@assistant-ui/react-ai-sdk 1.3.32 → 1.3.33

package/dist/assistant-stream/dist/core/tool/schema-utils.js.map

@@ -1 +1 @@
-{"version":3,"file":"schema-utils.js","names":[],"sources":["../../../../../../assistant-stream/dist/core/tool/schema-utils.js"],"sourcesContent":["//#region src/core/tool/schema-utils.ts\nfunction isStandardSchema(schema) {\n\treturn typeof schema === \"object\" && schema !== null && \"~standard\" in schema && typeof schema[\"~standard\"] === \"object\";\n}\nfunction hasToJSONSchemaMethod(schema) {\n\treturn typeof schema === \"object\" && schema !== null && \"toJSONSchema\" in schema && typeof schema.toJSONSchema === \"function\";\n}\nfunction hasToJSONMethod(schema) {\n\treturn typeof schema === \"object\" && schema !== null && \"toJSON\" in schema && typeof schema.toJSON === \"function\";\n}\n/**\n* Converts a schema to JSONSchema7.\n* Supports:\n* - StandardSchemaV1 with ~standard.toJSONSchema (e.g., Zod v4)\n* - StandardSchemaV1 with ~standard.jsonSchema.input() (e.g., Zod v4)\n* - Objects with toJSONSchema() method (e.g., Zod v4)\n* - Objects with toJSON() method\n* - Plain JSONSchema7 objects (must have a \"type\" property)\n*/\nfunction toJSONSchema(schema) {\n\tif (isStandardSchema(schema)) {\n\t\tconst toJSONSchemaMethod = schema[\"~standard\"].toJSONSchema;\n\t\tif (typeof toJSONSchemaMethod === \"function\") return toJSONSchemaMethod();\n\t\tconst jsonSchema = schema[\"~standard\"].jsonSchema;\n\t\tif (typeof jsonSchema === \"object\" && jsonSchema !== null && typeof jsonSchema.input === \"function\") return jsonSchema.input();\n\t}\n\tif (hasToJSONSchemaMethod(schema)) return schema.toJSONSchema();\n\tif (hasToJSONMethod(schema)) return schema.toJSON();\n\tif (isStandardSchema(schema)) throw new Error(\"Could not convert schema to JSON Schema. The schema implements Standard Schema but does not support JSON Schema conversion. If you are using Zod, please upgrade to Zod v4 (npm install zod@latest). Alternatively, pass a plain JSON Schema object instead.\");\n\treturn schema;\n}\n/**\n* Returns a copy of the JSON Schema with `required` removed recursively,\n* making every property optional. Array item schemas are left unchanged.\n*/\nfunction toPartialJSONSchema(schema) {\n\tconst { required: _, ...result } = schema;\n\tif (result.properties) result.properties = Object.fromEntries(Object.entries(result.properties).map(([key, prop]) => {\n\t\tif (typeof prop === \"object\" && prop !== null && !Array.isArray(prop)) {\n\t\t\tconst p = prop;\n\t\t\treturn [key, p.properties != null ? toPartialJSONSchema(p) : prop];\n\t\t}\n\t\treturn [key, prop];\n\t}));\n\treturn result;\n}\nfunction defaultToolFilter(_name, tool) {\n\treturn !tool.disabled && tool.type !== \"backend\" && (tool.type !== \"frontend\" || tool.execute !== void 0);\n}\nfunction toolHasUploadableParameters(tool) {\n\treturn tool.parameters !== void 0 && !tool.unstable_backendDefault?.parameters;\n}\n/**\n* Converts a record of tools to a record of tool definitions with JSON Schema parameters.\n* By default, filters out disabled tools and backend tools.\n*\n* Entries are emitted in alphabetical order so the resulting request body is\n* byte-identical regardless of the order in which tools were registered. This\n* keeps provider prompt caches stable across renders that mount tools in\n* different orders.\n*/\nfunction toToolsJSONSchema(tools, options = {}) {\n\tif (!tools) return {};\n\tconst filter = options.filter ?? defaultToolFilter;\n\treturn Object.fromEntries(Object.entries(tools).filter(([name, tool]) => filter(name, tool)).filter((entry) => toolHasUploadableParameters(entry[1])).sort(([a], [b]) => a < b ? -1 : a > b ? 1 : 0).map(([name, tool]) => [name, {\n\t\t...tool.description && { description: tool.description },\n\t\tparameters: toJSONSchema(tool.parameters),\n\t\t...tool.providerOptions && { providerOptions: tool.providerOptions }\n\t}]));\n}\n//#endregion\nexport { toJSONSchema, toPartialJSONSchema, toToolsJSONSchema };\n\n//# sourceMappingURL=schema-utils.js.map"],"mappings":";AACA,SAAS,iBAAiB,QAAQ;CACjC,OAAO,OAAO,WAAW,YAAY,WAAW,QAAQ,eAAe,UAAU,OAAO,OAAO,iBAAiB;AACjH;AACA,SAAS,sBAAsB,QAAQ;CACtC,OAAO,OAAO,WAAW,YAAY,WAAW,QAAQ,kBAAkB,UAAU,OAAO,OAAO,iBAAiB;AACpH;AACA,SAAS,gBAAgB,QAAQ;CAChC,OAAO,OAAO,WAAW,YAAY,WAAW,QAAQ,YAAY,UAAU,OAAO,OAAO,WAAW;AACxG;;;;;;;;;;AAUA,SAAS,aAAa,QAAQ;CAC7B,IAAI,iBAAiB,MAAM,GAAG;EAC7B,MAAM,qBAAqB,OAAO,aAAa;EAC/C,IAAI,OAAO,uBAAuB,YAAY,OAAO,mBAAmB;EACxE,MAAM,aAAa,OAAO,aAAa;EACvC,IAAI,OAAO,eAAe,YAAY,eAAe,QAAQ,OAAO,WAAW,UAAU,YAAY,OAAO,WAAW,MAAM;CAC9H;CACA,IAAI,sBAAsB,MAAM,GAAG,OAAO,OAAO,aAAa;CAC9D,IAAI,gBAAgB,MAAM,GAAG,OAAO,OAAO,OAAO;CAClD,IAAI,iBAAiB,MAAM,GAAG,MAAM,IAAI,MAAM,8PAA8P;CAC5S,OAAO;AACR;AAgBA,SAAS,kBAAkB,OAAO,MAAM;CACvC,OAAO,CAAC,KAAK,YAAY,KAAK,SAAS,cAAc,KAAK,SAAS,cAAc,KAAK,YAAY,KAAK;AACxG;AACA,SAAS,4BAA4B,MAAM;CAC1C,OAAO,KAAK,eAAe,KAAK,KAAK,CAAC,KAAK,yBAAyB;AACrE;;;;;;;;;;AAUA,SAAS,kBAAkB,OAAO,UAAU,CAAC,GAAG;CAC/C,IAAI,CAAC,OAAO,OAAO,CAAC;CACpB,MAAM,SAAS,QAAQ,UAAU;CACjC,OAAO,OAAO,YAAY,OAAO,QAAQ,KAAK,EAAE,QAAQ,CAAC,MAAM,UAAU,OAAO,MAAM,IAAI,CAAC,EAAE,QAAQ,UAAU,4BAA4B,MAAM,EAAE,CAAC,EAAE,MAAM,CAAC,IAAI,CAAC,OAAO,IAAI,IAAI,KAAK,IAAI,IAAI,IAAI,CAAC,EAAE,KAAK,CAAC,MAAM,UAAU,CAAC,MAAM;EACjO,GAAG,KAAK,eAAe,EAAE,aAAa,KAAK,YAAY;EACvD,YAAY,aAAa,KAAK,UAAU;EACxC,GAAG,KAAK,mBAAmB,EAAE,iBAAiB,KAAK,gBAAgB;CACpE,CAAC,CAAC,CAAC;AACJ"}
+{"version":3,"file":"schema-utils.js","names":[],"sources":["../../../../../../assistant-stream/dist/core/tool/schema-utils.js"],"sourcesContent":["//#region src/core/tool/schema-utils.ts\nfunction isStandardSchema(schema) {\n\treturn typeof schema === \"object\" && schema !== null && \"~standard\" in schema && typeof schema[\"~standard\"] === \"object\";\n}\nfunction hasToJSONSchemaMethod(schema) {\n\treturn typeof schema === \"object\" && schema !== null && \"toJSONSchema\" in schema && typeof schema.toJSONSchema === \"function\";\n}\nfunction hasToJSONMethod(schema) {\n\treturn typeof schema === \"object\" && schema !== null && \"toJSON\" in schema && typeof schema.toJSON === \"function\";\n}\n/**\n* Converts a schema to JSONSchema7.\n* Supports:\n* - StandardSchemaV1 with ~standard.toJSONSchema (e.g., Zod v4)\n* - StandardSchemaV1 with ~standard.jsonSchema.input() (e.g., Zod v4)\n* - Objects with toJSONSchema() method (e.g., Zod v4)\n* - Objects with toJSON() method\n* - Plain JSONSchema7 objects (must have a \"type\" property)\n*/\nfunction toJSONSchema(schema) {\n\tif (isStandardSchema(schema)) {\n\t\tconst toJSONSchemaMethod = schema[\"~standard\"].toJSONSchema;\n\t\tif (typeof toJSONSchemaMethod === \"function\") return toJSONSchemaMethod();\n\t\tconst jsonSchema = schema[\"~standard\"].jsonSchema;\n\t\tif (typeof jsonSchema === \"object\" && jsonSchema !== null && typeof jsonSchema.input === \"function\") return jsonSchema.input();\n\t}\n\tif (hasToJSONSchemaMethod(schema)) return schema.toJSONSchema();\n\tif (hasToJSONMethod(schema)) return schema.toJSON();\n\tif (isStandardSchema(schema)) throw new Error(\"Could not convert schema to JSON Schema. The schema implements Standard Schema but does not support JSON Schema conversion. If you are using Zod, please upgrade to Zod v4 (npm install zod@latest). Alternatively, pass a plain JSON Schema object instead.\");\n\treturn schema;\n}\n/**\n* Returns a copy of the JSON Schema with `required` removed recursively,\n* making every property optional. Array item schemas are left unchanged.\n*/\nfunction toPartialJSONSchema(schema) {\n\tconst { required: _, ...result } = schema;\n\tif (result.properties) result.properties = Object.fromEntries(Object.entries(result.properties).map(([key, prop]) => {\n\t\tif (typeof prop === \"object\" && prop !== null && !Array.isArray(prop)) {\n\t\t\tconst p = prop;\n\t\t\treturn [key, p.properties != null ? toPartialJSONSchema(p) : prop];\n\t\t}\n\t\treturn [key, prop];\n\t}));\n\treturn result;\n}\nfunction defaultToolFilter(_name, tool) {\n\treturn !tool.disabled && tool.type !== \"backend\" && (tool.type !== \"frontend\" || tool.execute !== void 0);\n}\nfunction toolHasUploadableParameters(tool) {\n\treturn tool.parameters !== void 0 && !tool.unstable_backendDefault?.parameters;\n}\n/**\n* Converts a record of tools to a record of tool definitions with JSON Schema parameters.\n* By default, filters out disabled tools and backend tools.\n*\n* Entries are emitted in alphabetical order so the resulting request body is\n* byte-identical regardless of the order in which tools were registered. This\n* keeps provider prompt caches stable across renders that mount tools in\n* different orders.\n*/\nfunction toToolsJSONSchema(tools, options = {}) {\n\tif (!tools) return {};\n\tconst filter = options.filter ?? defaultToolFilter;\n\treturn Object.fromEntries(Object.entries(tools).filter(([name, tool]) => filter(name, tool)).filter((entry) => toolHasUploadableParameters(entry[1])).sort(([a], [b]) => a < b ? -1 : a > b ? 1 : 0).map(([name, tool]) => [name, {\n\t\t...tool.description && { description: tool.description },\n\t\tparameters: toJSONSchema(tool.parameters),\n\t\t...tool.providerOptions && { providerOptions: tool.providerOptions }\n\t}]));\n}\n//#endregion\nexport { toJSONSchema, toPartialJSONSchema, toToolsJSONSchema };\n\n//# sourceMappingURL=schema-utils.js.map"],"mappings":";AACA,SAAS,iBAAiB,QAAQ;CACjC,OAAO,OAAO,WAAW,YAAY,WAAW,QAAQ,eAAe,UAAU,OAAO,OAAO,iBAAiB;AACjH;AACA,SAAS,sBAAsB,QAAQ;CACtC,OAAO,OAAO,WAAW,YAAY,WAAW,QAAQ,kBAAkB,UAAU,OAAO,OAAO,iBAAiB;AACpH;AACA,SAAS,gBAAgB,QAAQ;CAChC,OAAO,OAAO,WAAW,YAAY,WAAW,QAAQ,YAAY,UAAU,OAAO,OAAO,WAAW;AACxG;;;;;;;;;;AAUA,SAAS,aAAa,QAAQ;CAC7B,IAAI,iBAAiB,MAAM,GAAG;EAC7B,MAAM,qBAAqB,OAAO,YAAY,CAAC;EAC/C,IAAI,OAAO,uBAAuB,YAAY,OAAO,mBAAmB;EACxE,MAAM,aAAa,OAAO,YAAY,CAAC;EACvC,IAAI,OAAO,eAAe,YAAY,eAAe,QAAQ,OAAO,WAAW,UAAU,YAAY,OAAO,WAAW,MAAM;CAC9H;CACA,IAAI,sBAAsB,MAAM,GAAG,OAAO,OAAO,aAAa;CAC9D,IAAI,gBAAgB,MAAM,GAAG,OAAO,OAAO,OAAO;CAClD,IAAI,iBAAiB,MAAM,GAAG,MAAM,IAAI,MAAM,8PAA8P;CAC5S,OAAO;AACR;AAgBA,SAAS,kBAAkB,OAAO,MAAM;CACvC,OAAO,CAAC,KAAK,YAAY,KAAK,SAAS,cAAc,KAAK,SAAS,cAAc,KAAK,YAAY,KAAK;AACxG;AACA,SAAS,4BAA4B,MAAM;CAC1C,OAAO,KAAK,eAAe,KAAK,KAAK,CAAC,KAAK,yBAAyB;AACrE;;;;;;;;;;AAUA,SAAS,kBAAkB,OAAO,UAAU,CAAC,GAAG;CAC/C,IAAI,CAAC,OAAO,OAAO,CAAC;CACpB,MAAM,SAAS,QAAQ,UAAU;CACjC,OAAO,OAAO,YAAY,OAAO,QAAQ,KAAK,CAAC,CAAC,QAAQ,CAAC,MAAM,UAAU,OAAO,MAAM,IAAI,CAAC,CAAC,CAAC,QAAQ,UAAU,4BAA4B,MAAM,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,IAAI,IAAI,KAAK,IAAI,IAAI,IAAI,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,UAAU,CAAC,MAAM;EACjO,GAAG,KAAK,eAAe,EAAE,aAAa,KAAK,YAAY;EACvD,YAAY,aAAa,KAAK,UAAU;EACxC,GAAG,KAAK,mBAAmB,EAAE,iBAAiB,KAAK,gBAAgB;CACpE,CAAC,CAAC,CAAC;AACJ"}

package/dist/client.d.ts

@@ -0,0 +1,10 @@
+/// <reference types="@assistant-ui/core/react" />
+import { useAISDKRuntime } from "./ui/use-chat/useAISDKRuntime.js";
+import { UseChatRuntimeOptions, useChatRuntime } from "./ui/use-chat/useChatRuntime.js";
+import { RESUMABLE_STREAM_ID_HEADER } from "./assistant-stream/dist/resumable/createResumableAssistantStreamResponse.js";
+import { AssistantChatResumableOptions, ResumableClientStorage, createResumableSessionStorage } from "./ui/resumable.js";
+import { AssistantChatTransport } from "./ui/use-chat/AssistantChatTransport.js";
+import { frontendTools } from "./frontendTools.js";
+import { injectQuoteContext } from "./injectQuoteContext.js";
+import { ThreadTokenUsage, TokenUsageExtractableMessage, getThreadMessageTokenUsage, useThreadTokenUsage } from "./usage.js";
+export { type AssistantChatResumableOptions, AssistantChatTransport, RESUMABLE_STREAM_ID_HEADER, type ResumableClientStorage, type ThreadTokenUsage, type TokenUsageExtractableMessage, type UseChatRuntimeOptions, createResumableSessionStorage, frontendTools, getThreadMessageTokenUsage, injectQuoteContext, useAISDKRuntime, useChatRuntime, useThreadTokenUsage };

package/dist/client.js

@@ -0,0 +1,9 @@
+import { useAISDKRuntime } from "./ui/use-chat/useAISDKRuntime.js";
+import { RESUMABLE_STREAM_ID_HEADER } from "./assistant-stream/dist/resumable/createResumableAssistantStreamResponse.js";
+import { createResumableSessionStorage } from "./ui/resumable.js";
+import { AssistantChatTransport } from "./ui/use-chat/AssistantChatTransport.js";
+import { useChatRuntime } from "./ui/use-chat/useChatRuntime.js";
+import { frontendTools } from "./frontendTools.js";
+import { injectQuoteContext } from "./injectQuoteContext.js";
+import { getThreadMessageTokenUsage, useThreadTokenUsage } from "./usage.js";
+export { AssistantChatTransport, RESUMABLE_STREAM_ID_HEADER, createResumableSessionStorage, frontendTools, getThreadMessageTokenUsage, injectQuoteContext, useAISDKRuntime, useChatRuntime, useThreadTokenUsage };

package/dist/frontendTools.js.map

@@ -1 +1 @@
-{"version":3,"file":"frontendTools.js","names":[],"sources":["../src/frontendTools.ts"],"sourcesContent":["import { jsonSchema, type ToolSet } from \"ai\";\nimport type { ToolJSONSchema } from \"assistant-stream\";\nimport { unwrapModelContentEnvelope } from \"./modelContentEnvelope\";\nimport { toAISDKContent, toAISDKDefaultOutput } from \"./toolOutputConversion\";\n\nexport const defaultToModelOutput = ({ output }: { output: unknown }) => {\n  const { result, modelContent } = unwrapModelContentEnvelope(output);\n  if (modelContent !== undefined) {\n    return toAISDKContent(modelContent);\n  }\n  return toAISDKDefaultOutput(result);\n};\n\nexport const frontendTools = (tools: Record<string, ToolJSONSchema>): ToolSet =>\n  Object.fromEntries(\n    Object.entries(tools).map(([name, t]) => [\n      name,\n      {\n        ...(t.description !== undefined && { description: t.description }),\n        inputSchema: jsonSchema(t.parameters),\n        toModelOutput: defaultToModelOutput,\n        ...(t.providerOptions && { providerOptions: t.providerOptions }),\n      },\n    ]),\n  ) as ToolSet;\n"],"mappings":";;;;AAKA,MAAa,wBAAwB,EAAE,aAAkC;CACvE,MAAM,EAAE,QAAQ,iBAAiB,2BAA2B,MAAM;CAClE,IAAI,iBAAiB,KAAA,GACnB,OAAO,eAAe,YAAY;CAEpC,OAAO,qBAAqB,MAAM;AACpC;AAEA,MAAa,iBAAiB,UAC5B,OAAO,YACL,OAAO,QAAQ,KAAK,EAAE,KAAK,CAAC,MAAM,OAAO,CACvC,MACA;CACE,GAAI,EAAE,gBAAgB,KAAA,KAAa,EAAE,aAAa,EAAE,YAAY;CAChE,aAAa,WAAW,EAAE,UAAU;CACpC,eAAe;CACf,GAAI,EAAE,mBAAmB,EAAE,iBAAiB,EAAE,gBAAgB;AAChE,CACF,CAAC,CACH"}
+{"version":3,"file":"frontendTools.js","names":[],"sources":["../src/frontendTools.ts"],"sourcesContent":["import { jsonSchema, type ToolSet } from \"ai\";\nimport type { ToolJSONSchema } from \"assistant-stream\";\nimport { unwrapModelContentEnvelope } from \"./modelContentEnvelope\";\nimport { toAISDKContent, toAISDKDefaultOutput } from \"./toolOutputConversion\";\n\nexport const defaultToModelOutput = ({ output }: { output: unknown }) => {\n  const { result, modelContent } = unwrapModelContentEnvelope(output);\n  if (modelContent !== undefined) {\n    return toAISDKContent(modelContent);\n  }\n  return toAISDKDefaultOutput(result);\n};\n\nexport const frontendTools = (tools: Record<string, ToolJSONSchema>): ToolSet =>\n  Object.fromEntries(\n    Object.entries(tools).map(([name, t]) => [\n      name,\n      {\n        ...(t.description !== undefined && { description: t.description }),\n        inputSchema: jsonSchema(t.parameters),\n        toModelOutput: defaultToModelOutput,\n        ...(t.providerOptions && { providerOptions: t.providerOptions }),\n      },\n    ]),\n  ) as ToolSet;\n"],"mappings":";;;;AAKA,MAAa,wBAAwB,EAAE,aAAkC;CACvE,MAAM,EAAE,QAAQ,iBAAiB,2BAA2B,MAAM;CAClE,IAAI,iBAAiB,KAAA,GACnB,OAAO,eAAe,YAAY;CAEpC,OAAO,qBAAqB,MAAM;AACpC;AAEA,MAAa,iBAAiB,UAC5B,OAAO,YACL,OAAO,QAAQ,KAAK,CAAC,CAAC,KAAK,CAAC,MAAM,OAAO,CACvC,MACA;CACE,GAAI,EAAE,gBAAgB,KAAA,KAAa,EAAE,aAAa,EAAE,YAAY;CAChE,aAAa,WAAW,EAAE,UAAU;CACpC,eAAe;CACf,GAAI,EAAE,mBAAmB,EAAE,iBAAiB,EAAE,gBAAgB;AAChE,CACF,CAAC,CACH"}

package/dist/generativeTools.d.ts

@@ -3,6 +3,11 @@ import { ToolSet } from "ai";
 import { Toolkit } from "@assistant-ui/core/react";
 
 //#region src/generativeTools.d.ts
+/**
+ * @deprecated Options for the deprecated {@link generativeTools}. Use
+ * {@link AISDKToolkit} with {@link AISDKToolkitOptions} /
+ * {@link AISDKToolkitToolsOptions} instead.
+ */
 interface GenerativeToolsOptions {
   /**
    * The server build of a generative toolkit (schema + server `execute`). Typed
@@ -35,16 +40,25 @@ type AISDKToolkitToolsOptions = {
  * resolves to the server build — schema + `execute`, with `render` stripped) and
  * pass it here. Tools without an `execute` are still exposed to the model but
  * left for the client to fulfill. `frontendTools` lets the client contribute
- * tools that aren't in the static toolkit. Use {@link AISDKToolkit} when the
- * toolkit contains MCP entries.
+ * tools that aren't in the static toolkit.
+ *
+ * @deprecated Use {@link AISDKToolkit} instead:
+ * `new AISDKToolkit({ toolkit }).tools({ frontend })`. It is a strict superset
+ * (it also opens MCP server connections), so it replaces `generativeTools`
+ * everywhere. The `frontendTools` option is named `frontend` on `.tools()`, and
+ * `.tools()` is async. `generativeTools` will be removed in a future version.
  *
  * @example
  * ```ts
+ * // Define once at module scope so any MCP connections pool across requests.
+ * const aiToolkit = new AISDKToolkit({ toolkit: docsToolkit });
+ *
+ * // In your route handler:
  * const { tools } = await req.json();
  * streamText({
  *   model,
  *   messages,
- *   tools: generativeTools({ toolkit: docsToolkit, frontendTools: tools }),
+ *   tools: await aiToolkit.tools({ frontend: tools }),
  * });
  * ```
  */

package/dist/generativeTools.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"generativeTools.d.ts","names":[],"sources":["../src/generativeTools.ts"],"mappings":";;;;;UAiCiB,sBAAA;;;AAAjB;;;EAME,OAAA,EAAS,OAAA;EAMsB;;;;;EAA/B,aAAA,GAAgB,MAAA,SAAe,cAAA;AAAA;AAAA,KAGrB,mBAAA;EACV,OAAA,EAAS,OAAO;AAAA;AAAA,KAGN,wBAAA;EAJA;;;EAQV,QAAA,GAAW,MAAM,SAAS,cAAA;AAAA;AAJ5B;;;;;;;;AAI0C;AAyB1C;;;;;;;;AAUC;AAED;;;;AAzCA,cA6Ba,eAAA,GAAmB,OAAA,EAAS,sBAAA,KAAyB,OAUjE;AAAA,cAEY,YAAA;EAAA;cAIC,OAAA,EAAS,mBAAA;EAIf,KAAA,CAAM,OAAA,GAAS,wBAAA,GAAgC,OAAA,CAAQ,OAAA;EASvD,KAAA,IAAS,OAAA;AAAA"}
+{"version":3,"file":"generativeTools.d.ts","names":[],"sources":["../src/generativeTools.ts"],"mappings":";;;;;;;;AAsCA;;UAAiB,sBAAA;EAMN;;;;;EAAT,OAAA,EAAS,OAAA;EAAA;;;;;EAMT,aAAA,GAAgB,MAAA,SAAe,cAAA;AAAA;AAAA,KAGrB,mBAAA;EACV,OAAA,EAAS,OAAO;AAAA;AAAA,KAGN,wBAAA;EAAA;;;EAIV,QAAA,GAAW,MAAM,SAAS,cAAA;AAAA;;;;AAAc;AAkC1C;;;;;;;;AAUC;AAED;;;;;;;;;;;;;;;;;;cAZa,eAAA,GAAmB,OAAA,EAAS,sBAAA,KAAyB,OAUjE;AAAA,cAEY,YAAA;EAAA;cAIC,OAAA,EAAS,mBAAA;EAIf,KAAA,CAAM,OAAA,GAAS,wBAAA,GAAgC,OAAA,CAAQ,OAAA;EASvD,KAAA,IAAS,OAAA;AAAA"}

package/dist/generativeTools.js

@@ -1,10 +1,10 @@
 import { unwrapModelContentEnvelope } from "./modelContentEnvelope.js";
+import { toJSONSchema } from "./assistant-stream/dist/core/tool/schema-utils.js";
 import { toAISDKContent, toAISDKDefaultOutput } from "./toolOutputConversion.js";
 import { frontendTools } from "./frontendTools.js";
-import { toJSONSchema } from "./assistant-stream/dist/core/tool/schema-utils.js";
 import { jsonSchema } from "ai";
 import { createMCPClient } from "@ai-sdk/mcp";
-import { Experimental_StdioMCPTransport } from "@ai-sdk/mcp/mcp-stdio";
+import { Experimental_StdioMCPTransport } from "#mcp-stdio";
 //#region src/generativeTools.ts
 const EMPTY_SCHEMA = {
 	type: "object",
@@ -24,16 +24,25 @@ const parametersToInputSchema = (parameters) => jsonSchema(parameters ? toJSONSc
 * resolves to the server build — schema + `execute`, with `render` stripped) and
 * pass it here. Tools without an `execute` are still exposed to the model but
 * left for the client to fulfill. `frontendTools` lets the client contribute
-* tools that aren't in the static toolkit. Use {@link AISDKToolkit} when the
-* toolkit contains MCP entries.
+* tools that aren't in the static toolkit.
+*
+* @deprecated Use {@link AISDKToolkit} instead:
+* `new AISDKToolkit({ toolkit }).tools({ frontend })`. It is a strict superset
+* (it also opens MCP server connections), so it replaces `generativeTools`
+* everywhere. The `frontendTools` option is named `frontend` on `.tools()`, and
+* `.tools()` is async. `generativeTools` will be removed in a future version.
 *
 * @example
 * ```ts
+* // Define once at module scope so any MCP connections pool across requests.
+* const aiToolkit = new AISDKToolkit({ toolkit: docsToolkit });
+*
+* // In your route handler:
 * const { tools } = await req.json();
 * streamText({
 *   model,
 *   messages,
-*   tools: generativeTools({ toolkit: docsToolkit, frontendTools: tools }),
+*   tools: await aiToolkit.tools({ frontend: tools }),
 * });
 * ```
 */

package/dist/generativeTools.js.map

@@ -1 +1 @@
-{"version":3,"file":"generativeTools.js","names":["#toolkit","#mcpClients","#mcpTools","#mcpClient"],"sources":["../src/generativeTools.ts"],"sourcesContent":["import { jsonSchema, type ToolSet } from \"ai\";\nimport type { MCPClient, MCPClientConfig } from \"@ai-sdk/mcp\";\nimport { createMCPClient } from \"@ai-sdk/mcp\";\nimport { Experimental_StdioMCPTransport } from \"@ai-sdk/mcp/mcp-stdio\";\nimport {\n  toJSONSchema,\n  type Tool,\n  type McpServerConfig,\n  type ToolJSONSchema,\n  type ToolModelOutputFunction,\n} from \"assistant-stream\";\nimport type { Toolkit, ToolkitDefinition } from \"@assistant-ui/core/react\";\nimport { frontendTools } from \"./frontendTools\";\nimport { toAISDKContent, toAISDKDefaultOutput } from \"./toolOutputConversion\";\nimport {\n  unwrapModelContentEnvelope,\n  type ModelContentEnvelope,\n} from \"./modelContentEnvelope\";\n\nconst EMPTY_SCHEMA = { type: \"object\" as const, properties: {} };\n\nconst humanNotSupported = (): never => {\n  throw new Error(\n    \"`human()` is not available during server-side tool execution.\",\n  );\n};\n\n// AI SDK leaves `abortSignal` optional; assistant-ui's execute requires one.\nconst neverAbort = new AbortController().signal;\n\nconst parametersToInputSchema = (parameters: Tool[\"parameters\"] | undefined) =>\n  jsonSchema(parameters ? toJSONSchema(parameters) : EMPTY_SCHEMA);\n\nexport interface GenerativeToolsOptions {\n  /**\n   * The server build of a generative toolkit (schema + server `execute`). Typed\n   * as the canonical {@link Toolkit} so callers don't need to cast; the server\n   * build carries `execute`, recovered internally as {@link ToolkitDefinition}.\n   */\n  toolkit: Toolkit;\n  /**\n   * Tools uploaded by the frontend (the request body's `tools`). Merged in\n   * alongside the `toolkit`; a server `execute` from `toolkit` takes precedence\n   * over an uploaded entry of the same name.\n   */\n  frontendTools?: Record<string, ToolJSONSchema>;\n}\n\nexport type AISDKToolkitOptions = {\n  toolkit: Toolkit;\n};\n\nexport type AISDKToolkitToolsOptions = {\n  /**\n   * Tools uploaded by the frontend request body.\n   */\n  frontend?: Record<string, ToolJSONSchema>;\n};\n\n/**\n * Builds an AI SDK `ToolSet` for server-side use with `streamText` /\n * `generateText` from a generative `toolkit` and the frontend-uploaded tools.\n *\n * Each toolkit tool's `execute` runs on the server. Pair this with the\n * `\"use generative\"` compiler: import the toolkit in a server route (where it\n * resolves to the server build — schema + `execute`, with `render` stripped) and\n * pass it here. Tools without an `execute` are still exposed to the model but\n * left for the client to fulfill. `frontendTools` lets the client contribute\n * tools that aren't in the static toolkit. Use {@link AISDKToolkit} when the\n * toolkit contains MCP entries.\n *\n * @example\n * ```ts\n * const { tools } = await req.json();\n * streamText({\n *   model,\n *   messages,\n *   tools: generativeTools({ toolkit: docsToolkit, frontendTools: tools }),\n * });\n * ```\n */\nexport const generativeTools = (options: GenerativeToolsOptions): ToolSet => {\n  assertNoMcpToolkitTools(options.toolkit);\n  return {\n    ...(options.frontendTools ? frontendTools(options.frontendTools) : {}),\n    // `toolkit` last so its server-side `execute` wins over an uploaded entry of\n    // the same name. The cast recovers the declaration shape — the server build\n    // carries `execute`, which the canonical `Toolkit` type erases.\n    ...toProviderToolSet(options.toolkit),\n    ...toServerToolSet(options.toolkit as ToolkitDefinition),\n  };\n};\n\nexport class AISDKToolkit {\n  readonly #toolkit: Toolkit;\n  readonly #mcpClients = new Map<string, Promise<MCPClient>>();\n\n  constructor(options: AISDKToolkitOptions) {\n    this.#toolkit = options.toolkit;\n  }\n\n  async tools(options: AISDKToolkitToolsOptions = {}): Promise<ToolSet> {\n    return {\n      ...(options.frontend ? frontendTools(options.frontend) : {}),\n      ...(await this.#mcpTools()),\n      ...toProviderToolSet(this.#toolkit),\n      ...toServerToolSet(this.#toolkit as ToolkitDefinition),\n    };\n  }\n\n  async close(): Promise<void> {\n    const clientPromises = [...this.#mcpClients.values()];\n    this.#mcpClients.clear();\n    const results = await Promise.allSettled(clientPromises);\n    const clients = results.flatMap((result) =>\n      result.status === \"fulfilled\" ? [result.value] : [],\n    );\n    const closeResults = await Promise.allSettled(\n      clients.map((client) => client.close()),\n    );\n    const errors = [\n      ...results.flatMap((result) =>\n        result.status === \"rejected\" ? [result.reason] : [],\n      ),\n      ...closeResults.flatMap((result) =>\n        result.status === \"rejected\" ? [result.reason] : [],\n      ),\n    ];\n    if (errors.length === 1) throw errors[0];\n    if (errors.length > 1) {\n      throw new AggregateError(\n        errors,\n        \"Failed to close one or more MCP clients\",\n      );\n    }\n  }\n\n  async #mcpTools(): Promise<ToolSet> {\n    const toolSets = await Promise.all(\n      Object.entries(this.#toolkit)\n        .filter((entry): entry is [string, McpToolkitTool] =>\n          isMcpToolkitTool(entry[1]),\n        )\n        .map(async ([name, tool]) => {\n          const client = await this.#mcpClient(name, tool.server);\n          return [name, await client.tools()] as const;\n        }),\n    );\n\n    const tools: ToolSet = {};\n    const toolSources = new Map<string, string>();\n    for (const [serverName, toolSet] of toolSets) {\n      for (const [toolName, tool] of Object.entries(toolSet)) {\n        const existingServerName = toolSources.get(toolName);\n        if (existingServerName) {\n          throw new Error(\n            `MCP tool name collision: \"${toolName}\" is exposed by both \"${existingServerName}\" and \"${serverName}\". Rename one of the toolkit entries or expose distinct MCP tool names.`,\n          );\n        }\n        toolSources.set(toolName, serverName);\n        tools[toolName] = tool;\n      }\n    }\n    return tools;\n  }\n\n  #mcpClient(name: string, config: McpServerConfig): Promise<MCPClient> {\n    const existing = this.#mcpClients.get(name);\n    if (existing) return existing;\n    let next: Promise<MCPClient>;\n    next = createMCPClient(toMCPClientConfig(config)).catch((error) => {\n      if (this.#mcpClients.get(name) === next) {\n        this.#mcpClients.delete(name);\n      }\n      throw error;\n    });\n    this.#mcpClients.set(name, next);\n    return next;\n  }\n}\n\nconst toMCPClientConfig = (config: McpServerConfig): MCPClientConfig => {\n  if (config.type === \"stdio\") {\n    return {\n      transport: new Experimental_StdioMCPTransport({\n        command: config.command,\n        ...(config.args && { args: [...config.args] }),\n        ...(config.env && { env: config.env }),\n        ...(config.cwd && { cwd: config.cwd }),\n      }),\n    };\n  }\n\n  return {\n    transport: {\n      type: config.type,\n      url: config.url,\n      ...(config.headers && { headers: config.headers }),\n      ...(config.redirect && { redirect: config.redirect }),\n    },\n  };\n};\n\ntype ToolkitTool = Toolkit[string];\n\ntype McpToolkitTool = ToolkitTool & {\n  type: \"mcp\";\n  server: McpServerConfig;\n};\n\nconst isMcpToolkitTool = (tool: ToolkitTool): tool is McpToolkitTool =>\n  tool.type === \"mcp\" && !tool.disabled;\n\nconst assertNoMcpToolkitTools = (toolkit: Toolkit): void => {\n  const mcpToolName = Object.entries(toolkit).find(([, tool]) =>\n    isMcpToolkitTool(tool),\n  )?.[0];\n  if (!mcpToolName) return;\n\n  throw new Error(\n    `MCP toolkit entry \"${mcpToolName}\" requires AISDKToolkit. Use new AISDKToolkit({ toolkit }).tools(...) instead of generativeTools(...).`,\n  );\n};\n\ntype AISDKToModelOutputOptions<TArgs, TResult> = Omit<\n  Parameters<ToolModelOutputFunction<TArgs, TResult>>[0],\n  \"output\"\n> & {\n  output: TResult | ModelContentEnvelope<TResult>;\n};\n\nconst toAISDKToModelOutput =\n  <TArgs, TResult>(toModelOutput?: ToolModelOutputFunction<TArgs, TResult>) =>\n  async (options: AISDKToModelOutputOptions<TArgs, TResult>) => {\n    const { result, modelContent } = unwrapModelContentEnvelope(options.output);\n\n    if (modelContent !== undefined) {\n      return toAISDKContent(modelContent);\n    }\n\n    if (!toModelOutput) {\n      return toAISDKDefaultOutput(result);\n    }\n\n    const parts = await toModelOutput({\n      ...options,\n      output: result,\n    });\n    return toAISDKContent(parts);\n  };\n\nconst toServerToolSet = (toolkit: ToolkitDefinition): ToolSet =>\n  Object.fromEntries(\n    Object.entries(toolkit)\n      .filter(\n        ([, t]) => t.type !== \"mcp\" && t.type !== \"provider\" && !t.disabled,\n      )\n      .map(([name, t]) => {\n        const execute = t.execute;\n        return [\n          name,\n          {\n            ...(t.description !== undefined && { description: t.description }),\n            inputSchema: parametersToInputSchema(t.parameters),\n            toModelOutput: toAISDKToModelOutput(t.toModelOutput),\n            ...(t.providerOptions && { providerOptions: t.providerOptions }),\n            ...(execute && {\n              execute: (\n                args: unknown,\n                callOptions: { toolCallId: string; abortSignal?: AbortSignal },\n              ) =>\n                execute(args as never, {\n                  toolCallId: callOptions.toolCallId,\n                  abortSignal: callOptions.abortSignal ?? neverAbort,\n                  human: humanNotSupported,\n                }),\n            }),\n          },\n        ];\n      }),\n  ) as ToolSet;\n\nconst toProviderToolSet = (toolkit: Toolkit): ToolSet =>\n  Object.fromEntries(\n    Object.entries(toolkit)\n      .filter((entry): entry is [string, ProviderToolkitTool] =>\n        isProviderToolkitTool(entry[1]),\n      )\n      .map(([name, t]) => [\n        name,\n        {\n          type: \"provider\",\n          id: t.providerId,\n          args: t.args,\n          ...(t.parameters && {\n            inputSchema: parametersToInputSchema(t.parameters),\n          }),\n          ...(t.providerOptions && { providerOptions: t.providerOptions }),\n          ...(t.supportsDeferredResults !== undefined && {\n            supportsDeferredResults: t.supportsDeferredResults,\n          }),\n        },\n      ]),\n  ) as ToolSet;\n\ntype ProviderToolkitTool = Extract<Toolkit[string], { type: \"provider\" }>;\n\nconst isProviderToolkitTool = (\n  tool: Toolkit[string],\n): tool is ProviderToolkitTool => tool.type === \"provider\" && !tool.disabled;\n"],"mappings":";;;;;;;;AAmBA,MAAM,eAAe;CAAE,MAAM;CAAmB,YAAY,CAAC;AAAE;AAE/D,MAAM,0BAAiC;CACrC,MAAM,IAAI,MACR,+DACF;AACF;AAGA,MAAM,aAAa,IAAI,gBAAgB,EAAE;AAEzC,MAAM,2BAA2B,eAC/B,WAAW,aAAa,aAAa,UAAU,IAAI,YAAY;;;;;;;;;;;;;;;;;;;;;;;AAkDjE,MAAa,mBAAmB,YAA6C;CAC3E,wBAAwB,QAAQ,OAAO;CACvC,OAAO;EACL,GAAI,QAAQ,gBAAgB,cAAc,QAAQ,aAAa,IAAI,CAAC;EAIpE,GAAG,kBAAkB,QAAQ,OAAO;EACpC,GAAG,gBAAgB,QAAQ,OAA4B;CACzD;AACF;AAEA,IAAa,eAAb,MAA0B;CACxB;CACA,8BAAuB,IAAI,IAAgC;CAE3D,YAAY,SAA8B;EACxC,KAAKA,WAAW,QAAQ;CAC1B;CAEA,MAAM,MAAM,UAAoC,CAAC,GAAqB;EACpE,OAAO;GACL,GAAI,QAAQ,WAAW,cAAc,QAAQ,QAAQ,IAAI,CAAC;GAC1D,GAAI,MAAM,KAAKE,UAAU;GACzB,GAAG,kBAAkB,KAAKF,QAAQ;GAClC,GAAG,gBAAgB,KAAKA,QAA6B;EACvD;CACF;CAEA,MAAM,QAAuB;EAC3B,MAAM,iBAAiB,CAAC,GAAG,KAAKC,YAAY,OAAO,CAAC;EACpD,KAAKA,YAAY,MAAM;EACvB,MAAM,UAAU,MAAM,QAAQ,WAAW,cAAc;EACvD,MAAM,UAAU,QAAQ,SAAS,WAC/B,OAAO,WAAW,cAAc,CAAC,OAAO,KAAK,IAAI,CAAC,CACpD;EACA,MAAM,eAAe,MAAM,QAAQ,WACjC,QAAQ,KAAK,WAAW,OAAO,MAAM,CAAC,CACxC;EACA,MAAM,SAAS,CACb,GAAG,QAAQ,SAAS,WAClB,OAAO,WAAW,aAAa,CAAC,OAAO,MAAM,IAAI,CAAC,CACpD,GACA,GAAG,aAAa,SAAS,WACvB,OAAO,WAAW,aAAa,CAAC,OAAO,MAAM,IAAI,CAAC,CACpD,CACF;EACA,IAAI,OAAO,WAAW,GAAG,MAAM,OAAO;EACtC,IAAI,OAAO,SAAS,GAClB,MAAM,IAAI,eACR,QACA,yCACF;CAEJ;CAEA,MAAMC,YAA8B;EAClC,MAAM,WAAW,MAAM,QAAQ,IAC7B,OAAO,QAAQ,KAAKF,QAAQ,EACzB,QAAQ,UACP,iBAAiB,MAAM,EAAE,CAC3B,EACC,IAAI,OAAO,CAAC,MAAM,UAAU;GAE3B,OAAO,CAAC,MAAM,OAAM,MADC,KAAKG,WAAW,MAAM,KAAK,MAAM,GAC3B,MAAM,CAAC;EACpC,CAAC,CACL;EAEA,MAAM,QAAiB,CAAC;EACxB,MAAM,8BAAc,IAAI,IAAoB;EAC5C,KAAK,MAAM,CAAC,YAAY,YAAY,UAClC,KAAK,MAAM,CAAC,UAAU,SAAS,OAAO,QAAQ,OAAO,GAAG;GACtD,MAAM,qBAAqB,YAAY,IAAI,QAAQ;GACnD,IAAI,oBACF,MAAM,IAAI,MACR,6BAA6B,SAAS,wBAAwB,mBAAmB,SAAS,WAAW,wEACvG;GAEF,YAAY,IAAI,UAAU,UAAU;GACpC,MAAM,YAAY;EACpB;EAEF,OAAO;CACT;CAEA,WAAW,MAAc,QAA6C;EACpE,MAAM,WAAW,KAAKF,YAAY,IAAI,IAAI;EAC1C,IAAI,UAAU,OAAO;EACrB,IAAI;EACJ,OAAO,gBAAgB,kBAAkB,MAAM,CAAC,EAAE,OAAO,UAAU;GACjE,IAAI,KAAKA,YAAY,IAAI,IAAI,MAAM,MACjC,KAAKA,YAAY,OAAO,IAAI;GAE9B,MAAM;EACR,CAAC;EACD,KAAKA,YAAY,IAAI,MAAM,IAAI;EAC/B,OAAO;CACT;AACF;AAEA,MAAM,qBAAqB,WAA6C;CACtE,IAAI,OAAO,SAAS,SAClB,OAAO,EACL,WAAW,IAAI,+BAA+B;EAC5C,SAAS,OAAO;EAChB,GAAI,OAAO,QAAQ,EAAE,MAAM,CAAC,GAAG,OAAO,IAAI,EAAE;EAC5C,GAAI,OAAO,OAAO,EAAE,KAAK,OAAO,IAAI;EACpC,GAAI,OAAO,OAAO,EAAE,KAAK,OAAO,IAAI;CACtC,CAAC,EACH;CAGF,OAAO,EACL,WAAW;EACT,MAAM,OAAO;EACb,KAAK,OAAO;EACZ,GAAI,OAAO,WAAW,EAAE,SAAS,OAAO,QAAQ;EAChD,GAAI,OAAO,YAAY,EAAE,UAAU,OAAO,SAAS;CACrD,EACF;AACF;AASA,MAAM,oBAAoB,SACxB,KAAK,SAAS,SAAS,CAAC,KAAK;AAE/B,MAAM,2BAA2B,YAA2B;CAC1D,MAAM,cAAc,OAAO,QAAQ,OAAO,EAAE,MAAM,GAAG,UACnD,iBAAiB,IAAI,CACvB,IAAI;CACJ,IAAI,CAAC,aAAa;CAElB,MAAM,IAAI,MACR,sBAAsB,YAAY,uGACpC;AACF;AASA,MAAM,wBACa,kBACjB,OAAO,YAAuD;CAC5D,MAAM,EAAE,QAAQ,iBAAiB,2BAA2B,QAAQ,MAAM;CAE1E,IAAI,iBAAiB,KAAA,GACnB,OAAO,eAAe,YAAY;CAGpC,IAAI,CAAC,eACH,OAAO,qBAAqB,MAAM;CAOpC,OAAO,eAAe,MAJF,cAAc;EAChC,GAAG;EACH,QAAQ;CACV,CAAC,CAC0B;AAC7B;AAEF,MAAM,mBAAmB,YACvB,OAAO,YACL,OAAO,QAAQ,OAAO,EACnB,QACE,GAAG,OAAO,EAAE,SAAS,SAAS,EAAE,SAAS,cAAc,CAAC,EAAE,QAC7D,EACC,KAAK,CAAC,MAAM,OAAO;CAClB,MAAM,UAAU,EAAE;CAClB,OAAO,CACL,MACA;EACE,GAAI,EAAE,gBAAgB,KAAA,KAAa,EAAE,aAAa,EAAE,YAAY;EAChE,aAAa,wBAAwB,EAAE,UAAU;EACjD,eAAe,qBAAqB,EAAE,aAAa;EACnD,GAAI,EAAE,mBAAmB,EAAE,iBAAiB,EAAE,gBAAgB;EAC9D,GAAI,WAAW,EACb,UACE,MACA,gBAEA,QAAQ,MAAe;GACrB,YAAY,YAAY;GACxB,aAAa,YAAY,eAAe;GACxC,OAAO;EACT,CAAC,EACL;CACF,CACF;AACF,CAAC,CACL;AAEF,MAAM,qBAAqB,YACzB,OAAO,YACL,OAAO,QAAQ,OAAO,EACnB,QAAQ,UACP,sBAAsB,MAAM,EAAE,CAChC,EACC,KAAK,CAAC,MAAM,OAAO,CAClB,MACA;CACE,MAAM;CACN,IAAI,EAAE;CACN,MAAM,EAAE;CACR,GAAI,EAAE,cAAc,EAClB,aAAa,wBAAwB,EAAE,UAAU,EACnD;CACA,GAAI,EAAE,mBAAmB,EAAE,iBAAiB,EAAE,gBAAgB;CAC9D,GAAI,EAAE,4BAA4B,KAAA,KAAa,EAC7C,yBAAyB,EAAE,wBAC7B;AACF,CACF,CAAC,CACL;AAIF,MAAM,yBACJ,SACgC,KAAK,SAAS,cAAc,CAAC,KAAK"}
+{"version":3,"file":"generativeTools.js","names":["#toolkit","#mcpClients","#mcpTools","#mcpClient"],"sources":["../src/generativeTools.ts"],"sourcesContent":["import { jsonSchema, type ToolSet } from \"ai\";\nimport type { MCPClient, MCPClientConfig } from \"@ai-sdk/mcp\";\nimport { createMCPClient } from \"@ai-sdk/mcp\";\nimport { Experimental_StdioMCPTransport } from \"#mcp-stdio\";\nimport {\n  toJSONSchema,\n  type Tool,\n  type McpServerConfig,\n  type ToolJSONSchema,\n  type ToolModelOutputFunction,\n} from \"assistant-stream\";\nimport type { Toolkit, ToolkitDefinition } from \"@assistant-ui/core/react\";\nimport { frontendTools } from \"./frontendTools\";\nimport { toAISDKContent, toAISDKDefaultOutput } from \"./toolOutputConversion\";\nimport {\n  unwrapModelContentEnvelope,\n  type ModelContentEnvelope,\n} from \"./modelContentEnvelope\";\n\nconst EMPTY_SCHEMA = { type: \"object\" as const, properties: {} };\n\nconst humanNotSupported = (): never => {\n  throw new Error(\n    \"`human()` is not available during server-side tool execution.\",\n  );\n};\n\n// AI SDK leaves `abortSignal` optional; assistant-ui's execute requires one.\nconst neverAbort = new AbortController().signal;\n\nconst parametersToInputSchema = (parameters: Tool[\"parameters\"] | undefined) =>\n  jsonSchema(parameters ? toJSONSchema(parameters) : EMPTY_SCHEMA);\n\n/**\n * @deprecated Options for the deprecated {@link generativeTools}. Use\n * {@link AISDKToolkit} with {@link AISDKToolkitOptions} /\n * {@link AISDKToolkitToolsOptions} instead.\n */\nexport interface GenerativeToolsOptions {\n  /**\n   * The server build of a generative toolkit (schema + server `execute`). Typed\n   * as the canonical {@link Toolkit} so callers don't need to cast; the server\n   * build carries `execute`, recovered internally as {@link ToolkitDefinition}.\n   */\n  toolkit: Toolkit;\n  /**\n   * Tools uploaded by the frontend (the request body's `tools`). Merged in\n   * alongside the `toolkit`; a server `execute` from `toolkit` takes precedence\n   * over an uploaded entry of the same name.\n   */\n  frontendTools?: Record<string, ToolJSONSchema>;\n}\n\nexport type AISDKToolkitOptions = {\n  toolkit: Toolkit;\n};\n\nexport type AISDKToolkitToolsOptions = {\n  /**\n   * Tools uploaded by the frontend request body.\n   */\n  frontend?: Record<string, ToolJSONSchema>;\n};\n\n/**\n * Builds an AI SDK `ToolSet` for server-side use with `streamText` /\n * `generateText` from a generative `toolkit` and the frontend-uploaded tools.\n *\n * Each toolkit tool's `execute` runs on the server. Pair this with the\n * `\"use generative\"` compiler: import the toolkit in a server route (where it\n * resolves to the server build — schema + `execute`, with `render` stripped) and\n * pass it here. Tools without an `execute` are still exposed to the model but\n * left for the client to fulfill. `frontendTools` lets the client contribute\n * tools that aren't in the static toolkit.\n *\n * @deprecated Use {@link AISDKToolkit} instead:\n * `new AISDKToolkit({ toolkit }).tools({ frontend })`. It is a strict superset\n * (it also opens MCP server connections), so it replaces `generativeTools`\n * everywhere. The `frontendTools` option is named `frontend` on `.tools()`, and\n * `.tools()` is async. `generativeTools` will be removed in a future version.\n *\n * @example\n * ```ts\n * // Define once at module scope so any MCP connections pool across requests.\n * const aiToolkit = new AISDKToolkit({ toolkit: docsToolkit });\n *\n * // In your route handler:\n * const { tools } = await req.json();\n * streamText({\n *   model,\n *   messages,\n *   tools: await aiToolkit.tools({ frontend: tools }),\n * });\n * ```\n */\nexport const generativeTools = (options: GenerativeToolsOptions): ToolSet => {\n  assertNoMcpToolkitTools(options.toolkit);\n  return {\n    ...(options.frontendTools ? frontendTools(options.frontendTools) : {}),\n    // `toolkit` last so its server-side `execute` wins over an uploaded entry of\n    // the same name. The cast recovers the declaration shape — the server build\n    // carries `execute`, which the canonical `Toolkit` type erases.\n    ...toProviderToolSet(options.toolkit),\n    ...toServerToolSet(options.toolkit as ToolkitDefinition),\n  };\n};\n\nexport class AISDKToolkit {\n  readonly #toolkit: Toolkit;\n  readonly #mcpClients = new Map<string, Promise<MCPClient>>();\n\n  constructor(options: AISDKToolkitOptions) {\n    this.#toolkit = options.toolkit;\n  }\n\n  async tools(options: AISDKToolkitToolsOptions = {}): Promise<ToolSet> {\n    return {\n      ...(options.frontend ? frontendTools(options.frontend) : {}),\n      ...(await this.#mcpTools()),\n      ...toProviderToolSet(this.#toolkit),\n      ...toServerToolSet(this.#toolkit as ToolkitDefinition),\n    };\n  }\n\n  async close(): Promise<void> {\n    const clientPromises = [...this.#mcpClients.values()];\n    this.#mcpClients.clear();\n    const results = await Promise.allSettled(clientPromises);\n    const clients = results.flatMap((result) =>\n      result.status === \"fulfilled\" ? [result.value] : [],\n    );\n    const closeResults = await Promise.allSettled(\n      clients.map((client) => client.close()),\n    );\n    const errors = [\n      ...results.flatMap((result) =>\n        result.status === \"rejected\" ? [result.reason] : [],\n      ),\n      ...closeResults.flatMap((result) =>\n        result.status === \"rejected\" ? [result.reason] : [],\n      ),\n    ];\n    if (errors.length === 1) throw errors[0];\n    if (errors.length > 1) {\n      throw new AggregateError(\n        errors,\n        \"Failed to close one or more MCP clients\",\n      );\n    }\n  }\n\n  async #mcpTools(): Promise<ToolSet> {\n    const toolSets = await Promise.all(\n      Object.entries(this.#toolkit)\n        .filter((entry): entry is [string, McpToolkitTool] =>\n          isMcpToolkitTool(entry[1]),\n        )\n        .map(async ([name, tool]) => {\n          const client = await this.#mcpClient(name, tool.server);\n          return [name, await client.tools()] as const;\n        }),\n    );\n\n    const tools: ToolSet = {};\n    const toolSources = new Map<string, string>();\n    for (const [serverName, toolSet] of toolSets) {\n      for (const [toolName, tool] of Object.entries(toolSet)) {\n        const existingServerName = toolSources.get(toolName);\n        if (existingServerName) {\n          throw new Error(\n            `MCP tool name collision: \"${toolName}\" is exposed by both \"${existingServerName}\" and \"${serverName}\". Rename one of the toolkit entries or expose distinct MCP tool names.`,\n          );\n        }\n        toolSources.set(toolName, serverName);\n        tools[toolName] = tool;\n      }\n    }\n    return tools;\n  }\n\n  #mcpClient(name: string, config: McpServerConfig): Promise<MCPClient> {\n    const existing = this.#mcpClients.get(name);\n    if (existing) return existing;\n    let next: Promise<MCPClient>;\n    next = createMCPClient(toMCPClientConfig(config)).catch((error) => {\n      if (this.#mcpClients.get(name) === next) {\n        this.#mcpClients.delete(name);\n      }\n      throw error;\n    });\n    this.#mcpClients.set(name, next);\n    return next;\n  }\n}\n\nconst toMCPClientConfig = (config: McpServerConfig): MCPClientConfig => {\n  if (config.type === \"stdio\") {\n    return {\n      transport: new Experimental_StdioMCPTransport({\n        command: config.command,\n        ...(config.args && { args: [...config.args] }),\n        ...(config.env && { env: config.env }),\n        ...(config.cwd && { cwd: config.cwd }),\n      }),\n    };\n  }\n\n  return {\n    transport: {\n      type: config.type,\n      url: config.url,\n      ...(config.headers && { headers: config.headers }),\n      ...(config.redirect && { redirect: config.redirect }),\n    },\n  };\n};\n\ntype ToolkitTool = Toolkit[string];\n\ntype McpToolkitTool = ToolkitTool & {\n  type: \"mcp\";\n  server: McpServerConfig;\n};\n\nconst isMcpToolkitTool = (tool: ToolkitTool): tool is McpToolkitTool =>\n  tool.type === \"mcp\" && !tool.disabled;\n\nconst assertNoMcpToolkitTools = (toolkit: Toolkit): void => {\n  const mcpToolName = Object.entries(toolkit).find(([, tool]) =>\n    isMcpToolkitTool(tool),\n  )?.[0];\n  if (!mcpToolName) return;\n\n  throw new Error(\n    `MCP toolkit entry \"${mcpToolName}\" requires AISDKToolkit. Use new AISDKToolkit({ toolkit }).tools(...) instead of generativeTools(...).`,\n  );\n};\n\ntype AISDKToModelOutputOptions<TArgs, TResult> = Omit<\n  Parameters<ToolModelOutputFunction<TArgs, TResult>>[0],\n  \"output\"\n> & {\n  output: TResult | ModelContentEnvelope<TResult>;\n};\n\nconst toAISDKToModelOutput =\n  <TArgs, TResult>(toModelOutput?: ToolModelOutputFunction<TArgs, TResult>) =>\n  async (options: AISDKToModelOutputOptions<TArgs, TResult>) => {\n    const { result, modelContent } = unwrapModelContentEnvelope(options.output);\n\n    if (modelContent !== undefined) {\n      return toAISDKContent(modelContent);\n    }\n\n    if (!toModelOutput) {\n      return toAISDKDefaultOutput(result);\n    }\n\n    const parts = await toModelOutput({\n      ...options,\n      output: result,\n    });\n    return toAISDKContent(parts);\n  };\n\nconst toServerToolSet = (toolkit: ToolkitDefinition): ToolSet =>\n  Object.fromEntries(\n    Object.entries(toolkit)\n      .filter(\n        ([, t]) => t.type !== \"mcp\" && t.type !== \"provider\" && !t.disabled,\n      )\n      .map(([name, t]) => {\n        const execute = t.execute;\n        return [\n          name,\n          {\n            ...(t.description !== undefined && { description: t.description }),\n            inputSchema: parametersToInputSchema(t.parameters),\n            toModelOutput: toAISDKToModelOutput(t.toModelOutput),\n            ...(t.providerOptions && { providerOptions: t.providerOptions }),\n            ...(execute && {\n              execute: (\n                args: unknown,\n                callOptions: { toolCallId: string; abortSignal?: AbortSignal },\n              ) =>\n                execute(args as never, {\n                  toolCallId: callOptions.toolCallId,\n                  abortSignal: callOptions.abortSignal ?? neverAbort,\n                  human: humanNotSupported,\n                }),\n            }),\n          },\n        ];\n      }),\n  ) as ToolSet;\n\nconst toProviderToolSet = (toolkit: Toolkit): ToolSet =>\n  Object.fromEntries(\n    Object.entries(toolkit)\n      .filter((entry): entry is [string, ProviderToolkitTool] =>\n        isProviderToolkitTool(entry[1]),\n      )\n      .map(([name, t]) => [\n        name,\n        {\n          type: \"provider\",\n          id: t.providerId,\n          args: t.args,\n          ...(t.parameters && {\n            inputSchema: parametersToInputSchema(t.parameters),\n          }),\n          ...(t.providerOptions && { providerOptions: t.providerOptions }),\n          ...(t.supportsDeferredResults !== undefined && {\n            supportsDeferredResults: t.supportsDeferredResults,\n          }),\n        },\n      ]),\n  ) as ToolSet;\n\ntype ProviderToolkitTool = Extract<Toolkit[string], { type: \"provider\" }>;\n\nconst isProviderToolkitTool = (\n  tool: Toolkit[string],\n): tool is ProviderToolkitTool => tool.type === \"provider\" && !tool.disabled;\n"],"mappings":";;;;;;;;AAmBA,MAAM,eAAe;CAAE,MAAM;CAAmB,YAAY,CAAC;AAAE;AAE/D,MAAM,0BAAiC;CACrC,MAAM,IAAI,MACR,+DACF;AACF;AAGA,MAAM,aAAa,IAAI,gBAAgB,CAAC,CAAC;AAEzC,MAAM,2BAA2B,eAC/B,WAAW,aAAa,aAAa,UAAU,IAAI,YAAY;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAgEjE,MAAa,mBAAmB,YAA6C;CAC3E,wBAAwB,QAAQ,OAAO;CACvC,OAAO;EACL,GAAI,QAAQ,gBAAgB,cAAc,QAAQ,aAAa,IAAI,CAAC;EAIpE,GAAG,kBAAkB,QAAQ,OAAO;EACpC,GAAG,gBAAgB,QAAQ,OAA4B;CACzD;AACF;AAEA,IAAa,eAAb,MAA0B;CACxB;CACA,8BAAuB,IAAI,IAAgC;CAE3D,YAAY,SAA8B;EACxC,KAAKA,WAAW,QAAQ;CAC1B;CAEA,MAAM,MAAM,UAAoC,CAAC,GAAqB;EACpE,OAAO;GACL,GAAI,QAAQ,WAAW,cAAc,QAAQ,QAAQ,IAAI,CAAC;GAC1D,GAAI,MAAM,KAAKE,UAAU;GACzB,GAAG,kBAAkB,KAAKF,QAAQ;GAClC,GAAG,gBAAgB,KAAKA,QAA6B;EACvD;CACF;CAEA,MAAM,QAAuB;EAC3B,MAAM,iBAAiB,CAAC,GAAG,KAAKC,YAAY,OAAO,CAAC;EACpD,KAAKA,YAAY,MAAM;EACvB,MAAM,UAAU,MAAM,QAAQ,WAAW,cAAc;EACvD,MAAM,UAAU,QAAQ,SAAS,WAC/B,OAAO,WAAW,cAAc,CAAC,OAAO,KAAK,IAAI,CAAC,CACpD;EACA,MAAM,eAAe,MAAM,QAAQ,WACjC,QAAQ,KAAK,WAAW,OAAO,MAAM,CAAC,CACxC;EACA,MAAM,SAAS,CACb,GAAG,QAAQ,SAAS,WAClB,OAAO,WAAW,aAAa,CAAC,OAAO,MAAM,IAAI,CAAC,CACpD,GACA,GAAG,aAAa,SAAS,WACvB,OAAO,WAAW,aAAa,CAAC,OAAO,MAAM,IAAI,CAAC,CACpD,CACF;EACA,IAAI,OAAO,WAAW,GAAG,MAAM,OAAO;EACtC,IAAI,OAAO,SAAS,GAClB,MAAM,IAAI,eACR,QACA,yCACF;CAEJ;CAEA,MAAMC,YAA8B;EAClC,MAAM,WAAW,MAAM,QAAQ,IAC7B,OAAO,QAAQ,KAAKF,QAAQ,CAAC,CAC1B,QAAQ,UACP,iBAAiB,MAAM,EAAE,CAC3B,CAAC,CACA,IAAI,OAAO,CAAC,MAAM,UAAU;GAE3B,OAAO,CAAC,MAAM,OAAM,MADC,KAAKG,WAAW,MAAM,KAAK,MAAM,EAAA,CAC3B,MAAM,CAAC;EACpC,CAAC,CACL;EAEA,MAAM,QAAiB,CAAC;EACxB,MAAM,8BAAc,IAAI,IAAoB;EAC5C,KAAK,MAAM,CAAC,YAAY,YAAY,UAClC,KAAK,MAAM,CAAC,UAAU,SAAS,OAAO,QAAQ,OAAO,GAAG;GACtD,MAAM,qBAAqB,YAAY,IAAI,QAAQ;GACnD,IAAI,oBACF,MAAM,IAAI,MACR,6BAA6B,SAAS,wBAAwB,mBAAmB,SAAS,WAAW,wEACvG;GAEF,YAAY,IAAI,UAAU,UAAU;GACpC,MAAM,YAAY;EACpB;EAEF,OAAO;CACT;CAEA,WAAW,MAAc,QAA6C;EACpE,MAAM,WAAW,KAAKF,YAAY,IAAI,IAAI;EAC1C,IAAI,UAAU,OAAO;EACrB,IAAI;EACJ,OAAO,gBAAgB,kBAAkB,MAAM,CAAC,CAAC,CAAC,OAAO,UAAU;GACjE,IAAI,KAAKA,YAAY,IAAI,IAAI,MAAM,MACjC,KAAKA,YAAY,OAAO,IAAI;GAE9B,MAAM;EACR,CAAC;EACD,KAAKA,YAAY,IAAI,MAAM,IAAI;EAC/B,OAAO;CACT;AACF;AAEA,MAAM,qBAAqB,WAA6C;CACtE,IAAI,OAAO,SAAS,SAClB,OAAO,EACL,WAAW,IAAI,+BAA+B;EAC5C,SAAS,OAAO;EAChB,GAAI,OAAO,QAAQ,EAAE,MAAM,CAAC,GAAG,OAAO,IAAI,EAAE;EAC5C,GAAI,OAAO,OAAO,EAAE,KAAK,OAAO,IAAI;EACpC,GAAI,OAAO,OAAO,EAAE,KAAK,OAAO,IAAI;CACtC,CAAC,EACH;CAGF,OAAO,EACL,WAAW;EACT,MAAM,OAAO;EACb,KAAK,OAAO;EACZ,GAAI,OAAO,WAAW,EAAE,SAAS,OAAO,QAAQ;EAChD,GAAI,OAAO,YAAY,EAAE,UAAU,OAAO,SAAS;CACrD,EACF;AACF;AASA,MAAM,oBAAoB,SACxB,KAAK,SAAS,SAAS,CAAC,KAAK;AAE/B,MAAM,2BAA2B,YAA2B;CAC1D,MAAM,cAAc,OAAO,QAAQ,OAAO,CAAC,CAAC,MAAM,GAAG,UACnD,iBAAiB,IAAI,CACvB,CAAC,GAAG;CACJ,IAAI,CAAC,aAAa;CAElB,MAAM,IAAI,MACR,sBAAsB,YAAY,uGACpC;AACF;AASA,MAAM,wBACa,kBACjB,OAAO,YAAuD;CAC5D,MAAM,EAAE,QAAQ,iBAAiB,2BAA2B,QAAQ,MAAM;CAE1E,IAAI,iBAAiB,KAAA,GACnB,OAAO,eAAe,YAAY;CAGpC,IAAI,CAAC,eACH,OAAO,qBAAqB,MAAM;CAOpC,OAAO,eAAe,MAJF,cAAc;EAChC,GAAG;EACH,QAAQ;CACV,CAAC,CAC0B;AAC7B;AAEF,MAAM,mBAAmB,YACvB,OAAO,YACL,OAAO,QAAQ,OAAO,CAAC,CACpB,QACE,GAAG,OAAO,EAAE,SAAS,SAAS,EAAE,SAAS,cAAc,CAAC,EAAE,QAC7D,CAAC,CACA,KAAK,CAAC,MAAM,OAAO;CAClB,MAAM,UAAU,EAAE;CAClB,OAAO,CACL,MACA;EACE,GAAI,EAAE,gBAAgB,KAAA,KAAa,EAAE,aAAa,EAAE,YAAY;EAChE,aAAa,wBAAwB,EAAE,UAAU;EACjD,eAAe,qBAAqB,EAAE,aAAa;EACnD,GAAI,EAAE,mBAAmB,EAAE,iBAAiB,EAAE,gBAAgB;EAC9D,GAAI,WAAW,EACb,UACE,MACA,gBAEA,QAAQ,MAAe;GACrB,YAAY,YAAY;GACxB,aAAa,YAAY,eAAe;GACxC,OAAO;EACT,CAAC,EACL;CACF,CACF;AACF,CAAC,CACL;AAEF,MAAM,qBAAqB,YACzB,OAAO,YACL,OAAO,QAAQ,OAAO,CAAC,CACpB,QAAQ,UACP,sBAAsB,MAAM,EAAE,CAChC,CAAC,CACA,KAAK,CAAC,MAAM,OAAO,CAClB,MACA;CACE,MAAM;CACN,IAAI,EAAE;CACN,MAAM,EAAE;CACR,GAAI,EAAE,cAAc,EAClB,aAAa,wBAAwB,EAAE,UAAU,EACnD;CACA,GAAI,EAAE,mBAAmB,EAAE,iBAAiB,EAAE,gBAAgB;CAC9D,GAAI,EAAE,4BAA4B,KAAA,KAAa,EAC7C,yBAAyB,EAAE,wBAC7B;AACF,CACF,CAAC,CACL;AAIF,MAAM,yBACJ,SACgC,KAAK,SAAS,cAAc,CAAC,KAAK"}

package/dist/index.d.ts

@@ -1,11 +1,11 @@
 /// <reference types="@assistant-ui/core/react" />
-import { frontendTools } from "./frontendTools.js";
-import { AISDKToolkit, AISDKToolkitOptions, AISDKToolkitToolsOptions, GenerativeToolsOptions, generativeTools } from "./generativeTools.js";
 import { useAISDKRuntime } from "./ui/use-chat/useAISDKRuntime.js";
 import { UseChatRuntimeOptions, useChatRuntime } from "./ui/use-chat/useChatRuntime.js";
 import { RESUMABLE_STREAM_ID_HEADER } from "./assistant-stream/dist/resumable/createResumableAssistantStreamResponse.js";
 import { AssistantChatResumableOptions, ResumableClientStorage, createResumableSessionStorage } from "./ui/resumable.js";
 import { AssistantChatTransport } from "./ui/use-chat/AssistantChatTransport.js";
+import { frontendTools } from "./frontendTools.js";
 import { injectQuoteContext } from "./injectQuoteContext.js";
 import { ThreadTokenUsage, TokenUsageExtractableMessage, getThreadMessageTokenUsage, useThreadTokenUsage } from "./usage.js";
+import { AISDKToolkit, AISDKToolkitOptions, AISDKToolkitToolsOptions, GenerativeToolsOptions, generativeTools } from "./generativeTools.js";
 export { AISDKToolkit, type AISDKToolkitOptions, type AISDKToolkitToolsOptions, type AssistantChatResumableOptions, AssistantChatTransport, type GenerativeToolsOptions, RESUMABLE_STREAM_ID_HEADER, type ResumableClientStorage, type ThreadTokenUsage, type TokenUsageExtractableMessage, type UseChatRuntimeOptions, createResumableSessionStorage, frontendTools, generativeTools, getThreadMessageTokenUsage, injectQuoteContext, useAISDKRuntime, useChatRuntime, useThreadTokenUsage };

package/dist/index.js

@@ -1,10 +1,10 @@
-import { frontendTools } from "./frontendTools.js";
-import { AISDKToolkit, generativeTools } from "./generativeTools.js";
 import { useAISDKRuntime } from "./ui/use-chat/useAISDKRuntime.js";
 import { RESUMABLE_STREAM_ID_HEADER } from "./assistant-stream/dist/resumable/createResumableAssistantStreamResponse.js";
 import { createResumableSessionStorage } from "./ui/resumable.js";
 import { AssistantChatTransport } from "./ui/use-chat/AssistantChatTransport.js";
 import { useChatRuntime } from "./ui/use-chat/useChatRuntime.js";
+import { frontendTools } from "./frontendTools.js";
 import { injectQuoteContext } from "./injectQuoteContext.js";
 import { getThreadMessageTokenUsage, useThreadTokenUsage } from "./usage.js";
+import { AISDKToolkit, generativeTools } from "./generativeTools.js";
 export { AISDKToolkit, AssistantChatTransport, RESUMABLE_STREAM_ID_HEADER, createResumableSessionStorage, frontendTools, generativeTools, getThreadMessageTokenUsage, injectQuoteContext, useAISDKRuntime, useChatRuntime, useThreadTokenUsage };

package/dist/index.native.d.ts

@@ -0,0 +1,10 @@
+/// <reference types="@assistant-ui/core/react" />
+import { useAISDKRuntime } from "./ui/use-chat/useAISDKRuntime.js";
+import { UseChatRuntimeOptions, useChatRuntime } from "./ui/use-chat/useChatRuntime.js";
+import { RESUMABLE_STREAM_ID_HEADER } from "./assistant-stream/dist/resumable/createResumableAssistantStreamResponse.js";
+import { AssistantChatResumableOptions, ResumableClientStorage, createResumableSessionStorage } from "./ui/resumable.js";
+import { AssistantChatTransport } from "./ui/use-chat/AssistantChatTransport.js";
+import { frontendTools } from "./frontendTools.js";
+import { injectQuoteContext } from "./injectQuoteContext.js";
+import { ThreadTokenUsage, TokenUsageExtractableMessage, getThreadMessageTokenUsage, useThreadTokenUsage } from "./usage.js";
+export { type AssistantChatResumableOptions, AssistantChatTransport, RESUMABLE_STREAM_ID_HEADER, type ResumableClientStorage, type ThreadTokenUsage, type TokenUsageExtractableMessage, type UseChatRuntimeOptions, createResumableSessionStorage, frontendTools, getThreadMessageTokenUsage, injectQuoteContext, useAISDKRuntime, useChatRuntime, useThreadTokenUsage };

package/dist/index.native.js

@@ -0,0 +1,9 @@
+import { useAISDKRuntime } from "./ui/use-chat/useAISDKRuntime.js";
+import { RESUMABLE_STREAM_ID_HEADER } from "./assistant-stream/dist/resumable/createResumableAssistantStreamResponse.js";
+import { createResumableSessionStorage } from "./ui/resumable.js";
+import { AssistantChatTransport } from "./ui/use-chat/AssistantChatTransport.js";
+import { useChatRuntime } from "./ui/use-chat/useChatRuntime.js";
+import { frontendTools } from "./frontendTools.js";
+import { injectQuoteContext } from "./injectQuoteContext.js";
+import { getThreadMessageTokenUsage, useThreadTokenUsage } from "./usage.js";
+export { AssistantChatTransport, RESUMABLE_STREAM_ID_HEADER, createResumableSessionStorage, frontendTools, getThreadMessageTokenUsage, injectQuoteContext, useAISDKRuntime, useChatRuntime, useThreadTokenUsage };

package/dist/injectQuoteContext.js.map

@@ -1 +1 @@
-{"version":3,"file":"injectQuoteContext.js","names":[],"sources":["../src/injectQuoteContext.ts"],"sourcesContent":["import type { UIMessage } from \"ai\";\n\nconst getQuoteText = (metadata: unknown): string | undefined => {\n  if (!metadata || typeof metadata !== \"object\") return undefined;\n\n  const custom = (metadata as Record<string, unknown>).custom;\n  if (!custom || typeof custom !== \"object\") return undefined;\n\n  const quote = (custom as Record<string, unknown>).quote;\n  if (!quote || typeof quote !== \"object\") return undefined;\n\n  const text = (quote as Record<string, unknown>).text;\n  return typeof text === \"string\" ? text : undefined;\n};\n\n/**\n * Injects quote context into messages as markdown blockquotes.\n *\n * Use this in your route handler before `convertToModelMessages` so the LLM\n * sees the quoted text that the user is referring to.\n *\n * @example\n * ```ts\n * import { convertToModelMessages, streamText } from \"ai\";\n * import { injectQuoteContext } from \"@assistant-ui/react-ai-sdk\";\n *\n * export async function POST(req: Request) {\n *   const { messages } = await req.json();\n *   const result = streamText({\n *     model: myModel,\n *     messages: await convertToModelMessages(injectQuoteContext(messages)),\n *   });\n *   return result.toUIMessageStreamResponse();\n * }\n * ```\n */\nexport function injectQuoteContext(messages: UIMessage[]): UIMessage[] {\n  return messages.map((msg) => {\n    if (msg.role !== \"user\") return msg;\n\n    const text = getQuoteText(msg.metadata);\n    if (!text) return msg;\n\n    const blockquote = text\n      .split(/\\r?\\n/)\n      .map((line) => `> ${line}`)\n      .join(\"\\n\");\n\n    const alreadyInjected =\n      msg.parts[0]?.type === \"text\" &&\n      msg.parts[0].text === `${blockquote}\\n\\n`;\n    if (alreadyInjected) return msg;\n\n    return {\n      ...msg,\n      parts: [\n        { type: \"text\" as const, text: `${blockquote}\\n\\n` },\n        ...(msg.parts ?? []),\n      ],\n    };\n  });\n}\n"],"mappings":";AAEA,MAAM,gBAAgB,aAA0C;CAC9D,IAAI,CAAC,YAAY,OAAO,aAAa,UAAU,OAAO,KAAA;CAEtD,MAAM,SAAU,SAAqC;CACrD,IAAI,CAAC,UAAU,OAAO,WAAW,UAAU,OAAO,KAAA;CAElD,MAAM,QAAS,OAAmC;CAClD,IAAI,CAAC,SAAS,OAAO,UAAU,UAAU,OAAO,KAAA;CAEhD,MAAM,OAAQ,MAAkC;CAChD,OAAO,OAAO,SAAS,WAAW,OAAO,KAAA;AAC3C;;;;;;;;;;;;;;;;;;;;;;AAuBA,SAAgB,mBAAmB,UAAoC;CACrE,OAAO,SAAS,KAAK,QAAQ;EAC3B,IAAI,IAAI,SAAS,QAAQ,OAAO;EAEhC,MAAM,OAAO,aAAa,IAAI,QAAQ;EACtC,IAAI,CAAC,MAAM,OAAO;EAElB,MAAM,aAAa,KAChB,MAAM,OAAO,EACb,KAAK,SAAS,KAAK,MAAM,EACzB,KAAK,IAAI;EAKZ,IAFE,IAAI,MAAM,IAAI,SAAS,UACvB,IAAI,MAAM,GAAG,SAAS,GAAG,WAAW,OACjB,OAAO;EAE5B,OAAO;GACL,GAAG;GACH,OAAO,CACL;IAAE,MAAM;IAAiB,MAAM,GAAG,WAAW;GAAM,GACnD,GAAI,IAAI,SAAS,CAAC,CACpB;EACF;CACF,CAAC;AACH"}
+{"version":3,"file":"injectQuoteContext.js","names":[],"sources":["../src/injectQuoteContext.ts"],"sourcesContent":["import type { UIMessage } from \"ai\";\n\nconst getQuoteText = (metadata: unknown): string | undefined => {\n  if (!metadata || typeof metadata !== \"object\") return undefined;\n\n  const custom = (metadata as Record<string, unknown>).custom;\n  if (!custom || typeof custom !== \"object\") return undefined;\n\n  const quote = (custom as Record<string, unknown>).quote;\n  if (!quote || typeof quote !== \"object\") return undefined;\n\n  const text = (quote as Record<string, unknown>).text;\n  return typeof text === \"string\" ? text : undefined;\n};\n\n/**\n * Injects quote context into messages as markdown blockquotes.\n *\n * Use this in your route handler before `convertToModelMessages` so the LLM\n * sees the quoted text that the user is referring to.\n *\n * @example\n * ```ts\n * import { convertToModelMessages, streamText } from \"ai\";\n * import { injectQuoteContext } from \"@assistant-ui/react-ai-sdk\";\n *\n * export async function POST(req: Request) {\n *   const { messages } = await req.json();\n *   const result = streamText({\n *     model: myModel,\n *     messages: await convertToModelMessages(injectQuoteContext(messages)),\n *   });\n *   return result.toUIMessageStreamResponse();\n * }\n * ```\n */\nexport function injectQuoteContext(messages: UIMessage[]): UIMessage[] {\n  return messages.map((msg) => {\n    if (msg.role !== \"user\") return msg;\n\n    const text = getQuoteText(msg.metadata);\n    if (!text) return msg;\n\n    const blockquote = text\n      .split(/\\r?\\n/)\n      .map((line) => `> ${line}`)\n      .join(\"\\n\");\n\n    const alreadyInjected =\n      msg.parts[0]?.type === \"text\" &&\n      msg.parts[0].text === `${blockquote}\\n\\n`;\n    if (alreadyInjected) return msg;\n\n    return {\n      ...msg,\n      parts: [\n        { type: \"text\" as const, text: `${blockquote}\\n\\n` },\n        ...(msg.parts ?? []),\n      ],\n    };\n  });\n}\n"],"mappings":";AAEA,MAAM,gBAAgB,aAA0C;CAC9D,IAAI,CAAC,YAAY,OAAO,aAAa,UAAU,OAAO,KAAA;CAEtD,MAAM,SAAU,SAAqC;CACrD,IAAI,CAAC,UAAU,OAAO,WAAW,UAAU,OAAO,KAAA;CAElD,MAAM,QAAS,OAAmC;CAClD,IAAI,CAAC,SAAS,OAAO,UAAU,UAAU,OAAO,KAAA;CAEhD,MAAM,OAAQ,MAAkC;CAChD,OAAO,OAAO,SAAS,WAAW,OAAO,KAAA;AAC3C;;;;;;;;;;;;;;;;;;;;;;AAuBA,SAAgB,mBAAmB,UAAoC;CACrE,OAAO,SAAS,KAAK,QAAQ;EAC3B,IAAI,IAAI,SAAS,QAAQ,OAAO;EAEhC,MAAM,OAAO,aAAa,IAAI,QAAQ;EACtC,IAAI,CAAC,MAAM,OAAO;EAElB,MAAM,aAAa,KAChB,MAAM,OAAO,CAAC,CACd,KAAK,SAAS,KAAK,MAAM,CAAC,CAC1B,KAAK,IAAI;EAKZ,IAFE,IAAI,MAAM,EAAE,EAAE,SAAS,UACvB,IAAI,MAAM,EAAE,CAAC,SAAS,GAAG,WAAW,OACjB,OAAO;EAE5B,OAAO;GACL,GAAG;GACH,OAAO,CACL;IAAE,MAAM;IAAiB,MAAM,GAAG,WAAW;GAAM,GACnD,GAAI,IAAI,SAAS,CAAC,CACpB;EACF;CACF,CAAC;AACH"}

package/dist/mcp-stdio.node.d.ts

@@ -0,0 +1,2 @@
+import { Experimental_StdioMCPTransport } from "@ai-sdk/mcp/mcp-stdio";
+export { Experimental_StdioMCPTransport };

package/dist/mcp-stdio.node.js

@@ -0,0 +1,2 @@
+import { Experimental_StdioMCPTransport } from "@ai-sdk/mcp/mcp-stdio";
+export { Experimental_StdioMCPTransport };

package/dist/mcp-stdio.unsupported.d.ts

@@ -0,0 +1,7 @@
+import { Experimental_StdioMCPTransport as Experimental_StdioMCPTransport$1 } from "@ai-sdk/mcp/mcp-stdio";
+
+//#region src/mcp-stdio.unsupported.d.ts
+declare const Experimental_StdioMCPTransport: typeof Experimental_StdioMCPTransport$1;
+//#endregion
+export { Experimental_StdioMCPTransport };
+//# sourceMappingURL=mcp-stdio.unsupported.d.ts.map

package/dist/mcp-stdio.unsupported.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"mcp-stdio.unsupported.d.ts","names":[],"sources":["../src/mcp-stdio.unsupported.ts"],"mappings":";;;cAUa,8BAAA,SACuC,gCAAqB"}

package/dist/mcp-stdio.unsupported.js

@@ -0,0 +1,11 @@
+//#region src/mcp-stdio.unsupported.ts
+var UnsupportedStdioMCPTransport = class {
+	constructor() {
+		throw new Error("stdio MCP transport requires a runtime that can spawn a subprocess, such as Node, Bun, or Deno (with --allow-run). Use an HTTP or SSE MCP server config in browser, React Native, edge, or worker runtimes.");
+	}
+};
+const Experimental_StdioMCPTransport = UnsupportedStdioMCPTransport;
+//#endregion
+export { Experimental_StdioMCPTransport };
+
+//# sourceMappingURL=mcp-stdio.unsupported.js.map

package/dist/mcp-stdio.unsupported.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"mcp-stdio.unsupported.js","names":[],"sources":["../src/mcp-stdio.unsupported.ts"],"sourcesContent":["import type { Experimental_StdioMCPTransport as NodeStdioMCPTransport } from \"@ai-sdk/mcp/mcp-stdio\";\n\nclass UnsupportedStdioMCPTransport {\n  constructor() {\n    throw new Error(\n      \"stdio MCP transport requires a runtime that can spawn a subprocess, such as Node, Bun, or Deno (with --allow-run). Use an HTTP or SSE MCP server config in browser, React Native, edge, or worker runtimes.\",\n    );\n  }\n}\n\nexport const Experimental_StdioMCPTransport =\n  UnsupportedStdioMCPTransport as unknown as typeof NodeStdioMCPTransport;\n"],"mappings":";AAEA,IAAM,+BAAN,MAAmC;CACjC,cAAc;EACZ,MAAM,IAAI,MACR,6MACF;CACF;AACF;AAEA,MAAa,iCACX"}

package/dist/ui/use-chat/AssistantChatTransport.js.map

@@ -1 +1 @@
-{"version":3,"file":"AssistantChatTransport.js","names":[],"sources":["../../../src/ui/use-chat/AssistantChatTransport.ts"],"sourcesContent":["import type {\n  AssistantRuntime,\n  ThreadListItemRuntime,\n} from \"@assistant-ui/core\";\nimport {\n  DefaultChatTransport,\n  type HttpChatTransportInitOptions,\n  type UIMessage,\n} from \"ai\";\nimport { toToolsJSONSchema } from \"assistant-stream\";\nimport {\n  RESUMABLE_STREAM_ID_HEADER,\n  type AssistantChatResumableOptions,\n} from \"../resumable\";\n\ntype InitializableThreadListItem = Pick<ThreadListItemRuntime, \"initialize\">;\n\nconst FINISH_MARKER = '\"type\":\"finish\"';\nconst FINISH_BUFFER_LIMIT = 4096;\nconst FINISH_BUFFER_TAIL = 1024;\n\nexport type AssistantChatTransportInitOptions<UI_MESSAGE extends UIMessage> =\n  HttpChatTransportInitOptions<UI_MESSAGE> & {\n    resumable?: AssistantChatResumableOptions;\n  };\n\nexport class AssistantChatTransport<\n  UI_MESSAGE extends UIMessage,\n> extends DefaultChatTransport<UI_MESSAGE> {\n  private runtime: AssistantRuntime | undefined;\n  private getThreadListItem:\n    | (() => InitializableThreadListItem | undefined)\n    | undefined;\n  private readonly resumable: AssistantChatResumableOptions | undefined;\n\n  constructor(initOptions?: AssistantChatTransportInitOptions<UI_MESSAGE>) {\n    const { resumable, ...rest } = initOptions ?? {};\n    const userFetch = rest.fetch;\n    const userPrepareReconnect = rest.prepareReconnectToStreamRequest;\n\n    super({\n      ...rest,\n      ...(resumable && {\n        fetch: wrapFetchWithResumable(resumable, userFetch),\n        prepareReconnectToStreamRequest: wrapPrepareReconnect(\n          resumable,\n          userPrepareReconnect,\n        ),\n      }),\n      prepareSendMessagesRequest: async (options) => {\n        const context = this.runtime?.thread.getModelContext();\n        const threadListItem =\n          this.getThreadListItem?.() ?? this.runtime?.threads.mainItem;\n        const id = (await threadListItem?.initialize())?.remoteId ?? options.id;\n\n        const optionsEx = {\n          ...options,\n          body: {\n            callSettings: context?.callSettings,\n            system: context?.system,\n            config: context?.config,\n            tools: toToolsJSONSchema(context?.tools ?? {}),\n            ...options?.body,\n          },\n        };\n        const preparedRequest =\n          await rest.prepareSendMessagesRequest?.(optionsEx);\n\n        return {\n          ...preparedRequest,\n          body: preparedRequest?.body ?? {\n            ...optionsEx.body,\n            id,\n            messages: options.messages,\n            trigger: options.trigger,\n            messageId: options.messageId,\n            metadata: options.requestMetadata,\n          },\n        };\n      },\n    });\n\n    this.resumable = resumable;\n  }\n\n  setRuntime(runtime: AssistantRuntime) {\n    this.runtime = runtime;\n  }\n\n  getResumableAdapter(): AssistantChatResumableOptions | undefined {\n    return this.resumable;\n  }\n\n  __internal_setGetThreadListItem(\n    getter: () => InitializableThreadListItem | undefined,\n  ) {\n    this.getThreadListItem = getter;\n  }\n}\n\nfunction wrapFetchWithResumable(\n  resumable: AssistantChatResumableOptions,\n  userFetch: HttpChatTransportInitOptions<UIMessage>[\"fetch\"],\n): NonNullable<HttpChatTransportInitOptions<UIMessage>[\"fetch\"]> {\n  const baseFetch: typeof globalThis.fetch = userFetch\n    ? (input, init) => userFetch(input as RequestInfo | URL, init)\n    : globalThis.fetch.bind(globalThis);\n\n  return async (input, init) => {\n    const res = await baseFetch(input, init);\n    const id = res.headers.get(RESUMABLE_STREAM_ID_HEADER);\n    if (id) resumable.storage.setStreamId(id);\n    if (!res.body) return res;\n\n    const detectFinish = resumable.isFinishEvent ?? defaultIsFinishEvent;\n    // a single decoder is required so multi-byte sequences split across\n    // chunks buffer via stream: true rather than getting dropped.\n    const decoder = new TextDecoder();\n    let accumulator = \"\";\n    const tap = new TransformStream<Uint8Array, Uint8Array>({\n      transform(chunk, controller) {\n        controller.enqueue(chunk);\n        accumulator += decoder.decode(chunk, { stream: true });\n        if (detectFinish(chunk, accumulator)) {\n          resumable.storage.clear();\n          accumulator = \"\";\n        } else if (accumulator.length > FINISH_BUFFER_LIMIT) {\n          accumulator = accumulator.slice(-FINISH_BUFFER_TAIL);\n        }\n      },\n    });\n\n    return new Response(res.body.pipeThrough(tap), {\n      status: res.status,\n      statusText: res.statusText,\n      headers: res.headers,\n    });\n  };\n}\n\nfunction defaultIsFinishEvent(_chunk: Uint8Array, accumulator: string) {\n  return accumulator.includes(FINISH_MARKER);\n}\n\nfunction wrapPrepareReconnect(\n  resumable: AssistantChatResumableOptions,\n  userPrepareReconnect: HttpChatTransportInitOptions<UIMessage>[\"prepareReconnectToStreamRequest\"],\n): NonNullable<\n  HttpChatTransportInitOptions<UIMessage>[\"prepareReconnectToStreamRequest\"]\n> {\n  return async (options) => {\n    const streamId = resumable.storage.getStreamId();\n    if (!streamId) {\n      throw new Error(\n        \"AssistantChatTransport: no resumable stream id available; nothing to resume\",\n      );\n    }\n    const api =\n      typeof resumable.resumeApi === \"function\"\n        ? resumable.resumeApi(streamId)\n        : resumable.resumeApi;\n    const userPrepared = await userPrepareReconnect?.({ ...options, api });\n    return {\n      ...userPrepared,\n      api: userPrepared?.api ?? api,\n    };\n  };\n}\n"],"mappings":";;;;AAiBA,MAAM,gBAAgB;AACtB,MAAM,sBAAsB;AAQ5B,IAAa,yBAAb,cAEU,qBAAiC;CACzC;CACA;CAGA;CAEA,YAAY,aAA6D;EACvE,MAAM,EAAE,WAAW,GAAG,SAAS,eAAe,CAAC;EAC/C,MAAM,YAAY,KAAK;EACvB,MAAM,uBAAuB,KAAK;EAElC,MAAM;GACJ,GAAG;GACH,GAAI,aAAa;IACf,OAAO,uBAAuB,WAAW,SAAS;IAClD,iCAAiC,qBAC/B,WACA,oBACF;GACF;GACA,4BAA4B,OAAO,YAAY;IAC7C,MAAM,UAAU,KAAK,SAAS,OAAO,gBAAgB;IAGrD,MAAM,MAAM,OADV,KAAK,oBAAoB,KAAK,KAAK,SAAS,QAAQ,WACpB,WAAW,IAAI,YAAY,QAAQ;IAErE,MAAM,YAAY;KAChB,GAAG;KACH,MAAM;MACJ,cAAc,SAAS;MACvB,QAAQ,SAAS;MACjB,QAAQ,SAAS;MACjB,OAAO,kBAAkB,SAAS,SAAS,CAAC,CAAC;MAC7C,GAAG,SAAS;KACd;IACF;IACA,MAAM,kBACJ,MAAM,KAAK,6BAA6B,SAAS;IAEnD,OAAO;KACL,GAAG;KACH,MAAM,iBAAiB,QAAQ;MAC7B,GAAG,UAAU;MACb;MACA,UAAU,QAAQ;MAClB,SAAS,QAAQ;MACjB,WAAW,QAAQ;MACnB,UAAU,QAAQ;KACpB;IACF;GACF;EACF,CAAC;EAED,KAAK,YAAY;CACnB;CAEA,WAAW,SAA2B;EACpC,KAAK,UAAU;CACjB;CAEA,sBAAiE;EAC/D,OAAO,KAAK;CACd;CAEA,gCACE,QACA;EACA,KAAK,oBAAoB;CAC3B;AACF;AAEA,SAAS,uBACP,WACA,WAC+D;CAC/D,MAAM,YAAqC,aACtC,OAAO,SAAS,UAAU,OAA4B,IAAI,IAC3D,WAAW,MAAM,KAAK,UAAU;CAEpC,OAAO,OAAO,OAAO,SAAS;EAC5B,MAAM,MAAM,MAAM,UAAU,OAAO,IAAI;EACvC,MAAM,KAAK,IAAI,QAAQ,IAAI,0BAA0B;EACrD,IAAI,IAAI,UAAU,QAAQ,YAAY,EAAE;EACxC,IAAI,CAAC,IAAI,MAAM,OAAO;EAEtB,MAAM,eAAe,UAAU,iBAAiB;EAGhD,MAAM,UAAU,IAAI,YAAY;EAChC,IAAI,cAAc;EAClB,MAAM,MAAM,IAAI,gBAAwC,EACtD,UAAU,OAAO,YAAY;GAC3B,WAAW,QAAQ,KAAK;GACxB,eAAe,QAAQ,OAAO,OAAO,EAAE,QAAQ,KAAK,CAAC;GACrD,IAAI,aAAa,OAAO,WAAW,GAAG;IACpC,UAAU,QAAQ,MAAM;IACxB,cAAc;GAChB,OAAO,IAAI,YAAY,SAAS,qBAC9B,cAAc,YAAY,MAAM,KAAmB;EAEvD,EACF,CAAC;EAED,OAAO,IAAI,SAAS,IAAI,KAAK,YAAY,GAAG,GAAG;GAC7C,QAAQ,IAAI;GACZ,YAAY,IAAI;GAChB,SAAS,IAAI;EACf,CAAC;CACH;AACF;AAEA,SAAS,qBAAqB,QAAoB,aAAqB;CACrE,OAAO,YAAY,SAAS,aAAa;AAC3C;AAEA,SAAS,qBACP,WACA,sBAGA;CACA,OAAO,OAAO,YAAY;EACxB,MAAM,WAAW,UAAU,QAAQ,YAAY;EAC/C,IAAI,CAAC,UACH,MAAM,IAAI,MACR,6EACF;EAEF,MAAM,MACJ,OAAO,UAAU,cAAc,aAC3B,UAAU,UAAU,QAAQ,IAC5B,UAAU;EAChB,MAAM,eAAe,MAAM,uBAAuB;GAAE,GAAG;GAAS;EAAI,CAAC;EACrE,OAAO;GACL,GAAG;GACH,KAAK,cAAc,OAAO;EAC5B;CACF;AACF"}
+{"version":3,"file":"AssistantChatTransport.js","names":[],"sources":["../../../src/ui/use-chat/AssistantChatTransport.ts"],"sourcesContent":["import type {\n  AssistantRuntime,\n  ThreadListItemRuntime,\n} from \"@assistant-ui/core\";\nimport {\n  DefaultChatTransport,\n  type HttpChatTransportInitOptions,\n  type UIMessage,\n} from \"ai\";\nimport { toToolsJSONSchema } from \"assistant-stream\";\nimport {\n  RESUMABLE_STREAM_ID_HEADER,\n  type AssistantChatResumableOptions,\n} from \"../resumable\";\n\ntype InitializableThreadListItem = Pick<ThreadListItemRuntime, \"initialize\">;\n\nconst FINISH_MARKER = '\"type\":\"finish\"';\nconst FINISH_BUFFER_LIMIT = 4096;\nconst FINISH_BUFFER_TAIL = 1024;\n\nexport type AssistantChatTransportInitOptions<UI_MESSAGE extends UIMessage> =\n  HttpChatTransportInitOptions<UI_MESSAGE> & {\n    resumable?: AssistantChatResumableOptions;\n  };\n\nexport class AssistantChatTransport<\n  UI_MESSAGE extends UIMessage,\n> extends DefaultChatTransport<UI_MESSAGE> {\n  private runtime: AssistantRuntime | undefined;\n  private getThreadListItem:\n    | (() => InitializableThreadListItem | undefined)\n    | undefined;\n  private readonly resumable: AssistantChatResumableOptions | undefined;\n\n  constructor(initOptions?: AssistantChatTransportInitOptions<UI_MESSAGE>) {\n    const { resumable, ...rest } = initOptions ?? {};\n    const userFetch = rest.fetch;\n    const userPrepareReconnect = rest.prepareReconnectToStreamRequest;\n\n    super({\n      ...rest,\n      ...(resumable && {\n        fetch: wrapFetchWithResumable(resumable, userFetch),\n        prepareReconnectToStreamRequest: wrapPrepareReconnect(\n          resumable,\n          userPrepareReconnect,\n        ),\n      }),\n      prepareSendMessagesRequest: async (options) => {\n        const context = this.runtime?.thread.getModelContext();\n        const threadListItem =\n          this.getThreadListItem?.() ?? this.runtime?.threads.mainItem;\n        const id = (await threadListItem?.initialize())?.remoteId ?? options.id;\n\n        const optionsEx = {\n          ...options,\n          body: {\n            callSettings: context?.callSettings,\n            system: context?.system,\n            config: context?.config,\n            tools: toToolsJSONSchema(context?.tools ?? {}),\n            ...options?.body,\n          },\n        };\n        const preparedRequest =\n          await rest.prepareSendMessagesRequest?.(optionsEx);\n\n        return {\n          ...preparedRequest,\n          body: preparedRequest?.body ?? {\n            ...optionsEx.body,\n            id,\n            messages: options.messages,\n            trigger: options.trigger,\n            messageId: options.messageId,\n            metadata: options.requestMetadata,\n          },\n        };\n      },\n    });\n\n    this.resumable = resumable;\n  }\n\n  setRuntime(runtime: AssistantRuntime) {\n    this.runtime = runtime;\n  }\n\n  getResumableAdapter(): AssistantChatResumableOptions | undefined {\n    return this.resumable;\n  }\n\n  __internal_setGetThreadListItem(\n    getter: () => InitializableThreadListItem | undefined,\n  ) {\n    this.getThreadListItem = getter;\n  }\n}\n\nfunction wrapFetchWithResumable(\n  resumable: AssistantChatResumableOptions,\n  userFetch: HttpChatTransportInitOptions<UIMessage>[\"fetch\"],\n): NonNullable<HttpChatTransportInitOptions<UIMessage>[\"fetch\"]> {\n  const baseFetch: typeof globalThis.fetch = userFetch\n    ? (input, init) => userFetch(input as RequestInfo | URL, init)\n    : globalThis.fetch.bind(globalThis);\n\n  return async (input, init) => {\n    const res = await baseFetch(input, init);\n    const id = res.headers.get(RESUMABLE_STREAM_ID_HEADER);\n    if (id) resumable.storage.setStreamId(id);\n    if (!res.body) return res;\n\n    const detectFinish = resumable.isFinishEvent ?? defaultIsFinishEvent;\n    // a single decoder is required so multi-byte sequences split across\n    // chunks buffer via stream: true rather than getting dropped.\n    const decoder = new TextDecoder();\n    let accumulator = \"\";\n    const tap = new TransformStream<Uint8Array, Uint8Array>({\n      transform(chunk, controller) {\n        controller.enqueue(chunk);\n        accumulator += decoder.decode(chunk, { stream: true });\n        if (detectFinish(chunk, accumulator)) {\n          resumable.storage.clear();\n          accumulator = \"\";\n        } else if (accumulator.length > FINISH_BUFFER_LIMIT) {\n          accumulator = accumulator.slice(-FINISH_BUFFER_TAIL);\n        }\n      },\n    });\n\n    return new Response(res.body.pipeThrough(tap), {\n      status: res.status,\n      statusText: res.statusText,\n      headers: res.headers,\n    });\n  };\n}\n\nfunction defaultIsFinishEvent(_chunk: Uint8Array, accumulator: string) {\n  return accumulator.includes(FINISH_MARKER);\n}\n\nfunction wrapPrepareReconnect(\n  resumable: AssistantChatResumableOptions,\n  userPrepareReconnect: HttpChatTransportInitOptions<UIMessage>[\"prepareReconnectToStreamRequest\"],\n): NonNullable<\n  HttpChatTransportInitOptions<UIMessage>[\"prepareReconnectToStreamRequest\"]\n> {\n  return async (options) => {\n    const streamId = resumable.storage.getStreamId();\n    if (!streamId) {\n      throw new Error(\n        \"AssistantChatTransport: no resumable stream id available; nothing to resume\",\n      );\n    }\n    const api =\n      typeof resumable.resumeApi === \"function\"\n        ? resumable.resumeApi(streamId)\n        : resumable.resumeApi;\n    const userPrepared = await userPrepareReconnect?.({ ...options, api });\n    return {\n      ...userPrepared,\n      api: userPrepared?.api ?? api,\n    };\n  };\n}\n"],"mappings":";;;;AAiBA,MAAM,gBAAgB;AACtB,MAAM,sBAAsB;AAQ5B,IAAa,yBAAb,cAEU,qBAAiC;CACzC;CACA;CAGA;CAEA,YAAY,aAA6D;EACvE,MAAM,EAAE,WAAW,GAAG,SAAS,eAAe,CAAC;EAC/C,MAAM,YAAY,KAAK;EACvB,MAAM,uBAAuB,KAAK;EAElC,MAAM;GACJ,GAAG;GACH,GAAI,aAAa;IACf,OAAO,uBAAuB,WAAW,SAAS;IAClD,iCAAiC,qBAC/B,WACA,oBACF;GACF;GACA,4BAA4B,OAAO,YAAY;IAC7C,MAAM,UAAU,KAAK,SAAS,OAAO,gBAAgB;IAGrD,MAAM,MAAM,OADV,KAAK,oBAAoB,KAAK,KAAK,SAAS,QAAQ,SAAA,EACpB,WAAW,EAAA,EAAI,YAAY,QAAQ;IAErE,MAAM,YAAY;KAChB,GAAG;KACH,MAAM;MACJ,cAAc,SAAS;MACvB,QAAQ,SAAS;MACjB,QAAQ,SAAS;MACjB,OAAO,kBAAkB,SAAS,SAAS,CAAC,CAAC;MAC7C,GAAG,SAAS;KACd;IACF;IACA,MAAM,kBACJ,MAAM,KAAK,6BAA6B,SAAS;IAEnD,OAAO;KACL,GAAG;KACH,MAAM,iBAAiB,QAAQ;MAC7B,GAAG,UAAU;MACb;MACA,UAAU,QAAQ;MAClB,SAAS,QAAQ;MACjB,WAAW,QAAQ;MACnB,UAAU,QAAQ;KACpB;IACF;GACF;EACF,CAAC;EAED,KAAK,YAAY;CACnB;CAEA,WAAW,SAA2B;EACpC,KAAK,UAAU;CACjB;CAEA,sBAAiE;EAC/D,OAAO,KAAK;CACd;CAEA,gCACE,QACA;EACA,KAAK,oBAAoB;CAC3B;AACF;AAEA,SAAS,uBACP,WACA,WAC+D;CAC/D,MAAM,YAAqC,aACtC,OAAO,SAAS,UAAU,OAA4B,IAAI,IAC3D,WAAW,MAAM,KAAK,UAAU;CAEpC,OAAO,OAAO,OAAO,SAAS;EAC5B,MAAM,MAAM,MAAM,UAAU,OAAO,IAAI;EACvC,MAAM,KAAK,IAAI,QAAQ,IAAI,0BAA0B;EACrD,IAAI,IAAI,UAAU,QAAQ,YAAY,EAAE;EACxC,IAAI,CAAC,IAAI,MAAM,OAAO;EAEtB,MAAM,eAAe,UAAU,iBAAiB;EAGhD,MAAM,UAAU,IAAI,YAAY;EAChC,IAAI,cAAc;EAClB,MAAM,MAAM,IAAI,gBAAwC,EACtD,UAAU,OAAO,YAAY;GAC3B,WAAW,QAAQ,KAAK;GACxB,eAAe,QAAQ,OAAO,OAAO,EAAE,QAAQ,KAAK,CAAC;GACrD,IAAI,aAAa,OAAO,WAAW,GAAG;IACpC,UAAU,QAAQ,MAAM;IACxB,cAAc;GAChB,OAAO,IAAI,YAAY,SAAS,qBAC9B,cAAc,YAAY,MAAM,KAAmB;EAEvD,EACF,CAAC;EAED,OAAO,IAAI,SAAS,IAAI,KAAK,YAAY,GAAG,GAAG;GAC7C,QAAQ,IAAI;GACZ,YAAY,IAAI;GAChB,SAAS,IAAI;EACf,CAAC;CACH;AACF;AAEA,SAAS,qBAAqB,QAAoB,aAAqB;CACrE,OAAO,YAAY,SAAS,aAAa;AAC3C;AAEA,SAAS,qBACP,WACA,sBAGA;CACA,OAAO,OAAO,YAAY;EACxB,MAAM,WAAW,UAAU,QAAQ,YAAY;EAC/C,IAAI,CAAC,UACH,MAAM,IAAI,MACR,6EACF;EAEF,MAAM,MACJ,OAAO,UAAU,cAAc,aAC3B,UAAU,UAAU,QAAQ,IAC5B,UAAU;EAChB,MAAM,eAAe,MAAM,uBAAuB;GAAE,GAAG;GAAS;EAAI,CAAC;EACrE,OAAO;GACL,GAAG;GACH,KAAK,cAAc,OAAO;EAC5B;CACF;AACF"}

package/dist/ui/use-chat/useAISDKRuntime.d.ts

@@ -1,3 +1,4 @@
+import { JoinStrategy } from "@assistant-ui/core/react";
 import { AppendMessage, AssistantRuntime, ExternalStoreAdapter, ExternalStoreSharedOptions, ThreadHistoryAdapter } from "@assistant-ui/core";
 import { CreateUIMessage, UIMessage, useChat } from "@ai-sdk/react";
 
@@ -25,6 +26,15 @@ type AISDKRuntimeAdapter = ExternalStoreSharedOptions & {
    * (for example, an SSE reconnect endpoint keyed by turn id).
    */
   onResume?: ExternalStoreAdapter["onResume"];
+  /**
+   * How consecutive assistant messages are rendered.
+   *
+   * `"concat-content"` (the default) merges them into a single thread message.
+   * `"none"` keeps each assistant message as its own thread message, which is
+   * useful when a backend persists proactive or consecutive assistant messages
+   * as separate entries.
+   */
+  joinStrategy?: JoinStrategy | undefined;
 };
 declare const useAISDKRuntime: <UI_MESSAGE extends UIMessage = UIMessage>(chatHelpers: ReturnType<typeof useChat<UI_MESSAGE>>, adapter?: AISDKRuntimeAdapter) => AssistantRuntime;
 //#endregion

package/dist/ui/use-chat/useAISDKRuntime.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"useAISDKRuntime.d.ts","names":[],"sources":["../../../src/ui/use-chat/useAISDKRuntime.ts"],"mappings":";;;;KA4CY,6BAAA,uBACS,SAAA,GAAY,SAAA,EAE/B,OAAA,EAAS,aAAA,KACN,eAAA,CAAgB,UAAA;AAAA,KAYT,mBAAA,GAAsB,0BAAA;EAChC,QAAA,IACK,WAAA,CAAY,oBAAA;IACX,OAAA,GAAU,oBAAA;EAAA;EAGhB,eAAA,GAAkB,6BAAA;EArBa;;;;;;;;EA8B/B,4BAAA;EA5BS;;;;;AACoB;AAY/B;EAuBE,QAAA,GAAW,oBAAA;AAAA;AAAA,cAGA,eAAA,sBAAsC,SAAA,GAAY,SAAA,EAC7D,WAAA,EAAa,UAAA,QAAkB,OAAA,CAAQ,UAAA,IACvC,OAAA,GAAS,mBAAA,KAAwB,gBAAA"}
+{"version":3,"file":"useAISDKRuntime.d.ts","names":[],"sources":["../../../src/ui/use-chat/useAISDKRuntime.ts"],"mappings":";;;;;KA6CY,6BAAA,uBACS,SAAA,GAAY,SAAA,EAE/B,OAAA,EAAS,aAAA,KACN,eAAA,CAAgB,UAAA;AAAA,KAYT,mBAAA,GAAsB,0BAAA;EAChC,QAAA,IACK,WAAA,CAAY,oBAAA;IACX,OAAA,GAAU,oBAAA;EAAA;EAGhB,eAAA,GAAkB,6BAAA;EArBa;;;;;;;;EA8B/B,4BAAA;EA5BS;;;;;AACoB;AAY/B;EAuBE,QAAA,GAAW,oBAAA;;;;;;;;;EASX,YAAA,GAAe,YAAA;AAAA;AAAA,cAGJ,eAAA,sBAAsC,SAAA,GAAY,SAAA,EAC7D,WAAA,EAAa,UAAA,QAAkB,OAAA,CAAQ,UAAA,IACvC,OAAA,GAAS,mBAAA,KAAwB,gBAAA"}

package/dist/ui/use-chat/useAISDKRuntime.js

@@ -1,15 +1,15 @@
 "use client";
-import { wrapModelContentEnvelope } from "../../modelContentEnvelope.js";
 import { sliceMessagesUntil } from "../utils/sliceMessagesUntil.js";
 import { toCreateMessage } from "../utils/toCreateMessage.js";
 import { vercelAttachmentAdapter } from "../utils/vercelAttachmentAdapter.js";
 import { getVercelAIMessages } from "../getVercelAIMessages.js";
+import { wrapModelContentEnvelope } from "../../modelContentEnvelope.js";
 import { AISDKMessageConverter } from "../utils/convertMessage.js";
 import { aiSDKV6FormatAdapter } from "../adapters/aiSDKFormatAdapter.js";
 import { toExportedMessageRepository, useExternalHistory } from "./useExternalHistory.js";
 import { useStreamingTiming } from "./useStreamingTiming.js";
-import { generateId, isToolUIPart } from "ai";
 import { useMemo, useRef, useState } from "react";
+import { generateId, isToolUIPart } from "ai";
 import { useExternalStoreRuntime, useRuntimeAdapters } from "@assistant-ui/core/react";
 import { getExternalStoreMessages, pickExternalStoreSharedOptions } from "@assistant-ui/core";
 //#region src/ui/use-chat/useAISDKRuntime.ts
@@ -19,7 +19,7 @@ const toUIMessage = (createMessage, fallbackRole) => ({
 	role: createMessage.role ?? fallbackRole
 });
 const useAISDKRuntime = (chatHelpers, adapter = {}) => {
-	const { adapters, toCreateMessage: customToCreateMessage, cancelPendingToolCallsOnSend = true, onResume } = adapter;
+	const { adapters, toCreateMessage: customToCreateMessage, cancelPendingToolCallsOnSend = true, onResume, joinStrategy } = adapter;
 	const contextAdapters = useRuntimeAdapters();
 	const [toolStatuses, setToolStatuses] = useState({});
 	const toolArgsKeyOrderCacheRef = useRef(/* @__PURE__ */ new Map());
@@ -34,6 +34,7 @@ const useAISDKRuntime = (chatHelpers, adapter = {}) => {
 	const messages = AISDKMessageConverter.useThreadMessages({
 		isRunning,
 		messages: chatHelpers.messages,
+		joinStrategy,
 		metadata: useMemo(() => ({
 			toolStatuses,
 			messageTiming,

package/dist/ui/use-chat/useAISDKRuntime.js.map

@@ -1 +1 @@
-{"version":3,"file":"useAISDKRuntime.js","names":[],"sources":["../../../src/ui/use-chat/useAISDKRuntime.ts"],"sourcesContent":["\"use client\";\n\nimport { useMemo, useRef, useState } from \"react\";\nimport type { UIMessage, useChat, CreateUIMessage } from \"@ai-sdk/react\";\nimport { isToolUIPart, generateId } from \"ai\";\nimport {\n  useExternalStoreRuntime,\n  useRuntimeAdapters,\n} from \"@assistant-ui/core/react\";\nimport type { ToolExecutionStatus } from \"@assistant-ui/core\";\nimport type {\n  ExternalStoreAdapter,\n  ExternalStoreSharedOptions,\n  ThreadHistoryAdapter,\n  AssistantRuntime,\n  ThreadMessage,\n  MessageFormatAdapter,\n  MessageFormatItem,\n  MessageFormatRepository,\n  AppendMessage,\n  RunConfig,\n  McpAppMetadata,\n} from \"@assistant-ui/core\";\nimport {\n  getExternalStoreMessages,\n  pickExternalStoreSharedOptions,\n} from \"@assistant-ui/core\";\nimport type { ReadonlyJSONObject } from \"assistant-stream/utils\";\nimport { sliceMessagesUntil } from \"../utils/sliceMessagesUntil\";\nimport { toCreateMessage } from \"../utils/toCreateMessage\";\nimport { vercelAttachmentAdapter } from \"../utils/vercelAttachmentAdapter\";\nimport { getVercelAIMessages } from \"../getVercelAIMessages\";\nimport { AISDKMessageConverter } from \"../utils/convertMessage\";\nimport { wrapModelContentEnvelope } from \"../../modelContentEnvelope\";\nimport {\n  type AISDKStorageFormat,\n  aiSDKV6FormatAdapter,\n} from \"../adapters/aiSDKFormatAdapter\";\nimport {\n  useExternalHistory,\n  toExportedMessageRepository,\n} from \"./useExternalHistory\";\nimport { useStreamingTiming } from \"./useStreamingTiming\";\n\nexport type CustomToCreateMessageFunction = <\n  UI_MESSAGE extends UIMessage = UIMessage,\n>(\n  message: AppendMessage,\n) => CreateUIMessage<UI_MESSAGE>;\n\nconst toUIMessage = <UI_MESSAGE extends UIMessage>(\n  createMessage: CreateUIMessage<UI_MESSAGE>,\n  fallbackRole: UI_MESSAGE[\"role\"],\n): UI_MESSAGE =>\n  ({\n    ...createMessage,\n    id: createMessage.id ?? generateId(),\n    role: createMessage.role ?? fallbackRole,\n  }) as UI_MESSAGE;\n\nexport type AISDKRuntimeAdapter = ExternalStoreSharedOptions & {\n  adapters?:\n    | (NonNullable<ExternalStoreAdapter[\"adapters\"]> & {\n        history?: ThreadHistoryAdapter | undefined;\n      })\n    | undefined;\n  toCreateMessage?: CustomToCreateMessageFunction;\n  /**\n   * Whether to automatically cancel pending interactive tool calls when the user sends a new message.\n   *\n   * When enabled (default), the pending tool calls will be marked as failed with an error message\n   * indicating the user cancelled the tool call by sending a new message.\n   *\n   * @default true\n   */\n  cancelPendingToolCallsOnSend?: boolean | undefined;\n  /**\n   * Called when `runtime.thread.resumeRun(config)` is invoked.\n   *\n   * When omitted, `resumeRun` throws `\"Runtime does not support resuming runs.\"`.\n   * Provide this to bridge resume invocations into a custom replay channel\n   * (for example, an SSE reconnect endpoint keyed by turn id).\n   */\n  onResume?: ExternalStoreAdapter[\"onResume\"];\n};\n\nexport const useAISDKRuntime = <UI_MESSAGE extends UIMessage = UIMessage>(\n  chatHelpers: ReturnType<typeof useChat<UI_MESSAGE>>,\n  adapter: AISDKRuntimeAdapter = {},\n) => {\n  const {\n    adapters,\n    toCreateMessage: customToCreateMessage,\n    cancelPendingToolCallsOnSend = true,\n    onResume,\n  } = adapter;\n  const contextAdapters = useRuntimeAdapters();\n  const [toolStatuses, setToolStatuses] = useState<\n    Record<string, ToolExecutionStatus>\n  >({});\n  const toolArgsKeyOrderCacheRef = useRef<Map<string, Map<string, string[]>>>(\n    new Map(),\n  );\n  const toolLastInputCacheRef = useRef<Map<string, ReadonlyJSONObject>>(\n    new Map(),\n  );\n  const mcpAppMetadataCacheRef = useRef<Map<string, McpAppMetadata>>(new Map());\n  const lastRunConfigRef = useRef<RunConfig | undefined>(undefined);\n\n  const hasExecutingTools = Object.values(toolStatuses).some(\n    (s) => s?.type === \"executing\",\n  );\n  const isRunning =\n    chatHelpers.status === \"submitted\" ||\n    chatHelpers.status === \"streaming\" ||\n    hasExecutingTools;\n\n  const messageTiming = useStreamingTiming(chatHelpers.messages, isRunning);\n\n  // Flag the streaming message optimistic: its id can be swapped for a server\n  // id mid-run, and the repository then drops the orphaned pre-swap id (#4037).\n  const lastMessage = chatHelpers.messages.at(-1);\n  const optimisticMessageId =\n    isRunning && lastMessage?.role === \"assistant\" ? lastMessage.id : undefined;\n\n  const messages = AISDKMessageConverter.useThreadMessages({\n    isRunning,\n    messages: chatHelpers.messages,\n    metadata: useMemo(\n      () => ({\n        toolStatuses,\n        messageTiming,\n        toolArgsKeyOrderCache: toolArgsKeyOrderCacheRef.current,\n        toolLastInputCache: toolLastInputCacheRef.current,\n        mcpAppMetadataCache: mcpAppMetadataCacheRef.current,\n        ...(optimisticMessageId && { optimisticMessageId }),\n        ...(chatHelpers.error && { error: chatHelpers.error.message }),\n      }),\n      [toolStatuses, messageTiming, optimisticMessageId, chatHelpers.error],\n    ),\n  });\n\n  const [runtimeRef] = useState(() => ({\n    get current(): AssistantRuntime {\n      return runtime;\n    },\n  }));\n\n  const isLoading = useExternalHistory(\n    runtimeRef,\n    adapters?.history ?? contextAdapters?.history,\n    AISDKMessageConverter.toThreadMessages as (\n      messages: UI_MESSAGE[],\n    ) => ThreadMessage[],\n    aiSDKV6FormatAdapter as MessageFormatAdapter<\n      UI_MESSAGE,\n      AISDKStorageFormat\n    >,\n    (messages) => {\n      chatHelpers.setMessages(messages);\n    },\n  );\n\n  const completePendingToolCalls = async () => {\n    if (!cancelPendingToolCallsOnSend) return;\n\n    // The runtime auto-aborts in-flight tool invocations when a new run\n    // is dispatched (append() / startRun()). All we need to do here is\n    // mark any tool without a result as cancelled in the UI message list.\n\n    // Mark any tool without a result as cancelled (uses setMessages to avoid triggering sendAutomaticallyWhen)\n    chatHelpers.setMessages((messages) => {\n      const lastMessage = messages.at(-1);\n      if (lastMessage?.role !== \"assistant\") return messages;\n\n      let hasChanges = false;\n      const parts = lastMessage.parts?.map((part) => {\n        if (!isToolUIPart(part)) return part;\n        if (part.state === \"output-available\" || part.state === \"output-error\")\n          return part;\n\n        hasChanges = true;\n        const { approval: _approval, ...rest } = part;\n        return {\n          ...rest,\n          state: \"output-error\" as const,\n          errorText: \"User cancelled tool call by sending a new message.\",\n        };\n      });\n\n      if (!hasChanges) return messages;\n      return [...messages.slice(0, -1), { ...lastMessage, parts }];\n    });\n  };\n\n  const runtime = useExternalStoreRuntime({\n    isRunning,\n    messages,\n    unstable_enableToolInvocations: true,\n    setToolStatuses,\n    setMessages: (messages) =>\n      chatHelpers.setMessages(\n        messages\n          .map(getVercelAIMessages<UI_MESSAGE>)\n          .filter(Boolean)\n          .flat(),\n      ),\n    onImport: (messages) =>\n      chatHelpers.setMessages(\n        messages\n          .map(getVercelAIMessages<UI_MESSAGE>)\n          .filter(Boolean)\n          .flat(),\n      ),\n    onExportExternalState: (): MessageFormatRepository<UI_MESSAGE> => {\n      const exported = runtimeRef.current.thread.export();\n\n      const expandedMessages: MessageFormatItem<UI_MESSAGE>[] = [];\n      const lastInnerIdMap = new Map<string, string>();\n\n      for (const item of exported.messages) {\n        const innerMessages = getExternalStoreMessages<UI_MESSAGE>(\n          item.message,\n        );\n        let parentId =\n          item.parentId != null\n            ? (lastInnerIdMap.get(item.parentId) ?? item.parentId)\n            : null;\n        for (const innerMessage of innerMessages) {\n          expandedMessages.push({ parentId, message: innerMessage });\n          parentId = aiSDKV6FormatAdapter.getId(innerMessage as UIMessage);\n        }\n        if (innerMessages.length > 0) {\n          lastInnerIdMap.set(\n            item.message.id,\n            aiSDKV6FormatAdapter.getId(\n              innerMessages[innerMessages.length - 1]! as UIMessage,\n            ),\n          );\n        }\n      }\n\n      const result: MessageFormatRepository<UI_MESSAGE> = {\n        messages: expandedMessages,\n      };\n\n      if (exported.headId != null) {\n        result.headId = lastInnerIdMap.get(exported.headId) ?? exported.headId;\n      }\n\n      return result;\n    },\n    onLoadExternalState: (repo: MessageFormatRepository<UI_MESSAGE>) => {\n      // Convert MessageFormatRepository to ExportedMessageRepository\n      const exportedRepo = toExportedMessageRepository(\n        AISDKMessageConverter.toThreadMessages,\n        repo,\n      );\n\n      // Import into the thread's MessageRepository\n      runtimeRef.current.thread.import(exportedRepo);\n    },\n    onCancel: async () => {\n      chatHelpers.stop();\n    },\n    onNew: async (message) => {\n      const createMessage = (\n        customToCreateMessage ?? toCreateMessage\n      )<UI_MESSAGE>(message);\n\n      if (!(message.startRun ?? message.role === \"user\")) {\n        chatHelpers.setMessages((current) => [\n          ...current,\n          toUIMessage<UI_MESSAGE>(createMessage, message.role),\n        ]);\n        return;\n      }\n\n      lastRunConfigRef.current = message.runConfig;\n      await completePendingToolCalls();\n      await chatHelpers.sendMessage(createMessage, {\n        metadata: message.runConfig,\n      });\n    },\n    onEdit: async (message) => {\n      const createMessage = (\n        customToCreateMessage ?? toCreateMessage\n      )<UI_MESSAGE>(message);\n\n      if (!(message.startRun ?? message.role === \"user\")) {\n        chatHelpers.setMessages((current) => [\n          ...sliceMessagesUntil(current, message.parentId),\n          toUIMessage<UI_MESSAGE>(createMessage, message.role),\n        ]);\n        return;\n      }\n\n      lastRunConfigRef.current = message.runConfig;\n      chatHelpers.setMessages((current) =>\n        sliceMessagesUntil(current, message.parentId),\n      );\n      await chatHelpers.sendMessage(createMessage, {\n        metadata: message.runConfig,\n      });\n    },\n    onReload: async (parentId: string | null, config) => {\n      lastRunConfigRef.current = config.runConfig;\n      const newMessages = sliceMessagesUntil(chatHelpers.messages, parentId);\n      chatHelpers.setMessages(newMessages);\n\n      await chatHelpers.regenerate({ metadata: config.runConfig });\n    },\n    onAddToolResult: ({\n      toolCallId,\n      toolName,\n      result,\n      isError,\n      modelContent,\n    }) => {\n      const options = { metadata: lastRunConfigRef.current };\n      if (isError) {\n        chatHelpers.addToolOutput({\n          state: \"output-error\",\n          tool: toolName ?? toolCallId,\n          toolCallId,\n          errorText:\n            typeof result === \"string\" ? result : JSON.stringify(result),\n          options,\n        });\n      } else {\n        const output =\n          modelContent !== undefined\n            ? wrapModelContentEnvelope(result, modelContent)\n            : result;\n        chatHelpers.addToolResult({\n          tool: toolName,\n          toolCallId,\n          output,\n          options,\n        });\n      }\n    },\n    onRespondToToolApproval: ({ approvalId, approved, reason }) => {\n      void chatHelpers.addToolApprovalResponse({\n        id: approvalId,\n        approved,\n        ...(reason != null && { reason }),\n        options: { metadata: lastRunConfigRef.current },\n      });\n    },\n    ...pickExternalStoreSharedOptions(adapter),\n    ...(onResume && { onResume }),\n    adapters: {\n      attachments: vercelAttachmentAdapter,\n      ...contextAdapters,\n      ...adapters,\n    },\n    isLoading,\n  });\n\n  return runtime;\n};\n"],"mappings":";;;;;;;;;;;;;;;AAkDA,MAAM,eACJ,eACA,kBAEC;CACC,GAAG;CACH,IAAI,cAAc,MAAM,WAAW;CACnC,MAAM,cAAc,QAAQ;AAC9B;AA4BF,MAAa,mBACX,aACA,UAA+B,CAAC,MAC7B;CACH,MAAM,EACJ,UACA,iBAAiB,uBACjB,+BAA+B,MAC/B,aACE;CACJ,MAAM,kBAAkB,mBAAmB;CAC3C,MAAM,CAAC,cAAc,mBAAmB,SAEtC,CAAC,CAAC;CACJ,MAAM,2BAA2B,uBAC/B,IAAI,IAAI,CACV;CACA,MAAM,wBAAwB,uBAC5B,IAAI,IAAI,CACV;CACA,MAAM,yBAAyB,uBAAoC,IAAI,IAAI,CAAC;CAC5E,MAAM,mBAAmB,OAA8B,KAAA,CAAS;CAEhE,MAAM,oBAAoB,OAAO,OAAO,YAAY,EAAE,MACnD,MAAM,GAAG,SAAS,WACrB;CACA,MAAM,YACJ,YAAY,WAAW,eACvB,YAAY,WAAW,eACvB;CAEF,MAAM,gBAAgB,mBAAmB,YAAY,UAAU,SAAS;CAIxE,MAAM,cAAc,YAAY,SAAS,GAAG,EAAE;CAC9C,MAAM,sBACJ,aAAa,aAAa,SAAS,cAAc,YAAY,KAAK,KAAA;CAEpE,MAAM,WAAW,sBAAsB,kBAAkB;EACvD;EACA,UAAU,YAAY;EACtB,UAAU,eACD;GACL;GACA;GACA,uBAAuB,yBAAyB;GAChD,oBAAoB,sBAAsB;GAC1C,qBAAqB,uBAAuB;GAC5C,GAAI,uBAAuB,EAAE,oBAAoB;GACjD,GAAI,YAAY,SAAS,EAAE,OAAO,YAAY,MAAM,QAAQ;EAC9D,IACA;GAAC;GAAc;GAAe;GAAqB,YAAY;EAAK,CACtE;CACF,CAAC;CAED,MAAM,CAAC,cAAc,gBAAgB,EACnC,IAAI,UAA4B;EAC9B,OAAO;CACT,EACF,EAAE;CAEF,MAAM,YAAY,mBAChB,YACA,UAAU,WAAW,iBAAiB,SACtC,sBAAsB,kBAGtB,uBAIC,aAAa;EACZ,YAAY,YAAY,QAAQ;CAClC,CACF;CAEA,MAAM,2BAA2B,YAAY;EAC3C,IAAI,CAAC,8BAA8B;EAOnC,YAAY,aAAa,aAAa;GACpC,MAAM,cAAc,SAAS,GAAG,EAAE;GAClC,IAAI,aAAa,SAAS,aAAa,OAAO;GAE9C,IAAI,aAAa;GACjB,MAAM,QAAQ,YAAY,OAAO,KAAK,SAAS;IAC7C,IAAI,CAAC,aAAa,IAAI,GAAG,OAAO;IAChC,IAAI,KAAK,UAAU,sBAAsB,KAAK,UAAU,gBACtD,OAAO;IAET,aAAa;IACb,MAAM,EAAE,UAAU,WAAW,GAAG,SAAS;IACzC,OAAO;KACL,GAAG;KACH,OAAO;KACP,WAAW;IACb;GACF,CAAC;GAED,IAAI,CAAC,YAAY,OAAO;GACxB,OAAO,CAAC,GAAG,SAAS,MAAM,GAAG,EAAE,GAAG;IAAE,GAAG;IAAa;GAAM,CAAC;EAC7D,CAAC;CACH;CAEA,MAAM,UAAU,wBAAwB;EACtC;EACA;EACA,gCAAgC;EAChC;EACA,cAAc,aACZ,YAAY,YACV,SACG,IAAI,mBAA+B,EACnC,OAAO,OAAO,EACd,KAAK,CACV;EACF,WAAW,aACT,YAAY,YACV,SACG,IAAI,mBAA+B,EACnC,OAAO,OAAO,EACd,KAAK,CACV;EACF,6BAAkE;GAChE,MAAM,WAAW,WAAW,QAAQ,OAAO,OAAO;GAElD,MAAM,mBAAoD,CAAC;GAC3D,MAAM,iCAAiB,IAAI,IAAoB;GAE/C,KAAK,MAAM,QAAQ,SAAS,UAAU;IACpC,MAAM,gBAAgB,yBACpB,KAAK,OACP;IACA,IAAI,WACF,KAAK,YAAY,OACZ,eAAe,IAAI,KAAK,QAAQ,KAAK,KAAK,WAC3C;IACN,KAAK,MAAM,gBAAgB,eAAe;KACxC,iBAAiB,KAAK;MAAE;MAAU,SAAS;KAAa,CAAC;KACzD,WAAW,qBAAqB,MAAM,YAAyB;IACjE;IACA,IAAI,cAAc,SAAS,GACzB,eAAe,IACb,KAAK,QAAQ,IACb,qBAAqB,MACnB,cAAc,cAAc,SAAS,EACvC,CACF;GAEJ;GAEA,MAAM,SAA8C,EAClD,UAAU,iBACZ;GAEA,IAAI,SAAS,UAAU,MACrB,OAAO,SAAS,eAAe,IAAI,SAAS,MAAM,KAAK,SAAS;GAGlE,OAAO;EACT;EACA,sBAAsB,SAA8C;GAElE,MAAM,eAAe,4BACnB,sBAAsB,kBACtB,IACF;GAGA,WAAW,QAAQ,OAAO,OAAO,YAAY;EAC/C;EACA,UAAU,YAAY;GACpB,YAAY,KAAK;EACnB;EACA,OAAO,OAAO,YAAY;GACxB,MAAM,iBACJ,yBAAyB,iBACb,OAAO;GAErB,IAAI,EAAE,QAAQ,YAAY,QAAQ,SAAS,SAAS;IAClD,YAAY,aAAa,YAAY,CACnC,GAAG,SACH,YAAwB,eAAe,QAAQ,IAAI,CACrD,CAAC;IACD;GACF;GAEA,iBAAiB,UAAU,QAAQ;GACnC,MAAM,yBAAyB;GAC/B,MAAM,YAAY,YAAY,eAAe,EAC3C,UAAU,QAAQ,UACpB,CAAC;EACH;EACA,QAAQ,OAAO,YAAY;GACzB,MAAM,iBACJ,yBAAyB,iBACb,OAAO;GAErB,IAAI,EAAE,QAAQ,YAAY,QAAQ,SAAS,SAAS;IAClD,YAAY,aAAa,YAAY,CACnC,GAAG,mBAAmB,SAAS,QAAQ,QAAQ,GAC/C,YAAwB,eAAe,QAAQ,IAAI,CACrD,CAAC;IACD;GACF;GAEA,iBAAiB,UAAU,QAAQ;GACnC,YAAY,aAAa,YACvB,mBAAmB,SAAS,QAAQ,QAAQ,CAC9C;GACA,MAAM,YAAY,YAAY,eAAe,EAC3C,UAAU,QAAQ,UACpB,CAAC;EACH;EACA,UAAU,OAAO,UAAyB,WAAW;GACnD,iBAAiB,UAAU,OAAO;GAClC,MAAM,cAAc,mBAAmB,YAAY,UAAU,QAAQ;GACrE,YAAY,YAAY,WAAW;GAEnC,MAAM,YAAY,WAAW,EAAE,UAAU,OAAO,UAAU,CAAC;EAC7D;EACA,kBAAkB,EAChB,YACA,UACA,QACA,SACA,mBACI;GACJ,MAAM,UAAU,EAAE,UAAU,iBAAiB,QAAQ;GACrD,IAAI,SACF,YAAY,cAAc;IACxB,OAAO;IACP,MAAM,YAAY;IAClB;IACA,WACE,OAAO,WAAW,WAAW,SAAS,KAAK,UAAU,MAAM;IAC7D;GACF,CAAC;QACI;IACL,MAAM,SACJ,iBAAiB,KAAA,IACb,yBAAyB,QAAQ,YAAY,IAC7C;IACN,YAAY,cAAc;KACxB,MAAM;KACN;KACA;KACA;IACF,CAAC;GACH;EACF;EACA,0BAA0B,EAAE,YAAY,UAAU,aAAa;GAC7D,YAAiB,wBAAwB;IACvC,IAAI;IACJ;IACA,GAAI,UAAU,QAAQ,EAAE,OAAO;IAC/B,SAAS,EAAE,UAAU,iBAAiB,QAAQ;GAChD,CAAC;EACH;EACA,GAAG,+BAA+B,OAAO;EACzC,GAAI,YAAY,EAAE,SAAS;EAC3B,UAAU;GACR,aAAa;GACb,GAAG;GACH,GAAG;EACL;EACA;CACF,CAAC;CAED,OAAO;AACT"}
+{"version":3,"file":"useAISDKRuntime.js","names":[],"sources":["../../../src/ui/use-chat/useAISDKRuntime.ts"],"sourcesContent":["\"use client\";\n\nimport { useMemo, useRef, useState } from \"react\";\nimport type { UIMessage, useChat, CreateUIMessage } from \"@ai-sdk/react\";\nimport { isToolUIPart, generateId } from \"ai\";\nimport {\n  useExternalStoreRuntime,\n  useRuntimeAdapters,\n  type JoinStrategy,\n} from \"@assistant-ui/core/react\";\nimport type { ToolExecutionStatus } from \"@assistant-ui/core\";\nimport type {\n  ExternalStoreAdapter,\n  ExternalStoreSharedOptions,\n  ThreadHistoryAdapter,\n  AssistantRuntime,\n  ThreadMessage,\n  MessageFormatAdapter,\n  MessageFormatItem,\n  MessageFormatRepository,\n  AppendMessage,\n  RunConfig,\n  McpAppMetadata,\n} from \"@assistant-ui/core\";\nimport {\n  getExternalStoreMessages,\n  pickExternalStoreSharedOptions,\n} from \"@assistant-ui/core\";\nimport type { ReadonlyJSONObject } from \"assistant-stream/utils\";\nimport { sliceMessagesUntil } from \"../utils/sliceMessagesUntil\";\nimport { toCreateMessage } from \"../utils/toCreateMessage\";\nimport { vercelAttachmentAdapter } from \"../utils/vercelAttachmentAdapter\";\nimport { getVercelAIMessages } from \"../getVercelAIMessages\";\nimport { AISDKMessageConverter } from \"../utils/convertMessage\";\nimport { wrapModelContentEnvelope } from \"../../modelContentEnvelope\";\nimport {\n  type AISDKStorageFormat,\n  aiSDKV6FormatAdapter,\n} from \"../adapters/aiSDKFormatAdapter\";\nimport {\n  useExternalHistory,\n  toExportedMessageRepository,\n} from \"./useExternalHistory\";\nimport { useStreamingTiming } from \"./useStreamingTiming\";\n\nexport type CustomToCreateMessageFunction = <\n  UI_MESSAGE extends UIMessage = UIMessage,\n>(\n  message: AppendMessage,\n) => CreateUIMessage<UI_MESSAGE>;\n\nconst toUIMessage = <UI_MESSAGE extends UIMessage>(\n  createMessage: CreateUIMessage<UI_MESSAGE>,\n  fallbackRole: UI_MESSAGE[\"role\"],\n): UI_MESSAGE =>\n  ({\n    ...createMessage,\n    id: createMessage.id ?? generateId(),\n    role: createMessage.role ?? fallbackRole,\n  }) as UI_MESSAGE;\n\nexport type AISDKRuntimeAdapter = ExternalStoreSharedOptions & {\n  adapters?:\n    | (NonNullable<ExternalStoreAdapter[\"adapters\"]> & {\n        history?: ThreadHistoryAdapter | undefined;\n      })\n    | undefined;\n  toCreateMessage?: CustomToCreateMessageFunction;\n  /**\n   * Whether to automatically cancel pending interactive tool calls when the user sends a new message.\n   *\n   * When enabled (default), the pending tool calls will be marked as failed with an error message\n   * indicating the user cancelled the tool call by sending a new message.\n   *\n   * @default true\n   */\n  cancelPendingToolCallsOnSend?: boolean | undefined;\n  /**\n   * Called when `runtime.thread.resumeRun(config)` is invoked.\n   *\n   * When omitted, `resumeRun` throws `\"Runtime does not support resuming runs.\"`.\n   * Provide this to bridge resume invocations into a custom replay channel\n   * (for example, an SSE reconnect endpoint keyed by turn id).\n   */\n  onResume?: ExternalStoreAdapter[\"onResume\"];\n  /**\n   * How consecutive assistant messages are rendered.\n   *\n   * `\"concat-content\"` (the default) merges them into a single thread message.\n   * `\"none\"` keeps each assistant message as its own thread message, which is\n   * useful when a backend persists proactive or consecutive assistant messages\n   * as separate entries.\n   */\n  joinStrategy?: JoinStrategy | undefined;\n};\n\nexport const useAISDKRuntime = <UI_MESSAGE extends UIMessage = UIMessage>(\n  chatHelpers: ReturnType<typeof useChat<UI_MESSAGE>>,\n  adapter: AISDKRuntimeAdapter = {},\n) => {\n  const {\n    adapters,\n    toCreateMessage: customToCreateMessage,\n    cancelPendingToolCallsOnSend = true,\n    onResume,\n    joinStrategy,\n  } = adapter;\n  const contextAdapters = useRuntimeAdapters();\n  const [toolStatuses, setToolStatuses] = useState<\n    Record<string, ToolExecutionStatus>\n  >({});\n  const toolArgsKeyOrderCacheRef = useRef<Map<string, Map<string, string[]>>>(\n    new Map(),\n  );\n  const toolLastInputCacheRef = useRef<Map<string, ReadonlyJSONObject>>(\n    new Map(),\n  );\n  const mcpAppMetadataCacheRef = useRef<Map<string, McpAppMetadata>>(new Map());\n  const lastRunConfigRef = useRef<RunConfig | undefined>(undefined);\n\n  const hasExecutingTools = Object.values(toolStatuses).some(\n    (s) => s?.type === \"executing\",\n  );\n  const isRunning =\n    chatHelpers.status === \"submitted\" ||\n    chatHelpers.status === \"streaming\" ||\n    hasExecutingTools;\n\n  const messageTiming = useStreamingTiming(chatHelpers.messages, isRunning);\n\n  // Flag the streaming message optimistic: its id can be swapped for a server\n  // id mid-run, and the repository then drops the orphaned pre-swap id (#4037).\n  const lastMessage = chatHelpers.messages.at(-1);\n  const optimisticMessageId =\n    isRunning && lastMessage?.role === \"assistant\" ? lastMessage.id : undefined;\n\n  const messages = AISDKMessageConverter.useThreadMessages({\n    isRunning,\n    messages: chatHelpers.messages,\n    joinStrategy,\n    metadata: useMemo(\n      () => ({\n        toolStatuses,\n        messageTiming,\n        toolArgsKeyOrderCache: toolArgsKeyOrderCacheRef.current,\n        toolLastInputCache: toolLastInputCacheRef.current,\n        mcpAppMetadataCache: mcpAppMetadataCacheRef.current,\n        ...(optimisticMessageId && { optimisticMessageId }),\n        ...(chatHelpers.error && { error: chatHelpers.error.message }),\n      }),\n      [toolStatuses, messageTiming, optimisticMessageId, chatHelpers.error],\n    ),\n  });\n\n  const [runtimeRef] = useState(() => ({\n    get current(): AssistantRuntime {\n      return runtime;\n    },\n  }));\n\n  const isLoading = useExternalHistory(\n    runtimeRef,\n    adapters?.history ?? contextAdapters?.history,\n    AISDKMessageConverter.toThreadMessages as (\n      messages: UI_MESSAGE[],\n    ) => ThreadMessage[],\n    aiSDKV6FormatAdapter as MessageFormatAdapter<\n      UI_MESSAGE,\n      AISDKStorageFormat\n    >,\n    (messages) => {\n      chatHelpers.setMessages(messages);\n    },\n  );\n\n  const completePendingToolCalls = async () => {\n    if (!cancelPendingToolCallsOnSend) return;\n\n    // The runtime auto-aborts in-flight tool invocations when a new run\n    // is dispatched (append() / startRun()). All we need to do here is\n    // mark any tool without a result as cancelled in the UI message list.\n\n    // Mark any tool without a result as cancelled (uses setMessages to avoid triggering sendAutomaticallyWhen)\n    chatHelpers.setMessages((messages) => {\n      const lastMessage = messages.at(-1);\n      if (lastMessage?.role !== \"assistant\") return messages;\n\n      let hasChanges = false;\n      const parts = lastMessage.parts?.map((part) => {\n        if (!isToolUIPart(part)) return part;\n        if (part.state === \"output-available\" || part.state === \"output-error\")\n          return part;\n\n        hasChanges = true;\n        const { approval: _approval, ...rest } = part;\n        return {\n          ...rest,\n          state: \"output-error\" as const,\n          errorText: \"User cancelled tool call by sending a new message.\",\n        };\n      });\n\n      if (!hasChanges) return messages;\n      return [...messages.slice(0, -1), { ...lastMessage, parts }];\n    });\n  };\n\n  const runtime = useExternalStoreRuntime({\n    isRunning,\n    messages,\n    unstable_enableToolInvocations: true,\n    setToolStatuses,\n    setMessages: (messages) =>\n      chatHelpers.setMessages(\n        messages\n          .map(getVercelAIMessages<UI_MESSAGE>)\n          .filter(Boolean)\n          .flat(),\n      ),\n    onImport: (messages) =>\n      chatHelpers.setMessages(\n        messages\n          .map(getVercelAIMessages<UI_MESSAGE>)\n          .filter(Boolean)\n          .flat(),\n      ),\n    onExportExternalState: (): MessageFormatRepository<UI_MESSAGE> => {\n      const exported = runtimeRef.current.thread.export();\n\n      const expandedMessages: MessageFormatItem<UI_MESSAGE>[] = [];\n      const lastInnerIdMap = new Map<string, string>();\n\n      for (const item of exported.messages) {\n        const innerMessages = getExternalStoreMessages<UI_MESSAGE>(\n          item.message,\n        );\n        let parentId =\n          item.parentId != null\n            ? (lastInnerIdMap.get(item.parentId) ?? item.parentId)\n            : null;\n        for (const innerMessage of innerMessages) {\n          expandedMessages.push({ parentId, message: innerMessage });\n          parentId = aiSDKV6FormatAdapter.getId(innerMessage as UIMessage);\n        }\n        if (innerMessages.length > 0) {\n          lastInnerIdMap.set(\n            item.message.id,\n            aiSDKV6FormatAdapter.getId(\n              innerMessages[innerMessages.length - 1]! as UIMessage,\n            ),\n          );\n        }\n      }\n\n      const result: MessageFormatRepository<UI_MESSAGE> = {\n        messages: expandedMessages,\n      };\n\n      if (exported.headId != null) {\n        result.headId = lastInnerIdMap.get(exported.headId) ?? exported.headId;\n      }\n\n      return result;\n    },\n    onLoadExternalState: (repo: MessageFormatRepository<UI_MESSAGE>) => {\n      // Convert MessageFormatRepository to ExportedMessageRepository\n      const exportedRepo = toExportedMessageRepository(\n        AISDKMessageConverter.toThreadMessages,\n        repo,\n      );\n\n      // Import into the thread's MessageRepository\n      runtimeRef.current.thread.import(exportedRepo);\n    },\n    onCancel: async () => {\n      chatHelpers.stop();\n    },\n    onNew: async (message) => {\n      const createMessage = (\n        customToCreateMessage ?? toCreateMessage\n      )<UI_MESSAGE>(message);\n\n      if (!(message.startRun ?? message.role === \"user\")) {\n        chatHelpers.setMessages((current) => [\n          ...current,\n          toUIMessage<UI_MESSAGE>(createMessage, message.role),\n        ]);\n        return;\n      }\n\n      lastRunConfigRef.current = message.runConfig;\n      await completePendingToolCalls();\n      await chatHelpers.sendMessage(createMessage, {\n        metadata: message.runConfig,\n      });\n    },\n    onEdit: async (message) => {\n      const createMessage = (\n        customToCreateMessage ?? toCreateMessage\n      )<UI_MESSAGE>(message);\n\n      if (!(message.startRun ?? message.role === \"user\")) {\n        chatHelpers.setMessages((current) => [\n          ...sliceMessagesUntil(current, message.parentId),\n          toUIMessage<UI_MESSAGE>(createMessage, message.role),\n        ]);\n        return;\n      }\n\n      lastRunConfigRef.current = message.runConfig;\n      chatHelpers.setMessages((current) =>\n        sliceMessagesUntil(current, message.parentId),\n      );\n      await chatHelpers.sendMessage(createMessage, {\n        metadata: message.runConfig,\n      });\n    },\n    onReload: async (parentId: string | null, config) => {\n      lastRunConfigRef.current = config.runConfig;\n      const newMessages = sliceMessagesUntil(chatHelpers.messages, parentId);\n      chatHelpers.setMessages(newMessages);\n\n      await chatHelpers.regenerate({ metadata: config.runConfig });\n    },\n    onAddToolResult: ({\n      toolCallId,\n      toolName,\n      result,\n      isError,\n      modelContent,\n    }) => {\n      const options = { metadata: lastRunConfigRef.current };\n      if (isError) {\n        chatHelpers.addToolOutput({\n          state: \"output-error\",\n          tool: toolName ?? toolCallId,\n          toolCallId,\n          errorText:\n            typeof result === \"string\" ? result : JSON.stringify(result),\n          options,\n        });\n      } else {\n        const output =\n          modelContent !== undefined\n            ? wrapModelContentEnvelope(result, modelContent)\n            : result;\n        chatHelpers.addToolResult({\n          tool: toolName,\n          toolCallId,\n          output,\n          options,\n        });\n      }\n    },\n    onRespondToToolApproval: ({ approvalId, approved, reason }) => {\n      void chatHelpers.addToolApprovalResponse({\n        id: approvalId,\n        approved,\n        ...(reason != null && { reason }),\n        options: { metadata: lastRunConfigRef.current },\n      });\n    },\n    ...pickExternalStoreSharedOptions(adapter),\n    ...(onResume && { onResume }),\n    adapters: {\n      attachments: vercelAttachmentAdapter,\n      ...contextAdapters,\n      ...adapters,\n    },\n    isLoading,\n  });\n\n  return runtime;\n};\n"],"mappings":";;;;;;;;;;;;;;;AAmDA,MAAM,eACJ,eACA,kBAEC;CACC,GAAG;CACH,IAAI,cAAc,MAAM,WAAW;CACnC,MAAM,cAAc,QAAQ;AAC9B;AAqCF,MAAa,mBACX,aACA,UAA+B,CAAC,MAC7B;CACH,MAAM,EACJ,UACA,iBAAiB,uBACjB,+BAA+B,MAC/B,UACA,iBACE;CACJ,MAAM,kBAAkB,mBAAmB;CAC3C,MAAM,CAAC,cAAc,mBAAmB,SAEtC,CAAC,CAAC;CACJ,MAAM,2BAA2B,uBAC/B,IAAI,IAAI,CACV;CACA,MAAM,wBAAwB,uBAC5B,IAAI,IAAI,CACV;CACA,MAAM,yBAAyB,uBAAoC,IAAI,IAAI,CAAC;CAC5E,MAAM,mBAAmB,OAA8B,KAAA,CAAS;CAEhE,MAAM,oBAAoB,OAAO,OAAO,YAAY,CAAC,CAAC,MACnD,MAAM,GAAG,SAAS,WACrB;CACA,MAAM,YACJ,YAAY,WAAW,eACvB,YAAY,WAAW,eACvB;CAEF,MAAM,gBAAgB,mBAAmB,YAAY,UAAU,SAAS;CAIxE,MAAM,cAAc,YAAY,SAAS,GAAG,EAAE;CAC9C,MAAM,sBACJ,aAAa,aAAa,SAAS,cAAc,YAAY,KAAK,KAAA;CAEpE,MAAM,WAAW,sBAAsB,kBAAkB;EACvD;EACA,UAAU,YAAY;EACtB;EACA,UAAU,eACD;GACL;GACA;GACA,uBAAuB,yBAAyB;GAChD,oBAAoB,sBAAsB;GAC1C,qBAAqB,uBAAuB;GAC5C,GAAI,uBAAuB,EAAE,oBAAoB;GACjD,GAAI,YAAY,SAAS,EAAE,OAAO,YAAY,MAAM,QAAQ;EAC9D,IACA;GAAC;GAAc;GAAe;GAAqB,YAAY;EAAK,CACtE;CACF,CAAC;CAED,MAAM,CAAC,cAAc,gBAAgB,EACnC,IAAI,UAA4B;EAC9B,OAAO;CACT,EACF,EAAE;CAEF,MAAM,YAAY,mBAChB,YACA,UAAU,WAAW,iBAAiB,SACtC,sBAAsB,kBAGtB,uBAIC,aAAa;EACZ,YAAY,YAAY,QAAQ;CAClC,CACF;CAEA,MAAM,2BAA2B,YAAY;EAC3C,IAAI,CAAC,8BAA8B;EAOnC,YAAY,aAAa,aAAa;GACpC,MAAM,cAAc,SAAS,GAAG,EAAE;GAClC,IAAI,aAAa,SAAS,aAAa,OAAO;GAE9C,IAAI,aAAa;GACjB,MAAM,QAAQ,YAAY,OAAO,KAAK,SAAS;IAC7C,IAAI,CAAC,aAAa,IAAI,GAAG,OAAO;IAChC,IAAI,KAAK,UAAU,sBAAsB,KAAK,UAAU,gBACtD,OAAO;IAET,aAAa;IACb,MAAM,EAAE,UAAU,WAAW,GAAG,SAAS;IACzC,OAAO;KACL,GAAG;KACH,OAAO;KACP,WAAW;IACb;GACF,CAAC;GAED,IAAI,CAAC,YAAY,OAAO;GACxB,OAAO,CAAC,GAAG,SAAS,MAAM,GAAG,EAAE,GAAG;IAAE,GAAG;IAAa;GAAM,CAAC;EAC7D,CAAC;CACH;CAEA,MAAM,UAAU,wBAAwB;EACtC;EACA;EACA,gCAAgC;EAChC;EACA,cAAc,aACZ,YAAY,YACV,SACG,IAAI,mBAA+B,CAAC,CACpC,OAAO,OAAO,CAAC,CACf,KAAK,CACV;EACF,WAAW,aACT,YAAY,YACV,SACG,IAAI,mBAA+B,CAAC,CACpC,OAAO,OAAO,CAAC,CACf,KAAK,CACV;EACF,6BAAkE;GAChE,MAAM,WAAW,WAAW,QAAQ,OAAO,OAAO;GAElD,MAAM,mBAAoD,CAAC;GAC3D,MAAM,iCAAiB,IAAI,IAAoB;GAE/C,KAAK,MAAM,QAAQ,SAAS,UAAU;IACpC,MAAM,gBAAgB,yBACpB,KAAK,OACP;IACA,IAAI,WACF,KAAK,YAAY,OACZ,eAAe,IAAI,KAAK,QAAQ,KAAK,KAAK,WAC3C;IACN,KAAK,MAAM,gBAAgB,eAAe;KACxC,iBAAiB,KAAK;MAAE;MAAU,SAAS;KAAa,CAAC;KACzD,WAAW,qBAAqB,MAAM,YAAyB;IACjE;IACA,IAAI,cAAc,SAAS,GACzB,eAAe,IACb,KAAK,QAAQ,IACb,qBAAqB,MACnB,cAAc,cAAc,SAAS,EACvC,CACF;GAEJ;GAEA,MAAM,SAA8C,EAClD,UAAU,iBACZ;GAEA,IAAI,SAAS,UAAU,MACrB,OAAO,SAAS,eAAe,IAAI,SAAS,MAAM,KAAK,SAAS;GAGlE,OAAO;EACT;EACA,sBAAsB,SAA8C;GAElE,MAAM,eAAe,4BACnB,sBAAsB,kBACtB,IACF;GAGA,WAAW,QAAQ,OAAO,OAAO,YAAY;EAC/C;EACA,UAAU,YAAY;GACpB,YAAY,KAAK;EACnB;EACA,OAAO,OAAO,YAAY;GACxB,MAAM,iBACJ,yBAAyB,gBAAA,CACb,OAAO;GAErB,IAAI,EAAE,QAAQ,YAAY,QAAQ,SAAS,SAAS;IAClD,YAAY,aAAa,YAAY,CACnC,GAAG,SACH,YAAwB,eAAe,QAAQ,IAAI,CACrD,CAAC;IACD;GACF;GAEA,iBAAiB,UAAU,QAAQ;GACnC,MAAM,yBAAyB;GAC/B,MAAM,YAAY,YAAY,eAAe,EAC3C,UAAU,QAAQ,UACpB,CAAC;EACH;EACA,QAAQ,OAAO,YAAY;GACzB,MAAM,iBACJ,yBAAyB,gBAAA,CACb,OAAO;GAErB,IAAI,EAAE,QAAQ,YAAY,QAAQ,SAAS,SAAS;IAClD,YAAY,aAAa,YAAY,CACnC,GAAG,mBAAmB,SAAS,QAAQ,QAAQ,GAC/C,YAAwB,eAAe,QAAQ,IAAI,CACrD,CAAC;IACD;GACF;GAEA,iBAAiB,UAAU,QAAQ;GACnC,YAAY,aAAa,YACvB,mBAAmB,SAAS,QAAQ,QAAQ,CAC9C;GACA,MAAM,YAAY,YAAY,eAAe,EAC3C,UAAU,QAAQ,UACpB,CAAC;EACH;EACA,UAAU,OAAO,UAAyB,WAAW;GACnD,iBAAiB,UAAU,OAAO;GAClC,MAAM,cAAc,mBAAmB,YAAY,UAAU,QAAQ;GACrE,YAAY,YAAY,WAAW;GAEnC,MAAM,YAAY,WAAW,EAAE,UAAU,OAAO,UAAU,CAAC;EAC7D;EACA,kBAAkB,EAChB,YACA,UACA,QACA,SACA,mBACI;GACJ,MAAM,UAAU,EAAE,UAAU,iBAAiB,QAAQ;GACrD,IAAI,SACF,YAAY,cAAc;IACxB,OAAO;IACP,MAAM,YAAY;IAClB;IACA,WACE,OAAO,WAAW,WAAW,SAAS,KAAK,UAAU,MAAM;IAC7D;GACF,CAAC;QACI;IACL,MAAM,SACJ,iBAAiB,KAAA,IACb,yBAAyB,QAAQ,YAAY,IAC7C;IACN,YAAY,cAAc;KACxB,MAAM;KACN;KACA;KACA;IACF,CAAC;GACH;EACF;EACA,0BAA0B,EAAE,YAAY,UAAU,aAAa;GAC7D,YAAiB,wBAAwB;IACvC,IAAI;IACJ;IACA,GAAI,UAAU,QAAQ,EAAE,OAAO;IAC/B,SAAS,EAAE,UAAU,iBAAiB,QAAQ;GAChD,CAAC;EACH;EACA,GAAG,+BAA+B,OAAO;EACzC,GAAI,YAAY,EAAE,SAAS;EAC3B,UAAU;GACR,aAAa;GACb,GAAG;GACH,GAAG;EACL;EACA;CACF,CAAC;CAED,OAAO;AACT"}

package/dist/ui/use-chat/useChatRuntime.d.ts

@@ -10,6 +10,7 @@ type UseChatRuntimeOptions<UI_MESSAGE extends UIMessage$1 = UIMessage$1> = ChatI
   adapters?: AISDKRuntimeAdapter["adapters"] | undefined;
   toCreateMessage?: CustomToCreateMessageFunction;
   onResume?: AISDKRuntimeAdapter["onResume"];
+  joinStrategy?: AISDKRuntimeAdapter["joinStrategy"];
 };
 declare const useChatRuntime: <UI_MESSAGE extends UIMessage$1 = UIMessage$1>({
   cloud,

package/dist/ui/use-chat/useChatRuntime.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"useChatRuntime.d.ts","names":[],"sources":["../../../src/ui/use-chat/useChatRuntime.ts"],"mappings":";;;;;;;KAwBY,qBAAA,oBAAyC,WAAA,GAAY,WAAA,IAC/D,QAAA,CAAS,UAAA,IACP,0BAAA;EACE,KAAA,GAAQ,cAAA;EACR,QAAA,GAAW,mBAAA;EACX,eAAA,GAAkB,6BAAA;EAClB,QAAA,GAAW,mBAAA;AAAA;AAAA,cAwGJ,cAAA,sBAAqC,WAAA,GAAY,WAAA;EAAW,KAAA;EAAA,GAAA;AAAA,IAGtE,qBAAA,CAAsB,UAAA,MAAmB,gBAAA"}
+{"version":3,"file":"useChatRuntime.d.ts","names":[],"sources":["../../../src/ui/use-chat/useChatRuntime.ts"],"mappings":";;;;;;;KAwBY,qBAAA,oBAAyC,WAAA,GAAY,WAAA,IAC/D,QAAA,CAAS,UAAA,IACP,0BAAA;EACE,KAAA,GAAQ,cAAA;EACR,QAAA,GAAW,mBAAA;EACX,eAAA,GAAkB,6BAAA;EAClB,QAAA,GAAW,mBAAA;EACX,YAAA,GAAe,mBAAA;AAAA;AAAA,cA0GR,cAAA,sBAAqC,WAAA,GAAY,WAAA;EAAW,KAAA;EAAA,GAAA;AAAA,IAGtE,qBAAA,CAAsB,UAAA,MAAmB,gBAAA"}

package/dist/ui/use-chat/useChatRuntime.js

@@ -24,7 +24,7 @@ const getResumableAdapter = (transport) => {
 	return candidate.call(transport);
 };
 const useChatThreadRuntime = (options) => {
-	const { adapters, transport: transportOptions, toCreateMessage, isDisabled: _isDisabled, isSendDisabled: _isSendDisabled, unstable_capabilities: _unstable_capabilities, suggestions: _suggestions, onResume, ...chatOptions } = options ?? {};
+	const { adapters, transport: transportOptions, toCreateMessage, isDisabled: _isDisabled, isSendDisabled: _isSendDisabled, unstable_capabilities: _unstable_capabilities, suggestions: _suggestions, onResume, joinStrategy, ...chatOptions } = options ?? {};
 	const transport = useDynamicChatTransport(transportOptions ?? new AssistantChatTransport());
 	const id = useAuiState((s) => s.threadListItem.id);
 	const aui = useAui();
@@ -37,7 +37,8 @@ const useChatThreadRuntime = (options) => {
 		adapters,
 		...pickExternalStoreSharedOptions(options ?? {}),
 		...toCreateMessage && { toCreateMessage },
-		...onResume && { onResume }
+		...onResume && { onResume },
+		...joinStrategy && { joinStrategy }
 	});
 	if (transport instanceof AssistantChatTransport) {
 		transport.setRuntime(runtime);

package/dist/ui/use-chat/useChatRuntime.js.map

@@ -1 +1 @@
-{"version":3,"file":"useChatRuntime.js","names":[],"sources":["../../../src/ui/use-chat/useChatRuntime.ts"],"sourcesContent":["\"use client\";\n\nimport { useChat, type UIMessage } from \"@ai-sdk/react\";\nimport type { AssistantCloud } from \"assistant-cloud\";\nimport {\n  pickExternalStoreSharedOptions,\n  type AssistantRuntime,\n  type ExternalStoreSharedOptions,\n} from \"@assistant-ui/core\";\nimport {\n  useCloudThreadListAdapter,\n  useRemoteThreadListRuntime,\n} from \"@assistant-ui/core/react\";\nimport { useAui, useAuiState } from \"@assistant-ui/store\";\nimport {\n  useAISDKRuntime,\n  type AISDKRuntimeAdapter,\n  type CustomToCreateMessageFunction,\n} from \"./useAISDKRuntime\";\nimport type { ChatInit, ChatTransport } from \"ai\";\nimport { AssistantChatTransport } from \"./AssistantChatTransport\";\nimport type { AssistantChatResumableOptions } from \"../resumable\";\nimport { useEffect, useMemo, useRef } from \"react\";\n\nexport type UseChatRuntimeOptions<UI_MESSAGE extends UIMessage = UIMessage> =\n  ChatInit<UI_MESSAGE> &\n    ExternalStoreSharedOptions & {\n      cloud?: AssistantCloud | undefined;\n      adapters?: AISDKRuntimeAdapter[\"adapters\"] | undefined;\n      toCreateMessage?: CustomToCreateMessageFunction;\n      onResume?: AISDKRuntimeAdapter[\"onResume\"];\n    };\n\nconst useDynamicChatTransport = <UI_MESSAGE extends UIMessage = UIMessage>(\n  transport: ChatTransport<UI_MESSAGE>,\n): ChatTransport<UI_MESSAGE> => {\n  const transportRef = useRef<ChatTransport<UI_MESSAGE>>(transport);\n  useEffect(() => {\n    transportRef.current = transport;\n  });\n  const dynamicTransport = useMemo(\n    () =>\n      new Proxy(transportRef.current, {\n        get(_, prop) {\n          const res =\n            transportRef.current[prop as keyof ChatTransport<UI_MESSAGE>];\n          return typeof res === \"function\"\n            ? res.bind(transportRef.current)\n            : res;\n        },\n      }),\n    [],\n  );\n  return dynamicTransport;\n};\n\nconst getResumableAdapter = <UI_MESSAGE extends UIMessage>(\n  transport: ChatTransport<UI_MESSAGE>,\n): AssistantChatResumableOptions | undefined => {\n  if (transport instanceof AssistantChatTransport) {\n    return transport.getResumableAdapter();\n  }\n  const candidate = (transport as { getResumableAdapter?: () => unknown })\n    .getResumableAdapter;\n  if (typeof candidate !== \"function\") return undefined;\n  return candidate.call(transport) as AssistantChatResumableOptions | undefined;\n};\n\nconst useChatThreadRuntime = <UI_MESSAGE extends UIMessage = UIMessage>(\n  options?: UseChatRuntimeOptions<UI_MESSAGE>,\n): AssistantRuntime => {\n  const {\n    adapters,\n    transport: transportOptions,\n    toCreateMessage,\n    isDisabled: _isDisabled,\n    isSendDisabled: _isSendDisabled,\n    unstable_capabilities: _unstable_capabilities,\n    suggestions: _suggestions,\n    onResume,\n    ...chatOptions\n  } = options ?? {};\n  // peel guard: any shared key left in `chatOptions` collapses this to `never`\n  true satisfies keyof typeof chatOptions &\n    keyof ExternalStoreSharedOptions extends never\n    ? true\n    : never;\n\n  const transport = useDynamicChatTransport(\n    transportOptions ?? new AssistantChatTransport(),\n  );\n\n  const id = useAuiState((s) => s.threadListItem.id);\n  const aui = useAui();\n  const chat = useChat({\n    ...chatOptions,\n    id,\n    transport,\n  });\n\n  const runtime = useAISDKRuntime(chat, {\n    adapters,\n    ...pickExternalStoreSharedOptions(options ?? {}),\n    ...(toCreateMessage && { toCreateMessage }),\n    ...(onResume && { onResume }),\n  });\n\n  if (transport instanceof AssistantChatTransport) {\n    transport.setRuntime(runtime);\n    transport.__internal_setGetThreadListItem(() =>\n      aui.threadListItem.source ? aui.threadListItem() : undefined,\n    );\n  }\n\n  const resumeFiredRef = useRef(false);\n  useEffect(() => {\n    if (resumeFiredRef.current) return;\n    const adapter = getResumableAdapter(transport);\n    if (!adapter) return;\n    const pending = adapter.storage.getStreamId();\n    if (!pending) return;\n    resumeFiredRef.current = true;\n    chat.resumeStream().catch((err: unknown) => {\n      console.warn(\n        \"[assistant-ui] resumable: resume failed; clearing stored stream id\",\n        err,\n      );\n      adapter.storage.clear();\n    });\n  }, [transport, chat]);\n\n  return runtime;\n};\n\nexport const useChatRuntime = <UI_MESSAGE extends UIMessage = UIMessage>({\n  cloud,\n  ...options\n}: UseChatRuntimeOptions<UI_MESSAGE> = {}): AssistantRuntime => {\n  const cloudAdapter = useCloudThreadListAdapter({ cloud });\n  return useRemoteThreadListRuntime({\n    runtimeHook: function RuntimeHook() {\n      return useChatThreadRuntime(options);\n    },\n    adapter: cloudAdapter,\n    allowNesting: true,\n  });\n};\n"],"mappings":";;;;;;;;;AAiCA,MAAM,2BACJ,cAC8B;CAC9B,MAAM,eAAe,OAAkC,SAAS;CAChE,gBAAgB;EACd,aAAa,UAAU;CACzB,CAAC;CAcD,OAbyB,cAErB,IAAI,MAAM,aAAa,SAAS,EAC9B,IAAI,GAAG,MAAM;EACX,MAAM,MACJ,aAAa,QAAQ;EACvB,OAAO,OAAO,QAAQ,aAClB,IAAI,KAAK,aAAa,OAAO,IAC7B;CACN,EACF,CAAC,GACH,CAAC,CAEmB;AACxB;AAEA,MAAM,uBACJ,cAC8C;CAC9C,IAAI,qBAAqB,wBACvB,OAAO,UAAU,oBAAoB;CAEvC,MAAM,YAAa,UAChB;CACH,IAAI,OAAO,cAAc,YAAY,OAAO,KAAA;CAC5C,OAAO,UAAU,KAAK,SAAS;AACjC;AAEA,MAAM,wBACJ,YACqB;CACrB,MAAM,EACJ,UACA,WAAW,kBACX,iBACA,YAAY,aACZ,gBAAgB,iBAChB,uBAAuB,wBACvB,aAAa,cACb,UACA,GAAG,gBACD,WAAW,CAAC;CAOhB,MAAM,YAAY,wBAChB,oBAAoB,IAAI,uBAAuB,CACjD;CAEA,MAAM,KAAK,aAAa,MAAM,EAAE,eAAe,EAAE;CACjD,MAAM,MAAM,OAAO;CACnB,MAAM,OAAO,QAAQ;EACnB,GAAG;EACH;EACA;CACF,CAAC;CAED,MAAM,UAAU,gBAAgB,MAAM;EACpC;EACA,GAAG,+BAA+B,WAAW,CAAC,CAAC;EAC/C,GAAI,mBAAmB,EAAE,gBAAgB;EACzC,GAAI,YAAY,EAAE,SAAS;CAC7B,CAAC;CAED,IAAI,qBAAqB,wBAAwB;EAC/C,UAAU,WAAW,OAAO;EAC5B,UAAU,sCACR,IAAI,eAAe,SAAS,IAAI,eAAe,IAAI,KAAA,CACrD;CACF;CAEA,MAAM,iBAAiB,OAAO,KAAK;CACnC,gBAAgB;EACd,IAAI,eAAe,SAAS;EAC5B,MAAM,UAAU,oBAAoB,SAAS;EAC7C,IAAI,CAAC,SAAS;EAEd,IAAI,CADY,QAAQ,QAAQ,YACrB,GAAG;EACd,eAAe,UAAU;EACzB,KAAK,aAAa,EAAE,OAAO,QAAiB;GAC1C,QAAQ,KACN,sEACA,GACF;GACA,QAAQ,QAAQ,MAAM;EACxB,CAAC;CACH,GAAG,CAAC,WAAW,IAAI,CAAC;CAEpB,OAAO;AACT;AAEA,MAAa,kBAA4D,EACvE,OACA,GAAG,YACkC,CAAC,MAAwB;CAE9D,OAAO,2BAA2B;EAChC,aAAa,SAAS,cAAc;GAClC,OAAO,qBAAqB,OAAO;EACrC;EACA,SALmB,0BAA0B,EAAE,MAAM,CAKjC;EACpB,cAAc;CAChB,CAAC;AACH"}
+{"version":3,"file":"useChatRuntime.js","names":[],"sources":["../../../src/ui/use-chat/useChatRuntime.ts"],"sourcesContent":["\"use client\";\n\nimport { useChat, type UIMessage } from \"@ai-sdk/react\";\nimport type { AssistantCloud } from \"assistant-cloud\";\nimport {\n  pickExternalStoreSharedOptions,\n  type AssistantRuntime,\n  type ExternalStoreSharedOptions,\n} from \"@assistant-ui/core\";\nimport {\n  useCloudThreadListAdapter,\n  useRemoteThreadListRuntime,\n} from \"@assistant-ui/core/react\";\nimport { useAui, useAuiState } from \"@assistant-ui/store\";\nimport {\n  useAISDKRuntime,\n  type AISDKRuntimeAdapter,\n  type CustomToCreateMessageFunction,\n} from \"./useAISDKRuntime\";\nimport type { ChatInit, ChatTransport } from \"ai\";\nimport { AssistantChatTransport } from \"./AssistantChatTransport\";\nimport type { AssistantChatResumableOptions } from \"../resumable\";\nimport { useEffect, useMemo, useRef } from \"react\";\n\nexport type UseChatRuntimeOptions<UI_MESSAGE extends UIMessage = UIMessage> =\n  ChatInit<UI_MESSAGE> &\n    ExternalStoreSharedOptions & {\n      cloud?: AssistantCloud | undefined;\n      adapters?: AISDKRuntimeAdapter[\"adapters\"] | undefined;\n      toCreateMessage?: CustomToCreateMessageFunction;\n      onResume?: AISDKRuntimeAdapter[\"onResume\"];\n      joinStrategy?: AISDKRuntimeAdapter[\"joinStrategy\"];\n    };\n\nconst useDynamicChatTransport = <UI_MESSAGE extends UIMessage = UIMessage>(\n  transport: ChatTransport<UI_MESSAGE>,\n): ChatTransport<UI_MESSAGE> => {\n  const transportRef = useRef<ChatTransport<UI_MESSAGE>>(transport);\n  useEffect(() => {\n    transportRef.current = transport;\n  });\n  const dynamicTransport = useMemo(\n    () =>\n      new Proxy(transportRef.current, {\n        get(_, prop) {\n          const res =\n            transportRef.current[prop as keyof ChatTransport<UI_MESSAGE>];\n          return typeof res === \"function\"\n            ? res.bind(transportRef.current)\n            : res;\n        },\n      }),\n    [],\n  );\n  return dynamicTransport;\n};\n\nconst getResumableAdapter = <UI_MESSAGE extends UIMessage>(\n  transport: ChatTransport<UI_MESSAGE>,\n): AssistantChatResumableOptions | undefined => {\n  if (transport instanceof AssistantChatTransport) {\n    return transport.getResumableAdapter();\n  }\n  const candidate = (transport as { getResumableAdapter?: () => unknown })\n    .getResumableAdapter;\n  if (typeof candidate !== \"function\") return undefined;\n  return candidate.call(transport) as AssistantChatResumableOptions | undefined;\n};\n\nconst useChatThreadRuntime = <UI_MESSAGE extends UIMessage = UIMessage>(\n  options?: UseChatRuntimeOptions<UI_MESSAGE>,\n): AssistantRuntime => {\n  const {\n    adapters,\n    transport: transportOptions,\n    toCreateMessage,\n    isDisabled: _isDisabled,\n    isSendDisabled: _isSendDisabled,\n    unstable_capabilities: _unstable_capabilities,\n    suggestions: _suggestions,\n    onResume,\n    joinStrategy,\n    ...chatOptions\n  } = options ?? {};\n  // peel guard: any shared key left in `chatOptions` collapses this to `never`\n  true satisfies keyof typeof chatOptions &\n    keyof ExternalStoreSharedOptions extends never\n    ? true\n    : never;\n\n  const transport = useDynamicChatTransport(\n    transportOptions ?? new AssistantChatTransport(),\n  );\n\n  const id = useAuiState((s) => s.threadListItem.id);\n  const aui = useAui();\n  const chat = useChat({\n    ...chatOptions,\n    id,\n    transport,\n  });\n\n  const runtime = useAISDKRuntime(chat, {\n    adapters,\n    ...pickExternalStoreSharedOptions(options ?? {}),\n    ...(toCreateMessage && { toCreateMessage }),\n    ...(onResume && { onResume }),\n    ...(joinStrategy && { joinStrategy }),\n  });\n\n  if (transport instanceof AssistantChatTransport) {\n    transport.setRuntime(runtime);\n    transport.__internal_setGetThreadListItem(() =>\n      aui.threadListItem.source ? aui.threadListItem() : undefined,\n    );\n  }\n\n  const resumeFiredRef = useRef(false);\n  useEffect(() => {\n    if (resumeFiredRef.current) return;\n    const adapter = getResumableAdapter(transport);\n    if (!adapter) return;\n    const pending = adapter.storage.getStreamId();\n    if (!pending) return;\n    resumeFiredRef.current = true;\n    chat.resumeStream().catch((err: unknown) => {\n      console.warn(\n        \"[assistant-ui] resumable: resume failed; clearing stored stream id\",\n        err,\n      );\n      adapter.storage.clear();\n    });\n  }, [transport, chat]);\n\n  return runtime;\n};\n\nexport const useChatRuntime = <UI_MESSAGE extends UIMessage = UIMessage>({\n  cloud,\n  ...options\n}: UseChatRuntimeOptions<UI_MESSAGE> = {}): AssistantRuntime => {\n  const cloudAdapter = useCloudThreadListAdapter({ cloud });\n  return useRemoteThreadListRuntime({\n    runtimeHook: function RuntimeHook() {\n      return useChatThreadRuntime(options);\n    },\n    adapter: cloudAdapter,\n    allowNesting: true,\n  });\n};\n"],"mappings":";;;;;;;;;AAkCA,MAAM,2BACJ,cAC8B;CAC9B,MAAM,eAAe,OAAkC,SAAS;CAChE,gBAAgB;EACd,aAAa,UAAU;CACzB,CAAC;CAcD,OAbyB,cAErB,IAAI,MAAM,aAAa,SAAS,EAC9B,IAAI,GAAG,MAAM;EACX,MAAM,MACJ,aAAa,QAAQ;EACvB,OAAO,OAAO,QAAQ,aAClB,IAAI,KAAK,aAAa,OAAO,IAC7B;CACN,EACF,CAAC,GACH,CAAC,CAEmB;AACxB;AAEA,MAAM,uBACJ,cAC8C;CAC9C,IAAI,qBAAqB,wBACvB,OAAO,UAAU,oBAAoB;CAEvC,MAAM,YAAa,UAChB;CACH,IAAI,OAAO,cAAc,YAAY,OAAO,KAAA;CAC5C,OAAO,UAAU,KAAK,SAAS;AACjC;AAEA,MAAM,wBACJ,YACqB;CACrB,MAAM,EACJ,UACA,WAAW,kBACX,iBACA,YAAY,aACZ,gBAAgB,iBAChB,uBAAuB,wBACvB,aAAa,cACb,UACA,cACA,GAAG,gBACD,WAAW,CAAC;CAOhB,MAAM,YAAY,wBAChB,oBAAoB,IAAI,uBAAuB,CACjD;CAEA,MAAM,KAAK,aAAa,MAAM,EAAE,eAAe,EAAE;CACjD,MAAM,MAAM,OAAO;CACnB,MAAM,OAAO,QAAQ;EACnB,GAAG;EACH;EACA;CACF,CAAC;CAED,MAAM,UAAU,gBAAgB,MAAM;EACpC;EACA,GAAG,+BAA+B,WAAW,CAAC,CAAC;EAC/C,GAAI,mBAAmB,EAAE,gBAAgB;EACzC,GAAI,YAAY,EAAE,SAAS;EAC3B,GAAI,gBAAgB,EAAE,aAAa;CACrC,CAAC;CAED,IAAI,qBAAqB,wBAAwB;EAC/C,UAAU,WAAW,OAAO;EAC5B,UAAU,sCACR,IAAI,eAAe,SAAS,IAAI,eAAe,IAAI,KAAA,CACrD;CACF;CAEA,MAAM,iBAAiB,OAAO,KAAK;CACnC,gBAAgB;EACd,IAAI,eAAe,SAAS;EAC5B,MAAM,UAAU,oBAAoB,SAAS;EAC7C,IAAI,CAAC,SAAS;EAEd,IAAI,CADY,QAAQ,QAAQ,YACrB,GAAG;EACd,eAAe,UAAU;EACzB,KAAK,aAAa,CAAC,CAAC,OAAO,QAAiB;GAC1C,QAAQ,KACN,sEACA,GACF;GACA,QAAQ,QAAQ,MAAM;EACxB,CAAC;CACH,GAAG,CAAC,WAAW,IAAI,CAAC;CAEpB,OAAO;AACT;AAEA,MAAa,kBAA4D,EACvE,OACA,GAAG,YACkC,CAAC,MAAwB;CAE9D,OAAO,2BAA2B;EAChC,aAAa,SAAS,cAAc;GAClC,OAAO,qBAAqB,OAAO;EACrC;EACA,SALmB,0BAA0B,EAAE,MAAM,CAKjC;EACpB,cAAc;CAChB,CAAC;AACH"}