feat: add support for Vertex AI Search engine

Google APIs · copybara-github · commit e02c606e5d01 · 2025-03-26T12:27:43.000-07:00
PiperOrigin-RevId: 740862834
diff --git a/google/cloud/aiplatform/v1/vertex_rag_data.proto b/google/cloud/aiplatform/v1/vertex_rag_data.proto
@@ -142,6 +142,15 @@ message FileStatus {
   string error_status = 2 [(google.api.field_behavior) = OUTPUT_ONLY];
 }
 
+// Config for the Vertex AI Search.
+message VertexAiSearchConfig {
+  // Vertex AI Search Serving Config resource full name. For example,
+  // `projects/{project}/locations/{location}/collections/{collection}/engines/{engine}/servingConfigs/{serving_config}`
+  // or
+  // `projects/{project}/locations/{location}/collections/{collection}/dataStores/{data_store}/servingConfigs/{serving_config}`.
+  string serving_config = 1;
+}
+
 // RagCorpus status.
 message CorpusStatus {
   // RagCorpus life state.
@@ -207,6 +216,12 @@ message RagCorpus {
       (google.api.field_behavior) = OPTIONAL,
       (google.api.field_behavior) = IMMUTABLE
     ];
+
+    // Optional. Immutable. The config for the Vertex AI Search.
+    VertexAiSearchConfig vertex_ai_search_config = 10 [
+      (google.api.field_behavior) = OPTIONAL,
+      (google.api.field_behavior) = IMMUTABLE
+    ];
   }
 }
 
