fix: tiny fix in cot retreiver

2026-01-20 10:05:06 +01:00 · 2026-01-20 10:05:06 +01:00 · 58071288ec
commit 58071288ec
parent 8c7b309199
1 changed files with 1 additions and 1 deletions
--- 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,
                    }