poe-code 10.0.1 → 10.0.2

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "poe-code",
-  "version": "10.0.1",
+  "version": "10.0.2",
   "description": "CLI tool to configure Poe API for developer workflows.",
   "license": "MIT",
   "type": "module",

package/packages/agent-harness/dist/loader/run.d.ts

@@ -1,4 +1,5 @@
 import { lint, run, type Diagnostic, type OtelSink, type SnapshotBackend, type SpawnUsageTotal } from "../../../safejs/dist/index.js";
+type RunOptions = NonNullable<Parameters<typeof run>[1]>;
 type LintOptions = NonNullable<Parameters<typeof lint>[1]>;
 export type RunResult = Awaited<ReturnType<typeof run>> & {
     usage: SpawnUsageTotal;
@@ -18,6 +19,7 @@ export type HarnessImportMeta = {
 };
 export type RunHarnessPairOptions = {
     allowedGlobals?: LintOptions["allowedGlobals"];
+    budget?: RunOptions["budget"];
     clock?: {
         now: () => number;
     };

package/packages/agent-harness/dist/loader/run.js

@@ -118,6 +118,7 @@ export async function runHarnessPair(mdPath, options) {
         let result;
         try {
             result = await runWithSpawnUsageAccumulator(usageAccumulator, () => run(executableSource, {
+                budget: options.budget,
                 clock: runtimeClock,
                 importMeta: meta,
                 entryPointArgs: [validated],

package/packages/safejs/dist/chunks/{chunk-HNGGGIMP.js → chunk-EUCMIR4E.js}

@@ -14,7 +14,7 @@ import {
   safeEndSpan,
   safeRecordException,
   safeStartSpan
-} from "./chunk-E6LJNTDX.js";
+} from "./chunk-RSK47E4C.js";
 
 // packages/safejs/src/error/format.ts
 var DEFAULT_MAX_MESSAGE_LENGTH = 1e4;
@@ -7886,4 +7886,4 @@ export {
   parseMcpConfig,
   makeMcpModule
 };
-//# sourceMappingURL=chunk-HNGGGIMP.js.map
+//# sourceMappingURL=chunk-EUCMIR4E.js.map

package/packages/safejs/dist/chunks/{chunk-E6LJNTDX.js → chunk-RSK47E4C.js}

@@ -17781,9 +17781,14 @@ var PromiseReplay = class {
     const id = this.nextPromise++;
     if (!this.replaying) {
       const settled = () => {
+        if (this.failure !== void 0) return;
         this.events.push({ kind: "promise", id, step: this.steps });
       };
-      void promise.then(settled, settled);
+      void promise.then(settled, (reason) => {
+        if (reason instanceof SandboxError && (reason.code === "budgetExceeded" || reason.code === "reentry"))
+          return;
+        settled();
+      });
       return promise;
     }
     const tracked = new Promise((resolve, reject) => {
@@ -17799,6 +17804,11 @@ var PromiseReplay = class {
         },
         (reason) => {
           if (this.failure !== void 0) return;
+          if (reason instanceof SandboxError && (reason.code === "budgetExceeded" || reason.code === "reentry")) {
+            this.rejectors.delete(id);
+            reject(reason);
+            return;
+          }
           this.ready.set(id, () => {
             this.rejectors.delete(id);
             reject(reason);
@@ -22702,14 +22712,12 @@ function createDumpController(lifecycle) {
       }
       return this.requestSnapshot();
     },
-    requestSnapshot() {
+    requestSnapshot(options = {}) {
       assertDumpAllowed();
       if (failed !== void 0) {
-        if (isDataBudgetError(failed.error) && (latestSnapshot !== void 0 || latestSnapshotFactory !== void 0)) {
+        if ((options.onFailure === "checkpoint" || options.onFailure === void 0 && isDataBudgetError(failed.error)) && finalSnapshot !== void 0) {
           try {
-            return Promise.resolve(
-              serializeRunSnapshot(latestSnapshot ?? latestSnapshotFactory())
-            );
+            return Promise.resolve(serializeRunSnapshot(finalSnapshot));
           } catch (error) {
             return Promise.reject(error);
           }
@@ -22768,10 +22776,10 @@ function createDumpController(lifecycle) {
 function isDataBudgetError(error) {
   return typeof error === "object" && error !== null && "code" in error && error.code === "budgetExceeded" && "budget" in error && error.budget === "dataSize";
 }
-function dump(result) {
+function dump(result, options = {}) {
   const controller = readDumpController(result);
   if (controller !== void 0) {
-    return controller.requestSnapshot();
+    return controller.requestSnapshot(options);
   }
   if (hasSnapshot(result)) {
     try {
@@ -28308,8 +28316,17 @@ function run(source, options = {}) {
       let leaveHostReplay;
       let leaveInputReplay;
       let createFailureSnapshot;
-      let lastCheckpoint;
+      let cancellationSnapshot;
+      let cancellationSnapshotError;
       let snapshotScheduler;
+      const captureCancellationSnapshot = () => {
+        try {
+          cancellationSnapshot = createFailureSnapshot?.();
+        } catch (reason) {
+          cancellationSnapshotError = { reason };
+        }
+      };
+      options.signal?.addEventListener("abort", captureCancellationSnapshot, { once: true });
       try {
         const restoredSnapshot = options.snapshot === void 0 ? void 0 : restore(options.snapshot, { source });
         if (restoredSnapshot !== void 0) {
@@ -28446,6 +28463,9 @@ function run(source, options = {}) {
           bindings: executionScope.snapshot().bindings,
           clock: options.clock,
           hostCalls: hostCalls.snapshot(),
+          replay: hostCalls.snapshotReplay(),
+          initialInputs: initialInputs.snapshot,
+          promiseReplay: promiseReplay.snapshot(),
           random,
           sourceHash
         });
@@ -28477,7 +28497,6 @@ function run(source, options = {}) {
                 randomResumeState: typeof yieldPoint.replayState === "number" ? yieldPoint.replayState : void 0,
                 sourceHash
               });
-              lastCheckpoint = snapshot2;
               return snapshot2;
             };
             activeSnapshotScheduler.onYield(createSnapshot);
@@ -28519,7 +28538,6 @@ function run(source, options = {}) {
                 randomResumeState: typeof yieldPoint.replayState === "number" ? yieldPoint.replayState : void 0,
                 sourceHash
               });
-              lastCheckpoint = snapshot2;
               return snapshot2;
             };
             activeSnapshotScheduler.onYield(createSnapshot);
@@ -28561,9 +28579,10 @@ function run(source, options = {}) {
         materializeWrappedErrorCause(error);
         if (createFailureSnapshot !== void 0 && snapshotScheduler !== void 0) {
           try {
-            const snapshot = lastCheckpoint ?? createFailureSnapshot();
-            await snapshotScheduler.write(snapshot);
+            if (cancellationSnapshotError !== void 0) throw cancellationSnapshotError.reason;
+            const snapshot = cancellationSnapshot ?? createFailureSnapshot();
             dumpController.finalize(snapshot);
+            await snapshotScheduler.write(snapshot);
           } catch (snapshotError) {
             if (snapshotError instanceof UnsnapshotableValueError) {
               console.warn(`Skipping failure snapshot: ${snapshotError.message}`);
@@ -28575,6 +28594,7 @@ function run(source, options = {}) {
         dumpController.fail(error);
         throw error;
       } finally {
+        options.signal?.removeEventListener("abort", captureCancellationSnapshot);
         leaveInputReplay?.();
         leaveHostReplay?.();
         promiseReplay.dispose();
@@ -28795,4 +28815,4 @@ export {
   FileSnapshotBackend,
   run
 };
-//# sourceMappingURL=chunk-E6LJNTDX.js.map
+//# sourceMappingURL=chunk-RSK47E4C.js.map