Update agent.py

agent.py

@@ -1,4 +1,4 @@
-from smolagents import CodeAgent, LiteLLMModel, tool, Tool, load_tool, DuckDuckGoSearchTool, WikipediaSearchTool
+from smolagents import CodeAgent, ToolCallingAgent, LiteLLMModel, tool, Tool, load_tool, DuckDuckGoSearchTool, WikipediaSearchTool
 import asyncio
 import os
 import re
@@ -201,11 +201,11 @@ class MagAgent:
         ]
 
         self.tools = [
-            RetryDuckDuckGoSearchTool(),
-            WikipediaSearchTool(),
+#            RetryDuckDuckGoSearchTool(),
+#            WikipediaSearchTool(),
             SpeechToTextTool(),
             ExcelReaderTool(),
-            VisitWebpageTool(),
+#            VisitWebpageTool(),
             PythonCodeReaderTool(),
             search_arxiv,
         ]
@@ -235,8 +235,21 @@ class MagAgent:
             """
         )
 
+        web_agent = ToolCallingAgent(
+            tools=[RetryDuckDuckGoSearchTool(),
+                WikipediaSearchTool(),
+                SpeechToTextTool(),
+                VisitWebpageTool(),
+            ],
+            model=model,
+            max_steps=15,
+            name="web_search_agent",
+            description="Runs web searches for you.",
+        )        
+        
         self.agent = CodeAgent(
             model=model,
+            managed_agents=[web_agent],
             tools=self.tools,
             add_base_tools=True,
             additional_authorized_imports=self.imports,