From 79fce87e9370cfc487fa9a350cfcc4e854aeb401 Mon Sep 17 00:00:00 2001 From: wxg0103 <727495428@qq.com> Date: Tue, 28 Oct 2025 15:54:56 +0800 Subject: [PATCH] fix: add video format detection for uploaded files in image.py --- .../volcanic_engine_model_provider/model/image.py | 14 ++++++++++++-- 1 file changed, 12 insertions(+), 2 deletions(-) diff --git a/apps/models_provider/impl/volcanic_engine_model_provider/model/image.py b/apps/models_provider/impl/volcanic_engine_model_provider/model/image.py index 87b5702e7..810a0b8ea 100644 --- a/apps/models_provider/impl/volcanic_engine_model_provider/model/image.py +++ b/apps/models_provider/impl/volcanic_engine_model_provider/model/image.py @@ -25,9 +25,19 @@ class VolcanicEngineImage(MaxKBBaseModel, BaseChatOpenAI): def is_cache_model(): return False - def upload_file_and_get_url(self, file_stream, file_name): """上传文件并获取文件URL""" base64_video = base64.b64encode(file_stream).decode("utf-8") - video_format = mimetypes.guess_type(file_name)[0] + video_format = get_video_format(file_name) return f'data:{video_format};base64,{base64_video}' + + +def get_video_format(file_name): + extension = file_name.split('.')[-1].lower() + format_map = { + 'mp4': 'video/mp4', + 'avi': 'video/avi', + 'mov': 'video/mov', + 'wmv': 'video/x-ms-wmv' + } + return format_map.get(extension, 'video/mp4')