diff --git a/apps/models_provider/impl/aliyun_bai_lian_model_provider/model/image.py b/apps/models_provider/impl/aliyun_bai_lian_model_provider/model/image.py index ddb90a5b6..e09245c74 100644 --- a/apps/models_provider/impl/aliyun_bai_lian_model_provider/model/image.py +++ b/apps/models_provider/impl/aliyun_bai_lian_model_provider/model/image.py @@ -86,7 +86,6 @@ class QwenVLChatModel(MaxKBBaseModel, BaseChatOpenAI): policy_data = self.get_upload_policy(self.openai_api_key.get_secret_value(), self.model_name) # 2. 上传文件到OSS oss_url = self.upload_file_to_oss(policy_data, file_stream, file_name) - print(oss_url) return oss_url 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 2d71b3681..87b5702e7 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 @@ -1,3 +1,5 @@ +import base64 +import mimetypes from typing import Dict from models_provider.base_model_provider import MaxKBBaseModel @@ -22,3 +24,10 @@ class VolcanicEngineImage(MaxKBBaseModel, BaseChatOpenAI): @staticmethod 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] + return f'data:{video_format};base64,{base64_video}'