From 102d2f4d58b81863cd5fe15194cdd5b3b0376211 Mon Sep 17 00:00:00 2001 From: QuantumGhost Date: Tue, 10 Jun 2025 16:29:46 +0800 Subject: [PATCH] feat(api): Implement `version` method in `parameter_extractor_node` and raise NotImplementedError in `BaseNode` --- api/core/workflow/nodes/base/node.py | 2 +- .../nodes/parameter_extractor/parameter_extractor_node.py | 4 ++++ 2 files changed, 5 insertions(+), 1 deletion(-) diff --git a/api/core/workflow/nodes/base/node.py b/api/core/workflow/nodes/base/node.py index fa11d8893d..6973401429 100644 --- a/api/core/workflow/nodes/base/node.py +++ b/api/core/workflow/nodes/base/node.py @@ -178,7 +178,7 @@ class BaseNode(Generic[GenericNodeData]): # # If you have introduced a new node type, please add it to `NODE_TYPE_CLASSES_MAPPING` # in `api/core/workflow/nodes/__init__.py`. - pass + raise NotImplementedError("subclasses of BaseNode must implement `version` method.") @property def should_continue_on_error(self) -> bool: diff --git a/api/core/workflow/nodes/parameter_extractor/parameter_extractor_node.py b/api/core/workflow/nodes/parameter_extractor/parameter_extractor_node.py index 2552784762..bde319ebe2 100644 --- a/api/core/workflow/nodes/parameter_extractor/parameter_extractor_node.py +++ b/api/core/workflow/nodes/parameter_extractor/parameter_extractor_node.py @@ -109,6 +109,10 @@ class ParameterExtractorNode(BaseNode): } } + @classmethod + def version(cls) -> str: + return "1" + def _run(self): """ Run the node.