From 820243d6c47e6c7f21a7fef5ae1008443c2aa6b2 Mon Sep 17 00:00:00 2001 From: vasilije Date: Sun, 13 Jul 2025 14:34:49 +0200 Subject: [PATCH] added fix --- cognee/tests/test_falkordb.py | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/cognee/tests/test_falkordb.py b/cognee/tests/test_falkordb.py index 7c3ceef64..a424d70a5 100755 --- a/cognee/tests/test_falkordb.py +++ b/cognee/tests/test_falkordb.py @@ -91,7 +91,16 @@ async def main(): from cognee.infrastructure.databases.vector import get_vector_engine vector_engine = get_vector_engine() - random_node = (await vector_engine.search("entity.name", "AI"))[0] + search_results = await vector_engine.search("Entity_name", "AI") + if not search_results: + # If "AI" is not found, try searching for "LLM" which is more likely to be an entity + search_results = await vector_engine.search("Entity_name", "LLM") + if not search_results: + # If still no results, try a broader search + search_results = await vector_engine.search("Entity_name", "language model") + + assert len(search_results) > 0, "No entities found in the vector database" + random_node = search_results[0] random_node_name = random_node.payload["text"] search_results = await cognee.search(