@ai-sdk/react 2.0.0-alpha.3 → 2.0.0-alpha.5
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 +27 -0
- package/dist/index.d.mts +10 -8
- package/dist/index.d.ts +10 -8
- package/dist/index.js +80 -25
- package/dist/index.js.map +1 -1
- package/dist/index.mjs +74 -17
- package/dist/index.mjs.map +1 -1
- package/package.json +3 -3
package/CHANGELOG.md
CHANGED
|
@@ -1,5 +1,32 @@
|
|
|
1
1
|
# @ai-sdk/react
|
|
2
2
|
|
|
3
|
+
## 2.0.0-alpha.5
|
|
4
|
+
|
|
5
|
+
### Patch Changes
|
|
6
|
+
|
|
7
|
+
- Updated dependencies [655cf3c]
|
|
8
|
+
- Updated dependencies [1675396]
|
|
9
|
+
- Updated dependencies [cf9af6e]
|
|
10
|
+
- Updated dependencies [ef256ed]
|
|
11
|
+
- Updated dependencies [1ed0287]
|
|
12
|
+
- Updated dependencies [825e8d7]
|
|
13
|
+
- Updated dependencies [7324c21]
|
|
14
|
+
- ai@5.0.0-alpha.5
|
|
15
|
+
|
|
16
|
+
## 2.0.0-alpha.4
|
|
17
|
+
|
|
18
|
+
### Patch Changes
|
|
19
|
+
|
|
20
|
+
- 6b14724: chore (ai/react): add experimental throttle back to useChat
|
|
21
|
+
- Updated dependencies [b32c141]
|
|
22
|
+
- Updated dependencies [72d7d72]
|
|
23
|
+
- Updated dependencies [9315076]
|
|
24
|
+
- Updated dependencies [7d97ab6]
|
|
25
|
+
- Updated dependencies [37a916d]
|
|
26
|
+
- Updated dependencies [5f2b3d4]
|
|
27
|
+
- ai@5.0.0-alpha.4
|
|
28
|
+
- @ai-sdk/provider-utils@3.0.0-alpha.4
|
|
29
|
+
|
|
3
30
|
## 2.0.0-alpha.3
|
|
4
31
|
|
|
5
32
|
### Patch Changes
|
package/dist/index.d.mts
CHANGED
|
@@ -1,9 +1,9 @@
|
|
|
1
|
-
import {
|
|
1
|
+
import { UIDataPartSchemas, UIMessage, InferUIDataParts, CreateUIMessage, ChatRequestOptions, FileUIPart, UseChatOptions, CompletionRequestOptions, UseCompletionOptions, Schema, DeepPartial, ChatStoreOptions, ChatStore } from 'ai';
|
|
2
2
|
export { CreateUIMessage, UIMessage, UseChatOptions, UseCompletionOptions } from 'ai';
|
|
3
3
|
import { FetchFunction } from '@ai-sdk/provider-utils';
|
|
4
4
|
import z from 'zod';
|
|
5
5
|
|
|
6
|
-
type UseChatHelpers<MESSAGE_METADATA = unknown,
|
|
6
|
+
type UseChatHelpers<MESSAGE_METADATA = unknown, DATA_PART_SCHEMAS extends UIDataPartSchemas = UIDataPartSchemas> = {
|
|
7
7
|
/**
|
|
8
8
|
* The id of the chat.
|
|
9
9
|
*/
|
|
@@ -18,7 +18,7 @@ type UseChatHelpers<MESSAGE_METADATA = unknown, DATA_TYPE_SCHEMAS extends UIData
|
|
|
18
18
|
*/
|
|
19
19
|
readonly status: 'submitted' | 'streaming' | 'ready' | 'error';
|
|
20
20
|
/** Current messages in the chat */
|
|
21
|
-
readonly messages: UIMessage<MESSAGE_METADATA,
|
|
21
|
+
readonly messages: UIMessage<MESSAGE_METADATA, InferUIDataParts<DATA_PART_SCHEMAS>>[];
|
|
22
22
|
/** The error object of the API request */
|
|
23
23
|
readonly error: undefined | Error;
|
|
24
24
|
/**
|
|
@@ -28,7 +28,7 @@ type UseChatHelpers<MESSAGE_METADATA = unknown, DATA_TYPE_SCHEMAS extends UIData
|
|
|
28
28
|
* @param message The message to append
|
|
29
29
|
* @param options Additional options to pass to the API call
|
|
30
30
|
*/
|
|
31
|
-
append: (message: CreateUIMessage<MESSAGE_METADATA,
|
|
31
|
+
append: (message: CreateUIMessage<MESSAGE_METADATA, InferUIDataParts<DATA_PART_SCHEMAS>>, options?: ChatRequestOptions) => Promise<void>;
|
|
32
32
|
/**
|
|
33
33
|
* Reload the last AI chat response for the given chat history. If the last
|
|
34
34
|
* message isn't from the assistant, it will request the API to generate a
|
|
@@ -48,7 +48,7 @@ type UseChatHelpers<MESSAGE_METADATA = unknown, DATA_TYPE_SCHEMAS extends UIData
|
|
|
48
48
|
* edit the messages on the client, and then trigger the `reload` method
|
|
49
49
|
* manually to regenerate the AI response.
|
|
50
50
|
*/
|
|
51
|
-
setMessages: (messages: UIMessage<MESSAGE_METADATA,
|
|
51
|
+
setMessages: (messages: UIMessage<MESSAGE_METADATA, InferUIDataParts<DATA_PART_SCHEMAS>>[] | ((messages: UIMessage<MESSAGE_METADATA, InferUIDataParts<DATA_PART_SCHEMAS>>[]) => UIMessage<MESSAGE_METADATA, InferUIDataParts<DATA_PART_SCHEMAS>>[])) => void;
|
|
52
52
|
/** The current value of the input */
|
|
53
53
|
input: string;
|
|
54
54
|
/** setState-powered method to update the input value */
|
|
@@ -66,13 +66,13 @@ type UseChatHelpers<MESSAGE_METADATA = unknown, DATA_TYPE_SCHEMAS extends UIData
|
|
|
66
66
|
result: any;
|
|
67
67
|
}) => void;
|
|
68
68
|
};
|
|
69
|
-
declare function useChat<MESSAGE_METADATA = unknown,
|
|
69
|
+
declare function useChat<MESSAGE_METADATA = unknown, DATA_PART_SCHEMAS extends UIDataPartSchemas = UIDataPartSchemas>({ chatId, initialInput, onToolCall, onFinish, onError, generateId, experimental_throttle: throttleWaitMs, chatStore: chatStoreArg, }?: UseChatOptions<MESSAGE_METADATA, DATA_PART_SCHEMAS> & {
|
|
70
70
|
/**
|
|
71
71
|
Custom throttle wait in ms for the chat messages and data updates.
|
|
72
72
|
Default is undefined, which disables throttling.
|
|
73
73
|
*/
|
|
74
74
|
experimental_throttle?: number;
|
|
75
|
-
}): UseChatHelpers<MESSAGE_METADATA,
|
|
75
|
+
}): UseChatHelpers<MESSAGE_METADATA, DATA_PART_SCHEMAS>;
|
|
76
76
|
|
|
77
77
|
type UseCompletionHelpers = {
|
|
78
78
|
/** The current completion result */
|
|
@@ -205,4 +205,6 @@ declare function useObject<RESULT, INPUT = any>({ api, id, schema, // required,
|
|
|
205
205
|
initialValue, fetch, onError, onFinish, headers, credentials, }: Experimental_UseObjectOptions<RESULT>): Experimental_UseObjectHelpers<RESULT, INPUT>;
|
|
206
206
|
declare const experimental_useObject: typeof useObject;
|
|
207
207
|
|
|
208
|
-
|
|
208
|
+
declare function createChatStore<MESSAGE_METADATA = unknown, DATA_PART_SCHEMAS extends UIDataPartSchemas = UIDataPartSchemas>(options: ChatStoreOptions<MESSAGE_METADATA, DATA_PART_SCHEMAS>): ChatStore<MESSAGE_METADATA, DATA_PART_SCHEMAS>;
|
|
209
|
+
|
|
210
|
+
export { Experimental_UseObjectHelpers, Experimental_UseObjectOptions, UseChatHelpers, UseCompletionHelpers, createChatStore, experimental_useObject, useChat, useCompletion };
|
package/dist/index.d.ts
CHANGED
|
@@ -1,9 +1,9 @@
|
|
|
1
|
-
import {
|
|
1
|
+
import { UIDataPartSchemas, UIMessage, InferUIDataParts, CreateUIMessage, ChatRequestOptions, FileUIPart, UseChatOptions, CompletionRequestOptions, UseCompletionOptions, Schema, DeepPartial, ChatStoreOptions, ChatStore } from 'ai';
|
|
2
2
|
export { CreateUIMessage, UIMessage, UseChatOptions, UseCompletionOptions } from 'ai';
|
|
3
3
|
import { FetchFunction } from '@ai-sdk/provider-utils';
|
|
4
4
|
import z from 'zod';
|
|
5
5
|
|
|
6
|
-
type UseChatHelpers<MESSAGE_METADATA = unknown,
|
|
6
|
+
type UseChatHelpers<MESSAGE_METADATA = unknown, DATA_PART_SCHEMAS extends UIDataPartSchemas = UIDataPartSchemas> = {
|
|
7
7
|
/**
|
|
8
8
|
* The id of the chat.
|
|
9
9
|
*/
|
|
@@ -18,7 +18,7 @@ type UseChatHelpers<MESSAGE_METADATA = unknown, DATA_TYPE_SCHEMAS extends UIData
|
|
|
18
18
|
*/
|
|
19
19
|
readonly status: 'submitted' | 'streaming' | 'ready' | 'error';
|
|
20
20
|
/** Current messages in the chat */
|
|
21
|
-
readonly messages: UIMessage<MESSAGE_METADATA,
|
|
21
|
+
readonly messages: UIMessage<MESSAGE_METADATA, InferUIDataParts<DATA_PART_SCHEMAS>>[];
|
|
22
22
|
/** The error object of the API request */
|
|
23
23
|
readonly error: undefined | Error;
|
|
24
24
|
/**
|
|
@@ -28,7 +28,7 @@ type UseChatHelpers<MESSAGE_METADATA = unknown, DATA_TYPE_SCHEMAS extends UIData
|
|
|
28
28
|
* @param message The message to append
|
|
29
29
|
* @param options Additional options to pass to the API call
|
|
30
30
|
*/
|
|
31
|
-
append: (message: CreateUIMessage<MESSAGE_METADATA,
|
|
31
|
+
append: (message: CreateUIMessage<MESSAGE_METADATA, InferUIDataParts<DATA_PART_SCHEMAS>>, options?: ChatRequestOptions) => Promise<void>;
|
|
32
32
|
/**
|
|
33
33
|
* Reload the last AI chat response for the given chat history. If the last
|
|
34
34
|
* message isn't from the assistant, it will request the API to generate a
|
|
@@ -48,7 +48,7 @@ type UseChatHelpers<MESSAGE_METADATA = unknown, DATA_TYPE_SCHEMAS extends UIData
|
|
|
48
48
|
* edit the messages on the client, and then trigger the `reload` method
|
|
49
49
|
* manually to regenerate the AI response.
|
|
50
50
|
*/
|
|
51
|
-
setMessages: (messages: UIMessage<MESSAGE_METADATA,
|
|
51
|
+
setMessages: (messages: UIMessage<MESSAGE_METADATA, InferUIDataParts<DATA_PART_SCHEMAS>>[] | ((messages: UIMessage<MESSAGE_METADATA, InferUIDataParts<DATA_PART_SCHEMAS>>[]) => UIMessage<MESSAGE_METADATA, InferUIDataParts<DATA_PART_SCHEMAS>>[])) => void;
|
|
52
52
|
/** The current value of the input */
|
|
53
53
|
input: string;
|
|
54
54
|
/** setState-powered method to update the input value */
|
|
@@ -66,13 +66,13 @@ type UseChatHelpers<MESSAGE_METADATA = unknown, DATA_TYPE_SCHEMAS extends UIData
|
|
|
66
66
|
result: any;
|
|
67
67
|
}) => void;
|
|
68
68
|
};
|
|
69
|
-
declare function useChat<MESSAGE_METADATA = unknown,
|
|
69
|
+
declare function useChat<MESSAGE_METADATA = unknown, DATA_PART_SCHEMAS extends UIDataPartSchemas = UIDataPartSchemas>({ chatId, initialInput, onToolCall, onFinish, onError, generateId, experimental_throttle: throttleWaitMs, chatStore: chatStoreArg, }?: UseChatOptions<MESSAGE_METADATA, DATA_PART_SCHEMAS> & {
|
|
70
70
|
/**
|
|
71
71
|
Custom throttle wait in ms for the chat messages and data updates.
|
|
72
72
|
Default is undefined, which disables throttling.
|
|
73
73
|
*/
|
|
74
74
|
experimental_throttle?: number;
|
|
75
|
-
}): UseChatHelpers<MESSAGE_METADATA,
|
|
75
|
+
}): UseChatHelpers<MESSAGE_METADATA, DATA_PART_SCHEMAS>;
|
|
76
76
|
|
|
77
77
|
type UseCompletionHelpers = {
|
|
78
78
|
/** The current completion result */
|
|
@@ -205,4 +205,6 @@ declare function useObject<RESULT, INPUT = any>({ api, id, schema, // required,
|
|
|
205
205
|
initialValue, fetch, onError, onFinish, headers, credentials, }: Experimental_UseObjectOptions<RESULT>): Experimental_UseObjectHelpers<RESULT, INPUT>;
|
|
206
206
|
declare const experimental_useObject: typeof useObject;
|
|
207
207
|
|
|
208
|
-
|
|
208
|
+
declare function createChatStore<MESSAGE_METADATA = unknown, DATA_PART_SCHEMAS extends UIDataPartSchemas = UIDataPartSchemas>(options: ChatStoreOptions<MESSAGE_METADATA, DATA_PART_SCHEMAS>): ChatStore<MESSAGE_METADATA, DATA_PART_SCHEMAS>;
|
|
209
|
+
|
|
210
|
+
export { Experimental_UseObjectHelpers, Experimental_UseObjectOptions, UseChatHelpers, UseCompletionHelpers, createChatStore, experimental_useObject, useChat, useCompletion };
|
package/dist/index.js
CHANGED
|
@@ -30,6 +30,7 @@ var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: tru
|
|
|
30
30
|
// src/index.ts
|
|
31
31
|
var src_exports = {};
|
|
32
32
|
__export(src_exports, {
|
|
33
|
+
createChatStore: () => createChatStore,
|
|
33
34
|
experimental_useObject: () => experimental_useObject,
|
|
34
35
|
useChat: () => useChat,
|
|
35
36
|
useCompletion: () => useCompletion
|
|
@@ -37,25 +38,83 @@ __export(src_exports, {
|
|
|
37
38
|
module.exports = __toCommonJS(src_exports);
|
|
38
39
|
|
|
39
40
|
// src/use-chat.ts
|
|
40
|
-
var
|
|
41
|
+
var import_ai2 = require("ai");
|
|
41
42
|
var import_react = require("react");
|
|
43
|
+
|
|
44
|
+
// src/chat-store.ts
|
|
45
|
+
var import_ai = require("ai");
|
|
46
|
+
var ReactChat = class {
|
|
47
|
+
constructor(messages) {
|
|
48
|
+
this.status = "ready";
|
|
49
|
+
this.error = void 0;
|
|
50
|
+
this.activeResponse = void 0;
|
|
51
|
+
this.jobExecutor = new import_ai.SerialJobExecutor();
|
|
52
|
+
this.setStatus = (status) => {
|
|
53
|
+
this.status = status;
|
|
54
|
+
};
|
|
55
|
+
this.setError = (error) => {
|
|
56
|
+
this.error = error;
|
|
57
|
+
};
|
|
58
|
+
this.setActiveResponse = (activeResponse) => {
|
|
59
|
+
this.activeResponse = activeResponse;
|
|
60
|
+
};
|
|
61
|
+
this.setMessages = (messages) => {
|
|
62
|
+
this.messages = [...messages];
|
|
63
|
+
};
|
|
64
|
+
this.pushMessage = (message) => {
|
|
65
|
+
this.messages = this.messages.concat(message);
|
|
66
|
+
};
|
|
67
|
+
this.popMessage = () => {
|
|
68
|
+
this.messages = this.messages.slice(0, -1);
|
|
69
|
+
};
|
|
70
|
+
this.replaceMessage = (index, message) => {
|
|
71
|
+
this.messages = [
|
|
72
|
+
...this.messages.slice(0, index),
|
|
73
|
+
message,
|
|
74
|
+
...this.messages.slice(index + 1)
|
|
75
|
+
];
|
|
76
|
+
};
|
|
77
|
+
this.snapshot = (value) => {
|
|
78
|
+
return structuredClone(value);
|
|
79
|
+
};
|
|
80
|
+
this.messages = messages != null ? messages : [];
|
|
81
|
+
}
|
|
82
|
+
};
|
|
83
|
+
function createChatStore(options) {
|
|
84
|
+
return new import_ai.ChatStore({
|
|
85
|
+
...options,
|
|
86
|
+
createChat: (options2) => new ReactChat(
|
|
87
|
+
options2.messages
|
|
88
|
+
)
|
|
89
|
+
});
|
|
90
|
+
}
|
|
91
|
+
|
|
92
|
+
// src/throttle.ts
|
|
93
|
+
var import_throttleit = __toESM(require("throttleit"));
|
|
94
|
+
function throttle(fn, waitMs) {
|
|
95
|
+
return waitMs != null ? (0, import_throttleit.default)(fn, waitMs) : fn;
|
|
96
|
+
}
|
|
97
|
+
|
|
98
|
+
// src/use-chat.ts
|
|
42
99
|
function useChat({
|
|
43
100
|
chatId,
|
|
44
101
|
initialInput = "",
|
|
45
102
|
onToolCall,
|
|
46
103
|
onFinish,
|
|
47
104
|
onError,
|
|
48
|
-
generateId =
|
|
105
|
+
generateId = import_ai2.generateId,
|
|
49
106
|
experimental_throttle: throttleWaitMs,
|
|
50
107
|
chatStore: chatStoreArg
|
|
51
108
|
} = {}) {
|
|
52
109
|
const [hookId] = (0, import_react.useState)(generateId);
|
|
53
110
|
const stableChatId = chatId != null ? chatId : hookId;
|
|
54
111
|
const chatStore = (0, import_react.useRef)(
|
|
55
|
-
chatStoreArg
|
|
56
|
-
|
|
57
|
-
|
|
58
|
-
|
|
112
|
+
chatStoreArg == null ? createChatStore(
|
|
113
|
+
(0, import_ai2.defaultChatStoreOptions)({
|
|
114
|
+
api: "/api/chat",
|
|
115
|
+
generateId
|
|
116
|
+
})()
|
|
117
|
+
) : typeof chatStoreArg === "function" ? createChatStore(chatStoreArg()) : chatStoreArg
|
|
59
118
|
);
|
|
60
119
|
if (!chatStore.current.hasChat(stableChatId)) {
|
|
61
120
|
chatStore.current.addChat(stableChatId, []);
|
|
@@ -67,9 +126,8 @@ function useChat({
|
|
|
67
126
|
}) => {
|
|
68
127
|
return chatStore.current.subscribe({
|
|
69
128
|
onChatChanged: (event) => {
|
|
70
|
-
if (event.chatId !== stableChatId || event.type !== eventType)
|
|
129
|
+
if (event.chatId !== stableChatId || event.type !== eventType)
|
|
71
130
|
return;
|
|
72
|
-
}
|
|
73
131
|
onStoreChange();
|
|
74
132
|
}
|
|
75
133
|
});
|
|
@@ -99,13 +157,17 @@ function useChat({
|
|
|
99
157
|
() => chatStore.current.getStatus(stableChatId),
|
|
100
158
|
() => chatStore.current.getStatus(stableChatId)
|
|
101
159
|
);
|
|
102
|
-
const
|
|
160
|
+
const subscribeToChatStoreForMessages = (0, import_react.useCallback)(
|
|
103
161
|
(callback) => {
|
|
104
162
|
return subscribe({
|
|
105
|
-
onStoreChange: callback,
|
|
163
|
+
onStoreChange: throttleWaitMs ? throttle(callback, throttleWaitMs) : callback,
|
|
106
164
|
eventType: "chat-messages-changed"
|
|
107
165
|
});
|
|
108
166
|
},
|
|
167
|
+
[subscribe, throttleWaitMs]
|
|
168
|
+
);
|
|
169
|
+
const messages = (0, import_react.useSyncExternalStore)(
|
|
170
|
+
(callback) => subscribeToChatStoreForMessages(callback),
|
|
109
171
|
() => chatStore.current.getMessages(stableChatId),
|
|
110
172
|
() => chatStore.current.getMessages(stableChatId)
|
|
111
173
|
);
|
|
@@ -159,7 +221,7 @@ function useChat({
|
|
|
159
221
|
async (event, options = {}) => {
|
|
160
222
|
var _a;
|
|
161
223
|
(_a = event == null ? void 0 : event.preventDefault) == null ? void 0 : _a.call(event);
|
|
162
|
-
const fileParts = Array.isArray(options == null ? void 0 : options.files) ? options.files : await (0,
|
|
224
|
+
const fileParts = Array.isArray(options == null ? void 0 : options.files) ? options.files : await (0, import_ai2.convertFileListToFileUIParts)(options == null ? void 0 : options.files);
|
|
163
225
|
if (!input && fileParts.length === 0)
|
|
164
226
|
return;
|
|
165
227
|
append(
|
|
@@ -200,17 +262,9 @@ function useChat({
|
|
|
200
262
|
}
|
|
201
263
|
|
|
202
264
|
// src/use-completion.ts
|
|
203
|
-
var
|
|
265
|
+
var import_ai3 = require("ai");
|
|
204
266
|
var import_react2 = require("react");
|
|
205
267
|
var import_swr = __toESM(require("swr"));
|
|
206
|
-
|
|
207
|
-
// src/throttle.ts
|
|
208
|
-
var import_throttleit = __toESM(require("throttleit"));
|
|
209
|
-
function throttle(fn, waitMs) {
|
|
210
|
-
return waitMs != null ? (0, import_throttleit.default)(fn, waitMs) : fn;
|
|
211
|
-
}
|
|
212
|
-
|
|
213
|
-
// src/use-completion.ts
|
|
214
268
|
function useCompletion({
|
|
215
269
|
api = "/api/completion",
|
|
216
270
|
id,
|
|
@@ -250,7 +304,7 @@ function useCompletion({
|
|
|
250
304
|
};
|
|
251
305
|
}, [credentials, headers, body]);
|
|
252
306
|
const triggerRequest = (0, import_react2.useCallback)(
|
|
253
|
-
async (prompt, options) => (0,
|
|
307
|
+
async (prompt, options) => (0, import_ai3.callCompletionApi)({
|
|
254
308
|
api,
|
|
255
309
|
prompt,
|
|
256
310
|
credentials: extraMetadataRef.current.credentials,
|
|
@@ -335,7 +389,7 @@ function useCompletion({
|
|
|
335
389
|
|
|
336
390
|
// src/use-object.ts
|
|
337
391
|
var import_provider_utils = require("@ai-sdk/provider-utils");
|
|
338
|
-
var
|
|
392
|
+
var import_ai4 = require("ai");
|
|
339
393
|
var import_react3 = require("react");
|
|
340
394
|
var import_swr2 = __toESM(require("swr"));
|
|
341
395
|
var getOriginalFetch = () => fetch;
|
|
@@ -404,9 +458,9 @@ function useObject({
|
|
|
404
458
|
new WritableStream({
|
|
405
459
|
async write(chunk) {
|
|
406
460
|
accumulatedText += chunk;
|
|
407
|
-
const { value } = await (0,
|
|
461
|
+
const { value } = await (0, import_ai4.parsePartialJson)(accumulatedText);
|
|
408
462
|
const currentObject = value;
|
|
409
|
-
if (!(0,
|
|
463
|
+
if (!(0, import_ai4.isDeepEqualData)(latestObject, currentObject)) {
|
|
410
464
|
latestObject = currentObject;
|
|
411
465
|
mutate(currentObject);
|
|
412
466
|
}
|
|
@@ -417,7 +471,7 @@ function useObject({
|
|
|
417
471
|
if (onFinish != null) {
|
|
418
472
|
const validationResult = await (0, import_provider_utils.safeValidateTypes)({
|
|
419
473
|
value: latestObject,
|
|
420
|
-
schema: (0,
|
|
474
|
+
schema: (0, import_ai4.asSchema)(schema)
|
|
421
475
|
});
|
|
422
476
|
onFinish(
|
|
423
477
|
validationResult.success ? { object: validationResult.value, error: void 0 } : { object: void 0, error: validationResult.error }
|
|
@@ -448,6 +502,7 @@ function useObject({
|
|
|
448
502
|
var experimental_useObject = useObject;
|
|
449
503
|
// Annotate the CommonJS export names for ESM import in node:
|
|
450
504
|
0 && (module.exports = {
|
|
505
|
+
createChatStore,
|
|
451
506
|
experimental_useObject,
|
|
452
507
|
useChat,
|
|
453
508
|
useCompletion
|
package/dist/index.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"sources":["../src/index.ts","../src/use-chat.ts","../src/use-completion.ts","../src/throttle.ts","../src/use-object.ts"],"sourcesContent":["export * from './use-chat';\nexport * from './use-completion';\nexport * from './use-object';\n","import {\n ChatStore,\n convertFileListToFileUIParts,\n defaultChatStore,\n generateId as generateIdFunc,\n InferUIDataTypes,\n UIDataTypesSchemas,\n type ChatRequestOptions,\n type ChatStoreEvent,\n type CreateUIMessage,\n type FileUIPart,\n type UIMessage,\n type UseChatOptions,\n} from 'ai';\nimport { useCallback, useRef, useState, useSyncExternalStore } from 'react';\n\nexport type { CreateUIMessage, UIMessage, UseChatOptions };\n\nexport type UseChatHelpers<\n MESSAGE_METADATA = unknown,\n DATA_TYPE_SCHEMAS extends UIDataTypesSchemas = UIDataTypesSchemas,\n> = {\n /**\n * The id of the chat.\n */\n readonly chatId: string;\n\n /**\n * Hook status:\n *\n * - `submitted`: The message has been sent to the API and we're awaiting the start of the response stream.\n * - `streaming`: The response is actively streaming in from the API, receiving chunks of data.\n * - `ready`: The full response has been received and processed; a new user message can be submitted.\n * - `error`: An error occurred during the API request, preventing successful completion.\n */\n readonly status: 'submitted' | 'streaming' | 'ready' | 'error';\n\n /** Current messages in the chat */\n readonly messages: UIMessage<\n MESSAGE_METADATA,\n InferUIDataTypes<DATA_TYPE_SCHEMAS>\n >[];\n\n /** The error object of the API request */\n readonly error: undefined | Error;\n\n /**\n * Append a user message to the chat list. This triggers the API call to fetch\n * the assistant's response.\n *\n * @param message The message to append\n * @param options Additional options to pass to the API call\n */\n append: (\n message: CreateUIMessage<\n MESSAGE_METADATA,\n InferUIDataTypes<DATA_TYPE_SCHEMAS>\n >,\n options?: ChatRequestOptions,\n ) => Promise<void>;\n\n /**\n * Reload the last AI chat response for the given chat history. If the last\n * message isn't from the assistant, it will request the API to generate a\n * new response.\n */\n reload: (\n chatRequestOptions?: ChatRequestOptions,\n ) => Promise<string | null | undefined>;\n\n /**\n * Abort the current request immediately, keep the generated tokens if any.\n */\n stop: () => void;\n\n /**\n * Resume an ongoing chat generation stream. This does not resume an aborted generation.\n */\n experimental_resume: () => void;\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:\n | UIMessage<MESSAGE_METADATA, InferUIDataTypes<DATA_TYPE_SCHEMAS>>[]\n | ((\n messages: UIMessage<\n MESSAGE_METADATA,\n InferUIDataTypes<DATA_TYPE_SCHEMAS>\n >[],\n ) => UIMessage<\n MESSAGE_METADATA,\n InferUIDataTypes<DATA_TYPE_SCHEMAS>\n >[]),\n ) => void;\n\n /** The current value of the input */\n input: string;\n\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 handleInputChange: (\n e:\n | React.ChangeEvent<HTMLInputElement>\n | React.ChangeEvent<HTMLTextAreaElement>,\n ) => void;\n\n /** Form submission handler to automatically reset input and append a user message */\n handleSubmit: (\n event?: { preventDefault?: () => void },\n chatRequestOptions?: ChatRequestOptions & {\n files?: FileList | FileUIPart[];\n },\n ) => void;\n\n addToolResult: ({\n toolCallId,\n result,\n }: {\n toolCallId: string;\n result: any;\n }) => void;\n};\n\nexport function useChat<\n MESSAGE_METADATA = unknown,\n DATA_TYPE_SCHEMAS extends UIDataTypesSchemas = UIDataTypesSchemas,\n>({\n chatId,\n initialInput = '',\n onToolCall,\n onFinish,\n onError,\n generateId = generateIdFunc,\n experimental_throttle: throttleWaitMs,\n chatStore: chatStoreArg,\n}: UseChatOptions<MESSAGE_METADATA, DATA_TYPE_SCHEMAS> & {\n /**\nCustom throttle wait in ms for the chat messages and data updates.\nDefault is undefined, which disables throttling.\n */\n experimental_throttle?: number;\n} = {}): UseChatHelpers<MESSAGE_METADATA, DATA_TYPE_SCHEMAS> {\n // Generate ID once, store in state for stability across re-renders\n const [hookId] = useState(generateId);\n\n // Use the caller-supplied ID if available; otherwise, fall back to our stable ID\n const stableChatId = chatId ?? hookId;\n\n // chat store setup\n // TODO enable as arg\n const chatStore = useRef(\n chatStoreArg ??\n defaultChatStore<MESSAGE_METADATA, DATA_TYPE_SCHEMAS>({\n api: '/api/chat',\n generateId,\n }),\n );\n\n // ensure the chat is in the store\n if (!chatStore.current.hasChat(stableChatId)) {\n chatStore.current.addChat(stableChatId, []);\n }\n\n const subscribe = useCallback(\n ({\n onStoreChange,\n eventType,\n }: {\n onStoreChange: () => void;\n eventType: ChatStoreEvent['type'];\n }) => {\n return chatStore.current.subscribe({\n onChatChanged: event => {\n if (event.chatId !== stableChatId || event.type !== eventType) {\n return;\n }\n\n onStoreChange();\n },\n });\n },\n [chatStore, stableChatId],\n );\n\n const addToolResult = useCallback(\n (\n options: Omit<\n Parameters<\n ChatStore<MESSAGE_METADATA, DATA_TYPE_SCHEMAS>['addToolResult']\n >[0],\n 'chatId'\n >,\n ) => chatStore.current.addToolResult({ chatId: stableChatId, ...options }),\n [chatStore, stableChatId],\n );\n\n const stopStream = useCallback(() => {\n chatStore.current.stopStream({ chatId: stableChatId });\n }, [chatStore, stableChatId]);\n\n const error = useSyncExternalStore(\n callback =>\n subscribe({\n onStoreChange: callback,\n eventType: 'chat-status-changed',\n }),\n () => chatStore.current.getError(stableChatId),\n () => chatStore.current.getError(stableChatId),\n );\n\n const status = useSyncExternalStore(\n callback =>\n subscribe({\n onStoreChange: callback,\n eventType: 'chat-status-changed',\n }),\n () => chatStore.current.getStatus(stableChatId),\n () => chatStore.current.getStatus(stableChatId),\n );\n\n const messages = useSyncExternalStore(\n callback => {\n return subscribe({\n onStoreChange: callback,\n eventType: 'chat-messages-changed',\n });\n },\n () => chatStore.current.getMessages(stableChatId),\n () => chatStore.current.getMessages(stableChatId),\n );\n\n const append = useCallback(\n (\n message: CreateUIMessage<\n MESSAGE_METADATA,\n InferUIDataTypes<DATA_TYPE_SCHEMAS>\n >,\n { headers, body }: ChatRequestOptions = {},\n ) =>\n chatStore.current.submitMessage({\n chatId: stableChatId,\n message,\n headers,\n body,\n onError,\n onToolCall,\n onFinish,\n }),\n [chatStore, stableChatId, onError, onToolCall, onFinish],\n );\n\n const reload = useCallback(\n async ({ headers, body }: ChatRequestOptions = {}) =>\n chatStore.current.resubmitLastUserMessage({\n chatId: stableChatId,\n headers,\n body,\n onError,\n onToolCall,\n onFinish,\n }),\n [chatStore, stableChatId, onError, onToolCall, onFinish],\n );\n const stop = useCallback(() => stopStream(), [stopStream]);\n\n const experimental_resume = useCallback(\n async () =>\n chatStore.current.resumeStream({\n chatId: stableChatId,\n onError,\n onToolCall,\n onFinish,\n }),\n [chatStore, stableChatId, onError, onToolCall, onFinish],\n );\n\n const setMessages = useCallback(\n (\n messagesParam:\n | UIMessage<MESSAGE_METADATA, InferUIDataTypes<DATA_TYPE_SCHEMAS>>[]\n | ((\n messages: UIMessage<\n MESSAGE_METADATA,\n InferUIDataTypes<DATA_TYPE_SCHEMAS>\n >[],\n ) => UIMessage<\n MESSAGE_METADATA,\n InferUIDataTypes<DATA_TYPE_SCHEMAS>\n >[]),\n ) => {\n if (typeof messagesParam === 'function') {\n messagesParam = messagesParam(messages);\n }\n\n chatStore.current.setMessages({\n id: stableChatId,\n messages: messagesParam,\n });\n },\n [stableChatId, messages],\n );\n\n // Input state and handlers.\n const [input, setInput] = useState(initialInput);\n\n const handleSubmit = useCallback(\n async (\n event?: { preventDefault?: () => void },\n options: ChatRequestOptions & {\n files?: FileList | FileUIPart[];\n } = {},\n ) => {\n event?.preventDefault?.();\n\n const fileParts = Array.isArray(options?.files)\n ? options.files\n : await convertFileListToFileUIParts(options?.files);\n\n if (!input && fileParts.length === 0) return;\n\n append(\n {\n id: generateId(),\n role: 'user',\n metadata: undefined,\n parts: [...fileParts, { type: 'text', text: input }],\n },\n {\n headers: options.headers,\n body: options.body,\n },\n );\n\n setInput('');\n },\n [input, generateId, append],\n );\n\n const handleInputChange = (e: any) => {\n setInput(e.target.value);\n };\n\n return {\n messages,\n chatId: stableChatId,\n setMessages,\n error,\n append,\n reload,\n stop,\n experimental_resume,\n input,\n setInput,\n handleInputChange,\n handleSubmit,\n status,\n addToolResult,\n };\n}\n","import {\n CompletionRequestOptions,\n UseCompletionOptions,\n callCompletionApi,\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 return input ? complete(input) : undefined;\n },\n [input, complete],\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 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 FetchFunction,\n isAbortError,\n safeValidateTypes,\n} from '@ai-sdk/provider-utils';\nimport {\n asSchema,\n DeepPartial,\n isDeepEqualData,\n parsePartialJson,\n Schema,\n} from 'ai';\nimport { useCallback, useId, useRef, useState } from 'react';\nimport useSWR from 'swr';\nimport z from 'zod';\n\n// use function to allow for mocking in tests:\nconst getOriginalFetch = () => fetch;\n\nexport type Experimental_UseObjectOptions<RESULT> = {\n /**\n * The API endpoint. It should stream JSON that matches the schema as chunked text.\n */\n api: string;\n\n /**\n * A Zod schema that defines the shape of the complete object.\n */\n schema: z.Schema<RESULT, z.ZodTypeDef, any> | Schema<RESULT>;\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 /**\nCustom fetch implementation. You can use it as a middleware to intercept requests,\nor to provide a custom fetch implementation for e.g. testing.\n */\n fetch?: FetchFunction;\n\n /**\nCallback that is called when the stream has finished.\n */\n onFinish?: (event: {\n /**\nThe generated object (typed according to the schema).\nCan be undefined if the final object does not match the schema.\n */\n object: RESULT | undefined;\n\n /**\nOptional 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 */\n headers?: 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\nfunction useObject<RESULT, INPUT = any>({\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<RESULT>): Experimental_UseObjectHelpers<\n RESULT,\n INPUT\n> {\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 [api, completionId],\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 try {\n mutate(undefined); // reset the data\n setIsLoading(true);\n setError(undefined);\n\n const abortController = new AbortController();\n abortControllerRef.current = abortController;\n\n const actualFetch = fetch ?? getOriginalFetch();\n const response = await actualFetch(api, {\n method: 'POST',\n headers: {\n 'Content-Type': 'application/json',\n ...headers,\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 mutate(currentObject);\n }\n },\n\n async close() {\n setIsLoading(false);\n abortControllerRef.current = null;\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 }),\n );\n } catch (error) {\n if (isAbortError(error)) {\n return;\n }\n\n if (onError && error instanceof Error) {\n onError(error);\n }\n\n setIsLoading(false);\n setError(error instanceof Error ? error : new Error(String(error)));\n }\n };\n\n return {\n submit,\n object: data,\n error,\n isLoading,\n stop,\n };\n}\n\nexport const experimental_useObject = useObject;\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;;;ACAA,gBAaO;AACP,mBAAoE;AAmH7D,SAAS,QAGd;AAAA,EACA;AAAA,EACA,eAAe;AAAA,EACf;AAAA,EACA;AAAA,EACA;AAAA,EACA,aAAa,UAAAA;AAAA,EACb,uBAAuB;AAAA,EACvB,WAAW;AACb,IAMI,CAAC,GAAwD;AAE3D,QAAM,CAAC,MAAM,QAAI,uBAAS,UAAU;AAGpC,QAAM,eAAe,0BAAU;AAI/B,QAAM,gBAAY;AAAA,IAChB,0CACE,4BAAsD;AAAA,MACpD,KAAK;AAAA,MACL;AAAA,IACF,CAAC;AAAA,EACL;AAGA,MAAI,CAAC,UAAU,QAAQ,QAAQ,YAAY,GAAG;AAC5C,cAAU,QAAQ,QAAQ,cAAc,CAAC,CAAC;AAAA,EAC5C;AAEA,QAAM,gBAAY;AAAA,IAChB,CAAC;AAAA,MACC;AAAA,MACA;AAAA,IACF,MAGM;AACJ,aAAO,UAAU,QAAQ,UAAU;AAAA,QACjC,eAAe,WAAS;AACtB,cAAI,MAAM,WAAW,gBAAgB,MAAM,SAAS,WAAW;AAC7D;AAAA,UACF;AAEA,wBAAc;AAAA,QAChB;AAAA,MACF,CAAC;AAAA,IACH;AAAA,IACA,CAAC,WAAW,YAAY;AAAA,EAC1B;AAEA,QAAM,oBAAgB;AAAA,IACpB,CACE,YAMG,UAAU,QAAQ,cAAc,EAAE,QAAQ,cAAc,GAAG,QAAQ,CAAC;AAAA,IACzE,CAAC,WAAW,YAAY;AAAA,EAC1B;AAEA,QAAM,iBAAa,0BAAY,MAAM;AACnC,cAAU,QAAQ,WAAW,EAAE,QAAQ,aAAa,CAAC;AAAA,EACvD,GAAG,CAAC,WAAW,YAAY,CAAC;AAE5B,QAAM,YAAQ;AAAA,IACZ,cACE,UAAU;AAAA,MACR,eAAe;AAAA,MACf,WAAW;AAAA,IACb,CAAC;AAAA,IACH,MAAM,UAAU,QAAQ,SAAS,YAAY;AAAA,IAC7C,MAAM,UAAU,QAAQ,SAAS,YAAY;AAAA,EAC/C;AAEA,QAAM,aAAS;AAAA,IACb,cACE,UAAU;AAAA,MACR,eAAe;AAAA,MACf,WAAW;AAAA,IACb,CAAC;AAAA,IACH,MAAM,UAAU,QAAQ,UAAU,YAAY;AAAA,IAC9C,MAAM,UAAU,QAAQ,UAAU,YAAY;AAAA,EAChD;AAEA,QAAM,eAAW;AAAA,IACf,cAAY;AACV,aAAO,UAAU;AAAA,QACf,eAAe;AAAA,QACf,WAAW;AAAA,MACb,CAAC;AAAA,IACH;AAAA,IACA,MAAM,UAAU,QAAQ,YAAY,YAAY;AAAA,IAChD,MAAM,UAAU,QAAQ,YAAY,YAAY;AAAA,EAClD;AAEA,QAAM,aAAS;AAAA,IACb,CACE,SAIA,EAAE,SAAS,KAAK,IAAwB,CAAC,MAEzC,UAAU,QAAQ,cAAc;AAAA,MAC9B,QAAQ;AAAA,MACR;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,IACF,CAAC;AAAA,IACH,CAAC,WAAW,cAAc,SAAS,YAAY,QAAQ;AAAA,EACzD;AAEA,QAAM,aAAS;AAAA,IACb,OAAO,EAAE,SAAS,KAAK,IAAwB,CAAC,MAC9C,UAAU,QAAQ,wBAAwB;AAAA,MACxC,QAAQ;AAAA,MACR;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,IACF,CAAC;AAAA,IACH,CAAC,WAAW,cAAc,SAAS,YAAY,QAAQ;AAAA,EACzD;AACA,QAAM,WAAO,0BAAY,MAAM,WAAW,GAAG,CAAC,UAAU,CAAC;AAEzD,QAAM,0BAAsB;AAAA,IAC1B,YACE,UAAU,QAAQ,aAAa;AAAA,MAC7B,QAAQ;AAAA,MACR;AAAA,MACA;AAAA,MACA;AAAA,IACF,CAAC;AAAA,IACH,CAAC,WAAW,cAAc,SAAS,YAAY,QAAQ;AAAA,EACzD;AAEA,QAAM,kBAAc;AAAA,IAClB,CACE,kBAWG;AACH,UAAI,OAAO,kBAAkB,YAAY;AACvC,wBAAgB,cAAc,QAAQ;AAAA,MACxC;AAEA,gBAAU,QAAQ,YAAY;AAAA,QAC5B,IAAI;AAAA,QACJ,UAAU;AAAA,MACZ,CAAC;AAAA,IACH;AAAA,IACA,CAAC,cAAc,QAAQ;AAAA,EACzB;AAGA,QAAM,CAAC,OAAO,QAAQ,QAAI,uBAAS,YAAY;AAE/C,QAAM,mBAAe;AAAA,IACnB,OACE,OACA,UAEI,CAAC,MACF;AA7TT;AA8TM,2CAAO,mBAAP;AAEA,YAAM,YAAY,MAAM,QAAQ,mCAAS,KAAK,IAC1C,QAAQ,QACR,UAAM,wCAA6B,mCAAS,KAAK;AAErD,UAAI,CAAC,SAAS,UAAU,WAAW;AAAG;AAEtC;AAAA,QACE;AAAA,UACE,IAAI,WAAW;AAAA,UACf,MAAM;AAAA,UACN,UAAU;AAAA,UACV,OAAO,CAAC,GAAG,WAAW,EAAE,MAAM,QAAQ,MAAM,MAAM,CAAC;AAAA,QACrD;AAAA,QACA;AAAA,UACE,SAAS,QAAQ;AAAA,UACjB,MAAM,QAAQ;AAAA,QAChB;AAAA,MACF;AAEA,eAAS,EAAE;AAAA,IACb;AAAA,IACA,CAAC,OAAO,YAAY,MAAM;AAAA,EAC5B;AAEA,QAAM,oBAAoB,CAAC,MAAW;AACpC,aAAS,EAAE,OAAO,KAAK;AAAA,EACzB;AAEA,SAAO;AAAA,IACL;AAAA,IACA,QAAQ;AAAA,IACR;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF;AACF;;;AC5WA,IAAAC,aAIO;AACP,IAAAC,gBAAgE;AAChE,iBAAmB;;;ACNnB,wBAA6B;AAEtB,SAAS,SACd,IACA,QACG;AACH,SAAO,UAAU,WAAO,kBAAAC,SAAiB,IAAI,MAAM,IAAI;AACzD;;;ADwDO,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,aAAO,QAAQ,SAAS,KAAK,IAAI;AAAA,IACnC;AAAA,IACA,CAAC,OAAO,QAAQ;AAAA,EAClB;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;;;AE/MA,4BAIO;AACP,IAAAC,aAMO;AACP,IAAAC,gBAAqD;AACrD,IAAAC,cAAmB;AAInB,IAAM,mBAAmB,MAAM;AA4F/B,SAAS,UAA+B;AAAA,EACtC;AAAA,EACA;AAAA,EACA;AAAA;AAAA,EACA;AAAA,EACA,OAAAC;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AACF,GAGE;AAEA,QAAM,aAAS,qBAAM;AACrB,QAAM,eAAe,kBAAM;AAG3B,QAAM,EAAE,MAAM,OAAO,QAAI,YAAAC;AAAA,IACvB,CAAC,KAAK,YAAY;AAAA,IAClB;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;AA5IjC;AA6II,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;AAtJzC;AAuJI,QAAI;AACF,aAAO,MAAS;AAChB,mBAAa,IAAI;AACjB,eAAS,MAAS;AAElB,YAAM,kBAAkB,IAAI,gBAAgB;AAC5C,yBAAmB,UAAU;AAE7B,YAAM,cAAcD,UAAA,OAAAA,SAAS,iBAAiB;AAC9C,YAAM,WAAW,MAAM,YAAY,KAAK;AAAA,QACtC,QAAQ;AAAA,QACR,SAAS;AAAA,UACP,gBAAgB;AAAA,UAChB,GAAG;AAAA,QACL;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,qBAAO,aAAa;AAAA,YACtB;AAAA,UACF;AAAA,UAEA,MAAM,QAAQ;AACZ,yBAAa,KAAK;AAClB,+BAAmB,UAAU;AAE7B,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;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,mBAAa,KAAK;AAClB,eAASA,kBAAiB,QAAQA,SAAQ,IAAI,MAAM,OAAOA,MAAK,CAAC,CAAC;AAAA,IACpE;AAAA,EACF;AAEA,SAAO;AAAA,IACL;AAAA,IACA,QAAQ;AAAA,IACR;AAAA,IACA;AAAA,IACA;AAAA,EACF;AACF;AAEO,IAAM,yBAAyB;","names":["generateIdFunc","import_ai","import_react","throttleFunction","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-store.ts","../src/throttle.ts","../src/use-completion.ts","../src/use-object.ts"],"sourcesContent":["export * from './use-chat';\nexport * from './use-completion';\nexport * from './use-object';\nexport { createChatStore } from './chat-store';\n","import {\n ChatStore,\n convertFileListToFileUIParts,\n defaultChatStoreOptions,\n generateId as generateIdFunc,\n InferUIDataParts,\n UIDataPartSchemas,\n type ChatRequestOptions,\n type ChatStoreEvent,\n type CreateUIMessage,\n type FileUIPart,\n type UIMessage,\n type UseChatOptions,\n} from 'ai';\nimport { useCallback, useRef, useState, useSyncExternalStore } from 'react';\nimport { createChatStore } from './chat-store';\nimport { throttle } from './throttle';\n\nexport type { CreateUIMessage, UIMessage, UseChatOptions };\n\nexport type UseChatHelpers<\n MESSAGE_METADATA = unknown,\n DATA_PART_SCHEMAS extends UIDataPartSchemas = UIDataPartSchemas,\n> = {\n /**\n * The id of the chat.\n */\n readonly chatId: string;\n\n /**\n * Hook status:\n *\n * - `submitted`: The message has been sent to the API and we're awaiting the start of the response stream.\n * - `streaming`: The response is actively streaming in from the API, receiving chunks of data.\n * - `ready`: The full response has been received and processed; a new user message can be submitted.\n * - `error`: An error occurred during the API request, preventing successful completion.\n */\n readonly status: 'submitted' | 'streaming' | 'ready' | 'error';\n\n /** Current messages in the chat */\n readonly messages: UIMessage<\n MESSAGE_METADATA,\n InferUIDataParts<DATA_PART_SCHEMAS>\n >[];\n\n /** The error object of the API request */\n readonly error: undefined | Error;\n\n /**\n * Append a user message to the chat list. This triggers the API call to fetch\n * the assistant's response.\n *\n * @param message The message to append\n * @param options Additional options to pass to the API call\n */\n append: (\n message: CreateUIMessage<\n MESSAGE_METADATA,\n InferUIDataParts<DATA_PART_SCHEMAS>\n >,\n options?: ChatRequestOptions,\n ) => Promise<void>;\n\n /**\n * Reload the last AI chat response for the given chat history. If the last\n * message isn't from the assistant, it will request the API to generate a\n * new response.\n */\n reload: (\n chatRequestOptions?: ChatRequestOptions,\n ) => Promise<string | null | undefined>;\n\n /**\n * Abort the current request immediately, keep the generated tokens if any.\n */\n stop: () => void;\n\n /**\n * Resume an ongoing chat generation stream. This does not resume an aborted generation.\n */\n experimental_resume: () => void;\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:\n | UIMessage<MESSAGE_METADATA, InferUIDataParts<DATA_PART_SCHEMAS>>[]\n | ((\n messages: UIMessage<\n MESSAGE_METADATA,\n InferUIDataParts<DATA_PART_SCHEMAS>\n >[],\n ) => UIMessage<\n MESSAGE_METADATA,\n InferUIDataParts<DATA_PART_SCHEMAS>\n >[]),\n ) => void;\n\n /** The current value of the input */\n input: string;\n\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 handleInputChange: (\n e:\n | React.ChangeEvent<HTMLInputElement>\n | React.ChangeEvent<HTMLTextAreaElement>,\n ) => void;\n\n /** Form submission handler to automatically reset input and append a user message */\n handleSubmit: (\n event?: { preventDefault?: () => void },\n chatRequestOptions?: ChatRequestOptions & {\n files?: FileList | FileUIPart[];\n },\n ) => void;\n\n addToolResult: ({\n toolCallId,\n result,\n }: {\n toolCallId: string;\n result: any;\n }) => void;\n};\n\nexport function useChat<\n MESSAGE_METADATA = unknown,\n DATA_PART_SCHEMAS extends UIDataPartSchemas = UIDataPartSchemas,\n>({\n chatId,\n initialInput = '',\n onToolCall,\n onFinish,\n onError,\n generateId = generateIdFunc,\n experimental_throttle: throttleWaitMs,\n chatStore: chatStoreArg,\n}: UseChatOptions<MESSAGE_METADATA, DATA_PART_SCHEMAS> & {\n /**\nCustom throttle wait in ms for the chat messages and data updates.\nDefault is undefined, which disables throttling.\n */\n experimental_throttle?: number;\n} = {}): UseChatHelpers<MESSAGE_METADATA, DATA_PART_SCHEMAS> {\n // Generate ID once, store in state for stability across re-renders\n const [hookId] = useState(generateId);\n\n // Use the caller-supplied ID if available; otherwise, fall back to our stable ID\n const stableChatId = chatId ?? hookId;\n\n // chat store setup\n const chatStore = useRef(\n chatStoreArg == null\n ? createChatStore(\n defaultChatStoreOptions<MESSAGE_METADATA, DATA_PART_SCHEMAS>({\n api: '/api/chat',\n generateId,\n })(),\n )\n : typeof chatStoreArg === 'function'\n ? createChatStore(chatStoreArg())\n : chatStoreArg,\n );\n\n // ensure the chat is in the store\n if (!chatStore.current.hasChat(stableChatId)) {\n chatStore.current.addChat(stableChatId, []);\n }\n\n const subscribe = useCallback(\n ({\n onStoreChange,\n eventType,\n }: {\n onStoreChange: () => void;\n eventType: ChatStoreEvent['type'];\n }) => {\n return chatStore.current.subscribe({\n onChatChanged: event => {\n if (event.chatId !== stableChatId || event.type !== eventType) return;\n onStoreChange();\n },\n });\n },\n [chatStore, stableChatId],\n );\n\n const addToolResult = useCallback(\n (\n options: Omit<\n Parameters<\n ChatStore<MESSAGE_METADATA, DATA_PART_SCHEMAS>['addToolResult']\n >[0],\n 'chatId'\n >,\n ) => chatStore.current.addToolResult({ chatId: stableChatId, ...options }),\n [chatStore, stableChatId],\n );\n\n const stopStream = useCallback(() => {\n chatStore.current.stopStream({ chatId: stableChatId });\n }, [chatStore, stableChatId]);\n\n const error = useSyncExternalStore(\n callback =>\n subscribe({\n onStoreChange: callback,\n eventType: 'chat-status-changed',\n }),\n () => chatStore.current.getError(stableChatId),\n () => chatStore.current.getError(stableChatId),\n );\n\n const status = useSyncExternalStore(\n callback =>\n subscribe({\n onStoreChange: callback,\n eventType: 'chat-status-changed',\n }),\n () => chatStore.current.getStatus(stableChatId),\n () => chatStore.current.getStatus(stableChatId),\n );\n\n const subscribeToChatStoreForMessages = useCallback(\n (callback: () => void) => {\n return subscribe({\n onStoreChange: throttleWaitMs\n ? throttle(callback, throttleWaitMs)\n : callback,\n eventType: 'chat-messages-changed',\n });\n },\n [subscribe, throttleWaitMs],\n );\n\n const messages = useSyncExternalStore(\n callback => subscribeToChatStoreForMessages(callback),\n () => chatStore.current.getMessages(stableChatId),\n () => chatStore.current.getMessages(stableChatId),\n );\n\n const append = useCallback(\n (\n message: CreateUIMessage<\n MESSAGE_METADATA,\n InferUIDataParts<DATA_PART_SCHEMAS>\n >,\n { headers, body }: ChatRequestOptions = {},\n ) =>\n chatStore.current.submitMessage({\n chatId: stableChatId,\n message,\n headers,\n body,\n onError,\n onToolCall,\n onFinish,\n }),\n [chatStore, stableChatId, onError, onToolCall, onFinish],\n );\n\n const reload = useCallback(\n async ({ headers, body }: ChatRequestOptions = {}) =>\n chatStore.current.resubmitLastUserMessage({\n chatId: stableChatId,\n headers,\n body,\n onError,\n onToolCall,\n onFinish,\n }),\n [chatStore, stableChatId, onError, onToolCall, onFinish],\n );\n const stop = useCallback(() => stopStream(), [stopStream]);\n\n const experimental_resume = useCallback(\n async () =>\n chatStore.current.resumeStream({\n chatId: stableChatId,\n onError,\n onToolCall,\n onFinish,\n }),\n [chatStore, stableChatId, onError, onToolCall, onFinish],\n );\n\n const setMessages = useCallback(\n (\n messagesParam:\n | UIMessage<MESSAGE_METADATA, InferUIDataParts<DATA_PART_SCHEMAS>>[]\n | ((\n messages: UIMessage<\n MESSAGE_METADATA,\n InferUIDataParts<DATA_PART_SCHEMAS>\n >[],\n ) => UIMessage<\n MESSAGE_METADATA,\n InferUIDataParts<DATA_PART_SCHEMAS>\n >[]),\n ) => {\n if (typeof messagesParam === 'function') {\n messagesParam = messagesParam(messages);\n }\n\n chatStore.current.setMessages({\n id: stableChatId,\n messages: messagesParam,\n });\n },\n [stableChatId, messages],\n );\n\n // Input state and handlers.\n const [input, setInput] = useState(initialInput);\n\n const handleSubmit = useCallback(\n async (\n event?: { preventDefault?: () => void },\n options: ChatRequestOptions & {\n files?: FileList | FileUIPart[];\n } = {},\n ) => {\n event?.preventDefault?.();\n\n const fileParts = Array.isArray(options?.files)\n ? options.files\n : await convertFileListToFileUIParts(options?.files);\n\n if (!input && fileParts.length === 0) return;\n\n append(\n {\n id: generateId(),\n role: 'user',\n metadata: undefined,\n parts: [...fileParts, { type: 'text', text: input }],\n },\n {\n headers: options.headers,\n body: options.body,\n },\n );\n\n setInput('');\n },\n [input, generateId, append],\n );\n\n const handleInputChange = (e: any) => {\n setInput(e.target.value);\n };\n\n return {\n messages,\n chatId: stableChatId,\n setMessages,\n error,\n append,\n reload,\n stop,\n experimental_resume,\n input,\n setInput,\n handleInputChange,\n handleSubmit,\n status,\n addToolResult,\n };\n}\n","import {\n ChatStatus,\n ChatStore,\n ChatStoreOptions,\n InferUIDataParts,\n SerialJobExecutor,\n UIDataPartSchemas,\n UIDataTypes,\n UIMessage,\n type ActiveResponse,\n type Chat,\n} from 'ai';\n\nclass ReactChat<MESSAGE_METADATA, DATA_TYPES extends UIDataTypes>\n implements Chat<MESSAGE_METADATA, DATA_TYPES>\n{\n messages: UIMessage<MESSAGE_METADATA, DATA_TYPES>[];\n status: ChatStatus = 'ready';\n error: Error | undefined = undefined;\n activeResponse: ActiveResponse<MESSAGE_METADATA> | undefined = undefined;\n jobExecutor = new SerialJobExecutor();\n\n constructor(messages?: UIMessage<MESSAGE_METADATA, DATA_TYPES>[]) {\n this.messages = messages ?? [];\n }\n\n setStatus = (status: ChatStatus) => {\n this.status = status;\n };\n\n setError = (error: Error | undefined) => {\n this.error = error;\n };\n\n setActiveResponse = (\n activeResponse: ActiveResponse<MESSAGE_METADATA> | undefined,\n ) => {\n this.activeResponse = activeResponse;\n };\n\n setMessages = (messages: UIMessage<MESSAGE_METADATA, DATA_TYPES>[]) => {\n this.messages = [...messages];\n };\n\n pushMessage = (message: UIMessage<MESSAGE_METADATA, DATA_TYPES>) => {\n this.messages = this.messages.concat(message);\n };\n\n popMessage = () => {\n this.messages = this.messages.slice(0, -1);\n };\n\n replaceMessage = (\n index: number,\n message: UIMessage<MESSAGE_METADATA, DATA_TYPES>,\n ) => {\n this.messages = [\n ...this.messages.slice(0, index),\n message,\n ...this.messages.slice(index + 1),\n ];\n };\n\n snapshot = <T>(value: T): T => {\n return structuredClone(value);\n };\n}\n\nexport function createChatStore<\n MESSAGE_METADATA = unknown,\n DATA_PART_SCHEMAS extends UIDataPartSchemas = UIDataPartSchemas,\n>(\n options: ChatStoreOptions<MESSAGE_METADATA, DATA_PART_SCHEMAS>,\n): ChatStore<MESSAGE_METADATA, DATA_PART_SCHEMAS> {\n return new ChatStore<MESSAGE_METADATA, DATA_PART_SCHEMAS>({\n ...options,\n createChat: options =>\n new ReactChat<MESSAGE_METADATA, InferUIDataParts<DATA_PART_SCHEMAS>>(\n options.messages,\n ),\n });\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 CompletionRequestOptions,\n UseCompletionOptions,\n callCompletionApi,\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 return input ? complete(input) : undefined;\n },\n [input, complete],\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 FetchFunction,\n isAbortError,\n safeValidateTypes,\n} from '@ai-sdk/provider-utils';\nimport {\n asSchema,\n DeepPartial,\n isDeepEqualData,\n parsePartialJson,\n Schema,\n} from 'ai';\nimport { useCallback, useId, useRef, useState } from 'react';\nimport useSWR from 'swr';\nimport z from 'zod';\n\n// use function to allow for mocking in tests:\nconst getOriginalFetch = () => fetch;\n\nexport type Experimental_UseObjectOptions<RESULT> = {\n /**\n * The API endpoint. It should stream JSON that matches the schema as chunked text.\n */\n api: string;\n\n /**\n * A Zod schema that defines the shape of the complete object.\n */\n schema: z.Schema<RESULT, z.ZodTypeDef, any> | Schema<RESULT>;\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 /**\nCustom fetch implementation. You can use it as a middleware to intercept requests,\nor to provide a custom fetch implementation for e.g. testing.\n */\n fetch?: FetchFunction;\n\n /**\nCallback that is called when the stream has finished.\n */\n onFinish?: (event: {\n /**\nThe generated object (typed according to the schema).\nCan be undefined if the final object does not match the schema.\n */\n object: RESULT | undefined;\n\n /**\nOptional 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 */\n headers?: 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\nfunction useObject<RESULT, INPUT = any>({\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<RESULT>): Experimental_UseObjectHelpers<\n RESULT,\n INPUT\n> {\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 [api, completionId],\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 try {\n mutate(undefined); // reset the data\n setIsLoading(true);\n setError(undefined);\n\n const abortController = new AbortController();\n abortControllerRef.current = abortController;\n\n const actualFetch = fetch ?? getOriginalFetch();\n const response = await actualFetch(api, {\n method: 'POST',\n headers: {\n 'Content-Type': 'application/json',\n ...headers,\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 mutate(currentObject);\n }\n },\n\n async close() {\n setIsLoading(false);\n abortControllerRef.current = null;\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 }),\n );\n } catch (error) {\n if (isAbortError(error)) {\n return;\n }\n\n if (onError && error instanceof Error) {\n onError(error);\n }\n\n setIsLoading(false);\n setError(error instanceof Error ? error : new Error(String(error)));\n }\n };\n\n return {\n submit,\n object: data,\n error,\n isLoading,\n stop,\n };\n}\n\nexport const experimental_useObject = useObject;\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;;;ACAA,IAAAA,aAaO;AACP,mBAAoE;;;ACdpE,gBAWO;AAEP,IAAM,YAAN,MAEA;AAAA,EAOE,YAAY,UAAsD;AALlE,kBAAqB;AACrB,iBAA2B;AAC3B,0BAA+D;AAC/D,uBAAc,IAAI,4BAAkB;AAMpC,qBAAY,CAAC,WAAuB;AAClC,WAAK,SAAS;AAAA,IAChB;AAEA,oBAAW,CAAC,UAA6B;AACvC,WAAK,QAAQ;AAAA,IACf;AAEA,6BAAoB,CAClB,mBACG;AACH,WAAK,iBAAiB;AAAA,IACxB;AAEA,uBAAc,CAAC,aAAwD;AACrE,WAAK,WAAW,CAAC,GAAG,QAAQ;AAAA,IAC9B;AAEA,uBAAc,CAAC,YAAqD;AAClE,WAAK,WAAW,KAAK,SAAS,OAAO,OAAO;AAAA,IAC9C;AAEA,sBAAa,MAAM;AACjB,WAAK,WAAW,KAAK,SAAS,MAAM,GAAG,EAAE;AAAA,IAC3C;AAEA,0BAAiB,CACf,OACA,YACG;AACH,WAAK,WAAW;AAAA,QACd,GAAG,KAAK,SAAS,MAAM,GAAG,KAAK;AAAA,QAC/B;AAAA,QACA,GAAG,KAAK,SAAS,MAAM,QAAQ,CAAC;AAAA,MAClC;AAAA,IACF;AAEA,oBAAW,CAAI,UAAgB;AAC7B,aAAO,gBAAgB,KAAK;AAAA,IAC9B;AA1CE,SAAK,WAAW,8BAAY,CAAC;AAAA,EAC/B;AA0CF;AAEO,SAAS,gBAId,SACgD;AAChD,SAAO,IAAI,oBAA+C;AAAA,IACxD,GAAG;AAAA,IACH,YAAY,CAAAC,aACV,IAAI;AAAA,MACFA,SAAQ;AAAA,IACV;AAAA,EACJ,CAAC;AACH;;;ACjFA,wBAA6B;AAEtB,SAAS,SACd,IACA,QACG;AACH,SAAO,UAAU,WAAO,kBAAAC,SAAiB,IAAI,MAAM,IAAI;AACzD;;;AF4HO,SAAS,QAGd;AAAA,EACA;AAAA,EACA,eAAe;AAAA,EACf;AAAA,EACA;AAAA,EACA;AAAA,EACA,aAAa,WAAAC;AAAA,EACb,uBAAuB;AAAA,EACvB,WAAW;AACb,IAMI,CAAC,GAAwD;AAE3D,QAAM,CAAC,MAAM,QAAI,uBAAS,UAAU;AAGpC,QAAM,eAAe,0BAAU;AAG/B,QAAM,gBAAY;AAAA,IAChB,gBAAgB,OACZ;AAAA,UACE,oCAA6D;AAAA,QAC3D,KAAK;AAAA,QACL;AAAA,MACF,CAAC,EAAE;AAAA,IACL,IACA,OAAO,iBAAiB,aACtB,gBAAgB,aAAa,CAAC,IAC9B;AAAA,EACR;AAGA,MAAI,CAAC,UAAU,QAAQ,QAAQ,YAAY,GAAG;AAC5C,cAAU,QAAQ,QAAQ,cAAc,CAAC,CAAC;AAAA,EAC5C;AAEA,QAAM,gBAAY;AAAA,IAChB,CAAC;AAAA,MACC;AAAA,MACA;AAAA,IACF,MAGM;AACJ,aAAO,UAAU,QAAQ,UAAU;AAAA,QACjC,eAAe,WAAS;AACtB,cAAI,MAAM,WAAW,gBAAgB,MAAM,SAAS;AAAW;AAC/D,wBAAc;AAAA,QAChB;AAAA,MACF,CAAC;AAAA,IACH;AAAA,IACA,CAAC,WAAW,YAAY;AAAA,EAC1B;AAEA,QAAM,oBAAgB;AAAA,IACpB,CACE,YAMG,UAAU,QAAQ,cAAc,EAAE,QAAQ,cAAc,GAAG,QAAQ,CAAC;AAAA,IACzE,CAAC,WAAW,YAAY;AAAA,EAC1B;AAEA,QAAM,iBAAa,0BAAY,MAAM;AACnC,cAAU,QAAQ,WAAW,EAAE,QAAQ,aAAa,CAAC;AAAA,EACvD,GAAG,CAAC,WAAW,YAAY,CAAC;AAE5B,QAAM,YAAQ;AAAA,IACZ,cACE,UAAU;AAAA,MACR,eAAe;AAAA,MACf,WAAW;AAAA,IACb,CAAC;AAAA,IACH,MAAM,UAAU,QAAQ,SAAS,YAAY;AAAA,IAC7C,MAAM,UAAU,QAAQ,SAAS,YAAY;AAAA,EAC/C;AAEA,QAAM,aAAS;AAAA,IACb,cACE,UAAU;AAAA,MACR,eAAe;AAAA,MACf,WAAW;AAAA,IACb,CAAC;AAAA,IACH,MAAM,UAAU,QAAQ,UAAU,YAAY;AAAA,IAC9C,MAAM,UAAU,QAAQ,UAAU,YAAY;AAAA,EAChD;AAEA,QAAM,sCAAkC;AAAA,IACtC,CAAC,aAAyB;AACxB,aAAO,UAAU;AAAA,QACf,eAAe,iBACX,SAAS,UAAU,cAAc,IACjC;AAAA,QACJ,WAAW;AAAA,MACb,CAAC;AAAA,IACH;AAAA,IACA,CAAC,WAAW,cAAc;AAAA,EAC5B;AAEA,QAAM,eAAW;AAAA,IACf,cAAY,gCAAgC,QAAQ;AAAA,IACpD,MAAM,UAAU,QAAQ,YAAY,YAAY;AAAA,IAChD,MAAM,UAAU,QAAQ,YAAY,YAAY;AAAA,EAClD;AAEA,QAAM,aAAS;AAAA,IACb,CACE,SAIA,EAAE,SAAS,KAAK,IAAwB,CAAC,MAEzC,UAAU,QAAQ,cAAc;AAAA,MAC9B,QAAQ;AAAA,MACR;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,IACF,CAAC;AAAA,IACH,CAAC,WAAW,cAAc,SAAS,YAAY,QAAQ;AAAA,EACzD;AAEA,QAAM,aAAS;AAAA,IACb,OAAO,EAAE,SAAS,KAAK,IAAwB,CAAC,MAC9C,UAAU,QAAQ,wBAAwB;AAAA,MACxC,QAAQ;AAAA,MACR;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,IACF,CAAC;AAAA,IACH,CAAC,WAAW,cAAc,SAAS,YAAY,QAAQ;AAAA,EACzD;AACA,QAAM,WAAO,0BAAY,MAAM,WAAW,GAAG,CAAC,UAAU,CAAC;AAEzD,QAAM,0BAAsB;AAAA,IAC1B,YACE,UAAU,QAAQ,aAAa;AAAA,MAC7B,QAAQ;AAAA,MACR;AAAA,MACA;AAAA,MACA;AAAA,IACF,CAAC;AAAA,IACH,CAAC,WAAW,cAAc,SAAS,YAAY,QAAQ;AAAA,EACzD;AAEA,QAAM,kBAAc;AAAA,IAClB,CACE,kBAWG;AACH,UAAI,OAAO,kBAAkB,YAAY;AACvC,wBAAgB,cAAc,QAAQ;AAAA,MACxC;AAEA,gBAAU,QAAQ,YAAY;AAAA,QAC5B,IAAI;AAAA,QACJ,UAAU;AAAA,MACZ,CAAC;AAAA,IACH;AAAA,IACA,CAAC,cAAc,QAAQ;AAAA,EACzB;AAGA,QAAM,CAAC,OAAO,QAAQ,QAAI,uBAAS,YAAY;AAE/C,QAAM,mBAAe;AAAA,IACnB,OACE,OACA,UAEI,CAAC,MACF;AAvUT;AAwUM,2CAAO,mBAAP;AAEA,YAAM,YAAY,MAAM,QAAQ,mCAAS,KAAK,IAC1C,QAAQ,QACR,UAAM,yCAA6B,mCAAS,KAAK;AAErD,UAAI,CAAC,SAAS,UAAU,WAAW;AAAG;AAEtC;AAAA,QACE;AAAA,UACE,IAAI,WAAW;AAAA,UACf,MAAM;AAAA,UACN,UAAU;AAAA,UACV,OAAO,CAAC,GAAG,WAAW,EAAE,MAAM,QAAQ,MAAM,MAAM,CAAC;AAAA,QACrD;AAAA,QACA;AAAA,UACE,SAAS,QAAQ;AAAA,UACjB,MAAM,QAAQ;AAAA,QAChB;AAAA,MACF;AAEA,eAAS,EAAE;AAAA,IACb;AAAA,IACA,CAAC,OAAO,YAAY,MAAM;AAAA,EAC5B;AAEA,QAAM,oBAAoB,CAAC,MAAW;AACpC,aAAS,EAAE,OAAO,KAAK;AAAA,EACzB;AAEA,SAAO;AAAA,IACL;AAAA,IACA,QAAQ;AAAA,IACR;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF;AACF;;;AGtXA,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,aAAO,QAAQ,SAAS,KAAK,IAAI;AAAA,IACnC;AAAA,IACA,CAAC,OAAO,QAAQ;AAAA,EAClB;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;;;AC/MA,4BAIO;AACP,IAAAC,aAMO;AACP,IAAAC,gBAAqD;AACrD,IAAAC,cAAmB;AAInB,IAAM,mBAAmB,MAAM;AA4F/B,SAAS,UAA+B;AAAA,EACtC;AAAA,EACA;AAAA,EACA;AAAA;AAAA,EACA;AAAA,EACA,OAAAC;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AACF,GAGE;AAEA,QAAM,aAAS,qBAAM;AACrB,QAAM,eAAe,kBAAM;AAG3B,QAAM,EAAE,MAAM,OAAO,QAAI,YAAAC;AAAA,IACvB,CAAC,KAAK,YAAY;AAAA,IAClB;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;AA5IjC;AA6II,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;AAtJzC;AAuJI,QAAI;AACF,aAAO,MAAS;AAChB,mBAAa,IAAI;AACjB,eAAS,MAAS;AAElB,YAAM,kBAAkB,IAAI,gBAAgB;AAC5C,yBAAmB,UAAU;AAE7B,YAAM,cAAcD,UAAA,OAAAA,SAAS,iBAAiB;AAC9C,YAAM,WAAW,MAAM,YAAY,KAAK;AAAA,QACtC,QAAQ;AAAA,QACR,SAAS;AAAA,UACP,gBAAgB;AAAA,UAChB,GAAG;AAAA,QACL;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,qBAAO,aAAa;AAAA,YACtB;AAAA,UACF;AAAA,UAEA,MAAM,QAAQ;AACZ,yBAAa,KAAK;AAClB,+BAAmB,UAAU;AAE7B,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;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,mBAAa,KAAK;AAClB,eAASA,kBAAiB,QAAQA,SAAQ,IAAI,MAAM,OAAOA,MAAK,CAAC,CAAC;AAAA,IACpE;AAAA,EACF;AAEA,SAAO;AAAA,IACL;AAAA,IACA,QAAQ;AAAA,IACR;AAAA,IACA;AAAA,IACA;AAAA,EACF;AACF;AAEO,IAAM,yBAAyB;","names":["import_ai","options","throttleFunction","generateIdFunc","import_ai","import_react","fetch","useSWR","completion","import_ai","import_react","import_swr","fetch","useSWR","error"]}
|
package/dist/index.mjs
CHANGED
|
@@ -1,10 +1,69 @@
|
|
|
1
1
|
// src/use-chat.ts
|
|
2
2
|
import {
|
|
3
3
|
convertFileListToFileUIParts,
|
|
4
|
-
|
|
4
|
+
defaultChatStoreOptions,
|
|
5
5
|
generateId as generateIdFunc
|
|
6
6
|
} from "ai";
|
|
7
7
|
import { useCallback, useRef, useState, useSyncExternalStore } from "react";
|
|
8
|
+
|
|
9
|
+
// src/chat-store.ts
|
|
10
|
+
import {
|
|
11
|
+
ChatStore,
|
|
12
|
+
SerialJobExecutor
|
|
13
|
+
} from "ai";
|
|
14
|
+
var ReactChat = class {
|
|
15
|
+
constructor(messages) {
|
|
16
|
+
this.status = "ready";
|
|
17
|
+
this.error = void 0;
|
|
18
|
+
this.activeResponse = void 0;
|
|
19
|
+
this.jobExecutor = new SerialJobExecutor();
|
|
20
|
+
this.setStatus = (status) => {
|
|
21
|
+
this.status = status;
|
|
22
|
+
};
|
|
23
|
+
this.setError = (error) => {
|
|
24
|
+
this.error = error;
|
|
25
|
+
};
|
|
26
|
+
this.setActiveResponse = (activeResponse) => {
|
|
27
|
+
this.activeResponse = activeResponse;
|
|
28
|
+
};
|
|
29
|
+
this.setMessages = (messages) => {
|
|
30
|
+
this.messages = [...messages];
|
|
31
|
+
};
|
|
32
|
+
this.pushMessage = (message) => {
|
|
33
|
+
this.messages = this.messages.concat(message);
|
|
34
|
+
};
|
|
35
|
+
this.popMessage = () => {
|
|
36
|
+
this.messages = this.messages.slice(0, -1);
|
|
37
|
+
};
|
|
38
|
+
this.replaceMessage = (index, message) => {
|
|
39
|
+
this.messages = [
|
|
40
|
+
...this.messages.slice(0, index),
|
|
41
|
+
message,
|
|
42
|
+
...this.messages.slice(index + 1)
|
|
43
|
+
];
|
|
44
|
+
};
|
|
45
|
+
this.snapshot = (value) => {
|
|
46
|
+
return structuredClone(value);
|
|
47
|
+
};
|
|
48
|
+
this.messages = messages != null ? messages : [];
|
|
49
|
+
}
|
|
50
|
+
};
|
|
51
|
+
function createChatStore(options) {
|
|
52
|
+
return new ChatStore({
|
|
53
|
+
...options,
|
|
54
|
+
createChat: (options2) => new ReactChat(
|
|
55
|
+
options2.messages
|
|
56
|
+
)
|
|
57
|
+
});
|
|
58
|
+
}
|
|
59
|
+
|
|
60
|
+
// src/throttle.ts
|
|
61
|
+
import throttleFunction from "throttleit";
|
|
62
|
+
function throttle(fn, waitMs) {
|
|
63
|
+
return waitMs != null ? throttleFunction(fn, waitMs) : fn;
|
|
64
|
+
}
|
|
65
|
+
|
|
66
|
+
// src/use-chat.ts
|
|
8
67
|
function useChat({
|
|
9
68
|
chatId,
|
|
10
69
|
initialInput = "",
|
|
@@ -18,10 +77,12 @@ function useChat({
|
|
|
18
77
|
const [hookId] = useState(generateId);
|
|
19
78
|
const stableChatId = chatId != null ? chatId : hookId;
|
|
20
79
|
const chatStore = useRef(
|
|
21
|
-
chatStoreArg
|
|
22
|
-
|
|
23
|
-
|
|
24
|
-
|
|
80
|
+
chatStoreArg == null ? createChatStore(
|
|
81
|
+
defaultChatStoreOptions({
|
|
82
|
+
api: "/api/chat",
|
|
83
|
+
generateId
|
|
84
|
+
})()
|
|
85
|
+
) : typeof chatStoreArg === "function" ? createChatStore(chatStoreArg()) : chatStoreArg
|
|
25
86
|
);
|
|
26
87
|
if (!chatStore.current.hasChat(stableChatId)) {
|
|
27
88
|
chatStore.current.addChat(stableChatId, []);
|
|
@@ -33,9 +94,8 @@ function useChat({
|
|
|
33
94
|
}) => {
|
|
34
95
|
return chatStore.current.subscribe({
|
|
35
96
|
onChatChanged: (event) => {
|
|
36
|
-
if (event.chatId !== stableChatId || event.type !== eventType)
|
|
97
|
+
if (event.chatId !== stableChatId || event.type !== eventType)
|
|
37
98
|
return;
|
|
38
|
-
}
|
|
39
99
|
onStoreChange();
|
|
40
100
|
}
|
|
41
101
|
});
|
|
@@ -65,13 +125,17 @@ function useChat({
|
|
|
65
125
|
() => chatStore.current.getStatus(stableChatId),
|
|
66
126
|
() => chatStore.current.getStatus(stableChatId)
|
|
67
127
|
);
|
|
68
|
-
const
|
|
128
|
+
const subscribeToChatStoreForMessages = useCallback(
|
|
69
129
|
(callback) => {
|
|
70
130
|
return subscribe({
|
|
71
|
-
onStoreChange: callback,
|
|
131
|
+
onStoreChange: throttleWaitMs ? throttle(callback, throttleWaitMs) : callback,
|
|
72
132
|
eventType: "chat-messages-changed"
|
|
73
133
|
});
|
|
74
134
|
},
|
|
135
|
+
[subscribe, throttleWaitMs]
|
|
136
|
+
);
|
|
137
|
+
const messages = useSyncExternalStore(
|
|
138
|
+
(callback) => subscribeToChatStoreForMessages(callback),
|
|
75
139
|
() => chatStore.current.getMessages(stableChatId),
|
|
76
140
|
() => chatStore.current.getMessages(stableChatId)
|
|
77
141
|
);
|
|
@@ -171,14 +235,6 @@ import {
|
|
|
171
235
|
} from "ai";
|
|
172
236
|
import { useCallback as useCallback2, useEffect, useId, useRef as useRef2, useState as useState2 } from "react";
|
|
173
237
|
import useSWR from "swr";
|
|
174
|
-
|
|
175
|
-
// src/throttle.ts
|
|
176
|
-
import throttleFunction from "throttleit";
|
|
177
|
-
function throttle(fn, waitMs) {
|
|
178
|
-
return waitMs != null ? throttleFunction(fn, waitMs) : fn;
|
|
179
|
-
}
|
|
180
|
-
|
|
181
|
-
// src/use-completion.ts
|
|
182
238
|
function useCompletion({
|
|
183
239
|
api = "/api/completion",
|
|
184
240
|
id,
|
|
@@ -422,6 +478,7 @@ function useObject({
|
|
|
422
478
|
}
|
|
423
479
|
var experimental_useObject = useObject;
|
|
424
480
|
export {
|
|
481
|
+
createChatStore,
|
|
425
482
|
experimental_useObject,
|
|
426
483
|
useChat,
|
|
427
484
|
useCompletion
|
package/dist/index.mjs.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"sources":["../src/use-chat.ts","../src/use-completion.ts","../src/throttle.ts","../src/use-object.ts"],"sourcesContent":["import {\n ChatStore,\n convertFileListToFileUIParts,\n defaultChatStore,\n generateId as generateIdFunc,\n InferUIDataTypes,\n UIDataTypesSchemas,\n type ChatRequestOptions,\n type ChatStoreEvent,\n type CreateUIMessage,\n type FileUIPart,\n type UIMessage,\n type UseChatOptions,\n} from 'ai';\nimport { useCallback, useRef, useState, useSyncExternalStore } from 'react';\n\nexport type { CreateUIMessage, UIMessage, UseChatOptions };\n\nexport type UseChatHelpers<\n MESSAGE_METADATA = unknown,\n DATA_TYPE_SCHEMAS extends UIDataTypesSchemas = UIDataTypesSchemas,\n> = {\n /**\n * The id of the chat.\n */\n readonly chatId: string;\n\n /**\n * Hook status:\n *\n * - `submitted`: The message has been sent to the API and we're awaiting the start of the response stream.\n * - `streaming`: The response is actively streaming in from the API, receiving chunks of data.\n * - `ready`: The full response has been received and processed; a new user message can be submitted.\n * - `error`: An error occurred during the API request, preventing successful completion.\n */\n readonly status: 'submitted' | 'streaming' | 'ready' | 'error';\n\n /** Current messages in the chat */\n readonly messages: UIMessage<\n MESSAGE_METADATA,\n InferUIDataTypes<DATA_TYPE_SCHEMAS>\n >[];\n\n /** The error object of the API request */\n readonly error: undefined | Error;\n\n /**\n * Append a user message to the chat list. This triggers the API call to fetch\n * the assistant's response.\n *\n * @param message The message to append\n * @param options Additional options to pass to the API call\n */\n append: (\n message: CreateUIMessage<\n MESSAGE_METADATA,\n InferUIDataTypes<DATA_TYPE_SCHEMAS>\n >,\n options?: ChatRequestOptions,\n ) => Promise<void>;\n\n /**\n * Reload the last AI chat response for the given chat history. If the last\n * message isn't from the assistant, it will request the API to generate a\n * new response.\n */\n reload: (\n chatRequestOptions?: ChatRequestOptions,\n ) => Promise<string | null | undefined>;\n\n /**\n * Abort the current request immediately, keep the generated tokens if any.\n */\n stop: () => void;\n\n /**\n * Resume an ongoing chat generation stream. This does not resume an aborted generation.\n */\n experimental_resume: () => void;\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:\n | UIMessage<MESSAGE_METADATA, InferUIDataTypes<DATA_TYPE_SCHEMAS>>[]\n | ((\n messages: UIMessage<\n MESSAGE_METADATA,\n InferUIDataTypes<DATA_TYPE_SCHEMAS>\n >[],\n ) => UIMessage<\n MESSAGE_METADATA,\n InferUIDataTypes<DATA_TYPE_SCHEMAS>\n >[]),\n ) => void;\n\n /** The current value of the input */\n input: string;\n\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 handleInputChange: (\n e:\n | React.ChangeEvent<HTMLInputElement>\n | React.ChangeEvent<HTMLTextAreaElement>,\n ) => void;\n\n /** Form submission handler to automatically reset input and append a user message */\n handleSubmit: (\n event?: { preventDefault?: () => void },\n chatRequestOptions?: ChatRequestOptions & {\n files?: FileList | FileUIPart[];\n },\n ) => void;\n\n addToolResult: ({\n toolCallId,\n result,\n }: {\n toolCallId: string;\n result: any;\n }) => void;\n};\n\nexport function useChat<\n MESSAGE_METADATA = unknown,\n DATA_TYPE_SCHEMAS extends UIDataTypesSchemas = UIDataTypesSchemas,\n>({\n chatId,\n initialInput = '',\n onToolCall,\n onFinish,\n onError,\n generateId = generateIdFunc,\n experimental_throttle: throttleWaitMs,\n chatStore: chatStoreArg,\n}: UseChatOptions<MESSAGE_METADATA, DATA_TYPE_SCHEMAS> & {\n /**\nCustom throttle wait in ms for the chat messages and data updates.\nDefault is undefined, which disables throttling.\n */\n experimental_throttle?: number;\n} = {}): UseChatHelpers<MESSAGE_METADATA, DATA_TYPE_SCHEMAS> {\n // Generate ID once, store in state for stability across re-renders\n const [hookId] = useState(generateId);\n\n // Use the caller-supplied ID if available; otherwise, fall back to our stable ID\n const stableChatId = chatId ?? hookId;\n\n // chat store setup\n // TODO enable as arg\n const chatStore = useRef(\n chatStoreArg ??\n defaultChatStore<MESSAGE_METADATA, DATA_TYPE_SCHEMAS>({\n api: '/api/chat',\n generateId,\n }),\n );\n\n // ensure the chat is in the store\n if (!chatStore.current.hasChat(stableChatId)) {\n chatStore.current.addChat(stableChatId, []);\n }\n\n const subscribe = useCallback(\n ({\n onStoreChange,\n eventType,\n }: {\n onStoreChange: () => void;\n eventType: ChatStoreEvent['type'];\n }) => {\n return chatStore.current.subscribe({\n onChatChanged: event => {\n if (event.chatId !== stableChatId || event.type !== eventType) {\n return;\n }\n\n onStoreChange();\n },\n });\n },\n [chatStore, stableChatId],\n );\n\n const addToolResult = useCallback(\n (\n options: Omit<\n Parameters<\n ChatStore<MESSAGE_METADATA, DATA_TYPE_SCHEMAS>['addToolResult']\n >[0],\n 'chatId'\n >,\n ) => chatStore.current.addToolResult({ chatId: stableChatId, ...options }),\n [chatStore, stableChatId],\n );\n\n const stopStream = useCallback(() => {\n chatStore.current.stopStream({ chatId: stableChatId });\n }, [chatStore, stableChatId]);\n\n const error = useSyncExternalStore(\n callback =>\n subscribe({\n onStoreChange: callback,\n eventType: 'chat-status-changed',\n }),\n () => chatStore.current.getError(stableChatId),\n () => chatStore.current.getError(stableChatId),\n );\n\n const status = useSyncExternalStore(\n callback =>\n subscribe({\n onStoreChange: callback,\n eventType: 'chat-status-changed',\n }),\n () => chatStore.current.getStatus(stableChatId),\n () => chatStore.current.getStatus(stableChatId),\n );\n\n const messages = useSyncExternalStore(\n callback => {\n return subscribe({\n onStoreChange: callback,\n eventType: 'chat-messages-changed',\n });\n },\n () => chatStore.current.getMessages(stableChatId),\n () => chatStore.current.getMessages(stableChatId),\n );\n\n const append = useCallback(\n (\n message: CreateUIMessage<\n MESSAGE_METADATA,\n InferUIDataTypes<DATA_TYPE_SCHEMAS>\n >,\n { headers, body }: ChatRequestOptions = {},\n ) =>\n chatStore.current.submitMessage({\n chatId: stableChatId,\n message,\n headers,\n body,\n onError,\n onToolCall,\n onFinish,\n }),\n [chatStore, stableChatId, onError, onToolCall, onFinish],\n );\n\n const reload = useCallback(\n async ({ headers, body }: ChatRequestOptions = {}) =>\n chatStore.current.resubmitLastUserMessage({\n chatId: stableChatId,\n headers,\n body,\n onError,\n onToolCall,\n onFinish,\n }),\n [chatStore, stableChatId, onError, onToolCall, onFinish],\n );\n const stop = useCallback(() => stopStream(), [stopStream]);\n\n const experimental_resume = useCallback(\n async () =>\n chatStore.current.resumeStream({\n chatId: stableChatId,\n onError,\n onToolCall,\n onFinish,\n }),\n [chatStore, stableChatId, onError, onToolCall, onFinish],\n );\n\n const setMessages = useCallback(\n (\n messagesParam:\n | UIMessage<MESSAGE_METADATA, InferUIDataTypes<DATA_TYPE_SCHEMAS>>[]\n | ((\n messages: UIMessage<\n MESSAGE_METADATA,\n InferUIDataTypes<DATA_TYPE_SCHEMAS>\n >[],\n ) => UIMessage<\n MESSAGE_METADATA,\n InferUIDataTypes<DATA_TYPE_SCHEMAS>\n >[]),\n ) => {\n if (typeof messagesParam === 'function') {\n messagesParam = messagesParam(messages);\n }\n\n chatStore.current.setMessages({\n id: stableChatId,\n messages: messagesParam,\n });\n },\n [stableChatId, messages],\n );\n\n // Input state and handlers.\n const [input, setInput] = useState(initialInput);\n\n const handleSubmit = useCallback(\n async (\n event?: { preventDefault?: () => void },\n options: ChatRequestOptions & {\n files?: FileList | FileUIPart[];\n } = {},\n ) => {\n event?.preventDefault?.();\n\n const fileParts = Array.isArray(options?.files)\n ? options.files\n : await convertFileListToFileUIParts(options?.files);\n\n if (!input && fileParts.length === 0) return;\n\n append(\n {\n id: generateId(),\n role: 'user',\n metadata: undefined,\n parts: [...fileParts, { type: 'text', text: input }],\n },\n {\n headers: options.headers,\n body: options.body,\n },\n );\n\n setInput('');\n },\n [input, generateId, append],\n );\n\n const handleInputChange = (e: any) => {\n setInput(e.target.value);\n };\n\n return {\n messages,\n chatId: stableChatId,\n setMessages,\n error,\n append,\n reload,\n stop,\n experimental_resume,\n input,\n setInput,\n handleInputChange,\n handleSubmit,\n status,\n addToolResult,\n };\n}\n","import {\n CompletionRequestOptions,\n UseCompletionOptions,\n callCompletionApi,\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 return input ? complete(input) : undefined;\n },\n [input, complete],\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 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 FetchFunction,\n isAbortError,\n safeValidateTypes,\n} from '@ai-sdk/provider-utils';\nimport {\n asSchema,\n DeepPartial,\n isDeepEqualData,\n parsePartialJson,\n Schema,\n} from 'ai';\nimport { useCallback, useId, useRef, useState } from 'react';\nimport useSWR from 'swr';\nimport z from 'zod';\n\n// use function to allow for mocking in tests:\nconst getOriginalFetch = () => fetch;\n\nexport type Experimental_UseObjectOptions<RESULT> = {\n /**\n * The API endpoint. It should stream JSON that matches the schema as chunked text.\n */\n api: string;\n\n /**\n * A Zod schema that defines the shape of the complete object.\n */\n schema: z.Schema<RESULT, z.ZodTypeDef, any> | Schema<RESULT>;\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 /**\nCustom fetch implementation. You can use it as a middleware to intercept requests,\nor to provide a custom fetch implementation for e.g. testing.\n */\n fetch?: FetchFunction;\n\n /**\nCallback that is called when the stream has finished.\n */\n onFinish?: (event: {\n /**\nThe generated object (typed according to the schema).\nCan be undefined if the final object does not match the schema.\n */\n object: RESULT | undefined;\n\n /**\nOptional 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 */\n headers?: 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\nfunction useObject<RESULT, INPUT = any>({\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<RESULT>): Experimental_UseObjectHelpers<\n RESULT,\n INPUT\n> {\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 [api, completionId],\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 try {\n mutate(undefined); // reset the data\n setIsLoading(true);\n setError(undefined);\n\n const abortController = new AbortController();\n abortControllerRef.current = abortController;\n\n const actualFetch = fetch ?? getOriginalFetch();\n const response = await actualFetch(api, {\n method: 'POST',\n headers: {\n 'Content-Type': 'application/json',\n ...headers,\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 mutate(currentObject);\n }\n },\n\n async close() {\n setIsLoading(false);\n abortControllerRef.current = null;\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 }),\n );\n } catch (error) {\n if (isAbortError(error)) {\n return;\n }\n\n if (onError && error instanceof Error) {\n onError(error);\n }\n\n setIsLoading(false);\n setError(error instanceof Error ? error : new Error(String(error)));\n }\n };\n\n return {\n submit,\n object: data,\n error,\n isLoading,\n stop,\n };\n}\n\nexport const experimental_useObject = useObject;\n"],"mappings":";AAAA;AAAA,EAEE;AAAA,EACA;AAAA,EACA,cAAc;AAAA,OAST;AACP,SAAS,aAAa,QAAQ,UAAU,4BAA4B;AAmH7D,SAAS,QAGd;AAAA,EACA;AAAA,EACA,eAAe;AAAA,EACf;AAAA,EACA;AAAA,EACA;AAAA,EACA,aAAa;AAAA,EACb,uBAAuB;AAAA,EACvB,WAAW;AACb,IAMI,CAAC,GAAwD;AAE3D,QAAM,CAAC,MAAM,IAAI,SAAS,UAAU;AAGpC,QAAM,eAAe,0BAAU;AAI/B,QAAM,YAAY;AAAA,IAChB,sCACE,iBAAsD;AAAA,MACpD,KAAK;AAAA,MACL;AAAA,IACF,CAAC;AAAA,EACL;AAGA,MAAI,CAAC,UAAU,QAAQ,QAAQ,YAAY,GAAG;AAC5C,cAAU,QAAQ,QAAQ,cAAc,CAAC,CAAC;AAAA,EAC5C;AAEA,QAAM,YAAY;AAAA,IAChB,CAAC;AAAA,MACC;AAAA,MACA;AAAA,IACF,MAGM;AACJ,aAAO,UAAU,QAAQ,UAAU;AAAA,QACjC,eAAe,WAAS;AACtB,cAAI,MAAM,WAAW,gBAAgB,MAAM,SAAS,WAAW;AAC7D;AAAA,UACF;AAEA,wBAAc;AAAA,QAChB;AAAA,MACF,CAAC;AAAA,IACH;AAAA,IACA,CAAC,WAAW,YAAY;AAAA,EAC1B;AAEA,QAAM,gBAAgB;AAAA,IACpB,CACE,YAMG,UAAU,QAAQ,cAAc,EAAE,QAAQ,cAAc,GAAG,QAAQ,CAAC;AAAA,IACzE,CAAC,WAAW,YAAY;AAAA,EAC1B;AAEA,QAAM,aAAa,YAAY,MAAM;AACnC,cAAU,QAAQ,WAAW,EAAE,QAAQ,aAAa,CAAC;AAAA,EACvD,GAAG,CAAC,WAAW,YAAY,CAAC;AAE5B,QAAM,QAAQ;AAAA,IACZ,cACE,UAAU;AAAA,MACR,eAAe;AAAA,MACf,WAAW;AAAA,IACb,CAAC;AAAA,IACH,MAAM,UAAU,QAAQ,SAAS,YAAY;AAAA,IAC7C,MAAM,UAAU,QAAQ,SAAS,YAAY;AAAA,EAC/C;AAEA,QAAM,SAAS;AAAA,IACb,cACE,UAAU;AAAA,MACR,eAAe;AAAA,MACf,WAAW;AAAA,IACb,CAAC;AAAA,IACH,MAAM,UAAU,QAAQ,UAAU,YAAY;AAAA,IAC9C,MAAM,UAAU,QAAQ,UAAU,YAAY;AAAA,EAChD;AAEA,QAAM,WAAW;AAAA,IACf,cAAY;AACV,aAAO,UAAU;AAAA,QACf,eAAe;AAAA,QACf,WAAW;AAAA,MACb,CAAC;AAAA,IACH;AAAA,IACA,MAAM,UAAU,QAAQ,YAAY,YAAY;AAAA,IAChD,MAAM,UAAU,QAAQ,YAAY,YAAY;AAAA,EAClD;AAEA,QAAM,SAAS;AAAA,IACb,CACE,SAIA,EAAE,SAAS,KAAK,IAAwB,CAAC,MAEzC,UAAU,QAAQ,cAAc;AAAA,MAC9B,QAAQ;AAAA,MACR;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,IACF,CAAC;AAAA,IACH,CAAC,WAAW,cAAc,SAAS,YAAY,QAAQ;AAAA,EACzD;AAEA,QAAM,SAAS;AAAA,IACb,OAAO,EAAE,SAAS,KAAK,IAAwB,CAAC,MAC9C,UAAU,QAAQ,wBAAwB;AAAA,MACxC,QAAQ;AAAA,MACR;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,IACF,CAAC;AAAA,IACH,CAAC,WAAW,cAAc,SAAS,YAAY,QAAQ;AAAA,EACzD;AACA,QAAM,OAAO,YAAY,MAAM,WAAW,GAAG,CAAC,UAAU,CAAC;AAEzD,QAAM,sBAAsB;AAAA,IAC1B,YACE,UAAU,QAAQ,aAAa;AAAA,MAC7B,QAAQ;AAAA,MACR;AAAA,MACA;AAAA,MACA;AAAA,IACF,CAAC;AAAA,IACH,CAAC,WAAW,cAAc,SAAS,YAAY,QAAQ;AAAA,EACzD;AAEA,QAAM,cAAc;AAAA,IAClB,CACE,kBAWG;AACH,UAAI,OAAO,kBAAkB,YAAY;AACvC,wBAAgB,cAAc,QAAQ;AAAA,MACxC;AAEA,gBAAU,QAAQ,YAAY;AAAA,QAC5B,IAAI;AAAA,QACJ,UAAU;AAAA,MACZ,CAAC;AAAA,IACH;AAAA,IACA,CAAC,cAAc,QAAQ;AAAA,EACzB;AAGA,QAAM,CAAC,OAAO,QAAQ,IAAI,SAAS,YAAY;AAE/C,QAAM,eAAe;AAAA,IACnB,OACE,OACA,UAEI,CAAC,MACF;AA7TT;AA8TM,2CAAO,mBAAP;AAEA,YAAM,YAAY,MAAM,QAAQ,mCAAS,KAAK,IAC1C,QAAQ,QACR,MAAM,6BAA6B,mCAAS,KAAK;AAErD,UAAI,CAAC,SAAS,UAAU,WAAW;AAAG;AAEtC;AAAA,QACE;AAAA,UACE,IAAI,WAAW;AAAA,UACf,MAAM;AAAA,UACN,UAAU;AAAA,UACV,OAAO,CAAC,GAAG,WAAW,EAAE,MAAM,QAAQ,MAAM,MAAM,CAAC;AAAA,QACrD;AAAA,QACA;AAAA,UACE,SAAS,QAAQ;AAAA,UACjB,MAAM,QAAQ;AAAA,QAChB;AAAA,MACF;AAEA,eAAS,EAAE;AAAA,IACb;AAAA,IACA,CAAC,OAAO,YAAY,MAAM;AAAA,EAC5B;AAEA,QAAM,oBAAoB,CAAC,MAAW;AACpC,aAAS,EAAE,OAAO,KAAK;AAAA,EACzB;AAEA,SAAO;AAAA,IACL;AAAA,IACA,QAAQ;AAAA,IACR;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF;AACF;;;AC5WA;AAAA,EAGE;AAAA,OACK;AACP,SAAS,eAAAA,cAAa,WAAW,OAAO,UAAAC,SAAQ,YAAAC,iBAAgB;AAChE,OAAO,YAAY;;;ACNnB,OAAO,sBAAsB;AAEtB,SAAS,SACd,IACA,QACG;AACH,SAAO,UAAU,OAAO,iBAAiB,IAAI,MAAM,IAAI;AACzD;;;ADwDO,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,IAAIC,UAA4B,MAAS;AAC/D,QAAM,aAAa;AAGnB,QAAM,CAAC,iBAAiB,kBAAkB,IACxCA,UAAiC,IAAI;AAEvC,QAAM,mBAAmBC,QAAO;AAAA,IAC9B;AAAA,IACA;AAAA,IACA;AAAA,EACF,CAAC;AAED,YAAU,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,IAAIF,UAAS,YAAY;AAE/C,QAAM,eAAeE;AAAA,IACnB,CAAC,UAA4C;AArLjD;AAsLM,2CAAO,mBAAP;AACA,aAAO,QAAQ,SAAS,KAAK,IAAI;AAAA,IACnC;AAAA,IACA,CAAC,OAAO,QAAQ;AAAA,EAClB;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;;;AE/MA;AAAA,EAEE;AAAA,EACA;AAAA,OACK;AACP;AAAA,EACE;AAAA,EAEA;AAAA,EACA;AAAA,OAEK;AACP,SAAS,eAAAE,cAAa,SAAAC,QAAO,UAAAC,SAAQ,YAAAC,iBAAgB;AACrD,OAAOC,aAAY;AAInB,IAAM,mBAAmB,MAAM;AA4F/B,SAAS,UAA+B;AAAA,EACtC;AAAA,EACA;AAAA,EACA;AAAA;AAAA,EACA;AAAA,EACA,OAAAC;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AACF,GAGE;AAEA,QAAM,SAASJ,OAAM;AACrB,QAAM,eAAe,kBAAM;AAG3B,QAAM,EAAE,MAAM,OAAO,IAAIG;AAAA,IACvB,CAAC,KAAK,YAAY;AAAA,IAClB;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;AA5IjC;AA6II,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;AAtJzC;AAuJI,QAAI;AACF,aAAO,MAAS;AAChB,mBAAa,IAAI;AACjB,eAAS,MAAS;AAElB,YAAM,kBAAkB,IAAI,gBAAgB;AAC5C,yBAAmB,UAAU;AAE7B,YAAM,cAAcK,UAAA,OAAAA,SAAS,iBAAiB;AAC9C,YAAM,WAAW,MAAM,YAAY,KAAK;AAAA,QACtC,QAAQ;AAAA,QACR,SAAS;AAAA,UACP,gBAAgB;AAAA,UAChB,GAAG;AAAA,QACL;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,qBAAO,aAAa;AAAA,YACtB;AAAA,UACF;AAAA,UAEA,MAAM,QAAQ;AACZ,yBAAa,KAAK;AAClB,+BAAmB,UAAU;AAE7B,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;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,mBAAa,KAAK;AAClB,eAASA,kBAAiB,QAAQA,SAAQ,IAAI,MAAM,OAAOA,MAAK,CAAC,CAAC;AAAA,IACpE;AAAA,EACF;AAEA,SAAO;AAAA,IACL;AAAA,IACA,QAAQ;AAAA,IACR;AAAA,IACA;AAAA,IACA;AAAA,EACF;AACF;AAEO,IAAM,yBAAyB;","names":["useCallback","useRef","useState","fetch","useState","useRef","useCallback","completion","useCallback","useId","useRef","useState","useSWR","fetch","error"]}
|
|
1
|
+
{"version":3,"sources":["../src/use-chat.ts","../src/chat-store.ts","../src/throttle.ts","../src/use-completion.ts","../src/use-object.ts"],"sourcesContent":["import {\n ChatStore,\n convertFileListToFileUIParts,\n defaultChatStoreOptions,\n generateId as generateIdFunc,\n InferUIDataParts,\n UIDataPartSchemas,\n type ChatRequestOptions,\n type ChatStoreEvent,\n type CreateUIMessage,\n type FileUIPart,\n type UIMessage,\n type UseChatOptions,\n} from 'ai';\nimport { useCallback, useRef, useState, useSyncExternalStore } from 'react';\nimport { createChatStore } from './chat-store';\nimport { throttle } from './throttle';\n\nexport type { CreateUIMessage, UIMessage, UseChatOptions };\n\nexport type UseChatHelpers<\n MESSAGE_METADATA = unknown,\n DATA_PART_SCHEMAS extends UIDataPartSchemas = UIDataPartSchemas,\n> = {\n /**\n * The id of the chat.\n */\n readonly chatId: string;\n\n /**\n * Hook status:\n *\n * - `submitted`: The message has been sent to the API and we're awaiting the start of the response stream.\n * - `streaming`: The response is actively streaming in from the API, receiving chunks of data.\n * - `ready`: The full response has been received and processed; a new user message can be submitted.\n * - `error`: An error occurred during the API request, preventing successful completion.\n */\n readonly status: 'submitted' | 'streaming' | 'ready' | 'error';\n\n /** Current messages in the chat */\n readonly messages: UIMessage<\n MESSAGE_METADATA,\n InferUIDataParts<DATA_PART_SCHEMAS>\n >[];\n\n /** The error object of the API request */\n readonly error: undefined | Error;\n\n /**\n * Append a user message to the chat list. This triggers the API call to fetch\n * the assistant's response.\n *\n * @param message The message to append\n * @param options Additional options to pass to the API call\n */\n append: (\n message: CreateUIMessage<\n MESSAGE_METADATA,\n InferUIDataParts<DATA_PART_SCHEMAS>\n >,\n options?: ChatRequestOptions,\n ) => Promise<void>;\n\n /**\n * Reload the last AI chat response for the given chat history. If the last\n * message isn't from the assistant, it will request the API to generate a\n * new response.\n */\n reload: (\n chatRequestOptions?: ChatRequestOptions,\n ) => Promise<string | null | undefined>;\n\n /**\n * Abort the current request immediately, keep the generated tokens if any.\n */\n stop: () => void;\n\n /**\n * Resume an ongoing chat generation stream. This does not resume an aborted generation.\n */\n experimental_resume: () => void;\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:\n | UIMessage<MESSAGE_METADATA, InferUIDataParts<DATA_PART_SCHEMAS>>[]\n | ((\n messages: UIMessage<\n MESSAGE_METADATA,\n InferUIDataParts<DATA_PART_SCHEMAS>\n >[],\n ) => UIMessage<\n MESSAGE_METADATA,\n InferUIDataParts<DATA_PART_SCHEMAS>\n >[]),\n ) => void;\n\n /** The current value of the input */\n input: string;\n\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 handleInputChange: (\n e:\n | React.ChangeEvent<HTMLInputElement>\n | React.ChangeEvent<HTMLTextAreaElement>,\n ) => void;\n\n /** Form submission handler to automatically reset input and append a user message */\n handleSubmit: (\n event?: { preventDefault?: () => void },\n chatRequestOptions?: ChatRequestOptions & {\n files?: FileList | FileUIPart[];\n },\n ) => void;\n\n addToolResult: ({\n toolCallId,\n result,\n }: {\n toolCallId: string;\n result: any;\n }) => void;\n};\n\nexport function useChat<\n MESSAGE_METADATA = unknown,\n DATA_PART_SCHEMAS extends UIDataPartSchemas = UIDataPartSchemas,\n>({\n chatId,\n initialInput = '',\n onToolCall,\n onFinish,\n onError,\n generateId = generateIdFunc,\n experimental_throttle: throttleWaitMs,\n chatStore: chatStoreArg,\n}: UseChatOptions<MESSAGE_METADATA, DATA_PART_SCHEMAS> & {\n /**\nCustom throttle wait in ms for the chat messages and data updates.\nDefault is undefined, which disables throttling.\n */\n experimental_throttle?: number;\n} = {}): UseChatHelpers<MESSAGE_METADATA, DATA_PART_SCHEMAS> {\n // Generate ID once, store in state for stability across re-renders\n const [hookId] = useState(generateId);\n\n // Use the caller-supplied ID if available; otherwise, fall back to our stable ID\n const stableChatId = chatId ?? hookId;\n\n // chat store setup\n const chatStore = useRef(\n chatStoreArg == null\n ? createChatStore(\n defaultChatStoreOptions<MESSAGE_METADATA, DATA_PART_SCHEMAS>({\n api: '/api/chat',\n generateId,\n })(),\n )\n : typeof chatStoreArg === 'function'\n ? createChatStore(chatStoreArg())\n : chatStoreArg,\n );\n\n // ensure the chat is in the store\n if (!chatStore.current.hasChat(stableChatId)) {\n chatStore.current.addChat(stableChatId, []);\n }\n\n const subscribe = useCallback(\n ({\n onStoreChange,\n eventType,\n }: {\n onStoreChange: () => void;\n eventType: ChatStoreEvent['type'];\n }) => {\n return chatStore.current.subscribe({\n onChatChanged: event => {\n if (event.chatId !== stableChatId || event.type !== eventType) return;\n onStoreChange();\n },\n });\n },\n [chatStore, stableChatId],\n );\n\n const addToolResult = useCallback(\n (\n options: Omit<\n Parameters<\n ChatStore<MESSAGE_METADATA, DATA_PART_SCHEMAS>['addToolResult']\n >[0],\n 'chatId'\n >,\n ) => chatStore.current.addToolResult({ chatId: stableChatId, ...options }),\n [chatStore, stableChatId],\n );\n\n const stopStream = useCallback(() => {\n chatStore.current.stopStream({ chatId: stableChatId });\n }, [chatStore, stableChatId]);\n\n const error = useSyncExternalStore(\n callback =>\n subscribe({\n onStoreChange: callback,\n eventType: 'chat-status-changed',\n }),\n () => chatStore.current.getError(stableChatId),\n () => chatStore.current.getError(stableChatId),\n );\n\n const status = useSyncExternalStore(\n callback =>\n subscribe({\n onStoreChange: callback,\n eventType: 'chat-status-changed',\n }),\n () => chatStore.current.getStatus(stableChatId),\n () => chatStore.current.getStatus(stableChatId),\n );\n\n const subscribeToChatStoreForMessages = useCallback(\n (callback: () => void) => {\n return subscribe({\n onStoreChange: throttleWaitMs\n ? throttle(callback, throttleWaitMs)\n : callback,\n eventType: 'chat-messages-changed',\n });\n },\n [subscribe, throttleWaitMs],\n );\n\n const messages = useSyncExternalStore(\n callback => subscribeToChatStoreForMessages(callback),\n () => chatStore.current.getMessages(stableChatId),\n () => chatStore.current.getMessages(stableChatId),\n );\n\n const append = useCallback(\n (\n message: CreateUIMessage<\n MESSAGE_METADATA,\n InferUIDataParts<DATA_PART_SCHEMAS>\n >,\n { headers, body }: ChatRequestOptions = {},\n ) =>\n chatStore.current.submitMessage({\n chatId: stableChatId,\n message,\n headers,\n body,\n onError,\n onToolCall,\n onFinish,\n }),\n [chatStore, stableChatId, onError, onToolCall, onFinish],\n );\n\n const reload = useCallback(\n async ({ headers, body }: ChatRequestOptions = {}) =>\n chatStore.current.resubmitLastUserMessage({\n chatId: stableChatId,\n headers,\n body,\n onError,\n onToolCall,\n onFinish,\n }),\n [chatStore, stableChatId, onError, onToolCall, onFinish],\n );\n const stop = useCallback(() => stopStream(), [stopStream]);\n\n const experimental_resume = useCallback(\n async () =>\n chatStore.current.resumeStream({\n chatId: stableChatId,\n onError,\n onToolCall,\n onFinish,\n }),\n [chatStore, stableChatId, onError, onToolCall, onFinish],\n );\n\n const setMessages = useCallback(\n (\n messagesParam:\n | UIMessage<MESSAGE_METADATA, InferUIDataParts<DATA_PART_SCHEMAS>>[]\n | ((\n messages: UIMessage<\n MESSAGE_METADATA,\n InferUIDataParts<DATA_PART_SCHEMAS>\n >[],\n ) => UIMessage<\n MESSAGE_METADATA,\n InferUIDataParts<DATA_PART_SCHEMAS>\n >[]),\n ) => {\n if (typeof messagesParam === 'function') {\n messagesParam = messagesParam(messages);\n }\n\n chatStore.current.setMessages({\n id: stableChatId,\n messages: messagesParam,\n });\n },\n [stableChatId, messages],\n );\n\n // Input state and handlers.\n const [input, setInput] = useState(initialInput);\n\n const handleSubmit = useCallback(\n async (\n event?: { preventDefault?: () => void },\n options: ChatRequestOptions & {\n files?: FileList | FileUIPart[];\n } = {},\n ) => {\n event?.preventDefault?.();\n\n const fileParts = Array.isArray(options?.files)\n ? options.files\n : await convertFileListToFileUIParts(options?.files);\n\n if (!input && fileParts.length === 0) return;\n\n append(\n {\n id: generateId(),\n role: 'user',\n metadata: undefined,\n parts: [...fileParts, { type: 'text', text: input }],\n },\n {\n headers: options.headers,\n body: options.body,\n },\n );\n\n setInput('');\n },\n [input, generateId, append],\n );\n\n const handleInputChange = (e: any) => {\n setInput(e.target.value);\n };\n\n return {\n messages,\n chatId: stableChatId,\n setMessages,\n error,\n append,\n reload,\n stop,\n experimental_resume,\n input,\n setInput,\n handleInputChange,\n handleSubmit,\n status,\n addToolResult,\n };\n}\n","import {\n ChatStatus,\n ChatStore,\n ChatStoreOptions,\n InferUIDataParts,\n SerialJobExecutor,\n UIDataPartSchemas,\n UIDataTypes,\n UIMessage,\n type ActiveResponse,\n type Chat,\n} from 'ai';\n\nclass ReactChat<MESSAGE_METADATA, DATA_TYPES extends UIDataTypes>\n implements Chat<MESSAGE_METADATA, DATA_TYPES>\n{\n messages: UIMessage<MESSAGE_METADATA, DATA_TYPES>[];\n status: ChatStatus = 'ready';\n error: Error | undefined = undefined;\n activeResponse: ActiveResponse<MESSAGE_METADATA> | undefined = undefined;\n jobExecutor = new SerialJobExecutor();\n\n constructor(messages?: UIMessage<MESSAGE_METADATA, DATA_TYPES>[]) {\n this.messages = messages ?? [];\n }\n\n setStatus = (status: ChatStatus) => {\n this.status = status;\n };\n\n setError = (error: Error | undefined) => {\n this.error = error;\n };\n\n setActiveResponse = (\n activeResponse: ActiveResponse<MESSAGE_METADATA> | undefined,\n ) => {\n this.activeResponse = activeResponse;\n };\n\n setMessages = (messages: UIMessage<MESSAGE_METADATA, DATA_TYPES>[]) => {\n this.messages = [...messages];\n };\n\n pushMessage = (message: UIMessage<MESSAGE_METADATA, DATA_TYPES>) => {\n this.messages = this.messages.concat(message);\n };\n\n popMessage = () => {\n this.messages = this.messages.slice(0, -1);\n };\n\n replaceMessage = (\n index: number,\n message: UIMessage<MESSAGE_METADATA, DATA_TYPES>,\n ) => {\n this.messages = [\n ...this.messages.slice(0, index),\n message,\n ...this.messages.slice(index + 1),\n ];\n };\n\n snapshot = <T>(value: T): T => {\n return structuredClone(value);\n };\n}\n\nexport function createChatStore<\n MESSAGE_METADATA = unknown,\n DATA_PART_SCHEMAS extends UIDataPartSchemas = UIDataPartSchemas,\n>(\n options: ChatStoreOptions<MESSAGE_METADATA, DATA_PART_SCHEMAS>,\n): ChatStore<MESSAGE_METADATA, DATA_PART_SCHEMAS> {\n return new ChatStore<MESSAGE_METADATA, DATA_PART_SCHEMAS>({\n ...options,\n createChat: options =>\n new ReactChat<MESSAGE_METADATA, InferUIDataParts<DATA_PART_SCHEMAS>>(\n options.messages,\n ),\n });\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 CompletionRequestOptions,\n UseCompletionOptions,\n callCompletionApi,\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 return input ? complete(input) : undefined;\n },\n [input, complete],\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 FetchFunction,\n isAbortError,\n safeValidateTypes,\n} from '@ai-sdk/provider-utils';\nimport {\n asSchema,\n DeepPartial,\n isDeepEqualData,\n parsePartialJson,\n Schema,\n} from 'ai';\nimport { useCallback, useId, useRef, useState } from 'react';\nimport useSWR from 'swr';\nimport z from 'zod';\n\n// use function to allow for mocking in tests:\nconst getOriginalFetch = () => fetch;\n\nexport type Experimental_UseObjectOptions<RESULT> = {\n /**\n * The API endpoint. It should stream JSON that matches the schema as chunked text.\n */\n api: string;\n\n /**\n * A Zod schema that defines the shape of the complete object.\n */\n schema: z.Schema<RESULT, z.ZodTypeDef, any> | Schema<RESULT>;\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 /**\nCustom fetch implementation. You can use it as a middleware to intercept requests,\nor to provide a custom fetch implementation for e.g. testing.\n */\n fetch?: FetchFunction;\n\n /**\nCallback that is called when the stream has finished.\n */\n onFinish?: (event: {\n /**\nThe generated object (typed according to the schema).\nCan be undefined if the final object does not match the schema.\n */\n object: RESULT | undefined;\n\n /**\nOptional 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 */\n headers?: 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\nfunction useObject<RESULT, INPUT = any>({\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<RESULT>): Experimental_UseObjectHelpers<\n RESULT,\n INPUT\n> {\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 [api, completionId],\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 try {\n mutate(undefined); // reset the data\n setIsLoading(true);\n setError(undefined);\n\n const abortController = new AbortController();\n abortControllerRef.current = abortController;\n\n const actualFetch = fetch ?? getOriginalFetch();\n const response = await actualFetch(api, {\n method: 'POST',\n headers: {\n 'Content-Type': 'application/json',\n ...headers,\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 mutate(currentObject);\n }\n },\n\n async close() {\n setIsLoading(false);\n abortControllerRef.current = null;\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 }),\n );\n } catch (error) {\n if (isAbortError(error)) {\n return;\n }\n\n if (onError && error instanceof Error) {\n onError(error);\n }\n\n setIsLoading(false);\n setError(error instanceof Error ? error : new Error(String(error)));\n }\n };\n\n return {\n submit,\n object: data,\n error,\n isLoading,\n stop,\n };\n}\n\nexport const experimental_useObject = useObject;\n"],"mappings":";AAAA;AAAA,EAEE;AAAA,EACA;AAAA,EACA,cAAc;AAAA,OAST;AACP,SAAS,aAAa,QAAQ,UAAU,4BAA4B;;;ACdpE;AAAA,EAEE;AAAA,EAGA;AAAA,OAMK;AAEP,IAAM,YAAN,MAEA;AAAA,EAOE,YAAY,UAAsD;AALlE,kBAAqB;AACrB,iBAA2B;AAC3B,0BAA+D;AAC/D,uBAAc,IAAI,kBAAkB;AAMpC,qBAAY,CAAC,WAAuB;AAClC,WAAK,SAAS;AAAA,IAChB;AAEA,oBAAW,CAAC,UAA6B;AACvC,WAAK,QAAQ;AAAA,IACf;AAEA,6BAAoB,CAClB,mBACG;AACH,WAAK,iBAAiB;AAAA,IACxB;AAEA,uBAAc,CAAC,aAAwD;AACrE,WAAK,WAAW,CAAC,GAAG,QAAQ;AAAA,IAC9B;AAEA,uBAAc,CAAC,YAAqD;AAClE,WAAK,WAAW,KAAK,SAAS,OAAO,OAAO;AAAA,IAC9C;AAEA,sBAAa,MAAM;AACjB,WAAK,WAAW,KAAK,SAAS,MAAM,GAAG,EAAE;AAAA,IAC3C;AAEA,0BAAiB,CACf,OACA,YACG;AACH,WAAK,WAAW;AAAA,QACd,GAAG,KAAK,SAAS,MAAM,GAAG,KAAK;AAAA,QAC/B;AAAA,QACA,GAAG,KAAK,SAAS,MAAM,QAAQ,CAAC;AAAA,MAClC;AAAA,IACF;AAEA,oBAAW,CAAI,UAAgB;AAC7B,aAAO,gBAAgB,KAAK;AAAA,IAC9B;AA1CE,SAAK,WAAW,8BAAY,CAAC;AAAA,EAC/B;AA0CF;AAEO,SAAS,gBAId,SACgD;AAChD,SAAO,IAAI,UAA+C;AAAA,IACxD,GAAG;AAAA,IACH,YAAY,CAAAA,aACV,IAAI;AAAA,MACFA,SAAQ;AAAA,IACV;AAAA,EACJ,CAAC;AACH;;;ACjFA,OAAO,sBAAsB;AAEtB,SAAS,SACd,IACA,QACG;AACH,SAAO,UAAU,OAAO,iBAAiB,IAAI,MAAM,IAAI;AACzD;;;AF4HO,SAAS,QAGd;AAAA,EACA;AAAA,EACA,eAAe;AAAA,EACf;AAAA,EACA;AAAA,EACA;AAAA,EACA,aAAa;AAAA,EACb,uBAAuB;AAAA,EACvB,WAAW;AACb,IAMI,CAAC,GAAwD;AAE3D,QAAM,CAAC,MAAM,IAAI,SAAS,UAAU;AAGpC,QAAM,eAAe,0BAAU;AAG/B,QAAM,YAAY;AAAA,IAChB,gBAAgB,OACZ;AAAA,MACE,wBAA6D;AAAA,QAC3D,KAAK;AAAA,QACL;AAAA,MACF,CAAC,EAAE;AAAA,IACL,IACA,OAAO,iBAAiB,aACtB,gBAAgB,aAAa,CAAC,IAC9B;AAAA,EACR;AAGA,MAAI,CAAC,UAAU,QAAQ,QAAQ,YAAY,GAAG;AAC5C,cAAU,QAAQ,QAAQ,cAAc,CAAC,CAAC;AAAA,EAC5C;AAEA,QAAM,YAAY;AAAA,IAChB,CAAC;AAAA,MACC;AAAA,MACA;AAAA,IACF,MAGM;AACJ,aAAO,UAAU,QAAQ,UAAU;AAAA,QACjC,eAAe,WAAS;AACtB,cAAI,MAAM,WAAW,gBAAgB,MAAM,SAAS;AAAW;AAC/D,wBAAc;AAAA,QAChB;AAAA,MACF,CAAC;AAAA,IACH;AAAA,IACA,CAAC,WAAW,YAAY;AAAA,EAC1B;AAEA,QAAM,gBAAgB;AAAA,IACpB,CACE,YAMG,UAAU,QAAQ,cAAc,EAAE,QAAQ,cAAc,GAAG,QAAQ,CAAC;AAAA,IACzE,CAAC,WAAW,YAAY;AAAA,EAC1B;AAEA,QAAM,aAAa,YAAY,MAAM;AACnC,cAAU,QAAQ,WAAW,EAAE,QAAQ,aAAa,CAAC;AAAA,EACvD,GAAG,CAAC,WAAW,YAAY,CAAC;AAE5B,QAAM,QAAQ;AAAA,IACZ,cACE,UAAU;AAAA,MACR,eAAe;AAAA,MACf,WAAW;AAAA,IACb,CAAC;AAAA,IACH,MAAM,UAAU,QAAQ,SAAS,YAAY;AAAA,IAC7C,MAAM,UAAU,QAAQ,SAAS,YAAY;AAAA,EAC/C;AAEA,QAAM,SAAS;AAAA,IACb,cACE,UAAU;AAAA,MACR,eAAe;AAAA,MACf,WAAW;AAAA,IACb,CAAC;AAAA,IACH,MAAM,UAAU,QAAQ,UAAU,YAAY;AAAA,IAC9C,MAAM,UAAU,QAAQ,UAAU,YAAY;AAAA,EAChD;AAEA,QAAM,kCAAkC;AAAA,IACtC,CAAC,aAAyB;AACxB,aAAO,UAAU;AAAA,QACf,eAAe,iBACX,SAAS,UAAU,cAAc,IACjC;AAAA,QACJ,WAAW;AAAA,MACb,CAAC;AAAA,IACH;AAAA,IACA,CAAC,WAAW,cAAc;AAAA,EAC5B;AAEA,QAAM,WAAW;AAAA,IACf,cAAY,gCAAgC,QAAQ;AAAA,IACpD,MAAM,UAAU,QAAQ,YAAY,YAAY;AAAA,IAChD,MAAM,UAAU,QAAQ,YAAY,YAAY;AAAA,EAClD;AAEA,QAAM,SAAS;AAAA,IACb,CACE,SAIA,EAAE,SAAS,KAAK,IAAwB,CAAC,MAEzC,UAAU,QAAQ,cAAc;AAAA,MAC9B,QAAQ;AAAA,MACR;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,IACF,CAAC;AAAA,IACH,CAAC,WAAW,cAAc,SAAS,YAAY,QAAQ;AAAA,EACzD;AAEA,QAAM,SAAS;AAAA,IACb,OAAO,EAAE,SAAS,KAAK,IAAwB,CAAC,MAC9C,UAAU,QAAQ,wBAAwB;AAAA,MACxC,QAAQ;AAAA,MACR;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,IACF,CAAC;AAAA,IACH,CAAC,WAAW,cAAc,SAAS,YAAY,QAAQ;AAAA,EACzD;AACA,QAAM,OAAO,YAAY,MAAM,WAAW,GAAG,CAAC,UAAU,CAAC;AAEzD,QAAM,sBAAsB;AAAA,IAC1B,YACE,UAAU,QAAQ,aAAa;AAAA,MAC7B,QAAQ;AAAA,MACR;AAAA,MACA;AAAA,MACA;AAAA,IACF,CAAC;AAAA,IACH,CAAC,WAAW,cAAc,SAAS,YAAY,QAAQ;AAAA,EACzD;AAEA,QAAM,cAAc;AAAA,IAClB,CACE,kBAWG;AACH,UAAI,OAAO,kBAAkB,YAAY;AACvC,wBAAgB,cAAc,QAAQ;AAAA,MACxC;AAEA,gBAAU,QAAQ,YAAY;AAAA,QAC5B,IAAI;AAAA,QACJ,UAAU;AAAA,MACZ,CAAC;AAAA,IACH;AAAA,IACA,CAAC,cAAc,QAAQ;AAAA,EACzB;AAGA,QAAM,CAAC,OAAO,QAAQ,IAAI,SAAS,YAAY;AAE/C,QAAM,eAAe;AAAA,IACnB,OACE,OACA,UAEI,CAAC,MACF;AAvUT;AAwUM,2CAAO,mBAAP;AAEA,YAAM,YAAY,MAAM,QAAQ,mCAAS,KAAK,IAC1C,QAAQ,QACR,MAAM,6BAA6B,mCAAS,KAAK;AAErD,UAAI,CAAC,SAAS,UAAU,WAAW;AAAG;AAEtC;AAAA,QACE;AAAA,UACE,IAAI,WAAW;AAAA,UACf,MAAM;AAAA,UACN,UAAU;AAAA,UACV,OAAO,CAAC,GAAG,WAAW,EAAE,MAAM,QAAQ,MAAM,MAAM,CAAC;AAAA,QACrD;AAAA,QACA;AAAA,UACE,SAAS,QAAQ;AAAA,UACjB,MAAM,QAAQ;AAAA,QAChB;AAAA,MACF;AAEA,eAAS,EAAE;AAAA,IACb;AAAA,IACA,CAAC,OAAO,YAAY,MAAM;AAAA,EAC5B;AAEA,QAAM,oBAAoB,CAAC,MAAW;AACpC,aAAS,EAAE,OAAO,KAAK;AAAA,EACzB;AAEA,SAAO;AAAA,IACL;AAAA,IACA,QAAQ;AAAA,IACR;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF;AACF;;;AGtXA;AAAA,EAGE;AAAA,OACK;AACP,SAAS,eAAAC,cAAa,WAAW,OAAO,UAAAC,SAAQ,YAAAC,iBAAgB;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,IAAIC,UAA4B,MAAS;AAC/D,QAAM,aAAa;AAGnB,QAAM,CAAC,iBAAiB,kBAAkB,IACxCA,UAAiC,IAAI;AAEvC,QAAM,mBAAmBC,QAAO;AAAA,IAC9B;AAAA,IACA;AAAA,IACA;AAAA,EACF,CAAC;AAED,YAAU,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,IAAIF,UAAS,YAAY;AAE/C,QAAM,eAAeE;AAAA,IACnB,CAAC,UAA4C;AArLjD;AAsLM,2CAAO,mBAAP;AACA,aAAO,QAAQ,SAAS,KAAK,IAAI;AAAA,IACnC;AAAA,IACA,CAAC,OAAO,QAAQ;AAAA,EAClB;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;;;AC/MA;AAAA,EAEE;AAAA,EACA;AAAA,OACK;AACP;AAAA,EACE;AAAA,EAEA;AAAA,EACA;AAAA,OAEK;AACP,SAAS,eAAAE,cAAa,SAAAC,QAAO,UAAAC,SAAQ,YAAAC,iBAAgB;AACrD,OAAOC,aAAY;AAInB,IAAM,mBAAmB,MAAM;AA4F/B,SAAS,UAA+B;AAAA,EACtC;AAAA,EACA;AAAA,EACA;AAAA;AAAA,EACA;AAAA,EACA,OAAAC;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AACF,GAGE;AAEA,QAAM,SAASJ,OAAM;AACrB,QAAM,eAAe,kBAAM;AAG3B,QAAM,EAAE,MAAM,OAAO,IAAIG;AAAA,IACvB,CAAC,KAAK,YAAY;AAAA,IAClB;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;AA5IjC;AA6II,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;AAtJzC;AAuJI,QAAI;AACF,aAAO,MAAS;AAChB,mBAAa,IAAI;AACjB,eAAS,MAAS;AAElB,YAAM,kBAAkB,IAAI,gBAAgB;AAC5C,yBAAmB,UAAU;AAE7B,YAAM,cAAcK,UAAA,OAAAA,SAAS,iBAAiB;AAC9C,YAAM,WAAW,MAAM,YAAY,KAAK;AAAA,QACtC,QAAQ;AAAA,QACR,SAAS;AAAA,UACP,gBAAgB;AAAA,UAChB,GAAG;AAAA,QACL;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,qBAAO,aAAa;AAAA,YACtB;AAAA,UACF;AAAA,UAEA,MAAM,QAAQ;AACZ,yBAAa,KAAK;AAClB,+BAAmB,UAAU;AAE7B,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;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,mBAAa,KAAK;AAClB,eAASA,kBAAiB,QAAQA,SAAQ,IAAI,MAAM,OAAOA,MAAK,CAAC,CAAC;AAAA,IACpE;AAAA,EACF;AAEA,SAAO;AAAA,IACL;AAAA,IACA,QAAQ;AAAA,IACR;AAAA,IACA;AAAA,IACA;AAAA,EACF;AACF;AAEO,IAAM,yBAAyB;","names":["options","useCallback","useRef","useState","fetch","useState","useRef","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": "2.0.0-alpha.
|
|
3
|
+
"version": "2.0.0-alpha.5",
|
|
4
4
|
"license": "Apache-2.0",
|
|
5
5
|
"sideEffects": false,
|
|
6
6
|
"main": "./dist/index.js",
|
|
@@ -21,8 +21,8 @@
|
|
|
21
21
|
"dependencies": {
|
|
22
22
|
"swr": "^2.2.5",
|
|
23
23
|
"throttleit": "2.1.0",
|
|
24
|
-
"ai": "
|
|
25
|
-
"
|
|
24
|
+
"@ai-sdk/provider-utils": "3.0.0-alpha.4",
|
|
25
|
+
"ai": "5.0.0-alpha.5"
|
|
26
26
|
},
|
|
27
27
|
"devDependencies": {
|
|
28
28
|
"@testing-library/jest-dom": "^6.6.3",
|