weavatrix-search-vector 0.3.1

Persistent, mutable, bounded vector candidate search for Rust and Weavatrix
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38

use crate::error::SearchError;
use crate::hit::SearchHit;
use crate::hnsw::VectorIndex;
use crate::metadata::{MetadataFilter, MetadataIndex};
use crate::storage::MappedVectorIndex;

impl MetadataIndex {
    /// Searches an owned index and guarantees enough filtered hits through the
    /// index's exact fallback.
    ///
    /// # Errors
    ///
    /// Returns a typed query or allocation error.
    pub fn search(
        &self,
        index: &VectorIndex,
        query: &[f32],
        count: usize,
        filter: &MetadataFilter,
    ) -> Result<Vec<SearchHit>, SearchError> {
        index.search_filtered(query, count, |key| self.matches(key, filter))
    }

    /// Searches a memory-mapped index with the same filter semantics.
    ///
    /// # Errors
    ///
    /// Returns a typed query or allocation error.
    pub fn search_mapped(
        &self,
        index: &MappedVectorIndex,
        query: &[f32],
        count: usize,
        filter: &MetadataFilter,
    ) -> Result<Vec<SearchHit>, SearchError> {
        index.search_filtered(query, count, |key| self.matches(key, filter))
    }
}