mcp-client / app.py
ttiaggoc's picture
Fix mcp server endpoint
6557e5b
raw
history blame contribute delete
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()