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

google/genai/tunings.py

@@ -15,6 +15,7 @@
 
 # Code generated by the Google Gen AI SDK generator DO NOT EDIT.
 
+import json
 import logging
 from typing import Any, Optional, Union
 from urllib.parse import urlencode
@@ -23,7 +24,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
@@ -32,7 +32,6 @@ logger = logging.getLogger('google_genai.tunings')
 
 
 def _GetTuningJobParameters_to_mldev(
-    api_client: BaseApiClient,
     from_object: Union[dict[str, Any], object],
     parent_object: Optional[dict[str, Any]] = None,
 ) -> dict[str, Any]:
@@ -47,7 +46,6 @@ def _GetTuningJobParameters_to_mldev(
 
 
 def _ListTuningJobsConfig_to_mldev(
-    api_client: BaseApiClient,
     from_object: Union[dict[str, Any], object],
     parent_object: Optional[dict[str, Any]] = None,
 ) -> dict[str, Any]:
@@ -72,7 +70,6 @@ def _ListTuningJobsConfig_to_mldev(
 
 
 def _ListTuningJobsParameters_to_mldev(
-    api_client: BaseApiClient,
     from_object: Union[dict[str, Any], object],
     parent_object: Optional[dict[str, Any]] = None,
 ) -> dict[str, Any]:
@@ -82,7 +79,7 @@ def _ListTuningJobsParameters_to_mldev(
         to_object,
         ['config'],
         _ListTuningJobsConfig_to_mldev(
-            api_client, getv(from_object, ['config']), to_object
+            getv(from_object, ['config']), to_object
         ),
     )
 
@@ -90,7 +87,6 @@ def _ListTuningJobsParameters_to_mldev(
 
 
 def _TuningExample_to_mldev(
-    api_client: BaseApiClient,
     from_object: Union[dict[str, Any], object],
     parent_object: Optional[dict[str, Any]] = None,
 ) -> dict[str, Any]:
@@ -105,7 +101,6 @@ def _TuningExample_to_mldev(
 
 
 def _TuningDataset_to_mldev(
-    api_client: BaseApiClient,
     from_object: Union[dict[str, Any], object],
     parent_object: Optional[dict[str, Any]] = None,
 ) -> dict[str, Any]:
@@ -118,7 +113,7 @@ def _TuningDataset_to_mldev(
         to_object,
         ['examples', 'examples'],
         [
-            _TuningExample_to_mldev(api_client, item, to_object)
+            _TuningExample_to_mldev(item, to_object)
             for item in getv(from_object, ['examples'])
         ],
     )
@@ -127,7 +122,6 @@ def _TuningDataset_to_mldev(
 
 
 def _TuningValidationDataset_to_mldev(
-    api_client: BaseApiClient,
     from_object: Union[dict[str, Any], object],
     parent_object: Optional[dict[str, Any]] = None,
 ) -> dict[str, Any]:
@@ -139,7 +133,6 @@ def _TuningValidationDataset_to_mldev(
 
 
 def _CreateTuningJobConfig_to_mldev(
-    api_client: BaseApiClient,
     from_object: Union[dict[str, Any], object],
     parent_object: Optional[dict[str, Any]] = None,
 ) -> dict[str, Any]:
@@ -200,7 +193,6 @@ def _CreateTuningJobConfig_to_mldev(
 
 
 def _CreateTuningJobParameters_to_mldev(
-    api_client: BaseApiClient,
     from_object: Union[dict[str, Any], object],
     parent_object: Optional[dict[str, Any]] = None,
 ) -> dict[str, Any]:
@@ -213,7 +205,7 @@ def _CreateTuningJobParameters_to_mldev(
         to_object,
         ['tuningTask', 'trainingData'],
         _TuningDataset_to_mldev(
-            api_client, getv(from_object, ['training_dataset']), to_object
+            getv(from_object, ['training_dataset']), to_object
         ),
     )
 
@@ -222,7 +214,7 @@ def _CreateTuningJobParameters_to_mldev(
         to_object,
         ['config'],
         _CreateTuningJobConfig_to_mldev(
-            api_client, getv(from_object, ['config']), to_object
+            getv(from_object, ['config']), to_object
         ),
     )
 
@@ -230,7 +222,6 @@ def _CreateTuningJobParameters_to_mldev(
 
 
 def _GetTuningJobParameters_to_vertex(
-    api_client: BaseApiClient,
     from_object: Union[dict[str, Any], object],
     parent_object: Optional[dict[str, Any]] = None,
 ) -> dict[str, Any]:
@@ -245,7 +236,6 @@ def _GetTuningJobParameters_to_vertex(
 
 
 def _ListTuningJobsConfig_to_vertex(
-    api_client: BaseApiClient,
     from_object: Union[dict[str, Any], object],
     parent_object: Optional[dict[str, Any]] = None,
 ) -> dict[str, Any]:
@@ -270,7 +260,6 @@ def _ListTuningJobsConfig_to_vertex(
 
 
 def _ListTuningJobsParameters_to_vertex(
-    api_client: BaseApiClient,
     from_object: Union[dict[str, Any], object],
     parent_object: Optional[dict[str, Any]] = None,
 ) -> dict[str, Any]:
@@ -280,7 +269,7 @@ def _ListTuningJobsParameters_to_vertex(
         to_object,
         ['config'],
         _ListTuningJobsConfig_to_vertex(
-            api_client, getv(from_object, ['config']), to_object
+            getv(from_object, ['config']), to_object
         ),
     )
 
@@ -288,7 +277,6 @@ def _ListTuningJobsParameters_to_vertex(
 
 
 def _TuningExample_to_vertex(
-    api_client: BaseApiClient,
     from_object: Union[dict[str, Any], object],
     parent_object: Optional[dict[str, Any]] = None,
 ) -> dict[str, Any]:
@@ -303,7 +291,6 @@ def _TuningExample_to_vertex(
 
 
 def _TuningDataset_to_vertex(
-    api_client: BaseApiClient,
     from_object: Union[dict[str, Any], object],
     parent_object: Optional[dict[str, Any]] = None,
 ) -> dict[str, Any]:
@@ -322,7 +309,6 @@ def _TuningDataset_to_vertex(
 
 
 def _TuningValidationDataset_to_vertex(
-    api_client: BaseApiClient,
     from_object: Union[dict[str, Any], object],
     parent_object: Optional[dict[str, Any]] = None,
 ) -> dict[str, Any]:
@@ -334,7 +320,6 @@ def _TuningValidationDataset_to_vertex(
 
 
 def _CreateTuningJobConfig_to_vertex(
-    api_client: BaseApiClient,
     from_object: Union[dict[str, Any], object],
     parent_object: Optional[dict[str, Any]] = None,
 ) -> dict[str, Any]:
@@ -345,7 +330,7 @@ def _CreateTuningJobConfig_to_vertex(
         parent_object,
         ['supervisedTuningSpec'],
         _TuningValidationDataset_to_vertex(
-            api_client, getv(from_object, ['validation_dataset']), to_object
+            getv(from_object, ['validation_dataset']), to_object
         ),
     )
 
@@ -397,7 +382,6 @@ def _CreateTuningJobConfig_to_vertex(
 
 
 def _CreateTuningJobParameters_to_vertex(
-    api_client: BaseApiClient,
     from_object: Union[dict[str, Any], object],
     parent_object: Optional[dict[str, Any]] = None,
 ) -> dict[str, Any]:
@@ -410,7 +394,7 @@ def _CreateTuningJobParameters_to_vertex(
         to_object,
         ['supervisedTuningSpec', 'trainingDatasetUri'],
         _TuningDataset_to_vertex(
-            api_client, getv(from_object, ['training_dataset']), to_object
+            getv(from_object, ['training_dataset']), to_object
         ),
     )
 
@@ -419,7 +403,7 @@ def _CreateTuningJobParameters_to_vertex(
         to_object,
         ['config'],
         _CreateTuningJobConfig_to_vertex(
-            api_client, getv(from_object, ['config']), to_object
+            getv(from_object, ['config']), to_object
         ),
     )
 
@@ -427,7 +411,6 @@ def _CreateTuningJobParameters_to_vertex(
 
 
 def _TunedModelCheckpoint_from_mldev(
-    api_client: BaseApiClient,
     from_object: Union[dict[str, Any], object],
     parent_object: Optional[dict[str, Any]] = None,
 ) -> dict[str, Any]:
@@ -437,7 +420,6 @@ def _TunedModelCheckpoint_from_mldev(
 
 
 def _TunedModel_from_mldev(
-    api_client: BaseApiClient,
     from_object: Union[dict[str, Any], object],
     parent_object: Optional[dict[str, Any]] = None,
 ) -> dict[str, Any]:
@@ -452,7 +434,6 @@ def _TunedModel_from_mldev(
 
 
 def _TuningJob_from_mldev(
-    api_client: BaseApiClient,
     from_object: Union[dict[str, Any], object],
     parent_object: Optional[dict[str, Any]] = None,
 ) -> dict[str, Any]:
@@ -464,7 +445,7 @@ def _TuningJob_from_mldev(
     setv(
         to_object,
         ['state'],
-        t.t_tuning_job_status(api_client, getv(from_object, ['state'])),
+        t.t_tuning_job_status(getv(from_object, ['state'])),
     )
 
   if getv(from_object, ['createTime']) is not None:
@@ -497,9 +478,7 @@ def _TuningJob_from_mldev(
     setv(
         to_object,
         ['tuned_model'],
-        _TunedModel_from_mldev(
-            api_client, getv(from_object, ['_self']), to_object
-        ),
+        _TunedModel_from_mldev(getv(from_object, ['_self']), to_object),
     )
 
   if getv(from_object, ['distillationSpec']) is not None:
@@ -532,7 +511,6 @@ def _TuningJob_from_mldev(
 
 
 def _ListTuningJobsResponse_from_mldev(
-    api_client: BaseApiClient,
     from_object: Union[dict[str, Any], object],
     parent_object: Optional[dict[str, Any]] = None,
 ) -> dict[str, Any]:
@@ -545,7 +523,7 @@ def _ListTuningJobsResponse_from_mldev(
         to_object,
         ['tuning_jobs'],
         [
-            _TuningJob_from_mldev(api_client, item, to_object)
+            _TuningJob_from_mldev(item, to_object)
             for item in getv(from_object, ['tunedModels'])
         ],
     )
@@ -554,7 +532,6 @@ def _ListTuningJobsResponse_from_mldev(
 
 
 def _Operation_from_mldev(
-    api_client: BaseApiClient,
     from_object: Union[dict[str, Any], object],
     parent_object: Optional[dict[str, Any]] = None,
 ) -> dict[str, Any]:
@@ -575,7 +552,6 @@ def _Operation_from_mldev(
 
 
 def _TunedModelCheckpoint_from_vertex(
-    api_client: BaseApiClient,
     from_object: Union[dict[str, Any], object],
     parent_object: Optional[dict[str, Any]] = None,
 ) -> dict[str, Any]:
@@ -596,7 +572,6 @@ def _TunedModelCheckpoint_from_vertex(
 
 
 def _TunedModel_from_vertex(
-    api_client: BaseApiClient,
     from_object: Union[dict[str, Any], object],
     parent_object: Optional[dict[str, Any]] = None,
 ) -> dict[str, Any]:
@@ -612,7 +587,7 @@ def _TunedModel_from_vertex(
         to_object,
         ['checkpoints'],
         [
-            _TunedModelCheckpoint_from_vertex(api_client, item, to_object)
+            _TunedModelCheckpoint_from_vertex(item, to_object)
             for item in getv(from_object, ['checkpoints'])
         ],
     )
@@ -621,7 +596,6 @@ def _TunedModel_from_vertex(
 
 
 def _TuningJob_from_vertex(
-    api_client: BaseApiClient,
     from_object: Union[dict[str, Any], object],
     parent_object: Optional[dict[str, Any]] = None,
 ) -> dict[str, Any]:
@@ -633,7 +607,7 @@ def _TuningJob_from_vertex(
     setv(
         to_object,
         ['state'],
-        t.t_tuning_job_status(api_client, getv(from_object, ['state'])),
+        t.t_tuning_job_status(getv(from_object, ['state'])),
     )
 
   if getv(from_object, ['createTime']) is not None:
@@ -661,9 +635,7 @@ def _TuningJob_from_vertex(
     setv(
         to_object,
         ['tuned_model'],
-        _TunedModel_from_vertex(
-            api_client, getv(from_object, ['tunedModel']), to_object
-        ),
+        _TunedModel_from_vertex(getv(from_object, ['tunedModel']), to_object),
     )
 
   if getv(from_object, ['supervisedTuningSpec']) is not None:
@@ -718,7 +690,6 @@ def _TuningJob_from_vertex(
 
 
 def _ListTuningJobsResponse_from_vertex(
-    api_client: BaseApiClient,
     from_object: Union[dict[str, Any], object],
     parent_object: Optional[dict[str, Any]] = None,
 ) -> dict[str, Any]:
@@ -731,7 +702,7 @@ def _ListTuningJobsResponse_from_vertex(
         to_object,
         ['tuning_jobs'],
         [
-            _TuningJob_from_vertex(api_client, item, to_object)
+            _TuningJob_from_vertex(item, to_object)
             for item in getv(from_object, ['tuningJobs'])
         ],
     )
@@ -764,18 +735,14 @@ class Tunings(_api_module.BaseModule):
     request_url_dict: Optional[dict[str, str]]
 
     if self._api_client.vertexai:
-      request_dict = _GetTuningJobParameters_to_vertex(
-          self._api_client, parameter_model
-      )
+      request_dict = _GetTuningJobParameters_to_vertex(parameter_model)
       request_url_dict = request_dict.get('_url')
       if request_url_dict:
         path = '{name}'.format_map(request_url_dict)
       else:
         path = '{name}'
     else:
-      request_dict = _GetTuningJobParameters_to_mldev(
-          self._api_client, parameter_model
-      )
+      request_dict = _GetTuningJobParameters_to_mldev(parameter_model)
       request_url_dict = request_dict.get('_url')
       if request_url_dict:
         path = '{name}'.format_map(request_url_dict)
@@ -797,15 +764,15 @@ class Tunings(_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 self._api_client.vertexai:
-      response_dict = _TuningJob_from_vertex(self._api_client, response_dict)
+      response_dict = _TuningJob_from_vertex(response_dict)
 
     else:
-      response_dict = _TuningJob_from_mldev(self._api_client, response_dict)
+      response_dict = _TuningJob_from_mldev(response_dict)
 
     return_value = types.TuningJob._from_response(
         response=response_dict, kwargs=parameter_model.model_dump()
@@ -832,18 +799,14 @@ class Tunings(_api_module.BaseModule):
     request_url_dict: Optional[dict[str, str]]
 
     if self._api_client.vertexai:
-      request_dict = _ListTuningJobsParameters_to_vertex(
-          self._api_client, parameter_model
-      )
+      request_dict = _ListTuningJobsParameters_to_vertex(parameter_model)
       request_url_dict = request_dict.get('_url')
       if request_url_dict:
         path = 'tuningJobs'.format_map(request_url_dict)
       else:
         path = 'tuningJobs'
     else:
-      request_dict = _ListTuningJobsParameters_to_mldev(
-          self._api_client, parameter_model
-      )
+      request_dict = _ListTuningJobsParameters_to_mldev(parameter_model)
       request_url_dict = request_dict.get('_url')
       if request_url_dict:
         path = 'tunedModels'.format_map(request_url_dict)
@@ -865,19 +828,15 @@ class Tunings(_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 self._api_client.vertexai:
-      response_dict = _ListTuningJobsResponse_from_vertex(
-          self._api_client, response_dict
-      )
+      response_dict = _ListTuningJobsResponse_from_vertex(response_dict)
 
     else:
-      response_dict = _ListTuningJobsResponse_from_mldev(
-          self._api_client, response_dict
-      )
+      response_dict = _ListTuningJobsResponse_from_mldev(response_dict)
 
     return_value = types.ListTuningJobsResponse._from_response(
         response=response_dict, kwargs=parameter_model.model_dump()
@@ -913,9 +872,7 @@ class Tunings(_api_module.BaseModule):
     if not self._api_client.vertexai:
       raise ValueError('This method is only supported in the Vertex AI client.')
     else:
-      request_dict = _CreateTuningJobParameters_to_vertex(
-          self._api_client, parameter_model
-      )
+      request_dict = _CreateTuningJobParameters_to_vertex(parameter_model)
       request_url_dict = request_dict.get('_url')
       if request_url_dict:
         path = 'tuningJobs'.format_map(request_url_dict)
@@ -938,12 +895,14 @@ class Tunings(_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
     )
 
+    response_dict = '' if not response.body else json.loads(response.body)
+
     if self._api_client.vertexai:
-      response_dict = _TuningJob_from_vertex(self._api_client, response_dict)
+      response_dict = _TuningJob_from_vertex(response_dict)
 
     return_value = types.TuningJob._from_response(
         response=response_dict, kwargs=parameter_model.model_dump()
@@ -981,9 +940,7 @@ class Tunings(_api_module.BaseModule):
           'This method is only supported in the Gemini Developer client.'
       )
     else:
-      request_dict = _CreateTuningJobParameters_to_mldev(
-          self._api_client, parameter_model
-      )
+      request_dict = _CreateTuningJobParameters_to_mldev(parameter_model)
       request_url_dict = request_dict.get('_url')
       if request_url_dict:
         path = 'tunedModels'.format_map(request_url_dict)
@@ -1006,12 +963,14 @@ class Tunings(_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
     )
 
+    response_dict = '' if not response.body else json.loads(response.body)
+
     if not self._api_client.vertexai:
-      response_dict = _Operation_from_mldev(self._api_client, response_dict)
+      response_dict = _Operation_from_mldev(response_dict)
 
     return_value = types.Operation._from_response(
         response=response_dict, kwargs=parameter_model.model_dump()
@@ -1112,18 +1071,14 @@ class AsyncTunings(_api_module.BaseModule):
     request_url_dict: Optional[dict[str, str]]
 
     if self._api_client.vertexai:
-      request_dict = _GetTuningJobParameters_to_vertex(
-          self._api_client, parameter_model
-      )
+      request_dict = _GetTuningJobParameters_to_vertex(parameter_model)
       request_url_dict = request_dict.get('_url')
       if request_url_dict:
         path = '{name}'.format_map(request_url_dict)
       else:
         path = '{name}'
     else:
-      request_dict = _GetTuningJobParameters_to_mldev(
-          self._api_client, parameter_model
-      )
+      request_dict = _GetTuningJobParameters_to_mldev(parameter_model)
       request_url_dict = request_dict.get('_url')
       if request_url_dict:
         path = '{name}'.format_map(request_url_dict)
@@ -1145,15 +1100,17 @@ class AsyncTunings(_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 self._api_client.vertexai:
-      response_dict = _TuningJob_from_vertex(self._api_client, response_dict)
+      response_dict = _TuningJob_from_vertex(response_dict)
 
     else:
-      response_dict = _TuningJob_from_mldev(self._api_client, response_dict)
+      response_dict = _TuningJob_from_mldev(response_dict)
 
     return_value = types.TuningJob._from_response(
         response=response_dict, kwargs=parameter_model.model_dump()
@@ -1180,18 +1137,14 @@ class AsyncTunings(_api_module.BaseModule):
     request_url_dict: Optional[dict[str, str]]
 
     if self._api_client.vertexai:
-      request_dict = _ListTuningJobsParameters_to_vertex(
-          self._api_client, parameter_model
-      )
+      request_dict = _ListTuningJobsParameters_to_vertex(parameter_model)
       request_url_dict = request_dict.get('_url')
       if request_url_dict:
         path = 'tuningJobs'.format_map(request_url_dict)
       else:
         path = 'tuningJobs'
     else:
-      request_dict = _ListTuningJobsParameters_to_mldev(
-          self._api_client, parameter_model
-      )
+      request_dict = _ListTuningJobsParameters_to_mldev(parameter_model)
       request_url_dict = request_dict.get('_url')
       if request_url_dict:
         path = 'tunedModels'.format_map(request_url_dict)
@@ -1213,19 +1166,17 @@ class AsyncTunings(_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 self._api_client.vertexai:
-      response_dict = _ListTuningJobsResponse_from_vertex(
-          self._api_client, response_dict
-      )
+      response_dict = _ListTuningJobsResponse_from_vertex(response_dict)
 
     else:
-      response_dict = _ListTuningJobsResponse_from_mldev(
-          self._api_client, response_dict
-      )
+      response_dict = _ListTuningJobsResponse_from_mldev(response_dict)
 
     return_value = types.ListTuningJobsResponse._from_response(
         response=response_dict, kwargs=parameter_model.model_dump()
@@ -1261,9 +1212,7 @@ class AsyncTunings(_api_module.BaseModule):
     if not self._api_client.vertexai:
       raise ValueError('This method is only supported in the Vertex AI client.')
     else:
-      request_dict = _CreateTuningJobParameters_to_vertex(
-          self._api_client, parameter_model
-      )
+      request_dict = _CreateTuningJobParameters_to_vertex(parameter_model)
       request_url_dict = request_dict.get('_url')
       if request_url_dict:
         path = 'tuningJobs'.format_map(request_url_dict)
@@ -1286,12 +1235,14 @@ class AsyncTunings(_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
     )
 
+    response_dict = '' if not response.body else json.loads(response.body)
+
     if self._api_client.vertexai:
-      response_dict = _TuningJob_from_vertex(self._api_client, response_dict)
+      response_dict = _TuningJob_from_vertex(response_dict)
 
     return_value = types.TuningJob._from_response(
         response=response_dict, kwargs=parameter_model.model_dump()
@@ -1329,9 +1280,7 @@ class AsyncTunings(_api_module.BaseModule):
           'This method is only supported in the Gemini Developer client.'
       )
     else:
-      request_dict = _CreateTuningJobParameters_to_mldev(
-          self._api_client, parameter_model
-      )
+      request_dict = _CreateTuningJobParameters_to_mldev(parameter_model)
       request_url_dict = request_dict.get('_url')
       if request_url_dict:
         path = 'tunedModels'.format_map(request_url_dict)
@@ -1354,12 +1303,14 @@ class AsyncTunings(_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
     )
 
+    response_dict = '' if not response.body else json.loads(response.body)
+
     if not self._api_client.vertexai:
-      response_dict = _Operation_from_mldev(self._api_client, response_dict)
+      response_dict = _Operation_from_mldev(response_dict)
 
     return_value = types.Operation._from_response(
         response=response_dict, kwargs=parameter_model.model_dump()