Skip to content

Commit 6ca4c3c

Browse files
committed
fix(run-engine): carryover batchId after PENDING_EXECUTING stalls
1 parent 743b8db commit 6ca4c3c

File tree

2 files changed

+4
-0
lines changed

2 files changed

+4
-0
lines changed

internal-packages/run-engine/src/engine/index.ts

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1454,6 +1454,7 @@ export class RunEngine {
14541454
projectId: latestSnapshot.projectId,
14551455
checkpointId: latestSnapshot.checkpointId ?? undefined,
14561456
completedWaitpoints: latestSnapshot.completedWaitpoints,
1457+
batchId: latestSnapshot.batchId ?? undefined,
14571458
error: {
14581459
type: "INTERNAL_ERROR",
14591460
code: "TASK_RUN_DEQUEUED_MAX_RETRIES",

internal-packages/run-engine/src/engine/systems/runAttemptSystem.ts

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1140,6 +1140,7 @@ export class RunAttemptSystem {
11401140
runnerId,
11411141
checkpointId,
11421142
completedWaitpoints,
1143+
batchId,
11431144
tx,
11441145
}: {
11451146
run: TaskRun;
@@ -1159,6 +1160,7 @@ export class RunAttemptSystem {
11591160
id: string;
11601161
index?: number;
11611162
}[];
1163+
batchId?: string;
11621164
}): Promise<{ wasRequeued: boolean } & ExecutionResult> {
11631165
const prisma = tx ?? this.$.prisma;
11641166

@@ -1207,6 +1209,7 @@ export class RunAttemptSystem {
12071209
runnerId,
12081210
checkpointId,
12091211
completedWaitpoints,
1212+
batchId,
12101213
});
12111214

12121215
return {

0 commit comments

Comments
 (0)