From 7dc36bb4c78fe1b834ac3e788425340c10d21af0 Mon Sep 17 00:00:00 2001 From: Boris Arzentar Date: Mon, 3 Jun 2024 15:04:26 +0200 Subject: [PATCH] fix: await search in tests --- cognee/tests/test_library.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/cognee/tests/test_library.py b/cognee/tests/test_library.py index a0256a9aa..e279bcc51 100755 --- a/cognee/tests/test_library.py +++ b/cognee/tests/test_library.py @@ -48,25 +48,25 @@ async def main(): await cognee.cognify([ "cs_explanations"]) - search_results = cognee.search("SIMILARITY", "computer science") + search_results = await cognee.search("SIMILARITY", "computer science") assert len(search_results) != 0, "The search results list is empty." print("The search results list is not empty.") - search_results = cognee.search("CATEGORIES", "DefaultGraphModel__default_user") + search_results = await cognee.search("CATEGORIES", "DefaultGraphModel__default_user") assert len(search_results) != 0, "The search results list is empty." print("The search results list is not empty.") - search_results = cognee.search("NEIGHBOR", "DefaultGraphModel__default_user") + search_results = await cognee.search("NEIGHBOR", "DefaultGraphModel__default_user") assert len(search_results) != 0, "The search results list is empty." print("The search results list is not empty.") - search_results = cognee.search("SUMMARY", "Work and computers") + search_results = await cognee.search("SUMMARY", "Work and computers") assert len(search_results) != 0, "The search results list is empty." print("The search results list is not empty.") - search_results = cognee.search("ADJACENT", "DefaultGraphModel__default_user") + search_results = await cognee.search("ADJACENT", "DefaultGraphModel__default_user") assert len(search_results) != 0, "The search results list is empty." print("The search results list is not empty.")