From e3221c89b7d2ab6a0b65ab0f4dfd37a6358659b1 Mon Sep 17 00:00:00 2001 From: nightcityblade Date: Sun, 26 Apr 2026 23:10:28 +0800 Subject: [PATCH] types: add gpt-image-2 image model alias --- src/openai/types/image_generate_params.py | 3 +-- src/openai/types/image_model.py | 2 +- tests/test_image_model.py | 7 +++++++ 3 files changed, 9 insertions(+), 3 deletions(-) create mode 100644 tests/test_image_model.py 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)