feat: Add RayMetricSpec to persistent resource

Google APIs · copybara-github · commit 5d1dd8e7686a · 2024-02-21T11:52:44.000-08:00
PiperOrigin-RevId: 609073560
diff --git a/google/cloud/aiplatform/v1beta1/persistent_resource.proto b/google/cloud/aiplatform/v1beta1/persistent_resource.proto
@@ -237,6 +237,9 @@ message RaySpec {
   // set.
   string head_node_resource_pool_id = 7
       [(google.api.field_behavior) = OPTIONAL];
+
+  // Optional. Ray metrics configurations.
+  RayMetricSpec ray_metric_spec = 8 [(google.api.field_behavior) = OPTIONAL];
 }
 
 // Persistent Cluster runtime information as output
@@ -274,3 +277,9 @@ message ServiceAccountSpec {
   // Required if any containers are specified in `ResourceRuntimeSpec`.
   string service_account = 2 [(google.api.field_behavior) = OPTIONAL];
 }
+
+// Configuration for the Ray metrics.
+message RayMetricSpec {
+  // Optional. Flag to disable the Ray metrics collection.
+  bool disabled = 1 [(google.api.field_behavior) = OPTIONAL];
+}
