From e0980361a1c7cfee22ce55a28750cc379a7299c0 Mon Sep 17 00:00:00 2001 From: Rita Aleksziev Date: Wed, 22 Jan 2025 10:51:08 +0100 Subject: [PATCH] include insights search in cognee option --- evals/qa_context_provider_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/evals/qa_context_provider_utils.py b/evals/qa_context_provider_utils.py index 8fe31918d..2cef1e628 100644 --- a/evals/qa_context_provider_utils.py +++ b/evals/qa_context_provider_utils.py @@ -70,7 +70,7 @@ def _insight_to_string(triplet: tuple) -> str: async def get_context_with_cognee( instance: dict, task_indices: list[int] = None, - search_types: list[SearchType] = [SearchType.SUMMARIES, SearchType.CHUNKS], + search_types: list[SearchType] = [SearchType.INSIGHTS, SearchType.SUMMARIES, SearchType.CHUNKS], ) -> str: await cognify_instance(instance, task_indices)