diff --git a/cognee/tasks/feedback/extract_feedback_interactions.py b/cognee/tasks/feedback/extract_feedback_interactions.py index c80173188..521d35472 100644 --- a/cognee/tasks/feedback/extract_feedback_interactions.py +++ b/cognee/tasks/feedback/extract_feedback_interactions.py @@ -177,9 +177,7 @@ async def _build_feedback_interaction_records( return feedback_interaction_records -async def extract_feedback_interactions( - subgraphs: List, last_n: Optional[int] = None -) -> List[FeedbackEnrichment]: +async def extract_feedback_interactions(last_n: Optional[int] = None) -> List[FeedbackEnrichment]: """Extract negative feedback-interaction pairs and create FeedbackEnrichment DataPoints.""" graph_nodes, graph_edges = await _fetch_feedback_and_interaction_graph_data() if not graph_nodes: diff --git a/examples/python/feedback_enrichment_minimal_example.py b/examples/python/feedback_enrichment_minimal_example.py index 3af838214..11ef20830 100644 --- a/examples/python/feedback_enrichment_minimal_example.py +++ b/examples/python/feedback_enrichment_minimal_example.py @@ -47,7 +47,7 @@ async def run_question_and_submit_feedback(question_text: str) -> bool: await cognee.search( query_type=SearchType.FEEDBACK, query_text=feedback_text, - last_k=2, + last_k=1, ) return mentions_mallory