@assistant-ui/react 0.5.51 → 0.5.53

package/dist/index.mjs

@@ -373,7 +373,8 @@ var fromCoreMessages = (message) => {
 };
 var fromCoreMessage = (message, {
   id = generateId(),
-  status = { type: "complete", reason: "unknown" }
+  status = { type: "complete", reason: "unknown" },
+  attachments = []
 } = {}) => {
   const commonProps = {
     id,
@@ -401,7 +402,7 @@ var fromCoreMessage = (message, {
         ...commonProps,
         role,
         content: message.content,
-        attachments: []
+        attachments
       };
     case "system":
       return {
@@ -1218,20 +1219,16 @@ var LocalThreadRuntime = class {
     this.notifySubscribers();
   }
   async append(message) {
-    if (message.role !== "user")
-      throw new Error(
-        "Only appending user messages are supported in LocalRuntime. This is likely an internal bug in assistant-ui."
-      );
-    const userMessageId = generateId();
-    const userMessage = {
-      id: userMessageId,
-      role: "user",
-      content: message.content,
-      attachments: message.attachments ?? [],
-      createdAt: /* @__PURE__ */ new Date()
-    };
-    this.repository.addOrUpdateMessage(message.parentId, userMessage);
-    await this.startRun(userMessageId);
+    const newMessage = fromCoreMessage(message, {
+      attachments: message.attachments
+    });
+    this.repository.addOrUpdateMessage(message.parentId, newMessage);
+    if (message.role === "user") {
+      await this.startRun(newMessage.id);
+    } else {
+      this.repository.resetHead(newMessage.id);
+      this.notifySubscribers();
+    }
   }
   async startRun(parentId) {
     this.repository.resetHead(parentId);
@@ -2271,14 +2268,47 @@ var AssistantRuntimeProviderImpl = ({ children, runtime }) => {
 };
 var AssistantRuntimeProvider = memo(AssistantRuntimeProviderImpl);
 
+// src/context/providers/TextContentPartProvider.tsx
+import { useState as useState9 } from "react";
+import { create as create12 } from "zustand";
+
+// src/context/react/ContentPartContext.ts
+import { createContext as createContext4, useContext as useContext4 } from "react";
+var ContentPartContext = createContext4(
+  null
+);
+function useContentPartContext(options) {
+  const context = useContext4(ContentPartContext);
+  if (!options?.optional && !context)
+    throw new Error(
+      "This component can only be used inside a component passed to <MessagePrimitive.Content components={...} >."
+    );
+  return context;
+}
+
+// src/context/providers/TextContentPartProvider.tsx
+import { jsx as jsx9 } from "react/jsx-runtime";
+var TextContentPartProvider = ({ children, text }) => {
+  const [context] = useState9(() => {
+    const useContentPart = create12(() => ({
+      status: { type: "complete" },
+      part: { type: "text", text }
+    }));
+    return {
+      useContentPart
+    };
+  });
+  return /* @__PURE__ */ jsx9(ContentPartContext.Provider, { value: context, children });
+};
+
 // src/context/react/ComposerContext.ts
 import { useMemo as useMemo3 } from "react";
 
 // src/context/react/MessageContext.ts
-import { createContext as createContext4, useContext as useContext4 } from "react";
-var MessageContext = createContext4(null);
+import { createContext as createContext5, useContext as useContext5 } from "react";
+var MessageContext = createContext5(null);
 function useMessageContext(options) {
-  const context = useContext4(MessageContext);
+  const context = useContext5(MessageContext);
   if (!options?.optional && !context)
     throw new Error(
       "This component can only be used inside a component passed to <ThreadPrimitive.Messages components={...} />."
@@ -2299,20 +2329,6 @@ var useComposerContext = () => {
   );
 };
 
-// src/context/react/ContentPartContext.ts
-import { createContext as createContext5, useContext as useContext5 } from "react";
-var ContentPartContext = createContext5(
-  null
-);
-function useContentPartContext(options) {
-  const context = useContext5(ContentPartContext);
-  if (!options?.optional && !context)
-    throw new Error(
-      "This component can only be used inside a component passed to <MessagePrimitive.Content components={...} >."
-    );
-  return context;
-}
-
 // src/hooks/useAppendMessage.tsx
 import { useCallback } from "react";
 var toAppendMessage = (useThreadMessages, message) => {
@@ -2487,11 +2503,8 @@ var useActionBarCopy = ({
 // src/primitive-hooks/actionBar/useActionBarEdit.tsx
 import { useCallback as useCallback4 } from "react";
 var useActionBarEdit = () => {
-  const { useMessage, useEditComposer } = useMessageContext();
-  const disabled = useCombinedStore(
-    [useMessage, useEditComposer],
-    (m, c) => m.message.role !== "user" || c.isEditing
-  );
+  const { useEditComposer } = useMessageContext();
+  const disabled = useEditComposer((c) => c.isEditing);
   const callback = useCallback4(() => {
     const { edit } = useEditComposer.getState();
     edit();
@@ -2829,7 +2842,7 @@ var useActionBarFloatStatus = ({
 };
 
 // src/primitives/actionBar/ActionBarRoot.tsx
-import { jsx as jsx9 } from "react/jsx-runtime";
+import { jsx as jsx10 } from "react/jsx-runtime";
 var ActionBarPrimitiveRoot = forwardRef5(({ hideWhenRunning, autohide, autohideFloat, ...rest }, ref) => {
   const hideAndfloatStatus = useActionBarFloatStatus({
     hideWhenRunning,
@@ -2837,7 +2850,7 @@ var ActionBarPrimitiveRoot = forwardRef5(({ hideWhenRunning, autohide, autohideF
     autohideFloat
   });
   if (hideAndfloatStatus === "hidden" /* Hidden */) return null;
-  return /* @__PURE__ */ jsx9(
+  return /* @__PURE__ */ jsx10(
     Primitive2.div,
     {
       ...hideAndfloatStatus === "floating" /* Floating */ ? { "data-floating": "true" } : null,
@@ -2852,7 +2865,7 @@ ActionBarPrimitiveRoot.displayName = "ActionBarPrimitive.Root";
 import { forwardRef as forwardRef6 } from "react";
 import { Primitive as Primitive3 } from "@radix-ui/react-primitive";
 import { composeEventHandlers } from "@radix-ui/primitive";
-import { jsx as jsx10 } from "react/jsx-runtime";
+import { jsx as jsx11 } from "react/jsx-runtime";
 var createActionButton = (displayName, useActionButton, forwardProps = []) => {
   const ActionButton = forwardRef6((props, forwardedRef) => {
     const forwardedProps = {};
@@ -2865,7 +2878,7 @@ var createActionButton = (displayName, useActionButton, forwardProps = []) => {
       }
     });
     const callback = useActionButton(forwardedProps);
-    return /* @__PURE__ */ jsx10(
+    return /* @__PURE__ */ jsx11(
       Primitive3.button,
       {
         type: "button",
@@ -2912,7 +2925,7 @@ import { forwardRef as forwardRef7 } from "react";
 import { useEscapeKeydown } from "@radix-ui/react-use-escape-keydown";
 import { Primitive as Primitive4 } from "@radix-ui/react-primitive";
 import { composeEventHandlers as composeEventHandlers2 } from "@radix-ui/primitive";
-import { jsx as jsx11 } from "react/jsx-runtime";
+import { jsx as jsx12 } from "react/jsx-runtime";
 var ActionBarPrimitiveStopSpeaking = forwardRef7((props, ref) => {
   const callback = useActionBarStopSpeaking();
   useEscapeKeydown((e) => {
@@ -2921,7 +2934,7 @@ var ActionBarPrimitiveStopSpeaking = forwardRef7((props, ref) => {
       callback();
     }
   });
-  return /* @__PURE__ */ jsx11(
+  return /* @__PURE__ */ jsx12(
     Primitive4.button,
     {
       type: "button",
@@ -2945,7 +2958,7 @@ __export(assistantModal_exports, {
 });
 
 // src/primitives/assistantModal/AssistantModalRoot.tsx
-import { useState as useState9 } from "react";
+import { useState as useState10 } from "react";
 import * as PopoverPrimitive2 from "@radix-ui/react-popover";
 import { composeEventHandlers as composeEventHandlers3 } from "@radix-ui/primitive";
 
@@ -2967,9 +2980,9 @@ import * as PopoverPrimitive from "@radix-ui/react-popover";
 var usePopoverScope = PopoverPrimitive.createPopoverScope();
 
 // src/primitives/assistantModal/AssistantModalRoot.tsx
-import { jsx as jsx12 } from "react/jsx-runtime";
+import { jsx as jsx13 } from "react/jsx-runtime";
 var useAssistantModalOpenState = (defaultOpen = false) => {
-  const state = useState9(defaultOpen);
+  const state = useState10(defaultOpen);
   const [, setOpen] = state;
   useOnComposerFocus(() => {
     setOpen(true);
@@ -2985,7 +2998,7 @@ var AssistantModalPrimitiveRoot = ({
 }) => {
   const scope = usePopoverScope(__scopeAssistantModal);
   const [modalOpen, setOpen] = useAssistantModalOpenState(defaultOpen);
-  return /* @__PURE__ */ jsx12(
+  return /* @__PURE__ */ jsx13(
     PopoverPrimitive2.Root,
     {
       ...scope,
@@ -3000,14 +3013,14 @@ AssistantModalPrimitiveRoot.displayName = "AssistantModalPrimitive.Root";
 // src/primitives/assistantModal/AssistantModalTrigger.tsx
 import { forwardRef as forwardRef8 } from "react";
 import * as PopoverPrimitive3 from "@radix-ui/react-popover";
-import { jsx as jsx13 } from "react/jsx-runtime";
+import { jsx as jsx14 } from "react/jsx-runtime";
 var AssistantModalPrimitiveTrigger = forwardRef8(
   ({
     __scopeAssistantModal,
     ...rest
   }, ref) => {
     const scope = usePopoverScope(__scopeAssistantModal);
-    return /* @__PURE__ */ jsx13(PopoverPrimitive3.Trigger, { ...scope, ...rest, ref });
+    return /* @__PURE__ */ jsx14(PopoverPrimitive3.Trigger, { ...scope, ...rest, ref });
   }
 );
 AssistantModalPrimitiveTrigger.displayName = "AssistantModalPrimitive.Trigger";
@@ -3016,7 +3029,7 @@ AssistantModalPrimitiveTrigger.displayName = "AssistantModalPrimitive.Trigger";
 import { forwardRef as forwardRef9 } from "react";
 import * as PopoverPrimitive4 from "@radix-ui/react-popover";
 import { composeEventHandlers as composeEventHandlers4 } from "@radix-ui/primitive";
-import { jsx as jsx14 } from "react/jsx-runtime";
+import { jsx as jsx15 } from "react/jsx-runtime";
 var AssistantModalPrimitiveContent = forwardRef9(
   ({
     __scopeAssistantModal,
@@ -3027,7 +3040,7 @@ var AssistantModalPrimitiveContent = forwardRef9(
     ...props
   }, forwardedRef) => {
     const scope = usePopoverScope(__scopeAssistantModal);
-    return /* @__PURE__ */ jsx14(PopoverPrimitive4.Portal, { ...scope, children: /* @__PURE__ */ jsx14(
+    return /* @__PURE__ */ jsx15(PopoverPrimitive4.Portal, { ...scope, children: /* @__PURE__ */ jsx15(
       PopoverPrimitive4.Content,
       {
         ...scope,
@@ -3048,14 +3061,14 @@ AssistantModalPrimitiveContent.displayName = "AssistantModalPrimitive.Content";
 // src/primitives/assistantModal/AssistantModalAnchor.tsx
 import { forwardRef as forwardRef10 } from "react";
 import * as PopoverPrimitive5 from "@radix-ui/react-popover";
-import { jsx as jsx15 } from "react/jsx-runtime";
+import { jsx as jsx16 } from "react/jsx-runtime";
 var AssistantModalPrimitiveAnchor = forwardRef10(
   ({
     __scopeAssistantModal,
     ...rest
   }, ref) => {
     const scope = usePopoverScope(__scopeAssistantModal);
-    return /* @__PURE__ */ jsx15(PopoverPrimitive5.Anchor, { ...scope, ...rest, ref });
+    return /* @__PURE__ */ jsx16(PopoverPrimitive5.Anchor, { ...scope, ...rest, ref });
   }
 );
 AssistantModalPrimitiveAnchor.displayName = "AssistantModalPrimitive.Anchor";
@@ -3083,18 +3096,18 @@ var BranchPickerPrevious = createActionButton(
 );
 
 // src/primitives/branchPicker/BranchPickerCount.tsx
-import { Fragment, jsx as jsx16 } from "react/jsx-runtime";
+import { Fragment, jsx as jsx17 } from "react/jsx-runtime";
 var BranchPickerPrimitiveCount = () => {
   const branchCount = useBranchPickerCount();
-  return /* @__PURE__ */ jsx16(Fragment, { children: branchCount });
+  return /* @__PURE__ */ jsx17(Fragment, { children: branchCount });
 };
 BranchPickerPrimitiveCount.displayName = "BranchPickerPrimitive.Count";
 
 // src/primitives/branchPicker/BranchPickerNumber.tsx
-import { Fragment as Fragment2, jsx as jsx17 } from "react/jsx-runtime";
+import { Fragment as Fragment2, jsx as jsx18 } from "react/jsx-runtime";
 var BranchPickerPrimitiveNumber = () => {
   const branchNumber = useBranchPickerNumber();
-  return /* @__PURE__ */ jsx17(Fragment2, { children: branchNumber });
+  return /* @__PURE__ */ jsx18(Fragment2, { children: branchNumber });
 };
 BranchPickerPrimitiveNumber.displayName = "BranchPickerPrimitive.Number";
 
@@ -3139,7 +3152,7 @@ var useManagedRef = (callback) => {
 
 // src/primitives/message/MessageRoot.tsx
 import { useComposedRefs } from "@radix-ui/react-compose-refs";
-import { jsx as jsx18 } from "react/jsx-runtime";
+import { jsx as jsx19 } from "react/jsx-runtime";
 var useIsHoveringRef = () => {
   const { useMessageUtils } = useMessageContext();
   const callbackRef = useCallback16(
@@ -3166,7 +3179,7 @@ var useIsHoveringRef = () => {
 var MessagePrimitiveRoot = forwardRef11(({ onMouseEnter, onMouseLeave, ...rest }, forwardRef30) => {
   const isHoveringRef = useIsHoveringRef();
   const ref = useComposedRefs(forwardRef30, isHoveringRef);
-  return /* @__PURE__ */ jsx18(Primitive5.div, { ...rest, ref });
+  return /* @__PURE__ */ jsx19(Primitive5.div, { ...rest, ref });
 });
 MessagePrimitiveRoot.displayName = "MessagePrimitive.Root";
 
@@ -3184,9 +3197,9 @@ MessagePrimitiveIf.displayName = "MessagePrimitive.If";
 import { memo as memo2 } from "react";
 
 // src/context/providers/ContentPartProvider.tsx
-import { useEffect as useEffect9, useState as useState10 } from "react";
-import { create as create12 } from "zustand";
-import { jsx as jsx19 } from "react/jsx-runtime";
+import { useEffect as useEffect9, useState as useState11 } from "react";
+import { create as create13 } from "zustand";
+import { jsx as jsx20 } from "react/jsx-runtime";
 var COMPLETE_STATUS = {
   type: "complete"
 };
@@ -3214,6 +3227,8 @@ var getContentPartState = ({ message }, useContentPart, partIndex) => {
     } else {
       return null;
     }
+  } else if (message.content.length === 1 && part.type === "text" && part.text.length === 0) {
+    part = EMPTY_CONTENT;
   }
   const status = toContentPartStatus(message, partIndex, part);
   const currentState = useContentPart?.getState();
@@ -3223,8 +3238,8 @@ var getContentPartState = ({ message }, useContentPart, partIndex) => {
 };
 var useContentPartContext2 = (partIndex) => {
   const { useMessage } = useMessageContext();
-  const [context] = useState10(() => {
-    const useContentPart = create12(
+  const [context] = useState11(() => {
+    const useContentPart = create13(
       () => getContentPartState(useMessage.getState(), void 0, partIndex)
     );
     return { useContentPart };
@@ -3249,32 +3264,32 @@ var ContentPartProvider = ({
   children
 }) => {
   const context = useContentPartContext2(partIndex);
-  return /* @__PURE__ */ jsx19(ContentPartContext.Provider, { value: context, children });
+  return /* @__PURE__ */ jsx20(ContentPartContext.Provider, { value: context, children });
 };
 
 // src/primitives/contentPart/ContentPartText.tsx
 import {
   forwardRef as forwardRef12
 } from "react";
-import { jsx as jsx20 } from "react/jsx-runtime";
+import { jsx as jsx21 } from "react/jsx-runtime";
 var ContentPartPrimitiveText = forwardRef12(({ smooth = true, component: Component = "span", ...rest }, forwardedRef) => {
   const {
     part: { text },
     status
   } = useSmooth(useContentPartText(), smooth);
-  return /* @__PURE__ */ jsx20(Component, { "data-status": status.type, ...rest, ref: forwardedRef, children: text });
+  return /* @__PURE__ */ jsx21(Component, { "data-status": status.type, ...rest, ref: forwardedRef, children: text });
 });
 ContentPartPrimitiveText.displayName = "ContentPartPrimitive.Text";
 
 // src/primitives/contentPart/ContentPartImage.tsx
 import { Primitive as Primitive6 } from "@radix-ui/react-primitive";
 import { forwardRef as forwardRef13 } from "react";
-import { jsx as jsx21 } from "react/jsx-runtime";
+import { jsx as jsx22 } from "react/jsx-runtime";
 var ContentPartPrimitiveImage = forwardRef13((props, forwardedRef) => {
   const {
     part: { image }
   } = useContentPartImage();
-  return /* @__PURE__ */ jsx21(Primitive6.img, { src: image, ...props, ref: forwardedRef });
+  return /* @__PURE__ */ jsx22(Primitive6.img, { src: image, ...props, ref: forwardedRef });
 });
 ContentPartPrimitiveImage.displayName = "ContentPartPrimitive.Image";
 
@@ -3296,7 +3311,7 @@ var ContentPartPrimitiveInProgress = ({ children }) => {
 ContentPartPrimitiveInProgress.displayName = "ContentPartPrimitive.InProgress";
 
 // src/primitives/message/MessageContent.tsx
-import { jsx as jsx22, jsxs as jsxs3 } from "react/jsx-runtime";
+import { jsx as jsx23, jsxs as jsxs3 } from "react/jsx-runtime";
 var ToolUIDisplay = ({
   UI,
   ...props
@@ -3304,20 +3319,20 @@ var ToolUIDisplay = ({
   const { useToolUIs } = useAssistantContext();
   const Render = useToolUIs((s) => s.getToolUI(props.part.toolName)) ?? UI;
   if (!Render) return null;
-  return /* @__PURE__ */ jsx22(Render, { ...props });
+  return /* @__PURE__ */ jsx23(Render, { ...props });
 };
 var defaultComponents = {
   Text: () => /* @__PURE__ */ jsxs3("p", { style: { whiteSpace: "pre-line" }, children: [
-    /* @__PURE__ */ jsx22(ContentPartPrimitiveText, {}),
-    /* @__PURE__ */ jsx22(ContentPartPrimitiveInProgress, { children: /* @__PURE__ */ jsx22("span", { style: { fontFamily: "revert" }, children: " \u25CF" }) })
+    /* @__PURE__ */ jsx23(ContentPartPrimitiveText, {}),
+    /* @__PURE__ */ jsx23(ContentPartPrimitiveInProgress, { children: /* @__PURE__ */ jsx23("span", { style: { fontFamily: "revert" }, children: " \u25CF" }) })
   ] }),
-  Image: () => /* @__PURE__ */ jsx22(ContentPartPrimitiveImage, {}),
-  UI: () => /* @__PURE__ */ jsx22(ContentPartPrimitiveDisplay, {})
+  Image: () => /* @__PURE__ */ jsx23(ContentPartPrimitiveImage, {}),
+  UI: () => /* @__PURE__ */ jsx23(ContentPartPrimitiveDisplay, {})
 };
 var MessageContentPartComponent = ({
   components: {
-    Empty = defaultComponents.Text,
     Text: Text2 = defaultComponents.Text,
+    Empty,
     Image: Image2 = defaultComponents.Image,
     UI = defaultComponents.UI,
     tools: { by_name = {}, Fallback: Fallback2 = void 0 } = {}
@@ -3333,16 +3348,18 @@ var MessageContentPartComponent = ({
     case "text":
       if (status.type === "requires-action")
         throw new Error("Encountered unexpected requires-action status");
-      if (part === EMPTY_CONTENT) return /* @__PURE__ */ jsx22(Empty, { part, status });
-      return /* @__PURE__ */ jsx22(Text2, { part, status });
+      if (part === EMPTY_CONTENT && !!Empty) {
+        return /* @__PURE__ */ jsx23(Empty, { status });
+      }
+      return /* @__PURE__ */ jsx23(Text2, { part, status });
     case "image":
       if (status.type === "requires-action")
         throw new Error("Encountered unexpected requires-action status");
-      return /* @__PURE__ */ jsx22(Image2, { part, status });
+      return /* @__PURE__ */ jsx23(Image2, { part, status });
     case "ui":
       if (status.type === "requires-action")
         throw new Error("Encountered unexpected requires-action status");
-      return /* @__PURE__ */ jsx22(UI, { part, status });
+      return /* @__PURE__ */ jsx23(UI, { part, status });
     case "tool-call": {
       const Tool = by_name[part.toolName] || Fallback2;
       const addResult = (result) => addToolResult({
@@ -3351,7 +3368,7 @@ var MessageContentPartComponent = ({
         toolCallId: part.toolCallId,
         result
       });
-      return /* @__PURE__ */ jsx22(
+      return /* @__PURE__ */ jsx23(
         ToolUIDisplay,
         {
           UI: Tool,
@@ -3370,7 +3387,7 @@ var MessageContentPartImpl = ({
   partIndex,
   components
 }) => {
-  return /* @__PURE__ */ jsx22(ContentPartProvider, { partIndex, children: /* @__PURE__ */ jsx22(MessageContentPartComponent, { components }) });
+  return /* @__PURE__ */ jsx23(ContentPartProvider, { partIndex, children: /* @__PURE__ */ jsx23(MessageContentPartComponent, { components }) });
 };
 var MessageContentPart = memo2(
   MessageContentPartImpl,
@@ -3381,7 +3398,7 @@ var MessagePrimitiveContent = ({
 }) => {
   const { useMessage } = useMessageContext();
   const contentLength = useMessage((s) => s.message.content.length) || 1;
-  return Array.from({ length: contentLength }, (_, index) => /* @__PURE__ */ jsx22(MessageContentPart, { partIndex: index, components }, index));
+  return Array.from({ length: contentLength }, (_, index) => /* @__PURE__ */ jsx23(MessageContentPart, { partIndex: index, components }, index));
 };
 MessagePrimitiveContent.displayName = "MessagePrimitive.Content";
 
@@ -3417,9 +3434,9 @@ function useAttachmentContext(options) {
 }
 
 // src/context/providers/MessageAttachmentProvider.tsx
-import { useEffect as useEffect10, useState as useState11 } from "react";
-import { create as create13 } from "zustand";
-import { jsx as jsx23 } from "react/jsx-runtime";
+import { useEffect as useEffect10, useState as useState12 } from "react";
+import { create as create14 } from "zustand";
+import { jsx as jsx24 } from "react/jsx-runtime";
 var getAttachment = ({ message }, useAttachment, partIndex) => {
   if (message.role !== "user") return null;
   const attachments = message.attachments;
@@ -3431,9 +3448,9 @@ var getAttachment = ({ message }, useAttachment, partIndex) => {
 };
 var useMessageAttachmentContext = (partIndex) => {
   const { useMessage } = useMessageContext();
-  const [context] = useState11(
+  const [context] = useState12(
     () => {
-      const useAttachment = create13(
+      const useAttachment = create14(
         () => getAttachment(useMessage.getState(), void 0, partIndex)
       );
       return { type: "message", useAttachment };
@@ -3459,11 +3476,11 @@ var MessageAttachmentProvider = ({
   children
 }) => {
   const context = useMessageAttachmentContext(partIndex);
-  return /* @__PURE__ */ jsx23(AttachmentContext.Provider, { value: context, children });
+  return /* @__PURE__ */ jsx24(AttachmentContext.Provider, { value: context, children });
 };
 
 // src/primitives/message/MessageAttachments.tsx
-import { jsx as jsx24 } from "react/jsx-runtime";
+import { jsx as jsx25 } from "react/jsx-runtime";
 var getComponent = (components, attachment) => {
   const type = attachment.type;
   switch (type) {
@@ -3484,10 +3501,10 @@ var AttachmentComponent = ({ components }) => {
     (a) => getComponent(components, a.attachment)
   );
   if (!Component) return null;
-  return /* @__PURE__ */ jsx24(Component, {});
+  return /* @__PURE__ */ jsx25(Component, {});
 };
 var MessageAttachmentImpl = ({ components, attachmentIndex }) => {
-  return /* @__PURE__ */ jsx24(MessageAttachmentProvider, { attachmentIndex, children: /* @__PURE__ */ jsx24(AttachmentComponent, { components }) });
+  return /* @__PURE__ */ jsx25(MessageAttachmentProvider, { attachmentIndex, children: /* @__PURE__ */ jsx25(AttachmentComponent, { components }) });
 };
 var MessageAttachment = memo3(
   MessageAttachmentImpl,
@@ -3499,7 +3516,7 @@ var MessagePrimitiveAttachments = ({ components }) => {
     if (message.role !== "user") return 0;
     return message.attachments.length;
   });
-  return Array.from({ length: attachmentsCount }, (_, index) => /* @__PURE__ */ jsx24(
+  return Array.from({ length: attachmentsCount }, (_, index) => /* @__PURE__ */ jsx25(
     MessageAttachment,
     {
       attachmentIndex: index,
@@ -3511,9 +3528,9 @@ var MessagePrimitiveAttachments = ({ components }) => {
 MessagePrimitiveAttachments.displayName = "MessagePrimitive.Attachments";
 
 // src/primitives/branchPicker/BranchPickerRoot.tsx
-import { jsx as jsx25 } from "react/jsx-runtime";
+import { jsx as jsx26 } from "react/jsx-runtime";
 var BranchPickerPrimitiveRoot = forwardRef14(({ hideWhenSingleBranch, ...rest }, ref) => {
-  return /* @__PURE__ */ jsx25(MessagePrimitiveIf, { hasBranches: hideWhenSingleBranch ? true : void 0, children: /* @__PURE__ */ jsx25(Primitive7.div, { ...rest, ref }) });
+  return /* @__PURE__ */ jsx26(MessagePrimitiveIf, { hasBranches: hideWhenSingleBranch ? true : void 0, children: /* @__PURE__ */ jsx26(Primitive7.div, { ...rest, ref }) });
 });
 BranchPickerPrimitiveRoot.displayName = "BranchPickerPrimitive.Root";
 
@@ -3535,7 +3552,7 @@ import { Primitive as Primitive8 } from "@radix-ui/react-primitive";
 import {
   forwardRef as forwardRef15
 } from "react";
-import { jsx as jsx26 } from "react/jsx-runtime";
+import { jsx as jsx27 } from "react/jsx-runtime";
 var ComposerPrimitiveRoot = forwardRef15(({ onSubmit, ...rest }, forwardedRef) => {
   const send = useComposerSend();
   const handleSubmit = (e) => {
@@ -3543,7 +3560,7 @@ var ComposerPrimitiveRoot = forwardRef15(({ onSubmit, ...rest }, forwardedRef) =
     if (!send) return;
     send();
   };
-  return /* @__PURE__ */ jsx26(
+  return /* @__PURE__ */ jsx27(
     Primitive8.form,
     {
       ...rest,
@@ -3566,7 +3583,7 @@ import {
 } from "react";
 import TextareaAutosize from "react-textarea-autosize";
 import { useEscapeKeydown as useEscapeKeydown2 } from "@radix-ui/react-use-escape-keydown";
-import { jsx as jsx27 } from "react/jsx-runtime";
+import { jsx as jsx28 } from "react/jsx-runtime";
 var ComposerPrimitiveInput = forwardRef16(
   ({
     autoFocus = false,
@@ -3620,7 +3637,7 @@ var ComposerPrimitiveInput = forwardRef16(
         focus();
       }
     });
-    return /* @__PURE__ */ jsx27(
+    return /* @__PURE__ */ jsx28(
       Component,
       {
         name: "input",
@@ -3662,9 +3679,9 @@ var ComposerPrimitiveAddAttachment = createActionButton(
 import { memo as memo4 } from "react";
 
 // src/context/providers/ComposerAttachmentProvider.tsx
-import { useEffect as useEffect12, useState as useState12 } from "react";
-import { create as create14 } from "zustand";
-import { jsx as jsx28 } from "react/jsx-runtime";
+import { useEffect as useEffect12, useState as useState13 } from "react";
+import { create as create15 } from "zustand";
+import { jsx as jsx29 } from "react/jsx-runtime";
 var getAttachment2 = ({ attachments }, useAttachment, partIndex) => {
   let attachment = attachments[partIndex];
   if (!attachment) return null;
@@ -3674,9 +3691,9 @@ var getAttachment2 = ({ attachments }, useAttachment, partIndex) => {
 };
 var useComposerAttachmentContext = (partIndex) => {
   const { useComposer } = useThreadContext();
-  const [context] = useState12(
+  const [context] = useState13(
     () => {
-      const useAttachment = create14(
+      const useAttachment = create15(
         () => getAttachment2(useComposer.getState(), void 0, partIndex)
       );
       return { type: "composer", useAttachment };
@@ -3699,11 +3716,11 @@ var useComposerAttachmentContext = (partIndex) => {
 };
 var ComposerAttachmentProvider = ({ attachmentIndex: partIndex, children }) => {
   const context = useComposerAttachmentContext(partIndex);
-  return /* @__PURE__ */ jsx28(AttachmentContext.Provider, { value: context, children });
+  return /* @__PURE__ */ jsx29(AttachmentContext.Provider, { value: context, children });
 };
 
 // src/primitives/composer/ComposerAttachments.tsx
-import { jsx as jsx29 } from "react/jsx-runtime";
+import { jsx as jsx30 } from "react/jsx-runtime";
 var getComponent2 = (components, attachment) => {
   const type = attachment.type;
   switch (type) {
@@ -3724,10 +3741,10 @@ var AttachmentComponent2 = ({ components }) => {
     (a) => getComponent2(components, a.attachment)
   );
   if (!Component) return null;
-  return /* @__PURE__ */ jsx29(Component, {});
+  return /* @__PURE__ */ jsx30(Component, {});
 };
 var ComposerAttachmentImpl = ({ components, attachmentIndex }) => {
-  return /* @__PURE__ */ jsx29(ComposerAttachmentProvider, { attachmentIndex, children: /* @__PURE__ */ jsx29(AttachmentComponent2, { components }) });
+  return /* @__PURE__ */ jsx30(ComposerAttachmentProvider, { attachmentIndex, children: /* @__PURE__ */ jsx30(AttachmentComponent2, { components }) });
 };
 var ComposerAttachment = memo4(
   ComposerAttachmentImpl,
@@ -3736,7 +3753,7 @@ var ComposerAttachment = memo4(
 var ComposerPrimitiveAttachments = ({ components }) => {
   const { useComposer } = useThreadContext();
   const attachmentsCount = useComposer((s) => s.attachments.length);
-  return Array.from({ length: attachmentsCount }, (_, index) => /* @__PURE__ */ jsx29(
+  return Array.from({ length: attachmentsCount }, (_, index) => /* @__PURE__ */ jsx30(
     ComposerAttachment,
     {
       attachmentIndex: index,
@@ -3781,9 +3798,9 @@ __export(thread_exports, {
 // src/primitives/thread/ThreadRoot.tsx
 import { Primitive as Primitive9 } from "@radix-ui/react-primitive";
 import { forwardRef as forwardRef17 } from "react";
-import { jsx as jsx30 } from "react/jsx-runtime";
+import { jsx as jsx31 } from "react/jsx-runtime";
 var ThreadPrimitiveRoot = forwardRef17((props, ref) => {
-  return /* @__PURE__ */ jsx30(Primitive9.div, { ...props, ref });
+  return /* @__PURE__ */ jsx31(Primitive9.div, { ...props, ref });
 });
 ThreadPrimitiveRoot.displayName = "ThreadPrimitive.Root";
 
@@ -3918,13 +3935,13 @@ var useThreadViewportAutoScroll = ({
 };
 
 // src/primitives/thread/ThreadViewport.tsx
-import { jsx as jsx31 } from "react/jsx-runtime";
+import { jsx as jsx32 } from "react/jsx-runtime";
 var ThreadPrimitiveViewport = forwardRef18(({ autoScroll, onScroll, children, ...rest }, forwardedRef) => {
   const autoScrollRef = useThreadViewportAutoScroll({
     autoScroll
   });
   const ref = useComposedRefs4(forwardedRef, autoScrollRef);
-  return /* @__PURE__ */ jsx31(Primitive10.div, { ...rest, ref, children });
+  return /* @__PURE__ */ jsx32(Primitive10.div, { ...rest, ref, children });
 });
 ThreadPrimitiveViewport.displayName = "ThreadPrimitive.Viewport";
 
@@ -3932,15 +3949,15 @@ ThreadPrimitiveViewport.displayName = "ThreadPrimitive.Viewport";
 import { memo as memo5 } from "react";
 
 // src/context/providers/MessageProvider.tsx
-import { useEffect as useEffect14, useState as useState13 } from "react";
-import { create as create17 } from "zustand";
+import { useEffect as useEffect14, useState as useState14 } from "react";
+import { create as create18 } from "zustand";
 
 // src/context/stores/EditComposer.ts
-import { create as create15 } from "zustand";
+import { create as create16 } from "zustand";
 var makeEditComposerStore = ({
   onEdit,
   onSend
-}) => create15()((set, get) => ({
+}) => create16()((set, get) => ({
   get value() {
     return get().text;
   },
@@ -3974,8 +3991,8 @@ var makeEditComposerStore = ({
 }));
 
 // src/context/stores/MessageUtils.ts
-import { create as create16 } from "zustand";
-var makeMessageUtilsStore = () => create16((set) => {
+import { create as create17 } from "zustand";
+var makeMessageUtilsStore = () => create17((set) => {
   let utterance = null;
   return {
     isCopied: false,
@@ -4001,7 +4018,7 @@ var makeMessageUtilsStore = () => create16((set) => {
 });
 
 // src/context/providers/MessageProvider.tsx
-import { jsx as jsx32 } from "react/jsx-runtime";
+import { jsx as jsx33 } from "react/jsx-runtime";
 var getIsLast = (messages, message) => {
   return messages[messages.length - 1]?.id === message.id;
 };
@@ -4023,8 +4040,8 @@ var getMessageState = (messages, getBranches, useMessage, messageIndex) => {
 };
 var useMessageContext2 = (messageIndex) => {
   const { useThreadMessages, useThreadActions } = useThreadContext();
-  const [context] = useState13(() => {
-    const useMessage = create17(
+  const [context] = useState14(() => {
+    const useMessage = create18(
       () => getMessageState(
         useThreadMessages.getState(),
         useThreadActions.getState().getBranches,
@@ -4036,19 +4053,11 @@ var useMessageContext2 = (messageIndex) => {
     const useEditComposer = makeEditComposerStore({
       onEdit: () => {
         const message = useMessage.getState().message;
-        if (message.role !== "user")
-          throw new Error(
-            "Tried to edit a non-user message. Editing is only supported for user messages. This is likely an internal bug in assistant-ui."
-          );
         const text = getThreadMessageText(message);
         return text;
       },
       onSend: (text) => {
         const { message, parentId } = useMessage.getState();
-        if (message.role !== "user")
-          throw new Error(
-            "Tried to edit a non-user message. Editing is only supported for user messages. This is likely an internal bug in assistant-ui."
-          );
         const previousText = getThreadMessageText(message);
         if (previousText === text) return;
         const nonTextParts = message.content.filter(
@@ -4056,7 +4065,7 @@ var useMessageContext2 = (messageIndex) => {
         );
         useThreadActions.getState().append({
           parentId,
-          role: "user",
+          role: message.role,
           content: [{ type: "text", text }, ...nonTextParts],
           attachments: message.attachments
         });
@@ -4085,37 +4094,58 @@ var MessageProvider = ({
   children
 }) => {
   const context = useMessageContext2(messageIndex);
-  return /* @__PURE__ */ jsx32(MessageContext.Provider, { value: context, children });
+  return /* @__PURE__ */ jsx33(MessageContext.Provider, { value: context, children });
 };
 
 // src/primitives/thread/ThreadMessages.tsx
-import { jsx as jsx33, jsxs as jsxs4 } from "react/jsx-runtime";
+import { jsx as jsx34 } from "react/jsx-runtime";
+var isComponentsSame = (prev, next) => {
+  return prev.Message === next.Message && prev.EditComposer === next.EditComposer && prev.UserEditComposer === next.UserEditComposer && prev.AssistantEditComposer === next.AssistantEditComposer && prev.SystemEditComposer === next.SystemEditComposer && prev.UserMessage === next.UserMessage && prev.AssistantMessage === next.AssistantMessage && prev.SystemMessage === next.SystemMessage;
+};
 var DEFAULT_SYSTEM_MESSAGE = () => null;
-var getComponents = (components) => {
-  return {
-    EditComposer: components.EditComposer ?? components.UserMessage ?? components.Message,
-    UserMessage: components.UserMessage ?? components.Message,
-    AssistantMessage: components.AssistantMessage ?? components.Message,
-    SystemMessage: components.SystemMessage ?? DEFAULT_SYSTEM_MESSAGE
-  };
+var getComponent3 = (components, role, isEditing) => {
+  switch (role) {
+    case "user":
+      if (isEditing) {
+        return components.UserEditComposer ?? components.EditComposer ?? components.UserMessage ?? components.Message;
+      } else {
+        return components.UserMessage ?? components.Message;
+      }
+    case "assistant":
+      if (isEditing) {
+        return components.AssistantEditComposer ?? components.EditComposer ?? components.AssistantMessage ?? components.Message;
+      } else {
+        return components.AssistantMessage ?? components.Message;
+      }
+    case "system":
+      if (isEditing) {
+        return components.SystemEditComposer ?? components.EditComposer ?? components.SystemMessage ?? components.Message;
+      } else {
+        return components.SystemMessage ?? DEFAULT_SYSTEM_MESSAGE;
+      }
+    default:
+      const _exhaustiveCheck = role;
+      throw new Error(`Unknown message role: ${_exhaustiveCheck}`);
+  }
+};
+var ThreadMessageComponent = ({
+  components
+}) => {
+  const { useMessage, useEditComposer } = useMessageContext();
+  const role = useMessage((m) => m.message.role);
+  const isEditing = useEditComposer((c) => c.isEditing);
+  const Component = getComponent3(components, role, isEditing);
+  return /* @__PURE__ */ jsx34(Component, {});
 };
 var ThreadMessageImpl = ({
   messageIndex,
   components
 }) => {
-  const { UserMessage: UserMessage2, EditComposer: EditComposer2, AssistantMessage: AssistantMessage2, SystemMessage: SystemMessage2 } = getComponents(components);
-  return /* @__PURE__ */ jsxs4(MessageProvider, { messageIndex, children: [
-    /* @__PURE__ */ jsxs4(MessagePrimitiveIf, { user: true, children: [
-      /* @__PURE__ */ jsx33(ComposerPrimitiveIf, { editing: false, children: /* @__PURE__ */ jsx33(UserMessage2, {}) }),
-      /* @__PURE__ */ jsx33(ComposerPrimitiveIf, { editing: true, children: /* @__PURE__ */ jsx33(EditComposer2, {}) })
-    ] }),
-    /* @__PURE__ */ jsx33(MessagePrimitiveIf, { assistant: true, children: /* @__PURE__ */ jsx33(AssistantMessage2, {}) }),
-    /* @__PURE__ */ jsx33(MessagePrimitiveIf, { system: true, children: /* @__PURE__ */ jsx33(SystemMessage2, {}) })
-  ] });
+  return /* @__PURE__ */ jsx34(MessageProvider, { messageIndex, children: /* @__PURE__ */ jsx34(ThreadMessageComponent, { components }) });
 };
 var ThreadMessage = memo5(
   ThreadMessageImpl,
-  (prev, next) => prev.messageIndex === next.messageIndex && prev.components.Message === next.components.Message && prev.components.UserMessage === next.components.UserMessage && prev.components.EditComposer === next.components.EditComposer && prev.components.AssistantMessage === next.components.AssistantMessage && prev.components.SystemMessage === next.components.SystemMessage
+  (prev, next) => prev.messageIndex === next.messageIndex && isComponentsSame(prev.components, next.components)
 );
 var ThreadPrimitiveMessagesImpl = ({
   components
@@ -4123,12 +4153,12 @@ var ThreadPrimitiveMessagesImpl = ({
   const { useThreadMessages } = useThreadContext();
   const messagesLength = useThreadMessages((t) => t.length);
   if (messagesLength === 0) return null;
-  return Array.from({ length: messagesLength }, (_, index) => /* @__PURE__ */ jsx33(ThreadMessage, { messageIndex: index, components }, index));
+  return Array.from({ length: messagesLength }, (_, index) => /* @__PURE__ */ jsx34(ThreadMessage, { messageIndex: index, components }, index));
 };
 ThreadPrimitiveMessagesImpl.displayName = "ThreadPrimitive.Messages";
 var ThreadPrimitiveMessages = memo5(
   ThreadPrimitiveMessagesImpl,
-  (prev, next) => prev.components?.Message === next.components?.Message && prev.components?.UserMessage === next.components?.UserMessage && prev.components?.EditComposer === next.components?.EditComposer && prev.components?.AssistantMessage === next.components?.AssistantMessage && prev.components?.SystemMessage === next.components?.SystemMessage
+  (prev, next) => isComponentsSame(prev.components, next.components)
 );
 
 // src/primitives/thread/ThreadScrollToBottom.tsx
@@ -4149,7 +4179,7 @@ import {
   createContext as createContext7,
   useContext as useContext7
 } from "react";
-import { Fragment as Fragment3, jsx as jsx34 } from "react/jsx-runtime";
+import { Fragment as Fragment3, jsx as jsx35 } from "react/jsx-runtime";
 var ThreadConfigContext = createContext7({});
 var useThreadConfig = () => {
   return useContext7(ThreadConfigContext);
@@ -4159,14 +4189,14 @@ var ThreadConfigProvider = ({
   config
 }) => {
   const assistant = useAssistantContext({ optional: true });
-  const configProvider = config && Object.keys(config ?? {}).length > 0 ? /* @__PURE__ */ jsx34(ThreadConfigContext.Provider, { value: config, children }) : /* @__PURE__ */ jsx34(Fragment3, { children });
+  const configProvider = config && Object.keys(config ?? {}).length > 0 ? /* @__PURE__ */ jsx35(ThreadConfigContext.Provider, { value: config, children }) : /* @__PURE__ */ jsx35(Fragment3, { children });
   if (!config?.runtime) return configProvider;
   if (assistant) {
     throw new Error(
       "You provided a runtime to <Thread> while simulataneously using <AssistantRuntimeProvider>. This is not allowed."
     );
   }
-  return /* @__PURE__ */ jsx34(AssistantRuntimeProvider, { runtime: config.runtime, children: configProvider });
+  return /* @__PURE__ */ jsx35(AssistantRuntimeProvider, { runtime: config.runtime, children: configProvider });
 };
 ThreadConfigProvider.displayName = "ThreadConfigProvider";
 
@@ -4179,7 +4209,7 @@ import {
   RefreshCwIcon,
   StopCircleIcon
 } from "lucide-react";
-import { Fragment as Fragment4, jsx as jsx35, jsxs as jsxs5 } from "react/jsx-runtime";
+import { Fragment as Fragment4, jsx as jsx36, jsxs as jsxs4 } from "react/jsx-runtime";
 var useAllowCopy = (ensureCapability = false) => {
   const { assistantMessage: { allowCopy = true } = {} } = useThreadConfig();
   const { useThread } = useThreadContext();
@@ -4203,16 +4233,16 @@ var AssistantActionBar = () => {
   const allowReload = useAllowReload(true);
   const allowSpeak = useAllowSpeak(true);
   if (!allowCopy && !allowReload && !allowSpeak) return null;
-  return /* @__PURE__ */ jsxs5(
+  return /* @__PURE__ */ jsxs4(
     AssistantActionBarRoot,
     {
       hideWhenRunning: true,
       autohide: "not-last",
       autohideFloat: "single-branch",
       children: [
-        allowSpeak && /* @__PURE__ */ jsx35(AssistantActionBarSpeechControl, {}),
-        allowCopy && /* @__PURE__ */ jsx35(AssistantActionBarCopy, {}),
-        allowReload && /* @__PURE__ */ jsx35(AssistantActionBarReload, {})
+        allowSpeak && /* @__PURE__ */ jsx36(AssistantActionBarSpeechControl, {}),
+        allowCopy && /* @__PURE__ */ jsx36(AssistantActionBarCopy, {}),
+        allowReload && /* @__PURE__ */ jsx36(AssistantActionBarReload, {})
       ]
     }
   );
@@ -4228,16 +4258,16 @@ var AssistantActionBarCopy = forwardRef19((props, ref) => {
       assistantMessage: { copy: { tooltip = "Copy" } = {} } = {}
     } = {}
   } = useThreadConfig();
-  return /* @__PURE__ */ jsx35(actionBar_exports.Copy, { asChild: true, children: /* @__PURE__ */ jsx35(TooltipIconButton, { tooltip, ...props, ref, children: props.children ?? /* @__PURE__ */ jsxs5(Fragment4, { children: [
-    /* @__PURE__ */ jsx35(message_exports.If, { copied: true, children: /* @__PURE__ */ jsx35(CheckIcon, {}) }),
-    /* @__PURE__ */ jsx35(message_exports.If, { copied: false, children: /* @__PURE__ */ jsx35(CopyIcon, {}) })
+  return /* @__PURE__ */ jsx36(actionBar_exports.Copy, { asChild: true, children: /* @__PURE__ */ jsx36(TooltipIconButton, { tooltip, ...props, ref, children: props.children ?? /* @__PURE__ */ jsxs4(Fragment4, { children: [
+    /* @__PURE__ */ jsx36(message_exports.If, { copied: true, children: /* @__PURE__ */ jsx36(CheckIcon, {}) }),
+    /* @__PURE__ */ jsx36(message_exports.If, { copied: false, children: /* @__PURE__ */ jsx36(CopyIcon, {}) })
   ] }) }) });
 });
 AssistantActionBarCopy.displayName = "AssistantActionBarCopy";
 var AssistantActionBarSpeechControl = () => {
-  return /* @__PURE__ */ jsxs5(Fragment4, { children: [
-    /* @__PURE__ */ jsx35(message_exports.If, { speaking: false, children: /* @__PURE__ */ jsx35(AssistantActionBarSpeak, {}) }),
-    /* @__PURE__ */ jsx35(message_exports.If, { speaking: true, children: /* @__PURE__ */ jsx35(AssistantActionBarStopSpeaking, {}) })
+  return /* @__PURE__ */ jsxs4(Fragment4, { children: [
+    /* @__PURE__ */ jsx36(message_exports.If, { speaking: false, children: /* @__PURE__ */ jsx36(AssistantActionBarSpeak, {}) }),
+    /* @__PURE__ */ jsx36(message_exports.If, { speaking: true, children: /* @__PURE__ */ jsx36(AssistantActionBarStopSpeaking, {}) })
   ] });
 };
 var AssistantActionBarSpeak = forwardRef19((props, ref) => {
@@ -4247,7 +4277,7 @@ var AssistantActionBarSpeak = forwardRef19((props, ref) => {
     } = {}
   } = useThreadConfig();
   const allowSpeak = useAllowSpeak();
-  return /* @__PURE__ */ jsx35(actionBar_exports.Speak, { disabled: !allowSpeak, asChild: true, children: /* @__PURE__ */ jsx35(TooltipIconButton, { tooltip, ...props, ref, children: props.children ?? /* @__PURE__ */ jsx35(AudioLinesIcon, {}) }) });
+  return /* @__PURE__ */ jsx36(actionBar_exports.Speak, { disabled: !allowSpeak, asChild: true, children: /* @__PURE__ */ jsx36(TooltipIconButton, { tooltip, ...props, ref, children: props.children ?? /* @__PURE__ */ jsx36(AudioLinesIcon, {}) }) });
 });
 AssistantActionBarSpeak.displayName = "AssistantActionBarSpeak";
 var AssistantActionBarStopSpeaking = forwardRef19((props, ref) => {
@@ -4259,7 +4289,7 @@ var AssistantActionBarStopSpeaking = forwardRef19((props, ref) => {
     } = {}
   } = useThreadConfig();
   const allowSpeak = useAllowSpeak();
-  return /* @__PURE__ */ jsx35(actionBar_exports.StopSpeaking, { disabled: !allowSpeak, asChild: true, children: /* @__PURE__ */ jsx35(TooltipIconButton, { tooltip: stopTooltip, ...props, ref, children: props.children ?? /* @__PURE__ */ jsx35(StopCircleIcon, {}) }) });
+  return /* @__PURE__ */ jsx36(actionBar_exports.StopSpeaking, { disabled: !allowSpeak, asChild: true, children: /* @__PURE__ */ jsx36(TooltipIconButton, { tooltip: stopTooltip, ...props, ref, children: props.children ?? /* @__PURE__ */ jsx36(StopCircleIcon, {}) }) });
 });
 AssistantActionBarStopSpeaking.displayName = "AssistantActionBarStopSpeaking";
 var AssistantActionBarReload = forwardRef19((props, ref) => {
@@ -4269,7 +4299,7 @@ var AssistantActionBarReload = forwardRef19((props, ref) => {
     } = {}
   } = useThreadConfig();
   const allowReload = useAllowReload();
-  return /* @__PURE__ */ jsx35(actionBar_exports.Reload, { disabled: !allowReload, asChild: true, children: /* @__PURE__ */ jsx35(TooltipIconButton, { tooltip, ...props, ref, children: /* @__PURE__ */ jsx35(RefreshCwIcon, {}) }) });
+  return /* @__PURE__ */ jsx36(actionBar_exports.Reload, { disabled: !allowReload, asChild: true, children: /* @__PURE__ */ jsx36(TooltipIconButton, { tooltip, ...props, ref, children: /* @__PURE__ */ jsx36(RefreshCwIcon, {}) }) });
 });
 AssistantActionBarReload.displayName = "AssistantActionBarReload";
 var exports = {
@@ -4291,7 +4321,7 @@ import { forwardRef as forwardRef21, useMemo as useMemo5 } from "react";
 // src/ui/branch-picker.tsx
 import { forwardRef as forwardRef20 } from "react";
 import { ChevronLeftIcon, ChevronRightIcon } from "lucide-react";
-import { jsx as jsx36, jsxs as jsxs6 } from "react/jsx-runtime";
+import { jsx as jsx37, jsxs as jsxs5 } from "react/jsx-runtime";
 var useAllowBranchPicker = (ensureCapability = false) => {
   const { branchPicker: { allowBranchPicker = true } = {} } = useThreadConfig();
   const { useThread } = useThreadContext();
@@ -4301,10 +4331,10 @@ var useAllowBranchPicker = (ensureCapability = false) => {
 var BranchPicker = () => {
   const allowBranchPicker = useAllowBranchPicker();
   if (!allowBranchPicker) return null;
-  return /* @__PURE__ */ jsxs6(BranchPickerRoot, { hideWhenSingleBranch: true, children: [
-    /* @__PURE__ */ jsx36(BranchPickerPrevious2, {}),
-    /* @__PURE__ */ jsx36(BranchPickerState, {}),
-    /* @__PURE__ */ jsx36(BranchPickerNext, {})
+  return /* @__PURE__ */ jsxs5(BranchPickerRoot, { hideWhenSingleBranch: true, children: [
+    /* @__PURE__ */ jsx37(BranchPickerPrevious2, {}),
+    /* @__PURE__ */ jsx37(BranchPickerState, {}),
+    /* @__PURE__ */ jsx37(BranchPickerNext, {})
   ] });
 };
 BranchPicker.displayName = "BranchPicker";
@@ -4319,17 +4349,17 @@ var BranchPickerPrevious2 = forwardRef20((props, ref) => {
     } = {}
   } = useThreadConfig();
   const allowBranchPicker = useAllowBranchPicker();
-  return /* @__PURE__ */ jsx36(branchPicker_exports.Previous, { disabled: !allowBranchPicker, asChild: true, children: /* @__PURE__ */ jsx36(TooltipIconButton, { tooltip, ...props, ref, children: props.children ?? /* @__PURE__ */ jsx36(ChevronLeftIcon, {}) }) });
+  return /* @__PURE__ */ jsx37(branchPicker_exports.Previous, { disabled: !allowBranchPicker, asChild: true, children: /* @__PURE__ */ jsx37(TooltipIconButton, { tooltip, ...props, ref, children: props.children ?? /* @__PURE__ */ jsx37(ChevronLeftIcon, {}) }) });
 });
 BranchPickerPrevious2.displayName = "BranchPickerPrevious";
 var BranchPickerStateWrapper = withDefaults("span", {
   className: "aui-branch-picker-state"
 });
 var BranchPickerState = forwardRef20((props, ref) => {
-  return /* @__PURE__ */ jsxs6(BranchPickerStateWrapper, { ...props, ref, children: [
-    /* @__PURE__ */ jsx36(branchPicker_exports.Number, {}),
+  return /* @__PURE__ */ jsxs5(BranchPickerStateWrapper, { ...props, ref, children: [
+    /* @__PURE__ */ jsx37(branchPicker_exports.Number, {}),
     " / ",
-    /* @__PURE__ */ jsx36(branchPicker_exports.Count, {})
+    /* @__PURE__ */ jsx37(branchPicker_exports.Count, {})
   ] });
 });
 BranchPickerState.displayName = "BranchPickerState";
@@ -4338,7 +4368,7 @@ var BranchPickerNext = forwardRef20((props, ref) => {
     strings: { branchPicker: { next: { tooltip = "Next" } = {} } = {} } = {}
   } = useThreadConfig();
   const allowBranchPicker = useAllowBranchPicker();
-  return /* @__PURE__ */ jsx36(branchPicker_exports.Next, { disabled: !allowBranchPicker, asChild: true, children: /* @__PURE__ */ jsx36(TooltipIconButton, { tooltip, ...props, ref, children: props.children ?? /* @__PURE__ */ jsx36(ChevronRightIcon, {}) }) });
+  return /* @__PURE__ */ jsx37(branchPicker_exports.Next, { disabled: !allowBranchPicker, asChild: true, children: /* @__PURE__ */ jsx37(TooltipIconButton, { tooltip, ...props, ref, children: props.children ?? /* @__PURE__ */ jsx37(ChevronRightIcon, {}) }) });
 });
 BranchPickerNext.displayName = "BranchPickerNext";
 var exports2 = {
@@ -4350,12 +4380,12 @@ var branch_picker_default = Object.assign(BranchPicker, exports2);
 
 // src/ui/base/avatar.tsx
 import * as AvatarPrimitive from "@radix-ui/react-avatar";
-import { jsx as jsx37, jsxs as jsxs7 } from "react/jsx-runtime";
+import { jsx as jsx38, jsxs as jsxs6 } from "react/jsx-runtime";
 var Avatar = ({ src, alt, fallback }) => {
   if (src == null && fallback == null) return null;
-  return /* @__PURE__ */ jsxs7(AvatarRoot, { children: [
-    src != null && /* @__PURE__ */ jsx37(AvatarImage, { src, alt }),
-    fallback != null && /* @__PURE__ */ jsx37(AvatarFallback, { children: fallback })
+  return /* @__PURE__ */ jsxs6(AvatarRoot, { children: [
+    src != null && /* @__PURE__ */ jsx38(AvatarImage, { src, alt }),
+    fallback != null && /* @__PURE__ */ jsx38(AvatarFallback, { children: fallback })
   ] });
 };
 Avatar.displayName = "Avatar";
@@ -4374,10 +4404,10 @@ AvatarFallback.displayName = "AvatarFallback";
 
 // src/ui/content-part.tsx
 import classNames2 from "classnames";
-import { jsx as jsx38 } from "react/jsx-runtime";
+import { jsx as jsx39 } from "react/jsx-runtime";
 var Text = () => {
   const status = useSmoothStatus();
-  return /* @__PURE__ */ jsx38(
+  return /* @__PURE__ */ jsx39(
     contentPart_exports.Text,
     {
       className: classNames2(
@@ -4392,19 +4422,19 @@ var exports3 = { Text: withSmoothContextProvider(Text) };
 var content_part_default = exports3;
 
 // src/ui/assistant-message.tsx
-import { jsx as jsx39, jsxs as jsxs8 } from "react/jsx-runtime";
+import { jsx as jsx40, jsxs as jsxs7 } from "react/jsx-runtime";
 var AssistantMessage = () => {
-  return /* @__PURE__ */ jsxs8(AssistantMessageRoot, { children: [
-    /* @__PURE__ */ jsx39(AssistantMessageAvatar, {}),
-    /* @__PURE__ */ jsx39(AssistantMessageContent, {}),
-    /* @__PURE__ */ jsx39(branch_picker_default, {}),
-    /* @__PURE__ */ jsx39(assistant_action_bar_default, {})
+  return /* @__PURE__ */ jsxs7(AssistantMessageRoot, { children: [
+    /* @__PURE__ */ jsx40(AssistantMessageAvatar, {}),
+    /* @__PURE__ */ jsx40(AssistantMessageContent, {}),
+    /* @__PURE__ */ jsx40(branch_picker_default, {}),
+    /* @__PURE__ */ jsx40(assistant_action_bar_default, {})
   ] });
 };
 AssistantMessage.displayName = "AssistantMessage";
 var AssistantMessageAvatar = () => {
   const { assistantAvatar: avatar = { fallback: "A" } } = useThreadConfig();
-  return /* @__PURE__ */ jsx39(Avatar, { ...avatar });
+  return /* @__PURE__ */ jsx40(Avatar, { ...avatar });
 };
 var AssistantMessageRoot = withDefaults(message_exports.Root, {
   className: "aui-assistant-message-root"
@@ -4428,7 +4458,7 @@ var AssistantMessageContent = forwardRef21(({ components: componentsProp, ...res
     // eslint-disable-next-line react-hooks/exhaustive-deps
     [...tools ?? [], components.ToolFallback]
   );
-  return /* @__PURE__ */ jsx39(AssistantMessageContentWrapper, { ...rest, ref, children: /* @__PURE__ */ jsx39(
+  return /* @__PURE__ */ jsx40(AssistantMessageContentWrapper, { ...rest, ref, children: /* @__PURE__ */ jsx40(
     message_exports.Content,
     {
       components: {
@@ -4463,9 +4493,9 @@ import { forwardRef as forwardRef23 } from "react";
 import { PaperclipIcon, SendHorizontalIcon } from "lucide-react";
 
 // src/ui/base/CircleStopIcon.tsx
-import { jsx as jsx40 } from "react/jsx-runtime";
+import { jsx as jsx41 } from "react/jsx-runtime";
 var CircleStopIcon = () => {
-  return /* @__PURE__ */ jsx40(
+  return /* @__PURE__ */ jsx41(
     "svg",
     {
       xmlns: "http://www.w3.org/2000/svg",
@@ -4473,7 +4503,7 @@ var CircleStopIcon = () => {
       fill: "currentColor",
       width: "16",
       height: "16",
-      children: /* @__PURE__ */ jsx40("rect", { width: "10", height: "10", x: "3", y: "3", rx: "2" })
+      children: /* @__PURE__ */ jsx41("rect", { width: "10", height: "10", x: "3", y: "3", rx: "2" })
     }
   );
 };
@@ -4482,7 +4512,7 @@ CircleStopIcon.displayName = "CircleStopIcon";
 // src/ui/composer-attachment.tsx
 import { forwardRef as forwardRef22 } from "react";
 import { CircleXIcon } from "lucide-react";
-import { jsx as jsx41, jsxs as jsxs9 } from "react/jsx-runtime";
+import { jsx as jsx42, jsxs as jsxs8 } from "react/jsx-runtime";
 var ComposerAttachmentRoot = withDefaults("div", {
   className: "aui-composer-attachment-root"
 });
@@ -4490,10 +4520,10 @@ ComposerAttachmentRoot.displayName = "ComposerAttachmentRoot";
 var ComposerAttachment2 = () => {
   const { useAttachment } = useAttachmentContext({ type: "composer" });
   const attachment = useAttachment((a) => a.attachment);
-  return /* @__PURE__ */ jsxs9(ComposerAttachmentRoot, { children: [
+  return /* @__PURE__ */ jsxs8(ComposerAttachmentRoot, { children: [
     ".",
     attachment.name.split(".").pop(),
-    /* @__PURE__ */ jsx41(ComposerAttachmentRemove, {})
+    /* @__PURE__ */ jsx42(ComposerAttachmentRemove, {})
   ] });
 };
 ComposerAttachment2.displayName = "ComposerAttachment";
@@ -4508,7 +4538,7 @@ var ComposerAttachmentRemove = forwardRef22((props, ref) => {
   const handleRemoveAttachment = () => {
     useComposer.getState().removeAttachment(useAttachment.getState().attachment.id);
   };
-  return /* @__PURE__ */ jsx41(
+  return /* @__PURE__ */ jsx42(
     TooltipIconButton,
     {
       tooltip,
@@ -4517,7 +4547,7 @@ var ComposerAttachmentRemove = forwardRef22((props, ref) => {
       ...props,
       onClick: handleRemoveAttachment,
       ref,
-      children: props.children ?? /* @__PURE__ */ jsx41(CircleXIcon, {})
+      children: props.children ?? /* @__PURE__ */ jsx42(CircleXIcon, {})
     }
   );
 });
@@ -4532,7 +4562,7 @@ var composer_attachment_default = Object.assign(
 );
 
 // src/ui/composer.tsx
-import { Fragment as Fragment5, jsx as jsx42, jsxs as jsxs10 } from "react/jsx-runtime";
+import { Fragment as Fragment5, jsx as jsx43, jsxs as jsxs9 } from "react/jsx-runtime";
 var useAllowAttachments = (ensureCapability = false) => {
   const { composer: { allowAttachments = true } = {} } = useThreadConfig();
   const { useThread } = useThreadContext();
@@ -4541,11 +4571,11 @@ var useAllowAttachments = (ensureCapability = false) => {
 };
 var Composer = () => {
   const allowAttachments = useAllowAttachments(true);
-  return /* @__PURE__ */ jsxs10(ComposerRoot, { children: [
-    allowAttachments && /* @__PURE__ */ jsx42(ComposerAttachments, {}),
-    allowAttachments && /* @__PURE__ */ jsx42(ComposerAddAttachment, {}),
-    /* @__PURE__ */ jsx42(ComposerInput, { autoFocus: true }),
-    /* @__PURE__ */ jsx42(ComposerAction, {})
+  return /* @__PURE__ */ jsxs9(ComposerRoot, { children: [
+    allowAttachments && /* @__PURE__ */ jsx43(ComposerAttachments, {}),
+    allowAttachments && /* @__PURE__ */ jsx43(ComposerAddAttachment, {}),
+    /* @__PURE__ */ jsx43(ComposerInput, { autoFocus: true }),
+    /* @__PURE__ */ jsx43(ComposerAction, {})
   ] });
 };
 Composer.displayName = "Composer";
@@ -4565,7 +4595,7 @@ var ComposerInput = forwardRef23(
         composer: { input: { placeholder = "Write a message..." } = {} } = {}
       } = {}
     } = useThreadConfig();
-    return /* @__PURE__ */ jsx42(ComposerInputStyled, { placeholder, ...props, ref });
+    return /* @__PURE__ */ jsx43(ComposerInputStyled, { placeholder, ...props, ref });
   }
 );
 ComposerInput.displayName = "ComposerInput";
@@ -4573,7 +4603,7 @@ var ComposerAttachmentsContainer = withDefaults("div", {
   className: "aui-composer-attachments"
 });
 var ComposerAttachments = ({ components }) => {
-  return /* @__PURE__ */ jsx42(ComposerAttachmentsContainer, { children: /* @__PURE__ */ jsx42(
+  return /* @__PURE__ */ jsx43(ComposerAttachmentsContainer, { children: /* @__PURE__ */ jsx43(
     composer_exports.Attachments,
     {
       components: {
@@ -4594,14 +4624,14 @@ var ComposerAddAttachment = forwardRef23((props, ref) => {
     } = {}
   } = useThreadConfig();
   const allowAttachments = useAllowAttachments();
-  return /* @__PURE__ */ jsx42(composer_exports.AddAttachment, { disabled: !allowAttachments, asChild: true, children: /* @__PURE__ */ jsx42(
+  return /* @__PURE__ */ jsx43(composer_exports.AddAttachment, { disabled: !allowAttachments, asChild: true, children: /* @__PURE__ */ jsx43(
     ComposerAttachButton,
     {
       tooltip,
       variant: "ghost",
       ...props,
       ref,
-      children: props.children ?? /* @__PURE__ */ jsx42(PaperclipIcon, {})
+      children: props.children ?? /* @__PURE__ */ jsx43(PaperclipIcon, {})
     }
   ) });
 });
@@ -4613,10 +4643,10 @@ var useAllowCancel = () => {
 };
 var ComposerAction = () => {
   const allowCancel = useAllowCancel();
-  if (!allowCancel) return /* @__PURE__ */ jsx42(ComposerSend, {});
-  return /* @__PURE__ */ jsxs10(Fragment5, { children: [
-    /* @__PURE__ */ jsx42(thread_exports.If, { running: false, children: /* @__PURE__ */ jsx42(ComposerSend, {}) }),
-    /* @__PURE__ */ jsx42(thread_exports.If, { running: true, children: /* @__PURE__ */ jsx42(ComposerCancel, {}) })
+  if (!allowCancel) return /* @__PURE__ */ jsx43(ComposerSend, {});
+  return /* @__PURE__ */ jsxs9(Fragment5, { children: [
+    /* @__PURE__ */ jsx43(thread_exports.If, { running: false, children: /* @__PURE__ */ jsx43(ComposerSend, {}) }),
+    /* @__PURE__ */ jsx43(thread_exports.If, { running: true, children: /* @__PURE__ */ jsx43(ComposerCancel, {}) })
   ] });
 };
 ComposerAction.displayName = "ComposerAction";
@@ -4628,7 +4658,7 @@ var ComposerSend = forwardRef23((props, ref) => {
   const {
     strings: { composer: { send: { tooltip = "Send" } = {} } = {} } = {}
   } = useThreadConfig();
-  return /* @__PURE__ */ jsx42(composer_exports.Send, { asChild: true, children: /* @__PURE__ */ jsx42(ComposerSendButton, { tooltip, ...props, ref, children: props.children ?? /* @__PURE__ */ jsx42(SendHorizontalIcon, {}) }) });
+  return /* @__PURE__ */ jsx43(composer_exports.Send, { asChild: true, children: /* @__PURE__ */ jsx43(ComposerSendButton, { tooltip, ...props, ref, children: props.children ?? /* @__PURE__ */ jsx43(SendHorizontalIcon, {}) }) });
 });
 ComposerSend.displayName = "ComposerSend";
 var ComposerCancelButton = withDefaults(TooltipIconButton, {
@@ -4639,7 +4669,7 @@ var ComposerCancel = forwardRef23((props, ref) => {
   const {
     strings: { composer: { cancel: { tooltip = "Cancel" } = {} } = {} } = {}
   } = useThreadConfig();
-  return /* @__PURE__ */ jsx42(composer_exports.Cancel, { asChild: true, children: /* @__PURE__ */ jsx42(ComposerCancelButton, { tooltip, ...props, ref, children: props.children ?? /* @__PURE__ */ jsx42(CircleStopIcon, {}) }) });
+  return /* @__PURE__ */ jsx43(composer_exports.Cancel, { asChild: true, children: /* @__PURE__ */ jsx43(ComposerCancelButton, { tooltip, ...props, ref, children: props.children ?? /* @__PURE__ */ jsx43(CircleStopIcon, {}) }) });
 });
 ComposerCancel.displayName = "ComposerCancel";
 var exports6 = {
@@ -4655,14 +4685,14 @@ var composer_default = Object.assign(Composer, exports6);
 
 // src/ui/thread-welcome.tsx
 import { forwardRef as forwardRef24 } from "react";
-import { jsx as jsx43, jsxs as jsxs11 } from "react/jsx-runtime";
+import { jsx as jsx44, jsxs as jsxs10 } from "react/jsx-runtime";
 var ThreadWelcome = () => {
-  return /* @__PURE__ */ jsxs11(ThreadWelcomeRoot, { children: [
-    /* @__PURE__ */ jsxs11(ThreadWelcomeCenter, { children: [
-      /* @__PURE__ */ jsx43(ThreadWelcomeAvatar, {}),
-      /* @__PURE__ */ jsx43(ThreadWelcomeMessage, {})
+  return /* @__PURE__ */ jsxs10(ThreadWelcomeRoot, { children: [
+    /* @__PURE__ */ jsxs10(ThreadWelcomeCenter, { children: [
+      /* @__PURE__ */ jsx44(ThreadWelcomeAvatar, {}),
+      /* @__PURE__ */ jsx44(ThreadWelcomeMessage, {})
     ] }),
-    /* @__PURE__ */ jsx43(ThreadWelcomeSuggestions, {})
+    /* @__PURE__ */ jsx44(ThreadWelcomeSuggestions, {})
   ] });
 };
 ThreadWelcome.displayName = "ThreadWelcome";
@@ -4674,20 +4704,20 @@ var ThreadWelcomeCenter = withDefaults("div", {
 });
 var ThreadWelcomeRoot = forwardRef24(
   (props, ref) => {
-    return /* @__PURE__ */ jsx43(thread_exports.Empty, { children: /* @__PURE__ */ jsx43(ThreadWelcomeRootStyled, { ...props, ref }) });
+    return /* @__PURE__ */ jsx44(thread_exports.Empty, { children: /* @__PURE__ */ jsx44(ThreadWelcomeRootStyled, { ...props, ref }) });
   }
 );
 ThreadWelcomeRoot.displayName = "ThreadWelcomeRoot";
 var ThreadWelcomeAvatar = () => {
   const { assistantAvatar: avatar = { fallback: "A" } } = useThreadConfig();
-  return /* @__PURE__ */ jsx43(Avatar, { ...avatar });
+  return /* @__PURE__ */ jsx44(Avatar, { ...avatar });
 };
 var ThreadWelcomeMessageStyled = withDefaults("p", {
   className: "aui-thread-welcome-message"
 });
 var ThreadWelcomeMessage = forwardRef24(({ message: messageProp, ...rest }, ref) => {
   const { welcome: { message = "How can I help you today?" } = {} } = useThreadConfig();
-  return /* @__PURE__ */ jsx43(ThreadWelcomeMessageStyled, { ...rest, ref, children: messageProp ?? message });
+  return /* @__PURE__ */ jsx44(ThreadWelcomeMessageStyled, { ...rest, ref, children: messageProp ?? message });
 });
 ThreadWelcomeMessage.displayName = "ThreadWelcomeMessage";
 var ThreadWelcomeSuggestionContainer = withDefaults("div", {
@@ -4699,21 +4729,21 @@ var ThreadWelcomeSuggestionStyled = withDefaults(thread_exports.Suggestion, {
 var ThreadWelcomeSuggestion = ({
   suggestion: { text, prompt }
 }) => {
-  return /* @__PURE__ */ jsx43(
+  return /* @__PURE__ */ jsx44(
     ThreadWelcomeSuggestionStyled,
     {
       prompt,
       method: "replace",
       autoSend: true,
-      children: /* @__PURE__ */ jsx43("span", { className: "aui-thread-welcome-suggestion-text", children: text ?? prompt })
+      children: /* @__PURE__ */ jsx44("span", { className: "aui-thread-welcome-suggestion-text", children: text ?? prompt })
     }
   );
 };
 var ThreadWelcomeSuggestions = () => {
   const { welcome: { suggestions } = {} } = useThreadConfig();
-  return /* @__PURE__ */ jsx43(ThreadWelcomeSuggestionContainer, { children: suggestions?.map((suggestion, idx) => {
+  return /* @__PURE__ */ jsx44(ThreadWelcomeSuggestionContainer, { children: suggestions?.map((suggestion, idx) => {
     const key = `${suggestion.prompt}-${idx}`;
-    return /* @__PURE__ */ jsx43(ThreadWelcomeSuggestion, { suggestion }, key);
+    return /* @__PURE__ */ jsx44(ThreadWelcomeSuggestion, { suggestion }, key);
   }) });
 };
 ThreadWelcomeSuggestions.displayName = "ThreadWelcomeSuggestions";
@@ -4733,7 +4763,7 @@ import { forwardRef as forwardRef26 } from "react";
 // src/ui/user-action-bar.tsx
 import { forwardRef as forwardRef25 } from "react";
 import { PencilIcon } from "lucide-react";
-import { jsx as jsx44 } from "react/jsx-runtime";
+import { jsx as jsx45 } from "react/jsx-runtime";
 var useAllowEdit = (ensureCapability = false) => {
   const { userMessage: { allowEdit = true } = {} } = useThreadConfig();
   const { useThread } = useThreadContext();
@@ -4743,7 +4773,7 @@ var useAllowEdit = (ensureCapability = false) => {
 var UserActionBar = () => {
   const allowEdit = useAllowEdit(true);
   if (!allowEdit) return null;
-  return /* @__PURE__ */ jsx44(UserActionBarRoot, { hideWhenRunning: true, autohide: "not-last", children: /* @__PURE__ */ jsx44(UserActionBarEdit, {}) });
+  return /* @__PURE__ */ jsx45(UserActionBarRoot, { hideWhenRunning: true, autohide: "not-last", children: /* @__PURE__ */ jsx45(UserActionBarEdit, {}) });
 };
 UserActionBar.displayName = "UserActionBar";
 var UserActionBarRoot = withDefaults(actionBar_exports.Root, {
@@ -4755,7 +4785,7 @@ var UserActionBarEdit = forwardRef25((props, ref) => {
     strings: { userMessage: { edit: { tooltip = "Edit" } = {} } = {} } = {}
   } = useThreadConfig();
   const allowEdit = useAllowEdit();
-  return /* @__PURE__ */ jsx44(actionBar_exports.Edit, { disabled: !allowEdit, asChild: true, children: /* @__PURE__ */ jsx44(TooltipIconButton, { tooltip, ...props, ref, children: props.children ?? /* @__PURE__ */ jsx44(PencilIcon, {}) }) });
+  return /* @__PURE__ */ jsx45(actionBar_exports.Edit, { disabled: !allowEdit, asChild: true, children: /* @__PURE__ */ jsx45(TooltipIconButton, { tooltip, ...props, ref, children: props.children ?? /* @__PURE__ */ jsx45(PencilIcon, {}) }) });
 });
 UserActionBarEdit.displayName = "UserActionBarEdit";
 var exports8 = {
@@ -4765,7 +4795,7 @@ var exports8 = {
 var user_action_bar_default = Object.assign(UserActionBar, exports8);
 
 // src/ui/user-message-attachment.tsx
-import { jsxs as jsxs12 } from "react/jsx-runtime";
+import { jsxs as jsxs11 } from "react/jsx-runtime";
 var UserMessageAttachmentRoot = withDefaults("div", {
   className: "aui-user-message-attachment-root"
 });
@@ -4773,7 +4803,7 @@ UserMessageAttachmentRoot.displayName = "UserMessageAttachmentRoot";
 var UserMessageAttachment = () => {
   const { useAttachment } = useAttachmentContext();
   const attachment = useAttachment((a) => a.attachment);
-  return /* @__PURE__ */ jsxs12(UserMessageAttachmentRoot, { children: [
+  return /* @__PURE__ */ jsxs11(UserMessageAttachmentRoot, { children: [
     ".",
     attachment.name.split(".").pop()
   ] });
@@ -4788,13 +4818,13 @@ var user_message_attachment_default = Object.assign(
 );
 
 // src/ui/user-message.tsx
-import { jsx as jsx45, jsxs as jsxs13 } from "react/jsx-runtime";
+import { jsx as jsx46, jsxs as jsxs12 } from "react/jsx-runtime";
 var UserMessage = () => {
-  return /* @__PURE__ */ jsxs13(UserMessageRoot, { children: [
-    /* @__PURE__ */ jsx45(UserMessageAttachments, {}),
-    /* @__PURE__ */ jsx45(user_action_bar_default, {}),
-    /* @__PURE__ */ jsx45(UserMessageContent, {}),
-    /* @__PURE__ */ jsx45(branch_picker_default, {})
+  return /* @__PURE__ */ jsxs12(UserMessageRoot, { children: [
+    /* @__PURE__ */ jsx46(UserMessageAttachments, {}),
+    /* @__PURE__ */ jsx46(user_action_bar_default, {}),
+    /* @__PURE__ */ jsx46(UserMessageContent, {}),
+    /* @__PURE__ */ jsx46(branch_picker_default, {})
   ] });
 };
 UserMessage.displayName = "UserMessage";
@@ -4807,7 +4837,7 @@ var UserMessageContentWrapper = withDefaults("div", {
 });
 var UserMessageContent = forwardRef26(
   ({ components, ...props }, ref) => {
-    return /* @__PURE__ */ jsx45(UserMessageContentWrapper, { ...props, ref, children: /* @__PURE__ */ jsx45(
+    return /* @__PURE__ */ jsx46(UserMessageContentWrapper, { ...props, ref, children: /* @__PURE__ */ jsx46(
       message_exports.Content,
       {
         components: {
@@ -4825,7 +4855,7 @@ var UserMessageAttachmentsContainer = withDefaults("div", {
 var UserMessageAttachments = ({
   components
 }) => {
-  return /* @__PURE__ */ jsx45(message_exports.If, { hasAttachments: true, children: /* @__PURE__ */ jsx45(UserMessageAttachmentsContainer, { children: /* @__PURE__ */ jsx45(
+  return /* @__PURE__ */ jsx46(message_exports.If, { hasAttachments: true, children: /* @__PURE__ */ jsx46(UserMessageAttachmentsContainer, { children: /* @__PURE__ */ jsx46(
     message_exports.Attachments,
     {
       components: {
@@ -4844,13 +4874,13 @@ var user_message_default = Object.assign(UserMessage, exports10);
 
 // src/ui/edit-composer.tsx
 import { forwardRef as forwardRef27 } from "react";
-import { jsx as jsx46, jsxs as jsxs14 } from "react/jsx-runtime";
+import { jsx as jsx47, jsxs as jsxs13 } from "react/jsx-runtime";
 var EditComposer = () => {
-  return /* @__PURE__ */ jsxs14(EditComposerRoot, { children: [
-    /* @__PURE__ */ jsx46(EditComposerInput, {}),
-    /* @__PURE__ */ jsxs14(EditComposerFooter, { children: [
-      /* @__PURE__ */ jsx46(EditComposerCancel, {}),
-      /* @__PURE__ */ jsx46(EditComposerSend, {})
+  return /* @__PURE__ */ jsxs13(EditComposerRoot, { children: [
+    /* @__PURE__ */ jsx47(EditComposerInput, {}),
+    /* @__PURE__ */ jsxs13(EditComposerFooter, { children: [
+      /* @__PURE__ */ jsx47(EditComposerCancel, {}),
+      /* @__PURE__ */ jsx47(EditComposerSend, {})
     ] })
   ] });
 };
@@ -4874,7 +4904,7 @@ var EditComposerCancel = forwardRef27(
         editComposer: { cancel: { label = "Cancel" } = {} } = {}
       } = {}
     } = useThreadConfig();
-    return /* @__PURE__ */ jsx46(composer_exports.Cancel, { asChild: true, children: /* @__PURE__ */ jsx46(Button, { variant: "ghost", ...props, ref, children: props.children ?? label }) });
+    return /* @__PURE__ */ jsx47(composer_exports.Cancel, { asChild: true, children: /* @__PURE__ */ jsx47(Button, { variant: "ghost", ...props, ref, children: props.children ?? label }) });
   }
 );
 EditComposerCancel.displayName = "EditComposerCancel";
@@ -4883,7 +4913,7 @@ var EditComposerSend = forwardRef27(
     const {
       strings: { editComposer: { send: { label = "Send" } = {} } = {} } = {}
     } = useThreadConfig();
-    return /* @__PURE__ */ jsx46(composer_exports.Send, { asChild: true, children: /* @__PURE__ */ jsx46(Button, { ...props, ref, children: props.children ?? label }) });
+    return /* @__PURE__ */ jsx47(composer_exports.Send, { asChild: true, children: /* @__PURE__ */ jsx47(Button, { ...props, ref, children: props.children ?? label }) });
   }
 );
 EditComposerSend.displayName = "EditComposerSend";
@@ -4897,14 +4927,14 @@ var exports11 = {
 var edit_composer_default = Object.assign(EditComposer, exports11);
 
 // src/ui/thread.tsx
-import { jsx as jsx47, jsxs as jsxs15 } from "react/jsx-runtime";
+import { jsx as jsx48, jsxs as jsxs14 } from "react/jsx-runtime";
 var Thread = (config) => {
-  return /* @__PURE__ */ jsx47(ThreadRoot, { config, children: /* @__PURE__ */ jsxs15(ThreadViewport, { children: [
-    /* @__PURE__ */ jsx47(thread_welcome_default, {}),
-    /* @__PURE__ */ jsx47(ThreadMessages, {}),
-    /* @__PURE__ */ jsxs15(ThreadViewportFooter, { children: [
-      /* @__PURE__ */ jsx47(ThreadScrollToBottom, {}),
-      /* @__PURE__ */ jsx47(composer_default, {})
+  return /* @__PURE__ */ jsx48(ThreadRoot, { config, children: /* @__PURE__ */ jsxs14(ThreadViewport, { children: [
+    /* @__PURE__ */ jsx48(thread_welcome_default, {}),
+    /* @__PURE__ */ jsx48(ThreadMessages, {}),
+    /* @__PURE__ */ jsxs14(ThreadViewportFooter, { children: [
+      /* @__PURE__ */ jsx48(ThreadScrollToBottom, {}),
+      /* @__PURE__ */ jsx48(composer_default, {})
     ] })
   ] }) });
 };
@@ -4913,7 +4943,7 @@ var ThreadRootStyled = withDefaults(thread_exports.Root, {
 });
 var ThreadRoot = forwardRef28(
   ({ config, ...props }, ref) => {
-    return /* @__PURE__ */ jsx47(ThreadConfigProvider, { config, children: /* @__PURE__ */ jsx47(ThreadRootStyled, { ...props, ref }) });
+    return /* @__PURE__ */ jsx48(ThreadConfigProvider, { config, children: /* @__PURE__ */ jsx48(ThreadRootStyled, { ...props, ref }) });
   }
 );
 ThreadRoot.displayName = "ThreadRoot";
@@ -4927,7 +4957,7 @@ var ThreadViewportFooter = withDefaults("div", {
 ThreadViewportFooter.displayName = "ThreadViewportFooter";
 var SystemMessage = () => null;
 var ThreadMessages = ({ components, ...rest }) => {
-  return /* @__PURE__ */ jsx47(
+  return /* @__PURE__ */ jsx48(
     thread_exports.Messages,
     {
       components: {
@@ -4951,7 +4981,7 @@ var ThreadScrollToBottom = forwardRef28((props, ref) => {
       thread: { scrollToBottom: { tooltip = "Scroll to bottom" } = {} } = {}
     } = {}
   } = useThreadConfig();
-  return /* @__PURE__ */ jsx47(thread_exports.ScrollToBottom, { asChild: true, children: /* @__PURE__ */ jsx47(ThreadScrollToBottomIconButton, { tooltip, ...props, ref, children: props.children ?? /* @__PURE__ */ jsx47(ArrowDownIcon, {}) }) });
+  return /* @__PURE__ */ jsx48(thread_exports.ScrollToBottom, { asChild: true, children: /* @__PURE__ */ jsx48(ThreadScrollToBottomIconButton, { tooltip, ...props, ref, children: props.children ?? /* @__PURE__ */ jsx48(ArrowDownIcon, {}) }) });
 });
 ThreadScrollToBottom.displayName = "ThreadScrollToBottom";
 var exports12 = {
@@ -4964,20 +4994,20 @@ var exports12 = {
 var thread_default = Object.assign(Thread, exports12);
 
 // src/ui/assistant-modal.tsx
-import { Fragment as Fragment6, jsx as jsx48, jsxs as jsxs16 } from "react/jsx-runtime";
+import { Fragment as Fragment6, jsx as jsx49, jsxs as jsxs15 } from "react/jsx-runtime";
 var AssistantModal = (config) => {
-  return /* @__PURE__ */ jsxs16(AssistantModalRoot, { config, children: [
-    /* @__PURE__ */ jsx48(AssistantModalTrigger, {}),
-    /* @__PURE__ */ jsx48(AssistantModalContent, { children: /* @__PURE__ */ jsx48(thread_default, {}) })
+  return /* @__PURE__ */ jsxs15(AssistantModalRoot, { config, children: [
+    /* @__PURE__ */ jsx49(AssistantModalTrigger, {}),
+    /* @__PURE__ */ jsx49(AssistantModalContent, { children: /* @__PURE__ */ jsx49(thread_default, {}) })
   ] });
 };
 AssistantModal.displayName = "AssistantModal";
 var AssistantModalRoot = ({ config, ...props }) => {
-  return /* @__PURE__ */ jsx48(ThreadConfigProvider, { config, children: /* @__PURE__ */ jsx48(assistantModal_exports.Root, { ...props }) });
+  return /* @__PURE__ */ jsx49(ThreadConfigProvider, { config, children: /* @__PURE__ */ jsx49(assistantModal_exports.Root, { ...props }) });
 };
 AssistantModalRoot.displayName = "AssistantModalRoot";
 var AssistantModalTrigger = forwardRef29((props, ref) => {
-  return /* @__PURE__ */ jsx48(AssistantModalAnchor, { children: /* @__PURE__ */ jsx48(assistantModal_exports.Trigger, { asChild: true, children: /* @__PURE__ */ jsx48(AssistantModalButton, { ...props, ref }) }) });
+  return /* @__PURE__ */ jsx49(AssistantModalAnchor, { children: /* @__PURE__ */ jsx49(assistantModal_exports.Trigger, { asChild: true, children: /* @__PURE__ */ jsx49(AssistantModalButton, { ...props, ref }) }) });
 });
 AssistantModalTrigger.displayName = "AssistantModalTrigger";
 var AssistantModalAnchor = withDefaults(assistantModal_exports.Anchor, {
@@ -5002,7 +5032,7 @@ var AssistantModalButton = forwardRef29(({ "data-state": state, ...rest }, ref)
     } = {}
   } = useThreadConfig();
   const tooltip = state === "open" ? openTooltip : closedTooltip;
-  return /* @__PURE__ */ jsx48(
+  return /* @__PURE__ */ jsx49(
     ModalButtonStyled,
     {
       side: "left",
@@ -5010,15 +5040,15 @@ var AssistantModalButton = forwardRef29(({ "data-state": state, ...rest }, ref)
       "data-state": state,
       ...rest,
       ref,
-      children: rest.children ?? /* @__PURE__ */ jsxs16(Fragment6, { children: [
-        /* @__PURE__ */ jsx48(
+      children: rest.children ?? /* @__PURE__ */ jsxs15(Fragment6, { children: [
+        /* @__PURE__ */ jsx49(
           BotIcon,
           {
             "data-state": state,
             className: "aui-modal-button-closed-icon"
           }
         ),
-        /* @__PURE__ */ jsx48(
+        /* @__PURE__ */ jsx49(
           ChevronDownIcon,
           {
             "data-state": state,
@@ -5065,6 +5095,7 @@ export {
   message_exports as MessagePrimitive,
   SimpleImageAttachmentAdapter,
   SimpleTextAttachmentAdapter,
+  TextContentPartProvider,
   thread_default as Thread,
   ThreadConfigProvider,
   thread_exports as ThreadPrimitive,