jyapayne/dify
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

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

Browse source 
Signed-off-by: -LAN- <laipz8200@outlook.com>
This commit is contained in:
-LAN- 2024-12-24 15:56:59 +08:00 • committed by GitHub
commit 094343739b
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 8 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
api/core/variables/variables.py
View file

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

7
api/tests/unit_tests/core/app/segments/test_variables.py
View file

@ -2,6 +2,8 @@ import pytest
from pydantic import ValidationError from pydantic import ValidationError
from core.variables import ( from core.variables import (
ArrayFileVariable,
ArrayVariable,
FloatVariable, FloatVariable,
IntegerVariable, IntegerVariable,
ObjectVariable, ObjectVariable,
@ -81,3 +83,8 @@ def test_variable_to_object():
assert var.to_object() == 3.14 assert var.to_object() == 3.14
var = SecretVariable(name="secret", value="secret_value") var = SecretVariable(name="secret", value="secret_value")
assert var.to_object() == "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)