tt-retrieval 0.1.0

RAG / context compression: chunking, embedding, HNSW retrieval, <retrievable> tag substitution.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23

use async_trait::async_trait;
use uuid::Uuid;

use crate::error::RetrievalError;
use crate::types::{Chunk, RetrievalResult};

#[async_trait]
pub trait RetrievalStore: Send + Sync {
    async fn insert(&self, chunk: Chunk) -> Result<(), RetrievalError>;
    async fn search(
        &self,
        org_id: Uuid,
        corpus: &str,
        query_embedding: &[f32],
        k: usize,
        embedding_model: &str,
    ) -> Result<Vec<RetrievalResult>, RetrievalError>;
    async fn delete_corpus(&self, org_id: Uuid, corpus: &str) -> Result<u64, RetrievalError>;
}

pub mod memory;
#[cfg(feature = "postgres")]
pub mod postgres;