Struct HybridRetriever 

pub struct HybridRetriever<E: Embedder> { /* private fields */ }

Hybrid retriever combining dense and sparse retrieval

Implementations

impl<E: Embedder> HybridRetriever<E>

pub fn new(dense: VectorStore, sparse: BM25Index, embedder: E) -> Self

Create a new hybrid retriever

pub fn with_config(self, config: HybridRetrieverConfig) -> Self

Set the configuration

pub fn dense_store(&self) -> &VectorStore

Get the dense store

pub fn dense_store_mut(&mut self) -> &mut VectorStore

Get the dense store mutably

pub fn sparse_index(&self) -> &BM25Index

Get the sparse index

pub fn sparse_index_mut(&mut self) -> &mut BM25Index

Get the sparse index mutably

pub fn index(&mut self, chunk: Chunk) -> Result<()>

Index a chunk (adds to both dense and sparse indices)

pub fn index_batch(&mut self, chunks: Vec<Chunk>) -> Result<()>

Index multiple chunks

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

Retrieve relevant chunks for a query

pub fn retrieve_dense( &self, query: &str, k: usize, ) -> Result<Vec<RetrievalResult>>

Retrieve using only dense (vector) search

pub fn retrieve_sparse( &self, query: &str, k: usize, ) -> Result<Vec<RetrievalResult>>

Retrieve using only sparse (BM25) search

pub fn len(&self) -> usize

Get the number of indexed chunks

pub fn is_empty(&self) -> bool

Check if the retriever is empty