Struct RetrievalSystem 

pub struct RetrievalSystem { /* private fields */ }

Document retrieval system for RAG

Implementations

impl RetrievalSystem

pub fn new(config: RetrievalConfig) -> Result<Self>

Create a new retrieval system

Attempts to use ONNX embeddings (semantic) with fallback to simple embeddings (hash-based)

pub fn with_embedder( config: RetrievalConfig, embedder: Arc<dyn EmbeddingGenerator>, ) -> Result<Self>

Create a new retrieval system with a custom embedding generator

pub fn index_document( &self, document_id: &str, content: &str, source: &str, metadata: Option<String>, ) -> Result<usize>

Index a document by splitting it into chunks

pub fn retrieve(&self, query: &str, top_k: usize) -> Result<Vec<SearchResult>>

Retrieve relevant documents for a query

pub fn chunk_count(&self) -> usize

Get the number of indexed chunks

pub fn clear(&self)

Clear all indexed documents

pub fn stats(&self) -> RetrievalStats

Get retrieval statistics