Update tools.py

tools.py

@@ -2,9 +2,6 @@ from smolagents import Tool
 import requests
 
 def fetch_wikipedia_article(query: str) -> str:
-    """
-    Fetch the plain text of a Wikipedia article using the Wikipedia API.
-    """
     url = "https://en.wikipedia.org/w/api.php"
     params = {
         "action": "query",
@@ -20,9 +17,6 @@ def fetch_wikipedia_article(query: str) -> str:
     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.
-    """
     url = "https://en.wikipedia.org/w/api.php"
     params = {
         "action": "query",
@@ -41,6 +35,7 @@ class WikipediaTool(Tool):
     def __init__(self):
         self.name = "WikipediaTool"
         self.description = "Fetches plain text from a Wikipedia article. Input should be the article title."
+        self.inputs = ["query"]  # <- Add inputs attribute here
         super().__init__(name=self.name, description=self.description)
 
     def __call__(self, query: str) -> str:
@@ -50,6 +45,7 @@ class WikipediaSearchTool(Tool):
     def __init__(self):
         self.name = "WikipediaSearchTool"
         self.description = "Searches Wikipedia for article titles related to a query. Input should be a question or topic."
+        self.inputs = ["query"]  # <- Add inputs attribute here
         super().__init__(name=self.name, description=self.description)
 
     def __call__(self, query: str) -> str: