@assistant-ui/ai-sdk 0.0.5 → 0.0.7

package/dist/runtime/AISDKThreads.js

@@ -1,29 +1,31 @@
 "use client";
 import { AssistantChatTransport } from "../transport/AssistantChatTransport.js";
 import { useResourceCleanup } from "./useResourceCleanup.js";
-import { splitChatThreadOptions, useChatThread } from "./useChatThread.js";
+import { createChat, splitChatThreadOptions, useChatThread } from "./useChatThread.js";
+import { AI_SDK_SDK } from "./sdkIdentity.js";
 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 { MessageRepository } from "@assistant-ui/core/internal";
 import { useAui } from "@assistant-ui/store";
-import { Chat } from "@ai-sdk/react";
 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";
 import { ThreadClient } from "@assistant-ui/core/store/internal";
 //#region src/runtime/AISDKThreads.ts
 const createChatEntry = (threadId, options) => {
+	const optionsRef = { current: options };
 	const { chatInit } = splitChatThreadOptions(options);
 	const transport = typeof options?.transport === "function" ? options.transport() : options?.transport === void 0 ? new AssistantChatTransport() : options.transport instanceof AssistantChatTransport ? options.transport.__internal_clone() : options.transport;
 	return {
-		chat: new Chat({
+		chat: createChat({
 			...chatInit,
 			id: threadId,
 			transport
-		}),
+		}, optionsRef),
 		transport,
-		repository: new MessageRepository()
+		repository: new MessageRepository(),
+		optionsRef
 	};
 };
 const getOrCreateChatEntry = (threadId, options, chats) => {
@@ -34,7 +36,7 @@ const getOrCreateChatEntry = (threadId, options, chats) => {
 	return created;
 };
 const useAISDKChatThread = (t0) => {
-	const $ = c(35);
+	const $ = c(39);
 	const { threadId, options, chats, cloud } = t0;
 	let t1;
 	if ($[0] !== cloud || $[1] !== options || $[2] !== threadId) {
@@ -54,95 +56,106 @@ const useAISDKChatThread = (t0) => {
 		$[7] = threadId;
 		$[8] = t2;
 	} else t2 = $[8];
-	const { chat, transport, repository } = t2;
+	const { chat, transport, repository, optionsRef } = t2;
 	let t3;
-	let t4;
-	if ($[9] !== chat || $[10] !== cloud) {
+	if ($[9] !== cloud || $[10] !== options || $[11] !== optionsRef) {
 		t3 = () => {
+			if (cloud) optionsRef.current = options;
+		};
+		$[9] = cloud;
+		$[10] = options;
+		$[11] = optionsRef;
+		$[12] = t3;
+	} else t3 = $[12];
+	useEffect(t3);
+	let t4;
+	let t5;
+	if ($[13] !== chat || $[14] !== cloud) {
+		t4 = () => {
 			if (!cloud) return;
 			return () => {
 				chat.stop().catch(_temp);
 			};
 		};
-		t4 = [chat, cloud];
-		$[9] = chat;
-		$[10] = cloud;
-		$[11] = t3;
-		$[12] = t4;
+		t5 = [chat, cloud];
+		$[13] = chat;
+		$[14] = cloud;
+		$[15] = t4;
+		$[16] = t5;
 	} else {
-		t3 = $[11];
-		t4 = $[12];
+		t4 = $[15];
+		t5 = $[16];
 	}
-	useEffect(t3, t4);
+	useEffect(t4, t5);
 	const aui = useAui();
-	let t5;
-	if ($[13] !== threadId) {
-		t5 = { initialize: async () => ({
+	let t6;
+	if ($[17] !== threadId) {
+		t6 = { initialize: async () => ({
 			remoteId: threadId,
 			externalId: void 0
 		}) };
-		$[13] = threadId;
-		$[14] = t5;
-	} else t5 = $[14];
-	const fallbackItem = t5;
-	let t6;
-	if ($[15] !== options || $[16] !== transport) {
-		t6 = {
+		$[17] = threadId;
+		$[18] = t6;
+	} else t6 = $[18];
+	const fallbackItem = t6;
+	let t7;
+	if ($[19] !== options || $[20] !== transport) {
+		t7 = {
 			...options,
 			transport
 		};
-		$[15] = options;
-		$[16] = transport;
-		$[17] = t6;
-	} else t6 = $[17];
-	let t7;
-	if ($[18] !== aui || $[19] !== cloud || $[20] !== fallbackItem) {
-		t7 = () => cloud ? aui.threadListItem.source ? aui.threadListItem : void 0 : fallbackItem;
-		$[18] = aui;
-		$[19] = cloud;
-		$[20] = fallbackItem;
+		$[19] = options;
+		$[20] = transport;
 		$[21] = t7;
 	} else t7 = $[21];
 	let t8;
-	if ($[22] !== chat || $[23] !== cloud || $[24] !== repository || $[25] !== t7 || $[26] !== threadId) {
-		t8 = {
+	if ($[22] !== aui || $[23] !== cloud || $[24] !== fallbackItem) {
+		t8 = () => cloud ? aui.threadListItem.source ? aui.threadListItem : void 0 : fallbackItem;
+		$[22] = aui;
+		$[23] = cloud;
+		$[24] = fallbackItem;
+		$[25] = t8;
+	} else t8 = $[25];
+	let t9;
+	if ($[26] !== chat || $[27] !== cloud || $[28] !== repository || $[29] !== t8 || $[30] !== threadId) {
+		t9 = {
 			id: threadId,
 			isMainThread: true,
-			getThreadListItem: t7,
+			getThreadListItem: t8,
 			chat,
 			messageRepositoryInstance: repository,
 			stopOnClientDestroy: cloud
 		};
-		$[22] = chat;
-		$[23] = cloud;
-		$[24] = repository;
-		$[25] = t7;
-		$[26] = threadId;
-		$[27] = t8;
-	} else t8 = $[27];
-	const runtime = useChatThread(t6, t8);
+		$[26] = chat;
+		$[27] = cloud;
+		$[28] = repository;
+		$[29] = t8;
+		$[30] = threadId;
+		$[31] = t9;
+	} else t9 = $[31];
+	const runtime = useChatThread(t7, t9);
 	const clientRef = useAssistantClientRef();
-	let t9;
-	if ($[28] !== clientRef || $[29] !== runtime) {
-		t9 = () => runtime.registerModelContextProvider(clientRef.current.modelContext());
-		$[28] = clientRef;
-		$[29] = runtime;
-		$[30] = t9;
-	} else t9 = $[30];
 	let t10;
-	if ($[31] !== runtime) {
-		t10 = [runtime];
-		$[31] = runtime;
-		$[32] = t10;
-	} else t10 = $[32];
-	useAssistantScopeEffect("modelContext", t9, t10);
+	if ($[32] !== clientRef || $[33] !== runtime) {
+		t10 = () => runtime.registerModelContextProvider(clientRef.current.modelContext());
+		$[32] = clientRef;
+		$[33] = runtime;
+		$[34] = t10;
+	} else t10 = $[34];
 	let t11;
-	if ($[33] !== runtime.thread) {
-		t11 = ThreadClient({ runtime: runtime.thread });
-		$[33] = runtime.thread;
-		$[34] = t11;
-	} else t11 = $[34];
-	return useResource(t11);
+	if ($[35] !== runtime) {
+		t11 = [runtime];
+		$[35] = runtime;
+		$[36] = t11;
+	} else t11 = $[36];
+	useAssistantScopeEffect("modelContext", t10, t11);
+	let t12;
+	if ($[37] !== runtime.thread) {
+		t12 = ThreadClient({ runtime: runtime.thread });
+		$[37] = runtime.thread;
+		$[38] = t12;
+	} else t12 = $[38];
+	return useResource(t12);
 };
 const AISDKChatThread = resource(useAISDKChatThread);
 const useAISDKThreads = (options) => {
@@ -150,6 +163,9 @@ const useAISDKThreads = (options) => {
 	const { cloud, threadId, onThreadIdChange, ...threadOptions } = options ?? {};
 	const [chats] = useState(_temp2);
 	const bindCloud = cloud !== void 0;
+	useEffect(() => {
+		for (const { optionsRef } of chats.values()) optionsRef.current = threadOptions;
+	});
 	let t0;
 	if ($[0] !== chats) {
 		t0 = () => {
@@ -159,7 +175,10 @@ const useAISDKThreads = (options) => {
 		$[1] = t0;
 	} else t0 = $[1];
 	useResourceCleanup(true, t0);
-	const cloudAdapter = useCloudThreadListAdapter({ cloud });
+	const cloudAdapter = useCloudThreadListAdapter({
+		cloud,
+		sdk: AI_SDK_SDK
+	});
 	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","splitChatThreadOptions","useChatThread","ChatThreadOptions","MessageRepository","useResourceCleanup","AISDKThreadsOptions","Omit","UI_MESSAGE","transport","cloud","threadId","onThreadIdChange","AISDKThreadChatOptions","ChatEntry","chat","repository","createChatEntry","options","chatInit","undefined","__internal_clone","id","getOrCreateChatEntry","chats","Map","existing","get","created","set","useAISDKChatThread","t0","$","t1","owned","t2","t3","t4","stop","catch","_temp","aui","t5","initialize","remoteId","externalId","fallbackItem","t6","t7","threadListItem","source","t8","isMainThread","getThreadListItem","messageRepositoryInstance","stopOnClientDestroy","runtime","clientRef","t9","registerModelContextProvider","current","modelContext","t10","t11","thread","AISDKChatThread","useAISDKThreads","threadOptions","_temp2","bindCloud","values","_temp3","cloudAdapter","element","adapter","backgroundThreads","onDelete","id_0","_temp4","delete","AISDKThreads"],"sources":["../../src/runtime/AISDKThreads.ts"],"sourcesContent":["\"use client\";\n\nimport { resource, useResource, withKey } from \"@assistant-ui/tap\";\nimport { useEffect, useMemo, useState } from \"react\";\nimport { Chat, type 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  splitChatThreadOptions,\n  useChatThread,\n  type ChatThreadOptions,\n} from \"./useChatThread\";\nimport { MessageRepository } from \"@assistant-ui/core/internal\";\nimport { useResourceCleanup } from \"./useResourceCleanup\";\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 ChatEntry<UI_MESSAGE extends UIMessage> = {\n  chat: Chat<UI_MESSAGE>;\n  transport: ChatTransport<UI_MESSAGE>;\n  repository: MessageRepository;\n};\n\nconst createChatEntry = <UI_MESSAGE extends UIMessage>(\n  threadId: string,\n  options: AISDKThreadChatOptions<UI_MESSAGE> | undefined,\n): ChatEntry<UI_MESSAGE> => {\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: new Chat<UI_MESSAGE>({ ...chatInit, id: threadId, transport }),\n    transport,\n    repository: new MessageRepository(),\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 } =\n    owned ?? getOrCreateChatEntry(threadId, options, chats);\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: cloud,\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  useResourceCleanup(true, () => {\n    for (const { chat } of chats.values()) {\n      void chat.stop().catch(() => {});\n    }\n  });\n\n  const cloudAdapter = useCloudThreadListAdapter({ cloud });\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":";;;;;;;;;;;;;;;AAsEA,MAAMsC,mBACJN,UACAO,YAC0B;CAC1B,MAAM,EAAEC,aAAalB,uBACnBiB,OACF;CACA,MAAMT,YACJ,OAAOS,SAAST,cAAc,aAC1BS,QAAQT,UAAU,IAClBS,SAAST,cAAcW,KAAAA,IACrB,IAAIpB,uBAAuB,IAC3BkB,QAAQT,qBAAqBT,yBAC3BkB,QAAQT,UAAUY,iBAAiB,IACnCH,QAAQT;CAClB,OAAO;EACLM,MAAM,IAAI5B,KAAiB;GAAE,GAAGgC;GAAUG,IAAIX;GAAUF;EAAU,CAAC;EACnEA;EACAO,YAAY,IAAIZ,kBAAkB;CACpC;AACF;AAEA,MAAMmB,wBACJZ,UACAO,SACAM,UAC0B;CAC1B,MAAME,WAAWF,MAAMG,IAAIhB,QAAQ;CACnC,IAAIe,UAAU,OAAOA;CACrB,MAAME,UAAUX,gBAAgBN,UAAUO,OAAO;CACjDM,MAAMK,IAAIlB,UAAUiB,OAAO;CAC3B,OAAOA;AACT;AAEA,MAAME,sBAAqBC,OAAA;CAAA,MAAAC,IAAApD,EAAA,EAAA;CAA2C,MAAA,EAAA+B,UAAAO,SAAAM,OAAAd,UAAAqB;CAUrE,IAAAE;CAAA,IAAAD,EAAA,OAAAtB,SAAAsB,EAAA,OAAAd,WAAAc,EAAA,OAAArB,UAAA;EAC0BsB,WACvBvB,QAAQO,gBAAgBN,UAAUO,OAAmB,IAArDE,KAAAA;EAAsDY,EAAA,KAAAtB;EAAAsB,EAAA,KAAAd;EAAAc,EAAA,KAAArB;EAAAqB,EAAA,KAAAC;CAAA,OAAAA,KAAAD,EAAA;CADxD,MAAA,CAAAE,SAAgBhD,SAAS+C,EAEzB;CAAE,IAAAE;CAAA,IAAAH,EAAA,OAAAR,SAAAQ,EAAA,OAAAd,WAAAc,EAAA,OAAAE,SAAAF,EAAA,OAAArB,UAAA;EAEAwB,KAAAD,SAASX,qBAAqBZ,UAAUO,SAASM,KAAK;EAACQ,EAAA,KAAAR;EAAAQ,EAAA,KAAAd;EAAAc,EAAA,KAAAE;EAAAF,EAAA,KAAArB;EAAAqB,EAAA,KAAAG;CAAA,OAAAA,KAAAH,EAAA;CADzD,MAAA,EAAAjB,MAAAN,WAAAO,eACEmB;CAAwD,IAAAC;CAAA,IAAAC;CAAA,IAAAL,EAAA,OAAAjB,QAAAiB,EAAA,QAAAtB,OAAA;EAEhD0B,WAAA;GACR,IAAI,CAAC1B,OAAK;GAAmB,aACtB;IACAK,KAAIuB,KAAM,CAAC,CAAAC,MAAOC,KAAQ;GAAC;EACjC;EACAH,KAAA,CAACtB,MAAML,KAAK;EAACsB,EAAA,KAAAjB;EAAAiB,EAAA,MAAAtB;EAAAsB,EAAA,MAAAI;EAAAJ,EAAA,MAAAK;CAAA,OAAA;EAAAD,KAAAJ,EAAA;EAAAK,KAAAL,EAAA;CAAA;CALhBhD,UAAUoD,IAKPC,EAAa;CAEhB,MAAAI,MAAY1C,OAAO;CAAE,IAAA2C;CAAA,IAAAV,EAAA,QAAArB,UAAA;EAEZ+B,KAAA,EAAAC,YACO,aAAa;GAAAC,UACbjC;GAAQkC,YACNzB,KAAAA;EACd,GACF;EAACY,EAAA,MAAArB;EAAAqB,EAAA,MAAAU;CAAA,OAAAA,KAAAV,EAAA;CANH,MAAAc,eACSJ;CAOP,IAAAK;CAAA,IAAAf,EAAA,QAAAd,WAAAc,EAAA,QAAAvB,WAAA;EAEAsC,KAAA;GAAA,GAAK7B;GAAOT;EAAY;EAACuB,EAAA,MAAAd;EAAAc,EAAA,MAAAvB;EAAAuB,EAAA,MAAAe;CAAA,OAAAA,KAAAf,EAAA;CAAA,IAAAgB;CAAA,IAAAhB,EAAA,QAAAS,OAAAT,EAAA,QAAAtB,SAAAsB,EAAA,QAAAc,cAAA;EAIJE,WACjBtC,QACI+B,IAAGQ,eAAeC,SAChBT,IAAGQ,iBADL7B,KAAAA,IADJ0B;EAIgBd,EAAA,MAAAS;EAAAT,EAAA,MAAAtB;EAAAsB,EAAA,MAAAc;EAAAd,EAAA,MAAAgB;CAAA,OAAAA,KAAAhB,EAAA;CAAA,IAAAmB;CAAA,IAAAnB,EAAA,QAAAjB,QAAAiB,EAAA,QAAAtB,SAAAsB,EAAA,QAAAhB,cAAAgB,EAAA,QAAAgB,MAAAhB,EAAA,QAAArB,UAAA;EARpBwC,KAAA;GAAA7B,IACMX;GAAQyC,cACE;GAAIC,mBACCL;GAKDjC;GAAAuC,2BAEStC;GAAUuC,qBAChB7C;EACvB;EAACsB,EAAA,MAAAjB;EAAAiB,EAAA,MAAAtB;EAAAsB,EAAA,MAAAhB;EAAAgB,EAAA,MAAAgB;EAAAhB,EAAA,MAAArB;EAAAqB,EAAA,MAAAmB;CAAA,OAAAA,KAAAnB,EAAA;CAdH,MAAAwB,UAAgBtD,cACd6C,IACAI,EAaF;CAEA,MAAAM,YAAkB5D,sBAAsB;CAAE,IAAA6D;CAAA,IAAA1B,EAAA,QAAAyB,aAAAzB,EAAA,QAAAwB,SAAA;EAGxCE,WACEF,QAAOG,6BAA8BF,UAASG,QAAQC,aAAe,CAAC;EAAC7B,EAAA,MAAAyB;EAAAzB,EAAA,MAAAwB;EAAAxB,EAAA,MAAA0B;CAAA,OAAAA,KAAA1B,EAAA;CAAA,IAAA8B;CAAA,IAAA9B,EAAA,QAAAwB,SAAA;EACzEM,MAAA,CAACN,OAAO;EAACxB,EAAA,MAAAwB;EAAAxB,EAAA,MAAA8B;CAAA,OAAAA,MAAA9B,EAAA;CAJXlC,wBACE,gBACA4D,IAEAI,GACF;CAAC,IAAAC;CAAA,IAAA/B,EAAA,QAAAwB,QAAAQ,QAAA;EAEkBD,MAAArE,aAAa,EAAA8D,SAAWA,QAAOQ,OAAQ,CAAC;EAAChC,EAAA,MAAAwB,QAAAQ;EAAAhC,EAAA,MAAA+B;CAAA,OAAAA,MAAA/B,EAAA;CAAA,OAArDlD,YAAYiF,GAAyC;AAAC;AAG/D,MAAME,kBAAkBpF,SAASiD,kBAAkB;AAEnD,MAAMoC,mBAAkBhD,YAAA;CAAA,MAAAc,IAAApD,EAAA,CAAA;CAGtB,MAAA,EAAA8B,OAAAC,UAAAC,kBAAA,GAAAuD,kBAAgEjD,WAAA,CAAY;CAC5E,MAAA,CAAAM,SAAgBtC,SAASkF,MAA8C;CACvE,MAAAC,YAAkB3D,UAAUU,KAAAA;CAAU,IAAAW;CAAA,IAAAC,EAAA,OAAAR,OAAA;EAEbO,WAAA;GACvB,KAAK,MAAA,EAAAhB,UAAkBS,MAAK8C,OAAQ,GAC7BvD,KAAIuB,KAAM,CAAC,CAAAC,MAAOgC,MAAQ;EAChC;EACFvC,EAAA,KAAAR;EAAAQ,EAAA,KAAAD;CAAA,OAAAA,KAAAC,EAAA;CAJD3B,mBAAmB,MAAM0B,EAIxB;CAED,MAAAyC,eAAqB7E,0BAA0B,EAAAe,MAAQ,CAAC;CACxD,MAAAsD,UAAe1C,OAAA;EACb,MAAAmD,UAAgBR,gBAAgB;GAAAtD,UACpBW;GAAEJ,SACHiD;GAAa3C;GAAAd,OAEf2D;EACT,CAAC;EAAE,OACIA,YAAYtF,QAAQuC,IAAImD,OAAiB,IAAzCA;CAA0C;CACjD,OAEK3F,YACLuF,YACI7E,iBAAiB;EAAAkF,SACNF;EAAYG,mBACF;EAAIX;EAAArD;EAAAC;CAIzB,CAUC,IATDrB,mBAAmB;EAAAyE;EAAAY,WAEPC,SAAA;GACHrD,MAAKG,IACHL,IACA,CAAC,EAAAP,KAAKuB,KACN,CAAC,CAAAC,MAACuC,MAAQ;GACjBtD,MAAKuD,OAAQzD,IAAE;EAAC;CAEpB,CAAC,CACP;AAAC;;;;;;;;;;;;;;AAgBH,MAAa0D,eAAenG,SAASqF,eAAe;AAEpDtE,sBAAsBsE,iBAAiBzE,iCAAiC;AA7H7C,SAAA+C,QAAA,CAAA;AAgEH,SAAA4B,SAAA;CAAA,uBAIS,IAAI3C,IAAmC;AAAC;AAJjD,SAAA8C,SAAA,CAAA;AAAA,SAAAO,SAAA,CAAA"}
+{"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"],"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: cloud,\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,QAAAzB,SAAAyB,EAAA,QAAAjB,cAAAiB,EAAA,QAAAiB,MAAAjB,EAAA,QAAAxB,UAAA;EARpB4C,KAAA;GAAA9B,IACMd;GAAQ6C,cACE;GAAIC,mBACCL;GAKDnC;GAAAyC,2BAESxC;GAAUyC,qBAChBjD;EACvB;EAACyB,EAAA,MAAAlB;EAAAkB,EAAA,MAAAzB;EAAAyB,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"}

package/dist/runtime/sdkIdentity.d.ts

@@ -0,0 +1,5 @@
+import { SdkIdentity } from "assistant-cloud";
+//#region src/runtime/sdkIdentity.d.ts
+export declare const AI_SDK_SDK: SdkIdentity;
+//#endregion
+//# sourceMappingURL=sdkIdentity.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"sdkIdentity.d.ts","names":[],"sources":["../../src/runtime/sdkIdentity.ts"],"mappings":";;qBAEa,YAAY"}

package/dist/runtime/sdkIdentity.js

@@ -0,0 +1,9 @@
+//#region src/runtime/sdkIdentity.ts
+const AI_SDK_SDK = {
+	name: "@assistant-ui/ai-sdk",
+	version: "0.0.7"
+};
+//#endregion
+export { AI_SDK_SDK };
+
+//# sourceMappingURL=sdkIdentity.js.map

package/dist/runtime/sdkIdentity.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"sdkIdentity.js","names":["SdkIdentity","AI_SDK_SDK","name","version","__AUI_PACKAGE_VERSION__"],"sources":["../../src/runtime/sdkIdentity.ts"],"sourcesContent":["import type { SdkIdentity } from \"assistant-cloud\";\n\nexport const AI_SDK_SDK: SdkIdentity = {\n  name: \"@assistant-ui/ai-sdk\",\n  version:\n    typeof __AUI_PACKAGE_VERSION__ === \"string\"\n      ? __AUI_PACKAGE_VERSION__\n      : \"0.0.0\",\n};\n"],"mappings":";AAEA,MAAaC,aAA0B;CACrCC,MAAM;CACNC,SAAAA;AAIF"}

package/dist/runtime/useAISDKRuntime.d.ts

@@ -1,10 +1,10 @@
 import { JoinStrategy } from "@assistant-ui/core/react";
-import { AppendMessage, AssistantRuntime, ExternalStoreAdapter, ExternalStoreSharedOptions, MessageFormatRepository, SuggestionAdapter, ThreadHistoryAdapter } from "@assistant-ui/core";
+import { AppendMessage, AssistantRuntime, ExternalStoreAdapter, ExternalStoreSharedOptions, MessageFormatRepository, RespondToToolApprovalOptions, SuggestionAdapter, ThreadHistoryAdapter } from "@assistant-ui/core";
 import { MessageRepository } from "@assistant-ui/core/internal";
 import { CreateUIMessage, UIMessage, useChat } from "@ai-sdk/react";
 //#region src/runtime/useAISDKRuntime.d.ts
-type CustomToCreateMessageFunction = <UI_MESSAGE extends UIMessage = UIMessage>(message: AppendMessage) => CreateUIMessage<UI_MESSAGE>;
-type AISDKRuntimeAdapter<UI_MESSAGE extends UIMessage = UIMessage> = ExternalStoreSharedOptions & {
+export type CustomToCreateMessageFunction = <UI_MESSAGE extends UIMessage = UIMessage>(message: AppendMessage) => CreateUIMessage<UI_MESSAGE>;
+export type AISDKRuntimeAdapter<UI_MESSAGE extends UIMessage = UIMessage> = ExternalStoreSharedOptions & {
   adapters?: (NonNullable<ExternalStoreAdapter["adapters"]> & {
     history?: ThreadHistoryAdapter | undefined;
     suggestion?: SuggestionAdapter | undefined;
@@ -35,6 +35,19 @@ type AISDKRuntimeAdapter<UI_MESSAGE extends UIMessage = UIMessage> = ExternalSto
    * Provide this to bridge resume-tool-call invocations into a custom handler.
    */
   onResumeToolCall?: ExternalStoreAdapter["onResumeToolCall"];
+  /**
+   * Answers tool approval requests through a host-owned channel instead of the AI SDK's `addToolApprovalResponse`.
+   *
+   * Called for every approval request in the thread with the complete response, including option and free-form answers. Hand requests the host does not own to `respondViaAISDK`, which is what runs when this option is omitted. The answer applies to the approval when the handler starts and is removed if it throws. It is never written into the `useChat` messages, so `sendAutomaticallyWhen` cannot forward it, and it lasts as long as this runtime: until then a second response to the same request rejects, and a runtime mounted again over the same chat shows the request open until the resumed run records its resolution in the chat.
+   *
+   * While a handler is set, an approval's `display`, `allowFreeform` and `options` reach the renderer, because the handler can receive answers the AI SDK cannot carry.
+   */
+  onRespondToToolApproval?: ((response: RespondToToolApprovalOptions, context: {
+    toolCallId: string;
+    toolName: string;
+    /** Sends this response through the AI SDK's `addToolApprovalResponse`, which carries only `approved` and `reason`. */
+    respondViaAISDK: () => Promise<void>;
+  }) => Promise<void> | void) | undefined;
   /**
    * How consecutive assistant messages are rendered.
    *
@@ -58,7 +71,6 @@ type AISDKRuntimeAdapter<UI_MESSAGE extends UIMessage = UIMessage> = ExternalSto
    */
   unstable_onBranchChange?: ExternalStoreAdapter["unstable_onBranchChange"];
 };
-declare const useAISDKRuntime: <UI_MESSAGE extends UIMessage = UIMessage>(chatHelpers: ReturnType<typeof useChat<UI_MESSAGE>>, adapter?: AISDKRuntimeAdapter<UI_MESSAGE>) => AssistantRuntime;
+export declare const useAISDKRuntime: <UI_MESSAGE extends UIMessage = UIMessage>(chatHelpers: ReturnType<typeof useChat<UI_MESSAGE>>, adapter?: AISDKRuntimeAdapter<UI_MESSAGE>) => AssistantRuntime;
 //#endregion
-export { AISDKRuntimeAdapter, CustomToCreateMessageFunction, useAISDKRuntime };
 //# sourceMappingURL=useAISDKRuntime.d.ts.map

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

@@ -1 +1 @@
-{"version":3,"file":"useAISDKRuntime.d.ts","names":[],"sources":["../../src/runtime/useAISDKRuntime.ts"],"mappings":";;;;;KAkEY,iCACV,mBAAmB,YAAY,WAE/B,SAAS,kBACN,gBAAgB;KAYT,oBAAoB,mBAAmB,YAAY,aAC7D;EACE,YACK,YAAY;IACX,UAAU;IACV,aAAa;;EAGnB,kBAAkB;EAClB,qCAAqC;;;;;;;;;EASrC;;;;;;;;EAQA,WAAW;;;;;;;EAOX,mBAAmB;;;;;;;;;EASnB,eAAe;;;;;;EAMf,oBAAoB,wBAAwB;;;;;;;EAO5C,0BAA0B;;cAoFjB,kBAAmB,mBAAmB,YAAY,WAC7D,aAAa,kBAAkB,QAAQ,cACvC,UAAS,oBAAoB,gBAAgB"}
+{"version":3,"file":"useAISDKRuntime.d.ts","names":[],"sources":["../../src/runtime/useAISDKRuntime.ts"],"mappings":";;;;;YAuEY,iCACV,mBAAmB,YAAY,WAE/B,SAAS,kBACN,gBAAgB;YA6BT,oBAAoB,mBAAmB,YAAY,aAC7D;EACE,YACK,YAAY;IACX,UAAU;IACV,aAAa;;EAGnB,kBAAkB;EAClB,qCAAqC;;;;;;;;;EASrC;;;;;;;;EAQA,WAAW;;;;;;;EAOX,mBAAmB;;;;;;;;EAQnB,4BAEM,UAAU,8BACV;IACE;IACA;;IAEA,uBAAuB;QAEtB;;;;;;;;;EAUT,eAAe;;;;;;EAMf,oBAAoB,wBAAwB;;;;;;;EAO5C,0BAA0B;;qBAsGjB,kBAAmB,mBAAmB,YAAY,WAC7D,aAAa,kBAAkB,QAAQ,cACvC,UAAS,oBAAoB,gBAAgB"}

package/dist/runtime/useAISDKRuntime.js

@@ -12,7 +12,7 @@ import { useStreamingTiming as useStreamingTiming$1 } from "./useStreamingTiming
 import { useExternalStoreRuntime, useRuntimeAdapters } from "@assistant-ui/core/react";
 import { c } from "@assistant-ui/tap/react-shim/compiler-runtime";
 import { useCallback, useEffect, useInsertionEffect, useMemo, useRef, useState } from "@assistant-ui/tap/react-shim";
-import { generateId, isToolUIPart } from "ai";
+import { generateId, getToolName, isToolUIPart } from "ai";
 import { getExternalStoreMessages, pickExternalStoreSharedOptions } from "@assistant-ui/core";
 import { MessageRepository, consumeSuggestionResult } from "@assistant-ui/core/internal";
 //#region src/runtime/useAISDKRuntime.ts
@@ -21,6 +21,18 @@ const toUIMessage = (createMessage, fallbackRole) => ({
 	id: createMessage.id ?? generateId(),
 	role: createMessage.role ?? fallbackRole
 });
+const toVoiceTranscriptUIMessage = (message) => ({
+	id: message.id,
+	role: message.role,
+	parts: message.content.filter((part) => part.type === "text").map((part) => ({
+		type: "text",
+		text: part.text
+	})),
+	metadata: {
+		modality: "voice",
+		...Object.keys(message.metadata.custom).length > 0 && { custom: message.metadata.custom }
+	}
+});
 const EMPTY_SUGGESTIONS = [];
 const useGeneratedSuggestions = (suggestionAdapter, messages, isRunning) => {
 	const $ = c(13);
@@ -129,12 +141,22 @@ const useGeneratedSuggestions = (suggestionAdapter, messages, isRunning) => {
 	return suggestions;
 };
 const NO_CANCELLED_MESSAGE_IDS = /* @__PURE__ */ new Set();
+const NO_TOOL_APPROVAL_RESPONSES = /* @__PURE__ */ new Map();
+const toChatError = (error) => {
+	const code = error.code;
+	return {
+		code: typeof code === "string" ? code : error.name !== "Error" ? error.name : "unknown",
+		message: error.message
+	};
+};
 const useAISDKRuntime = (chatHelpers, adapter = {}) => {
-	const { adapters, toCreateMessage: customToCreateMessage, cancelPendingToolCallsOnSend = true, onResume, onResumeToolCall, joinStrategy, messageRepository, unstable_onBranchChange } = adapter;
+	const { adapters, toCreateMessage: customToCreateMessage, cancelPendingToolCallsOnSend = true, onResume, onResumeToolCall, onRespondToToolApproval: customOnRespondToToolApproval, joinStrategy, messageRepository, unstable_onBranchChange } = adapter;
 	const suggestionAdapter = adapters?.suggestion;
 	const contextAdapters = useRuntimeAdapters();
 	const [toolStatuses, setToolStatuses] = useState({});
 	const [cancelledMessages, setCancelledMessages] = useState(null);
+	const [toolApprovalResponses, setToolApprovalResponses] = useState(NO_TOOL_APPROVAL_RESPONSES);
+	const hostApprovalIdsRef = useRef(/* @__PURE__ */ new Set());
 	const toolArgsKeyOrderCacheRef = useRef(/* @__PURE__ */ new Map());
 	const toolLastInputCacheRef = useRef(/* @__PURE__ */ new Map());
 	const mcpAppMetadataCacheRef = useRef(/* @__PURE__ */ new Map());
@@ -147,6 +169,11 @@ const useAISDKRuntime = (chatHelpers, adapter = {}) => {
 	const lastMessage = chatHelpers.messages.at(-1);
 	const optimisticMessageId = isRunning && lastMessage?.role === "assistant" ? lastMessage.id : void 0;
 	const cancelledMessageIds = cancelledMessages?.chatId === chatHelpers.id ? cancelledMessages.ids : NO_CANCELLED_MESSAGE_IDS;
+	const supportsRichToolApprovalResponses = customOnRespondToToolApproval != null;
+	const toThreadMessages = useCallback((sourceMessages) => {
+		const metadata = { supportsRichToolApprovalResponses };
+		return AISDKMessageConverter.toThreadMessages(sourceMessages, false, metadata);
+	}, [supportsRichToolApprovalResponses]);
 	const retractCancellation = useCallback((chatId, messageId) => {
 		setCancelledMessages((prev) => {
 			if (prev?.chatId !== chatId || !prev.ids.has(messageId)) return prev;
@@ -180,27 +207,31 @@ const useAISDKRuntime = (chatHelpers, adapter = {}) => {
 			toolArgsKeyOrderCache: toolArgsKeyOrderCacheRef.current,
 			toolLastInputCache: toolLastInputCacheRef.current,
 			mcpAppMetadataCache: mcpAppMetadataCacheRef.current,
+			supportsRichToolApprovalResponses,
 			...optimisticMessageId && { optimisticMessageId },
-			...chatHelpers.error && { error: chatHelpers.error.message },
-			...cancelledMessageIds.size > 0 && { cancelledMessageIds }
+			...chatHelpers.error && { error: toChatError(chatHelpers.error) },
+			...cancelledMessageIds.size > 0 && { cancelledMessageIds },
+			...toolApprovalResponses.size > 0 && { toolApprovalResponses }
 		}), [
 			toolStatuses,
 			messageTiming,
 			optimisticMessageId,
 			chatHelpers.error,
-			cancelledMessageIds
+			cancelledMessageIds,
+			toolApprovalResponses,
+			supportsRichToolApprovalResponses
 		])
 	});
 	const exportedMessageRepository = useMemo(() => {
 		if (!messageRepository) return void 0;
-		const converted = toExportedMessageRepository(AISDKMessageConverter.toThreadMessages, messageRepository);
+		const converted = toExportedMessageRepository(toThreadMessages, messageRepository);
 		return converted.messages.length > 0 ? converted : void 0;
-	}, [messageRepository]);
+	}, [messageRepository, toThreadMessages]);
 	const generatedSuggestions = useGeneratedSuggestions(suggestionAdapter, messages, isRunning);
 	const [runtimeRef] = useState(() => ({ get current() {
 		return runtime;
 	} }));
-	const { isLoading, deleteMessage: deleteHistoryMessage } = useExternalHistory(runtimeRef, adapters?.history ?? contextAdapters?.history, AISDKMessageConverter.toThreadMessages, aiSDKV6FormatAdapter, (messages_0) => {
+	const { isLoading, deleteMessage: deleteHistoryMessage } = useExternalHistory(runtimeRef, adapters?.history ?? contextAdapters?.history, toThreadMessages, aiSDKV6FormatAdapter, (messages_0) => {
 		chatHelpers.setMessages(messages_0);
 	});
 	const { id: chatId_0, messages: chatMessages, status: chatStatus, error } = chatHelpers;
@@ -237,6 +268,44 @@ const useAISDKRuntime = (chatHelpers, adapter = {}) => {
 			}];
 		});
 	};
+	const respondViaAISDK = ({ approvalId, approved, reason }) => Promise.resolve(chatHelpers.addToolApprovalResponse({
+		id: approvalId,
+		approved,
+		...reason != null && { reason },
+		options: { metadata: lastRunConfigRef.current }
+	}));
+	const respondViaHost = async (onRespond, response) => {
+		const { approvalId: approvalId_0 } = response;
+		const requested = chatHelpers.messages.flatMap((message) => message.parts).filter(isToolUIPart).find((part_0) => part_0.state === "approval-requested" && part_0.approval.id === approvalId_0);
+		if (!requested || hostApprovalIdsRef.current.has(approvalId_0)) throw new Error(`Tool approval ${approvalId_0} is not waiting for a response.`);
+		const applyResponse = (applied) => {
+			if (applied) hostApprovalIdsRef.current.add(approvalId_0);
+			else hostApprovalIdsRef.current.delete(approvalId_0);
+			setToolApprovalResponses((prev_0) => {
+				const responses = new Map(prev_0);
+				if (applied) responses.set(approvalId_0, response);
+				else responses.delete(approvalId_0);
+				return responses;
+			});
+		};
+		applyResponse(true);
+		try {
+			await onRespond(response, {
+				toolCallId: requested.toolCallId,
+				toolName: getToolName(requested),
+				respondViaAISDK: async () => {
+					try {
+						await respondViaAISDK(response);
+					} finally {
+						applyResponse(false);
+					}
+				}
+			});
+		} catch (error_0) {
+			if (hostApprovalIdsRef.current.has(approvalId_0)) applyResponse(false);
+			throw error_0;
+		}
+	};
 	const hasSeededRepositoryRef = useRef(false);
 	const shouldFeedRepository = exportedMessageRepository != null && !hasSeededRepositoryRef.current && messages.length === 0;
 	const runtime = useExternalStoreRuntime({
@@ -246,6 +315,7 @@ const useAISDKRuntime = (chatHelpers, adapter = {}) => {
 		setToolStatuses,
 		setMessages: (messages_2) => chatHelpers.setMessages(messages_2.map(getVercelAIMessages).filter(Boolean).flat()),
 		onImport: (messages_3) => chatHelpers.setMessages(messages_3.map(getVercelAIMessages).filter(Boolean).flat()),
+		onVoiceTranscript: (message_0) => chatHelpers.setMessages((current) => [...current, toVoiceTranscriptUIMessage(message_0)]),
 		onExportExternalState: () => {
 			const exported = runtimeRef.current.thread.export();
 			const expandedMessages = [];
@@ -267,16 +337,16 @@ const useAISDKRuntime = (chatHelpers, adapter = {}) => {
 			return result;
 		},
 		onLoadExternalState: (repo) => {
-			const exportedRepo = toExportedMessageRepository(AISDKMessageConverter.toThreadMessages, repo);
+			const exportedRepo = toExportedMessageRepository(toThreadMessages, repo);
 			runtimeRef.current.thread.import(exportedRepo);
 		},
 		onCancel: async () => {
-			const message = chatHelpers.messages.at(-1);
-			const cancelledId = isRunning && message?.role === "assistant" ? message.id : void 0;
+			const message_1 = chatHelpers.messages.at(-1);
+			const cancelledId = isRunning && message_1?.role === "assistant" ? message_1.id : void 0;
 			if (cancelledId) {
 				const liveIds = new Set(chatHelpers.messages.map((m) => m.id));
-				setCancelledMessages((prev_0) => {
-					const kept = prev_0?.chatId === chatHelpers.id ? [...prev_0.ids].filter((id) => liveIds.has(id)) : [];
+				setCancelledMessages((prev_1) => {
+					const kept = prev_1?.chatId === chatHelpers.id ? [...prev_1.ids].filter((id) => liveIds.has(id)) : [];
 					return {
 						chatId: chatHelpers.id,
 						ids: /* @__PURE__ */ new Set([...kept, cancelledId])
@@ -285,40 +355,40 @@ const useAISDKRuntime = (chatHelpers, adapter = {}) => {
 			}
 			try {
 				await chatHelpers.stop();
-			} catch (error_0) {
-				if (!(error_0 instanceof Error && error_0.name === "AbortError")) {
+			} catch (error_1) {
+				if (!(error_1 instanceof Error && error_1.name === "AbortError")) {
 					if (cancelledId) retractCancellation(chatHelpers.id, cancelledId);
-					throw error_0;
+					throw error_1;
 				}
 			}
 		},
-		onNew: async (message_0) => {
-			const createMessage = (customToCreateMessage ?? toCreateMessage)(message_0);
-			if (!(message_0.startRun ?? message_0.role === "user")) {
-				chatHelpers.setMessages((current) => [...current, toUIMessage(createMessage, message_0.role)]);
+		onNew: async (message_2) => {
+			const createMessage = (customToCreateMessage ?? toCreateMessage)(message_2);
+			if (!(message_2.startRun ?? message_2.role === "user")) {
+				chatHelpers.setMessages((current_0) => [...current_0, toUIMessage(createMessage, message_2.role)]);
 				return;
 			}
-			lastRunConfigRef.current = message_0.runConfig;
+			lastRunConfigRef.current = message_2.runConfig;
 			await completePendingToolCalls();
-			await chatHelpers.sendMessage(createMessage, { metadata: message_0.runConfig });
+			await chatHelpers.sendMessage(createMessage, { metadata: message_2.runConfig });
 		},
-		onEdit: async (message_1) => {
-			const createMessage_0 = (customToCreateMessage ?? toCreateMessage)(message_1);
-			if (!(message_1.startRun ?? message_1.role === "user")) {
-				chatHelpers.setMessages((current_0) => [...sliceMessagesUntil(current_0, message_1.parentId), toUIMessage(createMessage_0, message_1.role)]);
+		onEdit: async (message_3) => {
+			const createMessage_0 = (customToCreateMessage ?? toCreateMessage)(message_3);
+			if (!(message_3.startRun ?? message_3.role === "user")) {
+				chatHelpers.setMessages((current_1) => [...sliceMessagesUntil(current_1, message_3.parentId), toUIMessage(createMessage_0, message_3.role)]);
 				return;
 			}
-			lastRunConfigRef.current = message_1.runConfig;
-			chatHelpers.setMessages((current_1) => sliceMessagesUntil(current_1, message_1.parentId));
-			await chatHelpers.sendMessage(createMessage_0, { metadata: message_1.runConfig });
+			lastRunConfigRef.current = message_3.runConfig;
+			chatHelpers.setMessages((current_2) => sliceMessagesUntil(current_2, message_3.parentId));
+			await chatHelpers.sendMessage(createMessage_0, { metadata: message_3.runConfig });
 		},
 		onDelete: async (messageId_0) => {
 			const threadMessages = runtimeRef.current.thread.getState().messages;
-			const messageIndex = threadMessages.findIndex((message_2) => message_2.id === messageId_0);
+			const messageIndex = threadMessages.findIndex((message_4) => message_4.id === messageId_0);
 			if (messageIndex === -1) return;
 			await deleteHistoryMessage(messageId_0);
-			const deleteIds = new Set(getExternalStoreMessages(threadMessages[messageIndex]).map((message_3) => message_3.id));
-			chatHelpers.setMessages((current_2) => current_2.filter((message_4) => !deleteIds.has(message_4.id)));
+			const deleteIds = new Set(getExternalStoreMessages(threadMessages[messageIndex]).map((message_5) => message_5.id));
+			chatHelpers.setMessages((current_3) => current_3.filter((message_6) => !deleteIds.has(message_6.id)));
 		},
 		onReload: async (parentId_0, config) => {
 			lastRunConfigRef.current = config.runConfig;
@@ -345,12 +415,7 @@ const useAISDKRuntime = (chatHelpers, adapter = {}) => {
 				}));
 			}
 		},
-		onRespondToToolApproval: ({ approvalId, approved, reason }) => Promise.resolve(chatHelpers.addToolApprovalResponse({
-			id: approvalId,
-			approved,
-			...reason != null && { reason },
-			options: { metadata: lastRunConfigRef.current }
-		})),
+		onRespondToToolApproval: customOnRespondToToolApproval ? (response_0) => respondViaHost(customOnRespondToToolApproval, response_0) : respondViaAISDK,
 		...pickExternalStoreSharedOptions(adapter),
 		...adapter.unstable_messageRepositoryInstance && { unstable_messageRepositoryInstance: adapter.unstable_messageRepositoryInstance },
 		...suggestionAdapter ? { suggestions: generatedSuggestions } : {},