Struct EmbeddingStore 

pub struct EmbeddingStore { /* private fields */ }

Implementations

impl EmbeddingStore

pub fn new(url: &str, pool: SqlitePool) -> Result<Self, MemoryError>

Create a new EmbeddingStore connected to the given Qdrant URL.

The pool is used for SQLite metadata operations on the embeddings_metadata table (which must already exist via sqlx migrations).

Errors

Returns an error if the Qdrant client cannot be created.

pub fn new_sqlite(pool: SqlitePool) -> Self

Create a new EmbeddingStore backed by SQLite for vector storage.

Uses the same pool for both vector data and metadata. No external Qdrant required.

pub fn with_store(store: Box<dyn VectorStore>, pool: SqlitePool) -> Self

pub async fn health_check(&self) -> bool

pub async fn ensure_collection( &self, vector_size: u64, ) -> Result<(), MemoryError>

Ensure the collection exists in Qdrant with the given vector size.

Idempotent: no-op if the collection already exists.

Errors

Returns an error if Qdrant cannot be reached or collection creation fails.

pub async fn store( &self, message_id: MessageId, conversation_id: ConversationId, role: &str, vector: Vec<f32>, kind: MessageKind, model: &str, ) -> Result<String, MemoryError>

Store a vector in Qdrant and persist metadata to SQLite.

Returns the UUID of the newly created Qdrant point.

Errors

Returns an error if the Qdrant upsert or SQLite insert fails.

pub async fn search( &self, query_vector: &[f32], limit: usize, filter: Option<SearchFilter>, ) -> Result<Vec<SearchResult>, MemoryError>

Search for similar vectors in Qdrant, returning up to limit results.

Errors

Returns an error if the Qdrant search fails.

pub async fn ensure_named_collection( &self, name: &str, vector_size: u64, ) -> Result<(), MemoryError>

Ensure a named collection exists in Qdrant with the given vector size.

Errors

Returns an error if Qdrant cannot be reached or collection creation fails.

pub async fn store_to_collection( &self, collection: &str, payload: Value, vector: Vec<f32>, ) -> Result<String, MemoryError>

Store a vector in a named Qdrant collection with arbitrary payload.

Returns the UUID of the newly created point.

Errors

Returns an error if the Qdrant upsert fails.

pub async fn search_collection( &self, collection: &str, query_vector: &[f32], limit: usize, filter: Option<VectorFilter>, ) -> Result<Vec<ScoredVectorPoint>, MemoryError>

Search a named Qdrant collection, returning scored points with payloads.

Errors

Returns an error if the Qdrant search fails.

pub async fn get_vectors( &self, ids: &[MessageId], ) -> Result<HashMap<MessageId, Vec<f32>>, MemoryError>

Fetch raw vectors for the given message IDs from the SQLite vector store.

Returns an empty map when using Qdrant backend (vectors not locally stored).

Errors

Returns an error if the SQLite query fails.

pub async fn has_embedding( &self, message_id: MessageId, ) -> Result<bool, MemoryError>

Check whether an embedding already exists for the given message ID.

Errors

Returns an error if the SQLite query fails.

Trait Implementations

impl Debug for EmbeddingStore

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.