@assistant-ui/react-ai-sdk 1.3.26 → 1.3.28

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

@@ -1,92 +1,71 @@
 "use client";
-import { useChat } from "@ai-sdk/react";
-import { useCloudThreadListAdapter, useRemoteThreadListRuntime, } from "@assistant-ui/core/react";
-import { useAui, useAuiState } from "@assistant-ui/store";
-import { useAISDKRuntime, } from "./useAISDKRuntime.js";
+import { useAISDKRuntime } from "./useAISDKRuntime.js";
 import { AssistantChatTransport } from "./AssistantChatTransport.js";
 import { useEffect, useMemo, useRef } from "react";
+import { useCloudThreadListAdapter, useRemoteThreadListRuntime } from "@assistant-ui/core/react";
+import { useAui, useAuiState } from "@assistant-ui/store";
+import { useChat } from "@ai-sdk/react";
+//#region src/ui/use-chat/useChatRuntime.ts
 const useDynamicChatTransport = (transport) => {
-    // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage
-    const transportRef = useRef(transport);
-    // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage
-    useEffect(() => {
-        transportRef.current = transport;
-    });
-    // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage
-    const dynamicTransport = useMemo(() => new Proxy(transportRef.current, {
-        get(_, prop) {
-            const res = transportRef.current[prop];
-            return typeof res === "function"
-                ? res.bind(transportRef.current)
-                : res;
-        },
-    }), []);
-    return dynamicTransport;
+	const transportRef = useRef(transport);
+	useEffect(() => {
+		transportRef.current = transport;
+	});
+	return useMemo(() => new Proxy(transportRef.current, { get(_, prop) {
+		const res = transportRef.current[prop];
+		return typeof res === "function" ? res.bind(transportRef.current) : res;
+	} }), []);
 };
 const getResumableAdapter = (transport) => {
-    if (transport instanceof AssistantChatTransport) {
-        return transport.getResumableAdapter();
-    }
-    const candidate = transport
-        .getResumableAdapter;
-    if (typeof candidate !== "function")
-        return undefined;
-    return candidate.call(transport);
+	if (transport instanceof AssistantChatTransport) return transport.getResumableAdapter();
+	const candidate = transport.getResumableAdapter;
+	if (typeof candidate !== "function") return void 0;
+	return candidate.call(transport);
 };
 const useChatThreadRuntime = (options) => {
-    const { adapters, transport: transportOptions, toCreateMessage, onResume, suggestions, ...chatOptions } = options ?? {};
-    // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage
-    const transport = useDynamicChatTransport(transportOptions ?? new AssistantChatTransport());
-    // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage
-    const id = useAuiState((s) => s.threadListItem.id);
-    // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage
-    const aui = useAui();
-    // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage
-    const chat = useChat({
-        ...chatOptions,
-        id,
-        transport,
-    });
-    // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage
-    const runtime = useAISDKRuntime(chat, {
-        adapters,
-        ...(toCreateMessage && { toCreateMessage }),
-        ...(onResume && { onResume }),
-        ...(suggestions && { suggestions }),
-    });
-    if (transport instanceof AssistantChatTransport) {
-        transport.setRuntime(runtime);
-        transport.__internal_setGetThreadListItem(() => aui.threadListItem.source ? aui.threadListItem() : undefined);
-    }
-    // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage
-    const resumeFiredRef = useRef(false);
-    // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage
-    useEffect(() => {
-        if (resumeFiredRef.current)
-            return;
-        const adapter = getResumableAdapter(transport);
-        if (!adapter)
-            return;
-        const pending = adapter.storage.getStreamId();
-        if (!pending)
-            return;
-        resumeFiredRef.current = true;
-        chat.resumeStream().catch((err) => {
-            console.warn("[assistant-ui] resumable: resume failed; clearing stored stream id", err);
-            adapter.storage.clear();
-        });
-    }, [transport, chat]);
-    return runtime;
+	const { adapters, transport: transportOptions, toCreateMessage, onResume, suggestions, ...chatOptions } = options ?? {};
+	const transport = useDynamicChatTransport(transportOptions ?? new AssistantChatTransport());
+	const id = useAuiState((s) => s.threadListItem.id);
+	const aui = useAui();
+	const chat = useChat({
+		...chatOptions,
+		id,
+		transport
+	});
+	const runtime = useAISDKRuntime(chat, {
+		adapters,
+		...toCreateMessage && { toCreateMessage },
+		...onResume && { onResume },
+		...suggestions && { suggestions }
+	});
+	if (transport instanceof AssistantChatTransport) {
+		transport.setRuntime(runtime);
+		transport.__internal_setGetThreadListItem(() => aui.threadListItem.source ? aui.threadListItem() : void 0);
+	}
+	const resumeFiredRef = useRef(false);
+	useEffect(() => {
+		if (resumeFiredRef.current) return;
+		const adapter = getResumableAdapter(transport);
+		if (!adapter) return;
+		if (!adapter.storage.getStreamId()) return;
+		resumeFiredRef.current = true;
+		chat.resumeStream().catch((err) => {
+			console.warn("[assistant-ui] resumable: resume failed; clearing stored stream id", err);
+			adapter.storage.clear();
+		});
+	}, [transport, chat]);
+	return runtime;
 };
-export const useChatRuntime = ({ cloud, ...options } = {}) => {
-    const cloudAdapter = useCloudThreadListAdapter({ cloud });
-    return useRemoteThreadListRuntime({
-        runtimeHook: function RuntimeHook() {
-            // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage
-            return useChatThreadRuntime(options);
-        },
-        adapter: cloudAdapter,
-        allowNesting: true,
-    });
+const useChatRuntime = ({ cloud, ...options } = {}) => {
+	return useRemoteThreadListRuntime({
+		runtimeHook: function RuntimeHook() {
+			return useChatThreadRuntime(options);
+		},
+		adapter: useCloudThreadListAdapter({ cloud }),
+		allowNesting: true
+	});
 };
+//#endregion
+export { useChatRuntime };
+
 //# sourceMappingURL=useChatRuntime.js.map

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

@@ -1 +1 @@
-{"version":3,"file":"useChatRuntime.js","sourceRoot":"","sources":["../../../src/ui/use-chat/useChatRuntime.ts"],"names":[],"mappings":"AAAA,YAAY,CAAC;AAEb,OAAO,EAAE,OAAO,EAAkB,MAAM,eAAe,CAAC;AAGxD,OAAO,EACL,yBAAyB,EACzB,0BAA0B,GAC3B,MAAM,0BAA0B,CAAC;AAClC,OAAO,EAAE,MAAM,EAAE,WAAW,EAAE,MAAM,qBAAqB,CAAC;AAC1D,OAAO,EACL,eAAe,GAGhB,6BAA0B;AAE3B,OAAO,EAAE,sBAAsB,EAAE,oCAAiC;AAElE,OAAO,EAAE,SAAS,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,OAAO,CAAC;AAWnD,MAAM,uBAAuB,GAAG,CAC9B,SAAoC,EACT,EAAE;IAC7B,6FAA6F;IAC7F,MAAM,YAAY,GAAG,MAAM,CAA4B,SAAS,CAAC,CAAC;IAClE,6FAA6F;IAC7F,SAAS,CAAC,GAAG,EAAE;QACb,YAAY,CAAC,OAAO,GAAG,SAAS,CAAC;IACnC,CAAC,CAAC,CAAC;IACH,6FAA6F;IAC7F,MAAM,gBAAgB,GAAG,OAAO,CAC9B,GAAG,EAAE,CACH,IAAI,KAAK,CAAC,YAAY,CAAC,OAAO,EAAE;QAC9B,GAAG,CAAC,CAAC,EAAE,IAAI;YACT,MAAM,GAAG,GACP,YAAY,CAAC,OAAO,CAAC,IAAuC,CAAC,CAAC;YAChE,OAAO,OAAO,GAAG,KAAK,UAAU;gBAC9B,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC;gBAChC,CAAC,CAAC,GAAG,CAAC;QACV,CAAC;KACF,CAAC,EACJ,EAAE,CACH,CAAC;IACF,OAAO,gBAAgB,CAAC;AAC1B,CAAC,CAAC;AAEF,MAAM,mBAAmB,GAAG,CAC1B,SAAoC,EACO,EAAE;IAC7C,IAAI,SAAS,YAAY,sBAAsB,EAAE,CAAC;QAChD,OAAO,SAAS,CAAC,mBAAmB,EAAE,CAAC;IACzC,CAAC;IACD,MAAM,SAAS,GAAI,SAAqD;SACrE,mBAAmB,CAAC;IACvB,IAAI,OAAO,SAAS,KAAK,UAAU;QAAE,OAAO,SAAS,CAAC;IACtD,OAAO,SAAS,CAAC,IAAI,CAAC,SAAS,CAA8C,CAAC;AAChF,CAAC,CAAC;AAEF,MAAM,oBAAoB,GAAG,CAC3B,OAA2C,EACzB,EAAE;IACpB,MAAM,EACJ,QAAQ,EACR,SAAS,EAAE,gBAAgB,EAC3B,eAAe,EACf,QAAQ,EACR,WAAW,EACX,GAAG,WAAW,EACf,GAAG,OAAO,IAAI,EAAE,CAAC;IAElB,6FAA6F;IAC7F,MAAM,SAAS,GAAG,uBAAuB,CACvC,gBAAgB,IAAI,IAAI,sBAAsB,EAAE,CACjD,CAAC;IAEF,6FAA6F;IAC7F,MAAM,EAAE,GAAG,WAAW,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,cAAc,CAAC,EAAE,CAAC,CAAC;IACnD,6FAA6F;IAC7F,MAAM,GAAG,GAAG,MAAM,EAAE,CAAC;IACrB,6FAA6F;IAC7F,MAAM,IAAI,GAAG,OAAO,CAAC;QACnB,GAAG,WAAW;QACd,EAAE;QACF,SAAS;KACV,CAAC,CAAC;IAEH,6FAA6F;IAC7F,MAAM,OAAO,GAAG,eAAe,CAAC,IAAI,EAAE;QACpC,QAAQ;QACR,GAAG,CAAC,eAAe,IAAI,EAAE,eAAe,EAAE,CAAC;QAC3C,GAAG,CAAC,QAAQ,IAAI,EAAE,QAAQ,EAAE,CAAC;QAC7B,GAAG,CAAC,WAAW,IAAI,EAAE,WAAW,EAAE,CAAC;KACpC,CAAC,CAAC;IAEH,IAAI,SAAS,YAAY,sBAAsB,EAAE,CAAC;QAChD,SAAS,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC;QAC9B,SAAS,CAAC,+BAA+B,CAAC,GAAG,EAAE,CAC7C,GAAG,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC,CAAC,GAAG,CAAC,cAAc,EAAE,CAAC,CAAC,CAAC,SAAS,CAC7D,CAAC;IACJ,CAAC;IAED,6FAA6F;IAC7F,MAAM,cAAc,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC;IACrC,6FAA6F;IAC7F,SAAS,CAAC,GAAG,EAAE;QACb,IAAI,cAAc,CAAC,OAAO;YAAE,OAAO;QACnC,MAAM,OAAO,GAAG,mBAAmB,CAAC,SAAS,CAAC,CAAC;QAC/C,IAAI,CAAC,OAAO;YAAE,OAAO;QACrB,MAAM,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC;QAC9C,IAAI,CAAC,OAAO;YAAE,OAAO;QACrB,cAAc,CAAC,OAAO,GAAG,IAAI,CAAC;QAC9B,IAAI,CAAC,YAAY,EAAE,CAAC,KAAK,CAAC,CAAC,GAAY,EAAE,EAAE;YACzC,OAAO,CAAC,IAAI,CACV,oEAAoE,EACpE,GAAG,CACJ,CAAC;YACF,OAAO,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;QAC1B,CAAC,CAAC,CAAC;IACL,CAAC,EAAE,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC,CAAC;IAEtB,OAAO,OAAO,CAAC;AACjB,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,cAAc,GAAG,CAA2C,EACvE,KAAK,EACL,GAAG,OAAO,KAC2B,EAAE,EAAoB,EAAE;IAC7D,MAAM,YAAY,GAAG,yBAAyB,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC;IAC1D,OAAO,0BAA0B,CAAC;QAChC,WAAW,EAAE,SAAS,WAAW;YAC/B,6FAA6F;YAC7F,OAAO,oBAAoB,CAAC,OAAO,CAAC,CAAC;QACvC,CAAC;QACD,OAAO,EAAE,YAAY;QACrB,YAAY,EAAE,IAAI;KACnB,CAAC,CAAC;AACL,CAAC,CAAC"}
+{"version":3,"file":"useChatRuntime.js","names":[],"sources":["../../../src/ui/use-chat/useChatRuntime.ts"],"sourcesContent":["\"use client\";\n\nimport { useChat, type UIMessage } from \"@ai-sdk/react\";\nimport type { AssistantCloud } from \"assistant-cloud\";\nimport type { AssistantRuntime } from \"@assistant-ui/core\";\nimport {\n  useCloudThreadListAdapter,\n  useRemoteThreadListRuntime,\n} from \"@assistant-ui/core/react\";\nimport { useAui, useAuiState } from \"@assistant-ui/store\";\nimport {\n  useAISDKRuntime,\n  type AISDKRuntimeAdapter,\n  type CustomToCreateMessageFunction,\n} from \"./useAISDKRuntime\";\nimport type { ChatInit, ChatTransport } from \"ai\";\nimport { AssistantChatTransport } from \"./AssistantChatTransport\";\nimport type { AssistantChatResumableOptions } from \"../resumable\";\nimport { useEffect, useMemo, useRef } from \"react\";\n\nexport type UseChatRuntimeOptions<UI_MESSAGE extends UIMessage = UIMessage> =\n  ChatInit<UI_MESSAGE> & {\n    cloud?: AssistantCloud | undefined;\n    adapters?: AISDKRuntimeAdapter[\"adapters\"] | undefined;\n    toCreateMessage?: CustomToCreateMessageFunction;\n    onResume?: AISDKRuntimeAdapter[\"onResume\"];\n    suggestions?: AISDKRuntimeAdapter[\"suggestions\"];\n  };\n\nconst useDynamicChatTransport = <UI_MESSAGE extends UIMessage = UIMessage>(\n  transport: ChatTransport<UI_MESSAGE>,\n): ChatTransport<UI_MESSAGE> => {\n  // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage\n  const transportRef = useRef<ChatTransport<UI_MESSAGE>>(transport);\n  // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage\n  useEffect(() => {\n    transportRef.current = transport;\n  });\n  // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage\n  const dynamicTransport = useMemo(\n    () =>\n      new Proxy(transportRef.current, {\n        get(_, prop) {\n          const res =\n            transportRef.current[prop as keyof ChatTransport<UI_MESSAGE>];\n          return typeof res === \"function\"\n            ? res.bind(transportRef.current)\n            : res;\n        },\n      }),\n    [],\n  );\n  return dynamicTransport;\n};\n\nconst getResumableAdapter = <UI_MESSAGE extends UIMessage>(\n  transport: ChatTransport<UI_MESSAGE>,\n): AssistantChatResumableOptions | undefined => {\n  if (transport instanceof AssistantChatTransport) {\n    return transport.getResumableAdapter();\n  }\n  const candidate = (transport as { getResumableAdapter?: () => unknown })\n    .getResumableAdapter;\n  if (typeof candidate !== \"function\") return undefined;\n  return candidate.call(transport) as AssistantChatResumableOptions | undefined;\n};\n\nconst useChatThreadRuntime = <UI_MESSAGE extends UIMessage = UIMessage>(\n  options?: UseChatRuntimeOptions<UI_MESSAGE>,\n): AssistantRuntime => {\n  const {\n    adapters,\n    transport: transportOptions,\n    toCreateMessage,\n    onResume,\n    suggestions,\n    ...chatOptions\n  } = options ?? {};\n\n  // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage\n  const transport = useDynamicChatTransport(\n    transportOptions ?? new AssistantChatTransport(),\n  );\n\n  // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage\n  const id = useAuiState((s) => s.threadListItem.id);\n  // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage\n  const aui = useAui();\n  // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage\n  const chat = useChat({\n    ...chatOptions,\n    id,\n    transport,\n  });\n\n  // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage\n  const runtime = useAISDKRuntime(chat, {\n    adapters,\n    ...(toCreateMessage && { toCreateMessage }),\n    ...(onResume && { onResume }),\n    ...(suggestions && { suggestions }),\n  });\n\n  if (transport instanceof AssistantChatTransport) {\n    transport.setRuntime(runtime);\n    transport.__internal_setGetThreadListItem(() =>\n      aui.threadListItem.source ? aui.threadListItem() : undefined,\n    );\n  }\n\n  // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage\n  const resumeFiredRef = useRef(false);\n  // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage\n  useEffect(() => {\n    if (resumeFiredRef.current) return;\n    const adapter = getResumableAdapter(transport);\n    if (!adapter) return;\n    const pending = adapter.storage.getStreamId();\n    if (!pending) return;\n    resumeFiredRef.current = true;\n    chat.resumeStream().catch((err: unknown) => {\n      console.warn(\n        \"[assistant-ui] resumable: resume failed; clearing stored stream id\",\n        err,\n      );\n      adapter.storage.clear();\n    });\n  }, [transport, chat]);\n\n  return runtime;\n};\n\nexport const useChatRuntime = <UI_MESSAGE extends UIMessage = UIMessage>({\n  cloud,\n  ...options\n}: UseChatRuntimeOptions<UI_MESSAGE> = {}): AssistantRuntime => {\n  const cloudAdapter = useCloudThreadListAdapter({ cloud });\n  return useRemoteThreadListRuntime({\n    runtimeHook: function RuntimeHook() {\n      // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage\n      return useChatThreadRuntime(options);\n    },\n    adapter: cloudAdapter,\n    allowNesting: true,\n  });\n};\n"],"mappings":";;;;;;;;AA6BA,MAAM,2BACJ,cAC8B;CAE9B,MAAM,eAAe,OAAkC,SAAS;CAEhE,gBAAgB;EACd,aAAa,UAAU;CACzB,CAAC;CAeD,OAbyB,cAErB,IAAI,MAAM,aAAa,SAAS,EAC9B,IAAI,GAAG,MAAM;EACX,MAAM,MACJ,aAAa,QAAQ;EACvB,OAAO,OAAO,QAAQ,aAClB,IAAI,KAAK,aAAa,OAAO,IAC7B;CACN,EACF,CAAC,GACH,CAAC,CAEmB;AACxB;AAEA,MAAM,uBACJ,cAC8C;CAC9C,IAAI,qBAAqB,wBACvB,OAAO,UAAU,oBAAoB;CAEvC,MAAM,YAAa,UAChB;CACH,IAAI,OAAO,cAAc,YAAY,OAAO,KAAA;CAC5C,OAAO,UAAU,KAAK,SAAS;AACjC;AAEA,MAAM,wBACJ,YACqB;CACrB,MAAM,EACJ,UACA,WAAW,kBACX,iBACA,UACA,aACA,GAAG,gBACD,WAAW,CAAC;CAGhB,MAAM,YAAY,wBAChB,oBAAoB,IAAI,uBAAuB,CACjD;CAGA,MAAM,KAAK,aAAa,MAAM,EAAE,eAAe,EAAE;CAEjD,MAAM,MAAM,OAAO;CAEnB,MAAM,OAAO,QAAQ;EACnB,GAAG;EACH;EACA;CACF,CAAC;CAGD,MAAM,UAAU,gBAAgB,MAAM;EACpC;EACA,GAAI,mBAAmB,EAAE,gBAAgB;EACzC,GAAI,YAAY,EAAE,SAAS;EAC3B,GAAI,eAAe,EAAE,YAAY;CACnC,CAAC;CAED,IAAI,qBAAqB,wBAAwB;EAC/C,UAAU,WAAW,OAAO;EAC5B,UAAU,sCACR,IAAI,eAAe,SAAS,IAAI,eAAe,IAAI,KAAA,CACrD;CACF;CAGA,MAAM,iBAAiB,OAAO,KAAK;CAEnC,gBAAgB;EACd,IAAI,eAAe,SAAS;EAC5B,MAAM,UAAU,oBAAoB,SAAS;EAC7C,IAAI,CAAC,SAAS;EAEd,IAAI,CADY,QAAQ,QAAQ,YACrB,GAAG;EACd,eAAe,UAAU;EACzB,KAAK,aAAa,EAAE,OAAO,QAAiB;GAC1C,QAAQ,KACN,sEACA,GACF;GACA,QAAQ,QAAQ,MAAM;EACxB,CAAC;CACH,GAAG,CAAC,WAAW,IAAI,CAAC;CAEpB,OAAO;AACT;AAEA,MAAa,kBAA4D,EACvE,OACA,GAAG,YACkC,CAAC,MAAwB;CAE9D,OAAO,2BAA2B;EAChC,aAAa,SAAS,cAAc;GAElC,OAAO,qBAAqB,OAAO;EACrC;EACA,SANmB,0BAA0B,EAAE,MAAM,CAMjC;EACpB,cAAc;CAChB,CAAC;AACH"}

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

@@ -1,5 +1,9 @@
-import type { AssistantRuntime, ThreadHistoryAdapter, ThreadMessage, MessageFormatAdapter, MessageFormatRepository, ExportedMessageRepository } from "@assistant-ui/core";
-import { type RefObject } from "react";
-export declare const toExportedMessageRepository: <TMessage>(toThreadMessages: (messages: TMessage[]) => ThreadMessage[], messages: MessageFormatRepository<TMessage>) => ExportedMessageRepository;
-export declare const useExternalHistory: <TMessage>(runtimeRef: RefObject<AssistantRuntime>, historyAdapter: ThreadHistoryAdapter | undefined, toThreadMessages: (messages: TMessage[]) => ThreadMessage[], storageFormatAdapter: MessageFormatAdapter<TMessage, any>, onSetMessages: (messages: TMessage[]) => void) => boolean;
+import { RefObject } from "react";
+import { AssistantRuntime, ExportedMessageRepository, MessageFormatAdapter, MessageFormatRepository, ThreadHistoryAdapter, ThreadMessage } from "@assistant-ui/core";
+
+//#region src/ui/use-chat/useExternalHistory.d.ts
+declare const toExportedMessageRepository: <TMessage>(toThreadMessages: (messages: TMessage[]) => ThreadMessage[], messages: MessageFormatRepository<TMessage>) => ExportedMessageRepository;
+declare const useExternalHistory: <TMessage>(runtimeRef: RefObject<AssistantRuntime>, historyAdapter: ThreadHistoryAdapter | undefined, toThreadMessages: (messages: TMessage[]) => ThreadMessage[], storageFormatAdapter: MessageFormatAdapter<TMessage, any>, onSetMessages: (messages: TMessage[]) => void) => boolean;
+//#endregion
+export { toExportedMessageRepository, useExternalHistory };
 //# sourceMappingURL=useExternalHistory.d.ts.map

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

@@ -1 +1 @@
-{"version":3,"file":"useExternalHistory.d.ts","sourceRoot":"","sources":["../../../src/ui/use-chat/useExternalHistory.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EACV,gBAAgB,EAChB,oBAAoB,EACpB,aAAa,EACb,oBAAoB,EACpB,uBAAuB,EACvB,yBAAyB,EAC1B,MAAM,oBAAoB,CAAC;AAI5B,OAAO,EAIL,KAAK,SAAS,EAGf,MAAM,OAAO,CAAC;AAEf,eAAO,MAAM,2BAA2B,GAAI,QAAQ,EAClD,kBAAkB,CAAC,QAAQ,EAAE,QAAQ,EAAE,KAAK,aAAa,EAAE,EAC3D,UAAU,uBAAuB,CAAC,QAAQ,CAAC,KAC1C,yBAWF,CAAC;AAEF,eAAO,MAAM,kBAAkB,GAAI,QAAQ,EACzC,YAAY,SAAS,CAAC,gBAAgB,CAAC,EACvC,gBAAgB,oBAAoB,GAAG,SAAS,EAChD,kBAAkB,CAAC,QAAQ,EAAE,QAAQ,EAAE,KAAK,aAAa,EAAE,EAC3D,sBAAsB,oBAAoB,CAAC,QAAQ,EAAE,GAAG,CAAC,EACzD,eAAe,CAAC,QAAQ,EAAE,QAAQ,EAAE,KAAK,IAAI,YAmP9C,CAAC"}
+{"version":3,"file":"useExternalHistory.d.ts","names":[],"sources":["../../../src/ui/use-chat/useExternalHistory.ts"],"mappings":";;;;cAsBa,2BAAA,aACX,gBAAA,GAAmB,QAAA,EAAU,QAAA,OAAe,aAAA,IAC5C,QAAA,EAAU,uBAAA,CAAwB,QAAA,MACjC,yBAAA;AAAA,cAaU,kBAAA,aACX,UAAA,EAAY,SAAA,CAAU,gBAAA,GACtB,cAAA,EAAgB,oBAAA,cAChB,gBAAA,GAAmB,QAAA,EAAU,QAAA,OAAe,aAAA,IAC5C,oBAAA,EAAsB,oBAAA,CAAqB,QAAA,QAC3C,aAAA,GAAgB,QAAA,EAAU,QAAA"}

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

@@ -1,207 +1,185 @@
 "use client";
+import { useCallback, useEffect, useMemo, useRef, useState } from "react";
 import { getExternalStoreMessages } from "@assistant-ui/core";
 import { MessageRepository } from "@assistant-ui/core/internal";
 import { useAui } from "@assistant-ui/store";
-import { useRef, useEffect, useState, useCallback, useMemo, } from "react";
-export const toExportedMessageRepository = (toThreadMessages, messages) => {
-    return {
-        headId: messages.headId,
-        messages: messages.messages.map((m) => {
-            const message = toThreadMessages([m.message])[0];
-            return {
-                ...m,
-                message,
-            };
-        }),
-    };
+//#region src/ui/use-chat/useExternalHistory.ts
+const toExportedMessageRepository = (toThreadMessages, messages) => {
+	return {
+		headId: messages.headId,
+		messages: messages.messages.map((m) => {
+			const message = toThreadMessages([m.message])[0];
+			return {
+				...m,
+				message
+			};
+		})
+	};
 };
-export const useExternalHistory = (runtimeRef, historyAdapter, toThreadMessages, storageFormatAdapter, onSetMessages) => {
-    const loadedRef = useRef(false);
-    const aui = useAui();
-    const optionalThreadListItem = useCallback(() => (aui.threadListItem.source ? aui.threadListItem() : null), [aui]);
-    const [isLoading, setIsLoading] = useState(false);
-    const historyIds = useRef(new Set());
-    const onSetMessagesRef = useRef(onSetMessages);
-    useEffect(() => {
-        onSetMessagesRef.current = onSetMessages;
-    });
-    const formatAdapter = useMemo(() => {
-        if (!historyAdapter)
-            return undefined;
-        if (!historyAdapter.withFormat) {
-            throw new Error("useAISDKRuntime: ThreadHistoryAdapter is missing the required `withFormat` method.");
-        }
-        return historyAdapter.withFormat(storageFormatAdapter);
-    }, [historyAdapter, storageFormatAdapter]);
-    useEffect(() => {
-        if (!formatAdapter || loadedRef.current)
-            return;
-        const loadHistory = async () => {
-            setIsLoading(true);
-            try {
-                const repo = await formatAdapter.load();
-                if (repo && repo.messages.length > 0) {
-                    const converted = toExportedMessageRepository(toThreadMessages, repo);
-                    runtimeRef.current.thread.import(converted);
-                    const tempRepo = new MessageRepository();
-                    tempRepo.import(converted);
-                    const messages = tempRepo.getMessages();
-                    onSetMessagesRef.current(messages.flatMap((getExternalStoreMessages)));
-                    historyIds.current = new Set(converted.messages.map((m) => m.message.id));
-                }
-            }
-            catch (error) {
-                console.error("Failed to load message history:", error);
-            }
-            finally {
-                setIsLoading(false);
-            }
-        };
-        loadedRef.current = true;
-        if (!optionalThreadListItem()?.getState().remoteId) {
-            setIsLoading(false);
-            return;
-        }
-        loadHistory();
-    }, [formatAdapter, toThreadMessages, runtimeRef, optionalThreadListItem]);
-    const runStartRef = useRef(null);
-    const persistTimerRef = useRef(null);
-    const stepBoundariesRef = useRef([]);
-    const wasRunningRef = useRef(false);
-    const toolCallCountRef = useRef(0);
-    useEffect(() => {
-        if (!formatAdapter)
-            return;
-        const unsubscribe = runtimeRef.current.thread.subscribe(() => {
-            const { isRunning } = runtimeRef.current.thread.getState();
-            const wasRunning = wasRunningRef.current;
-            wasRunningRef.current = isRunning;
-            // Track step boundaries by content changes (more reliable than isRunning)
-            if (runStartRef.current != null) {
-                const lastMsg = runtimeRef.current.thread.getState().messages.at(-1);
-                if (lastMsg?.role === "assistant") {
-                    const currentToolCallCount = lastMsg.content.filter((p) => p.type === "tool-call").length;
-                    while (toolCallCountRef.current < currentToolCallCount) {
-                        stepBoundariesRef.current.push(Date.now() - runStartRef.current);
-                        toolCallCountRef.current++;
-                    }
-                }
-            }
-            if (isRunning) {
-                if (runStartRef.current == null) {
-                    runStartRef.current = Date.now();
-                    stepBoundariesRef.current = [];
-                    toolCallCountRef.current = 0;
-                }
-                // Cancel any pending persist — isRunning went back to true
-                if (persistTimerRef.current) {
-                    clearTimeout(persistTimerRef.current);
-                    persistTimerRef.current = null;
-                }
-                return;
-            }
-            // Only act on the true→false transition
-            if (!wasRunning)
-                return;
-            // Record step boundary offset (synchronous for accuracy)
-            if (runStartRef.current != null) {
-                stepBoundariesRef.current.push(Date.now() - runStartRef.current);
-            }
-            // Debounce: wait one macrotask so agentic step flickers are absorbed
-            if (persistTimerRef.current)
-                clearTimeout(persistTimerRef.current);
-            persistTimerRef.current = setTimeout(async () => {
-                persistTimerRef.current = null;
-                // Re-read latest state — may have changed since the timeout was scheduled
-                const latest = runtimeRef.current.thread.getState();
-                if (latest.isRunning)
-                    return; // was just a flicker
-                // Derive durationMs from the last boundary (covers all steps)
-                const boundaries = stepBoundariesRef.current;
-                const durationMs = boundaries.length > 0 ? boundaries.at(-1) : undefined;
-                // Fallback: if only 1 boundary but message has multiple steps, distribute evenly
-                if (boundaries.length === 1 && durationMs != null) {
-                    const lastAssistant = latest.messages.findLast((m) => m.role === "assistant");
-                    if (lastAssistant) {
-                        const tcCount = lastAssistant.content.filter((p) => p.type === "tool-call").length;
-                        if (tcCount > 0) {
-                            const totalSteps = tcCount + 1;
-                            const stepDur = durationMs / totalSteps;
-                            boundaries.length = 0;
-                            for (let i = 0; i < totalSteps; i++) {
-                                boundaries.push(Math.round((i + 1) * stepDur));
-                            }
-                        }
-                    }
-                }
-                // Build per-step timestamps when there are multiple steps
-                const stepTimestamps = boundaries.length > 1
-                    ? boundaries.map((endMs, i) => ({
-                        start_ms: i === 0 ? 0 : boundaries[i - 1],
-                        end_ms: endMs,
-                    }))
-                    : undefined;
-                runStartRef.current = null;
-                stepBoundariesRef.current = [];
-                const telemetryOptions = {
-                    ...(durationMs != null ? { durationMs } : undefined),
-                    ...(stepTimestamps != null ? { stepTimestamps } : undefined),
-                };
-                const { messages } = latest;
-                let lastInnerMessageId = null;
-                const getLastInnerId = (msgs) => msgs.length > 0 ? storageFormatAdapter.getId(msgs.at(-1)) : null;
-                const toBatchItems = (msgs) => msgs.map((msg, idx) => ({
-                    parentId: idx === 0
-                        ? lastInnerMessageId
-                        : storageFormatAdapter.getId(msgs[idx - 1]),
-                    message: msg,
-                }));
-                for (const message of messages) {
-                    const innerMessages = getExternalStoreMessages(message);
-                    const isReady = message.status === undefined ||
-                        message.status.type === "complete" ||
-                        message.status.type === "incomplete";
-                    if (!isReady) {
-                        lastInnerMessageId =
-                            getLastInnerId(innerMessages) ?? lastInnerMessageId;
-                        continue;
-                    }
-                    if (historyIds.current.has(message.id)) {
-                        if (durationMs !== undefined) {
-                            let parentId = lastInnerMessageId;
-                            for (const innerMessage of innerMessages) {
-                                try {
-                                    await formatAdapter.update?.({ parentId, message: innerMessage }, storageFormatAdapter.getId(innerMessage));
-                                }
-                                catch {
-                                    // ignore update failures to avoid breaking the message processing loop
-                                }
-                                parentId = storageFormatAdapter.getId(innerMessage);
-                            }
-                        }
-                        lastInnerMessageId =
-                            getLastInnerId(innerMessages) ?? lastInnerMessageId;
-                        continue;
-                    }
-                    historyIds.current.add(message.id);
-                    const batchItems = toBatchItems(innerMessages);
-                    for (const item of batchItems) {
-                        await formatAdapter.append(item);
-                    }
-                    lastInnerMessageId =
-                        getLastInnerId(innerMessages) ?? lastInnerMessageId;
-                    formatAdapter.reportTelemetry?.(batchItems, telemetryOptions);
-                }
-            }, 0);
-        });
-        return () => {
-            unsubscribe();
-            if (persistTimerRef.current) {
-                clearTimeout(persistTimerRef.current);
-                persistTimerRef.current = null;
-            }
-        };
-    }, [formatAdapter, storageFormatAdapter, runtimeRef]);
-    return isLoading;
+const useExternalHistory = (runtimeRef, historyAdapter, toThreadMessages, storageFormatAdapter, onSetMessages) => {
+	const loadedRef = useRef(false);
+	const aui = useAui();
+	const optionalThreadListItem = useCallback(() => aui.threadListItem.source ? aui.threadListItem() : null, [aui]);
+	const [isLoading, setIsLoading] = useState(false);
+	const historyIds = useRef(/* @__PURE__ */ new Set());
+	const onSetMessagesRef = useRef(onSetMessages);
+	useEffect(() => {
+		onSetMessagesRef.current = onSetMessages;
+	});
+	const formatAdapter = useMemo(() => {
+		if (!historyAdapter) return void 0;
+		if (!historyAdapter.withFormat) throw new Error("useAISDKRuntime: ThreadHistoryAdapter is missing the required `withFormat` method.");
+		return historyAdapter.withFormat(storageFormatAdapter);
+	}, [historyAdapter, storageFormatAdapter]);
+	useEffect(() => {
+		if (!formatAdapter || loadedRef.current) return;
+		const loadHistory = async () => {
+			setIsLoading(true);
+			try {
+				const repo = await formatAdapter.load();
+				if (repo && repo.messages.length > 0) {
+					const converted = toExportedMessageRepository(toThreadMessages, repo);
+					runtimeRef.current.thread.import(converted);
+					const tempRepo = new MessageRepository();
+					tempRepo.import(converted);
+					const messages = tempRepo.getMessages();
+					onSetMessagesRef.current(messages.flatMap(getExternalStoreMessages));
+					historyIds.current = new Set(converted.messages.map((m) => m.message.id));
+				}
+			} catch (error) {
+				console.error("Failed to load message history:", error);
+			} finally {
+				setIsLoading(false);
+			}
+		};
+		loadedRef.current = true;
+		if (!optionalThreadListItem()?.getState().remoteId) {
+			setIsLoading(false);
+			return;
+		}
+		loadHistory();
+	}, [
+		formatAdapter,
+		toThreadMessages,
+		runtimeRef,
+		optionalThreadListItem
+	]);
+	const runStartRef = useRef(null);
+	const persistTimerRef = useRef(null);
+	const stepBoundariesRef = useRef([]);
+	const wasRunningRef = useRef(false);
+	const toolCallCountRef = useRef(0);
+	useEffect(() => {
+		if (!formatAdapter) return;
+		const unsubscribe = runtimeRef.current.thread.subscribe(() => {
+			const { isRunning } = runtimeRef.current.thread.getState();
+			const wasRunning = wasRunningRef.current;
+			wasRunningRef.current = isRunning;
+			if (runStartRef.current != null) {
+				const lastMsg = runtimeRef.current.thread.getState().messages.at(-1);
+				if (lastMsg?.role === "assistant") {
+					const currentToolCallCount = lastMsg.content.filter((p) => p.type === "tool-call").length;
+					while (toolCallCountRef.current < currentToolCallCount) {
+						stepBoundariesRef.current.push(Date.now() - runStartRef.current);
+						toolCallCountRef.current++;
+					}
+				}
+			}
+			if (isRunning) {
+				if (runStartRef.current == null) {
+					runStartRef.current = Date.now();
+					stepBoundariesRef.current = [];
+					toolCallCountRef.current = 0;
+				}
+				if (persistTimerRef.current) {
+					clearTimeout(persistTimerRef.current);
+					persistTimerRef.current = null;
+				}
+				return;
+			}
+			if (!wasRunning) return;
+			if (runStartRef.current != null) stepBoundariesRef.current.push(Date.now() - runStartRef.current);
+			if (persistTimerRef.current) clearTimeout(persistTimerRef.current);
+			persistTimerRef.current = setTimeout(async () => {
+				persistTimerRef.current = null;
+				const latest = runtimeRef.current.thread.getState();
+				if (latest.isRunning) return;
+				const boundaries = stepBoundariesRef.current;
+				const durationMs = boundaries.length > 0 ? boundaries.at(-1) : void 0;
+				if (boundaries.length === 1 && durationMs != null) {
+					const lastAssistant = latest.messages.findLast((m) => m.role === "assistant");
+					if (lastAssistant) {
+						const tcCount = lastAssistant.content.filter((p) => p.type === "tool-call").length;
+						if (tcCount > 0) {
+							const totalSteps = tcCount + 1;
+							const stepDur = durationMs / totalSteps;
+							boundaries.length = 0;
+							for (let i = 0; i < totalSteps; i++) boundaries.push(Math.round((i + 1) * stepDur));
+						}
+					}
+				}
+				const stepTimestamps = boundaries.length > 1 ? boundaries.map((endMs, i) => ({
+					start_ms: i === 0 ? 0 : boundaries[i - 1],
+					end_ms: endMs
+				})) : void 0;
+				runStartRef.current = null;
+				stepBoundariesRef.current = [];
+				const telemetryOptions = {
+					...durationMs != null ? { durationMs } : void 0,
+					...stepTimestamps != null ? { stepTimestamps } : void 0
+				};
+				const { messages } = latest;
+				let lastInnerMessageId = null;
+				const getLastInnerId = (msgs) => msgs.length > 0 ? storageFormatAdapter.getId(msgs.at(-1)) : null;
+				const toBatchItems = (msgs) => msgs.map((msg, idx) => ({
+					parentId: idx === 0 ? lastInnerMessageId : storageFormatAdapter.getId(msgs[idx - 1]),
+					message: msg
+				}));
+				for (const message of messages) {
+					const innerMessages = getExternalStoreMessages(message);
+					if (!(message.status === void 0 || message.status.type === "complete" || message.status.type === "incomplete")) {
+						lastInnerMessageId = getLastInnerId(innerMessages) ?? lastInnerMessageId;
+						continue;
+					}
+					if (historyIds.current.has(message.id)) {
+						if (durationMs !== void 0) {
+							let parentId = lastInnerMessageId;
+							for (const innerMessage of innerMessages) {
+								try {
+									await formatAdapter.update?.({
+										parentId,
+										message: innerMessage
+									}, storageFormatAdapter.getId(innerMessage));
+								} catch {}
+								parentId = storageFormatAdapter.getId(innerMessage);
+							}
+						}
+						lastInnerMessageId = getLastInnerId(innerMessages) ?? lastInnerMessageId;
+						continue;
+					}
+					historyIds.current.add(message.id);
+					const batchItems = toBatchItems(innerMessages);
+					for (const item of batchItems) await formatAdapter.append(item);
+					lastInnerMessageId = getLastInnerId(innerMessages) ?? lastInnerMessageId;
+					formatAdapter.reportTelemetry?.(batchItems, telemetryOptions);
+				}
+			}, 0);
+		});
+		return () => {
+			unsubscribe();
+			if (persistTimerRef.current) {
+				clearTimeout(persistTimerRef.current);
+				persistTimerRef.current = null;
+			}
+		};
+	}, [
+		formatAdapter,
+		storageFormatAdapter,
+		runtimeRef
+	]);
+	return isLoading;
 };
+//#endregion
+export { toExportedMessageRepository, useExternalHistory };
+
 //# sourceMappingURL=useExternalHistory.js.map