Refactor wiki_search function: replace WikipediaLoader with WikipediaQueryRun

tools/wikipedia.py

@@ -1,5 +1,6 @@
 from langchain_core.tools import tool
-from langchain_community.document_loaders import WikipediaLoader
+from langchain_community.tools import WikipediaQueryRun
+from langchain_community.utilities import WikipediaAPIWrapper
 
 @tool
 def wiki_search(query: str) -> str:
@@ -9,5 +10,6 @@ def wiki_search(query: str) -> str:
     Args:
         query (str): The search query to find relevant Wikipedia articles.
     """
-    loader = WikipediaLoader(query=query, load_max_docs=2).load()
-    return "\n---\n".join([f'<Document source="{doc.metadata["source"]}"/>\n{doc.page_content}\n</Document>' for doc in loader])
+    wikipedia = WikipediaQueryRun(api_wrapper=WikipediaAPIWrapper(top_k_results=1, doc_content_chars_max=32000)) # type: ignore
+    result = wikipedia.run(query)
+    return result