fix: iteration none output error

langgenius · Nov 5, 2024 · 6eebafc · 6eebafc
1 parent 5f21d13
commit 6eebafc
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/api/factories/variable_factory.py b/api/factories/variable_factory.py
@@ -91,6 +91,8 @@ def build_segment(value: Any, /) -> Segment:
                 return ArrayObjectSegment(value=value)
             case SegmentType.FILE:
                 return ArrayFileSegment(value=value)
+            case SegmentType.NONE:
+                return ArrayAnySegment(value=value)
             case _:
                 raise ValueError(f"not supported value {value}")
     raise ValueError(f"not supported value {value}")
diff --git a/api/tests/unit_tests/core/app/segments/test_factory.py b/api/tests/unit_tests/core/app/segments/test_factory.py
@@ -13,6 +13,7 @@
     StringVariable,
 )
 from core.variables.exc import VariableError
+from core.variables.segments import ArrayAnySegment
 from factories import variable_factory
 
 
@@ -156,3 +157,9 @@ def test_variable_cannot_large_than_200_kb():
                 "value": "a" * 1024 * 201,
             }
         )
+
+
+def test_array_none_variable():
+    var = variable_factory.build_segment([None, None, None, None])
+    assert isinstance(var, ArrayAnySegment)
+    assert var.value == [None, None, None, None]