Underground-Digital/Workflow-Engine

1from typing import Literal, Optional2 3from pydantic import BaseModel4 5from core.workflow.nodes.base import BaseNodeData6 7 8class AdvancedSettings(BaseModel):9    """10    Advanced setting.11    """12 13    group_enabled: bool14 15    class Group(BaseModel):16        """17        Group.18        """19 20        output_type: Literal["string", "number", "object", "array[string]", "array[number]", "array[object]"]21        variables: list[list[str]]22        group_name: str23 24    groups: list[Group]25 26 27class VariableAssignerNodeData(BaseNodeData):28    """29    Knowledge retrieval Node Data.30    """31 32    type: str = "variable-assigner"33    output_type: str34    variables: list[list[str]]35    advanced_settings: Optional[AdvancedSettings] = None36