Update langgraph_agent.py

langgraph_agent.py

@@ -40,12 +40,20 @@ def modulus(a: int, b: int) -> int:
 @tool
 def wiki_search(query: str) -> dict:
     """Search Wikipedia for a query and return up to 2 documents."""
-    docs = WikipediaLoader(query=query, load_max_docs=2).load()
-    formatted = "\n\n---\n\n".join(
-        f'<Document source="{d.metadata["source"]}"/>\n{d.page_content}'
-        for d in docs
-    )
-    return {"wiki_results": formatted}
+    try:
+        docs = WikipediaLoader(query=query, load_max_docs=2, lang="en").load() # Added lang="en" for clarity
+        if not docs:
+            return {"wiki_results": f"No documents found on Wikipedia for '{query}'."}
+        formatted = "\n\n---\n\n".join(
+            f'<Document source="{d.metadata.get("source", "N/A")}"/>\n{d.page_content}' # Added .get for safety
+            for d in docs
+        )
+        return {"wiki_results": formatted}
+    except Exception as e:
+        # Log the full error for debugging if possible
+        print(f"Error in wiki_search tool: {e}")
+        return {"wiki_results": f"Error occurred while searching Wikipedia for '{query}'. Details: {str(e)}"}
+
 
 @tool
 def web_search(query: str) -> dict: