tinycortex 0.1.1

Rust core for the TinyCortex memory system
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
121
122
123
124
125

//! End-to-end deterministic-extraction tests over the gmail + notion fixtures.
//!
//! Adapted from OpenHuman's `memory/ingestion/tests.rs`. The original drove the
//! full `UnifiedMemory::ingest_document` path and asserted on namespace query /
//! recall / graph reads; those storage surfaces are out of TinyCortex's
//! ownership boundary. Here we assert directly on the deterministic extractor's
//! recovered entities, relations, preferences, and decisions.

use super::{extract_document, ExtractionMode, MemoryIngestionConfig};

fn fixture(path: &str) -> String {
    let base = std::path::Path::new(env!("CARGO_MANIFEST_DIR"));
    std::fs::read_to_string(
        base.join("tests")
            .join("fixtures")
            .join("ingestion")
            .join(path),
    )
    .expect("fixture should load")
}

#[test]
fn gmail_fixture_extraction_recovers_required_signals() {
    let content = fixture("gmail_thread_example.txt");
    let result = extract_document(
        &content,
        "Memory integration plan for OpenHuman desktop",
        &MemoryIngestionConfig::default(),
    );

    assert!(result.entities.iter().any(|e| e.name == "SANIL JAIN"));
    assert!(result.entities.iter().any(|e| e.name == "RAVI KULKARNI"));
    assert!(result.entities.iter().any(|e| e.name == "ASHA MEHTA"));
    assert!(result.entities.iter().any(|e| e.name == "OPENHUMAN"));
    assert!(result.relations.iter().any(|r| r.subject == "OPENHUMAN"
        && r.predicate == "USES"
        && r.object.contains("JSON-RPC")));
    assert!(result
        .relations
        .iter()
        .any(|r| r.subject == "RAVI KULKARNI" && r.predicate == "OWNS"));
    assert!(result.preference_count >= 1);
    assert!(result.decision_count >= 1);
}

#[test]
fn notion_fixture_extraction_recovers_required_signals() {
    let content = fixture("notion_page_example.txt");
    let result = extract_document(
        &content,
        "OpenHuman Memory Layer Roadmap",
        &MemoryIngestionConfig::default(),
    );

    assert!(result.entities.iter().any(|e| e.name == "OPENHUMAN"));
    assert!(result.entities.iter().any(|e| e.name == "SANIL JAIN"));
    assert!(result.relations.iter().any(|r| r.subject == "OPENHUMAN"
        && r.predicate == "USES"
        && r.object.contains("JSON-RPC")));
    assert!(result
        .relations
        .iter()
        .any(|r| r.subject == "CORE CONTRACT LOCKED" && r.predicate == "HAS_DEADLINE"));
    assert!(result
        .relations
        .iter()
        .any(|r| r.subject == "SANIL JAIN" && r.predicate == "PREFERS"));
    assert!(result.preference_count >= 1);
    assert!(result.decision_count >= 1);
}

#[test]
fn chunk_mode_extraction_recovers_spatial_relations_and_recipient_units() {
    let config = MemoryIngestionConfig {
        extraction_mode: ExtractionMode::Chunk,
        relation_threshold: 0.1,
        ..MemoryIngestionConfig::default()
    };
    let result = extract_document(
        "Alice Smith sent the launch checklist to Bob Jones. \
         Kitchen is north of Garden. East Room is east of Office. \
         West Room is west of Kitchen. Garden is south of Kitchen.",
        "Office map handoff",
        &config,
    );

    assert!(result.entities.iter().any(|e| e.name == "ALICE SMITH"));
    assert!(result.entities.iter().any(|e| e.name == "BOB JONES"));
    assert!(result
        .relations
        .iter()
        .any(|r| { r.subject == "KITCHEN" && r.predicate == "NORTH_OF" && r.object == "GARDEN" }));
    assert!(result
        .relations
        .iter()
        .any(|r| { r.subject == "GARDEN" && r.predicate == "SOUTH_OF" && r.object == "KITCHEN" }));
}

#[test]
fn sentence_mode_extraction_recovers_east_and_west_spatial_relations() {
    let config = MemoryIngestionConfig {
        relation_threshold: 0.1,
        ..MemoryIngestionConfig::default()
    };
    let result = extract_document(
        "East Room is east of Office. West Room is west of Kitchen.",
        "Office directions",
        &config,
    );

    assert!(result
        .relations
        .iter()
        .any(|r| { r.subject == "EAST ROOM" && r.predicate == "EAST_OF" && r.object == "OFFICE" }));
    assert!(result
        .relations
        .iter()
        .any(|r| { r.subject == "OFFICE" && r.predicate == "WEST_OF" && r.object == "EAST ROOM" }));
    assert!(result.relations.iter().any(|r| {
        r.subject == "WEST ROOM" && r.predicate == "WEST_OF" && r.object == "KITCHEN"
    }));
    assert!(result.relations.iter().any(|r| {
        r.subject == "KITCHEN" && r.predicate == "EAST_OF" && r.object == "WEST ROOM"
    }));
}