zhuchaowe/maxkb
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
shaohuzhang1 2025-01-07 11:22:54 +08:00 committed by GitHub
parent 54381ffaf3
commit 4bddd1aab1
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
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, TencentTTIModelCredential,
TencentTextToImageModel)] TencentTextToImageModel)]
model_info_manage = ModelInfoManage.builder() \ model_info_manage = ModelInfoManage.builder() \
.append_model_info_list(model_info_list) \ .append_model_info_list(model_info_list) \
.append_model_info_list(model_info_embedding_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_model_info_list(model_info_tti_list) \
.append_default_model_info(model_info_tti_list[0]) \ .append_default_model_info(model_info_tti_list[0]) \
.append_default_model_info(model_info_list[0]) \ .append_default_model_info(model_info_list[0]) \
.append_default_model_info(tencent_embedding_model_info) \
.build() .build()
return model_info_manage 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( .append_default_model_info(
ModelInfo('tts', '', ModelTypeConst.TTS, tts_model_credential, XFSparkTextToSpeech)) ModelInfo('tts', '', ModelTypeConst.TTS, tts_model_credential, XFSparkTextToSpeech))
.append_default_model_info(
ModelInfo('embedding', '', ModelTypeConst.EMBEDDING, embedding_model_credential, XFEmbedding))
.build() .build()
) )