@trigger.dev/core 4.5.3 → 4.5.4

package/dist/esm/v3/schemas/messages.js

@@ -1,106 +1,15 @@
 import { z } from "zod";
 import { ImportTaskFileErrors, WorkerManifest } from "./build.js";
-import { MachinePreset, TaskRunExecution, TaskRunExecutionResult, TaskRunFailedExecutionResult, TaskRunInternalError, V3TaskRunExecution, } from "./common.js";
-import { TaskResource } from "./resources.js";
-import { EnvironmentType, V3ProdTaskRunExecution, V3ProdTaskRunExecutionPayload, RunEngineVersionSchema, TaskRunExecutionLazyAttemptPayload, TaskRunExecutionMetrics, WaitReason, } from "./schemas.js";
+import { TaskRunExecution, TaskRunExecutionResult } from "./common.js";
+import { RunEngineVersionSchema, TaskRunExecutionMetrics } from "./schemas.js";
 import { CompletedWaitpoint } from "./runEngine.js";
 import { DebugLogPropertiesInput } from "../runEngineWorker/supervisor/schemas.js";
-export const AckCallbackResult = z.discriminatedUnion("success", [
-    z.object({
-        success: z.literal(false),
-        error: z.object({
-            name: z.string(),
-            message: z.string(),
-            stack: z.string().optional(),
-            stderr: z.string().optional(),
-        }),
-    }),
-    z.object({
-        success: z.literal(true),
-    }),
-]);
-export const BackgroundWorkerServerMessages = z.discriminatedUnion("type", [
-    z.object({
-        type: z.literal("CANCEL_ATTEMPT"),
-        taskAttemptId: z.string(),
-        taskRunId: z.string(),
-    }),
-    z.object({
-        type: z.literal("SCHEDULE_ATTEMPT"),
-        image: z.string(),
-        version: z.string(),
-        machine: MachinePreset,
-        nextAttemptNumber: z.number().optional(),
-        // identifiers
-        id: z.string().optional(), // TODO: Remove this completely in a future release
-        envId: z.string(),
-        envType: EnvironmentType,
-        orgId: z.string(),
-        projectId: z.string(),
-        runId: z.string(),
-        dequeuedAt: z.number().optional(),
-    }),
-    z.object({
-        type: z.literal("EXECUTE_RUN_LAZY_ATTEMPT"),
-        payload: TaskRunExecutionLazyAttemptPayload,
-    }),
-]);
-export const serverWebsocketMessages = {
-    SERVER_READY: z.object({
-        version: z.literal("v1").default("v1"),
-        id: z.string(),
-    }),
-    BACKGROUND_WORKER_MESSAGE: z.object({
-        version: z.literal("v1").default("v1"),
-        backgroundWorkerId: z.string(),
-        data: BackgroundWorkerServerMessages,
-    }),
-};
-export const BackgroundWorkerClientMessages = z.discriminatedUnion("type", [
-    z.object({
-        version: z.literal("v1").default("v1"),
-        type: z.literal("TASK_RUN_COMPLETED"),
-        completion: TaskRunExecutionResult,
-        execution: V3TaskRunExecution,
-    }),
-    z.object({
-        version: z.literal("v1").default("v1"),
-        type: z.literal("TASK_RUN_FAILED_TO_RUN"),
-        completion: TaskRunFailedExecutionResult,
-    }),
-    z.object({
-        version: z.literal("v1").default("v1"),
-        type: z.literal("TASK_HEARTBEAT"),
-        id: z.string(),
-    }),
-    z.object({
-        version: z.literal("v1").default("v1"),
-        type: z.literal("TASK_RUN_HEARTBEAT"),
-        id: z.string(),
-    }),
-]);
 export const ServerBackgroundWorker = z.object({
     id: z.string(),
     version: z.string(),
     contentHash: z.string(),
     engine: RunEngineVersionSchema.optional(),
 });
-export const clientWebsocketMessages = {
-    READY_FOR_TASKS: z.object({
-        version: z.literal("v1").default("v1"),
-        backgroundWorkerId: z.string(),
-        inProgressRuns: z.string().array().optional(),
-    }),
-    BACKGROUND_WORKER_DEPRECATED: z.object({
-        version: z.literal("v1").default("v1"),
-        backgroundWorkerId: z.string(),
-    }),
-    BACKGROUND_WORKER_MESSAGE: z.object({
-        version: z.literal("v1").default("v1"),
-        backgroundWorkerId: z.string(),
-        data: BackgroundWorkerClientMessages,
-    }),
-};
 export const UncaughtExceptionMessage = z.object({
     version: z.literal("v1").default("v1"),
     error: z.object({
@@ -200,656 +109,4 @@ export const WorkerToExecutorMessageCatalog = {
         }),
     },
 };
-export const ProviderToPlatformMessages = {
-    LOG: {
-        message: z.object({
-            version: z.literal("v1").default("v1"),
-            data: z.string(),
-        }),
-    },
-    LOG_WITH_ACK: {
-        message: z.object({
-            version: z.literal("v1").default("v1"),
-            data: z.string(),
-        }),
-        callback: z.object({
-            status: z.literal("ok"),
-        }),
-    },
-    WORKER_CRASHED: {
-        message: z.object({
-            version: z.literal("v1").default("v1"),
-            runId: z.string(),
-            reason: z.string().optional(),
-            exitCode: z.number().optional(),
-            message: z.string().optional(),
-            logs: z.string().optional(),
-            /** This means we should only update the error if one exists */
-            overrideCompletion: z.boolean().optional(),
-            errorCode: TaskRunInternalError.shape.code.optional(),
-        }),
-    },
-    INDEXING_FAILED: {
-        message: z.object({
-            version: z.literal("v1").default("v1"),
-            deploymentId: z.string(),
-            error: z.object({
-                name: z.string(),
-                message: z.string(),
-                stack: z.string().optional(),
-                stderr: z.string().optional(),
-            }),
-            overrideCompletion: z.boolean().optional(),
-        }),
-    },
-};
-export const PlatformToProviderMessages = {
-    INDEX: {
-        message: z.object({
-            version: z.literal("v1").default("v1"),
-            imageTag: z.string(),
-            shortCode: z.string(),
-            apiKey: z.string(),
-            apiUrl: z.string(),
-            // identifiers
-            envId: z.string(),
-            envType: EnvironmentType,
-            orgId: z.string(),
-            projectId: z.string(),
-            deploymentId: z.string(),
-        }),
-        callback: AckCallbackResult,
-    },
-    RESTORE: {
-        message: z.object({
-            version: z.literal("v1").default("v1"),
-            type: z.enum(["DOCKER", "KUBERNETES"]),
-            location: z.string(),
-            reason: z.string().optional(),
-            imageRef: z.string(),
-            attemptNumber: z.number().optional(),
-            machine: MachinePreset,
-            // identifiers
-            checkpointId: z.string(),
-            envId: z.string(),
-            envType: EnvironmentType,
-            orgId: z.string(),
-            projectId: z.string(),
-            runId: z.string(),
-        }),
-    },
-    PRE_PULL_DEPLOYMENT: {
-        message: z.object({
-            version: z.literal("v1").default("v1"),
-            imageRef: z.string(),
-            shortCode: z.string(),
-            // identifiers
-            envId: z.string(),
-            envType: EnvironmentType,
-            orgId: z.string(),
-            projectId: z.string(),
-            deploymentId: z.string(),
-        }),
-    },
-};
-const CreateWorkerMessage = z.object({
-    projectRef: z.string(),
-    envId: z.string(),
-    deploymentId: z.string(),
-    metadata: z.object({
-        cliPackageVersion: z.string().optional(),
-        contentHash: z.string(),
-        packageVersion: z.string(),
-        tasks: TaskResource.array(),
-    }),
-});
-export const CoordinatorToPlatformMessages = {
-    LOG: {
-        message: z.object({
-            version: z.literal("v1").default("v1"),
-            metadata: z.any(),
-            text: z.string(),
-        }),
-    },
-    CREATE_WORKER: {
-        message: z.discriminatedUnion("version", [
-            CreateWorkerMessage.extend({
-                version: z.literal("v1"),
-            }),
-            CreateWorkerMessage.extend({
-                version: z.literal("v2"),
-                supportsLazyAttempts: z.boolean(),
-            }),
-        ]),
-        callback: z.discriminatedUnion("success", [
-            z.object({
-                success: z.literal(false),
-            }),
-            z.object({
-                success: z.literal(true),
-            }),
-        ]),
-    },
-    CREATE_TASK_RUN_ATTEMPT: {
-        message: z.object({
-            version: z.literal("v1").default("v1"),
-            runId: z.string(),
-            envId: z.string(),
-        }),
-        callback: z.discriminatedUnion("success", [
-            z.object({
-                success: z.literal(false),
-                reason: z.string().optional(),
-            }),
-            z.object({
-                success: z.literal(true),
-                executionPayload: V3ProdTaskRunExecutionPayload,
-            }),
-        ]),
-    },
-    // Deprecated: Only workers without lazy attempt support will use this
-    READY_FOR_EXECUTION: {
-        message: z.object({
-            version: z.literal("v1").default("v1"),
-            runId: z.string(),
-            totalCompletions: z.number(),
-        }),
-        callback: z.discriminatedUnion("success", [
-            z.object({
-                success: z.literal(false),
-            }),
-            z.object({
-                success: z.literal(true),
-                payload: V3ProdTaskRunExecutionPayload,
-            }),
-        ]),
-    },
-    READY_FOR_LAZY_ATTEMPT: {
-        message: z.object({
-            version: z.literal("v1").default("v1"),
-            runId: z.string(),
-            envId: z.string(),
-            totalCompletions: z.number(),
-        }),
-        callback: z.discriminatedUnion("success", [
-            z.object({
-                success: z.literal(false),
-                reason: z.string().optional(),
-            }),
-            z.object({
-                success: z.literal(true),
-                lazyPayload: TaskRunExecutionLazyAttemptPayload,
-            }),
-        ]),
-    },
-    READY_FOR_RESUME: {
-        message: z.object({
-            version: z.literal("v1").default("v1"),
-            attemptFriendlyId: z.string(),
-            type: WaitReason,
-        }),
-    },
-    TASK_RUN_COMPLETED: {
-        message: z.object({
-            version: z.enum(["v1", "v2"]).default("v1"),
-            execution: V3ProdTaskRunExecution,
-            completion: TaskRunExecutionResult,
-            checkpoint: z
-                .object({
-                docker: z.boolean(),
-                location: z.string(),
-            })
-                .optional(),
-        }),
-    },
-    TASK_RUN_COMPLETED_WITH_ACK: {
-        message: z.object({
-            version: z.enum(["v1", "v2"]).default("v2"),
-            execution: V3ProdTaskRunExecution,
-            completion: TaskRunExecutionResult,
-            checkpoint: z
-                .object({
-                docker: z.boolean(),
-                location: z.string(),
-            })
-                .optional(),
-        }),
-        callback: AckCallbackResult,
-    },
-    TASK_RUN_FAILED_TO_RUN: {
-        message: z.object({
-            version: z.literal("v1").default("v1"),
-            completion: TaskRunFailedExecutionResult,
-        }),
-    },
-    TASK_HEARTBEAT: {
-        message: z.object({
-            version: z.literal("v1").default("v1"),
-            attemptFriendlyId: z.string(),
-        }),
-    },
-    TASK_RUN_HEARTBEAT: {
-        message: z.object({
-            version: z.literal("v1").default("v1"),
-            runId: z.string(),
-        }),
-    },
-    CHECKPOINT_CREATED: {
-        message: z.object({
-            version: z.literal("v1").default("v1"),
-            runId: z.string().optional(),
-            attemptFriendlyId: z.string(),
-            docker: z.boolean(),
-            location: z.string(),
-            reason: z.discriminatedUnion("type", [
-                z.object({
-                    type: z.literal("WAIT_FOR_DURATION"),
-                    ms: z.number(),
-                    now: z.number(),
-                }),
-                z.object({
-                    type: z.literal("WAIT_FOR_BATCH"),
-                    batchFriendlyId: z.string(),
-                    runFriendlyIds: z.string().array(),
-                }),
-                z.object({
-                    type: z.literal("WAIT_FOR_TASK"),
-                    friendlyId: z.string(),
-                }),
-                z.object({
-                    type: z.literal("RETRYING_AFTER_FAILURE"),
-                    attemptNumber: z.number(),
-                }),
-                z.object({
-                    type: z.literal("MANUAL"),
-                    /** If unspecified it will be restored immediately, e.g. for live migration */
-                    restoreAtUnixTimeMs: z.number().optional(),
-                }),
-            ]),
-        }),
-        callback: z.object({
-            version: z.literal("v1").default("v1"),
-            keepRunAlive: z.boolean(),
-        }),
-    },
-    INDEXING_FAILED: {
-        message: z.object({
-            version: z.literal("v1").default("v1"),
-            deploymentId: z.string(),
-            error: z.object({
-                name: z.string(),
-                message: z.string(),
-                stack: z.string().optional(),
-                stderr: z.string().optional(),
-            }),
-        }),
-    },
-    RUN_CRASHED: {
-        message: z.object({
-            version: z.literal("v1").default("v1"),
-            runId: z.string(),
-            error: z.object({
-                name: z.string(),
-                message: z.string(),
-                stack: z.string().optional(),
-            }),
-        }),
-    },
-};
-export const PlatformToCoordinatorMessages = {
-    /** @deprecated use RESUME_AFTER_DEPENDENCY_WITH_ACK instead  */
-    RESUME_AFTER_DEPENDENCY: {
-        message: z.object({
-            version: z.literal("v1").default("v1"),
-            runId: z.string(),
-            attemptId: z.string(),
-            attemptFriendlyId: z.string(),
-            completions: TaskRunExecutionResult.array(),
-            executions: TaskRunExecution.array(),
-        }),
-    },
-    RESUME_AFTER_DEPENDENCY_WITH_ACK: {
-        message: z.object({
-            version: z.literal("v1").default("v1"),
-            runId: z.string(),
-            attemptId: z.string(),
-            attemptFriendlyId: z.string(),
-            completions: TaskRunExecutionResult.array(),
-            executions: TaskRunExecution.array(),
-        }),
-        callback: AckCallbackResult,
-    },
-    RESUME_AFTER_DURATION: {
-        message: z.object({
-            version: z.literal("v1").default("v1"),
-            attemptId: z.string(),
-            attemptFriendlyId: z.string(),
-        }),
-    },
-    REQUEST_ATTEMPT_CANCELLATION: {
-        message: z.object({
-            version: z.literal("v1").default("v1"),
-            attemptId: z.string(),
-            attemptFriendlyId: z.string(),
-        }),
-    },
-    REQUEST_RUN_CANCELLATION: {
-        message: z.object({
-            version: z.literal("v1").default("v1"),
-            runId: z.string(),
-            delayInMs: z.number().optional(),
-        }),
-    },
-    READY_FOR_RETRY: {
-        message: z.object({
-            version: z.literal("v1").default("v1"),
-            runId: z.string(),
-        }),
-    },
-    DYNAMIC_CONFIG: {
-        message: z.object({
-            version: z.literal("v1").default("v1"),
-            checkpointThresholdInMs: z.number(),
-        }),
-    },
-};
-export const ClientToSharedQueueMessages = {
-    READY_FOR_TASKS: {
-        message: z.object({
-            version: z.literal("v1").default("v1"),
-            backgroundWorkerId: z.string(),
-        }),
-    },
-    BACKGROUND_WORKER_DEPRECATED: {
-        message: z.object({
-            version: z.literal("v1").default("v1"),
-            backgroundWorkerId: z.string(),
-        }),
-    },
-    BACKGROUND_WORKER_MESSAGE: {
-        message: z.object({
-            version: z.literal("v1").default("v1"),
-            backgroundWorkerId: z.string(),
-            data: BackgroundWorkerClientMessages,
-        }),
-    },
-};
-export const SharedQueueToClientMessages = {
-    SERVER_READY: {
-        message: z.object({
-            version: z.literal("v1").default("v1"),
-            id: z.string(),
-        }),
-    },
-    BACKGROUND_WORKER_MESSAGE: {
-        message: z.object({
-            version: z.literal("v1").default("v1"),
-            backgroundWorkerId: z.string(),
-            data: BackgroundWorkerServerMessages,
-        }),
-    },
-};
-const IndexTasksMessage = z.object({
-    version: z.literal("v1"),
-    deploymentId: z.string(),
-    tasks: TaskResource.array(),
-    packageVersion: z.string(),
-});
-export const ProdWorkerToCoordinatorMessages = {
-    TEST: {
-        message: z.object({
-            version: z.literal("v1").default("v1"),
-        }),
-        callback: z.void(),
-    },
-    INDEX_TASKS: {
-        message: z.discriminatedUnion("version", [
-            IndexTasksMessage.extend({
-                version: z.literal("v1"),
-            }),
-            IndexTasksMessage.extend({
-                version: z.literal("v2"),
-                supportsLazyAttempts: z.boolean(),
-            }),
-        ]),
-        callback: z.discriminatedUnion("success", [
-            z.object({
-                success: z.literal(false),
-            }),
-            z.object({
-                success: z.literal(true),
-            }),
-        ]),
-    },
-    // Deprecated: Only workers without lazy attempt support will use this
-    READY_FOR_EXECUTION: {
-        message: z.object({
-            version: z.literal("v1").default("v1"),
-            runId: z.string(),
-            totalCompletions: z.number(),
-        }),
-    },
-    READY_FOR_LAZY_ATTEMPT: {
-        message: z.object({
-            version: z.literal("v1").default("v1"),
-            runId: z.string(),
-            totalCompletions: z.number(),
-            startTime: z.number().optional(),
-        }),
-    },
-    READY_FOR_RESUME: {
-        message: z.discriminatedUnion("version", [
-            z.object({
-                version: z.literal("v1"),
-                attemptFriendlyId: z.string(),
-                type: WaitReason,
-            }),
-            z.object({
-                version: z.literal("v2"),
-                attemptFriendlyId: z.string(),
-                attemptNumber: z.number(),
-                type: WaitReason,
-            }),
-        ]),
-    },
-    READY_FOR_CHECKPOINT: {
-        message: z.object({
-            version: z.literal("v1").default("v1"),
-        }),
-    },
-    CANCEL_CHECKPOINT: {
-        message: z
-            .discriminatedUnion("version", [
-            z.object({
-                version: z.literal("v1"),
-            }),
-            z.object({
-                version: z.literal("v2"),
-                reason: WaitReason.optional(),
-            }),
-        ])
-            .default({ version: "v1" }),
-        callback: z.object({
-            version: z.literal("v2").default("v2"),
-            checkpointCanceled: z.boolean(),
-            reason: WaitReason.optional(),
-        }),
-    },
-    TASK_HEARTBEAT: {
-        message: z.object({
-            version: z.literal("v1").default("v1"),
-            attemptFriendlyId: z.string(),
-        }),
-    },
-    TASK_RUN_HEARTBEAT: {
-        message: z.object({
-            version: z.literal("v1").default("v1"),
-            runId: z.string(),
-        }),
-    },
-    TASK_RUN_COMPLETED: {
-        message: z.object({
-            version: z.enum(["v1", "v2"]).default("v1"),
-            execution: V3ProdTaskRunExecution,
-            completion: TaskRunExecutionResult,
-        }),
-        callback: z.object({
-            willCheckpointAndRestore: z.boolean(),
-            shouldExit: z.boolean(),
-        }),
-    },
-    TASK_RUN_FAILED_TO_RUN: {
-        message: z.object({
-            version: z.literal("v1").default("v1"),
-            completion: TaskRunFailedExecutionResult,
-        }),
-    },
-    WAIT_FOR_DURATION: {
-        message: z.object({
-            version: z.literal("v1").default("v1"),
-            ms: z.number(),
-            now: z.number(),
-            attemptFriendlyId: z.string(),
-        }),
-        callback: z.object({
-            willCheckpointAndRestore: z.boolean(),
-        }),
-    },
-    WAIT_FOR_TASK: {
-        message: z.object({
-            version: z.enum(["v1", "v2"]).default("v1"),
-            friendlyId: z.string(),
-            // This is the attempt that is waiting
-            attemptFriendlyId: z.string(),
-        }),
-        callback: z.object({
-            willCheckpointAndRestore: z.boolean(),
-        }),
-    },
-    WAIT_FOR_BATCH: {
-        message: z.object({
-            version: z.enum(["v1", "v2"]).default("v1"),
-            batchFriendlyId: z.string(),
-            runFriendlyIds: z.string().array(),
-            // This is the attempt that is waiting
-            attemptFriendlyId: z.string(),
-        }),
-        callback: z.object({
-            willCheckpointAndRestore: z.boolean(),
-        }),
-    },
-    INDEXING_FAILED: {
-        message: z.object({
-            version: z.literal("v1").default("v1"),
-            deploymentId: z.string(),
-            error: z.object({
-                name: z.string(),
-                message: z.string(),
-                stack: z.string().optional(),
-                stderr: z.string().optional(),
-            }),
-        }),
-    },
-    CREATE_TASK_RUN_ATTEMPT: {
-        message: z.object({
-            version: z.literal("v1").default("v1"),
-            runId: z.string(),
-        }),
-        callback: z.discriminatedUnion("success", [
-            z.object({
-                success: z.literal(false),
-                reason: z.string().optional(),
-            }),
-            z.object({
-                success: z.literal(true),
-                executionPayload: V3ProdTaskRunExecutionPayload,
-            }),
-        ]),
-    },
-    UNRECOVERABLE_ERROR: {
-        message: z.object({
-            version: z.literal("v1").default("v1"),
-            error: z.object({
-                name: z.string(),
-                message: z.string(),
-                stack: z.string().optional(),
-            }),
-        }),
-    },
-    SET_STATE: {
-        message: z.object({
-            version: z.literal("v1").default("v1"),
-            attemptFriendlyId: z.string().optional(),
-            attemptNumber: z.string().optional(),
-        }),
-    },
-};
-// TODO: The coordinator can only safely use v1 worker messages, higher versions will need a new flag, e.g. SUPPORTS_VERSIONED_MESSAGES
-export const CoordinatorToProdWorkerMessages = {
-    RESUME_AFTER_DEPENDENCY: {
-        message: z.object({
-            version: z.literal("v1").default("v1"),
-            attemptId: z.string(),
-            completions: TaskRunExecutionResult.array(),
-            executions: TaskRunExecution.array(),
-        }),
-    },
-    RESUME_AFTER_DURATION: {
-        message: z.object({
-            version: z.literal("v1").default("v1"),
-            attemptId: z.string(),
-        }),
-    },
-    // Deprecated: Only workers without lazy attempt support will use this
-    EXECUTE_TASK_RUN: {
-        message: z.object({
-            version: z.literal("v1").default("v1"),
-            executionPayload: V3ProdTaskRunExecutionPayload,
-        }),
-    },
-    EXECUTE_TASK_RUN_LAZY_ATTEMPT: {
-        message: z.object({
-            version: z.literal("v1").default("v1"),
-            lazyPayload: TaskRunExecutionLazyAttemptPayload,
-        }),
-    },
-    REQUEST_ATTEMPT_CANCELLATION: {
-        message: z.object({
-            version: z.literal("v1").default("v1"),
-            attemptId: z.string(),
-        }),
-    },
-    REQUEST_EXIT: {
-        message: z.discriminatedUnion("version", [
-            z.object({
-                version: z.literal("v1"),
-            }),
-            z.object({
-                version: z.literal("v2"),
-                delayInMs: z.number().optional(),
-            }),
-        ]),
-    },
-    READY_FOR_RETRY: {
-        message: z.object({
-            version: z.literal("v1").default("v1"),
-            runId: z.string(),
-        }),
-    },
-};
-export const ProdWorkerSocketData = z.object({
-    contentHash: z.string(),
-    projectRef: z.string(),
-    envId: z.string(),
-    runId: z.string(),
-    attemptFriendlyId: z.string().optional(),
-    attemptNumber: z.string().optional(),
-    podName: z.string(),
-    deploymentId: z.string(),
-    deploymentVersion: z.string(),
-    requiresCheckpointResumeWithMessage: z.string().optional(),
-});
-export const CoordinatorSocketData = z.object({
-    supportsDynamicConfig: z.string().optional(),
-});
 //# sourceMappingURL=messages.js.map