mirror of
https://github.com/openai/codex.git
synced 2026-10-30 19:18:49 +00:00
Set rollback reference baseline at resume/fork callsites
This commit is contained in:
@@ -1622,10 +1622,16 @@ impl Session {
|
||||
.reconstruct_history_from_rollout(&turn_context, &rollout_items)
|
||||
.await;
|
||||
let previous_model = reconstructed.previous_model.clone();
|
||||
let reference_context_item = reconstructed
|
||||
.user_turn_baselines
|
||||
.last()
|
||||
.filter(|frame| !frame.invalidated_by_following_compaction)
|
||||
.map(|frame| frame.turn_context_item.clone());
|
||||
let curr = turn_context.model_info.slug.as_str();
|
||||
{
|
||||
let mut state = self.state.lock().await;
|
||||
state.set_user_turn_baselines(reconstructed.user_turn_baselines.clone());
|
||||
state.set_reference_context_item(reference_context_item);
|
||||
state.set_previous_model(previous_model.clone());
|
||||
}
|
||||
|
||||
@@ -1673,6 +1679,7 @@ impl Session {
|
||||
{
|
||||
let mut state = self.state.lock().await;
|
||||
state.set_user_turn_baselines(reconstructed.user_turn_baselines.clone());
|
||||
state.set_reference_context_item(None);
|
||||
state.set_previous_model(reconstructed.previous_model.clone());
|
||||
}
|
||||
|
||||
|
||||
@@ -86,8 +86,6 @@ impl ContextManager {
|
||||
|
||||
pub(crate) fn set_user_turn_baselines(&mut self, baselines: Vec<UserTurnBaselineFrame>) {
|
||||
self.user_turn_baselines = baselines;
|
||||
self.reference_context_item =
|
||||
self.effective_reference_context_item_from_user_turn_baselines();
|
||||
}
|
||||
|
||||
pub(crate) fn user_turn_baselines(&self) -> Vec<UserTurnBaselineFrame> {
|
||||
|
||||
Reference in New Issue
Block a user