15308555518/ MaxKB
1
0
Fork 0
mirror of https://github.com/1Panel-dev/MaxKB.git synced 2025-12-26 01:33:05 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

fix: azure llm model not show some recommends 

--bug=1051028 --user=刘瑞斌 【模型管理】添加Azure OpenAI的大语言模型，基础模型的默认下拉列表模型信息错误 https://www.tapd.cn/57709429/s/1639196
This commit is contained in:
CaptainB 2024-12-30 17:44:00 +08:00
parent 1e56d7877f
commit 7f9a11dbb8
1 changed files with 16 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
apps/setting/models_provider/impl/azure_model_provider/azure_model_provider.py
View File

@ -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)