Update tools.py

tools.py

@@ -33,20 +33,24 @@ def search_wikipedia(query: str) -> str:
 
 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": "string"}
-        super().__init__(name=self.name, description=self.description, inputs=self.inputs)
+        super().__init__(
+            name="WikipediaTool",
+            description="Fetches plain text from a Wikipedia article. Input should be the article title.",
+            inputs={"query": "string"},
+            output_type="string"
+        )
 
     def __call__(self, query: str) -> str:
         return fetch_wikipedia_article(query)
 
 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": "string"}
-        super().__init__(name=self.name, description=self.description, inputs=self.inputs)
+        super().__init__(
+            name="WikipediaSearchTool",
+            description="Searches Wikipedia for article titles related to a query. Input should be a question or topic.",
+            inputs={"query": "string"},
+            output_type="string"
+        )
 
     def __call__(self, query: str) -> str:
         return search_wikipedia(query)