소스 검색

fix: tool file id

Yeuoly 5 달 전
부모
커밋
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)