vantadb 0.1.4

VantaDB: An embedded persistent memory and vector retrieval engine for local-first AI applications.
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
39
40
41
42

//! Query Executor & Result Projection Modernized Test Suite
//! Part of the Vanta Certification ecosystem.

#[path = "../common/mod.rs"]
mod common;

use common::{TerminalReporter, VantaHarness};
use vantadb::index::{CPIndex, VectorRepresentations};

#[test]
fn engine_executor_certification() {
    let mut harness = VantaHarness::new("LOGIC LAYER (QUERY EXECUTOR)");

    harness.execute("Math: Cosine Similarity Projection", || {
        let vec_a = VectorRepresentations::Full(vec![1.0, 0.0, 0.0]);
        let vec_b = VectorRepresentations::Full(vec![1.0, 0.0, 0.0]);
        let vec_c = VectorRepresentations::Full(vec![0.0, 1.0, 0.0]);

        assert!((vec_a.cosine_similarity(&vec_b).unwrap() - 1.0).abs() < f32::EPSILON);
        assert!((vec_a.cosine_similarity(&vec_c).unwrap() - 0.0).abs() < f32::EPSILON);
        TerminalReporter::success("Scalar similarity math validated.");
    });

    harness.execute("Search: Bitset + Nearest Neighbor Projection", || {
        let idx = CPIndex::new();
        TerminalReporter::sub_step("Setting up tiered bitmask dataset...");
        // Match mask + High sim
        idx.add(1, 0b11, VectorRepresentations::Full(vec![1.0, 0.0]), 0);
        // Match mask + Low sim
        idx.add(2, 0b11, VectorRepresentations::Full(vec![0.0, 1.0]), 0);
        // Fails mask
        idx.add(3, 0b00, VectorRepresentations::Full(vec![1.0, 0.0]), 0);

        let res = idx.search_nearest(&[1.0, 0.0], None, None, 0b10, 2, None);

        assert_eq!(res.len(), 2, "Failed to ignore bitset-filtered nodes");
        assert_eq!(res[0].0, 1, "Incorrect result ranking");
        assert_eq!(res[1].0, 2);

        TerminalReporter::success("Bitset filter and NN ranking integrated correctly.");
    });
}