Explorar o código

fix: tool image url response

Yeuoly hai 5 meses
pai
achega
eead6abe85
Modificáronse 1 ficheiros con 4 adicións e 6 borrados
  1. 4 6
      api/core/workflow/nodes/tool/tool_node.py

+ 4 - 6
api/core/workflow/nodes/tool/tool_node.py

@@ -178,14 +178,12 @@ class ToolNode(BaseNode[ToolNodeData]):
         for message in message_stream:
             if message.type in {ToolInvokeMessage.MessageType.IMAGE_LINK, ToolInvokeMessage.MessageType.IMAGE}:
                 assert isinstance(message.message, ToolInvokeMessage.TextMessage)
-                assert message.meta
 
                 url = message.message.text
-                ext = path.splitext(url)[1]
-                tool_file_id = str(url).split("/")[-1].split(".")[0]
-                mimetype = message.meta.get("mime_type", "image/jpeg")
-                filename = message.save_as or url.split("/")[-1]
-                transfer_method = message.meta.get("transfer_method", FileTransferMethod.TOOL_FILE)
+                if message.meta:
+                    transfer_method = message.meta.get("transfer_method", FileTransferMethod.TOOL_FILE)
+                else:
+                    transfer_method = FileTransferMethod.TOOL_FILE
 
                 with Session(db.engine) as session:
                     stmt = select(ToolFile).where(ToolFile.id == tool_file_id)