Complete the attempt span if it was an OOM error

Author: matt-aitken

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

@@ -2968,6 +2968,22 @@ export class RunEngine {
             const nextAttemptNumber =
               latestSnapshot.attemptNumber === null ? 1 : latestSnapshot.attemptNumber + 1;
 
+            if (retryResult.wasOOMError) {
+              this.eventBus.emit("runAttemptFailed", {
+                time: failedAt,
+                run: {
+                  id: runId,
+                  status: run.status,
+                  spanId: run.spanId,
+                  error: completion.error,
+                  attemptNumber: latestSnapshot.attemptNumber ?? 0,
+                  createdAt: run.createdAt,
+                  completedAt: run.completedAt,
+                  taskEventStore: run.taskEventStore,
+                },
+              });
+            }
+
             this.eventBus.emit("runRetryScheduled", {
               time: failedAt,
               run: {

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

@@ -35,6 +35,7 @@ export type RetryOutcome =
       method: "queue" | "immediate";
       settings: TaskRunExecutionRetry;
       machine?: string;
+      wasOOMError?: boolean;
     };
 
 export async function retryOutcomeFromCompletion(
@@ -67,6 +68,7 @@ export async function retryOutcomeFromCompletion(
       method: "queue",
       machine: oomResult.machine,
       settings: { timestamp: Date.now() + delay, delay },
+      wasOOMError: true,
     };
   }