Merge branch 'main' into user/parshva/mem0_1.0.0

tests/configs/test_prompts.py

@@ -17,3 +17,35 @@ def test_get_update_memory_messages():
     ##
     result = prompts.get_update_memory_messages(retrieved_old_memory_dict, response_content, None)
     assert result.startswith(prompts.DEFAULT_UPDATE_MEMORY_PROMPT)
+
+
+def test_get_update_memory_messages_empty_memory():
+    # Test with None for retrieved_old_memory_dict
+    result = prompts.get_update_memory_messages(
+        None, 
+        ["new fact"], 
+        None
+    )
+    assert "Current memory is empty" in result
+
+    # Test with empty list for retrieved_old_memory_dict
+    result = prompts.get_update_memory_messages(
+        [], 
+        ["new fact"], 
+        None
+    )
+    assert "Current memory is empty" in result
+
+
+def test_get_update_memory_messages_non_empty_memory():
+    # Non-empty memory scenario
+    memory_data = [{"id": "1", "text": "existing memory"}]
+    result = prompts.get_update_memory_messages(
+        memory_data, 
+        ["new fact"], 
+        None
+    )
+    # Check that the memory data is displayed
+    assert str(memory_data) in result
+    # And that the non-empty memory message is present
+    assert "current content of my memory" in result

tests/memory/test_kuzu.py

@@ -11,6 +11,7 @@ class TestKuzu:
         "alice": np.random.uniform(0.0, 0.9, 384).tolist(),
         "bob": np.random.uniform(0.0, 0.9, 384).tolist(),
         "charlie": np.random.uniform(0.0, 0.9, 384).tolist(),
+        "dave": np.random.uniform(0.0, 0.9, 384).tolist(),
     }
 
     @pytest.fixture
@@ -78,6 +79,7 @@ class TestKuzu:
         assert kuzu_memory.llm == mock_llm
         assert kuzu_memory.threshold == 0.7
 
+
     @patch("mem0.memory.kuzu_memory.EmbedderFactory")
     @patch("mem0.memory.kuzu_memory.LlmFactory")
     def test_kuzu(self, mock_llm_factory, mock_embedder_factory, mock_config, mock_embedding_model, mock_llm):
@@ -109,12 +111,21 @@ class TestKuzu:
         assert get_node_count(kuzu_memory) == 3
         assert get_edge_count(kuzu_memory) == 4
 
+        data3 = [
+            {"source": "dave", "destination": "alice", "relationship": "admires"}
+        ]
+        result = kuzu_memory._add_entities(data3, filters, {})
+        assert result[0] == [{"source": "dave", "relationship": "admires", "target": "alice"}]
+        assert get_node_count(kuzu_memory) == 4  # dave is new
+        assert get_edge_count(kuzu_memory) == 5
+
         results = kuzu_memory.get_all(filters)
         assert set([f"{result['source']}_{result['relationship']}_{result['target']}" for result in results]) == set([
             "alice_knows_bob",
             "bob_knows_charlie",
             "charlie_likes_alice",
-            "charlie_knows_alice"
+            "charlie_knows_alice",
+            "dave_admires_alice"
         ])
 
         results = kuzu_memory._search_graph_db(["bob"], filters, threshold=0.8)
@@ -125,15 +136,15 @@ class TestKuzu:
 
         result = kuzu_memory._delete_entities(data2, filters)
         assert result[0] == [{"source": "charlie", "relationship": "likes", "target": "alice"}]
-        assert get_node_count(kuzu_memory) == 3
-        assert get_edge_count(kuzu_memory) == 3
+        assert get_node_count(kuzu_memory) == 4
+        assert get_edge_count(kuzu_memory) == 4
 
         result = kuzu_memory._delete_entities(data1, filters)
         assert result[0] == [{"source": "alice", "relationship": "knows", "target": "bob"}]
         assert result[1] == [{"source": "bob", "relationship": "knows", "target": "charlie"}]
         assert result[2] == [{"source": "charlie", "relationship": "knows", "target": "alice"}]
-        assert get_node_count(kuzu_memory) == 3
-        assert get_edge_count(kuzu_memory) == 0
+        assert get_node_count(kuzu_memory) == 4
+        assert get_edge_count(kuzu_memory) == 1
 
         result = kuzu_memory.delete_all(filters)
         assert get_node_count(kuzu_memory) == 0