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

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/interactions.py

@@ -0,0 +1,17 @@
+# Copyright 2025 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+"""Expose stainless types."""
+
+from ._interactions.types import *

google/genai/live.py

@@ -975,8 +975,9 @@ class AsyncLive(_api_module.BaseModule):
               ).model_dump(exclude_none=True),
           )
       )
-      del request_dict['config']
 
+      del request_dict['config']
+      request_dict = _common.encode_unserializable_types(request_dict)
       setv(request_dict, ['setup', 'model'], transformed_model)
 
       request = json.dumps(request_dict)
@@ -998,7 +999,7 @@ class AsyncLive(_api_module.BaseModule):
           )
       )
       del request_dict['config']
-
+      request_dict = _common.encode_unserializable_types(request_dict)
       setv(request_dict, ['setup', 'model'], transformed_model)
 
       request = json.dumps(request_dict)
@@ -1061,7 +1062,7 @@ class AsyncLive(_api_module.BaseModule):
           )
       )
       del request_dict['config']
-
+      request_dict = _common.encode_unserializable_types(request_dict)
       if (
           getv(
               request_dict, ['setup', 'generationConfig', 'responseModalities']

google/genai/models.py

@@ -1110,6 +1110,13 @@ def _GenerateContentConfig_to_mldev(
         _ImageConfig_to_mldev(getv(from_object, ['image_config']), to_object),
     )
 
+  if getv(from_object, ['enable_enhanced_civic_answers']) is not None:
+    setv(
+        to_object,
+        ['enableEnhancedCivicAnswers'],
+        getv(from_object, ['enable_enhanced_civic_answers']),
+    )
+
   return to_object
 
 
@@ -1269,6 +1276,11 @@ def _GenerateContentConfig_to_vertex(
         _ImageConfig_to_vertex(getv(from_object, ['image_config']), to_object),
     )
 
+  if getv(from_object, ['enable_enhanced_civic_answers']) is not None:
+    raise ValueError(
+        'enable_enhanced_civic_answers parameter is not supported in Vertex AI.'
+    )
+
   return to_object
 
 
@@ -3420,12 +3432,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/tests/__init__.py

@@ -0,0 +1,21 @@
+# Copyright 2025 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+
+"""Tests for the Google GenAI SDK."""
+
+
+import pytest
+pytest.register_assert_rewrite('genai.replay_api_client')

google/genai/tests/afc/__init__.py

@@ -0,0 +1,21 @@
+# Copyright 2025 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+
+"""AFC helper function unit tests.
+
+For AFC end to end test, please write test cases in
+test_generate_content_tools.py module
+"""