MCP_client / app.py
zihaddataimage's picture
Update app.py
9970ed4 verified
Raw
History Blame Contribute Delete
882 Bytes
import gradio as gr
from mcp.client.stdio import StdioServerParameters
from smolagents import ToolCollection, CodeAgent
from smolagents import CodeAgent, InferenceClientModel
from smolagents.mcp_client import MCPClient
mcp_client = None
try:
mcp_client = MCPClient(
{"url": "https://mcp.so/server/smolagents"}
#{"url": "http://localhost:7860/gradio_api/mcp/sse"}
)
tools = mcp_client.get_tools()
model = InferenceClientModel()
agent = CodeAgent(tools=[*tools], model=model)
demo = gr.ChatInterface(
fn=lambda message, history: str(agent.run(message)),
type="messages",
examples=["Prime factorization of 68"],
title="Agent with MCP Tools",
description="This is a simple agent that uses MCP tools to answer questions.",
)
demo.launch()
finally:
if MCPClient:
mcp_client.close()