Skip to content

Commit 7f9a11d

Browse files
committed
fix: azure llm model not show some recommends
--bug=1051028 --user=刘瑞斌 【模型管理】添加Azure OpenAI的大语言模型,基础模型的默认下拉列表模型信息错误 https://www.tapd.cn/57709429/s/1639196
1 parent 1e56d78 commit 7f9a11d

File tree

1 file changed

+16
-5
lines changed

1 file changed

+16
-5
lines changed

apps/setting/models_provider/impl/azure_model_provider/azure_model_provider.py

Lines changed: 16 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -32,9 +32,20 @@
3232
base_azure_tts_model_credential = AzureOpenAITTSModelCredential()
3333
base_azure_stt_model_credential = AzureOpenAISTTModelCredential()
3434

35-
default_model_info = ModelInfo('Azure OpenAI', '具体的基础模型由部署名决定', ModelTypeConst.LLM,
36-
base_azure_llm_model_credential, AzureChatModel, api_version='2024-02-15-preview'
37-
)
35+
default_model_info = [
36+
ModelInfo('Azure OpenAI', '具体的基础模型由部署名决定', ModelTypeConst.LLM,
37+
base_azure_llm_model_credential, AzureChatModel, api_version='2024-02-15-preview'
38+
),
39+
ModelInfo('gpt-4', '', ModelTypeConst.LLM,
40+
base_azure_llm_model_credential, AzureChatModel, api_version='2024-02-15-preview'
41+
),
42+
ModelInfo('gpt-4o', '', ModelTypeConst.LLM,
43+
base_azure_llm_model_credential, AzureChatModel, api_version='2024-02-15-preview'
44+
),
45+
ModelInfo('gpt-4o-mini', '', ModelTypeConst.LLM,
46+
base_azure_llm_model_credential, AzureChatModel, api_version='2024-02-15-preview'
47+
),
48+
]
3849

3950
embedding_model_info = [
4051
ModelInfo('text-embedding-3-large', '具体的基础模型由部署名决定', ModelTypeConst.EMBEDDING,
@@ -77,8 +88,8 @@
7788

7889
model_info_manage = (
7990
ModelInfoManage.builder()
80-
.append_default_model_info(default_model_info)
81-
.append_model_info(default_model_info)
91+
.append_default_model_info(default_model_info[0])
92+
.append_model_info_list(default_model_info)
8293
.append_model_info_list(embedding_model_info)
8394
.append_default_model_info(embedding_model_info[0])
8495
.append_model_info_list(image_model_info)

0 commit comments

Comments
 (0)