Struct EmbeddingStore 

pub struct EmbeddingStore { /* private fields */ }

Persistent store for embeddings, backed by SQLite.

Implementations

impl EmbeddingStore

pub fn new( conn: Arc<Mutex<Connection>>, embedder: Box<dyn Embedder>, ) -> PunchResult<Self>

Create a new embedding store, creating the embeddings table if it does not already exist.

pub fn store( &self, text: &str, metadata: HashMap<String, String>, ) -> PunchResult<String>

Store text with its embedding vector.

pub fn search( &self, query: &str, k: usize, ) -> PunchResult<Vec<(f32, Embedding)>>

Search for the top-k most similar embeddings to query.

pub fn delete(&self, id: &str) -> PunchResult<()>

Delete an embedding by ID.

pub fn count(&self) -> PunchResult<usize>

Return the total number of stored embeddings.

pub fn rebuild_index(&self) -> PunchResult<usize>

Re-embed all stored texts. Useful when the embedder changes (e.g., after re-fitting the built-in TF-IDF vocabulary).

pub fn embedder(&self) -> &dyn Embedder

Provide a reference to the current embedder.