google-genai 0.3.0__py3-none-any.whl → 0.5.0__py3-none-any.whl

google/genai/tunings.py

@@ -13,6 +13,8 @@
 # limitations under the License.
 #
 
+# Code generated by the Google Gen AI SDK generator DO NOT EDIT.
+
 from typing import Optional, Union
 from urllib.parse import urlencode
 from . import _common
@@ -195,10 +197,10 @@ def _TuningExample_to_vertex(
     parent_object: dict = None,
 ) -> dict:
   to_object = {}
-  if getv(from_object, ['text_input']):
+  if getv(from_object, ['text_input']) is not None:
     raise ValueError('text_input parameter is not supported in Vertex AI.')
 
-  if getv(from_object, ['output']):
+  if getv(from_object, ['output']) is not None:
     raise ValueError('output parameter is not supported in Vertex AI.')
 
   return to_object
@@ -210,7 +212,7 @@ def _TuningDataset_to_mldev(
     parent_object: dict = None,
 ) -> dict:
   to_object = {}
-  if getv(from_object, ['gcs_uri']):
+  if getv(from_object, ['gcs_uri']) is not None:
     raise ValueError('gcs_uri parameter is not supported in Google AI.')
 
   if getv(from_object, ['examples']) is not None:
@@ -239,7 +241,7 @@ def _TuningDataset_to_vertex(
         getv(from_object, ['gcs_uri']),
     )
 
-  if getv(from_object, ['examples']):
+  if getv(from_object, ['examples']) is not None:
     raise ValueError('examples parameter is not supported in Vertex AI.')
 
   return to_object
@@ -251,7 +253,7 @@ def _TuningValidationDataset_to_mldev(
     parent_object: dict = None,
 ) -> dict:
   to_object = {}
-  if getv(from_object, ['gcs_uri']):
+  if getv(from_object, ['gcs_uri']) is not None:
     raise ValueError('gcs_uri parameter is not supported in Google AI.')
 
   return to_object
@@ -278,7 +280,7 @@ def _CreateTuningJobConfig_to_mldev(
   if getv(from_object, ['http_options']) is not None:
     setv(to_object, ['httpOptions'], getv(from_object, ['http_options']))
 
-  if getv(from_object, ['validation_dataset']):
+  if getv(from_object, ['validation_dataset']) is not None:
     raise ValueError(
         'validation_dataset parameter is not supported in Google AI.'
     )
@@ -290,7 +292,7 @@ def _CreateTuningJobConfig_to_mldev(
         getv(from_object, ['tuned_model_display_name']),
     )
 
-  if getv(from_object, ['description']):
+  if getv(from_object, ['description']) is not None:
     raise ValueError('description parameter is not supported in Google AI.')
 
   if getv(from_object, ['epoch_count']) is not None:
@@ -307,7 +309,7 @@ def _CreateTuningJobConfig_to_mldev(
         getv(from_object, ['learning_rate_multiplier']),
     )
 
-  if getv(from_object, ['adapter_size']):
+  if getv(from_object, ['adapter_size']) is not None:
     raise ValueError('adapter_size parameter is not supported in Google AI.')
 
   if getv(from_object, ['batch_size']) is not None:
@@ -376,10 +378,10 @@ def _CreateTuningJobConfig_to_vertex(
         getv(from_object, ['adapter_size']),
     )
 
-  if getv(from_object, ['batch_size']):
+  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']):
+  if getv(from_object, ['learning_rate']) is not None:
     raise ValueError('learning_rate parameter is not supported in Vertex AI.')
 
   return to_object
@@ -451,7 +453,7 @@ def _DistillationDataset_to_mldev(
     parent_object: dict = None,
 ) -> dict:
   to_object = {}
-  if getv(from_object, ['gcs_uri']):
+  if getv(from_object, ['gcs_uri']) is not None:
     raise ValueError('gcs_uri parameter is not supported in Google AI.')
 
   return to_object
@@ -479,7 +481,7 @@ def _DistillationValidationDataset_to_mldev(
     parent_object: dict = None,
 ) -> dict:
   to_object = {}
-  if getv(from_object, ['gcs_uri']):
+  if getv(from_object, ['gcs_uri']) is not None:
     raise ValueError('gcs_uri parameter is not supported in Google AI.')
 
   return to_object
@@ -506,7 +508,7 @@ def _CreateDistillationJobConfig_to_mldev(
   if getv(from_object, ['http_options']) is not None:
     setv(to_object, ['httpOptions'], getv(from_object, ['http_options']))
 
-  if getv(from_object, ['validation_dataset']):
+  if getv(from_object, ['validation_dataset']) is not None:
     raise ValueError(
         'validation_dataset parameter is not supported in Google AI.'
     )
@@ -532,10 +534,10 @@ def _CreateDistillationJobConfig_to_mldev(
         getv(from_object, ['learning_rate_multiplier']),
     )
 
-  if getv(from_object, ['adapter_size']):
+  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']):
+  if getv(from_object, ['pipeline_root_directory']) is not None:
     raise ValueError(
         'pipeline_root_directory parameter is not supported in Google AI.'
     )
@@ -605,10 +607,10 @@ def _CreateDistillationJobParameters_to_mldev(
     parent_object: dict = None,
 ) -> dict:
   to_object = {}
-  if getv(from_object, ['student_model']):
+  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']):
+  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:
@@ -970,14 +972,14 @@ class Tunings(_common.BaseModule):
         config=config,
     )
 
-    if self.api_client.vertexai:
+    if self._api_client.vertexai:
       request_dict = _GetTuningJobParameters_to_vertex(
-          self.api_client, parameter_model
+          self._api_client, parameter_model
       )
       path = '{name}'.format_map(request_dict.get('_url'))
     else:
       request_dict = _GetTuningJobParameters_to_mldev(
-          self.api_client, parameter_model
+          self._api_client, parameter_model
       )
       path = '{name}'.format_map(request_dict.get('_url'))
     query_params = request_dict.get('_query')
@@ -989,19 +991,19 @@ class Tunings(_common.BaseModule):
     request_dict = _common.convert_to_dict(request_dict)
     request_dict = _common.apply_base64_encoding(request_dict)
 
-    response_dict = self.api_client.request(
+    response_dict = self._api_client.request(
         'get', path, request_dict, http_options
     )
 
-    if self.api_client.vertexai:
-      response_dict = _TuningJob_from_vertex(self.api_client, response_dict)
+    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)
+      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)
+    self._api_client._verify_response(return_value)
     return return_value
 
   def _list(
@@ -1020,14 +1022,14 @@ class Tunings(_common.BaseModule):
         config=config,
     )
 
-    if self.api_client.vertexai:
+    if self._api_client.vertexai:
       request_dict = _ListTuningJobsParameters_to_vertex(
-          self.api_client, parameter_model
+          self._api_client, parameter_model
       )
       path = 'tuningJobs'.format_map(request_dict.get('_url'))
     else:
       request_dict = _ListTuningJobsParameters_to_mldev(
-          self.api_client, parameter_model
+          self._api_client, parameter_model
       )
       path = 'tunedModels'.format_map(request_dict.get('_url'))
     query_params = request_dict.get('_query')
@@ -1039,23 +1041,23 @@ class Tunings(_common.BaseModule):
     request_dict = _common.convert_to_dict(request_dict)
     request_dict = _common.apply_base64_encoding(request_dict)
 
-    response_dict = self.api_client.request(
+    response_dict = self._api_client.request(
         'get', path, request_dict, http_options
     )
 
-    if self.api_client.vertexai:
+    if self._api_client.vertexai:
       response_dict = _ListTuningJobsResponse_from_vertex(
-          self.api_client, response_dict
+          self._api_client, response_dict
       )
     else:
       response_dict = _ListTuningJobsResponse_from_mldev(
-          self.api_client, response_dict
+          self._api_client, response_dict
       )
 
     return_value = types.ListTuningJobsResponse._from_response(
         response_dict, parameter_model
     )
-    self.api_client._verify_response(return_value)
+    self._api_client._verify_response(return_value)
     return return_value
 
   def _tune(
@@ -1082,14 +1084,14 @@ class Tunings(_common.BaseModule):
         config=config,
     )
 
-    if self.api_client.vertexai:
+    if self._api_client.vertexai:
       request_dict = _CreateTuningJobParameters_to_vertex(
-          self.api_client, parameter_model
+          self._api_client, parameter_model
       )
       path = 'tuningJobs'.format_map(request_dict.get('_url'))
     else:
       request_dict = _CreateTuningJobParameters_to_mldev(
-          self.api_client, parameter_model
+          self._api_client, parameter_model
       )
       path = 'tunedModels'.format_map(request_dict.get('_url'))
     query_params = request_dict.get('_query')
@@ -1101,23 +1103,23 @@ class Tunings(_common.BaseModule):
     request_dict = _common.convert_to_dict(request_dict)
     request_dict = _common.apply_base64_encoding(request_dict)
 
-    response_dict = self.api_client.request(
+    response_dict = self._api_client.request(
         'post', path, request_dict, http_options
     )
 
-    if self.api_client.vertexai:
+    if self._api_client.vertexai:
       response_dict = _TuningJobOrOperation_from_vertex(
-          self.api_client, response_dict
+          self._api_client, response_dict
       )
     else:
       response_dict = _TuningJobOrOperation_from_mldev(
-          self.api_client, response_dict
+          self._api_client, response_dict
       )
 
     return_value = types.TuningJobOrOperation._from_response(
         response_dict, parameter_model
     ).tuning_job
-    self.api_client._verify_response(return_value)
+    self._api_client._verify_response(return_value)
     return return_value
 
   def distill(
@@ -1147,11 +1149,11 @@ class Tunings(_common.BaseModule):
         config=config,
     )
 
-    if not self.api_client.vertexai:
+    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
+          self._api_client, parameter_model
       )
       path = 'tuningJobs'.format_map(request_dict.get('_url'))
 
@@ -1164,19 +1166,19 @@ class Tunings(_common.BaseModule):
     request_dict = _common.convert_to_dict(request_dict)
     request_dict = _common.apply_base64_encoding(request_dict)
 
-    response_dict = self.api_client.request(
+    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)
+    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)
+      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)
+    self._api_client._verify_response(return_value)
     return return_value
 
   def list(
@@ -1196,10 +1198,10 @@ class Tunings(_common.BaseModule):
       config: Optional[types.GetTuningJobConfigOrDict] = None,
   ) -> types.TuningJob:
     job = self._get(name=name, config=config)
-    if job.experiment and self.api_client.vertexai:
+    if job.experiment and self._api_client.vertexai:
       _IpythonUtils.display_experiment_button(
           experiment=job.experiment,
-          project=self.api_client.project,
+          project=self._api_client.project,
       )
     return job
 
@@ -1215,7 +1217,7 @@ class Tunings(_common.BaseModule):
         training_dataset=training_dataset,
         config=config,
     )
-    if result.name and self.api_client.vertexai:
+    if result.name and self._api_client.vertexai:
       _IpythonUtils.display_model_tuning_button(tuning_job_resource=result.name)
     return result
 
@@ -1242,14 +1244,14 @@ class AsyncTunings(_common.BaseModule):
         config=config,
     )
 
-    if self.api_client.vertexai:
+    if self._api_client.vertexai:
       request_dict = _GetTuningJobParameters_to_vertex(
-          self.api_client, parameter_model
+          self._api_client, parameter_model
       )
       path = '{name}'.format_map(request_dict.get('_url'))
     else:
       request_dict = _GetTuningJobParameters_to_mldev(
-          self.api_client, parameter_model
+          self._api_client, parameter_model
       )
       path = '{name}'.format_map(request_dict.get('_url'))
     query_params = request_dict.get('_query')
@@ -1261,19 +1263,19 @@ class AsyncTunings(_common.BaseModule):
     request_dict = _common.convert_to_dict(request_dict)
     request_dict = _common.apply_base64_encoding(request_dict)
 
-    response_dict = await self.api_client.async_request(
+    response_dict = await self._api_client.async_request(
         'get', path, request_dict, http_options
     )
 
-    if self.api_client.vertexai:
-      response_dict = _TuningJob_from_vertex(self.api_client, response_dict)
+    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)
+      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)
+    self._api_client._verify_response(return_value)
     return return_value
 
   async def _list(
@@ -1292,14 +1294,14 @@ class AsyncTunings(_common.BaseModule):
         config=config,
     )
 
-    if self.api_client.vertexai:
+    if self._api_client.vertexai:
       request_dict = _ListTuningJobsParameters_to_vertex(
-          self.api_client, parameter_model
+          self._api_client, parameter_model
       )
       path = 'tuningJobs'.format_map(request_dict.get('_url'))
     else:
       request_dict = _ListTuningJobsParameters_to_mldev(
-          self.api_client, parameter_model
+          self._api_client, parameter_model
       )
       path = 'tunedModels'.format_map(request_dict.get('_url'))
     query_params = request_dict.get('_query')
@@ -1311,23 +1313,23 @@ class AsyncTunings(_common.BaseModule):
     request_dict = _common.convert_to_dict(request_dict)
     request_dict = _common.apply_base64_encoding(request_dict)
 
-    response_dict = await self.api_client.async_request(
+    response_dict = await self._api_client.async_request(
         'get', path, request_dict, http_options
     )
 
-    if self.api_client.vertexai:
+    if self._api_client.vertexai:
       response_dict = _ListTuningJobsResponse_from_vertex(
-          self.api_client, response_dict
+          self._api_client, response_dict
       )
     else:
       response_dict = _ListTuningJobsResponse_from_mldev(
-          self.api_client, response_dict
+          self._api_client, response_dict
       )
 
     return_value = types.ListTuningJobsResponse._from_response(
         response_dict, parameter_model
     )
-    self.api_client._verify_response(return_value)
+    self._api_client._verify_response(return_value)
     return return_value
 
   async def _tune(
@@ -1354,14 +1356,14 @@ class AsyncTunings(_common.BaseModule):
         config=config,
     )
 
-    if self.api_client.vertexai:
+    if self._api_client.vertexai:
       request_dict = _CreateTuningJobParameters_to_vertex(
-          self.api_client, parameter_model
+          self._api_client, parameter_model
       )
       path = 'tuningJobs'.format_map(request_dict.get('_url'))
     else:
       request_dict = _CreateTuningJobParameters_to_mldev(
-          self.api_client, parameter_model
+          self._api_client, parameter_model
       )
       path = 'tunedModels'.format_map(request_dict.get('_url'))
     query_params = request_dict.get('_query')
@@ -1373,23 +1375,23 @@ class AsyncTunings(_common.BaseModule):
     request_dict = _common.convert_to_dict(request_dict)
     request_dict = _common.apply_base64_encoding(request_dict)
 
-    response_dict = await self.api_client.async_request(
+    response_dict = await self._api_client.async_request(
         'post', path, request_dict, http_options
     )
 
-    if self.api_client.vertexai:
+    if self._api_client.vertexai:
       response_dict = _TuningJobOrOperation_from_vertex(
-          self.api_client, response_dict
+          self._api_client, response_dict
       )
     else:
       response_dict = _TuningJobOrOperation_from_mldev(
-          self.api_client, response_dict
+          self._api_client, response_dict
       )
 
     return_value = types.TuningJobOrOperation._from_response(
         response_dict, parameter_model
     ).tuning_job
-    self.api_client._verify_response(return_value)
+    self._api_client._verify_response(return_value)
     return return_value
 
   async def distill(
@@ -1419,11 +1421,11 @@ class AsyncTunings(_common.BaseModule):
         config=config,
     )
 
-    if not self.api_client.vertexai:
+    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
+          self._api_client, parameter_model
       )
       path = 'tuningJobs'.format_map(request_dict.get('_url'))
 
@@ -1436,19 +1438,19 @@ class AsyncTunings(_common.BaseModule):
     request_dict = _common.convert_to_dict(request_dict)
     request_dict = _common.apply_base64_encoding(request_dict)
 
-    response_dict = await self.api_client.async_request(
+    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)
+    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)
+      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)
+    self._api_client._verify_response(return_value)
     return return_value
 
   async def list(
@@ -1468,10 +1470,10 @@ class AsyncTunings(_common.BaseModule):
       config: Optional[types.GetTuningJobConfigOrDict] = None,
   ) -> types.TuningJob:
     job = await self._get(name=name, config=config)
-    if job.experiment and self.api_client.vertexai:
+    if job.experiment and self._api_client.vertexai:
       _IpythonUtils.display_experiment_button(
           experiment=job.experiment,
-          project=self.api_client.project,
+          project=self._api_client.project,
       )
     return job
 
@@ -1487,7 +1489,7 @@ class AsyncTunings(_common.BaseModule):
         training_dataset=training_dataset,
         config=config,
     )
-    if result.name and self.api_client.vertexai:
+    if result.name and self._api_client.vertexai:
       _IpythonUtils.display_model_tuning_button(tuning_job_resource=result.name)
     return result