alopex-embedded 0.6.0

Embedded database interface for Alopex DB
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
43
44
45

use alopex_embedded::{Database, DatabaseOptions, Metric, TxnMode};

#[cfg_attr(not(feature = "lane_ci"), ignore)]
#[test]
fn vector_search_in_memory_orders_results() {
    let opts = DatabaseOptions::in_memory().with_memory_limit(64 * 1024);
    let db = Database::open_in_memory_with_options(opts).unwrap();

    let mut txn = db.begin(TxnMode::ReadWrite).unwrap();
    txn.upsert_vector(b"v1", b"m1", &[1.0, 0.0], Metric::Cosine)
        .unwrap();
    txn.upsert_vector(b"v2", b"m2", &[0.0, 1.0], Metric::Cosine)
        .unwrap();
    txn.upsert_vector(b"v3", b"m3", &[1.0, 1.0], Metric::Cosine)
        .unwrap();
    txn.commit().unwrap();

    let mut rtxn = db.begin(TxnMode::ReadOnly).unwrap();
    let res = rtxn
        .search_similar(&[1.0, 0.0], Metric::Cosine, 2, None)
        .unwrap();
    assert_eq!(res.len(), 2);
    assert_eq!(res[0].key, b"v1");
    assert_eq!(res[1].key, b"v3");
}

#[cfg_attr(not(feature = "lane_ci"), ignore)]
#[test]
fn vector_search_honors_memory_limit_option() {
    let opts = DatabaseOptions::in_memory().with_memory_limit(8 * 1024);
    let db = Database::open_in_memory_with_options(opts).unwrap();

    let mut txn = db.begin(TxnMode::ReadWrite).unwrap();
    txn.upsert_vector(b"l2:a", b"ma", &[0.0, 0.0], Metric::L2)
        .unwrap();
    txn.upsert_vector(b"l2:b", b"mb", &[3.0, 4.0], Metric::L2)
        .unwrap();
    txn.commit().unwrap();

    let mut rtxn = db.begin(TxnMode::ReadOnly).unwrap();
    let res = rtxn
        .search_similar(&[0.0, 0.0], Metric::L2, 1, None)
        .unwrap();
    assert_eq!(res[0].key, b"l2:a");
}