lumen-rag 0.2.1

A modular, database-agnostic RAG framework for Rust supporting MongoDB, Qdrant, and SAP HANA Cloud.
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

use anyhow::Result;
use async_trait::async_trait;
use lumen_rag::{Passage, VectorStore};
use std::sync::{Arc, Mutex};

pub struct MockStore {
    pub storage: Arc<Mutex<Vec<Passage>>>,
}

impl MockStore {
    pub fn new() -> Self {
        Self {
            storage: Arc::new(Mutex::new(Vec::new())),
        }
    }
}

#[async_trait]
impl VectorStore for MockStore {
    async fn add_passages(&self, passages: Vec<Passage>) -> Result<Vec<String>> {
        let mut store = self.storage.lock().unwrap();
        let mut ids = Vec::new();
        for p in passages {
            ids.push("mock_id".to_string());
            store.push(p);
        }
        Ok(ids)
    }

    async fn search(&self, _query_embedding: &[f32], _limit: usize) -> Result<Vec<Passage>> {
        let store = self.storage.lock().unwrap();
        Ok(store.clone())
    }
}