from collections.abc import Sequence
from enum import StrEnum

from dify_graph.nodes.base import BaseNodeData


class WriteMode(StrEnum):
    OVER_WRITE = "over-write"
    APPEND = "append"
    CLEAR = "clear"


class VariableAssignerData(BaseNodeData):
    assigned_variable_selector: Sequence[str]
    write_mode: WriteMode
    input_variable_selector: Sequence[str]