From 58071288ecf9b0c5fd69abd5d85128dc484410fc Mon Sep 17 00:00:00 2001 From: Andrej Milicevic Date: Tue, 20 Jan 2026 10:05:06 +0100 Subject: [PATCH] fix: tiny fix in cot retreiver --- cognee/modules/retrieval/graph_completion_cot_retriever.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/cognee/modules/retrieval/graph_completion_cot_retriever.py b/cognee/modules/retrieval/graph_completion_cot_retriever.py index 58c9bad15..0a4dfb9b6 100644 --- a/cognee/modules/retrieval/graph_completion_cot_retriever.py +++ b/cognee/modules/retrieval/graph_completion_cot_retriever.py @@ -257,7 +257,7 @@ class GraphCompletionCotRetriever(GraphCompletionRetriever): for batched_query, batched_query_state in query_state_tracker.items(): followup_args = { - "query": query, + "query": batched_query, "answer": batched_query_state.answer_text, "reasoning": batched_query_state.reasoning, }