@bastani/atomic 0.9.16-alpha.2 → 0.9.16-alpha.3

package/dist/builtin/workflows/src/shared/persistence-restore-helpers.ts

@@ -1,613 +0,0 @@
-import { Value } from "typebox/value";
-import { isModelAttempts } from "../durable/dbos-envelope.js";
-import type { NormalizedSessionEntry as SessionEntry } from "./persistence-restore.js";
-import { workflowSerializableObjectSchema } from "./serializable.js";
-import type { Store } from "./store.js";
-import type {
-	RunBudgetSnapshot,
-	RunBudgetState,
-	RunStatus,
-	StageSnapshot,
-	StageStatus,
-	ToolNodeSnapshot,
-	WorkflowActor,
-	WorkflowFailureCode,
-	WorkflowFailureDisposition,
-	WorkflowFailureKind,
-} from "./store-types.js";
-import type { WorkflowExitStatus, WorkflowInputValues, WorkflowOutputValues } from "./types.js";
-import {
-	isWorkflowFailureCode,
-	isWorkflowFailureDisposition,
-	isWorkflowFailureKind,
-	isWorkflowFailureRecoverability,
-} from "./workflow-failures.js";
-
-interface RestoredRunBlockedMetadata {
-	readonly failedStageId?: string;
-	readonly error: string;
-	readonly failureKind: WorkflowFailureKind;
-	readonly failureCode?: WorkflowFailureCode;
-	readonly failureRecoverability: "recoverable";
-	readonly failureDisposition?: WorkflowFailureDisposition;
-	readonly failureMessage?: string;
-	readonly retryAfterMs?: number;
-	readonly resumable: true;
-	readonly endedAt?: number;
-	readonly result?: WorkflowOutputValues;
-	readonly budgetState?: RunBudgetState;
-	readonly ts: number;
-}
-
-export function _buildStageSnapshots(
-	entries: readonly SessionEntry[],
-	runId: string,
-	blockedMeta?: RestoredRunBlockedMetadata,
-): StageSnapshot[] {
-	const stageMap = new Map<string, StageSnapshot>();
-	const endedStages = new Set<string>();
-
-	for (const entry of entries) {
-		if (entry.payload.runId !== runId) continue;
-
-		if (entry.type === "workflow.stage.start") {
-			const stageId = entry.payload.stageId;
-			const name = entry.payload.name;
-			const parentIds = entry.payload.parentIds;
-			const ts = entry.payload.ts;
-			if (typeof stageId !== "string" || typeof name !== "string") continue;
-			if (!stageMap.has(stageId)) {
-				stageMap.set(stageId, {
-					id: stageId,
-					name,
-					status: "running",
-					parentIds: Array.isArray(parentIds) ? (parentIds as string[]) : [],
-					startedAt: typeof ts === "number" ? ts : undefined,
-					...replayMetadata(entry.payload),
-					toolEvents: [],
-				});
-			}
-		}
-
-		if (entry.type === "workflow.stage.end") {
-			const stageId = entry.payload.stageId;
-			const status = entry.payload.status;
-			const durationMs = entry.payload.durationMs;
-			const summary = entry.payload.summary;
-			const error = entry.payload.error;
-			const failureKind = entry.payload.failureKind;
-			const failureCode = entry.payload.failureCode;
-			const failureRecoverability = entry.payload.failureRecoverability;
-			const failureDisposition = entry.payload.failureDisposition;
-			const retryAfterMs = entry.payload.retryAfterMs;
-			const failureMessage = entry.payload.failureMessage;
-			const skippedReason = entry.payload.skippedReason;
-			const sessionId = entry.payload.sessionId;
-			const sessionFile = entry.payload.sessionFile;
-			const modelAttempts = entry.payload.modelAttempts;
-			if (typeof stageId !== "string") continue;
-			endedStages.add(stageId);
-			const snap = stageMap.get(stageId);
-			if (snap) {
-				snap.status = restoreStageStatus(status);
-				if (typeof durationMs === "number") snap.durationMs = durationMs;
-				if (typeof summary === "string") snap.result = summary;
-				if (typeof error === "string") snap.error = error;
-				if (typeof failureKind === "string" && isWorkflowFailureKind(failureKind)) snap.failureKind = failureKind;
-				if (typeof failureCode === "string" && isWorkflowFailureCode(failureCode)) snap.failureCode = failureCode;
-				if (typeof failureRecoverability === "string" && isWorkflowFailureRecoverability(failureRecoverability))
-					snap.failureRecoverability = failureRecoverability;
-				if (typeof failureDisposition === "string" && isWorkflowFailureDisposition(failureDisposition))
-					snap.failureDisposition = failureDisposition;
-				if (typeof retryAfterMs === "number") snap.retryAfterMs = retryAfterMs;
-				if (typeof failureMessage === "string") snap.failureMessage = failureMessage;
-				if (typeof skippedReason === "string") snap.skippedReason = skippedReason;
-				if (typeof sessionId === "string") snap.sessionId = sessionId;
-				if (typeof sessionFile === "string") snap.sessionFile = sessionFile;
-				if (isModelAttempts(modelAttempts)) snap.modelAttempts = modelAttempts as StageSnapshot["modelAttempts"];
-				Object.assign(snap, replayMetadata(entry.payload), workflowChildMetadata(entry.payload));
-			}
-		}
-	}
-
-	if (blockedMeta !== undefined) {
-		restoreBlockedStageState(stageMap, endedStages, blockedMeta);
-	} else {
-		// Mark any stage that didn't get an end entry as crashed.
-		for (const [stageId, snap] of stageMap) {
-			if (endedStages.has(stageId)) continue;
-			snap.status = "failed";
-			snap.error = "Stage did not complete — process was interrupted.";
-		}
-	}
-
-	return [...stageMap.values()];
-}
-
-function hasRestoredAncestor(
-	stageMap: ReadonlyMap<string, StageSnapshot>,
-	stage: StageSnapshot,
-	ancestorId: string,
-): boolean {
-	const queue = [...stage.parentIds];
-	const seen = new Set<string>();
-
-	while (queue.length > 0) {
-		const next = queue.shift();
-		if (next === undefined || seen.has(next)) continue;
-		if (next === ancestorId) return true;
-		seen.add(next);
-		queue.push(...(stageMap.get(next)?.parentIds ?? []));
-	}
-
-	return false;
-}
-
-function markRestoredBlockedFailureStage(snap: StageSnapshot, blockedMeta: RestoredRunBlockedMetadata): void {
-	snap.status = "failed";
-	snap.error = blockedMeta.error;
-	snap.failureKind = blockedMeta.failureKind;
-	snap.failureCode = blockedMeta.failureCode;
-	snap.failureRecoverability = blockedMeta.failureRecoverability;
-	snap.failureDisposition = blockedMeta.failureDisposition;
-	snap.failureMessage = blockedMeta.failureMessage;
-	snap.retryAfterMs = blockedMeta.retryAfterMs;
-}
-function restoreBlockedStageState(
-	stageMap: Map<string, StageSnapshot>,
-	endedStages: ReadonlySet<string>,
-	blockedMeta: RestoredRunBlockedMetadata,
-): void {
-	if (blockedMeta.failedStageId === undefined) return;
-	for (const [stageId, snap] of stageMap) {
-		if (endedStages.has(stageId)) continue;
-		if (stageId === blockedMeta.failedStageId) {
-			markRestoredBlockedFailureStage(snap, blockedMeta);
-			continue;
-		}
-		if (hasRestoredAncestor(stageMap, snap, blockedMeta.failedStageId)) {
-			snap.status = "blocked";
-			snap.blockedByStageId = blockedMeta.failedStageId;
-		}
-	}
-}
-
-function replayMetadata(
-	payload: Record<string, unknown>,
-): Pick<StageSnapshot, "replayKey" | "replayedFromStageId" | "replayed"> {
-	const replayKey = payload.replayKey;
-	const replayedFromStageId = payload.replayedFromStageId;
-	const replayed = payload.replayed;
-	return {
-		...(typeof replayKey === "string" ? { replayKey } : {}),
-		...(typeof replayedFromStageId === "string" ? { replayedFromStageId } : {}),
-		...(typeof replayed === "boolean" ? { replayed } : {}),
-	};
-}
-
-function isRecord(value: unknown): value is Record<string, unknown> {
-	return value !== null && typeof value === "object" && !Array.isArray(value);
-}
-
-function serializableObject(value: unknown): WorkflowOutputValues | undefined {
-	return Value.Check(workflowSerializableObjectSchema, value) ? (value as WorkflowOutputValues) : undefined;
-}
-
-export function serializableObjectOrEmpty(value: unknown): WorkflowOutputValues {
-	return serializableObject(value) ?? {};
-}
-
-function restoreBudgetSnapshot(value: unknown): RunBudgetSnapshot | undefined {
-	if (!isRecord(value)) return undefined;
-	const maxDurationMs = numericDuration(value.maxDurationMs);
-	const maxTokens = numericDuration(value.maxTokens),
-		maxCost = numericDuration(value.maxCost);
-	const warnAtPercent = numericDuration(value.warnAtPercent);
-	if (maxDurationMs === undefined || warnAtPercent === undefined) return undefined;
-	return {
-		maxDurationMs,
-		...(maxTokens !== undefined ? { maxTokens } : {}),
-		...(maxCost !== undefined ? { maxCost } : {}),
-		warnAtPercent,
-	};
-}
-export function restoreBudgetState(value: unknown): RunBudgetState | undefined {
-	if (!isRecord(value)) return undefined;
-	const report = (candidate: unknown, dimension: "duration" | "tokens" | "cost") => {
-		if (!isRecord(candidate) || (dimension !== "duration" && dimension !== "tokens" && dimension !== "cost"))
-			return undefined;
-		const reading = numericDuration(candidate.reading);
-		const ceiling = numericDuration(candidate.ceiling);
-		const percent = numericDuration(candidate.percent);
-		return reading === undefined || ceiling === undefined || percent === undefined
-			? undefined
-			: { dimension, reading, ceiling, percent };
-	};
-	const duration = report(value.duration, "duration") as RunBudgetState["duration"];
-	const tokens = report(value.tokens, "tokens") as RunBudgetState["tokens"];
-	const cost = report(value.cost, "cost") as RunBudgetState["cost"];
-	const warning = isRecord(value.warning)
-		? report(value.warning, value.warning.dimension as "duration" | "tokens" | "cost")
-		: undefined;
-	const warnings: Partial<Record<"duration" | "tokens" | "cost", RunBudgetState["warning"]>> = {};
-	const warningValues = isRecord(value.warnings) ? value.warnings : undefined;
-	for (const dimension of ["duration", "tokens", "cost"] as const) {
-		const restored = report(warningValues?.[dimension], dimension);
-		if (restored !== undefined) warnings[dimension] = restored;
-	}
-	const requiredNumbers = (candidate: unknown, fields: readonly string[]): boolean =>
-		isRecord(candidate) && fields.every((field) => numericDuration(candidate[field]) !== undefined);
-	const accountingValue = value.accounting;
-	const accounting =
-		isRecord(accountingValue) &&
-		requiredNumbers(accountingValue, ["tokens", "cost"]) &&
-		requiredNumbers(accountingValue.baseline, ["input", "output", "cacheRead", "cacheWrite", "cost"]) &&
-		requiredNumbers(accountingValue.perCounter, ["input", "output", "cacheRead", "cacheWrite"])
-			? (accountingValue as RunBudgetState["accounting"])
-			: undefined;
-	const state = {
-		...(duration !== undefined ? { duration } : {}),
-		...(tokens !== undefined ? { tokens } : {}),
-		...(cost !== undefined ? { cost } : {}),
-		...(warning !== undefined ? { warning } : {}),
-		...(Object.keys(warnings).length > 0 ? { warnings } : {}),
-		...(accounting !== undefined ? { accounting } : {}),
-		...(value.warned === true ? { warned: true } : {}),
-		...(value.wrapUpDelivered === true ? { wrapUpDelivered: true } : {}),
-		...(value.wrapUpCompleted === true ? { wrapUpCompleted: true } : {}),
-		...(value.systemOwnedStop === true ? { systemOwnedStop: true } : {}),
-		...(typeof value.wrapUpSummary === "string" ? { wrapUpSummary: value.wrapUpSummary } : {}),
-		...(isRecord(value.wrapUpUsage) ? { wrapUpUsage: value.wrapUpUsage as RunBudgetState["wrapUpUsage"] } : {}),
-	};
-	return Object.keys(state).length > 0 ? state : undefined;
-}
-function isWorkflowChildReplayStatus(status: unknown): status is WorkflowExitStatus {
-	return (
-		status === "completed" ||
-		status === "skipped" ||
-		status === "cancelled" ||
-		status === "blocked" ||
-		status === "failed"
-	);
-}
-
-function workflowChildMetadata(payload: Record<string, unknown>): Pick<StageSnapshot, "workflowChild"> {
-	if (payload.status !== "completed") return {};
-	const workflowChild = payload.workflowChild;
-	if (!isRecord(workflowChild)) return {};
-	const alias = workflowChild.alias;
-	const workflow = workflowChild.workflow;
-	const childRunId = workflowChild.runId;
-	const status = workflowChild.status;
-	const outputs = workflowChild.outputs;
-	const exited = workflowChild.exited;
-	const exitReason = workflowChild.exitReason;
-	if (
-		typeof alias !== "string" ||
-		typeof workflow !== "string" ||
-		typeof childRunId !== "string" ||
-		!isWorkflowChildReplayStatus(status) ||
-		!isRecord(outputs)
-	) {
-		return {};
-	}
-
-	// `structuredClone` detaches the restored snapshot from the parsed JSONL
-	// payload with a guaranteed deep copy, independent of the TypeBox
-	// serializable check. Declared `outputs` are the child contract, so a
-	// non-serializable value bails the whole child snapshot.
-	let clonedOutputs: WorkflowOutputValues;
-	try {
-		const serializableOutputs = serializableObject(outputs);
-		if (serializableOutputs === undefined) return {};
-		clonedOutputs = structuredClone(serializableOutputs);
-	} catch {
-		return {};
-	}
-
-	return {
-		workflowChild: {
-			alias,
-			workflow,
-			runId: childRunId,
-			status,
-			...(typeof exited === "boolean"
-				? { exited }
-				: status === "failed"
-					? {}
-					: status !== "completed" || typeof exitReason === "string"
-						? { exited: true }
-						: {}),
-			outputs: clonedOutputs,
-			...(typeof exitReason === "string" ? { exitReason } : {}),
-		},
-	};
-}
-
-function restoreStageStatus(status: unknown): StageStatus {
-	switch (status) {
-		case "completed":
-		case "failed":
-		case "skipped":
-		case "blocked":
-			return status;
-		default:
-			return "failed";
-	}
-}
-
-function numericRetryAfterMs(value: unknown): number | undefined {
-	return typeof value === "number" && Number.isFinite(value) && value >= 0 ? value : undefined;
-}
-
-function numericTimestamp(value: unknown): number | undefined {
-	return typeof value === "number" && Number.isFinite(value) && value >= 0 ? value : undefined;
-}
-
-function numericDuration(value: unknown): number | undefined {
-	return typeof value === "number" && Number.isFinite(value) && value >= 0 ? value : undefined;
-}
-
-function restoreFailedToolNode(value: unknown): ToolNodeSnapshot | undefined {
-	if (value === null || typeof value !== "object" || Array.isArray(value)) return undefined;
-	const node = value as Partial<ToolNodeSnapshot>;
-	if (
-		node.kind !== "tool" ||
-		typeof node.id !== "string" ||
-		typeof node.name !== "string" ||
-		typeof node.argsHash !== "string" ||
-		typeof node.ordinal !== "number" ||
-		!Number.isInteger(node.ordinal) ||
-		!Array.isArray(node.parentIds) ||
-		!node.parentIds.every((parentId) => typeof parentId === "string") ||
-		node.status !== "failed" ||
-		typeof node.error !== "string"
-	)
-		return undefined;
-	return {
-		kind: "tool",
-		id: node.id,
-		name: node.name,
-		argsHash: node.argsHash,
-		ordinal: node.ordinal,
-		parentIds: Object.freeze([...node.parentIds]),
-		status: "failed",
-		...(typeof node.executionOrder === "number" && Number.isFinite(node.executionOrder)
-			? { executionOrder: node.executionOrder }
-			: {}),
-		...(numericTimestamp(node.startedAt) !== undefined ? { startedAt: node.startedAt } : {}),
-		...(numericTimestamp(node.endedAt) !== undefined ? { endedAt: node.endedAt } : {}),
-		error: node.error,
-		attachable: false,
-	};
-}
-
-export function findRunBlockedMetadata(
-	entries: readonly SessionEntry[],
-	runId: string,
-): RestoredRunBlockedMetadata | undefined {
-	let latest: RestoredRunBlockedMetadata | undefined;
-	for (const entry of entries) {
-		if (entry.type !== "workflow.run.blocked" || entry.payload.runId !== runId) continue;
-		const failedStageId = entry.payload.failedStageId;
-		const error = entry.payload.error;
-		const failureKind = entry.payload.failureKind;
-		const failureCode = entry.payload.failureCode;
-		const failureRecoverability = entry.payload.failureRecoverability;
-		const failureDisposition = entry.payload.failureDisposition;
-		const failureMessage = entry.payload.failureMessage;
-		const retryAfterMs = numericRetryAfterMs(entry.payload.retryAfterMs);
-		const resumable = entry.payload.resumable;
-		const ts = entry.payload.ts;
-		const result = serializableObject(entry.payload.result);
-		const budgetState = restoreBudgetState(entry.payload.budgetState);
-		const isBudgetStop = budgetState?.systemOwnedStop === true && result?.status === "budget_exceeded";
-		if (
-			typeof error !== "string" ||
-			typeof failureKind !== "string" ||
-			!isWorkflowFailureKind(failureKind) ||
-			failureRecoverability !== "recoverable" ||
-			resumable !== true ||
-			typeof ts !== "number" ||
-			(!isBudgetStop && typeof failedStageId !== "string")
-		)
-			continue;
-		latest = {
-			...(typeof failedStageId === "string" ? { failedStageId } : {}),
-			error,
-			failureKind,
-			...(typeof failureCode === "string" && isWorkflowFailureCode(failureCode) ? { failureCode } : {}),
-			failureRecoverability: "recoverable",
-			...(typeof failureDisposition === "string" && isWorkflowFailureDisposition(failureDisposition)
-				? { failureDisposition }
-				: {}),
-			...(typeof failureMessage === "string" ? { failureMessage } : {}),
-			...(retryAfterMs !== undefined ? { retryAfterMs } : {}),
-			resumable: true,
-			...(typeof entry.payload.endedAt === "number" ? { endedAt: entry.payload.endedAt } : {}),
-			...(result !== undefined ? { result } : {}),
-			...(budgetState !== undefined ? { budgetState } : {}),
-			ts,
-		};
-	}
-	return latest;
-}
-export function restoreTerminalRuns(entries: readonly SessionEntry[], store: Store): void {
-	const started = new Map<
-		string,
-		{ readonly name: string; readonly inputs: Readonly<WorkflowInputValues>; readonly startTs: number }
-	>();
-	const ended = new Map<string, Record<string, unknown>>();
-
-	for (const entry of entries) {
-		if (entry.type === "workflow.run.start") {
-			const runId = entry.payload.runId;
-			const name = entry.payload.name;
-			const inputs = entry.payload.inputs;
-			const ts = entry.payload.ts;
-			if (typeof runId === "string" && typeof name === "string" && typeof ts === "number") {
-				started.set(runId, {
-					name,
-					inputs: serializableObjectOrEmpty(inputs),
-					startTs: ts,
-				});
-			}
-		}
-		if (entry.type === "workflow.run.end") {
-			const runId = entry.payload.runId;
-			if (typeof runId === "string") ended.set(runId, entry.payload);
-		}
-	}
-
-	for (const [runId, start] of started) {
-		if (store.runs().some((run) => run.id === runId)) continue;
-		const end = ended.get(runId);
-		const status = restoreTerminalRunStatus(end?.status);
-		if (end === undefined || status === undefined) continue;
-
-		const runMeta = findRunStartMetadata(entries, runId);
-		const stages = _buildStageSnapshots(entries, runId);
-		const exited = end.exited;
-		const exitReason = end.exitReason;
-		const resumable = end.resumable;
-		const failedToolNodeId = end.failedToolNodeId;
-		const failedToolNode = restoreFailedToolNode(end.failedToolNode);
-		// Failed is both an author-exit status and the engine's ordinary failure
-		// status, so an exit reason alone cannot establish author intent.
-		const restoredAuthorExit =
-			isWorkflowExitTerminalStatus(status) &&
-			(exited === true || (status !== "failed" && typeof exitReason === "string"));
-		if (status === "completed" && !restoredAuthorExit && stages.some((stage) => stage.status !== "completed"))
-			continue;
-		store.recordRunStart({
-			id: runId,
-			name: start.name,
-			inputs: start.inputs,
-			status: "running",
-			stages,
-			...(typeof failedToolNodeId === "string" && failedToolNode?.id === failedToolNodeId
-				? { toolNodes: [failedToolNode] }
-				: {}),
-			startedAt: start.startTs,
-			...(runMeta.parentRunId !== undefined ? { parentRunId: runMeta.parentRunId } : {}),
-			...(runMeta.parentStageId !== undefined ? { parentStageId: runMeta.parentStageId } : {}),
-			...(runMeta.rootRunId !== undefined ? { rootRunId: runMeta.rootRunId } : {}),
-			...(runMeta.resumedFromRunId !== undefined ? { resumedFromRunId: runMeta.resumedFromRunId } : {}),
-			...(runMeta.resumeFromStageId !== undefined ? { resumeFromStageId: runMeta.resumeFromStageId } : {}),
-			...(runMeta.origin !== undefined ? { origin: runMeta.origin } : {}),
-			...(runMeta.accumulatedDurationMs !== undefined
-				? { accumulatedDurationMs: runMeta.accumulatedDurationMs }
-				: {}),
-			...(runMeta.budget !== undefined ? { budget: runMeta.budget } : {}),
-			...(runMeta.budgetState !== undefined ? { budgetState: runMeta.budgetState } : {}),
-		});
-
-		const error = end.error;
-		const result = serializableObject(end.result);
-		const failureKind = end.failureKind;
-		const failureCode = end.failureCode;
-		const failureRecoverability = end.failureRecoverability;
-		const failureDisposition = end.failureDisposition;
-		const retryAfterMs = numericRetryAfterMs(end.retryAfterMs);
-		const failureMessage = end.failureMessage;
-		const failedStageId = end.failedStageId;
-		const restoredEndedAt = numericTimestamp(end.endedAt);
-		const restoredDurationMs = numericDuration(end.durationMs);
-		store.recordRunEnd(runId, status, result, typeof error === "string" ? error : undefined, {
-			...(typeof failureKind === "string" && isWorkflowFailureKind(failureKind) ? { failureKind } : {}),
-			...(typeof failureCode === "string" && isWorkflowFailureCode(failureCode) ? { failureCode } : {}),
-			...(typeof failureRecoverability === "string" && isWorkflowFailureRecoverability(failureRecoverability)
-				? { failureRecoverability }
-				: {}),
-			...(typeof failureDisposition === "string" && isWorkflowFailureDisposition(failureDisposition)
-				? { failureDisposition }
-				: {}),
-			...(retryAfterMs !== undefined ? { retryAfterMs } : {}),
-			...(typeof failureMessage === "string" ? { failureMessage } : {}),
-			...(typeof failedStageId === "string" ? { failedStageId } : {}),
-			...(typeof failedToolNodeId === "string" ? { failedToolNodeId } : {}),
-			...(typeof resumable === "boolean"
-				? { resumable }
-				: isWorkflowExitTerminalStatus(status) && restoredAuthorExit
-					? { resumable: false }
-					: {}),
-			...(restoredAuthorExit && isWorkflowExitTerminalStatus(status) ? { exited: true } : {}),
-			...(typeof exitReason === "string" ? { exitReason } : {}),
-		});
-		const restoredRun = store.runs().find((run) => run.id === runId);
-		if (restoredRun !== undefined && (restoredEndedAt !== undefined || restoredDurationMs !== undefined)) {
-			Object.assign(restoredRun, {
-				...(restoredEndedAt !== undefined ? { endedAt: restoredEndedAt } : {}),
-				...(restoredDurationMs !== undefined ? { durationMs: restoredDurationMs } : {}),
-			});
-		}
-	}
-}
-
-function isWorkflowExitTerminalStatus(status: RunStatus): status is WorkflowExitStatus {
-	return (
-		status === "completed" ||
-		status === "skipped" ||
-		status === "cancelled" ||
-		status === "blocked" ||
-		status === "failed"
-	);
-}
-
-function restoreTerminalRunStatus(status: unknown): RunStatus | undefined {
-	switch (status) {
-		case "completed":
-		case "skipped":
-		case "cancelled":
-		case "blocked":
-		case "failed":
-		case "killed":
-			return status;
-		default:
-			return undefined;
-	}
-}
-
-export function findRunStartMetadata(
-	entries: readonly SessionEntry[],
-	runId: string,
-): {
-	readonly parentRunId?: string;
-	readonly parentStageId?: string;
-	readonly rootRunId?: string;
-	readonly resumedFromRunId?: string;
-	readonly resumeFromStageId?: string;
-	readonly accumulatedDurationMs?: number;
-	readonly origin?: WorkflowActor;
-	readonly budget?: RunBudgetSnapshot;
-	readonly budgetState?: RunBudgetState;
-} {
-	for (const entry of entries) {
-		if (entry.type !== "workflow.run.start" || entry.payload.runId !== runId) continue;
-		const parentRunId = entry.payload.parentRunId;
-		const parentStageId = entry.payload.parentStageId;
-		const rootRunId = entry.payload.rootRunId;
-		const resumedFromRunId = entry.payload.resumedFromRunId;
-		const resumeFromStageId = entry.payload.resumeFromStageId;
-		const accumulatedDurationMs = entry.payload.accumulatedDurationMs;
-		const budget = restoreBudgetSnapshot(entry.payload.budget);
-		const budgetState = restoreBudgetState(entry.payload.budgetState);
-		const origin = entry.payload.origin;
-		return {
-			...(typeof parentRunId === "string" ? { parentRunId } : {}),
-			...(typeof parentStageId === "string" ? { parentStageId } : {}),
-			...(typeof rootRunId === "string" ? { rootRunId } : {}),
-			...(typeof resumedFromRunId === "string" ? { resumedFromRunId } : {}),
-			...(typeof resumeFromStageId === "string" ? { resumeFromStageId } : {}),
-			...(typeof accumulatedDurationMs === "number" &&
-			Number.isFinite(accumulatedDurationMs) &&
-			accumulatedDurationMs > 0
-				? { accumulatedDurationMs }
-				: {}),
-			...(origin === "user" || origin === "agent" ? { origin } : {}),
-			...(budget !== undefined ? { budget } : {}),
-			...(budgetState !== undefined ? { budgetState } : {}),
-		};
-	}
-	return {};
-}