Update tools.py

tools.py

@@ -37,7 +37,8 @@ class WikipediaTool(Tool):
     inputs = {"query": "string"}
     output_type = "string"
 
-    def __call__(self, query: str) -> str:
+    def __call__(self, inputs: dict) -> str:
+        query = inputs["query"]
         return fetch_wikipedia_article(query)
 
 class WikipediaSearchTool(Tool):
@@ -46,7 +47,8 @@ class WikipediaSearchTool(Tool):
     inputs = {"query": "string"}
     output_type = "string"
 
-    def __call__(self, query: str) -> str:
+    def __call__(self, inputs: dict) -> str:
+        query = inputs["query"]
         return search_wikipedia(query)