From 4bddd1aab1a5d554a679d83f4eaaf8aa4eeb14eb Mon Sep 17 00:00:00 2001 From: shaohuzhang1 <80892890+shaohuzhang1@users.noreply.github.com> Date: Tue, 7 Jan 2025 11:22:54 +0800 Subject: [PATCH] fix: tencent and xf embedding model set default model (#1985) --- .../impl/tencent_model_provider/tencent_model_provider.py | 2 +- .../models_provider/impl/xf_model_provider/xf_model_provider.py | 2 ++ 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/apps/setting/models_provider/impl/tencent_model_provider/tencent_model_provider.py b/apps/setting/models_provider/impl/tencent_model_provider/tencent_model_provider.py index 90edaabb0..c7402c07a 100644 --- a/apps/setting/models_provider/impl/tencent_model_provider/tencent_model_provider.py +++ b/apps/setting/models_provider/impl/tencent_model_provider/tencent_model_provider.py @@ -96,7 +96,6 @@ def _initialize_model_info(): TencentTTIModelCredential, TencentTextToImageModel)] - model_info_manage = ModelInfoManage.builder() \ .append_model_info_list(model_info_list) \ .append_model_info_list(model_info_embedding_list) \ @@ -105,6 +104,7 @@ def _initialize_model_info(): .append_model_info_list(model_info_tti_list) \ .append_default_model_info(model_info_tti_list[0]) \ .append_default_model_info(model_info_list[0]) \ + .append_default_model_info(tencent_embedding_model_info) \ .build() return model_info_manage diff --git a/apps/setting/models_provider/impl/xf_model_provider/xf_model_provider.py b/apps/setting/models_provider/impl/xf_model_provider/xf_model_provider.py index b72958424..c89456b04 100644 --- a/apps/setting/models_provider/impl/xf_model_provider/xf_model_provider.py +++ b/apps/setting/models_provider/impl/xf_model_provider/xf_model_provider.py @@ -50,6 +50,8 @@ model_info_manage = ( ) .append_default_model_info( ModelInfo('tts', '', ModelTypeConst.TTS, tts_model_credential, XFSparkTextToSpeech)) + .append_default_model_info( + ModelInfo('embedding', '', ModelTypeConst.EMBEDDING, embedding_model_credential, XFEmbedding)) .build() )