klieo-memory-sqlite 0.2.0

SQLite-backed implementations of klieo-core's memory traits.
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

//! Integration test: sqlite-vec MATCH-based recall returns the
//! nearest fact under cosine ordering (FakeEmbedder produces unit
//! vectors so L2 ordering matches cosine).

#![cfg(feature = "sqlite-vec")]

use klieo_core::memory::{Fact, Scope};
use klieo_memory_sqlite::{FakeEmbedder, MemorySqlite};
use std::sync::Arc;

#[tokio::test]
async fn vec_recall_returns_closest_fact() {
    let mem = MemorySqlite::new(":memory:", Arc::new(FakeEmbedder::new(8)))
        .await
        .unwrap();
    mem.long_term
        .remember(
            Scope::Workspace("w".into()),
            Fact {
                text: "the cat sat on the mat".into(),
                metadata: serde_json::Value::Null,
            },
        )
        .await
        .unwrap();
    mem.long_term
        .remember(
            Scope::Workspace("w".into()),
            Fact {
                text: "completely unrelated topic about rust".into(),
                metadata: serde_json::Value::Null,
            },
        )
        .await
        .unwrap();
    let hits = mem
        .long_term
        .recall(Scope::Workspace("w".into()), "the cat sat on the mat", 1)
        .await
        .unwrap();
    assert_eq!(hits.len(), 1);
    assert_eq!(hits[0].text, "the cat sat on the mat");
}

#[tokio::test]
async fn vec_recall_isolates_by_scope() {
    let mem = MemorySqlite::new(":memory:", Arc::new(FakeEmbedder::new(8)))
        .await
        .unwrap();
    mem.long_term
        .remember(
            Scope::Workspace("alpha".into()),
            Fact {
                text: "in alpha workspace".into(),
                metadata: serde_json::Value::Null,
            },
        )
        .await
        .unwrap();
    mem.long_term
        .remember(
            Scope::Workspace("beta".into()),
            Fact {
                text: "in beta workspace".into(),
                metadata: serde_json::Value::Null,
            },
        )
        .await
        .unwrap();
    let hits = mem
        .long_term
        .recall(Scope::Workspace("alpha".into()), "any query", 10)
        .await
        .unwrap();
    assert_eq!(hits.len(), 1);
    assert_eq!(hits[0].text, "in alpha workspace");
}