feat: add Scaled tier for RagEngineConfig to v1beta, equivalent to Enterprise

Google APIs · copybara-github · commit 67a660f1bdc1 · 2025-06-16T15:10:05.000-07:00
feat: add Unprovisioned tier to RagEngineConfig in v1beta1 that can disable RagEngine service and delete all data within the service
docs: Enterprise tier in RagEngineConfig, use Scaled tier instead.

PiperOrigin-RevId: 772188314
diff --git a/google/cloud/aiplatform/v1beta1/vertex_rag_data.proto b/google/cloud/aiplatform/v1beta1/vertex_rag_data.proto
@@ -758,29 +758,52 @@ message ImportRagFilesConfig {
 
 // Configuration message for RagManagedDb used by RagEngine.
 message RagManagedDbConfig {
+  // Deprecated: Please use `Scaled` tier instead.
   // Enterprise tier offers production grade performance along with
   // autoscaling functionality. It is suitable for customers with large
   // amounts of data or performance sensitive workloads.
-  //
-  // NOTE: This is the default tier if not explicitly chosen.
-  message Enterprise {}
+  message Enterprise {
+    option deprecated = true;
+  }
+
+  // Scaled tier offers production grade performance along with
+  // autoscaling functionality. It is suitable for customers with large
+  // amounts of data or performance sensitive workloads.
+  message Scaled {}
 
   // Basic tier is a cost-effective and low compute tier suitable for
   // the following cases:
   // * Experimenting with RagManagedDb.
   // * Small data size.
   // * Latency insensitive workload.
   // * Only using RAG Engine with external vector DBs.
+  //
+  // NOTE: This is the default tier if not explicitly chosen.
   message Basic {}
 
+  // Disables the RAG Engine service and deletes all your data held
+  // within this service. This will halt the billing of the service.
+  //
+  // NOTE: Once deleted the data cannot be recovered. To start using
+  // RAG Engine again, you will need to update the tier by calling the
+  // UpdateRagEngineConfig API.
+  message Unprovisioned {}
+
   // The tier of the RagManagedDb.
   oneof tier {
+    // Deprecated: Please use `Scaled` tier instead.
     // Sets the RagManagedDb to the Enterprise tier. This is the default tier
     // if not explicitly chosen.
-    Enterprise enterprise = 1;
+    Enterprise enterprise = 1 [deprecated = true];
+
+    // Sets the RagManagedDb to the Scaled tier.
+    Scaled scaled = 4;
 
     // Sets the RagManagedDb to the Basic tier.
     Basic basic = 2;
+
+    // Sets the RagManagedDb to the Unprovisioned tier.
+    Unprovisioned unprovisioned = 3;
   }
 }
 
