mirror of
https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git
synced 2025-08-14 00:35:56 +08:00
chore: update default endpoint for ark provider (#7741)
This commit is contained in:
parent
693fe912f2
commit
7cfebffbb8
@ -32,6 +32,9 @@ from core.model_runtime.entities.message_entities import (
|
||||
UserPromptMessage,
|
||||
)
|
||||
|
||||
DEFAULT_V2_ENDPOINT = "maas-api.ml-platform-cn-beijing.volces.com"
|
||||
DEFAULT_V3_ENDPOINT = "https://ark.cn-beijing.volces.com/api/v3"
|
||||
|
||||
|
||||
class ArkClientV3:
|
||||
endpoint_id: Optional[str] = None
|
||||
@ -43,16 +46,24 @@ class ArkClientV3:
|
||||
|
||||
@staticmethod
|
||||
def is_legacy(credentials: dict) -> bool:
|
||||
# match default v2 endpoint
|
||||
if ArkClientV3.is_compatible_with_legacy(credentials):
|
||||
return False
|
||||
sdk_version = credentials.get("sdk_version", "v2")
|
||||
return sdk_version != "v3"
|
||||
# match default v3 endpoint
|
||||
if credentials.get("api_endpoint_host") == DEFAULT_V3_ENDPOINT:
|
||||
return False
|
||||
# only v3 support api_key
|
||||
if credentials.get("auth_method") == "api_key":
|
||||
return False
|
||||
# these cases are considered as sdk v2
|
||||
# - modified default v2 endpoint
|
||||
# - modified default v3 endpoint and auth without api_key
|
||||
return True
|
||||
|
||||
@staticmethod
|
||||
def is_compatible_with_legacy(credentials: dict) -> bool:
|
||||
sdk_version = credentials.get("sdk_version")
|
||||
endpoint = credentials.get("api_endpoint_host")
|
||||
return sdk_version is None and endpoint == "maas-api.ml-platform-cn-beijing.volces.com"
|
||||
return endpoint == DEFAULT_V2_ENDPOINT
|
||||
|
||||
@classmethod
|
||||
def from_credentials(cls, credentials):
|
||||
@ -64,7 +75,7 @@ class ArkClientV3:
|
||||
"sk": credentials['volc_secret_access_key'],
|
||||
}
|
||||
if cls.is_compatible_with_legacy(credentials):
|
||||
args["base_url"] = "https://ark.cn-beijing.volces.com/api/v3"
|
||||
args["base_url"] = DEFAULT_V3_ENDPOINT
|
||||
|
||||
client = ArkClientV3(
|
||||
**args
|
||||
|
@ -64,7 +64,7 @@ model_credential_schema:
|
||||
en_US: API Endpoint Host
|
||||
zh_Hans: API Endpoint Host
|
||||
type: text-input
|
||||
default: maas-api.ml-platform-cn-beijing.volces.com
|
||||
default: https://ark.cn-beijing.volces.com/api/v3
|
||||
placeholder:
|
||||
en_US: Enter your API Endpoint Host
|
||||
zh_Hans: 输入 API Endpoint Host
|
||||
|
Loading…
x
Reference in New Issue
Block a user