google-genai 1.38.0__py3-none-any.whl → 1.39.0__py3-none-any.whl

google/genai/tunings.py

@@ -32,57 +32,36 @@ from .pagers import AsyncPager, Pager
 logger = logging.getLogger('google_genai.tunings')
 
 
-def _GetTuningJobParameters_to_mldev(
+def _AutoraterConfig_from_vertex(
     from_object: Union[dict[str, Any], object],
     parent_object: Optional[dict[str, Any]] = None,
 ) -> dict[str, Any]:
   to_object: dict[str, Any] = {}
-  if getv(from_object, ['name']) is not None:
-    setv(to_object, ['_url', 'name'], getv(from_object, ['name']))
+  if getv(from_object, ['samplingCount']) is not None:
+    setv(to_object, ['sampling_count'], getv(from_object, ['samplingCount']))
 
-  if getv(from_object, ['config']) is not None:
-    setv(to_object, ['config'], getv(from_object, ['config']))
+  if getv(from_object, ['flipEnabled']) is not None:
+    setv(to_object, ['flip_enabled'], getv(from_object, ['flipEnabled']))
+
+  if getv(from_object, ['autoraterModel']) is not None:
+    setv(to_object, ['autorater_model'], getv(from_object, ['autoraterModel']))
 
   return to_object
 
 
-def _ListTuningJobsConfig_to_mldev(
+def _AutoraterConfig_to_vertex(
     from_object: Union[dict[str, Any], object],
     parent_object: Optional[dict[str, Any]] = None,
 ) -> dict[str, Any]:
   to_object: dict[str, Any] = {}
+  if getv(from_object, ['sampling_count']) is not None:
+    setv(to_object, ['samplingCount'], getv(from_object, ['sampling_count']))
 
-  if getv(from_object, ['page_size']) is not None:
-    setv(
-        parent_object, ['_query', 'pageSize'], getv(from_object, ['page_size'])
-    )
-
-  if getv(from_object, ['page_token']) is not None:
-    setv(
-        parent_object,
-        ['_query', 'pageToken'],
-        getv(from_object, ['page_token']),
-    )
-
-  if getv(from_object, ['filter']) is not None:
-    setv(parent_object, ['_query', 'filter'], getv(from_object, ['filter']))
-
-  return to_object
-
+  if getv(from_object, ['flip_enabled']) is not None:
+    setv(to_object, ['flipEnabled'], getv(from_object, ['flip_enabled']))
 
-def _ListTuningJobsParameters_to_mldev(
-    from_object: Union[dict[str, Any], object],
-    parent_object: Optional[dict[str, Any]] = None,
-) -> dict[str, Any]:
-  to_object: dict[str, Any] = {}
-  if getv(from_object, ['config']) is not None:
-    setv(
-        to_object,
-        ['config'],
-        _ListTuningJobsConfig_to_mldev(
-            getv(from_object, ['config']), to_object
-        ),
-    )
+  if getv(from_object, ['autorater_model']) is not None:
+    setv(to_object, ['autoraterModel'], getv(from_object, ['autorater_model']))
 
   return to_object
 
@@ -95,48 +74,16 @@ def _CancelTuningJobParameters_to_mldev(
   if getv(from_object, ['name']) is not None:
     setv(to_object, ['_url', 'name'], getv(from_object, ['name']))
 
-  if getv(from_object, ['config']) is not None:
-    setv(to_object, ['config'], getv(from_object, ['config']))
-
   return to_object
 
 
-def _TuningExample_to_mldev(
-    from_object: Union[dict[str, Any], object],
-    parent_object: Optional[dict[str, Any]] = None,
-) -> dict[str, Any]:
-  to_object: dict[str, Any] = {}
-  if getv(from_object, ['text_input']) is not None:
-    setv(to_object, ['textInput'], getv(from_object, ['text_input']))
-
-  if getv(from_object, ['output']) is not None:
-    setv(to_object, ['output'], getv(from_object, ['output']))
-
-  return to_object
-
-
-def _TuningDataset_to_mldev(
+def _CancelTuningJobParameters_to_vertex(
     from_object: Union[dict[str, Any], object],
     parent_object: Optional[dict[str, Any]] = None,
 ) -> dict[str, Any]:
   to_object: dict[str, Any] = {}
-  if getv(from_object, ['gcs_uri']) is not None:
-    raise ValueError('gcs_uri parameter is not supported in Gemini API.')
-
-  if getv(from_object, ['vertex_dataset_resource']) is not None:
-    raise ValueError(
-        'vertex_dataset_resource parameter is not supported in Gemini API.'
-    )
-
-  if getv(from_object, ['examples']) is not None:
-    setv(
-        to_object,
-        ['examples', 'examples'],
-        [
-            _TuningExample_to_mldev(item, to_object)
-            for item in getv(from_object, ['examples'])
-        ],
-    )
+  if getv(from_object, ['name']) is not None:
+    setv(to_object, ['_url', 'name'], getv(from_object, ['name']))
 
   return to_object
 
@@ -215,197 +162,161 @@ def _CreateTuningJobConfig_to_mldev(
   return to_object
 
 
-def _CreateTuningJobParametersPrivate_to_mldev(
+def _CreateTuningJobConfig_to_vertex(
     from_object: Union[dict[str, Any], object],
     parent_object: Optional[dict[str, Any]] = None,
 ) -> dict[str, Any]:
   to_object: dict[str, Any] = {}
-  if getv(from_object, ['base_model']) is not None:
-    setv(to_object, ['baseModel'], getv(from_object, ['base_model']))
-
-  if getv(from_object, ['pre_tuned_model']) is not None:
-    setv(to_object, ['preTunedModel'], getv(from_object, ['pre_tuned_model']))
 
-  if getv(from_object, ['training_dataset']) is not None:
+  if getv(from_object, ['validation_dataset']) is not None:
     setv(
-        to_object,
-        ['tuningTask', 'trainingData'],
-        _TuningDataset_to_mldev(
-            getv(from_object, ['training_dataset']), to_object
+        parent_object,
+        ['supervisedTuningSpec'],
+        _TuningValidationDataset_to_vertex(
+            getv(from_object, ['validation_dataset']), to_object
         ),
     )
 
-  if getv(from_object, ['config']) is not None:
+  if getv(from_object, ['tuned_model_display_name']) is not None:
     setv(
-        to_object,
-        ['config'],
-        _CreateTuningJobConfig_to_mldev(
-            getv(from_object, ['config']), to_object
-        ),
+        parent_object,
+        ['tunedModelDisplayName'],
+        getv(from_object, ['tuned_model_display_name']),
     )
 
-  return to_object
-
-
-def _GetTuningJobParameters_to_vertex(
-    from_object: Union[dict[str, Any], object],
-    parent_object: Optional[dict[str, Any]] = None,
-) -> dict[str, Any]:
-  to_object: dict[str, Any] = {}
-  if getv(from_object, ['name']) is not None:
-    setv(to_object, ['_url', 'name'], getv(from_object, ['name']))
-
-  if getv(from_object, ['config']) is not None:
-    setv(to_object, ['config'], getv(from_object, ['config']))
-
-  return to_object
-
+  if getv(from_object, ['description']) is not None:
+    setv(parent_object, ['description'], getv(from_object, ['description']))
 
-def _ListTuningJobsConfig_to_vertex(
-    from_object: Union[dict[str, Any], object],
-    parent_object: Optional[dict[str, Any]] = None,
-) -> dict[str, Any]:
-  to_object: dict[str, Any] = {}
+  if getv(from_object, ['epoch_count']) is not None:
+    setv(
+        parent_object,
+        ['supervisedTuningSpec', 'hyperParameters', 'epochCount'],
+        getv(from_object, ['epoch_count']),
+    )
 
-  if getv(from_object, ['page_size']) is not None:
+  if getv(from_object, ['learning_rate_multiplier']) is not None:
     setv(
-        parent_object, ['_query', 'pageSize'], getv(from_object, ['page_size'])
+        parent_object,
+        ['supervisedTuningSpec', 'hyperParameters', 'learningRateMultiplier'],
+        getv(from_object, ['learning_rate_multiplier']),
     )
 
-  if getv(from_object, ['page_token']) is not None:
+  if getv(from_object, ['export_last_checkpoint_only']) is not None:
     setv(
         parent_object,
-        ['_query', 'pageToken'],
-        getv(from_object, ['page_token']),
+        ['supervisedTuningSpec', 'exportLastCheckpointOnly'],
+        getv(from_object, ['export_last_checkpoint_only']),
     )
 
-  if getv(from_object, ['filter']) is not None:
-    setv(parent_object, ['_query', 'filter'], getv(from_object, ['filter']))
+  if getv(from_object, ['adapter_size']) is not None:
+    setv(
+        parent_object,
+        ['supervisedTuningSpec', 'hyperParameters', 'adapterSize'],
+        getv(from_object, ['adapter_size']),
+    )
 
-  return to_object
+  if getv(from_object, ['batch_size']) is not None:
+    raise ValueError('batch_size parameter is not supported in Vertex AI.')
 
+  if getv(from_object, ['learning_rate']) is not None:
+    raise ValueError('learning_rate parameter is not supported in Vertex AI.')
 
-def _ListTuningJobsParameters_to_vertex(
-    from_object: Union[dict[str, Any], object],
-    parent_object: Optional[dict[str, Any]] = None,
-) -> dict[str, Any]:
-  to_object: dict[str, Any] = {}
-  if getv(from_object, ['config']) is not None:
+  if getv(from_object, ['evaluation_config']) is not None:
     setv(
-        to_object,
-        ['config'],
-        _ListTuningJobsConfig_to_vertex(
-            getv(from_object, ['config']), to_object
+        parent_object,
+        ['supervisedTuningSpec', 'evaluationConfig'],
+        _EvaluationConfig_to_vertex(
+            getv(from_object, ['evaluation_config']), to_object
         ),
     )
 
-  return to_object
-
-
-def _CancelTuningJobParameters_to_vertex(
-    from_object: Union[dict[str, Any], object],
-    parent_object: Optional[dict[str, Any]] = None,
-) -> dict[str, Any]:
-  to_object: dict[str, Any] = {}
-  if getv(from_object, ['name']) is not None:
-    setv(to_object, ['_url', 'name'], getv(from_object, ['name']))
-
-  if getv(from_object, ['config']) is not None:
-    setv(to_object, ['config'], getv(from_object, ['config']))
+  if getv(from_object, ['labels']) is not None:
+    setv(parent_object, ['labels'], getv(from_object, ['labels']))
 
   return to_object
 
 
-def _TuningDataset_to_vertex(
+def _CreateTuningJobParametersPrivate_to_mldev(
     from_object: Union[dict[str, Any], object],
     parent_object: Optional[dict[str, Any]] = None,
 ) -> dict[str, Any]:
   to_object: dict[str, Any] = {}
-  if getv(from_object, ['gcs_uri']) is not None:
-    setv(
-        parent_object,
-        ['supervisedTuningSpec', 'trainingDatasetUri'],
-        getv(from_object, ['gcs_uri']),
-    )
+  if getv(from_object, ['base_model']) is not None:
+    setv(to_object, ['baseModel'], getv(from_object, ['base_model']))
 
-  if getv(from_object, ['vertex_dataset_resource']) is not None:
+  if getv(from_object, ['pre_tuned_model']) is not None:
+    setv(to_object, ['preTunedModel'], getv(from_object, ['pre_tuned_model']))
+
+  if getv(from_object, ['training_dataset']) is not None:
     setv(
-        parent_object,
-        ['supervisedTuningSpec', 'trainingDatasetUri'],
-        getv(from_object, ['vertex_dataset_resource']),
+        to_object,
+        ['tuningTask', 'trainingData'],
+        _TuningDataset_to_mldev(
+            getv(from_object, ['training_dataset']), to_object
+        ),
     )
 
-  if getv(from_object, ['examples']) is not None:
-    raise ValueError('examples parameter is not supported in Vertex AI.')
+  if getv(from_object, ['config']) is not None:
+    _CreateTuningJobConfig_to_mldev(getv(from_object, ['config']), to_object)
 
   return to_object
 
 
-def _TuningValidationDataset_to_vertex(
+def _CreateTuningJobParametersPrivate_to_vertex(
     from_object: Union[dict[str, Any], object],
     parent_object: Optional[dict[str, Any]] = None,
 ) -> dict[str, Any]:
   to_object: dict[str, Any] = {}
-  if getv(from_object, ['gcs_uri']) is not None:
-    setv(to_object, ['validationDatasetUri'], getv(from_object, ['gcs_uri']))
+  if getv(from_object, ['base_model']) is not None:
+    setv(to_object, ['baseModel'], getv(from_object, ['base_model']))
 
-  if getv(from_object, ['vertex_dataset_resource']) is not None:
+  if getv(from_object, ['pre_tuned_model']) is not None:
+    setv(to_object, ['preTunedModel'], getv(from_object, ['pre_tuned_model']))
+
+  if getv(from_object, ['training_dataset']) is not None:
     setv(
-        parent_object,
+        to_object,
         ['supervisedTuningSpec', 'trainingDatasetUri'],
-        getv(from_object, ['vertex_dataset_resource']),
+        _TuningDataset_to_vertex(
+            getv(from_object, ['training_dataset']), to_object
+        ),
     )
 
+  if getv(from_object, ['config']) is not None:
+    _CreateTuningJobConfig_to_vertex(getv(from_object, ['config']), to_object)
+
   return to_object
 
 
-def _GcsDestination_to_vertex(
+def _EvaluationConfig_from_vertex(
     from_object: Union[dict[str, Any], object],
     parent_object: Optional[dict[str, Any]] = None,
 ) -> dict[str, Any]:
   to_object: dict[str, Any] = {}
-  if getv(from_object, ['output_uri_prefix']) is not None:
+  if getv(from_object, ['metrics']) is not None:
+    setv(to_object, ['metrics'], t.t_metrics(getv(from_object, ['metrics'])))
+
+  if getv(from_object, ['outputConfig']) is not None:
     setv(
-        to_object, ['outputUriPrefix'], getv(from_object, ['output_uri_prefix'])
+        to_object,
+        ['output_config'],
+        _OutputConfig_from_vertex(
+            getv(from_object, ['outputConfig']), to_object
+        ),
     )
 
-  return to_object
-
-
-def _OutputConfig_to_vertex(
-    from_object: Union[dict[str, Any], object],
-    parent_object: Optional[dict[str, Any]] = None,
-) -> dict[str, Any]:
-  to_object: dict[str, Any] = {}
-  if getv(from_object, ['gcs_destination']) is not None:
+  if getv(from_object, ['autoraterConfig']) is not None:
     setv(
         to_object,
-        ['gcsDestination'],
-        _GcsDestination_to_vertex(
-            getv(from_object, ['gcs_destination']), to_object
+        ['autorater_config'],
+        _AutoraterConfig_from_vertex(
+            getv(from_object, ['autoraterConfig']), to_object
         ),
     )
 
   return to_object
 
 
-def _AutoraterConfig_to_vertex(
-    from_object: Union[dict[str, Any], object],
-    parent_object: Optional[dict[str, Any]] = None,
-) -> dict[str, Any]:
-  to_object: dict[str, Any] = {}
-  if getv(from_object, ['sampling_count']) is not None:
-    setv(to_object, ['samplingCount'], getv(from_object, ['sampling_count']))
-
-  if getv(from_object, ['flip_enabled']) is not None:
-    setv(to_object, ['flipEnabled'], getv(from_object, ['flip_enabled']))
-
-  if getv(from_object, ['autorater_model']) is not None:
-    setv(to_object, ['autoraterModel'], getv(from_object, ['autorater_model']))
-
-  return to_object
-
-
 def _EvaluationConfig_to_vertex(
     from_object: Union[dict[str, Any], object],
     parent_object: Optional[dict[str, Any]] = None,
@@ -435,212 +346,120 @@ def _EvaluationConfig_to_vertex(
   return to_object
 
 
-def _CreateTuningJobConfig_to_vertex(
+def _GcsDestination_from_vertex(
     from_object: Union[dict[str, Any], object],
     parent_object: Optional[dict[str, Any]] = None,
 ) -> dict[str, Any]:
   to_object: dict[str, Any] = {}
-
-  if getv(from_object, ['validation_dataset']) is not None:
-    setv(
-        parent_object,
-        ['supervisedTuningSpec'],
-        _TuningValidationDataset_to_vertex(
-            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, ['description']) is not None:
-    setv(parent_object, ['description'], getv(from_object, ['description']))
-
-  if getv(from_object, ['epoch_count']) is not None:
-    setv(
-        parent_object,
-        ['supervisedTuningSpec', 'hyperParameters', 'epochCount'],
-        getv(from_object, ['epoch_count']),
-    )
-
-  if getv(from_object, ['learning_rate_multiplier']) is not None:
-    setv(
-        parent_object,
-        ['supervisedTuningSpec', 'hyperParameters', 'learningRateMultiplier'],
-        getv(from_object, ['learning_rate_multiplier']),
-    )
-
-  if getv(from_object, ['export_last_checkpoint_only']) is not None:
-    setv(
-        parent_object,
-        ['supervisedTuningSpec', 'exportLastCheckpointOnly'],
-        getv(from_object, ['export_last_checkpoint_only']),
-    )
-
-  if getv(from_object, ['pre_tuned_model_checkpoint_id']) is not None:
-    setv(
-        to_object,
-        ['preTunedModel', 'checkpointId'],
-        getv(from_object, ['pre_tuned_model_checkpoint_id']),
-    )
-
-  if getv(from_object, ['adapter_size']) is not None:
+  if getv(from_object, ['outputUriPrefix']) is not None:
     setv(
-        parent_object,
-        ['supervisedTuningSpec', 'hyperParameters', 'adapterSize'],
-        getv(from_object, ['adapter_size']),
+        to_object, ['output_uri_prefix'], getv(from_object, ['outputUriPrefix'])
     )
 
-  if getv(from_object, ['batch_size']) is not None:
-    raise ValueError('batch_size parameter is not supported in Vertex AI.')
+  return to_object
 
-  if getv(from_object, ['learning_rate']) is not None:
-    raise ValueError('learning_rate parameter is not supported in Vertex AI.')
 
-  if getv(from_object, ['evaluation_config']) is not None:
+def _GcsDestination_to_vertex(
+    from_object: Union[dict[str, Any], object],
+    parent_object: Optional[dict[str, Any]] = None,
+) -> dict[str, Any]:
+  to_object: dict[str, Any] = {}
+  if getv(from_object, ['output_uri_prefix']) is not None:
     setv(
-        parent_object,
-        ['supervisedTuningSpec', 'evaluationConfig'],
-        _EvaluationConfig_to_vertex(
-            getv(from_object, ['evaluation_config']), to_object
-        ),
+        to_object, ['outputUriPrefix'], getv(from_object, ['output_uri_prefix'])
     )
 
-  if getv(from_object, ['labels']) is not None:
-    setv(parent_object, ['labels'], getv(from_object, ['labels']))
-
   return to_object
 
 
-def _CreateTuningJobParametersPrivate_to_vertex(
+def _GetTuningJobParameters_to_mldev(
     from_object: Union[dict[str, Any], object],
     parent_object: Optional[dict[str, Any]] = None,
 ) -> dict[str, Any]:
   to_object: dict[str, Any] = {}
-  if getv(from_object, ['base_model']) is not None:
-    setv(to_object, ['baseModel'], getv(from_object, ['base_model']))
-
-  if getv(from_object, ['pre_tuned_model']) is not None:
-    setv(to_object, ['preTunedModel'], getv(from_object, ['pre_tuned_model']))
-
-  if getv(from_object, ['training_dataset']) is not None:
-    setv(
-        to_object,
-        ['supervisedTuningSpec', 'trainingDatasetUri'],
-        _TuningDataset_to_vertex(
-            getv(from_object, ['training_dataset']), to_object
-        ),
-    )
-
-  if getv(from_object, ['config']) is not None:
-    setv(
-        to_object,
-        ['config'],
-        _CreateTuningJobConfig_to_vertex(
-            getv(from_object, ['config']), to_object
-        ),
-    )
+  if getv(from_object, ['name']) is not None:
+    setv(to_object, ['_url', 'name'], getv(from_object, ['name']))
 
   return to_object
 
 
-def _TunedModel_from_mldev(
+def _GetTuningJobParameters_to_vertex(
     from_object: Union[dict[str, Any], object],
     parent_object: Optional[dict[str, Any]] = None,
 ) -> dict[str, Any]:
   to_object: dict[str, Any] = {}
   if getv(from_object, ['name']) is not None:
-    setv(to_object, ['model'], getv(from_object, ['name']))
-
-  if getv(from_object, ['name']) is not None:
-    setv(to_object, ['endpoint'], getv(from_object, ['name']))
+    setv(to_object, ['_url', 'name'], getv(from_object, ['name']))
 
   return to_object
 
 
-def _TuningJob_from_mldev(
+def _ListTuningJobsConfig_to_mldev(
     from_object: Union[dict[str, Any], object],
     parent_object: Optional[dict[str, Any]] = None,
 ) -> dict[str, Any]:
   to_object: dict[str, Any] = {}
-  if getv(from_object, ['sdkHttpResponse']) is not None:
-    setv(
-        to_object, ['sdk_http_response'], getv(from_object, ['sdkHttpResponse'])
-    )
 
-  if getv(from_object, ['name']) is not None:
-    setv(to_object, ['name'], getv(from_object, ['name']))
-
-  if getv(from_object, ['state']) is not None:
+  if getv(from_object, ['page_size']) is not None:
     setv(
-        to_object,
-        ['state'],
-        t.t_tuning_job_status(getv(from_object, ['state'])),
+        parent_object, ['_query', 'pageSize'], getv(from_object, ['page_size'])
     )
 
-  if getv(from_object, ['createTime']) is not None:
-    setv(to_object, ['create_time'], getv(from_object, ['createTime']))
-
-  if getv(from_object, ['tuningTask', 'startTime']) is not None:
+  if getv(from_object, ['page_token']) is not None:
     setv(
-        to_object,
-        ['start_time'],
-        getv(from_object, ['tuningTask', 'startTime']),
+        parent_object,
+        ['_query', 'pageToken'],
+        getv(from_object, ['page_token']),
     )
 
-  if getv(from_object, ['tuningTask', 'completeTime']) is not None:
-    setv(
-        to_object,
-        ['end_time'],
-        getv(from_object, ['tuningTask', 'completeTime']),
-    )
+  if getv(from_object, ['filter']) is not None:
+    setv(parent_object, ['_query', 'filter'], getv(from_object, ['filter']))
 
-  if getv(from_object, ['updateTime']) is not None:
-    setv(to_object, ['update_time'], getv(from_object, ['updateTime']))
+  return to_object
 
-  if getv(from_object, ['description']) is not None:
-    setv(to_object, ['description'], getv(from_object, ['description']))
 
-  if getv(from_object, ['baseModel']) is not None:
-    setv(to_object, ['base_model'], getv(from_object, ['baseModel']))
+def _ListTuningJobsConfig_to_vertex(
+    from_object: Union[dict[str, Any], object],
+    parent_object: Optional[dict[str, Any]] = None,
+) -> dict[str, Any]:
+  to_object: dict[str, Any] = {}
 
-  if getv(from_object, ['_self']) is not None:
+  if getv(from_object, ['page_size']) is not None:
     setv(
-        to_object,
-        ['tuned_model'],
-        _TunedModel_from_mldev(getv(from_object, ['_self']), to_object),
+        parent_object, ['_query', 'pageSize'], getv(from_object, ['page_size'])
     )
 
-  if getv(from_object, ['customBaseModel']) is not None:
+  if getv(from_object, ['page_token']) is not None:
     setv(
-        to_object, ['custom_base_model'], getv(from_object, ['customBaseModel'])
+        parent_object,
+        ['_query', 'pageToken'],
+        getv(from_object, ['page_token']),
     )
 
-  if getv(from_object, ['experiment']) is not None:
-    setv(to_object, ['experiment'], getv(from_object, ['experiment']))
+  if getv(from_object, ['filter']) is not None:
+    setv(parent_object, ['_query', 'filter'], getv(from_object, ['filter']))
 
-  if getv(from_object, ['labels']) is not None:
-    setv(to_object, ['labels'], getv(from_object, ['labels']))
+  return to_object
 
-  if getv(from_object, ['outputUri']) is not None:
-    setv(to_object, ['output_uri'], getv(from_object, ['outputUri']))
 
-  if getv(from_object, ['pipelineJob']) is not None:
-    setv(to_object, ['pipeline_job'], getv(from_object, ['pipelineJob']))
+def _ListTuningJobsParameters_to_mldev(
+    from_object: Union[dict[str, Any], object],
+    parent_object: Optional[dict[str, Any]] = None,
+) -> dict[str, Any]:
+  to_object: dict[str, Any] = {}
+  if getv(from_object, ['config']) is not None:
+    _ListTuningJobsConfig_to_mldev(getv(from_object, ['config']), to_object)
 
-  if getv(from_object, ['serviceAccount']) is not None:
-    setv(to_object, ['service_account'], getv(from_object, ['serviceAccount']))
+  return to_object
 
-  if getv(from_object, ['tunedModelDisplayName']) is not None:
-    setv(
-        to_object,
-        ['tuned_model_display_name'],
-        getv(from_object, ['tunedModelDisplayName']),
-    )
+
+def _ListTuningJobsParameters_to_vertex(
+    from_object: Union[dict[str, Any], object],
+    parent_object: Optional[dict[str, Any]] = None,
+) -> dict[str, Any]:
+  to_object: dict[str, Any] = {}
+  if getv(from_object, ['config']) is not None:
+    _ListTuningJobsConfig_to_vertex(getv(from_object, ['config']), to_object)
 
   return to_object
 
@@ -671,7 +490,7 @@ def _ListTuningJobsResponse_from_mldev(
   return to_object
 
 
-def _TuningOperation_from_mldev(
+def _ListTuningJobsResponse_from_vertex(
     from_object: Union[dict[str, Any], object],
     parent_object: Optional[dict[str, Any]] = None,
 ) -> dict[str, Any]:
@@ -681,17 +500,52 @@ def _TuningOperation_from_mldev(
         to_object, ['sdk_http_response'], getv(from_object, ['sdkHttpResponse'])
     )
 
-  if getv(from_object, ['name']) is not None:
-    setv(to_object, ['name'], getv(from_object, ['name']))
+  if getv(from_object, ['nextPageToken']) is not None:
+    setv(to_object, ['next_page_token'], getv(from_object, ['nextPageToken']))
 
-  if getv(from_object, ['metadata']) is not None:
-    setv(to_object, ['metadata'], getv(from_object, ['metadata']))
+  if getv(from_object, ['tuningJobs']) is not None:
+    setv(
+        to_object,
+        ['tuning_jobs'],
+        [
+            _TuningJob_from_vertex(item, to_object)
+            for item in getv(from_object, ['tuningJobs'])
+        ],
+    )
 
-  if getv(from_object, ['done']) is not None:
-    setv(to_object, ['done'], getv(from_object, ['done']))
+  return to_object
 
-  if getv(from_object, ['error']) is not None:
-    setv(to_object, ['error'], getv(from_object, ['error']))
+
+def _OutputConfig_from_vertex(
+    from_object: Union[dict[str, Any], object],
+    parent_object: Optional[dict[str, Any]] = None,
+) -> dict[str, Any]:
+  to_object: dict[str, Any] = {}
+  if getv(from_object, ['gcsDestination']) is not None:
+    setv(
+        to_object,
+        ['gcs_destination'],
+        _GcsDestination_from_vertex(
+            getv(from_object, ['gcsDestination']), to_object
+        ),
+    )
+
+  return to_object
+
+
+def _OutputConfig_to_vertex(
+    from_object: Union[dict[str, Any], object],
+    parent_object: Optional[dict[str, Any]] = None,
+) -> dict[str, Any]:
+  to_object: dict[str, Any] = {}
+  if getv(from_object, ['gcs_destination']) is not None:
+    setv(
+        to_object,
+        ['gcsDestination'],
+        _GcsDestination_to_vertex(
+            getv(from_object, ['gcs_destination']), to_object
+        ),
+    )
 
   return to_object
 
@@ -716,6 +570,20 @@ def _TunedModelCheckpoint_from_vertex(
   return to_object
 
 
+def _TunedModel_from_mldev(
+    from_object: Union[dict[str, Any], object],
+    parent_object: Optional[dict[str, Any]] = None,
+) -> dict[str, Any]:
+  to_object: dict[str, Any] = {}
+  if getv(from_object, ['name']) is not None:
+    setv(to_object, ['model'], getv(from_object, ['name']))
+
+  if getv(from_object, ['name']) is not None:
+    setv(to_object, ['endpoint'], getv(from_object, ['name']))
+
+  return to_object
+
+
 def _TunedModel_from_vertex(
     from_object: Union[dict[str, Any], object],
     parent_object: Optional[dict[str, Any]] = None,
@@ -740,77 +608,149 @@ def _TunedModel_from_vertex(
   return to_object
 
 
-def _GcsDestination_from_vertex(
+def _TuningDataset_to_mldev(
     from_object: Union[dict[str, Any], object],
     parent_object: Optional[dict[str, Any]] = None,
 ) -> dict[str, Any]:
   to_object: dict[str, Any] = {}
-  if getv(from_object, ['outputUriPrefix']) is not None:
+  if getv(from_object, ['gcs_uri']) is not None:
+    raise ValueError('gcs_uri parameter is not supported in Gemini API.')
+
+  if getv(from_object, ['vertex_dataset_resource']) is not None:
+    raise ValueError(
+        'vertex_dataset_resource parameter is not supported in Gemini API.'
+    )
+
+  if getv(from_object, ['examples']) is not None:
     setv(
-        to_object, ['output_uri_prefix'], getv(from_object, ['outputUriPrefix'])
+        to_object,
+        ['examples', 'examples'],
+        [
+            _TuningExample_to_mldev(item, to_object)
+            for item in getv(from_object, ['examples'])
+        ],
     )
 
   return to_object
 
 
-def _OutputConfig_from_vertex(
+def _TuningDataset_to_vertex(
     from_object: Union[dict[str, Any], object],
     parent_object: Optional[dict[str, Any]] = None,
 ) -> dict[str, Any]:
   to_object: dict[str, Any] = {}
-  if getv(from_object, ['gcsDestination']) is not None:
+  if getv(from_object, ['gcs_uri']) is not None:
     setv(
-        to_object,
-        ['gcs_destination'],
-        _GcsDestination_from_vertex(
-            getv(from_object, ['gcsDestination']), to_object
-        ),
+        parent_object,
+        ['supervisedTuningSpec', 'trainingDatasetUri'],
+        getv(from_object, ['gcs_uri']),
+    )
+
+  if getv(from_object, ['vertex_dataset_resource']) is not None:
+    setv(
+        parent_object,
+        ['supervisedTuningSpec', 'trainingDatasetUri'],
+        getv(from_object, ['vertex_dataset_resource']),
     )
 
+  if getv(from_object, ['examples']) is not None:
+    raise ValueError('examples parameter is not supported in Vertex AI.')
+
   return to_object
 
 
-def _AutoraterConfig_from_vertex(
+def _TuningExample_to_mldev(
     from_object: Union[dict[str, Any], object],
     parent_object: Optional[dict[str, Any]] = None,
 ) -> dict[str, Any]:
   to_object: dict[str, Any] = {}
-  if getv(from_object, ['samplingCount']) is not None:
-    setv(to_object, ['sampling_count'], getv(from_object, ['samplingCount']))
-
-  if getv(from_object, ['flipEnabled']) is not None:
-    setv(to_object, ['flip_enabled'], getv(from_object, ['flipEnabled']))
+  if getv(from_object, ['text_input']) is not None:
+    setv(to_object, ['textInput'], getv(from_object, ['text_input']))
 
-  if getv(from_object, ['autoraterModel']) is not None:
-    setv(to_object, ['autorater_model'], getv(from_object, ['autoraterModel']))
+  if getv(from_object, ['output']) is not None:
+    setv(to_object, ['output'], getv(from_object, ['output']))
 
   return to_object
 
 
-def _EvaluationConfig_from_vertex(
+def _TuningJob_from_mldev(
     from_object: Union[dict[str, Any], object],
     parent_object: Optional[dict[str, Any]] = None,
 ) -> dict[str, Any]:
   to_object: dict[str, Any] = {}
-  if getv(from_object, ['metrics']) is not None:
-    setv(to_object, ['metrics'], t.t_metrics(getv(from_object, ['metrics'])))
+  if getv(from_object, ['sdkHttpResponse']) is not None:
+    setv(
+        to_object, ['sdk_http_response'], getv(from_object, ['sdkHttpResponse'])
+    )
 
-  if getv(from_object, ['outputConfig']) is not None:
+  if getv(from_object, ['name']) is not None:
+    setv(to_object, ['name'], getv(from_object, ['name']))
+
+  if getv(from_object, ['state']) is not None:
     setv(
         to_object,
-        ['output_config'],
-        _OutputConfig_from_vertex(
-            getv(from_object, ['outputConfig']), to_object
-        ),
+        ['state'],
+        t.t_tuning_job_status(getv(from_object, ['state'])),
     )
 
-  if getv(from_object, ['autoraterConfig']) is not None:
+  if getv(from_object, ['createTime']) is not None:
+    setv(to_object, ['create_time'], getv(from_object, ['createTime']))
+
+  if getv(from_object, ['tuningTask', 'startTime']) is not None:
     setv(
         to_object,
-        ['autorater_config'],
-        _AutoraterConfig_from_vertex(
-            getv(from_object, ['autoraterConfig']), to_object
-        ),
+        ['start_time'],
+        getv(from_object, ['tuningTask', 'startTime']),
+    )
+
+  if getv(from_object, ['tuningTask', 'completeTime']) is not None:
+    setv(
+        to_object,
+        ['end_time'],
+        getv(from_object, ['tuningTask', 'completeTime']),
+    )
+
+  if getv(from_object, ['updateTime']) is not None:
+    setv(to_object, ['update_time'], getv(from_object, ['updateTime']))
+
+  if getv(from_object, ['description']) is not None:
+    setv(to_object, ['description'], getv(from_object, ['description']))
+
+  if getv(from_object, ['baseModel']) is not None:
+    setv(to_object, ['base_model'], getv(from_object, ['baseModel']))
+
+  if getv(from_object, ['_self']) is not None:
+    setv(
+        to_object,
+        ['tuned_model'],
+        _TunedModel_from_mldev(getv(from_object, ['_self']), to_object),
+    )
+
+  if getv(from_object, ['customBaseModel']) is not None:
+    setv(
+        to_object, ['custom_base_model'], getv(from_object, ['customBaseModel'])
+    )
+
+  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, ['outputUri']) is not None:
+    setv(to_object, ['output_uri'], getv(from_object, ['outputUri']))
+
+  if getv(from_object, ['pipelineJob']) is not None:
+    setv(to_object, ['pipeline_job'], getv(from_object, ['pipelineJob']))
+
+  if getv(from_object, ['serviceAccount']) is not None:
+    setv(to_object, ['service_account'], getv(from_object, ['serviceAccount']))
+
+  if getv(from_object, ['tunedModelDisplayName']) is not None:
+    setv(
+        to_object,
+        ['tuned_model_display_name'],
+        getv(from_object, ['tunedModelDisplayName']),
     )
 
   return to_object
@@ -928,7 +868,7 @@ def _TuningJob_from_vertex(
   return to_object
 
 
-def _ListTuningJobsResponse_from_vertex(
+def _TuningOperation_from_mldev(
     from_object: Union[dict[str, Any], object],
     parent_object: Optional[dict[str, Any]] = None,
 ) -> dict[str, Any]:
@@ -938,17 +878,34 @@ def _ListTuningJobsResponse_from_vertex(
         to_object, ['sdk_http_response'], getv(from_object, ['sdkHttpResponse'])
     )
 
-  if getv(from_object, ['nextPageToken']) is not None:
-    setv(to_object, ['next_page_token'], getv(from_object, ['nextPageToken']))
+  if getv(from_object, ['name']) is not None:
+    setv(to_object, ['name'], getv(from_object, ['name']))
 
-  if getv(from_object, ['tuningJobs']) is not None:
+  if getv(from_object, ['metadata']) is not None:
+    setv(to_object, ['metadata'], getv(from_object, ['metadata']))
+
+  if getv(from_object, ['done']) is not None:
+    setv(to_object, ['done'], getv(from_object, ['done']))
+
+  if getv(from_object, ['error']) is not None:
+    setv(to_object, ['error'], getv(from_object, ['error']))
+
+  return to_object
+
+
+def _TuningValidationDataset_to_vertex(
+    from_object: Union[dict[str, Any], object],
+    parent_object: Optional[dict[str, Any]] = None,
+) -> dict[str, Any]:
+  to_object: dict[str, Any] = {}
+  if getv(from_object, ['gcs_uri']) is not None:
+    setv(to_object, ['validationDatasetUri'], getv(from_object, ['gcs_uri']))
+
+  if getv(from_object, ['vertex_dataset_resource']) is not None:
     setv(
-        to_object,
-        ['tuning_jobs'],
-        [
-            _TuningJob_from_vertex(item, to_object)
-            for item in getv(from_object, ['tuningJobs'])
-        ],
+        parent_object,
+        ['supervisedTuningSpec', 'trainingDatasetUri'],
+        getv(from_object, ['vertex_dataset_resource']),
     )
 
   return to_object
@@ -1334,7 +1291,12 @@ class Tunings(_api_module.BaseModule):
   ) -> types.TuningJob:
     if self._api_client.vertexai:
       if base_model.startswith('projects/'):  # Pre-tuned model
-        pre_tuned_model = types.PreTunedModel(tuned_model_name=base_model)
+        checkpoint_id = None
+        if config:
+          checkpoint_id = getattr(config, 'pre_tuned_model_checkpoint_id', None)
+        pre_tuned_model = types.PreTunedModel(
+            tuned_model_name=base_model, checkpoint_id=checkpoint_id
+        )
         tuning_job = self._tune(
             pre_tuned_model=pre_tuned_model,
             training_dataset=training_dataset,
@@ -1784,7 +1746,12 @@ class AsyncTunings(_api_module.BaseModule):
   ) -> types.TuningJob:
     if self._api_client.vertexai:
       if base_model.startswith('projects/'):  # Pre-tuned model
-        pre_tuned_model = types.PreTunedModel(tuned_model_name=base_model)
+        checkpoint_id = None
+        if config:
+          checkpoint_id = getattr(config, 'pre_tuned_model_checkpoint_id', None)
+        pre_tuned_model = types.PreTunedModel(
+            tuned_model_name=base_model, checkpoint_id=checkpoint_id
+        )
 
         tuning_job = await self._tune(
             pre_tuned_model=pre_tuned_model,