@zihanw/pi-forge 0.2.0 → 0.3.1

package/dist/preview.js

@@ -0,0 +1,179 @@
+import { agentMessageToPreviewText, compileMessages, compileSystemPrompt, createPromptVariableStore, getLatestUserMessage, } from "./compiler.js";
+import { estimatePayloadTokens } from "./payload-capture.js";
+export function renderPreview(ctx, target, sessionVariables) {
+    return buildPreview(ctx, target, sessionVariables).text;
+}
+export function buildPreview(ctx, target, sessionVariables, optionsOverride) {
+    const options = optionsOverride ?? ctx.getSystemPromptOptions();
+    const sessionMessages = getPreviewSessionMessages(ctx);
+    const latestUserMessage = getLatestUserMessage(sessionMessages);
+    const previewVariables = createPromptVariableStore(sessionVariables);
+    const runtime = { options, ctx, latestUserMessage, now: new Date(), variables: previewVariables };
+    const system = compileSystemPrompt(target.stack, runtime, ctx.getSystemPrompt());
+    const messages = compileMessages(target.stack, runtime, sessionMessages);
+    const diagnostics = [...target.diagnostics, ...system.diagnostics, ...messages.diagnostics];
+    const messageSections = messages.messages.map((message, index) => {
+        const content = agentMessageToPreviewText(message);
+        const source = messages.messageSources[index];
+        return previewSection(`message-${index}`, previewMessageTitle(source, index), content, message.role);
+    });
+    const systemSection = previewSection("system", "System prompt", system.systemPrompt || "(empty)");
+    const totalChars = systemSection.chars + messageSections.reduce((sum, section) => sum + section.chars, 0);
+    const preview = {
+        stackId: target.stack.id,
+        generatedAt: new Date().toISOString(),
+        system: systemSection,
+        messages: messageSections,
+        totalChars,
+        approxTokens: estimatePayloadTokens(`${system.systemPrompt}\n${messageSections.map((section) => section.content).join("\n")}`),
+    };
+    const text = [
+        `# Prompt stack preview: ${target.stack.id}`,
+        "",
+        "## System prompt",
+        "",
+        system.systemPrompt || "(empty)",
+        "",
+        "## Message layout",
+        "",
+        renderPreviewSectionText(messageSections),
+        "",
+        "## Diagnostics",
+        "",
+        renderDiagnostics(diagnostics),
+    ].join("\n");
+    return { text, preview, diagnostics };
+}
+function getPreviewSessionMessages(ctx) {
+    const entries = getCurrentBranchEntries(ctx).map(asSessionEntry);
+    const compaction = latestCompactionEntry(entries);
+    const messages = [];
+    const appendMessage = (entry) => {
+        if (entry.type === "message" && isAgentMessage(entry.message)) {
+            messages.push(entry.message);
+            return;
+        }
+        if (entry.type === "custom_message" && typeof entry.customType === "string") {
+            messages.push({
+                role: "custom",
+                customType: entry.customType,
+                content: typeof entry.content === "string" || Array.isArray(entry.content) ? entry.content : "",
+                display: entry.display === true,
+                details: entry.details,
+                timestamp: entryTimestamp(entry),
+            });
+            return;
+        }
+        if (entry.type === "branch_summary" && typeof entry.summary === "string" && entry.summary) {
+            messages.push({
+                role: "branchSummary",
+                summary: entry.summary,
+                fromId: typeof entry.fromId === "string" ? entry.fromId : "",
+                timestamp: entryTimestamp(entry),
+            });
+        }
+    };
+    if (compaction && typeof compaction.summary === "string") {
+        messages.push({
+            role: "compactionSummary",
+            summary: compaction.summary,
+            tokensBefore: typeof compaction.tokensBefore === "number" ? compaction.tokensBefore : 0,
+            timestamp: entryTimestamp(compaction),
+        });
+        const compactionIndex = entries.findIndex((entry) => entry.type === "compaction" && entry.id === compaction.id);
+        let foundFirstKept = false;
+        for (let index = 0; index < compactionIndex; index++) {
+            const entry = entries[index];
+            if (entry.id === compaction.firstKeptEntryId)
+                foundFirstKept = true;
+            if (foundFirstKept)
+                appendMessage(entry);
+        }
+        for (let index = compactionIndex + 1; index < entries.length; index++) {
+            appendMessage(entries[index]);
+        }
+        return messages;
+    }
+    for (const entry of entries)
+        appendMessage(entry);
+    return messages;
+}
+function latestCompactionEntry(entries) {
+    for (let index = entries.length - 1; index >= 0; index--) {
+        const entry = entries[index];
+        if (entry.type === "compaction")
+            return entry;
+    }
+    return undefined;
+}
+function getCurrentBranchEntries(ctx) {
+    const leafId = ctx.sessionManager.getLeafId();
+    if (leafId === null)
+        return [];
+    const sessionManager = ctx.sessionManager;
+    return sessionManager.getBranch ? sessionManager.getBranch(leafId ?? undefined) : sessionManager.getEntries();
+}
+function asSessionEntry(value) {
+    return value && typeof value === "object" ? value : {};
+}
+function isAgentMessage(value) {
+    return !!value && typeof value === "object" && typeof value.role === "string";
+}
+function entryTimestamp(entry) {
+    if (typeof entry.timestamp === "number")
+        return entry.timestamp;
+    if (typeof entry.timestamp === "string") {
+        const timestamp = new Date(entry.timestamp).getTime();
+        if (Number.isFinite(timestamp))
+            return timestamp;
+    }
+    return Date.now();
+}
+function previewMessageTitle(source, index) {
+    if (source?.kind === "stack-item") {
+        return source.itemName?.trim() || source.itemId || `Stack item ${index + 1}`;
+    }
+    if (source?.kind === "chat-history") {
+        const label = source.itemName?.trim() || "Chat history";
+        return `${label} #${source.historyIndex ?? index + 1}`;
+    }
+    if (source?.kind === "implicit-history") {
+        return `Conversation history #${source.historyIndex ?? index + 1}`;
+    }
+    return `Message ${index + 1}`;
+}
+function renderPreviewSectionText(sections, maxChars = 8000) {
+    let text = "";
+    for (const section of sections) {
+        const role = section.role ? ` (${section.role})` : "";
+        text += `\n--- ${section.title}${role} ---\n`;
+        text += section.content;
+        text += "\n";
+        if (text.length > maxChars)
+            return `${text.slice(0, maxChars)}\n\n[preview truncated]`;
+    }
+    return text.trimStart();
+}
+function previewSection(id, title, content, role) {
+    return {
+        id,
+        title,
+        role,
+        content,
+        chars: content.length,
+        approxTokens: estimatePayloadTokens(content),
+    };
+}
+export function renderDiagnostics(diagnostics) {
+    if (diagnostics.length === 0)
+        return "No diagnostics.";
+    return diagnostics.map((d) => `${d.level.toUpperCase()}${d.itemId ? ` [${d.itemId}]` : ""}: ${d.message}`).join("\n");
+}
+export async function showText(ctx, title, text) {
+    if (ctx.hasUI) {
+        await ctx.ui.editor(title, text);
+        return;
+    }
+    console.log(text);
+}
+//# sourceMappingURL=preview.js.map

package/dist/preview.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"preview.js","sourceRoot":"","sources":["../src/preview.ts"],"names":[],"mappings":"AAEA,OAAO,EACN,yBAAyB,EACzB,eAAe,EACf,mBAAmB,EACnB,yBAAyB,EACzB,oBAAoB,GACpB,MAAM,eAAe,CAAC;AACvB,OAAO,EAAE,qBAAqB,EAAE,MAAM,sBAAsB,CAAC;AAI7D,MAAM,UAAU,aAAa,CAC5B,GAA4B,EAC5B,MAAyB,EACzB,gBAAqD;IAErD,OAAO,YAAY,CAAC,GAAG,EAAE,MAAM,EAAE,gBAAgB,CAAC,CAAC,IAAI,CAAC;AACzD,CAAC;AAED,MAAM,UAAU,YAAY,CAC3B,GAA4B,EAC5B,MAAyB,EACzB,gBAAqD,EACrD,eAA0C;IAE1C,MAAM,OAAO,GAAG,eAAe,IAAI,GAAG,CAAC,sBAAsB,EAAE,CAAC;IAChE,MAAM,eAAe,GAAG,yBAAyB,CAAC,GAAG,CAAC,CAAC;IACvD,MAAM,iBAAiB,GAAG,oBAAoB,CAAC,eAAe,CAAC,CAAC;IAChE,MAAM,gBAAgB,GAAG,yBAAyB,CAAC,gBAAgB,CAAC,CAAC;IACrE,MAAM,OAAO,GAAG,EAAE,OAAO,EAAE,GAAG,EAAE,iBAAiB,EAAE,GAAG,EAAE,IAAI,IAAI,EAAE,EAAE,SAAS,EAAE,gBAAgB,EAAE,CAAC;IAClG,MAAM,MAAM,GAAG,mBAAmB,CAAC,MAAM,CAAC,KAAK,EAAE,OAAO,EAAE,GAAG,CAAC,eAAe,EAAE,CAAC,CAAC;IACjF,MAAM,QAAQ,GAAG,eAAe,CAAC,MAAM,CAAC,KAAK,EAAE,OAAO,EAAE,eAAe,CAAC,CAAC;IACzE,MAAM,WAAW,GAAG,CAAC,GAAG,MAAM,CAAC,WAAW,EAAE,GAAG,MAAM,CAAC,WAAW,EAAE,GAAG,QAAQ,CAAC,WAAW,CAAC,CAAC;IAC5F,MAAM,eAAe,GAAG,QAAQ,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,KAAK,EAAE,EAAE;QAChE,MAAM,OAAO,GAAG,yBAAyB,CAAC,OAAO,CAAC,CAAC;QACnD,MAAM,MAAM,GAAG,QAAQ,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC;QAC9C,OAAO,cAAc,CAAC,WAAW,KAAK,EAAE,EAAE,mBAAmB,CAAC,MAAM,EAAE,KAAK,CAAC,EAAE,OAAO,EAAE,OAAO,CAAC,IAAI,CAAC,CAAC;IACtG,CAAC,CAAC,CAAC;IACH,MAAM,aAAa,GAAG,cAAc,CAAC,QAAQ,EAAE,eAAe,EAAE,MAAM,CAAC,YAAY,IAAI,SAAS,CAAC,CAAC;IAClG,MAAM,UAAU,GAAG,aAAa,CAAC,KAAK,GAAG,eAAe,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,OAAO,EAAE,EAAE,CAAC,GAAG,GAAG,OAAO,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;IAC1G,MAAM,OAAO,GAAqB;QACjC,OAAO,EAAE,MAAM,CAAC,KAAK,CAAC,EAAE;QACxB,WAAW,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;QACrC,MAAM,EAAE,aAAa;QACrB,QAAQ,EAAE,eAAe;QACzB,UAAU;QACV,YAAY,EAAE,qBAAqB,CAAC,GAAG,MAAM,CAAC,YAAY,KAAK,eAAe,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;KAC9H,CAAC;IAEF,MAAM,IAAI,GAAG;QACZ,2BAA2B,MAAM,CAAC,KAAK,CAAC,EAAE,EAAE;QAC5C,EAAE;QACF,kBAAkB;QAClB,EAAE;QACF,MAAM,CAAC,YAAY,IAAI,SAAS;QAChC,EAAE;QACF,mBAAmB;QACnB,EAAE;QACF,wBAAwB,CAAC,eAAe,CAAC;QACzC,EAAE;QACF,gBAAgB;QAChB,EAAE;QACF,iBAAiB,CAAC,WAAW,CAAC;KAC9B,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAEb,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,WAAW,EAAE,CAAC;AACvC,CAAC;AAkBD,SAAS,yBAAyB,CAAC,GAA4B;IAC9D,MAAM,OAAO,GAAG,uBAAuB,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;IACjE,MAAM,UAAU,GAAG,qBAAqB,CAAC,OAAO,CAAC,CAAC;IAClD,MAAM,QAAQ,GAAmB,EAAE,CAAC;IAEpC,MAAM,aAAa,GAAG,CAAC,KAAuB,EAAQ,EAAE;QACvD,IAAI,KAAK,CAAC,IAAI,KAAK,SAAS,IAAI,cAAc,CAAC,KAAK,CAAC,OAAO,CAAC,EAAE,CAAC;YAC/D,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;YAC7B,OAAO;QACR,CAAC;QACD,IAAI,KAAK,CAAC,IAAI,KAAK,gBAAgB,IAAI,OAAO,KAAK,CAAC,UAAU,KAAK,QAAQ,EAAE,CAAC;YAC7E,QAAQ,CAAC,IAAI,CAAC;gBACb,IAAI,EAAE,QAAQ;gBACd,UAAU,EAAE,KAAK,CAAC,UAAU;gBAC5B,OAAO,EAAE,OAAO,KAAK,CAAC,OAAO,KAAK,QAAQ,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE;gBAC/F,OAAO,EAAE,KAAK,CAAC,OAAO,KAAK,IAAI;gBAC/B,OAAO,EAAE,KAAK,CAAC,OAAO;gBACtB,SAAS,EAAE,cAAc,CAAC,KAAK,CAAC;aAChB,CAAC,CAAC;YACnB,OAAO;QACR,CAAC;QACD,IAAI,KAAK,CAAC,IAAI,KAAK,gBAAgB,IAAI,OAAO,KAAK,CAAC,OAAO,KAAK,QAAQ,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;YAC3F,QAAQ,CAAC,IAAI,CAAC;gBACb,IAAI,EAAE,eAAe;gBACrB,OAAO,EAAE,KAAK,CAAC,OAAO;gBACtB,MAAM,EAAE,OAAO,KAAK,CAAC,MAAM,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE;gBAC5D,SAAS,EAAE,cAAc,CAAC,KAAK,CAAC;aAChB,CAAC,CAAC;QACpB,CAAC;IACF,CAAC,CAAC;IAEF,IAAI,UAAU,IAAI,OAAO,UAAU,CAAC,OAAO,KAAK,QAAQ,EAAE,CAAC;QAC1D,QAAQ,CAAC,IAAI,CAAC;YACb,IAAI,EAAE,mBAAmB;YACzB,OAAO,EAAE,UAAU,CAAC,OAAO;YAC3B,YAAY,EAAE,OAAO,UAAU,CAAC,YAAY,KAAK,QAAQ,CAAC,CAAC,CAAC,UAAU,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACvF,SAAS,EAAE,cAAc,CAAC,UAAU,CAAC;SACrB,CAAC,CAAC;QAEnB,MAAM,eAAe,GAAG,OAAO,CAAC,SAAS,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,YAAY,IAAI,KAAK,CAAC,EAAE,KAAK,UAAU,CAAC,EAAE,CAAC,CAAC;QAChH,IAAI,cAAc,GAAG,KAAK,CAAC;QAC3B,KAAK,IAAI,KAAK,GAAG,CAAC,EAAE,KAAK,GAAG,eAAe,EAAE,KAAK,EAAE,EAAE,CAAC;YACtD,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAE,CAAC;YAC9B,IAAI,KAAK,CAAC,EAAE,KAAK,UAAU,CAAC,gBAAgB;gBAAE,cAAc,GAAG,IAAI,CAAC;YACpE,IAAI,cAAc;gBAAE,aAAa,CAAC,KAAK,CAAC,CAAC;QAC1C,CAAC;QACD,KAAK,IAAI,KAAK,GAAG,eAAe,GAAG,CAAC,EAAE,KAAK,GAAG,OAAO,CAAC,MAAM,EAAE,KAAK,EAAE,EAAE,CAAC;YACvE,aAAa,CAAC,OAAO,CAAC,KAAK,CAAE,CAAC,CAAC;QAChC,CAAC;QACD,OAAO,QAAQ,CAAC;IACjB,CAAC;IAED,KAAK,MAAM,KAAK,IAAI,OAAO;QAAE,aAAa,CAAC,KAAK,CAAC,CAAC;IAClD,OAAO,QAAQ,CAAC;AACjB,CAAC;AAED,SAAS,qBAAqB,CAAC,OAA2B;IACzD,KAAK,IAAI,KAAK,GAAG,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,KAAK,IAAI,CAAC,EAAE,KAAK,EAAE,EAAE,CAAC;QAC1D,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAE,CAAC;QAC9B,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY;YAAE,OAAO,KAAK,CAAC;IAC/C,CAAC;IACD,OAAO,SAAS,CAAC;AAClB,CAAC;AAED,SAAS,uBAAuB,CAAC,GAA4B;IAC5D,MAAM,MAAM,GAAG,GAAG,CAAC,cAAc,CAAC,SAAS,EAAE,CAAC;IAC9C,IAAI,MAAM,KAAK,IAAI;QAAE,OAAO,EAAE,CAAC;IAC/B,MAAM,cAAc,GAAG,GAAG,CAAC,cAG1B,CAAC;IACF,OAAO,cAAc,CAAC,SAAS,CAAC,CAAC,CAAC,cAAc,CAAC,SAAS,CAAC,MAAM,IAAI,SAAS,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,UAAU,EAAE,CAAC;AAC/G,CAAC;AAED,SAAS,cAAc,CAAC,KAAc;IACrC,OAAO,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAyB,CAAC,CAAC,CAAC,EAAE,CAAC;AAC5E,CAAC;AAED,SAAS,cAAc,CAAC,KAAc;IACrC,OAAO,CAAC,CAAC,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,OAAQ,KAA4B,CAAC,IAAI,KAAK,QAAQ,CAAC;AACvG,CAAC;AAED,SAAS,cAAc,CAAC,KAAuB;IAC9C,IAAI,OAAO,KAAK,CAAC,SAAS,KAAK,QAAQ;QAAE,OAAO,KAAK,CAAC,SAAS,CAAC;IAChE,IAAI,OAAO,KAAK,CAAC,SAAS,KAAK,QAAQ,EAAE,CAAC;QACzC,MAAM,SAAS,GAAG,IAAI,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,OAAO,EAAE,CAAC;QACtD,IAAI,MAAM,CAAC,QAAQ,CAAC,SAAS,CAAC;YAAE,OAAO,SAAS,CAAC;IAClD,CAAC;IACD,OAAO,IAAI,CAAC,GAAG,EAAE,CAAC;AACnB,CAAC;AAED,SAAS,mBAAmB,CAAC,MAAwC,EAAE,KAAa;IACnF,IAAI,MAAM,EAAE,IAAI,KAAK,YAAY,EAAE,CAAC;QACnC,OAAO,MAAM,CAAC,QAAQ,EAAE,IAAI,EAAE,IAAI,MAAM,CAAC,MAAM,IAAI,cAAc,KAAK,GAAG,CAAC,EAAE,CAAC;IAC9E,CAAC;IACD,IAAI,MAAM,EAAE,IAAI,KAAK,cAAc,EAAE,CAAC;QACrC,MAAM,KAAK,GAAG,MAAM,CAAC,QAAQ,EAAE,IAAI,EAAE,IAAI,cAAc,CAAC;QACxD,OAAO,GAAG,KAAK,KAAK,MAAM,CAAC,YAAY,IAAI,KAAK,GAAG,CAAC,EAAE,CAAC;IACxD,CAAC;IACD,IAAI,MAAM,EAAE,IAAI,KAAK,kBAAkB,EAAE,CAAC;QACzC,OAAO,yBAAyB,MAAM,CAAC,YAAY,IAAI,KAAK,GAAG,CAAC,EAAE,CAAC;IACpE,CAAC;IACD,OAAO,WAAW,KAAK,GAAG,CAAC,EAAE,CAAC;AAC/B,CAAC;AAED,SAAS,wBAAwB,CAAC,QAAmC,EAAE,QAAQ,GAAG,IAAI;IACrF,IAAI,IAAI,GAAG,EAAE,CAAC;IACd,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;QAChC,MAAM,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,KAAK,OAAO,CAAC,IAAI,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC;QACtD,IAAI,IAAI,SAAS,OAAO,CAAC,KAAK,GAAG,IAAI,QAAQ,CAAC;QAC9C,IAAI,IAAI,OAAO,CAAC,OAAO,CAAC;QACxB,IAAI,IAAI,IAAI,CAAC;QACb,IAAI,IAAI,CAAC,MAAM,GAAG,QAAQ;YAAE,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,QAAQ,CAAC,yBAAyB,CAAC;IACxF,CAAC;IACD,OAAO,IAAI,CAAC,SAAS,EAAE,CAAC;AACzB,CAAC;AAED,SAAS,cAAc,CAAC,EAAU,EAAE,KAAa,EAAE,OAAe,EAAE,IAAa;IAChF,OAAO;QACN,EAAE;QACF,KAAK;QACL,IAAI;QACJ,OAAO;QACP,KAAK,EAAE,OAAO,CAAC,MAAM;QACrB,YAAY,EAAE,qBAAqB,CAAC,OAAO,CAAC;KAC5C,CAAC;AACH,CAAC;AAED,MAAM,UAAU,iBAAiB,CAAC,WAAoC;IACrE,IAAI,WAAW,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,iBAAiB,CAAC;IACvD,OAAO,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC,KAAK,CAAC,WAAW,EAAE,GAAG,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AACvH,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,QAAQ,CAAC,GAA4B,EAAE,KAAa,EAAE,IAAY;IACvF,IAAI,GAAG,CAAC,KAAK,EAAE,CAAC;QACf,MAAM,GAAG,CAAC,EAAE,CAAC,MAAM,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;QACjC,OAAO;IACR,CAAC;IACD,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;AACnB,CAAC"}

package/dist/regex.d.ts

@@ -0,0 +1,7 @@
+import type { AgentMessage } from "@earendil-works/pi-agent-core";
+import type { PromptRegexStage, PromptRegexTarget, PromptStack, PromptStackDiagnostic } from "./types.ts";
+export declare function validateRegexConfig(config: unknown): PromptStackDiagnostic[];
+export declare function applyRegexRulesToString(stack: PromptStack, text: string, stage: PromptRegexStage, target: PromptRegexTarget, diagnostics: PromptStackDiagnostic[]): string;
+export declare function applyRegexRulesToMessages(stack: PromptStack, messages: AgentMessage[], stage: PromptRegexStage, diagnostics: PromptStackDiagnostic[]): AgentMessage[];
+export declare function applyFinalizeRegexRulesToMessage(stack: PromptStack, message: AgentMessage, diagnostics: PromptStackDiagnostic[]): AgentMessage | undefined;
+//# sourceMappingURL=regex.d.ts.map

package/dist/regex.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"regex.d.ts","sourceRoot":"","sources":["../src/regex.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,+BAA+B,CAAC;AAClE,OAAO,KAAK,EAGX,gBAAgB,EAChB,iBAAiB,EACjB,WAAW,EACX,qBAAqB,EACrB,MAAM,YAAY,CAAC;AAiCpB,wBAAgB,mBAAmB,CAAC,MAAM,EAAE,OAAO,GAAG,qBAAqB,EAAE,CAwB5E;AAED,wBAAgB,uBAAuB,CACtC,KAAK,EAAE,WAAW,EAClB,IAAI,EAAE,MAAM,EACZ,KAAK,EAAE,gBAAgB,EACvB,MAAM,EAAE,iBAAiB,EACzB,WAAW,EAAE,qBAAqB,EAAE,GAClC,MAAM,CAWR;AAED,wBAAgB,yBAAyB,CACxC,KAAK,EAAE,WAAW,EAClB,QAAQ,EAAE,YAAY,EAAE,EACxB,KAAK,EAAE,gBAAgB,EACvB,WAAW,EAAE,qBAAqB,EAAE,GAClC,YAAY,EAAE,CAQhB;AAED,wBAAgB,gCAAgC,CAC/C,KAAK,EAAE,WAAW,EAClB,OAAO,EAAE,YAAY,EACrB,WAAW,EAAE,qBAAqB,EAAE,GAClC,YAAY,GAAG,SAAS,CAiB1B"}

package/dist/regex.js

@@ -0,0 +1,432 @@
+const ALLOWED_REGEX_FLAGS = new Set(["g", "i", "m", "s", "u"]);
+const VALID_STAGES = new Set(["history", "compiled"]);
+const VALID_EFFECTS = new Set(["outgoing", "display", "both", "finalize"]);
+const VALID_TARGETS = new Set(["system", "messages"]);
+export function validateRegexConfig(config) {
+    const diagnostics = [];
+    if (config === undefined)
+        return diagnostics;
+    if (!isPlainObject(config)) {
+        diagnostics.push({ level: "warning", message: "regex must be an object when provided." });
+        return diagnostics;
+    }
+    if (config.schemaVersion !== undefined && config.schemaVersion !== 1) {
+        diagnostics.push({ level: "warning", message: "Missing or unsupported regex.schemaVersion; assuming 1." });
+    }
+    const rawRules = config.rules;
+    if (rawRules === undefined)
+        return diagnostics;
+    if (!Array.isArray(rawRules)) {
+        diagnostics.push({ level: "error", message: "regex.rules must be an array when provided." });
+        return diagnostics;
+    }
+    const seenIds = new Set();
+    for (const [index, rawRule] of rawRules.entries()) {
+        validateRule(rawRule, index, seenIds, diagnostics);
+    }
+    return diagnostics;
+}
+export function applyRegexRulesToString(stack, text, stage, target, diagnostics) {
+    let result = text;
+    for (const rule of regexRulesFor(stack, stage, target, "outgoing", diagnostics)) {
+        const transformed = transformString(result, rule);
+        result = transformed.text;
+        addRuleStats(diagnostics, rule, stage, target, {
+            matches: transformed.matches,
+            changedSegments: transformed.changed ? 1 : 0,
+        });
+    }
+    return result;
+}
+export function applyRegexRulesToMessages(stack, messages, stage, diagnostics) {
+    let result = messages;
+    for (const rule of regexRulesFor(stack, stage, "messages", "outgoing", diagnostics)) {
+        const stats = { matches: 0, changedSegments: 0 };
+        result = transformMessages(result, rule, stats);
+        addRuleStats(diagnostics, rule, stage, "messages", stats);
+    }
+    return result;
+}
+export function applyFinalizeRegexRulesToMessage(stack, message, diagnostics) {
+    if (String(message.role) !== "assistant")
+        return undefined;
+    let result = message;
+    for (const rule of regexRulesFor(stack, "compiled", "messages", "finalize", diagnostics)) {
+        const stats = { matches: 0, changedSegments: 0 };
+        const [next = result] = transformMessages([result], rule, stats);
+        result = next;
+        addRuleStats(diagnostics, rule, "finalize", "messages", stats);
+    }
+    if (result === message)
+        return undefined;
+    diagnostics.push({
+        level: "warning",
+        message: "Finalize regex replaced finalized assistant message content; original model output is not preserved in the stored transcript.",
+    });
+    return result;
+}
+function validateRule(rawRule, index, seenIds, diagnostics) {
+    const label = `regex rule ${index + 1}`;
+    if (!isPlainObject(rawRule)) {
+        diagnostics.push({ level: "error", message: `${label} must be an object.` });
+        return;
+    }
+    const id = typeof rawRule.id === "string" ? rawRule.id.trim() : "";
+    if (!id) {
+        diagnostics.push({ level: "error", message: `${label} must have a non-empty id.` });
+    }
+    else if (seenIds.has(id)) {
+        diagnostics.push({ level: "error", message: `Duplicate regex rule id: ${id}.` });
+    }
+    else {
+        seenIds.add(id);
+    }
+    if (rawRule.enabled !== undefined && typeof rawRule.enabled !== "boolean") {
+        diagnostics.push({ level: "warning", message: `${regexRuleLabel(id, index)} enabled must be a boolean when provided.` });
+    }
+    if (!VALID_STAGES.has(rawRule.stage)) {
+        diagnostics.push({ level: "error", message: `${regexRuleLabel(id, index)} stage must be "history" or "compiled".` });
+    }
+    const effect = typeof rawRule.effect === "string" ? rawRule.effect : undefined;
+    if (rawRule.effect !== undefined) {
+        if (!VALID_EFFECTS.has(rawRule.effect)) {
+            diagnostics.push({ level: "error", message: `${regexRuleLabel(id, index)} effect must be "outgoing", "display", "both", or "finalize".` });
+        }
+        else if (rawRule.effect === "display") {
+            diagnostics.push({ level: "warning", message: `${regexRuleLabel(id, index)} effect "display" is not a runtime effect in pi-forge yet and will be ignored. Use "finalize" only when destructive transcript cleanup is intended.` });
+        }
+        else if (rawRule.effect === "both") {
+            diagnostics.push({ level: "warning", message: `${regexRuleLabel(id, index)} effect "both" is not implemented and will be ignored; use separate outgoing and finalize rules.` });
+        }
+        else if (rawRule.effect === "finalize") {
+            diagnostics.push({ level: "warning", message: `${regexRuleLabel(id, index)} effect "finalize" rewrites finalized assistant messages and replaces the original model output in the stored transcript.` });
+        }
+    }
+    if (effect === "finalize" && rawRule.stage !== "compiled") {
+        diagnostics.push({ level: "error", message: `${regexRuleLabel(id, index)} effect "finalize" requires stage "compiled".` });
+    }
+    if (typeof rawRule.pattern !== "string" || rawRule.pattern.length === 0) {
+        diagnostics.push({ level: "error", message: `${regexRuleLabel(id, index)} pattern must be a non-empty string.` });
+    }
+    else {
+        const flagsError = validateRegexFlags(rawRule.flags, id, index);
+        if (flagsError)
+            diagnostics.push({ level: "error", message: flagsError });
+        else {
+            try {
+                new RegExp(rawRule.pattern, typeof rawRule.flags === "string" ? rawRule.flags : "");
+            }
+            catch (error) {
+                diagnostics.push({ level: "error", message: `${regexRuleLabel(id, index)} pattern failed to compile: ${error instanceof Error ? error.message : String(error)}` });
+            }
+        }
+    }
+    if (rawRule.replace !== undefined && typeof rawRule.replace !== "string") {
+        diagnostics.push({ level: "error", message: `${regexRuleLabel(id, index)} replace must be a string when provided.` });
+    }
+    else if (typeof rawRule.replace === "string") {
+        if (/\{\{\s*match\s*\}\}/i.test(rawRule.replace)) {
+            diagnostics.push({ level: "warning", message: `${regexRuleLabel(id, index)} replacement contains SillyTavern {{match}}; use JavaScript $& or re-import the SillyTavern preset to convert it.` });
+        }
+    }
+    if (rawRule.trimStrings !== undefined && !isStringArray(rawRule.trimStrings)) {
+        diagnostics.push({ level: "error", message: `${regexRuleLabel(id, index)} trimStrings must be an array of strings when provided.` });
+    }
+    if (rawRule.roles !== undefined && !isStringArray(rawRule.roles)) {
+        diagnostics.push({ level: "error", message: `${regexRuleLabel(id, index)} roles must be an array of strings when provided.` });
+    }
+    if (rawRule.targets !== undefined) {
+        if (!isStringArray(rawRule.targets)) {
+            diagnostics.push({ level: "error", message: `${regexRuleLabel(id, index)} targets must be an array of strings when provided.` });
+        }
+        else {
+            for (const target of rawRule.targets) {
+                if (!VALID_TARGETS.has(target)) {
+                    diagnostics.push({ level: "error", message: `${regexRuleLabel(id, index)} target must be "system" or "messages": ${target}.` });
+                }
+                else if (effect === "finalize" && target !== "messages") {
+                    diagnostics.push({ level: "error", message: `${regexRuleLabel(id, index)} effect "finalize" only supports target "messages": ${target}.` });
+                }
+            }
+        }
+    }
+    if (effect === "finalize" && isStringArray(rawRule.roles) && !rawRule.roles.includes("assistant")) {
+        diagnostics.push({ level: "warning", message: `${regexRuleLabel(id, index)} effect "finalize" only runs for finalized assistant messages, but roles does not include "assistant".` });
+    }
+    if (rawRule.maxMessages !== undefined && !isPositiveInteger(rawRule.maxMessages)) {
+        diagnostics.push({ level: "error", message: `${regexRuleLabel(id, index)} maxMessages must be a positive integer when provided.` });
+    }
+    if (rawRule.maxChars !== undefined && !isPositiveInteger(rawRule.maxChars)) {
+        diagnostics.push({ level: "error", message: `${regexRuleLabel(id, index)} maxChars must be a positive integer when provided.` });
+    }
+    if (rawRule.minDepth !== undefined && !isNonNegativeInteger(rawRule.minDepth)) {
+        diagnostics.push({ level: "error", message: `${regexRuleLabel(id, index)} minDepth must be a non-negative integer when provided.` });
+    }
+    if (rawRule.maxDepth !== undefined && !isNonNegativeInteger(rawRule.maxDepth)) {
+        diagnostics.push({ level: "error", message: `${regexRuleLabel(id, index)} maxDepth must be a non-negative integer when provided.` });
+    }
+    if (isNonNegativeInteger(rawRule.minDepth) && isNonNegativeInteger(rawRule.maxDepth) && rawRule.maxDepth < rawRule.minDepth) {
+        diagnostics.push({ level: "error", message: `${regexRuleLabel(id, index)} maxDepth must be greater than or equal to minDepth.` });
+    }
+}
+function regexRulesFor(stack, stage, target, effect, diagnostics) {
+    const rules = Array.isArray(stack.regex?.rules) ? stack.regex.rules : [];
+    const compiled = [];
+    for (const rawRule of rules) {
+        if (!isPlainObject(rawRule))
+            continue;
+        if (rawRule.enabled === false)
+            continue;
+        if ((rawRule.effect ?? "outgoing") !== effect)
+            continue;
+        if (rawRule.stage !== stage)
+            continue;
+        if (stage === "compiled" && Array.isArray(rawRule.targets) && !rawRule.targets.includes(target))
+            continue;
+        const rule = compileRuntimeRule(rawRule, diagnostics);
+        if (rule)
+            compiled.push(rule);
+    }
+    return compiled;
+}
+function compileRuntimeRule(rule, diagnostics) {
+    if (typeof rule.id !== "string" || !rule.id.trim())
+        return undefined;
+    if (rule.stage !== "history" && rule.stage !== "compiled")
+        return undefined;
+    const effect = rule.effect ?? "outgoing";
+    if (!VALID_EFFECTS.has(effect))
+        return undefined;
+    if (typeof rule.pattern !== "string" || rule.pattern.length === 0)
+        return undefined;
+    const flags = typeof rule.flags === "string" ? rule.flags : "";
+    const flagsError = validateRegexFlags(flags, rule.id, -1);
+    if (flagsError)
+        return undefined;
+    try {
+        return {
+            id: rule.id.trim(),
+            stage: rule.stage,
+            effect,
+            targets: normalizeTargets(rule.targets),
+            roles: isStringArray(rule.roles) ? rule.roles : undefined,
+            maxMessages: isPositiveInteger(rule.maxMessages) ? Math.floor(rule.maxMessages) : undefined,
+            maxChars: isPositiveInteger(rule.maxChars) ? Math.floor(rule.maxChars) : undefined,
+            minDepth: isNonNegativeInteger(rule.minDepth) ? Math.floor(rule.minDepth) : undefined,
+            maxDepth: isNonNegativeInteger(rule.maxDepth) ? Math.floor(rule.maxDepth) : undefined,
+            regexp: new RegExp(rule.pattern, flags),
+            replace: typeof rule.replace === "string" ? rule.replace : "",
+            trimStrings: isStringArray(rule.trimStrings) ? rule.trimStrings : undefined,
+        };
+    }
+    catch (error) {
+        diagnostics.push({ level: "error", message: `Regex rule ${rule.id} failed to compile: ${error instanceof Error ? error.message : String(error)}` });
+        return undefined;
+    }
+}
+function transformMessages(messages, rule, stats) {
+    const eligible = eligibleMessageIndexes(messages, rule);
+    const eligibleSet = new Set(rule.maxMessages ? eligible.slice(-rule.maxMessages) : eligible);
+    let changed = false;
+    const transformed = messages.map((message, index) => {
+        if (!eligibleSet.has(index))
+            return message;
+        const next = transformMessage(message, rule, stats);
+        if (next !== message)
+            changed = true;
+        return next;
+    });
+    return changed ? transformed : messages;
+}
+function eligibleMessageIndexes(messages, rule) {
+    const indexes = [];
+    for (const [index, message] of messages.entries()) {
+        if (rule.roles && !rule.roles.includes(String(message.role)))
+            continue;
+        const depth = messages.length - 1 - index;
+        if (rule.minDepth !== undefined && depth < rule.minDepth)
+            continue;
+        if (rule.maxDepth !== undefined && depth > rule.maxDepth)
+            continue;
+        indexes.push(index);
+    }
+    return indexes;
+}
+function transformMessage(message, rule, stats) {
+    const content = message.content;
+    if (typeof content === "string") {
+        const result = transformString(content, rule);
+        mergeStringStats(stats, result);
+        return result.changed ? { ...message, content: result.text } : message;
+    }
+    if (!Array.isArray(content))
+        return message;
+    let changed = false;
+    const nextContent = content.map((part) => {
+        if (!isPlainObject(part) || part.type !== "text" || typeof part.text !== "string")
+            return part;
+        const result = transformString(part.text, rule);
+        mergeStringStats(stats, result);
+        if (!result.changed)
+            return part;
+        changed = true;
+        return { ...part, text: result.text };
+    });
+    return changed ? { ...message, content: nextContent } : message;
+}
+function transformString(text, rule) {
+    const headLength = rule.maxChars && text.length > rule.maxChars ? text.length - rule.maxChars : 0;
+    const head = headLength > 0 ? text.slice(0, headLength) : "";
+    const body = headLength > 0 ? text.slice(headLength) : text;
+    const matches = countReplacementMatches(body, rule.regexp);
+    const replaced = rule.trimStrings && rule.trimStrings.length > 0
+        ? replaceWithTrimStrings(body, rule)
+        : body.replace(rule.regexp, convertDollarZeroToFullMatch(rule.replace));
+    const result = head + replaced;
+    return {
+        text: result,
+        matches,
+        changed: result !== text,
+    };
+}
+/**
+ * Normalizes a JavaScript replacement string so that `$0` (not followed by another
+ * digit) behaves as the full match, matching the custom trimStrings expander and
+ * SillyTavern conventions. `$$0` stays a literal `$0`. This keeps `$0` consistent
+ * across the native and trimStrings replacement paths.
+ */
+function convertDollarZeroToFullMatch(replace) {
+    let result = "";
+    for (let i = 0; i < replace.length;) {
+        const ch = replace[i];
+        if (ch !== undefined && ch !== "$") {
+            result += ch;
+            i++;
+            continue;
+        }
+        const next = replace[i + 1];
+        if (next === "$") {
+            result += "$$";
+            i += 2;
+            continue;
+        }
+        if (next === "0" && !isDigitCode(replace.charCodeAt(i + 2))) {
+            result += "$&";
+            i += 2;
+            continue;
+        }
+        result += "$";
+        i++;
+    }
+    return result;
+}
+function isDigitCode(code) {
+    return code >= 48 && code <= 57;
+}
+function replaceWithTrimStrings(text, rule) {
+    return text.replace(rule.regexp, (...args) => {
+        const groups = isPlainObject(args.at(-1)) ? args.pop() : undefined;
+        const input = String(args.pop() ?? "");
+        const offset = Number(args.pop() ?? 0);
+        const [match = "", ...captures] = args.map((arg) => typeof arg === "string" ? arg : "");
+        return expandReplacementTemplate(rule.replace, {
+            match,
+            captures,
+            offset,
+            input,
+            groups,
+            trimStrings: rule.trimStrings ?? [],
+        });
+    });
+}
+function expandReplacementTemplate(template, context) {
+    return template.replace(/\$([$&`']|0(?!\d)|[1-9]\d?|<[^>]+>)/g, (token) => {
+        if (token === "$$")
+            return "$";
+        if (token === "$&" || token === "$0")
+            return trimMatchedValue(context.match, context.trimStrings);
+        if (token === "$`")
+            return context.input.slice(0, context.offset);
+        if (token === "$'")
+            return context.input.slice(context.offset + context.match.length);
+        if (token.startsWith("$<")) {
+            const name = token.slice(2, -1);
+            const value = context.groups?.[name];
+            return typeof value === "string" ? trimMatchedValue(value, context.trimStrings) : "";
+        }
+        const captureIndex = Number(token.slice(1)) - 1;
+        const capture = context.captures[captureIndex];
+        return capture === undefined ? "" : trimMatchedValue(capture, context.trimStrings);
+    });
+}
+function trimMatchedValue(value, trimStrings) {
+    let result = value;
+    for (const trimString of trimStrings) {
+        if (!trimString)
+            continue;
+        result = result.split(trimString).join("");
+    }
+    return result;
+}
+function countReplacementMatches(text, regexp) {
+    if (!regexp.global)
+        return new RegExp(regexp.source, regexp.flags.replace("g", "")).test(text) ? 1 : 0;
+    const matcher = new RegExp(regexp.source, regexp.flags);
+    let count = 0;
+    let match;
+    while ((match = matcher.exec(text)) !== null) {
+        count++;
+        if (match[0] === "")
+            matcher.lastIndex++;
+    }
+    return count;
+}
+function mergeStringStats(stats, result) {
+    stats.matches += result.matches;
+    if (result.changed)
+        stats.changedSegments++;
+}
+function addRuleStats(diagnostics, rule, stage, target, stats) {
+    if (stats.matches === 0)
+        return;
+    diagnostics.push({
+        level: "info",
+        message: `Regex rule ${rule.id} matched ${stats.matches} time(s) and changed ${stats.changedSegments} text segment(s) in ${stage}/${target}.`,
+    });
+}
+function validateRegexFlags(value, id, index) {
+    if (value === undefined)
+        return undefined;
+    if (typeof value !== "string")
+        return `${regexRuleLabel(id, index)} flags must be a string when provided.`;
+    const seen = new Set();
+    for (const flag of value) {
+        if (!ALLOWED_REGEX_FLAGS.has(flag))
+            return `${regexRuleLabel(id, index)} has unsupported regex flag: ${flag}.`;
+        if (seen.has(flag))
+            return `${regexRuleLabel(id, index)} has duplicate regex flag: ${flag}.`;
+        seen.add(flag);
+    }
+    return undefined;
+}
+function normalizeTargets(value) {
+    if (!isStringArray(value))
+        return undefined;
+    return value.filter((target) => target === "system" || target === "messages");
+}
+function regexRuleLabel(id, index) {
+    return id ? `Regex rule ${id}` : `regex rule ${index + 1}`;
+}
+function isStringArray(value) {
+    return Array.isArray(value) && value.every((item) => typeof item === "string");
+}
+function isPositiveInteger(value) {
+    return typeof value === "number" && Number.isInteger(value) && value > 0;
+}
+function isNonNegativeInteger(value) {
+    return typeof value === "number" && Number.isInteger(value) && value >= 0;
+}
+function isPlainObject(value) {
+    return !!value && typeof value === "object" && !Array.isArray(value);
+}
+//# sourceMappingURL=regex.js.map