From 952618d5616079d1e0fa45f4e36e57dfe061fc1a Mon Sep 17 00:00:00 2001 From: Andrej Milicevic Date: Tue, 20 Jan 2026 10:18:56 +0100 Subject: [PATCH] fix: small class definition change in cot retriever --- .../graph_completion_cot_retriever.py | 21 ++++++++++--------- 1 file changed, 11 insertions(+), 10 deletions(-) diff --git a/cognee/modules/retrieval/graph_completion_cot_retriever.py b/cognee/modules/retrieval/graph_completion_cot_retriever.py index 0a4dfb9b6..a985900e8 100644 --- a/cognee/modules/retrieval/graph_completion_cot_retriever.py +++ b/cognee/modules/retrieval/graph_completion_cot_retriever.py @@ -30,18 +30,19 @@ class QueryState: enable as many parallel calls to llms as possible. """ - completion: str = "" - triplets: List[Edge] = [] - context_text: str = "" + def __init__(self): + self.completion: str = "" + self.triplets: List[Edge] = [] + self.context_text: str = "" - answer_text: str = "" - valid_user_prompt: str = "" - valid_system_prompt: str = "" - reasoning: str = "" + self.answer_text: str = "" + self.valid_user_prompt: str = "" + self.valid_system_prompt: str = "" + self.reasoning: str = "" - followup_question: str = "" - followup_prompt: str = "" - followup_system: str = "" + self.followup_question: str = "" + self.followup_prompt: str = "" + self.followup_system: str = "" def _as_answer_text(completion: Any) -> str: