feat: add from_variable_selector for stream chunk / message event (#8228)

2024-09-10 22:15:50 +08:00 · 2024-09-10 22:15:50 +08:00 · cee0c51dbb
commit cee0c51dbb
parent fdbbdb706f
5 changed files with 23 additions and 6 deletions
--- a/api/core/workflow/nodes/answer/answer_stream_processor.py
+++ b/api/core/workflow/nodes/answer/answer_stream_processor.py
@ -108,6 +108,7 @@ class AnswerStreamProcessor(StreamProcessor):
                        route_node_state=event.route_node_state,
                        parallel_id=event.parallel_id,
                        parallel_start_node_id=event.parallel_start_node_id,
+                        from_variable_selector=[answer_node_id, "answer"],
                    )
                else:
                    route_chunk = cast(VarGenerateRouteChunk, route_chunk)