poe-code 10.0.1 → 10.0.3

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "poe-code",
-  "version": "10.0.1",
+  "version": "10.0.3",
   "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-E6LJNTDX.js → chunk-A2XBEN3H.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 {
@@ -28300,6 +28308,7 @@ function run(source, options = {}) {
   const lifecycle = { hostCallbackDepth: 0 };
   const dumpController = createDumpController(lifecycle);
   const promiseTracker = createSandboxPromiseRejectionTracker();
+  let completedSnapshot;
   const execute = async () => {
     const promiseReplay = new PromiseReplay(options.snapshot?.promiseReplay);
     return promiseReplayContext.run(promiseReplay, async () => {
@@ -28308,8 +28317,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 +28464,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 +28498,6 @@ function run(source, options = {}) {
                 randomResumeState: typeof yieldPoint.replayState === "number" ? yieldPoint.replayState : void 0,
                 sourceHash
               });
-              lastCheckpoint = snapshot2;
               return snapshot2;
             };
             activeSnapshotScheduler.onYield(createSnapshot);
@@ -28519,7 +28539,6 @@ function run(source, options = {}) {
                 randomResumeState: typeof yieldPoint.replayState === "number" ? yieldPoint.replayState : void 0,
                 sourceHash
               });
-              lastCheckpoint = snapshot2;
               return snapshot2;
             };
             activeSnapshotScheduler.onYield(createSnapshot);
@@ -28552,6 +28571,7 @@ function run(source, options = {}) {
         });
         if (replayError !== void 0) snapshot.replayError = replayError;
         dumpController.finalize(snapshot);
+        completedSnapshot = snapshot;
         return {
           ...result2,
           snapshot
@@ -28561,9 +28581,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 +28596,7 @@ function run(source, options = {}) {
         dumpController.fail(error);
         throw error;
       } finally {
+        options.signal?.removeEventListener("abort", captureCancellationSnapshot);
         leaveInputReplay?.();
         leaveHostReplay?.();
         promiseReplay.dispose();
@@ -28586,7 +28608,17 @@ function run(source, options = {}) {
   const result = withRunResources(
     options.signal,
     () => withSandboxPromiseRejectionTracker(promiseTracker, execute)
-  );
+  ).catch(async (error) => {
+    if (completedSnapshot !== void 0) {
+      try {
+        await createSnapshotScheduler(options).write(completedSnapshot);
+      } catch (snapshotError) {
+        console.warn("Failed to write failure snapshot.", snapshotError);
+      }
+    }
+    dumpController.fail(error);
+    throw error;
+  });
   return attachDumpController(result, dumpController);
 }
 async function throwIfReturnedPromiseRejected(result) {
@@ -28795,4 +28827,4 @@ export {
   FileSnapshotBackend,
   run
 };
-//# sourceMappingURL=chunk-E6LJNTDX.js.map
+//# sourceMappingURL=chunk-A2XBEN3H.js.map