@averyyy/pi-coding-agent 0.82.1-piclient.9 → 0.83.0-piclient.1

package/dist/core/agent-session.js

@@ -182,7 +182,7 @@ export class AgentSession {
     _retryAbortController = undefined;
     _retryAttempt = 0;
     // Bash execution state
-    _bashAbortController = undefined;
+    _bashAbortControllers = new Set();
     _pendingBashMessages = [];
     _validationModifiedFiles = new Set();
     _validationFailure = undefined;
@@ -2353,6 +2353,7 @@ export class AgentSession {
             setThinkingLevel: (level) => this.setThinkingLevel(level),
         }, {
             getModel: () => this.model,
+            getScopedModels: () => this._scopedModels,
             isIdle: () => this.isIdle,
             isProjectTrusted: () => this.settingsManager.isProjectTrusted(),
             getSignal: () => this.agent.signal,
@@ -2682,7 +2683,8 @@ export class AgentSession {
      * @param options.operations Custom BashOperations for remote execution
      */
     async executeBash(command, onChunk, options) {
-        this._bashAbortController = new AbortController();
+        const abortController = new AbortController();
+        this._bashAbortControllers.add(abortController);
         // Apply command prefix if configured (e.g., "shopt -s expand_aliases" for alias support)
         const prefix = this.settingsManager.getShellCommandPrefix();
         const shellPath = this.settingsManager.getShellPath();
@@ -2693,13 +2695,13 @@ export class AgentSession {
                     onChunk?.(delta);
                     this._emit({ type: "bash_execution_update", id: options?.id, delta });
                 },
-                signal: this._bashAbortController.signal,
+                signal: abortController.signal,
             });
             this.recordBashResult(command, result, options);
             return result;
         }
         finally {
-            this._bashAbortController = undefined;
+            this._bashAbortControllers.delete(abortController);
         }
     }
     /**
@@ -2734,11 +2736,13 @@ export class AgentSession {
      * Cancel running bash command.
      */
     abortBash() {
-        this._bashAbortController?.abort();
+        for (const abortController of [...this._bashAbortControllers]) {
+            abortController.abort();
+        }
     }
     /** Whether a bash command is currently running */
     get isBashRunning() {
-        return this._bashAbortController !== undefined;
+        return this._bashAbortControllers.size > 0;
     }
     /** Whether there are pending bash messages waiting to be flushed */
     get hasPendingBashMessages() {
@@ -2786,6 +2790,9 @@ export class AgentSession {
      * @returns Result with editorText (if user message) and cancelled status
      */
     async navigateTree(targetId, options = {}) {
+        if (this.isStreaming) {
+            throw new Error("Wait for the current response to finish before navigating the session tree.");
+        }
         const oldLeafId = this.sessionManager.getLeafId();
         // No-op if already at target
         if (targetId === oldLeafId) {