From 4d63bd208347defe071e8fce90592534d734176b Mon Sep 17 00:00:00 2001
From: -LAN- <laipz8200@outlook.com>
Date: Thu, 4 Sep 2025 17:47:13 +0800
Subject: [PATCH] refactor(graph_engine): rename SimpleWorkerPool to WorkerPool

---
 api/core/workflow/graph_engine/graph_engine.py                | 4 ++--
 .../graph_engine/orchestration/execution_coordinator.py       | 4 ++--
 api/core/workflow/graph_engine/worker_management/__init__.py  | 4 ++--
 .../{simple_worker_pool.py => worker_pool.py}                 | 2 +-
 4 files changed, 7 insertions(+), 7 deletions(-)
 rename api/core/workflow/graph_engine/worker_management/{simple_worker_pool.py => worker_pool.py} (99%)

diff --git a/api/core/workflow/graph_engine/graph_engine.py b/api/core/workflow/graph_engine/graph_engine.py
index fb3c0aadf6..45f3ada7f5 100644
--- a/api/core/workflow/graph_engine/graph_engine.py
+++ b/api/core/workflow/graph_engine/graph_engine.py
@@ -39,7 +39,7 @@ from .orchestration import Dispatcher, ExecutionCoordinator
 from .protocols.command_channel import CommandChannel
 from .response_coordinator import ResponseStreamCoordinator
 from .state_management import UnifiedStateManager
-from .worker_management import SimpleWorkerPool
+from .worker_management import WorkerPool
 
 logger = logging.getLogger(__name__)
 
@@ -187,7 +187,7 @@ class GraphEngine:
         context_vars = contextvars.copy_context()
 
         # Create worker pool for parallel node execution
-        self._worker_pool = SimpleWorkerPool(
+        self._worker_pool = WorkerPool(
             ready_queue=self._ready_queue,
             event_queue=self._event_queue,
             graph=self._graph,
diff --git a/api/core/workflow/graph_engine/orchestration/execution_coordinator.py b/api/core/workflow/graph_engine/orchestration/execution_coordinator.py
index 3dd443ddb3..234a3607c3 100644
--- a/api/core/workflow/graph_engine/orchestration/execution_coordinator.py
+++ b/api/core/workflow/graph_engine/orchestration/execution_coordinator.py
@@ -8,7 +8,7 @@ from ..command_processing import CommandProcessor
 from ..domain import GraphExecution
 from ..event_management import EventManager
 from ..state_management import UnifiedStateManager
-from ..worker_management import SimpleWorkerPool
+from ..worker_management import WorkerPool
 
 if TYPE_CHECKING:
     from ..event_management import EventHandler
@@ -30,7 +30,7 @@ class ExecutionCoordinator:
         event_handler: "EventHandler",
         event_collector: EventManager,
         command_processor: CommandProcessor,
-        worker_pool: SimpleWorkerPool,
+        worker_pool: WorkerPool,
     ) -> None:
         """
         Initialize the execution coordinator.
diff --git a/api/core/workflow/graph_engine/worker_management/__init__.py b/api/core/workflow/graph_engine/worker_management/__init__.py
index 5b25dbc79a..03de1f6daa 100644
--- a/api/core/workflow/graph_engine/worker_management/__init__.py
+++ b/api/core/workflow/graph_engine/worker_management/__init__.py
@@ -5,8 +5,8 @@ This package manages the worker pool, including creation,
 scaling, and activity tracking.
 """
 
-from .simple_worker_pool import SimpleWorkerPool
+from .worker_pool import WorkerPool
 
 __all__ = [
-    "SimpleWorkerPool",
+    "WorkerPool",
 ]
diff --git a/api/core/workflow/graph_engine/worker_management/simple_worker_pool.py b/api/core/workflow/graph_engine/worker_management/worker_pool.py
similarity index 99%
rename from api/core/workflow/graph_engine/worker_management/simple_worker_pool.py
rename to api/core/workflow/graph_engine/worker_management/worker_pool.py
index 367c2b36fc..40de0ffa08 100644
--- a/api/core/workflow/graph_engine/worker_management/simple_worker_pool.py
+++ b/api/core/workflow/graph_engine/worker_management/worker_pool.py
@@ -26,7 +26,7 @@ if TYPE_CHECKING:
 
 
 @final
-class SimpleWorkerPool:
+class WorkerPool:
     """
     Simple worker pool with integrated management.