From 305fbc7c923bcf06a77201ec565c18a19c360d13 Mon Sep 17 00:00:00 2001 From: Kalo Chin <91766386+fdb02983rhy@users.noreply.github.com> Date: Mon, 18 Nov 2024 10:45:59 +0900 Subject: [PATCH] fix: fal ai wizper also return text msg (#10789) --- api/core/tools/provider/builtin/fal/tools/wizper.py | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) diff --git a/api/core/tools/provider/builtin/fal/tools/wizper.py b/api/core/tools/provider/builtin/fal/tools/wizper.py index 67a5e45118..ba05a62073 100644 --- a/api/core/tools/provider/builtin/fal/tools/wizper.py +++ b/api/core/tools/provider/builtin/fal/tools/wizper.py @@ -19,7 +19,7 @@ class WizperTool(BuiltinTool): version = tool_parameters.get("version", "3") if audio_file.type != FileType.AUDIO: - return [self.create_text_message("Not a valid audio file.")] + return self.create_text_message("Not a valid audio file.") api_key = self.runtime.credentials["fal_api_key"] @@ -31,9 +31,8 @@ class WizperTool(BuiltinTool): try: audio_url = fal_client.upload(file_data, mime_type) - except Exception as e: - return [self.create_text_message(f"Error uploading audio file: {str(e)}")] + return self.create_text_message(f"Error uploading audio file: {str(e)}") arguments = { "audio_url": audio_url, @@ -49,4 +48,9 @@ class WizperTool(BuiltinTool): with_logs=False, ) - return self.create_json_message(result) + json_message = self.create_json_message(result) + + text = result.get("text", "") + text_message = self.create_text_message(text) + + return [json_message, text_message]