ragit 0.4.5

git-like rag pipeline
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

import json
from utils import cargo_run, goto_root, mk_and_cd_tmp_dir

def meta():
    goto_root()
    mk_and_cd_tmp_dir()
    cargo_run(["init"])
    metadata = {}

    for command, args in [
        ("get", ["key1"]),
        ("set", ["key1", "value1"]),
        ("get", ["key1"]),
        ("set", ["key1", "value2"]),
        ("get", ["key1"]),
        ("set", ["key2", "value3"]),
        ("remove", ["key1"]),
        ("remove-all", []),
    ]:
        if command == "get":
            key = args[0]

            if key not in metadata:
                assert cargo_run(["meta", "--get", key], check=False) != 0

            else:
                assert metadata[key] == cargo_run(["meta", "--get", key], stdout=True).strip()

        elif command == "set":
            key, value = args
            set_result = cargo_run(["meta", "--set", key, value], stdout=True)

            if key in metadata:
                prev_value = metadata[key]
                assert prev_value in set_result and value in set_result

            else:
                assert value in set_result

            metadata[key] = value

        elif command == "remove":
            key = args[0]

            if key in metadata:
                assert metadata[key] in cargo_run(["meta", "--remove", key], stdout=True)
                metadata.pop(key)

            else:
                assert cargo_run(["meta", "--remove", key], check=False) != 0

        assert json.loads(cargo_run(["meta", "--get-all", "--json"], stdout=True)) == metadata