demo-mcp / app.py
knight7561's picture
Reverting try due to unavailability of client.close()
ccf3bb9 verified
raw
history blame contribute delete
847 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": "https://abidlabs-mcp-tools.hf.space/gradio_api/mcp/sse"}
)
# try:
tools = mcp_client.get_tools()
model = InferenceClientModel()
agent = CodeAgent(tools=[*tools], model=model)
def call_agent(message, history):
return str(agent.run(message))
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()
# finally:
# mcp_client.close()