fix endpoint of pai-eas vpc network

aliyun · Aug 16, 2024 · 690916f · 690916f
1 parent c4bece7
commit 690916f
Showing 1 changed file with 6 additions and 1 deletion.
diff --git 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-manage-{network.value.lower()}"
+            else:
+                subdomain = f"{service_name}-{network.value.lower()}"
         else:
             subdomain = service_name
         return DEFAULT_SERVICE_ENDPOINT_PATTERN.format(subdomain, region_id)