feat: Introduce RagManagedVertexVectorSearch as a new vector db option

Google APIs · copybara-github · commit 2faf4a6d5795 · 2025-12-02T14:38:39.000-08:00
feat: add RagCorpus.satisfies_pzs and RagCorpus.satisfies_pzi
docs: A comment for field `rag_files_count` in message `.google.cloud.aiplatform.v1beta1.RagCorpus` is changed
docs: A comment for field `user_metadata` in message `.google.cloud.aiplatform.v1beta1.RagFile` is changed
docs: A comment for field `gcs_metadata_schema_source` in message `.google.cloud.aiplatform.v1beta1.RagFileMetadataConfig` is changed
docs: A comment for field `google_drive_metadata_schema_source` in message `.google.cloud.aiplatform.v1beta1.RagFileMetadataConfig` is changed
docs: A comment for field `gcs_metadata_source` in message `.google.cloud.aiplatform.v1beta1.RagFileMetadataConfig` is changed
docs: A comment for field `google_drive_metadata_source` in message `.google.cloud.aiplatform.v1beta1.RagFileMetadataConfig` is changed
docs: A comment for message `RagManagedDbConfig` is changed
docs: A comment for field `enterprise` in message `.google.cloud.aiplatform.v1beta1.RagManagedDbConfig` is changed
docs: A comment for field `scaled` in message `.google.cloud.aiplatform.v1beta1.RagManagedDbConfig` is changed

PiperOrigin-RevId: 839431594
diff --git a/google/cloud/aiplatform/v1beta1/vertex_rag_data.proto b/google/cloud/aiplatform/v1beta1/vertex_rag_data.proto
@@ -201,6 +201,15 @@ message RagVectorDbConfig {
     string index = 2;
   }
 
+  // The config for the RAG-managed Vertex Vector Search 2.0.
+  message RagManagedVertexVectorSearch {
+    // Output only. The resource name of the Vector Search 2.0 Collection that
+    // RAG Created for the corpus. Only populated after the corpus is
+    // successfully created. Format:
+    // `projects/{project}/locations/{location}/collections/{collection_id}`
+    string collection_name = 1 [(google.api.field_behavior) = OUTPUT_ONLY];
+  }
+
   // The config for the Vector DB.
   oneof vector_db {
     // The config for the RAG-managed Vector DB.
@@ -217,6 +226,9 @@ message RagVectorDbConfig {
 
     // The config for the Vertex Vector Search.
     VertexVectorSearch vertex_vector_search = 6;
+
+    // The config for the RAG-managed Vertex Vector Search 2.0.
+    RagManagedVertexVectorSearch rag_managed_vertex_vector_search = 8;
   }
 
   // Authentication config for the chosen Vector DB.
@@ -371,6 +383,9 @@ message RagCorpus {
   CorpusStatus corpus_status = 8 [(google.api.field_behavior) = OUTPUT_ONLY];
 
   // Output only. Number of RagFiles in the RagCorpus.
+  //
+  // NOTE: This field is not populated in the response of
+  // [VertexRagDataService.ListRagCorpora][google.cloud.aiplatform.v1beta1.VertexRagDataService.ListRagCorpora].
   int32 rag_files_count = 11 [(google.api.field_behavior) = OUTPUT_ONLY];
 
   // Optional. Immutable. The CMEK key name used to encrypt at-rest data related
@@ -385,6 +400,12 @@ message RagCorpus {
   // Optional. The corpus type config of the RagCorpus.
   CorpusTypeConfig corpus_type_config = 13
       [(google.api.field_behavior) = OPTIONAL];
+
+  // Output only. Reserved for future use.
+  bool satisfies_pzs = 19 [(google.api.field_behavior) = OUTPUT_ONLY];
+
+  // Output only. Reserved for future use.
+  bool satisfies_pzi = 20 [(google.api.field_behavior) = OUTPUT_ONLY];
 }
 
 // A RagFile contains user data for chunking, embedding and indexing.
@@ -463,7 +484,7 @@ message RagFile {
   // Output only. State of the RagFile.
   FileStatus file_status = 13 [(google.api.field_behavior) = OUTPUT_ONLY];
 
-  // Output only. The metadata for metadata search. The contents will be
+  // Output only. The metadata for metadata search. The user_metadata Needs to
   // be in JSON format.
   string user_metadata = 15 [(google.api.field_behavior) = OUTPUT_ONLY];
 }
@@ -604,13 +625,13 @@ message RagFileMetadataConfig {
     // well as entire Google Cloud Storage directories. Sample formats:
     // - `gs://bucket_name/my_directory/object_name/metadata_schema.json`
     // - `gs://bucket_name/my_directory`
-    // If providing a directory, the metadata schema will be read from
+    // If the user provides a directory, the metadata schema will be read from
     // the files that ends with "metadata_schema.json" in the directory.
     GcsSource gcs_metadata_schema_source = 1;
 
     // Google Drive location. Supports importing individual files as
     // well as Google Drive folders.
-    // If providing a folder, the metadata schema will be read from
+    // If the user provides a folder, the metadata schema will be read from
     // the files that ends with "metadata_schema.json" in the directory.
     GoogleDriveSource google_drive_metadata_schema_source = 2;
 
@@ -624,13 +645,13 @@ message RagFileMetadataConfig {
     // well as entire Google Cloud Storage directories. Sample formats:
     // - `gs://bucket_name/my_directory/object_name/metadata.json`
     // - `gs://bucket_name/my_directory`
-    // If providing a directory, the metadata will be read from
+    // If the user provides a directory, the metadata will be read from
     // the files that ends with "metadata.json" in the directory.
     GcsSource gcs_metadata_source = 4;
 
     // Google Drive location. Supports importing individual files as
     // well as Google Drive folders.
-    // If providing a directory, the metadata will be read from
+    // If the user provides a directory, the metadata will be read from
     // the files that ends with "metadata.json" in the directory.
     GoogleDriveSource google_drive_metadata_source = 5;
 
@@ -758,7 +779,6 @@ 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.
@@ -791,12 +811,11 @@ message RagManagedDbConfig {
 
   // 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.
+    // Sets the RagManagedDb to the Enterprise tier.
     Enterprise enterprise = 1 [deprecated = true];
 
-    // Sets the RagManagedDb to the Scaled tier.
+    // Sets the RagManagedDb to the Scaled tier. This is the default tier
+    // if not explicitly chosen.
     Scaled scaled = 4;
 
     // Sets the RagManagedDb to the Basic tier.
