refactor: Rename tasks
This commit is contained in:
parent
2847569616
commit
90ef8c30d2
3 changed files with 10 additions and 10 deletions
|
|
@ -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()
|
||||
|
|
|
|||
|
|
@ -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"],
|
||||
)
|
||||
|
||||
|
|
|
|||
|
|
@ -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)
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue