@agwab/pi-workflow 0.1.2 → 0.2.1

package/src/compiler.ts

@@ -31,6 +31,13 @@ import {
 	type WorkflowToolSpec,
 	type WorktreePolicy,
 } from "./types.js";
+import {
+	resolveWorkflowRuntime,
+	selectWorkflowRuntime,
+	type WorkflowModelInfo,
+	type WorkflowRuntimeDefaults,
+	type WorkflowRuntimeResolutionInput,
+} from "./workflow-runtime.js";
 
 const DELEGATION_TOOLS = new Set([
 	"skill_test_subagent",
@@ -57,6 +64,7 @@ const DEFAULT_DYNAMIC_DECISION_LOOP_MAX_STALLS = 3;
 interface CompileOptions {
 	cwd: string;
 	specPath?: string;
+	availableModels?: WorkflowModelInfo[];
 }
 
 interface ArtifactGraphCompilePlanBuildResult {
@@ -544,7 +552,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);
@@ -610,11 +619,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;
@@ -660,9 +683,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[] = [];
@@ -714,6 +737,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,
@@ -724,10 +763,22 @@ async function compileArtifactGraphPlan(
 				specDir,
 				workflowInputText,
 				options.task,
-				defaultModel,
-				defaultThinking,
+				resolvedDynamicRuntime,
+				{
+					runtimeOverrides,
+					runtimeDefaults,
+					specRuntimeDefaults,
+					stageRuntime: runtimeSettings(stage),
+				},
 				overrides,
 			);
+			dynamicTask.runtime = {
+				...dynamicTask.runtime,
+				...resolvedDynamicRuntime,
+			};
+			if (options.availableModels?.length) {
+				dynamicTask.dynamic.availableModels = options.availableModels;
+			}
 			if (dynamicToolSelection.tools || dynamicToolSelection.toolProviders) {
 				dynamicTask.runtime = {
 					...dynamicTask.runtime,
@@ -804,11 +855,28 @@ async function compileArtifactGraphPlan(
 		validateToolSubset(toolSelection.tools, stageAgent, issues, toolPath);
 		validateDelegationBoundary(toolSelection.tools, issues, toolPath);
 		const filteredToolSelection = filterToolSelection(toolSelection);
+		const requestedRuntime = selectWorkflowRuntime(
+			runtimeOverrides,
+			runtimeSettings(stage),
+			runtimeDefaults,
+			specRuntimeDefaults,
+		);
+		const resolvedRuntime = await resolveWorkflowRuntime(
+			requestedRuntime,
+			{
+				taskKey: key,
+				stageId: stage.id,
+				taskId,
+				agent: stageAgentName,
+			},
+			{
+				availableModels: options.availableModels,
+			},
+		);
 		const runtime = {
 			approvalMode:
 				stage.approvalMode ?? spec.defaults?.approvalMode ?? "non-interactive",
-			model: stage.model ?? defaultModel,
-			thinking: stage.thinking ?? defaultThinking,
+			...resolvedRuntime,
 			tools: filteredToolSelection.tools,
 			...(filteredToolSelection.toolProviders
 				? { toolProviders: filteredToolSelection.toolProviders }
@@ -1169,12 +1237,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;
 }
@@ -1262,8 +1352,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 ?? {};
@@ -1326,8 +1421,7 @@ function buildDynamicTask(
 	}
 	const decisionLoop = compileDynamicDecisionLoop(
 		dynamic.decisionLoop,
-		defaultModel,
-		defaultThinking,
+		runtimePriority,
 	);
 
 	return {
@@ -1347,8 +1441,7 @@ function buildDynamicTask(
 		explicitWorktreePolicy: false,
 		runtime: {
 			approvalMode: "non-interactive",
-			model: defaultModel,
-			thinking: defaultThinking,
+			...controllerRuntime,
 			maxRuntimeMs:
 				dynamic.budget?.maxRuntimeMs ?? DEFAULT_DYNAMIC_MAX_RUNTIME_MS,
 		},
@@ -1392,6 +1485,9 @@ function buildDynamicTask(
 			helpers,
 			workflows,
 			...(decisionLoop ? { decisionLoop } : {}),
+			...(runtimePriority.runtimeOverrides
+				? { runtimeOverrides: runtimePriority.runtimeOverrides }
+				: {}),
 		},
 		...overrides,
 	};
@@ -1399,8 +1495,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(
@@ -1416,25 +1516,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
@@ -1489,8 +1582,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(
@@ -1499,20 +1596,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-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,