Skip to content

Commit 4bddd1a

Browse files
authored
fix: tencent and xf embedding model set default model (#1985)
1 parent 54381ff commit 4bddd1a

File tree

2 files changed

+3
-1
lines changed

2 files changed

+3
-1
lines changed

apps/setting/models_provider/impl/tencent_model_provider/tencent_model_provider.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -96,7 +96,6 @@ def _initialize_model_info():
9696
TencentTTIModelCredential,
9797
TencentTextToImageModel)]
9898

99-
10099
model_info_manage = ModelInfoManage.builder() \
101100
.append_model_info_list(model_info_list) \
102101
.append_model_info_list(model_info_embedding_list) \
@@ -105,6 +104,7 @@ def _initialize_model_info():
105104
.append_model_info_list(model_info_tti_list) \
106105
.append_default_model_info(model_info_tti_list[0]) \
107106
.append_default_model_info(model_info_list[0]) \
107+
.append_default_model_info(tencent_embedding_model_info) \
108108
.build()
109109

110110
return model_info_manage

apps/setting/models_provider/impl/xf_model_provider/xf_model_provider.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -50,6 +50,8 @@
5050
)
5151
.append_default_model_info(
5252
ModelInfo('tts', '', ModelTypeConst.TTS, tts_model_credential, XFSparkTextToSpeech))
53+
.append_default_model_info(
54+
ModelInfo('embedding', '', ModelTypeConst.EMBEDDING, embedding_model_credential, XFEmbedding))
5355
.build()
5456
)
5557

0 commit comments

Comments
 (0)