google-genai 0.2.2__py3-none-any.whl → 0.4.0__py3-none-any.whl

google/genai/types.py

@@ -441,6 +441,10 @@ class Part(_common.BaseModel):
   video_metadata: Optional[VideoMetadata] = Field(
       default=None, description="""Metadata for a given video."""
   )
+  thought: Optional[bool] = Field(
+      default=None,
+      description="""Indicates if the part is thought from the model.""",
+  )
   code_execution_result: Optional[CodeExecutionResult] = Field(
       default=None,
       description="""Optional. Result of executing the [ExecutableCode].""",
@@ -525,6 +529,9 @@ class PartDict(TypedDict, total=False):
   video_metadata: Optional[VideoMetadataDict]
   """Metadata for a given video."""
 
+  thought: Optional[bool]
+  """Indicates if the part is thought from the model."""
+
   code_execution_result: Optional[CodeExecutionResultDict]
   """Optional. Result of executing the [ExecutableCode]."""
 
@@ -562,7 +569,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.""",
   )
 
 
@@ -574,7 +583,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]
@@ -591,7 +602,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.""",
   )
@@ -618,22 +629,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.""",
   )
@@ -645,7 +656,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.""",
   )
@@ -683,7 +694,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]
@@ -704,19 +715,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]
@@ -725,7 +736,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]
@@ -813,11 +824,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 (
@@ -826,7 +861,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(
@@ -838,13 +873,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
@@ -853,7 +888,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,
@@ -864,6 +899,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.
@@ -2410,7 +2453,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]:
@@ -2430,18 +2476,44 @@ class GenerateContentResponse(_common.BaseModel):
     text = ""
     any_text_part_text = False
     for part in self.candidates[0].content.parts:
-      for field_name, field_value in part.dict(exclude={"text"}).items():
+      for field_name, field_value in part.dict(
+          exclude={"text", "thought"}
+      ).items():
         if field_value is not None:
           raise ValueError(
               "GenerateContentResponse.text only supports text parts, but got"
               f" {field_name} part{part}"
           )
       if isinstance(part.text, str):
+        if isinstance(part.thought, bool) and part.thought:
+          continue
         any_text_part_text = True
         text += part.text
     # 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]
@@ -2456,12 +2528,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
 
@@ -3511,10 +3592,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
@@ -3543,9 +3620,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."""
@@ -3578,6 +3652,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."""
   )
@@ -3592,6 +3670,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."""
 
@@ -4182,7 +4263,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(
@@ -4196,7 +4277,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]]
@@ -4344,7 +4425,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.""",
   )
@@ -4360,7 +4441,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]
@@ -4409,7 +4490,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.""",
   )
@@ -4424,7 +4505,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]
@@ -4510,19 +4591,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.""",
   )
@@ -4547,16 +4628,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]]
@@ -4635,7 +4716,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.""",
   )
@@ -4669,7 +4750,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.""",
   )
@@ -4678,7 +4759,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]]
@@ -4702,7 +4783,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."""
 
 
@@ -4714,31 +4795,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.""",
   )
@@ -4769,25 +4850,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]]
@@ -4863,7 +4944,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.""",
   )
@@ -4879,7 +4960,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]
@@ -5567,6 +5648,11 @@ class CreateCachedContentConfig(_common.BaseModel):
       description="""The user-generated meaningful display name of the cached content.
       """,
   )
+  contents: Optional[ContentListUnion] = Field(
+      default=None,
+      description="""The content to cache.
+      """,
+  )
   system_instruction: Optional[ContentUnion] = Field(
       default=None,
       description="""Developer set system instruction.
@@ -5600,6 +5686,10 @@ class CreateCachedContentConfigDict(TypedDict, total=False):
   """The user-generated meaningful display name of the cached content.
       """
 
+  contents: Optional[ContentListUnionDict]
+  """The content to cache.
+      """
+
   system_instruction: Optional[ContentUnionDict]
   """Developer set system instruction.
       """
@@ -5625,11 +5715,6 @@ class _CreateCachedContentParameters(_common.BaseModel):
       default=None,
       description="""ID of the model to use. Example: gemini-1.5-flash""",
   )
-  contents: Optional[ContentListUnion] = Field(
-      default=None,
-      description="""The content to cache.
-      """,
-  )
   config: Optional[CreateCachedContentConfig] = Field(
       default=None,
       description="""Configuration that contains optional parameters.
@@ -5643,10 +5728,6 @@ class _CreateCachedContentParametersDict(TypedDict, total=False):
   model: Optional[str]
   """ID of the model to use. Example: gemini-1.5-flash"""
 
-  contents: Optional[ContentListUnionDict]
-  """The content to cache.
-      """
-
   config: Optional[CreateCachedContentConfigDict]
   """Configuration that contains optional parameters.
       """
@@ -6481,7 +6562,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".
       """,
   )
@@ -6498,7 +6579,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".
       """
 
@@ -6518,7 +6599,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".
       """,
   )
@@ -6537,7 +6618,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".
       """
 
@@ -7139,10 +7220,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
@@ -7170,9 +7247,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."""
@@ -7197,6 +7271,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."""
   )
@@ -7211,6 +7289,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."""
 
@@ -7644,7 +7725,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."""
   )
 
@@ -7657,7 +7738,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."""
 
 
@@ -7699,6 +7780,8 @@ class LiveServerMessage(_common.BaseModel):
     text = ""
     for part in self.server_content.model_turn.parts:
       if isinstance(part.text, str):
+        if isinstance(part.thought, bool) and part.thought:
+          continue
         text += part.text
     return text if text else None
 
@@ -7943,7 +8026,7 @@ class LiveClientMessage(_common.BaseModel):
       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(
@@ -7961,7 +8044,7 @@ class LiveClientMessageDict(TypedDict, total=False):
   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]