From 63c09ed2127cd7d3e2e1b9222c00c9d255f67b1b Mon Sep 17 00:00:00 2001
From: Krish Chainani <kchainani@openai.com>
Date: Tue, 15 Sep 2026 21:07:57 +0000
Subject: [PATCH] Preserve `ImageUserInput` in the Python SDK (#45796)

Keep the existing public class name for URL-based image input when regenerating the SDK. Map `UrlUserInput` to `ImageUserInput` during artifact generation and update the generated model and `UserInput` union accordingly.

Extend the class-name stability test to assert that `ImageUserInput` remains importable under its expected name.

GitOrigin-RevId: 6cbbd555e2e07904cc5fbc279e1e61ff089ac0a9
---
 sdk/python/scripts/update_sdk_artifacts.py              | 1 +
 sdk/python/src/openai_codex/generated/v2_all.py         | 6 +++---
 sdk/python/tests/test_artifact_workflow_and_binaries.py | 2 ++
 3 files changed, 6 insertions(+), 3 deletions(-)

diff --git a/sdk/python/scripts/update_sdk_artifacts.py b/sdk/python/scripts/update_sdk_artifacts.py
index 61d98e29e4..4c36b595f0 100755
--- a/sdk/python/scripts/update_sdk_artifacts.py
+++ b/sdk/python/scripts/update_sdk_artifacts.py
@@ -587,6 +587,7 @@ def _preserve_inline_image_class_names(out_path: Path) -> None:
     """Keep the public class names used before ImageReference was introduced."""
     source = out_path.read_text()
     stable_names = {
+        "UrlUserInput": "ImageUserInput",
         "ImageUrlContentItem": "InputImageContentItem",
         "ImageUrlFunctionCallOutputContentItem": "InputImageFunctionCallOutputContentItem",
     }
diff --git a/sdk/python/src/openai_codex/generated/v2_all.py b/sdk/python/src/openai_codex/generated/v2_all.py
index 0a76ae8fc4..8f500002c2 100644
--- a/sdk/python/src/openai_codex/generated/v2_all.py
+++ b/sdk/python/src/openai_codex/generated/v2_all.py
@@ -6240,7 +6240,7 @@ class TextUserInput(BaseModel):
     type: Annotated[Literal["text"], Field(title="TextUserInputType")]
 
 
-class UrlUserInput(BaseModel):
+class ImageUserInput(BaseModel):
     model_config = ConfigDict(
         populate_by_name=True,
     )
@@ -6304,7 +6304,7 @@ class MentionUserInput(BaseModel):
 class UserInput(
     RootModel[
         TextUserInput
-        | UrlUserInput
+        | ImageUserInput
         | FileIdUserInput
         | LocalImageUserInput
         | AudioUserInput
@@ -6318,7 +6318,7 @@ class UserInput(
     )
     root: (
         TextUserInput
-        | UrlUserInput
+        | ImageUserInput
         | FileIdUserInput
         | LocalImageUserInput
         | AudioUserInput
diff --git a/sdk/python/tests/test_artifact_workflow_and_binaries.py b/sdk/python/tests/test_artifact_workflow_and_binaries.py
index 7fae0b7a57..1d2e23170b 100644
--- a/sdk/python/tests/test_artifact_workflow_and_binaries.py
+++ b/sdk/python/tests/test_artifact_workflow_and_binaries.py
@@ -539,10 +539,12 @@ def test_generated_chatgpt_account_email_is_required_nullable() -> None:
 def test_generated_inline_image_class_names_remain_stable() -> None:
     """Keep the existing Python class names when image references expand."""
     from openai_codex.generated.v2_all import (
+        ImageUserInput,
         InputImageContentItem,
         InputImageFunctionCallOutputContentItem,
     )
 
+    assert ImageUserInput.__name__ == "ImageUserInput"
     assert InputImageContentItem.__name__ == "InputImageContentItem"
     assert (
         InputImageFunctionCallOutputContentItem.__name__