google-genai 0.6.0__py3-none-any.whl → 0.8.0__py3-none-any.whl

google/genai/types.py

@@ -16,15 +16,32 @@
 # 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, GenericAlias, 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
+
+  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
+
 
 class Outcome(_common.CaseInSensitiveEnum):
   """Required. Outcome of the code execution."""
@@ -164,7 +181,7 @@ class DeploymentResourcesType(_common.CaseInSensitiveEnum):
 
 
 class JobState(_common.CaseInSensitiveEnum):
-  """Config class for the job state."""
+  """Job state."""
 
   JOB_STATE_UNSPECIFIED = 'JOB_STATE_UNSPECIFIED'
   JOB_STATE_QUEUED = 'JOB_STATE_QUEUED'
@@ -192,14 +209,14 @@ class AdapterSize(_common.CaseInSensitiveEnum):
 
 
 class DynamicRetrievalConfigMode(_common.CaseInSensitiveEnum):
-  """Config class for the dynamic retrieval config mode."""
+  """Config for the dynamic retrieval config mode."""
 
   MODE_UNSPECIFIED = 'MODE_UNSPECIFIED'
   MODE_DYNAMIC = 'MODE_DYNAMIC'
 
 
 class FunctionCallingConfigMode(_common.CaseInSensitiveEnum):
-  """Config class for the function calling config mode."""
+  """Config for the function calling config mode."""
 
   MODE_UNSPECIFIED = 'MODE_UNSPECIFIED'
   AUTO = 'AUTO'
@@ -302,7 +319,7 @@ class FileSource(_common.CaseInSensitiveEnum):
 
 
 class Modality(_common.CaseInSensitiveEnum):
-  """Config class for the server content modalities."""
+  """Server content modalities."""
 
   MODALITY_UNSPECIFIED = 'MODALITY_UNSPECIFIED'
   TEXT = 'TEXT'
@@ -494,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."""
@@ -509,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."""
@@ -566,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,
@@ -583,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]
+      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)
 
@@ -683,6 +696,51 @@ class ContentDict(TypedDict, total=False):
 ContentOrDict = Union[Content, ContentDict]
 
 
+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):
   """Schema that defines the format of input and output data.
 
@@ -910,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',
+      *,
+      client,
+      callable: Callable,
   ) -> 'FunctionDeclaration':
-    """Converts a function to a FunctionDeclaration based on an API endpoint.
-
-    Supported endpoints are: 'GOOGLE_AI', 'VERTEX_AI', or 'DEFAULT'.
-    """
+    """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',
           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',
                 inspect.Parameter.POSITIONAL_OR_KEYWORD,
                 annotation=return_annotation,
             ),
-            func.__name__,
+            callable.__name__,
         )
     )
     return declaration
 
-  @classmethod
-  def from_callable(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.
@@ -1610,8 +1641,10 @@ class FileDict(TypedDict, total=False):
 
 FileOrDict = Union[File, FileDict]
 
-
-PartUnion = Union[File, Part, PIL.Image.Image, str]
+if _is_pillow_image_imported:
+  PartUnion = Union[File, Part, PIL_Image, str]
+else:
+  PartUnion = Union[File, Part, str]
 
 
 PartUnionDict = Union[PartUnion, PartDict]
@@ -1709,12 +1742,15 @@ SpeechConfigUnionDict = Union[SpeechConfigUnion, SpeechConfigDict]
 
 
 class GenerateContentConfig(_common.BaseModel):
-  """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[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.
@@ -1868,12 +1904,15 @@ class GenerateContentConfig(_common.BaseModel):
 
 
 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
@@ -2012,7 +2051,7 @@ 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,
@@ -2032,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
@@ -2154,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,
@@ -2166,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
@@ -2559,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,
@@ -2607,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.
@@ -2743,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.""",
   )
@@ -2806,7 +2845,7 @@ 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)
 
@@ -2814,8 +2853,12 @@ class GenerateContentResponse(_common.BaseModel):
     response_schema = _common.get_value_by_path(
         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:
@@ -2823,27 +2866,36 @@ class GenerateContentResponse(_common.BaseModel):
       # may not be a valid json per stream response
       except pydantic.ValidationError:
         pass
-
-    elif isinstance(response_schema, GenericAlias) and issubclass(
-        response_schema.__args__[0], pydantic.BaseModel
+      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
     ):
-      # Handle cases where `list[pydantic.BaseModel]` was provided.
-      result.parsed = []
-      pydantic_model_class = response_schema.__args__[0]
-      response_list_json = json.loads(result.text)
-      for json_instance in response_list_json:
-        try:
-          pydantic_model_instance = pydantic_model_class.model_validate_json(
-              json.dumps(json_instance)
-          )
-          result.parsed.append(pydantic_model_instance)
-        # may not be a valid json per stream response
-        except pydantic.ValidationError:
-          pass
+
+      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, pydantic.BaseModel
+        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)
@@ -2877,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(
@@ -2916,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]
@@ -3117,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(
@@ -3190,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]
@@ -3260,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,
@@ -3281,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,
@@ -3327,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:
@@ -3359,11 +3425,11 @@ class Image(_common.BaseModel):
       location = urllib.parse.urlunparse(parsed_url)
 
     if parsed_url.scheme == 'gs':
-      return Image(gcs_uri=location)
+      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):
@@ -3376,11 +3442,7 @@ 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
@@ -3395,7 +3457,7 @@ class Image(_common.BaseModel):
       if not PIL_Image:
         raise RuntimeError(
             'The PIL module is not available. Please install the Pillow'
-            ' package.'
+            ' package. `pip install pillow`'
         )
       self._loaded_image = PIL_Image.open(io.BytesIO(self.image_bytes))
     return self._loaded_image
@@ -3438,7 +3500,7 @@ 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
@@ -3450,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,
@@ -3468,10 +3533,16 @@ class GeneratedImage(_common.BaseModel):
       response.
       """,
   )
+  enhanced_prompt: Optional[str] = Field(
+      default=None,
+      description="""The rewritten prompt used for the image generation if the prompt
+      enhancer is enabled.
+      """,
+  )
 
 
 class GeneratedImageDict(TypedDict, total=False):
-  """Class that represents an output image."""
+  """An output image."""
 
   image: Optional[ImageDict]
   """The output image data.
@@ -3482,12 +3553,17 @@ class GeneratedImageDict(TypedDict, total=False):
       response.
       """
 
+  enhanced_prompt: Optional[str]
+  """The rewritten prompt used for the image generation if the prompt
+      enhancer is enabled.
+      """
+
 
 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,
@@ -3496,16 +3572,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
 ]
 
 
@@ -3690,7 +3766,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(
@@ -3767,7 +3843,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]
@@ -3894,7 +3970,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(
@@ -3922,7 +3998,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]
@@ -4005,9 +4081,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):
@@ -4015,6 +4112,9 @@ class _GetModelParametersDict(TypedDict, total=False):
   model: Optional[str]
   """"""
 
+  config: Optional[GetModelConfigDict]
+  """Optional parameters for the request."""
+
 
 _GetModelParametersOrDict = Union[_GetModelParameters, _GetModelParametersDict]
 
@@ -4166,7 +4266,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="""""")
@@ -4180,7 +4280,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]
@@ -4235,12 +4335,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]
   """"""
 
@@ -4271,9 +4377,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):
@@ -4281,6 +4406,9 @@ class _DeleteModelParametersDict(TypedDict, total=False):
   model: Optional[str]
   """"""
 
+  config: Optional[DeleteModelConfigDict]
+  """Optional parameters for the request."""
+
 
 _DeleteModelParametersOrDict = Union[
     _DeleteModelParameters, _DeleteModelParametersDict
@@ -4412,7 +4540,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(
@@ -4437,7 +4565,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]
@@ -4521,7 +4649,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."""
   )
 
@@ -4529,7 +4657,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."""
 
 
@@ -4629,7 +4757,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."""
   )
 
@@ -4637,7 +4765,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."""
 
 
@@ -5256,6 +5384,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."""
 
@@ -5351,41 +5514,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."""
 
@@ -5427,7 +5555,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."""
@@ -5440,16 +5568,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,
@@ -5459,6 +5583,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.""",
@@ -5506,7 +5634,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."""
@@ -5517,14 +5645,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."""
@@ -5532,6 +5657,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."""
 
@@ -5542,6 +5670,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="""""")
@@ -5550,6 +5681,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]
   """"""
 
@@ -5635,7 +5769,7 @@ TuningExampleOrDict = Union[TuningExample, TuningExampleDict]
 
 
 class TuningDataset(_common.BaseModel):
-  """Supervised fune-tuning training dataset."""
+  """Supervised fine-tuning training dataset."""
 
   gcs_uri: Optional[str] = Field(
       default=None,
@@ -5648,7 +5782,7 @@ class TuningDataset(_common.BaseModel):
 
 
 class TuningDatasetDict(TypedDict, total=False):
-  """Supervised fune-tuning training dataset."""
+  """Supervised fine-tuning training dataset."""
 
   gcs_uri: Optional[str]
   """GCS URI of the file containing training dataset in JSONL format."""
@@ -5682,7 +5816,7 @@ TuningValidationDatasetOrDict = Union[
 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[TuningValidationDataset] = Field(
@@ -5720,7 +5854,7 @@ class CreateTuningJobConfig(_common.BaseModel):
 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[TuningValidationDatasetDict]
@@ -5805,152 +5939,10 @@ TuningJobOrOperationOrDict = Union[
 ]
 
 
-class DistillationDataset(_common.BaseModel):
-  """Training dataset."""
-
-  gcs_uri: Optional[str] = Field(
-      default=None,
-      description="""GCS URI of the file containing training dataset in JSONL format.""",
-  )
-
-
-class DistillationDatasetDict(TypedDict, total=False):
-  """Training dataset."""
-
-  gcs_uri: Optional[str]
-  """GCS URI of the file containing training dataset in JSONL format."""
-
-
-DistillationDatasetOrDict = Union[DistillationDataset, DistillationDatasetDict]
-
-
-class DistillationValidationDataset(_common.BaseModel):
-  """Validation dataset."""
-
-  gcs_uri: Optional[str] = Field(
-      default=None,
-      description="""GCS URI of the file containing validation dataset in JSONL format.""",
-  )
-
-
-class DistillationValidationDatasetDict(TypedDict, total=False):
-  """Validation dataset."""
-
-  gcs_uri: Optional[str]
-  """GCS URI of the file containing validation dataset in JSONL format."""
-
-
-DistillationValidationDatasetOrDict = Union[
-    DistillationValidationDataset, DistillationValidationDatasetDict
-]
-
-
-class CreateDistillationJobConfig(_common.BaseModel):
-  """Distillation job creation request - optional fields."""
-
-  http_options: Optional[dict[str, Any]] = Field(
-      default=None, description="""Used to override HTTP request options."""
-  )
-  validation_dataset: Optional[DistillationValidationDataset] = Field(
-      default=None,
-      description="""Cloud Storage path to file containing validation 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.""",
-  )
-  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."""
-  )
-  pipeline_root_directory: Optional[str] = Field(
-      default=None,
-      description="""The resource name of the PipelineJob associated with the TuningJob. Format:`projects/{project}/locations/{location}/pipelineJobs/{pipeline_job}`.""",
-  )
-
-
-class CreateDistillationJobConfigDict(TypedDict, total=False):
-  """Distillation job creation request - optional fields."""
-
-  http_options: Optional[dict[str, Any]]
-  """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."""
-
-  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."""
-
-  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."""
-
-  pipeline_root_directory: Optional[str]
-  """The resource name of the PipelineJob associated with the TuningJob. Format:`projects/{project}/locations/{location}/pipelineJobs/{pipeline_job}`."""
-
-
-CreateDistillationJobConfigOrDict = Union[
-    CreateDistillationJobConfig, CreateDistillationJobConfigDict
-]
-
-
-class _CreateDistillationJobParameters(_common.BaseModel):
-  """Distillation 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(
-      default=None,
-      description="""The teacher model that is being distilled from, e.g. ``gemini-1.0-pro-002``.""",
-  )
-  training_dataset: Optional[DistillationDataset] = 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."""
-  )
-
-
-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``."""
-
-  teacher_model: Optional[str]
-  """The teacher model that is being distilled from, e.g. ``gemini-1.0-pro-002``."""
-
-  training_dataset: Optional[DistillationDatasetDict]
-  """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."""
-
-
-_CreateDistillationJobParametersOrDict = Union[
-    _CreateDistillationJobParameters, _CreateDistillationJobParametersDict
-]
-
-
 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(
@@ -5989,9 +5981,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]
@@ -6117,7 +6109,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,
@@ -6145,7 +6137,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."""
@@ -6163,7 +6155,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."""
   )
 
@@ -6171,7 +6163,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."""
 
 
@@ -6215,7 +6207,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."""
   )
 
@@ -6223,7 +6215,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."""
 
 
@@ -6284,7 +6276,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(
@@ -6300,7 +6292,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]
@@ -6348,6 +6340,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="""""")
 
@@ -6355,6 +6350,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]
   """"""
 
@@ -6418,7 +6416,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="""""")
@@ -6428,7 +6426,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]
@@ -6489,7 +6487,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."""
   )
 
@@ -6497,7 +6495,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."""
 
 
@@ -6560,7 +6558,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."""
   )
 
@@ -6568,7 +6566,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."""
 
 
@@ -6604,7 +6602,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."""
   )
 
@@ -6612,7 +6610,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."""
 
 
@@ -6663,7 +6661,7 @@ DeleteFileResponseOrDict = Union[DeleteFileResponse, DeleteFileResponseDict]
 
 
 class BatchJobSource(_common.BaseModel):
-  """Config class for `src` parameter."""
+  """Config for `src` parameter."""
 
   format: Optional[str] = Field(
       default=None,
@@ -6684,7 +6682,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:
@@ -6704,7 +6702,7 @@ BatchJobSourceOrDict = Union[BatchJobSource, BatchJobSourceDict]
 
 
 class BatchJobDestination(_common.BaseModel):
-  """Config class for `des` parameter."""
+  """Config for `des` parameter."""
 
   format: Optional[str] = Field(
       default=None,
@@ -6725,7 +6723,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:
@@ -6745,9 +6743,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(
@@ -6764,9 +6762,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]
@@ -6785,7 +6783,7 @@ CreateBatchJobConfigOrDict = Union[
 
 
 class _CreateBatchJobParameters(_common.BaseModel):
-  """Config class for batches.create parameters."""
+  """Config for batches.create parameters."""
 
   model: Optional[str] = Field(
       default=None,
@@ -6806,7 +6804,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.
@@ -6828,7 +6826,7 @@ _CreateBatchJobParametersOrDict = Union[
 
 
 class JobError(_common.BaseModel):
-  """Config class for the job error."""
+  """Job error."""
 
   details: Optional[list[str]] = Field(
       default=None,
@@ -6842,7 +6840,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."""
@@ -6858,7 +6856,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."""
@@ -6908,7 +6906,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."""
@@ -6953,7 +6951,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."""
   )
 
@@ -6961,7 +6959,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."""
 
 
@@ -6969,7 +6967,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,
@@ -6984,7 +6982,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.
@@ -7004,7 +7002,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."""
   )
 
@@ -7012,7 +7010,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."""
 
 
@@ -7022,7 +7020,7 @@ CancelBatchJobConfigOrDict = Union[
 
 
 class _CancelBatchJobParameters(_common.BaseModel):
-  """Config class for batches.cancel parameters."""
+  """Config for batches.cancel parameters."""
 
   name: Optional[str] = Field(
       default=None,
@@ -7037,7 +7035,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.
@@ -7054,10 +7052,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="""""")
@@ -7065,10 +7063,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]
@@ -7081,36 +7079,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]
   """"""
@@ -7119,13 +7119,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,
@@ -7134,10 +7154,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.
@@ -7145,6 +7168,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
@@ -7152,7 +7178,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="""""")
@@ -7160,7 +7186,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]
   """"""
@@ -7175,6 +7201,147 @@ class DeleteResourceJobDict(TypedDict, total=False):
 DeleteResourceJobOrDict = Union[DeleteResourceJob, DeleteResourceJobDict]
 
 
+class GetOperationConfig(_common.BaseModel):
+
+  http_options: Optional[HttpOptions] = Field(
+      default=None, description="""Used to override HTTP request options."""
+  )
+
+
+class GetOperationConfigDict(TypedDict, total=False):
+
+  http_options: Optional[HttpOptionsDict]
+  """Used to override HTTP request options."""
+
+
+GetOperationConfigOrDict = Union[GetOperationConfig, GetOperationConfigDict]
+
+
+class _GetOperationParameters(_common.BaseModel):
+  """Parameters for the GET method."""
+
+  operation_name: Optional[str] = Field(
+      default=None,
+      description="""The server-assigned name for the operation.""",
+  )
+  config: Optional[GetOperationConfig] = Field(
+      default=None,
+      description="""Used to override the default configuration.""",
+  )
+
+
+class _GetOperationParametersDict(TypedDict, total=False):
+  """Parameters for the GET method."""
+
+  operation_name: Optional[str]
+  """The server-assigned name for the operation."""
+
+  config: Optional[GetOperationConfigDict]
+  """Used to override the default configuration."""
+
+
+_GetOperationParametersOrDict = Union[
+    _GetOperationParameters, _GetOperationParametersDict
+]
+
+
+class Operation(_common.BaseModel):
+  """A long-running operation."""
+
+  name: Optional[str] = Field(
+      default=None,
+      description="""The server-assigned name, which is only unique within the same service that originally returns it. If you use the default HTTP mapping, the `name` should be a resource name ending with `operations/{unique_id}`.""",
+  )
+  metadata: Optional[dict[str, Any]] = Field(
+      default=None,
+      description="""Service-specific metadata associated with the operation. It typically contains progress information and common metadata such as create time. Some services might not provide such metadata.  Any method that returns a long-running operation should document the metadata type, if any.""",
+  )
+  done: Optional[bool] = Field(
+      default=None,
+      description="""If the value is `false`, it means the operation is still in progress. If `true`, the operation is completed, and either `error` or `response` is available.""",
+  )
+  error: Optional[dict[str, Any]] = Field(
+      default=None,
+      description="""The error result of the operation in case of failure or cancellation.""",
+  )
+  response: Optional[dict[str, Any]] = Field(
+      default=None,
+      description="""The normal response of the operation in case of success.""",
+  )
+
+
+class OperationDict(TypedDict, total=False):
+  """A long-running operation."""
+
+  name: Optional[str]
+  """The server-assigned name, which is only unique within the same service that originally returns it. If you use the default HTTP mapping, the `name` should be a resource name ending with `operations/{unique_id}`."""
+
+  metadata: Optional[dict[str, Any]]
+  """Service-specific metadata associated with the operation. It typically contains progress information and common metadata such as create time. Some services might not provide such metadata.  Any method that returns a long-running operation should document the metadata type, if any."""
+
+  done: Optional[bool]
+  """If the value is `false`, it means the operation is still in progress. If `true`, the operation is completed, and either `error` or `response` is available."""
+
+  error: Optional[dict[str, Any]]
+  """The error result of the operation in case of failure or cancellation."""
+
+  response: Optional[dict[str, Any]]
+  """The normal response of the operation in case of success."""
+
+
+OperationOrDict = Union[Operation, OperationDict]
+
+
+class FetchPredictOperationConfig(_common.BaseModel):
+
+  http_options: Optional[HttpOptions] = Field(
+      default=None, description="""Used to override HTTP request options."""
+  )
+
+
+class FetchPredictOperationConfigDict(TypedDict, total=False):
+
+  http_options: Optional[HttpOptionsDict]
+  """Used to override HTTP request options."""
+
+
+FetchPredictOperationConfigOrDict = Union[
+    FetchPredictOperationConfig, FetchPredictOperationConfigDict
+]
+
+
+class _FetchPredictOperationParameters(_common.BaseModel):
+  """Parameters for the fetchPredictOperation method."""
+
+  operation_name: Optional[str] = Field(
+      default=None,
+      description="""The server-assigned name for the operation.""",
+  )
+  resource_name: Optional[str] = Field(default=None, description="""""")
+  config: Optional[FetchPredictOperationConfig] = Field(
+      default=None,
+      description="""Used to override the default configuration.""",
+  )
+
+
+class _FetchPredictOperationParametersDict(TypedDict, total=False):
+  """Parameters for the fetchPredictOperation method."""
+
+  operation_name: Optional[str]
+  """The server-assigned name for the operation."""
+
+  resource_name: Optional[str]
+  """"""
+
+  config: Optional[FetchPredictOperationConfigDict]
+  """Used to override the default configuration."""
+
+
+_FetchPredictOperationParametersOrDict = Union[
+    _FetchPredictOperationParameters, _FetchPredictOperationParametersDict
+]
+
+
 class TestTableItem(_common.BaseModel):
 
   name: Optional[str] = Field(
@@ -7369,7 +7536,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(
@@ -7388,7 +7555,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]
@@ -7407,7 +7574,7 @@ UploadFileConfigOrDict = Union[UploadFileConfig, UploadFileConfigDict]
 class DownloadFileConfig(_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."""
   )
 
@@ -7415,7 +7582,7 @@ class DownloadFileConfig(_common.BaseModel):
 class DownloadFileConfigDict(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."""
 
 
@@ -7430,7 +7597,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(
@@ -7457,7 +7624,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]
@@ -7515,7 +7682,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
@@ -7546,7 +7713,7 @@ class RawReferenceImage(_common.BaseModel):
 
 
 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
@@ -7567,7 +7734,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
@@ -7612,7 +7779,7 @@ class MaskReferenceImage(_common.BaseModel):
 
 
 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
@@ -7640,7 +7807,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
@@ -7685,7 +7852,7 @@ class ControlReferenceImage(_common.BaseModel):
 
 
 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
@@ -7715,7 +7882,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.
@@ -7758,7 +7925,7 @@ class StyleReferenceImage(_common.BaseModel):
 
 
 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.
@@ -7784,7 +7951,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.
@@ -7827,7 +7994,7 @@ class SubjectReferenceImage(_common.BaseModel):
 
 
 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.
@@ -7888,7 +8055,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.""",
   )
 
 
@@ -7906,7 +8073,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]
@@ -8046,7 +8213,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,
@@ -8074,7 +8251,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.
@@ -8106,7 +8293,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.
       """,
   )
@@ -8131,7 +8318,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.
       """
 
@@ -8148,16 +8335,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(
@@ -8169,16 +8357,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]]
@@ -8268,7 +8457,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,
@@ -8302,7 +8491,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."""