Update wikipedia_searcher.py

wikipedia_searcher.py

@@ -1,24 +1,23 @@
+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.
-    """
-
-    def __init__(self) -> None:
-        # Initialize the Wikipedia query runner with the API wrapper
+class WikipediaSearcher(Tool):
+    name = "wikipedia_search"
+    description = "Search Wikipedia and return a summary for a given query."
+    inputs = {
+        "query": {
+            "type": "string",
+            "description": "Search term for Wikipedia"
+        }
+    }
+    output_type = "string"
+
+    def __init__(self):
+        super().__init__()
         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 forward(self, query: str) -> str:
         try:
             return self.wikipedia.run(query)
         except Exception as e:
@@ -28,4 +27,3 @@ class WikipediaSearcher:
 
 
 
-