prevent uncancellable checkpoints

2024-10-18 00:34:40 +01:00
parent a9928be68e
commit 2d84b7cdc4
1 changed files with 4 additions and 1 deletions
@@ -436,7 +436,10 @@ export class Checkpointer {
        this.#logger.error("Error during cleanup", { ...metadata, error });
      }

-      this.#abortControllers.delete(runId);
+      // Ensure only the current controller is removed
+      if (this.#abortControllers.get(runId) === controller) {
+        this.#abortControllers.delete(runId);
+      }
      controller.signal.removeEventListener("abort", onAbort);
    };