Updated tools.

tools/excel_analysis_tool.py

@@ -4,11 +4,14 @@ from huggingface_hub import hf_hub_download
 
 class ExcelAnalysisTool(Tool):
     def __init__(self):
-        self.name = "ExcelAnalysisTool"
-        self.description = (
-            "Loads an Excel file from the GAIA dataset on Hugging Face and calculates "
-            "the total sales for items labeled as 'food', excluding drinks. "
-            "Provide input as a string with the filename, e.g., 'sales_data.xlsx'."
+        super().__init__(
+            name="ExcelAnalysisTool",
+            description=(
+                "Loads an Excel file from the GAIA dataset on Hugging Face and calculates "
+                "the total sales for items labeled as 'food', excluding drinks. "
+                "Provide input as a string with the filename, e.g., 'sales_data.xlsx'."
+            ),
+            inputs=["filename"]
         )
         self.repo_id = "gaia-benchmark/GAIA"
 

tools/wiki_tool.py

@@ -3,12 +3,15 @@ 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'}"
+        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"]
         )
 
     def fetch_summary(self, topic: str) -> str:
@@ -22,7 +25,7 @@ class WikiTool(Tool):
         except Exception as e:
             return f"An unexpected error occurred: {str(e)}"
 
-    def is_historical_country(self, country_name: str) -> bool:
+    def is_historical_country(self, country_name: str) -> str:
         try:
             summary = wikipedia.summary(country_name, sentences=2).lower()
             keywords = [
@@ -30,11 +33,12 @@ class WikiTool(Tool):
                 "was a country", "defunct", "dissolved", "existed until",
                 "disestablished", "merged into"
             ]
-            return any(keyword in summary for keyword in keywords)
+            result = any(keyword in summary for keyword in keywords)
+            return "Yes" if result else "No"
         except Exception:
-            return False
+            return "Unknown (error during lookup)"
 
-    def __call__(self, args: dict):
+    def __call__(self, args: dict) -> str:
         action = args.get("action")
 
         if action == "summary":