Update agent.py

agent.py

@@ -162,20 +162,25 @@ class DownloadTaskAttachmentTool(Tool):
         except requests.exceptions.RequestException as e:
             return f"Error downloading file for task {task_id}: {str(e)}"
 
-@tool
-def SpeechToTextTool(audio_path: str) -> str:
-    """Tool for converting an audio file to text using OpenAI Whisper.
-    Args:
-        audio_path (str): Path to audio file
-    Returns:
-        str: audio speech text
-    """
-    model = whisper.load_model("base")
+class SpeechToTextTool(Tool):
+    name = "speech_to_text"
+    description = (
+        "Converts an audio file to text using OpenAI Whisper."
+    )
+    inputs = {
+        "audio_path": {"type": "string", "description": "Path to audio file (.mp3, .wav)"},
+    }
+    output_type = "string"
+
+    def __init__(self):
+        super().__init__()
+        self.model = whisper.load_model("base")
 
-    if not os.path.exists(audio_path):
+    def forward(self, audio_path: str) -> str:
+        if not os.path.exists(audio_path):
             return f"Error: File not found at {audio_path}"
-    result = model.transcribe(audio_path)
-    return result.get("text", "")
+        result = self.model.transcribe(audio_path)
+        return result.get("text", "")
 
 class ExcelReaderTool(Tool):
     name = "excel_reader"