@bastani/atomic 0.9.11-alpha.5 → 0.9.11-alpha.7

package/dist/core/agent-session-message-queue.js

@@ -1,6 +1,21 @@
 import { customMessageExcludesContext, drainAgentMessageQueue, normalizeInterruptAbortMessage } from "./agent-session-types.js";
-import { createSessionAsyncDeliveryHandler } from "./async/session-manager.js";
+import { forwardedMessageOptions, resolveWorkflowStageDeliveryTarget } from "./agent-session-delivery-forwarding.js";
+import { transferWorkflowStageDeliveriesTo } from "./agent-session-transfer.js";
+import { abort, pauseQueuedMessages, resumeQueuedMessages } from "./agent-session-queue-pause.js";
+import { restoreProtectedStreamingCustomMessages } from "./agent-session-persistent-custom-messages.js";
+import { commitAdmittedCustomMessage, commitAdmittedCustomMessages } from "./agent-session-custom-message-commit.js";
+export { transferWorkflowStageDeliveriesTo };
+const interruptMutationQueues = new WeakMap();
+function serializeInterruptMutation(owner, operation) {
+    const previous = interruptMutationQueues.get(owner) ?? Promise.resolve();
+    const delivery = previous.then(operation);
+    interruptMutationQueues.set(owner, delivery.catch(() => undefined));
+    return delivery;
+}
 export async function _queueSteer(text, images) {
+    const owner = resolveWorkflowStageDeliveryTarget(this);
+    if (owner !== this)
+        return owner._queueSteer(text, images);
     this._steeringMessages.push(text);
     this._emitQueueUpdate();
     const content = [{ type: "text", text }];
@@ -17,6 +32,9 @@ export async function _queueSteer(text, images) {
  * Internal: Queue a follow-up message (already expanded, no extension command check).
  */
 export async function _queueFollowUp(text, images) {
+    const owner = resolveWorkflowStageDeliveryTarget(this);
+    if (owner !== this)
+        return owner._queueFollowUp(text, images);
     this._followUpMessages.push(text);
     this._emitQueueUpdate();
     const content = [{ type: "text", text }];
@@ -55,6 +73,9 @@ export function _throwIfExtensionCommand(text) {
  * @param options.deliverAs Delivery mode: "steer", "followUp", "nextTurn", or "interrupt"
  */
 export async function sendCustomMessage(message, options) {
+    const currentOwner = resolveWorkflowStageDeliveryTarget(this);
+    if (currentOwner !== this)
+        return currentOwner.sendCustomMessage(message, options);
     const appMessage = {
         role: "custom",
         customType: message.customType,
@@ -68,44 +89,7 @@ export async function sendCustomMessage(message, options) {
     const deliver = async () => {
         if (boundary && options?.stageAdmissionBarrier)
             await options.stageAdmissionBarrier();
-        if (options?.deliverAs === "nextTurn") {
-            this._pendingNextTurnMessages.push(appMessage);
-        }
-        else if (options?.deliverAs === "interrupt" && options.triggerTurn) {
-            const interrupt = this._enqueueInterruptCustomMessage(appMessage, options);
-            boundary?.trackAdmittedWork(interrupt);
-            void interrupt.catch(() => { });
-        }
-        else if (this.isStreaming && options?.excludeFromContext === true && options.triggerTurn !== true && options.deliverAs === undefined) {
-            this._appendCustomMessage(appMessage);
-        }
-        else if (this.isStreaming && options?.persistWhenStreaming === true) {
-            // Persist the notice to the transcript so a streaming parent's cleared
-            // steer queue or aborted turn cannot silently drop it; it stays visible
-            // and in-context for the model's next step.
-            this._appendCustomMessage(appMessage);
-        }
-        else if (this.isStreaming) {
-            this._queueAgentMessage(appMessage, options?.deliverAs === "followUp" ? "followUp" : "steer");
-        }
-        else if (options?.triggerTurn) {
-            // ES2022 build target: use a manually constructed deferred instead of
-            // Promise.withResolvers (ES2024). Resolve on prompt-start so a later
-            // model-turn rejection cannot un-admit an already-delivered notice;
-            // reject only when the turn fails before that start signal fires.
-            let resolveAdmission;
-            let rejectAdmission;
-            const admission = new Promise((resolve, reject) => {
-                resolveAdmission = resolve;
-                rejectAdmission = reject;
-            });
-            const turn = this._runAgentPrompt(appMessage, resolveAdmission);
-            void turn.then(resolveAdmission, rejectAdmission);
-            await admission;
-        }
-        else {
-            this._appendCustomMessage(appMessage);
-        }
+        await commitAdmittedCustomMessage(this, appMessage, options);
     };
     if (boundary === undefined)
         return deliver();
@@ -118,6 +102,9 @@ export async function sendCustomMessage(message, options) {
 }
 /** Atomically admits a custom-message batch in array order. */
 export async function sendCustomMessages(messages, options) {
+    const currentOwner = resolveWorkflowStageDeliveryTarget(this);
+    if (currentOwner !== this)
+        return currentOwner.sendCustomMessages(messages, options);
     const timestamp = Date.now();
     const appMessages = messages.map((message) => ({
         role: "custom",
@@ -134,26 +121,7 @@ export async function sendCustomMessages(messages, options) {
     const deliver = async () => {
         if (boundary && options?.stageAdmissionBarrier)
             await options.stageAdmissionBarrier();
-        if (options?.deliverAs === "nextTurn") {
-            this._pendingNextTurnMessages.push(...appMessages);
-        }
-        else if (this.isStreaming && options?.excludeFromContext === true && options.triggerTurn !== true && options.deliverAs === undefined) {
-            for (const item of appMessages)
-                this._appendCustomMessage(item);
-        }
-        else if (this.isStreaming) {
-            const delivery = options?.deliverAs === "followUp" ? "followUp" : "steer";
-            for (const item of appMessages)
-                this._queueAgentMessage(item, delivery);
-        }
-        else if (options?.triggerTurn) {
-            const turn = this._runAgentPrompt(appMessages);
-            void turn.catch(() => { });
-        }
-        else {
-            for (const item of appMessages)
-                this._appendCustomMessage(item);
-        }
+        await commitAdmittedCustomMessages(this, appMessages, options);
     };
     if (boundary === undefined)
         return deliver();
@@ -172,25 +140,19 @@ export async function closeWorkflowStageGeneration() {
     await this.agent?.waitForIdle?.();
     await this._agentEventQueue;
 }
-export function transferWorkflowStageDeliveriesTo(target) {
-    const next = target;
-    const queued = this._drainQueuedAgentMessages();
-    if (this._activeInterruptQueueHold) {
-        queued.steering.push(...this._activeInterruptQueueHold.steering);
-        queued.followUp.push(...this._activeInterruptQueueHold.followUp);
-        this._activeInterruptQueueHold = undefined;
-    }
-    next._pendingNextTurnMessages.unshift(...this._pendingNextTurnMessages.splice(0));
-    next._restoreQueuedAgentMessages(queued);
-    this._asyncJobManager.transferSessionDeliveries(this._asyncJobManagerSessionId, next._asyncJobManagerSessionId, createSessionAsyncDeliveryHandler(next, next._asyncJobManager, next._asyncJobManagerSessionId));
-}
 export function _appendCustomMessage(message) {
+    const owner = resolveWorkflowStageDeliveryTarget(this);
+    if (owner !== this)
+        return owner._appendCustomMessage(message);
     this.agent.state.messages.push(message);
     this.sessionManager.appendCustomMessageEntry(message.customType, message.content, message.display, message.details, customMessageExcludesContext(message));
     this._emit({ type: "message_start", message });
     this._emit({ type: "message_end", message });
 }
 export function _enqueueInterruptCustomMessage(message, options) {
+    const owner = resolveWorkflowStageDeliveryTarget(this);
+    if (owner !== this)
+        return owner._enqueueInterruptCustomMessage(message, forwardedMessageOptions(options));
     this._pendingInterruptDeliveries += 1;
     // Establish the hold synchronously when the interrupt is enqueued, not when
     // the serialized delivery callback later starts. Callers commonly fire and
@@ -203,33 +165,52 @@ export function _enqueueInterruptCustomMessage(message, options) {
             await this._sendInterruptCustomMessageNow(message, options);
         }
         finally {
-            this._pendingInterruptDeliveries -= 1;
-            if (this._pendingInterruptDeliveries === 0) {
-                this._restoreAndClearActiveInterruptQueueHold();
+            // Retirement moves the pending count and hold synchronously. Settle the
+            // same live owner so its queue cannot remain stranded on the replacement.
+            const liveOwner = resolveWorkflowStageDeliveryTarget(this);
+            liveOwner._pendingInterruptDeliveries -= 1;
+            if (liveOwner._pendingInterruptDeliveries === 0) {
+                liveOwner._restoreAndClearActiveInterruptQueueHold();
             }
         }
     });
     this._interruptDeliveryQueue = delivery.catch(() => undefined);
     return delivery;
 }
-export async function _sendInterruptCustomMessageNow(message, options) {
-    this.abortRetry();
-    this._ensureActiveInterruptQueueHold();
-    if (this.isStreaming) {
-        const previousAbortMessage = this._activeInterruptAbortMessage;
-        this._activeInterruptAbortMessage = normalizeInterruptAbortMessage(options?.interruptAbortMessage);
+async function sendInterruptCustomMessageUnlocked(session, message, options) {
+    session.abortRetry();
+    session._ensureActiveInterruptQueueHold();
+    if (session.isStreaming) {
+        const previousAbortMessage = session._activeInterruptAbortMessage;
+        session._activeInterruptAbortMessage = normalizeInterruptAbortMessage(options?.interruptAbortMessage);
         try {
-            this.agent.abort();
-            await this.agent.waitForIdle();
-            await this._agentEventQueue;
+            session.agent.abort();
+            await session.agent.waitForIdle();
+            await session._agentEventQueue;
         }
         finally {
-            this._activeInterruptAbortMessage = previousAbortMessage;
+            session._activeInterruptAbortMessage = previousAbortMessage;
         }
     }
-    await this.agent.prompt(message);
+    const owner = resolveWorkflowStageDeliveryTarget(session);
+    if (owner !== session)
+        return owner._sendInterruptCustomMessageNow(message, forwardedMessageOptions(options));
+    if (session._queuedMessagesPaused) {
+        session._queueAgentMessage(message, "steer");
+        return;
+    }
+    await session.agent.prompt(message);
+}
+export function _sendInterruptCustomMessageNow(message, options) {
+    const owner = resolveWorkflowStageDeliveryTarget(this);
+    if (owner !== this)
+        return owner._sendInterruptCustomMessageNow(message, forwardedMessageOptions(options));
+    return serializeInterruptMutation(this, () => sendInterruptCustomMessageUnlocked(this, message, options));
 }
 export function _ensureActiveInterruptQueueHold() {
+    const owner = resolveWorkflowStageDeliveryTarget(this);
+    if (owner !== this)
+        return owner._ensureActiveInterruptQueueHold();
     if (this._activeInterruptQueueHold !== undefined) {
         return this._activeInterruptQueueHold;
     }
@@ -241,10 +222,14 @@ export function _ensureActiveInterruptQueueHold() {
     return this._activeInterruptQueueHold;
 }
 export function _restoreAndClearActiveInterruptQueueHold() {
+    const owner = resolveWorkflowStageDeliveryTarget(this);
+    if (owner !== this)
+        return owner._restoreAndClearActiveInterruptQueueHold();
+    if (this._queuedMessagesPaused)
+        return;
     const hold = this._activeInterruptQueueHold;
-    if (hold === undefined) {
+    if (hold === undefined)
         return;
-    }
     const currentCoreQueues = this._drainQueuedAgentMessages();
     this._restoreQueuedAgentMessages({
         steering: [...hold.steering, ...currentCoreQueues.steering],
@@ -253,6 +238,9 @@ export function _restoreAndClearActiveInterruptQueueHold() {
     this._activeInterruptQueueHold = undefined;
 }
 export function _queueAgentMessage(message, delivery) {
+    const owner = resolveWorkflowStageDeliveryTarget(this);
+    if (owner !== this)
+        return owner._queueAgentMessage(message, delivery);
     const hold = this._activeInterruptQueueHold;
     if (hold !== undefined) {
         if (delivery === "followUp") {
@@ -272,8 +260,8 @@ export function _queueAgentMessage(message, delivery) {
 }
 export function _drainQueuedAgentMessages() {
     // pi-agent-core exposes public clear methods but no public drain/restore pair.
-    // Interrupts need to prevent the aborting run from consuming queued steer/follow-up
-    // messages while still preserving those queues for a later turn.
+    // Interrupt and pause holds prevent an aborting run from consuming queued raw
+    // messages while preserving every entry for the existing resume boundary.
     const agentWithQueues = this.agent;
     return {
         steering: drainAgentMessageQueue(agentWithQueues.steeringQueue),
@@ -296,14 +284,23 @@ export function _restoreQueuedAgentMessages(queues) {
  * @param options.deliverAs Delivery mode when streaming: "steer" or "followUp"
  */
 export function clearQueue() {
+    const owner = resolveWorkflowStageDeliveryTarget(this);
+    if (owner !== this)
+        return owner.clearQueue();
     const steering = [...this._steeringMessages];
     const followUp = [...this._followUpMessages];
     this._steeringMessages = [];
     this._followUpMessages = [];
-    this.agent.clearAllQueues();
+    const removed = this._drainQueuedAgentMessages();
     if (this._activeInterruptQueueHold !== undefined) {
-        this._activeInterruptQueueHold.steering.length = 0;
-        this._activeInterruptQueueHold.followUp.length = 0;
+        removed.steering.push(...this._activeInterruptQueueHold.steering.splice(0));
+        removed.followUp.push(...this._activeInterruptQueueHold.followUp.splice(0));
+    }
+    restoreProtectedStreamingCustomMessages(this, removed);
+    const hold = this._activeInterruptQueueHold;
+    if (!this._queuedMessagesPaused && this._pendingInterruptDeliveries === 0 && hold !== undefined
+        && hold.steering.length === 0 && hold.followUp.length === 0) {
+        this._activeInterruptQueueHold = undefined;
     }
     this._emitQueueUpdate();
     return { steering, followUp };
@@ -316,11 +313,6 @@ export function getSteeringMessages() {
 export function getFollowUpMessages() {
     return this._followUpMessages;
 }
-export async function abort() {
-    this.abortRetry();
-    this.agent.abort();
-    await this.agent.waitForIdle();
-}
 // =========================================================================
 // Model Management
 // =========================================================================
@@ -353,6 +345,8 @@ export const agentSessionMessageQueueMethods = {
     _sendInterruptCustomMessageNow,
     _ensureActiveInterruptQueueHold,
     _restoreAndClearActiveInterruptQueueHold,
+    pauseQueuedMessages,
+    resumeQueuedMessages,
     _queueAgentMessage,
     _drainQueuedAgentMessages,
     _restoreQueuedAgentMessages,

package/dist/core/agent-session-message-queue.js.map

@@ -1 +1 @@
-{"version":3,"file":"agent-session-message-queue.js","sourceRoot":"","sources":["../../src/core/agent-session-message-queue.ts"],"names":[],"mappings":"AAKA,OAAO,EAAE,4BAA4B,EAAE,sBAAsB,EAAE,8BAA8B,EAA2E,MAAM,0BAA0B,CAAC;AACzM,OAAO,EAAE,iCAAiC,EAAE,MAAM,4BAA4B,CAAC;AAE/E,MAAM,CAAC,KAAK,UAAU,WAAW,CAAqB,IAAY,EAAE,MAAuB;IAC1F,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAClC,IAAI,CAAC,gBAAgB,EAAE,CAAC;IACxB,MAAM,OAAO,GAAmC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;IACzE,IAAI,MAAM,EAAE,CAAC;QACZ,OAAO,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,CAAC;IACzB,CAAC;IACD,IAAI,CAAC,kBAAkB,CACtB;QACC,IAAI,EAAE,MAAM;QACZ,OAAO;QACP,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;KACrB,EACD,OAAO,CACP,CAAC;AACH,CAAC;AAED;;GAEG;AAEH,MAAM,CAAC,KAAK,UAAU,cAAc,CAAqB,IAAY,EAAE,MAAuB;IAC7F,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAClC,IAAI,CAAC,gBAAgB,EAAE,CAAC;IACxB,MAAM,OAAO,GAAmC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;IACzE,IAAI,MAAM,EAAE,CAAC;QACZ,OAAO,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,CAAC;IACzB,CAAC;IACD,IAAI,CAAC,kBAAkB,CACtB;QACC,IAAI,EAAE,MAAM;QACZ,OAAO;QACP,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;KACrB,EACD,UAAU,CACV,CAAC;AACH,CAAC;AAED;;GAEG;AAEH,MAAM,UAAU,wBAAwB,CAAqB,IAAY;IACxE,MAAM,UAAU,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;IACrC,MAAM,WAAW,GAAG,UAAU,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,UAAU,CAAC,CAAC;IAClF,MAAM,OAAO,GAAG,IAAI,CAAC,gBAAgB,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC;IAE9D,IAAI,OAAO,EAAE,CAAC;QACb,MAAM,IAAI,KAAK,CACd,uBAAuB,WAAW,6EAA6E,CAC/G,CAAC;IACH,CAAC;AACF,CAAC;AAED;;;;;;;;;;;;;GAaG;AAEH,MAAM,CAAC,KAAK,UAAU,iBAAiB,CACtC,OAAiF,EACjF,OAA4B;IAE5B,MAAM,UAAU,GAAG;QAClB,IAAI,EAAE,QAAiB;QACvB,UAAU,EAAE,OAAO,CAAC,UAAU;QAC9B,OAAO,EAAE,OAAO,CAAC,OAAO,IAAI,EAAE;QAC9B,OAAO,EAAE,OAAO,CAAC,OAAO;QACxB,OAAO,EAAE,OAAO,CAAC,OAAO;QACxB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;QACrB,GAAG,CAAC,OAAO,EAAE,kBAAkB,KAAK,IAAI,CAAC,CAAC,CAAC,EAAE,kBAAkB,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KAClD,CAAC;IAC7B,MAAM,QAAQ,GAAG,IAAI,CAAC,uBAAuB,CAAC;IAC9C,MAAM,OAAO,GAAG,KAAK,IAAmB,EAAE;QACzC,IAAI,QAAQ,IAAI,OAAO,EAAE,qBAAqB;YAAE,MAAM,OAAO,CAAC,qBAAqB,EAAE,CAAC;QACtF,IAAI,OAAO,EAAE,SAAS,KAAK,UAAU,EAAE,CAAC;YACvC,IAAI,CAAC,wBAAwB,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAChD,CAAC;aAAM,IAAI,OAAO,EAAE,SAAS,KAAK,WAAW,IAAI,OAAO,CAAC,WAAW,EAAE,CAAC;YACtE,MAAM,SAAS,GAAG,IAAI,CAAC,8BAA8B,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;YAC3E,QAAQ,EAAE,iBAAiB,CAAC,SAAS,CAAC,CAAC;YACvC,KAAK,SAAS,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;QAChC,CAAC;aAAM,IAAI,IAAI,CAAC,WAAW,IAAI,OAAO,EAAE,kBAAkB,KAAK,IAAI,IAAI,OAAO,CAAC,WAAW,KAAK,IAAI,IAAI,OAAO,CAAC,SAAS,KAAK,SAAS,EAAE,CAAC;YACxI,IAAI,CAAC,oBAAoB,CAAC,UAAU,CAAC,CAAC;QACvC,CAAC;aAAM,IAAI,IAAI,CAAC,WAAW,IAAI,OAAO,EAAE,oBAAoB,KAAK,IAAI,EAAE,CAAC;YACvE,uEAAuE;YACvE,wEAAwE;YACxE,4CAA4C;YAC5C,IAAI,CAAC,oBAAoB,CAAC,UAAU,CAAC,CAAC;QACvC,CAAC;aAAM,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YAC7B,IAAI,CAAC,kBAAkB,CAAC,UAAU,EAAE,OAAO,EAAE,SAAS,KAAK,UAAU,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;QAC/F,CAAC;aAAM,IAAI,OAAO,EAAE,WAAW,EAAE,CAAC;YACjC,sEAAsE;YACtE,qEAAqE;YACrE,oEAAoE;YACpE,kEAAkE;YAClE,IAAI,gBAA6B,CAAC;YAClC,IAAI,eAA0C,CAAC;YAC/C,MAAM,SAAS,GAAG,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;gBACvD,gBAAgB,GAAG,OAAO,CAAC;gBAC3B,eAAe,GAAG,MAAM,CAAC;YAC1B,CAAC,CAAC,CAAC;YACH,MAAM,IAAI,GAAG,IAAI,CAAC,eAAe,CAAC,UAAU,EAAE,gBAAgB,CAAC,CAAC;YAChE,KAAK,IAAI,CAAC,IAAI,CAAC,gBAAgB,EAAE,eAAe,CAAC,CAAC;YAClD,MAAM,SAAS,CAAC;QACjB,CAAC;aAAM,CAAC;YACP,IAAI,CAAC,oBAAoB,CAAC,UAAU,CAAC,CAAC;QACvC,CAAC;IACF,CAAC,CAAC;IACF,IAAI,QAAQ,KAAK,SAAS;QAAE,OAAO,OAAO,EAAE,CAAC;IAC7C,MAAM,QAAQ,CAAC,KAAK,CACnB,OAAO,EAAE,iBAAiB,EAC1B,OAAO,EACP,GAAG,EAAE;QACJ,MAAM,MAAM,GAAG,IAAI,CAAC,qBAAqB,EAAE,iBAAiB,CAAC;QAC7D,IAAI,MAAM,KAAK,SAAS;YAAE,MAAM,IAAI,KAAK,CAAC,qDAAqD,CAAC,CAAC;QACjG,OAAO,MAAM,CAAC,YAAY,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;IAC9C,CAAC,CACD,CAAC,UAAU,CAAC;AACd,CAAC;AAED,+DAA+D;AAC/D,MAAM,CAAC,KAAK,UAAU,kBAAkB,CACvC,QAAyF,EACzF,OAA6B;IAE7B,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAC7B,MAAM,WAAW,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;QAC9C,IAAI,EAAE,QAAiB;QACvB,UAAU,EAAE,OAAO,CAAC,UAAU;QAC9B,OAAO,EAAE,OAAO,CAAC,OAAO,IAAI,EAAE;QAC9B,OAAO,EAAE,OAAO,CAAC,OAAO;QACxB,OAAO,EAAE,OAAO,CAAC,OAAO;QACxB,SAAS;QACT,GAAG,CAAC,OAAO,EAAE,kBAAkB,KAAK,IAAI,CAAC,CAAC,CAAC,EAAE,kBAAkB,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KACjD,CAAA,CAAC,CAAC;IAC/B,IAAI,WAAW,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO;IACrC,MAAM,QAAQ,GAAG,IAAI,CAAC,uBAAuB,CAAC;IAC9C,MAAM,OAAO,GAAG,KAAK,IAAmB,EAAE;QACzC,IAAI,QAAQ,IAAI,OAAO,EAAE,qBAAqB;YAAE,MAAM,OAAO,CAAC,qBAAqB,EAAE,CAAC;QACtF,IAAI,OAAO,EAAE,SAAS,KAAK,UAAU,EAAE,CAAC;YACvC,IAAI,CAAC,wBAAwB,CAAC,IAAI,CAAC,GAAG,WAAW,CAAC,CAAC;QACpD,CAAC;aAAM,IAAI,IAAI,CAAC,WAAW,IAAI,OAAO,EAAE,kBAAkB,KAAK,IAAI,IAAI,OAAO,CAAC,WAAW,KAAK,IAAI,IAAI,OAAO,CAAC,SAAS,KAAK,SAAS,EAAE,CAAC;YACxI,KAAK,MAAM,IAAI,IAAI,WAAW;gBAAE,IAAI,CAAC,oBAAoB,CAAC,IAAI,CAAC,CAAC;QACjE,CAAC;aAAM,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YAC7B,MAAM,QAAQ,GAAG,OAAO,EAAE,SAAS,KAAK,UAAU,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,OAAO,CAAC;YAC1E,KAAK,MAAM,IAAI,IAAI,WAAW;gBAAE,IAAI,CAAC,kBAAkB,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;QACzE,CAAC;aAAM,IAAI,OAAO,EAAE,WAAW,EAAE,CAAC;YACjC,MAAM,IAAI,GAAG,IAAI,CAAC,eAAe,CAAC,WAAW,CAAC,CAAC;YAC/C,KAAK,IAAI,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;QAC3B,CAAC;aAAM,CAAC;YACP,KAAK,MAAM,IAAI,IAAI,WAAW;gBAAE,IAAI,CAAC,oBAAoB,CAAC,IAAI,CAAC,CAAC;QACjE,CAAC;IACF,CAAC,CAAC;IACF,IAAI,QAAQ,KAAK,SAAS;QAAE,OAAO,OAAO,EAAE,CAAC;IAC7C,MAAM,QAAQ,CAAC,KAAK,CACnB,OAAO,EAAE,iBAAiB,EAC1B,OAAO,EACP,GAAG,EAAE;QACJ,MAAM,MAAM,GAAG,IAAI,CAAC,qBAAqB,EAAE,iBAAiB,CAAC;QAC7D,IAAI,MAAM,KAAK,SAAS;YAAE,MAAM,IAAI,KAAK,CAAC,qDAAqD,CAAC,CAAC;QACjG,OAAO,MAAM,CAAC,aAAa,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;IAChD,CAAC,CACD,CAAC,UAAU,CAAC;AACd,CAAC;AAGD,MAAM,UAAU,2BAA2B;IAC1C,IAAI,CAAC,uBAAuB,EAAE,IAAI,EAAE,CAAC;AACtC,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,4BAA4B;IACjD,MAAM,IAAI,CAAC,uBAAuB,EAAE,KAAK,EAAE,CAAC;IAC5C,MAAM,IAAI,CAAC,KAAK,EAAE,WAAW,EAAE,EAAE,CAAC;IAClC,MAAM,IAAI,CAAC,gBAAgB,CAAC;AAC7B,CAAC;AAED,MAAM,UAAU,iCAAiC,CAAqB,MAAc;IACnF,MAAM,IAAI,GAAG,MAAsB,CAAC;IACpC,MAAM,MAAM,GAAG,IAAI,CAAC,yBAAyB,EAAE,CAAC;IAChD,IAAI,IAAI,CAAC,yBAAyB,EAAE,CAAC;QACpC,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,yBAAyB,CAAC,QAAQ,CAAC,CAAC;QACjE,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,yBAAyB,CAAC,QAAQ,CAAC,CAAC;QACjE,IAAI,CAAC,yBAAyB,GAAG,SAAS,CAAC;IAC5C,CAAC;IACD,IAAI,CAAC,wBAAwB,CAAC,OAAO,CAAC,GAAG,IAAI,CAAC,wBAAwB,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;IAClF,IAAI,CAAC,2BAA2B,CAAC,MAAM,CAAC,CAAC;IACzC,IAAI,CAAC,gBAAgB,CAAC,yBAAyB,CAC9C,IAAI,CAAC,yBAAyB,EAC9B,IAAI,CAAC,yBAAyB,EAC9B,iCAAiC,CAAC,IAAI,EAAE,IAAI,CAAC,gBAAgB,EAAE,IAAI,CAAC,yBAAyB,CAAC,CAC9F,CAAC;AACH,CAAC;AACD,MAAM,UAAU,oBAAoB,CAAwB,OAAyB;IACpF,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACxC,IAAI,CAAC,cAAc,CAAC,wBAAwB,CAC3C,OAAO,CAAC,UAAU,EAClB,OAAO,CAAC,OAAO,EACf,OAAO,CAAC,OAAO,EACf,OAAO,CAAC,OAAO,EACf,4BAA4B,CAAC,OAAO,CAAC,CACrC,CAAC;IACF,IAAI,CAAC,KAAK,CAAC,EAAE,IAAI,EAAE,eAAe,EAAE,OAAO,EAAE,CAAC,CAAC;IAC/C,IAAI,CAAC,KAAK,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,OAAO,EAAE,CAAC,CAAC;AAC9C,CAAC;AAGD,MAAM,UAAU,8BAA8B,CAAwB,OAAyB,EAAE,OAA4B;IAC5H,IAAI,CAAC,2BAA2B,IAAI,CAAC,CAAC;IACtC,4EAA4E;IAC5E,2EAA2E;IAC3E,6EAA6E;IAC7E,6EAA6E;IAC7E,uEAAuE;IACvE,IAAI,CAAC,+BAA+B,EAAE,CAAC;IACvC,MAAM,QAAQ,GAAG,IAAI,CAAC,uBAAuB,CAAC,IAAI,CAAC,KAAK,IAAI,EAAE;QAC7D,IAAI,CAAC;YACJ,MAAM,IAAI,CAAC,8BAA8B,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;QAC7D,CAAC;gBAAS,CAAC;YACV,IAAI,CAAC,2BAA2B,IAAI,CAAC,CAAC;YACtC,IAAI,IAAI,CAAC,2BAA2B,KAAK,CAAC,EAAE,CAAC;gBAC5C,IAAI,CAAC,wCAAwC,EAAE,CAAC;YACjD,CAAC;QACF,CAAC;IACF,CAAC,CAAC,CAAC;IACH,IAAI,CAAC,uBAAuB,GAAG,QAAQ,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;IAC/D,OAAO,QAAQ,CAAC;AACjB,CAAC;AAGD,MAAM,CAAC,KAAK,UAAU,8BAA8B,CACnD,OAAyB,EACzB,OAA4B;IAE5B,IAAI,CAAC,UAAU,EAAE,CAAC;IAClB,IAAI,CAAC,+BAA+B,EAAE,CAAC;IACvC,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;QACtB,MAAM,oBAAoB,GAAG,IAAI,CAAC,4BAA4B,CAAC;QAC/D,IAAI,CAAC,4BAA4B,GAAG,8BAA8B,CAAC,OAAO,EAAE,qBAAqB,CAAC,CAAC;QACnG,IAAI,CAAC;YACJ,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;YACnB,MAAM,IAAI,CAAC,KAAK,CAAC,WAAW,EAAE,CAAC;YAC/B,MAAM,IAAI,CAAC,gBAAgB,CAAC;QAC7B,CAAC;gBAAS,CAAC;YACV,IAAI,CAAC,4BAA4B,GAAG,oBAAoB,CAAC;QAC1D,CAAC;IACF,CAAC;IACD,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;AAClC,CAAC;AAGD,MAAM,UAAU,+BAA+B;IAC9C,IAAI,IAAI,CAAC,yBAAyB,KAAK,SAAS,EAAE,CAAC;QAClD,OAAO,IAAI,CAAC,yBAAyB,CAAC;IACvC,CAAC;IACD,MAAM,OAAO,GAAG,IAAI,CAAC,yBAAyB,EAAE,CAAC;IACjD,IAAI,CAAC,yBAAyB,GAAG;QAChC,QAAQ,EAAE,CAAC,GAAG,OAAO,CAAC,QAAQ,CAAC;QAC/B,QAAQ,EAAE,CAAC,GAAG,OAAO,CAAC,QAAQ,CAAC;KAC/B,CAAC;IACF,OAAO,IAAI,CAAC,yBAAyB,CAAC;AACvC,CAAC;AAGD,MAAM,UAAU,wCAAwC;IACvD,MAAM,IAAI,GAAG,IAAI,CAAC,yBAAyB,CAAC;IAC5C,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;QACxB,OAAO;IACR,CAAC;IACD,MAAM,iBAAiB,GAAG,IAAI,CAAC,yBAAyB,EAAE,CAAC;IAC3D,IAAI,CAAC,2BAA2B,CAAC;QAChC,QAAQ,EAAE,CAAC,GAAG,IAAI,CAAC,QAAQ,EAAE,GAAG,iBAAiB,CAAC,QAAQ,CAAC;QAC3D,QAAQ,EAAE,CAAC,GAAG,IAAI,CAAC,QAAQ,EAAE,GAAG,iBAAiB,CAAC,QAAQ,CAAC;KAC3D,CAAC,CAAC;IACH,IAAI,CAAC,yBAAyB,GAAG,SAAS,CAAC;AAC5C,CAAC;AAGD,MAAM,UAAU,kBAAkB,CAAqB,OAAqB,EAAE,QAA8B;IAC3G,MAAM,IAAI,GAAG,IAAI,CAAC,yBAAyB,CAAC;IAC5C,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;QACxB,IAAI,QAAQ,KAAK,UAAU,EAAE,CAAC;YAC7B,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC7B,CAAC;aAAM,CAAC;YACP,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC7B,CAAC;QACD,OAAO;IACR,CAAC;IACD,IAAI,QAAQ,KAAK,UAAU,EAAE,CAAC;QAC7B,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;IAC9B,CAAC;SAAM,CAAC;QACP,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;IAC3B,CAAC;AACF,CAAC;AAGD,MAAM,UAAU,yBAAyB;IACxC,+EAA+E;IAC/E,oFAAoF;IACpF,iEAAiE;IACjE,MAAM,eAAe,GAAG,IAAI,CAAC,KAAoC,CAAC;IAClE,OAAO;QACN,QAAQ,EAAE,sBAAsB,CAAC,eAAe,CAAC,aAAa,CAAC;QAC/D,QAAQ,EAAE,sBAAsB,CAAC,eAAe,CAAC,aAAa,CAAC;KAC/D,CAAC;AACH,CAAC;AAGD,MAAM,UAAU,2BAA2B,CAAqB,MAA0B;IACzF,KAAK,MAAM,OAAO,IAAI,MAAM,CAAC,QAAQ,EAAE,CAAC;QACvC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;IAC3B,CAAC;IACD,KAAK,MAAM,OAAO,IAAI,MAAM,CAAC,QAAQ,EAAE,CAAC;QACvC,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;IAC9B,CAAC;AACF,CAAC;AAED;;;;;;GAMG;AAEH,MAAM,UAAU,UAAU;IACzB,MAAM,QAAQ,GAAG,CAAC,GAAG,IAAI,CAAC,iBAAiB,CAAC,CAAC;IAC7C,MAAM,QAAQ,GAAG,CAAC,GAAG,IAAI,CAAC,iBAAiB,CAAC,CAAC;IAC7C,IAAI,CAAC,iBAAiB,GAAG,EAAE,CAAC;IAC5B,IAAI,CAAC,iBAAiB,GAAG,EAAE,CAAC;IAC5B,IAAI,CAAC,KAAK,CAAC,cAAc,EAAE,CAAC;IAC5B,IAAI,IAAI,CAAC,yBAAyB,KAAK,SAAS,EAAE,CAAC;QAClD,IAAI,CAAC,yBAAyB,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC;QACnD,IAAI,CAAC,yBAAyB,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC;IACpD,CAAC;IACD,IAAI,CAAC,gBAAgB,EAAE,CAAC;IACxB,OAAO,EAAE,QAAQ,EAAE,QAAQ,EAAE,CAAC;AAC/B,CAAC;AAED,wEAAwE;AAExE,MAAM,UAAU,mBAAmB;IAClC,OAAO,IAAI,CAAC,iBAAiB,CAAC;AAC/B,CAAC;AAED,iDAAiD;AAEjD,MAAM,UAAU,mBAAmB;IAClC,OAAO,IAAI,CAAC,iBAAiB,CAAC;AAC/B,CAAC;AAGD,MAAM,CAAC,KAAK,UAAU,KAAK;IAC1B,IAAI,CAAC,UAAU,EAAE,CAAC;IAClB,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;IACnB,MAAM,IAAI,CAAC,KAAK,CAAC,WAAW,EAAE,CAAC;AAChC,CAAC;AAED,4EAA4E;AAC5E,mBAAmB;AACnB,4EAA4E;AAG5E,MAAM,UAAU,eAAe,CAAqB,IAA6B;IAChF,IAAI,CAAC,KAAK,CAAC,YAAY,GAAG,IAAI,CAAC;IAC/B,IAAI,CAAC,eAAe,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;AAC5C,CAAC;AAED;;;GAGG;AAEH,MAAM,UAAU,eAAe,CAAqB,IAA6B;IAChF,IAAI,CAAC,KAAK,CAAC,YAAY,GAAG,IAAI,CAAC;IAC/B,IAAI,CAAC,eAAe,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;AAC5C,CAAC;AAED,4EAA4E;AAC5E,8BAA8B;AAC9B,4EAA4E;AAC5E,MAAM,CAAC,MAAM,+BAA+B,GAAG;IAC9C,WAAW;IACX,cAAc;IACd,wBAAwB;IACxB,iCAAiC;IACjC,2BAA2B;IAC3B,iBAAiB;IACjB,kBAAkB;IAClB,4BAA4B;IAC5B,oBAAoB;IACpB,8BAA8B;IAC9B,8BAA8B;IAC9B,+BAA+B;IAC/B,wCAAwC;IACxC,kBAAkB;IAClB,yBAAyB;IACzB,2BAA2B;IAC3B,UAAU;IACV,mBAAmB;IACnB,mBAAmB;IACnB,KAAK;IACL,eAAe;IACf,eAAe;CACf,CAAC","sourcesContent":["import type { AgentMessage } from \"@earendil-works/pi-agent-core\";\nimport type { ImageContent, TextContent } from \"@earendil-works/pi-ai/compat\";\nimport type { CustomMessage } from \"./messages.ts\";\nimport type { SendMessageOptions, SendMessagesOptions } from \"./extensions/index.ts\";\nimport type { AgentSessionInternalSurface as AgentSession } from \"./agent-session-methods.ts\";\nimport { customMessageExcludesContext, drainAgentMessageQueue, normalizeInterruptAbortMessage, type AgentQueueAccess, type DrainedAgentQueues, type InterruptQueueHold } from \"./agent-session-types.ts\";\nimport { createSessionAsyncDeliveryHandler } from \"./async/session-manager.js\";\n\nexport async function _queueSteer(this: AgentSession, text: string, images?: ImageContent[]): Promise<void> {\n\tthis._steeringMessages.push(text);\n\tthis._emitQueueUpdate();\n\tconst content: (TextContent | ImageContent)[] = [{ type: \"text\", text }];\n\tif (images) {\n\t\tcontent.push(...images);\n\t}\n\tthis._queueAgentMessage(\n\t\t{\n\t\t\trole: \"user\",\n\t\t\tcontent,\n\t\t\ttimestamp: Date.now(),\n\t\t},\n\t\t\"steer\",\n\t);\n}\n\n/**\n * Internal: Queue a follow-up message (already expanded, no extension command check).\n */\n\nexport async function _queueFollowUp(this: AgentSession, text: string, images?: ImageContent[]): Promise<void> {\n\tthis._followUpMessages.push(text);\n\tthis._emitQueueUpdate();\n\tconst content: (TextContent | ImageContent)[] = [{ type: \"text\", text }];\n\tif (images) {\n\t\tcontent.push(...images);\n\t}\n\tthis._queueAgentMessage(\n\t\t{\n\t\t\trole: \"user\",\n\t\t\tcontent,\n\t\t\ttimestamp: Date.now(),\n\t\t},\n\t\t\"followUp\",\n\t);\n}\n\n/**\n * Throw an error if the text is an extension command.\n */\n\nexport function _throwIfExtensionCommand(this: AgentSession, text: string): void {\n\tconst spaceIndex = text.indexOf(\" \");\n\tconst commandName = spaceIndex === -1 ? text.slice(1) : text.slice(1, spaceIndex);\n\tconst command = this._extensionRunner.getCommand(commandName);\n\n\tif (command) {\n\t\tthrow new Error(\n\t\t\t`Extension command \"/${commandName}\" cannot be queued. Use prompt() or execute the command when not streaming.`,\n\t\t);\n\t}\n}\n\n/**\n * Send a custom message to the session. Creates a CustomMessageEntry.\n *\n * Handles five cases:\n * - Streaming + interrupt trigger: aborts the active run and starts an immediate custom-message turn\n * - Streaming + explicit display-only context exclusion: appends to state/session, no turn and no queue\n * - Streaming otherwise: queues message, processed when loop pulls from queue\n * - Not streaming + triggerTurn: appends to state/session, starts new turn\n * - Not streaming + no trigger: appends to state/session, no turn\n *\n * @param message Custom message with customType, content, display, details\n * @param options.triggerTurn If true and not streaming, triggers a new LLM turn\n * @param options.deliverAs Delivery mode: \"steer\", \"followUp\", \"nextTurn\", or \"interrupt\"\n */\n\nexport async function sendCustomMessage<T = unknown>(this: AgentSession,\n\tmessage: Pick<CustomMessage<T>, \"customType\" | \"content\" | \"display\" | \"details\">,\n\toptions?: SendMessageOptions,\n): Promise<void> {\n\tconst appMessage = {\n\t\trole: \"custom\" as const,\n\t\tcustomType: message.customType,\n\t\tcontent: message.content ?? [],\n\t\tdisplay: message.display,\n\t\tdetails: message.details,\n\t\ttimestamp: Date.now(),\n\t\t...(options?.excludeFromContext === true ? { excludeFromContext: true } : {}),\n\t} satisfies CustomMessage<T>;\n\tconst boundary = this._workflowStageAdmission;\n\tconst deliver = async (): Promise<void> => {\n\t\tif (boundary && options?.stageAdmissionBarrier) await options.stageAdmissionBarrier();\n\t\tif (options?.deliverAs === \"nextTurn\") {\n\t\t\tthis._pendingNextTurnMessages.push(appMessage);\n\t\t} else if (options?.deliverAs === \"interrupt\" && options.triggerTurn) {\n\t\t\tconst interrupt = this._enqueueInterruptCustomMessage(appMessage, options);\n\t\t\tboundary?.trackAdmittedWork(interrupt);\n\t\t\tvoid interrupt.catch(() => {});\n\t\t} else if (this.isStreaming && options?.excludeFromContext === true && options.triggerTurn !== true && options.deliverAs === undefined) {\n\t\t\tthis._appendCustomMessage(appMessage);\n\t\t} else if (this.isStreaming && options?.persistWhenStreaming === true) {\n\t\t\t// Persist the notice to the transcript so a streaming parent's cleared\n\t\t\t// steer queue or aborted turn cannot silently drop it; it stays visible\n\t\t\t// and in-context for the model's next step.\n\t\t\tthis._appendCustomMessage(appMessage);\n\t\t} else if (this.isStreaming) {\n\t\t\tthis._queueAgentMessage(appMessage, options?.deliverAs === \"followUp\" ? \"followUp\" : \"steer\");\n\t\t} else if (options?.triggerTurn) {\n\t\t\t// ES2022 build target: use a manually constructed deferred instead of\n\t\t\t// Promise.withResolvers (ES2024). Resolve on prompt-start so a later\n\t\t\t// model-turn rejection cannot un-admit an already-delivered notice;\n\t\t\t// reject only when the turn fails before that start signal fires.\n\t\t\tlet resolveAdmission!: () => void;\n\t\t\tlet rejectAdmission!: (error: unknown) => void;\n\t\t\tconst admission = new Promise<void>((resolve, reject) => {\n\t\t\t\tresolveAdmission = resolve;\n\t\t\t\trejectAdmission = reject;\n\t\t\t});\n\t\t\tconst turn = this._runAgentPrompt(appMessage, resolveAdmission);\n\t\t\tvoid turn.then(resolveAdmission, rejectAdmission);\n\t\t\tawait admission;\n\t\t} else {\n\t\t\tthis._appendCustomMessage(appMessage);\n\t\t}\n\t};\n\tif (boundary === undefined) return deliver();\n\tawait boundary.admit(\n\t\toptions?.stageAdmissionKey,\n\t\tdeliver,\n\t\t() => {\n\t\t\tconst router = this._orchestrationContext?.lateMessageRouter;\n\t\t\tif (router === undefined) throw new Error(\"Workflow stage closed without a late-message router\");\n\t\t\treturn router.routeMessage(message, options);\n\t\t},\n\t).completion;\n}\n\n/** Atomically admits a custom-message batch in array order. */\nexport async function sendCustomMessages<T = unknown>(this: AgentSession,\n\tmessages: Array<Pick<CustomMessage<T>, \"customType\" | \"content\" | \"display\" | \"details\">>,\n\toptions?: SendMessagesOptions,\n): Promise<void> {\n\tconst timestamp = Date.now();\n\tconst appMessages = messages.map((message) => ({\n\t\trole: \"custom\" as const,\n\t\tcustomType: message.customType,\n\t\tcontent: message.content ?? [],\n\t\tdisplay: message.display,\n\t\tdetails: message.details,\n\t\ttimestamp,\n\t\t...(options?.excludeFromContext === true ? { excludeFromContext: true } : {}),\n\t} satisfies CustomMessage<T>));\n\tif (appMessages.length === 0) return;\n\tconst boundary = this._workflowStageAdmission;\n\tconst deliver = async (): Promise<void> => {\n\t\tif (boundary && options?.stageAdmissionBarrier) await options.stageAdmissionBarrier();\n\t\tif (options?.deliverAs === \"nextTurn\") {\n\t\t\tthis._pendingNextTurnMessages.push(...appMessages);\n\t\t} else if (this.isStreaming && options?.excludeFromContext === true && options.triggerTurn !== true && options.deliverAs === undefined) {\n\t\t\tfor (const item of appMessages) this._appendCustomMessage(item);\n\t\t} else if (this.isStreaming) {\n\t\t\tconst delivery = options?.deliverAs === \"followUp\" ? \"followUp\" : \"steer\";\n\t\t\tfor (const item of appMessages) this._queueAgentMessage(item, delivery);\n\t\t} else if (options?.triggerTurn) {\n\t\t\tconst turn = this._runAgentPrompt(appMessages);\n\t\t\tvoid turn.catch(() => {});\n\t\t} else {\n\t\t\tfor (const item of appMessages) this._appendCustomMessage(item);\n\t\t}\n\t};\n\tif (boundary === undefined) return deliver();\n\tawait boundary.admit(\n\t\toptions?.stageAdmissionKey,\n\t\tdeliver,\n\t\t() => {\n\t\t\tconst router = this._orchestrationContext?.lateMessageRouter;\n\t\t\tif (router === undefined) throw new Error(\"Workflow stage closed without a late-message router\");\n\t\t\treturn router.routeMessages(messages, options);\n\t\t},\n\t).completion;\n}\n\n\nexport function sealWorkflowStageGeneration(this: AgentSession): void {\n\tthis._workflowStageAdmission?.seal();\n}\n\nexport async function closeWorkflowStageGeneration(this: AgentSession): Promise<void> {\n\tawait this._workflowStageAdmission?.close();\n\tawait this.agent?.waitForIdle?.();\n\tawait this._agentEventQueue;\n}\n\nexport function transferWorkflowStageDeliveriesTo(this: AgentSession, target: object): void {\n\tconst next = target as AgentSession;\n\tconst queued = this._drainQueuedAgentMessages();\n\tif (this._activeInterruptQueueHold) {\n\t\tqueued.steering.push(...this._activeInterruptQueueHold.steering);\n\t\tqueued.followUp.push(...this._activeInterruptQueueHold.followUp);\n\t\tthis._activeInterruptQueueHold = undefined;\n\t}\n\tnext._pendingNextTurnMessages.unshift(...this._pendingNextTurnMessages.splice(0));\n\tnext._restoreQueuedAgentMessages(queued);\n\tthis._asyncJobManager.transferSessionDeliveries(\n\t\tthis._asyncJobManagerSessionId,\n\t\tnext._asyncJobManagerSessionId,\n\t\tcreateSessionAsyncDeliveryHandler(next, next._asyncJobManager, next._asyncJobManagerSessionId),\n\t);\n}\nexport function _appendCustomMessage<T>(this: AgentSession, message: CustomMessage<T>): void {\n\tthis.agent.state.messages.push(message);\n\tthis.sessionManager.appendCustomMessageEntry(\n\t\tmessage.customType,\n\t\tmessage.content,\n\t\tmessage.display,\n\t\tmessage.details,\n\t\tcustomMessageExcludesContext(message),\n\t);\n\tthis._emit({ type: \"message_start\", message });\n\tthis._emit({ type: \"message_end\", message });\n}\n\n\nexport function _enqueueInterruptCustomMessage<T>(this: AgentSession, message: CustomMessage<T>, options?: SendMessageOptions): Promise<void> {\n\tthis._pendingInterruptDeliveries += 1;\n\t// Establish the hold synchronously when the interrupt is enqueued, not when\n\t// the serialized delivery callback later starts. Callers commonly fire and\n\t// forget sendCustomMessage(), then queue additional steer/follow-up messages\n\t// before the promise chain gets a microtask; those messages must be captured\n\t// in the active interrupt hold instead of pi-agent-core's live queues.\n\tthis._ensureActiveInterruptQueueHold();\n\tconst delivery = this._interruptDeliveryQueue.then(async () => {\n\t\ttry {\n\t\t\tawait this._sendInterruptCustomMessageNow(message, options);\n\t\t} finally {\n\t\t\tthis._pendingInterruptDeliveries -= 1;\n\t\t\tif (this._pendingInterruptDeliveries === 0) {\n\t\t\t\tthis._restoreAndClearActiveInterruptQueueHold();\n\t\t\t}\n\t\t}\n\t});\n\tthis._interruptDeliveryQueue = delivery.catch(() => undefined);\n\treturn delivery;\n}\n\n\nexport async function _sendInterruptCustomMessageNow<T>(this: AgentSession, \n\tmessage: CustomMessage<T>,\n\toptions?: SendMessageOptions,\n): Promise<void> {\n\tthis.abortRetry();\n\tthis._ensureActiveInterruptQueueHold();\n\tif (this.isStreaming) {\n\t\tconst previousAbortMessage = this._activeInterruptAbortMessage;\n\t\tthis._activeInterruptAbortMessage = normalizeInterruptAbortMessage(options?.interruptAbortMessage);\n\t\ttry {\n\t\t\tthis.agent.abort();\n\t\t\tawait this.agent.waitForIdle();\n\t\t\tawait this._agentEventQueue;\n\t\t} finally {\n\t\t\tthis._activeInterruptAbortMessage = previousAbortMessage;\n\t\t}\n\t}\n\tawait this.agent.prompt(message);\n}\n\n\nexport function _ensureActiveInterruptQueueHold(this: AgentSession): InterruptQueueHold {\n\tif (this._activeInterruptQueueHold !== undefined) {\n\t\treturn this._activeInterruptQueueHold;\n\t}\n\tconst drained = this._drainQueuedAgentMessages();\n\tthis._activeInterruptQueueHold = {\n\t\tsteering: [...drained.steering],\n\t\tfollowUp: [...drained.followUp],\n\t};\n\treturn this._activeInterruptQueueHold;\n}\n\n\nexport function _restoreAndClearActiveInterruptQueueHold(this: AgentSession): void {\n\tconst hold = this._activeInterruptQueueHold;\n\tif (hold === undefined) {\n\t\treturn;\n\t}\n\tconst currentCoreQueues = this._drainQueuedAgentMessages();\n\tthis._restoreQueuedAgentMessages({\n\t\tsteering: [...hold.steering, ...currentCoreQueues.steering],\n\t\tfollowUp: [...hold.followUp, ...currentCoreQueues.followUp],\n\t});\n\tthis._activeInterruptQueueHold = undefined;\n}\n\n\nexport function _queueAgentMessage(this: AgentSession, message: AgentMessage, delivery: \"steer\" | \"followUp\"): void {\n\tconst hold = this._activeInterruptQueueHold;\n\tif (hold !== undefined) {\n\t\tif (delivery === \"followUp\") {\n\t\t\thold.followUp.push(message);\n\t\t} else {\n\t\t\thold.steering.push(message);\n\t\t}\n\t\treturn;\n\t}\n\tif (delivery === \"followUp\") {\n\t\tthis.agent.followUp(message);\n\t} else {\n\t\tthis.agent.steer(message);\n\t}\n}\n\n\nexport function _drainQueuedAgentMessages(this: AgentSession): DrainedAgentQueues {\n\t// pi-agent-core exposes public clear methods but no public drain/restore pair.\n\t// Interrupts need to prevent the aborting run from consuming queued steer/follow-up\n\t// messages while still preserving those queues for a later turn.\n\tconst agentWithQueues = this.agent as unknown as AgentQueueAccess;\n\treturn {\n\t\tsteering: drainAgentMessageQueue(agentWithQueues.steeringQueue),\n\t\tfollowUp: drainAgentMessageQueue(agentWithQueues.followUpQueue),\n\t};\n}\n\n\nexport function _restoreQueuedAgentMessages(this: AgentSession, queues: DrainedAgentQueues): void {\n\tfor (const message of queues.steering) {\n\t\tthis.agent.steer(message);\n\t}\n\tfor (const message of queues.followUp) {\n\t\tthis.agent.followUp(message);\n\t}\n}\n\n/**\n * Send a user message to the agent. Always triggers a turn.\n * When the agent is streaming, use deliverAs to specify how to queue the message.\n *\n * @param content User message content (string or content array)\n * @param options.deliverAs Delivery mode when streaming: \"steer\" or \"followUp\"\n */\n\nexport function clearQueue(this: AgentSession): { steering: string[]; followUp: string[] } {\n\tconst steering = [...this._steeringMessages];\n\tconst followUp = [...this._followUpMessages];\n\tthis._steeringMessages = [];\n\tthis._followUpMessages = [];\n\tthis.agent.clearAllQueues();\n\tif (this._activeInterruptQueueHold !== undefined) {\n\t\tthis._activeInterruptQueueHold.steering.length = 0;\n\t\tthis._activeInterruptQueueHold.followUp.length = 0;\n\t}\n\tthis._emitQueueUpdate();\n\treturn { steering, followUp };\n}\n\n/** Number of pending messages (includes both steering and follow-up) */\n\nexport function getSteeringMessages(this: AgentSession): readonly string[] {\n\treturn this._steeringMessages;\n}\n\n/** Get pending follow-up messages (read-only) */\n\nexport function getFollowUpMessages(this: AgentSession): readonly string[] {\n\treturn this._followUpMessages;\n}\n\n\nexport async function abort(this: AgentSession): Promise<void> {\n\tthis.abortRetry();\n\tthis.agent.abort();\n\tawait this.agent.waitForIdle();\n}\n\n// =========================================================================\n// Model Management\n// =========================================================================\n\n\nexport function setSteeringMode(this: AgentSession, mode: \"all\" | \"one-at-a-time\"): void {\n\tthis.agent.steeringMode = mode;\n\tthis.settingsManager.setSteeringMode(mode);\n}\n\n/**\n * Set follow-up message mode.\n * Saves to settings.\n */\n\nexport function setFollowUpMode(this: AgentSession, mode: \"all\" | \"one-at-a-time\"): void {\n\tthis.agent.followUpMode = mode;\n\tthis.settingsManager.setFollowUpMode(mode);\n}\n\n// =========================================================================\n// Queue and delivery settings\n// =========================================================================\nexport const agentSessionMessageQueueMethods = {\n\t_queueSteer,\n\t_queueFollowUp,\n\t_throwIfExtensionCommand,\n\ttransferWorkflowStageDeliveriesTo,\n\tsealWorkflowStageGeneration,\n\tsendCustomMessage,\n\tsendCustomMessages,\n\tcloseWorkflowStageGeneration,\n\t_appendCustomMessage,\n\t_enqueueInterruptCustomMessage,\n\t_sendInterruptCustomMessageNow,\n\t_ensureActiveInterruptQueueHold,\n\t_restoreAndClearActiveInterruptQueueHold,\n\t_queueAgentMessage,\n\t_drainQueuedAgentMessages,\n\t_restoreQueuedAgentMessages,\n\tclearQueue,\n\tgetSteeringMessages,\n\tgetFollowUpMessages,\n\tabort,\n\tsetSteeringMode,\n\tsetFollowUpMode,\n};\n"]}
+{"version":3,"file":"agent-session-message-queue.js","sourceRoot":"","sources":["../../src/core/agent-session-message-queue.ts"],"names":[],"mappings":"AAKA,OAAO,EAAE,4BAA4B,EAAE,sBAAsB,EAAE,8BAA8B,EAA2E,MAAM,0BAA0B,CAAC;AACzM,OAAO,EAAE,uBAAuB,EAAE,kCAAkC,EAAE,MAAM,wCAAwC,CAAC;AACrH,OAAO,EAAE,iCAAiC,EAAE,MAAM,6BAA6B,CAAC;AAChF,OAAO,EAAE,KAAK,EAAE,mBAAmB,EAAE,oBAAoB,EAAE,MAAM,gCAAgC,CAAC;AAClG,OAAO,EAAE,uCAAuC,EAAE,MAAM,+CAA+C,CAAC;AACxG,OAAO,EAAE,2BAA2B,EAAE,4BAA4B,EAAE,MAAM,0CAA0C,CAAC;AACrH,OAAO,EAAE,iCAAiC,EAAE,CAAC;AAE7C,MAAM,uBAAuB,GAAG,IAAI,OAAO,EAA+B,CAAC;AAE3E,SAAS,0BAA0B,CAAC,KAAmB,EAAE,SAA8B;IACtF,MAAM,QAAQ,GAAG,uBAAuB,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;IACzE,MAAM,QAAQ,GAAG,QAAQ,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IAC1C,uBAAuB,CAAC,GAAG,CAAC,KAAK,EAAE,QAAQ,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC,CAAC;IACpE,OAAO,QAAQ,CAAC;AACjB,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,WAAW,CAAqB,IAAY,EAAE,MAAuB;IAC1F,MAAM,KAAK,GAAG,kCAAkC,CAAC,IAAI,CAAC,CAAC;IACvD,IAAI,KAAK,KAAK,IAAI;QAAE,OAAO,KAAK,CAAC,WAAW,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;IAC3D,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAClC,IAAI,CAAC,gBAAgB,EAAE,CAAC;IACxB,MAAM,OAAO,GAAmC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;IACzE,IAAI,MAAM,EAAE,CAAC;QACZ,OAAO,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,CAAC;IACzB,CAAC;IACD,IAAI,CAAC,kBAAkB,CACtB;QACC,IAAI,EAAE,MAAM;QACZ,OAAO;QACP,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;KACrB,EACD,OAAO,CACP,CAAC;AACH,CAAC;AAED;;GAEG;AAEH,MAAM,CAAC,KAAK,UAAU,cAAc,CAAqB,IAAY,EAAE,MAAuB;IAC7F,MAAM,KAAK,GAAG,kCAAkC,CAAC,IAAI,CAAC,CAAC;IACvD,IAAI,KAAK,KAAK,IAAI;QAAE,OAAO,KAAK,CAAC,cAAc,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;IAC9D,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAClC,IAAI,CAAC,gBAAgB,EAAE,CAAC;IACxB,MAAM,OAAO,GAAmC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;IACzE,IAAI,MAAM,EAAE,CAAC;QACZ,OAAO,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,CAAC;IACzB,CAAC;IACD,IAAI,CAAC,kBAAkB,CACtB;QACC,IAAI,EAAE,MAAM;QACZ,OAAO;QACP,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;KACrB,EACD,UAAU,CACV,CAAC;AACH,CAAC;AAED;;GAEG;AAEH,MAAM,UAAU,wBAAwB,CAAqB,IAAY;IACxE,MAAM,UAAU,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;IACrC,MAAM,WAAW,GAAG,UAAU,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,UAAU,CAAC,CAAC;IAClF,MAAM,OAAO,GAAG,IAAI,CAAC,gBAAgB,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC;IAE9D,IAAI,OAAO,EAAE,CAAC;QACb,MAAM,IAAI,KAAK,CACd,uBAAuB,WAAW,6EAA6E,CAC/G,CAAC;IACH,CAAC;AACF,CAAC;AAED;;;;;;;;;;;;;GAaG;AAEH,MAAM,CAAC,KAAK,UAAU,iBAAiB,CACtC,OAAiF,EACjF,OAA4B;IAE5B,MAAM,YAAY,GAAG,kCAAkC,CAAC,IAAI,CAAC,CAAC;IAC9D,IAAI,YAAY,KAAK,IAAI;QAAE,OAAO,YAAY,CAAC,iBAAiB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;IACnF,MAAM,UAAU,GAAG;QAClB,IAAI,EAAE,QAAiB;QACvB,UAAU,EAAE,OAAO,CAAC,UAAU;QAC9B,OAAO,EAAE,OAAO,CAAC,OAAO,IAAI,EAAE;QAC9B,OAAO,EAAE,OAAO,CAAC,OAAO;QACxB,OAAO,EAAE,OAAO,CAAC,OAAO;QACxB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;QACrB,GAAG,CAAC,OAAO,EAAE,kBAAkB,KAAK,IAAI,CAAC,CAAC,CAAC,EAAE,kBAAkB,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KAClD,CAAC;IAC7B,MAAM,QAAQ,GAAG,IAAI,CAAC,uBAAuB,CAAC;IAC9C,MAAM,OAAO,GAAG,KAAK,IAAmB,EAAE;QACzC,IAAI,QAAQ,IAAI,OAAO,EAAE,qBAAqB;YAAE,MAAM,OAAO,CAAC,qBAAqB,EAAE,CAAC;QACtF,MAAM,2BAA2B,CAAC,IAAI,EAAE,UAAU,EAAE,OAAO,CAAC,CAAC;IAC9D,CAAC,CAAC;IACF,IAAI,QAAQ,KAAK,SAAS;QAAE,OAAO,OAAO,EAAE,CAAC;IAC7C,MAAM,QAAQ,CAAC,KAAK,CACnB,OAAO,EAAE,iBAAiB,EAC1B,OAAO,EACP,GAAG,EAAE;QACJ,MAAM,MAAM,GAAG,IAAI,CAAC,qBAAqB,EAAE,iBAAiB,CAAC;QAC7D,IAAI,MAAM,KAAK,SAAS;YAAE,MAAM,IAAI,KAAK,CAAC,qDAAqD,CAAC,CAAC;QACjG,OAAO,MAAM,CAAC,YAAY,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;IAC9C,CAAC,CACD,CAAC,UAAU,CAAC;AACd,CAAC;AAED,+DAA+D;AAC/D,MAAM,CAAC,KAAK,UAAU,kBAAkB,CACvC,QAAyF,EACzF,OAA6B;IAE7B,MAAM,YAAY,GAAG,kCAAkC,CAAC,IAAI,CAAC,CAAC;IAC9D,IAAI,YAAY,KAAK,IAAI;QAAE,OAAO,YAAY,CAAC,kBAAkB,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;IACrF,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAC7B,MAAM,WAAW,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;QAC9C,IAAI,EAAE,QAAiB;QACvB,UAAU,EAAE,OAAO,CAAC,UAAU;QAC9B,OAAO,EAAE,OAAO,CAAC,OAAO,IAAI,EAAE;QAC9B,OAAO,EAAE,OAAO,CAAC,OAAO;QACxB,OAAO,EAAE,OAAO,CAAC,OAAO;QACxB,SAAS;QACT,GAAG,CAAC,OAAO,EAAE,kBAAkB,KAAK,IAAI,CAAC,CAAC,CAAC,EAAE,kBAAkB,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KACjD,CAAA,CAAC,CAAC;IAC/B,IAAI,WAAW,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO;IACrC,MAAM,QAAQ,GAAG,IAAI,CAAC,uBAAuB,CAAC;IAC9C,MAAM,OAAO,GAAG,KAAK,IAAmB,EAAE;QACzC,IAAI,QAAQ,IAAI,OAAO,EAAE,qBAAqB;YAAE,MAAM,OAAO,CAAC,qBAAqB,EAAE,CAAC;QACtF,MAAM,4BAA4B,CAAC,IAAI,EAAE,WAAW,EAAE,OAAO,CAAC,CAAC;IAChE,CAAC,CAAC;IACF,IAAI,QAAQ,KAAK,SAAS;QAAE,OAAO,OAAO,EAAE,CAAC;IAC7C,MAAM,QAAQ,CAAC,KAAK,CACnB,OAAO,EAAE,iBAAiB,EAC1B,OAAO,EACP,GAAG,EAAE;QACJ,MAAM,MAAM,GAAG,IAAI,CAAC,qBAAqB,EAAE,iBAAiB,CAAC;QAC7D,IAAI,MAAM,KAAK,SAAS;YAAE,MAAM,IAAI,KAAK,CAAC,qDAAqD,CAAC,CAAC;QACjG,OAAO,MAAM,CAAC,aAAa,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;IAChD,CAAC,CACD,CAAC,UAAU,CAAC;AACd,CAAC;AAED,MAAM,UAAU,2BAA2B;IAC1C,IAAI,CAAC,uBAAuB,EAAE,IAAI,EAAE,CAAC;AACtC,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,4BAA4B;IACjD,MAAM,IAAI,CAAC,uBAAuB,EAAE,KAAK,EAAE,CAAC;IAC5C,MAAM,IAAI,CAAC,KAAK,EAAE,WAAW,EAAE,EAAE,CAAC;IAClC,MAAM,IAAI,CAAC,gBAAgB,CAAC;AAC7B,CAAC;AAED,MAAM,UAAU,oBAAoB,CAAwB,OAAyB;IACpF,MAAM,KAAK,GAAG,kCAAkC,CAAC,IAAI,CAAC,CAAC;IACvD,IAAI,KAAK,KAAK,IAAI;QAAE,OAAO,KAAK,CAAC,oBAAoB,CAAC,OAAO,CAAC,CAAC;IAC/D,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACxC,IAAI,CAAC,cAAc,CAAC,wBAAwB,CAC3C,OAAO,CAAC,UAAU,EAClB,OAAO,CAAC,OAAO,EACf,OAAO,CAAC,OAAO,EACf,OAAO,CAAC,OAAO,EACf,4BAA4B,CAAC,OAAO,CAAC,CACrC,CAAC;IACF,IAAI,CAAC,KAAK,CAAC,EAAE,IAAI,EAAE,eAAe,EAAE,OAAO,EAAE,CAAC,CAAC;IAC/C,IAAI,CAAC,KAAK,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,OAAO,EAAE,CAAC,CAAC;AAC9C,CAAC;AAED,MAAM,UAAU,8BAA8B,CAAwB,OAAyB,EAAE,OAA4B;IAC5H,MAAM,KAAK,GAAG,kCAAkC,CAAC,IAAI,CAAC,CAAC;IACvD,IAAI,KAAK,KAAK,IAAI;QAAE,OAAO,KAAK,CAAC,8BAA8B,CAAC,OAAO,EAAE,uBAAuB,CAAC,OAAO,CAAC,CAAC,CAAC;IAC3G,IAAI,CAAC,2BAA2B,IAAI,CAAC,CAAC;IACtC,4EAA4E;IAC5E,2EAA2E;IAC3E,6EAA6E;IAC7E,6EAA6E;IAC7E,uEAAuE;IACvE,IAAI,CAAC,+BAA+B,EAAE,CAAC;IACvC,MAAM,QAAQ,GAAG,IAAI,CAAC,uBAAuB,CAAC,IAAI,CAAC,KAAK,IAAI,EAAE;QAC7D,IAAI,CAAC;YACJ,MAAM,IAAI,CAAC,8BAA8B,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;QAC7D,CAAC;gBAAS,CAAC;YACV,wEAAwE;YACxE,0EAA0E;YAC1E,MAAM,SAAS,GAAG,kCAAkC,CAAC,IAAI,CAAC,CAAC;YAC3D,SAAS,CAAC,2BAA2B,IAAI,CAAC,CAAC;YAC3C,IAAI,SAAS,CAAC,2BAA2B,KAAK,CAAC,EAAE,CAAC;gBACjD,SAAS,CAAC,wCAAwC,EAAE,CAAC;YACtD,CAAC;QACF,CAAC;IACF,CAAC,CAAC,CAAC;IACH,IAAI,CAAC,uBAAuB,GAAG,QAAQ,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;IAC/D,OAAO,QAAQ,CAAC;AACjB,CAAC;AAGD,KAAK,UAAU,kCAAkC,CAChD,OAAqB,EACrB,OAAyB,EACzB,OAA4B;IAE5B,OAAO,CAAC,UAAU,EAAE,CAAC;IACrB,OAAO,CAAC,+BAA+B,EAAE,CAAC;IAC1C,IAAI,OAAO,CAAC,WAAW,EAAE,CAAC;QACzB,MAAM,oBAAoB,GAAG,OAAO,CAAC,4BAA4B,CAAC;QAClE,OAAO,CAAC,4BAA4B,GAAG,8BAA8B,CAAC,OAAO,EAAE,qBAAqB,CAAC,CAAC;QACtG,IAAI,CAAC;YACJ,OAAO,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;YACtB,MAAM,OAAO,CAAC,KAAK,CAAC,WAAW,EAAE,CAAC;YAClC,MAAM,OAAO,CAAC,gBAAgB,CAAC;QAChC,CAAC;gBAAS,CAAC;YACV,OAAO,CAAC,4BAA4B,GAAG,oBAAoB,CAAC;QAC7D,CAAC;IACF,CAAC;IACD,MAAM,KAAK,GAAG,kCAAkC,CAAC,OAAO,CAAC,CAAC;IAC1D,IAAI,KAAK,KAAK,OAAO;QAAE,OAAO,KAAK,CAAC,8BAA8B,CAAC,OAAO,EAAE,uBAAuB,CAAC,OAAO,CAAC,CAAC,CAAC;IAC9G,IAAI,OAAO,CAAC,qBAAqB,EAAE,CAAC;QACnC,OAAO,CAAC,kBAAkB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;QAC7C,OAAO;IACR,CAAC;IACD,MAAM,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;AACrC,CAAC;AAED,MAAM,UAAU,8BAA8B,CAC7C,OAAyB,EACzB,OAA4B;IAE5B,MAAM,KAAK,GAAG,kCAAkC,CAAC,IAAI,CAAC,CAAC;IACvD,IAAI,KAAK,KAAK,IAAI;QAAE,OAAO,KAAK,CAAC,8BAA8B,CAAC,OAAO,EAAE,uBAAuB,CAAC,OAAO,CAAC,CAAC,CAAC;IAC3G,OAAO,0BAA0B,CAAC,IAAI,EAAE,GAAG,EAAE,CAAC,kCAAkC,CAAC,IAAI,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC;AAC3G,CAAC;AAGD,MAAM,UAAU,+BAA+B;IAC9C,MAAM,KAAK,GAAG,kCAAkC,CAAC,IAAI,CAAC,CAAC;IACvD,IAAI,KAAK,KAAK,IAAI;QAAE,OAAO,KAAK,CAAC,+BAA+B,EAAE,CAAC;IACnE,IAAI,IAAI,CAAC,yBAAyB,KAAK,SAAS,EAAE,CAAC;QAClD,OAAO,IAAI,CAAC,yBAAyB,CAAC;IACvC,CAAC;IACD,MAAM,OAAO,GAAG,IAAI,CAAC,yBAAyB,EAAE,CAAC;IACjD,IAAI,CAAC,yBAAyB,GAAG;QAChC,QAAQ,EAAE,CAAC,GAAG,OAAO,CAAC,QAAQ,CAAC;QAC/B,QAAQ,EAAE,CAAC,GAAG,OAAO,CAAC,QAAQ,CAAC;KAC/B,CAAC;IACF,OAAO,IAAI,CAAC,yBAAyB,CAAC;AACvC,CAAC;AAGD,MAAM,UAAU,wCAAwC;IACvD,MAAM,KAAK,GAAG,kCAAkC,CAAC,IAAI,CAAC,CAAC;IACvD,IAAI,KAAK,KAAK,IAAI;QAAE,OAAO,KAAK,CAAC,wCAAwC,EAAE,CAAC;IAC5E,IAAI,IAAI,CAAC,qBAAqB;QAAE,OAAO;IACvC,MAAM,IAAI,GAAG,IAAI,CAAC,yBAAyB,CAAC;IAC5C,IAAI,IAAI,KAAK,SAAS;QAAE,OAAO;IAC/B,MAAM,iBAAiB,GAAG,IAAI,CAAC,yBAAyB,EAAE,CAAC;IAC3D,IAAI,CAAC,2BAA2B,CAAC;QAChC,QAAQ,EAAE,CAAC,GAAG,IAAI,CAAC,QAAQ,EAAE,GAAG,iBAAiB,CAAC,QAAQ,CAAC;QAC3D,QAAQ,EAAE,CAAC,GAAG,IAAI,CAAC,QAAQ,EAAE,GAAG,iBAAiB,CAAC,QAAQ,CAAC;KAC3D,CAAC,CAAC;IACH,IAAI,CAAC,yBAAyB,GAAG,SAAS,CAAC;AAC5C,CAAC;AAGD,MAAM,UAAU,kBAAkB,CAAqB,OAAqB,EAAE,QAA8B;IAC3G,MAAM,KAAK,GAAG,kCAAkC,CAAC,IAAI,CAAC,CAAC;IACvD,IAAI,KAAK,KAAK,IAAI;QAAE,OAAO,KAAK,CAAC,kBAAkB,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;IACvE,MAAM,IAAI,GAAG,IAAI,CAAC,yBAAyB,CAAC;IAC5C,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;QACxB,IAAI,QAAQ,KAAK,UAAU,EAAE,CAAC;YAC7B,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC7B,CAAC;aAAM,CAAC;YACP,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC7B,CAAC;QACD,OAAO;IACR,CAAC;IACD,IAAI,QAAQ,KAAK,UAAU,EAAE,CAAC;QAC7B,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;IAC9B,CAAC;SAAM,CAAC;QACP,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;IAC3B,CAAC;AACF,CAAC;AAGD,MAAM,UAAU,yBAAyB;IACxC,+EAA+E;IAC/E,8EAA8E;IAC9E,0EAA0E;IAC1E,MAAM,eAAe,GAAG,IAAI,CAAC,KAAoC,CAAC;IAClE,OAAO;QACN,QAAQ,EAAE,sBAAsB,CAAC,eAAe,CAAC,aAAa,CAAC;QAC/D,QAAQ,EAAE,sBAAsB,CAAC,eAAe,CAAC,aAAa,CAAC;KAC/D,CAAC;AACH,CAAC;AAGD,MAAM,UAAU,2BAA2B,CAAqB,MAA0B;IACzF,KAAK,MAAM,OAAO,IAAI,MAAM,CAAC,QAAQ,EAAE,CAAC;QACvC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;IAC3B,CAAC;IACD,KAAK,MAAM,OAAO,IAAI,MAAM,CAAC,QAAQ,EAAE,CAAC;QACvC,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;IAC9B,CAAC;AACF,CAAC;AACD;;;;;;GAMG;AACH,MAAM,UAAU,UAAU;IACzB,MAAM,KAAK,GAAG,kCAAkC,CAAC,IAAI,CAAC,CAAC;IACvD,IAAI,KAAK,KAAK,IAAI;QAAE,OAAO,KAAK,CAAC,UAAU,EAAE,CAAC;IAC9C,MAAM,QAAQ,GAAG,CAAC,GAAG,IAAI,CAAC,iBAAiB,CAAC,CAAC;IAC7C,MAAM,QAAQ,GAAG,CAAC,GAAG,IAAI,CAAC,iBAAiB,CAAC,CAAC;IAC7C,IAAI,CAAC,iBAAiB,GAAG,EAAE,CAAC;IAC5B,IAAI,CAAC,iBAAiB,GAAG,EAAE,CAAC;IAC5B,MAAM,OAAO,GAAG,IAAI,CAAC,yBAAyB,EAAE,CAAC;IACjD,IAAI,IAAI,CAAC,yBAAyB,KAAK,SAAS,EAAE,CAAC;QAClD,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,yBAAyB,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;QAC5E,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,yBAAyB,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;IAC7E,CAAC;IACD,uCAAuC,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;IACvD,MAAM,IAAI,GAAG,IAAI,CAAC,yBAAyB,CAAC;IAC5C,IAAI,CAAC,IAAI,CAAC,qBAAqB,IAAI,IAAI,CAAC,2BAA2B,KAAK,CAAC,IAAI,IAAI,KAAK,SAAS;WAC3F,IAAI,CAAC,QAAQ,CAAC,MAAM,KAAK,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC9D,IAAI,CAAC,yBAAyB,GAAG,SAAS,CAAC;IAC5C,CAAC;IACD,IAAI,CAAC,gBAAgB,EAAE,CAAC;IACxB,OAAO,EAAE,QAAQ,EAAE,QAAQ,EAAE,CAAC;AAC/B,CAAC;AAED,wEAAwE;AACxE,MAAM,UAAU,mBAAmB;IAClC,OAAO,IAAI,CAAC,iBAAiB,CAAC;AAC/B,CAAC;AAED,iDAAiD;AACjD,MAAM,UAAU,mBAAmB;IAClC,OAAO,IAAI,CAAC,iBAAiB,CAAC;AAC/B,CAAC;AAGD,4EAA4E;AAC5E,mBAAmB;AACnB,4EAA4E;AAG5E,MAAM,UAAU,eAAe,CAAqB,IAA6B;IAChF,IAAI,CAAC,KAAK,CAAC,YAAY,GAAG,IAAI,CAAC;IAC/B,IAAI,CAAC,eAAe,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;AAC5C,CAAC;AAED;;;GAGG;AAEH,MAAM,UAAU,eAAe,CAAqB,IAA6B;IAChF,IAAI,CAAC,KAAK,CAAC,YAAY,GAAG,IAAI,CAAC;IAC/B,IAAI,CAAC,eAAe,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;AAC5C,CAAC;AAED,4EAA4E;AAC5E,8BAA8B;AAC9B,4EAA4E;AAC5E,MAAM,CAAC,MAAM,+BAA+B,GAAG;IAC9C,WAAW;IACX,cAAc;IACd,wBAAwB;IACxB,iCAAiC;IACjC,2BAA2B;IAC3B,iBAAiB;IACjB,kBAAkB;IAClB,4BAA4B;IAC5B,oBAAoB;IACpB,8BAA8B;IAC9B,8BAA8B;IAC9B,+BAA+B;IAC/B,wCAAwC;IACxC,mBAAmB;IACnB,oBAAoB;IACpB,kBAAkB;IAClB,yBAAyB;IACzB,2BAA2B;IAC3B,UAAU;IACV,mBAAmB;IACnB,mBAAmB;IACnB,KAAK;IACL,eAAe;IACf,eAAe;CACf,CAAC","sourcesContent":["import type { AgentMessage } from \"@earendil-works/pi-agent-core\";\nimport type { ImageContent, TextContent } from \"@earendil-works/pi-ai/compat\";\nimport type { CustomMessage } from \"./messages.ts\";\nimport type { SendMessageOptions, SendMessagesOptions } from \"./extensions/index.ts\";\nimport type { AgentSessionInternalSurface as AgentSession } from \"./agent-session-methods.ts\";\nimport { customMessageExcludesContext, drainAgentMessageQueue, normalizeInterruptAbortMessage, type AgentQueueAccess, type DrainedAgentQueues, type InterruptQueueHold } from \"./agent-session-types.ts\";\nimport { forwardedMessageOptions, resolveWorkflowStageDeliveryTarget } from \"./agent-session-delivery-forwarding.ts\";\nimport { transferWorkflowStageDeliveriesTo } from \"./agent-session-transfer.ts\";\nimport { abort, pauseQueuedMessages, resumeQueuedMessages } from \"./agent-session-queue-pause.ts\";\nimport { restoreProtectedStreamingCustomMessages } from \"./agent-session-persistent-custom-messages.ts\";\nimport { commitAdmittedCustomMessage, commitAdmittedCustomMessages } from \"./agent-session-custom-message-commit.ts\";\nexport { transferWorkflowStageDeliveriesTo };\n\nconst interruptMutationQueues = new WeakMap<AgentSession, Promise<void>>();\n\nfunction serializeInterruptMutation(owner: AgentSession, operation: () => Promise<void>): Promise<void> {\n\tconst previous = interruptMutationQueues.get(owner) ?? Promise.resolve();\n\tconst delivery = previous.then(operation);\n\tinterruptMutationQueues.set(owner, delivery.catch(() => undefined));\n\treturn delivery;\n}\n\nexport async function _queueSteer(this: AgentSession, text: string, images?: ImageContent[]): Promise<void> {\n\tconst owner = resolveWorkflowStageDeliveryTarget(this);\n\tif (owner !== this) return owner._queueSteer(text, images);\n\tthis._steeringMessages.push(text);\n\tthis._emitQueueUpdate();\n\tconst content: (TextContent | ImageContent)[] = [{ type: \"text\", text }];\n\tif (images) {\n\t\tcontent.push(...images);\n\t}\n\tthis._queueAgentMessage(\n\t\t{\n\t\t\trole: \"user\",\n\t\t\tcontent,\n\t\t\ttimestamp: Date.now(),\n\t\t},\n\t\t\"steer\",\n\t);\n}\n\n/**\n * Internal: Queue a follow-up message (already expanded, no extension command check).\n */\n\nexport async function _queueFollowUp(this: AgentSession, text: string, images?: ImageContent[]): Promise<void> {\n\tconst owner = resolveWorkflowStageDeliveryTarget(this);\n\tif (owner !== this) return owner._queueFollowUp(text, images);\n\tthis._followUpMessages.push(text);\n\tthis._emitQueueUpdate();\n\tconst content: (TextContent | ImageContent)[] = [{ type: \"text\", text }];\n\tif (images) {\n\t\tcontent.push(...images);\n\t}\n\tthis._queueAgentMessage(\n\t\t{\n\t\t\trole: \"user\",\n\t\t\tcontent,\n\t\t\ttimestamp: Date.now(),\n\t\t},\n\t\t\"followUp\",\n\t);\n}\n\n/**\n * Throw an error if the text is an extension command.\n */\n\nexport function _throwIfExtensionCommand(this: AgentSession, text: string): void {\n\tconst spaceIndex = text.indexOf(\" \");\n\tconst commandName = spaceIndex === -1 ? text.slice(1) : text.slice(1, spaceIndex);\n\tconst command = this._extensionRunner.getCommand(commandName);\n\n\tif (command) {\n\t\tthrow new Error(\n\t\t\t`Extension command \"/${commandName}\" cannot be queued. Use prompt() or execute the command when not streaming.`,\n\t\t);\n\t}\n}\n\n/**\n * Send a custom message to the session. Creates a CustomMessageEntry.\n *\n * Handles five cases:\n * - Streaming + interrupt trigger: aborts the active run and starts an immediate custom-message turn\n * - Streaming + explicit display-only context exclusion: appends to state/session, no turn and no queue\n * - Streaming otherwise: queues message, processed when loop pulls from queue\n * - Not streaming + triggerTurn: appends to state/session, starts new turn\n * - Not streaming + no trigger: appends to state/session, no turn\n *\n * @param message Custom message with customType, content, display, details\n * @param options.triggerTurn If true and not streaming, triggers a new LLM turn\n * @param options.deliverAs Delivery mode: \"steer\", \"followUp\", \"nextTurn\", or \"interrupt\"\n */\n\nexport async function sendCustomMessage<T = unknown>(this: AgentSession,\n\tmessage: Pick<CustomMessage<T>, \"customType\" | \"content\" | \"display\" | \"details\">,\n\toptions?: SendMessageOptions,\n): Promise<void> {\n\tconst currentOwner = resolveWorkflowStageDeliveryTarget(this);\n\tif (currentOwner !== this) return currentOwner.sendCustomMessage(message, options);\n\tconst appMessage = {\n\t\trole: \"custom\" as const,\n\t\tcustomType: message.customType,\n\t\tcontent: message.content ?? [],\n\t\tdisplay: message.display,\n\t\tdetails: message.details,\n\t\ttimestamp: Date.now(),\n\t\t...(options?.excludeFromContext === true ? { excludeFromContext: true } : {}),\n\t} satisfies CustomMessage<T>;\n\tconst boundary = this._workflowStageAdmission;\n\tconst deliver = async (): Promise<void> => {\n\t\tif (boundary && options?.stageAdmissionBarrier) await options.stageAdmissionBarrier();\n\t\tawait commitAdmittedCustomMessage(this, appMessage, options);\n\t};\n\tif (boundary === undefined) return deliver();\n\tawait boundary.admit(\n\t\toptions?.stageAdmissionKey,\n\t\tdeliver,\n\t\t() => {\n\t\t\tconst router = this._orchestrationContext?.lateMessageRouter;\n\t\t\tif (router === undefined) throw new Error(\"Workflow stage closed without a late-message router\");\n\t\t\treturn router.routeMessage(message, options);\n\t\t},\n\t).completion;\n}\n\n/** Atomically admits a custom-message batch in array order. */\nexport async function sendCustomMessages<T = unknown>(this: AgentSession,\n\tmessages: Array<Pick<CustomMessage<T>, \"customType\" | \"content\" | \"display\" | \"details\">>,\n\toptions?: SendMessagesOptions,\n): Promise<void> {\n\tconst currentOwner = resolveWorkflowStageDeliveryTarget(this);\n\tif (currentOwner !== this) return currentOwner.sendCustomMessages(messages, options);\n\tconst timestamp = Date.now();\n\tconst appMessages = messages.map((message) => ({\n\t\trole: \"custom\" as const,\n\t\tcustomType: message.customType,\n\t\tcontent: message.content ?? [],\n\t\tdisplay: message.display,\n\t\tdetails: message.details,\n\t\ttimestamp,\n\t\t...(options?.excludeFromContext === true ? { excludeFromContext: true } : {}),\n\t} satisfies CustomMessage<T>));\n\tif (appMessages.length === 0) return;\n\tconst boundary = this._workflowStageAdmission;\n\tconst deliver = async (): Promise<void> => {\n\t\tif (boundary && options?.stageAdmissionBarrier) await options.stageAdmissionBarrier();\n\t\tawait commitAdmittedCustomMessages(this, appMessages, options);\n\t};\n\tif (boundary === undefined) return deliver();\n\tawait boundary.admit(\n\t\toptions?.stageAdmissionKey,\n\t\tdeliver,\n\t\t() => {\n\t\t\tconst router = this._orchestrationContext?.lateMessageRouter;\n\t\t\tif (router === undefined) throw new Error(\"Workflow stage closed without a late-message router\");\n\t\t\treturn router.routeMessages(messages, options);\n\t\t},\n\t).completion;\n}\n\nexport function sealWorkflowStageGeneration(this: AgentSession): void {\n\tthis._workflowStageAdmission?.seal();\n}\n\nexport async function closeWorkflowStageGeneration(this: AgentSession): Promise<void> {\n\tawait this._workflowStageAdmission?.close();\n\tawait this.agent?.waitForIdle?.();\n\tawait this._agentEventQueue;\n}\n\nexport function _appendCustomMessage<T>(this: AgentSession, message: CustomMessage<T>): void {\n\tconst owner = resolveWorkflowStageDeliveryTarget(this);\n\tif (owner !== this) return owner._appendCustomMessage(message);\n\tthis.agent.state.messages.push(message);\n\tthis.sessionManager.appendCustomMessageEntry(\n\t\tmessage.customType,\n\t\tmessage.content,\n\t\tmessage.display,\n\t\tmessage.details,\n\t\tcustomMessageExcludesContext(message),\n\t);\n\tthis._emit({ type: \"message_start\", message });\n\tthis._emit({ type: \"message_end\", message });\n}\n\nexport function _enqueueInterruptCustomMessage<T>(this: AgentSession, message: CustomMessage<T>, options?: SendMessageOptions): Promise<void> {\n\tconst owner = resolveWorkflowStageDeliveryTarget(this);\n\tif (owner !== this) return owner._enqueueInterruptCustomMessage(message, forwardedMessageOptions(options));\n\tthis._pendingInterruptDeliveries += 1;\n\t// Establish the hold synchronously when the interrupt is enqueued, not when\n\t// the serialized delivery callback later starts. Callers commonly fire and\n\t// forget sendCustomMessage(), then queue additional steer/follow-up messages\n\t// before the promise chain gets a microtask; those messages must be captured\n\t// in the active interrupt hold instead of pi-agent-core's live queues.\n\tthis._ensureActiveInterruptQueueHold();\n\tconst delivery = this._interruptDeliveryQueue.then(async () => {\n\t\ttry {\n\t\t\tawait this._sendInterruptCustomMessageNow(message, options);\n\t\t} finally {\n\t\t\t// Retirement moves the pending count and hold synchronously. Settle the\n\t\t\t// same live owner so its queue cannot remain stranded on the replacement.\n\t\t\tconst liveOwner = resolveWorkflowStageDeliveryTarget(this);\n\t\t\tliveOwner._pendingInterruptDeliveries -= 1;\n\t\t\tif (liveOwner._pendingInterruptDeliveries === 0) {\n\t\t\t\tliveOwner._restoreAndClearActiveInterruptQueueHold();\n\t\t\t}\n\t\t}\n\t});\n\tthis._interruptDeliveryQueue = delivery.catch(() => undefined);\n\treturn delivery;\n}\n\n\nasync function sendInterruptCustomMessageUnlocked<T>(\n\tsession: AgentSession,\n\tmessage: CustomMessage<T>,\n\toptions?: SendMessageOptions,\n): Promise<void> {\n\tsession.abortRetry();\n\tsession._ensureActiveInterruptQueueHold();\n\tif (session.isStreaming) {\n\t\tconst previousAbortMessage = session._activeInterruptAbortMessage;\n\t\tsession._activeInterruptAbortMessage = normalizeInterruptAbortMessage(options?.interruptAbortMessage);\n\t\ttry {\n\t\t\tsession.agent.abort();\n\t\t\tawait session.agent.waitForIdle();\n\t\t\tawait session._agentEventQueue;\n\t\t} finally {\n\t\t\tsession._activeInterruptAbortMessage = previousAbortMessage;\n\t\t}\n\t}\n\tconst owner = resolveWorkflowStageDeliveryTarget(session);\n\tif (owner !== session) return owner._sendInterruptCustomMessageNow(message, forwardedMessageOptions(options));\n\tif (session._queuedMessagesPaused) {\n\t\tsession._queueAgentMessage(message, \"steer\");\n\t\treturn;\n\t}\n\tawait session.agent.prompt(message);\n}\n\nexport function _sendInterruptCustomMessageNow<T>(this: AgentSession,\n\tmessage: CustomMessage<T>,\n\toptions?: SendMessageOptions,\n): Promise<void> {\n\tconst owner = resolveWorkflowStageDeliveryTarget(this);\n\tif (owner !== this) return owner._sendInterruptCustomMessageNow(message, forwardedMessageOptions(options));\n\treturn serializeInterruptMutation(this, () => sendInterruptCustomMessageUnlocked(this, message, options));\n}\n\n\nexport function _ensureActiveInterruptQueueHold(this: AgentSession): InterruptQueueHold {\n\tconst owner = resolveWorkflowStageDeliveryTarget(this);\n\tif (owner !== this) return owner._ensureActiveInterruptQueueHold();\n\tif (this._activeInterruptQueueHold !== undefined) {\n\t\treturn this._activeInterruptQueueHold;\n\t}\n\tconst drained = this._drainQueuedAgentMessages();\n\tthis._activeInterruptQueueHold = {\n\t\tsteering: [...drained.steering],\n\t\tfollowUp: [...drained.followUp],\n\t};\n\treturn this._activeInterruptQueueHold;\n}\n\n\nexport function _restoreAndClearActiveInterruptQueueHold(this: AgentSession): void {\n\tconst owner = resolveWorkflowStageDeliveryTarget(this);\n\tif (owner !== this) return owner._restoreAndClearActiveInterruptQueueHold();\n\tif (this._queuedMessagesPaused) return;\n\tconst hold = this._activeInterruptQueueHold;\n\tif (hold === undefined) return;\n\tconst currentCoreQueues = this._drainQueuedAgentMessages();\n\tthis._restoreQueuedAgentMessages({\n\t\tsteering: [...hold.steering, ...currentCoreQueues.steering],\n\t\tfollowUp: [...hold.followUp, ...currentCoreQueues.followUp],\n\t});\n\tthis._activeInterruptQueueHold = undefined;\n}\n\n\nexport function _queueAgentMessage(this: AgentSession, message: AgentMessage, delivery: \"steer\" | \"followUp\"): void {\n\tconst owner = resolveWorkflowStageDeliveryTarget(this);\n\tif (owner !== this) return owner._queueAgentMessage(message, delivery);\n\tconst hold = this._activeInterruptQueueHold;\n\tif (hold !== undefined) {\n\t\tif (delivery === \"followUp\") {\n\t\t\thold.followUp.push(message);\n\t\t} else {\n\t\t\thold.steering.push(message);\n\t\t}\n\t\treturn;\n\t}\n\tif (delivery === \"followUp\") {\n\t\tthis.agent.followUp(message);\n\t} else {\n\t\tthis.agent.steer(message);\n\t}\n}\n\n\nexport function _drainQueuedAgentMessages(this: AgentSession): DrainedAgentQueues {\n\t// pi-agent-core exposes public clear methods but no public drain/restore pair.\n\t// Interrupt and pause holds prevent an aborting run from consuming queued raw\n\t// messages while preserving every entry for the existing resume boundary.\n\tconst agentWithQueues = this.agent as unknown as AgentQueueAccess;\n\treturn {\n\t\tsteering: drainAgentMessageQueue(agentWithQueues.steeringQueue),\n\t\tfollowUp: drainAgentMessageQueue(agentWithQueues.followUpQueue),\n\t};\n}\n\n\nexport function _restoreQueuedAgentMessages(this: AgentSession, queues: DrainedAgentQueues): void {\n\tfor (const message of queues.steering) {\n\t\tthis.agent.steer(message);\n\t}\n\tfor (const message of queues.followUp) {\n\t\tthis.agent.followUp(message);\n\t}\n}\n/**\n * Send a user message to the agent. Always triggers a turn.\n * When the agent is streaming, use deliverAs to specify how to queue the message.\n *\n * @param content User message content (string or content array)\n * @param options.deliverAs Delivery mode when streaming: \"steer\" or \"followUp\"\n */\nexport function clearQueue(this: AgentSession): { steering: string[]; followUp: string[] } {\n\tconst owner = resolveWorkflowStageDeliveryTarget(this);\n\tif (owner !== this) return owner.clearQueue();\n\tconst steering = [...this._steeringMessages];\n\tconst followUp = [...this._followUpMessages];\n\tthis._steeringMessages = [];\n\tthis._followUpMessages = [];\n\tconst removed = this._drainQueuedAgentMessages();\n\tif (this._activeInterruptQueueHold !== undefined) {\n\t\tremoved.steering.push(...this._activeInterruptQueueHold.steering.splice(0));\n\t\tremoved.followUp.push(...this._activeInterruptQueueHold.followUp.splice(0));\n\t}\n\trestoreProtectedStreamingCustomMessages(this, removed);\n\tconst hold = this._activeInterruptQueueHold;\n\tif (!this._queuedMessagesPaused && this._pendingInterruptDeliveries === 0 && hold !== undefined\n\t\t&& hold.steering.length === 0 && hold.followUp.length === 0) {\n\t\tthis._activeInterruptQueueHold = undefined;\n\t}\n\tthis._emitQueueUpdate();\n\treturn { steering, followUp };\n}\n\n/** Number of pending messages (includes both steering and follow-up) */\nexport function getSteeringMessages(this: AgentSession): readonly string[] {\n\treturn this._steeringMessages;\n}\n\n/** Get pending follow-up messages (read-only) */\nexport function getFollowUpMessages(this: AgentSession): readonly string[] {\n\treturn this._followUpMessages;\n}\n\n\n// =========================================================================\n// Model Management\n// =========================================================================\n\n\nexport function setSteeringMode(this: AgentSession, mode: \"all\" | \"one-at-a-time\"): void {\n\tthis.agent.steeringMode = mode;\n\tthis.settingsManager.setSteeringMode(mode);\n}\n\n/**\n * Set follow-up message mode.\n * Saves to settings.\n */\n\nexport function setFollowUpMode(this: AgentSession, mode: \"all\" | \"one-at-a-time\"): void {\n\tthis.agent.followUpMode = mode;\n\tthis.settingsManager.setFollowUpMode(mode);\n}\n\n// =========================================================================\n// Queue and delivery settings\n// =========================================================================\nexport const agentSessionMessageQueueMethods = {\n\t_queueSteer,\n\t_queueFollowUp,\n\t_throwIfExtensionCommand,\n\ttransferWorkflowStageDeliveriesTo,\n\tsealWorkflowStageGeneration,\n\tsendCustomMessage,\n\tsendCustomMessages,\n\tcloseWorkflowStageGeneration,\n\t_appendCustomMessage,\n\t_enqueueInterruptCustomMessage,\n\t_sendInterruptCustomMessageNow,\n\t_ensureActiveInterruptQueueHold,\n\t_restoreAndClearActiveInterruptQueueHold,\n\tpauseQueuedMessages,\n\tresumeQueuedMessages,\n\t_queueAgentMessage,\n\t_drainQueuedAgentMessages,\n\t_restoreQueuedAgentMessages,\n\tclearQueue,\n\tgetSteeringMessages,\n\tgetFollowUpMessages,\n\tabort,\n\tsetSteeringMode,\n\tsetFollowUpMode,\n};\n"]}

package/dist/core/agent-session-methods.d.ts

@@ -3,8 +3,9 @@ import type { ProviderHeaders } from "@earendil-works/pi-ai";
 import type { Api, AssistantMessage, ImageContent, Message, Model, TextContent } from "@earendil-works/pi-ai/compat";
 import type { PendingPostToolCompactionGuard } from "./agent-session-post-tool-compaction.ts";
 import type { BashResult } from "./bash-executor.ts";
-import type { VerbatimCompactionParameters, VerbatimCompactionResult } from "./compaction/index.ts";
+import type { CompactionUrgency, PlannerAuth, VerbatimCompactionParameters, VerbatimCompactionResult } from "./compaction/index.ts";
 import type { ContextUsage, ExtensionCommandContextActions, ExtensionErrorListener, ExtensionMode, ExtensionRunner, ExtensionUIContext, OrchestrationContext, ReplacedSessionContext, SessionStartEvent, ToolDefinition, ToolInfo } from "./extensions/index.ts";
+import type { PathMetadata } from "./package-manager.ts";
 import type { BashExecutionMessage, CustomMessage } from "./messages.ts";
 import type { ModelRegistry } from "./model-registry.ts";
 import type { PromptTemplate } from "./prompt-templates.ts";
@@ -14,20 +15,29 @@ import type { SettingsManager } from "./settings-manager.ts";
 import type { BuildSystemPromptOptions } from "./system-prompt.ts";
 import type { AsyncJobManager } from "./async/job-manager.js";
 import type { BashOperations } from "./tools/bash.ts";
-import type { AgentSessionEvent, AgentSessionEventListener, AgentSessionReloadOptions, ContextWindowReplayRequest, ContextWindowReplaySource, DrainedAgentQueues, ExtensionBindings, InterruptQueueHold, ModelCycleResult, PromptOptions, SessionStats, ToolDefinitionEntry } from "./agent-session-types.ts";
+import type { AgentSessionEvent, AgentSessionEventListener, AgentSessionReloadOptions, DrainedAgentQueues, ExtensionBindings, InterruptQueueHold, ModelCycleResult, PromptOptions, SessionStats, ToolDefinitionEntry } from "./agent-session-types.ts";
 import type { SendMessageOptions, SendMessagesOptions } from "./extensions/index.ts";
 export interface VerbatimCompactionApplyOptions {
-    resolvePlannerAuth: () => Promise<{
-        apiKey: string;
-        headers?: ProviderHeaders;
-        baseUrl?: string;
-    } | undefined>;
+    /** Per-model planner credentials; a borrowed fallback uses its own, never the session model's. */
+    resolvePlannerAuth: (model: Model<Api>) => Promise<PlannerAuth | undefined>;
     abortController: AbortController;
     backupLabel: string;
     compression_ratio?: number;
     preserve_recent?: number;
     query?: string;
     reason: "manual" | "threshold" | "overflow";
+    /** Only `load_bearing` may reach the context-destroying fresh rung. */
+    urgency: CompactionUrgency;
+    /**
+     * Admit a compactable region below the planner minimum.
+     *
+     * Narrow on purpose. Overflow recovery sets it because a real provider
+     * overflow is already known, and the post-tool preflight sets it only when its
+     * projected context is genuinely over the provider hard input limit. A
+     * threshold crossing that still fits must not clear context: the follow-up
+     * request can be sent as-is.
+     */
+    allowSmallRegion?: boolean;
 }
 export interface ExtensionResourcePathEntry {
     path: string;
@@ -35,24 +45,20 @@ export interface ExtensionResourcePathEntry {
 }
 export interface ExtensionResourcePathResult {
     path: string;
-    metadata: {
-        source: string;
-        scope: "temporary";
-        origin: "top-level";
-        baseDir?: string;
-    };
+    metadata: PathMetadata;
 }
 export interface RuntimeBuildOptions {
     activeToolNames?: string[];
     flagValues?: Map<string, boolean | string>;
     includeAllExtensionTools?: boolean;
 }
-export interface ContextWindowReplayResult {
-    model: Model<Api>;
-    contextWindow: number;
-    wouldWarn: boolean;
+export interface AgentSessionQueuePauseControl {
+    readonly queuedMessagesPaused: boolean;
+    pauseQueuedMessages(): void;
+    /** Release the hold without starting a turn; true means raw queued work was released. */
+    resumeQueuedMessages(beforeRelease?: () => void): Promise<boolean>;
 }
-export interface AgentSessionMethodSurface {
+export interface AgentSessionMethodSurface extends AgentSessionQueuePauseControl {
     readonly orchestrationContext: import("./extensions/index.ts").OrchestrationContext | undefined;
     readonly modelRegistry: ModelRegistry;
     readonly state: AgentState;
@@ -81,9 +87,9 @@ export interface AgentSessionMethodSurface {
     readonly isBashRunning: boolean;
     readonly hasPendingBashMessages: boolean;
     readonly extensionRunner: ExtensionRunner;
-    _handleAgentEvent(event: AgentEvent): void;
+    _handleAgentEvent(event: AgentEvent): Promise<void> | void;
     _getRequiredRequestAuth(model: Model<Api>): Promise<{
-        apiKey: string;
+        apiKey?: string;
         headers?: ProviderHeaders;
         baseUrl?: string;
     }>;
@@ -134,6 +140,8 @@ export interface AgentSessionMethodSurface {
     _throwIfExtensionCommand(text: string): void;
     sendCustomMessage<T = unknown>(message: Pick<CustomMessage<T>, "customType" | "content" | "display" | "details">, options?: SendMessageOptions): Promise<void>;
     sendCustomMessages<T = unknown>(messages: Array<Pick<CustomMessage<T>, "customType" | "content" | "display" | "details">>, options?: SendMessagesOptions): Promise<void>;
+    _commitAdmittedCustomMessage<T>(message: CustomMessage<T>, options?: SendMessageOptions): Promise<void>;
+    _commitAdmittedCustomMessages<T>(messages: CustomMessage<T>[], options?: SendMessagesOptions): Promise<void>;
     _appendCustomMessage<T>(message: CustomMessage<T>): void;
     _enqueueInterruptCustomMessage<T>(message: CustomMessage<T>, options?: SendMessageOptions): Promise<void>;
     _sendInterruptCustomMessageNow<T>(message: CustomMessage<T>, options?: SendMessageOptions): Promise<void>;
@@ -163,25 +171,11 @@ export interface AgentSessionMethodSurface {
     supportsThinking(): boolean;
     _getThinkingLevelForModelSwitch(explicitLevel?: ThinkingLevel): ThinkingLevel;
     _clampThinkingLevel(level: ThinkingLevel, availableLevels: ThinkingLevel[]): ThinkingLevel;
-    getAvailableContextWindows(): number[];
-    supportsContextWindowSelection(): boolean;
-    setContextWindow(contextWindow: number, options?: {
-        persistDefault?: boolean;
-    }): void;
-    _withContextWindowForModelSwitch(model: Model<Api>): Model<Api>;
-    _shouldCarryCurrentContextWindowForModelSwitch(currentModel: Model<Api>, settingsDefaultContextWindow: number | undefined): boolean;
-    _getSettingsContextWindowRequestForModel(model: Model<Api>): ContextWindowReplayRequest | undefined;
-    _getContextWindowReplayForModel(model: Model<Api>, requestedContextWindow: number | undefined, source: ContextWindowReplaySource | undefined): ContextWindowReplayResult;
-    _getDefaultContextWindowReplayForModel(model: Model<Api>, wouldWarn: boolean): ContextWindowReplayResult;
-    _getResumeContextWindowReplayForModel(model: Model<Api>): ContextWindowReplayResult;
-    _applyContextWindowReplay(contextWindow: number | undefined): void;
-    _appendContextWindowChangeIfChanged(previousModel: Model<Api> | undefined, nextModel: Model<Api>): void;
     _applyVerbatimCompaction(options: VerbatimCompactionApplyOptions): Promise<VerbatimCompactionResult | undefined>;
     compact(options?: Partial<VerbatimCompactionParameters>): Promise<VerbatimCompactionResult>;
     abortCompaction(): void;
     abortBranchSummary(): void;
     _checkCompaction(assistantMessage: AssistantMessage, skipAbortedCheck?: boolean): Promise<void>;
-    _isCopilotServerCapBelowSelectedContextWindow(assistantMessage: AssistantMessage): boolean;
     _dropTrailingAutoCompactionRetryAssistantIfPresent(): void;
     _schedulePostAutoCompactionContinuationProbe(reason: "overflow" | "threshold", willRetry: boolean): void;
     _awaitPendingPostCompactionContinuation(): Promise<void>;
@@ -206,7 +200,6 @@ export interface AgentSessionMethodSurface {
     _buildRuntime(options: RuntimeBuildOptions): void;
     reload(options?: AgentSessionReloadOptions): Promise<void>;
     _isRetryableError(message: AssistantMessage): boolean;
-    _normalizePersistedGeminiToolArgs(message: AssistantMessage): void;
     _isEmptyCompletion(message: AssistantMessage): boolean;
     _isSafetyRefusal(message: AssistantMessage): boolean;
     _handleRetryableError(message: AssistantMessage): Promise<boolean>;
@@ -214,14 +207,20 @@ export interface AgentSessionMethodSurface {
     abortRetry(): void;
     waitForRetry(): Promise<void>;
     setAutoRetryEnabled(enabled: boolean): void;
-    executeBash(command: string, onChunk?: (chunk: string) => void, options?: {
+    executeBash(command: string, onChunk?: (chunk: string, channel: "stdout" | "stderr") => void, options?: {
         excludeFromContext?: boolean;
+        id?: string;
         operations?: BashOperations;
+        pty?: boolean;
+        emitEvent?: boolean;
+        recordResult?: boolean;
     }): Promise<BashResult>;
     recordBashResult(command: string, result: BashResult, options?: {
         excludeFromContext?: boolean;
+        persist?: boolean;
+        defer?: boolean;
     }): void;
-    abortBash(): void;
+    abortBash(id?: string): void;
     _flushPendingBashMessages(): void;
     setSessionName(name: string): void;
     navigateTree(targetId: string, options?: {
@@ -254,7 +253,7 @@ export interface AgentSessionMethodSurface {
     transferWorkflowStageDeliveriesTo(target: object): void;
     hasExtensionHandlers(eventType: string): boolean;
 }
-export interface AgentSessionPublicSurface extends Pick<AgentSessionMethodSurface, "orchestrationContext" | "modelRegistry" | "state" | "model" | "thinkingLevel" | "isStreaming" | "systemPrompt" | "retryAttempt" | "isCompacting" | "messages" | "steeringMode" | "followUpMode" | "sessionFile" | "sessionId" | "sessionName" | "scopedModels" | "promptTemplates" | "pendingMessageCount" | "resourceLoader" | "autoCompactionEnabled" | "isRetrying" | "autoRetryEnabled" | "isBashRunning" | "hasPendingBashMessages" | "extensionRunner" | "subscribe" | "dispose" | "getActiveToolNames" | "getAllTools" | "getToolDefinition" | "setActiveToolsByName" | "setScopedModels" | "prompt" | "steer" | "followUp" | "sendCustomMessage" | "sendUserMessage" | "clearQueue" | "getSteeringMessages" | "getFollowUpMessages" | "abort" | "setModel" | "cycleModel" | "setThinkingLevel" | "cycleThinkingLevel" | "getAvailableThinkingLevels" | "supportsThinking" | "getAvailableContextWindows" | "supportsContextWindowSelection" | "setContextWindow" | "setSteeringMode" | "setFollowUpMode" | "compact" | "abortCompaction" | "abortBranchSummary" | "setAutoCompactionEnabled" | "bindExtensions" | "refreshCurrentModelFromRegistry" | "reload" | "abortRetry" | "setAutoRetryEnabled" | "executeBash" | "recordBashResult" | "abortBash" | "setSessionName" | "navigateTree" | "getUserMessagesForForking" | "getSessionStats" | "getContextUsage" | "exportToHtml" | "exportToJsonl" | "getLastAssistantText" | "createReplacedSessionContext" | "hasExtensionHandlers"> {
+export interface AgentSessionPublicSurface extends Pick<AgentSessionMethodSurface, "orchestrationContext" | "modelRegistry" | "state" | "model" | "thinkingLevel" | "isStreaming" | "systemPrompt" | "retryAttempt" | "isCompacting" | "messages" | "steeringMode" | "followUpMode" | "sessionFile" | "sessionId" | "sessionName" | "scopedModels" | "promptTemplates" | "pendingMessageCount" | "resourceLoader" | "autoCompactionEnabled" | "isRetrying" | "autoRetryEnabled" | "isBashRunning" | "hasPendingBashMessages" | "extensionRunner" | "queuedMessagesPaused" | "pauseQueuedMessages" | "resumeQueuedMessages" | "subscribe" | "dispose" | "getActiveToolNames" | "getAllTools" | "getToolDefinition" | "setActiveToolsByName" | "setScopedModels" | "prompt" | "steer" | "followUp" | "sendCustomMessage" | "sendUserMessage" | "clearQueue" | "getSteeringMessages" | "getFollowUpMessages" | "abort" | "setModel" | "cycleModel" | "setThinkingLevel" | "cycleThinkingLevel" | "getAvailableThinkingLevels" | "supportsThinking" | "setSteeringMode" | "setFollowUpMode" | "compact" | "abortCompaction" | "abortBranchSummary" | "setAutoCompactionEnabled" | "bindExtensions" | "refreshCurrentModelFromRegistry" | "reload" | "abortRetry" | "setAutoRetryEnabled" | "executeBash" | "recordBashResult" | "abortBash" | "setSessionName" | "navigateTree" | "getUserMessagesForForking" | "getSessionStats" | "getContextUsage" | "exportToHtml" | "exportToJsonl" | "getLastAssistantText" | "createReplacedSessionContext" | "hasExtensionHandlers"> {
     readonly agent: Agent;
     readonly sessionManager: SessionManager;
     readonly settingsManager: SettingsManager;
@@ -281,9 +280,18 @@ export interface AgentSessionInternalSurface extends AgentSessionMethodSurface,
     _terminatingToolCallIds: Set<string>;
     _pendingInterruptDeliveries: number;
     _activeInterruptQueueHold: InterruptQueueHold | undefined;
+    _queuedMessagesPaused: boolean;
+    _queuedMessagesPauseAbortBoundary: Promise<void> | undefined;
+    _workflowStageDeliveryForwardTarget: AgentSessionInternalSurface | undefined;
     _activeInterruptAbortMessage: string | undefined;
     _pendingNextTurnMessages: CustomMessage[];
+    _protectedStreamingCustomMessages: Array<{
+        message: CustomMessage;
+        delivery: "steer" | "followUp";
+        phase: "queued" | "consumed-unpersisted" | "persistence-failed";
+    }>;
     _compactionAbortController: AbortController | undefined;
+    _manualCompactionPromise: Promise<VerbatimCompactionResult> | undefined;
     _autoCompactionAbortController: AbortController | undefined;
     _overflowRecoveryAttempted: boolean;
     _branchSummaryAbortController: AbortController | undefined;
@@ -291,7 +299,7 @@ export interface AgentSessionInternalSurface extends AgentSessionMethodSurface,
     _retryAttempt: number;
     _retryPromise: Promise<void> | undefined;
     _retryResolve: (() => void) | undefined;
-    _bashAbortController: AbortController | undefined;
+    _bashAbortControllers: Map<string | symbol, AbortController>;
     _pendingBashMessages: BashExecutionMessage[];
     _extensionRunner: ExtensionRunner;
     _turnIndex: number;