[graph_memory]: improve delete/add graph memory (#2073)

2025-01-03 22:21:05 +05:30
parent 542153ad4f
commit 78a2ef41d7
7 changed files with 439 additions and 225 deletions
--- a/tests/test_proxy.py
+++ b/tests/test_proxy.py
@@ -94,4 +94,4 @@ def test_completions_create_with_system_message(mock_memory_client, mock_litellm

    call_args = mock_litellm.completion.call_args[1]
    assert call_args["messages"][0]["role"] == "system"
-    assert call_args["messages"][0]["content"] == MEMORY_ANSWER_PROMPT
+    assert call_args["messages"][0]["content"] == "You are a helpful assistant."