Skip to content

Commit 79fce87

Browse files
committed
fix: add video format detection for uploaded files in image.py
1 parent 6af51a4 commit 79fce87

File tree

1 file changed

+12
-2
lines changed
  • apps/models_provider/impl/volcanic_engine_model_provider/model

1 file changed

+12
-2
lines changed

apps/models_provider/impl/volcanic_engine_model_provider/model/image.py

Lines changed: 12 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -25,9 +25,19 @@ def new_instance(model_type, model_name, model_credential: Dict[str, object], **
2525
def is_cache_model():
2626
return False
2727

28-
2928
def upload_file_and_get_url(self, file_stream, file_name):
3029
"""上传文件并获取文件URL"""
3130
base64_video = base64.b64encode(file_stream).decode("utf-8")
32-
video_format = mimetypes.guess_type(file_name)[0]
31+
video_format = get_video_format(file_name)
3332
return f'data:{video_format};base64,{base64_video}'
33+
34+
35+
def get_video_format(file_name):
36+
extension = file_name.split('.')[-1].lower()
37+
format_map = {
38+
'mp4': 'video/mp4',
39+
'avi': 'video/avi',
40+
'mov': 'video/mov',
41+
'wmv': 'video/x-ms-wmv'
42+
}
43+
return format_map.get(extension, 'video/mp4')

0 commit comments

Comments
 (0)