ABVM commited on
Commit
efbf494
·
verified ·
1 Parent(s): 858f29d

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +3 -3
agent.py CHANGED
@@ -1,6 +1,6 @@
1
  from smolagents import CodeAgent, HfApiModel
2
- from smolagents import VisitWebpageTool, WebSearchTool, WikipediaSearchTool, FinalAnswerTool
3
- from calculator_tool import CalculatorTool
4
 
5
 
6
  model = HfApiModel(
@@ -22,7 +22,7 @@ agent = CodeAgent(
22
  VisitWebpageTool(),
23
  WebSearchTool(),
24
  WikipediaSearchTool(),
25
- CalculatorTool(),
26
  FinalAnswerTool(),
27
  ],
28
  max_steps=10,
 
1
  from smolagents import CodeAgent, HfApiModel
2
+ from smolagents import VisitWebpageTool, WebSearchTool, WikipediaSearchTool, PythonInterpreterTool,FinalAnswerTool
3
+
4
 
5
 
6
  model = HfApiModel(
 
22
  VisitWebpageTool(),
23
  WebSearchTool(),
24
  WikipediaSearchTool(),
25
+ PythonInterpreterTool(),
26
  FinalAnswerTool(),
27
  ],
28
  max_steps=10,