pi-kot 0.1.21 → 0.1.22

package/server/compaction-cards.js

@@ -0,0 +1,180 @@
+function isCompactionEntry(e) {
+    return e.type === "compaction";
+}
+function isMessageEntry(e) {
+    return e.type === "message";
+}
+/**
+ * Walk `session.sessionManager.getEntries()` once and produce one
+ * CompactionEvent per CompactionEntry. The mapping back to a position
+ * in `session.messages` is the tricky part: the post-compaction
+ * messages array is rebuilt from the entries that come AT or AFTER
+ * the latest compaction's `firstKeptEntryId`, with the compaction
+ * summary itself synthesised in. We compute `insertBeforeIndex` by
+ * counting how many "kept" message entries appear in `messages`
+ * before the position where this compaction's first-kept entry lands.
+ */
+export function buildCompactionHistory(session) {
+    const entries = session.sessionManager.getEntries();
+    const compactions = [];
+    for (const e of entries)
+        if (isCompactionEntry(e))
+            compactions.push(e);
+    if (compactions.length === 0)
+        return [];
+    // Walk entries in order; for each compaction event, capture the
+    // message entries between the previous compaction (or session start)
+    // and this one. Those are the "archived" messages — the ones that
+    // were summarised and dropped.
+    const events = [];
+    let archiveBuffer = [];
+    for (const entry of entries) {
+        if (isMessageEntry(entry)) {
+            archiveBuffer.push(entry.message);
+            continue;
+        }
+        if (isCompactionEntry(entry)) {
+            // Estimate what the kept context size was AFTER this compaction
+            // by counting tokens for all message entries from firstKeptEntryId
+            // to the end of the entries array.
+            const estimatedTokensAfter = estimateKeptTokens(entries, entry.firstKeptEntryId);
+            events.push({
+                id: entry.id,
+                timestamp: entry.timestamp,
+                summary: entry.summary,
+                tokensBefore: entry.tokensBefore,
+                estimatedTokensAfter: estimatedTokensAfter > 0 ? estimatedTokensAfter : undefined,
+                // Filled in below — depends on the index of `firstKeptEntryId`
+                // within the *kept* portion of the messages stream.
+                insertBeforeIndex: 0,
+                archivedMessages: archiveBuffer,
+            });
+            // Reset for the next archive window. Anything between this
+            // compaction and the next one (or end of session) belongs to
+            // the next event's archive — but only if there IS a next
+            // compaction. If not, those messages live in `session.messages`
+            // and don't need archiving.
+            archiveBuffer = [];
+        }
+    }
+    // Compute insertBeforeIndex for each event. Given a compaction whose
+    // firstKeptEntryId points at message-entry M, the number of message
+    // entries BEFORE M (within the post-this-compaction kept portion of
+    // the entries array) tells us where to splice the card.
+    //
+    // For the LAST compaction, the kept portion lines up with what
+    // session.messages actually contains, so we can count message
+    // entries from `firstKeptEntryId` forward up to the compaction's
+    // own position. Earlier compactions' kept portions were re-archived
+    // by later compactions — those events sit at the TOP of the
+    // (current) display, so insertBeforeIndex = 0 for all but the last.
+    // (Once a compaction's kept window has itself been archived, no
+    // post-compaction message in `session.messages` corresponds to it,
+    // so the only sensible position is "before everything else".)
+    for (let i = 0; i < events.length - 1; i++) {
+        const ev = events[i];
+        if (ev !== undefined)
+            ev.insertBeforeIndex = 0;
+    }
+    const last = events[events.length - 1];
+    if (last !== undefined) {
+        const lastCompaction = compactions[compactions.length - 1];
+        if (lastCompaction !== undefined) {
+            last.insertBeforeIndex = countMessagesBetween(entries, lastCompaction.firstKeptEntryId, lastCompaction.id);
+        }
+    }
+    return events;
+}
+/**
+ * Estimate token count for a single message using char/4 heuristic.
+ * Mirrors the SDK's `estimateTokens()` in `core/compaction/compaction.ts`.
+ */
+function estimateMessageTokens(msg) {
+    const m = msg;
+    let chars = 0;
+    // Roles with a `content` field: user, assistant, toolResult, custom
+    const content = m.content;
+    if (typeof content === "string") {
+        chars += content.length;
+    }
+    else if (Array.isArray(content)) {
+        for (const block of content) {
+            const b = block;
+            if (b.type === "text" && typeof b.text === "string") {
+                chars += b.text.length;
+            }
+            else if (b.type === "thinking" && typeof b.thinking === "string") {
+                chars += b.thinking.length;
+            }
+            else if (b.type === "toolCall") {
+                chars += typeof b.name === "string" ? b.name.length : 0;
+                try {
+                    chars += typeof b.arguments === "string"
+                        ? b.arguments.length
+                        : JSON.stringify(b.arguments).length;
+                }
+                catch { /* skip unstringifiable */ }
+            }
+            else if (b.type === "image") {
+                chars += 4800; // ESTIMATED_IMAGE_CHARS — same value as the SDK
+            }
+        }
+    }
+    // compactionSummary / branchSummary roles: carry `summary` instead of `content`
+    if (typeof m.summary === "string") {
+        chars = m.summary.length;
+    }
+    // bashExecution: carries `command` + `output`
+    if (typeof m.command === "string") {
+        chars += m.command.length;
+    }
+    if (typeof m.output === "string") {
+        chars += m.output.length;
+    }
+    return Math.ceil(chars / 4);
+}
+/**
+ * Walk message entries from `firstKeptEntryId` to the end of the entries
+ * array and estimate their total tokens. This gives us the
+ * `estimatedTokensAfter` — the approximate size of the kept context
+ * right after compaction.
+ */
+function estimateKeptTokens(entries, firstKeptEntryId) {
+    const startIdx = entries.findIndex((e) => e.id === firstKeptEntryId);
+    if (startIdx === -1)
+        return 0;
+    let total = 0;
+    for (let i = startIdx; i < entries.length; i++) {
+        const e = entries[i];
+        if (e === undefined)
+            break;
+        if (isMessageEntry(e)) {
+            total += estimateMessageTokens(e.message);
+        }
+    }
+    return total;
+}
+/**
+ * Count message-typed entries between two entry ids (exclusive of the
+ * end id). Returns the position where the compaction card should
+ * splice into `session.messages`. When `firstKeptEntryId` doesn't
+ * resolve (legacy session, manual JSONL edit), returns 0 — the card
+ * lands at the top of the chat, which is the safe fallback.
+ */
+function countMessagesBetween(entries, firstKeptEntryId, endEntryId) {
+    const startIdx = entries.findIndex((e) => e.id === firstKeptEntryId);
+    if (startIdx === -1)
+        return 0;
+    let count = 0;
+    for (let i = startIdx; i < entries.length; i++) {
+        const e = entries[i];
+        if (e === undefined)
+            break;
+        if (e.id === endEntryId)
+            break;
+        if (isMessageEntry(e))
+            count++;
+    }
+    return count;
+}
+//# sourceMappingURL=compaction-cards.js.map

package/server/compaction-cards.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"compaction-cards.js","sourceRoot":"","sources":["../src/compaction-cards.ts"],"names":[],"mappings":"AAoEA,SAAS,iBAAiB,CAAC,CAAe;IACxC,OAAO,CAAC,CAAC,IAAI,KAAK,YAAY,CAAC;AACjC,CAAC;AAED,SAAS,cAAc,CAAC,CAAe;IACrC,OAAO,CAAC,CAAC,IAAI,KAAK,SAAS,CAAC;AAC9B,CAAC;AAED;;;;;;;;;GASG;AACH,MAAM,UAAU,sBAAsB,CAAC,OAAqB;IAC1D,MAAM,OAAO,GAAG,OAAO,CAAC,cAAc,CAAC,UAAU,EAAE,CAAC;IACpD,MAAM,WAAW,GAA4C,EAAE,CAAC;IAChE,KAAK,MAAM,CAAC,IAAI,OAAO;QAAE,IAAI,iBAAiB,CAAC,CAAC,CAAC;YAAE,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACvE,IAAI,WAAW,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,EAAE,CAAC;IAExC,gEAAgE;IAChE,qEAAqE;IACrE,kEAAkE;IAClE,+BAA+B;IAC/B,MAAM,MAAM,GAAsB,EAAE,CAAC;IACrC,IAAI,aAAa,GAAmB,EAAE,CAAC;IACvC,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;QAC5B,IAAI,cAAc,CAAC,KAAK,CAAC,EAAE,CAAC;YAC1B,aAAa,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;YAClC,SAAS;QACX,CAAC;QACD,IAAI,iBAAiB,CAAC,KAAK,CAAC,EAAE,CAAC;YAC7B,gEAAgE;YAChE,mEAAmE;YACnE,mCAAmC;YACnC,MAAM,oBAAoB,GAAG,kBAAkB,CAAC,OAAO,EAAE,KAAK,CAAC,gBAAgB,CAAC,CAAC;YAEjF,MAAM,CAAC,IAAI,CAAC;gBACV,EAAE,EAAE,KAAK,CAAC,EAAE;gBACZ,SAAS,EAAE,KAAK,CAAC,SAAS;gBAC1B,OAAO,EAAE,KAAK,CAAC,OAAO;gBACtB,YAAY,EAAE,KAAK,CAAC,YAAY;gBAChC,oBAAoB,EAAE,oBAAoB,GAAG,CAAC,CAAC,CAAC,CAAC,oBAAoB,CAAC,CAAC,CAAC,SAAS;gBACjF,+DAA+D;gBAC/D,oDAAoD;gBACpD,iBAAiB,EAAE,CAAC;gBACpB,gBAAgB,EAAE,aAAa;aAChC,CAAC,CAAC;YACH,2DAA2D;YAC3D,6DAA6D;YAC7D,yDAAyD;YACzD,gEAAgE;YAChE,4BAA4B;YAC5B,aAAa,GAAG,EAAE,CAAC;QACrB,CAAC;IACH,CAAC;IAED,qEAAqE;IACrE,oEAAoE;IACpE,oEAAoE;IACpE,wDAAwD;IACxD,EAAE;IACF,+DAA+D;IAC/D,8DAA8D;IAC9D,iEAAiE;IACjE,oEAAoE;IACpE,4DAA4D;IAC5D,oEAAoE;IACpE,gEAAgE;IAChE,mEAAmE;IACnE,8DAA8D;IAC9D,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;QAC3C,MAAM,EAAE,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC;QACrB,IAAI,EAAE,KAAK,SAAS;YAAE,EAAE,CAAC,iBAAiB,GAAG,CAAC,CAAC;IACjD,CAAC;IACD,MAAM,IAAI,GAAG,MAAM,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;IACvC,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;QACvB,MAAM,cAAc,GAAG,WAAW,CAAC,WAAW,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QAC3D,IAAI,cAAc,KAAK,SAAS,EAAE,CAAC;YACjC,IAAI,CAAC,iBAAiB,GAAG,oBAAoB,CAC3C,OAAO,EACP,cAAc,CAAC,gBAAgB,EAC/B,cAAc,CAAC,EAAE,CAClB,CAAC;QACJ,CAAC;IACH,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC;AAED;;;GAGG;AACH,SAAS,qBAAqB,CAAC,GAAiB;IAC9C,MAAM,CAAC,GAAG,GAAyC,CAAC;IACpD,IAAI,KAAK,GAAG,CAAC,CAAC;IAEd,oEAAoE;IACpE,MAAM,OAAO,GAAG,CAAC,CAAC,OAAO,CAAC;IAC1B,IAAI,OAAO,OAAO,KAAK,QAAQ,EAAE,CAAC;QAChC,KAAK,IAAI,OAAO,CAAC,MAAM,CAAC;IAC1B,CAAC;SAAM,IAAI,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;QAClC,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;YAC5B,MAAM,CAAC,GAAG,KAAgC,CAAC;YAC3C,IAAI,CAAC,CAAC,IAAI,KAAK,MAAM,IAAI,OAAO,CAAC,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;gBACpD,KAAK,IAAI,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC;YACzB,CAAC;iBAAM,IAAI,CAAC,CAAC,IAAI,KAAK,UAAU,IAAI,OAAO,CAAC,CAAC,QAAQ,KAAK,QAAQ,EAAE,CAAC;gBACnE,KAAK,IAAI,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC;YAC7B,CAAC;iBAAM,IAAI,CAAC,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;gBACjC,KAAK,IAAI,OAAO,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;gBACxD,IAAI,CAAC;oBACH,KAAK,IAAI,OAAO,CAAC,CAAC,SAAS,KAAK,QAAQ;wBACtC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,MAAM;wBACpB,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,MAAM,CAAC;gBACzC,CAAC;gBAAC,MAAM,CAAC,CAAC,0BAA0B,CAAC,CAAC;YACxC,CAAC;iBAAM,IAAI,CAAC,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;gBAC9B,KAAK,IAAI,IAAI,CAAC,CAAC,gDAAgD;YACjE,CAAC;QACH,CAAC;IACH,CAAC;IAED,gFAAgF;IAChF,IAAI,OAAO,CAAC,CAAC,OAAO,KAAK,QAAQ,EAAE,CAAC;QAClC,KAAK,GAAG,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC;IAC3B,CAAC;IACD,8CAA8C;IAC9C,IAAI,OAAO,CAAC,CAAC,OAAO,KAAK,QAAQ,EAAE,CAAC;QAClC,KAAK,IAAI,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC;IAC5B,CAAC;IACD,IAAI,OAAO,CAAC,CAAC,MAAM,KAAK,QAAQ,EAAE,CAAC;QACjC,KAAK,IAAI,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC;IAC3B,CAAC;IAED,OAAO,IAAI,CAAC,IAAI,CAAC,KAAK,GAAG,CAAC,CAAC,CAAC;AAC9B,CAAC;AAED;;;;;GAKG;AACH,SAAS,kBAAkB,CAAC,OAAuB,EAAE,gBAAwB;IAC3E,MAAM,QAAQ,GAAG,OAAO,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,gBAAgB,CAAC,CAAC;IACrE,IAAI,QAAQ,KAAK,CAAC,CAAC;QAAE,OAAO,CAAC,CAAC;IAC9B,IAAI,KAAK,GAAG,CAAC,CAAC;IACd,KAAK,IAAI,CAAC,GAAG,QAAQ,EAAE,CAAC,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QAC/C,MAAM,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;QACrB,IAAI,CAAC,KAAK,SAAS;YAAE,MAAM;QAC3B,IAAI,cAAc,CAAC,CAAC,CAAC,EAAE,CAAC;YACtB,KAAK,IAAI,qBAAqB,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;QAC5C,CAAC;IACH,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC;AAED;;;;;;GAMG;AACH,SAAS,oBAAoB,CAC3B,OAAuB,EACvB,gBAAwB,EACxB,UAAkB;IAElB,MAAM,QAAQ,GAAG,OAAO,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,gBAAgB,CAAC,CAAC;IACrE,IAAI,QAAQ,KAAK,CAAC,CAAC;QAAE,OAAO,CAAC,CAAC;IAC9B,IAAI,KAAK,GAAG,CAAC,CAAC;IACd,KAAK,IAAI,CAAC,GAAG,QAAQ,EAAE,CAAC,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QAC/C,MAAM,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;QACrB,IAAI,CAAC,KAAK,SAAS;YAAE,MAAM;QAC3B,IAAI,CAAC,CAAC,EAAE,KAAK,UAAU;YAAE,MAAM;QAC/B,IAAI,cAAc,CAAC,CAAC,CAAC;YAAE,KAAK,EAAE,CAAC;IACjC,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC"}

package/server/compaction-continuation.d.ts

@@ -0,0 +1,112 @@
+/**
+ * In-process pi extension that nudges the model to KEEP WORKING on
+ * the in-progress task after an overflow-driven auto-compaction,
+ * instead of producing a prose status report that paraphrases the
+ * compaction summary's "Next Steps" section.
+ *
+ * ## The problem this solves
+ *
+ * When the LLM rejects a request with a context-overflow error, the
+ * pi SDK auto-compacts and calls `agent.continue()` to resume the
+ * loop. The model then sees:
+ *
+ *   - system prompt + tool defs
+ *   - <kept-recent messages — whatever fit under keepRecentTokens>
+ *   - <compactionSummary> — a structured doc with sections "## Goal",
+ *     "## Progress (Done / In Progress / Blocked)", "## Next Steps",
+ *     "## Critical Context"
+ *
+ * The failed assistant message was stripped by the SDK (per the
+ * overflow path in `agent-session.js:_checkCompaction`), so the
+ * `compactionSummary` is the LAST message and the next model output
+ * starts a fresh turn. Strong models (Claude, GPT-4) usually infer
+ * "I'm mid-task — pick up where I left off." Weaker / smaller local
+ * models (Gemma-class on vLLM in particular) read the structured
+ * summary as a status report addressed TO them and respond with
+ * prose paraphrasing the "Next Steps" section instead of actually
+ * doing the work. End-effect: the agent stops making progress right
+ * when the user needs it to keep going.
+ *
+ * The next action might be a tool call, a final answer, a follow-up
+ * question, or anything else the task requires — we don't prescribe
+ * the shape, we just say "keep going."
+ *
+ * ## How this extension fixes it
+ *
+ * Registers a `context` event handler. The `context` event fires
+ * BEFORE every LLM request, with the messages array that's about to
+ * be sent. We never mutate the session itself — we append a
+ * synthetic user message to the OUTBOUND copy only. The handler:
+ *
+ *   1. Looks at the messages array.
+ *   2. If the LAST message is a `compactionSummary`, the model is
+ *      about to receive a fresh post-compaction context with no
+ *      forward prompt. Append `NUDGE_MESSAGE` — a one-line
+ *      imperative telling the model to resume by calling tools.
+ *   3. Otherwise (last message is a user prompt, assistant turn, or
+ *      tool result), do nothing. Threshold compactions don't auto-
+ *      continue so the user types the next prompt themselves — which
+ *      becomes the last message — and provides the imperative
+ *      naturally.
+ *
+ * The nudge is sent to the LLM but NOT persisted to the session
+ * JSONL, so it doesn't leak into compaction summaries, tree views,
+ * or session exports.
+ *
+ * ## Why the `context` hook and not `session_before_compact`
+ *
+ * `session_before_compact` would let us replace the entire
+ * compaction (cancel it, or provide our own). To append text to the
+ * SDK-generated summary we'd have to re-implement summarization
+ * (call the SDK's exported `compact()` helper, mutate the result,
+ * return it). That works but is heavier and couples us to internal
+ * SDK shapes. The `context` hook is lighter: SDK does its own
+ * summarization, we just inject the imperative at LLM-call time.
+ *
+ * ## Why a user message and not a system / custom message
+ *
+ * The `context` hook returns an `AgentMessage[]` that the SDK passes
+ * to the provider. Providers expect a system prompt + a user/assistant
+ * dialogue; injecting a synthetic `system` mid-stream is non-standard
+ * and reorders cache. A trailing user message is the cheapest signal
+ * the model interprets as "respond to this now."
+ *
+ * Useful for nudging the model to continue after an auto-compaction.
+ */
+import type { ExtensionFactory } from "@earendil-works/pi-coding-agent";
+import type { AgentMessage } from "@earendil-works/pi-agent-core";
+/**
+ * The text the LLM sees as a synthetic user message after an
+ * overflow-driven auto-compaction. Tuned to be imperative and brief:
+ *
+ *   - "Continue the task in progress" — open-ended about what the
+ *     next step is. The agent may need to call a tool, write a
+ *     final answer, ask a follow-up question, or anything else the
+ *     task requires. We don't prescribe the shape; we just say
+ *     keep going.
+ *   - "Do not write a summary of what you were doing" — names the
+ *     observed failure mode (model paraphrases the summary's
+ *     "Next Steps" instead of acting).
+ *   - "Pick up from where you left off" — frames the summary as
+ *     reference material, not as a fresh prompt addressed to the
+ *     model.
+ *
+ * Kept under 250 characters so the addition to LLM input is
+ * negligible (~50 tokens).
+ */
+export declare const NUDGE_MESSAGE: string;
+/**
+ * Inspect a messages array and decide whether the post-compaction
+ * nudge should be appended. Exported separately so the unit test can
+ * exercise the boundary cases (empty, only summary, summary then user
+ * prompt, summary then assistant turn) without booting a full SDK
+ * session.
+ */
+export declare function shouldNudgeAfterCompaction(messages: readonly AgentMessage[]): boolean;
+/**
+ * Pi extension factory. Registered via `DefaultResourceLoader`'s
+ * `extensionFactories` option in `session-store.ts`. No I/O,
+ * no async setup — just registers the handler and returns.
+ */
+export declare const compactionContinuationExtension: ExtensionFactory;
+//# sourceMappingURL=compaction-continuation.d.ts.map

package/server/compaction-continuation.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"compaction-continuation.d.ts","sourceRoot":"","sources":["../src/compaction-continuation.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA0EG;AACH,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,iCAAiC,CAAC;AACxE,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,+BAA+B,CAAC;AAElE;;;;;;;;;;;;;;;;;;GAkBG;AACH,eAAO,MAAM,aAAa,QAG0C,CAAC;AAErE;;;;;;GAMG;AACH,wBAAgB,0BAA0B,CAAC,QAAQ,EAAE,SAAS,YAAY,EAAE,GAAG,OAAO,CAIrF;AAiBD;;;;GAIG;AACH,eAAO,MAAM,+BAA+B,EAAE,gBAS7C,CAAC"}

package/server/compaction-continuation.js

@@ -0,0 +1,65 @@
+/**
+ * The text the LLM sees as a synthetic user message after an
+ * overflow-driven auto-compaction. Tuned to be imperative and brief:
+ *
+ *   - "Continue the task in progress" — open-ended about what the
+ *     next step is. The agent may need to call a tool, write a
+ *     final answer, ask a follow-up question, or anything else the
+ *     task requires. We don't prescribe the shape; we just say
+ *     keep going.
+ *   - "Do not write a summary of what you were doing" — names the
+ *     observed failure mode (model paraphrases the summary's
+ *     "Next Steps" instead of acting).
+ *   - "Pick up from where you left off" — frames the summary as
+ *     reference material, not as a fresh prompt addressed to the
+ *     model.
+ *
+ * Kept under 250 characters so the addition to LLM input is
+ * negligible (~50 tokens).
+ */
+export const NUDGE_MESSAGE = "[continuation] Continue the task in progress — pick up from where you left off " +
+    "based on the summary above. Do not write a status update or summary of what you " +
+    "were doing; just proceed with the next action the task requires.";
+/**
+ * Inspect a messages array and decide whether the post-compaction
+ * nudge should be appended. Exported separately so the unit test can
+ * exercise the boundary cases (empty, only summary, summary then user
+ * prompt, summary then assistant turn) without booting a full SDK
+ * session.
+ */
+export function shouldNudgeAfterCompaction(messages) {
+    const last = messages[messages.length - 1];
+    if (last === undefined)
+        return false;
+    return last.role === "compactionSummary";
+}
+/**
+ * Build the synthetic user message used as the nudge. Timestamp is
+ * `Date.now()` so the message sorts after any prior content if a
+ * downstream consumer ever inspects timestamps; in practice the SDK
+ * doesn't reorder by timestamp at LLM-call time, but it costs nothing
+ * to keep monotonic ordering.
+ */
+function buildNudgeMessage() {
+    return {
+        role: "user",
+        content: [{ type: "text", text: NUDGE_MESSAGE }],
+        timestamp: Date.now(),
+    };
+}
+/**
+ * Pi extension factory. Registered via `DefaultResourceLoader`'s
+ * `extensionFactories` option in `session-store.ts`. No I/O,
+ * no async setup — just registers the handler and returns.
+ */
+export const compactionContinuationExtension = (pi) => {
+    pi.on("context", (event) => {
+        if (!shouldNudgeAfterCompaction(event.messages)) {
+            return undefined;
+        }
+        return {
+            messages: [...event.messages, buildNudgeMessage()],
+        };
+    });
+};
+//# sourceMappingURL=compaction-continuation.js.map

package/server/compaction-continuation.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"compaction-continuation.js","sourceRoot":"","sources":["../src/compaction-continuation.ts"],"names":[],"mappings":"AA8EA;;;;;;;;;;;;;;;;;;GAkBG;AACH,MAAM,CAAC,MAAM,aAAa,GACxB,iFAAiF;IACjF,kFAAkF;IAClF,kEAAkE,CAAC;AAErE;;;;;;GAMG;AACH,MAAM,UAAU,0BAA0B,CAAC,QAAiC;IAC1E,MAAM,IAAI,GAAG,QAAQ,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;IAC3C,IAAI,IAAI,KAAK,SAAS;QAAE,OAAO,KAAK,CAAC;IACrC,OAAO,IAAI,CAAC,IAAI,KAAK,mBAAmB,CAAC;AAC3C,CAAC;AAED;;;;;;GAMG;AACH,SAAS,iBAAiB;IACxB,OAAO;QACL,IAAI,EAAE,MAAM;QACZ,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,aAAa,EAAE,CAAC;QAChD,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;KACtB,CAAC;AACJ,CAAC;AAED;;;;GAIG;AACH,MAAM,CAAC,MAAM,+BAA+B,GAAqB,CAAC,EAAE,EAAE,EAAE;IACtE,EAAE,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC,KAAK,EAAE,EAAE;QACzB,IAAI,CAAC,0BAA0B,CAAC,KAAK,CAAC,QAAQ,CAAC,EAAE,CAAC;YAChD,OAAO,SAAS,CAAC;QACnB,CAAC;QACD,OAAO;YACL,QAAQ,EAAE,CAAC,GAAG,KAAK,CAAC,QAAQ,EAAE,iBAAiB,EAAE,CAAC;SACnD,CAAC;IACJ,CAAC,CAAC,CAAC;AACL,CAAC,CAAC"}

package/server/config-store.d.ts

@@ -0,0 +1,97 @@
+interface ProbeRequest {
+    baseUrl: string;
+    apiKey?: string;
+    apiType?: string;
+    headers?: Record<string, string>;
+}
+interface ProbeResult {
+    reachable: boolean;
+    error?: string;
+    detectedApiType?: string;
+    models?: Array<{
+        id: string;
+        name?: string;
+    }>;
+    suggestedName?: string;
+    /** Full response from the models endpoint for debugging */
+    rawResponse?: unknown;
+}
+/**
+ * Probe a provider endpoint to test connectivity, detect API type,
+ * and fetch available models.
+ */
+export declare function probeProvider(req: ProbeRequest): Promise<ProbeResult>;
+/**
+ * Add or update a custom provider in models.json.
+ */
+export declare function addCustomProvider(providerName: string, config: Record<string, unknown>): Promise<{
+    providers: Record<string, unknown>;
+}>;
+/**
+ * Remove a custom provider from models.json.
+ */
+export declare function removeCustomProvider(providerName: string): Promise<{
+    providers: Record<string, unknown>;
+}>;
+export declare class AuthProviderNotFoundError extends Error {
+    constructor(provider: string);
+}
+export interface AuthPresence {
+    configured: boolean;
+    source?: string;
+    label?: string;
+}
+export interface AuthSummary {
+    providers: Record<string, AuthPresence>;
+}
+export declare function readAuthSummary(): AuthSummary;
+export declare function writeApiKey(provider: string, apiKey: string): void;
+export declare function removeApiKey(provider: string): void;
+export declare function readSettings(): Record<string, unknown>;
+export declare function updateSettings(patch: Record<string, unknown>): Record<string, unknown>;
+export interface ModelsJson {
+    providers: Record<string, unknown>;
+}
+/**
+ * Read models.json, redacting secret fields (apiKey, apiKeyCommand).
+ */
+export declare function readModelsJsonRedacted(): Promise<ModelsJson>;
+export declare function writeModelsJson(data: ModelsJson): Promise<void>;
+/**
+ * Write settings.json atomically. Used by the per-session setModel route
+ * to restore the global default after the SDK mutates it as a side effect
+ * of session.setModel().
+ * Uses a temp-file + rename pattern for atomic writes.
+ */
+export declare function writeSettings(settings: Record<string, unknown>): Promise<void>;
+/**
+ * Read the enabledModels array from settings via the SDK SettingsManager.
+ * Returns undefined when no scoping is active (all models visible).
+ */
+export declare function readEnabledModels(): string[] | undefined;
+/**
+ * Persist the enabledModels array to settings. Pass undefined or null
+ * to disable scoping (all models visible).
+ */
+export declare function writeEnabledModels(patterns: string[] | null | undefined): void;
+export interface ProvidersListingOptions {
+    /** When true, only return models listed in enabledModels. */
+    scoped?: boolean;
+}
+export declare function liveProvidersListing(opts?: ProvidersListingOptions): {
+    providers: Array<{
+        provider: string;
+        models: Array<{
+            id: string;
+            name: string;
+            contextWindow: number;
+            maxTokens: number;
+            reasoning: boolean;
+            input: string[];
+            hasAuth: boolean;
+            supportedThinkingLevels: string[];
+        }>;
+    }>;
+};
+export {};
+//# sourceMappingURL=config-store.d.ts.map

package/server/config-store.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"config-store.d.ts","sourceRoot":"","sources":["../src/config-store.ts"],"names":[],"mappings":"AAkBA,UAAU,YAAY;IACpB,OAAO,EAAE,MAAM,CAAC;IAChB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;CAClC;AAED,UAAU,WAAW;IACnB,SAAS,EAAE,OAAO,CAAC;IACnB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,MAAM,CAAC,EAAE,KAAK,CAAC;QAAE,EAAE,EAAE,MAAM,CAAC;QAAC,IAAI,CAAC,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IAC9C,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,2DAA2D;IAC3D,WAAW,CAAC,EAAE,OAAO,CAAC;CACvB;AAED;;;GAGG;AACH,wBAAsB,aAAa,CAAC,GAAG,EAAE,YAAY,GAAG,OAAO,CAAC,WAAW,CAAC,CA+G3E;AAED;;GAEG;AACH,wBAAsB,iBAAiB,CACrC,YAAY,EAAE,MAAM,EACpB,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAC9B,OAAO,CAAC;IAAE,SAAS,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAA;CAAE,CAAC,CAKjD;AAED;;GAEG;AACH,wBAAsB,oBAAoB,CACxC,YAAY,EAAE,MAAM,GACnB,OAAO,CAAC;IAAE,SAAS,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAA;CAAE,CAAC,CAKjD;AAyBD,qBAAa,yBAA0B,SAAQ,KAAK;IAClD,YAAY,QAAQ,EAAE,MAAM,EAG3B;CACF;AAID,MAAM,WAAW,YAAY;IAC3B,UAAU,EAAE,OAAO,CAAC;IACpB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB;AAED,MAAM,WAAW,WAAW;IAC1B,SAAS,EAAE,MAAM,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;CACzC;AAED,wBAAgB,eAAe,IAAI,WAAW,CAU7C;AAED,wBAAgB,WAAW,CAAC,QAAQ,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,GAAG,IAAI,CAGlE;AAED,wBAAgB,YAAY,CAAC,QAAQ,EAAE,MAAM,GAAG,IAAI,CAMnD;AAID,wBAAgB,YAAY,IAAI,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAGtD;AAED,wBAAgB,cAAc,CAAC,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAkCtF;AAID,MAAM,WAAW,UAAU;IACzB,SAAS,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACpC;AAED;;GAEG;AACH,wBAAsB,sBAAsB,IAAI,OAAO,CAAC,UAAU,CAAC,CA6BlE;AAED,wBAAsB,eAAe,CAAC,IAAI,EAAE,UAAU,GAAG,OAAO,CAAC,IAAI,CAAC,CAGrE;AAkBD;;;;;GAKG;AACH,wBAAsB,aAAa,CAAC,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,OAAO,CAAC,IAAI,CAAC,CAGpF;AAID;;;GAGG;AACH,wBAAgB,iBAAiB,IAAI,MAAM,EAAE,GAAG,SAAS,CAGxD;AAED;;;GAGG;AACH,wBAAgB,kBAAkB,CAAC,QAAQ,EAAE,MAAM,EAAE,GAAG,IAAI,GAAG,SAAS,GAAG,IAAI,CAG9E;AAyBD,MAAM,WAAW,uBAAuB;IACtC,6DAA6D;IAC7D,MAAM,CAAC,EAAE,OAAO,CAAC;CAClB;AAED,wBAAgB,oBAAoB,CAAC,IAAI,CAAC,EAAE,uBAAuB,GAAG;IACpE,SAAS,EAAE,KAAK,CAAC;QACf,QAAQ,EAAE,MAAM,CAAC;QACjB,MAAM,EAAE,KAAK,CAAC;YACZ,EAAE,EAAE,MAAM,CAAC;YACX,IAAI,EAAE,MAAM,CAAC;YACb,aAAa,EAAE,MAAM,CAAC;YACtB,SAAS,EAAE,MAAM,CAAC;YAClB,SAAS,EAAE,OAAO,CAAC;YACnB,KAAK,EAAE,MAAM,EAAE,CAAC;YAChB,OAAO,EAAE,OAAO,CAAC;YACjB,uBAAuB,EAAE,MAAM,EAAE,CAAC;SACnC,CAAC,CAAC;KACJ,CAAC,CAAC;CACJ,CAyCA"}