hz/ebiz-dify-ai
1
0
Fork 0
mirror of http://112.124.100.131/huang.ze/ebiz-dify-ai.git synced 2025-12-15 22:06:52 +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
4 changed files with 6 additions and 6 deletions
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]):
_node_data_cls: type[BaseNodeData]
_node_data_cls: type[GenericNodeData]
_node_type: NodeType
def __init__(
@@ -57,7 +57,7 @@ class BaseNode(Generic[GenericNodeData]):
self.node_id = node_id
node_data = self._node_data_cls.model_validate(config.get("data", {}))
self.node_data = cast(GenericNodeData, node_data)
self.node_data = node_data
@abstractmethod
def _run(self) -> NodeRunResult | Generator[Union[NodeEvent, "InNodeEvent"], None, None]: