manjuldube commited on
Commit
5704b57
·
verified ·
1 Parent(s): e0dce4a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +25 -25
app.py CHANGED
@@ -1,26 +1,26 @@
1
- import gradio as gr
2
- import os
3
-
4
- from smolagents import InferenceClientModel, CodeAgent, MCPClient
5
-
6
-
7
- try:
8
- mcp_client = MCPClient(
9
- {"url": "https://abidlabs-mcp-tool-http.hf.space/gradio_api/mcp/sse"}
10
- )
11
- tools = mcp_client.get_tools()
12
-
13
- model = InferenceClientModel(token=os.getenv("HUGGINGFACE_API_TOKEN"))
14
- agent = CodeAgent(tools=[*tools], model=model, additional_authorized_imports=["json", "ast", "urllib", "base64"])
15
-
16
- demo = gr.ChatInterface(
17
- fn=lambda message, history: str(agent.run(message)),
18
- type="messages",
19
- examples=["Analyze the sentiment of the following text 'This is awesome'"],
20
- title="Agent with MCP Tools",
21
- description="This is a simple agent that uses MCP tools to answer questions.",
22
- )
23
-
24
- demo.launch()
25
- finally:
26
  mcp_client.disconnect()
 
1
+ import gradio as gr
2
+ import os
3
+
4
+ from smolagents import InferenceClientModel, CodeAgent, MCPClient
5
+
6
+
7
+ try:
8
+ mcp_client = MCPClient(
9
+ {"url": "https://abidlabs-mcp-tool-http.hf.space/gradio_api/mcp/sse"}
10
+ )
11
+ tools = mcp_client.get_tools()
12
+
13
+ model = InferenceClientModel(token=os.getenv("HUGGINGFACE_API_TOKEN"))
14
+ agent = CodeAgent(tools=[*tools], model=model, additional_authorized_imports=["json", "ast", "urllib", "base64"])
15
+
16
+ demo = gr.ChatInterface(
17
+ fn=lambda message, history: str(agent.run(message)),
18
+ type="messages",
19
+ examples=["Analyze the sentiment of the following text 'This is awesome'"],
20
+ title="Agent with MCP Tools",
21
+ description="This is a simple agent that uses MCP tools to answer questions.",
22
+ )
23
+
24
+ demo.launch()
25
+ finally:
26
  mcp_client.disconnect()