From f02a9cb45dd181fb05daa7d2564727c4f4ce1c9b Mon Sep 17 00:00:00 2001 From: LiangQuan Date: Fri, 16 Aug 2024 11:17:30 +0800 Subject: [PATCH] fix endpoint of pai-eas vpc network --- pai/api/client_factory.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/pai/api/client_factory.py b/pai/api/client_factory.py index 07d3dd1..fa76c84 100644 --- a/pai/api/client_factory.py +++ b/pai/api/client_factory.py @@ -87,7 +87,12 @@ def get_endpoint( raise ValueError("Please provide region_id to get the endpoint.") if network and network != Network.PUBLIC: - subdomain = f"{service_name}-{network.value.lower()}" + if service_name == "pai-eas": + # see endpoint list provided by PAI-EAS + # https://next.api.aliyun.com/product/eas + subdomain = f"pai-eas-manager-{network.value.lower()}" + else: + subdomain = f"{service_name}-{network.value.lower()}" else: subdomain = service_name return DEFAULT_SERVICE_ENDPOINT_PATTERN.format(subdomain, region_id)