@caupulican/pi-adaptative 0.81.3 → 0.81.4

package/dist/core/agent-session.js

@@ -1,17 +1,3 @@
-/**
- * AgentSession - Core abstraction for agent lifecycle and session management.
- *
- * This class is shared between all run modes (interactive, print, rpc).
- * It encapsulates:
- * - Agent state access
- * - Event subscription with automatic session persistence
- * - Model and thinking level management
- * - Compaction (manual and auto)
- * - Bash execution
- * - Session switching and branching
- *
- * Modes use this class and add their own I/O layer on top.
- */
 import { readFileSync } from "node:fs";
 import { basename, dirname, join } from "node:path";
 import { classifyFailure, compactToolResultDetailsForRetention, computeRetryDelayMs, createCustomMessage, DEFAULT_RETRY_POLICY, RetryController, sleepAbortable, withStreamIdleWatchdog, } from "@caupulican/pi-agent-core";
@@ -30,6 +16,7 @@ import { buildContextCompositionReport, formatContextCompositionDashboard, } fro
 import { ContextPipeline } from "./context-pipeline.js";
 import { downgradeReasoning, estimateTurnCostUsd, evaluateCostGuard } from "./cost-guard.js";
 import { appendWorkerResultSnapshot, getWorkerResultSnapshots } from "./delegation/session-worker-result.js";
+import { FailureCorpusRecorder } from "./failure-corpus.js";
 import { GatewayRegistry } from "./gateways/channel-provider.js";
 import { GoalLoopController } from "./goal-loop-controller.js";
 import { buildGoalRuntimeSnapshot, } from "./goals/goal-runtime-snapshot.js";
@@ -185,6 +172,7 @@ export class AgentSession {
      * {@link _runAgentPrompt} so the drive loop stays host-side. */
     _modelRouter;
     _billingFailover;
+    _failureCorpus;
     _skillCuratorInstance;
     _disposed = false;
     _reflectionAbort = new AbortController();
@@ -329,6 +317,12 @@ export class AgentSession {
             getModelRegistry: () => this._modelRegistry,
             isRawStream: () => this._isRawStreamSimple(this.agent.streamFn),
             getRequiredRequestAuth: (model) => this._getRequiredRequestAuth(model),
+            isModelExhausted: (ref) => this._billingFailover.isExhausted(ref),
+            getStoredFitnessReport: (ref) => this.getStoredFitnessReports().find((entry) => entry.model === ref)?.report,
+            // Live context is an over-estimate of the span to summarize (includes the kept tail) —
+            // conservative in the safe direction for the summarizer capacity check.
+            estimateSummarizationInputTokens: () => this._pipeline.estimateCurrentContextTokens(this.agent.state.messages),
+            emitWarning: (message) => this._emit({ type: "warning", message }),
         });
         this._pipeline = new ContextPipeline({
             getTurnIndex: () => this._turnIndex,
@@ -343,12 +337,16 @@ export class AgentSession {
             addSpawnedUsage: (usage, opts) => this.addSpawnedUsage(usage, opts),
             runIsolatedCompletion: (opts) => this.runIsolatedCompletion(opts),
         });
+        this._failureCorpus = new FailureCorpusRecorder({
+            filePath: join(this._agentDir, "state", "failure-corpus.jsonl"),
+        });
         this._billingFailover = new BillingFailoverController({
             agent: this.agent,
             modelRegistry: this._modelRegistry,
             exhausted: new ExhaustedProviderRegistry(),
             subscriptionHop: this.settingsManager.getFailoverSettings().subscriptionHop,
             emit: (event) => this._emit(event),
+            recordFailure: (args) => this._failureCorpus.record(args),
         });
         this._modelRouter = new ModelRouterController({
             getAgent: () => this.agent,
@@ -357,7 +355,7 @@ export class AgentSession {
             getSessionManager: () => this.sessionManager,
             getModelRegistry: () => this._modelRegistry,
             isModelExhausted: (model) => this._billingFailover.isExhausted(`${model.provider}/${model.id}`),
-            getFailoverStatus: () => this._billingFailover.getStatus(),
+            getFailoverStatus: () => ({ ...this._billingFailover.getStatus(), failureStats: this._failureCorpus.stats() }),
             getAgentDir: () => this._agentDir,
             getReflectionSignal: () => this._reflectionAbort.signal,
             getBaseSystemPrompt: () => this._baseSystemPrompt,
@@ -2551,6 +2549,7 @@ export class AgentSession {
                     throw error;
                 const message = error instanceof Error ? error.message : String(error);
                 const classified = classifyFailure({ message, provider });
+                this._failureCorpus.record({ provider, message, classified });
                 if (!classified.retryable)
                     throw error;
                 await sleepAbortable(computeRetryDelayMs(policy, attempt, { retryAfterMs: classified.retryAfterMs }), signal);
@@ -2836,11 +2835,18 @@ export class AgentSession {
         if (message.stopReason !== "error" || !message.errorMessage)
             return false;
         const contextWindow = this.model?.contextWindow ?? 0;
-        return classifyFailure({
+        const classified = classifyFailure({
             message: message.errorMessage,
             contextOverflow: isContextOverflow(message, contextWindow),
             provider: message.provider,
-        }).retryable;
+        });
+        this._failureCorpus.record({
+            provider: message.provider,
+            modelId: message.model,
+            message: message.errorMessage,
+            classified,
+        });
+        return classified.retryable;
     }
     /**
      * Prepare a retryable error for continuation with exponential backoff.