Struct VectorStore 

pub struct VectorStore { /* private fields */ }

Vector store for managing embeddings

Implementations

impl VectorStore

pub fn new(config: VectorStoreConfig) -> Self

Create a new vector store

pub fn with_dim(dim: usize) -> Self

Create with default config for given dimension

pub fn len(&self) -> usize

Get the number of vectors

pub fn is_empty(&self) -> bool

Check if empty

pub fn put(&mut self, entry: VectorEntry) -> Result<()>

Insert a vector

pub fn put_embedding( &mut self, id: impl Into<String>, embedding: impl Into<Embedding>, ) -> Result<()>

Insert a vector with just ID and embedding

pub fn put_batch(&mut self, entries: Vec<VectorEntry>) -> Result<usize>

Batch insert

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

Search for similar vectors

pub fn search_with_filter<F>( &self, query: &Embedding, k: usize, filter: F, ) -> Vec<SearchResult>

where F: Fn(&VectorEntry) -> bool,

Search with a filter on metadata

pub fn search_namespace( &self, query: &Embedding, k: usize, namespace: &str, ) -> Vec<SearchResult>

Search by namespace

pub fn get(&self, id: &str) -> Option<&VectorEntry>

Get a vector by ID

pub fn delete(&mut self, id: &str) -> bool

Delete a vector

pub fn ids(&self) -> Vec<&String>

Get all IDs

pub fn similarity(&self, id1: &str, id2: &str) -> Option<f32>

Compute similarity between two vectors by ID

pub fn upsert(&mut self, entry: VectorEntry) -> Result<bool>

Upsert: insert or update a vector

pub fn clear(&mut self)

Clear all vectors from the store

pub fn stats(&self) -> VectorStoreStats

Get store statistics

pub fn find_similar(&self, id: &str, k: usize) -> Vec<SearchResult>

Find vectors similar to a given ID

pub fn contains(&self, id: &str) -> bool

Check if a vector exists

pub fn config(&self) -> &VectorStoreConfig

Get the configuration

pub fn update_metadata( &mut self, id: &str, key: impl Into<String>, value: impl Into<Value>, ) -> bool

Update metadata for an existing vector

pub fn get_namespace(&self, namespace: &str) -> Vec<&VectorEntry>

Get all vectors in a namespace

pub fn count_namespace(&self, namespace: &str) -> usize

Count vectors in a namespace

pub fn delete_namespace(&mut self, namespace: &str) -> usize

Delete all vectors in a namespace

pub fn centroid(&self, namespace: Option<&str>) -> Option<Embedding>

Compute centroid of all vectors (or a subset by namespace)

Trait Implementations

impl Debug for VectorStore

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

Formats the value using the given formatter.