diff --git a/cognee/tests/integration/retrieval/test_graph_completion_retriever_context_extension.py b/cognee/tests/integration/retrieval/test_graph_completion_retriever_context_extension.py index 0d5413ff8..c87de16ef 100644 --- a/cognee/tests/integration/retrieval/test_graph_completion_retriever_context_extension.py +++ b/cognee/tests/integration/retrieval/test_graph_completion_retriever_context_extension.py @@ -213,3 +213,14 @@ async def test_get_graph_completion_extension_context_on_empty_graph(setup_test_ assert all(isinstance(item, str) and item.strip() for item in answer), ( "Answer must contain only non-empty strings" ) + + +@pytest.mark.asyncio +async def test_graph_completion_extension_get_triplets_empty(setup_test_environment_empty): + """Integration test: verify GraphCompletionContextExtensionRetriever get_triplets handles empty graph.""" + retriever = GraphCompletionContextExtensionRetriever() + + 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"