Update agent.py

agent.py

@@ -21,8 +21,7 @@ import logging
 logger = logging.getLogger(__name__)
 
 @tool
-def search_arxiv(query
-: str) -> str:
+def search_arxiv(query: str) -> str:
     """Search Arxiv for a query and return maximum 3 result.
     
     Args:
@@ -30,6 +29,7 @@ def search_arxiv(query
      Returns:
         str: Formatted search results
     """
+
     search_docs = ArxivLoader(query=query, load_max_docs=3).load()
     formatted_search_docs = "\n\n---\n\n".join(
         [
@@ -189,7 +189,7 @@ class MagAgent:
             WikipediaSearchTool(),
             SpeechToTextTool(),
             ExcelReaderTool(),
-#            VisitWebpageTool(),
+            VisitWebpageTool(),
             PythonCodeReaderTool(),
             search_arxiv,
         ]