google-genai 0.0.1__py3-none-any.whl → 0.2.0__py3-none-any.whl

google/genai/types.py

@@ -154,6 +154,14 @@ DynamicRetrievalConfigMode = Literal["MODE_UNSPECIFIED", "MODE_DYNAMIC"]
 FunctionCallingConfigMode = Literal["MODE_UNSPECIFIED", "AUTO", "ANY", "NONE"]
 
 
+MediaResolution = Literal[
+    "MEDIA_RESOLUTION_UNSPECIFIED",
+    "MEDIA_RESOLUTION_LOW",
+    "MEDIA_RESOLUTION_MEDIUM",
+    "MEDIA_RESOLUTION_HIGH",
+]
+
+
 SafetyFilterLevel = Literal[
     "BLOCK_LOW_AND_ABOVE",
     "BLOCK_MEDIUM_AND_ABOVE",
@@ -322,8 +330,13 @@ FileDataOrDict = Union[FileData, FileDataDict]
 
 
 class FunctionCall(_common.BaseModel):
-  """A predicted [FunctionCall] returned from the model that contains a string representing the [FunctionDeclaration.name] and a structured JSON object containing the parameters and their values."""
+  """A function call."""
 
+  id: Optional[str] = Field(
+      default=None,
+      description="""The unique id of the function call. If populated, the client to execute the
+   `function_call` and return the response with the matching `id`.""",
+  )
   args: Optional[dict[str, Any]] = Field(
       default=None,
       description="""Optional. Required. The function parameters and values in JSON object format. See [FunctionDeclaration.parameters] for parameter details.""",
@@ -335,7 +348,11 @@ class FunctionCall(_common.BaseModel):
 
 
 class FunctionCallDict(TypedDict, total=False):
-  """A predicted [FunctionCall] returned from the model that contains a string representing the [FunctionDeclaration.name] and a structured JSON object containing the parameters and their values."""
+  """A function call."""
+
+  id: Optional[str]
+  """The unique id of the function call. If populated, the client to execute the
+   `function_call` and return the response with the matching `id`."""
 
   args: Optional[dict[str, Any]]
   """Optional. Required. The function parameters and values in JSON object format. See [FunctionDeclaration.parameters] for parameter details."""
@@ -348,12 +365,13 @@ FunctionCallOrDict = Union[FunctionCall, FunctionCallDict]
 
 
 class FunctionResponse(_common.BaseModel):
-  """The result output from a [FunctionCall] that contains a string representing the [FunctionDeclaration.name] and a structured JSON object containing any output from the function is used as context to the model.
-
-  This should contain the result of a [FunctionCall] made based on model
-  prediction.
-  """
+  """A function response."""
 
+  id: Optional[str] = Field(
+      default=None,
+      description="""The id of the function call this response is for. Populated by the client
+   to match the corresponding function call `id`.""",
+  )
   name: Optional[str] = Field(
       default=None,
       description="""Required. The name of the function to call. Matches [FunctionDeclaration.name] and [FunctionCall.name].""",
@@ -365,11 +383,11 @@ class FunctionResponse(_common.BaseModel):
 
 
 class FunctionResponseDict(TypedDict, total=False):
-  """The result output from a [FunctionCall] that contains a string representing the [FunctionDeclaration.name] and a structured JSON object containing any output from the function is used as context to the model.
+  """A function response."""
 
-  This should contain the result of a [FunctionCall] made based on model
-  prediction.
-  """
+  id: Optional[str]
+  """The id of the function call this response is for. Populated by the client
+   to match the corresponding function call `id`."""
 
   name: Optional[str]
   """Required. The name of the function to call. Matches [FunctionDeclaration.name] and [FunctionCall.name]."""
@@ -795,6 +813,57 @@ class FunctionDeclaration(_common.BaseModel):
       description="""Optional. Describes the parameters to this function in JSON Schema Object format. Reflects the Open API 3.03 Parameter Object. string Key: the name of the parameter. Parameter names are case sensitive. Schema Value: the Schema defining the type used for the parameter. For function with no parameters, this can be left unset. Parameter names must start with a letter or an underscore and must only contain chars a-z, A-Z, 0-9, or underscores with a maximum length of 64. Example with 1 required and 1 optional parameter: type: OBJECT properties: param1: type: STRING param2: type: INTEGER required: - param1""",
   )
 
+  @staticmethod
+  def from_function(client, func: Callable) -> "FunctionDeclaration":
+    """Converts a function to a FunctionDeclaration."""
+    from . import _automatic_function_calling_util
+
+    parameters_properties = {}
+    for name, param in inspect.signature(func).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(
+            client, param, func.__name__
+        )
+        parameters_properties[name] = schema
+    declaration = FunctionDeclaration(
+        name=func.__name__,
+        description=func.__doc__,
+    )
+    if parameters_properties:
+      declaration.parameters = Schema(
+          type="OBJECT",
+          properties=parameters_properties,
+      )
+      if client.vertexai:
+        declaration.parameters.required = (
+            _automatic_function_calling_util._get_required_fields(
+                declaration.parameters
+            )
+        )
+    if not client.vertexai:
+      return declaration
+
+    return_annotation = inspect.signature(func).return_annotation
+    if return_annotation is inspect._empty:
+      return declaration
+
+    declaration.response = (
+        _automatic_function_calling_util._parse_schema_from_parameter(
+            client,
+            inspect.Parameter(
+                "return_value",
+                inspect.Parameter.POSITIONAL_OR_KEYWORD,
+                annotation=return_annotation,
+            ),
+            func.__name__,
+        )
+    )
+    return declaration
+
 
 class FunctionDeclarationDict(TypedDict, total=False):
   """Defines a function that the model can generate JSON inputs for.
@@ -1218,6 +1287,15 @@ class AutomaticFunctionCallingConfig(_common.BaseModel):
       If not set, SDK will set maximum number of remote calls to 10.
       """,
   )
+  ignore_call_history: Optional[bool] = Field(
+      default=None,
+      description="""If automatic function calling is enabled,
+      whether to ignore call history to the response.
+      If not set, SDK will set ignore_call_history to false,
+      and will append the call history to
+      GenerateContentResponse.automatic_function_calling_history.
+      """,
+  )
 
 
 class AutomaticFunctionCallingConfigDict(TypedDict, total=False):
@@ -1236,6 +1314,14 @@ class AutomaticFunctionCallingConfigDict(TypedDict, total=False):
       If not set, SDK will set maximum number of remote calls to 10.
       """
 
+  ignore_call_history: Optional[bool]
+  """If automatic function calling is enabled,
+      whether to ignore call history to the response.
+      If not set, SDK will set ignore_call_history to false,
+      and will append the call history to
+      GenerateContentResponse.automatic_function_calling_history.
+      """
+
 
 AutomaticFunctionCallingConfigOrDict = Union[
     AutomaticFunctionCallingConfig, AutomaticFunctionCallingConfigDict
@@ -1446,6 +1532,11 @@ class GenerateContentConfig(_common.BaseModel):
       modalities that the model can return.
       """,
   )
+  media_resolution: Optional[MediaResolution] = Field(
+      default=None,
+      description="""If specified, the media resolution specified will be used.
+    """,
+  )
   speech_config: Optional[SpeechConfigUnion] = Field(
       default=None,
       description="""The speech generation configuration.
@@ -1569,6 +1660,10 @@ class GenerateContentConfigDict(TypedDict, total=False):
       modalities that the model can return.
       """
 
+  media_resolution: Optional[MediaResolution]
+  """If specified, the media resolution specified will be used.
+    """
+
   speech_config: Optional[SpeechConfigUnionDict]
   """The speech generation configuration.
       """
@@ -2314,6 +2409,7 @@ class GenerateContentResponse(_common.BaseModel):
   usage_metadata: Optional[GenerateContentResponseUsageMetadata] = Field(
       default=None, description="""Usage metadata about the response(s)."""
   )
+  automatic_function_calling_history: Optional[list[Content]] = None
   parsed: Union[pydantic.BaseModel, dict] = None
 
   @property
@@ -2393,57 +2489,114 @@ GenerateContentResponseOrDict = Union[
 
 
 class EmbedContentConfig(_common.BaseModel):
+  """Class for configuring optional parameters for the embed_content method."""
 
-  task_type: Optional[str] = Field(default=None, description="""""")
-  title: Optional[str] = Field(default=None, description="""""")
-  output_dimensionality: Optional[int] = Field(default=None, description="""""")
-  mime_type: Optional[str] = Field(default=None, description="""""")
-  auto_truncate: Optional[bool] = Field(default=None, description="""""")
+  http_options: Optional[dict[str, Any]] = Field(
+      default=None, description="""Used to override HTTP request options."""
+  )
+  task_type: Optional[str] = Field(
+      default=None,
+      description="""Type of task for which the embedding will be used.
+      """,
+  )
+  title: Optional[str] = Field(
+      default=None,
+      description="""Title for the text. Only applicable when TaskType is
+      `RETRIEVAL_DOCUMENT`.
+      """,
+  )
+  output_dimensionality: Optional[int] = Field(
+      default=None,
+      description="""Reduced dimension for the output embedding. If set,
+      excessive values in the output embedding are truncated from the end.
+      Supported by newer models since 2024 only. You cannot set this value if
+      using the earlier model (`models/embedding-001`).
+      """,
+  )
+  mime_type: Optional[str] = Field(
+      default=None,
+      description="""Vertex API only. The MIME type of the input.
+      """,
+  )
+  auto_truncate: Optional[bool] = Field(
+      default=None,
+      description="""Vertex API only. Whether to silently truncate inputs longer than
+      the max sequence length. If this option is set to false, oversized inputs
+      will lead to an INVALID_ARGUMENT error, similar to other text APIs.
+      """,
+  )
 
 
 class EmbedContentConfigDict(TypedDict, total=False):
+  """Class for configuring optional parameters for the embed_content method."""
+
+  http_options: Optional[dict[str, Any]]
+  """Used to override HTTP request options."""
 
   task_type: Optional[str]
-  """"""
+  """Type of task for which the embedding will be used.
+      """
 
   title: Optional[str]
-  """"""
+  """Title for the text. Only applicable when TaskType is
+      `RETRIEVAL_DOCUMENT`.
+      """
 
   output_dimensionality: Optional[int]
-  """"""
+  """Reduced dimension for the output embedding. If set,
+      excessive values in the output embedding are truncated from the end.
+      Supported by newer models since 2024 only. You cannot set this value if
+      using the earlier model (`models/embedding-001`).
+      """
 
   mime_type: Optional[str]
-  """"""
+  """Vertex API only. The MIME type of the input.
+      """
 
   auto_truncate: Optional[bool]
-  """"""
+  """Vertex API only. Whether to silently truncate inputs longer than
+      the max sequence length. If this option is set to false, oversized inputs
+      will lead to an INVALID_ARGUMENT error, similar to other text APIs.
+      """
 
 
 EmbedContentConfigOrDict = Union[EmbedContentConfig, EmbedContentConfigDict]
 
 
 class _EmbedContentParameters(_common.BaseModel):
+  """Parameters for the embed_content method."""
 
   model: Optional[str] = Field(
       default=None,
       description="""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>`_.""",
   )
-  contents: Optional[ContentListUnion] = Field(default=None, description="""""")
-  config: Optional[EmbedContentConfig] = Field(default=None, description="""""")
+  contents: Optional[ContentListUnion] = Field(
+      default=None,
+      description="""The content to embed. Only the `parts.text` fields will be counted.
+      """,
+  )
+  config: Optional[EmbedContentConfig] = Field(
+      default=None,
+      description="""Configuration that contains optional parameters.
+      """,
+  )
 
 
 class _EmbedContentParametersDict(TypedDict, total=False):
+  """Parameters for the embed_content method."""
 
   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>`_."""
 
   contents: Optional[ContentListUnionDict]
-  """"""
+  """The content to embed. Only the `parts.text` fields will be counted.
+      """
 
   config: Optional[EmbedContentConfigDict]
-  """"""
+  """Configuration that contains optional parameters.
+      """
 
 
 _EmbedContentParametersOrDict = Union[
@@ -2452,18 +2605,32 @@ _EmbedContentParametersOrDict = Union[
 
 
 class ContentEmbeddingStatistics(_common.BaseModel):
+  """Statistics of the input text associated with the result of content embedding."""
 
-  truncated: Optional[bool] = Field(default=None, description="""""")
-  token_count: Optional[float] = Field(default=None, description="""""")
+  truncated: Optional[bool] = Field(
+      default=None,
+      description="""Vertex API only. If the input text was truncated due to having
+      a length longer than the allowed maximum input.
+      """,
+  )
+  token_count: Optional[float] = Field(
+      default=None,
+      description="""Vertex API only. Number of tokens of the input text.
+      """,
+  )
 
 
 class ContentEmbeddingStatisticsDict(TypedDict, total=False):
+  """Statistics of the input text associated with the result of content embedding."""
 
   truncated: Optional[bool]
-  """"""
+  """Vertex API only. If the input text was truncated due to having
+      a length longer than the allowed maximum input.
+      """
 
   token_count: Optional[float]
-  """"""
+  """Vertex API only. Number of tokens of the input text.
+      """
 
 
 ContentEmbeddingStatisticsOrDict = Union[
@@ -2472,36 +2639,55 @@ ContentEmbeddingStatisticsOrDict = Union[
 
 
 class ContentEmbedding(_common.BaseModel):
+  """The embedding generated from an input content."""
 
-  values: Optional[list[float]] = Field(default=None, description="""""")
+  values: Optional[list[float]] = Field(
+      default=None,
+      description="""A list of floats representing an embedding.
+      """,
+  )
   statistics: Optional[ContentEmbeddingStatistics] = Field(
-      default=None, description=""""""
+      default=None,
+      description="""Vertex API only. Statistics of the input text associated with this
+      embedding.
+      """,
   )
 
 
 class ContentEmbeddingDict(TypedDict, total=False):
+  """The embedding generated from an input content."""
 
   values: Optional[list[float]]
-  """"""
+  """A list of floats representing an embedding.
+      """
 
   statistics: Optional[ContentEmbeddingStatisticsDict]
-  """"""
+  """Vertex API only. Statistics of the input text associated with this
+      embedding.
+      """
 
 
 ContentEmbeddingOrDict = Union[ContentEmbedding, ContentEmbeddingDict]
 
 
 class EmbedContentMetadata(_common.BaseModel):
+  """Request-level metadata for the Vertex Embed Content API."""
 
   billable_character_count: Optional[int] = Field(
-      default=None, description=""""""
+      default=None,
+      description="""Vertex API only. The total number of billable characters included
+      in the request.
+      """,
   )
 
 
 class EmbedContentMetadataDict(TypedDict, total=False):
+  """Request-level metadata for the Vertex Embed Content API."""
 
   billable_character_count: Optional[int]
-  """"""
+  """Vertex API only. The total number of billable characters included
+      in the request.
+      """
 
 
 EmbedContentMetadataOrDict = Union[
@@ -2510,22 +2696,32 @@ EmbedContentMetadataOrDict = Union[
 
 
 class EmbedContentResponse(_common.BaseModel):
+  """Response for the embed_content method."""
 
   embeddings: Optional[list[ContentEmbedding]] = Field(
-      default=None, description=""""""
+      default=None,
+      description="""The embeddings for each request, in the same order as provided in
+      the batch request.
+      """,
   )
   metadata: Optional[EmbedContentMetadata] = Field(
-      default=None, description=""""""
+      default=None,
+      description="""Vertex API only. Metadata about the request.
+      """,
   )
 
 
 class EmbedContentResponseDict(TypedDict, total=False):
+  """Response for the embed_content method."""
 
   embeddings: Optional[list[ContentEmbeddingDict]]
-  """"""
+  """The embeddings for each request, in the same order as provided in
+      the batch request.
+      """
 
   metadata: Optional[EmbedContentMetadataDict]
-  """"""
+  """Vertex API only. Metadata about the request.
+      """
 
 
 EmbedContentResponseOrDict = Union[
@@ -2536,6 +2732,9 @@ EmbedContentResponseOrDict = Union[
 class GenerateImageConfig(_common.BaseModel):
   """Class that represents the config for generating an image."""
 
+  http_options: Optional[dict[str, Any]] = Field(
+      default=None, description="""Used to override HTTP request options."""
+  )
   output_gcs_uri: Optional[str] = Field(
       default=None,
       description="""Cloud Storage URI used to store the generated images.
@@ -2616,6 +2815,9 @@ class GenerateImageConfig(_common.BaseModel):
 class GenerateImageConfigDict(TypedDict, total=False):
   """Class that represents the config for generating an image."""
 
+  http_options: Optional[dict[str, Any]]
+  """Used to override HTTP request options."""
+
   output_gcs_uri: Optional[str]
   """Cloud Storage URI used to store the generated images.
       """
@@ -3102,6 +3304,9 @@ _ReferenceImageAPIOrDict = Union[_ReferenceImageAPI, _ReferenceImageAPIDict]
 class EditImageConfig(_common.BaseModel):
   """Configuration for editing an image."""
 
+  http_options: Optional[dict[str, Any]] = Field(
+      default=None, description="""Used to override HTTP request options."""
+  )
   output_gcs_uri: Optional[str] = Field(
       default=None,
       description="""Cloud Storage URI used to store the generated images.
@@ -3176,6 +3381,9 @@ class EditImageConfig(_common.BaseModel):
 class EditImageConfigDict(TypedDict, total=False):
   """Configuration for editing an image."""
 
+  http_options: Optional[dict[str, Any]]
+  """Used to override HTTP request options."""
+
   output_gcs_uri: Optional[str]
   """Cloud Storage URI used to store the generated images.
       """
@@ -3300,6 +3508,9 @@ class _UpscaleImageAPIConfig(_common.BaseModel):
   to be modifiable or exposed to users in the SDK method.
   """
 
+  http_options: Optional[dict[str, Any]] = Field(
+      default=None, description="""Used to override HTTP request options."""
+  )
   upscale_factor: Optional[str] = Field(
       default=None,
       description="""The factor to which the image will be upscaled.""",
@@ -3329,6 +3540,9 @@ class _UpscaleImageAPIConfigDict(TypedDict, total=False):
   to be modifiable or exposed to users in the SDK method.
   """
 
+  http_options: Optional[dict[str, Any]]
+  """Used to override HTTP request options."""
+
   upscale_factor: Optional[str]
   """The factor to which the image will be upscaled."""
 
@@ -4013,10 +4227,31 @@ ComputeTokensResponseOrDict = Union[
 ]
 
 
+class GetTuningJobConfig(_common.BaseModel):
+  """Optional parameters for tunings.get method."""
+
+  http_options: Optional[dict[str, Any]] = Field(
+      default=None, description="""Used to override HTTP request options."""
+  )
+
+
+class GetTuningJobConfigDict(TypedDict, total=False):
+  """Optional parameters for tunings.get method."""
+
+  http_options: Optional[dict[str, Any]]
+  """Used to override HTTP request options."""
+
+
+GetTuningJobConfigOrDict = Union[GetTuningJobConfig, GetTuningJobConfigDict]
+
+
 class _GetTuningJobParameters(_common.BaseModel):
   """Parameters for the get method."""
 
   name: Optional[str] = Field(default=None, description="""""")
+  config: Optional[GetTuningJobConfig] = Field(
+      default=None, description="""Optional parameters for the request."""
+  )
 
 
 class _GetTuningJobParametersDict(TypedDict, total=False):
@@ -4025,6 +4260,9 @@ class _GetTuningJobParametersDict(TypedDict, total=False):
   name: Optional[str]
   """"""
 
+  config: Optional[GetTuningJobConfigDict]
+  """Optional parameters for the request."""
+
 
 _GetTuningJobParametersOrDict = Union[
     _GetTuningJobParameters, _GetTuningJobParametersDict
@@ -5045,6 +5283,9 @@ TuningValidationDatasetOrDict = Union[
 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.""",
@@ -5080,6 +5321,9 @@ class CreateTuningJobConfig(_common.BaseModel):
 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."""
 
@@ -5205,6 +5449,9 @@ DistillationValidationDatasetOrDict = Union[
 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.""",
@@ -5233,6 +5480,9 @@ class CreateDistillationJobConfig(_common.BaseModel):
 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."""
 
@@ -6139,22 +6389,39 @@ DeleteFileResponseOrDict = Union[DeleteFileResponse, DeleteFileResponseDict]
 class BatchJobSource(_common.BaseModel):
   """Config class for `src` parameter."""
 
-  format: Optional[str] = Field(default=None, description="""""")
-  gcs_uri: Optional[list[str]] = Field(default=None, description="""""")
-  bigquery_uri: Optional[str] = Field(default=None, description="""""")
+  format: Optional[str] = Field(
+      default=None,
+      description="""Storage format of the input files. Must be one of:
+      'jsonl', 'bigquery'.
+      """,
+  )
+  gcs_uri: Optional[list[str]] = Field(
+      default=None,
+      description="""The Google Cloud Storage URIs to input files.
+      """,
+  )
+  bigquery_uri: Optional[str] = Field(
+      default=None,
+      description="""The BigQuery URI to input table.
+      """,
+  )
 
 
 class BatchJobSourceDict(TypedDict, total=False):
   """Config class for `src` parameter."""
 
   format: Optional[str]
-  """"""
+  """Storage format of the input files. Must be one of:
+      'jsonl', 'bigquery'.
+      """
 
   gcs_uri: Optional[list[str]]
-  """"""
+  """The Google Cloud Storage URIs to input files.
+      """
 
   bigquery_uri: Optional[str]
-  """"""
+  """The BigQuery URI to input table.
+      """
 
 
 BatchJobSourceOrDict = Union[BatchJobSource, BatchJobSourceDict]
@@ -6163,22 +6430,39 @@ BatchJobSourceOrDict = Union[BatchJobSource, BatchJobSourceDict]
 class BatchJobDestination(_common.BaseModel):
   """Config class for `des` parameter."""
 
-  format: Optional[str] = Field(default=None, description="""""")
-  gcs_uri: Optional[str] = Field(default=None, description="""""")
-  bigquery_uri: Optional[str] = Field(default=None, description="""""")
+  format: Optional[str] = Field(
+      default=None,
+      description="""Storage format of the output files. Must be one of:
+      'jsonl', 'bigquery'.
+      """,
+  )
+  gcs_uri: Optional[str] = Field(
+      default=None,
+      description="""The Google Cloud Storage URI to the output file.
+      """,
+  )
+  bigquery_uri: Optional[str] = Field(
+      default=None,
+      description="""The BigQuery URI to the output table.
+      """,
+  )
 
 
 class BatchJobDestinationDict(TypedDict, total=False):
   """Config class for `des` parameter."""
 
   format: Optional[str]
-  """"""
+  """Storage format of the output files. Must be one of:
+      'jsonl', 'bigquery'.
+      """
 
   gcs_uri: Optional[str]
-  """"""
+  """The Google Cloud Storage URI to the output file.
+      """
 
   bigquery_uri: Optional[str]
-  """"""
+  """The BigQuery URI to the output table.
+      """
 
 
 BatchJobDestinationOrDict = Union[BatchJobDestination, BatchJobDestinationDict]
@@ -6190,8 +6474,17 @@ class CreateBatchJobConfig(_common.BaseModel):
   http_options: Optional[dict[str, Any]] = Field(
       default=None, description="""Used to override HTTP request options."""
   )
-  display_name: Optional[str] = Field(default=None, description="""""")
-  dest: Optional[str] = Field(default=None, description="""""")
+  display_name: Optional[str] = Field(
+      default=None,
+      description="""The user-defined name of this BatchJob.
+      """,
+  )
+  dest: Optional[str] = Field(
+      default=None,
+      description="""GCS or Bigquery URI prefix for the output predictions. Example:
+      "gs://path/to/output/data" or "bq://projectId.bqDatasetId.bqTableId".
+      """,
+  )
 
 
 class CreateBatchJobConfigDict(TypedDict, total=False):
@@ -6201,10 +6494,13 @@ class CreateBatchJobConfigDict(TypedDict, total=False):
   """Used to override HTTP request options."""
 
   display_name: Optional[str]
-  """"""
+  """The user-defined name of this BatchJob.
+      """
 
   dest: Optional[str]
-  """"""
+  """GCS or Bigquery URI prefix for the output predictions. Example:
+      "gs://path/to/output/data" or "bq://projectId.bqDatasetId.bqTableId".
+      """
 
 
 CreateBatchJobConfigOrDict = Union[
@@ -6215,10 +6511,21 @@ CreateBatchJobConfigOrDict = Union[
 class _CreateBatchJobParameters(_common.BaseModel):
   """Config class for batches.create parameters."""
 
-  model: Optional[str] = Field(default=None, description="""""")
-  src: Optional[str] = Field(default=None, description="""""")
+  model: Optional[str] = Field(
+      default=None,
+      description="""The name of the model to produces the predictions via the BatchJob.
+      """,
+  )
+  src: Optional[str] = Field(
+      default=None,
+      description="""GCS URI(-s) or Bigquery URI to your input data to run batch job.
+      Example: "gs://path/to/input/data" or "bq://projectId.bqDatasetId.bqTableId".
+      """,
+  )
   config: Optional[CreateBatchJobConfig] = Field(
-      default=None, description=""""""
+      default=None,
+      description="""Optional parameters for creating a BatchJob.
+      """,
   )
 
 
@@ -6226,13 +6533,17 @@ class _CreateBatchJobParametersDict(TypedDict, total=False):
   """Config class for batches.create parameters."""
 
   model: Optional[str]
-  """"""
+  """The name of the model to produces the predictions via the BatchJob.
+      """
 
   src: Optional[str]
-  """"""
+  """GCS URI(-s) or Bigquery URI to your input data to run batch job.
+      Example: "gs://path/to/input/data" or "bq://projectId.bqDatasetId.bqTableId".
+      """
 
   config: Optional[CreateBatchJobConfigDict]
-  """"""
+  """Optional parameters for creating a BatchJob.
+      """
 
 
 _CreateBatchJobParametersOrDict = Union[
@@ -6303,9 +6614,21 @@ class BatchJob(_common.BaseModel):
       default=None,
       description="""Output only. Time when the Job was most recently updated.""",
   )
-  model: Optional[str] = Field(default=None, description="""""")
-  src: Optional[BatchJobSource] = Field(default=None, description="""""")
-  dest: Optional[BatchJobDestination] = Field(default=None, description="""""")
+  model: Optional[str] = Field(
+      default=None,
+      description="""The name of the model that produces the predictions via the BatchJob.
+      """,
+  )
+  src: Optional[BatchJobSource] = Field(
+      default=None,
+      description="""Configuration for the input data.
+      """,
+  )
+  dest: Optional[BatchJobDestination] = Field(
+      default=None,
+      description="""Configuration for the output data.
+      """,
+  )
 
 
 class BatchJobDict(TypedDict, total=False):
@@ -6336,29 +6659,65 @@ class BatchJobDict(TypedDict, total=False):
   """Output only. Time when the Job was most recently updated."""
 
   model: Optional[str]
-  """"""
+  """The name of the model that produces the predictions via the BatchJob.
+      """
 
   src: Optional[BatchJobSourceDict]
-  """"""
+  """Configuration for the input data.
+      """
 
   dest: Optional[BatchJobDestinationDict]
-  """"""
+  """Configuration for the output data.
+      """
 
 
 BatchJobOrDict = Union[BatchJob, BatchJobDict]
 
 
+class GetBatchJobConfig(_common.BaseModel):
+  """Optional parameters."""
+
+  http_options: Optional[dict[str, Any]] = Field(
+      default=None, description="""Used to override HTTP request options."""
+  )
+
+
+class GetBatchJobConfigDict(TypedDict, total=False):
+  """Optional parameters."""
+
+  http_options: Optional[dict[str, Any]]
+  """Used to override HTTP request options."""
+
+
+GetBatchJobConfigOrDict = Union[GetBatchJobConfig, GetBatchJobConfigDict]
+
+
 class _GetBatchJobParameters(_common.BaseModel):
   """Config class for batches.get parameters."""
 
-  name: Optional[str] = Field(default=None, description="""""")
+  name: Optional[str] = Field(
+      default=None,
+      description="""A fully-qualified BatchJob resource name or ID.
+    Example: "projects/.../locations/.../batchPredictionJobs/456"
+    or "456" when project and location are initialized in the client.
+    """,
+  )
+  config: Optional[GetBatchJobConfig] = Field(
+      default=None, description="""Optional parameters for the request."""
+  )
 
 
 class _GetBatchJobParametersDict(TypedDict, total=False):
   """Config class for batches.get parameters."""
 
   name: Optional[str]
-  """"""
+  """A fully-qualified BatchJob resource name or ID.
+    Example: "projects/.../locations/.../batchPredictionJobs/456"
+    or "456" when project and location are initialized in the client.
+    """
+
+  config: Optional[GetBatchJobConfigDict]
+  """Optional parameters for the request."""
 
 
 _GetBatchJobParametersOrDict = Union[
@@ -6366,17 +6725,52 @@ _GetBatchJobParametersOrDict = Union[
 ]
 
 
+class CancelBatchJobConfig(_common.BaseModel):
+  """Optional parameters."""
+
+  http_options: Optional[dict[str, Any]] = Field(
+      default=None, description="""Used to override HTTP request options."""
+  )
+
+
+class CancelBatchJobConfigDict(TypedDict, total=False):
+  """Optional parameters."""
+
+  http_options: Optional[dict[str, Any]]
+  """Used to override HTTP request options."""
+
+
+CancelBatchJobConfigOrDict = Union[
+    CancelBatchJobConfig, CancelBatchJobConfigDict
+]
+
+
 class _CancelBatchJobParameters(_common.BaseModel):
   """Config class for batches.cancel parameters."""
 
-  name: Optional[str] = Field(default=None, description="""""")
+  name: Optional[str] = Field(
+      default=None,
+      description="""A fully-qualified BatchJob resource name or ID.
+    Example: "projects/.../locations/.../batchPredictionJobs/456"
+    or "456" when project and location are initialized in the client.
+    """,
+  )
+  config: Optional[CancelBatchJobConfig] = Field(
+      default=None, description="""Optional parameters for the request."""
+  )
 
 
 class _CancelBatchJobParametersDict(TypedDict, total=False):
   """Config class for batches.cancel parameters."""
 
   name: Optional[str]
-  """"""
+  """A fully-qualified BatchJob resource name or ID.
+    Example: "projects/.../locations/.../batchPredictionJobs/456"
+    or "456" when project and location are initialized in the client.
+    """
+
+  config: Optional[CancelBatchJobConfigDict]
+  """Optional parameters for the request."""
 
 
 _CancelBatchJobParametersOrDict = Union[
@@ -6457,14 +6851,23 @@ ListBatchJobResponseOrDict = Union[
 class _DeleteBatchJobParameters(_common.BaseModel):
   """Config class for batches.delete parameters."""
 
-  name: Optional[str] = Field(default=None, description="""""")
+  name: Optional[str] = Field(
+      default=None,
+      description="""A fully-qualified BatchJob resource name or ID.
+    Example: "projects/.../locations/.../batchPredictionJobs/456"
+    or "456" when project and location are initialized in the client.
+    """,
+  )
 
 
 class _DeleteBatchJobParametersDict(TypedDict, total=False):
   """Config class for batches.delete parameters."""
 
   name: Optional[str]
-  """"""
+  """A fully-qualified BatchJob resource name or ID.
+    Example: "projects/.../locations/.../batchPredictionJobs/456"
+    or "456" when project and location are initialized in the client.
+    """
 
 
 _DeleteBatchJobParametersOrDict = Union[
@@ -6733,6 +7136,9 @@ class UpscaleImageConfig(_common.BaseModel):
   <https://cloud.google.com/vertex-ai/generative-ai/docs/model-reference/imagen-api>`_.
   """
 
+  http_options: Optional[dict[str, Any]] = Field(
+      default=None, description="""Used to override HTTP request options."""
+  )
   upscale_factor: Optional[str] = Field(
       default=None,
       description="""The factor to which the image will be upscaled.""",
@@ -6761,6 +7167,9 @@ class UpscaleImageConfigDict(TypedDict, total=False):
   <https://cloud.google.com/vertex-ai/generative-ai/docs/model-reference/imagen-api>`_.
   """
 
+  http_options: Optional[dict[str, Any]]
+  """Used to override HTTP request options."""
+
   upscale_factor: Optional[str]
   """The factor to which the image will be upscaled."""
 
@@ -7289,12 +7698,6 @@ class LiveServerMessage(_common.BaseModel):
       return None
     text = ""
     for part in self.server_content.model_turn.parts:
-      for field_name, field_value in part.dict(exclude={"text"}).items():
-        if field_value is not None:
-          raise ValueError(
-              "LiveServerMessage.text only supports text parts, but got"
-              f" {field_name} part{part}"
-          )
       if isinstance(part.text, str):
         text += part.text
     return text if text else None
@@ -7311,12 +7714,6 @@ class LiveServerMessage(_common.BaseModel):
       return None
     concatenated_data = b""
     for part in self.server_content.model_turn.parts:
-      for field_name, field_value in part.dict(exclude={"inline_data"}).items():
-        if field_value is not None:
-          raise ValueError(
-              "LiveServerMessage.text only supports inline_data parts, but got"
-              f" {field_name} part{part}"
-          )
       if part.inline_data and isinstance(part.inline_data.data, bytes):
         concatenated_data += part.inline_data.data
     return concatenated_data if len(concatenated_data) > 0 else None