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

fix: cot retriever fix

Browse source
This commit is contained in:
Andrej Milicevic 2026-01-20 10:59:39 +01:00
parent 952618d561
commit 05b5add480
1 changed files with 3 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -182,12 +182,9 @@ class GraphCompletionCotRetriever(GraphCompletionRetriever):
query_state_tracker[batched_query].context_text = batched_context_text query_state_tracker[batched_query].context_text = batched_context_text
else: else:
# Find new triplets, and update existing query states # Find new triplets, and update existing query states
followup_triplets_batch = await self.get_context( triplets_batch = await self.get_context(query_batch=followup_question_batch)
query_batch=followup_question_batch
) for batched_query, batched_followup_triplets in zip(query_batch, triplets_batch):
for batched_query, batched_followup_triplets in zip(
query_batch, followup_triplets_batch
):
query_state_tracker[batched_query].triplets = list( query_state_tracker[batched_query].triplets = list(
dict.fromkeys( dict.fromkeys(
query_state_tracker[batched_query].triplets + batched_followup_triplets query_state_tracker[batched_query].triplets + batched_followup_triplets