fix(segments): Support NoneType. (#6581)

2024-07-23 17:59:32 +08:00 · 2024-07-23 17:59:32 +08:00 · 2bc0632d0d
commit 2bc0632d0d
parent 75445a0c66
6 changed files with 45 additions and 4 deletions
--- a/api/core/app/segments/segments.py
+++ b/api/core/app/segments/segments.py
@ -43,6 +43,23 @@ class Segment(BaseModel):
        return self.value


+class NoneSegment(Segment):
+    value_type: SegmentType = SegmentType.NONE
+    value: None = None
+
+    @property
+    def text(self) -> str:
+        return 'null'
+
+    @property
+    def log(self) -> str:
+        return 'null'
+
+    @property
+    def markdown(self) -> str:
+        return 'null'
+
+
 class StringSegment(Segment):
    value_type: SegmentType = SegmentType.STRING
    value: str