google-genai 1.52.0__tar.gz → 1.54.0__tar.gz

{google_genai-1.52.0/google_genai.egg-info → google_genai-1.54.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: google-genai
-Version: 1.52.0
+Version: 1.54.0
 Summary: GenAI Python SDK
 Author-email: Google LLC <googleapis-packages@google.com>
 License-Expression: Apache-2.0
@@ -20,7 +20,7 @@ Requires-Python: >=3.10
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: anyio<5.0.0,>=4.8.0
-Requires-Dist: google-auth<3.0.0,>=2.14.1
+Requires-Dist: google-auth[requests]<3.0.0,>=2.14.1
 Requires-Dist: httpx<1.0.0,>=0.28.1
 Requires-Dist: pydantic<3.0.0,>=2.9.0
 Requires-Dist: requests<3.0.0,>=2.28.1
@@ -51,6 +51,16 @@ Google's generative models into their Python applications. It supports the
 [Vertex AI](https://cloud.google.com/vertex-ai/generative-ai/docs/learn/overview)
 APIs.
 
+## Code Generation
+
+Generative models are often unaware of recent API and SDK updates and may suggest outdated or legacy code.
+
+We recommend using our Code Generation instructions [codegen_instructions.md](https://raw.githubusercontent.com/googleapis/python-genai/refs/heads/main/codegen_instructions.md) when generating Google Gen AI SDK code to guide your model towards using the more recent SDK features.
+
+Copy and paste the instructions from [this file](https://raw.githubusercontent.com/googleapis/python-genai/refs/heads/main/codegen_instructions.md)
+into your development environment to provide the model with the necessary
+context
+
 ## Installation
 
 ```sh
@@ -91,6 +101,44 @@ client = genai.Client(
 )
 ```
 
+## Using types
+
+All API methods support Pydantic types and dictionaries, which you can access
+from `google.genai.types`. You can import the types module with the following:
+
+```python
+from google.genai import types
+```
+
+Below is an example `generate_content()` call using types from the types module:
+
+```python
+response = client.models.generate_content(
+    model='gemini-2.0-flash-001',
+    contents=types.Part.from_text(text='Why is the sky blue?'),
+    config=types.GenerateContentConfig(
+        temperature=0,
+        top_p=0.95,
+        top_k=20,
+    ),
+)
+```
+
+Alternatively, you can accomplish the same request using dictionaries instead of
+types:
+
+```python
+response = client.models.generate_content(
+    model='gemini-2.0-flash-001',
+    contents={'text': 'Why is the sky blue?'},
+    config={
+        'temperature': 0,
+        'top_p': 0.95,
+        'top_k': 20,
+    },
+)
+```
+
 **(Optional) Using environment variables:**
 
 You can create a client by configuring the necessary environment variables.
@@ -165,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
@@ -583,33 +631,6 @@ response = client.models.generate_content(
 print(response.text)
 ```
 
-### Typed Config
-
-All API methods support Pydantic types for parameters as well as
-dictionaries. You can get the type from `google.genai.types`.
-
-```python
-from google.genai import types
-
-response = client.models.generate_content(
-    model='gemini-2.0-flash-001',
-    contents=types.Part.from_text(text='Why is the sky blue?'),
-    config=types.GenerateContentConfig(
-        temperature=0,
-        top_p=0.95,
-        top_k=20,
-        candidate_count=1,
-        seed=5,
-        max_output_tokens=100,
-        stop_sequences=['STOP!'],
-        presence_penalty=0.0,
-        frequency_penalty=0.0,
-    ),
-)
-
-print(response.text)
-```
-
 ### List Base Models
 
 To retrieve tuned models, see [list tuned models](#list-tuned-models).

{google_genai-1.52.0 → google_genai-1.54.0}/README.md

@@ -15,6 +15,16 @@ Google's generative models into their Python applications. It supports the
 [Vertex AI](https://cloud.google.com/vertex-ai/generative-ai/docs/learn/overview)
 APIs.
 
+## Code Generation
+
+Generative models are often unaware of recent API and SDK updates and may suggest outdated or legacy code.
+
+We recommend using our Code Generation instructions [codegen_instructions.md](https://raw.githubusercontent.com/googleapis/python-genai/refs/heads/main/codegen_instructions.md) when generating Google Gen AI SDK code to guide your model towards using the more recent SDK features.
+
+Copy and paste the instructions from [this file](https://raw.githubusercontent.com/googleapis/python-genai/refs/heads/main/codegen_instructions.md)
+into your development environment to provide the model with the necessary
+context
+
 ## Installation
 
 ```sh
@@ -55,6 +65,44 @@ client = genai.Client(
 )
 ```
 
+## Using types
+
+All API methods support Pydantic types and dictionaries, which you can access
+from `google.genai.types`. You can import the types module with the following:
+
+```python
+from google.genai import types
+```
+
+Below is an example `generate_content()` call using types from the types module:
+
+```python
+response = client.models.generate_content(
+    model='gemini-2.0-flash-001',
+    contents=types.Part.from_text(text='Why is the sky blue?'),
+    config=types.GenerateContentConfig(
+        temperature=0,
+        top_p=0.95,
+        top_k=20,
+    ),
+)
+```
+
+Alternatively, you can accomplish the same request using dictionaries instead of
+types:
+
+```python
+response = client.models.generate_content(
+    model='gemini-2.0-flash-001',
+    contents={'text': 'Why is the sky blue?'},
+    config={
+        'temperature': 0,
+        'top_p': 0.95,
+        'top_k': 20,
+    },
+)
+```
+
 **(Optional) Using environment variables:**
 
 You can create a client by configuring the necessary environment variables.
@@ -129,7 +177,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
@@ -547,33 +595,6 @@ response = client.models.generate_content(
 print(response.text)
 ```
 
-### Typed Config
-
-All API methods support Pydantic types for parameters as well as
-dictionaries. You can get the type from `google.genai.types`.
-
-```python
-from google.genai import types
-
-response = client.models.generate_content(
-    model='gemini-2.0-flash-001',
-    contents=types.Part.from_text(text='Why is the sky blue?'),
-    config=types.GenerateContentConfig(
-        temperature=0,
-        top_p=0.95,
-        top_k=20,
-        candidate_count=1,
-        seed=5,
-        max_output_tokens=100,
-        stop_sequences=['STOP!'],
-        presence_penalty=0.0,
-        frequency_penalty=0.0,
-    ),
-)
-
-print(response.text)
-```
-
 ### List Base Models
 
 To retrieve tuned models, see [list tuned models](#list-tuned-models).

{google_genai-1.52.0 → google_genai-1.54.0}/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-1.52.0 → google_genai-1.54.0}/google/genai/batches.py

@@ -2012,16 +2012,15 @@ class Batches(_api_module.BaseModule):
     Usage:
 
     .. code-block:: python
-
-      batch_jobs = client.batches.list(config={"page_size": 10})
-      for batch_job in batch_jobs:
-        print(f"Batch job: {batch_job.name}, state {batch_job.state}")
+      config = {'page_size': 10}
+      for batch_job in client.batches.list(config):
+        print(batch_job.name)
     """
-    if config is None:
-      config = types.ListBatchJobsConfig()
+
+    list_request = self._list
     return Pager(
         'batch_jobs',
-        self._list,
+        list_request,
         self._list(config=config),
         config,
     )
@@ -2568,17 +2567,14 @@ class AsyncBatches(_api_module.BaseModule):
     Usage:
 
     .. code-block:: python
-
-      batch_jobs = await client.aio.batches.list(config={'page_size': 5})
-      print(f"current page: {batch_jobs.page}")
-      await batch_jobs_pager.next_page()
-      print(f"next page: {batch_jobs_pager.page}")
+      async for batch_job in await client.aio.batches.list():
+        print(batch_job.name)
     """
-    if config is None:
-      config = types.ListBatchJobsConfig()
+
+    list_request = self._list
     return AsyncPager(
         'batch_jobs',
-        self._list,
+        list_request,
         await self._list(config=config),
         config,
     )

{google_genai-1.52.0 → google_genai-1.54.0}/google/genai/caches.py

@@ -1129,15 +1129,6 @@ class Caches(_api_module.BaseModule):
   def _list(
       self, *, config: Optional[types.ListCachedContentsConfigOrDict] = None
   ) -> types.ListCachedContentsResponse:
-    """Lists cached content configurations.
-
-    .. code-block:: python
-
-      cached_contents = client.caches.list(config={'page_size': 2})
-      for cached_content in cached_contents:
-        print(cached_content)
-    """
-
     parameter_model = types._ListCachedContentsParameters(
         config=config,
     )
@@ -1196,9 +1187,28 @@ class Caches(_api_module.BaseModule):
   def list(
       self, *, config: Optional[types.ListCachedContentsConfigOrDict] = None
   ) -> Pager[types.CachedContent]:
+    """Lists cached contents.
+
+    Args:
+      config (ListCachedContentsConfig): Optional configuration for the list
+        request.
+
+    Returns:
+      A Pager object that contains one page of cached contents. When iterating
+      over
+      the pager, it automatically fetches the next page if there are more.
+
+    Usage:
+
+    .. code-block:: python
+      for cached_content in client.caches.list():
+        print(cached_content.name)
+    """
+
+    list_request = self._list
     return Pager(
         'cached_contents',
-        self._list,
+        list_request,
         self._list(config=config),
         config,
     )
@@ -1505,15 +1515,6 @@ class AsyncCaches(_api_module.BaseModule):
   async def _list(
       self, *, config: Optional[types.ListCachedContentsConfigOrDict] = None
   ) -> types.ListCachedContentsResponse:
-    """Lists cached content configurations.
-
-    .. code-block:: python
-
-      cached_contents = await client.aio.caches.list(config={'page_size': 2})
-      async for cached_content in cached_contents:
-        print(cached_content)
-    """
-
     parameter_model = types._ListCachedContentsParameters(
         config=config,
     )
@@ -1574,9 +1575,28 @@ class AsyncCaches(_api_module.BaseModule):
   async def list(
       self, *, config: Optional[types.ListCachedContentsConfigOrDict] = None
   ) -> AsyncPager[types.CachedContent]:
+    """Lists cached contents asynchronously.
+
+    Args:
+      config (ListCachedContentsConfig): Optional configuration for the list
+        request.
+
+    Returns:
+      A Pager object that contains one page of cached contents. When iterating
+      over
+      the pager, it automatically fetches the next page if there are more.
+
+    Usage:
+
+    .. code-block:: python
+      async for cached_content in await client.aio.caches.list():
+        print(cached_content.name)
+    """
+
+    list_request = self._list
     return AsyncPager(
         'cached_contents',
-        self._list,
+        list_request,
         await self._list(config=config),
         config,
     )

{google_genai-1.52.0 → google_genai-1.54.0}/google/genai/documents.py

@@ -249,17 +249,6 @@ class Documents(_api_module.BaseModule):
       parent: str,
       config: Optional[types.ListDocumentsConfigOrDict] = None,
   ) -> types.ListDocumentsResponse:
-    """Lists all Documents in a FileSearchStore.
-
-    Args:
-      parent (str): The name of the FileSearchStore containing the Documents.
-      config (ListDocumentsConfig | None): Optional parameters for the request,
-        such as page_size.
-
-    Returns:
-      ListDocumentsResponse: A paginated list of Documents.
-    """
-
     parameter_model = types._ListDocumentsParameters(
         parent=parent,
         config=config,
@@ -328,6 +317,7 @@ class Documents(_api_module.BaseModule):
       for document in client.documents.list(parent='rag_store_name'):
         print(f"document: {document.name} - {document.display_name}")
     """
+
     list_request = partial(self._list, parent=parent)
     return Pager(
         'documents',
@@ -461,17 +451,6 @@ class AsyncDocuments(_api_module.BaseModule):
       parent: str,
       config: Optional[types.ListDocumentsConfigOrDict] = None,
   ) -> types.ListDocumentsResponse:
-    """Lists all Documents in a FileSearchStore.
-
-    Args:
-      parent (str): The name of the FileSearchStore containing the Documents.
-      config (ListDocumentsConfig | None): Optional parameters for the request,
-        such as page_size.
-
-    Returns:
-      ListDocumentsResponse: A paginated list of Documents.
-    """
-
     parameter_model = types._ListDocumentsParameters(
         parent=parent,
         config=config,
@@ -540,9 +519,10 @@ class AsyncDocuments(_api_module.BaseModule):
     Usage:
     .. code-block:: python
       async for document in await
-      client.documents.list(parent='rag_store_name'):
+      client.aio.documents.list(parent='rag_store_name'):
         print(f"document: {document.name} - {document.display_name}")
     """
+
     list_request = partial(self._list, parent=parent)
     return AsyncPager(
         'documents',

{google_genai-1.52.0 → google_genai-1.54.0}/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-1.52.0 → google_genai-1.54.0}/google/genai/file_search_stores.py

@@ -499,16 +499,6 @@ class FileSearchStores(_api_module.BaseModule):
   def _list(
       self, *, config: Optional[types.ListFileSearchStoresConfigOrDict] = None
   ) -> types.ListFileSearchStoresResponse:
-    """Lists all FileSearchStore owned by the user.
-
-    Args:
-      config (ListFileSearchStoresConfig | None): Optional parameters for the
-        request, such as page_size.
-
-    Returns:
-      ListFileSearchStoresResponse: A paginated list of FileSearchStore.
-    """
-
     parameter_model = types._ListFileSearchStoresParameters(
         config=config,
     )
@@ -802,9 +792,11 @@ class FileSearchStores(_api_module.BaseModule):
         print(f"file search store: {file_search_store.name} -
         {file_search_store.display_name}")
     """
+
+    list_request = self._list
     return Pager(
         'file_search_stores',
-        self._list,
+        list_request,
         self._list(config=config),
         config,
     )
@@ -999,16 +991,6 @@ class AsyncFileSearchStores(_api_module.BaseModule):
   async def _list(
       self, *, config: Optional[types.ListFileSearchStoresConfigOrDict] = None
   ) -> types.ListFileSearchStoresResponse:
-    """Lists all FileSearchStore owned by the user.
-
-    Args:
-      config (ListFileSearchStoresConfig | None): Optional parameters for the
-        request, such as page_size.
-
-    Returns:
-      ListFileSearchStoresResponse: A paginated list of FileSearchStore.
-    """
-
     parameter_model = types._ListFileSearchStoresParameters(
         config=config,
     )
@@ -1289,8 +1271,8 @@ class AsyncFileSearchStores(_api_module.BaseModule):
     """Lists FileSearchStores asynchronously.
 
     Args:
-      config (ListFileSearchStoresConfig): Optional configuration for the list
-        request.
+      config (ListFileSearchStoresConfig): Optional parameters for the request,
+        such as page_size.
 
     Returns:
       A Pager object that contains one page of FileSearchStores. When iterating
@@ -1300,13 +1282,15 @@ class AsyncFileSearchStores(_api_module.BaseModule):
     Usage:
 
     .. code-block:: python
-      async for file_search_store in await client.file_search_stores.list():
+      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',
-        self._list,
+        list_request,
         await self._list(config=config),
         config,
     )