Upload 2 files

tools.py

@@ -17,6 +17,7 @@ class PythonRunTool(Tool):
     inputs = {
         "code": {"type": str, "description": "Python code to execute", "required": True}
     }
+    output_type = "str"
 
     def forward(self, code: str) -> str:
         buf, ns = io.StringIO(), {}
@@ -43,15 +44,19 @@ class ExcelLoaderTool(Tool):
             "type": str,
             "description": "Sheet name or index (optional, required for Excel files only)",
             "required": False,
-            "default": None,
+            "default": "",
         }
     }
+    output_type = "list"
 
     def forward(self, path: str, sheet: str | int | None = None) -> List[Dict[str, Any]]:
         import pandas as pd
         if not os.path.isfile(path):
             raise FileNotFoundError(path)
         ext = os.path.splitext(path)[1].lower()
+         # Handle empty string as None for sheet
+        if sheet == "":
+            sheet = None
         if ext == ".csv":
             df = pd.read_csv(path)
         else:
@@ -67,6 +72,7 @@ class YouTubeTranscriptTool(Tool):
         "url": {"type": str, "description": "YouTube URL", "required": True},
         "lang": {"type": str, "description": "Transcript language (default: en)", "required": False, "default": "en"}
     }
+    output_type = "str"
 
     def forward(self, url: str, lang: str = "en") -> str:
         from urllib.parse import urlparse, parse_qs
@@ -83,19 +89,16 @@ class AudioTranscriptionTool(Tool):
         "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 = "str"
 
     def forward(self, path: str, model: str = "whisper-1") -> str:
         import openai
         if not os.path.isfile(path):
             raise FileNotFoundError(path)
-        openai.api_key = os.getenv("OPENAI_API_KEY")
-        if not hasattr(openai, "Audio"):
-            raise ImportError(
-                "Your OpenAI package does not support Audio. "
-                "Please upgrade it with: pip install --upgrade openai"
-            )
+        client = openai.OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
         with open(path, "rb") as fp:
-            return openai.Audio.transcribe(model=model, file=fp)["text"].strip()
+            transcript = client.audio.transcriptions.create(model=model, file=fp)
+        return transcript.text.strip()
 
 # ---- 5. SimpleOCRTool ------------------------------------------------------
 class SimpleOCRTool(Tool):
@@ -104,6 +107,7 @@ class SimpleOCRTool(Tool):
     inputs = {
         "path": {"type": str, "description": "Path to image file", "required": True}
     }
+    output_type = "str"
 
     def forward(self, path: str) -> str:
         from PIL import Image