Refactor wiki_search function: replace WikipediaQueryRun with wikipediaapi for full page content retrieval

tools/wikipedia.py

@@ -1,15 +1,26 @@
 from langchain_core.tools import tool
-from langchain_community.tools import WikipediaQueryRun
-from langchain_community.utilities import WikipediaAPIWrapper
+import wikipediaapi
 
 @tool
 def wiki_search(query: str) -> str:
     """
-    Search Wikipedia for a given query and return the summary.
+    Search Wikipedia for a given query and return the full page content.
 
     Args:
         query (str): The search query to find relevant Wikipedia articles.
     """
-    wikipedia = WikipediaQueryRun(api_wrapper=WikipediaAPIWrapper(top_k_results=1, doc_content_chars_max=32000)) # type: ignore
-    result = wikipedia.run(query)
-    return result
+    # Initialize Wikipedia API with a proper user agent
+    wiki_wiki = wikipediaapi.Wikipedia(
+        user_agent='OracleBot/0.1.0 (https://github.com/abtsousa/oraclebot)',
+        language='en',
+    )
+    
+    # Get the page
+    page = wiki_wiki.page(query)
+    
+    # Check if page exists
+    if not page.exists():
+        return f"No Wikipedia page found for '{query}'. Please try a different search term."
+    
+    # Return the full text content (summary + all sections)
+    return f"Title: {page.title}\n\nURL: {page.fullurl}\n\n{page.text}"