google-genai 1.55.0__py3-none-any.whl → 1.56.0__py3-none-any.whl

google/genai/_interactions/types/text_content_param.py

@@ -29,7 +29,6 @@ class TextContentParam(TypedDict, total=False):
     """A text content block."""
 
     type: Required[Literal["text"]]
-    """Used as the OpenAPI type discriminator for the content oneof."""
 
     annotations: Iterable[AnnotationParam]
     """Citation information for model-generated content."""

google/genai/_interactions/types/thinking_level.py

@@ -19,4 +19,4 @@ from typing_extensions import Literal, TypeAlias
 
 __all__ = ["ThinkingLevel"]
 
-ThinkingLevel: TypeAlias = Literal["low", "high"]
+ThinkingLevel: TypeAlias = Literal["minimal", "low", "medium", "high"]

google/genai/_interactions/types/thought_content.py

@@ -32,7 +32,6 @@ class ThoughtContent(BaseModel):
     """A thought content block."""
 
     type: Literal["thought"]
-    """Used as the OpenAPI type discriminator for the content oneof."""
 
     signature: Optional[str] = None
     """Signature to match the backend source to be part of the generation."""

google/genai/_interactions/types/thought_content_param.py

@@ -35,7 +35,6 @@ class ThoughtContentParam(TypedDict, total=False):
     """A thought content block."""
 
     type: Required[Literal["thought"]]
-    """Used as the OpenAPI type discriminator for the content oneof."""
 
     signature: Annotated[Union[str, Base64FileInput], PropertyInfo(format="base64")]
     """Signature to match the backend source to be part of the generation."""

google/genai/_interactions/types/url_context_call_content.py

@@ -28,7 +28,6 @@ class URLContextCallContent(BaseModel):
     """URL context content."""
 
     type: Literal["url_context_call"]
-    """Used as the OpenAPI type discriminator for the content oneof."""
 
     id: Optional[str] = None
     """A unique ID for this specific tool call."""

google/genai/_interactions/types/url_context_call_content_param.py

@@ -28,7 +28,6 @@ class URLContextCallContentParam(TypedDict, total=False):
     """URL context content."""
 
     type: Required[Literal["url_context_call"]]
-    """Used as the OpenAPI type discriminator for the content oneof."""
 
     id: str
     """A unique ID for this specific tool call."""

google/genai/_interactions/types/url_context_result_content.py

@@ -28,7 +28,6 @@ class URLContextResultContent(BaseModel):
     """URL context result content."""
 
     type: Literal["url_context_result"]
-    """Used as the OpenAPI type discriminator for the content oneof."""
 
     call_id: Optional[str] = None
     """ID to match the ID from the url context call block."""

google/genai/_interactions/types/url_context_result_content_param.py

@@ -29,7 +29,6 @@ class URLContextResultContentParam(TypedDict, total=False):
     """URL context result content."""
 
     type: Required[Literal["url_context_result"]]
-    """Used as the OpenAPI type discriminator for the content oneof."""
 
     call_id: str
     """ID to match the ID from the url context call block."""

google/genai/_interactions/types/video_content.py

@@ -28,14 +28,13 @@ class VideoContent(BaseModel):
     """A video content block."""
 
     type: Literal["video"]
-    """Used as the OpenAPI type discriminator for the content oneof."""
 
     data: Optional[str] = None
 
     mime_type: Optional[VideoMimeType] = None
     """The mime type of the video."""
 
-    resolution: Optional[Literal["low", "medium", "high"]] = None
+    resolution: Optional[Literal["low", "medium", "high", "ultra_high"]] = None
     """The resolution of the media."""
 
     uri: Optional[str] = None

google/genai/_interactions/types/video_content_param.py

@@ -32,14 +32,13 @@ class VideoContentParam(TypedDict, total=False):
     """A video content block."""
 
     type: Required[Literal["video"]]
-    """Used as the OpenAPI type discriminator for the content oneof."""
 
     data: Annotated[Union[str, Base64FileInput], PropertyInfo(format="base64")]
 
     mime_type: VideoMimeTypeParam
     """The mime type of the video."""
 
-    resolution: Literal["low", "medium", "high"]
+    resolution: Literal["low", "medium", "high", "ultra_high"]
     """The resolution of the media."""
 
     uri: str

google/genai/_live_converters.py

@@ -226,13 +226,7 @@ def _GenerationConfig_to_vertex(
     setv(to_object, ['seed'], getv(from_object, ['seed']))
 
   if getv(from_object, ['speech_config']) is not None:
-    setv(
-        to_object,
-        ['speechConfig'],
-        _SpeechConfig_to_vertex(
-            getv(from_object, ['speech_config']), to_object
-        ),
-    )
+    setv(to_object, ['speechConfig'], getv(from_object, ['speech_config']))
 
   if getv(from_object, ['stop_sequences']) is not None:
     setv(to_object, ['stopSequences'], getv(from_object, ['stop_sequences']))
@@ -842,10 +836,7 @@ def _LiveConnectConfig_to_vertex(
     setv(
         parent_object,
         ['setup', 'generationConfig', 'speechConfig'],
-        _SpeechConfig_to_vertex(
-            t.t_live_speech_config(getv(from_object, ['speech_config'])),
-            to_object,
-        ),
+        t.t_live_speech_config(getv(from_object, ['speech_config'])),
     )
 
   if getv(from_object, ['thinking_config']) is not None:
@@ -1291,25 +1282,6 @@ def _SessionResumptionConfig_to_mldev(
   return to_object
 
 
-def _SpeechConfig_to_vertex(
-    from_object: Union[dict[str, Any], object],
-    parent_object: Optional[dict[str, Any]] = None,
-) -> dict[str, Any]:
-  to_object: dict[str, Any] = {}
-  if getv(from_object, ['voice_config']) is not None:
-    setv(to_object, ['voiceConfig'], getv(from_object, ['voice_config']))
-
-  if getv(from_object, ['language_code']) is not None:
-    setv(to_object, ['languageCode'], getv(from_object, ['language_code']))
-
-  if getv(from_object, ['multi_speaker_voice_config']) is not None:
-    raise ValueError(
-        'multi_speaker_voice_config parameter is not supported in Vertex AI.'
-    )
-
-  return to_object
-
-
 def _Tool_to_mldev(
     from_object: Union[dict[str, Any], object],
     parent_object: Optional[dict[str, Any]] = None,

google/genai/client.py

@@ -16,7 +16,7 @@
 import asyncio
 import os
 from types import TracebackType
-from typing import Optional, Union
+from typing import Optional, Union, cast
 
 import google.auth
 import pydantic
@@ -116,6 +116,7 @@ class AsyncClient:
     self._nextgen_client_instance = AsyncGeminiNextGenAPIClient(
         base_url=http_opts.base_url,
         api_key=self._api_client.api_key,
+        api_version=http_opts.api_version,
         default_headers=http_opts.headers,
         http_client=http_client,
         # uSDk expects ms, nextgen uses a httpx Timeout -> expects seconds.
@@ -485,6 +486,7 @@ class Client:
     self._nextgen_client_instance = GeminiNextGenAPIClient(
         base_url=http_opts.base_url,
         api_key=self._api_client.api_key,
+        api_version=http_opts.api_version,
         default_headers=http_opts.headers,
         http_client=self._api_client._httpx_client,
         # uSDk expects ms, nextgen uses a httpx Timeout -> expects seconds.

google/genai/models.py

@@ -1258,9 +1258,7 @@ def _GenerateContentConfig_to_vertex(
     setv(
         to_object,
         ['speechConfig'],
-        _SpeechConfig_to_vertex(
-            t.t_speech_config(getv(from_object, ['speech_config'])), to_object
-        ),
+        t.t_speech_config(getv(from_object, ['speech_config'])),
     )
 
   if getv(from_object, ['audio_timestamp']) is not None:
@@ -2446,13 +2444,7 @@ def _GenerationConfig_to_vertex(
     setv(to_object, ['seed'], getv(from_object, ['seed']))
 
   if getv(from_object, ['speech_config']) is not None:
-    setv(
-        to_object,
-        ['speechConfig'],
-        _SpeechConfig_to_vertex(
-            getv(from_object, ['speech_config']), to_object
-        ),
-    )
+    setv(to_object, ['speechConfig'], getv(from_object, ['speech_config']))
 
   if getv(from_object, ['stop_sequences']) is not None:
     setv(to_object, ['stopSequences'], getv(from_object, ['stop_sequences']))
@@ -3427,25 +3419,6 @@ def _SegmentImageSource_to_vertex(
   return to_object
 
 
-def _SpeechConfig_to_vertex(
-    from_object: Union[dict[str, Any], object],
-    parent_object: Optional[dict[str, Any]] = None,
-) -> dict[str, Any]:
-  to_object: dict[str, Any] = {}
-  if getv(from_object, ['voice_config']) is not None:
-    setv(to_object, ['voiceConfig'], getv(from_object, ['voice_config']))
-
-  if getv(from_object, ['language_code']) is not None:
-    setv(to_object, ['languageCode'], getv(from_object, ['language_code']))
-
-  if getv(from_object, ['multi_speaker_voice_config']) is not None:
-    raise ValueError(
-        'multi_speaker_voice_config parameter is not supported in Vertex AI.'
-    )
-
-  return to_object
-
-
 def _ToolConfig_to_mldev(
     from_object: Union[dict[str, Any], object],
     parent_object: Optional[dict[str, Any]] = None,

google/genai/tunings.py

@@ -674,13 +674,7 @@ def _GenerationConfig_to_vertex(
     setv(to_object, ['seed'], getv(from_object, ['seed']))
 
   if getv(from_object, ['speech_config']) is not None:
-    setv(
-        to_object,
-        ['speechConfig'],
-        _SpeechConfig_to_vertex(
-            getv(from_object, ['speech_config']), to_object, root_object
-        ),
-    )
+    setv(to_object, ['speechConfig'], getv(from_object, ['speech_config']))
 
   if getv(from_object, ['stop_sequences']) is not None:
     setv(to_object, ['stopSequences'], getv(from_object, ['stop_sequences']))
@@ -861,26 +855,6 @@ def _ListTuningJobsResponse_from_vertex(
   return to_object
 
 
-def _SpeechConfig_to_vertex(
-    from_object: Union[dict[str, Any], object],
-    parent_object: Optional[dict[str, Any]] = None,
-    root_object: Optional[Union[dict[str, Any], object]] = None,
-) -> dict[str, Any]:
-  to_object: dict[str, Any] = {}
-  if getv(from_object, ['voice_config']) is not None:
-    setv(to_object, ['voiceConfig'], getv(from_object, ['voice_config']))
-
-  if getv(from_object, ['language_code']) is not None:
-    setv(to_object, ['languageCode'], getv(from_object, ['language_code']))
-
-  if getv(from_object, ['multi_speaker_voice_config']) is not None:
-    raise ValueError(
-        'multi_speaker_voice_config parameter is not supported in Vertex AI.'
-    )
-
-  return to_object
-
-
 def _TunedModel_from_mldev(
     from_object: Union[dict[str, Any], object],
     parent_object: Optional[dict[str, Any]] = None,

google/genai/types.py

@@ -86,11 +86,6 @@ else:
 
 if typing.TYPE_CHECKING:
   import yaml
-else:
-  try:
-    import yaml
-  except ImportError:
-    yaml = None
 
 _is_httpx_imported = False
 if typing.TYPE_CHECKING:
@@ -267,14 +262,18 @@ class PhishBlockThreshold(_common.CaseInSensitiveEnum):
 
 
 class ThinkingLevel(_common.CaseInSensitiveEnum):
-  """The level of thoughts tokens that the model should generate."""
+  """The number of thoughts tokens that the model should generate."""
 
   THINKING_LEVEL_UNSPECIFIED = 'THINKING_LEVEL_UNSPECIFIED'
-  """Default value."""
+  """Unspecified thinking level."""
   LOW = 'LOW'
   """Low thinking level."""
+  MEDIUM = 'MEDIUM'
+  """Medium thinking level."""
   HIGH = 'HIGH'
   """High thinking level."""
+  MINIMAL = 'MINIMAL'
+  """MINIMAL thinking level."""
 
 
 class HarmCategory(_common.CaseInSensitiveEnum):
@@ -577,6 +576,8 @@ class PartMediaResolutionLevel(_common.CaseInSensitiveEnum):
   """Media resolution set to medium."""
   MEDIA_RESOLUTION_HIGH = 'MEDIA_RESOLUTION_HIGH'
   """Media resolution set to high."""
+  MEDIA_RESOLUTION_ULTRA_HIGH = 'MEDIA_RESOLUTION_ULTRA_HIGH'
+  """Media resolution set to ultra high."""
 
 
 class ResourceScope(_common.CaseInSensitiveEnum):
@@ -4484,24 +4485,19 @@ SpeakerVoiceConfigOrDict = Union[SpeakerVoiceConfig, SpeakerVoiceConfigDict]
 
 
 class MultiSpeakerVoiceConfig(_common.BaseModel):
-  """The configuration for the multi-speaker setup.
-
-  This data type is not supported in Vertex AI.
-  """
+  """Configuration for a multi-speaker text-to-speech request."""
 
   speaker_voice_configs: Optional[list[SpeakerVoiceConfig]] = Field(
-      default=None, description="""Required. All the enabled speaker voices."""
+      default=None,
+      description="""Required. A list of configurations for the voices of the speakers. Exactly two speaker voice configurations must be provided.""",
   )
 
 
 class MultiSpeakerVoiceConfigDict(TypedDict, total=False):
-  """The configuration for the multi-speaker setup.
-
-  This data type is not supported in Vertex AI.
-  """
+  """Configuration for a multi-speaker text-to-speech request."""
 
   speaker_voice_configs: Optional[list[SpeakerVoiceConfigDict]]
-  """Required. All the enabled speaker voices."""
+  """Required. A list of configurations for the voices of the speakers. Exactly two speaker voice configurations must be provided."""
 
 
 MultiSpeakerVoiceConfigOrDict = Union[
@@ -4521,7 +4517,7 @@ class SpeechConfig(_common.BaseModel):
   )
   multi_speaker_voice_config: Optional[MultiSpeakerVoiceConfig] = Field(
       default=None,
-      description="""Optional. The configuration for the multi-speaker setup. It is mutually exclusive with the voice_config field. This field is not supported in Vertex AI.""",
+      description="""The configuration for a multi-speaker text-to-speech request. This field is mutually exclusive with `voice_config`.""",
   )
 
 
@@ -4534,7 +4530,7 @@ class SpeechConfigDict(TypedDict, total=False):
   """Optional. Language code (ISO 639. e.g. en-US) for the speech synthesization."""
 
   multi_speaker_voice_config: Optional[MultiSpeakerVoiceConfigDict]
-  """Optional. The configuration for the multi-speaker setup. It is mutually exclusive with the voice_config field. This field is not supported in Vertex AI."""
+  """The configuration for a multi-speaker text-to-speech request. This field is mutually exclusive with `voice_config`."""
 
 
 SpeechConfigOrDict = Union[SpeechConfig, SpeechConfigDict]
@@ -4614,7 +4610,7 @@ class ThinkingConfig(_common.BaseModel):
   )
   thinking_level: Optional[ThinkingLevel] = Field(
       default=None,
-      description="""Optional. The level of thoughts tokens that the model should generate.""",
+      description="""Optional. The number of thoughts tokens that the model should generate.""",
   )
 
 
@@ -4630,7 +4626,7 @@ class ThinkingConfigDict(TypedDict, total=False):
       """
 
   thinking_level: Optional[ThinkingLevel]
-  """Optional. The level of thoughts tokens that the model should generate."""
+  """Optional. The number of thoughts tokens that the model should generate."""
 
 
 ThinkingConfigOrDict = Union[ThinkingConfig, ThinkingConfigDict]
@@ -10645,7 +10641,9 @@ class Metric(_common.BaseModel):
     Raises:
         ImportError: If the pyyaml library is not installed.
     """
-    if yaml is None:
+    try:
+      import yaml
+    except ImportError:
       raise ImportError(
           'YAML serialization requires the pyyaml library. Please install'
           " it using 'pip install google-cloud-aiplatform[evaluation]'."

google/genai/version.py

@@ -13,4 +13,4 @@
 # limitations under the License.
 #
 
-__version__ = '1.55.0' # x-release-please-version
+__version__ = '1.56.0' # x-release-please-version