mirror of
https://github.com/openai/codex.git
synced 2026-10-22 17:58:46 +00:00
## What changed - Add a feature-gated `TurnSettings` operation that can update the model, reasoning effort, reasoning summary, and service tier for subsequent steps of a named running turn without changing future thread settings. - Report whether an update was applied, rejected, or lost its live target. - Revalidate managed constraints and preserve the turn's admitted approval and Guardian safety properties before publishing an updated settings snapshot. ## Testing - Cover sparse updates, step capture and ordering, target replacement, managed policy changes, model metadata safety, and separation from future settings. GitOrigin-RevId: ecaaaa95b4fd4ec9d62265bef551582dddb36d78
912 lines
31 KiB
Rust
912 lines
31 KiB
Rust
use anyhow::Result;
|
|
use codex_core::CodexThread;
|
|
use codex_core::TurnInputRequest;
|
|
use codex_core::config::Constrained;
|
|
use codex_features::Feature;
|
|
use codex_login::CodexAuth;
|
|
use codex_models_manager::bundled_models_response;
|
|
use codex_models_manager::manager::RefreshStrategy;
|
|
use codex_protocol::config_types::ApprovalsReviewer;
|
|
use codex_protocol::config_types::ReasoningSummary;
|
|
use codex_protocol::config_types::ServiceTier;
|
|
use codex_protocol::openai_models::ModelInfo;
|
|
use codex_protocol::openai_models::ModelsResponse;
|
|
use codex_protocol::openai_models::ReasoningEffort;
|
|
use codex_protocol::protocol::AskForApproval;
|
|
use codex_protocol::protocol::EventMsg;
|
|
use codex_protocol::protocol::Op;
|
|
use codex_protocol::protocol::SafetyBufferingEvent;
|
|
use codex_protocol::protocol::ThreadSettingsOverrides;
|
|
use codex_protocol::protocol::TurnSettingsUpdate;
|
|
use codex_protocol::protocol::TurnSettingsUpdateOutcome;
|
|
use codex_protocol::request_user_input::RequestUserInputAnswer;
|
|
use codex_protocol::request_user_input::RequestUserInputEvent;
|
|
use codex_protocol::request_user_input::RequestUserInputResponse;
|
|
use codex_protocol::user_input::UserInput;
|
|
use core_test_support::responses::ResponsesRequest;
|
|
use core_test_support::responses::ev_completed;
|
|
use core_test_support::responses::ev_function_call;
|
|
use core_test_support::responses::ev_response_created;
|
|
use core_test_support::responses::mount_models_once;
|
|
use core_test_support::responses::mount_response_sequence;
|
|
use core_test_support::responses::mount_sse_sequence;
|
|
use core_test_support::responses::sse;
|
|
use core_test_support::responses::sse_completed;
|
|
use core_test_support::responses::sse_response;
|
|
use core_test_support::responses::start_mock_server;
|
|
use core_test_support::skip_if_no_network;
|
|
use core_test_support::test_codex::TestCodexBuilder;
|
|
use core_test_support::test_codex::test_codex;
|
|
use core_test_support::wait_for_event;
|
|
use core_test_support::wait_for_event_match;
|
|
use pretty_assertions::assert_eq;
|
|
use serde_json::Value;
|
|
use serde_json::json;
|
|
use std::collections::HashMap;
|
|
use test_case::test_case;
|
|
|
|
const MODEL_A: &str = "step-settings-a";
|
|
const MODEL_B: &str = "step-settings-b";
|
|
const MODEL_C: &str = "step-settings-c";
|
|
const TURN_STATE_HEADER: &str = "x-codex-turn-state";
|
|
|
|
fn step_settings_models() -> Vec<ModelInfo> {
|
|
let model = bundled_models_response()
|
|
.expect("bundled models should parse")
|
|
.models
|
|
.into_iter()
|
|
.find(|model| model.slug == "gpt-5.4")
|
|
.expect("bundled gpt-5.4 model");
|
|
[MODEL_A, MODEL_B, MODEL_C]
|
|
.into_iter()
|
|
.map(|slug| {
|
|
let mut model = model.clone();
|
|
// Tests add model-owned differences when they need to exercise
|
|
// activation or an explicit safety restriction.
|
|
model.slug = slug.to_string();
|
|
model
|
|
})
|
|
.collect()
|
|
}
|
|
|
|
fn step_settings_test() -> TestCodexBuilder {
|
|
test_codex().with_model(MODEL_A).with_config(move |config| {
|
|
for feature in [
|
|
Feature::StepModelSwitching,
|
|
Feature::DefaultModeRequestUserInput,
|
|
Feature::FastMode,
|
|
] {
|
|
config
|
|
.features
|
|
.enable(feature)
|
|
.expect("test config should allow feature update");
|
|
}
|
|
config.model_catalog = Some(ModelsResponse {
|
|
models: step_settings_models(),
|
|
});
|
|
config.model_reasoning_effort = Some(ReasoningEffort::Low);
|
|
config.model_reasoning_summary = Some(ReasoningSummary::Concise);
|
|
config.service_tier = None;
|
|
config.permissions.approval_policy = Constrained::allow_any(AskForApproval::OnRequest);
|
|
config.approvals_reviewer = ApprovalsReviewer::User;
|
|
})
|
|
}
|
|
|
|
fn paused_response(response_id: &str, call_id: &str) -> String {
|
|
sse(vec![
|
|
ev_response_created(response_id),
|
|
ev_function_call(
|
|
call_id,
|
|
"request_user_input",
|
|
&json!({
|
|
"questions": [{
|
|
"id": "continue",
|
|
"header": "Continue",
|
|
"question": "Continue after the settings update?",
|
|
"options": [{
|
|
"label": "Yes (Recommended)",
|
|
"description": "Continue the current turn."
|
|
}, {
|
|
"label": "No",
|
|
"description": "Stop the current turn."
|
|
}]
|
|
}]
|
|
})
|
|
.to_string(),
|
|
),
|
|
ev_completed(response_id),
|
|
])
|
|
}
|
|
|
|
async fn start_paused_turn(thread: &CodexThread) -> Result<RequestUserInputEvent> {
|
|
thread
|
|
.start_or_steer_turn(TurnInputRequest::user_input(vec![UserInput::Text {
|
|
text: "pause before continuing".into(),
|
|
text_elements: Vec::new(),
|
|
}]))
|
|
.await?;
|
|
Ok(wait_for_event_match(thread, |event| match event {
|
|
EventMsg::RequestUserInput(request) => Some(request.clone()),
|
|
_ => None,
|
|
})
|
|
.await)
|
|
}
|
|
|
|
async fn answer_paused_turn(thread: &CodexThread, turn_id: &str) -> Result<()> {
|
|
thread
|
|
.submit(Op::UserInputAnswer {
|
|
id: turn_id.to_string(),
|
|
response: RequestUserInputResponse {
|
|
answers: HashMap::from([(
|
|
"continue".to_string(),
|
|
RequestUserInputAnswer {
|
|
answers: vec!["Yes (Recommended)".to_string()],
|
|
},
|
|
)]),
|
|
},
|
|
})
|
|
.await?;
|
|
Ok(())
|
|
}
|
|
|
|
async fn submit_turn_settings(
|
|
thread: &CodexThread,
|
|
turn_id: &str,
|
|
update: TurnSettingsUpdate,
|
|
) -> Result<TurnSettingsUpdateOutcome> {
|
|
let (reply, outcome) = tokio::sync::oneshot::channel();
|
|
thread
|
|
.submit(Op::TurnSettings {
|
|
turn_id: turn_id.to_string(),
|
|
update,
|
|
reply,
|
|
})
|
|
.await?;
|
|
Ok(tokio::time::timeout(std::time::Duration::from_secs(/*secs*/ 10), outcome).await??)
|
|
}
|
|
|
|
fn request_settings(request: &ResponsesRequest) -> Value {
|
|
let body = request.body_json();
|
|
json!({
|
|
"model": body["model"],
|
|
"reasoning": body["reasoning"],
|
|
"service_tier": body.get("service_tier"),
|
|
})
|
|
}
|
|
|
|
fn request_turn_id(request: &ResponsesRequest) -> String {
|
|
let metadata: Value = serde_json::from_str(
|
|
&request
|
|
.header("x-codex-turn-metadata")
|
|
.expect("request should include turn metadata"),
|
|
)
|
|
.expect("valid turn metadata");
|
|
metadata["turn_id"]
|
|
.as_str()
|
|
.expect("request should include turn_id")
|
|
.to_string()
|
|
}
|
|
|
|
#[derive(Clone, Copy)]
|
|
enum SettingsTarget {
|
|
Thread,
|
|
Turn,
|
|
}
|
|
|
|
#[test_case(SettingsTarget::Thread; "thread updates stay next-turn-only with the feature enabled")]
|
|
#[test_case(SettingsTarget::Turn; "turn updates leave future settings unchanged")]
|
|
#[tokio::test(flavor = "multi_thread", worker_threads = 2)]
|
|
async fn settings_updates_preserve_turn_identity_and_target(target: SettingsTarget) -> Result<()> {
|
|
skip_if_no_network!(Ok(()));
|
|
|
|
let server = start_mock_server().await;
|
|
let response_mock = mount_response_sequence(
|
|
&server,
|
|
vec![
|
|
sse_response(paused_response("resp-1", "pause-turn"))
|
|
.insert_header(TURN_STATE_HEADER, "original-turn-state"),
|
|
sse_response(sse_completed("resp-2")),
|
|
sse_response(sse_completed("resp-3")),
|
|
],
|
|
)
|
|
.await;
|
|
let test = step_settings_test().build_with_auto_env(&server).await?;
|
|
let original_settings = test.codex.thread_settings_snapshot().await;
|
|
let request = start_paused_turn(&test.codex).await?;
|
|
|
|
match target {
|
|
SettingsTarget::Thread => {
|
|
test.codex
|
|
.submit(Op::ThreadSettings {
|
|
thread_settings: ThreadSettingsOverrides {
|
|
model: Some(MODEL_B.to_string()),
|
|
effort: Some(Some(ReasoningEffort::High)),
|
|
summary: Some(ReasoningSummary::Detailed),
|
|
service_tier: Some(Some(ServiceTier::Fast.request_value().to_string())),
|
|
..Default::default()
|
|
},
|
|
})
|
|
.await?;
|
|
}
|
|
SettingsTarget::Turn => {
|
|
let update = TurnSettingsUpdate {
|
|
model: Some(MODEL_B.to_string()),
|
|
effort: Some(Some(ReasoningEffort::High)),
|
|
summary: Some(ReasoningSummary::Detailed),
|
|
service_tier: Some(Some(ServiceTier::Fast.request_value().to_string())),
|
|
};
|
|
assert_eq!(
|
|
submit_turn_settings(&test.codex, "different-turn", update.clone()).await?,
|
|
TurnSettingsUpdateOutcome::TargetUnavailable
|
|
);
|
|
assert_eq!(
|
|
submit_turn_settings(&test.codex, &request.turn_id, update).await?,
|
|
TurnSettingsUpdateOutcome::Applied
|
|
);
|
|
}
|
|
}
|
|
answer_paused_turn(&test.codex, &request.turn_id).await?;
|
|
let mut settings_events = Vec::new();
|
|
let mut new_turns = Vec::new();
|
|
let completion = wait_for_event(&test.codex, |event| match event {
|
|
EventMsg::ThreadSettingsApplied(event) => {
|
|
settings_events.push(event.thread_settings.clone());
|
|
false
|
|
}
|
|
EventMsg::TurnStarted(event) => {
|
|
new_turns.push(event.turn_id.clone());
|
|
false
|
|
}
|
|
EventMsg::Error(error) => panic!("settings activation failed: {}", error.message),
|
|
EventMsg::TurnComplete(_) => true,
|
|
_ => false,
|
|
})
|
|
.await;
|
|
let EventMsg::TurnComplete(completion) = completion else {
|
|
unreachable!("waited for turn completion")
|
|
};
|
|
assert_eq!(completion.turn_id, request.turn_id);
|
|
assert_eq!(new_turns, Vec::<String>::new());
|
|
|
|
let mut changed_settings = original_settings.clone();
|
|
changed_settings.model = MODEL_B.to_string();
|
|
changed_settings.reasoning_effort = Some(ReasoningEffort::High);
|
|
changed_settings.reasoning_summary = Some(ReasoningSummary::Detailed);
|
|
changed_settings.service_tier = Some(ServiceTier::Fast.request_value().to_string());
|
|
changed_settings.collaboration_mode = changed_settings.collaboration_mode.with_updates(
|
|
Some(MODEL_B.to_string()),
|
|
Some(Some(ReasoningEffort::High)),
|
|
/*developer_instructions*/ None,
|
|
);
|
|
let expected_future_settings = match target {
|
|
SettingsTarget::Turn => {
|
|
assert_eq!(settings_events, Vec::new());
|
|
original_settings
|
|
}
|
|
SettingsTarget::Thread => {
|
|
assert_eq!(settings_events, vec![changed_settings.clone()]);
|
|
changed_settings
|
|
}
|
|
};
|
|
assert_eq!(
|
|
test.codex.thread_settings_snapshot().await,
|
|
expected_future_settings
|
|
);
|
|
test.submit_text_turn("start the next turn").await?;
|
|
|
|
let original_request_settings = json!({
|
|
"model": MODEL_A,
|
|
"reasoning": { "effort": "low", "summary": "concise" },
|
|
"service_tier": null,
|
|
});
|
|
let changed_request_settings = json!({
|
|
"model": MODEL_B,
|
|
"reasoning": { "effort": "high", "summary": "detailed" },
|
|
"service_tier": "priority",
|
|
});
|
|
let (continued_settings, next_turn_settings) = match target {
|
|
SettingsTarget::Thread => (original_request_settings.clone(), changed_request_settings),
|
|
SettingsTarget::Turn => (changed_request_settings, original_request_settings.clone()),
|
|
};
|
|
let requests = response_mock.requests();
|
|
assert_eq!(
|
|
requests.iter().map(request_settings).collect::<Vec<_>>(),
|
|
vec![
|
|
original_request_settings,
|
|
continued_settings,
|
|
next_turn_settings,
|
|
]
|
|
);
|
|
assert_eq!(request_turn_id(&requests[0]), request.turn_id);
|
|
assert_eq!(request_turn_id(&requests[1]), request.turn_id);
|
|
assert_ne!(request_turn_id(&requests[2]), request.turn_id);
|
|
let session_id = requests[0]
|
|
.header("session-id")
|
|
.expect("initial request session id");
|
|
assert_eq!(requests[1].header("session-id"), Some(session_id));
|
|
assert_eq!(requests[0].header(TURN_STATE_HEADER), None);
|
|
assert_eq!(
|
|
requests[1].header(TURN_STATE_HEADER),
|
|
Some("original-turn-state".to_string())
|
|
);
|
|
assert_eq!(requests[2].header(TURN_STATE_HEADER), None);
|
|
|
|
Ok(())
|
|
}
|
|
|
|
#[tokio::test(flavor = "multi_thread", worker_threads = 2)]
|
|
async fn response_metadata_uses_the_captured_model_after_a_turn_update() -> Result<()> {
|
|
skip_if_no_network!(Ok(()));
|
|
|
|
let server = start_mock_server().await;
|
|
let mut created = ev_response_created("resp-2");
|
|
created["safety_buffering"] = json!({
|
|
"use_cases": ["cyber"],
|
|
"reasons": ["policy-check"],
|
|
"retry_model": MODEL_C,
|
|
});
|
|
let response_mock = mount_response_sequence(
|
|
&server,
|
|
vec![
|
|
sse_response(paused_response("resp-1", "pause-turn")),
|
|
sse_response(sse(vec![created, ev_completed("resp-2")]))
|
|
.insert_header("OpenAI-Model", MODEL_B),
|
|
],
|
|
)
|
|
.await;
|
|
let test = step_settings_test().build_with_auto_env(&server).await?;
|
|
let request = start_paused_turn(&test.codex).await?;
|
|
assert_eq!(
|
|
submit_turn_settings(
|
|
&test.codex,
|
|
&request.turn_id,
|
|
TurnSettingsUpdate {
|
|
model: Some(MODEL_B.to_string()),
|
|
..Default::default()
|
|
},
|
|
)
|
|
.await?,
|
|
TurnSettingsUpdateOutcome::Applied
|
|
);
|
|
answer_paused_turn(&test.codex, &request.turn_id).await?;
|
|
|
|
let mut reroutes = Vec::new();
|
|
let mut buffering_events = Vec::new();
|
|
wait_for_event(&test.codex, |event| match event {
|
|
EventMsg::ModelReroute(event) => {
|
|
reroutes.push(event.clone());
|
|
false
|
|
}
|
|
EventMsg::SafetyBuffering(event) => {
|
|
buffering_events.push(event.clone());
|
|
false
|
|
}
|
|
EventMsg::Error(error) => panic!("sampling failed: {}", error.message),
|
|
EventMsg::TurnComplete(_) => true,
|
|
_ => false,
|
|
})
|
|
.await;
|
|
|
|
assert_eq!(
|
|
response_mock
|
|
.requests()
|
|
.iter()
|
|
.map(|request| request.body_json()["model"].clone())
|
|
.collect::<Vec<_>>(),
|
|
vec![json!(MODEL_A), json!(MODEL_B)]
|
|
);
|
|
// B's matching response header is not a reroute from the turn's initial A.
|
|
// Buffering metadata likewise belongs to the captured B step.
|
|
assert_eq!(
|
|
(reroutes, buffering_events),
|
|
(
|
|
vec![],
|
|
vec![SafetyBufferingEvent {
|
|
model: MODEL_B.to_string(),
|
|
use_cases: vec!["cyber".to_string()],
|
|
reasons: vec!["policy-check".to_string()],
|
|
show_buffering_ui: true,
|
|
faster_model: Some(MODEL_C.to_string()),
|
|
}],
|
|
)
|
|
);
|
|
Ok(())
|
|
}
|
|
|
|
#[tokio::test(flavor = "multi_thread", worker_threads = 2)]
|
|
async fn sparse_updates_preserve_divergent_active_and_future_models() -> Result<()> {
|
|
skip_if_no_network!(Ok(()));
|
|
|
|
let server = start_mock_server().await;
|
|
let response_mock = mount_sse_sequence(
|
|
&server,
|
|
vec![
|
|
paused_response("resp-1", "pause-first-step"),
|
|
paused_response("resp-2", "pause-second-step"),
|
|
sse_completed("resp-3"),
|
|
sse_completed("resp-4"),
|
|
],
|
|
)
|
|
.await;
|
|
let test = step_settings_test().build_with_auto_env(&server).await?;
|
|
let request = start_paused_turn(&test.codex).await?;
|
|
|
|
core_test_support::submit_thread_settings(
|
|
&test.codex,
|
|
ThreadSettingsOverrides {
|
|
model: Some(MODEL_B.to_string()),
|
|
..Default::default()
|
|
},
|
|
)
|
|
.await?;
|
|
assert_eq!(
|
|
submit_turn_settings(
|
|
&test.codex,
|
|
&request.turn_id,
|
|
TurnSettingsUpdate {
|
|
model: Some(MODEL_C.to_string()),
|
|
effort: Some(Some(ReasoningEffort::High)),
|
|
..Default::default()
|
|
}
|
|
)
|
|
.await?,
|
|
TurnSettingsUpdateOutcome::Applied
|
|
);
|
|
answer_paused_turn(&test.codex, &request.turn_id).await?;
|
|
let second_request = wait_for_event_match(&test.codex, |event| match event {
|
|
EventMsg::RequestUserInput(request) => Some(request.clone()),
|
|
_ => None,
|
|
})
|
|
.await;
|
|
assert_eq!(second_request.turn_id, request.turn_id);
|
|
assert_eq!(second_request.call_id, "pause-second-step");
|
|
|
|
test.codex
|
|
.submit(Op::ThreadSettings {
|
|
thread_settings: ThreadSettingsOverrides {
|
|
service_tier: Some(Some(ServiceTier::Fast.request_value().to_string())),
|
|
..Default::default()
|
|
},
|
|
})
|
|
.await?;
|
|
assert_eq!(
|
|
submit_turn_settings(
|
|
&test.codex,
|
|
&request.turn_id,
|
|
TurnSettingsUpdate {
|
|
service_tier: Some(Some(ServiceTier::Fast.request_value().to_string())),
|
|
..Default::default()
|
|
}
|
|
)
|
|
.await?,
|
|
TurnSettingsUpdateOutcome::Applied
|
|
);
|
|
let durable_settings = wait_for_event_match(&test.codex, |event| match event {
|
|
EventMsg::ThreadSettingsApplied(event) => Some(event.thread_settings.clone()),
|
|
_ => None,
|
|
})
|
|
.await;
|
|
assert_eq!(durable_settings.model, MODEL_B);
|
|
assert_eq!(
|
|
durable_settings.reasoning_effort,
|
|
Some(ReasoningEffort::Low)
|
|
);
|
|
assert_eq!(
|
|
durable_settings.service_tier,
|
|
Some(ServiceTier::Fast.request_value().to_string())
|
|
);
|
|
answer_paused_turn(&test.codex, &second_request.turn_id).await?;
|
|
wait_for_event(&test.codex, |event| {
|
|
matches!(event, EventMsg::TurnComplete(_))
|
|
})
|
|
.await;
|
|
test.submit_text_turn("start the next turn").await?;
|
|
|
|
let requests = response_mock.requests();
|
|
assert_eq!(
|
|
requests.iter().map(request_settings).collect::<Vec<_>>(),
|
|
vec![
|
|
json!({
|
|
"model": MODEL_A,
|
|
"reasoning": { "effort": "low", "summary": "concise" },
|
|
"service_tier": null,
|
|
}),
|
|
json!({
|
|
"model": MODEL_C,
|
|
"reasoning": { "effort": "high", "summary": "concise" },
|
|
"service_tier": null,
|
|
}),
|
|
json!({
|
|
"model": MODEL_C,
|
|
"reasoning": { "effort": "high", "summary": "concise" },
|
|
"service_tier": "priority",
|
|
}),
|
|
json!({
|
|
"model": MODEL_B,
|
|
"reasoning": { "effort": "low", "summary": "concise" },
|
|
"service_tier": "priority",
|
|
}),
|
|
]
|
|
);
|
|
assert_eq!(
|
|
requests[..3]
|
|
.iter()
|
|
.map(request_turn_id)
|
|
.collect::<Vec<_>>(),
|
|
vec![request.turn_id.clone(); 3]
|
|
);
|
|
assert_ne!(request_turn_id(&requests[3]), request.turn_id);
|
|
|
|
Ok(())
|
|
}
|
|
|
|
#[tokio::test(flavor = "multi_thread", worker_threads = 2)]
|
|
async fn turn_settings_do_not_target_idle_or_finished_turns() -> Result<()> {
|
|
skip_if_no_network!(Ok(()));
|
|
|
|
let server = start_mock_server().await;
|
|
let response_mock = mount_sse_sequence(
|
|
&server,
|
|
vec![sse_completed("resp-1"), sse_completed("resp-2")],
|
|
)
|
|
.await;
|
|
let test = step_settings_test().build_with_auto_env(&server).await?;
|
|
|
|
for (discarded_model, model) in [(MODEL_C, MODEL_B), (MODEL_A, MODEL_C)] {
|
|
let turn_id = response_mock
|
|
.requests()
|
|
.last()
|
|
.map(request_turn_id)
|
|
.unwrap_or_else(|| "never-started".to_string());
|
|
let before = test.codex.thread_settings_snapshot().await;
|
|
assert_eq!(
|
|
submit_turn_settings(
|
|
&test.codex,
|
|
&turn_id,
|
|
TurnSettingsUpdate {
|
|
model: Some(discarded_model.to_string()),
|
|
..Default::default()
|
|
}
|
|
)
|
|
.await?,
|
|
TurnSettingsUpdateOutcome::TargetUnavailable
|
|
);
|
|
assert_eq!(test.codex.thread_settings_snapshot().await, before);
|
|
core_test_support::submit_thread_settings(
|
|
&test.codex,
|
|
ThreadSettingsOverrides {
|
|
model: Some(model.to_string()),
|
|
..Default::default()
|
|
},
|
|
)
|
|
.await?;
|
|
test.submit_text_turn("start the next turn").await?;
|
|
}
|
|
|
|
assert_eq!(
|
|
response_mock
|
|
.requests()
|
|
.iter()
|
|
.map(request_settings)
|
|
.collect::<Vec<_>>(),
|
|
vec![
|
|
json!({
|
|
"model": MODEL_B,
|
|
"reasoning": { "effort": "low", "summary": "concise" },
|
|
"service_tier": null,
|
|
}),
|
|
json!({
|
|
"model": MODEL_C,
|
|
"reasoning": { "effort": "low", "summary": "concise" },
|
|
"service_tier": null,
|
|
}),
|
|
]
|
|
);
|
|
|
|
Ok(())
|
|
}
|
|
|
|
#[test_case(None, "detailed"; "unset summary follows the destination model")]
|
|
#[test_case(Some(ReasoningSummary::Concise), "concise"; "explicit summary is preserved")]
|
|
#[tokio::test(flavor = "multi_thread", worker_threads = 2)]
|
|
async fn model_activation_uses_destination_metadata_defaults(
|
|
configured_summary: Option<ReasoningSummary>,
|
|
expected_summary: &str,
|
|
) -> Result<()> {
|
|
skip_if_no_network!(Ok(()));
|
|
|
|
let server = start_mock_server().await;
|
|
let response_mock = mount_sse_sequence(
|
|
&server,
|
|
vec![
|
|
paused_response("resp-1", "pause-before-model-change"),
|
|
paused_response("resp-2", "pause-before-restoring-model"),
|
|
sse_completed("resp-3"),
|
|
],
|
|
)
|
|
.await;
|
|
let test = step_settings_test()
|
|
.with_config(move |config| {
|
|
config.model_context_window = None;
|
|
config.model_auto_compact_token_limit = None;
|
|
config.model_reasoning_summary = configured_summary;
|
|
config.service_tier = Some(ServiceTier::Fast.request_value().to_string());
|
|
config.base_instructions = None;
|
|
for model in &mut config
|
|
.model_catalog
|
|
.as_mut()
|
|
.expect("controlled model catalog")
|
|
.models
|
|
{
|
|
model.context_window = Some(256_000);
|
|
model.auto_compact_token_limit = Some(200_000);
|
|
model.default_reasoning_summary = ReasoningSummary::Concise;
|
|
if model.slug == MODEL_B {
|
|
model.context_window = Some(128_000);
|
|
model.auto_compact_token_limit = Some(100_000);
|
|
model.default_reasoning_summary = ReasoningSummary::Detailed;
|
|
model
|
|
.service_tiers
|
|
.retain(|tier| tier.id != ServiceTier::Fast.request_value());
|
|
model
|
|
.model_messages
|
|
.as_mut()
|
|
.expect("model instruction metadata")
|
|
.instructions_template =
|
|
Some("Destination model instructions.".to_string());
|
|
}
|
|
}
|
|
})
|
|
.build_with_auto_env(&server)
|
|
.await?;
|
|
let request = start_paused_turn(&test.codex).await?;
|
|
|
|
assert_eq!(
|
|
submit_turn_settings(
|
|
&test.codex,
|
|
&request.turn_id,
|
|
TurnSettingsUpdate {
|
|
model: Some(MODEL_B.to_string()),
|
|
..Default::default()
|
|
}
|
|
)
|
|
.await?,
|
|
TurnSettingsUpdateOutcome::Applied
|
|
);
|
|
answer_paused_turn(&test.codex, &request.turn_id).await?;
|
|
let paused = wait_for_event_match(&test.codex, |event| match event {
|
|
EventMsg::RequestUserInput(request) => Some(request.clone()),
|
|
_ => None,
|
|
})
|
|
.await;
|
|
// B cannot use the requested tier. Switching back must recover that
|
|
// selection and preserve an unset summary, not reuse B's effective values.
|
|
assert_eq!(
|
|
submit_turn_settings(
|
|
&test.codex,
|
|
&request.turn_id,
|
|
TurnSettingsUpdate {
|
|
model: Some(MODEL_A.to_string()),
|
|
..Default::default()
|
|
}
|
|
)
|
|
.await?,
|
|
TurnSettingsUpdateOutcome::Applied
|
|
);
|
|
answer_paused_turn(&test.codex, &paused.turn_id).await?;
|
|
wait_for_event(&test.codex, |event| match event {
|
|
EventMsg::Error(error) => panic!("model activation failed: {}", error.message),
|
|
EventMsg::TurnComplete(_) => true,
|
|
_ => false,
|
|
})
|
|
.await;
|
|
|
|
let requests = response_mock.requests();
|
|
assert_eq!(
|
|
requests.iter().map(request_settings).collect::<Vec<_>>(),
|
|
vec![
|
|
json!({
|
|
"model": MODEL_A,
|
|
"reasoning": { "effort": "low", "summary": "concise" },
|
|
"service_tier": "priority",
|
|
}),
|
|
json!({
|
|
"model": MODEL_B,
|
|
"reasoning": { "effort": "low", "summary": expected_summary },
|
|
"service_tier": null,
|
|
}),
|
|
json!({
|
|
"model": MODEL_A,
|
|
"reasoning": { "effort": "low", "summary": "concise" },
|
|
"service_tier": "priority",
|
|
}),
|
|
]
|
|
);
|
|
assert_eq!(
|
|
requests.iter().map(request_turn_id).collect::<Vec<_>>(),
|
|
vec![request.turn_id; 3],
|
|
);
|
|
|
|
Ok(())
|
|
}
|
|
|
|
#[tokio::test(flavor = "multi_thread", worker_threads = 2)]
|
|
async fn turn_settings_rejection_preserves_independent_future_settings() -> Result<()> {
|
|
skip_if_no_network!(Ok(()));
|
|
|
|
let server = start_mock_server().await;
|
|
let response_mock = mount_sse_sequence(
|
|
&server,
|
|
vec![
|
|
paused_response("resp-1", "pause-before-node-repl-restriction"),
|
|
sse_completed("resp-2"),
|
|
sse_completed("resp-3"),
|
|
],
|
|
)
|
|
.await;
|
|
let test = step_settings_test()
|
|
.with_config(|config| {
|
|
for model in &mut config
|
|
.model_catalog
|
|
.as_mut()
|
|
.expect("controlled model catalog")
|
|
.models
|
|
{
|
|
model.node_repl_disabled = model.slug == MODEL_B;
|
|
}
|
|
})
|
|
.build_with_auto_env(&server)
|
|
.await?;
|
|
let request = start_paused_turn(&test.codex).await?;
|
|
core_test_support::submit_thread_settings(
|
|
&test.codex,
|
|
ThreadSettingsOverrides {
|
|
model: Some(MODEL_B.to_string()),
|
|
..Default::default()
|
|
},
|
|
)
|
|
.await?;
|
|
let future = test.codex.thread_settings_snapshot().await;
|
|
assert_eq!(
|
|
submit_turn_settings(
|
|
&test.codex,
|
|
&request.turn_id,
|
|
TurnSettingsUpdate {
|
|
model: Some(MODEL_B.to_string()),
|
|
..Default::default()
|
|
}
|
|
)
|
|
.await?,
|
|
TurnSettingsUpdateOutcome::Rejected {
|
|
reason: "the destination changes the admitted node REPL availability restriction"
|
|
.to_string(),
|
|
}
|
|
);
|
|
assert_eq!(test.codex.thread_settings_snapshot().await, future);
|
|
answer_paused_turn(&test.codex, &request.turn_id).await?;
|
|
wait_for_event(&test.codex, |event| {
|
|
matches!(event, EventMsg::TurnComplete(_))
|
|
})
|
|
.await;
|
|
test.submit_text_turn("admit the next turn").await?;
|
|
|
|
let requests = response_mock.requests();
|
|
assert_eq!(
|
|
requests.iter().map(request_settings).collect::<Vec<_>>(),
|
|
[MODEL_A, MODEL_A, MODEL_B]
|
|
.into_iter()
|
|
.map(|model| {
|
|
json!({
|
|
"model": model,
|
|
"reasoning": { "effort": "low", "summary": "concise" },
|
|
"service_tier": null,
|
|
})
|
|
})
|
|
.collect::<Vec<_>>()
|
|
);
|
|
assert_eq!(request_turn_id(&requests[1]), request.turn_id);
|
|
assert_ne!(request_turn_id(&requests[2]), request.turn_id);
|
|
|
|
Ok(())
|
|
}
|
|
|
|
#[tokio::test(flavor = "multi_thread", worker_threads = 2)]
|
|
async fn request_preference_activation_keeps_admitted_model_metadata() -> Result<()> {
|
|
skip_if_no_network!(Ok(()));
|
|
|
|
// This test owns both catalog responses. The generic mock-server helper
|
|
// installs an extra one-shot /models response that would shift the sequence.
|
|
let server = wiremock::MockServer::start().await;
|
|
let response_mock = mount_sse_sequence(
|
|
&server,
|
|
vec![
|
|
paused_response("resp-1", "pause-before-models-refresh"),
|
|
sse_completed("resp-2"),
|
|
sse_completed("resp-3"),
|
|
],
|
|
)
|
|
.await;
|
|
let mut models = step_settings_models();
|
|
for model in &mut models {
|
|
model.default_reasoning_summary = ReasoningSummary::Concise;
|
|
}
|
|
let initial_catalog = mount_models_once(
|
|
&server,
|
|
ModelsResponse {
|
|
models: models.clone(),
|
|
},
|
|
)
|
|
.await;
|
|
let test = step_settings_test()
|
|
.with_auth(CodexAuth::create_dummy_chatgpt_auth_for_testing())
|
|
.with_config(|config| {
|
|
// Use the real refreshable models manager, and retain an unset
|
|
// preference so the selected model's default is observable.
|
|
config.model_catalog = None;
|
|
config.model_reasoning_summary = None;
|
|
})
|
|
.build_with_auto_env(&server)
|
|
.await?;
|
|
let request = start_paused_turn(&test.codex).await?;
|
|
assert_eq!(initial_catalog.requests().len(), 1);
|
|
|
|
for model in &mut models {
|
|
model.default_reasoning_summary = ReasoningSummary::Detailed;
|
|
}
|
|
let refresh = mount_models_once(&server, ModelsResponse { models }).await;
|
|
test.thread_manager
|
|
.get_models_manager()
|
|
.list_models(
|
|
RefreshStrategy::Online,
|
|
codex_core::test_support::default_http_client_factory(),
|
|
)
|
|
.await;
|
|
assert_eq!(refresh.requests().len(), 1);
|
|
|
|
assert_eq!(
|
|
submit_turn_settings(
|
|
&test.codex,
|
|
&request.turn_id,
|
|
TurnSettingsUpdate {
|
|
effort: Some(Some(ReasoningEffort::High)),
|
|
..Default::default()
|
|
}
|
|
)
|
|
.await?,
|
|
TurnSettingsUpdateOutcome::Applied
|
|
);
|
|
answer_paused_turn(&test.codex, &request.turn_id).await?;
|
|
wait_for_event(&test.codex, |event| {
|
|
matches!(event, EventMsg::TurnComplete(_))
|
|
})
|
|
.await;
|
|
test.submit_text_turn("admit the next turn").await?;
|
|
|
|
assert_eq!(
|
|
response_mock
|
|
.requests()
|
|
.iter()
|
|
.map(request_settings)
|
|
.collect::<Vec<_>>(),
|
|
vec![
|
|
json!({
|
|
"model": MODEL_A,
|
|
"reasoning": { "effort": "low", "summary": "concise" },
|
|
"service_tier": null,
|
|
}),
|
|
json!({
|
|
"model": MODEL_A,
|
|
"reasoning": { "effort": "high", "summary": "concise" },
|
|
"service_tier": null,
|
|
}),
|
|
json!({
|
|
"model": MODEL_A,
|
|
"reasoning": { "effort": "low", "summary": "detailed" },
|
|
"service_tier": null,
|
|
}),
|
|
]
|
|
);
|
|
|
|
Ok(())
|
|
}
|