google-genai 1.20.0__py3-none-any.whl → 1.21.0__py3-none-any.whl

google/genai/chats.py

@@ -250,7 +250,7 @@ class Chat(_BaseChat):
           f"Message must be a valid part type: {types.PartUnion} or"
           f" {types.PartUnionDict}, got {type(message)}"
       )
-    input_content = t.t_content(self._modules._api_client, message)
+    input_content = t.t_content(message)
     response = self._modules.generate_content(
         model=self._model,
         contents=self._curated_history + [input_content],  # type: ignore[arg-type]
@@ -303,7 +303,7 @@ class Chat(_BaseChat):
           f"Message must be a valid part type: {types.PartUnion} or"
           f" {types.PartUnionDict}, got {type(message)}"
       )
-    input_content = t.t_content(self._modules._api_client, message)
+    input_content = t.t_content(message)
     output_contents = []
     finish_reason = None
     is_valid = True
@@ -412,7 +412,7 @@ class AsyncChat(_BaseChat):
           f"Message must be a valid part type: {types.PartUnion} or"
           f" {types.PartUnionDict}, got {type(message)}"
       )
-    input_content = t.t_content(self._modules._api_client, message)
+    input_content = t.t_content(message)
     response = await self._modules.generate_content(
         model=self._model,
         contents=self._curated_history + [input_content],  # type: ignore[arg-type]
@@ -464,7 +464,7 @@ class AsyncChat(_BaseChat):
           f"Message must be a valid part type: {types.PartUnion} or"
           f" {types.PartUnionDict}, got {type(message)}"
       )
-    input_content = t.t_content(self._modules._api_client, message)
+    input_content = t.t_content(message)
 
     async def async_generator():  # type: ignore[no-untyped-def]
       output_contents = []

google/genai/client.py

@@ -31,7 +31,7 @@ from .models import AsyncModels, Models
 from .operations import AsyncOperations, Operations
 from .tokens import AsyncTokens, Tokens
 from .tunings import AsyncTunings, Tunings
-from .types import HttpOptions, HttpOptionsDict
+from .types import HttpOptions, HttpOptionsDict, HttpRetryOptions
 
 
 class AsyncClient:

google/genai/files.py

@@ -16,10 +16,10 @@
 # Code generated by the Google Gen AI SDK generator DO NOT EDIT.
 
 import io
+import json
 import logging
 import mimetypes
 import os
-import pathlib
 from typing import Any, Optional, Union
 from urllib.parse import urlencode
 
@@ -27,7 +27,6 @@ from . import _api_module
 from . import _common
 from . import _transformers as t
 from . import types
-from ._api_client import BaseApiClient
 from ._common import get_value_by_path as getv
 from ._common import set_value_by_path as setv
 from .pagers import AsyncPager, Pager
@@ -36,7 +35,6 @@ logger = logging.getLogger('google_genai.files')
 
 
 def _ListFilesConfig_to_mldev(
-    api_client: BaseApiClient,
     from_object: Union[dict[str, Any], object],
     parent_object: Optional[dict[str, Any]] = None,
 ) -> dict[str, Any]:
@@ -58,7 +56,6 @@ def _ListFilesConfig_to_mldev(
 
 
 def _ListFilesParameters_to_mldev(
-    api_client: BaseApiClient,
     from_object: Union[dict[str, Any], object],
     parent_object: Optional[dict[str, Any]] = None,
 ) -> dict[str, Any]:
@@ -67,16 +64,13 @@ def _ListFilesParameters_to_mldev(
     setv(
         to_object,
         ['config'],
-        _ListFilesConfig_to_mldev(
-            api_client, getv(from_object, ['config']), to_object
-        ),
+        _ListFilesConfig_to_mldev(getv(from_object, ['config']), to_object),
     )
 
   return to_object
 
 
 def _FileStatus_to_mldev(
-    api_client: BaseApiClient,
     from_object: Union[dict[str, Any], object],
     parent_object: Optional[dict[str, Any]] = None,
 ) -> dict[str, Any]:
@@ -94,7 +88,6 @@ def _FileStatus_to_mldev(
 
 
 def _File_to_mldev(
-    api_client: BaseApiClient,
     from_object: Union[dict[str, Any], object],
     parent_object: Optional[dict[str, Any]] = None,
 ) -> dict[str, Any]:
@@ -142,16 +135,13 @@ def _File_to_mldev(
     setv(
         to_object,
         ['error'],
-        _FileStatus_to_mldev(
-            api_client, getv(from_object, ['error']), to_object
-        ),
+        _FileStatus_to_mldev(getv(from_object, ['error']), to_object),
     )
 
   return to_object
 
 
 def _CreateFileParameters_to_mldev(
-    api_client: BaseApiClient,
     from_object: Union[dict[str, Any], object],
     parent_object: Optional[dict[str, Any]] = None,
 ) -> dict[str, Any]:
@@ -160,7 +150,7 @@ def _CreateFileParameters_to_mldev(
     setv(
         to_object,
         ['file'],
-        _File_to_mldev(api_client, getv(from_object, ['file']), to_object),
+        _File_to_mldev(getv(from_object, ['file']), to_object),
     )
 
   if getv(from_object, ['config']) is not None:
@@ -170,16 +160,13 @@ def _CreateFileParameters_to_mldev(
 
 
 def _GetFileParameters_to_mldev(
-    api_client: BaseApiClient,
     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, ['name']) is not None:
     setv(
-        to_object,
-        ['_url', 'file'],
-        t.t_file_name(api_client, getv(from_object, ['name'])),
+        to_object, ['_url', 'file'], t.t_file_name(getv(from_object, ['name']))
     )
 
   if getv(from_object, ['config']) is not None:
@@ -189,16 +176,13 @@ def _GetFileParameters_to_mldev(
 
 
 def _DeleteFileParameters_to_mldev(
-    api_client: BaseApiClient,
     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, ['name']) is not None:
     setv(
-        to_object,
-        ['_url', 'file'],
-        t.t_file_name(api_client, getv(from_object, ['name'])),
+        to_object, ['_url', 'file'], t.t_file_name(getv(from_object, ['name']))
     )
 
   if getv(from_object, ['config']) is not None:
@@ -208,7 +192,6 @@ def _DeleteFileParameters_to_mldev(
 
 
 def _FileStatus_from_mldev(
-    api_client: BaseApiClient,
     from_object: Union[dict[str, Any], object],
     parent_object: Optional[dict[str, Any]] = None,
 ) -> dict[str, Any]:
@@ -226,7 +209,6 @@ def _FileStatus_from_mldev(
 
 
 def _File_from_mldev(
-    api_client: BaseApiClient,
     from_object: Union[dict[str, Any], object],
     parent_object: Optional[dict[str, Any]] = None,
 ) -> dict[str, Any]:
@@ -274,16 +256,13 @@ def _File_from_mldev(
     setv(
         to_object,
         ['error'],
-        _FileStatus_from_mldev(
-            api_client, getv(from_object, ['error']), to_object
-        ),
+        _FileStatus_from_mldev(getv(from_object, ['error']), to_object),
     )
 
   return to_object
 
 
 def _ListFilesResponse_from_mldev(
-    api_client: BaseApiClient,
     from_object: Union[dict[str, Any], object],
     parent_object: Optional[dict[str, Any]] = None,
 ) -> dict[str, Any]:
@@ -296,7 +275,7 @@ def _ListFilesResponse_from_mldev(
         to_object,
         ['files'],
         [
-            _File_from_mldev(api_client, item, to_object)
+            _File_from_mldev(item, to_object)
             for item in getv(from_object, ['files'])
         ],
     )
@@ -305,19 +284,15 @@ def _ListFilesResponse_from_mldev(
 
 
 def _CreateFileResponse_from_mldev(
-    api_client: BaseApiClient,
     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, ['httpHeaders']) is not None:
-    setv(to_object, ['http_headers'], getv(from_object, ['httpHeaders']))
 
   return to_object
 
 
 def _DeleteFileResponse_from_mldev(
-    api_client: BaseApiClient,
     from_object: Union[dict[str, Any], object],
     parent_object: Optional[dict[str, Any]] = None,
 ) -> dict[str, Any]:
@@ -358,9 +333,7 @@ class Files(_api_module.BaseModule):
           'This method is only supported in the Gemini Developer client.'
       )
     else:
-      request_dict = _ListFilesParameters_to_mldev(
-          self._api_client, parameter_model
-      )
+      request_dict = _ListFilesParameters_to_mldev(parameter_model)
       request_url_dict = request_dict.get('_url')
       if request_url_dict:
         path = 'files'.format_map(request_url_dict)
@@ -383,14 +356,12 @@ class Files(_api_module.BaseModule):
     request_dict = _common.convert_to_dict(request_dict)
     request_dict = _common.encode_unserializable_types(request_dict)
 
-    response_dict = self._api_client.request(
-        'get', path, request_dict, http_options
-    )
+    response = self._api_client.request('get', path, request_dict, http_options)
+
+    response_dict = '' if not response.body else json.loads(response.body)
 
     if not self._api_client.vertexai:
-      response_dict = _ListFilesResponse_from_mldev(
-          self._api_client, response_dict
-      )
+      response_dict = _ListFilesResponse_from_mldev(response_dict)
 
     return_value = types.ListFilesResponse._from_response(
         response=response_dict, kwargs=parameter_model.model_dump()
@@ -415,9 +386,7 @@ class Files(_api_module.BaseModule):
           'This method is only supported in the Gemini Developer client.'
       )
     else:
-      request_dict = _CreateFileParameters_to_mldev(
-          self._api_client, parameter_model
-      )
+      request_dict = _CreateFileParameters_to_mldev(parameter_model)
       request_url_dict = request_dict.get('_url')
       if request_url_dict:
         path = 'upload/v1beta/files'.format_map(request_url_dict)
@@ -440,14 +409,20 @@ class Files(_api_module.BaseModule):
     request_dict = _common.convert_to_dict(request_dict)
     request_dict = _common.encode_unserializable_types(request_dict)
 
-    response_dict = self._api_client.request(
+    response = self._api_client.request(
         'post', path, request_dict, http_options
     )
 
+    if config is not None and getattr(
+        config, 'should_return_http_response', None
+    ):
+      return_value = types.CreateFileResponse(sdk_http_response=response)
+      return return_value
+
+    response_dict = '' if not response.body else json.loads(response.body)
+
     if not self._api_client.vertexai:
-      response_dict = _CreateFileResponse_from_mldev(
-          self._api_client, response_dict
-      )
+      response_dict = _CreateFileResponse_from_mldev(response_dict)
 
     return_value = types.CreateFileResponse._from_response(
         response=response_dict, kwargs=parameter_model.model_dump()
@@ -486,9 +461,7 @@ class Files(_api_module.BaseModule):
           'This method is only supported in the Gemini Developer client.'
       )
     else:
-      request_dict = _GetFileParameters_to_mldev(
-          self._api_client, parameter_model
-      )
+      request_dict = _GetFileParameters_to_mldev(parameter_model)
       request_url_dict = request_dict.get('_url')
       if request_url_dict:
         path = 'files/{file}'.format_map(request_url_dict)
@@ -511,12 +484,12 @@ class Files(_api_module.BaseModule):
     request_dict = _common.convert_to_dict(request_dict)
     request_dict = _common.encode_unserializable_types(request_dict)
 
-    response_dict = self._api_client.request(
-        'get', path, request_dict, http_options
-    )
+    response = self._api_client.request('get', path, request_dict, http_options)
+
+    response_dict = '' if not response.body else json.loads(response.body)
 
     if not self._api_client.vertexai:
-      response_dict = _File_from_mldev(self._api_client, response_dict)
+      response_dict = _File_from_mldev(response_dict)
 
     return_value = types.File._from_response(
         response=response_dict, kwargs=parameter_model.model_dump()
@@ -554,9 +527,7 @@ class Files(_api_module.BaseModule):
           'This method is only supported in the Gemini Developer client.'
       )
     else:
-      request_dict = _DeleteFileParameters_to_mldev(
-          self._api_client, parameter_model
-      )
+      request_dict = _DeleteFileParameters_to_mldev(parameter_model)
       request_url_dict = request_dict.get('_url')
       if request_url_dict:
         path = 'files/{file}'.format_map(request_url_dict)
@@ -579,14 +550,14 @@ class Files(_api_module.BaseModule):
     request_dict = _common.convert_to_dict(request_dict)
     request_dict = _common.encode_unserializable_types(request_dict)
 
-    response_dict = self._api_client.request(
+    response = self._api_client.request(
         'delete', path, request_dict, http_options
     )
 
+    response_dict = '' if not response.body else json.loads(response.body)
+
     if not self._api_client.vertexai:
-      response_dict = _DeleteFileResponse_from_mldev(
-          self._api_client, response_dict
-      )
+      response_dict = _DeleteFileResponse_from_mldev(response_dict)
 
     return_value = types.DeleteFileResponse._from_response(
         response=response_dict, kwargs=parameter_model.model_dump()
@@ -597,7 +568,7 @@ class Files(_api_module.BaseModule):
   def upload(
       self,
       *,
-      file: Union[str, pathlib.Path, os.PathLike[str], io.IOBase],
+      file: Union[str, os.PathLike[str], io.IOBase],
       config: Optional[types.UploadFileConfigOrDict] = None,
   ) -> types.File:
     """Calls the API to upload a file using a supported file service.
@@ -679,18 +650,22 @@ class Files(_api_module.BaseModule):
           },
       )
     response = self._create(
-        file=file_obj, config=types.CreateFileConfig(http_options=http_options)
+        file=file_obj,
+        config=types.CreateFileConfig(
+            http_options=http_options, should_return_http_response=True
+        ),
     )
 
     if (
-        response.http_headers is None
-        or 'x-goog-upload-url' not in response.http_headers
+        response.sdk_http_response is None
+        or response.sdk_http_response.headers is None
+        or 'x-goog-upload-url' not in response.sdk_http_response.headers
     ):
       raise KeyError(
           'Failed to create file. Upload URL did not returned from the create'
           ' file request.'
       )
-    upload_url = response.http_headers['x-goog-upload-url']
+    upload_url = response.sdk_http_response.headers['x-goog-upload-url']
 
     if isinstance(file, io.IOBase):
       return_file = self._api_client.upload_file(
@@ -702,7 +677,7 @@ class Files(_api_module.BaseModule):
       )
 
     return types.File._from_response(
-        response=_File_from_mldev(self._api_client, return_file.json['file']),
+        response=_File_from_mldev(return_file.json['file']),
         kwargs=config_model.model_dump() if config else {},
     )
 
@@ -775,7 +750,7 @@ class Files(_api_module.BaseModule):
           'downloaded. You can tell which files are downloadable by checking '
           'the `source` or `download_uri` property.'
       )
-    name = t.t_file_name(self._api_client, file)
+    name = t.t_file_name(file)
 
     path = f'files/{name}:download'
 
@@ -830,9 +805,7 @@ class AsyncFiles(_api_module.BaseModule):
           'This method is only supported in the Gemini Developer client.'
       )
     else:
-      request_dict = _ListFilesParameters_to_mldev(
-          self._api_client, parameter_model
-      )
+      request_dict = _ListFilesParameters_to_mldev(parameter_model)
       request_url_dict = request_dict.get('_url')
       if request_url_dict:
         path = 'files'.format_map(request_url_dict)
@@ -855,14 +828,14 @@ class AsyncFiles(_api_module.BaseModule):
     request_dict = _common.convert_to_dict(request_dict)
     request_dict = _common.encode_unserializable_types(request_dict)
 
-    response_dict = await self._api_client.async_request(
+    response = await self._api_client.async_request(
         'get', path, request_dict, http_options
     )
 
+    response_dict = '' if not response.body else json.loads(response.body)
+
     if not self._api_client.vertexai:
-      response_dict = _ListFilesResponse_from_mldev(
-          self._api_client, response_dict
-      )
+      response_dict = _ListFilesResponse_from_mldev(response_dict)
 
     return_value = types.ListFilesResponse._from_response(
         response=response_dict, kwargs=parameter_model.model_dump()
@@ -887,9 +860,7 @@ class AsyncFiles(_api_module.BaseModule):
           'This method is only supported in the Gemini Developer client.'
       )
     else:
-      request_dict = _CreateFileParameters_to_mldev(
-          self._api_client, parameter_model
-      )
+      request_dict = _CreateFileParameters_to_mldev(parameter_model)
       request_url_dict = request_dict.get('_url')
       if request_url_dict:
         path = 'upload/v1beta/files'.format_map(request_url_dict)
@@ -912,14 +883,20 @@ class AsyncFiles(_api_module.BaseModule):
     request_dict = _common.convert_to_dict(request_dict)
     request_dict = _common.encode_unserializable_types(request_dict)
 
-    response_dict = await self._api_client.async_request(
+    response = await self._api_client.async_request(
         'post', path, request_dict, http_options
     )
 
+    if config is not None and getattr(
+        config, 'should_return_http_response', None
+    ):
+      return_value = types.CreateFileResponse(sdk_http_response=response)
+      return return_value
+
+    response_dict = '' if not response.body else json.loads(response.body)
+
     if not self._api_client.vertexai:
-      response_dict = _CreateFileResponse_from_mldev(
-          self._api_client, response_dict
-      )
+      response_dict = _CreateFileResponse_from_mldev(response_dict)
 
     return_value = types.CreateFileResponse._from_response(
         response=response_dict, kwargs=parameter_model.model_dump()
@@ -958,9 +935,7 @@ class AsyncFiles(_api_module.BaseModule):
           'This method is only supported in the Gemini Developer client.'
       )
     else:
-      request_dict = _GetFileParameters_to_mldev(
-          self._api_client, parameter_model
-      )
+      request_dict = _GetFileParameters_to_mldev(parameter_model)
       request_url_dict = request_dict.get('_url')
       if request_url_dict:
         path = 'files/{file}'.format_map(request_url_dict)
@@ -983,12 +958,14 @@ class AsyncFiles(_api_module.BaseModule):
     request_dict = _common.convert_to_dict(request_dict)
     request_dict = _common.encode_unserializable_types(request_dict)
 
-    response_dict = await self._api_client.async_request(
+    response = await self._api_client.async_request(
         'get', path, request_dict, http_options
     )
 
+    response_dict = '' if not response.body else json.loads(response.body)
+
     if not self._api_client.vertexai:
-      response_dict = _File_from_mldev(self._api_client, response_dict)
+      response_dict = _File_from_mldev(response_dict)
 
     return_value = types.File._from_response(
         response=response_dict, kwargs=parameter_model.model_dump()
@@ -1026,9 +1003,7 @@ class AsyncFiles(_api_module.BaseModule):
           'This method is only supported in the Gemini Developer client.'
       )
     else:
-      request_dict = _DeleteFileParameters_to_mldev(
-          self._api_client, parameter_model
-      )
+      request_dict = _DeleteFileParameters_to_mldev(parameter_model)
       request_url_dict = request_dict.get('_url')
       if request_url_dict:
         path = 'files/{file}'.format_map(request_url_dict)
@@ -1051,14 +1026,14 @@ class AsyncFiles(_api_module.BaseModule):
     request_dict = _common.convert_to_dict(request_dict)
     request_dict = _common.encode_unserializable_types(request_dict)
 
-    response_dict = await self._api_client.async_request(
+    response = await self._api_client.async_request(
         'delete', path, request_dict, http_options
     )
 
+    response_dict = '' if not response.body else json.loads(response.body)
+
     if not self._api_client.vertexai:
-      response_dict = _DeleteFileResponse_from_mldev(
-          self._api_client, response_dict
-      )
+      response_dict = _DeleteFileResponse_from_mldev(response_dict)
 
     return_value = types.DeleteFileResponse._from_response(
         response=response_dict, kwargs=parameter_model.model_dump()
@@ -1069,7 +1044,7 @@ class AsyncFiles(_api_module.BaseModule):
   async def upload(
       self,
       *,
-      file: Union[str, pathlib.Path, os.PathLike[str], io.IOBase],
+      file: Union[str, os.PathLike[str], io.IOBase],
       config: Optional[types.UploadFileConfigOrDict] = None,
   ) -> types.File:
     """Calls the API to upload a file asynchronously using a supported file service.
@@ -1151,20 +1126,27 @@ class AsyncFiles(_api_module.BaseModule):
           },
       )
     response = await self._create(
-        file=file_obj, config=types.CreateFileConfig(http_options=http_options)
+        file=file_obj,
+        config=types.CreateFileConfig(
+            http_options=http_options, should_return_http_response=True
+        ),
     )
-    if response.http_headers is None or (
-        'x-goog-upload-url' not in response.http_headers
-        and 'X-Goog-Upload-URL' not in response.http_headers
+    if (
+        response.sdk_http_response is None
+        or response.sdk_http_response.headers is None
+        or (
+            'x-goog-upload-url' not in response.sdk_http_response.headers
+            and 'X-Goog-Upload-URL' not in response.sdk_http_response.headers
+        )
     ):
       raise KeyError(
           'Failed to create file. Upload URL did not returned from the create'
           ' file request.'
       )
-    elif 'x-goog-upload-url' in response.http_headers:
-      upload_url = response.http_headers['x-goog-upload-url']
+    elif 'x-goog-upload-url' in response.sdk_http_response.headers:
+      upload_url = response.sdk_http_response.headers['x-goog-upload-url']
     else:
-      upload_url = response.http_headers['X-Goog-Upload-URL']
+      upload_url = response.sdk_http_response.headers['X-Goog-Upload-URL']
 
     if isinstance(file, io.IOBase):
       return_file = await self._api_client.async_upload_file(
@@ -1176,7 +1158,7 @@ class AsyncFiles(_api_module.BaseModule):
       )
 
     return types.File._from_response(
-        response=_File_from_mldev(self._api_client, return_file.json['file']),
+        response=_File_from_mldev(return_file.json['file']),
         kwargs=config_model.model_dump() if config else {},
     )
 
@@ -1237,7 +1219,7 @@ class AsyncFiles(_api_module.BaseModule):
       else:
         config_model = config
 
-    name = t.t_file_name(self._api_client, file)
+    name = t.t_file_name(file)
 
     path = f'files/{name}:download'
 

google/genai/live.py

@@ -79,6 +79,7 @@ _FUNCTION_RESPONSE_REQUIRES_ID = (
     ' response of a ToolCall.FunctionalCalls in Google AI.'
 )
 
+
 class AsyncSession:
   """[Preview] AsyncSession."""
 
@@ -215,11 +216,11 @@ class AsyncSession:
 
     if self._api_client.vertexai:
       client_content_dict = live_converters._LiveClientContent_to_vertex(
-          api_client=self._api_client, from_object=client_content
+          from_object=client_content
       )
     else:
       client_content_dict = live_converters._LiveClientContent_to_mldev(
-          api_client=self._api_client, from_object=client_content
+          from_object=client_content
       )
 
     await self._ws.send(json.dumps({'client_content': client_content_dict}))
@@ -313,13 +314,13 @@ class AsyncSession:
     if self._api_client.vertexai:
       realtime_input_dict = (
           live_converters._LiveSendRealtimeInputParameters_to_vertex(
-              api_client=self._api_client, from_object=realtime_input
+              from_object=realtime_input
           )
       )
     else:
       realtime_input_dict = (
           live_converters._LiveSendRealtimeInputParameters_to_mldev(
-              api_client=self._api_client, from_object=realtime_input
+              from_object=realtime_input
           )
       )
     realtime_input_dict = _common.convert_to_dict(realtime_input_dict)
@@ -399,11 +400,11 @@ class AsyncSession:
     tool_response = t.t_tool_response(function_responses)
     if self._api_client.vertexai:
       tool_response_dict = live_converters._LiveClientToolResponse_to_vertex(
-          api_client=self._api_client, from_object=tool_response
+          from_object=tool_response
       )
     else:
       tool_response_dict = live_converters._LiveClientToolResponse_to_mldev(
-          api_client=self._api_client, from_object=tool_response
+          from_object=tool_response
       )
       for response in tool_response_dict.get('functionResponses', []):
         if response.get('id') is None:
@@ -527,13 +528,9 @@ class AsyncSession:
       response = {}
 
     if self._api_client.vertexai:
-      response_dict = live_converters._LiveServerMessage_from_vertex(
-          self._api_client, response
-      )
+      response_dict = live_converters._LiveServerMessage_from_vertex(response)
     else:
-      response_dict = live_converters._LiveServerMessage_from_mldev(
-          self._api_client, response
-      )
+      response_dict = live_converters._LiveServerMessage_from_mldev(response)
 
     return types.LiveServerMessage._from_response(
         response=response_dict, kwargs=parameter_model.model_dump()
@@ -647,13 +644,13 @@ class AsyncSession:
           content_input_parts.append(item)
       if self._api_client.vertexai:
         contents = [
-            _Content_to_vertex(self._api_client, item, to_object)
-            for item in t.t_contents(self._api_client, content_input_parts)
+            _Content_to_vertex(item, to_object)
+            for item in t.t_contents(content_input_parts)
         ]
       else:
         contents = [
-            _Content_to_mldev(self._api_client, item, to_object)
-            for item in t.t_contents(self._api_client, content_input_parts)
+            _Content_to_mldev(item, to_object)
+            for item in t.t_contents(content_input_parts)
         ]
 
       content_dict_list: list[types.ContentDict] = []
@@ -1063,7 +1060,7 @@ async def _t_live_connect_config(
   elif isinstance(config, dict):
     if getv(config, ['system_instruction']) is not None:
       converted_system_instruction = t.t_content(
-          api_client, getv(config, ['system_instruction'])
+          getv(config, ['system_instruction'])
       )
     else:
       converted_system_instruction = None
@@ -1073,9 +1070,7 @@ async def _t_live_connect_config(
     if config.system_instruction is None:
       system_instruction = None
     else:
-      system_instruction = t.t_content(
-          api_client, getv(config, ['system_instruction'])
-      )
+      system_instruction = t.t_content(getv(config, ['system_instruction']))
     parameter_model = config
     parameter_model.system_instruction = system_instruction