@statelyai/agent 2.0.0-alpha.6 → 2.0.0-alpha.8

package/dist/ai-sdk.cjs

@@ -1,5 +1,6 @@
 Object.defineProperty(exports, Symbol.toStringTag, { value: "Module" });
-const require_decision = require("./decision-CX3YdwrO.cjs");
+const require_run_agent = require("./run-agent-BQ3vV7UI.cjs");
+const require_decision = require("./decision-b-lkcs4L.cjs");
 let ai = require("ai");
 //#region src/ai-sdk/index.ts
 /**
@@ -85,7 +86,7 @@ function resolveAiSdkModel(options, modelRef) {
 /**
 * AI SDK request-mapping settings shared by `generateText`/`streamText`.
 * `AgentTextRequest.messages` (`AgentMessage[]`) and AI SDK's `ModelMessage[]`
-* are structurally compatible by design (§1 of docs/p0-design.md) — the cast
+* are structurally compatible by design (§1 of .scratch/p0-design.md) — the cast
 * below is a typed identity mapping, not a semantic conversion.
 */
 function toAiSdkCallSettings(request) {
@@ -115,6 +116,56 @@ function isStructuredOutputRequest(request) {
 	return require_decision.getAgentOutputMode(request.outputSchema) === "structured";
 }
 /**
+* Extracts the first complete top-level JSON value from `text`, or returns
+* `undefined` when there is nothing to repair (no complete value, or the value
+* already spans the whole text). Models occasionally emit two structured-output
+* envelopes back to back (`{"result":{…}}{"result":{…}}`), which fails JSON
+* parsing wholesale; the balanced scan below recovers the first value and
+* drops the rest.
+*/
+function extractFirstJsonValue(text) {
+	const start = text.search(/[{[]/);
+	if (start === -1) return;
+	let depth = 0;
+	let inString = false;
+	let escaped = false;
+	for (let i = start; i < text.length; i++) {
+		const char = text[i];
+		if (inString) {
+			if (escaped) escaped = false;
+			else if (char === "\\") escaped = true;
+			else if (char === "\"") inString = false;
+			continue;
+		}
+		if (char === "\"") inString = true;
+		else if (char === "{" || char === "[") depth++;
+		else if (char === "}" || char === "]") {
+			depth--;
+			if (depth === 0) {
+				const value = text.slice(start, i + 1);
+				return value === text.trim() ? void 0 : value;
+			}
+		}
+	}
+}
+function withJsonRepair(output) {
+	return {
+		...output,
+		parseCompleteOutput: async (options, context) => {
+			try {
+				return await output.parseCompleteOutput(options, context);
+			} catch (error) {
+				const repaired = extractFirstJsonValue(options.text);
+				if (repaired === void 0) throw error;
+				return await output.parseCompleteOutput({
+					...options,
+					text: repaired
+				}, context);
+			}
+		}
+	};
+}
+/**
 * The canonical Vercel AI SDK adapter: builds the `{ generateText, streamText,
 * decide }` executor set consumed by `runAgent`/`executeAgentRequest`. `ai`
 * must not become a dependency of core `src/` files — this subpath is the one
@@ -138,10 +189,21 @@ function createAiSdkExecutors(options) {
 		};
 		if (isStructuredOutputRequest(request)) {
 			const envelope = require_decision.buildEnvelopeSchema(request.outputSchema, { reasoning: request.reasoning });
-			const result = await (0, ai.generateText)({
-				...common,
-				output: ai.Output.object({ schema: envelope })
-			});
+			const structuredOutput = withJsonRepair(ai.Output.object({ schema: envelope }));
+			const canRetry = !request.tools || Object.keys(request.tools).length === 0;
+			let result;
+			try {
+				result = await (0, ai.generateText)({
+					...common,
+					output: structuredOutput
+				});
+			} catch (error) {
+				if (!canRetry || !ai.NoObjectGeneratedError.isInstance(error) || info?.signal?.aborted) throw error;
+				result = await (0, ai.generateText)({
+					...common,
+					output: structuredOutput
+				});
+			}
 			const { result: output, reasoning } = result.output;
 			return {
 				output,
@@ -213,8 +275,36 @@ function createAiSdkExecutors(options) {
 		decide
 	};
 }
+/** AI SDK host for a machine authored with `setupAgent({ models })`. */
+function runAgent(machine, options) {
+	const models = require_decision.getRegisteredAgentModels(machine);
+	if (!models || Object.keys(models).length === 0) throw new Error("AI SDK runAgent: machine has no models. Pass `models` to setupAgent, or use core runAgent with explicit executors.");
+	return require_run_agent.runAgent(machine, {
+		...options,
+		executors: createAiSdkExecutors({ models })
+	});
+}
+function createAgent(config) {
+	const { model, models: configuredModels, schemas, ...machineConfig } = config;
+	const models = configuredModels ?? { default: model };
+	const executors = createAiSdkExecutors({ models });
+	const machine = require_run_agent.setupAgent({
+		...schemas,
+		models
+	}).createMachine(machineConfig);
+	return {
+		machine,
+		run(runInput, options = {}) {
+			return require_run_agent.runAgent(machine, {
+				...options,
+				input: runInput,
+				executors
+			});
+		}
+	};
+}
 /** One AI SDK `tool()` per candidate event — the "tool-per-event +
-* toolChoice: 'required'" recipe from docs/p0-design.md §2.6. */
+* toolChoice: 'required'" recipe from .scratch/p0-design.md §2.6. */
 function toAiSdkEventTools(events) {
 	return Object.fromEntries(events.map((event) => [event.toolName, (0, ai.tool)({
 		description: `Choose the '${event.type}' move.`,
@@ -239,9 +329,12 @@ function toDecisionMessages(request) {
 	return messages;
 }
 //#endregion
+exports.createAgent = createAgent;
 exports.createAiSdkExecutors = createAiSdkExecutors;
 exports.defineModels = defineModels;
+exports.extractFirstJsonValue = extractFirstJsonValue;
 exports.isStructuredOutputRequest = isStructuredOutputRequest;
+exports.runAgent = runAgent;
 exports.toAiSdkCallSettings = toAiSdkCallSettings;
 exports.toAiSdkEventTools = toAiSdkEventTools;
 exports.toAiSdkToolChoice = toAiSdkToolChoice;

package/dist/ai-sdk.d.cts

@@ -1,5 +1,7 @@
-import { c as AgentTools, f as ChosenEvent } from "./types-BHjeDdch.cjs";
-import { B as AgentEventDescriptor, D as AgentDecisionExecutor, i as AgentRequestExecutor, k as AgentDecisionRequest, l as AgentTextRequest, s as AgentRequestExecutors } from "./text-logic-1ZQkO3zr.cjs";
+import { C as StandardSchemaV1, m as ChosenEvent, n as AgentEventSchemaInputMap, u as AgentTools } from "./types-C9QiMjre.cjs";
+import { H as AgentEventDescriptor, i as AgentRequestExecutor, j as AgentDecisionRequest, k as AgentDecisionExecutor, l as AgentTextRequest, s as AgentRequestExecutors } from "./text-logic-CZjyACzQ.cjs";
+import { E as AgentMachineConfig, T as AgentMachine, c as RunAgentResult, s as RunAgentOptions } from "./run-agent-BzW4emV_.cjs";
+import * as _$xstate from "xstate";
 import { FinishReason, LanguageModel, LanguageModelUsage, ModelMessage, Tool, ToolSet, TypedToolCall, TypedToolResult } from "ai";
 
 //#region src/ai-sdk/index.d.ts
@@ -54,7 +56,7 @@ type CreateAiSdkExecutorsOptions<TModels extends AiSdkModelMap = AiSdkModelMap>
 /**
  * AI SDK request-mapping settings shared by `generateText`/`streamText`.
  * `AgentTextRequest.messages` (`AgentMessage[]`) and AI SDK's `ModelMessage[]`
- * are structurally compatible by design (§1 of docs/p0-design.md) — the cast
+ * are structurally compatible by design (§1 of .scratch/p0-design.md) — the cast
  * below is a typed identity mapping, not a semantic conversion.
  */
 declare function toAiSdkCallSettings(request: AgentTextRequest & {
@@ -99,6 +101,15 @@ declare function toAiSdkToolChoice(toolChoice: AgentTextRequest["toolChoice"]):
 } | undefined;
 /** `true` when the request should use AI SDK structured `Output.object`. */
 declare function isStructuredOutputRequest(request: Pick<AgentTextRequest, "outputSchema">): boolean;
+/**
+ * Extracts the first complete top-level JSON value from `text`, or returns
+ * `undefined` when there is nothing to repair (no complete value, or the value
+ * already spans the whole text). Models occasionally emit two structured-output
+ * envelopes back to back (`{"result":{…}}{"result":{…}}`), which fails JSON
+ * parsing wholesale; the balanced scan below recovers the first value and
+ * drops the rest.
+ */
+declare function extractFirstJsonValue(text: string): string | undefined;
 /**
  * Raw result shape from {@link AiSdkExecutors.generateText} — the `{ output }`
  * envelope (the validated structured object for structured-output requests,
@@ -153,8 +164,35 @@ interface AiSdkExecutors extends AgentRequestExecutors<AiSdkGenerateResult, AiSd
  * ```
  */
 declare function createAiSdkExecutors<TModels extends AiSdkModelMap>(options: CreateAiSdkExecutorsOptions<TModels>): AiSdkExecutors;
+/** AI SDK host for a machine authored with `setupAgent({ models })`. */
+declare function runAgent<TMachine extends _$xstate.AnyStateMachine>(machine: TMachine, options: Omit<RunAgentOptions<TMachine>, "executors">): Promise<RunAgentResult<TMachine>>;
+type CreateAgentMachineConfig<TContextSchema extends StandardSchemaV1<Record<string, unknown>>, TInputSchema extends StandardSchemaV1, TEventSchemas extends AgentEventSchemaInputMap, TOutputSchema extends StandardSchemaV1, TModels extends AiSdkModelMap> = Omit<AgentMachineConfig<TContextSchema, TInputSchema, TEventSchemas, TOutputSchema, TModels>, "schemas">;
+type CreateAgentBaseConfig<TContextSchema extends StandardSchemaV1<Record<string, unknown>>, TInputSchema extends StandardSchemaV1, TEventSchemas extends AgentEventSchemaInputMap, TOutputSchema extends StandardSchemaV1, TModels extends AiSdkModelMap> = CreateAgentMachineConfig<TContextSchema, TInputSchema, TEventSchemas, TOutputSchema, TModels> & {
+  schemas: {
+    context: TContextSchema;
+    input: TInputSchema;
+    events?: TEventSchemas;
+    output?: TOutputSchema;
+  };
+};
+interface CreatedAgent<TMachine extends _$xstate.AnyStateMachine> {
+  machine: TMachine;
+  run(input: _$xstate.InputFrom<TMachine>, options?: Omit<RunAgentOptions<TMachine>, "input" | "executors">): Promise<RunAgentResult<TMachine>>;
+}
+/**
+ * One-call AI SDK entry point for the common case: creates a typed machine and
+ * a `run(input)` method with model executors already wired.
+ */
+declare function createAgent<TContextSchema extends StandardSchemaV1<Record<string, unknown>>, TInputSchema extends StandardSchemaV1, TEventSchemas extends AgentEventSchemaInputMap = {}, TOutputSchema extends StandardSchemaV1 = StandardSchemaV1>(config: CreateAgentBaseConfig<TContextSchema, TInputSchema, TEventSchemas, TOutputSchema, AiSdkModelMap<"default">> & {
+  model: LanguageModel;
+  models?: never;
+}): CreatedAgent<AgentMachine<TContextSchema, TInputSchema, TEventSchemas, TOutputSchema, AiSdkModelMap<"default">>>;
+declare function createAgent<TModels extends AiSdkModelMap, TContextSchema extends StandardSchemaV1<Record<string, unknown>>, TInputSchema extends StandardSchemaV1, TEventSchemas extends AgentEventSchemaInputMap = {}, TOutputSchema extends StandardSchemaV1 = StandardSchemaV1>(config: CreateAgentBaseConfig<TContextSchema, TInputSchema, TEventSchemas, TOutputSchema, TModels> & {
+  models: TModels;
+  model?: never;
+}): CreatedAgent<AgentMachine<TContextSchema, TInputSchema, TEventSchemas, TOutputSchema, TModels>>;
 /** One AI SDK `tool()` per candidate event — the "tool-per-event +
- * toolChoice: 'required'" recipe from docs/p0-design.md §2.6. */
+ * toolChoice: 'required'" recipe from .scratch/p0-design.md §2.6. */
 declare function toAiSdkEventTools(events: AgentEventDescriptor[]): {
   [k: string]: Tool<unknown, never>;
 };
@@ -165,4 +203,4 @@ declare function toAiSdkEventTools(events: AgentEventDescriptor[]): {
  */
 declare function toDecisionMessages(request: Pick<AgentDecisionRequest, "messages" | "prompt" | "events" | "attempts">): ModelMessage[] | undefined;
 //#endregion
-export { AiSdkDecideResult, AiSdkExecutors, AiSdkGenerateResult, AiSdkModelMap, AiSdkStreamResult, CreateAiSdkExecutorsOptions, createAiSdkExecutors, defineModels, isStructuredOutputRequest, toAiSdkCallSettings, toAiSdkEventTools, toAiSdkToolChoice, toAiSdkTools, toDecisionMessages };
+export { AiSdkDecideResult, AiSdkExecutors, AiSdkGenerateResult, AiSdkModelMap, AiSdkStreamResult, CreateAiSdkExecutorsOptions, CreatedAgent, createAgent, createAiSdkExecutors, defineModels, extractFirstJsonValue, isStructuredOutputRequest, runAgent, toAiSdkCallSettings, toAiSdkEventTools, toAiSdkToolChoice, toAiSdkTools, toDecisionMessages };

package/dist/ai-sdk.d.mts

@@ -1,5 +1,7 @@
-import { c as AgentTools, f as ChosenEvent } from "./types-Cq1YlAQ6.mjs";
-import { B as AgentEventDescriptor, D as AgentDecisionExecutor, i as AgentRequestExecutor, k as AgentDecisionRequest, l as AgentTextRequest, s as AgentRequestExecutors } from "./text-logic-2EMJIS-n.mjs";
+import { C as StandardSchemaV1, m as ChosenEvent, n as AgentEventSchemaInputMap, u as AgentTools } from "./types-qm00QF91.mjs";
+import { H as AgentEventDescriptor, i as AgentRequestExecutor, j as AgentDecisionRequest, k as AgentDecisionExecutor, l as AgentTextRequest, s as AgentRequestExecutors } from "./text-logic-C7WJpCIc.mjs";
+import { E as AgentMachineConfig, T as AgentMachine, c as RunAgentResult, s as RunAgentOptions } from "./run-agent-XYjmBxHi.mjs";
+import * as _$xstate from "xstate";
 import { FinishReason, LanguageModel, LanguageModelUsage, ModelMessage, Tool, ToolSet, TypedToolCall, TypedToolResult } from "ai";
 
 //#region src/ai-sdk/index.d.ts
@@ -54,7 +56,7 @@ type CreateAiSdkExecutorsOptions<TModels extends AiSdkModelMap = AiSdkModelMap>
 /**
  * AI SDK request-mapping settings shared by `generateText`/`streamText`.
  * `AgentTextRequest.messages` (`AgentMessage[]`) and AI SDK's `ModelMessage[]`
- * are structurally compatible by design (§1 of docs/p0-design.md) — the cast
+ * are structurally compatible by design (§1 of .scratch/p0-design.md) — the cast
  * below is a typed identity mapping, not a semantic conversion.
  */
 declare function toAiSdkCallSettings(request: AgentTextRequest & {
@@ -99,6 +101,15 @@ declare function toAiSdkToolChoice(toolChoice: AgentTextRequest["toolChoice"]):
 } | undefined;
 /** `true` when the request should use AI SDK structured `Output.object`. */
 declare function isStructuredOutputRequest(request: Pick<AgentTextRequest, "outputSchema">): boolean;
+/**
+ * Extracts the first complete top-level JSON value from `text`, or returns
+ * `undefined` when there is nothing to repair (no complete value, or the value
+ * already spans the whole text). Models occasionally emit two structured-output
+ * envelopes back to back (`{"result":{…}}{"result":{…}}`), which fails JSON
+ * parsing wholesale; the balanced scan below recovers the first value and
+ * drops the rest.
+ */
+declare function extractFirstJsonValue(text: string): string | undefined;
 /**
  * Raw result shape from {@link AiSdkExecutors.generateText} — the `{ output }`
  * envelope (the validated structured object for structured-output requests,
@@ -153,8 +164,35 @@ interface AiSdkExecutors extends AgentRequestExecutors<AiSdkGenerateResult, AiSd
  * ```
  */
 declare function createAiSdkExecutors<TModels extends AiSdkModelMap>(options: CreateAiSdkExecutorsOptions<TModels>): AiSdkExecutors;
+/** AI SDK host for a machine authored with `setupAgent({ models })`. */
+declare function runAgent<TMachine extends _$xstate.AnyStateMachine>(machine: TMachine, options: Omit<RunAgentOptions<TMachine>, "executors">): Promise<RunAgentResult<TMachine>>;
+type CreateAgentMachineConfig<TContextSchema extends StandardSchemaV1<Record<string, unknown>>, TInputSchema extends StandardSchemaV1, TEventSchemas extends AgentEventSchemaInputMap, TOutputSchema extends StandardSchemaV1, TModels extends AiSdkModelMap> = Omit<AgentMachineConfig<TContextSchema, TInputSchema, TEventSchemas, TOutputSchema, TModels>, "schemas">;
+type CreateAgentBaseConfig<TContextSchema extends StandardSchemaV1<Record<string, unknown>>, TInputSchema extends StandardSchemaV1, TEventSchemas extends AgentEventSchemaInputMap, TOutputSchema extends StandardSchemaV1, TModels extends AiSdkModelMap> = CreateAgentMachineConfig<TContextSchema, TInputSchema, TEventSchemas, TOutputSchema, TModels> & {
+  schemas: {
+    context: TContextSchema;
+    input: TInputSchema;
+    events?: TEventSchemas;
+    output?: TOutputSchema;
+  };
+};
+interface CreatedAgent<TMachine extends _$xstate.AnyStateMachine> {
+  machine: TMachine;
+  run(input: _$xstate.InputFrom<TMachine>, options?: Omit<RunAgentOptions<TMachine>, "input" | "executors">): Promise<RunAgentResult<TMachine>>;
+}
+/**
+ * One-call AI SDK entry point for the common case: creates a typed machine and
+ * a `run(input)` method with model executors already wired.
+ */
+declare function createAgent<TContextSchema extends StandardSchemaV1<Record<string, unknown>>, TInputSchema extends StandardSchemaV1, TEventSchemas extends AgentEventSchemaInputMap = {}, TOutputSchema extends StandardSchemaV1 = StandardSchemaV1>(config: CreateAgentBaseConfig<TContextSchema, TInputSchema, TEventSchemas, TOutputSchema, AiSdkModelMap<"default">> & {
+  model: LanguageModel;
+  models?: never;
+}): CreatedAgent<AgentMachine<TContextSchema, TInputSchema, TEventSchemas, TOutputSchema, AiSdkModelMap<"default">>>;
+declare function createAgent<TModels extends AiSdkModelMap, TContextSchema extends StandardSchemaV1<Record<string, unknown>>, TInputSchema extends StandardSchemaV1, TEventSchemas extends AgentEventSchemaInputMap = {}, TOutputSchema extends StandardSchemaV1 = StandardSchemaV1>(config: CreateAgentBaseConfig<TContextSchema, TInputSchema, TEventSchemas, TOutputSchema, TModels> & {
+  models: TModels;
+  model?: never;
+}): CreatedAgent<AgentMachine<TContextSchema, TInputSchema, TEventSchemas, TOutputSchema, TModels>>;
 /** One AI SDK `tool()` per candidate event — the "tool-per-event +
- * toolChoice: 'required'" recipe from docs/p0-design.md §2.6. */
+ * toolChoice: 'required'" recipe from .scratch/p0-design.md §2.6. */
 declare function toAiSdkEventTools(events: AgentEventDescriptor[]): {
   [k: string]: Tool<unknown, never>;
 };
@@ -165,4 +203,4 @@ declare function toAiSdkEventTools(events: AgentEventDescriptor[]): {
  */
 declare function toDecisionMessages(request: Pick<AgentDecisionRequest, "messages" | "prompt" | "events" | "attempts">): ModelMessage[] | undefined;
 //#endregion
-export { AiSdkDecideResult, AiSdkExecutors, AiSdkGenerateResult, AiSdkModelMap, AiSdkStreamResult, CreateAiSdkExecutorsOptions, createAiSdkExecutors, defineModels, isStructuredOutputRequest, toAiSdkCallSettings, toAiSdkEventTools, toAiSdkToolChoice, toAiSdkTools, toDecisionMessages };
+export { AiSdkDecideResult, AiSdkExecutors, AiSdkGenerateResult, AiSdkModelMap, AiSdkStreamResult, CreateAiSdkExecutorsOptions, CreatedAgent, createAgent, createAiSdkExecutors, defineModels, extractFirstJsonValue, isStructuredOutputRequest, runAgent, toAiSdkCallSettings, toAiSdkEventTools, toAiSdkToolChoice, toAiSdkTools, toDecisionMessages };

package/dist/ai-sdk.mjs

@@ -1,5 +1,6 @@
-import { G as isStandardSchema, T as getAgentOutputMode, l as renderDecisionAttempts, x as buildEnvelopeSchema } from "./decision-D1654JdD.mjs";
-import { Output, generateText, stepCountIs, streamText, tool } from "ai";
+import { a as runAgent$1, c as setupAgent } from "./run-agent-7OaHM7SB.mjs";
+import { J as isStandardSchema, L as getRegisteredAgentModels, T as getAgentOutputMode, l as renderDecisionAttempts, x as buildEnvelopeSchema } from "./decision-CQdrKc8k.mjs";
+import { NoObjectGeneratedError, Output, generateText, stepCountIs, streamText, tool } from "ai";
 //#region src/ai-sdk/index.ts
 /**
 * Maps an {@link AgentTools} map onto AI SDK `tool()` definitions. A tool that
@@ -84,7 +85,7 @@ function resolveAiSdkModel(options, modelRef) {
 /**
 * AI SDK request-mapping settings shared by `generateText`/`streamText`.
 * `AgentTextRequest.messages` (`AgentMessage[]`) and AI SDK's `ModelMessage[]`
-* are structurally compatible by design (§1 of docs/p0-design.md) — the cast
+* are structurally compatible by design (§1 of .scratch/p0-design.md) — the cast
 * below is a typed identity mapping, not a semantic conversion.
 */
 function toAiSdkCallSettings(request) {
@@ -114,6 +115,56 @@ function isStructuredOutputRequest(request) {
 	return getAgentOutputMode(request.outputSchema) === "structured";
 }
 /**
+* Extracts the first complete top-level JSON value from `text`, or returns
+* `undefined` when there is nothing to repair (no complete value, or the value
+* already spans the whole text). Models occasionally emit two structured-output
+* envelopes back to back (`{"result":{…}}{"result":{…}}`), which fails JSON
+* parsing wholesale; the balanced scan below recovers the first value and
+* drops the rest.
+*/
+function extractFirstJsonValue(text) {
+	const start = text.search(/[{[]/);
+	if (start === -1) return;
+	let depth = 0;
+	let inString = false;
+	let escaped = false;
+	for (let i = start; i < text.length; i++) {
+		const char = text[i];
+		if (inString) {
+			if (escaped) escaped = false;
+			else if (char === "\\") escaped = true;
+			else if (char === "\"") inString = false;
+			continue;
+		}
+		if (char === "\"") inString = true;
+		else if (char === "{" || char === "[") depth++;
+		else if (char === "}" || char === "]") {
+			depth--;
+			if (depth === 0) {
+				const value = text.slice(start, i + 1);
+				return value === text.trim() ? void 0 : value;
+			}
+		}
+	}
+}
+function withJsonRepair(output) {
+	return {
+		...output,
+		parseCompleteOutput: async (options, context) => {
+			try {
+				return await output.parseCompleteOutput(options, context);
+			} catch (error) {
+				const repaired = extractFirstJsonValue(options.text);
+				if (repaired === void 0) throw error;
+				return await output.parseCompleteOutput({
+					...options,
+					text: repaired
+				}, context);
+			}
+		}
+	};
+}
+/**
 * The canonical Vercel AI SDK adapter: builds the `{ generateText, streamText,
 * decide }` executor set consumed by `runAgent`/`executeAgentRequest`. `ai`
 * must not become a dependency of core `src/` files — this subpath is the one
@@ -137,10 +188,21 @@ function createAiSdkExecutors(options) {
 		};
 		if (isStructuredOutputRequest(request)) {
 			const envelope = buildEnvelopeSchema(request.outputSchema, { reasoning: request.reasoning });
-			const result = await generateText({
-				...common,
-				output: Output.object({ schema: envelope })
-			});
+			const structuredOutput = withJsonRepair(Output.object({ schema: envelope }));
+			const canRetry = !request.tools || Object.keys(request.tools).length === 0;
+			let result;
+			try {
+				result = await generateText({
+					...common,
+					output: structuredOutput
+				});
+			} catch (error) {
+				if (!canRetry || !NoObjectGeneratedError.isInstance(error) || info?.signal?.aborted) throw error;
+				result = await generateText({
+					...common,
+					output: structuredOutput
+				});
+			}
 			const { result: output, reasoning } = result.output;
 			return {
 				output,
@@ -212,8 +274,36 @@ function createAiSdkExecutors(options) {
 		decide
 	};
 }
+/** AI SDK host for a machine authored with `setupAgent({ models })`. */
+function runAgent(machine, options) {
+	const models = getRegisteredAgentModels(machine);
+	if (!models || Object.keys(models).length === 0) throw new Error("AI SDK runAgent: machine has no models. Pass `models` to setupAgent, or use core runAgent with explicit executors.");
+	return runAgent$1(machine, {
+		...options,
+		executors: createAiSdkExecutors({ models })
+	});
+}
+function createAgent(config) {
+	const { model, models: configuredModels, schemas, ...machineConfig } = config;
+	const models = configuredModels ?? { default: model };
+	const executors = createAiSdkExecutors({ models });
+	const machine = setupAgent({
+		...schemas,
+		models
+	}).createMachine(machineConfig);
+	return {
+		machine,
+		run(runInput, options = {}) {
+			return runAgent$1(machine, {
+				...options,
+				input: runInput,
+				executors
+			});
+		}
+	};
+}
 /** One AI SDK `tool()` per candidate event — the "tool-per-event +
-* toolChoice: 'required'" recipe from docs/p0-design.md §2.6. */
+* toolChoice: 'required'" recipe from .scratch/p0-design.md §2.6. */
 function toAiSdkEventTools(events) {
 	return Object.fromEntries(events.map((event) => [event.toolName, tool({
 		description: `Choose the '${event.type}' move.`,
@@ -238,4 +328,4 @@ function toDecisionMessages(request) {
 	return messages;
 }
 //#endregion
-export { createAiSdkExecutors, defineModels, isStructuredOutputRequest, toAiSdkCallSettings, toAiSdkEventTools, toAiSdkToolChoice, toAiSdkTools, toDecisionMessages };
+export { createAgent, createAiSdkExecutors, defineModels, extractFirstJsonValue, isStructuredOutputRequest, runAgent, toAiSdkCallSettings, toAiSdkEventTools, toAiSdkToolChoice, toAiSdkTools, toDecisionMessages };

package/dist/cli.cjs

@@ -1,5 +1,6 @@
 #!/usr/bin/env node
-const require_src = require("./src-wBfi-kTA.cjs");
+const require_run_agent = require("./run-agent-BQ3vV7UI.cjs");
+const require_src = require("./src-CFtSqm-c.cjs");
 let node_fs = require("node:fs");
 //#region src/cli.ts
 /**
@@ -44,7 +45,7 @@ function main(argv) {
 	}
 	let machine;
 	try {
-		machine = require_src.setupAgent.fromConfig(config, { compileSchema: stubCompileSchema });
+		machine = require_run_agent.setupAgent.fromConfig(config, { compileSchema: stubCompileSchema });
 	} catch (error) {
 		process.stderr.write(`statelyai-agent: '${file}' is not a valid agent-machine config: ${error instanceof Error ? error.message : String(error)}\n`);
 		return 2;

package/dist/cli.mjs

@@ -1,5 +1,6 @@
 #!/usr/bin/env node
-import { r as lintAgentMachine, v as setupAgent } from "./src-CEa947Dm.mjs";
+import { c as setupAgent } from "./run-agent-7OaHM7SB.mjs";
+import { r as lintAgentMachine } from "./src-d-jhiOgP.mjs";
 import { readFileSync } from "node:fs";
 //#region src/cli.ts
 /**

package/dist/{decision-D1654JdD.mjs → decision-CQdrKc8k.mjs}

@@ -221,6 +221,9 @@ function validateSchemaSync(schema, value) {
 //#endregion
 //#region src/internal/registry.ts
 const agentExecutionOptions = /* @__PURE__ */ new WeakMap();
+function getRegisteredAgentModels(machine) {
+	return agentExecutionOptions.get(machine)?.models;
+}
 /**
 * Machine-carried wait-state predicates, keyed on the machine's root `config`
 * object. `config` is shared by reference across `machine.provide(...)` (unlike
@@ -266,6 +269,27 @@ const DECIDE_ACTOR = "agent.decide";
 const PLAN_ACTOR = "agent.plan";
 /** Synthetic `src` stamped on trace requests produced by `getRequests` state interpretation — NOT a registered actor source (nothing is invoked; the pass makes the call directly). */
 const INTERPRET_SOURCE = "agent.interpret";
+/**
+* Splits a portable `"provider/model-id"` model ref (the convention JSON
+* workflows and registry-less hosts use, e.g. `"openai/gpt-5.4-mini"`) into
+* its parts. A ref with no `/` has no provider — `modelId` is the whole ref.
+* The standard building block for a host's `resolveModel`:
+*
+* @example
+* ```ts
+* const resolveModel = (ref: string) => openai(parseModelRef(ref).modelId);
+* ```
+*/
+function parseModelRef(modelRef) {
+	const slash = modelRef.indexOf("/");
+	return slash === -1 ? {
+		provider: void 0,
+		modelId: modelRef
+	} : {
+		provider: modelRef.slice(0, slash),
+		modelId: modelRef.slice(slash + 1)
+	};
+}
 const agentTextInputSchema = { "~standard": {
 	version: 1,
 	vendor: "statelyai-agent",
@@ -336,7 +360,7 @@ const builtinTextActors = {
 	[GENERATE_TEXT_ACTOR]: createBuiltinTextActor(GENERATE_TEXT_ACTOR, "generate", unknownOutputSchema),
 	[STREAM_TEXT_ACTOR]: createBuiltinTextActor(STREAM_TEXT_ACTOR, "stream", stringOutputSchema)
 };
-/** The unbound `agent.userInput` builtin registered by setupAgent (an unbound-placeholder logic — see internal/registry.ts). @internal */
+/** The unbound `agent.userInput` builtin registered by setupAgent (an unbound-placeholder logic — see internal/registry.ts). Output is `string` — what the human typed. @internal */
 const userInputActor = createAsyncLogic({ run: async () => {
 	throw new Error(`'${USER_INPUT_ACTOR}' has no host execution. Provide an implementation with machine.provide({ actorSources: { '${USER_INPUT_ACTOR}': ... } }).`);
 } });
@@ -444,12 +468,15 @@ function createTextLogic(config, execute) {
 * });
 * ```
 */
-function bindRequestExecutor(logic, executor) {
+function bindRequestExecutor(logic, executor, info) {
 	return logic.withExecutor(async ({ request, signal }) => {
 		const { output } = await executor({
 			...request,
 			tools: request.tools ?? {}
-		}, { signal });
+		}, {
+			signal,
+			onChunk: info?.onChunk
+		});
 		return { output };
 	});
 }
@@ -524,6 +551,17 @@ function buildEnvelopeSchema(inner, options = {}) {
 		} }
 	} };
 }
+/**
+* Validates a raw provider value against the structured-output envelope for
+* `request` and returns the unwrapped `{ result, reasoning? }` — the checked
+* replacement for `raw as StructuredOutputEnvelope` in hand-written hosts.
+* Pair with {@link buildEnvelopeSchema} (which produced the schema the
+* provider was asked to satisfy).
+*/
+function parseStructuredEnvelope(request, value) {
+	if (!request.outputSchema) throw new Error("parseStructuredEnvelope: the request declares no outputSchema.");
+	return validateSchemaSync(buildEnvelopeSchema(request.outputSchema, { reasoning: request.reasoning }), value);
+}
 function getStandardSchemaJson(schema) {
 	const jsonSchema = (schema?.["~standard"])?.jsonSchema?.input?.();
 	return jsonSchema && !(jsonSchema instanceof Promise) ? jsonSchema : void 0;
@@ -937,4 +975,4 @@ async function resolveDecision(request, executor, options = {}) {
 	throw new DecisionExhaustedError(attempts);
 }
 //#endregion
-export { userInputActor as A, getAgentMessages as B, createTextLogic as C, isTextLogic as D, isStructuredOutputSchema as E, isUnboundPlaceholder as F, isStandardSchema as G, getJsonSchemaSync as H, machineSuspensionPredicates as I, toolMessage as J, persistSnapshot as K, missingActor as L, executorBoundLogics as M, getMachineSuspensionPredicate as N, normalizeGeneratorResult as O, getRegisteredAgentExecutionOptions as P, assistantMessage as R, builtinTextActors as S, getAgentOutputMode as T, getMachineStructuralHash as U, getJsonSchema as V, getStateMeta as W, validateSchemaSync as X, userMessage as Y, INTERPRET_SOURCE as _, createPlanActor as a, bindRequestExecutor as b, isPlanLogic as c, EVENT_TOOL_PREFIX as d, getAcceptedEvents as f, DECIDE_ACTOR as g, sanitizeEventToolName as h, createDecideActor as i, agentExecutionOptions as j, parseOutput as k, renderDecisionAttempts as l, parseAgentEvent as m, PLAN_DONE_EVENT_TYPE as n, initialPlanLedger as o, matchesEventPattern as p, systemMessage as q, advancePlanLedger as r, isDecisionLogic as s, DecisionExhaustedError as t, resolveDecision as u, PLAN_ACTOR as v, executeAgentTextRequest as w, buildEnvelopeSchema as x, USER_INPUT_ACTOR as y, findNonSerializableContextPaths as z };
+export { validateSchemaSync as $, parseOutput as A, missingActor as B, createTextLogic as C, isTextLogic as D, isStructuredOutputSchema as E, getMachineSuspensionPredicate as F, getJsonSchemaSync as G, findNonSerializableContextPaths as H, getRegisteredAgentExecutionOptions as I, isStandardSchema as J, getMachineStructuralHash as K, getRegisteredAgentModels as L, userInputActor as M, agentExecutionOptions as N, normalizeGeneratorResult as O, executorBoundLogics as P, userMessage as Q, isUnboundPlaceholder as R, builtinTextActors as S, getAgentOutputMode as T, getAgentMessages as U, assistantMessage as V, getJsonSchema as W, systemMessage as X, persistSnapshot as Y, toolMessage as Z, INTERPRET_SOURCE as _, createPlanActor as a, bindRequestExecutor as b, isPlanLogic as c, EVENT_TOOL_PREFIX as d, getAcceptedEvents as f, DECIDE_ACTOR as g, sanitizeEventToolName as h, createDecideActor as i, parseStructuredEnvelope as j, parseModelRef as k, renderDecisionAttempts as l, parseAgentEvent as m, PLAN_DONE_EVENT_TYPE as n, initialPlanLedger as o, matchesEventPattern as p, getStateMeta as q, advancePlanLedger as r, isDecisionLogic as s, DecisionExhaustedError as t, resolveDecision as u, PLAN_ACTOR as v, executeAgentTextRequest as w, buildEnvelopeSchema as x, USER_INPUT_ACTOR as y, machineSuspensionPredicates as z };