@bastani/atomic 0.9.14-alpha.4 → 0.9.14-alpha.5

package/dist/builtin/workflows/src/extension/config-file-loader.ts

@@ -1,3 +1,4 @@
+import { validateWorkflowBudget } from "../shared/budget.js";
 import type { ConfigDiagnostic, WorkflowExtensionConfig } from "./config-loader.js";
 import { WORKFLOW_LIFECYCLE_NOTICE_KINDS, type WorkflowLifecycleNoticeKind } from "./lifecycle-notifications.js";
 
@@ -63,6 +64,11 @@ function validateConfig(value: unknown): string | null {
 		}
 	}
 
+	if ("budget" in c) {
+		const reason = validateWorkflowBudget(c.budget, "budget");
+		if (reason !== null) return reason;
+	}
+
 	if ("workflowNotifications" in c) {
 		const value = c.workflowNotifications;
 		if (value === null || typeof value !== "object" || Array.isArray(value)) {

package/dist/builtin/workflows/src/extension/config-loader.ts

@@ -18,6 +18,7 @@
 
 import { isAbsolute, join } from "node:path";
 import { CONFIG_DIR_NAME, CONFIG_DIR_NAMES, getAgentDir, getAgentDirs, getProjectConfigPaths } from "@bastani/atomic";
+import { type EffectiveBudget, resolve_budget, type WorkflowBudget } from "../shared/budget.js";
 import { loadConfigFile } from "./config-file-loader.js";
 import type { WorkflowLifecycleNoticeKind } from "./lifecycle-notifications.js";
 
@@ -49,6 +50,8 @@ export interface WorkflowWorktreeConfig {
 export interface WorkflowExtensionConfig {
 	/** Explicit named workflows to register by module path. */
 	readonly workflows?: Readonly<Record<string, WorkflowConfigEntry>>;
+	/** Default budget for every workflow run. Each `0` disables its dimension. */
+	readonly budget?: WorkflowBudget;
 	/** Maximum workflow recursion depth. Default: 4. */
 	readonly maxDepth?: number;
 	/** Default stage concurrency. Default: 4. */
@@ -135,6 +138,7 @@ export interface LoadWorkflowConfigOpts {
 function mergeConfigs(base: WorkflowExtensionConfig, override: WorkflowExtensionConfig): WorkflowExtensionConfig {
 	const workflows =
 		base.workflows || override.workflows ? { ...(base.workflows ?? {}), ...(override.workflows ?? {}) } : undefined;
+	const budget = base.budget || override.budget ? { ...(base.budget ?? {}), ...(override.budget ?? {}) } : undefined;
 
 	return {
 		...(base.maxDepth !== undefined || override.maxDepth !== undefined
@@ -163,6 +167,7 @@ function mergeConfigs(base: WorkflowExtensionConfig, override: WorkflowExtension
 		...(base.worktree !== undefined || override.worktree !== undefined
 			? { worktree: { ...(base.worktree ?? {}), ...(override.worktree ?? {}) } }
 			: {}),
+		...(budget !== undefined ? { budget } : {}),
 		...(workflows !== undefined ? { workflows } : {}),
 	};
 }
@@ -184,9 +189,25 @@ export const WORKFLOW_CONFIG_DEFAULTS = {
 	persistRuns: true,
 	statusFile: false,
 	resumeInFlight: "ask" as const,
+	budget: {
+		maxDurationMs: 0,
+		maxTokens: 0,
+		maxCost: 0,
+		warnAtPercent: 80,
+	},
 	workflowNotifications: {
 		enabled: true,
-		notifyOn: ["started", "completed", "failed", "blocked", "awaiting_input", "paused", "quit", "resumed"] as const,
+		notifyOn: [
+			"started",
+			"completed",
+			"failed",
+			"blocked",
+			"budget_warning",
+			"awaiting_input",
+			"paused",
+			"quit",
+			"resumed",
+		] as const,
 	},
 	worktree: {
 		symlinkDirectories: ["node_modules"] as readonly string[],
@@ -203,6 +224,7 @@ export interface WorkflowEffectiveConfig {
 	readonly persistRuns: boolean;
 	readonly statusFile: boolean;
 	readonly resumeInFlight: "ask" | "auto" | "never";
+	readonly budget: EffectiveBudget;
 	readonly workflowNotifications: {
 		readonly enabled: boolean;
 		readonly notifyOn: readonly WorkflowLifecycleNoticeKind[];
@@ -226,6 +248,7 @@ export function withWorkflowDefaults(config: WorkflowExtensionConfig): WorkflowE
 		persistRuns: config.persistRuns ?? WORKFLOW_CONFIG_DEFAULTS.persistRuns,
 		statusFile: config.statusFile ?? WORKFLOW_CONFIG_DEFAULTS.statusFile,
 		resumeInFlight: config.resumeInFlight ?? WORKFLOW_CONFIG_DEFAULTS.resumeInFlight,
+		budget: resolve_budget({ config: config.budget ?? WORKFLOW_CONFIG_DEFAULTS.budget }),
 		workflowNotifications: {
 			enabled: config.workflowNotifications?.enabled ?? WORKFLOW_CONFIG_DEFAULTS.workflowNotifications.enabled,
 			notifyOn: config.workflowNotifications?.notifyOn ?? WORKFLOW_CONFIG_DEFAULTS.workflowNotifications.notifyOn,

package/dist/builtin/workflows/src/extension/dispatcher.ts

@@ -13,6 +13,7 @@ import { jobTracker as defaultJobTracker, type JobTracker } from "../runs/backgr
 import { launchDetachedUntilStartup, workflowStartupFailureMessage } from "../runs/background/startup-admission.js";
 import { resolveAndValidateInputs } from "../runs/foreground/executor.js";
 import type { StageAdapters } from "../runs/foreground/stage-runner.js";
+import { resolve_budget } from "../shared/budget.js";
 import { effectiveRunStatus } from "../shared/returned-run-status.js";
 import { deriveInputFields, schemaIsRequired } from "../shared/schema-introspection.js";
 import { store as defaultStore, type Store } from "../shared/store.js";
@@ -163,13 +164,12 @@ export async function dispatch(args: WorkflowToolArgs, opts: DispatcherOpts): Pr
 			}
 
 			const policy = opts.policy ?? INTERACTIVE_WORKFLOW_POLICY;
-
-			// Pre-validate inputs against the workflow's declared schema. The
-			// executor would otherwise throw the same TypeError deep inside the
-			// background promise — silently from the caller's perspective. Catch
-			// it here so the dispatch result names what's missing.
+			// Pre-validate inputs and budget declarations. The executor would otherwise
+			// reject them deep in the background promise, silently from the caller's
+			// perspective. Catch errors here so the dispatch result names the problem.
 			try {
 				resolveAndValidateInputs(def.inputs, inputs, `workflow "${def.name}"`);
+				resolve_budget({ config: opts.config?.budget, definition: def.budget, run: args.budget });
 			} catch (err) {
 				return failedRunResult(def.name, "", err instanceof Error ? err.message : String(err));
 			}
@@ -186,6 +186,7 @@ export async function dispatch(args: WorkflowToolArgs, opts: DispatcherOpts): Pr
 					persistence: opts.persistence,
 					mcp: opts.mcp,
 					config: opts.config,
+					budget: args.budget,
 					models: opts.models,
 					executionMode: policy.mode,
 					cwd: opts.cwd,

package/dist/builtin/workflows/src/extension/extension-runtime-state.ts

@@ -127,6 +127,7 @@ export function createWorkflowExtensionRuntimeState(
 			persistRuns: WORKFLOW_CONFIG_DEFAULTS.persistRuns,
 			statusFile: WORKFLOW_CONFIG_DEFAULTS.statusFile,
 			resumeInFlight: WORKFLOW_CONFIG_DEFAULTS.resumeInFlight,
+			budget: withWorkflowDefaults({}).budget,
 			worktree: WORKFLOW_CONFIG_DEFAULTS.worktree,
 		},
 	};
@@ -355,6 +356,7 @@ export function createWorkflowExtensionRuntimeState(
 			persistRuns: effectiveConfig.persistRuns,
 			statusFile: effectiveConfig.statusFile,
 			resumeInFlight: effectiveConfig.resumeInFlight,
+			budget: effectiveConfig.budget,
 			worktree: effectiveConfig.worktree,
 		};
 		lifecycleNotificationConfigRef.current = effectiveConfig.workflowNotifications;