@mariozechner/pi-coding-agent 0.16.0 → 0.18.0

package/dist/core/agent-session.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"agent-session.d.ts","sourceRoot":"","sources":["../../src/core/agent-session.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;GAaG;AAEH,OAAO,KAAK,EAAE,KAAK,EAAE,UAAU,EAAE,UAAU,EAAE,UAAU,EAAE,UAAU,EAAE,aAAa,EAAE,MAAM,6BAA6B,CAAC;AACxH,OAAO,KAAK,EAAoB,KAAK,EAAE,MAAM,qBAAqB,CAAC;AAEnE,OAAO,EAAE,KAAK,UAAU,EAAqC,MAAM,oBAAoB,CAAC;AAKxF,OAAO,EAA0B,KAAK,cAAc,EAAE,MAAM,sBAAsB,CAAC;AACnF,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,uBAAuB,CAAC;AAC7D,OAAO,EAAsB,KAAK,gBAAgB,EAAE,MAAM,qBAAqB,CAAC;AAEhF,8DAA8D;AAC9D,MAAM,MAAM,iBAAiB,GAC1B,UAAU,GACV;IAAE,IAAI,EAAE,uBAAuB,CAAA;CAAE,GACjC;IAAE,IAAI,EAAE,qBAAqB,CAAC;IAAC,MAAM,EAAE,gBAAgB,GAAG,IAAI,CAAC;IAAC,OAAO,EAAE,OAAO,CAAA;CAAE,CAAC;AAEtF,iDAAiD;AACjD,MAAM,MAAM,yBAAyB,GAAG,CAAC,KAAK,EAAE,iBAAiB,KAAK,IAAI,CAAC;AAM3E,MAAM,WAAW,kBAAkB;IAClC,KAAK,EAAE,KAAK,CAAC;IACb,cAAc,EAAE,cAAc,CAAC;IAC/B,eAAe,EAAE,eAAe,CAAC;IACjC,+DAA+D;IAC/D,YAAY,CAAC,EAAE,KAAK,CAAC;QAAE,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC;QAAC,aAAa,EAAE,aAAa,CAAA;KAAE,CAAC,CAAC;IAC1E,8CAA8C;IAC9C,YAAY,CAAC,EAAE,gBAAgB,EAAE,CAAC;CAClC;AAED,wCAAwC;AACxC,MAAM,WAAW,aAAa;IAC7B,kEAAkE;IAClE,mBAAmB,CAAC,EAAE,OAAO,CAAC;IAC9B,6BAA6B;IAC7B,WAAW,CAAC,EAAE,UAAU,EAAE,CAAC;CAC3B;AAED,+BAA+B;AAC/B,MAAM,WAAW,gBAAgB;IAChC,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC;IAClB,aAAa,EAAE,aAAa,CAAC;IAC7B,6EAA6E;IAC7E,QAAQ,EAAE,OAAO,CAAC;CAClB;AAED,qDAAqD;AACrD,MAAM,WAAW,gBAAgB;IAChC,YAAY,EAAE,MAAM,CAAC;IACrB,OAAO,EAAE,MAAM,CAAC;CAChB;AAED,8CAA8C;AAC9C,MAAM,WAAW,YAAY;IAC5B,WAAW,EAAE,MAAM,CAAC;IACpB,SAAS,EAAE,MAAM,CAAC;IAClB,YAAY,EAAE,MAAM,CAAC;IACrB,iBAAiB,EAAE,MAAM,CAAC;IAC1B,SAAS,EAAE,MAAM,CAAC;IAClB,WAAW,EAAE,MAAM,CAAC;IACpB,aAAa,EAAE,MAAM,CAAC;IACtB,MAAM,EAAE;QACP,KAAK,EAAE,MAAM,CAAC;QACd,MAAM,EAAE,MAAM,CAAC;QACf,SAAS,EAAE,MAAM,CAAC;QAClB,UAAU,EAAE,MAAM,CAAC;QACnB,KAAK,EAAE,MAAM,CAAC;KACd,CAAC;IACF,IAAI,EAAE,MAAM,CAAC;CACb;AAMD,qBAAa,YAAY;IACxB,QAAQ,CAAC,KAAK,EAAE,KAAK,CAAC;IACtB,QAAQ,CAAC,cAAc,EAAE,cAAc,CAAC;IACxC,QAAQ,CAAC,eAAe,EAAE,eAAe,CAAC;IAE1C,OAAO,CAAC,aAAa,CAA6D;IAClF,OAAO,CAAC,aAAa,CAAqB;IAG1C,OAAO,CAAC,iBAAiB,CAAC,CAAa;IACvC,OAAO,CAAC,eAAe,CAAmC;IAG1D,OAAO,CAAC,eAAe,CAAgB;IAGvC,OAAO,CAAC,0BAA0B,CAAgC;IAClE,OAAO,CAAC,8BAA8B,CAAgC;IAGtE,OAAO,CAAC,oBAAoB,CAAgC;IAC5D,OAAO,CAAC,oBAAoB,CAA8B;IAE1D,YAAY,MAAM,EAAE,kBAAkB,EAMrC;IAMD,qCAAqC;IACrC,OAAO,CAAC,KAAK;IAOb,OAAO,CAAC,qBAAqB,CAAiC;IAE9D,4EAA4E;IAC5E,OAAO,CAAC,iBAAiB,CAyBvB;IAEF;;;;OAIG;IACH,SAAS,CAAC,QAAQ,EAAE,yBAAyB,GAAG,MAAM,IAAI,CAezD;IAED;;;;OAIG;IACH,OAAO,CAAC,oBAAoB;IAO5B;;;OAGG;IACH,OAAO,CAAC,iBAAiB;IAKzB;;;OAGG;IACH,OAAO,IAAI,IAAI,CAGd;IAMD,uBAAuB;IACvB,IAAI,KAAK,IAAI,UAAU,CAEtB;IAED,sDAAsD;IACtD,IAAI,KAAK,IAAI,KAAK,CAAC,GAAG,CAAC,GAAG,IAAI,CAE7B;IAED,6BAA6B;IAC7B,IAAI,aAAa,IAAI,aAAa,CAEjC;IAED,sDAAsD;IACtD,IAAI,WAAW,IAAI,OAAO,CAEzB;IAED,oEAAoE;IACpE,IAAI,QAAQ,IAAI,UAAU,EAAE,CAE3B;IAED,yBAAyB;IACzB,IAAI,SAAS,IAAI,KAAK,GAAG,eAAe,CAEvC;IAED,gCAAgC;IAChC,IAAI,WAAW,IAAI,MAAM,CAExB;IAED,yBAAyB;IACzB,IAAI,SAAS,IAAI,MAAM,CAEtB;IAED,qDAAqD;IACrD,IAAI,YAAY,IAAI,aAAa,CAAC;QAAE,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC;QAAC,aAAa,EAAE,aAAa,CAAA;KAAE,CAAC,CAErF;IAED,gCAAgC;IAChC,IAAI,YAAY,IAAI,aAAa,CAAC,gBAAgB,CAAC,CAElD;IAMD;;;;;OAKG;IACG,MAAM,CAAC,IAAI,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,aAAa,GAAG,OAAO,CAAC,IAAI,CAAC,CA6BjE;IAED;;;OAGG;IACG,YAAY,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAO9C;IAED;;;OAGG;IACH,UAAU,IAAI,MAAM,EAAE,CAKrB;IAED,0CAA0C;IAC1C,IAAI,kBAAkB,IAAI,MAAM,CAE/B;IAED,sCAAsC;IACtC,iBAAiB,IAAI,SAAS,MAAM,EAAE,CAErC;IAED;;OAEG;IACG,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC,CAG3B;IAED;;;;OAIG;IACG,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC,CAO3B;IAMD;;;;OAIG;IACG,QAAQ,CAAC,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,GAAG,OAAO,CAAC,IAAI,CAAC,CAS/C;IAED;;;;OAIG;IACG,UAAU,IAAI,OAAO,CAAC,gBAAgB,GAAG,IAAI,CAAC,CAKnD;YAEa,iBAAiB;YAgCjB,oBAAoB;IA0BlC;;OAEG;IACG,kBAAkB,IAAI,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC,CAIhD;IAMD;;;;OAIG;IACH,gBAAgB,CAAC,KAAK,EAAE,aAAa,GAAG,IAAI,CAK3C;IAED;;;OAGG;IACH,kBAAkB,IAAI,aAAa,GAAG,IAAI,CAezC;IAED;;OAEG;IACH,gBAAgB,IAAI,OAAO,CAE1B;IAMD;;;OAGG;IACH,YAAY,CAAC,IAAI,EAAE,KAAK,GAAG,eAAe,GAAG,IAAI,CAGhD;IAMD;;;;OAIG;IACG,OAAO,CAAC,kBAAkB,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,gBAAgB,CAAC,CA8CpE;IAED;;OAEG;IACH,eAAe,IAAI,IAAI,CAGtB;YAMa,kBAAkB;IA4DhC;;OAEG;IACH,wBAAwB,CAAC,OAAO,EAAE,OAAO,GAAG,IAAI,CAE/C;IAED,yCAAyC;IACzC,IAAI,qBAAqB,IAAI,OAAO,CAEnC;IAMD;;;;;OAKG;IACG,WAAW,CAAC,OAAO,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,IAAI,GAAG,OAAO,CAAC,UAAU,CAAC,CA0CzF;IAED;;OAEG;IACH,SAAS,IAAI,IAAI,CAEhB;IAED,kDAAkD;IAClD,IAAI,aAAa,IAAI,OAAO,CAE3B;IAED,oEAAoE;IACpE,IAAI,sBAAsB,IAAI,OAAO,CAEpC;IAED;;;OAGG;IACH,OAAO,CAAC,yBAAyB;IAuBjC;;;;OAIG;IACG,aAAa,CAAC,WAAW,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CA6BtD;IAED;;;;OAIG;IACH,MAAM,CAAC,UAAU,EAAE,MAAM,GAAG,MAAM,CAmBjC;IAED;;OAEG;IACH,2BAA2B,IAAI,KAAK,CAAC;QAAE,UAAU,EAAE,MAAM,CAAC;QAAC,IAAI,EAAE,MAAM,CAAA;KAAE,CAAC,CAgBzE;IAED,OAAO,CAAC,uBAAuB;IAW/B;;OAEG;IACH,eAAe,IAAI,YAAY,CA0C9B;IAED;;;;OAIG;IACH,YAAY,CAAC,UAAU,CAAC,EAAE,MAAM,GAAG,MAAM,CAExC;IAMD;;;;OAIG;IACH,oBAAoB,IAAI,MAAM,GAAG,IAAI,CAgBpC;CACD","sourcesContent":["/**\n * AgentSession - Core abstraction for agent lifecycle and session management.\n *\n * This class is shared between all run modes (interactive, print, rpc).\n * It encapsulates:\n * - Agent state access\n * - Event subscription with automatic session persistence\n * - Model and thinking level management\n * - Compaction (manual and auto)\n * - Bash execution\n * - Session switching and branching\n *\n * Modes use this class and add their own I/O layer on top.\n */\n\nimport type { Agent, AgentEvent, AgentState, AppMessage, Attachment, ThinkingLevel } from \"@mariozechner/pi-agent-core\";\nimport type { AssistantMessage, Model } from \"@mariozechner/pi-ai\";\nimport { getModelsPath } from \"../config.js\";\nimport { type BashResult, executeBash as executeBashCommand } from \"./bash-executor.js\";\nimport { calculateContextTokens, compact, shouldCompact } from \"./compaction.js\";\nimport { exportSessionToHtml } from \"./export-html.js\";\nimport type { BashExecutionMessage } from \"./messages.js\";\nimport { getApiKeyForModel, getAvailableModels } from \"./model-config.js\";\nimport { loadSessionFromEntries, type SessionManager } from \"./session-manager.js\";\nimport type { SettingsManager } from \"./settings-manager.js\";\nimport { expandSlashCommand, type FileSlashCommand } from \"./slash-commands.js\";\n\n/** Session-specific events that extend the core AgentEvent */\nexport type AgentSessionEvent =\n\t| AgentEvent\n\t| { type: \"auto_compaction_start\" }\n\t| { type: \"auto_compaction_end\"; result: CompactionResult | null; aborted: boolean };\n\n/** Listener function for agent session events */\nexport type AgentSessionEventListener = (event: AgentSessionEvent) => void;\n\n// ============================================================================\n// Types\n// ============================================================================\n\nexport interface AgentSessionConfig {\n\tagent: Agent;\n\tsessionManager: SessionManager;\n\tsettingsManager: SettingsManager;\n\t/** Models to cycle through with Ctrl+P (from --models flag) */\n\tscopedModels?: Array<{ model: Model<any>; thinkingLevel: ThinkingLevel }>;\n\t/** File-based slash commands for expansion */\n\tfileCommands?: FileSlashCommand[];\n}\n\n/** Options for AgentSession.prompt() */\nexport interface PromptOptions {\n\t/** Whether to expand file-based slash commands (default: true) */\n\texpandSlashCommands?: boolean;\n\t/** Image/file attachments */\n\tattachments?: Attachment[];\n}\n\n/** Result from cycleModel() */\nexport interface ModelCycleResult {\n\tmodel: Model<any>;\n\tthinkingLevel: ThinkingLevel;\n\t/** Whether cycling through scoped models (--models flag) or all available */\n\tisScoped: boolean;\n}\n\n/** Result from compact() or checkAutoCompaction() */\nexport interface CompactionResult {\n\ttokensBefore: number;\n\tsummary: string;\n}\n\n/** Session statistics for /session command */\nexport interface SessionStats {\n\tsessionFile: string;\n\tsessionId: string;\n\tuserMessages: number;\n\tassistantMessages: number;\n\ttoolCalls: number;\n\ttoolResults: number;\n\ttotalMessages: number;\n\ttokens: {\n\t\tinput: number;\n\t\toutput: number;\n\t\tcacheRead: number;\n\t\tcacheWrite: number;\n\t\ttotal: number;\n\t};\n\tcost: number;\n}\n\n// ============================================================================\n// AgentSession Class\n// ============================================================================\n\nexport class AgentSession {\n\treadonly agent: Agent;\n\treadonly sessionManager: SessionManager;\n\treadonly settingsManager: SettingsManager;\n\n\tprivate _scopedModels: Array<{ model: Model<any>; thinkingLevel: ThinkingLevel }>;\n\tprivate _fileCommands: FileSlashCommand[];\n\n\t// Event subscription state\n\tprivate _unsubscribeAgent?: () => void;\n\tprivate _eventListeners: AgentSessionEventListener[] = [];\n\n\t// Message queue state\n\tprivate _queuedMessages: string[] = [];\n\n\t// Compaction state\n\tprivate _compactionAbortController: AbortController | null = null;\n\tprivate _autoCompactionAbortController: AbortController | null = null;\n\n\t// Bash execution state\n\tprivate _bashAbortController: AbortController | null = null;\n\tprivate _pendingBashMessages: BashExecutionMessage[] = [];\n\n\tconstructor(config: AgentSessionConfig) {\n\t\tthis.agent = config.agent;\n\t\tthis.sessionManager = config.sessionManager;\n\t\tthis.settingsManager = config.settingsManager;\n\t\tthis._scopedModels = config.scopedModels ?? [];\n\t\tthis._fileCommands = config.fileCommands ?? [];\n\t}\n\n\t// =========================================================================\n\t// Event Subscription\n\t// =========================================================================\n\n\t/** Emit an event to all listeners */\n\tprivate _emit(event: AgentSessionEvent): void {\n\t\tfor (const l of this._eventListeners) {\n\t\t\tl(event);\n\t\t}\n\t}\n\n\t// Track last assistant message for auto-compaction check\n\tprivate _lastAssistantMessage: AssistantMessage | null = null;\n\n\t/** Internal handler for agent events - shared by subscribe and reconnect */\n\tprivate _handleAgentEvent = async (event: AgentEvent): Promise<void> => {\n\t\t// Notify all listeners\n\t\tthis._emit(event);\n\n\t\t// Handle session persistence\n\t\tif (event.type === \"message_end\") {\n\t\t\tthis.sessionManager.saveMessage(event.message);\n\n\t\t\t// Initialize session after first user+assistant exchange\n\t\t\tif (this.sessionManager.shouldInitializeSession(this.agent.state.messages)) {\n\t\t\t\tthis.sessionManager.startSession(this.agent.state);\n\t\t\t}\n\n\t\t\t// Track assistant message for auto-compaction (checked on agent_end)\n\t\t\tif (event.message.role === \"assistant\") {\n\t\t\t\tthis._lastAssistantMessage = event.message as AssistantMessage;\n\t\t\t}\n\t\t}\n\n\t\t// Check auto-compaction after agent completes (after agent_end clears UI)\n\t\tif (event.type === \"agent_end\" && this._lastAssistantMessage) {\n\t\t\tconst msg = this._lastAssistantMessage;\n\t\t\tthis._lastAssistantMessage = null;\n\t\t\tthis._runAutoCompaction(msg).catch(() => {});\n\t\t}\n\t};\n\n\t/**\n\t * Subscribe to agent events.\n\t * Session persistence is handled internally (saves messages on message_end).\n\t * Multiple listeners can be added. Returns unsubscribe function for this listener.\n\t */\n\tsubscribe(listener: AgentSessionEventListener): () => void {\n\t\tthis._eventListeners.push(listener);\n\n\t\t// Set up agent subscription if not already done\n\t\tif (!this._unsubscribeAgent) {\n\t\t\tthis._unsubscribeAgent = this.agent.subscribe(this._handleAgentEvent);\n\t\t}\n\n\t\t// Return unsubscribe function for this specific listener\n\t\treturn () => {\n\t\t\tconst index = this._eventListeners.indexOf(listener);\n\t\t\tif (index !== -1) {\n\t\t\t\tthis._eventListeners.splice(index, 1);\n\t\t\t}\n\t\t};\n\t}\n\n\t/**\n\t * Temporarily disconnect from agent events.\n\t * User listeners are preserved and will receive events again after resubscribe().\n\t * Used internally during operations that need to pause event processing.\n\t */\n\tprivate _disconnectFromAgent(): void {\n\t\tif (this._unsubscribeAgent) {\n\t\t\tthis._unsubscribeAgent();\n\t\t\tthis._unsubscribeAgent = undefined;\n\t\t}\n\t}\n\n\t/**\n\t * Reconnect to agent events after _disconnectFromAgent().\n\t * Preserves all existing listeners.\n\t */\n\tprivate _reconnectToAgent(): void {\n\t\tif (this._unsubscribeAgent) return; // Already connected\n\t\tthis._unsubscribeAgent = this.agent.subscribe(this._handleAgentEvent);\n\t}\n\n\t/**\n\t * Remove all listeners and disconnect from agent.\n\t * Call this when completely done with the session.\n\t */\n\tdispose(): void {\n\t\tthis._disconnectFromAgent();\n\t\tthis._eventListeners = [];\n\t}\n\n\t// =========================================================================\n\t// Read-only State Access\n\t// =========================================================================\n\n\t/** Full agent state */\n\tget state(): AgentState {\n\t\treturn this.agent.state;\n\t}\n\n\t/** Current model (may be null if not yet selected) */\n\tget model(): Model<any> | null {\n\t\treturn this.agent.state.model;\n\t}\n\n\t/** Current thinking level */\n\tget thinkingLevel(): ThinkingLevel {\n\t\treturn this.agent.state.thinkingLevel;\n\t}\n\n\t/** Whether agent is currently streaming a response */\n\tget isStreaming(): boolean {\n\t\treturn this.agent.state.isStreaming;\n\t}\n\n\t/** All messages including custom types like BashExecutionMessage */\n\tget messages(): AppMessage[] {\n\t\treturn this.agent.state.messages;\n\t}\n\n\t/** Current queue mode */\n\tget queueMode(): \"all\" | \"one-at-a-time\" {\n\t\treturn this.agent.getQueueMode();\n\t}\n\n\t/** Current session file path */\n\tget sessionFile(): string {\n\t\treturn this.sessionManager.getSessionFile();\n\t}\n\n\t/** Current session ID */\n\tget sessionId(): string {\n\t\treturn this.sessionManager.getSessionId();\n\t}\n\n\t/** Scoped models for cycling (from --models flag) */\n\tget scopedModels(): ReadonlyArray<{ model: Model<any>; thinkingLevel: ThinkingLevel }> {\n\t\treturn this._scopedModels;\n\t}\n\n\t/** File-based slash commands */\n\tget fileCommands(): ReadonlyArray<FileSlashCommand> {\n\t\treturn this._fileCommands;\n\t}\n\n\t// =========================================================================\n\t// Prompting\n\t// =========================================================================\n\n\t/**\n\t * Send a prompt to the agent.\n\t * - Validates model and API key before sending\n\t * - Expands file-based slash commands by default\n\t * @throws Error if no model selected or no API key available\n\t */\n\tasync prompt(text: string, options?: PromptOptions): Promise<void> {\n\t\t// Flush any pending bash messages before the new prompt\n\t\tthis._flushPendingBashMessages();\n\n\t\tconst expandCommands = options?.expandSlashCommands ?? true;\n\n\t\t// Validate model\n\t\tif (!this.model) {\n\t\t\tthrow new Error(\n\t\t\t\t\"No model selected.\\n\\n\" +\n\t\t\t\t\t\"Set an API key (ANTHROPIC_API_KEY, OPENAI_API_KEY, etc.)\\n\" +\n\t\t\t\t\t`or create ${getModelsPath()}\\n\\n` +\n\t\t\t\t\t\"Then use /model to select a model.\",\n\t\t\t);\n\t\t}\n\n\t\t// Validate API key\n\t\tconst apiKey = await getApiKeyForModel(this.model);\n\t\tif (!apiKey) {\n\t\t\tthrow new Error(\n\t\t\t\t`No API key found for ${this.model.provider}.\\n\\n` +\n\t\t\t\t\t`Set the appropriate environment variable or update ${getModelsPath()}`,\n\t\t\t);\n\t\t}\n\n\t\t// Expand slash commands if requested\n\t\tconst expandedText = expandCommands ? expandSlashCommand(text, [...this._fileCommands]) : text;\n\n\t\tawait this.agent.prompt(expandedText, options?.attachments);\n\t}\n\n\t/**\n\t * Queue a message to be sent after the current response completes.\n\t * Use when agent is currently streaming.\n\t */\n\tasync queueMessage(text: string): Promise<void> {\n\t\tthis._queuedMessages.push(text);\n\t\tawait this.agent.queueMessage({\n\t\t\trole: \"user\",\n\t\t\tcontent: [{ type: \"text\", text }],\n\t\t\ttimestamp: Date.now(),\n\t\t});\n\t}\n\n\t/**\n\t * Clear queued messages and return them.\n\t * Useful for restoring to editor when user aborts.\n\t */\n\tclearQueue(): string[] {\n\t\tconst queued = [...this._queuedMessages];\n\t\tthis._queuedMessages = [];\n\t\tthis.agent.clearMessageQueue();\n\t\treturn queued;\n\t}\n\n\t/** Number of messages currently queued */\n\tget queuedMessageCount(): number {\n\t\treturn this._queuedMessages.length;\n\t}\n\n\t/** Get queued messages (read-only) */\n\tgetQueuedMessages(): readonly string[] {\n\t\treturn this._queuedMessages;\n\t}\n\n\t/**\n\t * Abort current operation and wait for agent to become idle.\n\t */\n\tasync abort(): Promise<void> {\n\t\tthis.agent.abort();\n\t\tawait this.agent.waitForIdle();\n\t}\n\n\t/**\n\t * Reset agent and session to start fresh.\n\t * Clears all messages and starts a new session.\n\t * Listeners are preserved and will continue receiving events.\n\t */\n\tasync reset(): Promise<void> {\n\t\tthis._disconnectFromAgent();\n\t\tawait this.abort();\n\t\tthis.agent.reset();\n\t\tthis.sessionManager.reset();\n\t\tthis._queuedMessages = [];\n\t\tthis._reconnectToAgent();\n\t}\n\n\t// =========================================================================\n\t// Model Management\n\t// =========================================================================\n\n\t/**\n\t * Set model directly.\n\t * Validates API key, saves to session and settings.\n\t * @throws Error if no API key available for the model\n\t */\n\tasync setModel(model: Model<any>): Promise<void> {\n\t\tconst apiKey = await getApiKeyForModel(model);\n\t\tif (!apiKey) {\n\t\t\tthrow new Error(`No API key for ${model.provider}/${model.id}`);\n\t\t}\n\n\t\tthis.agent.setModel(model);\n\t\tthis.sessionManager.saveModelChange(model.provider, model.id);\n\t\tthis.settingsManager.setDefaultModelAndProvider(model.provider, model.id);\n\t}\n\n\t/**\n\t * Cycle to next model.\n\t * Uses scoped models (from --models flag) if available, otherwise all available models.\n\t * @returns The new model info, or null if only one model available\n\t */\n\tasync cycleModel(): Promise<ModelCycleResult | null> {\n\t\tif (this._scopedModels.length > 0) {\n\t\t\treturn this._cycleScopedModel();\n\t\t}\n\t\treturn this._cycleAvailableModel();\n\t}\n\n\tprivate async _cycleScopedModel(): Promise<ModelCycleResult | null> {\n\t\tif (this._scopedModels.length <= 1) return null;\n\n\t\tconst currentModel = this.model;\n\t\tlet currentIndex = this._scopedModels.findIndex(\n\t\t\t(sm) => sm.model.id === currentModel?.id && sm.model.provider === currentModel?.provider,\n\t\t);\n\n\t\tif (currentIndex === -1) currentIndex = 0;\n\t\tconst nextIndex = (currentIndex + 1) % this._scopedModels.length;\n\t\tconst next = this._scopedModels[nextIndex];\n\n\t\t// Validate API key\n\t\tconst apiKey = await getApiKeyForModel(next.model);\n\t\tif (!apiKey) {\n\t\t\tthrow new Error(`No API key for ${next.model.provider}/${next.model.id}`);\n\t\t}\n\n\t\t// Apply model\n\t\tthis.agent.setModel(next.model);\n\t\tthis.sessionManager.saveModelChange(next.model.provider, next.model.id);\n\t\tthis.settingsManager.setDefaultModelAndProvider(next.model.provider, next.model.id);\n\n\t\t// Apply thinking level (silently use \"off\" if not supported)\n\t\tconst effectiveThinking = next.model.reasoning ? next.thinkingLevel : \"off\";\n\t\tthis.agent.setThinkingLevel(effectiveThinking);\n\t\tthis.sessionManager.saveThinkingLevelChange(effectiveThinking);\n\t\tthis.settingsManager.setDefaultThinkingLevel(effectiveThinking);\n\n\t\treturn { model: next.model, thinkingLevel: effectiveThinking, isScoped: true };\n\t}\n\n\tprivate async _cycleAvailableModel(): Promise<ModelCycleResult | null> {\n\t\tconst { models: availableModels, error } = await getAvailableModels();\n\t\tif (error) throw new Error(`Failed to load models: ${error}`);\n\t\tif (availableModels.length <= 1) return null;\n\n\t\tconst currentModel = this.model;\n\t\tlet currentIndex = availableModels.findIndex(\n\t\t\t(m) => m.id === currentModel?.id && m.provider === currentModel?.provider,\n\t\t);\n\n\t\tif (currentIndex === -1) currentIndex = 0;\n\t\tconst nextIndex = (currentIndex + 1) % availableModels.length;\n\t\tconst nextModel = availableModels[nextIndex];\n\n\t\tconst apiKey = await getApiKeyForModel(nextModel);\n\t\tif (!apiKey) {\n\t\t\tthrow new Error(`No API key for ${nextModel.provider}/${nextModel.id}`);\n\t\t}\n\n\t\tthis.agent.setModel(nextModel);\n\t\tthis.sessionManager.saveModelChange(nextModel.provider, nextModel.id);\n\t\tthis.settingsManager.setDefaultModelAndProvider(nextModel.provider, nextModel.id);\n\n\t\treturn { model: nextModel, thinkingLevel: this.thinkingLevel, isScoped: false };\n\t}\n\n\t/**\n\t * Get all available models with valid API keys.\n\t */\n\tasync getAvailableModels(): Promise<Model<any>[]> {\n\t\tconst { models, error } = await getAvailableModels();\n\t\tif (error) throw new Error(error);\n\t\treturn models;\n\t}\n\n\t// =========================================================================\n\t// Thinking Level Management\n\t// =========================================================================\n\n\t/**\n\t * Set thinking level.\n\t * Silently uses \"off\" if model doesn't support thinking.\n\t * Saves to session and settings.\n\t */\n\tsetThinkingLevel(level: ThinkingLevel): void {\n\t\tconst effectiveLevel = this.supportsThinking() ? level : \"off\";\n\t\tthis.agent.setThinkingLevel(effectiveLevel);\n\t\tthis.sessionManager.saveThinkingLevelChange(effectiveLevel);\n\t\tthis.settingsManager.setDefaultThinkingLevel(effectiveLevel);\n\t}\n\n\t/**\n\t * Cycle to next thinking level.\n\t * @returns New level, or null if model doesn't support thinking\n\t */\n\tcycleThinkingLevel(): ThinkingLevel | null {\n\t\tif (!this.supportsThinking()) return null;\n\n\t\tconst modelId = this.model?.id || \"\";\n\t\tconst supportsXhigh = modelId.includes(\"codex-max\");\n\t\tconst levels: ThinkingLevel[] = supportsXhigh\n\t\t\t? [\"off\", \"minimal\", \"low\", \"medium\", \"high\", \"xhigh\"]\n\t\t\t: [\"off\", \"minimal\", \"low\", \"medium\", \"high\"];\n\n\t\tconst currentIndex = levels.indexOf(this.thinkingLevel);\n\t\tconst nextIndex = (currentIndex + 1) % levels.length;\n\t\tconst nextLevel = levels[nextIndex];\n\n\t\tthis.setThinkingLevel(nextLevel);\n\t\treturn nextLevel;\n\t}\n\n\t/**\n\t * Check if current model supports thinking/reasoning.\n\t */\n\tsupportsThinking(): boolean {\n\t\treturn !!this.model?.reasoning;\n\t}\n\n\t// =========================================================================\n\t// Queue Mode Management\n\t// =========================================================================\n\n\t/**\n\t * Set message queue mode.\n\t * Saves to settings.\n\t */\n\tsetQueueMode(mode: \"all\" | \"one-at-a-time\"): void {\n\t\tthis.agent.setQueueMode(mode);\n\t\tthis.settingsManager.setQueueMode(mode);\n\t}\n\n\t// =========================================================================\n\t// Compaction\n\t// =========================================================================\n\n\t/**\n\t * Manually compact the session context.\n\t * Aborts current agent operation first.\n\t * @param customInstructions Optional instructions for the compaction summary\n\t */\n\tasync compact(customInstructions?: string): Promise<CompactionResult> {\n\t\t// Abort any running operation\n\t\tthis._disconnectFromAgent();\n\t\tawait this.abort();\n\n\t\t// Create abort controller\n\t\tthis._compactionAbortController = new AbortController();\n\n\t\ttry {\n\t\t\tif (!this.model) {\n\t\t\t\tthrow new Error(\"No model selected\");\n\t\t\t}\n\n\t\t\tconst apiKey = await getApiKeyForModel(this.model);\n\t\t\tif (!apiKey) {\n\t\t\t\tthrow new Error(`No API key for ${this.model.provider}`);\n\t\t\t}\n\n\t\t\tconst entries = this.sessionManager.loadEntries();\n\t\t\tconst settings = this.settingsManager.getCompactionSettings();\n\t\t\tconst compactionEntry = await compact(\n\t\t\t\tentries,\n\t\t\t\tthis.model,\n\t\t\t\tsettings,\n\t\t\t\tapiKey,\n\t\t\t\tthis._compactionAbortController.signal,\n\t\t\t\tcustomInstructions,\n\t\t\t);\n\n\t\t\tif (this._compactionAbortController.signal.aborted) {\n\t\t\t\tthrow new Error(\"Compaction cancelled\");\n\t\t\t}\n\n\t\t\t// Save and reload\n\t\t\tthis.sessionManager.saveCompaction(compactionEntry);\n\t\t\tconst loaded = loadSessionFromEntries(this.sessionManager.loadEntries());\n\t\t\tthis.agent.replaceMessages(loaded.messages);\n\n\t\t\treturn {\n\t\t\t\ttokensBefore: compactionEntry.tokensBefore,\n\t\t\t\tsummary: compactionEntry.summary,\n\t\t\t};\n\t\t} finally {\n\t\t\tthis._compactionAbortController = null;\n\t\t\tthis._reconnectToAgent();\n\t\t}\n\t}\n\n\t/**\n\t * Cancel in-progress compaction (manual or auto).\n\t */\n\tabortCompaction(): void {\n\t\tthis._compactionAbortController?.abort();\n\t\tthis._autoCompactionAbortController?.abort();\n\t}\n\n\t/**\n\t * Internal: Run auto-compaction with events.\n\t * Called after assistant messages complete.\n\t */\n\tprivate async _runAutoCompaction(assistantMessage: AssistantMessage): Promise<void> {\n\t\tconst settings = this.settingsManager.getCompactionSettings();\n\t\tif (!settings.enabled) return;\n\n\t\t// Skip if message was aborted\n\t\tif (assistantMessage.stopReason === \"aborted\") return;\n\n\t\tconst contextTokens = calculateContextTokens(assistantMessage.usage);\n\t\tconst contextWindow = this.model?.contextWindow ?? 0;\n\n\t\tif (!shouldCompact(contextTokens, contextWindow, settings)) return;\n\n\t\t// Emit start event\n\t\tthis._emit({ type: \"auto_compaction_start\" });\n\t\tthis._autoCompactionAbortController = new AbortController();\n\n\t\ttry {\n\t\t\tif (!this.model) {\n\t\t\t\tthis._emit({ type: \"auto_compaction_end\", result: null, aborted: false });\n\t\t\t\treturn;\n\t\t\t}\n\n\t\t\tconst apiKey = await getApiKeyForModel(this.model);\n\t\t\tif (!apiKey) {\n\t\t\t\tthis._emit({ type: \"auto_compaction_end\", result: null, aborted: false });\n\t\t\t\treturn;\n\t\t\t}\n\n\t\t\t// Load entries (sync file read) then yield to let UI render\n\t\t\tconst entries = this.sessionManager.loadEntries();\n\t\t\tconst compactionEntry = await compact(\n\t\t\t\tentries,\n\t\t\t\tthis.model,\n\t\t\t\tsettings,\n\t\t\t\tapiKey,\n\t\t\t\tthis._autoCompactionAbortController.signal,\n\t\t\t);\n\n\t\t\tif (this._autoCompactionAbortController.signal.aborted) {\n\t\t\t\tthis._emit({ type: \"auto_compaction_end\", result: null, aborted: true });\n\t\t\t\treturn;\n\t\t\t}\n\n\t\t\tthis.sessionManager.saveCompaction(compactionEntry);\n\t\t\tconst loaded = loadSessionFromEntries(this.sessionManager.loadEntries());\n\t\t\tthis.agent.replaceMessages(loaded.messages);\n\n\t\t\tconst result: CompactionResult = {\n\t\t\t\ttokensBefore: compactionEntry.tokensBefore,\n\t\t\t\tsummary: compactionEntry.summary,\n\t\t\t};\n\t\t\tthis._emit({ type: \"auto_compaction_end\", result, aborted: false });\n\t\t} catch {\n\t\t\t// Silently fail auto-compaction but emit end event\n\t\t\tthis._emit({ type: \"auto_compaction_end\", result: null, aborted: false });\n\t\t} finally {\n\t\t\tthis._autoCompactionAbortController = null;\n\t\t}\n\t}\n\n\t/**\n\t * Toggle auto-compaction setting.\n\t */\n\tsetAutoCompactionEnabled(enabled: boolean): void {\n\t\tthis.settingsManager.setCompactionEnabled(enabled);\n\t}\n\n\t/** Whether auto-compaction is enabled */\n\tget autoCompactionEnabled(): boolean {\n\t\treturn this.settingsManager.getCompactionEnabled();\n\t}\n\n\t// =========================================================================\n\t// Bash Execution\n\t// =========================================================================\n\n\t/**\n\t * Execute a bash command.\n\t * Adds result to agent context and session.\n\t * @param command The bash command to execute\n\t * @param onChunk Optional streaming callback for output\n\t */\n\tasync executeBash(command: string, onChunk?: (chunk: string) => void): Promise<BashResult> {\n\t\tthis._bashAbortController = new AbortController();\n\n\t\ttry {\n\t\t\tconst result = await executeBashCommand(command, {\n\t\t\t\tonChunk,\n\t\t\t\tsignal: this._bashAbortController.signal,\n\t\t\t});\n\n\t\t\t// Create and save message\n\t\t\tconst bashMessage: BashExecutionMessage = {\n\t\t\t\trole: \"bashExecution\",\n\t\t\t\tcommand,\n\t\t\t\toutput: result.output,\n\t\t\t\texitCode: result.exitCode,\n\t\t\t\tcancelled: result.cancelled,\n\t\t\t\ttruncated: result.truncated,\n\t\t\t\tfullOutputPath: result.fullOutputPath,\n\t\t\t\ttimestamp: Date.now(),\n\t\t\t};\n\n\t\t\t// If agent is streaming, defer adding to avoid breaking tool_use/tool_result ordering\n\t\t\tif (this.isStreaming) {\n\t\t\t\t// Queue for later - will be flushed on agent_end\n\t\t\t\tthis._pendingBashMessages.push(bashMessage);\n\t\t\t} else {\n\t\t\t\t// Add to agent state immediately\n\t\t\t\tthis.agent.appendMessage(bashMessage);\n\n\t\t\t\t// Save to session\n\t\t\t\tthis.sessionManager.saveMessage(bashMessage);\n\n\t\t\t\t// Initialize session if needed\n\t\t\t\tif (this.sessionManager.shouldInitializeSession(this.agent.state.messages)) {\n\t\t\t\t\tthis.sessionManager.startSession(this.agent.state);\n\t\t\t\t}\n\t\t\t}\n\n\t\t\treturn result;\n\t\t} finally {\n\t\t\tthis._bashAbortController = null;\n\t\t}\n\t}\n\n\t/**\n\t * Cancel running bash command.\n\t */\n\tabortBash(): void {\n\t\tthis._bashAbortController?.abort();\n\t}\n\n\t/** Whether a bash command is currently running */\n\tget isBashRunning(): boolean {\n\t\treturn this._bashAbortController !== null;\n\t}\n\n\t/** Whether there are pending bash messages waiting to be flushed */\n\tget hasPendingBashMessages(): boolean {\n\t\treturn this._pendingBashMessages.length > 0;\n\t}\n\n\t/**\n\t * Flush pending bash messages to agent state and session.\n\t * Called after agent turn completes to maintain proper message ordering.\n\t */\n\tprivate _flushPendingBashMessages(): void {\n\t\tif (this._pendingBashMessages.length === 0) return;\n\n\t\tfor (const bashMessage of this._pendingBashMessages) {\n\t\t\t// Add to agent state\n\t\t\tthis.agent.appendMessage(bashMessage);\n\n\t\t\t// Save to session\n\t\t\tthis.sessionManager.saveMessage(bashMessage);\n\t\t}\n\n\t\t// Initialize session if needed\n\t\tif (this.sessionManager.shouldInitializeSession(this.agent.state.messages)) {\n\t\t\tthis.sessionManager.startSession(this.agent.state);\n\t\t}\n\n\t\tthis._pendingBashMessages = [];\n\t}\n\n\t// =========================================================================\n\t// Session Management\n\t// =========================================================================\n\n\t/**\n\t * Switch to a different session file.\n\t * Aborts current operation, loads messages, restores model/thinking.\n\t * Listeners are preserved and will continue receiving events.\n\t */\n\tasync switchSession(sessionPath: string): Promise<void> {\n\t\tthis._disconnectFromAgent();\n\t\tawait this.abort();\n\t\tthis._queuedMessages = [];\n\n\t\t// Set new session\n\t\tthis.sessionManager.setSessionFile(sessionPath);\n\n\t\t// Reload messages\n\t\tconst loaded = loadSessionFromEntries(this.sessionManager.loadEntries());\n\t\tthis.agent.replaceMessages(loaded.messages);\n\n\t\t// Restore model if saved\n\t\tconst savedModel = this.sessionManager.loadModel();\n\t\tif (savedModel) {\n\t\t\tconst availableModels = (await getAvailableModels()).models;\n\t\t\tconst match = availableModels.find((m) => m.provider === savedModel.provider && m.id === savedModel.modelId);\n\t\t\tif (match) {\n\t\t\t\tthis.agent.setModel(match);\n\t\t\t}\n\t\t}\n\n\t\t// Restore thinking level if saved\n\t\tconst savedThinking = this.sessionManager.loadThinkingLevel();\n\t\tif (savedThinking) {\n\t\t\tthis.agent.setThinkingLevel(savedThinking as ThinkingLevel);\n\t\t}\n\n\t\tthis._reconnectToAgent();\n\t}\n\n\t/**\n\t * Create a branch from a specific entry index.\n\t * @param entryIndex Index into session entries to branch from\n\t * @returns The text of the selected user message (for editor pre-fill)\n\t */\n\tbranch(entryIndex: number): string {\n\t\tconst entries = this.sessionManager.loadEntries();\n\t\tconst selectedEntry = entries[entryIndex];\n\n\t\tif (!selectedEntry || selectedEntry.type !== \"message\" || selectedEntry.message.role !== \"user\") {\n\t\t\tthrow new Error(\"Invalid entry index for branching\");\n\t\t}\n\n\t\tconst selectedText = this._extractUserMessageText(selectedEntry.message.content);\n\n\t\t// Create branched session\n\t\tconst newSessionFile = this.sessionManager.createBranchedSessionFromEntries(entries, entryIndex);\n\t\tthis.sessionManager.setSessionFile(newSessionFile);\n\n\t\t// Reload\n\t\tconst loaded = loadSessionFromEntries(this.sessionManager.loadEntries());\n\t\tthis.agent.replaceMessages(loaded.messages);\n\n\t\treturn selectedText;\n\t}\n\n\t/**\n\t * Get all user messages from session for branch selector.\n\t */\n\tgetUserMessagesForBranching(): Array<{ entryIndex: number; text: string }> {\n\t\tconst entries = this.sessionManager.loadEntries();\n\t\tconst result: Array<{ entryIndex: number; text: string }> = [];\n\n\t\tfor (let i = 0; i < entries.length; i++) {\n\t\t\tconst entry = entries[i];\n\t\t\tif (entry.type !== \"message\") continue;\n\t\t\tif (entry.message.role !== \"user\") continue;\n\n\t\t\tconst text = this._extractUserMessageText(entry.message.content);\n\t\t\tif (text) {\n\t\t\t\tresult.push({ entryIndex: i, text });\n\t\t\t}\n\t\t}\n\n\t\treturn result;\n\t}\n\n\tprivate _extractUserMessageText(content: string | Array<{ type: string; text?: string }>): string {\n\t\tif (typeof content === \"string\") return content;\n\t\tif (Array.isArray(content)) {\n\t\t\treturn content\n\t\t\t\t.filter((c): c is { type: \"text\"; text: string } => c.type === \"text\")\n\t\t\t\t.map((c) => c.text)\n\t\t\t\t.join(\"\");\n\t\t}\n\t\treturn \"\";\n\t}\n\n\t/**\n\t * Get session statistics.\n\t */\n\tgetSessionStats(): SessionStats {\n\t\tconst state = this.state;\n\t\tconst userMessages = state.messages.filter((m) => m.role === \"user\").length;\n\t\tconst assistantMessages = state.messages.filter((m) => m.role === \"assistant\").length;\n\t\tconst toolResults = state.messages.filter((m) => m.role === \"toolResult\").length;\n\n\t\tlet toolCalls = 0;\n\t\tlet totalInput = 0;\n\t\tlet totalOutput = 0;\n\t\tlet totalCacheRead = 0;\n\t\tlet totalCacheWrite = 0;\n\t\tlet totalCost = 0;\n\n\t\tfor (const message of state.messages) {\n\t\t\tif (message.role === \"assistant\") {\n\t\t\t\tconst assistantMsg = message as AssistantMessage;\n\t\t\t\ttoolCalls += assistantMsg.content.filter((c) => c.type === \"toolCall\").length;\n\t\t\t\ttotalInput += assistantMsg.usage.input;\n\t\t\t\ttotalOutput += assistantMsg.usage.output;\n\t\t\t\ttotalCacheRead += assistantMsg.usage.cacheRead;\n\t\t\t\ttotalCacheWrite += assistantMsg.usage.cacheWrite;\n\t\t\t\ttotalCost += assistantMsg.usage.cost.total;\n\t\t\t}\n\t\t}\n\n\t\treturn {\n\t\t\tsessionFile: this.sessionFile,\n\t\t\tsessionId: this.sessionId,\n\t\t\tuserMessages,\n\t\t\tassistantMessages,\n\t\t\ttoolCalls,\n\t\t\ttoolResults,\n\t\t\ttotalMessages: state.messages.length,\n\t\t\ttokens: {\n\t\t\t\tinput: totalInput,\n\t\t\t\toutput: totalOutput,\n\t\t\t\tcacheRead: totalCacheRead,\n\t\t\t\tcacheWrite: totalCacheWrite,\n\t\t\t\ttotal: totalInput + totalOutput + totalCacheRead + totalCacheWrite,\n\t\t\t},\n\t\t\tcost: totalCost,\n\t\t};\n\t}\n\n\t/**\n\t * Export session to HTML.\n\t * @param outputPath Optional output path (defaults to session directory)\n\t * @returns Path to exported file\n\t */\n\texportToHtml(outputPath?: string): string {\n\t\treturn exportSessionToHtml(this.sessionManager, this.state, outputPath);\n\t}\n\n\t// =========================================================================\n\t// Utilities\n\t// =========================================================================\n\n\t/**\n\t * Get text content of last assistant message.\n\t * Useful for /copy command.\n\t * @returns Text content, or null if no assistant message exists\n\t */\n\tgetLastAssistantText(): string | null {\n\t\tconst lastAssistant = this.messages\n\t\t\t.slice()\n\t\t\t.reverse()\n\t\t\t.find((m) => m.role === \"assistant\");\n\n\t\tif (!lastAssistant) return null;\n\n\t\tlet text = \"\";\n\t\tfor (const content of (lastAssistant as AssistantMessage).content) {\n\t\t\tif (content.type === \"text\") {\n\t\t\t\ttext += content.text;\n\t\t\t}\n\t\t}\n\n\t\treturn text.trim() || null;\n\t}\n}\n"]}
+{"version":3,"file":"agent-session.d.ts","sourceRoot":"","sources":["../../src/core/agent-session.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;GAaG;AAEH,OAAO,KAAK,EAAE,KAAK,EAAE,UAAU,EAAE,UAAU,EAAE,UAAU,EAAE,UAAU,EAAE,aAAa,EAAE,MAAM,6BAA6B,CAAC;AACxH,OAAO,KAAK,EAA6B,KAAK,EAAe,MAAM,qBAAqB,CAAC;AAGzF,OAAO,EAAE,KAAK,UAAU,EAAqC,MAAM,oBAAoB,CAAC;AAGxF,OAAO,KAAK,EAAqB,UAAU,EAAgC,MAAM,kBAAkB,CAAC;AAGpG,OAAO,EAA0B,KAAK,cAAc,EAAE,MAAM,sBAAsB,CAAC;AACnF,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,uBAAuB,CAAC;AAC7D,OAAO,EAAsB,KAAK,gBAAgB,EAAE,MAAM,qBAAqB,CAAC;AAEhF,8DAA8D;AAC9D,MAAM,MAAM,iBAAiB,GAC1B,UAAU,GACV;IAAE,IAAI,EAAE,uBAAuB,CAAC;IAAC,MAAM,EAAE,WAAW,GAAG,UAAU,CAAA;CAAE,GACnE;IAAE,IAAI,EAAE,qBAAqB,CAAC;IAAC,MAAM,EAAE,gBAAgB,GAAG,IAAI,CAAC;IAAC,OAAO,EAAE,OAAO,CAAC;IAAC,SAAS,EAAE,OAAO,CAAA;CAAE,CAAC;AAE1G,iDAAiD;AACjD,MAAM,MAAM,yBAAyB,GAAG,CAAC,KAAK,EAAE,iBAAiB,KAAK,IAAI,CAAC;AAM3E,MAAM,WAAW,kBAAkB;IAClC,KAAK,EAAE,KAAK,CAAC;IACb,cAAc,EAAE,cAAc,CAAC;IAC/B,eAAe,EAAE,eAAe,CAAC;IACjC,+DAA+D;IAC/D,YAAY,CAAC,EAAE,KAAK,CAAC;QAAE,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC;QAAC,aAAa,EAAE,aAAa,CAAA;KAAE,CAAC,CAAC;IAC1E,8CAA8C;IAC9C,YAAY,CAAC,EAAE,gBAAgB,EAAE,CAAC;IAClC,0DAA0D;IAC1D,UAAU,CAAC,EAAE,UAAU,GAAG,IAAI,CAAC;CAC/B;AAED,wCAAwC;AACxC,MAAM,WAAW,aAAa;IAC7B,kEAAkE;IAClE,mBAAmB,CAAC,EAAE,OAAO,CAAC;IAC9B,6BAA6B;IAC7B,WAAW,CAAC,EAAE,UAAU,EAAE,CAAC;CAC3B;AAED,+BAA+B;AAC/B,MAAM,WAAW,gBAAgB;IAChC,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC;IAClB,aAAa,EAAE,aAAa,CAAC;IAC7B,6EAA6E;IAC7E,QAAQ,EAAE,OAAO,CAAC;CAClB;AAED,qDAAqD;AACrD,MAAM,WAAW,gBAAgB;IAChC,YAAY,EAAE,MAAM,CAAC;IACrB,OAAO,EAAE,MAAM,CAAC;CAChB;AAED,8CAA8C;AAC9C,MAAM,WAAW,YAAY;IAC5B,WAAW,EAAE,MAAM,CAAC;IACpB,SAAS,EAAE,MAAM,CAAC;IAClB,YAAY,EAAE,MAAM,CAAC;IACrB,iBAAiB,EAAE,MAAM,CAAC;IAC1B,SAAS,EAAE,MAAM,CAAC;IAClB,WAAW,EAAE,MAAM,CAAC;IACpB,aAAa,EAAE,MAAM,CAAC;IACtB,MAAM,EAAE;QACP,KAAK,EAAE,MAAM,CAAC;QACd,MAAM,EAAE,MAAM,CAAC;QACf,SAAS,EAAE,MAAM,CAAC;QAClB,UAAU,EAAE,MAAM,CAAC;QACnB,KAAK,EAAE,MAAM,CAAC;KACd,CAAC;IACF,IAAI,EAAE,MAAM,CAAC;CACb;AAMD,qBAAa,YAAY;IACxB,QAAQ,CAAC,KAAK,EAAE,KAAK,CAAC;IACtB,QAAQ,CAAC,cAAc,EAAE,cAAc,CAAC;IACxC,QAAQ,CAAC,eAAe,EAAE,eAAe,CAAC;IAE1C,OAAO,CAAC,aAAa,CAA6D;IAClF,OAAO,CAAC,aAAa,CAAqB;IAG1C,OAAO,CAAC,iBAAiB,CAAC,CAAa;IACvC,OAAO,CAAC,eAAe,CAAmC;IAG1D,OAAO,CAAC,eAAe,CAAgB;IAGvC,OAAO,CAAC,0BAA0B,CAAgC;IAClE,OAAO,CAAC,8BAA8B,CAAgC;IAGtE,OAAO,CAAC,oBAAoB,CAAgC;IAC5D,OAAO,CAAC,oBAAoB,CAA8B;IAG1D,OAAO,CAAC,WAAW,CAA2B;IAC9C,OAAO,CAAC,UAAU,CAAK;IAEvB,YAAY,MAAM,EAAE,kBAAkB,EAOrC;IAMD,qCAAqC;IACrC,OAAO,CAAC,KAAK;IAOb,OAAO,CAAC,qBAAqB,CAAiC;IAE9D,4EAA4E;IAC5E,OAAO,CAAC,iBAAiB,CA0CvB;IAEF,0CAA0C;IAC1C,OAAO,CAAC,mBAAmB;YASb,cAAc;IA2B5B;;;;OAIG;IACH,SAAS,CAAC,QAAQ,EAAE,yBAAyB,GAAG,MAAM,IAAI,CAezD;IAED;;;;OAIG;IACH,OAAO,CAAC,oBAAoB;IAO5B;;;OAGG;IACH,OAAO,CAAC,iBAAiB;IAKzB;;;OAGG;IACH,OAAO,IAAI,IAAI,CAGd;IAMD,uBAAuB;IACvB,IAAI,KAAK,IAAI,UAAU,CAEtB;IAED,sDAAsD;IACtD,IAAI,KAAK,IAAI,KAAK,CAAC,GAAG,CAAC,GAAG,IAAI,CAE7B;IAED,6BAA6B;IAC7B,IAAI,aAAa,IAAI,aAAa,CAEjC;IAED,sDAAsD;IACtD,IAAI,WAAW,IAAI,OAAO,CAEzB;IAED,mDAAmD;IACnD,IAAI,YAAY,IAAI,OAAO,CAE1B;IAED,oEAAoE;IACpE,IAAI,QAAQ,IAAI,UAAU,EAAE,CAE3B;IAED,yBAAyB;IACzB,IAAI,SAAS,IAAI,KAAK,GAAG,eAAe,CAEvC;IAED,gCAAgC;IAChC,IAAI,WAAW,IAAI,MAAM,CAExB;IAED,yBAAyB;IACzB,IAAI,SAAS,IAAI,MAAM,CAEtB;IAED,qDAAqD;IACrD,IAAI,YAAY,IAAI,aAAa,CAAC;QAAE,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC;QAAC,aAAa,EAAE,aAAa,CAAA;KAAE,CAAC,CAErF;IAED,gCAAgC;IAChC,IAAI,YAAY,IAAI,aAAa,CAAC,gBAAgB,CAAC,CAElD;IAMD;;;;;OAKG;IACG,MAAM,CAAC,IAAI,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,aAAa,GAAG,OAAO,CAAC,IAAI,CAAC,CA6BjE;IAED;;;OAGG;IACG,YAAY,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAO9C;IAED;;;OAGG;IACH,UAAU,IAAI,MAAM,EAAE,CAKrB;IAED,0CAA0C;IAC1C,IAAI,kBAAkB,IAAI,MAAM,CAE/B;IAED,sCAAsC;IACtC,iBAAiB,IAAI,SAAS,MAAM,EAAE,CAErC;IAED;;OAEG;IACG,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC,CAG3B;IAED;;;;OAIG;IACG,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC,CAO3B;IAMD;;;;OAIG;IACG,QAAQ,CAAC,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,GAAG,OAAO,CAAC,IAAI,CAAC,CAS/C;IAED;;;;OAIG;IACG,UAAU,IAAI,OAAO,CAAC,gBAAgB,GAAG,IAAI,CAAC,CAKnD;YAEa,iBAAiB;YAgCjB,oBAAoB;IA0BlC;;OAEG;IACG,kBAAkB,IAAI,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC,CAIhD;IAMD;;;;OAIG;IACH,gBAAgB,CAAC,KAAK,EAAE,aAAa,GAAG,IAAI,CAK3C;IAED;;;OAGG;IACH,kBAAkB,IAAI,aAAa,GAAG,IAAI,CAezC;IAED;;OAEG;IACH,gBAAgB,IAAI,OAAO,CAE1B;IAMD;;;OAGG;IACH,YAAY,CAAC,IAAI,EAAE,KAAK,GAAG,eAAe,GAAG,IAAI,CAGhD;IAMD;;;;OAIG;IACG,OAAO,CAAC,kBAAkB,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,gBAAgB,CAAC,CA8CpE;IAED;;OAEG;IACH,eAAe,IAAI,IAAI,CAGtB;YAQa,yBAAyB;YAkCzB,kBAAkB;IA0EhC;;OAEG;IACH,wBAAwB,CAAC,OAAO,EAAE,OAAO,GAAG,IAAI,CAE/C;IAED,yCAAyC;IACzC,IAAI,qBAAqB,IAAI,OAAO,CAEnC;IAMD;;;;;OAKG;IACG,WAAW,CAAC,OAAO,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,IAAI,GAAG,OAAO,CAAC,UAAU,CAAC,CA0CzF;IAED;;OAEG;IACH,SAAS,IAAI,IAAI,CAEhB;IAED,kDAAkD;IAClD,IAAI,aAAa,IAAI,OAAO,CAE3B;IAED,oEAAoE;IACpE,IAAI,sBAAsB,IAAI,OAAO,CAEpC;IAED;;;OAGG;IACH,OAAO,CAAC,yBAAyB;IAuBjC;;;;OAIG;IACG,aAAa,CAAC,WAAW,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CA0CtD;IAED;;;;;;;;OAQG;IACG,MAAM,CAAC,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC;QAAE,YAAY,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,OAAO,CAAA;KAAE,CAAC,CA8CpF;IAED;;OAEG;IACH,2BAA2B,IAAI,KAAK,CAAC;QAAE,UAAU,EAAE,MAAM,CAAC;QAAC,IAAI,EAAE,MAAM,CAAA;KAAE,CAAC,CAgBzE;IAED,OAAO,CAAC,uBAAuB;IAW/B;;OAEG;IACH,eAAe,IAAI,YAAY,CA0C9B;IAED;;;;OAIG;IACH,YAAY,CAAC,UAAU,CAAC,EAAE,MAAM,GAAG,MAAM,CAExC;IAMD;;;;OAIG;IACH,oBAAoB,IAAI,MAAM,GAAG,IAAI,CAgBpC;IAMD;;OAEG;IACH,eAAe,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAE1C;IAED;;OAEG;IACH,IAAI,UAAU,IAAI,UAAU,GAAG,IAAI,CAElC;CACD","sourcesContent":["/**\n * AgentSession - Core abstraction for agent lifecycle and session management.\n *\n * This class is shared between all run modes (interactive, print, rpc).\n * It encapsulates:\n * - Agent state access\n * - Event subscription with automatic session persistence\n * - Model and thinking level management\n * - Compaction (manual and auto)\n * - Bash execution\n * - Session switching and branching\n *\n * Modes use this class and add their own I/O layer on top.\n */\n\nimport type { Agent, AgentEvent, AgentState, AppMessage, Attachment, ThinkingLevel } from \"@mariozechner/pi-agent-core\";\nimport type { AssistantMessage, Message, Model, TextContent } from \"@mariozechner/pi-ai\";\nimport { isContextOverflow } from \"@mariozechner/pi-ai\";\nimport { getModelsPath } from \"../config.js\";\nimport { type BashResult, executeBash as executeBashCommand } from \"./bash-executor.js\";\nimport { calculateContextTokens, compact, shouldCompact } from \"./compaction.js\";\nimport { exportSessionToHtml } from \"./export-html.js\";\nimport type { BranchEventResult, HookRunner, TurnEndEvent, TurnStartEvent } from \"./hooks/index.js\";\nimport type { BashExecutionMessage } from \"./messages.js\";\nimport { getApiKeyForModel, getAvailableModels } from \"./model-config.js\";\nimport { loadSessionFromEntries, type SessionManager } from \"./session-manager.js\";\nimport type { SettingsManager } from \"./settings-manager.js\";\nimport { expandSlashCommand, type FileSlashCommand } from \"./slash-commands.js\";\n\n/** Session-specific events that extend the core AgentEvent */\nexport type AgentSessionEvent =\n\t| AgentEvent\n\t| { type: \"auto_compaction_start\"; reason: \"threshold\" | \"overflow\" }\n\t| { type: \"auto_compaction_end\"; result: CompactionResult | null; aborted: boolean; willRetry: boolean };\n\n/** Listener function for agent session events */\nexport type AgentSessionEventListener = (event: AgentSessionEvent) => void;\n\n// ============================================================================\n// Types\n// ============================================================================\n\nexport interface AgentSessionConfig {\n\tagent: Agent;\n\tsessionManager: SessionManager;\n\tsettingsManager: SettingsManager;\n\t/** Models to cycle through with Ctrl+P (from --models flag) */\n\tscopedModels?: Array<{ model: Model<any>; thinkingLevel: ThinkingLevel }>;\n\t/** File-based slash commands for expansion */\n\tfileCommands?: FileSlashCommand[];\n\t/** Hook runner (created in main.ts with wrapped tools) */\n\thookRunner?: HookRunner | null;\n}\n\n/** Options for AgentSession.prompt() */\nexport interface PromptOptions {\n\t/** Whether to expand file-based slash commands (default: true) */\n\texpandSlashCommands?: boolean;\n\t/** Image/file attachments */\n\tattachments?: Attachment[];\n}\n\n/** Result from cycleModel() */\nexport interface ModelCycleResult {\n\tmodel: Model<any>;\n\tthinkingLevel: ThinkingLevel;\n\t/** Whether cycling through scoped models (--models flag) or all available */\n\tisScoped: boolean;\n}\n\n/** Result from compact() or checkAutoCompaction() */\nexport interface CompactionResult {\n\ttokensBefore: number;\n\tsummary: string;\n}\n\n/** Session statistics for /session command */\nexport interface SessionStats {\n\tsessionFile: string;\n\tsessionId: string;\n\tuserMessages: number;\n\tassistantMessages: number;\n\ttoolCalls: number;\n\ttoolResults: number;\n\ttotalMessages: number;\n\ttokens: {\n\t\tinput: number;\n\t\toutput: number;\n\t\tcacheRead: number;\n\t\tcacheWrite: number;\n\t\ttotal: number;\n\t};\n\tcost: number;\n}\n\n// ============================================================================\n// AgentSession Class\n// ============================================================================\n\nexport class AgentSession {\n\treadonly agent: Agent;\n\treadonly sessionManager: SessionManager;\n\treadonly settingsManager: SettingsManager;\n\n\tprivate _scopedModels: Array<{ model: Model<any>; thinkingLevel: ThinkingLevel }>;\n\tprivate _fileCommands: FileSlashCommand[];\n\n\t// Event subscription state\n\tprivate _unsubscribeAgent?: () => void;\n\tprivate _eventListeners: AgentSessionEventListener[] = [];\n\n\t// Message queue state\n\tprivate _queuedMessages: string[] = [];\n\n\t// Compaction state\n\tprivate _compactionAbortController: AbortController | null = null;\n\tprivate _autoCompactionAbortController: AbortController | null = null;\n\n\t// Bash execution state\n\tprivate _bashAbortController: AbortController | null = null;\n\tprivate _pendingBashMessages: BashExecutionMessage[] = [];\n\n\t// Hook system\n\tprivate _hookRunner: HookRunner | null = null;\n\tprivate _turnIndex = 0;\n\n\tconstructor(config: AgentSessionConfig) {\n\t\tthis.agent = config.agent;\n\t\tthis.sessionManager = config.sessionManager;\n\t\tthis.settingsManager = config.settingsManager;\n\t\tthis._scopedModels = config.scopedModels ?? [];\n\t\tthis._fileCommands = config.fileCommands ?? [];\n\t\tthis._hookRunner = config.hookRunner ?? null;\n\t}\n\n\t// =========================================================================\n\t// Event Subscription\n\t// =========================================================================\n\n\t/** Emit an event to all listeners */\n\tprivate _emit(event: AgentSessionEvent): void {\n\t\tfor (const l of this._eventListeners) {\n\t\t\tl(event);\n\t\t}\n\t}\n\n\t// Track last assistant message for auto-compaction check\n\tprivate _lastAssistantMessage: AssistantMessage | null = null;\n\n\t/** Internal handler for agent events - shared by subscribe and reconnect */\n\tprivate _handleAgentEvent = async (event: AgentEvent): Promise<void> => {\n\t\t// When a user message starts, check if it's from the queue and remove it BEFORE emitting\n\t\t// This ensures the UI sees the updated queue state\n\t\tif (event.type === \"message_start\" && event.message.role === \"user\" && this._queuedMessages.length > 0) {\n\t\t\t// Extract text content from the message\n\t\t\tconst messageText = this._getUserMessageText(event.message);\n\t\t\tif (messageText && this._queuedMessages.includes(messageText)) {\n\t\t\t\t// Remove the first occurrence of this message from the queue\n\t\t\t\tconst index = this._queuedMessages.indexOf(messageText);\n\t\t\t\tif (index !== -1) {\n\t\t\t\t\tthis._queuedMessages.splice(index, 1);\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\n\t\t// Emit to hooks first\n\t\tawait this._emitHookEvent(event);\n\n\t\t// Notify all listeners\n\t\tthis._emit(event);\n\n\t\t// Handle session persistence\n\t\tif (event.type === \"message_end\") {\n\t\t\tthis.sessionManager.saveMessage(event.message);\n\n\t\t\t// Initialize session after first user+assistant exchange\n\t\t\tif (this.sessionManager.shouldInitializeSession(this.agent.state.messages)) {\n\t\t\t\tthis.sessionManager.startSession(this.agent.state);\n\t\t\t}\n\n\t\t\t// Track assistant message for auto-compaction (checked on agent_end)\n\t\t\tif (event.message.role === \"assistant\") {\n\t\t\t\tthis._lastAssistantMessage = event.message;\n\t\t\t}\n\t\t}\n\n\t\t// Check auto-compaction after agent completes\n\t\tif (event.type === \"agent_end\" && this._lastAssistantMessage) {\n\t\t\tconst msg = this._lastAssistantMessage;\n\t\t\tthis._lastAssistantMessage = null;\n\t\t\tawait this._handleAgentEndCompaction(msg);\n\t\t}\n\t};\n\n\t/** Extract text content from a message */\n\tprivate _getUserMessageText(message: Message): string {\n\t\tif (message.role !== \"user\") return \"\";\n\t\tconst content = message.content;\n\t\tif (typeof content === \"string\") return content;\n\t\tconst textBlocks = content.filter((c) => c.type === \"text\");\n\t\treturn textBlocks.map((c) => (c as TextContent).text).join(\"\");\n\t}\n\n\t/** Emit hook events based on agent events */\n\tprivate async _emitHookEvent(event: AgentEvent): Promise<void> {\n\t\tif (!this._hookRunner) return;\n\n\t\tif (event.type === \"agent_start\") {\n\t\t\tthis._turnIndex = 0;\n\t\t\tawait this._hookRunner.emit({ type: \"agent_start\" });\n\t\t} else if (event.type === \"agent_end\") {\n\t\t\tawait this._hookRunner.emit({ type: \"agent_end\", messages: event.messages });\n\t\t} else if (event.type === \"turn_start\") {\n\t\t\tconst hookEvent: TurnStartEvent = {\n\t\t\t\ttype: \"turn_start\",\n\t\t\t\tturnIndex: this._turnIndex,\n\t\t\t\ttimestamp: Date.now(),\n\t\t\t};\n\t\t\tawait this._hookRunner.emit(hookEvent);\n\t\t} else if (event.type === \"turn_end\") {\n\t\t\tconst hookEvent: TurnEndEvent = {\n\t\t\t\ttype: \"turn_end\",\n\t\t\t\tturnIndex: this._turnIndex,\n\t\t\t\tmessage: event.message,\n\t\t\t\ttoolResults: event.toolResults,\n\t\t\t};\n\t\t\tawait this._hookRunner.emit(hookEvent);\n\t\t\tthis._turnIndex++;\n\t\t}\n\t}\n\n\t/**\n\t * Subscribe to agent events.\n\t * Session persistence is handled internally (saves messages on message_end).\n\t * Multiple listeners can be added. Returns unsubscribe function for this listener.\n\t */\n\tsubscribe(listener: AgentSessionEventListener): () => void {\n\t\tthis._eventListeners.push(listener);\n\n\t\t// Set up agent subscription if not already done\n\t\tif (!this._unsubscribeAgent) {\n\t\t\tthis._unsubscribeAgent = this.agent.subscribe(this._handleAgentEvent);\n\t\t}\n\n\t\t// Return unsubscribe function for this specific listener\n\t\treturn () => {\n\t\t\tconst index = this._eventListeners.indexOf(listener);\n\t\t\tif (index !== -1) {\n\t\t\t\tthis._eventListeners.splice(index, 1);\n\t\t\t}\n\t\t};\n\t}\n\n\t/**\n\t * Temporarily disconnect from agent events.\n\t * User listeners are preserved and will receive events again after resubscribe().\n\t * Used internally during operations that need to pause event processing.\n\t */\n\tprivate _disconnectFromAgent(): void {\n\t\tif (this._unsubscribeAgent) {\n\t\t\tthis._unsubscribeAgent();\n\t\t\tthis._unsubscribeAgent = undefined;\n\t\t}\n\t}\n\n\t/**\n\t * Reconnect to agent events after _disconnectFromAgent().\n\t * Preserves all existing listeners.\n\t */\n\tprivate _reconnectToAgent(): void {\n\t\tif (this._unsubscribeAgent) return; // Already connected\n\t\tthis._unsubscribeAgent = this.agent.subscribe(this._handleAgentEvent);\n\t}\n\n\t/**\n\t * Remove all listeners and disconnect from agent.\n\t * Call this when completely done with the session.\n\t */\n\tdispose(): void {\n\t\tthis._disconnectFromAgent();\n\t\tthis._eventListeners = [];\n\t}\n\n\t// =========================================================================\n\t// Read-only State Access\n\t// =========================================================================\n\n\t/** Full agent state */\n\tget state(): AgentState {\n\t\treturn this.agent.state;\n\t}\n\n\t/** Current model (may be null if not yet selected) */\n\tget model(): Model<any> | null {\n\t\treturn this.agent.state.model;\n\t}\n\n\t/** Current thinking level */\n\tget thinkingLevel(): ThinkingLevel {\n\t\treturn this.agent.state.thinkingLevel;\n\t}\n\n\t/** Whether agent is currently streaming a response */\n\tget isStreaming(): boolean {\n\t\treturn this.agent.state.isStreaming;\n\t}\n\n\t/** Whether auto-compaction is currently running */\n\tget isCompacting(): boolean {\n\t\treturn this._autoCompactionAbortController !== null || this._compactionAbortController !== null;\n\t}\n\n\t/** All messages including custom types like BashExecutionMessage */\n\tget messages(): AppMessage[] {\n\t\treturn this.agent.state.messages;\n\t}\n\n\t/** Current queue mode */\n\tget queueMode(): \"all\" | \"one-at-a-time\" {\n\t\treturn this.agent.getQueueMode();\n\t}\n\n\t/** Current session file path */\n\tget sessionFile(): string {\n\t\treturn this.sessionManager.getSessionFile();\n\t}\n\n\t/** Current session ID */\n\tget sessionId(): string {\n\t\treturn this.sessionManager.getSessionId();\n\t}\n\n\t/** Scoped models for cycling (from --models flag) */\n\tget scopedModels(): ReadonlyArray<{ model: Model<any>; thinkingLevel: ThinkingLevel }> {\n\t\treturn this._scopedModels;\n\t}\n\n\t/** File-based slash commands */\n\tget fileCommands(): ReadonlyArray<FileSlashCommand> {\n\t\treturn this._fileCommands;\n\t}\n\n\t// =========================================================================\n\t// Prompting\n\t// =========================================================================\n\n\t/**\n\t * Send a prompt to the agent.\n\t * - Validates model and API key before sending\n\t * - Expands file-based slash commands by default\n\t * @throws Error if no model selected or no API key available\n\t */\n\tasync prompt(text: string, options?: PromptOptions): Promise<void> {\n\t\t// Flush any pending bash messages before the new prompt\n\t\tthis._flushPendingBashMessages();\n\n\t\tconst expandCommands = options?.expandSlashCommands ?? true;\n\n\t\t// Validate model\n\t\tif (!this.model) {\n\t\t\tthrow new Error(\n\t\t\t\t\"No model selected.\\n\\n\" +\n\t\t\t\t\t\"Set an API key (ANTHROPIC_API_KEY, OPENAI_API_KEY, etc.)\\n\" +\n\t\t\t\t\t`or create ${getModelsPath()}\\n\\n` +\n\t\t\t\t\t\"Then use /model to select a model.\",\n\t\t\t);\n\t\t}\n\n\t\t// Validate API key\n\t\tconst apiKey = await getApiKeyForModel(this.model);\n\t\tif (!apiKey) {\n\t\t\tthrow new Error(\n\t\t\t\t`No API key found for ${this.model.provider}.\\n\\n` +\n\t\t\t\t\t`Set the appropriate environment variable or update ${getModelsPath()}`,\n\t\t\t);\n\t\t}\n\n\t\t// Expand slash commands if requested\n\t\tconst expandedText = expandCommands ? expandSlashCommand(text, [...this._fileCommands]) : text;\n\n\t\tawait this.agent.prompt(expandedText, options?.attachments);\n\t}\n\n\t/**\n\t * Queue a message to be sent after the current response completes.\n\t * Use when agent is currently streaming.\n\t */\n\tasync queueMessage(text: string): Promise<void> {\n\t\tthis._queuedMessages.push(text);\n\t\tawait this.agent.queueMessage({\n\t\t\trole: \"user\",\n\t\t\tcontent: [{ type: \"text\", text }],\n\t\t\ttimestamp: Date.now(),\n\t\t});\n\t}\n\n\t/**\n\t * Clear queued messages and return them.\n\t * Useful for restoring to editor when user aborts.\n\t */\n\tclearQueue(): string[] {\n\t\tconst queued = [...this._queuedMessages];\n\t\tthis._queuedMessages = [];\n\t\tthis.agent.clearMessageQueue();\n\t\treturn queued;\n\t}\n\n\t/** Number of messages currently queued */\n\tget queuedMessageCount(): number {\n\t\treturn this._queuedMessages.length;\n\t}\n\n\t/** Get queued messages (read-only) */\n\tgetQueuedMessages(): readonly string[] {\n\t\treturn this._queuedMessages;\n\t}\n\n\t/**\n\t * Abort current operation and wait for agent to become idle.\n\t */\n\tasync abort(): Promise<void> {\n\t\tthis.agent.abort();\n\t\tawait this.agent.waitForIdle();\n\t}\n\n\t/**\n\t * Reset agent and session to start fresh.\n\t * Clears all messages and starts a new session.\n\t * Listeners are preserved and will continue receiving events.\n\t */\n\tasync reset(): Promise<void> {\n\t\tthis._disconnectFromAgent();\n\t\tawait this.abort();\n\t\tthis.agent.reset();\n\t\tthis.sessionManager.reset();\n\t\tthis._queuedMessages = [];\n\t\tthis._reconnectToAgent();\n\t}\n\n\t// =========================================================================\n\t// Model Management\n\t// =========================================================================\n\n\t/**\n\t * Set model directly.\n\t * Validates API key, saves to session and settings.\n\t * @throws Error if no API key available for the model\n\t */\n\tasync setModel(model: Model<any>): Promise<void> {\n\t\tconst apiKey = await getApiKeyForModel(model);\n\t\tif (!apiKey) {\n\t\t\tthrow new Error(`No API key for ${model.provider}/${model.id}`);\n\t\t}\n\n\t\tthis.agent.setModel(model);\n\t\tthis.sessionManager.saveModelChange(model.provider, model.id);\n\t\tthis.settingsManager.setDefaultModelAndProvider(model.provider, model.id);\n\t}\n\n\t/**\n\t * Cycle to next model.\n\t * Uses scoped models (from --models flag) if available, otherwise all available models.\n\t * @returns The new model info, or null if only one model available\n\t */\n\tasync cycleModel(): Promise<ModelCycleResult | null> {\n\t\tif (this._scopedModels.length > 0) {\n\t\t\treturn this._cycleScopedModel();\n\t\t}\n\t\treturn this._cycleAvailableModel();\n\t}\n\n\tprivate async _cycleScopedModel(): Promise<ModelCycleResult | null> {\n\t\tif (this._scopedModels.length <= 1) return null;\n\n\t\tconst currentModel = this.model;\n\t\tlet currentIndex = this._scopedModels.findIndex(\n\t\t\t(sm) => sm.model.id === currentModel?.id && sm.model.provider === currentModel?.provider,\n\t\t);\n\n\t\tif (currentIndex === -1) currentIndex = 0;\n\t\tconst nextIndex = (currentIndex + 1) % this._scopedModels.length;\n\t\tconst next = this._scopedModels[nextIndex];\n\n\t\t// Validate API key\n\t\tconst apiKey = await getApiKeyForModel(next.model);\n\t\tif (!apiKey) {\n\t\t\tthrow new Error(`No API key for ${next.model.provider}/${next.model.id}`);\n\t\t}\n\n\t\t// Apply model\n\t\tthis.agent.setModel(next.model);\n\t\tthis.sessionManager.saveModelChange(next.model.provider, next.model.id);\n\t\tthis.settingsManager.setDefaultModelAndProvider(next.model.provider, next.model.id);\n\n\t\t// Apply thinking level (silently use \"off\" if not supported)\n\t\tconst effectiveThinking = next.model.reasoning ? next.thinkingLevel : \"off\";\n\t\tthis.agent.setThinkingLevel(effectiveThinking);\n\t\tthis.sessionManager.saveThinkingLevelChange(effectiveThinking);\n\t\tthis.settingsManager.setDefaultThinkingLevel(effectiveThinking);\n\n\t\treturn { model: next.model, thinkingLevel: effectiveThinking, isScoped: true };\n\t}\n\n\tprivate async _cycleAvailableModel(): Promise<ModelCycleResult | null> {\n\t\tconst { models: availableModels, error } = await getAvailableModels();\n\t\tif (error) throw new Error(`Failed to load models: ${error}`);\n\t\tif (availableModels.length <= 1) return null;\n\n\t\tconst currentModel = this.model;\n\t\tlet currentIndex = availableModels.findIndex(\n\t\t\t(m) => m.id === currentModel?.id && m.provider === currentModel?.provider,\n\t\t);\n\n\t\tif (currentIndex === -1) currentIndex = 0;\n\t\tconst nextIndex = (currentIndex + 1) % availableModels.length;\n\t\tconst nextModel = availableModels[nextIndex];\n\n\t\tconst apiKey = await getApiKeyForModel(nextModel);\n\t\tif (!apiKey) {\n\t\t\tthrow new Error(`No API key for ${nextModel.provider}/${nextModel.id}`);\n\t\t}\n\n\t\tthis.agent.setModel(nextModel);\n\t\tthis.sessionManager.saveModelChange(nextModel.provider, nextModel.id);\n\t\tthis.settingsManager.setDefaultModelAndProvider(nextModel.provider, nextModel.id);\n\n\t\treturn { model: nextModel, thinkingLevel: this.thinkingLevel, isScoped: false };\n\t}\n\n\t/**\n\t * Get all available models with valid API keys.\n\t */\n\tasync getAvailableModels(): Promise<Model<any>[]> {\n\t\tconst { models, error } = await getAvailableModels();\n\t\tif (error) throw new Error(error);\n\t\treturn models;\n\t}\n\n\t// =========================================================================\n\t// Thinking Level Management\n\t// =========================================================================\n\n\t/**\n\t * Set thinking level.\n\t * Silently uses \"off\" if model doesn't support thinking.\n\t * Saves to session and settings.\n\t */\n\tsetThinkingLevel(level: ThinkingLevel): void {\n\t\tconst effectiveLevel = this.supportsThinking() ? level : \"off\";\n\t\tthis.agent.setThinkingLevel(effectiveLevel);\n\t\tthis.sessionManager.saveThinkingLevelChange(effectiveLevel);\n\t\tthis.settingsManager.setDefaultThinkingLevel(effectiveLevel);\n\t}\n\n\t/**\n\t * Cycle to next thinking level.\n\t * @returns New level, or null if model doesn't support thinking\n\t */\n\tcycleThinkingLevel(): ThinkingLevel | null {\n\t\tif (!this.supportsThinking()) return null;\n\n\t\tconst modelId = this.model?.id || \"\";\n\t\tconst supportsXhigh = modelId.includes(\"codex-max\");\n\t\tconst levels: ThinkingLevel[] = supportsXhigh\n\t\t\t? [\"off\", \"minimal\", \"low\", \"medium\", \"high\", \"xhigh\"]\n\t\t\t: [\"off\", \"minimal\", \"low\", \"medium\", \"high\"];\n\n\t\tconst currentIndex = levels.indexOf(this.thinkingLevel);\n\t\tconst nextIndex = (currentIndex + 1) % levels.length;\n\t\tconst nextLevel = levels[nextIndex];\n\n\t\tthis.setThinkingLevel(nextLevel);\n\t\treturn nextLevel;\n\t}\n\n\t/**\n\t * Check if current model supports thinking/reasoning.\n\t */\n\tsupportsThinking(): boolean {\n\t\treturn !!this.model?.reasoning;\n\t}\n\n\t// =========================================================================\n\t// Queue Mode Management\n\t// =========================================================================\n\n\t/**\n\t * Set message queue mode.\n\t * Saves to settings.\n\t */\n\tsetQueueMode(mode: \"all\" | \"one-at-a-time\"): void {\n\t\tthis.agent.setQueueMode(mode);\n\t\tthis.settingsManager.setQueueMode(mode);\n\t}\n\n\t// =========================================================================\n\t// Compaction\n\t// =========================================================================\n\n\t/**\n\t * Manually compact the session context.\n\t * Aborts current agent operation first.\n\t * @param customInstructions Optional instructions for the compaction summary\n\t */\n\tasync compact(customInstructions?: string): Promise<CompactionResult> {\n\t\t// Abort any running operation\n\t\tthis._disconnectFromAgent();\n\t\tawait this.abort();\n\n\t\t// Create abort controller\n\t\tthis._compactionAbortController = new AbortController();\n\n\t\ttry {\n\t\t\tif (!this.model) {\n\t\t\t\tthrow new Error(\"No model selected\");\n\t\t\t}\n\n\t\t\tconst apiKey = await getApiKeyForModel(this.model);\n\t\t\tif (!apiKey) {\n\t\t\t\tthrow new Error(`No API key for ${this.model.provider}`);\n\t\t\t}\n\n\t\t\tconst entries = this.sessionManager.loadEntries();\n\t\t\tconst settings = this.settingsManager.getCompactionSettings();\n\t\t\tconst compactionEntry = await compact(\n\t\t\t\tentries,\n\t\t\t\tthis.model,\n\t\t\t\tsettings,\n\t\t\t\tapiKey,\n\t\t\t\tthis._compactionAbortController.signal,\n\t\t\t\tcustomInstructions,\n\t\t\t);\n\n\t\t\tif (this._compactionAbortController.signal.aborted) {\n\t\t\t\tthrow new Error(\"Compaction cancelled\");\n\t\t\t}\n\n\t\t\t// Save and reload\n\t\t\tthis.sessionManager.saveCompaction(compactionEntry);\n\t\t\tconst loaded = loadSessionFromEntries(this.sessionManager.loadEntries());\n\t\t\tthis.agent.replaceMessages(loaded.messages);\n\n\t\t\treturn {\n\t\t\t\ttokensBefore: compactionEntry.tokensBefore,\n\t\t\t\tsummary: compactionEntry.summary,\n\t\t\t};\n\t\t} finally {\n\t\t\tthis._compactionAbortController = null;\n\t\t\tthis._reconnectToAgent();\n\t\t}\n\t}\n\n\t/**\n\t * Cancel in-progress compaction (manual or auto).\n\t */\n\tabortCompaction(): void {\n\t\tthis._compactionAbortController?.abort();\n\t\tthis._autoCompactionAbortController?.abort();\n\t}\n\n\t/**\n\t * Handle compaction after agent_end.\n\t * Two cases:\n\t * 1. Overflow: LLM returned context overflow error, remove error message from agent state, compact, auto-retry\n\t * 2. Threshold: Turn succeeded but context over threshold, compact, NO auto-retry (user continues manually)\n\t */\n\tprivate async _handleAgentEndCompaction(assistantMessage: AssistantMessage): Promise<void> {\n\t\tconst settings = this.settingsManager.getCompactionSettings();\n\t\tif (!settings.enabled) return;\n\n\t\t// Skip if message was aborted (user cancelled)\n\t\tif (assistantMessage.stopReason === \"aborted\") return;\n\n\t\tconst contextWindow = this.model?.contextWindow ?? 0;\n\n\t\t// Case 1: Overflow - LLM returned context overflow error\n\t\tif (isContextOverflow(assistantMessage, contextWindow)) {\n\t\t\t// Remove the error message from agent state (it IS saved to session for history,\n\t\t\t// but we don't want it in context for the retry)\n\t\t\tconst messages = this.agent.state.messages;\n\t\t\tif (messages.length > 0 && messages[messages.length - 1].role === \"assistant\") {\n\t\t\t\tthis.agent.replaceMessages(messages.slice(0, -1));\n\t\t\t}\n\t\t\tawait this._runAutoCompaction(\"overflow\", true);\n\t\t\treturn;\n\t\t}\n\n\t\t// Case 2: Threshold - turn succeeded but context is getting large\n\t\t// Skip if this was an error (non-overflow errors don't have usage data)\n\t\tif (assistantMessage.stopReason === \"error\") return;\n\n\t\tconst contextTokens = calculateContextTokens(assistantMessage.usage);\n\t\tif (shouldCompact(contextTokens, contextWindow, settings)) {\n\t\t\tawait this._runAutoCompaction(\"threshold\", false);\n\t\t}\n\t}\n\n\t/**\n\t * Internal: Run auto-compaction with events.\n\t */\n\tprivate async _runAutoCompaction(reason: \"overflow\" | \"threshold\", willRetry: boolean): Promise<void> {\n\t\tconst settings = this.settingsManager.getCompactionSettings();\n\n\t\tthis._emit({ type: \"auto_compaction_start\", reason });\n\t\tthis._autoCompactionAbortController = new AbortController();\n\n\t\ttry {\n\t\t\tif (!this.model) {\n\t\t\t\tthis._emit({ type: \"auto_compaction_end\", result: null, aborted: false, willRetry: false });\n\t\t\t\treturn;\n\t\t\t}\n\n\t\t\tconst apiKey = await getApiKeyForModel(this.model);\n\t\t\tif (!apiKey) {\n\t\t\t\tthis._emit({ type: \"auto_compaction_end\", result: null, aborted: false, willRetry: false });\n\t\t\t\treturn;\n\t\t\t}\n\n\t\t\tconst entries = this.sessionManager.loadEntries();\n\t\t\tconst compactionEntry = await compact(\n\t\t\t\tentries,\n\t\t\t\tthis.model,\n\t\t\t\tsettings,\n\t\t\t\tapiKey,\n\t\t\t\tthis._autoCompactionAbortController.signal,\n\t\t\t);\n\n\t\t\tif (this._autoCompactionAbortController.signal.aborted) {\n\t\t\t\tthis._emit({ type: \"auto_compaction_end\", result: null, aborted: true, willRetry: false });\n\t\t\t\treturn;\n\t\t\t}\n\n\t\t\tthis.sessionManager.saveCompaction(compactionEntry);\n\t\t\tconst loaded = loadSessionFromEntries(this.sessionManager.loadEntries());\n\t\t\tthis.agent.replaceMessages(loaded.messages);\n\n\t\t\tconst result: CompactionResult = {\n\t\t\t\ttokensBefore: compactionEntry.tokensBefore,\n\t\t\t\tsummary: compactionEntry.summary,\n\t\t\t};\n\t\t\tthis._emit({ type: \"auto_compaction_end\", result, aborted: false, willRetry });\n\n\t\t\t// Auto-retry if needed - use continue() since user message is already in context\n\t\t\tif (willRetry) {\n\t\t\t\t// Remove trailing error message from agent state (it's kept in session file for history)\n\t\t\t\t// This is needed because continue() requires last message to be user or toolResult\n\t\t\t\tconst messages = this.agent.state.messages;\n\t\t\t\tconst lastMsg = messages[messages.length - 1];\n\t\t\t\tif (lastMsg?.role === \"assistant\" && (lastMsg as AssistantMessage).stopReason === \"error\") {\n\t\t\t\t\tthis.agent.replaceMessages(messages.slice(0, -1));\n\t\t\t\t}\n\n\t\t\t\t// Use setTimeout to break out of the event handler chain\n\t\t\t\tsetTimeout(() => {\n\t\t\t\t\tthis.agent.continue().catch(() => {\n\t\t\t\t\t\t// Retry failed - silently ignore, user can manually retry\n\t\t\t\t\t});\n\t\t\t\t}, 100);\n\t\t\t}\n\t\t} catch (error) {\n\t\t\t// Compaction failed - emit end event without retry\n\t\t\tthis._emit({ type: \"auto_compaction_end\", result: null, aborted: false, willRetry: false });\n\n\t\t\t// If this was overflow recovery and compaction failed, we have a hard stop\n\t\t\tif (reason === \"overflow\") {\n\t\t\t\tthrow new Error(\n\t\t\t\t\t`Context overflow: ${error instanceof Error ? error.message : \"compaction failed\"}. Your input may be too large for the context window.`,\n\t\t\t\t);\n\t\t\t}\n\t\t} finally {\n\t\t\tthis._autoCompactionAbortController = null;\n\t\t}\n\t}\n\n\t/**\n\t * Toggle auto-compaction setting.\n\t */\n\tsetAutoCompactionEnabled(enabled: boolean): void {\n\t\tthis.settingsManager.setCompactionEnabled(enabled);\n\t}\n\n\t/** Whether auto-compaction is enabled */\n\tget autoCompactionEnabled(): boolean {\n\t\treturn this.settingsManager.getCompactionEnabled();\n\t}\n\n\t// =========================================================================\n\t// Bash Execution\n\t// =========================================================================\n\n\t/**\n\t * Execute a bash command.\n\t * Adds result to agent context and session.\n\t * @param command The bash command to execute\n\t * @param onChunk Optional streaming callback for output\n\t */\n\tasync executeBash(command: string, onChunk?: (chunk: string) => void): Promise<BashResult> {\n\t\tthis._bashAbortController = new AbortController();\n\n\t\ttry {\n\t\t\tconst result = await executeBashCommand(command, {\n\t\t\t\tonChunk,\n\t\t\t\tsignal: this._bashAbortController.signal,\n\t\t\t});\n\n\t\t\t// Create and save message\n\t\t\tconst bashMessage: BashExecutionMessage = {\n\t\t\t\trole: \"bashExecution\",\n\t\t\t\tcommand,\n\t\t\t\toutput: result.output,\n\t\t\t\texitCode: result.exitCode,\n\t\t\t\tcancelled: result.cancelled,\n\t\t\t\ttruncated: result.truncated,\n\t\t\t\tfullOutputPath: result.fullOutputPath,\n\t\t\t\ttimestamp: Date.now(),\n\t\t\t};\n\n\t\t\t// If agent is streaming, defer adding to avoid breaking tool_use/tool_result ordering\n\t\t\tif (this.isStreaming) {\n\t\t\t\t// Queue for later - will be flushed on agent_end\n\t\t\t\tthis._pendingBashMessages.push(bashMessage);\n\t\t\t} else {\n\t\t\t\t// Add to agent state immediately\n\t\t\t\tthis.agent.appendMessage(bashMessage);\n\n\t\t\t\t// Save to session\n\t\t\t\tthis.sessionManager.saveMessage(bashMessage);\n\n\t\t\t\t// Initialize session if needed\n\t\t\t\tif (this.sessionManager.shouldInitializeSession(this.agent.state.messages)) {\n\t\t\t\t\tthis.sessionManager.startSession(this.agent.state);\n\t\t\t\t}\n\t\t\t}\n\n\t\t\treturn result;\n\t\t} finally {\n\t\t\tthis._bashAbortController = null;\n\t\t}\n\t}\n\n\t/**\n\t * Cancel running bash command.\n\t */\n\tabortBash(): void {\n\t\tthis._bashAbortController?.abort();\n\t}\n\n\t/** Whether a bash command is currently running */\n\tget isBashRunning(): boolean {\n\t\treturn this._bashAbortController !== null;\n\t}\n\n\t/** Whether there are pending bash messages waiting to be flushed */\n\tget hasPendingBashMessages(): boolean {\n\t\treturn this._pendingBashMessages.length > 0;\n\t}\n\n\t/**\n\t * Flush pending bash messages to agent state and session.\n\t * Called after agent turn completes to maintain proper message ordering.\n\t */\n\tprivate _flushPendingBashMessages(): void {\n\t\tif (this._pendingBashMessages.length === 0) return;\n\n\t\tfor (const bashMessage of this._pendingBashMessages) {\n\t\t\t// Add to agent state\n\t\t\tthis.agent.appendMessage(bashMessage);\n\n\t\t\t// Save to session\n\t\t\tthis.sessionManager.saveMessage(bashMessage);\n\t\t}\n\n\t\t// Initialize session if needed\n\t\tif (this.sessionManager.shouldInitializeSession(this.agent.state.messages)) {\n\t\t\tthis.sessionManager.startSession(this.agent.state);\n\t\t}\n\n\t\tthis._pendingBashMessages = [];\n\t}\n\n\t// =========================================================================\n\t// Session Management\n\t// =========================================================================\n\n\t/**\n\t * Switch to a different session file.\n\t * Aborts current operation, loads messages, restores model/thinking.\n\t * Listeners are preserved and will continue receiving events.\n\t */\n\tasync switchSession(sessionPath: string): Promise<void> {\n\t\tconst previousSessionFile = this.sessionFile;\n\n\t\tthis._disconnectFromAgent();\n\t\tawait this.abort();\n\t\tthis._queuedMessages = [];\n\n\t\t// Set new session\n\t\tthis.sessionManager.setSessionFile(sessionPath);\n\n\t\t// Emit session_switch event\n\t\tif (this._hookRunner) {\n\t\t\tthis._hookRunner.setSessionFile(sessionPath);\n\t\t\tawait this._hookRunner.emit({\n\t\t\t\ttype: \"session_switch\",\n\t\t\t\tnewSessionFile: sessionPath,\n\t\t\t\tpreviousSessionFile,\n\t\t\t\treason: \"switch\",\n\t\t\t});\n\t\t}\n\n\t\t// Reload messages\n\t\tconst loaded = loadSessionFromEntries(this.sessionManager.loadEntries());\n\t\tthis.agent.replaceMessages(loaded.messages);\n\n\t\t// Restore model if saved\n\t\tconst savedModel = this.sessionManager.loadModel();\n\t\tif (savedModel) {\n\t\t\tconst availableModels = (await getAvailableModels()).models;\n\t\t\tconst match = availableModels.find((m) => m.provider === savedModel.provider && m.id === savedModel.modelId);\n\t\t\tif (match) {\n\t\t\t\tthis.agent.setModel(match);\n\t\t\t}\n\t\t}\n\n\t\t// Restore thinking level if saved\n\t\tconst savedThinking = this.sessionManager.loadThinkingLevel();\n\t\tif (savedThinking) {\n\t\t\tthis.agent.setThinkingLevel(savedThinking as ThinkingLevel);\n\t\t}\n\n\t\tthis._reconnectToAgent();\n\t}\n\n\t/**\n\t * Create a branch from a specific entry index.\n\t * Emits branch event to hooks, which can control the branch behavior.\n\t *\n\t * @param entryIndex Index into session entries to branch from\n\t * @returns Object with:\n\t *   - selectedText: The text of the selected user message (for editor pre-fill)\n\t *   - skipped: True if a hook requested to skip conversation restore\n\t */\n\tasync branch(entryIndex: number): Promise<{ selectedText: string; skipped: boolean }> {\n\t\tconst previousSessionFile = this.sessionFile;\n\t\tconst entries = this.sessionManager.loadEntries();\n\t\tconst selectedEntry = entries[entryIndex];\n\n\t\tif (!selectedEntry || selectedEntry.type !== \"message\" || selectedEntry.message.role !== \"user\") {\n\t\t\tthrow new Error(\"Invalid entry index for branching\");\n\t\t}\n\n\t\tconst selectedText = this._extractUserMessageText(selectedEntry.message.content);\n\n\t\t// Emit branch event to hooks\n\t\tlet hookResult: BranchEventResult | undefined;\n\t\tif (this._hookRunner?.hasHandlers(\"branch\")) {\n\t\t\thookResult = (await this._hookRunner.emit({\n\t\t\t\ttype: \"branch\",\n\t\t\t\ttargetTurnIndex: entryIndex,\n\t\t\t\tentries,\n\t\t\t})) as BranchEventResult | undefined;\n\t\t}\n\n\t\t// If hook says skip conversation restore, don't branch\n\t\tif (hookResult?.skipConversationRestore) {\n\t\t\treturn { selectedText, skipped: true };\n\t\t}\n\n\t\t// Create branched session\n\t\tconst newSessionFile = this.sessionManager.createBranchedSessionFromEntries(entries, entryIndex);\n\t\tthis.sessionManager.setSessionFile(newSessionFile);\n\n\t\t// Emit session_switch event\n\t\tif (this._hookRunner) {\n\t\t\tthis._hookRunner.setSessionFile(newSessionFile);\n\t\t\tawait this._hookRunner.emit({\n\t\t\t\ttype: \"session_switch\",\n\t\t\t\tnewSessionFile,\n\t\t\t\tpreviousSessionFile,\n\t\t\t\treason: \"branch\",\n\t\t\t});\n\t\t}\n\n\t\t// Reload\n\t\tconst loaded = loadSessionFromEntries(this.sessionManager.loadEntries());\n\t\tthis.agent.replaceMessages(loaded.messages);\n\n\t\treturn { selectedText, skipped: false };\n\t}\n\n\t/**\n\t * Get all user messages from session for branch selector.\n\t */\n\tgetUserMessagesForBranching(): Array<{ entryIndex: number; text: string }> {\n\t\tconst entries = this.sessionManager.loadEntries();\n\t\tconst result: Array<{ entryIndex: number; text: string }> = [];\n\n\t\tfor (let i = 0; i < entries.length; i++) {\n\t\t\tconst entry = entries[i];\n\t\t\tif (entry.type !== \"message\") continue;\n\t\t\tif (entry.message.role !== \"user\") continue;\n\n\t\t\tconst text = this._extractUserMessageText(entry.message.content);\n\t\t\tif (text) {\n\t\t\t\tresult.push({ entryIndex: i, text });\n\t\t\t}\n\t\t}\n\n\t\treturn result;\n\t}\n\n\tprivate _extractUserMessageText(content: string | Array<{ type: string; text?: string }>): string {\n\t\tif (typeof content === \"string\") return content;\n\t\tif (Array.isArray(content)) {\n\t\t\treturn content\n\t\t\t\t.filter((c): c is { type: \"text\"; text: string } => c.type === \"text\")\n\t\t\t\t.map((c) => c.text)\n\t\t\t\t.join(\"\");\n\t\t}\n\t\treturn \"\";\n\t}\n\n\t/**\n\t * Get session statistics.\n\t */\n\tgetSessionStats(): SessionStats {\n\t\tconst state = this.state;\n\t\tconst userMessages = state.messages.filter((m) => m.role === \"user\").length;\n\t\tconst assistantMessages = state.messages.filter((m) => m.role === \"assistant\").length;\n\t\tconst toolResults = state.messages.filter((m) => m.role === \"toolResult\").length;\n\n\t\tlet toolCalls = 0;\n\t\tlet totalInput = 0;\n\t\tlet totalOutput = 0;\n\t\tlet totalCacheRead = 0;\n\t\tlet totalCacheWrite = 0;\n\t\tlet totalCost = 0;\n\n\t\tfor (const message of state.messages) {\n\t\t\tif (message.role === \"assistant\") {\n\t\t\t\tconst assistantMsg = message as AssistantMessage;\n\t\t\t\ttoolCalls += assistantMsg.content.filter((c) => c.type === \"toolCall\").length;\n\t\t\t\ttotalInput += assistantMsg.usage.input;\n\t\t\t\ttotalOutput += assistantMsg.usage.output;\n\t\t\t\ttotalCacheRead += assistantMsg.usage.cacheRead;\n\t\t\t\ttotalCacheWrite += assistantMsg.usage.cacheWrite;\n\t\t\t\ttotalCost += assistantMsg.usage.cost.total;\n\t\t\t}\n\t\t}\n\n\t\treturn {\n\t\t\tsessionFile: this.sessionFile,\n\t\t\tsessionId: this.sessionId,\n\t\t\tuserMessages,\n\t\t\tassistantMessages,\n\t\t\ttoolCalls,\n\t\t\ttoolResults,\n\t\t\ttotalMessages: state.messages.length,\n\t\t\ttokens: {\n\t\t\t\tinput: totalInput,\n\t\t\t\toutput: totalOutput,\n\t\t\t\tcacheRead: totalCacheRead,\n\t\t\t\tcacheWrite: totalCacheWrite,\n\t\t\t\ttotal: totalInput + totalOutput + totalCacheRead + totalCacheWrite,\n\t\t\t},\n\t\t\tcost: totalCost,\n\t\t};\n\t}\n\n\t/**\n\t * Export session to HTML.\n\t * @param outputPath Optional output path (defaults to session directory)\n\t * @returns Path to exported file\n\t */\n\texportToHtml(outputPath?: string): string {\n\t\treturn exportSessionToHtml(this.sessionManager, this.state, outputPath);\n\t}\n\n\t// =========================================================================\n\t// Utilities\n\t// =========================================================================\n\n\t/**\n\t * Get text content of last assistant message.\n\t * Useful for /copy command.\n\t * @returns Text content, or null if no assistant message exists\n\t */\n\tgetLastAssistantText(): string | null {\n\t\tconst lastAssistant = this.messages\n\t\t\t.slice()\n\t\t\t.reverse()\n\t\t\t.find((m) => m.role === \"assistant\");\n\n\t\tif (!lastAssistant) return null;\n\n\t\tlet text = \"\";\n\t\tfor (const content of (lastAssistant as AssistantMessage).content) {\n\t\t\tif (content.type === \"text\") {\n\t\t\t\ttext += content.text;\n\t\t\t}\n\t\t}\n\n\t\treturn text.trim() || null;\n\t}\n\n\t// =========================================================================\n\t// Hook System\n\t// =========================================================================\n\n\t/**\n\t * Check if hooks have handlers for a specific event type.\n\t */\n\thasHookHandlers(eventType: string): boolean {\n\t\treturn this._hookRunner?.hasHandlers(eventType) ?? false;\n\t}\n\n\t/**\n\t * Get the hook runner (for setting UI context and error handlers).\n\t */\n\tget hookRunner(): HookRunner | null {\n\t\treturn this._hookRunner;\n\t}\n}\n"]}

package/dist/core/agent-session.js

@@ -12,6 +12,7 @@
  *
  * Modes use this class and add their own I/O layer on top.
  */
+import { isContextOverflow } from "@mariozechner/pi-ai";
 import { getModelsPath } from "../config.js";
 import { executeBash as executeBashCommand } from "./bash-executor.js";
 import { calculateContextTokens, compact, shouldCompact } from "./compaction.js";
@@ -39,12 +40,16 @@ export class AgentSession {
     // Bash execution state
     _bashAbortController = null;
     _pendingBashMessages = [];
+    // Hook system
+    _hookRunner = null;
+    _turnIndex = 0;
     constructor(config) {
         this.agent = config.agent;
         this.sessionManager = config.sessionManager;
         this.settingsManager = config.settingsManager;
         this._scopedModels = config.scopedModels ?? [];
         this._fileCommands = config.fileCommands ?? [];
+        this._hookRunner = config.hookRunner ?? null;
     }
     // =========================================================================
     // Event Subscription
@@ -59,6 +64,21 @@ export class AgentSession {
     _lastAssistantMessage = null;
     /** Internal handler for agent events - shared by subscribe and reconnect */
     _handleAgentEvent = async (event) => {
+        // When a user message starts, check if it's from the 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._queuedMessages.length > 0) {
+            // Extract text content from the message
+            const messageText = this._getUserMessageText(event.message);
+            if (messageText && this._queuedMessages.includes(messageText)) {
+                // Remove the first occurrence of this message from the queue
+                const index = this._queuedMessages.indexOf(messageText);
+                if (index !== -1) {
+                    this._queuedMessages.splice(index, 1);
+                }
+            }
+        }
+        // Emit to hooks first
+        await this._emitHookEvent(event);
         // Notify all listeners
         this._emit(event);
         // Handle session persistence
@@ -73,13 +93,53 @@ export class AgentSession {
                 this._lastAssistantMessage = event.message;
             }
         }
-        // Check auto-compaction after agent completes (after agent_end clears UI)
+        // Check auto-compaction after agent completes
         if (event.type === "agent_end" && this._lastAssistantMessage) {
             const msg = this._lastAssistantMessage;
             this._lastAssistantMessage = null;
-            this._runAutoCompaction(msg).catch(() => { });
+            await this._handleAgentEndCompaction(msg);
         }
     };
+    /** 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("");
+    }
+    /** Emit hook events based on agent events */
+    async _emitHookEvent(event) {
+        if (!this._hookRunner)
+            return;
+        if (event.type === "agent_start") {
+            this._turnIndex = 0;
+            await this._hookRunner.emit({ type: "agent_start" });
+        }
+        else if (event.type === "agent_end") {
+            await this._hookRunner.emit({ type: "agent_end", messages: event.messages });
+        }
+        else if (event.type === "turn_start") {
+            const hookEvent = {
+                type: "turn_start",
+                turnIndex: this._turnIndex,
+                timestamp: Date.now(),
+            };
+            await this._hookRunner.emit(hookEvent);
+        }
+        else if (event.type === "turn_end") {
+            const hookEvent = {
+                type: "turn_end",
+                turnIndex: this._turnIndex,
+                message: event.message,
+                toolResults: event.toolResults,
+            };
+            await this._hookRunner.emit(hookEvent);
+            this._turnIndex++;
+        }
+    }
     /**
      * Subscribe to agent events.
      * Session persistence is handled internally (saves messages on message_end).
@@ -146,6 +206,10 @@ export class AgentSession {
     get isStreaming() {
         return this.agent.state.isStreaming;
     }
+    /** Whether auto-compaction is currently running */
+    get isCompacting() {
+        return this._autoCompactionAbortController !== null || this._compactionAbortController !== null;
+    }
     /** All messages including custom types like BashExecutionMessage */
     get messages() {
         return this.agent.state.messages;
@@ -432,38 +496,60 @@ export class AgentSession {
         this._autoCompactionAbortController?.abort();
     }
     /**
-     * Internal: Run auto-compaction with events.
-     * Called after assistant messages complete.
+     * Handle compaction after agent_end.
+     * Two cases:
+     * 1. Overflow: LLM returned context overflow error, remove error message from agent state, compact, auto-retry
+     * 2. Threshold: Turn succeeded but context over threshold, compact, NO auto-retry (user continues manually)
      */
-    async _runAutoCompaction(assistantMessage) {
+    async _handleAgentEndCompaction(assistantMessage) {
         const settings = this.settingsManager.getCompactionSettings();
         if (!settings.enabled)
             return;
-        // Skip if message was aborted
+        // Skip if message was aborted (user cancelled)
         if (assistantMessage.stopReason === "aborted")
             return;
-        const contextTokens = calculateContextTokens(assistantMessage.usage);
         const contextWindow = this.model?.contextWindow ?? 0;
-        if (!shouldCompact(contextTokens, contextWindow, settings))
+        // Case 1: Overflow - LLM returned context overflow error
+        if (isContextOverflow(assistantMessage, contextWindow)) {
+            // 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.replaceMessages(messages.slice(0, -1));
+            }
+            await this._runAutoCompaction("overflow", true);
+            return;
+        }
+        // Case 2: Threshold - turn succeeded but context is getting large
+        // Skip if this was an error (non-overflow errors don't have usage data)
+        if (assistantMessage.stopReason === "error")
             return;
-        // Emit start event
-        this._emit({ type: "auto_compaction_start" });
+        const contextTokens = calculateContextTokens(assistantMessage.usage);
+        if (shouldCompact(contextTokens, contextWindow, settings)) {
+            await this._runAutoCompaction("threshold", false);
+        }
+    }
+    /**
+     * Internal: Run auto-compaction with events.
+     */
+    async _runAutoCompaction(reason, willRetry) {
+        const settings = this.settingsManager.getCompactionSettings();
+        this._emit({ type: "auto_compaction_start", reason });
         this._autoCompactionAbortController = new AbortController();
         try {
             if (!this.model) {
-                this._emit({ type: "auto_compaction_end", result: null, aborted: false });
+                this._emit({ type: "auto_compaction_end", result: null, aborted: false, willRetry: false });
                 return;
             }
             const apiKey = await getApiKeyForModel(this.model);
             if (!apiKey) {
-                this._emit({ type: "auto_compaction_end", result: null, aborted: false });
+                this._emit({ type: "auto_compaction_end", result: null, aborted: false, willRetry: false });
                 return;
             }
-            // Load entries (sync file read) then yield to let UI render
             const entries = this.sessionManager.loadEntries();
             const compactionEntry = await compact(entries, this.model, settings, apiKey, this._autoCompactionAbortController.signal);
             if (this._autoCompactionAbortController.signal.aborted) {
-                this._emit({ type: "auto_compaction_end", result: null, aborted: true });
+                this._emit({ type: "auto_compaction_end", result: null, aborted: true, willRetry: false });
                 return;
             }
             this.sessionManager.saveCompaction(compactionEntry);
@@ -473,11 +559,31 @@ export class AgentSession {
                 tokensBefore: compactionEntry.tokensBefore,
                 summary: compactionEntry.summary,
             };
-            this._emit({ type: "auto_compaction_end", result, aborted: false });
+            this._emit({ type: "auto_compaction_end", result, aborted: false, willRetry });
+            // Auto-retry if needed - use continue() since user message is already in context
+            if (willRetry) {
+                // Remove trailing error message from agent state (it's kept in session file for history)
+                // This is needed because continue() requires last message to be user or toolResult
+                const messages = this.agent.state.messages;
+                const lastMsg = messages[messages.length - 1];
+                if (lastMsg?.role === "assistant" && lastMsg.stopReason === "error") {
+                    this.agent.replaceMessages(messages.slice(0, -1));
+                }
+                // Use setTimeout to break out of the event handler chain
+                setTimeout(() => {
+                    this.agent.continue().catch(() => {
+                        // Retry failed - silently ignore, user can manually retry
+                    });
+                }, 100);
+            }
         }
-        catch {
-            // Silently fail auto-compaction but emit end event
-            this._emit({ type: "auto_compaction_end", result: null, aborted: false });
+        catch (error) {
+            // Compaction failed - emit end event without retry
+            this._emit({ type: "auto_compaction_end", result: null, aborted: false, willRetry: false });
+            // If this was overflow recovery and compaction failed, we have a hard stop
+            if (reason === "overflow") {
+                throw new Error(`Context overflow: ${error instanceof Error ? error.message : "compaction failed"}. Your input may be too large for the context window.`);
+            }
         }
         finally {
             this._autoCompactionAbortController = null;
@@ -583,11 +689,22 @@ export class AgentSession {
      * Listeners are preserved and will continue receiving events.
      */
     async switchSession(sessionPath) {
+        const previousSessionFile = this.sessionFile;
         this._disconnectFromAgent();
         await this.abort();
         this._queuedMessages = [];
         // Set new session
         this.sessionManager.setSessionFile(sessionPath);
+        // Emit session_switch event
+        if (this._hookRunner) {
+            this._hookRunner.setSessionFile(sessionPath);
+            await this._hookRunner.emit({
+                type: "session_switch",
+                newSessionFile: sessionPath,
+                previousSessionFile,
+                reason: "switch",
+            });
+        }
         // Reload messages
         const loaded = loadSessionFromEntries(this.sessionManager.loadEntries());
         this.agent.replaceMessages(loaded.messages);
@@ -609,23 +726,51 @@ export class AgentSession {
     }
     /**
      * Create a branch from a specific entry index.
+     * Emits branch event to hooks, which can control the branch behavior.
+     *
      * @param entryIndex Index into session entries to branch from
-     * @returns The text of the selected user message (for editor pre-fill)
+     * @returns Object with:
+     *   - selectedText: The text of the selected user message (for editor pre-fill)
+     *   - skipped: True if a hook requested to skip conversation restore
      */
-    branch(entryIndex) {
+    async branch(entryIndex) {
+        const previousSessionFile = this.sessionFile;
         const entries = this.sessionManager.loadEntries();
         const selectedEntry = entries[entryIndex];
         if (!selectedEntry || selectedEntry.type !== "message" || selectedEntry.message.role !== "user") {
             throw new Error("Invalid entry index for branching");
         }
         const selectedText = this._extractUserMessageText(selectedEntry.message.content);
+        // Emit branch event to hooks
+        let hookResult;
+        if (this._hookRunner?.hasHandlers("branch")) {
+            hookResult = (await this._hookRunner.emit({
+                type: "branch",
+                targetTurnIndex: entryIndex,
+                entries,
+            }));
+        }
+        // If hook says skip conversation restore, don't branch
+        if (hookResult?.skipConversationRestore) {
+            return { selectedText, skipped: true };
+        }
         // Create branched session
         const newSessionFile = this.sessionManager.createBranchedSessionFromEntries(entries, entryIndex);
         this.sessionManager.setSessionFile(newSessionFile);
+        // Emit session_switch event
+        if (this._hookRunner) {
+            this._hookRunner.setSessionFile(newSessionFile);
+            await this._hookRunner.emit({
+                type: "session_switch",
+                newSessionFile,
+                previousSessionFile,
+                reason: "branch",
+            });
+        }
         // Reload
         const loaded = loadSessionFromEntries(this.sessionManager.loadEntries());
         this.agent.replaceMessages(loaded.messages);
-        return selectedText;
+        return { selectedText, skipped: false };
     }
     /**
      * Get all user messages from session for branch selector.
@@ -731,5 +876,20 @@ export class AgentSession {
         }
         return text.trim() || null;
     }
+    // =========================================================================
+    // Hook System
+    // =========================================================================
+    /**
+     * Check if hooks have handlers for a specific event type.
+     */
+    hasHookHandlers(eventType) {
+        return this._hookRunner?.hasHandlers(eventType) ?? false;
+    }
+    /**
+     * Get the hook runner (for setting UI context and error handlers).
+     */
+    get hookRunner() {
+        return this._hookRunner;
+    }
 }
 //# sourceMappingURL=agent-session.js.map