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

google/genai/caches.py

@@ -13,6 +13,8 @@
 # limitations under the License.
 #
 
+# Code generated by the Google Gen AI SDK generator DO NOT EDIT.
+
 from typing import Optional, Union
 from urllib.parse import urlencode
 from . import _common
@@ -1272,14 +1274,14 @@ class Caches(_common.BaseModule):
         config=config,
     )
 
-    if self.api_client.vertexai:
+    if self._api_client.vertexai:
       request_dict = _CreateCachedContentParameters_to_vertex(
-          self.api_client, parameter_model
+          self._api_client, parameter_model
       )
       path = 'cachedContents'.format_map(request_dict.get('_url'))
     else:
       request_dict = _CreateCachedContentParameters_to_mldev(
-          self.api_client, parameter_model
+          self._api_client, parameter_model
       )
       path = 'cachedContents'.format_map(request_dict.get('_url'))
     query_params = request_dict.get('_query')
@@ -1289,21 +1291,23 @@ class Caches(_common.BaseModule):
     config = request_dict.pop('config', None)
     http_options = config.pop('httpOptions', None) if config else None
     request_dict = _common.convert_to_dict(request_dict)
-    request_dict = _common.apply_base64_encoding(request_dict)
+    request_dict = _common.encode_unserializable_types(request_dict)
 
-    response_dict = self.api_client.request(
+    response_dict = self._api_client.request(
         'post', path, request_dict, http_options
     )
 
-    if self.api_client.vertexai:
-      response_dict = _CachedContent_from_vertex(self.api_client, response_dict)
+    if self._api_client.vertexai:
+      response_dict = _CachedContent_from_vertex(
+          self._api_client, response_dict
+      )
     else:
-      response_dict = _CachedContent_from_mldev(self.api_client, response_dict)
+      response_dict = _CachedContent_from_mldev(self._api_client, response_dict)
 
     return_value = types.CachedContent._from_response(
         response_dict, parameter_model
     )
-    self.api_client._verify_response(return_value)
+    self._api_client._verify_response(return_value)
     return return_value
 
   def get(
@@ -1325,14 +1329,14 @@ class Caches(_common.BaseModule):
         config=config,
     )
 
-    if self.api_client.vertexai:
+    if self._api_client.vertexai:
       request_dict = _GetCachedContentParameters_to_vertex(
-          self.api_client, parameter_model
+          self._api_client, parameter_model
       )
       path = '{name}'.format_map(request_dict.get('_url'))
     else:
       request_dict = _GetCachedContentParameters_to_mldev(
-          self.api_client, parameter_model
+          self._api_client, parameter_model
       )
       path = '{name}'.format_map(request_dict.get('_url'))
     query_params = request_dict.get('_query')
@@ -1342,21 +1346,23 @@ class Caches(_common.BaseModule):
     config = request_dict.pop('config', None)
     http_options = config.pop('httpOptions', None) if config else None
     request_dict = _common.convert_to_dict(request_dict)
-    request_dict = _common.apply_base64_encoding(request_dict)
+    request_dict = _common.encode_unserializable_types(request_dict)
 
-    response_dict = self.api_client.request(
+    response_dict = self._api_client.request(
         'get', path, request_dict, http_options
     )
 
-    if self.api_client.vertexai:
-      response_dict = _CachedContent_from_vertex(self.api_client, response_dict)
+    if self._api_client.vertexai:
+      response_dict = _CachedContent_from_vertex(
+          self._api_client, response_dict
+      )
     else:
-      response_dict = _CachedContent_from_mldev(self.api_client, response_dict)
+      response_dict = _CachedContent_from_mldev(self._api_client, response_dict)
 
     return_value = types.CachedContent._from_response(
         response_dict, parameter_model
     )
-    self.api_client._verify_response(return_value)
+    self._api_client._verify_response(return_value)
     return return_value
 
   def delete(
@@ -1380,14 +1386,14 @@ class Caches(_common.BaseModule):
         config=config,
     )
 
-    if self.api_client.vertexai:
+    if self._api_client.vertexai:
       request_dict = _DeleteCachedContentParameters_to_vertex(
-          self.api_client, parameter_model
+          self._api_client, parameter_model
       )
       path = '{name}'.format_map(request_dict.get('_url'))
     else:
       request_dict = _DeleteCachedContentParameters_to_mldev(
-          self.api_client, parameter_model
+          self._api_client, parameter_model
       )
       path = '{name}'.format_map(request_dict.get('_url'))
     query_params = request_dict.get('_query')
@@ -1397,25 +1403,25 @@ class Caches(_common.BaseModule):
     config = request_dict.pop('config', None)
     http_options = config.pop('httpOptions', None) if config else None
     request_dict = _common.convert_to_dict(request_dict)
-    request_dict = _common.apply_base64_encoding(request_dict)
+    request_dict = _common.encode_unserializable_types(request_dict)
 
-    response_dict = self.api_client.request(
+    response_dict = self._api_client.request(
         'delete', path, request_dict, http_options
     )
 
-    if self.api_client.vertexai:
+    if self._api_client.vertexai:
       response_dict = _DeleteCachedContentResponse_from_vertex(
-          self.api_client, response_dict
+          self._api_client, response_dict
       )
     else:
       response_dict = _DeleteCachedContentResponse_from_mldev(
-          self.api_client, response_dict
+          self._api_client, response_dict
       )
 
     return_value = types.DeleteCachedContentResponse._from_response(
         response_dict, parameter_model
     )
-    self.api_client._verify_response(return_value)
+    self._api_client._verify_response(return_value)
     return return_value
 
   def update(
@@ -1441,14 +1447,14 @@ class Caches(_common.BaseModule):
         config=config,
     )
 
-    if self.api_client.vertexai:
+    if self._api_client.vertexai:
       request_dict = _UpdateCachedContentParameters_to_vertex(
-          self.api_client, parameter_model
+          self._api_client, parameter_model
       )
       path = '{name}'.format_map(request_dict.get('_url'))
     else:
       request_dict = _UpdateCachedContentParameters_to_mldev(
-          self.api_client, parameter_model
+          self._api_client, parameter_model
       )
       path = '{name}'.format_map(request_dict.get('_url'))
     query_params = request_dict.get('_query')
@@ -1458,21 +1464,23 @@ class Caches(_common.BaseModule):
     config = request_dict.pop('config', None)
     http_options = config.pop('httpOptions', None) if config else None
     request_dict = _common.convert_to_dict(request_dict)
-    request_dict = _common.apply_base64_encoding(request_dict)
+    request_dict = _common.encode_unserializable_types(request_dict)
 
-    response_dict = self.api_client.request(
+    response_dict = self._api_client.request(
         'patch', path, request_dict, http_options
     )
 
-    if self.api_client.vertexai:
-      response_dict = _CachedContent_from_vertex(self.api_client, response_dict)
+    if self._api_client.vertexai:
+      response_dict = _CachedContent_from_vertex(
+          self._api_client, response_dict
+      )
     else:
-      response_dict = _CachedContent_from_mldev(self.api_client, response_dict)
+      response_dict = _CachedContent_from_mldev(self._api_client, response_dict)
 
     return_value = types.CachedContent._from_response(
         response_dict, parameter_model
     )
-    self.api_client._verify_response(return_value)
+    self._api_client._verify_response(return_value)
     return return_value
 
   def _list(
@@ -1491,14 +1499,14 @@ class Caches(_common.BaseModule):
         config=config,
     )
 
-    if self.api_client.vertexai:
+    if self._api_client.vertexai:
       request_dict = _ListCachedContentsParameters_to_vertex(
-          self.api_client, parameter_model
+          self._api_client, parameter_model
       )
       path = 'cachedContents'.format_map(request_dict.get('_url'))
     else:
       request_dict = _ListCachedContentsParameters_to_mldev(
-          self.api_client, parameter_model
+          self._api_client, parameter_model
       )
       path = 'cachedContents'.format_map(request_dict.get('_url'))
     query_params = request_dict.get('_query')
@@ -1508,25 +1516,25 @@ class Caches(_common.BaseModule):
     config = request_dict.pop('config', None)
     http_options = config.pop('httpOptions', None) if config else None
     request_dict = _common.convert_to_dict(request_dict)
-    request_dict = _common.apply_base64_encoding(request_dict)
+    request_dict = _common.encode_unserializable_types(request_dict)
 
-    response_dict = self.api_client.request(
+    response_dict = self._api_client.request(
         'get', path, request_dict, http_options
     )
 
-    if self.api_client.vertexai:
+    if self._api_client.vertexai:
       response_dict = _ListCachedContentsResponse_from_vertex(
-          self.api_client, response_dict
+          self._api_client, response_dict
       )
     else:
       response_dict = _ListCachedContentsResponse_from_mldev(
-          self.api_client, response_dict
+          self._api_client, response_dict
       )
 
     return_value = types.ListCachedContentsResponse._from_response(
         response_dict, parameter_model
     )
-    self.api_client._verify_response(return_value)
+    self._api_client._verify_response(return_value)
     return return_value
 
   def list(
@@ -1574,14 +1582,14 @@ class AsyncCaches(_common.BaseModule):
         config=config,
     )
 
-    if self.api_client.vertexai:
+    if self._api_client.vertexai:
       request_dict = _CreateCachedContentParameters_to_vertex(
-          self.api_client, parameter_model
+          self._api_client, parameter_model
       )
       path = 'cachedContents'.format_map(request_dict.get('_url'))
     else:
       request_dict = _CreateCachedContentParameters_to_mldev(
-          self.api_client, parameter_model
+          self._api_client, parameter_model
       )
       path = 'cachedContents'.format_map(request_dict.get('_url'))
     query_params = request_dict.get('_query')
@@ -1591,21 +1599,23 @@ class AsyncCaches(_common.BaseModule):
     config = request_dict.pop('config', None)
     http_options = config.pop('httpOptions', None) if config else None
     request_dict = _common.convert_to_dict(request_dict)
-    request_dict = _common.apply_base64_encoding(request_dict)
+    request_dict = _common.encode_unserializable_types(request_dict)
 
-    response_dict = await self.api_client.async_request(
+    response_dict = await self._api_client.async_request(
         'post', path, request_dict, http_options
     )
 
-    if self.api_client.vertexai:
-      response_dict = _CachedContent_from_vertex(self.api_client, response_dict)
+    if self._api_client.vertexai:
+      response_dict = _CachedContent_from_vertex(
+          self._api_client, response_dict
+      )
     else:
-      response_dict = _CachedContent_from_mldev(self.api_client, response_dict)
+      response_dict = _CachedContent_from_mldev(self._api_client, response_dict)
 
     return_value = types.CachedContent._from_response(
         response_dict, parameter_model
     )
-    self.api_client._verify_response(return_value)
+    self._api_client._verify_response(return_value)
     return return_value
 
   async def get(
@@ -1627,14 +1637,14 @@ class AsyncCaches(_common.BaseModule):
         config=config,
     )
 
-    if self.api_client.vertexai:
+    if self._api_client.vertexai:
       request_dict = _GetCachedContentParameters_to_vertex(
-          self.api_client, parameter_model
+          self._api_client, parameter_model
       )
       path = '{name}'.format_map(request_dict.get('_url'))
     else:
       request_dict = _GetCachedContentParameters_to_mldev(
-          self.api_client, parameter_model
+          self._api_client, parameter_model
       )
       path = '{name}'.format_map(request_dict.get('_url'))
     query_params = request_dict.get('_query')
@@ -1644,21 +1654,23 @@ class AsyncCaches(_common.BaseModule):
     config = request_dict.pop('config', None)
     http_options = config.pop('httpOptions', None) if config else None
     request_dict = _common.convert_to_dict(request_dict)
-    request_dict = _common.apply_base64_encoding(request_dict)
+    request_dict = _common.encode_unserializable_types(request_dict)
 
-    response_dict = await self.api_client.async_request(
+    response_dict = await self._api_client.async_request(
         'get', path, request_dict, http_options
     )
 
-    if self.api_client.vertexai:
-      response_dict = _CachedContent_from_vertex(self.api_client, response_dict)
+    if self._api_client.vertexai:
+      response_dict = _CachedContent_from_vertex(
+          self._api_client, response_dict
+      )
     else:
-      response_dict = _CachedContent_from_mldev(self.api_client, response_dict)
+      response_dict = _CachedContent_from_mldev(self._api_client, response_dict)
 
     return_value = types.CachedContent._from_response(
         response_dict, parameter_model
     )
-    self.api_client._verify_response(return_value)
+    self._api_client._verify_response(return_value)
     return return_value
 
   async def delete(
@@ -1682,14 +1694,14 @@ class AsyncCaches(_common.BaseModule):
         config=config,
     )
 
-    if self.api_client.vertexai:
+    if self._api_client.vertexai:
       request_dict = _DeleteCachedContentParameters_to_vertex(
-          self.api_client, parameter_model
+          self._api_client, parameter_model
       )
       path = '{name}'.format_map(request_dict.get('_url'))
     else:
       request_dict = _DeleteCachedContentParameters_to_mldev(
-          self.api_client, parameter_model
+          self._api_client, parameter_model
       )
       path = '{name}'.format_map(request_dict.get('_url'))
     query_params = request_dict.get('_query')
@@ -1699,25 +1711,25 @@ class AsyncCaches(_common.BaseModule):
     config = request_dict.pop('config', None)
     http_options = config.pop('httpOptions', None) if config else None
     request_dict = _common.convert_to_dict(request_dict)
-    request_dict = _common.apply_base64_encoding(request_dict)
+    request_dict = _common.encode_unserializable_types(request_dict)
 
-    response_dict = await self.api_client.async_request(
+    response_dict = await self._api_client.async_request(
         'delete', path, request_dict, http_options
     )
 
-    if self.api_client.vertexai:
+    if self._api_client.vertexai:
       response_dict = _DeleteCachedContentResponse_from_vertex(
-          self.api_client, response_dict
+          self._api_client, response_dict
       )
     else:
       response_dict = _DeleteCachedContentResponse_from_mldev(
-          self.api_client, response_dict
+          self._api_client, response_dict
       )
 
     return_value = types.DeleteCachedContentResponse._from_response(
         response_dict, parameter_model
     )
-    self.api_client._verify_response(return_value)
+    self._api_client._verify_response(return_value)
     return return_value
 
   async def update(
@@ -1743,14 +1755,14 @@ class AsyncCaches(_common.BaseModule):
         config=config,
     )
 
-    if self.api_client.vertexai:
+    if self._api_client.vertexai:
       request_dict = _UpdateCachedContentParameters_to_vertex(
-          self.api_client, parameter_model
+          self._api_client, parameter_model
       )
       path = '{name}'.format_map(request_dict.get('_url'))
     else:
       request_dict = _UpdateCachedContentParameters_to_mldev(
-          self.api_client, parameter_model
+          self._api_client, parameter_model
       )
       path = '{name}'.format_map(request_dict.get('_url'))
     query_params = request_dict.get('_query')
@@ -1760,21 +1772,23 @@ class AsyncCaches(_common.BaseModule):
     config = request_dict.pop('config', None)
     http_options = config.pop('httpOptions', None) if config else None
     request_dict = _common.convert_to_dict(request_dict)
-    request_dict = _common.apply_base64_encoding(request_dict)
+    request_dict = _common.encode_unserializable_types(request_dict)
 
-    response_dict = await self.api_client.async_request(
+    response_dict = await self._api_client.async_request(
         'patch', path, request_dict, http_options
     )
 
-    if self.api_client.vertexai:
-      response_dict = _CachedContent_from_vertex(self.api_client, response_dict)
+    if self._api_client.vertexai:
+      response_dict = _CachedContent_from_vertex(
+          self._api_client, response_dict
+      )
     else:
-      response_dict = _CachedContent_from_mldev(self.api_client, response_dict)
+      response_dict = _CachedContent_from_mldev(self._api_client, response_dict)
 
     return_value = types.CachedContent._from_response(
         response_dict, parameter_model
     )
-    self.api_client._verify_response(return_value)
+    self._api_client._verify_response(return_value)
     return return_value
 
   async def _list(
@@ -1793,14 +1807,14 @@ class AsyncCaches(_common.BaseModule):
         config=config,
     )
 
-    if self.api_client.vertexai:
+    if self._api_client.vertexai:
       request_dict = _ListCachedContentsParameters_to_vertex(
-          self.api_client, parameter_model
+          self._api_client, parameter_model
       )
       path = 'cachedContents'.format_map(request_dict.get('_url'))
     else:
       request_dict = _ListCachedContentsParameters_to_mldev(
-          self.api_client, parameter_model
+          self._api_client, parameter_model
       )
       path = 'cachedContents'.format_map(request_dict.get('_url'))
     query_params = request_dict.get('_query')
@@ -1810,25 +1824,25 @@ class AsyncCaches(_common.BaseModule):
     config = request_dict.pop('config', None)
     http_options = config.pop('httpOptions', None) if config else None
     request_dict = _common.convert_to_dict(request_dict)
-    request_dict = _common.apply_base64_encoding(request_dict)
+    request_dict = _common.encode_unserializable_types(request_dict)
 
-    response_dict = await self.api_client.async_request(
+    response_dict = await self._api_client.async_request(
         'get', path, request_dict, http_options
     )
 
-    if self.api_client.vertexai:
+    if self._api_client.vertexai:
       response_dict = _ListCachedContentsResponse_from_vertex(
-          self.api_client, response_dict
+          self._api_client, response_dict
       )
     else:
       response_dict = _ListCachedContentsResponse_from_mldev(
-          self.api_client, response_dict
+          self._api_client, response_dict
       )
 
     return_value = types.ListCachedContentsResponse._from_response(
         response_dict, parameter_model
     )
-    self.api_client._verify_response(return_value)
+    self._api_client._verify_response(return_value)
     return return_value
 
   async def list(

google/genai/chats.py

@@ -21,9 +21,7 @@ from .models import AsyncModels, Models
 from .types import Content, ContentDict, GenerateContentConfigOrDict, GenerateContentResponse, Part, PartUnionDict
 
 
-def _validate_response(
-    response: GenerateContentResponse
-) -> bool:
+def _validate_response(response: GenerateContentResponse) -> bool:
   if not response.candidates:
     return False
   if not response.candidates[0].content:
@@ -77,7 +75,7 @@ class Chat(_BaseChat):
       response = chat.send_message('tell me a story')
     """
 
-    input_content = t.t_content(self._modules.api_client, message)
+    input_content = t.t_content(self._modules._api_client, message)
     response = self._modules.generate_content(
         model=self._model,
         contents=self._curated_history + [input_content],
@@ -113,7 +111,7 @@ class Chat(_BaseChat):
         print(chunk.text)
     """
 
-    input_content = t.t_content(self._modules.api_client, message)
+    input_content = t.t_content(self._modules._api_client, message)
     output_contents = []
     finish_reason = None
     for chunk in self._modules.generate_content_stream(
@@ -184,7 +182,7 @@ class AsyncChat(_BaseChat):
       response = await chat.send_message('tell me a story')
     """
 
-    input_content = t.t_content(self._modules.api_client, message)
+    input_content = t.t_content(self._modules._api_client, message)
     response = await self._modules.generate_content(
         model=self._model,
         contents=self._curated_history + [input_content],
@@ -219,7 +217,7 @@ class AsyncChat(_BaseChat):
         print(chunk.text)
     """
 
-    input_content = t.t_content(self._modules.api_client, message)
+    input_content = t.t_content(self._modules._api_client, message)
     output_contents = []
     finish_reason = None
     async for chunk in self._modules.generate_content_stream(
@@ -240,7 +238,6 @@ class AsyncChat(_BaseChat):
 class AsyncChats:
   """A util class to create async chat sessions."""
 
-
   def __init__(self, modules: AsyncModels):
     self._modules = modules
 

google/genai/client.py

@@ -187,7 +187,8 @@ class Client:
     self._debug_config = debug_config or DebugConfig()
 
     # Throw ValueError if response_payload is set in http_options due to
-    # unpredical behavior when running multiple coroutines through client.aio.
+    # unpredictable behavior when running multiple coroutines through
+    # client.aio.
     if http_options and 'response_payload' in http_options:
       raise ValueError(
           'Setting response_payload in http_options is not supported.'

google/genai/errors.py

@@ -114,7 +114,7 @@ class ServerError(APIError):
   pass
 
 
-class UnkownFunctionCallArgumentError(ValueError):
+class UnknownFunctionCallArgumentError(ValueError):
   """Raised when the function call argument cannot be converted to the parameter annotation."""
 
   pass