AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-14 00:35:56 +08:00
Code Issues Packages Projects Releases Wiki Activity

chore(api): improve type hints for BaseNode and its subclasses (#15826)

Browse Source
This commit is contained in:
QuantumGhost 2025-03-14 18:09:11 +08:00 committed by GitHub
parent 5646442931
commit 23ed3a520b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
4 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
api/core/workflow/nodes/base/node.py
View File

@ -22,7 +22,7 @@ GenericNodeData = TypeVar("GenericNodeData", bound=BaseNodeData)
class BaseNode(Generic[GenericNodeData]): class BaseNode(Generic[GenericNodeData]):
_node_data_cls: type[BaseNodeData] _node_data_cls: type[GenericNodeData]
_node_type: NodeType _node_type: NodeType
def __init__( def __init__(
@ -57,7 +57,7 @@ class BaseNode(Generic[GenericNodeData]):
self.node_id = node_id self.node_id = node_id
node_data = self._node_data_cls.model_validate(config.get("data", {})) node_data = self._node_data_cls.model_validate(config.get("data", {}))
self.node_data = cast(GenericNodeData, node_data) self.node_data = node_data
@abstractmethod @abstractmethod
def _run(self) -> NodeRunResult | Generator[Union[NodeEvent, "InNodeEvent"], None, None]: def _run(self) -> NodeRunResult | Generator[Union[NodeEvent, "InNodeEvent"], None, None]:

2
api/core/workflow/nodes/iteration/iteration_start_node.py
View File

@ -5,7 +5,7 @@ from core.workflow.nodes.iteration.entities import IterationStartNodeData
from models.workflow import WorkflowNodeExecutionStatus from models.workflow import WorkflowNodeExecutionStatus
class IterationStartNode(BaseNode): class IterationStartNode(BaseNode[IterationStartNodeData]):
""" """
Iteration Start Node. Iteration Start Node.
""" """

2
api/core/workflow/nodes/loop/loop_start_node.py
View File

@ -5,7 +5,7 @@ from core.workflow.nodes.loop.entities import LoopStartNodeData
from models.workflow import WorkflowNodeExecutionStatus from models.workflow import WorkflowNodeExecutionStatus
class LoopStartNode(BaseNode): class LoopStartNode(BaseNode[LoopStartNodeData]):
""" """
Loop Start Node. Loop Start Node.
""" """

4
api/core/workflow/nodes/variable_assigner/v1/node.py
View File

@ -1,6 +1,6 @@
from core.variables import SegmentType, Variable from core.variables import SegmentType, Variable
from core.workflow.entities.node_entities import NodeRunResult from core.workflow.entities.node_entities import NodeRunResult
from core.workflow.nodes.base import BaseNode, BaseNodeData from core.workflow.nodes.base import BaseNode
from core.workflow.nodes.enums import NodeType from core.workflow.nodes.enums import NodeType
from core.workflow.nodes.variable_assigner.common import helpers as common_helpers from core.workflow.nodes.variable_assigner.common import helpers as common_helpers
from core.workflow.nodes.variable_assigner.common.exc import VariableOperatorNodeError from core.workflow.nodes.variable_assigner.common.exc import VariableOperatorNodeError
@ -11,7 +11,7 @@ from .node_data import VariableAssignerData, WriteMode
class VariableAssignerNode(BaseNode[VariableAssignerData]): class VariableAssignerNode(BaseNode[VariableAssignerData]):
_node_data_cls: type[BaseNodeData] = VariableAssignerData _node_data_cls = VariableAssignerData
_node_type = NodeType.VARIABLE_ASSIGNER _node_type = NodeType.VARIABLE_ASSIGNER
def _run(self) -> NodeRunResult: def _run(self) -> NodeRunResult: