@agwab/pi-workflow 0.2.0 → 0.3.0

package/node_modules/@agwab/pi-subagent/src/runners/headless-model.ts

@@ -79,6 +79,13 @@ export interface PiJsonParseResult {
 	metadata: Partial<ResultMetadata>;
 }
 
+export interface ContextLengthResolution {
+	rawContextLengthExceeded: boolean;
+	contextLengthExceeded: boolean;
+	contextOverflowRecovered: boolean;
+	recoveredStreamErrors: string[];
+}
+
 const CONTEXT_LENGTH_ERROR_PATTERN =
 	/\bcontext[_ -]?length[_ -]?exceeded\b|\bcontext[_ -]?window[_ -]?(?:exceeded|overflow|exhausted)\b|\b(?:maximum|max)[_ -]?context[_ -]?length\b|\btoo many tokens\b|\b(?:prompt|input|request)[^\n]{0,80}\btoo large\b|\bcontext_length_exceeded\b/i;
 
@@ -94,6 +101,32 @@ export function detectContextLengthExceeded(signals: {
 	return CONTEXT_LENGTH_ERROR_PATTERN.test(text);
 }
 
+export function resolveContextLengthState(
+	parsed: PiJsonParseResult,
+	rawContextLengthExceeded: boolean,
+): ContextLengthResolution {
+	const contextOverflowRecovered =
+		rawContextLengthExceeded && finalAssistantSucceeded(parsed);
+	return {
+		rawContextLengthExceeded,
+		contextLengthExceeded:
+			rawContextLengthExceeded && !contextOverflowRecovered,
+		contextOverflowRecovered,
+		recoveredStreamErrors: contextOverflowRecovered
+			? parsed.errors.filter((error) =>
+					detectContextLengthExceeded({ errors: [error] }),
+				)
+			: [],
+	};
+}
+
+function finalAssistantSucceeded(parsed: PiJsonParseResult): boolean {
+	return (
+		parsed.finalAssistantText.length > 0 &&
+		parsed.metadata.stopReason !== "error"
+	);
+}
+
 function normalizeTimeoutMs(timeoutMs: number | undefined): number | undefined {
 	if (timeoutMs === undefined) return undefined;
 	if (!Number.isFinite(timeoutMs) || timeoutMs <= 0) {
@@ -390,18 +423,29 @@ export function resolvePiJsonOutcome(
 
 export function resultMetadataFromParse(
 	parsed: PiJsonParseResult,
-	contextLengthExceeded: boolean,
+	contextLength: ContextLengthResolution,
 	outcome: ProcessOutcome,
 ): Partial<ResultMetadata> {
 	return {
 		...parsed.metadata,
-		contextLengthExceeded,
+		contextLengthExceeded: contextLength.contextLengthExceeded,
+		...(contextLength.contextOverflowRecovered
+			? { contextOverflowRecovered: true }
+			: {}),
 		...(parsed.errors.length === 0
 			? {}
 			: { streamErrors: parsed.errors.slice(0, MAX_METADATA_ERRORS) }),
 		...(outcome.status === "completed" && parsed.errors.length > 0
 			? { nonFatalStreamErrors: parsed.errors.slice(0, MAX_METADATA_ERRORS) }
 			: {}),
+		...(contextLength.recoveredStreamErrors.length === 0
+			? {}
+			: {
+					recoveredStreamErrors: contextLength.recoveredStreamErrors.slice(
+						0,
+						MAX_METADATA_ERRORS,
+					),
+				}),
 		...(parsed.parseErrors.length === 0
 			? {}
 			: { parseErrors: parsed.parseErrors.slice(0, MAX_METADATA_ERRORS) }),
@@ -774,14 +818,18 @@ export async function runHeadlessModel(
 		stderrText,
 		stderrContextLengthExceeded,
 	} = processResult;
-	const contextLengthExceeded =
+	const rawContextLengthExceeded =
 		stderrContextLengthExceeded ||
 		detectContextLengthExceeded({ stderrText, errors: parsed.errors });
+	const contextLength = resolveContextLengthState(
+		parsed,
+		rawContextLengthExceeded,
+	);
 
 	const outcome = resolvePiJsonOutcome(
 		processOutcome,
 		parsed,
-		contextLengthExceeded,
+		contextLength.contextLengthExceeded,
 	);
 
 	const completedAt = new Date();
@@ -811,7 +859,7 @@ export async function runHeadlessModel(
 		artifacts,
 		correlationId: options.correlationId,
 		metadata: {
-			...resultMetadataFromParse(parsed, contextLengthExceeded, outcome),
+			...resultMetadataFromParse(parsed, contextLength, outcome),
 			...sessionMetadata,
 			...(options.parentSessionId === undefined
 				? {}

package/node_modules/@agwab/pi-subagent/src/runners/tmux.ts

@@ -20,6 +20,7 @@ import {
 	detectContextLengthExceeded,
 	parsePiJsonFile,
 	parsePiJsonLines,
+	resolveContextLengthState,
 	resolvePiJsonOutcome,
 	resultMetadataFromParse,
 	resultSessionMetadata,
@@ -157,9 +158,7 @@ function workerScript(
 	return `import { spawn } from "node:child_process";\nimport { appendFileSync, closeSync, openSync, writeFileSync } from "node:fs";\nconst argv = ${JSON.stringify(argv)};\nconst cwd = ${JSON.stringify(cwd)};\nconst eventPath = ${JSON.stringify(eventPath)};\nconst stderrPath = ${JSON.stringify(stderrPath)};\nconst metaPath = ${JSON.stringify(metaPath)};\nconst messageUpdatePattern = /"type"\\s*:\\s*"message_update"/;\nconst maxStdoutLogLineChars = 64 * 1024 * 1024;\ncloseSync(openSync(eventPath, "w"));\ncloseSync(openSync(stderrPath, "w"));\nlet settled = false;\nlet stdoutBuffer = "";\nlet discardingOversizedLine = false;\nlet omittedMessageUpdates = 0;\nlet omittedMessageUpdateBytes = 0;\nlet omittedOversizedLines = 0;\nlet omittedOversizedBytes = 0;\nfunction writeStdoutLine(line) {\n  if (messageUpdatePattern.test(line)) {\n    omittedMessageUpdates += 1;\n    omittedMessageUpdateBytes += Buffer.byteLength(line, "utf8");\n    return;\n  }\n  appendFileSync(eventPath, line);\n  process.stdout.write(line);\n}\nfunction handleStdoutChunk(chunk) {\n  let text = chunk.toString("utf8");\n  while (text.length > 0) {\n    if (discardingOversizedLine) {\n      const newline = text.indexOf("\\n");\n      omittedOversizedBytes += Buffer.byteLength(newline < 0 ? text : text.slice(0, newline + 1), "utf8");\n      if (newline < 0) return;\n      discardingOversizedLine = false;\n      text = text.slice(newline + 1);\n      continue;\n    }\n    const newline = text.indexOf("\\n");\n    const segment = newline < 0 ? text : text.slice(0, newline + 1);\n    stdoutBuffer += segment;\n    text = newline < 0 ? "" : text.slice(newline + 1);\n    if (stdoutBuffer.length > maxStdoutLogLineChars) {\n      omittedOversizedLines += 1;\n      omittedOversizedBytes += Buffer.byteLength(stdoutBuffer, "utf8");\n      stdoutBuffer = "";\n      discardingOversizedLine = newline < 0;\n      continue;\n    }\n    if (newline >= 0) {\n      writeStdoutLine(stdoutBuffer);\n      stdoutBuffer = "";\n    }\n  }\n}\nfunction finishStdoutFilter() {\n  if (!discardingOversizedLine && stdoutBuffer.length > 0) writeStdoutLine(stdoutBuffer);\n  stdoutBuffer = "";\n  if (omittedMessageUpdates > 0 || omittedOversizedLines > 0) {\n    appendFileSync(eventPath, JSON.stringify({ type: "pi-subagent.stdout_filter", omitted: { messageUpdateEvents: omittedMessageUpdates, messageUpdateBytes: omittedMessageUpdateBytes, oversizedLines: omittedOversizedLines, oversizedBytes: omittedOversizedBytes }, reason: "cumulative message_update snapshots are omitted from durable stdout artifacts; final assistant text is stored in output.log" }) + "\\n");\n  }\n}\nfunction writeMeta(meta) {\n  if (settled) return;\n  settled = true;\n  finishStdoutFilter();\n  writeFileSync(metaPath, JSON.stringify(meta, null, 2) + "\\n");\n}\nconst env = { ...process.env };\ndelete env.TMUX;\nconst child = spawn(argv[0], argv.slice(1), { cwd, shell: false, stdio: ["ignore", "pipe", "pipe"], env });\nchild.stdout?.on("data", handleStdoutChunk);\nchild.stderr?.on("data", (chunk) => { appendFileSync(stderrPath, chunk); process.stderr.write(chunk); });\nchild.on("error", () => { writeMeta({ status: "failed", failureKind: "spawn", exitCode: null, signal: null }); });\nchild.on("close", (exitCode, signal) => {\n  const failureKind = exitCode === 0 ? null : "exit";\n  writeMeta({ status: failureKind === null ? "completed" : "failed", failureKind, exitCode, signal });\n});\n`;
 }
 
-async function runTmuxProcess(
-	options: RunTmuxProcessOptions,
-): Promise<{
+async function runTmuxProcess(options: RunTmuxProcessOptions): Promise<{
 	result: TmuxRunResult | null;
 	store: Awaited<ReturnType<typeof createAttemptArtifactStore>>;
 	cwd: string;
@@ -439,11 +438,19 @@ export async function runTmuxModel(
 		parsePiJsonLines(""),
 	);
 	await unlink(result.eventPath).catch(() => undefined);
-	const contextLengthExceeded = detectContextLengthExceeded({
+	const rawContextLengthExceeded = detectContextLengthExceeded({
 		stderrText,
 		errors: parsed.errors,
 	});
-	const meta = resolvePiJsonOutcome(result.meta, parsed, contextLengthExceeded);
+	const contextLength = resolveContextLengthState(
+		parsed,
+		rawContextLengthExceeded,
+	);
+	const meta = resolvePiJsonOutcome(
+		result.meta,
+		parsed,
+		contextLength.contextLengthExceeded,
+	);
 
 	const outputRef = await store.writeTextArtifact(
 		"output",
@@ -464,7 +471,7 @@ export async function runTmuxModel(
 		tmux: result.tmux,
 		correlationId: options.correlationId,
 		metadata: {
-			...resultMetadataFromParse(parsed, contextLengthExceeded, meta),
+			...resultMetadataFromParse(parsed, contextLength, meta),
 			...sessionMetadata,
 			...(options.parentSessionId === undefined
 				? {}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@agwab/pi-workflow",
-  "version": "0.2.0",
+  "version": "0.3.0",
   "description": "Workflow orchestration for Pi subagents.",
   "private": false,
   "type": "module",
@@ -76,7 +76,7 @@
     "node": ">=22.19.0"
   },
   "dependencies": {
-    "@agwab/pi-subagent": "^0.3.6",
+    "@agwab/pi-subagent": "^0.4.0",
     "pi-web-access": "^0.10.7",
     "typebox": "^1.1.39"
   },

package/src/compiler.ts

@@ -3,6 +3,7 @@ import { dirname, resolve } from "node:path";
 
 import { loadAgentByName } from "./agents.js";
 import { DYNAMIC_OUTPUT_PROFILES } from "./dynamic-profiles.js";
+import { compileRole } from "./roles.js";
 import {
 	classifyToolCapability,
 	effectiveToolClassification,
@@ -33,7 +34,10 @@ import {
 } from "./types.js";
 import {
 	resolveWorkflowRuntime,
+	selectWorkflowRuntime,
 	type WorkflowModelInfo,
+	type WorkflowRuntimeDefaults,
+	type WorkflowRuntimeResolutionInput,
 } from "./workflow-runtime.js";
 
 const DELEGATION_TOOLS = new Set([
@@ -549,7 +553,8 @@ export async function compileWorkflow(
 	spec: ArtifactGraphWorkflowSpec,
 	options: CompileOptions & {
 		task?: string;
-		runtimeDefaults?: { model?: string; thinking?: ThinkingLevel };
+		runtimeOverrides?: WorkflowRuntimeDefaults;
+		runtimeDefaults?: WorkflowRuntimeDefaults;
 	},
 ): Promise<any> {
 	const compilePlan = buildArtifactGraphCompilePlan(spec, options);
@@ -615,11 +620,25 @@ async function collectForeachPathWarnings(
 	return warnings;
 }
 
+function runtimeSettings(value: unknown): WorkflowRuntimeDefaults | undefined {
+	if (!isPlainRecord(value)) return undefined;
+	const model =
+		typeof value.model === "string" && value.model.trim()
+			? value.model.trim()
+			: undefined;
+	const thinking =
+		typeof value.thinking === "string" && value.thinking.trim()
+			? (value.thinking.trim() as ThinkingLevel)
+			: undefined;
+	return model || thinking ? { model, thinking } : undefined;
+}
+
 async function compileArtifactGraphPlan(
 	spec: any,
 	options: CompileOptions & {
 		task?: string;
-		runtimeDefaults?: { model?: string; thinking?: ThinkingLevel };
+		runtimeOverrides?: WorkflowRuntimeDefaults;
+		runtimeDefaults?: WorkflowRuntimeDefaults;
 	},
 ): Promise<any> {
 	const stages = spec.stages;
@@ -645,15 +664,19 @@ async function compileArtifactGraphPlan(
 		return defaultAgent;
 	};
 	const roleEntries = Object.entries(spec.roles ?? {});
-	const roles = roleEntries.map(([name, role]: [string, any]) => ({
-		name,
-		fromAgent: role.fromAgent,
-		content: role.prompt ?? "",
-		maxChars: role.maxChars ?? 8000,
-		truncated: false,
-		includedSections: [],
-		excludedSections: [],
-	}));
+	const roles = await Promise.all(
+		roleEntries.map(async ([name, role]: [string, any]) => {
+			const sourceAgent = role.fromAgent
+				? await loadWorkflowAgent(
+						role.fromAgent,
+						options.cwd,
+						agentCache,
+						`$.roles.${name}.fromAgent`,
+					)
+				: undefined;
+			return compileRole(name, role, sourceAgent);
+		}),
+	);
 	const roleText = roles.length
 		? `# Role Context\n\n${roles.map((r) => `## Role: ${r.name}\n${r.content}`).join("\n\n")}`
 		: "";
@@ -665,9 +688,9 @@ async function compileArtifactGraphPlan(
 		Object.keys(workflowInput).length > 0
 			? `# Workflow Input\n\n${JSON.stringify(workflowInput, null, 2)}`
 			: "";
-	const defaultModel = options.runtimeDefaults?.model ?? spec.defaults?.model;
-	const defaultThinking =
-		options.runtimeDefaults?.thinking ?? spec.defaults?.thinking;
+	const runtimeOverrides = options.runtimeOverrides;
+	const runtimeDefaults = options.runtimeDefaults;
+	const specRuntimeDefaults = runtimeSettings(spec.defaults);
 	const tasks: any[] = [];
 	const stageRecords: any[] = [];
 	const issues: ValidationIssue[] = [];
@@ -719,6 +742,22 @@ async function compileArtifactGraphPlan(
 				dynamicToolPath,
 			);
 			const dynamicToolSelection = filterToolSelection(rawDynamicToolSelection);
+			const requestedRuntime = selectWorkflowRuntime(
+				runtimeOverrides,
+				runtimeSettings(stage),
+				runtimeDefaults,
+				specRuntimeDefaults,
+			);
+			const resolvedDynamicRuntime = await resolveWorkflowRuntime(
+				requestedRuntime,
+				{
+					taskKey: key,
+					stageId: stage.id,
+					taskId,
+					agent: "dynamic",
+				},
+				{ availableModels: options.availableModels },
+			);
 			const dynamicTask = buildDynamicTask(
 				stage,
 				taskId,
@@ -729,24 +768,22 @@ async function compileArtifactGraphPlan(
 				specDir,
 				workflowInputText,
 				options.task,
-				defaultModel,
-				defaultThinking,
-				overrides,
-			);
-			const resolvedDynamicRuntime = await resolveWorkflowRuntime(
-				{ model: defaultModel, thinking: defaultThinking },
+				resolvedDynamicRuntime,
 				{
-					taskKey: key,
-					stageId: stage.id,
-					taskId,
-					agent: "dynamic",
+					runtimeOverrides,
+					runtimeDefaults,
+					specRuntimeDefaults,
+					stageRuntime: runtimeSettings(stage),
 				},
-				{ availableModels: options.availableModels },
+				overrides,
 			);
 			dynamicTask.runtime = {
 				...dynamicTask.runtime,
 				...resolvedDynamicRuntime,
 			};
+			if (options.availableModels?.length) {
+				dynamicTask.dynamic.availableModels = options.availableModels;
+			}
 			if (dynamicToolSelection.tools || dynamicToolSelection.toolProviders) {
 				dynamicTask.runtime = {
 					...dynamicTask.runtime,
@@ -823,15 +860,12 @@ async function compileArtifactGraphPlan(
 		validateToolSubset(toolSelection.tools, stageAgent, issues, toolPath);
 		validateDelegationBoundary(toolSelection.tools, issues, toolPath);
 		const filteredToolSelection = filterToolSelection(toolSelection);
-		// Explicit runtime overrides outrank stage pins; spec defaults fill last.
-		const requestedRuntime = {
-			model:
-				options.runtimeDefaults?.model ?? stage.model ?? spec.defaults?.model,
-			thinking:
-				options.runtimeDefaults?.thinking ??
-				stage.thinking ??
-				spec.defaults?.thinking,
-		};
+		const requestedRuntime = selectWorkflowRuntime(
+			runtimeOverrides,
+			runtimeSettings(stage),
+			runtimeDefaults,
+			specRuntimeDefaults,
+		);
 		const resolvedRuntime = await resolveWorkflowRuntime(
 			requestedRuntime,
 			{
@@ -1208,12 +1242,34 @@ async function compileArtifactGraphPlan(
 		tasks,
 		warnings,
 		budget: {
-			models: defaultModel ? [{ model: defaultModel }] : [],
+			models: budgetModelRows(tasks),
 			unratedModels: [],
 		},
 	};
 }
 
+function budgetModelRows(tasks: any[]): Array<{ model: string }> {
+	const models = new Set<string>();
+	for (const task of tasks) {
+		if (typeof task?.runtime?.model === "string" && task.runtime.model.trim()) {
+			models.add(task.runtime.model.trim());
+		}
+		const loop = task?.dynamic?.decisionLoop;
+		if (!loop || typeof loop !== "object") continue;
+		for (const profile of [
+			loop.planner,
+			loop.workerDefaults,
+			loop.verifier,
+			loop.synthesis,
+		]) {
+			if (typeof profile?.model === "string" && profile.model.trim()) {
+				models.add(profile.model.trim());
+			}
+		}
+	}
+	return [...models].sort().map((model) => ({ model }));
+}
+
 function isSupportStage(stage: any): boolean {
 	return stage?.support !== undefined && stage?.type === undefined;
 }
@@ -1301,8 +1357,13 @@ function buildDynamicTask(
 	specDir: string,
 	workflowInputText: string,
 	runtimeTask: string | undefined,
-	defaultModel: string | undefined,
-	defaultThinking: ThinkingLevel | undefined,
+	controllerRuntime: WorkflowRuntimeResolutionInput,
+	runtimePriority: {
+		runtimeOverrides?: WorkflowRuntimeDefaults;
+		runtimeDefaults?: WorkflowRuntimeDefaults;
+		specRuntimeDefaults?: WorkflowRuntimeDefaults;
+		stageRuntime?: WorkflowRuntimeDefaults;
+	},
 	overrides: Partial<CompiledTask> & Record<string, unknown>,
 ): any {
 	const dynamic = stage.dynamic ?? {};
@@ -1365,8 +1426,7 @@ function buildDynamicTask(
 	}
 	const decisionLoop = compileDynamicDecisionLoop(
 		dynamic.decisionLoop,
-		defaultModel,
-		defaultThinking,
+		runtimePriority,
 	);
 
 	return {
@@ -1386,8 +1446,7 @@ function buildDynamicTask(
 		explicitWorktreePolicy: false,
 		runtime: {
 			approvalMode: "non-interactive",
-			model: defaultModel,
-			thinking: defaultThinking,
+			...controllerRuntime,
 			maxRuntimeMs:
 				dynamic.budget?.maxRuntimeMs ?? DEFAULT_DYNAMIC_MAX_RUNTIME_MS,
 		},
@@ -1431,6 +1490,9 @@ function buildDynamicTask(
 			helpers,
 			workflows,
 			...(decisionLoop ? { decisionLoop } : {}),
+			...(runtimePriority.runtimeOverrides
+				? { runtimeOverrides: runtimePriority.runtimeOverrides }
+				: {}),
 		},
 		...overrides,
 	};
@@ -1438,8 +1500,12 @@ function buildDynamicTask(
 
 function compileDynamicDecisionLoop(
 	value: unknown,
-	defaultModel?: string,
-	defaultThinking?: ThinkingLevel,
+	runtimePriority: {
+		runtimeOverrides?: WorkflowRuntimeDefaults;
+		runtimeDefaults?: WorkflowRuntimeDefaults;
+		specRuntimeDefaults?: WorkflowRuntimeDefaults;
+		stageRuntime?: WorkflowRuntimeDefaults;
+	},
 ): any | undefined {
 	if (!isPlainRecord(value)) return undefined;
 	const allowedToolSelection = filterToolSelection(
@@ -1455,25 +1521,18 @@ function compileDynamicDecisionLoop(
 		recordValue(value.stateIndex, "requiredFindingIds"),
 	);
 	return {
-		planner: compileDynamicDecisionLoopProfile(
-			value.planner,
-			defaultModel,
-			defaultThinking,
-		),
+		planner: compileDynamicDecisionLoopProfile(value.planner, runtimePriority),
 		workerDefaults: compileDynamicDecisionLoopProfile(
 			value.workerDefaults,
-			defaultModel,
-			defaultThinking,
+			runtimePriority,
 		),
 		verifier: compileDynamicDecisionLoopProfile(
 			value.verifier,
-			defaultModel,
-			defaultThinking,
+			runtimePriority,
 		),
 		synthesis: compileDynamicDecisionLoopProfile(
 			value.synthesis,
-			defaultModel,
-			defaultThinking,
+			runtimePriority,
 		),
 		allowedAgents: stringArray(value.allowedAgents),
 		...(allowedToolSelection.tools
@@ -1528,8 +1587,12 @@ function compileDynamicDecisionLoop(
 
 function compileDynamicDecisionLoopProfile(
 	value: unknown,
-	defaultModel?: string,
-	defaultThinking?: ThinkingLevel,
+	runtimePriority: {
+		runtimeOverrides?: WorkflowRuntimeDefaults;
+		runtimeDefaults?: WorkflowRuntimeDefaults;
+		specRuntimeDefaults?: WorkflowRuntimeDefaults;
+		stageRuntime?: WorkflowRuntimeDefaults;
+	},
 ): any | undefined {
 	if (!isPlainRecord(value)) return undefined;
 	const toolSelection = filterToolSelection(
@@ -1538,20 +1601,18 @@ function compileDynamicDecisionLoopProfile(
 			undefined,
 		),
 	);
-	const model =
-		typeof value.model === "string" && value.model.trim()
-			? value.model.trim()
-			: defaultModel;
-	const thinking =
-		typeof value.thinking === "string" && value.thinking.trim()
-			? value.thinking.trim()
-			: defaultThinking;
+	const runtime = selectWorkflowRuntime(
+		runtimePriority.runtimeOverrides,
+		runtimeSettings(value),
+		runtimePriority.stageRuntime,
+		runtimePriority.runtimeDefaults,
+		runtimePriority.specRuntimeDefaults,
+	);
 	return {
 		...(typeof value.agent === "string" && value.agent.trim()
 			? { agent: value.agent.trim() }
 			: {}),
-		...(model ? { model } : {}),
-		...(thinking ? { thinking } : {}),
+		...runtime,
 		...(toolSelection.tools ? { tools: toolSelection.tools } : {}),
 		...(toolSelection.toolProviders
 			? { toolProviders: toolSelection.toolProviders }

package/src/dynamic-decision.ts

@@ -283,17 +283,6 @@ export function validateDynamicDecision(
 	};
 }
 
-export function assertValidDynamicDecision(
-	value: unknown,
-	context: DynamicDecisionValidationContext = {},
-): NormalizedDynamicDecision {
-	const result = validateDynamicDecision(value, context);
-	if (!result.ok || !result.decision) {
-		throw new Error(`invalid dynamic decision: ${result.errors.join("; ")}`);
-	}
-	return result.decision;
-}
-
 export function hashDynamicDecision(value: unknown): string {
 	return createHash("sha256")
 		.update(stableStringify(toJsonNormalizedValue(value)))

package/src/dynamic-generated-task-runtime.ts

@@ -24,6 +24,11 @@ import type {
 	WorkflowRunRecord,
 	WorkflowTaskRunRecord,
 } from "./types.js";
+import {
+	resolveWorkflowRuntime,
+	selectWorkflowRuntime,
+	type WorkflowModelInfo,
+} from "./workflow-runtime.js";
 
 const DYNAMIC_OUTPUT_MAX_DIGEST_CHARS = 1000;
 const DYNAMIC_DELEGATION_TOOLS = new Set([
@@ -92,6 +97,7 @@ export async function buildDynamicGeneratedCompiledTask(input: {
 	branchId?: string;
 	request: DynamicAgentRequest;
 	dynamic: CompiledDynamicWorkflowTask;
+	availableModels?: WorkflowModelInfo[];
 }): Promise<CompiledTask> {
 	if (input.dynamic.budget.maxAgents <= 0) {
 		throw new Error("dynamic agent budget is exhausted");
@@ -172,6 +178,23 @@ export async function buildDynamicGeneratedCompiledTask(input: {
 				),
 			),
 		);
+	const selectedRuntime = selectWorkflowRuntime(
+		input.dynamic.runtimeOverrides,
+		runtimeSettings(input.request),
+		runtimeSettings(executionProfile),
+		runtimeSettings(input.controllerCompiledTask.runtime),
+		runtimeSettings(agentDefinition),
+	);
+	const resolvedRuntime = await resolveWorkflowRuntime(
+		selectedRuntime,
+		{
+			taskKey: input.generatedSpecId,
+			stageId: input.controllerStageId,
+			taskId: input.request.id,
+			agent: requestedAgent,
+		},
+		{ availableModels: input.availableModels ?? input.dynamic.availableModels },
+	);
 	const unknownTools = (tools ?? []).filter(
 		(tool) => effectiveToolClassification(tool, toolProviders) === undefined,
 	);
@@ -253,16 +276,7 @@ export async function buildDynamicGeneratedCompiledTask(input: {
 		explicitWorktreePolicy: requiresWorktree,
 		runtime: {
 			approvalMode: "non-interactive",
-			model:
-				input.request.model ??
-				executionProfile?.model ??
-				input.controllerCompiledTask.runtime.model ??
-				agentDefinition.model,
-			thinking:
-				input.request.thinking ??
-				executionProfile?.thinking ??
-				input.controllerCompiledTask.runtime.thinking ??
-				agentDefinition.thinking,
+			...resolvedRuntime,
 			tools,
 			...(toolProviders ? { toolProviders } : {}),
 			maxRuntimeMs:
@@ -419,7 +433,9 @@ function dynamicDecisionLoopProfile(
 	);
 }
 
-export function isDynamicCompiledTaskPayload(value: unknown): value is CompiledTask {
+export function isDynamicCompiledTaskPayload(
+	value: unknown,
+): value is CompiledTask {
 	return (
 		!!value &&
 		typeof value === "object" &&
@@ -679,7 +695,9 @@ export async function readDynamicGeneratedTaskResult(
 	};
 }
 
-export function normalizeDynamicAgentRequest(value: unknown): DynamicAgentRequest {
+export function normalizeDynamicAgentRequest(
+	value: unknown,
+): DynamicAgentRequest {
 	if (!value || typeof value !== "object" || Array.isArray(value)) {
 		throw new Error("ctx.agent() request must be an object");
 	}
@@ -730,6 +748,23 @@ function requiredDynamicString(
 	return value.trim();
 }
 
+function runtimeSettings(
+	value: unknown,
+): { model?: string; thinking?: ThinkingLevel } | undefined {
+	if (!value || typeof value !== "object" || Array.isArray(value))
+		return undefined;
+	const record = value as Record<string, unknown>;
+	const model =
+		typeof record.model === "string" && record.model.trim()
+			? record.model.trim()
+			: undefined;
+	const thinking =
+		typeof record.thinking === "string" && record.thinking.trim()
+			? (record.thinking.trim() as ThinkingLevel)
+			: undefined;
+	return model || thinking ? { model, thinking } : undefined;
+}
+
 function optionalDynamicString(
 	value: unknown,
 	field: string,

package/src/dynamic-profiles.ts

@@ -44,7 +44,3 @@ export function isTerminalDynamicOutputProfile(
 ): value is (typeof DYNAMIC_TERMINAL_OUTPUT_PROFILES)[number] {
 	return typeof value === "string" && TERMINAL_OUTPUT_PROFILE_SET.has(value);
 }
-
-export function dynamicOutputProfileValues(): string[] {
-	return [...DYNAMIC_OUTPUT_PROFILES];
-}