fix(workflow): refine variable type checks in LLMNode (#10051)

2024-10-30 16:23:12 +08:00
parent 4d38798dd5
commit 3b53e06e0d
2 changed files with 128 additions and 5 deletions
--- a/api/core/workflow/nodes/llm/node.py
+++ b/api/core/workflow/nodes/llm/node.py
@@ -349,13 +349,11 @@ class LLMNode(BaseNode[LLMNodeData]):
        variable = self.graph_runtime_state.variable_pool.get(selector)
        if variable is None:
            return []
-        if isinstance(variable, FileSegment):
+        elif isinstance(variable, FileSegment):
            return [variable.value]
-        if isinstance(variable, ArrayFileSegment):
+        elif isinstance(variable, ArrayFileSegment):
            return variable.value
-        # FIXME: Temporary fix for empty array,
-        # all variables added to variable pool should be a Segment instance.
-        if isinstance(variable, ArrayAnySegment) and len(variable.value) == 0:
+        elif isinstance(variable, NoneSegment | ArrayAnySegment):
            return []
        raise ValueError(f"Invalid variable type: {type(variable)}")