Spaces:
Runtime error
Runtime error
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() | |