From 7370d9f2364b215206e1c2345e94d06b327f1786 Mon Sep 17 00:00:00 2001 From: Mayank Solanki Date: Thu, 26 Sep 2024 01:10:19 +0530 Subject: [PATCH] minor --- tests/test_main.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/tests/test_main.py b/tests/test_main.py index 8ed2224586..8141be5709 100644 --- 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