Yeuoly месяцев назад: 5
Родитель
Сommit
687455ca31
1 измененных файлов с 2 добавлено и 0 удалено
  1. 2 0
      api/core/workflow/nodes/tool/tool_node.py

+ 2 - 0
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)