Skip to content

Commit acd4aeb

Browse files
Google APIscopybara-github
authored andcommitted
feat: adding ThinkingConfig to v1 client library
PiperOrigin-RevId: 751263476
1 parent 6e084af commit acd4aeb

1 file changed

Lines changed: 12 additions & 0 deletions

File tree

google/cloud/aiplatform/v1/content.proto

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -211,6 +211,13 @@ message GenerationConfig {
211211
}
212212
}
213213

214+
// Config for thinking features.
215+
message ThinkingConfig {
216+
// Optional. Indicates the thinking budget in tokens.
217+
// This is only applied when enable_thinking is true.
218+
optional int32 thinking_budget = 3 [(google.api.field_behavior) = OPTIONAL];
219+
}
220+
214221
// Optional. Controls the randomness of predictions.
215222
optional float temperature = 1 [(google.api.field_behavior) = OPTIONAL];
216223

@@ -265,6 +272,11 @@ message GenerationConfig {
265272
// Optional. Routing configuration.
266273
optional RoutingConfig routing_config = 17
267274
[(google.api.field_behavior) = OPTIONAL];
275+
276+
// Optional. Config for thinking features.
277+
// An error will be returned if this field is set for models that don't
278+
// support thinking.
279+
ThinkingConfig thinking_config = 25 [(google.api.field_behavior) = OPTIONAL];
268280
}
269281

270282
// Safety settings.

0 commit comments

Comments
 (0)