jpuri commited on
Commit
562cf5d
·
1 Parent(s): b4a0202
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -1,4 +1,4 @@
1
- from smolagents import DuckDuckGoSearchTool, GradioUI, InferenceClientModel, Tool, ToolCallingAgent
2
 
3
 
4
  class AddTool(Tool):
@@ -27,9 +27,9 @@ class SubtractTool(Tool):
27
  return a - b
28
 
29
 
30
- agent = ToolCallingAgent(
31
  tools=[AddTool(), SubtractTool(), DuckDuckGoSearchTool()],
32
- model=InferenceClientModel(model_id="Qwen/Qwen2.5-72B-Instruct"),
33
  name="calculator_agent",
34
  description="An agent that can perform arithmetic (addition, subtraction) and search the web.",
35
  max_steps=10,
 
1
+ from smolagents import CodeAgent, DuckDuckGoSearchTool, GradioUI, InferenceClientModel, Tool
2
 
3
 
4
  class AddTool(Tool):
 
27
  return a - b
28
 
29
 
30
+ agent = CodeAgent(
31
  tools=[AddTool(), SubtractTool(), DuckDuckGoSearchTool()],
32
+ model=InferenceClientModel(model_id="Qwen/Qwen2.5-Coder-32B-Instruct"),
33
  name="calculator_agent",
34
  description="An agent that can perform arithmetic (addition, subtraction) and search the web.",
35
  max_steps=10,