@assistant-ui/ai-sdk 0.0.10 → 0.0.12

package/dist/model-context/injectInteractableContext.js

@@ -15,6 +15,8 @@ import { unstable_formatInteractableSnapshot, unstable_getInteractableSnapshots
 * `format` must also handle entries with `partial: true`, whose `state` carries
 * only the fields that changed since the model's last known state.
 *
+* @deprecated Experimental since 2026-06-23. Not scheduled for removal; the API may change in any release.
+*
 * @example
 * ```ts
 * import { convertToModelMessages, streamText } from "ai";
@@ -23,6 +25,7 @@ import { unstable_formatInteractableSnapshot, unstable_getInteractableSnapshots
 * export async function POST(req: Request) {
 *   const { messages } = await req.json();
 *   const result = streamText({
+*     abortSignal: req.signal,
 *     model: myModel,
 *     messages: await convertToModelMessages(unstable_injectInteractableContext(messages)),
 *   });

package/dist/model-context/injectInteractableContext.js.map

@@ -1 +1 @@
-{"version":3,"file":"injectInteractableContext.js","names":["UIMessage","unstable_formatInteractableSnapshot","unstable_getInteractableSnapshots","Unstable_InteractableSnapshotEntry","unstable_injectInteractableContext","messages","format","item","map","msg","role","items","length","text","join","alreadyInjected","parts","type","const"],"sources":["../../src/model-context/injectInteractableContext.ts"],"sourcesContent":["import type { UIMessage } from \"ai\";\nimport {\n  unstable_formatInteractableSnapshot,\n  unstable_getInteractableSnapshots,\n  type Unstable_InteractableSnapshotEntry,\n} from \"@assistant-ui/core\";\n\n/**\n * Injects interactable state snapshots into messages as model-visible text.\n *\n * Mirrors {@link injectQuoteContext}: reads the frozen snapshot stamped on a user\n * message's `metadata.custom.interactables` (by the interactables scope at send\n * time) and prepends a text part. Run this in your route handler before\n * `convertToModelMessages`, which otherwise ignores `metadata.custom`.\n *\n * Wording is consumer-owned — pass `format` to control how each snapshot reads.\n * A snapshot may originate from a user edit or an agent `update_*` call, so the\n * default phrasing is neutral. Keep the instance id visible in custom wording:\n * the model needs it to address the `update_*` tool's `id` parameter. A custom\n * `format` must also handle entries with `partial: true`, whose `state` carries\n * only the fields that changed since the model's last known state.\n *\n * @example\n * ```ts\n * import { convertToModelMessages, streamText } from \"ai\";\n * import { unstable_injectInteractableContext } from \"@assistant-ui/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(unstable_injectInteractableContext(messages)),\n *   });\n *   return result.toUIMessageStreamResponse();\n * }\n * ```\n */\nexport function unstable_injectInteractableContext(\n  messages: UIMessage[],\n  format: (\n    item: Unstable_InteractableSnapshotEntry,\n  ) => string = unstable_formatInteractableSnapshot,\n): UIMessage[] {\n  return messages.map((msg) => {\n    if (msg.role !== \"user\") return msg;\n\n    const items = unstable_getInteractableSnapshots(msg);\n    if (!items?.length) return msg;\n\n    const text = `${items.map(format).join(\"\\n\")}\\n\\n`;\n\n    const alreadyInjected =\n      msg.parts?.[0]?.type === \"text\" && msg.parts[0].text === text;\n    if (alreadyInjected) return msg;\n\n    return {\n      ...msg,\n      parts: [{ type: \"text\" as const, text }, ...(msg.parts ?? [])],\n    };\n  });\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAqCA,SAAgBI,mCACdC,UACAC,SAEcL,qCACD;CACb,OAAOI,SAASG,KAAKC,QAAQ;EAC3B,IAAIA,IAAIC,SAAS,QAAQ,OAAOD;EAEhC,MAAME,QAAQT,kCAAkCO,GAAG;EACnD,IAAI,CAACE,OAAOC,QAAQ,OAAOH;EAE3B,MAAMI,OAAO,GAAGF,MAAMH,IAAIF,MAAM,CAAC,CAACQ,KAAK,IAAI,EAAC;EAI5C,IADEL,IAAIO,QAAQ,EAAE,EAAEC,SAAS,UAAUR,IAAIO,MAAM,EAAE,CAACH,SAASA,MACtC,OAAOJ;EAE5B,OAAO;GACL,GAAGA;GACHO,OAAO,CAAC;IAAEC,MAAM;IAAiBJ;GAAK,GAAG,GAAIJ,IAAIO,SAAS,CAAA,CAAG;EAC/D;CACF,CAAC;AACH"}
+{"version":3,"file":"injectInteractableContext.js","names":["UIMessage","unstable_formatInteractableSnapshot","unstable_getInteractableSnapshots","Unstable_InteractableSnapshotEntry","unstable_injectInteractableContext","messages","format","item","map","msg","role","items","length","text","join","alreadyInjected","parts","type","const"],"sources":["../../src/model-context/injectInteractableContext.ts"],"sourcesContent":["import type { UIMessage } from \"ai\";\nimport {\n  unstable_formatInteractableSnapshot,\n  unstable_getInteractableSnapshots,\n  type Unstable_InteractableSnapshotEntry,\n} from \"@assistant-ui/core\";\n\n/**\n * Injects interactable state snapshots into messages as model-visible text.\n *\n * Mirrors {@link injectQuoteContext}: reads the frozen snapshot stamped on a user\n * message's `metadata.custom.interactables` (by the interactables scope at send\n * time) and prepends a text part. Run this in your route handler before\n * `convertToModelMessages`, which otherwise ignores `metadata.custom`.\n *\n * Wording is consumer-owned — pass `format` to control how each snapshot reads.\n * A snapshot may originate from a user edit or an agent `update_*` call, so the\n * default phrasing is neutral. Keep the instance id visible in custom wording:\n * the model needs it to address the `update_*` tool's `id` parameter. A custom\n * `format` must also handle entries with `partial: true`, whose `state` carries\n * only the fields that changed since the model's last known state.\n *\n * @deprecated Experimental since 2026-06-23. Not scheduled for removal; the API may change in any release.\n *\n * @example\n * ```ts\n * import { convertToModelMessages, streamText } from \"ai\";\n * import { unstable_injectInteractableContext } from \"@assistant-ui/ai-sdk\";\n *\n * export async function POST(req: Request) {\n *   const { messages } = await req.json();\n *   const result = streamText({\n *     abortSignal: req.signal,\n *     model: myModel,\n *     messages: await convertToModelMessages(unstable_injectInteractableContext(messages)),\n *   });\n *   return result.toUIMessageStreamResponse();\n * }\n * ```\n */\nexport function unstable_injectInteractableContext(\n  messages: UIMessage[],\n  format: (\n    item: Unstable_InteractableSnapshotEntry,\n  ) => string = unstable_formatInteractableSnapshot,\n): UIMessage[] {\n  return messages.map((msg) => {\n    if (msg.role !== \"user\") return msg;\n\n    const items = unstable_getInteractableSnapshots(msg);\n    if (!items?.length) return msg;\n\n    const text = `${items.map(format).join(\"\\n\")}\\n\\n`;\n\n    const alreadyInjected =\n      msg.parts?.[0]?.type === \"text\" && msg.parts[0].text === text;\n    if (alreadyInjected) return msg;\n\n    return {\n      ...msg,\n      parts: [{ type: \"text\" as const, text }, ...(msg.parts ?? [])],\n    };\n  });\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAwCA,SAAgBI,mCACdC,UACAC,SAEcL,qCACD;CACb,OAAOI,SAASG,KAAKC,QAAQ;EAC3B,IAAIA,IAAIC,SAAS,QAAQ,OAAOD;EAEhC,MAAME,QAAQT,kCAAkCO,GAAG;EACnD,IAAI,CAACE,OAAOC,QAAQ,OAAOH;EAE3B,MAAMI,OAAO,GAAGF,MAAMH,IAAIF,MAAM,CAAC,CAACQ,KAAK,IAAI,EAAC;EAI5C,IADEL,IAAIO,QAAQ,EAAE,EAAEC,SAAS,UAAUR,IAAIO,MAAM,EAAE,CAACH,SAASA,MACtC,OAAOJ;EAE5B,OAAO;GACL,GAAGA;GACHO,OAAO,CAAC;IAAEC,MAAM;IAAiBJ;GAAK,GAAG,GAAIJ,IAAIO,SAAS,CAAA,CAAG;EAC/D;CACF,CAAC;AACH"}

package/dist/model-context/injectQuoteContext.d.ts

@@ -13,6 +13,7 @@ import type { UIMessage } from "ai";
  * export async function POST(req: Request) {
  *   const { messages } = await req.json();
  *   const result = streamText({
+ *     abortSignal: req.signal,
  *     model: myModel,
  *     messages: await convertToModelMessages(injectQuoteContext(messages)),
  *   });

package/dist/model-context/injectQuoteContext.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"injectQuoteContext.d.ts","sourceRoot":"","sources":["../../src/model-context/injectQuoteContext.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,IAAI,CAAC;AAepC;;;;;;;;;;;;;;;;;;;;GAoBG;AACH,wBAAgB,kBAAkB,CAAC,QAAQ,EAAE,SAAS,EAAE,GAAG,SAAS,EAAE,CAyBrE"}
+{"version":3,"file":"injectQuoteContext.d.ts","sourceRoot":"","sources":["../../src/model-context/injectQuoteContext.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,IAAI,CAAC;AAepC;;;;;;;;;;;;;;;;;;;;;GAqBG;AACH,wBAAgB,kBAAkB,CAAC,QAAQ,EAAE,SAAS,EAAE,GAAG,SAAS,EAAE,CAyBrE"}

package/dist/model-context/injectQuoteContext.js

@@ -22,6 +22,7 @@ const getQuoteText = (metadata) => {
 * export async function POST(req: Request) {
 *   const { messages } = await req.json();
 *   const result = streamText({
+*     abortSignal: req.signal,
 *     model: myModel,
 *     messages: await convertToModelMessages(injectQuoteContext(messages)),
 *   });

package/dist/model-context/injectQuoteContext.js.map

@@ -1 +1 @@
-{"version":3,"file":"injectQuoteContext.js","names":["UIMessage","getQuoteText","metadata","undefined","custom","Record","quote","text","injectQuoteContext","messages","map","msg","role","blockquote","split","line","join","alreadyInjected","parts","type","const"],"sources":["../../src/model-context/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/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,MAAMC,gBAAgBC,aAA0C;CAC9D,IAAI,CAACA,YAAY,OAAOA,aAAa,UAAU,OAAOC,KAAAA;CAEtD,MAAMC,SAAUF,SAAqCE;CACrD,IAAI,CAACA,UAAU,OAAOA,WAAW,UAAU,OAAOD,KAAAA;CAElD,MAAMG,QAASF,OAAmCE;CAClD,IAAI,CAACA,SAAS,OAAOA,UAAU,UAAU,OAAOH,KAAAA;CAEhD,MAAMI,OAAQD,MAAkCC;CAChD,OAAO,OAAOA,SAAS,WAAWA,OAAOJ,KAAAA;AAC3C;;;;;;;;;;;;;;;;;;;;;;AAuBA,SAAgBK,mBAAmBC,UAAoC;CACrE,OAAOA,SAASC,KAAKC,QAAQ;EAC3B,IAAIA,IAAIC,SAAS,QAAQ,OAAOD;EAEhC,MAAMJ,OAAON,aAAaU,IAAIT,QAAQ;EACtC,IAAI,CAACK,MAAM,OAAOI;EAElB,MAAME,aAAaN,KAChBO,MAAM,OAAO,CAAC,CACdJ,KAAKK,SAAS,KAAKA,MAAM,CAAC,CAC1BC,KAAK,IAAI;EAKZ,IAFEL,IAAIO,QAAQ,EAAE,EAAEC,SAAS,UACzBR,IAAIO,MAAM,EAAE,CAACX,SAAS,GAAGM,WAAU,OAChB,OAAOF;EAE5B,OAAO;GACL,GAAGA;GACHO,OAAO,CACL;IAAEC,MAAM;IAAiBZ,MAAM,GAAGM,WAAU;GAAO,GACnD,GAAIF,IAAIO,SAAS,CAAA,CAAG;EAExB;CACF,CAAC;AACH"}
+{"version":3,"file":"injectQuoteContext.js","names":["UIMessage","getQuoteText","metadata","undefined","custom","Record","quote","text","injectQuoteContext","messages","map","msg","role","blockquote","split","line","join","alreadyInjected","parts","type","const"],"sources":["../../src/model-context/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/ai-sdk\";\n *\n * export async function POST(req: Request) {\n *   const { messages } = await req.json();\n *   const result = streamText({\n *     abortSignal: req.signal,\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,MAAMC,gBAAgBC,aAA0C;CAC9D,IAAI,CAACA,YAAY,OAAOA,aAAa,UAAU,OAAOC,KAAAA;CAEtD,MAAMC,SAAUF,SAAqCE;CACrD,IAAI,CAACA,UAAU,OAAOA,WAAW,UAAU,OAAOD,KAAAA;CAElD,MAAMG,QAASF,OAAmCE;CAClD,IAAI,CAACA,SAAS,OAAOA,UAAU,UAAU,OAAOH,KAAAA;CAEhD,MAAMI,OAAQD,MAAkCC;CAChD,OAAO,OAAOA,SAAS,WAAWA,OAAOJ,KAAAA;AAC3C;;;;;;;;;;;;;;;;;;;;;;;AAwBA,SAAgBK,mBAAmBC,UAAoC;CACrE,OAAOA,SAASC,KAAKC,QAAQ;EAC3B,IAAIA,IAAIC,SAAS,QAAQ,OAAOD;EAEhC,MAAMJ,OAAON,aAAaU,IAAIT,QAAQ;EACtC,IAAI,CAACK,MAAM,OAAOI;EAElB,MAAME,aAAaN,KAChBO,MAAM,OAAO,CAAC,CACdJ,KAAKK,SAAS,KAAKA,MAAM,CAAC,CAC1BC,KAAK,IAAI;EAKZ,IAFEL,IAAIO,QAAQ,EAAE,EAAEC,SAAS,UACzBR,IAAIO,MAAM,EAAE,CAACX,SAAS,GAAGM,WAAU,OAChB,OAAOF;EAE5B,OAAO;GACL,GAAGA;GACHO,OAAO,CACL;IAAEC,MAAM;IAAiBZ,MAAM,GAAGM,WAAU;GAAO,GACnD,GAAIF,IAAIO,SAAS,CAAA,CAAG;EAExB;CACF,CAAC;AACH"}

package/dist/runtime/AISDKChat.js

@@ -1,8 +1,8 @@
 "use client";
 import { useChatThread } from "./useChatThread.js";
 import { generateId } from "ai";
-import { c } from "@assistant-ui/tap/react-shim/compiler-runtime";
 import { useState } from "@assistant-ui/tap/react-shim";
+import { c } from "@assistant-ui/tap/react-shim/compiler-runtime";
 import { resource, useResource } from "@assistant-ui/tap";
 import { RuntimeAdapter, runtimeAdapterTransformScopes } from "@assistant-ui/core/store";
 import { attachTransformScopes } from "@assistant-ui/store/client";

package/dist/runtime/AISDKThreads.d.ts

@@ -18,6 +18,11 @@ export type AISDKThreadsOptions<UI_MESSAGE extends UIMessage = UIMessage> = Omit
      * switch and stops on delete; per-thread history loads once per thread.
      */
     cloud?: AssistantCloud | undefined;
+    /**
+     * Stable identity for the account or workspace owning Cloud runtime state.
+     * Provide it from the first render and change it when that scope changes.
+     */
+    scopeId?: string | undefined;
     /**
      * Controlled thread id for the cloud list. Ignored without `cloud`.
      */

package/dist/runtime/AISDKThreads.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"AISDKThreads.d.ts","sourceRoot":"","sources":["../../src/runtime/AISDKThreads.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,EAAQ,SAAS,EAAE,MAAM,eAAe,CAAC;AACrD,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,IAAI,CAAC;AACxC,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,iBAAiB,CAAC;AAetD,OAAO,EAIL,KAAK,iBAAiB,EACvB,2BAAwB;AAKzB,MAAM,MAAM,mBAAmB,CAAC,UAAU,SAAS,SAAS,GAAG,SAAS,IACtE,IAAI,CAAC,iBAAiB,CAAC,UAAU,CAAC,EAAE,IAAI,GAAG,WAAW,GAAG,UAAU,CAAC,GAAG;IACrE;;;;;;OAMG;IACH,SAAS,CAAC,EACN,aAAa,CAAC,UAAU,CAAC,GACzB,CAAC,MAAM,aAAa,CAAC,UAAU,CAAC,CAAC,GACjC,SAAS,CAAC;IACd;;;;;OAKG;IACH,KAAK,CAAC,EAAE,cAAc,GAAG,SAAS,CAAC;IACnC;;OAEG;IACH,QAAQ,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAC9B;;OAEG;IACH,gBAAgB,CAAC,EAAE,CAAC,CAAC,QAAQ,EAAE,MAAM,GAAG,SAAS,KAAK,IAAI,CAAC,GAAG,SAAS,CAAC;CACzE,CAAC;AA8KJ;;;;;;;;;;;;GAYG;AACH,eAAO,MAAM,YAAY,GAjEA,UAAU,SAAS,SAAS,mOAiEA,CAAC"}
+{"version":3,"file":"AISDKThreads.d.ts","sourceRoot":"","sources":["../../src/runtime/AISDKThreads.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,EAAQ,SAAS,EAAE,MAAM,eAAe,CAAC;AACrD,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,IAAI,CAAC;AACxC,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,iBAAiB,CAAC;AAetD,OAAO,EAIL,KAAK,iBAAiB,EACvB,2BAAwB;AAKzB,MAAM,MAAM,mBAAmB,CAAC,UAAU,SAAS,SAAS,GAAG,SAAS,IACtE,IAAI,CAAC,iBAAiB,CAAC,UAAU,CAAC,EAAE,IAAI,GAAG,WAAW,GAAG,UAAU,CAAC,GAAG;IACrE;;;;;;OAMG;IACH,SAAS,CAAC,EACN,aAAa,CAAC,UAAU,CAAC,GACzB,CAAC,MAAM,aAAa,CAAC,UAAU,CAAC,CAAC,GACjC,SAAS,CAAC;IACd;;;;;OAKG;IACH,KAAK,CAAC,EAAE,cAAc,GAAG,SAAS,CAAC;IACnC;;;OAGG;IACH,OAAO,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAC7B;;OAEG;IACH,QAAQ,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAC9B;;OAEG;IACH,gBAAgB,CAAC,EAAE,CAAC,CAAC,QAAQ,EAAE,MAAM,GAAG,SAAS,KAAK,IAAI,CAAC,GAAG,SAAS,CAAC;CACzE,CAAC;AAoMJ;;;;;;;;;;;;GAYG;AACH,eAAO,MAAM,YAAY,GAvFA,UAAU,SAAS,SAAS,mOAuFA,CAAC"}

package/dist/runtime/AISDKThreads.js

@@ -5,9 +5,9 @@ import { createChat, splitChatThreadOptions, useChatThread } from "./useChatThre
 import { AI_SDK_SDK } from "./sdkIdentity.js";
 import { MessageRepository } from "@assistant-ui/core/internal";
 import { useCloudThreadListAdapter } from "@assistant-ui/core/react";
-import { c } from "@assistant-ui/tap/react-shim/compiler-runtime";
-import { useEffect, useState } from "@assistant-ui/tap/react-shim";
+import { useEffect, useRef, useState } from "@assistant-ui/tap/react-shim";
 import { useAui } from "@assistant-ui/store";
+import { c } from "@assistant-ui/tap/react-shim/compiler-runtime";
 import { resource, useResource, withKey } from "@assistant-ui/tap";
 import { InMemoryThreadList, RemoteThreadList, inMemoryThreadListTransformScopes } from "@assistant-ui/core/store";
 import { attachTransformScopes, useAssistantClientRef, useAssistantScopeEffect } from "@assistant-ui/store/client";
@@ -158,8 +158,8 @@ const useAISDKChatThread = (t0) => {
 };
 const AISDKChatThread = resource(useAISDKChatThread);
 const useAISDKThreads = (options) => {
-	const $ = c(2);
-	const { cloud, threadId, onThreadIdChange, ...threadOptions } = options ?? {};
+	const $ = c(7);
+	const { cloud, scopeId, threadId, onThreadIdChange, ...threadOptions } = options ?? {};
 	const [chats] = useState(_temp2);
 	const bindCloud = cloud !== void 0;
 	useEffect(() => {
@@ -176,8 +176,41 @@ const useAISDKThreads = (options) => {
 	useResourceCleanup(true, t0);
 	const cloudAdapter = useCloudThreadListAdapter({
 		cloud,
+		scopeId,
 		sdk: AI_SDK_SDK
 	});
+	const clientRef = useAssistantClientRef();
+	const previousCloudAdapter = useRef(bindCloud ? cloudAdapter : void 0);
+	let t1;
+	let t2;
+	if ($[2] !== bindCloud || $[3] !== clientRef || $[4] !== cloudAdapter) {
+		t1 = () => {
+			const previous = previousCloudAdapter.current;
+			previousCloudAdapter.current = bindCloud ? cloudAdapter : void 0;
+			if (!bindCloud || previous === void 0 || previous === cloudAdapter) return;
+			let cancelled = false;
+			queueMicrotask(() => {
+				if (!cancelled) clientRef.current.threads.reload();
+			});
+			return () => {
+				cancelled = true;
+			};
+		};
+		t2 = [
+			bindCloud,
+			clientRef,
+			cloudAdapter
+		];
+		$[2] = bindCloud;
+		$[3] = clientRef;
+		$[4] = cloudAdapter;
+		$[5] = t1;
+		$[6] = t2;
+	} else {
+		t1 = $[5];
+		t2 = $[6];
+	}
+	useEffect(t1, t2);
 	const thread = (id) => {
 		const element = AISDKChatThread({
 			threadId: id,

package/dist/runtime/AISDKThreads.js.map

@@ -1 +1 @@
-{"version":3,"file":"AISDKThreads.js","names":["c","_c","resource","useResource","withKey","useEffect","useMemo","useState","Chat","UIMessage","ChatTransport","AssistantCloud","InMemoryThreadList","RemoteThreadList","inMemoryThreadListTransformScopes","ThreadClient","useCloudThreadListAdapter","attachTransformScopes","useAssistantClientRef","useAssistantScopeEffect","useAui","AssistantChatTransport","createChat","splitChatThreadOptions","useChatThread","ChatThreadOptions","MessageRepository","useResourceCleanup","AI_SDK_SDK","AISDKThreadsOptions","Omit","UI_MESSAGE","transport","cloud","threadId","onThreadIdChange","AISDKThreadChatOptions","ChatOptionsRef","current","ChatEntry","chat","repository","optionsRef","createChatEntry","options","chatInit","undefined","__internal_clone","id","getOrCreateChatEntry","chats","Map","existing","get","created","set","useAISDKChatThread","t0","$","t1","owned","t2","t3","t4","t5","stop","catch","_temp","aui","t6","initialize","remoteId","externalId","fallbackItem","t7","t8","threadListItem","source","t9","isMainThread","getThreadListItem","messageRepositoryInstance","stopOnClientDestroy","runtime","clientRef","t10","registerModelContextProvider","modelContext","t11","t12","thread","AISDKChatThread","useAISDKThreads","threadOptions","_temp2","bindCloud","values","_temp3","cloudAdapter","sdk","element","adapter","backgroundThreads","onDelete","id_0","_temp4","delete","AISDKThreads","_c"],"sources":["../../src/runtime/AISDKThreads.ts"],"sourcesContent":["\"use client\";\n\nimport { resource, useResource, withKey } from \"@assistant-ui/tap\";\nimport { useEffect, useMemo, useState } from \"react\";\nimport type { Chat, UIMessage } from \"@ai-sdk/react\";\nimport type { ChatTransport } from \"ai\";\nimport type { AssistantCloud } from \"assistant-cloud\";\nimport {\n  InMemoryThreadList,\n  RemoteThreadList,\n  inMemoryThreadListTransformScopes,\n} from \"@assistant-ui/core/store\";\nimport { ThreadClient } from \"@assistant-ui/core/store/internal\";\nimport { useCloudThreadListAdapter } from \"@assistant-ui/core/react\";\nimport {\n  attachTransformScopes,\n  useAssistantClientRef,\n  useAssistantScopeEffect,\n} from \"@assistant-ui/store/client\";\nimport { useAui } from \"@assistant-ui/store\";\nimport { AssistantChatTransport } from \"../transport/AssistantChatTransport\";\nimport {\n  createChat,\n  splitChatThreadOptions,\n  useChatThread,\n  type ChatThreadOptions,\n} from \"./useChatThread\";\nimport { MessageRepository } from \"@assistant-ui/core/internal\";\nimport { useResourceCleanup } from \"./useResourceCleanup\";\nimport { AI_SDK_SDK } from \"./sdkIdentity\";\n\nexport type AISDKThreadsOptions<UI_MESSAGE extends UIMessage = UIMessage> =\n  Omit<ChatThreadOptions<UI_MESSAGE>, \"id\" | \"transport\" | \"messages\"> & {\n    /**\n     * The transport threads send through. A factory is invoked once per\n     * thread so each thread owns its instance. A plain\n     * `AssistantChatTransport` instance is cloned per thread (its\n     * assistant-ui wiring is per thread); any other transport instance is\n     * shared as-is. Defaults to one `AssistantChatTransport` per thread.\n     */\n    transport?:\n      | ChatTransport<UI_MESSAGE>\n      | (() => ChatTransport<UI_MESSAGE>)\n      | undefined;\n    /**\n     * When set, the thread list is a `RemoteThreadList` backed by this\n     * assistant-cloud. Omit it to keep the in-memory list. Every visited\n     * cloud thread stays mounted, so an in-flight run continues after a\n     * switch and stops on delete; per-thread history loads once per thread.\n     */\n    cloud?: AssistantCloud | undefined;\n    /**\n     * Controlled thread id for the cloud list. Ignored without `cloud`.\n     */\n    threadId?: string | undefined;\n    /**\n     * Called with the settled remote id when the cloud list changes thread.\n     */\n    onThreadIdChange?: ((threadId: string | undefined) => void) | undefined;\n  };\n\ntype AISDKThreadChatOptions<UI_MESSAGE extends UIMessage = UIMessage> = Omit<\n  AISDKThreadsOptions<UI_MESSAGE>,\n  \"cloud\" | \"threadId\" | \"onThreadIdChange\"\n>;\n\ntype ChatOptionsRef<UI_MESSAGE extends UIMessage> = {\n  current: AISDKThreadChatOptions<UI_MESSAGE> | undefined;\n};\n\ntype ChatEntry<UI_MESSAGE extends UIMessage> = {\n  chat: Chat<UI_MESSAGE>;\n  transport: ChatTransport<UI_MESSAGE>;\n  repository: MessageRepository;\n  optionsRef: ChatOptionsRef<UI_MESSAGE>;\n};\n\nconst createChatEntry = <UI_MESSAGE extends UIMessage>(\n  threadId: string,\n  options: AISDKThreadChatOptions<UI_MESSAGE> | undefined,\n): ChatEntry<UI_MESSAGE> => {\n  const optionsRef: ChatOptionsRef<UI_MESSAGE> = { current: options };\n  const { chatInit } = splitChatThreadOptions(\n    options as ChatThreadOptions<UI_MESSAGE> | undefined,\n  );\n  const transport =\n    typeof options?.transport === \"function\"\n      ? options.transport()\n      : options?.transport === undefined\n        ? new AssistantChatTransport()\n        : options.transport instanceof AssistantChatTransport\n          ? options.transport.__internal_clone()\n          : options.transport;\n  return {\n    chat: createChat({ ...chatInit, id: threadId, transport }, optionsRef),\n    transport,\n    repository: new MessageRepository(),\n    optionsRef,\n  };\n};\n\nconst getOrCreateChatEntry = <UI_MESSAGE extends UIMessage>(\n  threadId: string,\n  options: AISDKThreadChatOptions<UI_MESSAGE> | undefined,\n  chats: Map<string, ChatEntry<UI_MESSAGE>>,\n): ChatEntry<UI_MESSAGE> => {\n  const existing = chats.get(threadId);\n  if (existing) return existing;\n  const created = createChatEntry(threadId, options);\n  chats.set(threadId, created);\n  return created;\n};\n\nconst useAISDKChatThread = <UI_MESSAGE extends UIMessage = UIMessage>({\n  threadId,\n  options,\n  chats,\n  cloud,\n}: {\n  threadId: string;\n  options: AISDKThreadChatOptions<UI_MESSAGE> | undefined;\n  chats: Map<string, ChatEntry<UI_MESSAGE>>;\n  cloud: boolean;\n}) => {\n  const [owned] = useState(() =>\n    cloud ? createChatEntry(threadId, options) : undefined,\n  );\n  const { chat, transport, repository, optionsRef } =\n    owned ?? getOrCreateChatEntry(threadId, options, chats);\n\n  useEffect(() => {\n    if (cloud) optionsRef.current = options;\n  });\n\n  useEffect(() => {\n    if (!cloud) return undefined;\n    return () => {\n      void chat.stop().catch(() => {});\n    };\n  }, [chat, cloud]);\n\n  const aui = useAui();\n  const fallbackItem = useMemo(\n    () => ({\n      initialize: async () => ({\n        remoteId: threadId,\n        externalId: undefined,\n      }),\n    }),\n    [threadId],\n  );\n  const runtime = useChatThread(\n    { ...options, transport } as ChatThreadOptions<UI_MESSAGE>,\n    {\n      id: threadId,\n      isMainThread: true,\n      getThreadListItem: () =>\n        cloud\n          ? aui.threadListItem.source\n            ? aui.threadListItem\n            : undefined\n          : fallbackItem,\n      chat,\n      messageRepositoryInstance: repository,\n      stopOnClientDestroy: false,\n    },\n  );\n\n  const clientRef = useAssistantClientRef();\n  useAssistantScopeEffect(\n    \"modelContext\",\n    () =>\n      runtime.registerModelContextProvider(clientRef.current!.modelContext()),\n    [runtime],\n  );\n\n  return useResource(ThreadClient({ runtime: runtime.thread }));\n};\n\nconst AISDKChatThread = resource(useAISDKChatThread);\n\nconst useAISDKThreads = <UI_MESSAGE extends UIMessage = UIMessage>(\n  options?: AISDKThreadsOptions<UI_MESSAGE>,\n) => {\n  const { cloud, threadId, onThreadIdChange, ...threadOptions } = options ?? {};\n  const [chats] = useState(() => new Map<string, ChatEntry<UI_MESSAGE>>());\n  const bindCloud = cloud !== undefined;\n\n  useEffect(() => {\n    for (const { optionsRef } of chats.values()) {\n      optionsRef.current = threadOptions;\n    }\n  });\n\n  useResourceCleanup(true, () => {\n    for (const { chat } of chats.values()) {\n      void chat.stop().catch(() => {});\n    }\n  });\n\n  const cloudAdapter = useCloudThreadListAdapter({ cloud, sdk: AI_SDK_SDK });\n  const thread = (id: string) => {\n    const element = AISDKChatThread({\n      threadId: id,\n      options: threadOptions,\n      chats,\n      cloud: bindCloud,\n    });\n    return bindCloud ? withKey(id, element) : element;\n  };\n\n  return useResource(\n    bindCloud\n      ? RemoteThreadList({\n          adapter: cloudAdapter,\n          backgroundThreads: true,\n          thread,\n          threadId,\n          onThreadIdChange,\n        })\n      : InMemoryThreadList({\n          thread,\n          onDelete: (id) => {\n            void chats\n              .get(id)\n              ?.chat.stop()\n              .catch(() => {});\n            chats.delete(id);\n          },\n        }),\n  );\n};\n\n/**\n * `AuiConfig` entry that runs one AI SDK chat per thread. Hosts the same\n * per-thread orchestration as {@link AISDKChat} inside the client's own\n * resource tree, so it works with any `AssistantClient` host, React or not.\n * Without `cloud`, threads live in memory for the client's lifetime and keep\n * their history across switches; only the visible thread is mounted, and a\n * switched-away chat keeps streaming into its stored state until it settles\n * or the thread is deleted. With `cloud`, the list is a\n * `RemoteThreadList` with `backgroundThreads`: every visited thread stays\n * mounted with its own history, a run continues after a switch and stops on\n * delete, and a freshly created thread titles itself. Model context is\n * registered on every mounted thread.\n */\nexport const AISDKThreads = resource(useAISDKThreads);\n\nattachTransformScopes(useAISDKThreads, inMemoryThreadListTransformScopes);\n"],"mappings":";;;;;;;;;;;;;;;AA6EA,MAAM2C,mBACJT,UACAU,YAC0B;CAC1B,MAAMF,aAAyC,EAAEJ,SAASM,QAAQ;CAClE,MAAM,EAAEC,aAAatB,uBACnBqB,OACF;CACA,MAAMZ,YACJ,OAAOY,SAASZ,cAAc,aAC1BY,QAAQZ,UAAU,IAClBY,SAASZ,cAAcc,KAAAA,IACrB,IAAIzB,uBAAuB,IAC3BuB,QAAQZ,qBAAqBX,yBAC3BuB,QAAQZ,UAAUe,iBAAiB,IACnCH,QAAQZ;CAClB,OAAO;EACLQ,MAAMlB,WAAW;GAAE,GAAGuB;GAAUG,IAAId;GAAUF;EAAU,GAAGU,UAAU;EACrEV;EACAS,YAAY,IAAIf,kBAAkB;EAClCgB;CACF;AACF;AAEA,MAAMO,wBACJf,UACAU,SACAM,UAC0B;CAC1B,MAAME,WAAWF,MAAMG,IAAInB,QAAQ;CACnC,IAAIkB,UAAU,OAAOA;CACrB,MAAME,UAAUX,gBAAgBT,UAAUU,OAAO;CACjDM,MAAMK,IAAIrB,UAAUoB,OAAO;CAC3B,OAAOA;AACT;AAEA,MAAME,sBAAqBC,OAAA;CAAA,MAAAC,IAAAzD,EAAA,EAAA;CAA2C,MAAA,EAAAiC,UAAAU,SAAAM,OAAAjB,UAAAwB;CAUrE,IAAAE;CAAA,IAAAD,EAAA,OAAAzB,SAAAyB,EAAA,OAAAd,WAAAc,EAAA,OAAAxB,UAAA;EAC0ByB,WACvB1B,QAAQU,gBAAgBT,UAAUU,OAAmB,IAArDE,KAAAA;EAAsDY,EAAA,KAAAzB;EAAAyB,EAAA,KAAAd;EAAAc,EAAA,KAAAxB;EAAAwB,EAAA,KAAAC;CAAA,OAAAA,KAAAD,EAAA;CADxD,MAAA,CAAAE,SAAgBrD,SAASoD,EAEzB;CAAE,IAAAE;CAAA,IAAAH,EAAA,OAAAR,SAAAQ,EAAA,OAAAd,WAAAc,EAAA,OAAAE,SAAAF,EAAA,OAAAxB,UAAA;EAEA2B,KAAAD,SAASX,qBAAqBf,UAAUU,SAASM,KAAK;EAACQ,EAAA,KAAAR;EAAAQ,EAAA,KAAAd;EAAAc,EAAA,KAAAE;EAAAF,EAAA,KAAAxB;EAAAwB,EAAA,KAAAG;CAAA,OAAAA,KAAAH,EAAA;CADzD,MAAA,EAAAlB,MAAAR,WAAAS,YAAAC,eACEmB;CAAwD,IAAAC;CAAA,IAAAJ,EAAA,OAAAzB,SAAAyB,EAAA,QAAAd,WAAAc,EAAA,QAAAhB,YAAA;EAEhDoB,WAAA;GACR,IAAI7B,OAAOS,WAAUJ,UAAWM;EAAQ;EACzCc,EAAA,KAAAzB;EAAAyB,EAAA,MAAAd;EAAAc,EAAA,MAAAhB;EAAAgB,EAAA,MAAAI;CAAA,OAAAA,KAAAJ,EAAA;CAFDrD,UAAUyD,EAET;CAAC,IAAAC;CAAA,IAAAC;CAAA,IAAAN,EAAA,QAAAlB,QAAAkB,EAAA,QAAAzB,OAAA;EAEQ8B,WAAA;GACR,IAAI,CAAC9B,OAAK;GAAmB,aACtB;IACAO,KAAIyB,KAAM,CAAC,CAAAC,MAAOC,KAAQ;GAAC;EACjC;EACAH,KAAA,CAACxB,MAAMP,KAAK;EAACyB,EAAA,MAAAlB;EAAAkB,EAAA,MAAAzB;EAAAyB,EAAA,MAAAK;EAAAL,EAAA,MAAAM;CAAA,OAAA;EAAAD,KAAAL,EAAA;EAAAM,KAAAN,EAAA;CAAA;CALhBrD,UAAU0D,IAKPC,EAAa;CAEhB,MAAAI,MAAYhD,OAAO;CAAE,IAAAiD;CAAA,IAAAX,EAAA,QAAAxB,UAAA;EAEZmC,KAAA,EAAAC,YACO,aAAa;GAAAC,UACbrC;GAAQsC,YACN1B,KAAAA;EACd,GACF;EAACY,EAAA,MAAAxB;EAAAwB,EAAA,MAAAW;CAAA,OAAAA,KAAAX,EAAA;CANH,MAAAe,eACSJ;CAOP,IAAAK;CAAA,IAAAhB,EAAA,QAAAd,WAAAc,EAAA,QAAA1B,WAAA;EAEA0C,KAAA;GAAA,GAAK9B;GAAOZ;EAAY;EAAC0B,EAAA,MAAAd;EAAAc,EAAA,MAAA1B;EAAA0B,EAAA,MAAAgB;CAAA,OAAAA,KAAAhB,EAAA;CAAA,IAAAiB;CAAA,IAAAjB,EAAA,QAAAU,OAAAV,EAAA,QAAAzB,SAAAyB,EAAA,QAAAe,cAAA;EAIJE,WACjB1C,QACImC,IAAGQ,eAAeC,SAChBT,IAAGQ,iBADL9B,KAAAA,IADJ2B;EAIgBf,EAAA,MAAAU;EAAAV,EAAA,MAAAzB;EAAAyB,EAAA,MAAAe;EAAAf,EAAA,MAAAiB;CAAA,OAAAA,KAAAjB,EAAA;CAAA,IAAAoB;CAAA,IAAApB,EAAA,QAAAlB,QAAAkB,EAAA,QAAAjB,cAAAiB,EAAA,QAAAiB,MAAAjB,EAAA,QAAAxB,UAAA;EARpB4C,KAAA;GAAA9B,IACMd;GAAQ6C,cACE;GAAIC,mBACCL;GAKDnC;GAAAyC,2BAESxC;GAAUyC,qBAChB;EACvB;EAACxB,EAAA,MAAAlB;EAAAkB,EAAA,MAAAjB;EAAAiB,EAAA,MAAAiB;EAAAjB,EAAA,MAAAxB;EAAAwB,EAAA,MAAAoB;CAAA,OAAAA,KAAApB,EAAA;CAdH,MAAAyB,UAAgB3D,cACdkD,IACAI,EAaF;CAEA,MAAAM,YAAkBlE,sBAAsB;CAAE,IAAAmE;CAAA,IAAA3B,EAAA,QAAA0B,aAAA1B,EAAA,QAAAyB,SAAA;EAGxCE,YACEF,QAAOG,6BAA8BF,UAAS9C,QAAQiD,aAAe,CAAC;EAAC7B,EAAA,MAAA0B;EAAA1B,EAAA,MAAAyB;EAAAzB,EAAA,MAAA2B;CAAA,OAAAA,MAAA3B,EAAA;CAAA,IAAA8B;CAAA,IAAA9B,EAAA,QAAAyB,SAAA;EACzEK,MAAA,CAACL,OAAO;EAACzB,EAAA,MAAAyB;EAAAzB,EAAA,MAAA8B;CAAA,OAAAA,MAAA9B,EAAA;CAJXvC,wBACE,gBACAkE,KAEAG,GACF;CAAC,IAAAC;CAAA,IAAA/B,EAAA,QAAAyB,QAAAO,QAAA;EAEkBD,MAAA1E,aAAa,EAAAoE,SAAWA,QAAOO,OAAQ,CAAC;EAAChC,EAAA,MAAAyB,QAAAO;EAAAhC,EAAA,MAAA+B;CAAA,OAAAA,MAAA/B,EAAA;CAAA,OAArDvD,YAAYsF,GAAyC;AAAC;AAG/D,MAAME,kBAAkBzF,SAASsD,kBAAkB;AAEnD,MAAMoC,mBAAkBhD,YAAA;CAAA,MAAAc,IAAAzD,EAAA,CAAA;CAGtB,MAAA,EAAAgC,OAAAC,UAAAC,kBAAA,GAAA0D,kBAAgEjD,WAAA,CAAY;CAC5E,MAAA,CAAAM,SAAgB3C,SAASuF,MAA8C;CACvE,MAAAC,YAAkB9D,UAAUa,KAAAA;CAE5BzC,gBAAU;EACR,KAAK,MAAA,EAAAqC,gBAAwBQ,MAAK8C,OAAQ,GACxCtD,WAAUJ,UAAWuD;CACtB,CACF;CAAC,IAAApC;CAAA,IAAAC,EAAA,OAAAR,OAAA;EAEuBO,WAAA;GACvB,KAAK,MAAA,EAAAjB,UAAkBU,MAAK8C,OAAQ,GAC7BxD,KAAIyB,KAAM,CAAC,CAAAC,MAAO+B,MAAQ;EAChC;EACFvC,EAAA,KAAAR;EAAAQ,EAAA,KAAAD;CAAA,OAAAA,KAAAC,EAAA;CAJD/B,mBAAmB,MAAM8B,EAIxB;CAED,MAAAyC,eAAqBlF,0BAA0B;EAAAiB;EAAAkE,KAAcvE;CAAW,CAAC;CACzE,MAAA8D,UAAe1C,OAAA;EACb,MAAAoD,UAAgBT,gBAAgB;GAAAzD,UACpBc;GAAEJ,SACHiD;GAAa3C;GAAAjB,OAEf8D;EACT,CAAC;EAAE,OACIA,YAAY3F,QAAQ4C,IAAIoD,OAAiB,IAAzCA;CAA0C;CACjD,OAEKjG,YACL4F,YACIlF,iBAAiB;EAAAwF,SACNH;EAAYI,mBACF;EAAIZ;EAAAxD;EAAAC;CAIzB,CAUC,IATDvB,mBAAmB;EAAA8E;EAAAa,WAEPC,SAAA;GACHtD,MAAKG,IACHL,IACA,CAAC,EAAAR,KAAKyB,KACN,CAAC,CAAAC,MAACuC,MAAQ;GACjBvD,MAAKwD,OAAQ1D,IAAE;EAAC;CAEpB,CAAC,CACP;AAAC;;;;;;;;;;;;;;AAgBH,MAAa2D,eAAezG,SAAS0F,eAAe;AAEpD3E,sBAAsB2E,iBAAiB9E,iCAAiC;AAvI7C,SAAAqD,QAAA,CAAA;AAoEH,SAAA2B,SAAA;CAAA,uBAIS,IAAI3C,IAAmC;AAAC;AAJjD,SAAA8C,SAAA,CAAA;AAAA,SAAAQ,SAAA,CAAA"}
+{"version":3,"file":"AISDKThreads.js","names":["c","_c","resource","useResource","withKey","useEffect","useMemo","useRef","useState","Chat","UIMessage","ChatTransport","AssistantCloud","InMemoryThreadList","RemoteThreadList","inMemoryThreadListTransformScopes","ThreadClient","useCloudThreadListAdapter","attachTransformScopes","useAssistantClientRef","useAssistantScopeEffect","useAui","AssistantChatTransport","createChat","splitChatThreadOptions","useChatThread","ChatThreadOptions","MessageRepository","useResourceCleanup","AI_SDK_SDK","AISDKThreadsOptions","Omit","UI_MESSAGE","transport","cloud","scopeId","threadId","onThreadIdChange","AISDKThreadChatOptions","ChatOptionsRef","current","ChatEntry","chat","repository","optionsRef","createChatEntry","options","chatInit","undefined","__internal_clone","id","getOrCreateChatEntry","chats","Map","existing","get","created","set","useAISDKChatThread","t0","$","t1","owned","t2","t3","t4","t5","stop","catch","_temp","aui","t6","initialize","remoteId","externalId","fallbackItem","t7","t8","threadListItem","source","t9","isMainThread","getThreadListItem","messageRepositoryInstance","stopOnClientDestroy","runtime","clientRef","t10","registerModelContextProvider","modelContext","t11","t12","thread","AISDKChatThread","useAISDKThreads","threadOptions","_temp2","bindCloud","values","_temp3","cloudAdapter","sdk","previousCloudAdapter","previous","cancelled","queueMicrotask","threads","reload","element","adapter","backgroundThreads","onDelete","id_0","_temp4","delete","AISDKThreads","_c"],"sources":["../../src/runtime/AISDKThreads.ts"],"sourcesContent":["\"use client\";\n\nimport { resource, useResource, withKey } from \"@assistant-ui/tap\";\nimport { useEffect, useMemo, useRef, useState } from \"react\";\nimport type { Chat, UIMessage } from \"@ai-sdk/react\";\nimport type { ChatTransport } from \"ai\";\nimport type { AssistantCloud } from \"assistant-cloud\";\nimport {\n  InMemoryThreadList,\n  RemoteThreadList,\n  inMemoryThreadListTransformScopes,\n} from \"@assistant-ui/core/store\";\nimport { ThreadClient } from \"@assistant-ui/core/store/internal\";\nimport { useCloudThreadListAdapter } from \"@assistant-ui/core/react\";\nimport {\n  attachTransformScopes,\n  useAssistantClientRef,\n  useAssistantScopeEffect,\n} from \"@assistant-ui/store/client\";\nimport { useAui } from \"@assistant-ui/store\";\nimport { AssistantChatTransport } from \"../transport/AssistantChatTransport\";\nimport {\n  createChat,\n  splitChatThreadOptions,\n  useChatThread,\n  type ChatThreadOptions,\n} from \"./useChatThread\";\nimport { MessageRepository } from \"@assistant-ui/core/internal\";\nimport { useResourceCleanup } from \"./useResourceCleanup\";\nimport { AI_SDK_SDK } from \"./sdkIdentity\";\n\nexport type AISDKThreadsOptions<UI_MESSAGE extends UIMessage = UIMessage> =\n  Omit<ChatThreadOptions<UI_MESSAGE>, \"id\" | \"transport\" | \"messages\"> & {\n    /**\n     * The transport threads send through. A factory is invoked once per\n     * thread so each thread owns its instance. A plain\n     * `AssistantChatTransport` instance is cloned per thread (its\n     * assistant-ui wiring is per thread); any other transport instance is\n     * shared as-is. Defaults to one `AssistantChatTransport` per thread.\n     */\n    transport?:\n      | ChatTransport<UI_MESSAGE>\n      | (() => ChatTransport<UI_MESSAGE>)\n      | undefined;\n    /**\n     * When set, the thread list is a `RemoteThreadList` backed by this\n     * assistant-cloud. Omit it to keep the in-memory list. Every visited\n     * cloud thread stays mounted, so an in-flight run continues after a\n     * switch and stops on delete; per-thread history loads once per thread.\n     */\n    cloud?: AssistantCloud | undefined;\n    /**\n     * Stable identity for the account or workspace owning Cloud runtime state.\n     * Provide it from the first render and change it when that scope changes.\n     */\n    scopeId?: string | undefined;\n    /**\n     * Controlled thread id for the cloud list. Ignored without `cloud`.\n     */\n    threadId?: string | undefined;\n    /**\n     * Called with the settled remote id when the cloud list changes thread.\n     */\n    onThreadIdChange?: ((threadId: string | undefined) => void) | undefined;\n  };\n\ntype AISDKThreadChatOptions<UI_MESSAGE extends UIMessage = UIMessage> = Omit<\n  AISDKThreadsOptions<UI_MESSAGE>,\n  \"cloud\" | \"scopeId\" | \"threadId\" | \"onThreadIdChange\"\n>;\n\ntype ChatOptionsRef<UI_MESSAGE extends UIMessage> = {\n  current: AISDKThreadChatOptions<UI_MESSAGE> | undefined;\n};\n\ntype ChatEntry<UI_MESSAGE extends UIMessage> = {\n  chat: Chat<UI_MESSAGE>;\n  transport: ChatTransport<UI_MESSAGE>;\n  repository: MessageRepository;\n  optionsRef: ChatOptionsRef<UI_MESSAGE>;\n};\n\nconst createChatEntry = <UI_MESSAGE extends UIMessage>(\n  threadId: string,\n  options: AISDKThreadChatOptions<UI_MESSAGE> | undefined,\n): ChatEntry<UI_MESSAGE> => {\n  const optionsRef: ChatOptionsRef<UI_MESSAGE> = { current: options };\n  const { chatInit } = splitChatThreadOptions(\n    options as ChatThreadOptions<UI_MESSAGE> | undefined,\n  );\n  const transport =\n    typeof options?.transport === \"function\"\n      ? options.transport()\n      : options?.transport === undefined\n        ? new AssistantChatTransport()\n        : options.transport instanceof AssistantChatTransport\n          ? options.transport.__internal_clone()\n          : options.transport;\n  return {\n    chat: createChat({ ...chatInit, id: threadId, transport }, optionsRef),\n    transport,\n    repository: new MessageRepository(),\n    optionsRef,\n  };\n};\n\nconst getOrCreateChatEntry = <UI_MESSAGE extends UIMessage>(\n  threadId: string,\n  options: AISDKThreadChatOptions<UI_MESSAGE> | undefined,\n  chats: Map<string, ChatEntry<UI_MESSAGE>>,\n): ChatEntry<UI_MESSAGE> => {\n  const existing = chats.get(threadId);\n  if (existing) return existing;\n  const created = createChatEntry(threadId, options);\n  chats.set(threadId, created);\n  return created;\n};\n\nconst useAISDKChatThread = <UI_MESSAGE extends UIMessage = UIMessage>({\n  threadId,\n  options,\n  chats,\n  cloud,\n}: {\n  threadId: string;\n  options: AISDKThreadChatOptions<UI_MESSAGE> | undefined;\n  chats: Map<string, ChatEntry<UI_MESSAGE>>;\n  cloud: boolean;\n}) => {\n  const [owned] = useState(() =>\n    cloud ? createChatEntry(threadId, options) : undefined,\n  );\n  const { chat, transport, repository, optionsRef } =\n    owned ?? getOrCreateChatEntry(threadId, options, chats);\n\n  useEffect(() => {\n    if (cloud) optionsRef.current = options;\n  });\n\n  useEffect(() => {\n    if (!cloud) return undefined;\n    return () => {\n      void chat.stop().catch(() => {});\n    };\n  }, [chat, cloud]);\n\n  const aui = useAui();\n  const fallbackItem = useMemo(\n    () => ({\n      initialize: async () => ({\n        remoteId: threadId,\n        externalId: undefined,\n      }),\n    }),\n    [threadId],\n  );\n  const runtime = useChatThread(\n    { ...options, transport } as ChatThreadOptions<UI_MESSAGE>,\n    {\n      id: threadId,\n      isMainThread: true,\n      getThreadListItem: () =>\n        cloud\n          ? aui.threadListItem.source\n            ? aui.threadListItem\n            : undefined\n          : fallbackItem,\n      chat,\n      messageRepositoryInstance: repository,\n      stopOnClientDestroy: false,\n    },\n  );\n\n  const clientRef = useAssistantClientRef();\n  useAssistantScopeEffect(\n    \"modelContext\",\n    () =>\n      runtime.registerModelContextProvider(clientRef.current!.modelContext()),\n    [runtime],\n  );\n\n  return useResource(ThreadClient({ runtime: runtime.thread }));\n};\n\nconst AISDKChatThread = resource(useAISDKChatThread);\n\nconst useAISDKThreads = <UI_MESSAGE extends UIMessage = UIMessage>(\n  options?: AISDKThreadsOptions<UI_MESSAGE>,\n) => {\n  const { cloud, scopeId, threadId, onThreadIdChange, ...threadOptions } =\n    options ?? {};\n  const [chats] = useState(() => new Map<string, ChatEntry<UI_MESSAGE>>());\n  const bindCloud = cloud !== undefined;\n\n  useEffect(() => {\n    for (const { optionsRef } of chats.values()) {\n      optionsRef.current = threadOptions;\n    }\n  });\n\n  useResourceCleanup(true, () => {\n    for (const { chat } of chats.values()) {\n      void chat.stop().catch(() => {});\n    }\n  });\n\n  const cloudAdapter = useCloudThreadListAdapter({\n    cloud,\n    scopeId,\n    sdk: AI_SDK_SDK,\n  });\n  const clientRef = useAssistantClientRef();\n  const previousCloudAdapter = useRef(bindCloud ? cloudAdapter : undefined);\n  useEffect(() => {\n    const previous = previousCloudAdapter.current;\n    previousCloudAdapter.current = bindCloud ? cloudAdapter : undefined;\n    if (!bindCloud || previous === undefined || previous === cloudAdapter)\n      return;\n    // Wait until RemoteThreadList commits the replacement adapter before\n    // requesting its explicit reload.\n    let cancelled = false;\n    queueMicrotask(() => {\n      if (!cancelled) void clientRef.current!.threads.reload();\n    });\n    return () => {\n      cancelled = true;\n    };\n  }, [bindCloud, clientRef, cloudAdapter]);\n  const thread = (id: string) => {\n    const element = AISDKChatThread({\n      threadId: id,\n      options: threadOptions,\n      chats,\n      cloud: bindCloud,\n    });\n    return bindCloud ? withKey(id, element) : element;\n  };\n\n  return useResource(\n    bindCloud\n      ? RemoteThreadList({\n          adapter: cloudAdapter,\n          backgroundThreads: true,\n          thread,\n          threadId,\n          onThreadIdChange,\n        })\n      : InMemoryThreadList({\n          thread,\n          onDelete: (id) => {\n            void chats\n              .get(id)\n              ?.chat.stop()\n              .catch(() => {});\n            chats.delete(id);\n          },\n        }),\n  );\n};\n\n/**\n * `AuiConfig` entry that runs one AI SDK chat per thread. Hosts the same\n * per-thread orchestration as {@link AISDKChat} inside the client's own\n * resource tree, so it works with any `AssistantClient` host, React or not.\n * Without `cloud`, threads live in memory for the client's lifetime and keep\n * their history across switches; only the visible thread is mounted, and a\n * switched-away chat keeps streaming into its stored state until it settles\n * or the thread is deleted. With `cloud`, the list is a\n * `RemoteThreadList` with `backgroundThreads`: every visited thread stays\n * mounted with its own history, a run continues after a switch and stops on\n * delete, and a freshly created thread titles itself. Model context is\n * registered on every mounted thread.\n */\nexport const AISDKThreads = resource(useAISDKThreads);\n\nattachTransformScopes(useAISDKThreads, inMemoryThreadListTransformScopes);\n"],"mappings":";;;;;;;;;;;;;;;AAkFA,MAAM6C,mBACJT,UACAU,YAC0B;CAC1B,MAAMF,aAAyC,EAAEJ,SAASM,QAAQ;CAClE,MAAM,EAAEC,aAAavB,uBACnBsB,OACF;CACA,MAAMb,YACJ,OAAOa,SAASb,cAAc,aAC1Ba,QAAQb,UAAU,IAClBa,SAASb,cAAce,KAAAA,IACrB,IAAI1B,uBAAuB,IAC3BwB,QAAQb,qBAAqBX,yBAC3BwB,QAAQb,UAAUgB,iBAAiB,IACnCH,QAAQb;CAClB,OAAO;EACLS,MAAMnB,WAAW;GAAE,GAAGwB;GAAUG,IAAId;GAAUH;EAAU,GAAGW,UAAU;EACrEX;EACAU,YAAY,IAAIhB,kBAAkB;EAClCiB;CACF;AACF;AAEA,MAAMO,wBACJf,UACAU,SACAM,UAC0B;CAC1B,MAAME,WAAWF,MAAMG,IAAInB,QAAQ;CACnC,IAAIkB,UAAU,OAAOA;CACrB,MAAME,UAAUX,gBAAgBT,UAAUU,OAAO;CACjDM,MAAMK,IAAIrB,UAAUoB,OAAO;CAC3B,OAAOA;AACT;AAEA,MAAME,sBAAqBC,OAAA;CAAA,MAAAC,IAAA3D,EAAA,EAAA;CAA2C,MAAA,EAAAmC,UAAAU,SAAAM,OAAAlB,UAAAyB;CAUrE,IAAAE;CAAA,IAAAD,EAAA,OAAA1B,SAAA0B,EAAA,OAAAd,WAAAc,EAAA,OAAAxB,UAAA;EAC0ByB,WACvB3B,QAAQW,gBAAgBT,UAAUU,OAAmB,IAArDE,KAAAA;EAAsDY,EAAA,KAAA1B;EAAA0B,EAAA,KAAAd;EAAAc,EAAA,KAAAxB;EAAAwB,EAAA,KAAAC;CAAA,OAAAA,KAAAD,EAAA;CADxD,MAAA,CAAAE,SAAgBtD,SAASqD,EAEzB;CAAE,IAAAE;CAAA,IAAAH,EAAA,OAAAR,SAAAQ,EAAA,OAAAd,WAAAc,EAAA,OAAAE,SAAAF,EAAA,OAAAxB,UAAA;EAEA2B,KAAAD,SAASX,qBAAqBf,UAAUU,SAASM,KAAK;EAACQ,EAAA,KAAAR;EAAAQ,EAAA,KAAAd;EAAAc,EAAA,KAAAE;EAAAF,EAAA,KAAAxB;EAAAwB,EAAA,KAAAG;CAAA,OAAAA,KAAAH,EAAA;CADzD,MAAA,EAAAlB,MAAAT,WAAAU,YAAAC,eACEmB;CAAwD,IAAAC;CAAA,IAAAJ,EAAA,OAAA1B,SAAA0B,EAAA,QAAAd,WAAAc,EAAA,QAAAhB,YAAA;EAEhDoB,WAAA;GACR,IAAI9B,OAAOU,WAAUJ,UAAWM;EAAQ;EACzCc,EAAA,KAAA1B;EAAA0B,EAAA,MAAAd;EAAAc,EAAA,MAAAhB;EAAAgB,EAAA,MAAAI;CAAA,OAAAA,KAAAJ,EAAA;CAFDvD,UAAU2D,EAET;CAAC,IAAAC;CAAA,IAAAC;CAAA,IAAAN,EAAA,QAAAlB,QAAAkB,EAAA,QAAA1B,OAAA;EAEQ+B,WAAA;GACR,IAAI,CAAC/B,OAAK;GAAmB,aACtB;IACAQ,KAAIyB,KAAM,CAAC,CAAAC,MAAOC,KAAQ;GAAC;EACjC;EACAH,KAAA,CAACxB,MAAMR,KAAK;EAAC0B,EAAA,MAAAlB;EAAAkB,EAAA,MAAA1B;EAAA0B,EAAA,MAAAK;EAAAL,EAAA,MAAAM;CAAA,OAAA;EAAAD,KAAAL,EAAA;EAAAM,KAAAN,EAAA;CAAA;CALhBvD,UAAU4D,IAKPC,EAAa;CAEhB,MAAAI,MAAYjD,OAAO;CAAE,IAAAkD;CAAA,IAAAX,EAAA,QAAAxB,UAAA;EAEZmC,KAAA,EAAAC,YACO,aAAa;GAAAC,UACbrC;GAAQsC,YACN1B,KAAAA;EACd,GACF;EAACY,EAAA,MAAAxB;EAAAwB,EAAA,MAAAW;CAAA,OAAAA,KAAAX,EAAA;CANH,MAAAe,eACSJ;CAOP,IAAAK;CAAA,IAAAhB,EAAA,QAAAd,WAAAc,EAAA,QAAA3B,WAAA;EAEA2C,KAAA;GAAA,GAAK9B;GAAOb;EAAY;EAAC2B,EAAA,MAAAd;EAAAc,EAAA,MAAA3B;EAAA2B,EAAA,MAAAgB;CAAA,OAAAA,KAAAhB,EAAA;CAAA,IAAAiB;CAAA,IAAAjB,EAAA,QAAAU,OAAAV,EAAA,QAAA1B,SAAA0B,EAAA,QAAAe,cAAA;EAIJE,WACjB3C,QACIoC,IAAGQ,eAAeC,SAChBT,IAAGQ,iBADL9B,KAAAA,IADJ2B;EAIgBf,EAAA,MAAAU;EAAAV,EAAA,MAAA1B;EAAA0B,EAAA,MAAAe;EAAAf,EAAA,MAAAiB;CAAA,OAAAA,KAAAjB,EAAA;CAAA,IAAAoB;CAAA,IAAApB,EAAA,QAAAlB,QAAAkB,EAAA,QAAAjB,cAAAiB,EAAA,QAAAiB,MAAAjB,EAAA,QAAAxB,UAAA;EARpB4C,KAAA;GAAA9B,IACMd;GAAQ6C,cACE;GAAIC,mBACCL;GAKDnC;GAAAyC,2BAESxC;GAAUyC,qBAChB;EACvB;EAACxB,EAAA,MAAAlB;EAAAkB,EAAA,MAAAjB;EAAAiB,EAAA,MAAAiB;EAAAjB,EAAA,MAAAxB;EAAAwB,EAAA,MAAAoB;CAAA,OAAAA,KAAApB,EAAA;CAdH,MAAAyB,UAAgB5D,cACdmD,IACAI,EAaF;CAEA,MAAAM,YAAkBnE,sBAAsB;CAAE,IAAAoE;CAAA,IAAA3B,EAAA,QAAA0B,aAAA1B,EAAA,QAAAyB,SAAA;EAGxCE,YACEF,QAAOG,6BAA8BF,UAAS9C,QAAQiD,aAAe,CAAC;EAAC7B,EAAA,MAAA0B;EAAA1B,EAAA,MAAAyB;EAAAzB,EAAA,MAAA2B;CAAA,OAAAA,MAAA3B,EAAA;CAAA,IAAA8B;CAAA,IAAA9B,EAAA,QAAAyB,SAAA;EACzEK,MAAA,CAACL,OAAO;EAACzB,EAAA,MAAAyB;EAAAzB,EAAA,MAAA8B;CAAA,OAAAA,MAAA9B,EAAA;CAJXxC,wBACE,gBACAmE,KAEAG,GACF;CAAC,IAAAC;CAAA,IAAA/B,EAAA,QAAAyB,QAAAO,QAAA;EAEkBD,MAAA3E,aAAa,EAAAqE,SAAWA,QAAOO,OAAQ,CAAC;EAAChC,EAAA,MAAAyB,QAAAO;EAAAhC,EAAA,MAAA+B;CAAA,OAAAA,MAAA/B,EAAA;CAAA,OAArDzD,YAAYwF,GAAyC;AAAC;AAG/D,MAAME,kBAAkB3F,SAASwD,kBAAkB;AAEnD,MAAMoC,mBAAkBhD,YAAA;CAAA,MAAAc,IAAA3D,EAAA,CAAA;CAGtB,MAAA,EAAAiC,OAAAC,SAAAC,UAAAC,kBAAA,GAAA0D,kBACEjD,WAAA,CAAY;CACd,MAAA,CAAAM,SAAgB5C,SAASwF,MAA8C;CACvE,MAAAC,YAAkB/D,UAAUc,KAAAA;CAE5B3C,gBAAU;EACR,KAAK,MAAA,EAAAuC,gBAAwBQ,MAAK8C,OAAQ,GACxCtD,WAAUJ,UAAWuD;CACtB,CACF;CAAC,IAAApC;CAAA,IAAAC,EAAA,OAAAR,OAAA;EAEuBO,WAAA;GACvB,KAAK,MAAA,EAAAjB,UAAkBU,MAAK8C,OAAQ,GAC7BxD,KAAIyB,KAAM,CAAC,CAAAC,MAAO+B,MAAQ;EAChC;EACFvC,EAAA,KAAAR;EAAAQ,EAAA,KAAAD;CAAA,OAAAA,KAAAC,EAAA;CAJDhC,mBAAmB,MAAM+B,EAIxB;CAED,MAAAyC,eAAqBnF,0BAA0B;EAAAiB;EAAAC;EAAAkE,KAGxCxE;CACP,CAAC;CACD,MAAAyD,YAAkBnE,sBAAsB;CACxC,MAAAmF,uBAA6B/F,OAAO0F,YAAAG,eAAApD,KAAAA,CAAoC;CAAE,IAAAa;CAAA,IAAAE;CAAA,IAAAH,EAAA,OAAAqC,aAAArC,EAAA,OAAA0B,aAAA1B,EAAA,OAAAwC,cAAA;EAChEvC,WAAA;GACR,MAAA0C,WAAiBD,qBAAoB9D;GACrC8D,qBAAoB9D,UAAWyD,YAAAG,eAAApD,KAAAA;GAC/B,IAAI,CAACiD,aAAaM,aAAavD,KAAAA,KAAauD,aAAaH,cAAY;GAIrE,IAAAI,YAAgB;GAChBC,qBAAe;IACb,IAAI,CAACD,WAAgBlB,UAAS9C,QAAQkE,QAASC,OAAQ;GAAE,CAC1D;GAAC,aACK;IACLH,YAAYA;GAAH;EACV;EACAzC,KAAA;GAACkC;GAAWX;GAAWc;EAAY;EAACxC,EAAA,KAAAqC;EAAArC,EAAA,KAAA0B;EAAA1B,EAAA,KAAAwC;EAAAxC,EAAA,KAAAC;EAAAD,EAAA,KAAAG;CAAA,OAAA;EAAAF,KAAAD,EAAA;EAAAG,KAAAH,EAAA;CAAA;CAdvCvD,UAAUwD,IAcPE,EAAoC;CACvC,MAAA6B,UAAe1C,OAAA;EACb,MAAA0D,UAAgBf,gBAAgB;GAAAzD,UACpBc;GAAEJ,SACHiD;GAAa3C;GAAAlB,OAEf+D;EACT,CAAC;EAAE,OACIA,YAAY7F,QAAQ8C,IAAI0D,OAAiB,IAAzCA;CAA0C;CACjD,OAEKzG,YACL8F,YACInF,iBAAiB;EAAA+F,SACNT;EAAYU,mBACF;EAAIlB;EAAAxD;EAAAC;CAIzB,CAUC,IATDxB,mBAAmB;EAAA+E;EAAAmB,WAEPC,SAAA;GACH5D,MAAKG,IACHL,IACA,CAAC,EAAAR,KAAKyB,KACN,CAAC,CAAAC,MAAC6C,MAAQ;GACjB7D,MAAK8D,OAAQhE,IAAE;EAAC;CAEpB,CAAC,CACP;AAAC;;;;;;;;;;;;;;AAgBH,MAAaiE,eAAejH,SAAS4F,eAAe;AAEpD5E,sBAAsB4E,iBAAiB/E,iCAAiC;AA7J7C,SAAAsD,QAAA,CAAA;AAoEH,SAAA2B,SAAA;CAAA,uBAKS,IAAI3C,IAAmC;AAAC;AALjD,SAAA8C,SAAA,CAAA;AAAA,SAAAc,SAAA,CAAA"}

package/dist/runtime/DynamicChatTransport.d.ts

@@ -0,0 +1,49 @@
+import type { AssistantRuntime } from "@assistant-ui/core";
+import type { UIMessage } from "@ai-sdk/react";
+import type { ChatTransport } from "ai";
+import { type InitializableThreadListItem } from "../transport/AssistantChatTransport.js";
+import type { ResumableClientStorage } from "../transport/resumable.js";
+export declare const getResumedStreamIds: (storage: ResumableClientStorage | undefined) => Set<string>;
+declare class DynamicResumableStorage implements ResumableClientStorage {
+    private readonly subscriptions;
+    private storage;
+    private resumedStreamIds;
+    private hasPendingNotification;
+    constructor(storage: ResumableClientStorage | undefined);
+    setStorage(storage: ResumableClientStorage | undefined): void;
+    flushChange(): void;
+    getResumedStreamIds(): Set<string>;
+    getStreamId(threadId?: string): string | null;
+    setStreamId(id: string, threadId?: string): void;
+    clear(threadId?: string): void;
+    subscribe(listener: () => void, threadId?: string): () => void;
+}
+type ThreadTransportBinding = {
+    runtime: AssistantRuntime;
+    getThreadListItem: () => InitializableThreadListItem | undefined;
+};
+export declare class DynamicChatTransport<UI_MESSAGE extends UIMessage> implements ChatTransport<UI_MESSAGE> {
+    private readonly threadContexts;
+    private transport;
+    private readonly resumableStorage;
+    constructor(transport: ChatTransport<UI_MESSAGE>);
+    readonly sendMessages: ChatTransport<UI_MESSAGE>["sendMessages"];
+    readonly reconnectToStream: ChatTransport<UI_MESSAGE>["reconnectToStream"];
+    readonly getCurrentTransport: (chatId: string) => ChatTransport<UI_MESSAGE>;
+    readonly getCurrentResumableStorage: () => DynamicResumableStorage;
+    readonly getResumedStreamIds: () => Set<string>;
+    createThreadProxy(owner: object, getBinding: () => ThreadTransportBinding): ChatTransport<UI_MESSAGE>;
+    setTransport(transport: ChatTransport<UI_MESSAGE>): void;
+    flushTransportChange(): void;
+    registerThread(chatId: string, owner: object): void;
+    setThreadContext(chatId: string, owner: object, runtime: AssistantRuntime, getThreadListItem: () => InitializableThreadListItem | undefined): void;
+    unregisterThread(chatId: string, owner: object): void;
+    private getTransport;
+    private getBoundTransport;
+    private getOrCreateBoundTransport;
+    private getThreadTransport;
+    private createThreadTransport;
+    private wireTransport;
+}
+export {};
+//# sourceMappingURL=DynamicChatTransport.d.ts.map

package/dist/runtime/DynamicChatTransport.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"DynamicChatTransport.d.ts","sourceRoot":"","sources":["../../src/runtime/DynamicChatTransport.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,oBAAoB,CAAC;AAC3D,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,eAAe,CAAC;AAC/C,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,IAAI,CAAC;AACxC,OAAO,EAEL,KAAK,2BAA2B,EACjC,+CAA4C;AAC7C,OAAO,KAAK,EAAE,sBAAsB,EAAE,kCAA+B;AAQrE,eAAO,MAAM,mBAAmB,GAC9B,SAAS,sBAAsB,GAAG,SAAS,gBAS5C,CAAC;AAQF,cAAM,uBAAwB,YAAW,sBAAsB;IAC7D,OAAO,CAAC,QAAQ,CAAC,aAAa,CAA2C;IACzE,OAAO,CAAC,OAAO,CAAqC;IACpD,OAAO,CAAC,gBAAgB,CAAc;IACtC,OAAO,CAAC,sBAAsB,CAAS;gBAE3B,OAAO,EAAE,sBAAsB,GAAG,SAAS;IAKhD,UAAU,CAAC,OAAO,EAAE,sBAAsB,GAAG,SAAS;IAWtD,WAAW;IAaX,mBAAmB;IAInB,WAAW,CAAC,QAAQ,CAAC,EAAE,MAAM;IAI7B,WAAW,CAAC,EAAE,EAAE,MAAM,EAAE,QAAQ,CAAC,EAAE,MAAM;IAIzC,KAAK,CAAC,QAAQ,CAAC,EAAE,MAAM;IAIvB,SAAS,CAAC,QAAQ,EAAE,MAAM,IAAI,EAAE,QAAQ,CAAC,EAAE,MAAM;CAYzD;AAYD,KAAK,sBAAsB,GAAG;IAC5B,OAAO,EAAE,gBAAgB,CAAC;IAC1B,iBAAiB,EAAE,MAAM,2BAA2B,GAAG,SAAS,CAAC;CAClE,CAAC;AAEF,qBAAa,oBAAoB,CAC/B,UAAU,SAAS,SAAS,CAC5B,YAAW,aAAa,CAAC,UAAU,CAAC;IACpC,OAAO,CAAC,QAAQ,CAAC,cAAc,CAG3B;IACJ,OAAO,CAAC,SAAS,CAA4B;IAC7C,OAAO,CAAC,QAAQ,CAAC,gBAAgB,CAA0B;gBAE/C,SAAS,EAAE,aAAa,CAAC,UAAU,CAAC;IAOhD,SAAgB,YAAY,EAAE,aAAa,CAAC,UAAU,CAAC,CAAC,cAAc,CAAC,CAEV;IAE7D,SAAgB,iBAAiB,EAAE,aAAa,CAAC,UAAU,CAAC,CAAC,mBAAmB,CAAC,CACL;IAE5E,SAAgB,mBAAmB,GAAI,QAAQ,MAAM,+BACwB;IAE7E,SAAgB,0BAA0B,gCAA+B;IAEzE,SAAgB,mBAAmB,oBACW;IAEvC,iBAAiB,CACtB,KAAK,EAAE,MAAM,EACb,UAAU,EAAE,MAAM,sBAAsB,GACvC,aAAa,CAAC,UAAU,CAAC;IAiBrB,YAAY,CAAC,SAAS,EAAE,aAAa,CAAC,UAAU,CAAC;IAMjD,oBAAoB;IAIpB,cAAc,CAAC,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM;IAQ5C,gBAAgB,CACrB,MAAM,EAAE,MAAM,EACd,KAAK,EAAE,MAAM,EACb,OAAO,EAAE,gBAAgB,EACzB,iBAAiB,EAAE,MAAM,2BAA2B,GAAG,SAAS;IAkB3D,gBAAgB,CAAC,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM;IAMrD,OAAO,CAAC,YAAY;IAUpB,OAAO,CAAC,iBAAiB;IAYzB,OAAO,CAAC,yBAAyB;IAWjC,OAAO,CAAC,kBAAkB;IAY1B,OAAO,CAAC,qBAAqB;IAM7B,OAAO,CAAC,aAAa;CAUtB"}

package/dist/runtime/DynamicChatTransport.js

@@ -0,0 +1,147 @@
+import { AssistantChatTransport } from "../transport/AssistantChatTransport.js";
+import { getResumableAdapter } from "./getResumableAdapter.js";
+//#region src/runtime/DynamicChatTransport.ts
+const resumedStreamIdsByStorage = /* @__PURE__ */ new WeakMap();
+const getResumedStreamIds = (storage) => {
+	if (!storage) return /* @__PURE__ */ new Set();
+	let resumedStreamIds = resumedStreamIdsByStorage.get(storage);
+	if (!resumedStreamIds) {
+		resumedStreamIds = /* @__PURE__ */ new Set();
+		resumedStreamIdsByStorage.set(storage, resumedStreamIds);
+	}
+	return resumedStreamIds;
+};
+var DynamicResumableStorage = class {
+	subscriptions = /* @__PURE__ */ new Set();
+	storage;
+	resumedStreamIds;
+	hasPendingNotification = false;
+	constructor(storage) {
+		this.storage = storage;
+		this.resumedStreamIds = getResumedStreamIds(storage);
+	}
+	setStorage(storage) {
+		if (this.storage === storage) return;
+		this.storage = storage;
+		const nextResumedStreamIds = getResumedStreamIds(storage);
+		for (const streamId of this.resumedStreamIds) nextResumedStreamIds.add(streamId);
+		this.resumedStreamIds = nextResumedStreamIds;
+		this.hasPendingNotification = true;
+	}
+	flushChange() {
+		if (!this.hasPendingNotification) return;
+		this.hasPendingNotification = false;
+		for (const subscription of this.subscriptions) {
+			subscription.unsubscribe?.();
+			subscription.unsubscribe = this.storage?.subscribe?.(subscription.listener, subscription.threadId);
+			subscription.listener();
+		}
+	}
+	getResumedStreamIds() {
+		return this.resumedStreamIds;
+	}
+	getStreamId(threadId) {
+		return this.storage?.getStreamId(threadId) ?? null;
+	}
+	setStreamId(id, threadId) {
+		this.storage?.setStreamId(id, threadId);
+	}
+	clear(threadId) {
+		this.storage?.clear(threadId);
+	}
+	subscribe(listener, threadId) {
+		const subscription = {
+			listener,
+			threadId,
+			unsubscribe: this.storage?.subscribe?.(listener, threadId)
+		};
+		this.subscriptions.add(subscription);
+		return () => {
+			this.subscriptions.delete(subscription);
+			subscription.unsubscribe?.();
+		};
+	}
+};
+var DynamicChatTransport = class {
+	threadContexts = /* @__PURE__ */ new Map();
+	transport;
+	resumableStorage;
+	constructor(transport) {
+		this.transport = transport;
+		this.resumableStorage = new DynamicResumableStorage(getResumableAdapter(transport)?.storage);
+	}
+	sendMessages = (options) => this.getTransport(options.chatId).sendMessages(options);
+	reconnectToStream = (options) => this.getTransport(options.chatId).reconnectToStream(options);
+	getCurrentTransport = (chatId) => this.getThreadTransport(this.threadContexts.get(chatId)) ?? this.transport;
+	getCurrentResumableStorage = () => this.resumableStorage;
+	getResumedStreamIds = () => this.resumableStorage.getResumedStreamIds();
+	createThreadProxy(owner, getBinding) {
+		return {
+			sendMessages: (options) => this.getOrCreateBoundTransport(options.chatId, owner, getBinding).sendMessages(options),
+			reconnectToStream: (options) => this.getOrCreateBoundTransport(options.chatId, owner, getBinding).reconnectToStream(options)
+		};
+	}
+	setTransport(transport) {
+		if (this.transport === transport) return;
+		this.transport = transport;
+		this.resumableStorage.setStorage(getResumableAdapter(transport)?.storage);
+	}
+	flushTransportChange() {
+		this.resumableStorage.flushChange();
+	}
+	registerThread(chatId, owner) {
+		if (this.threadContexts.get(chatId)?.owner === owner) return;
+		this.threadContexts.set(chatId, { owner });
+	}
+	setThreadContext(chatId, owner, runtime, getThreadListItem) {
+		this.registerThread(chatId, owner);
+		const context = this.threadContexts.get(chatId);
+		context.runtime = runtime;
+		context.getThreadListItem = getThreadListItem;
+		if (context.transport === void 0) {
+			this.getThreadTransport(context);
+			return;
+		}
+		if (context.sourceTransport === this.transport && context.transport !== void 0) this.wireTransport(context, context.transport);
+	}
+	unregisterThread(chatId, owner) {
+		if (this.threadContexts.get(chatId)?.owner === owner) this.threadContexts.delete(chatId);
+	}
+	getTransport(chatId) {
+		const context = this.threadContexts.get(chatId);
+		if (!context) throw new Error(`DynamicChatTransport has no registered context for chat "${chatId}"`);
+		return this.getThreadTransport(context);
+	}
+	getBoundTransport(chatId, owner, binding) {
+		this.registerThread(chatId, owner);
+		const context = this.threadContexts.get(chatId);
+		context.runtime = binding.runtime;
+		context.getThreadListItem = binding.getThreadListItem;
+		return this.getThreadTransport(context);
+	}
+	getOrCreateBoundTransport(chatId, owner, getBinding) {
+		const context = this.threadContexts.get(chatId);
+		return context?.owner === owner ? this.getThreadTransport(context) : this.getBoundTransport(chatId, owner, getBinding());
+	}
+	getThreadTransport(context) {
+		if (!context) return void 0;
+		if (context.sourceTransport !== this.transport) {
+			context.transport = this.createThreadTransport(this.transport);
+			context.sourceTransport = this.transport;
+		}
+		this.wireTransport(context, context.transport);
+		return context.transport;
+	}
+	createThreadTransport(transport) {
+		return transport instanceof AssistantChatTransport ? transport.__internal_clone() : transport;
+	}
+	wireTransport(context, transport) {
+		if (!(transport instanceof AssistantChatTransport)) return;
+		if (context.runtime) transport.setRuntime(context.runtime);
+		if (context.getThreadListItem) transport.__internal_setGetThreadListItem(context.getThreadListItem);
+	}
+};
+//#endregion
+export { DynamicChatTransport, getResumedStreamIds };
+
+//# sourceMappingURL=DynamicChatTransport.js.map

package/dist/runtime/DynamicChatTransport.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"DynamicChatTransport.js","names":["AssistantRuntime","UIMessage","ChatTransport","AssistantChatTransport","InitializableThreadListItem","ResumableClientStorage","getResumableAdapter","resumedStreamIdsByStorage","WeakMap","Set","getResumedStreamIds","storage","resumedStreamIds","get","set","ResumableStorageSubscription","listener","threadId","unsubscribe","DynamicResumableStorage","subscriptions","hasPendingNotification","constructor","setStorage","nextResumedStreamIds","streamId","add","flushChange","subscription","subscribe","getStreamId","setStreamId","id","clear","delete","ThreadTransportContext","owner","sourceTransport","UI_MESSAGE","transport","runtime","getThreadListItem","ThreadTransportBinding","DynamicChatTransport","threadContexts","Map","resumableStorage","sendMessages","options","getTransport","chatId","reconnectToStream","getCurrentTransport","getThreadTransport","getCurrentResumableStorage","createThreadProxy","getBinding","getOrCreateBoundTransport","setTransport","flushTransportChange","registerThread","existing","setThreadContext","context","undefined","wireTransport","unregisterThread","Error","getBoundTransport","binding","createThreadTransport","__internal_clone","setRuntime","__internal_setGetThreadListItem"],"sources":["../../src/runtime/DynamicChatTransport.ts"],"sourcesContent":["import type { AssistantRuntime } from \"@assistant-ui/core\";\nimport type { UIMessage } from \"@ai-sdk/react\";\nimport type { ChatTransport } from \"ai\";\nimport {\n  AssistantChatTransport,\n  type InitializableThreadListItem,\n} from \"../transport/AssistantChatTransport\";\nimport type { ResumableClientStorage } from \"../transport/resumable\";\nimport { getResumableAdapter } from \"./getResumableAdapter\";\n\nconst resumedStreamIdsByStorage = new WeakMap<\n  ResumableClientStorage,\n  Set<string>\n>();\n\nexport const getResumedStreamIds = (\n  storage: ResumableClientStorage | undefined,\n) => {\n  if (!storage) return new Set<string>();\n  let resumedStreamIds = resumedStreamIdsByStorage.get(storage);\n  if (!resumedStreamIds) {\n    resumedStreamIds = new Set();\n    resumedStreamIdsByStorage.set(storage, resumedStreamIds);\n  }\n  return resumedStreamIds;\n};\n\ntype ResumableStorageSubscription = {\n  listener: () => void;\n  threadId: string | undefined;\n  unsubscribe: (() => void) | undefined;\n};\n\nclass DynamicResumableStorage implements ResumableClientStorage {\n  private readonly subscriptions = new Set<ResumableStorageSubscription>();\n  private storage: ResumableClientStorage | undefined;\n  private resumedStreamIds: Set<string>;\n  private hasPendingNotification = false;\n\n  constructor(storage: ResumableClientStorage | undefined) {\n    this.storage = storage;\n    this.resumedStreamIds = getResumedStreamIds(storage);\n  }\n\n  public setStorage(storage: ResumableClientStorage | undefined) {\n    if (this.storage === storage) return;\n    this.storage = storage;\n    const nextResumedStreamIds = getResumedStreamIds(storage);\n    for (const streamId of this.resumedStreamIds) {\n      nextResumedStreamIds.add(streamId);\n    }\n    this.resumedStreamIds = nextResumedStreamIds;\n    this.hasPendingNotification = true;\n  }\n\n  public flushChange() {\n    if (!this.hasPendingNotification) return;\n    this.hasPendingNotification = false;\n    for (const subscription of this.subscriptions) {\n      subscription.unsubscribe?.();\n      subscription.unsubscribe = this.storage?.subscribe?.(\n        subscription.listener,\n        subscription.threadId,\n      );\n      subscription.listener();\n    }\n  }\n\n  public getResumedStreamIds() {\n    return this.resumedStreamIds;\n  }\n\n  public getStreamId(threadId?: string) {\n    return this.storage?.getStreamId(threadId) ?? null;\n  }\n\n  public setStreamId(id: string, threadId?: string) {\n    this.storage?.setStreamId(id, threadId);\n  }\n\n  public clear(threadId?: string) {\n    this.storage?.clear(threadId);\n  }\n\n  public subscribe(listener: () => void, threadId?: string) {\n    const subscription: ResumableStorageSubscription = {\n      listener,\n      threadId,\n      unsubscribe: this.storage?.subscribe?.(listener, threadId),\n    };\n    this.subscriptions.add(subscription);\n    return () => {\n      this.subscriptions.delete(subscription);\n      subscription.unsubscribe?.();\n    };\n  }\n}\n\ntype ThreadTransportContext<UI_MESSAGE extends UIMessage> = {\n  owner: object;\n  sourceTransport?: ChatTransport<UI_MESSAGE> | undefined;\n  transport?: ChatTransport<UI_MESSAGE> | undefined;\n  runtime?: AssistantRuntime | undefined;\n  getThreadListItem?:\n    | (() => InitializableThreadListItem | undefined)\n    | undefined;\n};\n\ntype ThreadTransportBinding = {\n  runtime: AssistantRuntime;\n  getThreadListItem: () => InitializableThreadListItem | undefined;\n};\n\nexport class DynamicChatTransport<\n  UI_MESSAGE extends UIMessage,\n> implements ChatTransport<UI_MESSAGE> {\n  private readonly threadContexts = new Map<\n    string,\n    ThreadTransportContext<UI_MESSAGE>\n  >();\n  private transport: ChatTransport<UI_MESSAGE>;\n  private readonly resumableStorage: DynamicResumableStorage;\n\n  constructor(transport: ChatTransport<UI_MESSAGE>) {\n    this.transport = transport;\n    this.resumableStorage = new DynamicResumableStorage(\n      getResumableAdapter(transport)?.storage,\n    );\n  }\n\n  public readonly sendMessages: ChatTransport<UI_MESSAGE>[\"sendMessages\"] = (\n    options,\n  ) => this.getTransport(options.chatId).sendMessages(options);\n\n  public readonly reconnectToStream: ChatTransport<UI_MESSAGE>[\"reconnectToStream\"] =\n    (options) => this.getTransport(options.chatId).reconnectToStream(options);\n\n  public readonly getCurrentTransport = (chatId: string) =>\n    this.getThreadTransport(this.threadContexts.get(chatId)) ?? this.transport;\n\n  public readonly getCurrentResumableStorage = () => this.resumableStorage;\n\n  public readonly getResumedStreamIds = () =>\n    this.resumableStorage.getResumedStreamIds();\n\n  public createThreadProxy(\n    owner: object,\n    getBinding: () => ThreadTransportBinding,\n  ): ChatTransport<UI_MESSAGE> {\n    return {\n      sendMessages: (options) =>\n        this.getOrCreateBoundTransport(\n          options.chatId,\n          owner,\n          getBinding,\n        ).sendMessages(options),\n      reconnectToStream: (options) =>\n        this.getOrCreateBoundTransport(\n          options.chatId,\n          owner,\n          getBinding,\n        ).reconnectToStream(options),\n    };\n  }\n\n  public setTransport(transport: ChatTransport<UI_MESSAGE>) {\n    if (this.transport === transport) return;\n    this.transport = transport;\n    this.resumableStorage.setStorage(getResumableAdapter(transport)?.storage);\n  }\n\n  public flushTransportChange() {\n    this.resumableStorage.flushChange();\n  }\n\n  public registerThread(chatId: string, owner: object) {\n    const existing = this.threadContexts.get(chatId);\n    if (existing?.owner === owner) return;\n    this.threadContexts.set(chatId, {\n      owner,\n    });\n  }\n\n  public setThreadContext(\n    chatId: string,\n    owner: object,\n    runtime: AssistantRuntime,\n    getThreadListItem: () => InitializableThreadListItem | undefined,\n  ) {\n    this.registerThread(chatId, owner);\n    const context = this.threadContexts.get(chatId)!;\n    context.runtime = runtime;\n    context.getThreadListItem = getThreadListItem;\n    if (context.transport === undefined) {\n      this.getThreadTransport(context);\n      return;\n    }\n    if (\n      context.sourceTransport === this.transport &&\n      context.transport !== undefined\n    ) {\n      this.wireTransport(context, context.transport);\n    }\n  }\n\n  public unregisterThread(chatId: string, owner: object) {\n    if (this.threadContexts.get(chatId)?.owner === owner) {\n      this.threadContexts.delete(chatId);\n    }\n  }\n\n  private getTransport(chatId: string) {\n    const context = this.threadContexts.get(chatId);\n    if (!context) {\n      throw new Error(\n        `DynamicChatTransport has no registered context for chat \"${chatId}\"`,\n      );\n    }\n    return this.getThreadTransport(context)!;\n  }\n\n  private getBoundTransport(\n    chatId: string,\n    owner: object,\n    binding: ThreadTransportBinding,\n  ) {\n    this.registerThread(chatId, owner);\n    const context = this.threadContexts.get(chatId)!;\n    context.runtime = binding.runtime;\n    context.getThreadListItem = binding.getThreadListItem;\n    return this.getThreadTransport(context)!;\n  }\n\n  private getOrCreateBoundTransport(\n    chatId: string,\n    owner: object,\n    getBinding: () => ThreadTransportBinding,\n  ) {\n    const context = this.threadContexts.get(chatId);\n    return context?.owner === owner\n      ? this.getThreadTransport(context)!\n      : this.getBoundTransport(chatId, owner, getBinding());\n  }\n\n  private getThreadTransport(\n    context: ThreadTransportContext<UI_MESSAGE> | undefined,\n  ) {\n    if (!context) return undefined;\n    if (context.sourceTransport !== this.transport) {\n      context.transport = this.createThreadTransport(this.transport);\n      context.sourceTransport = this.transport;\n    }\n    this.wireTransport(context, context.transport!);\n    return context.transport!;\n  }\n\n  private createThreadTransport(transport: ChatTransport<UI_MESSAGE>) {\n    return transport instanceof AssistantChatTransport\n      ? transport.__internal_clone()\n      : transport;\n  }\n\n  private wireTransport(\n    context: ThreadTransportContext<UI_MESSAGE>,\n    transport: ChatTransport<UI_MESSAGE>,\n  ) {\n    if (!(transport instanceof AssistantChatTransport)) return;\n    if (context.runtime) transport.setRuntime(context.runtime);\n    if (context.getThreadListItem) {\n      transport.__internal_setGetThreadListItem(context.getThreadListItem);\n    }\n  }\n}\n"],"mappings":";;;AAUA,MAAMO,4CAA4B,IAAIC,QAGpC;AAEF,MAAaE,uBACXC,YACG;CACH,IAAI,CAACA,SAAS,uBAAO,IAAIF,IAAY;CACrC,IAAIG,mBAAmBL,0BAA0BM,IAAIF,OAAO;CAC5D,IAAI,CAACC,kBAAkB;EACrBA,mCAAmB,IAAIH,IAAI;EAC3BF,0BAA0BO,IAAIH,SAASC,gBAAgB;CACzD;CACA,OAAOA;AACT;AAQA,IAAMO,0BAAN,MAAgE;CAC9D,gCAAiC,IAAIV,IAAkC;CACvE;CACA;CACA,yBAAiC;CAEjCa,YAAYX,SAA6C;EACvD,KAAKA,UAAUA;EACf,KAAKC,mBAAmBF,oBAAoBC,OAAO;CACrD;CAEA,WAAkBA,SAA6C;EAC7D,IAAI,KAAKA,YAAYA,SAAS;EAC9B,KAAKA,UAAUA;EACf,MAAMa,uBAAuBd,oBAAoBC,OAAO;EACxD,KAAK,MAAMc,YAAY,KAAKb,kBAC1BY,qBAAqBE,IAAID,QAAQ;EAEnC,KAAKb,mBAAmBY;EACxB,KAAKH,yBAAyB;CAChC;CAEA,cAAqB;EACnB,IAAI,CAAC,KAAKA,wBAAwB;EAClC,KAAKA,yBAAyB;EAC9B,KAAK,MAAMO,gBAAgB,KAAKR,eAAe;GAC7CQ,aAAaV,cAAc;GAC3BU,aAAaV,cAAc,KAAKP,SAASkB,YACvCD,aAAaZ,UACbY,aAAaX,QACf;GACAW,aAAaZ,SAAS;EACxB;CACF;CAEA,sBAA6B;EAC3B,OAAO,KAAKJ;CACd;CAEA,YAAmBK,UAAmB;EACpC,OAAO,KAAKN,SAASmB,YAAYb,QAAQ,KAAK;CAChD;CAEA,YAAmBe,IAAYf,UAAmB;EAChD,KAAKN,SAASoB,YAAYC,IAAIf,QAAQ;CACxC;CAEA,MAAaA,UAAmB;EAC9B,KAAKN,SAASsB,MAAMhB,QAAQ;CAC9B;CAEA,UAAiBD,UAAsBC,UAAmB;EACxD,MAAMW,eAA6C;GACjDZ;GACAC;GACAC,aAAa,KAAKP,SAASkB,YAAYb,UAAUC,QAAQ;EAC3D;EACA,KAAKG,cAAcM,IAAIE,YAAY;EACnC,aAAa;GACX,KAAKR,cAAcc,OAAON,YAAY;GACtCA,aAAaV,cAAc;EAC7B;CACF;AACF;AAiBA,IAAayB,uBAAb,MAEuC;CACrC,iCAAkC,IAAIE,IAGpC;CACF;CACA;CAEAvB,YAAYiB,WAAsC;EAChD,KAAKA,YAAYA;EACjB,KAAKO,mBAAmB,IAAI3B,wBAC1Bb,oBAAoBiC,SAAS,CAAC,EAAE5B,OAClC;CACF;CAEA,gBACEqC,YACG,KAAKC,aAAaD,QAAQE,MAAM,CAAC,CAACH,aAAaC,OAAO;CAE3D,qBACGA,YAAY,KAAKC,aAAaD,QAAQE,MAAM,CAAC,CAACC,kBAAkBH,OAAO;CAE1E,uBAAuCE,WACrC,KAAKG,mBAAmB,KAAKT,eAAe/B,IAAIqC,MAAM,CAAC,KAAK,KAAKX;CAEnE,mCAAmD,KAAKO;CAExD,4BACE,KAAKA,iBAAiBpC,oBAAoB;CAE5C,kBACE0B,OACAoB,YAC2B;EAC3B,OAAO;GACLT,eAAeC,YACb,KAAKS,0BACHT,QAAQE,QACRd,OACAoB,UACF,CAAC,CAACT,aAAaC,OAAO;GACxBG,oBAAoBH,YAClB,KAAKS,0BACHT,QAAQE,QACRd,OACAoB,UACF,CAAC,CAACL,kBAAkBH,OAAO;EAC/B;CACF;CAEA,aAAoBT,WAAsC;EACxD,IAAI,KAAKA,cAAcA,WAAW;EAClC,KAAKA,YAAYA;EACjB,KAAKO,iBAAiBvB,WAAWjB,oBAAoBiC,SAAS,CAAC,EAAE5B,OAAO;CAC1E;CAEA,uBAA8B;EAC5B,KAAKmC,iBAAiBnB,YAAY;CACpC;CAEA,eAAsBuB,QAAgBd,OAAe;EAEnD,IADiB,KAAKQ,eAAe/B,IAAIqC,MACrCW,CAAQ,EAAEzB,UAAUA,OAAO;EAC/B,KAAKQ,eAAe9B,IAAIoC,QAAQ,EAC9Bd,MACF,CAAC;CACH;CAEA,iBACEc,QACAd,OACAI,SACAC,mBACA;EACA,KAAKmB,eAAeV,QAAQd,KAAK;EACjC,MAAM2B,UAAU,KAAKnB,eAAe/B,IAAIqC,MAAM;EAC9Ca,QAAQvB,UAAUA;EAClBuB,QAAQtB,oBAAoBA;EAC5B,IAAIsB,QAAQxB,cAAcyB,KAAAA,GAAW;GACnC,KAAKX,mBAAmBU,OAAO;GAC/B;EACF;EACA,IACEA,QAAQ1B,oBAAoB,KAAKE,aACjCwB,QAAQxB,cAAcyB,KAAAA,GAEtB,KAAKC,cAAcF,SAASA,QAAQxB,SAAS;CAEjD;CAEA,iBAAwBW,QAAgBd,OAAe;EACrD,IAAI,KAAKQ,eAAe/B,IAAIqC,MAAM,CAAC,EAAEd,UAAUA,OAC7C,KAAKQ,eAAeV,OAAOgB,MAAM;CAErC;CAEA,aAAqBA,QAAgB;EACnC,MAAMa,UAAU,KAAKnB,eAAe/B,IAAIqC,MAAM;EAC9C,IAAI,CAACa,SACH,MAAM,IAAII,MACR,4DAA4DjB,OAAM,EACpE;EAEF,OAAO,KAAKG,mBAAmBU,OAAO;CACxC;CAEA,kBACEb,QACAd,OACAiC,SACA;EACA,KAAKT,eAAeV,QAAQd,KAAK;EACjC,MAAM2B,UAAU,KAAKnB,eAAe/B,IAAIqC,MAAM;EAC9Ca,QAAQvB,UAAU6B,QAAQ7B;EAC1BuB,QAAQtB,oBAAoB4B,QAAQ5B;EACpC,OAAO,KAAKY,mBAAmBU,OAAO;CACxC;CAEA,0BACEb,QACAd,OACAoB,YACA;EACA,MAAMO,UAAU,KAAKnB,eAAe/B,IAAIqC,MAAM;EAC9C,OAAOa,SAAS3B,UAAUA,QACtB,KAAKiB,mBAAmBU,OAAO,IAC/B,KAAKK,kBAAkBlB,QAAQd,OAAOoB,WAAW,CAAC;CACxD;CAEA,mBACEO,SACA;EACA,IAAI,CAACA,SAAS,OAAOC,KAAAA;EACrB,IAAID,QAAQ1B,oBAAoB,KAAKE,WAAW;GAC9CwB,QAAQxB,YAAY,KAAK+B,sBAAsB,KAAK/B,SAAS;GAC7DwB,QAAQ1B,kBAAkB,KAAKE;EACjC;EACA,KAAK0B,cAAcF,SAASA,QAAQxB,SAAU;EAC9C,OAAOwB,QAAQxB;CACjB;CAEA,sBAA8BA,WAAsC;EAClE,OAAOA,qBAAqBpC,yBACxBoC,UAAUgC,iBAAiB,IAC3BhC;CACN;CAEA,cACEwB,SACAxB,WACA;EACA,IAAI,EAAEA,qBAAqBpC,yBAAyB;EACpD,IAAI4D,QAAQvB,SAASD,UAAUiC,WAAWT,QAAQvB,OAAO;EACzD,IAAIuB,QAAQtB,mBACVF,UAAUkC,gCAAgCV,QAAQtB,iBAAiB;CAEvE;AACF"}

package/dist/runtime/getResumableAdapter.d.ts

@@ -0,0 +1,5 @@
+import type { UIMessage } from "@ai-sdk/react";
+import type { ChatTransport } from "ai";
+import type { AssistantChatResumableOptions } from "../transport/resumable.js";
+export declare const getResumableAdapter: <UI_MESSAGE extends UIMessage>(transport: ChatTransport<UI_MESSAGE>) => AssistantChatResumableOptions | undefined;
+//# sourceMappingURL=getResumableAdapter.d.ts.map

package/dist/runtime/getResumableAdapter.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"getResumableAdapter.d.ts","sourceRoot":"","sources":["../../src/runtime/getResumableAdapter.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,eAAe,CAAC;AAC/C,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,IAAI,CAAC;AAExC,OAAO,KAAK,EAAE,6BAA6B,EAAE,kCAA+B;AAE5E,eAAO,MAAM,mBAAmB,GAAI,UAAU,SAAS,SAAS,EAC9D,WAAW,aAAa,CAAC,UAAU,CAAC,KACnC,6BAA6B,GAAG,SAQlC,CAAC"}

package/dist/runtime/getResumableAdapter.js

@@ -0,0 +1,12 @@
+import { AssistantChatTransport } from "../transport/AssistantChatTransport.js";
+//#region src/runtime/getResumableAdapter.ts
+const getResumableAdapter = (transport) => {
+	if (transport instanceof AssistantChatTransport) return transport.getResumableAdapter();
+	const candidate = transport.getResumableAdapter;
+	if (typeof candidate !== "function") return void 0;
+	return candidate.call(transport);
+};
+//#endregion
+export { getResumableAdapter };
+
+//# sourceMappingURL=getResumableAdapter.js.map

package/dist/runtime/getResumableAdapter.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"getResumableAdapter.js","names":["UIMessage","ChatTransport","AssistantChatTransport","AssistantChatResumableOptions","getResumableAdapter","transport","UI_MESSAGE","candidate","undefined","call"],"sources":["../../src/runtime/getResumableAdapter.ts"],"sourcesContent":["import type { UIMessage } from \"@ai-sdk/react\";\nimport type { ChatTransport } from \"ai\";\nimport { AssistantChatTransport } from \"../transport/AssistantChatTransport\";\nimport type { AssistantChatResumableOptions } from \"../transport/resumable\";\n\nexport const 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"],"mappings":";;AAKA,MAAaI,uBACXC,cAC8C;CAC9C,IAAIA,qBAAqBH,wBACvB,OAAOG,UAAUD,oBAAoB;CAEvC,MAAMG,YAAaF,UAChBD;CACH,IAAI,OAAOG,cAAc,YAAY,OAAOC,KAAAA;CAC5C,OAAOD,UAAUE,KAAKJ,SAAS;AACjC"}

package/dist/runtime/sdkIdentity.js

@@ -1,7 +1,7 @@
 //#region src/runtime/sdkIdentity.ts
 const AI_SDK_SDK = {
 	name: "@assistant-ui/ai-sdk",
-	version: "0.0.10"
+	version: "0.0.12"
 };
 //#endregion
 export { AI_SDK_SDK };

package/dist/runtime/toolHistoryCodec.d.ts

@@ -0,0 +1,20 @@
+import type { RespondToToolApprovalOptions, ThreadMessage, Unstable_ToolInteractionLog } from "@assistant-ui/core";
+export type StoredToolApprovalResponse = Omit<RespondToToolApprovalOptions, "approvalId">;
+export declare const collectToolArtifacts: (message: ThreadMessage, toolArtifacts: ReadonlyMap<string, unknown> | undefined) => {
+    [k: string]: {} | null;
+} | undefined;
+export declare const collectToolInteractions: (message: ThreadMessage, toolInteractions: ReadonlyMap<string, Unstable_ToolInteractionLog> | undefined) => {
+    [k: string]: Unstable_ToolInteractionLog;
+} | undefined;
+export declare const collectToolApprovalResponses: (message: ThreadMessage, toolApprovalResponses: ReadonlyMap<string, RespondToToolApprovalOptions> | undefined) => {
+    [k: string]: {
+        readonly reason?: string;
+        readonly answers?: Readonly<Record<string, import("@assistant-ui/core").ToolApprovalAnswer>>;
+        readonly text?: string;
+        readonly optionId?: string;
+        readonly approved: boolean;
+    };
+} | undefined;
+export declare const addToolData: <TMessage>(message: TMessage, toolArtifacts: Record<string, unknown> | undefined, toolInteractions: Record<string, Unstable_ToolInteractionLog> | undefined, toolApprovalResponses: Record<string, StoredToolApprovalResponse> | undefined) => TMessage;
+export declare const restoreToolData: <TMessage>(message: TMessage, toolArtifacts: Map<string, unknown> | undefined, toolInteractions: Map<string, Unstable_ToolInteractionLog> | undefined, toolApprovalResponses: Map<string, RespondToToolApprovalOptions> | undefined) => TMessage;
+//# sourceMappingURL=toolHistoryCodec.d.ts.map

package/dist/runtime/toolHistoryCodec.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"toolHistoryCodec.d.ts","sourceRoot":"","sources":["../../src/runtime/toolHistoryCodec.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,4BAA4B,EAC5B,aAAa,EACb,2BAA2B,EAC5B,MAAM,oBAAoB,CAAC;AAQ5B,MAAM,MAAM,0BAA0B,GAAG,IAAI,CAC3C,4BAA4B,EAC5B,YAAY,CACb,CAAC;AAEF,eAAO,MAAM,oBAAoB,GAC/B,SAAS,aAAa,EACtB,eAAe,WAAW,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,SAAS;;aASxD,CAAC;AAEF,eAAO,MAAM,uBAAuB,GAClC,SAAS,aAAa,EACtB,kBACI,WAAW,CAAC,MAAM,EAAE,2BAA2B,CAAC,GAChD,SAAS;;aAWd,CAAC;AAEF,eAAO,MAAM,4BAA4B,GACvC,SAAS,aAAa,EACtB,uBACI,WAAW,CAAC,MAAM,EAAE,4BAA4B,CAAC,GACjD,SAAS;;;;;;;;aAqBd,CAAC;AAEF,eAAO,MAAM,WAAW,GAAI,QAAQ,EAClC,SAAS,QAAQ,EACjB,eAAe,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,SAAS,EAClD,kBAAkB,MAAM,CAAC,MAAM,EAAE,2BAA2B,CAAC,GAAG,SAAS,EACzE,uBAAuB,MAAM,CAAC,MAAM,EAAE,0BAA0B,CAAC,GAAG,SAAS,KAC5E,QAgEF,CAAC;AAEF,eAAO,MAAM,eAAe,GAAI,QAAQ,EACtC,SAAS,QAAQ,EACjB,eAAe,GAAG,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,SAAS,EAC/C,kBAAkB,GAAG,CAAC,MAAM,EAAE,2BAA2B,CAAC,GAAG,SAAS,EACtE,uBAAuB,GAAG,CAAC,MAAM,EAAE,4BAA4B,CAAC,GAAG,SAAS,KAC3E,QAwDF,CAAC"}