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

//! Tests for the tool-call/tool-result stripping transform. Ported from
//! OpenHuman's inline `memory_archivist::clip` tests.

use super::*;
use chrono::Utc;

fn t(role: &str, content: &str, tool_calls: Option<&str>) -> Turn {
    Turn {
        role: role.into(),
        content: content.into(),
        tool_calls_json: tool_calls.map(|s| s.into()),
        timestamp: Utc::now(),
    }
}

#[test]
fn drops_tool_calls_json_on_assistant_turns() {
    let convo = vec![
        t("user", "what's the time?", None),
        t("assistant", "let me check", Some(r#"[{"name":"clock"}]"#)),
    ];
    let cleaned = clean_conversation(&convo);
    assert_eq!(cleaned.len(), 2);
    assert!(cleaned[1].tool_calls_json.is_none());
    assert_eq!(cleaned[1].content, "let me check");
}

#[test]
fn drops_tool_role_turns_entirely() {
    let convo = vec![
        t("user", "list files", None),
        t("assistant", "running ls", Some(r#"[{"name":"bash"}]"#)),
        t("tool", "a.txt\nb.txt", None),
        t("assistant", "two files", None),
    ];
    let cleaned = clean_conversation(&convo);
    assert_eq!(cleaned.len(), 3);
    assert!(cleaned.iter().all(|t| t.role != "tool"));
}

#[test]
fn preserves_user_and_system_turns_unchanged() {
    let convo = vec![t("system", "be brief", None), t("user", "hi", None)];
    let cleaned = clean_conversation(&convo);
    assert_eq!(cleaned.len(), 2);
    assert_eq!(cleaned[0].role, "system");
    assert_eq!(cleaned[1].content, "hi");
}