chore: merge current develop into ticket source

This commit is contained in:
2026-08-19 11:54:35 +09:00
4 changed files with 245 additions and 12 deletions
+113 -1
View File
@@ -271,6 +271,9 @@ pub struct App {
pub quit_confirm: Option<std::time::Instant>,
/// Full display history in render order.
pub blocks: Vec<Block>,
/// Turn/protocol errors retained when a real `SegmentStart` replaces the
/// replayable conversation rows during segment rotation.
run_error_messages: Vec<String>,
pub scroll: Scroll,
pub mode: Mode,
pub cache: FileCache,
@@ -347,6 +350,7 @@ impl App {
quit: false,
quit_confirm: None,
blocks: Vec::new(),
run_error_messages: Vec::new(),
scroll: Scroll::default(),
mode: Mode::Normal,
cache: FileCache::new(),
@@ -812,6 +816,15 @@ impl App {
});
}
fn push_run_error(&mut self, message: String) {
self.run_error_messages.push(message.clone());
self.blocks.push(Block::Alert {
level: AlertLevel::Error,
source: AlertSource::Worker,
message,
});
}
fn handle_error(&mut self, code: ErrorCode, message: String) {
let text = format!("[{code:?}] {message}");
let was_applying = if let Some(picker) = self.rewind_picker.as_mut() {
@@ -829,7 +842,7 @@ impl App {
Duration::from_secs(6),
);
}
self.push_error(text);
self.push_run_error(text);
}
fn rewind_submit_pending(&self) -> bool {
@@ -981,8 +994,16 @@ impl App {
self.assistant_streaming = false;
}
Event::SegmentRotated { entry } => {
let retained_run_errors = self.run_error_messages.clone();
self.reset_for_rotation();
self.apply_log_entry_raw(&entry);
for message in retained_run_errors {
self.blocks.push(Block::Alert {
level: AlertLevel::Error,
source: AlertSource::Worker,
message,
});
}
self.assistant_streaming = false;
}
Event::SystemItem { item } => {
@@ -2005,6 +2026,7 @@ impl App {
entries: &[serde_json::Value],
greeting: Option<protocol::Greeting>,
) {
self.run_error_messages.clear();
self.turn_index = 0;
self.blocks.clear();
self.cache = FileCache::new();
@@ -2081,6 +2103,9 @@ impl App {
} if domain == "yoi.compaction" => {
self.apply_compaction_extension(&payload);
}
session_store::LogEntry::RunErrored { message, .. } => {
self.push_run_error(message);
}
// Non-history-bearing variants don't affect the block view.
_ => {}
}
@@ -3261,6 +3286,93 @@ mod completion_flow_tests {
));
}
#[test]
fn snapshot_replaces_live_error_with_one_durable_run_error_block() {
let mut app = App::new("test".into());
app.handle_worker_event(Event::Error {
code: ErrorCode::ProviderError,
message: "provider unavailable".into(),
});
app.handle_worker_event(Event::Status {
status: WorkerStatus::Idle,
});
let live_errors = app
.blocks
.iter()
.filter_map(|block| match block {
Block::Alert {
level: AlertLevel::Error,
source: AlertSource::Worker,
message,
} => Some(message.as_str()),
_ => None,
})
.collect::<Vec<_>>();
assert_eq!(live_errors, ["[ProviderError] provider unavailable"]);
let run_errored = session_store::LogEntry::RunErrored {
ts: 3,
interrupted: false,
message: "provider unavailable".into(),
};
app.handle_worker_event(Event::Snapshot {
greeting: test_greeting(),
entries: vec![serde_json::to_value(run_errored).unwrap()],
status: WorkerStatus::Idle,
in_flight: Default::default(),
});
let errors = app
.blocks
.iter()
.filter_map(|block| match block {
Block::Alert {
level: AlertLevel::Error,
source: AlertSource::Worker,
message,
} => Some(message.as_str()),
_ => None,
})
.collect::<Vec<_>>();
assert_eq!(errors, ["provider unavailable"]);
}
#[test]
fn segment_rotation_retains_live_error_across_real_segment_start() {
let mut app = App::new("test".into());
app.handle_worker_event(Event::Error {
code: ErrorCode::ProviderError,
message: "provider unavailable".into(),
});
let segment_start = session_store::LogEntry::SegmentStart {
ts: 5,
session_id: uuid::Uuid::nil(),
system_prompt: None,
config: Default::default(),
history: Vec::new(),
forked_from: None,
compacted_from: None,
};
app.handle_worker_event(Event::SegmentRotated {
entry: serde_json::to_value(segment_start).unwrap(),
});
let errors = app
.blocks
.iter()
.filter_map(|block| match block {
Block::Alert {
level: AlertLevel::Error,
source: AlertSource::Worker,
message,
} => Some(message.as_str()),
_ => None,
})
.collect::<Vec<_>>();
assert_eq!(errors, ["[ProviderError] provider unavailable"]);
}
#[test]
fn snapshot_in_flight_blocks_continue_with_live_deltas() {
let mut app = App::new("test".into());