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

fix: small class definition change in cot retriever

Browse source
This commit is contained in:
Andrej Milicevic 2026-01-20 10:18:56 +01:00
parent 58071288ec
commit 952618d561
1 changed files with 11 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -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: