@trigger.dev/core 4.5.3 → 4.5.4

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (37) hide show
  1. package/dist/commonjs/v3/build/extensions.d.ts +2 -0
  2. package/dist/commonjs/v3/schemas/build.d.ts +36 -0
  3. package/dist/commonjs/v3/schemas/build.js +2 -0
  4. package/dist/commonjs/v3/schemas/build.js.map +1 -1
  5. package/dist/commonjs/v3/schemas/checkpoints.d.ts +4 -4
  6. package/dist/commonjs/v3/schemas/messages.d.ts +2110 -35077
  7. package/dist/commonjs/v3/schemas/messages.js +1 -744
  8. package/dist/commonjs/v3/schemas/messages.js.map +1 -1
  9. package/dist/commonjs/v3/serverOnly/index.d.ts +0 -1
  10. package/dist/commonjs/v3/serverOnly/index.js +0 -1
  11. package/dist/commonjs/v3/serverOnly/index.js.map +1 -1
  12. package/dist/commonjs/version.js +1 -1
  13. package/dist/esm/v3/build/extensions.d.ts +2 -0
  14. package/dist/esm/v3/schemas/build.d.ts +36 -0
  15. package/dist/esm/v3/schemas/build.js +2 -0
  16. package/dist/esm/v3/schemas/build.js.map +1 -1
  17. package/dist/esm/v3/schemas/checkpoints.d.ts +4 -4
  18. package/dist/esm/v3/schemas/messages.d.ts +2110 -35077
  19. package/dist/esm/v3/schemas/messages.js +2 -745
  20. package/dist/esm/v3/schemas/messages.js.map +1 -1
  21. package/dist/esm/v3/serverOnly/index.d.ts +0 -1
  22. package/dist/esm/v3/serverOnly/index.js +0 -1
  23. package/dist/esm/v3/serverOnly/index.js.map +1 -1
  24. package/dist/esm/version.js +1 -1
  25. package/package.json +1 -16
  26. package/dist/commonjs/v3/serverOnly/resourceMonitor.d.ts +0 -190
  27. package/dist/commonjs/v3/serverOnly/resourceMonitor.js +0 -596
  28. package/dist/commonjs/v3/serverOnly/resourceMonitor.js.map +0 -1
  29. package/dist/commonjs/v3/zodNamespace.d.ts +0 -36
  30. package/dist/commonjs/v3/zodNamespace.js +0 -101
  31. package/dist/commonjs/v3/zodNamespace.js.map +0 -1
  32. package/dist/esm/v3/serverOnly/resourceMonitor.d.ts +0 -190
  33. package/dist/esm/v3/serverOnly/resourceMonitor.js +0 -589
  34. package/dist/esm/v3/serverOnly/resourceMonitor.js.map +0 -1
  35. package/dist/esm/v3/zodNamespace.d.ts +0 -36
  36. package/dist/esm/v3/zodNamespace.js +0 -97
  37. package/dist/esm/v3/zodNamespace.js.map +0 -1
@@ -1,109 +1,18 @@
1
1
  "use strict";
2
2
  Object.defineProperty(exports, "__esModule", { value: true });
3
- exports.CoordinatorSocketData = exports.ProdWorkerSocketData = exports.CoordinatorToProdWorkerMessages = exports.ProdWorkerToCoordinatorMessages = exports.SharedQueueToClientMessages = exports.ClientToSharedQueueMessages = exports.PlatformToCoordinatorMessages = exports.CoordinatorToPlatformMessages = exports.PlatformToProviderMessages = exports.ProviderToPlatformMessages = exports.WorkerToExecutorMessageCatalog = exports.ExecutorToWorkerMessageCatalog = exports.indexerToWorkerMessages = exports.TaskMetadataFailedToParseData = exports.UncaughtExceptionMessage = exports.clientWebsocketMessages = exports.ServerBackgroundWorker = exports.BackgroundWorkerClientMessages = exports.serverWebsocketMessages = exports.BackgroundWorkerServerMessages = exports.AckCallbackResult = void 0;
3
+ exports.WorkerToExecutorMessageCatalog = exports.ExecutorToWorkerMessageCatalog = exports.indexerToWorkerMessages = exports.TaskMetadataFailedToParseData = exports.UncaughtExceptionMessage = exports.ServerBackgroundWorker = void 0;
4
4
  const zod_1 = require("zod");
5
5
  const build_js_1 = require("./build.js");
6
6
  const common_js_1 = require("./common.js");
7
- const resources_js_1 = require("./resources.js");
8
7
  const schemas_js_1 = require("./schemas.js");
9
8
  const runEngine_js_1 = require("./runEngine.js");
10
9
  const schemas_js_2 = require("../runEngineWorker/supervisor/schemas.js");
11
- exports.AckCallbackResult = zod_1.z.discriminatedUnion("success", [
12
- zod_1.z.object({
13
- success: zod_1.z.literal(false),
14
- error: zod_1.z.object({
15
- name: zod_1.z.string(),
16
- message: zod_1.z.string(),
17
- stack: zod_1.z.string().optional(),
18
- stderr: zod_1.z.string().optional(),
19
- }),
20
- }),
21
- zod_1.z.object({
22
- success: zod_1.z.literal(true),
23
- }),
24
- ]);
25
- exports.BackgroundWorkerServerMessages = zod_1.z.discriminatedUnion("type", [
26
- zod_1.z.object({
27
- type: zod_1.z.literal("CANCEL_ATTEMPT"),
28
- taskAttemptId: zod_1.z.string(),
29
- taskRunId: zod_1.z.string(),
30
- }),
31
- zod_1.z.object({
32
- type: zod_1.z.literal("SCHEDULE_ATTEMPT"),
33
- image: zod_1.z.string(),
34
- version: zod_1.z.string(),
35
- machine: common_js_1.MachinePreset,
36
- nextAttemptNumber: zod_1.z.number().optional(),
37
- // identifiers
38
- id: zod_1.z.string().optional(), // TODO: Remove this completely in a future release
39
- envId: zod_1.z.string(),
40
- envType: schemas_js_1.EnvironmentType,
41
- orgId: zod_1.z.string(),
42
- projectId: zod_1.z.string(),
43
- runId: zod_1.z.string(),
44
- dequeuedAt: zod_1.z.number().optional(),
45
- }),
46
- zod_1.z.object({
47
- type: zod_1.z.literal("EXECUTE_RUN_LAZY_ATTEMPT"),
48
- payload: schemas_js_1.TaskRunExecutionLazyAttemptPayload,
49
- }),
50
- ]);
51
- exports.serverWebsocketMessages = {
52
- SERVER_READY: zod_1.z.object({
53
- version: zod_1.z.literal("v1").default("v1"),
54
- id: zod_1.z.string(),
55
- }),
56
- BACKGROUND_WORKER_MESSAGE: zod_1.z.object({
57
- version: zod_1.z.literal("v1").default("v1"),
58
- backgroundWorkerId: zod_1.z.string(),
59
- data: exports.BackgroundWorkerServerMessages,
60
- }),
61
- };
62
- exports.BackgroundWorkerClientMessages = zod_1.z.discriminatedUnion("type", [
63
- zod_1.z.object({
64
- version: zod_1.z.literal("v1").default("v1"),
65
- type: zod_1.z.literal("TASK_RUN_COMPLETED"),
66
- completion: common_js_1.TaskRunExecutionResult,
67
- execution: common_js_1.V3TaskRunExecution,
68
- }),
69
- zod_1.z.object({
70
- version: zod_1.z.literal("v1").default("v1"),
71
- type: zod_1.z.literal("TASK_RUN_FAILED_TO_RUN"),
72
- completion: common_js_1.TaskRunFailedExecutionResult,
73
- }),
74
- zod_1.z.object({
75
- version: zod_1.z.literal("v1").default("v1"),
76
- type: zod_1.z.literal("TASK_HEARTBEAT"),
77
- id: zod_1.z.string(),
78
- }),
79
- zod_1.z.object({
80
- version: zod_1.z.literal("v1").default("v1"),
81
- type: zod_1.z.literal("TASK_RUN_HEARTBEAT"),
82
- id: zod_1.z.string(),
83
- }),
84
- ]);
85
10
  exports.ServerBackgroundWorker = zod_1.z.object({
86
11
  id: zod_1.z.string(),
87
12
  version: zod_1.z.string(),
88
13
  contentHash: zod_1.z.string(),
89
14
  engine: schemas_js_1.RunEngineVersionSchema.optional(),
90
15
  });
91
- exports.clientWebsocketMessages = {
92
- READY_FOR_TASKS: zod_1.z.object({
93
- version: zod_1.z.literal("v1").default("v1"),
94
- backgroundWorkerId: zod_1.z.string(),
95
- inProgressRuns: zod_1.z.string().array().optional(),
96
- }),
97
- BACKGROUND_WORKER_DEPRECATED: zod_1.z.object({
98
- version: zod_1.z.literal("v1").default("v1"),
99
- backgroundWorkerId: zod_1.z.string(),
100
- }),
101
- BACKGROUND_WORKER_MESSAGE: zod_1.z.object({
102
- version: zod_1.z.literal("v1").default("v1"),
103
- backgroundWorkerId: zod_1.z.string(),
104
- data: exports.BackgroundWorkerClientMessages,
105
- }),
106
- };
107
16
  exports.UncaughtExceptionMessage = zod_1.z.object({
108
17
  version: zod_1.z.literal("v1").default("v1"),
109
18
  error: zod_1.z.object({
@@ -203,656 +112,4 @@ exports.WorkerToExecutorMessageCatalog = {
203
112
  }),
204
113
  },
205
114
  };
206
- exports.ProviderToPlatformMessages = {
207
- LOG: {
208
- message: zod_1.z.object({
209
- version: zod_1.z.literal("v1").default("v1"),
210
- data: zod_1.z.string(),
211
- }),
212
- },
213
- LOG_WITH_ACK: {
214
- message: zod_1.z.object({
215
- version: zod_1.z.literal("v1").default("v1"),
216
- data: zod_1.z.string(),
217
- }),
218
- callback: zod_1.z.object({
219
- status: zod_1.z.literal("ok"),
220
- }),
221
- },
222
- WORKER_CRASHED: {
223
- message: zod_1.z.object({
224
- version: zod_1.z.literal("v1").default("v1"),
225
- runId: zod_1.z.string(),
226
- reason: zod_1.z.string().optional(),
227
- exitCode: zod_1.z.number().optional(),
228
- message: zod_1.z.string().optional(),
229
- logs: zod_1.z.string().optional(),
230
- /** This means we should only update the error if one exists */
231
- overrideCompletion: zod_1.z.boolean().optional(),
232
- errorCode: common_js_1.TaskRunInternalError.shape.code.optional(),
233
- }),
234
- },
235
- INDEXING_FAILED: {
236
- message: zod_1.z.object({
237
- version: zod_1.z.literal("v1").default("v1"),
238
- deploymentId: zod_1.z.string(),
239
- error: zod_1.z.object({
240
- name: zod_1.z.string(),
241
- message: zod_1.z.string(),
242
- stack: zod_1.z.string().optional(),
243
- stderr: zod_1.z.string().optional(),
244
- }),
245
- overrideCompletion: zod_1.z.boolean().optional(),
246
- }),
247
- },
248
- };
249
- exports.PlatformToProviderMessages = {
250
- INDEX: {
251
- message: zod_1.z.object({
252
- version: zod_1.z.literal("v1").default("v1"),
253
- imageTag: zod_1.z.string(),
254
- shortCode: zod_1.z.string(),
255
- apiKey: zod_1.z.string(),
256
- apiUrl: zod_1.z.string(),
257
- // identifiers
258
- envId: zod_1.z.string(),
259
- envType: schemas_js_1.EnvironmentType,
260
- orgId: zod_1.z.string(),
261
- projectId: zod_1.z.string(),
262
- deploymentId: zod_1.z.string(),
263
- }),
264
- callback: exports.AckCallbackResult,
265
- },
266
- RESTORE: {
267
- message: zod_1.z.object({
268
- version: zod_1.z.literal("v1").default("v1"),
269
- type: zod_1.z.enum(["DOCKER", "KUBERNETES"]),
270
- location: zod_1.z.string(),
271
- reason: zod_1.z.string().optional(),
272
- imageRef: zod_1.z.string(),
273
- attemptNumber: zod_1.z.number().optional(),
274
- machine: common_js_1.MachinePreset,
275
- // identifiers
276
- checkpointId: zod_1.z.string(),
277
- envId: zod_1.z.string(),
278
- envType: schemas_js_1.EnvironmentType,
279
- orgId: zod_1.z.string(),
280
- projectId: zod_1.z.string(),
281
- runId: zod_1.z.string(),
282
- }),
283
- },
284
- PRE_PULL_DEPLOYMENT: {
285
- message: zod_1.z.object({
286
- version: zod_1.z.literal("v1").default("v1"),
287
- imageRef: zod_1.z.string(),
288
- shortCode: zod_1.z.string(),
289
- // identifiers
290
- envId: zod_1.z.string(),
291
- envType: schemas_js_1.EnvironmentType,
292
- orgId: zod_1.z.string(),
293
- projectId: zod_1.z.string(),
294
- deploymentId: zod_1.z.string(),
295
- }),
296
- },
297
- };
298
- const CreateWorkerMessage = zod_1.z.object({
299
- projectRef: zod_1.z.string(),
300
- envId: zod_1.z.string(),
301
- deploymentId: zod_1.z.string(),
302
- metadata: zod_1.z.object({
303
- cliPackageVersion: zod_1.z.string().optional(),
304
- contentHash: zod_1.z.string(),
305
- packageVersion: zod_1.z.string(),
306
- tasks: resources_js_1.TaskResource.array(),
307
- }),
308
- });
309
- exports.CoordinatorToPlatformMessages = {
310
- LOG: {
311
- message: zod_1.z.object({
312
- version: zod_1.z.literal("v1").default("v1"),
313
- metadata: zod_1.z.any(),
314
- text: zod_1.z.string(),
315
- }),
316
- },
317
- CREATE_WORKER: {
318
- message: zod_1.z.discriminatedUnion("version", [
319
- CreateWorkerMessage.extend({
320
- version: zod_1.z.literal("v1"),
321
- }),
322
- CreateWorkerMessage.extend({
323
- version: zod_1.z.literal("v2"),
324
- supportsLazyAttempts: zod_1.z.boolean(),
325
- }),
326
- ]),
327
- callback: zod_1.z.discriminatedUnion("success", [
328
- zod_1.z.object({
329
- success: zod_1.z.literal(false),
330
- }),
331
- zod_1.z.object({
332
- success: zod_1.z.literal(true),
333
- }),
334
- ]),
335
- },
336
- CREATE_TASK_RUN_ATTEMPT: {
337
- message: zod_1.z.object({
338
- version: zod_1.z.literal("v1").default("v1"),
339
- runId: zod_1.z.string(),
340
- envId: zod_1.z.string(),
341
- }),
342
- callback: zod_1.z.discriminatedUnion("success", [
343
- zod_1.z.object({
344
- success: zod_1.z.literal(false),
345
- reason: zod_1.z.string().optional(),
346
- }),
347
- zod_1.z.object({
348
- success: zod_1.z.literal(true),
349
- executionPayload: schemas_js_1.V3ProdTaskRunExecutionPayload,
350
- }),
351
- ]),
352
- },
353
- // Deprecated: Only workers without lazy attempt support will use this
354
- READY_FOR_EXECUTION: {
355
- message: zod_1.z.object({
356
- version: zod_1.z.literal("v1").default("v1"),
357
- runId: zod_1.z.string(),
358
- totalCompletions: zod_1.z.number(),
359
- }),
360
- callback: zod_1.z.discriminatedUnion("success", [
361
- zod_1.z.object({
362
- success: zod_1.z.literal(false),
363
- }),
364
- zod_1.z.object({
365
- success: zod_1.z.literal(true),
366
- payload: schemas_js_1.V3ProdTaskRunExecutionPayload,
367
- }),
368
- ]),
369
- },
370
- READY_FOR_LAZY_ATTEMPT: {
371
- message: zod_1.z.object({
372
- version: zod_1.z.literal("v1").default("v1"),
373
- runId: zod_1.z.string(),
374
- envId: zod_1.z.string(),
375
- totalCompletions: zod_1.z.number(),
376
- }),
377
- callback: zod_1.z.discriminatedUnion("success", [
378
- zod_1.z.object({
379
- success: zod_1.z.literal(false),
380
- reason: zod_1.z.string().optional(),
381
- }),
382
- zod_1.z.object({
383
- success: zod_1.z.literal(true),
384
- lazyPayload: schemas_js_1.TaskRunExecutionLazyAttemptPayload,
385
- }),
386
- ]),
387
- },
388
- READY_FOR_RESUME: {
389
- message: zod_1.z.object({
390
- version: zod_1.z.literal("v1").default("v1"),
391
- attemptFriendlyId: zod_1.z.string(),
392
- type: schemas_js_1.WaitReason,
393
- }),
394
- },
395
- TASK_RUN_COMPLETED: {
396
- message: zod_1.z.object({
397
- version: zod_1.z.enum(["v1", "v2"]).default("v1"),
398
- execution: schemas_js_1.V3ProdTaskRunExecution,
399
- completion: common_js_1.TaskRunExecutionResult,
400
- checkpoint: zod_1.z
401
- .object({
402
- docker: zod_1.z.boolean(),
403
- location: zod_1.z.string(),
404
- })
405
- .optional(),
406
- }),
407
- },
408
- TASK_RUN_COMPLETED_WITH_ACK: {
409
- message: zod_1.z.object({
410
- version: zod_1.z.enum(["v1", "v2"]).default("v2"),
411
- execution: schemas_js_1.V3ProdTaskRunExecution,
412
- completion: common_js_1.TaskRunExecutionResult,
413
- checkpoint: zod_1.z
414
- .object({
415
- docker: zod_1.z.boolean(),
416
- location: zod_1.z.string(),
417
- })
418
- .optional(),
419
- }),
420
- callback: exports.AckCallbackResult,
421
- },
422
- TASK_RUN_FAILED_TO_RUN: {
423
- message: zod_1.z.object({
424
- version: zod_1.z.literal("v1").default("v1"),
425
- completion: common_js_1.TaskRunFailedExecutionResult,
426
- }),
427
- },
428
- TASK_HEARTBEAT: {
429
- message: zod_1.z.object({
430
- version: zod_1.z.literal("v1").default("v1"),
431
- attemptFriendlyId: zod_1.z.string(),
432
- }),
433
- },
434
- TASK_RUN_HEARTBEAT: {
435
- message: zod_1.z.object({
436
- version: zod_1.z.literal("v1").default("v1"),
437
- runId: zod_1.z.string(),
438
- }),
439
- },
440
- CHECKPOINT_CREATED: {
441
- message: zod_1.z.object({
442
- version: zod_1.z.literal("v1").default("v1"),
443
- runId: zod_1.z.string().optional(),
444
- attemptFriendlyId: zod_1.z.string(),
445
- docker: zod_1.z.boolean(),
446
- location: zod_1.z.string(),
447
- reason: zod_1.z.discriminatedUnion("type", [
448
- zod_1.z.object({
449
- type: zod_1.z.literal("WAIT_FOR_DURATION"),
450
- ms: zod_1.z.number(),
451
- now: zod_1.z.number(),
452
- }),
453
- zod_1.z.object({
454
- type: zod_1.z.literal("WAIT_FOR_BATCH"),
455
- batchFriendlyId: zod_1.z.string(),
456
- runFriendlyIds: zod_1.z.string().array(),
457
- }),
458
- zod_1.z.object({
459
- type: zod_1.z.literal("WAIT_FOR_TASK"),
460
- friendlyId: zod_1.z.string(),
461
- }),
462
- zod_1.z.object({
463
- type: zod_1.z.literal("RETRYING_AFTER_FAILURE"),
464
- attemptNumber: zod_1.z.number(),
465
- }),
466
- zod_1.z.object({
467
- type: zod_1.z.literal("MANUAL"),
468
- /** If unspecified it will be restored immediately, e.g. for live migration */
469
- restoreAtUnixTimeMs: zod_1.z.number().optional(),
470
- }),
471
- ]),
472
- }),
473
- callback: zod_1.z.object({
474
- version: zod_1.z.literal("v1").default("v1"),
475
- keepRunAlive: zod_1.z.boolean(),
476
- }),
477
- },
478
- INDEXING_FAILED: {
479
- message: zod_1.z.object({
480
- version: zod_1.z.literal("v1").default("v1"),
481
- deploymentId: zod_1.z.string(),
482
- error: zod_1.z.object({
483
- name: zod_1.z.string(),
484
- message: zod_1.z.string(),
485
- stack: zod_1.z.string().optional(),
486
- stderr: zod_1.z.string().optional(),
487
- }),
488
- }),
489
- },
490
- RUN_CRASHED: {
491
- message: zod_1.z.object({
492
- version: zod_1.z.literal("v1").default("v1"),
493
- runId: zod_1.z.string(),
494
- error: zod_1.z.object({
495
- name: zod_1.z.string(),
496
- message: zod_1.z.string(),
497
- stack: zod_1.z.string().optional(),
498
- }),
499
- }),
500
- },
501
- };
502
- exports.PlatformToCoordinatorMessages = {
503
- /** @deprecated use RESUME_AFTER_DEPENDENCY_WITH_ACK instead */
504
- RESUME_AFTER_DEPENDENCY: {
505
- message: zod_1.z.object({
506
- version: zod_1.z.literal("v1").default("v1"),
507
- runId: zod_1.z.string(),
508
- attemptId: zod_1.z.string(),
509
- attemptFriendlyId: zod_1.z.string(),
510
- completions: common_js_1.TaskRunExecutionResult.array(),
511
- executions: common_js_1.TaskRunExecution.array(),
512
- }),
513
- },
514
- RESUME_AFTER_DEPENDENCY_WITH_ACK: {
515
- message: zod_1.z.object({
516
- version: zod_1.z.literal("v1").default("v1"),
517
- runId: zod_1.z.string(),
518
- attemptId: zod_1.z.string(),
519
- attemptFriendlyId: zod_1.z.string(),
520
- completions: common_js_1.TaskRunExecutionResult.array(),
521
- executions: common_js_1.TaskRunExecution.array(),
522
- }),
523
- callback: exports.AckCallbackResult,
524
- },
525
- RESUME_AFTER_DURATION: {
526
- message: zod_1.z.object({
527
- version: zod_1.z.literal("v1").default("v1"),
528
- attemptId: zod_1.z.string(),
529
- attemptFriendlyId: zod_1.z.string(),
530
- }),
531
- },
532
- REQUEST_ATTEMPT_CANCELLATION: {
533
- message: zod_1.z.object({
534
- version: zod_1.z.literal("v1").default("v1"),
535
- attemptId: zod_1.z.string(),
536
- attemptFriendlyId: zod_1.z.string(),
537
- }),
538
- },
539
- REQUEST_RUN_CANCELLATION: {
540
- message: zod_1.z.object({
541
- version: zod_1.z.literal("v1").default("v1"),
542
- runId: zod_1.z.string(),
543
- delayInMs: zod_1.z.number().optional(),
544
- }),
545
- },
546
- READY_FOR_RETRY: {
547
- message: zod_1.z.object({
548
- version: zod_1.z.literal("v1").default("v1"),
549
- runId: zod_1.z.string(),
550
- }),
551
- },
552
- DYNAMIC_CONFIG: {
553
- message: zod_1.z.object({
554
- version: zod_1.z.literal("v1").default("v1"),
555
- checkpointThresholdInMs: zod_1.z.number(),
556
- }),
557
- },
558
- };
559
- exports.ClientToSharedQueueMessages = {
560
- READY_FOR_TASKS: {
561
- message: zod_1.z.object({
562
- version: zod_1.z.literal("v1").default("v1"),
563
- backgroundWorkerId: zod_1.z.string(),
564
- }),
565
- },
566
- BACKGROUND_WORKER_DEPRECATED: {
567
- message: zod_1.z.object({
568
- version: zod_1.z.literal("v1").default("v1"),
569
- backgroundWorkerId: zod_1.z.string(),
570
- }),
571
- },
572
- BACKGROUND_WORKER_MESSAGE: {
573
- message: zod_1.z.object({
574
- version: zod_1.z.literal("v1").default("v1"),
575
- backgroundWorkerId: zod_1.z.string(),
576
- data: exports.BackgroundWorkerClientMessages,
577
- }),
578
- },
579
- };
580
- exports.SharedQueueToClientMessages = {
581
- SERVER_READY: {
582
- message: zod_1.z.object({
583
- version: zod_1.z.literal("v1").default("v1"),
584
- id: zod_1.z.string(),
585
- }),
586
- },
587
- BACKGROUND_WORKER_MESSAGE: {
588
- message: zod_1.z.object({
589
- version: zod_1.z.literal("v1").default("v1"),
590
- backgroundWorkerId: zod_1.z.string(),
591
- data: exports.BackgroundWorkerServerMessages,
592
- }),
593
- },
594
- };
595
- const IndexTasksMessage = zod_1.z.object({
596
- version: zod_1.z.literal("v1"),
597
- deploymentId: zod_1.z.string(),
598
- tasks: resources_js_1.TaskResource.array(),
599
- packageVersion: zod_1.z.string(),
600
- });
601
- exports.ProdWorkerToCoordinatorMessages = {
602
- TEST: {
603
- message: zod_1.z.object({
604
- version: zod_1.z.literal("v1").default("v1"),
605
- }),
606
- callback: zod_1.z.void(),
607
- },
608
- INDEX_TASKS: {
609
- message: zod_1.z.discriminatedUnion("version", [
610
- IndexTasksMessage.extend({
611
- version: zod_1.z.literal("v1"),
612
- }),
613
- IndexTasksMessage.extend({
614
- version: zod_1.z.literal("v2"),
615
- supportsLazyAttempts: zod_1.z.boolean(),
616
- }),
617
- ]),
618
- callback: zod_1.z.discriminatedUnion("success", [
619
- zod_1.z.object({
620
- success: zod_1.z.literal(false),
621
- }),
622
- zod_1.z.object({
623
- success: zod_1.z.literal(true),
624
- }),
625
- ]),
626
- },
627
- // Deprecated: Only workers without lazy attempt support will use this
628
- READY_FOR_EXECUTION: {
629
- message: zod_1.z.object({
630
- version: zod_1.z.literal("v1").default("v1"),
631
- runId: zod_1.z.string(),
632
- totalCompletions: zod_1.z.number(),
633
- }),
634
- },
635
- READY_FOR_LAZY_ATTEMPT: {
636
- message: zod_1.z.object({
637
- version: zod_1.z.literal("v1").default("v1"),
638
- runId: zod_1.z.string(),
639
- totalCompletions: zod_1.z.number(),
640
- startTime: zod_1.z.number().optional(),
641
- }),
642
- },
643
- READY_FOR_RESUME: {
644
- message: zod_1.z.discriminatedUnion("version", [
645
- zod_1.z.object({
646
- version: zod_1.z.literal("v1"),
647
- attemptFriendlyId: zod_1.z.string(),
648
- type: schemas_js_1.WaitReason,
649
- }),
650
- zod_1.z.object({
651
- version: zod_1.z.literal("v2"),
652
- attemptFriendlyId: zod_1.z.string(),
653
- attemptNumber: zod_1.z.number(),
654
- type: schemas_js_1.WaitReason,
655
- }),
656
- ]),
657
- },
658
- READY_FOR_CHECKPOINT: {
659
- message: zod_1.z.object({
660
- version: zod_1.z.literal("v1").default("v1"),
661
- }),
662
- },
663
- CANCEL_CHECKPOINT: {
664
- message: zod_1.z
665
- .discriminatedUnion("version", [
666
- zod_1.z.object({
667
- version: zod_1.z.literal("v1"),
668
- }),
669
- zod_1.z.object({
670
- version: zod_1.z.literal("v2"),
671
- reason: schemas_js_1.WaitReason.optional(),
672
- }),
673
- ])
674
- .default({ version: "v1" }),
675
- callback: zod_1.z.object({
676
- version: zod_1.z.literal("v2").default("v2"),
677
- checkpointCanceled: zod_1.z.boolean(),
678
- reason: schemas_js_1.WaitReason.optional(),
679
- }),
680
- },
681
- TASK_HEARTBEAT: {
682
- message: zod_1.z.object({
683
- version: zod_1.z.literal("v1").default("v1"),
684
- attemptFriendlyId: zod_1.z.string(),
685
- }),
686
- },
687
- TASK_RUN_HEARTBEAT: {
688
- message: zod_1.z.object({
689
- version: zod_1.z.literal("v1").default("v1"),
690
- runId: zod_1.z.string(),
691
- }),
692
- },
693
- TASK_RUN_COMPLETED: {
694
- message: zod_1.z.object({
695
- version: zod_1.z.enum(["v1", "v2"]).default("v1"),
696
- execution: schemas_js_1.V3ProdTaskRunExecution,
697
- completion: common_js_1.TaskRunExecutionResult,
698
- }),
699
- callback: zod_1.z.object({
700
- willCheckpointAndRestore: zod_1.z.boolean(),
701
- shouldExit: zod_1.z.boolean(),
702
- }),
703
- },
704
- TASK_RUN_FAILED_TO_RUN: {
705
- message: zod_1.z.object({
706
- version: zod_1.z.literal("v1").default("v1"),
707
- completion: common_js_1.TaskRunFailedExecutionResult,
708
- }),
709
- },
710
- WAIT_FOR_DURATION: {
711
- message: zod_1.z.object({
712
- version: zod_1.z.literal("v1").default("v1"),
713
- ms: zod_1.z.number(),
714
- now: zod_1.z.number(),
715
- attemptFriendlyId: zod_1.z.string(),
716
- }),
717
- callback: zod_1.z.object({
718
- willCheckpointAndRestore: zod_1.z.boolean(),
719
- }),
720
- },
721
- WAIT_FOR_TASK: {
722
- message: zod_1.z.object({
723
- version: zod_1.z.enum(["v1", "v2"]).default("v1"),
724
- friendlyId: zod_1.z.string(),
725
- // This is the attempt that is waiting
726
- attemptFriendlyId: zod_1.z.string(),
727
- }),
728
- callback: zod_1.z.object({
729
- willCheckpointAndRestore: zod_1.z.boolean(),
730
- }),
731
- },
732
- WAIT_FOR_BATCH: {
733
- message: zod_1.z.object({
734
- version: zod_1.z.enum(["v1", "v2"]).default("v1"),
735
- batchFriendlyId: zod_1.z.string(),
736
- runFriendlyIds: zod_1.z.string().array(),
737
- // This is the attempt that is waiting
738
- attemptFriendlyId: zod_1.z.string(),
739
- }),
740
- callback: zod_1.z.object({
741
- willCheckpointAndRestore: zod_1.z.boolean(),
742
- }),
743
- },
744
- INDEXING_FAILED: {
745
- message: zod_1.z.object({
746
- version: zod_1.z.literal("v1").default("v1"),
747
- deploymentId: zod_1.z.string(),
748
- error: zod_1.z.object({
749
- name: zod_1.z.string(),
750
- message: zod_1.z.string(),
751
- stack: zod_1.z.string().optional(),
752
- stderr: zod_1.z.string().optional(),
753
- }),
754
- }),
755
- },
756
- CREATE_TASK_RUN_ATTEMPT: {
757
- message: zod_1.z.object({
758
- version: zod_1.z.literal("v1").default("v1"),
759
- runId: zod_1.z.string(),
760
- }),
761
- callback: zod_1.z.discriminatedUnion("success", [
762
- zod_1.z.object({
763
- success: zod_1.z.literal(false),
764
- reason: zod_1.z.string().optional(),
765
- }),
766
- zod_1.z.object({
767
- success: zod_1.z.literal(true),
768
- executionPayload: schemas_js_1.V3ProdTaskRunExecutionPayload,
769
- }),
770
- ]),
771
- },
772
- UNRECOVERABLE_ERROR: {
773
- message: zod_1.z.object({
774
- version: zod_1.z.literal("v1").default("v1"),
775
- error: zod_1.z.object({
776
- name: zod_1.z.string(),
777
- message: zod_1.z.string(),
778
- stack: zod_1.z.string().optional(),
779
- }),
780
- }),
781
- },
782
- SET_STATE: {
783
- message: zod_1.z.object({
784
- version: zod_1.z.literal("v1").default("v1"),
785
- attemptFriendlyId: zod_1.z.string().optional(),
786
- attemptNumber: zod_1.z.string().optional(),
787
- }),
788
- },
789
- };
790
- // TODO: The coordinator can only safely use v1 worker messages, higher versions will need a new flag, e.g. SUPPORTS_VERSIONED_MESSAGES
791
- exports.CoordinatorToProdWorkerMessages = {
792
- RESUME_AFTER_DEPENDENCY: {
793
- message: zod_1.z.object({
794
- version: zod_1.z.literal("v1").default("v1"),
795
- attemptId: zod_1.z.string(),
796
- completions: common_js_1.TaskRunExecutionResult.array(),
797
- executions: common_js_1.TaskRunExecution.array(),
798
- }),
799
- },
800
- RESUME_AFTER_DURATION: {
801
- message: zod_1.z.object({
802
- version: zod_1.z.literal("v1").default("v1"),
803
- attemptId: zod_1.z.string(),
804
- }),
805
- },
806
- // Deprecated: Only workers without lazy attempt support will use this
807
- EXECUTE_TASK_RUN: {
808
- message: zod_1.z.object({
809
- version: zod_1.z.literal("v1").default("v1"),
810
- executionPayload: schemas_js_1.V3ProdTaskRunExecutionPayload,
811
- }),
812
- },
813
- EXECUTE_TASK_RUN_LAZY_ATTEMPT: {
814
- message: zod_1.z.object({
815
- version: zod_1.z.literal("v1").default("v1"),
816
- lazyPayload: schemas_js_1.TaskRunExecutionLazyAttemptPayload,
817
- }),
818
- },
819
- REQUEST_ATTEMPT_CANCELLATION: {
820
- message: zod_1.z.object({
821
- version: zod_1.z.literal("v1").default("v1"),
822
- attemptId: zod_1.z.string(),
823
- }),
824
- },
825
- REQUEST_EXIT: {
826
- message: zod_1.z.discriminatedUnion("version", [
827
- zod_1.z.object({
828
- version: zod_1.z.literal("v1"),
829
- }),
830
- zod_1.z.object({
831
- version: zod_1.z.literal("v2"),
832
- delayInMs: zod_1.z.number().optional(),
833
- }),
834
- ]),
835
- },
836
- READY_FOR_RETRY: {
837
- message: zod_1.z.object({
838
- version: zod_1.z.literal("v1").default("v1"),
839
- runId: zod_1.z.string(),
840
- }),
841
- },
842
- };
843
- exports.ProdWorkerSocketData = zod_1.z.object({
844
- contentHash: zod_1.z.string(),
845
- projectRef: zod_1.z.string(),
846
- envId: zod_1.z.string(),
847
- runId: zod_1.z.string(),
848
- attemptFriendlyId: zod_1.z.string().optional(),
849
- attemptNumber: zod_1.z.string().optional(),
850
- podName: zod_1.z.string(),
851
- deploymentId: zod_1.z.string(),
852
- deploymentVersion: zod_1.z.string(),
853
- requiresCheckpointResumeWithMessage: zod_1.z.string().optional(),
854
- });
855
- exports.CoordinatorSocketData = zod_1.z.object({
856
- supportsDynamicConfig: zod_1.z.string().optional(),
857
- });
858
115
  //# sourceMappingURL=messages.js.map