Update wikipedia_searcher.py

wikipedia_searcher.py

@@ -1,30 +1,23 @@
-from langchain_community.tools import WikipediaQueryRun
+from smolagents import Tool
+from langchain_community.tools import WikipediaQueryRun 
 from langchain_community.utilities import WikipediaAPIWrapper
 
-class WikipediaSearcher:
-    """
-    A simple wrapper class to query Wikipedia and retrieve summaries using LangChain's Wikipedia utilities.
-    """
+class WikipediaSearcher(Tool):
+    name = "wikipedia_search"
+    description = "Search Wikipedia for factual answers to questions."
 
-    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.
-        """
+    def run(self, input_data: dict) -> str:
+        query = input_data.get("query", "")
+        if not query:
+            return "No query provided for Wikipedia search."
         try:
-            result = self.wikipedia.run(query)
-            return result
+            wikipedia = WikipediaQueryRun(api_wrapper=WikipediaAPIWrapper())
+            return wikipedia.run(query)
         except Exception as e:
             return f"Error retrieving Wikipedia data: {e}"
 
 
 
+
+
+