Struct RagChunkCache 

pub struct RagChunkCache { /* private fields */ }

RAG chunk cache

Implementations

impl RagChunkCache

pub fn new(max_size_mb: usize) -> Self

Create a new RAG chunk cache

pub fn get_chunk(&self, id: ChunkId) -> Option<Chunk>

Get a chunk by ID

pub fn get_chunks_by_embedding(&self, embedding: &[f32], k: usize) -> Vec<Chunk>

Get chunks by embedding similarity

pub fn get_document_chunks(&self, document_id: &str) -> Vec<Chunk>

Get all chunks for a document

pub fn insert_chunk(&self, chunk: Chunk)

Insert a chunk

pub fn insert_chunks(&self, chunks: Vec<Chunk>)

Insert multiple chunks (batch)

pub fn cache_embedding_result(&self, embedding: &[f32], chunk_ids: Vec<ChunkId>)

Cache embedding to chunk ID mapping

pub fn remove_chunk(&self, id: ChunkId)

Remove a chunk

pub fn remove_document(&self, document_id: &str)

Remove all chunks for a document

pub fn stats(&self) -> RagCacheStatsSnapshot

Get cache statistics

pub fn clear(&self)

Clear all cached chunks