From e7d6def1e8860329b5749de62bf593c9ad7df990 Mon Sep 17 00:00:00 2001 From: Yeuoly Date: Mon, 18 Mar 2024 19:59:54 +0800 Subject: [PATCH] fix: trim file extension --- api/core/workflow/nodes/tool/tool_node.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/api/core/workflow/nodes/tool/tool_node.py b/api/core/workflow/nodes/tool/tool_node.py index 816a173b34..4f5b3332ae 100644 --- a/api/core/workflow/nodes/tool/tool_node.py +++ b/api/core/workflow/nodes/tool/tool_node.py @@ -102,7 +102,7 @@ class ToolNode(BaseNode): filename = response.save_as or url.split('/')[-1] # get tool file id - tool_file_id = url.split('/')[-1] + tool_file_id = url.split('/')[-1].split('.')[0] result.append(FileVar( tenant_id=self.tenant_id, type=FileType.IMAGE, @@ -114,7 +114,7 @@ class ToolNode(BaseNode): )) elif response.type == ToolInvokeMessage.MessageType.BLOB: # get tool file id - tool_file_id = response.message.split('/')[-1] + tool_file_id = response.message.split('/')[-1].split('.')[0] result.append(FileVar( tenant_id=self.tenant_id, type=FileType.IMAGE,