refactor: move workflow package to dify_graph (#32844)

api/dify_graph/nodes/code/entities.py

@@ -0,0 +1,54 @@
+from enum import StrEnum
+from typing import Annotated, Literal
+
+from pydantic import AfterValidator, BaseModel
+
+from dify_graph.nodes.base import BaseNodeData
+from dify_graph.nodes.base.entities import VariableSelector
+from dify_graph.variables.types import SegmentType
+
+
+class CodeLanguage(StrEnum):
+    PYTHON3 = "python3"
+    JINJA2 = "jinja2"
+    JAVASCRIPT = "javascript"
+
+
+_ALLOWED_OUTPUT_FROM_CODE = frozenset(
+    [
+        SegmentType.STRING,
+        SegmentType.NUMBER,
+        SegmentType.OBJECT,
+        SegmentType.BOOLEAN,
+        SegmentType.ARRAY_STRING,
+        SegmentType.ARRAY_NUMBER,
+        SegmentType.ARRAY_OBJECT,
+        SegmentType.ARRAY_BOOLEAN,
+    ]
+)
+
+
+def _validate_type(segment_type: SegmentType) -> SegmentType:
+    if segment_type not in _ALLOWED_OUTPUT_FROM_CODE:
+        raise ValueError(f"invalid type for code output, expected {_ALLOWED_OUTPUT_FROM_CODE}, actual {segment_type}")
+    return segment_type
+
+
+class CodeNodeData(BaseNodeData):
+    """
+    Code Node Data.
+    """
+
+    class Output(BaseModel):
+        type: Annotated[SegmentType, AfterValidator(_validate_type)]
+        children: dict[str, "CodeNodeData.Output"] | None = None
+
+    class Dependency(BaseModel):
+        name: str
+        version: str
+
+    variables: list[VariableSelector]
+    code_language: Literal[CodeLanguage.PYTHON3, CodeLanguage.JAVASCRIPT]
+    code: str
+    outputs: dict[str, Output]
+    dependencies: list[Dependency] | None = None