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

feat: generate improved answers

Browse source
This commit is contained in:
lxobr 2025-10-20 23:45:18 +02:00
parent 1e1fac3261
commit ce418828b4
1 changed files with 8 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
cognee/tasks/feedback/generate_improved_answers.py
View file

@ -65,9 +65,10 @@ async def _generate_improved_answer_for_single_interaction(
) )
retrieved_context = await retriever.get_context(query_text) retrieved_context = await retriever.get_context(query_text)
completion, new_context_text = await retriever.get_structured_completion( completion = await retriever.get_structured_completion(
query=query_text, context=retrieved_context, response_model=ImprovedAnswerResponse query=query_text, context=retrieved_context, response_model=ImprovedAnswerResponse
) )
new_context_text = await retriever.resolve_edges_to_text(retrieved_context)
if completion: if completion:
return { return {
@ -114,8 +115,6 @@ async def generate_improved_answers(
) )
improved_answers: List[Dict] = [] improved_answers: List[Dict] = []
successful_count = 0
failed_count = 0
for feedback_interaction in feedback_interactions: for feedback_interaction in feedback_interactions:
result = await _generate_improved_answer_for_single_interaction( result = await _generate_improved_answer_for_single_interaction(
@ -124,9 +123,12 @@ async def generate_improved_answers(
if result: if result:
improved_answers.append(result) improved_answers.append(result)
successful_count += 1
else: else:
failed_count += 1 logger.warning(
"Failed to generate improved answer",
question=feedback_interaction.get("question"),
interaction_id=feedback_interaction.get("interaction_id"),
)
logger.info("Generated improved answers", successful=successful_count, failed=failed_count) logger.info("Generated improved answers", count=len(improved_answers))
return improved_answers return improved_answers