From 5f51a19de220025fa4b4d3759c16834b9d7a9397 Mon Sep 17 00:00:00 2001 From: Yeuoly Date: Fri, 3 Jan 2025 14:48:19 +0800 Subject: [PATCH] fix: allow meta to be None --- api/core/tools/utils/message_transformer.py | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) diff --git a/api/core/tools/utils/message_transformer.py b/api/core/tools/utils/message_transformer.py index 3d19c14cd8..dad3a651c0 100644 --- a/api/core/tools/utils/message_transformer.py +++ b/api/core/tools/utils/message_transformer.py @@ -56,9 +56,9 @@ class ToolFileMessageTransformer: ) elif message.type == ToolInvokeMessage.MessageType.BLOB: # 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 not isinstance(message.message, ToolInvokeMessage.BlobMessage): @@ -81,17 +81,17 @@ class ToolFileMessageTransformer: yield ToolInvokeMessage( type=ToolInvokeMessage.MessageType.IMAGE_LINK, 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: yield ToolInvokeMessage( type=ToolInvokeMessage.MessageType.BINARY_LINK, 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: - assert message.meta is not None - file = message.meta.get("file") + meta = message.meta or {} + file = meta.get("file") if isinstance(file, File): if file.transfer_method == FileTransferMethod.TOOL_FILE: assert file.related_id is not None @@ -100,13 +100,13 @@ class ToolFileMessageTransformer: yield ToolInvokeMessage( type=ToolInvokeMessage.MessageType.IMAGE_LINK, 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: yield ToolInvokeMessage( type=ToolInvokeMessage.MessageType.LINK, 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: yield message