feat: Add service tier to GenerateContent.

PiperOrigin-RevId: 890679445

Author: MarkDaoust

google/genai/batches.py

@@ -1052,6 +1052,9 @@ def _GenerateContentConfig_to_mldev(
         'model_armor_config parameter is not supported in Gemini API.'
     )
 
+  if getv(from_object, ['service_tier']) is not None:
+    setv(parent_object, ['serviceTier'], getv(from_object, ['service_tier']))
+
   return to_object
 
 

google/genai/models.py

@@ -1316,6 +1316,9 @@ def _GenerateContentConfig_to_mldev(
         'model_armor_config parameter is not supported in Gemini API.'
     )
 
+  if getv(from_object, ['service_tier']) is not None:
+    setv(parent_object, ['serviceTier'], getv(from_object, ['service_tier']))
+
   return to_object
 
 
@@ -1502,6 +1505,9 @@ def _GenerateContentConfig_to_vertex(
         getv(from_object, ['model_armor_config']),
     )
 
+  if getv(from_object, ['service_tier']) is not None:
+    raise ValueError('service_tier parameter is not supported in Vertex AI.')
+
   return to_object
 
 

google/genai/types.py

@@ -831,6 +831,19 @@ class ResourceScope(_common.CaseInSensitiveEnum):
       "https://aiplatform.googleapis.com/publishers/google/models/gemini-3-pro-preview"""
 
 
+class ServiceTier(_common.CaseInSensitiveEnum):
+  """Pricing and performance service tier."""
+
+  SERVICE_TIER_UNSPECIFIED = 'SERVICE_TIER_UNSPECIFIED'
+  """Default service tier, which is standard."""
+  SERVICE_TIER_FLEX = 'SERVICE_TIER_FLEX'
+  """Flex service tier."""
+  SERVICE_TIER_STANDARD = 'SERVICE_TIER_STANDARD'
+  """Standard service tier."""
+  SERVICE_TIER_PRIORITY = 'SERVICE_TIER_PRIORITY'
+  """Priority service tier."""
+
+
 class JSONSchemaType(Enum):
   """The type of the data supported by JSON Schema.
 
@@ -5937,6 +5950,10 @@ class GenerateContentConfig(_common.BaseModel):
       service. If supplied, safety_settings must not be supplied.
       """,
   )
+  service_tier: Optional[ServiceTier] = Field(
+      default=None,
+      description="""The service tier to use for the request. For example, SERVICE_TIER_FLEX.""",
+  )
 
   @pydantic.field_validator('response_schema', mode='before')
   @classmethod
@@ -6151,6 +6168,9 @@ class GenerateContentConfigDict(TypedDict, total=False):
       service. If supplied, safety_settings must not be supplied.
       """
 
+  service_tier: Optional[ServiceTier]
+  """The service tier to use for the request. For example, SERVICE_TIER_FLEX."""
+
 
 GenerateContentConfigOrDict = Union[
     GenerateContentConfig, GenerateContentConfigDict