Update wikipedia_searcher.py

wikipedia_searcher.py

@@ -1,19 +1,26 @@
-
-from smolagents import Tool
-from langchain_community.tools import WikipediaQueryRun 
+from langchain_community.tools import WikipediaQueryRun
 from langchain_community.utilities import WikipediaAPIWrapper
 
-class WikipediaSearcher(Tool):
-    name = "wikipedia_search"
-    description = "Search Wikipedia for factual answers to questions."
-
-    def run(self, input_data: dict) -> str:
-        query = input_data.get("query", "")
-        if not query:
-            return "No query provided for Wikipedia search."
+class WikipediaSearcher:
+    """
+    A simple wrapper class to query Wikipedia and retrieve summaries using LangChain's Wikipedia utilities.
+    """
+
+    def __init__(self) -> None:
+        # Initialize the Wikipedia query runner with the API wrapper
+        self.wikipedia = WikipediaQueryRun(api_wrapper=WikipediaAPIWrapper())
+
+    def search(self, query: str) -> str:
+        """
+        Search Wikipedia for a given query and return a summary or snippet.
+        
+        Args:
+            query (str): The search query string.
+        Returns:
+            str: The Wikipedia summary or an error message.
+        """
         try:
-            wikipedia = WikipediaQueryRun(api_wrapper=WikipediaAPIWrapper())
-            return wikipedia.run(query)
+            return self.wikipedia.run(query)
         except Exception as e:
             return f"Error retrieving Wikipedia data: {e}"