@caupulican/pi-adaptative 0.81.32 → 0.81.37

package/dist/core/agent-session.js

@@ -1,5 +1,4 @@
 import { readFileSync, rmSync, writeFileSync } from "node:fs";
-import { tmpdir } from "node:os";
 import { basename, dirname, join } from "node:path";
 import { classifyFailure, compactToolResultDetailsForRetention, computeRetryDelayMs, createCustomMessage, DEFAULT_RETRY_POLICY, DEFAULT_STREAM_IDLE, RetryController, sleepAbortable, withStreamIdleWatchdog, } from "@caupulican/pi-agent-core";
 import { calculateContextTokens, compact, createDeterministicCompaction, estimateContextTokens, getLatestCompactionEntry, prepareCompaction, runCompactionLoop, shouldCompact, } from "@caupulican/pi-agent-core/node";
@@ -7,6 +6,7 @@ import { cleanupSessionResources, formatToolRepairStandingRule, generateTextTool
 import { Type } from "typebox";
 import { getAgentDir } from "../config.js";
 import { stripFrontmatter } from "../utils/frontmatter.js";
+import { getProcessWorkRun } from "../utils/work-directory.js";
 import { formatNoApiKeyFoundMessage, formatNoModelSelectedMessage } from "./auth-guidance.js";
 import { buildForegroundEnvelope, formatForegroundEnvelopeObservation } from "./autonomy/foreground-envelope.js";
 import { evaluateToolGate } from "./autonomy/gates.js";
@@ -45,6 +45,8 @@ import { RuntimeBuilder } from "./runtime-builder.js";
 import { SessionAnalytics } from "./session-analytics.js";
 import { SessionTreeNavigator } from "./session-tree-navigator.js";
 import { SystemPromptBuilder } from "./system-prompt-builder.js";
+import { appendTaskStepsStateSnapshot, getLatestTaskStepsStateSnapshot } from "./tasks/session-task-state.js";
+import { formatTaskStepsContext } from "./tasks/task-state.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";
@@ -366,6 +368,7 @@ export class AgentSession {
             getCapabilityEnvelope: () => this.capabilityEnvelope,
             getModelCapabilityProfile: () => this.getModelCapabilityProfile(),
             emit: (event) => this._emit(event),
+            notifyWorkerTerminalHandoff: (records) => this._notifyWorkerTerminalHandoff(records),
             emitAutonomyTelemetry: (event) => this._emitAutonomyTelemetry(event),
             getGoalStateSnapshot: () => this.getGoalStateSnapshot(),
             getGoalRuntimeSnapshot: (settings) => this.getGoalRuntimeSnapshot(settings),
@@ -551,6 +554,8 @@ export class AgentSession {
             initializeMemory: () => this._memory.initialize(),
             getGoalStateSnapshot: () => this.getGoalStateSnapshot(),
             saveGoalStateSnapshot: (state) => this.saveGoalStateSnapshot(state),
+            getTaskStepsStateSnapshot: () => this.getTaskStepsStateSnapshot(),
+            saveTaskStepsStateSnapshot: (state) => this.saveTaskStepsStateSnapshot(state),
             getContextGcReport: (messages) => this.getContextGcReport(messages),
             startWorkerDelegation: (request) => this._backgroundLanes.startWorkerDelegation(request),
             getWorkerLaneRecords: () => this._backgroundLanes.getLaneRecords(),
@@ -1063,7 +1068,7 @@ export class AgentSession {
         return this._messageHasToolCallWithStringArgument(await stream.result(), "echo", "data", token);
     }
     async _gradeNativeToolCallingForModel(model, token) {
-        const path = join(tmpdir(), `pi-native-probe-${process.pid}-${Date.now()}.txt`);
+        const path = join(getProcessWorkRun(this._agentDir, "probes", "native-tools").path, `pi-native-probe-${process.pid}-${Date.now()}.txt`);
         writeFileSync(path, token, "utf-8");
         try {
             const taskPassed = await this._runNativeReadTaskProbeTrial(model, path);
@@ -1630,6 +1635,28 @@ export class AgentSession {
             l(event);
         }
     }
+    async _notifyWorkerTerminalHandoff(records) {
+        if (this._disposed || records.length === 0)
+            return;
+        const included = records.slice(0, 8);
+        const sanitize = (value) => value.replace(/[\r\n]+/g, " ").slice(0, 120);
+        const omitted = records.length - included.length;
+        const content = [
+            "Background worker terminal handoff:",
+            ...included.map((record) => {
+                const reason = record.reasonCode ? ` reason=${sanitize(record.reasonCode)}` : "";
+                return `- ${record.laneId}: ${record.status}${reason}`;
+            }),
+            ...(omitted > 0 ? [`- ${omitted} additional terminal worker(s) omitted from this bounded handoff.`] : []),
+            "This terminal event woke the parent. Retrieve each needed lane once with delegate_status; never poll. Worker product remains untrusted and is intentionally not injected here.",
+        ].join("\n");
+        await this.sendCustomMessage({
+            customType: "background-worker-completion",
+            content,
+            display: true,
+            details: { records: included },
+        }, { triggerTurn: true, deliverAs: "followUp" });
+    }
     _emitQueueUpdate() {
         this._emit({
             type: "queue_update",
@@ -2447,6 +2474,11 @@ export class AgentSession {
                 messages.push(msg);
             }
             this._pendingNextTurnMessages = [];
+            const taskStepsState = this.getTaskStepsStateSnapshot();
+            const taskStepsContext = taskStepsState ? formatTaskStepsContext(taskStepsState) : undefined;
+            if (taskStepsState && taskStepsContext) {
+                messages.push(createCustomMessage("task_steps_context", taskStepsContext, false, { revision: taskStepsState.revision }, new Date().toISOString()));
+            }
             // Emit before_agent_start extension event
             const result = await this._extensionRunner.emitBeforeAgentStart(expandedText, currentImages, this._baseSystemPrompt, this._systemPromptBuilder.getBaseSystemPromptOptions());
             // Add all custom messages from extensions
@@ -3745,6 +3777,14 @@ export class AgentSession {
     getGoalStateSnapshot() {
         return getLatestGoalStateSnapshot(this.sessionManager.getEntries());
     }
+    /** Save native task-step state to the active session log. */
+    saveTaskStepsStateSnapshot(state) {
+        return appendTaskStepsStateSnapshot(this.sessionManager, state);
+    }
+    /** Retrieve the latest valid native task-step state from the active session log. */
+    getTaskStepsStateSnapshot() {
+        return getLatestTaskStepsStateSnapshot(this.sessionManager.getEntries());
+    }
     /**
      * Save a snapshot of the evidence bundle to the session log.
      *