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

fix: allow meta to be None

Browse Source
This commit is contained in:
Yeuoly 2025-01-03 14:48:19 +08:00
parent 71e0bfcbd8
commit 5f51a19de2
1 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
api/core/tools/utils/message_transformer.py
View File

@ -56,9 +56,9 @@ class ToolFileMessageTransformer:
) )
elif message.type == ToolInvokeMessage.MessageType.BLOB: elif message.type == ToolInvokeMessage.MessageType.BLOB:
# get mime type and save blob to storage # get mime type and save blob to storage
assert message.meta meta = message.meta or {}
mimetype = message.meta.get("mime_type", "octet/stream") mimetype = meta.get("mime_type", "octet/stream")
# if message is str, encode it to bytes # if message is str, encode it to bytes
if not isinstance(message.message, ToolInvokeMessage.BlobMessage): if not isinstance(message.message, ToolInvokeMessage.BlobMessage):
@ -81,17 +81,17 @@ class ToolFileMessageTransformer:
yield ToolInvokeMessage( yield ToolInvokeMessage(
type=ToolInvokeMessage.MessageType.IMAGE_LINK, type=ToolInvokeMessage.MessageType.IMAGE_LINK,
message=ToolInvokeMessage.TextMessage(text=url), message=ToolInvokeMessage.TextMessage(text=url),
meta=message.meta.copy() if message.meta is not None else {}, meta=meta.copy() if meta is not None else {},
) )
else: else:
yield ToolInvokeMessage( yield ToolInvokeMessage(
type=ToolInvokeMessage.MessageType.BINARY_LINK, type=ToolInvokeMessage.MessageType.BINARY_LINK,
message=ToolInvokeMessage.TextMessage(text=url), message=ToolInvokeMessage.TextMessage(text=url),
meta=message.meta.copy() if message.meta is not None else {}, meta=meta.copy() if meta is not None else {},
) )
elif message.type == ToolInvokeMessage.MessageType.FILE: elif message.type == ToolInvokeMessage.MessageType.FILE:
assert message.meta is not None meta = message.meta or {}
file = message.meta.get("file") file = meta.get("file")
if isinstance(file, File): if isinstance(file, File):
if file.transfer_method == FileTransferMethod.TOOL_FILE: if file.transfer_method == FileTransferMethod.TOOL_FILE:
assert file.related_id is not None assert file.related_id is not None
@ -100,13 +100,13 @@ class ToolFileMessageTransformer:
yield ToolInvokeMessage( yield ToolInvokeMessage(
type=ToolInvokeMessage.MessageType.IMAGE_LINK, type=ToolInvokeMessage.MessageType.IMAGE_LINK,
message=ToolInvokeMessage.TextMessage(text=url), message=ToolInvokeMessage.TextMessage(text=url),
meta=message.meta.copy() if message.meta is not None else {}, meta=meta.copy() if meta is not None else {},
) )
else: else:
yield ToolInvokeMessage( yield ToolInvokeMessage(
type=ToolInvokeMessage.MessageType.LINK, type=ToolInvokeMessage.MessageType.LINK,
message=ToolInvokeMessage.TextMessage(text=url), message=ToolInvokeMessage.TextMessage(text=url),
meta=message.meta.copy() if message.meta is not None else {}, meta=meta.copy() if meta is not None else {},
) )
else: else:
yield message yield message