agent/pi
2
0
Fork 0
mirror of https://github.com/earendil-works/pi.git synced 2026-06-18 15:54:04 +08:00
Code Issues Packages Projects Releases Wiki Activity
1,532 Commits 32 Branches 288 Tags
aea2ced2b8dbf9569f6cb8d0cc7e37573d3b2f2e
Commit Graph

44 Commits

  • Consolidate session events: remove session_before_new/session_new, add reason field to switch events 
    - Remove session_before_new and session_new hook events
- Add reason: 'new' | 'resume' to session_before_switch and session_switch events
- Remove 'new' reason from custom tool onSession (use 'switch' for both /new and /resume)
- Rename reset() to newSession(options?) in AgentSession
- Add NewSessionOptions with optional parentSession for lineage tracking
- Rename branchedFrom to parentSession in SessionHeader
- Rename RPC reset command to new_session with optional parentSession
- Update example hooks to use new event structure
- Update documentation and changelog

Based on discussion in #293
    Mario Zechner · 2026-01-01 23:31:26 +01:00
    484d7e06bb
  • 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
  • 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
  • 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
  • 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
  • Add global debug key (Shift+Ctrl+D), iterative tree sorting to avoid stack overflow
    Mario Zechner · 2025-12-30 22:42:22 +01:00
    544814875e
  • 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
  • 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
  • Fix hook message duplication in TUI 
    Two bugs:
1. createCustomMessage was returning role: 'user' instead of preserving
   the hook message structure (role: 'hookMessage', customType, etc.)
2. rebuildChatFromMessages wasn't clearing the container before rebuilding
    Mario Zechner · 2025-12-30 22:42:21 +01:00
    ecef601d19
  • 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
  • Fix: handle branch_summary in kept messages before compaction
    Mario Zechner · 2025-12-30 22:42:18 +01:00
    754f55e3f6
  • Add CustomMessageEntry rendering infrastructure 
    - Add renderCustomMessage to HookAPI for registering custom renderers
- Add CustomMessageRenderer type and CustomMessageRenderOptions
- Store customMessageRenderers in LoadedHook
- Add getCustomMessageRenderer(customType) to HookRunner
- SessionContext.entries now aligned with messages (same length, corresponding indices)

TUI can now correlate messages with their source entries to identify
custom_message entries and use hook-provided renderers.
    Mario Zechner · 2025-12-30 22:42:18 +01:00
    11a7845ceb
  • Fix CustomMessageEntry content type to match UserMessage 
    content: string | (TextContent | ImageContent)[]

This matches the UserMessage type from pi-ai, so content can be
passed directly to AppMessage without conversion.
    Mario Zechner · 2025-12-30 22:42:18 +01:00
    3ecaaa5937
  • Add CustomMessageEntry for hook-injected messages in LLM context 
    - CustomMessageEntry<T> type with customType, content, display, details
- appendCustomMessageEntry() in SessionManager
- buildSessionContext() includes custom_message entries as user messages
- Exported CustomEntry and CustomMessageEntry from main index

CustomEntry is for hook state (not in context).
CustomMessageEntry is for hook-injected content (in context).
    Mario Zechner · 2025-12-30 22:42:18 +01:00
    9da36e5ac6
  • 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
  • Improve CustomEntry docs and make it generic 
    - Add detailed doc comment explaining purpose (hook state persistence)
- Make CustomEntry<T = unknown> generic
- Clarify difference from CustomMessageEntry in plan
- Update changelog
    Mario Zechner · 2025-12-30 22:42:18 +01:00
    efb1036d8e
  • Add label support for session entries 
    - Add LabelEntry type with targetId and label (string | undefined)
- Add labelsById map built on load via linear scan
- Add getLabel(id) and appendLabelChange(targetId, label) methods
- Add label field to SessionTreeNode, populated by getTree()
- Update createBranchedSession to preserve labels for entries on path
- Labels are ignored by buildSessionContext (not sent to LLM)
- Add comprehensive tests for label functionality
    Mario Zechner · 2025-12-30 22:42:18 +01:00
    9e68a59fed
  • 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
  • Refactor session manager: migration chain, validation, tests 
    - Add migrateV1ToV2/migrateToCurrentVersion for extensible migrations
- createSummaryMessage now takes timestamp from entry
- loadEntriesFromFile validates session header
- findMostRecentSession only returns valid session files (reads first 512 bytes)
- Remove ConversationEntry alias
- Fix mom context.ts TreeNode type

Tests:
- migration.test.ts: v1 migration, idempotency
- build-context.test.ts: 14 tests covering trivial, compaction, branches
- file-operations.test.ts: loadEntriesFromFile, findMostRecentSession
    Mario Zechner · 2025-12-30 22:42:18 +01:00
    beb70f126d
  • Use short 8-char IDs for session entries 
    - Replace custom uuidv4() with native crypto.randomUUID()
- Entry IDs use first 8 hex chars with collision checking
- Session IDs stay full UUIDs (used in filenames)
- ~0.01 collisions per 10k entries, retry handles it
    Mario Zechner · 2025-12-30 22:42:17 +01:00
    95312e00bb
  • Fix --session flag to use provided filename 
    When --session path was provided for a non-existent file, _initNewSession() was overwriting the path with an auto-generated one. Now it only generates a filename if sessionFile wasn't already set.
    Mario Zechner · 2025-12-30 22:42:17 +01:00
    77595b97f9
  • 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
  • feat(coding-agent): Add --session-dir flag for custom session directory 
    - Add --session-dir CLI flag to specify custom session directory
- SessionManager API: second param of create(), continueRecent(), list(), open()
  changed from agentDir to sessionDir (direct directory, no cwd encoding)
- When omitted, uses default (~/.pi/agent/sessions/<encoded-cwd>/)
- --session now derives sessionDir from file's parent if --session-dir not provided
- list() validates session header before processing files
- Closes #313

Co-authored-by: scutifer <scutifer@users.noreply.github.com>
    Mario Zechner · 2025-12-25 20:27:41 +01:00
    911963e777
  • Improve compaction hooks: add signal, no timeout, SessionManager cleanup, docs
    Mario Zechner · 2025-12-24 13:54:05 +01:00
    705ba5d4f2
  • Fix session persistence: flush all buffered entries on first assistant message 
    - SessionManager: add flushed flag, _persist() flushes all entries when first assistant seen
- SessionManager: reset() now creates session header
- MomSessionManager: same pattern, remove pendingEntries/sessionInitialized/startSession/shouldInitializeSession
    Mario Zechner · 2025-12-22 02:51:56 +01:00
    184c648334
  • Remove unused AgentState import from session-manager.ts
    Mario Zechner · 2025-12-22 02:43:56 +01:00
    974c8f57e5
  • Fix session-manager simplification issues 
    - Remove unused inspector import from session-manager.ts
- Remove dead code in _persist()
- Update tests for simplified SessionHeader
- Update mom context.ts: remove unused AgentState import, fix startSession(), rename isEnabled to isPersisted
    Mario Zechner · 2025-12-22 02:43:38 +01:00
    0faadfcd00
  • Add agentDir parameter to SessionManager factory methods
    Mario Zechner · 2025-12-22 01:31:20 +01:00
    6d4ff74430
  • Refactor SessionManager to use static factory methods 
    - Add factory methods: create(cwd), open(path), continueRecent(cwd), inMemory()
- Add static list(cwd) for session listing
- Make constructor private, pass cwd explicitly
- Update SDK to take sessionManager instead of sessionFile options
- Update main.ts to create SessionManager based on CLI flags
- Update SessionSelectorComponent to take sessions[] instead of SessionManager
- Update tests to use factory methods
    Mario Zechner · 2025-12-22 01:29:54 +01:00
    ace8ea3d5b
  • Enable more biome lints and fix things
    Mario Zechner · 2025-12-21 22:56:20 +01:00
    d5fd685901
  • Release v0.22.5
    Mario Zechner · 2025-12-17 01:22:13 +01:00
    295f51b53f
  • Fix branch selector for single message and --no-session mode 
    - Allow branch selector to open with single user message (changed <= 1 to === 0 check)
- Support in-memory branching for --no-session mode (no files created)
- Add isEnabled() getter to SessionManager
- Update sessionFile getter to return null when sessions disabled
- Update SessionSwitchEvent types to allow null session files
- Add branching tests for single message and --no-session scenarios

fixes #163
    Mario Zechner · 2025-12-10 22:41:32 +01:00
    3d35e7c469
  • Simplify compaction: remove proactive abort, use Agent.continue() for retry 
    - Add agentLoopContinue() to pi-ai for resuming from existing context
- Add Agent.continue() method and transport.continue() interface
- Simplify AgentSession compaction to two cases: overflow (auto-retry) and threshold (no retry)
- Remove proactive mid-turn compaction abort
- Merge turn prefix summary into main summary
- Add isCompacting property to AgentSession and RPC state
- Block input during compaction in interactive mode
- Show compaction count on session resume
- Rename RPC.md to rpc.md for consistency

Related to #128
    Mario Zechner · 2025-12-09 21:43:49 +01:00
    5a9d844f9a
  • feat(coding-agent): implement new compaction system with overflow recovery 
    Phase 1: Updated compaction.ts
- findCutPoint now returns CutPointResult with isSplitTurn and turnStartIndex
- Can cut at user, assistant, or bashExecution messages (never tool results)
- Added turnPrefixSummary support for split turns (parallel summarization)
- estimateTokens helper for context size estimation

Phase 2: Updated session-manager.ts
- CompactionEntry now has optional turnPrefixSummary field
- loadSessionFromEntries injects both summaries when turn was split

Phase 3: Updated agent-session.ts
- Overflow detection via isContextOverflow after agent_end
- Proactive compaction check on turn_end before next LLM call
- _abortingForCompaction flag to skip saving aborted messages
- Auto-retry after overflow recovery or proactive compaction
- New event fields: reason (overflow/threshold), willRetry

Phase 4: Updated interactive-mode.ts
- Shows reason in compaction status (Context overflow detected...)
- Shows retry status after compaction

Tests updated for new CutPointResult return type.
    Mario Zechner · 2025-12-09 17:18:53 +01:00
    a38e619095
  • feat(coding-agent): maintain in-memory session entries for --no-session compaction support 
    SessionManager now tracks all entries in memory regardless of whether file
persistence is enabled. This allows compaction to work in --no-session mode
where no session file is created.

- Added inMemoryEntries array to store session entries
- All save methods now push to inMemoryEntries
- loadEntries() returns in-memory entries when file persistence is disabled
- File persistence only writes when enabled flag is true
- Session loading (constructor, setSessionFile) populates in-memory entries
    Mario Zechner · 2025-12-09 16:55:42 +01:00
    8226975080
  • Move config.ts from utils/ to src/
    Mario Zechner · 2025-12-09 01:28:06 +01:00
    14d99b5f86
  • Reorganize file structure: core/, utils/, modes/interactive/components/, modes/interactive/theme/
    Mario Zechner · 2025-12-09 00:51:33 +01:00
    83a6c26969