ruchat 0.2.0

ollama/chroma command-line AI chat tool
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60

use crate::agent::event::{AgentEvent, StreamItem};
use crate::agent::pipeline::PipelineStream;
use crate::io::Io;
use crate::Result;
use tokio_stream::StreamExt;

/// Consumes a `PipelineStream` and renders it to the terminal — shared by
/// `ask.rs`'s `pipe` command and `Manager::execute_command`'s `Run` arm, so
/// Manager's output goes through the same colored/status/trace handling
/// instead of plain `println!`.
pub(crate) async fn render_pipeline_stream(mut stream: PipelineStream, cio: &mut Io) -> Result<()> {
    let mut status_line_active = false;
    while let Some(res) = stream.next().await {
        match res {
            Ok(StreamItem::Chunk(responses)) => {
                if status_line_active {
                    cio.clear_status_line().await?;
                    status_line_active = false;
                }
                for resp in responses {
                    cio.write_line(&resp.response).await?;
                }
            }
            Ok(StreamItem::ChatChunk(chunk)) => {
                if status_line_active {
                    cio.clear_status_line().await?;
                    status_line_active = false;
                }
                cio.write_line(&chunk.message.content).await?;
            }
            Ok(StreamItem::Event(AgentEvent::ColorChange(ansi_code))) => {
                cio.write_line(ansi_code).await?;
            }
            Ok(StreamItem::Event(AgentEvent::StatusUpdate(msg))) => {
                cio.clear_status_line().await?;
                cio.write_line(&format!("\x1b[2m   ... {msg} \x1b[0m\r\x1b[2K"))
                    .await?;
                status_line_active = !msg.is_empty();
            }
            Ok(StreamItem::Event(AgentEvent::Trace(msg))) => {
                if status_line_active {
                    cio.clear_status_line().await?;
                    status_line_active = false;
                }
                cio.write_line(&format!("\n\x1b[90m[TRACE] {msg}\x1b[0m\n"))
                    .await?;
            }
            Ok(StreamItem::Event(AgentEvent::Progress(pct))) => {
                cio.clear_status_line().await?;
                cio.write_line(&format!("\x1b[2m   ... {pct:.0}% \x1b[0m\r\x1b[2K"))
                    .await?;
                status_line_active = true;
            }
            Ok(StreamItem::Event(AgentEvent::Done)) => break,
            Err(e) => return Err(e),
        }
    }
    cio.write_line("\x1b[0m").await?;
    Ok(())
}