Update multiple_tools.py

multiple_tools.py

@@ -13,7 +13,9 @@ import os
 import chess
 import chess.engine
 import tempfile
+import wikipedia
 from PIL import Image
+import wikipedia
 #---------------------------------
 
 load_dotenv()
@@ -23,8 +25,37 @@ my_search_engine = os.getenv("Google_WebSearch_Engine")
 g_search = GoogleSearchToolSpec(key=google_key, engine=my_search_engine, num=3)
 
 #Wikipedia Search Tool
-wikipedia_tool = WikipediaToolSpec()
-wikipedia_search_tool = FunctionTool.from_defaults(wikipedia_tool.search_data)
+#wikipedia_tool = WikipediaToolSpec()
+#wikipedia_search_tool = FunctionTool.from_defaults(wikipedia_tool.search_data)
+
+wikipedia.set_lang("en")
+
+def wiki_search(query: str) -> str:
+    """
+    Safe Wikipedia summary tool with disambiguation and fallback protection.
+    """
+    try:
+        return wikipedia.summary(query, sentences=3)
+    except wikipedia.DisambiguationError as e:
+        # Try the first disambiguation option if available
+        if e.options:
+            try:
+                return wikipedia.summary(e.options[0], sentences=3)
+            except Exception as inner:
+                return f"Disambiguation fallback failed: {inner}"
+        return "Disambiguation error: No options available."
+    except wikipedia.PageError:
+        search_results = wikipedia.search(query)
+        if not search_results:
+            return "No relevant Wikipedia page found."
+        try:
+            return wikipedia.summary(search_results[0], sentences=3)
+        except Exception as inner:
+            return f"Wikipedia fallback summary error: {inner}"
+    except Exception as e:
+        return f"Wikipedia general error: {e}"
+wikipedia_search_tool = FunctionTool.from_defaults(wiki_search)
+
 
 def google_web_search(query : str) -> str:
     """
@@ -37,14 +68,11 @@ def google_web_search(query : str) -> str:
         str: The snippet of the first search result along with its source link.
     """
     result = g_search.google_search(query)
-    # data = result.content
-    if isinstance(result, str):
-        import ast
-        result = ast.literal_eval(result) 
-    first = result[0]
-    snippet = first["snippet"]
-    link = first["link"]
-    return f"Result: {snippet} Source: {link}"
+    output = result[0]
+    if "huggingface.co" in output["link"]:
+        output = result[1]
+    print(output)
+    return f"Result: {output["snippet"]} Source: {output["link"]}"
 
 google_web_search_tool = FunctionTool.from_defaults(google_web_search)