From 4765728cef53400fcad269bae4ac3e58eadd45a9 Mon Sep 17 00:00:00 2001 From: Novice Lee Date: Wed, 22 Jan 2025 09:59:40 +0800 Subject: [PATCH] feat: add agent node stream output --- api/core/workflow/nodes/end/end_stream_generate_router.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/api/core/workflow/nodes/end/end_stream_generate_router.py b/api/core/workflow/nodes/end/end_stream_generate_router.py index b3678a82b7..3cc176b9b2 100644 --- 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))