Struct EmbeddingStore 

pub struct EmbeddingStore { /* private fields */ }

Implementations

impl EmbeddingStore

pub fn new(pool: Arc<Pool<SqliteConnectionManager>>) -> Self

pub fn initialize_index(&self, model: &str) -> Result<()>

pub fn store_embedding(&self, embedding: &Embedding) -> Result<()>

pub fn find_similar_embeddings( &self, query_embedding: &[f32], model: &str, limit: i32, similarity_threshold: f32, ) -> Result<Vec<(i64, f32)>>

pub fn get_embedding_by_message_id( &self, message_id: i64, model: &str, ) -> Result<Option<Embedding>>

pub fn get_stats(&self) -> Result<EmbeddingStats>