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 + ]; } }