@assistant-ui/react 0.15.13 → 0.15.14

package/dist/client/InMemoryThreadList.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"InMemoryThreadList.d.ts","names":[],"sources":["../../src/client/InMemoryThreadList.ts"],"mappings":";;;KAeY;EACV,SAAS,qBAAqB,gBAAgB;EAC9C,oBAAoB;EACpB;;cA+LW,gDAAkB,SAAA,0BAAA,OAAA"}
+{"version":3,"file":"InMemoryThreadList.d.ts","names":[],"sources":["../../src/client/InMemoryThreadList.ts"],"mappings":";;;KAgBY;EACV,SAAS,qBAAqB,gBAAgB;EAC9C,oBAAoB;EACpB;;cAgMW,gDAAkB,SAAA,0BAAA,OAAA"}

package/dist/client/InMemoryThreadList.js

@@ -1,8 +1,9 @@
 import { Derived, attachTransformScopes, useClientLookup, useClientResource } from "@assistant-ui/store";
 import { c } from "@assistant-ui/tap/react-shim/compiler-runtime";
 import { DataRenderers, Tools } from "@assistant-ui/core/react";
+import { generateId } from "@assistant-ui/core";
 import { useState } from "@assistant-ui/tap/react-shim";
-import { ModelContext, Suggestions } from "@assistant-ui/core/store";
+import { ModelContext } from "@assistant-ui/core/store";
 import { resource, withKey } from "@assistant-ui/tap";
 //#region src/client/InMemoryThreadList.ts
 const RESOLVED_PROMISE = Promise.resolve();
@@ -71,7 +72,7 @@ const useThreadListItemClient = (props) => {
 };
 const ThreadListItemClient = resource(useThreadListItemClient);
 const useInMemoryThreadList = (props) => {
-	const $ = c(57);
+	const $ = c(60);
 	const { thread: threadFactory, onSwitchToThread, onSwitchToNewThread } = props;
 	const [mainThreadId, setMainThreadId] = useState("main");
 	const [threads, setThreads] = useState(_temp3);
@@ -142,7 +143,7 @@ const useInMemoryThreadList = (props) => {
 	let t6;
 	if ($[8] !== onSwitchToNewThread) {
 		t6 = () => {
-			const newId = `thread-${Date.now()}`;
+			const newId = `thread-${generateId()}`;
 			setThreads((prev_5) => [...prev_5, {
 				id: newId,
 				title: "New Thread",
@@ -259,8 +260,14 @@ const useInMemoryThreadList = (props) => {
 		$[44] = t17;
 	} else t17 = $[44];
 	let t18;
-	if ($[45] !== mainThreadId || $[46] !== threadListItems || $[47] !== threads) {
-		t18 = (selector) => {
+	if ($[45] !== mainThreadClient.methods) {
+		t18 = () => mainThreadClient.methods.unstable_refetchThread?.() ?? RESOLVED_PROMISE;
+		$[45] = mainThreadClient.methods;
+		$[46] = t18;
+	} else t18 = $[46];
+	let t19;
+	if ($[47] !== mainThreadId || $[48] !== threadListItems || $[49] !== threads) {
+		t19 = (selector) => {
 			if (selector === "main") {
 				const index = threads.findIndex((t_10) => t_10.id === mainThreadId);
 				return threadListItems.get({ index: index === -1 ? 0 : index });
@@ -271,38 +278,39 @@ const useInMemoryThreadList = (props) => {
 			}
 			return threadListItems.get(selector);
 		};
-		$[45] = mainThreadId;
-		$[46] = threadListItems;
-		$[47] = threads;
-		$[48] = t18;
-	} else t18 = $[48];
-	let t19;
-	if ($[49] !== mainThreadClient.methods) {
-		t19 = () => mainThreadClient.methods;
-		$[49] = mainThreadClient.methods;
+		$[47] = mainThreadId;
+		$[48] = threadListItems;
+		$[49] = threads;
 		$[50] = t19;
 	} else t19 = $[50];
 	let t20;
-	if ($[51] !== handleSwitchToNewThread || $[52] !== handleSwitchToThread || $[53] !== t17 || $[54] !== t18 || $[55] !== t19) {
-		t20 = {
+	if ($[51] !== mainThreadClient.methods) {
+		t20 = () => mainThreadClient.methods;
+		$[51] = mainThreadClient.methods;
+		$[52] = t20;
+	} else t20 = $[52];
+	let t21;
+	if ($[53] !== handleSwitchToNewThread || $[54] !== handleSwitchToThread || $[55] !== t17 || $[56] !== t18 || $[57] !== t19 || $[58] !== t20) {
+		t21 = {
 			getState: t17,
 			switchToThread: handleSwitchToThread,
 			switchToNewThread: handleSwitchToNewThread,
 			getLoadThreadsPromise: _temp8,
 			reload: _temp9,
-			reloadMainThread: _temp0,
-			loadMore: _temp1,
-			item: t18,
-			thread: t19
+			reloadMainThread: t18,
+			loadMore: _temp0,
+			item: t19,
+			thread: t20
 		};
-		$[51] = handleSwitchToNewThread;
-		$[52] = handleSwitchToThread;
-		$[53] = t17;
-		$[54] = t18;
-		$[55] = t19;
-		$[56] = t20;
-	} else t20 = $[56];
-	return t20;
+		$[53] = handleSwitchToNewThread;
+		$[54] = handleSwitchToThread;
+		$[55] = t17;
+		$[56] = t18;
+		$[57] = t19;
+		$[58] = t20;
+		$[59] = t21;
+	} else t21 = $[59];
+	return t21;
 };
 const InMemoryThreadList = resource(useInMemoryThreadList);
 attachTransformScopes(useInMemoryThreadList, (scopes, parent) => {
@@ -324,7 +332,11 @@ attachTransformScopes(useInMemoryThreadList, (scopes, parent) => {
 	if (!scopes.modelContext && parent.modelContext.source === null) scopes.modelContext = ModelContext();
 	if (!scopes.tools && parent.tools.source === null) scopes.tools = Tools({});
 	if (!scopes.dataRenderers && parent.dataRenderers.source === null) scopes.dataRenderers = DataRenderers();
-	if (!scopes.suggestions && parent.suggestions.source === null) scopes.suggestions = Suggestions();
+	if (!scopes.suggestions && parent.suggestions.source === null) scopes.suggestions = Derived({
+		source: "thread",
+		query: {},
+		get: (aui) => aui.thread.suggestions()
+	});
 });
 function _temp() {}
 function _temp2() {}
@@ -356,9 +368,6 @@ function _temp9() {
 function _temp0() {
 	return RESOLVED_PROMISE;
 }
-function _temp1() {
-	return RESOLVED_PROMISE;
-}
 //#endregion
 export { InMemoryThreadList };
 

package/dist/client/InMemoryThreadList.js.map

@@ -1 +1 @@
-{"version":3,"file":"InMemoryThreadList.js","names":["useState","useMemo","resource","withKey","ResourceElement","ClientOutput","useClientLookup","Derived","attachTransformScopes","useClientResource","ModelContext","Suggestions","Tools","DataRenderers","RESOLVED_PROMISE","Promise","resolve","InMemoryThreadListProps","thread","threadId","onSwitchToThread","onSwitchToNewThread","ThreadData","id","title","status","custom","Record","useThreadListItemClient","props","$","_c","data","isRunning","onSwitchTo","onRename","onUpdateCustom","onArchive","onUnarchive","onDelete","t0","remoteId","undefined","externalId","state","t1","t2","t3","getState","switchTo","rename","updateCustom","archive","unarchive","delete","generateTitle","_temp","initialize","detach","_temp2","ThreadListItemClient","useInMemoryThreadList","threadFactory","mainThreadId","setMainThreadId","threads","setThreads","_temp3","handleSwitchToThread","Symbol","for","threadId_0","prev","map","t","handleRename","threadId_1","prev_0","t_0","const","handleArchive","threadId_2","prev_1","t_1","handleUnarchive","t4","threadId_3","prev_2","t_2","handleUpdateCustom","t5","threadId_4","prev_3","filter","t_3","prev_4","find","t_4","handleDelete","t6","newId","Date","now","prev_5","handleSwitchToNewThread","t7","mainThreadClient","t8","t9","t_5","title_0","custom_0","threadListItems","t10","t11","t12","t13","t14","t15","regularThreads","_temp4","archivedThreads","_temp5","_temp6","_temp7","t16","newThreadId","isLoading","isLoadingMore","hasMore","threadIds","archivedThreadIds","threadItems","main","t17","t18","selector","index","findIndex","t_10","get","index_0","t_11","t19","methods","t20","switchToThread","switchToNewThread","getLoadThreadsPromise","_temp8","reload","_temp9","reloadMainThread","_temp0","loadMore","_temp1","item","InMemoryThreadList","scopes","parent","source","query","type","aui","threadListItem","composer","modelContext","tools","dataRenderers","suggestions","t_6","t_7","t_8","t_9"],"sources":["../../src/client/InMemoryThreadList.ts"],"sourcesContent":["import { useState, useMemo } from \"react\";\nimport { resource, withKey, type ResourceElement } from \"@assistant-ui/tap\";\nimport {\n  type ClientOutput,\n  useClientLookup,\n  Derived,\n  attachTransformScopes,\n  useClientResource,\n} from \"@assistant-ui/store\";\n\nimport { ModelContext, Suggestions } from \"@assistant-ui/core/store\";\nimport { Tools, DataRenderers } from \"@assistant-ui/core/react\";\n\nconst RESOLVED_PROMISE = Promise.resolve();\n\nexport type InMemoryThreadListProps = {\n  thread: (threadId: string) => ResourceElement<ClientOutput<\"thread\">>;\n  onSwitchToThread?: (threadId: string) => void;\n  onSwitchToNewThread?: () => void;\n};\n\ntype ThreadData = {\n  id: string;\n  title?: string;\n  status: \"regular\" | \"archived\";\n  custom?: Record<string, unknown> | undefined;\n};\n\n// ThreadListItem Client\nconst useThreadListItemClient = (props: {\n  data: ThreadData;\n  isRunning: boolean;\n  onSwitchTo: () => void;\n  onRename: (title: string) => void;\n  onUpdateCustom: (custom: Record<string, unknown> | undefined) => void;\n  onArchive: () => void;\n  onUnarchive: () => void;\n  onDelete: () => void;\n}): ClientOutput<\"threadListItem\"> => {\n  const {\n    data,\n    isRunning,\n    onSwitchTo,\n    onRename,\n    onUpdateCustom,\n    onArchive,\n    onUnarchive,\n    onDelete,\n  } = props;\n  const state = useMemo(\n    () => ({\n      id: data.id,\n      remoteId: undefined,\n      externalId: undefined,\n      title: data.title,\n      status: data.status,\n      custom: data.custom,\n      isRunning,\n    }),\n    [data.id, data.title, data.status, data.custom, isRunning],\n  );\n\n  return {\n    getState: () => state,\n    switchTo: onSwitchTo,\n    rename: onRename,\n    updateCustom: onUpdateCustom,\n    archive: onArchive,\n    unarchive: onUnarchive,\n    delete: onDelete,\n    generateTitle: () => {},\n    initialize: async () => ({ remoteId: data.id, externalId: undefined }),\n    detach: () => {},\n  };\n};\n\nconst ThreadListItemClient = resource(useThreadListItemClient);\n\n// InMemoryThreadList Client\nconst useInMemoryThreadList = (\n  props: InMemoryThreadListProps,\n): ClientOutput<\"threads\"> => {\n  const {\n    thread: threadFactory,\n    onSwitchToThread,\n    onSwitchToNewThread,\n  } = props;\n\n  const [mainThreadId, setMainThreadId] = useState(\"main\");\n  const [threads, setThreads] = useState<readonly ThreadData[]>(() => [\n    { id: \"main\", title: \"Main Thread\", status: \"regular\" },\n  ]);\n\n  const handleSwitchToThread = (threadId: string) => {\n    setMainThreadId(threadId);\n    onSwitchToThread?.(threadId);\n  };\n\n  const handleRename = (threadId: string, title: string) => {\n    setThreads((prev) =>\n      prev.map((t) => (t.id === threadId ? { ...t, title } : t)),\n    );\n  };\n\n  const handleArchive = (threadId: string) => {\n    setThreads((prev) =>\n      prev.map((t) =>\n        t.id === threadId ? { ...t, status: \"archived\" as const } : t,\n      ),\n    );\n  };\n\n  const handleUnarchive = (threadId: string) => {\n    setThreads((prev) =>\n      prev.map((t) =>\n        t.id === threadId ? { ...t, status: \"regular\" as const } : t,\n      ),\n    );\n  };\n\n  const handleUpdateCustom = (\n    threadId: string,\n    custom: Record<string, unknown> | undefined,\n  ) => {\n    setThreads((prev) =>\n      prev.map((t) => (t.id === threadId ? { ...t, custom } : t)),\n    );\n  };\n\n  const handleDelete = (threadId: string) => {\n    setThreads((prev) => prev.filter((t) => t.id !== threadId));\n    setMainThreadId((prev) =>\n      prev === threadId\n        ? threads.find((t) => t.id !== threadId)?.id || \"main\"\n        : prev,\n    );\n  };\n\n  const handleSwitchToNewThread = () => {\n    const newId = `thread-${Date.now()}`;\n    setThreads((prev) => [\n      ...prev,\n      { id: newId, title: \"New Thread\", status: \"regular\" },\n    ]);\n    setMainThreadId(newId);\n    onSwitchToNewThread?.();\n  };\n\n  // Only the main thread is mounted, so it is the only thread that can run.\n  const mainThreadClient = useClientResource(threadFactory(mainThreadId));\n\n  const threadListItems = useClientLookup(\n    threads.map((t) =>\n      withKey(\n        t.id,\n        ThreadListItemClient({\n          data: t,\n          isRunning: t.id === mainThreadId && mainThreadClient.state.isRunning,\n          onSwitchTo: () => handleSwitchToThread(t.id),\n          onRename: (title) => handleRename(t.id, title),\n          onUpdateCustom: (custom) => handleUpdateCustom(t.id, custom),\n          onArchive: () => handleArchive(t.id),\n          onUnarchive: () => handleUnarchive(t.id),\n          onDelete: () => handleDelete(t.id),\n        }),\n      ),\n    ),\n  );\n\n  const state = useMemo(() => {\n    const regularThreads = threads.filter((t) => t.status === \"regular\");\n    const archivedThreads = threads.filter((t) => t.status === \"archived\");\n\n    return {\n      mainThreadId,\n      newThreadId: null,\n      isLoading: false,\n      isLoadingMore: false,\n      hasMore: false,\n      threadIds: regularThreads.map((t) => t.id),\n      archivedThreadIds: archivedThreads.map((t) => t.id),\n      threadItems: threadListItems.state,\n      main: mainThreadClient.state,\n    };\n  }, [mainThreadId, threads, threadListItems.state, mainThreadClient.state]);\n\n  return {\n    getState: () => state,\n    switchToThread: handleSwitchToThread,\n    switchToNewThread: handleSwitchToNewThread,\n    getLoadThreadsPromise: () => RESOLVED_PROMISE,\n    reload: () => RESOLVED_PROMISE,\n    reloadMainThread: () => RESOLVED_PROMISE,\n    loadMore: () => RESOLVED_PROMISE,\n    item: (selector) => {\n      if (selector === \"main\") {\n        const index = threads.findIndex((t) => t.id === mainThreadId);\n        return threadListItems.get({ index: index === -1 ? 0 : index });\n      }\n      if (\"id\" in selector) {\n        const index = threads.findIndex((t) => t.id === selector.id);\n        return threadListItems.get({ index });\n      }\n      return threadListItems.get(selector);\n    },\n    thread: () => mainThreadClient.methods,\n  };\n};\n\nexport const InMemoryThreadList = resource(useInMemoryThreadList);\n\nattachTransformScopes(useInMemoryThreadList, (scopes, parent) => {\n  scopes.thread ??= Derived({\n    source: \"threads\",\n    query: { type: \"main\" },\n    get: (aui) => aui.threads.thread(\"main\"),\n  });\n  scopes.threadListItem ??= Derived({\n    source: \"threads\",\n    query: { type: \"main\" },\n    get: (aui) => aui.threads.item(\"main\"),\n  });\n  scopes.composer ??= Derived({\n    source: \"thread\",\n    query: {},\n    get: (aui) => aui.threads.thread(\"main\").composer(),\n  });\n\n  if (!scopes.modelContext && parent.modelContext.source === null) {\n    scopes.modelContext = ModelContext();\n  }\n  if (!scopes.tools && parent.tools.source === null) {\n    scopes.tools = Tools({});\n  }\n  if (!scopes.dataRenderers && parent.dataRenderers.source === null) {\n    scopes.dataRenderers = DataRenderers();\n  }\n  if (!scopes.suggestions && parent.suggestions.source === null) {\n    scopes.suggestions = Suggestions();\n  }\n});\n"],"mappings":";;;;;;;AAaA,MAAMc,mBAAmBC,QAAQC,QAAQ;AAgBzC,MAAMY,2BAA0BC,UAAA;CAAA,MAAAC,IAAAC,EAAA,EAAA;CAU9B,MAAA,EAAAC,MAAAC,WAAAC,YAAAC,UAAAC,gBAAAC,WAAAC,aAAAC,aASIV;CAAM,IAAAW;CAAA,IAAAV,EAAA,OAAAE,KAAAN,UAAAI,EAAA,OAAAE,KAAAT,MAAAO,EAAA,OAAAE,KAAAP,UAAAK,EAAA,OAAAE,KAAAR,SAAAM,EAAA,OAAAG,WAAA;EAEDO,KAAA;GAAAjB,IACDS,KAAIT;GAAGkB,UACDC,KAAAA;GAASC,YACPD,KAAAA;GAASlB,OACdQ,KAAIR;GAAMC,QACTO,KAAIP;GAAOC,QACXM,KAAIN;GAAOO;EAErB;EAACH,EAAA,KAAAE,KAAAN;EAAAI,EAAA,KAAAE,KAAAT;EAAAO,EAAA,KAAAE,KAAAP;EAAAK,EAAA,KAAAE,KAAAR;EAAAM,EAAA,KAAAG;EAAAH,EAAA,KAAAU;CAAA,OAAAA,KAAAV,EAAA;CATH,MAAAc,QACSJ;CAUP,IAAAK;CAAA,IAAAf,EAAA,OAAAc,OAAA;EAGUC,WAAMD;EAAKd,EAAA,KAAAc;EAAAd,EAAA,KAAAe;CAAA,OAAAA,KAAAf,EAAA;CAAA,IAAAgB;CAAA,IAAAhB,EAAA,OAAAE,KAAAT,IAAA;EAQTuB,KAAA,aAAa;GAAAL,UAAYT,KAAIT;GAAGoB,YAAcD,KAAAA;EAAU;EAAEZ,EAAA,KAAAE,KAAAT;EAAAO,EAAA,KAAAgB;CAAA,OAAAA,KAAAhB,EAAA;CAAA,IAAAiB;CAAA,IAAAjB,EAAA,QAAAO,aAAAP,EAAA,QAAAS,YAAAT,EAAA,QAAAK,YAAAL,EAAA,QAAAI,cAAAJ,EAAA,QAAAQ,eAAAR,EAAA,QAAAM,kBAAAN,EAAA,QAAAe,MAAAf,EAAA,QAAAgB,IAAA;EATjEC,KAAA;GAAAC,UACKH;GAAWI,UACXf;GAAUgB,QACZf;GAAQgB,cACFf;GAAcgB,SACnBf;GAASgB,WACPf;GAAWgB,QACdf;GAAQgB,eACDC;GAAQC,YACXX;GAA0DY,QAC9DC;EACV;EAAC7B,EAAA,MAAAO;EAAAP,EAAA,MAAAS;EAAAT,EAAA,MAAAK;EAAAL,EAAA,MAAAI;EAAAJ,EAAA,MAAAQ;EAAAR,EAAA,MAAAM;EAAAN,EAAA,MAAAe;EAAAf,EAAA,MAAAgB;EAAAhB,EAAA,MAAAiB;CAAA,OAAAA,KAAAjB,EAAA;CAAA,OAXMiB;AAWN;AAGH,MAAMa,uBAAuB1D,SAAS0B,uBAAuB;AAG7D,MAAMiC,yBAAwBhC,UAAA;CAAA,MAAAC,IAAAC,EAAA,EAAA;CAG5B,MAAA,EAAAb,QAAA4C,eAAA1C,kBAAAC,wBAIIQ;CAEJ,MAAA,CAAAkC,cAAAC,mBAAwChE,SAAS,MAAM;CACvD,MAAA,CAAAiE,SAAAC,cAA8BlE,SAAgCmE,MAE7D;CAAE,IAAA3B;CAAA,IAAAV,EAAA,OAAAV,kBAAA;EAE0BoB,MAAArB,aAAA;GAC3B6C,gBAAgB7C,QAAQ;GACxBC,mBAAmBD,QAAQ;EAAC;EAC7BW,EAAA,KAAAV;EAAAU,EAAA,KAAAU;CAAA,OAAAA,KAAAV,EAAA;CAHD,MAAAsC,uBAA6B5B;CAG3B,IAAAK;CAAA,IAAAf,EAAA,OAAAuC,OAAAC,IAAA,2BAAA,GAAA;EAEmBzB,MAAA0B,YAAA/C,UAAA;GACnB0C,YAAWM,SACTA,KAAIC,KAAKC,MAAQA,EAACnD,OAAQJ,aAAT;IAAA,GAAyBuD;IAAClD;GAAY,IAAtCkD,CAAwC,CAC3D;EAAC;EACF5C,EAAA,KAAAe;CAAA,OAAAA,KAAAf,EAAA;CAJD,MAAA6C,eAAqB9B;CAInB,IAAAC;CAAA,IAAAhB,EAAA,OAAAuC,OAAAC,IAAA,2BAAA,GAAA;EAEoBxB,MAAA8B,eAAA;GACpBV,YAAWW,WACTL,OAAIC,KAAKK,QACPJ,IAACnD,OAAQJ,aAAT;IAAA,GAAyBuD;IAACjD,QAAU;GAAwB,IAA5DqD,GACF,CACF;EAAC;EACFhD,EAAA,KAAAgB;CAAA,OAAAA,KAAAhB,EAAA;CAND,MAAAkD,gBAAsBlC;CAMpB,IAAAC;CAAA,IAAAjB,EAAA,OAAAuC,OAAAC,IAAA,2BAAA,GAAA;EAEsBvB,MAAAkC,eAAA;GACtBf,YAAWgB,WACTV,OAAIC,KAAKU,QACPT,IAACnD,OAAQJ,aAAT;IAAA,GAAyBuD;IAACjD,QAAU;GAAuB,IAA3D0D,GACF,CACF;EAAC;EACFrD,EAAA,KAAAiB;CAAA,OAAAA,KAAAjB,EAAA;CAND,MAAAsD,kBAAwBrC;CAMtB,IAAAsC;CAAA,IAAAvD,EAAA,OAAAuC,OAAAC,IAAA,2BAAA,GAAA;EAEyBe,MAAAC,YAAA5D,WAAA;GAIzBwC,YAAWqB,WACTf,OAAIC,KAAKe,QAAQd,IAACnD,OAAQJ,aAAT;IAAA,GAAyBuD;IAAChD;GAAa,IAAvC8D,GAAyC,CAC5D;EAAC;EACF1D,EAAA,KAAAuD;CAAA,OAAAA,KAAAvD,EAAA;CAPD,MAAA2D,qBAA2BJ;CAOzB,IAAAK;CAAA,IAAA5D,EAAA,OAAAmC,SAAA;EAEmByB,MAAAC,eAAA;GACnBzB,YAAW0B,WAAUpB,OAAIqB,QAAQC,QAAOpB,IAACnD,OAAQJ,UAAQ,CAAC;GAC1D6C,iBAAgB+B,WACdvB,WAASrD,aACL8C,QAAO+B,MAAMC,QAAOvB,IAACnD,OAAQJ,UAAY,CAAC,EAAAI,MAA1C,SADJwE,MAGF;EAAC;EACFjE,EAAA,KAAAmC;EAAAnC,EAAA,KAAA4D;CAAA,OAAAA,KAAA5D,EAAA;CAPD,MAAAoE,eAAqBR;CAOnB,IAAAS;CAAA,IAAArE,EAAA,OAAAT,qBAAA;EAE8B8E,WAAA;GAC9B,MAAAC,QAAc,UAAUC,KAAIC,IAAK;GACjCpC,YAAWqC,WAAU,CAAA,GAChB/B,QACH;IAAAjD,IAAM6E;IAAK5E,OAAS;IAAYC,QAAU;GAAU,CAAC,CACtD;GACDuC,gBAAgBoC,KAAK;GACrB/E,sBAAsB;EAAC;EACxBS,EAAA,KAAAT;EAAAS,EAAA,KAAAqE;CAAA,OAAAA,KAAArE,EAAA;CARD,MAAA0E,0BAAgCL;CAQ9B,IAAAM;CAAA,IAAA3E,EAAA,QAAAiC,gBAAAjC,EAAA,QAAAgC,eAAA;EAGyC2C,KAAA3C,cAAcC,YAAY;EAACjC,EAAA,MAAAiC;EAAAjC,EAAA,MAAAgC;EAAAhC,EAAA,MAAA2E;CAAA,OAAAA,KAAA3E,EAAA;CAAtE,MAAA4E,mBAAyBjG,kBAAkBgG,EAA2B;CAAE,IAAAE;CAAA,IAAA7E,EAAA,QAAAoE,gBAAApE,EAAA,QAAAsC,wBAAAtC,EAAA,QAAA4E,iBAAA9D,SAAAd,EAAA,QAAAiC,gBAAAjC,EAAA,QAAAmC,SAAA;EAAA,IAAA2C;EAAA,IAAA9E,EAAA,QAAAoE,gBAAApE,EAAA,QAAAsC,wBAAAtC,EAAA,QAAA4E,iBAAA9D,SAAAd,EAAA,QAAAiC,cAAA;GAG1D6C,MAAAC,QACV1G,QACEuE,IAACnD,IACDqC,qBAAqB;IAAA5B,MACb0C;IAACzC,WACIyC,IAACnD,OAAQwC,gBAAgB2C,iBAAgB9D,MAAMX;IAAUC,kBAClDkC,qBAAqBM,IAACnD,EAAG;IAACY,WAClC2E,YAAWnC,aAAaD,IAACnD,IAAKC,OAAK;IAACY,iBAC9B2E,aAAYtB,mBAAmBf,IAACnD,IAAKG,QAAM;IAACW,iBAC3C2C,cAAcN,IAACnD,EAAG;IAACe,mBACjB8C,gBAAgBV,IAACnD,EAAG;IAACgB,gBACxB2D,aAAaxB,IAACnD,EAAG;GACnC,CAAC,CACH;GAACO,EAAA,MAAAoE;GAAApE,EAAA,MAAAsC;GAAAtC,EAAA,MAAA4E,iBAAA9D;GAAAd,EAAA,MAAAiC;GAAAjC,EAAA,MAAA8E;EAAA,OAAAA,KAAA9E,EAAA;EAbH6E,KAAA1C,QAAOQ,IAAKmC,EAcZ;EAAC9E,EAAA,MAAAoE;EAAApE,EAAA,MAAAsC;EAAAtC,EAAA,MAAA4E,iBAAA9D;EAAAd,EAAA,MAAAiC;EAAAjC,EAAA,MAAAmC;EAAAnC,EAAA,MAAA6E;CAAA,OAAAA,KAAA7E,EAAA;CAfH,MAAAkF,kBAAwB1G,gBACtBqG,EAeF;CAAE,IAAAM;CAAA,IAAAC;CAAA,IAAAC;CAAA,IAAAC;CAAA,IAAAC;CAAA,IAAAC;CAAA,IAAAV;CAAA,IAAA9E,EAAA,QAAAiC,gBAAAjC,EAAA,QAAAmC,SAAA;EAGA,MAAAsD,iBAAuBtD,QAAO4B,OAAQ2B,MAA6B;EACnE,MAAAC,kBAAwBxD,QAAO4B,OAAQ6B,MAA8B;EAGnE3D,KAAAA;EACakD,MAAA;EACFC,MAAA;EACIC,MAAA;EACNC,MAAA;EACEC,MAAAE,eAAc9C,IAAKkD,MAAW;EACtBL,MAAAG,gBAAehD,IAAKmD,MAAW;EAAC9F,EAAA,MAAAiC;EAAAjC,EAAA,MAAAmC;EAAAnC,EAAA,MAAAmF;EAAAnF,EAAA,MAAAoF;EAAApF,EAAA,MAAAqF;EAAArF,EAAA,MAAAsF;EAAAtF,EAAA,MAAAuF;EAAAvF,EAAA,MAAAwF;EAAAxF,EAAA,MAAA8E;CAAA,OAAA;EAAAK,MAAAnF,EAAA;EAAAoF,MAAApF,EAAA;EAAAqF,MAAArF,EAAA;EAAAsF,MAAAtF,EAAA;EAAAuF,MAAAvF,EAAA;EAAAwF,MAAAxF,EAAA;EAAA8E,KAAA9E,EAAA;CAAA;CAAA,IAAA+F;CAAA,IAAA/F,EAAA,QAAA4E,iBAAA9D,SAAAd,EAAA,QAAAmF,OAAAnF,EAAA,QAAAoF,OAAApF,EAAA,QAAAqF,OAAArF,EAAA,QAAAsF,OAAAtF,EAAA,QAAAuF,OAAAvF,EAAA,QAAAwF,OAAAxF,EAAA,QAAA8E,MAAA9E,EAAA,QAAAkF,gBAAApE,OAAA;EAP9CiF,MAAA;GAAA9D,cACLA;GAAY+D,aACCb;GAAIc,WACNb;GAAKc,eACDb;GAAKc,SACXb;GAAKc,WACHb;GAA+Bc,mBACvBb;GAAgCc,aACtCpB,gBAAepE;GAAMyF,MAC5B3B,iBAAgB9D;EACxB;EAACd,EAAA,MAAA4E,iBAAA9D;EAAAd,EAAA,MAAAmF;EAAAnF,EAAA,MAAAoF;EAAApF,EAAA,MAAAqF;EAAArF,EAAA,MAAAsF;EAAAtF,EAAA,MAAAuF;EAAAvF,EAAA,MAAAwF;EAAAxF,EAAA,MAAA8E;EAAA9E,EAAA,MAAAkF,gBAAApE;EAAAd,EAAA,MAAA+F;CAAA,OAAAA,MAAA/F,EAAA;CAdH,MAAAc,QAIEiF;CAWyE,IAAAS;CAAA,IAAAxG,EAAA,QAAAc,OAAA;EAG/D0F,YAAM1F;EAAKd,EAAA,MAAAc;EAAAd,EAAA,MAAAwG;CAAA,OAAAA,MAAAxG,EAAA;CAAA,IAAAyG;CAAA,IAAAzG,EAAA,QAAAiC,gBAAAjC,EAAA,QAAAkF,mBAAAlF,EAAA,QAAAmC,SAAA;EAOfsE,OAAAC,aAAA;GACJ,IAAIA,aAAa,QAAM;IACrB,MAAAC,QAAcxE,QAAOyE,WAAWC,SAAOjE,KAACnD,OAAQwC,YAAY;IAAE,OACvDiD,gBAAe4B,IAAK,EAAAH,OAASA,UAAU,KAAV,IAAAA,MAAyB,CAAC;GAAC;GAEjE,IAAI,QAAQD,UAAQ;IAClB,MAAAK,UAAc5E,QAAOyE,WAAWI,SAAOpE,KAACnD,OAAQiH,SAAQjH,EAAG;IAAE,OACtDyF,gBAAe4B,IAAK,EAAAH,OAAEA,QAAM,CAAC;GAAC;GACtC,OACMzB,gBAAe4B,IAAKJ,QAAQ;EAAC;EACrC1G,EAAA,MAAAiC;EAAAjC,EAAA,MAAAkF;EAAAlF,EAAA,MAAAmC;EAAAnC,EAAA,MAAAyG;CAAA,OAAAA,MAAAzG,EAAA;CAAA,IAAAiH;CAAA,IAAAjH,EAAA,QAAA4E,iBAAAsC,SAAA;EACOD,YAAMrC,iBAAgBsC;EAAQlH,EAAA,MAAA4E,iBAAAsC;EAAAlH,EAAA,MAAAiH;CAAA,OAAAA,MAAAjH,EAAA;CAAA,IAAAmH;CAAA,IAAAnH,EAAA,QAAA0E,2BAAA1E,EAAA,QAAAsC,wBAAAtC,EAAA,QAAAwG,OAAAxG,EAAA,QAAAyG,OAAAzG,EAAA,QAAAiH,KAAA;EAnBjCE,MAAA;GAAAjG,UACKsF;GAAWY,gBACL9E;GAAoB+E,mBACjB3C;GAAuB4C,uBACnBC;GAAsBC,QACrCC;GAAsBC,kBACZC;GAAsBC,UAC9BC;GAAsBC,MAC1BrB;GAULrH,QACO6H;EACV;EAACjH,EAAA,MAAA0E;EAAA1E,EAAA,MAAAsC;EAAAtC,EAAA,MAAAwG;EAAAxG,EAAA,MAAAyG;EAAAzG,EAAA,MAAAiH;EAAAjH,EAAA,MAAAmH;CAAA,OAAAA,MAAAnH,EAAA;CAAA,OApBMmH;AAoBN;AAGH,MAAaY,qBAAqB3J,SAAS2D,qBAAqB;AAEhErD,sBAAsBqD,wBAAwBiG,QAAQC,WAAW;CAC/DD,OAAO5I,WAAWX,QAAQ;EACxByJ,QAAQ;EACRC,OAAO,EAAEC,MAAM,OAAO;EACtBtB,MAAMuB,QAAQA,IAAIlG,QAAQ/C,OAAO,MAAM;CACzC,CAAC;CACD4I,OAAOM,mBAAmB7J,QAAQ;EAChCyJ,QAAQ;EACRC,OAAO,EAAEC,MAAM,OAAO;EACtBtB,MAAMuB,QAAQA,IAAIlG,QAAQ2F,KAAK,MAAM;CACvC,CAAC;CACDE,OAAOO,aAAa9J,QAAQ;EAC1ByJ,QAAQ;EACRC,OAAO,CAAC;EACRrB,MAAMuB,QAAQA,IAAIlG,QAAQ/C,OAAO,MAAM,CAAC,CAACmJ,SAAS;CACpD,CAAC;CAED,IAAI,CAACP,OAAOQ,gBAAgBP,OAAOO,aAAaN,WAAW,MACzDF,OAAOQ,eAAe5J,aAAa;CAErC,IAAI,CAACoJ,OAAOS,SAASR,OAAOQ,MAAMP,WAAW,MAC3CF,OAAOS,QAAQ3J,MAAM,CAAC,CAAC;CAEzB,IAAI,CAACkJ,OAAOU,iBAAiBT,OAAOS,cAAcR,WAAW,MAC3DF,OAAOU,gBAAgB3J,cAAc;CAEvC,IAAI,CAACiJ,OAAOW,eAAeV,OAAOU,YAAYT,WAAW,MACvDF,OAAOW,cAAc9J,YAAY;AAErC,CAAC;AAnN+B,SAAA6C,QAAA,CAAA;AAAA,SAAAG,SAAA,CAAA;AAkDF,SAAAQ,SAAA;CAAA,OAUwC,CAClE;EAAA5C,IAAM;EAAMC,OAAS;EAAaC,QAAU;CAAU,CAAC;AACxD;AAZ2B,SAAA+F,OAAAkD,KAAA;CAAA,OA2FmBhG,IAACjD,WAAY;AAAS;AA3FzC,SAAAiG,OAAAiD,KAAA;CAAA,OA4FoBjG,IAACjD,WAAY;AAAU;AA5F3C,SAAAkG,OAAAiD,KAAA;CAAA,OAoGalG,IAACnD;AAAG;AApGjB,SAAAqG,OAAAiD,KAAA;CAAA,OAqGsBnG,IAACnD;AAAG;AArG1B,SAAA8H,SAAA;CAAA,OA+GGvI;AAAgB;AA/GnB,SAAAyI,SAAA;CAAA,OAgHZzI;AAAgB;AAhHJ,SAAA2I,SAAA;CAAA,OAiHF3I;AAAgB;AAjHd,SAAA6I,SAAA;CAAA,OAkHV7I;AAAgB"}
+{"version":3,"file":"InMemoryThreadList.js","names":["useState","useMemo","resource","withKey","ResourceElement","ClientOutput","useClientLookup","Derived","attachTransformScopes","useClientResource","generateId","ModelContext","Tools","DataRenderers","RESOLVED_PROMISE","Promise","resolve","InMemoryThreadListProps","thread","threadId","onSwitchToThread","onSwitchToNewThread","ThreadData","id","title","status","custom","Record","useThreadListItemClient","props","$","_c","data","isRunning","onSwitchTo","onRename","onUpdateCustom","onArchive","onUnarchive","onDelete","t0","remoteId","undefined","externalId","state","t1","t2","t3","getState","switchTo","rename","updateCustom","archive","unarchive","delete","generateTitle","_temp","initialize","detach","_temp2","ThreadListItemClient","useInMemoryThreadList","threadFactory","mainThreadId","setMainThreadId","threads","setThreads","_temp3","handleSwitchToThread","Symbol","for","threadId_0","prev","map","t","handleRename","threadId_1","prev_0","t_0","const","handleArchive","threadId_2","prev_1","t_1","handleUnarchive","t4","threadId_3","prev_2","t_2","handleUpdateCustom","t5","threadId_4","prev_3","filter","t_3","prev_4","find","t_4","handleDelete","t6","newId","prev_5","handleSwitchToNewThread","t7","mainThreadClient","t8","t9","t_5","title_0","custom_0","threadListItems","t10","t11","t12","t13","t14","t15","regularThreads","_temp4","archivedThreads","_temp5","_temp6","_temp7","t16","newThreadId","isLoading","isLoadingMore","hasMore","threadIds","archivedThreadIds","threadItems","main","t17","t18","methods","unstable_refetchThread","t19","selector","index","findIndex","t_10","get","index_0","t_11","t20","t21","switchToThread","switchToNewThread","getLoadThreadsPromise","_temp8","reload","_temp9","reloadMainThread","loadMore","_temp0","item","InMemoryThreadList","scopes","parent","source","query","type","aui","threadListItem","composer","modelContext","tools","dataRenderers","suggestions","t_6","t_7","t_8","t_9"],"sources":["../../src/client/InMemoryThreadList.ts"],"sourcesContent":["import { useState, useMemo } from \"react\";\nimport { resource, withKey, type ResourceElement } from \"@assistant-ui/tap\";\nimport {\n  type ClientOutput,\n  useClientLookup,\n  Derived,\n  attachTransformScopes,\n  useClientResource,\n} from \"@assistant-ui/store\";\nimport { generateId } from \"@assistant-ui/core\";\n\nimport { ModelContext } from \"@assistant-ui/core/store\";\nimport { Tools, DataRenderers } from \"@assistant-ui/core/react\";\n\nconst RESOLVED_PROMISE = Promise.resolve();\n\nexport type InMemoryThreadListProps = {\n  thread: (threadId: string) => ResourceElement<ClientOutput<\"thread\">>;\n  onSwitchToThread?: (threadId: string) => void;\n  onSwitchToNewThread?: () => void;\n};\n\ntype ThreadData = {\n  id: string;\n  title?: string;\n  status: \"regular\" | \"archived\";\n  custom?: Record<string, unknown> | undefined;\n};\n\n// ThreadListItem Client\nconst useThreadListItemClient = (props: {\n  data: ThreadData;\n  isRunning: boolean;\n  onSwitchTo: () => void;\n  onRename: (title: string) => void;\n  onUpdateCustom: (custom: Record<string, unknown> | undefined) => void;\n  onArchive: () => void;\n  onUnarchive: () => void;\n  onDelete: () => void;\n}): ClientOutput<\"threadListItem\"> => {\n  const {\n    data,\n    isRunning,\n    onSwitchTo,\n    onRename,\n    onUpdateCustom,\n    onArchive,\n    onUnarchive,\n    onDelete,\n  } = props;\n  const state = useMemo(\n    () => ({\n      id: data.id,\n      remoteId: undefined,\n      externalId: undefined,\n      title: data.title,\n      status: data.status,\n      custom: data.custom,\n      isRunning,\n    }),\n    [data.id, data.title, data.status, data.custom, isRunning],\n  );\n\n  return {\n    getState: () => state,\n    switchTo: onSwitchTo,\n    rename: onRename,\n    updateCustom: onUpdateCustom,\n    archive: onArchive,\n    unarchive: onUnarchive,\n    delete: onDelete,\n    generateTitle: () => {},\n    initialize: async () => ({ remoteId: data.id, externalId: undefined }),\n    detach: () => {},\n  };\n};\n\nconst ThreadListItemClient = resource(useThreadListItemClient);\n\n// InMemoryThreadList Client\nconst useInMemoryThreadList = (\n  props: InMemoryThreadListProps,\n): ClientOutput<\"threads\"> => {\n  const {\n    thread: threadFactory,\n    onSwitchToThread,\n    onSwitchToNewThread,\n  } = props;\n\n  const [mainThreadId, setMainThreadId] = useState(\"main\");\n  const [threads, setThreads] = useState<readonly ThreadData[]>(() => [\n    { id: \"main\", title: \"Main Thread\", status: \"regular\" },\n  ]);\n\n  const handleSwitchToThread = (threadId: string) => {\n    setMainThreadId(threadId);\n    onSwitchToThread?.(threadId);\n  };\n\n  const handleRename = (threadId: string, title: string) => {\n    setThreads((prev) =>\n      prev.map((t) => (t.id === threadId ? { ...t, title } : t)),\n    );\n  };\n\n  const handleArchive = (threadId: string) => {\n    setThreads((prev) =>\n      prev.map((t) =>\n        t.id === threadId ? { ...t, status: \"archived\" as const } : t,\n      ),\n    );\n  };\n\n  const handleUnarchive = (threadId: string) => {\n    setThreads((prev) =>\n      prev.map((t) =>\n        t.id === threadId ? { ...t, status: \"regular\" as const } : t,\n      ),\n    );\n  };\n\n  const handleUpdateCustom = (\n    threadId: string,\n    custom: Record<string, unknown> | undefined,\n  ) => {\n    setThreads((prev) =>\n      prev.map((t) => (t.id === threadId ? { ...t, custom } : t)),\n    );\n  };\n\n  const handleDelete = (threadId: string) => {\n    setThreads((prev) => prev.filter((t) => t.id !== threadId));\n    setMainThreadId((prev) =>\n      prev === threadId\n        ? threads.find((t) => t.id !== threadId)?.id || \"main\"\n        : prev,\n    );\n  };\n\n  const handleSwitchToNewThread = () => {\n    const newId = `thread-${generateId()}`;\n    setThreads((prev) => [\n      ...prev,\n      { id: newId, title: \"New Thread\", status: \"regular\" },\n    ]);\n    setMainThreadId(newId);\n    onSwitchToNewThread?.();\n  };\n\n  // Only the main thread is mounted, so it is the only thread that can run.\n  const mainThreadClient = useClientResource(threadFactory(mainThreadId));\n\n  const threadListItems = useClientLookup(\n    threads.map((t) =>\n      withKey(\n        t.id,\n        ThreadListItemClient({\n          data: t,\n          isRunning: t.id === mainThreadId && mainThreadClient.state.isRunning,\n          onSwitchTo: () => handleSwitchToThread(t.id),\n          onRename: (title) => handleRename(t.id, title),\n          onUpdateCustom: (custom) => handleUpdateCustom(t.id, custom),\n          onArchive: () => handleArchive(t.id),\n          onUnarchive: () => handleUnarchive(t.id),\n          onDelete: () => handleDelete(t.id),\n        }),\n      ),\n    ),\n  );\n\n  const state = useMemo(() => {\n    const regularThreads = threads.filter((t) => t.status === \"regular\");\n    const archivedThreads = threads.filter((t) => t.status === \"archived\");\n\n    return {\n      mainThreadId,\n      newThreadId: null,\n      isLoading: false,\n      isLoadingMore: false,\n      hasMore: false,\n      threadIds: regularThreads.map((t) => t.id),\n      archivedThreadIds: archivedThreads.map((t) => t.id),\n      threadItems: threadListItems.state,\n      main: mainThreadClient.state,\n    };\n  }, [mainThreadId, threads, threadListItems.state, mainThreadClient.state]);\n\n  return {\n    getState: () => state,\n    switchToThread: handleSwitchToThread,\n    switchToNewThread: handleSwitchToNewThread,\n    getLoadThreadsPromise: () => RESOLVED_PROMISE,\n    reload: () => RESOLVED_PROMISE,\n    reloadMainThread: () =>\n      mainThreadClient.methods.unstable_refetchThread?.() ?? RESOLVED_PROMISE,\n    loadMore: () => RESOLVED_PROMISE,\n    item: (selector) => {\n      if (selector === \"main\") {\n        const index = threads.findIndex((t) => t.id === mainThreadId);\n        return threadListItems.get({ index: index === -1 ? 0 : index });\n      }\n      if (\"id\" in selector) {\n        const index = threads.findIndex((t) => t.id === selector.id);\n        return threadListItems.get({ index });\n      }\n      return threadListItems.get(selector);\n    },\n    thread: () => mainThreadClient.methods,\n  };\n};\n\nexport const InMemoryThreadList = resource(useInMemoryThreadList);\n\nattachTransformScopes(useInMemoryThreadList, (scopes, parent) => {\n  scopes.thread ??= Derived({\n    source: \"threads\",\n    query: { type: \"main\" },\n    get: (aui) => aui.threads.thread(\"main\"),\n  });\n  scopes.threadListItem ??= Derived({\n    source: \"threads\",\n    query: { type: \"main\" },\n    get: (aui) => aui.threads.item(\"main\"),\n  });\n  scopes.composer ??= Derived({\n    source: \"thread\",\n    query: {},\n    get: (aui) => aui.threads.thread(\"main\").composer(),\n  });\n\n  if (!scopes.modelContext && parent.modelContext.source === null) {\n    scopes.modelContext = ModelContext();\n  }\n  if (!scopes.tools && parent.tools.source === null) {\n    scopes.tools = Tools({});\n  }\n  if (!scopes.dataRenderers && parent.dataRenderers.source === null) {\n    scopes.dataRenderers = DataRenderers();\n  }\n  if (!scopes.suggestions && parent.suggestions.source === null) {\n    scopes.suggestions = Derived({\n      source: \"thread\",\n      query: {},\n      get: (aui) => aui.thread.suggestions(),\n    });\n  }\n});\n"],"mappings":";;;;;;;;AAcA,MAAMc,mBAAmBC,QAAQC,QAAQ;AAgBzC,MAAMY,2BAA0BC,UAAA;CAAA,MAAAC,IAAAC,EAAA,EAAA;CAU9B,MAAA,EAAAC,MAAAC,WAAAC,YAAAC,UAAAC,gBAAAC,WAAAC,aAAAC,aASIV;CAAM,IAAAW;CAAA,IAAAV,EAAA,OAAAE,KAAAN,UAAAI,EAAA,OAAAE,KAAAT,MAAAO,EAAA,OAAAE,KAAAP,UAAAK,EAAA,OAAAE,KAAAR,SAAAM,EAAA,OAAAG,WAAA;EAEDO,KAAA;GAAAjB,IACDS,KAAIT;GAAGkB,UACDC,KAAAA;GAASC,YACPD,KAAAA;GAASlB,OACdQ,KAAIR;GAAMC,QACTO,KAAIP;GAAOC,QACXM,KAAIN;GAAOO;EAErB;EAACH,EAAA,KAAAE,KAAAN;EAAAI,EAAA,KAAAE,KAAAT;EAAAO,EAAA,KAAAE,KAAAP;EAAAK,EAAA,KAAAE,KAAAR;EAAAM,EAAA,KAAAG;EAAAH,EAAA,KAAAU;CAAA,OAAAA,KAAAV,EAAA;CATH,MAAAc,QACSJ;CAUP,IAAAK;CAAA,IAAAf,EAAA,OAAAc,OAAA;EAGUC,WAAMD;EAAKd,EAAA,KAAAc;EAAAd,EAAA,KAAAe;CAAA,OAAAA,KAAAf,EAAA;CAAA,IAAAgB;CAAA,IAAAhB,EAAA,OAAAE,KAAAT,IAAA;EAQTuB,KAAA,aAAa;GAAAL,UAAYT,KAAIT;GAAGoB,YAAcD,KAAAA;EAAU;EAAEZ,EAAA,KAAAE,KAAAT;EAAAO,EAAA,KAAAgB;CAAA,OAAAA,KAAAhB,EAAA;CAAA,IAAAiB;CAAA,IAAAjB,EAAA,QAAAO,aAAAP,EAAA,QAAAS,YAAAT,EAAA,QAAAK,YAAAL,EAAA,QAAAI,cAAAJ,EAAA,QAAAQ,eAAAR,EAAA,QAAAM,kBAAAN,EAAA,QAAAe,MAAAf,EAAA,QAAAgB,IAAA;EATjEC,KAAA;GAAAC,UACKH;GAAWI,UACXf;GAAUgB,QACZf;GAAQgB,cACFf;GAAcgB,SACnBf;GAASgB,WACPf;GAAWgB,QACdf;GAAQgB,eACDC;GAAQC,YACXX;GAA0DY,QAC9DC;EACV;EAAC7B,EAAA,MAAAO;EAAAP,EAAA,MAAAS;EAAAT,EAAA,MAAAK;EAAAL,EAAA,MAAAI;EAAAJ,EAAA,MAAAQ;EAAAR,EAAA,MAAAM;EAAAN,EAAA,MAAAe;EAAAf,EAAA,MAAAgB;EAAAhB,EAAA,MAAAiB;CAAA,OAAAA,KAAAjB,EAAA;CAAA,OAXMiB;AAWN;AAGH,MAAMa,uBAAuB1D,SAAS0B,uBAAuB;AAG7D,MAAMiC,yBAAwBhC,UAAA;CAAA,MAAAC,IAAAC,EAAA,EAAA;CAG5B,MAAA,EAAAb,QAAA4C,eAAA1C,kBAAAC,wBAIIQ;CAEJ,MAAA,CAAAkC,cAAAC,mBAAwChE,SAAS,MAAM;CACvD,MAAA,CAAAiE,SAAAC,cAA8BlE,SAAgCmE,MAE7D;CAAE,IAAA3B;CAAA,IAAAV,EAAA,OAAAV,kBAAA;EAE0BoB,MAAArB,aAAA;GAC3B6C,gBAAgB7C,QAAQ;GACxBC,mBAAmBD,QAAQ;EAAC;EAC7BW,EAAA,KAAAV;EAAAU,EAAA,KAAAU;CAAA,OAAAA,KAAAV,EAAA;CAHD,MAAAsC,uBAA6B5B;CAG3B,IAAAK;CAAA,IAAAf,EAAA,OAAAuC,OAAAC,IAAA,2BAAA,GAAA;EAEmBzB,MAAA0B,YAAA/C,UAAA;GACnB0C,YAAWM,SACTA,KAAIC,KAAKC,MAAQA,EAACnD,OAAQJ,aAAT;IAAA,GAAyBuD;IAAClD;GAAY,IAAtCkD,CAAwC,CAC3D;EAAC;EACF5C,EAAA,KAAAe;CAAA,OAAAA,KAAAf,EAAA;CAJD,MAAA6C,eAAqB9B;CAInB,IAAAC;CAAA,IAAAhB,EAAA,OAAAuC,OAAAC,IAAA,2BAAA,GAAA;EAEoBxB,MAAA8B,eAAA;GACpBV,YAAWW,WACTL,OAAIC,KAAKK,QACPJ,IAACnD,OAAQJ,aAAT;IAAA,GAAyBuD;IAACjD,QAAU;GAAwB,IAA5DqD,GACF,CACF;EAAC;EACFhD,EAAA,KAAAgB;CAAA,OAAAA,KAAAhB,EAAA;CAND,MAAAkD,gBAAsBlC;CAMpB,IAAAC;CAAA,IAAAjB,EAAA,OAAAuC,OAAAC,IAAA,2BAAA,GAAA;EAEsBvB,MAAAkC,eAAA;GACtBf,YAAWgB,WACTV,OAAIC,KAAKU,QACPT,IAACnD,OAAQJ,aAAT;IAAA,GAAyBuD;IAACjD,QAAU;GAAuB,IAA3D0D,GACF,CACF;EAAC;EACFrD,EAAA,KAAAiB;CAAA,OAAAA,KAAAjB,EAAA;CAND,MAAAsD,kBAAwBrC;CAMtB,IAAAsC;CAAA,IAAAvD,EAAA,OAAAuC,OAAAC,IAAA,2BAAA,GAAA;EAEyBe,MAAAC,YAAA5D,WAAA;GAIzBwC,YAAWqB,WACTf,OAAIC,KAAKe,QAAQd,IAACnD,OAAQJ,aAAT;IAAA,GAAyBuD;IAAChD;GAAa,IAAvC8D,GAAyC,CAC5D;EAAC;EACF1D,EAAA,KAAAuD;CAAA,OAAAA,KAAAvD,EAAA;CAPD,MAAA2D,qBAA2BJ;CAOzB,IAAAK;CAAA,IAAA5D,EAAA,OAAAmC,SAAA;EAEmByB,MAAAC,eAAA;GACnBzB,YAAW0B,WAAUpB,OAAIqB,QAAQC,QAAOpB,IAACnD,OAAQJ,UAAQ,CAAC;GAC1D6C,iBAAgB+B,WACdvB,WAASrD,aACL8C,QAAO+B,MAAMC,QAAOvB,IAACnD,OAAQJ,UAAY,CAAC,EAAAI,MAA1C,SADJwE,MAGF;EAAC;EACFjE,EAAA,KAAAmC;EAAAnC,EAAA,KAAA4D;CAAA,OAAAA,KAAA5D,EAAA;CAPD,MAAAoE,eAAqBR;CAOnB,IAAAS;CAAA,IAAArE,EAAA,OAAAT,qBAAA;EAE8B8E,WAAA;GAC9B,MAAAC,QAAc,UAAU1F,WAAW;GACnCwD,YAAWmC,WAAU,CAAA,GAChB7B,QACH;IAAAjD,IAAM6E;IAAK5E,OAAS;IAAYC,QAAU;GAAU,CAAC,CACtD;GACDuC,gBAAgBoC,KAAK;GACrB/E,sBAAsB;EAAC;EACxBS,EAAA,KAAAT;EAAAS,EAAA,KAAAqE;CAAA,OAAAA,KAAArE,EAAA;CARD,MAAAwE,0BAAgCH;CAQ9B,IAAAI;CAAA,IAAAzE,EAAA,QAAAiC,gBAAAjC,EAAA,QAAAgC,eAAA;EAGyCyC,KAAAzC,cAAcC,YAAY;EAACjC,EAAA,MAAAiC;EAAAjC,EAAA,MAAAgC;EAAAhC,EAAA,MAAAyE;CAAA,OAAAA,KAAAzE,EAAA;CAAtE,MAAA0E,mBAAyB/F,kBAAkB8F,EAA2B;CAAE,IAAAE;CAAA,IAAA3E,EAAA,QAAAoE,gBAAApE,EAAA,QAAAsC,wBAAAtC,EAAA,QAAA0E,iBAAA5D,SAAAd,EAAA,QAAAiC,gBAAAjC,EAAA,QAAAmC,SAAA;EAAA,IAAAyC;EAAA,IAAA5E,EAAA,QAAAoE,gBAAApE,EAAA,QAAAsC,wBAAAtC,EAAA,QAAA0E,iBAAA5D,SAAAd,EAAA,QAAAiC,cAAA;GAG1D2C,MAAAC,QACVxG,QACEuE,IAACnD,IACDqC,qBAAqB;IAAA5B,MACb0C;IAACzC,WACIyC,IAACnD,OAAQwC,gBAAgByC,iBAAgB5D,MAAMX;IAAUC,kBAClDkC,qBAAqBM,IAACnD,EAAG;IAACY,WAClCyE,YAAWjC,aAAaD,IAACnD,IAAKC,OAAK;IAACY,iBAC9ByE,aAAYpB,mBAAmBf,IAACnD,IAAKG,QAAM;IAACW,iBAC3C2C,cAAcN,IAACnD,EAAG;IAACe,mBACjB8C,gBAAgBV,IAACnD,EAAG;IAACgB,gBACxB2D,aAAaxB,IAACnD,EAAG;GACnC,CAAC,CACH;GAACO,EAAA,MAAAoE;GAAApE,EAAA,MAAAsC;GAAAtC,EAAA,MAAA0E,iBAAA5D;GAAAd,EAAA,MAAAiC;GAAAjC,EAAA,MAAA4E;EAAA,OAAAA,KAAA5E,EAAA;EAbH2E,KAAAxC,QAAOQ,IAAKiC,EAcZ;EAAC5E,EAAA,MAAAoE;EAAApE,EAAA,MAAAsC;EAAAtC,EAAA,MAAA0E,iBAAA5D;EAAAd,EAAA,MAAAiC;EAAAjC,EAAA,MAAAmC;EAAAnC,EAAA,MAAA2E;CAAA,OAAAA,KAAA3E,EAAA;CAfH,MAAAgF,kBAAwBxG,gBACtBmG,EAeF;CAAE,IAAAM;CAAA,IAAAC;CAAA,IAAAC;CAAA,IAAAC;CAAA,IAAAC;CAAA,IAAAC;CAAA,IAAAV;CAAA,IAAA5E,EAAA,QAAAiC,gBAAAjC,EAAA,QAAAmC,SAAA;EAGA,MAAAoD,iBAAuBpD,QAAO4B,OAAQyB,MAA6B;EACnE,MAAAC,kBAAwBtD,QAAO4B,OAAQ2B,MAA8B;EAGnEzD,KAAAA;EACagD,MAAA;EACFC,MAAA;EACIC,MAAA;EACNC,MAAA;EACEC,MAAAE,eAAc5C,IAAKgD,MAAW;EACtBL,MAAAG,gBAAe9C,IAAKiD,MAAW;EAAC5F,EAAA,MAAAiC;EAAAjC,EAAA,MAAAmC;EAAAnC,EAAA,MAAAiF;EAAAjF,EAAA,MAAAkF;EAAAlF,EAAA,MAAAmF;EAAAnF,EAAA,MAAAoF;EAAApF,EAAA,MAAAqF;EAAArF,EAAA,MAAAsF;EAAAtF,EAAA,MAAA4E;CAAA,OAAA;EAAAK,MAAAjF,EAAA;EAAAkF,MAAAlF,EAAA;EAAAmF,MAAAnF,EAAA;EAAAoF,MAAApF,EAAA;EAAAqF,MAAArF,EAAA;EAAAsF,MAAAtF,EAAA;EAAA4E,KAAA5E,EAAA;CAAA;CAAA,IAAA6F;CAAA,IAAA7F,EAAA,QAAA0E,iBAAA5D,SAAAd,EAAA,QAAAiF,OAAAjF,EAAA,QAAAkF,OAAAlF,EAAA,QAAAmF,OAAAnF,EAAA,QAAAoF,OAAApF,EAAA,QAAAqF,OAAArF,EAAA,QAAAsF,OAAAtF,EAAA,QAAA4E,MAAA5E,EAAA,QAAAgF,gBAAAlE,OAAA;EAP9C+E,MAAA;GAAA5D,cACLA;GAAY6D,aACCb;GAAIc,WACNb;GAAKc,eACDb;GAAKc,SACXb;GAAKc,WACHb;GAA+Bc,mBACvBb;GAAgCc,aACtCpB,gBAAelE;GAAMuF,MAC5B3B,iBAAgB5D;EACxB;EAACd,EAAA,MAAA0E,iBAAA5D;EAAAd,EAAA,MAAAiF;EAAAjF,EAAA,MAAAkF;EAAAlF,EAAA,MAAAmF;EAAAnF,EAAA,MAAAoF;EAAApF,EAAA,MAAAqF;EAAArF,EAAA,MAAAsF;EAAAtF,EAAA,MAAA4E;EAAA5E,EAAA,MAAAgF,gBAAAlE;EAAAd,EAAA,MAAA6F;CAAA,OAAAA,MAAA7F,EAAA;CAdH,MAAAc,QAIE+E;CAWyE,IAAAS;CAAA,IAAAtG,EAAA,QAAAc,OAAA;EAG/DwF,YAAMxF;EAAKd,EAAA,MAAAc;EAAAd,EAAA,MAAAsG;CAAA,OAAAA,MAAAtG,EAAA;CAAA,IAAAuG;CAAA,IAAAvG,EAAA,QAAA0E,iBAAA8B,SAAA;EAKHD,YAChB7B,iBAAgB8B,QAAQC,yBAA8C,KAAtEzH;EAAuEgB,EAAA,MAAA0E,iBAAA8B;EAAAxG,EAAA,MAAAuG;CAAA,OAAAA,MAAAvG,EAAA;CAAA,IAAA0G;CAAA,IAAA1G,EAAA,QAAAiC,gBAAAjC,EAAA,QAAAgF,mBAAAhF,EAAA,QAAAmC,SAAA;EAEnEuE,OAAAC,aAAA;GACJ,IAAIA,aAAa,QAAM;IACrB,MAAAC,QAAczE,QAAO0E,WAAWC,SAAOlE,KAACnD,OAAQwC,YAAY;IAAE,OACvD+C,gBAAe+B,IAAK,EAAAH,OAASA,UAAU,KAAV,IAAAA,MAAyB,CAAC;GAAC;GAEjE,IAAI,QAAQD,UAAQ;IAClB,MAAAK,UAAc7E,QAAO0E,WAAWI,SAAOrE,KAACnD,OAAQkH,SAAQlH,EAAG;IAAE,OACtDuF,gBAAe+B,IAAK,EAAAH,OAAEA,QAAM,CAAC;GAAC;GACtC,OACM5B,gBAAe+B,IAAKJ,QAAQ;EAAC;EACrC3G,EAAA,MAAAiC;EAAAjC,EAAA,MAAAgF;EAAAhF,EAAA,MAAAmC;EAAAnC,EAAA,MAAA0G;CAAA,OAAAA,MAAA1G,EAAA;CAAA,IAAAkH;CAAA,IAAAlH,EAAA,QAAA0E,iBAAA8B,SAAA;EACOU,YAAMxC,iBAAgB8B;EAAQxG,EAAA,MAAA0E,iBAAA8B;EAAAxG,EAAA,MAAAkH;CAAA,OAAAA,MAAAlH,EAAA;CAAA,IAAAmH;CAAA,IAAAnH,EAAA,QAAAwE,2BAAAxE,EAAA,QAAAsC,wBAAAtC,EAAA,QAAAsG,OAAAtG,EAAA,QAAAuG,OAAAvG,EAAA,QAAA0G,OAAA1G,EAAA,QAAAkH,KAAA;EApBjCC,MAAA;GAAAjG,UACKoF;GAAWc,gBACL9E;GAAoB+E,mBACjB7C;GAAuB8C,uBACnBC;GAAsBC,QACrCC;GAAsBC,kBACZnB;GACuDoB,UAC/DC;GAAsBC,MAC1BnB;GAULtH,QACO8H;EACV;EAAClH,EAAA,MAAAwE;EAAAxE,EAAA,MAAAsC;EAAAtC,EAAA,MAAAsG;EAAAtG,EAAA,MAAAuG;EAAAvG,EAAA,MAAA0G;EAAA1G,EAAA,MAAAkH;EAAAlH,EAAA,MAAAmH;CAAA,OAAAA,MAAAnH,EAAA;CAAA,OArBMmH;AAqBN;AAGH,MAAaW,qBAAqB1J,SAAS2D,qBAAqB;AAEhErD,sBAAsBqD,wBAAwBgG,QAAQC,WAAW;CAC/DD,OAAO3I,WAAWX,QAAQ;EACxBwJ,QAAQ;EACRC,OAAO,EAAEC,MAAM,OAAO;EACtBpB,MAAMqB,QAAQA,IAAIjG,QAAQ/C,OAAO,MAAM;CACzC,CAAC;CACD2I,OAAOM,mBAAmB5J,QAAQ;EAChCwJ,QAAQ;EACRC,OAAO,EAAEC,MAAM,OAAO;EACtBpB,MAAMqB,QAAQA,IAAIjG,QAAQ0F,KAAK,MAAM;CACvC,CAAC;CACDE,OAAOO,aAAa7J,QAAQ;EAC1BwJ,QAAQ;EACRC,OAAO,CAAC;EACRnB,MAAMqB,QAAQA,IAAIjG,QAAQ/C,OAAO,MAAM,CAAC,CAACkJ,SAAS;CACpD,CAAC;CAED,IAAI,CAACP,OAAOQ,gBAAgBP,OAAOO,aAAaN,WAAW,MACzDF,OAAOQ,eAAe1J,aAAa;CAErC,IAAI,CAACkJ,OAAOS,SAASR,OAAOQ,MAAMP,WAAW,MAC3CF,OAAOS,QAAQ1J,MAAM,CAAC,CAAC;CAEzB,IAAI,CAACiJ,OAAOU,iBAAiBT,OAAOS,cAAcR,WAAW,MAC3DF,OAAOU,gBAAgB1J,cAAc;CAEvC,IAAI,CAACgJ,OAAOW,eAAeV,OAAOU,YAAYT,WAAW,MACvDF,OAAOW,cAAcjK,QAAQ;EAC3BwJ,QAAQ;EACRC,OAAO,CAAC;EACRnB,MAAMqB,QAAQA,IAAIhJ,OAAOsJ,YAAY;CACvC,CAAC;AAEL,CAAC;AAxN+B,SAAAhH,QAAA,CAAA;AAAA,SAAAG,SAAA,CAAA;AAkDF,SAAAQ,SAAA;CAAA,OAUwC,CAClE;EAAA5C,IAAM;EAAMC,OAAS;EAAaC,QAAU;CAAU,CAAC;AACxD;AAZ2B,SAAA6F,OAAAmD,KAAA;CAAA,OA2FmB/F,IAACjD,WAAY;AAAS;AA3FzC,SAAA+F,OAAAkD,KAAA;CAAA,OA4FoBhG,IAACjD,WAAY;AAAU;AA5F3C,SAAAgG,OAAAkD,KAAA;CAAA,OAoGajG,IAACnD;AAAG;AApGjB,SAAAmG,OAAAkD,KAAA;CAAA,OAqGsBlG,IAACnD;AAAG;AArG1B,SAAA8H,SAAA;CAAA,OA+GGvI;AAAgB;AA/GnB,SAAAyI,SAAA;CAAA,OAgHZzI;AAAgB;AAhHJ,SAAA4I,SAAA;CAAA,OAmHV5I;AAAgB"}

package/dist/index.d.ts

@@ -75,7 +75,7 @@ import "./mcp-apps/index.js";
 import { AssistantClient, AssistantEventCallback, AssistantEventName, AssistantEventPayload, AssistantEventScope, AssistantEventSelector, AssistantState, AuiConfig, AuiIf, AuiProvider, useAui, useAuiEvent, useAuiState } from "@assistant-ui/store";
 import { AssistantContextConfig, AssistantDataUI, AssistantDataUIProps, AssistantInteractableProps, AssistantTool, AssistantToolProps, AssistantToolUI, AssistantToolUIProps, DataMessagePartComponent, DataMessagePartProps, DataRenderers, EmptyMessagePartComponent, EmptyMessagePartProps, EnrichedPartState, FileMessagePartComponent, FileMessagePartProps, GenerativeUIComponentRegistry, GenerativeUIMessagePartComponent, GenerativeUIMessagePartProps, GenerativeUIRender, GenerativeUIRenderError, GenerativeUIRenderProps, GroupByContext, ImageMessagePartComponent, ImageMessagePartProps, Interactables, McpAppResourceOutput, McpToolkitDefinition, McpToolkitEntry, McpToolkitToolConfig, PartState, ProviderToolConfig, QuoteMessagePartComponent, QuoteMessagePartProps, ReasoningGroupComponent, ReasoningGroupProps, ReasoningMessagePartComponent, ReasoningMessagePartProps, RuntimeAdapterProvider, RuntimeAdapters, SourceMessagePartComponent, SourceMessagePartProps, TextMessagePartComponent, TextMessagePartProps, ToolArgsStatus, ToolCallMessagePartComponent, ToolCallMessagePartProps, ToolCallText, ToolDefinition, Toolkit, ToolkitDefinition, ToolkitDefinitionEntry, Tools, Unstable_AudioMessagePartComponent, Unstable_AudioMessagePartProps, Unstable_InferInteractableState, Unstable_InteractableConfig, Unstable_InteractableDefinition, Unstable_InteractablePersistedState, Unstable_InteractablePersistenceAdapter, Unstable_InteractablePersistenceStatus, Unstable_InteractableRegistration, Unstable_InteractableStateSchema, Unstable_InteractableToolConfig, Unstable_InteractableToolRenderProps, Unstable_InteractableVersionInfo, Unstable_InteractablesClientSchema, Unstable_InteractablesConfig, Unstable_InteractablesMethods, Unstable_InteractablesState, defineMcpToolkit, defineToolkit, externalTool, groupPartByType, hitl, hitlTool, humanTool, makeAssistantDataUI, makeAssistantTool, makeAssistantToolUI, providerTool, stubTool, unstable_Interactables, unstable_interactableTool, unstable_useInteractable, unstable_useInteractableState, unstable_useInteractableVersions, unstable_useThreadMessageIds, useAssistantContext, useAssistantDataUI, useAssistantInstructions, useAssistantInteractable, useAssistantTool, useAssistantToolUI, useAuiToolOverrides, useExternalStoreSharedOptions, useInlineRender, useInteractableState, useRuntimeAdapters, useToolArgsStatus, useVoiceControls, useVoiceState, useVoiceVolume } from "@assistant-ui/core/react";
 import { AssistantCloud } from "assistant-cloud";
-import { AddToolResultOptions, AppendMessage, AssistantFrameHost, AssistantFrameProvider, AssistantRuntime, Attachment, AttachmentAdapter, AttachmentRuntime, AttachmentState, AttachmentStatus, ChatModelAdapter, ChatModelRunOptions, ChatModelRunResult, ChatModelRunUpdate, CompleteAttachment, ComposerRuntime, ComposerState, CompositeAttachmentAdapter, CreateAppendMessage, CreateAttachment, CreateResumeRunConfig, CreateStartRunConfig, DataMessagePart, DictationAdapter, DictationState, EditComposerRuntime, EditComposerState, ExportedMessageRepository, ExportedMessageRepositoryItem, ExternalStoreAdapter, ExternalStoreBranchChange, ExternalStoreMessageConverter, ExternalStoreSharedOptions, ExternalStoreThreadData, ExternalStoreThreadListAdapter, ExternalThreadBranchAdapter, ExternalThreadQueueAdapter, FRAME_MESSAGE_CHANNEL, FeedbackAdapter, FileMessagePart, FrameMessage, FrameMessageType, GenerativeUIMessagePart, GenerativeUINode, GenerativeUISpec, GenericThreadHistoryAdapter, ImageMessagePart, InMemoryThreadListAdapter, LanguageModelConfig, LanguageModelV1CallSettings, LocalRuntimeOptionsBase, MessageFormatAdapter, MessageFormatItem, MessageFormatRepository, MessagePartRuntime, MessagePartState, MessagePartStatus, MessagePartStreamStatus, MessageQueueController, MessageQueueDriver, MessageRuntime, MessageState, MessageStatus, MessageStorageEntry, MessageTiming, ModelContext, ModelContextProvider, ModelContextRegistry, ModelContextRegistryInstructionHandle, ModelContextRegistryProviderHandle, ModelContextRegistryToolHandle, PartProviderMetadata, PendingAttachment, QuoteInfo, RealtimeVoiceAdapter, ReasoningMessagePart, RemoteThreadListAdapter, RemoteThreadListProviderComponent, RespondToToolApprovalOptions, SerializedModelContext, SerializedTool, SimpleImageAttachmentAdapter, SimpleTextAttachmentAdapter, SourceMessagePart, SourceProviderMetadata, SpeechSynthesisAdapter, SubmitFeedbackOptions, SuggestionAdapter, TextMessagePart, ThreadAssistantMessage, ThreadAssistantMessagePart, ThreadComposerRuntime, ThreadComposerState, ThreadHistoryAdapter, ThreadListItemRuntime, ThreadListItemState, ThreadListItemStatus, ThreadListRuntime, ThreadListState, ThreadMessage, ThreadMessageLike, ThreadRuntime, ThreadState, ThreadSuggestion, ThreadSystemMessage, ThreadUserMessage, ThreadUserMessagePart, ToolApprovalOption, ToolApprovalOptionKind, ToolApprovalResponse, ToolCallMessagePart, ToolCallMessagePartStatus, ToolCallTiming, ToolModelContentPart, Unstable_AudioMessagePart, Unstable_DirectiveFormatter, Unstable_DirectiveSegment, Unstable_InteractableSnapshotEntry, Unstable_InteractableVersion, Unstable_TriggerItem, Unsubscribe, VoiceSessionControls, VoiceSessionHelpers, VoiceSessionState, WebSpeechDictationAdapter, WebSpeechSynthesisAdapter, bindExternalStoreMessage, createMessageQueue, createVoiceSession, fromThreadMessageLike, generateId, getExternalStoreMessages, mergeModelContexts, pickExternalStoreSharedOptions, tool, unstable_defaultDirectiveFormatter, unstable_formatInteractableSnapshot, unstable_getInteractableSnapshots, unstable_getInteractableVersions } from "@assistant-ui/core";
+import { AddToolResultOptions, AppendMessage, AssistantFrameHost, AssistantFrameProvider, AssistantRuntime, Attachment, AttachmentAdapter, AttachmentRuntime, AttachmentState, AttachmentStatus, ChatModelAdapter, ChatModelRunOptions, ChatModelRunResult, ChatModelRunUpdate, CompleteAttachment, ComposerRuntime, ComposerState, CompositeAttachmentAdapter, CreateAppendMessage, CreateAttachment, CreateResumeRunConfig, CreateStartRunConfig, DataMessagePart, DictationAdapter, DictationState, EditComposerRuntime, EditComposerState, ExportedMessageRepository, ExportedMessageRepositoryItem, ExternalStoreAdapter, ExternalStoreBranchChange, ExternalStoreMessageConverter, ExternalStoreSharedOptions, ExternalStoreThreadData, ExternalStoreThreadListAdapter, ExternalThreadBranchAdapter, ExternalThreadQueueAdapter, FRAME_MESSAGE_CHANNEL, FeedbackAdapter, FileMessagePart, FrameMessage, FrameMessageType, GenerativeUIMessagePart, GenerativeUINode, GenerativeUISpec, GenericThreadHistoryAdapter, ImageMessagePart, InMemoryThreadListAdapter, LanguageModelConfig, LanguageModelV1CallSettings, LocalRuntimeOptionsBase, MessageFormatAdapter, MessageFormatItem, MessageFormatRepository, MessageNotSentError, MessagePartRuntime, MessagePartState, MessagePartStatus, MessagePartStreamStatus, MessageQueueController, MessageQueueDriver, MessageRuntime, MessageState, MessageStatus, MessageStorageEntry, MessageTiming, ModelContext, ModelContextProvider, ModelContextRegistry, ModelContextRegistryInstructionHandle, ModelContextRegistryProviderHandle, ModelContextRegistryToolHandle, PartProviderMetadata, PendingAttachment, QuoteInfo, RealtimeVoiceAdapter, ReasoningMessagePart, RemoteThreadListAdapter, RemoteThreadListProviderComponent, RespondToToolApprovalOptions, SerializedModelContext, SerializedTool, SimpleImageAttachmentAdapter, SimpleTextAttachmentAdapter, SourceMessagePart, SourceProviderMetadata, SpeechSynthesisAdapter, SubmitFeedbackOptions, SuggestionAdapter, TextMessagePart, ThreadAssistantMessage, ThreadAssistantMessagePart, ThreadComposerRuntime, ThreadComposerState, ThreadHistoryAdapter, ThreadListItemRuntime, ThreadListItemState, ThreadListItemStatus, ThreadListRuntime, ThreadListState, ThreadMessage, ThreadMessageLike, ThreadRuntime, ThreadState, ThreadSuggestion, ThreadSystemMessage, ThreadUserMessage, ThreadUserMessagePart, ToolApprovalOption, ToolApprovalOptionKind, ToolApprovalResponse, ToolCallMessagePart, ToolCallMessagePartStatus, ToolCallTiming, ToolModelContentPart, Unstable_AudioMessagePart, Unstable_DirectiveFormatter, Unstable_DirectiveSegment, Unstable_InteractableSnapshotEntry, Unstable_InteractableVersion, Unstable_TriggerItem, Unsubscribe, VoiceSessionControls, VoiceSessionHelpers, VoiceSessionState, WebSpeechDictationAdapter, WebSpeechSynthesisAdapter, bindExternalStoreMessage, createMessageQueue, createVoiceSession, fromThreadMessageLike, generateId, getExternalStoreMessages, isMessageNotSentError, mergeModelContexts, pickExternalStoreSharedOptions, tool, unstable_defaultDirectiveFormatter, unstable_formatInteractableSnapshot, unstable_getInteractableSnapshots, unstable_getInteractableVersions } from "@assistant-ui/core";
 import { Tool } from "assistant-stream";
 import { ChainOfThoughtClient, ComposerSendOptions, ExternalThread, ExternalThreadMessage, ExternalThreadProps, ModelContext as ModelContextClient, QueueItemMethods, QueueItemState, SingleThreadList, SuggestionConfig, Suggestions } from "@assistant-ui/core/store";
-export { actionBarMore_d_exports as ActionBarMorePrimitive, actionBar_d_exports as ActionBarPrimitive, type AddToolResultOptions, type AppendMessage, type Assistant, type AssistantClient, AssistantCloud, type AssistantContextConfig, type AssistantDataUI, type AssistantDataUIProps, type AssistantEventCallback, type AssistantEventName, type AssistantEventPayload, type AssistantEventScope, type AssistantEventSelector, AssistantFrameHost, AssistantFrameProvider, type AssistantInteractableProps, assistantModal_d_exports as AssistantModalPrimitive, type AssistantRuntime, AssistantRuntimeProvider, type AssistantState, type AssistantTool, type AssistantToolProps, type AssistantToolUI, type AssistantToolUIProps, type AssistantTransportCommand, type AssistantTransportConnectionMetadata, type AssistantTransportProtocol, type Attachment, type AttachmentAdapter, attachment_d_exports as AttachmentPrimitive, type AttachmentRuntime, type AttachmentState, type AttachmentStatus, AuiConfig, AuiIf, AuiProvider, branchPicker_d_exports as BranchPickerPrimitive, ChainOfThoughtByIndicesProvider, ChainOfThoughtClient, chainOfThought_d_exports as ChainOfThoughtPrimitive, type ChatModelAdapter, type ChatModelRunOptions, type ChatModelRunResult, type ChatModelRunUpdate, CloudFileAttachmentAdapter, type CompleteAttachment, ComposerAttachmentByIndexProvider, composer_d_exports as ComposerPrimitive, type ComposerRuntime, type ComposerSendOptions, type ComposerState, CompositeAttachmentAdapter, type CreateAppendMessage, type CreateAttachment, type CreateResumeRunConfig, type CreateStartRunConfig, type DataMessagePart, type DataMessagePartComponent, type DataMessagePartProps, DataRenderers, DevToolsHooks, DevToolsProviderApi, type DictationAdapter, type DictationState, type EditComposerRuntime, type EditComposerState, type EmptyMessagePartComponent, type EmptyMessagePartProps, type EnrichedPartState, error_d_exports as ErrorPrimitive, ExportedMessageRepository, type ExportedMessageRepositoryItem, type ExternalStoreAdapter, type ExternalStoreBranchChange, type ExternalStoreMessageConverter, type ExternalStoreSharedOptions, type ExternalStoreThreadData, type ExternalStoreThreadListAdapter, ExternalThread, type ExternalThreadBranchAdapter, type ExternalThreadMessage, type ExternalThreadProps, type ExternalThreadQueueAdapter, FRAME_MESSAGE_CHANNEL, type FeedbackAdapter, type FileMessagePart, type FileMessagePartComponent, type FileMessagePartProps, type FrameMessage, type FrameMessageType, type GenerativeUIComponentRegistry, type GenerativeUIMessagePart, type GenerativeUIMessagePartComponent, type GenerativeUIMessagePartProps, type GenerativeUINode, GenerativeUIRender, GenerativeUIRenderError, type GenerativeUIRenderProps, type GenerativeUISpec, type GenericThreadHistoryAdapter, type GroupByContext, internal_d_exports as INTERNAL, type ImageMessagePart, type ImageMessagePartComponent, type ImageMessagePartProps, InMemoryThreadList, InMemoryThreadListAdapter, type InMemoryThreadListProps, Interactables, type LanguageModelConfig, type LanguageModelV1CallSettings, type LocalRuntimeOptions, type LocalRuntimeOptionsBase, type McpAppDisplayMode, type McpAppHostContext, type McpAppHostInfo, type McpAppMetadata, McpAppRenderer, type McpAppRendererOptions, type McpAppResource, type McpAppResourceCSP, type McpAppResourceMeta, type McpAppResourceOutput, type McpAppSandboxConfig, type McpAppToolCallParams, type McpAppsHost, McpAppsRemoteHost, type McpAppsRemoteHostOptions, type McpToolkitDefinition, type McpToolkitEntry, type McpToolkitToolConfig, MessageAttachmentByIndexProvider, MessageByIndexProvider, type MessageFormatAdapter, type MessageFormatItem, type MessageFormatRepository, messagePart_d_exports as MessagePartPrimitive, type MessagePartRuntime, type MessagePartState, type MessagePartStatus, type MessagePartStreamStatus, message_d_exports as MessagePrimitive, MessageProvider, type MessageQueueController, type MessageQueueDriver, type MessageRuntime, type MessageState, type MessageStatus, type MessageStorageEntry, type MessageTiming, type ModelContext, ModelContextClient, type ModelContextProvider, ModelContextRegistry, type ModelContextRegistryInstructionHandle, type ModelContextRegistryProviderHandle, type ModelContextRegistryToolHandle, PartByIndexProvider, type PartProviderMetadata, type PartState, type PendingAttachment, type ProviderToolConfig, type QueueItemMethods, queueItem_d_exports as QueueItemPrimitive, type QueueItemState, type QuoteInfo, type QuoteMessagePartComponent, type QuoteMessagePartProps, ReadonlyThreadProvider, type RealtimeVoiceAdapter, type ReasoningGroupComponent, type ReasoningGroupProps, type ReasoningMessagePart, type ReasoningMessagePartComponent, type ReasoningMessagePartProps, type RemoteThreadListAdapter, type RemoteThreadListProviderComponent, type RespondToToolApprovalOptions, RuntimeAdapterProvider, type RuntimeAdapters, selectionToolbar_d_exports as SelectionToolbarPrimitive, type SendCommandsRequestBody, type SerializedModelContext, type SerializedTool, SimpleImageAttachmentAdapter, SimpleTextAttachmentAdapter, SingleThreadList, type SmoothOptions, type SourceMessagePart, type SourceMessagePartComponent, type SourceMessagePartProps, type SourceProviderMetadata, type SpeechSynthesisAdapter, type SubmitFeedbackOptions, type SuggestionAdapter, SuggestionByIndexProvider, type SuggestionConfig, suggestion_d_exports as SuggestionPrimitive, Suggestions, type TextMessagePart, type TextMessagePartComponent, type TextMessagePartProps, TextMessagePartProvider, type ThreadAssistantMessage, type ThreadAssistantMessagePart, type ThreadComposerRuntime, type ThreadComposerState, type ThreadHistoryAdapter, ThreadListItemByIndexProvider, threadListItemMore_d_exports as ThreadListItemMorePrimitive, threadListItem_d_exports as ThreadListItemPrimitive, type ThreadListItemRuntime, ThreadListItemRuntimeProvider, type ThreadListItemState, type ThreadListItemStatus, threadList_d_exports as ThreadListPrimitive, type ThreadListRuntime, type ThreadListState, type ThreadMessage, type ThreadMessageLike, thread_d_exports as ThreadPrimitive, type ThreadRuntime, type ThreadState, type ThreadSuggestion, type ThreadSystemMessage, type ThreadUserMessage, type ThreadUserMessagePart, type ThreadViewportState, type Tool, type ToolApprovalOption, type ToolApprovalOptionKind, type ToolApprovalResponse, type ToolArgsStatus, type ToolCallMessagePart, type ToolCallMessagePartComponent, type ToolCallMessagePartMcpMetadata, type ToolCallMessagePartProps, type ToolCallMessagePartStatus, type ToolCallText, type ToolCallTiming, type ToolDefinition, type ToolExecutionStatus, type ToolModelContentPart, type Toolkit, type ToolkitDefinition, type ToolkitDefinitionEntry, Tools, type Unstable_AudioMessagePart, type Unstable_AudioMessagePartComponent, type Unstable_AudioMessagePartProps, type Unstable_ComposerInput, type Unstable_ComposerInputHistory, type Unstable_DirectiveFormatter, type Unstable_DirectiveSegment, type Unstable_IconComponent, type Unstable_InferInteractableState, type Unstable_InteractableConfig, type Unstable_InteractableDefinition, type Unstable_InteractablePersistedState, type Unstable_InteractablePersistenceAdapter, type Unstable_InteractablePersistenceStatus, type Unstable_InteractableRegistration, type Unstable_InteractableSnapshotEntry, type Unstable_InteractableStateSchema, type Unstable_InteractableToolConfig, type Unstable_InteractableToolRenderProps, type Unstable_InteractableVersion, type Unstable_InteractableVersionInfo, type Unstable_InteractablesClientSchema, type Unstable_InteractablesConfig, type Unstable_InteractablesMethods, type Unstable_InteractablesState, type Unstable_Mention, type Unstable_MentionCategory, type Unstable_MentionDirective, type Unstable_MessageStallDetection, type Unstable_MessageStallDetectionOptions, type Unstable_ModelContextToolsOptions, type RegisteredTrigger as Unstable_RegisteredTrigger, type Unstable_SlashCommand, type Unstable_SlashCommandAction, type TriggerBehavior as Unstable_TriggerBehavior, type Unstable_TriggerItem, type Unstable_TriggerPopoverAriaProps, type Unstable_UseComposerInputOptions, type Unstable_UseLiveCompletionAdapterOptions, type Unstable_UseMentionAdapterOptions, type Unstable_UseSlashCommandAdapterOptions, type Unsubscribe, type VoiceSessionControls, type VoiceSessionHelpers, type VoiceSessionState, WebSpeechDictationAdapter, WebSpeechSynthesisAdapter, bindExternalStoreMessage, createMessageQueue, createVoiceSession, defineMcpToolkit, defineToolkit, externalTool, fromThreadMessageLike, generateId, getExternalStoreMessages, getMcpAppFromToolPart, groupPartByType, hitl, hitlTool, humanTool, makeAssistantDataUI, makeAssistantTool, makeAssistantToolUI, makeAssistantVisible, mergeModelContexts, pickExternalStoreSharedOptions, providerTool, stubTool, tool, unstable_Interactables, convertExternalMessages as unstable_convertExternalMessages, createMessageConverter as unstable_createMessageConverter, unstable_defaultDirectiveFormatter, unstable_formatInteractableSnapshot, unstable_getInteractableSnapshots, unstable_getInteractableVersions, unstable_interactableTool, unstable_useComposerInput, unstable_useComposerInputHistory, unstable_useInteractable, unstable_useInteractableState, unstable_useInteractableVersions, unstable_useLiveCompletionAdapter, unstable_useMentionAdapter, unstable_useMessageStallDetection, unstable_useSlashCommandAdapter, unstable_useThreadMessageIds, unstable_useTriggerPopoverAriaProps, useTriggerPopoverRootContext as unstable_useTriggerPopoverRootContext, useTriggerPopoverRootContextOptional as unstable_useTriggerPopoverRootContextOptional, useTriggerPopoverScopeContext as unstable_useTriggerPopoverScopeContext, useTriggerPopoverScopeContextOptional as unstable_useTriggerPopoverScopeContextOptional, useTriggerPopoverTriggers as unstable_useTriggerPopoverTriggers, useTriggerPopoverTriggersOptional as unstable_useTriggerPopoverTriggersOptional, useAssistantContext, useAssistantDataUI, useAssistantFrameHost, useAssistantInstructions, useAssistantInteractable, useAssistantTool, useAssistantToolUI, useAssistantTransportRuntime, useAssistantTransportSendCommand, useAssistantTransportState, useAui, useAuiEvent, useAuiState, useAuiToolOverrides, useCloudThreadListAdapter, useCloudThreadListRuntime, useExternalMessageConverter, useExternalStoreRuntime, useExternalStoreSharedOptions, useInlineRender, useInteractableState, useLocalRuntime, useMessagePartData, useMessagePartFile, useMessagePartImage, useMessagePartReasoning, useMessagePartSource, useMessagePartText, useMessageQuote, useMessageTiming, useRemoteThreadListRuntime, useRuntimeAdapters, useScrollLock, useSmooth, useThreadViewport, useThreadViewportAutoScroll, useThreadViewportStore, useToolArgsStatus, useToolCallElapsed, useVoiceControls, useVoiceState, useVoiceVolume };
+export { actionBarMore_d_exports as ActionBarMorePrimitive, actionBar_d_exports as ActionBarPrimitive, type AddToolResultOptions, type AppendMessage, type Assistant, type AssistantClient, AssistantCloud, type AssistantContextConfig, type AssistantDataUI, type AssistantDataUIProps, type AssistantEventCallback, type AssistantEventName, type AssistantEventPayload, type AssistantEventScope, type AssistantEventSelector, AssistantFrameHost, AssistantFrameProvider, type AssistantInteractableProps, assistantModal_d_exports as AssistantModalPrimitive, type AssistantRuntime, AssistantRuntimeProvider, type AssistantState, type AssistantTool, type AssistantToolProps, type AssistantToolUI, type AssistantToolUIProps, type AssistantTransportCommand, type AssistantTransportConnectionMetadata, type AssistantTransportProtocol, type Attachment, type AttachmentAdapter, attachment_d_exports as AttachmentPrimitive, type AttachmentRuntime, type AttachmentState, type AttachmentStatus, AuiConfig, AuiIf, AuiProvider, branchPicker_d_exports as BranchPickerPrimitive, ChainOfThoughtByIndicesProvider, ChainOfThoughtClient, chainOfThought_d_exports as ChainOfThoughtPrimitive, type ChatModelAdapter, type ChatModelRunOptions, type ChatModelRunResult, type ChatModelRunUpdate, CloudFileAttachmentAdapter, type CompleteAttachment, ComposerAttachmentByIndexProvider, composer_d_exports as ComposerPrimitive, type ComposerRuntime, type ComposerSendOptions, type ComposerState, CompositeAttachmentAdapter, type CreateAppendMessage, type CreateAttachment, type CreateResumeRunConfig, type CreateStartRunConfig, type DataMessagePart, type DataMessagePartComponent, type DataMessagePartProps, DataRenderers, DevToolsHooks, DevToolsProviderApi, type DictationAdapter, type DictationState, type EditComposerRuntime, type EditComposerState, type EmptyMessagePartComponent, type EmptyMessagePartProps, type EnrichedPartState, error_d_exports as ErrorPrimitive, ExportedMessageRepository, type ExportedMessageRepositoryItem, type ExternalStoreAdapter, type ExternalStoreBranchChange, type ExternalStoreMessageConverter, type ExternalStoreSharedOptions, type ExternalStoreThreadData, type ExternalStoreThreadListAdapter, ExternalThread, type ExternalThreadBranchAdapter, type ExternalThreadMessage, type ExternalThreadProps, type ExternalThreadQueueAdapter, FRAME_MESSAGE_CHANNEL, type FeedbackAdapter, type FileMessagePart, type FileMessagePartComponent, type FileMessagePartProps, type FrameMessage, type FrameMessageType, type GenerativeUIComponentRegistry, type GenerativeUIMessagePart, type GenerativeUIMessagePartComponent, type GenerativeUIMessagePartProps, type GenerativeUINode, GenerativeUIRender, GenerativeUIRenderError, type GenerativeUIRenderProps, type GenerativeUISpec, type GenericThreadHistoryAdapter, type GroupByContext, internal_d_exports as INTERNAL, type ImageMessagePart, type ImageMessagePartComponent, type ImageMessagePartProps, InMemoryThreadList, InMemoryThreadListAdapter, type InMemoryThreadListProps, Interactables, type LanguageModelConfig, type LanguageModelV1CallSettings, type LocalRuntimeOptions, type LocalRuntimeOptionsBase, type McpAppDisplayMode, type McpAppHostContext, type McpAppHostInfo, type McpAppMetadata, McpAppRenderer, type McpAppRendererOptions, type McpAppResource, type McpAppResourceCSP, type McpAppResourceMeta, type McpAppResourceOutput, type McpAppSandboxConfig, type McpAppToolCallParams, type McpAppsHost, McpAppsRemoteHost, type McpAppsRemoteHostOptions, type McpToolkitDefinition, type McpToolkitEntry, type McpToolkitToolConfig, MessageAttachmentByIndexProvider, MessageByIndexProvider, type MessageFormatAdapter, type MessageFormatItem, type MessageFormatRepository, MessageNotSentError, messagePart_d_exports as MessagePartPrimitive, type MessagePartRuntime, type MessagePartState, type MessagePartStatus, type MessagePartStreamStatus, message_d_exports as MessagePrimitive, MessageProvider, type MessageQueueController, type MessageQueueDriver, type MessageRuntime, type MessageState, type MessageStatus, type MessageStorageEntry, type MessageTiming, type ModelContext, ModelContextClient, type ModelContextProvider, ModelContextRegistry, type ModelContextRegistryInstructionHandle, type ModelContextRegistryProviderHandle, type ModelContextRegistryToolHandle, PartByIndexProvider, type PartProviderMetadata, type PartState, type PendingAttachment, type ProviderToolConfig, type QueueItemMethods, queueItem_d_exports as QueueItemPrimitive, type QueueItemState, type QuoteInfo, type QuoteMessagePartComponent, type QuoteMessagePartProps, ReadonlyThreadProvider, type RealtimeVoiceAdapter, type ReasoningGroupComponent, type ReasoningGroupProps, type ReasoningMessagePart, type ReasoningMessagePartComponent, type ReasoningMessagePartProps, type RemoteThreadListAdapter, type RemoteThreadListProviderComponent, type RespondToToolApprovalOptions, RuntimeAdapterProvider, type RuntimeAdapters, selectionToolbar_d_exports as SelectionToolbarPrimitive, type SendCommandsRequestBody, type SerializedModelContext, type SerializedTool, SimpleImageAttachmentAdapter, SimpleTextAttachmentAdapter, SingleThreadList, type SmoothOptions, type SourceMessagePart, type SourceMessagePartComponent, type SourceMessagePartProps, type SourceProviderMetadata, type SpeechSynthesisAdapter, type SubmitFeedbackOptions, type SuggestionAdapter, SuggestionByIndexProvider, type SuggestionConfig, suggestion_d_exports as SuggestionPrimitive, Suggestions, type TextMessagePart, type TextMessagePartComponent, type TextMessagePartProps, TextMessagePartProvider, type ThreadAssistantMessage, type ThreadAssistantMessagePart, type ThreadComposerRuntime, type ThreadComposerState, type ThreadHistoryAdapter, ThreadListItemByIndexProvider, threadListItemMore_d_exports as ThreadListItemMorePrimitive, threadListItem_d_exports as ThreadListItemPrimitive, type ThreadListItemRuntime, ThreadListItemRuntimeProvider, type ThreadListItemState, type ThreadListItemStatus, threadList_d_exports as ThreadListPrimitive, type ThreadListRuntime, type ThreadListState, type ThreadMessage, type ThreadMessageLike, thread_d_exports as ThreadPrimitive, type ThreadRuntime, type ThreadState, type ThreadSuggestion, type ThreadSystemMessage, type ThreadUserMessage, type ThreadUserMessagePart, type ThreadViewportState, type Tool, type ToolApprovalOption, type ToolApprovalOptionKind, type ToolApprovalResponse, type ToolArgsStatus, type ToolCallMessagePart, type ToolCallMessagePartComponent, type ToolCallMessagePartMcpMetadata, type ToolCallMessagePartProps, type ToolCallMessagePartStatus, type ToolCallText, type ToolCallTiming, type ToolDefinition, type ToolExecutionStatus, type ToolModelContentPart, type Toolkit, type ToolkitDefinition, type ToolkitDefinitionEntry, Tools, type Unstable_AudioMessagePart, type Unstable_AudioMessagePartComponent, type Unstable_AudioMessagePartProps, type Unstable_ComposerInput, type Unstable_ComposerInputHistory, type Unstable_DirectiveFormatter, type Unstable_DirectiveSegment, type Unstable_IconComponent, type Unstable_InferInteractableState, type Unstable_InteractableConfig, type Unstable_InteractableDefinition, type Unstable_InteractablePersistedState, type Unstable_InteractablePersistenceAdapter, type Unstable_InteractablePersistenceStatus, type Unstable_InteractableRegistration, type Unstable_InteractableSnapshotEntry, type Unstable_InteractableStateSchema, type Unstable_InteractableToolConfig, type Unstable_InteractableToolRenderProps, type Unstable_InteractableVersion, type Unstable_InteractableVersionInfo, type Unstable_InteractablesClientSchema, type Unstable_InteractablesConfig, type Unstable_InteractablesMethods, type Unstable_InteractablesState, type Unstable_Mention, type Unstable_MentionCategory, type Unstable_MentionDirective, type Unstable_MessageStallDetection, type Unstable_MessageStallDetectionOptions, type Unstable_ModelContextToolsOptions, type RegisteredTrigger as Unstable_RegisteredTrigger, type Unstable_SlashCommand, type Unstable_SlashCommandAction, type TriggerBehavior as Unstable_TriggerBehavior, type Unstable_TriggerItem, type Unstable_TriggerPopoverAriaProps, type Unstable_UseComposerInputOptions, type Unstable_UseLiveCompletionAdapterOptions, type Unstable_UseMentionAdapterOptions, type Unstable_UseSlashCommandAdapterOptions, type Unsubscribe, type VoiceSessionControls, type VoiceSessionHelpers, type VoiceSessionState, WebSpeechDictationAdapter, WebSpeechSynthesisAdapter, bindExternalStoreMessage, createMessageQueue, createVoiceSession, defineMcpToolkit, defineToolkit, externalTool, fromThreadMessageLike, generateId, getExternalStoreMessages, getMcpAppFromToolPart, groupPartByType, hitl, hitlTool, humanTool, isMessageNotSentError, makeAssistantDataUI, makeAssistantTool, makeAssistantToolUI, makeAssistantVisible, mergeModelContexts, pickExternalStoreSharedOptions, providerTool, stubTool, tool, unstable_Interactables, convertExternalMessages as unstable_convertExternalMessages, createMessageConverter as unstable_createMessageConverter, unstable_defaultDirectiveFormatter, unstable_formatInteractableSnapshot, unstable_getInteractableSnapshots, unstable_getInteractableVersions, unstable_interactableTool, unstable_useComposerInput, unstable_useComposerInputHistory, unstable_useInteractable, unstable_useInteractableState, unstable_useInteractableVersions, unstable_useLiveCompletionAdapter, unstable_useMentionAdapter, unstable_useMessageStallDetection, unstable_useSlashCommandAdapter, unstable_useThreadMessageIds, unstable_useTriggerPopoverAriaProps, useTriggerPopoverRootContext as unstable_useTriggerPopoverRootContext, useTriggerPopoverRootContextOptional as unstable_useTriggerPopoverRootContextOptional, useTriggerPopoverScopeContext as unstable_useTriggerPopoverScopeContext, useTriggerPopoverScopeContextOptional as unstable_useTriggerPopoverScopeContextOptional, useTriggerPopoverTriggers as unstable_useTriggerPopoverTriggers, useTriggerPopoverTriggersOptional as unstable_useTriggerPopoverTriggersOptional, useAssistantContext, useAssistantDataUI, useAssistantFrameHost, useAssistantInstructions, useAssistantInteractable, useAssistantTool, useAssistantToolUI, useAssistantTransportRuntime, useAssistantTransportSendCommand, useAssistantTransportState, useAui, useAuiEvent, useAuiState, useAuiToolOverrides, useCloudThreadListAdapter, useCloudThreadListRuntime, useExternalMessageConverter, useExternalStoreRuntime, useExternalStoreSharedOptions, useInlineRender, useInteractableState, useLocalRuntime, useMessagePartData, useMessagePartFile, useMessagePartImage, useMessagePartReasoning, useMessagePartSource, useMessagePartText, useMessageQuote, useMessageTiming, useRemoteThreadListRuntime, useRuntimeAdapters, useScrollLock, useSmooth, useThreadViewport, useThreadViewportAutoScroll, useThreadViewportStore, useToolArgsStatus, useToolCallElapsed, useVoiceControls, useVoiceState, useVoiceVolume };

package/dist/index.js

@@ -66,6 +66,6 @@ import { McpAppsRemoteHost } from "./mcp-apps/McpAppsRemoteHost.js";
 import { AuiConfig, AuiIf, AuiProvider, useAui, useAuiEvent, useAuiState } from "@assistant-ui/store";
 import { DataRenderers, GenerativeUIRender, GenerativeUIRenderError, Interactables, RuntimeAdapterProvider, Tools, defineMcpToolkit, defineToolkit, externalTool, groupPartByType, hitl, hitlTool, humanTool, makeAssistantDataUI, makeAssistantTool, makeAssistantToolUI, providerTool, stubTool, unstable_Interactables, unstable_interactableTool, unstable_useInteractable, unstable_useInteractableState, unstable_useInteractableVersions, unstable_useThreadMessageIds, useAssistantContext, useAssistantDataUI, useAssistantInstructions, useAssistantInteractable, useAssistantTool, useAssistantToolUI, useAuiToolOverrides, useExternalStoreSharedOptions, useInlineRender, useInteractableState, useRuntimeAdapters, useToolArgsStatus, useVoiceControls, useVoiceState, useVoiceVolume } from "@assistant-ui/core/react";
 import { AssistantCloud } from "assistant-cloud";
-import { AssistantFrameHost, AssistantFrameProvider, CompositeAttachmentAdapter, ExportedMessageRepository, FRAME_MESSAGE_CHANNEL, InMemoryThreadListAdapter, ModelContextRegistry, SimpleImageAttachmentAdapter, SimpleTextAttachmentAdapter, WebSpeechDictationAdapter, WebSpeechSynthesisAdapter, bindExternalStoreMessage, createMessageQueue, createVoiceSession, fromThreadMessageLike, generateId, getExternalStoreMessages, mergeModelContexts, pickExternalStoreSharedOptions, tool, unstable_defaultDirectiveFormatter, unstable_formatInteractableSnapshot, unstable_getInteractableSnapshots, unstable_getInteractableVersions } from "@assistant-ui/core";
+import { AssistantFrameHost, AssistantFrameProvider, CompositeAttachmentAdapter, ExportedMessageRepository, FRAME_MESSAGE_CHANNEL, InMemoryThreadListAdapter, MessageNotSentError, ModelContextRegistry, SimpleImageAttachmentAdapter, SimpleTextAttachmentAdapter, WebSpeechDictationAdapter, WebSpeechSynthesisAdapter, bindExternalStoreMessage, createMessageQueue, createVoiceSession, fromThreadMessageLike, generateId, getExternalStoreMessages, isMessageNotSentError, mergeModelContexts, pickExternalStoreSharedOptions, tool, unstable_defaultDirectiveFormatter, unstable_formatInteractableSnapshot, unstable_getInteractableSnapshots, unstable_getInteractableVersions } from "@assistant-ui/core";
 import { ChainOfThoughtClient, ExternalThread, ModelContext as ModelContextClient, SingleThreadList, Suggestions } from "@assistant-ui/core/store";
-export { actionBarMore_exports as ActionBarMorePrimitive, actionBar_exports as ActionBarPrimitive, AssistantCloud, AssistantFrameHost, AssistantFrameProvider, assistantModal_exports as AssistantModalPrimitive, AssistantRuntimeProvider, attachment_exports as AttachmentPrimitive, AuiConfig, AuiIf, AuiProvider, branchPicker_exports as BranchPickerPrimitive, ChainOfThoughtByIndicesProvider, ChainOfThoughtClient, chainOfThought_exports as ChainOfThoughtPrimitive, CloudFileAttachmentAdapter, ComposerAttachmentByIndexProvider, composer_exports as ComposerPrimitive, CompositeAttachmentAdapter, DataRenderers, DevToolsHooks, DevToolsProviderApi, error_exports as ErrorPrimitive, ExportedMessageRepository, ExternalThread, FRAME_MESSAGE_CHANNEL, GenerativeUIRender, GenerativeUIRenderError, internal_exports as INTERNAL, InMemoryThreadList, InMemoryThreadListAdapter, Interactables, McpAppRenderer, McpAppsRemoteHost, MessageAttachmentByIndexProvider, MessageByIndexProvider, messagePart_exports as MessagePartPrimitive, message_exports as MessagePrimitive, MessageProvider, ModelContextClient, ModelContextRegistry, PartByIndexProvider, queueItem_exports as QueueItemPrimitive, ReadonlyThreadProvider, RuntimeAdapterProvider, selectionToolbar_exports as SelectionToolbarPrimitive, SimpleImageAttachmentAdapter, SimpleTextAttachmentAdapter, SingleThreadList, SuggestionByIndexProvider, suggestion_exports as SuggestionPrimitive, Suggestions, TextMessagePartProvider, ThreadListItemByIndexProvider, threadListItemMore_exports as ThreadListItemMorePrimitive, threadListItem_exports as ThreadListItemPrimitive, ThreadListItemRuntimeProvider, threadList_exports as ThreadListPrimitive, thread_exports as ThreadPrimitive, Tools, WebSpeechDictationAdapter, WebSpeechSynthesisAdapter, bindExternalStoreMessage, createMessageQueue, createVoiceSession, defineMcpToolkit, defineToolkit, externalTool, fromThreadMessageLike, generateId, getExternalStoreMessages, getMcpAppFromToolPart, groupPartByType, hitl, hitlTool, humanTool, makeAssistantDataUI, makeAssistantTool, makeAssistantToolUI, makeAssistantVisible, mergeModelContexts, pickExternalStoreSharedOptions, providerTool, stubTool, tool, unstable_Interactables, convertExternalMessages as unstable_convertExternalMessages, createMessageConverter as unstable_createMessageConverter, unstable_defaultDirectiveFormatter, unstable_formatInteractableSnapshot, unstable_getInteractableSnapshots, unstable_getInteractableVersions, unstable_interactableTool, unstable_useComposerInput, unstable_useComposerInputHistory, unstable_useInteractable, unstable_useInteractableState, unstable_useInteractableVersions, unstable_useLiveCompletionAdapter, unstable_useMentionAdapter, unstable_useMessageStallDetection, unstable_useSlashCommandAdapter, unstable_useThreadMessageIds, unstable_useTriggerPopoverAriaProps, useTriggerPopoverRootContext as unstable_useTriggerPopoverRootContext, useTriggerPopoverRootContextOptional as unstable_useTriggerPopoverRootContextOptional, useTriggerPopoverScopeContext as unstable_useTriggerPopoverScopeContext, useTriggerPopoverScopeContextOptional as unstable_useTriggerPopoverScopeContextOptional, useTriggerPopoverTriggers as unstable_useTriggerPopoverTriggers, useTriggerPopoverTriggersOptional as unstable_useTriggerPopoverTriggersOptional, useAssistantContext, useAssistantDataUI, useAssistantFrameHost, useAssistantInstructions, useAssistantInteractable, useAssistantTool, useAssistantToolUI, useAssistantTransportRuntime, useAssistantTransportSendCommand, useAssistantTransportState, useAui, useAuiEvent, useAuiState, useAuiToolOverrides, useCloudThreadListAdapter, useCloudThreadListRuntime, useExternalMessageConverter, useExternalStoreRuntime, useExternalStoreSharedOptions, useInlineRender, useInteractableState, useLocalRuntime, useMessagePartData, useMessagePartFile, useMessagePartImage, useMessagePartReasoning, useMessagePartSource, useMessagePartText, useMessageQuote, useMessageTiming, useRemoteThreadListRuntime, useRuntimeAdapters, useScrollLock, useSmooth, useThreadViewport, useThreadViewportAutoScroll, useThreadViewportStore, useToolArgsStatus, useToolCallElapsed, useVoiceControls, useVoiceState, useVoiceVolume };
+export { actionBarMore_exports as ActionBarMorePrimitive, actionBar_exports as ActionBarPrimitive, AssistantCloud, AssistantFrameHost, AssistantFrameProvider, assistantModal_exports as AssistantModalPrimitive, AssistantRuntimeProvider, attachment_exports as AttachmentPrimitive, AuiConfig, AuiIf, AuiProvider, branchPicker_exports as BranchPickerPrimitive, ChainOfThoughtByIndicesProvider, ChainOfThoughtClient, chainOfThought_exports as ChainOfThoughtPrimitive, CloudFileAttachmentAdapter, ComposerAttachmentByIndexProvider, composer_exports as ComposerPrimitive, CompositeAttachmentAdapter, DataRenderers, DevToolsHooks, DevToolsProviderApi, error_exports as ErrorPrimitive, ExportedMessageRepository, ExternalThread, FRAME_MESSAGE_CHANNEL, GenerativeUIRender, GenerativeUIRenderError, internal_exports as INTERNAL, InMemoryThreadList, InMemoryThreadListAdapter, Interactables, McpAppRenderer, McpAppsRemoteHost, MessageAttachmentByIndexProvider, MessageByIndexProvider, MessageNotSentError, messagePart_exports as MessagePartPrimitive, message_exports as MessagePrimitive, MessageProvider, ModelContextClient, ModelContextRegistry, PartByIndexProvider, queueItem_exports as QueueItemPrimitive, ReadonlyThreadProvider, RuntimeAdapterProvider, selectionToolbar_exports as SelectionToolbarPrimitive, SimpleImageAttachmentAdapter, SimpleTextAttachmentAdapter, SingleThreadList, SuggestionByIndexProvider, suggestion_exports as SuggestionPrimitive, Suggestions, TextMessagePartProvider, ThreadListItemByIndexProvider, threadListItemMore_exports as ThreadListItemMorePrimitive, threadListItem_exports as ThreadListItemPrimitive, ThreadListItemRuntimeProvider, threadList_exports as ThreadListPrimitive, thread_exports as ThreadPrimitive, Tools, WebSpeechDictationAdapter, WebSpeechSynthesisAdapter, bindExternalStoreMessage, createMessageQueue, createVoiceSession, defineMcpToolkit, defineToolkit, externalTool, fromThreadMessageLike, generateId, getExternalStoreMessages, getMcpAppFromToolPart, groupPartByType, hitl, hitlTool, humanTool, isMessageNotSentError, makeAssistantDataUI, makeAssistantTool, makeAssistantToolUI, makeAssistantVisible, mergeModelContexts, pickExternalStoreSharedOptions, providerTool, stubTool, tool, unstable_Interactables, convertExternalMessages as unstable_convertExternalMessages, createMessageConverter as unstable_createMessageConverter, unstable_defaultDirectiveFormatter, unstable_formatInteractableSnapshot, unstable_getInteractableSnapshots, unstable_getInteractableVersions, unstable_interactableTool, unstable_useComposerInput, unstable_useComposerInputHistory, unstable_useInteractable, unstable_useInteractableState, unstable_useInteractableVersions, unstable_useLiveCompletionAdapter, unstable_useMentionAdapter, unstable_useMessageStallDetection, unstable_useSlashCommandAdapter, unstable_useThreadMessageIds, unstable_useTriggerPopoverAriaProps, useTriggerPopoverRootContext as unstable_useTriggerPopoverRootContext, useTriggerPopoverRootContextOptional as unstable_useTriggerPopoverRootContextOptional, useTriggerPopoverScopeContext as unstable_useTriggerPopoverScopeContext, useTriggerPopoverScopeContextOptional as unstable_useTriggerPopoverScopeContextOptional, useTriggerPopoverTriggers as unstable_useTriggerPopoverTriggers, useTriggerPopoverTriggersOptional as unstable_useTriggerPopoverTriggersOptional, useAssistantContext, useAssistantDataUI, useAssistantFrameHost, useAssistantInstructions, useAssistantInteractable, useAssistantTool, useAssistantToolUI, useAssistantTransportRuntime, useAssistantTransportSendCommand, useAssistantTransportState, useAui, useAuiEvent, useAuiState, useAuiToolOverrides, useCloudThreadListAdapter, useCloudThreadListRuntime, useExternalMessageConverter, useExternalStoreRuntime, useExternalStoreSharedOptions, useInlineRender, useInteractableState, useLocalRuntime, useMessagePartData, useMessagePartFile, useMessagePartImage, useMessagePartReasoning, useMessagePartSource, useMessagePartText, useMessageQuote, useMessageTiming, useRemoteThreadListRuntime, useRuntimeAdapters, useScrollLock, useSmooth, useThreadViewport, useThreadViewportAutoScroll, useThreadViewportStore, useToolArgsStatus, useToolCallElapsed, useVoiceControls, useVoiceState, useVoiceVolume };

package/dist/legacy-runtime/runtime-cores/assistant-transport/replayBoundaryStream.js

@@ -92,6 +92,24 @@ const createReplayBoundaryStream = async (response, { setReplaying, waitForRende
 	const reader = body.getReader();
 	let bytesForwarded = 0;
 	let replayFinished = false;
+	let readerCleanup;
+	const releaseReader = () => {
+		if (readerCleanup) return readerCleanup;
+		reader.releaseLock();
+		readerCleanup = Promise.resolve();
+		return readerCleanup;
+	};
+	const cancelReader = (reason) => {
+		if (readerCleanup) return readerCleanup;
+		readerCleanup = (async () => {
+			try {
+				await reader.cancel(reason);
+			} finally {
+				reader.releaseLock();
+			}
+		})();
+		return readerCleanup;
+	};
 	const finishReplay = async () => {
 		if (replayFinished) return;
 		replayFinished = true;
@@ -101,37 +119,43 @@ const createReplayBoundaryStream = async (response, { setReplaying, waitForRende
 	};
 	return new ReadableStream({
 		async pull(controller) {
-			const { done, value } = await reader.read();
-			if (done) {
-				await finishReplay();
-				controller.close();
-				return;
-			}
-			if (replayFinished) {
-				controller.enqueue(value);
-				return;
-			}
-			const nextBytesForwarded = bytesForwarded + value.byteLength;
-			if (nextBytesForwarded < replayContentLength) {
-				bytesForwarded = nextBytesForwarded;
-				controller.enqueue(value);
-				return;
-			}
-			if (nextBytesForwarded === replayContentLength) {
-				controller.enqueue(value);
+			try {
+				const { done, value } = await reader.read();
+				if (done) {
+					await releaseReader();
+					await finishReplay();
+					controller.close();
+					return;
+				}
+				if (replayFinished) {
+					controller.enqueue(value);
+					return;
+				}
+				const nextBytesForwarded = bytesForwarded + value.byteLength;
+				if (nextBytesForwarded < replayContentLength) {
+					bytesForwarded = nextBytesForwarded;
+					controller.enqueue(value);
+					return;
+				}
+				if (nextBytesForwarded === replayContentLength) {
+					controller.enqueue(value);
+					await finishReplay();
+					return;
+				}
+				const replayBytesInChunk = replayContentLength - bytesForwarded;
+				controller.enqueue(value.subarray(0, replayBytesInChunk));
 				await finishReplay();
-				return;
+				controller.enqueue(value.subarray(replayBytesInChunk));
+			} catch (error) {
+				await cancelReader(error).catch(() => {});
+				throw error;
 			}
-			const replayBytesInChunk = replayContentLength - bytesForwarded;
-			controller.enqueue(value.subarray(0, replayBytesInChunk));
-			await finishReplay();
-			controller.enqueue(value.subarray(replayBytesInChunk));
 		},
 		async cancel(reason) {
 			const wasFinished = replayFinished;
 			replayFinished = true;
 			if (!wasFinished) setReplaying(false);
-			await reader.cancel(reason);
+			await cancelReader(reason);
 		}
 	});
 };

package/dist/legacy-runtime/runtime-cores/assistant-transport/replayBoundaryStream.js.map

@@ -1 +1 @@
-{"version":3,"file":"replayBoundaryStream.js","names":["c","_c","useCallback","useEffect","useRef","useState","REPLAY_CONTENT_LENGTH_HEADER","ReplayBoundaryStreamOptions","setReplaying","value","waitForRender","Promise","useReplayRenderWait","$","renderTicket","setRenderTicket","mountedRef","nextTicketRef","t0","Symbol","for","waitersRef","t1","committedTicket","pendingWaiters","waiter","current","undefined","ticket","resolve","push","resolveWaiters","t2","t3","t4","t5","t6","setTimeout","parseReplayContentLength","headers","Headers","raw","get","boundary","Number","isSafeInteger","createReplayBoundaryStream","response","Response","waitForReplayRender","ReadableStream","Uint8Array","body","replayContentLength","reader","getReader","bytesForwarded","replayFinished","finishReplay","pull","controller","done","read","close","enqueue","nextBytesForwarded","byteLength","replayBytesInChunk","subarray","cancel","reason","wasFinished"],"sources":["../../../../src/legacy-runtime/runtime-cores/assistant-transport/replayBoundaryStream.ts"],"sourcesContent":["\"use client\";\n\nimport { useCallback, useEffect, useRef, useState } from \"react\";\n\nexport const REPLAY_CONTENT_LENGTH_HEADER = \"Aui-Replay-Content-Length\";\n\ntype ReplayBoundaryStreamOptions = {\n  setReplaying: (value: boolean) => void;\n  waitForRender: () => Promise<void>;\n};\n\nexport const useReplayRenderWait = () => {\n  const [renderTicket, setRenderTicket] = useState(0);\n  const mountedRef = useRef(true);\n  const nextTicketRef = useRef(0);\n  const waitersRef = useRef<Array<{ ticket: number; resolve: () => void }>>([]);\n\n  const resolveWaiters = useCallback((committedTicket?: number) => {\n    const pendingWaiters = [];\n\n    for (const waiter of waitersRef.current) {\n      if (committedTicket === undefined || waiter.ticket <= committedTicket) {\n        waiter.resolve();\n      } else {\n        pendingWaiters.push(waiter);\n      }\n    }\n\n    waitersRef.current = pendingWaiters;\n  }, []);\n\n  useEffect(() => {\n    mountedRef.current = true;\n    resolveWaiters(renderTicket);\n  }, [renderTicket, resolveWaiters]);\n\n  useEffect(\n    () => () => {\n      mountedRef.current = false;\n      resolveWaiters();\n    },\n    [resolveWaiters],\n  );\n\n  return useCallback(\n    () =>\n      new Promise<void>((resolve) => {\n        setTimeout(() => {\n          if (!mountedRef.current) {\n            resolve();\n            return;\n          }\n\n          const ticket = nextTicketRef.current + 1;\n          nextTicketRef.current = ticket;\n          waitersRef.current.push({ ticket, resolve });\n          setRenderTicket(ticket);\n        }, 0);\n      }),\n    [],\n  );\n};\n\nconst parseReplayContentLength = (headers: Headers): number => {\n  const raw = headers.get(REPLAY_CONTENT_LENGTH_HEADER);\n  if (raw == null) return 0;\n\n  const boundary = Number(raw);\n  return Number.isSafeInteger(boundary) && boundary > 0 ? boundary : 0;\n};\n\n// Gates replay bytes until isReplaying:true renders, then releases live bytes after isReplaying:false renders.\nexport const createReplayBoundaryStream = async (\n  response: Response,\n  {\n    setReplaying,\n    waitForRender: waitForReplayRender,\n  }: ReplayBoundaryStreamOptions,\n): Promise<ReadableStream<Uint8Array>> => {\n  const body = response.body as ReadableStream<Uint8Array>;\n  const replayContentLength = parseReplayContentLength(response.headers);\n\n  if (replayContentLength <= 0) {\n    return body;\n  }\n\n  setReplaying(true);\n  await waitForReplayRender();\n\n  const reader = body.getReader();\n  let bytesForwarded = 0;\n  let replayFinished = false;\n\n  const finishReplay = async () => {\n    if (replayFinished) return;\n    replayFinished = true;\n\n    // Let replay bytes drain before rendering live mode, then render live mode before releasing live bytes.\n    await waitForReplayRender();\n    setReplaying(false);\n    await waitForReplayRender();\n  };\n\n  return new ReadableStream<Uint8Array>({\n    async pull(controller) {\n      const { done, value } = await reader.read();\n\n      if (done) {\n        await finishReplay();\n        controller.close();\n        return;\n      }\n\n      if (replayFinished) {\n        controller.enqueue(value);\n        return;\n      }\n\n      const nextBytesForwarded = bytesForwarded + value.byteLength;\n\n      if (nextBytesForwarded < replayContentLength) {\n        bytesForwarded = nextBytesForwarded;\n        controller.enqueue(value);\n        return;\n      }\n\n      if (nextBytesForwarded === replayContentLength) {\n        controller.enqueue(value);\n        await finishReplay();\n        return;\n      }\n\n      const replayBytesInChunk = replayContentLength - bytesForwarded;\n\n      controller.enqueue(value.subarray(0, replayBytesInChunk));\n      await finishReplay();\n      controller.enqueue(value.subarray(replayBytesInChunk));\n    },\n    async cancel(reason) {\n      const wasFinished = replayFinished;\n      replayFinished = true;\n      if (!wasFinished) setReplaying(false);\n      await reader.cancel(reason);\n    },\n  });\n};\n"],"mappings":";;;;AAIA,MAAaM,+BAA+B;AAO5C,MAAaM,4BAAsB;CAAA,MAAAC,IAAAZ,EAAA,CAAA;CACjC,MAAA,CAAAa,cAAAC,mBAAwCV,SAAS,CAAC;CAClD,MAAAW,aAAmBZ,OAAO,IAAI;CAC9B,MAAAa,gBAAsBb,OAAO,CAAC;CAAE,IAAAc;CAAA,IAAAL,EAAA,OAAAM,OAAAC,IAAA,2BAAA,GAAA;EAC0CF,KAAA,CAAA;EAAEL,EAAA,KAAAK;CAAA,OAAAA,KAAAL,EAAA;CAA5E,MAAAQ,aAAmBjB,OAAuDc,EAAE;CAAE,IAAAI;CAAA,IAAAT,EAAA,OAAAM,OAAAC,IAAA,2BAAA,GAAA;EAE3CE,MAAAC,oBAAA;GACjC,MAAAC,iBAAuB,CAAA;GAEvB,KAAK,MAAAC,UAAgBJ,WAAUK,SAC7B,IAAIH,oBAAoBI,KAAAA,KAAaF,OAAMG,UAAWL,iBACpDE,OAAMI,QAAS;QAEfL,eAAcM,KAAML,MAAM;GAI9BJ,WAAUK,UAAWF;EAAH;EACnBX,EAAA,KAAAS;CAAA,OAAAA,KAAAT,EAAA;CAZD,MAAAkB,iBAAuBT;CAYhB,IAAAU;CAAA,IAAAC;CAAA,IAAApB,EAAA,OAAAC,cAAA;EAEGkB,WAAA;GACRhB,WAAUU,UAAW;GACrBK,eAAejB,YAAY;EAAC;EAC3BmB,KAAA,CAACnB,cAAciB,cAAc;EAAClB,EAAA,KAAAC;EAAAD,EAAA,KAAAmB;EAAAnB,EAAA,KAAAoB;CAAA,OAAA;EAAAD,KAAAnB,EAAA;EAAAoB,KAAApB,EAAA;CAAA;CAHjCV,UAAU6B,IAGPC,EAA8B;CAAC,IAAAC;CAAA,IAAAC;CAAA,IAAAtB,EAAA,OAAAM,OAAAC,IAAA,2BAAA,GAAA;EAGhCc,iBAAM;GACJlB,WAAUU,UAAW;GACrBK,eAAe;EAAC;EAElBI,KAAA,CAACJ,cAAc;EAAClB,EAAA,KAAAqB;EAAArB,EAAA,KAAAsB;CAAA,OAAA;EAAAD,KAAArB,EAAA;EAAAsB,KAAAtB,EAAA;CAAA;CALlBV,UACE+B,IAIAC,EACF;CAAC,IAAAC;CAAA,IAAAvB,EAAA,OAAAM,OAAAC,IAAA,2BAAA,GAAA;EAGCgB,WACE,IAAIzB,SAAckB,YAAA;GAChBQ,iBAAW;IACT,IAAI,CAACrB,WAAUU,SAAQ;KACrBG,QAAQ;KAAC;IAAA;IAIX,MAAAD,SAAeX,cAAaS,UAAW;IACvCT,cAAaS,UAAWE;IACxBP,WAAUK,QAAQI,KAAM;KAAAF;KAAAC;IAAkB,CAAC;IAC3Cd,gBAAgBa,MAAM;GAAC,GACtB,CAAC;EAAC,CACN;EAACf,EAAA,KAAAuB;CAAA,OAAAA,KAAAvB,EAAA;CAAA,OAdCuB;AAgBN;AAGH,MAAME,4BAA4BC,YAA6B;CAC7D,MAAME,MAAMF,QAAQG,IAAIpC,4BAA4B;CACpD,IAAImC,OAAO,MAAM,OAAO;CAExB,MAAME,WAAWC,OAAOH,GAAG;CAC3B,OAAOG,OAAOC,cAAcF,QAAQ,KAAKA,WAAW,IAAIA,WAAW;AACrE;AAGA,MAAaG,6BAA6B,OACxCC,UACA,EACEvC,cACAE,eAAeuC,0BAEuB;CACxC,MAAMG,OAAOL,SAASK;CACtB,MAAMC,sBAAsBf,yBAAyBS,SAASR,OAAO;CAErE,IAAIc,uBAAuB,GACzB,OAAOD;CAGT5C,aAAa,IAAI;CACjB,MAAMyC,oBAAoB;CAE1B,MAAMK,SAASF,KAAKG,UAAU;CAC9B,IAAIC,iBAAiB;CACrB,IAAIC,iBAAiB;CAErB,MAAMC,eAAe,YAAY;EAC/B,IAAID,gBAAgB;EACpBA,iBAAiB;EAGjB,MAAMR,oBAAoB;EAC1BzC,aAAa,KAAK;EAClB,MAAMyC,oBAAoB;CAC5B;CAEA,OAAO,IAAIC,eAA2B;EACpC,MAAMS,KAAKC,YAAY;GACrB,MAAM,EAAEC,MAAMpD,UAAU,MAAM6C,OAAOQ,KAAK;GAE1C,IAAID,MAAM;IACR,MAAMH,aAAa;IACnBE,WAAWG,MAAM;IACjB;GACF;GAEA,IAAIN,gBAAgB;IAClBG,WAAWI,QAAQvD,KAAK;IACxB;GACF;GAEA,MAAMwD,qBAAqBT,iBAAiB/C,MAAMyD;GAElD,IAAID,qBAAqBZ,qBAAqB;IAC5CG,iBAAiBS;IACjBL,WAAWI,QAAQvD,KAAK;IACxB;GACF;GAEA,IAAIwD,uBAAuBZ,qBAAqB;IAC9CO,WAAWI,QAAQvD,KAAK;IACxB,MAAMiD,aAAa;IACnB;GACF;GAEA,MAAMS,qBAAqBd,sBAAsBG;GAEjDI,WAAWI,QAAQvD,MAAM2D,SAAS,GAAGD,kBAAkB,CAAC;GACxD,MAAMT,aAAa;GACnBE,WAAWI,QAAQvD,MAAM2D,SAASD,kBAAkB,CAAC;EACvD;EACA,MAAME,OAAOC,QAAQ;GACnB,MAAMC,cAAcd;GACpBA,iBAAiB;GACjB,IAAI,CAACc,aAAa/D,aAAa,KAAK;GACpC,MAAM8C,OAAOe,OAAOC,MAAM;EAC5B;CACF,CAAC;AACH"}
+{"version":3,"file":"replayBoundaryStream.js","names":["c","_c","useCallback","useEffect","useRef","useState","REPLAY_CONTENT_LENGTH_HEADER","ReplayBoundaryStreamOptions","setReplaying","value","waitForRender","Promise","useReplayRenderWait","$","renderTicket","setRenderTicket","mountedRef","nextTicketRef","t0","Symbol","for","waitersRef","t1","committedTicket","pendingWaiters","waiter","current","undefined","ticket","resolve","push","resolveWaiters","t2","t3","t4","t5","t6","setTimeout","parseReplayContentLength","headers","Headers","raw","get","boundary","Number","isSafeInteger","createReplayBoundaryStream","response","Response","waitForReplayRender","ReadableStream","Uint8Array","body","replayContentLength","reader","getReader","bytesForwarded","replayFinished","readerCleanup","releaseReader","releaseLock","cancelReader","reason","cancel","finishReplay","pull","controller","done","read","close","enqueue","nextBytesForwarded","byteLength","replayBytesInChunk","subarray","error","catch","wasFinished"],"sources":["../../../../src/legacy-runtime/runtime-cores/assistant-transport/replayBoundaryStream.ts"],"sourcesContent":["\"use client\";\n\nimport { useCallback, useEffect, useRef, useState } from \"react\";\n\nexport const REPLAY_CONTENT_LENGTH_HEADER = \"Aui-Replay-Content-Length\";\n\ntype ReplayBoundaryStreamOptions = {\n  setReplaying: (value: boolean) => void;\n  waitForRender: () => Promise<void>;\n};\n\nexport const useReplayRenderWait = () => {\n  const [renderTicket, setRenderTicket] = useState(0);\n  const mountedRef = useRef(true);\n  const nextTicketRef = useRef(0);\n  const waitersRef = useRef<Array<{ ticket: number; resolve: () => void }>>([]);\n\n  const resolveWaiters = useCallback((committedTicket?: number) => {\n    const pendingWaiters = [];\n\n    for (const waiter of waitersRef.current) {\n      if (committedTicket === undefined || waiter.ticket <= committedTicket) {\n        waiter.resolve();\n      } else {\n        pendingWaiters.push(waiter);\n      }\n    }\n\n    waitersRef.current = pendingWaiters;\n  }, []);\n\n  useEffect(() => {\n    mountedRef.current = true;\n    resolveWaiters(renderTicket);\n  }, [renderTicket, resolveWaiters]);\n\n  useEffect(\n    () => () => {\n      mountedRef.current = false;\n      resolveWaiters();\n    },\n    [resolveWaiters],\n  );\n\n  return useCallback(\n    () =>\n      new Promise<void>((resolve) => {\n        setTimeout(() => {\n          if (!mountedRef.current) {\n            resolve();\n            return;\n          }\n\n          const ticket = nextTicketRef.current + 1;\n          nextTicketRef.current = ticket;\n          waitersRef.current.push({ ticket, resolve });\n          setRenderTicket(ticket);\n        }, 0);\n      }),\n    [],\n  );\n};\n\nconst parseReplayContentLength = (headers: Headers): number => {\n  const raw = headers.get(REPLAY_CONTENT_LENGTH_HEADER);\n  if (raw == null) return 0;\n\n  const boundary = Number(raw);\n  return Number.isSafeInteger(boundary) && boundary > 0 ? boundary : 0;\n};\n\n// Gates replay bytes until isReplaying:true renders, then releases live bytes after isReplaying:false renders.\nexport const createReplayBoundaryStream = async (\n  response: Response,\n  {\n    setReplaying,\n    waitForRender: waitForReplayRender,\n  }: ReplayBoundaryStreamOptions,\n): Promise<ReadableStream<Uint8Array>> => {\n  const body = response.body as ReadableStream<Uint8Array>;\n  const replayContentLength = parseReplayContentLength(response.headers);\n\n  if (replayContentLength <= 0) {\n    return body;\n  }\n\n  setReplaying(true);\n  await waitForReplayRender();\n\n  const reader = body.getReader();\n  let bytesForwarded = 0;\n  let replayFinished = false;\n  let readerCleanup: Promise<void> | undefined;\n\n  const releaseReader = () => {\n    if (readerCleanup) return readerCleanup;\n    reader.releaseLock();\n    readerCleanup = Promise.resolve();\n    return readerCleanup;\n  };\n\n  const cancelReader = (reason?: unknown) => {\n    if (readerCleanup) return readerCleanup;\n    readerCleanup = (async () => {\n      try {\n        await reader.cancel(reason);\n      } finally {\n        reader.releaseLock();\n      }\n    })();\n    return readerCleanup;\n  };\n\n  const finishReplay = async () => {\n    if (replayFinished) return;\n    replayFinished = true;\n\n    // Let replay bytes drain before rendering live mode, then render live mode before releasing live bytes.\n    await waitForReplayRender();\n    setReplaying(false);\n    await waitForReplayRender();\n  };\n\n  return new ReadableStream<Uint8Array>({\n    async pull(controller) {\n      try {\n        const { done, value } = await reader.read();\n\n        if (done) {\n          await releaseReader();\n          await finishReplay();\n          controller.close();\n          return;\n        }\n\n        if (replayFinished) {\n          controller.enqueue(value);\n          return;\n        }\n\n        const nextBytesForwarded = bytesForwarded + value.byteLength;\n\n        if (nextBytesForwarded < replayContentLength) {\n          bytesForwarded = nextBytesForwarded;\n          controller.enqueue(value);\n          return;\n        }\n\n        if (nextBytesForwarded === replayContentLength) {\n          controller.enqueue(value);\n          await finishReplay();\n          return;\n        }\n\n        const replayBytesInChunk = replayContentLength - bytesForwarded;\n\n        controller.enqueue(value.subarray(0, replayBytesInChunk));\n        await finishReplay();\n        controller.enqueue(value.subarray(replayBytesInChunk));\n      } catch (error) {\n        await cancelReader(error).catch(() => {});\n        throw error;\n      }\n    },\n    async cancel(reason) {\n      const wasFinished = replayFinished;\n      replayFinished = true;\n      if (!wasFinished) setReplaying(false);\n      await cancelReader(reason);\n    },\n  });\n};\n"],"mappings":";;;;AAIA,MAAaM,+BAA+B;AAO5C,MAAaM,4BAAsB;CAAA,MAAAC,IAAAZ,EAAA,CAAA;CACjC,MAAA,CAAAa,cAAAC,mBAAwCV,SAAS,CAAC;CAClD,MAAAW,aAAmBZ,OAAO,IAAI;CAC9B,MAAAa,gBAAsBb,OAAO,CAAC;CAAE,IAAAc;CAAA,IAAAL,EAAA,OAAAM,OAAAC,IAAA,2BAAA,GAAA;EAC0CF,KAAA,CAAA;EAAEL,EAAA,KAAAK;CAAA,OAAAA,KAAAL,EAAA;CAA5E,MAAAQ,aAAmBjB,OAAuDc,EAAE;CAAE,IAAAI;CAAA,IAAAT,EAAA,OAAAM,OAAAC,IAAA,2BAAA,GAAA;EAE3CE,MAAAC,oBAAA;GACjC,MAAAC,iBAAuB,CAAA;GAEvB,KAAK,MAAAC,UAAgBJ,WAAUK,SAC7B,IAAIH,oBAAoBI,KAAAA,KAAaF,OAAMG,UAAWL,iBACpDE,OAAMI,QAAS;QAEfL,eAAcM,KAAML,MAAM;GAI9BJ,WAAUK,UAAWF;EAAH;EACnBX,EAAA,KAAAS;CAAA,OAAAA,KAAAT,EAAA;CAZD,MAAAkB,iBAAuBT;CAYhB,IAAAU;CAAA,IAAAC;CAAA,IAAApB,EAAA,OAAAC,cAAA;EAEGkB,WAAA;GACRhB,WAAUU,UAAW;GACrBK,eAAejB,YAAY;EAAC;EAC3BmB,KAAA,CAACnB,cAAciB,cAAc;EAAClB,EAAA,KAAAC;EAAAD,EAAA,KAAAmB;EAAAnB,EAAA,KAAAoB;CAAA,OAAA;EAAAD,KAAAnB,EAAA;EAAAoB,KAAApB,EAAA;CAAA;CAHjCV,UAAU6B,IAGPC,EAA8B;CAAC,IAAAC;CAAA,IAAAC;CAAA,IAAAtB,EAAA,OAAAM,OAAAC,IAAA,2BAAA,GAAA;EAGhCc,iBAAM;GACJlB,WAAUU,UAAW;GACrBK,eAAe;EAAC;EAElBI,KAAA,CAACJ,cAAc;EAAClB,EAAA,KAAAqB;EAAArB,EAAA,KAAAsB;CAAA,OAAA;EAAAD,KAAArB,EAAA;EAAAsB,KAAAtB,EAAA;CAAA;CALlBV,UACE+B,IAIAC,EACF;CAAC,IAAAC;CAAA,IAAAvB,EAAA,OAAAM,OAAAC,IAAA,2BAAA,GAAA;EAGCgB,WACE,IAAIzB,SAAckB,YAAA;GAChBQ,iBAAW;IACT,IAAI,CAACrB,WAAUU,SAAQ;KACrBG,QAAQ;KAAC;IAAA;IAIX,MAAAD,SAAeX,cAAaS,UAAW;IACvCT,cAAaS,UAAWE;IACxBP,WAAUK,QAAQI,KAAM;KAAAF;KAAAC;IAAkB,CAAC;IAC3Cd,gBAAgBa,MAAM;GAAC,GACtB,CAAC;EAAC,CACN;EAACf,EAAA,KAAAuB;CAAA,OAAAA,KAAAvB,EAAA;CAAA,OAdCuB;AAgBN;AAGH,MAAME,4BAA4BC,YAA6B;CAC7D,MAAME,MAAMF,QAAQG,IAAIpC,4BAA4B;CACpD,IAAImC,OAAO,MAAM,OAAO;CAExB,MAAME,WAAWC,OAAOH,GAAG;CAC3B,OAAOG,OAAOC,cAAcF,QAAQ,KAAKA,WAAW,IAAIA,WAAW;AACrE;AAGA,MAAaG,6BAA6B,OACxCC,UACA,EACEvC,cACAE,eAAeuC,0BAEuB;CACxC,MAAMG,OAAOL,SAASK;CACtB,MAAMC,sBAAsBf,yBAAyBS,SAASR,OAAO;CAErE,IAAIc,uBAAuB,GACzB,OAAOD;CAGT5C,aAAa,IAAI;CACjB,MAAMyC,oBAAoB;CAE1B,MAAMK,SAASF,KAAKG,UAAU;CAC9B,IAAIC,iBAAiB;CACrB,IAAIC,iBAAiB;CACrB,IAAIC;CAEJ,MAAMC,sBAAsB;EAC1B,IAAID,eAAe,OAAOA;EAC1BJ,OAAOM,YAAY;EACnBF,gBAAgB/C,QAAQkB,QAAQ;EAChC,OAAO6B;CACT;CAEA,MAAMG,gBAAgBC,WAAqB;EACzC,IAAIJ,eAAe,OAAOA;EAC1BA,iBAAiB,YAAY;GAC3B,IAAI;IACF,MAAMJ,OAAOS,OAAOD,MAAM;GAC5B,UAAU;IACRR,OAAOM,YAAY;GACrB;EACF,EAAA,CAAG;EACH,OAAOF;CACT;CAEA,MAAMM,eAAe,YAAY;EAC/B,IAAIP,gBAAgB;EACpBA,iBAAiB;EAGjB,MAAMR,oBAAoB;EAC1BzC,aAAa,KAAK;EAClB,MAAMyC,oBAAoB;CAC5B;CAEA,OAAO,IAAIC,eAA2B;EACpC,MAAMe,KAAKC,YAAY;GACrB,IAAI;IACF,MAAM,EAAEC,MAAM1D,UAAU,MAAM6C,OAAOc,KAAK;IAE1C,IAAID,MAAM;KACR,MAAMR,cAAc;KACpB,MAAMK,aAAa;KACnBE,WAAWG,MAAM;KACjB;IACF;IAEA,IAAIZ,gBAAgB;KAClBS,WAAWI,QAAQ7D,KAAK;KACxB;IACF;IAEA,MAAM8D,qBAAqBf,iBAAiB/C,MAAM+D;IAElD,IAAID,qBAAqBlB,qBAAqB;KAC5CG,iBAAiBe;KACjBL,WAAWI,QAAQ7D,KAAK;KACxB;IACF;IAEA,IAAI8D,uBAAuBlB,qBAAqB;KAC9Ca,WAAWI,QAAQ7D,KAAK;KACxB,MAAMuD,aAAa;KACnB;IACF;IAEA,MAAMS,qBAAqBpB,sBAAsBG;IAEjDU,WAAWI,QAAQ7D,MAAMiE,SAAS,GAAGD,kBAAkB,CAAC;IACxD,MAAMT,aAAa;IACnBE,WAAWI,QAAQ7D,MAAMiE,SAASD,kBAAkB,CAAC;GACvD,SAASE,OAAO;IACd,MAAMd,aAAac,KAAK,CAAC,CAACC,YAAY,CAAC,CAAC;IACxC,MAAMD;GACR;EACF;EACA,MAAMZ,OAAOD,QAAQ;GACnB,MAAMe,cAAcpB;GACpBA,iBAAiB;GACjB,IAAI,CAACoB,aAAarE,aAAa,KAAK;GACpC,MAAMqD,aAAaC,MAAM;EAC3B;CACF,CAAC;AACH"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@assistant-ui/react",
-  "version": "0.15.13",
+  "version": "0.15.14",
   "description": "Open-source TypeScript/React library for building production-grade AI chat experiences",
   "keywords": [
     "ai",
@@ -55,9 +55,9 @@
   ],
   "sideEffects": false,
   "dependencies": {
-    "@assistant-ui/core": "^0.3.12",
-    "@assistant-ui/store": "^0.3.8",
-    "@assistant-ui/tap": "^0.9.11",
+    "@assistant-ui/core": "^0.3.13",
+    "@assistant-ui/store": "^0.3.9",
+    "@assistant-ui/tap": "^0.9.12",
     "@radix-ui/primitive": "^1.1.7",
     "@radix-ui/react-collection": "^1.1.15",
     "@radix-ui/react-compose-refs": "^1.1.5",
@@ -66,7 +66,7 @@
     "@radix-ui/react-use-callback-ref": "^1.1.4",
     "@radix-ui/react-use-controllable-state": "^1.2.6",
     "@radix-ui/react-use-escape-keydown": "^1.1.5",
-    "assistant-cloud": "^0.1.39",
+    "assistant-cloud": "^0.1.40",
     "assistant-stream": "^0.3.37",
     "radix-ui": "^1.6.7",
     "react-textarea-autosize": "^8.5.9",
@@ -98,8 +98,8 @@
     "react": "^19.2.8",
     "react-dom": "^19.2.8",
     "vitest": "^4.1.10",
-    "@assistant-ui/x-buildutils": "0.0.22",
-    "@assistant-ui/vite": "0.0.12"
+    "@assistant-ui/vite": "0.0.12",
+    "@assistant-ui/x-buildutils": "0.0.22"
   },
   "publishConfig": {
     "access": "public",

package/src/client/InMemoryThreadList.ts

@@ -7,8 +7,9 @@ import {
   attachTransformScopes,
   useClientResource,
 } from "@assistant-ui/store";
+import { generateId } from "@assistant-ui/core";
 
-import { ModelContext, Suggestions } from "@assistant-ui/core/store";
+import { ModelContext } from "@assistant-ui/core/store";
 import { Tools, DataRenderers } from "@assistant-ui/core/react";
 
 const RESOLVED_PROMISE = Promise.resolve();
@@ -137,7 +138,7 @@ const useInMemoryThreadList = (
   };
 
   const handleSwitchToNewThread = () => {
-    const newId = `thread-${Date.now()}`;
+    const newId = `thread-${generateId()}`;
     setThreads((prev) => [
       ...prev,
       { id: newId, title: "New Thread", status: "regular" },
@@ -190,7 +191,8 @@ const useInMemoryThreadList = (
     switchToNewThread: handleSwitchToNewThread,
     getLoadThreadsPromise: () => RESOLVED_PROMISE,
     reload: () => RESOLVED_PROMISE,
-    reloadMainThread: () => RESOLVED_PROMISE,
+    reloadMainThread: () =>
+      mainThreadClient.methods.unstable_refetchThread?.() ?? RESOLVED_PROMISE,
     loadMore: () => RESOLVED_PROMISE,
     item: (selector) => {
       if (selector === "main") {
@@ -236,6 +238,10 @@ attachTransformScopes(useInMemoryThreadList, (scopes, parent) => {
     scopes.dataRenderers = DataRenderers();
   }
   if (!scopes.suggestions && parent.suggestions.source === null) {
-    scopes.suggestions = Suggestions();
+    scopes.suggestions = Derived({
+      source: "thread",
+      query: {},
+      get: (aui) => aui.thread.suggestions(),
+    });
   }
 });

package/src/index.ts

@@ -139,6 +139,7 @@ export type {
   ExternalStoreThreadData,
   ExternalStoreBranchChange,
 } from "@assistant-ui/core";
+export { MessageNotSentError, isMessageNotSentError } from "@assistant-ui/core";
 export {
   createMessageQueue,
   type MessageQueueDriver,

package/src/legacy-runtime/runtime-cores/assistant-transport/replayBoundaryStream.test.ts

@@ -272,8 +272,11 @@ describe("createReplayBoundaryStream", () => {
   it("clears replaying when the stream ends before the boundary", async () => {
     const { waitForRender, releaseNext } = createRenderWait();
     const setReplaying = vi.fn();
+    const body = createBody(["hi"]);
     const streamPromise = createReplayBoundaryStream(
-      createResponse(["hi"], 10),
+      new Response(body, {
+        headers: { [REPLAY_CONTENT_LENGTH_HEADER]: "10" },
+      }),
       {
         setReplaying,
         waitForRender,
@@ -288,6 +291,7 @@ describe("createReplayBoundaryStream", () => {
 
     await expect(text).resolves.toBe("hi");
     expect(setReplaying).toHaveBeenLastCalledWith(false);
+    expect(body.locked).toBe(false);
   });
 
   it("clears replaying when the gated stream is cancelled", async () => {
@@ -313,6 +317,27 @@ describe("createReplayBoundaryStream", () => {
 
     expect(setReplaying).toHaveBeenLastCalledWith(false);
     expect(cancelled).toBe(true);
+    expect(body.locked).toBe(false);
+  });
+
+  it("releases the response body reader after a read failure", async () => {
+    const { waitForRender, releaseNext } = createRenderWait();
+    const error = new Error("stream failed");
+    const body = new ReadableStream<Uint8Array>({
+      start(controller) {
+        controller.error(error);
+      },
+    });
+    const streamPromise = createReplayBoundaryStream(
+      new Response(body, { headers: { [REPLAY_CONTENT_LENGTH_HEADER]: "10" } }),
+      { setReplaying: vi.fn(), waitForRender },
+    );
+
+    await releaseNext();
+    const stream = await streamPromise;
+
+    await expect(stream.getReader().read()).rejects.toBe(error);
+    expect(body.locked).toBe(false);
   });
 
   it("does not wait for replay completion after cancellation unblocks a read", async () => {

package/src/legacy-runtime/runtime-cores/assistant-transport/replayBoundaryStream.ts

@@ -90,6 +90,26 @@ export const createReplayBoundaryStream = async (
   const reader = body.getReader();
   let bytesForwarded = 0;
   let replayFinished = false;
+  let readerCleanup: Promise<void> | undefined;
+
+  const releaseReader = () => {
+    if (readerCleanup) return readerCleanup;
+    reader.releaseLock();
+    readerCleanup = Promise.resolve();
+    return readerCleanup;
+  };
+
+  const cancelReader = (reason?: unknown) => {
+    if (readerCleanup) return readerCleanup;
+    readerCleanup = (async () => {
+      try {
+        await reader.cancel(reason);
+      } finally {
+        reader.releaseLock();
+      }
+    })();
+    return readerCleanup;
+  };
 
   const finishReplay = async () => {
     if (replayFinished) return;
@@ -103,44 +123,50 @@ export const createReplayBoundaryStream = async (
 
   return new ReadableStream<Uint8Array>({
     async pull(controller) {
-      const { done, value } = await reader.read();
-
-      if (done) {
-        await finishReplay();
-        controller.close();
-        return;
-      }
-
-      if (replayFinished) {
-        controller.enqueue(value);
-        return;
-      }
-
-      const nextBytesForwarded = bytesForwarded + value.byteLength;
-
-      if (nextBytesForwarded < replayContentLength) {
-        bytesForwarded = nextBytesForwarded;
-        controller.enqueue(value);
-        return;
-      }
-
-      if (nextBytesForwarded === replayContentLength) {
-        controller.enqueue(value);
+      try {
+        const { done, value } = await reader.read();
+
+        if (done) {
+          await releaseReader();
+          await finishReplay();
+          controller.close();
+          return;
+        }
+
+        if (replayFinished) {
+          controller.enqueue(value);
+          return;
+        }
+
+        const nextBytesForwarded = bytesForwarded + value.byteLength;
+
+        if (nextBytesForwarded < replayContentLength) {
+          bytesForwarded = nextBytesForwarded;
+          controller.enqueue(value);
+          return;
+        }
+
+        if (nextBytesForwarded === replayContentLength) {
+          controller.enqueue(value);
+          await finishReplay();
+          return;
+        }
+
+        const replayBytesInChunk = replayContentLength - bytesForwarded;
+
+        controller.enqueue(value.subarray(0, replayBytesInChunk));
         await finishReplay();
-        return;
+        controller.enqueue(value.subarray(replayBytesInChunk));
+      } catch (error) {
+        await cancelReader(error).catch(() => {});
+        throw error;
       }
-
-      const replayBytesInChunk = replayContentLength - bytesForwarded;
-
-      controller.enqueue(value.subarray(0, replayBytesInChunk));
-      await finishReplay();
-      controller.enqueue(value.subarray(replayBytesInChunk));
     },
     async cancel(reason) {
       const wasFinished = replayFinished;
       replayFinished = true;
       if (!wasFinished) setReplaying(false);
-      await reader.cancel(reason);
+      await cancelReader(reason);
     },
   });
 };

package/src/tests/in-memory-thread-list.test.tsx

@@ -2,7 +2,7 @@
 
 import { render, waitFor } from "@testing-library/react";
 import type { FC } from "react";
-import { describe, it, expect } from "vitest";
+import { describe, it, expect, vi } from "vitest";
 import { useAui, AuiProvider } from "@assistant-ui/store";
 import { InMemoryThreadList } from "../client/InMemoryThreadList";
 import { ExternalThread } from "../index";
@@ -29,7 +29,7 @@ const renderThreads = () => {
   return { aui: () => captured.aui! };
 };
 
-describe("InMemoryThreadList delete", () => {
+describe("InMemoryThreadList", () => {
   it("falls back to a live thread when the switch target is deleted in the same tick", async () => {
     const { aui } = renderThreads();
 
@@ -52,4 +52,88 @@ describe("InMemoryThreadList delete", () => {
       expect(state.threadIds).not.toContain(newId);
     });
   });
+
+  it("creates unique IDs for threads created in the same millisecond", async () => {
+    const now = vi.spyOn(Date, "now").mockReturnValue(1_000);
+
+    try {
+      const { aui } = renderThreads();
+      aui().threads.switchToNewThread();
+      aui().threads.switchToNewThread();
+
+      await waitFor(() => {
+        const generatedIds = aui()
+          .threads.getState()
+          .threadIds.filter((id) => id !== "main");
+        expect(generatedIds).toHaveLength(2);
+        expect(new Set(generatedIds).size).toBe(2);
+      });
+    } finally {
+      now.mockRestore();
+    }
+  });
+});
+
+describe("InMemoryThreadList reloadMainThread", () => {
+  it("routes reloadMainThread to the main thread's refetch callback", async () => {
+    const refetched: string[] = [];
+    const captured: { aui?: ReturnType<typeof useAui> } = {};
+    const Capture: FC = () => {
+      captured.aui = useAui();
+      return null;
+    };
+    const App: FC = () => {
+      const aui = useAui({
+        threads: InMemoryThreadList({
+          thread: (threadId) =>
+            ExternalThread({
+              messages: [],
+              onRefetchThread: async () => {
+                refetched.push(threadId);
+              },
+            }),
+        }),
+      });
+      return (
+        <AuiProvider value={aui}>
+          <Capture />
+        </AuiProvider>
+      );
+    };
+    render(<App />);
+    const aui = () => captured.aui!;
+
+    await waitFor(() => expect(captured.aui).toBeDefined());
+    await aui().threads.reloadMainThread();
+    expect(refetched).toEqual(["main"]);
+
+    aui().threads.switchToNewThread();
+    await waitFor(() =>
+      expect(aui().threads.getState().mainThreadId).not.toBe("main"),
+    );
+    const newId = aui().threads.getState().mainThreadId;
+
+    await aui().threads.reloadMainThread();
+    expect(refetched).toEqual(["main", newId]);
+  });
+
+  it("resolves reloadMainThread when the thread has no refetch callback", async () => {
+    const { aui } = renderThreads();
+
+    await waitFor(() => expect(aui()).toBeDefined());
+    await expect(aui().threads.reloadMainThread()).resolves.toBeUndefined();
+  });
+});
+
+describe("InMemoryThreadList suggestions", () => {
+  it("derives the suggestions scope from the main thread", async () => {
+    const { aui } = renderThreads();
+
+    await waitFor(() => {
+      expect(aui().suggestions.getState()).toEqual({ suggestions: [] });
+    });
+    expect(aui().suggestions.getState()).toBe(
+      aui().thread.suggestions().getState(),
+    );
+  });
 });