From 870b703e12dccfa4da364f284bacaa3bb170cea2 Mon Sep 17 00:00:00 2001
From: Google APIs <noreply@google.com>
Date: Mon, 13 Jan 2025 23:00:49 -0800
Subject: [PATCH] feat: add retrieval_config to ToolConfig v1

PiperOrigin-RevId: 715245748
---
 google/cloud/aiplatform/v1/BUILD.bazel |  2 ++
 google/cloud/aiplatform/v1/tool.proto  | 13 +++++++++++++
 2 files changed, 15 insertions(+)

diff --git a/google/cloud/aiplatform/v1/BUILD.bazel b/google/cloud/aiplatform/v1/BUILD.bazel
index aa5885d3b0..51ce3aad94 100644
--- a/google/cloud/aiplatform/v1/BUILD.bazel
+++ b/google/cloud/aiplatform/v1/BUILD.bazel
@@ -154,6 +154,7 @@ proto_library(
         "//google/rpc:status_proto",
         "//google/type:date_proto",
         "//google/type:interval_proto",
+        "//google/type:latlng_proto",
         "//google/type:money_proto",
         "@com_google_protobuf//:duration_proto",
         "@com_google_protobuf//:empty_proto",
@@ -301,6 +302,7 @@ go_proto_library(
         "//google/rpc:status_go_proto",
         "//google/type:date_go_proto",
         "//google/type:interval_go_proto",
+        "//google/type:latlng_go_proto",
         "//google/type:money_go_proto",
     ],
 )
diff --git a/google/cloud/aiplatform/v1/tool.proto b/google/cloud/aiplatform/v1/tool.proto
index 27af5c222b..263c5c38e9 100644
--- a/google/cloud/aiplatform/v1/tool.proto
+++ b/google/cloud/aiplatform/v1/tool.proto
@@ -20,6 +20,7 @@ import "google/api/field_behavior.proto";
 import "google/api/resource.proto";
 import "google/cloud/aiplatform/v1/openapi.proto";
 import "google/protobuf/struct.proto";
+import "google/type/latlng.proto";
 
 option csharp_namespace = "Google.Cloud.AIPlatform.V1";
 option go_package = "cloud.google.com/go/aiplatform/apiv1/aiplatformpb;aiplatformpb";
@@ -223,6 +224,9 @@ message ToolConfig {
   // Optional. Function calling config.
   FunctionCallingConfig function_calling_config = 1
       [(google.api.field_behavior) = OPTIONAL];
+
+  // Optional. Retrieval config.
+  RetrievalConfig retrieval_config = 2 [(google.api.field_behavior) = OPTIONAL];
 }
 
 // Function calling config.
@@ -257,6 +261,15 @@ message FunctionCallingConfig {
       [(google.api.field_behavior) = OPTIONAL];
 }
 
+// Retrieval config.
+message RetrievalConfig {
+  // The location of the user.
+  optional google.type.LatLng lat_lng = 1;
+
+  // The language code of the user.
+  optional string language_code = 2;
+}
+
 // Specifies the context retrieval config.
 message RagRetrievalConfig {
   // Config for filters.