[Improvement] update LLM memory get function (#1162)

Co-authored-by: Deven Patel <deven298@yahoo.com>
2024-01-12 12:33:07 +05:30
parent f582c1fe25
commit c020e65a50
4 changed files with 46 additions and 12 deletions
--- a/tests/memory/test_chat_memory.py
+++ b/tests/memory/test_chat_memory.py
@@ -44,6 +44,10 @@ def test_get(chat_memory_instance):

    assert len(recent_memories) == 5

+    all_memories = chat_memory_instance.get(app_id, fetch_all=True)
+
+    assert len(all_memories) == 6
+

 def test_delete_chat_history(chat_memory_instance):
    app_id = "test_app"