mirror of
https://github.com/langgenius/dify.git
synced 2026-03-31 10:58:48 +08:00
28 lines
734 B
Python
28 lines
734 B
Python
from graphon.graph_engine.layers import GraphEngineLayer
|
|
from graphon.graph_events import GraphEngineEvent, GraphRunPausedEvent
|
|
|
|
|
|
class SuspendLayer(GraphEngineLayer):
|
|
""" """
|
|
|
|
def __init__(self) -> None:
|
|
super().__init__()
|
|
self._paused = False
|
|
|
|
def on_graph_start(self):
|
|
self._paused = False
|
|
|
|
def on_event(self, event: GraphEngineEvent):
|
|
"""
|
|
Handle the paused event, stash runtime state into storage and wait for resume.
|
|
"""
|
|
if isinstance(event, GraphRunPausedEvent):
|
|
self._paused = True
|
|
|
|
def on_graph_end(self, error: Exception | None):
|
|
""" """
|
|
self._paused = False
|
|
|
|
def is_paused(self) -> bool:
|
|
return self._paused
|