victor-db 0.1.2

A browser-optimized vector database
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
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120

use crate::memory::{Db, DirectoryHandle};

#[tokio::test]
async fn store_and_retrieve() {
    let embedding = vec![1.0, 2.0, 3.0];

    let mut victor = Db::new(DirectoryHandle::new());

    victor.write("hello", embedding.clone(), vec![]).await;

    let result = victor
        .find_nearest_neighbor(embedding, vec![])
        .await
        .unwrap()
        .content;

    assert_eq!(result, "hello".to_string());
}

#[tokio::test]
async fn store_two_and_retrieve() {
    let embedding_1 = vec![1.0, 2.0, 3.0];
    let embedding_2 = vec![-1.0, -2.0, -3.0];

    let mut victor = Db::new(DirectoryHandle::new());

    victor.write("hello", embedding_1.clone(), vec![]).await;
    victor.write("goodbye", embedding_2.clone(), vec![]).await;

    {
        let result = victor
            .find_nearest_neighbor(embedding_1, vec![])
            .await
            .unwrap()
            .content;

        assert_eq!(result, "hello".to_string());
    }
    {
        let result = victor
            .find_nearest_neighbor(embedding_2, vec![])
            .await
            .unwrap()
            .content;

        assert_eq!(result, "goodbye".to_string());
    }
}

#[tokio::test]
async fn store_two_and_retrieve_with_tags() {
    let embedding_1 = vec![1.0, 2.0, 3.0];
    let embedding_2 = vec![-1.0, -2.0, -3.0];

    let mut victor = Db::new(DirectoryHandle::new());

    victor
        .write("hello", embedding_1.clone(), vec!["greetings".to_string()])
        .await;
    victor
        .write("goodbye", embedding_2.clone(), vec!["goodbyes".to_string()])
        .await;

    {
        let result = victor
            .find_nearest_neighbor(embedding_1.clone(), vec![])
            .await
            .unwrap()
            .content;

        assert_eq!(result, "hello".to_string());
    }
    {
        let result = victor
            .find_nearest_neighbor(embedding_2.clone(), vec![])
            .await
            .unwrap()
            .content;

        assert_eq!(result, "goodbye".to_string());
    }

    {
        let result = victor
            .find_nearest_neighbor(embedding_1.clone(), vec!["goodbyes".to_string()])
            .await
            .unwrap()
            .content;

        assert_eq!(result, "goodbye".to_string());
    }
    {
        let result = victor
            .find_nearest_neighbor(embedding_2, vec!["greetings".to_string()])
            .await
            .unwrap()
            .content;

        assert_eq!(result, "hello".to_string());
    }
    {
        let result = victor
            .find_nearest_neighbor(embedding_1, vec!["mysterious".to_string()])
            .await;

        assert_eq!(result, None);
    }
}

#[should_panic]
#[tokio::test]
async fn incompatible_size_panic() {
    let embedding_1 = vec![1.0, 2.0, 3.0];
    let embedding_2 = vec![1.0, 2.0, 3.0, 4.0];

    let mut victor = Db::new(DirectoryHandle::new());

    victor.write("hello", embedding_1, vec![]).await;
    victor.write("hello", embedding_2, vec![]).await;
}