refactor(api): move workflow knowledge nodes and trigger nodes (#33445)

api/dify_graph/nodes/loop/entities.py

@@ -4,7 +4,7 @@ from typing import Annotated, Any, Literal
 from pydantic import AfterValidator, BaseModel, Field, field_validator
 
 from dify_graph.entities.base_node_data import BaseNodeData
-from dify_graph.enums import NodeType
+from dify_graph.enums import BuiltinNodeTypes, NodeType
 from dify_graph.nodes.base import BaseLoopNodeData, BaseLoopState
 from dify_graph.utils.condition.entities import Condition
 from dify_graph.variables.types import SegmentType
@@ -41,7 +41,7 @@ class LoopVariableData(BaseModel):
 
 
 class LoopNodeData(BaseLoopNodeData):
-    type: NodeType = NodeType.LOOP
+    type: NodeType = BuiltinNodeTypes.LOOP
     loop_count: int  # Maximum number of loops
     break_conditions: list[Condition]  # Conditions to break the loop
     logical_operator: Literal["and", "or"]
@@ -61,7 +61,7 @@ class LoopStartNodeData(BaseNodeData):
     Loop Start Node Data.
     """
 
-    type: NodeType = NodeType.LOOP_START
+    type: NodeType = BuiltinNodeTypes.LOOP_START
 
 
 class LoopEndNodeData(BaseNodeData):
@@ -69,7 +69,7 @@ class LoopEndNodeData(BaseNodeData):
     Loop End Node Data.
     """
 
-    type: NodeType = NodeType.LOOP_END
+    type: NodeType = BuiltinNodeTypes.LOOP_END
 
 
 class LoopState(BaseLoopState):

api/dify_graph/nodes/loop/loop_end_node.py

@@ -1,4 +1,4 @@
-from dify_graph.enums import NodeType, WorkflowNodeExecutionStatus
+from dify_graph.enums import BuiltinNodeTypes, WorkflowNodeExecutionStatus
 from dify_graph.node_events import NodeRunResult
 from dify_graph.nodes.base.node import Node
 from dify_graph.nodes.loop.entities import LoopEndNodeData
@@ -9,7 +9,7 @@ class LoopEndNode(Node[LoopEndNodeData]):
     Loop End Node.
     """
 
-    node_type = NodeType.LOOP_END
+    node_type = BuiltinNodeTypes.LOOP_END
 
     @classmethod
     def version(cls) -> str:

api/dify_graph/nodes/loop/loop_node.py

@@ -7,8 +7,8 @@ from typing import TYPE_CHECKING, Any, Literal, cast
 
 from dify_graph.entities.graph_config import NodeConfigDictAdapter
 from dify_graph.enums import (
+    BuiltinNodeTypes,
     NodeExecutionType,
-    NodeType,
     WorkflowNodeExecutionMetadataKey,
     WorkflowNodeExecutionStatus,
 )
@@ -46,7 +46,7 @@ class LoopNode(LLMUsageTrackingMixin, Node[LoopNodeData]):
     Loop Node.
     """
 
-    node_type = NodeType.LOOP
+    node_type = BuiltinNodeTypes.LOOP
     execution_type = NodeExecutionType.CONTAINER
 
     @classmethod
@@ -250,11 +250,11 @@ class LoopNode(LLMUsageTrackingMixin, Node[LoopNodeData]):
             if isinstance(event, GraphNodeEventBase):
                 self._append_loop_info_to_event(event=event, loop_run_index=current_index)
 
-            if isinstance(event, GraphNodeEventBase) and event.node_type == NodeType.LOOP_START:
+            if isinstance(event, GraphNodeEventBase) and event.node_type == BuiltinNodeTypes.LOOP_START:
                 continue
             if isinstance(event, GraphNodeEventBase):
                 yield event
-            if isinstance(event, NodeRunSucceededEvent) and event.node_type == NodeType.LOOP_END:
+            if isinstance(event, NodeRunSucceededEvent) and event.node_type == BuiltinNodeTypes.LOOP_END:
                 reach_break_node = True
             if isinstance(event, GraphRunFailedEvent):
                 raise Exception(event.error)
@@ -315,12 +315,9 @@ class LoopNode(LLMUsageTrackingMixin, Node[LoopNodeData]):
 
             # variable selector to variable mapping
             try:
-                # Get node class
-                from dify_graph.nodes.node_mapping import get_node_type_classes_mapping
-
                 typed_sub_node_config = NodeConfigDictAdapter.validate_python(sub_node_config)
                 node_type = typed_sub_node_config["data"].type
-                node_mapping = get_node_type_classes_mapping()
+                node_mapping = Node.get_node_type_classes_mapping()
                 if node_type not in node_mapping:
                     continue
                 node_version = str(typed_sub_node_config["data"].version)

api/dify_graph/nodes/loop/loop_start_node.py

@@ -1,4 +1,4 @@
-from dify_graph.enums import NodeType, WorkflowNodeExecutionStatus
+from dify_graph.enums import BuiltinNodeTypes, WorkflowNodeExecutionStatus
 from dify_graph.node_events import NodeRunResult
 from dify_graph.nodes.base.node import Node
 from dify_graph.nodes.loop.entities import LoopStartNodeData
@@ -9,7 +9,7 @@ class LoopStartNode(Node[LoopStartNodeData]):
     Loop Start Node.
     """
 
-    node_type = NodeType.LOOP_START
+    node_type = BuiltinNodeTypes.LOOP_START
 
     @classmethod
     def version(cls) -> str: