tools fix and formatting (#2441)

2025-03-26 11:25:03 +05:30
parent 2517ccd489
commit 2004427acd
18 changed files with 536 additions and 1151 deletions
--- a/tests/test_main.py
+++ b/tests/test_main.py
@@ -140,7 +140,7 @@ def test_search(memory_instance, version, enable_graph):
        assert result["results"][0]["score"] == 0.9

    memory_instance.vector_store.search.assert_called_once_with(
-        query=[0.1, 0.2, 0.3], limit=100, filters={"user_id": "test_user"}
+        query="test query", vectors=[0.1, 0.2, 0.3], limit=100, filters={"user_id": "test_user"}
    )
    memory_instance.embedding_model.embed.assert_called_once_with("test query", "search")