SergeyO7 commited on
Commit
d8f9011
·
verified ·
1 Parent(s): b7a21e2

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +7 -7
agent.py CHANGED
@@ -206,10 +206,10 @@ class MagAgent:
206
  self.tools = [
207
  # RetryDuckDuckGoSearchTool(),
208
  # WikipediaSearchTool(),
209
- # SpeechToTextTool(),
210
- # ExcelReaderTool(),
211
  # VisitWebpageTool(),
212
- # PythonCodeReaderTool(),
213
  # search_arxiv,
214
  ]
215
 
@@ -241,10 +241,10 @@ class MagAgent:
241
  web_agent = ToolCallingAgent(
242
  tools=[RetryDuckDuckGoSearchTool(),
243
  WikipediaSearchTool(),
244
- SpeechToTextTool(),
245
  VisitWebpageTool(),
246
- ExcelReaderTool(),
247
- PythonCodeReaderTool(),
248
  search_arxiv,
249
  ],
250
  model=model,
@@ -257,7 +257,7 @@ class MagAgent:
257
  model=model,
258
  managed_agents=[web_agent],
259
  tools=self.tools,
260
- add_base_tools=True,
261
  additional_authorized_imports=self.imports,
262
  verbosity_level=2,
263
  max_steps=20
 
206
  self.tools = [
207
  # RetryDuckDuckGoSearchTool(),
208
  # WikipediaSearchTool(),
209
+ SpeechToTextTool(),
210
+ ExcelReaderTool(),
211
  # VisitWebpageTool(),
212
+ PythonCodeReaderTool(),
213
  # search_arxiv,
214
  ]
215
 
 
241
  web_agent = ToolCallingAgent(
242
  tools=[RetryDuckDuckGoSearchTool(),
243
  WikipediaSearchTool(),
244
+ # SpeechToTextTool(),
245
  VisitWebpageTool(),
246
+ # ExcelReaderTool(),
247
+ # PythonCodeReaderTool(),
248
  search_arxiv,
249
  ],
250
  model=model,
 
257
  model=model,
258
  managed_agents=[web_agent],
259
  tools=self.tools,
260
+ add_base_tools=False,
261
  additional_authorized_imports=self.imports,
262
  verbosity_level=2,
263
  max_steps=20