diff --git a/cognee/api/v1/tasks/graph_ontology/graph_ontology.py b/cognee/api/v1/tasks/graph_ontology/graph_ontology.py new file mode 100644 index 000000000..7a09b4f29 --- /dev/null +++ b/cognee/api/v1/tasks/graph_ontology/graph_ontology.py @@ -0,0 +1,11 @@ + + +from cognee.modules.pipelines.tasks.Task import Task + +from cognee.modules.data.extraction.knowledge_graph.establish_graph_topology import establish_graph_topology +from cognee.shared.data_models import KnowledgeGraph + + + +async def ontology_task(): + return Task(establish_graph_topology, topology_model = KnowledgeGraph, task_config = { "batch_size": 10 }) \ No newline at end of file