@earendil-works/pi-coding-agent 0.82.0 → 0.83.0

package/dist/core/agent-session.js

@@ -102,7 +102,7 @@ export class AgentSession {
     _retryAbortController = undefined;
     _retryAttempt = 0;
     // Bash execution state
-    _bashAbortController = undefined;
+    _bashAbortControllers = new Set();
     _pendingBashMessages = [];
     // Extension system
     _extensionRunner;
@@ -174,7 +174,7 @@ export class AgentSession {
             }
             throw error;
         }
-        if (result?.auth.apiKey) {
+        if (result && (result.auth.apiKey || result.auth.headers)) {
             return {
                 apiKey: result.auth.apiKey,
                 headers: withoutDeletedHeaders(result.auth.headers),
@@ -1599,12 +1599,7 @@ export class AgentSession {
             let headers;
             let env;
             if (this.agent.streamFunction === streamSimple) {
-                const authResult = await this._modelRuntime.getAuth(this.model);
-                if (!authResult?.auth.apiKey)
-                    return false;
-                apiKey = authResult.auth.apiKey;
-                headers = withoutDeletedHeaders(authResult.auth.headers);
-                env = authResult.env;
+                ({ apiKey, headers, env } = await this._getRequiredRequestAuth(this.model));
             }
             else {
                 ({ apiKey, headers, env } = await this._getSummarizationRequestAuth(this.model));
@@ -1897,6 +1892,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,
@@ -2199,7 +2195,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();
@@ -2210,13 +2207,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);
         }
     }
     /**
@@ -2251,11 +2248,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() {
@@ -2303,6 +2302,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) {