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

package/dist/core/agent-session.js

@@ -29,7 +29,7 @@ import { createToolHtmlRenderer } from "./export-html/tool-renderer.js";
 import { ExtensionRunner, wrapRegisteredTools, } from "./extensions/index.js";
 import { emitSessionShutdownEvent } from "./extensions/runner.js";
 import { ModelRegistry } from "./model-registry.js";
-import { compactPiServer, syncPiServerTree, } from "./pi-server-client.js";
+import { compactPiServer } from "./pi-server-client.js";
 import { expandPromptTemplate } from "./prompt-templates.js";
 import { CURRENT_SESSION_VERSION, getLatestCompactionEntry } from "./session-manager.js";
 import { createSyntheticSourceInfo } from "./source-info.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;
@@ -491,7 +491,7 @@ export class AgentSession {
     // Track last assistant message for auto-compaction check
     _lastAssistantMessage = undefined;
     /** Internal handler for agent events - shared by subscribe and reconnect */
-    _handleAgentEvent = async (event, signal) => {
+    _handleAgentEvent = async (event, _signal) => {
         // When a user message starts, check if it's from either queue and remove it BEFORE emitting
         // This ensures the UI sees the updated queue state
         if (event.type === "message_start" && event.message.role === "user") {
@@ -549,13 +549,6 @@ export class AgentSession {
                     });
                     this._retryAttempt = 0;
                 }
-                if (isPiServerMode() && !this._isTerminalAssistantFailure(assistantMsg)) {
-                    await syncPiServerTree(this.sessionId, {
-                        systemPrompt: this.systemPrompt,
-                        messages: this.agent.state.messages,
-                        tools: this.agent.state.tools,
-                    }, { entries: this.sessionManager.getEntries(), leafId: this.sessionManager.getLeafId() }, { signal, onHistoryReconciled: (snapshot) => this.reconcilePiServerHistory(snapshot) });
-                }
             }
         }
     };
@@ -2360,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,
@@ -2689,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();
@@ -2700,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);
         }
     }
     /**
@@ -2741,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() {
@@ -2793,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) {
@@ -2935,13 +2935,6 @@ export class AgentSession {
             // Update agent state
             const sessionContext = this.sessionManager.buildSessionContext();
             this.agent.state.messages = sessionContext.messages;
-            if (isPiServerMode()) {
-                await syncPiServerTree(this.sessionId, {
-                    systemPrompt: this.systemPrompt,
-                    messages: this.agent.state.messages,
-                    tools: this.agent.state.tools,
-                }, { entries: this.sessionManager.getEntries(), leafId: this.sessionManager.getLeafId() }, { onHistoryReconciled: (snapshot) => this.reconcilePiServerHistory(snapshot) });
-            }
             // Emit session_tree event
             await this._extensionRunner.emit({
                 type: "session_tree",