mirror of
https://github.com/openai/codex.git
synced 2026-10-18 17:18:52 +00:00
Support self-serve Business ProLite accounts (#35785)
## What changed - Recognize `self_serve_business_prolite` across authentication, account and rate-limit APIs, generated schemas, workspace classification, status display, cloud configuration gating, and usage-limit messaging. - Keep the Python SDK's `PlanType` enum compatible with non-empty string values introduced by newer Codex runtimes while preserving its known constants. ## Testing - Cover token parsing, account reads and notifications, backend rate-limit mapping, workspace behavior, error formatting, and Python SDK response coercion for the new plan value. GitOrigin-RevId: 70bc17a7c4ba4028cb10e4333cc1f2ac64da361f
This commit is contained in:
@@ -5,8 +5,12 @@ from pathlib import Path
|
||||
from openai_codex.client import CodexClient, _params_dict
|
||||
from openai_codex.generated.notification_registry import notification_turn_id
|
||||
from openai_codex.generated.v2_all import (
|
||||
AccountRateLimitsUpdatedNotification,
|
||||
AccountUpdatedNotification,
|
||||
AgentMessageDeltaNotification,
|
||||
ApprovalsReviewer,
|
||||
GetAccountResponse,
|
||||
PlanType,
|
||||
ReasoningEffort,
|
||||
ReasoningEffortOption,
|
||||
ThreadForkParams,
|
||||
@@ -37,6 +41,38 @@ def test_generated_v2_bundle_has_single_shared_plan_type_definition() -> None:
|
||||
assert source.count("class PlanType(") == 1
|
||||
|
||||
|
||||
def test_plan_type_accepts_business_prolite_from_newer_runtime() -> None:
|
||||
"""New runtime plan values should remain typed when using a codex_bin override."""
|
||||
plan_type = "self_serve_business_prolite"
|
||||
response = GetAccountResponse.model_validate(
|
||||
{
|
||||
"account": {
|
||||
"type": "chatgpt",
|
||||
"email": "user@example.com",
|
||||
"planType": plan_type,
|
||||
},
|
||||
"requiresOpenaiAuth": True,
|
||||
}
|
||||
)
|
||||
assert response.account is not None
|
||||
assert response.account.root.plan_type.value == plan_type
|
||||
|
||||
client = CodexClient()
|
||||
account_updated = client._coerce_notification(
|
||||
"account/updated",
|
||||
{"authMode": "chatgpt", "planType": plan_type},
|
||||
)
|
||||
assert isinstance(account_updated.payload, AccountUpdatedNotification)
|
||||
assert account_updated.payload.plan_type == PlanType(plan_type)
|
||||
|
||||
rate_limits_updated = client._coerce_notification(
|
||||
"account/rateLimits/updated",
|
||||
{"rateLimits": {"planType": plan_type}},
|
||||
)
|
||||
assert isinstance(rate_limits_updated.payload, AccountRateLimitsUpdatedNotification)
|
||||
assert rate_limits_updated.payload.rate_limits.plan_type == PlanType(plan_type)
|
||||
|
||||
|
||||
def test_reasoning_effort_preserves_enum_constants_and_accepts_future_values() -> None:
|
||||
"""Known effort members and new runtime values should share the enum-style API."""
|
||||
known_option = ReasoningEffortOption.model_validate(
|
||||
|
||||
Reference in New Issue
Block a user