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.
     
@@ -508,54 +508,3 @@ wikipedia_search_tool2 = Tool(
     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,
-)