Update agent.py

agent.py

@@ -169,6 +169,7 @@ class RetryDuckDuckGoSearchTool(DuckDuckGoSearchTool):
         wait=wait_exponential(multiplier=1, min=4, max=10),
         retry=retry_if_exception_type(Exception)
     )
+
     def forward(self, query: str) -> str:
         return super().forward(query)
 
@@ -198,16 +199,18 @@ class MagAgent:
             "openpyxl",
             "pathlib",
             "sys",
+            "bs4",
+            "arxiv",
         ]
 
         self.tools = [
 #            RetryDuckDuckGoSearchTool(),
 #            WikipediaSearchTool(),
-            SpeechToTextTool(),
-            ExcelReaderTool(),
+#            SpeechToTextTool(),
+#            ExcelReaderTool(),
 #            VisitWebpageTool(),
-            PythonCodeReaderTool(),
-            search_arxiv,
+#            PythonCodeReaderTool(),
+#            search_arxiv,
         ]
 
         self.prompt_template = (
@@ -240,6 +243,9 @@ class MagAgent:
                 WikipediaSearchTool(),
                 SpeechToTextTool(),
                 VisitWebpageTool(),
+                ExcelReaderTool(),
+                PythonCodeReaderTool(),
+                search_arxiv,
             ],
             model=model,
             max_steps=15,