@apholdings/jensen-code 0.0.3 → 0.0.4

package/dist/core/agent-session.js

@@ -396,6 +396,18 @@ export class AgentSession {
             return; // Already connected
         this._unsubscribeAgent = this.agent.subscribe(this._handleAgentEvent);
     }
+    /**
+     * Wait for already-enqueued agent event processing to finish.
+     * This prevents stale events from mutating session state after a reset/switch.
+     */
+    async _drainAgentEventQueue() {
+        try {
+            await this._agentEventQueue;
+        }
+        catch {
+            // The queue is kept alive by _handleAgentEvent(); ignore the prior failure here.
+        }
+    }
     /**
      * Remove all listeners and disconnect from agent.
      * Call this when completely done with the session.
@@ -961,6 +973,7 @@ export class AgentSession {
         }
         this._disconnectFromAgent();
         await this.abort();
+        await this._drainAgentEventQueue();
         this.agent.reset();
         this.sessionManager.newSession({ parentSession: options?.parentSession });
         this.agent.sessionId = this.sessionManager.getSessionId();
@@ -1214,6 +1227,7 @@ export class AgentSession {
     async compact(customInstructions) {
         this._disconnectFromAgent();
         await this.abort();
+        await this._drainAgentEventQueue();
         this._compactionAbortController = new AbortController();
         try {
             if (!this.model) {
@@ -2035,6 +2049,7 @@ export class AgentSession {
         }
         this._disconnectFromAgent();
         await this.abort();
+        await this._drainAgentEventQueue();
         this._steeringMessages = [];
         this._followUpMessages = [];
         this._pendingNextTurnMessages = [];