Refactor: centralize node data hydration (#27771)

api/core/workflow/nodes/tool/tool_node.py

@@ -16,14 +16,12 @@ from core.tools.workflow_as_tool.tool import WorkflowTool
 from core.variables.segments import ArrayAnySegment, ArrayFileSegment
 from core.variables.variables import ArrayAnyVariable
 from core.workflow.enums import (
-    ErrorStrategy,
     NodeType,
     SystemVariableKey,
     WorkflowNodeExecutionMetadataKey,
     WorkflowNodeExecutionStatus,
 )
 from core.workflow.node_events import NodeEventBase, NodeRunResult, StreamChunkEvent, StreamCompletedEvent
-from core.workflow.nodes.base.entities import BaseNodeData, RetryConfig
 from core.workflow.nodes.base.node import Node
 from core.workflow.nodes.base.variable_template_parser import VariableTemplateParser
 from extensions.ext_database import db
@@ -42,7 +40,7 @@ if TYPE_CHECKING:
     from core.workflow.runtime import VariablePool
 
 
-class ToolNode(Node):
+class ToolNode(Node[ToolNodeData]):
     """
     Tool Node
     """
@@ -51,9 +49,6 @@ class ToolNode(Node):
 
     _node_data: ToolNodeData
 
-    def init_node_data(self, data: Mapping[str, Any]):
-        self._node_data = ToolNodeData.model_validate(data)
-
     @classmethod
     def version(cls) -> str:
         return "1"
@@ -498,24 +493,6 @@ class ToolNode(Node):
 
         return result
 
-    def _get_error_strategy(self) -> ErrorStrategy | None:
-        return self._node_data.error_strategy
-
-    def _get_retry_config(self) -> RetryConfig:
-        return self._node_data.retry_config
-
-    def _get_title(self) -> str:
-        return self._node_data.title
-
-    def _get_description(self) -> str | None:
-        return self._node_data.desc
-
-    def _get_default_value_dict(self) -> dict[str, Any]:
-        return self._node_data.default_value_dict
-
-    def get_base_node_data(self) -> BaseNodeData:
-        return self._node_data
-
     @property
     def retry(self) -> bool:
         return self._node_data.retry_config.retry_enabled