@trigger.dev/core 3.0.0-beta.23 → 3.0.0-beta.24

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.
package/dist/v3/index.js CHANGED
@@ -578,7 +578,7 @@ var BatchTaskRunExecutionResult = zod.z.object({
578
578
  items: TaskRunExecutionResult.array()
579
579
  });
580
580
 
581
- // src/v3/schemas/messages.ts
581
+ // src/v3/schemas/schemas.ts
582
582
  var EnvironmentType = zod.z.enum([
583
583
  "PRODUCTION",
584
584
  "STAGING",
@@ -622,92 +622,6 @@ var ProdTaskRunExecutionPayload = zod.z.object({
622
622
  traceContext: zod.z.record(zod.z.unknown()),
623
623
  environment: zod.z.record(zod.z.string()).optional()
624
624
  });
625
- var BackgroundWorkerServerMessages = zod.z.discriminatedUnion("type", [
626
- zod.z.object({
627
- type: zod.z.literal("EXECUTE_RUNS"),
628
- payloads: zod.z.array(TaskRunExecutionPayload)
629
- }),
630
- zod.z.object({
631
- type: zod.z.literal("CANCEL_ATTEMPT"),
632
- taskAttemptId: zod.z.string(),
633
- taskRunId: zod.z.string()
634
- }),
635
- zod.z.object({
636
- type: zod.z.literal("SCHEDULE_ATTEMPT"),
637
- image: zod.z.string(),
638
- version: zod.z.string(),
639
- machine: Machine,
640
- // identifiers
641
- id: zod.z.string(),
642
- envId: zod.z.string(),
643
- envType: EnvironmentType,
644
- orgId: zod.z.string(),
645
- projectId: zod.z.string(),
646
- runId: zod.z.string()
647
- })
648
- ]);
649
- var serverWebsocketMessages = {
650
- SERVER_READY: zod.z.object({
651
- version: zod.z.literal("v1").default("v1"),
652
- id: zod.z.string()
653
- }),
654
- BACKGROUND_WORKER_MESSAGE: zod.z.object({
655
- version: zod.z.literal("v1").default("v1"),
656
- backgroundWorkerId: zod.z.string(),
657
- data: BackgroundWorkerServerMessages
658
- })
659
- };
660
- var BackgroundWorkerClientMessages = zod.z.discriminatedUnion("type", [
661
- zod.z.object({
662
- version: zod.z.literal("v1").default("v1"),
663
- type: zod.z.literal("TASK_RUN_COMPLETED"),
664
- completion: TaskRunExecutionResult,
665
- execution: TaskRunExecution
666
- }),
667
- zod.z.object({
668
- version: zod.z.literal("v1").default("v1"),
669
- type: zod.z.literal("TASK_HEARTBEAT"),
670
- id: zod.z.string()
671
- })
672
- ]);
673
- var BackgroundWorkerProperties = zod.z.object({
674
- id: zod.z.string(),
675
- version: zod.z.string(),
676
- contentHash: zod.z.string()
677
- });
678
- var clientWebsocketMessages = {
679
- READY_FOR_TASKS: zod.z.object({
680
- version: zod.z.literal("v1").default("v1"),
681
- backgroundWorkerId: zod.z.string()
682
- }),
683
- BACKGROUND_WORKER_DEPRECATED: zod.z.object({
684
- version: zod.z.literal("v1").default("v1"),
685
- backgroundWorkerId: zod.z.string()
686
- }),
687
- BACKGROUND_WORKER_MESSAGE: zod.z.object({
688
- version: zod.z.literal("v1").default("v1"),
689
- backgroundWorkerId: zod.z.string(),
690
- data: BackgroundWorkerClientMessages
691
- })
692
- };
693
- var workerToChildMessages = {
694
- EXECUTE_TASK_RUN: zod.z.object({
695
- version: zod.z.literal("v1").default("v1"),
696
- execution: TaskRunExecution,
697
- traceContext: zod.z.record(zod.z.unknown()),
698
- metadata: BackgroundWorkerProperties
699
- }),
700
- TASK_RUN_COMPLETED_NOTIFICATION: zod.z.object({
701
- version: zod.z.literal("v1").default("v1"),
702
- completion: TaskRunExecutionResult,
703
- execution: TaskRunExecution
704
- }),
705
- CLEANUP: zod.z.object({
706
- version: zod.z.literal("v1").default("v1"),
707
- flush: zod.z.boolean().default(false),
708
- kill: zod.z.boolean().default(true)
709
- })
710
- };
711
625
  var FixedWindowRateLimit = zod.z.object({
712
626
  type: zod.z.literal("fixed-window"),
713
627
  limit: zod.z.number(),
@@ -813,149 +727,45 @@ var TaskFileMetadata = zod.z.object({
813
727
  exportName: zod.z.string()
814
728
  });
815
729
  var TaskMetadataWithFilePath = TaskMetadata.merge(TaskFileMetadata);
816
- var UncaughtExceptionMessage = zod.z.object({
817
- version: zod.z.literal("v1").default("v1"),
818
- error: zod.z.object({
819
- name: zod.z.string(),
820
- message: zod.z.string(),
821
- stack: zod.z.string().optional()
822
- }),
823
- origin: zod.z.enum([
824
- "uncaughtException",
825
- "unhandledRejection"
826
- ])
730
+ var PostStartCauses = zod.z.enum([
731
+ "index",
732
+ "create",
733
+ "restore"
734
+ ]);
735
+ var PreStopCauses = zod.z.enum([
736
+ "terminate"
737
+ ]);
738
+ var RegexSchema = zod.z.custom((val) => {
739
+ try {
740
+ return typeof val.test === "function";
741
+ } catch {
742
+ return false;
743
+ }
827
744
  });
828
- var TaskMetadataFailedToParseData = zod.z.object({
829
- version: zod.z.literal("v1").default("v1"),
830
- tasks: zod.z.unknown(),
831
- zodIssues: zod.z.custom((v) => {
832
- return Array.isArray(v) && v.every((issue) => typeof issue === "object" && "message" in issue);
833
- })
745
+ var Config = zod.z.object({
746
+ project: zod.z.string(),
747
+ triggerDirectories: zod.z.string().array().optional(),
748
+ triggerUrl: zod.z.string().optional(),
749
+ projectDir: zod.z.string().optional(),
750
+ tsconfigPath: zod.z.string().optional(),
751
+ retries: zod.z.object({
752
+ enabledInDev: zod.z.boolean().default(true),
753
+ default: RetryOptions.optional()
754
+ }).optional(),
755
+ additionalPackages: zod.z.string().array().optional(),
756
+ additionalFiles: zod.z.string().array().optional(),
757
+ dependenciesToBundle: zod.z.array(zod.z.union([
758
+ zod.z.string(),
759
+ RegexSchema
760
+ ])).optional(),
761
+ logLevel: zod.z.string().optional(),
762
+ enableConsoleLogging: zod.z.boolean().optional()
834
763
  });
835
- var childToWorkerMessages = {
836
- TASK_RUN_COMPLETED: zod.z.object({
837
- version: zod.z.literal("v1").default("v1"),
838
- execution: TaskRunExecution,
839
- result: TaskRunExecutionResult
840
- }),
841
- TASKS_READY: zod.z.object({
842
- version: zod.z.literal("v1").default("v1"),
843
- tasks: TaskMetadataWithFilePath.array()
844
- }),
845
- TASKS_FAILED_TO_PARSE: TaskMetadataFailedToParseData,
846
- TASK_HEARTBEAT: zod.z.object({
847
- version: zod.z.literal("v1").default("v1"),
848
- id: zod.z.string()
849
- }),
850
- READY_TO_DISPOSE: zod.z.undefined(),
851
- WAIT_FOR_DURATION: zod.z.object({
852
- version: zod.z.literal("v1").default("v1"),
853
- ms: zod.z.number()
854
- }),
855
- WAIT_FOR_TASK: zod.z.object({
856
- version: zod.z.literal("v1").default("v1"),
857
- id: zod.z.string()
858
- }),
859
- WAIT_FOR_BATCH: zod.z.object({
860
- version: zod.z.literal("v1").default("v1"),
861
- id: zod.z.string(),
862
- runs: zod.z.string().array()
863
- }),
864
- UNCAUGHT_EXCEPTION: UncaughtExceptionMessage
865
- };
866
- var ProdChildToWorkerMessages = {
867
- TASK_RUN_COMPLETED: {
868
- message: zod.z.object({
869
- version: zod.z.literal("v1").default("v1"),
870
- execution: TaskRunExecution,
871
- result: TaskRunExecutionResult
872
- })
873
- },
874
- TASKS_READY: {
875
- message: zod.z.object({
876
- version: zod.z.literal("v1").default("v1"),
877
- tasks: TaskMetadataWithFilePath.array()
878
- })
879
- },
880
- TASKS_FAILED_TO_PARSE: {
881
- message: TaskMetadataFailedToParseData
882
- },
883
- TASK_HEARTBEAT: {
884
- message: zod.z.object({
885
- version: zod.z.literal("v1").default("v1"),
886
- id: zod.z.string()
887
- })
888
- },
889
- READY_TO_DISPOSE: {
890
- message: zod.z.undefined()
891
- },
892
- READY_FOR_CHECKPOINT: {
893
- message: zod.z.object({
894
- version: zod.z.literal("v1").default("v1")
895
- })
896
- },
897
- CANCEL_CHECKPOINT: {
898
- message: zod.z.object({
899
- version: zod.z.literal("v1").default("v1")
900
- })
901
- },
902
- WAIT_FOR_DURATION: {
903
- message: zod.z.object({
904
- version: zod.z.literal("v1").default("v1"),
905
- ms: zod.z.number(),
906
- now: zod.z.number()
907
- }),
908
- callback: zod.z.object({
909
- willCheckpointAndRestore: zod.z.boolean()
910
- })
911
- },
912
- WAIT_FOR_TASK: {
913
- message: zod.z.object({
914
- version: zod.z.literal("v1").default("v1"),
915
- friendlyId: zod.z.string()
916
- })
917
- },
918
- WAIT_FOR_BATCH: {
919
- message: zod.z.object({
920
- version: zod.z.literal("v1").default("v1"),
921
- batchFriendlyId: zod.z.string(),
922
- runFriendlyIds: zod.z.string().array()
923
- })
924
- },
925
- UNCAUGHT_EXCEPTION: {
926
- message: UncaughtExceptionMessage
927
- }
928
- };
929
- var ProdWorkerToChildMessages = {
930
- EXECUTE_TASK_RUN: {
931
- message: zod.z.object({
932
- version: zod.z.literal("v1").default("v1"),
933
- execution: TaskRunExecution,
934
- traceContext: zod.z.record(zod.z.unknown()),
935
- metadata: BackgroundWorkerProperties
936
- })
937
- },
938
- TASK_RUN_COMPLETED_NOTIFICATION: {
939
- message: zod.z.object({
940
- version: zod.z.literal("v1").default("v1"),
941
- completion: TaskRunExecutionResult,
942
- execution: TaskRunExecution
943
- })
944
- },
945
- CLEANUP: {
946
- message: zod.z.object({
947
- version: zod.z.literal("v1").default("v1"),
948
- flush: zod.z.boolean().default(false),
949
- kill: zod.z.boolean().default(true)
950
- }),
951
- callback: zod.z.void()
952
- },
953
- WAIT_COMPLETED_NOTIFICATION: {
954
- message: zod.z.object({
955
- version: zod.z.literal("v1").default("v1")
956
- })
957
- }
958
- };
764
+ var WaitReason = zod.z.enum([
765
+ "WAIT_FOR_DURATION",
766
+ "WAIT_FOR_TASK",
767
+ "WAIT_FOR_BATCH"
768
+ ]);
959
769
 
960
770
  // src/v3/schemas/resources.ts
961
771
  var TaskResource = zod.z.object({
@@ -1178,67 +988,271 @@ var ScheduleObject = zod.z.object({
1178
988
  expression: zod.z.string(),
1179
989
  description: zod.z.string()
1180
990
  }),
1181
- nextRun: zod.z.coerce.date().nullish(),
1182
- environments: zod.z.array(zod.z.object({
1183
- id: zod.z.string(),
1184
- type: zod.z.string(),
1185
- userName: zod.z.string().nullish()
1186
- }))
1187
- });
1188
- var DeletedScheduleObject = zod.z.object({
1189
- id: zod.z.string()
991
+ nextRun: zod.z.coerce.date().nullish(),
992
+ environments: zod.z.array(zod.z.object({
993
+ id: zod.z.string(),
994
+ type: zod.z.string(),
995
+ userName: zod.z.string().nullish()
996
+ }))
997
+ });
998
+ var DeletedScheduleObject = zod.z.object({
999
+ id: zod.z.string()
1000
+ });
1001
+ var ListSchedulesResult = zod.z.object({
1002
+ data: zod.z.array(ScheduleObject),
1003
+ pagination: zod.z.object({
1004
+ currentPage: zod.z.number(),
1005
+ totalPages: zod.z.number(),
1006
+ count: zod.z.number()
1007
+ })
1008
+ });
1009
+ var ListScheduleOptions = zod.z.object({
1010
+ page: zod.z.number().optional(),
1011
+ perPage: zod.z.number().optional()
1012
+ });
1013
+ var BackgroundWorkerServerMessages = zod.z.discriminatedUnion("type", [
1014
+ zod.z.object({
1015
+ type: zod.z.literal("EXECUTE_RUNS"),
1016
+ payloads: zod.z.array(TaskRunExecutionPayload)
1017
+ }),
1018
+ zod.z.object({
1019
+ type: zod.z.literal("CANCEL_ATTEMPT"),
1020
+ taskAttemptId: zod.z.string(),
1021
+ taskRunId: zod.z.string()
1022
+ }),
1023
+ zod.z.object({
1024
+ type: zod.z.literal("SCHEDULE_ATTEMPT"),
1025
+ image: zod.z.string(),
1026
+ version: zod.z.string(),
1027
+ machine: Machine,
1028
+ // identifiers
1029
+ id: zod.z.string(),
1030
+ envId: zod.z.string(),
1031
+ envType: EnvironmentType,
1032
+ orgId: zod.z.string(),
1033
+ projectId: zod.z.string(),
1034
+ runId: zod.z.string()
1035
+ })
1036
+ ]);
1037
+ var serverWebsocketMessages = {
1038
+ SERVER_READY: zod.z.object({
1039
+ version: zod.z.literal("v1").default("v1"),
1040
+ id: zod.z.string()
1041
+ }),
1042
+ BACKGROUND_WORKER_MESSAGE: zod.z.object({
1043
+ version: zod.z.literal("v1").default("v1"),
1044
+ backgroundWorkerId: zod.z.string(),
1045
+ data: BackgroundWorkerServerMessages
1046
+ })
1047
+ };
1048
+ var BackgroundWorkerClientMessages = zod.z.discriminatedUnion("type", [
1049
+ zod.z.object({
1050
+ version: zod.z.literal("v1").default("v1"),
1051
+ type: zod.z.literal("TASK_RUN_COMPLETED"),
1052
+ completion: TaskRunExecutionResult,
1053
+ execution: TaskRunExecution
1054
+ }),
1055
+ zod.z.object({
1056
+ version: zod.z.literal("v1").default("v1"),
1057
+ type: zod.z.literal("TASK_HEARTBEAT"),
1058
+ id: zod.z.string()
1059
+ })
1060
+ ]);
1061
+ var BackgroundWorkerProperties = zod.z.object({
1062
+ id: zod.z.string(),
1063
+ version: zod.z.string(),
1064
+ contentHash: zod.z.string()
1190
1065
  });
1191
- var ListSchedulesResult = zod.z.object({
1192
- data: zod.z.array(ScheduleObject),
1193
- pagination: zod.z.object({
1194
- currentPage: zod.z.number(),
1195
- totalPages: zod.z.number(),
1196
- count: zod.z.number()
1066
+ var clientWebsocketMessages = {
1067
+ READY_FOR_TASKS: zod.z.object({
1068
+ version: zod.z.literal("v1").default("v1"),
1069
+ backgroundWorkerId: zod.z.string()
1070
+ }),
1071
+ BACKGROUND_WORKER_DEPRECATED: zod.z.object({
1072
+ version: zod.z.literal("v1").default("v1"),
1073
+ backgroundWorkerId: zod.z.string()
1074
+ }),
1075
+ BACKGROUND_WORKER_MESSAGE: zod.z.object({
1076
+ version: zod.z.literal("v1").default("v1"),
1077
+ backgroundWorkerId: zod.z.string(),
1078
+ data: BackgroundWorkerClientMessages
1079
+ })
1080
+ };
1081
+ var workerToChildMessages = {
1082
+ EXECUTE_TASK_RUN: zod.z.object({
1083
+ version: zod.z.literal("v1").default("v1"),
1084
+ execution: TaskRunExecution,
1085
+ traceContext: zod.z.record(zod.z.unknown()),
1086
+ metadata: BackgroundWorkerProperties
1087
+ }),
1088
+ TASK_RUN_COMPLETED_NOTIFICATION: zod.z.object({
1089
+ version: zod.z.literal("v1").default("v1"),
1090
+ completion: TaskRunExecutionResult,
1091
+ execution: TaskRunExecution
1092
+ }),
1093
+ CLEANUP: zod.z.object({
1094
+ version: zod.z.literal("v1").default("v1"),
1095
+ flush: zod.z.boolean().default(false),
1096
+ kill: zod.z.boolean().default(true)
1197
1097
  })
1098
+ };
1099
+ var UncaughtExceptionMessage = zod.z.object({
1100
+ version: zod.z.literal("v1").default("v1"),
1101
+ error: zod.z.object({
1102
+ name: zod.z.string(),
1103
+ message: zod.z.string(),
1104
+ stack: zod.z.string().optional()
1105
+ }),
1106
+ origin: zod.z.enum([
1107
+ "uncaughtException",
1108
+ "unhandledRejection"
1109
+ ])
1198
1110
  });
1199
- var ListScheduleOptions = zod.z.object({
1200
- page: zod.z.number().optional(),
1201
- perPage: zod.z.number().optional()
1111
+ var TaskMetadataFailedToParseData = zod.z.object({
1112
+ version: zod.z.literal("v1").default("v1"),
1113
+ tasks: zod.z.unknown(),
1114
+ zodIssues: zod.z.custom((v) => {
1115
+ return Array.isArray(v) && v.every((issue) => typeof issue === "object" && "message" in issue);
1116
+ })
1202
1117
  });
1203
- var PostStartCauses = zod.z.enum([
1204
- "index",
1205
- "create",
1206
- "restore"
1207
- ]);
1208
- var PreStopCauses = zod.z.enum([
1209
- "terminate"
1210
- ]);
1211
- var RegexSchema = zod.z.custom((val) => {
1212
- try {
1213
- return typeof val.test === "function";
1214
- } catch {
1215
- return false;
1118
+ var childToWorkerMessages = {
1119
+ TASK_RUN_COMPLETED: zod.z.object({
1120
+ version: zod.z.literal("v1").default("v1"),
1121
+ execution: TaskRunExecution,
1122
+ result: TaskRunExecutionResult
1123
+ }),
1124
+ TASKS_READY: zod.z.object({
1125
+ version: zod.z.literal("v1").default("v1"),
1126
+ tasks: TaskMetadataWithFilePath.array()
1127
+ }),
1128
+ TASKS_FAILED_TO_PARSE: TaskMetadataFailedToParseData,
1129
+ TASK_HEARTBEAT: zod.z.object({
1130
+ version: zod.z.literal("v1").default("v1"),
1131
+ id: zod.z.string()
1132
+ }),
1133
+ READY_TO_DISPOSE: zod.z.undefined(),
1134
+ WAIT_FOR_DURATION: zod.z.object({
1135
+ version: zod.z.literal("v1").default("v1"),
1136
+ ms: zod.z.number()
1137
+ }),
1138
+ WAIT_FOR_TASK: zod.z.object({
1139
+ version: zod.z.literal("v1").default("v1"),
1140
+ id: zod.z.string()
1141
+ }),
1142
+ WAIT_FOR_BATCH: zod.z.object({
1143
+ version: zod.z.literal("v1").default("v1"),
1144
+ id: zod.z.string(),
1145
+ runs: zod.z.string().array()
1146
+ }),
1147
+ UNCAUGHT_EXCEPTION: UncaughtExceptionMessage
1148
+ };
1149
+ var ProdChildToWorkerMessages = {
1150
+ TASK_RUN_COMPLETED: {
1151
+ message: zod.z.object({
1152
+ version: zod.z.literal("v1").default("v1"),
1153
+ execution: TaskRunExecution,
1154
+ result: TaskRunExecutionResult
1155
+ })
1156
+ },
1157
+ TASKS_READY: {
1158
+ message: zod.z.object({
1159
+ version: zod.z.literal("v1").default("v1"),
1160
+ tasks: TaskMetadataWithFilePath.array()
1161
+ })
1162
+ },
1163
+ TASKS_FAILED_TO_PARSE: {
1164
+ message: TaskMetadataFailedToParseData
1165
+ },
1166
+ TASK_HEARTBEAT: {
1167
+ message: zod.z.object({
1168
+ version: zod.z.literal("v1").default("v1"),
1169
+ id: zod.z.string()
1170
+ })
1171
+ },
1172
+ READY_TO_DISPOSE: {
1173
+ message: zod.z.undefined()
1174
+ },
1175
+ READY_FOR_CHECKPOINT: {
1176
+ message: zod.z.object({
1177
+ version: zod.z.literal("v1").default("v1")
1178
+ })
1179
+ },
1180
+ CANCEL_CHECKPOINT: {
1181
+ message: zod.z.discriminatedUnion("version", [
1182
+ zod.z.object({
1183
+ version: zod.z.literal("v1")
1184
+ }),
1185
+ zod.z.object({
1186
+ version: zod.z.literal("v2"),
1187
+ reason: WaitReason.optional()
1188
+ })
1189
+ ]).default({
1190
+ version: "v1"
1191
+ }),
1192
+ callback: zod.z.object({
1193
+ // TODO: Figure out how best to handle callback schema parsing in zod IPC
1194
+ version: zod.z.literal("v2"),
1195
+ checkpointCanceled: zod.z.boolean(),
1196
+ reason: WaitReason.optional()
1197
+ })
1198
+ },
1199
+ WAIT_FOR_DURATION: {
1200
+ message: zod.z.object({
1201
+ version: zod.z.literal("v1").default("v1"),
1202
+ ms: zod.z.number(),
1203
+ now: zod.z.number()
1204
+ }),
1205
+ callback: zod.z.object({
1206
+ willCheckpointAndRestore: zod.z.boolean()
1207
+ })
1208
+ },
1209
+ WAIT_FOR_TASK: {
1210
+ message: zod.z.object({
1211
+ version: zod.z.literal("v1").default("v1"),
1212
+ friendlyId: zod.z.string()
1213
+ })
1214
+ },
1215
+ WAIT_FOR_BATCH: {
1216
+ message: zod.z.object({
1217
+ version: zod.z.literal("v1").default("v1"),
1218
+ batchFriendlyId: zod.z.string(),
1219
+ runFriendlyIds: zod.z.string().array()
1220
+ })
1221
+ },
1222
+ UNCAUGHT_EXCEPTION: {
1223
+ message: UncaughtExceptionMessage
1216
1224
  }
1217
- });
1218
- var Config = zod.z.object({
1219
- project: zod.z.string(),
1220
- triggerDirectories: zod.z.string().array().optional(),
1221
- triggerUrl: zod.z.string().optional(),
1222
- projectDir: zod.z.string().optional(),
1223
- tsconfigPath: zod.z.string().optional(),
1224
- retries: zod.z.object({
1225
- enabledInDev: zod.z.boolean().default(true),
1226
- default: RetryOptions.optional()
1227
- }).optional(),
1228
- additionalPackages: zod.z.string().array().optional(),
1229
- additionalFiles: zod.z.string().array().optional(),
1230
- dependenciesToBundle: zod.z.array(zod.z.union([
1231
- zod.z.string(),
1232
- RegexSchema
1233
- ])).optional(),
1234
- logLevel: zod.z.string().optional(),
1235
- enableConsoleLogging: zod.z.boolean().optional()
1236
- });
1237
- var WaitReason = zod.z.enum([
1238
- "WAIT_FOR_DURATION",
1239
- "WAIT_FOR_TASK",
1240
- "WAIT_FOR_BATCH"
1241
- ]);
1225
+ };
1226
+ var ProdWorkerToChildMessages = {
1227
+ EXECUTE_TASK_RUN: {
1228
+ message: zod.z.object({
1229
+ version: zod.z.literal("v1").default("v1"),
1230
+ execution: TaskRunExecution,
1231
+ traceContext: zod.z.record(zod.z.unknown()),
1232
+ metadata: BackgroundWorkerProperties
1233
+ })
1234
+ },
1235
+ TASK_RUN_COMPLETED_NOTIFICATION: {
1236
+ message: zod.z.object({
1237
+ version: zod.z.literal("v1").default("v1"),
1238
+ completion: TaskRunExecutionResult,
1239
+ execution: TaskRunExecution
1240
+ })
1241
+ },
1242
+ CLEANUP: {
1243
+ message: zod.z.object({
1244
+ version: zod.z.literal("v1").default("v1"),
1245
+ flush: zod.z.boolean().default(false),
1246
+ kill: zod.z.boolean().default(true)
1247
+ }),
1248
+ callback: zod.z.void()
1249
+ },
1250
+ WAIT_COMPLETED_NOTIFICATION: {
1251
+ message: zod.z.object({
1252
+ version: zod.z.literal("v1").default("v1")
1253
+ })
1254
+ }
1255
+ };
1242
1256
  var ProviderToPlatformMessages = {
1243
1257
  LOG: {
1244
1258
  message: zod.z.object({
@@ -1573,8 +1587,21 @@ var ProdWorkerToCoordinatorMessages = {
1573
1587
  })
1574
1588
  },
1575
1589
  CANCEL_CHECKPOINT: {
1576
- message: zod.z.object({
1577
- version: zod.z.literal("v1").default("v1")
1590
+ message: zod.z.discriminatedUnion("version", [
1591
+ zod.z.object({
1592
+ version: zod.z.literal("v1")
1593
+ }),
1594
+ zod.z.object({
1595
+ version: zod.z.literal("v2"),
1596
+ reason: WaitReason.optional()
1597
+ })
1598
+ ]).default({
1599
+ version: "v1"
1600
+ }),
1601
+ callback: zod.z.object({
1602
+ version: zod.z.literal("v2").default("v2"),
1603
+ checkpointCanceled: zod.z.boolean(),
1604
+ reason: WaitReason.optional()
1578
1605
  })
1579
1606
  },
1580
1607
  TASK_HEARTBEAT: {