From 545a948ba91f1ab80374ddf4671e9f877c418ca0 Mon Sep 17 00:00:00 2001 From: wxg0103 <727495428@qq.com> Date: Tue, 1 Sep 2026 14:56:17 +0800 Subject: [PATCH] fix: refactor TencentEmbeddingModel to inherit from MaxKBBaseModel and remove unused video model references --- .../tencent_model_provider/model/embedding.py | 6 ++-- .../tencent_model_provider.py | 33 ------------------- 2 files changed, 3 insertions(+), 36 deletions(-) diff --git a/apps/models_provider/impl/tencent_model_provider/model/embedding.py b/apps/models_provider/impl/tencent_model_provider/model/embedding.py index 8357a30a76a..02e143f8a13 100644 --- a/apps/models_provider/impl/tencent_model_provider/model/embedding.py +++ b/apps/models_provider/impl/tencent_model_provider/model/embedding.py @@ -5,10 +5,10 @@ import requests from common.utils.logger import maxkb_logger -from models_provider.base_model_provider import MaxKBBaseEmbeddingModel +from models_provider.base_model_provider import MaxKBBaseModel -class TencentEmbeddingModel(MaxKBBaseEmbeddingModel): +class TencentEmbeddingModel(MaxKBBaseModel): """腾讯 TokenHub 向量模型（OpenAI Embeddings 兼容接口）。 文本向量：POST /v1/embeddings @@ -30,7 +30,7 @@ def is_cache_model(): @staticmethod def new_instance(model_type: str, model_name: str, model_credential: Dict[str, object], **model_kwargs): - optional_params = MaxKBBaseEmbeddingModel.filter_optional_params(model_kwargs) + optional_params = MaxKBBaseModel.filter_optional_params(model_kwargs) return TencentEmbeddingModel( api_key=model_credential.get("api_key"), model_name=model_name, diff --git a/apps/models_provider/impl/tencent_model_provider/tencent_model_provider.py b/apps/models_provider/impl/tencent_model_provider/tencent_model_provider.py index 336a47dc079..2c3318049e3 100644 --- a/apps/models_provider/impl/tencent_model_provider/tencent_model_provider.py +++ b/apps/models_provider/impl/tencent_model_provider/tencent_model_provider.py @@ -16,13 +16,11 @@ from models_provider.impl.tencent_model_provider.credential.stt import TencentSTTModelCredential from models_provider.impl.tencent_model_provider.credential.tokenhub_stt import TencentTokenhubSTTModelCredential from models_provider.impl.tencent_model_provider.credential.tti import TencentTTIModelCredential -from models_provider.impl.tencent_model_provider.credential.ttv import TencentTTVModelCredential from models_provider.impl.tencent_model_provider.model.embedding import TencentEmbeddingModel from models_provider.impl.tencent_model_provider.model.image import TencentVision from models_provider.impl.tencent_model_provider.model.llm import TencentModel from models_provider.impl.tencent_model_provider.model.stt import TencentSpeechToText, TencentWandSpeechToText from models_provider.impl.tencent_model_provider.model.tti import TencentTextToImageModel -from models_provider.impl.tencent_model_provider.model.ttv import TencentVideoModel from maxkb.conf import PROJECT_DIR from django.utils.translation import gettext as _ @@ -174,33 +172,6 @@ def _initialize_model_info(): ) ] - model_info_ttv_list = [ - _create_model_info( - "hy-video-1.5", - _("Hunyuan HY-Video 1.5 text-to-video model."), - ModelTypeConst.TTV, - TencentTTVModelCredential, - TencentVideoModel, - ) - ] - - model_info_itv_list = [ - _create_model_info( - "hy-video-1.5", - _("Hunyuan HY-Video 1.5 image-to-video model."), - ModelTypeConst.ITV, - TencentTTVModelCredential, - TencentVideoModel, - ), - _create_model_info( - "yt-video-2.0", - _("Tencent YT-Video 2.0 image-to-video model."), - ModelTypeConst.ITV, - TencentTTVModelCredential, - TencentVideoModel, - ), - ] - model_info_manage = ( ModelInfoManage.builder() .append_model_info_list(model_info_list) @@ -209,10 +180,6 @@ def _initialize_model_info(): .append_default_model_info(model_info_vision_list[0]) .append_model_info_list(model_info_tti_list) .append_default_model_info(model_info_tti_list[0]) - .append_model_info_list(model_info_ttv_list) - .append_default_model_info(model_info_ttv_list[0]) - .append_model_info_list(model_info_itv_list) - .append_default_model_info(model_info_itv_list[0]) .append_default_model_info(model_info_list[0]) .append_default_model_info(tencent_embedding_model_info) .build()