Update tools.py

tools.py

@@ -35,8 +35,8 @@ 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)
+        self.inputs = {"query": "string"}
+        super().__init__(name=self.name, description=self.description, inputs=self.inputs)
 
     def __call__(self, query: str) -> str:
         return fetch_wikipedia_article(query)
@@ -45,8 +45,8 @@ 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)
+        self.inputs = {"query": "string"}
+        super().__init__(name=self.name, description=self.description, inputs=self.inputs)
 
     def __call__(self, query: str) -> str:
         return search_wikipedia(query)