added models to create graph_template

Author: NiveditJain

state-manager/app/models/db/graph_template_model.py

@@ -0,0 +1,23 @@
+from .base import BaseDatabaseModel
+from pydantic import Field
+from typing import Optional, List
+from ..graph_template_validation_status import GraphTemplateValidationStatus
+from ..node_template_model import NodeTemplate
+from pymongo import IndexModel
+
+
+class GraphTemplate(BaseDatabaseModel):
+    name: str = Field(..., description="Name of the graph")
+    namespace: str = Field(..., description="Namespace of the graph")
+    nodes: List[NodeTemplate] = Field(..., description="Nodes of the graph")
+    validation_status: GraphTemplateValidationStatus = Field(..., description="Validation status of the graph")
+    validation_errors: Optional[List[str]] = Field(None, description="Validation errors of the graph")
+
+    class Settings:
+        indexes = [
+            IndexModel(
+                keys=[("name", 1), ("namespace", 1)],
+                unique=True,
+                name="unique_name_namespace"
+            )
+        ]

state-manager/app/models/graph_models.py

@@ -0,0 +1,18 @@
+from .node_template_model import NodeTemplate
+from pydantic import BaseModel
+from typing import List
+from datetime import datetime
+
+
+class UpsertGraphTemplateRequest(BaseModel):
+    name: str
+    namespace: str
+    nodes: List[NodeTemplate]
+
+
+class UpsertGraphTemplateResponse(BaseModel):
+    name: str
+    namespace: str
+    nodes: List[NodeTemplate]
+    created_at: datetime
+    updated_at: datetime

state-manager/app/models/graph_template_validation_status.py

@@ -0,0 +1,8 @@
+from enum import Enum
+
+
+class GraphTemplateValidationStatus(Enum):
+    VALID = "VALID"
+    INVALID = "INVALID"
+    PENDING = "PENDING"
+    ONGOING = "ONGOING"

state-manager/app/models/node_template_model.py

@@ -0,0 +1,11 @@
+from pydantic import Field, BaseModel
+from typing import Any, Optional, List
+
+
+class NodeTemplate(BaseModel):
+    name: str = Field(..., description="Name of the node")
+    namespace: str = Field(..., description="Namespace of the node")
+    indentifier: str = Field(..., description="Indentifier of the node")
+    inputs: dict[str, Any] = Field(..., description="Inputs of the node")
+    store: dict[str, Any] = Field(..., description="Upsert data to store object for the node")
+    next_nodes: Optional[List[str]] = Field(None, description="Next nodes to execute")