feat: add retrieval_config to ToolConfig v1beta1

Google APIs · copybara-github · commit 79d2e1371246 · 2025-01-13T23:10:58.000-08:00
PiperOrigin-RevId: 715248038
diff --git a/google/cloud/aiplatform/v1beta1/BUILD.bazel b/google/cloud/aiplatform/v1beta1/BUILD.bazel
@@ -162,6 +162,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",
@@ -317,6 +318,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/v1beta1/tool.proto b/google/cloud/aiplatform/v1beta1/tool.proto
@@ -20,6 +20,7 @@ import "google/api/field_behavior.proto";
 import "google/api/resource.proto";
 import "google/cloud/aiplatform/v1beta1/openapi.proto";
 import "google/protobuf/struct.proto";
+import "google/type/latlng.proto";
 
 option csharp_namespace = "Google.Cloud.AIPlatform.V1Beta1";
 option go_package = "cloud.google.com/go/aiplatform/apiv1beta1/aiplatformpb;aiplatformpb";
@@ -343,6 +344,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.
@@ -377,6 +381,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 Hybrid Search.
