google-genai 1.2.0__py3-none-any.whl → 1.4.0__py3-none-any.whl

google/genai/batches.py

@@ -15,25 +15,28 @@
 
 # Code generated by the Google Gen AI SDK generator DO NOT EDIT.
 
-from typing import Optional, Union
+import logging
+from typing import Any, Optional, Union
 from urllib.parse import urlencode
 from . import _api_module
 from . import _common
 from . import _extra_utils
 from . import _transformers as t
 from . import types
-from ._api_client import ApiClient
+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
 
+logger = logging.getLogger('google_genai.batches')
+
 
 def _BatchJobSource_to_mldev(
-    api_client: ApiClient,
+    api_client: BaseApiClient,
     from_object: Union[dict, object],
-    parent_object: dict = None,
+    parent_object: Optional[dict] = None,
 ) -> dict:
-  to_object = {}
+  to_object: dict[str, Any] = {}
   if getv(from_object, ['format']) is not None:
     raise ValueError('format parameter is not supported in Gemini API.')
 
@@ -47,11 +50,11 @@ def _BatchJobSource_to_mldev(
 
 
 def _BatchJobSource_to_vertex(
-    api_client: ApiClient,
+    api_client: BaseApiClient,
     from_object: Union[dict, object],
-    parent_object: dict = None,
+    parent_object: Optional[dict] = None,
 ) -> dict:
-  to_object = {}
+  to_object: dict[str, Any] = {}
   if getv(from_object, ['format']) is not None:
     setv(to_object, ['instancesFormat'], getv(from_object, ['format']))
 
@@ -69,11 +72,11 @@ def _BatchJobSource_to_vertex(
 
 
 def _BatchJobDestination_to_mldev(
-    api_client: ApiClient,
+    api_client: BaseApiClient,
     from_object: Union[dict, object],
-    parent_object: dict = None,
+    parent_object: Optional[dict] = None,
 ) -> dict:
-  to_object = {}
+  to_object: dict[str, Any] = {}
   if getv(from_object, ['format']) is not None:
     raise ValueError('format parameter is not supported in Gemini API.')
 
@@ -87,11 +90,11 @@ def _BatchJobDestination_to_mldev(
 
 
 def _BatchJobDestination_to_vertex(
-    api_client: ApiClient,
+    api_client: BaseApiClient,
     from_object: Union[dict, object],
-    parent_object: dict = None,
+    parent_object: Optional[dict] = None,
 ) -> dict:
-  to_object = {}
+  to_object: dict[str, Any] = {}
   if getv(from_object, ['format']) is not None:
     setv(to_object, ['predictionsFormat'], getv(from_object, ['format']))
 
@@ -113,11 +116,11 @@ def _BatchJobDestination_to_vertex(
 
 
 def _CreateBatchJobConfig_to_mldev(
-    api_client: ApiClient,
+    api_client: BaseApiClient,
     from_object: Union[dict, object],
-    parent_object: dict = None,
+    parent_object: Optional[dict] = None,
 ) -> dict:
-  to_object = {}
+  to_object: dict[str, Any] = {}
 
   if getv(from_object, ['display_name']) is not None:
     setv(parent_object, ['displayName'], getv(from_object, ['display_name']))
@@ -129,11 +132,11 @@ def _CreateBatchJobConfig_to_mldev(
 
 
 def _CreateBatchJobConfig_to_vertex(
-    api_client: ApiClient,
+    api_client: BaseApiClient,
     from_object: Union[dict, object],
-    parent_object: dict = None,
+    parent_object: Optional[dict] = None,
 ) -> dict:
-  to_object = {}
+  to_object: dict[str, Any] = {}
 
   if getv(from_object, ['display_name']) is not None:
     setv(parent_object, ['displayName'], getv(from_object, ['display_name']))
@@ -153,11 +156,11 @@ def _CreateBatchJobConfig_to_vertex(
 
 
 def _CreateBatchJobParameters_to_mldev(
-    api_client: ApiClient,
+    api_client: BaseApiClient,
     from_object: Union[dict, object],
-    parent_object: dict = None,
+    parent_object: Optional[dict] = None,
 ) -> dict:
-  to_object = {}
+  to_object: dict[str, Any] = {}
   if getv(from_object, ['model']) is not None:
     raise ValueError('model parameter is not supported in Gemini API.')
 
@@ -177,11 +180,11 @@ def _CreateBatchJobParameters_to_mldev(
 
 
 def _CreateBatchJobParameters_to_vertex(
-    api_client: ApiClient,
+    api_client: BaseApiClient,
     from_object: Union[dict, object],
-    parent_object: dict = None,
+    parent_object: Optional[dict] = None,
 ) -> dict:
-  to_object = {}
+  to_object: dict[str, Any] = {}
   if getv(from_object, ['model']) is not None:
     setv(
         to_object,
@@ -213,11 +216,11 @@ def _CreateBatchJobParameters_to_vertex(
 
 
 def _GetBatchJobParameters_to_mldev(
-    api_client: ApiClient,
+    api_client: BaseApiClient,
     from_object: Union[dict, object],
-    parent_object: dict = None,
+    parent_object: Optional[dict] = None,
 ) -> dict:
-  to_object = {}
+  to_object: dict[str, Any] = {}
   if getv(from_object, ['name']) is not None:
     raise ValueError('name parameter is not supported in Gemini API.')
 
@@ -228,11 +231,11 @@ def _GetBatchJobParameters_to_mldev(
 
 
 def _GetBatchJobParameters_to_vertex(
-    api_client: ApiClient,
+    api_client: BaseApiClient,
     from_object: Union[dict, object],
-    parent_object: dict = None,
+    parent_object: Optional[dict] = None,
 ) -> dict:
-  to_object = {}
+  to_object: dict[str, Any] = {}
   if getv(from_object, ['name']) is not None:
     setv(
         to_object,
@@ -247,11 +250,11 @@ def _GetBatchJobParameters_to_vertex(
 
 
 def _CancelBatchJobParameters_to_mldev(
-    api_client: ApiClient,
+    api_client: BaseApiClient,
     from_object: Union[dict, object],
-    parent_object: dict = None,
+    parent_object: Optional[dict] = None,
 ) -> dict:
-  to_object = {}
+  to_object: dict[str, Any] = {}
   if getv(from_object, ['name']) is not None:
     raise ValueError('name parameter is not supported in Gemini API.')
 
@@ -262,11 +265,11 @@ def _CancelBatchJobParameters_to_mldev(
 
 
 def _CancelBatchJobParameters_to_vertex(
-    api_client: ApiClient,
+    api_client: BaseApiClient,
     from_object: Union[dict, object],
-    parent_object: dict = None,
+    parent_object: Optional[dict] = None,
 ) -> dict:
-  to_object = {}
+  to_object: dict[str, Any] = {}
   if getv(from_object, ['name']) is not None:
     setv(
         to_object,
@@ -281,11 +284,11 @@ def _CancelBatchJobParameters_to_vertex(
 
 
 def _ListBatchJobsConfig_to_mldev(
-    api_client: ApiClient,
+    api_client: BaseApiClient,
     from_object: Union[dict, object],
-    parent_object: dict = None,
+    parent_object: Optional[dict] = None,
 ) -> dict:
-  to_object = {}
+  to_object: dict[str, Any] = {}
 
   if getv(from_object, ['page_size']) is not None:
     setv(
@@ -306,11 +309,11 @@ def _ListBatchJobsConfig_to_mldev(
 
 
 def _ListBatchJobsConfig_to_vertex(
-    api_client: ApiClient,
+    api_client: BaseApiClient,
     from_object: Union[dict, object],
-    parent_object: dict = None,
+    parent_object: Optional[dict] = None,
 ) -> dict:
-  to_object = {}
+  to_object: dict[str, Any] = {}
 
   if getv(from_object, ['page_size']) is not None:
     setv(
@@ -331,11 +334,11 @@ def _ListBatchJobsConfig_to_vertex(
 
 
 def _ListBatchJobsParameters_to_mldev(
-    api_client: ApiClient,
+    api_client: BaseApiClient,
     from_object: Union[dict, object],
-    parent_object: dict = None,
+    parent_object: Optional[dict] = None,
 ) -> dict:
-  to_object = {}
+  to_object: dict[str, Any] = {}
   if getv(from_object, ['config']) is not None:
     raise ValueError('config parameter is not supported in Gemini API.')
 
@@ -343,11 +346,11 @@ def _ListBatchJobsParameters_to_mldev(
 
 
 def _ListBatchJobsParameters_to_vertex(
-    api_client: ApiClient,
+    api_client: BaseApiClient,
     from_object: Union[dict, object],
-    parent_object: dict = None,
+    parent_object: Optional[dict] = None,
 ) -> dict:
-  to_object = {}
+  to_object: dict[str, Any] = {}
   if getv(from_object, ['config']) is not None:
     setv(
         to_object,
@@ -361,11 +364,11 @@ def _ListBatchJobsParameters_to_vertex(
 
 
 def _DeleteBatchJobParameters_to_mldev(
-    api_client: ApiClient,
+    api_client: BaseApiClient,
     from_object: Union[dict, object],
-    parent_object: dict = None,
+    parent_object: Optional[dict] = None,
 ) -> dict:
-  to_object = {}
+  to_object: dict[str, Any] = {}
   if getv(from_object, ['name']) is not None:
     raise ValueError('name parameter is not supported in Gemini API.')
 
@@ -376,11 +379,11 @@ def _DeleteBatchJobParameters_to_mldev(
 
 
 def _DeleteBatchJobParameters_to_vertex(
-    api_client: ApiClient,
+    api_client: BaseApiClient,
     from_object: Union[dict, object],
-    parent_object: dict = None,
+    parent_object: Optional[dict] = None,
 ) -> dict:
-  to_object = {}
+  to_object: dict[str, Any] = {}
   if getv(from_object, ['name']) is not None:
     setv(
         to_object,
@@ -395,21 +398,21 @@ def _DeleteBatchJobParameters_to_vertex(
 
 
 def _JobError_from_mldev(
-    api_client: ApiClient,
+    api_client: BaseApiClient,
     from_object: Union[dict, object],
-    parent_object: dict = None,
+    parent_object: Optional[dict] = None,
 ) -> dict:
-  to_object = {}
+  to_object: dict[str, Any] = {}
 
   return to_object
 
 
 def _JobError_from_vertex(
-    api_client: ApiClient,
+    api_client: BaseApiClient,
     from_object: Union[dict, object],
-    parent_object: dict = None,
+    parent_object: Optional[dict] = None,
 ) -> dict:
-  to_object = {}
+  to_object: dict[str, Any] = {}
   if getv(from_object, ['details']) is not None:
     setv(to_object, ['details'], getv(from_object, ['details']))
 
@@ -423,21 +426,21 @@ def _JobError_from_vertex(
 
 
 def _BatchJobSource_from_mldev(
-    api_client: ApiClient,
+    api_client: BaseApiClient,
     from_object: Union[dict, object],
-    parent_object: dict = None,
+    parent_object: Optional[dict] = None,
 ) -> dict:
-  to_object = {}
+  to_object: dict[str, Any] = {}
 
   return to_object
 
 
 def _BatchJobSource_from_vertex(
-    api_client: ApiClient,
+    api_client: BaseApiClient,
     from_object: Union[dict, object],
-    parent_object: dict = None,
+    parent_object: Optional[dict] = None,
 ) -> dict:
-  to_object = {}
+  to_object: dict[str, Any] = {}
   if getv(from_object, ['instancesFormat']) is not None:
     setv(to_object, ['format'], getv(from_object, ['instancesFormat']))
 
@@ -455,21 +458,21 @@ def _BatchJobSource_from_vertex(
 
 
 def _BatchJobDestination_from_mldev(
-    api_client: ApiClient,
+    api_client: BaseApiClient,
     from_object: Union[dict, object],
-    parent_object: dict = None,
+    parent_object: Optional[dict] = None,
 ) -> dict:
-  to_object = {}
+  to_object: dict[str, Any] = {}
 
   return to_object
 
 
 def _BatchJobDestination_from_vertex(
-    api_client: ApiClient,
+    api_client: BaseApiClient,
     from_object: Union[dict, object],
-    parent_object: dict = None,
+    parent_object: Optional[dict] = None,
 ) -> dict:
-  to_object = {}
+  to_object: dict[str, Any] = {}
   if getv(from_object, ['predictionsFormat']) is not None:
     setv(to_object, ['format'], getv(from_object, ['predictionsFormat']))
 
@@ -491,21 +494,21 @@ def _BatchJobDestination_from_vertex(
 
 
 def _BatchJob_from_mldev(
-    api_client: ApiClient,
+    api_client: BaseApiClient,
     from_object: Union[dict, object],
-    parent_object: dict = None,
+    parent_object: Optional[dict] = None,
 ) -> dict:
-  to_object = {}
+  to_object: dict[str, Any] = {}
 
   return to_object
 
 
 def _BatchJob_from_vertex(
-    api_client: ApiClient,
+    api_client: BaseApiClient,
     from_object: Union[dict, object],
-    parent_object: dict = None,
+    parent_object: Optional[dict] = None,
 ) -> dict:
-  to_object = {}
+  to_object: dict[str, Any] = {}
   if getv(from_object, ['name']) is not None:
     setv(to_object, ['name'], getv(from_object, ['name']))
 
@@ -561,11 +564,11 @@ def _BatchJob_from_vertex(
 
 
 def _ListBatchJobsResponse_from_mldev(
-    api_client: ApiClient,
+    api_client: BaseApiClient,
     from_object: Union[dict, object],
-    parent_object: dict = None,
+    parent_object: Optional[dict] = None,
 ) -> dict:
-  to_object = {}
+  to_object: dict[str, Any] = {}
   if getv(from_object, ['nextPageToken']) is not None:
     setv(to_object, ['next_page_token'], getv(from_object, ['nextPageToken']))
 
@@ -573,11 +576,11 @@ def _ListBatchJobsResponse_from_mldev(
 
 
 def _ListBatchJobsResponse_from_vertex(
-    api_client: ApiClient,
+    api_client: BaseApiClient,
     from_object: Union[dict, object],
-    parent_object: dict = None,
+    parent_object: Optional[dict] = None,
 ) -> dict:
-  to_object = {}
+  to_object: dict[str, Any] = {}
   if getv(from_object, ['nextPageToken']) is not None:
     setv(to_object, ['next_page_token'], getv(from_object, ['nextPageToken']))
 
@@ -595,21 +598,21 @@ def _ListBatchJobsResponse_from_vertex(
 
 
 def _DeleteResourceJob_from_mldev(
-    api_client: ApiClient,
+    api_client: BaseApiClient,
     from_object: Union[dict, object],
-    parent_object: dict = None,
+    parent_object: Optional[dict] = None,
 ) -> dict:
-  to_object = {}
+  to_object: dict[str, Any] = {}
 
   return to_object
 
 
 def _DeleteResourceJob_from_vertex(
-    api_client: ApiClient,
+    api_client: BaseApiClient,
     from_object: Union[dict, object],
-    parent_object: dict = None,
+    parent_object: Optional[dict] = None,
 ) -> dict:
-  to_object = {}
+  to_object: dict[str, Any] = {}
   if getv(from_object, ['name']) is not None:
     setv(to_object, ['name'], getv(from_object, ['name']))
 
@@ -643,13 +646,18 @@ class Batches(_api_module.BaseModule):
         config=config,
     )
 
+    request_url_dict: Optional[dict[str, str]]
     if not self._api_client.vertexai:
       raise ValueError('This method is only supported in the Vertex AI client.')
     else:
       request_dict = _CreateBatchJobParameters_to_vertex(
           self._api_client, parameter_model
       )
-      path = 'batchPredictionJobs'.format_map(request_dict.get('_url'))
+      request_url_dict = request_dict.get('_url')
+      if request_url_dict:
+        path = 'batchPredictionJobs'.format_map(request_url_dict)
+      else:
+        path = 'batchPredictionJobs'
 
     query_params = request_dict.get('_query')
     if query_params:
@@ -657,7 +665,7 @@ class Batches(_api_module.BaseModule):
     # TODO: remove the hack that pops config.
     request_dict.pop('config', None)
 
-    http_options = None
+    http_options: Optional[types.HttpOptionsOrDict] = None
     if isinstance(config, dict):
       http_options = config.get('http_options', None)
     elif hasattr(config, 'http_options'):
@@ -676,7 +684,7 @@ class Batches(_api_module.BaseModule):
       response_dict = _BatchJob_from_mldev(self._api_client, response_dict)
 
     return_value = types.BatchJob._from_response(
-        response=response_dict, kwargs=parameter_model
+        response=response_dict, kwargs=parameter_model.model_dump()
     )
     self._api_client._verify_response(return_value)
     return return_value
@@ -707,13 +715,18 @@ class Batches(_api_module.BaseModule):
         config=config,
     )
 
+    request_url_dict: Optional[dict[str, str]]
     if not self._api_client.vertexai:
       raise ValueError('This method is only supported in the Vertex AI client.')
     else:
       request_dict = _GetBatchJobParameters_to_vertex(
           self._api_client, parameter_model
       )
-      path = 'batchPredictionJobs/{name}'.format_map(request_dict.get('_url'))
+      request_url_dict = request_dict.get('_url')
+      if request_url_dict:
+        path = 'batchPredictionJobs/{name}'.format_map(request_url_dict)
+      else:
+        path = 'batchPredictionJobs/{name}'
 
     query_params = request_dict.get('_query')
     if query_params:
@@ -721,7 +734,7 @@ class Batches(_api_module.BaseModule):
     # TODO: remove the hack that pops config.
     request_dict.pop('config', None)
 
-    http_options = None
+    http_options: Optional[types.HttpOptionsOrDict] = None
     if isinstance(config, dict):
       http_options = config.get('http_options', None)
     elif hasattr(config, 'http_options'):
@@ -740,7 +753,7 @@ class Batches(_api_module.BaseModule):
       response_dict = _BatchJob_from_mldev(self._api_client, response_dict)
 
     return_value = types.BatchJob._from_response(
-        response=response_dict, kwargs=parameter_model
+        response=response_dict, kwargs=parameter_model.model_dump()
     )
     self._api_client._verify_response(return_value)
     return return_value
@@ -772,15 +785,18 @@ class Batches(_api_module.BaseModule):
         config=config,
     )
 
+    request_url_dict: Optional[dict[str, str]]
     if not self._api_client.vertexai:
       raise ValueError('This method is only supported in the Vertex AI client.')
     else:
       request_dict = _CancelBatchJobParameters_to_vertex(
           self._api_client, parameter_model
       )
-      path = 'batchPredictionJobs/{name}:cancel'.format_map(
-          request_dict.get('_url')
-      )
+      request_url_dict = request_dict.get('_url')
+      if request_url_dict:
+        path = 'batchPredictionJobs/{name}:cancel'.format_map(request_url_dict)
+      else:
+        path = 'batchPredictionJobs/{name}:cancel'
 
     query_params = request_dict.get('_query')
     if query_params:
@@ -788,7 +804,7 @@ class Batches(_api_module.BaseModule):
     # TODO: remove the hack that pops config.
     request_dict.pop('config', None)
 
-    http_options = None
+    http_options: Optional[types.HttpOptionsOrDict] = None
     if isinstance(config, dict):
       http_options = config.get('http_options', None)
     elif hasattr(config, 'http_options'):
@@ -808,13 +824,18 @@ class Batches(_api_module.BaseModule):
         config=config,
     )
 
+    request_url_dict: Optional[dict[str, str]]
     if not self._api_client.vertexai:
       raise ValueError('This method is only supported in the Vertex AI client.')
     else:
       request_dict = _ListBatchJobsParameters_to_vertex(
           self._api_client, parameter_model
       )
-      path = 'batchPredictionJobs'.format_map(request_dict.get('_url'))
+      request_url_dict = request_dict.get('_url')
+      if request_url_dict:
+        path = 'batchPredictionJobs'.format_map(request_url_dict)
+      else:
+        path = 'batchPredictionJobs'
 
     query_params = request_dict.get('_query')
     if query_params:
@@ -822,7 +843,7 @@ class Batches(_api_module.BaseModule):
     # TODO: remove the hack that pops config.
     request_dict.pop('config', None)
 
-    http_options = None
+    http_options: Optional[types.HttpOptionsOrDict] = None
     if isinstance(config, dict):
       http_options = config.get('http_options', None)
     elif hasattr(config, 'http_options'):
@@ -845,7 +866,7 @@ class Batches(_api_module.BaseModule):
       )
 
     return_value = types.ListBatchJobsResponse._from_response(
-        response=response_dict, kwargs=parameter_model
+        response=response_dict, kwargs=parameter_model.model_dump()
     )
     self._api_client._verify_response(return_value)
     return return_value
@@ -878,13 +899,18 @@ class Batches(_api_module.BaseModule):
         config=config,
     )
 
+    request_url_dict: Optional[dict[str, str]]
     if not self._api_client.vertexai:
       raise ValueError('This method is only supported in the Vertex AI client.')
     else:
       request_dict = _DeleteBatchJobParameters_to_vertex(
           self._api_client, parameter_model
       )
-      path = 'batchPredictionJobs/{name}'.format_map(request_dict.get('_url'))
+      request_url_dict = request_dict.get('_url')
+      if request_url_dict:
+        path = 'batchPredictionJobs/{name}'.format_map(request_url_dict)
+      else:
+        path = 'batchPredictionJobs/{name}'
 
     query_params = request_dict.get('_query')
     if query_params:
@@ -892,7 +918,7 @@ class Batches(_api_module.BaseModule):
     # TODO: remove the hack that pops config.
     request_dict.pop('config', None)
 
-    http_options = None
+    http_options: Optional[types.HttpOptionsOrDict] = None
     if isinstance(config, dict):
       http_options = config.get('http_options', None)
     elif hasattr(config, 'http_options'):
@@ -915,7 +941,7 @@ class Batches(_api_module.BaseModule):
       )
 
     return_value = types.DeleteResourceJob._from_response(
-        response=response_dict, kwargs=parameter_model
+        response=response_dict, kwargs=parameter_model.model_dump()
     )
     self._api_client._verify_response(return_value)
     return return_value
@@ -995,13 +1021,18 @@ class AsyncBatches(_api_module.BaseModule):
         config=config,
     )
 
+    request_url_dict: Optional[dict[str, str]]
     if not self._api_client.vertexai:
       raise ValueError('This method is only supported in the Vertex AI client.')
     else:
       request_dict = _CreateBatchJobParameters_to_vertex(
           self._api_client, parameter_model
       )
-      path = 'batchPredictionJobs'.format_map(request_dict.get('_url'))
+      request_url_dict = request_dict.get('_url')
+      if request_url_dict:
+        path = 'batchPredictionJobs'.format_map(request_url_dict)
+      else:
+        path = 'batchPredictionJobs'
 
     query_params = request_dict.get('_query')
     if query_params:
@@ -1009,7 +1040,7 @@ class AsyncBatches(_api_module.BaseModule):
     # TODO: remove the hack that pops config.
     request_dict.pop('config', None)
 
-    http_options = None
+    http_options: Optional[types.HttpOptionsOrDict] = None
     if isinstance(config, dict):
       http_options = config.get('http_options', None)
     elif hasattr(config, 'http_options'):
@@ -1028,7 +1059,7 @@ class AsyncBatches(_api_module.BaseModule):
       response_dict = _BatchJob_from_mldev(self._api_client, response_dict)
 
     return_value = types.BatchJob._from_response(
-        response=response_dict, kwargs=parameter_model
+        response=response_dict, kwargs=parameter_model.model_dump()
     )
     self._api_client._verify_response(return_value)
     return return_value
@@ -1050,7 +1081,7 @@ class AsyncBatches(_api_module.BaseModule):
 
     .. code-block:: python
 
-      batch_job = client.batches.get(name='123456789')
+      batch_job = await client.aio.batches.get(name='123456789')
       print(f"Batch job: {batch_job.name}, state {batch_job.state}")
     """
 
@@ -1059,13 +1090,18 @@ class AsyncBatches(_api_module.BaseModule):
         config=config,
     )
 
+    request_url_dict: Optional[dict[str, str]]
     if not self._api_client.vertexai:
       raise ValueError('This method is only supported in the Vertex AI client.')
     else:
       request_dict = _GetBatchJobParameters_to_vertex(
           self._api_client, parameter_model
       )
-      path = 'batchPredictionJobs/{name}'.format_map(request_dict.get('_url'))
+      request_url_dict = request_dict.get('_url')
+      if request_url_dict:
+        path = 'batchPredictionJobs/{name}'.format_map(request_url_dict)
+      else:
+        path = 'batchPredictionJobs/{name}'
 
     query_params = request_dict.get('_query')
     if query_params:
@@ -1073,7 +1109,7 @@ class AsyncBatches(_api_module.BaseModule):
     # TODO: remove the hack that pops config.
     request_dict.pop('config', None)
 
-    http_options = None
+    http_options: Optional[types.HttpOptionsOrDict] = None
     if isinstance(config, dict):
       http_options = config.get('http_options', None)
     elif hasattr(config, 'http_options'):
@@ -1092,7 +1128,7 @@ class AsyncBatches(_api_module.BaseModule):
       response_dict = _BatchJob_from_mldev(self._api_client, response_dict)
 
     return_value = types.BatchJob._from_response(
-        response=response_dict, kwargs=parameter_model
+        response=response_dict, kwargs=parameter_model.model_dump()
     )
     self._api_client._verify_response(return_value)
     return return_value
@@ -1116,7 +1152,7 @@ class AsyncBatches(_api_module.BaseModule):
 
     .. code-block:: python
 
-      client.batches.cancel(name='123456789')
+      await client.aio.batches.cancel(name='123456789')
     """
 
     parameter_model = types._CancelBatchJobParameters(
@@ -1124,15 +1160,18 @@ class AsyncBatches(_api_module.BaseModule):
         config=config,
     )
 
+    request_url_dict: Optional[dict[str, str]]
     if not self._api_client.vertexai:
       raise ValueError('This method is only supported in the Vertex AI client.')
     else:
       request_dict = _CancelBatchJobParameters_to_vertex(
           self._api_client, parameter_model
       )
-      path = 'batchPredictionJobs/{name}:cancel'.format_map(
-          request_dict.get('_url')
-      )
+      request_url_dict = request_dict.get('_url')
+      if request_url_dict:
+        path = 'batchPredictionJobs/{name}:cancel'.format_map(request_url_dict)
+      else:
+        path = 'batchPredictionJobs/{name}:cancel'
 
     query_params = request_dict.get('_query')
     if query_params:
@@ -1140,7 +1179,7 @@ class AsyncBatches(_api_module.BaseModule):
     # TODO: remove the hack that pops config.
     request_dict.pop('config', None)
 
-    http_options = None
+    http_options: Optional[types.HttpOptionsOrDict] = None
     if isinstance(config, dict):
       http_options = config.get('http_options', None)
     elif hasattr(config, 'http_options'):
@@ -1160,13 +1199,18 @@ class AsyncBatches(_api_module.BaseModule):
         config=config,
     )
 
+    request_url_dict: Optional[dict[str, str]]
     if not self._api_client.vertexai:
       raise ValueError('This method is only supported in the Vertex AI client.')
     else:
       request_dict = _ListBatchJobsParameters_to_vertex(
           self._api_client, parameter_model
       )
-      path = 'batchPredictionJobs'.format_map(request_dict.get('_url'))
+      request_url_dict = request_dict.get('_url')
+      if request_url_dict:
+        path = 'batchPredictionJobs'.format_map(request_url_dict)
+      else:
+        path = 'batchPredictionJobs'
 
     query_params = request_dict.get('_query')
     if query_params:
@@ -1174,7 +1218,7 @@ class AsyncBatches(_api_module.BaseModule):
     # TODO: remove the hack that pops config.
     request_dict.pop('config', None)
 
-    http_options = None
+    http_options: Optional[types.HttpOptionsOrDict] = None
     if isinstance(config, dict):
       http_options = config.get('http_options', None)
     elif hasattr(config, 'http_options'):
@@ -1197,7 +1241,7 @@ class AsyncBatches(_api_module.BaseModule):
       )
 
     return_value = types.ListBatchJobsResponse._from_response(
-        response=response_dict, kwargs=parameter_model
+        response=response_dict, kwargs=parameter_model.model_dump()
     )
     self._api_client._verify_response(return_value)
     return return_value
@@ -1222,7 +1266,7 @@ class AsyncBatches(_api_module.BaseModule):
 
     .. code-block:: python
 
-      client.batches.delete(name='123456789')
+      await client.aio.batches.delete(name='123456789')
     """
 
     parameter_model = types._DeleteBatchJobParameters(
@@ -1230,13 +1274,18 @@ class AsyncBatches(_api_module.BaseModule):
         config=config,
     )
 
+    request_url_dict: Optional[dict[str, str]]
     if not self._api_client.vertexai:
       raise ValueError('This method is only supported in the Vertex AI client.')
     else:
       request_dict = _DeleteBatchJobParameters_to_vertex(
           self._api_client, parameter_model
       )
-      path = 'batchPredictionJobs/{name}'.format_map(request_dict.get('_url'))
+      request_url_dict = request_dict.get('_url')
+      if request_url_dict:
+        path = 'batchPredictionJobs/{name}'.format_map(request_url_dict)
+      else:
+        path = 'batchPredictionJobs/{name}'
 
     query_params = request_dict.get('_query')
     if query_params:
@@ -1244,7 +1293,7 @@ class AsyncBatches(_api_module.BaseModule):
     # TODO: remove the hack that pops config.
     request_dict.pop('config', None)
 
-    http_options = None
+    http_options: Optional[types.HttpOptionsOrDict] = None
     if isinstance(config, dict):
       http_options = config.get('http_options', None)
     elif hasattr(config, 'http_options'):
@@ -1267,7 +1316,7 @@ class AsyncBatches(_api_module.BaseModule):
       )
 
     return_value = types.DeleteResourceJob._from_response(
-        response=response_dict, kwargs=parameter_model
+        response=response_dict, kwargs=parameter_model.model_dump()
     )
     self._api_client._verify_response(return_value)
     return return_value