@ai-sdk/react 3.0.306 → 3.0.308
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/CHANGELOG.md +23 -0
- package/dist/index.d.mts +1 -1
- package/dist/index.d.ts +1 -1
- package/dist/index.js +55 -2
- package/dist/index.js.map +1 -1
- package/dist/index.mjs +55 -2
- package/dist/index.mjs.map +1 -1
- package/package.json +3 -3
- package/src/use-chat.ts +101 -3
package/CHANGELOG.md
CHANGED
|
@@ -1,5 +1,28 @@
|
|
|
1
1
|
# @ai-sdk/react
|
|
2
2
|
|
|
3
|
+
## 3.0.308
|
|
4
|
+
|
|
5
|
+
### Patch Changes
|
|
6
|
+
|
|
7
|
+
- Updated dependencies [604b4d3]
|
|
8
|
+
- @ai-sdk/provider-utils@4.0.58
|
|
9
|
+
- ai@6.0.305
|
|
10
|
+
|
|
11
|
+
## 3.0.307
|
|
12
|
+
|
|
13
|
+
### Patch Changes
|
|
14
|
+
|
|
15
|
+
- fade070: fix: resume interrupted chat streams when the page becomes visible again
|
|
16
|
+
|
|
17
|
+
`useChat` with `resume: true` no longer starts a second resume request on mount when the chat is already submitted or streaming, e.g. when remounting with a shared `Chat` instance mid-stream.
|
|
18
|
+
|
|
19
|
+
- Updated dependencies [fade070]
|
|
20
|
+
- Updated dependencies [8e74de0]
|
|
21
|
+
- Updated dependencies [a5c9cca]
|
|
22
|
+
- Updated dependencies [7888841]
|
|
23
|
+
- Updated dependencies [e466fb0]
|
|
24
|
+
- ai@6.0.304
|
|
25
|
+
|
|
3
26
|
## 3.0.306
|
|
4
27
|
|
|
5
28
|
### Patch Changes
|
package/dist/index.d.mts
CHANGED
|
@@ -32,7 +32,7 @@ type UseChatOptions<UI_MESSAGE extends UIMessage> = ({
|
|
|
32
32
|
*/
|
|
33
33
|
experimental_throttle?: number;
|
|
34
34
|
/**
|
|
35
|
-
* Whether to resume an ongoing chat generation stream.
|
|
35
|
+
* Whether to automatically resume an ongoing chat generation stream.
|
|
36
36
|
*/
|
|
37
37
|
resume?: boolean;
|
|
38
38
|
};
|
package/dist/index.d.ts
CHANGED
|
@@ -32,7 +32,7 @@ type UseChatOptions<UI_MESSAGE extends UIMessage> = ({
|
|
|
32
32
|
*/
|
|
33
33
|
experimental_throttle?: number;
|
|
34
34
|
/**
|
|
35
|
-
* Whether to resume an ongoing chat generation stream.
|
|
35
|
+
* Whether to automatically resume an ongoing chat generation stream.
|
|
36
36
|
*/
|
|
37
37
|
resume?: boolean;
|
|
38
38
|
};
|
package/dist/index.js
CHANGED
|
@@ -193,11 +193,61 @@ var Chat = class extends import_ai.AbstractChat {
|
|
|
193
193
|
_state = new WeakMap();
|
|
194
194
|
|
|
195
195
|
// src/use-chat.ts
|
|
196
|
+
var automaticResumeStates = /* @__PURE__ */ new WeakMap();
|
|
197
|
+
function resumeOnVisible({
|
|
198
|
+
chat,
|
|
199
|
+
state
|
|
200
|
+
}) {
|
|
201
|
+
if (document.visibilityState !== "visible" || state.pendingVisibilityResume != null) {
|
|
202
|
+
return;
|
|
203
|
+
}
|
|
204
|
+
const isRegistered = () => automaticResumeStates.get(chat) === state && state.registrations.size > 0;
|
|
205
|
+
const clearPendingResume = () => {
|
|
206
|
+
state.pendingVisibilityResume = void 0;
|
|
207
|
+
};
|
|
208
|
+
const resumableChat = chat;
|
|
209
|
+
state.pendingVisibilityResume = resumableChat["~resumeStreamIfDisconnected"]({
|
|
210
|
+
shouldResume: () => isRegistered() && document.visibilityState === "visible"
|
|
211
|
+
}).then(clearPendingResume, clearPendingResume);
|
|
212
|
+
}
|
|
213
|
+
function registerAutomaticResume({
|
|
214
|
+
chat,
|
|
215
|
+
registration
|
|
216
|
+
}) {
|
|
217
|
+
let state = automaticResumeStates.get(chat);
|
|
218
|
+
if (state == null) {
|
|
219
|
+
const newState = { registrations: /* @__PURE__ */ new Set() };
|
|
220
|
+
automaticResumeStates.set(chat, newState);
|
|
221
|
+
state = newState;
|
|
222
|
+
if (typeof document !== "undefined") {
|
|
223
|
+
const onVisibilityChange = () => resumeOnVisible({ chat, state: newState });
|
|
224
|
+
document.addEventListener("visibilitychange", onVisibilityChange);
|
|
225
|
+
newState.cleanupVisibilityListener = () => {
|
|
226
|
+
document.removeEventListener("visibilitychange", onVisibilityChange);
|
|
227
|
+
};
|
|
228
|
+
}
|
|
229
|
+
}
|
|
230
|
+
const { registrations } = state;
|
|
231
|
+
const shouldResume = registrations.size === 0;
|
|
232
|
+
registrations.add(registration);
|
|
233
|
+
if (shouldResume && chat.status !== "submitted" && chat.status !== "streaming") {
|
|
234
|
+
void chat.resumeStream();
|
|
235
|
+
}
|
|
236
|
+
return () => {
|
|
237
|
+
var _a;
|
|
238
|
+
registrations.delete(registration);
|
|
239
|
+
if (registrations.size === 0) {
|
|
240
|
+
(_a = state.cleanupVisibilityListener) == null ? void 0 : _a.call(state);
|
|
241
|
+
automaticResumeStates.delete(chat);
|
|
242
|
+
}
|
|
243
|
+
};
|
|
244
|
+
}
|
|
196
245
|
function useChat({
|
|
197
246
|
experimental_throttle: throttleWaitMs,
|
|
198
247
|
resume = false,
|
|
199
248
|
...options
|
|
200
249
|
} = {}) {
|
|
250
|
+
const automaticResumeRegistration = (0, import_react.useRef)({});
|
|
201
251
|
const latestRef = (0, import_react.useRef)({});
|
|
202
252
|
if (!("chat" in options)) {
|
|
203
253
|
latestRef.current = {
|
|
@@ -321,9 +371,12 @@ function useChat({
|
|
|
321
371
|
);
|
|
322
372
|
(0, import_react.useEffect)(() => {
|
|
323
373
|
if (resume) {
|
|
324
|
-
|
|
374
|
+
return registerAutomaticResume({
|
|
375
|
+
chat,
|
|
376
|
+
registration: automaticResumeRegistration.current
|
|
377
|
+
});
|
|
325
378
|
}
|
|
326
|
-
}, [resume,
|
|
379
|
+
}, [resume, chat]);
|
|
327
380
|
return {
|
|
328
381
|
id: chatRef.current.id,
|
|
329
382
|
messages,
|
package/dist/index.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"sources":["../src/index.ts","../src/use-chat.ts","../src/chat.react.ts","../src/throttle.ts","../src/use-completion.ts","../src/use-object.ts"],"sourcesContent":["export * from './use-chat';\nexport { Chat } from './chat.react';\nexport * from './use-completion';\nexport * from './use-object';\n","import {\n type AbstractChat,\n type ChatInit,\n type ChatTransport,\n type CreateUIMessage,\n type UIMessage,\n DefaultChatTransport,\n} from 'ai';\nimport { useCallback, useEffect, useRef, useSyncExternalStore } from 'react';\nimport { Chat } from './chat.react';\n\nexport type { CreateUIMessage, UIMessage };\n\nexport type UseChatHelpers<UI_MESSAGE extends UIMessage> = {\n /**\n * The id of the chat.\n */\n readonly id: string;\n\n /**\n * Update the `messages` state locally. This is useful when you want to\n * edit the messages on the client, and then trigger the `reload` method\n * manually to regenerate the AI response.\n */\n setMessages: (\n messages: UI_MESSAGE[] | ((messages: UI_MESSAGE[]) => UI_MESSAGE[]),\n ) => void;\n\n error: Error | undefined;\n} & Pick<\n AbstractChat<UI_MESSAGE>,\n | 'sendMessage'\n | 'regenerate'\n | 'stop'\n | 'resumeStream'\n | 'addToolResult'\n | 'addToolOutput'\n | 'addToolApprovalResponse'\n | 'status'\n | 'messages'\n | 'clearError'\n>;\n\nexport type UseChatOptions<UI_MESSAGE extends UIMessage> = (\n | { chat: Chat<UI_MESSAGE> }\n | ChatInit<UI_MESSAGE>\n) & {\n /**\n * Custom throttle wait in ms for the chat messages and data updates.\n * Default is undefined, which disables throttling.\n */\n experimental_throttle?: number;\n\n /**\n * Whether to resume an ongoing chat generation stream.\n */\n resume?: boolean;\n};\n\nexport function useChat<UI_MESSAGE extends UIMessage = UIMessage>({\n experimental_throttle: throttleWaitMs,\n resume = false,\n ...options\n}: UseChatOptions<UI_MESSAGE> = {}): UseChatHelpers<UI_MESSAGE> {\n // the Chat instance is created once and not recreated when options change,\n // so it would normally keep the callbacks/transport from the first render forever\n\n // keep latest values in a ref that is refreshed on every render,\n // and hand `Chat` stable wrappers that read from it to avoid stale closures\n const latestRef = useRef<\n Partial<\n Pick<\n ChatInit<UI_MESSAGE>,\n | 'onToolCall'\n | 'onData'\n | 'onFinish'\n | 'onError'\n | 'sendAutomaticallyWhen'\n | 'transport'\n >\n >\n >({});\n\n if (!('chat' in options)) {\n latestRef.current = {\n onToolCall: options.onToolCall,\n onData: options.onData,\n onFinish: options.onFinish,\n onError: options.onError,\n sendAutomaticallyWhen: options.sendAutomaticallyWhen,\n transport: options.transport,\n };\n }\n\n // resolve the latest transport and fallback to a lazily created default transport\n let defaultTransport: ChatTransport<UI_MESSAGE> | undefined;\n const getTransport = () =>\n latestRef.current.transport ??\n (defaultTransport ??= new DefaultChatTransport<UI_MESSAGE>());\n\n // give `Chat` stable wrappers that always read the latest values from `latestRef`\n const chatOptions: typeof options = {\n ...options,\n transport: {\n sendMessages: sendOptions => getTransport().sendMessages(sendOptions),\n reconnectToStream: reconnectOptions =>\n getTransport().reconnectToStream(reconnectOptions),\n },\n onToolCall: arg => latestRef.current.onToolCall?.(arg),\n onData: arg => latestRef.current.onData?.(arg),\n onFinish: arg => latestRef.current.onFinish?.(arg),\n onError: arg => latestRef.current.onError?.(arg),\n sendAutomaticallyWhen: arg =>\n latestRef.current.sendAutomaticallyWhen?.(arg) ?? false,\n };\n\n const chatRef = useRef<Chat<UI_MESSAGE>>(\n 'chat' in options ? options.chat : new Chat(chatOptions),\n );\n\n const shouldRecreateChat =\n ('chat' in options && options.chat !== chatRef.current) ||\n ('id' in options &&\n options.id != null &&\n chatRef.current.id !== options.id);\n\n if (shouldRecreateChat) {\n chatRef.current = 'chat' in options ? options.chat : new Chat(chatOptions);\n }\n\n const chat = chatRef.current;\n const messagesSnapshotRef = useRef({\n chat,\n messages: chat.messages,\n });\n\n if (messagesSnapshotRef.current.chat !== chat) {\n messagesSnapshotRef.current = { chat, messages: chat.messages };\n }\n\n const subscribeToMessages = useCallback(\n (update: () => void) => {\n let isSubscribed = true;\n\n const updateMessages = () => {\n if (!isSubscribed || messagesSnapshotRef.current.chat !== chat) {\n return;\n }\n\n messagesSnapshotRef.current = { chat, messages: chat.messages };\n update();\n };\n\n const unsubscribe = chat['~registerMessagesCallback'](\n updateMessages,\n throttleWaitMs,\n );\n\n // Synchronize changes that may have happened between render and\n // subscription. useSyncExternalStore checks the snapshot after\n // subscribing and schedules a render when it changed.\n messagesSnapshotRef.current = { chat, messages: chat.messages };\n\n return () => {\n isSubscribed = false;\n unsubscribe();\n };\n },\n [chat, throttleWaitMs],\n );\n\n const getMessagesSnapshot = useCallback(\n () => messagesSnapshotRef.current.messages,\n [],\n );\n\n const messages = useSyncExternalStore(\n subscribeToMessages,\n getMessagesSnapshot,\n getMessagesSnapshot,\n );\n\n const subscribeToStatus = useCallback(\n (update: () => void) =>\n chat['~registerStatusCallback'](() => {\n if (messagesSnapshotRef.current.chat !== chat) {\n return;\n }\n\n if (chat.status === 'ready' || chat.status === 'error') {\n // Publish the latest messages before the terminal status can render.\n messagesSnapshotRef.current = { chat, messages: chat.messages };\n }\n\n update();\n }),\n [chat],\n );\n\n const getStatusSnapshot = useCallback(() => chat.status, [chat]);\n\n const status = useSyncExternalStore(\n subscribeToStatus,\n getStatusSnapshot,\n getStatusSnapshot,\n );\n\n const error = useSyncExternalStore(\n chatRef.current['~registerErrorCallback'],\n () => chatRef.current.error,\n () => chatRef.current.error,\n );\n\n const setMessages = useCallback(\n (\n messagesParam: UI_MESSAGE[] | ((messages: UI_MESSAGE[]) => UI_MESSAGE[]),\n ) => {\n if (typeof messagesParam === 'function') {\n messagesParam = messagesParam(chatRef.current.messages);\n }\n chatRef.current.messages = messagesParam;\n },\n [chatRef],\n );\n\n useEffect(() => {\n if (resume) {\n chatRef.current.resumeStream();\n }\n }, [resume, chatRef]);\n\n return {\n id: chatRef.current.id,\n messages,\n setMessages,\n sendMessage: chatRef.current.sendMessage,\n regenerate: chatRef.current.regenerate,\n clearError: chatRef.current.clearError,\n stop: chatRef.current.stop,\n error,\n resumeStream: chatRef.current.resumeStream,\n status,\n /**\n * @deprecated Use `addToolOutput` instead.\n */\n addToolResult: chatRef.current.addToolOutput,\n addToolOutput: chatRef.current.addToolOutput,\n addToolApprovalResponse: chatRef.current.addToolApprovalResponse,\n };\n}\n","import {\n AbstractChat,\n type ChatInit,\n type ChatState,\n type ChatStatus,\n type UIMessage,\n} from 'ai';\nimport { throttle } from './throttle';\n\nfunction cloneMetadata<METADATA>(metadata: METADATA): METADATA {\n if (Array.isArray(metadata)) {\n return [...metadata] as METADATA;\n }\n\n if (\n metadata != null &&\n typeof metadata === 'object' &&\n (Object.getPrototypeOf(metadata) === Object.prototype ||\n Object.getPrototypeOf(metadata) === null)\n ) {\n return { ...metadata } as METADATA;\n }\n\n return metadata;\n}\n\nclass ReactChatState<\n UI_MESSAGE extends UIMessage,\n> implements ChatState<UI_MESSAGE> {\n #messages: UI_MESSAGE[];\n #status: ChatStatus = 'ready';\n #error: Error | undefined = undefined;\n\n #messagesCallbacks = new Set<() => void>();\n #statusCallbacks = new Set<() => void>();\n #errorCallbacks = new Set<() => void>();\n\n constructor(initialMessages: UI_MESSAGE[] = []) {\n this.#messages = initialMessages;\n }\n\n get status(): ChatStatus {\n return this.#status;\n }\n\n set status(newStatus: ChatStatus) {\n this.#status = newStatus;\n this.#callStatusCallbacks();\n }\n\n get error(): Error | undefined {\n return this.#error;\n }\n\n set error(newError: Error | undefined) {\n this.#error = newError;\n this.#callErrorCallbacks();\n }\n\n get messages(): UI_MESSAGE[] {\n return this.#messages;\n }\n\n set messages(newMessages: UI_MESSAGE[]) {\n this.#messages = [...newMessages];\n this.#callMessagesCallbacks();\n }\n\n pushMessage = (message: UI_MESSAGE) => {\n this.#messages = this.#messages.concat(message);\n this.#callMessagesCallbacks();\n };\n\n popMessage = () => {\n this.#messages = this.#messages.slice(0, -1);\n this.#callMessagesCallbacks();\n };\n\n replaceMessage = (index: number, message: UI_MESSAGE) => {\n this.#messages = [\n ...this.#messages.slice(0, index),\n this.snapshot(message),\n ...this.#messages.slice(index + 1),\n ];\n this.#callMessagesCallbacks();\n };\n\n snapshot = <T>(value: T): T => {\n if (\n value == null ||\n typeof value !== 'object' ||\n !('parts' in value) ||\n !Array.isArray(value.parts)\n ) {\n return value;\n }\n\n const message = value as UI_MESSAGE;\n const snapshot = {\n ...message,\n parts: message.parts.map(part => ({ ...part })),\n };\n\n if ('metadata' in message) {\n snapshot.metadata = cloneMetadata(message.metadata);\n }\n\n return snapshot as T;\n };\n\n '~registerMessagesCallback' = (\n onChange: () => void,\n throttleWaitMs?: number,\n ): (() => void) => {\n const callback = throttleWaitMs\n ? throttle(onChange, throttleWaitMs)\n : onChange;\n this.#messagesCallbacks.add(callback);\n return () => {\n this.#messagesCallbacks.delete(callback);\n };\n };\n\n '~registerStatusCallback' = (onChange: () => void): (() => void) => {\n this.#statusCallbacks.add(onChange);\n return () => {\n this.#statusCallbacks.delete(onChange);\n };\n };\n\n '~registerErrorCallback' = (onChange: () => void): (() => void) => {\n this.#errorCallbacks.add(onChange);\n return () => {\n this.#errorCallbacks.delete(onChange);\n };\n };\n\n #callMessagesCallbacks = () => {\n this.#messagesCallbacks.forEach(callback => callback());\n };\n\n #callStatusCallbacks = () => {\n this.#statusCallbacks.forEach(callback => callback());\n };\n\n #callErrorCallbacks = () => {\n this.#errorCallbacks.forEach(callback => callback());\n };\n}\n\nexport class Chat<\n UI_MESSAGE extends UIMessage,\n> extends AbstractChat<UI_MESSAGE> {\n #state: ReactChatState<UI_MESSAGE>;\n\n constructor({ messages, ...init }: ChatInit<UI_MESSAGE>) {\n const state = new ReactChatState(messages);\n super({ ...init, state });\n this.#state = state;\n }\n\n '~registerMessagesCallback' = (\n onChange: () => void,\n throttleWaitMs?: number,\n ): (() => void) =>\n this.#state['~registerMessagesCallback'](onChange, throttleWaitMs);\n\n '~registerStatusCallback' = (onChange: () => void): (() => void) =>\n this.#state['~registerStatusCallback'](onChange);\n\n '~registerErrorCallback' = (onChange: () => void): (() => void) =>\n this.#state['~registerErrorCallback'](onChange);\n}\n","import throttleFunction from 'throttleit';\n\nexport function throttle<T extends (...args: any[]) => any>(\n fn: T,\n waitMs: number | undefined,\n): T {\n return waitMs != null ? throttleFunction(fn, waitMs) : fn;\n}\n","import {\n callCompletionApi,\n type CompletionRequestOptions,\n type UseCompletionOptions,\n} from 'ai';\nimport { useCallback, useEffect, useId, useRef, useState } from 'react';\nimport useSWR from 'swr';\nimport { throttle } from './throttle';\n\nexport type { UseCompletionOptions };\n\nexport type UseCompletionHelpers = {\n /** The current completion result */\n completion: string;\n /**\n * Send a new prompt to the API endpoint and update the completion state.\n */\n complete: (\n prompt: string,\n options?: CompletionRequestOptions,\n ) => Promise<string | null | undefined>;\n /** The error object of the API request */\n error: undefined | Error;\n /**\n * Abort the current API request but keep the generated tokens.\n */\n stop: () => void;\n /**\n * Update the `completion` state locally.\n */\n setCompletion: (completion: string) => void;\n /** The current value of the input */\n input: string;\n /** setState-powered method to update the input value */\n setInput: React.Dispatch<React.SetStateAction<string>>;\n /**\n * An input/textarea-ready onChange handler to control the value of the input\n * @example\n * ```jsx\n * <input onChange={handleInputChange} value={input} />\n * ```\n */\n handleInputChange: (\n event:\n | React.ChangeEvent<HTMLInputElement>\n | React.ChangeEvent<HTMLTextAreaElement>,\n ) => void;\n\n /**\n * Form submission handler to automatically reset input and append a user message\n * @example\n * ```jsx\n * <form onSubmit={handleSubmit}>\n * <input onChange={handleInputChange} value={input} />\n * </form>\n * ```\n */\n handleSubmit: (event?: { preventDefault?: () => void }) => void;\n\n /** Whether the API request is in progress */\n isLoading: boolean;\n};\n\nexport function useCompletion({\n api = '/api/completion',\n id,\n initialCompletion = '',\n initialInput = '',\n credentials,\n headers,\n body,\n streamProtocol = 'data',\n fetch,\n onFinish,\n onError,\n experimental_throttle: throttleWaitMs,\n}: UseCompletionOptions & {\n /**\n * Custom throttle wait in ms for the completion and data updates.\n * Default is undefined, which disables throttling.\n */\n experimental_throttle?: number;\n} = {}): UseCompletionHelpers {\n // Generate an unique id for the completion if not provided.\n const hookId = useId();\n const completionId = id || hookId;\n\n // Store the completion state in SWR, using the completionId as the key to share states.\n const { data, mutate } = useSWR<string>([api, completionId], null, {\n fallbackData: initialCompletion,\n });\n\n const { data: isLoading = false, mutate: mutateLoading } = useSWR<boolean>(\n [completionId, 'loading'],\n null,\n );\n\n const [error, setError] = useState<undefined | Error>(undefined);\n const completion = data!;\n\n // Abort controller to cancel the current API call.\n const [abortController, setAbortController] =\n useState<AbortController | null>(null);\n\n const extraMetadataRef = useRef({\n credentials,\n headers,\n body,\n });\n\n useEffect(() => {\n extraMetadataRef.current = {\n credentials,\n headers,\n body,\n };\n }, [credentials, headers, body]);\n\n const triggerRequest = useCallback(\n async (prompt: string, options?: CompletionRequestOptions) =>\n callCompletionApi({\n api,\n prompt,\n credentials: extraMetadataRef.current.credentials,\n headers: { ...extraMetadataRef.current.headers, ...options?.headers },\n body: {\n ...extraMetadataRef.current.body,\n ...options?.body,\n },\n streamProtocol,\n fetch,\n // throttle streamed ui updates:\n setCompletion: throttle(\n (completion: string) => mutate(completion, false),\n throttleWaitMs,\n ),\n setLoading: mutateLoading,\n setError,\n setAbortController,\n onFinish,\n onError,\n }),\n [\n mutate,\n mutateLoading,\n api,\n extraMetadataRef,\n setAbortController,\n onFinish,\n onError,\n setError,\n streamProtocol,\n fetch,\n throttleWaitMs,\n ],\n );\n\n const stop = useCallback(() => {\n if (abortController) {\n abortController.abort();\n setAbortController(null);\n }\n }, [abortController]);\n\n const setCompletion = useCallback(\n (completion: string) => {\n mutate(completion, false);\n },\n [mutate],\n );\n\n const complete = useCallback<UseCompletionHelpers['complete']>(\n async (prompt, options) => {\n return triggerRequest(prompt, options);\n },\n [triggerRequest],\n );\n\n const [input, setInput] = useState(initialInput);\n\n const handleSubmit = useCallback(\n (event?: { preventDefault?: () => void }) => {\n event?.preventDefault?.();\n if (!input) {\n return;\n }\n\n const result = complete(input);\n setInput('');\n return result;\n },\n [input, complete, setInput],\n );\n\n const handleInputChange = useCallback(\n (e: any) => {\n setInput(e.target.value);\n },\n [setInput],\n );\n\n return {\n completion,\n complete,\n error,\n setCompletion,\n stop,\n input,\n setInput,\n handleInputChange,\n handleSubmit,\n isLoading,\n };\n}\n","import {\n isAbortError,\n resolve,\n normalizeHeaders,\n safeValidateTypes,\n type FetchFunction,\n type FlexibleSchema,\n type InferSchema,\n type Resolvable,\n} from '@ai-sdk/provider-utils';\nimport {\n asSchema,\n isDeepEqualData,\n parsePartialJson,\n type DeepPartial,\n} from 'ai';\nimport { useCallback, useId, useRef, useState } from 'react';\nimport useSWR from 'swr';\n\n// use function to allow for mocking in tests:\nconst getOriginalFetch = () => fetch;\n\nexport type Experimental_UseObjectOptions<\n SCHEMA extends FlexibleSchema,\n RESULT,\n> = {\n /**\n * The API endpoint. It should stream JSON that matches the schema as chunked text.\n */\n api: string;\n\n /**\n * A schema that defines the shape of the complete object.\n */\n schema: SCHEMA;\n\n /**\n * An unique identifier. If not provided, a random one will be\n * generated. When provided, the `useObject` hook with the same `id` will\n * have shared states across components.\n */\n id?: string;\n\n /**\n * An optional value for the initial object.\n */\n initialValue?: DeepPartial<RESULT>;\n\n /**\n * Custom fetch implementation. You can use it as a middleware to intercept requests,\n * or to provide a custom fetch implementation for e.g. testing.\n */\n fetch?: FetchFunction;\n\n /**\n * Callback that is called when the stream has finished.\n */\n onFinish?: (event: {\n /**\n * The generated object (typed according to the schema).\n * Can be undefined if the final object does not match the schema.\n */\n object: RESULT | undefined;\n\n /**\n * Optional error object. This is e.g. a TypeValidationError when the final object does not match the schema.\n */\n error: Error | undefined;\n }) => Promise<void> | void;\n\n /**\n * Callback function to be called when an error is encountered.\n */\n onError?: (error: Error) => void;\n\n /**\n * Additional HTTP headers to be included in the request.\n * Can be a static object, a function that returns headers, or an async function\n * for dynamic auth tokens.\n */\n headers?: Resolvable<Record<string, string> | Headers>;\n\n /**\n * The credentials mode to be used for the fetch request.\n * Possible values are: 'omit', 'same-origin', 'include'.\n * Defaults to 'same-origin'.\n */\n credentials?: RequestCredentials;\n};\n\nexport type Experimental_UseObjectHelpers<RESULT, INPUT> = {\n /**\n * Calls the API with the provided input as JSON body.\n */\n submit: (input: INPUT) => void;\n\n /**\n * The current value for the generated object. Updated as the API streams JSON chunks.\n */\n object: DeepPartial<RESULT> | undefined;\n\n /**\n * The error object of the API request if any.\n */\n error: Error | undefined;\n\n /**\n * Flag that indicates whether an API request is in progress.\n */\n isLoading: boolean;\n\n /**\n * Abort the current request immediately, keep the current partial object if any.\n */\n stop: () => void;\n\n /**\n * Clear the object state.\n */\n clear: () => void;\n};\n\nfunction useObject<\n SCHEMA extends FlexibleSchema,\n RESULT = InferSchema<SCHEMA>,\n INPUT = any,\n>({\n api,\n id,\n schema, // required, in the future we will use it for validation\n initialValue,\n fetch,\n onError,\n onFinish,\n headers,\n credentials,\n}: Experimental_UseObjectOptions<\n SCHEMA,\n RESULT\n>): Experimental_UseObjectHelpers<RESULT, INPUT> {\n // Generate an unique id if not provided.\n const hookId = useId();\n const completionId = id ?? hookId;\n\n // Store the completion state in SWR, using the completionId as the key to share states.\n const { data, mutate } = useSWR<DeepPartial<RESULT>>(\n [completionId, 'object'],\n null,\n { fallbackData: initialValue },\n );\n\n const [error, setError] = useState<undefined | Error>(undefined);\n const [isLoading, setIsLoading] = useState(false);\n\n // Abort controller to cancel the current API call.\n const abortControllerRef = useRef<AbortController | null>(null);\n\n const stop = useCallback(() => {\n try {\n abortControllerRef.current?.abort();\n } catch (ignored) {\n } finally {\n setIsLoading(false);\n abortControllerRef.current = null;\n }\n }, []);\n\n const submit = async (input: INPUT) => {\n const abortController = new AbortController();\n\n try {\n clearObject();\n\n setIsLoading(true);\n\n abortControllerRef.current = abortController;\n\n // Resolve headers at request time (supports async functions for dynamic auth tokens)\n const resolvedHeaders = await resolve(headers);\n\n const actualFetch = fetch ?? getOriginalFetch();\n const response = await actualFetch(api, {\n method: 'POST',\n headers: {\n 'Content-Type': 'application/json',\n ...normalizeHeaders(resolvedHeaders),\n },\n credentials,\n signal: abortController.signal,\n body: JSON.stringify(input),\n });\n\n if (!response.ok) {\n throw new Error(\n (await response.text()) ?? 'Failed to fetch the response.',\n );\n }\n\n if (response.body == null) {\n throw new Error('The response body is empty.');\n }\n\n let accumulatedText = '';\n let latestObject: DeepPartial<RESULT> | undefined = undefined;\n\n await response.body.pipeThrough(new TextDecoderStream()).pipeTo(\n new WritableStream<string>({\n async write(chunk) {\n accumulatedText += chunk;\n\n const { value } = await parsePartialJson(accumulatedText);\n const currentObject = value as DeepPartial<RESULT>;\n\n if (!isDeepEqualData(latestObject, currentObject)) {\n latestObject = currentObject;\n\n if (abortControllerRef.current === abortController) {\n mutate(currentObject);\n }\n }\n },\n\n async close() {\n if (abortControllerRef.current === abortController) {\n setIsLoading(false);\n }\n\n if (onFinish != null) {\n const validationResult = await safeValidateTypes({\n value: latestObject,\n schema: asSchema(schema),\n });\n\n onFinish(\n validationResult.success\n ? { object: validationResult.value, error: undefined }\n : { object: undefined, error: validationResult.error },\n );\n }\n\n if (abortControllerRef.current === abortController) {\n abortControllerRef.current = null;\n }\n },\n }),\n );\n } catch (error) {\n if (isAbortError(error)) {\n return;\n }\n\n if (onError && error instanceof Error) {\n onError(error);\n }\n\n if (abortControllerRef.current === abortController) {\n setIsLoading(false);\n abortControllerRef.current = null;\n setError(error instanceof Error ? error : new Error(String(error)));\n }\n }\n };\n\n const clear = () => {\n stop();\n clearObject();\n };\n\n const clearObject = () => {\n setError(undefined);\n setIsLoading(false);\n mutate(undefined);\n };\n\n return {\n submit,\n object: data,\n error,\n isLoading,\n stop,\n clear,\n };\n}\n\nexport const experimental_useObject = useObject;\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;;;ACAA,IAAAA,aAOO;AACP,mBAAqE;;;ACRrE,gBAMO;;;ACNP,wBAA6B;AAEtB,SAAS,SACd,IACA,QACG;AACH,SAAO,UAAU,WAAO,kBAAAC,SAAiB,IAAI,MAAM,IAAI;AACzD;;;ADEA,SAAS,cAAwB,UAA8B;AAC7D,MAAI,MAAM,QAAQ,QAAQ,GAAG;AAC3B,WAAO,CAAC,GAAG,QAAQ;AAAA,EACrB;AAEA,MACE,YAAY,QACZ,OAAO,aAAa,aACnB,OAAO,eAAe,QAAQ,MAAM,OAAO,aAC1C,OAAO,eAAe,QAAQ,MAAM,OACtC;AACA,WAAO,EAAE,GAAG,SAAS;AAAA,EACvB;AAEA,SAAO;AACT;AAxBA;AA0BA,IAAM,iBAAN,MAEmC;AAAA,EASjC,YAAY,kBAAgC,CAAC,GAAG;AARhD;AACA,gCAAsB;AACtB,+BAA4B;AAE5B,2CAAqB,oBAAI,IAAgB;AACzC,yCAAmB,oBAAI,IAAgB;AACvC,wCAAkB,oBAAI,IAAgB;AAiCtC,uBAAc,CAAC,YAAwB;AACrC,yBAAK,WAAY,mBAAK,WAAU,OAAO,OAAO;AAC9C,yBAAK,wBAAL;AAAA,IACF;AAEA,sBAAa,MAAM;AACjB,yBAAK,WAAY,mBAAK,WAAU,MAAM,GAAG,EAAE;AAC3C,yBAAK,wBAAL;AAAA,IACF;AAEA,0BAAiB,CAAC,OAAe,YAAwB;AACvD,yBAAK,WAAY;AAAA,QACf,GAAG,mBAAK,WAAU,MAAM,GAAG,KAAK;AAAA,QAChC,KAAK,SAAS,OAAO;AAAA,QACrB,GAAG,mBAAK,WAAU,MAAM,QAAQ,CAAC;AAAA,MACnC;AACA,yBAAK,wBAAL;AAAA,IACF;AAEA,oBAAW,CAAI,UAAgB;AAC7B,UACE,SAAS,QACT,OAAO,UAAU,YACjB,EAAE,WAAW,UACb,CAAC,MAAM,QAAQ,MAAM,KAAK,GAC1B;AACA,eAAO;AAAA,MACT;AAEA,YAAM,UAAU;AAChB,YAAM,WAAW;AAAA,QACf,GAAG;AAAA,QACH,OAAO,QAAQ,MAAM,IAAI,WAAS,EAAE,GAAG,KAAK,EAAE;AAAA,MAChD;AAEA,UAAI,cAAc,SAAS;AACzB,iBAAS,WAAW,cAAc,QAAQ,QAAQ;AAAA,MACpD;AAEA,aAAO;AAAA,IACT;AAEA,wCAA8B,CAC5B,UACA,mBACiB;AACjB,YAAM,WAAW,iBACb,SAAS,UAAU,cAAc,IACjC;AACJ,yBAAK,oBAAmB,IAAI,QAAQ;AACpC,aAAO,MAAM;AACX,2BAAK,oBAAmB,OAAO,QAAQ;AAAA,MACzC;AAAA,IACF;AAEA,sCAA4B,CAAC,aAAuC;AAClE,yBAAK,kBAAiB,IAAI,QAAQ;AAClC,aAAO,MAAM;AACX,2BAAK,kBAAiB,OAAO,QAAQ;AAAA,MACvC;AAAA,IACF;AAEA,qCAA2B,CAAC,aAAuC;AACjE,yBAAK,iBAAgB,IAAI,QAAQ;AACjC,aAAO,MAAM;AACX,2BAAK,iBAAgB,OAAO,QAAQ;AAAA,MACtC;AAAA,IACF;AAEA,+CAAyB,MAAM;AAC7B,yBAAK,oBAAmB,QAAQ,cAAY,SAAS,CAAC;AAAA,IACxD;AAEA,6CAAuB,MAAM;AAC3B,yBAAK,kBAAiB,QAAQ,cAAY,SAAS,CAAC;AAAA,IACtD;AAEA,4CAAsB,MAAM;AAC1B,yBAAK,iBAAgB,QAAQ,cAAY,SAAS,CAAC;AAAA,IACrD;AA7GE,uBAAK,WAAY;AAAA,EACnB;AAAA,EAEA,IAAI,SAAqB;AACvB,WAAO,mBAAK;AAAA,EACd;AAAA,EAEA,IAAI,OAAO,WAAuB;AAChC,uBAAK,SAAU;AACf,uBAAK,sBAAL;AAAA,EACF;AAAA,EAEA,IAAI,QAA2B;AAC7B,WAAO,mBAAK;AAAA,EACd;AAAA,EAEA,IAAI,MAAM,UAA6B;AACrC,uBAAK,QAAS;AACd,uBAAK,qBAAL;AAAA,EACF;AAAA,EAEA,IAAI,WAAyB;AAC3B,WAAO,mBAAK;AAAA,EACd;AAAA,EAEA,IAAI,SAAS,aAA2B;AACtC,uBAAK,WAAY,CAAC,GAAG,WAAW;AAChC,uBAAK,wBAAL;AAAA,EACF;AAkFF;AAvHE;AACA;AACA;AAEA;AACA;AACA;AAsGA;AAIA;AAIA;AAjJF;AAsJO,IAAM,OAAN,cAEG,uBAAyB;AAAA,EAGjC,YAAY,EAAE,UAAU,GAAG,KAAK,GAAyB;AACvD,UAAM,QAAQ,IAAI,eAAe,QAAQ;AACzC,UAAM,EAAE,GAAG,MAAM,MAAM,CAAC;AAJ1B;AAQA,wCAA8B,CAC5B,UACA,mBAEA,mBAAK,QAAO,2BAA2B,EAAE,UAAU,cAAc;AAEnE,sCAA4B,CAAC,aAC3B,mBAAK,QAAO,yBAAyB,EAAE,QAAQ;AAEjD,qCAA2B,CAAC,aAC1B,mBAAK,QAAO,wBAAwB,EAAE,QAAQ;AAb9C,uBAAK,QAAS;AAAA,EAChB;AAaF;AAnBE;;;AD9FK,SAAS,QAAkD;AAAA,EAChE,uBAAuB;AAAA,EACvB,SAAS;AAAA,EACT,GAAG;AACL,IAAgC,CAAC,GAA+B;AAM9D,QAAM,gBAAY,qBAYhB,CAAC,CAAC;AAEJ,MAAI,EAAE,UAAU,UAAU;AACxB,cAAU,UAAU;AAAA,MAClB,YAAY,QAAQ;AAAA,MACpB,QAAQ,QAAQ;AAAA,MAChB,UAAU,QAAQ;AAAA,MAClB,SAAS,QAAQ;AAAA,MACjB,uBAAuB,QAAQ;AAAA,MAC/B,WAAW,QAAQ;AAAA,IACrB;AAAA,EACF;AAGA,MAAI;AACJ,QAAM,eAAe,MAAG;AAhG1B;AAiGI,2BAAU,QAAQ,cAAlB,YACC,iEAAqB,IAAI,gCAAiC;AAAA;AAG7D,QAAM,cAA8B;AAAA,IAClC,GAAG;AAAA,IACH,WAAW;AAAA,MACT,cAAc,iBAAe,aAAa,EAAE,aAAa,WAAW;AAAA,MACpE,mBAAmB,sBACjB,aAAa,EAAE,kBAAkB,gBAAgB;AAAA,IACrD;AAAA,IACA,YAAY,SAAI;AA5GpB;AA4GuB,mCAAU,SAAQ,eAAlB,4BAA+B;AAAA;AAAA,IAClD,QAAQ,SAAI;AA7GhB;AA6GmB,mCAAU,SAAQ,WAAlB,4BAA2B;AAAA;AAAA,IAC1C,UAAU,SAAI;AA9GlB;AA8GqB,mCAAU,SAAQ,aAAlB,4BAA6B;AAAA;AAAA,IAC9C,SAAS,SAAI;AA/GjB;AA+GoB,mCAAU,SAAQ,YAAlB,4BAA4B;AAAA;AAAA,IAC5C,uBAAuB,SAAI;AAhH/B;AAiHM,yCAAU,SAAQ,0BAAlB,4BAA0C,SAA1C,YAAkD;AAAA;AAAA,EACtD;AAEA,QAAM,cAAU;AAAA,IACd,UAAU,UAAU,QAAQ,OAAO,IAAI,KAAK,WAAW;AAAA,EACzD;AAEA,QAAM,qBACH,UAAU,WAAW,QAAQ,SAAS,QAAQ,WAC9C,QAAQ,WACP,QAAQ,MAAM,QACd,QAAQ,QAAQ,OAAO,QAAQ;AAEnC,MAAI,oBAAoB;AACtB,YAAQ,UAAU,UAAU,UAAU,QAAQ,OAAO,IAAI,KAAK,WAAW;AAAA,EAC3E;AAEA,QAAM,OAAO,QAAQ;AACrB,QAAM,0BAAsB,qBAAO;AAAA,IACjC;AAAA,IACA,UAAU,KAAK;AAAA,EACjB,CAAC;AAED,MAAI,oBAAoB,QAAQ,SAAS,MAAM;AAC7C,wBAAoB,UAAU,EAAE,MAAM,UAAU,KAAK,SAAS;AAAA,EAChE;AAEA,QAAM,0BAAsB;AAAA,IAC1B,CAAC,WAAuB;AACtB,UAAI,eAAe;AAEnB,YAAM,iBAAiB,MAAM;AAC3B,YAAI,CAAC,gBAAgB,oBAAoB,QAAQ,SAAS,MAAM;AAC9D;AAAA,QACF;AAEA,4BAAoB,UAAU,EAAE,MAAM,UAAU,KAAK,SAAS;AAC9D,eAAO;AAAA,MACT;AAEA,YAAM,cAAc,KAAK,2BAA2B;AAAA,QAClD;AAAA,QACA;AAAA,MACF;AAKA,0BAAoB,UAAU,EAAE,MAAM,UAAU,KAAK,SAAS;AAE9D,aAAO,MAAM;AACX,uBAAe;AACf,oBAAY;AAAA,MACd;AAAA,IACF;AAAA,IACA,CAAC,MAAM,cAAc;AAAA,EACvB;AAEA,QAAM,0BAAsB;AAAA,IAC1B,MAAM,oBAAoB,QAAQ;AAAA,IAClC,CAAC;AAAA,EACH;AAEA,QAAM,eAAW;AAAA,IACf;AAAA,IACA;AAAA,IACA;AAAA,EACF;AAEA,QAAM,wBAAoB;AAAA,IACxB,CAAC,WACC,KAAK,yBAAyB,EAAE,MAAM;AACpC,UAAI,oBAAoB,QAAQ,SAAS,MAAM;AAC7C;AAAA,MACF;AAEA,UAAI,KAAK,WAAW,WAAW,KAAK,WAAW,SAAS;AAEtD,4BAAoB,UAAU,EAAE,MAAM,UAAU,KAAK,SAAS;AAAA,MAChE;AAEA,aAAO;AAAA,IACT,CAAC;AAAA,IACH,CAAC,IAAI;AAAA,EACP;AAEA,QAAM,wBAAoB,0BAAY,MAAM,KAAK,QAAQ,CAAC,IAAI,CAAC;AAE/D,QAAM,aAAS;AAAA,IACb;AAAA,IACA;AAAA,IACA;AAAA,EACF;AAEA,QAAM,YAAQ;AAAA,IACZ,QAAQ,QAAQ,wBAAwB;AAAA,IACxC,MAAM,QAAQ,QAAQ;AAAA,IACtB,MAAM,QAAQ,QAAQ;AAAA,EACxB;AAEA,QAAM,kBAAc;AAAA,IAClB,CACE,kBACG;AACH,UAAI,OAAO,kBAAkB,YAAY;AACvC,wBAAgB,cAAc,QAAQ,QAAQ,QAAQ;AAAA,MACxD;AACA,cAAQ,QAAQ,WAAW;AAAA,IAC7B;AAAA,IACA,CAAC,OAAO;AAAA,EACV;AAEA,8BAAU,MAAM;AACd,QAAI,QAAQ;AACV,cAAQ,QAAQ,aAAa;AAAA,IAC/B;AAAA,EACF,GAAG,CAAC,QAAQ,OAAO,CAAC;AAEpB,SAAO;AAAA,IACL,IAAI,QAAQ,QAAQ;AAAA,IACpB;AAAA,IACA;AAAA,IACA,aAAa,QAAQ,QAAQ;AAAA,IAC7B,YAAY,QAAQ,QAAQ;AAAA,IAC5B,YAAY,QAAQ,QAAQ;AAAA,IAC5B,MAAM,QAAQ,QAAQ;AAAA,IACtB;AAAA,IACA,cAAc,QAAQ,QAAQ;AAAA,IAC9B;AAAA;AAAA;AAAA;AAAA,IAIA,eAAe,QAAQ,QAAQ;AAAA,IAC/B,eAAe,QAAQ,QAAQ;AAAA,IAC/B,yBAAyB,QAAQ,QAAQ;AAAA,EAC3C;AACF;;;AGzPA,IAAAC,aAIO;AACP,IAAAC,gBAAgE;AAChE,iBAAmB;AAyDZ,SAAS,cAAc;AAAA,EAC5B,MAAM;AAAA,EACN;AAAA,EACA,oBAAoB;AAAA,EACpB,eAAe;AAAA,EACf;AAAA,EACA;AAAA,EACA;AAAA,EACA,iBAAiB;AAAA,EACjB,OAAAC;AAAA,EACA;AAAA,EACA;AAAA,EACA,uBAAuB;AACzB,IAMI,CAAC,GAAyB;AAE5B,QAAM,aAAS,qBAAM;AACrB,QAAM,eAAe,MAAM;AAG3B,QAAM,EAAE,MAAM,OAAO,QAAI,WAAAC,SAAe,CAAC,KAAK,YAAY,GAAG,MAAM;AAAA,IACjE,cAAc;AAAA,EAChB,CAAC;AAED,QAAM,EAAE,MAAM,YAAY,OAAO,QAAQ,cAAc,QAAI,WAAAA;AAAA,IACzD,CAAC,cAAc,SAAS;AAAA,IACxB;AAAA,EACF;AAEA,QAAM,CAAC,OAAO,QAAQ,QAAI,wBAA4B,MAAS;AAC/D,QAAM,aAAa;AAGnB,QAAM,CAAC,iBAAiB,kBAAkB,QACxC,wBAAiC,IAAI;AAEvC,QAAM,uBAAmB,sBAAO;AAAA,IAC9B;AAAA,IACA;AAAA,IACA;AAAA,EACF,CAAC;AAED,+BAAU,MAAM;AACd,qBAAiB,UAAU;AAAA,MACzB;AAAA,MACA;AAAA,MACA;AAAA,IACF;AAAA,EACF,GAAG,CAAC,aAAa,SAAS,IAAI,CAAC;AAE/B,QAAM,qBAAiB;AAAA,IACrB,OAAO,QAAgB,gBACrB,8BAAkB;AAAA,MAChB;AAAA,MACA;AAAA,MACA,aAAa,iBAAiB,QAAQ;AAAA,MACtC,SAAS,EAAE,GAAG,iBAAiB,QAAQ,SAAS,GAAG,mCAAS,QAAQ;AAAA,MACpE,MAAM;AAAA,QACJ,GAAG,iBAAiB,QAAQ;AAAA,QAC5B,GAAG,mCAAS;AAAA,MACd;AAAA,MACA;AAAA,MACA,OAAAD;AAAA;AAAA,MAEA,eAAe;AAAA,QACb,CAACE,gBAAuB,OAAOA,aAAY,KAAK;AAAA,QAChD;AAAA,MACF;AAAA,MACA,YAAY;AAAA,MACZ;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,IACF,CAAC;AAAA,IACH;AAAA,MACE;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACAF;AAAA,MACA;AAAA,IACF;AAAA,EACF;AAEA,QAAM,WAAO,2BAAY,MAAM;AAC7B,QAAI,iBAAiB;AACnB,sBAAgB,MAAM;AACtB,yBAAmB,IAAI;AAAA,IACzB;AAAA,EACF,GAAG,CAAC,eAAe,CAAC;AAEpB,QAAM,oBAAgB;AAAA,IACpB,CAACE,gBAAuB;AACtB,aAAOA,aAAY,KAAK;AAAA,IAC1B;AAAA,IACA,CAAC,MAAM;AAAA,EACT;AAEA,QAAM,eAAW;AAAA,IACf,OAAO,QAAQ,YAAY;AACzB,aAAO,eAAe,QAAQ,OAAO;AAAA,IACvC;AAAA,IACA,CAAC,cAAc;AAAA,EACjB;AAEA,QAAM,CAAC,OAAO,QAAQ,QAAI,wBAAS,YAAY;AAE/C,QAAM,mBAAe;AAAA,IACnB,CAAC,UAA4C;AArLjD;AAsLM,2CAAO,mBAAP;AACA,UAAI,CAAC,OAAO;AACV;AAAA,MACF;AAEA,YAAM,SAAS,SAAS,KAAK;AAC7B,eAAS,EAAE;AACX,aAAO;AAAA,IACT;AAAA,IACA,CAAC,OAAO,UAAU,QAAQ;AAAA,EAC5B;AAEA,QAAM,wBAAoB;AAAA,IACxB,CAAC,MAAW;AACV,eAAS,EAAE,OAAO,KAAK;AAAA,IACzB;AAAA,IACA,CAAC,QAAQ;AAAA,EACX;AAEA,SAAO;AAAA,IACL;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF;AACF;;;ACrNA,4BASO;AACP,IAAAC,aAKO;AACP,IAAAC,gBAAqD;AACrD,IAAAC,cAAmB;AAGnB,IAAM,mBAAmB,MAAM;AAsG/B,SAAS,UAIP;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA;AAAA,EACA;AAAA,EACA,OAAAC;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AACF,GAGiD;AAE/C,QAAM,aAAS,qBAAM;AACrB,QAAM,eAAe,kBAAM;AAG3B,QAAM,EAAE,MAAM,OAAO,QAAI,YAAAC;AAAA,IACvB,CAAC,cAAc,QAAQ;AAAA,IACvB;AAAA,IACA,EAAE,cAAc,aAAa;AAAA,EAC/B;AAEA,QAAM,CAAC,OAAO,QAAQ,QAAI,wBAA4B,MAAS;AAC/D,QAAM,CAAC,WAAW,YAAY,QAAI,wBAAS,KAAK;AAGhD,QAAM,yBAAqB,sBAA+B,IAAI;AAE9D,QAAM,WAAO,2BAAY,MAAM;AA7JjC;AA8JI,QAAI;AACF,+BAAmB,YAAnB,mBAA4B;AAAA,IAC9B,SAAS,SAAS;AAAA,IAClB,UAAE;AACA,mBAAa,KAAK;AAClB,yBAAmB,UAAU;AAAA,IAC/B;AAAA,EACF,GAAG,CAAC,CAAC;AAEL,QAAM,SAAS,OAAO,UAAiB;AAvKzC;AAwKI,UAAM,kBAAkB,IAAI,gBAAgB;AAE5C,QAAI;AACF,kBAAY;AAEZ,mBAAa,IAAI;AAEjB,yBAAmB,UAAU;AAG7B,YAAM,kBAAkB,UAAM,+BAAQ,OAAO;AAE7C,YAAM,cAAcD,UAAA,OAAAA,SAAS,iBAAiB;AAC9C,YAAM,WAAW,MAAM,YAAY,KAAK;AAAA,QACtC,QAAQ;AAAA,QACR,SAAS;AAAA,UACP,gBAAgB;AAAA,UAChB,OAAG,wCAAiB,eAAe;AAAA,QACrC;AAAA,QACA;AAAA,QACA,QAAQ,gBAAgB;AAAA,QACxB,MAAM,KAAK,UAAU,KAAK;AAAA,MAC5B,CAAC;AAED,UAAI,CAAC,SAAS,IAAI;AAChB,cAAM,IAAI;AAAA,WACP,WAAM,SAAS,KAAK,MAApB,YAA0B;AAAA,QAC7B;AAAA,MACF;AAEA,UAAI,SAAS,QAAQ,MAAM;AACzB,cAAM,IAAI,MAAM,6BAA6B;AAAA,MAC/C;AAEA,UAAI,kBAAkB;AACtB,UAAI,eAAgD;AAEpD,YAAM,SAAS,KAAK,YAAY,IAAI,kBAAkB,CAAC,EAAE;AAAA,QACvD,IAAI,eAAuB;AAAA,UACzB,MAAM,MAAM,OAAO;AACjB,+BAAmB;AAEnB,kBAAM,EAAE,MAAM,IAAI,UAAM,6BAAiB,eAAe;AACxD,kBAAM,gBAAgB;AAEtB,gBAAI,KAAC,4BAAgB,cAAc,aAAa,GAAG;AACjD,6BAAe;AAEf,kBAAI,mBAAmB,YAAY,iBAAiB;AAClD,uBAAO,aAAa;AAAA,cACtB;AAAA,YACF;AAAA,UACF;AAAA,UAEA,MAAM,QAAQ;AACZ,gBAAI,mBAAmB,YAAY,iBAAiB;AAClD,2BAAa,KAAK;AAAA,YACpB;AAEA,gBAAI,YAAY,MAAM;AACpB,oBAAM,mBAAmB,UAAM,yCAAkB;AAAA,gBAC/C,OAAO;AAAA,gBACP,YAAQ,qBAAS,MAAM;AAAA,cACzB,CAAC;AAED;AAAA,gBACE,iBAAiB,UACb,EAAE,QAAQ,iBAAiB,OAAO,OAAO,OAAU,IACnD,EAAE,QAAQ,QAAW,OAAO,iBAAiB,MAAM;AAAA,cACzD;AAAA,YACF;AAEA,gBAAI,mBAAmB,YAAY,iBAAiB;AAClD,iCAAmB,UAAU;AAAA,YAC/B;AAAA,UACF;AAAA,QACF,CAAC;AAAA,MACH;AAAA,IACF,SAASE,QAAO;AACd,cAAI,oCAAaA,MAAK,GAAG;AACvB;AAAA,MACF;AAEA,UAAI,WAAWA,kBAAiB,OAAO;AACrC,gBAAQA,MAAK;AAAA,MACf;AAEA,UAAI,mBAAmB,YAAY,iBAAiB;AAClD,qBAAa,KAAK;AAClB,2BAAmB,UAAU;AAC7B,iBAASA,kBAAiB,QAAQA,SAAQ,IAAI,MAAM,OAAOA,MAAK,CAAC,CAAC;AAAA,MACpE;AAAA,IACF;AAAA,EACF;AAEA,QAAM,QAAQ,MAAM;AAClB,SAAK;AACL,gBAAY;AAAA,EACd;AAEA,QAAM,cAAc,MAAM;AACxB,aAAS,MAAS;AAClB,iBAAa,KAAK;AAClB,WAAO,MAAS;AAAA,EAClB;AAEA,SAAO;AAAA,IACL;AAAA,IACA,QAAQ;AAAA,IACR;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF;AACF;AAEO,IAAM,yBAAyB;","names":["import_ai","throttleFunction","import_ai","import_react","fetch","useSWR","completion","import_ai","import_react","import_swr","fetch","useSWR","error"]}
|
|
1
|
+
{"version":3,"sources":["../src/index.ts","../src/use-chat.ts","../src/chat.react.ts","../src/throttle.ts","../src/use-completion.ts","../src/use-object.ts"],"sourcesContent":["export * from './use-chat';\nexport { Chat } from './chat.react';\nexport * from './use-completion';\nexport * from './use-object';\n","import {\n type AbstractChat,\n type ChatInit,\n type ChatTransport,\n type CreateUIMessage,\n type UIMessage,\n DefaultChatTransport,\n} from 'ai';\nimport { useCallback, useEffect, useRef, useSyncExternalStore } from 'react';\nimport { Chat } from './chat.react';\n\nexport type { CreateUIMessage, UIMessage };\n\nexport type UseChatHelpers<UI_MESSAGE extends UIMessage> = {\n /**\n * The id of the chat.\n */\n readonly id: string;\n\n /**\n * Update the `messages` state locally. This is useful when you want to\n * edit the messages on the client, and then trigger the `reload` method\n * manually to regenerate the AI response.\n */\n setMessages: (\n messages: UI_MESSAGE[] | ((messages: UI_MESSAGE[]) => UI_MESSAGE[]),\n ) => void;\n\n error: Error | undefined;\n} & Pick<\n AbstractChat<UI_MESSAGE>,\n | 'sendMessage'\n | 'regenerate'\n | 'stop'\n | 'resumeStream'\n | 'addToolResult'\n | 'addToolOutput'\n | 'addToolApprovalResponse'\n | 'status'\n | 'messages'\n | 'clearError'\n>;\n\nexport type UseChatOptions<UI_MESSAGE extends UIMessage> = (\n | { chat: Chat<UI_MESSAGE> }\n | ChatInit<UI_MESSAGE>\n) & {\n /**\n * Custom throttle wait in ms for the chat messages and data updates.\n * Default is undefined, which disables throttling.\n */\n experimental_throttle?: number;\n\n /**\n * Whether to automatically resume an ongoing chat generation stream.\n */\n resume?: boolean;\n};\n\ntype AutomaticResumeState = {\n registrations: Set<object>;\n cleanupVisibilityListener?: () => void;\n pendingVisibilityResume?: Promise<void>;\n};\n\nconst automaticResumeStates = new WeakMap<object, AutomaticResumeState>();\n\ntype AutomaticallyResumableChat = {\n '~resumeStreamIfDisconnected': (options: {\n shouldResume: () => boolean;\n }) => Promise<void>;\n};\n\n/**\n * When the document becomes visible again, resume the chat stream if it was\n * interrupted by a network disconnect while the page was in the background.\n */\nfunction resumeOnVisible<UI_MESSAGE extends UIMessage>({\n chat,\n state,\n}: {\n chat: Chat<UI_MESSAGE>;\n state: AutomaticResumeState;\n}) {\n if (\n document.visibilityState !== 'visible' ||\n state.pendingVisibilityResume != null\n ) {\n return;\n }\n\n const isRegistered = () =>\n automaticResumeStates.get(chat) === state && state.registrations.size > 0;\n\n const clearPendingResume = () => {\n state.pendingVisibilityResume = undefined;\n };\n\n const resumableChat = chat as Chat<UI_MESSAGE> & AutomaticallyResumableChat;\n\n state.pendingVisibilityResume = resumableChat['~resumeStreamIfDisconnected']({\n shouldResume: () =>\n isRegistered() && document.visibilityState === 'visible',\n }).then(clearPendingResume, clearPendingResume);\n}\n\nfunction registerAutomaticResume<UI_MESSAGE extends UIMessage>({\n chat,\n registration,\n}: {\n chat: Chat<UI_MESSAGE>;\n registration: object;\n}) {\n let state = automaticResumeStates.get(chat);\n\n if (state == null) {\n const newState: AutomaticResumeState = { registrations: new Set() };\n automaticResumeStates.set(chat, newState);\n state = newState;\n\n if (typeof document !== 'undefined') {\n const onVisibilityChange = () =>\n resumeOnVisible({ chat, state: newState });\n\n document.addEventListener('visibilitychange', onVisibilityChange);\n newState.cleanupVisibilityListener = () => {\n document.removeEventListener('visibilitychange', onVisibilityChange);\n };\n }\n }\n\n const { registrations } = state;\n const shouldResume = registrations.size === 0;\n registrations.add(registration);\n\n if (\n shouldResume &&\n chat.status !== 'submitted' &&\n chat.status !== 'streaming'\n ) {\n void chat.resumeStream();\n }\n\n return () => {\n registrations.delete(registration);\n\n if (registrations.size === 0) {\n state.cleanupVisibilityListener?.();\n automaticResumeStates.delete(chat);\n }\n };\n}\nexport function useChat<UI_MESSAGE extends UIMessage = UIMessage>({\n experimental_throttle: throttleWaitMs,\n resume = false,\n ...options\n}: UseChatOptions<UI_MESSAGE> = {}): UseChatHelpers<UI_MESSAGE> {\n const automaticResumeRegistration = useRef({});\n\n // the Chat instance is created once and not recreated when options change,\n // so it would normally keep the callbacks/transport from the first render forever\n\n // keep latest values in a ref that is refreshed on every render,\n // and hand `Chat` stable wrappers that read from it to avoid stale closures\n const latestRef = useRef<\n Partial<\n Pick<\n ChatInit<UI_MESSAGE>,\n | 'onToolCall'\n | 'onData'\n | 'onFinish'\n | 'onError'\n | 'sendAutomaticallyWhen'\n | 'transport'\n >\n >\n >({});\n\n if (!('chat' in options)) {\n latestRef.current = {\n onToolCall: options.onToolCall,\n onData: options.onData,\n onFinish: options.onFinish,\n onError: options.onError,\n sendAutomaticallyWhen: options.sendAutomaticallyWhen,\n transport: options.transport,\n };\n }\n\n // resolve the latest transport and fallback to a lazily created default transport\n let defaultTransport: ChatTransport<UI_MESSAGE> | undefined;\n const getTransport = () =>\n latestRef.current.transport ??\n (defaultTransport ??= new DefaultChatTransport<UI_MESSAGE>());\n\n // give `Chat` stable wrappers that always read the latest values from `latestRef`\n const chatOptions: typeof options = {\n ...options,\n transport: {\n sendMessages: sendOptions => getTransport().sendMessages(sendOptions),\n reconnectToStream: reconnectOptions =>\n getTransport().reconnectToStream(reconnectOptions),\n },\n onToolCall: arg => latestRef.current.onToolCall?.(arg),\n onData: arg => latestRef.current.onData?.(arg),\n onFinish: arg => latestRef.current.onFinish?.(arg),\n onError: arg => latestRef.current.onError?.(arg),\n sendAutomaticallyWhen: arg =>\n latestRef.current.sendAutomaticallyWhen?.(arg) ?? false,\n };\n\n const chatRef = useRef<Chat<UI_MESSAGE>>(\n 'chat' in options ? options.chat : new Chat(chatOptions),\n );\n\n const shouldRecreateChat =\n ('chat' in options && options.chat !== chatRef.current) ||\n ('id' in options &&\n options.id != null &&\n chatRef.current.id !== options.id);\n\n if (shouldRecreateChat) {\n chatRef.current = 'chat' in options ? options.chat : new Chat(chatOptions);\n }\n\n const chat = chatRef.current;\n const messagesSnapshotRef = useRef({\n chat,\n messages: chat.messages,\n });\n\n if (messagesSnapshotRef.current.chat !== chat) {\n messagesSnapshotRef.current = { chat, messages: chat.messages };\n }\n\n const subscribeToMessages = useCallback(\n (update: () => void) => {\n let isSubscribed = true;\n\n const updateMessages = () => {\n if (!isSubscribed || messagesSnapshotRef.current.chat !== chat) {\n return;\n }\n\n messagesSnapshotRef.current = { chat, messages: chat.messages };\n update();\n };\n\n const unsubscribe = chat['~registerMessagesCallback'](\n updateMessages,\n throttleWaitMs,\n );\n\n // Synchronize changes that may have happened between render and\n // subscription. useSyncExternalStore checks the snapshot after\n // subscribing and schedules a render when it changed.\n messagesSnapshotRef.current = { chat, messages: chat.messages };\n\n return () => {\n isSubscribed = false;\n unsubscribe();\n };\n },\n [chat, throttleWaitMs],\n );\n\n const getMessagesSnapshot = useCallback(\n () => messagesSnapshotRef.current.messages,\n [],\n );\n\n const messages = useSyncExternalStore(\n subscribeToMessages,\n getMessagesSnapshot,\n getMessagesSnapshot,\n );\n\n const subscribeToStatus = useCallback(\n (update: () => void) =>\n chat['~registerStatusCallback'](() => {\n if (messagesSnapshotRef.current.chat !== chat) {\n return;\n }\n\n if (chat.status === 'ready' || chat.status === 'error') {\n // Publish the latest messages before the terminal status can render.\n messagesSnapshotRef.current = { chat, messages: chat.messages };\n }\n\n update();\n }),\n [chat],\n );\n\n const getStatusSnapshot = useCallback(() => chat.status, [chat]);\n\n const status = useSyncExternalStore(\n subscribeToStatus,\n getStatusSnapshot,\n getStatusSnapshot,\n );\n\n const error = useSyncExternalStore(\n chatRef.current['~registerErrorCallback'],\n () => chatRef.current.error,\n () => chatRef.current.error,\n );\n\n const setMessages = useCallback(\n (\n messagesParam: UI_MESSAGE[] | ((messages: UI_MESSAGE[]) => UI_MESSAGE[]),\n ) => {\n if (typeof messagesParam === 'function') {\n messagesParam = messagesParam(chatRef.current.messages);\n }\n chatRef.current.messages = messagesParam;\n },\n [chatRef],\n );\n\n useEffect(() => {\n if (resume) {\n return registerAutomaticResume({\n chat,\n registration: automaticResumeRegistration.current,\n });\n }\n }, [resume, chat]);\n\n return {\n id: chatRef.current.id,\n messages,\n setMessages,\n sendMessage: chatRef.current.sendMessage,\n regenerate: chatRef.current.regenerate,\n clearError: chatRef.current.clearError,\n stop: chatRef.current.stop,\n error,\n resumeStream: chatRef.current.resumeStream,\n status,\n /**\n * @deprecated Use `addToolOutput` instead.\n */\n addToolResult: chatRef.current.addToolOutput,\n addToolOutput: chatRef.current.addToolOutput,\n addToolApprovalResponse: chatRef.current.addToolApprovalResponse,\n };\n}\n","import {\n AbstractChat,\n type ChatInit,\n type ChatState,\n type ChatStatus,\n type UIMessage,\n} from 'ai';\nimport { throttle } from './throttle';\n\nfunction cloneMetadata<METADATA>(metadata: METADATA): METADATA {\n if (Array.isArray(metadata)) {\n return [...metadata] as METADATA;\n }\n\n if (\n metadata != null &&\n typeof metadata === 'object' &&\n (Object.getPrototypeOf(metadata) === Object.prototype ||\n Object.getPrototypeOf(metadata) === null)\n ) {\n return { ...metadata } as METADATA;\n }\n\n return metadata;\n}\n\nclass ReactChatState<\n UI_MESSAGE extends UIMessage,\n> implements ChatState<UI_MESSAGE> {\n #messages: UI_MESSAGE[];\n #status: ChatStatus = 'ready';\n #error: Error | undefined = undefined;\n\n #messagesCallbacks = new Set<() => void>();\n #statusCallbacks = new Set<() => void>();\n #errorCallbacks = new Set<() => void>();\n\n constructor(initialMessages: UI_MESSAGE[] = []) {\n this.#messages = initialMessages;\n }\n\n get status(): ChatStatus {\n return this.#status;\n }\n\n set status(newStatus: ChatStatus) {\n this.#status = newStatus;\n this.#callStatusCallbacks();\n }\n\n get error(): Error | undefined {\n return this.#error;\n }\n\n set error(newError: Error | undefined) {\n this.#error = newError;\n this.#callErrorCallbacks();\n }\n\n get messages(): UI_MESSAGE[] {\n return this.#messages;\n }\n\n set messages(newMessages: UI_MESSAGE[]) {\n this.#messages = [...newMessages];\n this.#callMessagesCallbacks();\n }\n\n pushMessage = (message: UI_MESSAGE) => {\n this.#messages = this.#messages.concat(message);\n this.#callMessagesCallbacks();\n };\n\n popMessage = () => {\n this.#messages = this.#messages.slice(0, -1);\n this.#callMessagesCallbacks();\n };\n\n replaceMessage = (index: number, message: UI_MESSAGE) => {\n this.#messages = [\n ...this.#messages.slice(0, index),\n this.snapshot(message),\n ...this.#messages.slice(index + 1),\n ];\n this.#callMessagesCallbacks();\n };\n\n snapshot = <T>(value: T): T => {\n if (\n value == null ||\n typeof value !== 'object' ||\n !('parts' in value) ||\n !Array.isArray(value.parts)\n ) {\n return value;\n }\n\n const message = value as UI_MESSAGE;\n const snapshot = {\n ...message,\n parts: message.parts.map(part => ({ ...part })),\n };\n\n if ('metadata' in message) {\n snapshot.metadata = cloneMetadata(message.metadata);\n }\n\n return snapshot as T;\n };\n\n '~registerMessagesCallback' = (\n onChange: () => void,\n throttleWaitMs?: number,\n ): (() => void) => {\n const callback = throttleWaitMs\n ? throttle(onChange, throttleWaitMs)\n : onChange;\n this.#messagesCallbacks.add(callback);\n return () => {\n this.#messagesCallbacks.delete(callback);\n };\n };\n\n '~registerStatusCallback' = (onChange: () => void): (() => void) => {\n this.#statusCallbacks.add(onChange);\n return () => {\n this.#statusCallbacks.delete(onChange);\n };\n };\n\n '~registerErrorCallback' = (onChange: () => void): (() => void) => {\n this.#errorCallbacks.add(onChange);\n return () => {\n this.#errorCallbacks.delete(onChange);\n };\n };\n\n #callMessagesCallbacks = () => {\n this.#messagesCallbacks.forEach(callback => callback());\n };\n\n #callStatusCallbacks = () => {\n this.#statusCallbacks.forEach(callback => callback());\n };\n\n #callErrorCallbacks = () => {\n this.#errorCallbacks.forEach(callback => callback());\n };\n}\n\nexport class Chat<\n UI_MESSAGE extends UIMessage,\n> extends AbstractChat<UI_MESSAGE> {\n #state: ReactChatState<UI_MESSAGE>;\n\n constructor({ messages, ...init }: ChatInit<UI_MESSAGE>) {\n const state = new ReactChatState(messages);\n super({ ...init, state });\n this.#state = state;\n }\n\n '~registerMessagesCallback' = (\n onChange: () => void,\n throttleWaitMs?: number,\n ): (() => void) =>\n this.#state['~registerMessagesCallback'](onChange, throttleWaitMs);\n\n '~registerStatusCallback' = (onChange: () => void): (() => void) =>\n this.#state['~registerStatusCallback'](onChange);\n\n '~registerErrorCallback' = (onChange: () => void): (() => void) =>\n this.#state['~registerErrorCallback'](onChange);\n}\n","import throttleFunction from 'throttleit';\n\nexport function throttle<T extends (...args: any[]) => any>(\n fn: T,\n waitMs: number | undefined,\n): T {\n return waitMs != null ? throttleFunction(fn, waitMs) : fn;\n}\n","import {\n callCompletionApi,\n type CompletionRequestOptions,\n type UseCompletionOptions,\n} from 'ai';\nimport { useCallback, useEffect, useId, useRef, useState } from 'react';\nimport useSWR from 'swr';\nimport { throttle } from './throttle';\n\nexport type { UseCompletionOptions };\n\nexport type UseCompletionHelpers = {\n /** The current completion result */\n completion: string;\n /**\n * Send a new prompt to the API endpoint and update the completion state.\n */\n complete: (\n prompt: string,\n options?: CompletionRequestOptions,\n ) => Promise<string | null | undefined>;\n /** The error object of the API request */\n error: undefined | Error;\n /**\n * Abort the current API request but keep the generated tokens.\n */\n stop: () => void;\n /**\n * Update the `completion` state locally.\n */\n setCompletion: (completion: string) => void;\n /** The current value of the input */\n input: string;\n /** setState-powered method to update the input value */\n setInput: React.Dispatch<React.SetStateAction<string>>;\n /**\n * An input/textarea-ready onChange handler to control the value of the input\n * @example\n * ```jsx\n * <input onChange={handleInputChange} value={input} />\n * ```\n */\n handleInputChange: (\n event:\n | React.ChangeEvent<HTMLInputElement>\n | React.ChangeEvent<HTMLTextAreaElement>,\n ) => void;\n\n /**\n * Form submission handler to automatically reset input and append a user message\n * @example\n * ```jsx\n * <form onSubmit={handleSubmit}>\n * <input onChange={handleInputChange} value={input} />\n * </form>\n * ```\n */\n handleSubmit: (event?: { preventDefault?: () => void }) => void;\n\n /** Whether the API request is in progress */\n isLoading: boolean;\n};\n\nexport function useCompletion({\n api = '/api/completion',\n id,\n initialCompletion = '',\n initialInput = '',\n credentials,\n headers,\n body,\n streamProtocol = 'data',\n fetch,\n onFinish,\n onError,\n experimental_throttle: throttleWaitMs,\n}: UseCompletionOptions & {\n /**\n * Custom throttle wait in ms for the completion and data updates.\n * Default is undefined, which disables throttling.\n */\n experimental_throttle?: number;\n} = {}): UseCompletionHelpers {\n // Generate an unique id for the completion if not provided.\n const hookId = useId();\n const completionId = id || hookId;\n\n // Store the completion state in SWR, using the completionId as the key to share states.\n const { data, mutate } = useSWR<string>([api, completionId], null, {\n fallbackData: initialCompletion,\n });\n\n const { data: isLoading = false, mutate: mutateLoading } = useSWR<boolean>(\n [completionId, 'loading'],\n null,\n );\n\n const [error, setError] = useState<undefined | Error>(undefined);\n const completion = data!;\n\n // Abort controller to cancel the current API call.\n const [abortController, setAbortController] =\n useState<AbortController | null>(null);\n\n const extraMetadataRef = useRef({\n credentials,\n headers,\n body,\n });\n\n useEffect(() => {\n extraMetadataRef.current = {\n credentials,\n headers,\n body,\n };\n }, [credentials, headers, body]);\n\n const triggerRequest = useCallback(\n async (prompt: string, options?: CompletionRequestOptions) =>\n callCompletionApi({\n api,\n prompt,\n credentials: extraMetadataRef.current.credentials,\n headers: { ...extraMetadataRef.current.headers, ...options?.headers },\n body: {\n ...extraMetadataRef.current.body,\n ...options?.body,\n },\n streamProtocol,\n fetch,\n // throttle streamed ui updates:\n setCompletion: throttle(\n (completion: string) => mutate(completion, false),\n throttleWaitMs,\n ),\n setLoading: mutateLoading,\n setError,\n setAbortController,\n onFinish,\n onError,\n }),\n [\n mutate,\n mutateLoading,\n api,\n extraMetadataRef,\n setAbortController,\n onFinish,\n onError,\n setError,\n streamProtocol,\n fetch,\n throttleWaitMs,\n ],\n );\n\n const stop = useCallback(() => {\n if (abortController) {\n abortController.abort();\n setAbortController(null);\n }\n }, [abortController]);\n\n const setCompletion = useCallback(\n (completion: string) => {\n mutate(completion, false);\n },\n [mutate],\n );\n\n const complete = useCallback<UseCompletionHelpers['complete']>(\n async (prompt, options) => {\n return triggerRequest(prompt, options);\n },\n [triggerRequest],\n );\n\n const [input, setInput] = useState(initialInput);\n\n const handleSubmit = useCallback(\n (event?: { preventDefault?: () => void }) => {\n event?.preventDefault?.();\n if (!input) {\n return;\n }\n\n const result = complete(input);\n setInput('');\n return result;\n },\n [input, complete, setInput],\n );\n\n const handleInputChange = useCallback(\n (e: any) => {\n setInput(e.target.value);\n },\n [setInput],\n );\n\n return {\n completion,\n complete,\n error,\n setCompletion,\n stop,\n input,\n setInput,\n handleInputChange,\n handleSubmit,\n isLoading,\n };\n}\n","import {\n isAbortError,\n resolve,\n normalizeHeaders,\n safeValidateTypes,\n type FetchFunction,\n type FlexibleSchema,\n type InferSchema,\n type Resolvable,\n} from '@ai-sdk/provider-utils';\nimport {\n asSchema,\n isDeepEqualData,\n parsePartialJson,\n type DeepPartial,\n} from 'ai';\nimport { useCallback, useId, useRef, useState } from 'react';\nimport useSWR from 'swr';\n\n// use function to allow for mocking in tests:\nconst getOriginalFetch = () => fetch;\n\nexport type Experimental_UseObjectOptions<\n SCHEMA extends FlexibleSchema,\n RESULT,\n> = {\n /**\n * The API endpoint. It should stream JSON that matches the schema as chunked text.\n */\n api: string;\n\n /**\n * A schema that defines the shape of the complete object.\n */\n schema: SCHEMA;\n\n /**\n * An unique identifier. If not provided, a random one will be\n * generated. When provided, the `useObject` hook with the same `id` will\n * have shared states across components.\n */\n id?: string;\n\n /**\n * An optional value for the initial object.\n */\n initialValue?: DeepPartial<RESULT>;\n\n /**\n * Custom fetch implementation. You can use it as a middleware to intercept requests,\n * or to provide a custom fetch implementation for e.g. testing.\n */\n fetch?: FetchFunction;\n\n /**\n * Callback that is called when the stream has finished.\n */\n onFinish?: (event: {\n /**\n * The generated object (typed according to the schema).\n * Can be undefined if the final object does not match the schema.\n */\n object: RESULT | undefined;\n\n /**\n * Optional error object. This is e.g. a TypeValidationError when the final object does not match the schema.\n */\n error: Error | undefined;\n }) => Promise<void> | void;\n\n /**\n * Callback function to be called when an error is encountered.\n */\n onError?: (error: Error) => void;\n\n /**\n * Additional HTTP headers to be included in the request.\n * Can be a static object, a function that returns headers, or an async function\n * for dynamic auth tokens.\n */\n headers?: Resolvable<Record<string, string> | Headers>;\n\n /**\n * The credentials mode to be used for the fetch request.\n * Possible values are: 'omit', 'same-origin', 'include'.\n * Defaults to 'same-origin'.\n */\n credentials?: RequestCredentials;\n};\n\nexport type Experimental_UseObjectHelpers<RESULT, INPUT> = {\n /**\n * Calls the API with the provided input as JSON body.\n */\n submit: (input: INPUT) => void;\n\n /**\n * The current value for the generated object. Updated as the API streams JSON chunks.\n */\n object: DeepPartial<RESULT> | undefined;\n\n /**\n * The error object of the API request if any.\n */\n error: Error | undefined;\n\n /**\n * Flag that indicates whether an API request is in progress.\n */\n isLoading: boolean;\n\n /**\n * Abort the current request immediately, keep the current partial object if any.\n */\n stop: () => void;\n\n /**\n * Clear the object state.\n */\n clear: () => void;\n};\n\nfunction useObject<\n SCHEMA extends FlexibleSchema,\n RESULT = InferSchema<SCHEMA>,\n INPUT = any,\n>({\n api,\n id,\n schema, // required, in the future we will use it for validation\n initialValue,\n fetch,\n onError,\n onFinish,\n headers,\n credentials,\n}: Experimental_UseObjectOptions<\n SCHEMA,\n RESULT\n>): Experimental_UseObjectHelpers<RESULT, INPUT> {\n // Generate an unique id if not provided.\n const hookId = useId();\n const completionId = id ?? hookId;\n\n // Store the completion state in SWR, using the completionId as the key to share states.\n const { data, mutate } = useSWR<DeepPartial<RESULT>>(\n [completionId, 'object'],\n null,\n { fallbackData: initialValue },\n );\n\n const [error, setError] = useState<undefined | Error>(undefined);\n const [isLoading, setIsLoading] = useState(false);\n\n // Abort controller to cancel the current API call.\n const abortControllerRef = useRef<AbortController | null>(null);\n\n const stop = useCallback(() => {\n try {\n abortControllerRef.current?.abort();\n } catch (ignored) {\n } finally {\n setIsLoading(false);\n abortControllerRef.current = null;\n }\n }, []);\n\n const submit = async (input: INPUT) => {\n const abortController = new AbortController();\n\n try {\n clearObject();\n\n setIsLoading(true);\n\n abortControllerRef.current = abortController;\n\n // Resolve headers at request time (supports async functions for dynamic auth tokens)\n const resolvedHeaders = await resolve(headers);\n\n const actualFetch = fetch ?? getOriginalFetch();\n const response = await actualFetch(api, {\n method: 'POST',\n headers: {\n 'Content-Type': 'application/json',\n ...normalizeHeaders(resolvedHeaders),\n },\n credentials,\n signal: abortController.signal,\n body: JSON.stringify(input),\n });\n\n if (!response.ok) {\n throw new Error(\n (await response.text()) ?? 'Failed to fetch the response.',\n );\n }\n\n if (response.body == null) {\n throw new Error('The response body is empty.');\n }\n\n let accumulatedText = '';\n let latestObject: DeepPartial<RESULT> | undefined = undefined;\n\n await response.body.pipeThrough(new TextDecoderStream()).pipeTo(\n new WritableStream<string>({\n async write(chunk) {\n accumulatedText += chunk;\n\n const { value } = await parsePartialJson(accumulatedText);\n const currentObject = value as DeepPartial<RESULT>;\n\n if (!isDeepEqualData(latestObject, currentObject)) {\n latestObject = currentObject;\n\n if (abortControllerRef.current === abortController) {\n mutate(currentObject);\n }\n }\n },\n\n async close() {\n if (abortControllerRef.current === abortController) {\n setIsLoading(false);\n }\n\n if (onFinish != null) {\n const validationResult = await safeValidateTypes({\n value: latestObject,\n schema: asSchema(schema),\n });\n\n onFinish(\n validationResult.success\n ? { object: validationResult.value, error: undefined }\n : { object: undefined, error: validationResult.error },\n );\n }\n\n if (abortControllerRef.current === abortController) {\n abortControllerRef.current = null;\n }\n },\n }),\n );\n } catch (error) {\n if (isAbortError(error)) {\n return;\n }\n\n if (onError && error instanceof Error) {\n onError(error);\n }\n\n if (abortControllerRef.current === abortController) {\n setIsLoading(false);\n abortControllerRef.current = null;\n setError(error instanceof Error ? error : new Error(String(error)));\n }\n }\n };\n\n const clear = () => {\n stop();\n clearObject();\n };\n\n const clearObject = () => {\n setError(undefined);\n setIsLoading(false);\n mutate(undefined);\n };\n\n return {\n submit,\n object: data,\n error,\n isLoading,\n stop,\n clear,\n };\n}\n\nexport const experimental_useObject = useObject;\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;;;ACAA,IAAAA,aAOO;AACP,mBAAqE;;;ACRrE,gBAMO;;;ACNP,wBAA6B;AAEtB,SAAS,SACd,IACA,QACG;AACH,SAAO,UAAU,WAAO,kBAAAC,SAAiB,IAAI,MAAM,IAAI;AACzD;;;ADEA,SAAS,cAAwB,UAA8B;AAC7D,MAAI,MAAM,QAAQ,QAAQ,GAAG;AAC3B,WAAO,CAAC,GAAG,QAAQ;AAAA,EACrB;AAEA,MACE,YAAY,QACZ,OAAO,aAAa,aACnB,OAAO,eAAe,QAAQ,MAAM,OAAO,aAC1C,OAAO,eAAe,QAAQ,MAAM,OACtC;AACA,WAAO,EAAE,GAAG,SAAS;AAAA,EACvB;AAEA,SAAO;AACT;AAxBA;AA0BA,IAAM,iBAAN,MAEmC;AAAA,EASjC,YAAY,kBAAgC,CAAC,GAAG;AARhD;AACA,gCAAsB;AACtB,+BAA4B;AAE5B,2CAAqB,oBAAI,IAAgB;AACzC,yCAAmB,oBAAI,IAAgB;AACvC,wCAAkB,oBAAI,IAAgB;AAiCtC,uBAAc,CAAC,YAAwB;AACrC,yBAAK,WAAY,mBAAK,WAAU,OAAO,OAAO;AAC9C,yBAAK,wBAAL;AAAA,IACF;AAEA,sBAAa,MAAM;AACjB,yBAAK,WAAY,mBAAK,WAAU,MAAM,GAAG,EAAE;AAC3C,yBAAK,wBAAL;AAAA,IACF;AAEA,0BAAiB,CAAC,OAAe,YAAwB;AACvD,yBAAK,WAAY;AAAA,QACf,GAAG,mBAAK,WAAU,MAAM,GAAG,KAAK;AAAA,QAChC,KAAK,SAAS,OAAO;AAAA,QACrB,GAAG,mBAAK,WAAU,MAAM,QAAQ,CAAC;AAAA,MACnC;AACA,yBAAK,wBAAL;AAAA,IACF;AAEA,oBAAW,CAAI,UAAgB;AAC7B,UACE,SAAS,QACT,OAAO,UAAU,YACjB,EAAE,WAAW,UACb,CAAC,MAAM,QAAQ,MAAM,KAAK,GAC1B;AACA,eAAO;AAAA,MACT;AAEA,YAAM,UAAU;AAChB,YAAM,WAAW;AAAA,QACf,GAAG;AAAA,QACH,OAAO,QAAQ,MAAM,IAAI,WAAS,EAAE,GAAG,KAAK,EAAE;AAAA,MAChD;AAEA,UAAI,cAAc,SAAS;AACzB,iBAAS,WAAW,cAAc,QAAQ,QAAQ;AAAA,MACpD;AAEA,aAAO;AAAA,IACT;AAEA,wCAA8B,CAC5B,UACA,mBACiB;AACjB,YAAM,WAAW,iBACb,SAAS,UAAU,cAAc,IACjC;AACJ,yBAAK,oBAAmB,IAAI,QAAQ;AACpC,aAAO,MAAM;AACX,2BAAK,oBAAmB,OAAO,QAAQ;AAAA,MACzC;AAAA,IACF;AAEA,sCAA4B,CAAC,aAAuC;AAClE,yBAAK,kBAAiB,IAAI,QAAQ;AAClC,aAAO,MAAM;AACX,2BAAK,kBAAiB,OAAO,QAAQ;AAAA,MACvC;AAAA,IACF;AAEA,qCAA2B,CAAC,aAAuC;AACjE,yBAAK,iBAAgB,IAAI,QAAQ;AACjC,aAAO,MAAM;AACX,2BAAK,iBAAgB,OAAO,QAAQ;AAAA,MACtC;AAAA,IACF;AAEA,+CAAyB,MAAM;AAC7B,yBAAK,oBAAmB,QAAQ,cAAY,SAAS,CAAC;AAAA,IACxD;AAEA,6CAAuB,MAAM;AAC3B,yBAAK,kBAAiB,QAAQ,cAAY,SAAS,CAAC;AAAA,IACtD;AAEA,4CAAsB,MAAM;AAC1B,yBAAK,iBAAgB,QAAQ,cAAY,SAAS,CAAC;AAAA,IACrD;AA7GE,uBAAK,WAAY;AAAA,EACnB;AAAA,EAEA,IAAI,SAAqB;AACvB,WAAO,mBAAK;AAAA,EACd;AAAA,EAEA,IAAI,OAAO,WAAuB;AAChC,uBAAK,SAAU;AACf,uBAAK,sBAAL;AAAA,EACF;AAAA,EAEA,IAAI,QAA2B;AAC7B,WAAO,mBAAK;AAAA,EACd;AAAA,EAEA,IAAI,MAAM,UAA6B;AACrC,uBAAK,QAAS;AACd,uBAAK,qBAAL;AAAA,EACF;AAAA,EAEA,IAAI,WAAyB;AAC3B,WAAO,mBAAK;AAAA,EACd;AAAA,EAEA,IAAI,SAAS,aAA2B;AACtC,uBAAK,WAAY,CAAC,GAAG,WAAW;AAChC,uBAAK,wBAAL;AAAA,EACF;AAkFF;AAvHE;AACA;AACA;AAEA;AACA;AACA;AAsGA;AAIA;AAIA;AAjJF;AAsJO,IAAM,OAAN,cAEG,uBAAyB;AAAA,EAGjC,YAAY,EAAE,UAAU,GAAG,KAAK,GAAyB;AACvD,UAAM,QAAQ,IAAI,eAAe,QAAQ;AACzC,UAAM,EAAE,GAAG,MAAM,MAAM,CAAC;AAJ1B;AAQA,wCAA8B,CAC5B,UACA,mBAEA,mBAAK,QAAO,2BAA2B,EAAE,UAAU,cAAc;AAEnE,sCAA4B,CAAC,aAC3B,mBAAK,QAAO,yBAAyB,EAAE,QAAQ;AAEjD,qCAA2B,CAAC,aAC1B,mBAAK,QAAO,wBAAwB,EAAE,QAAQ;AAb9C,uBAAK,QAAS;AAAA,EAChB;AAaF;AAnBE;;;ADxFF,IAAM,wBAAwB,oBAAI,QAAsC;AAYxE,SAAS,gBAA8C;AAAA,EACrD;AAAA,EACA;AACF,GAGG;AACD,MACE,SAAS,oBAAoB,aAC7B,MAAM,2BAA2B,MACjC;AACA;AAAA,EACF;AAEA,QAAM,eAAe,MACnB,sBAAsB,IAAI,IAAI,MAAM,SAAS,MAAM,cAAc,OAAO;AAE1E,QAAM,qBAAqB,MAAM;AAC/B,UAAM,0BAA0B;AAAA,EAClC;AAEA,QAAM,gBAAgB;AAEtB,QAAM,0BAA0B,cAAc,6BAA6B,EAAE;AAAA,IAC3E,cAAc,MACZ,aAAa,KAAK,SAAS,oBAAoB;AAAA,EACnD,CAAC,EAAE,KAAK,oBAAoB,kBAAkB;AAChD;AAEA,SAAS,wBAAsD;AAAA,EAC7D;AAAA,EACA;AACF,GAGG;AACD,MAAI,QAAQ,sBAAsB,IAAI,IAAI;AAE1C,MAAI,SAAS,MAAM;AACjB,UAAM,WAAiC,EAAE,eAAe,oBAAI,IAAI,EAAE;AAClE,0BAAsB,IAAI,MAAM,QAAQ;AACxC,YAAQ;AAER,QAAI,OAAO,aAAa,aAAa;AACnC,YAAM,qBAAqB,MACzB,gBAAgB,EAAE,MAAM,OAAO,SAAS,CAAC;AAE3C,eAAS,iBAAiB,oBAAoB,kBAAkB;AAChE,eAAS,4BAA4B,MAAM;AACzC,iBAAS,oBAAoB,oBAAoB,kBAAkB;AAAA,MACrE;AAAA,IACF;AAAA,EACF;AAEA,QAAM,EAAE,cAAc,IAAI;AAC1B,QAAM,eAAe,cAAc,SAAS;AAC5C,gBAAc,IAAI,YAAY;AAE9B,MACE,gBACA,KAAK,WAAW,eAChB,KAAK,WAAW,aAChB;AACA,SAAK,KAAK,aAAa;AAAA,EACzB;AAEA,SAAO,MAAM;AA/If;AAgJI,kBAAc,OAAO,YAAY;AAEjC,QAAI,cAAc,SAAS,GAAG;AAC5B,kBAAM,8BAAN;AACA,4BAAsB,OAAO,IAAI;AAAA,IACnC;AAAA,EACF;AACF;AACO,SAAS,QAAkD;AAAA,EAChE,uBAAuB;AAAA,EACvB,SAAS;AAAA,EACT,GAAG;AACL,IAAgC,CAAC,GAA+B;AAC9D,QAAM,kCAA8B,qBAAO,CAAC,CAAC;AAO7C,QAAM,gBAAY,qBAYhB,CAAC,CAAC;AAEJ,MAAI,EAAE,UAAU,UAAU;AACxB,cAAU,UAAU;AAAA,MAClB,YAAY,QAAQ;AAAA,MACpB,QAAQ,QAAQ;AAAA,MAChB,UAAU,QAAQ;AAAA,MAClB,SAAS,QAAQ;AAAA,MACjB,uBAAuB,QAAQ;AAAA,MAC/B,WAAW,QAAQ;AAAA,IACrB;AAAA,EACF;AAGA,MAAI;AACJ,QAAM,eAAe,MAAG;AA/L1B;AAgMI,2BAAU,QAAQ,cAAlB,YACC,iEAAqB,IAAI,gCAAiC;AAAA;AAG7D,QAAM,cAA8B;AAAA,IAClC,GAAG;AAAA,IACH,WAAW;AAAA,MACT,cAAc,iBAAe,aAAa,EAAE,aAAa,WAAW;AAAA,MACpE,mBAAmB,sBACjB,aAAa,EAAE,kBAAkB,gBAAgB;AAAA,IACrD;AAAA,IACA,YAAY,SAAI;AA3MpB;AA2MuB,mCAAU,SAAQ,eAAlB,4BAA+B;AAAA;AAAA,IAClD,QAAQ,SAAI;AA5MhB;AA4MmB,mCAAU,SAAQ,WAAlB,4BAA2B;AAAA;AAAA,IAC1C,UAAU,SAAI;AA7MlB;AA6MqB,mCAAU,SAAQ,aAAlB,4BAA6B;AAAA;AAAA,IAC9C,SAAS,SAAI;AA9MjB;AA8MoB,mCAAU,SAAQ,YAAlB,4BAA4B;AAAA;AAAA,IAC5C,uBAAuB,SAAI;AA/M/B;AAgNM,yCAAU,SAAQ,0BAAlB,4BAA0C,SAA1C,YAAkD;AAAA;AAAA,EACtD;AAEA,QAAM,cAAU;AAAA,IACd,UAAU,UAAU,QAAQ,OAAO,IAAI,KAAK,WAAW;AAAA,EACzD;AAEA,QAAM,qBACH,UAAU,WAAW,QAAQ,SAAS,QAAQ,WAC9C,QAAQ,WACP,QAAQ,MAAM,QACd,QAAQ,QAAQ,OAAO,QAAQ;AAEnC,MAAI,oBAAoB;AACtB,YAAQ,UAAU,UAAU,UAAU,QAAQ,OAAO,IAAI,KAAK,WAAW;AAAA,EAC3E;AAEA,QAAM,OAAO,QAAQ;AACrB,QAAM,0BAAsB,qBAAO;AAAA,IACjC;AAAA,IACA,UAAU,KAAK;AAAA,EACjB,CAAC;AAED,MAAI,oBAAoB,QAAQ,SAAS,MAAM;AAC7C,wBAAoB,UAAU,EAAE,MAAM,UAAU,KAAK,SAAS;AAAA,EAChE;AAEA,QAAM,0BAAsB;AAAA,IAC1B,CAAC,WAAuB;AACtB,UAAI,eAAe;AAEnB,YAAM,iBAAiB,MAAM;AAC3B,YAAI,CAAC,gBAAgB,oBAAoB,QAAQ,SAAS,MAAM;AAC9D;AAAA,QACF;AAEA,4BAAoB,UAAU,EAAE,MAAM,UAAU,KAAK,SAAS;AAC9D,eAAO;AAAA,MACT;AAEA,YAAM,cAAc,KAAK,2BAA2B;AAAA,QAClD;AAAA,QACA;AAAA,MACF;AAKA,0BAAoB,UAAU,EAAE,MAAM,UAAU,KAAK,SAAS;AAE9D,aAAO,MAAM;AACX,uBAAe;AACf,oBAAY;AAAA,MACd;AAAA,IACF;AAAA,IACA,CAAC,MAAM,cAAc;AAAA,EACvB;AAEA,QAAM,0BAAsB;AAAA,IAC1B,MAAM,oBAAoB,QAAQ;AAAA,IAClC,CAAC;AAAA,EACH;AAEA,QAAM,eAAW;AAAA,IACf;AAAA,IACA;AAAA,IACA;AAAA,EACF;AAEA,QAAM,wBAAoB;AAAA,IACxB,CAAC,WACC,KAAK,yBAAyB,EAAE,MAAM;AACpC,UAAI,oBAAoB,QAAQ,SAAS,MAAM;AAC7C;AAAA,MACF;AAEA,UAAI,KAAK,WAAW,WAAW,KAAK,WAAW,SAAS;AAEtD,4BAAoB,UAAU,EAAE,MAAM,UAAU,KAAK,SAAS;AAAA,MAChE;AAEA,aAAO;AAAA,IACT,CAAC;AAAA,IACH,CAAC,IAAI;AAAA,EACP;AAEA,QAAM,wBAAoB,0BAAY,MAAM,KAAK,QAAQ,CAAC,IAAI,CAAC;AAE/D,QAAM,aAAS;AAAA,IACb;AAAA,IACA;AAAA,IACA;AAAA,EACF;AAEA,QAAM,YAAQ;AAAA,IACZ,QAAQ,QAAQ,wBAAwB;AAAA,IACxC,MAAM,QAAQ,QAAQ;AAAA,IACtB,MAAM,QAAQ,QAAQ;AAAA,EACxB;AAEA,QAAM,kBAAc;AAAA,IAClB,CACE,kBACG;AACH,UAAI,OAAO,kBAAkB,YAAY;AACvC,wBAAgB,cAAc,QAAQ,QAAQ,QAAQ;AAAA,MACxD;AACA,cAAQ,QAAQ,WAAW;AAAA,IAC7B;AAAA,IACA,CAAC,OAAO;AAAA,EACV;AAEA,8BAAU,MAAM;AACd,QAAI,QAAQ;AACV,aAAO,wBAAwB;AAAA,QAC7B;AAAA,QACA,cAAc,4BAA4B;AAAA,MAC5C,CAAC;AAAA,IACH;AAAA,EACF,GAAG,CAAC,QAAQ,IAAI,CAAC;AAEjB,SAAO;AAAA,IACL,IAAI,QAAQ,QAAQ;AAAA,IACpB;AAAA,IACA;AAAA,IACA,aAAa,QAAQ,QAAQ;AAAA,IAC7B,YAAY,QAAQ,QAAQ;AAAA,IAC5B,YAAY,QAAQ,QAAQ;AAAA,IAC5B,MAAM,QAAQ,QAAQ;AAAA,IACtB;AAAA,IACA,cAAc,QAAQ,QAAQ;AAAA,IAC9B;AAAA;AAAA;AAAA;AAAA,IAIA,eAAe,QAAQ,QAAQ;AAAA,IAC/B,eAAe,QAAQ,QAAQ;AAAA,IAC/B,yBAAyB,QAAQ,QAAQ;AAAA,EAC3C;AACF;;;AG3VA,IAAAC,aAIO;AACP,IAAAC,gBAAgE;AAChE,iBAAmB;AAyDZ,SAAS,cAAc;AAAA,EAC5B,MAAM;AAAA,EACN;AAAA,EACA,oBAAoB;AAAA,EACpB,eAAe;AAAA,EACf;AAAA,EACA;AAAA,EACA;AAAA,EACA,iBAAiB;AAAA,EACjB,OAAAC;AAAA,EACA;AAAA,EACA;AAAA,EACA,uBAAuB;AACzB,IAMI,CAAC,GAAyB;AAE5B,QAAM,aAAS,qBAAM;AACrB,QAAM,eAAe,MAAM;AAG3B,QAAM,EAAE,MAAM,OAAO,QAAI,WAAAC,SAAe,CAAC,KAAK,YAAY,GAAG,MAAM;AAAA,IACjE,cAAc;AAAA,EAChB,CAAC;AAED,QAAM,EAAE,MAAM,YAAY,OAAO,QAAQ,cAAc,QAAI,WAAAA;AAAA,IACzD,CAAC,cAAc,SAAS;AAAA,IACxB;AAAA,EACF;AAEA,QAAM,CAAC,OAAO,QAAQ,QAAI,wBAA4B,MAAS;AAC/D,QAAM,aAAa;AAGnB,QAAM,CAAC,iBAAiB,kBAAkB,QACxC,wBAAiC,IAAI;AAEvC,QAAM,uBAAmB,sBAAO;AAAA,IAC9B;AAAA,IACA;AAAA,IACA;AAAA,EACF,CAAC;AAED,+BAAU,MAAM;AACd,qBAAiB,UAAU;AAAA,MACzB;AAAA,MACA;AAAA,MACA;AAAA,IACF;AAAA,EACF,GAAG,CAAC,aAAa,SAAS,IAAI,CAAC;AAE/B,QAAM,qBAAiB;AAAA,IACrB,OAAO,QAAgB,gBACrB,8BAAkB;AAAA,MAChB;AAAA,MACA;AAAA,MACA,aAAa,iBAAiB,QAAQ;AAAA,MACtC,SAAS,EAAE,GAAG,iBAAiB,QAAQ,SAAS,GAAG,mCAAS,QAAQ;AAAA,MACpE,MAAM;AAAA,QACJ,GAAG,iBAAiB,QAAQ;AAAA,QAC5B,GAAG,mCAAS;AAAA,MACd;AAAA,MACA;AAAA,MACA,OAAAD;AAAA;AAAA,MAEA,eAAe;AAAA,QACb,CAACE,gBAAuB,OAAOA,aAAY,KAAK;AAAA,QAChD;AAAA,MACF;AAAA,MACA,YAAY;AAAA,MACZ;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,IACF,CAAC;AAAA,IACH;AAAA,MACE;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACAF;AAAA,MACA;AAAA,IACF;AAAA,EACF;AAEA,QAAM,WAAO,2BAAY,MAAM;AAC7B,QAAI,iBAAiB;AACnB,sBAAgB,MAAM;AACtB,yBAAmB,IAAI;AAAA,IACzB;AAAA,EACF,GAAG,CAAC,eAAe,CAAC;AAEpB,QAAM,oBAAgB;AAAA,IACpB,CAACE,gBAAuB;AACtB,aAAOA,aAAY,KAAK;AAAA,IAC1B;AAAA,IACA,CAAC,MAAM;AAAA,EACT;AAEA,QAAM,eAAW;AAAA,IACf,OAAO,QAAQ,YAAY;AACzB,aAAO,eAAe,QAAQ,OAAO;AAAA,IACvC;AAAA,IACA,CAAC,cAAc;AAAA,EACjB;AAEA,QAAM,CAAC,OAAO,QAAQ,QAAI,wBAAS,YAAY;AAE/C,QAAM,mBAAe;AAAA,IACnB,CAAC,UAA4C;AArLjD;AAsLM,2CAAO,mBAAP;AACA,UAAI,CAAC,OAAO;AACV;AAAA,MACF;AAEA,YAAM,SAAS,SAAS,KAAK;AAC7B,eAAS,EAAE;AACX,aAAO;AAAA,IACT;AAAA,IACA,CAAC,OAAO,UAAU,QAAQ;AAAA,EAC5B;AAEA,QAAM,wBAAoB;AAAA,IACxB,CAAC,MAAW;AACV,eAAS,EAAE,OAAO,KAAK;AAAA,IACzB;AAAA,IACA,CAAC,QAAQ;AAAA,EACX;AAEA,SAAO;AAAA,IACL;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF;AACF;;;ACrNA,4BASO;AACP,IAAAC,aAKO;AACP,IAAAC,gBAAqD;AACrD,IAAAC,cAAmB;AAGnB,IAAM,mBAAmB,MAAM;AAsG/B,SAAS,UAIP;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA;AAAA,EACA;AAAA,EACA,OAAAC;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AACF,GAGiD;AAE/C,QAAM,aAAS,qBAAM;AACrB,QAAM,eAAe,kBAAM;AAG3B,QAAM,EAAE,MAAM,OAAO,QAAI,YAAAC;AAAA,IACvB,CAAC,cAAc,QAAQ;AAAA,IACvB;AAAA,IACA,EAAE,cAAc,aAAa;AAAA,EAC/B;AAEA,QAAM,CAAC,OAAO,QAAQ,QAAI,wBAA4B,MAAS;AAC/D,QAAM,CAAC,WAAW,YAAY,QAAI,wBAAS,KAAK;AAGhD,QAAM,yBAAqB,sBAA+B,IAAI;AAE9D,QAAM,WAAO,2BAAY,MAAM;AA7JjC;AA8JI,QAAI;AACF,+BAAmB,YAAnB,mBAA4B;AAAA,IAC9B,SAAS,SAAS;AAAA,IAClB,UAAE;AACA,mBAAa,KAAK;AAClB,yBAAmB,UAAU;AAAA,IAC/B;AAAA,EACF,GAAG,CAAC,CAAC;AAEL,QAAM,SAAS,OAAO,UAAiB;AAvKzC;AAwKI,UAAM,kBAAkB,IAAI,gBAAgB;AAE5C,QAAI;AACF,kBAAY;AAEZ,mBAAa,IAAI;AAEjB,yBAAmB,UAAU;AAG7B,YAAM,kBAAkB,UAAM,+BAAQ,OAAO;AAE7C,YAAM,cAAcD,UAAA,OAAAA,SAAS,iBAAiB;AAC9C,YAAM,WAAW,MAAM,YAAY,KAAK;AAAA,QACtC,QAAQ;AAAA,QACR,SAAS;AAAA,UACP,gBAAgB;AAAA,UAChB,OAAG,wCAAiB,eAAe;AAAA,QACrC;AAAA,QACA;AAAA,QACA,QAAQ,gBAAgB;AAAA,QACxB,MAAM,KAAK,UAAU,KAAK;AAAA,MAC5B,CAAC;AAED,UAAI,CAAC,SAAS,IAAI;AAChB,cAAM,IAAI;AAAA,WACP,WAAM,SAAS,KAAK,MAApB,YAA0B;AAAA,QAC7B;AAAA,MACF;AAEA,UAAI,SAAS,QAAQ,MAAM;AACzB,cAAM,IAAI,MAAM,6BAA6B;AAAA,MAC/C;AAEA,UAAI,kBAAkB;AACtB,UAAI,eAAgD;AAEpD,YAAM,SAAS,KAAK,YAAY,IAAI,kBAAkB,CAAC,EAAE;AAAA,QACvD,IAAI,eAAuB;AAAA,UACzB,MAAM,MAAM,OAAO;AACjB,+BAAmB;AAEnB,kBAAM,EAAE,MAAM,IAAI,UAAM,6BAAiB,eAAe;AACxD,kBAAM,gBAAgB;AAEtB,gBAAI,KAAC,4BAAgB,cAAc,aAAa,GAAG;AACjD,6BAAe;AAEf,kBAAI,mBAAmB,YAAY,iBAAiB;AAClD,uBAAO,aAAa;AAAA,cACtB;AAAA,YACF;AAAA,UACF;AAAA,UAEA,MAAM,QAAQ;AACZ,gBAAI,mBAAmB,YAAY,iBAAiB;AAClD,2BAAa,KAAK;AAAA,YACpB;AAEA,gBAAI,YAAY,MAAM;AACpB,oBAAM,mBAAmB,UAAM,yCAAkB;AAAA,gBAC/C,OAAO;AAAA,gBACP,YAAQ,qBAAS,MAAM;AAAA,cACzB,CAAC;AAED;AAAA,gBACE,iBAAiB,UACb,EAAE,QAAQ,iBAAiB,OAAO,OAAO,OAAU,IACnD,EAAE,QAAQ,QAAW,OAAO,iBAAiB,MAAM;AAAA,cACzD;AAAA,YACF;AAEA,gBAAI,mBAAmB,YAAY,iBAAiB;AAClD,iCAAmB,UAAU;AAAA,YAC/B;AAAA,UACF;AAAA,QACF,CAAC;AAAA,MACH;AAAA,IACF,SAASE,QAAO;AACd,cAAI,oCAAaA,MAAK,GAAG;AACvB;AAAA,MACF;AAEA,UAAI,WAAWA,kBAAiB,OAAO;AACrC,gBAAQA,MAAK;AAAA,MACf;AAEA,UAAI,mBAAmB,YAAY,iBAAiB;AAClD,qBAAa,KAAK;AAClB,2BAAmB,UAAU;AAC7B,iBAASA,kBAAiB,QAAQA,SAAQ,IAAI,MAAM,OAAOA,MAAK,CAAC,CAAC;AAAA,MACpE;AAAA,IACF;AAAA,EACF;AAEA,QAAM,QAAQ,MAAM;AAClB,SAAK;AACL,gBAAY;AAAA,EACd;AAEA,QAAM,cAAc,MAAM;AACxB,aAAS,MAAS;AAClB,iBAAa,KAAK;AAClB,WAAO,MAAS;AAAA,EAClB;AAEA,SAAO;AAAA,IACL;AAAA,IACA,QAAQ;AAAA,IACR;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF;AACF;AAEO,IAAM,yBAAyB;","names":["import_ai","throttleFunction","import_ai","import_react","fetch","useSWR","completion","import_ai","import_react","import_swr","fetch","useSWR","error"]}
|
package/dist/index.mjs
CHANGED
|
@@ -159,11 +159,61 @@ var Chat = class extends AbstractChat {
|
|
|
159
159
|
_state = new WeakMap();
|
|
160
160
|
|
|
161
161
|
// src/use-chat.ts
|
|
162
|
+
var automaticResumeStates = /* @__PURE__ */ new WeakMap();
|
|
163
|
+
function resumeOnVisible({
|
|
164
|
+
chat,
|
|
165
|
+
state
|
|
166
|
+
}) {
|
|
167
|
+
if (document.visibilityState !== "visible" || state.pendingVisibilityResume != null) {
|
|
168
|
+
return;
|
|
169
|
+
}
|
|
170
|
+
const isRegistered = () => automaticResumeStates.get(chat) === state && state.registrations.size > 0;
|
|
171
|
+
const clearPendingResume = () => {
|
|
172
|
+
state.pendingVisibilityResume = void 0;
|
|
173
|
+
};
|
|
174
|
+
const resumableChat = chat;
|
|
175
|
+
state.pendingVisibilityResume = resumableChat["~resumeStreamIfDisconnected"]({
|
|
176
|
+
shouldResume: () => isRegistered() && document.visibilityState === "visible"
|
|
177
|
+
}).then(clearPendingResume, clearPendingResume);
|
|
178
|
+
}
|
|
179
|
+
function registerAutomaticResume({
|
|
180
|
+
chat,
|
|
181
|
+
registration
|
|
182
|
+
}) {
|
|
183
|
+
let state = automaticResumeStates.get(chat);
|
|
184
|
+
if (state == null) {
|
|
185
|
+
const newState = { registrations: /* @__PURE__ */ new Set() };
|
|
186
|
+
automaticResumeStates.set(chat, newState);
|
|
187
|
+
state = newState;
|
|
188
|
+
if (typeof document !== "undefined") {
|
|
189
|
+
const onVisibilityChange = () => resumeOnVisible({ chat, state: newState });
|
|
190
|
+
document.addEventListener("visibilitychange", onVisibilityChange);
|
|
191
|
+
newState.cleanupVisibilityListener = () => {
|
|
192
|
+
document.removeEventListener("visibilitychange", onVisibilityChange);
|
|
193
|
+
};
|
|
194
|
+
}
|
|
195
|
+
}
|
|
196
|
+
const { registrations } = state;
|
|
197
|
+
const shouldResume = registrations.size === 0;
|
|
198
|
+
registrations.add(registration);
|
|
199
|
+
if (shouldResume && chat.status !== "submitted" && chat.status !== "streaming") {
|
|
200
|
+
void chat.resumeStream();
|
|
201
|
+
}
|
|
202
|
+
return () => {
|
|
203
|
+
var _a;
|
|
204
|
+
registrations.delete(registration);
|
|
205
|
+
if (registrations.size === 0) {
|
|
206
|
+
(_a = state.cleanupVisibilityListener) == null ? void 0 : _a.call(state);
|
|
207
|
+
automaticResumeStates.delete(chat);
|
|
208
|
+
}
|
|
209
|
+
};
|
|
210
|
+
}
|
|
162
211
|
function useChat({
|
|
163
212
|
experimental_throttle: throttleWaitMs,
|
|
164
213
|
resume = false,
|
|
165
214
|
...options
|
|
166
215
|
} = {}) {
|
|
216
|
+
const automaticResumeRegistration = useRef({});
|
|
167
217
|
const latestRef = useRef({});
|
|
168
218
|
if (!("chat" in options)) {
|
|
169
219
|
latestRef.current = {
|
|
@@ -287,9 +337,12 @@ function useChat({
|
|
|
287
337
|
);
|
|
288
338
|
useEffect(() => {
|
|
289
339
|
if (resume) {
|
|
290
|
-
|
|
340
|
+
return registerAutomaticResume({
|
|
341
|
+
chat,
|
|
342
|
+
registration: automaticResumeRegistration.current
|
|
343
|
+
});
|
|
291
344
|
}
|
|
292
|
-
}, [resume,
|
|
345
|
+
}, [resume, chat]);
|
|
293
346
|
return {
|
|
294
347
|
id: chatRef.current.id,
|
|
295
348
|
messages,
|
package/dist/index.mjs.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"sources":["../src/use-chat.ts","../src/chat.react.ts","../src/throttle.ts","../src/use-completion.ts","../src/use-object.ts"],"sourcesContent":["import {\n type AbstractChat,\n type ChatInit,\n type ChatTransport,\n type CreateUIMessage,\n type UIMessage,\n DefaultChatTransport,\n} from 'ai';\nimport { useCallback, useEffect, useRef, useSyncExternalStore } from 'react';\nimport { Chat } from './chat.react';\n\nexport type { CreateUIMessage, UIMessage };\n\nexport type UseChatHelpers<UI_MESSAGE extends UIMessage> = {\n /**\n * The id of the chat.\n */\n readonly id: string;\n\n /**\n * Update the `messages` state locally. This is useful when you want to\n * edit the messages on the client, and then trigger the `reload` method\n * manually to regenerate the AI response.\n */\n setMessages: (\n messages: UI_MESSAGE[] | ((messages: UI_MESSAGE[]) => UI_MESSAGE[]),\n ) => void;\n\n error: Error | undefined;\n} & Pick<\n AbstractChat<UI_MESSAGE>,\n | 'sendMessage'\n | 'regenerate'\n | 'stop'\n | 'resumeStream'\n | 'addToolResult'\n | 'addToolOutput'\n | 'addToolApprovalResponse'\n | 'status'\n | 'messages'\n | 'clearError'\n>;\n\nexport type UseChatOptions<UI_MESSAGE extends UIMessage> = (\n | { chat: Chat<UI_MESSAGE> }\n | ChatInit<UI_MESSAGE>\n) & {\n /**\n * Custom throttle wait in ms for the chat messages and data updates.\n * Default is undefined, which disables throttling.\n */\n experimental_throttle?: number;\n\n /**\n * Whether to resume an ongoing chat generation stream.\n */\n resume?: boolean;\n};\n\nexport function useChat<UI_MESSAGE extends UIMessage = UIMessage>({\n experimental_throttle: throttleWaitMs,\n resume = false,\n ...options\n}: UseChatOptions<UI_MESSAGE> = {}): UseChatHelpers<UI_MESSAGE> {\n // the Chat instance is created once and not recreated when options change,\n // so it would normally keep the callbacks/transport from the first render forever\n\n // keep latest values in a ref that is refreshed on every render,\n // and hand `Chat` stable wrappers that read from it to avoid stale closures\n const latestRef = useRef<\n Partial<\n Pick<\n ChatInit<UI_MESSAGE>,\n | 'onToolCall'\n | 'onData'\n | 'onFinish'\n | 'onError'\n | 'sendAutomaticallyWhen'\n | 'transport'\n >\n >\n >({});\n\n if (!('chat' in options)) {\n latestRef.current = {\n onToolCall: options.onToolCall,\n onData: options.onData,\n onFinish: options.onFinish,\n onError: options.onError,\n sendAutomaticallyWhen: options.sendAutomaticallyWhen,\n transport: options.transport,\n };\n }\n\n // resolve the latest transport and fallback to a lazily created default transport\n let defaultTransport: ChatTransport<UI_MESSAGE> | undefined;\n const getTransport = () =>\n latestRef.current.transport ??\n (defaultTransport ??= new DefaultChatTransport<UI_MESSAGE>());\n\n // give `Chat` stable wrappers that always read the latest values from `latestRef`\n const chatOptions: typeof options = {\n ...options,\n transport: {\n sendMessages: sendOptions => getTransport().sendMessages(sendOptions),\n reconnectToStream: reconnectOptions =>\n getTransport().reconnectToStream(reconnectOptions),\n },\n onToolCall: arg => latestRef.current.onToolCall?.(arg),\n onData: arg => latestRef.current.onData?.(arg),\n onFinish: arg => latestRef.current.onFinish?.(arg),\n onError: arg => latestRef.current.onError?.(arg),\n sendAutomaticallyWhen: arg =>\n latestRef.current.sendAutomaticallyWhen?.(arg) ?? false,\n };\n\n const chatRef = useRef<Chat<UI_MESSAGE>>(\n 'chat' in options ? options.chat : new Chat(chatOptions),\n );\n\n const shouldRecreateChat =\n ('chat' in options && options.chat !== chatRef.current) ||\n ('id' in options &&\n options.id != null &&\n chatRef.current.id !== options.id);\n\n if (shouldRecreateChat) {\n chatRef.current = 'chat' in options ? options.chat : new Chat(chatOptions);\n }\n\n const chat = chatRef.current;\n const messagesSnapshotRef = useRef({\n chat,\n messages: chat.messages,\n });\n\n if (messagesSnapshotRef.current.chat !== chat) {\n messagesSnapshotRef.current = { chat, messages: chat.messages };\n }\n\n const subscribeToMessages = useCallback(\n (update: () => void) => {\n let isSubscribed = true;\n\n const updateMessages = () => {\n if (!isSubscribed || messagesSnapshotRef.current.chat !== chat) {\n return;\n }\n\n messagesSnapshotRef.current = { chat, messages: chat.messages };\n update();\n };\n\n const unsubscribe = chat['~registerMessagesCallback'](\n updateMessages,\n throttleWaitMs,\n );\n\n // Synchronize changes that may have happened between render and\n // subscription. useSyncExternalStore checks the snapshot after\n // subscribing and schedules a render when it changed.\n messagesSnapshotRef.current = { chat, messages: chat.messages };\n\n return () => {\n isSubscribed = false;\n unsubscribe();\n };\n },\n [chat, throttleWaitMs],\n );\n\n const getMessagesSnapshot = useCallback(\n () => messagesSnapshotRef.current.messages,\n [],\n );\n\n const messages = useSyncExternalStore(\n subscribeToMessages,\n getMessagesSnapshot,\n getMessagesSnapshot,\n );\n\n const subscribeToStatus = useCallback(\n (update: () => void) =>\n chat['~registerStatusCallback'](() => {\n if (messagesSnapshotRef.current.chat !== chat) {\n return;\n }\n\n if (chat.status === 'ready' || chat.status === 'error') {\n // Publish the latest messages before the terminal status can render.\n messagesSnapshotRef.current = { chat, messages: chat.messages };\n }\n\n update();\n }),\n [chat],\n );\n\n const getStatusSnapshot = useCallback(() => chat.status, [chat]);\n\n const status = useSyncExternalStore(\n subscribeToStatus,\n getStatusSnapshot,\n getStatusSnapshot,\n );\n\n const error = useSyncExternalStore(\n chatRef.current['~registerErrorCallback'],\n () => chatRef.current.error,\n () => chatRef.current.error,\n );\n\n const setMessages = useCallback(\n (\n messagesParam: UI_MESSAGE[] | ((messages: UI_MESSAGE[]) => UI_MESSAGE[]),\n ) => {\n if (typeof messagesParam === 'function') {\n messagesParam = messagesParam(chatRef.current.messages);\n }\n chatRef.current.messages = messagesParam;\n },\n [chatRef],\n );\n\n useEffect(() => {\n if (resume) {\n chatRef.current.resumeStream();\n }\n }, [resume, chatRef]);\n\n return {\n id: chatRef.current.id,\n messages,\n setMessages,\n sendMessage: chatRef.current.sendMessage,\n regenerate: chatRef.current.regenerate,\n clearError: chatRef.current.clearError,\n stop: chatRef.current.stop,\n error,\n resumeStream: chatRef.current.resumeStream,\n status,\n /**\n * @deprecated Use `addToolOutput` instead.\n */\n addToolResult: chatRef.current.addToolOutput,\n addToolOutput: chatRef.current.addToolOutput,\n addToolApprovalResponse: chatRef.current.addToolApprovalResponse,\n };\n}\n","import {\n AbstractChat,\n type ChatInit,\n type ChatState,\n type ChatStatus,\n type UIMessage,\n} from 'ai';\nimport { throttle } from './throttle';\n\nfunction cloneMetadata<METADATA>(metadata: METADATA): METADATA {\n if (Array.isArray(metadata)) {\n return [...metadata] as METADATA;\n }\n\n if (\n metadata != null &&\n typeof metadata === 'object' &&\n (Object.getPrototypeOf(metadata) === Object.prototype ||\n Object.getPrototypeOf(metadata) === null)\n ) {\n return { ...metadata } as METADATA;\n }\n\n return metadata;\n}\n\nclass ReactChatState<\n UI_MESSAGE extends UIMessage,\n> implements ChatState<UI_MESSAGE> {\n #messages: UI_MESSAGE[];\n #status: ChatStatus = 'ready';\n #error: Error | undefined = undefined;\n\n #messagesCallbacks = new Set<() => void>();\n #statusCallbacks = new Set<() => void>();\n #errorCallbacks = new Set<() => void>();\n\n constructor(initialMessages: UI_MESSAGE[] = []) {\n this.#messages = initialMessages;\n }\n\n get status(): ChatStatus {\n return this.#status;\n }\n\n set status(newStatus: ChatStatus) {\n this.#status = newStatus;\n this.#callStatusCallbacks();\n }\n\n get error(): Error | undefined {\n return this.#error;\n }\n\n set error(newError: Error | undefined) {\n this.#error = newError;\n this.#callErrorCallbacks();\n }\n\n get messages(): UI_MESSAGE[] {\n return this.#messages;\n }\n\n set messages(newMessages: UI_MESSAGE[]) {\n this.#messages = [...newMessages];\n this.#callMessagesCallbacks();\n }\n\n pushMessage = (message: UI_MESSAGE) => {\n this.#messages = this.#messages.concat(message);\n this.#callMessagesCallbacks();\n };\n\n popMessage = () => {\n this.#messages = this.#messages.slice(0, -1);\n this.#callMessagesCallbacks();\n };\n\n replaceMessage = (index: number, message: UI_MESSAGE) => {\n this.#messages = [\n ...this.#messages.slice(0, index),\n this.snapshot(message),\n ...this.#messages.slice(index + 1),\n ];\n this.#callMessagesCallbacks();\n };\n\n snapshot = <T>(value: T): T => {\n if (\n value == null ||\n typeof value !== 'object' ||\n !('parts' in value) ||\n !Array.isArray(value.parts)\n ) {\n return value;\n }\n\n const message = value as UI_MESSAGE;\n const snapshot = {\n ...message,\n parts: message.parts.map(part => ({ ...part })),\n };\n\n if ('metadata' in message) {\n snapshot.metadata = cloneMetadata(message.metadata);\n }\n\n return snapshot as T;\n };\n\n '~registerMessagesCallback' = (\n onChange: () => void,\n throttleWaitMs?: number,\n ): (() => void) => {\n const callback = throttleWaitMs\n ? throttle(onChange, throttleWaitMs)\n : onChange;\n this.#messagesCallbacks.add(callback);\n return () => {\n this.#messagesCallbacks.delete(callback);\n };\n };\n\n '~registerStatusCallback' = (onChange: () => void): (() => void) => {\n this.#statusCallbacks.add(onChange);\n return () => {\n this.#statusCallbacks.delete(onChange);\n };\n };\n\n '~registerErrorCallback' = (onChange: () => void): (() => void) => {\n this.#errorCallbacks.add(onChange);\n return () => {\n this.#errorCallbacks.delete(onChange);\n };\n };\n\n #callMessagesCallbacks = () => {\n this.#messagesCallbacks.forEach(callback => callback());\n };\n\n #callStatusCallbacks = () => {\n this.#statusCallbacks.forEach(callback => callback());\n };\n\n #callErrorCallbacks = () => {\n this.#errorCallbacks.forEach(callback => callback());\n };\n}\n\nexport class Chat<\n UI_MESSAGE extends UIMessage,\n> extends AbstractChat<UI_MESSAGE> {\n #state: ReactChatState<UI_MESSAGE>;\n\n constructor({ messages, ...init }: ChatInit<UI_MESSAGE>) {\n const state = new ReactChatState(messages);\n super({ ...init, state });\n this.#state = state;\n }\n\n '~registerMessagesCallback' = (\n onChange: () => void,\n throttleWaitMs?: number,\n ): (() => void) =>\n this.#state['~registerMessagesCallback'](onChange, throttleWaitMs);\n\n '~registerStatusCallback' = (onChange: () => void): (() => void) =>\n this.#state['~registerStatusCallback'](onChange);\n\n '~registerErrorCallback' = (onChange: () => void): (() => void) =>\n this.#state['~registerErrorCallback'](onChange);\n}\n","import throttleFunction from 'throttleit';\n\nexport function throttle<T extends (...args: any[]) => any>(\n fn: T,\n waitMs: number | undefined,\n): T {\n return waitMs != null ? throttleFunction(fn, waitMs) : fn;\n}\n","import {\n callCompletionApi,\n type CompletionRequestOptions,\n type UseCompletionOptions,\n} from 'ai';\nimport { useCallback, useEffect, useId, useRef, useState } from 'react';\nimport useSWR from 'swr';\nimport { throttle } from './throttle';\n\nexport type { UseCompletionOptions };\n\nexport type UseCompletionHelpers = {\n /** The current completion result */\n completion: string;\n /**\n * Send a new prompt to the API endpoint and update the completion state.\n */\n complete: (\n prompt: string,\n options?: CompletionRequestOptions,\n ) => Promise<string | null | undefined>;\n /** The error object of the API request */\n error: undefined | Error;\n /**\n * Abort the current API request but keep the generated tokens.\n */\n stop: () => void;\n /**\n * Update the `completion` state locally.\n */\n setCompletion: (completion: string) => void;\n /** The current value of the input */\n input: string;\n /** setState-powered method to update the input value */\n setInput: React.Dispatch<React.SetStateAction<string>>;\n /**\n * An input/textarea-ready onChange handler to control the value of the input\n * @example\n * ```jsx\n * <input onChange={handleInputChange} value={input} />\n * ```\n */\n handleInputChange: (\n event:\n | React.ChangeEvent<HTMLInputElement>\n | React.ChangeEvent<HTMLTextAreaElement>,\n ) => void;\n\n /**\n * Form submission handler to automatically reset input and append a user message\n * @example\n * ```jsx\n * <form onSubmit={handleSubmit}>\n * <input onChange={handleInputChange} value={input} />\n * </form>\n * ```\n */\n handleSubmit: (event?: { preventDefault?: () => void }) => void;\n\n /** Whether the API request is in progress */\n isLoading: boolean;\n};\n\nexport function useCompletion({\n api = '/api/completion',\n id,\n initialCompletion = '',\n initialInput = '',\n credentials,\n headers,\n body,\n streamProtocol = 'data',\n fetch,\n onFinish,\n onError,\n experimental_throttle: throttleWaitMs,\n}: UseCompletionOptions & {\n /**\n * Custom throttle wait in ms for the completion and data updates.\n * Default is undefined, which disables throttling.\n */\n experimental_throttle?: number;\n} = {}): UseCompletionHelpers {\n // Generate an unique id for the completion if not provided.\n const hookId = useId();\n const completionId = id || hookId;\n\n // Store the completion state in SWR, using the completionId as the key to share states.\n const { data, mutate } = useSWR<string>([api, completionId], null, {\n fallbackData: initialCompletion,\n });\n\n const { data: isLoading = false, mutate: mutateLoading } = useSWR<boolean>(\n [completionId, 'loading'],\n null,\n );\n\n const [error, setError] = useState<undefined | Error>(undefined);\n const completion = data!;\n\n // Abort controller to cancel the current API call.\n const [abortController, setAbortController] =\n useState<AbortController | null>(null);\n\n const extraMetadataRef = useRef({\n credentials,\n headers,\n body,\n });\n\n useEffect(() => {\n extraMetadataRef.current = {\n credentials,\n headers,\n body,\n };\n }, [credentials, headers, body]);\n\n const triggerRequest = useCallback(\n async (prompt: string, options?: CompletionRequestOptions) =>\n callCompletionApi({\n api,\n prompt,\n credentials: extraMetadataRef.current.credentials,\n headers: { ...extraMetadataRef.current.headers, ...options?.headers },\n body: {\n ...extraMetadataRef.current.body,\n ...options?.body,\n },\n streamProtocol,\n fetch,\n // throttle streamed ui updates:\n setCompletion: throttle(\n (completion: string) => mutate(completion, false),\n throttleWaitMs,\n ),\n setLoading: mutateLoading,\n setError,\n setAbortController,\n onFinish,\n onError,\n }),\n [\n mutate,\n mutateLoading,\n api,\n extraMetadataRef,\n setAbortController,\n onFinish,\n onError,\n setError,\n streamProtocol,\n fetch,\n throttleWaitMs,\n ],\n );\n\n const stop = useCallback(() => {\n if (abortController) {\n abortController.abort();\n setAbortController(null);\n }\n }, [abortController]);\n\n const setCompletion = useCallback(\n (completion: string) => {\n mutate(completion, false);\n },\n [mutate],\n );\n\n const complete = useCallback<UseCompletionHelpers['complete']>(\n async (prompt, options) => {\n return triggerRequest(prompt, options);\n },\n [triggerRequest],\n );\n\n const [input, setInput] = useState(initialInput);\n\n const handleSubmit = useCallback(\n (event?: { preventDefault?: () => void }) => {\n event?.preventDefault?.();\n if (!input) {\n return;\n }\n\n const result = complete(input);\n setInput('');\n return result;\n },\n [input, complete, setInput],\n );\n\n const handleInputChange = useCallback(\n (e: any) => {\n setInput(e.target.value);\n },\n [setInput],\n );\n\n return {\n completion,\n complete,\n error,\n setCompletion,\n stop,\n input,\n setInput,\n handleInputChange,\n handleSubmit,\n isLoading,\n };\n}\n","import {\n isAbortError,\n resolve,\n normalizeHeaders,\n safeValidateTypes,\n type FetchFunction,\n type FlexibleSchema,\n type InferSchema,\n type Resolvable,\n} from '@ai-sdk/provider-utils';\nimport {\n asSchema,\n isDeepEqualData,\n parsePartialJson,\n type DeepPartial,\n} from 'ai';\nimport { useCallback, useId, useRef, useState } from 'react';\nimport useSWR from 'swr';\n\n// use function to allow for mocking in tests:\nconst getOriginalFetch = () => fetch;\n\nexport type Experimental_UseObjectOptions<\n SCHEMA extends FlexibleSchema,\n RESULT,\n> = {\n /**\n * The API endpoint. It should stream JSON that matches the schema as chunked text.\n */\n api: string;\n\n /**\n * A schema that defines the shape of the complete object.\n */\n schema: SCHEMA;\n\n /**\n * An unique identifier. If not provided, a random one will be\n * generated. When provided, the `useObject` hook with the same `id` will\n * have shared states across components.\n */\n id?: string;\n\n /**\n * An optional value for the initial object.\n */\n initialValue?: DeepPartial<RESULT>;\n\n /**\n * Custom fetch implementation. You can use it as a middleware to intercept requests,\n * or to provide a custom fetch implementation for e.g. testing.\n */\n fetch?: FetchFunction;\n\n /**\n * Callback that is called when the stream has finished.\n */\n onFinish?: (event: {\n /**\n * The generated object (typed according to the schema).\n * Can be undefined if the final object does not match the schema.\n */\n object: RESULT | undefined;\n\n /**\n * Optional error object. This is e.g. a TypeValidationError when the final object does not match the schema.\n */\n error: Error | undefined;\n }) => Promise<void> | void;\n\n /**\n * Callback function to be called when an error is encountered.\n */\n onError?: (error: Error) => void;\n\n /**\n * Additional HTTP headers to be included in the request.\n * Can be a static object, a function that returns headers, or an async function\n * for dynamic auth tokens.\n */\n headers?: Resolvable<Record<string, string> | Headers>;\n\n /**\n * The credentials mode to be used for the fetch request.\n * Possible values are: 'omit', 'same-origin', 'include'.\n * Defaults to 'same-origin'.\n */\n credentials?: RequestCredentials;\n};\n\nexport type Experimental_UseObjectHelpers<RESULT, INPUT> = {\n /**\n * Calls the API with the provided input as JSON body.\n */\n submit: (input: INPUT) => void;\n\n /**\n * The current value for the generated object. Updated as the API streams JSON chunks.\n */\n object: DeepPartial<RESULT> | undefined;\n\n /**\n * The error object of the API request if any.\n */\n error: Error | undefined;\n\n /**\n * Flag that indicates whether an API request is in progress.\n */\n isLoading: boolean;\n\n /**\n * Abort the current request immediately, keep the current partial object if any.\n */\n stop: () => void;\n\n /**\n * Clear the object state.\n */\n clear: () => void;\n};\n\nfunction useObject<\n SCHEMA extends FlexibleSchema,\n RESULT = InferSchema<SCHEMA>,\n INPUT = any,\n>({\n api,\n id,\n schema, // required, in the future we will use it for validation\n initialValue,\n fetch,\n onError,\n onFinish,\n headers,\n credentials,\n}: Experimental_UseObjectOptions<\n SCHEMA,\n RESULT\n>): Experimental_UseObjectHelpers<RESULT, INPUT> {\n // Generate an unique id if not provided.\n const hookId = useId();\n const completionId = id ?? hookId;\n\n // Store the completion state in SWR, using the completionId as the key to share states.\n const { data, mutate } = useSWR<DeepPartial<RESULT>>(\n [completionId, 'object'],\n null,\n { fallbackData: initialValue },\n );\n\n const [error, setError] = useState<undefined | Error>(undefined);\n const [isLoading, setIsLoading] = useState(false);\n\n // Abort controller to cancel the current API call.\n const abortControllerRef = useRef<AbortController | null>(null);\n\n const stop = useCallback(() => {\n try {\n abortControllerRef.current?.abort();\n } catch (ignored) {\n } finally {\n setIsLoading(false);\n abortControllerRef.current = null;\n }\n }, []);\n\n const submit = async (input: INPUT) => {\n const abortController = new AbortController();\n\n try {\n clearObject();\n\n setIsLoading(true);\n\n abortControllerRef.current = abortController;\n\n // Resolve headers at request time (supports async functions for dynamic auth tokens)\n const resolvedHeaders = await resolve(headers);\n\n const actualFetch = fetch ?? getOriginalFetch();\n const response = await actualFetch(api, {\n method: 'POST',\n headers: {\n 'Content-Type': 'application/json',\n ...normalizeHeaders(resolvedHeaders),\n },\n credentials,\n signal: abortController.signal,\n body: JSON.stringify(input),\n });\n\n if (!response.ok) {\n throw new Error(\n (await response.text()) ?? 'Failed to fetch the response.',\n );\n }\n\n if (response.body == null) {\n throw new Error('The response body is empty.');\n }\n\n let accumulatedText = '';\n let latestObject: DeepPartial<RESULT> | undefined = undefined;\n\n await response.body.pipeThrough(new TextDecoderStream()).pipeTo(\n new WritableStream<string>({\n async write(chunk) {\n accumulatedText += chunk;\n\n const { value } = await parsePartialJson(accumulatedText);\n const currentObject = value as DeepPartial<RESULT>;\n\n if (!isDeepEqualData(latestObject, currentObject)) {\n latestObject = currentObject;\n\n if (abortControllerRef.current === abortController) {\n mutate(currentObject);\n }\n }\n },\n\n async close() {\n if (abortControllerRef.current === abortController) {\n setIsLoading(false);\n }\n\n if (onFinish != null) {\n const validationResult = await safeValidateTypes({\n value: latestObject,\n schema: asSchema(schema),\n });\n\n onFinish(\n validationResult.success\n ? { object: validationResult.value, error: undefined }\n : { object: undefined, error: validationResult.error },\n );\n }\n\n if (abortControllerRef.current === abortController) {\n abortControllerRef.current = null;\n }\n },\n }),\n );\n } catch (error) {\n if (isAbortError(error)) {\n return;\n }\n\n if (onError && error instanceof Error) {\n onError(error);\n }\n\n if (abortControllerRef.current === abortController) {\n setIsLoading(false);\n abortControllerRef.current = null;\n setError(error instanceof Error ? error : new Error(String(error)));\n }\n }\n };\n\n const clear = () => {\n stop();\n clearObject();\n };\n\n const clearObject = () => {\n setError(undefined);\n setIsLoading(false);\n mutate(undefined);\n };\n\n return {\n submit,\n object: data,\n error,\n isLoading,\n stop,\n clear,\n };\n}\n\nexport const experimental_useObject = useObject;\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AAAA;AAAA,EAME;AAAA,OACK;AACP,SAAS,aAAa,WAAW,QAAQ,4BAA4B;;;ACRrE;AAAA,EACE;AAAA,OAKK;;;ACNP,OAAO,sBAAsB;AAEtB,SAAS,SACd,IACA,QACG;AACH,SAAO,UAAU,OAAO,iBAAiB,IAAI,MAAM,IAAI;AACzD;;;ADEA,SAAS,cAAwB,UAA8B;AAC7D,MAAI,MAAM,QAAQ,QAAQ,GAAG;AAC3B,WAAO,CAAC,GAAG,QAAQ;AAAA,EACrB;AAEA,MACE,YAAY,QACZ,OAAO,aAAa,aACnB,OAAO,eAAe,QAAQ,MAAM,OAAO,aAC1C,OAAO,eAAe,QAAQ,MAAM,OACtC;AACA,WAAO,EAAE,GAAG,SAAS;AAAA,EACvB;AAEA,SAAO;AACT;AAxBA;AA0BA,IAAM,iBAAN,MAEmC;AAAA,EASjC,YAAY,kBAAgC,CAAC,GAAG;AARhD;AACA,gCAAsB;AACtB,+BAA4B;AAE5B,2CAAqB,oBAAI,IAAgB;AACzC,yCAAmB,oBAAI,IAAgB;AACvC,wCAAkB,oBAAI,IAAgB;AAiCtC,uBAAc,CAAC,YAAwB;AACrC,yBAAK,WAAY,mBAAK,WAAU,OAAO,OAAO;AAC9C,yBAAK,wBAAL;AAAA,IACF;AAEA,sBAAa,MAAM;AACjB,yBAAK,WAAY,mBAAK,WAAU,MAAM,GAAG,EAAE;AAC3C,yBAAK,wBAAL;AAAA,IACF;AAEA,0BAAiB,CAAC,OAAe,YAAwB;AACvD,yBAAK,WAAY;AAAA,QACf,GAAG,mBAAK,WAAU,MAAM,GAAG,KAAK;AAAA,QAChC,KAAK,SAAS,OAAO;AAAA,QACrB,GAAG,mBAAK,WAAU,MAAM,QAAQ,CAAC;AAAA,MACnC;AACA,yBAAK,wBAAL;AAAA,IACF;AAEA,oBAAW,CAAI,UAAgB;AAC7B,UACE,SAAS,QACT,OAAO,UAAU,YACjB,EAAE,WAAW,UACb,CAAC,MAAM,QAAQ,MAAM,KAAK,GAC1B;AACA,eAAO;AAAA,MACT;AAEA,YAAM,UAAU;AAChB,YAAM,WAAW;AAAA,QACf,GAAG;AAAA,QACH,OAAO,QAAQ,MAAM,IAAI,WAAS,EAAE,GAAG,KAAK,EAAE;AAAA,MAChD;AAEA,UAAI,cAAc,SAAS;AACzB,iBAAS,WAAW,cAAc,QAAQ,QAAQ;AAAA,MACpD;AAEA,aAAO;AAAA,IACT;AAEA,wCAA8B,CAC5B,UACA,mBACiB;AACjB,YAAM,WAAW,iBACb,SAAS,UAAU,cAAc,IACjC;AACJ,yBAAK,oBAAmB,IAAI,QAAQ;AACpC,aAAO,MAAM;AACX,2BAAK,oBAAmB,OAAO,QAAQ;AAAA,MACzC;AAAA,IACF;AAEA,sCAA4B,CAAC,aAAuC;AAClE,yBAAK,kBAAiB,IAAI,QAAQ;AAClC,aAAO,MAAM;AACX,2BAAK,kBAAiB,OAAO,QAAQ;AAAA,MACvC;AAAA,IACF;AAEA,qCAA2B,CAAC,aAAuC;AACjE,yBAAK,iBAAgB,IAAI,QAAQ;AACjC,aAAO,MAAM;AACX,2BAAK,iBAAgB,OAAO,QAAQ;AAAA,MACtC;AAAA,IACF;AAEA,+CAAyB,MAAM;AAC7B,yBAAK,oBAAmB,QAAQ,cAAY,SAAS,CAAC;AAAA,IACxD;AAEA,6CAAuB,MAAM;AAC3B,yBAAK,kBAAiB,QAAQ,cAAY,SAAS,CAAC;AAAA,IACtD;AAEA,4CAAsB,MAAM;AAC1B,yBAAK,iBAAgB,QAAQ,cAAY,SAAS,CAAC;AAAA,IACrD;AA7GE,uBAAK,WAAY;AAAA,EACnB;AAAA,EAEA,IAAI,SAAqB;AACvB,WAAO,mBAAK;AAAA,EACd;AAAA,EAEA,IAAI,OAAO,WAAuB;AAChC,uBAAK,SAAU;AACf,uBAAK,sBAAL;AAAA,EACF;AAAA,EAEA,IAAI,QAA2B;AAC7B,WAAO,mBAAK;AAAA,EACd;AAAA,EAEA,IAAI,MAAM,UAA6B;AACrC,uBAAK,QAAS;AACd,uBAAK,qBAAL;AAAA,EACF;AAAA,EAEA,IAAI,WAAyB;AAC3B,WAAO,mBAAK;AAAA,EACd;AAAA,EAEA,IAAI,SAAS,aAA2B;AACtC,uBAAK,WAAY,CAAC,GAAG,WAAW;AAChC,uBAAK,wBAAL;AAAA,EACF;AAkFF;AAvHE;AACA;AACA;AAEA;AACA;AACA;AAsGA;AAIA;AAIA;AAjJF;AAsJO,IAAM,OAAN,cAEG,aAAyB;AAAA,EAGjC,YAAY,EAAE,UAAU,GAAG,KAAK,GAAyB;AACvD,UAAM,QAAQ,IAAI,eAAe,QAAQ;AACzC,UAAM,EAAE,GAAG,MAAM,MAAM,CAAC;AAJ1B;AAQA,wCAA8B,CAC5B,UACA,mBAEA,mBAAK,QAAO,2BAA2B,EAAE,UAAU,cAAc;AAEnE,sCAA4B,CAAC,aAC3B,mBAAK,QAAO,yBAAyB,EAAE,QAAQ;AAEjD,qCAA2B,CAAC,aAC1B,mBAAK,QAAO,wBAAwB,EAAE,QAAQ;AAb9C,uBAAK,QAAS;AAAA,EAChB;AAaF;AAnBE;;;AD9FK,SAAS,QAAkD;AAAA,EAChE,uBAAuB;AAAA,EACvB,SAAS;AAAA,EACT,GAAG;AACL,IAAgC,CAAC,GAA+B;AAM9D,QAAM,YAAY,OAYhB,CAAC,CAAC;AAEJ,MAAI,EAAE,UAAU,UAAU;AACxB,cAAU,UAAU;AAAA,MAClB,YAAY,QAAQ;AAAA,MACpB,QAAQ,QAAQ;AAAA,MAChB,UAAU,QAAQ;AAAA,MAClB,SAAS,QAAQ;AAAA,MACjB,uBAAuB,QAAQ;AAAA,MAC/B,WAAW,QAAQ;AAAA,IACrB;AAAA,EACF;AAGA,MAAI;AACJ,QAAM,eAAe,MAAG;AAhG1B;AAiGI,2BAAU,QAAQ,cAAlB,YACC,iEAAqB,IAAI,qBAAiC;AAAA;AAG7D,QAAM,cAA8B;AAAA,IAClC,GAAG;AAAA,IACH,WAAW;AAAA,MACT,cAAc,iBAAe,aAAa,EAAE,aAAa,WAAW;AAAA,MACpE,mBAAmB,sBACjB,aAAa,EAAE,kBAAkB,gBAAgB;AAAA,IACrD;AAAA,IACA,YAAY,SAAI;AA5GpB;AA4GuB,mCAAU,SAAQ,eAAlB,4BAA+B;AAAA;AAAA,IAClD,QAAQ,SAAI;AA7GhB;AA6GmB,mCAAU,SAAQ,WAAlB,4BAA2B;AAAA;AAAA,IAC1C,UAAU,SAAI;AA9GlB;AA8GqB,mCAAU,SAAQ,aAAlB,4BAA6B;AAAA;AAAA,IAC9C,SAAS,SAAI;AA/GjB;AA+GoB,mCAAU,SAAQ,YAAlB,4BAA4B;AAAA;AAAA,IAC5C,uBAAuB,SAAI;AAhH/B;AAiHM,yCAAU,SAAQ,0BAAlB,4BAA0C,SAA1C,YAAkD;AAAA;AAAA,EACtD;AAEA,QAAM,UAAU;AAAA,IACd,UAAU,UAAU,QAAQ,OAAO,IAAI,KAAK,WAAW;AAAA,EACzD;AAEA,QAAM,qBACH,UAAU,WAAW,QAAQ,SAAS,QAAQ,WAC9C,QAAQ,WACP,QAAQ,MAAM,QACd,QAAQ,QAAQ,OAAO,QAAQ;AAEnC,MAAI,oBAAoB;AACtB,YAAQ,UAAU,UAAU,UAAU,QAAQ,OAAO,IAAI,KAAK,WAAW;AAAA,EAC3E;AAEA,QAAM,OAAO,QAAQ;AACrB,QAAM,sBAAsB,OAAO;AAAA,IACjC;AAAA,IACA,UAAU,KAAK;AAAA,EACjB,CAAC;AAED,MAAI,oBAAoB,QAAQ,SAAS,MAAM;AAC7C,wBAAoB,UAAU,EAAE,MAAM,UAAU,KAAK,SAAS;AAAA,EAChE;AAEA,QAAM,sBAAsB;AAAA,IAC1B,CAAC,WAAuB;AACtB,UAAI,eAAe;AAEnB,YAAM,iBAAiB,MAAM;AAC3B,YAAI,CAAC,gBAAgB,oBAAoB,QAAQ,SAAS,MAAM;AAC9D;AAAA,QACF;AAEA,4BAAoB,UAAU,EAAE,MAAM,UAAU,KAAK,SAAS;AAC9D,eAAO;AAAA,MACT;AAEA,YAAM,cAAc,KAAK,2BAA2B;AAAA,QAClD;AAAA,QACA;AAAA,MACF;AAKA,0BAAoB,UAAU,EAAE,MAAM,UAAU,KAAK,SAAS;AAE9D,aAAO,MAAM;AACX,uBAAe;AACf,oBAAY;AAAA,MACd;AAAA,IACF;AAAA,IACA,CAAC,MAAM,cAAc;AAAA,EACvB;AAEA,QAAM,sBAAsB;AAAA,IAC1B,MAAM,oBAAoB,QAAQ;AAAA,IAClC,CAAC;AAAA,EACH;AAEA,QAAM,WAAW;AAAA,IACf;AAAA,IACA;AAAA,IACA;AAAA,EACF;AAEA,QAAM,oBAAoB;AAAA,IACxB,CAAC,WACC,KAAK,yBAAyB,EAAE,MAAM;AACpC,UAAI,oBAAoB,QAAQ,SAAS,MAAM;AAC7C;AAAA,MACF;AAEA,UAAI,KAAK,WAAW,WAAW,KAAK,WAAW,SAAS;AAEtD,4BAAoB,UAAU,EAAE,MAAM,UAAU,KAAK,SAAS;AAAA,MAChE;AAEA,aAAO;AAAA,IACT,CAAC;AAAA,IACH,CAAC,IAAI;AAAA,EACP;AAEA,QAAM,oBAAoB,YAAY,MAAM,KAAK,QAAQ,CAAC,IAAI,CAAC;AAE/D,QAAM,SAAS;AAAA,IACb;AAAA,IACA;AAAA,IACA;AAAA,EACF;AAEA,QAAM,QAAQ;AAAA,IACZ,QAAQ,QAAQ,wBAAwB;AAAA,IACxC,MAAM,QAAQ,QAAQ;AAAA,IACtB,MAAM,QAAQ,QAAQ;AAAA,EACxB;AAEA,QAAM,cAAc;AAAA,IAClB,CACE,kBACG;AACH,UAAI,OAAO,kBAAkB,YAAY;AACvC,wBAAgB,cAAc,QAAQ,QAAQ,QAAQ;AAAA,MACxD;AACA,cAAQ,QAAQ,WAAW;AAAA,IAC7B;AAAA,IACA,CAAC,OAAO;AAAA,EACV;AAEA,YAAU,MAAM;AACd,QAAI,QAAQ;AACV,cAAQ,QAAQ,aAAa;AAAA,IAC/B;AAAA,EACF,GAAG,CAAC,QAAQ,OAAO,CAAC;AAEpB,SAAO;AAAA,IACL,IAAI,QAAQ,QAAQ;AAAA,IACpB;AAAA,IACA;AAAA,IACA,aAAa,QAAQ,QAAQ;AAAA,IAC7B,YAAY,QAAQ,QAAQ;AAAA,IAC5B,YAAY,QAAQ,QAAQ;AAAA,IAC5B,MAAM,QAAQ,QAAQ;AAAA,IACtB;AAAA,IACA,cAAc,QAAQ,QAAQ;AAAA,IAC9B;AAAA;AAAA;AAAA;AAAA,IAIA,eAAe,QAAQ,QAAQ;AAAA,IAC/B,eAAe,QAAQ,QAAQ;AAAA,IAC/B,yBAAyB,QAAQ,QAAQ;AAAA,EAC3C;AACF;;;AGzPA;AAAA,EACE;AAAA,OAGK;AACP,SAAS,eAAAA,cAAa,aAAAC,YAAW,OAAO,UAAAC,SAAQ,gBAAgB;AAChE,OAAO,YAAY;AAyDZ,SAAS,cAAc;AAAA,EAC5B,MAAM;AAAA,EACN;AAAA,EACA,oBAAoB;AAAA,EACpB,eAAe;AAAA,EACf;AAAA,EACA;AAAA,EACA;AAAA,EACA,iBAAiB;AAAA,EACjB,OAAAC;AAAA,EACA;AAAA,EACA;AAAA,EACA,uBAAuB;AACzB,IAMI,CAAC,GAAyB;AAE5B,QAAM,SAAS,MAAM;AACrB,QAAM,eAAe,MAAM;AAG3B,QAAM,EAAE,MAAM,OAAO,IAAI,OAAe,CAAC,KAAK,YAAY,GAAG,MAAM;AAAA,IACjE,cAAc;AAAA,EAChB,CAAC;AAED,QAAM,EAAE,MAAM,YAAY,OAAO,QAAQ,cAAc,IAAI;AAAA,IACzD,CAAC,cAAc,SAAS;AAAA,IACxB;AAAA,EACF;AAEA,QAAM,CAAC,OAAO,QAAQ,IAAI,SAA4B,MAAS;AAC/D,QAAM,aAAa;AAGnB,QAAM,CAAC,iBAAiB,kBAAkB,IACxC,SAAiC,IAAI;AAEvC,QAAM,mBAAmBC,QAAO;AAAA,IAC9B;AAAA,IACA;AAAA,IACA;AAAA,EACF,CAAC;AAED,EAAAC,WAAU,MAAM;AACd,qBAAiB,UAAU;AAAA,MACzB;AAAA,MACA;AAAA,MACA;AAAA,IACF;AAAA,EACF,GAAG,CAAC,aAAa,SAAS,IAAI,CAAC;AAE/B,QAAM,iBAAiBC;AAAA,IACrB,OAAO,QAAgB,YACrB,kBAAkB;AAAA,MAChB;AAAA,MACA;AAAA,MACA,aAAa,iBAAiB,QAAQ;AAAA,MACtC,SAAS,EAAE,GAAG,iBAAiB,QAAQ,SAAS,GAAG,mCAAS,QAAQ;AAAA,MACpE,MAAM;AAAA,QACJ,GAAG,iBAAiB,QAAQ;AAAA,QAC5B,GAAG,mCAAS;AAAA,MACd;AAAA,MACA;AAAA,MACA,OAAAH;AAAA;AAAA,MAEA,eAAe;AAAA,QACb,CAACI,gBAAuB,OAAOA,aAAY,KAAK;AAAA,QAChD;AAAA,MACF;AAAA,MACA,YAAY;AAAA,MACZ;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,IACF,CAAC;AAAA,IACH;AAAA,MACE;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACAJ;AAAA,MACA;AAAA,IACF;AAAA,EACF;AAEA,QAAM,OAAOG,aAAY,MAAM;AAC7B,QAAI,iBAAiB;AACnB,sBAAgB,MAAM;AACtB,yBAAmB,IAAI;AAAA,IACzB;AAAA,EACF,GAAG,CAAC,eAAe,CAAC;AAEpB,QAAM,gBAAgBA;AAAA,IACpB,CAACC,gBAAuB;AACtB,aAAOA,aAAY,KAAK;AAAA,IAC1B;AAAA,IACA,CAAC,MAAM;AAAA,EACT;AAEA,QAAM,WAAWD;AAAA,IACf,OAAO,QAAQ,YAAY;AACzB,aAAO,eAAe,QAAQ,OAAO;AAAA,IACvC;AAAA,IACA,CAAC,cAAc;AAAA,EACjB;AAEA,QAAM,CAAC,OAAO,QAAQ,IAAI,SAAS,YAAY;AAE/C,QAAM,eAAeA;AAAA,IACnB,CAAC,UAA4C;AArLjD;AAsLM,2CAAO,mBAAP;AACA,UAAI,CAAC,OAAO;AACV;AAAA,MACF;AAEA,YAAM,SAAS,SAAS,KAAK;AAC7B,eAAS,EAAE;AACX,aAAO;AAAA,IACT;AAAA,IACA,CAAC,OAAO,UAAU,QAAQ;AAAA,EAC5B;AAEA,QAAM,oBAAoBA;AAAA,IACxB,CAAC,MAAW;AACV,eAAS,EAAE,OAAO,KAAK;AAAA,IACzB;AAAA,IACA,CAAC,QAAQ;AAAA,EACX;AAEA,SAAO;AAAA,IACL;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF;AACF;;;ACrNA;AAAA,EACE;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,OAKK;AACP;AAAA,EACE;AAAA,EACA;AAAA,EACA;AAAA,OAEK;AACP,SAAS,eAAAE,cAAa,SAAAC,QAAO,UAAAC,SAAQ,YAAAC,iBAAgB;AACrD,OAAOC,aAAY;AAGnB,IAAM,mBAAmB,MAAM;AAsG/B,SAAS,UAIP;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA;AAAA,EACA;AAAA,EACA,OAAAC;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AACF,GAGiD;AAE/C,QAAM,SAASJ,OAAM;AACrB,QAAM,eAAe,kBAAM;AAG3B,QAAM,EAAE,MAAM,OAAO,IAAIG;AAAA,IACvB,CAAC,cAAc,QAAQ;AAAA,IACvB;AAAA,IACA,EAAE,cAAc,aAAa;AAAA,EAC/B;AAEA,QAAM,CAAC,OAAO,QAAQ,IAAID,UAA4B,MAAS;AAC/D,QAAM,CAAC,WAAW,YAAY,IAAIA,UAAS,KAAK;AAGhD,QAAM,qBAAqBD,QAA+B,IAAI;AAE9D,QAAM,OAAOF,aAAY,MAAM;AA7JjC;AA8JI,QAAI;AACF,+BAAmB,YAAnB,mBAA4B;AAAA,IAC9B,SAAS,SAAS;AAAA,IAClB,UAAE;AACA,mBAAa,KAAK;AAClB,yBAAmB,UAAU;AAAA,IAC/B;AAAA,EACF,GAAG,CAAC,CAAC;AAEL,QAAM,SAAS,OAAO,UAAiB;AAvKzC;AAwKI,UAAM,kBAAkB,IAAI,gBAAgB;AAE5C,QAAI;AACF,kBAAY;AAEZ,mBAAa,IAAI;AAEjB,yBAAmB,UAAU;AAG7B,YAAM,kBAAkB,MAAM,QAAQ,OAAO;AAE7C,YAAM,cAAcK,UAAA,OAAAA,SAAS,iBAAiB;AAC9C,YAAM,WAAW,MAAM,YAAY,KAAK;AAAA,QACtC,QAAQ;AAAA,QACR,SAAS;AAAA,UACP,gBAAgB;AAAA,UAChB,GAAG,iBAAiB,eAAe;AAAA,QACrC;AAAA,QACA;AAAA,QACA,QAAQ,gBAAgB;AAAA,QACxB,MAAM,KAAK,UAAU,KAAK;AAAA,MAC5B,CAAC;AAED,UAAI,CAAC,SAAS,IAAI;AAChB,cAAM,IAAI;AAAA,WACP,WAAM,SAAS,KAAK,MAApB,YAA0B;AAAA,QAC7B;AAAA,MACF;AAEA,UAAI,SAAS,QAAQ,MAAM;AACzB,cAAM,IAAI,MAAM,6BAA6B;AAAA,MAC/C;AAEA,UAAI,kBAAkB;AACtB,UAAI,eAAgD;AAEpD,YAAM,SAAS,KAAK,YAAY,IAAI,kBAAkB,CAAC,EAAE;AAAA,QACvD,IAAI,eAAuB;AAAA,UACzB,MAAM,MAAM,OAAO;AACjB,+BAAmB;AAEnB,kBAAM,EAAE,MAAM,IAAI,MAAM,iBAAiB,eAAe;AACxD,kBAAM,gBAAgB;AAEtB,gBAAI,CAAC,gBAAgB,cAAc,aAAa,GAAG;AACjD,6BAAe;AAEf,kBAAI,mBAAmB,YAAY,iBAAiB;AAClD,uBAAO,aAAa;AAAA,cACtB;AAAA,YACF;AAAA,UACF;AAAA,UAEA,MAAM,QAAQ;AACZ,gBAAI,mBAAmB,YAAY,iBAAiB;AAClD,2BAAa,KAAK;AAAA,YACpB;AAEA,gBAAI,YAAY,MAAM;AACpB,oBAAM,mBAAmB,MAAM,kBAAkB;AAAA,gBAC/C,OAAO;AAAA,gBACP,QAAQ,SAAS,MAAM;AAAA,cACzB,CAAC;AAED;AAAA,gBACE,iBAAiB,UACb,EAAE,QAAQ,iBAAiB,OAAO,OAAO,OAAU,IACnD,EAAE,QAAQ,QAAW,OAAO,iBAAiB,MAAM;AAAA,cACzD;AAAA,YACF;AAEA,gBAAI,mBAAmB,YAAY,iBAAiB;AAClD,iCAAmB,UAAU;AAAA,YAC/B;AAAA,UACF;AAAA,QACF,CAAC;AAAA,MACH;AAAA,IACF,SAASC,QAAO;AACd,UAAI,aAAaA,MAAK,GAAG;AACvB;AAAA,MACF;AAEA,UAAI,WAAWA,kBAAiB,OAAO;AACrC,gBAAQA,MAAK;AAAA,MACf;AAEA,UAAI,mBAAmB,YAAY,iBAAiB;AAClD,qBAAa,KAAK;AAClB,2BAAmB,UAAU;AAC7B,iBAASA,kBAAiB,QAAQA,SAAQ,IAAI,MAAM,OAAOA,MAAK,CAAC,CAAC;AAAA,MACpE;AAAA,IACF;AAAA,EACF;AAEA,QAAM,QAAQ,MAAM;AAClB,SAAK;AACL,gBAAY;AAAA,EACd;AAEA,QAAM,cAAc,MAAM;AACxB,aAAS,MAAS;AAClB,iBAAa,KAAK;AAClB,WAAO,MAAS;AAAA,EAClB;AAEA,SAAO;AAAA,IACL;AAAA,IACA,QAAQ;AAAA,IACR;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF;AACF;AAEO,IAAM,yBAAyB;","names":["useCallback","useEffect","useRef","fetch","useRef","useEffect","useCallback","completion","useCallback","useId","useRef","useState","useSWR","fetch","error"]}
|
|
1
|
+
{"version":3,"sources":["../src/use-chat.ts","../src/chat.react.ts","../src/throttle.ts","../src/use-completion.ts","../src/use-object.ts"],"sourcesContent":["import {\n type AbstractChat,\n type ChatInit,\n type ChatTransport,\n type CreateUIMessage,\n type UIMessage,\n DefaultChatTransport,\n} from 'ai';\nimport { useCallback, useEffect, useRef, useSyncExternalStore } from 'react';\nimport { Chat } from './chat.react';\n\nexport type { CreateUIMessage, UIMessage };\n\nexport type UseChatHelpers<UI_MESSAGE extends UIMessage> = {\n /**\n * The id of the chat.\n */\n readonly id: string;\n\n /**\n * Update the `messages` state locally. This is useful when you want to\n * edit the messages on the client, and then trigger the `reload` method\n * manually to regenerate the AI response.\n */\n setMessages: (\n messages: UI_MESSAGE[] | ((messages: UI_MESSAGE[]) => UI_MESSAGE[]),\n ) => void;\n\n error: Error | undefined;\n} & Pick<\n AbstractChat<UI_MESSAGE>,\n | 'sendMessage'\n | 'regenerate'\n | 'stop'\n | 'resumeStream'\n | 'addToolResult'\n | 'addToolOutput'\n | 'addToolApprovalResponse'\n | 'status'\n | 'messages'\n | 'clearError'\n>;\n\nexport type UseChatOptions<UI_MESSAGE extends UIMessage> = (\n | { chat: Chat<UI_MESSAGE> }\n | ChatInit<UI_MESSAGE>\n) & {\n /**\n * Custom throttle wait in ms for the chat messages and data updates.\n * Default is undefined, which disables throttling.\n */\n experimental_throttle?: number;\n\n /**\n * Whether to automatically resume an ongoing chat generation stream.\n */\n resume?: boolean;\n};\n\ntype AutomaticResumeState = {\n registrations: Set<object>;\n cleanupVisibilityListener?: () => void;\n pendingVisibilityResume?: Promise<void>;\n};\n\nconst automaticResumeStates = new WeakMap<object, AutomaticResumeState>();\n\ntype AutomaticallyResumableChat = {\n '~resumeStreamIfDisconnected': (options: {\n shouldResume: () => boolean;\n }) => Promise<void>;\n};\n\n/**\n * When the document becomes visible again, resume the chat stream if it was\n * interrupted by a network disconnect while the page was in the background.\n */\nfunction resumeOnVisible<UI_MESSAGE extends UIMessage>({\n chat,\n state,\n}: {\n chat: Chat<UI_MESSAGE>;\n state: AutomaticResumeState;\n}) {\n if (\n document.visibilityState !== 'visible' ||\n state.pendingVisibilityResume != null\n ) {\n return;\n }\n\n const isRegistered = () =>\n automaticResumeStates.get(chat) === state && state.registrations.size > 0;\n\n const clearPendingResume = () => {\n state.pendingVisibilityResume = undefined;\n };\n\n const resumableChat = chat as Chat<UI_MESSAGE> & AutomaticallyResumableChat;\n\n state.pendingVisibilityResume = resumableChat['~resumeStreamIfDisconnected']({\n shouldResume: () =>\n isRegistered() && document.visibilityState === 'visible',\n }).then(clearPendingResume, clearPendingResume);\n}\n\nfunction registerAutomaticResume<UI_MESSAGE extends UIMessage>({\n chat,\n registration,\n}: {\n chat: Chat<UI_MESSAGE>;\n registration: object;\n}) {\n let state = automaticResumeStates.get(chat);\n\n if (state == null) {\n const newState: AutomaticResumeState = { registrations: new Set() };\n automaticResumeStates.set(chat, newState);\n state = newState;\n\n if (typeof document !== 'undefined') {\n const onVisibilityChange = () =>\n resumeOnVisible({ chat, state: newState });\n\n document.addEventListener('visibilitychange', onVisibilityChange);\n newState.cleanupVisibilityListener = () => {\n document.removeEventListener('visibilitychange', onVisibilityChange);\n };\n }\n }\n\n const { registrations } = state;\n const shouldResume = registrations.size === 0;\n registrations.add(registration);\n\n if (\n shouldResume &&\n chat.status !== 'submitted' &&\n chat.status !== 'streaming'\n ) {\n void chat.resumeStream();\n }\n\n return () => {\n registrations.delete(registration);\n\n if (registrations.size === 0) {\n state.cleanupVisibilityListener?.();\n automaticResumeStates.delete(chat);\n }\n };\n}\nexport function useChat<UI_MESSAGE extends UIMessage = UIMessage>({\n experimental_throttle: throttleWaitMs,\n resume = false,\n ...options\n}: UseChatOptions<UI_MESSAGE> = {}): UseChatHelpers<UI_MESSAGE> {\n const automaticResumeRegistration = useRef({});\n\n // the Chat instance is created once and not recreated when options change,\n // so it would normally keep the callbacks/transport from the first render forever\n\n // keep latest values in a ref that is refreshed on every render,\n // and hand `Chat` stable wrappers that read from it to avoid stale closures\n const latestRef = useRef<\n Partial<\n Pick<\n ChatInit<UI_MESSAGE>,\n | 'onToolCall'\n | 'onData'\n | 'onFinish'\n | 'onError'\n | 'sendAutomaticallyWhen'\n | 'transport'\n >\n >\n >({});\n\n if (!('chat' in options)) {\n latestRef.current = {\n onToolCall: options.onToolCall,\n onData: options.onData,\n onFinish: options.onFinish,\n onError: options.onError,\n sendAutomaticallyWhen: options.sendAutomaticallyWhen,\n transport: options.transport,\n };\n }\n\n // resolve the latest transport and fallback to a lazily created default transport\n let defaultTransport: ChatTransport<UI_MESSAGE> | undefined;\n const getTransport = () =>\n latestRef.current.transport ??\n (defaultTransport ??= new DefaultChatTransport<UI_MESSAGE>());\n\n // give `Chat` stable wrappers that always read the latest values from `latestRef`\n const chatOptions: typeof options = {\n ...options,\n transport: {\n sendMessages: sendOptions => getTransport().sendMessages(sendOptions),\n reconnectToStream: reconnectOptions =>\n getTransport().reconnectToStream(reconnectOptions),\n },\n onToolCall: arg => latestRef.current.onToolCall?.(arg),\n onData: arg => latestRef.current.onData?.(arg),\n onFinish: arg => latestRef.current.onFinish?.(arg),\n onError: arg => latestRef.current.onError?.(arg),\n sendAutomaticallyWhen: arg =>\n latestRef.current.sendAutomaticallyWhen?.(arg) ?? false,\n };\n\n const chatRef = useRef<Chat<UI_MESSAGE>>(\n 'chat' in options ? options.chat : new Chat(chatOptions),\n );\n\n const shouldRecreateChat =\n ('chat' in options && options.chat !== chatRef.current) ||\n ('id' in options &&\n options.id != null &&\n chatRef.current.id !== options.id);\n\n if (shouldRecreateChat) {\n chatRef.current = 'chat' in options ? options.chat : new Chat(chatOptions);\n }\n\n const chat = chatRef.current;\n const messagesSnapshotRef = useRef({\n chat,\n messages: chat.messages,\n });\n\n if (messagesSnapshotRef.current.chat !== chat) {\n messagesSnapshotRef.current = { chat, messages: chat.messages };\n }\n\n const subscribeToMessages = useCallback(\n (update: () => void) => {\n let isSubscribed = true;\n\n const updateMessages = () => {\n if (!isSubscribed || messagesSnapshotRef.current.chat !== chat) {\n return;\n }\n\n messagesSnapshotRef.current = { chat, messages: chat.messages };\n update();\n };\n\n const unsubscribe = chat['~registerMessagesCallback'](\n updateMessages,\n throttleWaitMs,\n );\n\n // Synchronize changes that may have happened between render and\n // subscription. useSyncExternalStore checks the snapshot after\n // subscribing and schedules a render when it changed.\n messagesSnapshotRef.current = { chat, messages: chat.messages };\n\n return () => {\n isSubscribed = false;\n unsubscribe();\n };\n },\n [chat, throttleWaitMs],\n );\n\n const getMessagesSnapshot = useCallback(\n () => messagesSnapshotRef.current.messages,\n [],\n );\n\n const messages = useSyncExternalStore(\n subscribeToMessages,\n getMessagesSnapshot,\n getMessagesSnapshot,\n );\n\n const subscribeToStatus = useCallback(\n (update: () => void) =>\n chat['~registerStatusCallback'](() => {\n if (messagesSnapshotRef.current.chat !== chat) {\n return;\n }\n\n if (chat.status === 'ready' || chat.status === 'error') {\n // Publish the latest messages before the terminal status can render.\n messagesSnapshotRef.current = { chat, messages: chat.messages };\n }\n\n update();\n }),\n [chat],\n );\n\n const getStatusSnapshot = useCallback(() => chat.status, [chat]);\n\n const status = useSyncExternalStore(\n subscribeToStatus,\n getStatusSnapshot,\n getStatusSnapshot,\n );\n\n const error = useSyncExternalStore(\n chatRef.current['~registerErrorCallback'],\n () => chatRef.current.error,\n () => chatRef.current.error,\n );\n\n const setMessages = useCallback(\n (\n messagesParam: UI_MESSAGE[] | ((messages: UI_MESSAGE[]) => UI_MESSAGE[]),\n ) => {\n if (typeof messagesParam === 'function') {\n messagesParam = messagesParam(chatRef.current.messages);\n }\n chatRef.current.messages = messagesParam;\n },\n [chatRef],\n );\n\n useEffect(() => {\n if (resume) {\n return registerAutomaticResume({\n chat,\n registration: automaticResumeRegistration.current,\n });\n }\n }, [resume, chat]);\n\n return {\n id: chatRef.current.id,\n messages,\n setMessages,\n sendMessage: chatRef.current.sendMessage,\n regenerate: chatRef.current.regenerate,\n clearError: chatRef.current.clearError,\n stop: chatRef.current.stop,\n error,\n resumeStream: chatRef.current.resumeStream,\n status,\n /**\n * @deprecated Use `addToolOutput` instead.\n */\n addToolResult: chatRef.current.addToolOutput,\n addToolOutput: chatRef.current.addToolOutput,\n addToolApprovalResponse: chatRef.current.addToolApprovalResponse,\n };\n}\n","import {\n AbstractChat,\n type ChatInit,\n type ChatState,\n type ChatStatus,\n type UIMessage,\n} from 'ai';\nimport { throttle } from './throttle';\n\nfunction cloneMetadata<METADATA>(metadata: METADATA): METADATA {\n if (Array.isArray(metadata)) {\n return [...metadata] as METADATA;\n }\n\n if (\n metadata != null &&\n typeof metadata === 'object' &&\n (Object.getPrototypeOf(metadata) === Object.prototype ||\n Object.getPrototypeOf(metadata) === null)\n ) {\n return { ...metadata } as METADATA;\n }\n\n return metadata;\n}\n\nclass ReactChatState<\n UI_MESSAGE extends UIMessage,\n> implements ChatState<UI_MESSAGE> {\n #messages: UI_MESSAGE[];\n #status: ChatStatus = 'ready';\n #error: Error | undefined = undefined;\n\n #messagesCallbacks = new Set<() => void>();\n #statusCallbacks = new Set<() => void>();\n #errorCallbacks = new Set<() => void>();\n\n constructor(initialMessages: UI_MESSAGE[] = []) {\n this.#messages = initialMessages;\n }\n\n get status(): ChatStatus {\n return this.#status;\n }\n\n set status(newStatus: ChatStatus) {\n this.#status = newStatus;\n this.#callStatusCallbacks();\n }\n\n get error(): Error | undefined {\n return this.#error;\n }\n\n set error(newError: Error | undefined) {\n this.#error = newError;\n this.#callErrorCallbacks();\n }\n\n get messages(): UI_MESSAGE[] {\n return this.#messages;\n }\n\n set messages(newMessages: UI_MESSAGE[]) {\n this.#messages = [...newMessages];\n this.#callMessagesCallbacks();\n }\n\n pushMessage = (message: UI_MESSAGE) => {\n this.#messages = this.#messages.concat(message);\n this.#callMessagesCallbacks();\n };\n\n popMessage = () => {\n this.#messages = this.#messages.slice(0, -1);\n this.#callMessagesCallbacks();\n };\n\n replaceMessage = (index: number, message: UI_MESSAGE) => {\n this.#messages = [\n ...this.#messages.slice(0, index),\n this.snapshot(message),\n ...this.#messages.slice(index + 1),\n ];\n this.#callMessagesCallbacks();\n };\n\n snapshot = <T>(value: T): T => {\n if (\n value == null ||\n typeof value !== 'object' ||\n !('parts' in value) ||\n !Array.isArray(value.parts)\n ) {\n return value;\n }\n\n const message = value as UI_MESSAGE;\n const snapshot = {\n ...message,\n parts: message.parts.map(part => ({ ...part })),\n };\n\n if ('metadata' in message) {\n snapshot.metadata = cloneMetadata(message.metadata);\n }\n\n return snapshot as T;\n };\n\n '~registerMessagesCallback' = (\n onChange: () => void,\n throttleWaitMs?: number,\n ): (() => void) => {\n const callback = throttleWaitMs\n ? throttle(onChange, throttleWaitMs)\n : onChange;\n this.#messagesCallbacks.add(callback);\n return () => {\n this.#messagesCallbacks.delete(callback);\n };\n };\n\n '~registerStatusCallback' = (onChange: () => void): (() => void) => {\n this.#statusCallbacks.add(onChange);\n return () => {\n this.#statusCallbacks.delete(onChange);\n };\n };\n\n '~registerErrorCallback' = (onChange: () => void): (() => void) => {\n this.#errorCallbacks.add(onChange);\n return () => {\n this.#errorCallbacks.delete(onChange);\n };\n };\n\n #callMessagesCallbacks = () => {\n this.#messagesCallbacks.forEach(callback => callback());\n };\n\n #callStatusCallbacks = () => {\n this.#statusCallbacks.forEach(callback => callback());\n };\n\n #callErrorCallbacks = () => {\n this.#errorCallbacks.forEach(callback => callback());\n };\n}\n\nexport class Chat<\n UI_MESSAGE extends UIMessage,\n> extends AbstractChat<UI_MESSAGE> {\n #state: ReactChatState<UI_MESSAGE>;\n\n constructor({ messages, ...init }: ChatInit<UI_MESSAGE>) {\n const state = new ReactChatState(messages);\n super({ ...init, state });\n this.#state = state;\n }\n\n '~registerMessagesCallback' = (\n onChange: () => void,\n throttleWaitMs?: number,\n ): (() => void) =>\n this.#state['~registerMessagesCallback'](onChange, throttleWaitMs);\n\n '~registerStatusCallback' = (onChange: () => void): (() => void) =>\n this.#state['~registerStatusCallback'](onChange);\n\n '~registerErrorCallback' = (onChange: () => void): (() => void) =>\n this.#state['~registerErrorCallback'](onChange);\n}\n","import throttleFunction from 'throttleit';\n\nexport function throttle<T extends (...args: any[]) => any>(\n fn: T,\n waitMs: number | undefined,\n): T {\n return waitMs != null ? throttleFunction(fn, waitMs) : fn;\n}\n","import {\n callCompletionApi,\n type CompletionRequestOptions,\n type UseCompletionOptions,\n} from 'ai';\nimport { useCallback, useEffect, useId, useRef, useState } from 'react';\nimport useSWR from 'swr';\nimport { throttle } from './throttle';\n\nexport type { UseCompletionOptions };\n\nexport type UseCompletionHelpers = {\n /** The current completion result */\n completion: string;\n /**\n * Send a new prompt to the API endpoint and update the completion state.\n */\n complete: (\n prompt: string,\n options?: CompletionRequestOptions,\n ) => Promise<string | null | undefined>;\n /** The error object of the API request */\n error: undefined | Error;\n /**\n * Abort the current API request but keep the generated tokens.\n */\n stop: () => void;\n /**\n * Update the `completion` state locally.\n */\n setCompletion: (completion: string) => void;\n /** The current value of the input */\n input: string;\n /** setState-powered method to update the input value */\n setInput: React.Dispatch<React.SetStateAction<string>>;\n /**\n * An input/textarea-ready onChange handler to control the value of the input\n * @example\n * ```jsx\n * <input onChange={handleInputChange} value={input} />\n * ```\n */\n handleInputChange: (\n event:\n | React.ChangeEvent<HTMLInputElement>\n | React.ChangeEvent<HTMLTextAreaElement>,\n ) => void;\n\n /**\n * Form submission handler to automatically reset input and append a user message\n * @example\n * ```jsx\n * <form onSubmit={handleSubmit}>\n * <input onChange={handleInputChange} value={input} />\n * </form>\n * ```\n */\n handleSubmit: (event?: { preventDefault?: () => void }) => void;\n\n /** Whether the API request is in progress */\n isLoading: boolean;\n};\n\nexport function useCompletion({\n api = '/api/completion',\n id,\n initialCompletion = '',\n initialInput = '',\n credentials,\n headers,\n body,\n streamProtocol = 'data',\n fetch,\n onFinish,\n onError,\n experimental_throttle: throttleWaitMs,\n}: UseCompletionOptions & {\n /**\n * Custom throttle wait in ms for the completion and data updates.\n * Default is undefined, which disables throttling.\n */\n experimental_throttle?: number;\n} = {}): UseCompletionHelpers {\n // Generate an unique id for the completion if not provided.\n const hookId = useId();\n const completionId = id || hookId;\n\n // Store the completion state in SWR, using the completionId as the key to share states.\n const { data, mutate } = useSWR<string>([api, completionId], null, {\n fallbackData: initialCompletion,\n });\n\n const { data: isLoading = false, mutate: mutateLoading } = useSWR<boolean>(\n [completionId, 'loading'],\n null,\n );\n\n const [error, setError] = useState<undefined | Error>(undefined);\n const completion = data!;\n\n // Abort controller to cancel the current API call.\n const [abortController, setAbortController] =\n useState<AbortController | null>(null);\n\n const extraMetadataRef = useRef({\n credentials,\n headers,\n body,\n });\n\n useEffect(() => {\n extraMetadataRef.current = {\n credentials,\n headers,\n body,\n };\n }, [credentials, headers, body]);\n\n const triggerRequest = useCallback(\n async (prompt: string, options?: CompletionRequestOptions) =>\n callCompletionApi({\n api,\n prompt,\n credentials: extraMetadataRef.current.credentials,\n headers: { ...extraMetadataRef.current.headers, ...options?.headers },\n body: {\n ...extraMetadataRef.current.body,\n ...options?.body,\n },\n streamProtocol,\n fetch,\n // throttle streamed ui updates:\n setCompletion: throttle(\n (completion: string) => mutate(completion, false),\n throttleWaitMs,\n ),\n setLoading: mutateLoading,\n setError,\n setAbortController,\n onFinish,\n onError,\n }),\n [\n mutate,\n mutateLoading,\n api,\n extraMetadataRef,\n setAbortController,\n onFinish,\n onError,\n setError,\n streamProtocol,\n fetch,\n throttleWaitMs,\n ],\n );\n\n const stop = useCallback(() => {\n if (abortController) {\n abortController.abort();\n setAbortController(null);\n }\n }, [abortController]);\n\n const setCompletion = useCallback(\n (completion: string) => {\n mutate(completion, false);\n },\n [mutate],\n );\n\n const complete = useCallback<UseCompletionHelpers['complete']>(\n async (prompt, options) => {\n return triggerRequest(prompt, options);\n },\n [triggerRequest],\n );\n\n const [input, setInput] = useState(initialInput);\n\n const handleSubmit = useCallback(\n (event?: { preventDefault?: () => void }) => {\n event?.preventDefault?.();\n if (!input) {\n return;\n }\n\n const result = complete(input);\n setInput('');\n return result;\n },\n [input, complete, setInput],\n );\n\n const handleInputChange = useCallback(\n (e: any) => {\n setInput(e.target.value);\n },\n [setInput],\n );\n\n return {\n completion,\n complete,\n error,\n setCompletion,\n stop,\n input,\n setInput,\n handleInputChange,\n handleSubmit,\n isLoading,\n };\n}\n","import {\n isAbortError,\n resolve,\n normalizeHeaders,\n safeValidateTypes,\n type FetchFunction,\n type FlexibleSchema,\n type InferSchema,\n type Resolvable,\n} from '@ai-sdk/provider-utils';\nimport {\n asSchema,\n isDeepEqualData,\n parsePartialJson,\n type DeepPartial,\n} from 'ai';\nimport { useCallback, useId, useRef, useState } from 'react';\nimport useSWR from 'swr';\n\n// use function to allow for mocking in tests:\nconst getOriginalFetch = () => fetch;\n\nexport type Experimental_UseObjectOptions<\n SCHEMA extends FlexibleSchema,\n RESULT,\n> = {\n /**\n * The API endpoint. It should stream JSON that matches the schema as chunked text.\n */\n api: string;\n\n /**\n * A schema that defines the shape of the complete object.\n */\n schema: SCHEMA;\n\n /**\n * An unique identifier. If not provided, a random one will be\n * generated. When provided, the `useObject` hook with the same `id` will\n * have shared states across components.\n */\n id?: string;\n\n /**\n * An optional value for the initial object.\n */\n initialValue?: DeepPartial<RESULT>;\n\n /**\n * Custom fetch implementation. You can use it as a middleware to intercept requests,\n * or to provide a custom fetch implementation for e.g. testing.\n */\n fetch?: FetchFunction;\n\n /**\n * Callback that is called when the stream has finished.\n */\n onFinish?: (event: {\n /**\n * The generated object (typed according to the schema).\n * Can be undefined if the final object does not match the schema.\n */\n object: RESULT | undefined;\n\n /**\n * Optional error object. This is e.g. a TypeValidationError when the final object does not match the schema.\n */\n error: Error | undefined;\n }) => Promise<void> | void;\n\n /**\n * Callback function to be called when an error is encountered.\n */\n onError?: (error: Error) => void;\n\n /**\n * Additional HTTP headers to be included in the request.\n * Can be a static object, a function that returns headers, or an async function\n * for dynamic auth tokens.\n */\n headers?: Resolvable<Record<string, string> | Headers>;\n\n /**\n * The credentials mode to be used for the fetch request.\n * Possible values are: 'omit', 'same-origin', 'include'.\n * Defaults to 'same-origin'.\n */\n credentials?: RequestCredentials;\n};\n\nexport type Experimental_UseObjectHelpers<RESULT, INPUT> = {\n /**\n * Calls the API with the provided input as JSON body.\n */\n submit: (input: INPUT) => void;\n\n /**\n * The current value for the generated object. Updated as the API streams JSON chunks.\n */\n object: DeepPartial<RESULT> | undefined;\n\n /**\n * The error object of the API request if any.\n */\n error: Error | undefined;\n\n /**\n * Flag that indicates whether an API request is in progress.\n */\n isLoading: boolean;\n\n /**\n * Abort the current request immediately, keep the current partial object if any.\n */\n stop: () => void;\n\n /**\n * Clear the object state.\n */\n clear: () => void;\n};\n\nfunction useObject<\n SCHEMA extends FlexibleSchema,\n RESULT = InferSchema<SCHEMA>,\n INPUT = any,\n>({\n api,\n id,\n schema, // required, in the future we will use it for validation\n initialValue,\n fetch,\n onError,\n onFinish,\n headers,\n credentials,\n}: Experimental_UseObjectOptions<\n SCHEMA,\n RESULT\n>): Experimental_UseObjectHelpers<RESULT, INPUT> {\n // Generate an unique id if not provided.\n const hookId = useId();\n const completionId = id ?? hookId;\n\n // Store the completion state in SWR, using the completionId as the key to share states.\n const { data, mutate } = useSWR<DeepPartial<RESULT>>(\n [completionId, 'object'],\n null,\n { fallbackData: initialValue },\n );\n\n const [error, setError] = useState<undefined | Error>(undefined);\n const [isLoading, setIsLoading] = useState(false);\n\n // Abort controller to cancel the current API call.\n const abortControllerRef = useRef<AbortController | null>(null);\n\n const stop = useCallback(() => {\n try {\n abortControllerRef.current?.abort();\n } catch (ignored) {\n } finally {\n setIsLoading(false);\n abortControllerRef.current = null;\n }\n }, []);\n\n const submit = async (input: INPUT) => {\n const abortController = new AbortController();\n\n try {\n clearObject();\n\n setIsLoading(true);\n\n abortControllerRef.current = abortController;\n\n // Resolve headers at request time (supports async functions for dynamic auth tokens)\n const resolvedHeaders = await resolve(headers);\n\n const actualFetch = fetch ?? getOriginalFetch();\n const response = await actualFetch(api, {\n method: 'POST',\n headers: {\n 'Content-Type': 'application/json',\n ...normalizeHeaders(resolvedHeaders),\n },\n credentials,\n signal: abortController.signal,\n body: JSON.stringify(input),\n });\n\n if (!response.ok) {\n throw new Error(\n (await response.text()) ?? 'Failed to fetch the response.',\n );\n }\n\n if (response.body == null) {\n throw new Error('The response body is empty.');\n }\n\n let accumulatedText = '';\n let latestObject: DeepPartial<RESULT> | undefined = undefined;\n\n await response.body.pipeThrough(new TextDecoderStream()).pipeTo(\n new WritableStream<string>({\n async write(chunk) {\n accumulatedText += chunk;\n\n const { value } = await parsePartialJson(accumulatedText);\n const currentObject = value as DeepPartial<RESULT>;\n\n if (!isDeepEqualData(latestObject, currentObject)) {\n latestObject = currentObject;\n\n if (abortControllerRef.current === abortController) {\n mutate(currentObject);\n }\n }\n },\n\n async close() {\n if (abortControllerRef.current === abortController) {\n setIsLoading(false);\n }\n\n if (onFinish != null) {\n const validationResult = await safeValidateTypes({\n value: latestObject,\n schema: asSchema(schema),\n });\n\n onFinish(\n validationResult.success\n ? { object: validationResult.value, error: undefined }\n : { object: undefined, error: validationResult.error },\n );\n }\n\n if (abortControllerRef.current === abortController) {\n abortControllerRef.current = null;\n }\n },\n }),\n );\n } catch (error) {\n if (isAbortError(error)) {\n return;\n }\n\n if (onError && error instanceof Error) {\n onError(error);\n }\n\n if (abortControllerRef.current === abortController) {\n setIsLoading(false);\n abortControllerRef.current = null;\n setError(error instanceof Error ? error : new Error(String(error)));\n }\n }\n };\n\n const clear = () => {\n stop();\n clearObject();\n };\n\n const clearObject = () => {\n setError(undefined);\n setIsLoading(false);\n mutate(undefined);\n };\n\n return {\n submit,\n object: data,\n error,\n isLoading,\n stop,\n clear,\n };\n}\n\nexport const experimental_useObject = useObject;\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AAAA;AAAA,EAME;AAAA,OACK;AACP,SAAS,aAAa,WAAW,QAAQ,4BAA4B;;;ACRrE;AAAA,EACE;AAAA,OAKK;;;ACNP,OAAO,sBAAsB;AAEtB,SAAS,SACd,IACA,QACG;AACH,SAAO,UAAU,OAAO,iBAAiB,IAAI,MAAM,IAAI;AACzD;;;ADEA,SAAS,cAAwB,UAA8B;AAC7D,MAAI,MAAM,QAAQ,QAAQ,GAAG;AAC3B,WAAO,CAAC,GAAG,QAAQ;AAAA,EACrB;AAEA,MACE,YAAY,QACZ,OAAO,aAAa,aACnB,OAAO,eAAe,QAAQ,MAAM,OAAO,aAC1C,OAAO,eAAe,QAAQ,MAAM,OACtC;AACA,WAAO,EAAE,GAAG,SAAS;AAAA,EACvB;AAEA,SAAO;AACT;AAxBA;AA0BA,IAAM,iBAAN,MAEmC;AAAA,EASjC,YAAY,kBAAgC,CAAC,GAAG;AARhD;AACA,gCAAsB;AACtB,+BAA4B;AAE5B,2CAAqB,oBAAI,IAAgB;AACzC,yCAAmB,oBAAI,IAAgB;AACvC,wCAAkB,oBAAI,IAAgB;AAiCtC,uBAAc,CAAC,YAAwB;AACrC,yBAAK,WAAY,mBAAK,WAAU,OAAO,OAAO;AAC9C,yBAAK,wBAAL;AAAA,IACF;AAEA,sBAAa,MAAM;AACjB,yBAAK,WAAY,mBAAK,WAAU,MAAM,GAAG,EAAE;AAC3C,yBAAK,wBAAL;AAAA,IACF;AAEA,0BAAiB,CAAC,OAAe,YAAwB;AACvD,yBAAK,WAAY;AAAA,QACf,GAAG,mBAAK,WAAU,MAAM,GAAG,KAAK;AAAA,QAChC,KAAK,SAAS,OAAO;AAAA,QACrB,GAAG,mBAAK,WAAU,MAAM,QAAQ,CAAC;AAAA,MACnC;AACA,yBAAK,wBAAL;AAAA,IACF;AAEA,oBAAW,CAAI,UAAgB;AAC7B,UACE,SAAS,QACT,OAAO,UAAU,YACjB,EAAE,WAAW,UACb,CAAC,MAAM,QAAQ,MAAM,KAAK,GAC1B;AACA,eAAO;AAAA,MACT;AAEA,YAAM,UAAU;AAChB,YAAM,WAAW;AAAA,QACf,GAAG;AAAA,QACH,OAAO,QAAQ,MAAM,IAAI,WAAS,EAAE,GAAG,KAAK,EAAE;AAAA,MAChD;AAEA,UAAI,cAAc,SAAS;AACzB,iBAAS,WAAW,cAAc,QAAQ,QAAQ;AAAA,MACpD;AAEA,aAAO;AAAA,IACT;AAEA,wCAA8B,CAC5B,UACA,mBACiB;AACjB,YAAM,WAAW,iBACb,SAAS,UAAU,cAAc,IACjC;AACJ,yBAAK,oBAAmB,IAAI,QAAQ;AACpC,aAAO,MAAM;AACX,2BAAK,oBAAmB,OAAO,QAAQ;AAAA,MACzC;AAAA,IACF;AAEA,sCAA4B,CAAC,aAAuC;AAClE,yBAAK,kBAAiB,IAAI,QAAQ;AAClC,aAAO,MAAM;AACX,2BAAK,kBAAiB,OAAO,QAAQ;AAAA,MACvC;AAAA,IACF;AAEA,qCAA2B,CAAC,aAAuC;AACjE,yBAAK,iBAAgB,IAAI,QAAQ;AACjC,aAAO,MAAM;AACX,2BAAK,iBAAgB,OAAO,QAAQ;AAAA,MACtC;AAAA,IACF;AAEA,+CAAyB,MAAM;AAC7B,yBAAK,oBAAmB,QAAQ,cAAY,SAAS,CAAC;AAAA,IACxD;AAEA,6CAAuB,MAAM;AAC3B,yBAAK,kBAAiB,QAAQ,cAAY,SAAS,CAAC;AAAA,IACtD;AAEA,4CAAsB,MAAM;AAC1B,yBAAK,iBAAgB,QAAQ,cAAY,SAAS,CAAC;AAAA,IACrD;AA7GE,uBAAK,WAAY;AAAA,EACnB;AAAA,EAEA,IAAI,SAAqB;AACvB,WAAO,mBAAK;AAAA,EACd;AAAA,EAEA,IAAI,OAAO,WAAuB;AAChC,uBAAK,SAAU;AACf,uBAAK,sBAAL;AAAA,EACF;AAAA,EAEA,IAAI,QAA2B;AAC7B,WAAO,mBAAK;AAAA,EACd;AAAA,EAEA,IAAI,MAAM,UAA6B;AACrC,uBAAK,QAAS;AACd,uBAAK,qBAAL;AAAA,EACF;AAAA,EAEA,IAAI,WAAyB;AAC3B,WAAO,mBAAK;AAAA,EACd;AAAA,EAEA,IAAI,SAAS,aAA2B;AACtC,uBAAK,WAAY,CAAC,GAAG,WAAW;AAChC,uBAAK,wBAAL;AAAA,EACF;AAkFF;AAvHE;AACA;AACA;AAEA;AACA;AACA;AAsGA;AAIA;AAIA;AAjJF;AAsJO,IAAM,OAAN,cAEG,aAAyB;AAAA,EAGjC,YAAY,EAAE,UAAU,GAAG,KAAK,GAAyB;AACvD,UAAM,QAAQ,IAAI,eAAe,QAAQ;AACzC,UAAM,EAAE,GAAG,MAAM,MAAM,CAAC;AAJ1B;AAQA,wCAA8B,CAC5B,UACA,mBAEA,mBAAK,QAAO,2BAA2B,EAAE,UAAU,cAAc;AAEnE,sCAA4B,CAAC,aAC3B,mBAAK,QAAO,yBAAyB,EAAE,QAAQ;AAEjD,qCAA2B,CAAC,aAC1B,mBAAK,QAAO,wBAAwB,EAAE,QAAQ;AAb9C,uBAAK,QAAS;AAAA,EAChB;AAaF;AAnBE;;;ADxFF,IAAM,wBAAwB,oBAAI,QAAsC;AAYxE,SAAS,gBAA8C;AAAA,EACrD;AAAA,EACA;AACF,GAGG;AACD,MACE,SAAS,oBAAoB,aAC7B,MAAM,2BAA2B,MACjC;AACA;AAAA,EACF;AAEA,QAAM,eAAe,MACnB,sBAAsB,IAAI,IAAI,MAAM,SAAS,MAAM,cAAc,OAAO;AAE1E,QAAM,qBAAqB,MAAM;AAC/B,UAAM,0BAA0B;AAAA,EAClC;AAEA,QAAM,gBAAgB;AAEtB,QAAM,0BAA0B,cAAc,6BAA6B,EAAE;AAAA,IAC3E,cAAc,MACZ,aAAa,KAAK,SAAS,oBAAoB;AAAA,EACnD,CAAC,EAAE,KAAK,oBAAoB,kBAAkB;AAChD;AAEA,SAAS,wBAAsD;AAAA,EAC7D;AAAA,EACA;AACF,GAGG;AACD,MAAI,QAAQ,sBAAsB,IAAI,IAAI;AAE1C,MAAI,SAAS,MAAM;AACjB,UAAM,WAAiC,EAAE,eAAe,oBAAI,IAAI,EAAE;AAClE,0BAAsB,IAAI,MAAM,QAAQ;AACxC,YAAQ;AAER,QAAI,OAAO,aAAa,aAAa;AACnC,YAAM,qBAAqB,MACzB,gBAAgB,EAAE,MAAM,OAAO,SAAS,CAAC;AAE3C,eAAS,iBAAiB,oBAAoB,kBAAkB;AAChE,eAAS,4BAA4B,MAAM;AACzC,iBAAS,oBAAoB,oBAAoB,kBAAkB;AAAA,MACrE;AAAA,IACF;AAAA,EACF;AAEA,QAAM,EAAE,cAAc,IAAI;AAC1B,QAAM,eAAe,cAAc,SAAS;AAC5C,gBAAc,IAAI,YAAY;AAE9B,MACE,gBACA,KAAK,WAAW,eAChB,KAAK,WAAW,aAChB;AACA,SAAK,KAAK,aAAa;AAAA,EACzB;AAEA,SAAO,MAAM;AA/If;AAgJI,kBAAc,OAAO,YAAY;AAEjC,QAAI,cAAc,SAAS,GAAG;AAC5B,kBAAM,8BAAN;AACA,4BAAsB,OAAO,IAAI;AAAA,IACnC;AAAA,EACF;AACF;AACO,SAAS,QAAkD;AAAA,EAChE,uBAAuB;AAAA,EACvB,SAAS;AAAA,EACT,GAAG;AACL,IAAgC,CAAC,GAA+B;AAC9D,QAAM,8BAA8B,OAAO,CAAC,CAAC;AAO7C,QAAM,YAAY,OAYhB,CAAC,CAAC;AAEJ,MAAI,EAAE,UAAU,UAAU;AACxB,cAAU,UAAU;AAAA,MAClB,YAAY,QAAQ;AAAA,MACpB,QAAQ,QAAQ;AAAA,MAChB,UAAU,QAAQ;AAAA,MAClB,SAAS,QAAQ;AAAA,MACjB,uBAAuB,QAAQ;AAAA,MAC/B,WAAW,QAAQ;AAAA,IACrB;AAAA,EACF;AAGA,MAAI;AACJ,QAAM,eAAe,MAAG;AA/L1B;AAgMI,2BAAU,QAAQ,cAAlB,YACC,iEAAqB,IAAI,qBAAiC;AAAA;AAG7D,QAAM,cAA8B;AAAA,IAClC,GAAG;AAAA,IACH,WAAW;AAAA,MACT,cAAc,iBAAe,aAAa,EAAE,aAAa,WAAW;AAAA,MACpE,mBAAmB,sBACjB,aAAa,EAAE,kBAAkB,gBAAgB;AAAA,IACrD;AAAA,IACA,YAAY,SAAI;AA3MpB;AA2MuB,mCAAU,SAAQ,eAAlB,4BAA+B;AAAA;AAAA,IAClD,QAAQ,SAAI;AA5MhB;AA4MmB,mCAAU,SAAQ,WAAlB,4BAA2B;AAAA;AAAA,IAC1C,UAAU,SAAI;AA7MlB;AA6MqB,mCAAU,SAAQ,aAAlB,4BAA6B;AAAA;AAAA,IAC9C,SAAS,SAAI;AA9MjB;AA8MoB,mCAAU,SAAQ,YAAlB,4BAA4B;AAAA;AAAA,IAC5C,uBAAuB,SAAI;AA/M/B;AAgNM,yCAAU,SAAQ,0BAAlB,4BAA0C,SAA1C,YAAkD;AAAA;AAAA,EACtD;AAEA,QAAM,UAAU;AAAA,IACd,UAAU,UAAU,QAAQ,OAAO,IAAI,KAAK,WAAW;AAAA,EACzD;AAEA,QAAM,qBACH,UAAU,WAAW,QAAQ,SAAS,QAAQ,WAC9C,QAAQ,WACP,QAAQ,MAAM,QACd,QAAQ,QAAQ,OAAO,QAAQ;AAEnC,MAAI,oBAAoB;AACtB,YAAQ,UAAU,UAAU,UAAU,QAAQ,OAAO,IAAI,KAAK,WAAW;AAAA,EAC3E;AAEA,QAAM,OAAO,QAAQ;AACrB,QAAM,sBAAsB,OAAO;AAAA,IACjC;AAAA,IACA,UAAU,KAAK;AAAA,EACjB,CAAC;AAED,MAAI,oBAAoB,QAAQ,SAAS,MAAM;AAC7C,wBAAoB,UAAU,EAAE,MAAM,UAAU,KAAK,SAAS;AAAA,EAChE;AAEA,QAAM,sBAAsB;AAAA,IAC1B,CAAC,WAAuB;AACtB,UAAI,eAAe;AAEnB,YAAM,iBAAiB,MAAM;AAC3B,YAAI,CAAC,gBAAgB,oBAAoB,QAAQ,SAAS,MAAM;AAC9D;AAAA,QACF;AAEA,4BAAoB,UAAU,EAAE,MAAM,UAAU,KAAK,SAAS;AAC9D,eAAO;AAAA,MACT;AAEA,YAAM,cAAc,KAAK,2BAA2B;AAAA,QAClD;AAAA,QACA;AAAA,MACF;AAKA,0BAAoB,UAAU,EAAE,MAAM,UAAU,KAAK,SAAS;AAE9D,aAAO,MAAM;AACX,uBAAe;AACf,oBAAY;AAAA,MACd;AAAA,IACF;AAAA,IACA,CAAC,MAAM,cAAc;AAAA,EACvB;AAEA,QAAM,sBAAsB;AAAA,IAC1B,MAAM,oBAAoB,QAAQ;AAAA,IAClC,CAAC;AAAA,EACH;AAEA,QAAM,WAAW;AAAA,IACf;AAAA,IACA;AAAA,IACA;AAAA,EACF;AAEA,QAAM,oBAAoB;AAAA,IACxB,CAAC,WACC,KAAK,yBAAyB,EAAE,MAAM;AACpC,UAAI,oBAAoB,QAAQ,SAAS,MAAM;AAC7C;AAAA,MACF;AAEA,UAAI,KAAK,WAAW,WAAW,KAAK,WAAW,SAAS;AAEtD,4BAAoB,UAAU,EAAE,MAAM,UAAU,KAAK,SAAS;AAAA,MAChE;AAEA,aAAO;AAAA,IACT,CAAC;AAAA,IACH,CAAC,IAAI;AAAA,EACP;AAEA,QAAM,oBAAoB,YAAY,MAAM,KAAK,QAAQ,CAAC,IAAI,CAAC;AAE/D,QAAM,SAAS;AAAA,IACb;AAAA,IACA;AAAA,IACA;AAAA,EACF;AAEA,QAAM,QAAQ;AAAA,IACZ,QAAQ,QAAQ,wBAAwB;AAAA,IACxC,MAAM,QAAQ,QAAQ;AAAA,IACtB,MAAM,QAAQ,QAAQ;AAAA,EACxB;AAEA,QAAM,cAAc;AAAA,IAClB,CACE,kBACG;AACH,UAAI,OAAO,kBAAkB,YAAY;AACvC,wBAAgB,cAAc,QAAQ,QAAQ,QAAQ;AAAA,MACxD;AACA,cAAQ,QAAQ,WAAW;AAAA,IAC7B;AAAA,IACA,CAAC,OAAO;AAAA,EACV;AAEA,YAAU,MAAM;AACd,QAAI,QAAQ;AACV,aAAO,wBAAwB;AAAA,QAC7B;AAAA,QACA,cAAc,4BAA4B;AAAA,MAC5C,CAAC;AAAA,IACH;AAAA,EACF,GAAG,CAAC,QAAQ,IAAI,CAAC;AAEjB,SAAO;AAAA,IACL,IAAI,QAAQ,QAAQ;AAAA,IACpB;AAAA,IACA;AAAA,IACA,aAAa,QAAQ,QAAQ;AAAA,IAC7B,YAAY,QAAQ,QAAQ;AAAA,IAC5B,YAAY,QAAQ,QAAQ;AAAA,IAC5B,MAAM,QAAQ,QAAQ;AAAA,IACtB;AAAA,IACA,cAAc,QAAQ,QAAQ;AAAA,IAC9B;AAAA;AAAA;AAAA;AAAA,IAIA,eAAe,QAAQ,QAAQ;AAAA,IAC/B,eAAe,QAAQ,QAAQ;AAAA,IAC/B,yBAAyB,QAAQ,QAAQ;AAAA,EAC3C;AACF;;;AG3VA;AAAA,EACE;AAAA,OAGK;AACP,SAAS,eAAAA,cAAa,aAAAC,YAAW,OAAO,UAAAC,SAAQ,gBAAgB;AAChE,OAAO,YAAY;AAyDZ,SAAS,cAAc;AAAA,EAC5B,MAAM;AAAA,EACN;AAAA,EACA,oBAAoB;AAAA,EACpB,eAAe;AAAA,EACf;AAAA,EACA;AAAA,EACA;AAAA,EACA,iBAAiB;AAAA,EACjB,OAAAC;AAAA,EACA;AAAA,EACA;AAAA,EACA,uBAAuB;AACzB,IAMI,CAAC,GAAyB;AAE5B,QAAM,SAAS,MAAM;AACrB,QAAM,eAAe,MAAM;AAG3B,QAAM,EAAE,MAAM,OAAO,IAAI,OAAe,CAAC,KAAK,YAAY,GAAG,MAAM;AAAA,IACjE,cAAc;AAAA,EAChB,CAAC;AAED,QAAM,EAAE,MAAM,YAAY,OAAO,QAAQ,cAAc,IAAI;AAAA,IACzD,CAAC,cAAc,SAAS;AAAA,IACxB;AAAA,EACF;AAEA,QAAM,CAAC,OAAO,QAAQ,IAAI,SAA4B,MAAS;AAC/D,QAAM,aAAa;AAGnB,QAAM,CAAC,iBAAiB,kBAAkB,IACxC,SAAiC,IAAI;AAEvC,QAAM,mBAAmBC,QAAO;AAAA,IAC9B;AAAA,IACA;AAAA,IACA;AAAA,EACF,CAAC;AAED,EAAAC,WAAU,MAAM;AACd,qBAAiB,UAAU;AAAA,MACzB;AAAA,MACA;AAAA,MACA;AAAA,IACF;AAAA,EACF,GAAG,CAAC,aAAa,SAAS,IAAI,CAAC;AAE/B,QAAM,iBAAiBC;AAAA,IACrB,OAAO,QAAgB,YACrB,kBAAkB;AAAA,MAChB;AAAA,MACA;AAAA,MACA,aAAa,iBAAiB,QAAQ;AAAA,MACtC,SAAS,EAAE,GAAG,iBAAiB,QAAQ,SAAS,GAAG,mCAAS,QAAQ;AAAA,MACpE,MAAM;AAAA,QACJ,GAAG,iBAAiB,QAAQ;AAAA,QAC5B,GAAG,mCAAS;AAAA,MACd;AAAA,MACA;AAAA,MACA,OAAAH;AAAA;AAAA,MAEA,eAAe;AAAA,QACb,CAACI,gBAAuB,OAAOA,aAAY,KAAK;AAAA,QAChD;AAAA,MACF;AAAA,MACA,YAAY;AAAA,MACZ;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,IACF,CAAC;AAAA,IACH;AAAA,MACE;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACAJ;AAAA,MACA;AAAA,IACF;AAAA,EACF;AAEA,QAAM,OAAOG,aAAY,MAAM;AAC7B,QAAI,iBAAiB;AACnB,sBAAgB,MAAM;AACtB,yBAAmB,IAAI;AAAA,IACzB;AAAA,EACF,GAAG,CAAC,eAAe,CAAC;AAEpB,QAAM,gBAAgBA;AAAA,IACpB,CAACC,gBAAuB;AACtB,aAAOA,aAAY,KAAK;AAAA,IAC1B;AAAA,IACA,CAAC,MAAM;AAAA,EACT;AAEA,QAAM,WAAWD;AAAA,IACf,OAAO,QAAQ,YAAY;AACzB,aAAO,eAAe,QAAQ,OAAO;AAAA,IACvC;AAAA,IACA,CAAC,cAAc;AAAA,EACjB;AAEA,QAAM,CAAC,OAAO,QAAQ,IAAI,SAAS,YAAY;AAE/C,QAAM,eAAeA;AAAA,IACnB,CAAC,UAA4C;AArLjD;AAsLM,2CAAO,mBAAP;AACA,UAAI,CAAC,OAAO;AACV;AAAA,MACF;AAEA,YAAM,SAAS,SAAS,KAAK;AAC7B,eAAS,EAAE;AACX,aAAO;AAAA,IACT;AAAA,IACA,CAAC,OAAO,UAAU,QAAQ;AAAA,EAC5B;AAEA,QAAM,oBAAoBA;AAAA,IACxB,CAAC,MAAW;AACV,eAAS,EAAE,OAAO,KAAK;AAAA,IACzB;AAAA,IACA,CAAC,QAAQ;AAAA,EACX;AAEA,SAAO;AAAA,IACL;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF;AACF;;;ACrNA;AAAA,EACE;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,OAKK;AACP;AAAA,EACE;AAAA,EACA;AAAA,EACA;AAAA,OAEK;AACP,SAAS,eAAAE,cAAa,SAAAC,QAAO,UAAAC,SAAQ,YAAAC,iBAAgB;AACrD,OAAOC,aAAY;AAGnB,IAAM,mBAAmB,MAAM;AAsG/B,SAAS,UAIP;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA;AAAA,EACA;AAAA,EACA,OAAAC;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AACF,GAGiD;AAE/C,QAAM,SAASJ,OAAM;AACrB,QAAM,eAAe,kBAAM;AAG3B,QAAM,EAAE,MAAM,OAAO,IAAIG;AAAA,IACvB,CAAC,cAAc,QAAQ;AAAA,IACvB;AAAA,IACA,EAAE,cAAc,aAAa;AAAA,EAC/B;AAEA,QAAM,CAAC,OAAO,QAAQ,IAAID,UAA4B,MAAS;AAC/D,QAAM,CAAC,WAAW,YAAY,IAAIA,UAAS,KAAK;AAGhD,QAAM,qBAAqBD,QAA+B,IAAI;AAE9D,QAAM,OAAOF,aAAY,MAAM;AA7JjC;AA8JI,QAAI;AACF,+BAAmB,YAAnB,mBAA4B;AAAA,IAC9B,SAAS,SAAS;AAAA,IAClB,UAAE;AACA,mBAAa,KAAK;AAClB,yBAAmB,UAAU;AAAA,IAC/B;AAAA,EACF,GAAG,CAAC,CAAC;AAEL,QAAM,SAAS,OAAO,UAAiB;AAvKzC;AAwKI,UAAM,kBAAkB,IAAI,gBAAgB;AAE5C,QAAI;AACF,kBAAY;AAEZ,mBAAa,IAAI;AAEjB,yBAAmB,UAAU;AAG7B,YAAM,kBAAkB,MAAM,QAAQ,OAAO;AAE7C,YAAM,cAAcK,UAAA,OAAAA,SAAS,iBAAiB;AAC9C,YAAM,WAAW,MAAM,YAAY,KAAK;AAAA,QACtC,QAAQ;AAAA,QACR,SAAS;AAAA,UACP,gBAAgB;AAAA,UAChB,GAAG,iBAAiB,eAAe;AAAA,QACrC;AAAA,QACA;AAAA,QACA,QAAQ,gBAAgB;AAAA,QACxB,MAAM,KAAK,UAAU,KAAK;AAAA,MAC5B,CAAC;AAED,UAAI,CAAC,SAAS,IAAI;AAChB,cAAM,IAAI;AAAA,WACP,WAAM,SAAS,KAAK,MAApB,YAA0B;AAAA,QAC7B;AAAA,MACF;AAEA,UAAI,SAAS,QAAQ,MAAM;AACzB,cAAM,IAAI,MAAM,6BAA6B;AAAA,MAC/C;AAEA,UAAI,kBAAkB;AACtB,UAAI,eAAgD;AAEpD,YAAM,SAAS,KAAK,YAAY,IAAI,kBAAkB,CAAC,EAAE;AAAA,QACvD,IAAI,eAAuB;AAAA,UACzB,MAAM,MAAM,OAAO;AACjB,+BAAmB;AAEnB,kBAAM,EAAE,MAAM,IAAI,MAAM,iBAAiB,eAAe;AACxD,kBAAM,gBAAgB;AAEtB,gBAAI,CAAC,gBAAgB,cAAc,aAAa,GAAG;AACjD,6BAAe;AAEf,kBAAI,mBAAmB,YAAY,iBAAiB;AAClD,uBAAO,aAAa;AAAA,cACtB;AAAA,YACF;AAAA,UACF;AAAA,UAEA,MAAM,QAAQ;AACZ,gBAAI,mBAAmB,YAAY,iBAAiB;AAClD,2BAAa,KAAK;AAAA,YACpB;AAEA,gBAAI,YAAY,MAAM;AACpB,oBAAM,mBAAmB,MAAM,kBAAkB;AAAA,gBAC/C,OAAO;AAAA,gBACP,QAAQ,SAAS,MAAM;AAAA,cACzB,CAAC;AAED;AAAA,gBACE,iBAAiB,UACb,EAAE,QAAQ,iBAAiB,OAAO,OAAO,OAAU,IACnD,EAAE,QAAQ,QAAW,OAAO,iBAAiB,MAAM;AAAA,cACzD;AAAA,YACF;AAEA,gBAAI,mBAAmB,YAAY,iBAAiB;AAClD,iCAAmB,UAAU;AAAA,YAC/B;AAAA,UACF;AAAA,QACF,CAAC;AAAA,MACH;AAAA,IACF,SAASC,QAAO;AACd,UAAI,aAAaA,MAAK,GAAG;AACvB;AAAA,MACF;AAEA,UAAI,WAAWA,kBAAiB,OAAO;AACrC,gBAAQA,MAAK;AAAA,MACf;AAEA,UAAI,mBAAmB,YAAY,iBAAiB;AAClD,qBAAa,KAAK;AAClB,2BAAmB,UAAU;AAC7B,iBAASA,kBAAiB,QAAQA,SAAQ,IAAI,MAAM,OAAOA,MAAK,CAAC,CAAC;AAAA,MACpE;AAAA,IACF;AAAA,EACF;AAEA,QAAM,QAAQ,MAAM;AAClB,SAAK;AACL,gBAAY;AAAA,EACd;AAEA,QAAM,cAAc,MAAM;AACxB,aAAS,MAAS;AAClB,iBAAa,KAAK;AAClB,WAAO,MAAS;AAAA,EAClB;AAEA,SAAO;AAAA,IACL;AAAA,IACA,QAAQ;AAAA,IACR;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF;AACF;AAEO,IAAM,yBAAyB;","names":["useCallback","useEffect","useRef","fetch","useRef","useEffect","useCallback","completion","useCallback","useId","useRef","useState","useSWR","fetch","error"]}
|
package/package.json
CHANGED
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
{
|
|
2
2
|
"name": "@ai-sdk/react",
|
|
3
|
-
"version": "3.0.
|
|
3
|
+
"version": "3.0.308",
|
|
4
4
|
"license": "Apache-2.0",
|
|
5
5
|
"sideEffects": false,
|
|
6
6
|
"main": "./dist/index.js",
|
|
@@ -28,8 +28,8 @@
|
|
|
28
28
|
"dependencies": {
|
|
29
29
|
"swr": "^2.2.5",
|
|
30
30
|
"throttleit": "2.1.0",
|
|
31
|
-
"@ai-sdk/provider-utils": "4.0.
|
|
32
|
-
"ai": "6.0.
|
|
31
|
+
"@ai-sdk/provider-utils": "4.0.58",
|
|
32
|
+
"ai": "6.0.305"
|
|
33
33
|
},
|
|
34
34
|
"devDependencies": {
|
|
35
35
|
"@testing-library/jest-dom": "^6.6.3",
|
package/src/use-chat.ts
CHANGED
|
@@ -52,16 +52,111 @@ export type UseChatOptions<UI_MESSAGE extends UIMessage> = (
|
|
|
52
52
|
experimental_throttle?: number;
|
|
53
53
|
|
|
54
54
|
/**
|
|
55
|
-
* Whether to resume an ongoing chat generation stream.
|
|
55
|
+
* Whether to automatically resume an ongoing chat generation stream.
|
|
56
56
|
*/
|
|
57
57
|
resume?: boolean;
|
|
58
58
|
};
|
|
59
59
|
|
|
60
|
+
type AutomaticResumeState = {
|
|
61
|
+
registrations: Set<object>;
|
|
62
|
+
cleanupVisibilityListener?: () => void;
|
|
63
|
+
pendingVisibilityResume?: Promise<void>;
|
|
64
|
+
};
|
|
65
|
+
|
|
66
|
+
const automaticResumeStates = new WeakMap<object, AutomaticResumeState>();
|
|
67
|
+
|
|
68
|
+
type AutomaticallyResumableChat = {
|
|
69
|
+
'~resumeStreamIfDisconnected': (options: {
|
|
70
|
+
shouldResume: () => boolean;
|
|
71
|
+
}) => Promise<void>;
|
|
72
|
+
};
|
|
73
|
+
|
|
74
|
+
/**
|
|
75
|
+
* When the document becomes visible again, resume the chat stream if it was
|
|
76
|
+
* interrupted by a network disconnect while the page was in the background.
|
|
77
|
+
*/
|
|
78
|
+
function resumeOnVisible<UI_MESSAGE extends UIMessage>({
|
|
79
|
+
chat,
|
|
80
|
+
state,
|
|
81
|
+
}: {
|
|
82
|
+
chat: Chat<UI_MESSAGE>;
|
|
83
|
+
state: AutomaticResumeState;
|
|
84
|
+
}) {
|
|
85
|
+
if (
|
|
86
|
+
document.visibilityState !== 'visible' ||
|
|
87
|
+
state.pendingVisibilityResume != null
|
|
88
|
+
) {
|
|
89
|
+
return;
|
|
90
|
+
}
|
|
91
|
+
|
|
92
|
+
const isRegistered = () =>
|
|
93
|
+
automaticResumeStates.get(chat) === state && state.registrations.size > 0;
|
|
94
|
+
|
|
95
|
+
const clearPendingResume = () => {
|
|
96
|
+
state.pendingVisibilityResume = undefined;
|
|
97
|
+
};
|
|
98
|
+
|
|
99
|
+
const resumableChat = chat as Chat<UI_MESSAGE> & AutomaticallyResumableChat;
|
|
100
|
+
|
|
101
|
+
state.pendingVisibilityResume = resumableChat['~resumeStreamIfDisconnected']({
|
|
102
|
+
shouldResume: () =>
|
|
103
|
+
isRegistered() && document.visibilityState === 'visible',
|
|
104
|
+
}).then(clearPendingResume, clearPendingResume);
|
|
105
|
+
}
|
|
106
|
+
|
|
107
|
+
function registerAutomaticResume<UI_MESSAGE extends UIMessage>({
|
|
108
|
+
chat,
|
|
109
|
+
registration,
|
|
110
|
+
}: {
|
|
111
|
+
chat: Chat<UI_MESSAGE>;
|
|
112
|
+
registration: object;
|
|
113
|
+
}) {
|
|
114
|
+
let state = automaticResumeStates.get(chat);
|
|
115
|
+
|
|
116
|
+
if (state == null) {
|
|
117
|
+
const newState: AutomaticResumeState = { registrations: new Set() };
|
|
118
|
+
automaticResumeStates.set(chat, newState);
|
|
119
|
+
state = newState;
|
|
120
|
+
|
|
121
|
+
if (typeof document !== 'undefined') {
|
|
122
|
+
const onVisibilityChange = () =>
|
|
123
|
+
resumeOnVisible({ chat, state: newState });
|
|
124
|
+
|
|
125
|
+
document.addEventListener('visibilitychange', onVisibilityChange);
|
|
126
|
+
newState.cleanupVisibilityListener = () => {
|
|
127
|
+
document.removeEventListener('visibilitychange', onVisibilityChange);
|
|
128
|
+
};
|
|
129
|
+
}
|
|
130
|
+
}
|
|
131
|
+
|
|
132
|
+
const { registrations } = state;
|
|
133
|
+
const shouldResume = registrations.size === 0;
|
|
134
|
+
registrations.add(registration);
|
|
135
|
+
|
|
136
|
+
if (
|
|
137
|
+
shouldResume &&
|
|
138
|
+
chat.status !== 'submitted' &&
|
|
139
|
+
chat.status !== 'streaming'
|
|
140
|
+
) {
|
|
141
|
+
void chat.resumeStream();
|
|
142
|
+
}
|
|
143
|
+
|
|
144
|
+
return () => {
|
|
145
|
+
registrations.delete(registration);
|
|
146
|
+
|
|
147
|
+
if (registrations.size === 0) {
|
|
148
|
+
state.cleanupVisibilityListener?.();
|
|
149
|
+
automaticResumeStates.delete(chat);
|
|
150
|
+
}
|
|
151
|
+
};
|
|
152
|
+
}
|
|
60
153
|
export function useChat<UI_MESSAGE extends UIMessage = UIMessage>({
|
|
61
154
|
experimental_throttle: throttleWaitMs,
|
|
62
155
|
resume = false,
|
|
63
156
|
...options
|
|
64
157
|
}: UseChatOptions<UI_MESSAGE> = {}): UseChatHelpers<UI_MESSAGE> {
|
|
158
|
+
const automaticResumeRegistration = useRef({});
|
|
159
|
+
|
|
65
160
|
// the Chat instance is created once and not recreated when options change,
|
|
66
161
|
// so it would normally keep the callbacks/transport from the first render forever
|
|
67
162
|
|
|
@@ -225,9 +320,12 @@ export function useChat<UI_MESSAGE extends UIMessage = UIMessage>({
|
|
|
225
320
|
|
|
226
321
|
useEffect(() => {
|
|
227
322
|
if (resume) {
|
|
228
|
-
|
|
323
|
+
return registerAutomaticResume({
|
|
324
|
+
chat,
|
|
325
|
+
registration: automaticResumeRegistration.current,
|
|
326
|
+
});
|
|
229
327
|
}
|
|
230
|
-
}, [resume,
|
|
328
|
+
}, [resume, chat]);
|
|
231
329
|
|
|
232
330
|
return {
|
|
233
331
|
id: chatRef.current.id,
|