test_mcp_client / app.py
imessam's picture
token
4c76701
raw
history blame contribute delete
812 Bytes
import gradio as gr
import os
from smolagents import InferenceClientModel, CodeAgent, MCPClient
try:
mcp_client = MCPClient(
{"url": "https://imessam-test-mcp-sentiment.hf.space/gradio_api/mcp/sse", "transport": "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=["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()