Struct PostgresVectorStore

pub struct PostgresVectorStore<Model: EmbeddingModel> { /* private fields */ }

Implementations

impl<Model: EmbeddingModel> PostgresVectorStore<Model>

pub fn new( model: Model, pg_pool: PgPool, documents_table: Option<String>, distance_function: PgVectorDistanceFunction, ) -> Self

pub fn with_defaults(model: Model, pg_pool: PgPool) -> Self

pub async fn insert_documents<Doc: Serialize + Embed + Send>( &self, documents: Vec<(Doc, OneOrMany<Embedding>)>, ) -> Result<(), VectorStoreError>

Trait Implementations

impl<Model: EmbeddingModel> VectorStoreIndex for PostgresVectorStore<Model>

async fn top_n<T: for<'a> Deserialize<'a> + Send>( &self, query: &str, n: usize, ) -> Result<Vec<(f64, String, T)>, VectorStoreError>

Get the top n documents based on the distance to the given query. The result is a list of tuples of the form (score, id, document)

async fn top_n_ids( &self, query: &str, n: usize, ) -> Result<Vec<(f64, String)>, VectorStoreError>

Same as top_n but returns the document ids only.