@sema-agent/core 1.294.0 → 1.295.0

package/dist/core/runner/runtask.js

@@ -1224,7 +1224,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 +1236,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 +1286,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") {
@@ -2446,7 +2447,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 +2458,7 @@ export class Runner {
                         : "parent",
                     skipSessionScoped: true,
                     envDying: ownedEnvDying,
+                    retainProcesses: spec.retainBackgroundProcesses === true,
                 });
             }
             catch (e) {
@@ -2467,13 +2469,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);
                 }