Skip to content

fix: refactor TencentEmbeddingModel to inherit from MaxKBBaseModel and remove unused video model references #6840

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
wxg0103 merged 1 commit into from
Sep 1, 2026
Merged

fix: refactor TencentEmbeddingModel to inherit from MaxKBBaseModel and remove unused video model references #6840

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 3 additions & 3 deletions apps/models_provider/impl/tencent_model_provider/model/embedding.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand All @@ -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,
Expand Down
33 changes: 0 additions & 33 deletions apps/models_provider/impl/tencent_model_provider/tencent_model_provider.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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 _

Expand Down Expand Up @@ -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)
Expand All @@ -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()
Expand Down
Loading