@assistant-ui/react-ai-sdk 0.0.1

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.
@@ -0,0 +1,20 @@
1
+
2
+ > @assistant-ui/react-ai-sdk@0.0.0-alpha.1 build /Users/yonom/Documents/GitHub/assistant-ui/packages/react-ai-sdk
3
+ > tsup src/index.ts --format cjs,esm --dts --sourcemap
4
+
5
+ CLI Building entry: src/index.ts
6
+ CLI Using tsconfig: tsconfig.json
7
+ CLI tsup v8.1.0
8
+ CLI Target: esnext
9
+ CJS Build start
10
+ ESM Build start
11
+ CJS dist/index.cjs 15.24 KB
12
+ CJS dist/index.cjs.map 29.06 KB
13
+ CJS ⚡️ Build success in 20ms
14
+ ESM dist/index.js 13.93 KB
15
+ ESM dist/index.js.map 29.18 KB
16
+ ESM ⚡️ Build success in 20ms
17
+ DTS Build start
18
+ DTS ⚡️ Build success in 2187ms
19
+ DTS dist/index.d.cts 3.30 KB
20
+ DTS dist/index.d.ts 3.30 KB
package/LICENSE ADDED
@@ -0,0 +1,21 @@
1
+ MIT License
2
+
3
+ Copyright (c) 2024 Simon Farshid
4
+
5
+ Permission is hereby granted, free of charge, to any person obtaining a copy
6
+ of this software and associated documentation files (the "Software"), to deal
7
+ in the Software without restriction, including without limitation the rights
8
+ to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
9
+ copies of the Software, and to permit persons to whom the Software is
10
+ furnished to do so, subject to the following conditions:
11
+
12
+ The above copyright notice and this permission notice shall be included in all
13
+ copies or substantial portions of the Software.
14
+
15
+ THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
16
+ IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
17
+ FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
18
+ AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
19
+ LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
20
+ OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
21
+ SOFTWARE.
package/README.md ADDED
@@ -0,0 +1,38 @@
1
+ # `assistant-ui`
2
+
3
+ `assistant-ui` is a set of React components for AI chat.
4
+
5
+ - [Discord](https://discord.gg/S9dwgCNEFs)
6
+ - [Website](https://assistant-ui.com/)
7
+ - [Demo](https://assistant-ui-rsc-example.vercel.app/)
8
+
9
+ ## Documentation
10
+
11
+ - [Documentation](https://www.assistant-ui.com/docs/getting-started)
12
+
13
+ ## Minimal Example with Vercel AI SDK
14
+
15
+ ```sh
16
+ npx assistant-ui@latest add assistant-modal
17
+ ```
18
+
19
+ ```tsx
20
+ "use client";
21
+
22
+ import { useChat } from "@ai-sdk/react";
23
+ import { AssistantRuntimeProvider, useVercelUseChatRuntime } from "@assistant-ui/react";
24
+ import { AssistantModal } from "@/components/ui/assistant-ui/assistant-modal";
25
+
26
+ export default const MyApp = () => {
27
+ const chat = useChat({
28
+ api: "/api/chat" // your backend route
29
+ });
30
+ const runtime = useVercelUseChatRuntime(chat);
31
+
32
+ return (
33
+ <AssistantRuntimeProvider runtime={runtime}>
34
+ <AssistantModal />
35
+ </AssistantRuntimeProvider>
36
+ );
37
+ }
38
+ ```
package/dist/index.cjs ADDED
@@ -0,0 +1,494 @@
1
+ "use strict";
2
+ var __defProp = Object.defineProperty;
3
+ var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
4
+ var __getOwnPropNames = Object.getOwnPropertyNames;
5
+ var __hasOwnProp = Object.prototype.hasOwnProperty;
6
+ var __export = (target, all) => {
7
+ for (var name in all)
8
+ __defProp(target, name, { get: all[name], enumerable: true });
9
+ };
10
+ var __copyProps = (to, from, except, desc) => {
11
+ if (from && typeof from === "object" || typeof from === "function") {
12
+ for (let key of __getOwnPropNames(from))
13
+ if (!__hasOwnProp.call(to, key) && key !== except)
14
+ __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
15
+ }
16
+ return to;
17
+ };
18
+ var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
19
+
20
+ // src/index.ts
21
+ var src_exports = {};
22
+ __export(src_exports, {
23
+ getVercelAIMessage: () => getVercelAIMessage,
24
+ getVercelRSCMessage: () => getVercelRSCMessage,
25
+ unstable_VercelModelAdapter: () => VercelModelAdapter,
26
+ useVercelRSCRuntime: () => useVercelRSCRuntime,
27
+ useVercelUseAssistantRuntime: () => useVercelUseAssistantRuntime,
28
+ useVercelUseChatRuntime: () => useVercelUseChatRuntime
29
+ });
30
+ module.exports = __toCommonJS(src_exports);
31
+
32
+ // src/core/VercelModelAdapter.tsx
33
+ var import_ai = require("ai");
34
+ var VercelModelAdapter = class {
35
+ constructor(model) {
36
+ this.model = model;
37
+ }
38
+ async run({ messages, abortSignal, onUpdate }) {
39
+ const { fullStream } = await (0, import_ai.streamText)({
40
+ model: this.model,
41
+ abortSignal,
42
+ messages: messages.map((m) => ({
43
+ role: m.role,
44
+ content: m.content.filter((c) => c.type !== "ui")
45
+ }))
46
+ });
47
+ const content = [];
48
+ for await (const aiPart of fullStream) {
49
+ switch (aiPart.type) {
50
+ case "text-delta": {
51
+ let part = content.at(-1);
52
+ if (!part || part.type !== "text") {
53
+ part = { type: "text", text: "" };
54
+ content.push(part);
55
+ }
56
+ part.text += aiPart.textDelta;
57
+ break;
58
+ }
59
+ case "tool-call": {
60
+ content.push({
61
+ type: "tool-call",
62
+ name: aiPart.toolName,
63
+ args: aiPart.args
64
+ });
65
+ break;
66
+ }
67
+ }
68
+ onUpdate({ content });
69
+ }
70
+ return { content };
71
+ }
72
+ };
73
+
74
+ // src/rsc/useVercelRSCRuntime.tsx
75
+ var import_react2 = require("react");
76
+
77
+ // src/rsc/VercelRSCRuntime.tsx
78
+ var import_zustand = require("zustand");
79
+
80
+ // src/rsc/useVercelRSCSync.tsx
81
+ var import_react = require("react");
82
+
83
+ // src/utils/ThreadMessageConverter.ts
84
+ var ThreadMessageConverter = class {
85
+ cache = /* @__PURE__ */ new WeakMap();
86
+ convertMessages(messages, converter, keyMapper = (key) => key) {
87
+ return messages.map((m) => {
88
+ const key = keyMapper(m);
89
+ const cached = this.cache.get(key);
90
+ const newMessage = converter(m, cached);
91
+ this.cache.set(key, newMessage);
92
+ return newMessage;
93
+ });
94
+ }
95
+ };
96
+
97
+ // src/rsc/getVercelRSCMessage.tsx
98
+ var symbolInnerRSCMessage = Symbol("innerVercelRSCMessage");
99
+ var getVercelRSCMessage = (message) => {
100
+ return message[symbolInnerRSCMessage];
101
+ };
102
+
103
+ // src/rsc/useVercelRSCSync.tsx
104
+ var vercelToThreadMessage = (converter, rawMessage) => {
105
+ const message = converter(rawMessage);
106
+ return {
107
+ id: message.id,
108
+ role: message.role,
109
+ content: [{ type: "ui", display: message.display }],
110
+ createdAt: message.createdAt ?? /* @__PURE__ */ new Date(),
111
+ ...{ status: "done" },
112
+ [symbolInnerRSCMessage]: rawMessage
113
+ };
114
+ };
115
+ var useVercelRSCSync = (adapter, updateData) => {
116
+ const [converter, convertCallback] = (0, import_react.useMemo)(() => {
117
+ const rscConverter = adapter.convertMessage ?? ((m) => m);
118
+ const convertCallback2 = (m, cache) => {
119
+ if (cache) return cache;
120
+ return vercelToThreadMessage(rscConverter, m);
121
+ };
122
+ return [new ThreadMessageConverter(), convertCallback2];
123
+ }, [adapter.convertMessage]);
124
+ (0, import_react.useEffect)(() => {
125
+ updateData(converter.convertMessages(adapter.messages, convertCallback));
126
+ }, [updateData, converter, convertCallback, adapter.messages]);
127
+ };
128
+
129
+ // src/rsc/VercelRSCRuntime.tsx
130
+ var EMPTY_BRANCHES = Object.freeze([]);
131
+ var VercelRSCRuntime = class {
132
+ constructor(adapter) {
133
+ this.adapter = adapter;
134
+ this.useAdapter = (0, import_zustand.create)(() => ({
135
+ adapter
136
+ }));
137
+ }
138
+ useAdapter;
139
+ _subscriptions = /* @__PURE__ */ new Set();
140
+ isRunning = false;
141
+ messages = [];
142
+ withRunning = (callback) => {
143
+ this.isRunning = true;
144
+ return callback.finally(() => {
145
+ this.isRunning = false;
146
+ });
147
+ };
148
+ getBranches() {
149
+ return EMPTY_BRANCHES;
150
+ }
151
+ switchToBranch() {
152
+ throw new Error(
153
+ "Branch switching is not supported by VercelRSCAssistantProvider."
154
+ );
155
+ }
156
+ async append(message) {
157
+ if (message.parentId !== (this.messages.at(-1)?.id ?? null)) {
158
+ if (!this.adapter.edit)
159
+ throw new Error(
160
+ "Message editing is not enabled, please provide an edit callback to VercelRSCAssistantProvider."
161
+ );
162
+ await this.withRunning(this.adapter.edit(message));
163
+ } else {
164
+ await this.withRunning(this.adapter.append(message));
165
+ }
166
+ }
167
+ async startRun(parentId) {
168
+ if (!this.adapter.reload)
169
+ throw new Error(
170
+ "Message reloading is not enabled, please provide a reload callback to VercelRSCAssistantProvider."
171
+ );
172
+ await this.withRunning(this.adapter.reload(parentId));
173
+ }
174
+ cancelRun() {
175
+ if (process.env["NODE_ENV"] === "development") {
176
+ console.warn(
177
+ "Run cancellation is not supported by VercelRSCAssistantProvider."
178
+ );
179
+ }
180
+ }
181
+ subscribe(callback) {
182
+ this._subscriptions.add(callback);
183
+ return () => this._subscriptions.delete(callback);
184
+ }
185
+ onAdapterUpdated() {
186
+ if (this.useAdapter.getState().adapter !== this.adapter) {
187
+ this.useAdapter.setState({ adapter: this.adapter });
188
+ }
189
+ }
190
+ updateData = (messages) => {
191
+ this.messages = messages;
192
+ for (const callback of this._subscriptions) callback();
193
+ };
194
+ unstable_synchronizer = () => {
195
+ const { adapter } = this.useAdapter();
196
+ useVercelRSCSync(adapter, this.updateData);
197
+ return null;
198
+ };
199
+ };
200
+
201
+ // src/rsc/useVercelRSCRuntime.tsx
202
+ var useVercelRSCRuntime = (adapter) => {
203
+ const [runtime] = (0, import_react2.useState)(() => new VercelRSCRuntime(adapter));
204
+ (0, import_react2.useInsertionEffect)(() => {
205
+ runtime.adapter = adapter;
206
+ });
207
+ (0, import_react2.useEffect)(() => {
208
+ runtime.onAdapterUpdated();
209
+ });
210
+ return runtime;
211
+ };
212
+
213
+ // src/ui/use-chat/useVercelUseChatRuntime.tsx
214
+ var import_react7 = require("react");
215
+
216
+ // src/ui/VercelAIRuntime.tsx
217
+ var import_react6 = require("@assistant-ui/react");
218
+ var import_zustand2 = require("zustand");
219
+
220
+ // src/ui/getVercelAIMessage.tsx
221
+ var symbolInnerAIMessage = Symbol("innerVercelAIUIMessage");
222
+ var getVercelAIMessage = (message) => {
223
+ return message[symbolInnerAIMessage];
224
+ };
225
+
226
+ // src/ui/utils/sliceMessagesUntil.tsx
227
+ var sliceMessagesUntil = (messages, messageId) => {
228
+ if (messageId == null) return [];
229
+ let messageIdx = messages.findIndex((m) => m.id === messageId);
230
+ if (messageIdx === -1)
231
+ throw new Error(
232
+ "useVercelAIThreadState: Message not found. This is liekly an internal bug in assistant-ui."
233
+ );
234
+ while (messages[messageIdx + 1]?.role === "assistant") {
235
+ messageIdx++;
236
+ }
237
+ return messages.slice(0, messageIdx + 1);
238
+ };
239
+
240
+ // src/ui/utils/useVercelAIComposerSync.tsx
241
+ var import_react3 = require("@assistant-ui/react");
242
+ var import_react4 = require("react");
243
+ var useVercelAIComposerSync = (vercel) => {
244
+ const { useComposer } = (0, import_react3.unstable_useThreadContext)();
245
+ (0, import_react4.useEffect)(() => {
246
+ useComposer.setState({
247
+ value: vercel.input,
248
+ setValue: vercel.setInput
249
+ });
250
+ }, [useComposer, vercel.input, vercel.setInput]);
251
+ };
252
+
253
+ // src/ui/utils/useVercelAIThreadSync.tsx
254
+ var import_react5 = require("react");
255
+ var getIsRunning = (vercel) => {
256
+ if ("isLoading" in vercel) return vercel.isLoading;
257
+ return vercel.status === "in_progress";
258
+ };
259
+ var vercelToThreadMessage2 = (messages, status) => {
260
+ const firstMessage = messages[0];
261
+ if (!firstMessage) throw new Error("No messages found");
262
+ const common = {
263
+ id: firstMessage.id,
264
+ createdAt: firstMessage.createdAt ?? /* @__PURE__ */ new Date(),
265
+ [symbolInnerAIMessage]: messages
266
+ };
267
+ switch (firstMessage.role) {
268
+ case "user":
269
+ if (messages.length > 1) {
270
+ throw new Error(
271
+ "Multiple user messages found. This is likely an internal bug in assistant-ui."
272
+ );
273
+ }
274
+ return {
275
+ ...common,
276
+ role: "user",
277
+ content: [{ type: "text", text: firstMessage.content }]
278
+ };
279
+ case "assistant":
280
+ return {
281
+ ...common,
282
+ role: "assistant",
283
+ content: messages.flatMap((message) => [
284
+ ...message.content ? [{ type: "text", text: message.content }] : [],
285
+ ...message.toolInvocations?.map(
286
+ (t) => ({
287
+ type: "tool-call",
288
+ name: t.toolName,
289
+ args: t.args,
290
+ result: "result" in t ? t.result : void 0
291
+ })
292
+ ) ?? []
293
+ ]),
294
+ status
295
+ };
296
+ default:
297
+ throw new Error(
298
+ `You have a message with an unsupported role. The role ${firstMessage.role} is not supported.`
299
+ );
300
+ }
301
+ };
302
+ var hasItems = (messages) => messages.length > 0;
303
+ var chunkedMessages = (messages) => {
304
+ const chunks = [];
305
+ let currentChunk = [];
306
+ for (const message of messages) {
307
+ if (message.role === "assistant") {
308
+ currentChunk.push(message);
309
+ } else {
310
+ if (hasItems(currentChunk)) {
311
+ chunks.push(currentChunk);
312
+ currentChunk = [];
313
+ }
314
+ chunks.push([message]);
315
+ }
316
+ }
317
+ if (hasItems(currentChunk)) {
318
+ chunks.push(currentChunk);
319
+ }
320
+ return chunks;
321
+ };
322
+ var shallowArrayEqual = (a, b) => {
323
+ if (a.length !== b.length) return false;
324
+ for (let i = 0; i < a.length; i++) {
325
+ if (a[i] !== b[i]) return false;
326
+ }
327
+ return true;
328
+ };
329
+ var useVercelAIThreadSync = (vercel, updateData) => {
330
+ const isRunning = getIsRunning(vercel);
331
+ const converter = (0, import_react5.useMemo)(() => new ThreadMessageConverter(), []);
332
+ (0, import_react5.useEffect)(() => {
333
+ const lastMessageId = vercel.messages.at(-1)?.id;
334
+ const convertCallback = (messages2, cache) => {
335
+ const status = lastMessageId === messages2[0].id && isRunning ? "in_progress" : "done";
336
+ if (cache && shallowArrayEqual(cache.content, messages2) && (cache.role === "user" || cache.status === status))
337
+ return cache;
338
+ return vercelToThreadMessage2(messages2, status);
339
+ };
340
+ const messages = converter.convertMessages(
341
+ chunkedMessages(vercel.messages),
342
+ convertCallback,
343
+ (m) => m[0]
344
+ );
345
+ updateData(isRunning, messages);
346
+ }, [updateData, isRunning, vercel.messages, converter]);
347
+ };
348
+
349
+ // src/ui/VercelAIRuntime.tsx
350
+ var hasUpcomingMessage = (isRunning, messages) => {
351
+ return isRunning && messages[messages.length - 1]?.role !== "assistant";
352
+ };
353
+ var VercelAIRuntime = class {
354
+ constructor(vercel) {
355
+ this.vercel = vercel;
356
+ this.useVercel = (0, import_zustand2.create)(() => ({
357
+ vercel
358
+ }));
359
+ }
360
+ _subscriptions = /* @__PURE__ */ new Set();
361
+ repository = new import_react6.unstable_MessageRepository();
362
+ assistantOptimisticId = null;
363
+ useVercel;
364
+ messages = [];
365
+ isRunning = false;
366
+ getBranches(messageId) {
367
+ return this.repository.getBranches(messageId);
368
+ }
369
+ switchToBranch(branchId) {
370
+ this.repository.switchToBranch(branchId);
371
+ this.updateVercelMessages(this.repository.getMessages());
372
+ }
373
+ async append(message) {
374
+ if (message.content.length !== 1 || message.content[0]?.type !== "text")
375
+ throw new Error("Only text content is supported by Vercel AI SDK.");
376
+ const newMessages = sliceMessagesUntil(
377
+ this.vercel.messages,
378
+ message.parentId
379
+ );
380
+ this.vercel.setMessages(newMessages);
381
+ await this.vercel.append({
382
+ role: "user",
383
+ content: message.content[0].text
384
+ });
385
+ }
386
+ async startRun(parentId) {
387
+ const reloadMaybe = "reload" in this.vercel ? this.vercel.reload : void 0;
388
+ if (!reloadMaybe)
389
+ throw new Error(
390
+ "Reload is not supported by Vercel AI SDK's useAssistant."
391
+ );
392
+ const newMessages = sliceMessagesUntil(this.vercel.messages, parentId);
393
+ this.vercel.setMessages(newMessages);
394
+ await reloadMaybe();
395
+ }
396
+ cancelRun() {
397
+ const previousMessage = this.vercel.messages.at(-1);
398
+ this.vercel.stop();
399
+ if (this.assistantOptimisticId) {
400
+ this.repository.deleteMessage(this.assistantOptimisticId);
401
+ this.assistantOptimisticId = null;
402
+ }
403
+ let messages = this.repository.getMessages();
404
+ if (previousMessage?.role === "user" && previousMessage.id === messages.at(-1)?.id) {
405
+ this.vercel.setInput(previousMessage.content);
406
+ this.repository.deleteMessage(previousMessage.id);
407
+ messages = this.repository.getMessages();
408
+ }
409
+ setTimeout(() => {
410
+ this.updateVercelMessages(messages);
411
+ }, 0);
412
+ }
413
+ subscribe(callback) {
414
+ this._subscriptions.add(callback);
415
+ return () => this._subscriptions.delete(callback);
416
+ }
417
+ updateVercelMessages = (messages) => {
418
+ this.vercel.setMessages(
419
+ messages.flatMap(getVercelAIMessage).filter((m) => m != null)
420
+ );
421
+ };
422
+ onVercelUpdated() {
423
+ if (this.useVercel.getState().vercel !== this.vercel) {
424
+ this.useVercel.setState({ vercel: this.vercel });
425
+ }
426
+ }
427
+ updateData = (isRunning, vm) => {
428
+ for (let i = 0; i < vm.length; i++) {
429
+ const message = vm[i];
430
+ const parent = vm[i - 1];
431
+ this.repository.addOrUpdateMessage(parent?.id ?? null, message);
432
+ }
433
+ if (this.assistantOptimisticId) {
434
+ this.repository.deleteMessage(this.assistantOptimisticId);
435
+ this.assistantOptimisticId = null;
436
+ }
437
+ if (hasUpcomingMessage(isRunning, vm)) {
438
+ this.assistantOptimisticId = this.repository.appendOptimisticMessage(
439
+ vm.at(-1)?.id ?? null,
440
+ {
441
+ role: "assistant",
442
+ content: [{ type: "text", text: "" }]
443
+ }
444
+ );
445
+ }
446
+ this.repository.resetHead(
447
+ this.assistantOptimisticId ?? vm.at(-1)?.id ?? null
448
+ );
449
+ this.messages = this.repository.getMessages();
450
+ this.isRunning = isRunning;
451
+ for (const callback of this._subscriptions) callback();
452
+ };
453
+ unstable_synchronizer = () => {
454
+ const { vercel } = this.useVercel();
455
+ useVercelAIThreadSync(vercel, this.updateData);
456
+ useVercelAIComposerSync(vercel);
457
+ return null;
458
+ };
459
+ };
460
+
461
+ // src/ui/use-chat/useVercelUseChatRuntime.tsx
462
+ var useVercelUseChatRuntime = (chatHelpers) => {
463
+ const [runtime] = (0, import_react7.useState)(() => new VercelAIRuntime(chatHelpers));
464
+ (0, import_react7.useInsertionEffect)(() => {
465
+ runtime.vercel = chatHelpers;
466
+ });
467
+ (0, import_react7.useEffect)(() => {
468
+ runtime.onVercelUpdated();
469
+ });
470
+ return runtime;
471
+ };
472
+
473
+ // src/ui/use-assistant/useVercelUseAssistantRuntime.tsx
474
+ var import_react8 = require("react");
475
+ var useVercelUseAssistantRuntime = (assistantHelpers) => {
476
+ const [runtime] = (0, import_react8.useState)(() => new VercelAIRuntime(assistantHelpers));
477
+ (0, import_react8.useInsertionEffect)(() => {
478
+ runtime.vercel = assistantHelpers;
479
+ });
480
+ (0, import_react8.useEffect)(() => {
481
+ runtime.onVercelUpdated();
482
+ });
483
+ return runtime;
484
+ };
485
+ // Annotate the CommonJS export names for ESM import in node:
486
+ 0 && (module.exports = {
487
+ getVercelAIMessage,
488
+ getVercelRSCMessage,
489
+ unstable_VercelModelAdapter,
490
+ useVercelRSCRuntime,
491
+ useVercelUseAssistantRuntime,
492
+ useVercelUseChatRuntime
493
+ });
494
+ //# sourceMappingURL=index.cjs.map
@@ -0,0 +1 @@
1
+ {"version":3,"sources":["../src/index.ts","../src/core/VercelModelAdapter.tsx","../src/rsc/useVercelRSCRuntime.tsx","../src/rsc/VercelRSCRuntime.tsx","../src/rsc/useVercelRSCSync.tsx","../src/utils/ThreadMessageConverter.ts","../src/rsc/getVercelRSCMessage.tsx","../src/ui/use-chat/useVercelUseChatRuntime.tsx","../src/ui/VercelAIRuntime.tsx","../src/ui/getVercelAIMessage.tsx","../src/ui/utils/sliceMessagesUntil.tsx","../src/ui/utils/useVercelAIComposerSync.tsx","../src/ui/utils/useVercelAIThreadSync.tsx","../src/ui/use-assistant/useVercelUseAssistantRuntime.tsx"],"sourcesContent":["export * from \"./core\";\nexport * from \"./rsc\";\nexport * from \"./ui\";\n","\"use client\";\nimport type {\n AssistantContentPart,\n TextContentPart,\n} from \"@assistant-ui/react\";\nimport type {\n unstable_ChatModelAdapter as ChatModelAdapter,\n unstable_ChatModelRunOptions as ChatModelRunOptions,\n} from \"@assistant-ui/react\";\nimport { type LanguageModel, streamText } from \"ai\";\n\nexport class VercelModelAdapter implements ChatModelAdapter {\n constructor(private readonly model: LanguageModel) {}\n\n async run({ messages, abortSignal, onUpdate }: ChatModelRunOptions) {\n const { fullStream } = await streamText({\n model: this.model,\n abortSignal,\n messages: messages.map((m) => ({\n role: m.role,\n content: m.content.filter((c): c is TextContentPart => c.type !== \"ui\"),\n })),\n });\n\n const content: AssistantContentPart[] = [];\n for await (const aiPart of fullStream) {\n switch (aiPart.type) {\n case \"text-delta\": {\n let part = content.at(-1);\n if (!part || part.type !== \"text\") {\n part = { type: \"text\", text: \"\" };\n content.push(part);\n }\n part.text += aiPart.textDelta;\n break;\n }\n // TODO tool results\n case \"tool-call\": {\n content.push({\n type: \"tool-call\",\n name: aiPart.toolName,\n args: aiPart.args,\n });\n break;\n }\n }\n\n onUpdate({ content });\n }\n\n return { content };\n }\n}\n","\"use client\";\n\nimport { useEffect, useInsertionEffect, useState } from \"react\";\nimport type { VercelRSCAdapter } from \"./VercelRSCAdapter\";\nimport { VercelRSCRuntime } from \"./VercelRSCRuntime\";\n\nexport const useVercelRSCRuntime = <T extends WeakKey>(\n adapter: VercelRSCAdapter<T>,\n) => {\n const [runtime] = useState(() => new VercelRSCRuntime(adapter));\n\n useInsertionEffect(() => {\n runtime.adapter = adapter;\n });\n useEffect(() => {\n runtime.onAdapterUpdated();\n });\n\n return runtime;\n};\n","\"use client\";\n\nimport type {\n AppendMessage,\n AssistantRuntime,\n ReactThreadRuntime,\n ThreadMessage,\n} from \"@assistant-ui/react\";\nimport { type StoreApi, type UseBoundStore, create } from \"zustand\";\nimport type { VercelRSCAdapter } from \"./VercelRSCAdapter\";\nimport type { VercelRSCMessage } from \"./VercelRSCMessage\";\nimport { useVercelRSCSync } from \"./useVercelRSCSync\";\n\nconst EMPTY_BRANCHES: readonly never[] = Object.freeze([]);\n\nexport class VercelRSCRuntime<T extends WeakKey = VercelRSCMessage>\n implements AssistantRuntime, ReactThreadRuntime\n{\n private useAdapter: UseBoundStore<StoreApi<{ adapter: VercelRSCAdapter<T> }>>;\n\n private _subscriptions = new Set<() => void>();\n\n public isRunning = false;\n public messages: ThreadMessage[] = [];\n\n constructor(public adapter: VercelRSCAdapter<T>) {\n this.useAdapter = create(() => ({\n adapter,\n }));\n }\n\n private withRunning = (callback: Promise<unknown>) => {\n this.isRunning = true;\n return callback.finally(() => {\n this.isRunning = false;\n });\n };\n\n public getBranches(): readonly string[] {\n return EMPTY_BRANCHES;\n }\n\n public switchToBranch(): void {\n throw new Error(\n \"Branch switching is not supported by VercelRSCAssistantProvider.\",\n );\n }\n\n public async append(message: AppendMessage): Promise<void> {\n if (message.parentId !== (this.messages.at(-1)?.id ?? null)) {\n if (!this.adapter.edit)\n throw new Error(\n \"Message editing is not enabled, please provide an edit callback to VercelRSCAssistantProvider.\",\n );\n await this.withRunning(this.adapter.edit(message));\n } else {\n await this.withRunning(this.adapter.append(message));\n }\n }\n\n public async startRun(parentId: string | null): Promise<void> {\n if (!this.adapter.reload)\n throw new Error(\n \"Message reloading is not enabled, please provide a reload callback to VercelRSCAssistantProvider.\",\n );\n await this.withRunning(this.adapter.reload(parentId));\n }\n\n cancelRun(): void {\n // in dev mode, log a warning\n if (process.env[\"NODE_ENV\"] === \"development\") {\n console.warn(\n \"Run cancellation is not supported by VercelRSCAssistantProvider.\",\n );\n }\n }\n\n public subscribe(callback: () => void): () => void {\n this._subscriptions.add(callback);\n return () => this._subscriptions.delete(callback);\n }\n\n public onAdapterUpdated() {\n if (this.useAdapter.getState().adapter !== this.adapter) {\n this.useAdapter.setState({ adapter: this.adapter });\n }\n }\n\n private updateData = (messages: ThreadMessage[]) => {\n this.messages = messages;\n for (const callback of this._subscriptions) callback();\n };\n\n unstable_synchronizer = () => {\n const { adapter } = this.useAdapter();\n\n useVercelRSCSync(adapter, this.updateData);\n\n return null;\n };\n}\n","import type { ThreadMessage } from \"@assistant-ui/react\";\nimport { useEffect, useMemo } from \"react\";\nimport {\n type ConverterCallback,\n ThreadMessageConverter,\n} from \"../utils/ThreadMessageConverter\";\nimport type { VercelRSCAdapter } from \"./VercelRSCAdapter\";\nimport type { VercelRSCMessage } from \"./VercelRSCMessage\";\nimport {\n type VercelRSCThreadMessage,\n symbolInnerRSCMessage,\n} from \"./getVercelRSCMessage\";\n\nconst vercelToThreadMessage = <T,>(\n converter: (message: T) => VercelRSCMessage,\n rawMessage: T,\n): VercelRSCThreadMessage<T> => {\n const message = converter(rawMessage);\n\n return {\n id: message.id,\n role: message.role,\n content: [{ type: \"ui\", display: message.display }],\n createdAt: message.createdAt ?? new Date(),\n ...{ status: \"done\" },\n [symbolInnerRSCMessage]: rawMessage,\n };\n};\n\ntype UpdateDataCallback = (messages: ThreadMessage[]) => void;\n\nexport const useVercelRSCSync = <T extends WeakKey>(\n adapter: VercelRSCAdapter<T>,\n updateData: UpdateDataCallback,\n) => {\n // flush the converter cache when the convertMessage prop changes\n const [converter, convertCallback] = useMemo(() => {\n const rscConverter =\n adapter.convertMessage ?? ((m: T) => m as VercelRSCMessage);\n const convertCallback: ConverterCallback<T> = (m, cache) => {\n if (cache) return cache;\n return vercelToThreadMessage(rscConverter, m);\n };\n return [new ThreadMessageConverter(), convertCallback];\n }, [adapter.convertMessage]);\n\n useEffect(() => {\n updateData(converter.convertMessages(adapter.messages, convertCallback));\n }, [updateData, converter, convertCallback, adapter.messages]);\n};\n","import type { ThreadMessage } from \"@assistant-ui/react\";\n\nexport type ConverterCallback<TIn> = (\n message: TIn,\n cache: ThreadMessage | undefined,\n) => ThreadMessage;\n\nexport class ThreadMessageConverter {\n private readonly cache = new WeakMap<WeakKey, ThreadMessage>();\n\n convertMessages<TIn extends WeakKey>(\n messages: TIn[],\n converter: ConverterCallback<TIn>,\n keyMapper: (m: TIn) => WeakKey = (key) => key,\n ): ThreadMessage[] {\n return messages.map((m) => {\n const key = keyMapper(m);\n const cached = this.cache.get(key);\n const newMessage = converter(m, cached);\n this.cache.set(key, newMessage);\n return newMessage;\n });\n }\n}\n","import type { ThreadMessage } from \"@assistant-ui/react\";\n\nexport const symbolInnerRSCMessage = Symbol(\"innerVercelRSCMessage\");\n\nexport type VercelRSCThreadMessage<T> = ThreadMessage & {\n [symbolInnerRSCMessage]?: T;\n};\n\nexport const getVercelRSCMessage = <T,>(message: ThreadMessage) => {\n return (message as VercelRSCThreadMessage<T>)[symbolInnerRSCMessage];\n};\n","import type { UseChatHelpers } from \"@ai-sdk/react\";\nimport { useEffect, useInsertionEffect, useState } from \"react\";\nimport { VercelAIRuntime } from \"../VercelAIRuntime\";\n\nexport const useVercelUseChatRuntime = (chatHelpers: UseChatHelpers) => {\n const [runtime] = useState(() => new VercelAIRuntime(chatHelpers));\n\n useInsertionEffect(() => {\n runtime.vercel = chatHelpers;\n });\n useEffect(() => {\n runtime.onVercelUpdated();\n });\n\n return runtime;\n};\n","import {\n type AppendMessage,\n type AssistantRuntime,\n unstable_MessageRepository as MessageRepository,\n type ReactThreadRuntime,\n type ThreadMessage,\n} from \"@assistant-ui/react\";\nimport type { Message } from \"ai\";\nimport { type StoreApi, type UseBoundStore, create } from \"zustand\";\nimport { getVercelAIMessage } from \"./getVercelAIMessage\";\nimport type { VercelHelpers } from \"./utils/VercelHelpers\";\nimport { sliceMessagesUntil } from \"./utils/sliceMessagesUntil\";\nimport { useVercelAIComposerSync } from \"./utils/useVercelAIComposerSync\";\nimport { useVercelAIThreadSync } from \"./utils/useVercelAIThreadSync\";\n\nconst hasUpcomingMessage = (isRunning: boolean, messages: ThreadMessage[]) => {\n return isRunning && messages[messages.length - 1]?.role !== \"assistant\";\n};\n\nexport class VercelAIRuntime implements AssistantRuntime, ReactThreadRuntime {\n private _subscriptions = new Set<() => void>();\n private repository = new MessageRepository();\n private assistantOptimisticId: string | null = null;\n\n private useVercel: UseBoundStore<StoreApi<{ vercel: VercelHelpers }>>;\n\n public messages: ThreadMessage[] = [];\n public isRunning = false;\n\n constructor(public vercel: VercelHelpers) {\n this.useVercel = create(() => ({\n vercel,\n }));\n }\n\n public getBranches(messageId: string): string[] {\n return this.repository.getBranches(messageId);\n }\n\n public switchToBranch(branchId: string): void {\n this.repository.switchToBranch(branchId);\n this.updateVercelMessages(this.repository.getMessages());\n }\n\n public async append(message: AppendMessage): Promise<void> {\n // add user message\n if (message.content.length !== 1 || message.content[0]?.type !== \"text\")\n throw new Error(\"Only text content is supported by Vercel AI SDK.\");\n\n const newMessages = sliceMessagesUntil(\n this.vercel.messages,\n message.parentId,\n );\n this.vercel.setMessages(newMessages);\n\n await this.vercel.append({\n role: \"user\",\n content: message.content[0].text,\n });\n }\n\n public async startRun(parentId: string | null): Promise<void> {\n const reloadMaybe =\n \"reload\" in this.vercel ? this.vercel.reload : undefined;\n if (!reloadMaybe)\n throw new Error(\n \"Reload is not supported by Vercel AI SDK's useAssistant.\",\n );\n\n const newMessages = sliceMessagesUntil(this.vercel.messages, parentId);\n this.vercel.setMessages(newMessages);\n\n await reloadMaybe();\n }\n\n public cancelRun(): void {\n const previousMessage = this.vercel.messages.at(-1);\n\n this.vercel.stop();\n\n if (this.assistantOptimisticId) {\n this.repository.deleteMessage(this.assistantOptimisticId);\n this.assistantOptimisticId = null;\n }\n\n let messages = this.repository.getMessages();\n if (\n previousMessage?.role === \"user\" &&\n previousMessage.id === messages.at(-1)?.id // ensure the previous message is a leaf node\n ) {\n this.vercel.setInput(previousMessage.content);\n this.repository.deleteMessage(previousMessage.id);\n\n messages = this.repository.getMessages();\n }\n\n // resync messages\n setTimeout(() => {\n this.updateVercelMessages(messages);\n }, 0);\n }\n\n public subscribe(callback: () => void): () => void {\n this._subscriptions.add(callback);\n return () => this._subscriptions.delete(callback);\n }\n\n private updateVercelMessages = (messages: ThreadMessage[]) => {\n this.vercel.setMessages(\n messages\n .flatMap(getVercelAIMessage)\n .filter((m): m is Message => m != null),\n );\n };\n\n public onVercelUpdated() {\n if (this.useVercel.getState().vercel !== this.vercel) {\n this.useVercel.setState({ vercel: this.vercel });\n }\n }\n\n private updateData = (isRunning: boolean, vm: ThreadMessage[]) => {\n for (let i = 0; i < vm.length; i++) {\n const message = vm[i]!;\n const parent = vm[i - 1];\n this.repository.addOrUpdateMessage(parent?.id ?? null, message);\n }\n\n if (this.assistantOptimisticId) {\n this.repository.deleteMessage(this.assistantOptimisticId);\n this.assistantOptimisticId = null;\n }\n\n if (hasUpcomingMessage(isRunning, vm)) {\n this.assistantOptimisticId = this.repository.appendOptimisticMessage(\n vm.at(-1)?.id ?? null,\n {\n role: \"assistant\",\n content: [{ type: \"text\", text: \"\" }],\n },\n );\n }\n\n this.repository.resetHead(\n this.assistantOptimisticId ?? vm.at(-1)?.id ?? null,\n );\n\n this.messages = this.repository.getMessages();\n this.isRunning = isRunning;\n\n for (const callback of this._subscriptions) callback();\n };\n\n unstable_synchronizer = () => {\n const { vercel } = this.useVercel();\n\n useVercelAIThreadSync(vercel, this.updateData);\n useVercelAIComposerSync(vercel);\n\n return null;\n };\n}\n","import type { ThreadMessage } from \"@assistant-ui/react\";\nimport type { Message } from \"ai\";\n\nexport const symbolInnerAIMessage = Symbol(\"innerVercelAIUIMessage\");\n\nexport type VercelAIThreadMessage = ThreadMessage & {\n [symbolInnerAIMessage]?: Message[];\n};\n\nexport const getVercelAIMessage = (message: ThreadMessage) => {\n return (message as VercelAIThreadMessage)[symbolInnerAIMessage];\n};\n","import type { Message } from \"ai\";\n\nexport const sliceMessagesUntil = (\n messages: Message[],\n messageId: string | null,\n) => {\n if (messageId == null) return [];\n\n let messageIdx = messages.findIndex((m) => m.id === messageId);\n if (messageIdx === -1)\n throw new Error(\n \"useVercelAIThreadState: Message not found. This is liekly an internal bug in assistant-ui.\",\n );\n\n while (messages[messageIdx + 1]?.role === \"assistant\") {\n messageIdx++;\n }\n\n return messages.slice(0, messageIdx + 1);\n};\n","import { unstable_useThreadContext as useThreadContext } from \"@assistant-ui/react\";\nimport { useEffect } from \"react\";\nimport type { VercelHelpers } from \"./VercelHelpers\";\n\n// two way sync between vercel helpers input state and composer text state\nexport const useVercelAIComposerSync = (vercel: VercelHelpers) => {\n const { useComposer } = useThreadContext();\n\n useEffect(() => {\n useComposer.setState({\n value: vercel.input,\n setValue: vercel.setInput,\n });\n }, [useComposer, vercel.input, vercel.setInput]);\n};\n","import type {\n TextContentPart,\n ThreadMessage,\n unstable_ToolCallContentPart as ToolCallContentPart,\n} from \"@assistant-ui/react\";\nimport type { Message } from \"ai\";\nimport { useEffect, useMemo } from \"react\";\nimport {\n type ConverterCallback,\n ThreadMessageConverter,\n} from \"../../utils/ThreadMessageConverter\";\nimport {\n type VercelAIThreadMessage,\n symbolInnerAIMessage,\n} from \"../getVercelAIMessage\";\nimport type { VercelHelpers } from \"./VercelHelpers\";\n\nconst getIsRunning = (vercel: VercelHelpers) => {\n if (\"isLoading\" in vercel) return vercel.isLoading;\n return vercel.status === \"in_progress\";\n};\n\nconst vercelToThreadMessage = (\n messages: Message[],\n status: \"in_progress\" | \"done\" | \"error\",\n): VercelAIThreadMessage => {\n const firstMessage = messages[0];\n if (!firstMessage) throw new Error(\"No messages found\");\n\n const common = {\n id: firstMessage.id,\n createdAt: firstMessage.createdAt ?? new Date(),\n [symbolInnerAIMessage]: messages,\n };\n\n switch (firstMessage.role) {\n case \"user\":\n if (messages.length > 1) {\n throw new Error(\n \"Multiple user messages found. This is likely an internal bug in assistant-ui.\",\n );\n }\n\n return {\n ...common,\n role: \"user\",\n content: [{ type: \"text\", text: firstMessage.content }],\n };\n case \"assistant\":\n return {\n ...common,\n role: \"assistant\",\n content: messages.flatMap((message) => [\n ...(message.content\n ? [{ type: \"text\", text: message.content } as TextContentPart]\n : []),\n ...(message.toolInvocations?.map(\n (t) =>\n ({\n type: \"tool-call\",\n name: t.toolName,\n args: t.args,\n result: \"result\" in t ? t.result : undefined,\n }) as ToolCallContentPart,\n ) ?? []),\n ]),\n status,\n };\n default:\n throw new Error(\n `You have a message with an unsupported role. The role ${firstMessage.role} is not supported.`,\n );\n }\n};\n\ntype Chunk = [Message, ...Message[]];\nconst hasItems = (messages: Message[]): messages is Chunk =>\n messages.length > 0;\n\nconst chunkedMessages = (messages: Message[]): Chunk[] => {\n const chunks: Chunk[] = [];\n let currentChunk: Message[] = [];\n\n for (const message of messages) {\n if (message.role === \"assistant\") {\n currentChunk.push(message);\n } else {\n if (hasItems(currentChunk)) {\n chunks.push(currentChunk);\n currentChunk = [];\n }\n chunks.push([message]);\n }\n }\n\n if (hasItems(currentChunk)) {\n chunks.push(currentChunk);\n }\n\n return chunks;\n};\n\nconst shallowArrayEqual = (a: unknown[], b: unknown[]) => {\n if (a.length !== b.length) return false;\n for (let i = 0; i < a.length; i++) {\n if (a[i] !== b[i]) return false;\n }\n return true;\n};\n\ntype UpdateDataCallback = (isRunning: boolean, vm: ThreadMessage[]) => void;\n\nexport const useVercelAIThreadSync = (\n vercel: VercelHelpers,\n updateData: UpdateDataCallback,\n) => {\n const isRunning = getIsRunning(vercel);\n\n const converter = useMemo(() => new ThreadMessageConverter(), []);\n\n useEffect(() => {\n const lastMessageId = vercel.messages.at(-1)?.id;\n const convertCallback: ConverterCallback<Chunk> = (messages, cache) => {\n const status =\n lastMessageId === messages[0].id && isRunning ? \"in_progress\" : \"done\";\n\n if (\n cache &&\n shallowArrayEqual(cache.content, messages) &&\n (cache.role === \"user\" || cache.status === status)\n )\n return cache;\n\n return vercelToThreadMessage(messages, status);\n };\n\n const messages = converter.convertMessages(\n chunkedMessages(vercel.messages),\n convertCallback,\n (m) => m[0],\n );\n\n updateData(isRunning, messages);\n }, [updateData, isRunning, vercel.messages, converter]);\n};\n","import type { UseAssistantHelpers } from \"@ai-sdk/react\";\nimport { useEffect, useInsertionEffect, useState } from \"react\";\nimport { VercelAIRuntime } from \"../VercelAIRuntime\";\n\nexport const useVercelUseAssistantRuntime = (\n assistantHelpers: UseAssistantHelpers,\n) => {\n const [runtime] = useState(() => new VercelAIRuntime(assistantHelpers));\n\n useInsertionEffect(() => {\n runtime.vercel = assistantHelpers;\n });\n useEffect(() => {\n runtime.onVercelUpdated();\n });\n\n return runtime;\n};\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;;;ACSA,gBAA+C;AAExC,IAAM,qBAAN,MAAqD;AAAA,EAC1D,YAA6B,OAAsB;AAAtB;AAAA,EAAuB;AAAA,EAEpD,MAAM,IAAI,EAAE,UAAU,aAAa,SAAS,GAAwB;AAClE,UAAM,EAAE,WAAW,IAAI,UAAM,sBAAW;AAAA,MACtC,OAAO,KAAK;AAAA,MACZ;AAAA,MACA,UAAU,SAAS,IAAI,CAAC,OAAO;AAAA,QAC7B,MAAM,EAAE;AAAA,QACR,SAAS,EAAE,QAAQ,OAAO,CAAC,MAA4B,EAAE,SAAS,IAAI;AAAA,MACxE,EAAE;AAAA,IACJ,CAAC;AAED,UAAM,UAAkC,CAAC;AACzC,qBAAiB,UAAU,YAAY;AACrC,cAAQ,OAAO,MAAM;AAAA,QACnB,KAAK,cAAc;AACjB,cAAI,OAAO,QAAQ,GAAG,EAAE;AACxB,cAAI,CAAC,QAAQ,KAAK,SAAS,QAAQ;AACjC,mBAAO,EAAE,MAAM,QAAQ,MAAM,GAAG;AAChC,oBAAQ,KAAK,IAAI;AAAA,UACnB;AACA,eAAK,QAAQ,OAAO;AACpB;AAAA,QACF;AAAA,QAEA,KAAK,aAAa;AAChB,kBAAQ,KAAK;AAAA,YACX,MAAM;AAAA,YACN,MAAM,OAAO;AAAA,YACb,MAAM,OAAO;AAAA,UACf,CAAC;AACD;AAAA,QACF;AAAA,MACF;AAEA,eAAS,EAAE,QAAQ,CAAC;AAAA,IACtB;AAEA,WAAO,EAAE,QAAQ;AAAA,EACnB;AACF;;;AClDA,IAAAA,gBAAwD;;;ACMxD,qBAA0D;;;ACP1D,mBAAmC;;;ACM5B,IAAM,yBAAN,MAA6B;AAAA,EACjB,QAAQ,oBAAI,QAAgC;AAAA,EAE7D,gBACE,UACA,WACA,YAAiC,CAAC,QAAQ,KACzB;AACjB,WAAO,SAAS,IAAI,CAAC,MAAM;AACzB,YAAM,MAAM,UAAU,CAAC;AACvB,YAAM,SAAS,KAAK,MAAM,IAAI,GAAG;AACjC,YAAM,aAAa,UAAU,GAAG,MAAM;AACtC,WAAK,MAAM,IAAI,KAAK,UAAU;AAC9B,aAAO;AAAA,IACT,CAAC;AAAA,EACH;AACF;;;ACrBO,IAAM,wBAAwB,OAAO,uBAAuB;AAM5D,IAAM,sBAAsB,CAAK,YAA2B;AACjE,SAAQ,QAAsC,qBAAqB;AACrE;;;AFGA,IAAM,wBAAwB,CAC5B,WACA,eAC8B;AAC9B,QAAM,UAAU,UAAU,UAAU;AAEpC,SAAO;AAAA,IACL,IAAI,QAAQ;AAAA,IACZ,MAAM,QAAQ;AAAA,IACd,SAAS,CAAC,EAAE,MAAM,MAAM,SAAS,QAAQ,QAAQ,CAAC;AAAA,IAClD,WAAW,QAAQ,aAAa,oBAAI,KAAK;AAAA,IACzC,GAAG,EAAE,QAAQ,OAAO;AAAA,IACpB,CAAC,qBAAqB,GAAG;AAAA,EAC3B;AACF;AAIO,IAAM,mBAAmB,CAC9B,SACA,eACG;AAEH,QAAM,CAAC,WAAW,eAAe,QAAI,sBAAQ,MAAM;AACjD,UAAM,eACJ,QAAQ,mBAAmB,CAAC,MAAS;AACvC,UAAMC,mBAAwC,CAAC,GAAG,UAAU;AAC1D,UAAI,MAAO,QAAO;AAClB,aAAO,sBAAsB,cAAc,CAAC;AAAA,IAC9C;AACA,WAAO,CAAC,IAAI,uBAAuB,GAAGA,gBAAe;AAAA,EACvD,GAAG,CAAC,QAAQ,cAAc,CAAC;AAE3B,8BAAU,MAAM;AACd,eAAW,UAAU,gBAAgB,QAAQ,UAAU,eAAe,CAAC;AAAA,EACzE,GAAG,CAAC,YAAY,WAAW,iBAAiB,QAAQ,QAAQ,CAAC;AAC/D;;;ADpCA,IAAM,iBAAmC,OAAO,OAAO,CAAC,CAAC;AAElD,IAAM,mBAAN,MAEP;AAAA,EAQE,YAAmB,SAA8B;AAA9B;AACjB,SAAK,iBAAa,uBAAO,OAAO;AAAA,MAC9B;AAAA,IACF,EAAE;AAAA,EACJ;AAAA,EAXQ;AAAA,EAEA,iBAAiB,oBAAI,IAAgB;AAAA,EAEtC,YAAY;AAAA,EACZ,WAA4B,CAAC;AAAA,EAQ5B,cAAc,CAAC,aAA+B;AACpD,SAAK,YAAY;AACjB,WAAO,SAAS,QAAQ,MAAM;AAC5B,WAAK,YAAY;AAAA,IACnB,CAAC;AAAA,EACH;AAAA,EAEO,cAAiC;AACtC,WAAO;AAAA,EACT;AAAA,EAEO,iBAAuB;AAC5B,UAAM,IAAI;AAAA,MACR;AAAA,IACF;AAAA,EACF;AAAA,EAEA,MAAa,OAAO,SAAuC;AACzD,QAAI,QAAQ,cAAc,KAAK,SAAS,GAAG,EAAE,GAAG,MAAM,OAAO;AAC3D,UAAI,CAAC,KAAK,QAAQ;AAChB,cAAM,IAAI;AAAA,UACR;AAAA,QACF;AACF,YAAM,KAAK,YAAY,KAAK,QAAQ,KAAK,OAAO,CAAC;AAAA,IACnD,OAAO;AACL,YAAM,KAAK,YAAY,KAAK,QAAQ,OAAO,OAAO,CAAC;AAAA,IACrD;AAAA,EACF;AAAA,EAEA,MAAa,SAAS,UAAwC;AAC5D,QAAI,CAAC,KAAK,QAAQ;AAChB,YAAM,IAAI;AAAA,QACR;AAAA,MACF;AACF,UAAM,KAAK,YAAY,KAAK,QAAQ,OAAO,QAAQ,CAAC;AAAA,EACtD;AAAA,EAEA,YAAkB;AAEhB,QAAI,QAAQ,IAAI,UAAU,MAAM,eAAe;AAC7C,cAAQ;AAAA,QACN;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAAA,EAEO,UAAU,UAAkC;AACjD,SAAK,eAAe,IAAI,QAAQ;AAChC,WAAO,MAAM,KAAK,eAAe,OAAO,QAAQ;AAAA,EAClD;AAAA,EAEO,mBAAmB;AACxB,QAAI,KAAK,WAAW,SAAS,EAAE,YAAY,KAAK,SAAS;AACvD,WAAK,WAAW,SAAS,EAAE,SAAS,KAAK,QAAQ,CAAC;AAAA,IACpD;AAAA,EACF;AAAA,EAEQ,aAAa,CAAC,aAA8B;AAClD,SAAK,WAAW;AAChB,eAAW,YAAY,KAAK,eAAgB,UAAS;AAAA,EACvD;AAAA,EAEA,wBAAwB,MAAM;AAC5B,UAAM,EAAE,QAAQ,IAAI,KAAK,WAAW;AAEpC,qBAAiB,SAAS,KAAK,UAAU;AAEzC,WAAO;AAAA,EACT;AACF;;;AD9FO,IAAM,sBAAsB,CACjC,YACG;AACH,QAAM,CAAC,OAAO,QAAI,wBAAS,MAAM,IAAI,iBAAiB,OAAO,CAAC;AAE9D,wCAAmB,MAAM;AACvB,YAAQ,UAAU;AAAA,EACpB,CAAC;AACD,+BAAU,MAAM;AACd,YAAQ,iBAAiB;AAAA,EAC3B,CAAC;AAED,SAAO;AACT;;;AKlBA,IAAAC,gBAAwD;;;ACDxD,IAAAC,gBAMO;AAEP,IAAAC,kBAA0D;;;ACLnD,IAAM,uBAAuB,OAAO,wBAAwB;AAM5D,IAAM,qBAAqB,CAAC,YAA2B;AAC5D,SAAQ,QAAkC,oBAAoB;AAChE;;;ACTO,IAAM,qBAAqB,CAChC,UACA,cACG;AACH,MAAI,aAAa,KAAM,QAAO,CAAC;AAE/B,MAAI,aAAa,SAAS,UAAU,CAAC,MAAM,EAAE,OAAO,SAAS;AAC7D,MAAI,eAAe;AACjB,UAAM,IAAI;AAAA,MACR;AAAA,IACF;AAEF,SAAO,SAAS,aAAa,CAAC,GAAG,SAAS,aAAa;AACrD;AAAA,EACF;AAEA,SAAO,SAAS,MAAM,GAAG,aAAa,CAAC;AACzC;;;ACnBA,IAAAC,gBAA8D;AAC9D,IAAAA,gBAA0B;AAInB,IAAM,0BAA0B,CAAC,WAA0B;AAChE,QAAM,EAAE,YAAY,QAAI,cAAAC,2BAAiB;AAEzC,+BAAU,MAAM;AACd,gBAAY,SAAS;AAAA,MACnB,OAAO,OAAO;AAAA,MACd,UAAU,OAAO;AAAA,IACnB,CAAC;AAAA,EACH,GAAG,CAAC,aAAa,OAAO,OAAO,OAAO,QAAQ,CAAC;AACjD;;;ACRA,IAAAC,gBAAmC;AAWnC,IAAM,eAAe,CAAC,WAA0B;AAC9C,MAAI,eAAe,OAAQ,QAAO,OAAO;AACzC,SAAO,OAAO,WAAW;AAC3B;AAEA,IAAMC,yBAAwB,CAC5B,UACA,WAC0B;AAC1B,QAAM,eAAe,SAAS,CAAC;AAC/B,MAAI,CAAC,aAAc,OAAM,IAAI,MAAM,mBAAmB;AAEtD,QAAM,SAAS;AAAA,IACb,IAAI,aAAa;AAAA,IACjB,WAAW,aAAa,aAAa,oBAAI,KAAK;AAAA,IAC9C,CAAC,oBAAoB,GAAG;AAAA,EAC1B;AAEA,UAAQ,aAAa,MAAM;AAAA,IACzB,KAAK;AACH,UAAI,SAAS,SAAS,GAAG;AACvB,cAAM,IAAI;AAAA,UACR;AAAA,QACF;AAAA,MACF;AAEA,aAAO;AAAA,QACL,GAAG;AAAA,QACH,MAAM;AAAA,QACN,SAAS,CAAC,EAAE,MAAM,QAAQ,MAAM,aAAa,QAAQ,CAAC;AAAA,MACxD;AAAA,IACF,KAAK;AACH,aAAO;AAAA,QACL,GAAG;AAAA,QACH,MAAM;AAAA,QACN,SAAS,SAAS,QAAQ,CAAC,YAAY;AAAA,UACrC,GAAI,QAAQ,UACR,CAAC,EAAE,MAAM,QAAQ,MAAM,QAAQ,QAAQ,CAAoB,IAC3D,CAAC;AAAA,UACL,GAAI,QAAQ,iBAAiB;AAAA,YAC3B,CAAC,OACE;AAAA,cACC,MAAM;AAAA,cACN,MAAM,EAAE;AAAA,cACR,MAAM,EAAE;AAAA,cACR,QAAQ,YAAY,IAAI,EAAE,SAAS;AAAA,YACrC;AAAA,UACJ,KAAK,CAAC;AAAA,QACR,CAAC;AAAA,QACD;AAAA,MACF;AAAA,IACF;AACE,YAAM,IAAI;AAAA,QACR,yDAAyD,aAAa,IAAI;AAAA,MAC5E;AAAA,EACJ;AACF;AAGA,IAAM,WAAW,CAAC,aAChB,SAAS,SAAS;AAEpB,IAAM,kBAAkB,CAAC,aAAiC;AACxD,QAAM,SAAkB,CAAC;AACzB,MAAI,eAA0B,CAAC;AAE/B,aAAW,WAAW,UAAU;AAC9B,QAAI,QAAQ,SAAS,aAAa;AAChC,mBAAa,KAAK,OAAO;AAAA,IAC3B,OAAO;AACL,UAAI,SAAS,YAAY,GAAG;AAC1B,eAAO,KAAK,YAAY;AACxB,uBAAe,CAAC;AAAA,MAClB;AACA,aAAO,KAAK,CAAC,OAAO,CAAC;AAAA,IACvB;AAAA,EACF;AAEA,MAAI,SAAS,YAAY,GAAG;AAC1B,WAAO,KAAK,YAAY;AAAA,EAC1B;AAEA,SAAO;AACT;AAEA,IAAM,oBAAoB,CAAC,GAAc,MAAiB;AACxD,MAAI,EAAE,WAAW,EAAE,OAAQ,QAAO;AAClC,WAAS,IAAI,GAAG,IAAI,EAAE,QAAQ,KAAK;AACjC,QAAI,EAAE,CAAC,MAAM,EAAE,CAAC,EAAG,QAAO;AAAA,EAC5B;AACA,SAAO;AACT;AAIO,IAAM,wBAAwB,CACnC,QACA,eACG;AACH,QAAM,YAAY,aAAa,MAAM;AAErC,QAAM,gBAAY,uBAAQ,MAAM,IAAI,uBAAuB,GAAG,CAAC,CAAC;AAEhE,+BAAU,MAAM;AACd,UAAM,gBAAgB,OAAO,SAAS,GAAG,EAAE,GAAG;AAC9C,UAAM,kBAA4C,CAACC,WAAU,UAAU;AACrE,YAAM,SACJ,kBAAkBA,UAAS,CAAC,EAAE,MAAM,YAAY,gBAAgB;AAElE,UACE,SACA,kBAAkB,MAAM,SAASA,SAAQ,MACxC,MAAM,SAAS,UAAU,MAAM,WAAW;AAE3C,eAAO;AAET,aAAOD,uBAAsBC,WAAU,MAAM;AAAA,IAC/C;AAEA,UAAM,WAAW,UAAU;AAAA,MACzB,gBAAgB,OAAO,QAAQ;AAAA,MAC/B;AAAA,MACA,CAAC,MAAM,EAAE,CAAC;AAAA,IACZ;AAEA,eAAW,WAAW,QAAQ;AAAA,EAChC,GAAG,CAAC,YAAY,WAAW,OAAO,UAAU,SAAS,CAAC;AACxD;;;AJjIA,IAAM,qBAAqB,CAAC,WAAoB,aAA8B;AAC5E,SAAO,aAAa,SAAS,SAAS,SAAS,CAAC,GAAG,SAAS;AAC9D;AAEO,IAAM,kBAAN,MAAsE;AAAA,EAU3E,YAAmB,QAAuB;AAAvB;AACjB,SAAK,gBAAY,wBAAO,OAAO;AAAA,MAC7B;AAAA,IACF,EAAE;AAAA,EACJ;AAAA,EAbQ,iBAAiB,oBAAI,IAAgB;AAAA,EACrC,aAAa,IAAI,cAAAC,2BAAkB;AAAA,EACnC,wBAAuC;AAAA,EAEvC;AAAA,EAED,WAA4B,CAAC;AAAA,EAC7B,YAAY;AAAA,EAQZ,YAAY,WAA6B;AAC9C,WAAO,KAAK,WAAW,YAAY,SAAS;AAAA,EAC9C;AAAA,EAEO,eAAe,UAAwB;AAC5C,SAAK,WAAW,eAAe,QAAQ;AACvC,SAAK,qBAAqB,KAAK,WAAW,YAAY,CAAC;AAAA,EACzD;AAAA,EAEA,MAAa,OAAO,SAAuC;AAEzD,QAAI,QAAQ,QAAQ,WAAW,KAAK,QAAQ,QAAQ,CAAC,GAAG,SAAS;AAC/D,YAAM,IAAI,MAAM,kDAAkD;AAEpE,UAAM,cAAc;AAAA,MAClB,KAAK,OAAO;AAAA,MACZ,QAAQ;AAAA,IACV;AACA,SAAK,OAAO,YAAY,WAAW;AAEnC,UAAM,KAAK,OAAO,OAAO;AAAA,MACvB,MAAM;AAAA,MACN,SAAS,QAAQ,QAAQ,CAAC,EAAE;AAAA,IAC9B,CAAC;AAAA,EACH;AAAA,EAEA,MAAa,SAAS,UAAwC;AAC5D,UAAM,cACJ,YAAY,KAAK,SAAS,KAAK,OAAO,SAAS;AACjD,QAAI,CAAC;AACH,YAAM,IAAI;AAAA,QACR;AAAA,MACF;AAEF,UAAM,cAAc,mBAAmB,KAAK,OAAO,UAAU,QAAQ;AACrE,SAAK,OAAO,YAAY,WAAW;AAEnC,UAAM,YAAY;AAAA,EACpB;AAAA,EAEO,YAAkB;AACvB,UAAM,kBAAkB,KAAK,OAAO,SAAS,GAAG,EAAE;AAElD,SAAK,OAAO,KAAK;AAEjB,QAAI,KAAK,uBAAuB;AAC9B,WAAK,WAAW,cAAc,KAAK,qBAAqB;AACxD,WAAK,wBAAwB;AAAA,IAC/B;AAEA,QAAI,WAAW,KAAK,WAAW,YAAY;AAC3C,QACE,iBAAiB,SAAS,UAC1B,gBAAgB,OAAO,SAAS,GAAG,EAAE,GAAG,IACxC;AACA,WAAK,OAAO,SAAS,gBAAgB,OAAO;AAC5C,WAAK,WAAW,cAAc,gBAAgB,EAAE;AAEhD,iBAAW,KAAK,WAAW,YAAY;AAAA,IACzC;AAGA,eAAW,MAAM;AACf,WAAK,qBAAqB,QAAQ;AAAA,IACpC,GAAG,CAAC;AAAA,EACN;AAAA,EAEO,UAAU,UAAkC;AACjD,SAAK,eAAe,IAAI,QAAQ;AAChC,WAAO,MAAM,KAAK,eAAe,OAAO,QAAQ;AAAA,EAClD;AAAA,EAEQ,uBAAuB,CAAC,aAA8B;AAC5D,SAAK,OAAO;AAAA,MACV,SACG,QAAQ,kBAAkB,EAC1B,OAAO,CAAC,MAAoB,KAAK,IAAI;AAAA,IAC1C;AAAA,EACF;AAAA,EAEO,kBAAkB;AACvB,QAAI,KAAK,UAAU,SAAS,EAAE,WAAW,KAAK,QAAQ;AACpD,WAAK,UAAU,SAAS,EAAE,QAAQ,KAAK,OAAO,CAAC;AAAA,IACjD;AAAA,EACF;AAAA,EAEQ,aAAa,CAAC,WAAoB,OAAwB;AAChE,aAAS,IAAI,GAAG,IAAI,GAAG,QAAQ,KAAK;AAClC,YAAM,UAAU,GAAG,CAAC;AACpB,YAAM,SAAS,GAAG,IAAI,CAAC;AACvB,WAAK,WAAW,mBAAmB,QAAQ,MAAM,MAAM,OAAO;AAAA,IAChE;AAEA,QAAI,KAAK,uBAAuB;AAC9B,WAAK,WAAW,cAAc,KAAK,qBAAqB;AACxD,WAAK,wBAAwB;AAAA,IAC/B;AAEA,QAAI,mBAAmB,WAAW,EAAE,GAAG;AACrC,WAAK,wBAAwB,KAAK,WAAW;AAAA,QAC3C,GAAG,GAAG,EAAE,GAAG,MAAM;AAAA,QACjB;AAAA,UACE,MAAM;AAAA,UACN,SAAS,CAAC,EAAE,MAAM,QAAQ,MAAM,GAAG,CAAC;AAAA,QACtC;AAAA,MACF;AAAA,IACF;AAEA,SAAK,WAAW;AAAA,MACd,KAAK,yBAAyB,GAAG,GAAG,EAAE,GAAG,MAAM;AAAA,IACjD;AAEA,SAAK,WAAW,KAAK,WAAW,YAAY;AAC5C,SAAK,YAAY;AAEjB,eAAW,YAAY,KAAK,eAAgB,UAAS;AAAA,EACvD;AAAA,EAEA,wBAAwB,MAAM;AAC5B,UAAM,EAAE,OAAO,IAAI,KAAK,UAAU;AAElC,0BAAsB,QAAQ,KAAK,UAAU;AAC7C,4BAAwB,MAAM;AAE9B,WAAO;AAAA,EACT;AACF;;;AD7JO,IAAM,0BAA0B,CAAC,gBAAgC;AACtE,QAAM,CAAC,OAAO,QAAI,wBAAS,MAAM,IAAI,gBAAgB,WAAW,CAAC;AAEjE,wCAAmB,MAAM;AACvB,YAAQ,SAAS;AAAA,EACnB,CAAC;AACD,+BAAU,MAAM;AACd,YAAQ,gBAAgB;AAAA,EAC1B,CAAC;AAED,SAAO;AACT;;;AMdA,IAAAC,gBAAwD;AAGjD,IAAM,+BAA+B,CAC1C,qBACG;AACH,QAAM,CAAC,OAAO,QAAI,wBAAS,MAAM,IAAI,gBAAgB,gBAAgB,CAAC;AAEtE,wCAAmB,MAAM;AACvB,YAAQ,SAAS;AAAA,EACnB,CAAC;AACD,+BAAU,MAAM;AACd,YAAQ,gBAAgB;AAAA,EAC1B,CAAC;AAED,SAAO;AACT;","names":["import_react","convertCallback","import_react","import_react","import_zustand","import_react","useThreadContext","import_react","vercelToThreadMessage","messages","MessageRepository","import_react"]}