@sema-agent/core 1.295.0 → 1.296.0

package/dist/core/runner/prepare-task.js

@@ -19,7 +19,8 @@ import { Session, isSessionConflict, hasSessionFork } from "../session.js";
 import { createAgentTranscriptTool, createSendMessageTool, createSubagentWorktreeHelper, forkGovernanceDenial, AGENT_TRANSCRIPT_TOOL_NAME, SEND_MESSAGE_TOOL_NAME, SubagentRetainLedger } from "../../agents/subagent.js";
 import { combinePolicies, createTranscriptIntegrityPolicy, createUnverifiableDeletePolicy, resolveAsk } from "../tool-policy.js";
 import { ActiveSkillScope, createActiveSkillScopePolicy } from "./active-skill-scope.js";
-import { computeCallCap, createCallCapRef, softExecDeadlineMs, toolCutDeadlineMs } from "./call-cap.js";
+import { computeCallCap, createCallCapRef, softExecDeadlineMs, toolCutDeadlineMs, WALLTIME_STALL_CONNECT_MS, WALLTIME_STALL_FIRST_TOKEN_MS, WALLTIME_STALL_IDLE_MS } from "./call-cap.js";
+import { createCutKillRegistry } from "./cut-kill.js";
 import { CHANGED_FILES_MTIME_EPS_MS } from "./turn-attachments.js";
 import { emitTrace } from "../trace.js";
 import { createSessionRulePolicy } from "./session-rule-policy.js";
@@ -630,10 +631,13 @@ export async function prepareTask(spec, deps, sessions, resume, _memorySelector,
         if (!(spec.tools ?? []).some((t) => t.name === REPORT_FINDINGS_TOOL_NAME || t.aliases?.includes(REPORT_FINDINGS_TOOL_NAME))) {
             tools.push(makeReportFindingsTool());
         }
-        if (spec.enablePlanMode === true) {
-            tools.push(defineTool(createPresentPlanTool(requestReview)));
-            if ((spec.checkpointStore ?? deps.checkpointStore) !== undefined) {
-                tools.push(defineTool(createEnterPlanModeTool(enterPlanMode)));
+        if (spec.enablePlanMode === true && spec.interactiveTools !== false) {
+            const planReviewFace = (spec.checkpointStore ?? deps.checkpointStore) !== undefined;
+            if (spec.interactiveTools === true || planReviewFace) {
+                tools.push(defineTool(createPresentPlanTool(requestReview)));
+                if (planReviewFace) {
+                    tools.push(defineTool(createEnterPlanModeTool(enterPlanMode)));
+                }
             }
         }
         let runtimeCaps;
@@ -818,12 +822,15 @@ export async function prepareTask(spec, deps, sessions, resume, _memorySelector,
                     ? spec.limits.maxOutputTokens
                     : model.maxTokens;
         }
+        const cutKills = callCapRef ? createCutKillRegistry() : undefined;
+        const callIssuedAtRef = {};
         const execClamp = callCapRef
             ? {
                 deadlineMs: () => softExecDeadlineMs(callCapRef),
                 onClamp: () => {
                     callCapRef.toolClamps = (callCapRef.toolClamps ?? 0) + 1;
                 },
+                ...(cutKills ? { registerKillHandle: cutKills.register } : {}),
             }
             : undefined;
         const memoryWriteGateRef = {};
@@ -961,7 +968,9 @@ export async function prepareTask(spec, deps, sessions, resume, _memorySelector,
         if (offloadStore)
             tools.push(makeReadToolResultTool(offloadStore));
         const onQuestion = spec.onQuestion ?? deps.onQuestion;
-        if (onQuestion)
+        const durableQuestionFace = (spec.checkpointStore ?? deps.checkpointStore) !== undefined &&
+            (spec.durableApproval !== undefined || runtimeCaps?.forceDurableGate === true);
+        if (spec.interactiveTools === true || (spec.interactiveTools !== false && (onQuestion !== undefined || durableQuestionFace)))
             tools.push(createAskUserQuestionTool(onQuestion, { principal: spec.principal, sourceTaskId: sessionId }));
         if (spec.handsReadOnly !== true) {
             const schedScope = sessionId ?? spec.principal ?? spec.taskId ?? "default";
@@ -1542,7 +1551,15 @@ export async function prepareTask(spec, deps, sessions, resume, _memorySelector,
                 }
                 : {}),
             ...(callCapRef !== undefined
-                ? { callDeadlineMsPerCall: () => softExecDeadlineMs(callCapRef) }
+                ? {
+                    callDeadlineMsPerCall: () => {
+                        callIssuedAtRef.current = Date.now();
+                        return softExecDeadlineMs(callCapRef);
+                    },
+                    stallTimeoutsPerCall: () => softExecDeadlineMs(callCapRef) !== undefined
+                        ? { connectMs: WALLTIME_STALL_CONNECT_MS, firstTokenMs: WALLTIME_STALL_FIRST_TOKEN_MS, idleMs: WALLTIME_STALL_IDLE_MS }
+                        : undefined,
+                }
                 : {}),
             ...(callCapRef !== undefined
                 ? {
@@ -1550,6 +1567,9 @@ export async function prepareTask(spec, deps, sessions, resume, _memorySelector,
                         deadlineMs: () => toolCutDeadlineMs(callCapRef, Date.now()),
                         onCut: (info) => {
                             callCapRef.toolCuts = (callCapRef.toolCuts ?? 0) + 1;
+                            if (callCapRef.finalizeMode === true && cutKills !== undefined) {
+                                cutKills.killFor(info.toolCallId);
+                            }
                             callCapRef.pendingCutTools = (callCapRef.pendingCutTools ?? 0) + 1;
                             void info.settled.then(() => {
                                 callCapRef.pendingCutTools = (callCapRef.pendingCutTools ?? 0) - 1;
@@ -2486,7 +2506,7 @@ export async function prepareTask(spec, deps, sessions, resume, _memorySelector,
             .list({ owner: hostTaskId, scope: taskScope, sessionId })
             .filter((t) => t.status === "pending" || t.status === "running")
             .map((t) => ({ id: t.task_id, ...(t.description !== undefined ? { description: t.description } : {}), status: t.status }));
-        return { harness, session, sessionId, taskRootPath, model, thinking, compModel, mcp, blockedRef, outputRef, abortController, conflictRef, blockedToolCalls, nestedStats, cwdRef: handsCwdRef, denyNarrowingPolicy, releaseSignal, cacheBreakDetector, cacheFingerprint, promptManifest, activeTools, ownedEnv, suspendRef, reviewRef, reviewRequestRef, suspendLoopRef, suspendForResource, ...(callCapRef ? { callCapRef } : {}), suspendForReview, resourceLedger: priorLedger, liveSpendRef, humanReviewRef, now, tools, toolEffects, promptOverheadTokens, readTaskFile, recentlyReadFiles, compactionReuseRef, trimPressureRef, ...(memoryEngineSession ? { memoryEngineSession } : {}), ...(subagentRetain ? { subagentRetain } : {}), ...(lspDiagnostics && nudgeLspOnEdit ? { lspDiagnostics: { registry: lspDiagnostics, nudge: nudgeLspOnEdit } } : {}), planModeRef, ...(detectExternalChanges ? { detectExternalChanges } : {}), ...(toolsDeltaRef ? { toolsDeltaRef } : {}), ...(agentListing ? { agentListing } : {}), listBackgroundTasks };
+        return { harness, session, sessionId, taskRootPath, model, thinking, compModel, mcp, blockedRef, outputRef, abortController, conflictRef, blockedToolCalls, nestedStats, cwdRef: handsCwdRef, denyNarrowingPolicy, releaseSignal, cacheBreakDetector, cacheFingerprint, promptManifest, activeTools, ownedEnv, suspendRef, reviewRef, reviewRequestRef, suspendLoopRef, suspendForResource, ...(callCapRef ? { callCapRef } : {}), ...(cutKills ? { cutKills } : {}), ...(callCapRef ? { callIssuedAtRef } : {}), suspendForReview, resourceLedger: priorLedger, liveSpendRef, humanReviewRef, now, tools, toolEffects, promptOverheadTokens, readTaskFile, recentlyReadFiles, compactionReuseRef, trimPressureRef, ...(memoryEngineSession ? { memoryEngineSession } : {}), ...(subagentRetain ? { subagentRetain } : {}), ...(lspDiagnostics && nudgeLspOnEdit ? { lspDiagnostics: { registry: lspDiagnostics, nudge: nudgeLspOnEdit } } : {}), planModeRef, ...(detectExternalChanges ? { detectExternalChanges } : {}), ...(toolsDeltaRef ? { toolsDeltaRef } : {}), ...(agentListing ? { agentListing } : {}), listBackgroundTasks };
     }
     catch (prepareErr) {
         if (mcp) {