AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-19 12:39:12 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: tool image url response

Browse Source
This commit is contained in:
Yeuoly 2024-11-28 18:23:28 +08:00
parent 5c6d919a4a
commit eead6abe85
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)