Example task extraction

This commit is contained in:
Vasilije 2024-08-06 18:18:27 +02:00
parent 80e4f298fc
commit 64f54a32c6

View file

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