JDhruv14 commited on
Commit
ba8f320
·
verified ·
1 Parent(s): 9bac57b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +28 -28
app.py CHANGED
@@ -1,29 +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
-
8
- try:
9
- mcp_client = MCPClient(
10
- ## Try this working example on the hub:
11
- # {"url": "https://abidlabs-mcp-tools.hf.space/gradio_api/mcp/sse"}
12
- {"url": "http://localhost:7860/gradio_api/mcp/sse"}
13
- )
14
- tools = mcp_client.get_tools()
15
-
16
- model = InferenceClientModel()
17
- agent = CodeAgent(tools=[*tools], model=model)
18
-
19
- demo = gr.ChatInterface(
20
- fn=lambda message, history: str(agent.run(message)),
21
- type="messages",
22
- examples=["Prime factorization of 68"],
23
- title="Agent with MCP Tools",
24
- description="This is a simple agent that uses MCP tools to answer questions.",
25
- )
26
-
27
- demo.launch()
28
- finally:
29
  mcp_client.disconnect()
 
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
+
8
+ try:
9
+ mcp_client = MCPClient(
10
+ ## Try this working example on the hub:
11
+ # {"url": "https://abidlabs-mcp-tools.hf.space/gradio_api/mcp/sse"}
12
+ {"url": "http://localhost:7860/gradio_api/mcp/sse"}
13
+ )
14
+ tools = mcp_client.get_tools()
15
+
16
+ model = InferenceClientModel()
17
+ agent = CodeAgent(tools=[*tools], model=model)
18
+
19
+ demo = gr.ChatInterface(
20
+ fn=lambda message, history: str(agent.run(message)),
21
+ type="messages",
22
+ examples=["Prime factorization of 68"],
23
+ title="Agent with MCP Tools",
24
+ description="This is a simple agent that uses MCP tools to answer questions.",
25
+ )
26
+
27
+ demo.launch()
28
+ finally:
29
  mcp_client.disconnect()