chore: adhere to memify input convention
This commit is contained in:
parent
2d6188523a
commit
b09e4b7cc4
1 changed files with 10 additions and 2 deletions
|
|
@ -1,6 +1,6 @@
|
||||||
from __future__ import annotations
|
from __future__ import annotations
|
||||||
|
|
||||||
from typing import Dict, List, Optional, Tuple
|
from typing import Any, Dict, List, Optional, Tuple
|
||||||
from uuid import UUID, uuid5, NAMESPACE_OID
|
from uuid import UUID, uuid5, NAMESPACE_OID
|
||||||
|
|
||||||
from cognee.infrastructure.llm import LLMGateway
|
from cognee.infrastructure.llm import LLMGateway
|
||||||
|
|
@ -177,8 +177,16 @@ async def _build_feedback_interaction_records(
|
||||||
return feedback_interaction_records
|
return feedback_interaction_records
|
||||||
|
|
||||||
|
|
||||||
async def extract_feedback_interactions(last_n: Optional[int] = None) -> List[FeedbackEnrichment]:
|
async def extract_feedback_interactions(
|
||||||
|
data: Any, last_n: Optional[int] = None
|
||||||
|
) -> List[FeedbackEnrichment]:
|
||||||
"""Extract negative feedback-interaction pairs and create FeedbackEnrichment DataPoints."""
|
"""Extract negative feedback-interaction pairs and create FeedbackEnrichment DataPoints."""
|
||||||
|
if not data or data == [{}]:
|
||||||
|
logger.info(
|
||||||
|
"No data passed to the extraction task (extraction task fetches data from graph directly)",
|
||||||
|
data=data,
|
||||||
|
)
|
||||||
|
|
||||||
graph_nodes, graph_edges = await _fetch_feedback_and_interaction_graph_data()
|
graph_nodes, graph_edges = await _fetch_feedback_and_interaction_graph_data()
|
||||||
if not graph_nodes:
|
if not graph_nodes:
|
||||||
return []
|
return []
|
||||||
|
|
|
||||||
Loading…
Add table
Reference in a new issue