Update tools.py

tools.py

@@ -1,28 +1,24 @@
 import base64
 import requests
-from smolagents import DuckDuckGoSearchTool
-from smolagents import Tool
+try:
+    from smolagents import DuckDuckGoSearchTool
+    from smolagents import Tool
+except ImportError:
+    DuckDuckGoSearchTool = object
+    Tool = object
 
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 # Tool to download and read file attachments (text) from the scoring API
-class FileReaderTool(Tool):
+# Plain class (no Tool inheritance) — called directly, not via CodeAgent
+class FileReaderTool:
     name = "file_reader"
-    description = "Downloads and reads a text file attachment from the scoring API given a task_id. Returns the file content as text. Use this tool when the question references an attached text file (e.g. .txt, .csv, .json)."
-    inputs = {
-        "task_id": {
-            "type": "string",
-            "description": "The task_id of the question whose file attachment to download."
-        }
-    }
-    output_type = "string"
 
     def __init__(self, api_url: str = DEFAULT_API_URL):
-        super().__init__()
         self.api_url = api_url
         print("FileReaderTool initialized.")
 
-    def forward(self, task_id: str) -> str:
+    def __call__(self, task_id: str) -> str:
         file_url = f"{self.api_url}/files/{task_id}"
         print(f"FileReaderTool downloading from: {file_url}")
         try:
@@ -38,23 +34,15 @@ class FileReaderTool(Tool):
 
 
 # Tool to download image attachments and return them as base64
-class ImageReaderTool(Tool):
+# Plain class (no Tool inheritance) — called directly, not via CodeAgent
+class ImageReaderTool:
     name = "image_reader"
-    description = "Downloads an image file attachment from the scoring API given a task_id. Returns the image as a base64-encoded string. Use this tool when the question references an attached image file (e.g. .png, .jpg, .jpeg, .gif, .webp)."
-    inputs = {
-        "task_id": {
-            "type": "string",
-            "description": "The task_id of the question whose image attachment to download."
-        }
-    }
-    output_type = "string"
 
     def __init__(self, api_url: str = DEFAULT_API_URL):
-        super().__init__()
         self.api_url = api_url
         print("ImageReaderTool initialized.")
 
-    def forward(self, task_id: str) -> str:
+    def __call__(self, task_id: str) -> str:
         file_url = f"{self.api_url}/files/{task_id}"
         print(f"ImageReaderTool downloading from: {file_url}")
         try:
@@ -85,9 +73,9 @@ class WebSearchTool( DuckDuckGoSearchTool):
     def __init__(self):
         super().__init__()
         print("WebSearchTool initialized.")
-    def __call__(self, query: str) -> str:
+    def forward(self, query: str) -> str:
         print(f"WebSearchTool received query (first 50 chars): {query[:50]}...")
-        result = super().__call__(query)
+        result = super().forward(query)
         print(f"WebSearchTool returning result (first 100 chars): {result[:100]}...")
         return result