fix: azure llm model not show some recommends

--bug=1051028 --user=刘瑞斌【模型管理】添加Azure OpenAI的大语言模型，基础模型的默认下拉列表模型信息错误 https://www.tapd.cn/57709429/s/1639196
1Panel-dev · Dec 30, 2024 · 7f9a11d · 7f9a11d
1 parent 1e56d78
commit 7f9a11d
Showing 1 changed file with 16 additions and 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
@@ -32,9 +32,20 @@
 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 @@
 
 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)