From eead6abe858637c5dadf17e5c67c3f84e36dc4f7 Mon Sep 17 00:00:00 2001 From: Yeuoly Date: Thu, 28 Nov 2024 18:23:28 +0800 Subject: [PATCH] fix: tool image url response --- api/core/workflow/nodes/tool/tool_node.py | 10 ++++------ 1 file changed, 4 insertions(+), 6 deletions(-) diff --git a/api/core/workflow/nodes/tool/tool_node.py b/api/core/workflow/nodes/tool/tool_node.py index bcf92074f7..7f97558763 100644 --- a/api/core/workflow/nodes/tool/tool_node.py +++ b/api/core/workflow/nodes/tool/tool_node.py @@ -178,14 +178,12 @@ class ToolNode(BaseNode[ToolNodeData]): for message in message_stream: if message.type in {ToolInvokeMessage.MessageType.IMAGE_LINK, ToolInvokeMessage.MessageType.IMAGE}: assert isinstance(message.message, ToolInvokeMessage.TextMessage) - assert message.meta url = message.message.text - ext = path.splitext(url)[1] - tool_file_id = str(url).split("/")[-1].split(".")[0] - mimetype = message.meta.get("mime_type", "image/jpeg") - filename = message.save_as or url.split("/")[-1] - transfer_method = message.meta.get("transfer_method", FileTransferMethod.TOOL_FILE) + if message.meta: + transfer_method = message.meta.get("transfer_method", FileTransferMethod.TOOL_FILE) + else: + transfer_method = FileTransferMethod.TOOL_FILE with Session(db.engine) as session: stmt = select(ToolFile).where(ToolFile.id == tool_file_id)