@yolk-sdk/vercel-workflows-runtime 0.0.1-canary.1 → 0.0.1-canary.10

package/README.md

@@ -5,7 +5,7 @@ Vercel Workflow-specific contracts for durable agent model/tool step loops.
 ## Install
 
 ```bash
-pnpm add @yolk-sdk/vercel-workflows-runtime@canary @yolk-sdk/agent@canary effect workflow
+pnpm add @yolk-sdk/vercel-workflows-runtime@canary effect workflow
 ```
 
 Canary APIs are unstable. Keep all `@yolk-sdk/*` packages on the same version.
@@ -34,12 +34,55 @@ import {
 
 - model step: produce model events/tool calls
 - tool batch step: execute requested tools
-- awaiting-input state: carry pending HITL hook data between steps
+- awaiting-input state: carry pending HITL hook data, wait through `awaitInput`, then rerun the
+  same tool batch with accumulated responses
 - close step: flush/close output stream
-- terminal status: completed, step failure, close failure, or max turns exceeded
+- terminal status: completed, step failure, await-input failure, close failure, or max turns exceeded
 
 Continuation state stays plain serializable data for Workflow persistence.
 
+## Minimal workflow wrapper
+
+Host apps keep concrete Workflow directives local and pass them into the package loop:
+
+```ts
+import { createHook } from 'workflow'
+import { runVercelAgentWorkflow } from '@yolk-sdk/vercel-workflows-runtime/workflow'
+
+export async function runAgentWorkflow(input: { request: unknown; context: unknown }) {
+  'use workflow'
+
+  return await runVercelAgentWorkflow({
+    input,
+    runModelStep,
+    runToolBatchStep,
+    closeStream,
+    writeError,
+    awaitInput: async awaitingInput => {
+      using hook = createHook<unknown>({ token: awaitingInput.hookToken })
+
+      return await hook
+    }
+  })
+}
+```
+
+`runModelStep`, `runToolBatchStep`, `closeStream`, and `writeError` are host-owned
+`'use step'` functions. Keep provider calls, tools, persistence, telemetry, and Effect runtimes in
+those steps, not in the `'use workflow'` orchestration body.
+
+## Awaiting input
+
+When a tool batch needs HITL input, return `awaitingInput` from `runToolBatchStep`. The package
+loop will:
+
+1. pass that payload to `awaitInput`
+2. wait for the hook/webhook response
+3. rerun the same tool batch with `hitlResponses: [...previous, response]`
+
+The host owns `hookToken` construction, auth, resume routes, and response decoding. Use stable
+tokens that the resume side can reconstruct or store.
+
 ## Retry policy
 
 Default retry policy is `noWorkflowStepRetry` (`maxAttempts: 1`). Retries are opt-in because streamed retries can duplicate chunks unless host/client de-dupe is ready.

package/dist/workflow-loop.d.mts

@@ -72,6 +72,12 @@ type VercelAgentWorkflowRunResult = {
   readonly turn: number;
   readonly error: unknown;
   readonly state: SerializableWorkflowState;
+} | {
+  readonly _tag: 'AwaitInputFailed';
+  readonly turn: number;
+  readonly error: unknown;
+  readonly awaitingInput: VercelAgentWorkflowAwaitingInput;
+  readonly state: SerializableWorkflowState;
 } | {
   readonly _tag: 'CloseStreamFailed';
   readonly turns: number;
@@ -90,8 +96,10 @@ type VercelAgentWorkflowLoopConfig = {
   readonly runToolBatchStep: (input: VercelAgentWorkflowToolBatchStepInput) => Promise<VercelAgentWorkflowToolBatchStepResult>;
   readonly closeStream: () => Promise<void>;
   readonly writeError: (error: unknown) => Promise<void>;
+  readonly awaitInput?: (input: VercelAgentWorkflowAwaitingInput) => Promise<unknown>;
   readonly modelStepRetry?: VercelAgentWorkflowStepRetryPolicy;
   readonly toolBatchStepRetry?: VercelAgentWorkflowStepRetryPolicy;
+  readonly awaitInputRetry?: VercelAgentWorkflowStepRetryPolicy;
   readonly closeStreamRetry?: VercelAgentWorkflowStepRetryPolicy;
 };
 declare const defaultMaxWorkflowTurns = 500;

package/dist/workflow-loop.d.mts.map

@@ -1 +1 @@
-{"version":3,"file":"workflow-loop.d.mts","names":[],"sources":["../src/workflow-loop.ts"],"mappings":";KAAY,wBAAA;EAAA,SACD,OAAA;EAAA,SACA,OAAO;AAAA;AAAA,KAGN,yBAAA;EAAA,SACD,OAAA;EAAA,SACA,QAAA,GAAW,aAAA;EAAA,SACX,eAAA,EAAiB,aAAa;EAAA,SAC9B,KAAA;EAAA,SACA,IAAA;EAAA,SACA,aAAA;AAAA;AAAA,KAGC,iCAAA;EAAA,SACD,OAAA;EAAA,SACA,KAAA,EAAO,yBAAyB;AAAA;AAAA,KAG/B,kCAAA;EAAA,SACD,IAAA;EAAA,SACA,QAAA,EAAU,aAAA;EAAA,SACV,eAAA,EAAiB,aAAA;EAAA,SACjB,SAAA,EAAW,aAAA;EAAA,SACX,KAAA;EAAA,SACA,IAAA;EAAA,SACA,aAAA;AAAA;AAAA,KAGC,qCAAA;EAAA,SACD,OAAA;EAAA,SACA,OAAA;EAAA,SACA,KAAA,EAAO,aAAA;EAAA,SACP,eAAA,EAAiB,aAAA;EAAA,SACjB,aAAA,GAAgB,aAAA;EAAA,SAChB,KAAA;EAAA,SACA,IAAA;EAAA,SACA,aAAA;AAAA;AAAA,KAGC,gCAAA;EAAA,SACD,SAAA;EAAA,SACA,QAAA,EAAU,aAAA;EAAA,SACV,QAAA,EAAU,aAAa;EAAA,SACvB,KAAA;EAAA,SACA,KAAA;EAAA,SACA,aAAA;AAAA;AAAA,KAGC,sCAAA;EAAA,SACD,QAAA,EAAU,aAAA;EAAA,SACV,eAAA,EAAiB,aAAA;EAAA,SACjB,aAAA,GAAgB,gCAAA;EAAA,SAChB,aAAA;AAAA;AAAA,KAGC,kBAAA;EAAA,SAEG,IAAA;EAAA,SACA,KAAA,EAAO,CAAC;AAAA;EAAA,SAGR,IAAA;EAAA,SACA,KAAA;AAAA;AAAA,KAGH,kCAAA;EAAA,SACD,WAAW;AAAA;AAAA,KAGV,4BAAA;EAAA,SAEG,IAAA;EAAA,SACA,KAAA;EAAA,SACA,KAAA,EAAO,yBAAA;AAAA;EAAA,SAGP,IAAA;EAAA,SACA,IAAA;EAAA,SACA,KAAA;EAAA,SACA,KAAA,EAAO,yBAAA;AAAA;EAAA,SAGP,IAAA;EAAA,SACA,IAAA;EAAA,SACA,KAAA;EAAA,SACA,KAAA,EAAO,yBAAA;AAAA;EAAA,SAGP,IAAA;EAAA,SACA,KAAA;EAAA,SACA,KAAA;EAAA,SACA,KAAA,EAAO,yBAAA;AAAA;EAAA,SAGP,IAAA;EAAA,SACA,QAAA;EAAA,SACA,KAAA,EAAO,KAAA;EAAA,SACP,KAAA,EAAO,yBAAA;AAAA;AAAA,KAGV,6BAAA;EAAA,SACD,KAAA,EAAO,wBAAA;EAAA,SACP,QAAA;EAAA,SACA,YAAA,GACP,KAAA,EAAO,iCAAA,KACJ,OAAA,CAAQ,kCAAA;EAAA,SACJ,gBAAA,GACP,KAAA,EAAO,qCAAA,KACJ,OAAA,CAAQ,sCAAA;EAAA,SACJ,WAAA,QAAmB,OAAA;EAAA,SACnB,UAAA,GAAa,KAAA,cAAmB,OAAA;EAAA,SAChC,cAAA,GAAiB,kCAAA;EAAA,SACjB,kBAAA,GAAqB,kCAAA;EAAA,SACrB,gBAAA,GAAmB,kCAAA;AAAA;AAAA,cAGjB,uBAAA;AAAA,cACA,mBAAA,EAAqB,kCAAuD;AAAA,cAE5E,kBAAA,MAAyB,OAAA,EAAS,OAAA,CAAQ,CAAA,MAAK,OAAA,CAAQ,kBAAA,CAAmB,CAAA;AAAA,iBAejE,iBAAA,GAAA,CACpB,OAAA,QAAe,OAAA,CAAQ,CAAA,GACvB,MAAA,GAAS,kCAAA,GACR,OAAA,CAAQ,CAAA;AAAA,iBAiBW,sBAAA,CACpB,MAAA,EAAQ,6BAAA,GACP,OAAA,CAAQ,4BAAA"}
+{"version":3,"file":"workflow-loop.d.mts","names":[],"sources":["../src/workflow-loop.ts"],"mappings":";KAAY,wBAAA;EAAA,SACD,OAAA;EAAA,SACA,OAAO;AAAA;AAAA,KAGN,yBAAA;EAAA,SACD,OAAA;EAAA,SACA,QAAA,GAAW,aAAA;EAAA,SACX,eAAA,EAAiB,aAAa;EAAA,SAC9B,KAAA;EAAA,SACA,IAAA;EAAA,SACA,aAAA;AAAA;AAAA,KAGC,iCAAA;EAAA,SACD,OAAA;EAAA,SACA,KAAA,EAAO,yBAAyB;AAAA;AAAA,KAG/B,kCAAA;EAAA,SACD,IAAA;EAAA,SACA,QAAA,EAAU,aAAA;EAAA,SACV,eAAA,EAAiB,aAAA;EAAA,SACjB,SAAA,EAAW,aAAA;EAAA,SACX,KAAA;EAAA,SACA,IAAA;EAAA,SACA,aAAA;AAAA;AAAA,KAGC,qCAAA;EAAA,SACD,OAAA;EAAA,SACA,OAAA;EAAA,SACA,KAAA,EAAO,aAAA;EAAA,SACP,eAAA,EAAiB,aAAA;EAAA,SACjB,aAAA,GAAgB,aAAA;EAAA,SAChB,KAAA;EAAA,SACA,IAAA;EAAA,SACA,aAAA;AAAA;AAAA,KAGC,gCAAA;EAAA,SACD,SAAA;EAAA,SACA,QAAA,EAAU,aAAA;EAAA,SACV,QAAA,EAAU,aAAa;EAAA,SACvB,KAAA;EAAA,SACA,KAAA;EAAA,SACA,aAAA;AAAA;AAAA,KAGC,sCAAA;EAAA,SACD,QAAA,EAAU,aAAA;EAAA,SACV,eAAA,EAAiB,aAAA;EAAA,SACjB,aAAA,GAAgB,gCAAA;EAAA,SAChB,aAAA;AAAA;AAAA,KAGC,kBAAA;EAAA,SAEG,IAAA;EAAA,SACA,KAAA,EAAO,CAAC;AAAA;EAAA,SAGR,IAAA;EAAA,SACA,KAAA;AAAA;AAAA,KAGH,kCAAA;EAAA,SACD,WAAW;AAAA;AAAA,KAGV,4BAAA;EAAA,SAEG,IAAA;EAAA,SACA,KAAA;EAAA,SACA,KAAA,EAAO,yBAAA;AAAA;EAAA,SAGP,IAAA;EAAA,SACA,IAAA;EAAA,SACA,KAAA;EAAA,SACA,KAAA,EAAO,yBAAA;AAAA;EAAA,SAGP,IAAA;EAAA,SACA,IAAA;EAAA,SACA,KAAA;EAAA,SACA,KAAA,EAAO,yBAAA;AAAA;EAAA,SAGP,IAAA;EAAA,SACA,IAAA;EAAA,SACA,KAAA;EAAA,SACA,aAAA,EAAe,gCAAA;EAAA,SACf,KAAA,EAAO,yBAAA;AAAA;EAAA,SAGP,IAAA;EAAA,SACA,KAAA;EAAA,SACA,KAAA;EAAA,SACA,KAAA,EAAO,yBAAA;AAAA;EAAA,SAGP,IAAA;EAAA,SACA,QAAA;EAAA,SACA,KAAA,EAAO,KAAA;EAAA,SACP,KAAA,EAAO,yBAAA;AAAA;AAAA,KAGV,6BAAA;EAAA,SACD,KAAA,EAAO,wBAAA;EAAA,SACP,QAAA;EAAA,SACA,YAAA,GACP,KAAA,EAAO,iCAAA,KACJ,OAAA,CAAQ,kCAAA;EAAA,SACJ,gBAAA,GACP,KAAA,EAAO,qCAAA,KACJ,OAAA,CAAQ,sCAAA;EAAA,SACJ,WAAA,QAAmB,OAAA;EAAA,SACnB,UAAA,GAAa,KAAA,cAAmB,OAAA;EAAA,SAChC,UAAA,IAAc,KAAA,EAAO,gCAAA,KAAqC,OAAA;EAAA,SAC1D,cAAA,GAAiB,kCAAA;EAAA,SACjB,kBAAA,GAAqB,kCAAA;EAAA,SACrB,eAAA,GAAkB,kCAAA;EAAA,SAClB,gBAAA,GAAmB,kCAAA;AAAA;AAAA,cAGjB,uBAAA;AAAA,cACA,mBAAA,EAAqB,kCAAuD;AAAA,cAE5E,kBAAA,MAAyB,OAAA,EAAS,OAAA,CAAQ,CAAA,MAAK,OAAA,CAAQ,kBAAA,CAAmB,CAAA;AAAA,iBAkBjE,iBAAA,GAAA,CACpB,OAAA,QAAe,OAAA,CAAQ,CAAA,GACvB,MAAA,GAAS,kCAAA,GACR,OAAA,CAAQ,CAAA;AAAA,iBAiBW,sBAAA,CACpB,MAAA,EAAQ,6BAAA,GACP,OAAA,CAAQ,4BAAA"}

package/dist/workflow-loop.mjs

@@ -10,6 +10,7 @@ const settleWorkflowStep = (promise) => promise.then((value) => ({
 }));
 const workflowMaxTurnsError = (maxTurns) => /* @__PURE__ */ new Error(`Vercel agent workflow exceeded max turns: ${maxTurns}`);
 const writeErrorSafely = (writeError, error) => writeError(error).catch(() => void 0);
+const missingAwaitInputHandlerError = () => /* @__PURE__ */ new Error("Vercel agent workflow awaiting input requested but no awaitInput handler is configured");
 const maxRetryAttempts = (policy) => Math.max(1, Math.floor(policy?.maxAttempts ?? noWorkflowStepRetry.maxAttempts));
 async function retryWorkflowStep(runStep, policy) {
 	const maxAttempts = maxRetryAttempts(policy);
@@ -22,20 +23,21 @@ async function retryWorkflowStep(runStep, policy) {
 	}
 }
 async function runVercelAgentWorkflow(config) {
+	const { input, maxTurns: configuredMaxTurns, runModelStep, runToolBatchStep, closeStream, writeError, awaitInput, modelStepRetry, toolBatchStepRetry, awaitInputRetry, closeStreamRetry } = config;
 	let state = {
-		request: config.input.request,
+		request: input.request,
 		createdMessages: [],
 		turn: 1,
 		eventSequence: 0
 	};
-	const maxTurns = config.maxTurns ?? 500;
+	const maxTurns = configuredMaxTurns ?? 500;
 	for (let step = 0; step < maxTurns; step++) {
-		const modelResult = await settleWorkflowStep(retryWorkflowStep(() => config.runModelStep({
-			context: config.input.context,
+		const modelResult = await settleWorkflowStep(retryWorkflowStep(() => runModelStep({
+			context: input.context,
 			state
-		}), config.modelStepRetry));
+		}), modelStepRetry));
 		if (modelResult._tag === "Failure") {
-			await writeErrorSafely(config.writeError, modelResult.error);
+			await writeErrorSafely(writeError, modelResult.error);
 			return {
 				_tag: "ModelStepFailed",
 				turn: state.turn,
@@ -44,9 +46,9 @@ async function runVercelAgentWorkflow(config) {
 			};
 		}
 		if (modelResult.value.done) {
-			const closeResult = await settleWorkflowStep(retryWorkflowStep(config.closeStream, config.closeStreamRetry));
+			const closeResult = await settleWorkflowStep(retryWorkflowStep(closeStream, closeStreamRetry));
 			if (closeResult._tag === "Failure") {
-				await writeErrorSafely(config.writeError, closeResult.error);
+				await writeErrorSafely(writeError, closeResult.error);
 				return {
 					_tag: "CloseStreamFailed",
 					turns: modelResult.value.turn,
@@ -60,34 +62,72 @@ async function runVercelAgentWorkflow(config) {
 				state
 			};
 		}
-		const toolsResult = await settleWorkflowStep(retryWorkflowStep(() => config.runToolBatchStep({
-			context: config.input.context,
-			request: config.input.request,
-			calls: modelResult.value.toolCalls,
-			createdMessages: modelResult.value.createdMessages,
-			turn: modelResult.value.turn,
-			eventSequence: modelResult.value.eventSequence ?? state.eventSequence
-		}), config.toolBatchStepRetry));
-		if (toolsResult._tag === "Failure") {
-			await writeErrorSafely(config.writeError, toolsResult.error);
+		let completedToolsResult;
+		let toolHitlResponses = [];
+		let toolEventSequence = modelResult.value.eventSequence ?? state.eventSequence;
+		for (;;) {
+			const toolsResult = await settleWorkflowStep(retryWorkflowStep(() => runToolBatchStep({
+				context: input.context,
+				request: input.request,
+				calls: modelResult.value.toolCalls,
+				createdMessages: modelResult.value.createdMessages,
+				hitlResponses: toolHitlResponses,
+				usage: modelResult.value.usage,
+				turn: modelResult.value.turn,
+				eventSequence: toolEventSequence
+			}), toolBatchStepRetry));
+			if (toolsResult._tag === "Failure") {
+				await writeErrorSafely(writeError, toolsResult.error);
+				return {
+					_tag: "ToolBatchStepFailed",
+					turn: modelResult.value.turn,
+					error: toolsResult.error,
+					state
+				};
+			}
+			if (toolsResult.value.awaitingInput === void 0) {
+				completedToolsResult = toolsResult.value;
+				break;
+			}
+			const awaitingInput = toolsResult.value.awaitingInput;
+			toolEventSequence = awaitingInput.eventSequence ?? toolsResult.value.eventSequence ?? toolEventSequence;
+			const hitlResponse = await settleWorkflowStep(retryWorkflowStep(() => {
+				if (awaitInput === void 0) return Promise.reject(missingAwaitInputHandlerError());
+				return awaitInput(awaitingInput);
+			}, awaitInputRetry));
+			if (hitlResponse._tag === "Failure") {
+				await writeErrorSafely(writeError, hitlResponse.error);
+				return {
+					_tag: "AwaitInputFailed",
+					turn: modelResult.value.turn,
+					error: hitlResponse.error,
+					awaitingInput,
+					state
+				};
+			}
+			toolHitlResponses = [...toolHitlResponses, hitlResponse.value];
+		}
+		if (completedToolsResult === void 0) {
+			const error = /* @__PURE__ */ new Error("Vercel agent workflow tool batch did not complete");
+			await writeErrorSafely(writeError, error);
 			return {
 				_tag: "ToolBatchStepFailed",
 				turn: modelResult.value.turn,
-				error: toolsResult.error,
+				error,
 				state
 			};
 		}
 		state = {
-			request: config.input.request,
-			messages: [...modelResult.value.messages, ...toolsResult.value.messages],
-			createdMessages: toolsResult.value.createdMessages,
+			request: input.request,
+			messages: [...modelResult.value.messages, ...completedToolsResult.messages],
+			createdMessages: completedToolsResult.createdMessages,
 			usage: modelResult.value.usage,
 			turn: modelResult.value.turn + 1,
-			eventSequence: toolsResult.value.eventSequence ?? modelResult.value.eventSequence ?? state.eventSequence
+			eventSequence: completedToolsResult.eventSequence ?? toolEventSequence
 		};
 	}
 	const error = workflowMaxTurnsError(maxTurns);
-	await writeErrorSafely(config.writeError, error);
+	await writeErrorSafely(writeError, error);
 	return {
 		_tag: "MaxTurnsExceeded",
 		maxTurns,

package/dist/workflow-loop.mjs.map

@@ -1 +1 @@
-{"version":3,"file":"workflow-loop.mjs","names":[],"sources":["../src/workflow-loop.ts"],"sourcesContent":["export type VercelAgentWorkflowInput = {\n  readonly request: unknown\n  readonly context: unknown\n}\n\nexport type SerializableWorkflowState = {\n  readonly request: unknown\n  readonly messages?: ReadonlyArray<unknown>\n  readonly createdMessages: ReadonlyArray<unknown>\n  readonly usage?: unknown\n  readonly turn: number\n  readonly eventSequence?: number\n}\n\nexport type VercelAgentWorkflowModelStepInput = {\n  readonly context: unknown\n  readonly state: SerializableWorkflowState\n}\n\nexport type VercelAgentWorkflowModelStepResult = {\n  readonly done: boolean\n  readonly messages: ReadonlyArray<unknown>\n  readonly createdMessages: ReadonlyArray<unknown>\n  readonly toolCalls: ReadonlyArray<unknown>\n  readonly usage: unknown\n  readonly turn: number\n  readonly eventSequence?: number\n}\n\nexport type VercelAgentWorkflowToolBatchStepInput = {\n  readonly context: unknown\n  readonly request: unknown\n  readonly calls: ReadonlyArray<unknown>\n  readonly createdMessages: ReadonlyArray<unknown>\n  readonly hitlResponses?: ReadonlyArray<unknown>\n  readonly usage?: unknown\n  readonly turn?: number\n  readonly eventSequence?: number\n}\n\nexport type VercelAgentWorkflowAwaitingInput = {\n  readonly hookToken: string\n  readonly requests: ReadonlyArray<unknown>\n  readonly messages: ReadonlyArray<unknown>\n  readonly usage: unknown\n  readonly turns: number\n  readonly eventSequence?: number\n}\n\nexport type VercelAgentWorkflowToolBatchStepResult = {\n  readonly messages: ReadonlyArray<unknown>\n  readonly createdMessages: ReadonlyArray<unknown>\n  readonly awaitingInput?: VercelAgentWorkflowAwaitingInput\n  readonly eventSequence?: number\n}\n\nexport type WorkflowStepResult<A> =\n  | {\n      readonly _tag: 'Success'\n      readonly value: A\n    }\n  | {\n      readonly _tag: 'Failure'\n      readonly error: unknown\n    }\n\nexport type VercelAgentWorkflowStepRetryPolicy = {\n  readonly maxAttempts: number\n}\n\nexport type VercelAgentWorkflowRunResult =\n  | {\n      readonly _tag: 'Completed'\n      readonly turns: number\n      readonly state: SerializableWorkflowState\n    }\n  | {\n      readonly _tag: 'ModelStepFailed'\n      readonly turn: number\n      readonly error: unknown\n      readonly state: SerializableWorkflowState\n    }\n  | {\n      readonly _tag: 'ToolBatchStepFailed'\n      readonly turn: number\n      readonly error: unknown\n      readonly state: SerializableWorkflowState\n    }\n  | {\n      readonly _tag: 'CloseStreamFailed'\n      readonly turns: number\n      readonly error: unknown\n      readonly state: SerializableWorkflowState\n    }\n  | {\n      readonly _tag: 'MaxTurnsExceeded'\n      readonly maxTurns: number\n      readonly error: Error\n      readonly state: SerializableWorkflowState\n    }\n\nexport type VercelAgentWorkflowLoopConfig = {\n  readonly input: VercelAgentWorkflowInput\n  readonly maxTurns?: number\n  readonly runModelStep: (\n    input: VercelAgentWorkflowModelStepInput\n  ) => Promise<VercelAgentWorkflowModelStepResult>\n  readonly runToolBatchStep: (\n    input: VercelAgentWorkflowToolBatchStepInput\n  ) => Promise<VercelAgentWorkflowToolBatchStepResult>\n  readonly closeStream: () => Promise<void>\n  readonly writeError: (error: unknown) => Promise<void>\n  readonly modelStepRetry?: VercelAgentWorkflowStepRetryPolicy\n  readonly toolBatchStepRetry?: VercelAgentWorkflowStepRetryPolicy\n  readonly closeStreamRetry?: VercelAgentWorkflowStepRetryPolicy\n}\n\nexport const defaultMaxWorkflowTurns = 500\nexport const noWorkflowStepRetry: VercelAgentWorkflowStepRetryPolicy = { maxAttempts: 1 }\n\nexport const settleWorkflowStep = <A>(promise: Promise<A>): Promise<WorkflowStepResult<A>> =>\n  promise.then(\n    value => ({ _tag: 'Success', value }),\n    error => ({ _tag: 'Failure', error })\n  )\n\nconst workflowMaxTurnsError = (maxTurns: number) =>\n  new Error(`Vercel agent workflow exceeded max turns: ${maxTurns}`)\n\nconst writeErrorSafely = (writeError: (error: unknown) => Promise<void>, error: unknown) =>\n  writeError(error).catch(() => undefined)\n\nconst maxRetryAttempts = (policy: VercelAgentWorkflowStepRetryPolicy | undefined) =>\n  Math.max(1, Math.floor(policy?.maxAttempts ?? noWorkflowStepRetry.maxAttempts))\n\nexport async function retryWorkflowStep<A>(\n  runStep: () => Promise<A>,\n  policy?: VercelAgentWorkflowStepRetryPolicy\n): Promise<A> {\n  const maxAttempts = maxRetryAttempts(policy)\n  let attempt = 1\n\n  for (;;) {\n    try {\n      return await runStep()\n    } catch (error) {\n      if (attempt >= maxAttempts) {\n        throw error\n      }\n\n      attempt += 1\n    }\n  }\n}\n\nexport async function runVercelAgentWorkflow(\n  config: VercelAgentWorkflowLoopConfig\n): Promise<VercelAgentWorkflowRunResult> {\n  let state: SerializableWorkflowState = {\n    request: config.input.request,\n    createdMessages: [],\n    turn: 1,\n    eventSequence: 0\n  }\n  const maxTurns = config.maxTurns ?? defaultMaxWorkflowTurns\n\n  for (let step = 0; step < maxTurns; step++) {\n    const modelResult = await settleWorkflowStep(\n      retryWorkflowStep(\n        () => config.runModelStep({ context: config.input.context, state }),\n        config.modelStepRetry\n      )\n    )\n\n    if (modelResult._tag === 'Failure') {\n      await writeErrorSafely(config.writeError, modelResult.error)\n      return {\n        _tag: 'ModelStepFailed',\n        turn: state.turn,\n        error: modelResult.error,\n        state\n      }\n    }\n\n    if (modelResult.value.done) {\n      const closeResult = await settleWorkflowStep(\n        retryWorkflowStep(config.closeStream, config.closeStreamRetry)\n      )\n\n      if (closeResult._tag === 'Failure') {\n        await writeErrorSafely(config.writeError, closeResult.error)\n\n        return {\n          _tag: 'CloseStreamFailed',\n          turns: modelResult.value.turn,\n          error: closeResult.error,\n          state\n        }\n      }\n\n      return {\n        _tag: 'Completed',\n        turns: modelResult.value.turn,\n        state\n      }\n    }\n\n    const toolsResult = await settleWorkflowStep(\n      retryWorkflowStep(\n        () =>\n          config.runToolBatchStep({\n            context: config.input.context,\n            request: config.input.request,\n            calls: modelResult.value.toolCalls,\n            createdMessages: modelResult.value.createdMessages,\n            turn: modelResult.value.turn,\n            eventSequence: modelResult.value.eventSequence ?? state.eventSequence\n          }),\n        config.toolBatchStepRetry\n      )\n    )\n\n    if (toolsResult._tag === 'Failure') {\n      await writeErrorSafely(config.writeError, toolsResult.error)\n      return {\n        _tag: 'ToolBatchStepFailed',\n        turn: modelResult.value.turn,\n        error: toolsResult.error,\n        state\n      }\n    }\n\n    state = {\n      request: config.input.request,\n      messages: [...modelResult.value.messages, ...toolsResult.value.messages],\n      createdMessages: toolsResult.value.createdMessages,\n      usage: modelResult.value.usage,\n      turn: modelResult.value.turn + 1,\n      eventSequence: toolsResult.value.eventSequence ?? modelResult.value.eventSequence ?? state.eventSequence\n    }\n  }\n\n  const error = workflowMaxTurnsError(maxTurns)\n  await writeErrorSafely(config.writeError, error)\n\n  return {\n    _tag: 'MaxTurnsExceeded',\n    maxTurns,\n    error,\n    state\n  }\n}\n"],"mappings":";AAqHA,MAAa,0BAA0B;AACvC,MAAa,sBAA0D,EAAE,aAAa,EAAE;AAExF,MAAa,sBAAyB,YACpC,QAAQ,MACN,WAAU;CAAE,MAAM;CAAW;AAAM,KACnC,WAAU;CAAE,MAAM;CAAW;AAAM,EACrC;AAEF,MAAM,yBAAyB,6BAC7B,IAAI,MAAM,6CAA6C,UAAU;AAEnE,MAAM,oBAAoB,YAA+C,UACvE,WAAW,KAAK,EAAE,YAAY,KAAA,CAAS;AAEzC,MAAM,oBAAoB,WACxB,KAAK,IAAI,GAAG,KAAK,MAAM,QAAQ,eAAe,oBAAoB,WAAW,CAAC;AAEhF,eAAsB,kBACpB,SACA,QACY;CACZ,MAAM,cAAc,iBAAiB,MAAM;CAC3C,IAAI,UAAU;CAEd,SACE,IAAI;EACF,OAAO,MAAM,QAAQ;CACvB,SAAS,OAAO;EACd,IAAI,WAAW,aACb,MAAM;EAGR,WAAW;CACb;AAEJ;AAEA,eAAsB,uBACpB,QACuC;CACvC,IAAI,QAAmC;EACrC,SAAS,OAAO,MAAM;EACtB,iBAAiB,CAAC;EAClB,MAAM;EACN,eAAe;CACjB;CACA,MAAM,WAAW,OAAO,YAAA;CAExB,KAAK,IAAI,OAAO,GAAG,OAAO,UAAU,QAAQ;EAC1C,MAAM,cAAc,MAAM,mBACxB,wBACQ,OAAO,aAAa;GAAE,SAAS,OAAO,MAAM;GAAS;EAAM,CAAC,GAClE,OAAO,cACT,CACF;EAEA,IAAI,YAAY,SAAS,WAAW;GAClC,MAAM,iBAAiB,OAAO,YAAY,YAAY,KAAK;GAC3D,OAAO;IACL,MAAM;IACN,MAAM,MAAM;IACZ,OAAO,YAAY;IACnB;GACF;EACF;EAEA,IAAI,YAAY,MAAM,MAAM;GAC1B,MAAM,cAAc,MAAM,mBACxB,kBAAkB,OAAO,aAAa,OAAO,gBAAgB,CAC/D;GAEA,IAAI,YAAY,SAAS,WAAW;IAClC,MAAM,iBAAiB,OAAO,YAAY,YAAY,KAAK;IAE3D,OAAO;KACL,MAAM;KACN,OAAO,YAAY,MAAM;KACzB,OAAO,YAAY;KACnB;IACF;GACF;GAEA,OAAO;IACL,MAAM;IACN,OAAO,YAAY,MAAM;IACzB;GACF;EACF;EAEA,MAAM,cAAc,MAAM,mBACxB,wBAEI,OAAO,iBAAiB;GACtB,SAAS,OAAO,MAAM;GACtB,SAAS,OAAO,MAAM;GACtB,OAAO,YAAY,MAAM;GACzB,iBAAiB,YAAY,MAAM;GACnC,MAAM,YAAY,MAAM;GACxB,eAAe,YAAY,MAAM,iBAAiB,MAAM;EAC1D,CAAC,GACH,OAAO,kBACT,CACF;EAEA,IAAI,YAAY,SAAS,WAAW;GAClC,MAAM,iBAAiB,OAAO,YAAY,YAAY,KAAK;GAC3D,OAAO;IACL,MAAM;IACN,MAAM,YAAY,MAAM;IACxB,OAAO,YAAY;IACnB;GACF;EACF;EAEA,QAAQ;GACN,SAAS,OAAO,MAAM;GACtB,UAAU,CAAC,GAAG,YAAY,MAAM,UAAU,GAAG,YAAY,MAAM,QAAQ;GACvE,iBAAiB,YAAY,MAAM;GACnC,OAAO,YAAY,MAAM;GACzB,MAAM,YAAY,MAAM,OAAO;GAC/B,eAAe,YAAY,MAAM,iBAAiB,YAAY,MAAM,iBAAiB,MAAM;EAC7F;CACF;CAEA,MAAM,QAAQ,sBAAsB,QAAQ;CAC5C,MAAM,iBAAiB,OAAO,YAAY,KAAK;CAE/C,OAAO;EACL,MAAM;EACN;EACA;EACA;CACF;AACF"}
+{"version":3,"file":"workflow-loop.mjs","names":[],"sources":["../src/workflow-loop.ts"],"sourcesContent":["export type VercelAgentWorkflowInput = {\n  readonly request: unknown\n  readonly context: unknown\n}\n\nexport type SerializableWorkflowState = {\n  readonly request: unknown\n  readonly messages?: ReadonlyArray<unknown>\n  readonly createdMessages: ReadonlyArray<unknown>\n  readonly usage?: unknown\n  readonly turn: number\n  readonly eventSequence?: number\n}\n\nexport type VercelAgentWorkflowModelStepInput = {\n  readonly context: unknown\n  readonly state: SerializableWorkflowState\n}\n\nexport type VercelAgentWorkflowModelStepResult = {\n  readonly done: boolean\n  readonly messages: ReadonlyArray<unknown>\n  readonly createdMessages: ReadonlyArray<unknown>\n  readonly toolCalls: ReadonlyArray<unknown>\n  readonly usage: unknown\n  readonly turn: number\n  readonly eventSequence?: number\n}\n\nexport type VercelAgentWorkflowToolBatchStepInput = {\n  readonly context: unknown\n  readonly request: unknown\n  readonly calls: ReadonlyArray<unknown>\n  readonly createdMessages: ReadonlyArray<unknown>\n  readonly hitlResponses?: ReadonlyArray<unknown>\n  readonly usage?: unknown\n  readonly turn?: number\n  readonly eventSequence?: number\n}\n\nexport type VercelAgentWorkflowAwaitingInput = {\n  readonly hookToken: string\n  readonly requests: ReadonlyArray<unknown>\n  readonly messages: ReadonlyArray<unknown>\n  readonly usage: unknown\n  readonly turns: number\n  readonly eventSequence?: number\n}\n\nexport type VercelAgentWorkflowToolBatchStepResult = {\n  readonly messages: ReadonlyArray<unknown>\n  readonly createdMessages: ReadonlyArray<unknown>\n  readonly awaitingInput?: VercelAgentWorkflowAwaitingInput\n  readonly eventSequence?: number\n}\n\nexport type WorkflowStepResult<A> =\n  | {\n      readonly _tag: 'Success'\n      readonly value: A\n    }\n  | {\n      readonly _tag: 'Failure'\n      readonly error: unknown\n    }\n\nexport type VercelAgentWorkflowStepRetryPolicy = {\n  readonly maxAttempts: number\n}\n\nexport type VercelAgentWorkflowRunResult =\n  | {\n      readonly _tag: 'Completed'\n      readonly turns: number\n      readonly state: SerializableWorkflowState\n    }\n  | {\n      readonly _tag: 'ModelStepFailed'\n      readonly turn: number\n      readonly error: unknown\n      readonly state: SerializableWorkflowState\n    }\n  | {\n      readonly _tag: 'ToolBatchStepFailed'\n      readonly turn: number\n      readonly error: unknown\n      readonly state: SerializableWorkflowState\n    }\n  | {\n      readonly _tag: 'AwaitInputFailed'\n      readonly turn: number\n      readonly error: unknown\n      readonly awaitingInput: VercelAgentWorkflowAwaitingInput\n      readonly state: SerializableWorkflowState\n    }\n  | {\n      readonly _tag: 'CloseStreamFailed'\n      readonly turns: number\n      readonly error: unknown\n      readonly state: SerializableWorkflowState\n    }\n  | {\n      readonly _tag: 'MaxTurnsExceeded'\n      readonly maxTurns: number\n      readonly error: Error\n      readonly state: SerializableWorkflowState\n    }\n\nexport type VercelAgentWorkflowLoopConfig = {\n  readonly input: VercelAgentWorkflowInput\n  readonly maxTurns?: number\n  readonly runModelStep: (\n    input: VercelAgentWorkflowModelStepInput\n  ) => Promise<VercelAgentWorkflowModelStepResult>\n  readonly runToolBatchStep: (\n    input: VercelAgentWorkflowToolBatchStepInput\n  ) => Promise<VercelAgentWorkflowToolBatchStepResult>\n  readonly closeStream: () => Promise<void>\n  readonly writeError: (error: unknown) => Promise<void>\n  readonly awaitInput?: (input: VercelAgentWorkflowAwaitingInput) => Promise<unknown>\n  readonly modelStepRetry?: VercelAgentWorkflowStepRetryPolicy\n  readonly toolBatchStepRetry?: VercelAgentWorkflowStepRetryPolicy\n  readonly awaitInputRetry?: VercelAgentWorkflowStepRetryPolicy\n  readonly closeStreamRetry?: VercelAgentWorkflowStepRetryPolicy\n}\n\nexport const defaultMaxWorkflowTurns = 500\nexport const noWorkflowStepRetry: VercelAgentWorkflowStepRetryPolicy = { maxAttempts: 1 }\n\nexport const settleWorkflowStep = <A>(promise: Promise<A>): Promise<WorkflowStepResult<A>> =>\n  promise.then(\n    value => ({ _tag: 'Success', value }),\n    error => ({ _tag: 'Failure', error })\n  )\n\nconst workflowMaxTurnsError = (maxTurns: number) =>\n  new Error(`Vercel agent workflow exceeded max turns: ${maxTurns}`)\n\nconst writeErrorSafely = (writeError: (error: unknown) => Promise<void>, error: unknown) =>\n  writeError(error).catch(() => undefined)\n\nconst missingAwaitInputHandlerError = () =>\n  new Error('Vercel agent workflow awaiting input requested but no awaitInput handler is configured')\n\nconst maxRetryAttempts = (policy: VercelAgentWorkflowStepRetryPolicy | undefined) =>\n  Math.max(1, Math.floor(policy?.maxAttempts ?? noWorkflowStepRetry.maxAttempts))\n\nexport async function retryWorkflowStep<A>(\n  runStep: () => Promise<A>,\n  policy?: VercelAgentWorkflowStepRetryPolicy\n): Promise<A> {\n  const maxAttempts = maxRetryAttempts(policy)\n  let attempt = 1\n\n  for (;;) {\n    try {\n      return await runStep()\n    } catch (error) {\n      if (attempt >= maxAttempts) {\n        throw error\n      }\n\n      attempt += 1\n    }\n  }\n}\n\nexport async function runVercelAgentWorkflow(\n  config: VercelAgentWorkflowLoopConfig\n): Promise<VercelAgentWorkflowRunResult> {\n  const {\n    input,\n    maxTurns: configuredMaxTurns,\n    runModelStep,\n    runToolBatchStep,\n    closeStream,\n    writeError,\n    awaitInput,\n    modelStepRetry,\n    toolBatchStepRetry,\n    awaitInputRetry,\n    closeStreamRetry\n  } = config\n  let state: SerializableWorkflowState = {\n    request: input.request,\n    createdMessages: [],\n    turn: 1,\n    eventSequence: 0\n  }\n  const maxTurns = configuredMaxTurns ?? defaultMaxWorkflowTurns\n\n  for (let step = 0; step < maxTurns; step++) {\n    const modelResult = await settleWorkflowStep(\n      retryWorkflowStep(\n        () => runModelStep({ context: input.context, state }),\n        modelStepRetry\n      )\n    )\n\n    if (modelResult._tag === 'Failure') {\n      await writeErrorSafely(writeError, modelResult.error)\n      return {\n        _tag: 'ModelStepFailed',\n        turn: state.turn,\n        error: modelResult.error,\n        state\n      }\n    }\n\n    if (modelResult.value.done) {\n      const closeResult = await settleWorkflowStep(\n        retryWorkflowStep(closeStream, closeStreamRetry)\n      )\n\n      if (closeResult._tag === 'Failure') {\n        await writeErrorSafely(writeError, closeResult.error)\n\n        return {\n          _tag: 'CloseStreamFailed',\n          turns: modelResult.value.turn,\n          error: closeResult.error,\n          state\n        }\n      }\n\n      return {\n        _tag: 'Completed',\n        turns: modelResult.value.turn,\n        state\n      }\n    }\n\n    let completedToolsResult: VercelAgentWorkflowToolBatchStepResult | undefined\n    let toolHitlResponses: ReadonlyArray<unknown> = []\n    let toolEventSequence = modelResult.value.eventSequence ?? state.eventSequence\n\n    for (;;) {\n      const toolsResult = await settleWorkflowStep(\n        retryWorkflowStep(\n          () =>\n            runToolBatchStep({\n              context: input.context,\n              request: input.request,\n              calls: modelResult.value.toolCalls,\n              createdMessages: modelResult.value.createdMessages,\n              hitlResponses: toolHitlResponses,\n              usage: modelResult.value.usage,\n              turn: modelResult.value.turn,\n              eventSequence: toolEventSequence\n            }),\n          toolBatchStepRetry\n        )\n      )\n\n      if (toolsResult._tag === 'Failure') {\n        await writeErrorSafely(writeError, toolsResult.error)\n\n        return {\n          _tag: 'ToolBatchStepFailed',\n          turn: modelResult.value.turn,\n          error: toolsResult.error,\n          state\n        }\n      }\n\n      if (toolsResult.value.awaitingInput === undefined) {\n        completedToolsResult = toolsResult.value\n        break\n      }\n\n      const awaitingInput = toolsResult.value.awaitingInput\n      toolEventSequence = awaitingInput.eventSequence ?? toolsResult.value.eventSequence ?? toolEventSequence\n\n      const hitlResponse = await settleWorkflowStep(\n        retryWorkflowStep(\n          () => {\n            if (awaitInput === undefined) {\n              return Promise.reject(missingAwaitInputHandlerError())\n            }\n\n            return awaitInput(awaitingInput)\n          },\n          awaitInputRetry\n        )\n      )\n\n      if (hitlResponse._tag === 'Failure') {\n        await writeErrorSafely(writeError, hitlResponse.error)\n\n        return {\n          _tag: 'AwaitInputFailed',\n          turn: modelResult.value.turn,\n          error: hitlResponse.error,\n          awaitingInput,\n          state\n        }\n      }\n\n      toolHitlResponses = [...toolHitlResponses, hitlResponse.value]\n    }\n\n    if (completedToolsResult === undefined) {\n      const error = new Error('Vercel agent workflow tool batch did not complete')\n      await writeErrorSafely(writeError, error)\n\n      return {\n        _tag: 'ToolBatchStepFailed',\n        turn: modelResult.value.turn,\n        error,\n        state\n      }\n    }\n\n    state = {\n      request: input.request,\n      messages: [...modelResult.value.messages, ...completedToolsResult.messages],\n      createdMessages: completedToolsResult.createdMessages,\n      usage: modelResult.value.usage,\n      turn: modelResult.value.turn + 1,\n      eventSequence: completedToolsResult.eventSequence ?? toolEventSequence\n    }\n  }\n\n  const error = workflowMaxTurnsError(maxTurns)\n  await writeErrorSafely(writeError, error)\n\n  return {\n    _tag: 'MaxTurnsExceeded',\n    maxTurns,\n    error,\n    state\n  }\n}\n"],"mappings":";AA8HA,MAAa,0BAA0B;AACvC,MAAa,sBAA0D,EAAE,aAAa,EAAE;AAExF,MAAa,sBAAyB,YACpC,QAAQ,MACN,WAAU;CAAE,MAAM;CAAW;AAAM,KACnC,WAAU;CAAE,MAAM;CAAW;AAAM,EACrC;AAEF,MAAM,yBAAyB,6BAC7B,IAAI,MAAM,6CAA6C,UAAU;AAEnE,MAAM,oBAAoB,YAA+C,UACvE,WAAW,KAAK,EAAE,YAAY,KAAA,CAAS;AAEzC,MAAM,sDACJ,IAAI,MAAM,wFAAwF;AAEpG,MAAM,oBAAoB,WACxB,KAAK,IAAI,GAAG,KAAK,MAAM,QAAQ,eAAe,oBAAoB,WAAW,CAAC;AAEhF,eAAsB,kBACpB,SACA,QACY;CACZ,MAAM,cAAc,iBAAiB,MAAM;CAC3C,IAAI,UAAU;CAEd,SACE,IAAI;EACF,OAAO,MAAM,QAAQ;CACvB,SAAS,OAAO;EACd,IAAI,WAAW,aACb,MAAM;EAGR,WAAW;CACb;AAEJ;AAEA,eAAsB,uBACpB,QACuC;CACvC,MAAM,EACJ,OACA,UAAU,oBACV,cACA,kBACA,aACA,YACA,YACA,gBACA,oBACA,iBACA,qBACE;CACJ,IAAI,QAAmC;EACrC,SAAS,MAAM;EACf,iBAAiB,CAAC;EAClB,MAAM;EACN,eAAe;CACjB;CACA,MAAM,WAAW,sBAAA;CAEjB,KAAK,IAAI,OAAO,GAAG,OAAO,UAAU,QAAQ;EAC1C,MAAM,cAAc,MAAM,mBACxB,wBACQ,aAAa;GAAE,SAAS,MAAM;GAAS;EAAM,CAAC,GACpD,cACF,CACF;EAEA,IAAI,YAAY,SAAS,WAAW;GAClC,MAAM,iBAAiB,YAAY,YAAY,KAAK;GACpD,OAAO;IACL,MAAM;IACN,MAAM,MAAM;IACZ,OAAO,YAAY;IACnB;GACF;EACF;EAEA,IAAI,YAAY,MAAM,MAAM;GAC1B,MAAM,cAAc,MAAM,mBACxB,kBAAkB,aAAa,gBAAgB,CACjD;GAEA,IAAI,YAAY,SAAS,WAAW;IAClC,MAAM,iBAAiB,YAAY,YAAY,KAAK;IAEpD,OAAO;KACL,MAAM;KACN,OAAO,YAAY,MAAM;KACzB,OAAO,YAAY;KACnB;IACF;GACF;GAEA,OAAO;IACL,MAAM;IACN,OAAO,YAAY,MAAM;IACzB;GACF;EACF;EAEA,IAAI;EACJ,IAAI,oBAA4C,CAAC;EACjD,IAAI,oBAAoB,YAAY,MAAM,iBAAiB,MAAM;EAEjE,SAAS;GACP,MAAM,cAAc,MAAM,mBACxB,wBAEI,iBAAiB;IACf,SAAS,MAAM;IACf,SAAS,MAAM;IACf,OAAO,YAAY,MAAM;IACzB,iBAAiB,YAAY,MAAM;IACnC,eAAe;IACf,OAAO,YAAY,MAAM;IACzB,MAAM,YAAY,MAAM;IACxB,eAAe;GACjB,CAAC,GACH,kBACF,CACF;GAEA,IAAI,YAAY,SAAS,WAAW;IAClC,MAAM,iBAAiB,YAAY,YAAY,KAAK;IAEpD,OAAO;KACL,MAAM;KACN,MAAM,YAAY,MAAM;KACxB,OAAO,YAAY;KACnB;IACF;GACF;GAEA,IAAI,YAAY,MAAM,kBAAkB,KAAA,GAAW;IACjD,uBAAuB,YAAY;IACnC;GACF;GAEA,MAAM,gBAAgB,YAAY,MAAM;GACxC,oBAAoB,cAAc,iBAAiB,YAAY,MAAM,iBAAiB;GAEtF,MAAM,eAAe,MAAM,mBACzB,wBACQ;IACJ,IAAI,eAAe,KAAA,GACjB,OAAO,QAAQ,OAAO,8BAA8B,CAAC;IAGvD,OAAO,WAAW,aAAa;GACjC,GACA,eACF,CACF;GAEA,IAAI,aAAa,SAAS,WAAW;IACnC,MAAM,iBAAiB,YAAY,aAAa,KAAK;IAErD,OAAO;KACL,MAAM;KACN,MAAM,YAAY,MAAM;KACxB,OAAO,aAAa;KACpB;KACA;IACF;GACF;GAEA,oBAAoB,CAAC,GAAG,mBAAmB,aAAa,KAAK;EAC/D;EAEA,IAAI,yBAAyB,KAAA,GAAW;GACtC,MAAM,wBAAQ,IAAI,MAAM,mDAAmD;GAC3E,MAAM,iBAAiB,YAAY,KAAK;GAExC,OAAO;IACL,MAAM;IACN,MAAM,YAAY,MAAM;IACxB;IACA;GACF;EACF;EAEA,QAAQ;GACN,SAAS,MAAM;GACf,UAAU,CAAC,GAAG,YAAY,MAAM,UAAU,GAAG,qBAAqB,QAAQ;GAC1E,iBAAiB,qBAAqB;GACtC,OAAO,YAAY,MAAM;GACzB,MAAM,YAAY,MAAM,OAAO;GAC/B,eAAe,qBAAqB,iBAAiB;EACvD;CACF;CAEA,MAAM,QAAQ,sBAAsB,QAAQ;CAC5C,MAAM,iBAAiB,YAAY,KAAK;CAExC,OAAO;EACL,MAAM;EACN;EACA;EACA;CACF;AACF"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@yolk-sdk/vercel-workflows-runtime",
-  "version": "0.0.1-canary.1",
+  "version": "0.0.1-canary.10",
   "description": "Vercel Workflow orchestration primitives for Yolk agent loops.",
   "license": "MIT",
   "type": "module",

package/src/workflow-loop.ts

@@ -86,6 +86,13 @@ export type VercelAgentWorkflowRunResult =
       readonly error: unknown
       readonly state: SerializableWorkflowState
     }
+  | {
+      readonly _tag: 'AwaitInputFailed'
+      readonly turn: number
+      readonly error: unknown
+      readonly awaitingInput: VercelAgentWorkflowAwaitingInput
+      readonly state: SerializableWorkflowState
+    }
   | {
       readonly _tag: 'CloseStreamFailed'
       readonly turns: number
@@ -110,8 +117,10 @@ export type VercelAgentWorkflowLoopConfig = {
   ) => Promise<VercelAgentWorkflowToolBatchStepResult>
   readonly closeStream: () => Promise<void>
   readonly writeError: (error: unknown) => Promise<void>
+  readonly awaitInput?: (input: VercelAgentWorkflowAwaitingInput) => Promise<unknown>
   readonly modelStepRetry?: VercelAgentWorkflowStepRetryPolicy
   readonly toolBatchStepRetry?: VercelAgentWorkflowStepRetryPolicy
+  readonly awaitInputRetry?: VercelAgentWorkflowStepRetryPolicy
   readonly closeStreamRetry?: VercelAgentWorkflowStepRetryPolicy
 }
 
@@ -130,6 +139,9 @@ const workflowMaxTurnsError = (maxTurns: number) =>
 const writeErrorSafely = (writeError: (error: unknown) => Promise<void>, error: unknown) =>
   writeError(error).catch(() => undefined)
 
+const missingAwaitInputHandlerError = () =>
+  new Error('Vercel agent workflow awaiting input requested but no awaitInput handler is configured')
+
 const maxRetryAttempts = (policy: VercelAgentWorkflowStepRetryPolicy | undefined) =>
   Math.max(1, Math.floor(policy?.maxAttempts ?? noWorkflowStepRetry.maxAttempts))
 
@@ -156,24 +168,37 @@ export async function retryWorkflowStep<A>(
 export async function runVercelAgentWorkflow(
   config: VercelAgentWorkflowLoopConfig
 ): Promise<VercelAgentWorkflowRunResult> {
+  const {
+    input,
+    maxTurns: configuredMaxTurns,
+    runModelStep,
+    runToolBatchStep,
+    closeStream,
+    writeError,
+    awaitInput,
+    modelStepRetry,
+    toolBatchStepRetry,
+    awaitInputRetry,
+    closeStreamRetry
+  } = config
   let state: SerializableWorkflowState = {
-    request: config.input.request,
+    request: input.request,
     createdMessages: [],
     turn: 1,
     eventSequence: 0
   }
-  const maxTurns = config.maxTurns ?? defaultMaxWorkflowTurns
+  const maxTurns = configuredMaxTurns ?? defaultMaxWorkflowTurns
 
   for (let step = 0; step < maxTurns; step++) {
     const modelResult = await settleWorkflowStep(
       retryWorkflowStep(
-        () => config.runModelStep({ context: config.input.context, state }),
-        config.modelStepRetry
+        () => runModelStep({ context: input.context, state }),
+        modelStepRetry
       )
     )
 
     if (modelResult._tag === 'Failure') {
-      await writeErrorSafely(config.writeError, modelResult.error)
+      await writeErrorSafely(writeError, modelResult.error)
       return {
         _tag: 'ModelStepFailed',
         turn: state.turn,
@@ -184,11 +209,11 @@ export async function runVercelAgentWorkflow(
 
     if (modelResult.value.done) {
       const closeResult = await settleWorkflowStep(
-        retryWorkflowStep(config.closeStream, config.closeStreamRetry)
+        retryWorkflowStep(closeStream, closeStreamRetry)
       )
 
       if (closeResult._tag === 'Failure') {
-        await writeErrorSafely(config.writeError, closeResult.error)
+        await writeErrorSafely(writeError, closeResult.error)
 
         return {
           _tag: 'CloseStreamFailed',
@@ -205,43 +230,99 @@ export async function runVercelAgentWorkflow(
       }
     }
 
-    const toolsResult = await settleWorkflowStep(
-      retryWorkflowStep(
-        () =>
-          config.runToolBatchStep({
-            context: config.input.context,
-            request: config.input.request,
-            calls: modelResult.value.toolCalls,
-            createdMessages: modelResult.value.createdMessages,
-            turn: modelResult.value.turn,
-            eventSequence: modelResult.value.eventSequence ?? state.eventSequence
-          }),
-        config.toolBatchStepRetry
+    let completedToolsResult: VercelAgentWorkflowToolBatchStepResult | undefined
+    let toolHitlResponses: ReadonlyArray<unknown> = []
+    let toolEventSequence = modelResult.value.eventSequence ?? state.eventSequence
+
+    for (;;) {
+      const toolsResult = await settleWorkflowStep(
+        retryWorkflowStep(
+          () =>
+            runToolBatchStep({
+              context: input.context,
+              request: input.request,
+              calls: modelResult.value.toolCalls,
+              createdMessages: modelResult.value.createdMessages,
+              hitlResponses: toolHitlResponses,
+              usage: modelResult.value.usage,
+              turn: modelResult.value.turn,
+              eventSequence: toolEventSequence
+            }),
+          toolBatchStepRetry
+        )
       )
-    )
 
-    if (toolsResult._tag === 'Failure') {
-      await writeErrorSafely(config.writeError, toolsResult.error)
+      if (toolsResult._tag === 'Failure') {
+        await writeErrorSafely(writeError, toolsResult.error)
+
+        return {
+          _tag: 'ToolBatchStepFailed',
+          turn: modelResult.value.turn,
+          error: toolsResult.error,
+          state
+        }
+      }
+
+      if (toolsResult.value.awaitingInput === undefined) {
+        completedToolsResult = toolsResult.value
+        break
+      }
+
+      const awaitingInput = toolsResult.value.awaitingInput
+      toolEventSequence = awaitingInput.eventSequence ?? toolsResult.value.eventSequence ?? toolEventSequence
+
+      const hitlResponse = await settleWorkflowStep(
+        retryWorkflowStep(
+          () => {
+            if (awaitInput === undefined) {
+              return Promise.reject(missingAwaitInputHandlerError())
+            }
+
+            return awaitInput(awaitingInput)
+          },
+          awaitInputRetry
+        )
+      )
+
+      if (hitlResponse._tag === 'Failure') {
+        await writeErrorSafely(writeError, hitlResponse.error)
+
+        return {
+          _tag: 'AwaitInputFailed',
+          turn: modelResult.value.turn,
+          error: hitlResponse.error,
+          awaitingInput,
+          state
+        }
+      }
+
+      toolHitlResponses = [...toolHitlResponses, hitlResponse.value]
+    }
+
+    if (completedToolsResult === undefined) {
+      const error = new Error('Vercel agent workflow tool batch did not complete')
+      await writeErrorSafely(writeError, error)
+
       return {
         _tag: 'ToolBatchStepFailed',
         turn: modelResult.value.turn,
-        error: toolsResult.error,
+        error,
         state
       }
     }
 
     state = {
-      request: config.input.request,
-      messages: [...modelResult.value.messages, ...toolsResult.value.messages],
-      createdMessages: toolsResult.value.createdMessages,
+      request: input.request,
+      messages: [...modelResult.value.messages, ...completedToolsResult.messages],
+      createdMessages: completedToolsResult.createdMessages,
       usage: modelResult.value.usage,
       turn: modelResult.value.turn + 1,
-      eventSequence: toolsResult.value.eventSequence ?? modelResult.value.eventSequence ?? state.eventSequence
+      eventSequence: completedToolsResult.eventSequence ?? toolEventSequence
     }
   }
 
   const error = workflowMaxTurnsError(maxTurns)
-  await writeErrorSafely(config.writeError, error)
+  await writeErrorSafely(writeError, error)
 
   return {
     _tag: 'MaxTurnsExceeded',