feat: adding ThinkingConfig to v1beta1 client library

Google APIs · copybara-github · commit 6e084afa5c1a · 2025-04-24T21:46:40.000-07:00
PiperOrigin-RevId: 751263449
diff --git a/google/cloud/aiplatform/v1beta1/content.proto b/google/cloud/aiplatform/v1beta1/content.proto
@@ -236,6 +236,13 @@ message GenerationConfig {
     }
   }
 
+  // Config for thinking features.
+  message ThinkingConfig {
+    // Optional. Indicates the thinking budget in tokens.
+    // This is only applied when enable_thinking is true.
+    optional int32 thinking_budget = 3 [(google.api.field_behavior) = OPTIONAL];
+  }
+
   // Config for model selection.
   message ModelConfig {
     // Options for feature selection preference.
@@ -359,6 +366,11 @@ message GenerationConfig {
   optional SpeechConfig speech_config = 23
       [(google.api.field_behavior) = OPTIONAL];
 
+  // Optional. Config for thinking features.
+  // An error will be returned if this field is set for models that don't
+  // support thinking.
+  ThinkingConfig thinking_config = 25 [(google.api.field_behavior) = OPTIONAL];
+
   // Optional. Config for model selection.
   ModelConfig model_config = 27 [(google.api.field_behavior) = OPTIONAL];
 }
