gmakstutis/cognee
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

refactor: Rename tasks

Browse source
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
Show stats Download patch file Download diff file Expand all files Collapse all files

12
cognee/api/v1/cognify/memify.py
View file

@ -23,9 +23,9 @@ logger = get_logger("memify")
async def memify( async def memify(
preprocessing_tasks: List[Task], data_streaming_tasks: List[Task],
processing_tasks: List[Task] = [], data_processing_tasks: List[Task] = [],
postprocessing_tasks: List[Task] = [], data_persistence_tasks: List[Task] = [],
data: Optional[Any] = None, data: Optional[Any] = None,
datasets: Union[str, list[str], list[UUID]] = None, datasets: Union[str, list[str], list[UUID]] = None,
user: User = None, user: User = None,
@ -66,9 +66,9 @@ async def memify(
data = [memory_fragment] data = [memory_fragment]
memify_tasks = [ memify_tasks = [
*preprocessing_tasks, # Unpack tasks provided to memify pipeline *data_streaming_tasks, # Unpack tasks provided to memify pipeline
*processing_tasks, *data_processing_tasks,
*postprocessing_tasks, *data_persistence_tasks,
] ]
await setup() await setup()

4
examples/python/memify_coding_agent_example.py
View file

@ -62,8 +62,8 @@ async def main():
] ]
await memify( await memify(
preprocessing_tasks=subgraph_extraction_tasks, data_streaming_tasks=subgraph_extraction_tasks,
processing_tasks=rule_association_tasks, data_processing_tasks=rule_association_tasks,
node_name=["coding_rules"], node_name=["coding_rules"],
) )

4
examples/python/memify_coding_agent_example_chunks.py
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). # 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 # If data is explicitely specified in the arguments this specified data will be forwarded through the tasks instead
await memify( await memify(
preprocessing_tasks=subgraph_extraction_tasks, data_streaming_tasks=subgraph_extraction_tasks,
processing_tasks=coding_rules_association_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) # Find the new specific coding rules added to graph through memify (created based on chat conversation between team members)