From 55085a9ca2efbb67f58c1a86e105b40df95a80c5 Mon Sep 17 00:00:00 2001 From: -LAN- Date: Thu, 28 Aug 2025 02:38:56 +0800 Subject: [PATCH] chore(graph_engine): add type hint for event_queue Signed-off-by: -LAN- --- api/core/workflow/graph_engine/graph_engine.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/api/core/workflow/graph_engine/graph_engine.py b/api/core/workflow/graph_engine/graph_engine.py index dcea94b994..6f4f71bfe2 100644 --- a/api/core/workflow/graph_engine/graph_engine.py +++ b/api/core/workflow/graph_engine/graph_engine.py @@ -20,6 +20,7 @@ from core.workflow.enums import NodeExecutionType from core.workflow.graph import Graph from core.workflow.graph_events import ( GraphEngineEvent, + GraphNodeEventBase, GraphRunAbortedEvent, GraphRunFailedEvent, GraphRunStartedEvent, @@ -103,7 +104,7 @@ class GraphEngine: # Initialize queues self.ready_queue: queue.Queue[str] = queue.Queue() - self.event_queue: queue.Queue = queue.Queue() + self.event_queue: queue.Queue[GraphNodeEventBase] = queue.Queue() # Initialize subsystems self._initialize_subsystems()