diff --git a/google/genai/models.py b/google/genai/models.py index 3426a89d..73cff733 100644 --- a/google/genai/models.py +++ b/google/genai/models.py @@ -4267,7 +4267,7 @@ def _edit_image( Args: model (str): The model to use. prompt (str): A text description of the edit to apply to the image. - reference_images (list[Union[RawReferenceImage, MaskReferenceImage, + reference_images (list[Union[RawReferenceImage, MaskReferenceImage, ControlReferenceImage, StyleReferenceImage, SubjectReferenceImage]): The reference images for editing. config (EditImageConfig): Configuration for editing. @@ -5277,7 +5277,7 @@ def edit_image( Args: model (str): The model to use. prompt (str): A text description of the edit to apply to the image. - reference_images (list[Union[RawReferenceImage, MaskReferenceImage, + reference_images (list[Union[RawReferenceImage, MaskReferenceImage, ControlReferenceImage, StyleReferenceImage, SubjectReferenceImage]): The reference images for editing. config (EditImageConfig): Configuration for editing. @@ -5768,7 +5768,7 @@ async def _edit_image( Args: model (str): The model to use. prompt (str): A text description of the edit to apply to the image. - reference_images (list[Union[RawReferenceImage, MaskReferenceImage, + reference_images (list[Union[RawReferenceImage, MaskReferenceImage, ControlReferenceImage, StyleReferenceImage, SubjectReferenceImage]): The reference images for editing. config (EditImageConfig): Configuration for editing. @@ -6710,7 +6710,7 @@ async def edit_image( Args: model (str): The model to use. prompt (str): A text description of the edit to apply to the image. - reference_images (list[Union[RawReferenceImage, MaskReferenceImage, + reference_images (list[Union[RawReferenceImage, MaskReferenceImage, ControlReferenceImage, StyleReferenceImage, SubjectReferenceImage]): The reference images for editing. config (EditImageConfig): Configuration for editing.