minor

mem0ai · Sep 25, 2024 · 7370d9f · 7370d9f
1 parent a6af220
commit 7370d9f
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/tests/test_main.py b/tests/test_main.py
@@ -1,8 +1,10 @@
-import pytest
 import os
 from unittest.mock import Mock, patch
-from mem0.memory.main import Memory
+
+import pytest
+
 from mem0.configs.base import MemoryConfig
+from mem0.memory.main import Memory
 
 
 @pytest.fixture(autouse=True)
@@ -44,7 +46,7 @@ def test_add(memory_instance, version, enable_graph):
     assert result["relations"] == []
 
     memory_instance._add_to_vector_store.assert_called_once_with(
-        [{"role": "user", "content": "Test message"}], {"user_id": "test_user"}, {"user_id": "test_user"}
+        [{"role": "user", "content": "Test message"}], {"user_id": "test_user"}, {"user_id": "test_user"}, None, None
     )
 
     # Remove the conditional assertion for _add_to_graph