diff --git a/cognee/api/v1/cognify/memify.py b/cognee/api/v1/cognify/memify.py index 8237059ec..3c5f7be0f 100644 --- a/cognee/api/v1/cognify/memify.py +++ b/cognee/api/v1/cognify/memify.py @@ -23,9 +23,9 @@ logger = get_logger("memify") async def memify( - preprocessing_tasks: List[Task], - processing_tasks: List[Task] = [], - postprocessing_tasks: List[Task] = [], + data_streaming_tasks: List[Task], + data_processing_tasks: List[Task] = [], + data_persistence_tasks: List[Task] = [], data: Optional[Any] = None, datasets: Union[str, list[str], list[UUID]] = None, user: User = None, @@ -66,9 +66,9 @@ async def memify( data = [memory_fragment] memify_tasks = [ - *preprocessing_tasks, # Unpack tasks provided to memify pipeline - *processing_tasks, - *postprocessing_tasks, + *data_streaming_tasks, # Unpack tasks provided to memify pipeline + *data_processing_tasks, + *data_persistence_tasks, ] await setup() diff --git a/examples/python/memify_coding_agent_example.py b/examples/python/memify_coding_agent_example.py index 004a840f8..c0bda215a 100644 --- a/examples/python/memify_coding_agent_example.py +++ b/examples/python/memify_coding_agent_example.py @@ -62,8 +62,8 @@ async def main(): ] await memify( - preprocessing_tasks=subgraph_extraction_tasks, - processing_tasks=rule_association_tasks, + data_streaming_tasks=subgraph_extraction_tasks, + data_processing_tasks=rule_association_tasks, node_name=["coding_rules"], ) diff --git a/examples/python/memify_coding_agent_example_chunks.py b/examples/python/memify_coding_agent_example_chunks.py index b07bcb815..639b97396 100644 --- a/examples/python/memify_coding_agent_example_chunks.py +++ b/examples/python/memify_coding_agent_example_chunks.py @@ -80,8 +80,8 @@ async def main(): # Memify accepts these tasks and orchestrates forwarding of graph data through these tasks (if data is not specified). # If data is explicitely specified in the arguments this specified data will be forwarded through the tasks instead await memify( - preprocessing_tasks=subgraph_extraction_tasks, - processing_tasks=coding_rules_association_tasks, + data_streaming_tasks=subgraph_extraction_tasks, + data_processing_tasks=coding_rules_association_tasks, ) # Find the new specific coding rules added to graph through memify (created based on chat conversation between team members)