feat: add retrieval_config to ToolConfig v1

Google APIs · copybara-github · commit 870b703e12dc · 2025-01-13T23:01:32.000-08:00
PiperOrigin-RevId: 715245748
diff --git 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
@@ -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.
