feat: add agent node stream output

1 year ago · 4765728cef
parent 5063f27916
commit 4765728cef
1 changed files with 1 additions and 1 deletions
--- a/api/core/workflow/nodes/end/end_stream_generate_router.py
+++ b/api/core/workflow/nodes/end/end_stream_generate_router.py
@ -64,7 +64,7 @@ class EndStreamGeneratorRouter:
                node_type = node.get("data", {}).get("type")
                if (
                    variable_selector.value_selector not in value_selectors
-                    and node_type == NodeType.LLM.value
+                    and (node_type in (NodeType.LLM.value, NodeType.AGENT.value))
                    and variable_selector.value_selector[1] == "text"
                ):
                    value_selectors.append(list(variable_selector.value_selector))