feat: Support PSC-I ingress in Ray Client

PiperOrigin-RevId: 688719900
googleapis · Oct 22, 2024 · da76253 · da76253
1 parent f713417
commit da76253
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/google/cloud/aiplatform/vertex_ray/client_builder.py b/google/cloud/aiplatform/vertex_ray/client_builder.py
@@ -177,7 +177,10 @@ def connect(self) -> _VertexRayClientContext:
         public_address = self.response.resource_runtime.access_uris.get(
             "RAY_CLIENT_ENDPOINT"
         )
-        if public_address:
+        private_address = self.response.resource_runtime.access_uris.get(
+            "RAY_HEAD_NODE_INTERNAL_IP"
+        )
+        if public_address and not private_address:
             self._credentials = grpc.ssl_channel_credentials()
             bearer_token = _validation_utils.get_bearer_token()
             self._metadata = [