@averyyy/pi-coding-agent 0.84.4-piclient.2 → 0.85.1-piclient.1

package/dist/core/agent-session.js

@@ -498,7 +498,7 @@ export class AgentSession {
         return this._idleWaitPromise;
     }
     _resolveIdleWaitIfIdle() {
-        if (this._isAgentRunActive || !this._resolveIdleWait) {
+        if (!this.isIdle || !this._resolveIdleWait) {
             return;
         }
         const resolve = this._resolveIdleWait;
@@ -834,9 +834,9 @@ export class AgentSession {
     get isStreaming() {
         return this._isAgentRunActive;
     }
-    /** Whether the session has no active agent run, retry, auto-compaction, or queued continuation. */
+    /** Whether the session has no active agent run, compaction, branch summary, retry, or queued continuation. */
     get isIdle() {
-        return !this._isAgentRunActive;
+        return !this._isAgentRunActive && !this.isCompacting;
     }
     /** Current effective system prompt (includes any per-turn extension modifications) */
     get systemPrompt() {
@@ -1528,6 +1528,8 @@ export class AgentSession {
      */
     async abort() {
         this.abortRetry();
+        this.abortCompaction();
+        this.abortBranchSummary();
         this.agent.abort();
         await this.waitForIdle();
     }
@@ -1784,6 +1786,10 @@ export class AgentSession {
         }
         return compactionResult;
     }
+    _clearManualCompactionState() {
+        this._compactionAbortController = undefined;
+        this._resolveIdleWaitIfIdle();
+    }
     /**
      * Manually compact the session context.
      * Aborts current agent operation first.
@@ -1907,7 +1913,7 @@ export class AgentSession {
                 details,
             };
             // compaction_end listeners may submit queued prompts, so expose idle state before notifying them.
-            this._compactionAbortController = undefined;
+            this._clearManualCompactionState();
             this._emit({
                 type: "compaction_end",
                 reason: "manual",
@@ -1921,18 +1927,20 @@ export class AgentSession {
             const message = error instanceof Error ? error.message : String(error);
             const aborted = message === "Compaction cancelled" || (error instanceof Error && error.name === "AbortError");
             this._compactionAbortController = undefined;
+            const errorMessage = aborted ? undefined : `Compaction failed: ${message}`;
+            this._clearManualCompactionState();
             this._emit({
                 type: "compaction_end",
                 reason: "manual",
                 result: undefined,
                 aborted,
                 willRetry: false,
-                errorMessage: aborted ? undefined : `Compaction failed: ${message}`,
+                errorMessage,
             });
             throw error;
         }
         finally {
-            this._compactionAbortController = undefined;
+            this._clearManualCompactionState();
         }
     }
     /**
@@ -2239,6 +2247,7 @@ export class AgentSession {
         }
         finally {
             this._autoCompactionAbortController = undefined;
+            this._resolveIdleWaitIfIdle();
         }
     }
     /**
@@ -3060,6 +3069,7 @@ export class AgentSession {
         }
         finally {
             this._branchSummaryAbortController = undefined;
+            this._resolveIdleWaitIfIdle();
         }
     }
     /**