velai / dataflow /enums.py
cansik's picture
Upload folder via script
691f45a verified
from __future__ import annotations
from enum import Enum
class DataPortState(Enum):
CLEAN = "clean"
DIRTY = "dirty"
BLOCKED = "blocked"
class PortDirection(Enum):
INPUT = "input"
OUTPUT = "output"
class ConnectMultiplicity(Enum):
SINGLE = "single"
MULTIPLE = "multiple"
class DataTypeId(Enum):
TEXT = "text"
IMAGE = "image"
class NodeKind(Enum):
TEXT_DATA = "TextDataNode"
IMAGE_DATA = "ImageDataNode"
TEXT_TO_IMAGE = "TextToImageNode"