Upload 4 files

tools.py

@@ -10,14 +10,13 @@ from smolagents import Tool
 # ---- 1. PythonRunTool ------------------------------------------------------
 class PythonRunTool(Tool):
     name = "python_run"
-    description = (
-        "Execute trusted Python code and return printed output "
-        "+ repr() of the last expression (or _result variable)."
-    )
+    description = """
+        Execute trusted Python code and return printed output + repr() of the last expression (or _result variable).
+    """
     inputs = {
-        "code": {"type": "string", "description": "Python code to execute", "required": True}
+        "code": {"type": str, "description": "Python code to execute", "required": True}
     }
-    output_type = "string"
+    output_type = "str"
 
     def forward(self, code: str) -> str:
         buf, ns = io.StringIO(), {}
@@ -34,14 +33,13 @@ class PythonRunTool(Tool):
 # ---- 2. ExcelLoaderTool ----------------------------------------------------
 class ExcelLoaderTool(Tool):
     name = "load_spreadsheet"
-    description = (
-        "Read .xlsx/.xls/.csv from disk and return "
-        "rows as a list of dictionaries with string keys."
-    )
+    description = """
+        Read .xlsx/.xls/.csv from disk and return rows as a list of dictionaries with string keys.
+    """
     inputs = {
-        "path": {"type": "string", "description": "Path to .csv/.xls/.xlsx file", "required": True},
+        "path": {"type": str, "description": "Path to .csv/.xls/.xlsx file", "required": True},
         "sheet": {
-            "type": "string",
+            "type": str,
             "description": "Sheet name or index (optional, required for Excel files only)",
             "required": False,
             "default": "",
@@ -67,12 +65,14 @@ class ExcelLoaderTool(Tool):
 # ---- 3. YouTubeTranscriptTool ---------------------------------------------
 class YouTubeTranscriptTool(Tool):
     name = "youtube_transcript"
-    description = "Return the subtitles of a YouTube URL using youtube-transcript-api."
+    description = """
+        Return the subtitles of a YouTube URL using youtube-transcript-api.
+    """
     inputs = {
-        "url": {"type": "string", "description": "YouTube URL", "required": True},
-        "lang": {"type": "string", "description": "Transcript language (default: en)", "required": False, "default": "en"}
+        "url": {"type": str, "description": "YouTube URL", "required": True},
+        "lang": {"type": str, "description": "Transcript language (default: en)", "required": False, "default": "en"}
     }
-    output_type = "string"
+    output_type = "str"
 
     def forward(self, url: str, lang: str = "en") -> str:
         from urllib.parse import urlparse, parse_qs
@@ -84,12 +84,14 @@ class YouTubeTranscriptTool(Tool):
 # ---- 4. AudioTranscriptionTool --------------------------------------------
 class AudioTranscriptionTool(Tool):
     name = "transcribe_audio"
-    description = "Transcribe an audio file with OpenAI Whisper, returns plain text."
+    description = """
+        Transcribe an audio file with OpenAI Whisper, returns plain text."
+    """
     inputs = {
-        "path": {"type": "string", "description": "Path to audio file", "required": True},
-        "model": {"type": "string", "description": "Model name for transcription (default: whisper-1)", "required": False, "default": "whisper-1"}
+        "path": {"type": str, "description": "Path to audio file", "required": True},
+        "model": {"type": str, "description": "Model name for transcription (default: whisper-1)", "required": False, "default": "whisper-1"}
     }
-    output_type = "string"
+    output_type = "str"
 
     def forward(self, path: str, model: str = "whisper-1") -> str:
         import openai
@@ -103,11 +105,13 @@ class AudioTranscriptionTool(Tool):
 # ---- 5. SimpleOCRTool ------------------------------------------------------
 class SimpleOCRTool(Tool):
     name = "image_ocr"
-    description = "Return any text spotted in an image via pytesseract OCR."
+    description = """
+        Return any text spotted in an image via pytesseract OCR.
+    """
     inputs = {
-        "path": {"type": "string", "description": "Path to image file", "required": True}
+        "path": {"type": str, "description": "Path to image file", "required": True}
     }
-    output_type = "string"
+    output_type = "str"
 
     def forward(self, path: str) -> str:
         from PIL import Image