diff --git a/codex-rs/rollout/src/policy.rs b/codex-rs/rollout/src/policy.rs index 8631034b3..aeb92ade1 100644 --- a/codex-rs/rollout/src/policy.rs +++ b/codex-rs/rollout/src/policy.rs @@ -102,10 +102,12 @@ fn event_msg_persistence_mode(ev: &EventMsg) -> Option { | EventMsg::ContextCompacted(_) | EventMsg::EnteredReviewMode(_) | EventMsg::ExitedReviewMode(_) + | EventMsg::McpToolCallEnd(_) | EventMsg::ThreadRolledBack(_) | EventMsg::TurnAborted(_) | EventMsg::TurnStarted(_) | EventMsg::TurnComplete(_) + | EventMsg::WebSearchEnd(_) | EventMsg::ImageGenerationEnd(_) => Some(EventPersistenceMode::Limited), EventMsg::ItemCompleted(event) => { // Plan items are derived from streaming tags and are not part of the @@ -117,14 +119,9 @@ fn event_msg_persistence_mode(ev: &EventMsg) -> Option { None } } - EventMsg::McpToolCallEnd(event) if event.mcp_app_resource_uri.is_some() => { - Some(EventPersistenceMode::Limited) - } EventMsg::Error(_) | EventMsg::GuardianAssessment(_) - | EventMsg::WebSearchEnd(_) | EventMsg::ExecCommandEnd(_) - | EventMsg::McpToolCallEnd(_) | EventMsg::ViewImageToolCall(_) | EventMsg::CollabAgentSpawnEnd(_) | EventMsg::CollabAgentInteractionEnd(_)