opensource
/
dify
mirror of https://github.com/langgenius/dify.git
2
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'fix/chore-fix' into dev/plugin-deploy

This commit is contained in:
Yeuoly 2024-11-28 18:24:19 +08:00
parent 6bd483030f eead6abe85
commit e79dcf93fe
No known key found for this signature in database
GPG Key ID: A66E7E320FB19F61
1 changed files with 4 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
api/core/workflow/nodes/tool/tool_node.py
View File

@ -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)