refactor: Rename tasks

This commit is contained in:
Igor Ilic 2025-09-03 16:16:55 +02:00
parent 2847569616
commit 90ef8c30d2
3 changed files with 10 additions and 10 deletions

View file

@ -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()

View file

@ -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"],
)

View file

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