@imdigitalashish/zpi 0.1.1

package/dist/core/memory.js

@@ -0,0 +1,247 @@
+/**
+ * Persistent memory system — procedural, episodic, and semantic memories
+ * stored per-project in .pi/memory/ as JSON files.
+ */
+import { existsSync, mkdirSync, readFileSync, writeFileSync } from "node:fs";
+import { join } from "node:path";
+// ---------------------------------------------------------------------------
+// Storage
+// ---------------------------------------------------------------------------
+export function getMemoryDir(cwd) {
+    return join(cwd, ".pi", "memory");
+}
+function ensureMemoryDir(cwd) {
+    const dir = getMemoryDir(cwd);
+    if (!existsSync(dir)) {
+        mkdirSync(dir, { recursive: true });
+    }
+    return dir;
+}
+export function loadStore(cwd, filename) {
+    const filepath = join(getMemoryDir(cwd), filename);
+    if (!existsSync(filepath)) {
+        return { memories: [] };
+    }
+    try {
+        const raw = readFileSync(filepath, "utf-8");
+        return JSON.parse(raw);
+    }
+    catch {
+        return { memories: [] };
+    }
+}
+export function saveStore(cwd, filename, store) {
+    const dir = ensureMemoryDir(cwd);
+    const filepath = join(dir, filename);
+    writeFileSync(filepath, JSON.stringify(store, null, 2), "utf-8");
+}
+export function generateId(prefix, store) {
+    let max = 0;
+    for (const m of store.memories) {
+        const num = parseInt(m.id.replace(`${prefix}_`, ""), 10);
+        if (!Number.isNaN(num) && num > max)
+            max = num;
+    }
+    return `${prefix}_${String(max + 1).padStart(3, "0")}`;
+}
+export function nowISO() {
+    return new Date().toISOString();
+}
+export function getSessionId() {
+    return `session_${Date.now()}`;
+}
+const STORE_FILES = {
+    procedural: "procedural.json",
+    episodic: "episodic.json",
+    semantic: "semantic.json",
+};
+export function getStoreFile(type) {
+    return STORE_FILES[type];
+}
+// ---------------------------------------------------------------------------
+// Search
+// ---------------------------------------------------------------------------
+export function searchMemories(memories, query, getText) {
+    const lower = query.toLowerCase();
+    const queryWords = lower.split(/\s+/).filter(Boolean);
+    return memories.filter((m) => {
+        const text = getText(m).toLowerCase();
+        const tagText = m.tags.join(" ").toLowerCase();
+        const combined = `${text} ${tagText}`;
+        return queryWords.every((w) => combined.includes(w));
+    });
+}
+// ---------------------------------------------------------------------------
+// Compaction
+// ---------------------------------------------------------------------------
+export function compactEpisodicMemories(memories) {
+    if (memories.length <= 1)
+        return memories;
+    const tagGroups = new Map();
+    for (const m of memories) {
+        let placed = false;
+        for (const [key, group] of tagGroups) {
+            const groupTags = new Set(key.split(","));
+            if (m.tags.some((t) => groupTags.has(t))) {
+                group.push(m);
+                const mergedTags = new Set([...groupTags, ...m.tags]);
+                tagGroups.delete(key);
+                tagGroups.set([...mergedTags].join(","), group);
+                placed = true;
+                break;
+            }
+        }
+        if (!placed) {
+            tagGroups.set(m.tags.join(",") || m.id, [m]);
+        }
+    }
+    const compacted = [];
+    for (const [, group] of tagGroups) {
+        if (group.length <= 1) {
+            compacted.push(...group);
+            continue;
+        }
+        const allDetails = group.flatMap((m) => m.details);
+        const allTags = [...new Set(group.flatMap((m) => m.tags))];
+        const allMistakes = group.flatMap((m) => m.reflection?.mistakes ?? []);
+        const allLessons = group.flatMap((m) => m.reflection?.lessons ?? []);
+        const summaries = group.map((m) => m.summary);
+        compacted.push({
+            id: group[0].id,
+            summary: `Consolidated: ${summaries.join("; ")}`,
+            details: [...new Set(allDetails)],
+            reflection: allMistakes.length > 0 || allLessons.length > 0
+                ? { mistakes: [...new Set(allMistakes)], lessons: [...new Set(allLessons)] }
+                : undefined,
+            tags: allTags,
+            date: group[group.length - 1].date,
+            sourceSession: group[group.length - 1].sourceSession,
+        });
+    }
+    return compacted;
+}
+// ---------------------------------------------------------------------------
+// Display formatters
+// ---------------------------------------------------------------------------
+export function formatProceduralForDisplay(m) {
+    const steps = m.steps.map((s, i) => `  ${i + 1}. ${s}`).join("\n");
+    return `[${m.id}] ${m.name}\n  Trigger: ${m.trigger}\n  Tags: ${m.tags.join(", ")}\n  Steps:\n${steps}\n  Updated: ${m.updated}`;
+}
+export function formatEpisodicForDisplay(m) {
+    let text = `[${m.id}] ${m.summary}\n  Date: ${m.date}\n  Tags: ${m.tags.join(", ")}`;
+    if (m.details.length > 0) {
+        text += `\n  Details:\n${m.details.map((d) => `    - ${d}`).join("\n")}`;
+    }
+    if (m.reflection) {
+        if (m.reflection.mistakes.length > 0) {
+            text += `\n  Mistakes:\n${m.reflection.mistakes.map((x) => `    - ${x}`).join("\n")}`;
+        }
+        if (m.reflection.lessons.length > 0) {
+            text += `\n  Lessons:\n${m.reflection.lessons.map((x) => `    - ${x}`).join("\n")}`;
+        }
+    }
+    return text;
+}
+export function formatSemanticForDisplay(m) {
+    return `[${m.id}] (${m.category}) ${m.text}\n  Tags: ${m.tags.join(", ")}\n  Created: ${m.created}`;
+}
+// ---------------------------------------------------------------------------
+// System prompt section
+// ---------------------------------------------------------------------------
+export function buildMemoryPromptSection(cwd) {
+    const semantic = loadStore(cwd, "semantic.json");
+    const procedural = loadStore(cwd, "procedural.json");
+    const episodic = loadStore(cwd, "episodic.json");
+    if (semantic.memories.length === 0 && procedural.memories.length === 0 && episodic.memories.length === 0) {
+        return `\n\n<memory_system>
+You have a persistent memory system that stores knowledge across sessions.
+Currently no memories are stored. Use the memory_write tool to save:
+1. Procedural memories when the user guides you through a multi-step workflow.
+2. Semantic memories when the user states a preference, rule, correction, or fact.
+3. Episodic memories when a significant task completes.
+</memory_system>`;
+    }
+    const parts = [];
+    parts.push(`\n\n<memory_system>
+You have a persistent memory system that stores knowledge across sessions.
+You MUST use these memories to improve your responses. Do not ask the user
+to repeat things they have already taught you.
+
+AUTOMATIC MEMORY CAPTURE RULES:
+1. When the user guides you through a multi-step workflow (correcting steps,
+   adding steps, reordering), save it as a PROCEDURAL memory using the
+   memory_write tool. Extract the general workflow, not the specific instance.
+2. When the user states a preference or rule ("always do X", "never do Y",
+   "I prefer X"), save it as a SEMANTIC memory immediately.
+3. When the user corrects you or you make a mistake, save the lesson as a
+   SEMANTIC memory with category "convention".
+4. When a significant task completes, save a summary as an EPISODIC memory.
+5. If you detect a conflict with an existing memory, ask the user: "You
+   previously told me [old]. You are now saying [new]. Should I update this?"
+   Only update after confirmation.
+
+TIMESTAMP AWARENESS:
+Each user message includes a timestamp. Use these to detect time gaps.
+If the user returns after a significant gap (>10 minutes), acknowledge it
+naturally if relevant. Do not force it if the gap is not relevant.`);
+    if (semantic.memories.length > 0) {
+        parts.push("\n<semantic_memories>");
+        for (const m of semantic.memories) {
+            parts.push(`  <memory id="${m.id}" category="${m.category}">${escapeXml(m.text)}</memory>`);
+        }
+        parts.push("</semantic_memories>");
+    }
+    if (procedural.memories.length > 0) {
+        parts.push("\n<procedural_memories>");
+        for (const m of procedural.memories) {
+            const steps = m.steps.map((s, i) => `${i + 1}. ${s}`).join("; ");
+            parts.push(`  <procedure id="${m.id}" name="${escapeXml(m.name)}" trigger="${escapeXml(m.trigger)}">${escapeXml(steps)}</procedure>`);
+        }
+        parts.push("</procedural_memories>");
+    }
+    if (episodic.memories.length > 0) {
+        const recent = episodic.memories.slice(-10);
+        parts.push("\n<recent_episodic_memories>");
+        for (const m of recent) {
+            const details = m.details.join("; ");
+            let text = details;
+            if (m.reflection?.lessons.length) {
+                text += ` | Lessons: ${m.reflection.lessons.join("; ")}`;
+            }
+            parts.push(`  <episode id="${m.id}" date="${m.date}" summary="${escapeXml(m.summary)}">${escapeXml(text)}</episode>`);
+        }
+        parts.push("</recent_episodic_memories>");
+    }
+    parts.push("\n</memory_system>");
+    return parts.join("\n");
+}
+function escapeXml(str) {
+    return str
+        .replace(/&/g, "&amp;")
+        .replace(/</g, "&lt;")
+        .replace(/>/g, "&gt;")
+        .replace(/"/g, "&quot;")
+        .replace(/'/g, "&apos;");
+}
+// ---------------------------------------------------------------------------
+// Timestamp formatting
+// ---------------------------------------------------------------------------
+export function formatTimestamp(ts) {
+    const d = new Date(ts);
+    return d.toLocaleString("en-US", {
+        year: "numeric",
+        month: "short",
+        day: "numeric",
+        hour: "2-digit",
+        minute: "2-digit",
+        second: "2-digit",
+        hour12: false,
+    });
+}
+export function getMemoryCounts(cwd) {
+    const semantic = loadStore(cwd, "semantic.json").memories.length;
+    const procedural = loadStore(cwd, "procedural.json").memories.length;
+    const episodic = loadStore(cwd, "episodic.json").memories.length;
+    return { semantic, procedural, episodic, total: semantic + procedural + episodic };
+}
+//# sourceMappingURL=memory.js.map

package/dist/core/memory.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"memory.js","sourceRoot":"","sources":["../../src/core/memory.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,EAAE,UAAU,EAAE,SAAS,EAAE,YAAY,EAAE,aAAa,EAAE,MAAM,SAAS,CAAC;AAC7E,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AA6CjC,8EAA8E;AAC9E,UAAU;AACV,8EAA8E;AAE9E,MAAM,UAAU,YAAY,CAAC,GAAW,EAAU;IACjD,OAAO,IAAI,CAAC,GAAG,EAAE,KAAK,EAAE,QAAQ,CAAC,CAAC;AAAA,CAClC;AAED,SAAS,eAAe,CAAC,GAAW,EAAU;IAC7C,MAAM,GAAG,GAAG,YAAY,CAAC,GAAG,CAAC,CAAC;IAC9B,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;QACtB,SAAS,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IACrC,CAAC;IACD,OAAO,GAAG,CAAC;AAAA,CACX;AAED,MAAM,UAAU,SAAS,CAAI,GAAW,EAAE,QAAgB,EAAkB;IAC3E,MAAM,QAAQ,GAAG,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,EAAE,QAAQ,CAAC,CAAC;IACnD,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;QAC3B,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC;IACzB,CAAC;IACD,IAAI,CAAC;QACJ,MAAM,GAAG,GAAG,YAAY,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;QAC5C,OAAO,IAAI,CAAC,KAAK,CAAC,GAAG,CAAmB,CAAC;IAC1C,CAAC;IAAC,MAAM,CAAC;QACR,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC;IACzB,CAAC;AAAA,CACD;AAED,MAAM,UAAU,SAAS,CAAI,GAAW,EAAE,QAAgB,EAAE,KAAqB,EAAQ;IACxF,MAAM,GAAG,GAAG,eAAe,CAAC,GAAG,CAAC,CAAC;IACjC,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,QAAQ,CAAC,CAAC;IACrC,aAAa,CAAC,QAAQ,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC;AAAA,CACjE;AAED,MAAM,UAAU,UAAU,CAAC,MAAc,EAAE,KAAkC,EAAU;IACtF,IAAI,GAAG,GAAG,CAAC,CAAC;IACZ,KAAK,MAAM,CAAC,IAAI,KAAK,CAAC,QAAQ,EAAE,CAAC;QAChC,MAAM,GAAG,GAAG,QAAQ,CAAC,CAAC,CAAC,EAAE,CAAC,OAAO,CAAC,GAAG,MAAM,GAAG,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;QACzD,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,GAAG,GAAG,GAAG;YAAE,GAAG,GAAG,GAAG,CAAC;IAChD,CAAC;IACD,OAAO,GAAG,MAAM,IAAI,MAAM,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,CAAC;AAAA,CACvD;AAED,MAAM,UAAU,MAAM,GAAW;IAChC,OAAO,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;AAAA,CAChC;AAED,MAAM,UAAU,YAAY,GAAW;IACtC,OAAO,WAAW,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC;AAAA,CAC/B;AAED,MAAM,WAAW,GAA+B;IAC/C,UAAU,EAAE,iBAAiB;IAC7B,QAAQ,EAAE,eAAe;IACzB,QAAQ,EAAE,eAAe;CACzB,CAAC;AAEF,MAAM,UAAU,YAAY,CAAC,IAAgB,EAAU;IACtD,OAAO,WAAW,CAAC,IAAI,CAAC,CAAC;AAAA,CACzB;AAED,8EAA8E;AAC9E,SAAS;AACT,8EAA8E;AAE9E,MAAM,UAAU,cAAc,CAC7B,QAAa,EACb,KAAa,EACb,OAAyB,EACnB;IACN,MAAM,KAAK,GAAG,KAAK,CAAC,WAAW,EAAE,CAAC;IAClC,MAAM,UAAU,GAAG,KAAK,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IACtD,OAAO,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC;QAC7B,MAAM,IAAI,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC;QACtC,MAAM,OAAO,GAAG,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,WAAW,EAAE,CAAC;QAC/C,MAAM,QAAQ,GAAG,GAAG,IAAI,IAAI,OAAO,EAAE,CAAC;QACtC,OAAO,UAAU,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC;IAAA,CACrD,CAAC,CAAC;AAAA,CACH;AAED,8EAA8E;AAC9E,aAAa;AACb,8EAA8E;AAE9E,MAAM,UAAU,uBAAuB,CAAC,QAA0B,EAAoB;IACrF,IAAI,QAAQ,CAAC,MAAM,IAAI,CAAC;QAAE,OAAO,QAAQ,CAAC;IAE1C,MAAM,SAAS,GAAG,IAAI,GAAG,EAA4B,CAAC;IACtD,KAAK,MAAM,CAAC,IAAI,QAAQ,EAAE,CAAC;QAC1B,IAAI,MAAM,GAAG,KAAK,CAAC;QACnB,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,SAAS,EAAE,CAAC;YACtC,MAAM,SAAS,GAAG,IAAI,GAAG,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC;YAC1C,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;gBAC1C,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;gBACd,MAAM,UAAU,GAAG,IAAI,GAAG,CAAC,CAAC,GAAG,SAAS,EAAE,GAAG,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;gBACtD,SAAS,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;gBACtB,SAAS,CAAC,GAAG,CAAC,CAAC,GAAG,UAAU,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,KAAK,CAAC,CAAC;gBAChD,MAAM,GAAG,IAAI,CAAC;gBACd,MAAM;YACP,CAAC;QACF,CAAC;QACD,IAAI,CAAC,MAAM,EAAE,CAAC;YACb,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;QAC9C,CAAC;IACF,CAAC;IAED,MAAM,SAAS,GAAqB,EAAE,CAAC;IACvC,KAAK,MAAM,CAAC,EAAE,KAAK,CAAC,IAAI,SAAS,EAAE,CAAC;QACnC,IAAI,KAAK,CAAC,MAAM,IAAI,CAAC,EAAE,CAAC;YACvB,SAAS,CAAC,IAAI,CAAC,GAAG,KAAK,CAAC,CAAC;YACzB,SAAS;QACV,CAAC;QACD,MAAM,UAAU,GAAG,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;QACnD,MAAM,OAAO,GAAG,CAAC,GAAG,IAAI,GAAG,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC3D,MAAM,WAAW,GAAG,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,EAAE,QAAQ,IAAI,EAAE,CAAC,CAAC;QACvE,MAAM,UAAU,GAAG,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,EAAE,OAAO,IAAI,EAAE,CAAC,CAAC;QACrE,MAAM,SAAS,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;QAE9C,SAAS,CAAC,IAAI,CAAC;YACd,EAAE,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE;YACf,OAAO,EAAE,iBAAiB,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE;YAChD,OAAO,EAAE,CAAC,GAAG,IAAI,GAAG,CAAC,UAAU,CAAC,CAAC;YACjC,UAAU,EACT,WAAW,CAAC,MAAM,GAAG,CAAC,IAAI,UAAU,CAAC,MAAM,GAAG,CAAC;gBAC9C,CAAC,CAAC,EAAE,QAAQ,EAAE,CAAC,GAAG,IAAI,GAAG,CAAC,WAAW,CAAC,CAAC,EAAE,OAAO,EAAE,CAAC,GAAG,IAAI,GAAG,CAAC,UAAU,CAAC,CAAC,EAAE;gBAC5E,CAAC,CAAC,SAAS;YACb,IAAI,EAAE,OAAO;YACb,IAAI,EAAE,KAAK,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,IAAI;YAClC,aAAa,EAAE,KAAK,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,aAAa;SACpD,CAAC,CAAC;IACJ,CAAC;IACD,OAAO,SAAS,CAAC;AAAA,CACjB;AAED,8EAA8E;AAC9E,qBAAqB;AACrB,8EAA8E;AAE9E,MAAM,UAAU,0BAA0B,CAAC,CAAmB,EAAU;IACvE,MAAM,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACnE,OAAO,IAAI,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,IAAI,gBAAgB,CAAC,CAAC,OAAO,aAAa,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,eAAe,KAAK,gBAAgB,CAAC,CAAC,OAAO,EAAE,CAAC;AAAA,CACjI;AAED,MAAM,UAAU,wBAAwB,CAAC,CAAiB,EAAU;IACnE,IAAI,IAAI,GAAG,IAAI,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,OAAO,aAAa,CAAC,CAAC,IAAI,aAAa,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;IACrF,IAAI,CAAC,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC1B,IAAI,IAAI,iBAAiB,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;IAC1E,CAAC;IACD,IAAI,CAAC,CAAC,UAAU,EAAE,CAAC;QAClB,IAAI,CAAC,CAAC,UAAU,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACtC,IAAI,IAAI,kBAAkB,CAAC,CAAC,UAAU,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;QACvF,CAAC;QACD,IAAI,CAAC,CAAC,UAAU,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACrC,IAAI,IAAI,iBAAiB,CAAC,CAAC,UAAU,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;QACrF,CAAC;IACF,CAAC;IACD,OAAO,IAAI,CAAC;AAAA,CACZ;AAED,MAAM,UAAU,wBAAwB,CAAC,CAAiB,EAAU;IACnE,OAAO,IAAI,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC,QAAQ,KAAK,CAAC,CAAC,IAAI,aAAa,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC,OAAO,EAAE,CAAC;AAAA,CACpG;AAED,8EAA8E;AAC9E,wBAAwB;AACxB,8EAA8E;AAE9E,MAAM,UAAU,wBAAwB,CAAC,GAAW,EAAU;IAC7D,MAAM,QAAQ,GAAG,SAAS,CAAiB,GAAG,EAAE,eAAe,CAAC,CAAC;IACjE,MAAM,UAAU,GAAG,SAAS,CAAmB,GAAG,EAAE,iBAAiB,CAAC,CAAC;IACvE,MAAM,QAAQ,GAAG,SAAS,CAAiB,GAAG,EAAE,eAAe,CAAC,CAAC;IAEjE,IAAI,QAAQ,CAAC,QAAQ,CAAC,MAAM,KAAK,CAAC,IAAI,UAAU,CAAC,QAAQ,CAAC,MAAM,KAAK,CAAC,IAAI,QAAQ,CAAC,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC1G,OAAO;;;;;;iBAMQ,CAAC;IACjB,CAAC;IAED,MAAM,KAAK,GAAa,EAAE,CAAC;IAE3B,KAAK,CAAC,IAAI,CAAC;;;;;;;;;;;;;;;;;;;;;mEAqBuD,CAAC,CAAC;IAEpE,IAAI,QAAQ,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAClC,KAAK,CAAC,IAAI,CAAC,uBAAuB,CAAC,CAAC;QACpC,KAAK,MAAM,CAAC,IAAI,QAAQ,CAAC,QAAQ,EAAE,CAAC;YACnC,KAAK,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC,EAAE,eAAe,CAAC,CAAC,QAAQ,KAAK,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QAC7F,CAAC;QACD,KAAK,CAAC,IAAI,CAAC,sBAAsB,CAAC,CAAC;IACpC,CAAC;IAED,IAAI,UAAU,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACpC,KAAK,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC;QACtC,KAAK,MAAM,CAAC,IAAI,UAAU,CAAC,QAAQ,EAAE,CAAC;YACrC,MAAM,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACjE,KAAK,CAAC,IAAI,CACT,oBAAoB,CAAC,CAAC,EAAE,WAAW,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,cAAc,SAAS,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,SAAS,CAAC,KAAK,CAAC,cAAc,CACzH,CAAC;QACH,CAAC;QACD,KAAK,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;IACtC,CAAC;IAED,IAAI,QAAQ,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAClC,MAAM,MAAM,GAAG,QAAQ,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC;QAC5C,KAAK,CAAC,IAAI,CAAC,8BAA8B,CAAC,CAAC;QAC3C,KAAK,MAAM,CAAC,IAAI,MAAM,EAAE,CAAC;YACxB,MAAM,OAAO,GAAG,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACrC,IAAI,IAAI,GAAG,OAAO,CAAC;YACnB,IAAI,CAAC,CAAC,UAAU,EAAE,OAAO,CAAC,MAAM,EAAE,CAAC;gBAClC,IAAI,IAAI,eAAe,CAAC,CAAC,UAAU,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;YAC1D,CAAC;YACD,KAAK,CAAC,IAAI,CACT,kBAAkB,CAAC,CAAC,EAAE,WAAW,CAAC,CAAC,IAAI,cAAc,SAAS,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,SAAS,CAAC,IAAI,CAAC,YAAY,CACzG,CAAC;QACH,CAAC;QACD,KAAK,CAAC,IAAI,CAAC,6BAA6B,CAAC,CAAC;IAC3C,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;IAEjC,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAAA,CACxB;AAED,SAAS,SAAS,CAAC,GAAW,EAAU;IACvC,OAAO,GAAG;SACR,OAAO,CAAC,IAAI,EAAE,OAAO,CAAC;SACtB,OAAO,CAAC,IAAI,EAAE,MAAM,CAAC;SACrB,OAAO,CAAC,IAAI,EAAE,MAAM,CAAC;SACrB,OAAO,CAAC,IAAI,EAAE,QAAQ,CAAC;SACvB,OAAO,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;AAAA,CAC1B;AAED,8EAA8E;AAC9E,uBAAuB;AACvB,8EAA8E;AAE9E,MAAM,UAAU,eAAe,CAAC,EAAU,EAAU;IACnD,MAAM,CAAC,GAAG,IAAI,IAAI,CAAC,EAAE,CAAC,CAAC;IACvB,OAAO,CAAC,CAAC,cAAc,CAAC,OAAO,EAAE;QAChC,IAAI,EAAE,SAAS;QACf,KAAK,EAAE,OAAO;QACd,GAAG,EAAE,SAAS;QACd,IAAI,EAAE,SAAS;QACf,MAAM,EAAE,SAAS;QACjB,MAAM,EAAE,SAAS;QACjB,MAAM,EAAE,KAAK;KACb,CAAC,CAAC;AAAA,CACH;AAaD,MAAM,UAAU,eAAe,CAAC,GAAW,EAAgB;IAC1D,MAAM,QAAQ,GAAG,SAAS,CAAiB,GAAG,EAAE,eAAe,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC;IACjF,MAAM,UAAU,GAAG,SAAS,CAAmB,GAAG,EAAE,iBAAiB,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC;IACvF,MAAM,QAAQ,GAAG,SAAS,CAAiB,GAAG,EAAE,eAAe,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC;IACjF,OAAO,EAAE,QAAQ,EAAE,UAAU,EAAE,QAAQ,EAAE,KAAK,EAAE,QAAQ,GAAG,UAAU,GAAG,QAAQ,EAAE,CAAC;AAAA,CACnF","sourcesContent":["/**\r\n * Persistent memory system — procedural, episodic, and semantic memories\r\n * stored per-project in .pi/memory/ as JSON files.\r\n */\r\n\r\nimport { existsSync, mkdirSync, readFileSync, writeFileSync } from \"node:fs\";\r\nimport { join } from \"node:path\";\r\n\r\n// ---------------------------------------------------------------------------\r\n// Types\r\n// ---------------------------------------------------------------------------\r\n\r\nexport interface ProceduralMemory {\r\n\tid: string;\r\n\tname: string;\r\n\ttrigger: string;\r\n\tsteps: string[];\r\n\ttags: string[];\r\n\tcreated: string;\r\n\tupdated: string;\r\n\tsourceSession: string;\r\n}\r\n\r\nexport interface EpisodicMemory {\r\n\tid: string;\r\n\tsummary: string;\r\n\tdetails: string[];\r\n\treflection?: {\r\n\t\tmistakes: string[];\r\n\t\tlessons: string[];\r\n\t};\r\n\ttags: string[];\r\n\tdate: string;\r\n\tsourceSession: string;\r\n}\r\n\r\nexport interface SemanticMemory {\r\n\tid: string;\r\n\tcategory: \"preference\" | \"architecture\" | \"convention\" | \"fact\";\r\n\ttext: string;\r\n\ttags: string[];\r\n\tcreated: string;\r\n\tsourceSession: string;\r\n}\r\n\r\nexport interface MemoryStore<T> {\r\n\tmemories: T[];\r\n}\r\n\r\nexport type MemoryType = \"procedural\" | \"episodic\" | \"semantic\";\r\n\r\n// ---------------------------------------------------------------------------\r\n// Storage\r\n// ---------------------------------------------------------------------------\r\n\r\nexport function getMemoryDir(cwd: string): string {\r\n\treturn join(cwd, \".pi\", \"memory\");\r\n}\r\n\r\nfunction ensureMemoryDir(cwd: string): string {\r\n\tconst dir = getMemoryDir(cwd);\r\n\tif (!existsSync(dir)) {\r\n\t\tmkdirSync(dir, { recursive: true });\r\n\t}\r\n\treturn dir;\r\n}\r\n\r\nexport function loadStore<T>(cwd: string, filename: string): MemoryStore<T> {\r\n\tconst filepath = join(getMemoryDir(cwd), filename);\r\n\tif (!existsSync(filepath)) {\r\n\t\treturn { memories: [] };\r\n\t}\r\n\ttry {\r\n\t\tconst raw = readFileSync(filepath, \"utf-8\");\r\n\t\treturn JSON.parse(raw) as MemoryStore<T>;\r\n\t} catch {\r\n\t\treturn { memories: [] };\r\n\t}\r\n}\r\n\r\nexport function saveStore<T>(cwd: string, filename: string, store: MemoryStore<T>): void {\r\n\tconst dir = ensureMemoryDir(cwd);\r\n\tconst filepath = join(dir, filename);\r\n\twriteFileSync(filepath, JSON.stringify(store, null, 2), \"utf-8\");\r\n}\r\n\r\nexport function generateId(prefix: string, store: MemoryStore<{ id: string }>): string {\r\n\tlet max = 0;\r\n\tfor (const m of store.memories) {\r\n\t\tconst num = parseInt(m.id.replace(`${prefix}_`, \"\"), 10);\r\n\t\tif (!Number.isNaN(num) && num > max) max = num;\r\n\t}\r\n\treturn `${prefix}_${String(max + 1).padStart(3, \"0\")}`;\r\n}\r\n\r\nexport function nowISO(): string {\r\n\treturn new Date().toISOString();\r\n}\r\n\r\nexport function getSessionId(): string {\r\n\treturn `session_${Date.now()}`;\r\n}\r\n\r\nconst STORE_FILES: Record<MemoryType, string> = {\r\n\tprocedural: \"procedural.json\",\r\n\tepisodic: \"episodic.json\",\r\n\tsemantic: \"semantic.json\",\r\n};\r\n\r\nexport function getStoreFile(type: MemoryType): string {\r\n\treturn STORE_FILES[type];\r\n}\r\n\r\n// ---------------------------------------------------------------------------\r\n// Search\r\n// ---------------------------------------------------------------------------\r\n\r\nexport function searchMemories<T extends { tags: string[] }>(\r\n\tmemories: T[],\r\n\tquery: string,\r\n\tgetText: (m: T) => string,\r\n): T[] {\r\n\tconst lower = query.toLowerCase();\r\n\tconst queryWords = lower.split(/\\s+/).filter(Boolean);\r\n\treturn memories.filter((m) => {\r\n\t\tconst text = getText(m).toLowerCase();\r\n\t\tconst tagText = m.tags.join(\" \").toLowerCase();\r\n\t\tconst combined = `${text} ${tagText}`;\r\n\t\treturn queryWords.every((w) => combined.includes(w));\r\n\t});\r\n}\r\n\r\n// ---------------------------------------------------------------------------\r\n// Compaction\r\n// ---------------------------------------------------------------------------\r\n\r\nexport function compactEpisodicMemories(memories: EpisodicMemory[]): EpisodicMemory[] {\r\n\tif (memories.length <= 1) return memories;\r\n\r\n\tconst tagGroups = new Map<string, EpisodicMemory[]>();\r\n\tfor (const m of memories) {\r\n\t\tlet placed = false;\r\n\t\tfor (const [key, group] of tagGroups) {\r\n\t\t\tconst groupTags = new Set(key.split(\",\"));\r\n\t\t\tif (m.tags.some((t) => groupTags.has(t))) {\r\n\t\t\t\tgroup.push(m);\r\n\t\t\t\tconst mergedTags = new Set([...groupTags, ...m.tags]);\r\n\t\t\t\ttagGroups.delete(key);\r\n\t\t\t\ttagGroups.set([...mergedTags].join(\",\"), group);\r\n\t\t\t\tplaced = true;\r\n\t\t\t\tbreak;\r\n\t\t\t}\r\n\t\t}\r\n\t\tif (!placed) {\r\n\t\t\ttagGroups.set(m.tags.join(\",\") || m.id, [m]);\r\n\t\t}\r\n\t}\r\n\r\n\tconst compacted: EpisodicMemory[] = [];\r\n\tfor (const [, group] of tagGroups) {\r\n\t\tif (group.length <= 1) {\r\n\t\t\tcompacted.push(...group);\r\n\t\t\tcontinue;\r\n\t\t}\r\n\t\tconst allDetails = group.flatMap((m) => m.details);\r\n\t\tconst allTags = [...new Set(group.flatMap((m) => m.tags))];\r\n\t\tconst allMistakes = group.flatMap((m) => m.reflection?.mistakes ?? []);\r\n\t\tconst allLessons = group.flatMap((m) => m.reflection?.lessons ?? []);\r\n\t\tconst summaries = group.map((m) => m.summary);\r\n\r\n\t\tcompacted.push({\r\n\t\t\tid: group[0].id,\r\n\t\t\tsummary: `Consolidated: ${summaries.join(\"; \")}`,\r\n\t\t\tdetails: [...new Set(allDetails)],\r\n\t\t\treflection:\r\n\t\t\t\tallMistakes.length > 0 || allLessons.length > 0\r\n\t\t\t\t\t? { mistakes: [...new Set(allMistakes)], lessons: [...new Set(allLessons)] }\r\n\t\t\t\t\t: undefined,\r\n\t\t\ttags: allTags,\r\n\t\t\tdate: group[group.length - 1].date,\r\n\t\t\tsourceSession: group[group.length - 1].sourceSession,\r\n\t\t});\r\n\t}\r\n\treturn compacted;\r\n}\r\n\r\n// ---------------------------------------------------------------------------\r\n// Display formatters\r\n// ---------------------------------------------------------------------------\r\n\r\nexport function formatProceduralForDisplay(m: ProceduralMemory): string {\r\n\tconst steps = m.steps.map((s, i) => `  ${i + 1}. ${s}`).join(\"\\n\");\r\n\treturn `[${m.id}] ${m.name}\\n  Trigger: ${m.trigger}\\n  Tags: ${m.tags.join(\", \")}\\n  Steps:\\n${steps}\\n  Updated: ${m.updated}`;\r\n}\r\n\r\nexport function formatEpisodicForDisplay(m: EpisodicMemory): string {\r\n\tlet text = `[${m.id}] ${m.summary}\\n  Date: ${m.date}\\n  Tags: ${m.tags.join(\", \")}`;\r\n\tif (m.details.length > 0) {\r\n\t\ttext += `\\n  Details:\\n${m.details.map((d) => `    - ${d}`).join(\"\\n\")}`;\r\n\t}\r\n\tif (m.reflection) {\r\n\t\tif (m.reflection.mistakes.length > 0) {\r\n\t\t\ttext += `\\n  Mistakes:\\n${m.reflection.mistakes.map((x) => `    - ${x}`).join(\"\\n\")}`;\r\n\t\t}\r\n\t\tif (m.reflection.lessons.length > 0) {\r\n\t\t\ttext += `\\n  Lessons:\\n${m.reflection.lessons.map((x) => `    - ${x}`).join(\"\\n\")}`;\r\n\t\t}\r\n\t}\r\n\treturn text;\r\n}\r\n\r\nexport function formatSemanticForDisplay(m: SemanticMemory): string {\r\n\treturn `[${m.id}] (${m.category}) ${m.text}\\n  Tags: ${m.tags.join(\", \")}\\n  Created: ${m.created}`;\r\n}\r\n\r\n// ---------------------------------------------------------------------------\r\n// System prompt section\r\n// ---------------------------------------------------------------------------\r\n\r\nexport function buildMemoryPromptSection(cwd: string): string {\r\n\tconst semantic = loadStore<SemanticMemory>(cwd, \"semantic.json\");\r\n\tconst procedural = loadStore<ProceduralMemory>(cwd, \"procedural.json\");\r\n\tconst episodic = loadStore<EpisodicMemory>(cwd, \"episodic.json\");\r\n\r\n\tif (semantic.memories.length === 0 && procedural.memories.length === 0 && episodic.memories.length === 0) {\r\n\t\treturn `\\n\\n<memory_system>\r\nYou have a persistent memory system that stores knowledge across sessions.\r\nCurrently no memories are stored. Use the memory_write tool to save:\r\n1. Procedural memories when the user guides you through a multi-step workflow.\r\n2. Semantic memories when the user states a preference, rule, correction, or fact.\r\n3. Episodic memories when a significant task completes.\r\n</memory_system>`;\r\n\t}\r\n\r\n\tconst parts: string[] = [];\r\n\r\n\tparts.push(`\\n\\n<memory_system>\r\nYou have a persistent memory system that stores knowledge across sessions.\r\nYou MUST use these memories to improve your responses. Do not ask the user\r\nto repeat things they have already taught you.\r\n\r\nAUTOMATIC MEMORY CAPTURE RULES:\r\n1. When the user guides you through a multi-step workflow (correcting steps,\r\n   adding steps, reordering), save it as a PROCEDURAL memory using the\r\n   memory_write tool. Extract the general workflow, not the specific instance.\r\n2. When the user states a preference or rule (\"always do X\", \"never do Y\",\r\n   \"I prefer X\"), save it as a SEMANTIC memory immediately.\r\n3. When the user corrects you or you make a mistake, save the lesson as a\r\n   SEMANTIC memory with category \"convention\".\r\n4. When a significant task completes, save a summary as an EPISODIC memory.\r\n5. If you detect a conflict with an existing memory, ask the user: \"You\r\n   previously told me [old]. You are now saying [new]. Should I update this?\"\r\n   Only update after confirmation.\r\n\r\nTIMESTAMP AWARENESS:\r\nEach user message includes a timestamp. Use these to detect time gaps.\r\nIf the user returns after a significant gap (>10 minutes), acknowledge it\r\nnaturally if relevant. Do not force it if the gap is not relevant.`);\r\n\r\n\tif (semantic.memories.length > 0) {\r\n\t\tparts.push(\"\\n<semantic_memories>\");\r\n\t\tfor (const m of semantic.memories) {\r\n\t\t\tparts.push(`  <memory id=\"${m.id}\" category=\"${m.category}\">${escapeXml(m.text)}</memory>`);\r\n\t\t}\r\n\t\tparts.push(\"</semantic_memories>\");\r\n\t}\r\n\r\n\tif (procedural.memories.length > 0) {\r\n\t\tparts.push(\"\\n<procedural_memories>\");\r\n\t\tfor (const m of procedural.memories) {\r\n\t\t\tconst steps = m.steps.map((s, i) => `${i + 1}. ${s}`).join(\"; \");\r\n\t\t\tparts.push(\r\n\t\t\t\t`  <procedure id=\"${m.id}\" name=\"${escapeXml(m.name)}\" trigger=\"${escapeXml(m.trigger)}\">${escapeXml(steps)}</procedure>`,\r\n\t\t\t);\r\n\t\t}\r\n\t\tparts.push(\"</procedural_memories>\");\r\n\t}\r\n\r\n\tif (episodic.memories.length > 0) {\r\n\t\tconst recent = episodic.memories.slice(-10);\r\n\t\tparts.push(\"\\n<recent_episodic_memories>\");\r\n\t\tfor (const m of recent) {\r\n\t\t\tconst details = m.details.join(\"; \");\r\n\t\t\tlet text = details;\r\n\t\t\tif (m.reflection?.lessons.length) {\r\n\t\t\t\ttext += ` | Lessons: ${m.reflection.lessons.join(\"; \")}`;\r\n\t\t\t}\r\n\t\t\tparts.push(\r\n\t\t\t\t`  <episode id=\"${m.id}\" date=\"${m.date}\" summary=\"${escapeXml(m.summary)}\">${escapeXml(text)}</episode>`,\r\n\t\t\t);\r\n\t\t}\r\n\t\tparts.push(\"</recent_episodic_memories>\");\r\n\t}\r\n\r\n\tparts.push(\"\\n</memory_system>\");\r\n\r\n\treturn parts.join(\"\\n\");\r\n}\r\n\r\nfunction escapeXml(str: string): string {\r\n\treturn str\r\n\t\t.replace(/&/g, \"&amp;\")\r\n\t\t.replace(/</g, \"&lt;\")\r\n\t\t.replace(/>/g, \"&gt;\")\r\n\t\t.replace(/\"/g, \"&quot;\")\r\n\t\t.replace(/'/g, \"&apos;\");\r\n}\r\n\r\n// ---------------------------------------------------------------------------\r\n// Timestamp formatting\r\n// ---------------------------------------------------------------------------\r\n\r\nexport function formatTimestamp(ts: number): string {\r\n\tconst d = new Date(ts);\r\n\treturn d.toLocaleString(\"en-US\", {\r\n\t\tyear: \"numeric\",\r\n\t\tmonth: \"short\",\r\n\t\tday: \"numeric\",\r\n\t\thour: \"2-digit\",\r\n\t\tminute: \"2-digit\",\r\n\t\tsecond: \"2-digit\",\r\n\t\thour12: false,\r\n\t});\r\n}\r\n\r\n// ---------------------------------------------------------------------------\r\n// Memory counts (for status display)\r\n// ---------------------------------------------------------------------------\r\n\r\nexport interface MemoryCounts {\r\n\tsemantic: number;\r\n\tprocedural: number;\r\n\tepisodic: number;\r\n\ttotal: number;\r\n}\r\n\r\nexport function getMemoryCounts(cwd: string): MemoryCounts {\r\n\tconst semantic = loadStore<SemanticMemory>(cwd, \"semantic.json\").memories.length;\r\n\tconst procedural = loadStore<ProceduralMemory>(cwd, \"procedural.json\").memories.length;\r\n\tconst episodic = loadStore<EpisodicMemory>(cwd, \"episodic.json\").memories.length;\r\n\treturn { semantic, procedural, episodic, total: semantic + procedural + episodic };\r\n}\r\n"]}

package/dist/core/messages.d.ts

@@ -0,0 +1,77 @@
+/**
+ * Custom message types and transformers for the coding agent.
+ *
+ * Extends the base AgentMessage type with coding-agent specific message types,
+ * and provides a transformer to convert them to LLM-compatible messages.
+ */
+import type { AgentMessage } from "@mariozechner/pi-agent-core";
+import type { ImageContent, Message, TextContent } from "@mariozechner/pi-ai";
+export declare const COMPACTION_SUMMARY_PREFIX = "The conversation history before this point was compacted into the following summary:\n\n<summary>\n";
+export declare const COMPACTION_SUMMARY_SUFFIX = "\n</summary>";
+export declare const BRANCH_SUMMARY_PREFIX = "The following is a summary of a branch that this conversation came back from:\n\n<summary>\n";
+export declare const BRANCH_SUMMARY_SUFFIX = "</summary>";
+/**
+ * Message type for bash executions via the ! command.
+ */
+export interface BashExecutionMessage {
+    role: "bashExecution";
+    command: string;
+    output: string;
+    exitCode: number | undefined;
+    cancelled: boolean;
+    truncated: boolean;
+    fullOutputPath?: string;
+    timestamp: number;
+    /** If true, this message is excluded from LLM context (!! prefix) */
+    excludeFromContext?: boolean;
+}
+/**
+ * Message type for extension-injected messages via sendMessage().
+ * These are custom messages that extensions can inject into the conversation.
+ */
+export interface CustomMessage<T = unknown> {
+    role: "custom";
+    customType: string;
+    content: string | (TextContent | ImageContent)[];
+    display: boolean;
+    details?: T;
+    timestamp: number;
+}
+export interface BranchSummaryMessage {
+    role: "branchSummary";
+    summary: string;
+    fromId: string;
+    timestamp: number;
+}
+export interface CompactionSummaryMessage {
+    role: "compactionSummary";
+    summary: string;
+    tokensBefore: number;
+    timestamp: number;
+}
+declare module "@mariozechner/pi-agent-core" {
+    interface CustomAgentMessages {
+        bashExecution: BashExecutionMessage;
+        custom: CustomMessage;
+        branchSummary: BranchSummaryMessage;
+        compactionSummary: CompactionSummaryMessage;
+    }
+}
+/**
+ * Convert a BashExecutionMessage to user message text for LLM context.
+ */
+export declare function bashExecutionToText(msg: BashExecutionMessage): string;
+export declare function createBranchSummaryMessage(summary: string, fromId: string, timestamp: string): BranchSummaryMessage;
+export declare function createCompactionSummaryMessage(summary: string, tokensBefore: number, timestamp: string): CompactionSummaryMessage;
+/** Convert CustomMessageEntry to AgentMessage format */
+export declare function createCustomMessage(customType: string, content: string | (TextContent | ImageContent)[], display: boolean, details: unknown | undefined, timestamp: string): CustomMessage;
+/**
+ * Transform AgentMessages (including custom types) to LLM-compatible Messages.
+ *
+ * This is used by:
+ * - Agent's transormToLlm option (for prompt calls and queued messages)
+ * - Compaction's generateSummary (for summarization)
+ * - Custom extensions and tools
+ */
+export declare function convertToLlm(messages: AgentMessage[]): Message[];
+//# sourceMappingURL=messages.d.ts.map

package/dist/core/messages.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"messages.d.ts","sourceRoot":"","sources":["../../src/core/messages.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,6BAA6B,CAAC;AAChE,OAAO,KAAK,EAAE,YAAY,EAAE,OAAO,EAAE,WAAW,EAAE,MAAM,qBAAqB,CAAC;AAG9E,eAAO,MAAM,yBAAyB,wGAGrC,CAAC;AAEF,eAAO,MAAM,yBAAyB,iBAC3B,CAAC;AAEZ,eAAO,MAAM,qBAAqB,iGAGjC,CAAC;AAEF,eAAO,MAAM,qBAAqB,eAAe,CAAC;AAElD;;GAEG;AACH,MAAM,WAAW,oBAAoB;IACpC,IAAI,EAAE,eAAe,CAAC;IACtB,OAAO,EAAE,MAAM,CAAC;IAChB,MAAM,EAAE,MAAM,CAAC;IACf,QAAQ,EAAE,MAAM,GAAG,SAAS,CAAC;IAC7B,SAAS,EAAE,OAAO,CAAC;IACnB,SAAS,EAAE,OAAO,CAAC;IACnB,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,SAAS,EAAE,MAAM,CAAC;IAClB,qEAAqE;IACrE,kBAAkB,CAAC,EAAE,OAAO,CAAC;CAC7B;AAED;;;GAGG;AACH,MAAM,WAAW,aAAa,CAAC,CAAC,GAAG,OAAO;IACzC,IAAI,EAAE,QAAQ,CAAC;IACf,UAAU,EAAE,MAAM,CAAC;IACnB,OAAO,EAAE,MAAM,GAAG,CAAC,WAAW,GAAG,YAAY,CAAC,EAAE,CAAC;IACjD,OAAO,EAAE,OAAO,CAAC;IACjB,OAAO,CAAC,EAAE,CAAC,CAAC;IACZ,SAAS,EAAE,MAAM,CAAC;CAClB;AAED,MAAM,WAAW,oBAAoB;IACpC,IAAI,EAAE,eAAe,CAAC;IACtB,OAAO,EAAE,MAAM,CAAC;IAChB,MAAM,EAAE,MAAM,CAAC;IACf,SAAS,EAAE,MAAM,CAAC;CAClB;AAED,MAAM,WAAW,wBAAwB;IACxC,IAAI,EAAE,mBAAmB,CAAC;IAC1B,OAAO,EAAE,MAAM,CAAC;IAChB,YAAY,EAAE,MAAM,CAAC;IACrB,SAAS,EAAE,MAAM,CAAC;CAClB;AAGD,OAAO,QAAQ,6BAA6B,CAAC,CAAC;IAC7C,UAAU,mBAAmB;QAC5B,aAAa,EAAE,oBAAoB,CAAC;QACpC,MAAM,EAAE,aAAa,CAAC;QACtB,aAAa,EAAE,oBAAoB,CAAC;QACpC,iBAAiB,EAAE,wBAAwB,CAAC;KAC5C;CACD;AAED;;GAEG;AACH,wBAAgB,mBAAmB,CAAC,GAAG,EAAE,oBAAoB,GAAG,MAAM,CAgBrE;AAED,wBAAgB,0BAA0B,CAAC,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,GAAG,oBAAoB,CAOnH;AAED,wBAAgB,8BAA8B,CAC7C,OAAO,EAAE,MAAM,EACf,YAAY,EAAE,MAAM,EACpB,SAAS,EAAE,MAAM,GACf,wBAAwB,CAO1B;AAED,wDAAwD;AACxD,wBAAgB,mBAAmB,CAClC,UAAU,EAAE,MAAM,EAClB,OAAO,EAAE,MAAM,GAAG,CAAC,WAAW,GAAG,YAAY,CAAC,EAAE,EAChD,OAAO,EAAE,OAAO,EAChB,OAAO,EAAE,OAAO,GAAG,SAAS,EAC5B,SAAS,EAAE,MAAM,GACf,aAAa,CASf;AAED;;;;;;;GAOG;AACH,wBAAgB,YAAY,CAAC,QAAQ,EAAE,YAAY,EAAE,GAAG,OAAO,EAAE,CAgDhE","sourcesContent":["/**\r\n * Custom message types and transformers for the coding agent.\r\n *\r\n * Extends the base AgentMessage type with coding-agent specific message types,\r\n * and provides a transformer to convert them to LLM-compatible messages.\r\n */\r\n\r\nimport type { AgentMessage } from \"@mariozechner/pi-agent-core\";\r\nimport type { ImageContent, Message, TextContent } from \"@mariozechner/pi-ai\";\r\nimport { formatTimestamp } from \"./memory.js\";\r\n\r\nexport const COMPACTION_SUMMARY_PREFIX = `The conversation history before this point was compacted into the following summary:\r\n\r\n<summary>\r\n`;\r\n\r\nexport const COMPACTION_SUMMARY_SUFFIX = `\r\n</summary>`;\r\n\r\nexport const BRANCH_SUMMARY_PREFIX = `The following is a summary of a branch that this conversation came back from:\r\n\r\n<summary>\r\n`;\r\n\r\nexport const BRANCH_SUMMARY_SUFFIX = `</summary>`;\r\n\r\n/**\r\n * Message type for bash executions via the ! command.\r\n */\r\nexport interface BashExecutionMessage {\r\n\trole: \"bashExecution\";\r\n\tcommand: string;\r\n\toutput: string;\r\n\texitCode: number | undefined;\r\n\tcancelled: boolean;\r\n\ttruncated: boolean;\r\n\tfullOutputPath?: string;\r\n\ttimestamp: number;\r\n\t/** If true, this message is excluded from LLM context (!! prefix) */\r\n\texcludeFromContext?: boolean;\r\n}\r\n\r\n/**\r\n * Message type for extension-injected messages via sendMessage().\r\n * These are custom messages that extensions can inject into the conversation.\r\n */\r\nexport interface CustomMessage<T = unknown> {\r\n\trole: \"custom\";\r\n\tcustomType: string;\r\n\tcontent: string | (TextContent | ImageContent)[];\r\n\tdisplay: boolean;\r\n\tdetails?: T;\r\n\ttimestamp: number;\r\n}\r\n\r\nexport interface BranchSummaryMessage {\r\n\trole: \"branchSummary\";\r\n\tsummary: string;\r\n\tfromId: string;\r\n\ttimestamp: number;\r\n}\r\n\r\nexport interface CompactionSummaryMessage {\r\n\trole: \"compactionSummary\";\r\n\tsummary: string;\r\n\ttokensBefore: number;\r\n\ttimestamp: number;\r\n}\r\n\r\n// Extend CustomAgentMessages via declaration merging\r\ndeclare module \"@mariozechner/pi-agent-core\" {\r\n\tinterface CustomAgentMessages {\r\n\t\tbashExecution: BashExecutionMessage;\r\n\t\tcustom: CustomMessage;\r\n\t\tbranchSummary: BranchSummaryMessage;\r\n\t\tcompactionSummary: CompactionSummaryMessage;\r\n\t}\r\n}\r\n\r\n/**\r\n * Convert a BashExecutionMessage to user message text for LLM context.\r\n */\r\nexport function bashExecutionToText(msg: BashExecutionMessage): string {\r\n\tlet text = `Ran \\`${msg.command}\\`\\n`;\r\n\tif (msg.output) {\r\n\t\ttext += `\\`\\`\\`\\n${msg.output}\\n\\`\\`\\``;\r\n\t} else {\r\n\t\ttext += \"(no output)\";\r\n\t}\r\n\tif (msg.cancelled) {\r\n\t\ttext += \"\\n\\n(command cancelled)\";\r\n\t} else if (msg.exitCode !== null && msg.exitCode !== undefined && msg.exitCode !== 0) {\r\n\t\ttext += `\\n\\nCommand exited with code ${msg.exitCode}`;\r\n\t}\r\n\tif (msg.truncated && msg.fullOutputPath) {\r\n\t\ttext += `\\n\\n[Output truncated. Full output: ${msg.fullOutputPath}]`;\r\n\t}\r\n\treturn text;\r\n}\r\n\r\nexport function createBranchSummaryMessage(summary: string, fromId: string, timestamp: string): BranchSummaryMessage {\r\n\treturn {\r\n\t\trole: \"branchSummary\",\r\n\t\tsummary,\r\n\t\tfromId,\r\n\t\ttimestamp: new Date(timestamp).getTime(),\r\n\t};\r\n}\r\n\r\nexport function createCompactionSummaryMessage(\r\n\tsummary: string,\r\n\ttokensBefore: number,\r\n\ttimestamp: string,\r\n): CompactionSummaryMessage {\r\n\treturn {\r\n\t\trole: \"compactionSummary\",\r\n\t\tsummary: summary,\r\n\t\ttokensBefore,\r\n\t\ttimestamp: new Date(timestamp).getTime(),\r\n\t};\r\n}\r\n\r\n/** Convert CustomMessageEntry to AgentMessage format */\r\nexport function createCustomMessage(\r\n\tcustomType: string,\r\n\tcontent: string | (TextContent | ImageContent)[],\r\n\tdisplay: boolean,\r\n\tdetails: unknown | undefined,\r\n\ttimestamp: string,\r\n): CustomMessage {\r\n\treturn {\r\n\t\trole: \"custom\",\r\n\t\tcustomType,\r\n\t\tcontent,\r\n\t\tdisplay,\r\n\t\tdetails,\r\n\t\ttimestamp: new Date(timestamp).getTime(),\r\n\t};\r\n}\r\n\r\n/**\r\n * Transform AgentMessages (including custom types) to LLM-compatible Messages.\r\n *\r\n * This is used by:\r\n * - Agent's transormToLlm option (for prompt calls and queued messages)\r\n * - Compaction's generateSummary (for summarization)\r\n * - Custom extensions and tools\r\n */\r\nexport function convertToLlm(messages: AgentMessage[]): Message[] {\r\n\treturn messages\r\n\t\t.map((m): Message | undefined => {\r\n\t\t\tswitch (m.role) {\r\n\t\t\t\tcase \"bashExecution\":\r\n\t\t\t\t\t// Skip messages excluded from context (!! prefix)\r\n\t\t\t\t\tif (m.excludeFromContext) {\r\n\t\t\t\t\t\treturn undefined;\r\n\t\t\t\t\t}\r\n\t\t\t\t\treturn {\r\n\t\t\t\t\t\trole: \"user\",\r\n\t\t\t\t\t\tcontent: [{ type: \"text\", text: bashExecutionToText(m) }],\r\n\t\t\t\t\t\ttimestamp: m.timestamp,\r\n\t\t\t\t\t};\r\n\t\t\t\tcase \"custom\": {\r\n\t\t\t\t\tconst content = typeof m.content === \"string\" ? [{ type: \"text\" as const, text: m.content }] : m.content;\r\n\t\t\t\t\treturn {\r\n\t\t\t\t\t\trole: \"user\",\r\n\t\t\t\t\t\tcontent,\r\n\t\t\t\t\t\ttimestamp: m.timestamp,\r\n\t\t\t\t\t};\r\n\t\t\t\t}\r\n\t\t\t\tcase \"branchSummary\":\r\n\t\t\t\t\treturn {\r\n\t\t\t\t\t\trole: \"user\",\r\n\t\t\t\t\t\tcontent: [{ type: \"text\" as const, text: BRANCH_SUMMARY_PREFIX + m.summary + BRANCH_SUMMARY_SUFFIX }],\r\n\t\t\t\t\t\ttimestamp: m.timestamp,\r\n\t\t\t\t\t};\r\n\t\t\t\tcase \"compactionSummary\":\r\n\t\t\t\t\treturn {\r\n\t\t\t\t\t\trole: \"user\",\r\n\t\t\t\t\t\tcontent: [\r\n\t\t\t\t\t\t\t{ type: \"text\" as const, text: COMPACTION_SUMMARY_PREFIX + m.summary + COMPACTION_SUMMARY_SUFFIX },\r\n\t\t\t\t\t\t],\r\n\t\t\t\t\t\ttimestamp: m.timestamp,\r\n\t\t\t\t\t};\r\n\t\t\t\tcase \"user\":\r\n\t\t\t\t\treturn injectTimestamp(m);\r\n\t\t\t\tcase \"assistant\":\r\n\t\t\t\tcase \"toolResult\":\r\n\t\t\t\t\treturn m;\r\n\t\t\t\tdefault:\r\n\t\t\t\t\t// biome-ignore lint/correctness/noSwitchDeclarations: fine\r\n\t\t\t\t\tconst _exhaustiveCheck: never = m;\r\n\t\t\t\t\treturn undefined;\r\n\t\t\t}\r\n\t\t})\r\n\t\t.filter((m) => m !== undefined);\r\n}\r\n\r\n/**\r\n * Inject a human-readable timestamp prefix into a user message so the LLM\r\n * can see when each message was sent and detect time gaps between messages.\r\n */\r\nfunction injectTimestamp(m: Message & { role: \"user\" }): Message {\r\n\tif (!m.timestamp) return m;\r\n\tconst ts = formatTimestamp(m.timestamp);\r\n\tconst prefix = `[${ts}] `;\r\n\r\n\tif (typeof m.content === \"string\") {\r\n\t\tif (m.content.startsWith(\"[\")) return m; // Already has a timestamp prefix\r\n\t\treturn { ...m, content: prefix + m.content };\r\n\t}\r\n\r\n\tif (Array.isArray(m.content) && m.content.length > 0) {\r\n\t\tconst first = m.content[0];\r\n\t\tif (first.type === \"text\" && !first.text.startsWith(\"[\")) {\r\n\t\t\tconst newContent = [...m.content];\r\n\t\t\tnewContent[0] = { ...first, text: prefix + first.text };\r\n\t\t\treturn { ...m, content: newContent };\r\n\t\t}\r\n\t}\r\n\r\n\treturn m;\r\n}\r\n"]}

package/dist/core/messages.js

@@ -0,0 +1,149 @@
+/**
+ * Custom message types and transformers for the coding agent.
+ *
+ * Extends the base AgentMessage type with coding-agent specific message types,
+ * and provides a transformer to convert them to LLM-compatible messages.
+ */
+import { formatTimestamp } from "./memory.js";
+export const COMPACTION_SUMMARY_PREFIX = `The conversation history before this point was compacted into the following summary:
+
+<summary>
+`;
+export const COMPACTION_SUMMARY_SUFFIX = `
+</summary>`;
+export const BRANCH_SUMMARY_PREFIX = `The following is a summary of a branch that this conversation came back from:
+
+<summary>
+`;
+export const BRANCH_SUMMARY_SUFFIX = `</summary>`;
+/**
+ * Convert a BashExecutionMessage to user message text for LLM context.
+ */
+export function bashExecutionToText(msg) {
+    let text = `Ran \`${msg.command}\`\n`;
+    if (msg.output) {
+        text += `\`\`\`\n${msg.output}\n\`\`\``;
+    }
+    else {
+        text += "(no output)";
+    }
+    if (msg.cancelled) {
+        text += "\n\n(command cancelled)";
+    }
+    else if (msg.exitCode !== null && msg.exitCode !== undefined && msg.exitCode !== 0) {
+        text += `\n\nCommand exited with code ${msg.exitCode}`;
+    }
+    if (msg.truncated && msg.fullOutputPath) {
+        text += `\n\n[Output truncated. Full output: ${msg.fullOutputPath}]`;
+    }
+    return text;
+}
+export function createBranchSummaryMessage(summary, fromId, timestamp) {
+    return {
+        role: "branchSummary",
+        summary,
+        fromId,
+        timestamp: new Date(timestamp).getTime(),
+    };
+}
+export function createCompactionSummaryMessage(summary, tokensBefore, timestamp) {
+    return {
+        role: "compactionSummary",
+        summary: summary,
+        tokensBefore,
+        timestamp: new Date(timestamp).getTime(),
+    };
+}
+/** Convert CustomMessageEntry to AgentMessage format */
+export function createCustomMessage(customType, content, display, details, timestamp) {
+    return {
+        role: "custom",
+        customType,
+        content,
+        display,
+        details,
+        timestamp: new Date(timestamp).getTime(),
+    };
+}
+/**
+ * Transform AgentMessages (including custom types) to LLM-compatible Messages.
+ *
+ * This is used by:
+ * - Agent's transormToLlm option (for prompt calls and queued messages)
+ * - Compaction's generateSummary (for summarization)
+ * - Custom extensions and tools
+ */
+export function convertToLlm(messages) {
+    return messages
+        .map((m) => {
+        switch (m.role) {
+            case "bashExecution":
+                // Skip messages excluded from context (!! prefix)
+                if (m.excludeFromContext) {
+                    return undefined;
+                }
+                return {
+                    role: "user",
+                    content: [{ type: "text", text: bashExecutionToText(m) }],
+                    timestamp: m.timestamp,
+                };
+            case "custom": {
+                const content = typeof m.content === "string" ? [{ type: "text", text: m.content }] : m.content;
+                return {
+                    role: "user",
+                    content,
+                    timestamp: m.timestamp,
+                };
+            }
+            case "branchSummary":
+                return {
+                    role: "user",
+                    content: [{ type: "text", text: BRANCH_SUMMARY_PREFIX + m.summary + BRANCH_SUMMARY_SUFFIX }],
+                    timestamp: m.timestamp,
+                };
+            case "compactionSummary":
+                return {
+                    role: "user",
+                    content: [
+                        { type: "text", text: COMPACTION_SUMMARY_PREFIX + m.summary + COMPACTION_SUMMARY_SUFFIX },
+                    ],
+                    timestamp: m.timestamp,
+                };
+            case "user":
+                return injectTimestamp(m);
+            case "assistant":
+            case "toolResult":
+                return m;
+            default:
+                // biome-ignore lint/correctness/noSwitchDeclarations: fine
+                const _exhaustiveCheck = m;
+                return undefined;
+        }
+    })
+        .filter((m) => m !== undefined);
+}
+/**
+ * Inject a human-readable timestamp prefix into a user message so the LLM
+ * can see when each message was sent and detect time gaps between messages.
+ */
+function injectTimestamp(m) {
+    if (!m.timestamp)
+        return m;
+    const ts = formatTimestamp(m.timestamp);
+    const prefix = `[${ts}] `;
+    if (typeof m.content === "string") {
+        if (m.content.startsWith("["))
+            return m; // Already has a timestamp prefix
+        return { ...m, content: prefix + m.content };
+    }
+    if (Array.isArray(m.content) && m.content.length > 0) {
+        const first = m.content[0];
+        if (first.type === "text" && !first.text.startsWith("[")) {
+            const newContent = [...m.content];
+            newContent[0] = { ...first, text: prefix + first.text };
+            return { ...m, content: newContent };
+        }
+    }
+    return m;
+}
+//# sourceMappingURL=messages.js.map

package/dist/core/messages.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"messages.js","sourceRoot":"","sources":["../../src/core/messages.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAIH,OAAO,EAAE,eAAe,EAAE,MAAM,aAAa,CAAC;AAE9C,MAAM,CAAC,MAAM,yBAAyB,GAAG;;;CAGxC,CAAC;AAEF,MAAM,CAAC,MAAM,yBAAyB,GAAG;WAC9B,CAAC;AAEZ,MAAM,CAAC,MAAM,qBAAqB,GAAG;;;CAGpC,CAAC;AAEF,MAAM,CAAC,MAAM,qBAAqB,GAAG,YAAY,CAAC;AAuDlD;;GAEG;AACH,MAAM,UAAU,mBAAmB,CAAC,GAAyB,EAAU;IACtE,IAAI,IAAI,GAAG,SAAS,GAAG,CAAC,OAAO,MAAM,CAAC;IACtC,IAAI,GAAG,CAAC,MAAM,EAAE,CAAC;QAChB,IAAI,IAAI,WAAW,GAAG,CAAC,MAAM,UAAU,CAAC;IACzC,CAAC;SAAM,CAAC;QACP,IAAI,IAAI,aAAa,CAAC;IACvB,CAAC;IACD,IAAI,GAAG,CAAC,SAAS,EAAE,CAAC;QACnB,IAAI,IAAI,yBAAyB,CAAC;IACnC,CAAC;SAAM,IAAI,GAAG,CAAC,QAAQ,KAAK,IAAI,IAAI,GAAG,CAAC,QAAQ,KAAK,SAAS,IAAI,GAAG,CAAC,QAAQ,KAAK,CAAC,EAAE,CAAC;QACtF,IAAI,IAAI,gCAAgC,GAAG,CAAC,QAAQ,EAAE,CAAC;IACxD,CAAC;IACD,IAAI,GAAG,CAAC,SAAS,IAAI,GAAG,CAAC,cAAc,EAAE,CAAC;QACzC,IAAI,IAAI,uCAAuC,GAAG,CAAC,cAAc,GAAG,CAAC;IACtE,CAAC;IACD,OAAO,IAAI,CAAC;AAAA,CACZ;AAED,MAAM,UAAU,0BAA0B,CAAC,OAAe,EAAE,MAAc,EAAE,SAAiB,EAAwB;IACpH,OAAO;QACN,IAAI,EAAE,eAAe;QACrB,OAAO;QACP,MAAM;QACN,SAAS,EAAE,IAAI,IAAI,CAAC,SAAS,CAAC,CAAC,OAAO,EAAE;KACxC,CAAC;AAAA,CACF;AAED,MAAM,UAAU,8BAA8B,CAC7C,OAAe,EACf,YAAoB,EACpB,SAAiB,EACU;IAC3B,OAAO;QACN,IAAI,EAAE,mBAAmB;QACzB,OAAO,EAAE,OAAO;QAChB,YAAY;QACZ,SAAS,EAAE,IAAI,IAAI,CAAC,SAAS,CAAC,CAAC,OAAO,EAAE;KACxC,CAAC;AAAA,CACF;AAED,wDAAwD;AACxD,MAAM,UAAU,mBAAmB,CAClC,UAAkB,EAClB,OAAgD,EAChD,OAAgB,EAChB,OAA4B,EAC5B,SAAiB,EACD;IAChB,OAAO;QACN,IAAI,EAAE,QAAQ;QACd,UAAU;QACV,OAAO;QACP,OAAO;QACP,OAAO;QACP,SAAS,EAAE,IAAI,IAAI,CAAC,SAAS,CAAC,CAAC,OAAO,EAAE;KACxC,CAAC;AAAA,CACF;AAED;;;;;;;GAOG;AACH,MAAM,UAAU,YAAY,CAAC,QAAwB,EAAa;IACjE,OAAO,QAAQ;SACb,GAAG,CAAC,CAAC,CAAC,EAAuB,EAAE,CAAC;QAChC,QAAQ,CAAC,CAAC,IAAI,EAAE,CAAC;YAChB,KAAK,eAAe;gBACnB,kDAAkD;gBAClD,IAAI,CAAC,CAAC,kBAAkB,EAAE,CAAC;oBAC1B,OAAO,SAAS,CAAC;gBAClB,CAAC;gBACD,OAAO;oBACN,IAAI,EAAE,MAAM;oBACZ,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,mBAAmB,CAAC,CAAC,CAAC,EAAE,CAAC;oBACzD,SAAS,EAAE,CAAC,CAAC,SAAS;iBACtB,CAAC;YACH,KAAK,QAAQ,EAAE,CAAC;gBACf,MAAM,OAAO,GAAG,OAAO,CAAC,CAAC,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;gBACzG,OAAO;oBACN,IAAI,EAAE,MAAM;oBACZ,OAAO;oBACP,SAAS,EAAE,CAAC,CAAC,SAAS;iBACtB,CAAC;YACH,CAAC;YACD,KAAK,eAAe;gBACnB,OAAO;oBACN,IAAI,EAAE,MAAM;oBACZ,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,qBAAqB,GAAG,CAAC,CAAC,OAAO,GAAG,qBAAqB,EAAE,CAAC;oBACrG,SAAS,EAAE,CAAC,CAAC,SAAS;iBACtB,CAAC;YACH,KAAK,mBAAmB;gBACvB,OAAO;oBACN,IAAI,EAAE,MAAM;oBACZ,OAAO,EAAE;wBACR,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,yBAAyB,GAAG,CAAC,CAAC,OAAO,GAAG,yBAAyB,EAAE;qBAClG;oBACD,SAAS,EAAE,CAAC,CAAC,SAAS;iBACtB,CAAC;YACH,KAAK,MAAM;gBACV,OAAO,eAAe,CAAC,CAAC,CAAC,CAAC;YAC3B,KAAK,WAAW,CAAC;YACjB,KAAK,YAAY;gBAChB,OAAO,CAAC,CAAC;YACV;gBACC,2DAA2D;gBAC3D,MAAM,gBAAgB,GAAU,CAAC,CAAC;gBAClC,OAAO,SAAS,CAAC;QACnB,CAAC;IAAA,CACD,CAAC;SACD,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,SAAS,CAAC,CAAC;AAAA,CACjC;AAED;;;GAGG;AACH,SAAS,eAAe,CAAC,CAA6B,EAAW;IAChE,IAAI,CAAC,CAAC,CAAC,SAAS;QAAE,OAAO,CAAC,CAAC;IAC3B,MAAM,EAAE,GAAG,eAAe,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;IACxC,MAAM,MAAM,GAAG,IAAI,EAAE,IAAI,CAAC;IAE1B,IAAI,OAAO,CAAC,CAAC,OAAO,KAAK,QAAQ,EAAE,CAAC;QACnC,IAAI,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC,GAAG,CAAC;YAAE,OAAO,CAAC,CAAC,CAAC,iCAAiC;QAC1E,OAAO,EAAE,GAAG,CAAC,EAAE,OAAO,EAAE,MAAM,GAAG,CAAC,CAAC,OAAO,EAAE,CAAC;IAC9C,CAAC;IAED,IAAI,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACtD,MAAM,KAAK,GAAG,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;QAC3B,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;YAC1D,MAAM,UAAU,GAAG,CAAC,GAAG,CAAC,CAAC,OAAO,CAAC,CAAC;YAClC,UAAU,CAAC,CAAC,CAAC,GAAG,EAAE,GAAG,KAAK,EAAE,IAAI,EAAE,MAAM,GAAG,KAAK,CAAC,IAAI,EAAE,CAAC;YACxD,OAAO,EAAE,GAAG,CAAC,EAAE,OAAO,EAAE,UAAU,EAAE,CAAC;QACtC,CAAC;IACF,CAAC;IAED,OAAO,CAAC,CAAC;AAAA,CACT","sourcesContent":["/**\r\n * Custom message types and transformers for the coding agent.\r\n *\r\n * Extends the base AgentMessage type with coding-agent specific message types,\r\n * and provides a transformer to convert them to LLM-compatible messages.\r\n */\r\n\r\nimport type { AgentMessage } from \"@mariozechner/pi-agent-core\";\r\nimport type { ImageContent, Message, TextContent } from \"@mariozechner/pi-ai\";\r\nimport { formatTimestamp } from \"./memory.js\";\r\n\r\nexport const COMPACTION_SUMMARY_PREFIX = `The conversation history before this point was compacted into the following summary:\r\n\r\n<summary>\r\n`;\r\n\r\nexport const COMPACTION_SUMMARY_SUFFIX = `\r\n</summary>`;\r\n\r\nexport const BRANCH_SUMMARY_PREFIX = `The following is a summary of a branch that this conversation came back from:\r\n\r\n<summary>\r\n`;\r\n\r\nexport const BRANCH_SUMMARY_SUFFIX = `</summary>`;\r\n\r\n/**\r\n * Message type for bash executions via the ! command.\r\n */\r\nexport interface BashExecutionMessage {\r\n\trole: \"bashExecution\";\r\n\tcommand: string;\r\n\toutput: string;\r\n\texitCode: number | undefined;\r\n\tcancelled: boolean;\r\n\ttruncated: boolean;\r\n\tfullOutputPath?: string;\r\n\ttimestamp: number;\r\n\t/** If true, this message is excluded from LLM context (!! prefix) */\r\n\texcludeFromContext?: boolean;\r\n}\r\n\r\n/**\r\n * Message type for extension-injected messages via sendMessage().\r\n * These are custom messages that extensions can inject into the conversation.\r\n */\r\nexport interface CustomMessage<T = unknown> {\r\n\trole: \"custom\";\r\n\tcustomType: string;\r\n\tcontent: string | (TextContent | ImageContent)[];\r\n\tdisplay: boolean;\r\n\tdetails?: T;\r\n\ttimestamp: number;\r\n}\r\n\r\nexport interface BranchSummaryMessage {\r\n\trole: \"branchSummary\";\r\n\tsummary: string;\r\n\tfromId: string;\r\n\ttimestamp: number;\r\n}\r\n\r\nexport interface CompactionSummaryMessage {\r\n\trole: \"compactionSummary\";\r\n\tsummary: string;\r\n\ttokensBefore: number;\r\n\ttimestamp: number;\r\n}\r\n\r\n// Extend CustomAgentMessages via declaration merging\r\ndeclare module \"@mariozechner/pi-agent-core\" {\r\n\tinterface CustomAgentMessages {\r\n\t\tbashExecution: BashExecutionMessage;\r\n\t\tcustom: CustomMessage;\r\n\t\tbranchSummary: BranchSummaryMessage;\r\n\t\tcompactionSummary: CompactionSummaryMessage;\r\n\t}\r\n}\r\n\r\n/**\r\n * Convert a BashExecutionMessage to user message text for LLM context.\r\n */\r\nexport function bashExecutionToText(msg: BashExecutionMessage): string {\r\n\tlet text = `Ran \\`${msg.command}\\`\\n`;\r\n\tif (msg.output) {\r\n\t\ttext += `\\`\\`\\`\\n${msg.output}\\n\\`\\`\\``;\r\n\t} else {\r\n\t\ttext += \"(no output)\";\r\n\t}\r\n\tif (msg.cancelled) {\r\n\t\ttext += \"\\n\\n(command cancelled)\";\r\n\t} else if (msg.exitCode !== null && msg.exitCode !== undefined && msg.exitCode !== 0) {\r\n\t\ttext += `\\n\\nCommand exited with code ${msg.exitCode}`;\r\n\t}\r\n\tif (msg.truncated && msg.fullOutputPath) {\r\n\t\ttext += `\\n\\n[Output truncated. Full output: ${msg.fullOutputPath}]`;\r\n\t}\r\n\treturn text;\r\n}\r\n\r\nexport function createBranchSummaryMessage(summary: string, fromId: string, timestamp: string): BranchSummaryMessage {\r\n\treturn {\r\n\t\trole: \"branchSummary\",\r\n\t\tsummary,\r\n\t\tfromId,\r\n\t\ttimestamp: new Date(timestamp).getTime(),\r\n\t};\r\n}\r\n\r\nexport function createCompactionSummaryMessage(\r\n\tsummary: string,\r\n\ttokensBefore: number,\r\n\ttimestamp: string,\r\n): CompactionSummaryMessage {\r\n\treturn {\r\n\t\trole: \"compactionSummary\",\r\n\t\tsummary: summary,\r\n\t\ttokensBefore,\r\n\t\ttimestamp: new Date(timestamp).getTime(),\r\n\t};\r\n}\r\n\r\n/** Convert CustomMessageEntry to AgentMessage format */\r\nexport function createCustomMessage(\r\n\tcustomType: string,\r\n\tcontent: string | (TextContent | ImageContent)[],\r\n\tdisplay: boolean,\r\n\tdetails: unknown | undefined,\r\n\ttimestamp: string,\r\n): CustomMessage {\r\n\treturn {\r\n\t\trole: \"custom\",\r\n\t\tcustomType,\r\n\t\tcontent,\r\n\t\tdisplay,\r\n\t\tdetails,\r\n\t\ttimestamp: new Date(timestamp).getTime(),\r\n\t};\r\n}\r\n\r\n/**\r\n * Transform AgentMessages (including custom types) to LLM-compatible Messages.\r\n *\r\n * This is used by:\r\n * - Agent's transormToLlm option (for prompt calls and queued messages)\r\n * - Compaction's generateSummary (for summarization)\r\n * - Custom extensions and tools\r\n */\r\nexport function convertToLlm(messages: AgentMessage[]): Message[] {\r\n\treturn messages\r\n\t\t.map((m): Message | undefined => {\r\n\t\t\tswitch (m.role) {\r\n\t\t\t\tcase \"bashExecution\":\r\n\t\t\t\t\t// Skip messages excluded from context (!! prefix)\r\n\t\t\t\t\tif (m.excludeFromContext) {\r\n\t\t\t\t\t\treturn undefined;\r\n\t\t\t\t\t}\r\n\t\t\t\t\treturn {\r\n\t\t\t\t\t\trole: \"user\",\r\n\t\t\t\t\t\tcontent: [{ type: \"text\", text: bashExecutionToText(m) }],\r\n\t\t\t\t\t\ttimestamp: m.timestamp,\r\n\t\t\t\t\t};\r\n\t\t\t\tcase \"custom\": {\r\n\t\t\t\t\tconst content = typeof m.content === \"string\" ? [{ type: \"text\" as const, text: m.content }] : m.content;\r\n\t\t\t\t\treturn {\r\n\t\t\t\t\t\trole: \"user\",\r\n\t\t\t\t\t\tcontent,\r\n\t\t\t\t\t\ttimestamp: m.timestamp,\r\n\t\t\t\t\t};\r\n\t\t\t\t}\r\n\t\t\t\tcase \"branchSummary\":\r\n\t\t\t\t\treturn {\r\n\t\t\t\t\t\trole: \"user\",\r\n\t\t\t\t\t\tcontent: [{ type: \"text\" as const, text: BRANCH_SUMMARY_PREFIX + m.summary + BRANCH_SUMMARY_SUFFIX }],\r\n\t\t\t\t\t\ttimestamp: m.timestamp,\r\n\t\t\t\t\t};\r\n\t\t\t\tcase \"compactionSummary\":\r\n\t\t\t\t\treturn {\r\n\t\t\t\t\t\trole: \"user\",\r\n\t\t\t\t\t\tcontent: [\r\n\t\t\t\t\t\t\t{ type: \"text\" as const, text: COMPACTION_SUMMARY_PREFIX + m.summary + COMPACTION_SUMMARY_SUFFIX },\r\n\t\t\t\t\t\t],\r\n\t\t\t\t\t\ttimestamp: m.timestamp,\r\n\t\t\t\t\t};\r\n\t\t\t\tcase \"user\":\r\n\t\t\t\t\treturn injectTimestamp(m);\r\n\t\t\t\tcase \"assistant\":\r\n\t\t\t\tcase \"toolResult\":\r\n\t\t\t\t\treturn m;\r\n\t\t\t\tdefault:\r\n\t\t\t\t\t// biome-ignore lint/correctness/noSwitchDeclarations: fine\r\n\t\t\t\t\tconst _exhaustiveCheck: never = m;\r\n\t\t\t\t\treturn undefined;\r\n\t\t\t}\r\n\t\t})\r\n\t\t.filter((m) => m !== undefined);\r\n}\r\n\r\n/**\r\n * Inject a human-readable timestamp prefix into a user message so the LLM\r\n * can see when each message was sent and detect time gaps between messages.\r\n */\r\nfunction injectTimestamp(m: Message & { role: \"user\" }): Message {\r\n\tif (!m.timestamp) return m;\r\n\tconst ts = formatTimestamp(m.timestamp);\r\n\tconst prefix = `[${ts}] `;\r\n\r\n\tif (typeof m.content === \"string\") {\r\n\t\tif (m.content.startsWith(\"[\")) return m; // Already has a timestamp prefix\r\n\t\treturn { ...m, content: prefix + m.content };\r\n\t}\r\n\r\n\tif (Array.isArray(m.content) && m.content.length > 0) {\r\n\t\tconst first = m.content[0];\r\n\t\tif (first.type === \"text\" && !first.text.startsWith(\"[\")) {\r\n\t\t\tconst newContent = [...m.content];\r\n\t\t\tnewContent[0] = { ...first, text: prefix + first.text };\r\n\t\t\treturn { ...m, content: newContent };\r\n\t\t}\r\n\t}\r\n\r\n\treturn m;\r\n}\r\n"]}