From fb44cd87e78b939a4bb0c0e8dcccf8f181c285f2 Mon Sep 17 00:00:00 2001 From: Yeuoly Date: Fri, 29 Nov 2024 18:20:36 +0800 Subject: [PATCH] fix: image url message --- api/core/tools/utils/message_transformer.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/api/core/tools/utils/message_transformer.py b/api/core/tools/utils/message_transformer.py index 876f0486b3..8d436d7bed 100644 --- a/api/core/tools/utils/message_transformer.py +++ b/api/core/tools/utils/message_transformer.py @@ -25,7 +25,9 @@ class ToolFileMessageTransformer: for message in messages: if message.type in {ToolInvokeMessage.MessageType.TEXT, ToolInvokeMessage.MessageType.LINK}: yield message - elif message.type == ToolInvokeMessage.MessageType.IMAGE and isinstance(message.message, str): + elif message.type == ToolInvokeMessage.MessageType.IMAGE and isinstance( + message.message, ToolInvokeMessage.TextMessage + ): # try to download image try: if not conversation_id: @@ -49,11 +51,10 @@ class ToolFileMessageTransformer: meta=message.meta.copy() if message.meta is not None else {}, ) except Exception as e: - logger.exception(f"Failed to download image from {url}") yield ToolInvokeMessage( type=ToolInvokeMessage.MessageType.TEXT, message=ToolInvokeMessage.TextMessage( - text=f"Failed to download image: {message.message}, you can try to download it yourself." + text=f"Failed to download image: {message.message.text}: {e}" ), meta=message.meta.copy() if message.meta is not None else {}, save_as=message.save_as,