Struct SqliteStorage 

pub struct SqliteStorage { /* private fields */ }

SQLite-based storage implementation.

Provides persistent storage for RLM state with full ACID guarantees.

Examples

use rlm_rs::storage::{SqliteStorage, Storage};

let mut storage = SqliteStorage::open("rlm-state.db").unwrap();
storage.init().unwrap();

Implementations

impl SqliteStorage

pub fn open<P: AsRef<Path>>(path: P) -> Result<Self>

Opens or creates a SQLite database at the given path.

Arguments

• path - Path to the database file. Parent directory must exist.

Errors

Returns an error if the database cannot be opened or initialized.

pub fn in_memory() -> Result<Self>

Creates an in-memory SQLite database.

Useful for testing.

Errors

Returns an error if the database cannot be created.

pub fn path(&self) -> Option<&Path>

Returns the database path (None for in-memory).

impl SqliteStorage

pub fn get_chunks_by_ids(&self, ids: &[i64]) -> Result<HashMap<i64, Chunk>>

Retrieves multiple chunks by their IDs in a single query.

More efficient than calling Storage::get_chunk repeatedly when fetching several chunks at once (e.g., populating search result previews).

Returns a map from chunk ID to Chunk.

Errors

Returns an error if the query fails.

pub fn store_embedding( &mut self, chunk_id: i64, embedding: &[f32], model_name: Option<&str>, ) -> Result<()>

Stores an embedding for a chunk.

Arguments

• chunk_id - The chunk ID to associate the embedding with.
• embedding - The embedding vector (f32 array).
• model_name - Optional name of the model that generated the embedding.

Errors

Returns an error if the embedding cannot be stored.

pub fn get_embedding(&self, chunk_id: i64) -> Result<Option<Vec<f32>>>

Retrieves the embedding for a chunk.

Errors

Returns an error if the query fails.

pub fn get_embedding_models(&self, buffer_id: i64) -> Result<Vec<String>>

Gets the distinct model names used for embeddings in a buffer.

Returns the set of model names used to generate embeddings for chunks belonging to the specified buffer.

Errors

Returns an error if the query fails.

pub fn get_embedding_model_counts( &self, buffer_id: i64, ) -> Result<Vec<(Option<String>, i64)>>

Gets the count of embeddings by model name for a buffer.

Returns a list of (model_name, count) pairs.

Errors

Returns an error if the query fails.

pub fn store_embeddings_batch( &mut self, embeddings: &[(i64, Vec<f32>)], model_name: Option<&str>, ) -> Result<()>

Stores embeddings for multiple chunks in a batch.

Errors

Returns an error if any embedding cannot be stored.

pub fn delete_embedding(&mut self, chunk_id: i64) -> Result<()>

Deletes the embedding for a chunk.

Errors

Returns an error if deletion fails.

pub fn search_fts(&self, query: &str, limit: usize) -> Result<Vec<(i64, f64)>>

Performs FTS5 BM25 full-text search.

Returns chunk IDs and their BM25 scores (higher is better match).

Arguments

• query - The search query (supports FTS5 query syntax).
• limit - Maximum number of results to return.

Errors

Returns an error if the search fails.

pub fn get_all_embeddings(&self) -> Result<Vec<(i64, Vec<f32>)>>

Returns all chunk embeddings for vector similarity search.

Errors

Returns an error if the query fails.

pub fn embedding_count(&self) -> Result<usize>

Counts chunks with embeddings.

Errors

Returns an error if the count fails.

pub fn all_chunks_have_embeddings(&self, buffer_id: i64) -> Result<bool>

Returns true if every chunk in the buffer has an embedding (or the buffer has no chunks).

Uses a single NOT EXISTS query instead of per-chunk lookups, making it O(1) in terms of round-trips regardless of how many chunks the buffer contains.

Errors

Returns an error if the query fails.

pub fn has_embedding(&self, chunk_id: i64) -> Result<bool>

Checks if a chunk has an embedding.

Errors

Returns an error if the query fails.

pub fn get_chunks_needing_embedding( &self, buffer_id: i64, current_model: Option<&str>, ) -> Result<Vec<i64>>

Gets chunk IDs that need embedding (either no embedding or wrong model).

This is used for incremental embedding updates. Returns chunks that:

• Have no embedding at all, OR
• Have an embedding with a different model name (if current_model is provided)

Arguments

• buffer_id - The buffer to check.
• current_model - Optional model name to check against. If provided, chunks with different models are included.

Errors

Returns an error if the query fails.

pub fn get_chunks_without_embedding(&self, buffer_id: i64) -> Result<Vec<i64>>

Gets chunks without any embedding for a buffer.

Simpler version of get_chunks_needing_embedding when model doesn’t matter.

Errors

Returns an error if the query fails.

pub fn delete_embeddings_by_model( &mut self, buffer_id: i64, model_name: Option<&str>, ) -> Result<usize>

Deletes embeddings with a specific model name.

Useful for cleaning up embeddings from old models before re-embedding.

Arguments

• buffer_id - The buffer to clean.
• model_name - The model name to match (or None to match NULL).

Returns

The number of embeddings deleted.

Errors

Returns an error if deletion fails.

pub fn get_embedding_stats(&self, buffer_id: i64) -> Result<EmbeddingStats>

Gets embedding statistics for a buffer.

Returns counts of embedded vs total chunks, and model breakdown.

Errors

Returns an error if the query fails.

Trait Implementations

impl Storage for SqliteStorage

fn init(&mut self) -> Result<()>

Initializes storage (creates schema, runs migrations).

fn is_initialized(&self) -> Result<bool>

Checks if storage is initialized.

fn reset(&mut self) -> Result<()>

Resets all stored state.

fn save_context(&mut self, context: &Context) -> Result<()>

Saves the current context state.

fn load_context(&self) -> Result<Option<Context>>

Loads the context state.

fn delete_context(&mut self) -> Result<()>

Deletes the current context.

fn add_buffer(&mut self, buffer: &Buffer) -> Result<i64>

Adds a buffer to storage.

fn get_buffer(&self, id: i64) -> Result<Option<Buffer>>

Retrieves a buffer by ID.

fn get_buffer_by_name(&self, name: &str) -> Result<Option<Buffer>>

Retrieves a buffer by name.

fn list_buffers(&self) -> Result<Vec<Buffer>>

Lists all buffers.

fn update_buffer(&mut self, buffer: &Buffer) -> Result<()>

Updates an existing buffer.

fn delete_buffer(&mut self, id: i64) -> Result<()>

Deletes a buffer by ID.

fn buffer_count(&self) -> Result<usize>

Returns the count of buffers.

fn add_chunks(&mut self, buffer_id: i64, chunks: &[Chunk]) -> Result<()>

Adds chunks for a buffer.

fn get_chunks(&self, buffer_id: i64) -> Result<Vec<Chunk>>

Retrieves all chunks for a buffer.

fn get_chunk(&self, id: i64) -> Result<Option<Chunk>>

Retrieves a specific chunk by ID.

fn delete_chunks(&mut self, buffer_id: i64) -> Result<()>

Deletes all chunks for a buffer.

fn chunk_count(&self, buffer_id: i64) -> Result<usize>

Returns the count of chunks for a buffer.

fn export_buffers(&self) -> Result<String>

Exports all buffers as a concatenated string.

fn stats(&self) -> Result<StorageStats>

Gets storage statistics.