AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-14 06:15:53 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: fal ai wizper also return text msg (#10789)

Browse Source
This commit is contained in:
Kalo Chin 2024-11-18 10:45:59 +09:00 committed by GitHub
parent 29341d60aa
commit 305fbc7c92
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
api/core/tools/provider/builtin/fal/tools/wizper.py
View File

@ -19,7 +19,7 @@ class WizperTool(BuiltinTool):
version = tool_parameters.get("version", "3") version = tool_parameters.get("version", "3")
if audio_file.type != FileType.AUDIO: 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"] api_key = self.runtime.credentials["fal_api_key"]
@ -31,9 +31,8 @@ class WizperTool(BuiltinTool):
try: try:
audio_url = fal_client.upload(file_data, mime_type) audio_url = fal_client.upload(file_data, mime_type)
except Exception as e: 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 = { arguments = {
"audio_url": audio_url, "audio_url": audio_url,
@ -49,4 +48,9 @@ class WizperTool(BuiltinTool):
with_logs=False, 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]