GlcoinCore/googleapis
1
0
Fork 0
mirror of https://github.com/googleapis/googleapis.git synced 2026-08-20 13:27:47 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat: add retrieval_config to ToolConfig v1

Browse source 
PiperOrigin-RevId: 715245748
This commit is contained in:
Google APIs 2025-01-13 23:00:49 -08:00 • committed by Copybara-Service
parent ee799bd80b
commit 870b703e12
2 changed files with 15 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

2
google/cloud/aiplatform/v1/BUILD.bazel
View file

@ -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",
],
)

13
google/cloud/aiplatform/v1/tool.proto
View file

@ -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.