git-vdb 0.4.0

An embedded vector database backed by immutable Git trees
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

import os
from pathlib import Path
import tempfile
import unittest

from git_vdb import PersistentClient


class ClientTest(unittest.TestCase):
    def test_vector_crud_query_and_filters(self):
        with tempfile.TemporaryDirectory() as temporary:
            client = PersistentClient(Path(temporary) / "vectors.git")
            docs = client.get_or_create_collection("docs")
            docs.upsert(
                ids=["east", "north"],
                embeddings=[[1.0, 0.0], [0.0, 1.0]],
                metadatas=[{"kind": "guide"}, {"kind": "note"}],
                documents=["East guide", "North note"],
                batch_size=1,
            )
            self.assertEqual(docs.count(), 2)
            result = docs.query(
                query_embeddings=[[0.9, 0.1]],
                where={"kind": "guide"},
                n_results=1,
            )
            self.assertEqual(result["ids"], [["east"]])
            self.assertEqual(result["documents"], [["East guide"]])
            self.assertEqual(docs.get(ids=["north"])["ids"], ["north"])
            docs.delete(ids=["north"])
            self.assertEqual(docs.count(), 1)
            self.assertTrue(client.doctor()["valid"])


if __name__ == "__main__":
    unittest.main()