google-genai 1.53.0__py3-none-any.whl → 1.54.0__py3-none-any.whl

google/genai/_api_client.py

@@ -1254,7 +1254,7 @@ class BaseApiClient:
               url=http_request.url,
               headers=http_request.headers,
               data=data,
-              timeout=aiohttp.ClientTimeout(connect=http_request.timeout),
+              timeout=aiohttp.ClientTimeout(total=http_request.timeout),
               **self._async_client_session_request_args,
           )
         except (
@@ -1276,7 +1276,7 @@ class BaseApiClient:
               url=http_request.url,
               headers=http_request.headers,
               data=data,
-              timeout=aiohttp.ClientTimeout(connect=http_request.timeout),
+              timeout=aiohttp.ClientTimeout(total=http_request.timeout),
               **self._async_client_session_request_args,
           )
 
@@ -1306,7 +1306,7 @@ class BaseApiClient:
               url=http_request.url,
               headers=http_request.headers,
               data=data,
-              timeout=aiohttp.ClientTimeout(connect=http_request.timeout),
+              timeout=aiohttp.ClientTimeout(total=http_request.timeout),
               **self._async_client_session_request_args,
           )
           await errors.APIError.raise_for_async_response(response)
@@ -1330,7 +1330,7 @@ class BaseApiClient:
               url=http_request.url,
               headers=http_request.headers,
               data=data,
-              timeout=aiohttp.ClientTimeout(connect=http_request.timeout),
+              timeout=aiohttp.ClientTimeout(total=http_request.timeout),
               **self._async_client_session_request_args,
           )
           await errors.APIError.raise_for_async_response(response)
@@ -1718,7 +1718,7 @@ class BaseApiClient:
               url=upload_url,
               data=file_chunk,
               headers=upload_headers,
-              timeout=aiohttp.ClientTimeout(connect=timeout_in_seconds),
+              timeout=aiohttp.ClientTimeout(total=timeout_in_seconds),
           )
 
           if response.headers.get('X-Goog-Upload-Status'):
@@ -1864,7 +1864,7 @@ class BaseApiClient:
           url=http_request.url,
           headers=http_request.headers,
           data=data,
-          timeout=aiohttp.ClientTimeout(connect=http_request.timeout),
+          timeout=aiohttp.ClientTimeout(total=http_request.timeout),
       )
       await errors.APIError.raise_for_async_response(response)
 

google/genai/_live_converters.py

@@ -1265,12 +1265,12 @@ def _SpeechConfig_to_vertex(
     parent_object: Optional[dict[str, Any]] = None,
 ) -> dict[str, Any]:
   to_object: dict[str, Any] = {}
-  if getv(from_object, ['language_code']) is not None:
-    setv(to_object, ['languageCode'], getv(from_object, ['language_code']))
-
   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.'

google/genai/batches.py

@@ -1902,34 +1902,6 @@ class Batches(_api_module.BaseModule):
     self._api_client._verify_response(return_value)
     return return_value
 
-  def list(
-      self, *, config: Optional[types.ListBatchJobsConfigOrDict] = None
-  ) -> Pager[types.BatchJob]:
-    """Lists batch jobs.
-
-    Args:
-      config (ListBatchJobsConfig): Optional configuration for the list request.
-
-    Returns:
-      A Pager object that contains one page of batch jobs. When iterating over
-      the pager, it automatically fetches the next page if there are more.
-
-    Usage:
-
-    .. code-block:: python
-      config = {'page_size': 10}
-      for batch_job in client.batches.list(config):
-        print(batch_job.name)
-    """
-
-    list_request = self._list
-    return Pager(
-        'batch_jobs',
-        list_request,
-        self._list(config=config),
-        config,
-    )
-
   def create(
       self,
       *,
@@ -2025,6 +1997,34 @@ class Batches(_api_module.BaseModule):
     else:
       return self._create_embeddings(model=model, src=src, config=config)
 
+  def list(
+      self, *, config: Optional[types.ListBatchJobsConfigOrDict] = None
+  ) -> Pager[types.BatchJob]:
+    """Lists batch jobs.
+
+    Args:
+      config (ListBatchJobsConfig): Optional configuration for the list request.
+
+    Returns:
+      A Pager object that contains one page of batch jobs. When iterating over
+      the pager, it automatically fetches the next page if there are more.
+
+    Usage:
+
+    .. code-block:: python
+      config = {'page_size': 10}
+      for batch_job in client.batches.list(config):
+        print(batch_job.name)
+    """
+
+    list_request = self._list
+    return Pager(
+        'batch_jobs',
+        list_request,
+        self._list(config=config),
+        config,
+    )
+
 
 class AsyncBatches(_api_module.BaseModule):
 
@@ -2451,33 +2451,6 @@ class AsyncBatches(_api_module.BaseModule):
     self._api_client._verify_response(return_value)
     return return_value
 
-  async def list(
-      self, *, config: Optional[types.ListBatchJobsConfigOrDict] = None
-  ) -> AsyncPager[types.BatchJob]:
-    """Lists batch jobs asynchronously.
-
-    Args:
-      config (ListBatchJobsConfig): Optional configuration for the list request.
-
-    Returns:
-      A Pager object that contains one page of batch jobs. When iterating over
-      the pager, it automatically fetches the next page if there are more.
-
-    Usage:
-
-    .. code-block:: python
-      async for batch_job in await client.aio.batches.list():
-        print(batch_job.name)
-    """
-
-    list_request = self._list
-    return AsyncPager(
-        'batch_jobs',
-        list_request,
-        await self._list(config=config),
-        config,
-    )
-
   async def create(
       self,
       *,
@@ -2578,3 +2551,30 @@ class AsyncBatches(_api_module.BaseModule):
       raise ValueError('Vertex AI does not support batches.create_embeddings.')
     else:
       return await self._create_embeddings(model=model, src=src, config=config)
+
+  async def list(
+      self, *, config: Optional[types.ListBatchJobsConfigOrDict] = None
+  ) -> AsyncPager[types.BatchJob]:
+    """Lists batch jobs asynchronously.
+
+    Args:
+      config (ListBatchJobsConfig): Optional configuration for the list request.
+
+    Returns:
+      A Pager object that contains one page of batch jobs. When iterating over
+      the pager, it automatically fetches the next page if there are more.
+
+    Usage:
+
+    .. code-block:: python
+      async for batch_job in await client.aio.batches.list():
+        print(batch_job.name)
+    """
+
+    list_request = self._list
+    return AsyncPager(
+        'batch_jobs',
+        list_request,
+        await self._list(config=config),
+        config,
+    )

google/genai/errors.py

@@ -15,7 +15,7 @@
 
 """Error classes for the GenAI SDK."""
 
-from typing import Any, Optional, TYPE_CHECKING, Union
+from typing import Any, Callable, Optional, TYPE_CHECKING, Union
 import httpx
 import json
 from . import _common
@@ -53,6 +53,21 @@ class APIError(Exception):
 
     super().__init__(f'{self.code} {self.status}. {self.details}')
 
+  def __reduce__(
+      self,
+  ) -> tuple[Callable[..., 'APIError'], tuple[dict[str, Any]]]:
+    """Returns a tuple that can be used to reconstruct the error for pickling."""
+    state = self.__dict__.copy()
+    return (self.__class__._rebuild, (state,))
+
+  @staticmethod
+  def _rebuild(state: dict[str, Any]) -> 'APIError':
+    """Rebuilds the error from the state."""
+    obj = APIError.__new__(APIError)
+    obj.__dict__.update(state)
+    Exception.__init__(obj, f'{obj.code} {obj.status}. {obj.details}')
+    return obj
+
   def _get_status(self, response_json: Any) -> Any:
     return response_json.get(
         'status', response_json.get('error', {}).get('status', None)

google/genai/file_search_stores.py

@@ -695,36 +695,6 @@ class FileSearchStores(_api_module.BaseModule):
     self._api_client._verify_response(return_value)
     return return_value
 
-  def list(
-      self, *, config: Optional[types.ListFileSearchStoresConfigOrDict] = None
-  ) -> Pager[types.FileSearchStore]:
-    """Lists FileSearchStores.
-
-    Args:
-      config (ListFileSearchStoresConfig): Optional configuration for the list
-        request.
-
-    Returns:
-      A Pager object that contains one page of file search stores. When
-      iterating over
-      the pager, it automatically fetches the next page if there are more.
-
-    Usage:
-
-    .. code-block:: python
-      for file_search_store in client.file_search_stores.list():
-        print(f"file search store: {file_search_store.name} -
-        {file_search_store.display_name}")
-    """
-
-    list_request = self._list
-    return Pager(
-        'file_search_stores',
-        list_request,
-        self._list(config=config),
-        config,
-    )
-
   def upload_to_file_search_store(
       self,
       *,
@@ -801,6 +771,36 @@ class FileSearchStores(_api_module.BaseModule):
         response=response_dict, kwargs={}
     )
 
+  def list(
+      self, *, config: Optional[types.ListFileSearchStoresConfigOrDict] = None
+  ) -> Pager[types.FileSearchStore]:
+    """Lists FileSearchStores.
+
+    Args:
+      config (ListFileSearchStoresConfig): Optional configuration for the list
+        request.
+
+    Returns:
+      A Pager object that contains one page of file search stores. When
+      iterating over
+      the pager, it automatically fetches the next page if there are more.
+
+    Usage:
+
+    .. code-block:: python
+      for file_search_store in client.file_search_stores.list():
+        print(f"file search store: {file_search_store.name} -
+        {file_search_store.display_name}")
+    """
+
+    list_request = self._list
+    return Pager(
+        'file_search_stores',
+        list_request,
+        self._list(config=config),
+        config,
+    )
+
 
 class AsyncFileSearchStores(_api_module.BaseModule):
 
@@ -1189,36 +1189,6 @@ class AsyncFileSearchStores(_api_module.BaseModule):
     self._api_client._verify_response(return_value)
     return return_value
 
-  async def list(
-      self, *, config: Optional[types.ListFileSearchStoresConfigOrDict] = None
-  ) -> AsyncPager[types.FileSearchStore]:
-    """Lists FileSearchStores asynchronously.
-
-    Args:
-      config (ListFileSearchStoresConfig): Optional parameters for the request,
-        such as page_size.
-
-    Returns:
-      A Pager object that contains one page of FileSearchStores. When iterating
-      over
-      the pager, it automatically fetches the next page if there are more.
-
-    Usage:
-
-    .. code-block:: python
-      async for file_search_store in await client.aio.file_search_stores.list():
-        print(f"file search store: {file_search_store.name} -
-        {file_search_store.display_name}")
-    """
-
-    list_request = self._list
-    return AsyncPager(
-        'file_search_stores',
-        list_request,
-        await self._list(config=config),
-        config,
-    )
-
   async def upload_to_file_search_store(
       self,
       *,
@@ -1294,3 +1264,33 @@ class AsyncFileSearchStores(_api_module.BaseModule):
     return types.UploadToFileSearchStoreOperation._from_response(
         response=response_dict, kwargs={}
     )
+
+  async def list(
+      self, *, config: Optional[types.ListFileSearchStoresConfigOrDict] = None
+  ) -> AsyncPager[types.FileSearchStore]:
+    """Lists FileSearchStores asynchronously.
+
+    Args:
+      config (ListFileSearchStoresConfig): Optional parameters for the request,
+        such as page_size.
+
+    Returns:
+      A Pager object that contains one page of FileSearchStores. When iterating
+      over
+      the pager, it automatically fetches the next page if there are more.
+
+    Usage:
+
+    .. code-block:: python
+      async for file_search_store in await client.aio.file_search_stores.list():
+        print(f"file search store: {file_search_store.name} -
+        {file_search_store.display_name}")
+    """
+
+    list_request = self._list
+    return AsyncPager(
+        'file_search_stores',
+        list_request,
+        await self._list(config=config),
+        config,
+    )

google/genai/files.py

@@ -402,34 +402,6 @@ class Files(_api_module.BaseModule):
     self._api_client._verify_response(return_value)
     return return_value
 
-  def list(
-      self, *, config: Optional[types.ListFilesConfigOrDict] = None
-  ) -> Pager[types.File]:
-    """Lists all files from the service.
-
-    Args:
-      config (ListFilesConfig): Optional, configuration for the list method.
-
-    Returns:
-      A Pager object that contains one page of files. When iterating over
-      the pager, it automatically fetches the next page if there are more.
-
-    Usage:
-
-    .. code-block:: python
-
-      for file in client.files.list(config={'page_size': 10}):
-        print(file.name)
-    """
-
-    list_request = self._list
-    return Pager(
-        'files',
-        list_request,
-        self._list(config=config),
-        config,
-    )
-
   def upload(
       self,
       *,
@@ -587,6 +559,34 @@ class Files(_api_module.BaseModule):
 
     return data
 
+  def list(
+      self, *, config: Optional[types.ListFilesConfigOrDict] = None
+  ) -> Pager[types.File]:
+    """Lists all files from the service.
+
+    Args:
+      config (ListFilesConfig): Optional, configuration for the list method.
+
+    Returns:
+      A Pager object that contains one page of files. When iterating over
+      the pager, it automatically fetches the next page if there are more.
+
+    Usage:
+
+    .. code-block:: python
+
+      for file in client.files.list(config={'page_size': 10}):
+        print(file.name)
+    """
+
+    list_request = self._list
+    return Pager(
+        'files',
+        list_request,
+        self._list(config=config),
+        config,
+    )
+
 
 class AsyncFiles(_api_module.BaseModule):
 
@@ -845,34 +845,6 @@ class AsyncFiles(_api_module.BaseModule):
     self._api_client._verify_response(return_value)
     return return_value
 
-  async def list(
-      self, *, config: Optional[types.ListFilesConfigOrDict] = None
-  ) -> AsyncPager[types.File]:
-    """Lists all files from the service asynchronously.
-
-    Args:
-      config (ListFilesConfig): Optional, configuration for the list method.
-
-    Returns:
-      A Pager object that contains one page of files. When iterating over
-      the pager, it automatically fetches the next page if there are more.
-
-    Usage:
-
-    .. code-block:: python
-
-      async for file in await client.aio.files.list(config={'page_size': 10}):
-        print(file.name)
-    """
-
-    list_request = self._list
-    return AsyncPager(
-        'files',
-        list_request,
-        await self._list(config=config),
-        config,
-    )
-
   async def upload(
       self,
       *,
@@ -1019,3 +991,31 @@ class AsyncFiles(_api_module.BaseModule):
     )
 
     return data
+
+  async def list(
+      self, *, config: Optional[types.ListFilesConfigOrDict] = None
+  ) -> AsyncPager[types.File]:
+    """Lists all files from the service asynchronously.
+
+    Args:
+      config (ListFilesConfig): Optional, configuration for the list method.
+
+    Returns:
+      A Pager object that contains one page of files. When iterating over
+      the pager, it automatically fetches the next page if there are more.
+
+    Usage:
+
+    .. code-block:: python
+
+      async for file in await client.aio.files.list(config={'page_size': 10}):
+        print(file.name)
+    """
+
+    list_request = self._list
+    return AsyncPager(
+        'files',
+        list_request,
+        await self._list(config=config),
+        config,
+    )

google/genai/models.py

@@ -3420,12 +3420,12 @@ def _SpeechConfig_to_vertex(
     parent_object: Optional[dict[str, Any]] = None,
 ) -> dict[str, Any]:
   to_object: dict[str, Any] = {}
-  if getv(from_object, ['language_code']) is not None:
-    setv(to_object, ['languageCode'], getv(from_object, ['language_code']))
-
   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.'

google/genai/tunings.py

@@ -867,12 +867,12 @@ def _SpeechConfig_to_vertex(
     root_object: Optional[Union[dict[str, Any], object]] = None,
 ) -> dict[str, Any]:
   to_object: dict[str, Any] = {}
-  if getv(from_object, ['language_code']) is not None:
-    setv(to_object, ['languageCode'], getv(from_object, ['language_code']))
-
   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.'
@@ -1555,33 +1555,6 @@ class Tunings(_api_module.BaseModule):
     self._api_client._verify_response(return_value)
     return return_value
 
-  def list(
-      self, *, config: Optional[types.ListTuningJobsConfigOrDict] = None
-  ) -> Pager[types.TuningJob]:
-    """Lists `TuningJob` objects.
-
-    Args:
-      config: The configuration for the list request.
-
-    Returns:
-      A Pager object that contains one page of tuning jobs. When iterating over
-      the pager, it automatically fetches the next page if there are more.
-
-    Usage:
-
-    .. code-block:: python
-        for tuning_job in client.tunings.list():
-            print(tuning_job.name)
-    """
-
-    list_request = self._list
-    return Pager(
-        'tuning_jobs',
-        list_request,
-        self._list(config=config),
-        config,
-    )
-
   def get(
       self,
       *,
@@ -1683,6 +1656,33 @@ class Tunings(_api_module.BaseModule):
       )
     return tuning_job
 
+  def list(
+      self, *, config: Optional[types.ListTuningJobsConfigOrDict] = None
+  ) -> Pager[types.TuningJob]:
+    """Lists `TuningJob` objects.
+
+    Args:
+      config: The configuration for the list request.
+
+    Returns:
+      A Pager object that contains one page of tuning jobs. When iterating over
+      the pager, it automatically fetches the next page if there are more.
+
+    Usage:
+
+    .. code-block:: python
+        for tuning_job in client.tunings.list():
+            print(tuning_job.name)
+    """
+
+    list_request = self._list
+    return Pager(
+        'tuning_jobs',
+        list_request,
+        self._list(config=config),
+        config,
+    )
+
 
 class AsyncTunings(_api_module.BaseModule):
 
@@ -2049,33 +2049,6 @@ class AsyncTunings(_api_module.BaseModule):
     self._api_client._verify_response(return_value)
     return return_value
 
-  async def list(
-      self, *, config: Optional[types.ListTuningJobsConfigOrDict] = None
-  ) -> AsyncPager[types.TuningJob]:
-    """Lists `TuningJob` objects asynchronously.
-
-    Args:
-      config: The configuration for the list request.
-
-    Returns:
-      A Pager object that contains one page of tuning jobs. When iterating over
-      the pager, it automatically fetches the next page if there are more.
-
-    Usage:
-
-    .. code-block:: python
-        async for tuning_job in await client.aio.tunings.list():
-            print(tuning_job.name)
-    """
-
-    list_request = self._list
-    return AsyncPager(
-        'tuning_jobs',
-        list_request,
-        await self._list(config=config),
-        config,
-    )
-
   async def get(
       self,
       *,
@@ -2171,6 +2144,33 @@ class AsyncTunings(_api_module.BaseModule):
       )
     return tuning_job
 
+  async def list(
+      self, *, config: Optional[types.ListTuningJobsConfigOrDict] = None
+  ) -> AsyncPager[types.TuningJob]:
+    """Lists `TuningJob` objects asynchronously.
+
+    Args:
+      config: The configuration for the list request.
+
+    Returns:
+      A Pager object that contains one page of tuning jobs. When iterating over
+      the pager, it automatically fetches the next page if there are more.
+
+    Usage:
+
+    .. code-block:: python
+        async for tuning_job in await client.aio.tunings.list():
+            print(tuning_job.name)
+    """
+
+    list_request = self._list
+    return AsyncPager(
+        'tuning_jobs',
+        list_request,
+        await self._list(config=config),
+        config,
+    )
+
 
 class _IpythonUtils:
   """Temporary class to hold the IPython related functions."""

google/genai/types.py

@@ -4368,6 +4368,163 @@ class ToolConfigDict(TypedDict, total=False):
 ToolConfigOrDict = Union[ToolConfig, ToolConfigDict]
 
 
+class ReplicatedVoiceConfig(_common.BaseModel):
+  """ReplicatedVoiceConfig is used to configure replicated voice."""
+
+  mime_type: Optional[str] = Field(
+      default=None,
+      description="""The mime type of the replicated voice.
+      """,
+  )
+  voice_sample_audio: Optional[bytes] = Field(
+      default=None,
+      description="""The sample audio of the replicated voice.
+      """,
+  )
+
+
+class ReplicatedVoiceConfigDict(TypedDict, total=False):
+  """ReplicatedVoiceConfig is used to configure replicated voice."""
+
+  mime_type: Optional[str]
+  """The mime type of the replicated voice.
+      """
+
+  voice_sample_audio: Optional[bytes]
+  """The sample audio of the replicated voice.
+      """
+
+
+ReplicatedVoiceConfigOrDict = Union[
+    ReplicatedVoiceConfig, ReplicatedVoiceConfigDict
+]
+
+
+class PrebuiltVoiceConfig(_common.BaseModel):
+  """The configuration for the prebuilt speaker to use."""
+
+  voice_name: Optional[str] = Field(
+      default=None, description="""The name of the preset voice to use."""
+  )
+
+
+class PrebuiltVoiceConfigDict(TypedDict, total=False):
+  """The configuration for the prebuilt speaker to use."""
+
+  voice_name: Optional[str]
+  """The name of the preset voice to use."""
+
+
+PrebuiltVoiceConfigOrDict = Union[PrebuiltVoiceConfig, PrebuiltVoiceConfigDict]
+
+
+class VoiceConfig(_common.BaseModel):
+
+  replicated_voice_config: Optional[ReplicatedVoiceConfig] = Field(
+      default=None,
+      description="""If true, the model will use a replicated voice for the response.""",
+  )
+  prebuilt_voice_config: Optional[PrebuiltVoiceConfig] = Field(
+      default=None,
+      description="""The configuration for the prebuilt voice to use.""",
+  )
+
+
+class VoiceConfigDict(TypedDict, total=False):
+
+  replicated_voice_config: Optional[ReplicatedVoiceConfigDict]
+  """If true, the model will use a replicated voice for the response."""
+
+  prebuilt_voice_config: Optional[PrebuiltVoiceConfigDict]
+  """The configuration for the prebuilt voice to use."""
+
+
+VoiceConfigOrDict = Union[VoiceConfig, VoiceConfigDict]
+
+
+class SpeakerVoiceConfig(_common.BaseModel):
+  """Configuration for a single speaker in a multi speaker setup."""
+
+  speaker: Optional[str] = Field(
+      default=None,
+      description="""Required. The name of the speaker. This should be the same as the speaker name used in the prompt.""",
+  )
+  voice_config: Optional[VoiceConfig] = Field(
+      default=None,
+      description="""Required. The configuration for the voice of this speaker.""",
+  )
+
+
+class SpeakerVoiceConfigDict(TypedDict, total=False):
+  """Configuration for a single speaker in a multi speaker setup."""
+
+  speaker: Optional[str]
+  """Required. The name of the speaker. This should be the same as the speaker name used in the prompt."""
+
+  voice_config: Optional[VoiceConfigDict]
+  """Required. The configuration for the voice of this speaker."""
+
+
+SpeakerVoiceConfigOrDict = Union[SpeakerVoiceConfig, SpeakerVoiceConfigDict]
+
+
+class MultiSpeakerVoiceConfig(_common.BaseModel):
+  """The configuration for the multi-speaker setup.
+
+  This data type is not supported in Vertex AI.
+  """
+
+  speaker_voice_configs: Optional[list[SpeakerVoiceConfig]] = Field(
+      default=None, description="""Required. All the enabled speaker voices."""
+  )
+
+
+class MultiSpeakerVoiceConfigDict(TypedDict, total=False):
+  """The configuration for the multi-speaker setup.
+
+  This data type is not supported in Vertex AI.
+  """
+
+  speaker_voice_configs: Optional[list[SpeakerVoiceConfigDict]]
+  """Required. All the enabled speaker voices."""
+
+
+MultiSpeakerVoiceConfigOrDict = Union[
+    MultiSpeakerVoiceConfig, MultiSpeakerVoiceConfigDict
+]
+
+
+class SpeechConfig(_common.BaseModel):
+
+  voice_config: Optional[VoiceConfig] = Field(
+      default=None,
+      description="""Configuration for the voice of the response.""",
+  )
+  language_code: Optional[str] = Field(
+      default=None,
+      description="""Optional. Language code (ISO 639. e.g. en-US) for the speech synthesization.""",
+  )
+  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.""",
+  )
+
+
+class SpeechConfigDict(TypedDict, total=False):
+
+  voice_config: Optional[VoiceConfigDict]
+  """Configuration for the voice of the response."""
+
+  language_code: Optional[str]
+  """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."""
+
+
+SpeechConfigOrDict = Union[SpeechConfig, SpeechConfigDict]
+
+
 class AutomaticFunctionCallingConfig(_common.BaseModel):
   """The configuration for automatic function calling."""
 
@@ -4792,38 +4949,6 @@ class SafetySettingDict(TypedDict, total=False):
 SafetySettingOrDict = Union[SafetySetting, SafetySettingDict]
 
 
-class SpeechConfig(_common.BaseModel):
-  """The speech generation config."""
-
-  language_code: Optional[str] = Field(
-      default=None,
-      description="""Optional. Language code (ISO 639. e.g. en-US) for the speech synthesization.""",
-  )
-  voice_config: Optional['VoiceConfig'] = Field(
-      default=None, description="""The configuration for the speaker to use."""
-  )
-  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.""",
-  )
-
-
-class SpeechConfigDict(TypedDict, total=False):
-  """The speech generation config."""
-
-  language_code: Optional[str]
-  """Optional. Language code (ISO 639. e.g. en-US) for the speech synthesization."""
-
-  voice_config: Optional['VoiceConfigDict']
-  """The configuration for the speaker to use."""
-
-  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."""
-
-
-SpeechConfigOrDict = Union[SpeechConfig, SpeechConfigDict]
-
-
 SpeechConfigUnion = Union[str, SpeechConfig]
 
 
@@ -9068,95 +9193,6 @@ class DeleteModelResponseDict(TypedDict, total=False):
 DeleteModelResponseOrDict = Union[DeleteModelResponse, DeleteModelResponseDict]
 
 
-class PrebuiltVoiceConfig(_common.BaseModel):
-  """The configuration for the prebuilt speaker to use."""
-
-  voice_name: Optional[str] = Field(
-      default=None, description="""The name of the preset voice to use."""
-  )
-
-
-class PrebuiltVoiceConfigDict(TypedDict, total=False):
-  """The configuration for the prebuilt speaker to use."""
-
-  voice_name: Optional[str]
-  """The name of the preset voice to use."""
-
-
-PrebuiltVoiceConfigOrDict = Union[PrebuiltVoiceConfig, PrebuiltVoiceConfigDict]
-
-
-class VoiceConfig(_common.BaseModel):
-  """The configuration for the voice to use."""
-
-  prebuilt_voice_config: Optional[PrebuiltVoiceConfig] = Field(
-      default=None,
-      description="""The configuration for the prebuilt voice to use.""",
-  )
-
-
-class VoiceConfigDict(TypedDict, total=False):
-  """The configuration for the voice to use."""
-
-  prebuilt_voice_config: Optional[PrebuiltVoiceConfigDict]
-  """The configuration for the prebuilt voice to use."""
-
-
-VoiceConfigOrDict = Union[VoiceConfig, VoiceConfigDict]
-
-
-class SpeakerVoiceConfig(_common.BaseModel):
-  """Configuration for a single speaker in a multi speaker setup."""
-
-  speaker: Optional[str] = Field(
-      default=None,
-      description="""Required. The name of the speaker. This should be the same as the speaker name used in the prompt.""",
-  )
-  voice_config: Optional[VoiceConfig] = Field(
-      default=None,
-      description="""Required. The configuration for the voice of this speaker.""",
-  )
-
-
-class SpeakerVoiceConfigDict(TypedDict, total=False):
-  """Configuration for a single speaker in a multi speaker setup."""
-
-  speaker: Optional[str]
-  """Required. The name of the speaker. This should be the same as the speaker name used in the prompt."""
-
-  voice_config: Optional[VoiceConfigDict]
-  """Required. The configuration for the voice of this speaker."""
-
-
-SpeakerVoiceConfigOrDict = Union[SpeakerVoiceConfig, SpeakerVoiceConfigDict]
-
-
-class MultiSpeakerVoiceConfig(_common.BaseModel):
-  """The configuration for the multi-speaker setup.
-
-  This data type is not supported in Vertex AI.
-  """
-
-  speaker_voice_configs: Optional[list[SpeakerVoiceConfig]] = Field(
-      default=None, description="""Required. All the enabled speaker voices."""
-  )
-
-
-class MultiSpeakerVoiceConfigDict(TypedDict, total=False):
-  """The configuration for the multi-speaker setup.
-
-  This data type is not supported in Vertex AI.
-  """
-
-  speaker_voice_configs: Optional[list[SpeakerVoiceConfigDict]]
-  """Required. All the enabled speaker voices."""
-
-
-MultiSpeakerVoiceConfigOrDict = Union[
-    MultiSpeakerVoiceConfig, MultiSpeakerVoiceConfigDict
-]
-
-
 class GenerationConfig(_common.BaseModel):
   """Generation config."""
 

google/genai/version.py

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

{google_genai-1.53.0.dist-info → google_genai-1.54.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: google-genai
-Version: 1.53.0
+Version: 1.54.0
 Summary: GenAI Python SDK
 Author-email: Google LLC <googleapis-packages@google.com>
 License-Expression: Apache-2.0
@@ -213,7 +213,7 @@ await aclient.aclose()
 ## Client context managers
 
 By using the sync client context manager, it will close the underlying
- sync client when exiting the with block.
+ sync client when exiting the with block and avoid httpx "client has been closed" error like [issues#1763](https://github.com/googleapis/python-genai/issues/1763).
 
 ```python
 from google.genai import Client

{google_genai-1.53.0.dist-info → google_genai-1.54.0.dist-info}/RECORD

@@ -1,13 +1,13 @@
 google/genai/__init__.py,sha256=SKz_9WQKA3R4OpJIDJlgssVfizLNDG2tuWtOD9pxrPE,729
 google/genai/_adapters.py,sha256=Kok38miNYJff2n--l0zEK_hbq0y2rWOH7k75J7SMYbQ,1744
-google/genai/_api_client.py,sha256=xZTh53LjzEnNLvGRjSm8FgEq0Sgg4Dy3Iuc-EejpQeg,67228
+google/genai/_api_client.py,sha256=CrHSu9Hd4rhhEpv5ds5jU_5xk0XwixTY055pRQRbFnE,67216
 google/genai/_api_module.py,sha256=lj8eUWx8_LBGBz-49qz6_ywWm3GYp3d8Bg5JoOHbtbI,902
 google/genai/_automatic_function_calling_util.py,sha256=xXNkJR-pzSMkeSXMz3Jw-kMHFbTJEiRJ3wocuwtWW4I,11627
 google/genai/_base_transformers.py,sha256=wljA6m4tLl4XLGlBC2DNOls5N9-X9tffBq0M7i8jgpw,1034
 google/genai/_base_url.py,sha256=E5H4dew14Y16qfnB3XRnjSCi19cJVlkaMNoM_8ip-PM,1597
 google/genai/_common.py,sha256=9HOP8GtEnTGTF-5A92hQdcPWeaCS6ig3vgnmm-8jiTY,25421
 google/genai/_extra_utils.py,sha256=GXsUfyKAPZaW9_ySvdzSAXThvuzaKwXcAmdHW_hoHpY,25309
-google/genai/_live_converters.py,sha256=sn18ntVlaWsM8sRVXE5oLLnHNY3LjJyStXHFCxFokAE,43721
+google/genai/_live_converters.py,sha256=b1If4I7rpmJu9DMnFqKex0jeu5AVUGeTLo1XSd4HqN4,43721
 google/genai/_local_tokenizer_loader.py,sha256=7yvJfEkWIpw2ueY4ZmoOAamJD-G7bm_9g44glD6bVwM,7147
 google/genai/_mcp_utils.py,sha256=HuWJ8FUjquv40Mf_QjcL5r5yXWrS-JjINsjlOSbbyAc,3870
 google/genai/_operations_converters.py,sha256=zw8DqkgJWHphrw9waAjAiH98QbE2Rnum3zGm02dHQWo,11695
@@ -15,27 +15,27 @@ google/genai/_replay_api_client.py,sha256=-sZY8pM8JulqTfuqIKaKHCy9MNXF7HJnX6fCzy
 google/genai/_test_api_client.py,sha256=4ruFIy5_1qcbKqqIBu3HSQbpSOBrxiecBtDZaTGFR1s,4797
 google/genai/_tokens_converters.py,sha256=sAx3CjSKR2PX2eppuJPvt_LgjYJZF9zh-fSUvDwa0WQ,15431
 google/genai/_transformers.py,sha256=I19zzGXCrRd64lrBPpeawhBq7vifKn3fNUX9iIJ8Kwo,43266
-google/genai/batches.py,sha256=9uTRtJASo77rOUg0TWcm45L8bEnhWySpuxUvaxHZ-O4,78419
+google/genai/batches.py,sha256=MNQga5CkQduCf4fKIy4yNqcSc9YY64hL-dCNGnQTdxg,78419
 google/genai/caches.py,sha256=MPYIElWqWtXkxiZ2FFCZspRVdT9sTrZzWEkBpQ-XII0,48118
 google/genai/chats.py,sha256=pIBw8d13llupLn4a7vP6vnpbzDcvCCrZZ-Q2r8Cvo7g,16652
 google/genai/client.py,sha256=xKiAg6h4kB_l9uBwgJzSlSjZ0icyCMsWPTFvhJ0_HJQ,13526
 google/genai/documents.py,sha256=GuYWVgRaYBzHv9aMgcZBrVkFlpBeaAeZsfhgvC_SLTA,16218
-google/genai/errors.py,sha256=6UynlDiPinuh_Sf3RzRQ-pE4kNVQ7ESGGApcFh3_jU8,7582
-google/genai/file_search_stores.py,sha256=-5Sa6nZr075Vu9RUURw0S9fttwRmgRFpjq-oFROZKZc,41418
-google/genai/files.py,sha256=bRNveGpTk77_ByjlDQ_vzRhDb9IsfgfJ2tjRnkQTMsI,31766
+google/genai/errors.py,sha256=2Iux-5iGUJzEB6OWE3R18SckMEvNimOfpNCQIYL6cn8,8118
+google/genai/file_search_stores.py,sha256=4EW_hRjkaGYu4gbQyNbiswTWk5nUgooxB5pqMi_WAME,41418
+google/genai/files.py,sha256=g-geVkMmcgAsI7CPJqt3jzucEcjQf6He1fhnscUdxvs,31766
 google/genai/live.py,sha256=ajM3lTDo9eT1KYVF5fLivuQeOtfbJcedQuDXHxmtNPo,41560
 google/genai/live_music.py,sha256=Y7I7jh5SAKgyjBIMLboH0oTnZJ18uOT2SpRDKURvp94,6783
 google/genai/local_tokenizer.py,sha256=EKZ72cV2Zfutlo_efMOPnLRNZN4WQe57rD3G80cF340,14109
-google/genai/models.py,sha256=GTw9-wPbcKRaJSsrc-jyaflUmh5aF7qTXLOFLy97Zls,238803
+google/genai/models.py,sha256=UeqEVqK-JO0fAW4DIbiXMQmejoYlM5kVv0h_QMX4z6A,238803
 google/genai/operations.py,sha256=FXJOnYp2HA5Xqa5mHW67mSa9p1D_EaGGyUoKc09xFPQ,16384
 google/genai/pagers.py,sha256=1i8NXDuKuQznFin5H5-I0hmj6H5YMTYsEJP7S0ITSbY,7158
 google/genai/py.typed,sha256=RsMFoLwBkAvY05t6izop4UHZtqOPLiKp3GkIEizzmQY,40
 google/genai/tokens.py,sha256=4BPW0gGWFeFVk3INkuY2tfREnsrvzQDhouvRI6_F9Q8,12235
-google/genai/tunings.py,sha256=JW6WG0UCjv20G5_oUbL4B8v9ZjJWkkDFyEFGoV0WGPQ,76343
-google/genai/types.py,sha256=mSXysZYf9KtLAbx2-xMAJ12UdrK8lAT5iwtaXWvVjPA,644745
-google/genai/version.py,sha256=osaboP52k3VOgRQg7CrFm17cDJDQBWPUdlcJenuTCPc,627
-google_genai-1.53.0.dist-info/licenses/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
-google_genai-1.53.0.dist-info/METADATA,sha256=khIfjmcy8jFB7TyfaelGM2eRnlmi7hpy83O5QPgZf_w,47712
-google_genai-1.53.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-google_genai-1.53.0.dist-info/top_level.txt,sha256=_1QvSJIhFAGfxb79D6DhB7SUw2X6T4rwnz_LLrbcD3c,7
-google_genai-1.53.0.dist-info/RECORD,,
+google/genai/tunings.py,sha256=JR_ngNjjgIMd_iYDD53wBCYyHfY5V83njmtQOpBNpL0,76343
+google/genai/types.py,sha256=qrRxDiTGbfkog6DhOBhpVkSQCJ6jSXt1iLioLxLOUXs,645693
+google/genai/version.py,sha256=Bh860abHuAZdE4MCKzCPHNCKqbhUGiJldF39ctkV7nU,627
+google_genai-1.54.0.dist-info/licenses/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
+google_genai-1.54.0.dist-info/METADATA,sha256=kD0yIP7EQVoq00wLiIXdkfug9OFfN8J71TPEPiwaIe0,47834
+google_genai-1.54.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+google_genai-1.54.0.dist-info/top_level.txt,sha256=_1QvSJIhFAGfxb79D6DhB7SUw2X6T4rwnz_LLrbcD3c,7
+google_genai-1.54.0.dist-info/RECORD,,