@apholdings/jensen-code 0.0.3 → 0.0.5

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,12 +973,23 @@ 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();
         this._steeringMessages = [];
         this._followUpMessages = [];
         this._pendingNextTurnMessages = [];
+        this._pendingBashMessages = [];
+        this._lastAssistantMessage = undefined;
+        this._overflowRecoveryAttempted = false;
+        this._retryAttempt = 0;
+        this._retryAbortController = undefined;
+        this._compactionAbortController = undefined;
+        this._autoCompactionAbortController = undefined;
+        this._branchSummaryAbortController = undefined;
+        this._bashAbortController = undefined;
+        this._resolveRetry();
         this.sessionManager.appendThinkingLevelChange(this.thinkingLevel);
         // Run setup callback if provided (e.g., to append initial messages)
         if (options?.setup) {
@@ -1214,6 +1237,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 +2059,7 @@ export class AgentSession {
         }
         this._disconnectFromAgent();
         await this.abort();
+        await this._drainAgentEventQueue();
         this._steeringMessages = [];
         this._followUpMessages = [];
         this._pendingNextTurnMessages = [];