Upload agent.py

agent.py

@@ -80,34 +80,42 @@ def modulus(a: int, b: int) -> int:
 
 @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}
+    """Search Wikipedia for a query and return up to 5 results summarized."""
+    search_docs = WikipediaLoader(query=query, load_max_docs=5).load()
+    if not search_docs:
+        return "No Wikipedia results found."
+
+    summaries = []
+    for doc in search_docs:
+        content = doc.page_content
+        # Truncate if too long or summarize with a simple heuristic or model
+        content = content[:1000] + ("..." if len(content) > 1000 else "")
+        summaries.append(
+            f'<Document source="{doc.metadata["source"]}" page="{doc.metadata.get("page", "")}">\n{content}\n</Document>'
+        )
+    return {"wiki_results": "\n\n---\n\n".join(summaries)}
+
     
 
 
 @tool
 def web_search(query: str) -> str:
-    """Search Tavily for a query and return maximum 3 results.
-    
-    Args:
-        query: The search query."""
+    """Search Tavily for a query and return maximum 3 results."""
     search_docs = TavilySearch(max_results=3).invoke(query=query)
+    if not search_docs:
+        return "No web search results found."
+
     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>'
+            f'<Document source="{doc.metadata.get("source", "unknown")}" page="{doc.metadata.get("page", "")}"/>\n'
+            f'{doc.page_content[:1000]}'  # truncate to 1000 chars if needed
+            f'\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.
@@ -177,7 +185,7 @@ tools = [
 ]
 
 # Build graph function
-def build_graph(provider: str = "openai"):
+def build_graph(provider: str = "groq"):
     """Build the graph"""
     # Load environment variables from .env file
     if provider == "openai":