Create tools.py

tools.py

@@ -0,0 +1,93 @@
+import base64
+import requests
+from smolagents import DuckDuckGoSearchTool
+from smolagents import Tool
+
+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):
+    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 __call__(self, task_id: str) -> str:
+        file_url = f"{self.api_url}/files/{task_id}"
+        print(f"FileReaderTool downloading from: {file_url}")
+        try:
+            resp = requests.get(file_url, timeout=30)
+            resp.raise_for_status()
+            content = resp.text
+            print(f"FileReaderTool downloaded {len(content)} chars.")
+            return content
+        except requests.exceptions.RequestException as e:
+            error_msg = f"Failed to download file for task {task_id}: {e}"
+            print(error_msg)
+            return error_msg
+
+
+# Tool to download image attachments and return them as base64
+class ImageReaderTool(Tool):
+    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 __call__(self, task_id: str) -> str:
+        file_url = f"{self.api_url}/files/{task_id}"
+        print(f"ImageReaderTool downloading from: {file_url}")
+        try:
+            resp = requests.get(file_url, timeout=30)
+            resp.raise_for_status()
+            # Detect content type for the base64 data URI
+            content_type = resp.headers.get("Content-Type", "image/png")
+            image_b64 = base64.b64encode(resp.content).decode("utf-8")
+            print(f"ImageReaderTool downloaded image ({len(resp.content)} bytes, {content_type}).")
+            return f"data:{content_type};base64,{image_b64}"
+        except requests.exceptions.RequestException as e:
+            error_msg = f"Failed to download image for task {task_id}: {e}"
+            print(error_msg)
+            return error_msg
+
+
+#search web tool
+class WebSearchTool(Tool, DuckDuckGoSearchTool):
+    name = "web_search"
+    description = "Searches the web for information based on a query string."
+    inputs = {
+        "query": {
+            "type": "string",
+            "description": "The query string to search for."
+        }
+    }
+    output_type = "string"
+    def __init__(self):
+        super().__init__()
+        print("WebSearchTool initialized.")
+    def __call__(self, query: str) -> str:
+        print(f"WebSearchTool received query (first 50 chars): {query[:50]}...")
+        result = super().__call__(query)
+        print(f"WebSearchTool returning result (first 100 chars): {result[:100]}...")
+        return result
+