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模型如果是内部模型则使用内部api_version,反之使用用户的

This commit is contained in:
shaohuzhang1 2024-03-25 18:47:01 +08:00
parent cf003aa2d2
commit 436d0b7483
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -121,8 +121,8 @@ class AzureModelProvider(IModelProvider):
model_info: ModelInfo = model_dict.get(model_name)
azure_chat_open_ai = AzureChatOpenAI(
openai_api_base=model_credential.get('api_base'),
openai_api_version=model_credential.get(
'api_version') if 'api_version' in model_credential else model_info.api_version,
openai_api_version=model_info.api_version if model_name in model_dict else model_credential.get(
'api_version'),
deployment_name=model_credential.get('deployment_name'),
openai_api_key=model_credential.get('api_key'),
openai_api_type="azure"