fix(run-engine): carryover batchId after PENDING_EXECUTING stalls (#2563)

Author: nicktrn

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

@@ -1454,6 +1454,7 @@ export class RunEngine {
             projectId: latestSnapshot.projectId,
             checkpointId: latestSnapshot.checkpointId ?? undefined,
             completedWaitpoints: latestSnapshot.completedWaitpoints,
+            batchId: latestSnapshot.batchId ?? undefined,
             error: {
               type: "INTERNAL_ERROR",
               code: "TASK_RUN_DEQUEUED_MAX_RETRIES",

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

@@ -1140,6 +1140,7 @@ export class RunAttemptSystem {
     runnerId,
     checkpointId,
     completedWaitpoints,
+    batchId,
     tx,
   }: {
     run: TaskRun;
@@ -1159,6 +1160,7 @@ export class RunAttemptSystem {
       id: string;
       index?: number;
     }[];
+    batchId?: string;
   }): Promise<{ wasRequeued: boolean } & ExecutionResult> {
     const prisma = tx ?? this.$.prisma;
 
@@ -1207,6 +1209,7 @@ export class RunAttemptSystem {
         runnerId,
         checkpointId,
         completedWaitpoints,
+        batchId,
       });
 
       return {