Artyom Boyko commited on
Commit
f85270d
·
1 Parent(s): cf8c5c1

Add simple MCP Client.

Browse files
Files changed (2) hide show
  1. app_client/app.py +3 -2
  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=llm, max_steps=20, planning_interval=1)
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==1.7.1
22
  fastmcp
 
18
  yt-dlp==2025.5.22
19
  smolagents
20
  smolagents[mcp]
21
+ mcp
22
  fastmcp