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

google/genai/models.py

@@ -32,7 +32,7 @@ def _Part_to_mldev(
     parent_object: dict = None,
 ) -> dict:
   to_object = {}
-  if getv(from_object, ['video_metadata']):
+  if getv(from_object, ['video_metadata']) is not None:
     raise ValueError('video_metadata parameter is not supported in Google AI.')
 
   if getv(from_object, ['thought']) is not None:
@@ -164,51 +164,51 @@ def _Schema_to_mldev(
     parent_object: dict = None,
 ) -> dict:
   to_object = {}
-  if getv(from_object, ['min_items']):
+  if getv(from_object, ['min_items']) is not None:
     raise ValueError('min_items parameter is not supported in Google AI.')
 
-  if getv(from_object, ['example']):
+  if getv(from_object, ['example']) is not None:
     raise ValueError('example parameter is not supported in Google AI.')
 
-  if getv(from_object, ['property_ordering']):
+  if getv(from_object, ['property_ordering']) is not None:
     raise ValueError(
         'property_ordering parameter is not supported in Google AI.'
     )
 
-  if getv(from_object, ['pattern']):
+  if getv(from_object, ['pattern']) is not None:
     raise ValueError('pattern parameter is not supported in Google AI.')
 
-  if getv(from_object, ['minimum']):
+  if getv(from_object, ['minimum']) is not None:
     raise ValueError('minimum parameter is not supported in Google AI.')
 
-  if getv(from_object, ['default']):
+  if getv(from_object, ['default']) is not None:
     raise ValueError('default parameter is not supported in Google AI.')
 
-  if getv(from_object, ['any_of']):
+  if getv(from_object, ['any_of']) is not None:
     raise ValueError('any_of parameter is not supported in Google AI.')
 
-  if getv(from_object, ['max_length']):
+  if getv(from_object, ['max_length']) is not None:
     raise ValueError('max_length parameter is not supported in Google AI.')
 
-  if getv(from_object, ['title']):
+  if getv(from_object, ['title']) is not None:
     raise ValueError('title parameter is not supported in Google AI.')
 
-  if getv(from_object, ['min_length']):
+  if getv(from_object, ['min_length']) is not None:
     raise ValueError('min_length parameter is not supported in Google AI.')
 
-  if getv(from_object, ['min_properties']):
+  if getv(from_object, ['min_properties']) is not None:
     raise ValueError('min_properties parameter is not supported in Google AI.')
 
-  if getv(from_object, ['max_items']):
+  if getv(from_object, ['max_items']) is not None:
     raise ValueError('max_items parameter is not supported in Google AI.')
 
-  if getv(from_object, ['maximum']):
+  if getv(from_object, ['maximum']) is not None:
     raise ValueError('maximum parameter is not supported in Google AI.')
 
-  if getv(from_object, ['nullable']):
+  if getv(from_object, ['nullable']) is not None:
     raise ValueError('nullable parameter is not supported in Google AI.')
 
-  if getv(from_object, ['max_properties']):
+  if getv(from_object, ['max_properties']) is not None:
     raise ValueError('max_properties parameter is not supported in Google AI.')
 
   if getv(from_object, ['type']) is not None:
@@ -320,7 +320,7 @@ def _SafetySetting_to_mldev(
     parent_object: dict = None,
 ) -> dict:
   to_object = {}
-  if getv(from_object, ['method']):
+  if getv(from_object, ['method']) is not None:
     raise ValueError('method parameter is not supported in Google AI.')
 
   if getv(from_object, ['category']) is not None:
@@ -356,7 +356,7 @@ def _FunctionDeclaration_to_mldev(
     parent_object: dict = None,
 ) -> dict:
   to_object = {}
-  if getv(from_object, ['response']):
+  if getv(from_object, ['response']) is not None:
     raise ValueError('response parameter is not supported in Google AI.')
 
   if getv(from_object, ['description']) is not None:
@@ -512,7 +512,7 @@ def _Tool_to_mldev(
         ],
     )
 
-  if getv(from_object, ['retrieval']):
+  if getv(from_object, ['retrieval']) is not None:
     raise ValueError('retrieval parameter is not supported in Google AI.')
 
   if getv(from_object, ['google_search']) is not None:
@@ -797,13 +797,15 @@ def _GenerateContentConfig_to_mldev(
   if getv(from_object, ['stop_sequences']) is not None:
     setv(to_object, ['stopSequences'], getv(from_object, ['stop_sequences']))
 
-  if getv(from_object, ['response_logprobs']):
-    raise ValueError(
-        'response_logprobs parameter is not supported in Google AI.'
+  if getv(from_object, ['response_logprobs']) is not None:
+    setv(
+        to_object,
+        ['responseLogprobs'],
+        getv(from_object, ['response_logprobs']),
     )
 
-  if getv(from_object, ['logprobs']):
-    raise ValueError('logprobs parameter is not supported in Google AI.')
+  if getv(from_object, ['logprobs']) is not None:
+    setv(to_object, ['logprobs'], getv(from_object, ['logprobs']))
 
   if getv(from_object, ['presence_penalty']) is not None:
     setv(
@@ -838,7 +840,7 @@ def _GenerateContentConfig_to_mldev(
         ),
     )
 
-  if getv(from_object, ['routing_config']):
+  if getv(from_object, ['routing_config']) is not None:
     raise ValueError('routing_config parameter is not supported in Google AI.')
 
   if getv(from_object, ['safety_settings']) is not None:
@@ -886,7 +888,7 @@ def _GenerateContentConfig_to_mldev(
         getv(from_object, ['response_modalities']),
     )
 
-  if getv(from_object, ['media_resolution']):
+  if getv(from_object, ['media_resolution']) is not None:
     raise ValueError(
         'media_resolution parameter is not supported in Google AI.'
     )
@@ -1152,10 +1154,10 @@ def _EmbedContentConfig_to_mldev(
         getv(from_object, ['output_dimensionality']),
     )
 
-  if getv(from_object, ['mime_type']):
+  if getv(from_object, ['mime_type']) is not None:
     raise ValueError('mime_type parameter is not supported in Google AI.')
 
-  if getv(from_object, ['auto_truncate']):
+  if getv(from_object, ['auto_truncate']) is not None:
     raise ValueError('auto_truncate parameter is not supported in Google AI.')
 
   return to_object
@@ -1282,7 +1284,7 @@ def _GenerateImageConfig_to_mldev(
   if getv(from_object, ['http_options']) is not None:
     setv(to_object, ['httpOptions'], getv(from_object, ['http_options']))
 
-  if getv(from_object, ['output_gcs_uri']):
+  if getv(from_object, ['output_gcs_uri']) is not None:
     raise ValueError('output_gcs_uri parameter is not supported in Google AI.')
 
   if getv(from_object, ['negative_prompt']) is not None:
@@ -1306,7 +1308,7 @@ def _GenerateImageConfig_to_mldev(
         getv(from_object, ['guidance_scale']),
     )
 
-  if getv(from_object, ['seed']):
+  if getv(from_object, ['seed']) is not None:
     raise ValueError('seed parameter is not supported in Google AI.')
 
   if getv(from_object, ['safety_filter_level']) is not None:
@@ -1358,7 +1360,7 @@ def _GenerateImageConfig_to_mldev(
         getv(from_object, ['output_compression_quality']),
     )
 
-  if getv(from_object, ['add_watermark']):
+  if getv(from_object, ['add_watermark']) is not None:
     raise ValueError('add_watermark parameter is not supported in Google AI.')
 
   if getv(from_object, ['aspect_ratio']) is not None:
@@ -1539,7 +1541,7 @@ def _Image_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, ['image_bytes']) is not None:
@@ -1569,15 +1571,15 @@ def _MaskReferenceConfig_to_mldev(
     parent_object: dict = None,
 ) -> dict:
   to_object = {}
-  if getv(from_object, ['mask_mode']):
+  if getv(from_object, ['mask_mode']) is not None:
     raise ValueError('mask_mode parameter is not supported in Google AI.')
 
-  if getv(from_object, ['segmentation_classes']):
+  if getv(from_object, ['segmentation_classes']) is not None:
     raise ValueError(
         'segmentation_classes parameter is not supported in Google AI.'
     )
 
-  if getv(from_object, ['mask_dilation']):
+  if getv(from_object, ['mask_dilation']) is not None:
     raise ValueError('mask_dilation parameter is not supported in Google AI.')
 
   return to_object
@@ -1609,10 +1611,10 @@ def _ControlReferenceConfig_to_mldev(
     parent_object: dict = None,
 ) -> dict:
   to_object = {}
-  if getv(from_object, ['control_type']):
+  if getv(from_object, ['control_type']) is not None:
     raise ValueError('control_type parameter is not supported in Google AI.')
 
-  if getv(from_object, ['enable_control_image_computation']):
+  if getv(from_object, ['enable_control_image_computation']) is not None:
     raise ValueError(
         'enable_control_image_computation parameter is not supported in'
         ' Google AI.'
@@ -1646,7 +1648,7 @@ def _StyleReferenceConfig_to_mldev(
     parent_object: dict = None,
 ) -> dict:
   to_object = {}
-  if getv(from_object, ['style_description']):
+  if getv(from_object, ['style_description']) is not None:
     raise ValueError(
         'style_description parameter is not supported in Google AI.'
     )
@@ -1676,10 +1678,10 @@ def _SubjectReferenceConfig_to_mldev(
     parent_object: dict = None,
 ) -> dict:
   to_object = {}
-  if getv(from_object, ['subject_type']):
+  if getv(from_object, ['subject_type']) is not None:
     raise ValueError('subject_type parameter is not supported in Google AI.')
 
-  if getv(from_object, ['subject_description']):
+  if getv(from_object, ['subject_description']) is not None:
     raise ValueError(
         'subject_description parameter is not supported in Google AI.'
     )
@@ -1712,31 +1714,31 @@ def _ReferenceImageAPI_to_mldev(
     parent_object: dict = None,
 ) -> dict:
   to_object = {}
-  if getv(from_object, ['reference_image']):
+  if getv(from_object, ['reference_image']) is not None:
     raise ValueError('reference_image parameter is not supported in Google AI.')
 
-  if getv(from_object, ['reference_id']):
+  if getv(from_object, ['reference_id']) is not None:
     raise ValueError('reference_id parameter is not supported in Google AI.')
 
-  if getv(from_object, ['reference_type']):
+  if getv(from_object, ['reference_type']) is not None:
     raise ValueError('reference_type parameter is not supported in Google AI.')
 
-  if getv(from_object, ['mask_image_config']):
+  if getv(from_object, ['mask_image_config']) is not None:
     raise ValueError(
         'mask_image_config parameter is not supported in Google AI.'
     )
 
-  if getv(from_object, ['control_image_config']):
+  if getv(from_object, ['control_image_config']) is not None:
     raise ValueError(
         'control_image_config parameter is not supported in Google AI.'
     )
 
-  if getv(from_object, ['style_image_config']):
+  if getv(from_object, ['style_image_config']) is not None:
     raise ValueError(
         'style_image_config parameter is not supported in Google AI.'
     )
 
-  if getv(from_object, ['subject_image_config']):
+  if getv(from_object, ['subject_image_config']) is not None:
     raise ValueError(
         'subject_image_config parameter is not supported in Google AI.'
     )
@@ -1813,7 +1815,7 @@ def _EditImageConfig_to_mldev(
   if getv(from_object, ['http_options']) is not None:
     setv(to_object, ['httpOptions'], getv(from_object, ['http_options']))
 
-  if getv(from_object, ['output_gcs_uri']):
+  if getv(from_object, ['output_gcs_uri']) is not None:
     raise ValueError('output_gcs_uri parameter is not supported in Google AI.')
 
   if getv(from_object, ['negative_prompt']) is not None:
@@ -1837,7 +1839,7 @@ def _EditImageConfig_to_mldev(
         getv(from_object, ['guidance_scale']),
     )
 
-  if getv(from_object, ['seed']):
+  if getv(from_object, ['seed']) is not None:
     raise ValueError('seed parameter is not supported in Google AI.')
 
   if getv(from_object, ['safety_filter_level']) is not None:
@@ -2083,13 +2085,6 @@ def _UpscaleImageAPIConfig_to_mldev(
   if getv(from_object, ['http_options']) is not None:
     setv(to_object, ['httpOptions'], getv(from_object, ['http_options']))
 
-  if getv(from_object, ['upscale_factor']) is not None:
-    setv(
-        parent_object,
-        ['parameters', 'upscaleConfig', 'upscaleFactor'],
-        getv(from_object, ['upscale_factor']),
-    )
-
   if getv(from_object, ['include_rai_reason']) is not None:
     setv(
         parent_object,
@@ -2133,13 +2128,6 @@ def _UpscaleImageAPIConfig_to_vertex(
   if getv(from_object, ['http_options']) is not None:
     setv(to_object, ['httpOptions'], getv(from_object, ['http_options']))
 
-  if getv(from_object, ['upscale_factor']) is not None:
-    setv(
-        parent_object,
-        ['parameters', 'upscaleConfig', 'upscaleFactor'],
-        getv(from_object, ['upscale_factor']),
-    )
-
   if getv(from_object, ['include_rai_reason']) is not None:
     setv(
         parent_object,
@@ -2194,6 +2182,13 @@ def _UpscaleImageAPIParameters_to_mldev(
         _Image_to_mldev(api_client, getv(from_object, ['image']), to_object),
     )
 
+  if getv(from_object, ['upscale_factor']) is not None:
+    setv(
+        to_object,
+        ['parameters', 'upscaleConfig', 'upscaleFactor'],
+        getv(from_object, ['upscale_factor']),
+    )
+
   if getv(from_object, ['config']) is not None:
     setv(
         to_object,
@@ -2226,6 +2221,13 @@ def _UpscaleImageAPIParameters_to_vertex(
         _Image_to_vertex(api_client, getv(from_object, ['image']), to_object),
     )
 
+  if getv(from_object, ['upscale_factor']) is not None:
+    setv(
+        to_object,
+        ['parameters', 'upscaleConfig', 'upscaleFactor'],
+        getv(from_object, ['upscale_factor']),
+    )
+
   if getv(from_object, ['config']) is not None:
     setv(
         to_object,
@@ -2496,7 +2498,7 @@ def _CountTokensConfig_to_mldev(
         ],
     )
 
-  if getv(from_object, ['generation_config']):
+  if getv(from_object, ['generation_config']) is not None:
     raise ValueError(
         'generation_config parameter is not supported in Google AI.'
     )
@@ -2655,7 +2657,7 @@ def _ComputeTokensParameters_to_mldev(
         t.t_model(api_client, getv(from_object, ['model'])),
     )
 
-  if getv(from_object, ['contents']):
+  if getv(from_object, ['contents']) is not None:
     raise ValueError('contents parameter is not supported in Google AI.')
 
   if getv(from_object, ['config']) is not None:
@@ -3984,6 +3986,7 @@ class Models(_common.BaseModule):
       *,
       model: str,
       image: types.ImageOrDict,
+      upscale_factor: str,
       config: Optional[types._UpscaleImageAPIConfigOrDict] = None,
   ) -> types.UpscaleImageResponse:
     """Upscales an image.
@@ -3991,12 +3994,14 @@ class Models(_common.BaseModule):
     Args:
       model (str): The model to use.
       image (Image): The input image for upscaling.
+      upscale_factor (str): The factor to upscale the image (x2 or x4).
       config (_UpscaleImageAPIConfig): Configuration for upscaling.
     """
 
     parameter_model = types._UpscaleImageAPIParameters(
         model=model,
         image=image,
+        upscale_factor=upscale_factor,
         config=config,
     )
 
@@ -4443,13 +4448,15 @@ class Models(_common.BaseModule):
       *,
       model: str,
       image: types.ImageOrDict,
-      config: types.UpscaleImageConfigOrDict,
+      upscale_factor: str,
+      config: Optional[types.UpscaleImageConfigOrDict] = None,
   ) -> types.UpscaleImageResponse:
     """Makes an API request to upscale a provided image.
 
     Args:
       model (str): The model to use.
       image (Image): The input image for upscaling.
+      upscale_factor (str): The factor to upscale the image (x2 or x4).
       config (UpscaleImageConfig): Configuration for upscaling.
 
     Usage:
@@ -4462,9 +4469,7 @@ class Models(_common.BaseModule):
       response=client.models.upscale_image(
           model='imagen-3.0-generate-001',
           image=types.Image.from_file(IMAGE_FILE_PATH),
-                  config={
-                    'upscale_factor': 'x2',
-                  }
+          upscale_factor='x2',
       )
       response.generated_images[0].image.show()
       # Opens my-image.png which is upscaled by a factor of 2.
@@ -4474,10 +4479,12 @@ class Models(_common.BaseModule):
     types.UpscaleImageParameters(
         model=model,
         image=image,
+        upscale_factor=upscale_factor,
         config=config,
     )
 
     # Convert to API config.
+    config = config or {}
     config_dct = config if isinstance(config, dict) else config.dict()
     api_config = types._UpscaleImageAPIConfigDict(**config_dct)  # pylint: disable=protected-access
 
@@ -4485,7 +4492,12 @@ class Models(_common.BaseModule):
     api_config['mode'] = 'upscale'
     api_config['number_of_images'] = 1
 
-    return self._upscale_image(model=model, image=image, config=api_config)
+    return self._upscale_image(
+        model=model,
+        image=image,
+        upscale_factor=upscale_factor,
+        config=api_config,
+    )
 
   def list(
       self,
@@ -4882,6 +4894,7 @@ class AsyncModels(_common.BaseModule):
       *,
       model: str,
       image: types.ImageOrDict,
+      upscale_factor: str,
       config: Optional[types._UpscaleImageAPIConfigOrDict] = None,
   ) -> types.UpscaleImageResponse:
     """Upscales an image.
@@ -4889,12 +4902,14 @@ class AsyncModels(_common.BaseModule):
     Args:
       model (str): The model to use.
       image (Image): The input image for upscaling.
+      upscale_factor (str): The factor to upscale the image (x2 or x4).
       config (_UpscaleImageAPIConfig): Configuration for upscaling.
     """
 
     parameter_model = types._UpscaleImageAPIParameters(
         model=model,
         image=image,
+        upscale_factor=upscale_factor,
         config=config,
     )
 
@@ -5371,13 +5386,15 @@ class AsyncModels(_common.BaseModule):
       *,
       model: str,
       image: types.ImageOrDict,
-      config: types.UpscaleImageConfigOrDict,
+      upscale_factor: str,
+      config: Optional[types.UpscaleImageConfigOrDict] = None,
   ) -> types.UpscaleImageResponse:
     """Makes an API request to upscale a provided image.
 
     Args:
       model (str): The model to use.
       image (Image): The input image for upscaling.
+      upscale_factor (str): The factor to upscale the image (x2 or x4).
       config (UpscaleImageConfig): Configuration for upscaling.
 
     Usage:
@@ -5388,11 +5405,9 @@ class AsyncModels(_common.BaseModule):
 
       IMAGE_FILE_PATH="my-image.png"
       response = await client.aio.models.upscale_image(
-          model = 'imagen-3.0-generate-001',
-          image = types.Image.from_file(IMAGE_FILE_PATH),
-                  config = {
-                    'upscale_factor': 'x2',
-                  }
+          model='imagen-3.0-generate-001',
+          image=types.Image.from_file(IMAGE_FILE_PATH),
+          upscale_factor='x2',
       )
       response.generated_images[0].image.show()
       # Opens my-image.png which is upscaled by a factor of 2.
@@ -5402,10 +5417,12 @@ class AsyncModels(_common.BaseModule):
     types.UpscaleImageParameters(
         model=model,
         image=image,
+        upscale_factor=upscale_factor,
         config=config,
     )
 
     # Convert to API config.
+    config = config or {}
     config_dct = config if isinstance(config, dict) else config.dict()
     api_config = types._UpscaleImageAPIConfigDict(**config_dct)  # pylint: disable=protected-access
 
@@ -5414,5 +5431,8 @@ class AsyncModels(_common.BaseModule):
     api_config['number_of_images'] = 1
 
     return await self._upscale_image(
-        model=model, image=image, config=api_config
+        model=model,
+        image=image,
+        upscale_factor=upscale_factor,
+        config=api_config,
     )

google/genai/tunings.py

@@ -195,10 +195,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 +210,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 +239,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 +251,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 +278,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 +290,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 +307,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 +376,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 +451,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 +479,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 +506,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 +532,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 +605,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: