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

google/genai/types.py

@@ -13,6 +13,8 @@
 # limitations under the License.
 #
 
+# Code generated by the Google Gen AI SDK generator DO NOT EDIT.
+
 import datetime
 import inspect
 import json
@@ -569,7 +571,9 @@ class Content(_common.BaseModel):
   )
   role: Optional[str] = Field(
       default=None,
-      description="""Optional. The producer of the content. Must be either 'user' or 'model'. Useful to set for multi-turn conversations, otherwise can be left blank or unset.""",
+      description="""Optional. The producer of the content. Must be either 'user' or
+      'model'. Useful to set for multi-turn conversations, otherwise can be
+      left blank or unset. If role is not specified, SDK will determine the role.""",
   )
 
 
@@ -581,7 +585,9 @@ class ContentDict(TypedDict, total=False):
       a different IANA MIME type."""
 
   role: Optional[str]
-  """Optional. The producer of the content. Must be either 'user' or 'model'. Useful to set for multi-turn conversations, otherwise can be left blank or unset."""
+  """Optional. The producer of the content. Must be either 'user' or
+      'model'. Useful to set for multi-turn conversations, otherwise can be
+      left blank or unset. If role is not specified, SDK will determine the role."""
 
 
 ContentOrDict = Union[Content, ContentDict]
@@ -598,7 +604,7 @@ class Schema(_common.BaseModel):
   Represents a select subset of an OpenAPI 3.0 schema object.
   """
 
-  min_items: Optional[str] = Field(
+  min_items: Optional[int] = Field(
       default=None,
       description="""Optional. Minimum number of the elements for Type.ARRAY.""",
   )
@@ -625,22 +631,22 @@ class Schema(_common.BaseModel):
       default=None,
       description="""Optional. The value should be validated against any (one or more) of the subschemas in the list.""",
   )
-  max_length: Optional[str] = Field(
+  max_length: Optional[int] = Field(
       default=None,
       description="""Optional. Maximum length of the Type.STRING""",
   )
   title: Optional[str] = Field(
       default=None, description="""Optional. The title of the Schema."""
   )
-  min_length: Optional[str] = Field(
+  min_length: Optional[int] = Field(
       default=None,
       description="""Optional. SCHEMA FIELDS FOR TYPE STRING Minimum length of the Type.STRING""",
   )
-  min_properties: Optional[str] = Field(
+  min_properties: Optional[int] = Field(
       default=None,
       description="""Optional. Minimum number of the properties for Type.OBJECT.""",
   )
-  max_items: Optional[str] = Field(
+  max_items: Optional[int] = Field(
       default=None,
       description="""Optional. Maximum number of the elements for Type.ARRAY.""",
   )
@@ -652,7 +658,7 @@ class Schema(_common.BaseModel):
       default=None,
       description="""Optional. Indicates if the value may be null.""",
   )
-  max_properties: Optional[str] = Field(
+  max_properties: Optional[int] = Field(
       default=None,
       description="""Optional. Maximum number of the properties for Type.OBJECT.""",
   )
@@ -690,7 +696,7 @@ class SchemaDict(TypedDict, total=False):
   Represents a select subset of an OpenAPI 3.0 schema object.
   """
 
-  min_items: Optional[str]
+  min_items: Optional[int]
   """Optional. Minimum number of the elements for Type.ARRAY."""
 
   example: Optional[Any]
@@ -711,19 +717,19 @@ class SchemaDict(TypedDict, total=False):
   any_of: list["SchemaDict"]
   """Optional. The value should be validated against any (one or more) of the subschemas in the list."""
 
-  max_length: Optional[str]
+  max_length: Optional[int]
   """Optional. Maximum length of the Type.STRING"""
 
   title: Optional[str]
   """Optional. The title of the Schema."""
 
-  min_length: Optional[str]
+  min_length: Optional[int]
   """Optional. SCHEMA FIELDS FOR TYPE STRING Minimum length of the Type.STRING"""
 
-  min_properties: Optional[str]
+  min_properties: Optional[int]
   """Optional. Minimum number of the properties for Type.OBJECT."""
 
-  max_items: Optional[str]
+  max_items: Optional[int]
   """Optional. Maximum number of the elements for Type.ARRAY."""
 
   maximum: Optional[float]
@@ -732,7 +738,7 @@ class SchemaDict(TypedDict, total=False):
   nullable: Optional[bool]
   """Optional. Indicates if the value may be null."""
 
-  max_properties: Optional[str]
+  max_properties: Optional[int]
   """Optional. Maximum number of the properties for Type.OBJECT."""
 
   type: Optional[Type]
@@ -820,11 +826,35 @@ 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."""
+  @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(
+      cls,
+      func: Callable,
+      variant: Literal["GOOGLE_AI", "VERTEX_AI", "DEFAULT"] = "GOOGLE_AI",
+  ) -> "FunctionDeclaration":
+    """Converts a function to a FunctionDeclaration based on an API endpoint.
+
+    Supported endpoints are: 'GOOGLE_AI', 'VERTEX_AI', or 'DEFAULT'.
+    """
     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():
       if param.kind in (
@@ -833,7 +863,7 @@ class FunctionDeclaration(_common.BaseModel):
           inspect.Parameter.POSITIONAL_ONLY,
       ):
         schema = _automatic_function_calling_util._parse_schema_from_parameter(
-            client, param, func.__name__
+            variant, param, func.__name__
         )
         parameters_properties[name] = schema
     declaration = FunctionDeclaration(
@@ -845,13 +875,13 @@ class FunctionDeclaration(_common.BaseModel):
           type="OBJECT",
           properties=parameters_properties,
       )
-      if client.vertexai:
+      if variant == "VERTEX_AI":
         declaration.parameters.required = (
             _automatic_function_calling_util._get_required_fields(
                 declaration.parameters
             )
         )
-    if not client.vertexai:
+    if not variant == "VERTEX_AI":
       return declaration
 
     return_annotation = inspect.signature(func).return_annotation
@@ -860,7 +890,7 @@ class FunctionDeclaration(_common.BaseModel):
 
     declaration.response = (
         _automatic_function_calling_util._parse_schema_from_parameter(
-            client,
+            variant,
             inspect.Parameter(
                 "return_value",
                 inspect.Parameter.POSITIONAL_OR_KEYWORD,
@@ -871,6 +901,14 @@ class FunctionDeclaration(_common.BaseModel):
     )
     return declaration
 
+  @classmethod
+  def from_function(cls, client, func: Callable) -> "FunctionDeclaration":
+    """Converts a function to a FunctionDeclaration."""
+    return cls.from_function_with_options(
+        variant=cls._get_variant(client),
+        func=func,
+    )
+
 
 class FunctionDeclarationDict(TypedDict, total=False):
   """Defines a function that the model can generate JSON inputs for.
@@ -2417,7 +2455,10 @@ 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] = None
+  parsed: Union[pydantic.BaseModel, dict] = Field(
+      default=None,
+      description="""Parsed response if response_schema is provided. Not available for streaming.""",
+  )
 
   @property
   def text(self) -> Optional[str]:
@@ -2453,6 +2494,28 @@ class GenerateContentResponse(_common.BaseModel):
     # part.text == '' is different from part.text is None
     return text if any_text_part_text else None
 
+  @property
+  def function_calls(self) -> Optional[list[FunctionCall]]:
+    """Returns the list of function calls in the response."""
+    if (
+        not self.candidates
+        or not self.candidates[0].content
+        or not self.candidates[0].content.parts
+    ):
+      return None
+    if len(self.candidates) > 1:
+      logging.warning(
+          "Warning: there are multiple candidates in the response, returning"
+          " function calls from the first one."
+      )
+    function_calls = [
+        part.function_call
+        for part in self.candidates[0].content.parts
+        if part.function_call is not None
+    ]
+
+    return function_calls if function_calls else None
+
   @classmethod
   def _from_response(
       cls, response: dict[str, object], kwargs: dict[str, object]
@@ -2467,12 +2530,21 @@ class GenerateContentResponse(_common.BaseModel):
         response_schema, pydantic.BaseModel
     ):
       # Pydantic schema.
-      result.parsed = response_schema.model_validate_json(result.text)
+      try:
+        result.parsed = response_schema.model_validate_json(result.text)
+      # may not be a valid json per stream response
+      except pydantic.ValidationError:
+        pass
+
     elif isinstance(response_schema, dict) or isinstance(
         response_schema, pydantic.BaseModel
     ):
       # JSON schema.
-      result.parsed = json.loads(result.text)
+      try:
+        result.parsed = json.loads(result.text)
+      # may not be a valid json per stream response
+      except json.decoder.JSONDecodeError:
+        pass
 
     return result
 
@@ -3522,10 +3594,6 @@ class _UpscaleImageAPIConfig(_common.BaseModel):
   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.""",
-  )
   include_rai_reason: Optional[bool] = Field(
       default=None,
       description="""Whether to include a reason for filtered-out images in the
@@ -3554,9 +3622,6 @@ class _UpscaleImageAPIConfigDict(TypedDict, total=False):
   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."""
-
   include_rai_reason: Optional[bool]
   """Whether to include a reason for filtered-out images in the
       response."""
@@ -3589,6 +3654,10 @@ class _UpscaleImageAPIParameters(_common.BaseModel):
   image: Optional[Image] = Field(
       default=None, description="""The input image to upscale."""
   )
+  upscale_factor: Optional[str] = Field(
+      default=None,
+      description="""The factor to upscale the image (x2 or x4).""",
+  )
   config: Optional[_UpscaleImageAPIConfig] = Field(
       default=None, description="""Configuration for upscaling."""
   )
@@ -3603,6 +3672,9 @@ class _UpscaleImageAPIParametersDict(TypedDict, total=False):
   image: Optional[ImageDict]
   """The input image to upscale."""
 
+  upscale_factor: Optional[str]
+  """The factor to upscale the image (x2 or x4)."""
+
   config: Optional[_UpscaleImageAPIConfigDict]
   """Configuration for upscaling."""
 
@@ -3791,13 +3863,23 @@ ModelOrDict = Union[Model, ModelDict]
 
 class ListModelsConfig(_common.BaseModel):
 
+  http_options: Optional[dict[str, Any]] = 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="""""")
+  query_base: Optional[bool] = Field(
+      default=None,
+      description="""Set true to list base models, false to list tuned models.""",
+  )
 
 
 class ListModelsConfigDict(TypedDict, total=False):
 
+  http_options: Optional[dict[str, Any]]
+  """Used to override HTTP request options."""
+
   page_size: Optional[int]
   """"""
 
@@ -3807,6 +3889,9 @@ class ListModelsConfigDict(TypedDict, total=False):
   filter: Optional[str]
   """"""
 
+  query_base: Optional[bool]
+  """Set true to list base models, false to list tuned models."""
+
 
 ListModelsConfigOrDict = Union[ListModelsConfig, ListModelsConfigDict]
 
@@ -4193,7 +4278,7 @@ class TokensInfo(_common.BaseModel):
       default=None,
       description="""Optional. Optional fields for the role from the corresponding Content.""",
   )
-  token_ids: Optional[list[str]] = Field(
+  token_ids: Optional[list[int]] = Field(
       default=None, description="""A list of token ids from the input."""
   )
   tokens: Optional[list[bytes]] = Field(
@@ -4207,7 +4292,7 @@ class TokensInfoDict(TypedDict, total=False):
   role: Optional[str]
   """Optional. Optional fields for the role from the corresponding Content."""
 
-  token_ids: Optional[list[str]]
+  token_ids: Optional[list[int]]
   """A list of token ids from the input."""
 
   tokens: Optional[list[bytes]]
@@ -4355,7 +4440,7 @@ class SupervisedHyperParameters(_common.BaseModel):
   adapter_size: Optional[AdapterSize] = Field(
       default=None, description="""Optional. Adapter size for tuning."""
   )
-  epoch_count: Optional[str] = Field(
+  epoch_count: Optional[int] = Field(
       default=None,
       description="""Optional. Number of complete passes the model makes over the entire training dataset during training.""",
   )
@@ -4371,7 +4456,7 @@ class SupervisedHyperParametersDict(TypedDict, total=False):
   adapter_size: Optional[AdapterSize]
   """Optional. Adapter size for tuning."""
 
-  epoch_count: Optional[str]
+  epoch_count: Optional[int]
   """Optional. Number of complete passes the model makes over the entire training dataset during training."""
 
   learning_rate_multiplier: Optional[float]
@@ -4420,7 +4505,7 @@ SupervisedTuningSpecOrDict = Union[
 class DatasetDistributionDistributionBucket(_common.BaseModel):
   """Dataset bucket used to create a histogram for the distribution given a population of values."""
 
-  count: Optional[str] = Field(
+  count: Optional[int] = Field(
       default=None,
       description="""Output only. Number of values in the bucket.""",
   )
@@ -4435,7 +4520,7 @@ class DatasetDistributionDistributionBucket(_common.BaseModel):
 class DatasetDistributionDistributionBucketDict(TypedDict, total=False):
   """Dataset bucket used to create a histogram for the distribution given a population of values."""
 
-  count: Optional[str]
+  count: Optional[int]
   """Output only. Number of values in the bucket."""
 
   left: Optional[float]
@@ -4521,19 +4606,19 @@ DatasetDistributionOrDict = Union[DatasetDistribution, DatasetDistributionDict]
 class DatasetStats(_common.BaseModel):
   """Statistics computed over a tuning dataset."""
 
-  total_billable_character_count: Optional[str] = Field(
+  total_billable_character_count: Optional[int] = Field(
       default=None,
       description="""Output only. Number of billable characters in the tuning dataset.""",
   )
-  total_tuning_character_count: Optional[str] = Field(
+  total_tuning_character_count: Optional[int] = Field(
       default=None,
       description="""Output only. Number of tuning characters in the tuning dataset.""",
   )
-  tuning_dataset_example_count: Optional[str] = Field(
+  tuning_dataset_example_count: Optional[int] = Field(
       default=None,
       description="""Output only. Number of examples in the tuning dataset.""",
   )
-  tuning_step_count: Optional[str] = Field(
+  tuning_step_count: Optional[int] = Field(
       default=None,
       description="""Output only. Number of tuning steps for this Tuning Job.""",
   )
@@ -4558,16 +4643,16 @@ class DatasetStats(_common.BaseModel):
 class DatasetStatsDict(TypedDict, total=False):
   """Statistics computed over a tuning dataset."""
 
-  total_billable_character_count: Optional[str]
+  total_billable_character_count: Optional[int]
   """Output only. Number of billable characters in the tuning dataset."""
 
-  total_tuning_character_count: Optional[str]
+  total_tuning_character_count: Optional[int]
   """Output only. Number of tuning characters in the tuning dataset."""
 
-  tuning_dataset_example_count: Optional[str]
+  tuning_dataset_example_count: Optional[int]
   """Output only. Number of examples in the tuning dataset."""
 
-  tuning_step_count: Optional[str]
+  tuning_step_count: Optional[int]
   """Output only. Number of tuning steps for this Tuning Job."""
 
   user_dataset_examples: Optional[list[ContentDict]]
@@ -4646,7 +4731,7 @@ SupervisedTuningDatasetDistributionDatasetBucketOrDict = Union[
 class SupervisedTuningDatasetDistribution(_common.BaseModel):
   """Dataset distribution for Supervised Tuning."""
 
-  billable_sum: Optional[str] = Field(
+  billable_sum: Optional[int] = Field(
       default=None,
       description="""Output only. Sum of a given population of values that are billable.""",
   )
@@ -4680,7 +4765,7 @@ class SupervisedTuningDatasetDistribution(_common.BaseModel):
       default=None,
       description="""Output only. The 95th percentile of the values in the population.""",
   )
-  sum: Optional[str] = Field(
+  sum: Optional[int] = Field(
       default=None,
       description="""Output only. Sum of a given population of values.""",
   )
@@ -4689,7 +4774,7 @@ class SupervisedTuningDatasetDistribution(_common.BaseModel):
 class SupervisedTuningDatasetDistributionDict(TypedDict, total=False):
   """Dataset distribution for Supervised Tuning."""
 
-  billable_sum: Optional[str]
+  billable_sum: Optional[int]
   """Output only. Sum of a given population of values that are billable."""
 
   buckets: Optional[list[SupervisedTuningDatasetDistributionDatasetBucketDict]]
@@ -4713,7 +4798,7 @@ class SupervisedTuningDatasetDistributionDict(TypedDict, total=False):
   p95: Optional[float]
   """Output only. The 95th percentile of the values in the population."""
 
-  sum: Optional[str]
+  sum: Optional[int]
   """Output only. Sum of a given population of values."""
 
 
@@ -4725,31 +4810,31 @@ SupervisedTuningDatasetDistributionOrDict = Union[
 class SupervisedTuningDataStats(_common.BaseModel):
   """Tuning data statistics for Supervised Tuning."""
 
-  total_billable_character_count: Optional[str] = Field(
+  total_billable_character_count: Optional[int] = Field(
       default=None,
       description="""Output only. Number of billable characters in the tuning dataset.""",
   )
-  total_billable_token_count: Optional[str] = Field(
+  total_billable_token_count: Optional[int] = Field(
       default=None,
       description="""Output only. Number of billable tokens in the tuning dataset.""",
   )
-  total_truncated_example_count: Optional[str] = Field(
+  total_truncated_example_count: Optional[int] = Field(
       default=None,
       description="""The number of examples in the dataset that have been truncated by any amount.""",
   )
-  total_tuning_character_count: Optional[str] = Field(
+  total_tuning_character_count: Optional[int] = Field(
       default=None,
       description="""Output only. Number of tuning characters in the tuning dataset.""",
   )
-  truncated_example_indices: Optional[list[str]] = Field(
+  truncated_example_indices: Optional[list[int]] = Field(
       default=None,
       description="""A partial sample of the indices (starting from 1) of the truncated examples.""",
   )
-  tuning_dataset_example_count: Optional[str] = Field(
+  tuning_dataset_example_count: Optional[int] = Field(
       default=None,
       description="""Output only. Number of examples in the tuning dataset.""",
   )
-  tuning_step_count: Optional[str] = Field(
+  tuning_step_count: Optional[int] = Field(
       default=None,
       description="""Output only. Number of tuning steps for this Tuning Job.""",
   )
@@ -4780,25 +4865,25 @@ class SupervisedTuningDataStats(_common.BaseModel):
 class SupervisedTuningDataStatsDict(TypedDict, total=False):
   """Tuning data statistics for Supervised Tuning."""
 
-  total_billable_character_count: Optional[str]
+  total_billable_character_count: Optional[int]
   """Output only. Number of billable characters in the tuning dataset."""
 
-  total_billable_token_count: Optional[str]
+  total_billable_token_count: Optional[int]
   """Output only. Number of billable tokens in the tuning dataset."""
 
-  total_truncated_example_count: Optional[str]
+  total_truncated_example_count: Optional[int]
   """The number of examples in the dataset that have been truncated by any amount."""
 
-  total_tuning_character_count: Optional[str]
+  total_tuning_character_count: Optional[int]
   """Output only. Number of tuning characters in the tuning dataset."""
 
-  truncated_example_indices: Optional[list[str]]
+  truncated_example_indices: Optional[list[int]]
   """A partial sample of the indices (starting from 1) of the truncated examples."""
 
-  tuning_dataset_example_count: Optional[str]
+  tuning_dataset_example_count: Optional[int]
   """Output only. Number of examples in the tuning dataset."""
 
-  tuning_step_count: Optional[str]
+  tuning_step_count: Optional[int]
   """Output only. Number of tuning steps for this Tuning Job."""
 
   user_dataset_examples: Optional[list[ContentDict]]
@@ -4874,7 +4959,7 @@ class DistillationHyperParameters(_common.BaseModel):
   adapter_size: Optional[AdapterSize] = Field(
       default=None, description="""Optional. Adapter size for distillation."""
   )
-  epoch_count: Optional[str] = Field(
+  epoch_count: Optional[int] = Field(
       default=None,
       description="""Optional. Number of complete passes the model makes over the entire training dataset during training.""",
   )
@@ -4890,7 +4975,7 @@ class DistillationHyperParametersDict(TypedDict, total=False):
   adapter_size: Optional[AdapterSize]
   """Optional. Adapter size for distillation."""
 
-  epoch_count: Optional[str]
+  epoch_count: Optional[int]
   """Optional. Number of complete passes the model makes over the entire training dataset during training."""
 
   learning_rate_multiplier: Optional[float]
@@ -6492,7 +6577,7 @@ class CreateBatchJobConfig(_common.BaseModel):
   )
   dest: Optional[str] = Field(
       default=None,
-      description="""GCS or Bigquery URI prefix for the output predictions. Example:
+      description="""GCS or BigQuery URI prefix for the output predictions. Example:
       "gs://path/to/output/data" or "bq://projectId.bqDatasetId.bqTableId".
       """,
   )
@@ -6509,7 +6594,7 @@ class CreateBatchJobConfigDict(TypedDict, total=False):
       """
 
   dest: Optional[str]
-  """GCS or Bigquery URI prefix for the output predictions. Example:
+  """GCS or BigQuery URI prefix for the output predictions. Example:
       "gs://path/to/output/data" or "bq://projectId.bqDatasetId.bqTableId".
       """
 
@@ -6529,7 +6614,7 @@ class _CreateBatchJobParameters(_common.BaseModel):
   )
   src: Optional[str] = Field(
       default=None,
-      description="""GCS URI(-s) or Bigquery URI to your input data to run batch job.
+      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".
       """,
   )
@@ -6548,7 +6633,7 @@ class _CreateBatchJobParametersDict(TypedDict, total=False):
       """
 
   src: Optional[str]
-  """GCS URI(-s) or Bigquery URI to your input data to run batch job.
+  """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".
       """
 
@@ -7150,10 +7235,6 @@ class UpscaleImageConfig(_common.BaseModel):
   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.""",
-  )
   include_rai_reason: Optional[bool] = Field(
       default=None,
       description="""Whether to include a reason for filtered-out images in the
@@ -7181,9 +7262,6 @@ class UpscaleImageConfigDict(TypedDict, total=False):
   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."""
-
   include_rai_reason: Optional[bool]
   """Whether to include a reason for filtered-out images in the
       response."""
@@ -7208,6 +7286,10 @@ class UpscaleImageParameters(_common.BaseModel):
   image: Optional[Image] = Field(
       default=None, description="""The input image to upscale."""
   )
+  upscale_factor: Optional[str] = Field(
+      default=None,
+      description="""The factor to upscale the image (x2 or x4).""",
+  )
   config: Optional[UpscaleImageConfig] = Field(
       default=None, description="""Configuration for upscaling."""
   )
@@ -7222,6 +7304,9 @@ class UpscaleImageParametersDict(TypedDict, total=False):
   image: Optional[ImageDict]
   """The input image to upscale."""
 
+  upscale_factor: Optional[str]
+  """The factor to upscale the image (x2 or x4)."""
+
   config: Optional[UpscaleImageConfigDict]
   """Configuration for upscaling."""
 
@@ -7655,7 +7740,7 @@ class LiveServerToolCallCancellation(_common.BaseModel):
   server turns.
   """
 
-  ids: Optional[list[int]] = Field(
+  ids: Optional[list[str]] = Field(
       default=None, description="""The ids of the tool calls to be cancelled."""
   )
 
@@ -7668,7 +7753,7 @@ class LiveServerToolCallCancellationDict(TypedDict, total=False):
   server turns.
   """
 
-  ids: Optional[list[int]]
+  ids: Optional[list[str]]
   """The ids of the tool calls to be cancelled."""
 
 
@@ -7950,13 +8035,13 @@ class LiveClientMessage(_common.BaseModel):
 
   setup: Optional[LiveClientSetup] = Field(
       default=None,
-      description="""Message to be sent in the first and only first client message.""",
+      description="""Message to be sent by the system when connecting to the API. SDK users should not send this message.""",
   )
   client_content: Optional[LiveClientContent] = Field(
       default=None,
       description="""Incremental update of the current conversation delivered from the client.""",
   )
-  realtime_update: Optional[LiveClientRealtimeInput] = Field(
+  realtime_input: Optional[LiveClientRealtimeInput] = Field(
       default=None, description="""User input that is sent in real time."""
   )
   tool_response: Optional[LiveClientToolResponse] = Field(
@@ -7969,12 +8054,12 @@ class LiveClientMessageDict(TypedDict, total=False):
   """Messages sent by the client in the API call."""
 
   setup: Optional[LiveClientSetupDict]
-  """Message to be sent in the first and only first client message."""
+  """Message to be sent by the system when connecting to the API. SDK users should not send this message."""
 
   client_content: Optional[LiveClientContentDict]
   """Incremental update of the current conversation delivered from the client."""
 
-  realtime_update: Optional[LiveClientRealtimeInputDict]
+  realtime_input: Optional[LiveClientRealtimeInputDict]
   """User input that is sent in real time."""
 
   tool_response: Optional[LiveClientToolResponseDict]