ritikjain51 commited on
Commit
cb3b830
·
verified ·
1 Parent(s): 05d106b

Create app.py

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