@caupulican/pi-adaptative 0.81.21 → 0.81.23

package/dist/core/agent-session.js

@@ -44,6 +44,8 @@ import { SessionAnalytics } from "./session-analytics.js";
 import { SessionTreeNavigator } from "./session-tree-navigator.js";
 import { SystemPromptBuilder } from "./system-prompt-builder.js";
 import { ToolGateController } from "./tool-gate-controller.js";
+import { TOOL_RECOVERY_EVENT_LOG_FILE } from "./tool-recovery-log-records.js";
+import { ToolRecoveryLogger } from "./tool-recovery-logger.js";
 import { formatToolRepairHealthReport } from "./tool-repair-health.js";
 import { resolveCurrentToolRepairSettings } from "./tool-repair-settings.js";
 // ============================================================================
@@ -207,6 +209,8 @@ export class AgentSession {
     _modelRouter;
     _billingFailover;
     _failureCorpus;
+    _toolRecoveryLogger;
+    _toolRecoveryEventLogPath;
     _skillCuratorInstance;
     _disposed = false;
     _reflectionAbort = new AbortController();
@@ -397,8 +401,17 @@ export class AgentSession {
             addSpawnedUsage: (usage, opts) => this.addSpawnedUsage(usage, opts),
             runIsolatedCompletion: (opts) => this.runIsolatedCompletion(opts),
         });
+        const failureCorpusPath = join(this._agentDir, "state", "failure-corpus.jsonl");
+        this._toolRecoveryEventLogPath = join(this._agentDir, "state", TOOL_RECOVERY_EVENT_LOG_FILE);
+        const toolRepairSettings = this._toolRepairSettings();
         this._failureCorpus = new FailureCorpusRecorder({
-            filePath: join(this._agentDir, "state", "failure-corpus.jsonl"),
+            filePath: failureCorpusPath,
+        });
+        this._toolRecoveryLogger = new ToolRecoveryLogger({
+            enabled: toolRepairSettings.logging,
+            sessionId: this.sessionManager.getSessionId(),
+            eventLogPath: this._toolRecoveryEventLogPath,
+            failureCorpusPath,
         });
         this._billingFailover = new BillingFailoverController({
             agent: this.agent,
@@ -538,13 +551,16 @@ export class AgentSession {
             getSessionManager: () => this.sessionManager,
             getSettingsManager: () => this.settingsManager,
             getToolDefinition: (name) => this.getToolDefinition(name),
+            getToolRecoveryEventLogPath: () => this._toolRecoveryEventLogPath,
         });
         const previousToolArgumentValidation = this.agent.onToolArgumentValidation;
         this.agent.onToolArgumentValidation = (event) => {
             const taggedEvent = this._tagModelAdaptationRuleTeaching(event);
             previousToolArgumentValidation?.(taggedEvent);
-            this._analytics.recordToolArgumentValidation(taggedEvent);
-            this._recordToolValidationBounce(taggedEvent);
+            const logRecord = this._toolRecoveryLogger.recordToolArgumentValidation(taggedEvent);
+            if (!logRecord)
+                return;
+            this._analytics.recordToolArgumentValidation(logRecord);
             this._handleTextToolProtocolValidationOutcome(taggedEvent);
             this._handleModelAdaptationTelemetry(taggedEvent);
         };
@@ -920,7 +936,6 @@ export class AgentSession {
     }
     _applyToolRepairLayerSettings() {
         const settings = this._toolRepairSettings();
-        this.agent.toolArgumentRepairEnabled = settings.repair;
         this.agent.toolArgumentTeachEnabled = settings.teach;
     }
     _getModelAdaptationRulesForPrompt() {
@@ -1273,18 +1288,6 @@ export class AgentSession {
     _looksLikeTextToolProtocolAttempt(text) {
         return /<pi:call\b|<tool_call\b|```(?:tool|tool_call)[\s\S]*"name"\s*:/i.test(text);
     }
-    _recordToolValidationBounce(event) {
-        if (event.outcome !== "bounced" || !event.failureShape || event.failureShape.length === 0)
-            return;
-        this._failureCorpus.recordToolValidation({
-            provider: event.provider ?? this.agent.state.model?.provider,
-            modelId: event.model ?? this.agent.state.model?.id,
-            tool: event.tool,
-            failureModes: event.failureModes,
-            shape: event.failureShape,
-            errorKeywords: event.errorKeywords,
-        });
-    }
     _tagModelAdaptationRuleTeaching(event) {
         if (!this._toolRepairSettings().teach || event.taught !== "none")
             return event;
@@ -1492,7 +1495,10 @@ export class AgentSession {
         return formatContextCompositionDashboard(this.getContextCompositionReport());
     }
     formatToolRepairHealthReport() {
-        return formatToolRepairHealthReport(this._modelAdaptationStore);
+        return formatToolRepairHealthReport(this._modelAdaptationStore, new Date(), this._toolRecoveryLogger.getStats());
+    }
+    async flushToolRecoveryLogsForTests(timeoutMs = 1000) {
+        await this._toolRecoveryLogger.flush(timeoutMs);
     }
     removeToolRepairRule(model, mode) {
         return this._modelAdaptationStore.removeRule(model, mode);
@@ -1907,6 +1913,7 @@ export class AgentSession {
             .getMemoryManager()
             .shutdownAll()
             .catch(() => { });
+        void this._toolRecoveryLogger.shutdown().catch(() => { });
         cleanupSessionResources(this.sessionId);
         // Best-effort final sweep for any grep/find artifact already released (reference
         // count zero) but not yet reclaimed -- e.g. a release whose cleanup() call failed
@@ -3091,7 +3098,7 @@ export class AgentSession {
                     willRetry: false,
                     skipReason: "no model selected",
                 });
-                return false;
+                return hadQueuedMessages || this.agent.hasQueuedMessages();
             }
             const contextWindow = model.contextWindow;
             const margin = Math.max(0, Math.floor(0.01 * contextWindow));
@@ -3180,13 +3187,13 @@ export class AgentSession {
             if (outcome.kind === "failed") {
                 if (outcome.reason === "aborted") {
                     this._emit({ type: "compaction_end", reason, result: undefined, aborted: true, willRetry: false });
-                    return false;
+                    return hadQueuedMessages || this.agent.hasQueuedMessages();
                 }
                 throw new Error(outcome.reason);
             }
             if (extensionCancelled || signal.aborted) {
                 this._emit({ type: "compaction_end", reason, result: undefined, aborted: true, willRetry: false });
-                return false;
+                return hadQueuedMessages || this.agent.hasQueuedMessages();
             }
             const result = outcome.kind === "success" ? outcome.result : lastCompaction;
             if (!result) {
@@ -3215,7 +3222,7 @@ export class AgentSession {
                     ? `Context overflow recovery failed: ${errorMessage}`
                     : `Auto-compaction failed: ${errorMessage}`,
             });
-            return false;
+            return hadQueuedMessages || this.agent.hasQueuedMessages();
         }
         finally {
             this._autoCompactionAbortController = undefined;