15308555518/ MaxKB
1
0
Fork 0
mirror of https://github.com/1Panel-dev/MaxKB.git synced 2025-12-26 10:12:51 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

fix: tencent and xf embedding model set default model (#1985) 

(cherry picked from commit 4bddd1aab1)
This commit is contained in:
shaohuzhang1 2025-01-07 11:22:54 +08:00 committed by shaohuzhang1
parent 1bfaf2f024
commit b4ffe656ad
2 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
apps/setting/models_provider/impl/tencent_model_provider/tencent_model_provider.py
View File

@ -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

2
apps/setting/models_provider/impl/xf_model_provider/xf_model_provider.py
View File

@ -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()
)