Aymen Hassoun commited on
Commit
dc0d9a0
·
1 Parent(s): 0378d61

initial commit

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