Explorar el Código

fix/array file cannot use in iteration node (#12035)

Signed-off-by: -LAN- <laipz8200@outlook.com>
-LAN- hace 4 meses
padre
commit
094343739b

+ 1 - 1
api/core/variables/variables.py

@@ -90,5 +90,5 @@ class FileVariable(FileSegment, Variable):
     pass
 
 
-class ArrayFileVariable(ArrayFileSegment, Variable):
+class ArrayFileVariable(ArrayFileSegment, ArrayVariable):
     pass

+ 7 - 0
api/tests/unit_tests/core/app/segments/test_variables.py

@@ -2,6 +2,8 @@ import pytest
 from pydantic import ValidationError
 
 from core.variables import (
+    ArrayFileVariable,
+    ArrayVariable,
     FloatVariable,
     IntegerVariable,
     ObjectVariable,
@@ -81,3 +83,8 @@ def test_variable_to_object():
     assert var.to_object() == 3.14
     var = SecretVariable(name="secret", value="secret_value")
     assert var.to_object() == "secret_value"
+
+
+def test_array_file_variable_is_array_variable():
+    var = ArrayFileVariable(name="files", value=[])
+    assert isinstance(var, ArrayVariable)