experimental-a2 0.19.0 → 0.21.1

package/dist/ai-server.js

@@ -1,13 +1,13 @@
-import { i as A2Error } from "./session-clear-DGI4jTQ5.js";
-import { a as reducer, s as validateSync } from "./reducer-BAXc8JyF.js";
-import { h as installAmbientToolScopeStorage, m as failedPreviousAttempt, p as consumeSchedulerSendFailure } from "./internal-CscSY7K_.js";
+import { i as A2Error } from "./session-clear-wHd5GCpZ.js";
+import { a as reducer, s as validateSync } from "./reducer-CjmvZSAb.js";
+import { h as installAmbientToolScopeStorage, m as failedPreviousAttempt, p as consumeSchedulerSendFailure } from "./internal-Bc4PvfHk.js";
 import { t as boundedCache } from "./bounded-cache-CPHWF-8h.js";
-import { a as CONSOLE_TELEMETRY, o as setServerFetchHooks, t as createServer } from "./server-D-Y4pqwu.js";
+import { a as CONSOLE_TELEMETRY, n as createServer, o as setServerFetchHooks } from "./server-DvIB8kj5.js";
 import { a as unexplainedAdmission, i as steersOn, n as initialControlState, r as steeringEnd, t as STEERED_REASON } from "./ai-control-CvLbW10B.js";
-import { A as interruptUIMessage, C as reduceToolActivity, D as flattenProgressBatches, E as upsertResponse, M as assertMessageBase, N as restoreMessageBase, O as progressBatches, S as projectedMessage, T as upsertMessage, _ as applyControlChanges, a as events, b as findMessage, d as appSlice, f as isAgentContract, g as isUIMessage, h as aiEventId, j as pauseUIMessage, k as progressOwnedBy, m as AI_EVENT_ID_PREFIX, n as agentContract, p as withAIStorage, u as appReducerOf, v as controlCommit, w as retainMessageBase, y as advanceProjection } from "./ai-mMiWul1T.js";
-import { a as harnessDriver, c as harnessStates, d as registerHarnessStates, s as harnessLog } from "./ai-harness-BuuC-gJ3.js";
+import { A as interruptUIMessage, C as reduceToolActivity, D as flattenProgressBatches, E as upsertResponse, M as assertMessageBase, N as restoreMessageBase, O as progressBatches, S as projectedMessage, T as upsertMessage, _ as applyControlChanges, b as findMessage, d as appSlice, f as isAgentContract, g as isUIMessage, h as aiEventId, j as pauseUIMessage, k as progressOwnedBy, m as AI_EVENT_ID_PREFIX, n as agentContract, o as events, p as withAIStorage, u as appReducerOf, v as controlCommit, w as retainMessageBase, y as advanceProjection } from "./ai-CzE672Q3.js";
+import { a as gatewaySessionAffinity, c as harnessLog, f as registerHarnessStates, l as harnessStates, o as harnessDriver } from "./ai-harness-dQeJaR1t.js";
+import { t as runtimeConsumer } from "./runtime-consumer-BWmht60z.js";
 import { AsyncLocalStorage } from "node:async_hooks";
-import { createHash } from "node:crypto";
 import { asSchema, convertToModelMessages, isToolUIPart, stepCountIs, streamText, toUIMessageStream } from "ai";
 import { Buffer } from "node:buffer";
 //#region src/ai-coordinator.ts
@@ -1607,8 +1607,8 @@ const measuredInputTokens = (options) => {
 //#endregion
 //#region src/ai-agent.ts
 /**
-* The server-only implementation of the experimental-a2/ai protocol: the
-* handler table and `agent()`, its assembly as an ordinary A2 server.
+* The server-only implementation of the experimental-a2/ai protocol: its
+* consumers and `agent()`, its assembly as an ordinary A2 server.
 */
 const isAgentIngressEvent = (event) => {
 	if (event.type !== "ai.control.requested") return false;
@@ -1659,8 +1659,8 @@ const withModelSettings = (options) => {
 const instructionTexts = (instructions) => instructions === void 0 ? [] : typeof instructions === "string" ? [instructions] : (Array.isArray(instructions) ? instructions : [instructions]).map((message) => message.content);
 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) {
+/** Run `fn` with `ctx` as the ambient tool consumer context, for direct tool tests. */
+function runWithConsumerContext(ctx, fn) {
 	return toolScopeStorage.run({ context: ctx }, fn);
 }
 /** Sessions whose latest step prompt one process keeps for that step's tools. */
@@ -1736,7 +1736,10 @@ async function generateWithAISDK(options) {
 		...context.generation,
 		headers: Object.keys(headers).some((key) => key.toLowerCase() === "x-session-affinity") ? headers : {
 			...headers,
-			"x-session-affinity": createHash("sha256").update(JSON.stringify([agentName, context.event.sessionId])).digest("hex")
+			"x-session-affinity": gatewaySessionAffinity({
+				agentName,
+				sessionId: context.event.sessionId
+			})
 		},
 		providerOptions: {
 			...context.generation.providerOptions,
@@ -2116,7 +2119,7 @@ const assertNoLoopOptions = (options) => {
 		throw new TypeError(harness === "ai-sdk" ? `${key} belongs to the harness: agent({ harness: { type: 'ai-sdk', ${key} } })` : `${key} is an option of the AI SDK harness; ${harness}() runs its own loop`);
 	}
 };
-function createHandlers(options) {
+function createConsumers(options) {
 	if (options.model === void 0) throw new TypeError("agent 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");
@@ -2740,8 +2743,14 @@ function createHandlers(options) {
 			};
 		}
 	};
-	return {
-		"ai.model.metadata.requested": { handler: async (ctx) => {
+	const contract = options.contract;
+	const { toolLane } = options;
+	return [
+		runtimeConsumer({
+			contract,
+			name: "a2.ai.metadata",
+			types: ["ai.model.metadata.requested"]
+		}).run(async (ctx) => {
 			const limits = await readModelLimits({
 				modelId: ctx.event.payload.modelId,
 				signal: ctx.signal
@@ -2755,27 +2764,29 @@ function createHandlers(options) {
 					limits
 				}
 			};
-		} },
-		"ai.control.requested": {
+		}),
+		runtimeConsumer({
+			contract,
+			name: "a2.ai.control",
+			types: [
+				"ai.control.requested",
+				"ai.work.reported",
+				"ai.session.closed"
+			],
 			lane: ({ event }) => {
+				if (event.type !== "ai.control.requested") return "a2.ai.control";
 				const command = event.payload;
 				if (event.id === "") throw new A2Error("INVALID_PAYLOAD", "AI command IDs must be nonempty");
 				if (event.id?.startsWith("a2.ai:")) throw new A2Error("INVALID_PAYLOAD", `AI command IDs cannot use the reserved '${AI_EVENT_ID_PREFIX}' prefix`);
 				if ((command.action === "send" || command.action === "steer") && event.id !== command.message.id) throw new A2Error("INVALID_PAYLOAD", "AI input event ID must equal message.id");
 				return "a2.ai.control";
-			},
-			handler: controlHandler
-		},
-		"ai.work.reported": {
-			lane: "a2.ai.control",
-			handler: controlHandler
-		},
-		"ai.session.closed": {
-			lane: "a2.ai.control",
-			handler: controlHandler
-		},
-		"ai.generation.requested": {
-			lane: "a2.ai.model",
+			}
+		}).run(controlHandler),
+		runtimeConsumer({
+			contract,
+			name: "a2.ai.model",
+			types: ["ai.generation.requested"],
+			lane: () => "a2.ai.model",
 			abortOn: {
 				"ai.control.requested": (event, trigger) => event.payload.action === "stop" && event.payload.turnId === trigger.payload.control?.turnId,
 				"ai.control.committed": (event, trigger) => {
@@ -2783,32 +2794,34 @@ function createHandlers(options) {
 					return active?.turnId !== trigger.payload.control?.turnId || active?.version !== trigger.payload.control?.version;
 				},
 				"ai.session.closed": true
-			},
-			handler: (ctx) => telemetry.span("a2.ai.generate", {
-				"a2.contract": options.contract.name,
-				"a2.session_id": ctx.event.sessionId,
-				"a2.ai.generate.request_id": ctx.event.id,
-				"a2.ai.generate.attempt": ctx.attempt
-			}, async () => {
-				try {
-					return control.settled({
-						ctx,
-						events: await generationHandler(ctx)
-					});
-				} catch (error) {
-					if (control.cancelled({
-						error,
-						signal: ctx.signal
-					})) return control.settled({
-						ctx,
-						events: void 0
-					});
-					throw error;
-				}
-			})
-		},
-		"ai.tool.execution.requested": {
-			...options.toolLane === void 0 ? {} : { lane: options.toolLane },
+			}
+		}).run((ctx) => telemetry.span("a2.ai.generate", {
+			"a2.contract": options.contract.name,
+			"a2.session_id": ctx.event.sessionId,
+			"a2.ai.generate.request_id": ctx.event.id,
+			"a2.ai.generate.attempt": ctx.attempt
+		}, async () => {
+			try {
+				return control.settled({
+					ctx,
+					events: await generationHandler(ctx)
+				});
+			} catch (error) {
+				if (control.cancelled({
+					error,
+					signal: ctx.signal
+				})) return control.settled({
+					ctx,
+					events: void 0
+				});
+				throw error;
+			}
+		})),
+		runtimeConsumer({
+			contract,
+			name: "a2.ai.tools",
+			types: ["ai.tool.execution.requested"],
+			...toolLane === void 0 ? {} : { lane: () => toolLane },
 			abortOn: {
 				"ai.control.requested": (event, trigger) => event.payload.action === "stop" && event.payload.turnId === trigger.payload.turnId,
 				"ai.generation.failed": (event, trigger) => event.payload.generationId === trigger.payload.call.generationId,
@@ -2817,21 +2830,21 @@ function createHandlers(options) {
 					return active?.turnId !== trigger.payload.turnId || active?.version !== trigger.payload.version;
 				},
 				"ai.session.closed": true
-			},
-			handler: (ctx) => telemetry.span("a2.ai.tool", {
-				"a2.contract": options.contract.name,
-				"a2.session_id": ctx.event.sessionId,
-				"a2.ai.tool.request_id": ctx.event.payload.call.requestId,
-				"a2.ai.tool.call_id": ctx.event.payload.call.toolCallId,
-				"a2.ai.tool.name": ctx.event.payload.call.toolName,
-				"a2.ai.tool.attempt": ctx.attempt
-			}, () => handleToolExecution(ctx))
-		}
-	};
+			}
+		}).run((ctx) => telemetry.span("a2.ai.tool", {
+			"a2.contract": options.contract.name,
+			"a2.session_id": ctx.event.sessionId,
+			"a2.ai.tool.request_id": ctx.event.payload.call.requestId,
+			"a2.ai.tool.call_id": ctx.event.payload.call.toolCallId,
+			"a2.ai.tool.name": ctx.event.payload.call.toolName,
+			"a2.ai.tool.attempt": ctx.attempt
+		}, () => handleToolExecution(ctx)))
+	];
 }
 function agent(options) {
 	assertNoLoopOptions(options);
-	const { model, tools, instructions, maxSteps, harness, toolRecovery, progress, toolLane, handlers, ...rest } = options;
+	if (options.toolLane !== void 0 && (typeof options.toolLane !== "string" || options.toolLane.length === 0)) throw new TypeError("toolLane must be a non-empty string");
+	const { model, tools, instructions, maxSteps, harness, toolRecovery, progress, toolLane, ...rest } = options;
 	let contract;
 	let serverOptions;
 	if ("contract" in rest) {
@@ -2850,7 +2863,7 @@ function agent(options) {
 		serverOptions = remaining;
 	}
 	const states = harnessStates();
-	const builtIns = createHandlers({
+	const builtIns = createConsumers({
 		contract,
 		model,
 		harnessStates: states,
@@ -2863,14 +2876,10 @@ function agent(options) {
 		...progress === void 0 ? {} : { progress },
 		...toolLane === void 0 ? {} : { toolLane }
 	});
-	for (const type of Object.keys(handlers ?? {})) if (Object.hasOwn(builtIns, type)) throw new TypeError(`agent handles '${type}' itself`);
 	const server = createServer({
 		...serverOptions,
 		contract,
-		handlers: {
-			...builtIns,
-			...handlers
-		}
+		consumers: [...builtIns, ...serverOptions.consumers ?? []]
 	});
 	registerHarnessStates(server, states);
 	const applicationEvents = "contract" in rest;
@@ -2881,6 +2890,6 @@ function agent(options) {
 	return server;
 }
 //#endregion
-export { agent, runWithHandlerContext };
+export { agent, runWithConsumerContext };
 
 //# sourceMappingURL=ai-server.js.map