open-multi-agent-kit 0.79.1 → 0.80.0

package/dist/core/agent-session.js

@@ -1,2535 +0,0 @@
-/**
- * AgentSession - Core abstraction for agent lifecycle and session management.
- *
- * This class is shared between all run modes (interactive, print, rpc).
- * It encapsulates:
- * - Agent state access
- * - Event subscription with automatic session persistence
- * - Model and thinking level management
- * - Compaction (manual and auto)
- * - Bash execution
- * - Session switching and branching
- *
- * Modes use this class and add their own I/O layer on top.
- */
-import { existsSync, mkdirSync, readFileSync, writeFileSync } from "node:fs";
-import { basename, dirname } from "node:path";
-import { clampThinkingLevel, cleanupSessionResources, getSupportedThinkingLevels, isContextOverflow, modelsAreEqual, resetApiProviders, streamSimple, } from "@earendil-works/omk-ai";
-import { theme } from "../modes/interactive/theme/theme.js";
-import { stripFrontmatter } from "../utils/frontmatter.js";
-import { resolvePath } from "../utils/paths.js";
-import { sleep } from "../utils/sleep.js";
-import { resolveAdaptiveHeadroomThreshold } from "./adaptive-runtime.js";
-import { formatNoApiKeyFoundMessage, formatNoModelSelectedMessage } from "./auth-guidance.js";
-import { executeBashWithOperations } from "./bash-executor.js";
-import { calculateContextTokens, collectEntriesForBranchSummary, compact, estimateContextTokens, generateBranchSummary, prepareCompaction, shouldCompact, } from "./compaction/index.js";
-import { DEFAULT_THINKING_LEVEL } from "./defaults.js";
-import { exportSessionToHtml } from "./export-html/index.js";
-import { createToolHtmlRenderer } from "./export-html/tool-renderer.js";
-import { ExtensionRunner, wrapRegisteredTools, } from "./extensions/index.js";
-import { emitSessionShutdownEvent } from "./extensions/runner.js";
-import { expandPromptTemplate } from "./prompt-templates.js";
-import { CURRENT_SESSION_VERSION, getLatestCompactionEntry } from "./session-manager.js";
-import { createSyntheticSourceInfo } from "./source-info.js";
-import { buildSystemPrompt } from "./system-prompt.js";
-import { createLocalBashOperations } from "./tools/bash.js";
-import { createAllToolDefinitions } from "./tools/index.js";
-import { createToolDefinitionFromAgentTool } from "./tools/tool-definition-wrapper.js";
-/**
- * Parse a skill block from message text.
- * Returns null if the text doesn't contain a skill block.
- */
-export function parseSkillBlock(text) {
-    const match = text.match(/^<skill name="([^"]+)" location="([^"]+)">\n([\s\S]*?)\n<\/skill>(?:\n\n([\s\S]+))?$/);
-    if (!match)
-        return null;
-    return {
-        name: match[1],
-        location: match[2],
-        content: match[3],
-        userMessage: match[4]?.trim() || undefined,
-    };
-}
-// ============================================================================
-// Constants
-// ============================================================================
-/** Standard thinking levels */
-const THINKING_LEVELS = ["off", "minimal", "low", "medium", "high", "xhigh", "max"];
-// ============================================================================
-// AgentSession Class
-// ============================================================================
-export class AgentSession {
-    agent;
-    sessionManager;
-    settingsManager;
-    _scopedModels;
-    // Event subscription state
-    _unsubscribeAgent;
-    _eventListeners = [];
-    /** Tracks pending steering messages for UI display. Removed when delivered. */
-    _steeringMessages = [];
-    /** Tracks pending follow-up messages for UI display. Removed when delivered. */
-    _followUpMessages = [];
-    /** Messages queued to be included with the next user prompt as context ("asides"). */
-    _pendingNextTurnMessages = [];
-    // Compaction state
-    _compactionAbortController = undefined;
-    _autoCompactionAbortController = undefined;
-    _overflowRecoveryAttempted = false;
-    // Branch summarization state
-    _branchSummaryAbortController = undefined;
-    // Retry state
-    _retryAbortController = undefined;
-    _retryAttempt = 0;
-    // Bash execution state
-    _bashAbortController = undefined;
-    _pendingBashMessages = [];
-    // Extension system
-    _extensionRunner;
-    _turnIndex = 0;
-    _resourceLoader;
-    _customTools;
-    _baseToolDefinitions = new Map();
-    _cwd;
-    _extensionRunnerRef;
-    _initialActiveToolNames;
-    _allowedToolNames;
-    _excludedToolNames;
-    _baseToolsOverride;
-    _sessionStartEvent;
-    _extensionUIContext;
-    _extensionMode = "print";
-    _extensionCommandContextActions;
-    _extensionAbortHandler;
-    _extensionShutdownHandler;
-    _extensionErrorListener;
-    _extensionErrorUnsubscriber;
-    // Model registry for API key resolution
-    _modelRegistry;
-    // Tool registry for extension getTools/setTools
-    _toolRegistry = new Map();
-    _toolDefinitions = new Map();
-    _toolPromptSnippets = new Map();
-    _toolPromptGuidelines = new Map();
-    // Base system prompt (without extension appends) - used to apply fresh appends each turn
-    _baseSystemPrompt = "";
-    _baseSystemPromptOptions;
-    constructor(config) {
-        this.agent = config.agent;
-        this.sessionManager = config.sessionManager;
-        this.settingsManager = config.settingsManager;
-        this._scopedModels = config.scopedModels ?? [];
-        this._resourceLoader = config.resourceLoader;
-        this._customTools = config.customTools ?? [];
-        this._cwd = config.cwd;
-        this._modelRegistry = config.modelRegistry;
-        this._extensionRunnerRef = config.extensionRunnerRef;
-        this._initialActiveToolNames = config.initialActiveToolNames;
-        this._allowedToolNames = config.allowedToolNames ? new Set(config.allowedToolNames) : undefined;
-        this._excludedToolNames = config.excludedToolNames ? new Set(config.excludedToolNames) : undefined;
-        this._baseToolsOverride = config.baseToolsOverride;
-        this._sessionStartEvent = config.sessionStartEvent ?? { type: "session_start", reason: "startup" };
-        // Always subscribe to agent events for internal handling
-        // (session persistence, extensions, auto-compaction, retry logic)
-        this._unsubscribeAgent = this.agent.subscribe(this._handleAgentEvent);
-        this._installAgentToolHooks();
-        this._buildRuntime({
-            activeToolNames: this._initialActiveToolNames,
-            includeAllExtensionTools: true,
-        });
-    }
-    /** Model registry for API key resolution and model discovery */
-    get modelRegistry() {
-        return this._modelRegistry;
-    }
-    async _getRequiredRequestAuth(model) {
-        const result = await this._modelRegistry.getApiKeyAndHeaders(model);
-        if (!result.ok) {
-            if (result.error.startsWith("No API key found")) {
-                throw new Error(formatNoApiKeyFoundMessage(model.provider));
-            }
-            throw new Error(result.error);
-        }
-        if (result.apiKey) {
-            return { apiKey: result.apiKey, headers: result.headers };
-        }
-        const isOAuth = this._modelRegistry.isUsingOAuth(model);
-        if (isOAuth) {
-            throw new Error(`Authentication failed for "${model.provider}". ` +
-                `Credentials may have expired or network is unavailable. ` +
-                `Run '/login ${model.provider}' to re-authenticate.`);
-        }
-        throw new Error(formatNoApiKeyFoundMessage(model.provider));
-    }
-    async _getCompactionRequestAuth(model) {
-        if (this.agent.streamFn === streamSimple) {
-            return this._getRequiredRequestAuth(model);
-        }
-        const result = await this._modelRegistry.getApiKeyAndHeaders(model);
-        return result.ok ? { apiKey: result.apiKey, headers: result.headers } : {};
-    }
-    /**
-     * Install tool hooks once on the Agent instance.
-     *
-     * The callbacks read `this._extensionRunner` at execution time, so extension reload swaps in the
-     * new runner without reinstalling hooks. Extension-specific tool wrappers are still used to adapt
-     * registered tool execution to the extension context. Tool call and tool result interception now
-     * happens here instead of in wrappers.
-     */
-    _installAgentToolHooks() {
-        this.agent.beforeToolCall = async ({ toolCall, args }) => {
-            const runner = this._extensionRunner;
-            if (!runner.hasHandlers("tool_call")) {
-                return undefined;
-            }
-            try {
-                return await runner.emitToolCall({
-                    type: "tool_call",
-                    toolName: toolCall.name,
-                    toolCallId: toolCall.id,
-                    input: args,
-                });
-            }
-            catch (err) {
-                if (err instanceof Error) {
-                    throw err;
-                }
-                throw new Error(`Extension failed, blocking execution: ${String(err)}`);
-            }
-        };
-        this.agent.afterToolCall = async ({ toolCall, args, result, isError }) => {
-            const runner = this._extensionRunner;
-            if (!runner.hasHandlers("tool_result")) {
-                return undefined;
-            }
-            const hookResult = await runner.emitToolResult({
-                type: "tool_result",
-                toolName: toolCall.name,
-                toolCallId: toolCall.id,
-                input: args,
-                content: result.content,
-                details: result.details,
-                isError,
-            });
-            if (!hookResult) {
-                return undefined;
-            }
-            return {
-                content: hookResult.content,
-                details: hookResult.details,
-                isError: hookResult.isError ?? isError,
-            };
-        };
-    }
-    // =========================================================================
-    // Event Subscription
-    // =========================================================================
-    /** Emit an event to all listeners */
-    _emit(event) {
-        for (const l of this._eventListeners) {
-            l(event);
-        }
-    }
-    _emitQueueUpdate() {
-        this._emit({
-            type: "queue_update",
-            steering: [...this._steeringMessages],
-            followUp: [...this._followUpMessages],
-        });
-    }
-    // Track last assistant message for auto-compaction check
-    _lastAssistantMessage = undefined;
-    /** Internal handler for agent events - shared by subscribe and reconnect */
-    _handleAgentEvent = async (event) => {
-        // 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") {
-            this._overflowRecoveryAttempted = false;
-            const messageText = this._getUserMessageText(event.message);
-            if (messageText) {
-                // Check steering queue first
-                const steeringIndex = this._steeringMessages.indexOf(messageText);
-                if (steeringIndex !== -1) {
-                    this._steeringMessages.splice(steeringIndex, 1);
-                    this._emitQueueUpdate();
-                }
-                else {
-                    // Check follow-up queue
-                    const followUpIndex = this._followUpMessages.indexOf(messageText);
-                    if (followUpIndex !== -1) {
-                        this._followUpMessages.splice(followUpIndex, 1);
-                        this._emitQueueUpdate();
-                    }
-                }
-            }
-        }
-        // Emit to extensions first
-        await this._emitExtensionEvent(event);
-        // Notify all listeners
-        this._emit(event.type === "agent_end" ? { ...event, willRetry: this._willRetryAfterAgentEnd(event) } : event);
-        // Handle session persistence
-        if (event.type === "message_end") {
-            // Check if this is a custom message from extensions
-            if (event.message.role === "custom") {
-                // Persist as CustomMessageEntry
-                this.sessionManager.appendCustomMessageEntry(event.message.customType, event.message.content, event.message.display, event.message.details);
-            }
-            else if (event.message.role === "user" ||
-                event.message.role === "assistant" ||
-                event.message.role === "toolResult") {
-                // Regular LLM message - persist as SessionMessageEntry
-                this.sessionManager.appendMessage(event.message);
-            }
-            // Other message types (bashExecution, compactionSummary, branchSummary) are persisted elsewhere
-            // Track assistant message for auto-compaction (checked on agent_end)
-            if (event.message.role === "assistant") {
-                this._lastAssistantMessage = event.message;
-                const assistantMsg = event.message;
-                if (assistantMsg.stopReason !== "error") {
-                    this._overflowRecoveryAttempted = false;
-                }
-                // Reset retry counter immediately on successful assistant response
-                // This prevents accumulation across multiple LLM calls within a turn
-                if (assistantMsg.stopReason !== "error" && this._retryAttempt > 0) {
-                    this._emit({
-                        type: "auto_retry_end",
-                        success: true,
-                        attempt: this._retryAttempt,
-                    });
-                    this._retryAttempt = 0;
-                }
-            }
-        }
-    };
-    _willRetryAfterAgentEnd(event) {
-        const settings = this.settingsManager.getRetrySettings();
-        if (!settings.enabled || this._retryAttempt >= settings.maxRetries) {
-            return false;
-        }
-        for (let i = event.messages.length - 1; i >= 0; i--) {
-            const message = event.messages[i];
-            if (message.role === "assistant") {
-                return this._isRetryableError(message);
-            }
-        }
-        return false;
-    }
-    /** Extract text content from a message */
-    _getUserMessageText(message) {
-        if (message.role !== "user")
-            return "";
-        const content = message.content;
-        if (typeof content === "string")
-            return content;
-        const textBlocks = content.filter((c) => c.type === "text");
-        return textBlocks.map((c) => c.text).join("");
-    }
-    /** Find the last assistant message in agent state (including aborted ones) */
-    _findLastAssistantMessage() {
-        const messages = this.agent.state.messages;
-        for (let i = messages.length - 1; i >= 0; i--) {
-            const msg = messages[i];
-            if (msg.role === "assistant") {
-                return msg;
-            }
-        }
-        return undefined;
-    }
-    _replaceMessageInPlace(target, replacement) {
-        // Agent-core stores the finalized message object in its state before emitting message_end.
-        // SessionManager persistence happens later in _handleAgentEvent() with event.message.
-        // Mutating this object in place keeps agent state, later turn/agent events, listeners,
-        // and the eventual SessionManager.appendMessage(event.message) persistence in sync.
-        if (target === replacement) {
-            return;
-        }
-        const targetRecord = target;
-        for (const key of Object.keys(targetRecord)) {
-            delete targetRecord[key];
-        }
-        Object.assign(targetRecord, replacement);
-    }
-    /** Emit extension events based on agent events */
-    async _emitExtensionEvent(event) {
-        if (event.type === "agent_start") {
-            this._turnIndex = 0;
-            await this._extensionRunner.emit({ type: "agent_start" });
-        }
-        else if (event.type === "agent_end") {
-            await this._extensionRunner.emit({ type: "agent_end", messages: event.messages });
-        }
-        else if (event.type === "turn_start") {
-            const extensionEvent = {
-                type: "turn_start",
-                turnIndex: this._turnIndex,
-                timestamp: Date.now(),
-            };
-            await this._extensionRunner.emit(extensionEvent);
-        }
-        else if (event.type === "turn_end") {
-            const extensionEvent = {
-                type: "turn_end",
-                turnIndex: this._turnIndex,
-                message: event.message,
-                toolResults: event.toolResults,
-            };
-            await this._extensionRunner.emit(extensionEvent);
-            this._turnIndex++;
-        }
-        else if (event.type === "message_start") {
-            const extensionEvent = {
-                type: "message_start",
-                message: event.message,
-            };
-            await this._extensionRunner.emit(extensionEvent);
-        }
-        else if (event.type === "message_update") {
-            const extensionEvent = {
-                type: "message_update",
-                message: event.message,
-                assistantMessageEvent: event.assistantMessageEvent,
-            };
-            await this._extensionRunner.emit(extensionEvent);
-        }
-        else if (event.type === "message_end") {
-            const extensionEvent = {
-                type: "message_end",
-                message: event.message,
-            };
-            const replacement = await this._extensionRunner.emitMessageEnd(extensionEvent);
-            if (replacement) {
-                this._replaceMessageInPlace(event.message, replacement);
-            }
-        }
-        else if (event.type === "tool_execution_start") {
-            const extensionEvent = {
-                type: "tool_execution_start",
-                toolCallId: event.toolCallId,
-                toolName: event.toolName,
-                args: event.args,
-            };
-            await this._extensionRunner.emit(extensionEvent);
-        }
-        else if (event.type === "tool_execution_update") {
-            const extensionEvent = {
-                type: "tool_execution_update",
-                toolCallId: event.toolCallId,
-                toolName: event.toolName,
-                args: event.args,
-                partialResult: event.partialResult,
-            };
-            await this._extensionRunner.emit(extensionEvent);
-        }
-        else if (event.type === "tool_execution_end") {
-            const extensionEvent = {
-                type: "tool_execution_end",
-                toolCallId: event.toolCallId,
-                toolName: event.toolName,
-                result: event.result,
-                isError: event.isError,
-            };
-            await this._extensionRunner.emit(extensionEvent);
-        }
-    }
-    /**
-     * Subscribe to agent events.
-     * Session persistence is handled internally (saves messages on message_end).
-     * Multiple listeners can be added. Returns unsubscribe function for this listener.
-     */
-    subscribe(listener) {
-        this._eventListeners.push(listener);
-        // Return unsubscribe function for this specific listener
-        return () => {
-            const index = this._eventListeners.indexOf(listener);
-            if (index !== -1) {
-                this._eventListeners.splice(index, 1);
-            }
-        };
-    }
-    /**
-     * Temporarily disconnect from agent events.
-     * User listeners are preserved and will receive events again after resubscribe().
-     * Used internally during operations that need to pause event processing.
-     */
-    _disconnectFromAgent() {
-        if (this._unsubscribeAgent) {
-            this._unsubscribeAgent();
-            this._unsubscribeAgent = undefined;
-        }
-    }
-    /**
-     * Reconnect to agent events after _disconnectFromAgent().
-     * Preserves all existing listeners.
-     */
-    _reconnectToAgent() {
-        if (this._unsubscribeAgent)
-            return; // Already connected
-        this._unsubscribeAgent = this.agent.subscribe(this._handleAgentEvent);
-    }
-    /**
-     * Remove all listeners and disconnect from agent.
-     * Call this when completely done with the session.
-     */
-    dispose() {
-        try {
-            this.abortRetry();
-            this.abortCompaction();
-            this.abortBranchSummary();
-            this.abortBash();
-            this.agent.abort();
-        }
-        catch {
-            // Dispose must succeed even if an abort hook throws.
-        }
-        this._extensionRunner.invalidate("This extension ctx is stale after session replacement or reload. Do not use a captured pi or command ctx after ctx.newSession(), ctx.fork(), ctx.switchSession(), or ctx.reload(). For newSession, fork, and switchSession, move post-replacement work into withSession and use the ctx passed to withSession. For reload, do not use the old ctx after await ctx.reload().");
-        this._disconnectFromAgent();
-        this._eventListeners = [];
-        cleanupSessionResources(this.sessionId);
-    }
-    // =========================================================================
-    // Read-only State Access
-    // =========================================================================
-    /** Full agent state */
-    get state() {
-        return this.agent.state;
-    }
-    /** Current model (may be undefined if not yet selected) */
-    get model() {
-        return this.agent.state.model;
-    }
-    /** Current thinking level */
-    get thinkingLevel() {
-        return this.agent.state.thinkingLevel;
-    }
-    /** Whether agent is currently streaming a response */
-    get isStreaming() {
-        return this.agent.state.isStreaming;
-    }
-    /** Current effective system prompt (includes any per-turn extension modifications) */
-    get systemPrompt() {
-        return this.agent.state.systemPrompt;
-    }
-    /** Current retry attempt (0 if not retrying) */
-    get retryAttempt() {
-        return this._retryAttempt;
-    }
-    /**
-     * Get the names of currently active tools.
-     * Returns the names of tools currently set on the agent.
-     */
-    getActiveToolNames() {
-        return this.agent.state.tools.map((t) => t.name);
-    }
-    /**
-     * Get all configured tools with name, description, parameter schema, prompt guidelines, and source metadata.
-     */
-    getAllTools() {
-        return Array.from(this._toolDefinitions.values()).map(({ definition, sourceInfo }) => ({
-            name: definition.name,
-            description: definition.description,
-            parameters: definition.parameters,
-            promptGuidelines: definition.promptGuidelines,
-            sourceInfo,
-        }));
-    }
-    getToolDefinition(name) {
-        return this._toolDefinitions.get(name)?.definition;
-    }
-    /**
-     * Set active tools by name.
-     * Only tools in the registry can be enabled. Unknown tool names are ignored.
-     * Also rebuilds the system prompt to reflect the new tool set.
-     * Changes take effect on the next agent turn.
-     */
-    setActiveToolsByName(toolNames) {
-        const tools = [];
-        const validToolNames = [];
-        for (const name of toolNames) {
-            const tool = this._toolRegistry.get(name);
-            if (tool) {
-                tools.push(tool);
-                validToolNames.push(name);
-            }
-        }
-        this.agent.state.tools = tools;
-        // Rebuild base system prompt with new tool set
-        this._baseSystemPrompt = this._rebuildSystemPrompt(validToolNames);
-        this.agent.state.systemPrompt = this._baseSystemPrompt;
-    }
-    /** Whether compaction or branch summarization is currently running */
-    get isCompacting() {
-        return (this._autoCompactionAbortController !== undefined ||
-            this._compactionAbortController !== undefined ||
-            this._branchSummaryAbortController !== undefined);
-    }
-    /** All messages including custom types like BashExecutionMessage */
-    get messages() {
-        return this.agent.state.messages;
-    }
-    /** Current steering mode */
-    get steeringMode() {
-        return this.agent.steeringMode;
-    }
-    /** Current follow-up mode */
-    get followUpMode() {
-        return this.agent.followUpMode;
-    }
-    /** Current session file path, or undefined if sessions are disabled */
-    get sessionFile() {
-        return this.sessionManager.getSessionFile();
-    }
-    /** Current session ID */
-    get sessionId() {
-        return this.sessionManager.getSessionId();
-    }
-    /** Current session display name, if set */
-    get sessionName() {
-        return this.sessionManager.getSessionName();
-    }
-    /** Scoped models for cycling (from --models flag) */
-    get scopedModels() {
-        return this._scopedModels;
-    }
-    /** Update scoped models for cycling */
-    setScopedModels(scopedModels) {
-        this._scopedModels = scopedModels;
-    }
-    /** File-based prompt templates */
-    get promptTemplates() {
-        return this._resourceLoader.getPrompts().prompts;
-    }
-    _normalizePromptSnippet(text) {
-        if (!text)
-            return undefined;
-        const oneLine = text
-            .replace(/[\r\n]+/g, " ")
-            .replace(/\s+/g, " ")
-            .trim();
-        return oneLine.length > 0 ? oneLine : undefined;
-    }
-    _normalizePromptGuidelines(guidelines) {
-        if (!guidelines || guidelines.length === 0) {
-            return [];
-        }
-        const unique = new Set();
-        for (const guideline of guidelines) {
-            const normalized = guideline.trim();
-            if (normalized.length > 0) {
-                unique.add(normalized);
-            }
-        }
-        return Array.from(unique);
-    }
-    _rebuildSystemPrompt(toolNames) {
-        const validToolNames = toolNames.filter((name) => this._toolRegistry.has(name));
-        const toolSnippets = {};
-        const promptGuidelines = [];
-        for (const name of validToolNames) {
-            const snippet = this._toolPromptSnippets.get(name);
-            if (snippet) {
-                toolSnippets[name] = snippet;
-            }
-            const toolGuidelines = this._toolPromptGuidelines.get(name);
-            if (toolGuidelines) {
-                promptGuidelines.push(...toolGuidelines);
-            }
-        }
-        const loaderSystemPrompt = this._resourceLoader.getSystemPrompt();
-        const loaderAppendSystemPrompt = this._resourceLoader.getAppendSystemPrompt();
-        const appendSystemPrompt = loaderAppendSystemPrompt.length > 0 ? loaderAppendSystemPrompt.join("\n\n") : undefined;
-        const loadedSkills = this._resourceLoader.getSkills().skills;
-        const loadedContextFiles = this._resourceLoader.getAgentsFiles().agentsFiles;
-        this._baseSystemPromptOptions = {
-            cwd: this._cwd,
-            skills: loadedSkills,
-            contextFiles: loadedContextFiles,
-            customPrompt: loaderSystemPrompt,
-            appendSystemPrompt,
-            selectedTools: validToolNames,
-            toolSnippets,
-            promptGuidelines,
-        };
-        return buildSystemPrompt(this._baseSystemPromptOptions);
-    }
-    // =========================================================================
-    // Prompting
-    // =========================================================================
-    async _runAgentPrompt(messages) {
-        try {
-            await this.agent.prompt(messages);
-            while (await this._handlePostAgentRun()) {
-                await this.agent.continue();
-            }
-        }
-        finally {
-            this._flushPendingBashMessages();
-        }
-    }
-    async _handlePostAgentRun() {
-        const msg = this._lastAssistantMessage;
-        this._lastAssistantMessage = undefined;
-        if (!msg) {
-            return false;
-        }
-        if (this._isRetryableError(msg) && (await this._prepareRetry(msg))) {
-            return true;
-        }
-        if (msg.stopReason === "error" && this._retryAttempt > 0) {
-            this._emit({
-                type: "auto_retry_end",
-                success: false,
-                attempt: this._retryAttempt,
-                finalError: msg.errorMessage,
-            });
-            this._retryAttempt = 0;
-        }
-        if (await this._checkCompaction(msg)) {
-            return true;
-        }
-        // The agent loop drains both queues before emitting agent_end. Any messages
-        // here were queued by agent_end extension handlers and need a continuation.
-        return this.agent.hasQueuedMessages();
-    }
-    /**
-     * Send a prompt to the agent.
-     * - Handles extension commands (registered via pi.registerCommand) immediately, even during streaming
-     * - Expands file-based prompt templates by default
-     * - During streaming, queues via steer() or followUp() based on streamingBehavior option
-     * - Validates model and API key before sending (when not streaming)
-     * @throws Error if streaming and no streamingBehavior specified
-     * @throws Error if no model selected or no API key available (when not streaming)
-     */
-    async prompt(text, options) {
-        const expandPromptTemplates = options?.expandPromptTemplates ?? true;
-        const preflightResult = options?.preflightResult;
-        let messages;
-        try {
-            // Handle extension commands first (execute immediately, even during streaming)
-            // Extension commands manage their own LLM interaction via pi.sendMessage()
-            if (expandPromptTemplates && text.startsWith("/")) {
-                const handled = await this._tryExecuteExtensionCommand(text);
-                if (handled) {
-                    // Extension command executed, no prompt to send
-                    preflightResult?.(true);
-                    return;
-                }
-            }
-            // Emit input event for extension interception (before skill/template expansion)
-            let currentText = text;
-            let currentImages = options?.images;
-            if (this._extensionRunner.hasHandlers("input")) {
-                const inputResult = await this._extensionRunner.emitInput(currentText, currentImages, options?.source ?? "interactive", this.isStreaming ? options?.streamingBehavior : undefined);
-                if (inputResult.action === "handled") {
-                    preflightResult?.(true);
-                    return;
-                }
-                if (inputResult.action === "transform") {
-                    currentText = inputResult.text;
-                    currentImages = inputResult.images ?? currentImages;
-                }
-            }
-            // Expand skill commands (/skill:name args) and prompt templates (/template args)
-            let expandedText = currentText;
-            if (expandPromptTemplates) {
-                expandedText = this._expandSkillCommand(expandedText);
-                expandedText = expandPromptTemplate(expandedText, [...this.promptTemplates]);
-            }
-            // If streaming, queue via steer() or followUp() based on option
-            if (this.isStreaming) {
-                if (!options?.streamingBehavior) {
-                    throw new Error("Agent is already processing. Specify streamingBehavior ('steer' or 'followUp') to queue the message.");
-                }
-                if (options.streamingBehavior === "followUp") {
-                    await this._queueFollowUp(expandedText, currentImages);
-                }
-                else {
-                    await this._queueSteer(expandedText, currentImages);
-                }
-                preflightResult?.(true);
-                return;
-            }
-            // Flush any pending bash messages before the new prompt
-            this._flushPendingBashMessages();
-            // Validate model
-            if (!this.model) {
-                throw new Error(formatNoModelSelectedMessage());
-            }
-            if (!this._modelRegistry.hasConfiguredAuth(this.model)) {
-                const isOAuth = this._modelRegistry.isUsingOAuth(this.model);
-                if (isOAuth) {
-                    throw new Error(`Authentication failed for "${this.model.provider}". ` +
-                        `Credentials may have expired or network is unavailable. ` +
-                        `Run '/login ${this.model.provider}' to re-authenticate.`);
-                }
-                throw new Error(formatNoApiKeyFoundMessage(this.model.provider));
-            }
-            // Check if we need to compact before sending (catches aborted responses)
-            const lastAssistant = this._findLastAssistantMessage();
-            if (lastAssistant && (await this._checkCompaction(lastAssistant, false))) {
-                try {
-                    await this.agent.continue();
-                    while (await this._handlePostAgentRun()) {
-                        await this.agent.continue();
-                    }
-                }
-                finally {
-                    this._flushPendingBashMessages();
-                }
-            }
-            // Build messages array (custom message if any, then user message)
-            messages = [];
-            // Add user message
-            const userContent = [{ type: "text", text: expandedText }];
-            if (currentImages) {
-                userContent.push(...currentImages);
-            }
-            messages.push({
-                role: "user",
-                content: userContent,
-                timestamp: Date.now(),
-            });
-            // Inject any pending "nextTurn" messages as context alongside the user message
-            for (const msg of this._pendingNextTurnMessages) {
-                messages.push(msg);
-            }
-            this._pendingNextTurnMessages = [];
-            // Emit before_agent_start extension event
-            const result = await this._extensionRunner.emitBeforeAgentStart(expandedText, currentImages, this._baseSystemPrompt, this._baseSystemPromptOptions);
-            // Add all custom messages from extensions
-            if (result?.messages) {
-                for (const msg of result.messages) {
-                    messages.push({
-                        role: "custom",
-                        customType: msg.customType,
-                        content: msg.content,
-                        display: msg.display,
-                        details: msg.details,
-                        timestamp: Date.now(),
-                    });
-                }
-            }
-            // Apply extension-modified system prompt, or reset to base
-            if (result?.systemPrompt) {
-                this.agent.state.systemPrompt = result.systemPrompt;
-            }
-            else {
-                // Ensure we're using the base prompt (in case previous turn had modifications)
-                this.agent.state.systemPrompt = this._baseSystemPrompt;
-            }
-        }
-        catch (error) {
-            preflightResult?.(false);
-            throw error;
-        }
-        if (!messages) {
-            return;
-        }
-        preflightResult?.(true);
-        await this._runAgentPrompt(messages);
-    }
-    /**
-     * Try to execute an extension command. Returns true if command was found and executed.
-     */
-    async _tryExecuteExtensionCommand(text) {
-        // Parse command name and args
-        const spaceIndex = text.indexOf(" ");
-        const commandName = spaceIndex === -1 ? text.slice(1) : text.slice(1, spaceIndex);
-        const args = spaceIndex === -1 ? "" : text.slice(spaceIndex + 1);
-        const command = this._extensionRunner.getCommand(commandName);
-        if (!command)
-            return false;
-        // Get command context from extension runner (includes session control methods)
-        const ctx = this._extensionRunner.createCommandContext();
-        try {
-            await command.handler(args, ctx);
-            return true;
-        }
-        catch (err) {
-            // Emit error via extension runner
-            this._extensionRunner.emitError({
-                extensionPath: `command:${commandName}`,
-                event: "command",
-                error: err instanceof Error ? err.message : String(err),
-            });
-            return true;
-        }
-    }
-    /**
-     * Expand skill commands (/skill:name args) to their full content.
-     * Returns the expanded text, or the original text if not a skill command or skill not found.
-     * Emits errors via extension runner if file read fails.
-     */
-    _expandSkillCommand(text) {
-        if (!text.startsWith("/skill:"))
-            return text;
-        const spaceIndex = text.indexOf(" ");
-        const skillName = spaceIndex === -1 ? text.slice(7) : text.slice(7, spaceIndex);
-        const args = spaceIndex === -1 ? "" : text.slice(spaceIndex + 1).trim();
-        const skill = this.resourceLoader.getSkills().skills.find((s) => s.name === skillName);
-        if (!skill)
-            return text; // Unknown skill, pass through
-        try {
-            const content = readFileSync(skill.filePath, "utf-8");
-            const body = stripFrontmatter(content).trim();
-            const skillBlock = `<skill name="${skill.name}" location="${skill.filePath}">\nReferences are relative to ${skill.baseDir}.\n\n${body}\n</skill>`;
-            return args ? `${skillBlock}\n\n${args}` : skillBlock;
-        }
-        catch (err) {
-            // Emit error like extension commands do
-            this._extensionRunner.emitError({
-                extensionPath: skill.filePath,
-                event: "skill_expansion",
-                error: err instanceof Error ? err.message : String(err),
-            });
-            return text; // Return original on error
-        }
-    }
-    /**
-     * Queue a steering message while the agent is running.
-     * Delivered after the current assistant turn finishes executing its tool calls,
-     * before the next LLM call.
-     * Expands skill commands and prompt templates. Errors on extension commands.
-     * @param images Optional image attachments to include with the message
-     * @throws Error if text is an extension command
-     */
-    async steer(text, images) {
-        // Check for extension commands (cannot be queued)
-        if (text.startsWith("/")) {
-            this._throwIfExtensionCommand(text);
-        }
-        // Expand skill commands and prompt templates
-        let expandedText = this._expandSkillCommand(text);
-        expandedText = expandPromptTemplate(expandedText, [...this.promptTemplates]);
-        await this._queueSteer(expandedText, images);
-    }
-    /**
-     * Queue a follow-up message to be processed after the agent finishes.
-     * Delivered only when agent has no more tool calls or steering messages.
-     * Expands skill commands and prompt templates. Errors on extension commands.
-     * @param images Optional image attachments to include with the message
-     * @throws Error if text is an extension command
-     */
-    async followUp(text, images) {
-        // Check for extension commands (cannot be queued)
-        if (text.startsWith("/")) {
-            this._throwIfExtensionCommand(text);
-        }
-        // Expand skill commands and prompt templates
-        let expandedText = this._expandSkillCommand(text);
-        expandedText = expandPromptTemplate(expandedText, [...this.promptTemplates]);
-        await this._queueFollowUp(expandedText, images);
-    }
-    /**
-     * Internal: Queue a steering message (already expanded, no extension command check).
-     */
-    async _queueSteer(text, images) {
-        this._steeringMessages.push(text);
-        this._emitQueueUpdate();
-        const content = [{ type: "text", text }];
-        if (images) {
-            content.push(...images);
-        }
-        this.agent.steer({
-            role: "user",
-            content,
-            timestamp: Date.now(),
-        });
-    }
-    /**
-     * Internal: Queue a follow-up message (already expanded, no extension command check).
-     */
-    async _queueFollowUp(text, images) {
-        this._followUpMessages.push(text);
-        this._emitQueueUpdate();
-        const content = [{ type: "text", text }];
-        if (images) {
-            content.push(...images);
-        }
-        this.agent.followUp({
-            role: "user",
-            content,
-            timestamp: Date.now(),
-        });
-    }
-    /**
-     * Throw an error if the text is an extension command.
-     */
-    _throwIfExtensionCommand(text) {
-        const spaceIndex = text.indexOf(" ");
-        const commandName = spaceIndex === -1 ? text.slice(1) : text.slice(1, spaceIndex);
-        const command = this._extensionRunner.getCommand(commandName);
-        if (command) {
-            throw new Error(`Extension command "/${commandName}" cannot be queued. Use prompt() or execute the command when not streaming.`);
-        }
-    }
-    /**
-     * Send a custom message to the session. Creates a CustomMessageEntry.
-     *
-     * Handles three cases:
-     * - Streaming: queues message, processed when loop pulls from queue
-     * - Not streaming + triggerTurn: appends to state/session, starts new turn
-     * - Not streaming + no trigger: appends to state/session, no turn
-     *
-     * @param message Custom message with customType, content, display, details
-     * @param options.triggerTurn If true and not streaming, triggers a new LLM turn
-     * @param options.deliverAs Delivery mode: "steer", "followUp", or "nextTurn"
-     */
-    async sendCustomMessage(message, options) {
-        const appMessage = {
-            role: "custom",
-            customType: message.customType,
-            content: message.content,
-            display: message.display,
-            details: message.details,
-            timestamp: Date.now(),
-        };
-        if (options?.deliverAs === "nextTurn") {
-            this._pendingNextTurnMessages.push(appMessage);
-        }
-        else if (this.isStreaming) {
-            if (options?.deliverAs === "followUp") {
-                this.agent.followUp(appMessage);
-            }
-            else {
-                this.agent.steer(appMessage);
-            }
-        }
-        else if (options?.triggerTurn) {
-            await this._runAgentPrompt(appMessage);
-        }
-        else {
-            this.agent.state.messages.push(appMessage);
-            this.sessionManager.appendCustomMessageEntry(message.customType, message.content, message.display, message.details);
-            this._emit({ type: "message_start", message: appMessage });
-            this._emit({ type: "message_end", message: appMessage });
-        }
-    }
-    /**
-     * Send a user message to the agent. Always triggers a turn.
-     * When the agent is streaming, use deliverAs to specify how to queue the message.
-     *
-     * @param content User message content (string or content array)
-     * @param options.deliverAs Delivery mode when streaming: "steer" or "followUp"
-     */
-    async sendUserMessage(content, options) {
-        // Normalize content to text string + optional images
-        let text;
-        let images;
-        if (typeof content === "string") {
-            text = content;
-        }
-        else {
-            const textParts = [];
-            images = [];
-            for (const part of content) {
-                if (part.type === "text") {
-                    textParts.push(part.text);
-                }
-                else {
-                    images.push(part);
-                }
-            }
-            text = textParts.join("\n");
-            if (images.length === 0)
-                images = undefined;
-        }
-        // Use prompt() with expandPromptTemplates: false to skip command handling and template expansion
-        await this.prompt(text, {
-            expandPromptTemplates: false,
-            streamingBehavior: options?.deliverAs,
-            images,
-            source: "extension",
-        });
-    }
-    /**
-     * Clear all queued messages and return them.
-     * Useful for restoring to editor when user aborts.
-     * @returns Object with steering and followUp arrays
-     */
-    clearQueue() {
-        const steering = [...this._steeringMessages];
-        const followUp = [...this._followUpMessages];
-        this._steeringMessages = [];
-        this._followUpMessages = [];
-        this.agent.clearAllQueues();
-        this._emitQueueUpdate();
-        return { steering, followUp };
-    }
-    /** Number of pending messages (includes both steering and follow-up) */
-    get pendingMessageCount() {
-        return this._steeringMessages.length + this._followUpMessages.length;
-    }
-    /** Get pending steering messages (read-only) */
-    getSteeringMessages() {
-        return this._steeringMessages;
-    }
-    /** Get pending follow-up messages (read-only) */
-    getFollowUpMessages() {
-        return this._followUpMessages;
-    }
-    get resourceLoader() {
-        return this._resourceLoader;
-    }
-    /**
-     * Abort current operation and wait for agent to become idle.
-     */
-    async abort() {
-        this.abortRetry();
-        this.agent.abort();
-        await this.agent.waitForIdle();
-    }
-    // =========================================================================
-    // Model Management
-    // =========================================================================
-    async _emitModelSelect(nextModel, previousModel, source) {
-        if (modelsAreEqual(previousModel, nextModel))
-            return;
-        await this._extensionRunner.emit({
-            type: "model_select",
-            model: nextModel,
-            previousModel,
-            source,
-        });
-    }
-    /**
-     * Set model directly.
-     * Validates that auth is configured, saves to session and settings.
-     * @throws Error if no auth is configured for the model
-     */
-    async setModel(model) {
-        if (!this._modelRegistry.hasConfiguredAuth(model)) {
-            throw new Error(`No API key for ${model.provider}/${model.id}`);
-        }
-        const previousModel = this.model;
-        const thinkingLevel = this._getThinkingLevelForModelSwitch(model);
-        this.agent.state.model = model;
-        this.sessionManager.appendModelChange(model.provider, model.id);
-        this.settingsManager.setDefaultModelAndProvider(model.provider, model.id);
-        // Re-clamp thinking level for new model's capabilities
-        this.setThinkingLevel(thinkingLevel);
-        await this._emitModelSelect(model, previousModel, "set");
-    }
-    /**
-     * Cycle to next/previous model.
-     * Uses scoped models (from --models flag) if available, otherwise all available models.
-     * @param direction - "forward" (default) or "backward"
-     * @returns The new model info, or undefined if only one model available
-     */
-    async cycleModel(direction = "forward") {
-        if (this._scopedModels.length > 0) {
-            return this._cycleScopedModel(direction);
-        }
-        return this._cycleAvailableModel(direction);
-    }
-    async _cycleScopedModel(direction) {
-        const scopedModels = this._scopedModels.filter((scoped) => this._modelRegistry.hasConfiguredAuth(scoped.model));
-        if (scopedModels.length <= 1)
-            return undefined;
-        const currentModel = this.model;
-        let currentIndex = scopedModels.findIndex((sm) => modelsAreEqual(sm.model, currentModel));
-        if (currentIndex === -1)
-            currentIndex = 0;
-        const len = scopedModels.length;
-        const nextIndex = direction === "forward" ? (currentIndex + 1) % len : (currentIndex - 1 + len) % len;
-        const next = scopedModels[nextIndex];
-        const thinkingLevel = this._getThinkingLevelForModelSwitch(next.model, next.thinkingLevel);
-        // Apply model
-        this.agent.state.model = next.model;
-        this.sessionManager.appendModelChange(next.model.provider, next.model.id);
-        this.settingsManager.setDefaultModelAndProvider(next.model.provider, next.model.id);
-        // Apply thinking level.
-        // - Explicit scoped model thinking level overrides current session level
-        // - Undefined scoped model thinking level inherits the current session preference
-        // setThinkingLevel clamps to model capabilities.
-        this.setThinkingLevel(thinkingLevel);
-        await this._emitModelSelect(next.model, currentModel, "cycle");
-        return { model: next.model, thinkingLevel: this.thinkingLevel, isScoped: true };
-    }
-    async _cycleAvailableModel(direction) {
-        const availableModels = await this._modelRegistry.getAvailable();
-        if (availableModels.length <= 1)
-            return undefined;
-        const currentModel = this.model;
-        let currentIndex = availableModels.findIndex((m) => modelsAreEqual(m, currentModel));
-        if (currentIndex === -1)
-            currentIndex = 0;
-        const len = availableModels.length;
-        const nextIndex = direction === "forward" ? (currentIndex + 1) % len : (currentIndex - 1 + len) % len;
-        const nextModel = availableModels[nextIndex];
-        const thinkingLevel = this._getThinkingLevelForModelSwitch(nextModel);
-        this.agent.state.model = nextModel;
-        this.sessionManager.appendModelChange(nextModel.provider, nextModel.id);
-        this.settingsManager.setDefaultModelAndProvider(nextModel.provider, nextModel.id);
-        // Re-clamp thinking level for new model's capabilities
-        this.setThinkingLevel(thinkingLevel);
-        await this._emitModelSelect(nextModel, currentModel, "cycle");
-        return { model: nextModel, thinkingLevel: this.thinkingLevel, isScoped: false };
-    }
-    // =========================================================================
-    // Thinking Level Management
-    // =========================================================================
-    /**
-     * Set thinking level.
-     * Clamps to model capabilities based on available thinking levels.
-     * Saves to session and settings only if the level actually changes.
-     */
-    setThinkingLevel(level) {
-        const availableLevels = this.getAvailableThinkingLevels();
-        const effectiveLevel = availableLevels.includes(level) ? level : this._clampThinkingLevel(level, availableLevels);
-        // Only persist if actually changing
-        const previousLevel = this.agent.state.thinkingLevel;
-        const isChanging = effectiveLevel !== previousLevel;
-        this.agent.state.thinkingLevel = effectiveLevel;
-        if (isChanging) {
-            this.sessionManager.appendThinkingLevelChange(effectiveLevel);
-            if (this.supportsThinking() || effectiveLevel !== "off") {
-                this.settingsManager.setDefaultThinkingLevel(effectiveLevel);
-            }
-            this._emit({ type: "thinking_level_changed", level: effectiveLevel });
-            void this._extensionRunner.emit({
-                type: "thinking_level_select",
-                level: effectiveLevel,
-                previousLevel,
-            });
-        }
-    }
-    /**
-     * Cycle to next thinking level.
-     * @returns New level, or undefined if model doesn't support thinking
-     */
-    cycleThinkingLevel() {
-        if (!this.supportsThinking())
-            return undefined;
-        const levels = this.getAvailableThinkingLevels();
-        const currentIndex = levels.indexOf(this.thinkingLevel);
-        const nextIndex = (currentIndex + 1) % levels.length;
-        const nextLevel = levels[nextIndex];
-        this.setThinkingLevel(nextLevel);
-        return nextLevel;
-    }
-    /**
-     * Get available thinking levels for current model.
-     * The provider will clamp to what the specific model supports internally.
-     */
-    getAvailableThinkingLevels() {
-        if (!this.model)
-            return THINKING_LEVELS;
-        return getSupportedThinkingLevels(this.model);
-    }
-    /**
-     * Check if current model supports thinking/reasoning.
-     */
-    supportsThinking() {
-        return !!this.model?.reasoning;
-    }
-    _getThinkingLevelForModelSwitch(targetModel, explicitLevel) {
-        if (explicitLevel !== undefined) {
-            return explicitLevel;
-        }
-        const modelThinkingLevel = targetModel ? this.settingsManager.getModelThinkingLevel(targetModel) : undefined;
-        if (modelThinkingLevel) {
-            return modelThinkingLevel;
-        }
-        if (!this.supportsThinking()) {
-            return this.settingsManager.getDefaultThinkingLevel() ?? DEFAULT_THINKING_LEVEL;
-        }
-        return this.thinkingLevel;
-    }
-    _clampThinkingLevel(level, _availableLevels) {
-        return this.model ? clampThinkingLevel(this.model, level) : "off";
-    }
-    // =========================================================================
-    // Queue Mode Management
-    // =========================================================================
-    /**
-     * Set steering message mode.
-     * Saves to settings.
-     */
-    setSteeringMode(mode) {
-        this.agent.steeringMode = mode;
-        this.settingsManager.setSteeringMode(mode);
-    }
-    /**
-     * Set follow-up message mode.
-     * Saves to settings.
-     */
-    setFollowUpMode(mode) {
-        this.agent.followUpMode = mode;
-        this.settingsManager.setFollowUpMode(mode);
-    }
-    // =========================================================================
-    // Compaction
-    // =========================================================================
-    /**
-     * Manually compact the session context.
-     * Aborts current agent operation first.
-     * @param customInstructions Optional instructions for the compaction summary
-     */
-    async compact(customInstructions) {
-        this._disconnectFromAgent();
-        await this.abort();
-        this._compactionAbortController = new AbortController();
-        this._emit({ type: "compaction_start", reason: "manual" });
-        try {
-            if (!this.model) {
-                throw new Error(formatNoModelSelectedMessage());
-            }
-            const { apiKey, headers } = await this._getCompactionRequestAuth(this.model);
-            const pathEntries = this.sessionManager.getBranch();
-            const settings = this.settingsManager.getCompactionSettings();
-            const preparation = prepareCompaction(pathEntries, settings);
-            if (!preparation) {
-                // Check why we can't compact
-                const lastEntry = pathEntries[pathEntries.length - 1];
-                if (lastEntry?.type === "compaction") {
-                    throw new Error("Already compacted");
-                }
-                throw new Error("Nothing to compact (session too small)");
-            }
-            let extensionCompaction;
-            let fromExtension = false;
-            if (this._extensionRunner.hasHandlers("session_before_compact")) {
-                const result = (await this._extensionRunner.emit({
-                    type: "session_before_compact",
-                    preparation,
-                    branchEntries: pathEntries,
-                    customInstructions,
-                    signal: this._compactionAbortController.signal,
-                }));
-                if (result?.cancel) {
-                    throw new Error("Compaction cancelled");
-                }
-                if (result?.compaction) {
-                    extensionCompaction = result.compaction;
-                    fromExtension = true;
-                }
-            }
-            let summary;
-            let firstKeptEntryId;
-            let tokensBefore;
-            let details;
-            if (extensionCompaction) {
-                // Extension provided compaction content
-                summary = extensionCompaction.summary;
-                firstKeptEntryId = extensionCompaction.firstKeptEntryId;
-                tokensBefore = extensionCompaction.tokensBefore;
-                details = extensionCompaction.details;
-            }
-            else {
-                // Generate compaction result
-                const result = await compact(preparation, this.model, apiKey, headers, customInstructions, this._compactionAbortController.signal, this.thinkingLevel, this.agent.streamFn);
-                summary = result.summary;
-                firstKeptEntryId = result.firstKeptEntryId;
-                tokensBefore = result.tokensBefore;
-                details = result.details;
-            }
-            if (this._compactionAbortController.signal.aborted) {
-                throw new Error("Compaction cancelled");
-            }
-            this.sessionManager.appendCompaction(summary, firstKeptEntryId, tokensBefore, details, fromExtension);
-            const newEntries = this.sessionManager.getEntries();
-            const sessionContext = this.sessionManager.buildSessionContext();
-            this.agent.state.messages = sessionContext.messages;
-            // Get the saved compaction entry for the extension event
-            const savedCompactionEntry = newEntries.find((e) => e.type === "compaction" && e.summary === summary);
-            if (this._extensionRunner && savedCompactionEntry) {
-                await this._extensionRunner.emit({
-                    type: "session_compact",
-                    compactionEntry: savedCompactionEntry,
-                    fromExtension,
-                });
-            }
-            const compactionResult = {
-                summary,
-                firstKeptEntryId,
-                tokensBefore,
-                details,
-            };
-            this._emit({
-                type: "compaction_end",
-                reason: "manual",
-                result: compactionResult,
-                aborted: false,
-                willRetry: false,
-            });
-            return compactionResult;
-        }
-        catch (error) {
-            const message = error instanceof Error ? error.message : String(error);
-            const aborted = message === "Compaction cancelled" || (error instanceof Error && error.name === "AbortError");
-            this._emit({
-                type: "compaction_end",
-                reason: "manual",
-                result: undefined,
-                aborted,
-                willRetry: false,
-                errorMessage: aborted ? undefined : `Compaction failed: ${message}`,
-            });
-            throw error;
-        }
-        finally {
-            this._compactionAbortController = undefined;
-            this._reconnectToAgent();
-        }
-    }
-    /**
-     * Cancel in-progress compaction (manual or auto).
-     */
-    abortCompaction() {
-        this._compactionAbortController?.abort();
-        this._autoCompactionAbortController?.abort();
-    }
-    /**
-     * Cancel in-progress branch summarization.
-     */
-    abortBranchSummary() {
-        this._branchSummaryAbortController?.abort();
-    }
-    /**
-     * Check if compaction is needed and run it.
-     * Called after agent_end and before prompt submission.
-     *
-     * Two cases:
-     * 1. Overflow: LLM returned context overflow error, remove error message from agent state, compact, auto-retry
-     * 2. Threshold: Context over threshold, compact, NO auto-retry (user continues manually)
-     *
-     * @param assistantMessage The assistant message to check
-     * @param skipAbortedCheck If false, include aborted messages (for pre-prompt check). Default: true
-     */
-    async _checkCompaction(assistantMessage, skipAbortedCheck = true) {
-        const settings = this.settingsManager.getCompactionSettings();
-        if (!settings.enabled)
-            return false;
-        // Skip if message was aborted (user cancelled) - unless skipAbortedCheck is false
-        if (skipAbortedCheck && assistantMessage.stopReason === "aborted")
-            return false;
-        const contextWindow = this.model?.contextWindow ?? 0;
-        // Skip overflow check if the message came from a different model.
-        // This handles the case where user switched from a smaller-context model (e.g. opus)
-        // to a larger-context model (e.g. codex) - the overflow error from the old model
-        // shouldn't trigger compaction for the new model.
-        const sameModel = this.model && assistantMessage.provider === this.model.provider && assistantMessage.model === this.model.id;
-        // Skip compaction checks if this assistant message is older than the latest
-        // compaction boundary. This prevents a stale pre-compaction usage/error
-        // from retriggering compaction on the first prompt after compaction.
-        const compactionEntry = getLatestCompactionEntry(this.sessionManager.getBranch());
-        const assistantIsFromBeforeCompaction = compactionEntry !== null && assistantMessage.timestamp <= new Date(compactionEntry.timestamp).getTime();
-        if (assistantIsFromBeforeCompaction) {
-            return false;
-        }
-        // Case 1: Overflow - LLM returned context overflow error
-        if (sameModel && isContextOverflow(assistantMessage, contextWindow)) {
-            if (this._overflowRecoveryAttempted) {
-                this._emit({
-                    type: "compaction_end",
-                    reason: "overflow",
-                    result: undefined,
-                    aborted: false,
-                    willRetry: false,
-                    errorMessage: "Context overflow recovery failed after one compact-and-retry attempt. Try reducing context or switching to a larger-context model.",
-                });
-                return false;
-            }
-            this._overflowRecoveryAttempted = true;
-            // Remove the error message from agent state (it IS saved to session for history,
-            // but we don't want it in context for the retry)
-            const messages = this.agent.state.messages;
-            if (messages.length > 0 && messages[messages.length - 1].role === "assistant") {
-                this.agent.state.messages = messages.slice(0, -1);
-            }
-            return await this._runAutoCompaction("overflow", true);
-        }
-        // Case 2: Threshold - context is getting large
-        // For error messages (no usage data), estimate from last successful response.
-        // This ensures sessions that hit persistent API errors (e.g. 529) can still compact.
-        let contextTokens;
-        if (assistantMessage.stopReason === "error") {
-            const messages = this.agent.state.messages;
-            const estimate = estimateContextTokens(messages);
-            if (estimate.lastUsageIndex === null)
-                return false; // No usage data at all
-            // Verify the usage source is post-compaction. Kept pre-compaction messages
-            // have stale usage reflecting the old (larger) context and would falsely
-            // trigger compaction right after one just finished.
-            const usageMsg = messages[estimate.lastUsageIndex];
-            if (compactionEntry &&
-                usageMsg.role === "assistant" &&
-                usageMsg.timestamp <= new Date(compactionEntry.timestamp).getTime()) {
-                return false;
-            }
-            contextTokens = estimate.tokens;
-        }
-        else {
-            contextTokens = calculateContextTokens(assistantMessage.usage);
-        }
-        // Adaptive headroom threshold (default 90%); falls back to the compaction default when disabled.
-        const headroomThreshold = resolveAdaptiveHeadroomThreshold(process.env);
-        if (shouldCompact(contextTokens, contextWindow, settings, headroomThreshold > 0 ? headroomThreshold : undefined)) {
-            return await this._runAutoCompaction("threshold", false);
-        }
-        return false;
-    }
-    /**
-     * Internal: Run auto-compaction with events.
-     */
-    async _runAutoCompaction(reason, willRetry) {
-        const settings = this.settingsManager.getCompactionSettings();
-        this._emit({ type: "compaction_start", reason });
-        this._autoCompactionAbortController = new AbortController();
-        try {
-            if (!this.model) {
-                this._emit({
-                    type: "compaction_end",
-                    reason,
-                    result: undefined,
-                    aborted: false,
-                    willRetry: false,
-                });
-                return false;
-            }
-            let apiKey;
-            let headers;
-            if (this.agent.streamFn === streamSimple) {
-                const authResult = await this._modelRegistry.getApiKeyAndHeaders(this.model);
-                if (!authResult.ok || !authResult.apiKey) {
-                    this._emit({
-                        type: "compaction_end",
-                        reason,
-                        result: undefined,
-                        aborted: false,
-                        willRetry: false,
-                    });
-                    return false;
-                }
-                apiKey = authResult.apiKey;
-                headers = authResult.headers;
-            }
-            else {
-                ({ apiKey, headers } = await this._getCompactionRequestAuth(this.model));
-            }
-            const pathEntries = this.sessionManager.getBranch();
-            const preparation = prepareCompaction(pathEntries, settings);
-            if (!preparation) {
-                this._emit({
-                    type: "compaction_end",
-                    reason,
-                    result: undefined,
-                    aborted: false,
-                    willRetry: false,
-                });
-                return false;
-            }
-            let extensionCompaction;
-            let fromExtension = false;
-            if (this._extensionRunner.hasHandlers("session_before_compact")) {
-                const extensionResult = (await this._extensionRunner.emit({
-                    type: "session_before_compact",
-                    preparation,
-                    branchEntries: pathEntries,
-                    customInstructions: undefined,
-                    signal: this._autoCompactionAbortController.signal,
-                }));
-                if (extensionResult?.cancel) {
-                    this._emit({
-                        type: "compaction_end",
-                        reason,
-                        result: undefined,
-                        aborted: true,
-                        willRetry: false,
-                    });
-                    return false;
-                }
-                if (extensionResult?.compaction) {
-                    extensionCompaction = extensionResult.compaction;
-                    fromExtension = true;
-                }
-            }
-            let summary;
-            let firstKeptEntryId;
-            let tokensBefore;
-            let details;
-            if (extensionCompaction) {
-                // Extension provided compaction content
-                summary = extensionCompaction.summary;
-                firstKeptEntryId = extensionCompaction.firstKeptEntryId;
-                tokensBefore = extensionCompaction.tokensBefore;
-                details = extensionCompaction.details;
-            }
-            else {
-                // Generate compaction result
-                const compactResult = await compact(preparation, this.model, apiKey, headers, undefined, this._autoCompactionAbortController.signal, this.thinkingLevel, this.agent.streamFn);
-                summary = compactResult.summary;
-                firstKeptEntryId = compactResult.firstKeptEntryId;
-                tokensBefore = compactResult.tokensBefore;
-                details = compactResult.details;
-            }
-            if (this._autoCompactionAbortController.signal.aborted) {
-                this._emit({
-                    type: "compaction_end",
-                    reason,
-                    result: undefined,
-                    aborted: true,
-                    willRetry: false,
-                });
-                return false;
-            }
-            this.sessionManager.appendCompaction(summary, firstKeptEntryId, tokensBefore, details, fromExtension);
-            const newEntries = this.sessionManager.getEntries();
-            const sessionContext = this.sessionManager.buildSessionContext();
-            this.agent.state.messages = sessionContext.messages;
-            // Get the saved compaction entry for the extension event
-            const savedCompactionEntry = newEntries.find((e) => e.type === "compaction" && e.summary === summary);
-            if (this._extensionRunner && savedCompactionEntry) {
-                await this._extensionRunner.emit({
-                    type: "session_compact",
-                    compactionEntry: savedCompactionEntry,
-                    fromExtension,
-                });
-            }
-            const result = {
-                summary,
-                firstKeptEntryId,
-                tokensBefore,
-                details,
-            };
-            this._emit({ type: "compaction_end", reason, result, aborted: false, willRetry });
-            if (willRetry) {
-                const messages = this.agent.state.messages;
-                const lastMsg = messages[messages.length - 1];
-                if (lastMsg?.role === "assistant" && lastMsg.stopReason === "error") {
-                    this.agent.state.messages = messages.slice(0, -1);
-                }
-                return true;
-            }
-            // Auto-compaction can complete while follow-up/steering/custom messages are waiting.
-            // Continue once so queued messages are delivered.
-            return this.agent.hasQueuedMessages();
-        }
-        catch (error) {
-            const errorMessage = error instanceof Error ? error.message : "compaction failed";
-            this._emit({
-                type: "compaction_end",
-                reason,
-                result: undefined,
-                aborted: false,
-                willRetry: false,
-                errorMessage: reason === "overflow"
-                    ? `Context overflow recovery failed: ${errorMessage}`
-                    : `Auto-compaction failed: ${errorMessage}`,
-            });
-            return false;
-        }
-        finally {
-            this._autoCompactionAbortController = undefined;
-        }
-    }
-    /**
-     * Toggle auto-compaction setting.
-     */
-    setAutoCompactionEnabled(enabled) {
-        this.settingsManager.setCompactionEnabled(enabled);
-    }
-    /** Whether auto-compaction is enabled */
-    get autoCompactionEnabled() {
-        return this.settingsManager.getCompactionEnabled();
-    }
-    async bindExtensions(bindings) {
-        if (bindings.uiContext !== undefined) {
-            this._extensionUIContext = bindings.uiContext;
-        }
-        if (bindings.mode !== undefined) {
-            this._extensionMode = bindings.mode;
-        }
-        if (bindings.commandContextActions !== undefined) {
-            this._extensionCommandContextActions = bindings.commandContextActions;
-        }
-        if (bindings.abortHandler !== undefined) {
-            this._extensionAbortHandler = bindings.abortHandler;
-        }
-        if (bindings.shutdownHandler !== undefined) {
-            this._extensionShutdownHandler = bindings.shutdownHandler;
-        }
-        if (bindings.onError !== undefined) {
-            this._extensionErrorListener = bindings.onError;
-        }
-        this._applyExtensionBindings(this._extensionRunner);
-        await this._extensionRunner.emit(this._sessionStartEvent);
-        await this.extendResourcesFromExtensions(this._sessionStartEvent.reason === "reload" ? "reload" : "startup");
-    }
-    async extendResourcesFromExtensions(reason) {
-        if (!this._extensionRunner.hasHandlers("resources_discover")) {
-            return;
-        }
-        const { skillPaths, promptPaths, themePaths } = await this._extensionRunner.emitResourcesDiscover(this._cwd, reason);
-        if (skillPaths.length === 0 && promptPaths.length === 0 && themePaths.length === 0) {
-            return;
-        }
-        const extensionPaths = {
-            skillPaths: this.buildExtensionResourcePaths(skillPaths),
-            promptPaths: this.buildExtensionResourcePaths(promptPaths),
-            themePaths: this.buildExtensionResourcePaths(themePaths),
-        };
-        this._resourceLoader.extendResources(extensionPaths);
-        this._baseSystemPrompt = this._rebuildSystemPrompt(this.getActiveToolNames());
-        this.agent.state.systemPrompt = this._baseSystemPrompt;
-    }
-    buildExtensionResourcePaths(entries) {
-        return entries.map((entry) => {
-            const source = this.getExtensionSourceLabel(entry.extensionPath);
-            const baseDir = entry.extensionPath.startsWith("<") ? undefined : dirname(entry.extensionPath);
-            return {
-                path: entry.path,
-                metadata: {
-                    source,
-                    scope: "temporary",
-                    origin: "top-level",
-                    baseDir,
-                },
-            };
-        });
-    }
-    getExtensionSourceLabel(extensionPath) {
-        if (extensionPath.startsWith("<")) {
-            return `extension:${extensionPath.replace(/[<>]/g, "")}`;
-        }
-        const base = basename(extensionPath);
-        const name = base.replace(/\.(ts|js)$/, "");
-        return `extension:${name}`;
-    }
-    _applyExtensionBindings(runner) {
-        runner.setUIContext(this._extensionUIContext, this._extensionMode);
-        runner.bindCommandContext(this._extensionCommandContextActions);
-        this._extensionErrorUnsubscriber?.();
-        this._extensionErrorUnsubscriber = this._extensionErrorListener
-            ? runner.onError(this._extensionErrorListener)
-            : undefined;
-    }
-    _refreshCurrentModelFromRegistry() {
-        const currentModel = this.model;
-        if (!currentModel) {
-            return;
-        }
-        const refreshedModel = this._modelRegistry.find(currentModel.provider, currentModel.id);
-        if (!refreshedModel || refreshedModel === currentModel) {
-            return;
-        }
-        this.agent.state.model = refreshedModel;
-    }
-    _bindExtensionCore(runner) {
-        const getCommands = () => {
-            const extensionCommands = runner.getRegisteredCommands().map((command) => ({
-                name: command.invocationName,
-                description: command.description,
-                source: "extension",
-                sourceInfo: command.sourceInfo,
-            }));
-            const templates = this.promptTemplates.map((template) => ({
-                name: template.name,
-                description: template.description,
-                source: "prompt",
-                sourceInfo: template.sourceInfo,
-            }));
-            const skills = this._resourceLoader.getSkills().skills.map((skill) => ({
-                name: `skill:${skill.name}`,
-                description: skill.description,
-                source: "skill",
-                sourceInfo: skill.sourceInfo,
-            }));
-            return [...extensionCommands, ...templates, ...skills];
-        };
-        runner.bindCore({
-            sendMessage: (message, options) => {
-                this.sendCustomMessage(message, options).catch((err) => {
-                    runner.emitError({
-                        extensionPath: "<runtime>",
-                        event: "send_message",
-                        error: err instanceof Error ? err.message : String(err),
-                    });
-                });
-            },
-            sendUserMessage: (content, options) => {
-                this.sendUserMessage(content, options).catch((err) => {
-                    runner.emitError({
-                        extensionPath: "<runtime>",
-                        event: "send_user_message",
-                        error: err instanceof Error ? err.message : String(err),
-                    });
-                });
-            },
-            appendEntry: (customType, data) => {
-                this.sessionManager.appendCustomEntry(customType, data);
-            },
-            setSessionName: (name) => {
-                this.setSessionName(name);
-            },
-            getSessionName: () => {
-                return this.sessionManager.getSessionName();
-            },
-            setLabel: (entryId, label) => {
-                this.sessionManager.appendLabelChange(entryId, label);
-            },
-            getActiveTools: () => this.getActiveToolNames(),
-            getAllTools: () => this.getAllTools(),
-            setActiveTools: (toolNames) => this.setActiveToolsByName(toolNames),
-            refreshTools: () => this._refreshToolRegistry(),
-            getCommands,
-            setModel: async (model) => {
-                if (!this.modelRegistry.hasConfiguredAuth(model))
-                    return false;
-                await this.setModel(model);
-                return true;
-            },
-            getThinkingLevel: () => this.thinkingLevel,
-            setThinkingLevel: (level) => this.setThinkingLevel(level),
-        }, {
-            getModel: () => this.model,
-            isIdle: () => !this.isStreaming,
-            getSignal: () => this.agent.signal,
-            abort: () => {
-                if (this._extensionAbortHandler) {
-                    this._extensionAbortHandler();
-                    return;
-                }
-                void this.abort();
-            },
-            hasPendingMessages: () => this.pendingMessageCount > 0,
-            shutdown: () => {
-                this._extensionShutdownHandler?.();
-            },
-            getContextUsage: () => this.getContextUsage(),
-            compact: (options) => {
-                void (async () => {
-                    try {
-                        const result = await this.compact(options?.customInstructions);
-                        options?.onComplete?.(result);
-                    }
-                    catch (error) {
-                        const err = error instanceof Error ? error : new Error(String(error));
-                        options?.onError?.(err);
-                    }
-                })();
-            },
-            getSystemPrompt: () => this.systemPrompt,
-            getSystemPromptOptions: () => this._baseSystemPromptOptions,
-        }, {
-            registerProvider: (name, config) => {
-                this._modelRegistry.registerProvider(name, config);
-                this._refreshCurrentModelFromRegistry();
-            },
-            unregisterProvider: (name) => {
-                this._modelRegistry.unregisterProvider(name);
-                this._refreshCurrentModelFromRegistry();
-            },
-        });
-    }
-    _refreshToolRegistry(options) {
-        const previousRegistryNames = new Set(this._toolRegistry.keys());
-        const previousActiveToolNames = this.getActiveToolNames();
-        const allowedToolNames = this._allowedToolNames;
-        const excludedToolNames = this._excludedToolNames;
-        const isAllowedTool = (name) => (!allowedToolNames || allowedToolNames.has(name)) && !excludedToolNames?.has(name);
-        const registeredTools = this._extensionRunner.getAllRegisteredTools();
-        const allCustomTools = [
-            ...registeredTools,
-            ...this._customTools.map((definition) => ({
-                definition,
-                sourceInfo: createSyntheticSourceInfo(`<sdk:${definition.name}>`, { source: "sdk" }),
-            })),
-        ].filter((tool) => isAllowedTool(tool.definition.name));
-        const definitionRegistry = new Map(Array.from(this._baseToolDefinitions.entries())
-            .filter(([name]) => isAllowedTool(name))
-            .map(([name, definition]) => [
-            name,
-            {
-                definition,
-                sourceInfo: createSyntheticSourceInfo(`<builtin:${name}>`, { source: "builtin" }),
-            },
-        ]));
-        for (const tool of allCustomTools) {
-            definitionRegistry.set(tool.definition.name, {
-                definition: tool.definition,
-                sourceInfo: tool.sourceInfo,
-            });
-        }
-        this._toolDefinitions = definitionRegistry;
-        this._toolPromptSnippets = new Map(Array.from(definitionRegistry.values())
-            .map(({ definition }) => {
-            const snippet = this._normalizePromptSnippet(definition.promptSnippet);
-            return snippet ? [definition.name, snippet] : undefined;
-        })
-            .filter((entry) => entry !== undefined));
-        this._toolPromptGuidelines = new Map(Array.from(definitionRegistry.values())
-            .map(({ definition }) => {
-            const guidelines = this._normalizePromptGuidelines(definition.promptGuidelines);
-            return guidelines.length > 0 ? [definition.name, guidelines] : undefined;
-        })
-            .filter((entry) => entry !== undefined));
-        const runner = this._extensionRunner;
-        const wrappedExtensionTools = wrapRegisteredTools(allCustomTools, runner);
-        const wrappedBuiltInTools = wrapRegisteredTools(Array.from(this._baseToolDefinitions.values())
-            .filter((definition) => isAllowedTool(definition.name))
-            .map((definition) => ({
-            definition,
-            sourceInfo: createSyntheticSourceInfo(`<builtin:${definition.name}>`, { source: "builtin" }),
-        })), runner);
-        const toolRegistry = new Map(wrappedBuiltInTools.map((tool) => [tool.name, tool]));
-        for (const tool of wrappedExtensionTools) {
-            toolRegistry.set(tool.name, tool);
-        }
-        this._toolRegistry = toolRegistry;
-        const nextActiveToolNames = (options?.activeToolNames ? [...options.activeToolNames] : [...previousActiveToolNames]).filter((name) => isAllowedTool(name));
-        if (allowedToolNames) {
-            for (const toolName of this._toolRegistry.keys()) {
-                if (allowedToolNames.has(toolName)) {
-                    nextActiveToolNames.push(toolName);
-                }
-            }
-        }
-        else if (options?.includeAllExtensionTools) {
-            for (const tool of wrappedExtensionTools) {
-                nextActiveToolNames.push(tool.name);
-            }
-        }
-        else if (!options?.activeToolNames) {
-            for (const toolName of this._toolRegistry.keys()) {
-                if (!previousRegistryNames.has(toolName)) {
-                    nextActiveToolNames.push(toolName);
-                }
-            }
-        }
-        this.setActiveToolsByName([...new Set(nextActiveToolNames)]);
-    }
-    _buildRuntime(options) {
-        const autoResizeImages = this.settingsManager.getImageAutoResize();
-        const shellCommandPrefix = this.settingsManager.getShellCommandPrefix();
-        const shellPath = this.settingsManager.getShellPath();
-        const baseToolDefinitions = this._baseToolsOverride
-            ? Object.fromEntries(Object.entries(this._baseToolsOverride).map(([name, tool]) => [
-                name,
-                createToolDefinitionFromAgentTool(tool),
-            ]))
-            : createAllToolDefinitions(this._cwd, {
-                read: { autoResizeImages },
-                bash: { commandPrefix: shellCommandPrefix, shellPath },
-            });
-        this._baseToolDefinitions = new Map(Object.entries(baseToolDefinitions).map(([name, tool]) => [name, tool]));
-        const extensionsResult = this._resourceLoader.getExtensions();
-        if (options.flagValues) {
-            for (const [name, value] of options.flagValues) {
-                extensionsResult.runtime.flagValues.set(name, value);
-            }
-        }
-        this._extensionRunner = new ExtensionRunner(extensionsResult.extensions, extensionsResult.runtime, this._cwd, this.sessionManager, this._modelRegistry);
-        if (this._extensionRunnerRef) {
-            this._extensionRunnerRef.current = this._extensionRunner;
-        }
-        this._bindExtensionCore(this._extensionRunner);
-        this._applyExtensionBindings(this._extensionRunner);
-        const defaultActiveToolNames = this._baseToolsOverride
-            ? Object.keys(this._baseToolsOverride)
-            : ["read", "bash", "edit", "write"];
-        const baseActiveToolNames = options.activeToolNames ?? defaultActiveToolNames;
-        this._refreshToolRegistry({
-            activeToolNames: baseActiveToolNames,
-            includeAllExtensionTools: options.includeAllExtensionTools,
-        });
-    }
-    async reload() {
-        const previousFlagValues = this._extensionRunner.getFlagValues();
-        await emitSessionShutdownEvent(this._extensionRunner, { type: "session_shutdown", reason: "reload" });
-        await this.settingsManager.reload();
-        resetApiProviders();
-        await this._resourceLoader.reload();
-        this._buildRuntime({
-            activeToolNames: this.getActiveToolNames(),
-            flagValues: previousFlagValues,
-            includeAllExtensionTools: true,
-        });
-        const hasBindings = this._extensionUIContext ||
-            this._extensionCommandContextActions ||
-            this._extensionShutdownHandler ||
-            this._extensionErrorListener;
-        if (hasBindings) {
-            await this._extensionRunner.emit({ type: "session_start", reason: "reload" });
-            await this.extendResourcesFromExtensions("reload");
-        }
-    }
-    // =========================================================================
-    // Auto-Retry
-    // =========================================================================
-    _isNonRetryableProviderLimitError(errorMessage) {
-        return /GoUsageLimitError|FreeUsageLimitError|Monthly usage limit reached|available balance|insufficient_quota|out of budget|quota exceeded|billing/i.test(errorMessage);
-    }
-    /**
-     * Check if an error is retryable (overloaded, rate limit, server errors).
-     * Context overflow errors are NOT retryable (handled by compaction instead).
-     */
-    _isRetryableError(message) {
-        if (message.stopReason !== "error" || !message.errorMessage)
-            return false;
-        // Context overflow is handled by compaction, not retry
-        const contextWindow = this.model?.contextWindow ?? 0;
-        if (isContextOverflow(message, contextWindow))
-            return false;
-        const err = message.errorMessage;
-        if (this._isNonRetryableProviderLimitError(err))
-            return false;
-        // Match: overloaded_error, provider returned error, rate limit, 429, 500, 502, 503, 504, service unavailable, network/connection errors (including connection lost), WebSocket transport closes/errors, fetch failed, premature stream endings, HTTP/2 closed before response, terminated, retry delay exceeded
-        return /overloaded|provider.?returned.?error|rate.?limit|too many requests|429|500|502|503|504|service.?unavailable|server.?error|internal.?error|network.?error|connection.?error|connection.?refused|connection.?lost|websocket.?closed|websocket.?error|other side closed|fetch failed|upstream.?connect|reset before headers|socket hang up|ended without|stream ended before message_stop|http2 request did not get a response|timed? out|timeout|terminated|retry delay/i.test(err);
-    }
-    /**
-     * Prepare a retryable error for continuation with exponential backoff.
-     * @returns true if the caller should continue the agent, false otherwise
-     */
-    async _prepareRetry(message) {
-        const settings = this.settingsManager.getRetrySettings();
-        if (!settings.enabled) {
-            return false;
-        }
-        this._retryAttempt++;
-        if (this._retryAttempt > settings.maxRetries) {
-            // Preserve the completed attempt count so post-run handling can emit the final failure.
-            this._retryAttempt--;
-            return false;
-        }
-        const delayMs = settings.baseDelayMs * 2 ** (this._retryAttempt - 1);
-        this._emit({
-            type: "auto_retry_start",
-            attempt: this._retryAttempt,
-            maxAttempts: settings.maxRetries,
-            delayMs,
-            errorMessage: message.errorMessage || "Unknown error",
-        });
-        // Remove error message from agent state (keep in session for history)
-        const messages = this.agent.state.messages;
-        if (messages.length > 0 && messages[messages.length - 1].role === "assistant") {
-            this.agent.state.messages = messages.slice(0, -1);
-        }
-        // Wait with exponential backoff (abortable)
-        this._retryAbortController = new AbortController();
-        try {
-            await sleep(delayMs, this._retryAbortController.signal);
-        }
-        catch {
-            // Aborted during sleep - emit end event so UI can clean up
-            const attempt = this._retryAttempt;
-            this._retryAttempt = 0;
-            this._emit({
-                type: "auto_retry_end",
-                success: false,
-                attempt,
-                finalError: "Retry cancelled",
-            });
-            return false;
-        }
-        finally {
-            this._retryAbortController = undefined;
-        }
-        return true;
-    }
-    /**
-     * Cancel in-progress retry.
-     */
-    abortRetry() {
-        this._retryAbortController?.abort();
-    }
-    /** Whether auto-retry is currently in progress */
-    get isRetrying() {
-        return this._retryAbortController !== undefined;
-    }
-    /** Whether auto-retry is enabled */
-    get autoRetryEnabled() {
-        return this.settingsManager.getRetryEnabled();
-    }
-    /**
-     * Toggle auto-retry setting.
-     */
-    setAutoRetryEnabled(enabled) {
-        this.settingsManager.setRetryEnabled(enabled);
-    }
-    // =========================================================================
-    // Bash Execution
-    // =========================================================================
-    /**
-     * Execute a bash command.
-     * Adds result to agent context and session.
-     * @param command The bash command to execute
-     * @param onChunk Optional streaming callback for output
-     * @param options.excludeFromContext If true, command output won't be sent to LLM (!! prefix)
-     * @param options.operations Custom BashOperations for remote execution
-     */
-    async executeBash(command, onChunk, options) {
-        this._bashAbortController = new 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();
-        const resolvedCommand = prefix ? `${prefix}\n${command}` : command;
-        try {
-            const result = await executeBashWithOperations(resolvedCommand, this.sessionManager.getCwd(), options?.operations ?? createLocalBashOperations({ shellPath }), {
-                onChunk,
-                signal: this._bashAbortController.signal,
-            });
-            this.recordBashResult(command, result, options);
-            return result;
-        }
-        finally {
-            this._bashAbortController = undefined;
-        }
-    }
-    /**
-     * Record a bash execution result in session history.
-     * Used by executeBash and by extensions that handle bash execution themselves.
-     */
-    recordBashResult(command, result, options) {
-        const bashMessage = {
-            role: "bashExecution",
-            command,
-            output: result.output,
-            exitCode: result.exitCode,
-            cancelled: result.cancelled,
-            truncated: result.truncated,
-            fullOutputPath: result.fullOutputPath,
-            timestamp: Date.now(),
-            excludeFromContext: options?.excludeFromContext,
-        };
-        // If agent is streaming, defer adding to avoid breaking tool_use/tool_result ordering
-        if (this.isStreaming) {
-            // Queue for later - will be flushed on agent_end
-            this._pendingBashMessages.push(bashMessage);
-        }
-        else {
-            // Add to agent state immediately
-            this.agent.state.messages.push(bashMessage);
-            // Save to session
-            this.sessionManager.appendMessage(bashMessage);
-        }
-    }
-    /**
-     * Cancel running bash command.
-     */
-    abortBash() {
-        this._bashAbortController?.abort();
-    }
-    /** Whether a bash command is currently running */
-    get isBashRunning() {
-        return this._bashAbortController !== undefined;
-    }
-    /** Whether there are pending bash messages waiting to be flushed */
-    get hasPendingBashMessages() {
-        return this._pendingBashMessages.length > 0;
-    }
-    /**
-     * Flush pending bash messages to agent state and session.
-     * Called after agent turn completes to maintain proper message ordering.
-     */
-    _flushPendingBashMessages() {
-        if (this._pendingBashMessages.length === 0)
-            return;
-        for (const bashMessage of this._pendingBashMessages) {
-            // Add to agent state
-            this.agent.state.messages.push(bashMessage);
-            // Save to session
-            this.sessionManager.appendMessage(bashMessage);
-        }
-        this._pendingBashMessages = [];
-    }
-    // =========================================================================
-    // Session Management
-    // =========================================================================
-    /**
-     * Set a display name for the current session.
-     */
-    setSessionName(name) {
-        this.sessionManager.appendSessionInfo(name);
-        this._emit({ type: "session_info_changed", name: this.sessionManager.getSessionName() });
-    }
-    // =========================================================================
-    // Tree Navigation
-    // =========================================================================
-    /**
-     * Navigate to a different node in the session tree.
-     * Unlike fork() which creates a new session file, this stays in the same file.
-     *
-     * @param targetId The entry ID to navigate to
-     * @param options.summarize Whether user wants to summarize abandoned branch
-     * @param options.customInstructions Custom instructions for summarizer
-     * @param options.replaceInstructions If true, customInstructions replaces the default prompt
-     * @param options.label Label to attach to the branch summary entry
-     * @returns Result with editorText (if user message) and cancelled status
-     */
-    async navigateTree(targetId, options = {}) {
-        const oldLeafId = this.sessionManager.getLeafId();
-        // No-op if already at target
-        if (targetId === oldLeafId) {
-            return { cancelled: false };
-        }
-        // Model required for summarization
-        if (options.summarize && !this.model) {
-            throw new Error("No model available for summarization");
-        }
-        const targetEntry = this.sessionManager.getEntry(targetId);
-        if (!targetEntry) {
-            throw new Error(`Entry ${targetId} not found`);
-        }
-        // Collect entries to summarize (from old leaf to common ancestor)
-        const { entries: entriesToSummarize, commonAncestorId } = collectEntriesForBranchSummary(this.sessionManager, oldLeafId, targetId);
-        // Prepare event data - mutable so extensions can override
-        let customInstructions = options.customInstructions;
-        let replaceInstructions = options.replaceInstructions;
-        let label = options.label;
-        const preparation = {
-            targetId,
-            oldLeafId,
-            commonAncestorId,
-            entriesToSummarize,
-            userWantsSummary: options.summarize ?? false,
-            customInstructions,
-            replaceInstructions,
-            label,
-        };
-        // Set up abort controller for summarization
-        this._branchSummaryAbortController = new AbortController();
-        try {
-            let extensionSummary;
-            let fromExtension = false;
-            // Emit session_before_tree event
-            if (this._extensionRunner.hasHandlers("session_before_tree")) {
-                const result = (await this._extensionRunner.emit({
-                    type: "session_before_tree",
-                    preparation,
-                    signal: this._branchSummaryAbortController.signal,
-                }));
-                if (result?.cancel) {
-                    return { cancelled: true };
-                }
-                if (result?.summary && options.summarize) {
-                    extensionSummary = result.summary;
-                    fromExtension = true;
-                }
-                // Allow extensions to override instructions and label
-                if (result?.customInstructions !== undefined) {
-                    customInstructions = result.customInstructions;
-                }
-                if (result?.replaceInstructions !== undefined) {
-                    replaceInstructions = result.replaceInstructions;
-                }
-                if (result?.label !== undefined) {
-                    label = result.label;
-                }
-            }
-            // Run default summarizer if needed
-            let summaryText;
-            let summaryDetails;
-            if (options.summarize && entriesToSummarize.length > 0 && !extensionSummary) {
-                const model = this.model;
-                const { apiKey, headers } = await this._getRequiredRequestAuth(model);
-                const branchSummarySettings = this.settingsManager.getBranchSummarySettings();
-                const result = await generateBranchSummary(entriesToSummarize, {
-                    model,
-                    apiKey,
-                    headers,
-                    signal: this._branchSummaryAbortController.signal,
-                    customInstructions,
-                    replaceInstructions,
-                    reserveTokens: branchSummarySettings.reserveTokens,
-                    streamFn: this.agent.streamFn,
-                });
-                if (result.aborted) {
-                    return { cancelled: true, aborted: true };
-                }
-                if (result.error) {
-                    throw new Error(result.error);
-                }
-                summaryText = result.summary;
-                summaryDetails = {
-                    readFiles: result.readFiles || [],
-                    modifiedFiles: result.modifiedFiles || [],
-                };
-            }
-            else if (extensionSummary) {
-                summaryText = extensionSummary.summary;
-                summaryDetails = extensionSummary.details;
-            }
-            // Determine the new leaf position based on target type
-            let newLeafId;
-            let editorText;
-            if (targetEntry.type === "message" && targetEntry.message.role === "user") {
-                // User message: leaf = parent (null if root), text goes to editor
-                newLeafId = targetEntry.parentId;
-                editorText = this._extractUserMessageText(targetEntry.message.content);
-            }
-            else if (targetEntry.type === "custom_message") {
-                // Custom message: leaf = parent (null if root), text goes to editor
-                newLeafId = targetEntry.parentId;
-                editorText =
-                    typeof targetEntry.content === "string"
-                        ? targetEntry.content
-                        : targetEntry.content
-                            .filter((c) => c.type === "text")
-                            .map((c) => c.text)
-                            .join("");
-            }
-            else {
-                // Non-user message: leaf = selected node
-                newLeafId = targetId;
-            }
-            // Switch leaf (with or without summary)
-            // Summary is attached at the navigation target position (newLeafId), not the old branch
-            let summaryEntry;
-            if (summaryText) {
-                // Create summary at target position (can be null for root)
-                const summaryId = this.sessionManager.branchWithSummary(newLeafId, summaryText, summaryDetails, fromExtension);
-                summaryEntry = this.sessionManager.getEntry(summaryId);
-                // Attach label to the summary entry
-                if (label) {
-                    this.sessionManager.appendLabelChange(summaryId, label);
-                }
-            }
-            else if (newLeafId === null) {
-                // No summary, navigating to root - reset leaf
-                this.sessionManager.resetLeaf();
-            }
-            else {
-                // No summary, navigating to non-root
-                this.sessionManager.branch(newLeafId);
-            }
-            // Attach label to target entry when not summarizing (no summary entry to label)
-            if (label && !summaryText) {
-                this.sessionManager.appendLabelChange(targetId, label);
-            }
-            // Update agent state
-            const sessionContext = this.sessionManager.buildSessionContext();
-            this.agent.state.messages = sessionContext.messages;
-            // Emit session_tree event
-            await this._extensionRunner.emit({
-                type: "session_tree",
-                newLeafId: this.sessionManager.getLeafId(),
-                oldLeafId,
-                summaryEntry,
-                fromExtension: summaryText ? fromExtension : undefined,
-            });
-            // Emit to custom tools
-            return { editorText, cancelled: false, summaryEntry };
-        }
-        finally {
-            this._branchSummaryAbortController = undefined;
-        }
-    }
-    /**
-     * Get all user messages from session for fork selector.
-     */
-    getUserMessagesForForking() {
-        const entries = this.sessionManager.getEntries();
-        const result = [];
-        for (const entry of entries) {
-            if (entry.type !== "message")
-                continue;
-            if (entry.message.role !== "user")
-                continue;
-            const text = this._extractUserMessageText(entry.message.content);
-            if (text) {
-                result.push({ entryId: entry.id, text });
-            }
-        }
-        return result;
-    }
-    _extractUserMessageText(content) {
-        if (typeof content === "string")
-            return content;
-        if (Array.isArray(content)) {
-            return content
-                .filter((c) => c.type === "text")
-                .map((c) => c.text)
-                .join("");
-        }
-        return "";
-    }
-    /**
-     * Get session statistics.
-     */
-    getSessionStats() {
-        const state = this.state;
-        const userMessages = state.messages.filter((m) => m.role === "user").length;
-        const assistantMessages = state.messages.filter((m) => m.role === "assistant").length;
-        const toolResults = state.messages.filter((m) => m.role === "toolResult").length;
-        let toolCalls = 0;
-        let totalInput = 0;
-        let totalOutput = 0;
-        let totalCacheRead = 0;
-        let totalCacheWrite = 0;
-        let totalCost = 0;
-        for (const message of state.messages) {
-            if (message.role === "assistant") {
-                const assistantMsg = message;
-                toolCalls += assistantMsg.content.filter((c) => c.type === "toolCall").length;
-                totalInput += assistantMsg.usage.input;
-                totalOutput += assistantMsg.usage.output;
-                totalCacheRead += assistantMsg.usage.cacheRead;
-                totalCacheWrite += assistantMsg.usage.cacheWrite;
-                totalCost += assistantMsg.usage.cost.total;
-            }
-        }
-        return {
-            sessionFile: this.sessionFile,
-            sessionId: this.sessionId,
-            userMessages,
-            assistantMessages,
-            toolCalls,
-            toolResults,
-            totalMessages: state.messages.length,
-            tokens: {
-                input: totalInput,
-                output: totalOutput,
-                cacheRead: totalCacheRead,
-                cacheWrite: totalCacheWrite,
-                total: totalInput + totalOutput + totalCacheRead + totalCacheWrite,
-            },
-            cost: totalCost,
-            contextUsage: this.getContextUsage(),
-        };
-    }
-    getContextUsage() {
-        const model = this.model;
-        if (!model)
-            return undefined;
-        const contextWindow = model.contextWindow ?? 0;
-        if (contextWindow <= 0)
-            return undefined;
-        // After compaction, the last assistant usage reflects pre-compaction context size.
-        // We can only trust usage from an assistant that responded after the latest compaction.
-        // If no such assistant exists, context token count is unknown until the next LLM response.
-        const branchEntries = this.sessionManager.getBranch();
-        const latestCompaction = getLatestCompactionEntry(branchEntries);
-        if (latestCompaction) {
-            // Check if there's a valid assistant usage after the compaction boundary
-            const compactionIndex = branchEntries.lastIndexOf(latestCompaction);
-            let hasPostCompactionUsage = false;
-            for (let i = branchEntries.length - 1; i > compactionIndex; i--) {
-                const entry = branchEntries[i];
-                if (entry.type === "message" && entry.message.role === "assistant") {
-                    const assistant = entry.message;
-                    if (assistant.stopReason !== "aborted" && assistant.stopReason !== "error") {
-                        const contextTokens = calculateContextTokens(assistant.usage);
-                        if (contextTokens > 0) {
-                            hasPostCompactionUsage = true;
-                        }
-                        break;
-                    }
-                }
-            }
-            if (!hasPostCompactionUsage) {
-                return { tokens: null, contextWindow, percent: null };
-            }
-        }
-        const estimate = estimateContextTokens(this.messages);
-        const percent = (estimate.tokens / contextWindow) * 100;
-        return {
-            tokens: estimate.tokens,
-            contextWindow,
-            percent,
-        };
-    }
-    /**
-     * Export session to HTML.
-     * @param outputPath Optional output path (defaults to session directory)
-     * @returns Path to exported file
-     */
-    async exportToHtml(outputPath) {
-        const themeName = this.settingsManager.getTheme();
-        // Create tool renderer if we have an extension runner (for custom tool HTML rendering)
-        const toolRenderer = createToolHtmlRenderer({
-            getToolDefinition: (name) => this.getToolDefinition(name),
-            theme,
-            cwd: this.sessionManager.getCwd(),
-        });
-        return await exportSessionToHtml(this.sessionManager, this.state, {
-            outputPath,
-            themeName,
-            toolRenderer,
-        });
-    }
-    /**
-     * Export the current session branch to a JSONL file.
-     * Writes the session header followed by all entries on the current branch path.
-     * @param outputPath Target file path. If omitted, generates a timestamped file in cwd.
-     * @returns The resolved output file path.
-     */
-    exportToJsonl(outputPath) {
-        const filePath = resolvePath(outputPath ?? `session-${new Date().toISOString().replace(/[:.]/g, "-")}.jsonl`, process.cwd());
-        const dir = dirname(filePath);
-        if (!existsSync(dir)) {
-            mkdirSync(dir, { recursive: true });
-        }
-        const header = {
-            type: "session",
-            version: CURRENT_SESSION_VERSION,
-            id: this.sessionManager.getSessionId(),
-            timestamp: new Date().toISOString(),
-            cwd: this.sessionManager.getCwd(),
-        };
-        const branchEntries = this.sessionManager.getBranch();
-        const lines = [JSON.stringify(header)];
-        // Re-chain parentIds to form a linear sequence
-        let prevId = null;
-        for (const entry of branchEntries) {
-            const linear = { ...entry, parentId: prevId };
-            lines.push(JSON.stringify(linear));
-            prevId = entry.id;
-        }
-        writeFileSync(filePath, `${lines.join("\n")}\n`);
-        return filePath;
-    }
-    // =========================================================================
-    // Utilities
-    // =========================================================================
-    /**
-     * Get text content of last assistant message.
-     * Useful for /copy command.
-     * @returns Text content, or undefined if no assistant message exists
-     */
-    getLastAssistantText() {
-        const lastAssistant = this.messages
-            .slice()
-            .reverse()
-            .find((m) => {
-            if (m.role !== "assistant")
-                return false;
-            const msg = m;
-            // Skip aborted messages with no content
-            if (msg.stopReason === "aborted" && msg.content.length === 0)
-                return false;
-            return true;
-        });
-        if (!lastAssistant)
-            return undefined;
-        let text = "";
-        for (const content of lastAssistant.content) {
-            if (content.type === "text") {
-                text += content.text;
-            }
-        }
-        return text.trim() || undefined;
-    }
-    // =========================================================================
-    // Extension System
-    // =========================================================================
-    createReplacedSessionContext() {
-        const context = Object.defineProperties({}, Object.getOwnPropertyDescriptors(this._extensionRunner.createCommandContext()));
-        context.sendMessage = (message, options) => this.sendCustomMessage(message, options);
-        context.sendUserMessage = (content, options) => this.sendUserMessage(content, options);
-        return context;
-    }
-    /**
-     * Check if extensions have handlers for a specific event type.
-     */
-    hasExtensionHandlers(eventType) {
-        return this._extensionRunner.hasHandlers(eventType);
-    }
-    /**
-     * Get the extension runner (for setting UI context and error handlers).
-     */
-    get extensionRunner() {
-        return this._extensionRunner;
-    }
-}
-//# sourceMappingURL=agent-session.js.map