`UsageDelta` gains `prefillMs`, llama-server's own `timings.prompt_ms`, so the footer under a finished reply is "read 9.5s · 50.3 tok/s · 3:00 PM". Prefill is the half of a turn that was invisible and is often the larger: measured on the 0.6B here, 1m 4s for the first turn after a model loads against 22ms for the next, whose prompt the server still had cached. The clock moves to the end of the line. Everything in front of it is a provider's own measurement, so a session on another provider has fewer of them or none, and a reader who has learned where the time is should not have to find it again because the model changed. The costs grow leftwards into the space instead, and a test asserts every shape of the line ends with the same thing. Verified on the emulator against a real llama session: three replies reading "read 1m 4s · 193 tok/s · 3:54 PM", "read 25ms · 308 tok/s · 3:54 PM" and "read 22ms · 194 tok/s · 3:54 PM", with the clock in one column. Co-Authored-By: Claude Opus 5 <noreply@anthropic.com>
1023 lines
40 KiB
Rust
1023 lines
40 KiB
Rust
//! Append-only JSONL event log, one per session, with monotonically
|
|
//! increasing sequence numbers -- the phone's resume cursor.
|
|
//!
|
|
//! One line per event: `{"seq":N,"ts":...,"type":...,...}`. The writer
|
|
//! assigns sequence numbers; readers replay everything after a cursor.
|
|
//! Reopening an existing file continues the numbering, which is what makes
|
|
//! a backend restart invisible to a phone holding a cursor.
|
|
|
|
use std::fs::{File, OpenOptions};
|
|
use std::io::Write;
|
|
use std::ops::Range;
|
|
use std::os::unix::fs::OpenOptionsExt;
|
|
use std::path::Path;
|
|
|
|
use anyhow::{Context, Result};
|
|
use serde::{Deserialize, Serialize};
|
|
|
|
use super::driver::{Event, SessionStatus, context_after, context_limit_after};
|
|
|
|
/// One transcript line: an [`Event`] plus its position and time. The event
|
|
/// is flattened so the wire shape stays one flat object.
|
|
#[derive(Debug, Clone, PartialEq, Serialize, Deserialize)]
|
|
pub struct SeqEvent {
|
|
pub seq: u64,
|
|
/// Epoch seconds.
|
|
pub ts: f64,
|
|
#[serde(flatten)]
|
|
pub event: Event,
|
|
}
|
|
|
|
pub struct Transcript {
|
|
file: File,
|
|
next_seq: u64,
|
|
last_status: Option<SessionStatus>,
|
|
last_activity: Option<f64>,
|
|
context_tokens: Option<u64>,
|
|
context_limit: Option<u64>,
|
|
}
|
|
|
|
impl Transcript {
|
|
/// Opens (or creates) the log at `path`, continuing the sequence from
|
|
/// the last line if one exists.
|
|
pub fn open(path: &Path) -> Result<Self> {
|
|
// One pass for all three answers. They are wanted at the same moment
|
|
// by the same caller, and reading the file again for each doubled the
|
|
// cost of starting every session.
|
|
let existing = read_after(path, 0)?;
|
|
let last_seq = existing.last().map(|entry| entry.seq).unwrap_or(0);
|
|
let last_status = existing.iter().rev().find_map(|entry| match entry.event {
|
|
Event::Status { state } => Some(state),
|
|
_ => None,
|
|
});
|
|
// Owner-only: a transcript is the whole conversation, including
|
|
// whatever the session read, wrote, or was told.
|
|
let file = OpenOptions::new()
|
|
.create(true)
|
|
.append(true)
|
|
.mode(0o600)
|
|
.open(path)
|
|
.with_context(|| format!("open transcript {}", path.display()))?;
|
|
Ok(Self {
|
|
file,
|
|
next_seq: last_seq + 1,
|
|
last_status,
|
|
last_activity: existing.last().map(|entry| entry.ts),
|
|
// Folded rather than read off the newest usage entry: a clear or
|
|
// a compaction after it is what the answer is, and those events
|
|
// carry no usage of their own.
|
|
context_tokens: existing
|
|
.iter()
|
|
.fold(None, |current, entry| context_after(current, &entry.event)),
|
|
// The same fold for the same reason: a session whose process has
|
|
// since exited has no window, and the newest `ContextWindow` line
|
|
// alone would not know that.
|
|
context_limit: existing.iter().fold(None, |current, entry| {
|
|
context_limit_after(current, &entry.event)
|
|
}),
|
|
})
|
|
}
|
|
|
|
/// The state the session was last reported to be in, as of opening.
|
|
///
|
|
/// Read from the file rather than assumed, because a server that has just
|
|
/// restarted has been told nothing. Assuming idle claimed a session was
|
|
/// waiting for you when it had exited hours earlier.
|
|
///
|
|
/// `None` for a transcript that never carried a status.
|
|
pub fn last_status(&self) -> Option<SessionStatus> {
|
|
self.last_status
|
|
}
|
|
|
|
/// When this session last did anything, as of opening.
|
|
///
|
|
/// Read from the file for the reason [`Transcript::last_status`] is, and it
|
|
/// is the same mistake in the other direction: taking the clock instead
|
|
/// said every session it relaunched had been active this second. On the
|
|
/// phone that is every row reading "just now" and the list -- sorted by
|
|
/// this -- in an order that means nothing.
|
|
///
|
|
/// `None` for a transcript with no lines, which is a session that genuinely
|
|
/// has not done anything. Its caller answers with when the session was
|
|
/// created, not with the clock.
|
|
pub fn last_activity(&self) -> Option<f64> {
|
|
self.last_activity
|
|
}
|
|
|
|
/// How much context the session was holding, as of opening.
|
|
///
|
|
/// `None` for a transcript nothing has been measured in. That is not zero:
|
|
/// a server that has just restarted has been told nothing, and answering
|
|
/// zero would draw an empty context for a conversation that may be nearly
|
|
/// full.
|
|
pub fn context_tokens(&self) -> Option<u64> {
|
|
self.context_tokens
|
|
}
|
|
|
|
/// What that figure is out of, as of opening, and `None` where this
|
|
/// session's provider does not say.
|
|
pub fn context_limit(&self) -> Option<u64> {
|
|
self.context_limit
|
|
}
|
|
|
|
/// Appends `event`, assigning it the next sequence number. Flushed per
|
|
/// event: each line is tiny, and the transcript is the source of truth a
|
|
/// crash must not lose the tail of.
|
|
pub fn append(&mut self, event: Event, ts: f64) -> Result<SeqEvent> {
|
|
let entry = SeqEvent {
|
|
seq: self.next_seq,
|
|
ts,
|
|
event,
|
|
};
|
|
let mut line = serde_json::to_string(&entry).context("serialize event")?;
|
|
line.push('\n');
|
|
self.file
|
|
.write_all(line.as_bytes())
|
|
.context("append to transcript")?;
|
|
self.next_seq += 1;
|
|
Ok(entry)
|
|
}
|
|
}
|
|
|
|
/// A window of the transcript ending just before `before`, newest-biased.
|
|
///
|
|
/// The screen opens on the end of a conversation, and the end is all it can
|
|
/// show at once. Replaying the whole file to get there costs one network frame
|
|
/// per event -- on an 863-event import that was several seconds of messages
|
|
/// arriving oldest-first, which reads as the app loading top-down.
|
|
///
|
|
/// `before` pages backwards for history somebody actually scrolls to. Only the
|
|
/// window is parsed; see [`Indexed`] for why that is the whole cost.
|
|
///
|
|
/// `after` is a floor: nothing at or below it is returned, and the page stops
|
|
/// there rather than at `limit`. A phone holding a cached run passes the end of
|
|
/// what it already has, so the page is exactly the gap and never overlaps its
|
|
/// copy -- an overlap it cannot store, since a coalesced event cannot be cut at
|
|
/// a seq inside its own delta run.
|
|
pub fn read_window(
|
|
path: &Path,
|
|
before: Option<u64>,
|
|
after: Option<u64>,
|
|
limit: usize,
|
|
coalesce: bool,
|
|
) -> Result<Vec<SeqEvent>> {
|
|
let Some(indexed) = Indexed::read(path)? else {
|
|
return Ok(Vec::new());
|
|
};
|
|
let end = match before {
|
|
Some(before) => indexed.first_at_or_after(before)?,
|
|
None => indexed.lines.len(),
|
|
};
|
|
let start = match after {
|
|
Some(after) => indexed.first_at_or_after(after.saturating_add(1))?,
|
|
None => 0,
|
|
};
|
|
// A floor above the window is an empty page, not a walk backwards past it.
|
|
let start = start.min(end);
|
|
// Coalescing counts *rows*, not events, and would misread the newest
|
|
// window: a message still streaming there would fold to one event whose seq
|
|
// is its first delta, and the phone resumes its live stream from the newest
|
|
// seq it applied -- so the deltas the coalesced event hid would replay and
|
|
// double. Only settled history (`before` set) is safe.
|
|
if coalesce && before.is_some() {
|
|
indexed.parse_coalesced(start, end, limit)
|
|
} else {
|
|
indexed.parse(start.max(end.saturating_sub(limit))..end)
|
|
}
|
|
}
|
|
|
|
/// How far behind a reconnecting subscriber can be and still be handed the
|
|
/// backlog one event at a time.
|
|
///
|
|
/// Past this it is served better by rebuilding from the newest window. The
|
|
/// events are the same either way; what differs is that one arrives as a single
|
|
/// window and the other as thousands of frames a screen renders one by one. Set
|
|
/// well above a screenful so an ordinary blip still streams continuously.
|
|
pub const CATCH_UP_LIMIT: usize = 200;
|
|
|
|
/// What a subscriber asking for "everything after my cursor" gets back.
|
|
///
|
|
/// Two answers rather than one list, because they mean different things to the
|
|
/// screen holding the cursor: one continues what it has, the other replaces it.
|
|
/// Collapsing them would leave the client splicing a window onto rows it has no
|
|
/// way to know are no longer adjacent -- a seam that looks like ordinary output.
|
|
#[derive(Debug, Clone, PartialEq)]
|
|
pub enum CatchUp {
|
|
/// The events after the cursor, continuing what the subscriber holds.
|
|
Continue(Vec<SeqEvent>),
|
|
/// The subscriber was further behind than [`CATCH_UP_LIMIT`]: the newest
|
|
/// window, replacing whatever it holds. Earlier history is still there to be
|
|
/// paged backwards through.
|
|
Restart(Vec<SeqEvent>),
|
|
}
|
|
|
|
/// Everything after `after`, or the newest `limit` when that is more.
|
|
///
|
|
/// The window is chosen before anything is parsed, which matters most in the
|
|
/// case that looks least interesting: a subscriber with no cursor asks for the
|
|
/// whole conversation and is handed the last [`CATCH_UP_LIMIT`] events of it,
|
|
/// so parsing the discarded prefix is the whole file's work for a screenful.
|
|
pub fn catch_up(path: &Path, after: u64, limit: usize) -> Result<CatchUp> {
|
|
let Some(indexed) = Indexed::read(path)? else {
|
|
return Ok(CatchUp::Continue(Vec::new()));
|
|
};
|
|
let end = indexed.lines.len();
|
|
let start = indexed.first_at_or_after(after.saturating_add(1))?;
|
|
if end - start > limit {
|
|
return Ok(CatchUp::Restart(indexed.parse(end - limit..end)?));
|
|
}
|
|
Ok(CatchUp::Continue(indexed.parse(start..end)?))
|
|
}
|
|
|
|
/// Replays every event with `seq > after`, oldest first. A missing file is
|
|
/// an empty transcript, not an error -- the session just hasn't produced an
|
|
/// event yet.
|
|
pub fn read_after(path: &Path, after: u64) -> Result<Vec<SeqEvent>> {
|
|
let Some(indexed) = Indexed::read(path)? else {
|
|
return Ok(Vec::new());
|
|
};
|
|
let start = indexed.first_at_or_after(after.saturating_add(1))?;
|
|
indexed.parse(start..indexed.lines.len())
|
|
}
|
|
|
|
/// The transcript's lines located but not read, so a reader can find the range
|
|
/// it wants and parse only that.
|
|
///
|
|
/// Both readers above want a *range* of the file, and both used to reach it by
|
|
/// parsing every line and discarding the ones outside it -- the cost that grows
|
|
/// with the conversation rather than with the answer. Measured on a 21 MB,
|
|
/// 24,000-event transcript, one page took **500 ms of server time to return
|
|
/// 600 KB**, and the same 500 ms whichever page was asked for. A phone paging
|
|
/// back pays it per page, and every stream reconnect pays it again to discover
|
|
/// there is nothing new.
|
|
///
|
|
/// Sequence numbers only ever increase, so the boundary of a range is a
|
|
/// bisection: this parses one line per halving, and the caller parses only what
|
|
/// it returns. The file is still read whole, which is a deliberate stop --
|
|
/// going further means a chunked backwards reader, and locating a line is not
|
|
/// what the half-second was going to.
|
|
struct Indexed<'a> {
|
|
path: &'a Path,
|
|
text: String,
|
|
/// Byte range of each non-blank line, in the order they were written.
|
|
lines: Vec<Range<usize>>,
|
|
}
|
|
|
|
impl<'a> Indexed<'a> {
|
|
/// `None` for a file that isn't there, which is a session that has not
|
|
/// produced an event yet rather than a failure.
|
|
fn read(path: &'a Path) -> Result<Option<Self>> {
|
|
let text = match std::fs::read_to_string(path) {
|
|
Ok(text) => text,
|
|
Err(err) if err.kind() == std::io::ErrorKind::NotFound => return Ok(None),
|
|
Err(err) => {
|
|
return Err(err).with_context(|| format!("read transcript {}", path.display()));
|
|
}
|
|
};
|
|
let mut lines = Vec::new();
|
|
let mut start = 0;
|
|
while start < text.len() {
|
|
let end = text[start..]
|
|
.find('\n')
|
|
.map(|at| start + at)
|
|
.unwrap_or(text.len());
|
|
if !text[start..end].trim().is_empty() {
|
|
lines.push(start..end);
|
|
}
|
|
start = end + 1;
|
|
}
|
|
Ok(Some(Self { path, text, lines }))
|
|
}
|
|
|
|
/// The index of the first line numbered `seq` or higher, or the end when
|
|
/// every line is older than that.
|
|
///
|
|
/// A bisection, which is only correct because the file is in sequence order.
|
|
/// A line that cannot be read is reported here rather than silently treated
|
|
/// as out of range, because the answer would be a window off by however much
|
|
/// of the file the bad line hid.
|
|
fn first_at_or_after(&self, seq: u64) -> Result<usize> {
|
|
let (mut low, mut high) = (0, self.lines.len());
|
|
while low < high {
|
|
let middle = (low + high) / 2;
|
|
if self.seq_at(middle)? < seq {
|
|
low = middle + 1;
|
|
} else {
|
|
high = middle;
|
|
}
|
|
}
|
|
Ok(low)
|
|
}
|
|
|
|
/// One line's sequence number, without building the event on it.
|
|
fn seq_at(&self, index: usize) -> Result<u64> {
|
|
#[derive(Deserialize)]
|
|
struct JustSeq {
|
|
seq: u64,
|
|
}
|
|
let line = &self.text[self.lines[index].clone()];
|
|
let entry: JustSeq = serde_json::from_str(line)
|
|
.with_context(|| format!("bad transcript line in {}", self.path.display()))?;
|
|
Ok(entry.seq)
|
|
}
|
|
|
|
fn parse(&self, range: Range<usize>) -> Result<Vec<SeqEvent>> {
|
|
self.lines[range.clone()]
|
|
.iter()
|
|
.enumerate()
|
|
.map(|(offset, at)| self.parse_at(range.start + offset, at.clone()))
|
|
.collect()
|
|
}
|
|
|
|
fn parse_one(&self, index: usize) -> Result<SeqEvent> {
|
|
self.parse_at(index, self.lines[index].clone())
|
|
}
|
|
|
|
/// One line, degrading to [`Event::Unreadable`] rather than failing when
|
|
/// this build cannot make sense of the event on it.
|
|
///
|
|
/// **A transcript is append-only and permanent, so the set of kinds that
|
|
/// can appear in one only ever grows.** What this build writes is not what
|
|
/// it may have to read: a line may come from a newer server, or from an
|
|
/// older one that wrote a kind since dropped. Refusing the whole file for
|
|
/// one such line is what happened on 2026-09-06 -- a kind was removed after
|
|
/// transcripts had recorded it, every read of those files failed, and the
|
|
/// sessions in them could not be opened, listed, paged or sent to. One
|
|
/// unfamiliar word took down every conversation it appeared in.
|
|
///
|
|
/// So the line survives as a line. It keeps its seq, which is the part
|
|
/// everything downstream is addressed by, and says what it was rather than
|
|
/// pretending to be something -- there is a row for that on the phone
|
|
/// already.
|
|
///
|
|
/// The seq itself is still required, and this still fails without one: a
|
|
/// line that cannot say where it sits in the sequence is not a line this
|
|
/// file can hold, and quietly dropping it would hand out a seq the file
|
|
/// already contains.
|
|
fn parse_at(&self, index: usize, at: Range<usize>) -> Result<SeqEvent> {
|
|
let line = &self.text[at];
|
|
match serde_json::from_str(line) {
|
|
Ok(entry) => Ok(entry),
|
|
Err(err) => {
|
|
#[derive(Deserialize)]
|
|
struct JustPlace {
|
|
seq: u64,
|
|
ts: f64,
|
|
#[serde(rename = "type")]
|
|
kind: Option<String>,
|
|
}
|
|
let place: JustPlace = serde_json::from_str(line)
|
|
.with_context(|| format!("bad transcript line in {}", self.path.display()))?;
|
|
// Debug rather than a warning: a transcript written against a
|
|
// newer build has one of these per line it wrote, and the row
|
|
// on the phone is where this is actually reported.
|
|
tracing::debug!(
|
|
"transcript line {index} of {} (seq {}) is not one this build can read: {err}",
|
|
self.path.display(),
|
|
place.seq,
|
|
);
|
|
Ok(SeqEvent {
|
|
seq: place.seq,
|
|
ts: place.ts,
|
|
event: Event::Unreadable {
|
|
kind: place.kind.unwrap_or_else(|| "no kind".to_string()),
|
|
},
|
|
})
|
|
}
|
|
}
|
|
}
|
|
|
|
/// The newest `limit` *rows* ending at line `end`, with each run of
|
|
/// consecutive deltas of one streamed kind concatenated into one.
|
|
///
|
|
/// Two kinds stream a token at a time -- [`Event::AssistantText`] and
|
|
/// [`Event::Thinking`] -- and a run is of one of them, never of both: they
|
|
/// are two rows on screen, and welding them would put a model's working
|
|
/// inside what it said.
|
|
///
|
|
/// A reply is stored a token at a time, so a window counted in events is a
|
|
/// fraction of a row for a reply and a whole row for a tool call, and the
|
|
/// phone can neither predict how much a page will show nor fill a screen
|
|
/// without folding a page of near-duplicate events. Counted in rows, a page
|
|
/// is a page.
|
|
///
|
|
/// A run takes the seq and time of its *oldest* delta, matching the phone's
|
|
/// own rule -- so anchors and the `before` cursor land where they always
|
|
/// did. A run cut by the `limit` is emitted as the partial it is, and the
|
|
/// phone's `healSplitMessage` welds it to the next page. `start` is the same
|
|
/// kind of cut from the other end.
|
|
fn parse_coalesced(&self, start: usize, end: usize, limit: usize) -> Result<Vec<SeqEvent>> {
|
|
// Newest first while walking back, reversed to transcript order at the end.
|
|
let mut out: Vec<SeqEvent> = Vec::new();
|
|
// The run currently being gathered: which kind it is, its oldest
|
|
// seq/ts so far, and its deltas newest-first.
|
|
let mut run: Option<Run> = None;
|
|
let flush = |run: &mut Option<Run>, out: &mut Vec<SeqEvent>| {
|
|
if let Some(Run {
|
|
kind,
|
|
seq,
|
|
ts,
|
|
mut deltas,
|
|
}) = run.take()
|
|
{
|
|
deltas.reverse();
|
|
out.push(SeqEvent {
|
|
seq,
|
|
ts,
|
|
event: kind.of_delta(deltas.concat()),
|
|
});
|
|
}
|
|
};
|
|
let mut index = end;
|
|
while index > start {
|
|
// A row is counted when it lands in `out`; an open run is the row
|
|
// being gathered, so stopping while one is open would drop the
|
|
// deltas already read. Break only between rows.
|
|
if out.len() >= limit && run.is_none() {
|
|
break;
|
|
}
|
|
index -= 1;
|
|
let entry = self.parse_one(index)?;
|
|
match Streamed::of(entry.event) {
|
|
Ok((kind, delta)) => {
|
|
// A run of a different kind ends here, whatever it was
|
|
// gathering: the two are separate rows.
|
|
if run.as_ref().is_some_and(|open| open.kind != kind) {
|
|
flush(&mut run, &mut out);
|
|
}
|
|
match run {
|
|
Some(ref mut open) => {
|
|
open.seq = entry.seq;
|
|
open.ts = entry.ts;
|
|
open.deltas.push(delta);
|
|
}
|
|
None => {
|
|
run = Some(Run {
|
|
kind,
|
|
seq: entry.seq,
|
|
ts: entry.ts,
|
|
deltas: vec![delta],
|
|
})
|
|
}
|
|
}
|
|
}
|
|
Err(event) => {
|
|
// The run above this event (newer) is complete: it is a row, and so is this
|
|
// event.
|
|
flush(&mut run, &mut out);
|
|
out.push(SeqEvent { event, ..entry });
|
|
}
|
|
}
|
|
}
|
|
flush(&mut run, &mut out);
|
|
out.reverse();
|
|
Ok(out)
|
|
}
|
|
}
|
|
|
|
/// One run of same-kind deltas being gathered by [`Indexed::parse_coalesced`].
|
|
struct Run {
|
|
kind: Streamed,
|
|
seq: u64,
|
|
ts: f64,
|
|
deltas: Vec<String>,
|
|
}
|
|
|
|
/// The event kinds that arrive a fragment at a time and are read as one row.
|
|
#[derive(Clone, Copy, PartialEq, Eq)]
|
|
enum Streamed {
|
|
Text,
|
|
Thinking,
|
|
}
|
|
|
|
impl Streamed {
|
|
/// The kind and fragment of a streamed event, or the event back unchanged
|
|
/// when it is not one -- so the caller cannot forget to put it back.
|
|
fn of(event: Event) -> std::result::Result<(Self, String), Event> {
|
|
match event {
|
|
Event::AssistantText { delta } => Ok((Self::Text, delta)),
|
|
Event::Thinking { delta } => Ok((Self::Thinking, delta)),
|
|
other => Err(other),
|
|
}
|
|
}
|
|
|
|
/// The run put back together as the event it was a run of.
|
|
fn of_delta(self, delta: String) -> Event {
|
|
match self {
|
|
Self::Text => Event::AssistantText { delta },
|
|
Self::Thinking => Event::Thinking { delta },
|
|
}
|
|
}
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod tests {
|
|
use super::*;
|
|
use crate::session::driver::{QuestionOption, SessionStatus};
|
|
|
|
fn text(delta: &str) -> Event {
|
|
Event::AssistantText {
|
|
delta: delta.to_string(),
|
|
}
|
|
}
|
|
|
|
/// The failure that took every live session down on 2026-09-06: an event
|
|
/// kind was removed from the enum after transcripts had already recorded
|
|
/// it, so every read of those files failed and the sessions in them could
|
|
/// not be opened, listed, paged or sent to.
|
|
///
|
|
/// A transcript is append-only and permanent, so **the set of kinds that
|
|
/// can appear in one only ever grows** -- what this build writes is not
|
|
/// what it may have to read. A line it cannot make sense of has to be a
|
|
/// line, not the end of the file.
|
|
#[test]
|
|
fn a_line_of_a_kind_this_build_does_not_know_does_not_break_the_file() {
|
|
let dir = tempfile::tempdir().expect("tempdir");
|
|
let path = dir.path().join("transcript.jsonl");
|
|
std::fs::write(
|
|
&path,
|
|
concat!(
|
|
r#"{"seq":1,"ts":1.0,"type":"status","state":"running"}"#,
|
|
"\n",
|
|
r#"{"seq":2,"ts":2.0,"type":"assistantText","delta":"hi"}"#,
|
|
"\n",
|
|
r#"{"seq":3,"ts":3.0,"type":"taskNote","about":"t","title":"h","status":"completed"}"#,
|
|
"\n",
|
|
r#"{"seq":4,"ts":4.0,"type":"somethingFromTheFuture","whatever":[1,2]}"#,
|
|
"\n",
|
|
r#"{"seq":5,"ts":5.0,"type":"status","state":"idle"}"#,
|
|
"\n",
|
|
),
|
|
)
|
|
.expect("write");
|
|
|
|
let entries = read_after(&path, 0).expect("a strange line is not a broken file");
|
|
assert_eq!(entries.len(), 5, "every line is still a line: {entries:?}");
|
|
assert_eq!(entries[4].seq, 5);
|
|
|
|
let transcript = Transcript::open(&path).expect("open");
|
|
assert_eq!(transcript.last_status(), Some(SessionStatus::Idle));
|
|
// The next seq is counted from the newest *line*, whatever kind it is.
|
|
// Skipping the ones this build cannot read would hand out a seq the
|
|
// file already contains.
|
|
assert_eq!(transcript.next_seq, 6);
|
|
|
|
let window = read_window(&path, None, None, 80, false).expect("window");
|
|
assert_eq!(window.len(), 5, "{window:?}");
|
|
}
|
|
|
|
#[test]
|
|
fn assigns_increasing_seqs_and_replays_after_a_cursor() {
|
|
let dir = tempfile::tempdir().expect("tempdir");
|
|
let path = dir.path().join("transcript.jsonl");
|
|
|
|
let mut transcript = Transcript::open(&path).expect("open");
|
|
assert_eq!(transcript.append(text("a"), 1.0).expect("append").seq, 1);
|
|
assert_eq!(transcript.append(text("b"), 2.0).expect("append").seq, 2);
|
|
assert_eq!(transcript.append(text("c"), 3.0).expect("append").seq, 3);
|
|
|
|
let replay = read_after(&path, 1).expect("read");
|
|
assert_eq!(replay.len(), 2);
|
|
assert_eq!(replay[0].seq, 2);
|
|
assert_eq!(replay[0].event, text("b"));
|
|
assert_eq!(replay[1].seq, 3);
|
|
|
|
// A cursor at or past the end replays nothing.
|
|
assert!(read_after(&path, 3).expect("read").is_empty());
|
|
}
|
|
|
|
#[test]
|
|
fn reopening_continues_the_numbering() {
|
|
let dir = tempfile::tempdir().expect("tempdir");
|
|
let path = dir.path().join("transcript.jsonl");
|
|
|
|
let mut transcript = Transcript::open(&path).expect("open");
|
|
transcript.append(text("a"), 1.0).expect("append");
|
|
transcript.append(text("b"), 2.0).expect("append");
|
|
drop(transcript);
|
|
|
|
let mut reopened = Transcript::open(&path).expect("reopen");
|
|
assert_eq!(reopened.append(text("c"), 3.0).expect("append").seq, 3);
|
|
}
|
|
|
|
#[test]
|
|
fn a_short_backlog_continues_and_a_long_one_restarts() {
|
|
let dir = tempfile::tempdir().expect("tempdir");
|
|
let path = dir.path().join("transcript.jsonl");
|
|
let mut transcript = Transcript::open(&path).expect("open");
|
|
for n in 0..10 {
|
|
transcript
|
|
.append(text(&n.to_string()), 0.0)
|
|
.expect("append");
|
|
}
|
|
|
|
// Within the limit the subscriber keeps what it has.
|
|
let CatchUp::Continue(events) = catch_up(&path, 7, 5).expect("catch up") else {
|
|
panic!("a backlog of 3 should continue");
|
|
};
|
|
assert_eq!(events.len(), 3);
|
|
assert_eq!(events[0].seq, 8);
|
|
|
|
// Past it, the newest window replaces what it has -- and it is the
|
|
// newest, not the oldest, that survives the trim.
|
|
let CatchUp::Restart(events) = catch_up(&path, 0, 5).expect("catch up") else {
|
|
panic!("a backlog of 10 should restart");
|
|
};
|
|
assert_eq!(events.len(), 5);
|
|
assert_eq!(events[0].seq, 6);
|
|
assert_eq!(events[4].seq, 10);
|
|
|
|
// Exactly at the limit is still a continuation: the boundary belongs to
|
|
// the cheaper answer, so a client is not reset for being one event
|
|
// behind the threshold.
|
|
assert!(matches!(
|
|
catch_up(&path, 5, 5).expect("catch up"),
|
|
CatchUp::Continue(_)
|
|
));
|
|
}
|
|
|
|
#[test]
|
|
fn reopening_reports_the_state_it_was_last_left_in() {
|
|
let dir = tempfile::tempdir().expect("tempdir");
|
|
let path = dir.path().join("transcript.jsonl");
|
|
|
|
// Nothing recorded yet: no prior state to report, which is not the same
|
|
// as reporting idle.
|
|
assert_eq!(Transcript::open(&path).expect("open").last_status(), None);
|
|
|
|
let mut transcript = Transcript::open(&path).expect("open");
|
|
transcript
|
|
.append(
|
|
Event::Status {
|
|
state: SessionStatus::Running,
|
|
},
|
|
1.0,
|
|
)
|
|
.expect("append");
|
|
transcript
|
|
.append(
|
|
Event::Status {
|
|
state: SessionStatus::Exited,
|
|
},
|
|
2.0,
|
|
)
|
|
.expect("append");
|
|
// Events after the last status must not hide it.
|
|
transcript.append(text("trailing"), 3.0).expect("append");
|
|
drop(transcript);
|
|
|
|
let reopened = Transcript::open(&path).expect("reopen");
|
|
assert_eq!(reopened.last_status(), Some(SessionStatus::Exited));
|
|
// And the same pass still continues the numbering.
|
|
assert_eq!(reopened.next_seq, 4);
|
|
}
|
|
|
|
#[test]
|
|
fn a_window_is_the_events_before_a_cursor_and_nothing_else() {
|
|
let dir = tempfile::tempdir().expect("tempdir");
|
|
let path = dir.path().join("transcript.jsonl");
|
|
let mut transcript = Transcript::open(&path).expect("open");
|
|
for n in 1..=10 {
|
|
transcript
|
|
.append(text(&n.to_string()), 0.0)
|
|
.expect("append");
|
|
}
|
|
|
|
// No cursor is the newest page, which is what opening a session asks for.
|
|
let newest = read_window(&path, None, None, 3, false).expect("window");
|
|
assert_eq!(
|
|
newest.iter().map(|entry| entry.seq).collect::<Vec<_>>(),
|
|
[8, 9, 10]
|
|
);
|
|
|
|
// Then backwards from the oldest of those, exclusive: the page a phone
|
|
// scrolling up asks for must not repeat the row it is scrolling from.
|
|
let older = read_window(&path, Some(8), None, 3, false).expect("window");
|
|
assert_eq!(
|
|
older.iter().map(|entry| entry.seq).collect::<Vec<_>>(),
|
|
[5, 6, 7]
|
|
);
|
|
|
|
// Asking for more than there is gives what there is, rather than failing.
|
|
assert_eq!(
|
|
read_window(&path, None, None, 100, false)
|
|
.expect("window")
|
|
.len(),
|
|
10
|
|
);
|
|
|
|
// Nothing before the first event, which is how the phone learns to stop
|
|
// paging. An empty answer here is the end of the history, not a fault.
|
|
assert!(
|
|
read_window(&path, Some(1), None, 3, false)
|
|
.expect("window")
|
|
.is_empty()
|
|
);
|
|
assert!(
|
|
read_window(&dir.path().join("nope.jsonl"), None, None, 3, false)
|
|
.expect("window")
|
|
.is_empty()
|
|
);
|
|
}
|
|
|
|
#[test]
|
|
fn a_floor_stops_a_page_at_what_the_caller_already_holds() {
|
|
let dir = tempfile::tempdir().expect("tempdir");
|
|
let path = dir.path().join("transcript.jsonl");
|
|
let mut transcript = Transcript::open(&path).expect("open");
|
|
for n in 1..=10 {
|
|
transcript
|
|
.append(text(&n.to_string()), 0.0)
|
|
.expect("append");
|
|
}
|
|
|
|
// The floor is exclusive, like the SSE route's `after`, and it -- not the
|
|
// limit -- is what the page stops at. This is the gap between a phone's
|
|
// cached run and the window on its screen, fetched exactly.
|
|
let page = read_window(&path, Some(9), Some(5), 100, false).expect("window");
|
|
assert_eq!(
|
|
page.iter().map(|entry| entry.seq).collect::<Vec<_>>(),
|
|
[6, 7, 8]
|
|
);
|
|
|
|
// A limit smaller than the gap still bites; the floor is a bound, not a
|
|
// replacement for one.
|
|
let page = read_window(&path, Some(9), Some(2), 3, false).expect("window");
|
|
assert_eq!(
|
|
page.iter().map(|entry| entry.seq).collect::<Vec<_>>(),
|
|
[6, 7, 8]
|
|
);
|
|
|
|
// A floor at or above the window is an empty page, not a walk past it.
|
|
assert!(
|
|
read_window(&path, Some(4), Some(9), 10, false)
|
|
.expect("window")
|
|
.is_empty()
|
|
);
|
|
}
|
|
|
|
#[test]
|
|
/// Thinking streams a fragment at a time exactly as a reply does, so a page
|
|
/// counted in rows has to coalesce it too -- otherwise one block of working
|
|
/// is a whole page of near-duplicate events. And the two runs stay two: a
|
|
/// weld across the boundary would put the model's working inside what it
|
|
/// said, in the prompt as well as on screen.
|
|
fn thinking_deltas_coalesce_into_a_row_of_their_own() {
|
|
let dir = tempfile::tempdir().expect("tempdir");
|
|
let path = dir.path().join("transcript.jsonl");
|
|
let mut transcript = Transcript::open(&path).expect("append");
|
|
for delta in ["think", "ing"] {
|
|
transcript
|
|
.append(
|
|
Event::Thinking {
|
|
delta: delta.into(),
|
|
},
|
|
0.0,
|
|
)
|
|
.expect("append");
|
|
}
|
|
transcript
|
|
.append(Event::ThinkingDone { ms: 1200 }, 0.0)
|
|
.expect("append");
|
|
for delta in ["said", " it"] {
|
|
transcript.append(text(delta), 0.0).expect("append");
|
|
}
|
|
|
|
// `before` past the end, since coalescing is only ever done on settled
|
|
// history -- see [`read_window`].
|
|
let rows = read_window(&path, Some(6), None, 10, true).expect("window");
|
|
assert_eq!(rows.len(), 3);
|
|
assert!(matches!(
|
|
&rows[0],
|
|
SeqEvent { seq: 1, event: Event::Thinking { delta }, .. } if delta == "thinking"
|
|
));
|
|
assert!(matches!(
|
|
&rows[1],
|
|
SeqEvent {
|
|
seq: 3,
|
|
event: Event::ThinkingDone { ms: 1200 },
|
|
..
|
|
}
|
|
));
|
|
assert!(matches!(
|
|
&rows[2],
|
|
SeqEvent { seq: 4, event: Event::AssistantText { delta }, .. } if delta == "said it"
|
|
));
|
|
}
|
|
|
|
#[test]
|
|
fn a_floor_inside_a_delta_run_leaves_the_partial_run_it_cuts() {
|
|
let dir = tempfile::tempdir().expect("tempdir");
|
|
let path = dir.path().join("transcript.jsonl");
|
|
let mut transcript = Transcript::open(&path).expect("open");
|
|
for d in ["a", "b", "c", "d"] {
|
|
transcript.append(text(d), 0.0).expect("append"); // seq 1..4
|
|
}
|
|
transcript
|
|
.append(
|
|
Event::ToolStart {
|
|
id: "t".into(),
|
|
tool: "Bash".into(),
|
|
input: serde_json::Value::Null,
|
|
},
|
|
0.0,
|
|
)
|
|
.expect("append"); // seq 5
|
|
|
|
// Cut inside the run: what comes back is the deltas above the floor, seq'd
|
|
// at the first of them -- the partial the phone's `healSplitMessage` welds
|
|
// onto the rest, the same as a run cut by the limit.
|
|
let rows = read_window(&path, Some(6), Some(2), 10, true).expect("window");
|
|
assert_eq!(rows.len(), 2);
|
|
assert!(matches!(
|
|
&rows[0],
|
|
SeqEvent { seq: 3, event: Event::AssistantText { delta }, .. } if delta == "cd"
|
|
));
|
|
assert!(matches!(
|
|
&rows[1],
|
|
SeqEvent {
|
|
seq: 5,
|
|
event: Event::ToolStart { .. },
|
|
..
|
|
}
|
|
));
|
|
|
|
// And with no floor the whole run is one row, as before.
|
|
let rows = read_window(&path, Some(6), None, 10, true).expect("window");
|
|
assert_eq!(rows.len(), 2);
|
|
assert!(matches!(
|
|
&rows[0],
|
|
SeqEvent { seq: 1, event: Event::AssistantText { delta }, .. } if delta == "abcd"
|
|
));
|
|
}
|
|
|
|
#[test]
|
|
fn coalescing_counts_rows_and_joins_delta_runs() {
|
|
let dir = tempfile::tempdir().expect("tempdir");
|
|
let path = dir.path().join("transcript.jsonl");
|
|
let mut transcript = Transcript::open(&path).expect("open");
|
|
// Two replies of three deltas each, split by a tool call: the shape a turn writes, and
|
|
// the one an event-counted page cannot see a row of.
|
|
for d in ["a", "b", "c"] {
|
|
transcript.append(text(d), 0.0).expect("append"); // seq 1..3
|
|
}
|
|
transcript
|
|
.append(
|
|
Event::ToolStart {
|
|
id: "t".into(),
|
|
tool: "Bash".into(),
|
|
input: serde_json::Value::Null,
|
|
},
|
|
0.0,
|
|
)
|
|
.expect("append"); // seq 4
|
|
for d in ["d", "e", "f"] {
|
|
transcript.append(text(d), 0.0).expect("append"); // seq 5..7
|
|
}
|
|
|
|
// Three rows asked for, three rows returned -- each delta run one event -- where a raw
|
|
// window of three would have shown one and a half tokens of the newer reply.
|
|
let rows = read_window(&path, Some(8), None, 3, true).expect("window");
|
|
assert_eq!(rows.len(), 3);
|
|
// A run keeps its oldest delta's seq, so the phone anchors and pages from where it always
|
|
// did.
|
|
assert!(matches!(
|
|
&rows[0],
|
|
SeqEvent { seq: 1, event: Event::AssistantText { delta }, .. } if delta == "abc"
|
|
));
|
|
assert!(matches!(
|
|
&rows[1],
|
|
SeqEvent {
|
|
seq: 4,
|
|
event: Event::ToolStart { .. },
|
|
..
|
|
}
|
|
));
|
|
assert!(matches!(
|
|
&rows[2],
|
|
SeqEvent { seq: 5, event: Event::AssistantText { delta }, .. } if delta == "def"
|
|
));
|
|
|
|
// The next page pages from the oldest row's seq and returns the rest, no repeat, no gap.
|
|
let older = read_window(&path, Some(1), None, 3, true).expect("window");
|
|
assert!(older.is_empty());
|
|
|
|
// The newest window never coalesces even when asked: the live cursor depends on real seqs.
|
|
let newest = read_window(&path, None, None, 2, true).expect("window");
|
|
assert_eq!(newest.iter().map(|e| e.seq).collect::<Vec<_>>(), [6, 7]);
|
|
}
|
|
|
|
#[test]
|
|
fn a_line_read_back_is_the_line_that_was_written() {
|
|
let dir = tempfile::tempdir().expect("tempdir");
|
|
let path = dir.path().join("transcript.jsonl");
|
|
// A timestamp with enough digits to be lost: the clock produces these all day, and this
|
|
// one is real (2026-09-04). serde_json's default float parser is not correctly rounded,
|
|
// so it read this back as ...0755 and every reader got a line one bit different from the
|
|
// one in the file -- while the SSE stream, which serializes the same struct, had already
|
|
// sent the original. Two answers to "what is line 1", indistinguishable by eye.
|
|
//
|
|
// Nothing on screen showed it: a `ts` is drawn as a relative time. What found it was the
|
|
// phone's transcript cache, which keeps the line it was sent and checks it against the
|
|
// server's own answer before resuming a stream from it -- so the mismatch turned into a
|
|
// cache thrown away and a transcript downloaded again, silently and only sometimes. The
|
|
// `float_roundtrip` feature in Cargo.toml is the fix; this is what keeps it.
|
|
let mut transcript = Transcript::open(&path).expect("open");
|
|
transcript
|
|
.append(text("hello"), 1788546972.6030757)
|
|
.expect("append");
|
|
drop(transcript);
|
|
|
|
let written = std::fs::read_to_string(&path).expect("read");
|
|
let entry = read_window(&path, None, None, 10, false).expect("window");
|
|
assert_eq!(
|
|
serde_json::to_string(&entry[0]).expect("serialize"),
|
|
written.trim()
|
|
);
|
|
}
|
|
|
|
#[test]
|
|
fn a_missing_file_reads_as_empty() {
|
|
let dir = tempfile::tempdir().expect("tempdir");
|
|
assert!(
|
|
read_after(&dir.path().join("nope.jsonl"), 0)
|
|
.expect("read")
|
|
.is_empty()
|
|
);
|
|
}
|
|
|
|
#[test]
|
|
fn round_trips_every_event_shape() {
|
|
let dir = tempfile::tempdir().expect("tempdir");
|
|
let path = dir.path().join("transcript.jsonl");
|
|
let events = vec![
|
|
Event::UserMessage {
|
|
id: None,
|
|
text: "hi".into(),
|
|
attachments: Vec::new(),
|
|
},
|
|
text("hello"),
|
|
Event::AssistantTextFinal {
|
|
text: "hello, revised".into(),
|
|
},
|
|
Event::ToolStart {
|
|
id: "t1".into(),
|
|
tool: "bash".into(),
|
|
input: serde_json::json!({"command": "ls"}),
|
|
},
|
|
Event::ToolUpdate {
|
|
id: "t1".into(),
|
|
output: "partial".into(),
|
|
},
|
|
Event::ToolEnd {
|
|
id: "t1".into(),
|
|
output: "done".into(),
|
|
},
|
|
Event::Image {
|
|
image: "img1".into(),
|
|
about: None,
|
|
},
|
|
Event::Question {
|
|
id: "q1".into(),
|
|
prompt: "Allow?".into(),
|
|
header: None,
|
|
options: vec![QuestionOption::plain("Yes"), QuestionOption::plain("No")],
|
|
multi_select: false,
|
|
about: None,
|
|
},
|
|
Event::Answered {
|
|
id: "q1".into(),
|
|
answers: vec!["Yes".into()],
|
|
},
|
|
Event::BackgroundTasks { count: 5 },
|
|
Event::Status {
|
|
state: SessionStatus::Idle,
|
|
},
|
|
Event::UsageDelta {
|
|
tokens: 42,
|
|
context: Some(42),
|
|
tokens_per_second: None,
|
|
prefill_ms: None,
|
|
},
|
|
Event::AuthenticationRequired {
|
|
message: "sign in again".into(),
|
|
},
|
|
Event::Error {
|
|
message: "boom".into(),
|
|
},
|
|
];
|
|
|
|
let mut transcript = Transcript::open(&path).expect("open");
|
|
for event in &events {
|
|
transcript.append(event.clone(), 0.0).expect("append");
|
|
}
|
|
|
|
let replayed: Vec<Event> = read_after(&path, 0)
|
|
.expect("read")
|
|
.into_iter()
|
|
.map(|entry| entry.event)
|
|
.collect();
|
|
assert_eq!(replayed, events);
|
|
}
|
|
}
|