Skip to content

Commit 56d4e62

Browse files
Google APIscopybara-github
authored andcommitted
feat: add VertexAISearch.engine option
PiperOrigin-RevId: 732246835
1 parent 33b23a7 commit 56d4e62

2 files changed

Lines changed: 18 additions & 6 deletions

File tree

google/cloud/aiplatform/v1/tool.proto

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -261,13 +261,19 @@ message VertexRagStore {
261261
[(google.api.field_behavior) = OPTIONAL];
262262
}
263263

264-
// Retrieve from Vertex AI Search datastore for grounding.
264+
// Retrieve from Vertex AI Search datastore or engine for grounding.
265+
// datastore and engine are mutually exclusive.
265266
// See https://cloud.google.com/products/agent-builder
266267
message VertexAISearch {
267-
// Required. Fully-qualified Vertex AI Search data store resource ID.
268+
// Optional. Fully-qualified Vertex AI Search data store resource ID.
268269
// Format:
269270
// `projects/{project}/locations/{location}/collections/{collection}/dataStores/{dataStore}`
270-
string datastore = 1 [(google.api.field_behavior) = REQUIRED];
271+
string datastore = 1 [(google.api.field_behavior) = OPTIONAL];
272+
273+
// Optional. Fully-qualified Vertex AI Search engine resource ID.
274+
// Format:
275+
// `projects/{project}/locations/{location}/collections/{collection}/engines/{engine}`
276+
string engine = 2 [(google.api.field_behavior) = OPTIONAL];
271277
}
272278

273279
// Tool to retrieve public web data for grounding, powered by Google.

google/cloud/aiplatform/v1beta1/tool.proto

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -309,13 +309,19 @@ message VertexRagStore {
309309
[(google.api.field_behavior) = OPTIONAL];
310310
}
311311

312-
// Retrieve from Vertex AI Search datastore for grounding.
312+
// Retrieve from Vertex AI Search datastore or engine for grounding.
313+
// datastore and engine are mutually exclusive.
313314
// See https://cloud.google.com/products/agent-builder
314315
message VertexAISearch {
315-
// Required. Fully-qualified Vertex AI Search data store resource ID.
316+
// Optional. Fully-qualified Vertex AI Search data store resource ID.
316317
// Format:
317318
// `projects/{project}/locations/{location}/collections/{collection}/dataStores/{dataStore}`
318-
string datastore = 1 [(google.api.field_behavior) = REQUIRED];
319+
string datastore = 1 [(google.api.field_behavior) = OPTIONAL];
320+
321+
// Optional. Fully-qualified Vertex AI Search engine resource ID.
322+
// Format:
323+
// `projects/{project}/locations/{location}/collections/{collection}/engines/{engine}`
324+
string engine = 2 [(google.api.field_behavior) = OPTIONAL];
319325
}
320326

321327
// Tool to retrieve public web data for grounding, powered by Google.

0 commit comments

Comments
 (0)