|
|
|
|
@ -28,10 +28,9 @@ class AnswerStreamProcessor(StreamProcessor):
|
|
|
|
|
self.current_stream_chunk_generating_node_ids: dict[str, list[str]] = {}
|
|
|
|
|
self.has_output = False
|
|
|
|
|
self.output_node_ids: set[str] = set()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def process(self, generator: Generator[GraphEngineEvent, None, None]) -> Generator[GraphEngineEvent, None, None]:
|
|
|
|
|
for event in generator:
|
|
|
|
|
|
|
|
|
|
if isinstance(event, NodeRunStartedEvent):
|
|
|
|
|
if event.route_node_state.node_id == self.graph.root_node_id and not self.rest_node_ids:
|
|
|
|
|
self.reset()
|
|
|
|
|
|