demo-mcp / app.py
knight7561's picture
Add init MCP file
5f75330
raw
history blame
709 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 = MCPClient(
{"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.",
messages=[],
)
demo.launch()