@assistant-ui/react 0.2.2 → 0.2.3

package/dist/index.js

@@ -60,7 +60,6 @@ __export(src_exports, {
   useContentPartContext: () => useContentPartContext,
   useContentPartDisplay: () => useContentPartDisplay,
   useContentPartImage: () => useContentPartImage,
-  useContentPartInProgressIndicator: () => useContentPartInProgressIndicator,
   useContentPartText: () => useContentPartText,
   useLocalRuntime: () => useLocalRuntime,
   useMessageContext: () => useMessageContext,
@@ -211,7 +210,7 @@ var makeBaseComposer = (set) => ({
 });
 
 // src/context/stores/Composer.ts
-var makeComposerStore = (useThread, useThreadActions) => {
+var makeComposerStore = (useThread, useThreadMessages, useThreadActions) => {
   const focusListeners = /* @__PURE__ */ new Set();
   return (0, import_zustand3.create)()((set, get, store) => {
     return {
@@ -221,7 +220,7 @@ var makeComposerStore = (useThread, useThreadActions) => {
         const { setValue, value } = get();
         setValue("");
         useThreadActions.getState().append({
-          parentId: useThread.getState().messages.at(-1)?.id ?? null,
+          parentId: useThreadMessages.getState().at(-1)?.id ?? null,
           role: "user",
           content: [{ type: "text", text: value }]
         });
@@ -251,7 +250,6 @@ var makeComposerStore = (useThread, useThreadActions) => {
 var import_zustand4 = require("zustand");
 var makeThreadStore = (runtimeRef) => {
   return (0, import_zustand4.create)(() => ({
-    messages: runtimeRef.current.messages,
     isRunning: runtimeRef.current.isRunning
   }));
 };
@@ -291,6 +289,12 @@ var makeThreadActionStore = (runtimeRef) => {
   );
 };
 
+// src/context/stores/ThreadMessages.ts
+var import_zustand7 = require("zustand");
+var makeThreadMessagesStore = (runtimeRef) => {
+  return (0, import_zustand7.create)(() => runtimeRef.current.messages);
+};
+
 // src/context/providers/ThreadProvider.tsx
 var import_jsx_runtime = require("react/jsx-runtime");
 var ThreadProvider = ({
@@ -303,11 +307,17 @@ var ThreadProvider = ({
   });
   const [context] = (0, import_react3.useState)(() => {
     const useThread = makeThreadStore(runtimeRef);
+    const useThreadMessages = makeThreadMessagesStore(runtimeRef);
     const useThreadActions = makeThreadActionStore(runtimeRef);
     const useViewport = makeThreadViewportStore();
-    const useComposer = makeComposerStore(useThread, useThreadActions);
+    const useComposer = makeComposerStore(
+      useThread,
+      useThreadMessages,
+      useThreadActions
+    );
     return {
       useThread,
+      useThreadMessages,
       useThreadActions,
       useComposer,
       useViewport
@@ -317,11 +327,11 @@ var ThreadProvider = ({
     const onRuntimeUpdate = () => {
       context.useThread.setState(
         Object.freeze({
-          messages: runtimeRef.current.messages,
           isRunning: runtimeRef.current.isRunning
         }),
         true
       );
+      context.useThreadMessages.setState(Object.freeze(runtimeRef.current.messages), true);
     };
     onRuntimeUpdate();
     return runtime.subscribe(onRuntimeUpdate);
@@ -342,8 +352,8 @@ var ThreadProvider = ({
 };
 
 // src/context/stores/AssistantActions.tsx
-var import_zustand7 = require("zustand");
-var makeAssistantActionsStore = (runtimeRef) => (0, import_zustand7.create)(
+var import_zustand8 = require("zustand");
+var makeAssistantActionsStore = (runtimeRef) => (0, import_zustand8.create)(
   () => Object.freeze({
     switchToThread: () => runtimeRef.current.switchToThread(null)
   })
@@ -420,30 +430,30 @@ var useContentPartContext = () => {
 
 // src/hooks/useAppendMessage.tsx
 var import_react9 = require("react");
-var toAppendMessage = (useThread, message) => {
+var toAppendMessage = (useThreadMessages, message) => {
   if (typeof message === "string") {
     return {
-      parentId: useThread.getState().messages.at(-1)?.id ?? null,
+      parentId: useThreadMessages.getState().at(-1)?.id ?? null,
       role: "user",
       content: [{ type: "text", text: message }]
     };
   }
   return {
-    parentId: message.parentId ?? useThread.getState().messages.at(-1)?.id ?? null,
+    parentId: message.parentId ?? useThreadMessages.getState().at(-1)?.id ?? null,
     role: message.role ?? "user",
     content: message.content
   };
 };
 var useAppendMessage = () => {
-  const { useThread, useThreadActions, useViewport, useComposer } = useThreadContext();
+  const { useThreadMessages, useThreadActions, useViewport, useComposer } = useThreadContext();
   const append = (0, import_react9.useCallback)(
     (message) => {
-      const appendMessage = toAppendMessage(useThread, message);
+      const appendMessage = toAppendMessage(useThreadMessages, message);
       useThreadActions.getState().append(appendMessage);
       useViewport.getState().scrollToBottom();
       useComposer.getState().focus();
     },
-    [useThread, useThreadActions, useViewport, useComposer]
+    [useThreadMessages, useThreadActions, useViewport, useComposer]
   );
   return append;
 };
@@ -739,17 +749,6 @@ var useContentPartImage = () => {
   return image;
 };
 
-// src/primitive-hooks/contentPart/useContentPartInProgressIndicator.tsx
-var useContentPartInProgressIndicator = () => {
-  const { useMessageUtils } = useMessageContext();
-  const { useContentPart } = useContentPartContext();
-  const indicator = useCombinedStore(
-    [useMessageUtils, useContentPart],
-    (m, c) => c.status === "in_progress" ? m.inProgressIndicator : null
-  );
-  return indicator;
-};
-
 // src/primitive-hooks/contentPart/useContentPartText.tsx
 var useContentPartText = () => {
   const { useContentPart } = useContentPartContext();
@@ -782,14 +781,17 @@ var useMessageIf = (props) => {
 
 // src/primitive-hooks/thread/useThreadIf.tsx
 var useThreadIf = (props) => {
-  const { useThread } = useThreadContext();
-  return useThread((thread) => {
-    if (props.empty === true && thread.messages.length !== 0) return false;
-    if (props.empty === false && thread.messages.length === 0) return false;
-    if (props.running === true && !thread.isRunning) return false;
-    if (props.running === false && thread.isRunning) return false;
-    return true;
-  });
+  const { useThread, useThreadMessages } = useThreadContext();
+  return useCombinedStore(
+    [useThread, useThreadMessages],
+    (thread, messages) => {
+      if (props.empty === true && messages.length !== 0) return false;
+      if (props.empty === false && messages.length === 0) return false;
+      if (props.running === true && !thread.isRunning) return false;
+      if (props.running === false && thread.isRunning) return false;
+      return true;
+    }
+  );
 };
 
 // src/primitive-hooks/thread/useThreadEmpty.tsx
@@ -1077,7 +1079,7 @@ BranchPickerPrimitiveNumber.displayName = "BranchPickerPrimitive.Number";
 
 // src/primitives/branchPicker/BranchPickerRoot.tsx
 var import_react_primitive6 = require("@radix-ui/react-primitive");
-var import_react36 = require("react");
+var import_react37 = require("react");
 
 // src/primitives/message/index.ts
 var message_exports = {};
@@ -1125,11 +1127,11 @@ var MessagePrimitiveIf = ({
 MessagePrimitiveIf.displayName = "MessagePrimitive.If";
 
 // src/primitives/message/MessageContent.tsx
-var import_react34 = require("react");
+var import_react35 = require("react");
 
 // src/context/providers/ContentPartProvider.tsx
 var import_react32 = require("react");
-var import_zustand8 = require("zustand");
+var import_zustand9 = require("zustand");
 var import_jsx_runtime12 = require("react/jsx-runtime");
 var syncContentPart = ({ message }, useContentPart, partIndex) => {
   const part = message.content[partIndex];
@@ -1148,7 +1150,7 @@ var syncContentPart = ({ message }, useContentPart, partIndex) => {
 var useContentPartContext2 = (partIndex) => {
   const { useMessage } = useMessageContext();
   const [context] = (0, import_react32.useState)(() => {
-    const useContentPart = (0, import_zustand8.create)(
+    const useContentPart = (0, import_zustand9.create)(
       () => ({})
     );
     syncContentPart(useMessage.getState(), useContentPart, partIndex);
@@ -1177,38 +1179,61 @@ var ContentPartPrimitiveDisplay = () => {
 };
 ContentPartPrimitiveDisplay.displayName = "ContentPartPrimitive.Display";
 
+// src/utils/OutPortal.tsx
+var import_react33 = require("react");
+var import_jsx_runtime13 = require("react/jsx-runtime");
+var OutPortal = ({ node }) => {
+  const parentRef = (0, import_react33.useRef)(null);
+  (0, import_react33.useLayoutEffect)(() => {
+    const parent = parentRef.current;
+    if (!parent || !node) return;
+    parent.appendChild(node);
+    return () => {
+      parent.removeChild(node);
+    };
+  }, [node]);
+  if (!node) return null;
+  return /* @__PURE__ */ (0, import_jsx_runtime13.jsx)("span", { ref: parentRef });
+};
+
 // src/primitives/contentPart/ContentPartInProgressIndicator.tsx
+var import_jsx_runtime14 = require("react/jsx-runtime");
 var ContentPartPrimitiveInProgressIndicator = () => {
-  const indicator = useContentPartInProgressIndicator();
-  return indicator;
+  const { useMessageUtils } = useMessageContext();
+  const { useContentPart } = useContentPartContext();
+  const indicator = useCombinedStore(
+    [useMessageUtils, useContentPart],
+    (m, c) => c.status === "in_progress" ? m.inProgressIndicator : null
+  );
+  return /* @__PURE__ */ (0, import_jsx_runtime14.jsx)(OutPortal, { node: indicator });
 };
 ContentPartPrimitiveInProgressIndicator.displayName = "ContentPartPrimitive.InProgressIndicator";
 
 // src/primitives/contentPart/ContentPartText.tsx
 var import_react_primitive4 = require("@radix-ui/react-primitive");
-var import_react33 = require("react");
-var import_jsx_runtime13 = require("react/jsx-runtime");
-var ContentPartPrimitiveText = (0, import_react33.forwardRef)((props, forwardedRef) => {
+var import_react34 = require("react");
+var import_jsx_runtime15 = require("react/jsx-runtime");
+var ContentPartPrimitiveText = (0, import_react34.forwardRef)((props, forwardedRef) => {
   const text = useContentPartText();
-  return /* @__PURE__ */ (0, import_jsx_runtime13.jsx)(import_react_primitive4.Primitive.p, { ...props, ref: forwardedRef, children: text });
+  return /* @__PURE__ */ (0, import_jsx_runtime15.jsx)(import_react_primitive4.Primitive.span, { ...props, ref: forwardedRef, children: text });
 });
 ContentPartPrimitiveText.displayName = "ContentPartPrimitive.Text";
 
 // src/primitives/message/MessageContent.tsx
-var import_jsx_runtime14 = require("react/jsx-runtime");
+var import_jsx_runtime16 = require("react/jsx-runtime");
 var defaultComponents = {
-  Text: () => /* @__PURE__ */ (0, import_jsx_runtime14.jsxs)(import_jsx_runtime14.Fragment, { children: [
-    /* @__PURE__ */ (0, import_jsx_runtime14.jsx)(ContentPartPrimitiveText, { style: { whiteSpace: "pre-line" } }),
-    /* @__PURE__ */ (0, import_jsx_runtime14.jsx)(ContentPartPrimitiveInProgressIndicator, {})
+  Text: () => /* @__PURE__ */ (0, import_jsx_runtime16.jsxs)("p", { style: { whiteSpace: "pre-line" }, children: [
+    /* @__PURE__ */ (0, import_jsx_runtime16.jsx)(ContentPartPrimitiveText, {}),
+    /* @__PURE__ */ (0, import_jsx_runtime16.jsx)(ContentPartPrimitiveInProgressIndicator, {})
   ] }),
   Image: () => null,
-  UI: () => /* @__PURE__ */ (0, import_jsx_runtime14.jsx)(ContentPartPrimitiveDisplay, {}),
+  UI: () => /* @__PURE__ */ (0, import_jsx_runtime16.jsx)(ContentPartPrimitiveDisplay, {}),
   tools: {
     Fallback: (props) => {
       const { useToolUIs } = useAssistantContext();
       const Render = useToolUIs((s) => s.getToolUI(props.part.toolName));
       if (!Render) return null;
-      return /* @__PURE__ */ (0, import_jsx_runtime14.jsx)(Render, { ...props });
+      return /* @__PURE__ */ (0, import_jsx_runtime16.jsx)(Render, { ...props });
     }
   }
 };
@@ -1227,15 +1252,15 @@ var MessageContentPartComponent = ({
   const type = part.type;
   switch (type) {
     case "text":
-      return /* @__PURE__ */ (0, import_jsx_runtime14.jsx)(Text, { part, status });
+      return /* @__PURE__ */ (0, import_jsx_runtime16.jsx)(Text, { part, status });
     case "image":
-      return /* @__PURE__ */ (0, import_jsx_runtime14.jsx)(Image, { part, status });
+      return /* @__PURE__ */ (0, import_jsx_runtime16.jsx)(Image, { part, status });
     case "ui":
-      return /* @__PURE__ */ (0, import_jsx_runtime14.jsx)(UI, { part, status });
+      return /* @__PURE__ */ (0, import_jsx_runtime16.jsx)(UI, { part, status });
     case "tool-call": {
       const Tool = by_name[part.toolName] || Fallback;
       const addResult = (result) => addToolResult(part.toolCallId, result);
-      return /* @__PURE__ */ (0, import_jsx_runtime14.jsx)(Tool, { part, status, addResult });
+      return /* @__PURE__ */ (0, import_jsx_runtime16.jsx)(Tool, { part, status, addResult });
     }
     default:
       throw new Error(`Unknown content part type: ${type}`);
@@ -1245,9 +1270,9 @@ var MessageContentPartImpl = ({
   partIndex,
   components
 }) => {
-  return /* @__PURE__ */ (0, import_jsx_runtime14.jsx)(ContentPartProvider, { partIndex, children: /* @__PURE__ */ (0, import_jsx_runtime14.jsx)(MessageContentPartComponent, { components }) });
+  return /* @__PURE__ */ (0, import_jsx_runtime16.jsx)(ContentPartProvider, { partIndex, children: /* @__PURE__ */ (0, import_jsx_runtime16.jsx)(MessageContentPartComponent, { components }) });
 };
-var MessageContentPart = (0, import_react34.memo)(
+var MessageContentPart = (0, import_react35.memo)(
   MessageContentPartImpl,
   (prev, next) => prev.partIndex === next.partIndex && prev.components?.Text === next.components?.Text && prev.components?.Image === next.components?.Image && prev.components?.UI === next.components?.UI && prev.components?.tools === next.components?.tools
 );
@@ -1258,7 +1283,7 @@ var MessagePrimitiveContent = ({
   const contentLength = useMessage((s) => s.message.content.length);
   return new Array(contentLength).fill(null).map((_, idx) => {
     const partIndex = idx;
-    return /* @__PURE__ */ (0, import_jsx_runtime14.jsx)(
+    return /* @__PURE__ */ (0, import_jsx_runtime16.jsx)(
       MessageContentPart,
       {
         partIndex,
@@ -1271,25 +1296,21 @@ var MessagePrimitiveContent = ({
 MessagePrimitiveContent.displayName = "MessagePrimitive.Content";
 
 // src/primitives/message/MessageInProgress.tsx
+var import_react_dom = require("react-dom");
 var import_react_primitive5 = require("@radix-ui/react-primitive");
-var import_react35 = require("react");
-var import_jsx_runtime15 = require("react/jsx-runtime");
-var MessagePrimitiveInProgress = (0, import_react35.forwardRef)((props, ref) => {
+var import_react36 = require("react");
+var import_jsx_runtime17 = require("react/jsx-runtime");
+var MessagePrimitiveInProgress = (0, import_react36.forwardRef)((props, ref) => {
   const { useMessageUtils } = useMessageContext();
-  (0, import_react35.useEffect)(() => {
-    useMessageUtils.getState().setInProgressIndicator(/* @__PURE__ */ (0, import_jsx_runtime15.jsx)(import_react_primitive5.Primitive.span, { ...props, ref }));
-    return () => {
-      useMessageUtils.getState().setInProgressIndicator(null);
-    };
-  }, [useMessageUtils, props, ref]);
-  return null;
+  const portalNode = useMessageUtils((s) => s.inProgressIndicator);
+  return (0, import_react_dom.createPortal)(/* @__PURE__ */ (0, import_jsx_runtime17.jsx)(import_react_primitive5.Primitive.span, { ...props, ref }), portalNode);
 });
 MessagePrimitiveInProgress.displayName = "MessagePrimitive.InProgress";
 
 // src/primitives/branchPicker/BranchPickerRoot.tsx
-var import_jsx_runtime16 = require("react/jsx-runtime");
-var BranchPickerPrimitiveRoot = (0, import_react36.forwardRef)(({ hideWhenSingleBranch, ...rest }, ref) => {
-  return /* @__PURE__ */ (0, import_jsx_runtime16.jsx)(MessagePrimitiveIf, { hasBranches: hideWhenSingleBranch ? true : void 0, children: /* @__PURE__ */ (0, import_jsx_runtime16.jsx)(import_react_primitive6.Primitive.div, { ...rest, ref }) });
+var import_jsx_runtime18 = require("react/jsx-runtime");
+var BranchPickerPrimitiveRoot = (0, import_react37.forwardRef)(({ hideWhenSingleBranch, ...rest }, ref) => {
+  return /* @__PURE__ */ (0, import_jsx_runtime18.jsx)(MessagePrimitiveIf, { hasBranches: hideWhenSingleBranch ? true : void 0, children: /* @__PURE__ */ (0, import_jsx_runtime18.jsx)(import_react_primitive6.Primitive.div, { ...rest, ref }) });
 });
 BranchPickerPrimitiveRoot.displayName = "BranchPickerPrimitive.Root";
 
@@ -1306,16 +1327,16 @@ __export(composer_exports, {
 // src/primitives/composer/ComposerRoot.tsx
 var import_primitive5 = require("@radix-ui/primitive");
 var import_react_primitive7 = require("@radix-ui/react-primitive");
-var import_react37 = require("react");
-var import_jsx_runtime17 = require("react/jsx-runtime");
-var ComposerPrimitiveRoot = (0, import_react37.forwardRef)(({ onSubmit, ...rest }, forwardedRef) => {
+var import_react38 = require("react");
+var import_jsx_runtime19 = require("react/jsx-runtime");
+var ComposerPrimitiveRoot = (0, import_react38.forwardRef)(({ onSubmit, ...rest }, forwardedRef) => {
   const send = useComposerSend();
   const handleSubmit = (e) => {
     e.preventDefault();
     if (!send) return;
     send();
   };
-  return /* @__PURE__ */ (0, import_jsx_runtime17.jsx)(
+  return /* @__PURE__ */ (0, import_jsx_runtime19.jsx)(
     import_react_primitive7.Primitive.form,
     {
       ...rest,
@@ -1330,11 +1351,11 @@ ComposerPrimitiveRoot.displayName = "ComposerPrimitive.Root";
 var import_primitive6 = require("@radix-ui/primitive");
 var import_react_compose_refs = require("@radix-ui/react-compose-refs");
 var import_react_slot = require("@radix-ui/react-slot");
-var import_react38 = require("react");
+var import_react39 = require("react");
 var import_react_textarea_autosize = __toESM(require("react-textarea-autosize"));
 var import_react_use_escape_keydown = require("@radix-ui/react-use-escape-keydown");
-var import_jsx_runtime18 = require("react/jsx-runtime");
-var ComposerPrimitiveInput = (0, import_react38.forwardRef)(
+var import_jsx_runtime20 = require("react/jsx-runtime");
+var ComposerPrimitiveInput = (0, import_react39.forwardRef)(
   ({ autoFocus = false, asChild, disabled, onChange, onKeyDown, ...rest }, forwardedRef) => {
     const { useThread } = useThreadContext();
     const { useComposer, type } = useComposerContext();
@@ -1343,7 +1364,7 @@ var ComposerPrimitiveInput = (0, import_react38.forwardRef)(
       return c.value;
     });
     const Component = asChild ? import_react_slot.Slot : import_react_textarea_autosize.default;
-    const textareaRef = (0, import_react38.useRef)(null);
+    const textareaRef = (0, import_react39.useRef)(null);
     const ref = (0, import_react_compose_refs.useComposedRefs)(forwardedRef, textareaRef);
     (0, import_react_use_escape_keydown.useEscapeKeydown)((e) => {
       const composer = useComposer.getState();
@@ -1362,7 +1383,7 @@ var ComposerPrimitiveInput = (0, import_react38.forwardRef)(
       }
     };
     const autoFocusEnabled = autoFocus && !disabled;
-    const focus = (0, import_react38.useCallback)(() => {
+    const focus = (0, import_react39.useCallback)(() => {
       const textarea = textareaRef.current;
       if (!textarea || !autoFocusEnabled) return;
       textarea.focus({ preventScroll: true });
@@ -1371,13 +1392,13 @@ var ComposerPrimitiveInput = (0, import_react38.forwardRef)(
         textareaRef.current.value.length
       );
     }, [autoFocusEnabled]);
-    (0, import_react38.useEffect)(() => focus(), [focus]);
+    (0, import_react39.useEffect)(() => focus(), [focus]);
     useOnComposerFocus(() => {
       if (type === "new") {
         focus();
       }
     });
-    return /* @__PURE__ */ (0, import_jsx_runtime18.jsx)(
+    return /* @__PURE__ */ (0, import_jsx_runtime20.jsx)(
       Component,
       {
         value,
@@ -1397,13 +1418,13 @@ var ComposerPrimitiveInput = (0, import_react38.forwardRef)(
 ComposerPrimitiveInput.displayName = "ComposerPrimitive.Input";
 
 // src/primitives/composer/ComposerSend.tsx
-var import_react39 = require("react");
+var import_react40 = require("react");
 var import_react_primitive8 = require("@radix-ui/react-primitive");
-var import_jsx_runtime19 = require("react/jsx-runtime");
-var ComposerPrimitiveSend = (0, import_react39.forwardRef)(({ disabled, ...rest }, ref) => {
+var import_jsx_runtime21 = require("react/jsx-runtime");
+var ComposerPrimitiveSend = (0, import_react40.forwardRef)(({ disabled, ...rest }, ref) => {
   const { useComposer } = useComposerContext();
   const hasValue = useComposer((c) => c.isEditing && c.value.length > 0);
-  return /* @__PURE__ */ (0, import_jsx_runtime19.jsx)(
+  return /* @__PURE__ */ (0, import_jsx_runtime21.jsx)(
     import_react_primitive8.Primitive.button,
     {
       type: "submit",
@@ -1442,11 +1463,11 @@ __export(contentPart_exports, {
 
 // src/primitives/contentPart/ContentPartImage.tsx
 var import_react_primitive9 = require("@radix-ui/react-primitive");
-var import_react40 = require("react");
-var import_jsx_runtime20 = require("react/jsx-runtime");
-var ContentPartPrimitiveImage = (0, import_react40.forwardRef)((props, forwardedRef) => {
+var import_react41 = require("react");
+var import_jsx_runtime22 = require("react/jsx-runtime");
+var ContentPartPrimitiveImage = (0, import_react41.forwardRef)((props, forwardedRef) => {
   const image = useContentPartImage();
-  return /* @__PURE__ */ (0, import_jsx_runtime20.jsx)(import_react_primitive9.Primitive.img, { src: image, ...props, ref: forwardedRef });
+  return /* @__PURE__ */ (0, import_jsx_runtime22.jsx)(import_react_primitive9.Primitive.img, { src: image, ...props, ref: forwardedRef });
 });
 ContentPartPrimitiveImage.displayName = "ContentPartPrimitive.Image";
 
@@ -1464,10 +1485,10 @@ __export(thread_exports, {
 
 // src/primitives/thread/ThreadRoot.tsx
 var import_react_primitive10 = require("@radix-ui/react-primitive");
-var import_react41 = require("react");
-var import_jsx_runtime21 = require("react/jsx-runtime");
-var ThreadPrimitiveRoot = (0, import_react41.forwardRef)((props, ref) => {
-  return /* @__PURE__ */ (0, import_jsx_runtime21.jsx)(import_react_primitive10.Primitive.div, { ...props, ref });
+var import_react42 = require("react");
+var import_jsx_runtime23 = require("react/jsx-runtime");
+var ThreadPrimitiveRoot = (0, import_react42.forwardRef)((props, ref) => {
+  return /* @__PURE__ */ (0, import_jsx_runtime23.jsx)(import_react_primitive10.Primitive.div, { ...props, ref });
 });
 ThreadPrimitiveRoot.displayName = "ThreadPrimitive.Root";
 
@@ -1493,21 +1514,21 @@ ThreadPrimitiveIf.displayName = "ThreadPrimitive.If";
 // src/primitives/thread/ThreadViewport.tsx
 var import_react_compose_refs3 = require("@radix-ui/react-compose-refs");
 var import_react_primitive11 = require("@radix-ui/react-primitive");
-var import_react46 = require("react");
+var import_react47 = require("react");
 
 // src/primitive-hooks/thread/useThreadViewportAutoScroll.tsx
 var import_react_compose_refs2 = require("@radix-ui/react-compose-refs");
-var import_react45 = require("react");
+var import_react46 = require("react");
 
 // src/utils/hooks/useOnResizeContent.tsx
 var import_react_use_callback_ref2 = require("@radix-ui/react-use-callback-ref");
-var import_react43 = require("react");
+var import_react44 = require("react");
 
 // src/utils/hooks/useManagedRef.ts
-var import_react42 = require("react");
+var import_react43 = require("react");
 var useManagedRef = (callback) => {
-  const cleanupRef = (0, import_react42.useRef)();
-  const ref = (0, import_react42.useCallback)(
+  const cleanupRef = (0, import_react43.useRef)();
+  const ref = (0, import_react43.useCallback)(
     (el) => {
       if (cleanupRef.current) {
         cleanupRef.current();
@@ -1524,7 +1545,7 @@ var useManagedRef = (callback) => {
 // src/utils/hooks/useOnResizeContent.tsx
 var useOnResizeContent = (callback) => {
   const callbackRef = (0, import_react_use_callback_ref2.useCallbackRef)(callback);
-  const refCallback = (0, import_react43.useCallback)(
+  const refCallback = (0, import_react44.useCallback)(
     (el) => {
       const resizeObserver = new ResizeObserver(() => {
         callbackRef();
@@ -1561,11 +1582,11 @@ var useOnResizeContent = (callback) => {
 
 // src/utils/hooks/useOnScrollToBottom.tsx
 var import_react_use_callback_ref3 = require("@radix-ui/react-use-callback-ref");
-var import_react44 = require("react");
+var import_react45 = require("react");
 var useOnScrollToBottom = (callback) => {
   const callbackRef = (0, import_react_use_callback_ref3.useCallbackRef)(callback);
   const { useViewport } = useThreadContext();
-  (0, import_react44.useEffect)(() => {
+  (0, import_react45.useEffect)(() => {
     return useViewport.getState().onScrollToBottom(() => {
       callbackRef();
     });
@@ -1576,11 +1597,11 @@ var useOnScrollToBottom = (callback) => {
 var useThreadViewportAutoScroll = ({
   autoScroll = true
 }) => {
-  const divRef = (0, import_react45.useRef)(null);
+  const divRef = (0, import_react46.useRef)(null);
   const { useViewport } = useThreadContext();
-  const firstRenderRef = (0, import_react45.useRef)(true);
-  const lastScrollTop = (0, import_react45.useRef)(0);
-  const isScrollingToBottomRef = (0, import_react45.useRef)(false);
+  const firstRenderRef = (0, import_react46.useRef)(true);
+  const lastScrollTop = (0, import_react46.useRef)(0);
+  const isScrollingToBottomRef = (0, import_react46.useRef)(false);
   const scrollToBottom = () => {
     const div = divRef.current;
     if (!div || !autoScroll) return;
@@ -1626,29 +1647,29 @@ var useThreadViewportAutoScroll = ({
 };
 
 // src/primitives/thread/ThreadViewport.tsx
-var import_jsx_runtime22 = require("react/jsx-runtime");
-var ThreadPrimitiveViewport = (0, import_react46.forwardRef)(({ autoScroll, onScroll, children, ...rest }, forwardedRef) => {
+var import_jsx_runtime24 = require("react/jsx-runtime");
+var ThreadPrimitiveViewport = (0, import_react47.forwardRef)(({ autoScroll, onScroll, children, ...rest }, forwardedRef) => {
   const autoScrollRef = useThreadViewportAutoScroll({
     autoScroll
   });
   const ref = (0, import_react_compose_refs3.useComposedRefs)(forwardedRef, autoScrollRef);
-  return /* @__PURE__ */ (0, import_jsx_runtime22.jsx)(import_react_primitive11.Primitive.div, { ...rest, ref, children });
+  return /* @__PURE__ */ (0, import_jsx_runtime24.jsx)(import_react_primitive11.Primitive.div, { ...rest, ref, children });
 });
 ThreadPrimitiveViewport.displayName = "ThreadPrimitive.Viewport";
 
 // src/primitives/thread/ThreadMessages.tsx
-var import_react48 = require("react");
+var import_react49 = require("react");
 
 // src/context/providers/MessageProvider.tsx
-var import_react47 = require("react");
-var import_zustand11 = require("zustand");
+var import_react48 = require("react");
+var import_zustand12 = require("zustand");
 
 // src/context/stores/EditComposer.ts
-var import_zustand9 = require("zustand");
+var import_zustand10 = require("zustand");
 var makeEditComposerStore = ({
   onEdit,
   onSend
-}) => (0, import_zustand9.create)()((set, get, store) => ({
+}) => (0, import_zustand10.create)()((set, get, store) => ({
   ...makeBaseComposer(set, get, store),
   isEditing: false,
   edit: () => {
@@ -1668,12 +1689,9 @@ var makeEditComposerStore = ({
 }));
 
 // src/context/stores/MessageUtils.ts
-var import_zustand10 = require("zustand");
-var makeMessageUtilsStore = () => (0, import_zustand10.create)((set) => ({
-  inProgressIndicator: null,
-  setInProgressIndicator: (value) => {
-    set({ inProgressIndicator: value });
-  },
+var import_zustand11 = require("zustand");
+var makeMessageUtilsStore = () => (0, import_zustand11.create)((set) => ({
+  inProgressIndicator: document.createElement("span"),
   isCopied: false,
   setIsCopied: (value) => {
     set({ isCopied: value });
@@ -1685,15 +1703,15 @@ var makeMessageUtilsStore = () => (0, import_zustand10.create)((set) => ({
 }));
 
 // src/context/providers/MessageProvider.tsx
-var import_jsx_runtime23 = require("react/jsx-runtime");
-var getIsLast = (thread, message) => {
-  return thread.messages[thread.messages.length - 1]?.id === message.id;
+var import_jsx_runtime25 = require("react/jsx-runtime");
+var getIsLast = (messages, message) => {
+  return messages[messages.length - 1]?.id === message.id;
 };
-var syncMessage = (thread, getBranches, useMessage, messageIndex) => {
-  const parentId = thread.messages[messageIndex - 1]?.id ?? null;
-  const message = thread.messages[messageIndex];
+var syncMessage = (messages, getBranches, useMessage, messageIndex) => {
+  const parentId = messages[messageIndex - 1]?.id ?? null;
+  const message = messages[messageIndex];
   if (!message) return;
-  const isLast = getIsLast(thread, message);
+  const isLast = getIsLast(messages, message);
   const branches = getBranches(message.id);
   const currentState = useMessage.getState();
   if (currentState.message === message && currentState.parentId === parentId && currentState.branches === branches && currentState.isLast === isLast)
@@ -1706,9 +1724,9 @@ var syncMessage = (thread, getBranches, useMessage, messageIndex) => {
   });
 };
 var useMessageContext2 = (messageIndex) => {
-  const { useThread, useThreadActions } = useThreadContext();
-  const [context] = (0, import_react47.useState)(() => {
-    const useMessage = (0, import_zustand11.create)(() => ({}));
+  const { useThreadMessages, useThreadActions } = useThreadContext();
+  const [context] = (0, import_react48.useState)(() => {
+    const useMessage = (0, import_zustand12.create)(() => ({}));
     const useMessageUtils = makeMessageUtilsStore();
     const useEditComposer = makeEditComposerStore({
       onEdit: () => {
@@ -1737,15 +1755,15 @@ var useMessageContext2 = (messageIndex) => {
       }
     });
     syncMessage(
-      useThread.getState(),
+      useThreadMessages.getState(),
       useThreadActions.getState().getBranches,
       useMessage,
       messageIndex
     );
     return { useMessage, useMessageUtils, useEditComposer };
   });
-  (0, import_react47.useEffect)(() => {
-    return useThread.subscribe((thread) => {
+  (0, import_react48.useEffect)(() => {
+    return useThreadMessages.subscribe((thread) => {
       syncMessage(
         thread,
         useThreadActions.getState().getBranches,
@@ -1753,7 +1771,7 @@ var useMessageContext2 = (messageIndex) => {
         messageIndex
       );
     });
-  }, [useThread, useThreadActions, context, messageIndex]);
+  }, [useThreadMessages, useThreadActions, context, messageIndex]);
   return context;
 };
 var MessageProvider = ({
@@ -1761,11 +1779,11 @@ var MessageProvider = ({
   children
 }) => {
   const context = useMessageContext2(messageIndex);
-  return /* @__PURE__ */ (0, import_jsx_runtime23.jsx)(MessageContext.Provider, { value: context, children });
+  return /* @__PURE__ */ (0, import_jsx_runtime25.jsx)(MessageContext.Provider, { value: context, children });
 };
 
 // src/primitives/thread/ThreadMessages.tsx
-var import_jsx_runtime24 = require("react/jsx-runtime");
+var import_jsx_runtime26 = require("react/jsx-runtime");
 var getComponents = (components) => {
   return {
     EditComposer: components.EditComposer ?? components.UserMessage ?? components.Message,
@@ -1778,27 +1796,27 @@ var ThreadMessageImpl = ({
   components
 }) => {
   const { UserMessage, EditComposer, AssistantMessage } = getComponents(components);
-  return /* @__PURE__ */ (0, import_jsx_runtime24.jsxs)(MessageProvider, { messageIndex, children: [
-    /* @__PURE__ */ (0, import_jsx_runtime24.jsxs)(MessagePrimitiveIf, { user: true, children: [
-      /* @__PURE__ */ (0, import_jsx_runtime24.jsx)(ComposerPrimitiveIf, { editing: false, children: /* @__PURE__ */ (0, import_jsx_runtime24.jsx)(UserMessage, {}) }),
-      /* @__PURE__ */ (0, import_jsx_runtime24.jsx)(ComposerPrimitiveIf, { editing: true, children: /* @__PURE__ */ (0, import_jsx_runtime24.jsx)(EditComposer, {}) })
+  return /* @__PURE__ */ (0, import_jsx_runtime26.jsxs)(MessageProvider, { messageIndex, children: [
+    /* @__PURE__ */ (0, import_jsx_runtime26.jsxs)(MessagePrimitiveIf, { user: true, children: [
+      /* @__PURE__ */ (0, import_jsx_runtime26.jsx)(ComposerPrimitiveIf, { editing: false, children: /* @__PURE__ */ (0, import_jsx_runtime26.jsx)(UserMessage, {}) }),
+      /* @__PURE__ */ (0, import_jsx_runtime26.jsx)(ComposerPrimitiveIf, { editing: true, children: /* @__PURE__ */ (0, import_jsx_runtime26.jsx)(EditComposer, {}) })
     ] }),
-    /* @__PURE__ */ (0, import_jsx_runtime24.jsx)(MessagePrimitiveIf, { assistant: true, children: /* @__PURE__ */ (0, import_jsx_runtime24.jsx)(AssistantMessage, {}) })
+    /* @__PURE__ */ (0, import_jsx_runtime26.jsx)(MessagePrimitiveIf, { assistant: true, children: /* @__PURE__ */ (0, import_jsx_runtime26.jsx)(AssistantMessage, {}) })
   ] });
 };
-var ThreadMessage = (0, import_react48.memo)(
+var ThreadMessage = (0, import_react49.memo)(
   ThreadMessageImpl,
   (prev, next) => prev.messageIndex === next.messageIndex && prev.components.UserMessage === next.components.UserMessage && prev.components.EditComposer === next.components.EditComposer && prev.components.AssistantMessage === next.components.AssistantMessage
 );
 var ThreadPrimitiveMessages = ({
   components
 }) => {
-  const { useThread } = useThreadContext();
-  const messagesLength = useThread((t) => t.messages.length);
+  const { useThreadMessages } = useThreadContext();
+  const messagesLength = useThreadMessages((t) => t.length);
   if (messagesLength === 0) return null;
   return new Array(messagesLength).fill(null).map((_, idx) => {
     const messageIndex = idx;
-    return /* @__PURE__ */ (0, import_jsx_runtime24.jsx)(
+    return /* @__PURE__ */ (0, import_jsx_runtime26.jsx)(
       ThreadMessage,
       {
         messageIndex,
@@ -1823,7 +1841,7 @@ var ThreadPrimitiveSuggestion = createActionButton(
 );
 
 // src/runtime/local/useLocalRuntime.tsx
-var import_react49 = require("react");
+var import_react50 = require("react");
 
 // src/runtime/utils/idUtils.tsx
 var import_non_secure = require("nanoid/non-secure");
@@ -2155,8 +2173,8 @@ var LocalThreadRuntime = class {
 
 // src/runtime/local/useLocalRuntime.tsx
 var useLocalRuntime = (adapter) => {
-  const [runtime] = (0, import_react49.useState)(() => new LocalRuntime(adapter));
-  (0, import_react49.useInsertionEffect)(() => {
+  const [runtime] = (0, import_react50.useState)(() => new LocalRuntime(adapter));
+  (0, import_react50.useInsertionEffect)(() => {
     runtime.adapter = adapter;
   });
   return runtime;
@@ -2201,7 +2219,6 @@ __export(internal_exports, {
   useContentPartContext,
   useContentPartDisplay,
   useContentPartImage,
-  useContentPartInProgressIndicator,
   useContentPartText,
   useLocalRuntime,
   useMessageContext,