From d6240204cf579cd6ce332071cf3ceadc24ca81b1 Mon Sep 17 00:00:00 2001 From: GuanMu Date: Wed, 11 Jun 2025 13:07:49 +0000 Subject: [PATCH] fix: style --- api/core/app/entities/task_entities.py | 1 + api/core/workflow/nodes/answer/answer_stream_processor.py | 3 +-- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/api/core/app/entities/task_entities.py b/api/core/app/entities/task_entities.py index cca8ce7737..40ace038d2 100644 --- a/api/core/app/entities/task_entities.py +++ b/api/core/app/entities/task_entities.py @@ -52,6 +52,7 @@ class WorkflowTaskState(TaskState): answer: str = "" outputs: Optional[Mapping[str, Any]] = None + class StreamEvent(Enum): """ Stream event diff --git a/api/core/workflow/nodes/answer/answer_stream_processor.py b/api/core/workflow/nodes/answer/answer_stream_processor.py index e14e17d318..afd7e6f593 100644 --- a/api/core/workflow/nodes/answer/answer_stream_processor.py +++ b/api/core/workflow/nodes/answer/answer_stream_processor.py @@ -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()