deepline 0.3.270 → 0.3.272

package/dist/bundling-sources/shared_libs/play-runtime/scheduler-backends/postgres-progress.ts

@@ -1,3 +1,4 @@
+import { RUN_EVENT_KIND } from '../run-event-catalog';
 /**
  * Live step/progress ledger reporting for Node runtime workers.
  *
@@ -426,11 +427,17 @@ export function createPostgresSchedulerProgressReporter(input: {
   let totalRows = 0;
   let activeMapNodeId: string | null = null;
   let pendingDatasetLifecycleEvents: Array<
-    Extract<PlayExecutionEvent, { type: 'dataset.lifecycle' }>
+    Extract<
+      PlayExecutionEvent,
+      { type: typeof RUN_EVENT_KIND.datasetLifecycle }
+    >
   > = [];
   const pendingActivityObservations = new Map<
     string,
-    Extract<PlayExecutionEvent, { type: 'activity.observed' }>['observation']
+    Extract<
+      PlayExecutionEvent,
+      { type: typeof RUN_EVENT_KIND.activityObserved }
+    >['observation']
   >();
 
   type ProgressFlushRequest = {
@@ -466,7 +473,10 @@ export function createPostgresSchedulerProgressReporter(input: {
   const requeueProgressEvents = (input: {
     datasetLifecycleEvents: typeof pendingDatasetLifecycleEvents;
     activityObservations: Array<
-      Extract<PlayExecutionEvent, { type: 'activity.observed' }>['observation']
+      Extract<
+        PlayExecutionEvent,
+        { type: typeof RUN_EVENT_KIND.activityObserved }
+      >['observation']
     >;
   }): void => {
     pendingDatasetLifecycleEvents = [
@@ -954,7 +964,7 @@ export function createPostgresSchedulerProgressReporter(input: {
         return;
       }
     }
-    if (event.type === 'activity.observed') {
+    if (event.type === RUN_EVENT_KIND.activityObserved) {
       pendingActivityObservations.set(
         event.observation.activityId,
         event.observation,
@@ -962,7 +972,7 @@ export function createPostgresSchedulerProgressReporter(input: {
       scheduleFlush(true);
       return;
     }
-    if (event.type === 'dataset.lifecycle') {
+    if (event.type === RUN_EVENT_KIND.datasetLifecycle) {
       pendingDatasetLifecycleEvents.push(event);
       scheduleFlush(true);
       return;
@@ -1551,7 +1561,7 @@ export function buildFailedTerminalSettledProgressEvents(input: {
     const failed = Math.max(0, Math.floor(outcome.failed) || 0);
     if (completed <= 0 && failed <= 0) continue;
     events.push({
-      type: 'step.progress',
+      type: RUN_EVENT_KIND.stepProgress,
       runId: input.launch.runId,
       source: 'worker',
       occurredAt: input.occurredAt,
@@ -1643,7 +1653,7 @@ export async function collectGatewayRunnerTerminalProgressLedgerEvents(input: {
   });
   if (finalUpdate.workCounters) {
     progressEvents.push({
-      type: 'work.counters',
+      type: RUN_EVENT_KIND.workCounters,
       runId: input.launch.runId,
       source: 'worker',
       occurredAt: finalUpdate.lastCheckpointAt ?? Date.now(),
@@ -1653,7 +1663,7 @@ export async function collectGatewayRunnerTerminalProgressLedgerEvents(input: {
   const datasetEvents: PlayRunLedgerEvent[] = (
     finalUpdate.datasetLifecycleEvents ?? []
   ).map((event) => ({
-    type: 'dataset.lifecycle',
+    type: RUN_EVENT_KIND.datasetLifecycle,
     runId: input.launch.runId,
     source: 'worker',
     occurredAt: event.at,
@@ -1674,7 +1684,7 @@ export async function collectGatewayRunnerTerminalProgressLedgerEvents(input: {
   const activityEvents: PlayRunLedgerEvent[] = (
     finalUpdate.activityObservations ?? []
   ).map((observation: PlayActivityObservation) => ({
-    type: 'activity.observed',
+    type: RUN_EVENT_KIND.activityObserved,
     runId: input.launch.runId,
     source: 'worker',
     occurredAt: observation.observedAt,