mentedb 0.42.1

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
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120

//! The standing-rules (`scope:always`) cleanup policy lives in the engine so the
//! hosted platform, the local daemon, and every SDK share one behavior and it
//! cannot drift between them.

use mentedb::MenteDb;
use mentedb::prelude::*;
use mentedb_core::types::AgentId;

fn pinned(content: &str, extra_tags: &[&str]) -> MemoryNode {
    let mut node = MemoryNode::new(
        AgentId::nil(),
        MemoryType::Semantic,
        content.to_string(),
        vec![],
    );
    node.tags = std::iter::once("scope:always".to_string())
        .chain(extra_tags.iter().map(|t| t.to_string()))
        .collect();
    node
}

#[test]
fn prune_unpins_auto_keeps_manual_and_profile_and_dedups() {
    let dir = tempfile::tempdir().unwrap();
    let db = MenteDb::open(dir.path()).expect("open");

    // Auto-pinned (enrichment): must be un-pinned, memory kept.
    let auto = pinned("a distilled fact about deploys", &["source:enrichment"]);
    let auto_id = auto.id;
    db.store(auto).expect("store");

    // Explicitly pinned by the user: must be kept as scope:always.
    let manual = pinned("never add Co-Authored-By", &["source:manual"]);
    let manual_id = manual.id;
    db.store(manual).expect("store");

    // The user profile: engine-pinned deliberately, must be kept.
    let profile = pinned(
        "Nam is a technical founder",
        &["user_profile", "source:enrichment"],
    );
    let profile_id = profile.id;
    db.store(profile).expect("store");

    // Two exact-content duplicates: one is removed (healthiest kept).
    let dup1 = pinned("always use rtk", &["source:enrichment"]);
    let dup1_id = dup1.id;
    db.store(dup1).expect("store");
    let dup2 = pinned("always use rtk", &["source:enrichment"]);
    let dup2_id = dup2.id;
    db.store(dup2).expect("store");

    let report = db.prune_standing_rules().expect("prune");

    assert_eq!(report.total_always, 5);

    // Exactly one duplicate collapsed and forgotten.
    assert_eq!(report.duplicate_groups, 1);
    assert_eq!(report.pruned.len(), 1);
    let removed = report.pruned[0];
    assert!(removed == dup1_id || removed == dup2_id);
    assert!(
        db.get_memory(removed).is_err(),
        "removed duplicate must be forgotten"
    );

    let is_always = |id: MemoryId| {
        db.get_memory(id)
            .map(|n| n.tags.iter().any(|t| t == "scope:always"))
            .unwrap_or(false)
    };

    // Auto-pinned rule un-pinned but still present (recalled by relevance).
    assert!(!is_always(auto_id), "auto-pinned rule must be un-pinned");
    assert!(db.get_memory(auto_id).is_ok(), "un-pinned memory is kept");

    // The surviving duplicate is enrichment-sourced, so also un-pinned.
    let surviving_dup = if removed == dup1_id { dup2_id } else { dup1_id };
    assert!(!is_always(surviving_dup));

    // Manual pin and the profile survive with scope:always intact.
    assert!(is_always(manual_id), "manual pin must be kept pinned");
    assert!(is_always(profile_id), "profile must stay pinned");

    // The report reflects exactly what was un-pinned.
    assert!(report.unpinned.contains(&auto_id));
    assert!(report.unpinned.contains(&surviving_dup));
    assert!(!report.unpinned.contains(&manual_id));
    assert!(!report.unpinned.contains(&profile_id));
}

/// Pruning finds pinned rules through the tag index: with thousands of
/// ordinary memories it loads only the pinned ones, not the whole database,
/// because callers hold the account write lock while it runs.
#[test]
fn prune_loads_only_pinned_memories() {
    let dir = tempfile::tempdir().expect("tempdir");
    let db = MenteDb::open(dir.path()).expect("open");
    for i in 0..2_000 {
        let node = MemoryNode::new(
            AgentId::nil(),
            MemoryType::Episodic,
            format!("ordinary memory {i}"),
            vec![],
        );
        db.store(node).expect("store");
    }
    for i in 0..3 {
        db.store(pinned(&format!("pinned rule {i}"), &["source:manual"]))
            .expect("store pinned");
    }
    let before = db.metrics().memory_loads;
    let report = db.prune_standing_rules().expect("prune");
    let loaded = db.metrics().memory_loads - before;
    assert_eq!(report.total_always, 3);
    assert!(
        loaded < 50,
        "prune loaded {loaded} memories for 3 pinned rules out of 2003"
    );
}