ragit 0.4.5

git-like rag pipeline
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

use super::QueryTurn;
use serde::{Deserialize, Serialize};

#[derive(Clone, Debug, Deserialize, Serialize)]
pub struct RenderedQueryTurn {
    pub role: String,
    pub content: String,
}

pub fn render_query_turns(turns: &[QueryTurn]) -> Vec<RenderedQueryTurn> {
    let mut result = Vec::with_capacity(turns.len() * 2);

    for turn in turns.iter() {
        result.push(RenderedQueryTurn {
            role: String::from("user"),
            content: turn.query.clone(),
        });
        result.push(RenderedQueryTurn {
            // how about using the model name?
            role: String::from("assistant"),
            content: format!(
                "{}{}",
                turn.response.response,
                if turn.response.retrieved_chunks.is_empty() {
                    String::new()
                } else {
                    format!(
                        "\n---- source{} ----\n{}",
                        if turn.response.retrieved_chunks.len() == 1 { "" } else { "s" },
                        turn.response.retrieved_chunks.iter().map(
                            |chunk| format!("{} ({})", chunk.render_source(), chunk.uid.abbrev(8))
                        ).collect::<Vec<_>>().join("\n"),
                    )
                },
            ),
        });
    }

    result
}