update

Signed-off-by: Sajid Alam <sajid_alam@mckinsey.com>

package/kedro_viz/api/rest/responses.py

@@ -13,7 +13,7 @@
 
 from kedro_viz.api.rest.utils import get_package_compatibilities
 from kedro_viz.data_access import data_access_manager
-from kedro_viz.models.flowchart.flowchart import (
+from kedro_viz.models.flowchart import (
     DataNode,
     DataNodeMetadata,
     ParametersNodeMetadata,

package/kedro_viz/data_access/managers.py

@@ -14,7 +14,7 @@
 
 from kedro_viz.constants import DEFAULT_REGISTERED_PIPELINE_ID, ROOT_MODULAR_PIPELINE_ID
 from kedro_viz.integrations.utils import UnavailableDataset
-from kedro_viz.models.flowchart.flowchart import (
+from kedro_viz.models.flowchart import (
     DataNode,
     GraphEdge,
     GraphNode,

package/kedro_viz/data_access/repositories/graph.py

@@ -3,7 +3,7 @@
 # pylint: disable=missing-class-docstring,missing-function-docstring
 from typing import Dict, Generator, List, Optional, Set
 
-from kedro_viz.models.flowchart.flowchart import GraphEdge, GraphNode
+from kedro_viz.models.flowchart import GraphEdge, GraphNode
 
 
 class GraphNodesRepository:

package/kedro_viz/data_access/repositories/modular_pipelines.py

@@ -9,7 +9,7 @@
 from kedro.pipeline.node import Node as KedroNode
 
 from kedro_viz.constants import ROOT_MODULAR_PIPELINE_ID
-from kedro_viz.models.flowchart.flowchart import (
+from kedro_viz.models.flowchart import (
     GraphNode,
     GraphNodeType,
     ModularPipelineChild,

package/kedro_viz/data_access/repositories/registered_pipelines.py

@@ -4,7 +4,7 @@
 from collections import OrderedDict, defaultdict
 from typing import Dict, List, Optional, Set
 
-from kedro_viz.models.flowchart.flowchart import RegisteredPipeline
+from kedro_viz.models.flowchart import RegisteredPipeline
 
 
 class RegisteredPipelinesRepository:

package/kedro_viz/data_access/repositories/tags.py

@@ -3,7 +3,7 @@
 # pylint: disable=missing-class-docstring,missing-function-docstring
 from typing import Iterable, List, Set
 
-from kedro_viz.models.flowchart.flowchart import Tag
+from kedro_viz.models.flowchart import Tag
 
 
 class TagsRepository:

package/kedro_viz/models/flowchart/edges.py

@@ -1,5 +1,6 @@
 from pydantic import BaseModel
 
+
 class GraphEdge(BaseModel, frozen=True):
     """Represent an edge in the graph
 

package/kedro_viz/models/flowchart/entities.py

@@ -0,0 +1,30 @@
+from pydantic import BaseModel, Field, ValidationInfo, field_validator
+from typing import Optional
+
+
+class NamedEntity(BaseModel):
+    """Represent a named entity (Tag/Registered Pipeline) in a Kedro project."""
+
+    id: str
+    name: Optional[str] = Field(
+        default=None,
+        validate_default=True,
+        description="The name of the entity",
+    )
+
+    @field_validator("name")
+    @classmethod
+    def set_name(cls, _, info: ValidationInfo):
+        assert "id" in info.data
+        return info.data["id"]
+
+
+class RegisteredPipeline(NamedEntity):
+    """Represent a registered pipeline in a Kedro project."""
+
+
+class Tag(NamedEntity):
+    """Represent a tag in a Kedro project."""
+
+    def __hash__(self) -> int:
+        return hash(self.id)

package/kedro_viz/models/flowchart/enums.py

@@ -0,0 +1,10 @@
+from enum import Enum
+
+
+class GraphNodeType(str, Enum):
+    """Represent all possible node types in the graph representation of a Kedro pipeline."""
+
+    TASK = "task"
+    DATA = "data"
+    PARAMETERS = "parameters"
+    MODULAR_PIPELINE = "modularPipeline"  # CamelCase for frontend compatibility