@sema-agent/core 1.294.0 → 1.296.0

package/dist/core/runner/runtask.js

@@ -1103,6 +1103,9 @@ export class Runner {
                 turnUsage.cacheWriteTokens += u.cacheWrite || 0;
                 turnUsage.costMicroUsd += turnCostMicroUsd;
                 const callEndAt = Date.now();
+                const callIssuedAt = prepared.callIssuedAtRef?.current;
+                if (prepared.callIssuedAtRef)
+                    prepared.callIssuedAtRef.current = undefined;
                 const capLast = prepared.callCapRef?.state.last;
                 if (prepared.callCapRef) {
                     recordCallSample(prepared.callCapRef.state, u.output || 0, callStartAt !== undefined ? callEndAt - callStartAt : 0);
@@ -1123,6 +1126,7 @@ export class Runner {
                     cacheWrite: u.cacheWrite || 0,
                     latencyMs: callStartAt !== undefined ? callEndAt - callStartAt : 0,
                     firstTokenMs: callStartAt !== undefined && firstTokenAt !== undefined ? firstTokenAt - callStartAt : undefined,
+                    ...(callIssuedAt !== undefined ? { callStartedAt: callIssuedAt } : {}),
                     costMicroUsd: turnCostMicroUsd,
                     ...(capLast !== undefined ? { callCap: capLast.cap, capThinkingSkipped: capLast.thinkingSkipped } : {}),
                     ...(m.role === "assistant" && typeof m.stopReason === "string"
@@ -1224,7 +1228,8 @@ export class Runner {
         const batchArgs = postToolBatchHook ? new Map() : undefined;
         let toolBatch = [];
         const onToolStart = (event) => {
-            toolStartAt.set(event.toolCallId, Date.now());
+            const startAt = Date.now();
+            toolStartAt.set(event.toolCallId, startAt);
             startedToolCallIds.add(event.toolCallId);
             batchArgs?.set(event.toolCallId, event.args);
             stats.toolCalls += 1;
@@ -1235,7 +1240,7 @@ export class Runner {
                     groundingSignalPostR9 = true;
             }
             const activityArg = primaryActivityArg(event.args);
-            internals?.onActivity?.({ phase: "start", toolCallId: event.toolCallId, toolName: event.toolName, ...(activityArg !== undefined ? { arg: activityArg } : {}) });
+            internals?.onActivity?.({ phase: "start", toolCallId: event.toolCallId, toolName: event.toolName, at: startAt, ...(activityArg !== undefined ? { arg: activityArg } : {}) });
             pushContent({
                 type: "tool_start",
                 toolCallId: event.toolCallId,
@@ -1285,7 +1290,7 @@ export class Runner {
                 ok: !event.isError,
                 ts: toolNow,
             }));
-            internals?.onActivity?.({ phase: "end", toolCallId: event.toolCallId, toolName: event.toolName, isError: event.isError });
+            internals?.onActivity?.({ phase: "end", toolCallId: event.toolCallId, toolName: event.toolName, isError: event.isError, at: toolNow });
             if (prepared.lspDiagnostics && !event.isError) {
                 const name = canonicalToolName(event.toolName);
                 if (name === "Edit" || name === "Write" || name === "NotebookEdit") {
@@ -2316,7 +2321,8 @@ export class Runner {
             const toolClamps = prepared.callCapRef?.toolClamps ?? 0;
             const toolCuts = prepared.callCapRef?.toolCuts ?? 0;
             const pendingCutTools = prepared.callCapRef?.pendingCutTools ?? 0;
-            if (finalizeInjected || nudgesSent > 0 || capShrinks > 0 || callCutoffs > 0 || toolClamps > 0 || toolCuts > 0 || pendingCutTools > 0 || finalVerifyInjections > 0 || attachmentsInjected > 0 || repetitionCuts > 0 || repetitionSpared > 0) {
+            const toolCutKills = prepared.cutKills?.kills ?? 0;
+            if (finalizeInjected || nudgesSent > 0 || capShrinks > 0 || callCutoffs > 0 || toolClamps > 0 || toolCuts > 0 || toolCutKills > 0 || pendingCutTools > 0 || finalVerifyInjections > 0 || attachmentsInjected > 0 || repetitionCuts > 0 || repetitionSpared > 0) {
                 stats.mechanisms = {
                     ...(finalizeInjected ? { finalizeInjected: true } : {}),
                     ...(nudgesSent > 0 ? { nudgesSent } : {}),
@@ -2324,6 +2330,7 @@ export class Runner {
                     ...(callCutoffs > 0 ? { callCutoffs } : {}),
                     ...(toolClamps > 0 ? { toolClamps } : {}),
                     ...(toolCuts > 0 ? { toolCuts } : {}),
+                    ...(toolCutKills > 0 ? { toolCutKills } : {}),
                     ...(pendingCutTools > 0 ? { pendingCutTools } : {}),
                     ...(finalVerifyInjections > 0 ? { finalVerifyInjected: true } : {}),
                     ...(finalVerifyInjections > 0 ? { finalVerifyInjections } : {}),
@@ -2424,6 +2431,9 @@ export class Runner {
                 ...(stats.mechanisms !== undefined
                     ? { mechanisms: (({ repetitionEvents: _events, ...scalars }) => scalars)(stats.mechanisms) }
                     : {}),
+                ...(timeout.latenessMs !== undefined && timeout.latenessMs > TIMER_LATENESS_REPORT_MS
+                    ? { timerLatenessMs: timeout.latenessMs }
+                    : {}),
                 ts: Date.now(),
             }));
             setResult(result);
@@ -2446,7 +2456,7 @@ export class Runner {
                 prepared.suspendRef.token === undefined &&
                 prepared.reviewRef.token === undefined;
             try {
-                await defaultTaskRegistry.settleKilledForOwner({ owner: spec.taskId ?? prepared.sessionId, scope: spec.principal ?? "default" }, {
+                await defaultTaskRegistry.settleKilledForOwner({ owner: spec.taskId ?? prepared.sessionId, scope: spec.principal ?? "default", sessionId: prepared.sessionId }, {
                     source: abortedLive &&
                         !timeout.fired &&
                         !turnsExceeded &&
@@ -2457,6 +2467,7 @@ export class Runner {
                         : "parent",
                     skipSessionScoped: true,
                     envDying: ownedEnvDying,
+                    retainProcesses: spec.retainBackgroundProcesses === true,
                 });
             }
             catch (e) {
@@ -2467,13 +2478,15 @@ export class Runner {
                 }
             }
             const bgEnv = prepared.ownedEnv ?? this.deps.executionEnv;
-            if (bgEnv && hasBackgroundShell(bgEnv)) {
+            const retainBgBySpec = spec.retainBackgroundProcesses === true && !ownedEnvDying;
+            if (bgEnv && hasBackgroundShell(bgEnv) && !retainBgBySpec) {
                 try {
                     const keepAlive = ownedEnvDying && bgEnv === prepared.ownedEnv
                         ? []
                         : [
                             ...defaultTaskRegistry.sessionResidentShellIds(prepared.sessionId, bgEnv),
                             ...defaultTaskRegistry.timeoutResidentShellIds(bgEnv),
+                            ...defaultTaskRegistry.retainedShellIds(bgEnv),
                         ];
                     await bgEnv.disposeBackgroundShells(keepAlive.length > 0 ? { except: keepAlive } : undefined);
                 }
@@ -3098,12 +3111,15 @@ export function runTask(spec, deps) {
     return new Runner(deps).runTask(spec);
 }
 const WALLTIME_SUSPEND_GRACE_SEC = 120;
+const TIMER_LATENESS_REPORT_MS = 5_000;
 function startTimeout(harness, abortController, timeoutSec, softSuspendable = false) {
     const state = { fired: false, clear: () => { } };
     if (timeoutSec && timeoutSec > 0) {
         const hardSec = softSuspendable ? timeoutSec + WALLTIME_SUSPEND_GRACE_SEC : timeoutSec;
+        const scheduledAtMs = Date.now() + hardSec * 1000;
         const timer = setTimeout(() => {
             state.fired = true;
+            state.latenessMs = Math.max(0, Date.now() - scheduledAtMs);
             abortController.abort();
             void harness.abort();
         }, hardSec * 1000);