From 687455ca31faaced876765cf8e56606a9fa1eebc Mon Sep 17 00:00:00 2001 From: Yeuoly Date: Fri, 29 Nov 2024 14:09:34 +0800 Subject: [PATCH] fix: tool file id --- api/core/workflow/nodes/tool/tool_node.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/api/core/workflow/nodes/tool/tool_node.py b/api/core/workflow/nodes/tool/tool_node.py index bdc9cc9da6..79e0073df6 100644 --- a/api/core/workflow/nodes/tool/tool_node.py +++ b/api/core/workflow/nodes/tool/tool_node.py @@ -196,6 +196,8 @@ class ToolNode(BaseNode[ToolNodeData]): else: transfer_method = FileTransferMethod.TOOL_FILE + tool_file_id = str(url).split("/")[-1].split(".")[0] + with Session(db.engine) as session: stmt = select(ToolFile).where(ToolFile.id == tool_file_id) tool_file = session.scalar(stmt)