From 9a46d145bbf40c407e66b2de776adf88f70a4556 Mon Sep 17 00:00:00 2001
From: hajdul88 <52442977+hajdul88@users.noreply.github.com>
Date: Mon, 18 Aug 2025 14:53:47 +0200
Subject: [PATCH] chore: fix search db tests

---
 cognee/tests/test_search_db.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/cognee/tests/test_search_db.py b/cognee/tests/test_search_db.py
index 9eafb5c0c..8a99ebd8f 100644
--- a/cognee/tests/test_search_db.py
+++ b/cognee/tests/test_search_db.py
@@ -44,16 +44,16 @@ async def main():
 
     await cognee.cognify([dataset_name])
 
-    context_gk = await GraphCompletionRetriever().get_context(
+    context_gk, _ = await GraphCompletionRetriever().get_context(
         query="Next to which country is Germany located?"
     )
-    context_gk_cot = await GraphCompletionCotRetriever().get_context(
+    context_gk_cot, _ = await GraphCompletionCotRetriever().get_context(
         query="Next to which country is Germany located?"
     )
-    context_gk_ext = await GraphCompletionContextExtensionRetriever().get_context(
+    context_gk_ext, _ = await GraphCompletionContextExtensionRetriever().get_context(
         query="Next to which country is Germany located?"
     )
-    context_gk_sum = await GraphSummaryCompletionRetriever().get_context(
+    context_gk_sum, _ = await GraphSummaryCompletionRetriever().get_context(
         query="Next to which country is Germany located?"
     )