Add infer param and version bump (#2389)

Co-authored-by: Deshraj Yadav <deshrajdry@gmail.com>
2025-03-18 01:04:58 +05:30
parent e25dc4b504
commit d7a26bd0c3
4 changed files with 26 additions and 7 deletions
--- a/tests/test_main.py
+++ b/tests/test_main.py
@@ -51,7 +51,7 @@ def test_add(memory_instance, version, enable_graph):
        assert result["results"] == [{"memory": "Test memory", "event": "ADD"}]

    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"}, True
    )

    # Remove the conditional assertion for _add_to_graph