mcp-client / app.py
root
Initial commit
a6e221b
import os
import gradio as gr
from mcp import StdioServerParameters
from smolagents import InferenceClientModel, CodeAgent, ToolCollection, MCPClient
try:
server_parameters = {
"url": "https://abidlabs-mcp-tools2.hf.space/gradio_api/mcp/sse", # https://jonnhan-mcp-sentiment.hf.space/gradio_api/mcp/sse
"transport": "sse",
}
mcp_client = MCPClient(server_parameters)
tools = mcp_client.get_tools()
model = InferenceClientModel(token=os.getenv("HF_TOKEN"))
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:
mcp_client.disconnect()