[Client] Implement runtime_context().gcs_address (ray-project#36895)

Fixes gcs_address for RuntimeContexts on Ray Client.

Signed-off-by: e428265 <arvind.chandramouli@lmco.com>

python/ray/tests/test_client.py

@@ -902,6 +902,15 @@ def get_value(self):
         assert ray.get(deserialized.get_value.remote()) == 1234
 
 
+def test_get_runtime_context_gcs_client(call_ray_start_shared):
+    """
+    Tests get_runtime_context gcs_client
+    """
+    with ray_start_client_server_for_address(call_ray_start_shared) as ray:
+        context = ray.get_runtime_context()
+        assert context.gcs_address, "gcs_address not set"
+
+
 if __name__ == "__main__":
     if os.environ.get("PARALLEL_CI"):
         sys.exit(pytest.main(["-n", "auto", "--boxed", "-vs", __file__]))

python/ray/util/client/__init__.py

@@ -17,7 +17,7 @@
 
 # This version string is incremented to indicate breaking changes in the
 # protocol that require upgrading the client version.
-CURRENT_PROTOCOL_VERSION = "2022-12-06"
+CURRENT_PROTOCOL_VERSION = "2023-06-27"
 
 
 class _ClientContext:

python/ray/util/client/runtime_context.py

@@ -1,4 +1,5 @@
 from typing import TYPE_CHECKING
+from types import SimpleNamespace
 
 if TYPE_CHECKING:
     from ray import JobID, NodeID
@@ -55,3 +56,10 @@ def should_capture_child_tasks_in_placement_group(self) -> bool:
     @property
     def runtime_env(self) -> str:
         return self._fetch_runtime_context().runtime_env
+
+    def check_connected(self) -> bool:
+        return self.worker.ping_server()
+
+    @property
+    def gcs_client(self) -> str:
+        return SimpleNamespace(address=self._fetch_runtime_context().gcs_address)

python/ray/util/client/server/server.py

@@ -268,6 +268,7 @@ def ClusterInfo(self, request, context=None) -> ray_client_pb2.ClusterInfoRespon
                 ctx.capture_client_tasks = (
                     rtc.should_capture_child_tasks_in_placement_group
                 )
+                ctx.gcs_address = rtc.gcs_address
                 ctx.runtime_env = rtc.get_runtime_env_string()
             resp.runtime_context.CopyFrom(ctx)
         else:

src/ray/protobuf/ray_client.proto

@@ -208,6 +208,7 @@ message ClusterInfoResponse {
     string namespace = 3;
     string runtime_env = 4;
     bool capture_client_tasks = 5;
+    string gcs_address = 6;
   }
 
   ClusterInfoType.TypeEnum type = 1;