@caupulican/pi-adaptative 0.80.26 → 0.80.28

package/dist/core/agent-session.js

@@ -13,7 +13,7 @@
  * Modes use this class and add their own I/O layer on top.
  */
 import { existsSync, mkdirSync, readFileSync, writeFileSync } from "node:fs";
-import { basename, dirname } from "node:path";
+import { basename, dirname, join } from "node:path";
 import { clampThinkingLevel, cleanupSessionResources, getSupportedThinkingLevels, isContextOverflow, modelsAreEqual, resetApiProviders, streamSimple, } from "@caupulican/pi-ai";
 import { theme } from "../modes/interactive/theme/theme.js";
 import { stripFrontmatter } from "../utils/frontmatter.js";
@@ -22,6 +22,7 @@ import { sleep } from "../utils/sleep.js";
 import { formatNoApiKeyFoundMessage, formatNoModelSelectedMessage } from "./auth-guidance.js";
 import { executeBashWithOperations } from "./bash-executor.js";
 import { calculateContextTokens, collectEntriesForBranchSummary, compact, estimateContextTokens, generateBranchSummary, prepareCompaction, shouldCompact, } from "./compaction/index.js";
+import { applyContextGc } from "./context-gc.js";
 import { DEFAULT_THINKING_LEVEL } from "./defaults.js";
 import { exportSessionToHtml } from "./export-html/index.js";
 import { createToolHtmlRenderer } from "./export-html/tool-renderer.js";
@@ -79,10 +80,11 @@ export class AgentSession {
     _pendingNextTurnMessages = [];
     /** Serializes prompt() submissions made while streaming so queued steering/follow-ups keep user-typed FIFO order. */
     _streamingPromptSubmissionTail = Promise.resolve();
-    // Compaction state
+    // Compaction/context hygiene state
     _compactionAbortController = undefined;
     _autoCompactionAbortController = undefined;
     _overflowRecoveryAttempted = false;
+    _latestContextGcReport = undefined;
     // Branch summarization state
     _branchSummaryAbortController = undefined;
     // Retry state
@@ -213,10 +215,11 @@ export class AgentSession {
             catch {
                 currentMessages = authoritativeMessages;
             }
+            let finalMessages = currentMessages;
             if (this._extensionRunner.hasHandlers("context")) {
-                return await this._extensionRunner.emitContext(currentMessages);
+                finalMessages = await this._extensionRunner.emitContext(currentMessages);
             }
-            return currentMessages;
+            return this._applyContextGc(finalMessages, true).messages;
         };
     }
     _installAgentTurnRefresh() {
@@ -243,6 +246,45 @@ export class AgentSession {
             tools: this.agent.state.tools.slice(),
         };
     }
+    _contextGcStorageDir() {
+        return join(this.sessionManager.getSessionDir(), "context-gc", this.sessionManager.getSessionId());
+    }
+    _applyContextGc(messages, writePayloads) {
+        try {
+            const result = applyContextGc(messages, {
+                ...this.settingsManager.getContextGcSettings(),
+                cwd: this._cwd,
+                storageDir: this._contextGcStorageDir(),
+                writePayloads,
+            });
+            this._latestContextGcReport = result.report;
+            return result;
+        }
+        catch {
+            const report = {
+                enabled: false,
+                packedCount: 0,
+                originalTokens: 0,
+                packedTokens: 0,
+                savedTokens: 0,
+                records: [],
+            };
+            this._latestContextGcReport = report;
+            return { messages, report };
+        }
+    }
+    getContextGcReport(messages) {
+        if (messages)
+            return this._applyContextGc(messages, false).report;
+        return (this._latestContextGcReport ?? {
+            enabled: this.settingsManager.getContextGcSettings().enabled,
+            packedCount: 0,
+            originalTokens: 0,
+            packedTokens: 0,
+            savedTokens: 0,
+            records: [],
+        });
+    }
     _estimateCurrentContextTokens(messages) {
         const estimate = estimateContextTokens(messages);
         const compactionEntry = getLatestCompactionEntry(this.sessionManager.getBranch());
@@ -2178,7 +2220,7 @@ export class AgentSession {
             });
         this._baseToolDefinitions = new Map(Object.entries(baseToolDefinitions).map(([name, tool]) => [name, tool]));
         if (!this._baseToolsOverride) {
-            for (const definition of createCoreDiagnosticsToolDefinitions(() => this.getActiveToolNames(), () => this.getAllTools())) {
+            for (const definition of createCoreDiagnosticsToolDefinitions(() => this.getActiveToolNames(), () => this.getAllTools(), (messages) => this.getContextGcReport(messages))) {
                 this._baseToolDefinitions.set(definition.name, definition);
             }
         }