feat: add model_config field for model selection preference

Google APIs · copybara-github · commit 21a7867e1f98 · 2025-03-24T10:56:46.000-07:00
PiperOrigin-RevId: 739998881
diff --git a/google/cloud/aiplatform/v1beta1/content.proto b/google/cloud/aiplatform/v1beta1/content.proto
@@ -235,6 +235,28 @@ message GenerationConfig {
     }
   }
 
+  // Config for model selection.
+  message ModelConfig {
+    // Options for feature selection preference.
+    enum FeatureSelectionPreference {
+      // Unspecified feature selection preference.
+      FEATURE_SELECTION_PREFERENCE_UNSPECIFIED = 0;
+
+      // Prefer higher quality over lower cost.
+      PRIORITIZE_QUALITY = 1;
+
+      // Balanced feature selection preference.
+      BALANCED = 2;
+
+      // Prefer lower cost over higher quality.
+      PRIORITIZE_COST = 3;
+    }
+
+    // Required. Feature selection preference.
+    FeatureSelectionPreference feature_selection_preference = 1
+        [(google.api.field_behavior) = REQUIRED];
+  }
+
   // The modalities of the response.
   enum Modality {
     // Unspecified modality. Will be processed as text.
@@ -335,6 +357,9 @@ message GenerationConfig {
   // Optional. The speech generation config.
   optional SpeechConfig speech_config = 23
       [(google.api.field_behavior) = OPTIONAL];
+
+  // Optional. Config for model selection.
+  ModelConfig model_config = 27 [(google.api.field_behavior) = OPTIONAL];
 }
 
 // Safety settings.
