From 7f9a11dbb835259c3e05cfb6f04cd09c3fe0a99b Mon Sep 17 00:00:00 2001 From: CaptainB Date: Mon, 30 Dec 2024 17:44:00 +0800 Subject: [PATCH] fix: azure llm model not show some recommends MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --bug=1051028 --user=刘瑞斌 【模型管理】添加Azure OpenAI的大语言模型，基础模型的默认下拉列表模型信息错误 https://www.tapd.cn/57709429/s/1639196 --- .../azure_model_provider.py | 21 ++++++++++++++----- 1 file changed, 16 insertions(+), 5 deletions(-) diff --git a/apps/setting/models_provider/impl/azure_model_provider/azure_model_provider.py b/apps/setting/models_provider/impl/azure_model_provider/azure_model_provider.py index 735ea7f90..111c4390b 100644 --- a/apps/setting/models_provider/impl/azure_model_provider/azure_model_provider.py +++ b/apps/setting/models_provider/impl/azure_model_provider/azure_model_provider.py @@ -32,9 +32,20 @@ base_azure_tti_model_credential = AzureOpenAITextToImageModelCredential() base_azure_tts_model_credential = AzureOpenAITTSModelCredential() base_azure_stt_model_credential = AzureOpenAISTTModelCredential() -default_model_info = ModelInfo('Azure OpenAI', '具体的基础模型由部署名决定', ModelTypeConst.LLM, - base_azure_llm_model_credential, AzureChatModel, api_version='2024-02-15-preview' - ) +default_model_info = [ + ModelInfo('Azure OpenAI', '具体的基础模型由部署名决定', ModelTypeConst.LLM, + base_azure_llm_model_credential, AzureChatModel, api_version='2024-02-15-preview' + ), + ModelInfo('gpt-4', '', ModelTypeConst.LLM, + base_azure_llm_model_credential, AzureChatModel, api_version='2024-02-15-preview' + ), + ModelInfo('gpt-4o', '', ModelTypeConst.LLM, + base_azure_llm_model_credential, AzureChatModel, api_version='2024-02-15-preview' + ), + ModelInfo('gpt-4o-mini', '', ModelTypeConst.LLM, + base_azure_llm_model_credential, AzureChatModel, api_version='2024-02-15-preview' + ), +] embedding_model_info = [ ModelInfo('text-embedding-3-large', '具体的基础模型由部署名决定', ModelTypeConst.EMBEDDING, @@ -77,8 +88,8 @@ stt_model_info = [ model_info_manage = ( ModelInfoManage.builder() - .append_default_model_info(default_model_info) - .append_model_info(default_model_info) + .append_default_model_info(default_model_info[0]) + .append_model_info_list(default_model_info) .append_model_info_list(embedding_model_info) .append_default_model_info(embedding_model_info[0]) .append_model_info_list(image_model_info)