feat: Add RayMetricSpec to persistent resource

PiperOrigin-RevId: 609073560
googleapis · Feb 21, 2024 · 5d1dd8e · 5d1dd8e
1 parent 64497b3
commit 5d1dd8e
Showing 1 changed file with 9 additions and 0 deletions.
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];
+}