fix: style

pull/20921/head
GuanMu 10 months ago
parent 314013aa33
commit d6240204cf

@ -52,6 +52,7 @@ class WorkflowTaskState(TaskState):
answer: str = ""
outputs: Optional[Mapping[str, Any]] = None
class StreamEvent(Enum):
"""
Stream event

@ -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()

Loading…
Cancel
Save