From 094343739b42668d40f29c208d7c52f9617f76e8 Mon Sep 17 00:00:00 2001 From: -LAN- Date: Tue, 24 Dec 2024 15:56:59 +0800 Subject: [PATCH] fix/array file cannot use in iteration node (#12035) Signed-off-by: -LAN- --- api/core/variables/variables.py | 2 +- api/tests/unit_tests/core/app/segments/test_variables.py | 7 +++++++ 2 files changed, 8 insertions(+), 1 deletion(-) diff --git a/api/core/variables/variables.py b/api/core/variables/variables.py index f9268b52e6..973e420961 100644 --- a/api/core/variables/variables.py +++ b/api/core/variables/variables.py @@ -90,5 +90,5 @@ class FileVariable(FileSegment, Variable): pass -class ArrayFileVariable(ArrayFileSegment, Variable): +class ArrayFileVariable(ArrayFileSegment, ArrayVariable): pass diff --git a/api/tests/unit_tests/core/app/segments/test_variables.py b/api/tests/unit_tests/core/app/segments/test_variables.py index 0c264c15a0..426557c716 100644 --- a/api/tests/unit_tests/core/app/segments/test_variables.py +++ b/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)