@koda-sl/baker-cli 0.270.3-dev.e350bbc93 → 0.270.4-dev.719d8b6f3

package/dist/{chunk-LJ5DIC4F.js → chunk-Y2GYULGB.js}

@@ -3131,6 +3131,16 @@ function dfsCycle(u, color, stack, reverseAdj) {
   stack.pop();
   return null;
 }
+function nodesBeforeCheckpoint(graph, types, stopKinds) {
+  const all = new Set(graph.keys());
+  if (!stopKinds?.size) return all;
+  const running = /* @__PURE__ */ new Set();
+  for (const layer of topologicalLayers(graph)) {
+    if (layer.some((id) => stopKinds.has(types.get(id) ?? ""))) return running;
+    for (const id of layer) running.add(id);
+  }
+  return running;
+}
 
 // src/engine/engine/validator.ts
 import { readFile as readFile2 } from "fs/promises";
@@ -3763,6 +3773,19 @@ function checkSlotsForNode(ctx, n, _i) {
     });
   }
 }
+function estimateCreditsFor(canvas, registry, ids) {
+  let total = 0;
+  for (const n of canvas.nodes) {
+    if (!ids.has(n.id)) continue;
+    const def = registry.get(n.type);
+    if (!def?.cost) continue;
+    try {
+      total += def.cost({ params: def.params.parse(n.params ?? {}) }).credits;
+    } catch {
+    }
+  }
+  return total;
+}
 function estimateCredits(ctx) {
   let total = 0;
   for (const n of ctx.canvas.nodes) {
@@ -4587,6 +4610,13 @@ function unwrap(schema) {
 }
 
 // src/engine/engine/executor.ts
+function heldByCheckpoint(canvas, layer, kinds) {
+  if (!kinds?.size) return [];
+  return layer.filter((id) => {
+    const type = canvas.nodes.find((n) => n.id === id)?.type;
+    return type !== void 0 && kinds.has(type);
+  });
+}
 var Engine = class {
   registry;
   client;
@@ -4614,7 +4644,7 @@ var Engine = class {
   async run(input, opts = {}) {
     const validation = await this.validateDeep(input);
     if (!validation.ok) throw new ValidationError(validation.issues);
-    if (opts.max_credits !== void 0 && validation.estimatedCredits > opts.max_credits) {
+    if (!opts.stop_before_kinds?.size && opts.max_credits !== void 0 && validation.estimatedCredits > opts.max_credits) {
       throw new RunAbortedError(
         "cost_cap",
         `estimated ${validation.estimatedCredits} credits exceeds the ${opts.max_credits}-credit cap \u2014 nothing was billed; raise --max-credits or shrink the canvas`
@@ -4633,6 +4663,20 @@ var Engine = class {
     const counters = { cachedNodes: 0, totalCredits: 0 };
     const nodeRuns = [];
     const graph = this.pruneToOutput(canvas, buildGraph(canvas));
+    if (opts.max_credits !== void 0 && opts.stop_before_kinds?.size) {
+      const willRun = nodesBeforeCheckpoint(
+        graph,
+        new Map(canvas.nodes.map((n) => [n.id, n.type])),
+        opts.stop_before_kinds
+      );
+      const estimate = estimateCreditsFor(canvas, this.registry, willRun);
+      if (estimate > opts.max_credits) {
+        throw new RunAbortedError(
+          "cost_cap",
+          `estimated ${estimate} credits exceeds the ${opts.max_credits}-credit cap \u2014 nothing was billed; raise --max-credits or shrink the canvas`
+        );
+      }
+    }
     const needsBytes = computeNeedsLocalBytes(canvas, graph, this.registry);
     this.emitProgress(opts, {
       kind: "plan",
@@ -4677,6 +4721,13 @@ var Engine = class {
           `spent ${counters.totalCredits} credits, over the ${opts.max_credits}-credit cap \u2014 completed nodes are cached; raise --max-credits to continue where this stopped`
         );
       }
+      const held = heldByCheckpoint(canvas, layer, opts.stop_before_kinds);
+      if (held.length > 0) {
+        throw new RunAbortedError(
+          "checkpoint",
+          `stopped before ${held.join(", ")} \u2014 everything up to here is done and cached, so continuing this run id pays for none of it again`
+        );
+      }
       const settled = await mapWithConcurrency(layer, limit, (nodeId) => {
         if (opts.signal?.aborted) {
           return Promise.reject(new RunAbortedError("signal", "run aborted before node dispatch"));
@@ -9294,4 +9345,4 @@ export {
   defaultRegistry,
   createEngineFromEnv
 };
-//# sourceMappingURL=chunk-LJ5DIC4F.js.map
+//# sourceMappingURL=chunk-Y2GYULGB.js.map