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

Signed-off-by: -LAN- <laipz8200@outlook.com>
pull/12042/head
-LAN- 1 year ago committed by GitHub
parent 82134a1d50
commit 094343739b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

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

@ -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)

Loading…
Cancel
Save