aletheiadb 0.1.0

A high-performance bi-temporal graph database for LLM integration
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
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72

use aletheiadb::core::id::NodeId;
use aletheiadb::index::vector::{
    DistanceMetric, HnswConfig, HnswIndex, HnswIndexBuilder, Quantization, VectorIndex,
};
use std::fs::File;
use std::io::Write;

#[test]
fn test_verify_header_file_open_error() {
    let dir = tempfile::tempdir().unwrap();
    let path = dir.path().join("non_existent.index");
    let config = HnswConfig::default();

    let result = HnswIndex::load(&path, config);
    assert!(result.is_err());
    let msg = result.unwrap_err().to_string();
    assert!(msg.contains("Failed to open index file for verification"));
}

#[test]
fn test_verify_header_read_error() {
    let dir = tempfile::tempdir().unwrap();
    let path = dir.path().join("truncated.index");

    // Create a file smaller than 8 bytes
    let mut file = File::create(&path).unwrap();
    file.write_all(&[0u8; 4]).unwrap(); // Only 4 bytes

    let config = HnswConfig::default();

    let result = HnswIndex::load(&path, config);
    assert!(result.is_err());
    let msg = result.unwrap_err().to_string();
    assert!(msg.contains("Failed to read index header"));
}

#[test]
fn test_verify_header_quantization_coverage() {
    let dir = tempfile::tempdir().unwrap();

    // F16
    {
        let path = dir.path().join("f16_valid.index");
        let index = HnswIndexBuilder::new(16, DistanceMetric::Cosine)
            .quantization(Quantization::F16)
            .build()
            .unwrap();
        index.add(NodeId::new(1).unwrap(), &[1.0; 16]).unwrap();
        index.save(&path).unwrap();

        let config =
            HnswConfig::new(16, DistanceMetric::Cosine).with_quantization(Quantization::F16);
        let loaded = HnswIndex::load(&path, config);
        assert!(loaded.is_ok());
    }

    // I8
    {
        let path = dir.path().join("i8_valid.index");
        let index = HnswIndexBuilder::new(16, DistanceMetric::Cosine)
            .quantization(Quantization::I8)
            .build()
            .unwrap();
        index.add(NodeId::new(1).unwrap(), &[1.0; 16]).unwrap();
        index.save(&path).unwrap();

        let config =
            HnswConfig::new(16, DistanceMetric::Cosine).with_quantization(Quantization::I8);
        let loaded = HnswIndex::load(&path, config);
        assert!(loaded.is_ok());
    }
}