diff --git a/codex-rs/app-server/src/request_processors/external_agent_session_import.rs b/codex-rs/app-server/src/request_processors/external_agent_session_import.rs index 0e17cb573..dfa2ea94f 100644 --- a/codex-rs/app-server/src/request_processors/external_agent_session_import.rs +++ b/codex-rs/app-server/src/request_processors/external_agent_session_import.rs @@ -215,6 +215,7 @@ impl ExternalAgentSessionImporter { }, dynamic_tools: Vec::new(), multi_agent_version: Some(MultiAgentVersion::V1), + initial_window_id: uuid::Uuid::now_v7().to_string(), metadata: ThreadPersistenceMetadata { cwd: Some(cwd.clone()), model_provider: model_provider.clone(), diff --git a/codex-rs/app-server/tests/common/rollout.rs b/codex-rs/app-server/tests/common/rollout.rs index 7b389e8fc..ec01dacc4 100644 --- a/codex-rs/app-server/tests/common/rollout.rs +++ b/codex-rs/app-server/tests/common/rollout.rs @@ -202,6 +202,7 @@ fn create_fake_rollout_with_source_and_parent_thread_id( dynamic_tools: None, memory_mode: None, multi_agent_version: None, + context_window: None, }; let payload = serde_json::to_value(SessionMetaLine { meta, @@ -289,6 +290,7 @@ pub fn create_fake_rollout_with_text_elements( dynamic_tools: None, memory_mode: None, multi_agent_version: None, + context_window: None, }; let payload = serde_json::to_value(SessionMetaLine { meta, diff --git a/codex-rs/app-server/tests/suite/conversation_summary.rs b/codex-rs/app-server/tests/suite/conversation_summary.rs index 17f73db3e..d4d0cf061 100644 --- a/codex-rs/app-server/tests/suite/conversation_summary.rs +++ b/codex-rs/app-server/tests/suite/conversation_summary.rs @@ -131,6 +131,7 @@ async fn get_conversation_summary_by_thread_id_reads_pathless_store_thread() -> base_instructions: BaseInstructions::default(), dynamic_tools: Vec::new(), multi_agent_version: None, + initial_window_id: Uuid::now_v7().to_string(), metadata: ThreadPersistenceMetadata { cwd: None, model_provider: "test-provider".to_string(), diff --git a/codex-rs/app-server/tests/suite/v2/remote_thread_store.rs b/codex-rs/app-server/tests/suite/v2/remote_thread_store.rs index 6deec70ff..1a6d1eb4f 100644 --- a/codex-rs/app-server/tests/suite/v2/remote_thread_store.rs +++ b/codex-rs/app-server/tests/suite/v2/remote_thread_store.rs @@ -157,6 +157,7 @@ async fn thread_delete_with_non_local_thread_store_does_not_create_local_persist base_instructions: BaseInstructions::default(), dynamic_tools: Vec::new(), multi_agent_version: None, + initial_window_id: Uuid::now_v7().to_string(), metadata: ThreadPersistenceMetadata { cwd: Some(codex_home.path().to_path_buf()), model_provider: "mock_provider".to_string(), diff --git a/codex-rs/app-server/tests/suite/v2/thread_read.rs b/codex-rs/app-server/tests/suite/v2/thread_read.rs index 76239b4e4..0b4f101c0 100644 --- a/codex-rs/app-server/tests/suite/v2/thread_read.rs +++ b/codex-rs/app-server/tests/suite/v2/thread_read.rs @@ -1367,6 +1367,7 @@ async fn seed_pathless_store_thread( base_instructions: BaseInstructions::default(), dynamic_tools: Vec::new(), multi_agent_version: None, + initial_window_id: Uuid::now_v7().to_string(), metadata: ThreadPersistenceMetadata { cwd: None, model_provider: "test-provider".to_string(), diff --git a/codex-rs/app-server/tests/suite/v2/thread_resume.rs b/codex-rs/app-server/tests/suite/v2/thread_resume.rs index ea7a1a9d4..a67282a82 100644 --- a/codex-rs/app-server/tests/suite/v2/thread_resume.rs +++ b/codex-rs/app-server/tests/suite/v2/thread_resume.rs @@ -2072,6 +2072,7 @@ stream_max_retries = 0 dynamic_tools: None, memory_mode: None, multi_agent_version: None, + context_window: None, }; std::fs::write( &rollout_path, diff --git a/codex-rs/app-server/tests/suite/v2/thread_unarchive.rs b/codex-rs/app-server/tests/suite/v2/thread_unarchive.rs index 46a989545..db8b60d16 100644 --- a/codex-rs/app-server/tests/suite/v2/thread_unarchive.rs +++ b/codex-rs/app-server/tests/suite/v2/thread_unarchive.rs @@ -218,6 +218,7 @@ async fn thread_unarchive_preserves_pathless_store_metadata() -> Result<()> { base_instructions: BaseInstructions::default(), dynamic_tools: Vec::new(), multi_agent_version: None, + initial_window_id: Uuid::now_v7().to_string(), metadata: ThreadPersistenceMetadata { cwd: None, model_provider: "test-provider".to_string(), diff --git a/codex-rs/core/src/personality_migration_tests.rs b/codex-rs/core/src/personality_migration_tests.rs index 57f06317a..dd0da327c 100644 --- a/codex-rs/core/src/personality_migration_tests.rs +++ b/codex-rs/core/src/personality_migration_tests.rs @@ -61,6 +61,7 @@ async fn write_rollout_with_user_event(dir: &Path, thread_id: ThreadId) -> io::R dynamic_tools: None, memory_mode: None, multi_agent_version: None, + context_window: None, }, git: None, }; diff --git a/codex-rs/core/src/session/rollout_reconstruction.rs b/codex-rs/core/src/session/rollout_reconstruction.rs index 6989ff80d..b3fbeb103 100644 --- a/codex-rs/core/src/session/rollout_reconstruction.rs +++ b/codex-rs/core/src/session/rollout_reconstruction.rs @@ -1,5 +1,6 @@ use super::*; use crate::context_manager::is_user_turn_boundary; +use codex_protocol::protocol::SessionContextWindow; use uuid::Uuid; // Return value of `Session::reconstruct_history_from_rollout`, bundling the rebuilt history with @@ -113,6 +114,22 @@ impl Session { // stopping once a surviving replacement-history checkpoint and the required resume metadata // are both known; then replay only the buffered surviving tail forward to preserve exact // history semantics. + let has_legacy_compaction_without_window_number = + rollout_items.iter().any(|item| { + matches!(item, RolloutItem::Compacted(compacted) if compacted.window_number.is_none()) + }); + let initial_window = if has_legacy_compaction_without_window_number { + None + } else { + rollout_items.iter().find_map(|item| match item { + RolloutItem::SessionMeta(session_meta) => session_meta + .meta + .context_window + .as_ref() + .and_then(reconstructed_window_from_session_context_window), + _ => None, + }) + }; let mut base_replacement_history: Option<&[ResponseItem]> = None; let mut previous_turn_settings = None; let mut reference_context_item = TurnReferenceContextItem::NeverSet; @@ -348,7 +365,7 @@ impl Session { reference_context_item }; - let window = window.unwrap_or(ReconstructedWindow { + let window = window.or(initial_window).unwrap_or(ReconstructedWindow { number: fallback_window_number, first_id: None, previous_id: None, @@ -371,3 +388,15 @@ fn parse_uuid_v7(value: &str) -> Option { .ok() .filter(|uuid| uuid.get_version_num() == 7) } + +fn reconstructed_window_from_session_context_window( + context_window: &SessionContextWindow, +) -> Option { + let id = parse_uuid_v7(&context_window.window_id)?; + Some(ReconstructedWindow { + number: 0, + first_id: Some(id), + previous_id: None, + id: Some(id), + }) +} diff --git a/codex-rs/core/src/session/rollout_reconstruction_tests.rs b/codex-rs/core/src/session/rollout_reconstruction_tests.rs index 828c07858..b875e1f49 100644 --- a/codex-rs/core/src/session/rollout_reconstruction_tests.rs +++ b/codex-rs/core/src/session/rollout_reconstruction_tests.rs @@ -9,8 +9,12 @@ use codex_protocol::protocol::CompactedItem; use codex_protocol::protocol::InitialHistory; use codex_protocol::protocol::InterAgentCommunication; use codex_protocol::protocol::ResumedHistory; +use codex_protocol::protocol::SessionContextWindow; +use codex_protocol::protocol::SessionMeta; +use codex_protocol::protocol::SessionMetaLine; use pretty_assertions::assert_eq; use std::path::PathBuf; +use uuid::Uuid; fn user_message(text: &str) -> ResponseItem { ResponseItem::Message { @@ -909,6 +913,116 @@ async fn record_initial_history_resumed_does_not_seed_reference_context_item_aft assert!(session.reference_context_item().await.is_none()); } +#[tokio::test] +async fn reconstruct_history_restores_initial_window_from_session_meta() { + let (session, turn_context) = make_session_and_context().await; + let thread_id = ThreadId::default(); + let initial_window_id = Uuid::now_v7(); + let rollout_items = vec![RolloutItem::SessionMeta(SessionMetaLine { + meta: SessionMeta { + session_id: thread_id.into(), + id: thread_id, + context_window: Some(SessionContextWindow { + window_id: initial_window_id.to_string(), + }), + ..SessionMeta::default() + }, + git: None, + })]; + + let reconstructed = session + .reconstruct_history_from_rollout(&turn_context, &rollout_items) + .await; + + assert_eq!(reconstructed.window_number, 0); + assert_eq!(reconstructed.first_window_id, Some(initial_window_id)); + assert_eq!(reconstructed.previous_window_id, None); + assert_eq!(reconstructed.window_id, Some(initial_window_id)); +} + +#[tokio::test] +async fn reconstruct_history_prefers_compacted_window_over_session_meta() { + let (session, turn_context) = make_session_and_context().await; + let thread_id = ThreadId::default(); + let initial_window_id = Uuid::now_v7(); + let compacted_first_window_id = Uuid::now_v7(); + let compacted_previous_window_id = Uuid::now_v7(); + let compacted_window_id = Uuid::now_v7(); + let rollout_items = vec![ + RolloutItem::SessionMeta(SessionMetaLine { + meta: SessionMeta { + session_id: thread_id.into(), + id: thread_id, + context_window: Some(SessionContextWindow { + window_id: initial_window_id.to_string(), + }), + ..SessionMeta::default() + }, + git: None, + }), + RolloutItem::Compacted(CompactedItem { + message: String::new(), + replacement_history: Some(Vec::new()), + window_number: Some(2), + first_window_id: Some(compacted_first_window_id.to_string()), + previous_window_id: Some(compacted_previous_window_id.to_string()), + window_id: Some(compacted_window_id.to_string()), + }), + ]; + + let reconstructed = session + .reconstruct_history_from_rollout(&turn_context, &rollout_items) + .await; + + assert_eq!(reconstructed.window_number, 2); + assert_eq!( + reconstructed.first_window_id, + Some(compacted_first_window_id) + ); + assert_eq!( + reconstructed.previous_window_id, + Some(compacted_previous_window_id) + ); + assert_eq!(reconstructed.window_id, Some(compacted_window_id)); +} + +#[tokio::test] +async fn reconstruct_history_preserves_legacy_compaction_count_with_session_meta_window() { + let (session, turn_context) = make_session_and_context().await; + let thread_id = ThreadId::default(); + let initial_window_id = Uuid::now_v7(); + let rollout_items = vec![ + RolloutItem::SessionMeta(SessionMetaLine { + meta: SessionMeta { + session_id: thread_id.into(), + id: thread_id, + context_window: Some(SessionContextWindow { + window_id: initial_window_id.to_string(), + }), + ..SessionMeta::default() + }, + git: None, + }), + RolloutItem::Compacted(CompactedItem { + message: "legacy summary".to_string(), + replacement_history: None, + window_number: None, + first_window_id: None, + previous_window_id: None, + window_id: None, + }), + ]; + + let reconstructed = session + .reconstruct_history_from_rollout(&turn_context, &rollout_items) + .await; + + assert_eq!(reconstructed.window_number, 1); + assert_eq!(reconstructed.first_window_id, None); + assert_eq!(reconstructed.previous_window_id, None); + assert_eq!(reconstructed.window_id, None); +} + #[tokio::test] async fn reconstruct_history_legacy_compaction_without_replacement_history_does_not_inject_current_initial_context() { diff --git a/codex-rs/core/src/session/session.rs b/codex-rs/core/src/session/session.rs index 00c5b64e8..ee62155a8 100644 --- a/codex-rs/core/src/session/session.rs +++ b/codex-rs/core/src/session/session.rs @@ -546,6 +546,7 @@ impl Session { SessionId::from(thread_id) } }); + let initial_auto_compact_window_ids = AutoCompactWindowIds::new_initial(); let agent_control = agent_control.with_session_id( session_id, config @@ -585,6 +586,9 @@ impl Session { }, dynamic_tools: session_configuration.dynamic_tools.clone(), multi_agent_version: initial_multi_agent_version, + initial_window_id: initial_auto_compact_window_ids + .window_id + .to_string(), metadata: ThreadPersistenceMetadata { cwd: Some(config.cwd.to_path_buf()), model_provider: config.model_provider_id.clone(), @@ -889,7 +893,10 @@ impl Session { session_configuration.thread_name = thread_name.clone(); validate_config_lock_if_configured(&session_configuration).await?; export_config_lock_if_configured(&session_configuration, thread_id).await?; - let state = SessionState::new(session_configuration.clone()); + let state = SessionState::new_with_auto_compact_window_ids( + session_configuration.clone(), + initial_auto_compact_window_ids, + ); let managed_network_requirements_configured = config .config_layer_stack .requirements_toml() diff --git a/codex-rs/core/src/session/tests.rs b/codex-rs/core/src/session/tests.rs index 869047349..1ebb069f0 100644 --- a/codex-rs/core/src/session/tests.rs +++ b/codex-rs/core/src/session/tests.rs @@ -3844,6 +3844,7 @@ async fn attach_thread_persistence(session: &mut Session) -> PathBuf { base_instructions: BaseInstructions::default(), dynamic_tools: Vec::new(), multi_agent_version: None, + initial_window_id: Uuid::now_v7().to_string(), metadata: ThreadPersistenceMetadata { cwd: Some(config.cwd.to_path_buf()), model_provider: config.model_provider_id.clone(), @@ -6715,6 +6716,7 @@ async fn shutdown_complete_does_not_append_to_thread_store_after_shutdown() { base_instructions: BaseInstructions::default(), dynamic_tools: Vec::new(), multi_agent_version: None, + initial_window_id: Uuid::now_v7().to_string(), metadata: ThreadPersistenceMetadata { cwd: Some(config.cwd.to_path_buf()), model_provider: config.model_provider_id.clone(), diff --git a/codex-rs/core/src/state/auto_compact_window.rs b/codex-rs/core/src/state/auto_compact_window.rs index 419e85be0..6301231df 100644 --- a/codex-rs/core/src/state/auto_compact_window.rs +++ b/codex-rs/core/src/state/auto_compact_window.rs @@ -8,6 +8,17 @@ pub(crate) struct AutoCompactWindowIds { pub(crate) window_id: Uuid, } +impl AutoCompactWindowIds { + pub(crate) fn new_initial() -> Self { + let window_id = Uuid::now_v7(); + Self { + first_window_id: window_id, + previous_window_id: None, + window_id, + } + } +} + #[derive(Debug, Clone, Copy, PartialEq, Eq)] pub(crate) struct AutoCompactWindowSnapshot { pub(crate) prefill_input_tokens: Option, @@ -34,15 +45,10 @@ pub(super) struct AutoCompactWindow { } impl AutoCompactWindow { - pub(super) fn new() -> Self { - let window_id = Uuid::now_v7(); + pub(super) fn new_with_ids(ids: AutoCompactWindowIds) -> Self { Self { window_number: 0, - ids: AutoCompactWindowIds { - first_window_id: window_id, - previous_window_id: None, - window_id, - }, + ids, new_context_window_requested: false, prefill_input_tokens: None, token_budget_reminder_delivered: false, @@ -135,7 +141,7 @@ mod tests { #[test] fn tracks_prefill_and_window_boundaries() { - let mut window = AutoCompactWindow::new(); + let mut window = AutoCompactWindow::new_with_ids(AutoCompactWindowIds::new_initial()); assert_eq!(window.window_number(), 0); let initial_window_id = window.ids().window_id; diff --git a/codex-rs/core/src/state/session.rs b/codex-rs/core/src/state/session.rs index 6444dc8cf..521559486 100644 --- a/codex-rs/core/src/state/session.rs +++ b/codex-rs/core/src/state/session.rs @@ -47,7 +47,18 @@ pub(crate) struct SessionState { impl SessionState { /// Create a new session state mirroring previous `State::default()` semantics. + #[cfg(test)] pub(crate) fn new(session_configuration: SessionConfiguration) -> Self { + Self::new_with_auto_compact_window_ids( + session_configuration, + AutoCompactWindowIds::new_initial(), + ) + } + + pub(crate) fn new_with_auto_compact_window_ids( + session_configuration: SessionConfiguration, + auto_compact_window_ids: AutoCompactWindowIds, + ) -> Self { let history = ContextManager::new(); Self { session_configuration, @@ -57,7 +68,7 @@ impl SessionState { mcp_dependency_prompted: HashSet::new(), additional_context: AdditionalContextStore::default(), previous_turn_settings: None, - auto_compact_window: AutoCompactWindow::new(), + auto_compact_window: AutoCompactWindow::new_with_ids(auto_compact_window_ids), startup_prewarm: None, current_time_reminder: CurrentTimeReminderState::default(), active_connector_selection: HashSet::new(), diff --git a/codex-rs/core/tests/suite/personality_migration.rs b/codex-rs/core/tests/suite/personality_migration.rs index bbab96181..ca1d14872 100644 --- a/codex-rs/core/tests/suite/personality_migration.rs +++ b/codex-rs/core/tests/suite/personality_migration.rs @@ -77,6 +77,7 @@ async fn write_rollout_with_user_event(dir: &Path, thread_id: ThreadId) -> io::R dynamic_tools: None, memory_mode: None, multi_agent_version: None, + context_window: None, }, git: None, }; @@ -128,6 +129,7 @@ async fn write_rollout_with_meta_only(dir: &Path, thread_id: ThreadId) -> io::Re dynamic_tools: None, memory_mode: None, multi_agent_version: None, + context_window: None, }, git: None, }; diff --git a/codex-rs/core/tests/suite/sqlite_state.rs b/codex-rs/core/tests/suite/sqlite_state.rs index 36a51f80c..ddfc8ef21 100644 --- a/codex-rs/core/tests/suite/sqlite_state.rs +++ b/codex-rs/core/tests/suite/sqlite_state.rs @@ -374,6 +374,7 @@ async fn backfill_scans_existing_rollouts() -> Result<()> { dynamic_tools: None, memory_mode: None, multi_agent_version: None, + context_window: None, }, git: None, }; diff --git a/codex-rs/protocol/src/protocol.rs b/codex-rs/protocol/src/protocol.rs index 52464f706..cc0d25807 100644 --- a/codex-rs/protocol/src/protocol.rs +++ b/codex-rs/protocol/src/protocol.rs @@ -2918,6 +2918,18 @@ pub enum MultiAgentVersion { V2, } +#[derive(Serialize, Deserialize, Clone, Debug, PartialEq, Eq, JsonSchema, TS)] +pub struct SessionContextWindow { + /// UUIDv7 identity of this context window. + pub window_id: String, +} + +impl SessionContextWindow { + pub fn new(window_id: String) -> Self { + Self { window_id } + } +} + /// SessionMeta contains session-level data that doesn't correspond to a specific turn. /// /// NOTE: There used to be an `instructions` field here, which stored user_instructions, but we @@ -2964,6 +2976,9 @@ pub struct SessionMeta { pub memory_mode: Option, #[serde(skip_serializing_if = "Option::is_none")] pub multi_agent_version: Option, + /// Initial context-window identity for consumers that tail rollout JSONL before compaction. + #[serde(default, skip_serializing_if = "Option::is_none")] + pub context_window: Option, } impl Default for SessionMeta { @@ -2988,6 +3003,7 @@ impl Default for SessionMeta { dynamic_tools: None, memory_mode: None, multi_agent_version: None, + context_window: None, } } } diff --git a/codex-rs/rollout/src/compression_tests.rs b/codex-rs/rollout/src/compression_tests.rs index 60f47d078..a4db9a965 100644 --- a/codex-rs/rollout/src/compression_tests.rs +++ b/codex-rs/rollout/src/compression_tests.rs @@ -474,6 +474,7 @@ fn write_rollout(path: &std::path::Path, thread_id: ThreadId, message: &str) -> dynamic_tools: None, memory_mode: None, multi_agent_version: None, + context_window: None, }, git: None, }; diff --git a/codex-rs/rollout/src/metadata_tests.rs b/codex-rs/rollout/src/metadata_tests.rs index 5f923f16d..0744e7780 100644 --- a/codex-rs/rollout/src/metadata_tests.rs +++ b/codex-rs/rollout/src/metadata_tests.rs @@ -51,6 +51,7 @@ async fn extract_metadata_from_rollout_uses_session_meta() { dynamic_tools: None, memory_mode: None, multi_agent_version: None, + context_window: None, }; let session_meta_line = SessionMetaLine { meta: session_meta, @@ -107,6 +108,7 @@ async fn extract_metadata_from_rollout_returns_latest_memory_mode() { dynamic_tools: None, memory_mode: None, multi_agent_version: None, + context_window: None, }; let polluted_meta = SessionMeta { memory_mode: Some("polluted".to_string()), @@ -375,6 +377,7 @@ fn write_rollout_in_sessions_with_cwd( dynamic_tools: None, memory_mode: None, multi_agent_version: None, + context_window: None, }; let session_meta_line = SessionMetaLine { meta: session_meta, diff --git a/codex-rs/rollout/src/recorder.rs b/codex-rs/rollout/src/recorder.rs index 3229f5450..53bf115d8 100644 --- a/codex-rs/rollout/src/recorder.rs +++ b/codex-rs/rollout/src/recorder.rs @@ -57,6 +57,7 @@ use codex_protocol::protocol::MultiAgentVersion; use codex_protocol::protocol::ResumedHistory; use codex_protocol::protocol::RolloutItem; use codex_protocol::protocol::RolloutLine; +use codex_protocol::protocol::SessionContextWindow; use codex_protocol::protocol::SessionMeta; use codex_protocol::protocol::SessionMetaLine; use codex_protocol::protocol::SessionSource; @@ -91,6 +92,7 @@ pub enum RolloutRecorderParams { base_instructions: BaseInstructions, dynamic_tools: Vec, multi_agent_version: Option, + initial_window_id: Option, }, Resume { path: PathBuf, @@ -178,6 +180,7 @@ impl RolloutRecorderParams { base_instructions, dynamic_tools, multi_agent_version: None, + initial_window_id: None, } } @@ -202,6 +205,17 @@ impl RolloutRecorderParams { self } + pub fn with_initial_window_id(mut self, initial_window_id: String) -> Self { + if let Self::Create { + initial_window_id: window_id, + .. + } = &mut self + { + *window_id = Some(initial_window_id); + } + self + } + pub fn resume(path: PathBuf) -> Self { Self::Resume { path } } @@ -715,6 +729,7 @@ impl RolloutRecorder { base_instructions, dynamic_tools, multi_agent_version, + initial_window_id, } => { let log_file_info = precompute_log_file_info(config, conversation_id)?; let path = log_file_info.path.clone(); @@ -752,6 +767,7 @@ impl RolloutRecorder { }, memory_mode: (!config.generate_memories()).then_some("disabled".to_string()), multi_agent_version, + context_window: initial_window_id.map(SessionContextWindow::new), }; (None, Some(log_file_info), path, Some(session_meta)) diff --git a/codex-rs/rollout/src/recorder_tests.rs b/codex-rs/rollout/src/recorder_tests.rs index 738fde899..23618cb81 100644 --- a/codex-rs/rollout/src/recorder_tests.rs +++ b/codex-rs/rollout/src/recorder_tests.rs @@ -103,6 +103,7 @@ async fn state_db_init_backfills_before_returning() -> anyhow::Result<()> { dynamic_tools: None, memory_mode: None, multi_agent_version: None, + context_window: None, }, git: None, }; @@ -375,6 +376,7 @@ async fn recorder_materializes_on_flush_with_pending_items() -> std::io::Result< let config = test_config(home.path()); let session_id = SessionId::default(); let thread_id = ThreadId::new(); + let initial_window_id = Uuid::now_v7().to_string(); let recorder = RolloutRecorder::new( &config, RolloutRecorderParams::new( @@ -386,7 +388,8 @@ async fn recorder_materializes_on_flush_with_pending_items() -> std::io::Result< BaseInstructions::default(), Vec::new(), ) - .with_session_id(session_id), + .with_session_id(session_id) + .with_initial_window_id(initial_window_id.clone()), ) .await?; @@ -437,6 +440,13 @@ async fn recorder_materializes_on_flush_with_pending_items() -> std::io::Result< panic!("expected session metadata in rollout"); }; assert_eq!(session_meta.meta.session_id, session_id); + assert_eq!( + session_meta + .meta + .context_window + .map(|window| window.window_id), + Some(initial_window_id) + ); let buffered_idx = text .find("buffered-event") .expect("buffered event in rollout"); diff --git a/codex-rs/rollout/src/session_index_tests.rs b/codex-rs/rollout/src/session_index_tests.rs index 33e4ae4a5..7f57c2ba0 100644 --- a/codex-rs/rollout/src/session_index_tests.rs +++ b/codex-rs/rollout/src/session_index_tests.rs @@ -43,6 +43,7 @@ fn write_rollout_with_metadata(path: &Path, thread_id: ThreadId) -> std::io::Res dynamic_tools: None, memory_mode: None, multi_agent_version: None, + context_window: None, }, git: None, }), diff --git a/codex-rs/rollout/src/state_db_tests.rs b/codex-rs/rollout/src/state_db_tests.rs index 4b2ae4f25..5712b65ce 100644 --- a/codex-rs/rollout/src/state_db_tests.rs +++ b/codex-rs/rollout/src/state_db_tests.rs @@ -176,6 +176,7 @@ fn write_rollout_with_user_message( dynamic_tools: None, memory_mode: None, multi_agent_version: None, + context_window: None, }, git: None, }), diff --git a/codex-rs/rollout/src/tests.rs b/codex-rs/rollout/src/tests.rs index 983fff824..e6a3c728e 100644 --- a/codex-rs/rollout/src/tests.rs +++ b/codex-rs/rollout/src/tests.rs @@ -1290,6 +1290,7 @@ async fn test_updated_at_uses_file_mtime() -> Result<()> { dynamic_tools: None, memory_mode: None, multi_agent_version: None, + context_window: None, }, git: None, }), diff --git a/codex-rs/state/src/extract.rs b/codex-rs/state/src/extract.rs index aa1dd0403..5b8388bf8 100644 --- a/codex-rs/state/src/extract.rs +++ b/codex-rs/state/src/extract.rs @@ -341,6 +341,7 @@ mod tests { dynamic_tools: None, memory_mode: None, multi_agent_version: None, + context_window: None, }, git: None, }), @@ -532,6 +533,7 @@ mod tests { dynamic_tools: None, memory_mode: None, multi_agent_version: None, + context_window: None, }, git: None, }), diff --git a/codex-rs/state/src/runtime/threads.rs b/codex-rs/state/src/runtime/threads.rs index a448dce56..e4181d47d 100644 --- a/codex-rs/state/src/runtime/threads.rs +++ b/codex-rs/state/src/runtime/threads.rs @@ -1975,6 +1975,7 @@ mod tests { dynamic_tools: None, memory_mode: Some("polluted".to_string()), multi_agent_version: None, + context_window: None, }, git: None, })]; @@ -2037,6 +2038,7 @@ mod tests { dynamic_tools: None, memory_mode: None, multi_agent_version: None, + context_window: None, }, git: Some(GitInfo { commit_hash: Some(codex_git_utils::GitSha::new("rollout-sha")), diff --git a/codex-rs/thread-store/src/in_memory.rs b/codex-rs/thread-store/src/in_memory.rs index 84e0484f7..d7d85c3cf 100644 --- a/codex-rs/thread-store/src/in_memory.rs +++ b/codex-rs/thread-store/src/in_memory.rs @@ -10,6 +10,7 @@ use codex_protocol::ThreadId; use codex_protocol::models::PermissionProfile; use codex_protocol::protocol::AskForApproval; use codex_protocol::protocol::RolloutItem; +use codex_protocol::protocol::SessionContextWindow; use codex_protocol::protocol::SessionMeta; use codex_protocol::protocol::SessionMetaLine; use codex_protocol::protocol::ThreadMemoryMode; @@ -120,6 +121,7 @@ mod tests { base_instructions: BaseInstructions::default(), dynamic_tools: Vec::new(), multi_agent_version: None, + initial_window_id: uuid::Uuid::now_v7().to_string(), metadata: ThreadPersistenceMetadata { cwd: None, model_provider: "test-provider".to_string(), @@ -248,6 +250,7 @@ impl InMemoryThreadStore { memory_mode: matches!(params.metadata.memory_mode, ThreadMemoryMode::Disabled) .then_some("disabled".to_string()), multi_agent_version: params.multi_agent_version, + context_window: Some(SessionContextWindow::new(params.initial_window_id.clone())), ..SessionMeta::default() }; state diff --git a/codex-rs/thread-store/src/local/create_thread.rs b/codex-rs/thread-store/src/local/create_thread.rs index 51a96b081..36c3cdc13 100644 --- a/codex-rs/thread-store/src/local/create_thread.rs +++ b/codex-rs/thread-store/src/local/create_thread.rs @@ -37,7 +37,8 @@ pub(super) async fn create_thread( params.dynamic_tools, ) .with_session_id(params.session_id) - .with_multi_agent_version(params.multi_agent_version), + .with_multi_agent_version(params.multi_agent_version) + .with_initial_window_id(params.initial_window_id), ) .await .map_err(|err| ThreadStoreError::Internal { diff --git a/codex-rs/thread-store/src/local/mod.rs b/codex-rs/thread-store/src/local/mod.rs index b96e05cfd..d0a4d6518 100644 --- a/codex-rs/thread-store/src/local/mod.rs +++ b/codex-rs/thread-store/src/local/mod.rs @@ -1133,6 +1133,7 @@ mod tests { base_instructions: BaseInstructions::default(), dynamic_tools: Vec::new(), multi_agent_version: None, + initial_window_id: uuid::Uuid::now_v7().to_string(), metadata: thread_metadata(), } } diff --git a/codex-rs/thread-store/src/types.rs b/codex-rs/thread-store/src/types.rs index 2a82e4251..c18f39882 100644 --- a/codex-rs/thread-store/src/types.rs +++ b/codex-rs/thread-store/src/types.rs @@ -85,6 +85,8 @@ pub struct CreateThreadParams { pub dynamic_tools: Vec, /// Multi-agent runtime selected when the thread was created. pub multi_agent_version: Option, + /// Initial context-window identity captured when the thread was created. + pub initial_window_id: String, /// Metadata captured for the newly created thread. pub metadata: ThreadPersistenceMetadata, }