From 436d0b7483d31aa9d96827c0f90bfa2d3ab9ae02 Mon Sep 17 00:00:00 2001
From: shaohuzhang1 <shaohu.zhang@fit2cloud.com>
Date: Mon, 25 Mar 2024 18:47:01 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20azure=E6=A8=A1=E5=9E=8B=E5=A6=82?=
 =?UTF-8?q?=E6=9E=9C=E6=98=AF=E5=86=85=E9=83=A8=E6=A8=A1=E5=9E=8B=E5=88=99?=
 =?UTF-8?q?=E4=BD=BF=E7=94=A8=E5=86=85=E9=83=A8api=5Fversion,=E5=8F=8D?=
 =?UTF-8?q?=E4=B9=8B=E4=BD=BF=E7=94=A8=E7=94=A8=E6=88=B7=E7=9A=84?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../impl/azure_model_provider/azure_model_provider.py         | 4 ++--
 1 file changed, 2 insertions(+), 2 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 bfaf91308..35775d866 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
@@ -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"