@assistant-ui/react 0.1.2 → 0.1.3
Sign up to get free protection for your applications and to get access to all the features.
- package/package.json +3 -2
- package/dist/Thread-By2aBNAG.d.mts +0 -67
- package/dist/Thread-By2aBNAG.d.ts +0 -67
- package/dist/chunk-SCWPIDP6.mjs +0 -83
- package/dist/chunk-SCWPIDP6.mjs.map +0 -1
- package/dist/experimental.d.mts +0 -85
- package/dist/experimental.d.ts +0 -85
- package/dist/experimental.js +0 -123
- package/dist/experimental.js.map +0 -1
- package/dist/experimental.mjs +0 -33
- package/dist/experimental.mjs.map +0 -1
- package/dist/index.d.mts +0 -316
- package/dist/index.d.ts +0 -316
- package/dist/index.js +0 -1509
- package/dist/index.js.map +0 -1
- package/dist/index.mjs +0 -1440
- package/dist/index.mjs.map +0 -1
package/dist/index.js
DELETED
@@ -1,1509 +0,0 @@
|
|
1
|
-
"use strict";
|
2
|
-
var __create = Object.create;
|
3
|
-
var __defProp = Object.defineProperty;
|
4
|
-
var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
|
5
|
-
var __getOwnPropNames = Object.getOwnPropertyNames;
|
6
|
-
var __getProtoOf = Object.getPrototypeOf;
|
7
|
-
var __hasOwnProp = Object.prototype.hasOwnProperty;
|
8
|
-
var __export = (target, all) => {
|
9
|
-
for (var name in all)
|
10
|
-
__defProp(target, name, { get: all[name], enumerable: true });
|
11
|
-
};
|
12
|
-
var __copyProps = (to, from, except, desc) => {
|
13
|
-
if (from && typeof from === "object" || typeof from === "function") {
|
14
|
-
for (let key of __getOwnPropNames(from))
|
15
|
-
if (!__hasOwnProp.call(to, key) && key !== except)
|
16
|
-
__defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
|
17
|
-
}
|
18
|
-
return to;
|
19
|
-
};
|
20
|
-
var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__getProtoOf(mod)) : {}, __copyProps(
|
21
|
-
// If the importer is in node compatibility mode or this is not an ESM
|
22
|
-
// file that has been converted to a CommonJS file using a Babel-
|
23
|
-
// compatible transform (i.e. "__esModule" has not been set), then set
|
24
|
-
// "default" to the CommonJS "module.exports" for node compatibility.
|
25
|
-
isNodeMode || !mod || !mod.__esModule ? __defProp(target, "default", { value: mod, enumerable: true }) : target,
|
26
|
-
mod
|
27
|
-
));
|
28
|
-
var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
|
29
|
-
|
30
|
-
// src/index.ts
|
31
|
-
var src_exports = {};
|
32
|
-
__export(src_exports, {
|
33
|
-
ActionBarPrimitive: () => actionBar_exports,
|
34
|
-
AssistantRuntimeProvider: () => AssistantRuntimeProvider,
|
35
|
-
BranchPickerPrimitive: () => branchPicker_exports,
|
36
|
-
ComposerPrimitive: () => composer_exports,
|
37
|
-
ContentPartPrimitive: () => contentPart_exports,
|
38
|
-
INTERNAL: () => internal_exports,
|
39
|
-
MessagePrimitive: () => message_exports,
|
40
|
-
ThreadPrimitive: () => thread_exports,
|
41
|
-
useBeginMessageEdit: () => useBeginMessageEdit,
|
42
|
-
useCopyMessage: () => useCopyMessage,
|
43
|
-
useGoToNextBranch: () => useGoToNextBranch,
|
44
|
-
useGoToPreviousBranch: () => useGoToPreviousBranch,
|
45
|
-
useLocalRuntime: () => useLocalRuntime,
|
46
|
-
useReloadMessage: () => useReloadMessage
|
47
|
-
});
|
48
|
-
module.exports = __toCommonJS(src_exports);
|
49
|
-
|
50
|
-
// src/actions/useCopyMessage.tsx
|
51
|
-
var import_react4 = require("react");
|
52
|
-
|
53
|
-
// src/context/MessageContext.ts
|
54
|
-
var import_react = require("react");
|
55
|
-
var MessageContext = (0, import_react.createContext)(null);
|
56
|
-
var useMessageContext = () => {
|
57
|
-
const context = (0, import_react.useContext)(MessageContext);
|
58
|
-
if (!context)
|
59
|
-
throw new Error(
|
60
|
-
"This component can only be used inside a component passed to <ThreadPrimitive.Messages components={...} />."
|
61
|
-
);
|
62
|
-
return context;
|
63
|
-
};
|
64
|
-
|
65
|
-
// src/utils/combined/useCombinedStore.ts
|
66
|
-
var import_react3 = require("react");
|
67
|
-
|
68
|
-
// src/utils/combined/createCombinedStore.ts
|
69
|
-
var import_react2 = require("react");
|
70
|
-
var createCombinedStore = (stores) => {
|
71
|
-
const subscribe = (callback) => {
|
72
|
-
const unsubscribes = stores.map((store) => store.subscribe(callback));
|
73
|
-
return () => {
|
74
|
-
for (const unsub of unsubscribes) {
|
75
|
-
unsub();
|
76
|
-
}
|
77
|
-
};
|
78
|
-
};
|
79
|
-
return (selector) => {
|
80
|
-
const getSnapshot = () => selector(...stores.map((store) => store.getState()));
|
81
|
-
return (0, import_react2.useSyncExternalStore)(subscribe, getSnapshot, getSnapshot);
|
82
|
-
};
|
83
|
-
};
|
84
|
-
|
85
|
-
// src/utils/combined/useCombinedStore.ts
|
86
|
-
var useCombinedStore = (stores, selector) => {
|
87
|
-
const useCombined = (0, import_react3.useMemo)(() => createCombinedStore(stores), stores);
|
88
|
-
return useCombined(selector);
|
89
|
-
};
|
90
|
-
|
91
|
-
// src/utils/getMessageText.tsx
|
92
|
-
var getMessageText = (message) => {
|
93
|
-
const textParts = message.content.filter(
|
94
|
-
(part) => part.type === "text"
|
95
|
-
);
|
96
|
-
return textParts.map((part) => part.text).join("\n\n");
|
97
|
-
};
|
98
|
-
|
99
|
-
// src/actions/useCopyMessage.tsx
|
100
|
-
var useCopyMessage = ({ copiedDuration = 3e3 }) => {
|
101
|
-
const { useMessage, useComposer } = useMessageContext();
|
102
|
-
const hasCopyableContent = useCombinedStore(
|
103
|
-
[useMessage, useComposer],
|
104
|
-
(m, c) => {
|
105
|
-
return c.isEditing || m.message.content.some((c2) => c2.type === "text");
|
106
|
-
}
|
107
|
-
);
|
108
|
-
const callback = (0, import_react4.useCallback)(() => {
|
109
|
-
const { isEditing, value: composerValue } = useComposer.getState();
|
110
|
-
const { message, setIsCopied } = useMessage.getState();
|
111
|
-
const valueToCopy = isEditing ? composerValue : getMessageText(message);
|
112
|
-
navigator.clipboard.writeText(valueToCopy);
|
113
|
-
setIsCopied(true);
|
114
|
-
setTimeout(() => setIsCopied(false), copiedDuration);
|
115
|
-
}, [useComposer, useMessage, copiedDuration]);
|
116
|
-
if (!hasCopyableContent) return null;
|
117
|
-
return callback;
|
118
|
-
};
|
119
|
-
|
120
|
-
// src/actions/useReloadMessage.tsx
|
121
|
-
var import_react6 = require("react");
|
122
|
-
|
123
|
-
// src/context/ThreadContext.ts
|
124
|
-
var import_react5 = require("react");
|
125
|
-
var ThreadContext = (0, import_react5.createContext)(null);
|
126
|
-
var useThreadContext = () => {
|
127
|
-
const context = (0, import_react5.useContext)(ThreadContext);
|
128
|
-
if (!context)
|
129
|
-
throw new Error("This component must be used within an AssistantRuntimeProvider.");
|
130
|
-
return context;
|
131
|
-
};
|
132
|
-
|
133
|
-
// src/actions/useReloadMessage.tsx
|
134
|
-
var useReloadMessage = () => {
|
135
|
-
const { useThread, useViewport } = useThreadContext();
|
136
|
-
const { useMessage } = useMessageContext();
|
137
|
-
const disabled = useCombinedStore(
|
138
|
-
[useThread, useMessage],
|
139
|
-
(t, m) => t.isRunning || m.message.role !== "assistant"
|
140
|
-
);
|
141
|
-
const callback = (0, import_react6.useCallback)(() => {
|
142
|
-
const { parentId } = useMessage.getState();
|
143
|
-
useThread.getState().startRun(parentId);
|
144
|
-
useViewport.getState().scrollToBottom();
|
145
|
-
}, [useMessage, useThread, useViewport]);
|
146
|
-
if (disabled) return null;
|
147
|
-
return callback;
|
148
|
-
};
|
149
|
-
|
150
|
-
// src/actions/useBeginMessageEdit.tsx
|
151
|
-
var import_react7 = require("react");
|
152
|
-
var useBeginMessageEdit = () => {
|
153
|
-
const { useMessage, useComposer } = useMessageContext();
|
154
|
-
const disabled = useCombinedStore(
|
155
|
-
[useMessage, useComposer],
|
156
|
-
(m, c) => m.message.role !== "user" || c.isEditing
|
157
|
-
);
|
158
|
-
const callback = (0, import_react7.useCallback)(() => {
|
159
|
-
const { edit } = useComposer.getState();
|
160
|
-
edit();
|
161
|
-
}, [useComposer]);
|
162
|
-
if (disabled) return null;
|
163
|
-
return callback;
|
164
|
-
};
|
165
|
-
|
166
|
-
// src/actions/useGoToNextBranch.tsx
|
167
|
-
var import_react8 = require("react");
|
168
|
-
var useGoToNextBranch = () => {
|
169
|
-
const { useThread } = useThreadContext();
|
170
|
-
const { useMessage, useComposer } = useMessageContext();
|
171
|
-
const disabled = useCombinedStore(
|
172
|
-
[useMessage, useComposer],
|
173
|
-
(m, c) => c.isEditing || m.branches.indexOf(m.message.id) + 1 >= m.branches.length
|
174
|
-
);
|
175
|
-
const callback = (0, import_react8.useCallback)(() => {
|
176
|
-
const { message, branches } = useMessage.getState();
|
177
|
-
useThread.getState().switchToBranch(branches[branches.indexOf(message.id) + 1]);
|
178
|
-
}, [useMessage, useThread]);
|
179
|
-
if (disabled) return null;
|
180
|
-
return callback;
|
181
|
-
};
|
182
|
-
|
183
|
-
// src/actions/useGoToPreviousBranch.tsx
|
184
|
-
var import_react9 = require("react");
|
185
|
-
var useGoToPreviousBranch = () => {
|
186
|
-
const { useThread } = useThreadContext();
|
187
|
-
const { useMessage, useComposer } = useMessageContext();
|
188
|
-
const disabled = useCombinedStore(
|
189
|
-
[useMessage, useComposer],
|
190
|
-
(m, c) => c.isEditing || m.branches.indexOf(m.message.id) <= 0
|
191
|
-
);
|
192
|
-
const callback = (0, import_react9.useCallback)(() => {
|
193
|
-
const { message, branches } = useMessage.getState();
|
194
|
-
useThread.getState().switchToBranch(branches[branches.indexOf(message.id) - 1]);
|
195
|
-
}, [useMessage, useThread]);
|
196
|
-
if (disabled) return null;
|
197
|
-
return callback;
|
198
|
-
};
|
199
|
-
|
200
|
-
// src/primitives/thread/index.ts
|
201
|
-
var thread_exports = {};
|
202
|
-
__export(thread_exports, {
|
203
|
-
Empty: () => ThreadEmpty,
|
204
|
-
If: () => ThreadIf,
|
205
|
-
Messages: () => ThreadMessages,
|
206
|
-
Root: () => ThreadRoot,
|
207
|
-
ScrollToBottom: () => ThreadScrollToBottom,
|
208
|
-
Suggestion: () => ThreadSuggestion,
|
209
|
-
Viewport: () => ThreadViewport
|
210
|
-
});
|
211
|
-
|
212
|
-
// src/primitives/thread/ThreadRoot.tsx
|
213
|
-
var import_react_primitive = require("@radix-ui/react-primitive");
|
214
|
-
var import_react10 = require("react");
|
215
|
-
var import_jsx_runtime = require("react/jsx-runtime");
|
216
|
-
var ThreadRoot = (0, import_react10.forwardRef)(
|
217
|
-
(props, ref) => {
|
218
|
-
return /* @__PURE__ */ (0, import_jsx_runtime.jsx)(import_react_primitive.Primitive.div, { ...props, ref });
|
219
|
-
}
|
220
|
-
);
|
221
|
-
|
222
|
-
// src/primitives/thread/ThreadIf.tsx
|
223
|
-
var useThreadIf = (props) => {
|
224
|
-
const { useThread } = useThreadContext();
|
225
|
-
return useThread((thread) => {
|
226
|
-
if (props.empty === true && thread.messages.length !== 0) return false;
|
227
|
-
if (props.empty === false && thread.messages.length === 0) return false;
|
228
|
-
if (props.running === true && !thread.isRunning) return false;
|
229
|
-
if (props.running === false && thread.isRunning) return false;
|
230
|
-
return true;
|
231
|
-
});
|
232
|
-
};
|
233
|
-
var ThreadIf = ({ children, ...query }) => {
|
234
|
-
const result = useThreadIf(query);
|
235
|
-
return result ? children : null;
|
236
|
-
};
|
237
|
-
|
238
|
-
// src/primitives/thread/ThreadEmpty.tsx
|
239
|
-
var import_jsx_runtime2 = require("react/jsx-runtime");
|
240
|
-
var ThreadEmpty = ({ children }) => {
|
241
|
-
return /* @__PURE__ */ (0, import_jsx_runtime2.jsx)(ThreadIf, { empty: true, children });
|
242
|
-
};
|
243
|
-
|
244
|
-
// src/primitives/thread/ThreadViewport.tsx
|
245
|
-
var import_primitive = require("@radix-ui/primitive");
|
246
|
-
var import_react_compose_refs = require("@radix-ui/react-compose-refs");
|
247
|
-
var import_react_primitive2 = require("@radix-ui/react-primitive");
|
248
|
-
var import_react13 = require("react");
|
249
|
-
|
250
|
-
// src/utils/hooks/useOnResizeContent.tsx
|
251
|
-
var import_react_use_callback_ref = require("@radix-ui/react-use-callback-ref");
|
252
|
-
var import_react11 = require("react");
|
253
|
-
var useOnResizeContent = (ref, callback) => {
|
254
|
-
const callbackRef = (0, import_react_use_callback_ref.useCallbackRef)(callback);
|
255
|
-
(0, import_react11.useEffect)(() => {
|
256
|
-
const el = ref.current;
|
257
|
-
if (!el) return;
|
258
|
-
const resizeObserver = new ResizeObserver(() => {
|
259
|
-
callbackRef();
|
260
|
-
});
|
261
|
-
const mutationObserver = new MutationObserver((mutations) => {
|
262
|
-
for (const mutation of mutations) {
|
263
|
-
for (const node of mutation.addedNodes) {
|
264
|
-
if (node instanceof Element) {
|
265
|
-
resizeObserver.observe(node);
|
266
|
-
}
|
267
|
-
}
|
268
|
-
for (const node of mutation.removedNodes) {
|
269
|
-
if (node instanceof Element) {
|
270
|
-
resizeObserver.unobserve(node);
|
271
|
-
}
|
272
|
-
}
|
273
|
-
}
|
274
|
-
callbackRef();
|
275
|
-
});
|
276
|
-
resizeObserver.observe(el);
|
277
|
-
mutationObserver.observe(el, { childList: true });
|
278
|
-
for (const child of el.children) {
|
279
|
-
resizeObserver.observe(child);
|
280
|
-
}
|
281
|
-
return () => {
|
282
|
-
resizeObserver.disconnect();
|
283
|
-
mutationObserver.disconnect();
|
284
|
-
};
|
285
|
-
}, [ref.current, callbackRef]);
|
286
|
-
};
|
287
|
-
|
288
|
-
// src/utils/hooks/useOnScrollToBottom.tsx
|
289
|
-
var import_react_use_callback_ref2 = require("@radix-ui/react-use-callback-ref");
|
290
|
-
var import_react12 = require("react");
|
291
|
-
var useOnScrollToBottom = (callback) => {
|
292
|
-
const callbackRef = (0, import_react_use_callback_ref2.useCallbackRef)(callback);
|
293
|
-
const { useViewport } = useThreadContext();
|
294
|
-
(0, import_react12.useEffect)(() => {
|
295
|
-
return useViewport.getState().onScrollToBottom(() => {
|
296
|
-
callbackRef();
|
297
|
-
});
|
298
|
-
}, [useViewport, callbackRef]);
|
299
|
-
};
|
300
|
-
|
301
|
-
// src/primitives/thread/ThreadViewport.tsx
|
302
|
-
var import_jsx_runtime3 = require("react/jsx-runtime");
|
303
|
-
var ThreadViewport = (0, import_react13.forwardRef)(({ autoScroll = true, onScroll, children, ...rest }, forwardedRef) => {
|
304
|
-
const messagesEndRef = (0, import_react13.useRef)(null);
|
305
|
-
const divRef = (0, import_react13.useRef)(null);
|
306
|
-
const ref = (0, import_react_compose_refs.useComposedRefs)(forwardedRef, divRef);
|
307
|
-
const { useViewport } = useThreadContext();
|
308
|
-
const firstRenderRef = (0, import_react13.useRef)(true);
|
309
|
-
const isScrollingToBottomRef = (0, import_react13.useRef)(false);
|
310
|
-
const lastScrollTop = (0, import_react13.useRef)(0);
|
311
|
-
const scrollToBottom = () => {
|
312
|
-
const div = messagesEndRef.current;
|
313
|
-
if (!div || !autoScroll) return;
|
314
|
-
const behavior = firstRenderRef.current ? "instant" : "auto";
|
315
|
-
firstRenderRef.current = false;
|
316
|
-
isScrollingToBottomRef.current = true;
|
317
|
-
div.scrollIntoView({ behavior });
|
318
|
-
};
|
319
|
-
useOnResizeContent(divRef, () => {
|
320
|
-
if (!isScrollingToBottomRef.current && !useViewport.getState().isAtBottom) {
|
321
|
-
handleScroll();
|
322
|
-
} else {
|
323
|
-
scrollToBottom();
|
324
|
-
}
|
325
|
-
});
|
326
|
-
useOnScrollToBottom(() => {
|
327
|
-
scrollToBottom();
|
328
|
-
});
|
329
|
-
const handleScroll = () => {
|
330
|
-
const div = divRef.current;
|
331
|
-
if (!div) return;
|
332
|
-
const isAtBottom = useViewport.getState().isAtBottom;
|
333
|
-
const newIsAtBottom = div.scrollHeight - div.scrollTop <= div.clientHeight;
|
334
|
-
if (!newIsAtBottom && lastScrollTop.current < div.scrollTop) {
|
335
|
-
} else if (newIsAtBottom !== isAtBottom) {
|
336
|
-
isScrollingToBottomRef.current = false;
|
337
|
-
useViewport.setState({ isAtBottom: newIsAtBottom });
|
338
|
-
}
|
339
|
-
lastScrollTop.current = div.scrollTop;
|
340
|
-
};
|
341
|
-
return /* @__PURE__ */ (0, import_jsx_runtime3.jsxs)(
|
342
|
-
import_react_primitive2.Primitive.div,
|
343
|
-
{
|
344
|
-
...rest,
|
345
|
-
onScroll: (0, import_primitive.composeEventHandlers)(onScroll, handleScroll),
|
346
|
-
ref,
|
347
|
-
children: [
|
348
|
-
children,
|
349
|
-
/* @__PURE__ */ (0, import_jsx_runtime3.jsx)("div", { ref: messagesEndRef })
|
350
|
-
]
|
351
|
-
}
|
352
|
-
);
|
353
|
-
});
|
354
|
-
|
355
|
-
// src/context/providers/MessageProvider.tsx
|
356
|
-
var import_react14 = require("react");
|
357
|
-
var import_zustand2 = require("zustand");
|
358
|
-
|
359
|
-
// src/context/stores/MessageComposer.ts
|
360
|
-
var import_zustand = require("zustand");
|
361
|
-
|
362
|
-
// src/context/stores/BaseComposer.ts
|
363
|
-
var makeBaseComposer = (set) => ({
|
364
|
-
value: "",
|
365
|
-
setValue: (value) => {
|
366
|
-
set({ value });
|
367
|
-
}
|
368
|
-
});
|
369
|
-
|
370
|
-
// src/context/stores/MessageComposer.ts
|
371
|
-
var makeEditComposerStore = ({
|
372
|
-
onEdit,
|
373
|
-
onSend
|
374
|
-
}) => (0, import_zustand.create)()((set, get, store) => ({
|
375
|
-
...makeBaseComposer(set, get, store),
|
376
|
-
isEditing: false,
|
377
|
-
edit: () => {
|
378
|
-
const value = onEdit();
|
379
|
-
set({ isEditing: true, value });
|
380
|
-
},
|
381
|
-
send: () => {
|
382
|
-
const value = get().value;
|
383
|
-
set({ isEditing: false });
|
384
|
-
onSend(value);
|
385
|
-
},
|
386
|
-
cancel: () => {
|
387
|
-
if (!get().isEditing) return false;
|
388
|
-
set({ isEditing: false });
|
389
|
-
return true;
|
390
|
-
}
|
391
|
-
}));
|
392
|
-
|
393
|
-
// src/context/providers/MessageProvider.tsx
|
394
|
-
var import_jsx_runtime4 = require("react/jsx-runtime");
|
395
|
-
var getIsLast = (thread, message) => {
|
396
|
-
return thread.messages[thread.messages.length - 1]?.id === message.id;
|
397
|
-
};
|
398
|
-
var syncMessage = (thread, useMessage, messageIndex) => {
|
399
|
-
const parentId = thread.messages[messageIndex - 1]?.id ?? null;
|
400
|
-
const message = thread.messages[messageIndex];
|
401
|
-
if (!message) return;
|
402
|
-
const isLast = getIsLast(thread, message);
|
403
|
-
const branches = thread.getBranches(message.id);
|
404
|
-
const currentState = useMessage.getState();
|
405
|
-
if (currentState.message === message && currentState.parentId === parentId && currentState.branches === branches && currentState.isLast === isLast)
|
406
|
-
return;
|
407
|
-
useMessage.setState({
|
408
|
-
message,
|
409
|
-
parentId,
|
410
|
-
branches,
|
411
|
-
isLast
|
412
|
-
});
|
413
|
-
};
|
414
|
-
var useMessageContext2 = (messageIndex) => {
|
415
|
-
const { useThread } = useThreadContext();
|
416
|
-
const [context] = (0, import_react14.useState)(() => {
|
417
|
-
const useMessage = (0, import_zustand2.create)((set) => ({
|
418
|
-
message: null,
|
419
|
-
parentId: null,
|
420
|
-
branches: [],
|
421
|
-
isLast: false,
|
422
|
-
inProgressIndicator: null,
|
423
|
-
isCopied: false,
|
424
|
-
isHovering: false,
|
425
|
-
setInProgressIndicator: (value) => {
|
426
|
-
set({ inProgressIndicator: value });
|
427
|
-
},
|
428
|
-
setIsCopied: (value) => {
|
429
|
-
set({ isCopied: value });
|
430
|
-
},
|
431
|
-
setIsHovering: (value) => {
|
432
|
-
set({ isHovering: value });
|
433
|
-
}
|
434
|
-
}));
|
435
|
-
const useComposer = makeEditComposerStore({
|
436
|
-
onEdit: () => {
|
437
|
-
const message = useMessage.getState().message;
|
438
|
-
if (message.role !== "user")
|
439
|
-
throw new Error(
|
440
|
-
"Tried to edit a non-user message. Editing is only supported for user messages. This is likely an internal bug in assistant-ui."
|
441
|
-
);
|
442
|
-
const text = getMessageText(message);
|
443
|
-
return text;
|
444
|
-
},
|
445
|
-
onSend: (text) => {
|
446
|
-
const { message, parentId } = useMessage.getState();
|
447
|
-
if (message.role !== "user")
|
448
|
-
throw new Error(
|
449
|
-
"Tried to edit a non-user message. Editing is only supported for user messages. This is likely an internal bug in assistant-ui."
|
450
|
-
);
|
451
|
-
const nonTextParts = message.content.filter(
|
452
|
-
(part) => part.type !== "text" && part.type !== "ui"
|
453
|
-
);
|
454
|
-
useThread.getState().append({
|
455
|
-
parentId,
|
456
|
-
content: [{ type: "text", text }, ...nonTextParts]
|
457
|
-
});
|
458
|
-
}
|
459
|
-
});
|
460
|
-
syncMessage(useThread.getState(), useMessage, messageIndex);
|
461
|
-
return { useMessage, useComposer };
|
462
|
-
});
|
463
|
-
(0, import_react14.useEffect)(() => {
|
464
|
-
return useThread.subscribe((thread) => {
|
465
|
-
syncMessage(thread, context.useMessage, messageIndex);
|
466
|
-
});
|
467
|
-
}, [context, useThread, messageIndex]);
|
468
|
-
return context;
|
469
|
-
};
|
470
|
-
var MessageProvider = ({
|
471
|
-
messageIndex,
|
472
|
-
children
|
473
|
-
}) => {
|
474
|
-
const context = useMessageContext2(messageIndex);
|
475
|
-
return /* @__PURE__ */ (0, import_jsx_runtime4.jsx)(MessageContext.Provider, { value: context, children });
|
476
|
-
};
|
477
|
-
|
478
|
-
// src/context/ComposerContext.ts
|
479
|
-
var import_react15 = require("react");
|
480
|
-
var useComposerContext = () => {
|
481
|
-
const { useComposer } = useThreadContext();
|
482
|
-
const { useComposer: useEditComposer } = (0, import_react15.useContext)(MessageContext) ?? {};
|
483
|
-
return (0, import_react15.useMemo)(
|
484
|
-
() => ({
|
485
|
-
useComposer: useEditComposer ?? useComposer,
|
486
|
-
type: useEditComposer ? "edit" : "new"
|
487
|
-
}),
|
488
|
-
[useEditComposer, useComposer]
|
489
|
-
);
|
490
|
-
};
|
491
|
-
|
492
|
-
// src/primitives/composer/ComposerIf.tsx
|
493
|
-
var useComposerIf = (props) => {
|
494
|
-
const { useComposer } = useComposerContext();
|
495
|
-
return useComposer((composer) => {
|
496
|
-
if (props.editing === true && !composer.isEditing) return false;
|
497
|
-
if (props.editing === false && composer.isEditing) return false;
|
498
|
-
return true;
|
499
|
-
});
|
500
|
-
};
|
501
|
-
var ComposerIf = ({ children, ...query }) => {
|
502
|
-
const result = useComposerIf(query);
|
503
|
-
return result ? children : null;
|
504
|
-
};
|
505
|
-
|
506
|
-
// src/primitives/message/MessageIf.tsx
|
507
|
-
var useMessageIf = (props) => {
|
508
|
-
const { useMessage } = useMessageContext();
|
509
|
-
return useMessage(({ message, branches, isLast, isCopied, isHovering }) => {
|
510
|
-
if (props.hasBranches === true && branches.length < 2) return false;
|
511
|
-
if (props.user && message.role !== "user") return false;
|
512
|
-
if (props.assistant && message.role !== "assistant") return false;
|
513
|
-
if (props.lastOrHover === true && !isHovering && !isLast) return false;
|
514
|
-
if (props.copied === true && !isCopied) return false;
|
515
|
-
if (props.copied === false && isCopied) return false;
|
516
|
-
return true;
|
517
|
-
});
|
518
|
-
};
|
519
|
-
var MessageIf = ({ children, ...query }) => {
|
520
|
-
const result = useMessageIf(query);
|
521
|
-
return result ? children : null;
|
522
|
-
};
|
523
|
-
|
524
|
-
// src/primitives/thread/ThreadMessages.tsx
|
525
|
-
var import_jsx_runtime5 = require("react/jsx-runtime");
|
526
|
-
var getComponents = (components) => {
|
527
|
-
return {
|
528
|
-
EditComposer: components.EditComposer ?? components.UserMessage ?? components.Message,
|
529
|
-
UserMessage: components.UserMessage ?? components.Message,
|
530
|
-
AssistantMessage: components.AssistantMessage ?? components.Message
|
531
|
-
};
|
532
|
-
};
|
533
|
-
var ThreadMessages = ({ components }) => {
|
534
|
-
const { useThread } = useThreadContext();
|
535
|
-
const messagesLength = useThread((t) => t.messages.length);
|
536
|
-
const { UserMessage, EditComposer, AssistantMessage } = getComponents(components);
|
537
|
-
if (messagesLength === 0) return null;
|
538
|
-
return /* @__PURE__ */ (0, import_jsx_runtime5.jsx)(import_jsx_runtime5.Fragment, { children: new Array(messagesLength).fill(null).map((_, idx) => {
|
539
|
-
const messageIndex = idx;
|
540
|
-
return /* @__PURE__ */ (0, import_jsx_runtime5.jsxs)(
|
541
|
-
MessageProvider,
|
542
|
-
{
|
543
|
-
messageIndex,
|
544
|
-
children: [
|
545
|
-
/* @__PURE__ */ (0, import_jsx_runtime5.jsxs)(MessageIf, { user: true, children: [
|
546
|
-
/* @__PURE__ */ (0, import_jsx_runtime5.jsx)(ComposerIf, { editing: false, children: /* @__PURE__ */ (0, import_jsx_runtime5.jsx)(UserMessage, {}) }),
|
547
|
-
/* @__PURE__ */ (0, import_jsx_runtime5.jsx)(ComposerIf, { editing: true, children: /* @__PURE__ */ (0, import_jsx_runtime5.jsx)(EditComposer, {}) })
|
548
|
-
] }),
|
549
|
-
/* @__PURE__ */ (0, import_jsx_runtime5.jsx)(MessageIf, { assistant: true, children: /* @__PURE__ */ (0, import_jsx_runtime5.jsx)(AssistantMessage, {}) })
|
550
|
-
]
|
551
|
-
},
|
552
|
-
messageIndex
|
553
|
-
);
|
554
|
-
}) });
|
555
|
-
};
|
556
|
-
|
557
|
-
// src/primitives/thread/ThreadScrollToBottom.tsx
|
558
|
-
var import_primitive2 = require("@radix-ui/primitive");
|
559
|
-
var import_react_primitive3 = require("@radix-ui/react-primitive");
|
560
|
-
var import_react16 = require("react");
|
561
|
-
var import_jsx_runtime6 = require("react/jsx-runtime");
|
562
|
-
var ThreadScrollToBottom = (0, import_react16.forwardRef)(({ onClick, ...rest }, ref) => {
|
563
|
-
const { useViewport } = useThreadContext();
|
564
|
-
const isAtBottom = useViewport((s) => s.isAtBottom);
|
565
|
-
const handleScrollToBottom = () => {
|
566
|
-
useViewport.getState().scrollToBottom();
|
567
|
-
};
|
568
|
-
return /* @__PURE__ */ (0, import_jsx_runtime6.jsx)(
|
569
|
-
import_react_primitive3.Primitive.button,
|
570
|
-
{
|
571
|
-
...rest,
|
572
|
-
disabled: isAtBottom,
|
573
|
-
ref,
|
574
|
-
onClick: (0, import_primitive2.composeEventHandlers)(onClick, handleScrollToBottom)
|
575
|
-
}
|
576
|
-
);
|
577
|
-
});
|
578
|
-
|
579
|
-
// src/primitives/thread/ThreadSuggestion.tsx
|
580
|
-
var import_primitive3 = require("@radix-ui/primitive");
|
581
|
-
var import_react_primitive4 = require("@radix-ui/react-primitive");
|
582
|
-
var import_react17 = require("react");
|
583
|
-
var import_jsx_runtime7 = require("react/jsx-runtime");
|
584
|
-
var ThreadSuggestion = (0, import_react17.forwardRef)(({ onClick, prompt, method, autoSend: send, ...rest }, ref) => {
|
585
|
-
const { useThread, useComposer } = useThreadContext();
|
586
|
-
const isDisabled = useThread((t) => t.isRunning);
|
587
|
-
const handleApplySuggestion = () => {
|
588
|
-
const thread = useThread.getState();
|
589
|
-
const composer = useComposer.getState();
|
590
|
-
composer.setValue(prompt);
|
591
|
-
if (send && !thread.isRunning) {
|
592
|
-
composer.send();
|
593
|
-
}
|
594
|
-
};
|
595
|
-
return /* @__PURE__ */ (0, import_jsx_runtime7.jsx)(
|
596
|
-
import_react_primitive4.Primitive.button,
|
597
|
-
{
|
598
|
-
...rest,
|
599
|
-
disabled: isDisabled,
|
600
|
-
ref,
|
601
|
-
onClick: (0, import_primitive3.composeEventHandlers)(onClick, handleApplySuggestion)
|
602
|
-
}
|
603
|
-
);
|
604
|
-
});
|
605
|
-
|
606
|
-
// src/primitives/composer/index.ts
|
607
|
-
var composer_exports = {};
|
608
|
-
__export(composer_exports, {
|
609
|
-
Cancel: () => ComposerCancel,
|
610
|
-
If: () => ComposerIf,
|
611
|
-
Input: () => ComposerInput,
|
612
|
-
Root: () => ComposerRoot,
|
613
|
-
Send: () => ComposerSend
|
614
|
-
});
|
615
|
-
|
616
|
-
// src/primitives/composer/ComposerRoot.tsx
|
617
|
-
var import_primitive4 = require("@radix-ui/primitive");
|
618
|
-
var import_react_compose_refs2 = require("@radix-ui/react-compose-refs");
|
619
|
-
var import_react_primitive5 = require("@radix-ui/react-primitive");
|
620
|
-
var import_react18 = require("react");
|
621
|
-
var import_jsx_runtime8 = require("react/jsx-runtime");
|
622
|
-
var ComposerRoot = (0, import_react18.forwardRef)(
|
623
|
-
({ onSubmit, ...rest }, forwardedRef) => {
|
624
|
-
const { useViewport } = useThreadContext();
|
625
|
-
const { useComposer } = useComposerContext();
|
626
|
-
const formRef = (0, import_react18.useRef)(null);
|
627
|
-
const ref = (0, import_react_compose_refs2.useComposedRefs)(forwardedRef, formRef);
|
628
|
-
const handleSubmit = (e) => {
|
629
|
-
const composerState = useComposer.getState();
|
630
|
-
if (!composerState.isEditing) return;
|
631
|
-
e.preventDefault();
|
632
|
-
composerState.send();
|
633
|
-
useViewport.getState().scrollToBottom();
|
634
|
-
};
|
635
|
-
return /* @__PURE__ */ (0, import_jsx_runtime8.jsx)(
|
636
|
-
import_react_primitive5.Primitive.form,
|
637
|
-
{
|
638
|
-
...rest,
|
639
|
-
ref,
|
640
|
-
onSubmit: (0, import_primitive4.composeEventHandlers)(onSubmit, handleSubmit)
|
641
|
-
}
|
642
|
-
);
|
643
|
-
}
|
644
|
-
);
|
645
|
-
|
646
|
-
// src/primitives/composer/ComposerInput.tsx
|
647
|
-
var import_primitive5 = require("@radix-ui/primitive");
|
648
|
-
var import_react_compose_refs3 = require("@radix-ui/react-compose-refs");
|
649
|
-
var import_react_slot = require("@radix-ui/react-slot");
|
650
|
-
var import_react19 = require("react");
|
651
|
-
var import_react_textarea_autosize = __toESM(require("react-textarea-autosize"));
|
652
|
-
var import_jsx_runtime9 = require("react/jsx-runtime");
|
653
|
-
var ComposerInput = (0, import_react19.forwardRef)(
|
654
|
-
({ autoFocus = false, asChild, disabled, onChange, onKeyDown, ...rest }, forwardedRef) => {
|
655
|
-
const { useThread, useViewport } = useThreadContext();
|
656
|
-
const { useComposer, type } = useComposerContext();
|
657
|
-
const value = useComposer((c) => {
|
658
|
-
if (!c.isEditing) return "";
|
659
|
-
return c.value;
|
660
|
-
});
|
661
|
-
const Component = asChild ? import_react_slot.Slot : import_react_textarea_autosize.default;
|
662
|
-
const handleKeyPress = (e) => {
|
663
|
-
if (disabled) return;
|
664
|
-
const composer = useComposer.getState();
|
665
|
-
if (e.key === "Escape") {
|
666
|
-
if (useComposer.getState().cancel()) {
|
667
|
-
e.preventDefault();
|
668
|
-
}
|
669
|
-
} else if (e.key === "Enter" && e.shiftKey === false) {
|
670
|
-
const isRunning = useThread.getState().isRunning;
|
671
|
-
if (!isRunning) {
|
672
|
-
e.preventDefault();
|
673
|
-
composer.send();
|
674
|
-
useViewport.getState().scrollToBottom();
|
675
|
-
}
|
676
|
-
}
|
677
|
-
};
|
678
|
-
const textareaRef = (0, import_react19.useRef)(null);
|
679
|
-
const ref = (0, import_react_compose_refs3.useComposedRefs)(forwardedRef, textareaRef);
|
680
|
-
const autoFocusEnabled = autoFocus && !disabled;
|
681
|
-
const focus = (0, import_react19.useCallback)(() => {
|
682
|
-
const textarea = textareaRef.current;
|
683
|
-
if (!textarea || !autoFocusEnabled) return;
|
684
|
-
textarea.focus();
|
685
|
-
textarea.setSelectionRange(
|
686
|
-
textareaRef.current.value.length,
|
687
|
-
textareaRef.current.value.length
|
688
|
-
);
|
689
|
-
}, [autoFocusEnabled]);
|
690
|
-
(0, import_react19.useEffect)(() => focus(), [focus]);
|
691
|
-
useOnScrollToBottom(() => {
|
692
|
-
if (type === "new") {
|
693
|
-
focus();
|
694
|
-
}
|
695
|
-
});
|
696
|
-
return /* @__PURE__ */ (0, import_jsx_runtime9.jsx)(
|
697
|
-
Component,
|
698
|
-
{
|
699
|
-
value,
|
700
|
-
...rest,
|
701
|
-
ref,
|
702
|
-
disabled,
|
703
|
-
onChange: (0, import_primitive5.composeEventHandlers)(onChange, (e) => {
|
704
|
-
const composerState = useComposer.getState();
|
705
|
-
if (!composerState.isEditing) return;
|
706
|
-
return composerState.setValue(e.target.value);
|
707
|
-
}),
|
708
|
-
onKeyDown: (0, import_primitive5.composeEventHandlers)(onKeyDown, handleKeyPress)
|
709
|
-
}
|
710
|
-
);
|
711
|
-
}
|
712
|
-
);
|
713
|
-
|
714
|
-
// src/primitives/composer/ComposerSend.tsx
|
715
|
-
var import_react_primitive6 = require("@radix-ui/react-primitive");
|
716
|
-
var import_react20 = require("react");
|
717
|
-
var import_jsx_runtime10 = require("react/jsx-runtime");
|
718
|
-
var ComposerSend = (0, import_react20.forwardRef)(
|
719
|
-
({ disabled, ...rest }, ref) => {
|
720
|
-
const { useComposer } = useComposerContext();
|
721
|
-
const hasValue = useComposer((c) => c.isEditing && c.value.length > 0);
|
722
|
-
return /* @__PURE__ */ (0, import_jsx_runtime10.jsx)(
|
723
|
-
import_react_primitive6.Primitive.button,
|
724
|
-
{
|
725
|
-
type: "submit",
|
726
|
-
...rest,
|
727
|
-
ref,
|
728
|
-
disabled: disabled || !hasValue
|
729
|
-
}
|
730
|
-
);
|
731
|
-
}
|
732
|
-
);
|
733
|
-
|
734
|
-
// src/primitives/composer/ComposerCancel.tsx
|
735
|
-
var import_primitive6 = require("@radix-ui/primitive");
|
736
|
-
var import_react_primitive7 = require("@radix-ui/react-primitive");
|
737
|
-
var import_react21 = require("react");
|
738
|
-
var import_jsx_runtime11 = require("react/jsx-runtime");
|
739
|
-
var ComposerCancel = (0, import_react21.forwardRef)(({ onClick, ...rest }, ref) => {
|
740
|
-
const { useComposer } = useComposerContext();
|
741
|
-
const handleCancel = () => {
|
742
|
-
useComposer.getState().cancel();
|
743
|
-
};
|
744
|
-
return /* @__PURE__ */ (0, import_jsx_runtime11.jsx)(
|
745
|
-
import_react_primitive7.Primitive.button,
|
746
|
-
{
|
747
|
-
type: "button",
|
748
|
-
...rest,
|
749
|
-
ref,
|
750
|
-
onClick: (0, import_primitive6.composeEventHandlers)(onClick, handleCancel)
|
751
|
-
}
|
752
|
-
);
|
753
|
-
});
|
754
|
-
|
755
|
-
// src/primitives/message/index.ts
|
756
|
-
var message_exports = {};
|
757
|
-
__export(message_exports, {
|
758
|
-
Content: () => MessageContent,
|
759
|
-
If: () => MessageIf,
|
760
|
-
InProgress: () => MessageInProgress,
|
761
|
-
Root: () => MessageRoot
|
762
|
-
});
|
763
|
-
|
764
|
-
// src/primitives/message/MessageRoot.tsx
|
765
|
-
var import_primitive7 = require("@radix-ui/primitive");
|
766
|
-
var import_react_primitive8 = require("@radix-ui/react-primitive");
|
767
|
-
var import_react22 = require("react");
|
768
|
-
var import_jsx_runtime12 = require("react/jsx-runtime");
|
769
|
-
var MessageRoot = (0, import_react22.forwardRef)(
|
770
|
-
({ onMouseEnter, onMouseLeave, ...rest }, ref) => {
|
771
|
-
const { useMessage } = useMessageContext();
|
772
|
-
const setIsHovering = useMessage((s) => s.setIsHovering);
|
773
|
-
const handleMouseEnter = () => {
|
774
|
-
setIsHovering(true);
|
775
|
-
};
|
776
|
-
const handleMouseLeave = () => {
|
777
|
-
setIsHovering(false);
|
778
|
-
};
|
779
|
-
return /* @__PURE__ */ (0, import_jsx_runtime12.jsx)(
|
780
|
-
import_react_primitive8.Primitive.div,
|
781
|
-
{
|
782
|
-
...rest,
|
783
|
-
ref,
|
784
|
-
onMouseEnter: (0, import_primitive7.composeEventHandlers)(onMouseEnter, handleMouseEnter),
|
785
|
-
onMouseLeave: (0, import_primitive7.composeEventHandlers)(onMouseLeave, handleMouseLeave)
|
786
|
-
}
|
787
|
-
);
|
788
|
-
}
|
789
|
-
);
|
790
|
-
|
791
|
-
// src/context/providers/ContentPartProvider.tsx
|
792
|
-
var import_react24 = require("react");
|
793
|
-
var import_zustand3 = require("zustand");
|
794
|
-
|
795
|
-
// src/context/ContentPartContext.ts
|
796
|
-
var import_react23 = require("react");
|
797
|
-
var ContentPartContext = (0, import_react23.createContext)(
|
798
|
-
null
|
799
|
-
);
|
800
|
-
var useContentPartContext = () => {
|
801
|
-
const context = (0, import_react23.useContext)(ContentPartContext);
|
802
|
-
if (!context)
|
803
|
-
throw new Error(
|
804
|
-
"This component can only be used inside a component passed to <MessagePrimitive.Content components={...} >."
|
805
|
-
);
|
806
|
-
return context;
|
807
|
-
};
|
808
|
-
|
809
|
-
// src/context/providers/ContentPartProvider.tsx
|
810
|
-
var import_jsx_runtime13 = require("react/jsx-runtime");
|
811
|
-
var syncContentPart = ({ message }, useContentPart, partIndex) => {
|
812
|
-
const part = message.content[partIndex];
|
813
|
-
if (!part) return;
|
814
|
-
const messageStatus = message.role === "assistant" ? message.status : "done";
|
815
|
-
const status = partIndex === message.content.length - 1 ? messageStatus : "done";
|
816
|
-
useContentPart.setState({ part, status });
|
817
|
-
};
|
818
|
-
var useContentPartContext2 = (partIndex) => {
|
819
|
-
const { useMessage } = useMessageContext();
|
820
|
-
const [context] = (0, import_react24.useState)(() => {
|
821
|
-
const useContentPart = (0, import_zustand3.create)(() => ({
|
822
|
-
part: { type: "text", text: "" },
|
823
|
-
status: "done"
|
824
|
-
}));
|
825
|
-
syncContentPart(useMessage.getState(), useContentPart, partIndex);
|
826
|
-
return { useContentPart };
|
827
|
-
});
|
828
|
-
(0, import_react24.useEffect)(() => {
|
829
|
-
return useMessage.subscribe((message) => {
|
830
|
-
syncContentPart(message, context.useContentPart, partIndex);
|
831
|
-
});
|
832
|
-
}, [context, useMessage, partIndex]);
|
833
|
-
return context;
|
834
|
-
};
|
835
|
-
var ContentPartProvider = ({
|
836
|
-
partIndex,
|
837
|
-
children
|
838
|
-
}) => {
|
839
|
-
const context = useContentPartContext2(partIndex);
|
840
|
-
return /* @__PURE__ */ (0, import_jsx_runtime13.jsx)(ContentPartContext.Provider, { value: context, children });
|
841
|
-
};
|
842
|
-
|
843
|
-
// src/primitives/contentPart/ContentPartInProgressIndicator.tsx
|
844
|
-
var ContentPartInProgressIndicator = () => {
|
845
|
-
const { useMessage } = useMessageContext();
|
846
|
-
const { useContentPart } = useContentPartContext();
|
847
|
-
const indicator = useCombinedStore(
|
848
|
-
[useMessage, useContentPart],
|
849
|
-
(m, c) => c.status === "in_progress" ? m.inProgressIndicator : null
|
850
|
-
);
|
851
|
-
return indicator;
|
852
|
-
};
|
853
|
-
|
854
|
-
// src/primitives/message/MessageContent.tsx
|
855
|
-
var import_jsx_runtime14 = require("react/jsx-runtime");
|
856
|
-
var defaultComponents = {
|
857
|
-
Text: ({ part }) => /* @__PURE__ */ (0, import_jsx_runtime14.jsxs)(import_jsx_runtime14.Fragment, { children: [
|
858
|
-
part.text,
|
859
|
-
/* @__PURE__ */ (0, import_jsx_runtime14.jsx)(ContentPartInProgressIndicator, {})
|
860
|
-
] }),
|
861
|
-
Image: () => null,
|
862
|
-
UI: ({ part }) => part.display,
|
863
|
-
tools: {
|
864
|
-
Fallback: () => null
|
865
|
-
}
|
866
|
-
};
|
867
|
-
var MessageContent = ({
|
868
|
-
components: {
|
869
|
-
Text = defaultComponents.Text,
|
870
|
-
Image = defaultComponents.Image,
|
871
|
-
UI = defaultComponents.UI,
|
872
|
-
tools: { by_name = {}, Fallback = defaultComponents.tools.Fallback } = {}
|
873
|
-
} = {}
|
874
|
-
}) => {
|
875
|
-
const { useMessage } = useMessageContext();
|
876
|
-
const message = useMessage((s) => s.message);
|
877
|
-
const content = message.content;
|
878
|
-
return /* @__PURE__ */ (0, import_jsx_runtime14.jsx)(import_jsx_runtime14.Fragment, { children: content.map((part, i) => {
|
879
|
-
const partIndex = i;
|
880
|
-
const type = part.type;
|
881
|
-
let component = null;
|
882
|
-
switch (type) {
|
883
|
-
case "text":
|
884
|
-
component = /* @__PURE__ */ (0, import_jsx_runtime14.jsx)(Text, { part });
|
885
|
-
break;
|
886
|
-
case "image":
|
887
|
-
component = /* @__PURE__ */ (0, import_jsx_runtime14.jsx)(Image, { part });
|
888
|
-
break;
|
889
|
-
case "ui":
|
890
|
-
component = /* @__PURE__ */ (0, import_jsx_runtime14.jsx)(UI, { part });
|
891
|
-
break;
|
892
|
-
case "tool-call": {
|
893
|
-
const Tool = by_name[part.name] || Fallback;
|
894
|
-
component = /* @__PURE__ */ (0, import_jsx_runtime14.jsx)(Tool, { part });
|
895
|
-
break;
|
896
|
-
}
|
897
|
-
default:
|
898
|
-
throw new Error(`Unknown content part type: ${type}`);
|
899
|
-
}
|
900
|
-
return /* @__PURE__ */ (0, import_jsx_runtime14.jsx)(ContentPartProvider, { partIndex, children: component }, partIndex);
|
901
|
-
}) });
|
902
|
-
};
|
903
|
-
|
904
|
-
// src/primitives/message/MessageInProgress.tsx
|
905
|
-
var import_react_primitive9 = require("@radix-ui/react-primitive");
|
906
|
-
var import_react25 = require("react");
|
907
|
-
var import_jsx_runtime15 = require("react/jsx-runtime");
|
908
|
-
var MessageInProgress = (0, import_react25.forwardRef)((props, ref) => {
|
909
|
-
const { useMessage } = useMessageContext();
|
910
|
-
(0, import_react25.useMemo)(() => {
|
911
|
-
useMessage.getState().setInProgressIndicator(/* @__PURE__ */ (0, import_jsx_runtime15.jsx)(import_react_primitive9.Primitive.span, { ...props, ref }));
|
912
|
-
}, [useMessage, props, ref]);
|
913
|
-
return null;
|
914
|
-
});
|
915
|
-
|
916
|
-
// src/primitives/branchPicker/index.ts
|
917
|
-
var branchPicker_exports = {};
|
918
|
-
__export(branchPicker_exports, {
|
919
|
-
Count: () => BranchPickerCount,
|
920
|
-
Next: () => BranchPickerNext,
|
921
|
-
Number: () => BranchPickerNumber,
|
922
|
-
Previous: () => BranchPickerPrevious,
|
923
|
-
Root: () => BranchPickerRoot
|
924
|
-
});
|
925
|
-
|
926
|
-
// src/utils/createActionButton.tsx
|
927
|
-
var import_primitive8 = require("@radix-ui/primitive");
|
928
|
-
var import_react_primitive10 = require("@radix-ui/react-primitive");
|
929
|
-
var import_react26 = require("react");
|
930
|
-
var import_jsx_runtime16 = require("react/jsx-runtime");
|
931
|
-
var createActionButton = (useActionButton) => {
|
932
|
-
return (0, import_react26.forwardRef)(
|
933
|
-
(props, forwardedRef) => {
|
934
|
-
const onClick = useActionButton(props);
|
935
|
-
return /* @__PURE__ */ (0, import_jsx_runtime16.jsx)(
|
936
|
-
import_react_primitive10.Primitive.button,
|
937
|
-
{
|
938
|
-
type: "button",
|
939
|
-
disabled: !onClick,
|
940
|
-
...props,
|
941
|
-
ref: forwardedRef,
|
942
|
-
onClick: (0, import_primitive8.composeEventHandlers)(props.onClick, onClick ?? void 0)
|
943
|
-
}
|
944
|
-
);
|
945
|
-
}
|
946
|
-
);
|
947
|
-
};
|
948
|
-
|
949
|
-
// src/primitives/branchPicker/BranchPickerNext.tsx
|
950
|
-
var BranchPickerNext = createActionButton(useGoToNextBranch);
|
951
|
-
|
952
|
-
// src/primitives/branchPicker/BranchPickerPrevious.tsx
|
953
|
-
var BranchPickerPrevious = createActionButton(useGoToPreviousBranch);
|
954
|
-
|
955
|
-
// src/primitives/branchPicker/BranchPickerCount.tsx
|
956
|
-
var import_jsx_runtime17 = require("react/jsx-runtime");
|
957
|
-
var BranchPickerCount = () => {
|
958
|
-
const { useMessage } = useMessageContext();
|
959
|
-
const branchCount = useMessage((s) => s.branches.length);
|
960
|
-
return /* @__PURE__ */ (0, import_jsx_runtime17.jsx)(import_jsx_runtime17.Fragment, { children: branchCount });
|
961
|
-
};
|
962
|
-
|
963
|
-
// src/primitives/branchPicker/BranchPickerNumber.tsx
|
964
|
-
var import_jsx_runtime18 = require("react/jsx-runtime");
|
965
|
-
var BranchPickerNumber = () => {
|
966
|
-
const { useMessage } = useMessageContext();
|
967
|
-
const branchIdx = useMessage((s) => s.branches.indexOf(s.message.id));
|
968
|
-
return /* @__PURE__ */ (0, import_jsx_runtime18.jsx)(import_jsx_runtime18.Fragment, { children: branchIdx + 1 });
|
969
|
-
};
|
970
|
-
|
971
|
-
// src/primitives/branchPicker/BranchPickerRoot.tsx
|
972
|
-
var import_react_primitive11 = require("@radix-ui/react-primitive");
|
973
|
-
var import_react27 = require("react");
|
974
|
-
var import_jsx_runtime19 = require("react/jsx-runtime");
|
975
|
-
var BranchPickerRoot = (0, import_react27.forwardRef)(({ hideWhenSingleBranch, ...rest }, ref) => {
|
976
|
-
return /* @__PURE__ */ (0, import_jsx_runtime19.jsx)(MessageIf, { hasBranches: hideWhenSingleBranch ? true : void 0, children: /* @__PURE__ */ (0, import_jsx_runtime19.jsx)(import_react_primitive11.Primitive.div, { ...rest, ref }) });
|
977
|
-
});
|
978
|
-
|
979
|
-
// src/primitives/actionBar/index.ts
|
980
|
-
var actionBar_exports = {};
|
981
|
-
__export(actionBar_exports, {
|
982
|
-
Copy: () => ActionBarCopy,
|
983
|
-
Edit: () => ActionBarEdit,
|
984
|
-
Reload: () => ActionBarReload,
|
985
|
-
Root: () => ActionBarRoot
|
986
|
-
});
|
987
|
-
|
988
|
-
// src/primitives/actionBar/ActionBarRoot.tsx
|
989
|
-
var import_react_primitive12 = require("@radix-ui/react-primitive");
|
990
|
-
var import_react28 = require("react");
|
991
|
-
var import_jsx_runtime20 = require("react/jsx-runtime");
|
992
|
-
var ActionBarRoot = (0, import_react28.forwardRef)(({ hideWhenRunning, autohide, autohideFloat, ...rest }, ref) => {
|
993
|
-
const { useThread } = useThreadContext();
|
994
|
-
const { useMessage } = useMessageContext();
|
995
|
-
const hideAndfloatStatus = useCombinedStore(
|
996
|
-
[useThread, useMessage],
|
997
|
-
(t, m) => {
|
998
|
-
if (hideWhenRunning && t.isRunning) return "hidden" /* Hidden */;
|
999
|
-
const autohideEnabled = autohide === "always" || autohide === "not-last" && !m.isLast;
|
1000
|
-
if (!autohideEnabled) return "normal" /* Normal */;
|
1001
|
-
if (!m.isHovering) return "hidden" /* Hidden */;
|
1002
|
-
if (autohideFloat === "always" || autohideFloat === "single-branch" && m.branches.length <= 1)
|
1003
|
-
return "floating" /* Floating */;
|
1004
|
-
return "normal" /* Normal */;
|
1005
|
-
}
|
1006
|
-
);
|
1007
|
-
if (hideAndfloatStatus === "hidden" /* Hidden */) return null;
|
1008
|
-
return /* @__PURE__ */ (0, import_jsx_runtime20.jsx)(
|
1009
|
-
import_react_primitive12.Primitive.div,
|
1010
|
-
{
|
1011
|
-
"data-floating": hideAndfloatStatus === "floating" /* Floating */,
|
1012
|
-
...rest,
|
1013
|
-
ref
|
1014
|
-
}
|
1015
|
-
);
|
1016
|
-
});
|
1017
|
-
|
1018
|
-
// src/primitives/actionBar/ActionBarCopy.tsx
|
1019
|
-
var ActionBarCopy = createActionButton(useCopyMessage);
|
1020
|
-
|
1021
|
-
// src/primitives/actionBar/ActionBarReload.tsx
|
1022
|
-
var ActionBarReload = createActionButton(useReloadMessage);
|
1023
|
-
|
1024
|
-
// src/primitives/actionBar/ActionBarEdit.tsx
|
1025
|
-
var ActionBarEdit = createActionButton(useBeginMessageEdit);
|
1026
|
-
|
1027
|
-
// src/primitives/contentPart/index.ts
|
1028
|
-
var contentPart_exports = {};
|
1029
|
-
__export(contentPart_exports, {
|
1030
|
-
InProgressIndicator: () => ContentPartInProgressIndicator
|
1031
|
-
});
|
1032
|
-
|
1033
|
-
// src/runtime/local/useLocalRuntime.tsx
|
1034
|
-
var import_react29 = require("react");
|
1035
|
-
|
1036
|
-
// src/utils/ModelConfigTypes.ts
|
1037
|
-
var mergeModelConfigs = (configs) => {
|
1038
|
-
return configs.reduce((acc, config) => {
|
1039
|
-
if (config.system) {
|
1040
|
-
if (acc.system) {
|
1041
|
-
acc.system += `
|
1042
|
-
|
1043
|
-
${config.system}`;
|
1044
|
-
} else {
|
1045
|
-
acc.system = config.system;
|
1046
|
-
}
|
1047
|
-
}
|
1048
|
-
if (config.tools) {
|
1049
|
-
acc.tools = { ...acc.tools, ...config.tools };
|
1050
|
-
}
|
1051
|
-
return acc;
|
1052
|
-
}, {});
|
1053
|
-
};
|
1054
|
-
|
1055
|
-
// src/runtime/utils/idUtils.tsx
|
1056
|
-
var import_non_secure = require("nanoid/non-secure");
|
1057
|
-
var generateId = (0, import_non_secure.customAlphabet)(
|
1058
|
-
"0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz",
|
1059
|
-
7
|
1060
|
-
);
|
1061
|
-
var optimisticPrefix = "__optimistic__";
|
1062
|
-
var generateOptimisticId = () => `${optimisticPrefix}${generateId()}`;
|
1063
|
-
|
1064
|
-
// src/runtime/utils/MessageRepository.tsx
|
1065
|
-
var findHead = (message) => {
|
1066
|
-
if (message.next) return findHead(message.next);
|
1067
|
-
return message;
|
1068
|
-
};
|
1069
|
-
var MessageRepository = class {
|
1070
|
-
messages = /* @__PURE__ */ new Map();
|
1071
|
-
// message_id -> item
|
1072
|
-
head = null;
|
1073
|
-
root = {
|
1074
|
-
children: []
|
1075
|
-
};
|
1076
|
-
performOp(newParent, child, operation) {
|
1077
|
-
const parentOrRoot = child.prev ?? this.root;
|
1078
|
-
const newParentOrRoot = newParent ?? this.root;
|
1079
|
-
if (operation === "relink" && parentOrRoot === newParentOrRoot) return;
|
1080
|
-
if (operation !== "link") {
|
1081
|
-
parentOrRoot.children = parentOrRoot.children.filter(
|
1082
|
-
(m) => m !== child.current.id
|
1083
|
-
);
|
1084
|
-
if (child.prev?.next === child) {
|
1085
|
-
const fallbackId = child.prev.children.at(-1);
|
1086
|
-
const fallback = fallbackId ? this.messages.get(fallbackId) : null;
|
1087
|
-
if (fallback === void 0) {
|
1088
|
-
throw new Error(
|
1089
|
-
"MessageRepository(performOp/cut): Fallback sibling message not found. This is likely an internal bug in assistant-ui."
|
1090
|
-
);
|
1091
|
-
}
|
1092
|
-
child.prev.next = fallback;
|
1093
|
-
}
|
1094
|
-
}
|
1095
|
-
if (operation !== "cut") {
|
1096
|
-
newParentOrRoot.children = [
|
1097
|
-
...newParentOrRoot.children,
|
1098
|
-
child.current.id
|
1099
|
-
];
|
1100
|
-
if (newParent && (findHead(child) === this.head || newParent.next === null)) {
|
1101
|
-
newParent.next = child;
|
1102
|
-
}
|
1103
|
-
child.prev = newParent;
|
1104
|
-
}
|
1105
|
-
}
|
1106
|
-
getMessages() {
|
1107
|
-
const messages = new Array(this.head?.level ?? 0);
|
1108
|
-
for (let current = this.head; current; current = current.prev) {
|
1109
|
-
messages[current.level] = current.current;
|
1110
|
-
}
|
1111
|
-
return messages;
|
1112
|
-
}
|
1113
|
-
addOrUpdateMessage(parentId, message) {
|
1114
|
-
const existingItem = this.messages.get(message.id);
|
1115
|
-
const prev = parentId ? this.messages.get(parentId) : null;
|
1116
|
-
if (prev === void 0)
|
1117
|
-
throw new Error(
|
1118
|
-
"MessageRepository(addOrUpdateMessage): Parent message not found. This is likely an internal bug in assistant-ui."
|
1119
|
-
);
|
1120
|
-
if (existingItem) {
|
1121
|
-
existingItem.current = message;
|
1122
|
-
this.performOp(prev, existingItem, "relink");
|
1123
|
-
return;
|
1124
|
-
}
|
1125
|
-
const newItem = {
|
1126
|
-
prev,
|
1127
|
-
current: message,
|
1128
|
-
next: null,
|
1129
|
-
children: [],
|
1130
|
-
level: prev ? prev.level + 1 : 0
|
1131
|
-
};
|
1132
|
-
this.messages.set(message.id, newItem);
|
1133
|
-
this.performOp(prev, newItem, "link");
|
1134
|
-
if (this.head === prev) {
|
1135
|
-
this.head = newItem;
|
1136
|
-
}
|
1137
|
-
}
|
1138
|
-
appendOptimisticMessage(parentId, message) {
|
1139
|
-
let optimisticId;
|
1140
|
-
do {
|
1141
|
-
optimisticId = generateOptimisticId();
|
1142
|
-
} while (this.messages.has(optimisticId));
|
1143
|
-
this.addOrUpdateMessage(parentId, {
|
1144
|
-
...message,
|
1145
|
-
id: optimisticId,
|
1146
|
-
createdAt: /* @__PURE__ */ new Date(),
|
1147
|
-
...message.role === "assistant" ? { status: "in_progress" } : void 0
|
1148
|
-
});
|
1149
|
-
return optimisticId;
|
1150
|
-
}
|
1151
|
-
deleteMessage(messageId, replacementId) {
|
1152
|
-
const message = this.messages.get(messageId);
|
1153
|
-
if (!message)
|
1154
|
-
throw new Error(
|
1155
|
-
"MessageRepository(deleteMessage): Optimistic message not found. This is likely an internal bug in assistant-ui."
|
1156
|
-
);
|
1157
|
-
const replacement = replacementId === void 0 ? message.prev : replacementId === null ? null : this.messages.get(replacementId);
|
1158
|
-
if (replacement === void 0)
|
1159
|
-
throw new Error(
|
1160
|
-
"MessageRepository(deleteMessage): Replacement not found. This is likely an internal bug in assistant-ui."
|
1161
|
-
);
|
1162
|
-
for (const child of message.children) {
|
1163
|
-
const childMessage = this.messages.get(child);
|
1164
|
-
if (!childMessage)
|
1165
|
-
throw new Error(
|
1166
|
-
"MessageRepository(deleteMessage): Child message not found. This is likely an internal bug in assistant-ui."
|
1167
|
-
);
|
1168
|
-
this.performOp(replacement, childMessage, "relink");
|
1169
|
-
}
|
1170
|
-
this.performOp(null, message, "cut");
|
1171
|
-
this.messages.delete(messageId);
|
1172
|
-
if (this.head === message) {
|
1173
|
-
this.head = replacement ? findHead(replacement) : null;
|
1174
|
-
}
|
1175
|
-
}
|
1176
|
-
getBranches(messageId) {
|
1177
|
-
const message = this.messages.get(messageId);
|
1178
|
-
if (!message)
|
1179
|
-
throw new Error(
|
1180
|
-
"MessageRepository(getBranches): Message not found. This is likely an internal bug in assistant-ui."
|
1181
|
-
);
|
1182
|
-
const { children } = message.prev ?? this.root;
|
1183
|
-
return children;
|
1184
|
-
}
|
1185
|
-
switchToBranch(messageId) {
|
1186
|
-
const message = this.messages.get(messageId);
|
1187
|
-
if (!message)
|
1188
|
-
throw new Error(
|
1189
|
-
"MessageRepository(switchToBranch): Branch not found. This is likely an internal bug in assistant-ui."
|
1190
|
-
);
|
1191
|
-
if (message.prev) {
|
1192
|
-
message.prev.next = message;
|
1193
|
-
}
|
1194
|
-
this.head = findHead(message);
|
1195
|
-
}
|
1196
|
-
resetHead(messageId) {
|
1197
|
-
if (messageId === null) {
|
1198
|
-
this.head = null;
|
1199
|
-
return;
|
1200
|
-
}
|
1201
|
-
const message = this.messages.get(messageId);
|
1202
|
-
if (!message)
|
1203
|
-
throw new Error(
|
1204
|
-
"MessageRepository(resetHead): Branch not found. This is likely an internal bug in assistant-ui."
|
1205
|
-
);
|
1206
|
-
this.head = message;
|
1207
|
-
for (let current = message; current; current = current.prev) {
|
1208
|
-
if (current.prev) {
|
1209
|
-
current.prev.next = current;
|
1210
|
-
}
|
1211
|
-
}
|
1212
|
-
}
|
1213
|
-
};
|
1214
|
-
|
1215
|
-
// src/runtime/local/LocalRuntime.tsx
|
1216
|
-
var LocalRuntime = class {
|
1217
|
-
constructor(adapter) {
|
1218
|
-
this.adapter = adapter;
|
1219
|
-
}
|
1220
|
-
_subscriptions = /* @__PURE__ */ new Set();
|
1221
|
-
_configProviders = /* @__PURE__ */ new Set();
|
1222
|
-
abortController = null;
|
1223
|
-
repository = new MessageRepository();
|
1224
|
-
get messages() {
|
1225
|
-
return this.repository.getMessages();
|
1226
|
-
}
|
1227
|
-
get isRunning() {
|
1228
|
-
return this.abortController != null;
|
1229
|
-
}
|
1230
|
-
getBranches(messageId) {
|
1231
|
-
return this.repository.getBranches(messageId);
|
1232
|
-
}
|
1233
|
-
switchToBranch(branchId) {
|
1234
|
-
this.repository.switchToBranch(branchId);
|
1235
|
-
this.notifySubscribers();
|
1236
|
-
}
|
1237
|
-
async append(message) {
|
1238
|
-
const userMessageId = generateId();
|
1239
|
-
const userMessage = {
|
1240
|
-
id: userMessageId,
|
1241
|
-
role: "user",
|
1242
|
-
content: message.content,
|
1243
|
-
createdAt: /* @__PURE__ */ new Date()
|
1244
|
-
};
|
1245
|
-
this.repository.addOrUpdateMessage(message.parentId, userMessage);
|
1246
|
-
await this.startRun(userMessageId);
|
1247
|
-
}
|
1248
|
-
async startRun(parentId) {
|
1249
|
-
const id = generateId();
|
1250
|
-
this.repository.resetHead(parentId);
|
1251
|
-
const messages = this.repository.getMessages();
|
1252
|
-
const message = {
|
1253
|
-
id,
|
1254
|
-
role: "assistant",
|
1255
|
-
status: "in_progress",
|
1256
|
-
content: [{ type: "text", text: "" }],
|
1257
|
-
createdAt: /* @__PURE__ */ new Date()
|
1258
|
-
};
|
1259
|
-
this.repository.addOrUpdateMessage(parentId, { ...message });
|
1260
|
-
this.abortController?.abort();
|
1261
|
-
this.abortController = new AbortController();
|
1262
|
-
this.notifySubscribers();
|
1263
|
-
try {
|
1264
|
-
const updateHandler = ({ content }) => {
|
1265
|
-
message.content = content;
|
1266
|
-
this.repository.addOrUpdateMessage(parentId, { ...message });
|
1267
|
-
this.notifySubscribers();
|
1268
|
-
};
|
1269
|
-
const result = await this.adapter.run({
|
1270
|
-
messages,
|
1271
|
-
abortSignal: this.abortController.signal,
|
1272
|
-
config: mergeModelConfigs([...this._configProviders].map((p) => p())),
|
1273
|
-
onUpdate: updateHandler
|
1274
|
-
});
|
1275
|
-
updateHandler(result);
|
1276
|
-
message.status = "done";
|
1277
|
-
this.repository.addOrUpdateMessage(parentId, { ...message });
|
1278
|
-
} catch (e) {
|
1279
|
-
message.status = "error";
|
1280
|
-
this.repository.addOrUpdateMessage(parentId, { ...message });
|
1281
|
-
console.error(e);
|
1282
|
-
} finally {
|
1283
|
-
this.abortController = null;
|
1284
|
-
this.notifySubscribers();
|
1285
|
-
}
|
1286
|
-
}
|
1287
|
-
cancelRun() {
|
1288
|
-
if (!this.abortController) return;
|
1289
|
-
this.abortController.abort();
|
1290
|
-
this.abortController = null;
|
1291
|
-
this.notifySubscribers();
|
1292
|
-
}
|
1293
|
-
notifySubscribers() {
|
1294
|
-
for (const callback of this._subscriptions) callback();
|
1295
|
-
}
|
1296
|
-
subscribe(callback) {
|
1297
|
-
this._subscriptions.add(callback);
|
1298
|
-
return () => this._subscriptions.delete(callback);
|
1299
|
-
}
|
1300
|
-
registerModelConfigProvider(provider) {
|
1301
|
-
this._configProviders.add(provider);
|
1302
|
-
return () => this._configProviders.delete(provider);
|
1303
|
-
}
|
1304
|
-
};
|
1305
|
-
|
1306
|
-
// src/runtime/local/useLocalRuntime.tsx
|
1307
|
-
var useLocalRuntime = (adapter) => {
|
1308
|
-
const [runtime] = (0, import_react29.useState)(() => new LocalRuntime(adapter));
|
1309
|
-
(0, import_react29.useInsertionEffect)(() => {
|
1310
|
-
runtime.adapter = adapter;
|
1311
|
-
});
|
1312
|
-
return runtime;
|
1313
|
-
};
|
1314
|
-
|
1315
|
-
// src/context/providers/AssistantRuntimeProvider.tsx
|
1316
|
-
var import_react33 = require("react");
|
1317
|
-
|
1318
|
-
// src/context/providers/AssistantProvider.tsx
|
1319
|
-
var import_react32 = require("react");
|
1320
|
-
|
1321
|
-
// src/context/AssistantContext.ts
|
1322
|
-
var import_react30 = require("react");
|
1323
|
-
var AssistantContext = (0, import_react30.createContext)(
|
1324
|
-
null
|
1325
|
-
);
|
1326
|
-
|
1327
|
-
// src/context/stores/AssistantModelConfig.ts
|
1328
|
-
var import_zustand4 = require("zustand");
|
1329
|
-
|
1330
|
-
// src/utils/ProxyConfigProvider.ts
|
1331
|
-
var ProxyConfigProvider = class {
|
1332
|
-
_providers = /* @__PURE__ */ new Set();
|
1333
|
-
getModelConfig() {
|
1334
|
-
return mergeModelConfigs([...this._providers].map((p) => p()));
|
1335
|
-
}
|
1336
|
-
registerModelConfigProvider(provider) {
|
1337
|
-
this._providers.add(provider);
|
1338
|
-
return () => {
|
1339
|
-
this._providers.delete(provider);
|
1340
|
-
};
|
1341
|
-
}
|
1342
|
-
};
|
1343
|
-
|
1344
|
-
// src/context/stores/AssistantModelConfig.ts
|
1345
|
-
var makeAssistantModelConfigStore = () => (0, import_zustand4.create)(() => {
|
1346
|
-
const proxy = new ProxyConfigProvider();
|
1347
|
-
return {
|
1348
|
-
getModelConfig: () => {
|
1349
|
-
return proxy.getModelConfig();
|
1350
|
-
},
|
1351
|
-
registerModelConfigProvider: (provider) => {
|
1352
|
-
return proxy.registerModelConfigProvider(provider);
|
1353
|
-
}
|
1354
|
-
};
|
1355
|
-
});
|
1356
|
-
|
1357
|
-
// src/context/providers/ThreadProvider.tsx
|
1358
|
-
var import_react31 = require("react");
|
1359
|
-
|
1360
|
-
// src/context/stores/Composer.ts
|
1361
|
-
var import_zustand5 = require("zustand");
|
1362
|
-
var makeComposerStore = (useThread) => (0, import_zustand5.create)()((set, get, store) => {
|
1363
|
-
return {
|
1364
|
-
...makeBaseComposer(set, get, store),
|
1365
|
-
isEditing: true,
|
1366
|
-
send: () => {
|
1367
|
-
const { setValue, value } = get();
|
1368
|
-
setValue("");
|
1369
|
-
useThread.getState().append({
|
1370
|
-
parentId: useThread.getState().messages.at(-1)?.id ?? null,
|
1371
|
-
content: [{ type: "text", text: value }]
|
1372
|
-
});
|
1373
|
-
},
|
1374
|
-
cancel: () => {
|
1375
|
-
const thread = useThread.getState();
|
1376
|
-
if (!thread.isRunning) return false;
|
1377
|
-
useThread.getState().cancelRun();
|
1378
|
-
return true;
|
1379
|
-
}
|
1380
|
-
};
|
1381
|
-
});
|
1382
|
-
|
1383
|
-
// src/context/stores/Thread.ts
|
1384
|
-
var import_zustand6 = require("zustand");
|
1385
|
-
var makeThreadStore = (runtimeRef) => {
|
1386
|
-
const useThread = (0, import_zustand6.create)(() => ({
|
1387
|
-
messages: runtimeRef.current.messages,
|
1388
|
-
isRunning: runtimeRef.current.isRunning,
|
1389
|
-
getBranches: (messageId) => runtimeRef.current.getBranches(messageId),
|
1390
|
-
switchToBranch: (branchId) => runtimeRef.current.switchToBranch(branchId),
|
1391
|
-
startRun: (parentId) => runtimeRef.current.startRun(parentId),
|
1392
|
-
append: (message) => runtimeRef.current.append(message),
|
1393
|
-
cancelRun: () => runtimeRef.current.cancelRun()
|
1394
|
-
}));
|
1395
|
-
const onRuntimeUpdate = () => {
|
1396
|
-
useThread.setState({
|
1397
|
-
messages: runtimeRef.current.messages,
|
1398
|
-
isRunning: runtimeRef.current.isRunning
|
1399
|
-
});
|
1400
|
-
};
|
1401
|
-
return {
|
1402
|
-
useThread,
|
1403
|
-
onRuntimeUpdate
|
1404
|
-
};
|
1405
|
-
};
|
1406
|
-
|
1407
|
-
// src/context/stores/ThreadViewport.tsx
|
1408
|
-
var import_zustand7 = require("zustand");
|
1409
|
-
var makeThreadViewportStore = () => {
|
1410
|
-
const scrollToBottomListeners = /* @__PURE__ */ new Set();
|
1411
|
-
return (0, import_zustand7.create)(() => ({
|
1412
|
-
isAtBottom: true,
|
1413
|
-
scrollToBottom: () => {
|
1414
|
-
for (const listener of scrollToBottomListeners) {
|
1415
|
-
listener();
|
1416
|
-
}
|
1417
|
-
},
|
1418
|
-
onScrollToBottom: (callback) => {
|
1419
|
-
scrollToBottomListeners.add(callback);
|
1420
|
-
return () => {
|
1421
|
-
scrollToBottomListeners.delete(callback);
|
1422
|
-
};
|
1423
|
-
}
|
1424
|
-
}));
|
1425
|
-
};
|
1426
|
-
|
1427
|
-
// src/context/providers/ThreadProvider.tsx
|
1428
|
-
var import_jsx_runtime21 = require("react/jsx-runtime");
|
1429
|
-
var ThreadProvider = ({
|
1430
|
-
children,
|
1431
|
-
runtime
|
1432
|
-
}) => {
|
1433
|
-
const runtimeRef = (0, import_react31.useRef)(runtime);
|
1434
|
-
(0, import_react31.useInsertionEffect)(() => {
|
1435
|
-
runtimeRef.current = runtime;
|
1436
|
-
});
|
1437
|
-
const [{ context, onRuntimeUpdate }] = (0, import_react31.useState)(() => {
|
1438
|
-
const { useThread, onRuntimeUpdate: onRuntimeUpdate2 } = makeThreadStore(runtimeRef);
|
1439
|
-
const useViewport = makeThreadViewportStore();
|
1440
|
-
const useComposer = makeComposerStore(useThread);
|
1441
|
-
return {
|
1442
|
-
context: {
|
1443
|
-
useViewport,
|
1444
|
-
useThread,
|
1445
|
-
useComposer
|
1446
|
-
},
|
1447
|
-
onRuntimeUpdate: onRuntimeUpdate2
|
1448
|
-
};
|
1449
|
-
});
|
1450
|
-
(0, import_react31.useEffect)(() => {
|
1451
|
-
onRuntimeUpdate();
|
1452
|
-
return runtime.subscribe(onRuntimeUpdate);
|
1453
|
-
}, [onRuntimeUpdate, runtime]);
|
1454
|
-
const RuntimeSynchronizer = runtime.unstable_synchronizer;
|
1455
|
-
return /* @__PURE__ */ (0, import_jsx_runtime21.jsxs)(ThreadContext.Provider, { value: context, children: [
|
1456
|
-
RuntimeSynchronizer && /* @__PURE__ */ (0, import_jsx_runtime21.jsx)(RuntimeSynchronizer, {}),
|
1457
|
-
children
|
1458
|
-
] });
|
1459
|
-
};
|
1460
|
-
|
1461
|
-
// src/context/providers/AssistantProvider.tsx
|
1462
|
-
var import_jsx_runtime22 = require("react/jsx-runtime");
|
1463
|
-
var AssistantProvider = ({ children, runtime }) => {
|
1464
|
-
const runtimeRef = (0, import_react32.useRef)(runtime);
|
1465
|
-
(0, import_react32.useInsertionEffect)(() => {
|
1466
|
-
runtimeRef.current = runtime;
|
1467
|
-
});
|
1468
|
-
const [context] = (0, import_react32.useState)(() => {
|
1469
|
-
const useModelConfig = makeAssistantModelConfigStore();
|
1470
|
-
return { useModelConfig };
|
1471
|
-
});
|
1472
|
-
const getModelCOnfig = context.useModelConfig((c) => c.getModelConfig);
|
1473
|
-
(0, import_react32.useEffect)(() => {
|
1474
|
-
return runtime.registerModelConfigProvider(getModelCOnfig);
|
1475
|
-
}, [runtime, getModelCOnfig]);
|
1476
|
-
return /* @__PURE__ */ (0, import_jsx_runtime22.jsx)(AssistantContext.Provider, { value: context, children: /* @__PURE__ */ (0, import_jsx_runtime22.jsx)(ThreadProvider, { runtime, children }) });
|
1477
|
-
};
|
1478
|
-
|
1479
|
-
// src/context/providers/AssistantRuntimeProvider.tsx
|
1480
|
-
var import_jsx_runtime23 = require("react/jsx-runtime");
|
1481
|
-
var AssistantRuntimeProviderImpl = ({ children, runtime }) => {
|
1482
|
-
return /* @__PURE__ */ (0, import_jsx_runtime23.jsx)(AssistantProvider, { runtime, children });
|
1483
|
-
};
|
1484
|
-
var AssistantRuntimeProvider = (0, import_react33.memo)(AssistantRuntimeProviderImpl);
|
1485
|
-
|
1486
|
-
// src/internal.ts
|
1487
|
-
var internal_exports = {};
|
1488
|
-
__export(internal_exports, {
|
1489
|
-
MessageRepository: () => MessageRepository,
|
1490
|
-
ProxyConfigProvider: () => ProxyConfigProvider
|
1491
|
-
});
|
1492
|
-
// Annotate the CommonJS export names for ESM import in node:
|
1493
|
-
0 && (module.exports = {
|
1494
|
-
ActionBarPrimitive,
|
1495
|
-
AssistantRuntimeProvider,
|
1496
|
-
BranchPickerPrimitive,
|
1497
|
-
ComposerPrimitive,
|
1498
|
-
ContentPartPrimitive,
|
1499
|
-
INTERNAL,
|
1500
|
-
MessagePrimitive,
|
1501
|
-
ThreadPrimitive,
|
1502
|
-
useBeginMessageEdit,
|
1503
|
-
useCopyMessage,
|
1504
|
-
useGoToNextBranch,
|
1505
|
-
useGoToPreviousBranch,
|
1506
|
-
useLocalRuntime,
|
1507
|
-
useReloadMessage
|
1508
|
-
});
|
1509
|
-
//# sourceMappingURL=index.js.map
|