gmakstutis/cognee
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix: track followup question in cot retriever

Browse source
This commit is contained in:
Andrej Milicevic 2026-01-20 11:55:41 +01:00
parent aa2712ddb8
commit 8eee3990f7
1 changed files with 6 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
cognee/modules/retrieval/graph_completion_cot_retriever.py
View file

@ -277,6 +277,12 @@ class GraphCompletionCotRetriever(GraphCompletionRetriever):
for batched_query_state in query_state_tracker.values()
]
)
for batched_query, batched_followup_question in zip(
query_batch, followup_question_batch
):
query_state_tracker[batched_query].followup_question = batched_followup_question
logger.info(
f"Chain-of-thought: round {round_idx} - follow-up questions: {followup_question_batch}"
)