Austin/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2024-11-16 03:32:23 +08:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
呆萌闷油瓶 2024-03-14 09:14:27 +08:00 committed by GitHub
parent 1f92b55f58
commit e52448b84b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 18 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
api/core/model_runtime/model_providers/azure_openai/_common.py
View File

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

1
api/core/model_runtime/model_providers/azure_openai/_constant.py
View File

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

16
api/core/model_runtime/model_providers/azure_openai/azure_openai.yaml
View File

@ -46,6 +46,22 @@ model_credential_schema:
placeholder: placeholder:
zh_Hans: 在此输入您的 API Key zh_Hans: 在此输入您的 API Key
en_US: Enter your API key here 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 - variable: base_model_name
label: label:
en_US: Base Model en_US: Base Model