diff --git a/cognee/modules/retrieval/graph_completion_retriever.py b/cognee/modules/retrieval/graph_completion_retriever.py index a8cdee3ab..c831d8550 100644 --- a/cognee/modules/retrieval/graph_completion_retriever.py +++ b/cognee/modules/retrieval/graph_completion_retriever.py @@ -145,7 +145,7 @@ class GraphCompletionRetriever(BaseRetriever): if len(triplets) == 0: logger.warning("Empty context was provided to the completion") - return "" + return "", triplets context = await self.resolve_edges_to_text(triplets) diff --git a/cognee/tests/unit/modules/search/search_methods_test.py b/cognee/tests/unit/modules/search/search_methods_test.py index 8e9afff1c..8645d965a 100644 --- a/cognee/tests/unit/modules/search/search_methods_test.py +++ b/cognee/tests/unit/modules/search/search_methods_test.py @@ -65,6 +65,7 @@ async def test_search( top_k=10, node_type=None, node_name=None, + save_interaction=False, ) # Verify result logging