Update helper.py

helper.py

@@ -486,7 +486,7 @@ gemini_multimodal_tool = Tool(
     func=analyze_image_with_gemini,
 )
 
-
+'''
 def wiki_search(query: str) -> str:
     """Search Wikipedia for a query and return maximum 2 results.
     
@@ -507,4 +507,55 @@ wikipedia_search_tool2 = Tool(
     name="wikipedia_search_tool2",
     description=wiki_search.__doc__,
     func=wiki_search,
+)
+'''
+
+from langchain_community.document_loaders import WikipediaLoader
+from langchain_core.tools import Tool
+from typing import Optional, List
+
+def wiki_search(
+    query: str,
+    load_max_docs: int = 3,
+    auto_suggest: bool = False, # New parameter
+    lang: str = "en" # New parameter
+) -> str:
+    """Search Wikipedia for a query and return a specified maximum number of results.
+    
+    Args:
+        query: The search query.
+        load_max_docs: The maximum number of documents to load. Default is 2.
+        auto_suggest: If True, Wikipedia's auto-suggestion feature will be used to refine the query.
+                      This can lead to more consistent results for slightly misspelled or ambiguous queries.
+        lang: The language of the Wikipedia to search (e.g., "en" for English). Default is "en".
+    Returns:
+        A string with formatted Wikipedia search results.
+    """
+    try:
+        # Use auto_suggest to let Wikipedia refine the query
+        search_docs = WikipediaLoader(
+            query=query,
+            load_max_docs=load_max_docs,
+            auto_suggest=auto_suggest,
+            lang=lang
+        ).load()
+        
+        if not search_docs:
+            return "No relevant Wikipedia results found."
+
+        formatted_search_docs = "\n\n---\n\n".join(
+            [
+                f'<Document source="{doc.metadata.get("source", "")}" page="{doc.metadata.get("page", "")}"/>\n{doc.page_content}\n</Document>'
+                for doc in search_docs
+            ])
+        return formatted_search_docs
+    except Exception as e:
+        # Catching potential Wikipedia API errors (e.g., query too ambiguous)
+        return f"An error occurred during Wikipedia search: {e}"
+
+
+wikipedia_search_tool2 = Tool(
+    name="wikipedia_search_tool2",
+    description="Search Wikipedia for a query and return a specified maximum number of results. Can use auto-suggestion and specify language.",
+    func=wiki_search,
 )