@assistant-ui/react 0.5.21 → 0.5.22

package/dist/index.mjs

@@ -1,37 +1,13 @@
 "use client";
 import {
+  mergeModelConfigs,
   runResultStream,
   toCoreMessage,
   toCoreMessages,
   toLanguageModelMessages,
   toLanguageModelTools,
   toolResultStream
-} from "./chunk-2RKUKZSZ.mjs";
-import {
-  AssistantContext,
-  BaseAssistantRuntime,
-  Button,
-  ContentPartContext,
-  MessageContext,
-  MessageRepository,
-  ProxyConfigProvider,
-  ThreadContext,
-  TooltipIconButton,
-  fromCoreMessage,
-  fromCoreMessages,
-  generateId,
-  internal_exports,
-  useAssistantContext,
-  useComposerContext,
-  useContentPartContext,
-  useMessageContext,
-  useSmooth,
-  useSmoothStatus,
-  useThreadContext,
-  withDefaults,
-  withSmoothContextProvider
-} from "./chunk-V66MVXBH.mjs";
-import "./chunk-QBS6JLLN.mjs";
+} from "./chunk-ZWRFAYHH.mjs";
 import {
   __export
 } from "./chunk-BJPOCE4O.mjs";
@@ -42,8 +18,38 @@ import { memo } from "react";
 // src/context/providers/AssistantProvider.tsx
 import { useEffect, useInsertionEffect as useInsertionEffect2, useRef as useRef2, useState as useState2 } from "react";
 
+// src/context/react/AssistantContext.ts
+import { createContext, useContext } from "react";
+var AssistantContext = createContext(
+  null
+);
+function useAssistantContext(options) {
+  const context = useContext(AssistantContext);
+  if (!options?.optional && !context)
+    throw new Error(
+      "This component must be used within an AssistantRuntimeProvider."
+    );
+  return context;
+}
+
 // src/context/stores/AssistantModelConfig.ts
 import { create } from "zustand";
+
+// src/utils/ProxyConfigProvider.ts
+var ProxyConfigProvider = class {
+  _providers = /* @__PURE__ */ new Set();
+  getModelConfig() {
+    return mergeModelConfigs(this._providers);
+  }
+  registerModelConfigProvider(provider) {
+    this._providers.add(provider);
+    return () => {
+      this._providers.delete(provider);
+    };
+  }
+};
+
+// src/context/stores/AssistantModelConfig.ts
 var makeAssistantModelConfigStore = () => create(() => {
   const proxy = new ProxyConfigProvider();
   return Object.freeze({
@@ -91,6 +97,18 @@ var makeAssistantToolUIsStore = () => create2((set) => {
 // src/context/providers/ThreadProvider.tsx
 import { useCallback as useCallback2, useInsertionEffect, useState } from "react";
 
+// src/context/react/ThreadContext.ts
+import { createContext as createContext2, useContext as useContext2 } from "react";
+var ThreadContext = createContext2(null);
+function useThreadContext(options) {
+  const context = useContext2(ThreadContext);
+  if (!options?.optional && !context)
+    throw new Error(
+      "This component must be used within an AssistantRuntimeProvider."
+    );
+  return context;
+}
+
 // src/context/stores/Composer.ts
 import { create as create3 } from "zustand";
 
@@ -318,6 +336,48 @@ var AssistantRuntimeProviderImpl = ({ children, runtime }) => {
 };
 var AssistantRuntimeProvider = memo(AssistantRuntimeProviderImpl);
 
+// src/context/react/ComposerContext.ts
+import { useMemo } from "react";
+
+// src/context/react/MessageContext.ts
+import { createContext as createContext3, useContext as useContext3 } from "react";
+var MessageContext = createContext3(null);
+function useMessageContext(options) {
+  const context = useContext3(MessageContext);
+  if (!options?.optional && !context)
+    throw new Error(
+      "This component can only be used inside a component passed to <ThreadPrimitive.Messages components={...} />."
+    );
+  return context;
+}
+
+// src/context/react/ComposerContext.ts
+var useComposerContext = () => {
+  const { useComposer } = useThreadContext();
+  const { useEditComposer } = useMessageContext({ optional: true }) ?? {};
+  return useMemo(
+    () => ({
+      useComposer: useEditComposer ?? useComposer,
+      type: useEditComposer ? "edit" : "new"
+    }),
+    [useEditComposer, useComposer]
+  );
+};
+
+// 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/hooks/useAppendMessage.tsx
 import { useCallback as useCallback3 } from "react";
 var toAppendMessage = (useThreadMessages, message) => {
@@ -435,7 +495,7 @@ var useAssistantInstructions = (instruction) => {
 import { useCallback as useCallback5 } from "react";
 
 // src/utils/combined/useCombinedStore.ts
-import { useMemo } from "react";
+import { useMemo as useMemo2 } from "react";
 
 // src/utils/combined/createCombinedStore.ts
 import { useSyncExternalStore } from "react";
@@ -456,7 +516,7 @@ var createCombinedStore = (stores) => {
 
 // src/utils/combined/useCombinedStore.ts
 var useCombinedStore = (stores, selector) => {
-  const useCombined = useMemo(() => createCombinedStore(stores), stores);
+  const useCombined = useMemo2(() => createCombinedStore(stores), stores);
   return useCombined(selector);
 };
 
@@ -1011,7 +1071,7 @@ BranchPickerPrimitiveNumber.displayName = "BranchPickerPrimitive.Number";
 
 // src/primitives/branchPicker/BranchPickerRoot.tsx
 import { Primitive as Primitive5 } from "@radix-ui/react-primitive";
-import { forwardRef as forwardRef9 } from "react";
+import { forwardRef as forwardRef10 } from "react";
 
 // src/primitives/message/index.ts
 var message_exports = {};
@@ -1053,9 +1113,9 @@ var useIsHoveringRef = () => {
   );
   return useManagedRef(callbackRef);
 };
-var MessagePrimitiveRoot = forwardRef6(({ onMouseEnter, onMouseLeave, ...rest }, forwardRef25) => {
+var MessagePrimitiveRoot = forwardRef6(({ onMouseEnter, onMouseLeave, ...rest }, forwardRef29) => {
   const isHoveringRef = useIsHoveringRef();
-  const ref = useComposedRefs(forwardRef25, isHoveringRef);
+  const ref = useComposedRefs(forwardRef29, isHoveringRef);
   return /* @__PURE__ */ jsx12(Primitive3.div, { ...rest, ref });
 });
 MessagePrimitiveRoot.displayName = "MessagePrimitive.Root";
@@ -1144,27 +1204,170 @@ var ContentPartProvider = ({
 
 // src/primitives/contentPart/ContentPartText.tsx
 import {
-  forwardRef as forwardRef7
+  forwardRef as forwardRef8
+} from "react";
+
+// src/utils/smooth/useSmooth.tsx
+import { useEffect as useEffect7, useMemo as useMemo3, useRef as useRef3, useState as useState6 } from "react";
+
+// src/utils/smooth/SmoothContext.tsx
+import {
+  createContext as createContext5,
+  forwardRef as forwardRef7,
+  useContext as useContext5,
+  useState as useState5
 } from "react";
+import { create as create11 } from "zustand";
 import { jsx as jsx14 } from "react/jsx-runtime";
-var ContentPartPrimitiveText = forwardRef7(({ smooth = true, component: Component = "span", ...rest }, forwardedRef) => {
+var SmoothContext = createContext5(null);
+var makeSmoothContext = (initialState) => {
+  const useSmoothStatus2 = create11(() => initialState);
+  return { useSmoothStatus: useSmoothStatus2 };
+};
+var SmoothContextProvider = ({ children }) => {
+  const outer = useSmoothContext({ optional: true });
+  const { useContentPart } = useContentPartContext();
+  const [context] = useState5(
+    () => makeSmoothContext(useContentPart.getState().status)
+  );
+  if (outer) return children;
+  return /* @__PURE__ */ jsx14(SmoothContext.Provider, { value: context, children });
+};
+var withSmoothContextProvider = (Component) => {
+  const Wrapped = forwardRef7((props, ref) => {
+    return /* @__PURE__ */ jsx14(SmoothContextProvider, { children: /* @__PURE__ */ jsx14(Component, { ...props, ref }) });
+  });
+  Wrapped.displayName = Component.displayName;
+  return Wrapped;
+};
+function useSmoothContext(options) {
+  const context = useContext5(SmoothContext);
+  if (!options?.optional && !context)
+    throw new Error(
+      "This component must be used within a SmoothContextProvider."
+    );
+  return context;
+}
+var useSmoothStatus = () => {
+  const { useSmoothStatus: useSmoothStatus2 } = useSmoothContext();
+  return useSmoothStatus2();
+};
+
+// src/utils/smooth/useSmooth.tsx
+import { useCallbackRef as useCallbackRef2 } from "@radix-ui/react-use-callback-ref";
+var TextStreamAnimator = class {
+  constructor(currentText, setText) {
+    this.currentText = currentText;
+    this.setText = setText;
+  }
+  animationFrameId = null;
+  lastUpdateTime = Date.now();
+  targetText = "";
+  start() {
+    if (this.animationFrameId !== null) return;
+    this.lastUpdateTime = Date.now();
+    this.animate();
+  }
+  stop() {
+    if (this.animationFrameId !== null) {
+      cancelAnimationFrame(this.animationFrameId);
+      this.animationFrameId = null;
+    }
+  }
+  animate = () => {
+    const currentTime = Date.now();
+    const deltaTime = currentTime - this.lastUpdateTime;
+    let timeToConsume = deltaTime;
+    const remainingChars = this.targetText.length - this.currentText.length;
+    const baseTimePerChar = Math.min(5, 250 / remainingChars);
+    let charsToAdd = 0;
+    while (timeToConsume >= baseTimePerChar && charsToAdd < remainingChars) {
+      charsToAdd++;
+      timeToConsume -= baseTimePerChar;
+    }
+    if (charsToAdd !== remainingChars) {
+      this.animationFrameId = requestAnimationFrame(this.animate);
+    } else {
+      this.animationFrameId = null;
+    }
+    if (charsToAdd === 0) return;
+    this.currentText = this.targetText.slice(
+      0,
+      this.currentText.length + charsToAdd
+    );
+    this.lastUpdateTime = currentTime - timeToConsume;
+    this.setText(this.currentText);
+  };
+};
+var SMOOTH_STATUS = Object.freeze({
+  type: "running"
+});
+var useSmooth = (state, smooth = false) => {
+  const { useSmoothStatus: useSmoothStatus2 } = useSmoothContext({ optional: true }) ?? {};
+  const {
+    part: { text }
+  } = state;
+  const { useMessage } = useMessageContext();
+  const id = useMessage((m) => m.message.id);
+  const idRef = useRef3(id);
+  const [displayedText, setDisplayedText] = useState6(text);
+  const setText = useCallbackRef2((text2) => {
+    setDisplayedText(text2);
+    useSmoothStatus2?.setState(text2 !== state.part.text ? SMOOTH_STATUS : state.status);
+  });
+  const [animatorRef] = useState6(
+    new TextStreamAnimator(text, setText)
+  );
+  useEffect7(() => {
+    if (!smooth) {
+      animatorRef.stop();
+      return;
+    }
+    if (idRef.current !== id || !text.startsWith(animatorRef.targetText)) {
+      idRef.current = id;
+      setText(text);
+      animatorRef.currentText = text;
+      animatorRef.targetText = text;
+      animatorRef.stop();
+      return;
+    }
+    animatorRef.targetText = text;
+    animatorRef.start();
+  }, [setText, animatorRef, id, smooth, text]);
+  useEffect7(() => {
+    return () => {
+      animatorRef.stop();
+    };
+  }, [animatorRef]);
+  return useMemo3(
+    () => smooth ? {
+      part: { type: "text", text: displayedText },
+      status: text === displayedText ? state.status : SMOOTH_STATUS
+    } : state,
+    [smooth, displayedText, state, text]
+  );
+};
+
+// src/primitives/contentPart/ContentPartText.tsx
+import { jsx as jsx15 } from "react/jsx-runtime";
+var ContentPartPrimitiveText = forwardRef8(({ smooth = true, component: Component = "span", ...rest }, forwardedRef) => {
   const {
     part: { text },
     status
   } = useSmooth(useContentPartText(), smooth);
-  return /* @__PURE__ */ jsx14(Component, { "data-status": status.type, ...rest, ref: forwardedRef, children: text });
+  return /* @__PURE__ */ jsx15(Component, { "data-status": status.type, ...rest, ref: forwardedRef, children: text });
 });
 ContentPartPrimitiveText.displayName = "ContentPartPrimitive.Text";
 
 // src/primitives/contentPart/ContentPartImage.tsx
 import { Primitive as Primitive4 } from "@radix-ui/react-primitive";
-import { forwardRef as forwardRef8 } from "react";
-import { jsx as jsx15 } from "react/jsx-runtime";
-var ContentPartPrimitiveImage = forwardRef8((props, forwardedRef) => {
+import { forwardRef as forwardRef9 } from "react";
+import { jsx as jsx16 } from "react/jsx-runtime";
+var ContentPartPrimitiveImage = forwardRef9((props, forwardedRef) => {
   const {
     part: { image }
   } = useContentPartImage();
-  return /* @__PURE__ */ jsx15(Primitive4.img, { src: image, ...props, ref: forwardedRef });
+  return /* @__PURE__ */ jsx16(Primitive4.img, { src: image, ...props, ref: forwardedRef });
 });
 ContentPartPrimitiveImage.displayName = "ContentPartPrimitive.Image";
 
@@ -1186,20 +1389,20 @@ var ContentPartPrimitiveInProgress = ({ children }) => {
 ContentPartPrimitiveInProgress.displayName = "ContentPartPrimitive.InProgress";
 
 // src/primitives/message/MessageContent.tsx
-import { jsx as jsx16, jsxs as jsxs2 } from "react/jsx-runtime";
+import { jsx as jsx17, jsxs as jsxs2 } from "react/jsx-runtime";
 var defaultComponents = {
   Text: () => /* @__PURE__ */ jsxs2("p", { style: { whiteSpace: "pre-line" }, children: [
-    /* @__PURE__ */ jsx16(ContentPartPrimitiveText, {}),
-    /* @__PURE__ */ jsx16(ContentPartPrimitiveInProgress, { children: /* @__PURE__ */ jsx16("span", { style: { fontFamily: "revert" }, children: " \u25CF" }) })
+    /* @__PURE__ */ jsx17(ContentPartPrimitiveText, {}),
+    /* @__PURE__ */ jsx17(ContentPartPrimitiveInProgress, { children: /* @__PURE__ */ jsx17("span", { style: { fontFamily: "revert" }, children: " \u25CF" }) })
   ] }),
-  Image: () => /* @__PURE__ */ jsx16(ContentPartPrimitiveImage, {}),
-  UI: () => /* @__PURE__ */ jsx16(ContentPartPrimitiveDisplay, {}),
+  Image: () => /* @__PURE__ */ jsx17(ContentPartPrimitiveImage, {}),
+  UI: () => /* @__PURE__ */ jsx17(ContentPartPrimitiveDisplay, {}),
   tools: {
     Fallback: (props) => {
       const { useToolUIs } = useAssistantContext();
       const Render = useToolUIs((s) => s.getToolUI(props.part.toolName));
       if (!Render) return null;
-      return /* @__PURE__ */ jsx16(Render, { ...props });
+      return /* @__PURE__ */ jsx17(Render, { ...props });
     }
   }
 };
@@ -1222,16 +1425,16 @@ var MessageContentPartComponent = ({
     case "text":
       if (status.type === "requires-action")
         throw new Error("Encountered unexpected requires-action status");
-      if (part === EMPTY_CONTENT) return /* @__PURE__ */ jsx16(Empty, { part, status });
-      return /* @__PURE__ */ jsx16(Text2, { part, status });
+      if (part === EMPTY_CONTENT) return /* @__PURE__ */ jsx17(Empty, { part, status });
+      return /* @__PURE__ */ jsx17(Text2, { part, status });
     case "image":
       if (status.type === "requires-action")
         throw new Error("Encountered unexpected requires-action status");
-      return /* @__PURE__ */ jsx16(Image2, { part, status });
+      return /* @__PURE__ */ jsx17(Image2, { part, status });
     case "ui":
       if (status.type === "requires-action")
         throw new Error("Encountered unexpected requires-action status");
-      return /* @__PURE__ */ jsx16(UI, { part, status });
+      return /* @__PURE__ */ jsx17(UI, { part, status });
     case "tool-call": {
       const Tool = by_name[part.toolName] || Fallback2;
       const addResult = (result) => addToolResult({
@@ -1239,7 +1442,7 @@ var MessageContentPartComponent = ({
         toolCallId: part.toolCallId,
         result
       });
-      return /* @__PURE__ */ jsx16(Tool, { part, status, addResult });
+      return /* @__PURE__ */ jsx17(Tool, { part, status, addResult });
     }
     default:
       const unhandledType = type;
@@ -1250,7 +1453,7 @@ var MessageContentPartImpl = ({
   partIndex,
   components
 }) => {
-  return /* @__PURE__ */ jsx16(ContentPartProvider, { partIndex, children: /* @__PURE__ */ jsx16(MessageContentPartComponent, { components }) });
+  return /* @__PURE__ */ jsx17(ContentPartProvider, { partIndex, children: /* @__PURE__ */ jsx17(MessageContentPartComponent, { components }) });
 };
 var MessageContentPart = memo2(
   MessageContentPartImpl,
@@ -1263,7 +1466,7 @@ var MessagePrimitiveContent = ({
   const contentLength = useMessage((s) => s.message.content.length) || 1;
   return new Array(contentLength).fill(null).map((_, idx) => {
     const partIndex = idx;
-    return /* @__PURE__ */ jsx16(
+    return /* @__PURE__ */ jsx17(
       MessageContentPart,
       {
         partIndex,
@@ -1282,9 +1485,9 @@ var MessagePrimitiveInProgress = () => {
 MessagePrimitiveInProgress.displayName = "MessagePrimitive.InProgress";
 
 // src/primitives/branchPicker/BranchPickerRoot.tsx
-import { jsx as jsx17 } from "react/jsx-runtime";
-var BranchPickerPrimitiveRoot = forwardRef9(({ hideWhenSingleBranch, ...rest }, ref) => {
-  return /* @__PURE__ */ jsx17(MessagePrimitiveIf, { hasBranches: hideWhenSingleBranch ? true : void 0, children: /* @__PURE__ */ jsx17(Primitive5.div, { ...rest, ref }) });
+import { jsx as jsx18 } from "react/jsx-runtime";
+var BranchPickerPrimitiveRoot = forwardRef10(({ hideWhenSingleBranch, ...rest }, ref) => {
+  return /* @__PURE__ */ jsx18(MessagePrimitiveIf, { hasBranches: hideWhenSingleBranch ? true : void 0, children: /* @__PURE__ */ jsx18(Primitive5.div, { ...rest, ref }) });
 });
 BranchPickerPrimitiveRoot.displayName = "BranchPickerPrimitive.Root";
 
@@ -1302,17 +1505,17 @@ __export(composer_exports, {
 import { composeEventHandlers as composeEventHandlers4 } from "@radix-ui/primitive";
 import { Primitive as Primitive6 } from "@radix-ui/react-primitive";
 import {
-  forwardRef as forwardRef10
+  forwardRef as forwardRef11
 } from "react";
-import { jsx as jsx18 } from "react/jsx-runtime";
-var ComposerPrimitiveRoot = forwardRef10(({ onSubmit, ...rest }, forwardedRef) => {
+import { jsx as jsx19 } from "react/jsx-runtime";
+var ComposerPrimitiveRoot = forwardRef11(({ onSubmit, ...rest }, forwardedRef) => {
   const send = useComposerSend();
   const handleSubmit = (e) => {
     e.preventDefault();
     if (!send) return;
     send();
   };
-  return /* @__PURE__ */ jsx18(
+  return /* @__PURE__ */ jsx19(
     Primitive6.form,
     {
       ...rest,
@@ -1328,15 +1531,15 @@ import { composeEventHandlers as composeEventHandlers5 } from "@radix-ui/primiti
 import { useComposedRefs as useComposedRefs2 } from "@radix-ui/react-compose-refs";
 import { Slot } from "@radix-ui/react-slot";
 import {
-  forwardRef as forwardRef11,
+  forwardRef as forwardRef12,
   useCallback as useCallback15,
-  useEffect as useEffect7,
-  useRef as useRef3
+  useEffect as useEffect8,
+  useRef as useRef4
 } from "react";
 import TextareaAutosize from "react-textarea-autosize";
 import { useEscapeKeydown } from "@radix-ui/react-use-escape-keydown";
-import { jsx as jsx19 } from "react/jsx-runtime";
-var ComposerPrimitiveInput = forwardRef11(
+import { jsx as jsx20 } from "react/jsx-runtime";
+var ComposerPrimitiveInput = forwardRef12(
   ({
     autoFocus = false,
     asChild,
@@ -1353,7 +1556,7 @@ var ComposerPrimitiveInput = forwardRef11(
     });
     const Component = asChild ? Slot : TextareaAutosize;
     const isDisabled = useThread((t) => t.isDisabled) ?? disabledProp ?? false;
-    const textareaRef = useRef3(null);
+    const textareaRef = useRef4(null);
     const ref = useComposedRefs2(forwardedRef, textareaRef);
     useEscapeKeydown((e) => {
       const composer = useComposer.getState();
@@ -1382,13 +1585,13 @@ var ComposerPrimitiveInput = forwardRef11(
         textareaRef.current.value.length
       );
     }, [autoFocusEnabled]);
-    useEffect7(() => focus(), [focus]);
+    useEffect8(() => focus(), [focus]);
     useOnComposerFocus(() => {
       if (type === "new") {
         focus();
       }
     });
-    return /* @__PURE__ */ jsx19(
+    return /* @__PURE__ */ jsx20(
       Component,
       {
         name: "input",
@@ -1409,13 +1612,13 @@ var ComposerPrimitiveInput = forwardRef11(
 ComposerPrimitiveInput.displayName = "ComposerPrimitive.Input";
 
 // src/primitives/composer/ComposerSend.tsx
-import { forwardRef as forwardRef12 } from "react";
+import { forwardRef as forwardRef13 } from "react";
 import { Primitive as Primitive7 } from "@radix-ui/react-primitive";
-import { jsx as jsx20 } from "react/jsx-runtime";
-var ComposerPrimitiveSend = forwardRef12(({ disabled, ...rest }, ref) => {
+import { jsx as jsx21 } from "react/jsx-runtime";
+var ComposerPrimitiveSend = forwardRef13(({ disabled, ...rest }, ref) => {
   const { useComposer } = useComposerContext();
   const hasValue = useComposer((c) => c.isEditing && c.value.length > 0);
-  return /* @__PURE__ */ jsx20(
+  return /* @__PURE__ */ jsx21(
     Primitive7.button,
     {
       type: "submit",
@@ -1466,10 +1669,10 @@ __export(thread_exports, {
 
 // src/primitives/thread/ThreadRoot.tsx
 import { Primitive as Primitive8 } from "@radix-ui/react-primitive";
-import { forwardRef as forwardRef13 } from "react";
-import { jsx as jsx21 } from "react/jsx-runtime";
-var ThreadPrimitiveRoot = forwardRef13((props, ref) => {
-  return /* @__PURE__ */ jsx21(Primitive8.div, { ...props, ref });
+import { forwardRef as forwardRef14 } from "react";
+import { jsx as jsx22 } from "react/jsx-runtime";
+var ThreadPrimitiveRoot = forwardRef14((props, ref) => {
+  return /* @__PURE__ */ jsx22(Primitive8.div, { ...props, ref });
 });
 ThreadPrimitiveRoot.displayName = "ThreadPrimitive.Root";
 
@@ -1495,17 +1698,17 @@ ThreadPrimitiveIf.displayName = "ThreadPrimitive.If";
 // src/primitives/thread/ThreadViewport.tsx
 import { useComposedRefs as useComposedRefs4 } from "@radix-ui/react-compose-refs";
 import { Primitive as Primitive9 } from "@radix-ui/react-primitive";
-import { forwardRef as forwardRef14 } from "react";
+import { forwardRef as forwardRef15 } from "react";
 
 // src/primitive-hooks/thread/useThreadViewportAutoScroll.tsx
 import { useComposedRefs as useComposedRefs3 } from "@radix-ui/react-compose-refs";
-import { useRef as useRef4 } from "react";
+import { useRef as useRef5 } from "react";
 
 // src/utils/hooks/useOnResizeContent.tsx
-import { useCallbackRef as useCallbackRef2 } from "@radix-ui/react-use-callback-ref";
+import { useCallbackRef as useCallbackRef3 } from "@radix-ui/react-use-callback-ref";
 import { useCallback as useCallback16 } from "react";
 var useOnResizeContent = (callback) => {
-  const callbackRef = useCallbackRef2(callback);
+  const callbackRef = useCallbackRef3(callback);
   const refCallback = useCallback16(
     (el) => {
       const resizeObserver = new ResizeObserver(() => {
@@ -1542,12 +1745,12 @@ var useOnResizeContent = (callback) => {
 };
 
 // src/utils/hooks/useOnScrollToBottom.tsx
-import { useCallbackRef as useCallbackRef3 } from "@radix-ui/react-use-callback-ref";
-import { useEffect as useEffect8 } from "react";
+import { useCallbackRef as useCallbackRef4 } from "@radix-ui/react-use-callback-ref";
+import { useEffect as useEffect9 } from "react";
 var useOnScrollToBottom = (callback) => {
-  const callbackRef = useCallbackRef3(callback);
+  const callbackRef = useCallbackRef4(callback);
   const { useViewport } = useThreadContext();
-  useEffect8(() => {
+  useEffect9(() => {
     return useViewport.getState().onScrollToBottom(() => {
       callbackRef();
     });
@@ -1558,10 +1761,10 @@ var useOnScrollToBottom = (callback) => {
 var useThreadViewportAutoScroll = ({
   autoScroll = true
 }) => {
-  const divRef = useRef4(null);
+  const divRef = useRef5(null);
   const { useViewport } = useThreadContext();
-  const lastScrollTop = useRef4(0);
-  const isScrollingToBottomRef = useRef4(false);
+  const lastScrollTop = useRef5(0);
+  const isScrollingToBottomRef = useRef5(false);
   const scrollToBottom = (behavior) => {
     const div = divRef.current;
     if (!div || !autoScroll) return;
@@ -1606,13 +1809,13 @@ var useThreadViewportAutoScroll = ({
 };
 
 // src/primitives/thread/ThreadViewport.tsx
-import { jsx as jsx22 } from "react/jsx-runtime";
-var ThreadPrimitiveViewport = forwardRef14(({ autoScroll, onScroll, children, ...rest }, forwardedRef) => {
+import { jsx as jsx23 } from "react/jsx-runtime";
+var ThreadPrimitiveViewport = forwardRef15(({ autoScroll, onScroll, children, ...rest }, forwardedRef) => {
   const autoScrollRef = useThreadViewportAutoScroll({
     autoScroll
   });
   const ref = useComposedRefs4(forwardedRef, autoScrollRef);
-  return /* @__PURE__ */ jsx22(Primitive9.div, { ...rest, ref, children });
+  return /* @__PURE__ */ jsx23(Primitive9.div, { ...rest, ref, children });
 });
 ThreadPrimitiveViewport.displayName = "ThreadPrimitive.Viewport";
 
@@ -1620,15 +1823,15 @@ ThreadPrimitiveViewport.displayName = "ThreadPrimitive.Viewport";
 import { memo as memo3 } from "react";
 
 // src/context/providers/MessageProvider.tsx
-import { useEffect as useEffect9, useState as useState5 } from "react";
-import { create as create13 } from "zustand";
+import { useEffect as useEffect10, useState as useState7 } from "react";
+import { create as create14 } from "zustand";
 
 // src/context/stores/EditComposer.ts
-import { create as create11 } from "zustand";
+import { create as create12 } from "zustand";
 var makeEditComposerStore = ({
   onEdit,
   onSend
-}) => create11()((set, get, store) => ({
+}) => create12()((set, get, store) => ({
   ...makeBaseComposer(set, get, store),
   canCancel: false,
   isEditing: false,
@@ -1647,8 +1850,8 @@ var makeEditComposerStore = ({
 }));
 
 // src/context/stores/MessageUtils.ts
-import { create as create12 } from "zustand";
-var makeMessageUtilsStore = () => create12((set) => ({
+import { create as create13 } from "zustand";
+var makeMessageUtilsStore = () => create13((set) => ({
   isCopied: false,
   setIsCopied: (value) => {
     set({ isCopied: value });
@@ -1660,7 +1863,7 @@ var makeMessageUtilsStore = () => create12((set) => ({
 }));
 
 // src/context/providers/MessageProvider.tsx
-import { jsx as jsx23 } from "react/jsx-runtime";
+import { jsx as jsx24 } from "react/jsx-runtime";
 var getIsLast = (messages, message) => {
   return messages[messages.length - 1]?.id === message.id;
 };
@@ -1682,8 +1885,8 @@ var getMessageState = (messages, getBranches, useMessage, messageIndex) => {
 };
 var useMessageContext2 = (messageIndex) => {
   const { useThreadMessages, useThreadActions } = useThreadContext();
-  const [context] = useState5(() => {
-    const useMessage = create13(
+  const [context] = useState7(() => {
+    const useMessage = create14(
       () => getMessageState(
         useThreadMessages.getState(),
         useThreadActions.getState().getBranches,
@@ -1720,7 +1923,7 @@ var useMessageContext2 = (messageIndex) => {
     });
     return { useMessage, useMessageUtils, useEditComposer };
   });
-  useEffect9(() => {
+  useEffect10(() => {
     const syncMessage = (thread) => {
       const newState = getMessageState(
         thread,
@@ -1744,11 +1947,11 @@ var MessageProvider = ({
   children
 }) => {
   const context = useMessageContext2(messageIndex);
-  return /* @__PURE__ */ jsx23(MessageContext.Provider, { value: context, children });
+  return /* @__PURE__ */ jsx24(MessageContext.Provider, { value: context, children });
 };
 
 // src/primitives/thread/ThreadMessages.tsx
-import { jsx as jsx24, jsxs as jsxs3 } from "react/jsx-runtime";
+import { jsx as jsx25, jsxs as jsxs3 } from "react/jsx-runtime";
 var DEFAULT_SYSTEM_MESSAGE = () => null;
 var getComponents = (components) => {
   return {
@@ -1765,11 +1968,11 @@ var ThreadMessageImpl = ({
   const { UserMessage: UserMessage2, EditComposer: EditComposer2, AssistantMessage: AssistantMessage2, SystemMessage: SystemMessage2 } = getComponents(components);
   return /* @__PURE__ */ jsxs3(MessageProvider, { messageIndex, children: [
     /* @__PURE__ */ jsxs3(MessagePrimitiveIf, { user: true, children: [
-      /* @__PURE__ */ jsx24(ComposerPrimitiveIf, { editing: false, children: /* @__PURE__ */ jsx24(UserMessage2, {}) }),
-      /* @__PURE__ */ jsx24(ComposerPrimitiveIf, { editing: true, children: /* @__PURE__ */ jsx24(EditComposer2, {}) })
+      /* @__PURE__ */ jsx25(ComposerPrimitiveIf, { editing: false, children: /* @__PURE__ */ jsx25(UserMessage2, {}) }),
+      /* @__PURE__ */ jsx25(ComposerPrimitiveIf, { editing: true, children: /* @__PURE__ */ jsx25(EditComposer2, {}) })
     ] }),
-    /* @__PURE__ */ jsx24(MessagePrimitiveIf, { assistant: true, children: /* @__PURE__ */ jsx24(AssistantMessage2, {}) }),
-    /* @__PURE__ */ jsx24(MessagePrimitiveIf, { system: true, children: /* @__PURE__ */ jsx24(SystemMessage2, {}) })
+    /* @__PURE__ */ jsx25(MessagePrimitiveIf, { assistant: true, children: /* @__PURE__ */ jsx25(AssistantMessage2, {}) }),
+    /* @__PURE__ */ jsx25(MessagePrimitiveIf, { system: true, children: /* @__PURE__ */ jsx25(SystemMessage2, {}) })
   ] });
 };
 var ThreadMessage = memo3(
@@ -1784,7 +1987,7 @@ var ThreadPrimitiveMessagesImpl = ({
   if (messagesLength === 0) return null;
   return new Array(messagesLength).fill(null).map((_, idx) => {
     const messageIndex = idx;
-    return /* @__PURE__ */ jsx24(
+    return /* @__PURE__ */ jsx25(
       ThreadMessage,
       {
         messageIndex,
@@ -1814,7 +2017,359 @@ var ThreadPrimitiveSuggestion = createActionButton(
 );
 
 // src/runtimes/local/useLocalRuntime.tsx
-import { useInsertionEffect as useInsertionEffect3, useState as useState7 } from "react";
+import { useInsertionEffect as useInsertionEffect3, useState as useState9 } from "react";
+
+// src/runtimes/core/BaseAssistantRuntime.tsx
+var BaseAssistantRuntime = class {
+  constructor(_thread) {
+    this._thread = _thread;
+    this._thread = _thread;
+  }
+  get thread() {
+    return this._thread;
+  }
+  set thread(thread) {
+    this._thread = thread;
+    this.subscriptionHandler();
+  }
+  _subscriptions = /* @__PURE__ */ new Set();
+  subscribe(callback) {
+    this._subscriptions.add(callback);
+    return () => this._subscriptions.delete(callback);
+  }
+  subscriptionHandler = () => {
+    for (const callback of this._subscriptions) callback();
+  };
+};
+
+// src/internal.ts
+var internal_exports = {};
+__export(internal_exports, {
+  BaseAssistantRuntime: () => BaseAssistantRuntime,
+  MessageRepository: () => MessageRepository,
+  ProxyConfigProvider: () => ProxyConfigProvider,
+  TooltipIconButton: () => TooltipIconButton,
+  generateId: () => generateId,
+  useSmooth: () => useSmooth,
+  useSmoothStatus: () => useSmoothStatus,
+  withSmoothContextProvider: () => withSmoothContextProvider
+});
+
+// src/utils/idUtils.tsx
+import { customAlphabet } from "nanoid/non-secure";
+var generateId = customAlphabet(
+  "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz",
+  7
+);
+var optimisticPrefix = "__optimistic__";
+var generateOptimisticId = () => `${optimisticPrefix}${generateId()}`;
+
+// src/runtimes/edge/converters/fromCoreMessage.ts
+var fromCoreMessages = (message) => {
+  return message.map((message2) => fromCoreMessage(message2));
+};
+var fromCoreMessage = (message, {
+  id = generateId(),
+  status = { type: "complete", reason: "unknown" }
+} = {}) => {
+  const commonProps = {
+    id,
+    createdAt: /* @__PURE__ */ new Date()
+  };
+  const role = message.role;
+  switch (role) {
+    case "assistant":
+      return {
+        ...commonProps,
+        role,
+        content: message.content.map((part) => {
+          if (part.type === "tool-call") {
+            return {
+              ...part,
+              argsText: JSON.stringify(part.args)
+            };
+          }
+          return part;
+        }),
+        status
+      };
+    case "user":
+      return {
+        ...commonProps,
+        role,
+        content: message.content
+      };
+    case "system":
+      return {
+        ...commonProps,
+        role,
+        content: message.content
+      };
+    default: {
+      const unsupportedRole = role;
+      throw new Error(`Unknown message role: ${unsupportedRole}`);
+    }
+  }
+};
+
+// src/runtimes/utils/MessageRepository.tsx
+var findHead = (message) => {
+  if (message.next) return findHead(message.next);
+  return message;
+};
+var MessageRepository = class {
+  messages = /* @__PURE__ */ new Map();
+  // message_id -> item
+  head = null;
+  root = {
+    children: []
+  };
+  performOp(newParent, child, operation) {
+    const parentOrRoot = child.prev ?? this.root;
+    const newParentOrRoot = newParent ?? this.root;
+    if (operation === "relink" && parentOrRoot === newParentOrRoot) return;
+    if (operation !== "link") {
+      parentOrRoot.children = parentOrRoot.children.filter(
+        (m) => m !== child.current.id
+      );
+      if (child.prev?.next === child) {
+        const fallbackId = child.prev.children.at(-1);
+        const fallback = fallbackId ? this.messages.get(fallbackId) : null;
+        if (fallback === void 0) {
+          throw new Error(
+            "MessageRepository(performOp/cut): Fallback sibling message not found. This is likely an internal bug in assistant-ui."
+          );
+        }
+        child.prev.next = fallback;
+      }
+    }
+    if (operation !== "cut") {
+      newParentOrRoot.children = [
+        ...newParentOrRoot.children,
+        child.current.id
+      ];
+      if (newParent && (findHead(child) === this.head || newParent.next === null)) {
+        newParent.next = child;
+      }
+      child.prev = newParent;
+    }
+  }
+  getMessages() {
+    const messages = new Array(this.head?.level ?? 0);
+    for (let current = this.head; current; current = current.prev) {
+      messages[current.level] = current.current;
+    }
+    return messages;
+  }
+  addOrUpdateMessage(parentId, message) {
+    const existingItem = this.messages.get(message.id);
+    const prev = parentId ? this.messages.get(parentId) : null;
+    if (prev === void 0)
+      throw new Error(
+        "MessageRepository(addOrUpdateMessage): Parent message not found. This is likely an internal bug in assistant-ui."
+      );
+    if (existingItem) {
+      existingItem.current = message;
+      this.performOp(prev, existingItem, "relink");
+      return;
+    }
+    const newItem = {
+      prev,
+      current: message,
+      next: null,
+      children: [],
+      level: prev ? prev.level + 1 : 0
+    };
+    this.messages.set(message.id, newItem);
+    this.performOp(prev, newItem, "link");
+    if (this.head === prev) {
+      this.head = newItem;
+    }
+  }
+  getMessage(messageId) {
+    const message = this.messages.get(messageId);
+    if (!message)
+      throw new Error(
+        "MessageRepository(updateMessage): Message not found. This is likely an internal bug in assistant-ui."
+      );
+    return {
+      parentId: message.prev?.current.id ?? null,
+      message: message.current
+    };
+  }
+  appendOptimisticMessage(parentId, message) {
+    let optimisticId;
+    do {
+      optimisticId = generateOptimisticId();
+    } while (this.messages.has(optimisticId));
+    this.addOrUpdateMessage(
+      parentId,
+      fromCoreMessage(message, {
+        id: optimisticId,
+        status: { type: "running" }
+      })
+    );
+    return optimisticId;
+  }
+  deleteMessage(messageId, replacementId) {
+    const message = this.messages.get(messageId);
+    if (!message)
+      throw new Error(
+        "MessageRepository(deleteMessage): Optimistic message not found. This is likely an internal bug in assistant-ui."
+      );
+    const replacement = replacementId === void 0 ? message.prev : replacementId === null ? null : this.messages.get(replacementId);
+    if (replacement === void 0)
+      throw new Error(
+        "MessageRepository(deleteMessage): Replacement not found. This is likely an internal bug in assistant-ui."
+      );
+    for (const child of message.children) {
+      const childMessage = this.messages.get(child);
+      if (!childMessage)
+        throw new Error(
+          "MessageRepository(deleteMessage): Child message not found. This is likely an internal bug in assistant-ui."
+        );
+      this.performOp(replacement, childMessage, "relink");
+    }
+    this.performOp(null, message, "cut");
+    this.messages.delete(messageId);
+    if (this.head === message) {
+      this.head = replacement ? findHead(replacement) : null;
+    }
+  }
+  getBranches(messageId) {
+    const message = this.messages.get(messageId);
+    if (!message)
+      throw new Error(
+        "MessageRepository(getBranches): Message not found. This is likely an internal bug in assistant-ui."
+      );
+    const { children } = message.prev ?? this.root;
+    return children;
+  }
+  switchToBranch(messageId) {
+    const message = this.messages.get(messageId);
+    if (!message)
+      throw new Error(
+        "MessageRepository(switchToBranch): Branch not found. This is likely an internal bug in assistant-ui."
+      );
+    if (message.prev) {
+      message.prev.next = message;
+    }
+    this.head = findHead(message);
+  }
+  resetHead(messageId) {
+    if (messageId === null) {
+      this.head = null;
+      return;
+    }
+    const message = this.messages.get(messageId);
+    if (!message)
+      throw new Error(
+        "MessageRepository(resetHead): Branch not found. This is likely an internal bug in assistant-ui."
+      );
+    this.head = message;
+    for (let current = message; current; current = current.prev) {
+      if (current.prev) {
+        current.prev.next = current;
+      }
+    }
+  }
+};
+
+// src/ui/base/tooltip-icon-button.tsx
+import { forwardRef as forwardRef18 } from "react";
+
+// src/ui/base/tooltip.tsx
+import * as TooltipPrimitive from "@radix-ui/react-tooltip";
+
+// src/ui/utils/withDefaults.tsx
+import {
+  forwardRef as forwardRef16
+} from "react";
+import classNames from "classnames";
+import { jsx as jsx26 } from "react/jsx-runtime";
+var withDefaultProps = ({
+  className,
+  ...defaultProps
+}) => ({ className: classNameProp, ...props }) => {
+  return {
+    className: classNames(className, classNameProp),
+    ...defaultProps,
+    ...props
+  };
+};
+var withDefaults = (Component, defaultProps) => {
+  const getProps = withDefaultProps(defaultProps);
+  const WithDefaults = forwardRef16(
+    (props, ref) => {
+      const ComponentAsAny = Component;
+      return /* @__PURE__ */ jsx26(ComponentAsAny, { ...getProps(props), ref });
+    }
+  );
+  WithDefaults.displayName = "withDefaults(" + (typeof Component === "string" ? Component : Component.displayName) + ")";
+  return WithDefaults;
+};
+
+// src/ui/base/tooltip.tsx
+import { jsx as jsx27 } from "react/jsx-runtime";
+var Tooltip = (props) => {
+  return /* @__PURE__ */ jsx27(TooltipPrimitive.Provider, { children: /* @__PURE__ */ jsx27(TooltipPrimitive.Root, { ...props }) });
+};
+Tooltip.displayName = "Tooltip";
+var TooltipTrigger = TooltipPrimitive.Trigger;
+var TooltipContent = withDefaults(TooltipPrimitive.Content, {
+  sideOffset: 4,
+  className: "aui-tooltip-content"
+});
+TooltipContent.displayName = "TooltipContent";
+
+// src/ui/base/button.tsx
+import { cva } from "class-variance-authority";
+import { Primitive as Primitive10 } from "@radix-ui/react-primitive";
+import { forwardRef as forwardRef17 } from "react";
+import { jsx as jsx28 } from "react/jsx-runtime";
+var buttonVariants = cva("aui-button", {
+  variants: {
+    variant: {
+      default: "aui-button-primary",
+      outline: "aui-button-outline",
+      ghost: "aui-button-ghost"
+    },
+    size: {
+      default: "aui-button-medium",
+      icon: "aui-button-icon"
+    }
+  },
+  defaultVariants: {
+    variant: "default",
+    size: "default"
+  }
+});
+var Button = forwardRef17(
+  ({ className, variant, size, ...props }, ref) => {
+    return /* @__PURE__ */ jsx28(
+      Primitive10.button,
+      {
+        className: buttonVariants({ variant, size, className }),
+        ...props,
+        ref
+      }
+    );
+  }
+);
+Button.displayName = "Button";
+
+// src/ui/base/tooltip-icon-button.tsx
+import { jsx as jsx29, jsxs as jsxs4 } from "react/jsx-runtime";
+var TooltipIconButton = forwardRef18(({ children, tooltip, side = "bottom", ...rest }, ref) => {
+  return /* @__PURE__ */ jsxs4(Tooltip, { children: [
+    /* @__PURE__ */ jsx29(TooltipTrigger, { asChild: true, children: /* @__PURE__ */ jsxs4(Button, { variant: "ghost", size: "icon", ...rest, ref, children: [
+      children,
+      /* @__PURE__ */ jsx29("span", { className: "aui-sr-only", children: tooltip })
+    ] }) }),
+    /* @__PURE__ */ jsx29(TooltipContent, { side, children: tooltip })
+  ] });
+});
+TooltipIconButton.displayName = "TooltipIconButton";
 
 // src/runtimes/edge/converters/fromLanguageModelMessages.ts
 var fromLanguageModelMessages = (lm, { mergeRoundtrips }) => {
@@ -1934,7 +2489,7 @@ var fromLanguageModelTools = (tools) => {
 };
 
 // src/runtimes/edge/useEdgeRuntime.ts
-import { useState as useState6 } from "react";
+import { useState as useState8 } from "react";
 
 // src/runtimes/edge/streams/assistantDecoderStream.ts
 function assistantDecoderStream() {
@@ -2091,7 +2646,7 @@ var useEdgeRuntime = ({
   initialMessages,
   ...options
 }) => {
-  const [adapter] = useState6(() => new EdgeChatAdapter(options));
+  const [adapter] = useState8(() => new EdgeChatAdapter(options));
   return useLocalRuntime(adapter, { initialMessages });
 };
 
@@ -2305,7 +2860,7 @@ var LocalRuntime = class extends BaseAssistantRuntime {
 
 // src/runtimes/local/useLocalRuntime.tsx
 var useLocalRuntime = (adapter, options) => {
-  const [runtime] = useState7(() => new LocalRuntime(adapter, options));
+  const [runtime] = useState9(() => new LocalRuntime(adapter, options));
   useInsertionEffect3(() => {
     runtime.adapter = adapter;
   });
@@ -2313,7 +2868,7 @@ var useLocalRuntime = (adapter, options) => {
 };
 
 // src/runtimes/external-store/ExternalStoreThreadRuntime.tsx
-import { create as create14 } from "zustand";
+import { create as create15 } from "zustand";
 
 // src/runtimes/external-store/getExternalStoreMessage.tsx
 var symbolInnerMessage = Symbol("innerMessage");
@@ -2322,7 +2877,7 @@ var getExternalStoreMessage = (message) => {
 };
 
 // src/runtimes/external-store/useExternalStoreSync.tsx
-import { useEffect as useEffect10, useInsertionEffect as useInsertionEffect4, useMemo as useMemo2, useRef as useRef5 } from "react";
+import { useEffect as useEffect11, useInsertionEffect as useInsertionEffect4, useMemo as useMemo4, useRef as useRef6 } from "react";
 
 // src/runtimes/external-store/ThreadMessageConverter.ts
 var ThreadMessageConverter = class {
@@ -2416,11 +2971,11 @@ var fromThreadMessageLike = (like, fallbackId, fallbackStatus) => {
 
 // src/runtimes/external-store/useExternalStoreSync.tsx
 var useExternalStoreSync = (adapter, updateData) => {
-  const adapterRef = useRef5(adapter);
+  const adapterRef = useRef6(adapter);
   useInsertionEffect4(() => {
     adapterRef.current = adapter;
   });
-  const [converter, convertCallback] = useMemo2(() => {
+  const [converter, convertCallback] = useMemo4(() => {
     const converter2 = adapter.convertMessage ?? ((m) => m);
     const convertCallback2 = (cache, m, idx) => {
       const autoStatus = getAutoStatus(
@@ -2439,7 +2994,7 @@ var useExternalStoreSync = (adapter, updateData) => {
     };
     return [new ThreadMessageConverter(), convertCallback2];
   }, [adapter.convertMessage]);
-  useEffect10(() => {
+  useEffect11(() => {
     updateData(
       adapter.isDisabled ?? false,
       adapter.isRunning ?? false,
@@ -2467,7 +3022,7 @@ var ExternalStoreThreadRuntime = class {
       store.isRunning ?? false,
       store.messages
     );
-    this.useStore = create14(() => ({
+    this.useStore = create15(() => ({
       store
     }));
   }
@@ -2608,13 +3163,13 @@ var ExternalStoreRuntime = class extends BaseAssistantRuntime {
 };
 
 // src/runtimes/external-store/useExternalStoreRuntime.tsx
-import { useEffect as useEffect11, useInsertionEffect as useInsertionEffect5, useState as useState8 } from "react";
+import { useEffect as useEffect12, useInsertionEffect as useInsertionEffect5, useState as useState10 } from "react";
 var useExternalStoreRuntime = (store) => {
-  const [runtime] = useState8(() => new ExternalStoreRuntime(store));
+  const [runtime] = useState10(() => new ExternalStoreRuntime(store));
   useInsertionEffect5(() => {
     runtime.store = store;
   });
-  useEffect11(() => {
+  useEffect12(() => {
     runtime.onStoreUpdated();
   });
   return runtime;
@@ -2622,34 +3177,34 @@ var useExternalStoreRuntime = (store) => {
 
 // src/ui/thread-config.tsx
 import {
-  createContext,
-  useContext
+  createContext as createContext6,
+  useContext as useContext6
 } from "react";
-import { Fragment as Fragment3, jsx as jsx25 } from "react/jsx-runtime";
-var ThreadConfigContext = createContext({});
+import { Fragment as Fragment3, jsx as jsx30 } from "react/jsx-runtime";
+var ThreadConfigContext = createContext6({});
 var useThreadConfig = () => {
-  return useContext(ThreadConfigContext);
+  return useContext6(ThreadConfigContext);
 };
 var ThreadConfigProvider = ({
   children,
   config
 }) => {
   const assistant = useAssistantContext({ optional: true });
-  const configProvider = config && Object.keys(config ?? {}).length > 0 ? /* @__PURE__ */ jsx25(ThreadConfigContext.Provider, { value: config, children }) : /* @__PURE__ */ jsx25(Fragment3, { children });
+  const configProvider = config && Object.keys(config ?? {}).length > 0 ? /* @__PURE__ */ jsx30(ThreadConfigContext.Provider, { value: config, children }) : /* @__PURE__ */ jsx30(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__ */ jsx25(AssistantRuntimeProvider, { runtime: config.runtime, children: configProvider });
+  return /* @__PURE__ */ jsx30(AssistantRuntimeProvider, { runtime: config.runtime, children: configProvider });
 };
 ThreadConfigProvider.displayName = "ThreadConfigProvider";
 
 // src/ui/assistant-action-bar.tsx
-import { forwardRef as forwardRef15 } from "react";
+import { forwardRef as forwardRef19 } from "react";
 import { CheckIcon, CopyIcon, RefreshCwIcon } from "lucide-react";
-import { Fragment as Fragment4, jsx as jsx26, jsxs as jsxs4 } from "react/jsx-runtime";
+import { Fragment as Fragment4, jsx as jsx31, jsxs as jsxs5 } from "react/jsx-runtime";
 var useAllowCopy = () => {
   const { assistantMessage: { allowCopy = true } = {} } = useThreadConfig();
   const { useThreadActions } = useThreadContext();
@@ -2666,15 +3221,15 @@ var AssistantActionBar = () => {
   const allowCopy = useAllowCopy();
   const allowReload = useAllowReload();
   if (!allowCopy && !allowReload) return null;
-  return /* @__PURE__ */ jsxs4(
+  return /* @__PURE__ */ jsxs5(
     AssistantActionBarRoot,
     {
       hideWhenRunning: true,
       autohide: "not-last",
       autohideFloat: "single-branch",
       children: [
-        /* @__PURE__ */ jsx26(AssistantActionBarCopy, {}),
-        /* @__PURE__ */ jsx26(AssistantActionBarReload, {})
+        /* @__PURE__ */ jsx31(AssistantActionBarCopy, {}),
+        /* @__PURE__ */ jsx31(AssistantActionBarReload, {})
       ]
     }
   );
@@ -2684,7 +3239,7 @@ var AssistantActionBarRoot = withDefaults(actionBar_exports.Root, {
   className: "aui-assistant-action-bar-root"
 });
 AssistantActionBarRoot.displayName = "AssistantActionBarRoot";
-var AssistantActionBarCopy = forwardRef15((props, ref) => {
+var AssistantActionBarCopy = forwardRef19((props, ref) => {
   const {
     strings: {
       assistantMessage: { reload: { tooltip = "Copy" } = {} } = {}
@@ -2692,13 +3247,13 @@ var AssistantActionBarCopy = forwardRef15((props, ref) => {
   } = useThreadConfig();
   const allowCopy = useAllowCopy();
   if (!allowCopy) return null;
-  return /* @__PURE__ */ jsx26(actionBar_exports.Copy, { asChild: true, children: /* @__PURE__ */ jsx26(TooltipIconButton, { tooltip, ...props, ref, children: props.children ?? /* @__PURE__ */ jsxs4(Fragment4, { children: [
-    /* @__PURE__ */ jsx26(message_exports.If, { copied: true, children: /* @__PURE__ */ jsx26(CheckIcon, {}) }),
-    /* @__PURE__ */ jsx26(message_exports.If, { copied: false, children: /* @__PURE__ */ jsx26(CopyIcon, {}) })
+  return /* @__PURE__ */ jsx31(actionBar_exports.Copy, { asChild: true, children: /* @__PURE__ */ jsx31(TooltipIconButton, { tooltip, ...props, ref, children: props.children ?? /* @__PURE__ */ jsxs5(Fragment4, { children: [
+    /* @__PURE__ */ jsx31(message_exports.If, { copied: true, children: /* @__PURE__ */ jsx31(CheckIcon, {}) }),
+    /* @__PURE__ */ jsx31(message_exports.If, { copied: false, children: /* @__PURE__ */ jsx31(CopyIcon, {}) })
   ] }) }) });
 });
 AssistantActionBarCopy.displayName = "AssistantActionBarCopy";
-var AssistantActionBarReload = forwardRef15((props, ref) => {
+var AssistantActionBarReload = forwardRef19((props, ref) => {
   const {
     strings: {
       assistantMessage: { reload: { tooltip = "Refresh" } = {} } = {}
@@ -2706,7 +3261,7 @@ var AssistantActionBarReload = forwardRef15((props, ref) => {
   } = useThreadConfig();
   const allowReload = useAllowReload();
   if (!allowReload) return null;
-  return /* @__PURE__ */ jsx26(actionBar_exports.Reload, { asChild: true, children: /* @__PURE__ */ jsx26(TooltipIconButton, { tooltip, ...props, ref, children: /* @__PURE__ */ jsx26(RefreshCwIcon, {}) }) });
+  return /* @__PURE__ */ jsx31(actionBar_exports.Reload, { asChild: true, children: /* @__PURE__ */ jsx31(TooltipIconButton, { tooltip, ...props, ref, children: /* @__PURE__ */ jsx31(RefreshCwIcon, {}) }) });
 });
 AssistantActionBarReload.displayName = "AssistantActionBarReload";
 var exports = {
@@ -2720,12 +3275,12 @@ var assistant_action_bar_default = Object.assign(
 );
 
 // src/ui/assistant-message.tsx
-import { forwardRef as forwardRef17 } from "react";
+import { forwardRef as forwardRef21 } from "react";
 
 // src/ui/branch-picker.tsx
-import { forwardRef as forwardRef16 } from "react";
+import { forwardRef as forwardRef20 } from "react";
 import { ChevronLeftIcon, ChevronRightIcon } from "lucide-react";
-import { jsx as jsx27, jsxs as jsxs5 } from "react/jsx-runtime";
+import { jsx as jsx32, jsxs as jsxs6 } from "react/jsx-runtime";
 var useAllowBranchPicker = () => {
   const { branchPicker: { allowBranchPicker = true } = {} } = useThreadConfig();
   const { useThreadActions } = useThreadContext();
@@ -2735,10 +3290,10 @@ var useAllowBranchPicker = () => {
 var BranchPicker = () => {
   const allowBranchPicker = useAllowBranchPicker();
   if (!allowBranchPicker) return null;
-  return /* @__PURE__ */ jsxs5(BranchPickerRoot, { hideWhenSingleBranch: true, children: [
-    /* @__PURE__ */ jsx27(BranchPickerPrevious2, {}),
-    /* @__PURE__ */ jsx27(BranchPickerState, {}),
-    /* @__PURE__ */ jsx27(BranchPickerNext, {})
+  return /* @__PURE__ */ jsxs6(BranchPickerRoot, { hideWhenSingleBranch: true, children: [
+    /* @__PURE__ */ jsx32(BranchPickerPrevious2, {}),
+    /* @__PURE__ */ jsx32(BranchPickerState, {}),
+    /* @__PURE__ */ jsx32(BranchPickerNext, {})
   ] });
 };
 BranchPicker.displayName = "BranchPicker";
@@ -2746,31 +3301,31 @@ var BranchPickerRoot = withDefaults(branchPicker_exports.Root, {
   className: "aui-branch-picker-root"
 });
 BranchPickerRoot.displayName = "BranchPickerRoot";
-var BranchPickerPrevious2 = forwardRef16((props, ref) => {
+var BranchPickerPrevious2 = forwardRef20((props, ref) => {
   const {
     strings: {
       branchPicker: { previous: { tooltip = "Previous" } = {} } = {}
     } = {}
   } = useThreadConfig();
-  return /* @__PURE__ */ jsx27(branchPicker_exports.Previous, { asChild: true, children: /* @__PURE__ */ jsx27(TooltipIconButton, { tooltip, ...props, ref, children: props.children ?? /* @__PURE__ */ jsx27(ChevronLeftIcon, {}) }) });
+  return /* @__PURE__ */ jsx32(branchPicker_exports.Previous, { asChild: true, children: /* @__PURE__ */ jsx32(TooltipIconButton, { tooltip, ...props, ref, children: props.children ?? /* @__PURE__ */ jsx32(ChevronLeftIcon, {}) }) });
 });
 BranchPickerPrevious2.displayName = "BranchPickerPrevious";
 var BranchPickerStateWrapper = withDefaults("span", {
   className: "aui-branch-picker-state"
 });
-var BranchPickerState = forwardRef16((props, ref) => {
-  return /* @__PURE__ */ jsxs5(BranchPickerStateWrapper, { ...props, ref, children: [
-    /* @__PURE__ */ jsx27(branchPicker_exports.Number, {}),
+var BranchPickerState = forwardRef20((props, ref) => {
+  return /* @__PURE__ */ jsxs6(BranchPickerStateWrapper, { ...props, ref, children: [
+    /* @__PURE__ */ jsx32(branchPicker_exports.Number, {}),
     " / ",
-    /* @__PURE__ */ jsx27(branchPicker_exports.Count, {})
+    /* @__PURE__ */ jsx32(branchPicker_exports.Count, {})
   ] });
 });
 BranchPickerState.displayName = "BranchPickerState";
-var BranchPickerNext = forwardRef16((props, ref) => {
+var BranchPickerNext = forwardRef20((props, ref) => {
   const {
     strings: { branchPicker: { next: { tooltip = "Next" } = {} } = {} } = {}
   } = useThreadConfig();
-  return /* @__PURE__ */ jsx27(branchPicker_exports.Next, { asChild: true, children: /* @__PURE__ */ jsx27(TooltipIconButton, { tooltip, ...props, ref, children: props.children ?? /* @__PURE__ */ jsx27(ChevronRightIcon, {}) }) });
+  return /* @__PURE__ */ jsx32(branchPicker_exports.Next, { asChild: true, children: /* @__PURE__ */ jsx32(TooltipIconButton, { tooltip, ...props, ref, children: props.children ?? /* @__PURE__ */ jsx32(ChevronRightIcon, {}) }) });
 });
 BranchPickerNext.displayName = "BranchPickerNext";
 var exports2 = {
@@ -2782,12 +3337,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 jsx28, jsxs as jsxs6 } from "react/jsx-runtime";
+import { jsx as jsx33, jsxs as jsxs7 } from "react/jsx-runtime";
 var Avatar = ({ src, alt, fallback }) => {
   if (src == null && fallback == null) return null;
-  return /* @__PURE__ */ jsxs6(AvatarRoot, { children: [
-    src != null && /* @__PURE__ */ jsx28(AvatarImage, { src, alt }),
-    fallback != null && /* @__PURE__ */ jsx28(AvatarFallback, { children: fallback })
+  return /* @__PURE__ */ jsxs7(AvatarRoot, { children: [
+    src != null && /* @__PURE__ */ jsx33(AvatarImage, { src, alt }),
+    fallback != null && /* @__PURE__ */ jsx33(AvatarFallback, { children: fallback })
   ] });
 };
 Avatar.displayName = "Avatar";
@@ -2805,14 +3360,14 @@ var AvatarFallback = withDefaults(AvatarPrimitive.Fallback, {
 AvatarFallback.displayName = "AvatarFallback";
 
 // src/ui/content-part.tsx
-import classNames from "classnames";
-import { jsx as jsx29 } from "react/jsx-runtime";
+import classNames2 from "classnames";
+import { jsx as jsx34 } from "react/jsx-runtime";
 var Text = () => {
   const status = useSmoothStatus();
-  return /* @__PURE__ */ jsx29(
+  return /* @__PURE__ */ jsx34(
     contentPart_exports.Text,
     {
-      className: classNames(
+      className: classNames2(
         "aui-text",
         status.type === "running" && "aui-text-running"
       ),
@@ -2824,19 +3379,19 @@ var exports3 = { Text: withSmoothContextProvider(Text) };
 var content_part_default = exports3;
 
 // src/ui/assistant-message.tsx
-import { jsx as jsx30, jsxs as jsxs7 } from "react/jsx-runtime";
+import { jsx as jsx35, jsxs as jsxs8 } from "react/jsx-runtime";
 var AssistantMessage = () => {
-  return /* @__PURE__ */ jsxs7(AssistantMessageRoot, { children: [
-    /* @__PURE__ */ jsx30(AssistantMessageAvatar, {}),
-    /* @__PURE__ */ jsx30(AssistantMessageContent, {}),
-    /* @__PURE__ */ jsx30(branch_picker_default, {}),
-    /* @__PURE__ */ jsx30(assistant_action_bar_default, {})
+  return /* @__PURE__ */ jsxs8(AssistantMessageRoot, { children: [
+    /* @__PURE__ */ jsx35(AssistantMessageAvatar, {}),
+    /* @__PURE__ */ jsx35(AssistantMessageContent, {}),
+    /* @__PURE__ */ jsx35(branch_picker_default, {}),
+    /* @__PURE__ */ jsx35(assistant_action_bar_default, {})
   ] });
 };
 AssistantMessage.displayName = "AssistantMessage";
 var AssistantMessageAvatar = () => {
   const { assistantAvatar: avatar = { fallback: "A" } } = useThreadConfig();
-  return /* @__PURE__ */ jsx30(Avatar, { ...avatar });
+  return /* @__PURE__ */ jsx35(Avatar, { ...avatar });
 };
 var AssistantMessageRoot = withDefaults(message_exports.Root, {
   className: "aui-assistant-message-root"
@@ -2845,9 +3400,9 @@ AssistantMessageRoot.displayName = "AssistantMessageRoot";
 var AssistantMessageContentWrapper = withDefaults("div", {
   className: "aui-assistant-message-content"
 });
-var AssistantMessageContent = forwardRef17(({ components: componentsProp, ...rest }, ref) => {
+var AssistantMessageContent = forwardRef21(({ components: componentsProp, ...rest }, ref) => {
   const { assistantMessage: { components = {} } = {} } = useThreadConfig();
-  return /* @__PURE__ */ jsx30(AssistantMessageContentWrapper, { ...rest, ref, children: /* @__PURE__ */ jsx30(
+  return /* @__PURE__ */ jsx35(AssistantMessageContentWrapper, { ...rest, ref, children: /* @__PURE__ */ jsx35(
     message_exports.Content,
     {
       components: {
@@ -2869,38 +3424,38 @@ var assistant_message_default = Object.assign(
 );
 
 // src/ui/assistant-modal.tsx
-import { forwardRef as forwardRef24 } from "react";
+import { forwardRef as forwardRef28 } from "react";
 import { BotIcon, ChevronDownIcon } from "lucide-react";
 
 // src/ui/thread.tsx
-import { forwardRef as forwardRef23 } from "react";
+import { forwardRef as forwardRef27 } from "react";
 import { ArrowDownIcon } from "lucide-react";
 
 // src/ui/composer.tsx
-import { forwardRef as forwardRef18 } from "react";
+import { forwardRef as forwardRef22 } from "react";
 import { SendHorizontalIcon } from "lucide-react";
 
 // src/ui/base/CircleStopIcon.tsx
-import { jsx as jsx31 } from "react/jsx-runtime";
+import { jsx as jsx36 } from "react/jsx-runtime";
 var CircleStopIcon = () => {
-  return /* @__PURE__ */ jsx31(
+  return /* @__PURE__ */ jsx36(
     "svg",
     {
       xmlns: "http://www.w3.org/2000/svg",
       viewBox: "0 0 16 16",
       fill: "currentColor",
-      children: /* @__PURE__ */ jsx31("rect", { width: "10", height: "10", x: "3", y: "3", rx: "2" })
+      children: /* @__PURE__ */ jsx36("rect", { width: "10", height: "10", x: "3", y: "3", rx: "2" })
     }
   );
 };
 CircleStopIcon.displayName = "CircleStopIcon";
 
 // src/ui/composer.tsx
-import { Fragment as Fragment5, jsx as jsx32, jsxs as jsxs8 } from "react/jsx-runtime";
+import { Fragment as Fragment5, jsx as jsx37, jsxs as jsxs9 } from "react/jsx-runtime";
 var Composer = () => {
-  return /* @__PURE__ */ jsxs8(ComposerRoot, { children: [
-    /* @__PURE__ */ jsx32(ComposerInput, { autoFocus: true }),
-    /* @__PURE__ */ jsx32(ComposerAction, {})
+  return /* @__PURE__ */ jsxs9(ComposerRoot, { children: [
+    /* @__PURE__ */ jsx37(ComposerInput, { autoFocus: true }),
+    /* @__PURE__ */ jsx37(ComposerAction, {})
   ] });
 };
 Composer.displayName = "Composer";
@@ -2913,14 +3468,14 @@ var ComposerInputStyled = withDefaults(composer_exports.Input, {
   autoFocus: true,
   className: "aui-composer-input"
 });
-var ComposerInput = forwardRef18(
+var ComposerInput = forwardRef22(
   (props, ref) => {
     const {
       strings: {
         composer: { input: { placeholder = "Write a message..." } = {} } = {}
       } = {}
     } = useThreadConfig();
-    return /* @__PURE__ */ jsx32(ComposerInputStyled, { placeholder, ...props, ref });
+    return /* @__PURE__ */ jsx37(ComposerInputStyled, { placeholder, ...props, ref });
   }
 );
 ComposerInput.displayName = "ComposerInput";
@@ -2931,10 +3486,10 @@ var useAllowCancel = () => {
 };
 var ComposerAction = () => {
   const allowCancel = useAllowCancel();
-  if (!allowCancel) return /* @__PURE__ */ jsx32(ComposerSend, {});
-  return /* @__PURE__ */ jsxs8(Fragment5, { children: [
-    /* @__PURE__ */ jsx32(thread_exports.If, { running: false, children: /* @__PURE__ */ jsx32(ComposerSend, {}) }),
-    /* @__PURE__ */ jsx32(thread_exports.If, { running: true, children: /* @__PURE__ */ jsx32(ComposerCancel, {}) })
+  if (!allowCancel) return /* @__PURE__ */ jsx37(ComposerSend, {});
+  return /* @__PURE__ */ jsxs9(Fragment5, { children: [
+    /* @__PURE__ */ jsx37(thread_exports.If, { running: false, children: /* @__PURE__ */ jsx37(ComposerSend, {}) }),
+    /* @__PURE__ */ jsx37(thread_exports.If, { running: true, children: /* @__PURE__ */ jsx37(ComposerCancel, {}) })
   ] });
 };
 ComposerAction.displayName = "ComposerAction";
@@ -2942,22 +3497,22 @@ var ComposerSendButton = withDefaults(TooltipIconButton, {
   variant: "default",
   className: "aui-composer-send"
 });
-var ComposerSend = forwardRef18((props, ref) => {
+var ComposerSend = forwardRef22((props, ref) => {
   const {
     strings: { composer: { send: { tooltip = "Send" } = {} } = {} } = {}
   } = useThreadConfig();
-  return /* @__PURE__ */ jsx32(composer_exports.Send, { asChild: true, children: /* @__PURE__ */ jsx32(ComposerSendButton, { tooltip, ...props, ref, children: props.children ?? /* @__PURE__ */ jsx32(SendHorizontalIcon, {}) }) });
+  return /* @__PURE__ */ jsx37(composer_exports.Send, { asChild: true, children: /* @__PURE__ */ jsx37(ComposerSendButton, { tooltip, ...props, ref, children: props.children ?? /* @__PURE__ */ jsx37(SendHorizontalIcon, {}) }) });
 });
 ComposerSend.displayName = "ComposerSend";
 var ComposerCancelButton = withDefaults(TooltipIconButton, {
   variant: "default",
   className: "aui-composer-cancel"
 });
-var ComposerCancel = forwardRef18((props, ref) => {
+var ComposerCancel = forwardRef22((props, ref) => {
   const {
     strings: { composer: { cancel: { tooltip = "Cancel" } = {} } = {} } = {}
   } = useThreadConfig();
-  return /* @__PURE__ */ jsx32(composer_exports.Cancel, { asChild: true, children: /* @__PURE__ */ jsx32(ComposerCancelButton, { tooltip, ...props, ref, children: props.children ?? /* @__PURE__ */ jsx32(CircleStopIcon, {}) }) });
+  return /* @__PURE__ */ jsx37(composer_exports.Cancel, { asChild: true, children: /* @__PURE__ */ jsx37(ComposerCancelButton, { tooltip, ...props, ref, children: props.children ?? /* @__PURE__ */ jsx37(CircleStopIcon, {}) }) });
 });
 ComposerCancel.displayName = "ComposerCancel";
 var exports5 = {
@@ -2970,15 +3525,15 @@ var exports5 = {
 var composer_default = Object.assign(Composer, exports5);
 
 // src/ui/thread-welcome.tsx
-import { forwardRef as forwardRef19 } from "react";
-import { jsx as jsx33, jsxs as jsxs9 } from "react/jsx-runtime";
+import { forwardRef as forwardRef23 } from "react";
+import { jsx as jsx38, jsxs as jsxs10 } from "react/jsx-runtime";
 var ThreadWelcome = () => {
-  return /* @__PURE__ */ jsxs9(ThreadWelcomeRoot, { children: [
-    /* @__PURE__ */ jsxs9(ThreadWelcomeCenter, { children: [
-      /* @__PURE__ */ jsx33(ThreadWelcomeAvatar, {}),
-      /* @__PURE__ */ jsx33(ThreadWelcomeMessage, {})
+  return /* @__PURE__ */ jsxs10(ThreadWelcomeRoot, { children: [
+    /* @__PURE__ */ jsxs10(ThreadWelcomeCenter, { children: [
+      /* @__PURE__ */ jsx38(ThreadWelcomeAvatar, {}),
+      /* @__PURE__ */ jsx38(ThreadWelcomeMessage, {})
     ] }),
-    /* @__PURE__ */ jsx33(ThreadWelcomeSuggestions, {})
+    /* @__PURE__ */ jsx38(ThreadWelcomeSuggestions, {})
   ] });
 };
 ThreadWelcome.displayName = "ThreadWelcome";
@@ -2988,22 +3543,22 @@ var ThreadWelcomeRootStyled = withDefaults("div", {
 var ThreadWelcomeCenter = withDefaults("div", {
   className: "aui-thread-welcome-center"
 });
-var ThreadWelcomeRoot = forwardRef19(
+var ThreadWelcomeRoot = forwardRef23(
   (props, ref) => {
-    return /* @__PURE__ */ jsx33(thread_exports.Empty, { children: /* @__PURE__ */ jsx33(ThreadWelcomeRootStyled, { ...props, ref }) });
+    return /* @__PURE__ */ jsx38(thread_exports.Empty, { children: /* @__PURE__ */ jsx38(ThreadWelcomeRootStyled, { ...props, ref }) });
   }
 );
 ThreadWelcomeRoot.displayName = "ThreadWelcomeRoot";
 var ThreadWelcomeAvatar = () => {
   const { assistantAvatar: avatar = { fallback: "A" } } = useThreadConfig();
-  return /* @__PURE__ */ jsx33(Avatar, { ...avatar });
+  return /* @__PURE__ */ jsx38(Avatar, { ...avatar });
 };
 var ThreadWelcomeMessageStyled = withDefaults("p", {
   className: "aui-thread-welcome-message"
 });
-var ThreadWelcomeMessage = forwardRef19(({ message: messageProp, ...rest }, ref) => {
+var ThreadWelcomeMessage = forwardRef23(({ message: messageProp, ...rest }, ref) => {
   const { welcome: { message = "How can I help you today?" } = {} } = useThreadConfig();
-  return /* @__PURE__ */ jsx33(ThreadWelcomeMessageStyled, { ...rest, ref, children: messageProp ?? message });
+  return /* @__PURE__ */ jsx38(ThreadWelcomeMessageStyled, { ...rest, ref, children: messageProp ?? message });
 });
 ThreadWelcomeMessage.displayName = "ThreadWelcomeMessage";
 var ThreadWelcomeSuggestionContainer = withDefaults("div", {
@@ -3015,21 +3570,21 @@ var ThreadWelcomeSuggestionStyled = withDefaults(thread_exports.Suggestion, {
 var ThreadWelcomeSuggestion = ({
   suggestion: { text, prompt }
 }) => {
-  return /* @__PURE__ */ jsx33(
+  return /* @__PURE__ */ jsx38(
     ThreadWelcomeSuggestionStyled,
     {
       prompt,
       method: "replace",
       autoSend: true,
-      children: /* @__PURE__ */ jsx33("span", { className: "aui-thread-welcome-suggestion-text", children: text ?? prompt })
+      children: /* @__PURE__ */ jsx38("span", { className: "aui-thread-welcome-suggestion-text", children: text ?? prompt })
     }
   );
 };
 var ThreadWelcomeSuggestions = () => {
   const { welcome: { suggestions } = {} } = useThreadConfig();
-  return /* @__PURE__ */ jsx33(ThreadWelcomeSuggestionContainer, { children: suggestions?.map((suggestion, idx) => {
+  return /* @__PURE__ */ jsx38(ThreadWelcomeSuggestionContainer, { children: suggestions?.map((suggestion, idx) => {
     const key = `${suggestion.prompt}-${idx}`;
-    return /* @__PURE__ */ jsx33(ThreadWelcomeSuggestion, { suggestion }, key);
+    return /* @__PURE__ */ jsx38(ThreadWelcomeSuggestion, { suggestion }, key);
   }) });
 };
 ThreadWelcomeSuggestions.displayName = "ThreadWelcomeSuggestions";
@@ -3044,12 +3599,12 @@ var exports6 = {
 var thread_welcome_default = Object.assign(ThreadWelcome, exports6);
 
 // src/ui/user-message.tsx
-import { forwardRef as forwardRef21 } from "react";
+import { forwardRef as forwardRef25 } from "react";
 
 // src/ui/user-action-bar.tsx
-import { forwardRef as forwardRef20 } from "react";
+import { forwardRef as forwardRef24 } from "react";
 import { PencilIcon } from "lucide-react";
-import { jsx as jsx34 } from "react/jsx-runtime";
+import { jsx as jsx39 } from "react/jsx-runtime";
 var useAllowEdit = () => {
   const { userMessage: { allowEdit = true } = {} } = useThreadConfig();
   const { useThreadActions } = useThreadContext();
@@ -3059,20 +3614,20 @@ var useAllowEdit = () => {
 var UserActionBar = () => {
   const allowEdit = useAllowEdit();
   if (!allowEdit) return null;
-  return /* @__PURE__ */ jsx34(UserActionBarRoot, { hideWhenRunning: true, autohide: "not-last", children: /* @__PURE__ */ jsx34(UserActionBarEdit, {}) });
+  return /* @__PURE__ */ jsx39(UserActionBarRoot, { hideWhenRunning: true, autohide: "not-last", children: /* @__PURE__ */ jsx39(UserActionBarEdit, {}) });
 };
 UserActionBar.displayName = "UserActionBar";
 var UserActionBarRoot = withDefaults(actionBar_exports.Root, {
   className: "aui-user-action-bar-root"
 });
 UserActionBarRoot.displayName = "UserActionBarRoot";
-var UserActionBarEdit = forwardRef20((props, ref) => {
+var UserActionBarEdit = forwardRef24((props, ref) => {
   const {
     strings: { userMessage: { edit: { tooltip = "Edit" } = {} } = {} } = {}
   } = useThreadConfig();
   const allowEdit = useAllowEdit();
   if (!allowEdit) return null;
-  return /* @__PURE__ */ jsx34(actionBar_exports.Edit, { asChild: true, children: /* @__PURE__ */ jsx34(TooltipIconButton, { tooltip, ...props, ref, children: props.children ?? /* @__PURE__ */ jsx34(PencilIcon, {}) }) });
+  return /* @__PURE__ */ jsx39(actionBar_exports.Edit, { asChild: true, children: /* @__PURE__ */ jsx39(TooltipIconButton, { tooltip, ...props, ref, children: props.children ?? /* @__PURE__ */ jsx39(PencilIcon, {}) }) });
 });
 UserActionBarEdit.displayName = "UserActionBarEdit";
 var exports7 = {
@@ -3082,12 +3637,12 @@ var exports7 = {
 var user_action_bar_default = Object.assign(UserActionBar, exports7);
 
 // src/ui/user-message.tsx
-import { jsx as jsx35, jsxs as jsxs10 } from "react/jsx-runtime";
+import { jsx as jsx40, jsxs as jsxs11 } from "react/jsx-runtime";
 var UserMessage = () => {
-  return /* @__PURE__ */ jsxs10(UserMessageRoot, { children: [
-    /* @__PURE__ */ jsx35(user_action_bar_default, {}),
-    /* @__PURE__ */ jsx35(UserMessageContent, {}),
-    /* @__PURE__ */ jsx35(branch_picker_default, {})
+  return /* @__PURE__ */ jsxs11(UserMessageRoot, { children: [
+    /* @__PURE__ */ jsx40(user_action_bar_default, {}),
+    /* @__PURE__ */ jsx40(UserMessageContent, {}),
+    /* @__PURE__ */ jsx40(branch_picker_default, {})
   ] });
 };
 UserMessage.displayName = "UserMessage";
@@ -3098,9 +3653,9 @@ UserMessageRoot.displayName = "UserMessageRoot";
 var UserMessageContentWrapper = withDefaults("div", {
   className: "aui-user-message-content"
 });
-var UserMessageContent = forwardRef21(
+var UserMessageContent = forwardRef25(
   ({ components, ...props }, ref) => {
-    return /* @__PURE__ */ jsx35(UserMessageContentWrapper, { ...props, ref, children: /* @__PURE__ */ jsx35(
+    return /* @__PURE__ */ jsx40(UserMessageContentWrapper, { ...props, ref, children: /* @__PURE__ */ jsx40(
       message_exports.Content,
       {
         components: {
@@ -3119,14 +3674,14 @@ var exports8 = {
 var user_message_default = Object.assign(UserMessage, exports8);
 
 // src/ui/edit-composer.tsx
-import { forwardRef as forwardRef22 } from "react";
-import { jsx as jsx36, jsxs as jsxs11 } from "react/jsx-runtime";
+import { forwardRef as forwardRef26 } from "react";
+import { jsx as jsx41, jsxs as jsxs12 } from "react/jsx-runtime";
 var EditComposer = () => {
-  return /* @__PURE__ */ jsxs11(EditComposerRoot, { children: [
-    /* @__PURE__ */ jsx36(EditComposerInput, {}),
-    /* @__PURE__ */ jsxs11(EditComposerFooter, { children: [
-      /* @__PURE__ */ jsx36(EditComposerCancel, {}),
-      /* @__PURE__ */ jsx36(EditComposerSend, {})
+  return /* @__PURE__ */ jsxs12(EditComposerRoot, { children: [
+    /* @__PURE__ */ jsx41(EditComposerInput, {}),
+    /* @__PURE__ */ jsxs12(EditComposerFooter, { children: [
+      /* @__PURE__ */ jsx41(EditComposerCancel, {}),
+      /* @__PURE__ */ jsx41(EditComposerSend, {})
     ] })
   ] });
 };
@@ -3143,23 +3698,23 @@ var EditComposerFooter = withDefaults("div", {
   className: "aui-edit-composer-footer"
 });
 EditComposerFooter.displayName = "EditComposerFooter";
-var EditComposerCancel = forwardRef22(
+var EditComposerCancel = forwardRef26(
   (props, ref) => {
     const {
       strings: {
         editComposer: { cancel: { label = "Cancel" } = {} } = {}
       } = {}
     } = useThreadConfig();
-    return /* @__PURE__ */ jsx36(composer_exports.Cancel, { asChild: true, children: /* @__PURE__ */ jsx36(Button, { variant: "ghost", ...props, ref, children: props.children ?? label }) });
+    return /* @__PURE__ */ jsx41(composer_exports.Cancel, { asChild: true, children: /* @__PURE__ */ jsx41(Button, { variant: "ghost", ...props, ref, children: props.children ?? label }) });
   }
 );
 EditComposerCancel.displayName = "EditComposerCancel";
-var EditComposerSend = forwardRef22(
+var EditComposerSend = forwardRef26(
   (props, ref) => {
     const {
       strings: { editComposer: { send: { label = "Send" } = {} } = {} } = {}
     } = useThreadConfig();
-    return /* @__PURE__ */ jsx36(composer_exports.Send, { asChild: true, children: /* @__PURE__ */ jsx36(Button, { ...props, ref, children: props.children ?? label }) });
+    return /* @__PURE__ */ jsx41(composer_exports.Send, { asChild: true, children: /* @__PURE__ */ jsx41(Button, { ...props, ref, children: props.children ?? label }) });
   }
 );
 EditComposerSend.displayName = "EditComposerSend";
@@ -3173,23 +3728,23 @@ var exports9 = {
 var edit_composer_default = Object.assign(EditComposer, exports9);
 
 // src/ui/thread.tsx
-import { jsx as jsx37, jsxs as jsxs12 } from "react/jsx-runtime";
+import { jsx as jsx42, jsxs as jsxs13 } from "react/jsx-runtime";
 var Thread = (config) => {
-  return /* @__PURE__ */ jsx37(ThreadRoot, { config, children: /* @__PURE__ */ jsxs12(ThreadViewport, { children: [
-    /* @__PURE__ */ jsx37(thread_welcome_default, {}),
-    /* @__PURE__ */ jsx37(ThreadMessages, {}),
-    /* @__PURE__ */ jsxs12(ThreadViewportFooter, { children: [
-      /* @__PURE__ */ jsx37(ThreadScrollToBottom, {}),
-      /* @__PURE__ */ jsx37(composer_default, {})
+  return /* @__PURE__ */ jsx42(ThreadRoot, { config, children: /* @__PURE__ */ jsxs13(ThreadViewport, { children: [
+    /* @__PURE__ */ jsx42(thread_welcome_default, {}),
+    /* @__PURE__ */ jsx42(ThreadMessages, {}),
+    /* @__PURE__ */ jsxs13(ThreadViewportFooter, { children: [
+      /* @__PURE__ */ jsx42(ThreadScrollToBottom, {}),
+      /* @__PURE__ */ jsx42(composer_default, {})
     ] })
   ] }) });
 };
 var ThreadRootStyled = withDefaults(thread_exports.Root, {
   className: "aui-root aui-thread-root"
 });
-var ThreadRoot = forwardRef23(
+var ThreadRoot = forwardRef27(
   ({ config, ...props }, ref) => {
-    return /* @__PURE__ */ jsx37(ThreadConfigProvider, { config, children: /* @__PURE__ */ jsx37(ThreadRootStyled, { ...props, ref }) });
+    return /* @__PURE__ */ jsx42(ThreadConfigProvider, { config, children: /* @__PURE__ */ jsx42(ThreadRootStyled, { ...props, ref }) });
   }
 );
 ThreadRoot.displayName = "ThreadRoot";
@@ -3203,7 +3758,7 @@ var ThreadViewportFooter = withDefaults("div", {
 ThreadViewportFooter.displayName = "ThreadViewportFooter";
 var SystemMessage = () => null;
 var ThreadMessages = ({ components, ...rest }) => {
-  return /* @__PURE__ */ jsx37(
+  return /* @__PURE__ */ jsx42(
     thread_exports.Messages,
     {
       components: {
@@ -3221,13 +3776,13 @@ var ThreadScrollToBottomIconButton = withDefaults(TooltipIconButton, {
   variant: "outline",
   className: "aui-thread-scroll-to-bottom"
 });
-var ThreadScrollToBottom = forwardRef23((props, ref) => {
+var ThreadScrollToBottom = forwardRef27((props, ref) => {
   const {
     strings: {
       thread: { scrollToBottom: { tooltip = "Scroll to bottom" } = {} } = {}
     } = {}
   } = useThreadConfig();
-  return /* @__PURE__ */ jsx37(thread_exports.ScrollToBottom, { asChild: true, children: /* @__PURE__ */ jsx37(ThreadScrollToBottomIconButton, { tooltip, ...props, ref, children: props.children ?? /* @__PURE__ */ jsx37(ArrowDownIcon, {}) }) });
+  return /* @__PURE__ */ jsx42(thread_exports.ScrollToBottom, { asChild: true, children: /* @__PURE__ */ jsx42(ThreadScrollToBottomIconButton, { tooltip, ...props, ref, children: props.children ?? /* @__PURE__ */ jsx42(ArrowDownIcon, {}) }) });
 });
 ThreadScrollToBottom.displayName = "ThreadScrollToBottom";
 var exports10 = {
@@ -3240,20 +3795,20 @@ var exports10 = {
 var thread_default = Object.assign(Thread, exports10);
 
 // src/ui/assistant-modal.tsx
-import { Fragment as Fragment6, jsx as jsx38, jsxs as jsxs13 } from "react/jsx-runtime";
+import { Fragment as Fragment6, jsx as jsx43, jsxs as jsxs14 } from "react/jsx-runtime";
 var AssistantModal = (config) => {
-  return /* @__PURE__ */ jsxs13(AssistantModalRoot, { config, children: [
-    /* @__PURE__ */ jsx38(AssistantModalTrigger, {}),
-    /* @__PURE__ */ jsx38(AssistantModalContent, { children: /* @__PURE__ */ jsx38(thread_default, {}) })
+  return /* @__PURE__ */ jsxs14(AssistantModalRoot, { config, children: [
+    /* @__PURE__ */ jsx43(AssistantModalTrigger, {}),
+    /* @__PURE__ */ jsx43(AssistantModalContent, { children: /* @__PURE__ */ jsx43(thread_default, {}) })
   ] });
 };
 AssistantModal.displayName = "AssistantModal";
 var AssistantModalRoot = ({ config, ...props }) => {
-  return /* @__PURE__ */ jsx38(ThreadConfigProvider, { config, children: /* @__PURE__ */ jsx38(assistantModal_exports.Root, { ...props }) });
+  return /* @__PURE__ */ jsx43(ThreadConfigProvider, { config, children: /* @__PURE__ */ jsx43(assistantModal_exports.Root, { ...props }) });
 };
 AssistantModalRoot.displayName = "AssistantModalRoot";
-var AssistantModalTrigger = forwardRef24((props, ref) => {
-  return /* @__PURE__ */ jsx38(AssistantModalAnchor, { children: /* @__PURE__ */ jsx38(assistantModal_exports.Trigger, { asChild: true, children: /* @__PURE__ */ jsx38(AssistantModalButton, { ...props, ref }) }) });
+var AssistantModalTrigger = forwardRef28((props, ref) => {
+  return /* @__PURE__ */ jsx43(AssistantModalAnchor, { children: /* @__PURE__ */ jsx43(assistantModal_exports.Trigger, { asChild: true, children: /* @__PURE__ */ jsx43(AssistantModalButton, { ...props, ref }) }) });
 });
 AssistantModalTrigger.displayName = "AssistantModalTrigger";
 var AssistantModalAnchor = withDefaults(assistantModal_exports.Anchor, {
@@ -3264,7 +3819,7 @@ var ModalButtonStyled = withDefaults(TooltipIconButton, {
   variant: "default",
   className: "aui-modal-button"
 });
-var AssistantModalButton = forwardRef24(({ "data-state": state, ...rest }, ref) => {
+var AssistantModalButton = forwardRef28(({ "data-state": state, ...rest }, ref) => {
   const {
     strings: {
       assistantModal: {
@@ -3278,7 +3833,7 @@ var AssistantModalButton = forwardRef24(({ "data-state": state, ...rest }, ref)
     } = {}
   } = useThreadConfig();
   const tooltip = state === "open" ? openTooltip : closedTooltip;
-  return /* @__PURE__ */ jsx38(
+  return /* @__PURE__ */ jsx43(
     ModalButtonStyled,
     {
       side: "left",
@@ -3286,15 +3841,15 @@ var AssistantModalButton = forwardRef24(({ "data-state": state, ...rest }, ref)
       "data-state": state,
       ...rest,
       ref,
-      children: rest.children ?? /* @__PURE__ */ jsxs13(Fragment6, { children: [
-        /* @__PURE__ */ jsx38(
+      children: rest.children ?? /* @__PURE__ */ jsxs14(Fragment6, { children: [
+        /* @__PURE__ */ jsx43(
           BotIcon,
           {
             "data-state": state,
             className: "aui-modal-button-closed-icon"
           }
         ),
-        /* @__PURE__ */ jsx38(
+        /* @__PURE__ */ jsx43(
           ChevronDownIcon,
           {
             "data-state": state,