opt

tools/wikipedia_tool.py

@@ -7,23 +7,18 @@ wikipedia.set_lang("en")
 def wiki_search(query: str) -> str:
     """
     Search Wikipedia and return a short summary of the topic.
-
-    Args:
-        query (str): The search query.
-
-    Returns:
-        str: Summary text or error message.
     """
     try:
-        # Try to fetch a summary of the most relevant page
-        #return wikipedia.summary(query, sentences=3, auto_suggest=True, redirect=True)
-        try:
-            return wikipedia.summary(query, sentences=3)
-        except wikipedia.PageError:
-            search_results = wikipedia.search(query)
-            if search_results:
+        return wikipedia.summary(query, sentences=3)
+    except wikipedia.PageError:
+        search_results = wikipedia.search(query)
+        if search_results:
+            try:
                 return wikipedia.summary(search_results[0], sentences=3)
-            return "No relevant Wikipedia page found."
+            except Exception as inner:
+                return f"Wikipedia error while fetching fallback summary: {inner}"
+        return "No relevant Wikipedia page found."
     except wikipedia.DisambiguationError as e:
-        # If multiple results, return options
-        pass
+        return f"Disambiguation: {', '.join(e.options[:5])}"
+    except Exception as e:
+        return f"Wikipedia general error: {e}"