28 lines
947 B
Python
28 lines
947 B
Python
from traderai.memory import MemoryStore
|
|
|
|
|
|
def test_memory_store_recalls_saved_fact(tmp_path):
|
|
store = MemoryStore(str(tmp_path / "memory.sqlite3"))
|
|
store.remember("preference", "The user prefers Polaris Bits searches to include barter listings.", importance=5)
|
|
|
|
results = store.recall("Polaris barter")
|
|
|
|
assert results
|
|
assert "Polaris Bits" in results[0]["content"]
|
|
|
|
|
|
def test_memory_store_clear_selected_sections(tmp_path):
|
|
store = MemoryStore(str(tmp_path / "memory.sqlite3"))
|
|
store.remember("note", "Forgettable note")
|
|
store.add_conversation("user", "hello")
|
|
store.set_profile("configured_name", "Hudson")
|
|
|
|
deleted = store.clear(include_profile=False)
|
|
snapshot = store.inspect()
|
|
|
|
assert deleted["memories"] == 1
|
|
assert deleted["conversations"] == 1
|
|
assert snapshot["memories"] == []
|
|
assert snapshot["conversations"] == []
|
|
assert snapshot["profile"][0]["key"] == "configured_name"
|