Skip to content

Commit 0943437

Browse files
authored
fix/array file cannot use in iteration node (#12035)
Signed-off-by: -LAN- <laipz8200@outlook.com>
1 parent 82134a1 commit 0943437

File tree

2 files changed

+8
-1
lines changed

2 files changed

+8
-1
lines changed

api/core/variables/variables.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -90,5 +90,5 @@ class FileVariable(FileSegment, Variable):
9090
pass
9191

9292

93-
class ArrayFileVariable(ArrayFileSegment, Variable):
93+
class ArrayFileVariable(ArrayFileSegment, ArrayVariable):
9494
pass

api/tests/unit_tests/core/app/segments/test_variables.py

+7
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,8 @@
22
from pydantic import ValidationError
33

44
from core.variables import (
5+
ArrayFileVariable,
6+
ArrayVariable,
57
FloatVariable,
68
IntegerVariable,
79
ObjectVariable,
@@ -81,3 +83,8 @@ def test_variable_to_object():
8183
assert var.to_object() == 3.14
8284
var = SecretVariable(name="secret", value="secret_value")
8385
assert var.to_object() == "secret_value"
86+
87+
88+
def test_array_file_variable_is_array_variable():
89+
var = ArrayFileVariable(name="files", value=[])
90+
assert isinstance(var, ArrayVariable)

0 commit comments

Comments
 (0)