poe-code 11.0.22 → 11.0.23

package/dist/index.js

@@ -141613,7 +141613,7 @@ function invokeHostCallback(invoke2, options) {
   }
   lifecycle.hostCallbackDepth += 1;
   try {
-    const result = invoke2();
+    const result = lifecycle.hostCallbackContext.run(true, invoke2);
     if (isPromiseLike5(result)) {
       return Promise.resolve(result).finally(() => {
         lifecycle.hostCallbackDepth -= 1;
@@ -147663,8 +147663,8 @@ function createDumpController(lifecycle) {
       }
       settlePendingSnapshot(createSnapshot());
     },
-    requestCurrentSnapshot() {
-      assertDumpAllowed();
+    requestCurrentSnapshot(options = {}) {
+      assertDumpAllowed(options);
       if (failed !== void 0) {
         return Promise.reject(failed.error);
       }
@@ -147675,10 +147675,10 @@ function createDumpController(lifecycle) {
           return Promise.reject(error3);
         }
       }
-      return this.requestSnapshot();
+      return this.requestSnapshot(options);
     },
     requestSnapshot(options = {}) {
-      assertDumpAllowed();
+      assertDumpAllowed(options);
       if (failed !== void 0) {
         if ((options.onFailure === "checkpoint" || options.onFailure === void 0 && isDataBudgetError(failed.error)) && finalSnapshot !== void 0) {
           try {
@@ -147700,6 +147700,9 @@ function createDumpController(lifecycle) {
           return Promise.reject(error3);
         }
       }
+      if (options.mode === "replay" && (latestSnapshot !== void 0 || latestSnapshotFactory !== void 0)) {
+        return this.requestCurrentSnapshot(options);
+      }
       if (pendingRequest !== void 0) {
         return pendingRequest.promise;
       }
@@ -147717,8 +147720,8 @@ function createDumpController(lifecycle) {
       return promise;
     }
   };
-  function assertDumpAllowed() {
-    if ((lifecycle?.hostCallbackDepth ?? 0) > 0) {
+  function assertDumpAllowed(options) {
+    if ((lifecycle?.hostCallbackDepth ?? 0) > 0 && (options.mode !== "replay" || lifecycle?.hostCallbackContext.getStore() === true)) {
       throw new SandboxError("reentry");
     }
   }
@@ -148163,8 +148166,12 @@ var init_replay_inputs = __esm({
 });
 
 // packages/safejs/src/run.ts
+import { AsyncLocalStorage as AsyncLocalStorage8 } from "node:async_hooks";
 function run(source, options = {}) {
-  const lifecycle = { hostCallbackDepth: 0 };
+  const lifecycle = {
+    hostCallbackDepth: 0,
+    hostCallbackContext: new AsyncLocalStorage8()
+  };
   const dumpController = createDumpController(lifecycle);
   const promiseTracker = createSandboxPromiseRejectionTracker();
   let completedSnapshot;
@@ -148470,6 +148477,7 @@ function run(source, options = {}) {
         dumpController.fail(error3);
         throw error3;
       } finally {
+        lifecycle.hostCallbackContext.disable();
         options.signal?.removeEventListener("abort", captureCancellationSnapshot);
         leaveInputReplay?.();
         leaveHostReplay?.();
@@ -149037,7 +149045,7 @@ var init_find_exported = __esm({
 });
 
 // packages/safejs/src/modules/agent.ts
-import { AsyncLocalStorage as AsyncLocalStorage8 } from "node:async_hooks";
+import { AsyncLocalStorage as AsyncLocalStorage9 } from "node:async_hooks";
 function createSpawnUsageAccumulator() {
   let inputTokens = 0;
   let outputTokens = 0;
@@ -149940,7 +149948,7 @@ var init_agent3 = __esm({
         hostErrorData.set(this, { result });
       }
     };
-    activeUsageAccumulator = new AsyncLocalStorage8();
+    activeUsageAccumulator = new AsyncLocalStorage9();
     oneAttemptRetryOptions = {
       maxAttempts: 1,
       backoffMs: 0,
@@ -155709,7 +155717,7 @@ var init_package2 = __esm({
   "package.json"() {
     package_default2 = {
       name: "poe-code",
-      version: "11.0.22",
+      version: "11.0.23",
       description: "CLI tool to configure Poe API for developer workflows.",
       license: "MIT",
       type: "module",