opensource/dify
2
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-04-28 20:17:29 +08:00
Code Issues Packages Projects Releases Wiki Activity

r2

Browse Source
This commit is contained in:
jyong 2025-07-04 19:09:40 +08:00
parent 1b3888a13e
commit 1fb59adba9
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
api/core/datasource/utils/message_transformer.py
View File

@ -1,6 +1,6 @@
import logging import logging
from collections.abc import Generator from collections.abc import Generator
from mimetypes import guess_extension from mimetypes import guess_extension, guess_type
from typing import Optional from typing import Optional
from core.datasource.datasource_file_manager import DatasourceFileManager from core.datasource.datasource_file_manager import DatasourceFileManager
@ -57,10 +57,13 @@ class DatasourceFileMessageTransformer:
elif message.type == DatasourceMessage.MessageType.BLOB: elif message.type == DatasourceMessage.MessageType.BLOB:
# get mime type and save blob to storage # get mime type and save blob to storage
meta = message.meta or {} meta = message.meta or {}
mimetype = meta.get("mime_type", "application/octet-stream")
# get filename from meta # get filename from meta
filename = meta.get("file_name", None) 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 message is str, encode it to bytes
if not isinstance(message.message, DatasourceMessage.BlobMessage): if not isinstance(message.message, DatasourceMessage.BlobMessage):