sqlrite-engine 0.10.1

Light version of SQLite developed with Rust. Published as `sqlrite-engine` on crates.io; import as `use sqlrite::…`.
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

"""Memory-layer tests: recall combines vector + lexical + facts."""

from __future__ import annotations

from sqlrite_agent.memory import Memory, query_keywords


def test_keywords_filter_stop_words():
    kws = query_keywords("what is the weather like in lisbon today")
    assert "lisbon" in kws
    assert "the" not in kws
    assert "what" not in kws


def test_log_user_message_extracts_facts(memory: Memory):
    memory.log_message(
        conversation_id="c1",
        role="user",
        content="My dog's name is Mochi.",
    )
    facts = memory.all_facts()
    assert any(
        f.subject == "user.dog" and f.object == "Mochi" for f in facts
    )


def test_log_assistant_does_not_extract_facts(memory: Memory):
    memory.log_message(
        conversation_id="c1",
        role="assistant",
        content="Your dog's name is Mochi.",  # assistant echoing back
    )
    assert memory.all_facts() == []


def test_recall_pulls_messages_summaries_and_facts(memory: Memory):
    memory.log_message(
        conversation_id="c1",
        role="user",
        content="My dog's name is Mochi.",
    )
    memory.log_message(
        conversation_id="c1",
        role="user",
        content="Mochi loves carrots more than treats.",
    )
    memory.log_message(
        conversation_id="c1",
        role="user",
        content="The weather in Lisbon is sunny today.",
    )

    r = memory.recall("what does mochi like to eat", conversation_id="c1")
    assert any("Mochi" in f.object for f in r.facts)
    assert any("mochi" in m.content.lower() for m in r.messages)


def test_recall_works_without_conversation_filter(memory: Memory):
    memory.log_message(conversation_id="c1", role="user", content="apples are red")
    memory.log_message(conversation_id="c2", role="user", content="bananas are yellow")
    r = memory.recall("color of an apple")
    contents = " ".join(m.content for m in r.messages)
    assert "apple" in contents


def test_recall_persists_across_reopen(tmp_path, embedder):
    from sqlrite_agent.db import AgentDB

    path = str(tmp_path / "mem.sqlrite")
    db = AgentDB(path)
    mem = Memory(db, embedder)
    mem.log_message(
        conversation_id="c1",
        role="user",
        content="My favorite color is blue.",
    )
    db.close()

    db2 = AgentDB(path)
    mem2 = Memory(db2, embedder)
    r = mem2.recall("what color do i like", conversation_id="c1")
    assert any(
        f.predicate == "favorite_color" and f.object == "blue" for f in r.facts
    )
    db2.close()