diff --git a/cognee/tests/integration/retrieval/test_graph_completion_retriever.py b/cognee/tests/integration/retrieval/test_graph_completion_retriever.py index a79363b75..7367b353b 100644 --- a/cognee/tests/integration/retrieval/test_graph_completion_retriever.py +++ b/cognee/tests/integration/retrieval/test_graph_completion_retriever.py @@ -255,3 +255,14 @@ async def test_get_graph_completion_context_on_empty_graph(setup_test_environmen context = await retriever.get_context("Who works at Figma?") assert context == [], "Context should be empty on an empty graph" + + +@pytest.mark.asyncio +async def test_graph_completion_get_triplets_empty(setup_test_environment_empty): + """Integration test: verify GraphCompletionRetriever get_triplets handles empty graph.""" + retriever = GraphCompletionRetriever() + + triplets = await retriever.get_triplets("Who works at Figma?") + + assert isinstance(triplets, list), "Triplets should be a list" + assert len(triplets) == 0, "Should return empty list on empty graph"