fix: tool file id

2024-11-29 14:09:34 +08:00 · 2024-11-29 14:09:34 +08:00 · 687455ca31
parent 8c5928da2f
commit 687455ca31
1 changed files with 2 additions and 0 deletions
--- 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)