83 Commits

• Rework custom tools API with CustomToolContext

  - CustomAgentTool renamed to CustomTool
  - ToolAPI renamed to CustomToolAPI
  - ToolContext renamed to CustomToolContext
  - ToolSessionEvent renamed to CustomToolSessionEvent
  - Added CustomToolContext parameter to execute() and onSession()
  - CustomToolFactory now returns CustomTool<any, any> for type compatibility
  - dispose() replaced with onSession({ reason: 'shutdown' })
  - Added wrapCustomTool() to convert CustomTool to AgentTool
  - Session exposes setToolUIContext() instead of leaking internals
  - Fix ToolExecutionComponent to sync with toolOutputExpanded state
  - Update all custom tool examples for new API

  Mario Zechner · 2025-12-31 12:05:24 +01:00
  568150f18b
• Replace custom tool dispose() with shutdown session event

  Breaking change: CustomAgentTool.dispose() removed. Use onSession with
  reason 'shutdown' instead for cleanup.
  
  - Add 'shutdown' to SessionEvent.reason for custom tools
  - Remove dispose() method from CustomAgentTool interface
  - Make emitToolSessionEvent() public on AgentSession
  - Emit shutdown event to tools in InteractiveMode.shutdown()
  - Update custom-tools.md with new API and examples

  Mario Zechner · 2025-12-31 02:55:45 +01:00
  ff78ac2f84
• refactor(coding-agent): fix compaction for branched sessions, consolidate hook context types

  Compaction API:
  - prepareCompaction() now takes (pathEntries, settings) only
  - CompactionPreparation restructured: removed cutPoint/messagesToKeep/boundaryStart, added turnPrefixMessages/isSplitTurn/previousSummary/fileOps/settings
  - compact() now takes (preparation, model, apiKey, customInstructions?, signal?)
  - Fixed token overflow by using getPath() instead of getEntries()
  
  Hook types:
  - HookEventContext renamed to HookContext
  - HookCommandContext removed, RegisteredCommand.handler takes (args, ctx)
  - HookContext now includes model field
  - SessionBeforeCompactEvent: removed previousCompactions/model, added branchEntries
  - SessionBeforeTreeEvent: removed model (use ctx.model)
  - HookRunner.initialize() added for modes to set up callbacks

  Mario Zechner · 2025-12-31 02:24:24 +01:00
  ddda8b124c
• Pass fromHook to appendCompaction for both manual and auto compaction
  Mario Zechner · 2025-12-30 22:42:24 +01:00
  d4dc07ab20
• Add fromHook field to CompactionEntry and BranchSummaryEntry

  - fromHook: true = hook generated, skip file extraction
  - fromHook: undefined/false = pi generated, extract files (backward compatible)
  - branchWithSummary now accepts fromHook parameter
  - File extraction only runs for !entry.fromHook entries

  Mario Zechner · 2025-12-30 22:42:24 +01:00
  f118cdc67b
• Store file lists in BranchSummaryEntry.details for cumulative tracking

  - BranchSummaryResult now returns readFiles and modifiedFiles separately
  - BranchSummaryDetails type for details: { readFiles, modifiedFiles }
  - branchWithSummary accepts optional details parameter
  - Collect files from existing branch_summary.details when preparing entries
  - Files accumulate across nested branch summaries

  Mario Zechner · 2025-12-30 22:42:24 +01:00
  4ef3325cec
• Use reserveTokens for branch summary (tokens left for prompt + response)

  - tokenBudget = contextWindow - reserveTokens
  - Default 16384, same as compaction
  - Consistent naming with CompactionSettings.reserveTokens

  Mario Zechner · 2025-12-30 22:42:24 +01:00
  dc5fc4fc40
• Use token-based maxTokens instead of fraction-based reserveFraction

  - BranchSummarySettings.maxTokens (default 100000) instead of reserveFraction
  - More intuitive and consistent with CompactionSettings.keepRecentTokens

  Mario Zechner · 2025-12-30 22:42:24 +01:00
  f5f39f08f1
• Use AgentMessage in BranchPreparation and add BranchSummarySettings

  - BranchPreparation now uses AgentMessage[] instead of custom type
  - Reuse getMessageFromEntry pattern from compaction.ts
  - Add BranchSummarySettings with reserveFraction to settings.json
  - Add getBranchSummarySettings() to SettingsManager
  - Use settings for reserveFraction instead of hardcoded value

  Mario Zechner · 2025-12-30 22:42:23 +01:00
  839a46e6fe
• Add ReadonlySessionManager and refactor branch summarization

  - Add ReadonlySessionManager interface to session-manager.ts
  - Re-export from hooks/index.ts
  - Add collectEntriesForBranchSummary() to extract entries for summarization
  - Don't stop at compaction boundaries (include their summaries as context)
  - Add token budget support to prepareBranchEntries()
  - Walk entries newest-to-oldest to prioritize recent context
  - Use options object for generateBranchSummary()
  - Handle compaction entries as context summaries
  - Export new types: CollectEntriesResult, GenerateBranchSummaryOptions

  Mario Zechner · 2025-12-30 22:42:23 +01:00
  08fab16e2d
• Refactor: move compaction code to src/core/compaction/

  - Move compaction.ts to src/core/compaction/compaction.ts
  - Extract branch summarization to src/core/compaction/branch-summarization.ts
  - Add index.ts to re-export all compaction utilities
  - Update all imports across the codebase

  Mario Zechner · 2025-12-30 22:42:23 +01:00
  fd13b53b1c
• Add tree navigation tests and shared test utilities

  - Add test/utilities.ts with shared helpers (API_KEY, userMsg, assistantMsg, createTestSession)
  - Add agent-session-tree-navigation.test.ts with e2e tests for tree navigation
  - Add getChildren() method to SessionManager
  - Add summaryEntry to navigateTree return type
  - Update existing tests to use shared utilities

  Mario Zechner · 2025-12-30 22:42:23 +01:00
  31c5cd38d1
• Fix branch summarization abort handling and tree navigation

  - Check response.stopReason instead of catching errors for abort detection
  - Return result object from _generateBranchSummary instead of throwing
  - Fix summary attachment: attach to navigation target position, not old branch
  - Support root-level summaries (parentId=null) in branchWithSummary
  - Remove setTimeout hack for re-showing tree selector on abort

  Mario Zechner · 2025-12-30 22:42:23 +01:00
  01dae9ebcc
• WIP: Add branch summarization abort support with loader and escape handler
  Mario Zechner · 2025-12-30 22:42:23 +01:00
  9dac0a1423
• feat(coding-agent): implement /tree command for session tree navigation

  - Add TreeSelectorComponent with ASCII tree visualization
  - Add AgentSession.navigateTree() for switching branches
  - Add session_before_tree/session_tree hook events
  - Add SessionManager.resetLeaf() for navigating to root
  - Change leafId from string to string|null for consistency with parentId
  - Support optional branch summarization when switching
  - Update buildSessionContext() to handle null leafId
  - Add /tree to slash commands in interactive mode

  Mario Zechner · 2025-12-30 22:42:22 +01:00
  4958271dd3
• refactor(hooks): split session events into individual typed events

  Major changes:
  - Replace monolithic SessionEvent with reason discriminator with individual
    event types: session_start, session_before_switch, session_switch,
    session_before_new, session_new, session_before_branch, session_branch,
    session_before_compact, session_compact, session_shutdown
  - Each event has dedicated result type (SessionBeforeSwitchResult, etc.)
  - HookHandler type now allows bare return statements (void in return type)
  - HookAPI.on() has proper overloads for each event with correct typing
  
  Additional fixes:
  - AgentSession now always subscribes to agent in constructor (was only
    subscribing when external subscribe() called, breaking internal handlers)
  - Standardize on undefined over null throughout codebase
  - HookUIContext methods return undefined instead of null
  - SessionManager methods return undefined instead of null
  - Simplify hook exports to 'export type * from types.js'
  - Add detailed JSDoc for skipConversationRestore vs cancel
  - Fix createBranchedSession to rebuild index in persist mode
  - newSession() now returns the session file path
  
  Updated all example hooks, tests, and emission sites to use new event types.

  Mario Zechner · 2025-12-30 22:42:22 +01:00
  d6283f99dc
• Support multiple messages in agent.prompt() and agentLoop

  - agentLoop now accepts AgentMessage[] instead of single message
  - agent.prompt() accepts AgentMessage | AgentMessage[]
  - Emits message_start/end for each message in the array
  - AgentSession.prompt() builds array with hook message + user message
  - TUI now receives events for before_agent_start injected messages

  Mario Zechner · 2025-12-30 22:42:22 +01:00
  41af99cccf
• Implement before_agent_start hook event

  - Add BeforeAgentStartEvent and BeforeAgentStartEventResult types
  - Add emitBeforeAgentStart to HookRunner
  - Call in AgentSession.prompt() before agent.prompt()
  - Hook can return a message to inject into context (persisted + visible)
  - Add test hook demonstrating custom message rendering and before_agent_start

  Mario Zechner · 2025-12-30 22:42:22 +01:00
  57146de202
• Use exhaustive switch on message.role throughout coding-agent

  - addMessageToChat: exhaustive switch for all AgentMessage roles
  - renderSessionContext: delegates to addMessageToChat, special handling for assistant tool calls and tool results
  - export-html formatMessage: exhaustive switch for all AgentMessage roles
  - Removed isHookMessage, isBashExecutionMessage type guards in favor of role checks
  - Fixed imports and removed unused getLatestCompactionEntry

  Mario Zechner · 2025-12-30 22:42:21 +01:00
  b921298af7
• WIP: Major cleanup - move Attachment to consumers, simplify agent API

  - Removed Attachment from agent package (now in web-ui/coding-agent)
  - Agent.prompt now takes (text, images?: ImageContent[])
  - Removed transports from web-ui (duplicate of agent package)
  - Updated coding-agent to use local message types
  - Updated mom package for new agent API
  
  Remaining: Fix AgentInterface.ts to compose UserMessageWithAttachments

  Mario Zechner · 2025-12-30 22:42:20 +01:00
  6ddc7418da
• WIP: Refactor agent package - not compiling

  - Renamed AppMessage to AgentMessage throughout
  - New agent-loop.ts with AgentLoopContext, AgentLoopConfig
  - Removed transport abstraction, Agent now takes streamFn directly
  - Extracted streamProxy to proxy.ts utility
  - Removed agent-loop from pi-ai (now in agent package)
  - Updated consumers (coding-agent, mom) for AgentMessage rename
  - Tests updated but some consumers still need migration
  
  Known issues:
  - AgentTool, AgentToolResult not exported from pi-ai
  - Attachment not exported from pi-agent-core
  - ProviderTransport removed but still referenced
  - messageTransformer -> convertToLlm migration incomplete
  - CustomMessages declaration merging not working properly

  Mario Zechner · 2025-12-30 22:42:20 +01:00
  a055fd4481
• Cleanup: unify HookMessage naming and simplify SessionContext

  - Rename HookAppMessage to HookMessage, isHookAppMessage to isHookMessage
  - Remove entries array from SessionContext (use isHookMessage type guard instead)
  - HookMessage.content now accepts string directly (not just array)
  - Fix streamMessage type in AgentState (AppMessage, not Message)
  - Rename CustomMessageComponent to HookMessageComponent
  - Fix test hook to use pi.sendMessage

  Mario Zechner · 2025-12-30 22:42:20 +01:00
  204d27581b
• Move exec to HookAPI, sessionManager/modelRegistry to HookEventContext

  Breaking changes:
  - HookEventContext now has sessionManager and modelRegistry (moved from SessionEventBase)
  - HookAPI now has exec() method (moved from HookEventContext/HookCommandContext)
  - HookRunner constructor takes sessionManager and modelRegistry as required params
  - Session events no longer include sessionManager/modelRegistry fields
  
  Hook code migration:
  - event.sessionManager -> ctx.sessionManager
  - event.modelRegistry -> ctx.modelRegistry
  - ctx.exec() -> pi.exec()
  
  Updated:
  - src/core/hooks/types.ts - type changes
  - src/core/hooks/runner.ts - constructor, createContext
  - src/core/hooks/loader.ts - add exec to HookAPI
  - src/core/sdk.ts - pass sessionManager/modelRegistry to HookRunner
  - src/core/agent-session.ts - remove sessionManager/modelRegistry from events
  - src/modes/* - remove setSessionFile calls, update events
  - examples/hooks/* - update to new API

  Mario Zechner · 2025-12-30 22:42:19 +01:00
  29fec7848e
• Update plan: HookCommandContext without sendMessage (use pi closure)
  Mario Zechner · 2025-12-30 22:42:19 +01:00
  09e7e9196c
• Add Agent.prompt(AppMessage) overload for custom message types

  Instead of using continue() which validates roles, prompt() now accepts
  an AppMessage directly. This allows hook messages with role: 'hookMessage'
  to trigger proper agent loop with message events.
  
  - Add overloads: prompt(AppMessage) and prompt(string, attachments?)
  - sendHookMessage uses prompt(appMessage) instead of appendMessage+continue

  Mario Zechner · 2025-12-30 22:42:19 +01:00
  a6322fda59
• Use proper HookAppMessage type instead of _hookData marker

  Following the same pattern as BashExecutionMessage:
  - HookAppMessage has role: 'hookMessage' with customType, content, display, details
  - isHookAppMessage() type guard for checking message type
  - messageTransformer converts to user message for LLM context
  - TUI checks isHookAppMessage() for rendering as CustomMessageComponent
  
  This makes the API clean for anyone building on AgentSession - they can
  use the type guard instead of knowing about internal marker fields.

  Mario Zechner · 2025-12-30 22:42:19 +01:00
  75a9c3c714
• Add emitLastMessage flag to agent.continue()

  When calling continue() with emitLastMessage=true, the agent loop
  emits message_start/message_end events for the last message in context.
  This allows messages added outside the loop (e.g., hook messages via
  sendHookMessage) to trigger proper TUI rendering.
  
  Changes across packages:
  - packages/ai: agentLoopContinue() accepts emitLastMessage parameter
  - packages/agent: Agent.continue(), transports updated to pass flag
  - packages/coding-agent: sendHookMessage passes true when triggerTurn

  Mario Zechner · 2025-12-30 22:42:19 +01:00
  357bd946c2
• Hook commands: remove string return, use sendMessage() for prompting

  - Command handler now returns Promise<void> instead of Promise<string | undefined>
  - To trigger LLM response, use sendMessage() with triggerTurn: true
  - Simplify _tryExecuteHookCommand to return boolean
  
  Added example hook and slash command in .pi/:
  - .pi/hooks/test-command.ts - /greet command using sendMessage
  - .pi/commands/review.md - file-based /review command

  Mario Zechner · 2025-12-30 22:42:18 +01:00
  30cd723411
• Move hook command execution to AgentSession.prompt()

  Hook commands registered via pi.registerCommand() are now handled in
  AgentSession.prompt() alongside file-based slash commands. This:
  
  - Removes duplicate tryHandleHookCommand from interactive-mode and rpc-mode
  - All modes (interactive, RPC, print) share the same command handling logic
  - AgentSession._tryExecuteHookCommand() builds CommandContext using:
    - UI context from hookRunner (set by mode)
    - sessionManager, modelRegistry from AgentSession
    - sendMessage via sendHookMessage
    - exec via exported execCommand
  - Handler returning string uses it as prompt, undefined returns early
  
  Also:
  - Export execCommand from hooks/runner.ts
  - Add getUIContext() and getHasUI() to HookRunner
  - Make HookRunner.emitError() public for error reporting

  Mario Zechner · 2025-12-30 22:42:18 +01:00
  c8d9382aaa
• Hook API: replace send() with sendMessage(), add appendEntry() and registerCommand()

  Breaking changes to Hook API:
  - pi.send(text, attachments?) replaced with pi.sendMessage(message, triggerTurn?)
    - Creates CustomMessageEntry instead of user messages
    - Properly handles queuing during streaming via agent loop
    - Supports optional turn triggering when idle
  - New pi.appendEntry(customType, data?) for hook state persistence
  - New pi.registerCommand(name, options) for custom slash commands
  - Handler types renamed: SendHandler -> SendMessageHandler, new AppendEntryHandler
  
  Implementation:
  - AgentSession.sendHookMessage() handles all three cases:
    - Streaming: queues message with _hookData marker, agent loop processes it
    - Not streaming + triggerTurn: appends to state/session, calls agent.continue()
    - Not streaming + no trigger: appends to state/session only
  - message_end handler routes based on _hookData presence to correct persistence
  - HookRunner gains getRegisteredCommands() and getCommand() methods
  
  New types: HookMessage<T>, RegisteredCommand, CommandContext

  Mario Zechner · 2025-12-30 22:42:18 +01:00
  ba185b0571
• Refactor SessionEventBase to pass sessionManager and modelRegistry

  Breaking changes to hook types:
  - SessionEventBase now passes sessionManager and modelRegistry directly
  - before_compact: passes preparation, previousCompactions (newest first)
  - before_switch: has targetSessionFile; switch: has previousSessionFile
  - Removed resolveApiKey (use modelRegistry.getApiKey())
  - getSessionFile() returns string | undefined for in-memory sessions
  
  Updated:
  - All session event emissions in agent-session.ts
  - Hook examples (custom-compaction.ts, auto-commit-on-exit.ts, confirm-destructive.ts)
  - Tests (compaction-hooks.test.ts, compaction-hooks-example.test.ts)
  - export-html.ts guards for in-memory sessions

  Mario Zechner · 2025-12-30 22:42:18 +01:00
  9bba388ec5
• Make CompactionEntry and CompactionResult generic with details field

  - CompactionEntry<T> and CompactionResult<T> now have optional details?: T
  - appendCompaction() accepts optional details parameter
  - Hooks can return compaction.details to store custom data
  - Enables structured compaction with ArtifactIndex (see #314)
  - Fix CompactionResult export location (now from compaction.ts)
  - Update plan with remaining compaction refactor items

  Mario Zechner · 2025-12-30 22:42:18 +01:00
  d96375b5e5
• Session tree: simplify types, add branching API, comprehensive tests

  Types:
  - SessionEntryBase with type field, extended by all entry types
  - CustomEntry for hooks (type: 'custom', customType, data)
  - Remove XXXContent types and TreeNode (redundant)
  
  API:
  - Rename saveXXX to appendXXX with JSDoc explaining tree semantics
  - Rename branchInPlace to branch() with better docs
  - Add createBranchedSession(leafId) replacing index-based version
  - Add getTree() returning SessionTreeNode[] for tree traversal
  - Add appendCustomEntry(customType, data) for hooks
  
  Tests:
  - tree-traversal.test.ts: 28 tests covering append, getPath, getTree,
    branch, branchWithSummary, createBranchedSession
  - save-entry.test.ts: custom entry integration
  
  Docs:
  - Class-level JSDoc explaining append-only tree model
  - Method docs explaining leaf advancement and branching
  - CHANGELOG.md entry for all changes

  Mario Zechner · 2025-12-30 22:42:18 +01:00
  6f94e24629
• Fix SessionEntry type to exclude SessionHeader

  - SessionEntry now only contains conversation entries (messages, compaction, etc.)
  - SessionHeader is separate, not part of SessionEntry
  - FileEntry = SessionHeader | SessionEntry (for file storage)
  - getEntries() filters out header, returns SessionEntry[]
  - Added getHeader() for accessing session metadata
  - Updated compaction and tests to not expect header in entries
  - Updated mom package to use FileEntry for internal storage

  Mario Zechner · 2025-12-30 22:42:17 +01:00
  9478a3c1f5
• Session tree structure with id/parentId linking

  - Add TreeNode base type with id, parentId, timestamp
  - Add *Content types for clean input/output separation
  - Entry types are now TreeNode & *Content intersections
  - SessionManager assigns id/parentId on save, tracks leafId
  - Add migrateSessionEntries() for v1 to v2 conversion
  - Migration runs on load, rewrites file
  - buildSessionContext() uses tree traversal from leaf
  - Compaction returns CompactionResult (content only)
  - Hooks return compaction content, not full entries
  - Add firstKeptEntryId to before_compact hook event
  - Update mom package for tree fields
  - Better error messages for compaction failures

  Mario Zechner · 2025-12-30 22:42:17 +01:00
  c58d5f20a4
• Merge pull request #346 from ronyrus/main

  Improve error message when `apiKey` or `model` are absent

  Mario Zechner · 2025-12-29 13:38:21 +01:00
  65fb9116fb
• Fix /copy after aborts
  Armin Ronacher · 2025-12-29 11:54:45 +01:00
  e30c4e30cb
• Improve error message when apiKey or model are absent
  Rony Kelner · 2025-12-29 10:39:04 +09:00
  47b11bf798
• fix: use consistent model comparison including provider
  Kao Félix · 2025-12-25 22:10:08 +01:00
  a7efe3d4c1
• Reverse model switching and binding for dialog
  Armin Ronacher · 2025-12-25 18:07:36 +01:00
  b576a527c7
• Rename /clear to /new, update hook events to before_new/new

  Closes #305 - took direct rename approach instead of alias system
  
  Thanks @mitsuhiko for the nudge!

  Mario Zechner · 2025-12-25 04:15:10 +01:00
  454ea1d36a
• Refactor OAuth/API key handling: AuthStorage and ModelRegistry

  - Add AuthStorage class for credential storage (auth.json)
  - Add ModelRegistry class for model management with API key resolution
  - Add discoverAuthStorage() and discoverModels() discovery functions
  - Add migration from legacy oauth.json and settings.json apiKeys to auth.json
  - Remove configureOAuthStorage, defaultGetApiKey, findModel, discoverAvailableModels
  - Remove apiKeys from Settings type and SettingsManager methods
  - Rename getOAuthPath to getAuthPath
  - Update SDK, examples, docs, tests, and mom package
  
  Fixes #296

  Mario Zechner · 2025-12-25 03:48:36 +01:00
  54018b6cc0
• WIP: Rename model-config.ts to models-json.ts

  - loadCustomModels now takes file path instead of agentDir

  Mario Zechner · 2025-12-25 01:15:17 +01:00
  1c31d91c83
• Fix model selector not showing models with settings.json API keys

  Fixes #295

  Mario Zechner · 2025-12-24 21:23:44 +01:00
  ac5f4a77cc
• Improve compaction hooks: add signal, no timeout, SessionManager cleanup, docs
  Mario Zechner · 2025-12-24 13:54:05 +01:00
  705ba5d4f2
• Add previousSummary to before_compact hook event
  Mario Zechner · 2025-12-24 12:47:12 +01:00
  403faafdbe
• Improve before_compact hook: add messagesToKeep, replace apiKey with resolveApiKey
  Mario Zechner · 2025-12-24 12:41:22 +01:00
  d9a542763a
• Add resolveApiKey to before_compact hook event
  Mario Zechner · 2025-12-24 12:28:51 +01:00
  43a5447a80
• feat: Improve HTML export with markdown rendering and theme support

  - Render markdown server-side using marked (tables, headings, code blocks, etc.)
  - Honor user's chosen theme (light/dark) with appropriate color defaults
  - Add image rendering for user messages
  - Style code blocks with TUI-like ```language markers

  scutifer · 2025-12-24 12:13:15 +01:00
  531bb91337
• Add before_compact hook event (closes #281) (#285)

  * Add before_compact hook event (closes #281)
  
  * Add compact hook event and documentation
  
  - Add compact event that fires after compaction completes
  - Update hooks.md with lifecycle diagram, field docs, and example
  - Add CHANGELOG entry
  - Add comprehensive test coverage (10 tests) for before_compact and compact events
  - Tests cover: event emission, cancellation, custom entry, error handling, multiple hooks

  Nico Bailon · 2025-12-24 11:26:29 +01:00
  1e1a92ea47