Browse Source

feat:add api-version selection for azure openai APIs (#2821)

呆萌闷油瓶 1 năm trước cách đây
mục cha
commit
e52448b84b

+ 2 - 1
api/core/model_runtime/model_providers/azure_openai/_common.py

@@ -15,10 +15,11 @@ from core.model_runtime.model_providers.azure_openai._constant import AZURE_OPEN
 class _CommonAzureOpenAI:
     @staticmethod
     def _to_credential_kwargs(credentials: dict) -> dict:
+        api_version = credentials.get('openai_api_version', AZURE_OPENAI_API_VERSION)
         credentials_kwargs = {
             "api_key": credentials['openai_api_key'],
             "azure_endpoint": credentials['openai_api_base'],
-            "api_version": AZURE_OPENAI_API_VERSION,
+            "api_version": api_version,
             "timeout": Timeout(315.0, read=300.0, write=10.0, connect=5.0),
             "max_retries": 1,
         }

+ 0 - 1
api/core/model_runtime/model_providers/azure_openai/_constant.py

@@ -16,7 +16,6 @@ from core.model_runtime.entities.model_entities import (
 
 AZURE_OPENAI_API_VERSION = '2024-02-15-preview'
 
-
 def _get_max_tokens(default: int, min_val: int, max_val: int) -> ParameterRule:
     rule = ParameterRule(
         name='max_tokens',

+ 16 - 0
api/core/model_runtime/model_providers/azure_openai/azure_openai.yaml

@@ -46,6 +46,22 @@ model_credential_schema:
       placeholder:
         zh_Hans: 在此输入您的 API Key
         en_US: Enter your API key here
+    - variable: openai_api_version
+      label:
+        zh_Hans: API 版本
+        en_US: API Version
+      type: select
+      required: true
+      options:
+        - label:
+            en_US: 2024-02-15-preview
+          value: 2024-02-15-preview
+        - label:
+            en_US: 2023-12-01-preview
+          value: 2023-12-01-preview
+      placeholder:
+        zh_Hans: 在此选择您的 API 版本
+        en_US: Select your API Version here
     - variable: base_model_name
       label:
         en_US: Base Model