pi-subagents 0.68.0 → 0.70.0

package/src/runs/shared/single-output.d.ts

@@ -0,0 +1,61 @@
+import type { Message } from "@earendil-works/pi-ai";
+import type { OutputMode, SavedOutputReference } from "../../shared/types.ts";
+export interface SingleOutputSnapshot {
+    exists: boolean;
+    mtimeMs?: number;
+    size?: number;
+    error?: string;
+}
+/**
+ * Content the child itself sent to the configured output path, taken from its
+ * last `write` tool call whose tool result reports success. Unlike reading the
+ * path from disk, this cannot be polluted by a sibling run writing the same
+ * path (#420); requiring the successful tool result keeps failed, cancelled,
+ * or unanswered write calls from counting as authored output. Returns
+ * undefined when no such write exists (e.g. bash or edit-based construction),
+ * in which case callers must not assume file authorship.
+ */
+export declare function extractChildWrittenOutput(messages: Message[] | undefined, outputPath: string | undefined, cwd?: string): string | undefined;
+export declare function normalizeSingleOutputOverride(output: string | boolean | undefined, defaultOutput: string | undefined): string | false | undefined;
+export declare function resolveSingleOutputPath(output: string | boolean | undefined, runtimeCwd: string, requestedCwd?: string, relativeBaseDir?: string): string | undefined;
+interface OutputInstructionCapabilities {
+    tools?: string[];
+    mcpDirectTools?: string[];
+}
+export declare function requestedOutputPathFromTask(task: string): string | undefined;
+export declare function outputPathMappingFromTask(task: string, savedPath: string | undefined): {
+    requestedPath: string;
+    savedPath: string;
+} | undefined;
+export declare function injectSingleOutputInstruction(task: string, outputPath: string | undefined, capabilities?: OutputInstructionCapabilities): string;
+export declare function injectOutputPathSystemPrompt(systemPrompt: string, outputPath: string | undefined, capabilities?: OutputInstructionCapabilities): string;
+export declare function formatSavedOutputReference(savedPath: string, fullOutput: string): SavedOutputReference;
+export declare function validateFileOnlyOutputMode(outputMode: OutputMode | undefined, outputPath: string | undefined, context: string): string | undefined;
+export declare function resolveSingleOutputClaimPath(outputPath: string): string;
+export declare function captureSingleOutputSnapshot(outputPath: string | undefined): SingleOutputSnapshot | undefined;
+export declare function hasSingleOutputChangedSinceSnapshot(outputPath: string | undefined, beforeRun: SingleOutputSnapshot | undefined): boolean | undefined;
+export declare function resolveSingleOutput(outputPath: string | undefined, fallbackOutput: string, beforeRun: SingleOutputSnapshot | undefined, expectedClaimPath?: string): {
+    fullOutput: string;
+    savedPath?: string;
+    saveError?: string;
+    fatalError?: boolean;
+};
+export declare function finalizeSingleOutput(params: {
+    fullOutput: string;
+    truncatedOutput?: string;
+    outputPath?: string;
+    outputMode?: OutputMode;
+    exitCode: number;
+    /** Keep the saved-output reference when a post-run acceptance gate rejects an otherwise completed child. */
+    preserveSavedOutput?: boolean;
+    savedPath?: string;
+    outputReference?: SavedOutputReference;
+    saveError?: string;
+}): {
+    displayOutput: string;
+    savedPath?: string;
+    outputReference?: SavedOutputReference;
+    saveError?: string;
+};
+export {};
+//# sourceMappingURL=single-output.d.ts.map

package/src/runs/shared/single-output.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"single-output.d.ts","sourceRoot":"","sources":["../../../../src/runs/shared/single-output.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,uBAAuB,CAAC;AACrD,OAAO,KAAK,EAAE,UAAU,EAAE,oBAAoB,EAAE,MAAM,uBAAuB,CAAC;AAG9E,MAAM,WAAW,oBAAoB;IACpC,MAAM,EAAE,OAAO,CAAC;IAChB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,KAAK,CAAC,EAAE,MAAM,CAAC;CACf;AAYD;;;;;;;;GAQG;AACH,wBAAgB,yBAAyB,CACxC,QAAQ,EAAE,OAAO,EAAE,GAAG,SAAS,EAC/B,UAAU,EAAE,MAAM,GAAG,SAAS,EAC9B,GAAG,CAAC,EAAE,MAAM,GACV,MAAM,GAAG,SAAS,CA0BpB;AAED,wBAAgB,6BAA6B,CAC5C,MAAM,EAAE,MAAM,GAAG,OAAO,GAAG,SAAS,EACpC,aAAa,EAAE,MAAM,GAAG,SAAS,GAC/B,MAAM,GAAG,KAAK,GAAG,SAAS,CAK5B;AAED,wBAAgB,uBAAuB,CACtC,MAAM,EAAE,MAAM,GAAG,OAAO,GAAG,SAAS,EACpC,UAAU,EAAE,MAAM,EAClB,YAAY,CAAC,EAAE,MAAM,EACrB,eAAe,CAAC,EAAE,MAAM,GACtB,MAAM,GAAG,SAAS,CAQpB;AAED,UAAU,6BAA6B;IACtC,KAAK,CAAC,EAAE,MAAM,EAAE,CAAC;IACjB,cAAc,CAAC,EAAE,MAAM,EAAE,CAAC;CAC1B;AAiBD,wBAAgB,2BAA2B,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM,GAAG,SAAS,CAQ5E;AAED,wBAAgB,yBAAyB,CAAC,IAAI,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,GAAG,SAAS,GAAG;IAAE,aAAa,EAAE,MAAM,CAAC;IAAC,SAAS,EAAE,MAAM,CAAA;CAAE,GAAG,SAAS,CAK/I;AAED,wBAAgB,6BAA6B,CAAC,IAAI,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,GAAG,SAAS,EAAE,YAAY,CAAC,EAAE,6BAA6B,GAAG,MAAM,CAGhJ;AAED,wBAAgB,4BAA4B,CAAC,YAAY,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,GAAG,SAAS,EAAE,YAAY,CAAC,EAAE,6BAA6B,GAAG,MAAM,CAIvJ;AAoBD,wBAAgB,0BAA0B,CAAC,SAAS,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,GAAG,oBAAoB,CAUtG;AAED,wBAAgB,0BAA0B,CAAC,UAAU,EAAE,UAAU,GAAG,SAAS,EAAE,UAAU,EAAE,MAAM,GAAG,SAAS,EAAE,OAAO,EAAE,MAAM,GAAG,MAAM,GAAG,SAAS,CAKlJ;AAED,wBAAgB,4BAA4B,CAAC,UAAU,EAAE,MAAM,GAAG,MAAM,CAUvE;AAOD,wBAAgB,2BAA2B,CAAC,UAAU,EAAE,MAAM,GAAG,SAAS,GAAG,oBAAoB,GAAG,SAAS,CAQ5G;AAkBD,wBAAgB,mCAAmC,CAClD,UAAU,EAAE,MAAM,GAAG,SAAS,EAC9B,SAAS,EAAE,oBAAoB,GAAG,SAAS,GACzC,OAAO,GAAG,SAAS,CAIrB;AAmBD,wBAAgB,mBAAmB,CAClC,UAAU,EAAE,MAAM,GAAG,SAAS,EAC9B,cAAc,EAAE,MAAM,EACtB,SAAS,EAAE,oBAAoB,GAAG,SAAS,EAC3C,iBAAiB,CAAC,EAAE,MAAM,GACxB;IAAE,UAAU,EAAE,MAAM,CAAC;IAAC,SAAS,CAAC,EAAE,MAAM,CAAC;IAAC,SAAS,CAAC,EAAE,MAAM,CAAC;IAAC,UAAU,CAAC,EAAE,OAAO,CAAA;CAAE,CA2BtF;AAED,wBAAgB,oBAAoB,CAAC,MAAM,EAAE;IAC5C,UAAU,EAAE,MAAM,CAAC;IACnB,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,UAAU,CAAC,EAAE,UAAU,CAAC;IACxB,QAAQ,EAAE,MAAM,CAAC;IACjB,4GAA4G;IAC5G,mBAAmB,CAAC,EAAE,OAAO,CAAC;IAC9B,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,eAAe,CAAC,EAAE,oBAAoB,CAAC;IACvC,SAAS,CAAC,EAAE,MAAM,CAAC;CACnB,GAAG;IAAE,aAAa,EAAE,MAAM,CAAC;IAAC,SAAS,CAAC,EAAE,MAAM,CAAC;IAAC,eAAe,CAAC,EAAE,oBAAoB,CAAC;IAAC,SAAS,CAAC,EAAE,MAAM,CAAA;CAAE,CAe5G"}

package/src/runs/shared/single-output.js

@@ -0,0 +1,263 @@
+import * as fs from "node:fs";
+import * as path from "node:path";
+import { hasMutationToolCapability } from "./completion-guard.js";
+function missingFileStatError(error) {
+    const code = error?.code;
+    return code === "ENOENT" || code === "ENOTDIR";
+}
+function snapshotFromStatError(error) {
+    if (missingFileStatError(error))
+        return { exists: false };
+    return { exists: false, error: error instanceof Error ? error.message : String(error) };
+}
+/**
+ * Content the child itself sent to the configured output path, taken from its
+ * last `write` tool call whose tool result reports success. Unlike reading the
+ * path from disk, this cannot be polluted by a sibling run writing the same
+ * path (#420); requiring the successful tool result keeps failed, cancelled,
+ * or unanswered write calls from counting as authored output. Returns
+ * undefined when no such write exists (e.g. bash or edit-based construction),
+ * in which case callers must not assume file authorship.
+ */
+export function extractChildWrittenOutput(messages, outputPath, cwd) {
+    if (!messages?.length || !outputPath)
+        return undefined;
+    const resolvedTarget = path.resolve(cwd ?? ".", outputPath);
+    const comparableTarget = process.platform === "win32" ? resolvedTarget.toLowerCase() : resolvedTarget;
+    const successfulCallIds = new Set();
+    for (const message of messages) {
+        if (message.role === "toolResult" && message.isError === false && typeof message.toolCallId === "string") {
+            successfulCallIds.add(message.toolCallId);
+        }
+    }
+    let content;
+    for (const message of messages) {
+        if (message.role !== "assistant")
+            continue;
+        for (const part of message.content) {
+            if (part.type !== "toolCall" || part.name !== "write" || !successfulCallIds.has(part.id))
+                continue;
+            const args = typeof part.arguments === "object" && part.arguments !== null && !Array.isArray(part.arguments)
+                ? part.arguments
+                : {};
+            if (typeof args.path !== "string" || typeof args.content !== "string")
+                continue;
+            const resolvedWritePath = path.resolve(cwd ?? ".", args.path);
+            const comparableWritePath = process.platform === "win32" ? resolvedWritePath.toLowerCase() : resolvedWritePath;
+            if (comparableWritePath !== comparableTarget)
+                continue;
+            content = args.content;
+        }
+    }
+    return content;
+}
+export function normalizeSingleOutputOverride(output, defaultOutput) {
+    if (output === false || output === "false")
+        return false;
+    if (output === true || output === "true")
+        return defaultOutput;
+    if (typeof output === "string" && output.length > 0)
+        return output;
+    return undefined;
+}
+export function resolveSingleOutputPath(output, runtimeCwd, requestedCwd, relativeBaseDir) {
+    if (typeof output !== "string" || !output || output === "false" || output === "true")
+        return undefined;
+    if (path.isAbsolute(output))
+        return output;
+    if (relativeBaseDir)
+        return path.resolve(relativeBaseDir, output);
+    const baseCwd = requestedCwd
+        ? (path.isAbsolute(requestedCwd) ? requestedCwd : path.resolve(runtimeCwd, requestedCwd))
+        : runtimeCwd;
+    return path.resolve(baseCwd, output);
+}
+function formatOutputPathInstruction(outputPath, capabilities) {
+    const delivery = !capabilities || hasMutationToolCapability(capabilities.tools, capabilities.mcpDirectTools)
+        ? `Write your findings to exactly this path: ${outputPath}`
+        : [
+            "Return the complete artifact in your final response.",
+            `The runtime will persist it to exactly this path: ${outputPath}`,
+            "Do not call contact_supervisor merely because no write-capable tool is available.",
+        ].join("\n");
+    return [
+        delivery,
+        "This path is authoritative for this run.",
+        "Ignore any other output filename or output path mentioned elsewhere, including output destinations in the base agent prompt, system prompt, or task instructions.",
+    ].join("\n");
+}
+export function requestedOutputPathFromTask(task) {
+    for (const line of task.split(/\r?\n/).reverse()) {
+        const match = line.match(/^\s*(?:Write your findings to(?: exactly this path)?:|The runtime will persist it to exactly this path:)\s*(.+?)\s*$/i);
+        if (!match?.[1])
+            continue;
+        const requested = match[1].trim();
+        return requested.startsWith("`") && requested.endsWith("`") ? requested.slice(1, -1) : requested;
+    }
+    return undefined;
+}
+export function outputPathMappingFromTask(task, savedPath) {
+    const requestedPath = requestedOutputPathFromTask(task);
+    if (!requestedPath || !savedPath)
+        return undefined;
+    if (path.isAbsolute(requestedPath) && path.normalize(requestedPath) === path.normalize(savedPath))
+        return undefined;
+    return { requestedPath, savedPath };
+}
+export function injectSingleOutputInstruction(task, outputPath, capabilities) {
+    if (!outputPath)
+        return task;
+    return `${task}\n\n---\n**Output:**\n${formatOutputPathInstruction(outputPath, capabilities)}`;
+}
+export function injectOutputPathSystemPrompt(systemPrompt, outputPath, capabilities) {
+    if (!outputPath)
+        return systemPrompt;
+    const instruction = `Runtime output path override:\n${formatOutputPathInstruction(outputPath, capabilities)}`;
+    return systemPrompt ? `${systemPrompt}\n\n${instruction}` : instruction;
+}
+function countLines(text) {
+    if (!text)
+        return 0;
+    const newlineMatches = text.match(/\r\n|\r|\n/g);
+    return (newlineMatches?.length ?? 0) + (/[\r\n]$/.test(text) ? 0 : 1);
+}
+function formatByteSize(bytes) {
+    if (bytes < 1024)
+        return `${bytes} B`;
+    const units = ["KB", "MB", "GB", "TB"];
+    let value = bytes / 1024;
+    let unitIndex = 0;
+    while (value >= 1024 && unitIndex < units.length - 1) {
+        value /= 1024;
+        unitIndex++;
+    }
+    return `${value.toFixed(1)} ${units[unitIndex]}`;
+}
+export function formatSavedOutputReference(savedPath, fullOutput) {
+    const absolutePath = path.resolve(savedPath);
+    const bytes = Buffer.byteLength(fullOutput, "utf-8");
+    const lines = countLines(fullOutput);
+    return {
+        path: absolutePath,
+        bytes,
+        lines,
+        message: `Output saved to: ${absolutePath} (${formatByteSize(bytes)}, ${lines} ${lines === 1 ? "line" : "lines"}). Read this file if needed.`,
+    };
+}
+export function validateFileOnlyOutputMode(outputMode, outputPath, context) {
+    if (outputMode === "file-only" && !outputPath) {
+        return `${context} sets outputMode: "file-only" but does not configure an output file. Set output to a path or use outputMode: "inline".`;
+    }
+    return undefined;
+}
+export function resolveSingleOutputClaimPath(outputPath) {
+    let existing = path.resolve(outputPath);
+    const missingSegments = [];
+    while (!fs.existsSync(existing)) {
+        missingSegments.unshift(path.basename(existing));
+        const parent = path.dirname(existing);
+        if (parent === existing)
+            break;
+        existing = parent;
+    }
+    return path.join(fs.realpathSync(existing), ...missingSegments);
+}
+function outputClaimError(outputPath, expectedClaimPath) {
+    if (expectedClaimPath && resolveSingleOutputClaimPath(outputPath) !== expectedClaimPath)
+        return "Output path changed after it was claimed.";
+    return undefined;
+}
+export function captureSingleOutputSnapshot(outputPath) {
+    if (!outputPath)
+        return undefined;
+    try {
+        const stat = fs.statSync(outputPath);
+        return { exists: true, mtimeMs: stat.mtimeMs, size: stat.size };
+    }
+    catch (error) {
+        return snapshotFromStatError(error);
+    }
+}
+function inspectSingleOutputChange(outputPath, beforeRun) {
+    if (beforeRun?.error)
+        return { changed: false, error: beforeRun.error };
+    try {
+        const stat = fs.statSync(outputPath);
+        return {
+            changed: !beforeRun?.exists || stat.mtimeMs !== beforeRun.mtimeMs || stat.size !== beforeRun.size,
+        };
+    }
+    catch (error) {
+        if (missingFileStatError(error))
+            return { changed: false };
+        return { changed: false, error: error instanceof Error ? error.message : String(error) };
+    }
+}
+export function hasSingleOutputChangedSinceSnapshot(outputPath, beforeRun) {
+    if (!outputPath)
+        return undefined;
+    const inspected = inspectSingleOutputChange(outputPath, beforeRun);
+    return inspected.error ? undefined : inspected.changed;
+}
+function persistSingleOutput(outputPath, fullOutput, expectedClaimPath) {
+    if (!outputPath)
+        return {};
+    try {
+        const claimError = outputClaimError(outputPath, expectedClaimPath);
+        if (claimError)
+            return { error: claimError, fatalError: true };
+        fs.mkdirSync(path.dirname(outputPath), { recursive: true });
+        fs.writeFileSync(outputPath, fullOutput, "utf-8");
+        return { savedPath: outputPath };
+    }
+    catch (err) {
+        return { error: err instanceof Error ? err.message : String(err) };
+    }
+}
+export function resolveSingleOutput(outputPath, fallbackOutput, beforeRun, expectedClaimPath) {
+    if (!outputPath)
+        return { fullOutput: fallbackOutput };
+    const claimError = outputClaimError(outputPath, expectedClaimPath);
+    if (claimError)
+        return { fullOutput: fallbackOutput, saveError: claimError, fatalError: true };
+    const changedSinceStart = inspectSingleOutputChange(outputPath, beforeRun);
+    if (changedSinceStart.error) {
+        return {
+            fullOutput: fallbackOutput,
+            saveError: `Failed to inspect output file: ${changedSinceStart.error}`,
+        };
+    }
+    if (changedSinceStart.changed) {
+        try {
+            return { fullOutput: fs.readFileSync(outputPath, "utf-8"), savedPath: outputPath };
+        }
+        catch (error) {
+            return {
+                fullOutput: fallbackOutput,
+                saveError: `Failed to read changed output file: ${error instanceof Error ? error.message : String(error)}`,
+            };
+        }
+    }
+    const save = persistSingleOutput(outputPath, fallbackOutput, expectedClaimPath);
+    if (save.savedPath)
+        return { fullOutput: fallbackOutput, savedPath: save.savedPath };
+    return { fullOutput: fallbackOutput, saveError: save.error, ...(save.fatalError ? { fatalError: true } : {}) };
+}
+export function finalizeSingleOutput(params) {
+    let displayOutput = params.truncatedOutput || params.fullOutput;
+    if ((params.exitCode === 0 || params.preserveSavedOutput) && params.savedPath) {
+        const outputReference = params.outputReference ?? formatSavedOutputReference(params.savedPath, params.fullOutput);
+        if (params.outputMode === "file-only") {
+            return { displayOutput: outputReference.message, savedPath: params.savedPath, outputReference };
+        }
+        if (!displayOutput.includes(outputReference.message))
+            displayOutput += `\n\n${outputReference.message}`;
+        return { displayOutput, savedPath: params.savedPath, outputReference };
+    }
+    if (params.exitCode === 0 && params.saveError && params.outputPath) {
+        displayOutput += `\n\nOutput file error: ${params.outputPath}\n${params.saveError}`;
+        return { displayOutput, saveError: params.saveError };
+    }
+    return { displayOutput };
+}
+//# sourceMappingURL=single-output.js.map

package/src/runs/shared/single-output.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"single-output.js","sourceRoot":"","sources":["../../../../src/runs/shared/single-output.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,MAAM,SAAS,CAAC;AAC9B,OAAO,KAAK,IAAI,MAAM,WAAW,CAAC;AAGlC,OAAO,EAAE,yBAAyB,EAAE,MAAM,uBAAuB,CAAC;AASlE,SAAS,oBAAoB,CAAC,KAAc;IAC3C,MAAM,IAAI,GAAI,KAA2C,EAAE,IAAI,CAAC;IAChE,OAAO,IAAI,KAAK,QAAQ,IAAI,IAAI,KAAK,SAAS,CAAC;AAChD,CAAC;AAED,SAAS,qBAAqB,CAAC,KAAc;IAC5C,IAAI,oBAAoB,CAAC,KAAK,CAAC;QAAE,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC;IAC1D,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC;AACzF,CAAC;AAED;;;;;;;;GAQG;AACH,MAAM,UAAU,yBAAyB,CACxC,QAA+B,EAC/B,UAA8B,EAC9B,GAAY;IAEZ,IAAI,CAAC,QAAQ,EAAE,MAAM,IAAI,CAAC,UAAU;QAAE,OAAO,SAAS,CAAC;IACvD,MAAM,cAAc,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,IAAI,GAAG,EAAE,UAAU,CAAC,CAAC;IAC5D,MAAM,gBAAgB,GAAG,OAAO,CAAC,QAAQ,KAAK,OAAO,CAAC,CAAC,CAAC,cAAc,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,cAAc,CAAC;IACtG,MAAM,iBAAiB,GAAG,IAAI,GAAG,EAAU,CAAC;IAC5C,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;QAChC,IAAI,OAAO,CAAC,IAAI,KAAK,YAAY,IAAI,OAAO,CAAC,OAAO,KAAK,KAAK,IAAI,OAAO,OAAO,CAAC,UAAU,KAAK,QAAQ,EAAE,CAAC;YAC1G,iBAAiB,CAAC,GAAG,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;QAC3C,CAAC;IACF,CAAC;IACD,IAAI,OAA2B,CAAC;IAChC,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;QAChC,IAAI,OAAO,CAAC,IAAI,KAAK,WAAW;YAAE,SAAS;QAC3C,KAAK,MAAM,IAAI,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;YACpC,IAAI,IAAI,CAAC,IAAI,KAAK,UAAU,IAAI,IAAI,CAAC,IAAI,KAAK,OAAO,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC;gBAAE,SAAS;YACnG,MAAM,IAAI,GAAG,OAAO,IAAI,CAAC,SAAS,KAAK,QAAQ,IAAI,IAAI,CAAC,SAAS,KAAK,IAAI,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC;gBAC3G,CAAC,CAAC,IAAI,CAAC,SAAoC;gBAC3C,CAAC,CAAC,EAAE,CAAC;YACN,IAAI,OAAO,IAAI,CAAC,IAAI,KAAK,QAAQ,IAAI,OAAO,IAAI,CAAC,OAAO,KAAK,QAAQ;gBAAE,SAAS;YAChF,MAAM,iBAAiB,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,IAAI,GAAG,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC;YAC9D,MAAM,mBAAmB,GAAG,OAAO,CAAC,QAAQ,KAAK,OAAO,CAAC,CAAC,CAAC,iBAAiB,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,iBAAiB,CAAC;YAC/G,IAAI,mBAAmB,KAAK,gBAAgB;gBAAE,SAAS;YACvD,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC;QACxB,CAAC;IACF,CAAC;IACD,OAAO,OAAO,CAAC;AAChB,CAAC;AAED,MAAM,UAAU,6BAA6B,CAC5C,MAAoC,EACpC,aAAiC;IAEjC,IAAI,MAAM,KAAK,KAAK,IAAI,MAAM,KAAK,OAAO;QAAE,OAAO,KAAK,CAAC;IACzD,IAAI,MAAM,KAAK,IAAI,IAAI,MAAM,KAAK,MAAM;QAAE,OAAO,aAAa,CAAC;IAC/D,IAAI,OAAO,MAAM,KAAK,QAAQ,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC;QAAE,OAAO,MAAM,CAAC;IACnE,OAAO,SAAS,CAAC;AAClB,CAAC;AAED,MAAM,UAAU,uBAAuB,CACtC,MAAoC,EACpC,UAAkB,EAClB,YAAqB,EACrB,eAAwB;IAExB,IAAI,OAAO,MAAM,KAAK,QAAQ,IAAI,CAAC,MAAM,IAAI,MAAM,KAAK,OAAO,IAAI,MAAM,KAAK,MAAM;QAAE,OAAO,SAAS,CAAC;IACvG,IAAI,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC;QAAE,OAAO,MAAM,CAAC;IAC3C,IAAI,eAAe;QAAE,OAAO,IAAI,CAAC,OAAO,CAAC,eAAe,EAAE,MAAM,CAAC,CAAC;IAClE,MAAM,OAAO,GAAG,YAAY;QAC3B,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,UAAU,EAAE,YAAY,CAAC,CAAC;QACzF,CAAC,CAAC,UAAU,CAAC;IACd,OAAO,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;AACtC,CAAC;AAOD,SAAS,2BAA2B,CAAC,UAAkB,EAAE,YAA4C;IACpG,MAAM,QAAQ,GAAG,CAAC,YAAY,IAAI,yBAAyB,CAAC,YAAY,CAAC,KAAK,EAAE,YAAY,CAAC,cAAc,CAAC;QAC3G,CAAC,CAAC,6CAA6C,UAAU,EAAE;QAC3D,CAAC,CAAC;YACD,sDAAsD;YACtD,qDAAqD,UAAU,EAAE;YACjE,mFAAmF;SACnF,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACd,OAAO;QACN,QAAQ;QACR,0CAA0C;QAC1C,mKAAmK;KACnK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AACd,CAAC;AAED,MAAM,UAAU,2BAA2B,CAAC,IAAY;IACvD,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC;QAClD,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,uHAAuH,CAAC,CAAC;QAClJ,IAAI,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;YAAE,SAAS;QAC1B,MAAM,SAAS,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;QAClC,OAAO,SAAS,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,SAAS,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IAClG,CAAC;IACD,OAAO,SAAS,CAAC;AAClB,CAAC;AAED,MAAM,UAAU,yBAAyB,CAAC,IAAY,EAAE,SAA6B;IACpF,MAAM,aAAa,GAAG,2BAA2B,CAAC,IAAI,CAAC,CAAC;IACxD,IAAI,CAAC,aAAa,IAAI,CAAC,SAAS;QAAE,OAAO,SAAS,CAAC;IACnD,IAAI,IAAI,CAAC,UAAU,CAAC,aAAa,CAAC,IAAI,IAAI,CAAC,SAAS,CAAC,aAAa,CAAC,KAAK,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC;QAAE,OAAO,SAAS,CAAC;IACpH,OAAO,EAAE,aAAa,EAAE,SAAS,EAAE,CAAC;AACrC,CAAC;AAED,MAAM,UAAU,6BAA6B,CAAC,IAAY,EAAE,UAA8B,EAAE,YAA4C;IACvI,IAAI,CAAC,UAAU;QAAE,OAAO,IAAI,CAAC;IAC7B,OAAO,GAAG,IAAI,yBAAyB,2BAA2B,CAAC,UAAU,EAAE,YAAY,CAAC,EAAE,CAAC;AAChG,CAAC;AAED,MAAM,UAAU,4BAA4B,CAAC,YAAoB,EAAE,UAA8B,EAAE,YAA4C;IAC9I,IAAI,CAAC,UAAU;QAAE,OAAO,YAAY,CAAC;IACrC,MAAM,WAAW,GAAG,kCAAkC,2BAA2B,CAAC,UAAU,EAAE,YAAY,CAAC,EAAE,CAAC;IAC9G,OAAO,YAAY,CAAC,CAAC,CAAC,GAAG,YAAY,OAAO,WAAW,EAAE,CAAC,CAAC,CAAC,WAAW,CAAC;AACzE,CAAC;AAED,SAAS,UAAU,CAAC,IAAY;IAC/B,IAAI,CAAC,IAAI;QAAE,OAAO,CAAC,CAAC;IACpB,MAAM,cAAc,GAAG,IAAI,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC;IACjD,OAAO,CAAC,cAAc,EAAE,MAAM,IAAI,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;AACvE,CAAC;AAED,SAAS,cAAc,CAAC,KAAa;IACpC,IAAI,KAAK,GAAG,IAAI;QAAE,OAAO,GAAG,KAAK,IAAI,CAAC;IACtC,MAAM,KAAK,GAAG,CAAC,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC;IACvC,IAAI,KAAK,GAAG,KAAK,GAAG,IAAI,CAAC;IACzB,IAAI,SAAS,GAAG,CAAC,CAAC;IAClB,OAAO,KAAK,IAAI,IAAI,IAAI,SAAS,GAAG,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACtD,KAAK,IAAI,IAAI,CAAC;QACd,SAAS,EAAE,CAAC;IACb,CAAC;IACD,OAAO,GAAG,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,SAAS,CAAC,EAAE,CAAC;AAClD,CAAC;AAED,MAAM,UAAU,0BAA0B,CAAC,SAAiB,EAAE,UAAkB;IAC/E,MAAM,YAAY,GAAG,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;IAC7C,MAAM,KAAK,GAAG,MAAM,CAAC,UAAU,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;IACrD,MAAM,KAAK,GAAG,UAAU,CAAC,UAAU,CAAC,CAAC;IACrC,OAAO;QACN,IAAI,EAAE,YAAY;QAClB,KAAK;QACL,KAAK;QACL,OAAO,EAAE,oBAAoB,YAAY,KAAK,cAAc,CAAC,KAAK,CAAC,KAAK,KAAK,IAAI,KAAK,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,8BAA8B;KAC7I,CAAC;AACH,CAAC;AAED,MAAM,UAAU,0BAA0B,CAAC,UAAkC,EAAE,UAA8B,EAAE,OAAe;IAC7H,IAAI,UAAU,KAAK,WAAW,IAAI,CAAC,UAAU,EAAE,CAAC;QAC/C,OAAO,GAAG,OAAO,wHAAwH,CAAC;IAC3I,CAAC;IACD,OAAO,SAAS,CAAC;AAClB,CAAC;AAED,MAAM,UAAU,4BAA4B,CAAC,UAAkB;IAC9D,IAAI,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;IACxC,MAAM,eAAe,GAAa,EAAE,CAAC;IACrC,OAAO,CAAC,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;QACjC,eAAe,CAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC;QACjD,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QACtC,IAAI,MAAM,KAAK,QAAQ;YAAE,MAAM;QAC/B,QAAQ,GAAG,MAAM,CAAC;IACnB,CAAC;IACD,OAAO,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,YAAY,CAAC,QAAQ,CAAC,EAAE,GAAG,eAAe,CAAC,CAAC;AACjE,CAAC;AAED,SAAS,gBAAgB,CAAC,UAAkB,EAAE,iBAAqC;IAClF,IAAI,iBAAiB,IAAI,4BAA4B,CAAC,UAAU,CAAC,KAAK,iBAAiB;QAAE,OAAO,2CAA2C,CAAC;IAC5I,OAAO,SAAS,CAAC;AAClB,CAAC;AAED,MAAM,UAAU,2BAA2B,CAAC,UAA8B;IACzE,IAAI,CAAC,UAAU;QAAE,OAAO,SAAS,CAAC;IAClC,IAAI,CAAC;QACJ,MAAM,IAAI,GAAG,EAAE,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC;QACrC,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC;IACjE,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QAChB,OAAO,qBAAqB,CAAC,KAAK,CAAC,CAAC;IACrC,CAAC;AACF,CAAC;AAED,SAAS,yBAAyB,CACjC,UAAkB,EAClB,SAA2C;IAE3C,IAAI,SAAS,EAAE,KAAK;QAAE,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,SAAS,CAAC,KAAK,EAAE,CAAC;IACxE,IAAI,CAAC;QACJ,MAAM,IAAI,GAAG,EAAE,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC;QACrC,OAAO;YACN,OAAO,EAAE,CAAC,SAAS,EAAE,MAAM,IAAI,IAAI,CAAC,OAAO,KAAK,SAAS,CAAC,OAAO,IAAI,IAAI,CAAC,IAAI,KAAK,SAAS,CAAC,IAAI;SACjG,CAAC;IACH,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QAChB,IAAI,oBAAoB,CAAC,KAAK,CAAC;YAAE,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC;QAC3D,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC;IAC1F,CAAC;AACF,CAAC;AAED,MAAM,UAAU,mCAAmC,CAClD,UAA8B,EAC9B,SAA2C;IAE3C,IAAI,CAAC,UAAU;QAAE,OAAO,SAAS,CAAC;IAClC,MAAM,SAAS,GAAG,yBAAyB,CAAC,UAAU,EAAE,SAAS,CAAC,CAAC;IACnE,OAAO,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,OAAO,CAAC;AACxD,CAAC;AAED,SAAS,mBAAmB,CAC3B,UAA8B,EAC9B,UAAkB,EAClB,iBAA0B;IAE1B,IAAI,CAAC,UAAU;QAAE,OAAO,EAAE,CAAC;IAC3B,IAAI,CAAC;QACJ,MAAM,UAAU,GAAG,gBAAgB,CAAC,UAAU,EAAE,iBAAiB,CAAC,CAAC;QACnE,IAAI,UAAU;YAAE,OAAO,EAAE,KAAK,EAAE,UAAU,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC;QAC/D,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QAC5D,EAAE,CAAC,aAAa,CAAC,UAAU,EAAE,UAAU,EAAE,OAAO,CAAC,CAAC;QAClD,OAAO,EAAE,SAAS,EAAE,UAAU,EAAE,CAAC;IAClC,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACd,OAAO,EAAE,KAAK,EAAE,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC;IACpE,CAAC;AACF,CAAC;AAED,MAAM,UAAU,mBAAmB,CAClC,UAA8B,EAC9B,cAAsB,EACtB,SAA2C,EAC3C,iBAA0B;IAE1B,IAAI,CAAC,UAAU;QAAE,OAAO,EAAE,UAAU,EAAE,cAAc,EAAE,CAAC;IACvD,MAAM,UAAU,GAAG,gBAAgB,CAAC,UAAU,EAAE,iBAAiB,CAAC,CAAC;IACnE,IAAI,UAAU;QAAE,OAAO,EAAE,UAAU,EAAE,cAAc,EAAE,SAAS,EAAE,UAAU,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC;IAE/F,MAAM,iBAAiB,GAAG,yBAAyB,CAAC,UAAU,EAAE,SAAS,CAAC,CAAC;IAC3E,IAAI,iBAAiB,CAAC,KAAK,EAAE,CAAC;QAC7B,OAAO;YACN,UAAU,EAAE,cAAc;YAC1B,SAAS,EAAE,kCAAkC,iBAAiB,CAAC,KAAK,EAAE;SACtE,CAAC;IACH,CAAC;IAED,IAAI,iBAAiB,CAAC,OAAO,EAAE,CAAC;QAC/B,IAAI,CAAC;YACJ,OAAO,EAAE,UAAU,EAAE,EAAE,CAAC,YAAY,CAAC,UAAU,EAAE,OAAO,CAAC,EAAE,SAAS,EAAE,UAAU,EAAE,CAAC;QACpF,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAChB,OAAO;gBACN,UAAU,EAAE,cAAc;gBAC1B,SAAS,EAAE,uCAAuC,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE;aAC1G,CAAC;QACH,CAAC;IACF,CAAC;IAED,MAAM,IAAI,GAAG,mBAAmB,CAAC,UAAU,EAAE,cAAc,EAAE,iBAAiB,CAAC,CAAC;IAChF,IAAI,IAAI,CAAC,SAAS;QAAE,OAAO,EAAE,UAAU,EAAE,cAAc,EAAE,SAAS,EAAE,IAAI,CAAC,SAAS,EAAE,CAAC;IACrF,OAAO,EAAE,UAAU,EAAE,cAAc,EAAE,SAAS,EAAE,IAAI,CAAC,KAAK,EAAE,GAAG,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC;AAChH,CAAC;AAED,MAAM,UAAU,oBAAoB,CAAC,MAWpC;IACA,IAAI,aAAa,GAAG,MAAM,CAAC,eAAe,IAAI,MAAM,CAAC,UAAU,CAAC;IAChE,IAAI,CAAC,MAAM,CAAC,QAAQ,KAAK,CAAC,IAAI,MAAM,CAAC,mBAAmB,CAAC,IAAI,MAAM,CAAC,SAAS,EAAE,CAAC;QAC/E,MAAM,eAAe,GAAG,MAAM,CAAC,eAAe,IAAI,0BAA0B,CAAC,MAAM,CAAC,SAAS,EAAE,MAAM,CAAC,UAAU,CAAC,CAAC;QAClH,IAAI,MAAM,CAAC,UAAU,KAAK,WAAW,EAAE,CAAC;YACvC,OAAO,EAAE,aAAa,EAAE,eAAe,CAAC,OAAO,EAAE,SAAS,EAAE,MAAM,CAAC,SAAS,EAAE,eAAe,EAAE,CAAC;QACjG,CAAC;QACD,IAAI,CAAC,aAAa,CAAC,QAAQ,CAAC,eAAe,CAAC,OAAO,CAAC;YAAE,aAAa,IAAI,OAAO,eAAe,CAAC,OAAO,EAAE,CAAC;QACxG,OAAO,EAAE,aAAa,EAAE,SAAS,EAAE,MAAM,CAAC,SAAS,EAAE,eAAe,EAAE,CAAC;IACxE,CAAC;IACD,IAAI,MAAM,CAAC,QAAQ,KAAK,CAAC,IAAI,MAAM,CAAC,SAAS,IAAI,MAAM,CAAC,UAAU,EAAE,CAAC;QACpE,aAAa,IAAI,0BAA0B,MAAM,CAAC,UAAU,KAAK,MAAM,CAAC,SAAS,EAAE,CAAC;QACpF,OAAO,EAAE,aAAa,EAAE,SAAS,EAAE,MAAM,CAAC,SAAS,EAAE,CAAC;IACvD,CAAC;IACD,OAAO,EAAE,aAAa,EAAE,CAAC;AAC1B,CAAC"}

package/src/runs/shared/spawn-budget.d.ts

@@ -0,0 +1,21 @@
+import { type ExtensionConfig, type SpawnBudgetSnapshot, type SubagentState } from "../../shared/types.ts";
+export declare function getSpawnBudgetSnapshot(state: SubagentState, config: ExtensionConfig, sessionId?: string | null): SpawnBudgetSnapshot;
+export declare function formatSpawnBudgetSummary(snapshot: SpawnBudgetSnapshot): string;
+export declare function formatSpawnBudget(snapshot: SpawnBudgetSnapshot): string;
+export declare function preflightSpawnBudget(state: SubagentState, config: ExtensionConfig, sessionId: string | null, requested: number): {
+    snapshot: SpawnBudgetSnapshot;
+    error?: string;
+};
+export declare function reserveSpawnBudget(state: SubagentState, config: ExtensionConfig, sessionId: string | null, requested: number): {
+    snapshot: SpawnBudgetSnapshot;
+    error?: string;
+};
+export declare function preflightSpawnBudgetGrant(state: SubagentState, config: ExtensionConfig, sessionId: string, additional: number): {
+    snapshot: SpawnBudgetSnapshot;
+    error?: string;
+};
+export declare function grantSpawnBudget(state: SubagentState, config: ExtensionConfig, sessionId: string, additional: number, now?: number): {
+    snapshot: SpawnBudgetSnapshot;
+    error?: string;
+};
+//# sourceMappingURL=spawn-budget.d.ts.map

package/src/runs/shared/spawn-budget.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"spawn-budget.d.ts","sourceRoot":"","sources":["../../../../src/runs/shared/spawn-budget.ts"],"names":[],"mappings":"AAAA,OAAO,EAEN,KAAK,eAAe,EACpB,KAAK,mBAAmB,EACxB,KAAK,aAAa,EAClB,MAAM,uBAAuB,CAAC;AAwB/B,wBAAgB,sBAAsB,CACrC,KAAK,EAAE,aAAa,EACpB,MAAM,EAAE,eAAe,EACvB,SAAS,GAAE,MAAM,GAAG,IAA6B,GAC/C,mBAAmB,CAcrB;AAED,wBAAgB,wBAAwB,CAAC,QAAQ,EAAE,mBAAmB,GAAG,MAAM,CAG9E;AAED,wBAAgB,iBAAiB,CAAC,QAAQ,EAAE,mBAAmB,GAAG,MAAM,CAEvE;AAED,wBAAgB,oBAAoB,CACnC,KAAK,EAAE,aAAa,EACpB,MAAM,EAAE,eAAe,EACvB,SAAS,EAAE,MAAM,GAAG,IAAI,EACxB,SAAS,EAAE,MAAM,GACf;IAAE,QAAQ,EAAE,mBAAmB,CAAC;IAAC,KAAK,CAAC,EAAE,MAAM,CAAA;CAAE,CAOnD;AAED,wBAAgB,kBAAkB,CACjC,KAAK,EAAE,aAAa,EACpB,MAAM,EAAE,eAAe,EACvB,SAAS,EAAE,MAAM,GAAG,IAAI,EACxB,SAAS,EAAE,MAAM,GACf;IAAE,QAAQ,EAAE,mBAAmB,CAAC;IAAC,KAAK,CAAC,EAAE,MAAM,CAAA;CAAE,CAKnD;AAED,wBAAgB,yBAAyB,CACxC,KAAK,EAAE,aAAa,EACpB,MAAM,EAAE,eAAe,EACvB,SAAS,EAAE,MAAM,EACjB,UAAU,EAAE,MAAM,GAChB;IAAE,QAAQ,EAAE,mBAAmB,CAAC;IAAC,KAAK,CAAC,EAAE,MAAM,CAAA;CAAE,CAenD;AAED,wBAAgB,gBAAgB,CAC/B,KAAK,EAAE,aAAa,EACpB,MAAM,EAAE,eAAe,EACvB,SAAS,EAAE,MAAM,EACjB,UAAU,EAAE,MAAM,EAClB,GAAG,SAAa,GACd;IAAE,QAAQ,EAAE,mBAAmB,CAAC;IAAC,KAAK,CAAC,EAAE,MAAM,CAAA;CAAE,CAenD"}

package/src/runs/shared/spawn-budget.js

@@ -0,0 +1,94 @@
+import { resolveMaxSubagentSpawnsPerSession, } from "../../shared/types.js";
+const MAX_GRANT_HISTORY = 20;
+function sessionState(state, config, sessionId) {
+    let counters = state.subagentSpawns;
+    if (!counters || counters.sessionId !== sessionId) {
+        counters = {
+            sessionId,
+            count: 0,
+            configuredLimit: resolveMaxSubagentSpawnsPerSession(config.maxSubagentSpawnsPerSession) ?? null,
+            granted: 0,
+            grantHistory: [],
+        };
+        state.subagentSpawns = counters;
+    }
+    if (counters.configuredLimit === undefined) {
+        counters.configuredLimit = resolveMaxSubagentSpawnsPerSession(config.maxSubagentSpawnsPerSession) ?? null;
+    }
+    counters.granted ??= 0;
+    counters.grantHistory ??= [];
+    return counters;
+}
+export function getSpawnBudgetSnapshot(state, config, sessionId = state.currentSessionId) {
+    const counters = sessionState(state, config, sessionId);
+    const configuredLimit = counters.configuredLimit ?? null;
+    const granted = configuredLimit === null ? 0 : counters.granted ?? 0;
+    const limit = configuredLimit === null ? null : configuredLimit + granted;
+    return {
+        used: counters.count,
+        configuredLimit,
+        granted,
+        limit,
+        remaining: limit === null ? null : Math.max(0, limit - counters.count),
+        grantRemaining: configuredLimit === null ? null : Math.max(0, configuredLimit - granted),
+        grantHistory: [...(counters.grantHistory ?? [])],
+    };
+}
+export function formatSpawnBudgetSummary(snapshot) {
+    if (snapshot.limit === null)
+        return "unlimited";
+    return `${snapshot.used}/${snapshot.limit} used, ${snapshot.remaining} remaining (configured ${snapshot.configuredLimit}; granted ${snapshot.granted}; grant allowance ${snapshot.grantRemaining})`;
+}
+export function formatSpawnBudget(snapshot) {
+    return `Spawn budget: ${formatSpawnBudgetSummary(snapshot)}`;
+}
+export function preflightSpawnBudget(state, config, sessionId, requested) {
+    const snapshot = getSpawnBudgetSnapshot(state, config, sessionId);
+    if (requested <= 0 || snapshot.limit === null || requested <= (snapshot.remaining ?? 0))
+        return { snapshot };
+    return {
+        snapshot,
+        error: `Subagent spawn limit reached for this session (${snapshot.used}/${snapshot.limit} used, ${requested} requested). ${snapshot.remaining} remaining; the declared run cannot fit, so no children were started. Grant budget explicitly from the root interactive session or start a new session.`,
+    };
+}
+export function reserveSpawnBudget(state, config, sessionId, requested) {
+    const checked = preflightSpawnBudget(state, config, sessionId, requested);
+    if (checked.error || requested <= 0 || checked.snapshot.limit === null)
+        return checked;
+    state.subagentSpawns.count += requested;
+    return { snapshot: getSpawnBudgetSnapshot(state, config, sessionId) };
+}
+export function preflightSpawnBudgetGrant(state, config, sessionId, additional) {
+    const snapshot = getSpawnBudgetSnapshot(state, config, sessionId);
+    if (!Number.isInteger(additional) || additional <= 0) {
+        return { snapshot, error: "action='grant-spawn-budget' requires additional to be a positive integer." };
+    }
+    if (snapshot.configuredLimit === null || snapshot.limit === null) {
+        return { snapshot, error: "The current session has no configured spawn cap, so it does not need a budget grant." };
+    }
+    if (additional > (snapshot.grantRemaining ?? 0)) {
+        return {
+            snapshot,
+            error: `Spawn budget grant rejected: ${additional} requested but only ${snapshot.grantRemaining} of the original configured limit remains grantable.`,
+        };
+    }
+    return { snapshot };
+}
+export function grantSpawnBudget(state, config, sessionId, additional, now = Date.now()) {
+    const checked = preflightSpawnBudgetGrant(state, config, sessionId, additional);
+    if (checked.error)
+        return checked;
+    const before = checked.snapshot;
+    if (before.limit === null) {
+        return { snapshot: before, error: "The current session has no configured spawn cap, so it does not need a budget grant." };
+    }
+    const counters = state.subagentSpawns;
+    counters.granted = (counters.granted ?? 0) + additional;
+    const limit = before.limit + additional;
+    counters.grantHistory = [
+        ...(counters.grantHistory ?? []),
+        { sessionId, amount: additional, grantedAt: now, previousLimit: before.limit, limit },
+    ].slice(-MAX_GRANT_HISTORY);
+    return { snapshot: getSpawnBudgetSnapshot(state, config, sessionId) };
+}
+//# sourceMappingURL=spawn-budget.js.map

package/src/runs/shared/spawn-budget.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"spawn-budget.js","sourceRoot":"","sources":["../../../../src/runs/shared/spawn-budget.ts"],"names":[],"mappings":"AAAA,OAAO,EACN,kCAAkC,GAIlC,MAAM,uBAAuB,CAAC;AAE/B,MAAM,iBAAiB,GAAG,EAAE,CAAC;AAE7B,SAAS,YAAY,CAAC,KAAoB,EAAE,MAAuB,EAAE,SAAwB;IAC5F,IAAI,QAAQ,GAAG,KAAK,CAAC,cAAc,CAAC;IACpC,IAAI,CAAC,QAAQ,IAAI,QAAQ,CAAC,SAAS,KAAK,SAAS,EAAE,CAAC;QACnD,QAAQ,GAAG;YACV,SAAS;YACT,KAAK,EAAE,CAAC;YACR,eAAe,EAAE,kCAAkC,CAAC,MAAM,CAAC,2BAA2B,CAAC,IAAI,IAAI;YAC/F,OAAO,EAAE,CAAC;YACV,YAAY,EAAE,EAAE;SAChB,CAAC;QACF,KAAK,CAAC,cAAc,GAAG,QAAQ,CAAC;IACjC,CAAC;IACD,IAAI,QAAQ,CAAC,eAAe,KAAK,SAAS,EAAE,CAAC;QAC5C,QAAQ,CAAC,eAAe,GAAG,kCAAkC,CAAC,MAAM,CAAC,2BAA2B,CAAC,IAAI,IAAI,CAAC;IAC3G,CAAC;IACD,QAAQ,CAAC,OAAO,KAAK,CAAC,CAAC;IACvB,QAAQ,CAAC,YAAY,KAAK,EAAE,CAAC;IAC7B,OAAO,QAAQ,CAAC;AACjB,CAAC;AAED,MAAM,UAAU,sBAAsB,CACrC,KAAoB,EACpB,MAAuB,EACvB,YAA2B,KAAK,CAAC,gBAAgB;IAEjD,MAAM,QAAQ,GAAG,YAAY,CAAC,KAAK,EAAE,MAAM,EAAE,SAAS,CAAC,CAAC;IACxD,MAAM,eAAe,GAAG,QAAQ,CAAC,eAAe,IAAI,IAAI,CAAC;IACzD,MAAM,OAAO,GAAG,eAAe,KAAK,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,OAAO,IAAI,CAAC,CAAC;IACrE,MAAM,KAAK,GAAG,eAAe,KAAK,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,eAAe,GAAG,OAAO,CAAC;IAC1E,OAAO;QACN,IAAI,EAAE,QAAQ,CAAC,KAAK;QACpB,eAAe;QACf,OAAO;QACP,KAAK;QACL,SAAS,EAAE,KAAK,KAAK,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,KAAK,GAAG,QAAQ,CAAC,KAAK,CAAC;QACtE,cAAc,EAAE,eAAe,KAAK,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,eAAe,GAAG,OAAO,CAAC;QACxF,YAAY,EAAE,CAAC,GAAG,CAAC,QAAQ,CAAC,YAAY,IAAI,EAAE,CAAC,CAAC;KAChD,CAAC;AACH,CAAC;AAED,MAAM,UAAU,wBAAwB,CAAC,QAA6B;IACrE,IAAI,QAAQ,CAAC,KAAK,KAAK,IAAI;QAAE,OAAO,WAAW,CAAC;IAChD,OAAO,GAAG,QAAQ,CAAC,IAAI,IAAI,QAAQ,CAAC,KAAK,UAAU,QAAQ,CAAC,SAAS,0BAA0B,QAAQ,CAAC,eAAe,aAAa,QAAQ,CAAC,OAAO,qBAAqB,QAAQ,CAAC,cAAc,GAAG,CAAC;AACrM,CAAC;AAED,MAAM,UAAU,iBAAiB,CAAC,QAA6B;IAC9D,OAAO,iBAAiB,wBAAwB,CAAC,QAAQ,CAAC,EAAE,CAAC;AAC9D,CAAC;AAED,MAAM,UAAU,oBAAoB,CACnC,KAAoB,EACpB,MAAuB,EACvB,SAAwB,EACxB,SAAiB;IAEjB,MAAM,QAAQ,GAAG,sBAAsB,CAAC,KAAK,EAAE,MAAM,EAAE,SAAS,CAAC,CAAC;IAClE,IAAI,SAAS,IAAI,CAAC,IAAI,QAAQ,CAAC,KAAK,KAAK,IAAI,IAAI,SAAS,IAAI,CAAC,QAAQ,CAAC,SAAS,IAAI,CAAC,CAAC;QAAE,OAAO,EAAE,QAAQ,EAAE,CAAC;IAC7G,OAAO;QACN,QAAQ;QACR,KAAK,EAAE,kDAAkD,QAAQ,CAAC,IAAI,IAAI,QAAQ,CAAC,KAAK,UAAU,SAAS,gBAAgB,QAAQ,CAAC,SAAS,yJAAyJ;KACtS,CAAC;AACH,CAAC;AAED,MAAM,UAAU,kBAAkB,CACjC,KAAoB,EACpB,MAAuB,EACvB,SAAwB,EACxB,SAAiB;IAEjB,MAAM,OAAO,GAAG,oBAAoB,CAAC,KAAK,EAAE,MAAM,EAAE,SAAS,EAAE,SAAS,CAAC,CAAC;IAC1E,IAAI,OAAO,CAAC,KAAK,IAAI,SAAS,IAAI,CAAC,IAAI,OAAO,CAAC,QAAQ,CAAC,KAAK,KAAK,IAAI;QAAE,OAAO,OAAO,CAAC;IACvF,KAAK,CAAC,cAAe,CAAC,KAAK,IAAI,SAAS,CAAC;IACzC,OAAO,EAAE,QAAQ,EAAE,sBAAsB,CAAC,KAAK,EAAE,MAAM,EAAE,SAAS,CAAC,EAAE,CAAC;AACvE,CAAC;AAED,MAAM,UAAU,yBAAyB,CACxC,KAAoB,EACpB,MAAuB,EACvB,SAAiB,EACjB,UAAkB;IAElB,MAAM,QAAQ,GAAG,sBAAsB,CAAC,KAAK,EAAE,MAAM,EAAE,SAAS,CAAC,CAAC;IAClE,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,UAAU,CAAC,IAAI,UAAU,IAAI,CAAC,EAAE,CAAC;QACtD,OAAO,EAAE,QAAQ,EAAE,KAAK,EAAE,2EAA2E,EAAE,CAAC;IACzG,CAAC;IACD,IAAI,QAAQ,CAAC,eAAe,KAAK,IAAI,IAAI,QAAQ,CAAC,KAAK,KAAK,IAAI,EAAE,CAAC;QAClE,OAAO,EAAE,QAAQ,EAAE,KAAK,EAAE,sFAAsF,EAAE,CAAC;IACpH,CAAC;IACD,IAAI,UAAU,GAAG,CAAC,QAAQ,CAAC,cAAc,IAAI,CAAC,CAAC,EAAE,CAAC;QACjD,OAAO;YACN,QAAQ;YACR,KAAK,EAAE,gCAAgC,UAAU,uBAAuB,QAAQ,CAAC,cAAc,sDAAsD;SACrJ,CAAC;IACH,CAAC;IACD,OAAO,EAAE,QAAQ,EAAE,CAAC;AACrB,CAAC;AAED,MAAM,UAAU,gBAAgB,CAC/B,KAAoB,EACpB,MAAuB,EACvB,SAAiB,EACjB,UAAkB,EAClB,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE;IAEhB,MAAM,OAAO,GAAG,yBAAyB,CAAC,KAAK,EAAE,MAAM,EAAE,SAAS,EAAE,UAAU,CAAC,CAAC;IAChF,IAAI,OAAO,CAAC,KAAK;QAAE,OAAO,OAAO,CAAC;IAClC,MAAM,MAAM,GAAG,OAAO,CAAC,QAAQ,CAAC;IAChC,IAAI,MAAM,CAAC,KAAK,KAAK,IAAI,EAAE,CAAC;QAC3B,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,sFAAsF,EAAE,CAAC;IAC5H,CAAC;IACD,MAAM,QAAQ,GAAG,KAAK,CAAC,cAAe,CAAC;IACvC,QAAQ,CAAC,OAAO,GAAG,CAAC,QAAQ,CAAC,OAAO,IAAI,CAAC,CAAC,GAAG,UAAU,CAAC;IACxD,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,GAAG,UAAU,CAAC;IACxC,QAAQ,CAAC,YAAY,GAAG;QACvB,GAAG,CAAC,QAAQ,CAAC,YAAY,IAAI,EAAE,CAAC;QAChC,EAAE,SAAS,EAAE,MAAM,EAAE,UAAU,EAAE,SAAS,EAAE,GAAG,EAAE,aAAa,EAAE,MAAM,CAAC,KAAK,EAAE,KAAK,EAAE;KACrF,CAAC,KAAK,CAAC,CAAC,iBAAiB,CAAC,CAAC;IAC5B,OAAO,EAAE,QAAQ,EAAE,sBAAsB,CAAC,KAAK,EAAE,MAAM,EAAE,SAAS,CAAC,EAAE,CAAC;AACvE,CAAC"}

package/src/runs/shared/structured-output.d.ts

@@ -0,0 +1,57 @@
+import type { JsonSchemaObject } from "../../shared/types.ts";
+import type { ResolvedAcceptanceReportMode } from "./acceptance.ts";
+export declare const MISSING_STRUCTURED_OUTPUT_CALL_ERROR = "Missing structured_output call; this step has outputSchema and must finish by calling structured_output.";
+export declare const MISSING_STRUCTURED_ACCEPTANCE_REPORT_ERROR = "Missing acceptanceReport in structured_output call; acceptance.report is \"on\".";
+export interface StructuredOutputRuntime {
+    schema: JsonSchemaObject;
+    schemaPath: string;
+    outputPath: string;
+    acceptanceReportPath?: string;
+    acceptanceReportRequired?: boolean;
+}
+export declare function createStructuredOutputToolParameters(schema: JsonSchemaObject, options?: {
+    acceptanceReport?: "optional" | "required";
+}): JsonSchemaObject;
+interface CompiledJsonSchema {
+    Check(value: unknown): boolean;
+    Errors(value: unknown): Iterable<JsonSchemaValidationError>;
+}
+interface JsonSchemaValidationError {
+    keyword?: string;
+    schemaPath?: string;
+    instancePath?: string;
+    params?: {
+        failingKeyword?: string;
+        requiredProperties?: string[];
+    };
+    message?: string;
+}
+type CompileJsonSchema = (schema: unknown) => CompiledJsonSchema;
+export declare function resolveCompileFromPackageRoot(packageRoot: string): Promise<CompileJsonSchema | undefined>;
+export declare function assertJsonSchemaObject(schema: unknown, label?: string): asserts schema is JsonSchemaObject;
+export declare function createStructuredOutputRuntime(schema: JsonSchemaObject, baseDir?: string, options?: {
+    acceptanceReport?: ResolvedAcceptanceReportMode;
+}): StructuredOutputRuntime;
+export declare function validateStructuredOutputValue(schema: JsonSchemaObject, value: unknown): Promise<{
+    status: "valid";
+} | {
+    status: "invalid";
+    message: string;
+}>;
+export declare function readStructuredOutput(runtime: StructuredOutputRuntime): Promise<{
+    value?: unknown;
+    error?: string;
+}>;
+export declare function readStructuredOutputAcceptanceReport(runtime: StructuredOutputRuntime): {
+    value?: unknown;
+    error?: string;
+};
+/**
+ * Capture callback that persists the structured output (and acceptance report)
+ * to the runtime's files, for hosts that read the value back from disk.
+ */
+export declare function createStructuredOutputFileCapture(runtime: StructuredOutputRuntime): (value: unknown, acceptanceReport: unknown | undefined) => void;
+export declare function clearStructuredOutputCaptures(runtime: StructuredOutputRuntime): string | undefined;
+export declare function cleanupStructuredOutputRuntime(runtime: StructuredOutputRuntime | undefined): void;
+export {};
+//# sourceMappingURL=structured-output.d.ts.map

package/src/runs/shared/structured-output.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"structured-output.d.ts","sourceRoot":"","sources":["../../../../src/runs/shared/structured-output.ts"],"names":[],"mappings":"AAMA,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,uBAAuB,CAAC;AAC9D,OAAO,KAAK,EAAE,4BAA4B,EAAE,MAAM,iBAAiB,CAAC;AAEpE,eAAO,MAAM,oCAAoC,6GAA6G,CAAC;AAC/J,eAAO,MAAM,0CAA0C,qFAAqF,CAAC;AAE7I,MAAM,WAAW,uBAAuB;IACvC,MAAM,EAAE,gBAAgB,CAAC;IACzB,UAAU,EAAE,MAAM,CAAC;IACnB,UAAU,EAAE,MAAM,CAAC;IACnB,oBAAoB,CAAC,EAAE,MAAM,CAAC;IAC9B,wBAAwB,CAAC,EAAE,OAAO,CAAC;CACnC;AA6CD,wBAAgB,oCAAoC,CAAC,MAAM,EAAE,gBAAgB,EAAE,OAAO,GAAE;IAAE,gBAAgB,CAAC,EAAE,UAAU,GAAG,UAAU,CAAA;CAAO,GAAG,gBAAgB,CAU7J;AAED,UAAU,kBAAkB;IAC3B,KAAK,CAAC,KAAK,EAAE,OAAO,GAAG,OAAO,CAAC;IAC/B,MAAM,CAAC,KAAK,EAAE,OAAO,GAAG,QAAQ,CAAC,yBAAyB,CAAC,CAAC;CAC5D;AAED,UAAU,yBAAyB;IAClC,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,MAAM,CAAC,EAAE;QAAE,cAAc,CAAC,EAAE,MAAM,CAAC;QAAC,kBAAkB,CAAC,EAAE,MAAM,EAAE,CAAA;KAAE,CAAC;IACpE,OAAO,CAAC,EAAE,MAAM,CAAC;CACjB;AAED,KAAK,iBAAiB,GAAG,CAAC,MAAM,EAAE,OAAO,KAAK,kBAAkB,CAAC;AAIjE,wBAAsB,6BAA6B,CAAC,WAAW,EAAE,MAAM,GAAG,OAAO,CAAC,iBAAiB,GAAG,SAAS,CAAC,CAK/G;AA4KD,wBAAgB,sBAAsB,CAAC,MAAM,EAAE,OAAO,EAAE,KAAK,SAAiB,GAAG,OAAO,CAAC,MAAM,IAAI,gBAAgB,CAIlH;AAED,wBAAgB,6BAA6B,CAAC,MAAM,EAAE,gBAAgB,EAAE,OAAO,CAAC,EAAE,MAAM,EAAE,OAAO,GAAE;IAAE,gBAAgB,CAAC,EAAE,4BAA4B,CAAA;CAAO,GAAG,uBAAuB,CAgBpL;AAED,wBAAsB,6BAA6B,CAAC,MAAM,EAAE,gBAAgB,EAAE,KAAK,EAAE,OAAO,GAAG,OAAO,CAAC;IAAE,MAAM,EAAE,OAAO,CAAA;CAAE,GAAG;IAAE,MAAM,EAAE,SAAS,CAAC;IAAC,OAAO,EAAE,MAAM,CAAA;CAAE,CAAC,CAuBnK;AAED,wBAAsB,oBAAoB,CAAC,OAAO,EAAE,uBAAuB,GAAG,OAAO,CAAC;IAAE,KAAK,CAAC,EAAE,OAAO,CAAC;IAAC,KAAK,CAAC,EAAE,MAAM,CAAA;CAAE,CAAC,CAiBzH;AAED,wBAAgB,oCAAoC,CAAC,OAAO,EAAE,uBAAuB,GAAG;IAAE,KAAK,CAAC,EAAE,OAAO,CAAC;IAAC,KAAK,CAAC,EAAE,MAAM,CAAA;CAAE,CAU1H;AAED;;;GAGG;AACH,wBAAgB,iCAAiC,CAAC,OAAO,EAAE,uBAAuB,GAAG,CAAC,KAAK,EAAE,OAAO,EAAE,gBAAgB,EAAE,OAAO,GAAG,SAAS,KAAK,IAAI,CAWnJ;AAED,wBAAgB,6BAA6B,CAAC,OAAO,EAAE,uBAAuB,GAAG,MAAM,GAAG,SAAS,CAWlG;AAED,wBAAgB,8BAA8B,CAAC,OAAO,EAAE,uBAAuB,GAAG,SAAS,GAAG,IAAI,CAOjG"}