@caupulican/pi-adaptative 0.81.27 → 0.81.29

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;
@@ -402,6 +406,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(),
@@ -465,6 +470,7 @@ export class AgentSession {
             getReflectionSignal: () => this._reflectionAbort.signal,
             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 +550,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 +579,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 +641,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)
@@ -1752,6 +1793,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 +2060,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 +2093,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 +2392,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 +2507,7 @@ export class AgentSession {
                 this._memory.recordRecallOutcome(injectedRecall, recallQuery, responseText);
             }
         }
+        this._backgroundLanes.drainQueuedWorkerDelegations();
         this._backgroundLanes.scheduleGoalAutoContinueFromIdle(options);
         this._backgroundLanes.scheduleResearchLaneFromIdle();
     }