fix: Make master hold a queue of task data

2025-07-07 20:22:00 +01:00
parent e1894bc106
commit fe17aaf9f8
1 changed files with 3 additions and 2 deletions
@@ -24,6 +24,7 @@ from shared.types.networking.topology import (
 )
 from shared.types.profiling.common import NodePerformanceProfile
 from shared.types.states.shared import SharedState
+from shared.types.tasks.common import TaskData, TaskType
 from shared.types.worker.instances import InstanceData, InstanceId


@@ -64,8 +65,8 @@ class ControlPlaneNetworkState(State[EventCategories.ControlPlaneEventTypes]):
 class MasterState(SharedState):
    data_plane_network_state: DataPlaneNetworkState
    control_plane_network_state: ControlPlaneNetworkState
-    job_inbox: Queue[ExternalCommand]
-    job_outbox: Queue[ExternalCommand]
+    job_inbox: Queue[TaskData[TaskType]]
+    job_outbox: Queue[TaskData[TaskType]]
    cache_policy: CachePolicy[CachePolicyType]