@caupulican/pi-adaptative 0.81.39 → 0.81.41

package/dist/core/compaction-controller.d.ts

@@ -0,0 +1,90 @@
+import type { Agent, AgentMessage, ThinkingLevel } from "@caupulican/pi-agent-core";
+import { type CompactionResult, type CompactionSettings, type SessionManager } from "@caupulican/pi-agent-core/node";
+import type { Api, AssistantMessage, Model } from "@caupulican/pi-ai";
+import type { ExtensionRunner } from "./extensions/index.ts";
+import type { FailureCorpusRecorder } from "./failure-corpus.ts";
+import type { SettingsManager } from "./settings-manager.ts";
+export type AutoCompactionReason = "overflow" | "threshold";
+type CompactionControllerEvent = {
+    type: "compaction_start";
+    reason: "manual" | AutoCompactionReason;
+} | {
+    type: "compaction_end";
+    reason: "manual" | AutoCompactionReason;
+    result: CompactionResult | undefined;
+    aborted: boolean;
+    willRetry: boolean;
+    errorMessage?: string;
+    skipReason?: string;
+} | {
+    type: "warning";
+    message: string;
+};
+export interface CompactionControllerDeps {
+    agent: Agent;
+    sessionManager: SessionManager;
+    settingsManager: SettingsManager;
+    getModel(): Model<Api> | undefined;
+    getAdaptedSettings(): CompactionSettings;
+    getRequestAuth(model: Model<Api>): Promise<{
+        apiKey?: string;
+        headers?: Record<string, string>;
+    }>;
+    resolveModelAndAuth(compactionModel: Model<Api>, sessionModel: Model<Api>): Promise<{
+        model: Model<Api>;
+        apiKey?: string;
+        headers?: Record<string, string>;
+        failure?: string;
+    }>;
+    resolveModel(sessionModel: Model<Api>): Model<Api>;
+    getSelectionReason(): string | undefined;
+    resolveThinkingLevel(compactionModel: Model<Api>, sessionModel: Model<Api>): ThinkingLevel | undefined;
+    describeSummarizer(): string;
+    getExtensionRunner(): ExtensionRunner;
+    isRawStream(): boolean;
+    disconnectAgent(): void;
+    reconnectAgent(): void;
+    abortForeground(): Promise<void>;
+    emit(event: CompactionControllerEvent): void;
+    estimateCurrentContextTokens(messages: AgentMessage[]): number;
+    buildPreDigest(): ((text: string, signal?: AbortSignal) => Promise<string>) | undefined;
+    refreshAfterCompaction(): void;
+    getFailureCorpus(): FailureCorpusRecorder;
+    measureLiveContextTokens(): number;
+    runAutoCompaction(reason: AutoCompactionReason, willRetry: boolean): Promise<boolean>;
+    compactWithRetry(run: () => Promise<CompactionResult>, signal: AbortSignal, provider?: string): Promise<CompactionResult>;
+}
+export declare function runCompactionWithRetry<T>(options: {
+    run(): Promise<T>;
+    signal: AbortSignal;
+    provider?: string;
+    getRetrySettings(): {
+        enabled: boolean;
+        maxRetries: number;
+        baseDelayMs: number;
+    };
+    recordFailure(record: Parameters<FailureCorpusRecorder["record"]>[0]): void;
+}): Promise<T>;
+/** Owns compaction detection, execution, retry, persistence, notification, and cancellation. */
+export declare class CompactionController {
+    private manualAbortController;
+    private autoAbortController;
+    private overflowRecoveryAttempted;
+    private readonly deps;
+    constructor(deps: CompactionControllerDeps);
+    isRunning(): boolean;
+    resetOverflowRecovery(): void;
+    abort(): void;
+    checkContextWindowUsageWarning(): void;
+    maybeCompactBeforeContextTransform(messages: AgentMessage[], projectContextGc: (messages: AgentMessage[]) => AgentMessage[]): Promise<boolean>;
+    compact(customInstructions?: string): Promise<CompactionResult>;
+    check(assistantMessage: AssistantMessage, skipAbortedCheck?: boolean): Promise<boolean>;
+    measureLiveContextTokens(): number;
+    runAuto(reason: AutoCompactionReason, willRetry: boolean): Promise<boolean>;
+    compactWithRetry(run: () => Promise<CompactionResult>, signal: AbortSignal, provider?: string): Promise<CompactionResult>;
+    private findLastAssistantMessage;
+    private getExtensionCompaction;
+    private applyResult;
+}
+export {};
+//# sourceMappingURL=compaction-controller.d.ts.map

package/dist/core/compaction-controller.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"compaction-controller.d.ts","sourceRoot":"","sources":["../../src/core/compaction-controller.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,KAAK,EAAE,YAAY,EAAe,aAAa,EAAE,MAAM,2BAA2B,CAAC;AAEjG,OAAO,EAGN,KAAK,gBAAgB,EACrB,KAAK,kBAAkB,EAQvB,KAAK,cAAc,EAEnB,MAAM,gCAAgC,CAAC;AACxC,OAAO,KAAK,EAAE,GAAG,EAAE,gBAAgB,EAAE,KAAK,EAAE,MAAM,mBAAmB,CAAC;AAGtE,OAAO,KAAK,EAAE,eAAe,EAA8B,MAAM,uBAAuB,CAAC;AACzF,OAAO,KAAK,EAAE,qBAAqB,EAAE,MAAM,qBAAqB,CAAC;AACjE,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,uBAAuB,CAAC;AAE7D,MAAM,MAAM,oBAAoB,GAAG,UAAU,GAAG,WAAW,CAAC;AAE5D,KAAK,yBAAyB,GAC3B;IAAE,IAAI,EAAE,kBAAkB,CAAC;IAAC,MAAM,EAAE,QAAQ,GAAG,oBAAoB,CAAA;CAAE,GACrE;IACA,IAAI,EAAE,gBAAgB,CAAC;IACvB,MAAM,EAAE,QAAQ,GAAG,oBAAoB,CAAC;IACxC,MAAM,EAAE,gBAAgB,GAAG,SAAS,CAAC;IACrC,OAAO,EAAE,OAAO,CAAC;IACjB,SAAS,EAAE,OAAO,CAAC;IACnB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,UAAU,CAAC,EAAE,MAAM,CAAC;CACnB,GACD;IAAE,IAAI,EAAE,SAAS,CAAC;IAAC,OAAO,EAAE,MAAM,CAAA;CAAE,CAAC;AAExC,MAAM,WAAW,wBAAwB;IACxC,KAAK,EAAE,KAAK,CAAC;IACb,cAAc,EAAE,cAAc,CAAC;IAC/B,eAAe,EAAE,eAAe,CAAC;IACjC,QAAQ,IAAI,KAAK,CAAC,GAAG,CAAC,GAAG,SAAS,CAAC;IACnC,kBAAkB,IAAI,kBAAkB,CAAC;IACzC,cAAc,CAAC,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,GAAG,OAAO,CAAC;QAAE,MAAM,CAAC,EAAE,MAAM,CAAC;QAAC,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAA;KAAE,CAAC,CAAC;IAClG,mBAAmB,CAClB,eAAe,EAAE,KAAK,CAAC,GAAG,CAAC,EAC3B,YAAY,EAAE,KAAK,CAAC,GAAG,CAAC,GACtB,OAAO,CAAC;QAAE,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC;QAAC,MAAM,CAAC,EAAE,MAAM,CAAC;QAAC,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QAAC,OAAO,CAAC,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IACvG,YAAY,CAAC,YAAY,EAAE,KAAK,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC;IACnD,kBAAkB,IAAI,MAAM,GAAG,SAAS,CAAC;IACzC,oBAAoB,CAAC,eAAe,EAAE,KAAK,CAAC,GAAG,CAAC,EAAE,YAAY,EAAE,KAAK,CAAC,GAAG,CAAC,GAAG,aAAa,GAAG,SAAS,CAAC;IACvG,kBAAkB,IAAI,MAAM,CAAC;IAC7B,kBAAkB,IAAI,eAAe,CAAC;IACtC,WAAW,IAAI,OAAO,CAAC;IACvB,eAAe,IAAI,IAAI,CAAC;IACxB,cAAc,IAAI,IAAI,CAAC;IACvB,eAAe,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;IACjC,IAAI,CAAC,KAAK,EAAE,yBAAyB,GAAG,IAAI,CAAC;IAC7C,4BAA4B,CAAC,QAAQ,EAAE,YAAY,EAAE,GAAG,MAAM,CAAC;IAC/D,cAAc,IAAI,CAAC,CAAC,IAAI,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,WAAW,KAAK,OAAO,CAAC,MAAM,CAAC,CAAC,GAAG,SAAS,CAAC;IACxF,sBAAsB,IAAI,IAAI,CAAC;IAC/B,gBAAgB,IAAI,qBAAqB,CAAC;IAC1C,wBAAwB,IAAI,MAAM,CAAC;IACnC,iBAAiB,CAAC,MAAM,EAAE,oBAAoB,EAAE,SAAS,EAAE,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC;IACtF,gBAAgB,CACf,GAAG,EAAE,MAAM,OAAO,CAAC,gBAAgB,CAAC,EACpC,MAAM,EAAE,WAAW,EACnB,QAAQ,CAAC,EAAE,MAAM,GACf,OAAO,CAAC,gBAAgB,CAAC,CAAC;CAC7B;AAED,wBAAsB,sBAAsB,CAAC,CAAC,EAAE,OAAO,EAAE;IACxD,GAAG,IAAI,OAAO,CAAC,CAAC,CAAC,CAAC;IAClB,MAAM,EAAE,WAAW,CAAC;IACpB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,gBAAgB,IAAI;QAAE,OAAO,EAAE,OAAO,CAAC;QAAC,UAAU,EAAE,MAAM,CAAC;QAAC,WAAW,EAAE,MAAM,CAAA;KAAE,CAAC;IAClF,aAAa,CAAC,MAAM,EAAE,UAAU,CAAC,qBAAqB,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC;CAC5E,GAAG,OAAO,CAAC,CAAC,CAAC,CAwBb;AAED,gGAAgG;AAChG,qBAAa,oBAAoB;IAChC,OAAO,CAAC,qBAAqB,CAA8B;IAC3D,OAAO,CAAC,mBAAmB,CAA8B;IACzD,OAAO,CAAC,yBAAyB,CAAS;IAC1C,OAAO,CAAC,QAAQ,CAAC,IAAI,CAA2B;IAEhD,YAAY,IAAI,EAAE,wBAAwB,EAEzC;IAED,SAAS,IAAI,OAAO,CAEnB;IAED,qBAAqB,IAAI,IAAI,CAE5B;IAED,KAAK,IAAI,IAAI,CAGZ;IAED,8BAA8B,IAAI,IAAI,CA0BrC;IAEK,kCAAkC,CACvC,QAAQ,EAAE,YAAY,EAAE,EACxB,gBAAgB,EAAE,CAAC,QAAQ,EAAE,YAAY,EAAE,KAAK,YAAY,EAAE,GAC5D,OAAO,CAAC,OAAO,CAAC,CAiBlB;IAEK,OAAO,CAAC,kBAAkB,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,gBAAgB,CAAC,CAkIpE;IAEK,KAAK,CAAC,gBAAgB,EAAE,gBAAgB,EAAE,gBAAgB,UAAO,GAAG,OAAO,CAAC,OAAO,CAAC,CAiEzF;IAED,wBAAwB,IAAI,MAAM,CAWjC;IAEK,OAAO,CAAC,MAAM,EAAE,oBAAoB,EAAE,SAAS,EAAE,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,CAgJhF;IAEK,gBAAgB,CACrB,GAAG,EAAE,MAAM,OAAO,CAAC,gBAAgB,CAAC,EACpC,MAAM,EAAE,WAAW,EACnB,QAAQ,CAAC,EAAE,MAAM,GACf,OAAO,CAAC,gBAAgB,CAAC,CAQ3B;IAED,OAAO,CAAC,wBAAwB;YAQlB,sBAAsB;YAkBtB,WAAW;CAuBzB","sourcesContent":["import type { Agent, AgentMessage, RetryPolicy, ThinkingLevel } from \"@caupulican/pi-agent-core\";\nimport { classifyFailure, computeRetryDelayMs, DEFAULT_RETRY_POLICY, sleepAbortable } from \"@caupulican/pi-agent-core\";\nimport {\n\ttype CompactionEntry,\n\ttype CompactionPreparation,\n\ttype CompactionResult,\n\ttype CompactionSettings,\n\tcalculateContextTokens,\n\tcompact,\n\tcreateDeterministicCompaction,\n\testimateContextTokens,\n\tgetLatestCompactionEntry,\n\tprepareCompaction,\n\trunCompactionLoop,\n\ttype SessionManager,\n\tshouldCompact,\n} from \"@caupulican/pi-agent-core/node\";\nimport type { Api, AssistantMessage, Model } from \"@caupulican/pi-ai\";\nimport { isContextOverflow } from \"@caupulican/pi-ai\";\nimport { formatNoModelSelectedMessage } from \"./auth-guidance.ts\";\nimport type { ExtensionRunner, SessionBeforeCompactResult } from \"./extensions/index.ts\";\nimport type { FailureCorpusRecorder } from \"./failure-corpus.ts\";\nimport type { SettingsManager } from \"./settings-manager.ts\";\n\nexport type AutoCompactionReason = \"overflow\" | \"threshold\";\n\ntype CompactionControllerEvent =\n\t| { type: \"compaction_start\"; reason: \"manual\" | AutoCompactionReason }\n\t| {\n\t\t\ttype: \"compaction_end\";\n\t\t\treason: \"manual\" | AutoCompactionReason;\n\t\t\tresult: CompactionResult | undefined;\n\t\t\taborted: boolean;\n\t\t\twillRetry: boolean;\n\t\t\terrorMessage?: string;\n\t\t\tskipReason?: string;\n\t  }\n\t| { type: \"warning\"; message: string };\n\nexport interface CompactionControllerDeps {\n\tagent: Agent;\n\tsessionManager: SessionManager;\n\tsettingsManager: SettingsManager;\n\tgetModel(): Model<Api> | undefined;\n\tgetAdaptedSettings(): CompactionSettings;\n\tgetRequestAuth(model: Model<Api>): Promise<{ apiKey?: string; headers?: Record<string, string> }>;\n\tresolveModelAndAuth(\n\t\tcompactionModel: Model<Api>,\n\t\tsessionModel: Model<Api>,\n\t): Promise<{ model: Model<Api>; apiKey?: string; headers?: Record<string, string>; failure?: string }>;\n\tresolveModel(sessionModel: Model<Api>): Model<Api>;\n\tgetSelectionReason(): string | undefined;\n\tresolveThinkingLevel(compactionModel: Model<Api>, sessionModel: Model<Api>): ThinkingLevel | undefined;\n\tdescribeSummarizer(): string;\n\tgetExtensionRunner(): ExtensionRunner;\n\tisRawStream(): boolean;\n\tdisconnectAgent(): void;\n\treconnectAgent(): void;\n\tabortForeground(): Promise<void>;\n\temit(event: CompactionControllerEvent): void;\n\testimateCurrentContextTokens(messages: AgentMessage[]): number;\n\tbuildPreDigest(): ((text: string, signal?: AbortSignal) => Promise<string>) | undefined;\n\trefreshAfterCompaction(): void;\n\tgetFailureCorpus(): FailureCorpusRecorder;\n\tmeasureLiveContextTokens(): number;\n\trunAutoCompaction(reason: AutoCompactionReason, willRetry: boolean): Promise<boolean>;\n\tcompactWithRetry(\n\t\trun: () => Promise<CompactionResult>,\n\t\tsignal: AbortSignal,\n\t\tprovider?: string,\n\t): Promise<CompactionResult>;\n}\n\nexport async function runCompactionWithRetry<T>(options: {\n\trun(): Promise<T>;\n\tsignal: AbortSignal;\n\tprovider?: string;\n\tgetRetrySettings(): { enabled: boolean; maxRetries: number; baseDelayMs: number };\n\trecordFailure(record: Parameters<FailureCorpusRecorder[\"record\"]>[0]): void;\n}): Promise<T> {\n\tconst retrySettings = options.getRetrySettings();\n\tconst maxAttempts = retrySettings.enabled ? Math.max(1, retrySettings.maxRetries + 1) : 1;\n\tconst policy: RetryPolicy = {\n\t\tmaxAttempts,\n\t\tbaseDelayMs: retrySettings.baseDelayMs,\n\t\tmaxDelayMs: DEFAULT_RETRY_POLICY.maxDelayMs,\n\t\tjitterRatio: 0,\n\t};\n\tfor (let attempt = 1; ; attempt++) {\n\t\ttry {\n\t\t\treturn await options.run();\n\t\t} catch (error) {\n\t\t\tif (options.signal.aborted || attempt >= maxAttempts) throw error;\n\t\t\tconst message = error instanceof Error ? error.message : String(error);\n\t\t\tconst classified = classifyFailure({ message, provider: options.provider });\n\t\t\toptions.recordFailure({ provider: options.provider, message, classified });\n\t\t\tif (!classified.retryable) throw error;\n\t\t\tawait sleepAbortable(\n\t\t\t\tcomputeRetryDelayMs(policy, attempt, { retryAfterMs: classified.retryAfterMs }),\n\t\t\t\toptions.signal,\n\t\t\t);\n\t\t}\n\t}\n}\n\n/** Owns compaction detection, execution, retry, persistence, notification, and cancellation. */\nexport class CompactionController {\n\tprivate manualAbortController: AbortController | undefined;\n\tprivate autoAbortController: AbortController | undefined;\n\tprivate overflowRecoveryAttempted = false;\n\tprivate readonly deps: CompactionControllerDeps;\n\n\tconstructor(deps: CompactionControllerDeps) {\n\t\tthis.deps = deps;\n\t}\n\n\tisRunning(): boolean {\n\t\treturn this.manualAbortController !== undefined || this.autoAbortController !== undefined;\n\t}\n\n\tresetOverflowRecovery(): void {\n\t\tthis.overflowRecoveryAttempted = false;\n\t}\n\n\tabort(): void {\n\t\tthis.manualAbortController?.abort();\n\t\tthis.autoAbortController?.abort();\n\t}\n\n\tcheckContextWindowUsageWarning(): void {\n\t\tconst model = this.deps.getModel();\n\t\tif (!model) return;\n\t\tconst contextWindow = model.contextWindow ?? 0;\n\t\tif (contextWindow <= 0) return;\n\n\t\tconst systemPromptTokens = Math.ceil((this.deps.agent.state.systemPrompt ?? \"\").length / 4);\n\t\tlet toolsChars = 0;\n\t\tfor (const tool of this.deps.agent.state.tools || []) {\n\t\t\ttoolsChars += tool.name.length;\n\t\t\ttoolsChars += tool.description?.length ?? 0;\n\t\t\tif (tool.parameters) toolsChars += JSON.stringify(tool.parameters).length;\n\t\t}\n\t\tconst baseTokens = systemPromptTokens + Math.ceil(toolsChars / 4);\n\n\t\tif (baseTokens >= contextWindow) {\n\t\t\tthis.deps.emit({\n\t\t\t\ttype: \"warning\",\n\t\t\t\tmessage: `Base configuration (system prompt and active tools) consumes ${baseTokens} tokens, which exceeds the model's context window of ${contextWindow} tokens. The model cannot process any prompts in this state.`,\n\t\t\t});\n\t\t} else if (baseTokens >= contextWindow * 0.7) {\n\t\t\tthis.deps.emit({\n\t\t\t\ttype: \"warning\",\n\t\t\t\tmessage: `Base configuration (system prompt and active tools) consumes ${baseTokens} tokens (${Math.round((baseTokens / contextWindow) * 100)}% of the ${contextWindow} context window). This leaves very little room for conversation history and may cause immediate compaction or context overflow.`,\n\t\t\t});\n\t\t}\n\t}\n\n\tasync maybeCompactBeforeContextTransform(\n\t\tmessages: AgentMessage[],\n\t\tprojectContextGc: (messages: AgentMessage[]) => AgentMessage[],\n\t): Promise<boolean> {\n\t\tconst settings = this.deps.getAdaptedSettings();\n\t\tconst model = this.deps.getModel();\n\t\tconst contextWindow = model?.contextWindow ?? 0;\n\t\tif (!settings.enabled || !model || contextWindow <= 0 || this.isRunning()) return false;\n\n\t\tconst triggerTokens = model.autoCompactionTriggerTokens;\n\t\tconst contextTokens = this.deps.estimateCurrentContextTokens(messages);\n\t\tif (!shouldCompact(contextTokens, contextWindow, settings, triggerTokens)) return false;\n\t\tconst projectedMessages = projectContextGc(messages);\n\t\tconst projectedTokens = this.deps.estimateCurrentContextTokens(projectedMessages);\n\t\tif (!shouldCompact(projectedTokens, contextWindow, settings, triggerTokens)) return false;\n\n\t\tconst latestBefore = getLatestCompactionEntry(this.deps.sessionManager.getBranch())?.id;\n\t\tawait this.deps.runAutoCompaction(\"threshold\", false);\n\t\tconst latestAfter = getLatestCompactionEntry(this.deps.sessionManager.getBranch())?.id;\n\t\treturn Boolean(latestAfter && latestAfter !== latestBefore);\n\t}\n\n\tasync compact(customInstructions?: string): Promise<CompactionResult> {\n\t\tthis.deps.disconnectAgent();\n\t\tawait this.deps.abortForeground();\n\t\tthis.manualAbortController = new AbortController();\n\t\tthis.deps.emit({ type: \"compaction_start\", reason: \"manual\" });\n\n\t\ttry {\n\t\t\tconst sessionModel = this.deps.getModel();\n\t\t\tif (!sessionModel) throw new Error(formatNoModelSelectedMessage());\n\n\t\t\tconst selectedCompactionModel = this.deps.resolveModel(sessionModel);\n\t\t\tif (this.deps.isRawStream()) await this.deps.getRequestAuth(selectedCompactionModel);\n\t\t\tconst selectionReason = this.deps.getSelectionReason() ?? \"unknown\";\n\t\t\tconst settings = this.deps.getAdaptedSettings();\n\t\t\tconst initialBranch = this.deps.sessionManager.getBranch();\n\t\t\tconst initialPreparation = prepareCompaction(initialBranch, settings);\n\t\t\tif (!initialPreparation) {\n\t\t\t\tconst lastEntry = initialBranch[initialBranch.length - 1];\n\t\t\t\tif (lastEntry?.type === \"compaction\") throw new Error(\"Already compacted\");\n\t\t\t\tthrow new Error(\"Nothing to compact (session too small)\");\n\t\t\t}\n\n\t\t\tconst signal = this.manualAbortController.signal;\n\t\t\tconst extension = await this.getExtensionCompaction(\n\t\t\t\tinitialPreparation,\n\t\t\t\tinitialBranch,\n\t\t\t\tcustomInstructions,\n\t\t\t\tsignal,\n\t\t\t);\n\t\t\tif (extension.cancelled) throw new Error(\"Compaction cancelled\");\n\t\t\tif (extension.result) {\n\t\t\t\tawait this.applyResult(extension.result, true);\n\t\t\t\tthis.deps.emit({\n\t\t\t\t\ttype: \"compaction_end\",\n\t\t\t\t\treason: \"manual\",\n\t\t\t\t\tresult: extension.result,\n\t\t\t\t\taborted: false,\n\t\t\t\t\twillRetry: false,\n\t\t\t\t});\n\t\t\t\treturn extension.result;\n\t\t\t}\n\n\t\t\tlet appliedResult: CompactionResult | undefined;\n\t\t\tconst outcome = await runCompactionLoop({\n\t\t\t\tmeasureLiveTokens: () =>\n\t\t\t\t\tMath.max(this.deps.estimateCurrentContextTokens(this.deps.agent.state.messages), 1),\n\t\t\t\tshouldCompact: () => true,\n\t\t\t\tgetPostApplyMargin: () => 0,\n\t\t\t\tgetBranch: () => this.deps.sessionManager.getBranch(),\n\t\t\t\tgetBaseKeepRecentTokens: () => settings.keepRecentTokens,\n\t\t\t\tresolveModelAndAuth: async (modelTier) => {\n\t\t\t\t\tconst model = modelTier === \"cheap\" ? selectedCompactionModel : sessionModel;\n\t\t\t\t\treturn this.deps.resolveModelAndAuth(model, sessionModel);\n\t\t\t\t},\n\t\t\t\tsummarizeAndVerify: async (params, model, apiKey, headers, branch) => {\n\t\t\t\t\tconst preparation = prepareCompaction(\n\t\t\t\t\t\tbranch,\n\t\t\t\t\t\t{ ...settings, keepRecentTokens: params.keepRecentTokens },\n\t\t\t\t\t\t{ allowTrailingCompactionAsPrevious: true },\n\t\t\t\t\t);\n\t\t\t\t\tif (!preparation) throw new Error(\"Nothing to compact (session too small)\");\n\t\t\t\t\tconst compactionThinkingLevel = this.deps.resolveThinkingLevel(model, sessionModel);\n\t\t\t\t\tconst result = await this.deps.compactWithRetry(\n\t\t\t\t\t\t() =>\n\t\t\t\t\t\t\tcompact(\n\t\t\t\t\t\t\t\tpreparation,\n\t\t\t\t\t\t\t\tmodel,\n\t\t\t\t\t\t\t\tapiKey,\n\t\t\t\t\t\t\t\theaders,\n\t\t\t\t\t\t\t\tcustomInstructions,\n\t\t\t\t\t\t\t\tsignal,\n\t\t\t\t\t\t\t\tcompactionThinkingLevel,\n\t\t\t\t\t\t\t\tthis.deps.agent.streamFn,\n\t\t\t\t\t\t\t\tthis.deps.buildPreDigest(),\n\t\t\t\t\t\t\t\t{ chunked: params.chunked },\n\t\t\t\t\t\t\t),\n\t\t\t\t\t\tsignal,\n\t\t\t\t\t\tmodel.provider,\n\t\t\t\t\t);\n\t\t\t\t\treturn { result };\n\t\t\t\t},\n\t\t\t\tbuildDeterministicCheckpoint: () => ({ result: createDeterministicCompaction(initialPreparation) }),\n\t\t\t\tapply: async (result) => {\n\t\t\t\t\tif (signal.aborted) throw new Error(\"Compaction cancelled\");\n\t\t\t\t\tawait this.applyResult(result, false);\n\t\t\t\t\tappliedResult = result;\n\t\t\t\t},\n\t\t\t\tverifyPostApplyEffect: () => false,\n\t\t\t\tonTransition: ({ cycle, cause, detail }) => {\n\t\t\t\t\tthis.deps.emit({\n\t\t\t\t\t\ttype: \"warning\",\n\t\t\t\t\t\tmessage: `manual compaction cycle ${cycle}: ${cause}${detail ? ` (${detail})` : \"\"} — retrying from step 0 (${this.deps.describeSummarizer()})`,\n\t\t\t\t\t});\n\t\t\t\t},\n\t\t\t\tsignal,\n\t\t\t});\n\n\t\t\tif (outcome.kind === \"failed\") {\n\t\t\t\tif (outcome.reason === \"aborted\") throw new Error(\"Compaction cancelled\");\n\t\t\t\tthrow new Error(\n\t\t\t\t\t`manual compaction failed after retry ladder using ${selectedCompactionModel.provider}/${selectedCompactionModel.id} (${selectionReason}); first failure: ${outcome.reason}`,\n\t\t\t\t);\n\t\t\t}\n\t\t\tif (outcome.kind === \"skip\" || !appliedResult) {\n\t\t\t\tthrow new Error(outcome.kind === \"skip\" ? outcome.reason : \"Compaction failed\");\n\t\t\t}\n\t\t\tthis.deps.emit({\n\t\t\t\ttype: \"compaction_end\",\n\t\t\t\treason: \"manual\",\n\t\t\t\tresult: appliedResult,\n\t\t\t\taborted: false,\n\t\t\t\twillRetry: false,\n\t\t\t});\n\t\t\treturn appliedResult;\n\t\t} catch (error) {\n\t\t\tconst message = error instanceof Error ? error.message : String(error);\n\t\t\tconst aborted = message === \"Compaction cancelled\" || (error instanceof Error && error.name === \"AbortError\");\n\t\t\tthis.deps.emit({\n\t\t\t\ttype: \"compaction_end\",\n\t\t\t\treason: \"manual\",\n\t\t\t\tresult: undefined,\n\t\t\t\taborted,\n\t\t\t\twillRetry: false,\n\t\t\t\terrorMessage: aborted ? undefined : `Compaction failed: ${message}`,\n\t\t\t});\n\t\t\tthrow error;\n\t\t} finally {\n\t\t\tthis.manualAbortController = undefined;\n\t\t\tthis.deps.reconnectAgent();\n\t\t}\n\t}\n\n\tasync check(assistantMessage: AssistantMessage, skipAbortedCheck = true): Promise<boolean> {\n\t\tconst settings = this.deps.getAdaptedSettings();\n\t\tif (!settings.enabled) return false;\n\t\tif (skipAbortedCheck && assistantMessage.stopReason === \"aborted\") return false;\n\n\t\tconst model = this.deps.getModel();\n\t\tconst contextWindow = model?.contextWindow ?? 0;\n\t\tconst sameModel = model && assistantMessage.provider === model.provider && assistantMessage.model === model.id;\n\t\tconst compactionEntry = getLatestCompactionEntry(this.deps.sessionManager.getBranch());\n\t\tconst assistantIsFromBeforeCompaction =\n\t\t\tcompactionEntry !== null && assistantMessage.timestamp <= new Date(compactionEntry.timestamp).getTime();\n\t\tif (assistantIsFromBeforeCompaction) return false;\n\n\t\tif (sameModel && isContextOverflow(assistantMessage, contextWindow)) {\n\t\t\tif (this.overflowRecoveryAttempted) {\n\t\t\t\tthis.deps.emit({\n\t\t\t\t\ttype: \"compaction_end\",\n\t\t\t\t\treason: \"overflow\",\n\t\t\t\t\tresult: undefined,\n\t\t\t\t\taborted: false,\n\t\t\t\t\twillRetry: false,\n\t\t\t\t\terrorMessage:\n\t\t\t\t\t\t\"Context overflow recovery failed after one compact-and-retry attempt. Try reducing context or switching to a larger-context model.\",\n\t\t\t\t});\n\t\t\t\treturn false;\n\t\t\t}\n\t\t\tthis.overflowRecoveryAttempted = true;\n\t\t\tconst messages = this.deps.agent.state.messages;\n\t\t\tif (messages.length > 0 && messages[messages.length - 1].role === \"assistant\") {\n\t\t\t\tthis.deps.agent.state.messages = messages.slice(0, -1);\n\t\t\t}\n\t\t\treturn this.deps.runAutoCompaction(\"overflow\", true);\n\t\t}\n\n\t\tlet contextTokens: number;\n\t\tif (assistantMessage.stopReason === \"error\") {\n\t\t\tconst messages = this.deps.agent.state.messages;\n\t\t\tconst estimate = estimateContextTokens(messages);\n\t\t\tif (estimate.lastUsageIndex === null) return false;\n\t\t\tconst usageMessage = messages[estimate.lastUsageIndex];\n\t\t\tif (\n\t\t\t\tcompactionEntry &&\n\t\t\t\tusageMessage.role === \"assistant\" &&\n\t\t\t\tusageMessage.timestamp <= new Date(compactionEntry.timestamp).getTime()\n\t\t\t) {\n\t\t\t\treturn false;\n\t\t\t}\n\t\t\tcontextTokens = estimate.tokens;\n\t\t} else {\n\t\t\tcontextTokens = calculateContextTokens(assistantMessage.usage);\n\t\t\tconst estimate = estimateContextTokens(this.deps.agent.state.messages);\n\t\t\tif (estimate.lastUsageIndex !== null) {\n\t\t\t\tconst usageMessage = this.deps.agent.state.messages[estimate.lastUsageIndex];\n\t\t\t\tconst usageIsPostCompaction = !(\n\t\t\t\t\tcompactionEntry &&\n\t\t\t\t\tusageMessage.role === \"assistant\" &&\n\t\t\t\t\tusageMessage.timestamp <= new Date(compactionEntry.timestamp).getTime()\n\t\t\t\t);\n\t\t\t\tif (usageIsPostCompaction) contextTokens = Math.max(contextTokens, estimate.tokens);\n\t\t\t}\n\t\t}\n\t\tif (shouldCompact(contextTokens, contextWindow, settings, model?.autoCompactionTriggerTokens)) {\n\t\t\treturn this.deps.runAutoCompaction(\"threshold\", false);\n\t\t}\n\t\treturn false;\n\t}\n\n\tmeasureLiveContextTokens(): number {\n\t\tconst estimatedTokens = this.deps.estimateCurrentContextTokens(this.deps.agent.state.messages);\n\t\tconst assistantMessage = this.findLastAssistantMessage();\n\t\tif (!assistantMessage || assistantMessage.stopReason === \"error\" || assistantMessage.stopReason === \"aborted\") {\n\t\t\treturn estimatedTokens;\n\t\t}\n\t\tconst compactionEntry = getLatestCompactionEntry(this.deps.sessionManager.getBranch());\n\t\tif (compactionEntry && assistantMessage.timestamp <= new Date(compactionEntry.timestamp).getTime()) {\n\t\t\treturn estimatedTokens;\n\t\t}\n\t\treturn Math.max(calculateContextTokens(assistantMessage.usage), estimatedTokens);\n\t}\n\n\tasync runAuto(reason: AutoCompactionReason, willRetry: boolean): Promise<boolean> {\n\t\tconst settings = this.deps.getAdaptedSettings();\n\t\tconst model = this.deps.getModel();\n\t\tthis.deps.emit({ type: \"compaction_start\", reason });\n\t\tconst hadQueuedMessages = this.deps.agent.hasQueuedMessages();\n\t\tthis.autoAbortController = new AbortController();\n\t\tconst signal = this.autoAbortController.signal;\n\t\tlet fromExtension = false;\n\t\tlet lastCompaction: CompactionResult | undefined;\n\t\tlet extensionCancelled = false;\n\t\ttry {\n\t\t\tif (!model) {\n\t\t\t\tthis.deps.emit({\n\t\t\t\t\ttype: \"compaction_end\",\n\t\t\t\t\treason,\n\t\t\t\t\tresult: undefined,\n\t\t\t\t\taborted: false,\n\t\t\t\t\twillRetry: false,\n\t\t\t\t\tskipReason: \"no model selected\",\n\t\t\t\t});\n\t\t\t\treturn hadQueuedMessages || this.deps.agent.hasQueuedMessages();\n\t\t\t}\n\n\t\t\tconst contextWindow = model.contextWindow;\n\t\t\tconst margin = Math.max(0, Math.floor(0.01 * contextWindow));\n\t\t\tconst outcome = await runCompactionLoop({\n\t\t\t\tgetBranch: () => this.deps.sessionManager.getBranch(),\n\t\t\t\tmeasureLiveTokens: () => this.deps.measureLiveContextTokens(),\n\t\t\t\tshouldCompact:\n\t\t\t\t\treason === \"overflow\"\n\t\t\t\t\t\t? () => true\n\t\t\t\t\t\t: (tokens) => shouldCompact(tokens, contextWindow, settings, model.autoCompactionTriggerTokens),\n\t\t\t\tgetPostApplyMargin: () => margin,\n\t\t\t\tgetBaseKeepRecentTokens: () => settings.keepRecentTokens,\n\t\t\t\tresolveModelAndAuth: async (modelTier) =>\n\t\t\t\t\tthis.deps.resolveModelAndAuth(modelTier === \"session\" ? model : this.deps.resolveModel(model), model),\n\t\t\t\tsummarizeAndVerify: async (params, compactModel, apiKey, headers, branchEntries) => {\n\t\t\t\t\tfromExtension = false;\n\t\t\t\t\tconst preparation = prepareCompaction(branchEntries, {\n\t\t\t\t\t\t...settings,\n\t\t\t\t\t\tkeepRecentTokens: params.keepRecentTokens,\n\t\t\t\t\t});\n\t\t\t\t\tif (!preparation) throw new Error(\"already compacted\");\n\t\t\t\t\tconst compactionThinkingLevel = this.deps.resolveThinkingLevel(compactModel, model);\n\t\t\t\t\tconst extension = await this.getExtensionCompaction(preparation, branchEntries, undefined, signal);\n\t\t\t\t\tif (extension.cancelled) {\n\t\t\t\t\t\textensionCancelled = true;\n\t\t\t\t\t\tthrow new Error(\"auto-compaction-cancelled\");\n\t\t\t\t\t}\n\t\t\t\t\tif (extension.result) {\n\t\t\t\t\t\tfromExtension = true;\n\t\t\t\t\t\treturn { result: extension.result };\n\t\t\t\t\t}\n\t\t\t\t\tconst result = await this.deps.compactWithRetry(\n\t\t\t\t\t\t() =>\n\t\t\t\t\t\t\tcompact(\n\t\t\t\t\t\t\t\tpreparation,\n\t\t\t\t\t\t\t\tcompactModel,\n\t\t\t\t\t\t\t\tapiKey,\n\t\t\t\t\t\t\t\theaders,\n\t\t\t\t\t\t\t\tundefined,\n\t\t\t\t\t\t\t\tsignal,\n\t\t\t\t\t\t\t\tcompactionThinkingLevel,\n\t\t\t\t\t\t\t\tthis.deps.agent.streamFn,\n\t\t\t\t\t\t\t\tthis.deps.buildPreDigest(),\n\t\t\t\t\t\t\t\t{ chunked: params.chunked },\n\t\t\t\t\t\t\t),\n\t\t\t\t\t\tsignal,\n\t\t\t\t\t\tcompactModel.provider,\n\t\t\t\t\t);\n\t\t\t\t\treturn { result };\n\t\t\t\t},\n\t\t\t\tbuildDeterministicCheckpoint: () => {\n\t\t\t\t\tconst preparation = prepareCompaction(this.deps.sessionManager.getBranch(), settings);\n\t\t\t\t\tif (!preparation) throw new Error(\"already compacted\");\n\t\t\t\t\tfromExtension = false;\n\t\t\t\t\treturn { result: createDeterministicCompaction(preparation) };\n\t\t\t\t},\n\t\t\t\tapply: async (result) => {\n\t\t\t\t\tlastCompaction = result;\n\t\t\t\t\tawait this.applyResult(result, fromExtension);\n\t\t\t\t},\n\t\t\t\tverifyPostApplyEffect: reason === \"overflow\" ? () => false : undefined,\n\t\t\t\tonTransition: ({ cycle, cause, detail }) => {\n\t\t\t\t\tthis.deps.emit({\n\t\t\t\t\t\ttype: \"warning\",\n\t\t\t\t\t\tmessage: `auto-compaction cycle ${cycle}: ${cause}${detail ? ` (${detail})` : \"\"} — retrying from step 0 (${this.deps.describeSummarizer()})`,\n\t\t\t\t\t});\n\t\t\t\t},\n\t\t\t\tsignal,\n\t\t\t});\n\n\t\t\tif (outcome.kind === \"skip\") {\n\t\t\t\tthis.deps.emit({\n\t\t\t\t\ttype: \"compaction_end\",\n\t\t\t\t\treason,\n\t\t\t\t\tresult: undefined,\n\t\t\t\t\taborted: false,\n\t\t\t\t\twillRetry: false,\n\t\t\t\t\tskipReason: outcome.reason,\n\t\t\t\t});\n\t\t\t\treturn hadQueuedMessages || this.deps.agent.hasQueuedMessages();\n\t\t\t}\n\t\t\tif (outcome.kind === \"failed\") {\n\t\t\t\tif (outcome.reason === \"aborted\") {\n\t\t\t\t\tthis.deps.emit({ type: \"compaction_end\", reason, result: undefined, aborted: true, willRetry: false });\n\t\t\t\t\treturn hadQueuedMessages || this.deps.agent.hasQueuedMessages();\n\t\t\t\t}\n\t\t\t\tthrow new Error(outcome.reason);\n\t\t\t}\n\t\t\tif (extensionCancelled || signal.aborted) {\n\t\t\t\tthis.deps.emit({ type: \"compaction_end\", reason, result: undefined, aborted: true, willRetry: false });\n\t\t\t\treturn hadQueuedMessages || this.deps.agent.hasQueuedMessages();\n\t\t\t}\n\n\t\t\tconst result = outcome.kind === \"success\" ? outcome.result : lastCompaction;\n\t\t\tif (!result) throw new Error(\"Auto-compaction succeeded without a result\");\n\t\t\tthis.deps.emit({ type: \"compaction_end\", reason, result, aborted: false, willRetry });\n\t\t\tif (willRetry) {\n\t\t\t\tconst messages = this.deps.agent.state.messages;\n\t\t\t\tconst lastMessage = messages[messages.length - 1];\n\t\t\t\tif (lastMessage?.role === \"assistant\" && lastMessage.stopReason === \"error\") {\n\t\t\t\t\tthis.deps.agent.state.messages = messages.slice(0, -1);\n\t\t\t\t}\n\t\t\t\treturn true;\n\t\t\t}\n\t\t\treturn hadQueuedMessages || this.deps.agent.hasQueuedMessages();\n\t\t} catch (error) {\n\t\t\tconst errorMessage = error instanceof Error ? error.message : \"compaction failed\";\n\t\t\tthis.deps.emit({\n\t\t\t\ttype: \"compaction_end\",\n\t\t\t\treason,\n\t\t\t\tresult: undefined,\n\t\t\t\taborted: false,\n\t\t\t\twillRetry: false,\n\t\t\t\terrorMessage:\n\t\t\t\t\treason === \"overflow\"\n\t\t\t\t\t\t? `Context overflow recovery failed: ${errorMessage}`\n\t\t\t\t\t\t: `Auto-compaction failed: ${errorMessage}`,\n\t\t\t});\n\t\t\treturn hadQueuedMessages || this.deps.agent.hasQueuedMessages();\n\t\t} finally {\n\t\t\tthis.autoAbortController = undefined;\n\t\t}\n\t}\n\n\tasync compactWithRetry(\n\t\trun: () => Promise<CompactionResult>,\n\t\tsignal: AbortSignal,\n\t\tprovider?: string,\n\t): Promise<CompactionResult> {\n\t\treturn runCompactionWithRetry({\n\t\t\trun,\n\t\t\tsignal,\n\t\t\tprovider,\n\t\t\tgetRetrySettings: () => this.deps.settingsManager.getRetrySettings(),\n\t\t\trecordFailure: (record) => this.deps.getFailureCorpus().record(record),\n\t\t});\n\t}\n\n\tprivate findLastAssistantMessage(): AssistantMessage | undefined {\n\t\tfor (let index = this.deps.agent.state.messages.length - 1; index >= 0; index--) {\n\t\t\tconst message = this.deps.agent.state.messages[index];\n\t\t\tif (message.role === \"assistant\") return message;\n\t\t}\n\t\treturn undefined;\n\t}\n\n\tprivate async getExtensionCompaction(\n\t\tpreparation: CompactionPreparation,\n\t\tbranchEntries: ReturnType<SessionManager[\"getBranch\"]>,\n\t\tcustomInstructions: string | undefined,\n\t\tsignal: AbortSignal,\n\t): Promise<{ cancelled: boolean; result?: CompactionResult }> {\n\t\tconst extensionRunner = this.deps.getExtensionRunner();\n\t\tif (!extensionRunner.hasHandlers(\"session_before_compact\")) return { cancelled: false };\n\t\tconst extensionResult = (await extensionRunner.emit({\n\t\t\ttype: \"session_before_compact\",\n\t\t\tpreparation,\n\t\t\tbranchEntries,\n\t\t\tcustomInstructions,\n\t\t\tsignal,\n\t\t})) as SessionBeforeCompactResult | undefined;\n\t\treturn { cancelled: extensionResult?.cancel === true, result: extensionResult?.compaction };\n\t}\n\n\tprivate async applyResult(result: CompactionResult, fromExtension: boolean): Promise<void> {\n\t\tthis.deps.sessionManager.appendCompaction(\n\t\t\tresult.summary,\n\t\t\tresult.firstKeptEntryId,\n\t\t\tresult.tokensBefore,\n\t\t\tresult.details,\n\t\t\tfromExtension,\n\t\t\tresult.usage,\n\t\t);\n\t\tthis.deps.refreshAfterCompaction();\n\t\tconst savedEntry = this.deps.sessionManager\n\t\t\t.getEntries()\n\t\t\t.find((entry) => entry.type === \"compaction\" && entry.summary === result.summary) as\n\t\t\t| CompactionEntry\n\t\t\t| undefined;\n\t\tif (savedEntry) {\n\t\t\tawait this.deps.getExtensionRunner().emit({\n\t\t\t\ttype: \"session_compact\",\n\t\t\t\tcompactionEntry: savedEntry,\n\t\t\t\tfromExtension,\n\t\t\t});\n\t\t}\n\t}\n}\n"]}

package/dist/core/compaction-controller.js

@@ -0,0 +1,452 @@
+import { classifyFailure, computeRetryDelayMs, DEFAULT_RETRY_POLICY, sleepAbortable } from "@caupulican/pi-agent-core";
+import { calculateContextTokens, compact, createDeterministicCompaction, estimateContextTokens, getLatestCompactionEntry, prepareCompaction, runCompactionLoop, shouldCompact, } from "@caupulican/pi-agent-core/node";
+import { isContextOverflow } from "@caupulican/pi-ai";
+import { formatNoModelSelectedMessage } from "./auth-guidance.js";
+export async function runCompactionWithRetry(options) {
+    const retrySettings = options.getRetrySettings();
+    const maxAttempts = retrySettings.enabled ? Math.max(1, retrySettings.maxRetries + 1) : 1;
+    const policy = {
+        maxAttempts,
+        baseDelayMs: retrySettings.baseDelayMs,
+        maxDelayMs: DEFAULT_RETRY_POLICY.maxDelayMs,
+        jitterRatio: 0,
+    };
+    for (let attempt = 1;; attempt++) {
+        try {
+            return await options.run();
+        }
+        catch (error) {
+            if (options.signal.aborted || attempt >= maxAttempts)
+                throw error;
+            const message = error instanceof Error ? error.message : String(error);
+            const classified = classifyFailure({ message, provider: options.provider });
+            options.recordFailure({ provider: options.provider, message, classified });
+            if (!classified.retryable)
+                throw error;
+            await sleepAbortable(computeRetryDelayMs(policy, attempt, { retryAfterMs: classified.retryAfterMs }), options.signal);
+        }
+    }
+}
+/** Owns compaction detection, execution, retry, persistence, notification, and cancellation. */
+export class CompactionController {
+    manualAbortController;
+    autoAbortController;
+    overflowRecoveryAttempted = false;
+    deps;
+    constructor(deps) {
+        this.deps = deps;
+    }
+    isRunning() {
+        return this.manualAbortController !== undefined || this.autoAbortController !== undefined;
+    }
+    resetOverflowRecovery() {
+        this.overflowRecoveryAttempted = false;
+    }
+    abort() {
+        this.manualAbortController?.abort();
+        this.autoAbortController?.abort();
+    }
+    checkContextWindowUsageWarning() {
+        const model = this.deps.getModel();
+        if (!model)
+            return;
+        const contextWindow = model.contextWindow ?? 0;
+        if (contextWindow <= 0)
+            return;
+        const systemPromptTokens = Math.ceil((this.deps.agent.state.systemPrompt ?? "").length / 4);
+        let toolsChars = 0;
+        for (const tool of this.deps.agent.state.tools || []) {
+            toolsChars += tool.name.length;
+            toolsChars += tool.description?.length ?? 0;
+            if (tool.parameters)
+                toolsChars += JSON.stringify(tool.parameters).length;
+        }
+        const baseTokens = systemPromptTokens + Math.ceil(toolsChars / 4);
+        if (baseTokens >= contextWindow) {
+            this.deps.emit({
+                type: "warning",
+                message: `Base configuration (system prompt and active tools) consumes ${baseTokens} tokens, which exceeds the model's context window of ${contextWindow} tokens. The model cannot process any prompts in this state.`,
+            });
+        }
+        else if (baseTokens >= contextWindow * 0.7) {
+            this.deps.emit({
+                type: "warning",
+                message: `Base configuration (system prompt and active tools) consumes ${baseTokens} tokens (${Math.round((baseTokens / contextWindow) * 100)}% of the ${contextWindow} context window). This leaves very little room for conversation history and may cause immediate compaction or context overflow.`,
+            });
+        }
+    }
+    async maybeCompactBeforeContextTransform(messages, projectContextGc) {
+        const settings = this.deps.getAdaptedSettings();
+        const model = this.deps.getModel();
+        const contextWindow = model?.contextWindow ?? 0;
+        if (!settings.enabled || !model || contextWindow <= 0 || this.isRunning())
+            return false;
+        const triggerTokens = model.autoCompactionTriggerTokens;
+        const contextTokens = this.deps.estimateCurrentContextTokens(messages);
+        if (!shouldCompact(contextTokens, contextWindow, settings, triggerTokens))
+            return false;
+        const projectedMessages = projectContextGc(messages);
+        const projectedTokens = this.deps.estimateCurrentContextTokens(projectedMessages);
+        if (!shouldCompact(projectedTokens, contextWindow, settings, triggerTokens))
+            return false;
+        const latestBefore = getLatestCompactionEntry(this.deps.sessionManager.getBranch())?.id;
+        await this.deps.runAutoCompaction("threshold", false);
+        const latestAfter = getLatestCompactionEntry(this.deps.sessionManager.getBranch())?.id;
+        return Boolean(latestAfter && latestAfter !== latestBefore);
+    }
+    async compact(customInstructions) {
+        this.deps.disconnectAgent();
+        await this.deps.abortForeground();
+        this.manualAbortController = new AbortController();
+        this.deps.emit({ type: "compaction_start", reason: "manual" });
+        try {
+            const sessionModel = this.deps.getModel();
+            if (!sessionModel)
+                throw new Error(formatNoModelSelectedMessage());
+            const selectedCompactionModel = this.deps.resolveModel(sessionModel);
+            if (this.deps.isRawStream())
+                await this.deps.getRequestAuth(selectedCompactionModel);
+            const selectionReason = this.deps.getSelectionReason() ?? "unknown";
+            const settings = this.deps.getAdaptedSettings();
+            const initialBranch = this.deps.sessionManager.getBranch();
+            const initialPreparation = prepareCompaction(initialBranch, settings);
+            if (!initialPreparation) {
+                const lastEntry = initialBranch[initialBranch.length - 1];
+                if (lastEntry?.type === "compaction")
+                    throw new Error("Already compacted");
+                throw new Error("Nothing to compact (session too small)");
+            }
+            const signal = this.manualAbortController.signal;
+            const extension = await this.getExtensionCompaction(initialPreparation, initialBranch, customInstructions, signal);
+            if (extension.cancelled)
+                throw new Error("Compaction cancelled");
+            if (extension.result) {
+                await this.applyResult(extension.result, true);
+                this.deps.emit({
+                    type: "compaction_end",
+                    reason: "manual",
+                    result: extension.result,
+                    aborted: false,
+                    willRetry: false,
+                });
+                return extension.result;
+            }
+            let appliedResult;
+            const outcome = await runCompactionLoop({
+                measureLiveTokens: () => Math.max(this.deps.estimateCurrentContextTokens(this.deps.agent.state.messages), 1),
+                shouldCompact: () => true,
+                getPostApplyMargin: () => 0,
+                getBranch: () => this.deps.sessionManager.getBranch(),
+                getBaseKeepRecentTokens: () => settings.keepRecentTokens,
+                resolveModelAndAuth: async (modelTier) => {
+                    const model = modelTier === "cheap" ? selectedCompactionModel : sessionModel;
+                    return this.deps.resolveModelAndAuth(model, sessionModel);
+                },
+                summarizeAndVerify: async (params, model, apiKey, headers, branch) => {
+                    const preparation = prepareCompaction(branch, { ...settings, keepRecentTokens: params.keepRecentTokens }, { allowTrailingCompactionAsPrevious: true });
+                    if (!preparation)
+                        throw new Error("Nothing to compact (session too small)");
+                    const compactionThinkingLevel = this.deps.resolveThinkingLevel(model, sessionModel);
+                    const result = await this.deps.compactWithRetry(() => compact(preparation, model, apiKey, headers, customInstructions, signal, compactionThinkingLevel, this.deps.agent.streamFn, this.deps.buildPreDigest(), { chunked: params.chunked }), signal, model.provider);
+                    return { result };
+                },
+                buildDeterministicCheckpoint: () => ({ result: createDeterministicCompaction(initialPreparation) }),
+                apply: async (result) => {
+                    if (signal.aborted)
+                        throw new Error("Compaction cancelled");
+                    await this.applyResult(result, false);
+                    appliedResult = result;
+                },
+                verifyPostApplyEffect: () => false,
+                onTransition: ({ cycle, cause, detail }) => {
+                    this.deps.emit({
+                        type: "warning",
+                        message: `manual compaction cycle ${cycle}: ${cause}${detail ? ` (${detail})` : ""} — retrying from step 0 (${this.deps.describeSummarizer()})`,
+                    });
+                },
+                signal,
+            });
+            if (outcome.kind === "failed") {
+                if (outcome.reason === "aborted")
+                    throw new Error("Compaction cancelled");
+                throw new Error(`manual compaction failed after retry ladder using ${selectedCompactionModel.provider}/${selectedCompactionModel.id} (${selectionReason}); first failure: ${outcome.reason}`);
+            }
+            if (outcome.kind === "skip" || !appliedResult) {
+                throw new Error(outcome.kind === "skip" ? outcome.reason : "Compaction failed");
+            }
+            this.deps.emit({
+                type: "compaction_end",
+                reason: "manual",
+                result: appliedResult,
+                aborted: false,
+                willRetry: false,
+            });
+            return appliedResult;
+        }
+        catch (error) {
+            const message = error instanceof Error ? error.message : String(error);
+            const aborted = message === "Compaction cancelled" || (error instanceof Error && error.name === "AbortError");
+            this.deps.emit({
+                type: "compaction_end",
+                reason: "manual",
+                result: undefined,
+                aborted,
+                willRetry: false,
+                errorMessage: aborted ? undefined : `Compaction failed: ${message}`,
+            });
+            throw error;
+        }
+        finally {
+            this.manualAbortController = undefined;
+            this.deps.reconnectAgent();
+        }
+    }
+    async check(assistantMessage, skipAbortedCheck = true) {
+        const settings = this.deps.getAdaptedSettings();
+        if (!settings.enabled)
+            return false;
+        if (skipAbortedCheck && assistantMessage.stopReason === "aborted")
+            return false;
+        const model = this.deps.getModel();
+        const contextWindow = model?.contextWindow ?? 0;
+        const sameModel = model && assistantMessage.provider === model.provider && assistantMessage.model === model.id;
+        const compactionEntry = getLatestCompactionEntry(this.deps.sessionManager.getBranch());
+        const assistantIsFromBeforeCompaction = compactionEntry !== null && assistantMessage.timestamp <= new Date(compactionEntry.timestamp).getTime();
+        if (assistantIsFromBeforeCompaction)
+            return false;
+        if (sameModel && isContextOverflow(assistantMessage, contextWindow)) {
+            if (this.overflowRecoveryAttempted) {
+                this.deps.emit({
+                    type: "compaction_end",
+                    reason: "overflow",
+                    result: undefined,
+                    aborted: false,
+                    willRetry: false,
+                    errorMessage: "Context overflow recovery failed after one compact-and-retry attempt. Try reducing context or switching to a larger-context model.",
+                });
+                return false;
+            }
+            this.overflowRecoveryAttempted = true;
+            const messages = this.deps.agent.state.messages;
+            if (messages.length > 0 && messages[messages.length - 1].role === "assistant") {
+                this.deps.agent.state.messages = messages.slice(0, -1);
+            }
+            return this.deps.runAutoCompaction("overflow", true);
+        }
+        let contextTokens;
+        if (assistantMessage.stopReason === "error") {
+            const messages = this.deps.agent.state.messages;
+            const estimate = estimateContextTokens(messages);
+            if (estimate.lastUsageIndex === null)
+                return false;
+            const usageMessage = messages[estimate.lastUsageIndex];
+            if (compactionEntry &&
+                usageMessage.role === "assistant" &&
+                usageMessage.timestamp <= new Date(compactionEntry.timestamp).getTime()) {
+                return false;
+            }
+            contextTokens = estimate.tokens;
+        }
+        else {
+            contextTokens = calculateContextTokens(assistantMessage.usage);
+            const estimate = estimateContextTokens(this.deps.agent.state.messages);
+            if (estimate.lastUsageIndex !== null) {
+                const usageMessage = this.deps.agent.state.messages[estimate.lastUsageIndex];
+                const usageIsPostCompaction = !(compactionEntry &&
+                    usageMessage.role === "assistant" &&
+                    usageMessage.timestamp <= new Date(compactionEntry.timestamp).getTime());
+                if (usageIsPostCompaction)
+                    contextTokens = Math.max(contextTokens, estimate.tokens);
+            }
+        }
+        if (shouldCompact(contextTokens, contextWindow, settings, model?.autoCompactionTriggerTokens)) {
+            return this.deps.runAutoCompaction("threshold", false);
+        }
+        return false;
+    }
+    measureLiveContextTokens() {
+        const estimatedTokens = this.deps.estimateCurrentContextTokens(this.deps.agent.state.messages);
+        const assistantMessage = this.findLastAssistantMessage();
+        if (!assistantMessage || assistantMessage.stopReason === "error" || assistantMessage.stopReason === "aborted") {
+            return estimatedTokens;
+        }
+        const compactionEntry = getLatestCompactionEntry(this.deps.sessionManager.getBranch());
+        if (compactionEntry && assistantMessage.timestamp <= new Date(compactionEntry.timestamp).getTime()) {
+            return estimatedTokens;
+        }
+        return Math.max(calculateContextTokens(assistantMessage.usage), estimatedTokens);
+    }
+    async runAuto(reason, willRetry) {
+        const settings = this.deps.getAdaptedSettings();
+        const model = this.deps.getModel();
+        this.deps.emit({ type: "compaction_start", reason });
+        const hadQueuedMessages = this.deps.agent.hasQueuedMessages();
+        this.autoAbortController = new AbortController();
+        const signal = this.autoAbortController.signal;
+        let fromExtension = false;
+        let lastCompaction;
+        let extensionCancelled = false;
+        try {
+            if (!model) {
+                this.deps.emit({
+                    type: "compaction_end",
+                    reason,
+                    result: undefined,
+                    aborted: false,
+                    willRetry: false,
+                    skipReason: "no model selected",
+                });
+                return hadQueuedMessages || this.deps.agent.hasQueuedMessages();
+            }
+            const contextWindow = model.contextWindow;
+            const margin = Math.max(0, Math.floor(0.01 * contextWindow));
+            const outcome = await runCompactionLoop({
+                getBranch: () => this.deps.sessionManager.getBranch(),
+                measureLiveTokens: () => this.deps.measureLiveContextTokens(),
+                shouldCompact: reason === "overflow"
+                    ? () => true
+                    : (tokens) => shouldCompact(tokens, contextWindow, settings, model.autoCompactionTriggerTokens),
+                getPostApplyMargin: () => margin,
+                getBaseKeepRecentTokens: () => settings.keepRecentTokens,
+                resolveModelAndAuth: async (modelTier) => this.deps.resolveModelAndAuth(modelTier === "session" ? model : this.deps.resolveModel(model), model),
+                summarizeAndVerify: async (params, compactModel, apiKey, headers, branchEntries) => {
+                    fromExtension = false;
+                    const preparation = prepareCompaction(branchEntries, {
+                        ...settings,
+                        keepRecentTokens: params.keepRecentTokens,
+                    });
+                    if (!preparation)
+                        throw new Error("already compacted");
+                    const compactionThinkingLevel = this.deps.resolveThinkingLevel(compactModel, model);
+                    const extension = await this.getExtensionCompaction(preparation, branchEntries, undefined, signal);
+                    if (extension.cancelled) {
+                        extensionCancelled = true;
+                        throw new Error("auto-compaction-cancelled");
+                    }
+                    if (extension.result) {
+                        fromExtension = true;
+                        return { result: extension.result };
+                    }
+                    const result = await this.deps.compactWithRetry(() => compact(preparation, compactModel, apiKey, headers, undefined, signal, compactionThinkingLevel, this.deps.agent.streamFn, this.deps.buildPreDigest(), { chunked: params.chunked }), signal, compactModel.provider);
+                    return { result };
+                },
+                buildDeterministicCheckpoint: () => {
+                    const preparation = prepareCompaction(this.deps.sessionManager.getBranch(), settings);
+                    if (!preparation)
+                        throw new Error("already compacted");
+                    fromExtension = false;
+                    return { result: createDeterministicCompaction(preparation) };
+                },
+                apply: async (result) => {
+                    lastCompaction = result;
+                    await this.applyResult(result, fromExtension);
+                },
+                verifyPostApplyEffect: reason === "overflow" ? () => false : undefined,
+                onTransition: ({ cycle, cause, detail }) => {
+                    this.deps.emit({
+                        type: "warning",
+                        message: `auto-compaction cycle ${cycle}: ${cause}${detail ? ` (${detail})` : ""} — retrying from step 0 (${this.deps.describeSummarizer()})`,
+                    });
+                },
+                signal,
+            });
+            if (outcome.kind === "skip") {
+                this.deps.emit({
+                    type: "compaction_end",
+                    reason,
+                    result: undefined,
+                    aborted: false,
+                    willRetry: false,
+                    skipReason: outcome.reason,
+                });
+                return hadQueuedMessages || this.deps.agent.hasQueuedMessages();
+            }
+            if (outcome.kind === "failed") {
+                if (outcome.reason === "aborted") {
+                    this.deps.emit({ type: "compaction_end", reason, result: undefined, aborted: true, willRetry: false });
+                    return hadQueuedMessages || this.deps.agent.hasQueuedMessages();
+                }
+                throw new Error(outcome.reason);
+            }
+            if (extensionCancelled || signal.aborted) {
+                this.deps.emit({ type: "compaction_end", reason, result: undefined, aborted: true, willRetry: false });
+                return hadQueuedMessages || this.deps.agent.hasQueuedMessages();
+            }
+            const result = outcome.kind === "success" ? outcome.result : lastCompaction;
+            if (!result)
+                throw new Error("Auto-compaction succeeded without a result");
+            this.deps.emit({ type: "compaction_end", reason, result, aborted: false, willRetry });
+            if (willRetry) {
+                const messages = this.deps.agent.state.messages;
+                const lastMessage = messages[messages.length - 1];
+                if (lastMessage?.role === "assistant" && lastMessage.stopReason === "error") {
+                    this.deps.agent.state.messages = messages.slice(0, -1);
+                }
+                return true;
+            }
+            return hadQueuedMessages || this.deps.agent.hasQueuedMessages();
+        }
+        catch (error) {
+            const errorMessage = error instanceof Error ? error.message : "compaction failed";
+            this.deps.emit({
+                type: "compaction_end",
+                reason,
+                result: undefined,
+                aborted: false,
+                willRetry: false,
+                errorMessage: reason === "overflow"
+                    ? `Context overflow recovery failed: ${errorMessage}`
+                    : `Auto-compaction failed: ${errorMessage}`,
+            });
+            return hadQueuedMessages || this.deps.agent.hasQueuedMessages();
+        }
+        finally {
+            this.autoAbortController = undefined;
+        }
+    }
+    async compactWithRetry(run, signal, provider) {
+        return runCompactionWithRetry({
+            run,
+            signal,
+            provider,
+            getRetrySettings: () => this.deps.settingsManager.getRetrySettings(),
+            recordFailure: (record) => this.deps.getFailureCorpus().record(record),
+        });
+    }
+    findLastAssistantMessage() {
+        for (let index = this.deps.agent.state.messages.length - 1; index >= 0; index--) {
+            const message = this.deps.agent.state.messages[index];
+            if (message.role === "assistant")
+                return message;
+        }
+        return undefined;
+    }
+    async getExtensionCompaction(preparation, branchEntries, customInstructions, signal) {
+        const extensionRunner = this.deps.getExtensionRunner();
+        if (!extensionRunner.hasHandlers("session_before_compact"))
+            return { cancelled: false };
+        const extensionResult = (await extensionRunner.emit({
+            type: "session_before_compact",
+            preparation,
+            branchEntries,
+            customInstructions,
+            signal,
+        }));
+        return { cancelled: extensionResult?.cancel === true, result: extensionResult?.compaction };
+    }
+    async applyResult(result, fromExtension) {
+        this.deps.sessionManager.appendCompaction(result.summary, result.firstKeptEntryId, result.tokensBefore, result.details, fromExtension, result.usage);
+        this.deps.refreshAfterCompaction();
+        const savedEntry = this.deps.sessionManager
+            .getEntries()
+            .find((entry) => entry.type === "compaction" && entry.summary === result.summary);
+        if (savedEntry) {
+            await this.deps.getExtensionRunner().emit({
+                type: "session_compact",
+                compactionEntry: savedEntry,
+                fromExtension,
+            });
+        }
+    }
+}
+//# sourceMappingURL=compaction-controller.js.map