gmakstutis/cognee
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

test: tiny test fix

Browse source
This commit is contained in:
Andrej Milicevic 2026-01-19 10:56:50 +01:00
parent d258b1d7af
commit 8e0d112439
1 changed files with 6 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
cognee/tests/unit/modules/retrieval/graph_completion_retriever_context_extension_test.py
View file

@ -611,7 +611,12 @@ async def test_get_completion_batch_queries_context_extension_stops_early(mock_e
retriever = GraphCompletionContextExtensionRetriever()
with (
patch.object(retriever, "get_context", new_callable=AsyncMock, return_value=[[mock_edge]]),
patch.object(
retriever,
"get_context",
new_callable=AsyncMock,
return_value=[[mock_edge], [mock_edge]],
),
patch(
"cognee.modules.retrieval.graph_completion_retriever.resolve_edges_to_text",
return_value="Resolved context",