@caupulican/pi-adaptative 0.81.27 → 0.81.30

package/dist/core/agent-session.js

@@ -9,6 +9,7 @@ import { getAgentDir } from "../config.js";
 import { stripFrontmatter } from "../utils/frontmatter.js";
 import { formatNoApiKeyFoundMessage, formatNoModelSelectedMessage } from "./auth-guidance.js";
 import { buildForegroundEnvelope, formatForegroundEnvelopeObservation } from "./autonomy/foreground-envelope.js";
+import { evaluateToolGate } from "./autonomy/gates.js";
 import { AutonomyTelemetry } from "./autonomy-telemetry.js";
 import { BackgroundLaneController } from "./background-lane-controller.js";
 import { BashExecutionController } from "./bash-execution-controller.js";
@@ -49,6 +50,8 @@ 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";
+import { ToolPerformanceStore } from "./tool-selection/tool-performance-store.js";
+import { ToolSelectionController } from "./tool-selection/tool-selection-controller.js";
 // ============================================================================
 // Stream-idle watchdog wiring
 // ============================================================================
@@ -154,6 +157,7 @@ export class AgentSession {
     _bash;
     _profileFilter;
     _toolGate;
+    _toolSelection;
     _extensionRunner;
     _turnIndex = 0;
     _currentForegroundEnvelope;
@@ -368,6 +372,7 @@ export class AgentSession {
             getEvidenceBundleSnapshot: () => this.getEvidenceBundleSnapshot(),
             saveEvidenceBundleSnapshot: (bundle) => this.saveEvidenceBundleSnapshot(bundle),
             saveWorkerResultSnapshot: (result, request) => this.saveWorkerResultSnapshot(result, request),
+            readMemoryForLane: (query) => this._memory.readMemoryForLane(query),
             addSpawnedUsage: (usage, opts) => this.addSpawnedUsage(usage, opts),
             runIsolatedCompletion: (opts) => this.runIsolatedCompletion(opts),
             continueGoalLoop: (options) => this.continueGoalLoop(options),
@@ -402,6 +407,7 @@ export class AgentSession {
             getSessionManager: () => this.sessionManager,
             getSettingsManager: () => this.settingsManager,
             getModelRegistry: () => this._modelRegistry,
+            getModel: () => this.model,
             getAgentDir: () => this._agentDir,
             getCwd: () => this._cwd,
             getActiveToolNames: () => this.getActiveToolNames(),
@@ -463,8 +469,10 @@ export class AgentSession {
             isChildSession: () => this._isChildSession,
             isDisposed: () => this._disposed,
             getReflectionSignal: () => this._reflectionAbort.signal,
+            resolveTextToolCallProtocol: (model) => this._textProtocolFlag(model),
             archivePromotedSkill: (name) => this.archivePromotedSkill(name),
             emitAutonomyTelemetry: (event) => this._emitAutonomyTelemetry(event),
+            ensureModelReady: (model) => this._localRuntimeController.ensureIsolatedModelReady(model),
             addSpawnedUsage: (usage, opts) => this.addSpawnedUsage(usage, opts),
             saveLearningDecisionSnapshot: (decision) => this.saveLearningDecisionSnapshot(decision),
         });
@@ -544,6 +552,9 @@ export class AgentSession {
             getGoalStateSnapshot: () => this.getGoalStateSnapshot(),
             saveGoalStateSnapshot: (state) => this.saveGoalStateSnapshot(state),
             getContextGcReport: (messages) => this.getContextGcReport(messages),
+            startWorkerDelegation: (request) => this._backgroundLanes.startWorkerDelegation(request),
+            getWorkerLaneRecords: () => this._backgroundLanes.getLaneRecords(),
+            getWorkerResultSnapshots: () => this.getWorkerResultSnapshots(),
             runWorkerDelegationOnce: (request) => this.runWorkerDelegationOnce(request),
             runModelFitness: (args) => this.runModelFitness(args),
             resolveCurationModelIfFit: () => this._resolveCurationModelIfFit(),
@@ -570,6 +581,9 @@ export class AgentSession {
         const previousToolArgumentValidation = this.agent.onToolArgumentValidation;
         this.agent.onToolArgumentValidation = (event) => {
             const taggedEvent = this._tagModelAdaptationRuleTeaching(event);
+            if (taggedEvent.outcome === "repaired" || taggedEvent.outcome === "bounced") {
+                this._toolSelection.recordValidation(taggedEvent.tool, taggedEvent.outcome);
+            }
             previousToolArgumentValidation?.(taggedEvent);
             const logRecord = this._toolRecoveryLogger.recordToolArgumentValidation(taggedEvent);
             if (!logRecord)
@@ -629,12 +643,41 @@ export class AgentSession {
             isExplicitThinking: () => this._isExplicitThinking,
             setThinkingLevel: (level) => this.setThinkingLevel(level, { persistSettings: false }),
         });
+        this._toolSelection = new ToolSelectionController({
+            store: ToolPerformanceStore.forAgentDir(this._agentDir),
+            getModelRef: () => {
+                const model = this.model;
+                return model ? formatModelRouterModel(model) : "unknown";
+            },
+            getActiveTools: () => {
+                const activeNames = new Set(this.getActiveToolNames());
+                return this.getAllTools()
+                    .filter((tool) => activeNames.has(tool.name))
+                    .map((tool) => ({
+                    name: tool.name,
+                    description: tool.description,
+                    parameters: tool.parameters,
+                }));
+            },
+            isCandidateAllowed: (toolName) => {
+                const envelope = this.capabilityEnvelope;
+                if (!envelope)
+                    return true;
+                return (evaluateToolGate({
+                    toolName,
+                    args: {},
+                    cwd: this._cwd,
+                    envelope,
+                }).outcome === "allow");
+            },
+        });
         this._toolGate = new ToolGateController({
             maybeEscalateToolCall: (toolName, args) => this._modelRouter.maybeEscalateToolCall(toolName, args),
             getCwd: () => this._cwd,
             getCapabilityEnvelope: () => this.capabilityEnvelope,
             recordGateOutcome: (outcome) => this._recordGateOutcome(outcome),
             getExtensionRunner: () => this._extensionRunner,
+            getToolSelectionController: () => this._toolSelection,
         });
         // Always subscribe to agent events for internal handling
         // (session persistence, extensions, auto-compaction, retry logic)
@@ -695,17 +738,7 @@ export class AgentSession {
             if (controller.signal.aborted)
                 return;
             if (model.provider === OLLAMA_PROVIDER || model.provider === HF_TRANSFORMERS_PROVIDER) {
-                await this._localRuntimeController.ensureRouteModelReady({
-                    decision: {
-                        tier: "cheap",
-                        risk: "read-only",
-                        confidence: 1,
-                        reasonCode: "prefix_warmer",
-                        reasons: ["prefix warmer residency"],
-                        model: `${model.provider}/${model.id}`,
-                    },
-                    model,
-                });
+                await this._localRuntimeController.ensureIsolatedModelReady(model);
             }
             if (controller.signal.aborted)
                 return;
@@ -1436,6 +1469,16 @@ export class AgentSession {
     _buildCompactionPreDigest() {
         return this._pipeline.buildCompactionPreDigest();
     }
+    /** Drop provider-owned request/continuation caches whose prefix was invalidated by compaction. */
+    _refreshAfterCompaction() {
+        this.agent.state.messages = this.sessionManager.buildSessionContext().messages;
+        try {
+            cleanupSessionResources(this.sessionId);
+        }
+        catch {
+            // Provider cache cleanup is best-effort and must not turn an applied compaction into a failure.
+        }
+    }
     /**
      * Context composition dashboard data: decomposes the per-request payload (system prompt, tool
      * schemas, extension contributions, message classes incl. GC/policy stubs and recall pages)
@@ -1752,6 +1795,7 @@ export class AgentSession {
             await this._extensionRunner.emit({ type: "agent_end", messages: event.messages });
         }
         else if (event.type === "turn_start") {
+            this._toolSelection.startTurn();
             this._refreshForegroundEnvelope();
             const extensionEvent = {
                 type: "turn_start",
@@ -2018,8 +2062,9 @@ export class AgentSession {
      * Also rebuilds the system prompt to reflect the new tool set.
      * Changes take effect on the next agent turn.
      *
-     * artifact_retrieve is auto-activated as a companion whenever grep or find ends up
-     * in the resulting active set and artifact_retrieve is registered (i.e. not excluded/
+     * artifact_retrieve is auto-activated as a companion whenever an artifact-producing tool
+     * (grep, find, or run_toolkit_script) ends up in the resulting active set and artifact_retrieve
+     * is registered (i.e. not excluded/
      * blocked/outside an allowlist -- the registry itself is built with that same filter,
      * so registry presence already tracks "allowed"). This is enforced here, not just in
      * the settings/profile refresh flow, because this method is a public, extension-
@@ -2050,9 +2095,14 @@ export class AgentSession {
         for (const name of capabilityFiltered) {
             addIfRegistered(name);
         }
-        if (validToolNames.includes("grep") || validToolNames.includes("find")) {
+        if (validToolNames.includes("grep") ||
+            validToolNames.includes("find") ||
+            validToolNames.includes("run_toolkit_script")) {
             addIfRegistered("artifact_retrieve");
         }
+        if (validToolNames.includes("delegate")) {
+            addIfRegistered("delegate_status");
+        }
         this.agent.state.tools = tools;
         // Rebuild base system prompt with new tool set
         this._baseSystemPrompt = this._rebuildSystemPrompt(validToolNames);
@@ -2344,6 +2394,11 @@ export class AgentSession {
             if (!requestModel) {
                 throw new Error(formatNoModelSelectedMessage());
             }
+            // A manual/default local model has no RouteDecision, so the router readiness gate above is
+            // intentionally a no-op. It still needs the same managed-runtime boot/residency guarantee.
+            if (!resolvedRouteInfo) {
+                await this._localRuntimeController.ensureForegroundModelReady(requestModel);
+            }
             if (!this._modelRegistry.hasConfiguredAuth(requestModel)) {
                 const isOAuth = this._modelRegistry.isUsingOAuth(requestModel);
                 if (isOAuth) {
@@ -2454,6 +2509,7 @@ export class AgentSession {
                 this._memory.recordRecallOutcome(injectedRecall, recallQuery, responseText);
             }
         }
+        this._backgroundLanes.drainQueuedWorkerDelegations();
         this._backgroundLanes.scheduleGoalAutoContinueFromIdle(options);
         this._backgroundLanes.scheduleResearchLaneFromIdle();
     }
@@ -2850,7 +2906,7 @@ export class AgentSession {
             }
             if (extensionCompaction) {
                 this.sessionManager.appendCompaction(extensionCompaction.summary, extensionCompaction.firstKeptEntryId, extensionCompaction.tokensBefore, extensionCompaction.details, true);
-                this.agent.state.messages = this.sessionManager.buildSessionContext().messages;
+                this._refreshAfterCompaction();
                 const savedCompactionEntry = this.sessionManager
                     .getEntries()
                     .find((entry) => entry.type === "compaction" && entry.summary === extensionCompaction.summary);
@@ -2901,7 +2957,7 @@ export class AgentSession {
                     if (signal.aborted)
                         throw new Error("Compaction cancelled");
                     this.sessionManager.appendCompaction(result.summary, result.firstKeptEntryId, result.tokensBefore, result.details, fromExtension);
-                    this.agent.state.messages = this.sessionManager.buildSessionContext().messages;
+                    this._refreshAfterCompaction();
                     const savedCompactionEntry = this.sessionManager
                         .getEntries()
                         .find((entry) => entry.type === "compaction" && entry.summary === result.summary);
@@ -3185,7 +3241,7 @@ export class AgentSession {
                 apply: async (result) => {
                     lastCompaction = result;
                     this.sessionManager.appendCompaction(result.summary, result.firstKeptEntryId, result.tokensBefore, result.details, fromExtension);
-                    this.agent.state.messages = this.sessionManager.buildSessionContext().messages;
+                    this._refreshAfterCompaction();
                     const newEntries = this.sessionManager.getEntries();
                     const savedCompactionEntry = newEntries.find((entry) => entry.type === "compaction" && entry.summary === result.summary);
                     if (this._extensionRunner && savedCompactionEntry) {