File size: 998 Bytes
7404322 6a73182 7404322 6a73182 7404322 2abf61d 7404322 6a73182 7404322 6a73182 7404322 6a73182 7404322 6a73182 7404322 6a73182 2abf61d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 |
import gradio as gr
from mcp.client.stdio import StdioServerParameters
from smolagents import InferenceClientModel, CodeAgent
from smolagents.mcp_client import MCPClient
# Initialize the MCP client correctly
try:
mcp_client = MCPClient(
## Try this working example on the hub:
# {"url": "https://abidlabs-mcp-tools.hf.space/gradio_api/mcp/sse"}
{"url": "http://localhost:7860/gradio_api/mcp/sse"}
)
tools = mcp_client.get_tools()
model = InferenceClientModel()
agent = CodeAgent(tools=[*tools], model=model)
# Define Gradio ChatInterface
demo = gr.ChatInterface(
fn=lambda message, history: str(agent.run(message)),
title="Agent with MCP Tools",
description="This is a simple agent that uses MCP tools to get chapters of the Quran.",
)
demo.launch()
finally:
# Properly close the MCP client connection
# if 'mcp_client' in locals():
# mcp_client.disconnect()
mcp_client.disconnect()
|