From 1fb59adba9057915f7bf098b7def2195f10cd7fb Mon Sep 17 00:00:00 2001 From: jyong <718720800@qq.com> Date: Fri, 4 Jul 2025 19:09:40 +0800 Subject: [PATCH] r2 --- api/core/datasource/utils/message_transformer.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/api/core/datasource/utils/message_transformer.py b/api/core/datasource/utils/message_transformer.py index 9bc57235d8..12d2a71d69 100644 --- a/api/core/datasource/utils/message_transformer.py +++ b/api/core/datasource/utils/message_transformer.py @@ -1,6 +1,6 @@ import logging from collections.abc import Generator -from mimetypes import guess_extension +from mimetypes import guess_extension, guess_type from typing import Optional from core.datasource.datasource_file_manager import DatasourceFileManager @@ -57,10 +57,13 @@ class DatasourceFileMessageTransformer: elif message.type == DatasourceMessage.MessageType.BLOB: # get mime type and save blob to storage meta = message.meta or {} - - mimetype = meta.get("mime_type", "application/octet-stream") # get filename from meta filename = meta.get("file_name", None) + + mimetype = meta.get("mime_type") + if not mimetype: + mimetype = guess_type(filename)[0] or "application/octet-stream" + # if message is str, encode it to bytes if not isinstance(message.message, DatasourceMessage.BlobMessage):