From 64f54a32c6aed01f0da4b616f46cf9967d20a9c3 Mon Sep 17 00:00:00 2001
From: Vasilije <8619304+Vasilije1990@users.noreply.github.com>
Date: Tue, 6 Aug 2024 18:18:27 +0200
Subject: [PATCH] Example task extraction

---
 cognee/api/v1/tasks/graph_ontology/graph_ontology.py | 11 +++++++++++
 1 file changed, 11 insertions(+)
 create mode 100644 cognee/api/v1/tasks/graph_ontology/graph_ontology.py

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