experimental-a2 0.2.0 → 0.4.0

package/dist/ai.js

@@ -1,5 +1,6 @@
-import { t as assertSyncSchema } from "./validate-XKT4FSNn.js";
-import { t as contract } from "./contract-CG_adnu_.js";
+import { n as validateSync, t as assertSyncSchema } from "./validate-XKT4FSNn.js";
+import { t as contract } from "./contract-48bUMgcL.js";
+import { l as ambientToolScopeStorage } from "./internal-aEotMzu_.js";
 //#region src/ai-projector.ts
 const isRecord$1 = (value) => typeof value === "object" && value !== null && !Array.isArray(value);
 const cloneMessage = (messageId, base) => structuredClone(base ?? {
@@ -423,15 +424,18 @@ const messageValue = (value) => isRecord(value) && typeof value["id"] === "strin
 const validateMessage = (value, messageSchema) => {
 	if (!messageValue(value)) return issue("invalid AI SDK UIMessage");
 	if (!messageSchema) return { value };
-	const result = messageSchema["~standard"].validate(value);
-	if (result instanceof Promise) throw new TypeError("the AI message schema must validate synchronously");
-	return result;
+	return validateSync(messageSchema, value, "the AI message");
 };
 const messagePayloadSchema = (messageSchema) => schema("ai.message.created", (value) => {
 	if (!isRecord(value)) return issue("invalid ai.message.created payload");
 	const message = validateMessage(value["message"], messageSchema);
 	if (message.issues) return { issues: message.issues };
-	return { value: { message: message.value } };
+	const generate = value["generate"];
+	if (generate !== void 0 && typeof generate !== "boolean") return issue("invalid ai.message.created generate flag");
+	return { value: {
+		message: message.value,
+		...generate === false ? { generate: false } : {}
+	} };
 });
 const progressPayloadSchema = schema("ai.generation.progress", (value) => {
 	if (!isRecord(value) || !isJSONCompatible(value) || !stringField(value, "requestId") || !stringField(value, "messageId") || !stringField(value, "generationId") || !stringField(value, "responseMessageId") || typeof value["sequence"] !== "number" || !Number.isInteger(value["sequence"]) || value["sequence"] < 0 || !Array.isArray(value["chunks"]) || !value["chunks"].every((chunk) => isRecord(chunk) && typeof chunk["type"] === "string")) return issue("invalid ai.generation.progress payload");
@@ -464,7 +468,7 @@ function createEvents(options) {
 		"ai.session.closed": jsonObject("ai.session.closed", (value) => optionalStringField(value, "reason")),
 		"ai.message.created": messagePayloadSchema(messageSchema),
 		"ai.message.completed": jsonObject("ai.message.completed", (value) => stringField(value, "messageId")),
-		"ai.message.interrupted": jsonObject("ai.message.interrupted", (value) => stringField(value, "messageId") && optionalStringField(value, "generationId") && optionalStringField(value, "reason") && (value["lastSeenIndex"] === void 0 || typeof value["lastSeenIndex"] === "number" && Number.isInteger(value["lastSeenIndex"]) && value["lastSeenIndex"] >= 0)),
+		"ai.message.interrupted": jsonObject("ai.message.interrupted", (value) => stringField(value, "messageId") && (stringField(value, "generationId") && value["requestId"] === void 0 || stringField(value, "requestId") && value["generationId"] === void 0) && optionalStringField(value, "reason") && (value["lastSeenIndex"] === void 0 || typeof value["lastSeenIndex"] === "number" && Number.isInteger(value["lastSeenIndex"]) && value["lastSeenIndex"] >= 0)),
 		"ai.generation.requested": jsonObject("ai.generation.requested", (value) => stringField(value, "messageId") && optionalStringField(value, "responseMessageId") && (value["reason"] === "message" || value["reason"] === "tool" || value["reason"] === "input" || value["reason"] === "retry")),
 		"ai.retry.requested": jsonObject("ai.retry.requested", (value) => stringField(value, "messageId") && stringField(value, "responseMessageId") && stringField(value, "retryId")),
 		"ai.generation.started": jsonObject("ai.generation.started", (value) => stringField(value, "requestId") && stringField(value, "messageId") && stringField(value, "generationId") && stringField(value, "responseMessageId") && typeof value["attempt"] === "number" && Number.isInteger(value["attempt"]) && value["attempt"] > 0 && stringField(value, "model")),
@@ -583,7 +587,7 @@ const toolResultChunk = (payload) => {
 };
 const waitingStatus = (state) => {
 	if (state.pendingApprovals.length > 0 || state.pendingInputs.length > 0) return "waiting";
-	return state.activeGeneration !== null || state.tools.some((tool) => tool.status === "running") ? "generating" : "idle";
+	return state.activeGeneration !== null || state.tools.some((activity) => activity.status === "running") ? "generating" : "idle";
 };
 const lifecycleStatus = (previous, next) => previous.status === "failed" ? "failed" : waitingStatus(next);
 const isStaleGeneration = (state, messageId, generationId) => {
@@ -610,12 +614,12 @@ const reduceToolActivity = (tools, event) => {
 	if (event.type === "ai.tool.called") {
 		const payload = event.payload;
 		const { providerMetadata, ...called } = payload;
-		const tool = {
+		const activity = {
 			...called,
 			...providerMetadata === void 0 ? {} : { callProviderMetadata: providerMetadata },
 			status: "running"
 		};
-		return [...tools.filter((candidate) => candidate.generationId !== payload.generationId || candidate.toolCallId !== payload.toolCallId), tool];
+		return [...tools.filter((candidate) => candidate.generationId !== payload.generationId || candidate.toolCallId !== payload.toolCallId), activity];
 	}
 	const payload = event.payload;
 	const existing = tools.find((candidate) => candidate.generationId === payload.generationId && candidate.toolCallId === payload.toolCallId);
@@ -625,7 +629,7 @@ const reduceToolActivity = (tools, event) => {
 	const providerExecuted = payload.providerExecuted ?? existing?.providerExecuted;
 	const toolMetadata = payload.toolMetadata ?? existing?.toolMetadata;
 	const resultProviderMetadata = payload.providerMetadata ?? existing?.resultProviderMetadata;
-	const tool = {
+	const activity = {
 		requestId: payload.requestId,
 		messageId: payload.messageId,
 		generationId: payload.generationId,
@@ -645,19 +649,60 @@ const reduceToolActivity = (tools, event) => {
 		...existing?.title === void 0 ? {} : { title: existing.title },
 		status: payload.denied ? "denied" : payload.error ? "failed" : payload.preliminary ? "running" : "completed"
 	};
-	return [...tools.filter((candidate) => candidate.generationId !== payload.generationId || candidate.toolCallId !== payload.toolCallId), tool];
+	return [...tools.filter((candidate) => candidate.generationId !== payload.generationId || candidate.toolCallId !== payload.toolCallId), activity];
 };
 const projectedTools = (projection) => projection.toolEvents.reduce((tools, event) => reduceToolActivity(tools, event), projection.baseTools);
-const stopGenerationTools = (tools, generationId, responseMessageId, error) => tools.map((tool) => tool.generationId === generationId || tool.messageId === responseMessageId ? {
-	...tool,
+const restoreGenerationBase = (state, generationId, responseMessageId, reason) => {
+	const projection = state.activeProjection?.generationId === generationId ? state.activeProjection : null;
+	return {
+		...state,
+		messages: projection?.baseMessage === void 0 ? state.messages.filter((message) => message.id !== responseMessageId) : upsertMessage(state.messages, projection.baseMessage),
+		tools: projection === null ? state.tools.filter((activity) => activity.generationId !== generationId) : projection.baseTools,
+		pendingApprovals: state.pendingApprovals.filter((approval) => approval.generationId !== generationId),
+		pendingInputs: state.pendingInputs.filter((input) => input.generationId !== generationId),
+		activeGeneration: null,
+		activeProjection: null,
+		terminalGenerations: {
+			...state.terminalGenerations,
+			[generationId]: reason
+		},
+		compaction: projection === null ? state.compaction?.generationId === generationId ? null : state.compaction : projection.baseCompaction ?? null
+	};
+};
+const stopGenerationTools = (tools, generationId, responseMessageId, error) => tools.map((activity) => activity.status === "running" && (generationId === void 0 ? activity.messageId === responseMessageId : activity.generationId === generationId) ? {
+	...activity,
 	status: "failed",
 	error,
 	preliminary: false
-} : tool);
+} : activity);
+const interruptGenerationProjection = (state, generationId, responseMessageId, cutoff) => {
+	const projection = state.activeProjection?.generationId === generationId ? state.activeProjection : null;
+	if (projection === null) return null;
+	const retained = {
+		...projection,
+		batches: projection.batches.filter((batch) => batch.index <= cutoff),
+		toolEvents: projection.toolEvents.filter((toolEvent) => toolEvent.index <= cutoff),
+		approvalEvents: projection.approvalEvents.filter((approvalEvent) => approvalEvent.index <= cutoff)
+	};
+	const shouldProject = retained.baseMessage !== void 0 || retained.batches.length > 0;
+	const restored = restoreGenerationBase(state, generationId, responseMessageId, "interrupted");
+	return {
+		...restored,
+		messages: shouldProject ? upsertMessage(restored.messages, interruptUIMessage(projectedMessage(retained))) : restored.messages,
+		tools: stopGenerationTools(projectedTools(retained), generationId, responseMessageId, "Tool execution was interrupted."),
+		usage: projection.completionIndex === void 0 || projection.completionIndex <= cutoff ? restored.usage : restored.usage.filter((entry) => entry.generationId !== generationId)
+	};
+};
 const generationRequestId = (generationId) => {
 	const markerIndex = generationId.lastIndexOf(":generation:");
 	return markerIndex === -1 ? void 0 : generationId.slice(0, markerIndex);
 };
+const terminalGenerationReason = (state, generationId) => {
+	const terminal = state.terminalGenerations?.[generationId];
+	if (terminal !== void 0) return terminal;
+	const requestId = generationRequestId(generationId);
+	return requestId !== void 0 && state.terminalRequestIds?.[requestId] === true ? "interrupted" : void 0;
+};
 /** Pure projection of the built-in AI protocol. Unknown extension events are ignored. */
 function reduceAIState(state, event) {
 	if (state.status === "closed") return state;
@@ -687,6 +732,26 @@ function reduceAIState(state, event) {
 				messages: upsertMessage(state.messages, message)
 			};
 		}
+		case "ai.message.completed": {
+			const { messageId } = event.payload;
+			const generationId = state.responseGenerationIds[messageId];
+			if (generationId === void 0 || terminalGenerationReason(state, generationId) !== void 0) return state;
+			const requestId = generationRequestId(generationId);
+			return {
+				...state,
+				activeRequestId: null,
+				activeResponseMessageId: null,
+				activeProjection: state.activeProjection?.generationId === generationId ? null : state.activeProjection,
+				terminalGenerations: {
+					...state.terminalGenerations,
+					[generationId]: "completed"
+				},
+				...requestId === void 0 ? {} : { terminalRequestIds: {
+					...state.terminalRequestIds,
+					[requestId]: true
+				} }
+			};
+		}
 		case "ai.generation.requested": {
 			const payload = event.payload;
 			const schedulesToolContinuation = payload.reason === "tool";
@@ -694,10 +759,12 @@ function reduceAIState(state, event) {
 			const ownerGenerationId = state.responseGenerationIds[responseMessageId] ?? state.activeGeneration?.generationId;
 			const ownerRequestId = ownerGenerationId === void 0 ? void 0 : generationRequestId(ownerGenerationId);
 			const requestId = event.id;
+			if (requestId !== void 0 && state.terminalRequestIds?.[requestId] === true) return state;
 			if (!(payload.reason === "message" || (payload.reason === "tool" ? ownerGenerationId !== void 0 && requestId === `ai.generate:tools:${ownerGenerationId}` && state.activeRequestId === ownerRequestId && state.status !== "failed" : payload.reason === "retry" ? state.status === "failed" : ownerGenerationId !== void 0 && requestId?.startsWith(`ai.generate:input:${encodeURIComponent(responseMessageId)}:${encodeURIComponent(ownerGenerationId)}:`) === true && state.activeRequestId === ownerRequestId && state.status !== "failed"))) return state;
 			return {
 				...state,
 				messages: payload.reason === "retry" && payload.responseMessageId !== void 0 ? state.messages.filter((message) => message.id !== payload.responseMessageId) : state.messages,
+				activeProjection: payload.reason === "retry" ? null : state.activeProjection,
 				status: schedulesToolContinuation ? state.status : "generating",
 				activeRequestId: requestId ?? null,
 				activeResponseMessageId: responseMessageId,
@@ -706,27 +773,33 @@ function reduceAIState(state, event) {
 		}
 		case "ai.generation.started": {
 			const payload = event.payload;
+			if (state.terminalRequestIds?.[payload.requestId] === true || terminalGenerationReason(state, payload.generationId) !== void 0) return state;
 			if (state.activeRequestId !== null && state.activeRequestId !== payload.requestId) return state;
 			const currentOwner = state.responseGenerationIds[payload.responseMessageId];
+			const currentTerminal = currentOwner === void 0 ? void 0 : terminalGenerationReason(state, currentOwner);
+			if (currentTerminal === "completed" || currentTerminal === "interrupted") return state;
 			const marker = ":generation:";
 			const currentAttempt = currentOwner?.startsWith(`${payload.requestId}${marker}`) ? Number(currentOwner.slice(`${payload.requestId}${marker}`.length)) : void 0;
 			if (currentAttempt !== void 0 && Number.isFinite(currentAttempt) && currentAttempt >= payload.attempt) return state;
-			const baseMessage = state.messages.find((message) => message.id === payload.responseMessageId);
+			const activeGeneration = state.activeGeneration;
+			const baseState = activeGeneration !== null && activeGeneration.requestId === payload.requestId && activeGeneration.responseMessageId === payload.responseMessageId && activeGeneration.attempt < payload.attempt && state.activeProjection?.generationId === activeGeneration.generationId ? restoreGenerationBase(state, activeGeneration.generationId, activeGeneration.responseMessageId, "superseded") : state;
+			const baseMessage = baseState.messages.find((message) => message.id === payload.responseMessageId);
 			return {
-				...state,
+				...baseState,
 				status: "generating",
 				activeGeneration: payload,
 				activeResponseMessageId: null,
 				responseGenerationIds: {
-					...state.responseGenerationIds,
+					...baseState.responseGenerationIds,
 					[payload.responseMessageId]: payload.generationId
 				},
 				activeProjection: {
 					generationId: payload.generationId,
 					responseMessageId: payload.responseMessageId,
 					...baseMessage === void 0 ? {} : { baseMessage },
+					baseCompaction: baseState.compaction,
 					batches: [],
-					baseTools: state.tools,
+					baseTools: baseState.tools,
 					toolEvents: [],
 					approvalEvents: []
 				},
@@ -735,6 +808,7 @@ function reduceAIState(state, event) {
 		}
 		case "ai.generation.progress": {
 			const payload = event.payload;
+			if (terminalGenerationReason(state, payload.generationId) !== void 0) return state;
 			const activeProjection = state.activeProjection;
 			if (state.activeGeneration?.generationId !== payload.generationId || activeProjection?.generationId !== payload.generationId) return state;
 			const projection = {
@@ -753,6 +827,7 @@ function reduceAIState(state, event) {
 		}
 		case "ai.generation.completed": {
 			const payload = event.payload;
+			if (terminalGenerationReason(state, payload.generationId) !== void 0) return state;
 			if (state.activeGeneration?.generationId !== payload.generationId) return state;
 			const usage = payload.usage ? [...state.usage.filter((entry) => entry.generationId !== payload.generationId), {
 				generationId: payload.generationId,
@@ -761,7 +836,10 @@ function reduceAIState(state, event) {
 			const next = {
 				...state,
 				activeGeneration: null,
-				activeProjection: null,
+				activeProjection: state.activeProjection?.generationId === payload.generationId ? {
+					...state.activeProjection,
+					completionIndex: event.index
+				} : state.activeProjection,
 				usage
 			};
 			return {
@@ -771,77 +849,111 @@ function reduceAIState(state, event) {
 		}
 		case "ai.generation.failed": {
 			const payload = event.payload;
-			if (state.activeGeneration?.generationId !== payload.generationId && !(payload.stepLimit === true && state.responseGenerationIds[payload.responseMessageId] === payload.generationId)) return state;
-			if (payload.superseded === true) return {
-				...state,
-				messages: state.messages.filter((message) => message.id !== payload.responseMessageId),
-				tools: state.tools.filter((tool) => tool.generationId !== payload.generationId),
-				pendingApprovals: state.pendingApprovals.filter((approval) => approval.generationId !== payload.generationId),
-				pendingInputs: state.pendingInputs.filter((input) => input.generationId !== payload.generationId),
-				activeGeneration: null,
-				activeProjection: null,
-				compaction: state.compaction?.generationId === payload.generationId ? null : state.compaction
-			};
-			const projection = state.activeProjection;
-			const shouldProject = projection?.generationId === payload.generationId && (projection.baseMessage !== void 0 || projection.batches.length > 0);
+			if (terminalGenerationReason(state, payload.generationId) !== void 0) return state;
+			if (state.activeGeneration?.generationId !== payload.generationId && !(payload.stepLimit === true && state.responseGenerationIds[payload.responseMessageId] === payload.generationId && state.activeRequestId === `ai.generate:tools:${payload.generationId}` && state.activeResponseMessageId === payload.responseMessageId)) return state;
+			if (payload.superseded === true) return restoreGenerationBase(state, payload.generationId, payload.responseMessageId, "superseded");
+			const projection = state.activeProjection?.generationId === payload.generationId ? state.activeProjection : null;
+			const shouldProject = projection !== null && (projection.baseMessage !== void 0 || projection.batches.length > 0);
 			return {
 				...state,
 				messages: shouldProject ? upsertMessage(state.messages, interruptUIMessage(projectedMessage(projection), `Generation failed before tool completion: ${payload.error}`)) : state.messages,
 				status: "failed",
 				activeGeneration: null,
 				activeResponseMessageId: null,
-				activeProjection: null,
-				compaction: state.compaction?.generationId === payload.generationId ? null : state.compaction,
+				activeProjection: projection,
+				terminalGenerations: {
+					...state.terminalGenerations,
+					[payload.generationId]: "failed"
+				},
+				terminalRequestIds: {
+					...state.terminalRequestIds,
+					[payload.requestId]: true,
+					...state.activeRequestId === null ? {} : { [state.activeRequestId]: true }
+				},
+				compaction: projection?.generationId === payload.generationId ? projection.baseCompaction ?? null : state.compaction?.generationId === payload.generationId ? null : state.compaction,
 				pendingApprovals: state.pendingApprovals.filter((approval) => approval.generationId !== payload.generationId),
+				pendingInputs: state.pendingInputs.filter((input) => input.generationId !== payload.generationId),
 				tools: stopGenerationTools(state.tools, payload.generationId, payload.responseMessageId, payload.error),
 				error: payload.error
 			};
 		}
 		case "ai.message.interrupted": {
 			const payload = event.payload;
+			const currentGenerationId = state.responseGenerationIds[payload.messageId];
+			const targetRequestId = payload.requestId ?? generationRequestId(payload.generationId);
+			const ownsRequestPhase = payload.requestId !== void 0 && state.activeGeneration === null && state.activeRequestId === payload.requestId && state.activeResponseMessageId === payload.messageId;
+			if (currentGenerationId !== void 0 && !ownsRequestPhase && (payload.generationId !== void 0 ? currentGenerationId !== payload.generationId : generationRequestId(currentGenerationId) !== payload.requestId)) return state;
+			if (state.activeGeneration === null && state.activeRequestId !== null && state.activeResponseMessageId === payload.messageId && targetRequestId !== state.activeRequestId) return state;
+			const currentTerminal = currentGenerationId === void 0 ? void 0 : terminalGenerationReason(state, currentGenerationId);
+			if (currentTerminal === "completed" || currentTerminal === "interrupted" || currentTerminal === "superseded") return state;
 			const active = state.activeGeneration;
 			if (active === null) {
-				if (payload.generationId === void 0 && state.activeRequestId !== null && state.activeResponseMessageId === payload.messageId) return {
-					...state,
-					status: "idle",
-					activeRequestId: null,
-					activeResponseMessageId: null,
-					error: null
-				};
-				if (!(state.pendingApprovals.some((approval) => approval.messageId === payload.messageId) || state.pendingInputs.some((input) => input.messageId === payload.messageId) || state.tools.some((tool) => tool.messageId === payload.messageId && tool.status === "running"))) return state;
+				if (payload.requestId !== void 0 && state.activeRequestId !== null && state.activeRequestId === payload.requestId && state.activeResponseMessageId === payload.messageId) {
+					const baseState = (currentGenerationId === void 0 ? null : interruptGenerationProjection(state, currentGenerationId, payload.messageId, payload.lastSeenIndex ?? Number.POSITIVE_INFINITY)) ?? state;
+					return {
+						...baseState,
+						status: "idle",
+						activeRequestId: null,
+						activeResponseMessageId: null,
+						activeProjection: null,
+						terminalRequestIds: {
+							...baseState.terminalRequestIds,
+							[payload.requestId]: true
+						},
+						...currentGenerationId === void 0 ? {} : { terminalGenerations: {
+							...baseState.terminalGenerations,
+							[currentGenerationId]: "interrupted"
+						} },
+						error: null
+					};
+				}
+				const targetGenerationId = payload.generationId ?? (currentGenerationId !== void 0 && generationRequestId(currentGenerationId) === payload.requestId ? currentGenerationId : void 0);
+				if (targetGenerationId === void 0) return state;
+				if (currentGenerationId !== targetGenerationId) return state;
+				const matchesGeneration = (generationId) => targetGenerationId === generationId;
+				const hasMessageWork = state.pendingApprovals.some((approval) => approval.messageId === payload.messageId && matchesGeneration(approval.generationId)) || state.pendingInputs.some((input) => input.messageId === payload.messageId && matchesGeneration(input.generationId)) || state.tools.some((activity) => activity.messageId === payload.messageId && activity.status === "running" && matchesGeneration(activity.generationId));
+				const resolvesFailedGeneration = state.status === "failed" && currentGenerationId === targetGenerationId && (state.activeRequestId === null || state.activeRequestId === targetRequestId || state.activeRequestId === `ai.generate:tools:${targetGenerationId}`);
+				if (!hasMessageWork && !resolvesFailedGeneration && !(currentGenerationId === targetGenerationId && currentTerminal === void 0)) return state;
+				const pendingRequestId = state.activeRequestId;
+				const projectedState = interruptGenerationProjection(state, targetGenerationId, payload.messageId, payload.lastSeenIndex ?? Number.POSITIVE_INFINITY);
+				const baseState = projectedState ?? state;
 				const next = {
-					...state,
+					...baseState,
 					activeRequestId: null,
 					activeResponseMessageId: null,
-					messages: state.messages.map((message) => message.id === payload.messageId ? interruptUIMessage(message) : message),
-					pendingApprovals: state.pendingApprovals.filter((approval) => approval.messageId !== payload.messageId),
-					pendingInputs: state.pendingInputs.filter((input) => input.messageId !== payload.messageId),
-					tools: stopGenerationTools(state.tools, payload.generationId ?? "", payload.messageId, "Tool execution was interrupted.")
+					activeProjection: null,
+					messages: projectedState === null ? state.messages.map((message) => message.id === payload.messageId ? interruptUIMessage(message) : message) : baseState.messages,
+					pendingApprovals: baseState.pendingApprovals.filter((approval) => approval.messageId !== payload.messageId || !matchesGeneration(approval.generationId)),
+					pendingInputs: baseState.pendingInputs.filter((input) => input.messageId !== payload.messageId || !matchesGeneration(input.generationId)),
+					tools: projectedState === null ? stopGenerationTools(state.tools, targetGenerationId, payload.messageId, "Tool execution was interrupted.") : baseState.tools,
+					terminalGenerations: {
+						...baseState.terminalGenerations,
+						[targetGenerationId]: "interrupted"
+					},
+					...pendingRequestId === null ? {} : { terminalRequestIds: {
+						...baseState.terminalRequestIds,
+						[pendingRequestId]: true
+					} },
+					error: resolvesFailedGeneration ? null : state.error
 				};
 				return {
 					...next,
-					status: lifecycleStatus(state, next)
+					status: waitingStatus(next)
 				};
 			}
-			if (payload.generationId === void 0 ? payload.messageId !== active.responseMessageId : payload.generationId !== active.generationId) return state;
-			const projection = state.activeProjection;
+			if (payload.messageId !== active.responseMessageId || payload.generationId !== active.generationId && payload.requestId !== active.requestId) return state;
 			const cutoff = payload.lastSeenIndex ?? Number.POSITIVE_INFINITY;
-			const retained = projection?.generationId === active.generationId ? {
-				...projection,
-				batches: projection.batches.filter((batch) => batch.index <= cutoff),
-				toolEvents: projection.toolEvents.filter((toolEvent) => toolEvent.index <= cutoff),
-				approvalEvents: projection.approvalEvents.filter((approvalEvent) => approvalEvent.index <= cutoff)
-			} : null;
-			const shouldProject = retained !== null && (retained.baseMessage !== void 0 || retained.batches.length > 0);
+			const projectedState = interruptGenerationProjection(state, active.generationId, active.responseMessageId, cutoff);
+			const baseState = projectedState ?? restoreGenerationBase(state, active.generationId, active.responseMessageId, "interrupted");
 			const next = {
-				...state,
-				messages: shouldProject ? upsertMessage(state.messages, interruptUIMessage(projectedMessage(retained))) : state.messages,
-				activeGeneration: null,
+				...baseState,
 				activeRequestId: null,
 				activeResponseMessageId: null,
-				activeProjection: null,
-				pendingApprovals: state.pendingApprovals.filter((approval) => approval.generationId !== active.generationId),
-				tools: stopGenerationTools(retained === null ? state.tools : projectedTools(retained), active.generationId, active.responseMessageId, "Tool execution was interrupted.")
+				terminalRequestIds: {
+					...baseState.terminalRequestIds,
+					[active.requestId]: true
+				},
+				tools: projectedState === null ? stopGenerationTools(baseState.tools, active.generationId, active.responseMessageId, "Tool execution was interrupted.") : baseState.tools
 			};
 			return {
 				...next,
@@ -851,8 +963,8 @@ function reduceAIState(state, event) {
 		case "ai.tool.called": {
 			const payload = event.payload;
 			const projection = state.activeProjection;
-			if (isStaleGeneration(state, payload.messageId, payload.generationId)) return state;
-			if (state.tools.some((tool) => tool.generationId === payload.generationId && tool.toolCallId === payload.toolCallId && tool.status === "failed" && state.activeProjection?.generationId !== payload.generationId)) return state;
+			if (isStaleGeneration(state, payload.messageId, payload.generationId) || terminalGenerationReason(state, payload.generationId) !== void 0) return state;
+			if (state.tools.some((activity) => activity.generationId === payload.generationId && activity.toolCallId === payload.toolCallId && activity.status === "failed" && state.activeGeneration?.generationId !== payload.generationId)) return state;
 			if (projection?.generationId !== payload.generationId) {
 				const next = {
 					...state,
@@ -874,17 +986,23 @@ function reduceAIState(state, event) {
 					payload
 				}]
 			};
-			return {
+			const next = {
 				...state,
 				activeProjection,
 				tools: projectedTools(activeProjection)
 			};
+			return {
+				...next,
+				status: lifecycleStatus(state, next)
+			};
 		}
 		case "ai.tool.result": {
 			const payload = event.payload;
 			const projection = state.activeProjection;
-			if (isStaleGeneration(state, payload.messageId, payload.generationId)) return state;
-			if (state.tools.some((tool) => tool.generationId === payload.generationId && tool.toolCallId === payload.toolCallId && tool.status === "failed" && state.activeProjection?.generationId !== payload.generationId)) return state;
+			const terminalReason = terminalGenerationReason(state, payload.generationId);
+			const completesFailedTool = terminalReason === "failed" && state.status === "failed" && payload.preliminary !== true && projection?.generationId === payload.generationId && projectedTools(projection).some((activity) => activity.generationId === payload.generationId && activity.toolCallId === payload.toolCallId && activity.status === "running");
+			if (isStaleGeneration(state, payload.messageId, payload.generationId) || terminalReason !== void 0 && !completesFailedTool) return state;
+			if (state.tools.some((activity) => activity.generationId === payload.generationId && activity.toolCallId === payload.toolCallId && activity.status === "failed" && state.activeGeneration?.generationId !== payload.generationId && !completesFailedTool)) return state;
 			if (projection?.generationId !== payload.generationId) {
 				const next = {
 					...state,
@@ -907,18 +1025,25 @@ function reduceAIState(state, event) {
 					payload
 				}]
 			};
-			return {
+			const projected = projectedMessage(activeProjection);
+			const tools = projectedTools(activeProjection);
+			const failureError = state.error ?? "Generation failed.";
+			const next = {
 				...state,
-				messages: upsertMessage(state.messages, projectedMessage(activeProjection)),
+				messages: upsertMessage(state.messages, terminalReason === "failed" ? interruptUIMessage(projected, `Generation failed before tool completion: ${failureError}`) : projected),
 				activeProjection,
-				tools: projectedTools(activeProjection)
+				tools: terminalReason === "failed" ? stopGenerationTools(tools, payload.generationId, payload.messageId, failureError) : tools
+			};
+			return {
+				...next,
+				status: lifecycleStatus(state, next)
 			};
 		}
 		case "ai.approval.requested": {
 			const payload = event.payload;
 			const projection = state.activeProjection;
-			if (isStaleGeneration(state, payload.messageId, payload.generationId)) return state;
-			if (state.tools.some((tool) => tool.generationId === payload.generationId && tool.toolCallId === payload.toolCallId && tool.status === "failed" && projection?.generationId !== payload.generationId)) return state;
+			if (isStaleGeneration(state, payload.messageId, payload.generationId) || terminalGenerationReason(state, payload.generationId) !== void 0) return state;
+			if (state.tools.some((activity) => activity.generationId === payload.generationId && activity.toolCallId === payload.toolCallId && activity.status === "failed" && state.activeGeneration?.generationId !== payload.generationId)) return state;
 			const activeProjection = projection?.generationId === payload.generationId ? {
 				...projection,
 				approvalEvents: [...projection.approvalEvents, {
@@ -940,7 +1065,7 @@ function reduceAIState(state, event) {
 		}
 		case "ai.approval.responded": {
 			const payload = event.payload;
-			if (isStaleGeneration(state, payload.messageId, payload.generationId)) return state;
+			if (isStaleGeneration(state, payload.messageId, payload.generationId) || terminalGenerationReason(state, payload.generationId) !== void 0) return state;
 			const projection = state.activeProjection;
 			const pendingRequest = state.pendingApprovals.find((candidate) => candidate.approvalId === payload.approvalId && candidate.messageId === payload.messageId && candidate.generationId === payload.generationId);
 			const activeRequest = projection?.approvalEvents.findLast((candidate) => candidate.type === "ai.approval.requested" && candidate.payload.approvalId === payload.approvalId && candidate.payload.messageId === payload.messageId && candidate.payload.generationId === payload.generationId);
@@ -966,7 +1091,7 @@ function reduceAIState(state, event) {
 		}
 		case "ai.input.requested": {
 			const payload = event.payload;
-			if (isStaleGeneration(state, payload.messageId, payload.generationId)) return state;
+			if (isStaleGeneration(state, payload.messageId, payload.generationId) || terminalGenerationReason(state, payload.generationId) !== void 0) return state;
 			const next = {
 				...state,
 				pendingInputs: [...state.pendingInputs.filter((input) => input.messageId !== payload.messageId || input.generationId !== payload.generationId || input.inputId !== payload.inputId), payload]
@@ -978,7 +1103,7 @@ function reduceAIState(state, event) {
 		}
 		case "ai.input.responded": {
 			const payload = event.payload;
-			if (isStaleGeneration(state, payload.messageId, payload.generationId)) return state;
+			if (isStaleGeneration(state, payload.messageId, payload.generationId) || terminalGenerationReason(state, payload.generationId) !== void 0) return state;
 			const next = {
 				...state,
 				pendingInputs: state.pendingInputs.filter((input) => input.inputId !== payload.inputId || input.messageId !== payload.messageId || input.generationId !== payload.generationId)
@@ -1022,21 +1147,26 @@ function deriveUIMessages(history) {
 /** Build the standard AI state projection for any contract containing the protocol. */
 function createReducer(options) {
 	return options.contract.reducer({
-		name: options.name ?? "a2.ai.state.v6",
+		name: options.name ?? "a2.ai.state.v8",
 		initialState: initialState()
 	}).fold((state, event) => reduceAIState(state, event));
 }
-const seedMessage = (message) => [{
+const messageInput = (message, generate) => [{
 	type: "ai.message.created",
 	id: `ai.message:${message.id}`,
-	payload: { message }
+	payload: {
+		message,
+		...generate === false ? { generate: false } : {}
+	}
 }];
 /** Pure append inputs for the built-in AI protocol. */
 const inputs = {
-	message(message) {
-		return seedMessage(message);
+	message(message, options) {
+		return messageInput(message, options?.generate);
+	},
+	seed(message) {
+		return messageInput(message, false);
 	},
-	seed: seedMessage,
 	approval(response) {
 		return [{
 			type: "ai.approval.responded",
@@ -1068,12 +1198,27 @@ const inputs = {
 	interrupt(interruption) {
 		return [{
 			type: "ai.message.interrupted",
-			id: `ai.interrupt:${interruption.generationId ?? interruption.messageId}`,
+			id: `ai.interrupt:${interruption.generationId ?? interruption.requestId}`,
 			payload: interruption
 		}];
 	}
 };
 /**
+* Read the durable handler context ambient to the current AI tool
+* execution. A2 runs each local tool's `execute` inside this scope, so
+* plain AI SDK `tool()` definitions reach their `event`, `attempt`,
+* `session`, and `signal` without an A2-specific wrapper. The agent (or
+* contract) argument carries the event types and is verified against
+* the executing server's contract.
+*/
+function handlerContext(agentOrContract) {
+	const contract = "contract" in agentOrContract ? agentOrContract.contract : agentOrContract;
+	const scope = ambientToolScopeStorage()?.getStore();
+	if (scope === void 0) throw new TypeError(`handlerContext('${contract.name}') requires an ambient A2 tool execution: call it inside a local tool's execute, or wrap direct tool tests with runWithHandlerContext`);
+	if (scope.contract !== void 0 && scope.contract.name !== contract.name) throw new TypeError(`handlerContext('${contract.name}') was called during a tool execution for agent '${scope.contract.name}'`);
+	return scope.context;
+}
+/**
 * Define an agent: built-in AI protocol + application events and one state
 * reducer. The result is isomorphic.
 */
@@ -1091,10 +1236,10 @@ function agent(options) {
 	return {
 		contract: agentContract,
 		reducer: agentContract.reducer({
-			name: options.reducerName ?? "a2.ai.state.v6",
+			name: options.reducerName ?? "a2.ai.state.v8",
 			initialState: initialState()
 		}).fold((state, event) => reduceAIState(state, event))
 	};
 }
 //#endregion
-export { agent, createEvents, createReducer, deriveUIMessages, events, inputs, reduceAIState };
+export { agent, createEvents, createReducer, deriveUIMessages, events, handlerContext, inputs, reduceAIState };