sfc-gh-mcastro commited on
Commit
b863469
·
1 Parent(s): 2c6526e

Add application file

Browse files
Files changed (2) hide show
  1. requirements.txt +2 -0
  2. server.py +29 -0
requirements.txt ADDED
@@ -0,0 +1,2 @@
 
 
 
1
+ gradio[mcp]
2
+ smolagents[mcp]
server.py ADDED
@@ -0,0 +1,29 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+
3
+ from mcp.client.stdio import StdioServerParameters
4
+ from smolagents import InferenceClientModel, CodeAgent, ToolCollection
5
+ from smolagents.mcp_client import MCPClient
6
+
7
+ try:
8
+
9
+ mcp_client = MCPClient(
10
+ {"url": "http://localhost:7860/gradio_api/mcp/sse"}
11
+ )
12
+ tools = mcp_client.get_tools()
13
+
14
+ model = InferenceClientModel()
15
+ agent = CodeAgent(tools=[*tools], model=model)
16
+
17
+ demo = gr.ChatInterface(
18
+ fn=lambda message, history: str(agent.run(message)),
19
+ type="messages",
20
+ examples=["Prime factorization of 68"],
21
+ title="Agent with MCP Tools",
22
+ description="This is a simple agent that uses MCP tools to answer questions.",
23
+ messages=[],
24
+ )
25
+
26
+ demo.launch()
27
+
28
+ finally:
29
+ mcp_client.disconnect()