From ff2f1477bb380a004756ae834e3ac6354ed1d67d Mon Sep 17 00:00:00 2001 From: Dmitri Khokhlov Date: Wed, 1 May 2024 23:28:36 -0700 Subject: [PATCH] fix: TypeError: LongTermMemory.search() missing 1 required positional argument: 'latest_n' (#488) Signed-off-by: Dmitri Khokhlov --- tests/memory/long_term_memory_test.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/memory/long_term_memory_test.py b/tests/memory/long_term_memory_test.py index 578ec102c..3639054e3 100644 --- a/tests/memory/long_term_memory_test.py +++ b/tests/memory/long_term_memory_test.py @@ -20,7 +20,7 @@ def test_save_and_search(long_term_memory): metadata={"task": "test_task", "quality": 0.5}, ) long_term_memory.save(memory) - find = long_term_memory.search("test_task")[0] + find = long_term_memory.search("test_task", latest_n=5)[0] assert find["score"] == 0.5 assert find["datetime"] == "test_datetime" assert find["metadata"]["agent"] == "test_agent"