Add Responses API model constants (ResponsesModels)

Co-authored-by: Cursor <cursoragent@cursor.com>

Author: rushitat

src/gradient/__init__.py

@@ -42,6 +42,7 @@
 )
 from ._base_client import DefaultHttpxClient, DefaultAioHttpClient, DefaultAsyncHttpxClient
 from ._utils._logs import setup_logging as _setup_logging
+from .responses_models import ResponsesModels
 
 __all__ = [
     "types",
@@ -91,6 +92,7 @@
     "DefaultHttpxClient",
     "DefaultAsyncHttpxClient",
     "DefaultAioHttpClient",
+    "ResponsesModels",
 ]
 
 if not _t.TYPE_CHECKING:

src/gradient/responses_models.py

@@ -0,0 +1,8 @@
+"""Recommended model IDs for the Responses API."""
+
+
+class ResponsesModels:
+    """Model ID constants for use with the Responses API."""
+
+    GPT_5_2_PRO = "openai-gpt-5.2-pro"
+    GPT_5_1_CODEX_MAX = "openai-gpt-5.1-codex-max"

tests/test_responses_models.py

@@ -0,0 +1,18 @@
+"""Tests for Responses API model constants."""
+
+import pytest
+
+from gradient import ResponsesModels
+
+
+def test_responses_models_gpt_5_2_pro() -> None:
+    assert ResponsesModels.GPT_5_2_PRO == "openai-gpt-5.2-pro"
+
+
+def test_responses_models_gpt_5_1_codex_max() -> None:
+    assert ResponsesModels.GPT_5_1_CODEX_MAX == "openai-gpt-5.1-codex-max"
+
+
+def test_responses_models_constants_are_strings() -> None:
+    assert isinstance(ResponsesModels.GPT_5_2_PRO, str)
+    assert isinstance(ResponsesModels.GPT_5_1_CODEX_MAX, str)