diff --git a/codex-rs/core/src/codex.rs b/codex-rs/core/src/codex.rs
index faab4a8c2b..db70d82cc9 100644
--- a/codex-rs/core/src/codex.rs
+++ b/codex-rs/core/src/codex.rs
@@ -1846,34 +1846,113 @@ impl Session {
fn reconstruct_turn_context_history_from_rollout(
rollout_items: &[RolloutItem],
) -> Vec