mentedb 0.24.4

A purpose-built database engine for AI agent memory
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

//! The value-update supersession loop, end to end: storing a corrected fact
//! with the same sentence frame must hide the outdated one from recall (the
//! live bug: "cortado" and "flat white" favorites coexisted and recall served
//! both).

use mentedb::MenteDb;
use mentedb::prelude::*;

fn now_us() -> u64 {
    std::time::SystemTime::now()
        .duration_since(std::time::UNIX_EPOCH)
        .unwrap()
        .as_micros() as u64
}

#[test]
fn corrected_fact_supersedes_and_recall_serves_only_it() {
    let dir = tempfile::tempdir().unwrap();
    let db = MenteDb::open(dir.path()).unwrap();
    let agent = AgentId::new();

    let mut old = MemoryNode::new(
        agent,
        MemoryType::Semantic,
        "The user's favorite coffee order is a cortado".into(),
        vec![1.0, 0.05, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0],
    );
    old.created_at = now_us();
    let old_id = old.id;
    db.store(old).unwrap();

    let mut new = MemoryNode::new(
        agent,
        MemoryType::Semantic,
        "The user's favorite coffee order is a flat white".into(),
        vec![0.99, 0.08, 0.01, 0.0, 0.0, 0.0, 0.0, 0.0],
    );
    new.created_at = now_us() + 1;
    let new_id = new.id;
    db.store(new).unwrap();

    let hits = db
        .recall_hybrid_at(
            &[1.0, 0.06, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0],
            Some("favorite coffee order"),
            5,
            now_us() + 10,
            None,
            None,
            None,
        )
        .unwrap();
    let ids: Vec<MemoryId> = hits.iter().map(|(id, _)| *id).collect();
    assert!(
        ids.contains(&new_id),
        "the corrected fact must be recallable"
    );
    assert!(
        !ids.contains(&old_id),
        "the superseded fact must be hidden from recall"
    );
}

#[test]
fn different_subjects_both_survive() {
    let dir = tempfile::tempdir().unwrap();
    let db = MenteDb::open(dir.path()).unwrap();
    let agent = AgentId::new();

    let mut dog = MemoryNode::new(
        agent,
        MemoryType::Semantic,
        "My dog is allergic to chicken".into(),
        vec![1.0, 0.05, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0],
    );
    dog.created_at = now_us();
    let dog_id = dog.id;
    db.store(dog).unwrap();

    let mut cat = MemoryNode::new(
        agent,
        MemoryType::Semantic,
        "My cat is allergic to chicken".into(),
        vec![0.99, 0.08, 0.01, 0.0, 0.0, 0.0, 0.0, 0.0],
    );
    cat.created_at = now_us() + 1;
    let cat_id = cat.id;
    db.store(cat).unwrap();

    let hits = db
        .recall_hybrid_at(
            &[1.0, 0.06, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0],
            Some("allergic to chicken"),
            5,
            now_us() + 10,
            None,
            None,
            None,
        )
        .unwrap();
    let ids: Vec<MemoryId> = hits.iter().map(|(id, _)| *id).collect();
    assert!(ids.contains(&dog_id), "the dog fact must survive");
    assert!(ids.contains(&cat_id), "the cat fact must survive");
}