Default context workbench to Codex 5.5

Author: HaShiShark

backend/web_runtime.py

@@ -28,7 +28,14 @@
     ToolCallReadyEvent,
 )
 from simple_agent.agent import BridgedFunctionCall, SimpleAgent, ToolEvent, sanitize_text
-from simple_agent.config import CODEX_PROXY_BASE_URL, CODEX_PROXY_PROVIDER_ID, DEFAULT_CODEX_PROXY_MODELS, Settings
+from simple_agent.config import (
+    CODEX_PROXY_BASE_URL,
+    CODEX_PROXY_PROVIDER_ID,
+    DEFAULT_CODEX_PROXY_MODELS,
+    DEFAULT_CONTEXT_WORKBENCH_MODEL,
+    DEFAULT_CONTEXT_WORKBENCH_PROVIDER_ID,
+    Settings,
+)
 from simple_agent.codex_tool_registry import ToolExecution
 from simple_agent.provider_clients import ClaudeRESTClient, GeminiRESTClient
 try:
@@ -58,14 +65,13 @@
 )
 
 def context_workbench_settings_payload(settings: Settings) -> dict[str, object]:
-    default_context_model = sanitize_text(DEFAULT_CODEX_PROXY_MODELS[0].get("id") or "").strip() or "gpt-5.5"
     return {
         "context_workbench_model": sanitize_text(settings.context_workbench_model or "").strip()
-        or default_context_model,
+        or DEFAULT_CONTEXT_WORKBENCH_MODEL,
         "context_workbench_provider_id": sanitize_text(
             settings.context_workbench_provider_id or ""
         ).strip()
-        or CODEX_PROXY_PROVIDER_ID,
+        or DEFAULT_CONTEXT_WORKBENCH_PROVIDER_ID,
         "context_token_warning_threshold": int(settings.context_token_warning_threshold or 5000),
         "context_token_critical_threshold": int(settings.context_token_critical_threshold or 10000),
         "user_locale": sanitize_text(settings.user_locale or "").strip() or "en-US",
@@ -155,8 +161,7 @@ def build_context_chat_runtime(
         )
     )
 
-    default_context_model = sanitize_text(DEFAULT_CODEX_PROXY_MODELS[0].get("id") or "").strip() or "gpt-5.5"
-    request_model = sanitize_text(settings.context_workbench_model or "").strip() or default_context_model
+    request_model = sanitize_text(settings.context_workbench_model or "").strip() or DEFAULT_CONTEXT_WORKBENCH_MODEL
     instructions = "\n".join(
         [
             "You are a context-maintenance assistant for a Codex conversation.",
@@ -358,12 +363,11 @@ def context_workbench_provider(settings: Settings) -> dict[str, Any]:
         return provider
 
     return {
-        "id": CODEX_PROXY_PROVIDER_ID,
+        "id": DEFAULT_CONTEXT_WORKBENCH_PROVIDER_ID,
         "name": "Codex",
         "provider_type": "responses",
         "api_base_url": CODEX_PROXY_BASE_URL,
-        "default_model": sanitize_text(DEFAULT_CODEX_PROXY_MODELS[0].get("id") or "").strip()
-        or "gpt-5.5",
+        "default_model": DEFAULT_CONTEXT_WORKBENCH_MODEL,
     }
 
 
@@ -1313,8 +1317,7 @@ def context_workbench_provider_payloads(settings: Settings, *, refresh_models: b
                     "provider_type": "responses",
                     "enabled": True,
                     "api_base_url": CODEX_PROXY_BASE_URL,
-                    "default_model": sanitize_text(DEFAULT_CODEX_PROXY_MODELS[0].get("id") or "").strip()
-                    or "gpt-5.5",
+                    "default_model": DEFAULT_CONTEXT_WORKBENCH_MODEL,
                     "models": DEFAULT_CODEX_PROXY_MODELS,
                 },
                 settings,

scripts/test_prompt_settings.py

@@ -143,13 +143,26 @@ def test_compact_prompt_uses_configured_manual_and_auto_defaults() -> None:
         assert is_local_compact_prompt_text("custom auto compact") is True
 
 
+def test_context_workbench_defaults_to_codex_55() -> None:
+    with isolated_settings():
+        settings = config_module.load_settings()
+        assert settings.context_workbench_provider_id == config_module.DEFAULT_CONTEXT_WORKBENCH_PROVIDER_ID
+        assert settings.context_workbench_model == config_module.DEFAULT_CONTEXT_WORKBENCH_MODEL
+
+        config_module.save_settings(codex_system_prompt="custom context prompt")
+        settings = config_module.load_settings()
+        assert settings.context_workbench_provider_id == config_module.DEFAULT_CONTEXT_WORKBENCH_PROVIDER_ID
+        assert settings.context_workbench_model == config_module.DEFAULT_CONTEXT_WORKBENCH_MODEL
+
+
 def main() -> None:
     tests = [
         test_first_codex_instructions_updates_default_and_current_on_next_proxy_start,
         test_first_codex_instructions_preserves_user_custom_prompt,
         test_first_codex_instructions_only_scans_top_level_instructions,
         test_codex_system_prompt_override_replaces_forwarded_instructions,
         test_compact_prompt_uses_configured_manual_and_auto_defaults,
+        test_context_workbench_defaults_to_codex_55,
     ]
     for test in tests:
         test()

simple_agent/config.py

@@ -40,6 +40,8 @@ def _resolve_data_dir() -> Path:
     {"id": "gpt-5.4", "label": "gpt-5.4", "group": "Codex", "provider": "Codex"},
     {"id": "gpt-5.2", "label": "gpt-5.2", "group": "Codex", "provider": "Codex"},
 )
+DEFAULT_CONTEXT_WORKBENCH_PROVIDER_ID = CODEX_PROXY_PROVIDER_ID
+DEFAULT_CONTEXT_WORKBENCH_MODEL = "gpt-5.5"
 
 DEFAULT_RESPONSE_PROVIDERS: tuple[dict[str, object], ...] = (
     {
@@ -537,7 +539,7 @@ def load_settings() -> Settings:
         stored.get("context_workbench_provider_id")
         or os.getenv("HASH_CONTEXT_WORKBENCH_PROVIDER_ID"),
         response_providers,
-        fallback_provider_id=CODEX_PROXY_PROVIDER_ID,
+        fallback_provider_id=DEFAULT_CONTEXT_WORKBENCH_PROVIDER_ID,
     )
 
     active_provider = next(
@@ -562,7 +564,7 @@ def load_settings() -> Settings:
     context_workbench_model = (
         raw_context_workbench_model
         or _clean_string(context_provider.get("default_model"))
-        or _clean_string(DEFAULT_CODEX_PROXY_MODELS[0].get("id"))
+        or DEFAULT_CONTEXT_WORKBENCH_MODEL
         or model
     )
     tool_settings = normalize_tool_settings(stored.get("tool_settings"))
@@ -841,7 +843,7 @@ def save_settings(
         or current.get("context_workbench_provider_id")
         or loaded.context_workbench_provider_id,
         ordered_records,
-        fallback_provider_id=CODEX_PROXY_PROVIDER_ID,
+        fallback_provider_id=DEFAULT_CONTEXT_WORKBENCH_PROVIDER_ID,
     )
     context_provider = current_by_id.get(next_context_workbench_provider_id) or active_provider
     current["context_workbench_provider_id"] = next_context_workbench_provider_id
@@ -854,6 +856,7 @@ def save_settings(
         else:
             next_context_workbench_model = (
                 _clean_string(context_provider.get("default_model"))
+                or DEFAULT_CONTEXT_WORKBENCH_MODEL
                 or _clean_string(current.get("model"))
                 or loaded.model
             )
@@ -863,6 +866,7 @@ def save_settings(
     ):
         next_context_workbench_model = (
             _clean_string(context_provider.get("default_model"))
+            or DEFAULT_CONTEXT_WORKBENCH_MODEL
             or _clean_string(current.get("model"))
             or loaded.model
         )