Update tools.py

tools.py

@@ -31,6 +31,7 @@ def search_wikipedia(query: str) -> str:
         return "\n".join([r['title'] for r in results])
     return "No search results found."
 
+# ✅ Actual tools expecting dict inputs
 class WikipediaTool(Tool):
     name = "WikipediaTool"
     description = "Fetches plain text from a Wikipedia article. Input should be the article title."
@@ -38,8 +39,7 @@ class WikipediaTool(Tool):
     output_type = "string"
 
     def __call__(self, inputs: dict) -> str:
-        query = inputs["query"]
-        return fetch_wikipedia_article(query)
+        return fetch_wikipedia_article(inputs["query"])
 
 class WikipediaSearchTool(Tool):
     name = "WikipediaSearchTool"
@@ -48,7 +48,40 @@ class WikipediaSearchTool(Tool):
     output_type = "string"
 
     def __call__(self, inputs: dict) -> str:
-        query = inputs["query"]
-        return search_wikipedia(query)
+        return search_wikipedia(inputs["query"])
+
+# ✅ Wrapper Tools (these are called by CodeAgent)
+class WikipediaToolWrapper(Tool):
+    name = "WikipediaToolWrapper"
+    description = WikipediaTool.description
+    inputs = {"query": "string"}
+    output_type = "string"
+
+    def __init__(self):
+        self.tool = WikipediaTool()
+
+    def __call__(self, inputs: dict) -> str:
+        if isinstance(inputs, str):
+            inputs = {"query": inputs}
+        elif isinstance(inputs, dict) and "query" not in inputs:
+            inputs = {"query": next(iter(inputs.values()))}
+        return self.tool(inputs)
+
+class WikipediaSearchToolWrapper(Tool):
+    name = "WikipediaSearchToolWrapper"
+    description = WikipediaSearchTool.description
+    inputs = {"query": "string"}
+    output_type = "string"
+
+    def __init__(self):
+        self.tool = WikipediaSearchTool()
+
+    def __call__(self, inputs: dict) -> str:
+        if isinstance(inputs, str):
+            inputs = {"query": inputs}
+        elif isinstance(inputs, dict) and "query" not in inputs:
+            inputs = {"query": next(iter(inputs.values()))}
+        return self.tool(inputs)
+