From adaf69c1278cbdb58968b7ec4e8999215bf8c2f9 Mon Sep 17 00:00:00 2001
From: Leon Luithlen <leon@topoteretes.com>
Date: Tue, 12 Nov 2024 09:05:51 +0100
Subject: [PATCH] Readd infer_data_ontology models

---
 .../models/__pycache__/models.py              | 31 +++++++++++++++++++
 1 file changed, 31 insertions(+)
 create mode 100644 cognee/tasks/infer_data_ontology/models/__pycache__/models.py

diff --git a/cognee/tasks/infer_data_ontology/models/__pycache__/models.py b/cognee/tasks/infer_data_ontology/models/__pycache__/models.py
new file mode 100644
index 000000000..5b1108e6a
--- /dev/null
+++ b/cognee/tasks/infer_data_ontology/models/__pycache__/models.py
@@ -0,0 +1,31 @@
+from typing import Any, Dict, List, Optional, Union
+from pydantic import BaseModel, Field
+
+class RelationshipModel(BaseModel):
+    type: str
+    source: str
+    target: str
+
+class NodeModel(BaseModel):
+    node_id: str
+    name: str
+    default_relationship: Optional[RelationshipModel] = None
+    children: List[Union[Dict[str, Any], "NodeModel"]] = Field(default_factory=list)
+
+NodeModel.model_rebuild()
+
+
+class OntologyNode(BaseModel):
+    id: str = Field(..., description = "Unique identifier made from node name.")
+    name: str
+    description: str
+
+class OntologyEdge(BaseModel):
+    id: str
+    source_id: str
+    target_id: str
+    relationship_type: str
+
+class GraphOntology(BaseModel):
+    nodes: list[OntologyNode]
+    edges: list[OntologyEdge]
\ No newline at end of file