Merge branch 'feat/workflow-backend' into deploy/dev

langgenius · Mar 13, 2024 · 6857422 · 6857422
2 parents 9b81b2b + 6ef3542
commit 6857422
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/api/core/workflow/entities/variable_pool.py b/api/core/workflow/entities/variable_pool.py
@@ -13,7 +13,10 @@ class ValueType(Enum):
     STRING = "string"
     NUMBER = "number"
     OBJECT = "object"
-    ARRAY = "array"
+    ARRAY_STRING = "array[string]"
+    ARRAY_NUMBER = "array[number]"
+    ARRAY_OBJECT = "array[object]"
+    ARRAY_FILE = "array[file]"
     FILE = "file"
 
 
@@ -78,7 +81,10 @@ def get_variable_value(self, variable_selector: list[str],
             elif target_value_type == ValueType.OBJECT:
                 if not isinstance(value, dict):
                     raise ValueError('Invalid value type: object')
-            elif target_value_type == ValueType.ARRAY:
+            elif target_value_type in [ValueType.ARRAY_STRING,
+                                       ValueType.ARRAY_NUMBER,
+                                       ValueType.ARRAY_OBJECT,
+                                       ValueType.ARRAY_FILE]:
                 if not isinstance(value, list):
                     raise ValueError('Invalid value type: array')