Spaces:
Running
Running
Artyom Boyko
commited on
Commit
·
f85270d
1
Parent(s):
cf8c5c1
Add simple MCP Client.
Browse files- app_client/app.py +3 -2
- requirements.txt +1 -1
app_client/app.py
CHANGED
|
@@ -1,8 +1,9 @@
|
|
| 1 |
import gradio as gr
|
|
|
|
| 2 |
|
| 3 |
from mcp.client.stdio import StdioServerParameters
|
| 4 |
from smolagents import ToolCollection, CodeAgent
|
| 5 |
-
from smolagents import CodeAgent, ToolCallingAgent, InferenceClientModel
|
| 6 |
from smolagents.mcp_client import MCPClient
|
| 7 |
|
| 8 |
# model = InferenceClientModel()
|
|
@@ -19,7 +20,7 @@ try:
|
|
| 19 |
|
| 20 |
tools = mcp_client.get_tools()
|
| 21 |
# agent = ToolCallingAgent(tools=[*tools], model=model)
|
| 22 |
-
agent = ToolCallingAgent(tools=[*tools], model=
|
| 23 |
|
| 24 |
def call_agent(message, history):
|
| 25 |
return str(agent.run(message))
|
|
|
|
| 1 |
import gradio as gr
|
| 2 |
+
import torch
|
| 3 |
|
| 4 |
from mcp.client.stdio import StdioServerParameters
|
| 5 |
from smolagents import ToolCollection, CodeAgent
|
| 6 |
+
from smolagents import CodeAgent, ToolCallingAgent, InferenceClientModel, TransformersModel
|
| 7 |
from smolagents.mcp_client import MCPClient
|
| 8 |
|
| 9 |
# model = InferenceClientModel()
|
|
|
|
| 20 |
|
| 21 |
tools = mcp_client.get_tools()
|
| 22 |
# agent = ToolCallingAgent(tools=[*tools], model=model)
|
| 23 |
+
agent = ToolCallingAgent(tools=[*tools], model=model, max_steps=20, planning_interval=1)
|
| 24 |
|
| 25 |
def call_agent(message, history):
|
| 26 |
return str(agent.run(message))
|
requirements.txt
CHANGED
|
@@ -18,5 +18,5 @@ transformers==4.52.4
|
|
| 18 |
yt-dlp==2025.5.22
|
| 19 |
smolagents
|
| 20 |
smolagents[mcp]
|
| 21 |
-
mcp
|
| 22 |
fastmcp
|
|
|
|
| 18 |
yt-dlp==2025.5.22
|
| 19 |
smolagents
|
| 20 |
smolagents[mcp]
|
| 21 |
+
mcp
|
| 22 |
fastmcp
|