Skip to content

Commit a4f1b4e

Browse files
abrookinsclaude
andcommitted
Fix client test mock to match new method signature
Update test assertion to match the expanded get_working_memory method signature that includes user_id, model_name, and context_window_max parameters. 🤖 Generated with [Claude Code](https://claude.ai/code) Co-Authored-By: Claude <[email protected]>
1 parent 71de1e7 commit a4f1b4e

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

agent-memory-client/tests/test_client.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -88,7 +88,13 @@ async def test_promote_working_memories_to_long_term(self, enhanced_test_client)
8888
)
8989

9090
assert result.status == "ok"
91-
mock_get.assert_called_once_with(session_id=session_id, namespace=None)
91+
mock_get.assert_called_once_with(
92+
session_id=session_id,
93+
user_id=None,
94+
namespace=None,
95+
model_name=None,
96+
context_window_max=None,
97+
)
9298
mock_create.assert_called_once_with(memories)
9399

94100
@pytest.mark.asyncio

0 commit comments

Comments
 (0)