claude-baton 2.1.3 → 2.2.0

package/README.md

@@ -24,7 +24,7 @@ You're deep in a Claude Code session — 45 minutes in, multiple files changed,
 
 ## ✨ The Solution
 
-claude-baton gives Claude Code **persistent memory across sessions**. It automatically saves what was built, what's next, and why — then restores it perfectly when you come back.
+claude-baton gives Claude Code **persistent memory across sessions**. One command saves what was built, what's next, and why — one command brings it back. Context compaction? A PreCompact hook auto-saves before anything is lost.
 
 ```
 Session 1                          Session 2
@@ -41,11 +41,12 @@ Session 1                          Session 2
 
 ## 🎯 What You Get
 
-- **🔄 Seamless resume** — Start any session with `/memo-resume` and pick up exactly where you left off
+- **🔄 One-command resume** — Start any session with `/memo-resume` to restore what was built, decisions, blockers, and next steps
 - **💾 Auto-checkpoint** — PreCompact hook saves your session state *before* Claude compacts context. You never have to remember.
 - **📊 Diff intelligence** — Resume shows what changed since your last session (new commits, modified files, dependency changes)
 - **🧠 Decision memory** — Key decisions and their reasoning survive across sessions
 - **📋 EOD summaries** — Generate end-of-day reports from all sessions with `/memo-eod`
+- **💰 Cost transparency** — EOD shows daily LLM calls; `claude-baton status` shows all-time usage and DB size
 - **🔒 Fully local** — All data in a local SQLite database. No cloud, no API keys, no data leaves your machine.
 
 ## 🏗️ Claude Code vs Claude Code + Baton
@@ -202,7 +203,7 @@ fix: MCP server registration via claude mcp add
 
 ```bash
 claude-baton setup               # 🔧 one-time setup (MCP, hooks, commands)
-claude-baton status              # 📊 checkpoint counts, db size
+claude-baton status              # 📊 checkpoint counts, LLM calls, db size
 claude-baton projects            # 📁 list tracked projects
 claude-baton export [--project]  # 📤 export as JSON
 claude-baton import <file>       # 📥 import from JSON
@@ -268,7 +269,7 @@ git clone https://github.com/bakabaka91/claude-baton.git
 cd claude-baton
 npm install
 npm run build
-npm test            # 97 tests
+npm test            # 98 tests
 ```
 
 ## 🗑️ Uninstall

package/commands/memo-checkpoint.md

@@ -1,5 +1,7 @@
 Save session state before context loss.
 
+**Important: Run all git commands exactly as written below — do NOT add -C flags or path arguments. The working directory is already the project root.**
+
 ## Steps
 
 1. Detect the project from the current working directory.
@@ -8,7 +10,6 @@ Save session state before context loss.
    - `git branch --show-current`
    - `git status --short`
    - `git diff --name-only HEAD`
-   - `git log --since="2 hours ago" --format="%h %s"`
    - `git log --oneline -10` (this becomes the git_snapshot)
 
 3. Check if any plan documents exist (PLAN.md, docs/*.md specs, roadmaps). If one is being actively worked on, note the file path and the current phase/step (e.g. "docs/v2-plan.md Phase 2 Step 3").
@@ -20,6 +21,11 @@ Save session state before context loss.
    - **decisions**: key choices made and WHY
    - **blockers**: anything blocking progress
    - **plan_reference**: file path and section of the active plan document (e.g. "docs/v2-plan.md Phase 2 Step 3"), or omit if no plan is active
+   - **learnings**: review the conversation for corrections, failures, and wrong assumptions. For each, write a short actionable statement. Examples:
+     - "User corrected: always run tests before committing"
+     - "Assumed API returns array but it returns object"
+     - "Build failed because forgot to update imports after rename"
+     If nothing went wrong this session, use an empty array.
 
 5. Call the `save_checkpoint` MCP tool with all fields:
    - what_was_built
@@ -31,6 +37,7 @@ Save session state before context loss.
    - branch (from step 2)
    - uncommitted_files (array of lines from git status --short)
    - git_snapshot (from git log --oneline -10)
+   - learnings (array of strings from step 4)
 
 6. Print structured confirmation:
 
@@ -41,4 +48,7 @@ Save session state before context loss.
 📍 State: [current_state summary]
 🎯 Next: [next_steps]
 📝 Uncommitted: [count] files
+
+📜 Commits this session:
+[List commits made during THIS conversation from your context — or "No commits this session" if none were made]
 ```

package/commands/memo-eod.md

@@ -1,15 +1,17 @@
 End-of-day summary combining git activity with stored checkpoints.
 
+**Important: Run all git commands exactly as written below — do NOT add -C flags or path arguments. The working directory is already the project root.**
+
 ## Steps
 
 1. Detect the project from the current working directory.
 
-2. Collect git activity for today by running:
-   - `git log --since="$(date '+%Y-%m-%d') 00:00:00" --until="$(date '+%Y-%m-%d') 23:59:59" --format="%h|||%s|||%ai|||%an" --all`
+2. Collect git activity for today by running (replace YYYY-MM-DD with today's actual date as a literal string — do NOT use $() command substitution):
+   - `git log --since="YYYY-MM-DD 00:00:00" --until="YYYY-MM-DD 23:59:59" --format="%h|||%s|||%ai|||%an" --all`
    - Parse and group commits by conventional commit prefix (feat/fix/chore/refactor/test/docs).
 
 3. Review changed files:
-   - `git diff --name-only HEAD~10 HEAD 2>/dev/null | head -30`
+   - `git diff --name-only HEAD~10 HEAD`
 
 4. Call the `daily_summary` MCP tool (defaults to today).
    This internally gathers checkpoints and sends them to Sonnet for synthesis, then stores the result.
@@ -37,6 +39,10 @@ End-of-day summary combining git activity with stored checkpoints.
 - 📝 Commits today: [count]
 - 📁 Files changed: [count]
 - 🏷️ Feature areas: [list of commit prefixes]
+
+### 💰 Baton Usage
+- LLM calls today: [usage.llm_calls_today from daily_summary response] ([N] auto-checkpoints + 1 EOD)
+- Database: [usage.db_size from daily_summary response]
 ```
 
 7. Confirm: "✅ EOD saved for [Project Name] — [DATE]. [N] commits today."

package/commands/memo-resume.md

@@ -1,5 +1,7 @@
 Restore context from last checkpoint at session start.
 
+**Important: Run all git commands exactly as written below — do NOT add -C flags or path arguments. The working directory is already the project root.**
+
 ## Steps
 
 1. Detect the project from the current working directory.

package/commands/memo-retro.md

@@ -0,0 +1,84 @@
+Analyze recent checkpoints to surface learnings and suggest memory updates.
+
+Run periodically (weekly or after major milestones) to close the learning loop.
+
+## Steps
+
+1. Detect the project from the current working directory. Construct the project slug by replacing all `/` with `-` in the full path.
+
+2. Call the `list_checkpoints` MCP tool with `limit: 10` (no date) to fetch the 10 most recent checkpoints.
+
+3. Load current memory state:
+   - Read all files in `~/.claude/projects/[PROJECT_SLUG]/memory/`
+   - Read `MEMORY.md` in the memory directory
+   - Read the project's `CLAUDE.md` (in the working directory root)
+
+4. Check git history for correction patterns:
+   ```bash
+   git log --oneline -30
+   ```
+   Look for reverts, quick-follow fix commits (e.g., "fix: ..." immediately after a feature commit), and patterns indicating repeated mistakes.
+
+5. Analyze the checkpoint data:
+
+   **Extract learnings:**
+   - Pull the `learnings` array from each checkpoint (may be empty or absent on older checkpoints)
+   - Also extract signals from `decisions_made`, `blockers`, and `what_was_built`
+
+   **Frequency analysis:**
+   - Count how many times each learning (or semantically similar learning) appears
+   - Single occurrence = note only
+   - 2+ occurrences = high confidence, propose as rule
+
+   **Cross-reference:**
+   - Is this learning already captured in a memory file? → skip
+   - Is this learning already a CLAUDE.md rule? → skip
+   - Is this a correction/preference? → propose as feedback memory
+   - Is this about the project? → propose as project memory
+
+6. Present the report:
+
+```
+## 🔄 Retro — [PROJECT_NAME] — [DATE]
+
+### 🧠 Memory Health
+
+- 📁 Total memories: [count of files in memory dir]
+- 📏 MEMORY.md line count: [N]/200 (⚠️ warn if >150)
+- 🪦 Stale memories: [any referencing completed/outdated work]
+- 🔀 Duplicate/overlapping memories: [any found]
+
+### 📚 Learnings Summary
+
+**🔁 Recurring (2+ times) — propose as rules:**
+
+- [learning] — appeared in [N] checkpoints
+- ...
+
+**📌 Single occurrence — note only:**
+
+- [learning]
+- ...
+
+### 🔍 Git Correction Patterns
+
+[Reverts, quick-follow fixes that indicate recurring mistakes, or "✅ None detected"]
+
+### 🛠️ Proposed Changes
+
+**📝 Memory file actions:**
+
+- ➕ **Create:** [new memory file if a pattern warrants it, with suggested filename and content]
+- ✏️ **Update:** [existing memory with outdated info]
+- 🔗 **Merge:** [overlapping memories to consolidate]
+- 🗑️ **Prune:** [stale memories to delete]
+
+**📐 CLAUDE.md rules** (only for recurring learnings not already captured):
+
+1. Add to section [X]: "[exact rule wording]"
+   _Reason: appeared in [N] checkpoints, e.g., [quote from learning]_
+```
+
+7. Ask: **"Which changes should I apply? (all / list numbers / none)"**
+
+   Do NOT modify CLAUDE.md, memory files, or any project files without explicit approval.

package/dist/cli.js

@@ -6,7 +6,7 @@ import path from "path";
 import os from "os";
 import { createInterface } from "readline";
 import { initDatabase, getDefaultDbPath, saveDatabase, countAll, listProjects, insertCheckpoint, getLatestCheckpoint, getAllCheckpoints, getAllDailySummaries, deleteProjectData, deleteAllData, } from "./store.js";
-import { ensureDir, normalizeProjectPath } from "./utils.js";
+import { ensureDir, formatSize, normalizeProjectPath } from "./utils.js";
 import { callClaudeJson } from "./llm.js";
 const __dirname = path.dirname(fileURLToPath(import.meta.url));
 function readPromptTemplate(name) {
@@ -176,7 +176,8 @@ export async function handleSetup() {
         console.error("  Registered PreCompact hook");
     }
     // Register allowed tools for frictionless slash commands (idempotent)
-    const BATON_TOOLS = [
+    // Bash patterns use legacy allowedTools (still works for Bash tools)
+    const BATON_BASH_TOOLS = [
         "Bash(git status*)",
         "Bash(git log*)",
         "Bash(git diff*)",
@@ -185,7 +186,7 @@ export async function handleSetup() {
     ];
     const allowedTools = (settings.allowedTools ?? []);
     let toolsAdded = 0;
-    for (const tool of BATON_TOOLS) {
+    for (const tool of BATON_BASH_TOOLS) {
         if (!allowedTools.includes(tool)) {
             allowedTools.push(tool);
             toolsAdded++;
@@ -193,7 +194,29 @@ export async function handleSetup() {
     }
     if (toolsAdded > 0) {
         settings.allowedTools = allowedTools;
-        console.error(`  Registered ${toolsAdded} allowed tools`);
+    }
+    // MCP tools use permissions.allow (required for MCP tool auto-approval)
+    const BATON_MCP_TOOLS = [
+        "mcp__claude-baton__save_checkpoint",
+        "mcp__claude-baton__get_checkpoint",
+        "mcp__claude-baton__list_checkpoints",
+        "mcp__claude-baton__daily_summary",
+    ];
+    const permissions = (settings.permissions ?? {});
+    const allowList = (permissions.allow ?? []);
+    let mcpToolsAdded = 0;
+    for (const tool of BATON_MCP_TOOLS) {
+        if (!allowList.includes(tool)) {
+            allowList.push(tool);
+            mcpToolsAdded++;
+        }
+    }
+    if (mcpToolsAdded > 0) {
+        permissions.allow = allowList;
+        settings.permissions = permissions;
+    }
+    if (toolsAdded + mcpToolsAdded > 0) {
+        console.error(`  Registered ${toolsAdded + mcpToolsAdded} allowed tools`);
     }
     writeFileSync(settingsPath, JSON.stringify(settings, null, 2) + "\n");
     const dbPath = getDefaultDbPath();
@@ -289,21 +312,53 @@ export async function handleUninstall(opts) {
                 }
                 console.error("  Removed PreCompact hook");
             }
-            // Remove allowed tools
+            // Remove allowed tools (bash patterns)
             if (Array.isArray(settings.allowedTools)) {
-                const BATON_PATTERNS = [
+                const BATON_BASH_PATTERNS = [
                     "Bash(git status*)",
                     "Bash(git log*)",
                     "Bash(git diff*)",
                     "Bash(git branch*)",
+                    "Bash(git -C *status*)",
+                    "Bash(git -C *log*)",
+                    "Bash(git -C *diff*)",
+                    "Bash(git -C *branch*)",
                     "Bash(node *claude-baton*)",
                 ];
-                settings.allowedTools = settings.allowedTools.filter((t) => !BATON_PATTERNS.includes(t));
+                // Note: git -C patterns kept in uninstall to clean up from older installs
+                settings.allowedTools = settings.allowedTools.filter((t) => !BATON_BASH_PATTERNS.includes(t));
                 if (settings.allowedTools.length === 0) {
                     delete settings.allowedTools;
                 }
                 console.error("  Removed allowed tools");
             }
+            // Remove MCP tool permissions
+            if (settings.permissions &&
+                typeof settings.permissions === "object" &&
+                Array.isArray(settings.permissions.allow)) {
+                const perms = settings.permissions;
+                const BATON_MCP_PATTERNS = [
+                    "mcp__claude-baton__save_checkpoint",
+                    "mcp__claude-baton__get_checkpoint",
+                    "mcp__claude-baton__list_checkpoints",
+                    "mcp__claude-baton__daily_summary",
+                ];
+                perms.allow = perms.allow.filter((t) => !BATON_MCP_PATTERNS.includes(t));
+                if (perms.allow.length === 0) {
+                    delete perms.allow;
+                }
+                if (Object.keys(perms).length === 0) {
+                    delete settings.permissions;
+                }
+                console.error("  Removed MCP tool permissions");
+            }
+            // Clean up legacy MCP entries from allowedTools (from older versions)
+            if (Array.isArray(settings.allowedTools)) {
+                settings.allowedTools = settings.allowedTools.filter((t) => !t.startsWith("mcp__claude-baton__"));
+                if (settings.allowedTools.length === 0) {
+                    delete settings.allowedTools;
+                }
+            }
             writeFileSync(settingsPath, JSON.stringify(settings, null, 2) + "\n");
         }
         catch {
@@ -351,13 +406,15 @@ export async function handleStatus(opts) {
     const projectPath = opts.project ?? normalizeProjectPath(process.cwd());
     const counts = countAll(db, projectPath);
     const dbSize = statSync(dbPath).size;
+    const llmCalls = counts.auto_checkpoints + counts.daily_summaries;
     console.log(`Project: ${projectPath}`);
-    console.log(`Database: ${dbPath} (${(dbSize / 1024).toFixed(1)} KB)`);
+    console.log(`Database: ${dbPath} (${formatSize(dbSize)})`);
     console.log();
     console.log("Counts:");
-    for (const [key, value] of Object.entries(counts)) {
-        console.log(`  ${key}: ${value}`);
-    }
+    console.log(`  checkpoints: ${counts.checkpoints} (${counts.checkpoints - counts.auto_checkpoints} manual, ${counts.auto_checkpoints} auto)`);
+    console.log(`  daily_summaries: ${counts.daily_summaries}`);
+    console.log();
+    console.log(`LLM calls (claude -p): ${llmCalls} (${counts.auto_checkpoints} auto-checkpoints + ${counts.daily_summaries} EOD summaries)`);
 }
 // --- Projects command ---
 export async function handleProjects() {

package/dist/index.js

@@ -1,9 +1,9 @@
 import { Server } from "@modelcontextprotocol/sdk/server/index.js";
 import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
 import { CallToolRequestSchema, ListToolsRequestSchema, } from "@modelcontextprotocol/sdk/types.js";
-import { initDatabase, getDefaultDbPath, insertCheckpoint, getLatestCheckpoint, getCheckpoint, getCheckpointsByDate, insertDailySummary, } from "./store.js";
+import { initDatabase, getDefaultDbPath, insertCheckpoint, getLatestCheckpoint, getCheckpoint, getCheckpointsByDate, getRecentCheckpoints, insertDailySummary, countAll, } from "./store.js";
 import { callClaudeJson } from "./llm.js";
-import { normalizeProjectPath } from "./utils.js";
+import { formatSize, normalizeProjectPath } from "./utils.js";
 import { readFileSync, statSync } from "fs";
 import path from "path";
 import { fileURLToPath } from "url";
@@ -76,6 +76,11 @@ server.setRequestHandler(ListToolsRequestSchema, async () => ({
                         type: "string",
                         description: "Reference to active plan document and section, e.g. 'docs/plan.md Phase 2 Step 3'",
                     },
+                    learnings: {
+                        type: "array",
+                        items: { type: "string" },
+                        description: "Actionable learnings from the session — corrections, wrong assumptions, failed approaches",
+                    },
                     source: {
                         type: "string",
                         enum: ["manual", "auto"],
@@ -108,13 +113,17 @@ server.setRequestHandler(ListToolsRequestSchema, async () => ({
         },
         {
             name: "list_checkpoints",
-            description: "List all checkpoints for a date",
+            description: "List checkpoints by date or most recent",
             inputSchema: {
                 type: "object",
                 properties: {
                     date: {
                         type: "string",
-                        description: "YYYY-MM-DD (defaults to today)",
+                        description: "YYYY-MM-DD — returns checkpoints for this date. If omitted, returns most recent across all dates.",
+                    },
+                    limit: {
+                        type: "number",
+                        description: "Max checkpoints to return (default 10). Used when date is omitted.",
                     },
                     project: {
                         type: "string",
@@ -161,6 +170,7 @@ server.setRequestHandler(CallToolRequestSchema, async (request) => {
                     uncommittedFiles: a?.uncommitted_files,
                     gitSnapshot: a?.git_snapshot,
                     planReference: a?.plan_reference,
+                    learnings: a?.learnings,
                     source: a?.source ?? "manual",
                 }, dbPath);
                 return { content: [{ type: "text", text: `Checkpoint saved: ${id}` }] };
@@ -177,14 +187,19 @@ server.setRequestHandler(CallToolRequestSchema, async (request) => {
                 };
             }
             case "list_checkpoints": {
-                const cpDate = args?.date ?? new Date().toISOString().slice(0, 10);
-                const cps = getCheckpointsByDate(db, projectPath, cpDate);
+                const cpDate = args?.date;
+                const limit = args?.limit ?? 10;
+                const cps = cpDate
+                    ? getCheckpointsByDate(db, projectPath, cpDate)
+                    : getRecentCheckpoints(db, projectPath, limit);
                 const summary = cps.map((cp) => ({
                     id: cp.id,
                     created_at: cp.created_at,
                     what_was_built: cp.what_was_built,
                     branch: cp.branch,
                     current_state: cp.current_state,
+                    decisions_made: cp.decisions_made,
+                    learnings: cp.learnings,
                 }));
                 return {
                     content: [{ type: "text", text: JSON.stringify(summary, null, 2) }],
@@ -209,11 +224,21 @@ server.setRequestHandler(CallToolRequestSchema, async (request) => {
                     .replace("{{ACTIVITY}}", activityParts.join("\n\n"));
                 const summaryResult = await callClaudeJson(summaryPrompt, "sonnet", 60000);
                 insertDailySummary(db, projectPath, date, summaryResult, dbPath);
+                // Attach usage stats for cost transparency
+                const autoCheckpointsToday = checkpoints.filter((cp) => cp.source === "auto").length;
+                const counts = countAll(db, projectPath);
+                const dbSizeBytes = statSync(dbPath).size;
+                const usage = {
+                    llm_calls_today: autoCheckpointsToday + 1, // +1 for this EOD call
+                    auto_checkpoints_today: autoCheckpointsToday,
+                    total_llm_calls: counts.auto_checkpoints + counts.daily_summaries,
+                    db_size: formatSize(dbSizeBytes),
+                };
                 return {
                     content: [
                         {
                             type: "text",
-                            text: JSON.stringify(summaryResult, null, 2),
+                            text: JSON.stringify({ ...summaryResult, usage }, null, 2),
                         },
                     ],
                 };

package/dist/store.d.ts

@@ -12,10 +12,12 @@ export declare function insertCheckpoint(db: Database, projectPath: string, sess
     gitSnapshot?: string;
     planReference?: string;
     source?: "manual" | "auto";
+    learnings?: string[];
 }, dbPath?: string): string;
 export declare function getCheckpoint(db: Database, id: string): Checkpoint | null;
 export declare function getLatestCheckpoint(db: Database, projectPath: string): Checkpoint | null;
 export declare function getCheckpointsByDate(db: Database, projectPath: string, date: string): Checkpoint[];
+export declare function getRecentCheckpoints(db: Database, projectPath: string, limit?: number): Checkpoint[];
 export declare function getAllCheckpoints(db: Database, projectPath?: string): Checkpoint[];
 export declare function insertDailySummary(db: Database, projectPath: string, date: string, summary: Record<string, unknown>, dbPath?: string): string;
 export declare function getDailySummary(db: Database, projectPath: string, date: string): DailySummary | null;

package/dist/store.js

@@ -88,12 +88,21 @@ export function initSchema(db) {
         if (!msg.includes("duplicate column"))
             throw e;
     }
+    // Migration: add learnings column for existing databases
+    try {
+        db.exec("ALTER TABLE checkpoints ADD COLUMN learnings TEXT DEFAULT '[]'");
+    }
+    catch (e) {
+        const msg = e instanceof Error ? e.message : "";
+        if (!msg.includes("duplicate column"))
+            throw e;
+    }
 }
 // --- Checkpoints CRUD ---
 export function insertCheckpoint(db, projectPath, sessionId, currentState, whatWasBuilt, nextSteps, opts, dbPath) {
     const id = crypto.randomUUID();
-    db.run(`INSERT INTO checkpoints (id, project_path, session_id, branch, current_state, what_was_built, next_steps, decisions_made, blockers, uncommitted_files, git_snapshot, plan_reference, source, created_at)
-     VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`, [
+    db.run(`INSERT INTO checkpoints (id, project_path, session_id, branch, current_state, what_was_built, next_steps, decisions_made, blockers, uncommitted_files, git_snapshot, plan_reference, source, learnings, created_at)
+     VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`, [
         id,
         projectPath,
         sessionId,
@@ -107,6 +116,7 @@ export function insertCheckpoint(db, projectPath, sessionId, currentState, whatW
         opts?.gitSnapshot ?? null,
         opts?.planReference ?? null,
         opts?.source ?? "manual",
+        JSON.stringify(opts?.learnings ?? []),
         new Date().toISOString(),
     ]);
     if (dbPath)
@@ -137,6 +147,7 @@ function parseCheckpointRow(row) {
         uncommitted_files: JSON.parse(row.uncommitted_files),
         git_snapshot: row.git_snapshot ?? null,
         plan_reference: row.plan_reference ?? null,
+        learnings: JSON.parse(row.learnings ?? "[]"),
         source: row.source ?? "manual",
     };
 }
@@ -154,6 +165,16 @@ export function getCheckpointsByDate(db, projectPath, date) {
     stmt.free();
     return results;
 }
+export function getRecentCheckpoints(db, projectPath, limit = 10) {
+    const stmt = db.prepare("SELECT * FROM checkpoints WHERE project_path = ? ORDER BY created_at DESC, rowid DESC LIMIT ?");
+    stmt.bind([projectPath, limit]);
+    const results = [];
+    while (stmt.step()) {
+        results.push(parseCheckpointRow(stmt.getAsObject()));
+    }
+    stmt.free();
+    return results;
+}
 export function getAllCheckpoints(db, projectPath) {
     let sql = "SELECT * FROM checkpoints";
     const params = [];
@@ -233,11 +254,13 @@ export function countAll(db, projectPath) {
         const p = [projectPath];
         return {
             checkpoints: countTable(db, "SELECT COUNT(*) as count FROM checkpoints WHERE project_path = ?", p),
+            auto_checkpoints: countTable(db, "SELECT COUNT(*) as count FROM checkpoints WHERE project_path = ? AND source = 'auto'", p),
             daily_summaries: countTable(db, "SELECT COUNT(*) as count FROM daily_summaries WHERE project_path = ?", p),
         };
     }
     return {
         checkpoints: countTable(db, "SELECT COUNT(*) as count FROM checkpoints", []),
+        auto_checkpoints: countTable(db, "SELECT COUNT(*) as count FROM checkpoints WHERE source = 'auto'", []),
         daily_summaries: countTable(db, "SELECT COUNT(*) as count FROM daily_summaries", []),
     };
 }

package/dist/types.d.ts

@@ -12,6 +12,7 @@ export interface Checkpoint {
     uncommitted_files: string[];
     git_snapshot: string | null;
     plan_reference: string | null;
+    learnings: string[];
     source: "manual" | "auto";
     created_at: string;
 }

package/dist/utils.d.ts

@@ -1,3 +1,4 @@
 export declare function getProjectPath(): string;
 export declare function normalizeProjectPath(p: string): string;
 export declare function ensureDir(dirPath: string): void;
+export declare function formatSize(bytes: number): string;

package/dist/utils.js

@@ -21,3 +21,11 @@ export function normalizeProjectPath(p) {
 export function ensureDir(dirPath) {
     mkdirSync(dirPath, { recursive: true });
 }
+// --- Formatting helpers ---
+export function formatSize(bytes) {
+    if (bytes < 1024)
+        return `${bytes} B`;
+    if (bytes < 1024 * 1024)
+        return `${(bytes / 1024).toFixed(1)} KB`;
+    return `${(bytes / (1024 * 1024)).toFixed(1)} MB`;
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "claude-baton",
-  "version": "2.1.3",
+  "version": "2.2.0",
   "description": "Session lifecycle management for Claude Code",
   "type": "module",
   "main": "dist/index.js",

package/prompts/auto_checkpoint.txt

@@ -19,6 +19,7 @@ Analyze these inputs and produce a JSON object with these fields:
 - decisions_made: Key decisions made since the previous checkpoint, if any (1-2 sentences, or "None")
 - blockers: Anything blocking progress, if any (1 sentence, or "None")
 - plan_reference: If the session references an active plan document (e.g. PLAN.md, docs/plan.md, a spec file), include the file path and the specific section being worked on (e.g. "docs/v2-plan.md Phase 2 Step 3"). Null if no plan document is referenced.
+- learnings: Review the transcript for corrections made by the user, failed approaches, wrong assumptions, or mistakes by Claude. For each, write a short actionable learning statement. Examples: "User corrected: always run tests before committing", "Assumed API returns array but it returns object", "Build failed because forgot to update imports after rename". If nothing went wrong, output an empty array.
 
 Rules:
 - Be concise — each field should be 1-3 sentences max
@@ -40,5 +41,6 @@ Example output:
   "next_steps": "Wire up login form to POST /auth/login endpoint. Add token refresh middleware.",
   "decisions_made": "Using bcrypt over argon2 for password hashing due to simpler deployment. Storing refresh tokens in httpOnly cookies.",
   "blockers": "None",
-  "plan_reference": "docs/auth-plan.md Phase 2 Step 1"
+  "plan_reference": "docs/auth-plan.md Phase 2 Step 1",
+  "learnings": ["Assumed bcrypt was already installed but had to add it to dependencies"]
 }