Struct InMemoryVectorStoreBuilder 

pub struct InMemoryVectorStoreBuilder<D>
where
    D: Serialize,
{ /* private fields */ }

Builder for creating an InMemoryVectorStore with custom configuration.

Implementations

impl<D> InMemoryVectorStoreBuilder<D>

where D: Serialize + Eq,

pub fn new() -> InMemoryVectorStoreBuilder<D>

Create a new builder with default settings. Default index strategy is BruteForce.

pub fn index_strategy( self, index_strategy: IndexStrategy, ) -> InMemoryVectorStoreBuilder<D>

Set the index strategy for the vector store.

Examples

use rig::vector_store::{InMemoryVectorStoreBuilder, IndexStrategy};

let store = InMemoryVectorStoreBuilder::<String>::new()
    .index_strategy(IndexStrategy::LSH {
        num_tables: 5,
        num_hyperplanes: 10,
    })
    .build();

pub fn documents( self, documents: impl IntoIterator<Item = (D, OneOrMany<Embedding>)>, ) -> InMemoryVectorStoreBuilder<D>

Add documents with auto-generated IDs. IDs will have the form "doc{n}" where n is the index.

pub fn documents_with_ids( self, documents: impl IntoIterator<Item = (impl ToString, D, OneOrMany<Embedding>)>, ) -> InMemoryVectorStoreBuilder<D>

Add documents with explicit IDs.

pub fn documents_with_id_f( self, documents: impl IntoIterator<Item = (D, OneOrMany<Embedding>)>, f: fn(&D) -> String, ) -> InMemoryVectorStoreBuilder<D>

Add documents with IDs generated by a function.

pub fn build(self) -> InMemoryVectorStore<D>

Build the InMemoryVectorStore with the configured settings.

Trait Implementations

impl<D> Default for InMemoryVectorStoreBuilder<D>

where D: Serialize + Eq,

fn default() -> InMemoryVectorStoreBuilder<D>

Returns the “default value” for a type.