s

agent.py

@@ -69,19 +69,33 @@ def modulus(a: int, b: int) -> int:
     """
     return a % b
 
+@tool
+def wiki_search(query: str) -> str:
+    """Search Wikipedia for a query and return maximum 2 results.
+    
+    Args:
+        query: The search query."""
+    search_docs = WikipediaLoader(query=query, load_max_docs=2).load()
+    formatted_search_docs = "\n\n---\n\n".join(
+        [
+            f'<Document source="{doc.metadata["source"]}" page="{doc.metadata.get("page", "")}"/>\n{doc.page_content}\n</Document>'
+            for doc in search_docs
+        ])
+    return {"wiki_results": formatted_search_docs}
+
 # @tool
-# def wiki_search(query: str) -> str:
-#     """Search Wikipedia for a query and return maximum 2 results.
+# def web_search(query: str) -> str:
+#     """Search Tavily for a query and return maximum 3 results.
     
 #     Args:
 #         query: The search query."""
-#     search_docs = WikipediaLoader(query=query, load_max_docs=2).load()
+#     search_docs = TavilySearchResults(max_results=3).invoke(query=query)
 #     formatted_search_docs = "\n\n---\n\n".join(
 #         [
 #             f'<Document source="{doc.metadata["source"]}" page="{doc.metadata.get("page", "")}"/>\n{doc.page_content}\n</Document>'
 #             for doc in search_docs
 #         ])
-#     return {"wiki_results": formatted_search_docs}
+#     return {"web_results": formatted_search_docs}
 @tool
 def web_search(query: str) -> str:
     """Search Tavily for a query and return maximum 3 results.
@@ -122,20 +136,6 @@ def web_search(query: str) -> str:
         print(f"Error during Tavily search for query '{query}': {e}")
         return {"web_results": f"Error performing web search: {e}"}
 
-@tool
-def web_search(query: str) -> str:
-    """Search Tavily for a query and return maximum 3 results.
-    
-    Args:
-        query: The search query."""
-    search_docs = TavilySearchResults(max_results=3).invoke(query=query)
-    formatted_search_docs = "\n\n---\n\n".join(
-        [
-            f'<Document source="{doc.metadata["source"]}" page="{doc.metadata.get("page", "")}"/>\n{doc.page_content}\n</Document>'
-            for doc in search_docs
-        ])
-    return {"web_results": formatted_search_docs}
-
 @tool
 def arvix_search(query: str) -> str:
     """Search Arxiv for a query and return maximum 3 result.