@assemblyline-agents/runtime 6.1.4 → 7.0.0

package/dist/task-budget-config.js

@@ -0,0 +1,55 @@
+const FIELDS = ["maxModelRequests", "maxTokens", "maxCostUsd", "maxDurationMs"];
+export function taskBudgetLimits(defaults, requested) {
+    const result = {};
+    for (const source of [defaults, requested]) {
+        if (!source)
+            continue;
+        for (const key of Object.keys(source))
+            if (!FIELDS.includes(key))
+                throw new Error(`Unknown task budget limit: ${key}`);
+        for (const key of FIELDS) {
+            const value = source[key];
+            if (value === undefined)
+                continue;
+            if (!Number.isFinite(value) ||
+                value <= 0 ||
+                (key !== "maxCostUsd" && !Number.isSafeInteger(value)) ||
+                (key === "maxCostUsd" && (!Number.isSafeInteger(Math.ceil(value * 1e6)) || value < 1e-6))) {
+                throw new Error(`Task budget ${key} must be a positive ${key === "maxCostUsd" ? "USD amount of at least 0.000001" : "safe integer"}.`);
+            }
+            result[key] = Math.min(result[key] ?? Infinity, value);
+        }
+    }
+    return Object.keys(result).length ? result : undefined;
+}
+export function createTaskBudget(limits, now = new Date()) {
+    if (!limits)
+        return;
+    const startedAt = now.toISOString();
+    return {
+        limits,
+        startedAt,
+        ...(limits.maxDurationMs ? { deadlineAt: new Date(now.getTime() + limits.maxDurationMs).toISOString() } : {}),
+        modelRequests: 0,
+        pendingRequests: 0,
+        tokens: 0,
+        costMicros: 0,
+        requests: {}
+    };
+}
+export function taskBudgetFromEnv(env) {
+    const result = {};
+    const names = {
+        maxModelRequests: "MAX_MODEL_REQUESTS",
+        maxTokens: "MAX_TOKENS",
+        maxCostUsd: "MAX_COST_USD",
+        maxDurationMs: "MAX_DURATION_MS"
+    };
+    for (const key of FIELDS) {
+        const raw = env[`ASSEMBLY_LINE_TASK_BUDGET_${names[key]}`];
+        if (raw !== undefined)
+            result[key] = Number(raw);
+    }
+    return taskBudgetLimits(result);
+}
+//# sourceMappingURL=task-budget-config.js.map

package/dist/task-budget-config.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"task-budget-config.js","sourceRoot":"","sources":["../src/task-budget-config.ts"],"names":[],"mappings":"AAEA,MAAM,MAAM,GAAG,CAAC,kBAAkB,EAAE,WAAW,EAAE,YAAY,EAAE,eAAe,CAAU,CAAC;AAEzF,MAAM,UAAU,gBAAgB,CAC9B,QAA2B,EAC3B,SAA4B;IAE5B,MAAM,MAAM,GAAqB,EAAE,CAAC;IACpC,KAAK,MAAM,MAAM,IAAI,CAAC,QAAQ,EAAE,SAAS,CAAC,EAAE,CAAC;QAC3C,IAAI,CAAC,MAAM;YAAE,SAAS;QACtB,KAAK,MAAM,GAAG,IAAI,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC;YACnC,IAAI,CAAE,MAA4B,CAAC,QAAQ,CAAC,GAAG,CAAC;gBAAE,MAAM,IAAI,KAAK,CAAC,8BAA8B,GAAG,EAAE,CAAC,CAAC;QACzG,KAAK,MAAM,GAAG,IAAI,MAAM,EAAE,CAAC;YACzB,MAAM,KAAK,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC;YAC1B,IAAI,KAAK,KAAK,SAAS;gBAAE,SAAS;YAClC,IACE,CAAC,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC;gBACvB,KAAK,IAAI,CAAC;gBACV,CAAC,GAAG,KAAK,YAAY,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC;gBACtD,CAAC,GAAG,KAAK,YAAY,IAAI,CAAC,CAAC,MAAM,CAAC,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,GAAG,GAAG,CAAC,CAAC,IAAI,KAAK,GAAG,IAAI,CAAC,CAAC,EACzF,CAAC;gBACD,MAAM,IAAI,KAAK,CACb,eAAe,GAAG,uBAChB,GAAG,KAAK,YAAY,CAAC,CAAC,CAAC,iCAAiC,CAAC,CAAC,CAAC,cAC7D,GAAG,CACJ,CAAC;YACJ,CAAC;YACD,MAAM,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,IAAI,QAAQ,EAAE,KAAK,CAAC,CAAC;QACzD,CAAC;IACH,CAAC;IACD,OAAO,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS,CAAC;AACzD,CAAC;AAED,MAAM,UAAU,gBAAgB,CAAC,MAAoC,EAAE,GAAG,GAAG,IAAI,IAAI,EAAE;IACrF,IAAI,CAAC,MAAM;QAAE,OAAO;IACpB,MAAM,SAAS,GAAG,GAAG,CAAC,WAAW,EAAE,CAAC;IACpC,OAAO;QACL,MAAM;QACN,SAAS;QACT,GAAG,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,IAAI,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,GAAG,MAAM,CAAC,aAAa,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QAC7G,aAAa,EAAE,CAAC;QAChB,eAAe,EAAE,CAAC;QAClB,MAAM,EAAE,CAAC;QACT,UAAU,EAAE,CAAC;QACb,QAAQ,EAAE,EAAE;KACb,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,iBAAiB,CAAC,GAAuC;IACvE,MAAM,MAAM,GAAqB,EAAE,CAAC;IACpC,MAAM,KAAK,GAAG;QACZ,gBAAgB,EAAE,oBAAoB;QACtC,SAAS,EAAE,YAAY;QACvB,UAAU,EAAE,cAAc;QAC1B,aAAa,EAAE,iBAAiB;KACxB,CAAC;IACX,KAAK,MAAM,GAAG,IAAI,MAAM,EAAE,CAAC;QACzB,MAAM,GAAG,GAAG,GAAG,CAAC,6BAA6B,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QAC3D,IAAI,GAAG,KAAK,SAAS;YAAE,MAAM,CAAC,GAAG,CAAC,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC;IACnD,CAAC;IACD,OAAO,gBAAgB,CAAC,MAAM,CAAC,CAAC;AAClC,CAAC"}

package/dist/task-budget-harness.d.ts

@@ -0,0 +1,5 @@
+import type { AgentHarness, AgentHarnessContext } from "@assemblyline-agents/core";
+import type { StateAdapter } from "./state.js";
+/** Apply the task contract to a runtime-owned harness outside the interactive loop. */
+export declare function runTaskBudgetHarness(state: StateAdapter, harness: AgentHarness, input: Parameters<AgentHarness["runTurn"]>[0], context: AgentHarnessContext): Promise<import("@assemblyline-agents/core").AgentTurnResult>;
+//# sourceMappingURL=task-budget-harness.d.ts.map

package/dist/task-budget-harness.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"task-budget-harness.d.ts","sourceRoot":"","sources":["../src/task-budget-harness.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,mBAAmB,EAAE,MAAM,2BAA2B,CAAC;AACnF,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,YAAY,CAAC;AAK/C,uFAAuF;AACvF,wBAAsB,oBAAoB,CACxC,KAAK,EAAE,YAAY,EACnB,OAAO,EAAE,YAAY,EACrB,KAAK,EAAE,UAAU,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,EAC7C,OAAO,EAAE,mBAAmB,gEAwC7B"}

package/dist/task-budget-harness.js

@@ -0,0 +1,44 @@
+import { TaskBudgetAccounting } from "./task-budget-accounting.js";
+import { checkTaskBudget } from "./task-budget.js";
+import { armTaskDeadline } from "./task-deadline.js";
+/** Apply the task contract to a runtime-owned harness outside the interactive loop. */
+export async function runTaskBudgetHarness(state, harness, input, context) {
+    await checkTaskBudget(state, context.runId);
+    const accounting = new TaskBudgetAccounting(state, context.runId);
+    const controller = new AbortController();
+    const stopDeadline = armTaskDeadline(state, context.runId, () => controller.abort(), () => { });
+    const timer = setInterval(() => {
+        void checkTaskBudget(state, context.runId).catch(() => controller.abort());
+    }, 1000);
+    timer.unref?.();
+    try {
+        return await harness.runTurn(input, {
+            ...context,
+            signal: context.signal ? AbortSignal.any([context.signal, controller.signal]) : controller.signal,
+            shouldStop: () => controller.signal.aborted || (context.shouldStop?.() ?? false),
+            tools: async (options) => {
+                await checkTaskBudget(state, context.runId);
+                return context.tools(options);
+            },
+            executeTool: async (call) => {
+                await checkTaskBudget(state, context.runId);
+                return context.executeTool(call);
+            },
+            emit: async (event) => {
+                if (event.type === "request_started")
+                    await accounting.start(event.iteration);
+                if (event.type === "model_retry")
+                    await accounting.retry(event.iteration);
+                if (event.type === "response_completed")
+                    await accounting.complete(event.iteration, event.usage);
+                await context.emit(event);
+            }
+        });
+    }
+    finally {
+        clearInterval(timer);
+        stopDeadline();
+        await accounting.finalize();
+    }
+}
+//# sourceMappingURL=task-budget-harness.js.map

package/dist/task-budget-harness.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"task-budget-harness.js","sourceRoot":"","sources":["../src/task-budget-harness.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,oBAAoB,EAAE,MAAM,6BAA6B,CAAC;AACnE,OAAO,EAAE,eAAe,EAAE,MAAM,kBAAkB,CAAC;AACnD,OAAO,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AAErD,uFAAuF;AACvF,MAAM,CAAC,KAAK,UAAU,oBAAoB,CACxC,KAAmB,EACnB,OAAqB,EACrB,KAA6C,EAC7C,OAA4B;IAE5B,MAAM,eAAe,CAAC,KAAK,EAAE,OAAO,CAAC,KAAK,CAAC,CAAC;IAC5C,MAAM,UAAU,GAAG,IAAI,oBAAoB,CAAC,KAAK,EAAE,OAAO,CAAC,KAAK,CAAC,CAAC;IAClE,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;IACzC,MAAM,YAAY,GAAG,eAAe,CAClC,KAAK,EACL,OAAO,CAAC,KAAK,EACb,GAAG,EAAE,CAAC,UAAU,CAAC,KAAK,EAAE,EACxB,GAAG,EAAE,GAAE,CAAC,CACT,CAAC;IACF,MAAM,KAAK,GAAG,WAAW,CAAC,GAAG,EAAE;QAC7B,KAAK,eAAe,CAAC,KAAK,EAAE,OAAO,CAAC,KAAK,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,UAAU,CAAC,KAAK,EAAE,CAAC,CAAC;IAC7E,CAAC,EAAE,IAAI,CAAC,CAAC;IACT,KAAK,CAAC,KAAK,EAAE,EAAE,CAAC;IAChB,IAAI,CAAC;QACH,OAAO,MAAM,OAAO,CAAC,OAAO,CAAC,KAAK,EAAE;YAClC,GAAG,OAAO;YACV,MAAM,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,OAAO,CAAC,MAAM,EAAE,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,MAAM;YACjG,UAAU,EAAE,GAAG,EAAE,CAAC,UAAU,CAAC,MAAM,CAAC,OAAO,IAAI,CAAC,OAAO,CAAC,UAAU,EAAE,EAAE,IAAI,KAAK,CAAC;YAChF,KAAK,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE;gBACvB,MAAM,eAAe,CAAC,KAAK,EAAE,OAAO,CAAC,KAAK,CAAC,CAAC;gBAC5C,OAAO,OAAO,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;YAChC,CAAC;YACD,WAAW,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE;gBAC1B,MAAM,eAAe,CAAC,KAAK,EAAE,OAAO,CAAC,KAAK,CAAC,CAAC;gBAC5C,OAAO,OAAO,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;YACnC,CAAC;YACD,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE;gBACpB,IAAI,KAAK,CAAC,IAAI,KAAK,iBAAiB;oBAAE,MAAM,UAAU,CAAC,KAAK,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;gBAC9E,IAAI,KAAK,CAAC,IAAI,KAAK,aAAa;oBAAE,MAAM,UAAU,CAAC,KAAK,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;gBAC1E,IAAI,KAAK,CAAC,IAAI,KAAK,oBAAoB;oBAAE,MAAM,UAAU,CAAC,QAAQ,CAAC,KAAK,CAAC,SAAS,EAAE,KAAK,CAAC,KAAK,CAAC,CAAC;gBACjG,MAAM,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAC5B,CAAC;SACF,CAAC,CAAC;IACL,CAAC;YAAS,CAAC;QACT,aAAa,CAAC,KAAK,CAAC,CAAC;QACrB,YAAY,EAAE,CAAC;QACf,MAAM,UAAU,CAAC,QAAQ,EAAE,CAAC;IAC9B,CAAC;AACH,CAAC"}

package/dist/task-budget.d.ts

@@ -0,0 +1,14 @@
+import type { HarnessUsage } from "@assemblyline-agents/core";
+import type { StateAdapter, TaskBudget } from "./state.js";
+/** Short durable admission on the common root serializes concurrent siblings. */
+export declare function admitTaskBudgetRequest(state: StateAdapter, runId: string, requestId: string): Promise<string | undefined>;
+/** Budget enforcement is independent of the best-effort observability usage ledger. */
+export declare function completeTaskBudgetRequest(state: StateAdapter, rootId: string, requestId: string, usage?: HarnessUsage): Promise<void>;
+export declare function checkTaskBudget(state: StateAdapter, runId: string): Promise<void>;
+/** Dead workers cannot make unobserved token/cash usage look like zero on restart. */
+export declare function recoverTaskBudgets(state: StateAdapter): Promise<void>;
+export declare function getTaskBudget(state: StateAdapter, runId: string): Promise<{
+    rootRunId: string;
+    budget?: TaskBudget;
+}>;
+//# sourceMappingURL=task-budget.d.ts.map

package/dist/task-budget.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"task-budget.d.ts","sourceRoot":"","sources":["../src/task-budget.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,2BAA2B,CAAC;AAI9D,OAAO,KAAK,EAAmC,YAAY,EAAE,UAAU,EAAE,MAAM,YAAY,CAAC;AAkE5F,iFAAiF;AACjF,wBAAsB,sBAAsB,CAC1C,KAAK,EAAE,YAAY,EACnB,KAAK,EAAE,MAAM,EACb,SAAS,EAAE,MAAM,GAChB,OAAO,CAAC,MAAM,GAAG,SAAS,CAAC,CAuC7B;AAED,uFAAuF;AACvF,wBAAsB,yBAAyB,CAC7C,KAAK,EAAE,YAAY,EACnB,MAAM,EAAE,MAAM,EACd,SAAS,EAAE,MAAM,EACjB,KAAK,CAAC,EAAE,YAAY,GACnB,OAAO,CAAC,IAAI,CAAC,CAkEf;AAED,wBAAsB,eAAe,CAAC,KAAK,EAAE,YAAY,EAAE,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAYvF;AAED,sFAAsF;AACtF,wBAAsB,kBAAkB,CAAC,KAAK,EAAE,YAAY,GAAG,OAAO,CAAC,IAAI,CAAC,CAoB3E;AAED,wBAAsB,aAAa,CACjC,KAAK,EAAE,YAAY,EACnB,KAAK,EAAE,MAAM,GACZ,OAAO,CAAC;IAAE,SAAS,EAAE,MAAM,CAAC;IAAC,MAAM,CAAC,EAAE,UAAU,CAAA;CAAE,CAAC,CAMrD"}

package/dist/task-budget.js

@@ -0,0 +1,212 @@
+import { cancellationUpdate, propagateRunCancellation } from "./run-tree-cancellation.js";
+import { assertRunTreeOpen, runAncestry, RunTreeStoppedError } from "./run-tree.js";
+import { outsideRunExecution } from "./run-execution-lease.js";
+function stopUpdate(root, budget, reason) {
+    const at = new Date().toISOString();
+    const stopped = { ...budget, stopped: budget.stopped ?? { reason, at } };
+    const update = cancellationUpdate(root, root.lifecycle?.cancellation ?? {
+        id: `budget:${root.id}`,
+        rootRunId: root.id,
+        requestedAt: at,
+        reason,
+        pending: true
+    });
+    return {
+        ...update,
+        patch: {
+            ...update.patch,
+            lifecycle: { ...update.patch.lifecycle, taskBudget: stopped }
+        },
+        events: [
+            ...(update.events ?? []),
+            {
+                id: `task-budget-stopped:${root.id}`,
+                type: "task.budget_exhausted",
+                data: {
+                    reason,
+                    modelRequests: budget.modelRequests,
+                    tokens: budget.tokens,
+                    costMicros: budget.costMicros
+                }
+            }
+        ]
+    };
+}
+function reached(budget, admission = false) {
+    if (budget.stopped)
+        return budget.stopped.reason;
+    if (budget.deadlineAt && Date.parse(budget.deadlineAt) <= Date.now())
+        return "budget.deadline";
+    if (admission &&
+        budget.limits.maxModelRequests !== undefined &&
+        budget.modelRequests >= budget.limits.maxModelRequests)
+        return "budget.model_requests";
+    if (budget.limits.maxTokens !== undefined && budget.tokens >= budget.limits.maxTokens)
+        return "budget.tokens";
+    if (budget.limits.maxCostUsd !== undefined &&
+        budget.costMicros >= Math.floor(budget.limits.maxCostUsd * 1e6))
+        return "budget.cost";
+    if ((budget.limits.maxTokens !== undefined || budget.limits.maxCostUsd !== undefined) &&
+        Object.values(budget.requests).some((request) => request.status === "unknown"))
+        return "budget.usage_unavailable";
+}
+async function enforceStop(state, root) {
+    if (!root.lifecycle?.taskBudget?.stopped)
+        return;
+    await propagateRunCancellation(state, root.id);
+    throw new RunTreeStoppedError(root.id, root.lifecycle.taskBudget.stopped.reason);
+}
+/** Short durable admission on the common root serializes concurrent siblings. */
+export async function admitTaskBudgetRequest(state, runId, requestId) {
+    const ancestry = await assertRunTreeOpen(state, runId);
+    const root = ancestry.at(-1);
+    if (!root.lifecycle?.taskBudget)
+        return;
+    const result = await outsideRunExecution(() => state.transactRun(root.id, ({ run }) => {
+        const budget = run.lifecycle.taskBudget;
+        if (run.lifecycle?.cancellation)
+            return;
+        if (budget.requests[requestId])
+            return;
+        const reason = reached(budget, true);
+        if (reason)
+            return stopUpdate(run, budget, reason);
+        const next = {
+            ...budget,
+            modelRequests: budget.modelRequests + 1,
+            pendingRequests: budget.pendingRequests + 1,
+            requests: {
+                ...budget.requests,
+                [requestId]: {
+                    runId,
+                    status: "pending",
+                    startedAt: new Date().toISOString()
+                }
+            }
+        };
+        return {
+            patch: { lifecycle: { ...run.lifecycle, taskBudget: next } },
+            events: [
+                {
+                    type: "task.budget_request_admitted",
+                    data: { requestId, runId, modelRequests: next.modelRequests }
+                }
+            ]
+        };
+    }));
+    await enforceStop(state, result.run);
+    if (result.run.lifecycle?.cancellation)
+        throw new RunTreeStoppedError(root.id, result.run.lifecycle.cancellation.reason);
+    return root.id;
+}
+/** Budget enforcement is independent of the best-effort observability usage ledger. */
+export async function completeTaskBudgetRequest(state, rootId, requestId, usage) {
+    const result = await outsideRunExecution(() => state.transactRun(rootId, ({ run }) => {
+        const budget = run.lifecycle?.taskBudget;
+        const previous = budget?.requests[requestId];
+        if (!budget || !previous || previous.status === "settled")
+            return;
+        const valid = (value) => typeof value === "number" && Number.isFinite(value) && value >= 0;
+        const measuredTokens = valid(usage?.inputTokens) &&
+            valid(usage?.outputTokens) &&
+            Number.isSafeInteger(usage.inputTokens + usage.outputTokens) &&
+            Number.isSafeInteger(usage.inputTokens) &&
+            usage?.tokenSource !== "unavailable"
+            ? usage.inputTokens + usage.outputTokens
+            : undefined;
+        const measuredCostMicros = valid(usage?.costUsd) &&
+            Number.isSafeInteger(Math.ceil(usage.costUsd * 1e6)) &&
+            (usage.costSource === "provider_reported" || usage.costSource === "provider_reconciled") &&
+            (usage.currency === undefined || usage.currency === "USD")
+            ? Math.ceil(usage.costUsd * 1e6)
+            : undefined;
+        // A late provider receipt may fill in a lost observation, but never reopen a stopped task.
+        const tokens = previous.tokens ?? measuredTokens;
+        const costMicros = previous.costMicros ?? measuredCostMicros;
+        if (previous.status === "unknown" && tokens === previous.tokens && costMicros === previous.costMicros)
+            return;
+        const unknown = (budget.limits.maxTokens !== undefined && tokens === undefined) ||
+            (budget.limits.maxCostUsd !== undefined && costMicros === undefined);
+        const next = {
+            ...budget,
+            pendingRequests: budget.pendingRequests - (previous.status === "pending" ? 1 : 0),
+            tokens: budget.tokens + (tokens ?? 0) - (previous.tokens ?? 0),
+            costMicros: budget.costMicros + (costMicros ?? 0) - (previous.costMicros ?? 0),
+            requests: {
+                ...budget.requests,
+                [requestId]: {
+                    ...previous,
+                    status: unknown ? "unknown" : "settled",
+                    ...(tokens !== undefined ? { tokens } : {}),
+                    ...(costMicros !== undefined ? { costMicros } : {})
+                }
+            }
+        };
+        const reason = reached(next);
+        if (reason && !budget.stopped)
+            return stopUpdate(run, next, reason);
+        return {
+            patch: { lifecycle: { ...run.lifecycle, taskBudget: next } },
+            events: [
+                {
+                    type: "task.budget_usage_recorded",
+                    data: {
+                        requestId,
+                        status: next.requests[requestId].status,
+                        tokens: tokens ?? null,
+                        costMicros: costMicros ?? null
+                    }
+                }
+            ]
+        };
+    }));
+    // Cancellation remains durable even when the response recorder is interrupted.
+    if (result.run.lifecycle?.taskBudget?.stopped)
+        await propagateRunCancellation(state, rootId);
+}
+export async function checkTaskBudget(state, runId) {
+    const ancestry = await assertRunTreeOpen(state, runId);
+    const root = ancestry.at(-1);
+    if (!root.lifecycle?.taskBudget || !reached(root.lifecycle.taskBudget))
+        return;
+    const result = await outsideRunExecution(() => state.transactRun(root.id, ({ run }) => {
+        const budget = run.lifecycle?.taskBudget;
+        const reason = budget && reached(budget);
+        return reason && !budget.stopped ? stopUpdate(run, budget, reason) : undefined;
+    }));
+    await enforceStop(state, result.run);
+}
+/** Dead workers cannot make unobserved token/cash usage look like zero on restart. */
+export async function recoverTaskBudgets(state) {
+    for (const root of await state.queryRuns({ taskBudgetPending: true })) {
+        const budget = root.lifecycle.taskBudget;
+        for (const [id, request] of Object.entries(budget.requests)) {
+            if (request.status !== "pending")
+                continue;
+            const run = await state.getRun(request.runId);
+            const lease = run?.lifecycle?.lease;
+            const active = lease
+                ? Date.parse(lease.expiresAt) > Date.now()
+                : run &&
+                    (run.status === "running" || run.controlIntent === "cancel") &&
+                    Date.now() - Date.parse(run.updatedAt) < 5 * 60_000;
+            if (!active)
+                await completeTaskBudgetRequest(state, root.id, id);
+        }
+        try {
+            await checkTaskBudget(state, root.id);
+        }
+        catch (error) {
+            if (!(error instanceof RunTreeStoppedError))
+                throw error;
+        }
+    }
+}
+export async function getTaskBudget(state, runId) {
+    const root = (await runAncestry(state, runId)).at(-1);
+    return {
+        rootRunId: root.id,
+        ...(root.lifecycle?.taskBudget ? { budget: root.lifecycle.taskBudget } : {})
+    };
+}
+//# sourceMappingURL=task-budget.js.map

package/dist/task-budget.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"task-budget.js","sourceRoot":"","sources":["../src/task-budget.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,kBAAkB,EAAE,wBAAwB,EAAE,MAAM,4BAA4B,CAAC;AAC1F,OAAO,EAAE,iBAAiB,EAAE,WAAW,EAAE,mBAAmB,EAAE,MAAM,eAAe,CAAC;AACpF,OAAO,EAAE,mBAAmB,EAAE,MAAM,0BAA0B,CAAC;AAG/D,SAAS,UAAU,CAAC,IAAe,EAAE,MAAkB,EAAE,MAAc;IACrE,MAAM,EAAE,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;IACpC,MAAM,OAAO,GAAG,EAAE,GAAG,MAAM,EAAE,OAAO,EAAE,MAAM,CAAC,OAAO,IAAI,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,CAAC;IACzE,MAAM,MAAM,GAAG,kBAAkB,CAC/B,IAAI,EACJ,IAAI,CAAC,SAAS,EAAE,YAAY,IAAI;QAC9B,EAAE,EAAE,UAAU,IAAI,CAAC,EAAE,EAAE;QACvB,SAAS,EAAE,IAAI,CAAC,EAAE;QAClB,WAAW,EAAE,EAAE;QACf,MAAM;QACN,OAAO,EAAE,IAAI;KACd,CACF,CAAC;IACF,OAAO;QACL,GAAG,MAAM;QACT,KAAK,EAAE;YACL,GAAG,MAAM,CAAC,KAAK;YACf,SAAS,EAAE,EAAE,GAAG,MAAM,CAAC,KAAM,CAAC,SAAU,EAAE,UAAU,EAAE,OAAO,EAAE;SAChE;QACD,MAAM,EAAE;YACN,GAAG,CAAC,MAAM,CAAC,MAAM,IAAI,EAAE,CAAC;YACxB;gBACE,EAAE,EAAE,uBAAuB,IAAI,CAAC,EAAE,EAAE;gBACpC,IAAI,EAAE,uBAAuB;gBAC7B,IAAI,EAAE;oBACJ,MAAM;oBACN,aAAa,EAAE,MAAM,CAAC,aAAa;oBACnC,MAAM,EAAE,MAAM,CAAC,MAAM;oBACrB,UAAU,EAAE,MAAM,CAAC,UAAU;iBAC9B;aACF;SACF;KACF,CAAC;AACJ,CAAC;AAED,SAAS,OAAO,CAAC,MAAkB,EAAE,SAAS,GAAG,KAAK;IACpD,IAAI,MAAM,CAAC,OAAO;QAAE,OAAO,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC;IACjD,IAAI,MAAM,CAAC,UAAU,IAAI,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,UAAU,CAAC,IAAI,IAAI,CAAC,GAAG,EAAE;QAAE,OAAO,iBAAiB,CAAC;IAC/F,IACE,SAAS;QACT,MAAM,CAAC,MAAM,CAAC,gBAAgB,KAAK,SAAS;QAC5C,MAAM,CAAC,aAAa,IAAI,MAAM,CAAC,MAAM,CAAC,gBAAgB;QAEtD,OAAO,uBAAuB,CAAC;IACjC,IAAI,MAAM,CAAC,MAAM,CAAC,SAAS,KAAK,SAAS,IAAI,MAAM,CAAC,MAAM,IAAI,MAAM,CAAC,MAAM,CAAC,SAAS;QACnF,OAAO,eAAe,CAAC;IACzB,IACE,MAAM,CAAC,MAAM,CAAC,UAAU,KAAK,SAAS;QACtC,MAAM,CAAC,UAAU,IAAI,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,UAAU,GAAG,GAAG,CAAC;QAE/D,OAAO,aAAa,CAAC;IACvB,IACE,CAAC,MAAM,CAAC,MAAM,CAAC,SAAS,KAAK,SAAS,IAAI,MAAM,CAAC,MAAM,CAAC,UAAU,KAAK,SAAS,CAAC;QACjF,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,MAAM,KAAK,SAAS,CAAC;QAE9E,OAAO,0BAA0B,CAAC;AACtC,CAAC;AAED,KAAK,UAAU,WAAW,CAAC,KAAmB,EAAE,IAAe;IAC7D,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,UAAU,EAAE,OAAO;QAAE,OAAO;IACjD,MAAM,wBAAwB,CAAC,KAAK,EAAE,IAAI,CAAC,EAAE,CAAC,CAAC;IAC/C,MAAM,IAAI,mBAAmB,CAAC,IAAI,CAAC,EAAE,EAAE,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;AACnF,CAAC;AAED,iFAAiF;AACjF,MAAM,CAAC,KAAK,UAAU,sBAAsB,CAC1C,KAAmB,EACnB,KAAa,EACb,SAAiB;IAEjB,MAAM,QAAQ,GAAG,MAAM,iBAAiB,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;IACvD,MAAM,IAAI,GAAG,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAC,CAAE,CAAC;IAC9B,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,UAAU;QAAE,OAAO;IACxC,MAAM,MAAM,GAAG,MAAM,mBAAmB,CAAC,GAAG,EAAE,CAC5C,KAAK,CAAC,WAAW,CAAC,IAAI,CAAC,EAAE,EAAE,CAAC,EAAE,GAAG,EAAE,EAAE,EAAE;QACrC,MAAM,MAAM,GAAG,GAAG,CAAC,SAAU,CAAC,UAAW,CAAC;QAC1C,IAAI,GAAG,CAAC,SAAS,EAAE,YAAY;YAAE,OAAO;QACxC,IAAI,MAAM,CAAC,QAAQ,CAAC,SAAS,CAAC;YAAE,OAAO;QACvC,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;QACrC,IAAI,MAAM;YAAE,OAAO,UAAU,CAAC,GAAG,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC;QACnD,MAAM,IAAI,GAAe;YACvB,GAAG,MAAM;YACT,aAAa,EAAE,MAAM,CAAC,aAAa,GAAG,CAAC;YACvC,eAAe,EAAE,MAAM,CAAC,eAAe,GAAG,CAAC;YAC3C,QAAQ,EAAE;gBACR,GAAG,MAAM,CAAC,QAAQ;gBAClB,CAAC,SAAS,CAAC,EAAE;oBACX,KAAK;oBACL,MAAM,EAAE,SAAS;oBACjB,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;iBACpC;aACF;SACF,CAAC;QACF,OAAO;YACL,KAAK,EAAE,EAAE,SAAS,EAAE,EAAE,GAAG,GAAG,CAAC,SAAU,EAAE,UAAU,EAAE,IAAI,EAAE,EAAE;YAC7D,MAAM,EAAE;gBACN;oBACE,IAAI,EAAE,8BAA8B;oBACpC,IAAI,EAAE,EAAE,SAAS,EAAE,KAAK,EAAE,aAAa,EAAE,IAAI,CAAC,aAAa,EAAE;iBAC9D;aACF;SACF,CAAC;IACJ,CAAC,CAAC,CACH,CAAC;IACF,MAAM,WAAW,CAAC,KAAK,EAAE,MAAM,CAAC,GAAG,CAAC,CAAC;IACrC,IAAI,MAAM,CAAC,GAAG,CAAC,SAAS,EAAE,YAAY;QACpC,MAAM,IAAI,mBAAmB,CAAC,IAAI,CAAC,EAAE,EAAE,MAAM,CAAC,GAAG,CAAC,SAAS,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC;IACnF,OAAO,IAAI,CAAC,EAAE,CAAC;AACjB,CAAC;AAED,uFAAuF;AACvF,MAAM,CAAC,KAAK,UAAU,yBAAyB,CAC7C,KAAmB,EACnB,MAAc,EACd,SAAiB,EACjB,KAAoB;IAEpB,MAAM,MAAM,GAAG,MAAM,mBAAmB,CAAC,GAAG,EAAE,CAC5C,KAAK,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC,EAAE,GAAG,EAAE,EAAE,EAAE;QACpC,MAAM,MAAM,GAAG,GAAG,CAAC,SAAS,EAAE,UAAU,CAAC;QACzC,MAAM,QAAQ,GAAG,MAAM,EAAE,QAAQ,CAAC,SAAS,CAAC,CAAC;QAC7C,IAAI,CAAC,MAAM,IAAI,CAAC,QAAQ,IAAI,QAAQ,CAAC,MAAM,KAAK,SAAS;YAAE,OAAO;QAClE,MAAM,KAAK,GAAG,CAAC,KAAyB,EAAmB,EAAE,CAC3D,OAAO,KAAK,KAAK,QAAQ,IAAI,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,KAAK,IAAI,CAAC,CAAC;QACpE,MAAM,cAAc,GAClB,KAAK,CAAC,KAAK,EAAE,WAAW,CAAC;YACzB,KAAK,CAAC,KAAK,EAAE,YAAY,CAAC;YAC1B,MAAM,CAAC,aAAa,CAAC,KAAK,CAAC,WAAW,GAAG,KAAK,CAAC,YAAY,CAAC;YAC5D,MAAM,CAAC,aAAa,CAAC,KAAK,CAAC,WAAW,CAAC;YACvC,KAAK,EAAE,WAAW,KAAK,aAAa;YAClC,CAAC,CAAC,KAAK,CAAC,WAAW,GAAG,KAAK,CAAC,YAAY;YACxC,CAAC,CAAC,SAAS,CAAC;QAChB,MAAM,kBAAkB,GACtB,KAAK,CAAC,KAAK,EAAE,OAAO,CAAC;YACrB,MAAM,CAAC,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,GAAG,GAAG,CAAC,CAAC;YACpD,CAAC,KAAK,CAAC,UAAU,KAAK,mBAAmB,IAAI,KAAK,CAAC,UAAU,KAAK,qBAAqB,CAAC;YACxF,CAAC,KAAK,CAAC,QAAQ,KAAK,SAAS,IAAI,KAAK,CAAC,QAAQ,KAAK,KAAK,CAAC;YACxD,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,GAAG,GAAG,CAAC;YAChC,CAAC,CAAC,SAAS,CAAC;QAChB,2FAA2F;QAC3F,MAAM,MAAM,GAAG,QAAQ,CAAC,MAAM,IAAI,cAAc,CAAC;QACjD,MAAM,UAAU,GAAG,QAAQ,CAAC,UAAU,IAAI,kBAAkB,CAAC;QAC7D,IAAI,QAAQ,CAAC,MAAM,KAAK,SAAS,IAAI,MAAM,KAAK,QAAQ,CAAC,MAAM,IAAI,UAAU,KAAK,QAAQ,CAAC,UAAU;YACnG,OAAO;QACT,MAAM,OAAO,GACX,CAAC,MAAM,CAAC,MAAM,CAAC,SAAS,KAAK,SAAS,IAAI,MAAM,KAAK,SAAS,CAAC;YAC/D,CAAC,MAAM,CAAC,MAAM,CAAC,UAAU,KAAK,SAAS,IAAI,UAAU,KAAK,SAAS,CAAC,CAAC;QACvE,MAAM,IAAI,GAAe;YACvB,GAAG,MAAM;YACT,eAAe,EAAE,MAAM,CAAC,eAAe,GAAG,CAAC,QAAQ,CAAC,MAAM,KAAK,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YACjF,MAAM,EAAE,MAAM,CAAC,MAAM,GAAG,CAAC,MAAM,IAAI,CAAC,CAAC,GAAG,CAAC,QAAQ,CAAC,MAAM,IAAI,CAAC,CAAC;YAC9D,UAAU,EAAE,MAAM,CAAC,UAAU,GAAG,CAAC,UAAU,IAAI,CAAC,CAAC,GAAG,CAAC,QAAQ,CAAC,UAAU,IAAI,CAAC,CAAC;YAC9E,QAAQ,EAAE;gBACR,GAAG,MAAM,CAAC,QAAQ;gBAClB,CAAC,SAAS,CAAC,EAAE;oBACX,GAAG,QAAQ;oBACX,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS;oBACvC,GAAG,CAAC,MAAM,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;oBAC3C,GAAG,CAAC,UAAU,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;iBACpD;aACF;SACF,CAAC;QACF,MAAM,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;QAC7B,IAAI,MAAM,IAAI,CAAC,MAAM,CAAC,OAAO;YAAE,OAAO,UAAU,CAAC,GAAG,EAAE,IAAI,EAAE,MAAM,CAAC,CAAC;QACpE,OAAO;YACL,KAAK,EAAE,EAAE,SAAS,EAAE,EAAE,GAAG,GAAG,CAAC,SAAU,EAAE,UAAU,EAAE,IAAI,EAAE,EAAE;YAC7D,MAAM,EAAE;gBACN;oBACE,IAAI,EAAE,4BAA4B;oBAClC,IAAI,EAAE;wBACJ,SAAS;wBACT,MAAM,EAAE,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAE,CAAC,MAAM;wBACxC,MAAM,EAAE,MAAM,IAAI,IAAI;wBACtB,UAAU,EAAE,UAAU,IAAI,IAAI;qBAC/B;iBACF;aACF;SACF,CAAC;IACJ,CAAC,CAAC,CACH,CAAC;IACF,+EAA+E;IAC/E,IAAI,MAAM,CAAC,GAAG,CAAC,SAAS,EAAE,UAAU,EAAE,OAAO;QAAE,MAAM,wBAAwB,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;AAC/F,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,eAAe,CAAC,KAAmB,EAAE,KAAa;IACtE,MAAM,QAAQ,GAAG,MAAM,iBAAiB,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;IACvD,MAAM,IAAI,GAAG,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAC,CAAE,CAAC;IAC9B,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,UAAU,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC;QAAE,OAAO;IAC/E,MAAM,MAAM,GAAG,MAAM,mBAAmB,CAAC,GAAG,EAAE,CAC5C,KAAK,CAAC,WAAW,CAAC,IAAI,CAAC,EAAE,EAAE,CAAC,EAAE,GAAG,EAAE,EAAE,EAAE;QACrC,MAAM,MAAM,GAAG,GAAG,CAAC,SAAS,EAAE,UAAU,CAAC;QACzC,MAAM,MAAM,GAAG,MAAM,IAAI,OAAO,CAAC,MAAM,CAAC,CAAC;QACzC,OAAO,MAAM,IAAI,CAAC,MAAO,CAAC,OAAO,CAAC,CAAC,CAAC,UAAU,CAAC,GAAG,EAAE,MAAO,EAAE,MAAM,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IACnF,CAAC,CAAC,CACH,CAAC;IACF,MAAM,WAAW,CAAC,KAAK,EAAE,MAAM,CAAC,GAAG,CAAC,CAAC;AACvC,CAAC;AAED,sFAAsF;AACtF,MAAM,CAAC,KAAK,UAAU,kBAAkB,CAAC,KAAmB;IAC1D,KAAK,MAAM,IAAI,IAAI,MAAM,KAAK,CAAC,SAAS,CAAC,EAAE,iBAAiB,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC;QACtE,MAAM,MAAM,GAAG,IAAI,CAAC,SAAU,CAAC,UAAW,CAAC;QAC3C,KAAK,MAAM,CAAC,EAAE,EAAE,OAAO,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,EAAE,CAAC;YAC5D,IAAI,OAAO,CAAC,MAAM,KAAK,SAAS;gBAAE,SAAS;YAC3C,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;YAC9C,MAAM,KAAK,GAAG,GAAG,EAAE,SAAS,EAAE,KAAK,CAAC;YACpC,MAAM,MAAM,GAAG,KAAK;gBAClB,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,SAAS,CAAC,GAAG,IAAI,CAAC,GAAG,EAAE;gBAC1C,CAAC,CAAC,GAAG;oBACH,CAAC,GAAG,CAAC,MAAM,KAAK,SAAS,IAAI,GAAG,CAAC,aAAa,KAAK,QAAQ,CAAC;oBAC5D,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,SAAS,CAAC,GAAG,CAAC,GAAG,MAAM,CAAC;YACxD,IAAI,CAAC,MAAM;gBAAE,MAAM,yBAAyB,CAAC,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;QACnE,CAAC;QACD,IAAI,CAAC;YACH,MAAM,eAAe,CAAC,KAAK,EAAE,IAAI,CAAC,EAAE,CAAC,CAAC;QACxC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,CAAC,KAAK,YAAY,mBAAmB,CAAC;gBAAE,MAAM,KAAK,CAAC;QAC3D,CAAC;IACH,CAAC;AACH,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,aAAa,CACjC,KAAmB,EACnB,KAAa;IAEb,MAAM,IAAI,GAAG,CAAC,MAAM,WAAW,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAE,CAAC;IACvD,OAAO;QACL,SAAS,EAAE,IAAI,CAAC,EAAE;QAClB,GAAG,CAAC,IAAI,CAAC,SAAS,EAAE,UAAU,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KAC7E,CAAC;AACJ,CAAC"}

package/dist/task-deadline.d.ts

@@ -0,0 +1,4 @@
+import type { StateAdapter } from "./state.js";
+/** Timers are only an accelerator; the persisted deadline is checked again at every boundary. */
+export declare function armTaskDeadline(state: StateAdapter, runId: string, abort: () => void, onError: (error: unknown) => void): () => void;
+//# sourceMappingURL=task-deadline.d.ts.map

package/dist/task-deadline.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"task-deadline.d.ts","sourceRoot":"","sources":["../src/task-deadline.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,YAAY,CAAC;AAG/C,iGAAiG;AACjG,wBAAgB,eAAe,CAC7B,KAAK,EAAE,YAAY,EACnB,KAAK,EAAE,MAAM,EACb,KAAK,EAAE,MAAM,IAAI,EACjB,OAAO,EAAE,CAAC,KAAK,EAAE,OAAO,KAAK,IAAI,GAChC,MAAM,IAAI,CAkCZ"}

package/dist/task-deadline.js

@@ -0,0 +1,36 @@
+import { checkTaskBudget, getTaskBudget } from "./task-budget.js";
+/** Timers are only an accelerator; the persisted deadline is checked again at every boundary. */
+export function armTaskDeadline(state, runId, abort, onError) {
+    let stopped = false;
+    let timer;
+    const arm = async () => {
+        const { budget } = await getTaskBudget(state, runId);
+        if (stopped || !budget?.deadlineAt)
+            return;
+        const remaining = Date.parse(budget.deadlineAt) - Date.now();
+        if (remaining <= 0) {
+            abort();
+            try {
+                await checkTaskBudget(state, runId);
+            }
+            catch (error) {
+                onError(error);
+            }
+            return;
+        }
+        timer = setTimeout(() => void arm().catch((error) => {
+            abort();
+            onError(error);
+        }), Math.min(remaining, 2_147_483_647));
+        timer.unref?.();
+    };
+    void arm().catch((error) => {
+        abort();
+        onError(error);
+    });
+    return () => {
+        stopped = true;
+        clearTimeout(timer);
+    };
+}
+//# sourceMappingURL=task-deadline.js.map

package/dist/task-deadline.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"task-deadline.js","sourceRoot":"","sources":["../src/task-deadline.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,eAAe,EAAE,aAAa,EAAE,MAAM,kBAAkB,CAAC;AAElE,iGAAiG;AACjG,MAAM,UAAU,eAAe,CAC7B,KAAmB,EACnB,KAAa,EACb,KAAiB,EACjB,OAAiC;IAEjC,IAAI,OAAO,GAAG,KAAK,CAAC;IACpB,IAAI,KAAgD,CAAC;IACrD,MAAM,GAAG,GAAG,KAAK,IAAI,EAAE;QACrB,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,aAAa,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;QACrD,IAAI,OAAO,IAAI,CAAC,MAAM,EAAE,UAAU;YAAE,OAAO;QAC3C,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,UAAU,CAAC,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC7D,IAAI,SAAS,IAAI,CAAC,EAAE,CAAC;YACnB,KAAK,EAAE,CAAC;YACR,IAAI,CAAC;gBACH,MAAM,eAAe,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;YACtC,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,OAAO,CAAC,KAAK,CAAC,CAAC;YACjB,CAAC;YACD,OAAO;QACT,CAAC;QACD,KAAK,GAAG,UAAU,CAChB,GAAG,EAAE,CACH,KAAK,GAAG,EAAE,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;YACzB,KAAK,EAAE,CAAC;YACR,OAAO,CAAC,KAAK,CAAC,CAAC;QACjB,CAAC,CAAC,EACJ,IAAI,CAAC,GAAG,CAAC,SAAS,EAAE,aAAa,CAAC,CACnC,CAAC;QACF,KAAK,CAAC,KAAK,EAAE,EAAE,CAAC;IAClB,CAAC,CAAC;IACF,KAAK,GAAG,EAAE,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;QACzB,KAAK,EAAE,CAAC;QACR,OAAO,CAAC,KAAK,CAAC,CAAC;IACjB,CAAC,CAAC,CAAC;IACH,OAAO,GAAG,EAAE;QACV,OAAO,GAAG,IAAI,CAAC;QACf,YAAY,CAAC,KAAK,CAAC,CAAC;IACtB,CAAC,CAAC;AACJ,CAAC"}

package/dist/tool-execution-host.d.ts

@@ -17,6 +17,7 @@ export interface ToolRuntimeContext {
     manifest: AgentManifest;
     coreToolPolicy: ResolvedCoreToolPolicy;
     authoredToolExecution: AuthoredToolExecutionPolicy;
+    backgroundOnlySubagents: ReadonlySet<string>;
     toolOutputMaxChars: number;
     toolTimeoutMs: number;
     surfaceForRun(run: RunRecord): AgentSurface;
@@ -104,10 +105,12 @@ export interface ToolLearningServices {
 }
 export interface ToolSubagentServices {
     runSubagent(parentRunId: string, subagentName: string, task: JsonObject): Promise<RunTimeline>;
-    queueSubagent(parentRunId: string, subagentName: string, task: JsonObject): Promise<RunRecord>;
+    queueSubagent(parentRunId: string, subagentName: string, task: JsonObject, commandId?: string): Promise<RunRecord>;
+    queueSubagentResume(parentRunId: string, childRunId: string, subagentName: string, message: string, commandId?: string): Promise<RunRecord>;
     resumeSubagent(parentRunId: string, childRunId: string, subagentName: string, message: string): Promise<RunTimeline>;
     listBackgroundWork(rootRunId: string, includeTerminal?: boolean): Promise<JsonObject[]>;
     getBackgroundWork(rootRunId: string, childRunId: string): Promise<RunRecord>;
+    commandBackgroundWork(rootRunId: string, childRunId: string, input: import("./run-commands.js").RunCommandInput): Promise<import("./state.js").RunCommand>;
     cancelBackgroundWork(rootRunId: string, childRunId: string): Promise<JsonObject>;
 }
 export interface ToolPersistenceServices {
@@ -118,5 +121,6 @@ export interface ToolPersistenceServices {
     projectResourcesForRun(run: RunRecord, scope: MemoryScope, sandbox: SandboxSession, input: ResourceProjectInput): Promise<ResourceProjectResult>;
 }
 export interface ToolExecutionHost extends ToolRuntimeContext, ToolSandboxServices, ToolConnectionServices, ToolLearningServices, ToolSubagentServices, ToolPersistenceServices {
+    registerRunAbort(runId: string, controller: AbortController): () => void;
 }
 //# sourceMappingURL=tool-execution-host.d.ts.map

package/dist/tool-execution-host.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"tool-execution-host.d.ts","sourceRoot":"","sources":["../src/tool-execution-host.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,aAAa,EACb,gBAAgB,EAChB,kBAAkB,EAClB,YAAY,EACZ,UAAU,EACV,iBAAiB,EACjB,gBAAgB,EAChB,oBAAoB,EACpB,qBAAqB,EACrB,iBAAiB,EACjB,kBAAkB,EAClB,cAAc,EACd,cAAc,EACd,oBAAoB,EACrB,MAAM,2BAA2B,CAAC;AACnC,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AACvD,OAAO,KAAK,EACV,mCAAmC,EACnC,oBAAoB,EACrB,MAAM,kBAAkB,CAAC;AAC1B,OAAO,KAAK,EAAE,8BAA8B,EAAE,MAAM,+BAA+B,CAAC;AACpF,OAAO,KAAK,EAAE,sBAAsB,EAAE,MAAM,eAAe,CAAC;AAC5D,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,oBAAoB,CAAC;AAC3D,OAAO,KAAK,EACV,qBAAqB,EACrB,yBAAyB,EACzB,uBAAuB,EACvB,oBAAoB,EACpB,UAAU,EACV,wBAAwB,EACzB,MAAM,uBAAuB,CAAC;AAC/B,OAAO,KAAK,EACV,gBAAgB,EAChB,WAAW,EACX,SAAS,EACT,YAAY,EACb,MAAM,YAAY,CAAC;AACpB,OAAO,KAAK,EACV,2BAA2B,EAC3B,WAAW,EACX,2BAA2B,EAC3B,eAAe,EACf,WAAW,EACX,WAAW,EACX,oBAAoB,EACpB,cAAc,EACf,MAAM,oBAAoB,CAAC;AAC5B,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,wBAAwB,CAAC;AACnE,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,wBAAwB,CAAC;AAE/D,MAAM,WAAW,kBAAkB;IACjC,KAAK,EAAE,YAAY,CAAC;IACpB,IAAI,EAAE,WAAW,CAAC;IAClB,GAAG,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,SAAS,CAAC,CAAC;IACxC,KAAK,EAAE,OAAO,KAAK,CAAC;IACpB,QAAQ,EAAE,aAAa,CAAC;IACxB,cAAc,EAAE,sBAAsB,CAAC;IACvC,qBAAqB,EAAE,2BAA2B,CAAC;IACnD,kBAAkB,EAAE,MAAM,CAAC;IAC3B,aAAa,EAAE,MAAM,CAAC;IACtB,aAAa,CAAC,GAAG,EAAE,SAAS,GAAG,YAAY,CAAC;IAC5C,gBAAgB,CAAC,KAAK,EAAE,MAAM,GAAG,gBAAgB,CAAC;IAClD,iBAAiB,CAAC,KAAK,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,UAAU,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IACnE,uBAAuB,CAAC,GAAG,EAAE,SAAS,EAAE,SAAS,EAAE,MAAM,EAAE,KAAK,EAAE,UAAU,GAAG,OAAO,CAAC,UAAU,CAAC,CAAC;IACnG,eAAe,CAAC,GAAG,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,GAAG,CAAC,CAAC,KAAK,EAAE,OAAO,KAAK,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC,GAAG,SAAS,CAAC;IAChH,QAAQ,CAAC,IAAI,EAAE,YAAY,EAAE,GAAG,CAAC,EAAE,SAAS,GAAG,OAAO,CAAC,cAAc,CAAC,CAAC;IACvE,QAAQ,CAAC,IAAI,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,UAAU,EAAE,UAAU,EAAE,OAAO,CAAC,EAAE,eAAe,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAC5G,iBAAiB,CAAC,GAAG,EAAE,SAAS,EAAE,KAAK,CAAC,EAAE,UAAU,GAAG,UAAU,CAAC;IAClE,UAAU,CAAC,KAAK,EAAE,MAAM,GAAG,MAAM,CAAC;IAClC,qBAAqB,CAAC,KAAK,EAAE,OAAO,EAAE,MAAM,EAAE,OAAO,GAAG,UAAU,CAAC;CACpE;AAED,MAAM,WAAW,mBAAmB;IAClC,OAAO,EAAE,cAAc,CAAC;IACxB,sBAAsB,EAAE,MAAM,CAAC;IAC/B,iBAAiB,CAAC,GAAG,EAAE,SAAS,GAAG,MAAM,CAAC;IAC1C,yBAAyB,CAAC,IAAI,EAAE,YAAY,EAAE,GAAG,CAAC,EAAE,SAAS,GAAG,OAAO,CAAC;IACxE,qBAAqB,CAAC,IAAI,EAAE,YAAY,EAAE,aAAa,EAAE,oBAAoB,GAAG,OAAO,CAAC,cAAc,CAAC,CAAC;IACxG,oBAAoB,CAClB,IAAI,EAAE,YAAY,EAClB,KAAK,EAAE,OAAO,EACd,OAAO,EAAE,oBAAoB,EAC7B,aAAa,EAAE,oBAAoB,EACnC,MAAM,CAAC,EAAE,WAAW,GACnB,OAAO,CAAC;QAAE,MAAM,EAAE,OAAO,CAAC;QAAC,cAAc,EAAE,OAAO,CAAC;QAAC,WAAW,CAAC,EAAE,OAAO,CAAA;KAAE,CAAC,CAAC;IAChF,uBAAuB,CAAC,GAAG,EAAE,SAAS,EAAE,OAAO,EAAE,cAAc,EAAE,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IACnG,yBAAyB,CACvB,GAAG,EAAE,SAAS,EACd,OAAO,EAAE,cAAc,EACvB,IAAI,EAAE,MAAM,EACZ,OAAO,CAAC,EAAE;QAAE,IAAI,CAAC,EAAE,OAAO,CAAC;QAAC,KAAK,CAAC,EAAE,MAAM,CAAA;KAAE,GAC3C,OAAO,CAAC,IAAI,CAAC,CAAC;IACjB,sBAAsB,CAAC,GAAG,EAAE,SAAS,EAAE,OAAO,EAAE,cAAc,EAAE,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAC7F,oBAAoB,CAClB,GAAG,EAAE,SAAS,EACd,QAAQ,EAAE,MAAM,EAChB,KAAK,EAAE,OAAO,EACd,aAAa,EAAE,oBAAoB,GAClC,OAAO,CAAC,OAAO,CAAC,CAAC;CACrB;AAED,MAAM,WAAW,sBAAsB;IACrC,WAAW,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACrC,gBAAgB,EAAE,gBAAgB,CAAC;IACnC,oBAAoB,EAAE,oBAAoB,CAAC;IAC3C,mCAAmC,EAAE,mCAAmC,CAAC;IACzE,yBAAyB,EAAE,MAAM,GAAG,SAAS,CAAC;IAC9C,iBAAiB,EAAE,yBAAyB,CAAC;IAC7C,iBAAiB,IAAI,kBAAkB,EAAE,CAAC;IAC1C,wBAAwB,CAAC,UAAU,EAAE,kBAAkB,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC;IAC3E,wBAAwB,CAAC,KAAK,EAAE;QAC9B,IAAI,EAAE,MAAM,CAAC;QACb,QAAQ,EAAE,wBAAwB,CAAC;QACnC,WAAW,EAAE,MAAM,CAAC;QACpB,UAAU,EAAE,UAAU,CAAC;QACvB,KAAK,CAAC,EAAE,2BAA2B,CAAC;QACpC,KAAK,CAAC,EAAE,MAAM,CAAC;KAChB,GAAG,OAAO,CAAC;QAAE,IAAI,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IAC9B,yBAAyB,CAAC,KAAK,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,UAAU,EAAE,CAAC,CAAC;IACjE,0BAA0B,CAAC,IAAI,EAAE,MAAM,EAAE,KAAK,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;CACzE;AAED,MAAM,WAAW,oBAAoB;IACnC,UAAU,EAAE,UAAU,CAAC;IACvB,gBAAgB,EAAE,gBAAgB,CAAC;IACnC,WAAW,EAAE,WAAW,CAAC;IACzB,kBAAkB,CAAC,GAAG,EAAE,SAAS,GAAG,oBAAoB,CAAC;IACzD,eAAe,EAAE,uBAAuB,CAAC;IACzC,wBAAwB,CAAC,GAAG,EAAE,SAAS,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IACxD,gBAAgB,CAAC,GAAG,EAAE,SAAS,GAAG,MAAM,CAAC;IACzC,mBAAmB,CAAC,GAAG,EAAE,SAAS,GAAG,MAAM,CAAC;IAC5C,iBAAiB,CAAC,GAAG,EAAE,SAAS,GAAG,WAAW,CAAC;IAC/C,eAAe,CAAC,KAAK,EAAE,oBAAoB,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC;IAC/D,eAAe,CACb,KAAK,EAAE;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,IAAI,EAAE,MAAM,CAAC;QAAC,MAAM,CAAC,EAAE,OAAO,GAAG,MAAM,CAAA;KAAE,EAChE,GAAG,EAAE,SAAS,GACb,OAAO,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,OAAO,CAAA;KAAE,CAAC,CAAC;IAC/C,gBAAgB,CAAC,GAAG,EAAE,SAAS,GAAG,OAAO,CAAC,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,WAAW,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,OAAO,CAAA;KAAE,CAAC,CAAC,CAAC;IAC1H,iBAAiB,CAAC,IAAI,EAAE,MAAM,EAAE,GAAG,EAAE,SAAS,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAC/D,uBAAuB,CAAC,KAAK,EAAE,8BAA8B,GAAG,OAAO,CAAC,qBAAqB,CAAC,CAAC;IAC/F,sBAAsB,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,qBAAqB,EAAE,CAAC,CAAC;IACxE,uBAAuB,CAAC,KAAK,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,UAAU,GAAG,OAAO,CAAC,qBAAqB,CAAC,CAAC;IACtG,uBAAuB,CAAC,KAAK,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC;CACtE;AAED,MAAM,WAAW,oBAAoB;IACnC,WAAW,CAAC,WAAW,EAAE,MAAM,EAAE,YAAY,EAAE,MAAM,EAAE,IAAI,EAAE,UAAU,GAAG,OAAO,CAAC,WAAW,CAAC,CAAC;IAC/F,aAAa,CAAC,WAAW,EAAE,MAAM,EAAE,YAAY,EAAE,MAAM,EAAE,IAAI,EAAE,UAAU,GAAG,OAAO,CAAC,SAAS,CAAC,CAAC;IAC/F,cAAc,CAAC,WAAW,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,EAAE,YAAY,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,WAAW,CAAC,CAAC;IACrH,kBAAkB,CAAC,SAAS,EAAE,MAAM,EAAE,eAAe,CAAC,EAAE,OAAO,GAAG,OAAO,CAAC,UAAU,EAAE,CAAC,CAAC;IACxF,iBAAiB,CAAC,SAAS,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,SAAS,CAAC,CAAC;IAC7E,oBAAoB,CAAC,SAAS,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,UAAU,CAAC,CAAC;CAClF;AAED,MAAM,WAAW,uBAAuB;IACtC,gBAAgB,CAAC,KAAK,EAAE,IAAI,CAAC,gBAAgB,EAAE,IAAI,GAAG,WAAW,CAAC,GAAG,OAAO,CAAC,gBAAgB,CAAC,CAAC;IAC/F,uBAAuB,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,gBAAgB,EAAE,CAAC,CAAC;IACpE,mBAAmB,CAAC,GAAG,EAAE,SAAS,EAAE,KAAK,EAAE,WAAW,EAAE,KAAK,CAAC,EAAE,iBAAiB,GAAG,OAAO,CAAC,gBAAgB,EAAE,CAAC,CAAC;IAChH,kBAAkB,CAAC,GAAG,EAAE,SAAS,EAAE,KAAK,EAAE,WAAW,EAAE,KAAK,EAAE,iBAAiB,GAAG,OAAO,CAAC,kBAAkB,CAAC,CAAC;IAC9G,sBAAsB,CACpB,GAAG,EAAE,SAAS,EACd,KAAK,EAAE,WAAW,EAClB,OAAO,EAAE,cAAc,EACvB,KAAK,EAAE,oBAAoB,GAC1B,OAAO,CAAC,qBAAqB,CAAC,CAAC;CACnC;AAED,MAAM,WAAW,iBACf,SAAQ,kBAAkB,EACxB,mBAAmB,EACnB,sBAAsB,EACtB,oBAAoB,EACpB,oBAAoB,EACpB,uBAAuB;CAAG"}
+{"version":3,"file":"tool-execution-host.d.ts","sourceRoot":"","sources":["../src/tool-execution-host.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,aAAa,EACb,gBAAgB,EAChB,kBAAkB,EAClB,YAAY,EACZ,UAAU,EACV,iBAAiB,EACjB,gBAAgB,EAChB,oBAAoB,EACpB,qBAAqB,EACrB,iBAAiB,EACjB,kBAAkB,EAClB,cAAc,EACd,cAAc,EACd,oBAAoB,EACrB,MAAM,2BAA2B,CAAC;AACnC,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AACvD,OAAO,KAAK,EACV,mCAAmC,EACnC,oBAAoB,EACrB,MAAM,kBAAkB,CAAC;AAC1B,OAAO,KAAK,EAAE,8BAA8B,EAAE,MAAM,+BAA+B,CAAC;AACpF,OAAO,KAAK,EAAE,sBAAsB,EAAE,MAAM,eAAe,CAAC;AAC5D,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,oBAAoB,CAAC;AAC3D,OAAO,KAAK,EACV,qBAAqB,EACrB,yBAAyB,EACzB,uBAAuB,EACvB,oBAAoB,EACpB,UAAU,EACV,wBAAwB,EACzB,MAAM,uBAAuB,CAAC;AAC/B,OAAO,KAAK,EACV,gBAAgB,EAChB,WAAW,EACX,SAAS,EACT,YAAY,EACb,MAAM,YAAY,CAAC;AACpB,OAAO,KAAK,EACV,2BAA2B,EAC3B,WAAW,EACX,2BAA2B,EAC3B,eAAe,EACf,WAAW,EACX,WAAW,EACX,oBAAoB,EACpB,cAAc,EACf,MAAM,oBAAoB,CAAC;AAC5B,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,wBAAwB,CAAC;AACnE,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,wBAAwB,CAAC;AAE/D,MAAM,WAAW,kBAAkB;IACjC,KAAK,EAAE,YAAY,CAAC;IACpB,IAAI,EAAE,WAAW,CAAC;IAClB,GAAG,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,SAAS,CAAC,CAAC;IACxC,KAAK,EAAE,OAAO,KAAK,CAAC;IACpB,QAAQ,EAAE,aAAa,CAAC;IACxB,cAAc,EAAE,sBAAsB,CAAC;IACvC,qBAAqB,EAAE,2BAA2B,CAAC;IACnD,uBAAuB,EAAE,WAAW,CAAC,MAAM,CAAC,CAAC;IAC7C,kBAAkB,EAAE,MAAM,CAAC;IAC3B,aAAa,EAAE,MAAM,CAAC;IACtB,aAAa,CAAC,GAAG,EAAE,SAAS,GAAG,YAAY,CAAC;IAC5C,gBAAgB,CAAC,KAAK,EAAE,MAAM,GAAG,gBAAgB,CAAC;IAClD,iBAAiB,CAAC,KAAK,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,UAAU,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IACnE,uBAAuB,CAAC,GAAG,EAAE,SAAS,EAAE,SAAS,EAAE,MAAM,EAAE,KAAK,EAAE,UAAU,GAAG,OAAO,CAAC,UAAU,CAAC,CAAC;IACnG,eAAe,CAAC,GAAG,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,GAAG,CAAC,CAAC,KAAK,EAAE,OAAO,KAAK,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC,GAAG,SAAS,CAAC;IAChH,QAAQ,CAAC,IAAI,EAAE,YAAY,EAAE,GAAG,CAAC,EAAE,SAAS,GAAG,OAAO,CAAC,cAAc,CAAC,CAAC;IACvE,QAAQ,CAAC,IAAI,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,UAAU,EAAE,UAAU,EAAE,OAAO,CAAC,EAAE,eAAe,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAC5G,iBAAiB,CAAC,GAAG,EAAE,SAAS,EAAE,KAAK,CAAC,EAAE,UAAU,GAAG,UAAU,CAAC;IAClE,UAAU,CAAC,KAAK,EAAE,MAAM,GAAG,MAAM,CAAC;IAClC,qBAAqB,CAAC,KAAK,EAAE,OAAO,EAAE,MAAM,EAAE,OAAO,GAAG,UAAU,CAAC;CACpE;AAED,MAAM,WAAW,mBAAmB;IAClC,OAAO,EAAE,cAAc,CAAC;IACxB,sBAAsB,EAAE,MAAM,CAAC;IAC/B,iBAAiB,CAAC,GAAG,EAAE,SAAS,GAAG,MAAM,CAAC;IAC1C,yBAAyB,CAAC,IAAI,EAAE,YAAY,EAAE,GAAG,CAAC,EAAE,SAAS,GAAG,OAAO,CAAC;IACxE,qBAAqB,CAAC,IAAI,EAAE,YAAY,EAAE,aAAa,EAAE,oBAAoB,GAAG,OAAO,CAAC,cAAc,CAAC,CAAC;IACxG,oBAAoB,CAClB,IAAI,EAAE,YAAY,EAClB,KAAK,EAAE,OAAO,EACd,OAAO,EAAE,oBAAoB,EAC7B,aAAa,EAAE,oBAAoB,EACnC,MAAM,CAAC,EAAE,WAAW,GACnB,OAAO,CAAC;QAAE,MAAM,EAAE,OAAO,CAAC;QAAC,cAAc,EAAE,OAAO,CAAC;QAAC,WAAW,CAAC,EAAE,OAAO,CAAA;KAAE,CAAC,CAAC;IAChF,uBAAuB,CAAC,GAAG,EAAE,SAAS,EAAE,OAAO,EAAE,cAAc,EAAE,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IACnG,yBAAyB,CACvB,GAAG,EAAE,SAAS,EACd,OAAO,EAAE,cAAc,EACvB,IAAI,EAAE,MAAM,EACZ,OAAO,CAAC,EAAE;QAAE,IAAI,CAAC,EAAE,OAAO,CAAC;QAAC,KAAK,CAAC,EAAE,MAAM,CAAA;KAAE,GAC3C,OAAO,CAAC,IAAI,CAAC,CAAC;IACjB,sBAAsB,CAAC,GAAG,EAAE,SAAS,EAAE,OAAO,EAAE,cAAc,EAAE,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAC7F,oBAAoB,CAClB,GAAG,EAAE,SAAS,EACd,QAAQ,EAAE,MAAM,EAChB,KAAK,EAAE,OAAO,EACd,aAAa,EAAE,oBAAoB,GAClC,OAAO,CAAC,OAAO,CAAC,CAAC;CACrB;AAED,MAAM,WAAW,sBAAsB;IACrC,WAAW,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACrC,gBAAgB,EAAE,gBAAgB,CAAC;IACnC,oBAAoB,EAAE,oBAAoB,CAAC;IAC3C,mCAAmC,EAAE,mCAAmC,CAAC;IACzE,yBAAyB,EAAE,MAAM,GAAG,SAAS,CAAC;IAC9C,iBAAiB,EAAE,yBAAyB,CAAC;IAC7C,iBAAiB,IAAI,kBAAkB,EAAE,CAAC;IAC1C,wBAAwB,CAAC,UAAU,EAAE,kBAAkB,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC;IAC3E,wBAAwB,CAAC,KAAK,EAAE;QAC9B,IAAI,EAAE,MAAM,CAAC;QACb,QAAQ,EAAE,wBAAwB,CAAC;QACnC,WAAW,EAAE,MAAM,CAAC;QACpB,UAAU,EAAE,UAAU,CAAC;QACvB,KAAK,CAAC,EAAE,2BAA2B,CAAC;QACpC,KAAK,CAAC,EAAE,MAAM,CAAC;KAChB,GAAG,OAAO,CAAC;QAAE,IAAI,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IAC9B,yBAAyB,CAAC,KAAK,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,UAAU,EAAE,CAAC,CAAC;IACjE,0BAA0B,CAAC,IAAI,EAAE,MAAM,EAAE,KAAK,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;CACzE;AAED,MAAM,WAAW,oBAAoB;IACnC,UAAU,EAAE,UAAU,CAAC;IACvB,gBAAgB,EAAE,gBAAgB,CAAC;IACnC,WAAW,EAAE,WAAW,CAAC;IACzB,kBAAkB,CAAC,GAAG,EAAE,SAAS,GAAG,oBAAoB,CAAC;IACzD,eAAe,EAAE,uBAAuB,CAAC;IACzC,wBAAwB,CAAC,GAAG,EAAE,SAAS,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IACxD,gBAAgB,CAAC,GAAG,EAAE,SAAS,GAAG,MAAM,CAAC;IACzC,mBAAmB,CAAC,GAAG,EAAE,SAAS,GAAG,MAAM,CAAC;IAC5C,iBAAiB,CAAC,GAAG,EAAE,SAAS,GAAG,WAAW,CAAC;IAC/C,eAAe,CAAC,KAAK,EAAE,oBAAoB,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC;IAC/D,eAAe,CACb,KAAK,EAAE;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,IAAI,EAAE,MAAM,CAAC;QAAC,MAAM,CAAC,EAAE,OAAO,GAAG,MAAM,CAAA;KAAE,EAChE,GAAG,EAAE,SAAS,GACb,OAAO,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,OAAO,CAAA;KAAE,CAAC,CAAC;IAC/C,gBAAgB,CAAC,GAAG,EAAE,SAAS,GAAG,OAAO,CAAC,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,WAAW,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,OAAO,CAAA;KAAE,CAAC,CAAC,CAAC;IAC1H,iBAAiB,CAAC,IAAI,EAAE,MAAM,EAAE,GAAG,EAAE,SAAS,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAC/D,uBAAuB,CAAC,KAAK,EAAE,8BAA8B,GAAG,OAAO,CAAC,qBAAqB,CAAC,CAAC;IAC/F,sBAAsB,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,qBAAqB,EAAE,CAAC,CAAC;IACxE,uBAAuB,CAAC,KAAK,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,UAAU,GAAG,OAAO,CAAC,qBAAqB,CAAC,CAAC;IACtG,uBAAuB,CAAC,KAAK,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC;CACtE;AAED,MAAM,WAAW,oBAAoB;IACnC,WAAW,CAAC,WAAW,EAAE,MAAM,EAAE,YAAY,EAAE,MAAM,EAAE,IAAI,EAAE,UAAU,GAAG,OAAO,CAAC,WAAW,CAAC,CAAC;IAC/F,aAAa,CAAC,WAAW,EAAE,MAAM,EAAE,YAAY,EAAE,MAAM,EAAE,IAAI,EAAE,UAAU,EAAE,SAAS,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,SAAS,CAAC,CAAC;IACnH,mBAAmB,CAAC,WAAW,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,EAAE,YAAY,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,SAAS,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,SAAS,CAAC,CAAC;IAC5I,cAAc,CAAC,WAAW,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,EAAE,YAAY,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,WAAW,CAAC,CAAC;IACrH,kBAAkB,CAAC,SAAS,EAAE,MAAM,EAAE,eAAe,CAAC,EAAE,OAAO,GAAG,OAAO,CAAC,UAAU,EAAE,CAAC,CAAC;IACxF,iBAAiB,CAAC,SAAS,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,SAAS,CAAC,CAAC;IAC7E,qBAAqB,CAAC,SAAS,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,EAAE,KAAK,EAAE,OAAO,mBAAmB,EAAE,eAAe,GAAG,OAAO,CAAC,OAAO,YAAY,EAAE,UAAU,CAAC,CAAC;IAC3J,oBAAoB,CAAC,SAAS,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,UAAU,CAAC,CAAC;CAClF;AAED,MAAM,WAAW,uBAAuB;IACtC,gBAAgB,CAAC,KAAK,EAAE,IAAI,CAAC,gBAAgB,EAAE,IAAI,GAAG,WAAW,CAAC,GAAG,OAAO,CAAC,gBAAgB,CAAC,CAAC;IAC/F,uBAAuB,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,gBAAgB,EAAE,CAAC,CAAC;IACpE,mBAAmB,CAAC,GAAG,EAAE,SAAS,EAAE,KAAK,EAAE,WAAW,EAAE,KAAK,CAAC,EAAE,iBAAiB,GAAG,OAAO,CAAC,gBAAgB,EAAE,CAAC,CAAC;IAChH,kBAAkB,CAAC,GAAG,EAAE,SAAS,EAAE,KAAK,EAAE,WAAW,EAAE,KAAK,EAAE,iBAAiB,GAAG,OAAO,CAAC,kBAAkB,CAAC,CAAC;IAC9G,sBAAsB,CACpB,GAAG,EAAE,SAAS,EACd,KAAK,EAAE,WAAW,EAClB,OAAO,EAAE,cAAc,EACvB,KAAK,EAAE,oBAAoB,GAC1B,OAAO,CAAC,qBAAqB,CAAC,CAAC;CACnC;AAED,MAAM,WAAW,iBACf,SAAQ,kBAAkB,EACxB,mBAAmB,EACnB,sBAAsB,EACtB,oBAAoB,EACpB,oBAAoB,EACpB,uBAAuB;IACzB,gBAAgB,CAAC,KAAK,EAAE,MAAM,EAAE,UAAU,EAAE,eAAe,GAAG,MAAM,IAAI,CAAC;CAC1E"}

package/dist/tool-execution.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"tool-execution.d.ts","sourceRoot":"","sources":["../src/tool-execution.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAc,cAAc,EAAE,MAAM,2BAA2B,CAAC;AA+B1F,OAAO,KAAK,EAAE,eAAe,EAAE,oBAAoB,EAAE,MAAM,oBAAoB,CAAC;AAGhF,OAAO,KAAK,EACV,SAAS,EAET,cAAc,EACf,MAAM,YAAY,CAAC;AAEpB,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,0BAA0B,CAAC;AAElE;;;;;;;;;;;GAWG;AAEH,YAAY,EAAE,iBAAiB,EAAE,MAAM,0BAA0B,CAAC;AAElE,wBAAgB,cAAc,CAAC,IAAI,EAAE,iBAAiB,EAAE,OAAO,EAAE,eAAe,GAAG,YAAY,GAAG,SAAS,CAM1G;AAED,wBAAsB,eAAe,CACnC,IAAI,EAAE,iBAAiB,EACvB,GAAG,EAAE,SAAS,EACd,IAAI,EAAE,YAAY,EAClB,KAAK,EAAE,OAAO,EACd,aAAa,EAAE,oBAAoB,EACnC,QAAQ,EAAE,OAAO,EACjB,cAAc,SAAgC,EAC9C,MAAM,CAAC,EAAE,WAAW,GACnB,OAAO,CAAC;IACT,GAAG,EAAE,SAAS,CAAC;IACf,WAAW,CAAC,EAAE,OAAO,CAAC;IACtB,kBAAkB,CAAC,EAAE,OAAO,CAAC;IAC7B,eAAe,CAAC,EAAE,OAAO,CAAC;IAC1B,oBAAoB,CAAC,EAAE,OAAO,CAAC;IAC/B,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,sBAAsB,CAAC,EAAE,MAAM,CAAC;CACjC,CAAC,CAmED;AAED,wBAAsB,yBAAyB,CAC7C,IAAI,EAAE,iBAAiB,EACvB,GAAG,EAAE,SAAS,EACd,IAAI,EAAE,YAAY,EAClB,QAAQ,EAAE,cAAc,EACxB,KAAK,EAAE,OAAO,EACd,aAAa,EAAE,oBAAoB,EACnC,gBAAgB,UAAQ,EACxB,MAAM,CAAC,EAAE,WAAW,GACnB,OAAO,CAAC;IAAE,GAAG,EAAE,SAAS,CAAC;IAAC,WAAW,CAAC,EAAE,OAAO,CAAC;IAAC,eAAe,CAAC,EAAE,OAAO,CAAC;IAAC,UAAU,CAAC,EAAE,MAAM,CAAC;IAAC,QAAQ,CAAC,EAAE,MAAM,CAAA;CAAE,CAAC,CAgLvH;AAsBD,sIAAsI;AACtI,wBAAgB,mBAAmB,CAAC,SAAS,EAAE,MAAM,GAAG,SAAS,EAAE,GAAG,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,SAAS,CAAC,GAAG,MAAM,CAMlH;AA8CD,wBAAgB,eAAe,CAAC,IAAI,EAAE,iBAAiB,EAAE,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,cAAc,GAAG,cAAc,CAiD/G"}
+{"version":3,"file":"tool-execution.d.ts","sourceRoot":"","sources":["../src/tool-execution.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,YAAY,EAAc,cAAc,EAAE,MAAM,2BAA2B,CAAC;AA8B1F,OAAO,KAAK,EAAE,eAAe,EAAE,oBAAoB,EAAE,MAAM,oBAAoB,CAAC;AAGhF,OAAO,KAAK,EACV,SAAS,EAET,cAAc,EACf,MAAM,YAAY,CAAC;AAEpB,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,0BAA0B,CAAC;AAGlE;;;;;;;;;;;GAWG;AAEH,YAAY,EAAE,iBAAiB,EAAE,MAAM,0BAA0B,CAAC;AAElE,wBAAgB,cAAc,CAAC,IAAI,EAAE,iBAAiB,EAAE,OAAO,EAAE,eAAe,GAAG,YAAY,GAAG,SAAS,CAM1G;AAED,wBAAsB,eAAe,CACnC,IAAI,EAAE,iBAAiB,EACvB,GAAG,EAAE,SAAS,EACd,IAAI,EAAE,YAAY,EAClB,KAAK,EAAE,OAAO,EACd,aAAa,EAAE,oBAAoB,EACnC,QAAQ,EAAE,OAAO,EACjB,cAAc,SAAgC,EAC9C,MAAM,CAAC,EAAE,WAAW,GACnB,OAAO,CAAC;IACT,GAAG,EAAE,SAAS,CAAC;IACf,WAAW,CAAC,EAAE,OAAO,CAAC;IACtB,kBAAkB,CAAC,EAAE,OAAO,CAAC;IAC7B,eAAe,CAAC,EAAE,OAAO,CAAC;IAC1B,oBAAoB,CAAC,EAAE,OAAO,CAAC;IAC/B,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,sBAAsB,CAAC,EAAE,MAAM,CAAC;CACjC,CAAC,CAoED;AAED,wBAAsB,yBAAyB,CAC7C,IAAI,EAAE,iBAAiB,EACvB,GAAG,EAAE,SAAS,EACd,IAAI,EAAE,YAAY,EAClB,QAAQ,EAAE,cAAc,EACxB,KAAK,EAAE,OAAO,EACd,aAAa,EAAE,oBAAoB,EACnC,gBAAgB,UAAQ,EACxB,MAAM,CAAC,EAAE,WAAW,GACnB,OAAO,CAAC;IAAE,GAAG,EAAE,SAAS,CAAC;IAAC,WAAW,CAAC,EAAE,OAAO,CAAC;IAAC,eAAe,CAAC,EAAE,OAAO,CAAC;IAAC,UAAU,CAAC,EAAE,MAAM,CAAC;IAAC,QAAQ,CAAC,EAAE,MAAM,CAAA;CAAE,CAAC,CAQvH;AAmND,sIAAsI;AACtI,wBAAgB,mBAAmB,CAAC,SAAS,EAAE,MAAM,GAAG,SAAS,EAAE,GAAG,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,SAAS,CAAC,GAAG,MAAM,CAMlH;AA8CD,wBAAgB,eAAe,CAAC,IAAI,EAAE,iBAAiB,EAAE,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,cAAc,GAAG,cAAc,CAiD/G"}

package/dist/tool-execution.js

@@ -1,4 +1,5 @@
-import { boundModelOutput, normalizeApprovalPolicy, sandboxFilesystemContractMetadata, validateAgainstJsonSchema } from "@assemblyline-agents/core";
+import { checkTaskBudget } from "./task-budget.js";
+import { normalizeApprovalPolicy, sandboxFilesystemContractMetadata, validateAgainstJsonSchema } from "@assemblyline-agents/core";
 import { HISTORY_SEARCH_TOOL_NAME, FILES_MOUNT_TOOL_NAME, FILES_SEARCH_TOOL_NAME, runtimeToolsForSurface, TOOL_SEARCH_TOOL_NAME, WORKSPACE_CHECKPOINT_CREATE_TOOL_NAME, WORKSPACE_CHECKPOINT_LIST_TOOL_NAME, WORKSPACE_CHECKPOINT_RESTORE_TOOL_NAME, WORKSPACE_FORK_TOOL_NAME, WORKSPACE_SEARCH_TOOL_NAME, WORKSPACE_STATUS_TOOL_NAME } from "./builtin-tools.js";
 import { createConnectionRegistryForRunImpl, executeHistorySearchTool, executeToolSearchTool, failRuntimeHandledTool, } from "./builtin-tool-executors.js";
 import { resolvePositiveIntegerEnv } from "./env-config.js";
@@ -9,6 +10,7 @@ import { sandboxWriteByteLength } from "./resource-projection-util.js";
 import { stringValue, timestamp, toolRequiresSandbox } from "./runtime-util.js";
 import { applyCoreToolPolicy } from "./security.js";
 import { createToolContext, HumanInputRequired } from "./tool-apis.js";
+import { READ_TOOL_RESULT_TOOL_NAME, readToolResult, retainToolModelOutput } from "./tool-result-output.js";
 export function selectToolImpl(host, options) {
     if (options.toolName) {
         const surface = options.existingRun ? host.surfaceForRun(options.existingRun) : undefined;
@@ -17,6 +19,7 @@ export function selectToolImpl(host, options) {
     return undefined;
 }
 export async function executeToolImpl(host, run, tool, input, sandboxHandle, approved, idempotencyKey = `tool:${run.id}:${tool.name}`, signal) {
+    await checkTaskBudget(host.state, run.id);
     const policyTool = applyCoreToolPolicy(tool, host.coreToolPolicy);
     if (!policyTool) {
         await host.state.appendEvent(run.id, "tool.call_requested", { toolName: tool.name, input: redactedToolInput(input) });
@@ -84,6 +87,17 @@ export async function executeToolImpl(host, run, tool, input, sandboxHandle, app
     return executeToolWithRecordImpl(host, run, tool, toolCall, input, sandboxHandle, approved, signal);
 }
 export async function executeToolWithRecordImpl(host, run, tool, toolCall, input, sandboxHandle, approvedToolCall = false, signal) {
+    await checkTaskBudget(host.state, run.id);
+    const controller = new AbortController();
+    const unregister = host.registerRunAbort(run.id, controller);
+    try {
+        return await executeToolWithRecordControlled(host, run, tool, toolCall, input, sandboxHandle, approvedToolCall, signal ? AbortSignal.any([signal, controller.signal]) : controller.signal);
+    }
+    finally {
+        unregister();
+    }
+}
+async function executeToolWithRecordControlled(host, run, tool, toolCall, input, sandboxHandle, approvedToolCall = false, signal) {
     const authoredTool = !tool.source.path.startsWith("__builtin__");
     const sandboxedAuthoredTool = authoredTool && host.shouldSandboxAuthoredTool(tool, run);
     await host.state.updateToolCall(toolCall.id, { status: "running" });
@@ -113,17 +127,18 @@ export async function executeToolWithRecordImpl(host, run, tool, toolCall, input
             const ctx = createToolContext(host, run, sandboxHandle, tool, approvedToolCall, sandboxedAuthoredTool);
             if (sandboxedAuthoredTool) {
                 const sandboxed = await host.executeSandboxedTool(tool, input, ctx, sandboxHandle, operationSignal);
-                const modelOutput = sandboxed.hasModelOutput
-                    ? boundModelOutput(sandboxed.modelOutput, host.toolOutputMaxChars)
-                    : boundModelOutput(sandboxed.result, host.toolOutputMaxChars);
+                const modelOutput = await retainToolModelOutput(host, run.id, sandboxed.hasModelOutput ? sandboxed.modelOutput : sandboxed.result);
                 return { result: sandboxed.result, modelOutput };
             }
-            const result = workspaceTool
-                ? await host.executeWorkspaceTool(run, tool.name, input, sandboxHandle)
-                : await Promise.resolve(definition.execute(input, ctx));
-            const modelOutput = definition?.toModelOutput
+            const result = tool.name === READ_TOOL_RESULT_TOOL_NAME
+                ? await readToolResult(host, run.id, input)
+                : workspaceTool
+                    ? await host.executeWorkspaceTool(run, tool.name, input, sandboxHandle)
+                    : await Promise.resolve(definition.execute(input, ctx));
+            const projection = definition?.toModelOutput
                 ? await definition.toModelOutput(result)
-                : boundModelOutput(result, host.toolOutputMaxChars);
+                : result;
+            const modelOutput = await retainToolModelOutput(host, run.id, projection);
             return { result, modelOutput };
         }, definition?.timeoutMs ?? host.toolTimeoutMs, tool.name, signal);
         const { result, modelOutput } = executed;