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

google/genai/tunings.py

@@ -17,6 +17,7 @@
 
 from typing import Optional, Union
 from urllib.parse import urlencode
+from . import _api_module
 from . import _common
 from . import _transformers as t
 from . import types
@@ -26,30 +27,6 @@ from ._common import set_value_by_path as setv
 from .pagers import AsyncPager, Pager
 
 
-def _GetTuningJobConfig_to_mldev(
-    api_client: ApiClient,
-    from_object: Union[dict, object],
-    parent_object: dict = None,
-) -> dict:
-  to_object = {}
-  if getv(from_object, ['http_options']) is not None:
-    setv(to_object, ['httpOptions'], getv(from_object, ['http_options']))
-
-  return to_object
-
-
-def _GetTuningJobConfig_to_vertex(
-    api_client: ApiClient,
-    from_object: Union[dict, object],
-    parent_object: dict = None,
-) -> dict:
-  to_object = {}
-  if getv(from_object, ['http_options']) is not None:
-    setv(to_object, ['httpOptions'], getv(from_object, ['http_options']))
-
-  return to_object
-
-
 def _GetTuningJobParameters_to_mldev(
     api_client: ApiClient,
     from_object: Union[dict, object],
@@ -60,13 +37,7 @@ def _GetTuningJobParameters_to_mldev(
     setv(to_object, ['_url', 'name'], getv(from_object, ['name']))
 
   if getv(from_object, ['config']) is not None:
-    setv(
-        to_object,
-        ['config'],
-        _GetTuningJobConfig_to_mldev(
-            api_client, getv(from_object, ['config']), to_object
-        ),
-    )
+    setv(to_object, ['config'], getv(from_object, ['config']))
 
   return to_object
 
@@ -81,13 +52,7 @@ def _GetTuningJobParameters_to_vertex(
     setv(to_object, ['_url', 'name'], getv(from_object, ['name']))
 
   if getv(from_object, ['config']) is not None:
-    setv(
-        to_object,
-        ['config'],
-        _GetTuningJobConfig_to_vertex(
-            api_client, getv(from_object, ['config']), to_object
-        ),
-    )
+    setv(to_object, ['config'], getv(from_object, ['config']))
 
   return to_object
 
@@ -98,6 +63,7 @@ def _ListTuningJobsConfig_to_mldev(
     parent_object: dict = None,
 ) -> dict:
   to_object = {}
+
   if getv(from_object, ['page_size']) is not None:
     setv(
         parent_object, ['_query', 'pageSize'], getv(from_object, ['page_size'])
@@ -122,6 +88,7 @@ def _ListTuningJobsConfig_to_vertex(
     parent_object: dict = None,
 ) -> dict:
   to_object = {}
+
   if getv(from_object, ['page_size']) is not None:
     setv(
         parent_object, ['_query', 'pageSize'], getv(from_object, ['page_size'])
@@ -213,7 +180,7 @@ def _TuningDataset_to_mldev(
 ) -> dict:
   to_object = {}
   if getv(from_object, ['gcs_uri']) is not None:
-    raise ValueError('gcs_uri parameter is not supported in Google AI.')
+    raise ValueError('gcs_uri parameter is not supported in Gemini API.')
 
   if getv(from_object, ['examples']) is not None:
     setv(
@@ -254,7 +221,7 @@ def _TuningValidationDataset_to_mldev(
 ) -> dict:
   to_object = {}
   if getv(from_object, ['gcs_uri']) is not None:
-    raise ValueError('gcs_uri parameter is not supported in Google AI.')
+    raise ValueError('gcs_uri parameter is not supported in Gemini API.')
 
   return to_object
 
@@ -277,12 +244,10 @@ def _CreateTuningJobConfig_to_mldev(
     parent_object: dict = None,
 ) -> dict:
   to_object = {}
-  if getv(from_object, ['http_options']) is not None:
-    setv(to_object, ['httpOptions'], getv(from_object, ['http_options']))
 
   if getv(from_object, ['validation_dataset']) is not None:
     raise ValueError(
-        'validation_dataset parameter is not supported in Google AI.'
+        'validation_dataset parameter is not supported in Gemini API.'
     )
 
   if getv(from_object, ['tuned_model_display_name']) is not None:
@@ -293,7 +258,7 @@ def _CreateTuningJobConfig_to_mldev(
     )
 
   if getv(from_object, ['description']) is not None:
-    raise ValueError('description parameter is not supported in Google AI.')
+    raise ValueError('description parameter is not supported in Gemini API.')
 
   if getv(from_object, ['epoch_count']) is not None:
     setv(
@@ -310,7 +275,7 @@ def _CreateTuningJobConfig_to_mldev(
     )
 
   if getv(from_object, ['adapter_size']) is not None:
-    raise ValueError('adapter_size parameter is not supported in Google AI.')
+    raise ValueError('adapter_size parameter is not supported in Gemini API.')
 
   if getv(from_object, ['batch_size']) is not None:
     setv(
@@ -335,8 +300,6 @@ def _CreateTuningJobConfig_to_vertex(
     parent_object: dict = None,
 ) -> dict:
   to_object = {}
-  if getv(from_object, ['http_options']) is not None:
-    setv(to_object, ['httpOptions'], getv(from_object, ['http_options']))
 
   if getv(from_object, ['validation_dataset']) is not None:
     setv(
@@ -447,234 +410,6 @@ def _CreateTuningJobParameters_to_vertex(
   return to_object
 
 
-def _DistillationDataset_to_mldev(
-    api_client: ApiClient,
-    from_object: Union[dict, object],
-    parent_object: dict = None,
-) -> dict:
-  to_object = {}
-  if getv(from_object, ['gcs_uri']) is not None:
-    raise ValueError('gcs_uri parameter is not supported in Google AI.')
-
-  return to_object
-
-
-def _DistillationDataset_to_vertex(
-    api_client: ApiClient,
-    from_object: Union[dict, object],
-    parent_object: dict = None,
-) -> dict:
-  to_object = {}
-  if getv(from_object, ['gcs_uri']) is not None:
-    setv(
-        parent_object,
-        ['distillationSpec', 'trainingDatasetUri'],
-        getv(from_object, ['gcs_uri']),
-    )
-
-  return to_object
-
-
-def _DistillationValidationDataset_to_mldev(
-    api_client: ApiClient,
-    from_object: Union[dict, object],
-    parent_object: dict = None,
-) -> dict:
-  to_object = {}
-  if getv(from_object, ['gcs_uri']) is not None:
-    raise ValueError('gcs_uri parameter is not supported in Google AI.')
-
-  return to_object
-
-
-def _DistillationValidationDataset_to_vertex(
-    api_client: ApiClient,
-    from_object: Union[dict, object],
-    parent_object: dict = None,
-) -> dict:
-  to_object = {}
-  if getv(from_object, ['gcs_uri']) is not None:
-    setv(to_object, ['validationDatasetUri'], getv(from_object, ['gcs_uri']))
-
-  return to_object
-
-
-def _CreateDistillationJobConfig_to_mldev(
-    api_client: ApiClient,
-    from_object: Union[dict, object],
-    parent_object: dict = None,
-) -> dict:
-  to_object = {}
-  if getv(from_object, ['http_options']) is not None:
-    setv(to_object, ['httpOptions'], getv(from_object, ['http_options']))
-
-  if getv(from_object, ['validation_dataset']) is not None:
-    raise ValueError(
-        'validation_dataset parameter is not supported in Google AI.'
-    )
-
-  if getv(from_object, ['tuned_model_display_name']) is not None:
-    setv(
-        parent_object,
-        ['displayName'],
-        getv(from_object, ['tuned_model_display_name']),
-    )
-
-  if getv(from_object, ['epoch_count']) is not None:
-    setv(
-        parent_object,
-        ['tuningTask', 'hyperparameters', 'epochCount'],
-        getv(from_object, ['epoch_count']),
-    )
-
-  if getv(from_object, ['learning_rate_multiplier']) is not None:
-    setv(
-        parent_object,
-        ['tuningTask', 'hyperparameters', 'learningRateMultiplier'],
-        getv(from_object, ['learning_rate_multiplier']),
-    )
-
-  if getv(from_object, ['adapter_size']) is not None:
-    raise ValueError('adapter_size parameter is not supported in Google AI.')
-
-  if getv(from_object, ['pipeline_root_directory']) is not None:
-    raise ValueError(
-        'pipeline_root_directory parameter is not supported in Google AI.'
-    )
-
-  return to_object
-
-
-def _CreateDistillationJobConfig_to_vertex(
-    api_client: ApiClient,
-    from_object: Union[dict, object],
-    parent_object: dict = None,
-) -> dict:
-  to_object = {}
-  if getv(from_object, ['http_options']) is not None:
-    setv(to_object, ['httpOptions'], getv(from_object, ['http_options']))
-
-  if getv(from_object, ['validation_dataset']) is not None:
-    setv(
-        parent_object,
-        ['distillationSpec'],
-        _DistillationValidationDataset_to_vertex(
-            api_client, getv(from_object, ['validation_dataset']), to_object
-        ),
-    )
-
-  if getv(from_object, ['tuned_model_display_name']) is not None:
-    setv(
-        parent_object,
-        ['tunedModelDisplayName'],
-        getv(from_object, ['tuned_model_display_name']),
-    )
-
-  if getv(from_object, ['epoch_count']) is not None:
-    setv(
-        parent_object,
-        ['distillationSpec', 'hyperParameters', 'epochCount'],
-        getv(from_object, ['epoch_count']),
-    )
-
-  if getv(from_object, ['learning_rate_multiplier']) is not None:
-    setv(
-        parent_object,
-        ['distillationSpec', 'hyperParameters', 'learningRateMultiplier'],
-        getv(from_object, ['learning_rate_multiplier']),
-    )
-
-  if getv(from_object, ['adapter_size']) is not None:
-    setv(
-        parent_object,
-        ['distillationSpec', 'hyperParameters', 'adapterSize'],
-        getv(from_object, ['adapter_size']),
-    )
-
-  if getv(from_object, ['pipeline_root_directory']) is not None:
-    setv(
-        parent_object,
-        ['distillationSpec', 'pipelineRootDirectory'],
-        getv(from_object, ['pipeline_root_directory']),
-    )
-
-  return to_object
-
-
-def _CreateDistillationJobParameters_to_mldev(
-    api_client: ApiClient,
-    from_object: Union[dict, object],
-    parent_object: dict = None,
-) -> dict:
-  to_object = {}
-  if getv(from_object, ['student_model']) is not None:
-    raise ValueError('student_model parameter is not supported in Google AI.')
-
-  if getv(from_object, ['teacher_model']) is not None:
-    raise ValueError('teacher_model parameter is not supported in Google AI.')
-
-  if getv(from_object, ['training_dataset']) is not None:
-    setv(
-        to_object,
-        ['tuningTask', 'trainingData'],
-        _DistillationDataset_to_mldev(
-            api_client, getv(from_object, ['training_dataset']), to_object
-        ),
-    )
-
-  if getv(from_object, ['config']) is not None:
-    setv(
-        to_object,
-        ['config'],
-        _CreateDistillationJobConfig_to_mldev(
-            api_client, getv(from_object, ['config']), to_object
-        ),
-    )
-
-  return to_object
-
-
-def _CreateDistillationJobParameters_to_vertex(
-    api_client: ApiClient,
-    from_object: Union[dict, object],
-    parent_object: dict = None,
-) -> dict:
-  to_object = {}
-  if getv(from_object, ['student_model']) is not None:
-    setv(
-        to_object,
-        ['distillationSpec', 'studentModel'],
-        getv(from_object, ['student_model']),
-    )
-
-  if getv(from_object, ['teacher_model']) is not None:
-    setv(
-        to_object,
-        ['distillationSpec', 'baseTeacherModel'],
-        getv(from_object, ['teacher_model']),
-    )
-
-  if getv(from_object, ['training_dataset']) is not None:
-    setv(
-        to_object,
-        ['distillationSpec', 'trainingDatasetUri'],
-        _DistillationDataset_to_vertex(
-            api_client, getv(from_object, ['training_dataset']), to_object
-        ),
-    )
-
-  if getv(from_object, ['config']) is not None:
-    setv(
-        to_object,
-        ['config'],
-        _CreateDistillationJobConfig_to_vertex(
-            api_client, getv(from_object, ['config']), to_object
-        ),
-    )
-
-  return to_object
-
-
 def _TunedModel_from_mldev(
     api_client: ApiClient,
     from_object: Union[dict, object],
@@ -756,12 +491,22 @@ def _TuningJob_from_mldev(
         ),
     )
 
+  if getv(from_object, ['distillationSpec']) is not None:
+    setv(
+        to_object,
+        ['distillation_spec'],
+        getv(from_object, ['distillationSpec']),
+    )
+
   if getv(from_object, ['experiment']) is not None:
     setv(to_object, ['experiment'], getv(from_object, ['experiment']))
 
   if getv(from_object, ['labels']) is not None:
     setv(to_object, ['labels'], getv(from_object, ['labels']))
 
+  if getv(from_object, ['pipelineJob']) is not None:
+    setv(to_object, ['pipeline_job'], getv(from_object, ['pipelineJob']))
+
   if getv(from_object, ['tunedModelDisplayName']) is not None:
     setv(
         to_object,
@@ -833,13 +578,6 @@ def _TuningJob_from_vertex(
   if getv(from_object, ['encryptionSpec']) is not None:
     setv(to_object, ['encryption_spec'], getv(from_object, ['encryptionSpec']))
 
-  if getv(from_object, ['distillationSpec']) is not None:
-    setv(
-        to_object,
-        ['distillation_spec'],
-        getv(from_object, ['distillationSpec']),
-    )
-
   if getv(from_object, ['partnerModelTuningSpec']) is not None:
     setv(
         to_object,
@@ -847,8 +585,12 @@ def _TuningJob_from_vertex(
         getv(from_object, ['partnerModelTuningSpec']),
     )
 
-  if getv(from_object, ['pipelineJob']) is not None:
-    setv(to_object, ['pipeline_job'], getv(from_object, ['pipelineJob']))
+  if getv(from_object, ['distillationSpec']) is not None:
+    setv(
+        to_object,
+        ['distillation_spec'],
+        getv(from_object, ['distillationSpec']),
+    )
 
   if getv(from_object, ['experiment']) is not None:
     setv(to_object, ['experiment'], getv(from_object, ['experiment']))
@@ -856,6 +598,9 @@ def _TuningJob_from_vertex(
   if getv(from_object, ['labels']) is not None:
     setv(to_object, ['labels'], getv(from_object, ['labels']))
 
+  if getv(from_object, ['pipelineJob']) is not None:
+    setv(to_object, ['pipeline_job'], getv(from_object, ['pipelineJob']))
+
   if getv(from_object, ['tunedModelDisplayName']) is not None:
     setv(
         to_object,
@@ -950,7 +695,7 @@ def _TuningJobOrOperation_from_vertex(
   return to_object
 
 
-class Tunings(_common.BaseModule):
+class Tunings(_api_module.BaseModule):
 
   def _get(
       self,
@@ -986,8 +731,14 @@ class Tunings(_common.BaseModule):
     if query_params:
       path = f'{path}?{urlencode(query_params)}'
     # TODO: remove the hack that pops config.
-    config = request_dict.pop('config', None)
-    http_options = config.pop('httpOptions', None) if config else None
+    request_dict.pop('config', None)
+
+    http_options = None
+    if isinstance(config, dict):
+      http_options = config.get('http_options', None)
+    elif hasattr(config, 'http_options'):
+      http_options = config.http_options
+
     request_dict = _common.convert_to_dict(request_dict)
     request_dict = _common.encode_unserializable_types(request_dict)
 
@@ -1001,7 +752,7 @@ class Tunings(_common.BaseModule):
       response_dict = _TuningJob_from_mldev(self._api_client, response_dict)
 
     return_value = types.TuningJob._from_response(
-        response_dict, parameter_model
+        response=response_dict, kwargs=parameter_model
     )
     self._api_client._verify_response(return_value)
     return return_value
@@ -1036,8 +787,14 @@ class Tunings(_common.BaseModule):
     if query_params:
       path = f'{path}?{urlencode(query_params)}'
     # TODO: remove the hack that pops config.
-    config = request_dict.pop('config', None)
-    http_options = config.pop('httpOptions', None) if config else None
+    request_dict.pop('config', None)
+
+    http_options = None
+    if isinstance(config, dict):
+      http_options = config.get('http_options', None)
+    elif hasattr(config, 'http_options'):
+      http_options = config.http_options
+
     request_dict = _common.convert_to_dict(request_dict)
     request_dict = _common.encode_unserializable_types(request_dict)
 
@@ -1055,7 +812,7 @@ class Tunings(_common.BaseModule):
       )
 
     return_value = types.ListTuningJobsResponse._from_response(
-        response_dict, parameter_model
+        response=response_dict, kwargs=parameter_model
     )
     self._api_client._verify_response(return_value)
     return return_value
@@ -1098,8 +855,14 @@ class Tunings(_common.BaseModule):
     if query_params:
       path = f'{path}?{urlencode(query_params)}'
     # TODO: remove the hack that pops config.
-    config = request_dict.pop('config', None)
-    http_options = config.pop('httpOptions', None) if config else None
+    request_dict.pop('config', None)
+
+    http_options = None
+    if isinstance(config, dict):
+      http_options = config.get('http_options', None)
+    elif hasattr(config, 'http_options'):
+      http_options = config.http_options
+
     request_dict = _common.convert_to_dict(request_dict)
     request_dict = _common.encode_unserializable_types(request_dict)
 
@@ -1117,70 +880,11 @@ class Tunings(_common.BaseModule):
       )
 
     return_value = types.TuningJobOrOperation._from_response(
-        response_dict, parameter_model
+        response=response_dict, kwargs=parameter_model
     ).tuning_job
     self._api_client._verify_response(return_value)
     return return_value
 
-  def distill(
-      self,
-      *,
-      student_model: str,
-      teacher_model: str,
-      training_dataset: types.DistillationDatasetOrDict,
-      config: Optional[types.CreateDistillationJobConfigOrDict] = None,
-  ) -> types.TuningJob:
-    """Creates a distillation job.
-
-    Args:
-      student_model: The name of the model to tune.
-      teacher_model: The name of the model to distill from.
-      training_dataset: The training dataset to use.
-      config: The configuration to use for the distillation job.
-
-    Returns:
-      A TuningJob object.
-    """
-
-    parameter_model = types._CreateDistillationJobParameters(
-        student_model=student_model,
-        teacher_model=teacher_model,
-        training_dataset=training_dataset,
-        config=config,
-    )
-
-    if not self._api_client.vertexai:
-      raise ValueError('This method is only supported in the Vertex AI client.')
-    else:
-      request_dict = _CreateDistillationJobParameters_to_vertex(
-          self._api_client, parameter_model
-      )
-      path = 'tuningJobs'.format_map(request_dict.get('_url'))
-
-    query_params = request_dict.get('_query')
-    if query_params:
-      path = f'{path}?{urlencode(query_params)}'
-    # TODO: remove the hack that pops config.
-    config = request_dict.pop('config', None)
-    http_options = config.pop('httpOptions', None) if config else None
-    request_dict = _common.convert_to_dict(request_dict)
-    request_dict = _common.encode_unserializable_types(request_dict)
-
-    response_dict = self._api_client.request(
-        'post', path, request_dict, http_options
-    )
-
-    if self._api_client.vertexai:
-      response_dict = _TuningJob_from_vertex(self._api_client, response_dict)
-    else:
-      response_dict = _TuningJob_from_mldev(self._api_client, response_dict)
-
-    return_value = types.TuningJob._from_response(
-        response_dict, parameter_model
-    )
-    self._api_client._verify_response(return_value)
-    return return_value
-
   def list(
       self, *, config: Optional[types.ListTuningJobsConfigOrDict] = None
   ) -> Pager[types.TuningJob]:
@@ -1222,7 +926,7 @@ class Tunings(_common.BaseModule):
     return result
 
 
-class AsyncTunings(_common.BaseModule):
+class AsyncTunings(_api_module.BaseModule):
 
   async def _get(
       self,
@@ -1258,8 +962,14 @@ class AsyncTunings(_common.BaseModule):
     if query_params:
       path = f'{path}?{urlencode(query_params)}'
     # TODO: remove the hack that pops config.
-    config = request_dict.pop('config', None)
-    http_options = config.pop('httpOptions', None) if config else None
+    request_dict.pop('config', None)
+
+    http_options = None
+    if isinstance(config, dict):
+      http_options = config.get('http_options', None)
+    elif hasattr(config, 'http_options'):
+      http_options = config.http_options
+
     request_dict = _common.convert_to_dict(request_dict)
     request_dict = _common.encode_unserializable_types(request_dict)
 
@@ -1273,7 +983,7 @@ class AsyncTunings(_common.BaseModule):
       response_dict = _TuningJob_from_mldev(self._api_client, response_dict)
 
     return_value = types.TuningJob._from_response(
-        response_dict, parameter_model
+        response=response_dict, kwargs=parameter_model
     )
     self._api_client._verify_response(return_value)
     return return_value
@@ -1308,8 +1018,14 @@ class AsyncTunings(_common.BaseModule):
     if query_params:
       path = f'{path}?{urlencode(query_params)}'
     # TODO: remove the hack that pops config.
-    config = request_dict.pop('config', None)
-    http_options = config.pop('httpOptions', None) if config else None
+    request_dict.pop('config', None)
+
+    http_options = None
+    if isinstance(config, dict):
+      http_options = config.get('http_options', None)
+    elif hasattr(config, 'http_options'):
+      http_options = config.http_options
+
     request_dict = _common.convert_to_dict(request_dict)
     request_dict = _common.encode_unserializable_types(request_dict)
 
@@ -1327,7 +1043,7 @@ class AsyncTunings(_common.BaseModule):
       )
 
     return_value = types.ListTuningJobsResponse._from_response(
-        response_dict, parameter_model
+        response=response_dict, kwargs=parameter_model
     )
     self._api_client._verify_response(return_value)
     return return_value
@@ -1370,8 +1086,14 @@ class AsyncTunings(_common.BaseModule):
     if query_params:
       path = f'{path}?{urlencode(query_params)}'
     # TODO: remove the hack that pops config.
-    config = request_dict.pop('config', None)
-    http_options = config.pop('httpOptions', None) if config else None
+    request_dict.pop('config', None)
+
+    http_options = None
+    if isinstance(config, dict):
+      http_options = config.get('http_options', None)
+    elif hasattr(config, 'http_options'):
+      http_options = config.http_options
+
     request_dict = _common.convert_to_dict(request_dict)
     request_dict = _common.encode_unserializable_types(request_dict)
 
@@ -1389,70 +1111,11 @@ class AsyncTunings(_common.BaseModule):
       )
 
     return_value = types.TuningJobOrOperation._from_response(
-        response_dict, parameter_model
+        response=response_dict, kwargs=parameter_model
     ).tuning_job
     self._api_client._verify_response(return_value)
     return return_value
 
-  async def distill(
-      self,
-      *,
-      student_model: str,
-      teacher_model: str,
-      training_dataset: types.DistillationDatasetOrDict,
-      config: Optional[types.CreateDistillationJobConfigOrDict] = None,
-  ) -> types.TuningJob:
-    """Creates a distillation job.
-
-    Args:
-      student_model: The name of the model to tune.
-      teacher_model: The name of the model to distill from.
-      training_dataset: The training dataset to use.
-      config: The configuration to use for the distillation job.
-
-    Returns:
-      A TuningJob object.
-    """
-
-    parameter_model = types._CreateDistillationJobParameters(
-        student_model=student_model,
-        teacher_model=teacher_model,
-        training_dataset=training_dataset,
-        config=config,
-    )
-
-    if not self._api_client.vertexai:
-      raise ValueError('This method is only supported in the Vertex AI client.')
-    else:
-      request_dict = _CreateDistillationJobParameters_to_vertex(
-          self._api_client, parameter_model
-      )
-      path = 'tuningJobs'.format_map(request_dict.get('_url'))
-
-    query_params = request_dict.get('_query')
-    if query_params:
-      path = f'{path}?{urlencode(query_params)}'
-    # TODO: remove the hack that pops config.
-    config = request_dict.pop('config', None)
-    http_options = config.pop('httpOptions', None) if config else None
-    request_dict = _common.convert_to_dict(request_dict)
-    request_dict = _common.encode_unserializable_types(request_dict)
-
-    response_dict = await self._api_client.async_request(
-        'post', path, request_dict, http_options
-    )
-
-    if self._api_client.vertexai:
-      response_dict = _TuningJob_from_vertex(self._api_client, response_dict)
-    else:
-      response_dict = _TuningJob_from_mldev(self._api_client, response_dict)
-
-    return_value = types.TuningJob._from_response(
-        response_dict, parameter_model
-    )
-    self._api_client._verify_response(return_value)
-    return return_value
-
   async def list(
       self, *, config: Optional[types.ListTuningJobsConfigOrDict] = None
   ) -> AsyncPager[types.TuningJob]: