16 lines
584 B
Python
16 lines
584 B
Python
from src.memory import MemoryManager
|
|
|
|
|
|
def test_extract_memory_from_chat_skips_non_dict_messages(tmp_path):
|
|
# chat_history rows can be malformed (a non-dict slipping in from a partial
|
|
# session blob); the old loop did msg.get(...) and crashed on the first one.
|
|
m = MemoryManager(str(tmp_path))
|
|
history = [
|
|
{"role": "assistant", "content": "- remember to buy milk"},
|
|
"junk-msg",
|
|
None,
|
|
{"role": "user", "content": "hi"},
|
|
]
|
|
out = m.extract_memory_from_chat(history)
|
|
assert any(e["text"] == "remember to buy milk" for e in out)
|