experimental-a2 0.2.0 → 0.4.0

package/dist/ai-server.js

@@ -1,5 +1,7 @@
-import { t as A2Error } from "./errors-BJRMd-h6.js";
-import { t as createServer } from "./server-DJgD2YWP.js";
+import { t as A2Error } from "./errors-BQuJpe82.js";
+import { m as installAmbientToolScopeStorage, p as consumeSchedulerSendFailure } from "./internal-aEotMzu_.js";
+import { t as createServer } from "./server-CcNnFnoW.js";
+import { AsyncLocalStorage } from "node:async_hooks";
 import { convertToModelMessages, stepCountIs, streamText, toUIMessageStream } from "ai";
 //#region src/ai-sdk-step.ts
 const CONTROLLED_SETTINGS = [
@@ -19,8 +21,7 @@ const CONTROLLED_SETTINGS = [
 	"prepareStep",
 	"stopWhen",
 	"system",
-	"tools",
-	"toolsContext"
+	"tools"
 ];
 const safeSettings = (settings) => {
 	const result = { ...settings };
@@ -72,7 +73,8 @@ async function generateAISDKStep(input) {
 			tools,
 			originalMessages: input.messages,
 			generateMessageId: () => input.responseMessageId,
-			onError: errorMessage$1
+			onError: errorMessage$1,
+			...input.messageMetadata === void 0 ? {} : { messageMetadata: input.messageMetadata }
 		}),
 		completion() {
 			if (fatalError !== void 0) throw fatalError.value;
@@ -95,13 +97,15 @@ const foldCoordinator = (state, event) => {
 			queued: []
 		};
 		case "ai.message.created": {
-			const message = event.payload.message;
+			const payload = event.payload;
+			const message = payload.message;
 			if (message.role !== "user") return state;
 			return {
 				...state,
 				queued: [...state.queued.filter((item) => item.messageId !== message.id), {
 					index: event.index,
-					messageId: message.id
+					messageId: message.id,
+					generate: payload.generate !== false
 				}].toSorted((left, right) => left.index - right.index)
 			};
 		}
@@ -109,9 +113,10 @@ const foldCoordinator = (state, event) => {
 			const request = event.payload;
 			if (request.reason === "message") {
 				const responseMessageId = request.responseMessageId ?? `${request.messageId}:assistant`;
+				const requested = state.queued.find((item) => item.messageId === request.messageId);
 				return {
 					...state,
-					queued: state.queued.filter((item) => item.messageId !== request.messageId),
+					queued: requested === void 0 ? state.queued.filter((item) => item.messageId !== request.messageId) : state.queued.filter((item) => item.index > requested.index),
 					response: {
 						rootMessageId: request.messageId,
 						responseMessageId,
@@ -177,7 +182,7 @@ const foldCoordinator = (state, event) => {
 		case "ai.message.interrupted": {
 			const interruption = event.payload;
 			const response = state.response;
-			return response?.responseMessageId === interruption.messageId && (interruption.generationId === void 0 || interruption.generationId === response.generation?.generationId) ? {
+			return response?.responseMessageId === interruption.messageId && (interruption.generationId === response.generation?.generationId || interruption.requestId === response.activeRequestId) ? {
 				closed: state.closed,
 				queued: state.queued
 			} : state;
@@ -255,7 +260,7 @@ const foldCoordinator = (state, event) => {
 	}
 };
 const aiCoordinatorReducer = (contract) => contract.reducer({
-	name: "a2.ai.coordinator.v1",
+	name: "a2.ai.coordinator.v3",
 	initialState: {
 		closed: false,
 		queued: []
@@ -288,8 +293,18 @@ const modelName = (model) => {
 	return "custom";
 };
 const errorMessage = (error) => error instanceof Error ? error.message : String(error);
+const toolScopeStorage = installAmbientToolScopeStorage(() => new AsyncLocalStorage());
+/** Run `fn` with `ctx` as the ambient tool handler context, for direct tool tests. */
+function runWithHandlerContext(ctx, fn) {
+	return toolScopeStorage.run({ context: ctx }, fn);
+}
 const promptCacheKey = (sessionId, generationId) => `${sessionId}\u001f${generationId}`;
-const isAsyncIterable = (value) => typeof value === "object" && value !== null && Symbol.asyncIterator in value && typeof value[Symbol.asyncIterator] === "function";
+const asyncIterator = (value) => {
+	if (typeof value !== "object" || value === null) return void 0;
+	const factory = Reflect.get(value, Symbol.asyncIterator);
+	if (typeof factory !== "function") return void 0;
+	return factory.call(value);
+};
 const isBoundaryChunk = (chunk) => chunk !== void 0 && (chunk.type === "text-end" || chunk.type === "reasoning-end" || chunk.type === "tool-input-available" || chunk.type === "tool-input-error" || chunk.type === "tool-output-available" || chunk.type === "tool-output-error" || chunk.type === "tool-output-denied" || chunk.type === "tool-approval-request" || chunk.type === "finish" || chunk.type === "abort" || chunk.type === "error");
 const nextOrTimer = async (next, delayMs) => {
 	let timer;
@@ -305,7 +320,7 @@ const nextOrTimer = async (next, delayMs) => {
 		if (timer !== void 0) clearTimeout(timer);
 	}
 };
-async function generateWithAISDK(context) {
+async function generateWithAISDK(context, messageMetadata) {
 	const step = await generateAISDKStep({
 		model: context.model,
 		tools: context.tools,
@@ -313,7 +328,11 @@ async function generateWithAISDK(context) {
 		responseMessageId: context.responseMessageId,
 		abortSignal: context.signal,
 		...context.instructions === void 0 ? {} : { instructions: context.instructions },
-		settings: context.generation
+		settings: context.generation,
+		...messageMetadata === void 0 ? {} : { messageMetadata: ({ part }) => messageMetadata({
+			...context,
+			part
+		}) }
 	});
 	return {
 		stream: step.stream,
@@ -408,6 +427,10 @@ const activeContextMessages = (state, coordinator) => {
 	const queued = new Set(coordinator.queued.map((item) => item.messageId));
 	return contextMessages(state).filter((message) => !queued.has(message.id));
 };
+const generationRequestId = (generationId) => {
+	const markerIndex = generationId.lastIndexOf(":generation:");
+	return markerIndex === -1 ? void 0 : generationId.slice(0, markerIndex);
+};
 const toolCalledEvent = (payload) => ({
 	type: "ai.tool.called",
 	id: `${payload.generationId}:tool:${payload.toolCallId}:called`,
@@ -576,15 +599,27 @@ const lifecycleEvents = (options) => {
 	};
 };
 function queuedMessagesAt(history) {
-	const started = new Set(history.filter((event) => event.type === "ai.generation.requested").map((event) => event.payload).filter((request) => request.reason === "message").map((request) => request.messageId));
-	return history.flatMap((event) => {
-		if (event.type !== "ai.message.created") return [];
-		const message = event.payload.message;
-		return message.role === "user" && !started.has(message.id) ? [{
-			index: event.index,
-			messageId: message.id
-		}] : [];
-	});
+	const queued = [];
+	for (const event of history) {
+		if (event.type === "ai.message.created") {
+			const payload = event.payload;
+			if (payload.message.role !== "user") continue;
+			const duplicate = queued.findIndex((item) => item.messageId === payload.message.id);
+			if (duplicate !== -1) queued.splice(duplicate, 1);
+			queued.push({
+				index: event.index,
+				messageId: payload.message.id,
+				generate: payload.generate !== false
+			});
+			continue;
+		}
+		if (event.type !== "ai.generation.requested") continue;
+		const request = event.payload;
+		if (request.reason !== "message") continue;
+		const requested = queued.findIndex((item) => item.messageId === request.messageId);
+		if (requested !== -1) queued.splice(0, requested + 1);
+	}
+	return queued;
 }
 function withoutGenerationLifecycle(history, generationIds) {
 	if (generationIds.size === 0) return history;
@@ -612,10 +647,10 @@ function createHandlers(options) {
 	if (options.model === void 0) throw new TypeError("createHandlers requires model");
 	if (options.progress?.maxChunks !== void 0 && (!Number.isInteger(options.progress.maxChunks) || options.progress.maxChunks < 1)) throw new TypeError("progress.maxChunks must be a positive integer");
 	if (options.progress?.maxDelayMs !== void 0 && (!Number.isFinite(options.progress.maxDelayMs) || options.progress.maxDelayMs < 0)) throw new TypeError("progress.maxDelayMs must be a non-negative number");
-	if (options.maxSteps !== void 0 && (!Number.isInteger(options.maxSteps) || options.maxSteps < 1)) throw new TypeError("maxSteps must be a positive integer");
+	if (options.maxSteps !== void 0 && options.maxSteps !== Number.POSITIVE_INFINITY && (!Number.isInteger(options.maxSteps) || options.maxSteps < 1)) throw new TypeError("maxSteps must be a positive integer or Infinity");
 	const tools = options.tools ?? {};
 	const generation = options.generation ?? {};
-	const maxSteps = options.maxSteps ?? 20;
+	const maxSteps = options.maxSteps ?? Number.POSITIVE_INFINITY;
 	const coordinator = aiCoordinatorReducer(options.agent.contract);
 	const promptCache = /* @__PURE__ */ new Map();
 	const coordinatorStateAt = (history, frontier = Number.POSITIVE_INFINITY) => {
@@ -628,7 +663,7 @@ function createHandlers(options) {
 	};
 	const requestForNextMessage = (state) => {
 		if (state.closed || state.response !== void 0) return void 0;
-		const next = state.queued[0];
+		const next = state.queued.find((item) => item.generate !== false);
 		if (next === void 0) return void 0;
 		return {
 			type: "ai.generation.requested",
@@ -710,36 +745,72 @@ function createHandlers(options) {
 		});
 		return computation;
 	};
-	const executeTool = async (ctx, call) => {
-		const tool = tools[call.toolName];
-		if (tool?.execute === void 0) return resultEvent(call, "execution:error", { error: `Tool '${call.toolName}' has no server executor` });
-		const messages = await promptMessages(ctx.event.sessionId, call, ctx.session.history);
+	const toolExecutionFailure = (ctx, call, error) => {
+		const schedulerFailure = consumeSchedulerSendFailure(error);
+		if (ctx.signal.aborted) return;
+		if (schedulerFailure === "retryable") throw error;
+		return resultEvent(call, "execution:error", { error: errorMessage(error) });
+	};
+	const runToolExecution = async (ctx, call, execute, messages) => {
 		let output;
 		try {
-			output = await tool.execute(call.input, {
+			output = await execute(call.input, {
 				toolCallId: call.toolCallId,
 				messages,
-				abortSignal: ctx.signal,
-				context: void 0
+				abortSignal: ctx.signal
 			});
 		} catch (error) {
-			if (ctx.signal.aborted) return;
-			return resultEvent(call, "execution:error", { error: errorMessage(error) });
+			return toolExecutionFailure(ctx, call, error);
 		}
-		if (!isAsyncIterable(output)) return resultEvent(call, "execution:0", { output });
+		let iterator;
+		try {
+			iterator = asyncIterator(output);
+		} catch (error) {
+			return toolExecutionFailure(ctx, call, error);
+		}
+		if (iterator === void 0) return resultEvent(call, "execution:0", { output });
 		let last;
 		let sequence = 0;
-		for await (const value of output) {
-			if (ctx.signal.aborted) return;
-			last = value;
-			await ctx.session.append(`tool:${call.toolCallId}:preliminary:${sequence}`, resultEvent(call, `execution:${sequence}:preliminary`, {
-				output: value,
-				preliminary: true
-			}));
+		for (;;) {
+			let result;
+			try {
+				result = await iterator.next();
+			} catch (error) {
+				return toolExecutionFailure(ctx, call, error);
+			}
+			if (result.done) break;
+			if (ctx.signal.aborted) {
+				try {
+					await iterator.return?.();
+				} catch {}
+				return;
+			}
+			last = result.value;
+			try {
+				await ctx.session.append(`tool:${call.toolCallId}:preliminary:${sequence}`, resultEvent(call, `execution:${sequence}:preliminary`, {
+					output: result.value,
+					preliminary: true
+				}));
+			} catch (error) {
+				try {
+					await iterator.return?.();
+				} catch {}
+				throw error;
+			}
 			sequence += 1;
 		}
 		return resultEvent(call, `execution:${sequence}:final`, sequence === 0 ? {} : { output: last });
 	};
+	const executeTool = async (ctx, call) => {
+		const execute = tools[call.toolName]?.execute;
+		if (execute === void 0) return resultEvent(call, "execution:error", { error: `Tool '${call.toolName}' has no server executor` });
+		const messages = await promptMessages(ctx.event.sessionId, call, ctx.session.history);
+		const scope = {
+			contract: options.agent.contract,
+			context: ctx
+		};
+		return toolScopeStorage.run(scope, () => runToolExecution(ctx, call, execute, messages));
+	};
 	const handleToolCall = async (ctx) => {
 		const response = (await ctx.session.state(coordinator)).state.response;
 		const current = response?.calls.find((candidate) => candidate.index === ctx.event.index);
@@ -914,7 +985,7 @@ function createHandlers(options) {
 			};
 			const custom = options.generate !== void 0;
 			const updates = consumeGeneration({
-				source: options.generate === void 0 ? await generateWithAISDK(generateContext) : { stream: await options.generate(generateContext) },
+				source: options.generate === void 0 ? await generateWithAISDK(generateContext, options.messageMetadata) : { stream: await options.generate(generateContext) },
 				...options.progress === void 0 ? {} : { progress: options.progress }
 			});
 			let sequence = 0;
@@ -1008,7 +1079,7 @@ function createHandlers(options) {
 		}
 	};
 	const handleMessageCreated = async (ctx) => {
-		if (ctx.event.payload.message.role !== "user") return;
+		if (ctx.event.payload.message.role !== "user" || ctx.event.payload.generate === false) return;
 		return scheduleNext(ctx);
 	};
 	const handleRetry = async (ctx) => {
@@ -1071,7 +1142,7 @@ function createHandlers(options) {
 			abortOn: {
 				"ai.message.interrupted": (event, trigger, context) => {
 					const responseMessageId = trigger.payload.responseMessageId ?? (trigger.payload.reason === "message" ? `${trigger.payload.messageId}:assistant` : trigger.payload.messageId);
-					return event.payload.messageId === responseMessageId && (event.payload.generationId === void 0 || event.payload.generationId === `${trigger.id}:generation:${context.attempt}`);
+					return event.payload.messageId === responseMessageId && (event.payload.requestId === trigger.id || event.payload.generationId === `${trigger.id}:generation:${context.attempt}`);
 				},
 				"ai.session.closed": true
 			},
@@ -1081,7 +1152,7 @@ function createHandlers(options) {
 		"ai.tool.called": {
 			abortOn: {
 				"ai.generation.failed": (event, trigger) => event.payload.generationId === trigger.payload.generationId,
-				"ai.message.interrupted": (event, trigger) => event.payload.messageId === trigger.payload.messageId && (event.payload.generationId === void 0 || event.payload.generationId === trigger.payload.generationId),
+				"ai.message.interrupted": (event, trigger) => event.payload.messageId === trigger.payload.messageId && (event.payload.requestId === trigger.payload.requestId || event.payload.generationId === trigger.payload.generationId),
 				"ai.session.closed": true
 			},
 			handler: handleToolCall
@@ -1089,7 +1160,7 @@ function createHandlers(options) {
 		"ai.approval.responded": {
 			abortOn: {
 				"ai.generation.failed": (event, trigger) => event.payload.responseMessageId === trigger.payload.messageId && event.payload.generationId === trigger.payload.generationId,
-				"ai.message.interrupted": (event, trigger) => event.payload.messageId === trigger.payload.messageId && (event.payload.generationId === void 0 || event.payload.generationId === trigger.payload.generationId),
+				"ai.message.interrupted": (event, trigger) => event.payload.messageId === trigger.payload.messageId && (event.payload.requestId === generationRequestId(trigger.payload.generationId) || event.payload.generationId === trigger.payload.generationId),
 				"ai.session.closed": true
 			},
 			handler: handleApproval
@@ -1102,7 +1173,7 @@ function createHandlers(options) {
 }
 /** Assemble an A2 server with the built-in agent handlers and app extensions. */
 function createAgentServer(options) {
-	const { agent: definition, model, tools, instructions, generation, maxSteps, generate, compaction, progress, handlers, ...serverOptions } = options;
+	const { agent: definition, model, tools, instructions, generation, maxSteps, messageMetadata, generate, compaction, progress, handlers, ...serverOptions } = options;
 	const builtIns = createHandlers({
 		agent: definition,
 		model,
@@ -1110,6 +1181,7 @@ function createAgentServer(options) {
 		...instructions === void 0 ? {} : { instructions },
 		...generation === void 0 ? {} : { generation },
 		...maxSteps === void 0 ? {} : { maxSteps },
+		...messageMetadata === void 0 ? {} : { messageMetadata },
 		...generate === void 0 ? {} : { generate },
 		...compaction === void 0 ? {} : { compaction },
 		...progress === void 0 ? {} : { progress }
@@ -1125,4 +1197,4 @@ function createAgentServer(options) {
 	});
 }
 //#endregion
-export { createAgentServer, createHandlers, validateAgentPush };
+export { createAgentServer, createHandlers, runWithHandlerContext, validateAgentPush };

package/dist/ai.d.ts

@@ -1,303 +1,2 @@
-import { a as EventDefs, c as Reducer, n as Contract, t as AppendInput, u as StandardSchemaV1 } from "./contract-C_3dIIEU.js";
-import { FinishReason, JSONValue, LanguageModelUsage, UIMessage, UIMessageChunk } from "ai";
-//#region src/ai.d.ts
-type AgentStatus = "idle" | "generating" | "waiting" | "failed" | "closed";
-type GenerationReason = "message" | "tool" | "input" | "retry";
-type SessionCreatedPayload = {
-  metadata?: JSONValue;
-};
-type SessionClosedPayload = {
-  reason?: string;
-};
-type MessageCreatedPayload<M extends UIMessage> = {
-  message: M;
-};
-type MessageCompletedPayload = {
-  messageId: string;
-};
-type MessageInterruptedPayload = {
-  messageId: string;
-  generationId?: string;
-  reason?: string;
-  /** Last confirmed log index visible to the interrupter. */
-  lastSeenIndex?: number;
-};
-type GenerationRequestedPayload = {
-  messageId: string;
-  reason: GenerationReason;
-  responseMessageId?: string;
-};
-type RetryGenerationOptions = {
-  messageId: string;
-  responseMessageId: string;
-  retryId: string;
-};
-type RetryRequestedPayload = RetryGenerationOptions;
-type GenerationStartedPayload = {
-  requestId: string;
-  messageId: string;
-  generationId: string;
-  responseMessageId: string;
-  attempt: number;
-  model: string;
-};
-type GenerationProgressPayload = {
-  requestId: string;
-  messageId: string;
-  generationId: string;
-  responseMessageId: string;
-  sequence: number;
-  chunks: UIMessageChunk[];
-};
-type GenerationCompletedPayload = {
-  requestId: string;
-  messageId: string;
-  generationId: string;
-  responseMessageId: string;
-  finishReason?: FinishReason;
-  usage?: LanguageModelUsage;
-};
-type GenerationFailedPayload = {
-  requestId: string;
-  messageId: string;
-  generationId: string;
-  responseMessageId: string;
-  error: string;
-  /** True when a later attempt replaced an incomplete attempt. */
-  superseded?: boolean;
-  stepLimit?: boolean;
-};
-type ToolCalledPayload = {
-  requestId: string;
-  messageId: string;
-  generationId: string;
-  toolCallId: string;
-  toolName: string;
-  input: unknown;
-  dynamic?: boolean;
-  providerExecuted?: boolean;
-  supportsDeferredResults?: boolean;
-  providerMetadata?: unknown;
-  toolMetadata?: unknown;
-  title?: string;
-};
-type ToolResultPayload = {
-  requestId: string;
-  messageId: string;
-  generationId: string;
-  toolCallId: string;
-  toolName?: string;
-  input?: unknown;
-  rawInput?: unknown;
-  output?: unknown;
-  error?: string;
-  denied?: boolean;
-  preliminary?: boolean;
-  phase?: "input" | "execution";
-  dynamic?: boolean;
-  providerExecuted?: boolean;
-  providerMetadata?: unknown;
-  toolMetadata?: unknown;
-};
-type ApprovalRequestedPayload = {
-  messageId: string;
-  generationId: string;
-  approvalId: string;
-  toolCallId: string;
-  isAutomatic?: boolean;
-  signature?: string;
-};
-type ApprovalRespondedPayload = {
-  messageId: string;
-  generationId: string;
-  approvalId: string;
-  approved: boolean;
-  reason?: string;
-};
-type InputRequestedPayload = {
-  messageId: string;
-  generationId: string;
-  inputId: string;
-  name: string;
-  question?: JSONValue;
-};
-type InputRespondedPayload = {
-  messageId: string;
-  generationId: string;
-  inputId: string;
-  name: string;
-  value: JSONValue;
-};
-type CompactionRequestedPayload = {
-  generationId: string;
-  throughMessageId: string;
-};
-type CompactionCompletedPayload<M extends UIMessage> = {
-  generationId: string;
-  throughMessageId: string;
-  messages: M[];
-  retainedMessageIds?: string[];
-};
-type AIEventDefs<M extends UIMessage = UIMessage> = {
-  "ai.session.created": StandardSchemaV1<SessionCreatedPayload>;
-  "ai.session.closed": StandardSchemaV1<SessionClosedPayload>;
-  "ai.message.created": StandardSchemaV1<MessageCreatedPayload<M>>;
-  "ai.message.completed": StandardSchemaV1<MessageCompletedPayload>;
-  "ai.message.interrupted": StandardSchemaV1<MessageInterruptedPayload>;
-  "ai.generation.requested": StandardSchemaV1<GenerationRequestedPayload>;
-  "ai.retry.requested": StandardSchemaV1<RetryRequestedPayload>;
-  "ai.generation.started": StandardSchemaV1<GenerationStartedPayload>;
-  "ai.generation.progress": StandardSchemaV1<GenerationProgressPayload>;
-  "ai.generation.completed": StandardSchemaV1<GenerationCompletedPayload>;
-  "ai.generation.failed": StandardSchemaV1<GenerationFailedPayload>;
-  "ai.tool.called": StandardSchemaV1<ToolCalledPayload>;
-  "ai.tool.result": StandardSchemaV1<ToolResultPayload>;
-  "ai.approval.requested": StandardSchemaV1<ApprovalRequestedPayload>;
-  "ai.approval.responded": StandardSchemaV1<ApprovalRespondedPayload>;
-  "ai.input.requested": StandardSchemaV1<InputRequestedPayload>;
-  "ai.input.responded": StandardSchemaV1<InputRespondedPayload>;
-  "ai.compaction.requested": StandardSchemaV1<CompactionRequestedPayload>;
-  "ai.compaction.completed": StandardSchemaV1<CompactionCompletedPayload<M>>;
-};
-type ActiveGeneration = GenerationStartedPayload;
-type ActiveGenerationProjection<M extends UIMessage = UIMessage> = {
-  generationId: string;
-  responseMessageId: string;
-  baseMessage?: M;
-  batches: Array<{
-    index: number;
-    chunks: UIMessageChunk[];
-  }>;
-  baseTools: ToolActivity[];
-  toolEvents: Array<{
-    index: number;
-    type: "ai.tool.called";
-    payload: ToolCalledPayload;
-  } | {
-    index: number;
-    type: "ai.tool.result";
-    payload: ToolResultPayload;
-  }>;
-  approvalEvents: Array<{
-    index: number;
-    type: "ai.approval.requested";
-    payload: ApprovalRequestedPayload;
-  } | {
-    index: number;
-    type: "ai.approval.responded";
-    payload: ApprovalRespondedPayload;
-  }>;
-};
-type PendingApproval = ApprovalRequestedPayload;
-type PendingInput = InputRequestedPayload;
-type ToolActivity = {
-  requestId: string;
-  messageId: string;
-  generationId: string;
-  toolCallId: string;
-  toolName?: string;
-  input?: unknown;
-  rawInput?: unknown;
-  output?: unknown;
-  error?: string;
-  preliminary?: boolean;
-  phase?: "input" | "execution";
-  dynamic?: boolean;
-  providerExecuted?: boolean;
-  callProviderMetadata?: unknown;
-  resultProviderMetadata?: unknown;
-  toolMetadata?: unknown;
-  title?: string;
-  status: "running" | "completed" | "failed" | "denied";
-};
-type CompactionState<M extends UIMessage> = {
-  status: "running" | "completed";
-  generationId: string;
-  throughMessageId: string;
-  messages?: M[];
-  retainedMessageIds?: string[];
-};
-type GenerationUsage = {
-  generationId: string;
-  usage: LanguageModelUsage;
-};
-type AIState<M extends UIMessage = UIMessage> = {
-  session: {
-    created: boolean;
-    metadata?: JSONValue;
-  };
-  messages: M[];
-  status: AgentStatus;
-  activeGeneration: ActiveGeneration | null;
-  activeRequestId: string | null;
-  activeResponseMessageId: string | null;
-  activeProjection: ActiveGenerationProjection<M> | null;
-  responseGenerationIds: Record<string, string>;
-  pendingApprovals: PendingApproval[];
-  pendingInputs: PendingInput[];
-  tools: ToolActivity[];
-  compaction: CompactionState<M> | null;
-  usage: GenerationUsage[];
-  error: string | null;
-};
-type AgentDefinition<M extends UIMessage = UIMessage, D extends EventDefs = AIEventDefs<M>> = {
-  readonly contract: Contract<D>;
-  readonly reducer: Reducer<D, AIState<M>>;
-};
-type AgentOptions<M extends UIMessage, E extends EventDefs> = {
-  name: string;
-  events?: E;
-  /** Optional stricter schema for custom UIMessage metadata/data/tool types. */
-  messageSchema?: StandardSchemaV1<M>;
-  /** Snapshot identity. Bump when application-level projection semantics change. */
-  reducerName?: string;
-};
-type AIMessageFrom<D extends EventDefs> = D extends Record<"ai.message.created", infer S extends StandardSchemaV1> ? StandardSchemaV1.InferOutput<S> extends MessageCreatedPayload<infer M extends UIMessage> ? M : UIMessage : UIMessage;
-declare function createEvents<M extends UIMessage = UIMessage>(options?: {
-  messageSchema?: StandardSchemaV1<M>;
-}): AIEventDefs<M>;
-declare const events: AIEventDefs<UIMessage>;
-/** Pure projection of the built-in AI protocol. Unknown extension events are ignored. */
-declare function reduceAIState<M extends UIMessage>(state: AIState<M>, event: {
-  type: string;
-  payload: unknown;
-  index: number;
-  id?: string;
-}): AIState<M>;
-/** Derive AI SDK UI messages directly from an A2 event history. */
-declare function deriveUIMessages<M extends UIMessage = UIMessage>(history: readonly {
-  type: string;
-  payload: unknown;
-  index: number;
-}[]): M[];
-/** Build the standard AI state projection for any contract containing the protocol. */
-declare function createReducer<D extends EventDefs>(options: {
-  contract: Contract<D>;
-  name?: string;
-}): Reducer<D, AIState<AIMessageFrom<D>>>;
-type ProtocolInput<M extends UIMessage, K extends keyof AIEventDefs<M> & string> = AppendInput<Pick<AIEventDefs<M>, K>>;
-type AIInputs = {
-  /** Record a message. The server schedules user messages in arrival order. */
-  message<M extends UIMessage>(message: M): ProtocolInput<M, "ai.message.created">[];
-  /** Record a message without requesting a generation. */
-  seed<M extends UIMessage>(message: M): ProtocolInput<M, "ai.message.created">[];
-  /** Answer an AI SDK tool approval. The tool barrier owns continuation. */
-  approval(response: ApprovalRespondedPayload): ProtocolInput<UIMessage, "ai.approval.responded">[];
-  /** Answer an application-defined input request. */
-  input(response: InputRespondedPayload): ProtocolInput<UIMessage, "ai.input.responded">[];
-  /** Record an application-defined request for input. */
-  requestInput(request: InputRequestedPayload): ProtocolInput<UIMessage, "ai.input.requested">[];
-  /** Request a fresh attempt for a failed response. */
-  retry(options: RetryGenerationOptions): ProtocolInput<UIMessage, "ai.retry.requested">[];
-  /** Interrupt the active response for a message. */
-  interrupt(interruption: MessageInterruptedPayload): ProtocolInput<UIMessage, "ai.message.interrupted">[];
-};
-/** Pure append inputs for the built-in AI protocol. */
-declare const inputs: AIInputs;
-/**
- * Define an agent: built-in AI protocol + application events and one state
- * reducer. The result is isomorphic.
- */
-declare function agent<M extends UIMessage = UIMessage, E extends EventDefs = Record<never, never>>(options: AgentOptions<M, E>): AgentDefinition<M, AIEventDefs<M> & E>;
-//#endregion
-export { AIEventDefs, AIInputs, AIState, ActiveGeneration, ActiveGenerationProjection, AgentDefinition, AgentOptions, AgentStatus, ApprovalRequestedPayload, ApprovalRespondedPayload, CompactionCompletedPayload, CompactionRequestedPayload, CompactionState, type FinishReason, GenerationCompletedPayload, GenerationFailedPayload, GenerationProgressPayload, GenerationReason, GenerationRequestedPayload, GenerationStartedPayload, GenerationUsage, InputRequestedPayload, InputRespondedPayload, type JSONValue, type LanguageModelUsage, MessageCompletedPayload, MessageCreatedPayload, MessageInterruptedPayload, PendingApproval, PendingInput, RetryGenerationOptions, RetryRequestedPayload, SessionClosedPayload, SessionCreatedPayload, ToolActivity, ToolCalledPayload, ToolResultPayload, type UIMessage, type UIMessageChunk, agent, createEvents, createReducer, deriveUIMessages, events, inputs, reduceAIState };
+import { A as MessageInterruptedPayload, B as UIMessage, C as GenerationUsage, D as LanguageModelUsage, E as JSONValue, F as SessionClosedPayload, G as deriveUIMessages, H as agent, I as SessionCreatedPayload, J as inputs, K as events, L as ToolActivity, M as PendingInput, N as RetryGenerationOptions, O as MessageCompletedPayload, P as RetryRequestedPayload, R as ToolCalledPayload, S as GenerationStartedPayload, T as InputRespondedPayload, U as createEvents, V as UIMessageChunk, W as createReducer, Y as reduceAIState, _ as GenerationCompletedPayload, a as ActiveGenerationProjection, b as GenerationReason, c as AgentStatus, d as ApprovalRequestedPayload, f as ApprovalRespondedPayload, g as FinishReason, h as CompactionState, i as ActiveGeneration, j as PendingApproval, k as MessageCreatedPayload, l as AgentToolContext, m as CompactionRequestedPayload, n as AIInputs, o as AgentDefinition, p as CompactionCompletedPayload, q as handlerContext, r as AIState, s as AgentOptions, t as AIEventDefs, u as AgentToolEventDefs, v as GenerationFailedPayload, w as InputRequestedPayload, x as GenerationRequestedPayload, y as GenerationProgressPayload, z as ToolResultPayload } from "./ai-B4YhEnfw.js";
+export { AIEventDefs, AIInputs, AIState, ActiveGeneration, ActiveGenerationProjection, AgentDefinition, AgentOptions, AgentStatus, AgentToolContext, AgentToolEventDefs, ApprovalRequestedPayload, ApprovalRespondedPayload, CompactionCompletedPayload, CompactionRequestedPayload, CompactionState, type FinishReason, GenerationCompletedPayload, GenerationFailedPayload, GenerationProgressPayload, GenerationReason, GenerationRequestedPayload, GenerationStartedPayload, GenerationUsage, InputRequestedPayload, InputRespondedPayload, type JSONValue, type LanguageModelUsage, MessageCompletedPayload, MessageCreatedPayload, MessageInterruptedPayload, PendingApproval, PendingInput, RetryGenerationOptions, RetryRequestedPayload, SessionClosedPayload, SessionCreatedPayload, ToolActivity, ToolCalledPayload, ToolResultPayload, type UIMessage, type UIMessageChunk, agent, createEvents, createReducer, deriveUIMessages, events, handlerContext, inputs, reduceAIState };