mcp-client / app.py
ttiaggoc's picture
Fix mcp server endpoint
6557e5b
raw
history blame
946 Bytes
import gradio as gr
import os
from mcp import StdioServerParameters
from smolagents import InferenceClientModel, CodeAgent, ToolCollection, MCPClient
mcp_client = None
try:
mcp_client = MCPClient(
{"url": "https://ttiaggoc-mcp-sentiment.hf.space/gradio_api/mcp/sse"} # This is the MCP Server we created in the previous section
)
tools = mcp_client.get_tools()
model = InferenceClientModel(token=os.getenv("HUGGINGFACE_API_TOKEN"))
agent = CodeAgent(tools=[*tools], model=model)
demo = gr.ChatInterface(
fn=lambda message, history: str(agent.run(message)),
type="messages",
examples=["I'm feeling great!"],
title="Sentiment Analysis Chat",
description="This is a simple agent that uses MCP tools to answer questions.",
)
#demo.launch(server_name="0.0.0.0", server_port=9060)
demo.launch()
finally:
if(mcp_client != None):
mcp_client.disconnect()