quorumrag 0.1.0

Quorum-based retrieval-augmented generation: fuse multiple retrievers and keep only the evidence they agree on.
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

use serde::{Deserialize, Serialize};

/// A chunk of text from the corpus.
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct Chunk {
    pub id: String,
    pub text: String,
    pub embedding: Vec<f32>,
}

/// A query with its text and precomputed embedding.
#[derive(Debug, Clone)]
pub struct Query {
    pub text: String,
    pub embedding: Vec<f32>,
}
/// This is going to be a candidate returned by a single retriever.
#[derive(Debug, Clone)]
pub struct Candidate {
    pub chunk: Chunk,
    pub score: f32,
    pub retriever_id: String,
}

/// A cluster of semantically similar candidates from multiple retrievers.
#[derive(Debug, Clone)]
pub struct EvidenceCluster {
    pub members: Vec<Candidate>,
    pub support: usize,
    pub avg_score: f32,
}