From b6be61776a377e022ef683a7a6cbbff3767008d4 Mon Sep 17 00:00:00 2001
From: hajdul88 <52442977+hajdul88@users.noreply.github.com>
Date: Mon, 18 Aug 2025 13:50:21 +0200
Subject: [PATCH] fix: fixes tests

---
 cognee/modules/retrieval/graph_completion_retriever.py  | 2 +-
 cognee/tests/unit/modules/search/search_methods_test.py | 1 +
 2 files changed, 2 insertions(+), 1 deletion(-)

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