Fixed wiki

tools/wiki_tool.py

@@ -3,15 +3,36 @@ import wikipedia
 
 class WikiTool(Tool):
     def __init__(self):
+        self.name = "WikiTool"  # Явно задаем имя
+        self.description = (
+            "Performs Wikipedia lookups. Supports actions: 'summary' and 'is_historical_country'.\n"
+            "Usage:\n"
+            "- {'action': 'summary', 'topic': 'France'}\n"
+            "- {'action': 'is_historical_country', 'country_name': 'Yugoslavia'}"
+        )
+        self.inputs = {
+            "action": {
+                "type": "string",
+                "description": "The action to perform: 'summary' or 'is_historical_country'.",
+                "nullable": True  # Сделать action необязательным
+            },
+            "topic": {
+                "type": "string",
+                "description": "The topic to summarize (required for 'summary' action).",
+                "nullable": True
+            },
+            "country_name": {
+                "type": "string",
+                "description": "The country name to check (required for 'is_historical_country' action).",
+                "nullable": True
+            }
+        }
+        self.output_type = "string" 
         super().__init__(
-            name="WikiTool",
-            description=(
-                "Performs Wikipedia lookups. Supports actions: 'summary' and 'is_historical_country'.\n"
-                "Usage:\n"
-                "- {'action': 'summary', 'topic': 'France'}\n"
-                "- {'action': 'is_historical_country', 'country_name': 'Yugoslavia'}"
-            ),
-            inputs=["action", "topic", "country_name"]
+            name=self.name,
+            description=self.description,
+            inputs=self.inputs,
+            output_type=self.output_type
         )
 
     def fetch_summary(self, topic: str) -> str:
@@ -38,12 +59,10 @@ class WikiTool(Tool):
         except Exception:
             return "Unknown (error during lookup)"
 
-    def __call__(self, args: dict) -> str:
-        action = args.get("action")
-
+    def forward(self, action: str = "", topic: str = "", country_name: str = "") -> str:
         if action == "summary":
-            return self.fetch_summary(args.get("topic", ""))
+            return self.fetch_summary(topic)
         elif action == "is_historical_country":
-            return self.is_historical_country(args.get("country_name", ""))
+            return self.is_historical_country(country_name)
         else:
-            return "Error: Unknown action. Use 'summary' or 'is_historical_country'."
+            return "Error: Unknown action. Use 'summary' or 'is_historical_country'."