@mariozechner/pi-coding-agent 0.27.5 → 0.27.6

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (43) hide show
  1. package/CHANGELOG.md +14 -1
  2. package/dist/core/agent-session.d.ts +1 -1
  3. package/dist/core/agent-session.d.ts.map +1 -1
  4. package/dist/core/agent-session.js +49 -30
  5. package/dist/core/agent-session.js.map +1 -1
  6. package/dist/core/compaction.d.ts +3 -0
  7. package/dist/core/compaction.d.ts.map +1 -1
  8. package/dist/core/compaction.js +10 -1
  9. package/dist/core/compaction.js.map +1 -1
  10. package/dist/core/hooks/runner.d.ts.map +1 -1
  11. package/dist/core/hooks/runner.js +11 -3
  12. package/dist/core/hooks/runner.js.map +1 -1
  13. package/dist/core/hooks/types.d.ts +9 -1
  14. package/dist/core/hooks/types.d.ts.map +1 -1
  15. package/dist/core/hooks/types.js.map +1 -1
  16. package/dist/core/model-config.d.ts +7 -2
  17. package/dist/core/model-config.d.ts.map +1 -1
  18. package/dist/core/model-config.js +7 -2
  19. package/dist/core/model-config.js.map +1 -1
  20. package/dist/core/sdk.d.ts.map +1 -1
  21. package/dist/core/sdk.js +1 -1
  22. package/dist/core/sdk.js.map +1 -1
  23. package/dist/core/session-manager.d.ts +24 -11
  24. package/dist/core/session-manager.d.ts.map +1 -1
  25. package/dist/core/session-manager.js +25 -21
  26. package/dist/core/session-manager.js.map +1 -1
  27. package/dist/index.d.ts +1 -1
  28. package/dist/index.d.ts.map +1 -1
  29. package/dist/index.js +1 -1
  30. package/dist/index.js.map +1 -1
  31. package/dist/modes/interactive/interactive-mode.d.ts.map +1 -1
  32. package/dist/modes/interactive/interactive-mode.js +5 -5
  33. package/dist/modes/interactive/interactive-mode.js.map +1 -1
  34. package/dist/modes/print-mode.d.ts.map +1 -1
  35. package/dist/modes/print-mode.js +1 -1
  36. package/dist/modes/print-mode.js.map +1 -1
  37. package/dist/modes/rpc/rpc-mode.d.ts.map +1 -1
  38. package/dist/modes/rpc/rpc-mode.js +1 -1
  39. package/dist/modes/rpc/rpc-mode.js.map +1 -1
  40. package/docs/hooks.md +109 -48
  41. package/examples/hooks/README.md +3 -0
  42. package/examples/hooks/custom-compaction.ts +115 -0
  43. package/package.json +4 -4
package/CHANGELOG.md CHANGED
@@ -1,6 +1,19 @@
1
1
  # Changelog
2
2
 
3
- ## [Unreleased]
3
+ ## [0.27.6] - 2025-12-24
4
+
5
+ ### Added
6
+
7
+ - **Compaction hook improvements**: The `before_compact` session event now includes:
8
+ - `previousSummary`: Summary from the last compaction (if any), so hooks can preserve accumulated context
9
+ - `messagesToKeep`: Messages that will be kept after the summary (recent turns), in addition to `messagesToSummarize`
10
+ - `resolveApiKey`: Function to resolve API keys for any model (checks settings, OAuth, env vars)
11
+ - Removed `apiKey` string in favor of `resolveApiKey` for more flexibility
12
+
13
+ - **SessionManager API cleanup**:
14
+ - Renamed `loadSessionFromEntries()` to `buildSessionContext()` (builds LLM context from entries, handling compaction)
15
+ - Renamed `loadEntries()` to `getEntries()` (returns defensive copy of all session entries)
16
+ - Added `buildSessionContext()` method to SessionManager
4
17
 
5
18
  ## [0.27.5] - 2025-12-24
6
19
 
@@ -17,7 +17,7 @@ import type { Model } from "@mariozechner/pi-ai";
17
17
  import { type BashResult } from "./bash-executor.js";
18
18
  import type { LoadedCustomTool } from "./custom-tools/index.js";
19
19
  import type { HookRunner } from "./hooks/index.js";
20
- import { type SessionManager } from "./session-manager.js";
20
+ import type { SessionManager } from "./session-manager.js";
21
21
  import type { SettingsManager, SkillsSettings } from "./settings-manager.js";
22
22
  import { type FileSlashCommand } from "./slash-commands.js";
23
23
  /** Session-specific events that extend the core AgentEvent */
@@ -1 +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,EAA6B,KAAK,EAAe,MAAM,qBAAqB,CAAC;AAGzF,OAAO,EAAE,KAAK,UAAU,EAAqC,MAAM,oBAAoB,CAAC;AAExF,OAAO,KAAK,EAAE,gBAAgB,EAAoC,MAAM,yBAAyB,CAAC;AAElG,OAAO,KAAK,EAAE,UAAU,EAAoD,MAAM,kBAAkB,CAAC;AAGrG,OAAO,EAAgD,KAAK,cAAc,EAAE,MAAM,sBAAsB,CAAC;AACzG,OAAO,KAAK,EAAE,eAAe,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC;AAC7E,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,GACtG;IAAE,IAAI,EAAE,kBAAkB,CAAC;IAAC,OAAO,EAAE,MAAM,CAAC;IAAC,WAAW,EAAE,MAAM,CAAC;IAAC,OAAO,EAAE,MAAM,CAAC;IAAC,YAAY,EAAE,MAAM,CAAA;CAAE,GACzG;IAAE,IAAI,EAAE,gBAAgB,CAAC;IAAC,OAAO,EAAE,OAAO,CAAC;IAAC,OAAO,EAAE,MAAM,CAAC;IAAC,UAAU,CAAC,EAAE,MAAM,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;IAClC,0DAA0D;IAC1D,UAAU,CAAC,EAAE,UAAU,GAAG,IAAI,CAAC;IAC/B,gDAAgD;IAChD,WAAW,CAAC,EAAE,gBAAgB,EAAE,CAAC;IACjC,cAAc,CAAC,EAAE,QAAQ,CAAC,cAAc,CAAC,CAAC;IAC1C,8DAA8D;IAC9D,aAAa,CAAC,EAAE,CAAC,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,KAAK,OAAO,CAAC,MAAM,GAAG,SAAS,CAAC,CAAC;CACnE;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,GAAG,IAAI,CAAC;IAC3B,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;AAgBD,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,qBAAqB,CAAgC;IAC7D,OAAO,CAAC,aAAa,CAAK;IAC1B,OAAO,CAAC,aAAa,CAA8B;IACnD,OAAO,CAAC,aAAa,CAA6B;IAGlD,OAAO,CAAC,oBAAoB,CAAgC;IAC5D,OAAO,CAAC,oBAAoB,CAA8B;IAG1D,OAAO,CAAC,WAAW,CAA2B;IAC9C,OAAO,CAAC,UAAU,CAAK;IAGvB,OAAO,CAAC,YAAY,CAA0B;IAE9C,OAAO,CAAC,eAAe,CAAuC;IAG9D,OAAO,CAAC,cAAc,CAAqD;IAE3E,YAAY,MAAM,EAAE,kBAAkB,EAUrC;IAMD,qCAAqC;IACrC,OAAO,CAAC,KAAK;IAOb,OAAO,CAAC,qBAAqB,CAAiC;IAE9D,4EAA4E;IAC5E,OAAO,CAAC,iBAAiB,CAsDvB;IAEF,wCAAwC;IACxC,OAAO,CAAC,aAAa;IAQrB,0CAA0C;IAC1C,OAAO,CAAC,mBAAmB;IAQ3B,8EAA8E;IAC9E,OAAO,CAAC,yBAAyB;YAYnB,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,kEAAkE;IAClE,IAAI,WAAW,IAAI,MAAM,GAAG,IAAI,CAE/B;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,CAmCjE;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,IAAI,cAAc,IAAI,QAAQ,CAAC,cAAc,CAAC,GAAG,SAAS,CAEzD;IAED;;OAEG;IACG,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC,CAI3B;IAED;;;;;OAKG;IACG,KAAK,IAAI,OAAO,CAAC,OAAO,CAAC,CAyC9B;IAMD;;;;OAIG;IACG,QAAQ,CAAC,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,GAAG,OAAO,CAAC,IAAI,CAAC,CAY/C;IAED;;;;OAIG;IACG,UAAU,IAAI,OAAO,CAAC,gBAAgB,GAAG,IAAI,CAAC,CAKnD;YAEa,iBAAiB;YA6BjB,oBAAoB;IA6BlC;;OAEG;IACG,kBAAkB,IAAI,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC,CAIhD;IAMD;;;;OAIG;IACH,gBAAgB,CAAC,KAAK,EAAE,aAAa,GAAG,IAAI,CAU3C;IAED;;;OAGG;IACH,kBAAkB,IAAI,aAAa,GAAG,IAAI,CAUzC;IAED;;OAEG;IACH,0BAA0B,IAAI,aAAa,EAAE,CAE5C;IAED;;OAEG;IACH,qBAAqB,IAAI,OAAO,CAE/B;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,CA4FpE;IAED;;OAEG;IACH,eAAe,IAAI,IAAI,CAGtB;YAaa,gBAAgB;YAkChB,kBAAkB;IAsHhC;;OAEG;IACH,wBAAwB,CAAC,OAAO,EAAE,OAAO,GAAG,IAAI,CAE/C;IAED,yCAAyC;IACzC,IAAI,qBAAqB,IAAI,OAAO,CAEnC;IAMD;;;OAGG;IACH,OAAO,CAAC,iBAAiB;YAkBX,qBAAqB;IAwEnC;;OAEG;IACH,OAAO,CAAC,MAAM;IAgBd;;OAEG;IACH,UAAU,IAAI,IAAI,CAIjB;YAMa,YAAY;IAM1B,kDAAkD;IAClD,IAAI,UAAU,IAAI,OAAO,CAExB;IAED,oCAAoC;IACpC,IAAI,gBAAgB,IAAI,OAAO,CAE9B;IAED;;OAEG;IACH,mBAAmB,CAAC,OAAO,EAAE,OAAO,GAAG,IAAI,CAE1C;IAMD;;;;;OAKG;IACG,WAAW,CAAC,OAAO,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,IAAI,GAAG,OAAO,CAAC,UAAU,CAAC,CAqCzF;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;IAkBjC;;;;;OAKG;IACG,aAAa,CAAC,WAAW,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC,CAiEzD;IAED;;;;;;;;OAQG;IACG,MAAM,CAAC,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC;QAAE,YAAY,EAAE,MAAM,CAAC;QAAC,SAAS,EAAE,OAAO,CAAA;KAAE,CAAC,CA+DtF;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,CAGxC;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;IAED;;OAEG;IACH,IAAI,WAAW,IAAI,gBAAgB,EAAE,CAEpC;YAMa,qBAAqB;CAoBnC","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, supportsXhigh } from \"@mariozechner/pi-ai\";\nimport { getModelsPath } from \"../config.js\";\nimport { type BashResult, executeBash as executeBashCommand } from \"./bash-executor.js\";\nimport { calculateContextTokens, compact, prepareCompaction, shouldCompact } from \"./compaction.js\";\nimport type { LoadedCustomTool, SessionEvent as ToolSessionEvent } from \"./custom-tools/index.js\";\nimport { exportSessionToHtml } from \"./export-html.js\";\nimport type { HookRunner, SessionEventResult, TurnEndEvent, TurnStartEvent } from \"./hooks/index.js\";\nimport type { BashExecutionMessage } from \"./messages.js\";\nimport { getApiKeyForModel, getAvailableModels } from \"./model-config.js\";\nimport { type CompactionEntry, loadSessionFromEntries, type SessionManager } from \"./session-manager.js\";\nimport type { SettingsManager, SkillsSettings } 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\t| { type: \"auto_retry_start\"; attempt: number; maxAttempts: number; delayMs: number; errorMessage: string }\n\t| { type: \"auto_retry_end\"; success: boolean; attempt: number; finalError?: string };\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\t/** Custom tools for session lifecycle events */\n\tcustomTools?: LoadedCustomTool[];\n\tskillsSettings?: Required<SkillsSettings>;\n\t/** Resolve API key for a model. Default: getApiKeyForModel */\n\tresolveApiKey?: (model: Model<any>) => Promise<string | undefined>;\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 | null;\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// Constants\n// ============================================================================\n\n/** Standard thinking levels */\nconst THINKING_LEVELS: ThinkingLevel[] = [\"off\", \"minimal\", \"low\", \"medium\", \"high\"];\n\n/** Thinking levels including xhigh (for supported models) */\nconst THINKING_LEVELS_WITH_XHIGH: ThinkingLevel[] = [\"off\", \"minimal\", \"low\", \"medium\", \"high\", \"xhigh\"];\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// Retry state\n\tprivate _retryAbortController: AbortController | null = null;\n\tprivate _retryAttempt = 0;\n\tprivate _retryPromise: Promise<void> | null = null;\n\tprivate _retryResolve: (() => void) | 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\t// Custom tools for session lifecycle\n\tprivate _customTools: LoadedCustomTool[] = [];\n\n\tprivate _skillsSettings: Required<SkillsSettings> | undefined;\n\n\t// API key resolver\n\tprivate _resolveApiKey: (model: Model<any>) => Promise<string | undefined>;\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\tthis._customTools = config.customTools ?? [];\n\t\tthis._skillsSettings = config.skillsSettings;\n\t\tthis._resolveApiKey = config.resolveApiKey ?? getApiKeyForModel;\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// 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-retry and 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\n\t\t\t// Check for retryable errors first (overloaded, rate limit, server errors)\n\t\t\tif (this._isRetryableError(msg)) {\n\t\t\t\tconst didRetry = await this._handleRetryableError(msg);\n\t\t\t\tif (didRetry) return; // Retry was initiated, don't proceed to compaction\n\t\t\t} else if (this._retryAttempt > 0) {\n\t\t\t\t// Previous retry succeeded - emit success event and reset counter\n\t\t\t\tthis._emit({\n\t\t\t\t\ttype: \"auto_retry_end\",\n\t\t\t\t\tsuccess: true,\n\t\t\t\t\tattempt: this._retryAttempt,\n\t\t\t\t});\n\t\t\t\tthis._retryAttempt = 0;\n\t\t\t\t// Resolve the retry promise so waitForRetry() completes\n\t\t\t\tthis._resolveRetry();\n\t\t\t}\n\n\t\t\tawait this._checkCompaction(msg);\n\t\t}\n\t};\n\n\t/** Resolve the pending retry promise */\n\tprivate _resolveRetry(): void {\n\t\tif (this._retryResolve) {\n\t\t\tthis._retryResolve();\n\t\t\tthis._retryResolve = null;\n\t\t\tthis._retryPromise = null;\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/** Find the last assistant message in agent state (including aborted ones) */\n\tprivate _findLastAssistantMessage(): AssistantMessage | null {\n\t\tconst messages = this.agent.state.messages;\n\t\tfor (let i = messages.length - 1; i >= 0; i--) {\n\t\t\tconst msg = messages[i];\n\t\t\tif (msg.role === \"assistant\") {\n\t\t\t\treturn msg as AssistantMessage;\n\t\t\t}\n\t\t}\n\t\treturn null;\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, or null if sessions are disabled */\n\tget sessionFile(): string | null {\n\t\treturn this.sessionManager.isPersisted() ? this.sessionManager.getSessionFile() : null;\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`Use /login, set an API key environment variable, 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 this._resolveApiKey(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`Use /login, set an API key environment variable, or create ${getModelsPath()}`,\n\t\t\t);\n\t\t}\n\n\t\t// Check if we need to compact before sending (catches aborted responses)\n\t\tconst lastAssistant = this._findLastAssistantMessage();\n\t\tif (lastAssistant) {\n\t\t\tawait this._checkCompaction(lastAssistant, false);\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\tawait this.waitForRetry();\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\tget skillsSettings(): Required<SkillsSettings> | undefined {\n\t\treturn this._skillsSettings;\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.abortRetry();\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 * @returns true if reset completed, false if cancelled by hook\n\t */\n\tasync reset(): Promise<boolean> {\n\t\tconst previousSessionFile = this.sessionFile;\n\t\tconst entries = this.sessionManager.loadEntries();\n\n\t\t// Emit before_clear event (can be cancelled)\n\t\tif (this._hookRunner?.hasHandlers(\"session\")) {\n\t\t\tconst result = (await this._hookRunner.emit({\n\t\t\t\ttype: \"session\",\n\t\t\t\tentries,\n\t\t\t\tsessionFile: this.sessionFile,\n\t\t\t\tpreviousSessionFile: null,\n\t\t\t\treason: \"before_clear\",\n\t\t\t})) as SessionEventResult | undefined;\n\n\t\t\tif (result?.cancel) {\n\t\t\t\treturn false;\n\t\t\t}\n\t\t}\n\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\n\t\t// Emit session event with reason \"clear\" to hooks\n\t\tif (this._hookRunner) {\n\t\t\tthis._hookRunner.setSessionFile(this.sessionFile);\n\t\t\tawait this._hookRunner.emit({\n\t\t\t\ttype: \"session\",\n\t\t\t\tentries: [],\n\t\t\t\tsessionFile: this.sessionFile,\n\t\t\t\tpreviousSessionFile,\n\t\t\t\treason: \"clear\",\n\t\t\t});\n\t\t}\n\n\t\t// Emit session event to custom tools\n\t\tawait this._emitToolSessionEvent(\"clear\", previousSessionFile);\n\t\treturn true;\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 this._resolveApiKey(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\n\t\t// Re-clamp thinking level for new model's capabilities\n\t\tthis.setThinkingLevel(this.thinkingLevel);\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 this._resolveApiKey(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 (setThinkingLevel clamps to model capabilities)\n\t\tthis.setThinkingLevel(next.thinkingLevel);\n\n\t\treturn { model: next.model, thinkingLevel: this.thinkingLevel, 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 this._resolveApiKey(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\t// Re-clamp thinking level for new model's capabilities\n\t\tthis.setThinkingLevel(this.thinkingLevel);\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 * Clamps to model capabilities: \"off\" if no reasoning, \"high\" if xhigh unsupported.\n\t * Saves to session and settings.\n\t */\n\tsetThinkingLevel(level: ThinkingLevel): void {\n\t\tlet effectiveLevel = level;\n\t\tif (!this.supportsThinking()) {\n\t\t\teffectiveLevel = \"off\";\n\t\t} else if (level === \"xhigh\" && !this.supportsXhighThinking()) {\n\t\t\teffectiveLevel = \"high\";\n\t\t}\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 levels = this.getAvailableThinkingLevels();\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 * Get available thinking levels for current model.\n\t */\n\tgetAvailableThinkingLevels(): ThinkingLevel[] {\n\t\treturn this.supportsXhighThinking() ? THINKING_LEVELS_WITH_XHIGH : THINKING_LEVELS;\n\t}\n\n\t/**\n\t * Check if current model supports xhigh thinking level.\n\t */\n\tsupportsXhighThinking(): boolean {\n\t\treturn this.model ? supportsXhigh(this.model) : false;\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\tthis._disconnectFromAgent();\n\t\tawait this.abort();\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 this._resolveApiKey(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\n\t\t\tconst preparation = prepareCompaction(entries, settings);\n\t\t\tif (!preparation) {\n\t\t\t\tthrow new Error(\"Already compacted\");\n\t\t\t}\n\n\t\t\tlet compactionEntry: CompactionEntry | undefined;\n\t\t\tlet fromHook = false;\n\n\t\t\tif (this._hookRunner?.hasHandlers(\"session\")) {\n\t\t\t\tconst result = (await this._hookRunner.emit({\n\t\t\t\t\ttype: \"session\",\n\t\t\t\t\tentries,\n\t\t\t\t\tsessionFile: this.sessionFile,\n\t\t\t\t\tpreviousSessionFile: null,\n\t\t\t\t\treason: \"before_compact\",\n\t\t\t\t\tcutPoint: preparation.cutPoint,\n\t\t\t\t\tmessagesToSummarize: preparation.messagesToSummarize,\n\t\t\t\t\ttokensBefore: preparation.tokensBefore,\n\t\t\t\t\tcustomInstructions,\n\t\t\t\t\tmodel: this.model,\n\t\t\t\t\tapiKey,\n\t\t\t\t})) as SessionEventResult | undefined;\n\n\t\t\t\tif (result?.cancel) {\n\t\t\t\t\tthrow new Error(\"Compaction cancelled\");\n\t\t\t\t}\n\n\t\t\t\tif (result?.compactionEntry) {\n\t\t\t\t\tcompactionEntry = result.compactionEntry;\n\t\t\t\t\tfromHook = true;\n\t\t\t\t}\n\t\t\t}\n\n\t\t\tif (!compactionEntry) {\n\t\t\t\tcompactionEntry = await compact(\n\t\t\t\t\tentries,\n\t\t\t\t\tthis.model,\n\t\t\t\t\tsettings,\n\t\t\t\t\tapiKey,\n\t\t\t\t\tthis._compactionAbortController.signal,\n\t\t\t\t\tcustomInstructions,\n\t\t\t\t);\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\tthis.sessionManager.saveCompaction(compactionEntry);\n\t\t\tconst newEntries = this.sessionManager.loadEntries();\n\t\t\tconst loaded = loadSessionFromEntries(newEntries);\n\t\t\tthis.agent.replaceMessages(loaded.messages);\n\n\t\t\tif (this._hookRunner) {\n\t\t\t\tawait this._hookRunner.emit({\n\t\t\t\t\ttype: \"session\",\n\t\t\t\t\tentries: newEntries,\n\t\t\t\t\tsessionFile: this.sessionFile,\n\t\t\t\t\tpreviousSessionFile: null,\n\t\t\t\t\treason: \"compact\",\n\t\t\t\t\tcompactionEntry,\n\t\t\t\t\ttokensBefore: compactionEntry.tokensBefore,\n\t\t\t\t\tfromHook,\n\t\t\t\t});\n\t\t\t}\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 * Check if compaction is needed and run it.\n\t * Called after agent_end and before prompt submission.\n\t *\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: Context over threshold, compact, NO auto-retry (user continues manually)\n\t *\n\t * @param assistantMessage The assistant message to check\n\t * @param skipAbortedCheck If false, include aborted messages (for pre-prompt check). Default: true\n\t */\n\tprivate async _checkCompaction(assistantMessage: AssistantMessage, skipAbortedCheck = true): 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) - unless skipAbortedCheck is false\n\t\tif (skipAbortedCheck && 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 this._resolveApiKey(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\n\t\t\tconst preparation = prepareCompaction(entries, settings);\n\t\t\tif (!preparation) {\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\tlet compactionEntry: CompactionEntry | undefined;\n\t\t\tlet fromHook = false;\n\n\t\t\tif (this._hookRunner?.hasHandlers(\"session\")) {\n\t\t\t\tconst hookResult = (await this._hookRunner.emit({\n\t\t\t\t\ttype: \"session\",\n\t\t\t\t\tentries,\n\t\t\t\t\tsessionFile: this.sessionFile,\n\t\t\t\t\tpreviousSessionFile: null,\n\t\t\t\t\treason: \"before_compact\",\n\t\t\t\t\tcutPoint: preparation.cutPoint,\n\t\t\t\t\tmessagesToSummarize: preparation.messagesToSummarize,\n\t\t\t\t\ttokensBefore: preparation.tokensBefore,\n\t\t\t\t\tcustomInstructions: undefined,\n\t\t\t\t\tmodel: this.model,\n\t\t\t\t\tapiKey,\n\t\t\t\t})) as SessionEventResult | undefined;\n\n\t\t\t\tif (hookResult?.cancel) {\n\t\t\t\t\tthis._emit({ type: \"auto_compaction_end\", result: null, aborted: true, willRetry: false });\n\t\t\t\t\treturn;\n\t\t\t\t}\n\n\t\t\t\tif (hookResult?.compactionEntry) {\n\t\t\t\t\tcompactionEntry = hookResult.compactionEntry;\n\t\t\t\t\tfromHook = true;\n\t\t\t\t}\n\t\t\t}\n\n\t\t\tif (!compactionEntry) {\n\t\t\t\tcompactionEntry = await compact(\n\t\t\t\t\tentries,\n\t\t\t\t\tthis.model,\n\t\t\t\t\tsettings,\n\t\t\t\t\tapiKey,\n\t\t\t\t\tthis._autoCompactionAbortController.signal,\n\t\t\t\t);\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 newEntries = this.sessionManager.loadEntries();\n\t\t\tconst loaded = loadSessionFromEntries(newEntries);\n\t\t\tthis.agent.replaceMessages(loaded.messages);\n\n\t\t\tif (this._hookRunner) {\n\t\t\t\tawait this._hookRunner.emit({\n\t\t\t\t\ttype: \"session\",\n\t\t\t\t\tentries: newEntries,\n\t\t\t\t\tsessionFile: this.sessionFile,\n\t\t\t\t\tpreviousSessionFile: null,\n\t\t\t\t\treason: \"compact\",\n\t\t\t\t\tcompactionEntry,\n\t\t\t\t\ttokensBefore: compactionEntry.tokensBefore,\n\t\t\t\t\tfromHook,\n\t\t\t\t});\n\t\t\t}\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\tif (willRetry) {\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\tsetTimeout(() => {\n\t\t\t\t\tthis.agent.continue().catch(() => {});\n\t\t\t\t}, 100);\n\t\t\t}\n\t\t} catch (error) {\n\t\t\tthis._emit({ type: \"auto_compaction_end\", result: null, aborted: false, willRetry: false });\n\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// Auto-Retry\n\t// =========================================================================\n\n\t/**\n\t * Check if an error is retryable (overloaded, rate limit, server errors).\n\t * Context overflow errors are NOT retryable (handled by compaction instead).\n\t */\n\tprivate _isRetryableError(message: AssistantMessage): boolean {\n\t\tif (message.stopReason !== \"error\" || !message.errorMessage) return false;\n\n\t\t// Context overflow is handled by compaction, not retry\n\t\tconst contextWindow = this.model?.contextWindow ?? 0;\n\t\tif (isContextOverflow(message, contextWindow)) return false;\n\n\t\tconst err = message.errorMessage;\n\t\t// Match: overloaded_error, rate limit, 429, 500, 502, 503, 504, service unavailable, connection error\n\t\treturn /overloaded|rate.?limit|too many requests|429|500|502|503|504|service.?unavailable|server error|internal error|connection.?error/i.test(\n\t\t\terr,\n\t\t);\n\t}\n\n\t/**\n\t * Handle retryable errors with exponential backoff.\n\t * @returns true if retry was initiated, false if max retries exceeded or disabled\n\t */\n\tprivate async _handleRetryableError(message: AssistantMessage): Promise<boolean> {\n\t\tconst settings = this.settingsManager.getRetrySettings();\n\t\tif (!settings.enabled) return false;\n\n\t\tthis._retryAttempt++;\n\n\t\t// Create retry promise on first attempt so waitForRetry() can await it\n\t\tif (this._retryAttempt === 1 && !this._retryPromise) {\n\t\t\tthis._retryPromise = new Promise((resolve) => {\n\t\t\t\tthis._retryResolve = resolve;\n\t\t\t});\n\t\t}\n\n\t\tif (this._retryAttempt > settings.maxRetries) {\n\t\t\t// Max retries exceeded, emit final failure and reset\n\t\t\tthis._emit({\n\t\t\t\ttype: \"auto_retry_end\",\n\t\t\t\tsuccess: false,\n\t\t\t\tattempt: this._retryAttempt - 1,\n\t\t\t\tfinalError: message.errorMessage,\n\t\t\t});\n\t\t\tthis._retryAttempt = 0;\n\t\t\tthis._resolveRetry(); // Resolve so waitForRetry() completes\n\t\t\treturn false;\n\t\t}\n\n\t\tconst delayMs = settings.baseDelayMs * 2 ** (this._retryAttempt - 1);\n\n\t\tthis._emit({\n\t\t\ttype: \"auto_retry_start\",\n\t\t\tattempt: this._retryAttempt,\n\t\t\tmaxAttempts: settings.maxRetries,\n\t\t\tdelayMs,\n\t\t\terrorMessage: message.errorMessage || \"Unknown error\",\n\t\t});\n\n\t\t// Remove error message from agent state (keep in session for history)\n\t\tconst messages = this.agent.state.messages;\n\t\tif (messages.length > 0 && messages[messages.length - 1].role === \"assistant\") {\n\t\t\tthis.agent.replaceMessages(messages.slice(0, -1));\n\t\t}\n\n\t\t// Wait with exponential backoff (abortable)\n\t\tthis._retryAbortController = new AbortController();\n\t\ttry {\n\t\t\tawait this._sleep(delayMs, this._retryAbortController.signal);\n\t\t} catch {\n\t\t\t// Aborted during sleep - emit end event so UI can clean up\n\t\t\tconst attempt = this._retryAttempt;\n\t\t\tthis._retryAttempt = 0;\n\t\t\tthis._retryAbortController = null;\n\t\t\tthis._emit({\n\t\t\t\ttype: \"auto_retry_end\",\n\t\t\t\tsuccess: false,\n\t\t\t\tattempt,\n\t\t\t\tfinalError: \"Retry cancelled\",\n\t\t\t});\n\t\t\tthis._resolveRetry();\n\t\t\treturn false;\n\t\t}\n\t\tthis._retryAbortController = null;\n\n\t\t// Retry via continue() - use setTimeout to break out of event handler chain\n\t\tsetTimeout(() => {\n\t\t\tthis.agent.continue().catch(() => {\n\t\t\t\t// Retry failed - will be caught by next agent_end\n\t\t\t});\n\t\t}, 0);\n\n\t\treturn true;\n\t}\n\n\t/**\n\t * Sleep helper that respects abort signal.\n\t */\n\tprivate _sleep(ms: number, signal?: AbortSignal): Promise<void> {\n\t\treturn new Promise((resolve, reject) => {\n\t\t\tif (signal?.aborted) {\n\t\t\t\treject(new Error(\"Aborted\"));\n\t\t\t\treturn;\n\t\t\t}\n\n\t\t\tconst timeout = setTimeout(resolve, ms);\n\n\t\t\tsignal?.addEventListener(\"abort\", () => {\n\t\t\t\tclearTimeout(timeout);\n\t\t\t\treject(new Error(\"Aborted\"));\n\t\t\t});\n\t\t});\n\t}\n\n\t/**\n\t * Cancel in-progress retry.\n\t */\n\tabortRetry(): void {\n\t\tthis._retryAbortController?.abort();\n\t\tthis._retryAttempt = 0;\n\t\tthis._resolveRetry();\n\t}\n\n\t/**\n\t * Wait for any in-progress retry to complete.\n\t * Returns immediately if no retry is in progress.\n\t */\n\tprivate async waitForRetry(): Promise<void> {\n\t\tif (this._retryPromise) {\n\t\t\tawait this._retryPromise;\n\t\t}\n\t}\n\n\t/** Whether auto-retry is currently in progress */\n\tget isRetrying(): boolean {\n\t\treturn this._retryPromise !== null;\n\t}\n\n\t/** Whether auto-retry is enabled */\n\tget autoRetryEnabled(): boolean {\n\t\treturn this.settingsManager.getRetryEnabled();\n\t}\n\n\t/**\n\t * Toggle auto-retry setting.\n\t */\n\tsetAutoRetryEnabled(enabled: boolean): void {\n\t\tthis.settingsManager.setRetryEnabled(enabled);\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\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\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 * @returns true if switch completed, false if cancelled by hook\n\t */\n\tasync switchSession(sessionPath: string): Promise<boolean> {\n\t\tconst previousSessionFile = this.sessionFile;\n\t\tconst oldEntries = this.sessionManager.loadEntries();\n\n\t\t// Emit before_switch event (can be cancelled)\n\t\tif (this._hookRunner?.hasHandlers(\"session\")) {\n\t\t\tconst result = (await this._hookRunner.emit({\n\t\t\t\ttype: \"session\",\n\t\t\t\tentries: oldEntries,\n\t\t\t\tsessionFile: this.sessionFile,\n\t\t\t\tpreviousSessionFile: null,\n\t\t\t\treason: \"before_switch\",\n\t\t\t})) as SessionEventResult | undefined;\n\n\t\t\tif (result?.cancel) {\n\t\t\t\treturn false;\n\t\t\t}\n\t\t}\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// Reload messages\n\t\tconst entries = this.sessionManager.loadEntries();\n\t\tconst loaded = loadSessionFromEntries(entries);\n\n\t\t// Emit session event to hooks\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\",\n\t\t\t\tentries,\n\t\t\t\tsessionFile: sessionPath,\n\t\t\t\tpreviousSessionFile,\n\t\t\t\treason: \"switch\",\n\t\t\t});\n\t\t}\n\n\t\t// Emit session event to custom tools\n\t\tawait this._emitToolSessionEvent(\"switch\", previousSessionFile);\n\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 (setThinkingLevel clamps to model capabilities)\n\t\tconst savedThinking = this.sessionManager.loadThinkingLevel();\n\t\tif (savedThinking) {\n\t\t\tthis.setThinkingLevel(savedThinking as ThinkingLevel);\n\t\t}\n\n\t\tthis._reconnectToAgent();\n\t\treturn true;\n\t}\n\n\t/**\n\t * Create a branch from a specific entry index.\n\t * Emits before_branch/branch session events to hooks.\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 * - cancelled: True if a hook cancelled the branch\n\t */\n\tasync branch(entryIndex: number): Promise<{ selectedText: string; cancelled: 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\tlet skipConversationRestore = false;\n\n\t\t// Emit before_branch event (can be cancelled)\n\t\tif (this._hookRunner?.hasHandlers(\"session\")) {\n\t\t\tconst result = (await this._hookRunner.emit({\n\t\t\t\ttype: \"session\",\n\t\t\t\tentries,\n\t\t\t\tsessionFile: this.sessionFile,\n\t\t\t\tpreviousSessionFile: null,\n\t\t\t\treason: \"before_branch\",\n\t\t\t\ttargetTurnIndex: entryIndex,\n\t\t\t})) as SessionEventResult | undefined;\n\n\t\t\tif (result?.cancel) {\n\t\t\t\treturn { selectedText, cancelled: true };\n\t\t\t}\n\t\t\tskipConversationRestore = result?.skipConversationRestore ?? false;\n\t\t}\n\n\t\t// Create branched session (returns null in --no-session mode)\n\t\tconst newSessionFile = this.sessionManager.createBranchedSessionFromEntries(entries, entryIndex);\n\n\t\t// Update session file if we have one (file-based mode)\n\t\tif (newSessionFile !== null) {\n\t\t\tthis.sessionManager.setSessionFile(newSessionFile);\n\t\t}\n\n\t\t// Reload messages from entries (works for both file and in-memory mode)\n\t\tconst newEntries = this.sessionManager.loadEntries();\n\t\tconst loaded = loadSessionFromEntries(newEntries);\n\n\t\t// Emit branch event to hooks (after branch completes)\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\",\n\t\t\t\tentries: newEntries,\n\t\t\t\tsessionFile: newSessionFile,\n\t\t\t\tpreviousSessionFile,\n\t\t\t\treason: \"branch\",\n\t\t\t\ttargetTurnIndex: entryIndex,\n\t\t\t});\n\t\t}\n\n\t\t// Emit session event to custom tools (with reason \"branch\")\n\t\tawait this._emitToolSessionEvent(\"branch\", previousSessionFile);\n\n\t\tif (!skipConversationRestore) {\n\t\t\tthis.agent.replaceMessages(loaded.messages);\n\t\t}\n\n\t\treturn { selectedText, cancelled: 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\tconst themeName = this.settingsManager.getTheme();\n\t\treturn exportSessionToHtml(this.sessionManager, this.state, { outputPath, themeName });\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\t/**\n\t * Get custom tools (for setting UI context in modes).\n\t */\n\tget customTools(): LoadedCustomTool[] {\n\t\treturn this._customTools;\n\t}\n\n\t/**\n\t * Emit session event to all custom tools.\n\t * Called on session switch, branch, and clear.\n\t */\n\tprivate async _emitToolSessionEvent(\n\t\treason: ToolSessionEvent[\"reason\"],\n\t\tpreviousSessionFile: string | null,\n\t): Promise<void> {\n\t\tconst event: ToolSessionEvent = {\n\t\t\tentries: this.sessionManager.loadEntries(),\n\t\t\tsessionFile: this.sessionFile,\n\t\t\tpreviousSessionFile,\n\t\t\treason,\n\t\t};\n\t\tfor (const { tool } of this._customTools) {\n\t\t\tif (tool.onSession) {\n\t\t\t\ttry {\n\t\t\t\t\tawait tool.onSession(event);\n\t\t\t\t} catch (_err) {\n\t\t\t\t\t// Silently ignore tool errors during session events\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\t}\n}\n"]}
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,EAA6B,KAAK,EAAe,MAAM,qBAAqB,CAAC;AAGzF,OAAO,EAAE,KAAK,UAAU,EAAqC,MAAM,oBAAoB,CAAC;AAExF,OAAO,KAAK,EAAE,gBAAgB,EAAoC,MAAM,yBAAyB,CAAC;AAElG,OAAO,KAAK,EAAE,UAAU,EAAoD,MAAM,kBAAkB,CAAC;AAGrG,OAAO,KAAK,EAAmB,cAAc,EAAE,MAAM,sBAAsB,CAAC;AAC5E,OAAO,KAAK,EAAE,eAAe,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC;AAC7E,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,GACtG;IAAE,IAAI,EAAE,kBAAkB,CAAC;IAAC,OAAO,EAAE,MAAM,CAAC;IAAC,WAAW,EAAE,MAAM,CAAC;IAAC,OAAO,EAAE,MAAM,CAAC;IAAC,YAAY,EAAE,MAAM,CAAA;CAAE,GACzG;IAAE,IAAI,EAAE,gBAAgB,CAAC;IAAC,OAAO,EAAE,OAAO,CAAC;IAAC,OAAO,EAAE,MAAM,CAAC;IAAC,UAAU,CAAC,EAAE,MAAM,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;IAClC,0DAA0D;IAC1D,UAAU,CAAC,EAAE,UAAU,GAAG,IAAI,CAAC;IAC/B,gDAAgD;IAChD,WAAW,CAAC,EAAE,gBAAgB,EAAE,CAAC;IACjC,cAAc,CAAC,EAAE,QAAQ,CAAC,cAAc,CAAC,CAAC;IAC1C,8DAA8D;IAC9D,aAAa,CAAC,EAAE,CAAC,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,KAAK,OAAO,CAAC,MAAM,GAAG,SAAS,CAAC,CAAC;CACnE;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,GAAG,IAAI,CAAC;IAC3B,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;AAgBD,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,qBAAqB,CAAgC;IAC7D,OAAO,CAAC,aAAa,CAAK;IAC1B,OAAO,CAAC,aAAa,CAA8B;IACnD,OAAO,CAAC,aAAa,CAA6B;IAGlD,OAAO,CAAC,oBAAoB,CAAgC;IAC5D,OAAO,CAAC,oBAAoB,CAA8B;IAG1D,OAAO,CAAC,WAAW,CAA2B;IAC9C,OAAO,CAAC,UAAU,CAAK;IAGvB,OAAO,CAAC,YAAY,CAA0B;IAE9C,OAAO,CAAC,eAAe,CAAuC;IAG9D,OAAO,CAAC,cAAc,CAAqD;IAE3E,YAAY,MAAM,EAAE,kBAAkB,EAUrC;IAMD,qCAAqC;IACrC,OAAO,CAAC,KAAK;IAOb,OAAO,CAAC,qBAAqB,CAAiC;IAE9D,4EAA4E;IAC5E,OAAO,CAAC,iBAAiB,CAsDvB;IAEF,wCAAwC;IACxC,OAAO,CAAC,aAAa;IAQrB,0CAA0C;IAC1C,OAAO,CAAC,mBAAmB;IAQ3B,8EAA8E;IAC9E,OAAO,CAAC,yBAAyB;YAYnB,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,kEAAkE;IAClE,IAAI,WAAW,IAAI,MAAM,GAAG,IAAI,CAE/B;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,CAmCjE;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,IAAI,cAAc,IAAI,QAAQ,CAAC,cAAc,CAAC,GAAG,SAAS,CAEzD;IAED;;OAEG;IACG,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC,CAI3B;IAED;;;;;OAKG;IACG,KAAK,IAAI,OAAO,CAAC,OAAO,CAAC,CAyC9B;IAMD;;;;OAIG;IACG,QAAQ,CAAC,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,GAAG,OAAO,CAAC,IAAI,CAAC,CAY/C;IAED;;;;OAIG;IACG,UAAU,IAAI,OAAO,CAAC,gBAAgB,GAAG,IAAI,CAAC,CAKnD;YAEa,iBAAiB;YA6BjB,oBAAoB;IA6BlC;;OAEG;IACG,kBAAkB,IAAI,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC,CAIhD;IAMD;;;;OAIG;IACH,gBAAgB,CAAC,KAAK,EAAE,aAAa,GAAG,IAAI,CAU3C;IAED;;;OAGG;IACH,kBAAkB,IAAI,aAAa,GAAG,IAAI,CAUzC;IAED;;OAEG;IACH,0BAA0B,IAAI,aAAa,EAAE,CAE5C;IAED;;OAEG;IACH,qBAAqB,IAAI,OAAO,CAE/B;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,CAwGpE;IAED;;OAEG;IACH,eAAe,IAAI,IAAI,CAGtB;YAaa,gBAAgB;YAkChB,kBAAkB;IAkIhC;;OAEG;IACH,wBAAwB,CAAC,OAAO,EAAE,OAAO,GAAG,IAAI,CAE/C;IAED,yCAAyC;IACzC,IAAI,qBAAqB,IAAI,OAAO,CAEnC;IAMD;;;OAGG;IACH,OAAO,CAAC,iBAAiB;YAkBX,qBAAqB;IAwEnC;;OAEG;IACH,OAAO,CAAC,MAAM;IAgBd;;OAEG;IACH,UAAU,IAAI,IAAI,CAIjB;YAMa,YAAY;IAM1B,kDAAkD;IAClD,IAAI,UAAU,IAAI,OAAO,CAExB;IAED,oCAAoC;IACpC,IAAI,gBAAgB,IAAI,OAAO,CAE9B;IAED;;OAEG;IACH,mBAAmB,CAAC,OAAO,EAAE,OAAO,GAAG,IAAI,CAE1C;IAMD;;;;;OAKG;IACG,WAAW,CAAC,OAAO,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,IAAI,GAAG,OAAO,CAAC,UAAU,CAAC,CAqCzF;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;IAkBjC;;;;;OAKG;IACG,aAAa,CAAC,WAAW,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC,CAiEzD;IAED;;;;;;;;OAQG;IACG,MAAM,CAAC,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC;QAAE,YAAY,EAAE,MAAM,CAAC;QAAC,SAAS,EAAE,OAAO,CAAA;KAAE,CAAC,CA+DtF;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,CAGxC;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;IAED;;OAEG;IACH,IAAI,WAAW,IAAI,gBAAgB,EAAE,CAEpC;YAMa,qBAAqB;CAoBnC","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, supportsXhigh } from \"@mariozechner/pi-ai\";\nimport { getModelsPath } from \"../config.js\";\nimport { type BashResult, executeBash as executeBashCommand } from \"./bash-executor.js\";\nimport { calculateContextTokens, compact, prepareCompaction, shouldCompact } from \"./compaction.js\";\nimport type { LoadedCustomTool, SessionEvent as ToolSessionEvent } from \"./custom-tools/index.js\";\nimport { exportSessionToHtml } from \"./export-html.js\";\nimport type { HookRunner, SessionEventResult, TurnEndEvent, TurnStartEvent } from \"./hooks/index.js\";\nimport type { BashExecutionMessage } from \"./messages.js\";\nimport { getApiKeyForModel, getAvailableModels } from \"./model-config.js\";\nimport type { CompactionEntry, SessionManager } from \"./session-manager.js\";\nimport type { SettingsManager, SkillsSettings } 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\t| { type: \"auto_retry_start\"; attempt: number; maxAttempts: number; delayMs: number; errorMessage: string }\n\t| { type: \"auto_retry_end\"; success: boolean; attempt: number; finalError?: string };\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\t/** Custom tools for session lifecycle events */\n\tcustomTools?: LoadedCustomTool[];\n\tskillsSettings?: Required<SkillsSettings>;\n\t/** Resolve API key for a model. Default: getApiKeyForModel */\n\tresolveApiKey?: (model: Model<any>) => Promise<string | undefined>;\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 | null;\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// Constants\n// ============================================================================\n\n/** Standard thinking levels */\nconst THINKING_LEVELS: ThinkingLevel[] = [\"off\", \"minimal\", \"low\", \"medium\", \"high\"];\n\n/** Thinking levels including xhigh (for supported models) */\nconst THINKING_LEVELS_WITH_XHIGH: ThinkingLevel[] = [\"off\", \"minimal\", \"low\", \"medium\", \"high\", \"xhigh\"];\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// Retry state\n\tprivate _retryAbortController: AbortController | null = null;\n\tprivate _retryAttempt = 0;\n\tprivate _retryPromise: Promise<void> | null = null;\n\tprivate _retryResolve: (() => void) | 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\t// Custom tools for session lifecycle\n\tprivate _customTools: LoadedCustomTool[] = [];\n\n\tprivate _skillsSettings: Required<SkillsSettings> | undefined;\n\n\t// API key resolver\n\tprivate _resolveApiKey: (model: Model<any>) => Promise<string | undefined>;\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\tthis._customTools = config.customTools ?? [];\n\t\tthis._skillsSettings = config.skillsSettings;\n\t\tthis._resolveApiKey = config.resolveApiKey ?? getApiKeyForModel;\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// 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-retry and 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\n\t\t\t// Check for retryable errors first (overloaded, rate limit, server errors)\n\t\t\tif (this._isRetryableError(msg)) {\n\t\t\t\tconst didRetry = await this._handleRetryableError(msg);\n\t\t\t\tif (didRetry) return; // Retry was initiated, don't proceed to compaction\n\t\t\t} else if (this._retryAttempt > 0) {\n\t\t\t\t// Previous retry succeeded - emit success event and reset counter\n\t\t\t\tthis._emit({\n\t\t\t\t\ttype: \"auto_retry_end\",\n\t\t\t\t\tsuccess: true,\n\t\t\t\t\tattempt: this._retryAttempt,\n\t\t\t\t});\n\t\t\t\tthis._retryAttempt = 0;\n\t\t\t\t// Resolve the retry promise so waitForRetry() completes\n\t\t\t\tthis._resolveRetry();\n\t\t\t}\n\n\t\t\tawait this._checkCompaction(msg);\n\t\t}\n\t};\n\n\t/** Resolve the pending retry promise */\n\tprivate _resolveRetry(): void {\n\t\tif (this._retryResolve) {\n\t\t\tthis._retryResolve();\n\t\t\tthis._retryResolve = null;\n\t\t\tthis._retryPromise = null;\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/** Find the last assistant message in agent state (including aborted ones) */\n\tprivate _findLastAssistantMessage(): AssistantMessage | null {\n\t\tconst messages = this.agent.state.messages;\n\t\tfor (let i = messages.length - 1; i >= 0; i--) {\n\t\t\tconst msg = messages[i];\n\t\t\tif (msg.role === \"assistant\") {\n\t\t\t\treturn msg as AssistantMessage;\n\t\t\t}\n\t\t}\n\t\treturn null;\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, or null if sessions are disabled */\n\tget sessionFile(): string | null {\n\t\treturn this.sessionManager.isPersisted() ? this.sessionManager.getSessionFile() : null;\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`Use /login, set an API key environment variable, 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 this._resolveApiKey(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`Use /login, set an API key environment variable, or create ${getModelsPath()}`,\n\t\t\t);\n\t\t}\n\n\t\t// Check if we need to compact before sending (catches aborted responses)\n\t\tconst lastAssistant = this._findLastAssistantMessage();\n\t\tif (lastAssistant) {\n\t\t\tawait this._checkCompaction(lastAssistant, false);\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\tawait this.waitForRetry();\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\tget skillsSettings(): Required<SkillsSettings> | undefined {\n\t\treturn this._skillsSettings;\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.abortRetry();\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 * @returns true if reset completed, false if cancelled by hook\n\t */\n\tasync reset(): Promise<boolean> {\n\t\tconst previousSessionFile = this.sessionFile;\n\t\tconst entries = this.sessionManager.getEntries();\n\n\t\t// Emit before_clear event (can be cancelled)\n\t\tif (this._hookRunner?.hasHandlers(\"session\")) {\n\t\t\tconst result = (await this._hookRunner.emit({\n\t\t\t\ttype: \"session\",\n\t\t\t\tentries,\n\t\t\t\tsessionFile: this.sessionFile,\n\t\t\t\tpreviousSessionFile: null,\n\t\t\t\treason: \"before_clear\",\n\t\t\t})) as SessionEventResult | undefined;\n\n\t\t\tif (result?.cancel) {\n\t\t\t\treturn false;\n\t\t\t}\n\t\t}\n\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\n\t\t// Emit session event with reason \"clear\" to hooks\n\t\tif (this._hookRunner) {\n\t\t\tthis._hookRunner.setSessionFile(this.sessionFile);\n\t\t\tawait this._hookRunner.emit({\n\t\t\t\ttype: \"session\",\n\t\t\t\tentries: [],\n\t\t\t\tsessionFile: this.sessionFile,\n\t\t\t\tpreviousSessionFile,\n\t\t\t\treason: \"clear\",\n\t\t\t});\n\t\t}\n\n\t\t// Emit session event to custom tools\n\t\tawait this._emitToolSessionEvent(\"clear\", previousSessionFile);\n\t\treturn true;\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 this._resolveApiKey(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\n\t\t// Re-clamp thinking level for new model's capabilities\n\t\tthis.setThinkingLevel(this.thinkingLevel);\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 this._resolveApiKey(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 (setThinkingLevel clamps to model capabilities)\n\t\tthis.setThinkingLevel(next.thinkingLevel);\n\n\t\treturn { model: next.model, thinkingLevel: this.thinkingLevel, 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 this._resolveApiKey(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\t// Re-clamp thinking level for new model's capabilities\n\t\tthis.setThinkingLevel(this.thinkingLevel);\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 * Clamps to model capabilities: \"off\" if no reasoning, \"high\" if xhigh unsupported.\n\t * Saves to session and settings.\n\t */\n\tsetThinkingLevel(level: ThinkingLevel): void {\n\t\tlet effectiveLevel = level;\n\t\tif (!this.supportsThinking()) {\n\t\t\teffectiveLevel = \"off\";\n\t\t} else if (level === \"xhigh\" && !this.supportsXhighThinking()) {\n\t\t\teffectiveLevel = \"high\";\n\t\t}\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 levels = this.getAvailableThinkingLevels();\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 * Get available thinking levels for current model.\n\t */\n\tgetAvailableThinkingLevels(): ThinkingLevel[] {\n\t\treturn this.supportsXhighThinking() ? THINKING_LEVELS_WITH_XHIGH : THINKING_LEVELS;\n\t}\n\n\t/**\n\t * Check if current model supports xhigh thinking level.\n\t */\n\tsupportsXhighThinking(): boolean {\n\t\treturn this.model ? supportsXhigh(this.model) : false;\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\tthis._disconnectFromAgent();\n\t\tawait this.abort();\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 this._resolveApiKey(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.getEntries();\n\t\t\tconst settings = this.settingsManager.getCompactionSettings();\n\n\t\t\tconst preparation = prepareCompaction(entries, settings);\n\t\t\tif (!preparation) {\n\t\t\t\tthrow new Error(\"Already compacted\");\n\t\t\t}\n\n\t\t\t// Find previous compaction summary if any\n\t\t\tlet previousSummary: string | undefined;\n\t\t\tfor (let i = entries.length - 1; i >= 0; i--) {\n\t\t\t\tif (entries[i].type === \"compaction\") {\n\t\t\t\t\tpreviousSummary = (entries[i] as CompactionEntry).summary;\n\t\t\t\t\tbreak;\n\t\t\t\t}\n\t\t\t}\n\n\t\t\tlet compactionEntry: CompactionEntry | undefined;\n\t\t\tlet fromHook = false;\n\n\t\t\tif (this._hookRunner?.hasHandlers(\"session\")) {\n\t\t\t\tconst result = (await this._hookRunner.emit({\n\t\t\t\t\ttype: \"session\",\n\t\t\t\t\tentries,\n\t\t\t\t\tsessionFile: this.sessionFile,\n\t\t\t\t\tpreviousSessionFile: null,\n\t\t\t\t\treason: \"before_compact\",\n\t\t\t\t\tcutPoint: preparation.cutPoint,\n\t\t\t\t\tpreviousSummary,\n\t\t\t\t\tmessagesToSummarize: [...preparation.messagesToSummarize],\n\t\t\t\t\tmessagesToKeep: [...preparation.messagesToKeep],\n\t\t\t\t\ttokensBefore: preparation.tokensBefore,\n\t\t\t\t\tcustomInstructions,\n\t\t\t\t\tmodel: this.model,\n\t\t\t\t\tresolveApiKey: this._resolveApiKey,\n\t\t\t\t\tsignal: this._compactionAbortController.signal,\n\t\t\t\t})) as SessionEventResult | undefined;\n\n\t\t\t\tif (result?.cancel) {\n\t\t\t\t\tthrow new Error(\"Compaction cancelled\");\n\t\t\t\t}\n\n\t\t\t\tif (result?.compactionEntry) {\n\t\t\t\t\tcompactionEntry = result.compactionEntry;\n\t\t\t\t\tfromHook = true;\n\t\t\t\t}\n\t\t\t}\n\n\t\t\tif (!compactionEntry) {\n\t\t\t\tcompactionEntry = await compact(\n\t\t\t\t\tentries,\n\t\t\t\t\tthis.model,\n\t\t\t\t\tsettings,\n\t\t\t\t\tapiKey,\n\t\t\t\t\tthis._compactionAbortController.signal,\n\t\t\t\t\tcustomInstructions,\n\t\t\t\t);\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\tthis.sessionManager.saveCompaction(compactionEntry);\n\t\t\tconst newEntries = this.sessionManager.getEntries();\n\t\t\tconst sessionContext = this.sessionManager.buildSessionContext();\n\t\t\tthis.agent.replaceMessages(sessionContext.messages);\n\n\t\t\tif (this._hookRunner) {\n\t\t\t\tawait this._hookRunner.emit({\n\t\t\t\t\ttype: \"session\",\n\t\t\t\t\tentries: newEntries,\n\t\t\t\t\tsessionFile: this.sessionFile,\n\t\t\t\t\tpreviousSessionFile: null,\n\t\t\t\t\treason: \"compact\",\n\t\t\t\t\tcompactionEntry,\n\t\t\t\t\ttokensBefore: compactionEntry.tokensBefore,\n\t\t\t\t\tfromHook,\n\t\t\t\t});\n\t\t\t}\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 * Check if compaction is needed and run it.\n\t * Called after agent_end and before prompt submission.\n\t *\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: Context over threshold, compact, NO auto-retry (user continues manually)\n\t *\n\t * @param assistantMessage The assistant message to check\n\t * @param skipAbortedCheck If false, include aborted messages (for pre-prompt check). Default: true\n\t */\n\tprivate async _checkCompaction(assistantMessage: AssistantMessage, skipAbortedCheck = true): 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) - unless skipAbortedCheck is false\n\t\tif (skipAbortedCheck && 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 this._resolveApiKey(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.getEntries();\n\n\t\t\tconst preparation = prepareCompaction(entries, settings);\n\t\t\tif (!preparation) {\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\t// Find previous compaction summary if any\n\t\t\tlet previousSummary: string | undefined;\n\t\t\tfor (let i = entries.length - 1; i >= 0; i--) {\n\t\t\t\tif (entries[i].type === \"compaction\") {\n\t\t\t\t\tpreviousSummary = (entries[i] as CompactionEntry).summary;\n\t\t\t\t\tbreak;\n\t\t\t\t}\n\t\t\t}\n\n\t\t\tlet compactionEntry: CompactionEntry | undefined;\n\t\t\tlet fromHook = false;\n\n\t\t\tif (this._hookRunner?.hasHandlers(\"session\")) {\n\t\t\t\tconst hookResult = (await this._hookRunner.emit({\n\t\t\t\t\ttype: \"session\",\n\t\t\t\t\tentries,\n\t\t\t\t\tsessionFile: this.sessionFile,\n\t\t\t\t\tpreviousSessionFile: null,\n\t\t\t\t\treason: \"before_compact\",\n\t\t\t\t\tcutPoint: preparation.cutPoint,\n\t\t\t\t\tpreviousSummary,\n\t\t\t\t\tmessagesToSummarize: [...preparation.messagesToSummarize],\n\t\t\t\t\tmessagesToKeep: [...preparation.messagesToKeep],\n\t\t\t\t\ttokensBefore: preparation.tokensBefore,\n\t\t\t\t\tcustomInstructions: undefined,\n\t\t\t\t\tmodel: this.model,\n\t\t\t\t\tresolveApiKey: this._resolveApiKey,\n\t\t\t\t\tsignal: this._autoCompactionAbortController.signal,\n\t\t\t\t})) as SessionEventResult | undefined;\n\n\t\t\t\tif (hookResult?.cancel) {\n\t\t\t\t\tthis._emit({ type: \"auto_compaction_end\", result: null, aborted: true, willRetry: false });\n\t\t\t\t\treturn;\n\t\t\t\t}\n\n\t\t\t\tif (hookResult?.compactionEntry) {\n\t\t\t\t\tcompactionEntry = hookResult.compactionEntry;\n\t\t\t\t\tfromHook = true;\n\t\t\t\t}\n\t\t\t}\n\n\t\t\tif (!compactionEntry) {\n\t\t\t\tcompactionEntry = await compact(\n\t\t\t\t\tentries,\n\t\t\t\t\tthis.model,\n\t\t\t\t\tsettings,\n\t\t\t\t\tapiKey,\n\t\t\t\t\tthis._autoCompactionAbortController.signal,\n\t\t\t\t);\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 newEntries = this.sessionManager.getEntries();\n\t\t\tconst sessionContext = this.sessionManager.buildSessionContext();\n\t\t\tthis.agent.replaceMessages(sessionContext.messages);\n\n\t\t\tif (this._hookRunner) {\n\t\t\t\tawait this._hookRunner.emit({\n\t\t\t\t\ttype: \"session\",\n\t\t\t\t\tentries: newEntries,\n\t\t\t\t\tsessionFile: this.sessionFile,\n\t\t\t\t\tpreviousSessionFile: null,\n\t\t\t\t\treason: \"compact\",\n\t\t\t\t\tcompactionEntry,\n\t\t\t\t\ttokensBefore: compactionEntry.tokensBefore,\n\t\t\t\t\tfromHook,\n\t\t\t\t});\n\t\t\t}\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\tif (willRetry) {\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\tsetTimeout(() => {\n\t\t\t\t\tthis.agent.continue().catch(() => {});\n\t\t\t\t}, 100);\n\t\t\t}\n\t\t} catch (error) {\n\t\t\tthis._emit({ type: \"auto_compaction_end\", result: null, aborted: false, willRetry: false });\n\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// Auto-Retry\n\t// =========================================================================\n\n\t/**\n\t * Check if an error is retryable (overloaded, rate limit, server errors).\n\t * Context overflow errors are NOT retryable (handled by compaction instead).\n\t */\n\tprivate _isRetryableError(message: AssistantMessage): boolean {\n\t\tif (message.stopReason !== \"error\" || !message.errorMessage) return false;\n\n\t\t// Context overflow is handled by compaction, not retry\n\t\tconst contextWindow = this.model?.contextWindow ?? 0;\n\t\tif (isContextOverflow(message, contextWindow)) return false;\n\n\t\tconst err = message.errorMessage;\n\t\t// Match: overloaded_error, rate limit, 429, 500, 502, 503, 504, service unavailable, connection error\n\t\treturn /overloaded|rate.?limit|too many requests|429|500|502|503|504|service.?unavailable|server error|internal error|connection.?error/i.test(\n\t\t\terr,\n\t\t);\n\t}\n\n\t/**\n\t * Handle retryable errors with exponential backoff.\n\t * @returns true if retry was initiated, false if max retries exceeded or disabled\n\t */\n\tprivate async _handleRetryableError(message: AssistantMessage): Promise<boolean> {\n\t\tconst settings = this.settingsManager.getRetrySettings();\n\t\tif (!settings.enabled) return false;\n\n\t\tthis._retryAttempt++;\n\n\t\t// Create retry promise on first attempt so waitForRetry() can await it\n\t\tif (this._retryAttempt === 1 && !this._retryPromise) {\n\t\t\tthis._retryPromise = new Promise((resolve) => {\n\t\t\t\tthis._retryResolve = resolve;\n\t\t\t});\n\t\t}\n\n\t\tif (this._retryAttempt > settings.maxRetries) {\n\t\t\t// Max retries exceeded, emit final failure and reset\n\t\t\tthis._emit({\n\t\t\t\ttype: \"auto_retry_end\",\n\t\t\t\tsuccess: false,\n\t\t\t\tattempt: this._retryAttempt - 1,\n\t\t\t\tfinalError: message.errorMessage,\n\t\t\t});\n\t\t\tthis._retryAttempt = 0;\n\t\t\tthis._resolveRetry(); // Resolve so waitForRetry() completes\n\t\t\treturn false;\n\t\t}\n\n\t\tconst delayMs = settings.baseDelayMs * 2 ** (this._retryAttempt - 1);\n\n\t\tthis._emit({\n\t\t\ttype: \"auto_retry_start\",\n\t\t\tattempt: this._retryAttempt,\n\t\t\tmaxAttempts: settings.maxRetries,\n\t\t\tdelayMs,\n\t\t\terrorMessage: message.errorMessage || \"Unknown error\",\n\t\t});\n\n\t\t// Remove error message from agent state (keep in session for history)\n\t\tconst messages = this.agent.state.messages;\n\t\tif (messages.length > 0 && messages[messages.length - 1].role === \"assistant\") {\n\t\t\tthis.agent.replaceMessages(messages.slice(0, -1));\n\t\t}\n\n\t\t// Wait with exponential backoff (abortable)\n\t\tthis._retryAbortController = new AbortController();\n\t\ttry {\n\t\t\tawait this._sleep(delayMs, this._retryAbortController.signal);\n\t\t} catch {\n\t\t\t// Aborted during sleep - emit end event so UI can clean up\n\t\t\tconst attempt = this._retryAttempt;\n\t\t\tthis._retryAttempt = 0;\n\t\t\tthis._retryAbortController = null;\n\t\t\tthis._emit({\n\t\t\t\ttype: \"auto_retry_end\",\n\t\t\t\tsuccess: false,\n\t\t\t\tattempt,\n\t\t\t\tfinalError: \"Retry cancelled\",\n\t\t\t});\n\t\t\tthis._resolveRetry();\n\t\t\treturn false;\n\t\t}\n\t\tthis._retryAbortController = null;\n\n\t\t// Retry via continue() - use setTimeout to break out of event handler chain\n\t\tsetTimeout(() => {\n\t\t\tthis.agent.continue().catch(() => {\n\t\t\t\t// Retry failed - will be caught by next agent_end\n\t\t\t});\n\t\t}, 0);\n\n\t\treturn true;\n\t}\n\n\t/**\n\t * Sleep helper that respects abort signal.\n\t */\n\tprivate _sleep(ms: number, signal?: AbortSignal): Promise<void> {\n\t\treturn new Promise((resolve, reject) => {\n\t\t\tif (signal?.aborted) {\n\t\t\t\treject(new Error(\"Aborted\"));\n\t\t\t\treturn;\n\t\t\t}\n\n\t\t\tconst timeout = setTimeout(resolve, ms);\n\n\t\t\tsignal?.addEventListener(\"abort\", () => {\n\t\t\t\tclearTimeout(timeout);\n\t\t\t\treject(new Error(\"Aborted\"));\n\t\t\t});\n\t\t});\n\t}\n\n\t/**\n\t * Cancel in-progress retry.\n\t */\n\tabortRetry(): void {\n\t\tthis._retryAbortController?.abort();\n\t\tthis._retryAttempt = 0;\n\t\tthis._resolveRetry();\n\t}\n\n\t/**\n\t * Wait for any in-progress retry to complete.\n\t * Returns immediately if no retry is in progress.\n\t */\n\tprivate async waitForRetry(): Promise<void> {\n\t\tif (this._retryPromise) {\n\t\t\tawait this._retryPromise;\n\t\t}\n\t}\n\n\t/** Whether auto-retry is currently in progress */\n\tget isRetrying(): boolean {\n\t\treturn this._retryPromise !== null;\n\t}\n\n\t/** Whether auto-retry is enabled */\n\tget autoRetryEnabled(): boolean {\n\t\treturn this.settingsManager.getRetryEnabled();\n\t}\n\n\t/**\n\t * Toggle auto-retry setting.\n\t */\n\tsetAutoRetryEnabled(enabled: boolean): void {\n\t\tthis.settingsManager.setRetryEnabled(enabled);\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\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\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 * @returns true if switch completed, false if cancelled by hook\n\t */\n\tasync switchSession(sessionPath: string): Promise<boolean> {\n\t\tconst previousSessionFile = this.sessionFile;\n\t\tconst oldEntries = this.sessionManager.getEntries();\n\n\t\t// Emit before_switch event (can be cancelled)\n\t\tif (this._hookRunner?.hasHandlers(\"session\")) {\n\t\t\tconst result = (await this._hookRunner.emit({\n\t\t\t\ttype: \"session\",\n\t\t\t\tentries: oldEntries,\n\t\t\t\tsessionFile: this.sessionFile,\n\t\t\t\tpreviousSessionFile: null,\n\t\t\t\treason: \"before_switch\",\n\t\t\t})) as SessionEventResult | undefined;\n\n\t\t\tif (result?.cancel) {\n\t\t\t\treturn false;\n\t\t\t}\n\t\t}\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// Reload messages\n\t\tconst entries = this.sessionManager.getEntries();\n\t\tconst sessionContext = this.sessionManager.buildSessionContext();\n\n\t\t// Emit session event to hooks\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\",\n\t\t\t\tentries,\n\t\t\t\tsessionFile: sessionPath,\n\t\t\t\tpreviousSessionFile,\n\t\t\t\treason: \"switch\",\n\t\t\t});\n\t\t}\n\n\t\t// Emit session event to custom tools\n\t\tawait this._emitToolSessionEvent(\"switch\", previousSessionFile);\n\n\t\tthis.agent.replaceMessages(sessionContext.messages);\n\n\t\t// Restore model if saved\n\t\tif (sessionContext.model) {\n\t\t\tconst availableModels = (await getAvailableModels()).models;\n\t\t\tconst match = availableModels.find(\n\t\t\t\t(m) => m.provider === sessionContext.model!.provider && m.id === sessionContext.model!.modelId,\n\t\t\t);\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 (setThinkingLevel clamps to model capabilities)\n\t\tif (sessionContext.thinkingLevel) {\n\t\t\tthis.setThinkingLevel(sessionContext.thinkingLevel as ThinkingLevel);\n\t\t}\n\n\t\tthis._reconnectToAgent();\n\t\treturn true;\n\t}\n\n\t/**\n\t * Create a branch from a specific entry index.\n\t * Emits before_branch/branch session events to hooks.\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 * - cancelled: True if a hook cancelled the branch\n\t */\n\tasync branch(entryIndex: number): Promise<{ selectedText: string; cancelled: boolean }> {\n\t\tconst previousSessionFile = this.sessionFile;\n\t\tconst entries = this.sessionManager.getEntries();\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\tlet skipConversationRestore = false;\n\n\t\t// Emit before_branch event (can be cancelled)\n\t\tif (this._hookRunner?.hasHandlers(\"session\")) {\n\t\t\tconst result = (await this._hookRunner.emit({\n\t\t\t\ttype: \"session\",\n\t\t\t\tentries,\n\t\t\t\tsessionFile: this.sessionFile,\n\t\t\t\tpreviousSessionFile: null,\n\t\t\t\treason: \"before_branch\",\n\t\t\t\ttargetTurnIndex: entryIndex,\n\t\t\t})) as SessionEventResult | undefined;\n\n\t\t\tif (result?.cancel) {\n\t\t\t\treturn { selectedText, cancelled: true };\n\t\t\t}\n\t\t\tskipConversationRestore = result?.skipConversationRestore ?? false;\n\t\t}\n\n\t\t// Create branched session (returns null in --no-session mode)\n\t\tconst newSessionFile = this.sessionManager.createBranchedSessionFromEntries(entries, entryIndex);\n\n\t\t// Update session file if we have one (file-based mode)\n\t\tif (newSessionFile !== null) {\n\t\t\tthis.sessionManager.setSessionFile(newSessionFile);\n\t\t}\n\n\t\t// Reload messages from entries (works for both file and in-memory mode)\n\t\tconst newEntries = this.sessionManager.getEntries();\n\t\tconst sessionContext = this.sessionManager.buildSessionContext();\n\n\t\t// Emit branch event to hooks (after branch completes)\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\",\n\t\t\t\tentries: newEntries,\n\t\t\t\tsessionFile: newSessionFile,\n\t\t\t\tpreviousSessionFile,\n\t\t\t\treason: \"branch\",\n\t\t\t\ttargetTurnIndex: entryIndex,\n\t\t\t});\n\t\t}\n\n\t\t// Emit session event to custom tools (with reason \"branch\")\n\t\tawait this._emitToolSessionEvent(\"branch\", previousSessionFile);\n\n\t\tif (!skipConversationRestore) {\n\t\t\tthis.agent.replaceMessages(sessionContext.messages);\n\t\t}\n\n\t\treturn { selectedText, cancelled: 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.getEntries();\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\tconst themeName = this.settingsManager.getTheme();\n\t\treturn exportSessionToHtml(this.sessionManager, this.state, { outputPath, themeName });\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\t/**\n\t * Get custom tools (for setting UI context in modes).\n\t */\n\tget customTools(): LoadedCustomTool[] {\n\t\treturn this._customTools;\n\t}\n\n\t/**\n\t * Emit session event to all custom tools.\n\t * Called on session switch, branch, and clear.\n\t */\n\tprivate async _emitToolSessionEvent(\n\t\treason: ToolSessionEvent[\"reason\"],\n\t\tpreviousSessionFile: string | null,\n\t): Promise<void> {\n\t\tconst event: ToolSessionEvent = {\n\t\t\tentries: this.sessionManager.getEntries(),\n\t\t\tsessionFile: this.sessionFile,\n\t\t\tpreviousSessionFile,\n\t\t\treason,\n\t\t};\n\t\tfor (const { tool } of this._customTools) {\n\t\t\tif (tool.onSession) {\n\t\t\t\ttry {\n\t\t\t\t\tawait tool.onSession(event);\n\t\t\t\t} catch (_err) {\n\t\t\t\t\t// Silently ignore tool errors during session events\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\t}\n}\n"]}
@@ -18,7 +18,6 @@ import { executeBash as executeBashCommand } from "./bash-executor.js";
18
18
  import { calculateContextTokens, compact, prepareCompaction, shouldCompact } from "./compaction.js";
19
19
  import { exportSessionToHtml } from "./export-html.js";
20
20
  import { getApiKeyForModel, getAvailableModels } from "./model-config.js";
21
- import { loadSessionFromEntries } from "./session-manager.js";
22
21
  import { expandSlashCommand } from "./slash-commands.js";
23
22
  // ============================================================================
24
23
  // Constants
@@ -370,7 +369,7 @@ export class AgentSession {
370
369
  */
371
370
  async reset() {
372
371
  const previousSessionFile = this.sessionFile;
373
- const entries = this.sessionManager.loadEntries();
372
+ const entries = this.sessionManager.getEntries();
374
373
  // Emit before_clear event (can be cancelled)
375
374
  if (this._hookRunner?.hasHandlers("session")) {
376
375
  const result = (await this._hookRunner.emit({
@@ -572,12 +571,20 @@ export class AgentSession {
572
571
  if (!apiKey) {
573
572
  throw new Error(`No API key for ${this.model.provider}`);
574
573
  }
575
- const entries = this.sessionManager.loadEntries();
574
+ const entries = this.sessionManager.getEntries();
576
575
  const settings = this.settingsManager.getCompactionSettings();
577
576
  const preparation = prepareCompaction(entries, settings);
578
577
  if (!preparation) {
579
578
  throw new Error("Already compacted");
580
579
  }
580
+ // Find previous compaction summary if any
581
+ let previousSummary;
582
+ for (let i = entries.length - 1; i >= 0; i--) {
583
+ if (entries[i].type === "compaction") {
584
+ previousSummary = entries[i].summary;
585
+ break;
586
+ }
587
+ }
581
588
  let compactionEntry;
582
589
  let fromHook = false;
583
590
  if (this._hookRunner?.hasHandlers("session")) {
@@ -588,11 +595,14 @@ export class AgentSession {
588
595
  previousSessionFile: null,
589
596
  reason: "before_compact",
590
597
  cutPoint: preparation.cutPoint,
591
- messagesToSummarize: preparation.messagesToSummarize,
598
+ previousSummary,
599
+ messagesToSummarize: [...preparation.messagesToSummarize],
600
+ messagesToKeep: [...preparation.messagesToKeep],
592
601
  tokensBefore: preparation.tokensBefore,
593
602
  customInstructions,
594
603
  model: this.model,
595
- apiKey,
604
+ resolveApiKey: this._resolveApiKey,
605
+ signal: this._compactionAbortController.signal,
596
606
  }));
597
607
  if (result?.cancel) {
598
608
  throw new Error("Compaction cancelled");
@@ -609,9 +619,9 @@ export class AgentSession {
609
619
  throw new Error("Compaction cancelled");
610
620
  }
611
621
  this.sessionManager.saveCompaction(compactionEntry);
612
- const newEntries = this.sessionManager.loadEntries();
613
- const loaded = loadSessionFromEntries(newEntries);
614
- this.agent.replaceMessages(loaded.messages);
622
+ const newEntries = this.sessionManager.getEntries();
623
+ const sessionContext = this.sessionManager.buildSessionContext();
624
+ this.agent.replaceMessages(sessionContext.messages);
615
625
  if (this._hookRunner) {
616
626
  await this._hookRunner.emit({
617
627
  type: "session",
@@ -697,12 +707,20 @@ export class AgentSession {
697
707
  this._emit({ type: "auto_compaction_end", result: null, aborted: false, willRetry: false });
698
708
  return;
699
709
  }
700
- const entries = this.sessionManager.loadEntries();
710
+ const entries = this.sessionManager.getEntries();
701
711
  const preparation = prepareCompaction(entries, settings);
702
712
  if (!preparation) {
703
713
  this._emit({ type: "auto_compaction_end", result: null, aborted: false, willRetry: false });
704
714
  return;
705
715
  }
716
+ // Find previous compaction summary if any
717
+ let previousSummary;
718
+ for (let i = entries.length - 1; i >= 0; i--) {
719
+ if (entries[i].type === "compaction") {
720
+ previousSummary = entries[i].summary;
721
+ break;
722
+ }
723
+ }
706
724
  let compactionEntry;
707
725
  let fromHook = false;
708
726
  if (this._hookRunner?.hasHandlers("session")) {
@@ -713,11 +731,14 @@ export class AgentSession {
713
731
  previousSessionFile: null,
714
732
  reason: "before_compact",
715
733
  cutPoint: preparation.cutPoint,
716
- messagesToSummarize: preparation.messagesToSummarize,
734
+ previousSummary,
735
+ messagesToSummarize: [...preparation.messagesToSummarize],
736
+ messagesToKeep: [...preparation.messagesToKeep],
717
737
  tokensBefore: preparation.tokensBefore,
718
738
  customInstructions: undefined,
719
739
  model: this.model,
720
- apiKey,
740
+ resolveApiKey: this._resolveApiKey,
741
+ signal: this._autoCompactionAbortController.signal,
721
742
  }));
722
743
  if (hookResult?.cancel) {
723
744
  this._emit({ type: "auto_compaction_end", result: null, aborted: true, willRetry: false });
@@ -736,9 +757,9 @@ export class AgentSession {
736
757
  return;
737
758
  }
738
759
  this.sessionManager.saveCompaction(compactionEntry);
739
- const newEntries = this.sessionManager.loadEntries();
740
- const loaded = loadSessionFromEntries(newEntries);
741
- this.agent.replaceMessages(loaded.messages);
760
+ const newEntries = this.sessionManager.getEntries();
761
+ const sessionContext = this.sessionManager.buildSessionContext();
762
+ this.agent.replaceMessages(sessionContext.messages);
742
763
  if (this._hookRunner) {
743
764
  await this._hookRunner.emit({
744
765
  type: "session",
@@ -1004,7 +1025,7 @@ export class AgentSession {
1004
1025
  */
1005
1026
  async switchSession(sessionPath) {
1006
1027
  const previousSessionFile = this.sessionFile;
1007
- const oldEntries = this.sessionManager.loadEntries();
1028
+ const oldEntries = this.sessionManager.getEntries();
1008
1029
  // Emit before_switch event (can be cancelled)
1009
1030
  if (this._hookRunner?.hasHandlers("session")) {
1010
1031
  const result = (await this._hookRunner.emit({
@@ -1024,8 +1045,8 @@ export class AgentSession {
1024
1045
  // Set new session
1025
1046
  this.sessionManager.setSessionFile(sessionPath);
1026
1047
  // Reload messages
1027
- const entries = this.sessionManager.loadEntries();
1028
- const loaded = loadSessionFromEntries(entries);
1048
+ const entries = this.sessionManager.getEntries();
1049
+ const sessionContext = this.sessionManager.buildSessionContext();
1029
1050
  // Emit session event to hooks
1030
1051
  if (this._hookRunner) {
1031
1052
  this._hookRunner.setSessionFile(sessionPath);
@@ -1039,20 +1060,18 @@ export class AgentSession {
1039
1060
  }
1040
1061
  // Emit session event to custom tools
1041
1062
  await this._emitToolSessionEvent("switch", previousSessionFile);
1042
- this.agent.replaceMessages(loaded.messages);
1063
+ this.agent.replaceMessages(sessionContext.messages);
1043
1064
  // Restore model if saved
1044
- const savedModel = this.sessionManager.loadModel();
1045
- if (savedModel) {
1065
+ if (sessionContext.model) {
1046
1066
  const availableModels = (await getAvailableModels()).models;
1047
- const match = availableModels.find((m) => m.provider === savedModel.provider && m.id === savedModel.modelId);
1067
+ const match = availableModels.find((m) => m.provider === sessionContext.model.provider && m.id === sessionContext.model.modelId);
1048
1068
  if (match) {
1049
1069
  this.agent.setModel(match);
1050
1070
  }
1051
1071
  }
1052
1072
  // Restore thinking level if saved (setThinkingLevel clamps to model capabilities)
1053
- const savedThinking = this.sessionManager.loadThinkingLevel();
1054
- if (savedThinking) {
1055
- this.setThinkingLevel(savedThinking);
1073
+ if (sessionContext.thinkingLevel) {
1074
+ this.setThinkingLevel(sessionContext.thinkingLevel);
1056
1075
  }
1057
1076
  this._reconnectToAgent();
1058
1077
  return true;
@@ -1068,7 +1087,7 @@ export class AgentSession {
1068
1087
  */
1069
1088
  async branch(entryIndex) {
1070
1089
  const previousSessionFile = this.sessionFile;
1071
- const entries = this.sessionManager.loadEntries();
1090
+ const entries = this.sessionManager.getEntries();
1072
1091
  const selectedEntry = entries[entryIndex];
1073
1092
  if (!selectedEntry || selectedEntry.type !== "message" || selectedEntry.message.role !== "user") {
1074
1093
  throw new Error("Invalid entry index for branching");
@@ -1097,8 +1116,8 @@ export class AgentSession {
1097
1116
  this.sessionManager.setSessionFile(newSessionFile);
1098
1117
  }
1099
1118
  // Reload messages from entries (works for both file and in-memory mode)
1100
- const newEntries = this.sessionManager.loadEntries();
1101
- const loaded = loadSessionFromEntries(newEntries);
1119
+ const newEntries = this.sessionManager.getEntries();
1120
+ const sessionContext = this.sessionManager.buildSessionContext();
1102
1121
  // Emit branch event to hooks (after branch completes)
1103
1122
  if (this._hookRunner) {
1104
1123
  this._hookRunner.setSessionFile(newSessionFile);
@@ -1114,7 +1133,7 @@ export class AgentSession {
1114
1133
  // Emit session event to custom tools (with reason "branch")
1115
1134
  await this._emitToolSessionEvent("branch", previousSessionFile);
1116
1135
  if (!skipConversationRestore) {
1117
- this.agent.replaceMessages(loaded.messages);
1136
+ this.agent.replaceMessages(sessionContext.messages);
1118
1137
  }
1119
1138
  return { selectedText, cancelled: false };
1120
1139
  }
@@ -1122,7 +1141,7 @@ export class AgentSession {
1122
1141
  * Get all user messages from session for branch selector.
1123
1142
  */
1124
1143
  getUserMessagesForBranching() {
1125
- const entries = this.sessionManager.loadEntries();
1144
+ const entries = this.sessionManager.getEntries();
1126
1145
  const result = [];
1127
1146
  for (let i = 0; i < entries.length; i++) {
1128
1147
  const entry = entries[i];
@@ -1250,7 +1269,7 @@ export class AgentSession {
1250
1269
  */
1251
1270
  async _emitToolSessionEvent(reason, previousSessionFile) {
1252
1271
  const event = {
1253
- entries: this.sessionManager.loadEntries(),
1272
+ entries: this.sessionManager.getEntries(),
1254
1273
  sessionFile: this.sessionFile,
1255
1274
  previousSessionFile,
1256
1275
  reason,