@assistant-ui/react 0.5.19 → 0.5.21

package/dist/index.mjs

@@ -1,9 +1,40 @@
 "use client";
-var __defProp = Object.defineProperty;
-var __export = (target, all) => {
-  for (var name in all)
-    __defProp(target, name, { get: all[name], enumerable: true });
-};
+import {
+  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";
+import {
+  __export
+} from "./chunk-BJPOCE4O.mjs";
 
 // src/context/providers/AssistantRuntimeProvider.tsx
 import { memo } from "react";
@@ -11,93 +42,8 @@ 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/types/ModelConfigTypes.ts
-import { z } from "zod";
-var LanguageModelV1CallSettingsSchema = z.object({
-  maxTokens: z.number().int().positive().optional(),
-  temperature: z.number().optional(),
-  topP: z.number().optional(),
-  presencePenalty: z.number().optional(),
-  frequencyPenalty: z.number().optional(),
-  seed: z.number().int().optional(),
-  headers: z.record(z.string().optional()).optional()
-});
-var LanguageModelConfigSchema = z.object({
-  apiKey: z.string().optional(),
-  baseUrl: z.string().optional(),
-  modelName: z.string().optional()
-});
-var mergeModelConfigs = (configSet) => {
-  const configs = Array.from(configSet).map((c) => c.getModelConfig()).sort((a, b) => (b.priority ?? 0) - (a.priority ?? 0));
-  return configs.reduce((acc, config) => {
-    if (config.system) {
-      if (acc.system) {
-        acc.system += `
-
-${config.system}`;
-      } else {
-        acc.system = config.system;
-      }
-    }
-    if (config.tools) {
-      for (const [name, tool] of Object.entries(config.tools)) {
-        if (acc.tools?.[name]) {
-          throw new Error(
-            `You tried to define a tool with the name ${name}, but it already exists.`
-          );
-        }
-        if (!acc.tools) acc.tools = {};
-        acc.tools[name] = tool;
-      }
-    }
-    if (config.config) {
-      acc.config = {
-        ...acc.config,
-        ...config.config
-      };
-    }
-    if (config.callSettings) {
-      acc.callSettings = {
-        ...acc.callSettings,
-        ...config.callSettings
-      };
-    }
-    return acc;
-  }, {});
-};
-
-// 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({
@@ -145,18 +91,6 @@ 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";
 
@@ -384,48 +318,6 @@ 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) => {
@@ -543,7 +435,7 @@ var useAssistantInstructions = (instruction) => {
 import { useCallback as useCallback5 } from "react";
 
 // src/utils/combined/useCombinedStore.ts
-import { useMemo as useMemo2 } from "react";
+import { useMemo } from "react";
 
 // src/utils/combined/createCombinedStore.ts
 import { useSyncExternalStore } from "react";
@@ -564,7 +456,7 @@ var createCombinedStore = (stores) => {
 
 // src/utils/combined/useCombinedStore.ts
 var useCombinedStore = (stores, selector) => {
-  const useCombined = useMemo2(() => createCombinedStore(stores), stores);
+  const useCombined = useMemo(() => createCombinedStore(stores), stores);
   return useCombined(selector);
 };
 
@@ -1118,7 +1010,7 @@ var BranchPickerPrimitiveNumber = () => {
 BranchPickerPrimitiveNumber.displayName = "BranchPickerPrimitive.Number";
 
 // src/primitives/branchPicker/BranchPickerRoot.tsx
-import { Primitive as Primitive6 } from "@radix-ui/react-primitive";
+import { Primitive as Primitive5 } from "@radix-ui/react-primitive";
 import { forwardRef as forwardRef9 } from "react";
 
 // src/primitives/message/index.ts
@@ -1161,9 +1053,9 @@ var useIsHoveringRef = () => {
   );
   return useManagedRef(callbackRef);
 };
-var MessagePrimitiveRoot = forwardRef6(({ onMouseEnter, onMouseLeave, ...rest }, forwardRef28) => {
+var MessagePrimitiveRoot = forwardRef6(({ onMouseEnter, onMouseLeave, ...rest }, forwardRef25) => {
   const isHoveringRef = useIsHoveringRef();
-  const ref = useComposedRefs(forwardRef28, isHoveringRef);
+  const ref = useComposedRefs(forwardRef25, isHoveringRef);
   return /* @__PURE__ */ jsx12(Primitive3.div, { ...rest, ref });
 });
 MessagePrimitiveRoot.displayName = "MessagePrimitive.Root";
@@ -1225,7 +1117,6 @@ var useContentPartContext2 = (partIndex) => {
     const useContentPart = create10(
       () => getContentPartState(useMessage.getState(), void 0, partIndex)
     );
-    getContentPartState(useMessage.getState(), useContentPart, partIndex);
     return { useContentPart };
   });
   useEffect6(() => {
@@ -1252,108 +1143,28 @@ var ContentPartProvider = ({
 };
 
 // src/primitives/contentPart/ContentPartText.tsx
-import { Primitive as Primitive4 } from "@radix-ui/react-primitive";
-import { forwardRef as forwardRef7 } from "react";
-
-// src/utils/hooks/useSmooth.tsx
-import { useEffect as useEffect7, useRef as useRef3, useState as useState5 } from "react";
-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 useSmooth = (text, smooth = false) => {
-  const { useMessage } = useMessageContext();
-  const id = useMessage((m) => m.message.id);
-  const idRef = useRef3(id);
-  const [displayedText, setDisplayedText] = useState5(text);
-  const [animatorRef] = useState5(
-    new TextStreamAnimator(text, setDisplayedText)
-  );
-  useEffect7(() => {
-    if (!smooth) {
-      animatorRef.stop();
-      return;
-    }
-    if (idRef.current !== id || !text.startsWith(animatorRef.targetText)) {
-      idRef.current = id;
-      setDisplayedText(text);
-      animatorRef.currentText = text;
-      animatorRef.targetText = text;
-      animatorRef.stop();
-      return;
-    }
-    animatorRef.targetText = text;
-    animatorRef.start();
-  }, [animatorRef, id, smooth, text]);
-  useEffect7(() => {
-    return () => {
-      animatorRef.stop();
-    };
-  }, [animatorRef]);
-  return smooth ? displayedText : text;
-};
-
-// src/primitives/contentPart/ContentPartText.tsx
+import {
+  forwardRef as forwardRef7
+} from "react";
 import { jsx as jsx14 } from "react/jsx-runtime";
-var ContentPartPrimitiveText = forwardRef7(({ smooth = true, ...rest }, forwardedRef) => {
+var ContentPartPrimitiveText = forwardRef7(({ smooth = true, component: Component = "span", ...rest }, forwardedRef) => {
   const {
-    status,
-    part: { text }
-  } = useContentPartText();
-  const smoothText = useSmooth(text, smooth);
-  return /* @__PURE__ */ jsx14(Primitive4.span, { "data-status": status.type, ...rest, ref: forwardedRef, children: smoothText });
+    part: { text },
+    status
+  } = useSmooth(useContentPartText(), smooth);
+  return /* @__PURE__ */ jsx14(Component, { "data-status": status.type, ...rest, ref: forwardedRef, children: text });
 });
 ContentPartPrimitiveText.displayName = "ContentPartPrimitive.Text";
 
 // src/primitives/contentPart/ContentPartImage.tsx
-import { Primitive as Primitive5 } from "@radix-ui/react-primitive";
+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) => {
   const {
     part: { image }
   } = useContentPartImage();
-  return /* @__PURE__ */ jsx15(Primitive5.img, { src: image, ...props, ref: forwardedRef });
+  return /* @__PURE__ */ jsx15(Primitive4.img, { src: image, ...props, ref: forwardedRef });
 });
 ContentPartPrimitiveImage.displayName = "ContentPartPrimitive.Image";
 
@@ -1473,7 +1284,7 @@ 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(Primitive6.div, { ...rest, ref }) });
+  return /* @__PURE__ */ jsx17(MessagePrimitiveIf, { hasBranches: hideWhenSingleBranch ? true : void 0, children: /* @__PURE__ */ jsx17(Primitive5.div, { ...rest, ref }) });
 });
 BranchPickerPrimitiveRoot.displayName = "BranchPickerPrimitive.Root";
 
@@ -1489,7 +1300,7 @@ __export(composer_exports, {
 
 // src/primitives/composer/ComposerRoot.tsx
 import { composeEventHandlers as composeEventHandlers4 } from "@radix-ui/primitive";
-import { Primitive as Primitive7 } from "@radix-ui/react-primitive";
+import { Primitive as Primitive6 } from "@radix-ui/react-primitive";
 import {
   forwardRef as forwardRef10
 } from "react";
@@ -1502,7 +1313,7 @@ var ComposerPrimitiveRoot = forwardRef10(({ onSubmit, ...rest }, forwardedRef) =
     send();
   };
   return /* @__PURE__ */ jsx18(
-    Primitive7.form,
+    Primitive6.form,
     {
       ...rest,
       ref: forwardedRef,
@@ -1519,8 +1330,8 @@ import { Slot } from "@radix-ui/react-slot";
 import {
   forwardRef as forwardRef11,
   useCallback as useCallback15,
-  useEffect as useEffect8,
-  useRef as useRef4
+  useEffect as useEffect7,
+  useRef as useRef3
 } from "react";
 import TextareaAutosize from "react-textarea-autosize";
 import { useEscapeKeydown } from "@radix-ui/react-use-escape-keydown";
@@ -1542,7 +1353,7 @@ var ComposerPrimitiveInput = forwardRef11(
     });
     const Component = asChild ? Slot : TextareaAutosize;
     const isDisabled = useThread((t) => t.isDisabled) ?? disabledProp ?? false;
-    const textareaRef = useRef4(null);
+    const textareaRef = useRef3(null);
     const ref = useComposedRefs2(forwardedRef, textareaRef);
     useEscapeKeydown((e) => {
       const composer = useComposer.getState();
@@ -1571,7 +1382,7 @@ var ComposerPrimitiveInput = forwardRef11(
         textareaRef.current.value.length
       );
     }, [autoFocusEnabled]);
-    useEffect8(() => focus(), [focus]);
+    useEffect7(() => focus(), [focus]);
     useOnComposerFocus(() => {
       if (type === "new") {
         focus();
@@ -1599,13 +1410,13 @@ ComposerPrimitiveInput.displayName = "ComposerPrimitive.Input";
 
 // src/primitives/composer/ComposerSend.tsx
 import { forwardRef as forwardRef12 } from "react";
-import { Primitive as Primitive8 } from "@radix-ui/react-primitive";
+import { Primitive as Primitive7 } from "@radix-ui/react-primitive";
 import { jsx as jsx20 } from "react/jsx-runtime";
 var ComposerPrimitiveSend = forwardRef12(({ disabled, ...rest }, ref) => {
   const { useComposer } = useComposerContext();
   const hasValue = useComposer((c) => c.isEditing && c.value.length > 0);
   return /* @__PURE__ */ jsx20(
-    Primitive8.button,
+    Primitive7.button,
     {
       type: "submit",
       ...rest,
@@ -1654,11 +1465,11 @@ __export(thread_exports, {
 });
 
 // src/primitives/thread/ThreadRoot.tsx
-import { Primitive as Primitive9 } from "@radix-ui/react-primitive";
+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(Primitive9.div, { ...props, ref });
+  return /* @__PURE__ */ jsx21(Primitive8.div, { ...props, ref });
 });
 ThreadPrimitiveRoot.displayName = "ThreadPrimitive.Root";
 
@@ -1683,12 +1494,12 @@ ThreadPrimitiveIf.displayName = "ThreadPrimitive.If";
 
 // src/primitives/thread/ThreadViewport.tsx
 import { useComposedRefs as useComposedRefs4 } from "@radix-ui/react-compose-refs";
-import { Primitive as Primitive10 } from "@radix-ui/react-primitive";
+import { Primitive as Primitive9 } from "@radix-ui/react-primitive";
 import { forwardRef as forwardRef14 } from "react";
 
 // src/primitive-hooks/thread/useThreadViewportAutoScroll.tsx
 import { useComposedRefs as useComposedRefs3 } from "@radix-ui/react-compose-refs";
-import { useRef as useRef5 } from "react";
+import { useRef as useRef4 } from "react";
 
 // src/utils/hooks/useOnResizeContent.tsx
 import { useCallbackRef as useCallbackRef2 } from "@radix-ui/react-use-callback-ref";
@@ -1732,11 +1543,11 @@ var useOnResizeContent = (callback) => {
 
 // src/utils/hooks/useOnScrollToBottom.tsx
 import { useCallbackRef as useCallbackRef3 } from "@radix-ui/react-use-callback-ref";
-import { useEffect as useEffect9 } from "react";
+import { useEffect as useEffect8 } from "react";
 var useOnScrollToBottom = (callback) => {
   const callbackRef = useCallbackRef3(callback);
   const { useViewport } = useThreadContext();
-  useEffect9(() => {
+  useEffect8(() => {
     return useViewport.getState().onScrollToBottom(() => {
       callbackRef();
     });
@@ -1747,10 +1558,10 @@ var useOnScrollToBottom = (callback) => {
 var useThreadViewportAutoScroll = ({
   autoScroll = true
 }) => {
-  const divRef = useRef5(null);
+  const divRef = useRef4(null);
   const { useViewport } = useThreadContext();
-  const lastScrollTop = useRef5(0);
-  const isScrollingToBottomRef = useRef5(false);
+  const lastScrollTop = useRef4(0);
+  const isScrollingToBottomRef = useRef4(false);
   const scrollToBottom = (behavior) => {
     const div = divRef.current;
     if (!div || !autoScroll) return;
@@ -1801,7 +1612,7 @@ var ThreadPrimitiveViewport = forwardRef14(({ autoScroll, onScroll, children, ..
     autoScroll
   });
   const ref = useComposedRefs4(forwardedRef, autoScrollRef);
-  return /* @__PURE__ */ jsx22(Primitive10.div, { ...rest, ref, children });
+  return /* @__PURE__ */ jsx22(Primitive9.div, { ...rest, ref, children });
 });
 ThreadPrimitiveViewport.displayName = "ThreadPrimitive.Viewport";
 
@@ -1809,7 +1620,7 @@ ThreadPrimitiveViewport.displayName = "ThreadPrimitive.Viewport";
 import { memo as memo3 } from "react";
 
 // src/context/providers/MessageProvider.tsx
-import { useEffect as useEffect10, useState as useState6 } from "react";
+import { useEffect as useEffect9, useState as useState5 } from "react";
 import { create as create13 } from "zustand";
 
 // src/context/stores/EditComposer.ts
@@ -1871,7 +1682,7 @@ var getMessageState = (messages, getBranches, useMessage, messageIndex) => {
 };
 var useMessageContext2 = (messageIndex) => {
   const { useThreadMessages, useThreadActions } = useThreadContext();
-  const [context] = useState6(() => {
+  const [context] = useState5(() => {
     const useMessage = create13(
       () => getMessageState(
         useThreadMessages.getState(),
@@ -1909,7 +1720,7 @@ var useMessageContext2 = (messageIndex) => {
     });
     return { useMessage, useMessageUtils, useEditComposer };
   });
-  useEffect10(() => {
+  useEffect9(() => {
     const syncMessage = (thread) => {
       const newState = getMessageState(
         thread,
@@ -2003,617 +1814,112 @@ var ThreadPrimitiveSuggestion = createActionButton(
 );
 
 // src/runtimes/local/useLocalRuntime.tsx
-import { useInsertionEffect as useInsertionEffect3, useState as useState8 } 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
-});
-
-// src/utils/idUtils.tsx
-import { customAlphabet } from "nanoid/non-secure";
-var generateId = customAlphabet(
-  "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz",
-  7
-);
-var optimisticPrefix = "__optimistic__";
-var generateOptimisticId = () => `${optimisticPrefix}${generateId()}`;
+import { useInsertionEffect as useInsertionEffect3, useState as useState7 } from "react";
 
-// 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) => {
+// src/runtimes/edge/converters/fromLanguageModelMessages.ts
+var fromLanguageModelMessages = (lm, { mergeRoundtrips }) => {
+  const messages = [];
+  for (const lmMessage of lm) {
+    const role = lmMessage.role;
+    switch (role) {
+      case "system": {
+        messages.push({
+          role: "system",
+          content: [
+            {
+              type: "text",
+              text: lmMessage.content
+            }
+          ]
+        });
+        break;
+      }
+      case "user": {
+        messages.push({
+          role: "user",
+          content: lmMessage.content.map((part) => {
+            const type = part.type;
+            switch (type) {
+              case "text": {
+                return {
+                  type: "text",
+                  text: part.text
+                };
+              }
+              case "image": {
+                if (part.image instanceof URL) {
+                  return {
+                    type: "image",
+                    image: part.image.href
+                  };
+                }
+                throw new Error("Only images with URL data are supported");
+              }
+              default: {
+                const unhandledType = type;
+                throw new Error(`Unknown content part type: ${unhandledType}`);
+              }
+            }
+          })
+        });
+        break;
+      }
+      case "assistant": {
+        const newContent = lmMessage.content.map((part) => {
           if (part.type === "tool-call") {
             return {
-              ...part,
-              argsText: JSON.stringify(part.args)
+              type: "tool-call",
+              toolCallId: part.toolCallId,
+              toolName: part.toolName,
+              argsText: JSON.stringify(part.args),
+              args: 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) {
+        });
+        if (mergeRoundtrips) {
+          const previousMessage = messages[messages.length - 1];
+          if (previousMessage?.role === "assistant") {
+            previousMessage.content.push(...newContent);
+            break;
+          }
+        }
+        messages.push({
+          role: "assistant",
+          content: newContent
+        });
+        break;
+      }
+      case "tool": {
+        const previousMessage = messages[messages.length - 1];
+        if (previousMessage?.role !== "assistant")
           throw new Error(
-            "MessageRepository(performOp/cut): Fallback sibling message not found. This is likely an internal bug in assistant-ui."
+            "A tool message must be preceded by an assistant message."
+          );
+        for (const tool of lmMessage.content) {
+          const toolCall = previousMessage.content.find(
+            (c) => c.type === "tool-call" && c.toolCallId === tool.toolCallId
           );
+          if (!toolCall)
+            throw new Error("Received tool result for an unknown tool call.");
+          if (toolCall.toolName !== tool.toolName)
+            throw new Error("Tool call name mismatch.");
+          toolCall.result = tool.result;
+          if (tool.isError) {
+            toolCall.isError = true;
+          }
         }
-        child.prev.next = fallback;
+        break;
       }
-    }
-    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 forwardRef17 } from "react";
-
-// src/ui/base/tooltip.tsx
-import * as TooltipPrimitive from "@radix-ui/react-tooltip";
-
-// src/ui/utils/withDefaults.tsx
-import {
-  forwardRef as forwardRef15
-} from "react";
-import classNames from "classnames";
-import { jsx as jsx25 } 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 = forwardRef15(
-    (props, ref) => {
-      const ComponentAsAny = Component;
-      return /* @__PURE__ */ jsx25(ComponentAsAny, { ...getProps(props), ref });
-    }
-  );
-  WithDefaults.displayName = "withDefaults(" + (typeof Component === "string" ? Component : Component.displayName) + ")";
-  return WithDefaults;
-};
-
-// src/ui/base/tooltip.tsx
-import { jsx as jsx26 } from "react/jsx-runtime";
-var Tooltip = (props) => {
-  return /* @__PURE__ */ jsx26(TooltipPrimitive.Provider, { children: /* @__PURE__ */ jsx26(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 Primitive11 } from "@radix-ui/react-primitive";
-import { forwardRef as forwardRef16 } from "react";
-import { jsx as jsx27 } 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 = forwardRef16(
-  ({ className, variant, size, ...props }, ref) => {
-    return /* @__PURE__ */ jsx27(
-      Primitive11.button,
-      {
-        className: buttonVariants({ variant, size, className }),
-        ...props,
-        ref
-      }
-    );
-  }
-);
-Button.displayName = "Button";
-
-// src/ui/base/tooltip-icon-button.tsx
-import { jsx as jsx28, jsxs as jsxs4 } from "react/jsx-runtime";
-var TooltipIconButton = forwardRef17(({ children, tooltip, side = "bottom", ...rest }, ref) => {
-  return /* @__PURE__ */ jsxs4(Tooltip, { children: [
-    /* @__PURE__ */ jsx28(TooltipTrigger, { asChild: true, children: /* @__PURE__ */ jsxs4(Button, { variant: "ghost", size: "icon", ...rest, ref, children: [
-      children,
-      /* @__PURE__ */ jsx28("span", { className: "aui-sr-only", children: tooltip })
-    ] }) }),
-    /* @__PURE__ */ jsx28(TooltipContent, { side, children: tooltip })
-  ] });
-});
-TooltipIconButton.displayName = "TooltipIconButton";
-
-// src/runtimes/edge/converters/toLanguageModelMessages.ts
-var assistantMessageSplitter = () => {
-  const stash = [];
-  let assistantMessage = {
-    role: "assistant",
-    content: []
-  };
-  let toolMessage = {
-    role: "tool",
-    content: []
-  };
-  return {
-    addTextContentPart: (part) => {
-      if (toolMessage.content.length > 0) {
-        stash.push(assistantMessage);
-        stash.push(toolMessage);
-        assistantMessage = {
-          role: "assistant",
-          content: []
-        };
-        toolMessage = {
-          role: "tool",
-          content: []
-        };
-      }
-      assistantMessage.content.push(part);
-    },
-    addToolCallPart: (part) => {
-      assistantMessage.content.push({
-        type: "tool-call",
-        toolCallId: part.toolCallId,
-        toolName: part.toolName,
-        args: part.args
-      });
-      if (part.result) {
-        toolMessage.content.push({
-          type: "tool-result",
-          toolCallId: part.toolCallId,
-          toolName: part.toolName,
-          result: part.result
-          // isError
-        });
-      }
-    },
-    getMessages: () => {
-      if (toolMessage.content.length > 0) {
-        return [...stash, assistantMessage, toolMessage];
-      }
-      return [...stash, assistantMessage];
-    }
-  };
-};
-function toLanguageModelMessages(message) {
-  return message.flatMap((message2) => {
-    const role = message2.role;
-    switch (role) {
-      case "system": {
-        return [{ role: "system", content: message2.content[0].text }];
-      }
-      case "user": {
-        const msg = {
-          role: "user",
-          content: message2.content.map(
-            (part) => {
-              const type = part.type;
-              switch (type) {
-                case "text": {
-                  return part;
-                }
-                case "image": {
-                  return {
-                    type: "image",
-                    image: new URL(part.image)
-                  };
-                }
-                default: {
-                  const unhandledType = type;
-                  throw new Error(
-                    `Unspported content part type: ${unhandledType}`
-                  );
-                }
-              }
-            }
-          )
-        };
-        return [msg];
-      }
-      case "assistant": {
-        const splitter = assistantMessageSplitter();
-        for (const part of message2.content) {
-          const type = part.type;
-          switch (type) {
-            case "text": {
-              splitter.addTextContentPart(part);
-              break;
-            }
-            case "tool-call": {
-              splitter.addToolCallPart(part);
-              break;
-            }
-            default: {
-              const unhandledType = type;
-              throw new Error(`Unhandled content part type: ${unhandledType}`);
-            }
-          }
-        }
-        return splitter.getMessages();
-      }
-      default: {
-        const unhandledRole = role;
-        throw new Error(`Unknown message role: ${unhandledRole}`);
-      }
-    }
-  });
-}
-
-// src/runtimes/edge/converters/fromLanguageModelMessages.ts
-var fromLanguageModelMessages = (lm, { mergeRoundtrips }) => {
-  const messages = [];
-  for (const lmMessage of lm) {
-    const role = lmMessage.role;
-    switch (role) {
-      case "system": {
-        messages.push({
-          role: "system",
-          content: [
-            {
-              type: "text",
-              text: lmMessage.content
-            }
-          ]
-        });
-        break;
-      }
-      case "user": {
-        messages.push({
-          role: "user",
-          content: lmMessage.content.map((part) => {
-            const type = part.type;
-            switch (type) {
-              case "text": {
-                return {
-                  type: "text",
-                  text: part.text
-                };
-              }
-              case "image": {
-                if (part.image instanceof URL) {
-                  return {
-                    type: "image",
-                    image: part.image.href
-                  };
-                }
-                throw new Error("Only images with URL data are supported");
-              }
-              default: {
-                const unhandledType = type;
-                throw new Error(`Unknown content part type: ${unhandledType}`);
-              }
-            }
-          })
-        });
-        break;
-      }
-      case "assistant": {
-        const newContent = lmMessage.content.map((part) => {
-          if (part.type === "tool-call") {
-            return {
-              type: "tool-call",
-              toolCallId: part.toolCallId,
-              toolName: part.toolName,
-              argsText: JSON.stringify(part.args),
-              args: part.args
-            };
-          }
-          return part;
-        });
-        if (mergeRoundtrips) {
-          const previousMessage = messages[messages.length - 1];
-          if (previousMessage?.role === "assistant") {
-            previousMessage.content.push(...newContent);
-            break;
-          }
-        }
-        messages.push({
-          role: "assistant",
-          content: newContent
-        });
-        break;
-      }
-      case "tool": {
-        const previousMessage = messages[messages.length - 1];
-        if (previousMessage?.role !== "assistant")
-          throw new Error(
-            "A tool message must be preceded by an assistant message."
-          );
-        for (const tool of lmMessage.content) {
-          const toolCall = previousMessage.content.find(
-            (c) => c.type === "tool-call" && c.toolCallId === tool.toolCallId
-          );
-          if (!toolCall)
-            throw new Error("Received tool result for an unknown tool call.");
-          if (toolCall.toolName !== tool.toolName)
-            throw new Error("Tool call name mismatch.");
-          toolCall.result = tool.result;
-          if (tool.isError) {
-            toolCall.isError = true;
-          }
-        }
-        break;
-      }
-      default: {
-        const unhandledRole = role;
-        throw new Error(`Unknown message role: ${unhandledRole}`);
+      default: {
+        const unhandledRole = role;
+        throw new Error(`Unknown message role: ${unhandledRole}`);
       }
     }
   }
   return messages;
 };
 
-// src/runtimes/edge/converters/toCoreMessages.ts
-var toCoreMessages = (message) => {
-  return message.map(toCoreMessage);
-};
-var toCoreMessage = (message) => {
-  const role = message.role;
-  switch (role) {
-    case "assistant":
-      return {
-        role,
-        content: message.content.map((part) => {
-          if (part.type === "ui") throw new Error("UI parts are not supported");
-          if (part.type === "tool-call") {
-            const { argsText, ...rest } = part;
-            return rest;
-          }
-          return part;
-        })
-      };
-    case "user":
-      return {
-        role,
-        content: message.content.map((part) => {
-          if (part.type === "ui") throw new Error("UI parts are not supported");
-          return part;
-        })
-      };
-    case "system":
-      return {
-        role,
-        content: message.content
-      };
-    default: {
-      const unsupportedRole = role;
-      throw new Error(`Unknown message role: ${unsupportedRole}`);
-    }
-  }
-};
-
 // src/runtimes/edge/converters/fromLanguageModelTools.ts
 var fromLanguageModelTools = (tools) => {
   return Object.fromEntries(
@@ -2627,20 +1933,8 @@ var fromLanguageModelTools = (tools) => {
   );
 };
 
-// src/runtimes/edge/converters/toLanguageModelTools.ts
-import { z as z2 } from "zod";
-import zodToJsonSchema from "zod-to-json-schema";
-var toLanguageModelTools = (tools) => {
-  return Object.entries(tools).map(([name, tool]) => ({
-    type: "function",
-    name,
-    ...tool.description ? { description: tool.description } : void 0,
-    parameters: tool.parameters instanceof z2.ZodType ? zodToJsonSchema(tool.parameters) : tool.parameters
-  }));
-};
-
 // src/runtimes/edge/useEdgeRuntime.ts
-import { useState as useState7 } from "react";
+import { useState as useState6 } from "react";
 
 // src/runtimes/edge/streams/assistantDecoderStream.ts
 function assistantDecoderStream() {
@@ -2746,589 +2040,6 @@ function chunkByLineStream() {
   });
 }
 
-// src/runtimes/edge/partial-json/parse-partial-json.ts
-import sjson from "secure-json-parse";
-
-// src/runtimes/edge/partial-json/fix-json.ts
-function fixJson(input) {
-  const stack = ["ROOT"];
-  let lastValidIndex = -1;
-  let literalStart = null;
-  function processValueStart(char, i, swapState) {
-    {
-      switch (char) {
-        case '"': {
-          lastValidIndex = i;
-          stack.pop();
-          stack.push(swapState);
-          stack.push("INSIDE_STRING");
-          break;
-        }
-        case "f":
-        case "t":
-        case "n": {
-          lastValidIndex = i;
-          literalStart = i;
-          stack.pop();
-          stack.push(swapState);
-          stack.push("INSIDE_LITERAL");
-          break;
-        }
-        case "-": {
-          stack.pop();
-          stack.push(swapState);
-          stack.push("INSIDE_NUMBER");
-          break;
-        }
-        case "0":
-        case "1":
-        case "2":
-        case "3":
-        case "4":
-        case "5":
-        case "6":
-        case "7":
-        case "8":
-        case "9": {
-          lastValidIndex = i;
-          stack.pop();
-          stack.push(swapState);
-          stack.push("INSIDE_NUMBER");
-          break;
-        }
-        case "{": {
-          lastValidIndex = i;
-          stack.pop();
-          stack.push(swapState);
-          stack.push("INSIDE_OBJECT_START");
-          break;
-        }
-        case "[": {
-          lastValidIndex = i;
-          stack.pop();
-          stack.push(swapState);
-          stack.push("INSIDE_ARRAY_START");
-          break;
-        }
-      }
-    }
-  }
-  function processAfterObjectValue(char, i) {
-    switch (char) {
-      case ",": {
-        stack.pop();
-        stack.push("INSIDE_OBJECT_AFTER_COMMA");
-        break;
-      }
-      case "}": {
-        lastValidIndex = i;
-        stack.pop();
-        break;
-      }
-    }
-  }
-  function processAfterArrayValue(char, i) {
-    switch (char) {
-      case ",": {
-        stack.pop();
-        stack.push("INSIDE_ARRAY_AFTER_COMMA");
-        break;
-      }
-      case "]": {
-        lastValidIndex = i;
-        stack.pop();
-        break;
-      }
-    }
-  }
-  for (let i = 0; i < input.length; i++) {
-    const char = input[i];
-    const currentState = stack[stack.length - 1];
-    switch (currentState) {
-      case "ROOT":
-        processValueStart(char, i, "FINISH");
-        break;
-      case "INSIDE_OBJECT_START": {
-        switch (char) {
-          case '"': {
-            stack.pop();
-            stack.push("INSIDE_OBJECT_KEY");
-            break;
-          }
-          case "}": {
-            lastValidIndex = i;
-            stack.pop();
-            break;
-          }
-        }
-        break;
-      }
-      case "INSIDE_OBJECT_AFTER_COMMA": {
-        switch (char) {
-          case '"': {
-            stack.pop();
-            stack.push("INSIDE_OBJECT_KEY");
-            break;
-          }
-        }
-        break;
-      }
-      case "INSIDE_OBJECT_KEY": {
-        switch (char) {
-          case '"': {
-            stack.pop();
-            stack.push("INSIDE_OBJECT_AFTER_KEY");
-            break;
-          }
-        }
-        break;
-      }
-      case "INSIDE_OBJECT_AFTER_KEY": {
-        switch (char) {
-          case ":": {
-            stack.pop();
-            stack.push("INSIDE_OBJECT_BEFORE_VALUE");
-            break;
-          }
-        }
-        break;
-      }
-      case "INSIDE_OBJECT_BEFORE_VALUE": {
-        processValueStart(char, i, "INSIDE_OBJECT_AFTER_VALUE");
-        break;
-      }
-      case "INSIDE_OBJECT_AFTER_VALUE": {
-        processAfterObjectValue(char, i);
-        break;
-      }
-      case "INSIDE_STRING": {
-        switch (char) {
-          case '"': {
-            stack.pop();
-            lastValidIndex = i;
-            break;
-          }
-          case "\\": {
-            stack.push("INSIDE_STRING_ESCAPE");
-            break;
-          }
-          default: {
-            lastValidIndex = i;
-          }
-        }
-        break;
-      }
-      case "INSIDE_ARRAY_START": {
-        switch (char) {
-          case "]": {
-            lastValidIndex = i;
-            stack.pop();
-            break;
-          }
-          default: {
-            lastValidIndex = i;
-            processValueStart(char, i, "INSIDE_ARRAY_AFTER_VALUE");
-            break;
-          }
-        }
-        break;
-      }
-      case "INSIDE_ARRAY_AFTER_VALUE": {
-        switch (char) {
-          case ",": {
-            stack.pop();
-            stack.push("INSIDE_ARRAY_AFTER_COMMA");
-            break;
-          }
-          case "]": {
-            lastValidIndex = i;
-            stack.pop();
-            break;
-          }
-          default: {
-            lastValidIndex = i;
-            break;
-          }
-        }
-        break;
-      }
-      case "INSIDE_ARRAY_AFTER_COMMA": {
-        processValueStart(char, i, "INSIDE_ARRAY_AFTER_VALUE");
-        break;
-      }
-      case "INSIDE_STRING_ESCAPE": {
-        stack.pop();
-        lastValidIndex = i;
-        break;
-      }
-      case "INSIDE_NUMBER": {
-        switch (char) {
-          case "0":
-          case "1":
-          case "2":
-          case "3":
-          case "4":
-          case "5":
-          case "6":
-          case "7":
-          case "8":
-          case "9": {
-            lastValidIndex = i;
-            break;
-          }
-          case "e":
-          case "E":
-          case "-":
-          case ".": {
-            break;
-          }
-          case ",": {
-            stack.pop();
-            if (stack[stack.length - 1] === "INSIDE_ARRAY_AFTER_VALUE") {
-              processAfterArrayValue(char, i);
-            }
-            if (stack[stack.length - 1] === "INSIDE_OBJECT_AFTER_VALUE") {
-              processAfterObjectValue(char, i);
-            }
-            break;
-          }
-          case "}": {
-            stack.pop();
-            if (stack[stack.length - 1] === "INSIDE_OBJECT_AFTER_VALUE") {
-              processAfterObjectValue(char, i);
-            }
-            break;
-          }
-          case "]": {
-            stack.pop();
-            if (stack[stack.length - 1] === "INSIDE_ARRAY_AFTER_VALUE") {
-              processAfterArrayValue(char, i);
-            }
-            break;
-          }
-          default: {
-            stack.pop();
-            break;
-          }
-        }
-        break;
-      }
-      case "INSIDE_LITERAL": {
-        const partialLiteral = input.substring(literalStart, i + 1);
-        if (!"false".startsWith(partialLiteral) && !"true".startsWith(partialLiteral) && !"null".startsWith(partialLiteral)) {
-          stack.pop();
-          if (stack[stack.length - 1] === "INSIDE_OBJECT_AFTER_VALUE") {
-            processAfterObjectValue(char, i);
-          } else if (stack[stack.length - 1] === "INSIDE_ARRAY_AFTER_VALUE") {
-            processAfterArrayValue(char, i);
-          }
-        } else {
-          lastValidIndex = i;
-        }
-        break;
-      }
-    }
-  }
-  let result = input.slice(0, lastValidIndex + 1);
-  for (let i = stack.length - 1; i >= 0; i--) {
-    const state = stack[i];
-    switch (state) {
-      case "INSIDE_STRING": {
-        result += '"';
-        break;
-      }
-      case "INSIDE_OBJECT_KEY":
-      case "INSIDE_OBJECT_AFTER_KEY":
-      case "INSIDE_OBJECT_AFTER_COMMA":
-      case "INSIDE_OBJECT_START":
-      case "INSIDE_OBJECT_BEFORE_VALUE":
-      case "INSIDE_OBJECT_AFTER_VALUE": {
-        result += "}";
-        break;
-      }
-      case "INSIDE_ARRAY_START":
-      case "INSIDE_ARRAY_AFTER_COMMA":
-      case "INSIDE_ARRAY_AFTER_VALUE": {
-        result += "]";
-        break;
-      }
-      case "INSIDE_LITERAL": {
-        const partialLiteral = input.substring(literalStart, input.length);
-        if ("true".startsWith(partialLiteral)) {
-          result += "true".slice(partialLiteral.length);
-        } else if ("false".startsWith(partialLiteral)) {
-          result += "false".slice(partialLiteral.length);
-        } else if ("null".startsWith(partialLiteral)) {
-          result += "null".slice(partialLiteral.length);
-        }
-      }
-    }
-  }
-  return result;
-}
-
-// src/runtimes/edge/partial-json/parse-partial-json.ts
-var parsePartialJson = (json) => {
-  try {
-    return sjson.parse(json);
-  } catch {
-    try {
-      return sjson.parse(fixJson(json));
-    } catch {
-      return void 0;
-    }
-  }
-};
-
-// src/runtimes/edge/streams/runResultStream.ts
-function runResultStream() {
-  let message = {
-    content: [],
-    status: { type: "running" }
-  };
-  const currentToolCall = { toolCallId: "", argsText: "" };
-  return new TransformStream({
-    transform(chunk, controller) {
-      const chunkType = chunk.type;
-      switch (chunkType) {
-        case "text-delta": {
-          message = appendOrUpdateText(message, chunk.textDelta);
-          controller.enqueue(message);
-          break;
-        }
-        case "tool-call-delta": {
-          const { toolCallId, toolName, argsTextDelta } = chunk;
-          if (currentToolCall.toolCallId !== toolCallId) {
-            currentToolCall.toolCallId = toolCallId;
-            currentToolCall.argsText = argsTextDelta;
-          } else {
-            currentToolCall.argsText += argsTextDelta;
-          }
-          message = appendOrUpdateToolCall(
-            message,
-            toolCallId,
-            toolName,
-            currentToolCall.argsText
-          );
-          controller.enqueue(message);
-          break;
-        }
-        case "tool-call": {
-          break;
-        }
-        case "tool-result": {
-          message = appendOrUpdateToolResult(
-            message,
-            chunk.toolCallId,
-            chunk.toolName,
-            chunk.result
-          );
-          controller.enqueue(message);
-          break;
-        }
-        case "finish": {
-          message = appendOrUpdateFinish(message, chunk);
-          controller.enqueue(message);
-          break;
-        }
-        case "error": {
-          if (chunk.error instanceof Error && chunk.error.name === "AbortError") {
-            message = appendOrUpdateCancel(message);
-            controller.enqueue(message);
-            break;
-          } else {
-            throw chunk.error;
-          }
-        }
-        default: {
-          const unhandledType = chunkType;
-          throw new Error(`Unhandled chunk type: ${unhandledType}`);
-        }
-      }
-    }
-  });
-}
-var appendOrUpdateText = (message, textDelta) => {
-  let contentParts = message.content;
-  let contentPart = message.content.at(-1);
-  if (contentPart?.type !== "text") {
-    contentPart = { type: "text", text: textDelta };
-  } else {
-    contentParts = contentParts.slice(0, -1);
-    contentPart = { type: "text", text: contentPart.text + textDelta };
-  }
-  return {
-    ...message,
-    content: contentParts.concat([contentPart])
-  };
-};
-var appendOrUpdateToolCall = (message, toolCallId, toolName, argsText) => {
-  let contentParts = message.content;
-  let contentPart = message.content.at(-1);
-  if (contentPart?.type !== "tool-call" || contentPart.toolCallId !== toolCallId) {
-    contentPart = {
-      type: "tool-call",
-      toolCallId,
-      toolName,
-      argsText,
-      args: parsePartialJson(argsText)
-    };
-  } else {
-    contentParts = contentParts.slice(0, -1);
-    contentPart = {
-      ...contentPart,
-      argsText,
-      args: parsePartialJson(argsText)
-    };
-  }
-  return {
-    ...message,
-    content: contentParts.concat([contentPart])
-  };
-};
-var appendOrUpdateToolResult = (message, toolCallId, toolName, result) => {
-  let found = false;
-  const newContentParts = message.content.map((part) => {
-    if (part.type !== "tool-call" || part.toolCallId !== toolCallId)
-      return part;
-    found = true;
-    if (part.toolName !== toolName)
-      throw new Error(
-        `Tool call ${toolCallId} found with tool name ${part.toolName}, but expected ${toolName}`
-      );
-    return {
-      ...part,
-      result
-    };
-  });
-  if (!found)
-    throw new Error(
-      `Received tool result for unknown tool call "${toolName}" / "${toolCallId}". This is likely an internal bug in assistant-ui.`
-    );
-  return {
-    ...message,
-    content: newContentParts
-  };
-};
-var appendOrUpdateFinish = (message, chunk) => {
-  const { type, ...rest } = chunk;
-  return {
-    ...message,
-    status: getStatus(chunk),
-    roundtrips: [
-      ...message.roundtrips ?? [],
-      {
-        logprobs: rest.logprobs,
-        usage: rest.usage
-      }
-    ]
-  };
-};
-var getStatus = (chunk) => {
-  if (chunk.finishReason === "tool-calls") {
-    return {
-      type: "requires-action",
-      reason: "tool-calls"
-    };
-  } else if (chunk.finishReason === "stop" || chunk.finishReason === "unknown") {
-    return {
-      type: "complete",
-      reason: chunk.finishReason
-    };
-  } else {
-    return {
-      type: "incomplete",
-      reason: chunk.finishReason
-    };
-  }
-};
-var appendOrUpdateCancel = (message) => {
-  return {
-    ...message,
-    status: {
-      type: "incomplete",
-      reason: "cancelled"
-    }
-  };
-};
-
-// src/runtimes/edge/streams/toolResultStream.ts
-import { z as z3 } from "zod";
-import sjson2 from "secure-json-parse";
-function toolResultStream(tools) {
-  const toolCallExecutions = /* @__PURE__ */ new Map();
-  return new TransformStream({
-    transform(chunk, controller) {
-      controller.enqueue(chunk);
-      const chunkType = chunk.type;
-      switch (chunkType) {
-        case "tool-call": {
-          const { toolCallId, toolCallType, toolName, args: argsText } = chunk;
-          const tool = tools?.[toolName];
-          if (!tool || !tool.execute) return;
-          const args = sjson2.parse(argsText);
-          if (tool.parameters instanceof z3.ZodType) {
-            const result = tool.parameters.safeParse(args);
-            if (!result.success) {
-              controller.enqueue({
-                type: "tool-result",
-                toolCallType,
-                toolCallId,
-                toolName,
-                result: "Function parameter validation failed. " + JSON.stringify(result.error.issues),
-                isError: true
-              });
-              return;
-            } else {
-              toolCallExecutions.set(
-                toolCallId,
-                (async () => {
-                  try {
-                    const result2 = await tool.execute(args);
-                    controller.enqueue({
-                      type: "tool-result",
-                      toolCallType,
-                      toolCallId,
-                      toolName,
-                      result: result2
-                    });
-                  } catch (error) {
-                    console.error("Error: ", error);
-                    controller.enqueue({
-                      type: "tool-result",
-                      toolCallType,
-                      toolCallId,
-                      toolName,
-                      result: "Error: " + error,
-                      isError: true
-                    });
-                  } finally {
-                    toolCallExecutions.delete(toolCallId);
-                  }
-                })()
-              );
-            }
-          }
-          break;
-        }
-        case "text-delta":
-        case "tool-call-delta":
-        case "tool-result":
-        case "finish":
-        case "error":
-          break;
-        default: {
-          const unhandledType = chunkType;
-          throw new Error(`Unhandled chunk type: ${unhandledType}`);
-        }
-      }
-    },
-    async flush() {
-      await Promise.all(toolCallExecutions.values());
-    }
-  });
-}
-
 // src/runtimes/edge/EdgeChatAdapter.ts
 function asAsyncIterable(source) {
   return {
@@ -3380,7 +2091,7 @@ var useEdgeRuntime = ({
   initialMessages,
   ...options
 }) => {
-  const [adapter] = useState7(() => new EdgeChatAdapter(options));
+  const [adapter] = useState6(() => new EdgeChatAdapter(options));
   return useLocalRuntime(adapter, { initialMessages });
 };
 
@@ -3594,7 +2305,7 @@ var LocalRuntime = class extends BaseAssistantRuntime {
 
 // src/runtimes/local/useLocalRuntime.tsx
 var useLocalRuntime = (adapter, options) => {
-  const [runtime] = useState8(() => new LocalRuntime(adapter, options));
+  const [runtime] = useState7(() => new LocalRuntime(adapter, options));
   useInsertionEffect3(() => {
     runtime.adapter = adapter;
   });
@@ -3611,7 +2322,7 @@ var getExternalStoreMessage = (message) => {
 };
 
 // src/runtimes/external-store/useExternalStoreSync.tsx
-import { useEffect as useEffect11, useInsertionEffect as useInsertionEffect4, useMemo as useMemo3, useRef as useRef6 } from "react";
+import { useEffect as useEffect10, useInsertionEffect as useInsertionEffect4, useMemo as useMemo2, useRef as useRef5 } from "react";
 
 // src/runtimes/external-store/ThreadMessageConverter.ts
 var ThreadMessageConverter = class {
@@ -3705,11 +2416,11 @@ var fromThreadMessageLike = (like, fallbackId, fallbackStatus) => {
 
 // src/runtimes/external-store/useExternalStoreSync.tsx
 var useExternalStoreSync = (adapter, updateData) => {
-  const adapterRef = useRef6(adapter);
+  const adapterRef = useRef5(adapter);
   useInsertionEffect4(() => {
     adapterRef.current = adapter;
   });
-  const [converter, convertCallback] = useMemo3(() => {
+  const [converter, convertCallback] = useMemo2(() => {
     const converter2 = adapter.convertMessage ?? ((m) => m);
     const convertCallback2 = (cache, m, idx) => {
       const autoStatus = getAutoStatus(
@@ -3728,7 +2439,7 @@ var useExternalStoreSync = (adapter, updateData) => {
     };
     return [new ThreadMessageConverter(), convertCallback2];
   }, [adapter.convertMessage]);
-  useEffect11(() => {
+  useEffect10(() => {
     updateData(
       adapter.isDisabled ?? false,
       adapter.isRunning ?? false,
@@ -3897,13 +2608,13 @@ var ExternalStoreRuntime = class extends BaseAssistantRuntime {
 };
 
 // src/runtimes/external-store/useExternalStoreRuntime.tsx
-import { useEffect as useEffect12, useInsertionEffect as useInsertionEffect5, useState as useState9 } from "react";
+import { useEffect as useEffect11, useInsertionEffect as useInsertionEffect5, useState as useState8 } from "react";
 var useExternalStoreRuntime = (store) => {
-  const [runtime] = useState9(() => new ExternalStoreRuntime(store));
+  const [runtime] = useState8(() => new ExternalStoreRuntime(store));
   useInsertionEffect5(() => {
     runtime.store = store;
   });
-  useEffect12(() => {
+  useEffect11(() => {
     runtime.onStoreUpdated();
   });
   return runtime;
@@ -3911,34 +2622,34 @@ var useExternalStoreRuntime = (store) => {
 
 // src/ui/thread-config.tsx
 import {
-  createContext as createContext5,
-  useContext as useContext5
+  createContext,
+  useContext
 } from "react";
-import { Fragment as Fragment3, jsx as jsx29 } from "react/jsx-runtime";
-var ThreadConfigContext = createContext5({});
+import { Fragment as Fragment3, jsx as jsx25 } from "react/jsx-runtime";
+var ThreadConfigContext = createContext({});
 var useThreadConfig = () => {
-  return useContext5(ThreadConfigContext);
+  return useContext(ThreadConfigContext);
 };
 var ThreadConfigProvider = ({
   children,
   config
 }) => {
   const assistant = useAssistantContext({ optional: true });
-  const configProvider = config && Object.keys(config ?? {}).length > 0 ? /* @__PURE__ */ jsx29(ThreadConfigContext.Provider, { value: config, children }) : /* @__PURE__ */ jsx29(Fragment3, { children });
+  const configProvider = config && Object.keys(config ?? {}).length > 0 ? /* @__PURE__ */ jsx25(ThreadConfigContext.Provider, { value: config, children }) : /* @__PURE__ */ jsx25(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__ */ jsx29(AssistantRuntimeProvider, { runtime: config.runtime, children: configProvider });
+  return /* @__PURE__ */ jsx25(AssistantRuntimeProvider, { runtime: config.runtime, children: configProvider });
 };
 ThreadConfigProvider.displayName = "ThreadConfigProvider";
 
 // src/ui/assistant-action-bar.tsx
-import { forwardRef as forwardRef18 } from "react";
+import { forwardRef as forwardRef15 } from "react";
 import { CheckIcon, CopyIcon, RefreshCwIcon } from "lucide-react";
-import { Fragment as Fragment4, jsx as jsx30, jsxs as jsxs5 } from "react/jsx-runtime";
+import { Fragment as Fragment4, jsx as jsx26, jsxs as jsxs4 } from "react/jsx-runtime";
 var useAllowCopy = () => {
   const { assistantMessage: { allowCopy = true } = {} } = useThreadConfig();
   const { useThreadActions } = useThreadContext();
@@ -3955,15 +2666,15 @@ var AssistantActionBar = () => {
   const allowCopy = useAllowCopy();
   const allowReload = useAllowReload();
   if (!allowCopy && !allowReload) return null;
-  return /* @__PURE__ */ jsxs5(
+  return /* @__PURE__ */ jsxs4(
     AssistantActionBarRoot,
     {
       hideWhenRunning: true,
       autohide: "not-last",
       autohideFloat: "single-branch",
       children: [
-        /* @__PURE__ */ jsx30(AssistantActionBarCopy, {}),
-        /* @__PURE__ */ jsx30(AssistantActionBarReload, {})
+        /* @__PURE__ */ jsx26(AssistantActionBarCopy, {}),
+        /* @__PURE__ */ jsx26(AssistantActionBarReload, {})
       ]
     }
   );
@@ -3973,7 +2684,7 @@ var AssistantActionBarRoot = withDefaults(actionBar_exports.Root, {
   className: "aui-assistant-action-bar-root"
 });
 AssistantActionBarRoot.displayName = "AssistantActionBarRoot";
-var AssistantActionBarCopy = forwardRef18((props, ref) => {
+var AssistantActionBarCopy = forwardRef15((props, ref) => {
   const {
     strings: {
       assistantMessage: { reload: { tooltip = "Copy" } = {} } = {}
@@ -3981,13 +2692,13 @@ var AssistantActionBarCopy = forwardRef18((props, ref) => {
   } = useThreadConfig();
   const allowCopy = useAllowCopy();
   if (!allowCopy) return null;
-  return /* @__PURE__ */ jsx30(actionBar_exports.Copy, { asChild: true, children: /* @__PURE__ */ jsx30(TooltipIconButton, { tooltip, ...props, ref, children: props.children ?? /* @__PURE__ */ jsxs5(Fragment4, { children: [
-    /* @__PURE__ */ jsx30(message_exports.If, { copied: true, children: /* @__PURE__ */ jsx30(CheckIcon, {}) }),
-    /* @__PURE__ */ jsx30(message_exports.If, { copied: false, children: /* @__PURE__ */ jsx30(CopyIcon, {}) })
+  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, {}) })
   ] }) }) });
 });
 AssistantActionBarCopy.displayName = "AssistantActionBarCopy";
-var AssistantActionBarReload = forwardRef18((props, ref) => {
+var AssistantActionBarReload = forwardRef15((props, ref) => {
   const {
     strings: {
       assistantMessage: { reload: { tooltip = "Refresh" } = {} } = {}
@@ -3995,7 +2706,7 @@ var AssistantActionBarReload = forwardRef18((props, ref) => {
   } = useThreadConfig();
   const allowReload = useAllowReload();
   if (!allowReload) return null;
-  return /* @__PURE__ */ jsx30(actionBar_exports.Reload, { asChild: true, children: /* @__PURE__ */ jsx30(TooltipIconButton, { tooltip, ...props, ref, children: /* @__PURE__ */ jsx30(RefreshCwIcon, {}) }) });
+  return /* @__PURE__ */ jsx26(actionBar_exports.Reload, { asChild: true, children: /* @__PURE__ */ jsx26(TooltipIconButton, { tooltip, ...props, ref, children: /* @__PURE__ */ jsx26(RefreshCwIcon, {}) }) });
 });
 AssistantActionBarReload.displayName = "AssistantActionBarReload";
 var exports = {
@@ -4009,12 +2720,12 @@ var assistant_action_bar_default = Object.assign(
 );
 
 // src/ui/assistant-message.tsx
-import { forwardRef as forwardRef20 } from "react";
+import { forwardRef as forwardRef17 } from "react";
 
 // src/ui/branch-picker.tsx
-import { forwardRef as forwardRef19 } from "react";
+import { forwardRef as forwardRef16 } from "react";
 import { ChevronLeftIcon, ChevronRightIcon } from "lucide-react";
-import { jsx as jsx31, jsxs as jsxs6 } from "react/jsx-runtime";
+import { jsx as jsx27, jsxs as jsxs5 } from "react/jsx-runtime";
 var useAllowBranchPicker = () => {
   const { branchPicker: { allowBranchPicker = true } = {} } = useThreadConfig();
   const { useThreadActions } = useThreadContext();
@@ -4024,10 +2735,10 @@ var useAllowBranchPicker = () => {
 var BranchPicker = () => {
   const allowBranchPicker = useAllowBranchPicker();
   if (!allowBranchPicker) return null;
-  return /* @__PURE__ */ jsxs6(BranchPickerRoot, { hideWhenSingleBranch: true, children: [
-    /* @__PURE__ */ jsx31(BranchPickerPrevious2, {}),
-    /* @__PURE__ */ jsx31(BranchPickerState, {}),
-    /* @__PURE__ */ jsx31(BranchPickerNext, {})
+  return /* @__PURE__ */ jsxs5(BranchPickerRoot, { hideWhenSingleBranch: true, children: [
+    /* @__PURE__ */ jsx27(BranchPickerPrevious2, {}),
+    /* @__PURE__ */ jsx27(BranchPickerState, {}),
+    /* @__PURE__ */ jsx27(BranchPickerNext, {})
   ] });
 };
 BranchPicker.displayName = "BranchPicker";
@@ -4035,31 +2746,31 @@ var BranchPickerRoot = withDefaults(branchPicker_exports.Root, {
   className: "aui-branch-picker-root"
 });
 BranchPickerRoot.displayName = "BranchPickerRoot";
-var BranchPickerPrevious2 = forwardRef19((props, ref) => {
+var BranchPickerPrevious2 = forwardRef16((props, ref) => {
   const {
     strings: {
       branchPicker: { previous: { tooltip = "Previous" } = {} } = {}
     } = {}
   } = useThreadConfig();
-  return /* @__PURE__ */ jsx31(branchPicker_exports.Previous, { asChild: true, children: /* @__PURE__ */ jsx31(TooltipIconButton, { tooltip, ...props, ref, children: props.children ?? /* @__PURE__ */ jsx31(ChevronLeftIcon, {}) }) });
+  return /* @__PURE__ */ jsx27(branchPicker_exports.Previous, { asChild: true, children: /* @__PURE__ */ jsx27(TooltipIconButton, { tooltip, ...props, ref, children: props.children ?? /* @__PURE__ */ jsx27(ChevronLeftIcon, {}) }) });
 });
 BranchPickerPrevious2.displayName = "BranchPickerPrevious";
 var BranchPickerStateWrapper = withDefaults("span", {
   className: "aui-branch-picker-state"
 });
-var BranchPickerState = forwardRef19((props, ref) => {
-  return /* @__PURE__ */ jsxs6(BranchPickerStateWrapper, { ...props, ref, children: [
-    /* @__PURE__ */ jsx31(branchPicker_exports.Number, {}),
+var BranchPickerState = forwardRef16((props, ref) => {
+  return /* @__PURE__ */ jsxs5(BranchPickerStateWrapper, { ...props, ref, children: [
+    /* @__PURE__ */ jsx27(branchPicker_exports.Number, {}),
     " / ",
-    /* @__PURE__ */ jsx31(branchPicker_exports.Count, {})
+    /* @__PURE__ */ jsx27(branchPicker_exports.Count, {})
   ] });
 });
 BranchPickerState.displayName = "BranchPickerState";
-var BranchPickerNext = forwardRef19((props, ref) => {
+var BranchPickerNext = forwardRef16((props, ref) => {
   const {
     strings: { branchPicker: { next: { tooltip = "Next" } = {} } = {} } = {}
   } = useThreadConfig();
-  return /* @__PURE__ */ jsx31(branchPicker_exports.Next, { asChild: true, children: /* @__PURE__ */ jsx31(TooltipIconButton, { tooltip, ...props, ref, children: props.children ?? /* @__PURE__ */ jsx31(ChevronRightIcon, {}) }) });
+  return /* @__PURE__ */ jsx27(branchPicker_exports.Next, { asChild: true, children: /* @__PURE__ */ jsx27(TooltipIconButton, { tooltip, ...props, ref, children: props.children ?? /* @__PURE__ */ jsx27(ChevronRightIcon, {}) }) });
 });
 BranchPickerNext.displayName = "BranchPickerNext";
 var exports2 = {
@@ -4071,12 +2782,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 jsx32, jsxs as jsxs7 } from "react/jsx-runtime";
+import { jsx as jsx28, 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__ */ jsx32(AvatarImage, { src, alt }),
-    fallback != null && /* @__PURE__ */ jsx32(AvatarFallback, { children: fallback })
+  return /* @__PURE__ */ jsxs6(AvatarRoot, { children: [
+    src != null && /* @__PURE__ */ jsx28(AvatarImage, { src, alt }),
+    fallback != null && /* @__PURE__ */ jsx28(AvatarFallback, { children: fallback })
   ] });
 };
 Avatar.displayName = "Avatar";
@@ -4094,37 +2805,38 @@ var AvatarFallback = withDefaults(AvatarPrimitive.Fallback, {
 AvatarFallback.displayName = "AvatarFallback";
 
 // src/ui/content-part.tsx
-import classNames2 from "classnames";
-import { jsx as jsx33 } from "react/jsx-runtime";
-var Text = ({ status }) => {
-  return /* @__PURE__ */ jsx33(
-    "p",
+import classNames from "classnames";
+import { jsx as jsx29 } from "react/jsx-runtime";
+var Text = () => {
+  const status = useSmoothStatus();
+  return /* @__PURE__ */ jsx29(
+    contentPart_exports.Text,
     {
-      className: classNames2(
+      className: classNames(
         "aui-text",
-        status.type === "running" && "aui-text-in-progress"
+        status.type === "running" && "aui-text-running"
       ),
-      children: /* @__PURE__ */ jsx33(contentPart_exports.Text, {})
+      component: "p"
     }
   );
 };
-var exports3 = { Text };
+var exports3 = { Text: withSmoothContextProvider(Text) };
 var content_part_default = exports3;
 
 // src/ui/assistant-message.tsx
-import { jsx as jsx34, jsxs as jsxs8 } from "react/jsx-runtime";
+import { jsx as jsx30, jsxs as jsxs7 } from "react/jsx-runtime";
 var AssistantMessage = () => {
-  return /* @__PURE__ */ jsxs8(AssistantMessageRoot, { children: [
-    /* @__PURE__ */ jsx34(AssistantMessageAvatar, {}),
-    /* @__PURE__ */ jsx34(AssistantMessageContent, {}),
-    /* @__PURE__ */ jsx34(branch_picker_default, {}),
-    /* @__PURE__ */ jsx34(assistant_action_bar_default, {})
+  return /* @__PURE__ */ jsxs7(AssistantMessageRoot, { children: [
+    /* @__PURE__ */ jsx30(AssistantMessageAvatar, {}),
+    /* @__PURE__ */ jsx30(AssistantMessageContent, {}),
+    /* @__PURE__ */ jsx30(branch_picker_default, {}),
+    /* @__PURE__ */ jsx30(assistant_action_bar_default, {})
   ] });
 };
 AssistantMessage.displayName = "AssistantMessage";
 var AssistantMessageAvatar = () => {
   const { assistantAvatar: avatar = { fallback: "A" } } = useThreadConfig();
-  return /* @__PURE__ */ jsx34(Avatar, { ...avatar });
+  return /* @__PURE__ */ jsx30(Avatar, { ...avatar });
 };
 var AssistantMessageRoot = withDefaults(message_exports.Root, {
   className: "aui-assistant-message-root"
@@ -4133,9 +2845,9 @@ AssistantMessageRoot.displayName = "AssistantMessageRoot";
 var AssistantMessageContentWrapper = withDefaults("div", {
   className: "aui-assistant-message-content"
 });
-var AssistantMessageContent = forwardRef20(({ components: componentsProp, ...rest }, ref) => {
+var AssistantMessageContent = forwardRef17(({ components: componentsProp, ...rest }, ref) => {
   const { assistantMessage: { components = {} } = {} } = useThreadConfig();
-  return /* @__PURE__ */ jsx34(AssistantMessageContentWrapper, { ...rest, ref, children: /* @__PURE__ */ jsx34(
+  return /* @__PURE__ */ jsx30(AssistantMessageContentWrapper, { ...rest, ref, children: /* @__PURE__ */ jsx30(
     message_exports.Content,
     {
       components: {
@@ -4157,38 +2869,38 @@ var assistant_message_default = Object.assign(
 );
 
 // src/ui/assistant-modal.tsx
-import { forwardRef as forwardRef27 } from "react";
+import { forwardRef as forwardRef24 } from "react";
 import { BotIcon, ChevronDownIcon } from "lucide-react";
 
 // src/ui/thread.tsx
-import { forwardRef as forwardRef26 } from "react";
+import { forwardRef as forwardRef23 } from "react";
 import { ArrowDownIcon } from "lucide-react";
 
 // src/ui/composer.tsx
-import { forwardRef as forwardRef21 } from "react";
+import { forwardRef as forwardRef18 } from "react";
 import { SendHorizontalIcon } from "lucide-react";
 
 // src/ui/base/CircleStopIcon.tsx
-import { jsx as jsx35 } from "react/jsx-runtime";
+import { jsx as jsx31 } from "react/jsx-runtime";
 var CircleStopIcon = () => {
-  return /* @__PURE__ */ jsx35(
+  return /* @__PURE__ */ jsx31(
     "svg",
     {
       xmlns: "http://www.w3.org/2000/svg",
       viewBox: "0 0 16 16",
       fill: "currentColor",
-      children: /* @__PURE__ */ jsx35("rect", { width: "10", height: "10", x: "3", y: "3", rx: "2" })
+      children: /* @__PURE__ */ jsx31("rect", { width: "10", height: "10", x: "3", y: "3", rx: "2" })
     }
   );
 };
 CircleStopIcon.displayName = "CircleStopIcon";
 
 // src/ui/composer.tsx
-import { Fragment as Fragment5, jsx as jsx36, jsxs as jsxs9 } from "react/jsx-runtime";
+import { Fragment as Fragment5, jsx as jsx32, jsxs as jsxs8 } from "react/jsx-runtime";
 var Composer = () => {
-  return /* @__PURE__ */ jsxs9(ComposerRoot, { children: [
-    /* @__PURE__ */ jsx36(ComposerInput, { autoFocus: true }),
-    /* @__PURE__ */ jsx36(ComposerAction, {})
+  return /* @__PURE__ */ jsxs8(ComposerRoot, { children: [
+    /* @__PURE__ */ jsx32(ComposerInput, { autoFocus: true }),
+    /* @__PURE__ */ jsx32(ComposerAction, {})
   ] });
 };
 Composer.displayName = "Composer";
@@ -4201,14 +2913,14 @@ var ComposerInputStyled = withDefaults(composer_exports.Input, {
   autoFocus: true,
   className: "aui-composer-input"
 });
-var ComposerInput = forwardRef21(
+var ComposerInput = forwardRef18(
   (props, ref) => {
     const {
       strings: {
         composer: { input: { placeholder = "Write a message..." } = {} } = {}
       } = {}
     } = useThreadConfig();
-    return /* @__PURE__ */ jsx36(ComposerInputStyled, { placeholder, ...props, ref });
+    return /* @__PURE__ */ jsx32(ComposerInputStyled, { placeholder, ...props, ref });
   }
 );
 ComposerInput.displayName = "ComposerInput";
@@ -4219,10 +2931,10 @@ var useAllowCancel = () => {
 };
 var ComposerAction = () => {
   const allowCancel = useAllowCancel();
-  if (!allowCancel) return /* @__PURE__ */ jsx36(ComposerSend, {});
-  return /* @__PURE__ */ jsxs9(Fragment5, { children: [
-    /* @__PURE__ */ jsx36(thread_exports.If, { running: false, children: /* @__PURE__ */ jsx36(ComposerSend, {}) }),
-    /* @__PURE__ */ jsx36(thread_exports.If, { running: true, children: /* @__PURE__ */ jsx36(ComposerCancel, {}) })
+  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, {}) })
   ] });
 };
 ComposerAction.displayName = "ComposerAction";
@@ -4230,22 +2942,22 @@ var ComposerSendButton = withDefaults(TooltipIconButton, {
   variant: "default",
   className: "aui-composer-send"
 });
-var ComposerSend = forwardRef21((props, ref) => {
+var ComposerSend = forwardRef18((props, ref) => {
   const {
     strings: { composer: { send: { tooltip = "Send" } = {} } = {} } = {}
   } = useThreadConfig();
-  return /* @__PURE__ */ jsx36(composer_exports.Send, { asChild: true, children: /* @__PURE__ */ jsx36(ComposerSendButton, { tooltip, ...props, ref, children: props.children ?? /* @__PURE__ */ jsx36(SendHorizontalIcon, {}) }) });
+  return /* @__PURE__ */ jsx32(composer_exports.Send, { asChild: true, children: /* @__PURE__ */ jsx32(ComposerSendButton, { tooltip, ...props, ref, children: props.children ?? /* @__PURE__ */ jsx32(SendHorizontalIcon, {}) }) });
 });
 ComposerSend.displayName = "ComposerSend";
 var ComposerCancelButton = withDefaults(TooltipIconButton, {
   variant: "default",
   className: "aui-composer-cancel"
 });
-var ComposerCancel = forwardRef21((props, ref) => {
+var ComposerCancel = forwardRef18((props, ref) => {
   const {
     strings: { composer: { cancel: { tooltip = "Cancel" } = {} } = {} } = {}
   } = useThreadConfig();
-  return /* @__PURE__ */ jsx36(composer_exports.Cancel, { asChild: true, children: /* @__PURE__ */ jsx36(ComposerCancelButton, { tooltip, ...props, ref, children: props.children ?? /* @__PURE__ */ jsx36(CircleStopIcon, {}) }) });
+  return /* @__PURE__ */ jsx32(composer_exports.Cancel, { asChild: true, children: /* @__PURE__ */ jsx32(ComposerCancelButton, { tooltip, ...props, ref, children: props.children ?? /* @__PURE__ */ jsx32(CircleStopIcon, {}) }) });
 });
 ComposerCancel.displayName = "ComposerCancel";
 var exports5 = {
@@ -4258,15 +2970,15 @@ var exports5 = {
 var composer_default = Object.assign(Composer, exports5);
 
 // src/ui/thread-welcome.tsx
-import { forwardRef as forwardRef22 } from "react";
-import { jsx as jsx37, jsxs as jsxs10 } from "react/jsx-runtime";
+import { forwardRef as forwardRef19 } from "react";
+import { jsx as jsx33, jsxs as jsxs9 } from "react/jsx-runtime";
 var ThreadWelcome = () => {
-  return /* @__PURE__ */ jsxs10(ThreadWelcomeRoot, { children: [
-    /* @__PURE__ */ jsxs10(ThreadWelcomeCenter, { children: [
-      /* @__PURE__ */ jsx37(ThreadWelcomeAvatar, {}),
-      /* @__PURE__ */ jsx37(ThreadWelcomeMessage, {})
+  return /* @__PURE__ */ jsxs9(ThreadWelcomeRoot, { children: [
+    /* @__PURE__ */ jsxs9(ThreadWelcomeCenter, { children: [
+      /* @__PURE__ */ jsx33(ThreadWelcomeAvatar, {}),
+      /* @__PURE__ */ jsx33(ThreadWelcomeMessage, {})
     ] }),
-    /* @__PURE__ */ jsx37(ThreadWelcomeSuggestions, {})
+    /* @__PURE__ */ jsx33(ThreadWelcomeSuggestions, {})
   ] });
 };
 ThreadWelcome.displayName = "ThreadWelcome";
@@ -4276,22 +2988,22 @@ var ThreadWelcomeRootStyled = withDefaults("div", {
 var ThreadWelcomeCenter = withDefaults("div", {
   className: "aui-thread-welcome-center"
 });
-var ThreadWelcomeRoot = forwardRef22(
+var ThreadWelcomeRoot = forwardRef19(
   (props, ref) => {
-    return /* @__PURE__ */ jsx37(thread_exports.Empty, { children: /* @__PURE__ */ jsx37(ThreadWelcomeRootStyled, { ...props, ref }) });
+    return /* @__PURE__ */ jsx33(thread_exports.Empty, { children: /* @__PURE__ */ jsx33(ThreadWelcomeRootStyled, { ...props, ref }) });
   }
 );
 ThreadWelcomeRoot.displayName = "ThreadWelcomeRoot";
 var ThreadWelcomeAvatar = () => {
   const { assistantAvatar: avatar = { fallback: "A" } } = useThreadConfig();
-  return /* @__PURE__ */ jsx37(Avatar, { ...avatar });
+  return /* @__PURE__ */ jsx33(Avatar, { ...avatar });
 };
 var ThreadWelcomeMessageStyled = withDefaults("p", {
   className: "aui-thread-welcome-message"
 });
-var ThreadWelcomeMessage = forwardRef22(({ message: messageProp, ...rest }, ref) => {
+var ThreadWelcomeMessage = forwardRef19(({ message: messageProp, ...rest }, ref) => {
   const { welcome: { message = "How can I help you today?" } = {} } = useThreadConfig();
-  return /* @__PURE__ */ jsx37(ThreadWelcomeMessageStyled, { ...rest, ref, children: messageProp ?? message });
+  return /* @__PURE__ */ jsx33(ThreadWelcomeMessageStyled, { ...rest, ref, children: messageProp ?? message });
 });
 ThreadWelcomeMessage.displayName = "ThreadWelcomeMessage";
 var ThreadWelcomeSuggestionContainer = withDefaults("div", {
@@ -4303,21 +3015,21 @@ var ThreadWelcomeSuggestionStyled = withDefaults(thread_exports.Suggestion, {
 var ThreadWelcomeSuggestion = ({
   suggestion: { text, prompt }
 }) => {
-  return /* @__PURE__ */ jsx37(
+  return /* @__PURE__ */ jsx33(
     ThreadWelcomeSuggestionStyled,
     {
       prompt,
       method: "replace",
       autoSend: true,
-      children: /* @__PURE__ */ jsx37("span", { className: "aui-thread-welcome-suggestion-text", children: text ?? prompt })
+      children: /* @__PURE__ */ jsx33("span", { className: "aui-thread-welcome-suggestion-text", children: text ?? prompt })
     }
   );
 };
 var ThreadWelcomeSuggestions = () => {
   const { welcome: { suggestions } = {} } = useThreadConfig();
-  return /* @__PURE__ */ jsx37(ThreadWelcomeSuggestionContainer, { children: suggestions?.map((suggestion, idx) => {
+  return /* @__PURE__ */ jsx33(ThreadWelcomeSuggestionContainer, { children: suggestions?.map((suggestion, idx) => {
     const key = `${suggestion.prompt}-${idx}`;
-    return /* @__PURE__ */ jsx37(ThreadWelcomeSuggestion, { suggestion }, key);
+    return /* @__PURE__ */ jsx33(ThreadWelcomeSuggestion, { suggestion }, key);
   }) });
 };
 ThreadWelcomeSuggestions.displayName = "ThreadWelcomeSuggestions";
@@ -4332,12 +3044,12 @@ var exports6 = {
 var thread_welcome_default = Object.assign(ThreadWelcome, exports6);
 
 // src/ui/user-message.tsx
-import { forwardRef as forwardRef24 } from "react";
+import { forwardRef as forwardRef21 } from "react";
 
 // src/ui/user-action-bar.tsx
-import { forwardRef as forwardRef23 } from "react";
+import { forwardRef as forwardRef20 } from "react";
 import { PencilIcon } from "lucide-react";
-import { jsx as jsx38 } from "react/jsx-runtime";
+import { jsx as jsx34 } from "react/jsx-runtime";
 var useAllowEdit = () => {
   const { userMessage: { allowEdit = true } = {} } = useThreadConfig();
   const { useThreadActions } = useThreadContext();
@@ -4347,20 +3059,20 @@ var useAllowEdit = () => {
 var UserActionBar = () => {
   const allowEdit = useAllowEdit();
   if (!allowEdit) return null;
-  return /* @__PURE__ */ jsx38(UserActionBarRoot, { hideWhenRunning: true, autohide: "not-last", children: /* @__PURE__ */ jsx38(UserActionBarEdit, {}) });
+  return /* @__PURE__ */ jsx34(UserActionBarRoot, { hideWhenRunning: true, autohide: "not-last", children: /* @__PURE__ */ jsx34(UserActionBarEdit, {}) });
 };
 UserActionBar.displayName = "UserActionBar";
 var UserActionBarRoot = withDefaults(actionBar_exports.Root, {
   className: "aui-user-action-bar-root"
 });
 UserActionBarRoot.displayName = "UserActionBarRoot";
-var UserActionBarEdit = forwardRef23((props, ref) => {
+var UserActionBarEdit = forwardRef20((props, ref) => {
   const {
     strings: { userMessage: { edit: { tooltip = "Edit" } = {} } = {} } = {}
   } = useThreadConfig();
   const allowEdit = useAllowEdit();
   if (!allowEdit) return null;
-  return /* @__PURE__ */ jsx38(actionBar_exports.Edit, { asChild: true, children: /* @__PURE__ */ jsx38(TooltipIconButton, { tooltip, ...props, ref, children: props.children ?? /* @__PURE__ */ jsx38(PencilIcon, {}) }) });
+  return /* @__PURE__ */ jsx34(actionBar_exports.Edit, { asChild: true, children: /* @__PURE__ */ jsx34(TooltipIconButton, { tooltip, ...props, ref, children: props.children ?? /* @__PURE__ */ jsx34(PencilIcon, {}) }) });
 });
 UserActionBarEdit.displayName = "UserActionBarEdit";
 var exports7 = {
@@ -4370,12 +3082,12 @@ var exports7 = {
 var user_action_bar_default = Object.assign(UserActionBar, exports7);
 
 // src/ui/user-message.tsx
-import { jsx as jsx39, jsxs as jsxs11 } from "react/jsx-runtime";
+import { jsx as jsx35, jsxs as jsxs10 } from "react/jsx-runtime";
 var UserMessage = () => {
-  return /* @__PURE__ */ jsxs11(UserMessageRoot, { children: [
-    /* @__PURE__ */ jsx39(user_action_bar_default, {}),
-    /* @__PURE__ */ jsx39(UserMessageContent, {}),
-    /* @__PURE__ */ jsx39(branch_picker_default, {})
+  return /* @__PURE__ */ jsxs10(UserMessageRoot, { children: [
+    /* @__PURE__ */ jsx35(user_action_bar_default, {}),
+    /* @__PURE__ */ jsx35(UserMessageContent, {}),
+    /* @__PURE__ */ jsx35(branch_picker_default, {})
   ] });
 };
 UserMessage.displayName = "UserMessage";
@@ -4386,9 +3098,9 @@ UserMessageRoot.displayName = "UserMessageRoot";
 var UserMessageContentWrapper = withDefaults("div", {
   className: "aui-user-message-content"
 });
-var UserMessageContent = forwardRef24(
+var UserMessageContent = forwardRef21(
   ({ components, ...props }, ref) => {
-    return /* @__PURE__ */ jsx39(UserMessageContentWrapper, { ...props, ref, children: /* @__PURE__ */ jsx39(
+    return /* @__PURE__ */ jsx35(UserMessageContentWrapper, { ...props, ref, children: /* @__PURE__ */ jsx35(
       message_exports.Content,
       {
         components: {
@@ -4407,14 +3119,14 @@ var exports8 = {
 var user_message_default = Object.assign(UserMessage, exports8);
 
 // src/ui/edit-composer.tsx
-import { forwardRef as forwardRef25 } from "react";
-import { jsx as jsx40, jsxs as jsxs12 } from "react/jsx-runtime";
+import { forwardRef as forwardRef22 } from "react";
+import { jsx as jsx36, jsxs as jsxs11 } from "react/jsx-runtime";
 var EditComposer = () => {
-  return /* @__PURE__ */ jsxs12(EditComposerRoot, { children: [
-    /* @__PURE__ */ jsx40(EditComposerInput, {}),
-    /* @__PURE__ */ jsxs12(EditComposerFooter, { children: [
-      /* @__PURE__ */ jsx40(EditComposerCancel, {}),
-      /* @__PURE__ */ jsx40(EditComposerSend, {})
+  return /* @__PURE__ */ jsxs11(EditComposerRoot, { children: [
+    /* @__PURE__ */ jsx36(EditComposerInput, {}),
+    /* @__PURE__ */ jsxs11(EditComposerFooter, { children: [
+      /* @__PURE__ */ jsx36(EditComposerCancel, {}),
+      /* @__PURE__ */ jsx36(EditComposerSend, {})
     ] })
   ] });
 };
@@ -4431,23 +3143,23 @@ var EditComposerFooter = withDefaults("div", {
   className: "aui-edit-composer-footer"
 });
 EditComposerFooter.displayName = "EditComposerFooter";
-var EditComposerCancel = forwardRef25(
+var EditComposerCancel = forwardRef22(
   (props, ref) => {
     const {
       strings: {
         editComposer: { cancel: { label = "Cancel" } = {} } = {}
       } = {}
     } = useThreadConfig();
-    return /* @__PURE__ */ jsx40(composer_exports.Cancel, { asChild: true, children: /* @__PURE__ */ jsx40(Button, { variant: "ghost", ...props, ref, children: props.children ?? label }) });
+    return /* @__PURE__ */ jsx36(composer_exports.Cancel, { asChild: true, children: /* @__PURE__ */ jsx36(Button, { variant: "ghost", ...props, ref, children: props.children ?? label }) });
   }
 );
 EditComposerCancel.displayName = "EditComposerCancel";
-var EditComposerSend = forwardRef25(
+var EditComposerSend = forwardRef22(
   (props, ref) => {
     const {
       strings: { editComposer: { send: { label = "Send" } = {} } = {} } = {}
     } = useThreadConfig();
-    return /* @__PURE__ */ jsx40(composer_exports.Send, { asChild: true, children: /* @__PURE__ */ jsx40(Button, { ...props, ref, children: props.children ?? label }) });
+    return /* @__PURE__ */ jsx36(composer_exports.Send, { asChild: true, children: /* @__PURE__ */ jsx36(Button, { ...props, ref, children: props.children ?? label }) });
   }
 );
 EditComposerSend.displayName = "EditComposerSend";
@@ -4461,23 +3173,23 @@ var exports9 = {
 var edit_composer_default = Object.assign(EditComposer, exports9);
 
 // src/ui/thread.tsx
-import { jsx as jsx41, jsxs as jsxs13 } from "react/jsx-runtime";
+import { jsx as jsx37, jsxs as jsxs12 } from "react/jsx-runtime";
 var Thread = (config) => {
-  return /* @__PURE__ */ jsx41(ThreadRoot, { config, children: /* @__PURE__ */ jsxs13(ThreadViewport, { children: [
-    /* @__PURE__ */ jsx41(thread_welcome_default, {}),
-    /* @__PURE__ */ jsx41(ThreadMessages, {}),
-    /* @__PURE__ */ jsxs13(ThreadViewportFooter, { children: [
-      /* @__PURE__ */ jsx41(ThreadScrollToBottom, {}),
-      /* @__PURE__ */ jsx41(composer_default, {})
+  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, {})
     ] })
   ] }) });
 };
 var ThreadRootStyled = withDefaults(thread_exports.Root, {
   className: "aui-root aui-thread-root"
 });
-var ThreadRoot = forwardRef26(
+var ThreadRoot = forwardRef23(
   ({ config, ...props }, ref) => {
-    return /* @__PURE__ */ jsx41(ThreadConfigProvider, { config, children: /* @__PURE__ */ jsx41(ThreadRootStyled, { ...props, ref }) });
+    return /* @__PURE__ */ jsx37(ThreadConfigProvider, { config, children: /* @__PURE__ */ jsx37(ThreadRootStyled, { ...props, ref }) });
   }
 );
 ThreadRoot.displayName = "ThreadRoot";
@@ -4491,7 +3203,7 @@ var ThreadViewportFooter = withDefaults("div", {
 ThreadViewportFooter.displayName = "ThreadViewportFooter";
 var SystemMessage = () => null;
 var ThreadMessages = ({ components, ...rest }) => {
-  return /* @__PURE__ */ jsx41(
+  return /* @__PURE__ */ jsx37(
     thread_exports.Messages,
     {
       components: {
@@ -4509,13 +3221,13 @@ var ThreadScrollToBottomIconButton = withDefaults(TooltipIconButton, {
   variant: "outline",
   className: "aui-thread-scroll-to-bottom"
 });
-var ThreadScrollToBottom = forwardRef26((props, ref) => {
+var ThreadScrollToBottom = forwardRef23((props, ref) => {
   const {
     strings: {
       thread: { scrollToBottom: { tooltip = "Scroll to bottom" } = {} } = {}
     } = {}
   } = useThreadConfig();
-  return /* @__PURE__ */ jsx41(thread_exports.ScrollToBottom, { asChild: true, children: /* @__PURE__ */ jsx41(ThreadScrollToBottomIconButton, { tooltip, ...props, ref, children: props.children ?? /* @__PURE__ */ jsx41(ArrowDownIcon, {}) }) });
+  return /* @__PURE__ */ jsx37(thread_exports.ScrollToBottom, { asChild: true, children: /* @__PURE__ */ jsx37(ThreadScrollToBottomIconButton, { tooltip, ...props, ref, children: props.children ?? /* @__PURE__ */ jsx37(ArrowDownIcon, {}) }) });
 });
 ThreadScrollToBottom.displayName = "ThreadScrollToBottom";
 var exports10 = {
@@ -4528,20 +3240,20 @@ var exports10 = {
 var thread_default = Object.assign(Thread, exports10);
 
 // src/ui/assistant-modal.tsx
-import { Fragment as Fragment6, jsx as jsx42, jsxs as jsxs14 } from "react/jsx-runtime";
+import { Fragment as Fragment6, jsx as jsx38, jsxs as jsxs13 } from "react/jsx-runtime";
 var AssistantModal = (config) => {
-  return /* @__PURE__ */ jsxs14(AssistantModalRoot, { config, children: [
-    /* @__PURE__ */ jsx42(AssistantModalTrigger, {}),
-    /* @__PURE__ */ jsx42(AssistantModalContent, { children: /* @__PURE__ */ jsx42(thread_default, {}) })
+  return /* @__PURE__ */ jsxs13(AssistantModalRoot, { config, children: [
+    /* @__PURE__ */ jsx38(AssistantModalTrigger, {}),
+    /* @__PURE__ */ jsx38(AssistantModalContent, { children: /* @__PURE__ */ jsx38(thread_default, {}) })
   ] });
 };
 AssistantModal.displayName = "AssistantModal";
 var AssistantModalRoot = ({ config, ...props }) => {
-  return /* @__PURE__ */ jsx42(ThreadConfigProvider, { config, children: /* @__PURE__ */ jsx42(assistantModal_exports.Root, { ...props }) });
+  return /* @__PURE__ */ jsx38(ThreadConfigProvider, { config, children: /* @__PURE__ */ jsx38(assistantModal_exports.Root, { ...props }) });
 };
 AssistantModalRoot.displayName = "AssistantModalRoot";
-var AssistantModalTrigger = forwardRef27((props, ref) => {
-  return /* @__PURE__ */ jsx42(AssistantModalAnchor, { children: /* @__PURE__ */ jsx42(assistantModal_exports.Trigger, { asChild: true, children: /* @__PURE__ */ jsx42(AssistantModalButton, { ...props, ref }) }) });
+var AssistantModalTrigger = forwardRef24((props, ref) => {
+  return /* @__PURE__ */ jsx38(AssistantModalAnchor, { children: /* @__PURE__ */ jsx38(assistantModal_exports.Trigger, { asChild: true, children: /* @__PURE__ */ jsx38(AssistantModalButton, { ...props, ref }) }) });
 });
 AssistantModalTrigger.displayName = "AssistantModalTrigger";
 var AssistantModalAnchor = withDefaults(assistantModal_exports.Anchor, {
@@ -4552,7 +3264,7 @@ var ModalButtonStyled = withDefaults(TooltipIconButton, {
   variant: "default",
   className: "aui-modal-button"
 });
-var AssistantModalButton = forwardRef27(({ "data-state": state, ...rest }, ref) => {
+var AssistantModalButton = forwardRef24(({ "data-state": state, ...rest }, ref) => {
   const {
     strings: {
       assistantModal: {
@@ -4566,7 +3278,7 @@ var AssistantModalButton = forwardRef27(({ "data-state": state, ...rest }, ref)
     } = {}
   } = useThreadConfig();
   const tooltip = state === "open" ? openTooltip : closedTooltip;
-  return /* @__PURE__ */ jsx42(
+  return /* @__PURE__ */ jsx38(
     ModalButtonStyled,
     {
       side: "left",
@@ -4574,15 +3286,15 @@ var AssistantModalButton = forwardRef27(({ "data-state": state, ...rest }, ref)
       "data-state": state,
       ...rest,
       ref,
-      children: rest.children ?? /* @__PURE__ */ jsxs14(Fragment6, { children: [
-        /* @__PURE__ */ jsx42(
+      children: rest.children ?? /* @__PURE__ */ jsxs13(Fragment6, { children: [
+        /* @__PURE__ */ jsx38(
           BotIcon,
           {
             "data-state": state,
             className: "aui-modal-button-closed-icon"
           }
         ),
-        /* @__PURE__ */ jsx42(
+        /* @__PURE__ */ jsx38(
           ChevronDownIcon,
           {
             "data-state": state,