fix(variables): NoneVariable should inherit from NoneSegment. (#6584)

2024-07-23 21:46:08 +08:00
parent 8123a00e97
commit 85a883e281
2 changed files with 7 additions and 14 deletions
--- a/api/core/app/segments/variables.py
+++ b/api/core/app/segments/variables.py
@@ -6,7 +6,7 @@ from pydantic import Field
 from core.file.file_obj import FileVar
 from core.helper import encrypter

-from .segments import Segment, StringSegment
+from .segments import NoneSegment, Segment, StringSegment
 from .types import SegmentType


@@ -83,6 +83,6 @@ class SecretVariable(StringVariable):
        return encrypter.obfuscated_token(self.value)


-class NoneVariable(Variable):
+class NoneVariable(NoneSegment, Variable):
    value_type: SegmentType = SegmentType.NONE
-    value: None = None
+    value: None = None