@ai-sdk/react 2.0.0-alpha.7 → 2.0.0-alpha.9
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 +20 -0
- package/dist/index.d.mts +27 -54
- package/dist/index.d.ts +27 -54
- package/dist/index.js +81 -136
- package/dist/index.js.map +1 -1
- package/dist/index.mjs +83 -140
- package/dist/index.mjs.map +1 -1
- package/package.json +6 -6
package/CHANGELOG.md
CHANGED
|
@@ -1,5 +1,25 @@
|
|
|
1
1
|
# @ai-sdk/react
|
|
2
2
|
|
|
3
|
+
## 2.0.0-alpha.9
|
|
4
|
+
|
|
5
|
+
### Patch Changes
|
|
6
|
+
|
|
7
|
+
- Updated dependencies [8255639]
|
|
8
|
+
- Updated dependencies [9ae327d]
|
|
9
|
+
- ai@5.0.0-alpha.9
|
|
10
|
+
- @ai-sdk/provider-utils@3.0.0-alpha.9
|
|
11
|
+
|
|
12
|
+
## 2.0.0-alpha.8
|
|
13
|
+
|
|
14
|
+
### Patch Changes
|
|
15
|
+
|
|
16
|
+
- Updated dependencies [4fef487]
|
|
17
|
+
- Updated dependencies [6b1c55c]
|
|
18
|
+
- Updated dependencies [2e4f9e4]
|
|
19
|
+
- Updated dependencies [c25cbce]
|
|
20
|
+
- @ai-sdk/provider-utils@3.0.0-alpha.8
|
|
21
|
+
- ai@5.0.0-alpha.8
|
|
22
|
+
|
|
3
23
|
## 2.0.0-alpha.7
|
|
4
24
|
|
|
5
25
|
### Patch Changes
|
package/dist/index.d.mts
CHANGED
|
@@ -1,48 +1,18 @@
|
|
|
1
|
-
import { UIDataPartSchemas, UIMessage, InferUIDataParts,
|
|
2
|
-
export { CreateUIMessage, UIMessage,
|
|
3
|
-
import { FetchFunction } from '@ai-sdk/provider-utils';
|
|
4
|
-
import
|
|
1
|
+
import { UIDataPartSchemas, AbstractChat, BaseChatInit, UIMessage, InferUIDataParts, ChatRequestOptions, FileUIPart, CompletionRequestOptions, UseCompletionOptions, Schema, DeepPartial } from 'ai';
|
|
2
|
+
export { CreateUIMessage, UIMessage, UseCompletionOptions } from 'ai';
|
|
3
|
+
import { FetchFunction, InferSchema } from '@ai-sdk/provider-utils';
|
|
4
|
+
import * as z3 from 'zod/v3';
|
|
5
|
+
import * as z4 from 'zod/v4/core';
|
|
6
|
+
|
|
7
|
+
declare class Chat<MESSAGE_METADATA, UI_DATA_PART_SCHEMAS extends UIDataPartSchemas = UIDataPartSchemas> extends AbstractChat<MESSAGE_METADATA, UI_DATA_PART_SCHEMAS> {
|
|
8
|
+
constructor({ messages, ...init }: BaseChatInit<MESSAGE_METADATA, UI_DATA_PART_SCHEMAS>);
|
|
9
|
+
}
|
|
5
10
|
|
|
6
11
|
type UseChatHelpers<MESSAGE_METADATA = unknown, DATA_PART_SCHEMAS extends UIDataPartSchemas = UIDataPartSchemas> = {
|
|
7
12
|
/**
|
|
8
13
|
* The id of the chat.
|
|
9
14
|
*/
|
|
10
|
-
readonly
|
|
11
|
-
/**
|
|
12
|
-
* Hook status:
|
|
13
|
-
*
|
|
14
|
-
* - `submitted`: The message has been sent to the API and we're awaiting the start of the response stream.
|
|
15
|
-
* - `streaming`: The response is actively streaming in from the API, receiving chunks of data.
|
|
16
|
-
* - `ready`: The full response has been received and processed; a new user message can be submitted.
|
|
17
|
-
* - `error`: An error occurred during the API request, preventing successful completion.
|
|
18
|
-
*/
|
|
19
|
-
readonly status: 'submitted' | 'streaming' | 'ready' | 'error';
|
|
20
|
-
/** Current messages in the chat */
|
|
21
|
-
readonly messages: UIMessage<MESSAGE_METADATA, InferUIDataParts<DATA_PART_SCHEMAS>>[];
|
|
22
|
-
/** The error object of the API request */
|
|
23
|
-
readonly error: undefined | Error;
|
|
24
|
-
/**
|
|
25
|
-
* Append a user message to the chat list. This triggers the API call to fetch
|
|
26
|
-
* the assistant's response.
|
|
27
|
-
*
|
|
28
|
-
* @param message The message to append
|
|
29
|
-
* @param options Additional options to pass to the API call
|
|
30
|
-
*/
|
|
31
|
-
append: (message: CreateUIMessage<MESSAGE_METADATA, InferUIDataParts<DATA_PART_SCHEMAS>>, options?: ChatRequestOptions) => Promise<void>;
|
|
32
|
-
/**
|
|
33
|
-
* Reload the last AI chat response for the given chat history. If the last
|
|
34
|
-
* message isn't from the assistant, it will request the API to generate a
|
|
35
|
-
* new response.
|
|
36
|
-
*/
|
|
37
|
-
reload: (chatRequestOptions?: ChatRequestOptions) => Promise<string | null | undefined>;
|
|
38
|
-
/**
|
|
39
|
-
* Abort the current request immediately, keep the generated tokens if any.
|
|
40
|
-
*/
|
|
41
|
-
stop: () => void;
|
|
42
|
-
/**
|
|
43
|
-
* Resume an ongoing chat generation stream. This does not resume an aborted generation.
|
|
44
|
-
*/
|
|
45
|
-
experimental_resume: () => void;
|
|
15
|
+
readonly id: string;
|
|
46
16
|
/**
|
|
47
17
|
* Update the `messages` state locally. This is useful when you want to
|
|
48
18
|
* edit the messages on the client, and then trigger the `reload` method
|
|
@@ -61,18 +31,23 @@ type UseChatHelpers<MESSAGE_METADATA = unknown, DATA_PART_SCHEMAS extends UIData
|
|
|
61
31
|
}, chatRequestOptions?: ChatRequestOptions & {
|
|
62
32
|
files?: FileList | FileUIPart[];
|
|
63
33
|
}) => void;
|
|
64
|
-
|
|
65
|
-
|
|
66
|
-
|
|
67
|
-
|
|
68
|
-
}
|
|
69
|
-
|
|
34
|
+
error: Error | undefined;
|
|
35
|
+
} & Pick<AbstractChat<MESSAGE_METADATA, DATA_PART_SCHEMAS>, 'append' | 'reload' | 'stop' | 'experimental_resume' | 'addToolResult' | 'status' | 'messages'>;
|
|
36
|
+
type UseChatOptions<MESSAGE_METADATA = unknown, DATA_TYPE_SCHEMAS extends UIDataPartSchemas = UIDataPartSchemas> = ({
|
|
37
|
+
chat: Chat<MESSAGE_METADATA, DATA_TYPE_SCHEMAS>;
|
|
38
|
+
} | BaseChatInit<MESSAGE_METADATA, DATA_TYPE_SCHEMAS>) & {
|
|
39
|
+
/**
|
|
40
|
+
/**
|
|
41
|
+
* Initial input of the chat.
|
|
42
|
+
*/
|
|
43
|
+
initialInput?: string;
|
|
70
44
|
/**
|
|
71
45
|
Custom throttle wait in ms for the chat messages and data updates.
|
|
72
46
|
Default is undefined, which disables throttling.
|
|
73
47
|
*/
|
|
74
48
|
experimental_throttle?: number;
|
|
75
|
-
}
|
|
49
|
+
};
|
|
50
|
+
declare function useChat<MESSAGE_METADATA = unknown, DATA_PART_SCHEMAS extends UIDataPartSchemas = UIDataPartSchemas>({ initialInput, experimental_throttle: throttleWaitMs, ...options }?: UseChatOptions<MESSAGE_METADATA, DATA_PART_SCHEMAS>): UseChatHelpers<MESSAGE_METADATA, DATA_PART_SCHEMAS>;
|
|
76
51
|
|
|
77
52
|
type UseCompletionHelpers = {
|
|
78
53
|
/** The current completion result */
|
|
@@ -126,7 +101,7 @@ declare function useCompletion({ api, id, initialCompletion, initialInput, crede
|
|
|
126
101
|
experimental_throttle?: number;
|
|
127
102
|
}): UseCompletionHelpers;
|
|
128
103
|
|
|
129
|
-
type Experimental_UseObjectOptions<RESULT> = {
|
|
104
|
+
type Experimental_UseObjectOptions<SCHEMA extends z4.$ZodType | z3.Schema | Schema, RESULT> = {
|
|
130
105
|
/**
|
|
131
106
|
* The API endpoint. It should stream JSON that matches the schema as chunked text.
|
|
132
107
|
*/
|
|
@@ -134,7 +109,7 @@ type Experimental_UseObjectOptions<RESULT> = {
|
|
|
134
109
|
/**
|
|
135
110
|
* A Zod schema that defines the shape of the complete object.
|
|
136
111
|
*/
|
|
137
|
-
schema:
|
|
112
|
+
schema: SCHEMA;
|
|
138
113
|
/**
|
|
139
114
|
* An unique identifier. If not provided, a random one will be
|
|
140
115
|
* generated. When provided, the `useObject` hook with the same `id` will
|
|
@@ -201,10 +176,8 @@ type Experimental_UseObjectHelpers<RESULT, INPUT> = {
|
|
|
201
176
|
*/
|
|
202
177
|
stop: () => void;
|
|
203
178
|
};
|
|
204
|
-
declare function useObject<
|
|
205
|
-
initialValue, fetch, onError, onFinish, headers, credentials, }: Experimental_UseObjectOptions<RESULT>): Experimental_UseObjectHelpers<RESULT, INPUT>;
|
|
179
|
+
declare function useObject<SCHEMA extends z4.$ZodType | z3.Schema | Schema, RESULT = InferSchema<SCHEMA>, INPUT = any>({ api, id, schema, // required, in the future we will use it for validation
|
|
180
|
+
initialValue, fetch, onError, onFinish, headers, credentials, }: Experimental_UseObjectOptions<SCHEMA, RESULT>): Experimental_UseObjectHelpers<RESULT, INPUT>;
|
|
206
181
|
declare const experimental_useObject: typeof useObject;
|
|
207
182
|
|
|
208
|
-
|
|
209
|
-
|
|
210
|
-
export { Experimental_UseObjectHelpers, Experimental_UseObjectOptions, UseChatHelpers, UseCompletionHelpers, createChatStore, experimental_useObject, useChat, useCompletion };
|
|
183
|
+
export { Chat, Experimental_UseObjectHelpers, Experimental_UseObjectOptions, UseChatHelpers, UseChatOptions, UseCompletionHelpers, experimental_useObject, useChat, useCompletion };
|
package/dist/index.d.ts
CHANGED
|
@@ -1,48 +1,18 @@
|
|
|
1
|
-
import { UIDataPartSchemas, UIMessage, InferUIDataParts,
|
|
2
|
-
export { CreateUIMessage, UIMessage,
|
|
3
|
-
import { FetchFunction } from '@ai-sdk/provider-utils';
|
|
4
|
-
import
|
|
1
|
+
import { UIDataPartSchemas, AbstractChat, BaseChatInit, UIMessage, InferUIDataParts, ChatRequestOptions, FileUIPart, CompletionRequestOptions, UseCompletionOptions, Schema, DeepPartial } from 'ai';
|
|
2
|
+
export { CreateUIMessage, UIMessage, UseCompletionOptions } from 'ai';
|
|
3
|
+
import { FetchFunction, InferSchema } from '@ai-sdk/provider-utils';
|
|
4
|
+
import * as z3 from 'zod/v3';
|
|
5
|
+
import * as z4 from 'zod/v4/core';
|
|
6
|
+
|
|
7
|
+
declare class Chat<MESSAGE_METADATA, UI_DATA_PART_SCHEMAS extends UIDataPartSchemas = UIDataPartSchemas> extends AbstractChat<MESSAGE_METADATA, UI_DATA_PART_SCHEMAS> {
|
|
8
|
+
constructor({ messages, ...init }: BaseChatInit<MESSAGE_METADATA, UI_DATA_PART_SCHEMAS>);
|
|
9
|
+
}
|
|
5
10
|
|
|
6
11
|
type UseChatHelpers<MESSAGE_METADATA = unknown, DATA_PART_SCHEMAS extends UIDataPartSchemas = UIDataPartSchemas> = {
|
|
7
12
|
/**
|
|
8
13
|
* The id of the chat.
|
|
9
14
|
*/
|
|
10
|
-
readonly
|
|
11
|
-
/**
|
|
12
|
-
* Hook status:
|
|
13
|
-
*
|
|
14
|
-
* - `submitted`: The message has been sent to the API and we're awaiting the start of the response stream.
|
|
15
|
-
* - `streaming`: The response is actively streaming in from the API, receiving chunks of data.
|
|
16
|
-
* - `ready`: The full response has been received and processed; a new user message can be submitted.
|
|
17
|
-
* - `error`: An error occurred during the API request, preventing successful completion.
|
|
18
|
-
*/
|
|
19
|
-
readonly status: 'submitted' | 'streaming' | 'ready' | 'error';
|
|
20
|
-
/** Current messages in the chat */
|
|
21
|
-
readonly messages: UIMessage<MESSAGE_METADATA, InferUIDataParts<DATA_PART_SCHEMAS>>[];
|
|
22
|
-
/** The error object of the API request */
|
|
23
|
-
readonly error: undefined | Error;
|
|
24
|
-
/**
|
|
25
|
-
* Append a user message to the chat list. This triggers the API call to fetch
|
|
26
|
-
* the assistant's response.
|
|
27
|
-
*
|
|
28
|
-
* @param message The message to append
|
|
29
|
-
* @param options Additional options to pass to the API call
|
|
30
|
-
*/
|
|
31
|
-
append: (message: CreateUIMessage<MESSAGE_METADATA, InferUIDataParts<DATA_PART_SCHEMAS>>, options?: ChatRequestOptions) => Promise<void>;
|
|
32
|
-
/**
|
|
33
|
-
* Reload the last AI chat response for the given chat history. If the last
|
|
34
|
-
* message isn't from the assistant, it will request the API to generate a
|
|
35
|
-
* new response.
|
|
36
|
-
*/
|
|
37
|
-
reload: (chatRequestOptions?: ChatRequestOptions) => Promise<string | null | undefined>;
|
|
38
|
-
/**
|
|
39
|
-
* Abort the current request immediately, keep the generated tokens if any.
|
|
40
|
-
*/
|
|
41
|
-
stop: () => void;
|
|
42
|
-
/**
|
|
43
|
-
* Resume an ongoing chat generation stream. This does not resume an aborted generation.
|
|
44
|
-
*/
|
|
45
|
-
experimental_resume: () => void;
|
|
15
|
+
readonly id: string;
|
|
46
16
|
/**
|
|
47
17
|
* Update the `messages` state locally. This is useful when you want to
|
|
48
18
|
* edit the messages on the client, and then trigger the `reload` method
|
|
@@ -61,18 +31,23 @@ type UseChatHelpers<MESSAGE_METADATA = unknown, DATA_PART_SCHEMAS extends UIData
|
|
|
61
31
|
}, chatRequestOptions?: ChatRequestOptions & {
|
|
62
32
|
files?: FileList | FileUIPart[];
|
|
63
33
|
}) => void;
|
|
64
|
-
|
|
65
|
-
|
|
66
|
-
|
|
67
|
-
|
|
68
|
-
}
|
|
69
|
-
|
|
34
|
+
error: Error | undefined;
|
|
35
|
+
} & Pick<AbstractChat<MESSAGE_METADATA, DATA_PART_SCHEMAS>, 'append' | 'reload' | 'stop' | 'experimental_resume' | 'addToolResult' | 'status' | 'messages'>;
|
|
36
|
+
type UseChatOptions<MESSAGE_METADATA = unknown, DATA_TYPE_SCHEMAS extends UIDataPartSchemas = UIDataPartSchemas> = ({
|
|
37
|
+
chat: Chat<MESSAGE_METADATA, DATA_TYPE_SCHEMAS>;
|
|
38
|
+
} | BaseChatInit<MESSAGE_METADATA, DATA_TYPE_SCHEMAS>) & {
|
|
39
|
+
/**
|
|
40
|
+
/**
|
|
41
|
+
* Initial input of the chat.
|
|
42
|
+
*/
|
|
43
|
+
initialInput?: string;
|
|
70
44
|
/**
|
|
71
45
|
Custom throttle wait in ms for the chat messages and data updates.
|
|
72
46
|
Default is undefined, which disables throttling.
|
|
73
47
|
*/
|
|
74
48
|
experimental_throttle?: number;
|
|
75
|
-
}
|
|
49
|
+
};
|
|
50
|
+
declare function useChat<MESSAGE_METADATA = unknown, DATA_PART_SCHEMAS extends UIDataPartSchemas = UIDataPartSchemas>({ initialInput, experimental_throttle: throttleWaitMs, ...options }?: UseChatOptions<MESSAGE_METADATA, DATA_PART_SCHEMAS>): UseChatHelpers<MESSAGE_METADATA, DATA_PART_SCHEMAS>;
|
|
76
51
|
|
|
77
52
|
type UseCompletionHelpers = {
|
|
78
53
|
/** The current completion result */
|
|
@@ -126,7 +101,7 @@ declare function useCompletion({ api, id, initialCompletion, initialInput, crede
|
|
|
126
101
|
experimental_throttle?: number;
|
|
127
102
|
}): UseCompletionHelpers;
|
|
128
103
|
|
|
129
|
-
type Experimental_UseObjectOptions<RESULT> = {
|
|
104
|
+
type Experimental_UseObjectOptions<SCHEMA extends z4.$ZodType | z3.Schema | Schema, RESULT> = {
|
|
130
105
|
/**
|
|
131
106
|
* The API endpoint. It should stream JSON that matches the schema as chunked text.
|
|
132
107
|
*/
|
|
@@ -134,7 +109,7 @@ type Experimental_UseObjectOptions<RESULT> = {
|
|
|
134
109
|
/**
|
|
135
110
|
* A Zod schema that defines the shape of the complete object.
|
|
136
111
|
*/
|
|
137
|
-
schema:
|
|
112
|
+
schema: SCHEMA;
|
|
138
113
|
/**
|
|
139
114
|
* An unique identifier. If not provided, a random one will be
|
|
140
115
|
* generated. When provided, the `useObject` hook with the same `id` will
|
|
@@ -201,10 +176,8 @@ type Experimental_UseObjectHelpers<RESULT, INPUT> = {
|
|
|
201
176
|
*/
|
|
202
177
|
stop: () => void;
|
|
203
178
|
};
|
|
204
|
-
declare function useObject<
|
|
205
|
-
initialValue, fetch, onError, onFinish, headers, credentials, }: Experimental_UseObjectOptions<RESULT>): Experimental_UseObjectHelpers<RESULT, INPUT>;
|
|
179
|
+
declare function useObject<SCHEMA extends z4.$ZodType | z3.Schema | Schema, RESULT = InferSchema<SCHEMA>, INPUT = any>({ api, id, schema, // required, in the future we will use it for validation
|
|
180
|
+
initialValue, fetch, onError, onFinish, headers, credentials, }: Experimental_UseObjectOptions<SCHEMA, RESULT>): Experimental_UseObjectHelpers<RESULT, INPUT>;
|
|
206
181
|
declare const experimental_useObject: typeof useObject;
|
|
207
182
|
|
|
208
|
-
|
|
209
|
-
|
|
210
|
-
export { Experimental_UseObjectHelpers, Experimental_UseObjectOptions, UseChatHelpers, UseCompletionHelpers, createChatStore, experimental_useObject, useChat, useCompletion };
|
|
183
|
+
export { Chat, Experimental_UseObjectHelpers, Experimental_UseObjectOptions, UseChatHelpers, UseChatOptions, UseCompletionHelpers, experimental_useObject, useChat, useCompletion };
|
package/dist/index.js
CHANGED
|
@@ -26,11 +26,29 @@ var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__ge
|
|
|
26
26
|
mod
|
|
27
27
|
));
|
|
28
28
|
var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
|
|
29
|
+
var __accessCheck = (obj, member, msg) => {
|
|
30
|
+
if (!member.has(obj))
|
|
31
|
+
throw TypeError("Cannot " + msg);
|
|
32
|
+
};
|
|
33
|
+
var __privateGet = (obj, member, getter) => {
|
|
34
|
+
__accessCheck(obj, member, "read from private field");
|
|
35
|
+
return getter ? getter.call(obj) : member.get(obj);
|
|
36
|
+
};
|
|
37
|
+
var __privateAdd = (obj, member, value) => {
|
|
38
|
+
if (member.has(obj))
|
|
39
|
+
throw TypeError("Cannot add the same private member more than once");
|
|
40
|
+
member instanceof WeakSet ? member.add(obj) : member.set(obj, value);
|
|
41
|
+
};
|
|
42
|
+
var __privateSet = (obj, member, value, setter) => {
|
|
43
|
+
__accessCheck(obj, member, "write to private field");
|
|
44
|
+
setter ? setter.call(obj, value) : member.set(obj, value);
|
|
45
|
+
return value;
|
|
46
|
+
};
|
|
29
47
|
|
|
30
48
|
// src/index.ts
|
|
31
49
|
var src_exports = {};
|
|
32
50
|
__export(src_exports, {
|
|
33
|
-
|
|
51
|
+
Chat: () => Chat,
|
|
34
52
|
experimental_useObject: () => experimental_useObject,
|
|
35
53
|
useChat: () => useChat,
|
|
36
54
|
useCompletion: () => useCompletion
|
|
@@ -41,53 +59,46 @@ module.exports = __toCommonJS(src_exports);
|
|
|
41
59
|
var import_ai2 = require("ai");
|
|
42
60
|
var import_react = require("react");
|
|
43
61
|
|
|
44
|
-
// src/chat
|
|
62
|
+
// src/chat.react.ts
|
|
45
63
|
var import_ai = require("ai");
|
|
46
|
-
var
|
|
47
|
-
|
|
64
|
+
var _messages;
|
|
65
|
+
var ReactChatState = class {
|
|
66
|
+
constructor(messages = []) {
|
|
67
|
+
__privateAdd(this, _messages, void 0);
|
|
48
68
|
this.status = "ready";
|
|
49
69
|
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
70
|
this.pushMessage = (message) => {
|
|
65
|
-
this
|
|
71
|
+
__privateSet(this, _messages, this.messages.concat(message));
|
|
66
72
|
};
|
|
67
73
|
this.popMessage = () => {
|
|
68
|
-
this
|
|
74
|
+
__privateSet(this, _messages, this.messages.slice(0, -1));
|
|
69
75
|
};
|
|
70
76
|
this.replaceMessage = (index, message) => {
|
|
71
|
-
this
|
|
72
|
-
...this.
|
|
77
|
+
__privateSet(this, _messages, [
|
|
78
|
+
...__privateGet(this, _messages).slice(0, index),
|
|
73
79
|
message,
|
|
74
|
-
...this.
|
|
75
|
-
];
|
|
76
|
-
};
|
|
77
|
-
this.snapshot = (value) => {
|
|
78
|
-
return structuredClone(value);
|
|
80
|
+
...__privateGet(this, _messages).slice(index + 1)
|
|
81
|
+
]);
|
|
79
82
|
};
|
|
80
|
-
this.
|
|
83
|
+
this.snapshot = (value) => structuredClone(value);
|
|
84
|
+
__privateSet(this, _messages, messages);
|
|
85
|
+
}
|
|
86
|
+
get messages() {
|
|
87
|
+
return __privateGet(this, _messages);
|
|
88
|
+
}
|
|
89
|
+
set messages(messages) {
|
|
90
|
+
__privateSet(this, _messages, [...messages]);
|
|
91
|
+
}
|
|
92
|
+
};
|
|
93
|
+
_messages = new WeakMap();
|
|
94
|
+
var Chat = class extends import_ai.AbstractChat {
|
|
95
|
+
constructor({
|
|
96
|
+
messages,
|
|
97
|
+
...init
|
|
98
|
+
}) {
|
|
99
|
+
super({ ...init, state: new ReactChatState(messages) });
|
|
81
100
|
}
|
|
82
101
|
};
|
|
83
|
-
function createChatStore(options) {
|
|
84
|
-
return new import_ai.ChatStore({
|
|
85
|
-
...options,
|
|
86
|
-
createChat: (options2) => new ReactChat(
|
|
87
|
-
options2.messages
|
|
88
|
-
)
|
|
89
|
-
});
|
|
90
|
-
}
|
|
91
102
|
|
|
92
103
|
// src/throttle.ts
|
|
93
104
|
var import_throttleit = __toESM(require("throttleit"));
|
|
@@ -97,161 +108,95 @@ function throttle(fn, waitMs) {
|
|
|
97
108
|
|
|
98
109
|
// src/use-chat.ts
|
|
99
110
|
function useChat({
|
|
100
|
-
chatId,
|
|
101
111
|
initialInput = "",
|
|
102
|
-
onToolCall,
|
|
103
|
-
onFinish,
|
|
104
|
-
onError,
|
|
105
|
-
generateId = import_ai2.generateId,
|
|
106
112
|
experimental_throttle: throttleWaitMs,
|
|
107
|
-
|
|
113
|
+
...options
|
|
108
114
|
} = {}) {
|
|
109
|
-
const
|
|
110
|
-
const stableChatId = chatId != null ? chatId : hookId;
|
|
111
|
-
const chatStore = (0, import_react.useRef)(
|
|
112
|
-
chatStoreArg == null ? createChatStore(
|
|
113
|
-
(0, import_ai2.defaultChatStoreOptions)({
|
|
114
|
-
api: "/api/chat",
|
|
115
|
-
generateId
|
|
116
|
-
})()
|
|
117
|
-
) : typeof chatStoreArg === "function" ? createChatStore(chatStoreArg()) : chatStoreArg
|
|
118
|
-
);
|
|
119
|
-
if (!chatStore.current.hasChat(stableChatId)) {
|
|
120
|
-
chatStore.current.addChat(stableChatId, []);
|
|
121
|
-
}
|
|
115
|
+
const chatRef = (0, import_react.useRef)("chat" in options ? options.chat : new Chat(options));
|
|
122
116
|
const subscribe = (0, import_react.useCallback)(
|
|
123
117
|
({
|
|
124
118
|
onStoreChange,
|
|
125
119
|
eventType
|
|
126
|
-
}) => {
|
|
127
|
-
|
|
128
|
-
|
|
129
|
-
|
|
130
|
-
|
|
131
|
-
|
|
132
|
-
|
|
133
|
-
|
|
134
|
-
},
|
|
135
|
-
[chatStore, stableChatId]
|
|
120
|
+
}) => chatRef.current.subscribe({
|
|
121
|
+
onChange: (event) => {
|
|
122
|
+
if (event.type !== eventType)
|
|
123
|
+
return;
|
|
124
|
+
onStoreChange();
|
|
125
|
+
}
|
|
126
|
+
}),
|
|
127
|
+
[chatRef]
|
|
136
128
|
);
|
|
137
129
|
const addToolResult = (0, import_react.useCallback)(
|
|
138
|
-
(
|
|
139
|
-
[
|
|
140
|
-
);
|
|
141
|
-
const stopStream = (0, import_react.useCallback)(() => {
|
|
142
|
-
chatStore.current.stopStream({ chatId: stableChatId });
|
|
143
|
-
}, [chatStore, stableChatId]);
|
|
144
|
-
const error = (0, import_react.useSyncExternalStore)(
|
|
145
|
-
(callback) => subscribe({
|
|
146
|
-
onStoreChange: callback,
|
|
147
|
-
eventType: "chat-status-changed"
|
|
148
|
-
}),
|
|
149
|
-
() => chatStore.current.getError(stableChatId),
|
|
150
|
-
() => chatStore.current.getError(stableChatId)
|
|
130
|
+
(options2) => chatRef.current.addToolResult(options2),
|
|
131
|
+
[chatRef]
|
|
151
132
|
);
|
|
152
133
|
const status = (0, import_react.useSyncExternalStore)(
|
|
153
134
|
(callback) => subscribe({
|
|
154
135
|
onStoreChange: callback,
|
|
155
|
-
eventType: "
|
|
136
|
+
eventType: "status-changed"
|
|
156
137
|
}),
|
|
157
|
-
() =>
|
|
158
|
-
() =>
|
|
138
|
+
() => chatRef.current.status,
|
|
139
|
+
() => chatRef.current.status
|
|
159
140
|
);
|
|
160
141
|
const subscribeToChatStoreForMessages = (0, import_react.useCallback)(
|
|
161
142
|
(callback) => {
|
|
162
143
|
return subscribe({
|
|
163
144
|
onStoreChange: throttleWaitMs ? throttle(callback, throttleWaitMs) : callback,
|
|
164
|
-
eventType: "
|
|
145
|
+
eventType: "messages-changed"
|
|
165
146
|
});
|
|
166
147
|
},
|
|
167
148
|
[subscribe, throttleWaitMs]
|
|
168
149
|
);
|
|
169
150
|
const messages = (0, import_react.useSyncExternalStore)(
|
|
170
151
|
(callback) => subscribeToChatStoreForMessages(callback),
|
|
171
|
-
() =>
|
|
172
|
-
() =>
|
|
173
|
-
);
|
|
174
|
-
const append = (0, import_react.useCallback)(
|
|
175
|
-
(message, { headers, body } = {}) => chatStore.current.submitMessage({
|
|
176
|
-
chatId: stableChatId,
|
|
177
|
-
message,
|
|
178
|
-
headers,
|
|
179
|
-
body,
|
|
180
|
-
onError,
|
|
181
|
-
onToolCall,
|
|
182
|
-
onFinish
|
|
183
|
-
}),
|
|
184
|
-
[chatStore, stableChatId, onError, onToolCall, onFinish]
|
|
185
|
-
);
|
|
186
|
-
const reload = (0, import_react.useCallback)(
|
|
187
|
-
async ({ headers, body } = {}) => chatStore.current.resubmitLastUserMessage({
|
|
188
|
-
chatId: stableChatId,
|
|
189
|
-
headers,
|
|
190
|
-
body,
|
|
191
|
-
onError,
|
|
192
|
-
onToolCall,
|
|
193
|
-
onFinish
|
|
194
|
-
}),
|
|
195
|
-
[chatStore, stableChatId, onError, onToolCall, onFinish]
|
|
196
|
-
);
|
|
197
|
-
const stop = (0, import_react.useCallback)(() => stopStream(), [stopStream]);
|
|
198
|
-
const experimental_resume = (0, import_react.useCallback)(
|
|
199
|
-
async () => chatStore.current.resumeStream({
|
|
200
|
-
chatId: stableChatId,
|
|
201
|
-
onError,
|
|
202
|
-
onToolCall,
|
|
203
|
-
onFinish
|
|
204
|
-
}),
|
|
205
|
-
[chatStore, stableChatId, onError, onToolCall, onFinish]
|
|
152
|
+
() => chatRef.current.messages,
|
|
153
|
+
() => chatRef.current.messages
|
|
206
154
|
);
|
|
207
155
|
const setMessages = (0, import_react.useCallback)(
|
|
208
156
|
(messagesParam) => {
|
|
209
157
|
if (typeof messagesParam === "function") {
|
|
210
158
|
messagesParam = messagesParam(messages);
|
|
211
159
|
}
|
|
212
|
-
|
|
213
|
-
id: stableChatId,
|
|
214
|
-
messages: messagesParam
|
|
215
|
-
});
|
|
160
|
+
chatRef.current.messages = messagesParam;
|
|
216
161
|
},
|
|
217
|
-
[
|
|
162
|
+
[chatRef, messages]
|
|
218
163
|
);
|
|
219
164
|
const [input, setInput] = (0, import_react.useState)(initialInput);
|
|
220
165
|
const handleSubmit = (0, import_react.useCallback)(
|
|
221
|
-
async (event,
|
|
166
|
+
async (event, options2 = {}) => {
|
|
222
167
|
var _a;
|
|
223
168
|
(_a = event == null ? void 0 : event.preventDefault) == null ? void 0 : _a.call(event);
|
|
224
|
-
const fileParts = Array.isArray(
|
|
169
|
+
const fileParts = Array.isArray(options2 == null ? void 0 : options2.files) ? options2.files : await (0, import_ai2.convertFileListToFileUIParts)(options2 == null ? void 0 : options2.files);
|
|
225
170
|
if (!input && fileParts.length === 0)
|
|
226
171
|
return;
|
|
227
|
-
append(
|
|
172
|
+
chatRef.current.append(
|
|
228
173
|
{
|
|
229
|
-
id: generateId(),
|
|
174
|
+
id: chatRef.current.generateId(),
|
|
230
175
|
role: "user",
|
|
231
176
|
metadata: void 0,
|
|
232
177
|
parts: [...fileParts, { type: "text", text: input }]
|
|
233
178
|
},
|
|
234
179
|
{
|
|
235
|
-
headers:
|
|
236
|
-
body:
|
|
180
|
+
headers: options2.headers,
|
|
181
|
+
body: options2.body
|
|
237
182
|
}
|
|
238
183
|
);
|
|
239
184
|
setInput("");
|
|
240
185
|
},
|
|
241
|
-
[input,
|
|
186
|
+
[input, chatRef]
|
|
242
187
|
);
|
|
243
188
|
const handleInputChange = (e) => {
|
|
244
189
|
setInput(e.target.value);
|
|
245
190
|
};
|
|
246
191
|
return {
|
|
192
|
+
id: chatRef.current.id,
|
|
247
193
|
messages,
|
|
248
|
-
chatId: stableChatId,
|
|
249
194
|
setMessages,
|
|
250
|
-
|
|
251
|
-
|
|
252
|
-
|
|
253
|
-
|
|
254
|
-
experimental_resume,
|
|
195
|
+
append: chatRef.current.append,
|
|
196
|
+
reload: chatRef.current.reload,
|
|
197
|
+
stop: chatRef.current.stop,
|
|
198
|
+
error: chatRef.current.error,
|
|
199
|
+
experimental_resume: chatRef.current.experimental_resume,
|
|
255
200
|
input,
|
|
256
201
|
setInput,
|
|
257
202
|
handleInputChange,
|
|
@@ -502,7 +447,7 @@ function useObject({
|
|
|
502
447
|
var experimental_useObject = useObject;
|
|
503
448
|
// Annotate the CommonJS export names for ESM import in node:
|
|
504
449
|
0 && (module.exports = {
|
|
505
|
-
|
|
450
|
+
Chat,
|
|
506
451
|
experimental_useObject,
|
|
507
452
|
useChat,
|
|
508
453
|
useCompletion
|
package/dist/index.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
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"]}
|
|
1
|
+
{"version":3,"sources":["../src/index.ts","../src/use-chat.ts","../src/chat.react.ts","../src/throttle.ts","../src/use-completion.ts","../src/use-object.ts"],"sourcesContent":["export * from './use-chat';\nexport { Chat } from './chat.react';\nexport * from './use-completion';\nexport * from './use-object';\n","import {\n AbstractChat,\n BaseChatInit,\n ChatEvent,\n convertFileListToFileUIParts,\n InferUIDataParts,\n UIDataPartSchemas,\n type ChatRequestOptions,\n type CreateUIMessage,\n type FileUIPart,\n type UIMessage,\n} from 'ai';\nimport { useCallback, useRef, useState, useSyncExternalStore } from 'react';\nimport { Chat } from './chat.react';\nimport { throttle } from './throttle';\n\nexport type { CreateUIMessage, UIMessage };\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 id: string;\n\n /**\n * Update the `messages` state locally. This is useful when you want to\n * edit the messages on the client, and then trigger the `reload` method\n * manually to regenerate the AI response.\n */\n setMessages: (\n messages:\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 error: Error | undefined;\n} & Pick<\n AbstractChat<MESSAGE_METADATA, DATA_PART_SCHEMAS>,\n | 'append'\n | 'reload'\n | 'stop'\n | 'experimental_resume'\n | 'addToolResult'\n | 'status'\n | 'messages'\n>;\n\nexport type UseChatOptions<\n MESSAGE_METADATA = unknown,\n DATA_TYPE_SCHEMAS extends UIDataPartSchemas = UIDataPartSchemas,\n> = (\n | { chat: Chat<MESSAGE_METADATA, DATA_TYPE_SCHEMAS> }\n | BaseChatInit<MESSAGE_METADATA, DATA_TYPE_SCHEMAS>\n) & {\n /**\n /**\n * Initial input of the chat.\n */\n initialInput?: string;\n\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};\n\nexport function useChat<\n MESSAGE_METADATA = unknown,\n DATA_PART_SCHEMAS extends UIDataPartSchemas = UIDataPartSchemas,\n>({\n initialInput = '',\n experimental_throttle: throttleWaitMs,\n ...options\n}: UseChatOptions<MESSAGE_METADATA, DATA_PART_SCHEMAS> = {}): UseChatHelpers<\n MESSAGE_METADATA,\n DATA_PART_SCHEMAS\n> {\n const chatRef = useRef('chat' in options ? options.chat : new Chat(options));\n\n const subscribe = useCallback(\n ({\n onStoreChange,\n eventType,\n }: {\n onStoreChange: () => void;\n eventType: ChatEvent['type'];\n }) =>\n chatRef.current.subscribe({\n onChange: event => {\n if (event.type !== eventType) return;\n onStoreChange();\n },\n }),\n [chatRef],\n );\n\n const addToolResult = useCallback(\n (\n options: Parameters<\n Chat<MESSAGE_METADATA, DATA_PART_SCHEMAS>['addToolResult']\n >[0],\n ) => chatRef.current.addToolResult(options),\n [chatRef],\n );\n\n const status = useSyncExternalStore(\n callback =>\n subscribe({\n onStoreChange: callback,\n eventType: 'status-changed',\n }),\n () => chatRef.current.status,\n () => chatRef.current.status,\n );\n\n const subscribeToChatStoreForMessages = useCallback(\n (callback: () => void) => {\n return subscribe({\n onStoreChange: throttleWaitMs\n ? throttle(callback, throttleWaitMs)\n : callback,\n eventType: 'messages-changed',\n });\n },\n [subscribe, throttleWaitMs],\n );\n\n const messages = useSyncExternalStore(\n callback => subscribeToChatStoreForMessages(callback),\n () => chatRef.current.messages,\n () => chatRef.current.messages,\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 chatRef.current.messages = messagesParam;\n },\n [chatRef, 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 chatRef.current.append(\n {\n id: chatRef.current.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, chatRef],\n );\n\n const handleInputChange = (e: any) => {\n setInput(e.target.value);\n };\n\n return {\n id: chatRef.current.id,\n messages,\n setMessages,\n append: chatRef.current.append,\n reload: chatRef.current.reload,\n stop: chatRef.current.stop,\n error: chatRef.current.error,\n experimental_resume: chatRef.current.experimental_resume,\n input,\n setInput,\n handleInputChange,\n handleSubmit,\n status,\n addToolResult,\n };\n}\n","import {\n AbstractChat,\n BaseChatInit,\n ChatState,\n ChatStatus,\n UIDataPartSchemas,\n UIDataTypes,\n UIMessage,\n} from 'ai';\n\nclass ReactChatState<MESSAGE_METADATA, DATA_TYPES extends UIDataTypes>\n implements ChatState<MESSAGE_METADATA, DATA_TYPES>\n{\n #messages: UIMessage<MESSAGE_METADATA, DATA_TYPES>[];\n status: ChatStatus = 'ready';\n error: Error | undefined = undefined;\n\n constructor(messages: UIMessage<MESSAGE_METADATA, DATA_TYPES>[] = []) {\n this.#messages = messages;\n }\n\n get messages() {\n return this.#messages;\n }\n\n set messages(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 => structuredClone(value);\n}\n\nexport class Chat<\n MESSAGE_METADATA,\n UI_DATA_PART_SCHEMAS extends UIDataPartSchemas = UIDataPartSchemas,\n> extends AbstractChat<MESSAGE_METADATA, UI_DATA_PART_SCHEMAS> {\n constructor({\n messages,\n ...init\n }: BaseChatInit<MESSAGE_METADATA, UI_DATA_PART_SCHEMAS>) {\n super({ ...init, state: new ReactChatState(messages) });\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 InferSchema,\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 * as z3 from 'zod/v3';\nimport * as z4 from 'zod/v4/core';\n\n// use function to allow for mocking in tests:\nconst getOriginalFetch = () => fetch;\n\nexport type Experimental_UseObjectOptions<\n SCHEMA extends z4.$ZodType | z3.Schema | Schema,\n RESULT,\n> = {\n /**\n * The API endpoint. It should stream JSON that matches the schema as chunked text.\n */\n api: string;\n\n /**\n * A Zod schema that defines the shape of the complete object.\n */\n schema: SCHEMA;\n\n /**\n * An unique identifier. If not provided, a random one will be\n * generated. When provided, the `useObject` hook with the same `id` will\n * have shared states across components.\n */\n id?: string;\n\n /**\n * An optional value for the initial object.\n */\n initialValue?: DeepPartial<RESULT>;\n\n /**\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<\n SCHEMA extends z4.$ZodType | z3.Schema | Schema,\n RESULT = InferSchema<SCHEMA>,\n INPUT = any,\n>({\n api,\n id,\n schema, // required, in the future we will use it for validation\n initialValue,\n fetch,\n onError,\n onFinish,\n headers,\n credentials,\n}: Experimental_UseObjectOptions<\n SCHEMA,\n RESULT\n>): Experimental_UseObjectHelpers<RESULT, INPUT> {\n // Generate an unique id if not provided.\n const hookId = useId();\n const completionId = id ?? hookId;\n\n // Store the completion state in SWR, using the completionId as the key to share states.\n const { data, mutate } = useSWR<DeepPartial<RESULT>>(\n [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,aAWO;AACP,mBAAoE;;;ACZpE,gBAQO;AARP;AAUA,IAAM,iBAAN,MAEA;AAAA,EAKE,YAAY,WAAsD,CAAC,GAAG;AAJtE;AACA,kBAAqB;AACrB,iBAA2B;AAc3B,uBAAc,CAAC,YAAqD;AAClE,yBAAK,WAAY,KAAK,SAAS,OAAO,OAAO;AAAA,IAC/C;AAEA,sBAAa,MAAM;AACjB,yBAAK,WAAY,KAAK,SAAS,MAAM,GAAG,EAAE;AAAA,IAC5C;AAEA,0BAAiB,CACf,OACA,YACG;AACH,yBAAK,WAAY;AAAA,QACf,GAAG,mBAAK,WAAU,MAAM,GAAG,KAAK;AAAA,QAChC;AAAA,QACA,GAAG,mBAAK,WAAU,MAAM,QAAQ,CAAC;AAAA,MACnC;AAAA,IACF;AAEA,oBAAW,CAAI,UAAgB,gBAAgB,KAAK;AA9BlD,uBAAK,WAAY;AAAA,EACnB;AAAA,EAEA,IAAI,WAAW;AACb,WAAO,mBAAK;AAAA,EACd;AAAA,EAEA,IAAI,SAAS,UAAqD;AAChE,uBAAK,WAAY,CAAC,GAAG,QAAQ;AAAA,EAC/B;AAsBF;AApCE;AAsCK,IAAM,OAAN,cAGG,uBAAqD;AAAA,EAC7D,YAAY;AAAA,IACV;AAAA,IACA,GAAG;AAAA,EACL,GAAyD;AACvD,UAAM,EAAE,GAAG,MAAM,OAAO,IAAI,eAAe,QAAQ,EAAE,CAAC;AAAA,EACxD;AACF;;;AC7DA,wBAA6B;AAEtB,SAAS,SACd,IACA,QACG;AACH,SAAO,UAAU,WAAO,kBAAAC,SAAiB,IAAI,MAAM,IAAI;AACzD;;;AF4FO,SAAS,QAGd;AAAA,EACA,eAAe;AAAA,EACf,uBAAuB;AAAA,EACvB,GAAG;AACL,IAAyD,CAAC,GAGxD;AACA,QAAM,cAAU,qBAAO,UAAU,UAAU,QAAQ,OAAO,IAAI,KAAK,OAAO,CAAC;AAE3E,QAAM,gBAAY;AAAA,IAChB,CAAC;AAAA,MACC;AAAA,MACA;AAAA,IACF,MAIE,QAAQ,QAAQ,UAAU;AAAA,MACxB,UAAU,WAAS;AACjB,YAAI,MAAM,SAAS;AAAW;AAC9B,sBAAc;AAAA,MAChB;AAAA,IACF,CAAC;AAAA,IACH,CAAC,OAAO;AAAA,EACV;AAEA,QAAM,oBAAgB;AAAA,IACpB,CACEC,aAGG,QAAQ,QAAQ,cAAcA,QAAO;AAAA,IAC1C,CAAC,OAAO;AAAA,EACV;AAEA,QAAM,aAAS;AAAA,IACb,cACE,UAAU;AAAA,MACR,eAAe;AAAA,MACf,WAAW;AAAA,IACb,CAAC;AAAA,IACH,MAAM,QAAQ,QAAQ;AAAA,IACtB,MAAM,QAAQ,QAAQ;AAAA,EACxB;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,QAAQ,QAAQ;AAAA,IACtB,MAAM,QAAQ,QAAQ;AAAA,EACxB;AAEA,QAAM,kBAAc;AAAA,IAClB,CACE,kBAWG;AACH,UAAI,OAAO,kBAAkB,YAAY;AACvC,wBAAgB,cAAc,QAAQ;AAAA,MACxC;AAEA,cAAQ,QAAQ,WAAW;AAAA,IAC7B;AAAA,IACA,CAAC,SAAS,QAAQ;AAAA,EACpB;AAGA,QAAM,CAAC,OAAO,QAAQ,QAAI,uBAAS,YAAY;AAE/C,QAAM,mBAAe;AAAA,IACnB,OACE,OACAA,WAEI,CAAC,MACF;AAtMT;AAuMM,2CAAO,mBAAP;AAEA,YAAM,YAAY,MAAM,QAAQA,YAAA,gBAAAA,SAAS,KAAK,IAC1CA,SAAQ,QACR,UAAM,yCAA6BA,YAAA,gBAAAA,SAAS,KAAK;AAErD,UAAI,CAAC,SAAS,UAAU,WAAW;AAAG;AAEtC,cAAQ,QAAQ;AAAA,QACd;AAAA,UACE,IAAI,QAAQ,QAAQ,WAAW;AAAA,UAC/B,MAAM;AAAA,UACN,UAAU;AAAA,UACV,OAAO,CAAC,GAAG,WAAW,EAAE,MAAM,QAAQ,MAAM,MAAM,CAAC;AAAA,QACrD;AAAA,QACA;AAAA,UACE,SAASA,SAAQ;AAAA,UACjB,MAAMA,SAAQ;AAAA,QAChB;AAAA,MACF;AAEA,eAAS,EAAE;AAAA,IACb;AAAA,IACA,CAAC,OAAO,OAAO;AAAA,EACjB;AAEA,QAAM,oBAAoB,CAAC,MAAW;AACpC,aAAS,EAAE,OAAO,KAAK;AAAA,EACzB;AAEA,SAAO;AAAA,IACL,IAAI,QAAQ,QAAQ;AAAA,IACpB;AAAA,IACA;AAAA,IACA,QAAQ,QAAQ,QAAQ;AAAA,IACxB,QAAQ,QAAQ,QAAQ;AAAA,IACxB,MAAM,QAAQ,QAAQ;AAAA,IACtB,OAAO,QAAQ,QAAQ;AAAA,IACvB,qBAAqB,QAAQ,QAAQ;AAAA,IACrC;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF;AACF;;;AGrPA,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,4BAKO;AACP,IAAAC,aAMO;AACP,IAAAC,gBAAqD;AACrD,IAAAC,cAAmB;AAKnB,IAAM,mBAAmB,MAAM;AA+F/B,SAAS,UAIP;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA;AAAA,EACA;AAAA,EACA,OAAAC;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AACF,GAGiD;AAE/C,QAAM,aAAS,qBAAM;AACrB,QAAM,eAAe,kBAAM;AAG3B,QAAM,EAAE,MAAM,OAAO,QAAI,YAAAC;AAAA,IACvB,CAAC,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;AArJjC;AAsJI,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;AA/JzC;AAgKI,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","throttleFunction","options","import_ai","import_react","fetch","useSWR","completion","import_ai","import_react","import_swr","fetch","useSWR","error"]}
|
package/dist/index.mjs
CHANGED
|
@@ -1,61 +1,70 @@
|
|
|
1
|
+
var __accessCheck = (obj, member, msg) => {
|
|
2
|
+
if (!member.has(obj))
|
|
3
|
+
throw TypeError("Cannot " + msg);
|
|
4
|
+
};
|
|
5
|
+
var __privateGet = (obj, member, getter) => {
|
|
6
|
+
__accessCheck(obj, member, "read from private field");
|
|
7
|
+
return getter ? getter.call(obj) : member.get(obj);
|
|
8
|
+
};
|
|
9
|
+
var __privateAdd = (obj, member, value) => {
|
|
10
|
+
if (member.has(obj))
|
|
11
|
+
throw TypeError("Cannot add the same private member more than once");
|
|
12
|
+
member instanceof WeakSet ? member.add(obj) : member.set(obj, value);
|
|
13
|
+
};
|
|
14
|
+
var __privateSet = (obj, member, value, setter) => {
|
|
15
|
+
__accessCheck(obj, member, "write to private field");
|
|
16
|
+
setter ? setter.call(obj, value) : member.set(obj, value);
|
|
17
|
+
return value;
|
|
18
|
+
};
|
|
19
|
+
|
|
1
20
|
// src/use-chat.ts
|
|
2
21
|
import {
|
|
3
|
-
convertFileListToFileUIParts
|
|
4
|
-
defaultChatStoreOptions,
|
|
5
|
-
generateId as generateIdFunc
|
|
22
|
+
convertFileListToFileUIParts
|
|
6
23
|
} from "ai";
|
|
7
24
|
import { useCallback, useRef, useState, useSyncExternalStore } from "react";
|
|
8
25
|
|
|
9
|
-
// src/chat
|
|
26
|
+
// src/chat.react.ts
|
|
10
27
|
import {
|
|
11
|
-
|
|
12
|
-
SerialJobExecutor
|
|
28
|
+
AbstractChat
|
|
13
29
|
} from "ai";
|
|
14
|
-
var
|
|
15
|
-
|
|
30
|
+
var _messages;
|
|
31
|
+
var ReactChatState = class {
|
|
32
|
+
constructor(messages = []) {
|
|
33
|
+
__privateAdd(this, _messages, void 0);
|
|
16
34
|
this.status = "ready";
|
|
17
35
|
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
36
|
this.pushMessage = (message) => {
|
|
33
|
-
this
|
|
37
|
+
__privateSet(this, _messages, this.messages.concat(message));
|
|
34
38
|
};
|
|
35
39
|
this.popMessage = () => {
|
|
36
|
-
this
|
|
40
|
+
__privateSet(this, _messages, this.messages.slice(0, -1));
|
|
37
41
|
};
|
|
38
42
|
this.replaceMessage = (index, message) => {
|
|
39
|
-
this
|
|
40
|
-
...this.
|
|
43
|
+
__privateSet(this, _messages, [
|
|
44
|
+
...__privateGet(this, _messages).slice(0, index),
|
|
41
45
|
message,
|
|
42
|
-
...this.
|
|
43
|
-
];
|
|
46
|
+
...__privateGet(this, _messages).slice(index + 1)
|
|
47
|
+
]);
|
|
44
48
|
};
|
|
45
|
-
this.snapshot = (value) =>
|
|
46
|
-
|
|
47
|
-
|
|
48
|
-
|
|
49
|
+
this.snapshot = (value) => structuredClone(value);
|
|
50
|
+
__privateSet(this, _messages, messages);
|
|
51
|
+
}
|
|
52
|
+
get messages() {
|
|
53
|
+
return __privateGet(this, _messages);
|
|
54
|
+
}
|
|
55
|
+
set messages(messages) {
|
|
56
|
+
__privateSet(this, _messages, [...messages]);
|
|
57
|
+
}
|
|
58
|
+
};
|
|
59
|
+
_messages = new WeakMap();
|
|
60
|
+
var Chat = class extends AbstractChat {
|
|
61
|
+
constructor({
|
|
62
|
+
messages,
|
|
63
|
+
...init
|
|
64
|
+
}) {
|
|
65
|
+
super({ ...init, state: new ReactChatState(messages) });
|
|
49
66
|
}
|
|
50
67
|
};
|
|
51
|
-
function createChatStore(options) {
|
|
52
|
-
return new ChatStore({
|
|
53
|
-
...options,
|
|
54
|
-
createChat: (options2) => new ReactChat(
|
|
55
|
-
options2.messages
|
|
56
|
-
)
|
|
57
|
-
});
|
|
58
|
-
}
|
|
59
68
|
|
|
60
69
|
// src/throttle.ts
|
|
61
70
|
import throttleFunction from "throttleit";
|
|
@@ -65,161 +74,95 @@ function throttle(fn, waitMs) {
|
|
|
65
74
|
|
|
66
75
|
// src/use-chat.ts
|
|
67
76
|
function useChat({
|
|
68
|
-
chatId,
|
|
69
77
|
initialInput = "",
|
|
70
|
-
onToolCall,
|
|
71
|
-
onFinish,
|
|
72
|
-
onError,
|
|
73
|
-
generateId = generateIdFunc,
|
|
74
78
|
experimental_throttle: throttleWaitMs,
|
|
75
|
-
|
|
79
|
+
...options
|
|
76
80
|
} = {}) {
|
|
77
|
-
const
|
|
78
|
-
const stableChatId = chatId != null ? chatId : hookId;
|
|
79
|
-
const chatStore = useRef(
|
|
80
|
-
chatStoreArg == null ? createChatStore(
|
|
81
|
-
defaultChatStoreOptions({
|
|
82
|
-
api: "/api/chat",
|
|
83
|
-
generateId
|
|
84
|
-
})()
|
|
85
|
-
) : typeof chatStoreArg === "function" ? createChatStore(chatStoreArg()) : chatStoreArg
|
|
86
|
-
);
|
|
87
|
-
if (!chatStore.current.hasChat(stableChatId)) {
|
|
88
|
-
chatStore.current.addChat(stableChatId, []);
|
|
89
|
-
}
|
|
81
|
+
const chatRef = useRef("chat" in options ? options.chat : new Chat(options));
|
|
90
82
|
const subscribe = useCallback(
|
|
91
83
|
({
|
|
92
84
|
onStoreChange,
|
|
93
85
|
eventType
|
|
94
|
-
}) => {
|
|
95
|
-
|
|
96
|
-
|
|
97
|
-
|
|
98
|
-
|
|
99
|
-
|
|
100
|
-
|
|
101
|
-
|
|
102
|
-
},
|
|
103
|
-
[chatStore, stableChatId]
|
|
86
|
+
}) => chatRef.current.subscribe({
|
|
87
|
+
onChange: (event) => {
|
|
88
|
+
if (event.type !== eventType)
|
|
89
|
+
return;
|
|
90
|
+
onStoreChange();
|
|
91
|
+
}
|
|
92
|
+
}),
|
|
93
|
+
[chatRef]
|
|
104
94
|
);
|
|
105
95
|
const addToolResult = useCallback(
|
|
106
|
-
(
|
|
107
|
-
[
|
|
108
|
-
);
|
|
109
|
-
const stopStream = useCallback(() => {
|
|
110
|
-
chatStore.current.stopStream({ chatId: stableChatId });
|
|
111
|
-
}, [chatStore, stableChatId]);
|
|
112
|
-
const error = useSyncExternalStore(
|
|
113
|
-
(callback) => subscribe({
|
|
114
|
-
onStoreChange: callback,
|
|
115
|
-
eventType: "chat-status-changed"
|
|
116
|
-
}),
|
|
117
|
-
() => chatStore.current.getError(stableChatId),
|
|
118
|
-
() => chatStore.current.getError(stableChatId)
|
|
96
|
+
(options2) => chatRef.current.addToolResult(options2),
|
|
97
|
+
[chatRef]
|
|
119
98
|
);
|
|
120
99
|
const status = useSyncExternalStore(
|
|
121
100
|
(callback) => subscribe({
|
|
122
101
|
onStoreChange: callback,
|
|
123
|
-
eventType: "
|
|
102
|
+
eventType: "status-changed"
|
|
124
103
|
}),
|
|
125
|
-
() =>
|
|
126
|
-
() =>
|
|
104
|
+
() => chatRef.current.status,
|
|
105
|
+
() => chatRef.current.status
|
|
127
106
|
);
|
|
128
107
|
const subscribeToChatStoreForMessages = useCallback(
|
|
129
108
|
(callback) => {
|
|
130
109
|
return subscribe({
|
|
131
110
|
onStoreChange: throttleWaitMs ? throttle(callback, throttleWaitMs) : callback,
|
|
132
|
-
eventType: "
|
|
111
|
+
eventType: "messages-changed"
|
|
133
112
|
});
|
|
134
113
|
},
|
|
135
114
|
[subscribe, throttleWaitMs]
|
|
136
115
|
);
|
|
137
116
|
const messages = useSyncExternalStore(
|
|
138
117
|
(callback) => subscribeToChatStoreForMessages(callback),
|
|
139
|
-
() =>
|
|
140
|
-
() =>
|
|
141
|
-
);
|
|
142
|
-
const append = useCallback(
|
|
143
|
-
(message, { headers, body } = {}) => chatStore.current.submitMessage({
|
|
144
|
-
chatId: stableChatId,
|
|
145
|
-
message,
|
|
146
|
-
headers,
|
|
147
|
-
body,
|
|
148
|
-
onError,
|
|
149
|
-
onToolCall,
|
|
150
|
-
onFinish
|
|
151
|
-
}),
|
|
152
|
-
[chatStore, stableChatId, onError, onToolCall, onFinish]
|
|
153
|
-
);
|
|
154
|
-
const reload = useCallback(
|
|
155
|
-
async ({ headers, body } = {}) => chatStore.current.resubmitLastUserMessage({
|
|
156
|
-
chatId: stableChatId,
|
|
157
|
-
headers,
|
|
158
|
-
body,
|
|
159
|
-
onError,
|
|
160
|
-
onToolCall,
|
|
161
|
-
onFinish
|
|
162
|
-
}),
|
|
163
|
-
[chatStore, stableChatId, onError, onToolCall, onFinish]
|
|
164
|
-
);
|
|
165
|
-
const stop = useCallback(() => stopStream(), [stopStream]);
|
|
166
|
-
const experimental_resume = useCallback(
|
|
167
|
-
async () => chatStore.current.resumeStream({
|
|
168
|
-
chatId: stableChatId,
|
|
169
|
-
onError,
|
|
170
|
-
onToolCall,
|
|
171
|
-
onFinish
|
|
172
|
-
}),
|
|
173
|
-
[chatStore, stableChatId, onError, onToolCall, onFinish]
|
|
118
|
+
() => chatRef.current.messages,
|
|
119
|
+
() => chatRef.current.messages
|
|
174
120
|
);
|
|
175
121
|
const setMessages = useCallback(
|
|
176
122
|
(messagesParam) => {
|
|
177
123
|
if (typeof messagesParam === "function") {
|
|
178
124
|
messagesParam = messagesParam(messages);
|
|
179
125
|
}
|
|
180
|
-
|
|
181
|
-
id: stableChatId,
|
|
182
|
-
messages: messagesParam
|
|
183
|
-
});
|
|
126
|
+
chatRef.current.messages = messagesParam;
|
|
184
127
|
},
|
|
185
|
-
[
|
|
128
|
+
[chatRef, messages]
|
|
186
129
|
);
|
|
187
130
|
const [input, setInput] = useState(initialInput);
|
|
188
131
|
const handleSubmit = useCallback(
|
|
189
|
-
async (event,
|
|
132
|
+
async (event, options2 = {}) => {
|
|
190
133
|
var _a;
|
|
191
134
|
(_a = event == null ? void 0 : event.preventDefault) == null ? void 0 : _a.call(event);
|
|
192
|
-
const fileParts = Array.isArray(
|
|
135
|
+
const fileParts = Array.isArray(options2 == null ? void 0 : options2.files) ? options2.files : await convertFileListToFileUIParts(options2 == null ? void 0 : options2.files);
|
|
193
136
|
if (!input && fileParts.length === 0)
|
|
194
137
|
return;
|
|
195
|
-
append(
|
|
138
|
+
chatRef.current.append(
|
|
196
139
|
{
|
|
197
|
-
id: generateId(),
|
|
140
|
+
id: chatRef.current.generateId(),
|
|
198
141
|
role: "user",
|
|
199
142
|
metadata: void 0,
|
|
200
143
|
parts: [...fileParts, { type: "text", text: input }]
|
|
201
144
|
},
|
|
202
145
|
{
|
|
203
|
-
headers:
|
|
204
|
-
body:
|
|
146
|
+
headers: options2.headers,
|
|
147
|
+
body: options2.body
|
|
205
148
|
}
|
|
206
149
|
);
|
|
207
150
|
setInput("");
|
|
208
151
|
},
|
|
209
|
-
[input,
|
|
152
|
+
[input, chatRef]
|
|
210
153
|
);
|
|
211
154
|
const handleInputChange = (e) => {
|
|
212
155
|
setInput(e.target.value);
|
|
213
156
|
};
|
|
214
157
|
return {
|
|
158
|
+
id: chatRef.current.id,
|
|
215
159
|
messages,
|
|
216
|
-
chatId: stableChatId,
|
|
217
160
|
setMessages,
|
|
218
|
-
|
|
219
|
-
|
|
220
|
-
|
|
221
|
-
|
|
222
|
-
experimental_resume,
|
|
161
|
+
append: chatRef.current.append,
|
|
162
|
+
reload: chatRef.current.reload,
|
|
163
|
+
stop: chatRef.current.stop,
|
|
164
|
+
error: chatRef.current.error,
|
|
165
|
+
experimental_resume: chatRef.current.experimental_resume,
|
|
223
166
|
input,
|
|
224
167
|
setInput,
|
|
225
168
|
handleInputChange,
|
|
@@ -478,7 +421,7 @@ function useObject({
|
|
|
478
421
|
}
|
|
479
422
|
var experimental_useObject = useObject;
|
|
480
423
|
export {
|
|
481
|
-
|
|
424
|
+
Chat,
|
|
482
425
|
experimental_useObject,
|
|
483
426
|
useChat,
|
|
484
427
|
useCompletion
|
package/dist/index.mjs.map
CHANGED
|
@@ -1 +1 @@
|
|
|
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"]}
|
|
1
|
+
{"version":3,"sources":["../src/use-chat.ts","../src/chat.react.ts","../src/throttle.ts","../src/use-completion.ts","../src/use-object.ts"],"sourcesContent":["import {\n AbstractChat,\n BaseChatInit,\n ChatEvent,\n convertFileListToFileUIParts,\n InferUIDataParts,\n UIDataPartSchemas,\n type ChatRequestOptions,\n type CreateUIMessage,\n type FileUIPart,\n type UIMessage,\n} from 'ai';\nimport { useCallback, useRef, useState, useSyncExternalStore } from 'react';\nimport { Chat } from './chat.react';\nimport { throttle } from './throttle';\n\nexport type { CreateUIMessage, UIMessage };\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 id: string;\n\n /**\n * Update the `messages` state locally. This is useful when you want to\n * edit the messages on the client, and then trigger the `reload` method\n * manually to regenerate the AI response.\n */\n setMessages: (\n messages:\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 error: Error | undefined;\n} & Pick<\n AbstractChat<MESSAGE_METADATA, DATA_PART_SCHEMAS>,\n | 'append'\n | 'reload'\n | 'stop'\n | 'experimental_resume'\n | 'addToolResult'\n | 'status'\n | 'messages'\n>;\n\nexport type UseChatOptions<\n MESSAGE_METADATA = unknown,\n DATA_TYPE_SCHEMAS extends UIDataPartSchemas = UIDataPartSchemas,\n> = (\n | { chat: Chat<MESSAGE_METADATA, DATA_TYPE_SCHEMAS> }\n | BaseChatInit<MESSAGE_METADATA, DATA_TYPE_SCHEMAS>\n) & {\n /**\n /**\n * Initial input of the chat.\n */\n initialInput?: string;\n\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};\n\nexport function useChat<\n MESSAGE_METADATA = unknown,\n DATA_PART_SCHEMAS extends UIDataPartSchemas = UIDataPartSchemas,\n>({\n initialInput = '',\n experimental_throttle: throttleWaitMs,\n ...options\n}: UseChatOptions<MESSAGE_METADATA, DATA_PART_SCHEMAS> = {}): UseChatHelpers<\n MESSAGE_METADATA,\n DATA_PART_SCHEMAS\n> {\n const chatRef = useRef('chat' in options ? options.chat : new Chat(options));\n\n const subscribe = useCallback(\n ({\n onStoreChange,\n eventType,\n }: {\n onStoreChange: () => void;\n eventType: ChatEvent['type'];\n }) =>\n chatRef.current.subscribe({\n onChange: event => {\n if (event.type !== eventType) return;\n onStoreChange();\n },\n }),\n [chatRef],\n );\n\n const addToolResult = useCallback(\n (\n options: Parameters<\n Chat<MESSAGE_METADATA, DATA_PART_SCHEMAS>['addToolResult']\n >[0],\n ) => chatRef.current.addToolResult(options),\n [chatRef],\n );\n\n const status = useSyncExternalStore(\n callback =>\n subscribe({\n onStoreChange: callback,\n eventType: 'status-changed',\n }),\n () => chatRef.current.status,\n () => chatRef.current.status,\n );\n\n const subscribeToChatStoreForMessages = useCallback(\n (callback: () => void) => {\n return subscribe({\n onStoreChange: throttleWaitMs\n ? throttle(callback, throttleWaitMs)\n : callback,\n eventType: 'messages-changed',\n });\n },\n [subscribe, throttleWaitMs],\n );\n\n const messages = useSyncExternalStore(\n callback => subscribeToChatStoreForMessages(callback),\n () => chatRef.current.messages,\n () => chatRef.current.messages,\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 chatRef.current.messages = messagesParam;\n },\n [chatRef, 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 chatRef.current.append(\n {\n id: chatRef.current.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, chatRef],\n );\n\n const handleInputChange = (e: any) => {\n setInput(e.target.value);\n };\n\n return {\n id: chatRef.current.id,\n messages,\n setMessages,\n append: chatRef.current.append,\n reload: chatRef.current.reload,\n stop: chatRef.current.stop,\n error: chatRef.current.error,\n experimental_resume: chatRef.current.experimental_resume,\n input,\n setInput,\n handleInputChange,\n handleSubmit,\n status,\n addToolResult,\n };\n}\n","import {\n AbstractChat,\n BaseChatInit,\n ChatState,\n ChatStatus,\n UIDataPartSchemas,\n UIDataTypes,\n UIMessage,\n} from 'ai';\n\nclass ReactChatState<MESSAGE_METADATA, DATA_TYPES extends UIDataTypes>\n implements ChatState<MESSAGE_METADATA, DATA_TYPES>\n{\n #messages: UIMessage<MESSAGE_METADATA, DATA_TYPES>[];\n status: ChatStatus = 'ready';\n error: Error | undefined = undefined;\n\n constructor(messages: UIMessage<MESSAGE_METADATA, DATA_TYPES>[] = []) {\n this.#messages = messages;\n }\n\n get messages() {\n return this.#messages;\n }\n\n set messages(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 => structuredClone(value);\n}\n\nexport class Chat<\n MESSAGE_METADATA,\n UI_DATA_PART_SCHEMAS extends UIDataPartSchemas = UIDataPartSchemas,\n> extends AbstractChat<MESSAGE_METADATA, UI_DATA_PART_SCHEMAS> {\n constructor({\n messages,\n ...init\n }: BaseChatInit<MESSAGE_METADATA, UI_DATA_PART_SCHEMAS>) {\n super({ ...init, state: new ReactChatState(messages) });\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 InferSchema,\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 * as z3 from 'zod/v3';\nimport * as z4 from 'zod/v4/core';\n\n// use function to allow for mocking in tests:\nconst getOriginalFetch = () => fetch;\n\nexport type Experimental_UseObjectOptions<\n SCHEMA extends z4.$ZodType | z3.Schema | Schema,\n RESULT,\n> = {\n /**\n * The API endpoint. It should stream JSON that matches the schema as chunked text.\n */\n api: string;\n\n /**\n * A Zod schema that defines the shape of the complete object.\n */\n schema: SCHEMA;\n\n /**\n * An unique identifier. If not provided, a random one will be\n * generated. When provided, the `useObject` hook with the same `id` will\n * have shared states across components.\n */\n id?: string;\n\n /**\n * An optional value for the initial object.\n */\n initialValue?: DeepPartial<RESULT>;\n\n /**\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<\n SCHEMA extends z4.$ZodType | z3.Schema | Schema,\n RESULT = InferSchema<SCHEMA>,\n INPUT = any,\n>({\n api,\n id,\n schema, // required, in the future we will use it for validation\n initialValue,\n fetch,\n onError,\n onFinish,\n headers,\n credentials,\n}: Experimental_UseObjectOptions<\n SCHEMA,\n RESULT\n>): Experimental_UseObjectHelpers<RESULT, INPUT> {\n // Generate an unique id if not provided.\n const hookId = useId();\n const completionId = id ?? hookId;\n\n // Store the completion state in SWR, using the completionId as the key to share states.\n const { data, mutate } = useSWR<DeepPartial<RESULT>>(\n [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,EAIE;AAAA,OAOK;AACP,SAAS,aAAa,QAAQ,UAAU,4BAA4B;;;ACZpE;AAAA,EACE;AAAA,OAOK;AARP;AAUA,IAAM,iBAAN,MAEA;AAAA,EAKE,YAAY,WAAsD,CAAC,GAAG;AAJtE;AACA,kBAAqB;AACrB,iBAA2B;AAc3B,uBAAc,CAAC,YAAqD;AAClE,yBAAK,WAAY,KAAK,SAAS,OAAO,OAAO;AAAA,IAC/C;AAEA,sBAAa,MAAM;AACjB,yBAAK,WAAY,KAAK,SAAS,MAAM,GAAG,EAAE;AAAA,IAC5C;AAEA,0BAAiB,CACf,OACA,YACG;AACH,yBAAK,WAAY;AAAA,QACf,GAAG,mBAAK,WAAU,MAAM,GAAG,KAAK;AAAA,QAChC;AAAA,QACA,GAAG,mBAAK,WAAU,MAAM,QAAQ,CAAC;AAAA,MACnC;AAAA,IACF;AAEA,oBAAW,CAAI,UAAgB,gBAAgB,KAAK;AA9BlD,uBAAK,WAAY;AAAA,EACnB;AAAA,EAEA,IAAI,WAAW;AACb,WAAO,mBAAK;AAAA,EACd;AAAA,EAEA,IAAI,SAAS,UAAqD;AAChE,uBAAK,WAAY,CAAC,GAAG,QAAQ;AAAA,EAC/B;AAsBF;AApCE;AAsCK,IAAM,OAAN,cAGG,aAAqD;AAAA,EAC7D,YAAY;AAAA,IACV;AAAA,IACA,GAAG;AAAA,EACL,GAAyD;AACvD,UAAM,EAAE,GAAG,MAAM,OAAO,IAAI,eAAe,QAAQ,EAAE,CAAC;AAAA,EACxD;AACF;;;AC7DA,OAAO,sBAAsB;AAEtB,SAAS,SACd,IACA,QACG;AACH,SAAO,UAAU,OAAO,iBAAiB,IAAI,MAAM,IAAI;AACzD;;;AF4FO,SAAS,QAGd;AAAA,EACA,eAAe;AAAA,EACf,uBAAuB;AAAA,EACvB,GAAG;AACL,IAAyD,CAAC,GAGxD;AACA,QAAM,UAAU,OAAO,UAAU,UAAU,QAAQ,OAAO,IAAI,KAAK,OAAO,CAAC;AAE3E,QAAM,YAAY;AAAA,IAChB,CAAC;AAAA,MACC;AAAA,MACA;AAAA,IACF,MAIE,QAAQ,QAAQ,UAAU;AAAA,MACxB,UAAU,WAAS;AACjB,YAAI,MAAM,SAAS;AAAW;AAC9B,sBAAc;AAAA,MAChB;AAAA,IACF,CAAC;AAAA,IACH,CAAC,OAAO;AAAA,EACV;AAEA,QAAM,gBAAgB;AAAA,IACpB,CACEA,aAGG,QAAQ,QAAQ,cAAcA,QAAO;AAAA,IAC1C,CAAC,OAAO;AAAA,EACV;AAEA,QAAM,SAAS;AAAA,IACb,cACE,UAAU;AAAA,MACR,eAAe;AAAA,MACf,WAAW;AAAA,IACb,CAAC;AAAA,IACH,MAAM,QAAQ,QAAQ;AAAA,IACtB,MAAM,QAAQ,QAAQ;AAAA,EACxB;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,QAAQ,QAAQ;AAAA,IACtB,MAAM,QAAQ,QAAQ;AAAA,EACxB;AAEA,QAAM,cAAc;AAAA,IAClB,CACE,kBAWG;AACH,UAAI,OAAO,kBAAkB,YAAY;AACvC,wBAAgB,cAAc,QAAQ;AAAA,MACxC;AAEA,cAAQ,QAAQ,WAAW;AAAA,IAC7B;AAAA,IACA,CAAC,SAAS,QAAQ;AAAA,EACpB;AAGA,QAAM,CAAC,OAAO,QAAQ,IAAI,SAAS,YAAY;AAE/C,QAAM,eAAe;AAAA,IACnB,OACE,OACAA,WAEI,CAAC,MACF;AAtMT;AAuMM,2CAAO,mBAAP;AAEA,YAAM,YAAY,MAAM,QAAQA,YAAA,gBAAAA,SAAS,KAAK,IAC1CA,SAAQ,QACR,MAAM,6BAA6BA,YAAA,gBAAAA,SAAS,KAAK;AAErD,UAAI,CAAC,SAAS,UAAU,WAAW;AAAG;AAEtC,cAAQ,QAAQ;AAAA,QACd;AAAA,UACE,IAAI,QAAQ,QAAQ,WAAW;AAAA,UAC/B,MAAM;AAAA,UACN,UAAU;AAAA,UACV,OAAO,CAAC,GAAG,WAAW,EAAE,MAAM,QAAQ,MAAM,MAAM,CAAC;AAAA,QACrD;AAAA,QACA;AAAA,UACE,SAASA,SAAQ;AAAA,UACjB,MAAMA,SAAQ;AAAA,QAChB;AAAA,MACF;AAEA,eAAS,EAAE;AAAA,IACb;AAAA,IACA,CAAC,OAAO,OAAO;AAAA,EACjB;AAEA,QAAM,oBAAoB,CAAC,MAAW;AACpC,aAAS,EAAE,OAAO,KAAK;AAAA,EACzB;AAEA,SAAO;AAAA,IACL,IAAI,QAAQ,QAAQ;AAAA,IACpB;AAAA,IACA;AAAA,IACA,QAAQ,QAAQ,QAAQ;AAAA,IACxB,QAAQ,QAAQ,QAAQ;AAAA,IACxB,MAAM,QAAQ,QAAQ;AAAA,IACtB,OAAO,QAAQ,QAAQ;AAAA,IACvB,qBAAqB,QAAQ,QAAQ;AAAA,IACrC;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF;AACF;;;AGrPA;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,EAGE;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;AAKnB,IAAM,mBAAmB,MAAM;AA+F/B,SAAS,UAIP;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA;AAAA,EACA;AAAA,EACA,OAAAC;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AACF,GAGiD;AAE/C,QAAM,SAASJ,OAAM;AACrB,QAAM,eAAe,kBAAM;AAG3B,QAAM,EAAE,MAAM,OAAO,IAAIG;AAAA,IACvB,CAAC,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;AArJjC;AAsJI,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;AA/JzC;AAgKI,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.9",
|
|
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": "5.0.0-alpha.
|
|
25
|
-
"@ai-sdk/provider-utils": "3.0.0-alpha.
|
|
24
|
+
"ai": "5.0.0-alpha.9",
|
|
25
|
+
"@ai-sdk/provider-utils": "3.0.0-alpha.9"
|
|
26
26
|
},
|
|
27
27
|
"devDependencies": {
|
|
28
28
|
"@testing-library/jest-dom": "^6.6.3",
|
|
@@ -37,13 +37,13 @@
|
|
|
37
37
|
"react-dom": "^18",
|
|
38
38
|
"tsup": "^7.2.0",
|
|
39
39
|
"typescript": "5.8.3",
|
|
40
|
-
"zod": "3.
|
|
40
|
+
"zod": "3.25.49",
|
|
41
41
|
"@vercel/ai-tsconfig": "0.0.0",
|
|
42
42
|
"eslint-config-vercel-ai": "0.0.0"
|
|
43
43
|
},
|
|
44
44
|
"peerDependencies": {
|
|
45
45
|
"react": "^18 || ^19 || ^19.0.0-rc",
|
|
46
|
-
"zod": "^3.
|
|
46
|
+
"zod": "^3.25.49"
|
|
47
47
|
},
|
|
48
48
|
"peerDependenciesMeta": {
|
|
49
49
|
"zod": {
|
|
@@ -70,7 +70,7 @@
|
|
|
70
70
|
],
|
|
71
71
|
"scripts": {
|
|
72
72
|
"build": "pnpm clean && tsup --tsconfig tsconfig.build.json",
|
|
73
|
-
"build:watch": "pnpm clean && tsup --watch",
|
|
73
|
+
"build:watch": "pnpm clean && tsup --watch --tsconfig tsconfig.build.json",
|
|
74
74
|
"clean": "rm -rf dist *.tsbuildinfo",
|
|
75
75
|
"lint": "eslint \"./**/*.ts*\"",
|
|
76
76
|
"type-check": "tsc --build",
|