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

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

@@ -1,116 +1,106 @@
-import { DefaultChatTransport, } from "ai";
-import { toToolsJSONSchema } from "assistant-stream";
-import { RESUMABLE_STREAM_ID_HEADER, } from "../resumable.js";
-const FINISH_MARKER = '"type":"finish"';
+import { toToolsJSONSchema } from "../../packages/assistant-stream/dist/core/tool/schema-utils.js";
+import { RESUMABLE_STREAM_ID_HEADER } from "../../packages/assistant-stream/dist/resumable/createResumableAssistantStreamResponse.js";
+import { DefaultChatTransport } from "ai";
+//#region src/ui/use-chat/AssistantChatTransport.ts
+const FINISH_MARKER = "\"type\":\"finish\"";
 const FINISH_BUFFER_LIMIT = 4096;
-const FINISH_BUFFER_TAIL = 1024;
-export class AssistantChatTransport extends DefaultChatTransport {
-    runtime;
-    getThreadListItem;
-    resumable;
-    constructor(initOptions) {
-        const { resumable, ...rest } = initOptions ?? {};
-        const userFetch = rest.fetch;
-        const userPrepareReconnect = rest.prepareReconnectToStreamRequest;
-        super({
-            ...rest,
-            ...(resumable && {
-                fetch: wrapFetchWithResumable(resumable, userFetch),
-                prepareReconnectToStreamRequest: wrapPrepareReconnect(resumable, userPrepareReconnect),
-            }),
-            prepareSendMessagesRequest: async (options) => {
-                const context = this.runtime?.thread.getModelContext();
-                const threadListItem = this.getThreadListItem?.() ?? this.runtime?.threads.mainItem;
-                const id = (await threadListItem?.initialize())?.remoteId ?? options.id;
-                const optionsEx = {
-                    ...options,
-                    body: {
-                        callSettings: context?.callSettings,
-                        system: context?.system,
-                        config: context?.config,
-                        tools: toToolsJSONSchema(context?.tools ?? {}),
-                        ...options?.body,
-                    },
-                };
-                const preparedRequest = await rest.prepareSendMessagesRequest?.(optionsEx);
-                return {
-                    ...preparedRequest,
-                    body: preparedRequest?.body ?? {
-                        ...optionsEx.body,
-                        id,
-                        messages: options.messages,
-                        trigger: options.trigger,
-                        messageId: options.messageId,
-                        metadata: options.requestMetadata,
-                    },
-                };
-            },
-        });
-        this.resumable = resumable;
-    }
-    setRuntime(runtime) {
-        this.runtime = runtime;
-    }
-    getResumableAdapter() {
-        return this.resumable;
-    }
-    __internal_setGetThreadListItem(getter) {
-        this.getThreadListItem = getter;
-    }
-}
+var AssistantChatTransport = class extends DefaultChatTransport {
+	runtime;
+	getThreadListItem;
+	resumable;
+	constructor(initOptions) {
+		const { resumable, ...rest } = initOptions ?? {};
+		const userFetch = rest.fetch;
+		const userPrepareReconnect = rest.prepareReconnectToStreamRequest;
+		super({
+			...rest,
+			...resumable && {
+				fetch: wrapFetchWithResumable(resumable, userFetch),
+				prepareReconnectToStreamRequest: wrapPrepareReconnect(resumable, userPrepareReconnect)
+			},
+			prepareSendMessagesRequest: async (options) => {
+				const context = this.runtime?.thread.getModelContext();
+				const id = (await (this.getThreadListItem?.() ?? this.runtime?.threads.mainItem)?.initialize())?.remoteId ?? options.id;
+				const optionsEx = {
+					...options,
+					body: {
+						callSettings: context?.callSettings,
+						system: context?.system,
+						config: context?.config,
+						tools: toToolsJSONSchema(context?.tools ?? {}),
+						...options?.body
+					}
+				};
+				const preparedRequest = await rest.prepareSendMessagesRequest?.(optionsEx);
+				return {
+					...preparedRequest,
+					body: preparedRequest?.body ?? {
+						...optionsEx.body,
+						id,
+						messages: options.messages,
+						trigger: options.trigger,
+						messageId: options.messageId,
+						metadata: options.requestMetadata
+					}
+				};
+			}
+		});
+		this.resumable = resumable;
+	}
+	setRuntime(runtime) {
+		this.runtime = runtime;
+	}
+	getResumableAdapter() {
+		return this.resumable;
+	}
+	__internal_setGetThreadListItem(getter) {
+		this.getThreadListItem = getter;
+	}
+};
 function wrapFetchWithResumable(resumable, userFetch) {
-    const baseFetch = userFetch
-        ? (input, init) => userFetch(input, init)
-        : globalThis.fetch.bind(globalThis);
-    return async (input, init) => {
-        const res = await baseFetch(input, init);
-        const id = res.headers.get(RESUMABLE_STREAM_ID_HEADER);
-        if (id)
-            resumable.storage.setStreamId(id);
-        if (!res.body)
-            return res;
-        const detectFinish = resumable.isFinishEvent ?? defaultIsFinishEvent;
-        // a single decoder is required so multi-byte sequences split across
-        // chunks buffer via stream: true rather than getting dropped.
-        const decoder = new TextDecoder();
-        let accumulator = "";
-        const tap = new TransformStream({
-            transform(chunk, controller) {
-                controller.enqueue(chunk);
-                accumulator += decoder.decode(chunk, { stream: true });
-                if (detectFinish(chunk, accumulator)) {
-                    resumable.storage.clear();
-                    accumulator = "";
-                }
-                else if (accumulator.length > FINISH_BUFFER_LIMIT) {
-                    accumulator = accumulator.slice(-FINISH_BUFFER_TAIL);
-                }
-            },
-        });
-        return new Response(res.body.pipeThrough(tap), {
-            status: res.status,
-            statusText: res.statusText,
-            headers: res.headers,
-        });
-    };
+	const baseFetch = userFetch ? (input, init) => userFetch(input, init) : globalThis.fetch.bind(globalThis);
+	return async (input, init) => {
+		const res = await baseFetch(input, init);
+		const id = res.headers.get(RESUMABLE_STREAM_ID_HEADER);
+		if (id) resumable.storage.setStreamId(id);
+		if (!res.body) return res;
+		const detectFinish = resumable.isFinishEvent ?? defaultIsFinishEvent;
+		const decoder = new TextDecoder();
+		let accumulator = "";
+		const tap = new TransformStream({ transform(chunk, controller) {
+			controller.enqueue(chunk);
+			accumulator += decoder.decode(chunk, { stream: true });
+			if (detectFinish(chunk, accumulator)) {
+				resumable.storage.clear();
+				accumulator = "";
+			} else if (accumulator.length > FINISH_BUFFER_LIMIT) accumulator = accumulator.slice(-1024);
+		} });
+		return new Response(res.body.pipeThrough(tap), {
+			status: res.status,
+			statusText: res.statusText,
+			headers: res.headers
+		});
+	};
 }
 function defaultIsFinishEvent(_chunk, accumulator) {
-    return accumulator.includes(FINISH_MARKER);
+	return accumulator.includes(FINISH_MARKER);
 }
 function wrapPrepareReconnect(resumable, userPrepareReconnect) {
-    return async (options) => {
-        const streamId = resumable.storage.getStreamId();
-        if (!streamId) {
-            throw new Error("AssistantChatTransport: no resumable stream id available; nothing to resume");
-        }
-        const api = typeof resumable.resumeApi === "function"
-            ? resumable.resumeApi(streamId)
-            : resumable.resumeApi;
-        const userPrepared = await userPrepareReconnect?.({ ...options, api });
-        return {
-            ...userPrepared,
-            api: userPrepared?.api ?? api,
-        };
-    };
+	return async (options) => {
+		const streamId = resumable.storage.getStreamId();
+		if (!streamId) throw new Error("AssistantChatTransport: no resumable stream id available; nothing to resume");
+		const api = typeof resumable.resumeApi === "function" ? resumable.resumeApi(streamId) : resumable.resumeApi;
+		const userPrepared = await userPrepareReconnect?.({
+			...options,
+			api
+		});
+		return {
+			...userPrepared,
+			api: userPrepared?.api ?? api
+		};
+	};
 }
+//#endregion
+export { AssistantChatTransport };
+
 //# sourceMappingURL=AssistantChatTransport.js.map

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

@@ -1 +1 @@
-{"version":3,"file":"AssistantChatTransport.js","sourceRoot":"","sources":["../../../src/ui/use-chat/AssistantChatTransport.ts"],"names":[],"mappings":"AAIA,OAAO,EACL,oBAAoB,GAGrB,MAAM,IAAI,CAAC;AACZ,OAAO,EAAE,iBAAiB,EAAE,MAAM,kBAAkB,CAAC;AACrD,OAAO,EACL,0BAA0B,GAE3B,wBAAqB;AAItB,MAAM,aAAa,GAAG,iBAAiB,CAAC;AACxC,MAAM,mBAAmB,GAAG,IAAI,CAAC;AACjC,MAAM,kBAAkB,GAAG,IAAI,CAAC;AAOhC,MAAM,OAAO,sBAEX,SAAQ,oBAAgC;IAChC,OAAO,CAA+B;IACtC,iBAAiB,CAEX;IACG,SAAS,CAA4C;IAEtE,YAAY,WAA2D;QACrE,MAAM,EAAE,SAAS,EAAE,GAAG,IAAI,EAAE,GAAG,WAAW,IAAI,EAAE,CAAC;QACjD,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC;QAC7B,MAAM,oBAAoB,GAAG,IAAI,CAAC,+BAA+B,CAAC;QAElE,KAAK,CAAC;YACJ,GAAG,IAAI;YACP,GAAG,CAAC,SAAS,IAAI;gBACf,KAAK,EAAE,sBAAsB,CAAC,SAAS,EAAE,SAAS,CAAC;gBACnD,+BAA+B,EAAE,oBAAoB,CACnD,SAAS,EACT,oBAAoB,CACrB;aACF,CAAC;YACF,0BAA0B,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE;gBAC5C,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,EAAE,MAAM,CAAC,eAAe,EAAE,CAAC;gBACvD,MAAM,cAAc,GAClB,IAAI,CAAC,iBAAiB,EAAE,EAAE,IAAI,IAAI,CAAC,OAAO,EAAE,OAAO,CAAC,QAAQ,CAAC;gBAC/D,MAAM,EAAE,GAAG,CAAC,MAAM,cAAc,EAAE,UAAU,EAAE,CAAC,EAAE,QAAQ,IAAI,OAAO,CAAC,EAAE,CAAC;gBAExE,MAAM,SAAS,GAAG;oBAChB,GAAG,OAAO;oBACV,IAAI,EAAE;wBACJ,YAAY,EAAE,OAAO,EAAE,YAAY;wBACnC,MAAM,EAAE,OAAO,EAAE,MAAM;wBACvB,MAAM,EAAE,OAAO,EAAE,MAAM;wBACvB,KAAK,EAAE,iBAAiB,CAAC,OAAO,EAAE,KAAK,IAAI,EAAE,CAAC;wBAC9C,GAAG,OAAO,EAAE,IAAI;qBACjB;iBACF,CAAC;gBACF,MAAM,eAAe,GACnB,MAAM,IAAI,CAAC,0BAA0B,EAAE,CAAC,SAAS,CAAC,CAAC;gBAErD,OAAO;oBACL,GAAG,eAAe;oBAClB,IAAI,EAAE,eAAe,EAAE,IAAI,IAAI;wBAC7B,GAAG,SAAS,CAAC,IAAI;wBACjB,EAAE;wBACF,QAAQ,EAAE,OAAO,CAAC,QAAQ;wBAC1B,OAAO,EAAE,OAAO,CAAC,OAAO;wBACxB,SAAS,EAAE,OAAO,CAAC,SAAS;wBAC5B,QAAQ,EAAE,OAAO,CAAC,eAAe;qBAClC;iBACF,CAAC;YACJ,CAAC;SACF,CAAC,CAAC;QAEH,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;IAC7B,CAAC;IAED,UAAU,CAAC,OAAyB;QAClC,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;IACzB,CAAC;IAED,mBAAmB;QACjB,OAAO,IAAI,CAAC,SAAS,CAAC;IACxB,CAAC;IAED,+BAA+B,CAC7B,MAAqD;QAErD,IAAI,CAAC,iBAAiB,GAAG,MAAM,CAAC;IAClC,CAAC;CACF;AAED,SAAS,sBAAsB,CAC7B,SAAwC,EACxC,SAA2D;IAE3D,MAAM,SAAS,GAA4B,SAAS;QAClD,CAAC,CAAC,CAAC,KAAK,EAAE,IAAI,EAAE,EAAE,CAAC,SAAS,CAAC,KAA0B,EAAE,IAAI,CAAC;QAC9D,CAAC,CAAC,UAAU,CAAC,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IAEtC,OAAO,KAAK,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE;QAC3B,MAAM,GAAG,GAAG,MAAM,SAAS,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;QACzC,MAAM,EAAE,GAAG,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,0BAA0B,CAAC,CAAC;QACvD,IAAI,EAAE;YAAE,SAAS,CAAC,OAAO,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC;QAC1C,IAAI,CAAC,GAAG,CAAC,IAAI;YAAE,OAAO,GAAG,CAAC;QAE1B,MAAM,YAAY,GAAG,SAAS,CAAC,aAAa,IAAI,oBAAoB,CAAC;QACrE,oEAAoE;QACpE,8DAA8D;QAC9D,MAAM,OAAO,GAAG,IAAI,WAAW,EAAE,CAAC;QAClC,IAAI,WAAW,GAAG,EAAE,CAAC;QACrB,MAAM,GAAG,GAAG,IAAI,eAAe,CAAyB;YACtD,SAAS,CAAC,KAAK,EAAE,UAAU;gBACzB,UAAU,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;gBAC1B,WAAW,IAAI,OAAO,CAAC,MAAM,CAAC,KAAK,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;gBACvD,IAAI,YAAY,CAAC,KAAK,EAAE,WAAW,CAAC,EAAE,CAAC;oBACrC,SAAS,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;oBAC1B,WAAW,GAAG,EAAE,CAAC;gBACnB,CAAC;qBAAM,IAAI,WAAW,CAAC,MAAM,GAAG,mBAAmB,EAAE,CAAC;oBACpD,WAAW,GAAG,WAAW,CAAC,KAAK,CAAC,CAAC,kBAAkB,CAAC,CAAC;gBACvD,CAAC;YACH,CAAC;SACF,CAAC,CAAC;QAEH,OAAO,IAAI,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,EAAE;YAC7C,MAAM,EAAE,GAAG,CAAC,MAAM;YAClB,UAAU,EAAE,GAAG,CAAC,UAAU;YAC1B,OAAO,EAAE,GAAG,CAAC,OAAO;SACrB,CAAC,CAAC;IACL,CAAC,CAAC;AACJ,CAAC;AAED,SAAS,oBAAoB,CAAC,MAAkB,EAAE,WAAmB;IACnE,OAAO,WAAW,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAC;AAC7C,CAAC;AAED,SAAS,oBAAoB,CAC3B,SAAwC,EACxC,oBAAgG;IAIhG,OAAO,KAAK,EAAE,OAAO,EAAE,EAAE;QACvB,MAAM,QAAQ,GAAG,SAAS,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC;QACjD,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,MAAM,IAAI,KAAK,CACb,6EAA6E,CAC9E,CAAC;QACJ,CAAC;QACD,MAAM,GAAG,GACP,OAAO,SAAS,CAAC,SAAS,KAAK,UAAU;YACvC,CAAC,CAAC,SAAS,CAAC,SAAS,CAAC,QAAQ,CAAC;YAC/B,CAAC,CAAC,SAAS,CAAC,SAAS,CAAC;QAC1B,MAAM,YAAY,GAAG,MAAM,oBAAoB,EAAE,CAAC,EAAE,GAAG,OAAO,EAAE,GAAG,EAAE,CAAC,CAAC;QACvE,OAAO;YACL,GAAG,YAAY;YACf,GAAG,EAAE,YAAY,EAAE,GAAG,IAAI,GAAG;SAC9B,CAAC;IACJ,CAAC,CAAC;AACJ,CAAC"}
+{"version":3,"file":"AssistantChatTransport.js","names":[],"sources":["../../../src/ui/use-chat/AssistantChatTransport.ts"],"sourcesContent":["import type {\n  AssistantRuntime,\n  ThreadListItemRuntime,\n} from \"@assistant-ui/core\";\nimport {\n  DefaultChatTransport,\n  type HttpChatTransportInitOptions,\n  type UIMessage,\n} from \"ai\";\nimport { toToolsJSONSchema } from \"assistant-stream\";\nimport {\n  RESUMABLE_STREAM_ID_HEADER,\n  type AssistantChatResumableOptions,\n} from \"../resumable\";\n\ntype InitializableThreadListItem = Pick<ThreadListItemRuntime, \"initialize\">;\n\nconst FINISH_MARKER = '\"type\":\"finish\"';\nconst FINISH_BUFFER_LIMIT = 4096;\nconst FINISH_BUFFER_TAIL = 1024;\n\nexport type AssistantChatTransportInitOptions<UI_MESSAGE extends UIMessage> =\n  HttpChatTransportInitOptions<UI_MESSAGE> & {\n    resumable?: AssistantChatResumableOptions;\n  };\n\nexport class AssistantChatTransport<\n  UI_MESSAGE extends UIMessage,\n> extends DefaultChatTransport<UI_MESSAGE> {\n  private runtime: AssistantRuntime | undefined;\n  private getThreadListItem:\n    | (() => InitializableThreadListItem | undefined)\n    | undefined;\n  private readonly resumable: AssistantChatResumableOptions | undefined;\n\n  constructor(initOptions?: AssistantChatTransportInitOptions<UI_MESSAGE>) {\n    const { resumable, ...rest } = initOptions ?? {};\n    const userFetch = rest.fetch;\n    const userPrepareReconnect = rest.prepareReconnectToStreamRequest;\n\n    super({\n      ...rest,\n      ...(resumable && {\n        fetch: wrapFetchWithResumable(resumable, userFetch),\n        prepareReconnectToStreamRequest: wrapPrepareReconnect(\n          resumable,\n          userPrepareReconnect,\n        ),\n      }),\n      prepareSendMessagesRequest: async (options) => {\n        const context = this.runtime?.thread.getModelContext();\n        const threadListItem =\n          this.getThreadListItem?.() ?? this.runtime?.threads.mainItem;\n        const id = (await threadListItem?.initialize())?.remoteId ?? options.id;\n\n        const optionsEx = {\n          ...options,\n          body: {\n            callSettings: context?.callSettings,\n            system: context?.system,\n            config: context?.config,\n            tools: toToolsJSONSchema(context?.tools ?? {}),\n            ...options?.body,\n          },\n        };\n        const preparedRequest =\n          await rest.prepareSendMessagesRequest?.(optionsEx);\n\n        return {\n          ...preparedRequest,\n          body: preparedRequest?.body ?? {\n            ...optionsEx.body,\n            id,\n            messages: options.messages,\n            trigger: options.trigger,\n            messageId: options.messageId,\n            metadata: options.requestMetadata,\n          },\n        };\n      },\n    });\n\n    this.resumable = resumable;\n  }\n\n  setRuntime(runtime: AssistantRuntime) {\n    this.runtime = runtime;\n  }\n\n  getResumableAdapter(): AssistantChatResumableOptions | undefined {\n    return this.resumable;\n  }\n\n  __internal_setGetThreadListItem(\n    getter: () => InitializableThreadListItem | undefined,\n  ) {\n    this.getThreadListItem = getter;\n  }\n}\n\nfunction wrapFetchWithResumable(\n  resumable: AssistantChatResumableOptions,\n  userFetch: HttpChatTransportInitOptions<UIMessage>[\"fetch\"],\n): NonNullable<HttpChatTransportInitOptions<UIMessage>[\"fetch\"]> {\n  const baseFetch: typeof globalThis.fetch = userFetch\n    ? (input, init) => userFetch(input as RequestInfo | URL, init)\n    : globalThis.fetch.bind(globalThis);\n\n  return async (input, init) => {\n    const res = await baseFetch(input, init);\n    const id = res.headers.get(RESUMABLE_STREAM_ID_HEADER);\n    if (id) resumable.storage.setStreamId(id);\n    if (!res.body) return res;\n\n    const detectFinish = resumable.isFinishEvent ?? defaultIsFinishEvent;\n    // a single decoder is required so multi-byte sequences split across\n    // chunks buffer via stream: true rather than getting dropped.\n    const decoder = new TextDecoder();\n    let accumulator = \"\";\n    const tap = new TransformStream<Uint8Array, Uint8Array>({\n      transform(chunk, controller) {\n        controller.enqueue(chunk);\n        accumulator += decoder.decode(chunk, { stream: true });\n        if (detectFinish(chunk, accumulator)) {\n          resumable.storage.clear();\n          accumulator = \"\";\n        } else if (accumulator.length > FINISH_BUFFER_LIMIT) {\n          accumulator = accumulator.slice(-FINISH_BUFFER_TAIL);\n        }\n      },\n    });\n\n    return new Response(res.body.pipeThrough(tap), {\n      status: res.status,\n      statusText: res.statusText,\n      headers: res.headers,\n    });\n  };\n}\n\nfunction defaultIsFinishEvent(_chunk: Uint8Array, accumulator: string) {\n  return accumulator.includes(FINISH_MARKER);\n}\n\nfunction wrapPrepareReconnect(\n  resumable: AssistantChatResumableOptions,\n  userPrepareReconnect: HttpChatTransportInitOptions<UIMessage>[\"prepareReconnectToStreamRequest\"],\n): NonNullable<\n  HttpChatTransportInitOptions<UIMessage>[\"prepareReconnectToStreamRequest\"]\n> {\n  return async (options) => {\n    const streamId = resumable.storage.getStreamId();\n    if (!streamId) {\n      throw new Error(\n        \"AssistantChatTransport: no resumable stream id available; nothing to resume\",\n      );\n    }\n    const api =\n      typeof resumable.resumeApi === \"function\"\n        ? resumable.resumeApi(streamId)\n        : resumable.resumeApi;\n    const userPrepared = await userPrepareReconnect?.({ ...options, api });\n    return {\n      ...userPrepared,\n      api: userPrepared?.api ?? api,\n    };\n  };\n}\n"],"mappings":";;;;AAiBA,MAAM,gBAAgB;AACtB,MAAM,sBAAsB;AAQ5B,IAAa,yBAAb,cAEU,qBAAiC;CACzC;CACA;CAGA;CAEA,YAAY,aAA6D;EACvE,MAAM,EAAE,WAAW,GAAG,SAAS,eAAe,CAAC;EAC/C,MAAM,YAAY,KAAK;EACvB,MAAM,uBAAuB,KAAK;EAElC,MAAM;GACJ,GAAG;GACH,GAAI,aAAa;IACf,OAAO,uBAAuB,WAAW,SAAS;IAClD,iCAAiC,qBAC/B,WACA,oBACF;GACF;GACA,4BAA4B,OAAO,YAAY;IAC7C,MAAM,UAAU,KAAK,SAAS,OAAO,gBAAgB;IAGrD,MAAM,MAAM,OADV,KAAK,oBAAoB,KAAK,KAAK,SAAS,QAAQ,WACpB,WAAW,IAAI,YAAY,QAAQ;IAErE,MAAM,YAAY;KAChB,GAAG;KACH,MAAM;MACJ,cAAc,SAAS;MACvB,QAAQ,SAAS;MACjB,QAAQ,SAAS;MACjB,OAAO,kBAAkB,SAAS,SAAS,CAAC,CAAC;MAC7C,GAAG,SAAS;KACd;IACF;IACA,MAAM,kBACJ,MAAM,KAAK,6BAA6B,SAAS;IAEnD,OAAO;KACL,GAAG;KACH,MAAM,iBAAiB,QAAQ;MAC7B,GAAG,UAAU;MACb;MACA,UAAU,QAAQ;MAClB,SAAS,QAAQ;MACjB,WAAW,QAAQ;MACnB,UAAU,QAAQ;KACpB;IACF;GACF;EACF,CAAC;EAED,KAAK,YAAY;CACnB;CAEA,WAAW,SAA2B;EACpC,KAAK,UAAU;CACjB;CAEA,sBAAiE;EAC/D,OAAO,KAAK;CACd;CAEA,gCACE,QACA;EACA,KAAK,oBAAoB;CAC3B;AACF;AAEA,SAAS,uBACP,WACA,WAC+D;CAC/D,MAAM,YAAqC,aACtC,OAAO,SAAS,UAAU,OAA4B,IAAI,IAC3D,WAAW,MAAM,KAAK,UAAU;CAEpC,OAAO,OAAO,OAAO,SAAS;EAC5B,MAAM,MAAM,MAAM,UAAU,OAAO,IAAI;EACvC,MAAM,KAAK,IAAI,QAAQ,IAAI,0BAA0B;EACrD,IAAI,IAAI,UAAU,QAAQ,YAAY,EAAE;EACxC,IAAI,CAAC,IAAI,MAAM,OAAO;EAEtB,MAAM,eAAe,UAAU,iBAAiB;EAGhD,MAAM,UAAU,IAAI,YAAY;EAChC,IAAI,cAAc;EAClB,MAAM,MAAM,IAAI,gBAAwC,EACtD,UAAU,OAAO,YAAY;GAC3B,WAAW,QAAQ,KAAK;GACxB,eAAe,QAAQ,OAAO,OAAO,EAAE,QAAQ,KAAK,CAAC;GACrD,IAAI,aAAa,OAAO,WAAW,GAAG;IACpC,UAAU,QAAQ,MAAM;IACxB,cAAc;GAChB,OAAO,IAAI,YAAY,SAAS,qBAC9B,cAAc,YAAY,MAAM,KAAmB;EAEvD,EACF,CAAC;EAED,OAAO,IAAI,SAAS,IAAI,KAAK,YAAY,GAAG,GAAG;GAC7C,QAAQ,IAAI;GACZ,YAAY,IAAI;GAChB,SAAS,IAAI;EACf,CAAC;CACH;AACF;AAEA,SAAS,qBAAqB,QAAoB,aAAqB;CACrE,OAAO,YAAY,SAAS,aAAa;AAC3C;AAEA,SAAS,qBACP,WACA,sBAGA;CACA,OAAO,OAAO,YAAY;EACxB,MAAM,WAAW,UAAU,QAAQ,YAAY;EAC/C,IAAI,CAAC,UACH,MAAM,IAAI,MACR,6EACF;EAEF,MAAM,MACJ,OAAO,UAAU,cAAc,aAC3B,UAAU,UAAU,QAAQ,IAC5B,UAAU;EAChB,MAAM,eAAe,MAAM,uBAAuB;GAAE,GAAG;GAAS;EAAI,CAAC;EACrE,OAAO;GACL,GAAG;GACH,KAAK,cAAc,OAAO;EAC5B;CACF;AACF"}

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

@@ -1,35 +1,45 @@
-import type { UIMessage, useChat, CreateUIMessage } from "@ai-sdk/react";
-import type { ExternalStoreAdapter, ThreadHistoryAdapter, AssistantRuntime, ThreadSuggestion, AppendMessage } from "@assistant-ui/core";
-export type CustomToCreateMessageFunction = <UI_MESSAGE extends UIMessage = UIMessage>(message: AppendMessage) => CreateUIMessage<UI_MESSAGE>;
-export type AISDKRuntimeAdapter = {
-    adapters?: (NonNullable<ExternalStoreAdapter["adapters"]> & {
-        history?: ThreadHistoryAdapter | undefined;
-    }) | undefined;
-    toCreateMessage?: CustomToCreateMessageFunction;
-    /**
-     * Whether to automatically cancel pending interactive tool calls when the user sends a new message.
-     *
-     * When enabled (default), the pending tool calls will be marked as failed with an error message
-     * indicating the user cancelled the tool call by sending a new message.
-     *
-     * @default true
-     */
-    cancelPendingToolCallsOnSend?: boolean | undefined;
-    /**
-     * Called when `runtime.thread.resumeRun(config)` is invoked.
-     *
-     * When omitted, `resumeRun` throws `"Runtime does not support resuming runs."`.
-     * Provide this to bridge resume invocations into a custom replay channel
-     * (for example, an SSE reconnect endpoint keyed by turn id).
-     */
-    onResume?: ExternalStoreAdapter["onResume"];
-    /**
-     * Follow up suggestions to surface on the thread. Use this to drive
-     * dynamic suggestions from application state, tool results, or backend
-     * responses; flows into `thread.suggestions` and is rendered by
-     * components that read it (such as the shadcn `ThreadFollowupSuggestions`).
-     */
-    suggestions?: readonly ThreadSuggestion[] | undefined;
+import { AppendMessage, AssistantRuntime, ExternalStoreAdapter, ThreadHistoryAdapter, ThreadSuggestion } from "@assistant-ui/core";
+import { CreateUIMessage, UIMessage, useChat } from "@ai-sdk/react";
+
+//#region src/ui/use-chat/useAISDKRuntime.d.ts
+type CustomToCreateMessageFunction = <UI_MESSAGE extends UIMessage = UIMessage>(message: AppendMessage) => CreateUIMessage<UI_MESSAGE>;
+type AISDKRuntimeAdapter = {
+  adapters?: (NonNullable<ExternalStoreAdapter["adapters"]> & {
+    history?: ThreadHistoryAdapter | undefined;
+  }) | undefined;
+  toCreateMessage?: CustomToCreateMessageFunction;
+  /**
+   * Whether to automatically cancel pending interactive tool calls when the user sends a new message.
+   *
+   * When enabled (default), the pending tool calls will be marked as failed with an error message
+   * indicating the user cancelled the tool call by sending a new message.
+   *
+   * @default true
+   */
+  cancelPendingToolCallsOnSend?: boolean | undefined;
+  /**
+   * Called when `runtime.thread.resumeRun(config)` is invoked.
+   *
+   * When omitted, `resumeRun` throws `"Runtime does not support resuming runs."`.
+   * Provide this to bridge resume invocations into a custom replay channel
+   * (for example, an SSE reconnect endpoint keyed by turn id).
+   */
+  onResume?: ExternalStoreAdapter["onResume"];
+  /**
+   * Follow up suggestions to surface on the thread. Use this to drive
+   * dynamic suggestions from application state, tool results, or backend
+   * responses; flows into `thread.suggestions` and is rendered by
+   * components that read it (such as the shadcn `ThreadFollowupSuggestions`).
+   */
+  suggestions?: readonly ThreadSuggestion[] | undefined;
 };
-export declare const useAISDKRuntime: <UI_MESSAGE extends UIMessage = UIMessage>(chatHelpers: ReturnType<typeof useChat<UI_MESSAGE>>, { adapters, toCreateMessage: customToCreateMessage, cancelPendingToolCallsOnSend, onResume, suggestions, }?: AISDKRuntimeAdapter) => AssistantRuntime;
+declare const useAISDKRuntime: <UI_MESSAGE extends UIMessage = UIMessage>(chatHelpers: ReturnType<typeof useChat<UI_MESSAGE>>, {
+  adapters,
+  toCreateMessage: customToCreateMessage,
+  cancelPendingToolCallsOnSend,
+  onResume,
+  suggestions
+}?: AISDKRuntimeAdapter) => AssistantRuntime;
+//#endregion
+export { AISDKRuntimeAdapter, CustomToCreateMessageFunction, useAISDKRuntime };
 //# sourceMappingURL=useAISDKRuntime.d.ts.map

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

@@ -1 +1 @@
-{"version":3,"file":"useAISDKRuntime.d.ts","sourceRoot":"","sources":["../../../src/ui/use-chat/useAISDKRuntime.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,SAAS,EAAE,OAAO,EAAE,eAAe,EAAE,MAAM,eAAe,CAAC;AAQzE,OAAO,KAAK,EACV,oBAAoB,EACpB,oBAAoB,EACpB,gBAAgB,EAEhB,gBAAgB,EAIhB,aAAa,EAGd,MAAM,oBAAoB,CAAC;AAmB5B,MAAM,MAAM,6BAA6B,GAAG,CAC1C,UAAU,SAAS,SAAS,GAAG,SAAS,EAExC,OAAO,EAAE,aAAa,KACnB,eAAe,CAAC,UAAU,CAAC,CAAC;AAYjC,MAAM,MAAM,mBAAmB,GAAG;IAChC,QAAQ,CAAC,EACL,CAAC,WAAW,CAAC,oBAAoB,CAAC,UAAU,CAAC,CAAC,GAAG;QAC/C,OAAO,CAAC,EAAE,oBAAoB,GAAG,SAAS,CAAC;KAC5C,CAAC,GACF,SAAS,CAAC;IACd,eAAe,CAAC,EAAE,6BAA6B,CAAC;IAChD;;;;;;;OAOG;IACH,4BAA4B,CAAC,EAAE,OAAO,GAAG,SAAS,CAAC;IACnD;;;;;;OAMG;IACH,QAAQ,CAAC,EAAE,oBAAoB,CAAC,UAAU,CAAC,CAAC;IAC5C;;;;;OAKG;IACH,WAAW,CAAC,EAAE,SAAS,gBAAgB,EAAE,GAAG,SAAS,CAAC;CACvD,CAAC;AAEF,eAAO,MAAM,eAAe,GAAI,UAAU,SAAS,SAAS,GAAG,SAAS,EACtE,aAAa,UAAU,CAAC,OAAO,OAAO,CAAC,UAAU,CAAC,CAAC,EACnD,6GAMG,mBAAwB,qBAwQ5B,CAAC"}
+{"version":3,"file":"useAISDKRuntime.d.ts","names":[],"sources":["../../../src/ui/use-chat/useAISDKRuntime.ts"],"mappings":";;;;KA0CY,6BAAA,uBACS,SAAA,GAAY,SAAA,EAE/B,OAAA,EAAS,aAAA,KACN,eAAA,CAAgB,UAAA;AAAA,KAYT,mBAAA;EACV,QAAA,IACK,WAAA,CAAY,oBAAA;IACX,OAAA,GAAU,oBAAA;EAAA;EAGhB,eAAA,GAAkB,6BAAA;EArBa;;;;;;;;EA8B/B,4BAAA;EA5BS;;;;;AACoB;AAY/B;EAuBE,QAAA,GAAW,oBAAA;;;;;;;EAOX,WAAA,YAAuB,gBAAA;AAAA;AAAA,cAGZ,eAAA,sBAAsC,SAAA,GAAY,SAAA,EAC7D,WAAA,EAAa,UAAA,QAAkB,OAAA,CAAQ,UAAA;EACvC,QAAA;EAAA,eAAA,EAAA,qBAAA;EAAA,4BAAA;EAAA,QAAA;EAAA;AAAA,IAMG,mBAAA,KAAwB,gBAAA"}