@axlsdk/axl 0.24.3 → 0.25.0

package/dist/index.cjs

@@ -425,7 +425,7 @@ var TimeoutError = class extends AxlError {
   constructor(operation, timeoutMs, breakdown, agent2) {
     super(
       "TIMEOUT",
-      `${operation} exceeded timeout of ${timeoutMs}ms` + (breakdown ? ` (elapsed ${breakdown.elapsedMs}ms: queued ${breakdown.queuedMs}ms, retries ${breakdown.retryMs}ms, wire ${breakdown.wireMs}ms, other ${breakdown.otherMs}ms)` : "") + (agent2 ? ` for agent '${agent2}'` : "")
+      `${operation} exceeded timeout of ${timeoutMs}ms` + (breakdown ? ` (elapsed ${breakdown.elapsedMs}ms: queued ${breakdown.queuedMs}ms, retries ${breakdown.retryMs}ms, wire ${breakdown.wireMs}ms, other ${breakdown.otherMs}ms, charged ${breakdown.chargedMs}ms)` : "") + (agent2 ? ` for agent '${agent2}'` : "")
     );
     this.name = "TimeoutError";
     if (breakdown) this.breakdown = breakdown;
@@ -563,7 +563,7 @@ var sharedGuard = typeof installedGuard === "object" && installedGuard !== null
 var buildPath = typeof __AXL_BUILD_PATH__ === "string" ? __AXL_BUILD_PATH__ : "<source>";
 var thisCopy = {
   path: buildPath.startsWith("file:") ? (0, import_node_url.fileURLToPath)(buildPath) : buildPath,
-  version: true ? "0.24.3" : "unknown"
+  version: true ? "0.25.0" : "unknown"
 };
 function warnCrossCopy(other, joined) {
   if (other === thisCopy) return;
@@ -1547,6 +1547,41 @@ var OPENAI_DEFAULT_BASE_URL = "https://api.openai.com/v1";
 var ANTHROPIC_DEFAULT_BASE_URL = "https://api.anthropic.com/v1";
 var GEMINI_DEFAULT_BASE_URL = "https://generativelanguage.googleapis.com/v1beta";
 
+// src/ask-clock.ts
+var import_node_async_hooks2 = require("async_hooks");
+var askClockStorage = new import_node_async_hooks2.AsyncLocalStorage();
+function runWithAskClock(fn) {
+  const tracker = { pausedMs: 0, activePauses: 0 };
+  const enclosing = askClockStorage.getStore() ?? [];
+  return askClockStorage.run([...enclosing, tracker], fn);
+}
+function currentAskClock() {
+  return askClockStorage.getStore()?.at(-1);
+}
+function pausedAskClockMs(tracker, now = Date.now()) {
+  return tracker.pausedMs + (tracker.activePauses > 0 && tracker.pauseStartedAt !== void 0 ? now - tracker.pauseStartedAt : 0);
+}
+async function pauseAskClocks(wait) {
+  const trackers = askClockStorage.getStore();
+  if (!trackers || trackers.length === 0) return wait();
+  const startedAt = Date.now();
+  for (const tracker of trackers) {
+    if (tracker.activePauses++ === 0) tracker.pauseStartedAt = startedAt;
+  }
+  try {
+    return await wait();
+  } finally {
+    const finishedAt = Date.now();
+    for (const tracker of trackers) {
+      tracker.activePauses--;
+      if (tracker.activePauses === 0 && tracker.pauseStartedAt !== void 0) {
+        tracker.pausedMs += finishedAt - tracker.pauseStartedAt;
+        tracker.pauseStartedAt = void 0;
+      }
+    }
+  }
+}
+
 // src/providers/rate-limiter.ts
 function sanitizeRateLimitConfig(config) {
   const limits = {};
@@ -2609,7 +2644,7 @@ async function fetchWithRetry(input, init, opts) {
   let headersAt = 0;
   const waitSelfImposed = async (wait) => {
     const start = Date.now();
-    await wait();
+    await pauseAskClocks(wait);
     const waited = Date.now() - start;
     queuedMs += waited;
     if (dispatches > 0) queuedAfterFirstDispatchMs += waited;
@@ -9234,7 +9269,7 @@ var TranscriptionProviderRegistry = class {
 };
 
 // src/context.ts
-var import_node_async_hooks2 = require("async_hooks");
+var import_node_async_hooks3 = require("async_hooks");
 var import_node_crypto = require("crypto");
 var import_zod2 = require("zod");
 
@@ -12766,9 +12801,8 @@ function recordToolSpanOutcome(span, outcome) {
 }
 
 // src/context.ts
-var signalStorage = new import_node_async_hooks2.AsyncLocalStorage();
-var hardSignalStorage = new import_node_async_hooks2.AsyncLocalStorage();
-var timeoutTrackerStorage = new import_node_async_hooks2.AsyncLocalStorage();
+var signalStorage = new import_node_async_hooks3.AsyncLocalStorage();
+var hardSignalStorage = new import_node_async_hooks3.AsyncLocalStorage();
 var MAX_TIMER_DELAY_MS = 2147483647;
 function composeAbortSignals(...signals) {
   const active = signals.filter((signal) => signal !== void 0);
@@ -12800,9 +12834,6 @@ function raceWithAbortSignal(promise, signal) {
     );
   });
 }
-function pausedHumanWaitMs(tracker, now = Date.now()) {
-  return tracker.humanWaitMs + (tracker.activeHumanWaits > 0 && tracker.humanWaitStartedAt !== void 0 ? now - tracker.humanWaitStartedAt : 0);
-}
 var DEFAULT_BRANCH_DRAIN_TIMEOUT_MS = 5e3;
 function resolveBranchDrainTimeoutMs(value) {
   const resolved = value === void 0 ? DEFAULT_BRANCH_DRAIN_TIMEOUT_MS : value;
@@ -12873,7 +12904,7 @@ function parseHumanDecision(value) {
   }
   return invalid3();
 }
-var askStorage = new import_node_async_hooks2.AsyncLocalStorage();
+var askStorage = new import_node_async_hooks3.AsyncLocalStorage();
 var inlineSchemaCache = /* @__PURE__ */ new WeakMap();
 function zodToJsonSchema(schema) {
   const cached = inlineSchemaCache.get(schema);
@@ -13810,10 +13841,7 @@ var WorkflowContext = class _WorkflowContext {
     return hardAskSignal ? hardSignalStorage.run(hardAskSignal, runWithAskSignal) : runWithAskSignal();
   }
   async executeAgentCall(agent2, input, options, handoffMessages, usageCapture, sessionHistory = normalizeSessionHistory(this.sessionHistory)) {
-    const tracker = { humanWaitMs: 0, activeHumanWaits: 0 };
-    const enclosing = timeoutTrackerStorage.getStore() ?? [];
-    return timeoutTrackerStorage.run(
-      [...enclosing, tracker],
+    return runWithAskClock(
       () => this.executeAgentCallImpl(
         agent2,
         input,
@@ -14088,7 +14116,7 @@ ${appendedSchemaText}`;
       );
     }
     const startTime = Date.now();
-    const timeoutTracker = timeoutTrackerStorage.getStore()?.at(-1);
+    const askClock = currentAskClock();
     const timingTotals = { turns: 0, queuedMs: 0, retryMs: 0, wireMs: 0 };
     const currentMessages = [...messages];
     let turns = 0;
@@ -14105,13 +14133,16 @@ ${appendedSchemaText}`;
     while (turns < maxTurns) {
       this.currentSignal?.throwIfAborted();
       const now = Date.now();
-      const elapsedMs = now - startTime - (timeoutTracker ? pausedHumanWaitMs(timeoutTracker, now) : 0);
-      if (elapsedMs > timeoutMs) {
+      const elapsedMs = now - startTime;
+      const pausedMs = askClock ? pausedAskClockMs(askClock, now) : 0;
+      const chargedMs = Math.max(0, elapsedMs - Math.min(pausedMs, Math.max(0, elapsedMs)));
+      if (chargedMs > timeoutMs) {
         throw new TimeoutError(
           "ctx.ask()",
           timeoutMs,
           timingTotals.turns > 0 ? {
             elapsedMs,
+            chargedMs,
             queuedMs: timingTotals.queuedMs,
             retryMs: timingTotals.retryMs,
             wireMs: timingTotals.wireMs,
@@ -15981,12 +16012,7 @@ ${summaryResponse.content}`
   }
   // ── ctx.awaitHuman() ──────────────────────────────────────────────────
   async awaitHuman(options) {
-    const trackers = timeoutTrackerStorage.getStore();
-    const startedAt = Date.now();
-    for (const tracker of trackers ?? []) {
-      if (tracker.activeHumanWaits++ === 0) tracker.humanWaitStartedAt = startedAt;
-    }
-    try {
+    return pauseAskClocks(async () => {
       if (this.spanManager) {
         return await this.spanManager.withSpanAsync(
           "axl.ctx.awaitHuman",
@@ -16003,16 +16029,7 @@ ${summaryResponse.content}`
         );
       }
       return await this._awaitHumanImpl(options);
-    } finally {
-      const finishedAt = Date.now();
-      for (const tracker of trackers ?? []) {
-        tracker.activeHumanWaits--;
-        if (tracker.activeHumanWaits === 0 && tracker.humanWaitStartedAt !== void 0) {
-          tracker.humanWaitMs += finishedAt - tracker.humanWaitStartedAt;
-          tracker.humanWaitStartedAt = void 0;
-        }
-      }
-    }
+    });
   }
   async _awaitHumanImpl(options) {
     const branchSignal = this.currentSignal;
@@ -17009,7 +17026,7 @@ function workflow(config) {
 // src/runtime.ts
 var import_node_events2 = require("events");
 var import_node_crypto4 = require("crypto");
-var import_node_async_hooks3 = require("async_hooks");
+var import_node_async_hooks4 = require("async_hooks");
 var import_node_fs3 = require("fs");
 var import_node_path3 = require("path");
 
@@ -19189,7 +19206,7 @@ function assertNoRemovedObservationCallbacks(options) {
     `runtime.createContext() no longer accepts ${removed.join(", ")}. Observe ctx.events or use runtime.stream() instead. See docs/migration/stream-first-observation.md.`
   );
 }
-var costScopeStorage = new import_node_async_hooks3.AsyncLocalStorage();
+var costScopeStorage = new import_node_async_hooks4.AsyncLocalStorage();
 function forwardAbortSignal(external, internal) {
   if (!external) return () => {
   };