@caupulican/pi-adaptative 0.81.22 → 0.81.24

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();
@@ -326,9 +330,7 @@ export class AgentSession {
             getSessionManager: () => this.sessionManager,
             getLastModelRouterDecision: () => this._modelRouter.getLastDecision(),
             getLastResearchLaneSkipReason: () => this._backgroundLanes.getLastResearchLaneSkipReason(),
-            getSessionStats: () => this.getSessionStats(),
-            getSpawnedUsage: () => this.getSpawnedUsage(),
-            getDailyUsageTotals: () => this.getDailyUsageTotals?.(),
+            getCostSummary: () => this.getCostSummary(),
             getGoalStateSnapshot: () => this.getGoalStateSnapshot(),
             getActiveLaneCount: () => this._backgroundLanes.getActiveLaneCount(),
             getEvidenceBundleSnapshots: () => this.getEvidenceBundleSnapshots(),
@@ -397,8 +399,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 +549,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 +934,6 @@ export class AgentSession {
     }
     _applyToolRepairLayerSettings() {
         const settings = this._toolRepairSettings();
-        this.agent.toolArgumentRepairEnabled = settings.repair;
         this.agent.toolArgumentTeachEnabled = settings.teach;
     }
     _getModelAdaptationRulesForPrompt() {
@@ -1273,18 +1286,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 +1493,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 +1911,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
@@ -3626,6 +3631,9 @@ export class AgentSession {
     getDailyUsageTotals(now = new Date()) {
         return this._analytics.getDailyUsageTotals(now);
     }
+    getCostSummary(now = new Date()) {
+        return this._analytics.getCostSummary(now);
+    }
     getDailyUsageBreakdown(formatLabel, now = new Date()) {
         return this._analytics.getDailyUsageBreakdown(formatLabel, now);
     }