Spaces:
Sleeping
Sleeping
Update agent.py
Browse files
agent.py
CHANGED
|
@@ -206,10 +206,10 @@ class MagAgent:
|
|
| 206 |
self.tools = [
|
| 207 |
# RetryDuckDuckGoSearchTool(),
|
| 208 |
# WikipediaSearchTool(),
|
| 209 |
-
|
| 210 |
-
|
| 211 |
# VisitWebpageTool(),
|
| 212 |
-
|
| 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=
|
| 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
|