Stream the report to the terminal as the writer generates it #5

Merged
schaefera merged 5 commits from worktree-deep-research-streaming into master 2026-08-17 10:48:53 +00:00
Showing only changes of commit b12f156a1b - Show all commits

View file

@ -5,7 +5,6 @@ use clap::Parser;
use futures::StreamExt;
use rig::agent::MultiTurnStreamItem;
use rig::client::{AgentClientExt, Nothing};
use rig::message::Text;
use rig::providers::ollama;
use rig::streaming::{StreamedAssistantContent, StreamingPrompt};
use std::io::Write;
@ -186,15 +185,12 @@ async fn write_report(
while let Some(chunk) = response_stream.next().await {
match chunk? {
MultiTurnStreamItem::StreamAssistantItem(StreamedAssistantContent::Text(Text {
text,
..
})) => {
report.push_str(&text);
MultiTurnStreamItem::StreamAssistantItem(StreamedAssistantContent::Text(text)) => {
report.push_str(&text.text);
// Terminal stdout is line-buffered, so a flush is needed here —
// otherwise a chunk without a trailing newline sits in the
// buffer instead of appearing as it streams in.
write!(handle, "{text}")?;
write!(handle, "{}", text.text)?;
handle.flush()?;
}
_ => continue,