rag 0.1.5

A Rust library and CLI for Retrieval-Augmented Generation
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

//! Optional re-ranking hook for similarity lists (default: identity).

use std::future::Future;

use crate::errors::Result;
use crate::vector_store::Similarity;

pub trait SimilarityReranker: Send + Sync {
    fn rerank(
        &self,
        query: &str,
        items: Vec<Similarity>,
    ) -> impl Future<Output = Result<Vec<Similarity>>> + Send;
}

/// No-op reranker (preserves order and scores).
pub struct PassthroughReranker;

impl SimilarityReranker for PassthroughReranker {
    async fn rerank(&self, _query: &str, items: Vec<Similarity>) -> Result<Vec<Similarity>> {
        Ok(items)
    }
}

/// Apply a reranker to an owned similarity list.
pub async fn rerank_similarities<R: SimilarityReranker>(
    reranker: &R,
    query: &str,
    items: Vec<Similarity>,
) -> Result<Vec<Similarity>> {
    reranker.rerank(query, items).await
}

#[cfg(test)]
mod tests {
    use super::*;
    use crate::vector_store::Document;

    #[tokio::test]
    async fn passthrough_preserves() {
        let r = PassthroughReranker;
        let v = vec![Similarity {
            document: Document::new("a".to_string()),
            score: 0.5,
        }];
        let out = r.rerank("q", v).await.unwrap();
        assert_eq!(out.len(), 1);
        assert_eq!(out[0].document.content, "a");
    }
}