src/backend/base/langflow/graph/vertex/schema.py

from enum import Enum

from typing_extensions import NotRequired, TypedDict


class NodeTypeEnum(str, Enum):
    NoteNode = "noteNode"
    GenericNode = "genericNode"


class Position(TypedDict):
    x: float
    y: float


class NodeData(TypedDict):
    id: str
    data: dict
    dragging: NotRequired[bool]
    height: NotRequired[int]
    width: NotRequired[int]
    position: NotRequired[Position]
    positionAbsolute: NotRequired[Position]
    selected: NotRequired[bool]
    parent_node_id: NotRequired[str]
    type: NotRequired[NodeTypeEnum]