google-genai 0.5.0__py3-none-any.whl → 0.7.0__py3-none-any.whl

google/genai/types.py

@@ -16,206 +16,315 @@
 # Code generated by the Google Gen AI SDK generator DO NOT EDIT.
 
 import datetime
+from enum import Enum, EnumMeta
 import inspect
 import json
 import logging
-from typing import Any, Callable, Literal, Optional, TypedDict, Union
-import PIL.Image
+import typing
+from typing import Any, Callable, GenericAlias, Literal, Optional, Type, TypedDict, Union
 import pydantic
 from pydantic import Field
 from . import _common
 
+_is_pillow_image_imported = False
+if typing.TYPE_CHECKING:
+  import PIL.Image
 
-Outcome = Literal[
-    "OUTCOME_UNSPECIFIED",
-    "OUTCOME_OK",
-    "OUTCOME_FAILED",
-    "OUTCOME_DEADLINE_EXCEEDED",
-]
+  PIL_Image = PIL.Image.Image
+  _is_pillow_image_imported = True
+else:
+  PIL_Image: Type = Any
+  try:
+    import PIL.Image
 
+    PIL_Image = PIL.Image.Image
+    _is_pillow_image_imported = True
+  except ImportError:
+    PIL_Image = None
 
-Language = Literal["LANGUAGE_UNSPECIFIED", "PYTHON"]
 
+class Outcome(_common.CaseInSensitiveEnum):
+  """Required. Outcome of the code execution."""
 
-Type = Literal[
-    "TYPE_UNSPECIFIED",
-    "STRING",
-    "NUMBER",
-    "INTEGER",
-    "BOOLEAN",
-    "ARRAY",
-    "OBJECT",
-]
+  OUTCOME_UNSPECIFIED = 'OUTCOME_UNSPECIFIED'
+  OUTCOME_OK = 'OUTCOME_OK'
+  OUTCOME_FAILED = 'OUTCOME_FAILED'
+  OUTCOME_DEADLINE_EXCEEDED = 'OUTCOME_DEADLINE_EXCEEDED'
 
 
-HarmCategory = Literal[
-    "HARM_CATEGORY_UNSPECIFIED",
-    "HARM_CATEGORY_HATE_SPEECH",
-    "HARM_CATEGORY_DANGEROUS_CONTENT",
-    "HARM_CATEGORY_HARASSMENT",
-    "HARM_CATEGORY_SEXUALLY_EXPLICIT",
-    "HARM_CATEGORY_CIVIC_INTEGRITY",
-]
+class Language(_common.CaseInSensitiveEnum):
+  """Required. Programming language of the `code`."""
 
+  LANGUAGE_UNSPECIFIED = 'LANGUAGE_UNSPECIFIED'
+  PYTHON = 'PYTHON'
 
-HarmBlockMethod = Literal[
-    "HARM_BLOCK_METHOD_UNSPECIFIED", "SEVERITY", "PROBABILITY"
-]
 
+class Type(_common.CaseInSensitiveEnum):
+  """A basic data type."""
 
-HarmBlockThreshold = Literal[
-    "HARM_BLOCK_THRESHOLD_UNSPECIFIED",
-    "BLOCK_LOW_AND_ABOVE",
-    "BLOCK_MEDIUM_AND_ABOVE",
-    "BLOCK_ONLY_HIGH",
-    "BLOCK_NONE",
-    "OFF",
-]
+  TYPE_UNSPECIFIED = 'TYPE_UNSPECIFIED'
+  STRING = 'STRING'
+  NUMBER = 'NUMBER'
+  INTEGER = 'INTEGER'
+  BOOLEAN = 'BOOLEAN'
+  ARRAY = 'ARRAY'
+  OBJECT = 'OBJECT'
 
 
-Mode = Literal["MODE_UNSPECIFIED", "MODE_DYNAMIC"]
+class HarmCategory(_common.CaseInSensitiveEnum):
+  """Required. Harm category."""
 
+  HARM_CATEGORY_UNSPECIFIED = 'HARM_CATEGORY_UNSPECIFIED'
+  HARM_CATEGORY_HATE_SPEECH = 'HARM_CATEGORY_HATE_SPEECH'
+  HARM_CATEGORY_DANGEROUS_CONTENT = 'HARM_CATEGORY_DANGEROUS_CONTENT'
+  HARM_CATEGORY_HARASSMENT = 'HARM_CATEGORY_HARASSMENT'
+  HARM_CATEGORY_SEXUALLY_EXPLICIT = 'HARM_CATEGORY_SEXUALLY_EXPLICIT'
+  HARM_CATEGORY_CIVIC_INTEGRITY = 'HARM_CATEGORY_CIVIC_INTEGRITY'
 
-FinishReason = Literal[
-    "FINISH_REASON_UNSPECIFIED",
-    "STOP",
-    "MAX_TOKENS",
-    "SAFETY",
-    "RECITATION",
-    "OTHER",
-    "BLOCKLIST",
-    "PROHIBITED_CONTENT",
-    "SPII",
-    "MALFORMED_FUNCTION_CALL",
-]
 
+class HarmBlockMethod(_common.CaseInSensitiveEnum):
+  """Optional.
 
-HarmProbability = Literal[
-    "HARM_PROBABILITY_UNSPECIFIED", "NEGLIGIBLE", "LOW", "MEDIUM", "HIGH"
-]
+  Specify if the threshold is used for probability or severity score. If not
+  specified, the threshold is used for probability score.
+  """
 
+  HARM_BLOCK_METHOD_UNSPECIFIED = 'HARM_BLOCK_METHOD_UNSPECIFIED'
+  SEVERITY = 'SEVERITY'
+  PROBABILITY = 'PROBABILITY'
 
-HarmSeverity = Literal[
-    "HARM_SEVERITY_UNSPECIFIED",
-    "HARM_SEVERITY_NEGLIGIBLE",
-    "HARM_SEVERITY_LOW",
-    "HARM_SEVERITY_MEDIUM",
-    "HARM_SEVERITY_HIGH",
-]
 
+class HarmBlockThreshold(_common.CaseInSensitiveEnum):
+  """Required. The harm block threshold."""
 
-BlockedReason = Literal[
-    "BLOCKED_REASON_UNSPECIFIED",
-    "SAFETY",
-    "OTHER",
-    "BLOCKLIST",
-    "PROHIBITED_CONTENT",
-]
+  HARM_BLOCK_THRESHOLD_UNSPECIFIED = 'HARM_BLOCK_THRESHOLD_UNSPECIFIED'
+  BLOCK_LOW_AND_ABOVE = 'BLOCK_LOW_AND_ABOVE'
+  BLOCK_MEDIUM_AND_ABOVE = 'BLOCK_MEDIUM_AND_ABOVE'
+  BLOCK_ONLY_HIGH = 'BLOCK_ONLY_HIGH'
+  BLOCK_NONE = 'BLOCK_NONE'
+  OFF = 'OFF'
 
 
-DeploymentResourcesType = Literal[
-    "DEPLOYMENT_RESOURCES_TYPE_UNSPECIFIED",
-    "DEDICATED_RESOURCES",
-    "AUTOMATIC_RESOURCES",
-    "SHARED_RESOURCES",
-]
+class Mode(_common.CaseInSensitiveEnum):
+  """The mode of the predictor to be used in dynamic retrieval."""
 
+  MODE_UNSPECIFIED = 'MODE_UNSPECIFIED'
+  MODE_DYNAMIC = 'MODE_DYNAMIC'
 
-JobState = Literal[
-    "JOB_STATE_UNSPECIFIED",
-    "JOB_STATE_QUEUED",
-    "JOB_STATE_PENDING",
-    "JOB_STATE_RUNNING",
-    "JOB_STATE_SUCCEEDED",
-    "JOB_STATE_FAILED",
-    "JOB_STATE_CANCELLING",
-    "JOB_STATE_CANCELLED",
-    "JOB_STATE_PAUSED",
-    "JOB_STATE_EXPIRED",
-    "JOB_STATE_UPDATING",
-    "JOB_STATE_PARTIALLY_SUCCEEDED",
-]
 
+class State(_common.CaseInSensitiveEnum):
+  """Output only. RagFile state."""
 
-AdapterSize = Literal[
-    "ADAPTER_SIZE_UNSPECIFIED",
-    "ADAPTER_SIZE_ONE",
-    "ADAPTER_SIZE_FOUR",
-    "ADAPTER_SIZE_EIGHT",
-    "ADAPTER_SIZE_SIXTEEN",
-    "ADAPTER_SIZE_THIRTY_TWO",
-]
+  STATE_UNSPECIFIED = 'STATE_UNSPECIFIED'
+  ACTIVE = 'ACTIVE'
+  ERROR = 'ERROR'
 
 
-State = Literal["STATE_UNSPECIFIED", "ACTIVE", "ERROR"]
+class FinishReason(_common.CaseInSensitiveEnum):
+  """Output only.
+
+  The reason why the model stopped generating tokens. If empty, the model has
+  not stopped generating the tokens.
+  """
 
+  FINISH_REASON_UNSPECIFIED = 'FINISH_REASON_UNSPECIFIED'
+  STOP = 'STOP'
+  MAX_TOKENS = 'MAX_TOKENS'
+  SAFETY = 'SAFETY'
+  RECITATION = 'RECITATION'
+  OTHER = 'OTHER'
+  BLOCKLIST = 'BLOCKLIST'
+  PROHIBITED_CONTENT = 'PROHIBITED_CONTENT'
+  SPII = 'SPII'
+  MALFORMED_FUNCTION_CALL = 'MALFORMED_FUNCTION_CALL'
 
-DynamicRetrievalConfigMode = Literal["MODE_UNSPECIFIED", "MODE_DYNAMIC"]
 
+class HarmProbability(_common.CaseInSensitiveEnum):
+  """Output only. Harm probability levels in the content."""
 
-FunctionCallingConfigMode = Literal["MODE_UNSPECIFIED", "AUTO", "ANY", "NONE"]
+  HARM_PROBABILITY_UNSPECIFIED = 'HARM_PROBABILITY_UNSPECIFIED'
+  NEGLIGIBLE = 'NEGLIGIBLE'
+  LOW = 'LOW'
+  MEDIUM = 'MEDIUM'
+  HIGH = 'HIGH'
 
 
-MediaResolution = Literal[
-    "MEDIA_RESOLUTION_UNSPECIFIED",
-    "MEDIA_RESOLUTION_LOW",
-    "MEDIA_RESOLUTION_MEDIUM",
-    "MEDIA_RESOLUTION_HIGH",
-]
+class HarmSeverity(_common.CaseInSensitiveEnum):
+  """Output only. Harm severity levels in the content."""
 
+  HARM_SEVERITY_UNSPECIFIED = 'HARM_SEVERITY_UNSPECIFIED'
+  HARM_SEVERITY_NEGLIGIBLE = 'HARM_SEVERITY_NEGLIGIBLE'
+  HARM_SEVERITY_LOW = 'HARM_SEVERITY_LOW'
+  HARM_SEVERITY_MEDIUM = 'HARM_SEVERITY_MEDIUM'
+  HARM_SEVERITY_HIGH = 'HARM_SEVERITY_HIGH'
 
-SafetyFilterLevel = Literal[
-    "BLOCK_LOW_AND_ABOVE",
-    "BLOCK_MEDIUM_AND_ABOVE",
-    "BLOCK_ONLY_HIGH",
-    "BLOCK_NONE",
-]
 
+class BlockedReason(_common.CaseInSensitiveEnum):
+  """Output only. Blocked reason."""
 
-PersonGeneration = Literal["DONT_ALLOW", "ALLOW_ADULT", "ALLOW_ALL"]
+  BLOCKED_REASON_UNSPECIFIED = 'BLOCKED_REASON_UNSPECIFIED'
+  SAFETY = 'SAFETY'
+  OTHER = 'OTHER'
+  BLOCKLIST = 'BLOCKLIST'
+  PROHIBITED_CONTENT = 'PROHIBITED_CONTENT'
 
 
-ImagePromptLanguage = Literal["auto", "en", "ja", "ko", "hi"]
+class DeploymentResourcesType(_common.CaseInSensitiveEnum):
+  """"""
 
+  DEPLOYMENT_RESOURCES_TYPE_UNSPECIFIED = (
+      'DEPLOYMENT_RESOURCES_TYPE_UNSPECIFIED'
+  )
+  DEDICATED_RESOURCES = 'DEDICATED_RESOURCES'
+  AUTOMATIC_RESOURCES = 'AUTOMATIC_RESOURCES'
+  SHARED_RESOURCES = 'SHARED_RESOURCES'
 
-MaskReferenceMode = Literal[
-    "MASK_MODE_DEFAULT",
-    "MASK_MODE_USER_PROVIDED",
-    "MASK_MODE_BACKGROUND",
-    "MASK_MODE_FOREGROUND",
-    "MASK_MODE_SEMANTIC",
-]
 
+class JobState(_common.CaseInSensitiveEnum):
+  """Job state."""
 
-ControlReferenceType = Literal[
-    "CONTROL_TYPE_DEFAULT",
-    "CONTROL_TYPE_CANNY",
-    "CONTROL_TYPE_SCRIBBLE",
-    "CONTROL_TYPE_FACE_MESH",
-]
+  JOB_STATE_UNSPECIFIED = 'JOB_STATE_UNSPECIFIED'
+  JOB_STATE_QUEUED = 'JOB_STATE_QUEUED'
+  JOB_STATE_PENDING = 'JOB_STATE_PENDING'
+  JOB_STATE_RUNNING = 'JOB_STATE_RUNNING'
+  JOB_STATE_SUCCEEDED = 'JOB_STATE_SUCCEEDED'
+  JOB_STATE_FAILED = 'JOB_STATE_FAILED'
+  JOB_STATE_CANCELLING = 'JOB_STATE_CANCELLING'
+  JOB_STATE_CANCELLED = 'JOB_STATE_CANCELLED'
+  JOB_STATE_PAUSED = 'JOB_STATE_PAUSED'
+  JOB_STATE_EXPIRED = 'JOB_STATE_EXPIRED'
+  JOB_STATE_UPDATING = 'JOB_STATE_UPDATING'
+  JOB_STATE_PARTIALLY_SUCCEEDED = 'JOB_STATE_PARTIALLY_SUCCEEDED'
 
 
-SubjectReferenceType = Literal[
-    "SUBJECT_TYPE_DEFAULT",
-    "SUBJECT_TYPE_PERSON",
-    "SUBJECT_TYPE_ANIMAL",
-    "SUBJECT_TYPE_PRODUCT",
-]
+class AdapterSize(_common.CaseInSensitiveEnum):
+  """Optional. Adapter size for tuning."""
 
+  ADAPTER_SIZE_UNSPECIFIED = 'ADAPTER_SIZE_UNSPECIFIED'
+  ADAPTER_SIZE_ONE = 'ADAPTER_SIZE_ONE'
+  ADAPTER_SIZE_FOUR = 'ADAPTER_SIZE_FOUR'
+  ADAPTER_SIZE_EIGHT = 'ADAPTER_SIZE_EIGHT'
+  ADAPTER_SIZE_SIXTEEN = 'ADAPTER_SIZE_SIXTEEN'
+  ADAPTER_SIZE_THIRTY_TWO = 'ADAPTER_SIZE_THIRTY_TWO'
 
-EditMode = Literal[
-    "EDIT_MODE_DEFAULT",
-    "EDIT_MODE_INPAINT_REMOVAL",
-    "EDIT_MODE_INPAINT_INSERTION",
-    "EDIT_MODE_OUTPAINT",
-    "EDIT_MODE_CONTROLLED_EDITING",
-    "EDIT_MODE_STYLE",
-    "EDIT_MODE_BGSWAP",
-    "EDIT_MODE_PRODUCT_IMAGE",
-]
+
+class DynamicRetrievalConfigMode(_common.CaseInSensitiveEnum):
+  """Config for the dynamic retrieval config mode."""
+
+  MODE_UNSPECIFIED = 'MODE_UNSPECIFIED'
+  MODE_DYNAMIC = 'MODE_DYNAMIC'
+
+
+class FunctionCallingConfigMode(_common.CaseInSensitiveEnum):
+  """Config for the function calling config mode."""
+
+  MODE_UNSPECIFIED = 'MODE_UNSPECIFIED'
+  AUTO = 'AUTO'
+  ANY = 'ANY'
+  NONE = 'NONE'
+
+
+class MediaResolution(_common.CaseInSensitiveEnum):
+  """The media resolution to use."""
+
+  MEDIA_RESOLUTION_UNSPECIFIED = 'MEDIA_RESOLUTION_UNSPECIFIED'
+  MEDIA_RESOLUTION_LOW = 'MEDIA_RESOLUTION_LOW'
+  MEDIA_RESOLUTION_MEDIUM = 'MEDIA_RESOLUTION_MEDIUM'
+  MEDIA_RESOLUTION_HIGH = 'MEDIA_RESOLUTION_HIGH'
+
+
+class SafetyFilterLevel(_common.CaseInSensitiveEnum):
+  """Enum that controls the safety filter level for objectionable content."""
+
+  BLOCK_LOW_AND_ABOVE = 'BLOCK_LOW_AND_ABOVE'
+  BLOCK_MEDIUM_AND_ABOVE = 'BLOCK_MEDIUM_AND_ABOVE'
+  BLOCK_ONLY_HIGH = 'BLOCK_ONLY_HIGH'
+  BLOCK_NONE = 'BLOCK_NONE'
+
+
+class PersonGeneration(_common.CaseInSensitiveEnum):
+  """Enum that controls the generation of people."""
+
+  DONT_ALLOW = 'DONT_ALLOW'
+  ALLOW_ADULT = 'ALLOW_ADULT'
+  ALLOW_ALL = 'ALLOW_ALL'
+
+
+class ImagePromptLanguage(_common.CaseInSensitiveEnum):
+  """Enum that specifies the language of the text in the prompt."""
+
+  auto = 'auto'
+  en = 'en'
+  ja = 'ja'
+  ko = 'ko'
+  hi = 'hi'
+
+
+class MaskReferenceMode(_common.CaseInSensitiveEnum):
+  """Enum representing the mask mode of a mask reference image."""
+
+  MASK_MODE_DEFAULT = 'MASK_MODE_DEFAULT'
+  MASK_MODE_USER_PROVIDED = 'MASK_MODE_USER_PROVIDED'
+  MASK_MODE_BACKGROUND = 'MASK_MODE_BACKGROUND'
+  MASK_MODE_FOREGROUND = 'MASK_MODE_FOREGROUND'
+  MASK_MODE_SEMANTIC = 'MASK_MODE_SEMANTIC'
+
+
+class ControlReferenceType(_common.CaseInSensitiveEnum):
+  """Enum representing the control type of a control reference image."""
+
+  CONTROL_TYPE_DEFAULT = 'CONTROL_TYPE_DEFAULT'
+  CONTROL_TYPE_CANNY = 'CONTROL_TYPE_CANNY'
+  CONTROL_TYPE_SCRIBBLE = 'CONTROL_TYPE_SCRIBBLE'
+  CONTROL_TYPE_FACE_MESH = 'CONTROL_TYPE_FACE_MESH'
+
+
+class SubjectReferenceType(_common.CaseInSensitiveEnum):
+  """Enum representing the subject type of a subject reference image."""
+
+  SUBJECT_TYPE_DEFAULT = 'SUBJECT_TYPE_DEFAULT'
+  SUBJECT_TYPE_PERSON = 'SUBJECT_TYPE_PERSON'
+  SUBJECT_TYPE_ANIMAL = 'SUBJECT_TYPE_ANIMAL'
+  SUBJECT_TYPE_PRODUCT = 'SUBJECT_TYPE_PRODUCT'
+
+
+class EditMode(_common.CaseInSensitiveEnum):
+  """Enum representing the Imagen 3 Edit mode."""
+
+  EDIT_MODE_DEFAULT = 'EDIT_MODE_DEFAULT'
+  EDIT_MODE_INPAINT_REMOVAL = 'EDIT_MODE_INPAINT_REMOVAL'
+  EDIT_MODE_INPAINT_INSERTION = 'EDIT_MODE_INPAINT_INSERTION'
+  EDIT_MODE_OUTPAINT = 'EDIT_MODE_OUTPAINT'
+  EDIT_MODE_CONTROLLED_EDITING = 'EDIT_MODE_CONTROLLED_EDITING'
+  EDIT_MODE_STYLE = 'EDIT_MODE_STYLE'
+  EDIT_MODE_BGSWAP = 'EDIT_MODE_BGSWAP'
+  EDIT_MODE_PRODUCT_IMAGE = 'EDIT_MODE_PRODUCT_IMAGE'
+
+
+class FileState(_common.CaseInSensitiveEnum):
+  """State for the lifecycle of a File."""
+
+  STATE_UNSPECIFIED = 'STATE_UNSPECIFIED'
+  PROCESSING = 'PROCESSING'
+  ACTIVE = 'ACTIVE'
+  FAILED = 'FAILED'
 
 
-FileState = Literal["STATE_UNSPECIFIED", "PROCESSING", "ACTIVE", "FAILED"]
+class FileSource(_common.CaseInSensitiveEnum):
+  """Source of the File."""
+
+  SOURCE_UNSPECIFIED = 'SOURCE_UNSPECIFIED'
+  UPLOADED = 'UPLOADED'
+  GENERATED = 'GENERATED'
+
+
+class Modality(_common.CaseInSensitiveEnum):
+  """Server content modalities."""
+
+  MODALITY_UNSPECIFIED = 'MODALITY_UNSPECIFIED'
+  TEXT = 'TEXT'
+  IMAGE = 'IMAGE'
+  AUDIO = 'AUDIO'
 
 
 class VideoMetadata(_common.BaseModel):
@@ -402,10 +511,7 @@ FunctionResponseOrDict = Union[FunctionResponse, FunctionResponseDict]
 
 
 class Blob(_common.BaseModel):
-  """Content blob.
-
-  It's preferred to send as text directly rather than raw bytes.
-  """
+  """Content blob."""
 
   data: Optional[bytes] = Field(
       default=None, description="""Required. Raw bytes."""
@@ -417,10 +523,7 @@ class Blob(_common.BaseModel):
 
 
 class BlobDict(TypedDict, total=False):
-  """Content blob.
-
-  It's preferred to send as text directly rather than raw bytes.
-  """
+  """Content blob."""
 
   data: Optional[bytes]
   """Required. Raw bytes."""
@@ -474,16 +577,16 @@ class Part(_common.BaseModel):
   )
 
   @classmethod
-  def from_uri(cls, file_uri: str, mime_type: str) -> "Part":
+  def from_uri(cls, *, file_uri: str, mime_type: str) -> 'Part':
     file_data = FileData(file_uri=file_uri, mime_type=mime_type)
     return cls(file_data=file_data)
 
   @classmethod
-  def from_text(cls, text: str) -> "Part":
+  def from_text(cls, *, text: str) -> 'Part':
     return cls(text=text)
 
   @classmethod
-  def from_bytes(cls, data: bytes, mime_type: str) -> "Part":
+  def from_bytes(cls, *, data: bytes, mime_type: str) -> 'Part':
     inline_data = Blob(
         data=data,
         mime_type=mime_type,
@@ -491,31 +594,33 @@ class Part(_common.BaseModel):
     return cls(inline_data=inline_data)
 
   @classmethod
-  def from_function_call(cls, name: str, args: dict[str, Any]) -> "Part":
+  def from_function_call(cls, *, name: str, args: dict[str, Any]) -> 'Part':
     function_call = FunctionCall(name=name, args=args)
     return cls(function_call=function_call)
 
   @classmethod
   def from_function_response(
-      cls, name: str, response: dict[str, Any]
-  ) -> "Part":
+      cls, *, name: str, response: dict[str, Any]
+  ) -> 'Part':
     function_response = FunctionResponse(name=name, response=response)
     return cls(function_response=function_response)
 
   @classmethod
-  def from_video_metadata(cls, end_offset: str, start_offset: str) -> "Part":
+  def from_video_metadata(cls, *, start_offset: str, end_offset: str) -> 'Part':
     video_metadata = VideoMetadata(
         end_offset=end_offset, start_offset=start_offset
     )
     return cls(video_metadata=video_metadata)
 
   @classmethod
-  def from_executable_code(cls, code: str, language: Language) -> "Part":
+  def from_executable_code(cls, *, code: str, language: Language) -> 'Part':
     executable_code = ExecutableCode(code=code, language=language)
     return cls(executable_code=executable_code)
 
   @classmethod
-  def from_code_execution_result(cls, outcome: Outcome, output: str) -> "Part":
+  def from_code_execution_result(
+      cls, *, outcome: Outcome, output: str
+  ) -> 'Part':
     code_execution_result = CodeExecutionResult(outcome=outcome, output=output)
     return cls(code_execution_result=code_execution_result)
 
@@ -557,8 +662,6 @@ class PartDict(TypedDict, total=False):
 
 
 PartOrDict = Union[Part, PartDict]
-PartUnion = Union[Part, PIL.Image.Image, str]
-PartUnionDict = Union[PartUnion, PartDict]
 
 
 class Content(_common.BaseModel):
@@ -591,11 +694,51 @@ class ContentDict(TypedDict, total=False):
 
 
 ContentOrDict = Union[Content, ContentDict]
-ContentUnion = Union[Content, list[PartUnion], PartUnion]
-ContentUnionDict = Union[ContentUnion, ContentDict]
 
-ContentListUnion = Union[list[ContentUnion], ContentUnion]
-ContentListUnionDict = Union[list[ContentUnionDict], ContentUnionDict]
+
+class HttpOptions(_common.BaseModel):
+  """HTTP options to be used in each of the requests."""
+
+  base_url: Optional[str] = Field(
+      default=None,
+      description="""The base URL for the AI platform service endpoint.""",
+  )
+  api_version: Optional[str] = Field(
+      default=None, description="""Specifies the version of the API to use."""
+  )
+  headers: Optional[dict[str, str]] = Field(
+      default=None,
+      description="""Additional HTTP headers to be sent with the request.""",
+  )
+  timeout: Optional[int] = Field(
+      default=None, description="""Timeout for the request in milliseconds."""
+  )
+  deprecated_response_payload: Optional[dict[str, any]] = Field(
+      default=None,
+      description="""This field is deprecated. If set, the response payload will be returned int the supplied dict.""",
+  )
+
+
+class HttpOptionsDict(TypedDict, total=False):
+  """HTTP options to be used in each of the requests."""
+
+  base_url: Optional[str]
+  """The base URL for the AI platform service endpoint."""
+
+  api_version: Optional[str]
+  """Specifies the version of the API to use."""
+
+  headers: Optional[dict[str, str]]
+  """Additional HTTP headers to be sent with the request."""
+
+  timeout: Optional[int]
+  """Timeout for the request in milliseconds."""
+
+  deprecated_response_payload: Optional[dict[str, any]]
+  """This field is deprecated. If set, the response payload will be returned int the supplied dict."""
+
+
+HttpOptionsOrDict = Union[HttpOptions, HttpOptionsDict]
 
 
 class Schema(_common.BaseModel):
@@ -627,7 +770,7 @@ class Schema(_common.BaseModel):
   default: Optional[Any] = Field(
       default=None, description="""Optional. Default value of the data."""
   )
-  any_of: list["Schema"] = Field(
+  any_of: list['Schema'] = Field(
       default=None,
       description="""Optional. The value should be validated against any (one or more) of the subschemas in the list.""",
   )
@@ -676,11 +819,11 @@ class Schema(_common.BaseModel):
       default=None,
       description="""Optional. The format of the data. Supported formats: for NUMBER type: "float", "double" for INTEGER type: "int32", "int64" for STRING type: "email", "byte", etc""",
   )
-  items: "Schema" = Field(
+  items: 'Schema' = Field(
       default=None,
       description="""Optional. SCHEMA FIELDS FOR TYPE ARRAY Schema of the elements of Type.ARRAY.""",
   )
-  properties: dict[str, "Schema"] = Field(
+  properties: dict[str, 'Schema'] = Field(
       default=None,
       description="""Optional. SCHEMA FIELDS FOR TYPE OBJECT Properties of Type.OBJECT.""",
   )
@@ -714,7 +857,7 @@ class SchemaDict(TypedDict, total=False):
   default: Optional[Any]
   """Optional. Default value of the data."""
 
-  any_of: list["SchemaDict"]
+  any_of: list['SchemaDict']
   """Optional. The value should be validated against any (one or more) of the subschemas in the list."""
 
   max_length: Optional[int]
@@ -753,10 +896,10 @@ class SchemaDict(TypedDict, total=False):
   format: Optional[str]
   """Optional. The format of the data. Supported formats: for NUMBER type: "float", "double" for INTEGER type: "int32", "int64" for STRING type: "email", "byte", etc"""
 
-  items: "SchemaDict"
+  items: 'SchemaDict'
   """Optional. SCHEMA FIELDS FOR TYPE ARRAY Schema of the elements of Type.ARRAY."""
 
-  properties: dict[str, "SchemaDict"]
+  properties: dict[str, 'SchemaDict']
   """Optional. SCHEMA FIELDS FOR TYPE OBJECT Properties of Type.OBJECT."""
 
   required: Optional[list[str]]
@@ -764,8 +907,6 @@ class SchemaDict(TypedDict, total=False):
 
 
 SchemaOrDict = Union[Schema, SchemaDict]
-SchemaUnion = Union[dict, type, Schema]
-SchemaUnionDict = Union[SchemaUnion, SchemaDict]
 
 
 class SafetySetting(_common.BaseModel):
@@ -827,88 +968,61 @@ class FunctionDeclaration(_common.BaseModel):
   )
 
   @classmethod
-  def _get_variant(cls, client) -> str:
-    """Returns the function variant based on the provided client object."""
-    if client.vertexai:
-      return "VERTEX_AI"
-    else:
-      return "GOOGLE_AI"
-
-  @classmethod
-  def from_function_with_options(
+  def from_callable(
       cls,
-      func: Callable,
-      variant: Literal["GOOGLE_AI", "VERTEX_AI", "DEFAULT"] = "GOOGLE_AI",
-  ) -> "FunctionDeclaration":
-    """Converts a function to a FunctionDeclaration based on an API endpoint.
-
-    Supported endpoints are: 'GOOGLE_AI', 'VERTEX_AI', or 'DEFAULT'.
-    """
+      *,
+      client,
+      callable: Callable,
+  ) -> 'FunctionDeclaration':
+    """Converts a Callable to a FunctionDeclaration based on the client."""
     from . import _automatic_function_calling_util
 
-    supported_variants = ["GOOGLE_AI", "VERTEX_AI", "DEFAULT"]
-    if variant not in supported_variants:
-      raise ValueError(
-          f"Unsupported variant: {variant}. Supported variants are:"
-          f" {', '.join(supported_variants)}"
-      )
-
-    # TODO: b/382524014 - Add support for DEFAULT API endpoint.
-
     parameters_properties = {}
-    for name, param in inspect.signature(func).parameters.items():
+    for name, param in inspect.signature(callable).parameters.items():
       if param.kind in (
           inspect.Parameter.POSITIONAL_OR_KEYWORD,
           inspect.Parameter.KEYWORD_ONLY,
           inspect.Parameter.POSITIONAL_ONLY,
       ):
         schema = _automatic_function_calling_util._parse_schema_from_parameter(
-            variant, param, func.__name__
+            client, param, callable.__name__
         )
         parameters_properties[name] = schema
     declaration = FunctionDeclaration(
-        name=func.__name__,
-        description=func.__doc__,
+        name=callable.__name__,
+        description=callable.__doc__,
     )
     if parameters_properties:
       declaration.parameters = Schema(
-          type="OBJECT",
+          type='OBJECT',
           properties=parameters_properties,
       )
-      if variant == "VERTEX_AI":
+      if client.vertexai:
         declaration.parameters.required = (
             _automatic_function_calling_util._get_required_fields(
                 declaration.parameters
             )
         )
-    if not variant == "VERTEX_AI":
+    if not client.vertexai:
       return declaration
 
-    return_annotation = inspect.signature(func).return_annotation
+    return_annotation = inspect.signature(callable).return_annotation
     if return_annotation is inspect._empty:
       return declaration
 
     declaration.response = (
         _automatic_function_calling_util._parse_schema_from_parameter(
-            variant,
+            client,
             inspect.Parameter(
-                "return_value",
+                'return_value',
                 inspect.Parameter.POSITIONAL_OR_KEYWORD,
                 annotation=return_annotation,
             ),
-            func.__name__,
+            callable.__name__,
         )
     )
     return declaration
 
-  @classmethod
-  def from_function(cls, client, func: Callable) -> "FunctionDeclaration":
-    """Converts a function to a FunctionDeclaration."""
-    return cls.from_function_with_options(
-        variant=cls._get_variant(client),
-        func=func,
-    )
-
 
 class FunctionDeclarationDict(TypedDict, total=False):
   """Defines a function that the model can generate JSON inputs for.
@@ -1310,8 +1424,6 @@ class SpeechConfigDict(TypedDict, total=False):
 
 
 SpeechConfigOrDict = Union[SpeechConfig, SpeechConfigDict]
-SpeechConfigUnion = Union[SpeechConfig, str]
-SpeechConfigUnionDict = Union[SpeechConfigUnion, SpeechConfigDict]
 
 
 class AutomaticFunctionCallingConfig(_common.BaseModel):
@@ -1373,154 +1485,340 @@ AutomaticFunctionCallingConfigOrDict = Union[
 ]
 
 
-class GenerationConfigRoutingConfigAutoRoutingMode(_common.BaseModel):
-  """When automated routing is specified, the routing will be determined by the pretrained routing model and customer provided model routing preference."""
-
-  model_routing_preference: Optional[
-      Literal["UNKNOWN", "PRIORITIZE_QUALITY", "BALANCED", "PRIORITIZE_COST"]
-  ] = Field(default=None, description="""The model routing preference.""")
-
-
-class GenerationConfigRoutingConfigAutoRoutingModeDict(TypedDict, total=False):
-  """When automated routing is specified, the routing will be determined by the pretrained routing model and customer provided model routing preference."""
-
-  model_routing_preference: Optional[
-      Literal["UNKNOWN", "PRIORITIZE_QUALITY", "BALANCED", "PRIORITIZE_COST"]
-  ]
-  """The model routing preference."""
-
-
-GenerationConfigRoutingConfigAutoRoutingModeOrDict = Union[
-    GenerationConfigRoutingConfigAutoRoutingMode,
-    GenerationConfigRoutingConfigAutoRoutingModeDict,
-]
-
-
-class GenerationConfigRoutingConfigManualRoutingMode(_common.BaseModel):
-  """When manual routing is set, the specified model will be used directly."""
+class ThinkingConfig(_common.BaseModel):
+  """The thinking features configuration."""
 
-  model_name: Optional[str] = Field(
+  include_thoughts: Optional[bool] = Field(
       default=None,
-      description="""The model name to use. Only the public LLM models are accepted. e.g. 'gemini-1.5-pro-001'.""",
+      description="""Indicates whether to include thoughts in the response. If true, thoughts are returned only if the model supports thought and thoughts are available.
+      """,
   )
 
 
-class GenerationConfigRoutingConfigManualRoutingModeDict(
-    TypedDict, total=False
-):
-  """When manual routing is set, the specified model will be used directly."""
+class ThinkingConfigDict(TypedDict, total=False):
+  """The thinking features configuration."""
 
-  model_name: Optional[str]
-  """The model name to use. Only the public LLM models are accepted. e.g. 'gemini-1.5-pro-001'."""
+  include_thoughts: Optional[bool]
+  """Indicates whether to include thoughts in the response. If true, thoughts are returned only if the model supports thought and thoughts are available.
+      """
 
 
-GenerationConfigRoutingConfigManualRoutingModeOrDict = Union[
-    GenerationConfigRoutingConfigManualRoutingMode,
-    GenerationConfigRoutingConfigManualRoutingModeDict,
-]
+ThinkingConfigOrDict = Union[ThinkingConfig, ThinkingConfigDict]
 
 
-class GenerationConfigRoutingConfig(_common.BaseModel):
-  """The configuration for routing the request to a specific model."""
+class FileStatus(_common.BaseModel):
+  """Status of a File that uses a common error model."""
 
-  auto_mode: Optional[GenerationConfigRoutingConfigAutoRoutingMode] = Field(
-      default=None, description="""Automated routing."""
+  details: Optional[list[dict[str, Any]]] = Field(
+      default=None,
+      description="""A list of messages that carry the error details. There is a common set of message types for APIs to use.""",
   )
-  manual_mode: Optional[GenerationConfigRoutingConfigManualRoutingMode] = Field(
-      default=None, description="""Manual routing."""
+  message: Optional[str] = Field(
+      default=None,
+      description="""A list of messages that carry the error details. There is a common set of message types for APIs to use.""",
+  )
+  code: Optional[int] = Field(
+      default=None, description="""The status code. 0 for OK, 1 for CANCELLED"""
   )
 
 
-class GenerationConfigRoutingConfigDict(TypedDict, total=False):
-  """The configuration for routing the request to a specific model."""
+class FileStatusDict(TypedDict, total=False):
+  """Status of a File that uses a common error model."""
 
-  auto_mode: Optional[GenerationConfigRoutingConfigAutoRoutingModeDict]
-  """Automated routing."""
+  details: Optional[list[dict[str, Any]]]
+  """A list of messages that carry the error details. There is a common set of message types for APIs to use."""
 
-  manual_mode: Optional[GenerationConfigRoutingConfigManualRoutingModeDict]
-  """Manual routing."""
+  message: Optional[str]
+  """A list of messages that carry the error details. There is a common set of message types for APIs to use."""
 
+  code: Optional[int]
+  """The status code. 0 for OK, 1 for CANCELLED"""
 
-GenerationConfigRoutingConfigOrDict = Union[
-    GenerationConfigRoutingConfig, GenerationConfigRoutingConfigDict
-]
 
+FileStatusOrDict = Union[FileStatus, FileStatusDict]
 
-class GenerateContentConfig(_common.BaseModel):
-  """Class for configuring optional model parameters.
 
-  For more information, see `Content generation parameters
-  <https://cloud.google.com/vertex-ai/generative-ai/docs/multimodal/content-generation-parameters>`_.
-  """
+class File(_common.BaseModel):
+  """A file uploaded to the API."""
 
-  system_instruction: Optional[ContentUnion] = Field(
+  name: Optional[str] = Field(
       default=None,
-      description="""Instructions for the model to steer it toward better performance.
-      For example, "Answer as concisely as possible" or "Don't use technical
-      terms in your response".
-      """,
+      description="""The `File` resource name. The ID (name excluding the "files/" prefix) can contain up to 40 characters that are lowercase alphanumeric or dashes (-). The ID cannot start or end with a dash. If the name is empty on create, a unique name will be generated. Example: `files/123-456`""",
   )
-  temperature: Optional[float] = Field(
+  display_name: Optional[str] = Field(
       default=None,
-      description="""Value that controls the degree of randomness in token selection.
-      Lower temperatures are good for prompts that require a less open-ended or
-      creative response, while higher temperatures can lead to more diverse or
-      creative results.
-      """,
+      description="""Optional. The human-readable display name for the `File`. The display name must be no more than 512 characters in length, including spaces. Example: 'Welcome Image'""",
   )
-  top_p: Optional[float] = Field(
-      default=None,
-      description="""Tokens are selected from the most to least probable until the sum
-      of their probabilities equals this value. Use a lower value for less
-      random responses and a higher value for more random responses.
-      """,
+  mime_type: Optional[str] = Field(
+      default=None, description="""Output only. MIME type of the file."""
   )
-  top_k: Optional[float] = Field(
-      default=None,
-      description="""For each token selection step, the ``top_k`` tokens with the
-      highest probabilities are sampled. Then tokens are further filtered based
-      on ``top_p`` with the final token selected using temperature sampling. Use
-      a lower number for less random responses and a higher number for more
-      random responses.
-      """,
+  size_bytes: Optional[int] = Field(
+      default=None, description="""Output only. Size of the file in bytes."""
   )
-  candidate_count: Optional[int] = Field(
+  create_time: Optional[datetime.datetime] = Field(
       default=None,
-      description="""Number of response variations to return.
-      """,
+      description="""Output only. The timestamp of when the `File` was created.""",
   )
-  max_output_tokens: Optional[int] = Field(
+  expiration_time: Optional[datetime.datetime] = Field(
       default=None,
-      description="""Maximum number of tokens that can be generated in the response.
-      """,
+      description="""Optional. The human-readable display name for the `File`. The display name must be no more than 512 characters in length, including spaces. Example: 'Welcome Image'""",
   )
-  stop_sequences: Optional[list[str]] = Field(
+  update_time: Optional[datetime.datetime] = Field(
       default=None,
-      description="""List of strings that tells the model to stop generating text if one
-      of the strings is encountered in the response.
-      """,
+      description="""Output only. The timestamp of when the `File` was last updated.""",
   )
-  response_logprobs: Optional[bool] = Field(
+  sha256_hash: Optional[str] = Field(
       default=None,
-      description="""Whether to return the log probabilities of the tokens that were
-      chosen by the model at each step.
-      """,
+      description="""Output only. SHA-256 hash of the uploaded bytes.""",
   )
-  logprobs: Optional[int] = Field(
-      default=None,
-      description="""Number of top candidate tokens to return the log probabilities for
-      at each generation step.
-      """,
+  uri: Optional[str] = Field(
+      default=None, description="""Output only. The URI of the `File`."""
   )
-  presence_penalty: Optional[float] = Field(
+  download_uri: Optional[str] = Field(
       default=None,
-      description="""Positive values penalize tokens that already appear in the
-      generated text, increasing the probability of generating more diverse
-      content.
-      """,
+      description="""Output only. The URI of the `File`, only set for downloadable (generated) files.""",
   )
-  frequency_penalty: Optional[float] = Field(
-      default=None,
+  state: Optional[FileState] = Field(
+      default=None, description="""Output only. Processing state of the File."""
+  )
+  source: Optional[FileSource] = Field(
+      default=None, description="""Output only. The source of the `File`."""
+  )
+  video_metadata: Optional[dict[str, Any]] = Field(
+      default=None, description="""Output only. Metadata for a video."""
+  )
+  error: Optional[FileStatus] = Field(
+      default=None,
+      description="""Output only. Error status if File processing failed.""",
+  )
+
+
+class FileDict(TypedDict, total=False):
+  """A file uploaded to the API."""
+
+  name: Optional[str]
+  """The `File` resource name. The ID (name excluding the "files/" prefix) can contain up to 40 characters that are lowercase alphanumeric or dashes (-). The ID cannot start or end with a dash. If the name is empty on create, a unique name will be generated. Example: `files/123-456`"""
+
+  display_name: Optional[str]
+  """Optional. The human-readable display name for the `File`. The display name must be no more than 512 characters in length, including spaces. Example: 'Welcome Image'"""
+
+  mime_type: Optional[str]
+  """Output only. MIME type of the file."""
+
+  size_bytes: Optional[int]
+  """Output only. Size of the file in bytes."""
+
+  create_time: Optional[datetime.datetime]
+  """Output only. The timestamp of when the `File` was created."""
+
+  expiration_time: Optional[datetime.datetime]
+  """Optional. The human-readable display name for the `File`. The display name must be no more than 512 characters in length, including spaces. Example: 'Welcome Image'"""
+
+  update_time: Optional[datetime.datetime]
+  """Output only. The timestamp of when the `File` was last updated."""
+
+  sha256_hash: Optional[str]
+  """Output only. SHA-256 hash of the uploaded bytes."""
+
+  uri: Optional[str]
+  """Output only. The URI of the `File`."""
+
+  download_uri: Optional[str]
+  """Output only. The URI of the `File`, only set for downloadable (generated) files."""
+
+  state: Optional[FileState]
+  """Output only. Processing state of the File."""
+
+  source: Optional[FileSource]
+  """Output only. The source of the `File`."""
+
+  video_metadata: Optional[dict[str, Any]]
+  """Output only. Metadata for a video."""
+
+  error: Optional[FileStatusDict]
+  """Output only. Error status if File processing failed."""
+
+
+FileOrDict = Union[File, FileDict]
+
+if _is_pillow_image_imported:
+  PartUnion = Union[File, Part, PIL_Image, str]
+else:
+  PartUnion = Union[File, Part, str]
+
+
+PartUnionDict = Union[PartUnion, PartDict]
+
+
+ContentUnion = Union[Content, list[PartUnion], PartUnion]
+
+
+ContentUnionDict = Union[ContentUnion, ContentDict]
+
+
+SchemaUnion = Union[dict, type, Schema, GenericAlias]
+
+
+SchemaUnionDict = Union[SchemaUnion, SchemaDict]
+
+
+class GenerationConfigRoutingConfigAutoRoutingMode(_common.BaseModel):
+  """When automated routing is specified, the routing will be determined by the pretrained routing model and customer provided model routing preference."""
+
+  model_routing_preference: Optional[
+      Literal['UNKNOWN', 'PRIORITIZE_QUALITY', 'BALANCED', 'PRIORITIZE_COST']
+  ] = Field(default=None, description="""The model routing preference.""")
+
+
+class GenerationConfigRoutingConfigAutoRoutingModeDict(TypedDict, total=False):
+  """When automated routing is specified, the routing will be determined by the pretrained routing model and customer provided model routing preference."""
+
+  model_routing_preference: Optional[
+      Literal['UNKNOWN', 'PRIORITIZE_QUALITY', 'BALANCED', 'PRIORITIZE_COST']
+  ]
+  """The model routing preference."""
+
+
+GenerationConfigRoutingConfigAutoRoutingModeOrDict = Union[
+    GenerationConfigRoutingConfigAutoRoutingMode,
+    GenerationConfigRoutingConfigAutoRoutingModeDict,
+]
+
+
+class GenerationConfigRoutingConfigManualRoutingMode(_common.BaseModel):
+  """When manual routing is set, the specified model will be used directly."""
+
+  model_name: Optional[str] = Field(
+      default=None,
+      description="""The model name to use. Only the public LLM models are accepted. e.g. 'gemini-1.5-pro-001'.""",
+  )
+
+
+class GenerationConfigRoutingConfigManualRoutingModeDict(
+    TypedDict, total=False
+):
+  """When manual routing is set, the specified model will be used directly."""
+
+  model_name: Optional[str]
+  """The model name to use. Only the public LLM models are accepted. e.g. 'gemini-1.5-pro-001'."""
+
+
+GenerationConfigRoutingConfigManualRoutingModeOrDict = Union[
+    GenerationConfigRoutingConfigManualRoutingMode,
+    GenerationConfigRoutingConfigManualRoutingModeDict,
+]
+
+
+class GenerationConfigRoutingConfig(_common.BaseModel):
+  """The configuration for routing the request to a specific model."""
+
+  auto_mode: Optional[GenerationConfigRoutingConfigAutoRoutingMode] = Field(
+      default=None, description="""Automated routing."""
+  )
+  manual_mode: Optional[GenerationConfigRoutingConfigManualRoutingMode] = Field(
+      default=None, description="""Manual routing."""
+  )
+
+
+class GenerationConfigRoutingConfigDict(TypedDict, total=False):
+  """The configuration for routing the request to a specific model."""
+
+  auto_mode: Optional[GenerationConfigRoutingConfigAutoRoutingModeDict]
+  """Automated routing."""
+
+  manual_mode: Optional[GenerationConfigRoutingConfigManualRoutingModeDict]
+  """Manual routing."""
+
+
+GenerationConfigRoutingConfigOrDict = Union[
+    GenerationConfigRoutingConfig, GenerationConfigRoutingConfigDict
+]
+
+
+SpeechConfigUnion = Union[SpeechConfig, str]
+
+
+SpeechConfigUnionDict = Union[SpeechConfigUnion, SpeechConfigDict]
+
+
+class GenerateContentConfig(_common.BaseModel):
+  """Optional model configuration parameters.
+
+  For more information, see `Content generation parameters
+  <https://cloud.google.com/vertex-ai/generative-ai/docs/multimodal/content-generation-parameters>`_.
+  """
+
+  http_options: Optional[HttpOptions] = Field(
+      default=None, description="""Used to override HTTP request options."""
+  )
+  system_instruction: Optional[ContentUnion] = Field(
+      default=None,
+      description="""Instructions for the model to steer it toward better performance.
+      For example, "Answer as concisely as possible" or "Don't use technical
+      terms in your response".
+      """,
+  )
+  temperature: Optional[float] = Field(
+      default=None,
+      description="""Value that controls the degree of randomness in token selection.
+      Lower temperatures are good for prompts that require a less open-ended or
+      creative response, while higher temperatures can lead to more diverse or
+      creative results.
+      """,
+  )
+  top_p: Optional[float] = Field(
+      default=None,
+      description="""Tokens are selected from the most to least probable until the sum
+      of their probabilities equals this value. Use a lower value for less
+      random responses and a higher value for more random responses.
+      """,
+  )
+  top_k: Optional[float] = Field(
+      default=None,
+      description="""For each token selection step, the ``top_k`` tokens with the
+      highest probabilities are sampled. Then tokens are further filtered based
+      on ``top_p`` with the final token selected using temperature sampling. Use
+      a lower number for less random responses and a higher number for more
+      random responses.
+      """,
+  )
+  candidate_count: Optional[int] = Field(
+      default=None,
+      description="""Number of response variations to return.
+      """,
+  )
+  max_output_tokens: Optional[int] = Field(
+      default=None,
+      description="""Maximum number of tokens that can be generated in the response.
+      """,
+  )
+  stop_sequences: Optional[list[str]] = Field(
+      default=None,
+      description="""List of strings that tells the model to stop generating text if one
+      of the strings is encountered in the response.
+      """,
+  )
+  response_logprobs: Optional[bool] = Field(
+      default=None,
+      description="""Whether to return the log probabilities of the tokens that were
+      chosen by the model at each step.
+      """,
+  )
+  logprobs: Optional[int] = Field(
+      default=None,
+      description="""Number of top candidate tokens to return the log probabilities for
+      at each generation step.
+      """,
+  )
+  presence_penalty: Optional[float] = Field(
+      default=None,
+      description="""Positive values penalize tokens that already appear in the
+      generated text, increasing the probability of generating more diverse
+      content.
+      """,
+  )
+  frequency_penalty: Optional[float] = Field(
+      default=None,
       description="""Positive values penalize tokens that repeatedly appear in the
       generated text, increasing the probability of generating more diverse
       content.
@@ -1587,20 +1885,34 @@ class GenerateContentConfig(_common.BaseModel):
       description="""The speech generation configuration.
       """,
   )
+  audio_timestamp: Optional[bool] = Field(
+      default=None,
+      description="""If enabled, audio timestamp will be included in the request to the
+       model.
+      """,
+  )
   automatic_function_calling: Optional[AutomaticFunctionCallingConfig] = Field(
       default=None,
       description="""The configuration for automatic function calling.
       """,
   )
+  thinking_config: Optional[ThinkingConfig] = Field(
+      default=None,
+      description="""The thinking features configuration.
+      """,
+  )
 
 
 class GenerateContentConfigDict(TypedDict, total=False):
-  """Class for configuring optional model parameters.
+  """Optional model configuration parameters.
 
   For more information, see `Content generation parameters
   <https://cloud.google.com/vertex-ai/generative-ai/docs/multimodal/content-generation-parameters>`_.
   """
 
+  http_options: Optional[HttpOptionsDict]
+  """Used to override HTTP request options."""
+
   system_instruction: Optional[ContentUnionDict]
   """Instructions for the model to steer it toward better performance.
       For example, "Answer as concisely as possible" or "Don't use technical
@@ -1713,18 +2025,33 @@ class GenerateContentConfigDict(TypedDict, total=False):
   """The speech generation configuration.
       """
 
+  audio_timestamp: Optional[bool]
+  """If enabled, audio timestamp will be included in the request to the
+       model.
+      """
+
   automatic_function_calling: Optional[AutomaticFunctionCallingConfigDict]
   """The configuration for automatic function calling.
       """
 
+  thinking_config: Optional[ThinkingConfigDict]
+  """The thinking features configuration.
+      """
+
 
 GenerateContentConfigOrDict = Union[
     GenerateContentConfig, GenerateContentConfigDict
 ]
 
 
+ContentListUnion = Union[list[ContentUnion], ContentUnion]
+
+
+ContentListUnionDict = Union[list[ContentUnionDict], ContentUnionDict]
+
+
 class _GenerateContentParameters(_common.BaseModel):
-  """Class for configuring the content of the request to the model."""
+  """Config for models.generate_content parameters."""
 
   model: Optional[str] = Field(
       default=None,
@@ -1744,7 +2071,7 @@ class _GenerateContentParameters(_common.BaseModel):
 
 
 class _GenerateContentParametersDict(TypedDict, total=False):
-  """Class for configuring the content of the request to the model."""
+  """Config for models.generate_content parameters."""
 
   model: Optional[str]
   """ID of the model to use. For a list of models, see `Google models
@@ -1866,7 +2193,7 @@ CitationOrDict = Union[Citation, CitationDict]
 
 
 class CitationMetadata(_common.BaseModel):
-  """Class for citation information when the model quotes another source."""
+  """Citation information when the model quotes another source."""
 
   citations: Optional[list[Citation]] = Field(
       default=None,
@@ -1878,7 +2205,7 @@ class CitationMetadata(_common.BaseModel):
 
 
 class CitationMetadataDict(TypedDict, total=False):
-  """Class for citation information when the model quotes another source."""
+  """Citation information when the model quotes another source."""
 
   citations: Optional[list[CitationDict]]
   """Contains citation information when the model directly quotes, at
@@ -2271,7 +2598,7 @@ SafetyRatingOrDict = Union[SafetyRating, SafetyRatingDict]
 
 
 class Candidate(_common.BaseModel):
-  """Class containing a response candidate generated from the model."""
+  """A response candidate generated from the model."""
 
   content: Optional[Content] = Field(
       default=None,
@@ -2319,7 +2646,7 @@ class Candidate(_common.BaseModel):
 
 
 class CandidateDict(TypedDict, total=False):
-  """Class containing a response candidate generated from the model."""
+  """A response candidate generated from the model."""
 
   content: Optional[ContentDict]
   """Contains the multi-part content of the response.
@@ -2455,7 +2782,7 @@ class GenerateContentResponse(_common.BaseModel):
       default=None, description="""Usage metadata about the response(s)."""
   )
   automatic_function_calling_history: Optional[list[Content]] = None
-  parsed: Union[pydantic.BaseModel, dict] = Field(
+  parsed: Union[pydantic.BaseModel, dict, Enum] = Field(
       default=None,
       description="""Parsed response if response_schema is provided. Not available for streaming.""",
   )
@@ -2471,20 +2798,20 @@ class GenerateContentResponse(_common.BaseModel):
       return None
     if len(self.candidates) > 1:
       logging.warning(
-          f"there are {len(self.candidates)} candidates, returning text from"
-          " the first candidate.Access response.candidates directly to get"
-          " text from other candidates."
+          f'there are {len(self.candidates)} candidates, returning text from'
+          ' the first candidate.Access response.candidates directly to get'
+          ' text from other candidates.'
       )
-    text = ""
+    text = ''
     any_text_part_text = False
     for part in self.candidates[0].content.parts:
       for field_name, field_value in part.dict(
-          exclude={"text", "thought"}
+          exclude={'text', 'thought'}
       ).items():
         if field_value is not None:
           raise ValueError(
-              "GenerateContentResponse.text only supports text parts, but got"
-              f" {field_name} part{part}"
+              'GenerateContentResponse.text only supports text parts, but got'
+              f' {field_name} part{part}'
           )
       if isinstance(part.text, str):
         if isinstance(part.thought, bool) and part.thought:
@@ -2505,8 +2832,8 @@ class GenerateContentResponse(_common.BaseModel):
       return None
     if len(self.candidates) > 1:
       logging.warning(
-          "Warning: there are multiple candidates in the response, returning"
-          " function calls from the first one."
+          'Warning: there are multiple candidates in the response, returning'
+          ' function calls from the first one.'
       )
     function_calls = [
         part.function_call
@@ -2518,16 +2845,20 @@ class GenerateContentResponse(_common.BaseModel):
 
   @classmethod
   def _from_response(
-      cls, response: dict[str, object], kwargs: dict[str, object]
+      cls, *, response: dict[str, object], kwargs: dict[str, object]
   ):
     result = super()._from_response(response, kwargs)
 
     # Handles response schema.
     response_schema = _common.get_value_by_path(
-        kwargs, ["config", "response_schema"]
+        kwargs, ['config', 'response_schema']
     )
-    if inspect.isclass(response_schema) and issubclass(
-        response_schema, pydantic.BaseModel
+    if (
+        inspect.isclass(response_schema)
+        and not (
+            isinstance(response_schema, GenericAlias)
+        )  # Needed for Python 3.9 and 3.10
+        and issubclass(response_schema, pydantic.BaseModel)
     ):
       # Pydantic schema.
       try:
@@ -2535,14 +2866,40 @@ class GenerateContentResponse(_common.BaseModel):
       # may not be a valid json per stream response
       except pydantic.ValidationError:
         pass
-
-    elif isinstance(response_schema, dict) or isinstance(
-        response_schema, pydantic.BaseModel
-    ):
-      # JSON schema.
-      try:
-        result.parsed = json.loads(result.text)
-      # may not be a valid json per stream response
+      except json.decoder.JSONDecodeError:
+        pass
+    elif isinstance(response_schema, EnumMeta):
+      # Enum with "application/json" returns response in double quotes.
+      enum_value = result.text.replace('"', '')
+      try:
+        result.parsed = response_schema(enum_value)
+      except ValueError:
+        pass
+    elif isinstance(response_schema, GenericAlias) or isinstance(
+        response_schema, type
+    ):
+
+      class Placeholder(pydantic.BaseModel):
+        placeholder: response_schema
+
+      try:
+        parsed = {'placeholder': json.loads(result.text)}
+        placeholder = Placeholder.model_validate(parsed)
+        result.parsed = placeholder.placeholder
+      except json.decoder.JSONDecodeError:
+        pass
+      except pydantic.ValidationError:
+        pass
+
+    elif isinstance(response_schema, dict) or isinstance(
+        response_schema, Schema
+    ):
+      # With just the Schema, we don't know what pydantic model the user would
+      # want the result converted to. So just return json.
+      # JSON schema.
+      try:
+        result.parsed = json.loads(result.text)
+      # may not be a valid json per stream response
       except json.decoder.JSONDecodeError:
         pass
 
@@ -2572,9 +2929,9 @@ GenerateContentResponseOrDict = Union[
 
 
 class EmbedContentConfig(_common.BaseModel):
-  """Class for configuring optional parameters for the embed_content method."""
+  """Optional parameters for the embed_content method."""
 
-  http_options: Optional[dict[str, Any]] = Field(
+  http_options: Optional[HttpOptions] = Field(
       default=None, description="""Used to override HTTP request options."""
   )
   task_type: Optional[str] = Field(
@@ -2611,9 +2968,9 @@ class EmbedContentConfig(_common.BaseModel):
 
 
 class EmbedContentConfigDict(TypedDict, total=False):
-  """Class for configuring optional parameters for the embed_content method."""
+  """Optional parameters for the embed_content method."""
 
-  http_options: Optional[dict[str, Any]]
+  http_options: Optional[HttpOptionsDict]
   """Used to override HTTP request options."""
 
   task_type: Optional[str]
@@ -2812,10 +3169,10 @@ EmbedContentResponseOrDict = Union[
 ]
 
 
-class GenerateImageConfig(_common.BaseModel):
-  """Class that represents the config for generating an image."""
+class GenerateImagesConfig(_common.BaseModel):
+  """The config for generating an images."""
 
-  http_options: Optional[dict[str, Any]] = Field(
+  http_options: Optional[HttpOptions] = Field(
       default=None, description="""Used to override HTTP request options."""
   )
   output_gcs_uri: Optional[str] = Field(
@@ -2885,20 +3242,25 @@ class GenerateImageConfig(_common.BaseModel):
   )
   add_watermark: Optional[bool] = Field(
       default=None,
-      description="""Whether to add a watermark to the generated image.
+      description="""Whether to add a watermark to the generated images.
       """,
   )
   aspect_ratio: Optional[str] = Field(
       default=None,
-      description="""Aspect ratio of the generated image.
+      description="""Aspect ratio of the generated images.
+      """,
+  )
+  enhance_prompt: Optional[bool] = Field(
+      default=None,
+      description="""Whether to use the prompt rewriting logic.
       """,
   )
 
 
-class GenerateImageConfigDict(TypedDict, total=False):
-  """Class that represents the config for generating an image."""
+class GenerateImagesConfigDict(TypedDict, total=False):
+  """The config for generating an images."""
 
-  http_options: Optional[dict[str, Any]]
+  http_options: Optional[HttpOptionsDict]
   """Used to override HTTP request options."""
 
   output_gcs_uri: Optional[str]
@@ -2955,19 +3317,25 @@ class GenerateImageConfigDict(TypedDict, total=False):
       """
 
   add_watermark: Optional[bool]
-  """Whether to add a watermark to the generated image.
+  """Whether to add a watermark to the generated images.
       """
 
   aspect_ratio: Optional[str]
-  """Aspect ratio of the generated image.
+  """Aspect ratio of the generated images.
+      """
+
+  enhance_prompt: Optional[bool]
+  """Whether to use the prompt rewriting logic.
       """
 
 
-GenerateImageConfigOrDict = Union[GenerateImageConfig, GenerateImageConfigDict]
+GenerateImagesConfigOrDict = Union[
+    GenerateImagesConfig, GenerateImagesConfigDict
+]
 
 
-class _GenerateImageParameters(_common.BaseModel):
-  """Class that represents the parameters for generating an image."""
+class _GenerateImagesParameters(_common.BaseModel):
+  """The parameters for generating images."""
 
   model: Optional[str] = Field(
       default=None,
@@ -2976,39 +3344,39 @@ class _GenerateImageParameters(_common.BaseModel):
   )
   prompt: Optional[str] = Field(
       default=None,
-      description="""Text prompt that typically describes the image to output.
+      description="""Text prompt that typically describes the images to output.
       """,
   )
-  config: Optional[GenerateImageConfig] = Field(
+  config: Optional[GenerateImagesConfig] = Field(
       default=None,
-      description="""Configuration for generating an image.
+      description="""Configuration for generating images.
       """,
   )
 
 
-class _GenerateImageParametersDict(TypedDict, total=False):
-  """Class that represents the parameters for generating an image."""
+class _GenerateImagesParametersDict(TypedDict, total=False):
+  """The parameters for generating images."""
 
   model: Optional[str]
   """ID of the model to use. For a list of models, see `Google models
     <https://cloud.google.com/vertex-ai/generative-ai/docs/learn/models>`_."""
 
   prompt: Optional[str]
-  """Text prompt that typically describes the image to output.
+  """Text prompt that typically describes the images to output.
       """
 
-  config: Optional[GenerateImageConfigDict]
-  """Configuration for generating an image.
+  config: Optional[GenerateImagesConfigDict]
+  """Configuration for generating images.
       """
 
 
-_GenerateImageParametersOrDict = Union[
-    _GenerateImageParameters, _GenerateImageParametersDict
+_GenerateImagesParametersOrDict = Union[
+    _GenerateImagesParameters, _GenerateImagesParametersDict
 ]
 
 
 class Image(_common.BaseModel):
-  """Class that represents an image."""
+  """An image."""
 
   gcs_uri: Optional[str] = Field(
       default=None,
@@ -3022,13 +3390,16 @@ class Image(_common.BaseModel):
       or the ``gcs_uri`` field but not both.
       """,
   )
+  mime_type: Optional[str] = Field(
+      default=None, description="""The MIME type of the image."""
+  )
 
   _loaded_image = None
 
   """Image."""
 
-  @staticmethod
-  def from_file(location: str) -> "Image":
+  @classmethod
+  def from_file(cls, *, location: str) -> 'Image':
     """Lazy-loads an image from a local file or Google Cloud Storage.
 
     Args:
@@ -3043,22 +3414,22 @@ class Image(_common.BaseModel):
 
     parsed_url = urllib.parse.urlparse(location)
     if (
-        parsed_url.scheme == "https"
-        and parsed_url.netloc == "storage.googleapis.com"
+        parsed_url.scheme == 'https'
+        and parsed_url.netloc == 'storage.googleapis.com'
     ):
       parsed_url = parsed_url._replace(
-          scheme="gs",
-          netloc="",
-          path=f"/{urllib.parse.unquote(parsed_url.path)}",
+          scheme='gs',
+          netloc='',
+          path=f'/{urllib.parse.unquote(parsed_url.path)}',
       )
       location = urllib.parse.urlunparse(parsed_url)
 
-    if parsed_url.scheme == "gs":
-      return Image(gcs_uri=location)
+    if parsed_url.scheme == 'gs':
+      return cls(gcs_uri=location)
 
     # Load image from local path
     image_bytes = pathlib.Path(location).read_bytes()
-    image = Image(image_bytes=image_bytes)
+    image = cls(image_bytes=image_bytes)
     return image
 
   def show(self):
@@ -3071,15 +3442,11 @@ class Image(_common.BaseModel):
     except ImportError:
       IPython_display = None
 
-    try:
-      from PIL import Image as PIL_Image
-    except ImportError:
-      PIL_Image = None
-    if PIL_Image and IPython_display:
+    if IPython_display:
       IPython_display.display(self._pil_image)
 
   @property
-  def _pil_image(self) -> "PIL_Image.Image":
+  def _pil_image(self) -> 'PIL_Image.Image':
     try:
       from PIL import Image as PIL_Image
     except ImportError:
@@ -3089,8 +3456,8 @@ class Image(_common.BaseModel):
     if self._loaded_image is None:
       if not PIL_Image:
         raise RuntimeError(
-            "The PIL module is not available. Please install the Pillow"
-            " package."
+            'The PIL module is not available. Please install the Pillow'
+            ' package. `pip install pillow`'
         )
       self._loaded_image = PIL_Image.open(io.BytesIO(self.image_bytes))
     return self._loaded_image
@@ -3106,36 +3473,34 @@ class Image(_common.BaseModel):
     pathlib.Path(location).write_bytes(self.image_bytes)
 
 
-Modality = Literal["MODALITY_UNSPECIFIED", "TEXT", "IMAGE", "AUDIO"]
-
 JOB_STATES_SUCCEEDED_VERTEX = [
-    "JOB_STATE_SUCCEEDED",
+    'JOB_STATE_SUCCEEDED',
 ]
 
 JOB_STATES_SUCCEEDED_MLDEV = [
-    "ACTIVE",
+    'ACTIVE',
 ]
 
 JOB_STATES_SUCCEEDED = JOB_STATES_SUCCEEDED_VERTEX + JOB_STATES_SUCCEEDED_MLDEV
 
 
 JOB_STATES_ENDED_VERTEX = [
-    "JOB_STATE_SUCCEEDED",
-    "JOB_STATE_FAILED",
-    "JOB_STATE_CANCELLED",
-    "JOB_STATE_EXPIRED",
+    'JOB_STATE_SUCCEEDED',
+    'JOB_STATE_FAILED',
+    'JOB_STATE_CANCELLED',
+    'JOB_STATE_EXPIRED',
 ]
 
 JOB_STATES_ENDED_MLDEV = [
-    "ACTIVE",
-    "FAILED",
+    'ACTIVE',
+    'FAILED',
 ]
 
 JOB_STATES_ENDED = JOB_STATES_ENDED_VERTEX + JOB_STATES_ENDED_MLDEV
 
 
 class ImageDict(TypedDict, total=False):
-  """Class that represents an image."""
+  """An image."""
 
   gcs_uri: Optional[str]
   """The Cloud Storage URI of the image. ``Image`` can contain a value
@@ -3147,12 +3512,15 @@ class ImageDict(TypedDict, total=False):
       or the ``gcs_uri`` field but not both.
       """
 
+  mime_type: Optional[str]
+  """The MIME type of the image."""
+
 
 ImageOrDict = Union[Image, ImageDict]
 
 
 class GeneratedImage(_common.BaseModel):
-  """Class that represents an output image."""
+  """An output image."""
 
   image: Optional[Image] = Field(
       default=None,
@@ -3168,7 +3536,7 @@ class GeneratedImage(_common.BaseModel):
 
 
 class GeneratedImageDict(TypedDict, total=False):
-  """Class that represents an output image."""
+  """An output image."""
 
   image: Optional[ImageDict]
   """The output image data.
@@ -3183,8 +3551,8 @@ class GeneratedImageDict(TypedDict, total=False):
 GeneratedImageOrDict = Union[GeneratedImage, GeneratedImageDict]
 
 
-class GenerateImageResponse(_common.BaseModel):
-  """Class that represents the output image response."""
+class GenerateImagesResponse(_common.BaseModel):
+  """The output images response."""
 
   generated_images: Optional[list[GeneratedImage]] = Field(
       default=None,
@@ -3193,16 +3561,16 @@ class GenerateImageResponse(_common.BaseModel):
   )
 
 
-class GenerateImageResponseDict(TypedDict, total=False):
-  """Class that represents the output image response."""
+class GenerateImagesResponseDict(TypedDict, total=False):
+  """The output images response."""
 
   generated_images: Optional[list[GeneratedImageDict]]
   """List of generated images.
       """
 
 
-GenerateImageResponseOrDict = Union[
-    GenerateImageResponse, GenerateImageResponseDict
+GenerateImagesResponseOrDict = Union[
+    GenerateImagesResponse, GenerateImagesResponseDict
 ]
 
 
@@ -3387,7 +3755,7 @@ _ReferenceImageAPIOrDict = Union[_ReferenceImageAPI, _ReferenceImageAPIDict]
 class EditImageConfig(_common.BaseModel):
   """Configuration for editing an image."""
 
-  http_options: Optional[dict[str, Any]] = Field(
+  http_options: Optional[HttpOptions] = Field(
       default=None, description="""Used to override HTTP request options."""
   )
   output_gcs_uri: Optional[str] = Field(
@@ -3464,7 +3832,7 @@ class EditImageConfig(_common.BaseModel):
 class EditImageConfigDict(TypedDict, total=False):
   """Configuration for editing an image."""
 
-  http_options: Optional[dict[str, Any]]
+  http_options: Optional[HttpOptionsDict]
   """Used to override HTTP request options."""
 
   output_gcs_uri: Optional[str]
@@ -3591,7 +3959,7 @@ class _UpscaleImageAPIConfig(_common.BaseModel):
   to be modifiable or exposed to users in the SDK method.
   """
 
-  http_options: Optional[dict[str, Any]] = Field(
+  http_options: Optional[HttpOptions] = Field(
       default=None, description="""Used to override HTTP request options."""
   )
   include_rai_reason: Optional[bool] = Field(
@@ -3619,7 +3987,7 @@ class _UpscaleImageAPIConfigDict(TypedDict, total=False):
   to be modifiable or exposed to users in the SDK method.
   """
 
-  http_options: Optional[dict[str, Any]]
+  http_options: Optional[HttpOptionsDict]
   """Used to override HTTP request options."""
 
   include_rai_reason: Optional[bool]
@@ -3702,9 +4070,30 @@ UpscaleImageResponseOrDict = Union[
 ]
 
 
+class GetModelConfig(_common.BaseModel):
+  """Optional parameters for models.get method."""
+
+  http_options: Optional[HttpOptions] = Field(
+      default=None, description="""Used to override HTTP request options."""
+  )
+
+
+class GetModelConfigDict(TypedDict, total=False):
+  """Optional parameters for models.get method."""
+
+  http_options: Optional[HttpOptionsDict]
+  """Used to override HTTP request options."""
+
+
+GetModelConfigOrDict = Union[GetModelConfig, GetModelConfigDict]
+
+
 class _GetModelParameters(_common.BaseModel):
 
   model: Optional[str] = Field(default=None, description="""""")
+  config: Optional[GetModelConfig] = Field(
+      default=None, description="""Optional parameters for the request."""
+  )
 
 
 class _GetModelParametersDict(TypedDict, total=False):
@@ -3712,6 +4101,9 @@ class _GetModelParametersDict(TypedDict, total=False):
   model: Optional[str]
   """"""
 
+  config: Optional[GetModelConfigDict]
+  """Optional parameters for the request."""
+
 
 _GetModelParametersOrDict = Union[_GetModelParameters, _GetModelParametersDict]
 
@@ -3863,7 +4255,7 @@ ModelOrDict = Union[Model, ModelDict]
 
 class ListModelsConfig(_common.BaseModel):
 
-  http_options: Optional[dict[str, Any]] = Field(
+  http_options: Optional[HttpOptions] = Field(
       default=None, description="""Used to override HTTP request options."""
   )
   page_size: Optional[int] = Field(default=None, description="""""")
@@ -3877,7 +4269,7 @@ class ListModelsConfig(_common.BaseModel):
 
 class ListModelsConfigDict(TypedDict, total=False):
 
-  http_options: Optional[dict[str, Any]]
+  http_options: Optional[HttpOptionsDict]
   """Used to override HTTP request options."""
 
   page_size: Optional[int]
@@ -3932,12 +4324,18 @@ ListModelsResponseOrDict = Union[ListModelsResponse, ListModelsResponseDict]
 
 class UpdateModelConfig(_common.BaseModel):
 
+  http_options: Optional[HttpOptions] = Field(
+      default=None, description="""Used to override HTTP request options."""
+  )
   display_name: Optional[str] = Field(default=None, description="""""")
   description: Optional[str] = Field(default=None, description="""""")
 
 
 class UpdateModelConfigDict(TypedDict, total=False):
 
+  http_options: Optional[HttpOptionsDict]
+  """Used to override HTTP request options."""
+
   display_name: Optional[str]
   """"""
 
@@ -3968,9 +4366,28 @@ _UpdateModelParametersOrDict = Union[
 ]
 
 
+class DeleteModelConfig(_common.BaseModel):
+
+  http_options: Optional[HttpOptions] = Field(
+      default=None, description="""Used to override HTTP request options."""
+  )
+
+
+class DeleteModelConfigDict(TypedDict, total=False):
+
+  http_options: Optional[HttpOptionsDict]
+  """Used to override HTTP request options."""
+
+
+DeleteModelConfigOrDict = Union[DeleteModelConfig, DeleteModelConfigDict]
+
+
 class _DeleteModelParameters(_common.BaseModel):
 
   model: Optional[str] = Field(default=None, description="""""")
+  config: Optional[DeleteModelConfig] = Field(
+      default=None, description="""Optional parameters for the request."""
+  )
 
 
 class _DeleteModelParametersDict(TypedDict, total=False):
@@ -3978,6 +4395,9 @@ class _DeleteModelParametersDict(TypedDict, total=False):
   model: Optional[str]
   """"""
 
+  config: Optional[DeleteModelConfigDict]
+  """Optional parameters for the request."""
+
 
 _DeleteModelParametersOrDict = Union[
     _DeleteModelParameters, _DeleteModelParametersDict
@@ -4109,7 +4529,7 @@ GenerationConfigOrDict = Union[GenerationConfig, GenerationConfigDict]
 class CountTokensConfig(_common.BaseModel):
   """Config for the count_tokens method."""
 
-  http_options: Optional[dict[str, Any]] = Field(
+  http_options: Optional[HttpOptions] = Field(
       default=None, description="""Used to override HTTP request options."""
   )
   system_instruction: Optional[ContentUnion] = Field(
@@ -4134,7 +4554,7 @@ class CountTokensConfig(_common.BaseModel):
 class CountTokensConfigDict(TypedDict, total=False):
   """Config for the count_tokens method."""
 
-  http_options: Optional[dict[str, Any]]
+  http_options: Optional[HttpOptionsDict]
   """Used to override HTTP request options."""
 
   system_instruction: Optional[ContentUnionDict]
@@ -4218,7 +4638,7 @@ CountTokensResponseOrDict = Union[CountTokensResponse, CountTokensResponseDict]
 class ComputeTokensConfig(_common.BaseModel):
   """Optional parameters for computing tokens."""
 
-  http_options: Optional[dict[str, Any]] = Field(
+  http_options: Optional[HttpOptions] = Field(
       default=None, description="""Used to override HTTP request options."""
   )
 
@@ -4226,7 +4646,7 @@ class ComputeTokensConfig(_common.BaseModel):
 class ComputeTokensConfigDict(TypedDict, total=False):
   """Optional parameters for computing tokens."""
 
-  http_options: Optional[dict[str, Any]]
+  http_options: Optional[HttpOptionsDict]
   """Used to override HTTP request options."""
 
 
@@ -4326,7 +4746,7 @@ ComputeTokensResponseOrDict = Union[
 class GetTuningJobConfig(_common.BaseModel):
   """Optional parameters for tunings.get method."""
 
-  http_options: Optional[dict[str, Any]] = Field(
+  http_options: Optional[HttpOptions] = Field(
       default=None, description="""Used to override HTTP request options."""
   )
 
@@ -4334,7 +4754,7 @@ class GetTuningJobConfig(_common.BaseModel):
 class GetTuningJobConfigDict(TypedDict, total=False):
   """Optional parameters for tunings.get method."""
 
-  http_options: Optional[dict[str, Any]]
+  http_options: Optional[HttpOptionsDict]
   """Used to override HTTP request options."""
 
 
@@ -4953,6 +5373,41 @@ class EncryptionSpecDict(TypedDict, total=False):
 EncryptionSpecOrDict = Union[EncryptionSpec, EncryptionSpecDict]
 
 
+class PartnerModelTuningSpec(_common.BaseModel):
+  """Tuning spec for Partner models."""
+
+  hyper_parameters: Optional[dict[str, Any]] = Field(
+      default=None,
+      description="""Hyperparameters for tuning. The accepted hyper_parameters and their valid range of values will differ depending on the base model.""",
+  )
+  training_dataset_uri: Optional[str] = Field(
+      default=None,
+      description="""Required. Cloud Storage path to file containing training dataset for tuning. The dataset must be formatted as a JSONL file.""",
+  )
+  validation_dataset_uri: Optional[str] = Field(
+      default=None,
+      description="""Optional. Cloud Storage path to file containing validation dataset for tuning. The dataset must be formatted as a JSONL file.""",
+  )
+
+
+class PartnerModelTuningSpecDict(TypedDict, total=False):
+  """Tuning spec for Partner models."""
+
+  hyper_parameters: Optional[dict[str, Any]]
+  """Hyperparameters for tuning. The accepted hyper_parameters and their valid range of values will differ depending on the base model."""
+
+  training_dataset_uri: Optional[str]
+  """Required. Cloud Storage path to file containing training dataset for tuning. The dataset must be formatted as a JSONL file."""
+
+  validation_dataset_uri: Optional[str]
+  """Optional. Cloud Storage path to file containing validation dataset for tuning. The dataset must be formatted as a JSONL file."""
+
+
+PartnerModelTuningSpecOrDict = Union[
+    PartnerModelTuningSpec, PartnerModelTuningSpecDict
+]
+
+
 class DistillationHyperParameters(_common.BaseModel):
   """Hyperparameters for Distillation."""
 
@@ -5048,41 +5503,6 @@ class DistillationSpecDict(TypedDict, total=False):
 DistillationSpecOrDict = Union[DistillationSpec, DistillationSpecDict]
 
 
-class PartnerModelTuningSpec(_common.BaseModel):
-  """Tuning spec for Partner models."""
-
-  hyper_parameters: Optional[dict[str, Any]] = Field(
-      default=None,
-      description="""Hyperparameters for tuning. The accepted hyper_parameters and their valid range of values will differ depending on the base model.""",
-  )
-  training_dataset_uri: Optional[str] = Field(
-      default=None,
-      description="""Required. Cloud Storage path to file containing training dataset for tuning. The dataset must be formatted as a JSONL file.""",
-  )
-  validation_dataset_uri: Optional[str] = Field(
-      default=None,
-      description="""Optional. Cloud Storage path to file containing validation dataset for tuning. The dataset must be formatted as a JSONL file.""",
-  )
-
-
-class PartnerModelTuningSpecDict(TypedDict, total=False):
-  """Tuning spec for Partner models."""
-
-  hyper_parameters: Optional[dict[str, Any]]
-  """Hyperparameters for tuning. The accepted hyper_parameters and their valid range of values will differ depending on the base model."""
-
-  training_dataset_uri: Optional[str]
-  """Required. Cloud Storage path to file containing training dataset for tuning. The dataset must be formatted as a JSONL file."""
-
-  validation_dataset_uri: Optional[str]
-  """Optional. Cloud Storage path to file containing validation dataset for tuning. The dataset must be formatted as a JSONL file."""
-
-
-PartnerModelTuningSpecOrDict = Union[
-    PartnerModelTuningSpec, PartnerModelTuningSpecDict
-]
-
-
 class TuningJob(_common.BaseModel):
   """A tuning job."""
 
@@ -5124,7 +5544,7 @@ class TuningJob(_common.BaseModel):
   )
   tuned_model: Optional[TunedModel] = Field(
       default=None,
-      description="""Output only. The tuned model resources assiociated with this TuningJob.""",
+      description="""Output only. The tuned model resources associated with this TuningJob.""",
   )
   supervised_tuning_spec: Optional[SupervisedTuningSpec] = Field(
       default=None, description="""Tuning Spec for Supervised Fine Tuning."""
@@ -5137,16 +5557,12 @@ class TuningJob(_common.BaseModel):
       default=None,
       description="""Customer-managed encryption key options for a TuningJob. If this is set, then all resources created by the TuningJob will be encrypted with the provided encryption key.""",
   )
-  distillation_spec: Optional[DistillationSpec] = Field(
-      default=None, description="""Tuning Spec for Distillation."""
-  )
   partner_model_tuning_spec: Optional[PartnerModelTuningSpec] = Field(
       default=None,
       description="""Tuning Spec for open sourced and third party Partner models.""",
   )
-  pipeline_job: Optional[str] = Field(
-      default=None,
-      description="""Output only. The resource name of the PipelineJob associated with the TuningJob. Format: `projects/{project}/locations/{location}/pipelineJobs/{pipeline_job}`.""",
+  distillation_spec: Optional[DistillationSpec] = Field(
+      default=None, description="""Tuning Spec for Distillation."""
   )
   experiment: Optional[str] = Field(
       default=None,
@@ -5156,6 +5572,10 @@ class TuningJob(_common.BaseModel):
       default=None,
       description="""Optional. The labels with user-defined metadata to organize TuningJob and generated resources such as Model and Endpoint. Label keys and values can be no longer than 64 characters (Unicode codepoints), can only contain lowercase letters, numeric characters, underscores and dashes. International characters are allowed. See https://goo.gl/xmQnxf for more information and examples of labels.""",
   )
+  pipeline_job: Optional[str] = Field(
+      default=None,
+      description="""Output only. The resource name of the PipelineJob associated with the TuningJob. Format: `projects/{project}/locations/{location}/pipelineJobs/{pipeline_job}`.""",
+  )
   tuned_model_display_name: Optional[str] = Field(
       default=None,
       description="""Optional. The display name of the TunedModel. The name can be up to 128 characters long and can consist of any UTF-8 characters.""",
@@ -5203,7 +5623,7 @@ class TuningJobDict(TypedDict, total=False):
   """The base model that is being tuned, e.g., "gemini-1.0-pro-002". ."""
 
   tuned_model: Optional[TunedModelDict]
-  """Output only. The tuned model resources assiociated with this TuningJob."""
+  """Output only. The tuned model resources associated with this TuningJob."""
 
   supervised_tuning_spec: Optional[SupervisedTuningSpecDict]
   """Tuning Spec for Supervised Fine Tuning."""
@@ -5214,14 +5634,11 @@ class TuningJobDict(TypedDict, total=False):
   encryption_spec: Optional[EncryptionSpecDict]
   """Customer-managed encryption key options for a TuningJob. If this is set, then all resources created by the TuningJob will be encrypted with the provided encryption key."""
 
-  distillation_spec: Optional[DistillationSpecDict]
-  """Tuning Spec for Distillation."""
-
   partner_model_tuning_spec: Optional[PartnerModelTuningSpecDict]
   """Tuning Spec for open sourced and third party Partner models."""
 
-  pipeline_job: Optional[str]
-  """Output only. The resource name of the PipelineJob associated with the TuningJob. Format: `projects/{project}/locations/{location}/pipelineJobs/{pipeline_job}`."""
+  distillation_spec: Optional[DistillationSpecDict]
+  """Tuning Spec for Distillation."""
 
   experiment: Optional[str]
   """Output only. The Experiment associated with this TuningJob."""
@@ -5229,6 +5646,9 @@ class TuningJobDict(TypedDict, total=False):
   labels: Optional[dict[str, str]]
   """Optional. The labels with user-defined metadata to organize TuningJob and generated resources such as Model and Endpoint. Label keys and values can be no longer than 64 characters (Unicode codepoints), can only contain lowercase letters, numeric characters, underscores and dashes. International characters are allowed. See https://goo.gl/xmQnxf for more information and examples of labels."""
 
+  pipeline_job: Optional[str]
+  """Output only. The resource name of the PipelineJob associated with the TuningJob. Format: `projects/{project}/locations/{location}/pipelineJobs/{pipeline_job}`."""
+
   tuned_model_display_name: Optional[str]
   """Optional. The display name of the TunedModel. The name can be up to 128 characters long and can consist of any UTF-8 characters."""
 
@@ -5239,6 +5659,9 @@ TuningJobOrDict = Union[TuningJob, TuningJobDict]
 class ListTuningJobsConfig(_common.BaseModel):
   """Configuration for the list tuning jobs method."""
 
+  http_options: Optional[HttpOptions] = Field(
+      default=None, description="""Used to override HTTP request options."""
+  )
   page_size: Optional[int] = Field(default=None, description="""""")
   page_token: Optional[str] = Field(default=None, description="""""")
   filter: Optional[str] = Field(default=None, description="""""")
@@ -5247,6 +5670,9 @@ class ListTuningJobsConfig(_common.BaseModel):
 class ListTuningJobsConfigDict(TypedDict, total=False):
   """Configuration for the list tuning jobs method."""
 
+  http_options: Optional[HttpOptionsDict]
+  """Used to override HTTP request options."""
+
   page_size: Optional[int]
   """"""
 
@@ -5321,208 +5747,43 @@ class TuningExample(_common.BaseModel):
 
 class TuningExampleDict(TypedDict, total=False):
 
-  text_input: Optional[str]
-  """Text model input."""
-
-  output: Optional[str]
-  """The expected model output."""
-
-
-TuningExampleOrDict = Union[TuningExample, TuningExampleDict]
-
-
-class TuningDataset(_common.BaseModel):
-  """Supervised fune-tuning training dataset."""
-
-  gcs_uri: Optional[str] = Field(
-      default=None,
-      description="""GCS URI of the file containing training dataset in JSONL format.""",
-  )
-  examples: Optional[list[TuningExample]] = Field(
-      default=None,
-      description="""Inline examples with simple input/output text.""",
-  )
-
-
-class TuningDatasetDict(TypedDict, total=False):
-  """Supervised fune-tuning training dataset."""
-
-  gcs_uri: Optional[str]
-  """GCS URI of the file containing training dataset in JSONL format."""
-
-  examples: Optional[list[TuningExampleDict]]
-  """Inline examples with simple input/output text."""
-
-
-TuningDatasetOrDict = Union[TuningDataset, TuningDatasetDict]
-
-
-class TuningValidationDataset(_common.BaseModel):
-
-  gcs_uri: Optional[str] = Field(
-      default=None,
-      description="""GCS URI of the file containing validation dataset in JSONL format.""",
-  )
-
-
-class TuningValidationDatasetDict(TypedDict, total=False):
-
-  gcs_uri: Optional[str]
-  """GCS URI of the file containing validation dataset in JSONL format."""
-
-
-TuningValidationDatasetOrDict = Union[
-    TuningValidationDataset, TuningValidationDatasetDict
-]
-
-
-class CreateTuningJobConfig(_common.BaseModel):
-  """Supervised fine-tuning job creation request - optional fields."""
-
-  http_options: Optional[dict[str, Any]] = Field(
-      default=None, description="""Used to override HTTP request options."""
-  )
-  validation_dataset: Optional[TuningValidationDataset] = Field(
-      default=None,
-      description="""Cloud Storage path to file containing training dataset for tuning. The dataset must be formatted as a JSONL file.""",
-  )
-  tuned_model_display_name: Optional[str] = Field(
-      default=None,
-      description="""The display name of the tuned Model. The name can be up to 128 characters long and can consist of any UTF-8 characters.""",
-  )
-  description: Optional[str] = Field(
-      default=None, description="""The description of the TuningJob"""
-  )
-  epoch_count: Optional[int] = Field(
-      default=None,
-      description="""Number of complete passes the model makes over the entire training dataset during training.""",
-  )
-  learning_rate_multiplier: Optional[float] = Field(
-      default=None,
-      description="""Multiplier for adjusting the default learning rate.""",
-  )
-  adapter_size: Optional[AdapterSize] = Field(
-      default=None, description="""Adapter size for tuning."""
-  )
-  batch_size: Optional[int] = Field(
-      default=None,
-      description="""The batch size hyperparameter for tuning. If not set, a default of 4 or 16 will be used based on the number of training examples.""",
-  )
-  learning_rate: Optional[float] = Field(
-      default=None,
-      description="""The learning rate hyperparameter for tuning. If not set, a default of 0.001 or 0.0002 will be calculated based on the number of training examples.""",
-  )
-
-
-class CreateTuningJobConfigDict(TypedDict, total=False):
-  """Supervised fine-tuning job creation request - optional fields."""
-
-  http_options: Optional[dict[str, Any]]
-  """Used to override HTTP request options."""
-
-  validation_dataset: Optional[TuningValidationDatasetDict]
-  """Cloud Storage path to file containing training dataset for tuning. The dataset must be formatted as a JSONL file."""
-
-  tuned_model_display_name: Optional[str]
-  """The display name of the tuned Model. The name can be up to 128 characters long and can consist of any UTF-8 characters."""
-
-  description: Optional[str]
-  """The description of the TuningJob"""
-
-  epoch_count: Optional[int]
-  """Number of complete passes the model makes over the entire training dataset during training."""
-
-  learning_rate_multiplier: Optional[float]
-  """Multiplier for adjusting the default learning rate."""
-
-  adapter_size: Optional[AdapterSize]
-  """Adapter size for tuning."""
-
-  batch_size: Optional[int]
-  """The batch size hyperparameter for tuning. If not set, a default of 4 or 16 will be used based on the number of training examples."""
-
-  learning_rate: Optional[float]
-  """The learning rate hyperparameter for tuning. If not set, a default of 0.001 or 0.0002 will be calculated based on the number of training examples."""
-
-
-CreateTuningJobConfigOrDict = Union[
-    CreateTuningJobConfig, CreateTuningJobConfigDict
-]
-
-
-class _CreateTuningJobParameters(_common.BaseModel):
-  """Supervised fine-tuning job creation parameters - optional fields."""
-
-  base_model: Optional[str] = Field(
-      default=None,
-      description="""The base model that is being tuned, e.g., "gemini-1.0-pro-002".""",
-  )
-  training_dataset: Optional[TuningDataset] = Field(
-      default=None,
-      description="""Cloud Storage path to file containing training dataset for tuning. The dataset must be formatted as a JSONL file.""",
-  )
-  config: Optional[CreateTuningJobConfig] = Field(
-      default=None, description="""Configuration for the tuning job."""
-  )
-
-
-class _CreateTuningJobParametersDict(TypedDict, total=False):
-  """Supervised fine-tuning job creation parameters - optional fields."""
-
-  base_model: Optional[str]
-  """The base model that is being tuned, e.g., "gemini-1.0-pro-002"."""
-
-  training_dataset: Optional[TuningDatasetDict]
-  """Cloud Storage path to file containing training dataset for tuning. The dataset must be formatted as a JSONL file."""
-
-  config: Optional[CreateTuningJobConfigDict]
-  """Configuration for the tuning job."""
-
-
-_CreateTuningJobParametersOrDict = Union[
-    _CreateTuningJobParameters, _CreateTuningJobParametersDict
-]
-
-
-class TuningJobOrOperation(_common.BaseModel):
-  """A tuning job or an long-running-operation that resolves to a tuning job."""
-
-  tuning_job: Optional[TuningJob] = Field(default=None, description="""""")
-
-
-class TuningJobOrOperationDict(TypedDict, total=False):
-  """A tuning job or an long-running-operation that resolves to a tuning job."""
-
-  tuning_job: Optional[TuningJobDict]
-  """"""
+  text_input: Optional[str]
+  """Text model input."""
 
+  output: Optional[str]
+  """The expected model output."""
 
-TuningJobOrOperationOrDict = Union[
-    TuningJobOrOperation, TuningJobOrOperationDict
-]
+
+TuningExampleOrDict = Union[TuningExample, TuningExampleDict]
 
 
-class DistillationDataset(_common.BaseModel):
-  """Training dataset."""
+class TuningDataset(_common.BaseModel):
+  """Supervised fine-tuning training dataset."""
 
   gcs_uri: Optional[str] = Field(
       default=None,
       description="""GCS URI of the file containing training dataset in JSONL format.""",
   )
+  examples: Optional[list[TuningExample]] = Field(
+      default=None,
+      description="""Inline examples with simple input/output text.""",
+  )
 
 
-class DistillationDatasetDict(TypedDict, total=False):
-  """Training dataset."""
+class TuningDatasetDict(TypedDict, total=False):
+  """Supervised fine-tuning training dataset."""
 
   gcs_uri: Optional[str]
   """GCS URI of the file containing training dataset in JSONL format."""
 
+  examples: Optional[list[TuningExampleDict]]
+  """Inline examples with simple input/output text."""
+
 
-DistillationDatasetOrDict = Union[DistillationDataset, DistillationDatasetDict]
+TuningDatasetOrDict = Union[TuningDataset, TuningDatasetDict]
 
 
-class DistillationValidationDataset(_common.BaseModel):
-  """Validation dataset."""
+class TuningValidationDataset(_common.BaseModel):
 
   gcs_uri: Optional[str] = Field(
       default=None,
@@ -5530,32 +5791,34 @@ class DistillationValidationDataset(_common.BaseModel):
   )
 
 
-class DistillationValidationDatasetDict(TypedDict, total=False):
-  """Validation dataset."""
+class TuningValidationDatasetDict(TypedDict, total=False):
 
   gcs_uri: Optional[str]
   """GCS URI of the file containing validation dataset in JSONL format."""
 
 
-DistillationValidationDatasetOrDict = Union[
-    DistillationValidationDataset, DistillationValidationDatasetDict
+TuningValidationDatasetOrDict = Union[
+    TuningValidationDataset, TuningValidationDatasetDict
 ]
 
 
-class CreateDistillationJobConfig(_common.BaseModel):
-  """Distillation job creation request - optional fields."""
+class CreateTuningJobConfig(_common.BaseModel):
+  """Supervised fine-tuning job creation request - optional fields."""
 
-  http_options: Optional[dict[str, Any]] = Field(
+  http_options: Optional[HttpOptions] = Field(
       default=None, description="""Used to override HTTP request options."""
   )
-  validation_dataset: Optional[DistillationValidationDataset] = Field(
+  validation_dataset: Optional[TuningValidationDataset] = Field(
       default=None,
-      description="""Cloud Storage path to file containing validation dataset for tuning. The dataset must be formatted as a JSONL file.""",
+      description="""Cloud Storage path to file containing training dataset for tuning. The dataset must be formatted as a JSONL file.""",
   )
   tuned_model_display_name: Optional[str] = Field(
       default=None,
       description="""The display name of the tuned Model. The name can be up to 128 characters long and can consist of any UTF-8 characters.""",
   )
+  description: Optional[str] = Field(
+      default=None, description="""The description of the TuningJob"""
+  )
   epoch_count: Optional[int] = Field(
       default=None,
       description="""Number of complete passes the model makes over the entire training dataset during training.""",
@@ -5567,24 +5830,31 @@ class CreateDistillationJobConfig(_common.BaseModel):
   adapter_size: Optional[AdapterSize] = Field(
       default=None, description="""Adapter size for tuning."""
   )
-  pipeline_root_directory: Optional[str] = Field(
+  batch_size: Optional[int] = Field(
+      default=None,
+      description="""The batch size hyperparameter for tuning. If not set, a default of 4 or 16 will be used based on the number of training examples.""",
+  )
+  learning_rate: Optional[float] = Field(
       default=None,
-      description="""The resource name of the PipelineJob associated with the TuningJob. Format:`projects/{project}/locations/{location}/pipelineJobs/{pipeline_job}`.""",
+      description="""The learning rate hyperparameter for tuning. If not set, a default of 0.001 or 0.0002 will be calculated based on the number of training examples.""",
   )
 
 
-class CreateDistillationJobConfigDict(TypedDict, total=False):
-  """Distillation job creation request - optional fields."""
+class CreateTuningJobConfigDict(TypedDict, total=False):
+  """Supervised fine-tuning job creation request - optional fields."""
 
-  http_options: Optional[dict[str, Any]]
+  http_options: Optional[HttpOptionsDict]
   """Used to override HTTP request options."""
 
-  validation_dataset: Optional[DistillationValidationDatasetDict]
-  """Cloud Storage path to file containing validation dataset for tuning. The dataset must be formatted as a JSONL file."""
+  validation_dataset: Optional[TuningValidationDatasetDict]
+  """Cloud Storage path to file containing training dataset for tuning. The dataset must be formatted as a JSONL file."""
 
   tuned_model_display_name: Optional[str]
   """The display name of the tuned Model. The name can be up to 128 characters long and can consist of any UTF-8 characters."""
 
+  description: Optional[str]
+  """The description of the TuningJob"""
+
   epoch_count: Optional[int]
   """Number of complete passes the model makes over the entire training dataset during training."""
 
@@ -5594,60 +5864,74 @@ class CreateDistillationJobConfigDict(TypedDict, total=False):
   adapter_size: Optional[AdapterSize]
   """Adapter size for tuning."""
 
-  pipeline_root_directory: Optional[str]
-  """The resource name of the PipelineJob associated with the TuningJob. Format:`projects/{project}/locations/{location}/pipelineJobs/{pipeline_job}`."""
+  batch_size: Optional[int]
+  """The batch size hyperparameter for tuning. If not set, a default of 4 or 16 will be used based on the number of training examples."""
+
+  learning_rate: Optional[float]
+  """The learning rate hyperparameter for tuning. If not set, a default of 0.001 or 0.0002 will be calculated based on the number of training examples."""
 
 
-CreateDistillationJobConfigOrDict = Union[
-    CreateDistillationJobConfig, CreateDistillationJobConfigDict
+CreateTuningJobConfigOrDict = Union[
+    CreateTuningJobConfig, CreateTuningJobConfigDict
 ]
 
 
-class _CreateDistillationJobParameters(_common.BaseModel):
-  """Distillation job creation parameters - optional fields."""
+class _CreateTuningJobParameters(_common.BaseModel):
+  """Supervised fine-tuning job creation parameters - optional fields."""
 
-  student_model: Optional[str] = Field(
-      default=None,
-      description="""The student model that is being tuned, e.g. ``google/gemma-2b-1.1-it``.""",
-  )
-  teacher_model: Optional[str] = Field(
+  base_model: Optional[str] = Field(
       default=None,
-      description="""The teacher model that is being distilled from, e.g. ``gemini-1.0-pro-002``.""",
+      description="""The base model that is being tuned, e.g., "gemini-1.0-pro-002".""",
   )
-  training_dataset: Optional[DistillationDataset] = Field(
+  training_dataset: Optional[TuningDataset] = Field(
       default=None,
       description="""Cloud Storage path to file containing training dataset for tuning. The dataset must be formatted as a JSONL file.""",
   )
-  config: Optional[CreateDistillationJobConfig] = Field(
-      default=None, description="""Configuration for the distillation job."""
+  config: Optional[CreateTuningJobConfig] = Field(
+      default=None, description="""Configuration for the tuning job."""
   )
 
 
-class _CreateDistillationJobParametersDict(TypedDict, total=False):
-  """Distillation job creation parameters - optional fields."""
-
-  student_model: Optional[str]
-  """The student model that is being tuned, e.g. ``google/gemma-2b-1.1-it``."""
+class _CreateTuningJobParametersDict(TypedDict, total=False):
+  """Supervised fine-tuning job creation parameters - optional fields."""
 
-  teacher_model: Optional[str]
-  """The teacher model that is being distilled from, e.g. ``gemini-1.0-pro-002``."""
+  base_model: Optional[str]
+  """The base model that is being tuned, e.g., "gemini-1.0-pro-002"."""
 
-  training_dataset: Optional[DistillationDatasetDict]
+  training_dataset: Optional[TuningDatasetDict]
   """Cloud Storage path to file containing training dataset for tuning. The dataset must be formatted as a JSONL file."""
 
-  config: Optional[CreateDistillationJobConfigDict]
-  """Configuration for the distillation job."""
+  config: Optional[CreateTuningJobConfigDict]
+  """Configuration for the tuning job."""
+
+
+_CreateTuningJobParametersOrDict = Union[
+    _CreateTuningJobParameters, _CreateTuningJobParametersDict
+]
+
+
+class TuningJobOrOperation(_common.BaseModel):
+  """A tuning job or an long-running-operation that resolves to a tuning job."""
+
+  tuning_job: Optional[TuningJob] = Field(default=None, description="""""")
+
+
+class TuningJobOrOperationDict(TypedDict, total=False):
+  """A tuning job or an long-running-operation that resolves to a tuning job."""
 
+  tuning_job: Optional[TuningJobDict]
+  """"""
 
-_CreateDistillationJobParametersOrDict = Union[
-    _CreateDistillationJobParameters, _CreateDistillationJobParametersDict
+
+TuningJobOrOperationOrDict = Union[
+    TuningJobOrOperation, TuningJobOrOperationDict
 ]
 
 
 class CreateCachedContentConfig(_common.BaseModel):
-  """Class for configuring optional cached content creation parameters."""
+  """Optional configuration for cached content creation."""
 
-  http_options: Optional[dict[str, Any]] = Field(
+  http_options: Optional[HttpOptions] = Field(
       default=None, description="""Used to override HTTP request options."""
   )
   ttl: Optional[str] = Field(
@@ -5686,9 +5970,9 @@ class CreateCachedContentConfig(_common.BaseModel):
 
 
 class CreateCachedContentConfigDict(TypedDict, total=False):
-  """Class for configuring optional cached content creation parameters."""
+  """Optional configuration for cached content creation."""
 
-  http_options: Optional[dict[str, Any]]
+  http_options: Optional[HttpOptionsDict]
   """Used to override HTTP request options."""
 
   ttl: Optional[str]
@@ -5814,7 +6098,7 @@ class CachedContent(_common.BaseModel):
       description="""The name of the publisher model to use for cached content.""",
   )
   create_time: Optional[datetime.datetime] = Field(
-      default=None, description="""Creatation time of the cache entry."""
+      default=None, description="""Creation time of the cache entry."""
   )
   update_time: Optional[datetime.datetime] = Field(
       default=None,
@@ -5842,7 +6126,7 @@ class CachedContentDict(TypedDict, total=False):
   """The name of the publisher model to use for cached content."""
 
   create_time: Optional[datetime.datetime]
-  """Creatation time of the cache entry."""
+  """Creation time of the cache entry."""
 
   update_time: Optional[datetime.datetime]
   """When the cache entry was last updated in UTC time."""
@@ -5860,7 +6144,7 @@ CachedContentOrDict = Union[CachedContent, CachedContentDict]
 class GetCachedContentConfig(_common.BaseModel):
   """Optional parameters for caches.get method."""
 
-  http_options: Optional[dict[str, Any]] = Field(
+  http_options: Optional[HttpOptions] = Field(
       default=None, description="""Used to override HTTP request options."""
   )
 
@@ -5868,7 +6152,7 @@ class GetCachedContentConfig(_common.BaseModel):
 class GetCachedContentConfigDict(TypedDict, total=False):
   """Optional parameters for caches.get method."""
 
-  http_options: Optional[dict[str, Any]]
+  http_options: Optional[HttpOptionsDict]
   """Used to override HTTP request options."""
 
 
@@ -5912,7 +6196,7 @@ _GetCachedContentParametersOrDict = Union[
 class DeleteCachedContentConfig(_common.BaseModel):
   """Optional parameters for caches.delete method."""
 
-  http_options: Optional[dict[str, Any]] = Field(
+  http_options: Optional[HttpOptions] = Field(
       default=None, description="""Used to override HTTP request options."""
   )
 
@@ -5920,7 +6204,7 @@ class DeleteCachedContentConfig(_common.BaseModel):
 class DeleteCachedContentConfigDict(TypedDict, total=False):
   """Optional parameters for caches.delete method."""
 
-  http_options: Optional[dict[str, Any]]
+  http_options: Optional[HttpOptionsDict]
   """Used to override HTTP request options."""
 
 
@@ -5981,7 +6265,7 @@ DeleteCachedContentResponseOrDict = Union[
 class UpdateCachedContentConfig(_common.BaseModel):
   """Optional parameters for caches.update method."""
 
-  http_options: Optional[dict[str, Any]] = Field(
+  http_options: Optional[HttpOptions] = Field(
       default=None, description="""Used to override HTTP request options."""
   )
   ttl: Optional[str] = Field(
@@ -5997,7 +6281,7 @@ class UpdateCachedContentConfig(_common.BaseModel):
 class UpdateCachedContentConfigDict(TypedDict, total=False):
   """Optional parameters for caches.update method."""
 
-  http_options: Optional[dict[str, Any]]
+  http_options: Optional[HttpOptionsDict]
   """Used to override HTTP request options."""
 
   ttl: Optional[str]
@@ -6045,6 +6329,9 @@ _UpdateCachedContentParametersOrDict = Union[
 class ListCachedContentsConfig(_common.BaseModel):
   """Config for caches.list method."""
 
+  http_options: Optional[HttpOptions] = Field(
+      default=None, description="""Used to override HTTP request options."""
+  )
   page_size: Optional[int] = Field(default=None, description="""""")
   page_token: Optional[str] = Field(default=None, description="""""")
 
@@ -6052,6 +6339,9 @@ class ListCachedContentsConfig(_common.BaseModel):
 class ListCachedContentsConfigDict(TypedDict, total=False):
   """Config for caches.list method."""
 
+  http_options: Optional[HttpOptionsDict]
+  """Used to override HTTP request options."""
+
   page_size: Optional[int]
   """"""
 
@@ -6115,7 +6405,7 @@ ListCachedContentsResponseOrDict = Union[
 class ListFilesConfig(_common.BaseModel):
   """Used to override the default configuration."""
 
-  http_options: Optional[dict[str, Any]] = Field(
+  http_options: Optional[HttpOptions] = Field(
       default=None, description="""Used to override HTTP request options."""
   )
   page_size: Optional[int] = Field(default=None, description="""""")
@@ -6125,7 +6415,7 @@ class ListFilesConfig(_common.BaseModel):
 class ListFilesConfigDict(TypedDict, total=False):
   """Used to override the default configuration."""
 
-  http_options: Optional[dict[str, Any]]
+  http_options: Optional[HttpOptionsDict]
   """Used to override HTTP request options."""
 
   page_size: Optional[int]
@@ -6159,129 +6449,6 @@ _ListFilesParametersOrDict = Union[
 ]
 
 
-class FileStatus(_common.BaseModel):
-  """Status of a File that uses a common error model."""
-
-  details: Optional[list[dict[str, Any]]] = Field(
-      default=None,
-      description="""A list of messages that carry the error details. There is a common set of message types for APIs to use.""",
-  )
-  message: Optional[str] = Field(
-      default=None,
-      description="""A list of messages that carry the error details. There is a common set of message types for APIs to use.""",
-  )
-  code: Optional[int] = Field(
-      default=None, description="""The status code. 0 for OK, 1 for CANCELLED"""
-  )
-
-
-class FileStatusDict(TypedDict, total=False):
-  """Status of a File that uses a common error model."""
-
-  details: Optional[list[dict[str, Any]]]
-  """A list of messages that carry the error details. There is a common set of message types for APIs to use."""
-
-  message: Optional[str]
-  """A list of messages that carry the error details. There is a common set of message types for APIs to use."""
-
-  code: Optional[int]
-  """The status code. 0 for OK, 1 for CANCELLED"""
-
-
-FileStatusOrDict = Union[FileStatus, FileStatusDict]
-
-
-class File(_common.BaseModel):
-  """A file uploaded to the API."""
-
-  name: Optional[str] = Field(
-      default=None,
-      description="""The `File` resource name. The ID (name excluding the "files/" prefix) can contain up to 40 characters that are lowercase alphanumeric or dashes (-). The ID cannot start or end with a dash. If the name is empty on create, a unique name will be generated. Example: `files/123-456`""",
-  )
-  display_name: Optional[str] = Field(
-      default=None,
-      description="""Optional. The human-readable display name for the `File`. The display name must be no more than 512 characters in length, including spaces. Example: 'Welcome Image'""",
-  )
-  mime_type: Optional[str] = Field(
-      default=None, description="""Output only. MIME type of the file."""
-  )
-  size_bytes: Optional[int] = Field(
-      default=None, description="""Output only. Size of the file in bytes."""
-  )
-  create_time: Optional[datetime.datetime] = Field(
-      default=None,
-      description="""Output only. The timestamp of when the `File` was created.""",
-  )
-  expiration_time: Optional[datetime.datetime] = Field(
-      default=None,
-      description="""Optional. The human-readable display name for the `File`. The display name must be no more than 512 characters in length, including spaces. Example: 'Welcome Image'""",
-  )
-  update_time: Optional[datetime.datetime] = Field(
-      default=None,
-      description="""Output only. The timestamp of when the `File` was last updated.""",
-  )
-  sha256_hash: Optional[bytes] = Field(
-      default=None,
-      description="""Output only. SHA-256 hash of the uploaded bytes.""",
-  )
-  uri: Optional[str] = Field(
-      default=None, description="""Output only. The URI of the `File`."""
-  )
-  state: Optional[FileState] = Field(
-      default=None, description="""Output only. Processing state of the File."""
-  )
-  video_metadata: Optional[dict[str, Any]] = Field(
-      default=None, description="""Output only. Metadata for a video."""
-  )
-  error: Optional[FileStatus] = Field(
-      default=None,
-      description="""Output only. Error status if File processing failed.""",
-  )
-
-
-class FileDict(TypedDict, total=False):
-  """A file uploaded to the API."""
-
-  name: Optional[str]
-  """The `File` resource name. The ID (name excluding the "files/" prefix) can contain up to 40 characters that are lowercase alphanumeric or dashes (-). The ID cannot start or end with a dash. If the name is empty on create, a unique name will be generated. Example: `files/123-456`"""
-
-  display_name: Optional[str]
-  """Optional. The human-readable display name for the `File`. The display name must be no more than 512 characters in length, including spaces. Example: 'Welcome Image'"""
-
-  mime_type: Optional[str]
-  """Output only. MIME type of the file."""
-
-  size_bytes: Optional[int]
-  """Output only. Size of the file in bytes."""
-
-  create_time: Optional[datetime.datetime]
-  """Output only. The timestamp of when the `File` was created."""
-
-  expiration_time: Optional[datetime.datetime]
-  """Optional. The human-readable display name for the `File`. The display name must be no more than 512 characters in length, including spaces. Example: 'Welcome Image'"""
-
-  update_time: Optional[datetime.datetime]
-  """Output only. The timestamp of when the `File` was last updated."""
-
-  sha256_hash: Optional[bytes]
-  """Output only. SHA-256 hash of the uploaded bytes."""
-
-  uri: Optional[str]
-  """Output only. The URI of the `File`."""
-
-  state: Optional[FileState]
-  """Output only. Processing state of the File."""
-
-  video_metadata: Optional[dict[str, Any]]
-  """Output only. Metadata for a video."""
-
-  error: Optional[FileStatusDict]
-  """Output only. Error status if File processing failed."""
-
-
-FileOrDict = Union[File, FileDict]
-
-
 class ListFilesResponse(_common.BaseModel):
   """Response for the list files method."""
 
@@ -6309,7 +6476,7 @@ ListFilesResponseOrDict = Union[ListFilesResponse, ListFilesResponseDict]
 class CreateFileConfig(_common.BaseModel):
   """Used to override the default configuration."""
 
-  http_options: Optional[dict[str, Any]] = Field(
+  http_options: Optional[HttpOptions] = Field(
       default=None, description="""Used to override HTTP request options."""
   )
 
@@ -6317,7 +6484,7 @@ class CreateFileConfig(_common.BaseModel):
 class CreateFileConfigDict(TypedDict, total=False):
   """Used to override the default configuration."""
 
-  http_options: Optional[dict[str, Any]]
+  http_options: Optional[HttpOptionsDict]
   """Used to override HTTP request options."""
 
 
@@ -6380,7 +6547,7 @@ CreateFileResponseOrDict = Union[CreateFileResponse, CreateFileResponseDict]
 class GetFileConfig(_common.BaseModel):
   """Used to override the default configuration."""
 
-  http_options: Optional[dict[str, Any]] = Field(
+  http_options: Optional[HttpOptions] = Field(
       default=None, description="""Used to override HTTP request options."""
   )
 
@@ -6388,7 +6555,7 @@ class GetFileConfig(_common.BaseModel):
 class GetFileConfigDict(TypedDict, total=False):
   """Used to override the default configuration."""
 
-  http_options: Optional[dict[str, Any]]
+  http_options: Optional[HttpOptionsDict]
   """Used to override HTTP request options."""
 
 
@@ -6424,7 +6591,7 @@ _GetFileParametersOrDict = Union[_GetFileParameters, _GetFileParametersDict]
 class DeleteFileConfig(_common.BaseModel):
   """Used to override the default configuration."""
 
-  http_options: Optional[dict[str, Any]] = Field(
+  http_options: Optional[HttpOptions] = Field(
       default=None, description="""Used to override HTTP request options."""
   )
 
@@ -6432,7 +6599,7 @@ class DeleteFileConfig(_common.BaseModel):
 class DeleteFileConfigDict(TypedDict, total=False):
   """Used to override the default configuration."""
 
-  http_options: Optional[dict[str, Any]]
+  http_options: Optional[HttpOptionsDict]
   """Used to override HTTP request options."""
 
 
@@ -6483,7 +6650,7 @@ DeleteFileResponseOrDict = Union[DeleteFileResponse, DeleteFileResponseDict]
 
 
 class BatchJobSource(_common.BaseModel):
-  """Config class for `src` parameter."""
+  """Config for `src` parameter."""
 
   format: Optional[str] = Field(
       default=None,
@@ -6504,7 +6671,7 @@ class BatchJobSource(_common.BaseModel):
 
 
 class BatchJobSourceDict(TypedDict, total=False):
-  """Config class for `src` parameter."""
+  """Config for `src` parameter."""
 
   format: Optional[str]
   """Storage format of the input files. Must be one of:
@@ -6524,7 +6691,7 @@ BatchJobSourceOrDict = Union[BatchJobSource, BatchJobSourceDict]
 
 
 class BatchJobDestination(_common.BaseModel):
-  """Config class for `des` parameter."""
+  """Config for `des` parameter."""
 
   format: Optional[str] = Field(
       default=None,
@@ -6545,7 +6712,7 @@ class BatchJobDestination(_common.BaseModel):
 
 
 class BatchJobDestinationDict(TypedDict, total=False):
-  """Config class for `des` parameter."""
+  """Config for `des` parameter."""
 
   format: Optional[str]
   """Storage format of the output files. Must be one of:
@@ -6565,9 +6732,9 @@ BatchJobDestinationOrDict = Union[BatchJobDestination, BatchJobDestinationDict]
 
 
 class CreateBatchJobConfig(_common.BaseModel):
-  """Config class for optional parameters."""
+  """Config for optional parameters."""
 
-  http_options: Optional[dict[str, Any]] = Field(
+  http_options: Optional[HttpOptions] = Field(
       default=None, description="""Used to override HTTP request options."""
   )
   display_name: Optional[str] = Field(
@@ -6584,9 +6751,9 @@ class CreateBatchJobConfig(_common.BaseModel):
 
 
 class CreateBatchJobConfigDict(TypedDict, total=False):
-  """Config class for optional parameters."""
+  """Config for optional parameters."""
 
-  http_options: Optional[dict[str, Any]]
+  http_options: Optional[HttpOptionsDict]
   """Used to override HTTP request options."""
 
   display_name: Optional[str]
@@ -6605,7 +6772,7 @@ CreateBatchJobConfigOrDict = Union[
 
 
 class _CreateBatchJobParameters(_common.BaseModel):
-  """Config class for batches.create parameters."""
+  """Config for batches.create parameters."""
 
   model: Optional[str] = Field(
       default=None,
@@ -6626,7 +6793,7 @@ class _CreateBatchJobParameters(_common.BaseModel):
 
 
 class _CreateBatchJobParametersDict(TypedDict, total=False):
-  """Config class for batches.create parameters."""
+  """Config for batches.create parameters."""
 
   model: Optional[str]
   """The name of the model to produces the predictions via the BatchJob.
@@ -6648,7 +6815,7 @@ _CreateBatchJobParametersOrDict = Union[
 
 
 class JobError(_common.BaseModel):
-  """Config class for the job error."""
+  """Job error."""
 
   details: Optional[list[str]] = Field(
       default=None,
@@ -6662,7 +6829,7 @@ class JobError(_common.BaseModel):
 
 
 class JobErrorDict(TypedDict, total=False):
-  """Config class for the job error."""
+  """Job error."""
 
   details: Optional[list[str]]
   """A list of messages that carry the error details. There is a common set of message types for APIs to use."""
@@ -6678,7 +6845,7 @@ JobErrorOrDict = Union[JobError, JobErrorDict]
 
 
 class BatchJob(_common.BaseModel):
-  """Config class for batches.create return value."""
+  """Config for batches.create return value."""
 
   name: Optional[str] = Field(
       default=None, description="""Output only. Resource name of the Job."""
@@ -6728,7 +6895,7 @@ class BatchJob(_common.BaseModel):
 
 
 class BatchJobDict(TypedDict, total=False):
-  """Config class for batches.create return value."""
+  """Config for batches.create return value."""
 
   name: Optional[str]
   """Output only. Resource name of the Job."""
@@ -6773,7 +6940,7 @@ BatchJobOrDict = Union[BatchJob, BatchJobDict]
 class GetBatchJobConfig(_common.BaseModel):
   """Optional parameters."""
 
-  http_options: Optional[dict[str, Any]] = Field(
+  http_options: Optional[HttpOptions] = Field(
       default=None, description="""Used to override HTTP request options."""
   )
 
@@ -6781,7 +6948,7 @@ class GetBatchJobConfig(_common.BaseModel):
 class GetBatchJobConfigDict(TypedDict, total=False):
   """Optional parameters."""
 
-  http_options: Optional[dict[str, Any]]
+  http_options: Optional[HttpOptionsDict]
   """Used to override HTTP request options."""
 
 
@@ -6789,7 +6956,7 @@ GetBatchJobConfigOrDict = Union[GetBatchJobConfig, GetBatchJobConfigDict]
 
 
 class _GetBatchJobParameters(_common.BaseModel):
-  """Config class for batches.get parameters."""
+  """Config for batches.get parameters."""
 
   name: Optional[str] = Field(
       default=None,
@@ -6804,7 +6971,7 @@ class _GetBatchJobParameters(_common.BaseModel):
 
 
 class _GetBatchJobParametersDict(TypedDict, total=False):
-  """Config class for batches.get parameters."""
+  """Config for batches.get parameters."""
 
   name: Optional[str]
   """A fully-qualified BatchJob resource name or ID.
@@ -6824,7 +6991,7 @@ _GetBatchJobParametersOrDict = Union[
 class CancelBatchJobConfig(_common.BaseModel):
   """Optional parameters."""
 
-  http_options: Optional[dict[str, Any]] = Field(
+  http_options: Optional[HttpOptions] = Field(
       default=None, description="""Used to override HTTP request options."""
   )
 
@@ -6832,7 +6999,7 @@ class CancelBatchJobConfig(_common.BaseModel):
 class CancelBatchJobConfigDict(TypedDict, total=False):
   """Optional parameters."""
 
-  http_options: Optional[dict[str, Any]]
+  http_options: Optional[HttpOptionsDict]
   """Used to override HTTP request options."""
 
 
@@ -6842,7 +7009,7 @@ CancelBatchJobConfigOrDict = Union[
 
 
 class _CancelBatchJobParameters(_common.BaseModel):
-  """Config class for batches.cancel parameters."""
+  """Config for batches.cancel parameters."""
 
   name: Optional[str] = Field(
       default=None,
@@ -6857,7 +7024,7 @@ class _CancelBatchJobParameters(_common.BaseModel):
 
 
 class _CancelBatchJobParametersDict(TypedDict, total=False):
-  """Config class for batches.cancel parameters."""
+  """Config for batches.cancel parameters."""
 
   name: Optional[str]
   """A fully-qualified BatchJob resource name or ID.
@@ -6874,10 +7041,10 @@ _CancelBatchJobParametersOrDict = Union[
 ]
 
 
-class ListBatchJobConfig(_common.BaseModel):
-  """Config class for optional parameters."""
+class ListBatchJobsConfig(_common.BaseModel):
+  """Config for optional parameters."""
 
-  http_options: Optional[dict[str, Any]] = Field(
+  http_options: Optional[HttpOptions] = Field(
       default=None, description="""Used to override HTTP request options."""
   )
   page_size: Optional[int] = Field(default=None, description="""""")
@@ -6885,10 +7052,10 @@ class ListBatchJobConfig(_common.BaseModel):
   filter: Optional[str] = Field(default=None, description="""""")
 
 
-class ListBatchJobConfigDict(TypedDict, total=False):
-  """Config class for optional parameters."""
+class ListBatchJobsConfigDict(TypedDict, total=False):
+  """Config for optional parameters."""
 
-  http_options: Optional[dict[str, Any]]
+  http_options: Optional[HttpOptionsDict]
   """Used to override HTTP request options."""
 
   page_size: Optional[int]
@@ -6901,36 +7068,38 @@ class ListBatchJobConfigDict(TypedDict, total=False):
   """"""
 
 
-ListBatchJobConfigOrDict = Union[ListBatchJobConfig, ListBatchJobConfigDict]
+ListBatchJobsConfigOrDict = Union[ListBatchJobsConfig, ListBatchJobsConfigDict]
 
 
-class _ListBatchJobParameters(_common.BaseModel):
-  """Config class for batches.list parameters."""
+class _ListBatchJobsParameters(_common.BaseModel):
+  """Config for batches.list parameters."""
 
-  config: Optional[ListBatchJobConfig] = Field(default=None, description="""""")
+  config: Optional[ListBatchJobsConfig] = Field(
+      default=None, description=""""""
+  )
 
 
-class _ListBatchJobParametersDict(TypedDict, total=False):
-  """Config class for batches.list parameters."""
+class _ListBatchJobsParametersDict(TypedDict, total=False):
+  """Config for batches.list parameters."""
 
-  config: Optional[ListBatchJobConfigDict]
+  config: Optional[ListBatchJobsConfigDict]
   """"""
 
 
-_ListBatchJobParametersOrDict = Union[
-    _ListBatchJobParameters, _ListBatchJobParametersDict
+_ListBatchJobsParametersOrDict = Union[
+    _ListBatchJobsParameters, _ListBatchJobsParametersDict
 ]
 
 
-class ListBatchJobResponse(_common.BaseModel):
-  """Config class for batches.list return value."""
+class ListBatchJobsResponse(_common.BaseModel):
+  """Config for batches.list return value."""
 
   next_page_token: Optional[str] = Field(default=None, description="""""")
   batch_jobs: Optional[list[BatchJob]] = Field(default=None, description="""""")
 
 
-class ListBatchJobResponseDict(TypedDict, total=False):
-  """Config class for batches.list return value."""
+class ListBatchJobsResponseDict(TypedDict, total=False):
+  """Config for batches.list return value."""
 
   next_page_token: Optional[str]
   """"""
@@ -6939,13 +7108,33 @@ class ListBatchJobResponseDict(TypedDict, total=False):
   """"""
 
 
-ListBatchJobResponseOrDict = Union[
-    ListBatchJobResponse, ListBatchJobResponseDict
+ListBatchJobsResponseOrDict = Union[
+    ListBatchJobsResponse, ListBatchJobsResponseDict
+]
+
+
+class DeleteBatchJobConfig(_common.BaseModel):
+  """Optional parameters for models.get method."""
+
+  http_options: Optional[HttpOptions] = Field(
+      default=None, description="""Used to override HTTP request options."""
+  )
+
+
+class DeleteBatchJobConfigDict(TypedDict, total=False):
+  """Optional parameters for models.get method."""
+
+  http_options: Optional[HttpOptionsDict]
+  """Used to override HTTP request options."""
+
+
+DeleteBatchJobConfigOrDict = Union[
+    DeleteBatchJobConfig, DeleteBatchJobConfigDict
 ]
 
 
 class _DeleteBatchJobParameters(_common.BaseModel):
-  """Config class for batches.delete parameters."""
+  """Config for batches.delete parameters."""
 
   name: Optional[str] = Field(
       default=None,
@@ -6954,10 +7143,13 @@ class _DeleteBatchJobParameters(_common.BaseModel):
     or "456" when project and location are initialized in the client.
     """,
   )
+  config: Optional[DeleteBatchJobConfig] = Field(
+      default=None, description="""Optional parameters for the request."""
+  )
 
 
 class _DeleteBatchJobParametersDict(TypedDict, total=False):
-  """Config class for batches.delete parameters."""
+  """Config for batches.delete parameters."""
 
   name: Optional[str]
   """A fully-qualified BatchJob resource name or ID.
@@ -6965,6 +7157,9 @@ class _DeleteBatchJobParametersDict(TypedDict, total=False):
     or "456" when project and location are initialized in the client.
     """
 
+  config: Optional[DeleteBatchJobConfigDict]
+  """Optional parameters for the request."""
+
 
 _DeleteBatchJobParametersOrDict = Union[
     _DeleteBatchJobParameters, _DeleteBatchJobParametersDict
@@ -6972,7 +7167,7 @@ _DeleteBatchJobParametersOrDict = Union[
 
 
 class DeleteResourceJob(_common.BaseModel):
-  """Config class for the return value of delete operation."""
+  """The return value of delete operation."""
 
   name: Optional[str] = Field(default=None, description="""""")
   done: Optional[bool] = Field(default=None, description="""""")
@@ -6980,7 +7175,7 @@ class DeleteResourceJob(_common.BaseModel):
 
 
 class DeleteResourceJobDict(TypedDict, total=False):
-  """Config class for the return value of delete operation."""
+  """The return value of delete operation."""
 
   name: Optional[str]
   """"""
@@ -7189,7 +7384,7 @@ ReplayFileOrDict = Union[ReplayFile, ReplayFileDict]
 class UploadFileConfig(_common.BaseModel):
   """Used to override the default configuration."""
 
-  http_options: Optional[dict[str, Any]] = Field(
+  http_options: Optional[HttpOptions] = Field(
       default=None, description="""Used to override HTTP request options."""
   )
   name: Optional[str] = Field(
@@ -7208,7 +7403,7 @@ class UploadFileConfig(_common.BaseModel):
 class UploadFileConfigDict(TypedDict, total=False):
   """Used to override the default configuration."""
 
-  http_options: Optional[dict[str, Any]]
+  http_options: Optional[HttpOptionsDict]
   """Used to override HTTP request options."""
 
   name: Optional[str]
@@ -7224,6 +7419,24 @@ class UploadFileConfigDict(TypedDict, total=False):
 UploadFileConfigOrDict = Union[UploadFileConfig, UploadFileConfigDict]
 
 
+class DownloadFileConfig(_common.BaseModel):
+  """Used to override the default configuration."""
+
+  http_options: Optional[HttpOptions] = Field(
+      default=None, description="""Used to override HTTP request options."""
+  )
+
+
+class DownloadFileConfigDict(TypedDict, total=False):
+  """Used to override the default configuration."""
+
+  http_options: Optional[HttpOptionsDict]
+  """Used to override HTTP request options."""
+
+
+DownloadFileConfigOrDict = Union[DownloadFileConfig, DownloadFileConfigDict]
+
+
 class UpscaleImageConfig(_common.BaseModel):
   """Configuration for upscaling an image.
 
@@ -7232,7 +7445,7 @@ class UpscaleImageConfig(_common.BaseModel):
   <https://cloud.google.com/vertex-ai/generative-ai/docs/model-reference/imagen-api>`_.
   """
 
-  http_options: Optional[dict[str, Any]] = Field(
+  http_options: Optional[HttpOptions] = Field(
       default=None, description="""Used to override HTTP request options."""
   )
   include_rai_reason: Optional[bool] = Field(
@@ -7259,7 +7472,7 @@ class UpscaleImageConfigDict(TypedDict, total=False):
   <https://cloud.google.com/vertex-ai/generative-ai/docs/model-reference/imagen-api>`_.
   """
 
-  http_options: Optional[dict[str, Any]]
+  http_options: Optional[HttpOptionsDict]
   """Used to override HTTP request options."""
 
   include_rai_reason: Optional[bool]
@@ -7317,7 +7530,7 @@ UpscaleImageParametersOrDict = Union[
 
 
 class RawReferenceImage(_common.BaseModel):
-  """Class that represents a Raw reference image.
+  """A raw reference image.
 
   A raw reference image represents the base image to edit, provided by the user.
   It can optionally be provided in addition to a mask reference image or
@@ -7343,12 +7556,12 @@ class RawReferenceImage(_common.BaseModel):
     super().__init__(
         reference_image=reference_image,
         reference_id=reference_id,
-        reference_type="REFERENCE_TYPE_RAW",
+        reference_type='REFERENCE_TYPE_RAW',
     )
 
 
 class RawReferenceImageDict(TypedDict, total=False):
-  """Class that represents a Raw reference image.
+  """A raw reference image.
 
   A raw reference image represents the base image to edit, provided by the user.
   It can optionally be provided in addition to a mask reference image or
@@ -7369,7 +7582,7 @@ RawReferenceImageOrDict = Union[RawReferenceImage, RawReferenceImageDict]
 
 
 class MaskReferenceImage(_common.BaseModel):
-  """Class that represents a Mask reference image.
+  """A mask reference image.
 
   This encapsulates either a mask image provided by the user and configs for
   the user provided mask, or only config parameters for the model to generate
@@ -7395,7 +7608,7 @@ class MaskReferenceImage(_common.BaseModel):
       description="""Configuration for the mask reference image.""",
   )
   """Re-map config to mask_reference_config to send to API."""
-  mask_image_config: Optional["MaskReferenceConfig"] = Field(
+  mask_image_config: Optional['MaskReferenceConfig'] = Field(
       default=None, description=""""""
   )
 
@@ -7403,18 +7616,18 @@ class MaskReferenceImage(_common.BaseModel):
       self,
       reference_image: Optional[Image] = None,
       reference_id: Optional[int] = None,
-      config: Optional["MaskReferenceConfig"] = None,
+      config: Optional['MaskReferenceConfig'] = None,
   ):
     super().__init__(
         reference_image=reference_image,
         reference_id=reference_id,
-        reference_type="REFERENCE_TYPE_MASK",
+        reference_type='REFERENCE_TYPE_MASK',
     )
     self.mask_image_config = config
 
 
 class MaskReferenceImageDict(TypedDict, total=False):
-  """Class that represents a Mask reference image.
+  """A mask reference image.
 
   This encapsulates either a mask image provided by the user and configs for
   the user provided mask, or only config parameters for the model to generate
@@ -7442,7 +7655,7 @@ MaskReferenceImageOrDict = Union[MaskReferenceImage, MaskReferenceImageDict]
 
 
 class ControlReferenceImage(_common.BaseModel):
-  """Class that represents a Control reference image.
+  """A control reference image.
 
   The image of the control reference image is either a control image provided
   by the user, or a regular image which the backend will use to generate a
@@ -7468,7 +7681,7 @@ class ControlReferenceImage(_common.BaseModel):
       description="""Configuration for the control reference image.""",
   )
   """Re-map config to control_reference_config to send to API."""
-  control_image_config: Optional["ControlReferenceConfig"] = Field(
+  control_image_config: Optional['ControlReferenceConfig'] = Field(
       default=None, description=""""""
   )
 
@@ -7476,18 +7689,18 @@ class ControlReferenceImage(_common.BaseModel):
       self,
       reference_image: Optional[Image] = None,
       reference_id: Optional[int] = None,
-      config: Optional["ControlReferenceConfig"] = None,
+      config: Optional['ControlReferenceConfig'] = None,
   ):
     super().__init__(
         reference_image=reference_image,
         reference_id=reference_id,
-        reference_type="REFERENCE_TYPE_CONTROL",
+        reference_type='REFERENCE_TYPE_CONTROL',
     )
     self.control_image_config = config
 
 
 class ControlReferenceImageDict(TypedDict, total=False):
-  """Class that represents a Control reference image.
+  """A control reference image.
 
   The image of the control reference image is either a control image provided
   by the user, or a regular image which the backend will use to generate a
@@ -7517,7 +7730,7 @@ ControlReferenceImageOrDict = Union[
 
 
 class StyleReferenceImage(_common.BaseModel):
-  """Class that represents a Style reference image.
+  """A style reference image.
 
   This encapsulates a style reference image provided by the user, and
   additionally optional config parameters for the style reference image.
@@ -7541,7 +7754,7 @@ class StyleReferenceImage(_common.BaseModel):
       description="""Configuration for the style reference image.""",
   )
   """Re-map config to style_reference_config to send to API."""
-  style_image_config: Optional["StyleReferenceConfig"] = Field(
+  style_image_config: Optional['StyleReferenceConfig'] = Field(
       default=None, description=""""""
   )
 
@@ -7549,18 +7762,18 @@ class StyleReferenceImage(_common.BaseModel):
       self,
       reference_image: Optional[Image] = None,
       reference_id: Optional[int] = None,
-      config: Optional["StyleReferenceConfig"] = None,
+      config: Optional['StyleReferenceConfig'] = None,
   ):
     super().__init__(
         reference_image=reference_image,
         reference_id=reference_id,
-        reference_type="REFERENCE_TYPE_STYLE",
+        reference_type='REFERENCE_TYPE_STYLE',
     )
     self.style_image_config = config
 
 
 class StyleReferenceImageDict(TypedDict, total=False):
-  """Class that represents a Style reference image.
+  """A style reference image.
 
   This encapsulates a style reference image provided by the user, and
   additionally optional config parameters for the style reference image.
@@ -7586,7 +7799,7 @@ StyleReferenceImageOrDict = Union[StyleReferenceImage, StyleReferenceImageDict]
 
 
 class SubjectReferenceImage(_common.BaseModel):
-  """Class that represents a Subject reference image.
+  """A subject reference image.
 
   This encapsulates a subject reference image provided by the user, and
   additionally optional config parameters for the subject reference image.
@@ -7610,7 +7823,7 @@ class SubjectReferenceImage(_common.BaseModel):
       description="""Configuration for the subject reference image.""",
   )
   """Re-map config to subject_reference_config to send to API."""
-  subject_image_config: Optional["SubjectReferenceConfig"] = Field(
+  subject_image_config: Optional['SubjectReferenceConfig'] = Field(
       default=None, description=""""""
   )
 
@@ -7618,18 +7831,18 @@ class SubjectReferenceImage(_common.BaseModel):
       self,
       reference_image: Optional[Image] = None,
       reference_id: Optional[int] = None,
-      config: Optional["SubjectReferenceConfig"] = None,
+      config: Optional['SubjectReferenceConfig'] = None,
   ):
     super().__init__(
         reference_image=reference_image,
         reference_id=reference_id,
-        reference_type="REFERENCE_TYPE_SUBJECT",
+        reference_type='REFERENCE_TYPE_SUBJECT',
     )
     self.subject_image_config = config
 
 
 class SubjectReferenceImageDict(TypedDict, total=False):
-  """Class that represents a Subject reference image.
+  """A subject reference image.
 
   This encapsulates a subject reference image provided by the user, and
   additionally optional config parameters for the subject reference image.
@@ -7690,7 +7903,7 @@ class LiveServerContent(_common.BaseModel):
   )
   interrupted: Optional[bool] = Field(
       default=None,
-      description="""If true, indicates that a client message has interrupted current model generation. If the client is playing out the content in realtime, this is a good signal to stop and empty the current queue. If the client is playing out the content in realtime, this is a good signal to stop and empty the current playback queue.""",
+      description="""If true, indicates that a client message has interrupted current model generation. If the client is playing out the content in realtime, this is a good signal to stop and empty the current queue.""",
   )
 
 
@@ -7708,7 +7921,7 @@ class LiveServerContentDict(TypedDict, total=False):
   """If true, indicates that the model is done generating. Generation will only start in response to additional client messages. Can be set alongside `content`, indicating that the `content` is the last in the turn."""
 
   interrupted: Optional[bool]
-  """If true, indicates that a client message has interrupted current model generation. If the client is playing out the content in realtime, this is a good signal to stop and empty the current queue. If the client is playing out the content in realtime, this is a good signal to stop and empty the current playback queue."""
+  """If true, indicates that a client message has interrupted current model generation. If the client is playing out the content in realtime, this is a good signal to stop and empty the current queue."""
 
 
 LiveServerContentOrDict = Union[LiveServerContent, LiveServerContentDict]
@@ -7792,7 +8005,7 @@ class LiveServerMessage(_common.BaseModel):
         or not self.server_content.model_turn.parts
     ):
       return None
-    text = ""
+    text = ''
     for part in self.server_content.model_turn.parts:
       if isinstance(part.text, str):
         if isinstance(part.thought, bool) and part.thought:
@@ -7810,7 +8023,7 @@ class LiveServerMessage(_common.BaseModel):
         or not self.server_content.model_turn.parts
     ):
       return None
-    concatenated_data = b""
+    concatenated_data = b''
     for part in self.server_content.model_turn.parts:
       if part.inline_data and isinstance(part.inline_data.data, bytes):
         concatenated_data += part.inline_data.data
@@ -7848,7 +8061,17 @@ class LiveClientSetup(_common.BaseModel):
   )
   generation_config: Optional[GenerationConfig] = Field(
       default=None,
-      description="""The generation configuration for the session.""",
+      description="""The generation configuration for the session.
+
+The following fields are supported:
+- `response_logprobs`
+- `response_mime_type`
+- `logprobs`
+- `response_schema`
+- `stop_sequence`
+- `routing_config`
+- `audio_timestamp`
+      """,
   )
   system_instruction: Optional[Content] = Field(
       default=None,
@@ -7876,7 +8099,17 @@ class LiveClientSetupDict(TypedDict, total=False):
       """
 
   generation_config: Optional[GenerationConfigDict]
-  """The generation configuration for the session."""
+  """The generation configuration for the session.
+
+The following fields are supported:
+- `response_logprobs`
+- `response_mime_type`
+- `logprobs`
+- `response_schema`
+- `stop_sequence`
+- `routing_config`
+- `audio_timestamp`
+      """
 
   system_instruction: Optional[ContentDict]
   """The user provided system instructions for the model.
@@ -7908,7 +8141,7 @@ class LiveClientContent(_common.BaseModel):
       description="""The content appended to the current conversation with the model.
 
       For single-turn queries, this is a single instance. For multi-turn
-      queries, this is a repeated field that contains conversation history +
+      queries, this is a repeated field that contains conversation history and
       latest request.
       """,
   )
@@ -7933,7 +8166,7 @@ class LiveClientContentDict(TypedDict, total=False):
   """The content appended to the current conversation with the model.
 
       For single-turn queries, this is a single instance. For multi-turn
-      queries, this is a repeated field that contains conversation history +
+      queries, this is a repeated field that contains conversation history and
       latest request.
       """
 
@@ -7950,16 +8183,17 @@ class LiveClientRealtimeInput(_common.BaseModel):
   """User input that is sent in real time.
 
   This is different from `ClientContentUpdate` in a few ways:
-  - Can be sent continuously without interruption the model generation.
-  - If there is a need to mix data interleaved across the
-    `ClientContentUpdate` and the `RealtimeUpdate`, server will attempt to
-    optimize for best response, but there are no guarantees.
-  - End of turn is not explicitly specified, but is rather derived from user
-    activity, e.g. end of speech.
-  - Even before the end of turn, the data will be processed incrementally
-    to optimize for a fast start of the response from the model.
-  - Is always assumed to be the user's input (cannot be used to populate
-    conversation history).
+
+    - Can be sent continuously without interruption to model generation.
+    - If there is a need to mix data interleaved across the
+      `ClientContentUpdate` and the `RealtimeUpdate`, server attempts to
+      optimize for best response, but there are no guarantees.
+    - End of turn is not explicitly specified, but is rather derived from user
+      activity (for example, end of speech).
+    - Even before the end of turn, the data is processed incrementally
+      to optimize for a fast start of the response from the model.
+    - Is always assumed to be the user's input (cannot be used to populate
+      conversation history).
   """
 
   media_chunks: Optional[list[Blob]] = Field(
@@ -7971,16 +8205,17 @@ class LiveClientRealtimeInputDict(TypedDict, total=False):
   """User input that is sent in real time.
 
   This is different from `ClientContentUpdate` in a few ways:
-  - Can be sent continuously without interruption the model generation.
-  - If there is a need to mix data interleaved across the
-    `ClientContentUpdate` and the `RealtimeUpdate`, server will attempt to
-    optimize for best response, but there are no guarantees.
-  - End of turn is not explicitly specified, but is rather derived from user
-    activity, e.g. end of speech.
-  - Even before the end of turn, the data will be processed incrementally
-    to optimize for a fast start of the response from the model.
-  - Is always assumed to be the user's input (cannot be used to populate
-    conversation history).
+
+    - Can be sent continuously without interruption to model generation.
+    - If there is a need to mix data interleaved across the
+      `ClientContentUpdate` and the `RealtimeUpdate`, server attempts to
+      optimize for best response, but there are no guarantees.
+    - End of turn is not explicitly specified, but is rather derived from user
+      activity (for example, end of speech).
+    - Even before the end of turn, the data is processed incrementally
+      to optimize for a fast start of the response from the model.
+    - Is always assumed to be the user's input (cannot be used to populate
+      conversation history).
   """
 
   media_chunks: Optional[list[BlobDict]]
@@ -8070,7 +8305,7 @@ LiveClientMessageOrDict = Union[LiveClientMessage, LiveClientMessageDict]
 
 
 class LiveConnectConfig(_common.BaseModel):
-  """Config class for the session."""
+  """Session config for the API connection."""
 
   generation_config: Optional[GenerationConfig] = Field(
       default=None,
@@ -8104,7 +8339,7 @@ class LiveConnectConfig(_common.BaseModel):
 
 
 class LiveConnectConfigDict(TypedDict, total=False):
-  """Config class for the session."""
+  """Session config for the API connection."""
 
   generation_config: Optional[GenerationConfigDict]
   """The generation configuration for the session."""