dcdpr / jp / 26823662050

67%

Build Type

push

github

Committed by web-flow

Commit Message

fix(llm, cli): Preserve partial reasoning on stream interrupt (#725)

`peek_partial_content()` in `EventBuilder` only collected message text
from unflushed buffers, silently dropping any accumulated reasoning.
When a user interrupted mid-reasoning (or a retry fired before the first
flush), the partial reasoning was lost: the model resumed with no record
of the thinking it had started, and the user's interjection landed
against reasoning the model no longer had in context.

Replace `peek_partial_content() -> Option<String>` with
`peek_partial_events() -> Vec<ChatResponse>`. The new method returns
each non-empty buffer as its correctly-typed response — reasoning as
`ChatResponse::Reasoning`, message text as `ChatResponse::Message` — in
stream-index order. Tool-call and structured buffers are still excluded
(a partial tool call would orphan a `tool_use`, and partial JSON is not
a usable resume point).

All interrupt paths in `TurnCoordinator` (`handle_quit`,
`handle_streaming_interrupt` for Stop/Continue/Reply) and the retry
handler in `stream/retry.rs` now iterate over the returned events and
commit each one individually, so reasoning stays reasoning rather than
being folded into the assistant's answer text.

Signed-off-by: Jean Mertz <git@jeanmertz.com>

Coverage Stats

28 of 32 new or added lines in 3 files covered. (87.5%)

3 existing lines in 1 file now uncovered.

33426 of 50248 relevant lines covered (66.52%)

236.83 hits per line