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

google/genai/batches.py

@@ -16,14 +16,14 @@
 # Code generated by the Google Gen AI SDK generator DO NOT EDIT.
 
 import logging
-from typing import Optional, Union
+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
@@ -32,11 +32,11 @@ 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.')
 
@@ -50,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']))
 
@@ -72,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.')
 
@@ -90,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']))
 
@@ -116,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']))
@@ -132,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']))
@@ -156,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.')
 
@@ -180,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,
@@ -216,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.')
 
@@ -231,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,
@@ -250,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.')
 
@@ -265,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,
@@ -284,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(
@@ -309,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(
@@ -334,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.')
 
@@ -346,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,
@@ -364,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.')
 
@@ -379,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,
@@ -398,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']))
 
@@ -426,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']))
 
@@ -458,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']))
 
@@ -494,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']))
 
@@ -564,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']))
 
@@ -576,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']))
 
@@ -598,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']))
 
@@ -646,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:
@@ -660,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'):
@@ -679,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
@@ -710,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:
@@ -724,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'):
@@ -743,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
@@ -775,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:
@@ -791,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'):
@@ -811,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:
@@ -825,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'):
@@ -848,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
@@ -881,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:
@@ -895,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'):
@@ -918,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
@@ -998,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:
@@ -1012,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'):
@@ -1031,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
@@ -1062,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:
@@ -1076,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'):
@@ -1095,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
@@ -1127,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:
@@ -1143,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'):
@@ -1163,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:
@@ -1177,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'):
@@ -1200,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
@@ -1233,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:
@@ -1247,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'):
@@ -1270,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