From e02c606e5d01bbcd0cd997988edd18c2986d8f7e Mon Sep 17 00:00:00 2001
From: Google APIs <noreply@google.com>
Date: Wed, 26 Mar 2025 12:26:45 -0700
Subject: [PATCH] feat: add support for Vertex AI Search engine

PiperOrigin-RevId: 740862834
---
 google/cloud/aiplatform/v1/vertex_rag_data.proto | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/google/cloud/aiplatform/v1/vertex_rag_data.proto b/google/cloud/aiplatform/v1/vertex_rag_data.proto
index 0a0ad43b18..f6a432478f 100644
--- 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
+    ];
   }
 }