fix(server): fetch filtered dashboard memories beyond default page (#5753)
This commit is contained in:
@@ -618,6 +618,31 @@ class TestGetMemories:
|
||||
# 3. Verify the core logic: the param was mapped to the filters dict!
|
||||
_, kwargs = mock_memory.get_all.call_args
|
||||
assert kwargs["filters"] == {"user_id": "test_routing_user"}
|
||||
assert "top_k" not in kwargs
|
||||
|
||||
def test_get_memories_entity_filters_forward_top_k(self, client, mock_memory):
|
||||
response = client.get("/memories?user_id=test_routing_user&top_k=1000")
|
||||
|
||||
assert response.status_code == 200
|
||||
|
||||
_, kwargs = mock_memory.get_all.call_args
|
||||
assert kwargs["filters"] == {"user_id": "test_routing_user"}
|
||||
assert kwargs["top_k"] == 1000
|
||||
|
||||
def test_get_memories_admin_top_k_zero_not_defaulted(self, client, mock_memory):
|
||||
mock_memory.vector_store.list.return_value = []
|
||||
|
||||
response = client.get("/memories?top_k=0")
|
||||
|
||||
assert response.status_code == 200
|
||||
_, kwargs = mock_memory.vector_store.list.call_args
|
||||
assert kwargs["top_k"] == 0
|
||||
|
||||
def test_get_memories_rejects_top_k_above_limit(self, client, mock_memory):
|
||||
response = client.get("/memories?user_id=test_routing_user&top_k=1001")
|
||||
|
||||
assert response.status_code == 422
|
||||
mock_memory.get_all.assert_not_called()
|
||||
|
||||
|
||||
# ===========================================================================
|
||||
|
||||
Reference in New Issue
Block a user