schaefera/doubleo7
1
0
Fork 0
Code Issues Pull requests 1 Projects Releases Packages Wiki Activity Actions

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
Conversation 0 Commits 5 Files changed 7 +3 -7
1 changed files with 3 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files
Showing only changes of commit b12f156a1b - Show all commits

10
deep_research/src/core.rs
View file

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