Update tools.py

tools.py

@@ -1,3 +1,4 @@
+from smolagents import Tool
 import requests
 
 def fetch_wikipedia_article(query: str) -> str:
@@ -18,7 +19,6 @@ def fetch_wikipedia_article(query: str) -> str:
     page = next(iter(pages.values()))
     return page.get("extract", "No content found for this query.")
 
-
 def search_wikipedia(query: str) -> str:
     """
     Search for a Wikipedia article title using the Wikipedia API.
@@ -34,19 +34,23 @@ def search_wikipedia(query: str) -> str:
     data = response.json()
     results = data.get("query", {}).get("search", [])
     if results:
-        return "\n".join([f"{r['title']}" for r in results])
+        return "\n".join([r['title'] for r in results])
     return "No search results found."
 
-
-class WikipediaTool:
+class WikipediaTool(Tool):
     description = "Fetches plain text from a Wikipedia article. Input should be the article title."
 
+    def __init__(self):
+        super().__init__(name="WikipediaTool", description=self.description)
+
     def __call__(self, query: str) -> str:
         return fetch_wikipedia_article(query)
 
-
-class WikipediaSearchTool:
+class WikipediaSearchTool(Tool):
     description = "Searches Wikipedia for article titles related to a query. Input should be a question or topic."
 
+    def __init__(self):
+        super().__init__(name="WikipediaSearchTool", description=self.description)
+
     def __call__(self, query: str) -> str:
         return search_wikipedia(query)