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.