diff --git a/src/openai/types/image_generate_params.py b/src/openai/types/image_generate_params.py index 7a95b3dd3d..c9e176092f 100644 --- a/src/openai/types/image_generate_params.py +++ b/src/openai/types/image_generate_params.py @@ -32,8 +32,7 @@ class ImageGenerateParamsBase(TypedDict, total=False): model: Union[str, ImageModel, None] """The model to use for image generation. - One of `dall-e-2`, `dall-e-3`, or a GPT image model (`gpt-image-1`, - `gpt-image-1-mini`, `gpt-image-1.5`). Defaults to `dall-e-2` unless a parameter + One of `dall-e-2`, `dall-e-3`, or a GPT image model (`gpt-image-1`, `gpt-image-1-mini`, `gpt-image-1.5`, `gpt-image-2`). Defaults to `dall-e-2` unless a parameter specific to the GPT image models is used. """ diff --git a/src/openai/types/image_model.py b/src/openai/types/image_model.py index 8ea486fbb6..4fd9eaef80 100644 --- a/src/openai/types/image_model.py +++ b/src/openai/types/image_model.py @@ -4,4 +4,4 @@ __all__ = ["ImageModel"] -ImageModel: TypeAlias = Literal["gpt-image-1.5", "dall-e-2", "dall-e-3", "gpt-image-1", "gpt-image-1-mini"] +ImageModel: TypeAlias = Literal["gpt-image-1.5", "gpt-image-2", "dall-e-2", "dall-e-3", "gpt-image-1", "gpt-image-1-mini"] diff --git a/tests/test_image_model.py b/tests/test_image_model.py new file mode 100644 index 0000000000..8d98a2ff50 --- /dev/null +++ b/tests/test_image_model.py @@ -0,0 +1,7 @@ +from typing import get_args + +from openai.types import ImageModel + + +def test_image_model_includes_gpt_image_2() -> None: + assert "gpt-image-2" in get_args(ImageModel)