@bastani/atomic 0.8.31-alpha.5 → 0.9.0-alpha.1

package/dist/core/agent-session-compaction.js

@@ -0,0 +1,282 @@
+import { formatNoModelSelectedMessage } from "./auth-guidance.js";
+import { contextCompact as runContextCompact, prepareContextCompaction, validateContextDeletionRequest } from "./compaction/index.js";
+function deepFreeze(value) {
+    if (value && typeof value === "object") {
+        Object.freeze(value);
+        for (const nested of Object.values(value)) {
+            deepFreeze(nested);
+        }
+    }
+    return value;
+}
+export async function _applyContextVerbatimCompaction(options) {
+    if (!this.model) {
+        throw new Error(formatNoModelSelectedMessage());
+    }
+    // Capture the narrowed model now (control-flow narrowing holds immediately after the
+    // guard) so the lazy planner-fallback closure below can use a non-undefined model.
+    const model = this.model;
+    const compactionThinkingLevel = this.thinkingLevel;
+    const pathEntries = this.sessionManager.getBranch();
+    const settings = this.settingsManager.getCompactionSettings();
+    const preparation = prepareContextCompaction(pathEntries, settings, {
+        ...(options.compression_ratio === undefined ? {} : { compression_ratio: options.compression_ratio }),
+        ...(options.preserve_recent === undefined ? {} : { preserve_recent: options.preserve_recent }),
+        ...(options.query === undefined ? {} : { query: options.query }),
+    });
+    if (!preparation) {
+        return undefined;
+    }
+    const parameters = preparation.parameters;
+    // Planner fallback used when no extension supplies a deletionRequest. Auth is resolved
+    // lazily here so extension-provided deletion requests keep working offline. Returns
+    // undefined when auth is unavailable (auto-mode resolvers), signaling a no-op compaction.
+    const runPlanner = async () => {
+        const auth = await options.resolvePlannerAuth();
+        if (!auth)
+            return undefined;
+        return runContextCompact(preparation, model, auth.apiKey, auth.headers, options.abortController.signal, compactionThinkingLevel);
+    };
+    // Emit session_before_compact to allow extensions to cancel or provide a deletion request.
+    // This happens BEFORE any auth resolution so local extension deletion requests work
+    // without configured API credentials.
+    let fromExtension = false;
+    let validated;
+    if (this._extensionRunner.hasHandlers("session_before_compact")) {
+        // Deep-clone the preparation only when a before-compact handler actually exists. Extensions
+        // receive an isolated, frozen snapshot so they cannot mutate protection metadata
+        // (protectedEntryIds, entry .protected flags, etc.) on the internal preparation used for
+        // validation. Building it lazily avoids deep-cloning the transcript — largest exactly when
+        // compaction fires — on the common no-extension path.
+        let extensionPreparation;
+        try {
+            extensionPreparation = deepFreeze(structuredClone(preparation));
+        }
+        catch (error) {
+            // structuredClone only throws if an entry carries a non-cloneable value (a function or a
+            // class instance). Transcript entries are plain data today, so this guards a latent
+            // invariant: surface a clear error instead of letting a raw DataCloneError abort an
+            // otherwise-viable compaction.
+            throw new Error(`Failed to snapshot transcript for compaction extensions: ${error instanceof Error ? error.message : String(error)}`);
+        }
+        const hookResult = (await this._extensionRunner.emit({
+            type: "session_before_compact",
+            reason: options.reason,
+            parameters,
+            preparation: extensionPreparation,
+            branchEntries: pathEntries,
+            signal: options.abortController.signal,
+        }));
+        if (hookResult?.cancel) {
+            throw new Error("Compaction cancelled");
+        }
+        if (hookResult?.deletionRequest) {
+            const extensionDeletionRequest = hookResult.deletionRequest;
+            // Reject empty deletion requests before any side effects (backup, append, rebuild).
+            if (!Array.isArray(extensionDeletionRequest.deletions) || extensionDeletionRequest.deletions.length === 0) {
+                throw new Error("No safe context deletions proposed by extension");
+            }
+            // Validate against the internal transcript snapshot, not the extension-facing clone.
+            // Auth is NOT resolved here — local extension deletion requests work offline.
+            validated = validateContextDeletionRequest(extensionDeletionRequest, preparation.transcript);
+            // Reject if reconciliation reduced deletions to zero.
+            if (validated.deletedTargets.length === 0) {
+                throw new Error("No safe context deletions proposed by extension");
+            }
+            fromExtension = true;
+        }
+    }
+    // Planner fallback shared by both paths: no before-compact handler at all, or a handler that
+    // observed without supplying a deletionRequest. Resolves auth lazily; undefined means auth is
+    // unavailable (auto-mode resolvers), so compaction is a no-op.
+    if (!validated) {
+        const plannerResult = await runPlanner();
+        if (!plannerResult) {
+            return undefined;
+        }
+        validated = plannerResult;
+    }
+    if (options.abortController.signal.aborted) {
+        throw new Error("Compaction cancelled");
+    }
+    const backupPath = this.sessionManager.writeBackupSnapshot(options.backupLabel);
+    const compactionEntryId = this.sessionManager.appendContextCompaction(validated.deletedTargets, validated.protectedEntryIds, validated.stats, backupPath);
+    const sessionContext = this.sessionManager.buildSessionContext();
+    this.agent.state.messages = sessionContext.messages;
+    const result = {
+        ...validated,
+        promptVersion: 1,
+        parameters,
+        ...(backupPath ? { backupPath } : {}),
+    };
+    // Emit session_compact so extensions can observe the validated result. This is a pure
+    // observation hook fired AFTER the compaction has been committed (backup written,
+    // context_compaction entry persisted, active context rebuilt). A misbehaving observer must
+    // never turn a successful, already-persisted compaction into a reported failure, so any
+    // throw is routed to the non-fatal extension-error channel and compaction still reports
+    // success.
+    const contextCompactionEntry = this.sessionManager.getEntry(compactionEntryId);
+    try {
+        await this._extensionRunner.emit({
+            type: "session_compact",
+            reason: options.reason,
+            parameters,
+            result,
+            contextCompactionEntry,
+            fromExtension,
+        });
+    }
+    catch (error) {
+        this._extensionRunner.emitError({
+            extensionPath: "<session_compact>",
+            event: "session_compact",
+            error: error instanceof Error ? error.message : String(error),
+            stack: error instanceof Error ? error.stack : undefined,
+        });
+    }
+    return result;
+}
+/**
+ * Manually compact the session context using deletion-only verbatim context compaction.
+ * Aborts current agent operation first.
+ */
+export async function compact(options = {}) {
+    this._disconnectFromAgent();
+    await this.abort();
+    this._compactionAbortController = new AbortController();
+    this._emit({ type: "compaction_start", reason: "manual" });
+    try {
+        if (!this.model) {
+            throw new Error(formatNoModelSelectedMessage());
+        }
+        // Auth is resolved lazily: only called when the planner fallback is needed.
+        // Extensions that provide a deletionRequest work without configured credentials.
+        const model = this.model;
+        const result = await this._applyContextVerbatimCompaction({
+            resolvePlannerAuth: () => this._getRequiredRequestAuth(model),
+            abortController: this._compactionAbortController,
+            backupLabel: "compact",
+            reason: "manual",
+            ...(options.compression_ratio === undefined ? {} : { compression_ratio: options.compression_ratio }),
+            ...(options.preserve_recent === undefined ? {} : { preserve_recent: options.preserve_recent }),
+            ...(options.query === undefined ? {} : { query: options.query }),
+        });
+        if (!result) {
+            throw new Error("Nothing to compact (session too small)");
+        }
+        this._emit({
+            type: "compaction_end",
+            reason: "manual",
+            result,
+            aborted: false,
+            willRetry: false,
+        });
+        return result;
+    }
+    catch (error) {
+        const message = error instanceof Error ? error.message : String(error);
+        const aborted = message === "Compaction cancelled" || (error instanceof Error && error.name === "AbortError");
+        this._emit({
+            type: "compaction_end",
+            reason: "manual",
+            result: undefined,
+            aborted,
+            willRetry: false,
+            errorMessage: aborted ? undefined : `Compaction failed: ${message}`,
+        });
+        throw error;
+    }
+    finally {
+        this._compactionAbortController = undefined;
+        this._reconnectToAgent();
+    }
+}
+/**
+ * Manually compact the session context by applying validated logical deletions.
+ * Retained transcript entries/content blocks stay verbatim; no user prompt text is accepted.
+ */
+export async function contextCompact() {
+    this._disconnectFromAgent();
+    await this.abort();
+    this._compactionAbortController = new AbortController();
+    this._emit({ type: "context_compaction_start", reason: "manual" });
+    try {
+        if (!this.model) {
+            throw new Error(formatNoModelSelectedMessage());
+        }
+        // Auth is resolved lazily: only called when the planner fallback is needed.
+        // Extensions that provide a deletionRequest work without configured credentials.
+        const model = this.model;
+        const result = await this._applyContextVerbatimCompaction({
+            resolvePlannerAuth: () => this._getRequiredRequestAuth(model),
+            abortController: this._compactionAbortController,
+            backupLabel: "context-compact",
+            reason: "manual",
+        });
+        if (!result) {
+            throw new Error("Nothing to context-compact (session too small)");
+        }
+        this._emit({
+            type: "context_compaction_end",
+            reason: "manual",
+            result,
+            aborted: false,
+            willRetry: false,
+        });
+        return result;
+    }
+    catch (error) {
+        const message = error instanceof Error ? error.message : String(error);
+        const aborted = message === "Compaction cancelled" || (error instanceof Error && error.name === "AbortError");
+        this._emit({
+            type: "context_compaction_end",
+            reason: "manual",
+            result: undefined,
+            aborted,
+            willRetry: false,
+            errorMessage: aborted ? undefined : `Context compaction failed: ${message}`,
+        });
+        throw error;
+    }
+    finally {
+        this._compactionAbortController = undefined;
+        this._reconnectToAgent();
+    }
+}
+/**
+ * Cancel in-progress compaction (manual or auto).
+ */
+export function abortCompaction() {
+    this._compactionAbortController?.abort();
+    this._autoCompactionAbortController?.abort();
+}
+/**
+ * Cancel in-progress branch summarization.
+ */
+export function abortBranchSummary() {
+    this._branchSummaryAbortController?.abort();
+}
+/**
+ * Check if compaction is needed and run it.
+ * Called after agent_end and before prompt submission.
+ *
+ * Two cases:
+ * 1. Overflow: LLM returned context overflow error, remove error message from agent state, compact, auto-retry
+ * 2. Threshold: Context over threshold, compact, resume queued active-turn work if present; otherwise wait for user
+ *
+ * @param assistantMessage The assistant message to check
+ * @param skipAbortedCheck If false, include aborted messages (for pre-prompt check). Default: true
+ */
+export function setAutoCompactionEnabled(enabled) {
+    this.settingsManager.setCompactionEnabled(enabled);
+}
+/** Whether auto-compaction is enabled */
+export const agentSessionCompactionMethods = {
+    _applyContextVerbatimCompaction,
+    compact,
+    contextCompact,
+    abortCompaction,
+    abortBranchSummary,
+    setAutoCompactionEnabled,
+};
+//# sourceMappingURL=agent-session-compaction.js.map

package/dist/core/agent-session-compaction.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"agent-session-compaction.js","sourceRoot":"","sources":["../../src/core/agent-session-compaction.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,4BAA4B,EAAE,MAAM,oBAAoB,CAAC;AAGlE,OAAO,EAAuK,cAAc,IAAI,iBAAiB,EAAE,wBAAwB,EAAE,8BAA8B,EAAE,MAAM,uBAAuB,CAAC;AAI3S,SAAS,UAAU,CAAI,KAAQ;IAC9B,IAAI,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QACxC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QACrB,KAAK,MAAM,MAAM,IAAI,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC;YAC3C,UAAU,CAAC,MAAM,CAAC,CAAC;QACpB,CAAC;IACF,CAAC;IACD,OAAO,KAAK,CAAC;AACd,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,+BAA+B,CAEpD,OAAsC;IAEtC,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC;QACjB,MAAM,IAAI,KAAK,CAAC,4BAA4B,EAAE,CAAC,CAAC;IACjD,CAAC;IACD,qFAAqF;IACrF,mFAAmF;IACnF,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC;IACzB,MAAM,uBAAuB,GAAG,IAAI,CAAC,aAAa,CAAC;IAEnD,MAAM,WAAW,GAAG,IAAI,CAAC,cAAc,CAAC,SAAS,EAAE,CAAC;IACpD,MAAM,QAAQ,GAAG,IAAI,CAAC,eAAe,CAAC,qBAAqB,EAAE,CAAC;IAC9D,MAAM,WAAW,GAAG,wBAAwB,CAAC,WAAW,EAAE,QAAQ,EAAE;QACnE,GAAG,CAAC,OAAO,CAAC,iBAAiB,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,iBAAiB,EAAE,OAAO,CAAC,iBAAiB,EAAE,CAAC;QACpG,GAAG,CAAC,OAAO,CAAC,eAAe,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,eAAe,EAAE,OAAO,CAAC,eAAe,EAAE,CAAC;QAC9F,GAAG,CAAC,OAAO,CAAC,KAAK,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,OAAO,CAAC,KAAK,EAAE,CAAC;KAChE,CAAC,CAAC;IACH,IAAI,CAAC,WAAW,EAAE,CAAC;QAClB,OAAO,SAAS,CAAC;IAClB,CAAC;IACD,MAAM,UAAU,GAAgC,WAAW,CAAC,UAAU,CAAC;IAEvE,uFAAuF;IACvF,oFAAoF;IACpF,0FAA0F;IAC1F,MAAM,UAAU,GAAG,KAAK,IAAyD,EAAE;QAClF,MAAM,IAAI,GAAG,MAAM,OAAO,CAAC,kBAAkB,EAAE,CAAC;QAChD,IAAI,CAAC,IAAI;YAAE,OAAO,SAAS,CAAC;QAC5B,OAAO,iBAAiB,CACvB,WAAW,EACX,KAAK,EACL,IAAI,CAAC,MAAM,EACX,IAAI,CAAC,OAAO,EACZ,OAAO,CAAC,eAAe,CAAC,MAAM,EAC9B,uBAAuB,CACvB,CAAC;IACH,CAAC,CAAC;IAEF,2FAA2F;IAC3F,oFAAoF;IACpF,sCAAsC;IACtC,IAAI,aAAa,GAAG,KAAK,CAAC;IAC1B,IAAI,SAAqD,CAAC;IAE1D,IAAI,IAAI,CAAC,gBAAgB,CAAC,WAAW,CAAC,wBAAwB,CAAC,EAAE,CAAC;QACjE,4FAA4F;QAC5F,iFAAiF;QACjF,yFAAyF;QACzF,2FAA2F;QAC3F,sDAAsD;QACtD,IAAI,oBAAkD,CAAC;QACvD,IAAI,CAAC;YACJ,oBAAoB,GAAG,UAAU,CAAC,eAAe,CAAC,WAAW,CAAC,CAAC,CAAC;QACjE,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAChB,yFAAyF;YACzF,oFAAoF;YACpF,oFAAoF;YACpF,+BAA+B;YAC/B,MAAM,IAAI,KAAK,CACd,4DAA4D,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CACpH,CAAC;QACH,CAAC;QAED,MAAM,UAAU,GAAG,CAAC,MAAM,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC;YACpD,IAAI,EAAE,wBAAwB;YAC9B,MAAM,EAAE,OAAO,CAAC,MAAM;YACtB,UAAU;YACV,WAAW,EAAE,oBAAoB;YACjC,aAAa,EAAE,WAAW;YAC1B,MAAM,EAAE,OAAO,CAAC,eAAe,CAAC,MAAM;SACF,CAAC,CAA2C,CAAC;QAElF,IAAI,UAAU,EAAE,MAAM,EAAE,CAAC;YACxB,MAAM,IAAI,KAAK,CAAC,sBAAsB,CAAC,CAAC;QACzC,CAAC;QAED,IAAI,UAAU,EAAE,eAAe,EAAE,CAAC;YACjC,MAAM,wBAAwB,GAAG,UAAU,CAAC,eAAyC,CAAC;YACtF,oFAAoF;YACpF,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,wBAAwB,CAAC,SAAS,CAAC,IAAI,wBAAwB,CAAC,SAAS,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBAC3G,MAAM,IAAI,KAAK,CAAC,iDAAiD,CAAC,CAAC;YACpE,CAAC;YACD,qFAAqF;YACrF,8EAA8E;YAC9E,SAAS,GAAG,8BAA8B,CAAC,wBAAwB,EAAE,WAAW,CAAC,UAAU,CAAC,CAAC;YAC7F,sDAAsD;YACtD,IAAI,SAAS,CAAC,cAAc,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBAC3C,MAAM,IAAI,KAAK,CAAC,iDAAiD,CAAC,CAAC;YACpE,CAAC;YACD,aAAa,GAAG,IAAI,CAAC;QACtB,CAAC;IACF,CAAC;IAED,6FAA6F;IAC7F,8FAA8F;IAC9F,+DAA+D;IAC/D,IAAI,CAAC,SAAS,EAAE,CAAC;QAChB,MAAM,aAAa,GAAG,MAAM,UAAU,EAAE,CAAC;QACzC,IAAI,CAAC,aAAa,EAAE,CAAC;YACpB,OAAO,SAAS,CAAC;QAClB,CAAC;QACD,SAAS,GAAG,aAAa,CAAC;IAC3B,CAAC;IAED,IAAI,OAAO,CAAC,eAAe,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;QAC5C,MAAM,IAAI,KAAK,CAAC,sBAAsB,CAAC,CAAC;IACzC,CAAC;IAED,MAAM,UAAU,GAAG,IAAI,CAAC,cAAc,CAAC,mBAAmB,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;IAChF,MAAM,iBAAiB,GAAG,IAAI,CAAC,cAAc,CAAC,uBAAuB,CACpE,SAAS,CAAC,cAAc,EACxB,SAAS,CAAC,iBAAiB,EAC3B,SAAS,CAAC,KAAK,EACf,UAAU,CACV,CAAC;IACF,MAAM,cAAc,GAAG,IAAI,CAAC,cAAc,CAAC,mBAAmB,EAAE,CAAC;IACjE,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,QAAQ,GAAG,cAAc,CAAC,QAAQ,CAAC;IAEpD,MAAM,MAAM,GAA4B;QACvC,GAAG,SAAS;QACZ,aAAa,EAAE,CAAC;QAChB,UAAU;QACV,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KACrC,CAAC;IAEF,sFAAsF;IACtF,kFAAkF;IAClF,2FAA2F;IAC3F,wFAAwF;IACxF,wFAAwF;IACxF,WAAW;IACX,MAAM,sBAAsB,GAAG,IAAI,CAAC,cAAc,CAAC,QAAQ,CAAC,iBAAiB,CAA2B,CAAC;IACzG,IAAI,CAAC;QACJ,MAAM,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC;YAChC,IAAI,EAAE,iBAAiB;YACvB,MAAM,EAAE,OAAO,CAAC,MAAM;YACtB,UAAU;YACV,MAAM;YACN,sBAAsB;YACtB,aAAa;SACiB,CAAC,CAAC;IAClC,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QAChB,IAAI,CAAC,gBAAgB,CAAC,SAAS,CAAC;YAC/B,aAAa,EAAE,mBAAmB;YAClC,KAAK,EAAE,iBAAiB;YACxB,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;YAC7D,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS;SACvD,CAAC,CAAC;IACJ,CAAC;IAED,OAAO,MAAM,CAAC;AACf,CAAC;AAED;;;GAGG;AAEH,MAAM,CAAC,KAAK,UAAU,OAAO,CAAqB,OAAO,GAAyC,EAAE;IACnG,IAAI,CAAC,oBAAoB,EAAE,CAAC;IAC5B,MAAM,IAAI,CAAC,KAAK,EAAE,CAAC;IACnB,IAAI,CAAC,0BAA0B,GAAG,IAAI,eAAe,EAAE,CAAC;IACxD,IAAI,CAAC,KAAK,CAAC,EAAE,IAAI,EAAE,kBAAkB,EAAE,MAAM,EAAE,QAAQ,EAAE,CAAC,CAAC;IAE3D,IAAI,CAAC;QACJ,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CAAC,4BAA4B,EAAE,CAAC,CAAC;QACjD,CAAC;QAED,4EAA4E;QAC5E,iFAAiF;QACjF,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC;QACzB,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,+BAA+B,CAAC;YACzD,kBAAkB,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,uBAAuB,CAAC,KAAK,CAAC;YAC7D,eAAe,EAAE,IAAI,CAAC,0BAA0B;YAChD,WAAW,EAAE,SAAS;YACtB,MAAM,EAAE,QAAQ;YAChB,GAAG,CAAC,OAAO,CAAC,iBAAiB,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,iBAAiB,EAAE,OAAO,CAAC,iBAAiB,EAAE,CAAC;YACpG,GAAG,CAAC,OAAO,CAAC,eAAe,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,eAAe,EAAE,OAAO,CAAC,eAAe,EAAE,CAAC;YAC9F,GAAG,CAAC,OAAO,CAAC,KAAK,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,OAAO,CAAC,KAAK,EAAE,CAAC;SAChE,CAAC,CAAC;QACH,IAAI,CAAC,MAAM,EAAE,CAAC;YACb,MAAM,IAAI,KAAK,CAAC,wCAAwC,CAAC,CAAC;QAC3D,CAAC;QAED,IAAI,CAAC,KAAK,CAAC;YACV,IAAI,EAAE,gBAAgB;YACtB,MAAM,EAAE,QAAQ;YAChB,MAAM;YACN,OAAO,EAAE,KAAK;YACd,SAAS,EAAE,KAAK;SAChB,CAAC,CAAC;QACH,OAAO,MAAM,CAAC;IACf,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QAChB,MAAM,OAAO,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QACvE,MAAM,OAAO,GAAG,OAAO,KAAK,sBAAsB,IAAI,CAAC,KAAK,YAAY,KAAK,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY,CAAC,CAAC;QAC9G,IAAI,CAAC,KAAK,CAAC;YACV,IAAI,EAAE,gBAAgB;YACtB,MAAM,EAAE,QAAQ;YAChB,MAAM,EAAE,SAAS;YACjB,OAAO;YACP,SAAS,EAAE,KAAK;YAChB,YAAY,EAAE,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,sBAAsB,OAAO,EAAE;SACnE,CAAC,CAAC;QACH,MAAM,KAAK,CAAC;IACb,CAAC;YAAS,CAAC;QACV,IAAI,CAAC,0BAA0B,GAAG,SAAS,CAAC;QAC5C,IAAI,CAAC,iBAAiB,EAAE,CAAC;IAC1B,CAAC;AACF,CAAC;AAED;;;GAGG;AAEH,MAAM,CAAC,KAAK,UAAU,cAAc;IACnC,IAAI,CAAC,oBAAoB,EAAE,CAAC;IAC5B,MAAM,IAAI,CAAC,KAAK,EAAE,CAAC;IACnB,IAAI,CAAC,0BAA0B,GAAG,IAAI,eAAe,EAAE,CAAC;IACxD,IAAI,CAAC,KAAK,CAAC,EAAE,IAAI,EAAE,0BAA0B,EAAE,MAAM,EAAE,QAAQ,EAAE,CAAC,CAAC;IAEnE,IAAI,CAAC;QACJ,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CAAC,4BAA4B,EAAE,CAAC,CAAC;QACjD,CAAC;QAED,4EAA4E;QAC5E,iFAAiF;QACjF,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC;QACzB,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,+BAA+B,CAAC;YACzD,kBAAkB,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,uBAAuB,CAAC,KAAK,CAAC;YAC7D,eAAe,EAAE,IAAI,CAAC,0BAA0B;YAChD,WAAW,EAAE,iBAAiB;YAC9B,MAAM,EAAE,QAAQ;SAChB,CAAC,CAAC;QACH,IAAI,CAAC,MAAM,EAAE,CAAC;YACb,MAAM,IAAI,KAAK,CAAC,gDAAgD,CAAC,CAAC;QACnE,CAAC;QAED,IAAI,CAAC,KAAK,CAAC;YACV,IAAI,EAAE,wBAAwB;YAC9B,MAAM,EAAE,QAAQ;YAChB,MAAM;YACN,OAAO,EAAE,KAAK;YACd,SAAS,EAAE,KAAK;SAChB,CAAC,CAAC;QACH,OAAO,MAAM,CAAC;IACf,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QAChB,MAAM,OAAO,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QACvE,MAAM,OAAO,GAAG,OAAO,KAAK,sBAAsB,IAAI,CAAC,KAAK,YAAY,KAAK,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY,CAAC,CAAC;QAC9G,IAAI,CAAC,KAAK,CAAC;YACV,IAAI,EAAE,wBAAwB;YAC9B,MAAM,EAAE,QAAQ;YAChB,MAAM,EAAE,SAAS;YACjB,OAAO;YACP,SAAS,EAAE,KAAK;YAChB,YAAY,EAAE,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,8BAA8B,OAAO,EAAE;SAC3E,CAAC,CAAC;QACH,MAAM,KAAK,CAAC;IACb,CAAC;YAAS,CAAC;QACV,IAAI,CAAC,0BAA0B,GAAG,SAAS,CAAC;QAC5C,IAAI,CAAC,iBAAiB,EAAE,CAAC;IAC1B,CAAC;AACF,CAAC;AAED;;GAEG;AAEH,MAAM,UAAU,eAAe;IAC9B,IAAI,CAAC,0BAA0B,EAAE,KAAK,EAAE,CAAC;IACzC,IAAI,CAAC,8BAA8B,EAAE,KAAK,EAAE,CAAC;AAC9C,CAAC;AAED;;GAEG;AAEH,MAAM,UAAU,kBAAkB;IACjC,IAAI,CAAC,6BAA6B,EAAE,KAAK,EAAE,CAAC;AAC7C,CAAC;AAED;;;;;;;;;;GAUG;AAEH,MAAM,UAAU,wBAAwB,CAAqB,OAAgB;IAC5E,IAAI,CAAC,eAAe,CAAC,oBAAoB,CAAC,OAAO,CAAC,CAAC;AACpD,CAAC;AAED,yCAAyC;AAEzC,MAAM,CAAC,MAAM,6BAA6B,GAAG;IAC5C,+BAA+B;IAC/B,OAAO;IACP,cAAc;IACd,eAAe;IACf,kBAAkB;IAClB,wBAAwB;CACxB,CAAC","sourcesContent":["import { formatNoModelSelectedMessage } from \"./auth-guidance.ts\";\nimport type { AgentSessionInternalSurface as AgentSession } from \"./agent-session-methods.ts\";\nimport type { ContextCompactionApplyOptions } from \"./agent-session-methods.ts\";\nimport { type ContextCompactionParameters, type ContextCompactionPreparation, type ContextCompactionResult, type ContextDeletionRequest, type ValidatedContextDeletionResult, contextCompact as runContextCompact, prepareContextCompaction, validateContextDeletionRequest } from \"./compaction/index.ts\";\nimport type { SessionBeforeCompactEvent, SessionBeforeCompactResult, SessionCompactEvent } from \"./extensions/index.ts\";\nimport type { ContextCompactionEntry } from \"./session-manager.ts\";\n\nfunction deepFreeze<T>(value: T): T {\n\tif (value && typeof value === \"object\") {\n\t\tObject.freeze(value);\n\t\tfor (const nested of Object.values(value)) {\n\t\t\tdeepFreeze(nested);\n\t\t}\n\t}\n\treturn value;\n}\n\nexport async function _applyContextVerbatimCompaction(\n\tthis: AgentSession,\n\toptions: ContextCompactionApplyOptions,\n): Promise<ContextCompactionResult | undefined> {\n\tif (!this.model) {\n\t\tthrow new Error(formatNoModelSelectedMessage());\n\t}\n\t// Capture the narrowed model now (control-flow narrowing holds immediately after the\n\t// guard) so the lazy planner-fallback closure below can use a non-undefined model.\n\tconst model = this.model;\n\tconst compactionThinkingLevel = this.thinkingLevel;\n\n\tconst pathEntries = this.sessionManager.getBranch();\n\tconst settings = this.settingsManager.getCompactionSettings();\n\tconst preparation = prepareContextCompaction(pathEntries, settings, {\n\t\t...(options.compression_ratio === undefined ? {} : { compression_ratio: options.compression_ratio }),\n\t\t...(options.preserve_recent === undefined ? {} : { preserve_recent: options.preserve_recent }),\n\t\t...(options.query === undefined ? {} : { query: options.query }),\n\t});\n\tif (!preparation) {\n\t\treturn undefined;\n\t}\n\tconst parameters: ContextCompactionParameters = preparation.parameters;\n\n\t// Planner fallback used when no extension supplies a deletionRequest. Auth is resolved\n\t// lazily here so extension-provided deletion requests keep working offline. Returns\n\t// undefined when auth is unavailable (auto-mode resolvers), signaling a no-op compaction.\n\tconst runPlanner = async (): Promise<ValidatedContextDeletionResult | undefined> => {\n\t\tconst auth = await options.resolvePlannerAuth();\n\t\tif (!auth) return undefined;\n\t\treturn runContextCompact(\n\t\t\tpreparation,\n\t\t\tmodel,\n\t\t\tauth.apiKey,\n\t\t\tauth.headers,\n\t\t\toptions.abortController.signal,\n\t\t\tcompactionThinkingLevel,\n\t\t);\n\t};\n\n\t// Emit session_before_compact to allow extensions to cancel or provide a deletion request.\n\t// This happens BEFORE any auth resolution so local extension deletion requests work\n\t// without configured API credentials.\n\tlet fromExtension = false;\n\tlet validated: ValidatedContextDeletionResult | undefined;\n\n\tif (this._extensionRunner.hasHandlers(\"session_before_compact\")) {\n\t\t// Deep-clone the preparation only when a before-compact handler actually exists. Extensions\n\t\t// receive an isolated, frozen snapshot so they cannot mutate protection metadata\n\t\t// (protectedEntryIds, entry .protected flags, etc.) on the internal preparation used for\n\t\t// validation. Building it lazily avoids deep-cloning the transcript — largest exactly when\n\t\t// compaction fires — on the common no-extension path.\n\t\tlet extensionPreparation: ContextCompactionPreparation;\n\t\ttry {\n\t\t\textensionPreparation = deepFreeze(structuredClone(preparation));\n\t\t} catch (error) {\n\t\t\t// structuredClone only throws if an entry carries a non-cloneable value (a function or a\n\t\t\t// class instance). Transcript entries are plain data today, so this guards a latent\n\t\t\t// invariant: surface a clear error instead of letting a raw DataCloneError abort an\n\t\t\t// otherwise-viable compaction.\n\t\t\tthrow new Error(\n\t\t\t\t`Failed to snapshot transcript for compaction extensions: ${error instanceof Error ? error.message : String(error)}`,\n\t\t\t);\n\t\t}\n\n\t\tconst hookResult = (await this._extensionRunner.emit({\n\t\t\ttype: \"session_before_compact\",\n\t\t\treason: options.reason,\n\t\t\tparameters,\n\t\t\tpreparation: extensionPreparation,\n\t\t\tbranchEntries: pathEntries,\n\t\t\tsignal: options.abortController.signal,\n\t\t} satisfies SessionBeforeCompactEvent)) as SessionBeforeCompactResult | undefined;\n\n\t\tif (hookResult?.cancel) {\n\t\t\tthrow new Error(\"Compaction cancelled\");\n\t\t}\n\n\t\tif (hookResult?.deletionRequest) {\n\t\t\tconst extensionDeletionRequest = hookResult.deletionRequest as ContextDeletionRequest;\n\t\t\t// Reject empty deletion requests before any side effects (backup, append, rebuild).\n\t\t\tif (!Array.isArray(extensionDeletionRequest.deletions) || extensionDeletionRequest.deletions.length === 0) {\n\t\t\t\tthrow new Error(\"No safe context deletions proposed by extension\");\n\t\t\t}\n\t\t\t// Validate against the internal transcript snapshot, not the extension-facing clone.\n\t\t\t// Auth is NOT resolved here — local extension deletion requests work offline.\n\t\t\tvalidated = validateContextDeletionRequest(extensionDeletionRequest, preparation.transcript);\n\t\t\t// Reject if reconciliation reduced deletions to zero.\n\t\t\tif (validated.deletedTargets.length === 0) {\n\t\t\t\tthrow new Error(\"No safe context deletions proposed by extension\");\n\t\t\t}\n\t\t\tfromExtension = true;\n\t\t}\n\t}\n\n\t// Planner fallback shared by both paths: no before-compact handler at all, or a handler that\n\t// observed without supplying a deletionRequest. Resolves auth lazily; undefined means auth is\n\t// unavailable (auto-mode resolvers), so compaction is a no-op.\n\tif (!validated) {\n\t\tconst plannerResult = await runPlanner();\n\t\tif (!plannerResult) {\n\t\t\treturn undefined;\n\t\t}\n\t\tvalidated = plannerResult;\n\t}\n\n\tif (options.abortController.signal.aborted) {\n\t\tthrow new Error(\"Compaction cancelled\");\n\t}\n\n\tconst backupPath = this.sessionManager.writeBackupSnapshot(options.backupLabel);\n\tconst compactionEntryId = this.sessionManager.appendContextCompaction(\n\t\tvalidated.deletedTargets,\n\t\tvalidated.protectedEntryIds,\n\t\tvalidated.stats,\n\t\tbackupPath,\n\t);\n\tconst sessionContext = this.sessionManager.buildSessionContext();\n\tthis.agent.state.messages = sessionContext.messages;\n\n\tconst result: ContextCompactionResult = {\n\t\t...validated,\n\t\tpromptVersion: 1,\n\t\tparameters,\n\t\t...(backupPath ? { backupPath } : {}),\n\t};\n\n\t// Emit session_compact so extensions can observe the validated result. This is a pure\n\t// observation hook fired AFTER the compaction has been committed (backup written,\n\t// context_compaction entry persisted, active context rebuilt). A misbehaving observer must\n\t// never turn a successful, already-persisted compaction into a reported failure, so any\n\t// throw is routed to the non-fatal extension-error channel and compaction still reports\n\t// success.\n\tconst contextCompactionEntry = this.sessionManager.getEntry(compactionEntryId) as ContextCompactionEntry;\n\ttry {\n\t\tawait this._extensionRunner.emit({\n\t\t\ttype: \"session_compact\",\n\t\t\treason: options.reason,\n\t\t\tparameters,\n\t\t\tresult,\n\t\t\tcontextCompactionEntry,\n\t\t\tfromExtension,\n\t\t} satisfies SessionCompactEvent);\n\t} catch (error) {\n\t\tthis._extensionRunner.emitError({\n\t\t\textensionPath: \"<session_compact>\",\n\t\t\tevent: \"session_compact\",\n\t\t\terror: error instanceof Error ? error.message : String(error),\n\t\t\tstack: error instanceof Error ? error.stack : undefined,\n\t\t});\n\t}\n\n\treturn result;\n}\n\n/**\n * Manually compact the session context using deletion-only verbatim context compaction.\n * Aborts current agent operation first.\n */\n\nexport async function compact(this: AgentSession, options: Partial<ContextCompactionParameters> = {}): Promise<ContextCompactionResult> {\n\tthis._disconnectFromAgent();\n\tawait this.abort();\n\tthis._compactionAbortController = new AbortController();\n\tthis._emit({ type: \"compaction_start\", reason: \"manual\" });\n\n\ttry {\n\t\tif (!this.model) {\n\t\t\tthrow new Error(formatNoModelSelectedMessage());\n\t\t}\n\n\t\t// Auth is resolved lazily: only called when the planner fallback is needed.\n\t\t// Extensions that provide a deletionRequest work without configured credentials.\n\t\tconst model = this.model;\n\t\tconst result = await this._applyContextVerbatimCompaction({\n\t\t\tresolvePlannerAuth: () => this._getRequiredRequestAuth(model),\n\t\t\tabortController: this._compactionAbortController,\n\t\t\tbackupLabel: \"compact\",\n\t\t\treason: \"manual\",\n\t\t\t...(options.compression_ratio === undefined ? {} : { compression_ratio: options.compression_ratio }),\n\t\t\t...(options.preserve_recent === undefined ? {} : { preserve_recent: options.preserve_recent }),\n\t\t\t...(options.query === undefined ? {} : { query: options.query }),\n\t\t});\n\t\tif (!result) {\n\t\t\tthrow new Error(\"Nothing to compact (session too small)\");\n\t\t}\n\n\t\tthis._emit({\n\t\t\ttype: \"compaction_end\",\n\t\t\treason: \"manual\",\n\t\t\tresult,\n\t\t\taborted: false,\n\t\t\twillRetry: false,\n\t\t});\n\t\treturn result;\n\t} catch (error) {\n\t\tconst message = error instanceof Error ? error.message : String(error);\n\t\tconst aborted = message === \"Compaction cancelled\" || (error instanceof Error && error.name === \"AbortError\");\n\t\tthis._emit({\n\t\t\ttype: \"compaction_end\",\n\t\t\treason: \"manual\",\n\t\t\tresult: undefined,\n\t\t\taborted,\n\t\t\twillRetry: false,\n\t\t\terrorMessage: aborted ? undefined : `Compaction failed: ${message}`,\n\t\t});\n\t\tthrow error;\n\t} finally {\n\t\tthis._compactionAbortController = undefined;\n\t\tthis._reconnectToAgent();\n\t}\n}\n\n/**\n * Manually compact the session context by applying validated logical deletions.\n * Retained transcript entries/content blocks stay verbatim; no user prompt text is accepted.\n */\n\nexport async function contextCompact(this: AgentSession): Promise<ContextCompactionResult> {\n\tthis._disconnectFromAgent();\n\tawait this.abort();\n\tthis._compactionAbortController = new AbortController();\n\tthis._emit({ type: \"context_compaction_start\", reason: \"manual\" });\n\n\ttry {\n\t\tif (!this.model) {\n\t\t\tthrow new Error(formatNoModelSelectedMessage());\n\t\t}\n\n\t\t// Auth is resolved lazily: only called when the planner fallback is needed.\n\t\t// Extensions that provide a deletionRequest work without configured credentials.\n\t\tconst model = this.model;\n\t\tconst result = await this._applyContextVerbatimCompaction({\n\t\t\tresolvePlannerAuth: () => this._getRequiredRequestAuth(model),\n\t\t\tabortController: this._compactionAbortController,\n\t\t\tbackupLabel: \"context-compact\",\n\t\t\treason: \"manual\",\n\t\t});\n\t\tif (!result) {\n\t\t\tthrow new Error(\"Nothing to context-compact (session too small)\");\n\t\t}\n\n\t\tthis._emit({\n\t\t\ttype: \"context_compaction_end\",\n\t\t\treason: \"manual\",\n\t\t\tresult,\n\t\t\taborted: false,\n\t\t\twillRetry: false,\n\t\t});\n\t\treturn result;\n\t} catch (error) {\n\t\tconst message = error instanceof Error ? error.message : String(error);\n\t\tconst aborted = message === \"Compaction cancelled\" || (error instanceof Error && error.name === \"AbortError\");\n\t\tthis._emit({\n\t\t\ttype: \"context_compaction_end\",\n\t\t\treason: \"manual\",\n\t\t\tresult: undefined,\n\t\t\taborted,\n\t\t\twillRetry: false,\n\t\t\terrorMessage: aborted ? undefined : `Context compaction failed: ${message}`,\n\t\t});\n\t\tthrow error;\n\t} finally {\n\t\tthis._compactionAbortController = undefined;\n\t\tthis._reconnectToAgent();\n\t}\n}\n\n/**\n * Cancel in-progress compaction (manual or auto).\n */\n\nexport function abortCompaction(this: AgentSession): void {\n\tthis._compactionAbortController?.abort();\n\tthis._autoCompactionAbortController?.abort();\n}\n\n/**\n * Cancel in-progress branch summarization.\n */\n\nexport function abortBranchSummary(this: AgentSession): void {\n\tthis._branchSummaryAbortController?.abort();\n}\n\n/**\n * Check if compaction is needed and run it.\n * Called after agent_end and before prompt submission.\n *\n * Two cases:\n * 1. Overflow: LLM returned context overflow error, remove error message from agent state, compact, auto-retry\n * 2. Threshold: Context over threshold, compact, resume queued active-turn work if present; otherwise wait for user\n *\n * @param assistantMessage The assistant message to check\n * @param skipAbortedCheck If false, include aborted messages (for pre-prompt check). Default: true\n */\n\nexport function setAutoCompactionEnabled(this: AgentSession, enabled: boolean): void {\n\tthis.settingsManager.setCompactionEnabled(enabled);\n}\n\n/** Whether auto-compaction is enabled */\n\nexport const agentSessionCompactionMethods = {\n\t_applyContextVerbatimCompaction,\n\tcompact,\n\tcontextCompact,\n\tabortCompaction,\n\tabortBranchSummary,\n\tsetAutoCompactionEnabled,\n};\n"]}

package/dist/core/agent-session-events.d.ts

@@ -0,0 +1,65 @@
+import type { AgentEvent, AgentMessage } from "@earendil-works/pi-agent-core";
+import type { AssistantMessage, Message } from "@earendil-works/pi-ai";
+import type { AgentSessionInternalSurface as AgentSession } from "./agent-session-methods.ts";
+import { type AgentSessionEvent, type AgentSessionEventListener } from "./agent-session-types.ts";
+export declare function _emit(this: AgentSession, event: AgentSessionEvent): void;
+export declare function _emitQueueUpdate(this: AgentSession): void;
+/** Internal handler for agent events - shared by subscribe and reconnect */
+export declare function _handleAgentEvent(this: AgentSession, event: AgentEvent): void;
+export declare function _createRetryPromiseForAgentEnd(this: AgentSession, event: AgentEvent): void;
+export declare function _findLastAssistantInMessages(this: AgentSession, messages: AgentMessage[]): AssistantMessage | undefined;
+export declare function _processAgentEvent(this: AgentSession, event: AgentEvent): Promise<void>;
+export declare function _applyInterruptAbortMessage(this: AgentSession, event: AgentEvent): void;
+export declare function _applyProviderErrorGuidance(this: AgentSession, event: AgentEvent): void;
+/** Resolve the pending retry promise */
+export declare function _resolveRetry(this: AgentSession): void;
+/** Extract text content from a message */
+export declare function _getUserMessageText(this: AgentSession, message: Message): string;
+/** Find the last assistant message in agent state (including aborted ones) */
+export declare function _findLastAssistantMessage(this: AgentSession): AssistantMessage | undefined;
+export declare function _replaceMessageInPlace(this: AgentSession, target: AgentMessage, replacement: AgentMessage): void;
+/** Emit extension events based on agent events */
+export declare function _emitExtensionEvent(this: AgentSession, event: AgentEvent): Promise<void>;
+/**
+ * Subscribe to agent events.
+ * Session persistence is handled internally (saves messages on message_end).
+ * Multiple listeners can be added. Returns unsubscribe function for this listener.
+ */
+export declare function subscribe(this: AgentSession, listener: AgentSessionEventListener): () => void;
+/**
+ * Temporarily disconnect from agent events.
+ * User listeners are preserved and will receive events again after resubscribe().
+ * Used internally during operations that need to pause event processing.
+ */
+export declare function _disconnectFromAgent(this: AgentSession): void;
+/**
+ * Reconnect to agent events after _disconnectFromAgent().
+ * Preserves all existing listeners.
+ */
+export declare function _reconnectToAgent(this: AgentSession): void;
+/**
+ * Remove all listeners and disconnect from agent.
+ * Call this when completely done with the session.
+ */
+export declare function dispose(this: AgentSession): void;
+/** Full agent state */
+export declare const agentSessionEventsMethods: {
+    _emit: typeof _emit;
+    _emitQueueUpdate: typeof _emitQueueUpdate;
+    _handleAgentEvent: typeof _handleAgentEvent;
+    _createRetryPromiseForAgentEnd: typeof _createRetryPromiseForAgentEnd;
+    _findLastAssistantInMessages: typeof _findLastAssistantInMessages;
+    _processAgentEvent: typeof _processAgentEvent;
+    _applyInterruptAbortMessage: typeof _applyInterruptAbortMessage;
+    _applyProviderErrorGuidance: typeof _applyProviderErrorGuidance;
+    _resolveRetry: typeof _resolveRetry;
+    _getUserMessageText: typeof _getUserMessageText;
+    _findLastAssistantMessage: typeof _findLastAssistantMessage;
+    _replaceMessageInPlace: typeof _replaceMessageInPlace;
+    _emitExtensionEvent: typeof _emitExtensionEvent;
+    subscribe: typeof subscribe;
+    _disconnectFromAgent: typeof _disconnectFromAgent;
+    _reconnectToAgent: typeof _reconnectToAgent;
+    dispose: typeof dispose;
+};
+//# sourceMappingURL=agent-session-events.d.ts.map

package/dist/core/agent-session-events.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"agent-session-events.d.ts","sourceRoot":"","sources":["../../src/core/agent-session-events.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,UAAU,EAAE,YAAY,EAAE,MAAM,+BAA+B,CAAC;AAC9E,OAAO,KAAK,EAAE,gBAAgB,EAAE,OAAO,EAAe,MAAM,uBAAuB,CAAC;AAGpF,OAAO,KAAK,EAAE,2BAA2B,IAAI,YAAY,EAAE,MAAM,4BAA4B,CAAC;AAC9F,OAAO,EAA0F,KAAK,iBAAiB,EAAE,KAAK,yBAAyB,EAAE,MAAM,0BAA0B,CAAC;AAG1L,wBAAgB,KAAK,CAAC,IAAI,EAAE,YAAY,EAAE,KAAK,EAAE,iBAAiB,GAAG,IAAI,CAIxE;AAGD,wBAAgB,gBAAgB,CAAC,IAAI,EAAE,YAAY,GAAG,IAAI,CAMzD;AAED,4EAA4E;AAE5E,wBAAgB,iBAAiB,CAAC,IAAI,EAAE,YAAY,EAAE,KAAK,EAAE,UAAU,GAAG,IAAI,CAe7E;AAGD,wBAAgB,8BAA8B,CAAC,IAAI,EAAE,YAAY,EAAE,KAAK,EAAE,UAAU,GAAG,IAAI,CAuB1F;AAGD,wBAAgB,4BAA4B,CAAC,IAAI,EAAE,YAAY,EAAE,QAAQ,EAAE,YAAY,EAAE,GAAG,gBAAgB,GAAG,SAAS,CAQvH;AAGD,wBAAsB,kBAAkB,CAAC,IAAI,EAAE,YAAY,EAAE,KAAK,EAAE,UAAU,GAAG,OAAO,CAAC,IAAI,CAAC,CA0G7F;AAGD,wBAAgB,2BAA2B,CAAC,IAAI,EAAE,YAAY,EAAE,KAAK,EAAE,UAAU,GAAG,IAAI,CAsBvF;AAGD,wBAAgB,2BAA2B,CAAC,IAAI,EAAE,YAAY,EAAE,KAAK,EAAE,UAAU,GAAG,IAAI,CAWvF;AAED,wCAAwC;AAExC,wBAAgB,aAAa,CAAC,IAAI,EAAE,YAAY,GAAG,IAAI,CAMtD;AAED,0CAA0C;AAE1C,wBAAgB,mBAAmB,CAAC,IAAI,EAAE,YAAY,EAAE,OAAO,EAAE,OAAO,GAAG,MAAM,CAMhF;AAED,8EAA8E;AAE9E,wBAAgB,yBAAyB,CAAC,IAAI,EAAE,YAAY,GAAG,gBAAgB,GAAG,SAAS,CAS1F;AAGD,wBAAgB,sBAAsB,CAAC,IAAI,EAAE,YAAY,EAAE,MAAM,EAAE,YAAY,EAAE,WAAW,EAAE,YAAY,GAAG,IAAI,CAchH;AAED,kDAAkD;AAElD,wBAAsB,mBAAmB,CAAC,IAAI,EAAE,YAAY,EAAE,KAAK,EAAE,UAAU,GAAG,OAAO,CAAC,IAAI,CAAC,CAuE9F;AAED;;;;GAIG;AAEH,wBAAgB,SAAS,CAAC,IAAI,EAAE,YAAY,EAAE,QAAQ,EAAE,yBAAyB,GAAG,MAAM,IAAI,CAU7F;AAED;;;;GAIG;AAEH,wBAAgB,oBAAoB,CAAC,IAAI,EAAE,YAAY,GAAG,IAAI,CAK7D;AAED;;;GAGG;AAEH,wBAAgB,iBAAiB,CAAC,IAAI,EAAE,YAAY,GAAG,IAAI,CAG1D;AAED;;;GAGG;AAEH,wBAAgB,OAAO,CAAC,IAAI,EAAE,YAAY,GAAG,IAAI,CAOhD;AAMD,uBAAuB;AAEvB,eAAO,MAAM,yBAAyB;;;;;;;;;;;;;;;;;;CAkBrC,CAAC","sourcesContent":["import type { AgentEvent, AgentMessage } from \"@earendil-works/pi-agent-core\";\nimport type { AssistantMessage, Message, TextContent } from \"@earendil-works/pi-ai\";\nimport { cleanupSessionResources } from \"@earendil-works/pi-ai\";\nimport { formatCopilotProviderError } from \"./copilot-errors.ts\";\nimport type { AgentSessionInternalSurface as AgentSession } from \"./agent-session-methods.ts\";\nimport { customMessageExcludesContext, isSingleGenericAbortTextContent, replacementAbortContent, type AgentSessionEvent, type AgentSessionEventListener } from \"./agent-session-types.ts\";\nimport type { MessageEndEvent, MessageStartEvent, MessageUpdateEvent, ToolExecutionEndEvent, ToolExecutionStartEvent, ToolExecutionUpdateEvent, TurnEndEvent, TurnStartEvent } from \"./extensions/index.ts\";\n\nexport function _emit(this: AgentSession, event: AgentSessionEvent): void {\n\tfor (const l of this._eventListeners) {\n\t\tl(event);\n\t}\n}\n\n\nexport function _emitQueueUpdate(this: AgentSession): void {\n\tthis._emit({\n\t\ttype: \"queue_update\",\n\t\tsteering: [...this._steeringMessages],\n\t\tfollowUp: [...this._followUpMessages],\n\t});\n}\n\n/** Internal handler for agent events - shared by subscribe and reconnect */\n\nexport function _handleAgentEvent(this: AgentSession, event: AgentEvent): void {\n\t// Create retry promise synchronously before queueing async processing.\n\t// Agent.emit() calls this handler synchronously, and prompt() calls waitForRetry()\n\t// as soon as agent.prompt() resolves. If _retryPromise is created only inside\n\t// _processAgentEvent, slow earlier queued events can delay agent_end processing\n\t// and waitForRetry() can miss the in-flight retry.\n\tthis._createRetryPromiseForAgentEnd(event);\n\n\tthis._agentEventQueue = this._agentEventQueue.then(\n\t\t() => this._processAgentEvent(event),\n\t\t() => this._processAgentEvent(event),\n\t);\n\n\t// Keep queue alive if an event handler fails\n\tthis._agentEventQueue.catch(() => {});\n}\n\n\nexport function _createRetryPromiseForAgentEnd(this: AgentSession, event: AgentEvent): void {\n\tif (event.type !== \"agent_end\" || this._retryPromise) {\n\t\treturn;\n\t}\n\n\tconst settings = this.settingsManager.getRetrySettings();\n\tif (!settings.enabled) {\n\t\treturn;\n\t}\n\n\tconst lastAssistant = this._findLastAssistantInMessages(event.messages);\n\tif (!lastAssistant) {\n\t\treturn;\n\t}\n\n\tconst shouldRetry = this._isRetryableError(lastAssistant) || this._isEmptyCompletion(lastAssistant);\n\tif (!shouldRetry) {\n\t\treturn;\n\t}\n\n\tthis._retryPromise = new Promise((resolve) => {\n\t\tthis._retryResolve = resolve;\n\t});\n}\n\n\nexport function _findLastAssistantInMessages(this: AgentSession, messages: AgentMessage[]): AssistantMessage | undefined {\n\tfor (let i = messages.length - 1; i >= 0; i--) {\n\t\tconst message = messages[i];\n\t\tif (message.role === \"assistant\") {\n\t\t\treturn message as AssistantMessage;\n\t\t}\n\t}\n\treturn undefined;\n}\n\n\nexport async function _processAgentEvent(this: AgentSession, event: AgentEvent): Promise<void> {\n\t// When a user message starts, check if it's from either queue and remove it BEFORE emitting\n\t// This ensures the UI sees the updated queue state\n\tif (event.type === \"message_start\" && event.message.role === \"user\") {\n\t\tthis._overflowRecoveryAttempted = false;\n\t\tconst messageText = this._getUserMessageText(event.message);\n\t\tif (messageText) {\n\t\t\t// Check steering queue first\n\t\t\tconst steeringIndex = this._steeringMessages.indexOf(messageText);\n\t\t\tif (steeringIndex !== -1) {\n\t\t\t\tthis._steeringMessages.splice(steeringIndex, 1);\n\t\t\t\tthis._emitQueueUpdate();\n\t\t\t} else {\n\t\t\t\t// Check follow-up queue\n\t\t\t\tconst followUpIndex = this._followUpMessages.indexOf(messageText);\n\t\t\t\tif (followUpIndex !== -1) {\n\t\t\t\t\tthis._followUpMessages.splice(followUpIndex, 1);\n\t\t\t\t\tthis._emitQueueUpdate();\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\t}\n\n\tthis._applyInterruptAbortMessage(event);\n\tthis._applyProviderErrorGuidance(event);\n\n\t// Emit to extensions first\n\tawait this._emitExtensionEvent(event);\n\n\t// Notify all listeners\n\tthis._emit(event);\n\n\t// Handle session persistence\n\tif (event.type === \"message_end\") {\n\t\t// Check if this is a custom message from extensions\n\t\tif (event.message.role === \"custom\") {\n\t\t\t// Persist as CustomMessageEntry\n\t\t\tthis.sessionManager.appendCustomMessageEntry(\n\t\t\t\tevent.message.customType,\n\t\t\t\tevent.message.content,\n\t\t\t\tevent.message.display,\n\t\t\t\tevent.message.details,\n\t\t\t\tcustomMessageExcludesContext(event.message),\n\t\t\t);\n\t\t} else if (\n\t\t\tevent.message.role === \"user\" ||\n\t\t\tevent.message.role === \"assistant\" ||\n\t\t\tevent.message.role === \"toolResult\"\n\t\t) {\n\t\t\tif (event.message.role === \"assistant\") {\n\t\t\t\tthis._normalizePersistedGeminiToolArgs(event.message);\n\t\t\t}\n\t\t\t// Regular LLM message - persist as SessionMessageEntry\n\t\t\tthis.sessionManager.appendMessage(event.message);\n\t\t}\n\t\t// Other message types (bashExecution, branchSummary) are persisted elsewhere\n\n\t\t// Track assistant message for auto-compaction (checked on agent_end)\n\t\tif (event.message.role === \"assistant\") {\n\t\t\tthis._lastAssistantMessage = event.message;\n\n\t\t\tconst assistantMsg = event.message as AssistantMessage;\n\t\t\t// Treat degenerate empty completions (no content, zero output tokens) as\n\t\t\t// failures alongside stopReason === \"error\". Otherwise an empty turn that\n\t\t\t// stops with reason \"stop\" would reset the retry counter on every attempt,\n\t\t\t// causing unbounded retries instead of honoring maxRetries.\n\t\t\tconst assistantFailed = assistantMsg.stopReason === \"error\" || this._isEmptyCompletion(assistantMsg);\n\t\t\tif (!assistantFailed) {\n\t\t\t\tthis._overflowRecoveryAttempted = false;\n\t\t\t}\n\n\t\t\t// Reset retry counter immediately on successful assistant response\n\t\t\t// This prevents accumulation across multiple LLM calls within a turn\n\t\t\tif (!assistantFailed && this._retryAttempt > 0) {\n\t\t\t\tthis._emit({\n\t\t\t\t\ttype: \"auto_retry_end\",\n\t\t\t\t\tsuccess: true,\n\t\t\t\t\tattempt: this._retryAttempt,\n\t\t\t\t});\n\t\t\t\tthis._retryAttempt = 0;\n\t\t\t}\n\t\t}\n\t}\n\n\t// Check auto-retry and auto-compaction after agent completes\n\tif (event.type === \"agent_end\" && this._lastAssistantMessage) {\n\t\tconst msg = this._lastAssistantMessage;\n\t\tthis._lastAssistantMessage = undefined;\n\n\t\t// Check for retryable errors first (overloaded, rate limit, server errors,\n\t\t// transient provider finish_reason errors, or degenerate empty completions)\n\t\tconst retryableError = this._isRetryableError(msg);\n\t\tconst emptyCompletion = !retryableError && this._isEmptyCompletion(msg);\n\t\tif (retryableError || emptyCompletion) {\n\t\t\tif (emptyCompletion && !msg.errorMessage) {\n\t\t\t\t// Surface a clear reason in the retry banner; empty completions carry no\n\t\t\t\t// provider error message of their own.\n\t\t\t\tmsg.errorMessage = \"Provider returned an empty completion\";\n\t\t\t}\n\t\t\tconst didRetry = await this._handleRetryableError(msg);\n\t\t\tif (didRetry) return; // Retry was initiated, don't proceed to compaction\n\t\t}\n\n\t\tthis._resolveRetry();\n\t\tawait this._checkCompaction(msg);\n\t}\n}\n\n\nexport function _applyInterruptAbortMessage(this: AgentSession, event: AgentEvent): void {\n\tconst abortMessage = this._activeInterruptAbortMessage;\n\tif (!abortMessage) return;\n\n\tif (event.type === \"tool_execution_end\" && event.isError && isSingleGenericAbortTextContent(event.result.content)) {\n\t\tevent.result.content = replacementAbortContent(abortMessage);\n\t\treturn;\n\t}\n\n\tif (event.type !== \"message_start\" && event.type !== \"message_end\") return;\n\n\tif (event.message.role === \"toolResult\" && event.message.isError && isSingleGenericAbortTextContent(event.message.content)) {\n\t\tevent.message.content = replacementAbortContent(abortMessage);\n\t\treturn;\n\t}\n\n\tif (event.message.role === \"assistant\") {\n\t\tconst assistantMessage = event.message as AssistantMessage;\n\t\tif (assistantMessage.stopReason === \"aborted\") {\n\t\t\tassistantMessage.errorMessage = abortMessage;\n\t\t}\n\t}\n}\n\n\nexport function _applyProviderErrorGuidance(this: AgentSession, event: AgentEvent): void {\n\tif (event.type !== \"message_start\" && event.type !== \"message_update\" && event.type !== \"message_end\") return;\n\tif (event.message.role !== \"assistant\") return;\n\n\tconst assistantMessage = event.message as AssistantMessage;\n\tif (assistantMessage.stopReason !== \"error\" || !assistantMessage.errorMessage) return;\n\n\tassistantMessage.errorMessage = formatCopilotProviderError(\n\t\tassistantMessage.provider,\n\t\tassistantMessage.errorMessage,\n\t);\n}\n\n/** Resolve the pending retry promise */\n\nexport function _resolveRetry(this: AgentSession): void {\n\tif (this._retryResolve) {\n\t\tthis._retryResolve();\n\t\tthis._retryResolve = undefined;\n\t\tthis._retryPromise = undefined;\n\t}\n}\n\n/** Extract text content from a message */\n\nexport function _getUserMessageText(this: AgentSession, message: Message): string {\n\tif (message.role !== \"user\") return \"\";\n\tconst content = message.content;\n\tif (typeof content === \"string\") return content;\n\tconst textBlocks = content.filter((c) => c.type === \"text\");\n\treturn textBlocks.map((c) => (c as TextContent).text).join(\"\");\n}\n\n/** Find the last assistant message in agent state (including aborted ones) */\n\nexport function _findLastAssistantMessage(this: AgentSession): AssistantMessage | undefined {\n\tconst messages = this.agent.state.messages;\n\tfor (let i = messages.length - 1; i >= 0; i--) {\n\t\tconst msg = messages[i];\n\t\tif (msg.role === \"assistant\") {\n\t\t\treturn msg as AssistantMessage;\n\t\t}\n\t}\n\treturn undefined;\n}\n\n\nexport function _replaceMessageInPlace(this: AgentSession, target: AgentMessage, replacement: AgentMessage): void {\n\t// Agent-core stores the finalized message object in its state before emitting message_end.\n\t// SessionManager persistence happens later in _processAgentEvent() with event.message.\n\t// Mutating this object in place keeps agent state, later turn/agent events, listeners,\n\t// and the eventual SessionManager.appendMessage(event.message) persistence in sync.\n\tif (target === replacement) {\n\t\treturn;\n\t}\n\n\tconst targetRecord = target as unknown as Record<string, unknown>;\n\tfor (const key of Object.keys(targetRecord)) {\n\t\tdelete targetRecord[key];\n\t}\n\tObject.assign(targetRecord, replacement);\n}\n\n/** Emit extension events based on agent events */\n\nexport async function _emitExtensionEvent(this: AgentSession, event: AgentEvent): Promise<void> {\n\tif (event.type === \"agent_start\") {\n\t\tthis._turnIndex = 0;\n\t\tawait this._extensionRunner.emit({ type: \"agent_start\" });\n\t} else if (event.type === \"agent_end\") {\n\t\tawait this._extensionRunner.emit({ type: \"agent_end\", messages: event.messages });\n\t} else if (event.type === \"turn_start\") {\n\t\tconst extensionEvent: TurnStartEvent = {\n\t\t\ttype: \"turn_start\",\n\t\t\tturnIndex: this._turnIndex,\n\t\t\ttimestamp: Date.now(),\n\t\t};\n\t\tawait this._extensionRunner.emit(extensionEvent);\n\t} else if (event.type === \"turn_end\") {\n\t\tconst extensionEvent: TurnEndEvent = {\n\t\t\ttype: \"turn_end\",\n\t\t\tturnIndex: this._turnIndex,\n\t\t\tmessage: event.message,\n\t\t\ttoolResults: event.toolResults,\n\t\t};\n\t\tawait this._extensionRunner.emit(extensionEvent);\n\t\tthis._turnIndex++;\n\t} else if (event.type === \"message_start\") {\n\t\tconst extensionEvent: MessageStartEvent = {\n\t\t\ttype: \"message_start\",\n\t\t\tmessage: event.message,\n\t\t};\n\t\tawait this._extensionRunner.emit(extensionEvent);\n\t} else if (event.type === \"message_update\") {\n\t\tconst extensionEvent: MessageUpdateEvent = {\n\t\t\ttype: \"message_update\",\n\t\t\tmessage: event.message,\n\t\t\tassistantMessageEvent: event.assistantMessageEvent,\n\t\t};\n\t\tawait this._extensionRunner.emit(extensionEvent);\n\t} else if (event.type === \"message_end\") {\n\t\tconst extensionEvent: MessageEndEvent = {\n\t\t\ttype: \"message_end\",\n\t\t\tmessage: event.message,\n\t\t};\n\t\tconst replacement = await this._extensionRunner.emitMessageEnd(extensionEvent);\n\t\tif (replacement) {\n\t\t\tthis._replaceMessageInPlace(event.message, replacement);\n\t\t}\n\t} else if (event.type === \"tool_execution_start\") {\n\t\tconst extensionEvent: ToolExecutionStartEvent = {\n\t\t\ttype: \"tool_execution_start\",\n\t\t\ttoolCallId: event.toolCallId,\n\t\t\ttoolName: event.toolName,\n\t\t\targs: event.args,\n\t\t};\n\t\tawait this._extensionRunner.emit(extensionEvent);\n\t} else if (event.type === \"tool_execution_update\") {\n\t\tconst extensionEvent: ToolExecutionUpdateEvent = {\n\t\t\ttype: \"tool_execution_update\",\n\t\t\ttoolCallId: event.toolCallId,\n\t\t\ttoolName: event.toolName,\n\t\t\targs: event.args,\n\t\t\tpartialResult: event.partialResult,\n\t\t};\n\t\tawait this._extensionRunner.emit(extensionEvent);\n\t} else if (event.type === \"tool_execution_end\") {\n\t\tconst extensionEvent: ToolExecutionEndEvent = {\n\t\t\ttype: \"tool_execution_end\",\n\t\t\ttoolCallId: event.toolCallId,\n\t\t\ttoolName: event.toolName,\n\t\t\tresult: event.result,\n\t\t\tisError: event.isError,\n\t\t};\n\t\tawait this._extensionRunner.emit(extensionEvent);\n\t}\n}\n\n/**\n * Subscribe to agent events.\n * Session persistence is handled internally (saves messages on message_end).\n * Multiple listeners can be added. Returns unsubscribe function for this listener.\n */\n\nexport function subscribe(this: AgentSession, listener: AgentSessionEventListener): () => void {\n\tthis._eventListeners.push(listener);\n\n\t// Return unsubscribe function for this specific listener\n\treturn () => {\n\t\tconst index = this._eventListeners.indexOf(listener);\n\t\tif (index !== -1) {\n\t\t\tthis._eventListeners.splice(index, 1);\n\t\t}\n\t};\n}\n\n/**\n * Temporarily disconnect from agent events.\n * User listeners are preserved and will receive events again after resubscribe().\n * Used internally during operations that need to pause event processing.\n */\n\nexport function _disconnectFromAgent(this: AgentSession): void {\n\tif (this._unsubscribeAgent) {\n\t\tthis._unsubscribeAgent();\n\t\tthis._unsubscribeAgent = undefined;\n\t}\n}\n\n/**\n * Reconnect to agent events after _disconnectFromAgent().\n * Preserves all existing listeners.\n */\n\nexport function _reconnectToAgent(this: AgentSession): void {\n\tif (this._unsubscribeAgent) return; // Already connected\n\tthis._unsubscribeAgent = this.agent.subscribe(this._handleAgentEvent);\n}\n\n/**\n * Remove all listeners and disconnect from agent.\n * Call this when completely done with the session.\n */\n\nexport function dispose(this: AgentSession): void {\n\tthis._extensionRunner.invalidate(\n\t\t\"This extension ctx is stale after session replacement or reload. Do not use a captured pi or command ctx after ctx.newSession(), ctx.fork(), ctx.switchSession(), or ctx.reload(). For newSession, fork, and switchSession, move post-replacement work into withSession and use the ctx passed to withSession. For reload, do not use the old ctx after await ctx.reload().\",\n\t);\n\tthis._disconnectFromAgent();\n\tthis._eventListeners = [];\n\tcleanupSessionResources(this.sessionId);\n}\n\n// =========================================================================\n// Read-only State Access\n// =========================================================================\n\n/** Full agent state */\n\nexport const agentSessionEventsMethods = {\n\t_emit,\n\t_emitQueueUpdate,\n\t_handleAgentEvent,\n\t_createRetryPromiseForAgentEnd,\n\t_findLastAssistantInMessages,\n\t_processAgentEvent,\n\t_applyInterruptAbortMessage,\n\t_applyProviderErrorGuidance,\n\t_resolveRetry,\n\t_getUserMessageText,\n\t_findLastAssistantMessage,\n\t_replaceMessageInPlace,\n\t_emitExtensionEvent,\n\tsubscribe,\n\t_disconnectFromAgent,\n\t_reconnectToAgent,\n\tdispose,\n};\n"]}