@assistant-ui/react 0.5.66 → 0.5.68

package/dist/index.mjs

@@ -145,7 +145,9 @@ var { useThread, useThreadStore } = createContextStoreHook(
   useThreadContext,
   "useThread"
 );
-var { useThreadMessages, useThreadMessagesStore } = createContextStoreHook(useThreadContext, "useThreadMessages");
+var messages = createContextStoreHook(useThreadContext, "useThreadMessages");
+var useThreadMessages = messages.useThreadMessages;
+var useThreadMessagesStore = messages.useThreadMessagesStore;
 var {
   useComposer: useThreadComposer,
   useComposerStore: useThreadComposerStore
@@ -155,61 +157,11 @@ var {
   useViewportStore: useThreadViewportStore
 } = createContextStoreHook(useThreadContext, "useViewport");
 
-// src/context/stores/ThreadComposer.ts
-import { create as create2 } from "zustand";
-var makeThreadComposerStore = (runtime) => {
-  const focusListeners = /* @__PURE__ */ new Set();
-  return create2()((_, get) => {
-    return {
-      type: "thread",
-      get value() {
-        return get().text;
-      },
-      setValue(value) {
-        get().setText(value);
-      },
-      ...runtime.getState(),
-      canCancel: false,
-      // "TODO",
-      isEditing: true,
-      addAttachment: (file) => {
-        runtime.addAttachment(file);
-      },
-      removeAttachment: (attachmentId) => {
-        runtime.removeAttachment(attachmentId);
-      },
-      reset: () => {
-        runtime.reset();
-      },
-      setText: (text) => {
-        runtime.setText(text);
-      },
-      send: () => {
-        runtime.send();
-      },
-      cancel: () => {
-        runtime.cancel();
-      },
-      focus: () => {
-        for (const listener of focusListeners) {
-          listener();
-        }
-      },
-      onFocus: (listener) => {
-        focusListeners.add(listener);
-        return () => {
-          focusListeners.delete(listener);
-        };
-      }
-    };
-  });
-};
-
 // src/context/stores/ThreadViewport.tsx
-import { create as create3 } from "zustand";
+import { create as create2 } from "zustand";
 var makeThreadViewportStore = () => {
   const scrollToBottomListeners = /* @__PURE__ */ new Set();
-  return create3(() => ({
+  return create2(() => ({
     isAtBottom: true,
     scrollToBottom: () => {
       for (const listener of scrollToBottomListeners) {
@@ -231,17 +183,17 @@ var writableStore = (store) => {
 };
 
 // src/context/providers/ThreadRuntimeProvider.tsx
-import { create as create4 } from "zustand";
-import { jsx, jsxs } from "react/jsx-runtime";
+import { create as create3 } from "zustand";
+import { jsx } from "react/jsx-runtime";
 var useThreadRuntimeStore2 = (runtime) => {
-  const [store] = useState(() => create4(() => runtime));
+  const [store] = useState(() => create3(() => runtime));
   useEffect(() => {
     writableStore(store).setState(runtime, true);
   }, [runtime, store]);
   return store;
 };
 var useThreadStore2 = (runtime) => {
-  const [store] = useState(() => create4(() => runtime.getState()));
+  const [store] = useState(() => create3(() => runtime.getState()));
   useEffect(() => {
     const updateState = () => writableStore(store).setState(runtime.getState(), true);
     updateState();
@@ -250,7 +202,7 @@ var useThreadStore2 = (runtime) => {
   return store;
 };
 var useThreadMessagesStore2 = (runtime) => {
-  const [store] = useState(() => create4(() => runtime.messages));
+  const [store] = useState(() => create3(() => runtime.messages));
   useEffect(() => {
     const updateState = () => writableStore(store).setState(runtime.messages, true);
     updateState();
@@ -259,9 +211,9 @@ var useThreadMessagesStore2 = (runtime) => {
   return store;
 };
 var useThreadComposerStore2 = (runtime) => {
-  const [store] = useState(() => makeThreadComposerStore(runtime));
+  const [store] = useState(() => create3(() => runtime.getState()));
   useEffect(() => {
-    const updateState = () => writableStore(store).setState(runtime.getState());
+    const updateState = () => writableStore(store).setState(runtime.getState(), true);
     updateState();
     return runtime.subscribe(updateState);
   }, [runtime, store]);
@@ -283,20 +235,14 @@ var ThreadRuntimeProvider = ({ children, runtime }) => {
       useViewport
     };
   }, [useThread2, useThreadRuntime2, useThreadMessages2, useThreadComposer2]);
-  const Synchronizer = context.useThread(
-    (t) => t.unstable_synchronizer
-  );
-  return /* @__PURE__ */ jsxs(ThreadContext.Provider, { value: context, children: [
-    Synchronizer && /* @__PURE__ */ jsx(Synchronizer, {}),
-    children
-  ] });
+  return /* @__PURE__ */ jsx(ThreadContext.Provider, { value: context, children });
 };
 
 // src/context/providers/AssistantRuntimeProvider.tsx
-import { create as create5 } from "zustand";
+import { create as create4 } from "zustand";
 import { jsx as jsx2 } from "react/jsx-runtime";
 var useAssistantRuntimeStore2 = (runtime) => {
-  const [store] = useState2(() => create5(() => runtime));
+  const [store] = useState2(() => create4(() => runtime));
   useEffect2(() => {
     writableStore(store).setState(runtime, true);
   }, [runtime, store]);
@@ -321,7 +267,7 @@ var AssistantRuntimeProvider = memo(AssistantRuntimeProviderImpl);
 
 // src/context/providers/TextContentPartProvider.tsx
 import { useEffect as useEffect3, useState as useState3 } from "react";
-import { create as create6 } from "zustand";
+import { create as create5 } from "zustand";
 
 // src/context/react/ContentPartContext.ts
 import { createContext as createContext3 } from "react";
@@ -332,11 +278,47 @@ var useContentPartContext = createContextHook(
   ContentPartContext,
   "a component passed to <MessagePrimitive.Content components={...}>"
 );
+function useContentPartRuntime(options) {
+  const context = useContentPartContext(options);
+  if (!context) return null;
+  return context.useContentPartRuntime();
+}
 var { useContentPart, useContentPartStore } = createContextStoreHook(
   useContentPartContext,
   "useContentPart"
 );
 
+// src/api/ContentPartRuntime.ts
+var ContentPartRuntime = class {
+  constructor(contentBinding, messageApi, threadApi) {
+    this.contentBinding = contentBinding;
+    this.messageApi = messageApi;
+    this.threadApi = threadApi;
+  }
+  getState() {
+    return this.contentBinding.getState();
+  }
+  addToolResult(result) {
+    const message = this.messageApi.getState();
+    if (!message) throw new Error("Message is not available");
+    const state = this.contentBinding.getState();
+    if (!state) throw new Error("Content part is not available");
+    if (state.type !== "tool-call")
+      throw new Error("Tried to add tool result to non-tool content part");
+    const toolName = state.toolName;
+    const toolCallId = state.toolCallId;
+    this.threadApi.getState().addToolResult({
+      messageId: message.id,
+      toolName,
+      toolCallId,
+      result
+    });
+  }
+  subscribe(callback) {
+    return this.contentBinding.subscribe(callback);
+  }
+};
+
 // src/context/providers/TextContentPartProvider.tsx
 import { jsx as jsx3 } from "react/jsx-runtime";
 var COMPLETE_STATUS = {
@@ -347,18 +329,22 @@ var RUNNING_STATUS = {
 };
 var TextContentPartProvider = ({ children, text, isRunning }) => {
   const [context] = useState3(() => {
-    const useContentPart2 = create6(() => ({
+    const useContentPartRuntime2 = create5(
+      // TODO
+      () => new ContentPartRuntime(null, null, null)
+    );
+    const useContentPart2 = create5(() => ({
       status: isRunning ? RUNNING_STATUS : COMPLETE_STATUS,
-      part: { type: "text", text }
+      part: { type: "text", text },
+      type: "text",
+      text
     }));
-    return {
-      useContentPart: useContentPart2
-    };
+    return { useContentPartRuntime: useContentPartRuntime2, useContentPart: useContentPart2 };
   });
   useEffect3(() => {
     const state = context.useContentPart.getState();
-    const textUpdated = state.part.text !== text;
-    const targetTextPart = textUpdated ? { type: "text", text } : state.part;
+    const textUpdated = state.text !== text;
+    const targetTextPart = textUpdated ? { type: "text", text } : state;
     const targetStatus = isRunning ? RUNNING_STATUS : COMPLETE_STATUS;
     const statusUpdated = state.status !== targetStatus;
     if (!textUpdated && !statusUpdated) return;
@@ -401,20 +387,25 @@ var { useEditComposer, useEditComposerStore } = createContextStoreHook(
 // src/context/react/ComposerContext.ts
 import { useMemo as useMemo3 } from "react";
 var useComposerContext = () => {
-  const { useComposer: useComposer2 } = useThreadContext();
+  const { useComposer: useThreadComposer2 } = useThreadContext();
   const { useEditComposer: useEditComposer2 } = useMessageContext({ optional: true }) ?? {};
   return useMemo3(
     () => ({
-      useComposer: useEditComposer2 ?? useComposer2,
+      useComposer: useEditComposer2 ?? useThreadComposer2,
       type: useEditComposer2 ? "edit" : "new"
     }),
-    [useEditComposer2, useComposer2]
+    [useEditComposer2, useThreadComposer2]
   );
 };
 var { useComposer, useComposerStore } = createContextStoreHook(
   useComposerContext,
   "useComposer"
 );
+function useComposerRuntime(options) {
+  const messageRuntime = useMessageRuntime({ optional: true });
+  const threadRuntime = useThreadRuntime(options);
+  return messageRuntime ? messageRuntime.composer : threadRuntime?.composer ?? null;
+}
 
 // src/hooks/useAppendMessage.tsx
 import { useCallback } from "react";
@@ -580,12 +571,11 @@ var useActionBarCopy = ({
 // src/primitive-hooks/actionBar/useActionBarEdit.tsx
 import { useCallback as useCallback4 } from "react";
 var useActionBarEdit = () => {
-  const editComposerStore = useEditComposerStore();
+  const messageRuntime = useMessageRuntime();
   const disabled = useEditComposer((c) => c.isEditing);
   const callback = useCallback4(() => {
-    const { edit } = editComposerStore.getState();
-    edit();
-  }, [editComposerStore]);
+    messageRuntime.composer.beginEdit();
+  }, [messageRuntime]);
   if (disabled) return null;
   return callback;
 };
@@ -595,7 +585,7 @@ import { useCallback as useCallback5 } from "react";
 var useActionBarReload = () => {
   const messageStore = useMessageStore();
   const threadStore = useThreadStore();
-  const threadRuntime = useThreadRuntime();
+  const messageRuntime = useMessageRuntime();
   const threadComposerStore = useThreadComposerStore();
   const threadViewportStore = useThreadViewportStore();
   const disabled = useCombinedStore(
@@ -603,11 +593,10 @@ var useActionBarReload = () => {
     (t, m) => t.isRunning || t.isDisabled || m.role !== "assistant"
   );
   const callback = useCallback5(() => {
-    const { parentId } = messageStore.getState();
-    threadRuntime.startRun(parentId);
+    messageRuntime.reload();
     threadViewportStore.getState().scrollToBottom();
     threadComposerStore.getState().focus();
-  }, [threadRuntime, threadComposerStore, threadViewportStore, messageStore]);
+  }, [messageRuntime, threadComposerStore, threadViewportStore]);
   if (disabled) return null;
   return callback;
 };
@@ -617,19 +606,18 @@ import { useCallback as useCallback6 } from "react";
 var useActionBarSpeak = () => {
   const messageStore = useMessageStore();
   const editComposerStore = useEditComposerStore();
-  const threadRuntime = useThreadRuntime();
+  const messageRunime = useMessageRuntime();
   const messageUtilsStore = useMessageUtilsStore();
   const hasSpeakableContent = useCombinedStore(
     [messageStore, editComposerStore],
-    ({ message }, c) => {
+    (message, c) => {
       return !c.isEditing && (message.role !== "assistant" || message.status.type !== "running") && message.content.some((c2) => c2.type === "text" && c2.text.length > 0);
     }
   );
   const callback = useCallback6(async () => {
-    const { message } = messageStore.getState();
-    const utt = threadRuntime.speak(message.id);
+    const utt = messageRunime.speak();
     messageUtilsStore.getState().addUtterance(utt);
-  }, [threadRuntime, messageStore, messageUtilsStore]);
+  }, [messageRunime, messageUtilsStore]);
   if (!hasSpeakableContent) return null;
   return callback;
 };
@@ -649,32 +637,28 @@ var useActionBarStopSpeaking = () => {
 // src/primitive-hooks/actionBar/useActionBarFeedbackPositive.tsx
 import { useCallback as useCallback8 } from "react";
 var useActionBarFeedbackPositive = () => {
-  const threadRuntime = useThreadRuntime();
-  const messageStore = useMessageStore();
+  const messageRuntime = useMessageRuntime();
   const messageUtilsStore = useMessageUtilsStore();
   const callback = useCallback8(() => {
-    threadRuntime.submitFeedback({
-      messageId: messageStore.getState().id,
+    messageRuntime.submitFeedback({
       type: "positive"
     });
     messageUtilsStore.getState().setSubmittedFeedback("positive");
-  }, [messageStore, messageUtilsStore, threadRuntime]);
+  }, [messageUtilsStore, messageRuntime]);
   return callback;
 };
 
 // src/primitive-hooks/actionBar/useActionBarFeedbackNegative.tsx
 import { useCallback as useCallback9 } from "react";
 var useActionBarFeedbackNegative = () => {
-  const threadRuntime = useThreadRuntime();
-  const messageStore = useMessageStore();
+  const messageRuntime = useMessageRuntime();
   const messageUtilsStore = useMessageUtilsStore();
   const callback = useCallback9(() => {
-    threadRuntime.submitFeedback({
-      messageId: messageStore.getState().id,
+    messageRuntime.submitFeedback({
       type: "negative"
     });
     messageUtilsStore.getState().setSubmittedFeedback("negative");
-  }, [messageStore, messageUtilsStore, threadRuntime]);
+  }, [messageUtilsStore, messageRuntime]);
   return callback;
 };
 
@@ -796,7 +780,7 @@ var useComposerAddAttachment = () => {
 // src/primitive-hooks/contentPart/useContentPartDisplay.tsx
 var useContentPartDisplay = () => {
   const display = useContentPart((c) => {
-    if (c.part.type !== "ui")
+    if (c.type !== "ui")
       throw new Error(
         "This component can only be used inside ui content parts."
       );
@@ -808,7 +792,7 @@ var useContentPartDisplay = () => {
 // src/primitive-hooks/contentPart/useContentPartImage.tsx
 var useContentPartImage = () => {
   const image = useContentPart((c) => {
-    if (c.part.type !== "image")
+    if (c.type !== "image")
       throw new Error(
         "ContentPartImage can only be used inside image content parts."
       );
@@ -820,7 +804,7 @@ var useContentPartImage = () => {
 // src/primitive-hooks/contentPart/useContentPartText.tsx
 var useContentPartText = () => {
   const text = useContentPart((c) => {
-    if (c.part.type !== "text")
+    if (c.type !== "text")
       throw new Error(
         "ContentPartText can only be used inside text content parts."
       );
@@ -858,20 +842,15 @@ var useMessageIf = (props) => {
 
 // src/primitive-hooks/thread/useThreadIf.tsx
 var useThreadIf = (props) => {
-  const threadStore = useThreadStore();
-  const threadMessagesStore = useThreadMessagesStore();
-  return useCombinedStore(
-    [threadStore, threadMessagesStore],
-    (thread, messages) => {
-      if (props.empty === true && messages.length !== 0) return false;
-      if (props.empty === false && messages.length === 0) return false;
-      if (props.running === true && !thread.isRunning) return false;
-      if (props.running === false && thread.isRunning) return false;
-      if (props.disabled === true && thread.isDisabled) return false;
-      if (props.disabled === false && thread.isDisabled) return false;
-      return true;
-    }
-  );
+  return useThread((thread) => {
+    if (props.empty === true && thread.messages.length !== 0) return false;
+    if (props.empty === false && thread.messages.length === 0) return false;
+    if (props.running === true && !thread.isRunning) return false;
+    if (props.running === false && thread.isRunning) return false;
+    if (props.disabled === true && thread.isDisabled) return false;
+    if (props.disabled === false && thread.isDisabled) return false;
+    return true;
+  });
 };
 
 // src/primitive-hooks/thread/useThreadEmpty.tsx
@@ -1380,76 +1359,37 @@ var MessagePrimitiveIf = ({
 MessagePrimitiveIf.displayName = "MessagePrimitive.If";
 
 // src/primitives/message/MessageContent.tsx
-import { memo as memo2 } from "react";
+import { memo as memo2, useMemo as useMemo6 } from "react";
 
-// src/context/providers/ContentPartProvider.tsx
+// src/context/providers/ContentPartRuntimeProvider.tsx
 import { useEffect as useEffect8, useState as useState5 } from "react";
-import { create as create7 } from "zustand";
+import { create as create6 } from "zustand";
 import { jsx as jsx17 } from "react/jsx-runtime";
-var COMPLETE_STATUS2 = {
-  type: "complete"
-};
-var toContentPartStatus = (message, partIndex, part) => {
-  if (message.role !== "assistant") return COMPLETE_STATUS2;
-  const isLastPart = partIndex === Math.max(0, message.content.length - 1);
-  if (part.type !== "tool-call") {
-    if ("reason" in message.status && message.status.reason === "tool-calls" && isLastPart)
-      throw new Error(
-        "Encountered unexpected requires-action status. This is likely an internal bug in assistant-ui."
-      );
-    return isLastPart ? message.status : COMPLETE_STATUS2;
-  }
-  if (!!part.result) {
-    return COMPLETE_STATUS2;
-  }
-  return message.status;
-};
-var EMPTY_CONTENT = Object.freeze({ type: "text", text: "" });
-var getContentPartState = ({ message }, useContentPart2, partIndex) => {
-  let part = message.content[partIndex];
-  if (!part) {
-    if (message.content.length === 0 && partIndex === 0) {
-      part = EMPTY_CONTENT;
-    } else {
-      return null;
-    }
-  } else if (message.content.length === 1 && part.type === "text" && part.text.length === 0) {
-    part = EMPTY_CONTENT;
-  }
-  const status = toContentPartStatus(message, partIndex, part);
-  const currentState = useContentPart2?.getState();
-  if (currentState && currentState.part === part && currentState.status === status)
-    return null;
-  return Object.freeze({ part, status });
+var useContentPartRuntimeStore = (runtime) => {
+  const [store] = useState5(() => create6(() => runtime));
+  useEffect8(() => {
+    writableStore(store).setState(runtime, true);
+  }, [runtime, store]);
+  return store;
 };
-var useContentPartContext2 = (partIndex) => {
-  const messageStore = useMessageStore();
-  const [context] = useState5(() => {
-    const useContentPart2 = create7(
-      () => getContentPartState(messageStore.getState(), void 0, partIndex)
-    );
-    return { useContentPart: useContentPart2 };
-  });
+var useContentPartStore2 = (runtime) => {
+  const [store] = useState5(() => create6(() => runtime.getState()));
   useEffect8(() => {
-    const syncContentPart = (message) => {
-      const newState = getContentPartState(
-        message,
-        context.useContentPart,
-        partIndex
-      );
-      if (!newState) return;
-      writableStore(context.useContentPart).setState(newState, true);
-    };
-    syncContentPart(messageStore.getState());
-    return messageStore.subscribe(syncContentPart);
-  }, [context, messageStore, partIndex]);
-  return context;
+    const updateState = () => writableStore(store).setState(runtime.getState(), true);
+    updateState();
+    return runtime.subscribe(updateState);
+  }, [runtime, store]);
+  return store;
 };
-var ContentPartProvider = ({
-  partIndex,
+var ContentPartRuntimeProvider = ({
+  runtime,
   children
 }) => {
-  const context = useContentPartContext2(partIndex);
+  const useContentPartRuntime2 = useContentPartRuntimeStore(runtime);
+  const useContentPart2 = useContentPartStore2(runtime);
+  const [context] = useState5(() => {
+    return { useContentPartRuntime: useContentPartRuntime2, useContentPart: useContentPart2 };
+  });
   return /* @__PURE__ */ jsx17(ContentPartContext.Provider, { value: context, children });
 };
 
@@ -1469,11 +1409,11 @@ import {
   useContext as useContext2,
   useState as useState6
 } from "react";
-import { create as create8 } from "zustand";
+import { create as create7 } from "zustand";
 import { jsx as jsx18 } from "react/jsx-runtime";
 var SmoothContext = createContext5(null);
 var makeSmoothContext = (initialState) => {
-  const useSmoothStatus2 = create8(() => initialState);
+  const useSmoothStatus2 = create7(() => initialState);
   return { useSmoothStatus: useSmoothStatus2 };
 };
 var SmoothContextProvider = ({ children }) => {
@@ -1554,9 +1494,7 @@ var SMOOTH_STATUS = Object.freeze({
   type: "running"
 });
 var useSmooth = (state, smooth = false) => {
-  const {
-    part: { text }
-  } = state;
+  const { text } = state;
   const id = useMessage({
     optional: true,
     selector: (m) => m.id
@@ -1568,17 +1506,17 @@ var useSmooth = (state, smooth = false) => {
     setDisplayedText(text2);
     if (smoothStatusStore) {
       writableStore(smoothStatusStore).setState(
-        text2 !== state.part.text ? SMOOTH_STATUS : state.status
+        text2 !== state.text ? SMOOTH_STATUS : state.status
       );
     }
   });
   useEffect9(() => {
     if (smoothStatusStore) {
       writableStore(smoothStatusStore).setState(
-        text !== state.part.text ? SMOOTH_STATUS : state.status
+        text !== state.text ? SMOOTH_STATUS : state.status
       );
     }
-  }, [smoothStatusStore, text, displayedText, state.status, state.part.text]);
+  }, [smoothStatusStore, text, displayedText, state.status, state.text]);
   const [animatorRef] = useState7(
     new TextStreamAnimator(text, setText)
   );
@@ -1605,6 +1543,8 @@ var useSmooth = (state, smooth = false) => {
   }, [animatorRef]);
   return useMemo5(
     () => smooth ? {
+      type: "text",
+      text: displayedText,
       part: { type: "text", text: displayedText },
       status: text === displayedText ? state.status : SMOOTH_STATUS
     } : state,
@@ -1615,10 +1555,7 @@ var useSmooth = (state, smooth = false) => {
 // src/primitives/contentPart/ContentPartText.tsx
 import { jsx as jsx19 } from "react/jsx-runtime";
 var ContentPartPrimitiveText = forwardRef12(({ smooth = true, component: Component = "span", ...rest }, forwardedRef) => {
-  const {
-    part: { text },
-    status
-  } = useSmooth(useContentPartText(), smooth);
+  const { text, status } = useSmooth(useContentPartText(), smooth);
   return /* @__PURE__ */ jsx19(Component, { "data-status": status.type, ...rest, ref: forwardedRef, children: text });
 });
 ContentPartPrimitiveText.displayName = "ContentPartPrimitive.Text";
@@ -1628,18 +1565,14 @@ import { Primitive as Primitive8 } from "@radix-ui/react-primitive";
 import { forwardRef as forwardRef13 } from "react";
 import { jsx as jsx20 } from "react/jsx-runtime";
 var ContentPartPrimitiveImage = forwardRef13((props, forwardedRef) => {
-  const {
-    part: { image }
-  } = useContentPartImage();
+  const { image } = useContentPartImage();
   return /* @__PURE__ */ jsx20(Primitive8.img, { src: image, ...props, ref: forwardedRef });
 });
 ContentPartPrimitiveImage.displayName = "ContentPartPrimitive.Image";
 
 // src/primitives/contentPart/ContentPartDisplay.tsx
 var ContentPartPrimitiveDisplay = () => {
-  const {
-    part: { display }
-  } = useContentPartDisplay();
+  const { display } = useContentPartDisplay();
   return display ?? null;
 };
 ContentPartPrimitiveDisplay.displayName = "ContentPartPrimitive.Display";
@@ -1651,105 +1584,655 @@ var ContentPartPrimitiveInProgress = ({ children }) => {
 };
 ContentPartPrimitiveInProgress.displayName = "ContentPartPrimitive.InProgress";
 
-// src/primitives/message/MessageContent.tsx
-import { jsx as jsx21, jsxs as jsxs2 } from "react/jsx-runtime";
-var ToolUIDisplay = ({
-  UI,
-  ...props
-}) => {
-  const Render = useToolUIs((s) => s.getToolUI(props.part.toolName)) ?? UI;
-  if (!Render) return null;
-  return /* @__PURE__ */ jsx21(Render, { ...props });
+// src/api/AttachmentRuntime.ts
+var AttachmentRuntime = class {
+  constructor(_core) {
+    this._core = _core;
+  }
+  getState() {
+    return this._core.getState();
+  }
+  subscribe(callback) {
+    return this._core.subscribe(callback);
+  }
 };
-var defaultComponents = {
-  Text: () => /* @__PURE__ */ jsxs2("p", { style: { whiteSpace: "pre-line" }, children: [
-    /* @__PURE__ */ jsx21(ContentPartPrimitiveText, {}),
-    /* @__PURE__ */ jsx21(ContentPartPrimitiveInProgress, { children: /* @__PURE__ */ jsx21("span", { style: { fontFamily: "revert" }, children: " \u25CF" }) })
-  ] }),
-  Image: () => /* @__PURE__ */ jsx21(ContentPartPrimitiveImage, {}),
-  UI: () => /* @__PURE__ */ jsx21(ContentPartPrimitiveDisplay, {})
+var ComposerAttachmentRuntime = class extends AttachmentRuntime {
+  constructor(core, _composerApi) {
+    super(core);
+    this._composerApi = _composerApi;
+  }
+  remove() {
+    const core = this._composerApi.getState();
+    if (!core) throw new Error("Composer is not available");
+    return core.removeAttachment(this.getState().id);
+  }
 };
-var MessageContentPartComponent = ({
-  components: {
-    Text: Text2 = defaultComponents.Text,
-    Empty,
-    Image: Image2 = defaultComponents.Image,
-    UI = defaultComponents.UI,
-    tools: { by_name = {}, Fallback: Fallback2 = void 0 } = {}
-  } = {}
-}) => {
-  const messageStore = useMessageStore();
-  const threadRuntime = useThreadRuntime();
-  const { part, status } = useContentPart();
-  const type = part.type;
-  switch (type) {
-    case "text":
-      if (status.type === "requires-action")
-        throw new Error("Encountered unexpected requires-action status");
-      if (part === EMPTY_CONTENT && !!Empty) {
-        return /* @__PURE__ */ jsx21(Empty, { status });
-      }
-      return /* @__PURE__ */ jsx21(Text2, { part, status });
-    case "image":
-      if (status.type === "requires-action")
-        throw new Error("Encountered unexpected requires-action status");
-      return /* @__PURE__ */ jsx21(Image2, { part, status });
-    case "ui":
-      if (status.type === "requires-action")
-        throw new Error("Encountered unexpected requires-action status");
-      return /* @__PURE__ */ jsx21(UI, { part, status });
-    case "tool-call": {
-      const Tool = by_name[part.toolName] || Fallback2;
-      const addResult = (result) => threadRuntime.addToolResult({
-        messageId: messageStore.getState().id,
-        toolName: part.toolName,
-        toolCallId: part.toolCallId,
-        result
-      });
-      return /* @__PURE__ */ jsx21(
-        ToolUIDisplay,
-        {
-          UI: Tool,
-          part,
-          status,
-          addResult
-        }
-      );
-    }
-    default:
-      const unhandledType = type;
-      throw new Error(`Unknown content part type: ${unhandledType}`);
+var ThreadComposerAttachmentRuntime = class extends ComposerAttachmentRuntime {
+  get source() {
+    return "thread-composer";
   }
 };
-var MessageContentPartImpl = ({
-  partIndex,
-  components
-}) => {
-  return /* @__PURE__ */ jsx21(ContentPartProvider, { partIndex, children: /* @__PURE__ */ jsx21(MessageContentPartComponent, { components }) });
+var EditComposerAttachmentRuntime = class extends ComposerAttachmentRuntime {
+  get source() {
+    return "edit-composer";
+  }
 };
-var MessageContentPart = memo2(
-  MessageContentPartImpl,
-  (prev, next) => prev.partIndex === next.partIndex && prev.components?.Text === next.components?.Text && prev.components?.Image === next.components?.Image && prev.components?.UI === next.components?.UI && prev.components?.tools === next.components?.tools
-);
-var MessagePrimitiveContent = ({
-  components
-}) => {
-  const contentLength = useMessage((s) => s.content.length) || 1;
-  return Array.from({ length: contentLength }, (_, index) => /* @__PURE__ */ jsx21(MessageContentPart, { partIndex: index, components }, index));
+var MessageAttachmentRuntime = class extends AttachmentRuntime {
+  get source() {
+    return "message";
+  }
+  constructor(core) {
+    super(core);
+  }
+  remove() {
+    throw new Error("Message attachments cannot be removed");
+  }
 };
-MessagePrimitiveContent.displayName = "MessagePrimitive.Content";
 
-// src/primitives/message/MessageInProgress.tsx
-var MessagePrimitiveInProgress = () => {
-  return null;
+// src/api/subscribable/BaseSubject.ts
+var BaseSubject = class {
+  _subscriptions = /* @__PURE__ */ new Set();
+  _connection;
+  get isConnected() {
+    return !!this._connection;
+  }
+  notifySubscribers() {
+    for (const callback of this._subscriptions) callback();
+  }
+  _updateConnection() {
+    if (this._subscriptions.size > 0) {
+      if (this._connection) return;
+      this._connection = this._connect();
+    } else {
+      this._connection?.();
+      this._connection = void 0;
+    }
+  }
+  subscribe(callback) {
+    this._subscriptions.add(callback);
+    this._updateConnection();
+    return () => {
+      this._subscriptions.delete(callback);
+      this._updateConnection();
+    };
+  }
 };
-MessagePrimitiveInProgress.displayName = "MessagePrimitive.InProgress";
 
-// src/primitives/message/MessageAttachments.tsx
-import { memo as memo3 } from "react";
+// src/api/subscribable/SKIP_UPDATE.ts
+var SKIP_UPDATE = Symbol("skip-update");
 
-// src/context/react/AttachmentContext.ts
-import { createContext as createContext6, useContext as useContext3 } from "react";
+// src/api/subscribable/LazyMemoizeSubject.ts
+var LazyMemoizeSubject = class extends BaseSubject {
+  constructor(binding) {
+    super();
+    this.binding = binding;
+  }
+  _previousStateDirty = true;
+  _previousState;
+  getState = () => {
+    if (!this.isConnected || this._previousStateDirty) {
+      const newState = this.binding.getState();
+      if (newState !== SKIP_UPDATE) {
+        this._previousState = newState;
+      }
+      this._previousStateDirty = false;
+    }
+    if (this._previousState === void 0)
+      throw new Error("Entry not available in the store");
+    return this._previousState;
+  };
+  _connect() {
+    const callback = () => {
+      this._previousStateDirty = true;
+      this.notifySubscribers();
+    };
+    return this.binding.subscribe(callback);
+  }
+};
+
+// src/api/subscribable/shallowEqual.ts
+function shallowEqual(objA, objB) {
+  if (objA === void 0 && objB === void 0) return true;
+  if (objA === void 0) return false;
+  if (objB === void 0) return false;
+  for (const key of Object.keys(objA)) {
+    const valueA = objA[key];
+    const valueB = objB[key];
+    if (!Object.is(valueA, valueB)) return false;
+  }
+  return true;
+}
+
+// src/api/subscribable/ShallowMemoizeSubject.ts
+var ShallowMemoizeSubject = class extends BaseSubject {
+  constructor(binding) {
+    super();
+    this.binding = binding;
+    const state = binding.getState();
+    if (state === SKIP_UPDATE)
+      throw new Error("Entry not available in the store");
+    this._previousState = state;
+  }
+  _previousState;
+  getState = () => {
+    if (!this.isConnected) this._syncState();
+    return this._previousState;
+  };
+  _syncState() {
+    const state = this.binding.getState();
+    if (state === SKIP_UPDATE) return false;
+    if (shallowEqual(state, this._previousState)) return false;
+    this._previousState = state;
+    return true;
+  }
+  _connect() {
+    const callback = () => {
+      if (this._syncState()) {
+        this.notifySubscribers();
+      }
+    };
+    return this.binding.subscribe(callback);
+  }
+};
+
+// src/api/ComposerRuntime.ts
+var METHOD_NOT_SUPPORTED = () => {
+  throw new Error("Composer is not available");
+};
+var EMPTY_ARRAY = Object.freeze([]);
+var getThreadComposerState = (runtime, focus, onFocus) => {
+  return Object.freeze({
+    type: "thread",
+    isEditing: runtime?.isEditing ?? false,
+    canCancel: runtime?.canCancel ?? false,
+    isEmpty: runtime?.isEmpty ?? true,
+    text: runtime?.text ?? "",
+    attachments: runtime?.attachments ?? EMPTY_ARRAY,
+    attachmentAccept: runtime?.attachmentAccept ?? "*",
+    value: runtime?.text ?? "",
+    setValue: runtime?.setText.bind(runtime) ?? METHOD_NOT_SUPPORTED,
+    setText: runtime?.setText.bind(runtime) ?? METHOD_NOT_SUPPORTED,
+    // edit: beginEdit,
+    send: runtime?.send.bind(runtime) ?? METHOD_NOT_SUPPORTED,
+    cancel: runtime?.cancel.bind(runtime) ?? METHOD_NOT_SUPPORTED,
+    focus,
+    onFocus,
+    reset: runtime?.reset.bind(runtime) ?? METHOD_NOT_SUPPORTED,
+    addAttachment: runtime?.addAttachment.bind(runtime) ?? METHOD_NOT_SUPPORTED,
+    removeAttachment: runtime?.removeAttachment.bind(runtime) ?? METHOD_NOT_SUPPORTED
+  });
+};
+var getEditComposerState = (runtime, beginEdit) => {
+  return Object.freeze({
+    type: "edit",
+    isEditing: runtime?.isEditing ?? false,
+    canCancel: runtime?.canCancel ?? false,
+    isEmpty: runtime?.isEmpty ?? true,
+    text: runtime?.text ?? "",
+    attachments: runtime?.attachments ?? EMPTY_ARRAY,
+    attachmentAccept: runtime?.attachmentAccept ?? "*",
+    value: runtime?.text ?? "",
+    setValue: runtime?.setText.bind(runtime) ?? METHOD_NOT_SUPPORTED,
+    setText: runtime?.setText.bind(runtime) ?? METHOD_NOT_SUPPORTED,
+    edit: beginEdit,
+    send: runtime?.send.bind(runtime) ?? METHOD_NOT_SUPPORTED,
+    cancel: runtime?.cancel.bind(runtime) ?? METHOD_NOT_SUPPORTED
+  });
+};
+var ComposerRuntime = class {
+  constructor(_core) {
+    this._core = _core;
+  }
+  /**
+   * @deprecated Use `getState().isEditing` instead. This will be removed in 0.6.0.
+   */
+  get isEditing() {
+    return this.getState().isEditing;
+  }
+  /**
+   * @deprecated Use `getState().isEmpty` instead. This will be removed in 0.6.0.
+   */
+  get isEmpty() {
+    return this.getState().isEmpty;
+  }
+  /**
+   * @deprecated Use `getState().canCancel` instead. This will be removed in 0.6.0.
+   */
+  get canCancel() {
+    return this.getState().canCancel;
+  }
+  /**
+   * @deprecated Use `getState().text` instead. This will be removed in 0.6.0.
+   */
+  get text() {
+    return this.getState().text;
+  }
+  /**
+   * @deprecated Use `getState().attachmentAccept` instead. This will be removed in 0.6.0.
+   */
+  get attachmentAccept() {
+    return this.getState().attachmentAccept;
+  }
+  /**
+   * @deprecated Use `getState().attachments` instead. This will be removed in 0.6.0.
+   */
+  get attachments() {
+    return this.getState().attachments;
+  }
+  /**
+   * @deprecated Use `getState().text` instead. This will be removed in 0.6.0.
+   */
+  get value() {
+    return this.text;
+  }
+  setText(text) {
+    const core = this._core.getState();
+    if (!core) throw new Error("Composer is not available");
+    core.setText(text);
+  }
+  setValue(text) {
+    this.setText(text);
+  }
+  addAttachment(file) {
+    const core = this._core.getState();
+    if (!core) throw new Error("Composer is not available");
+    return core.addAttachment(file);
+  }
+  // /**
+  //  * @deprecated Use `getAttachmentById(id).removeAttachment` instead. This will be removed in 0.6.0.
+  //  */
+  removeAttachment(attachmentId) {
+    const core = this._core.getState();
+    if (!core) throw new Error("Composer is not available");
+    return core.removeAttachment(attachmentId);
+  }
+  /**
+   * @deprecated This method will be removed in 0.6.0. Submit feedback if you need this functionality.
+   */
+  reset() {
+    const core = this._core.getState();
+    if (!core) throw new Error("Composer is not available");
+    core.reset();
+  }
+  send() {
+    const core = this._core.getState();
+    if (!core) throw new Error("Composer is not available");
+    core.send();
+  }
+  cancel() {
+    const core = this._core.getState();
+    if (!core) throw new Error("Composer is not available");
+    core.cancel();
+  }
+  subscribe(callback) {
+    return this._core.subscribe(callback);
+  }
+};
+var ThreadComposerRuntime = class extends ComposerRuntime {
+  get type() {
+    return "thread";
+  }
+  _getState;
+  constructor(core) {
+    const stateBinding = new LazyMemoizeSubject({
+      getState: () => getThreadComposerState(
+        core.getState(),
+        this.focus.bind(this),
+        this.onFocus.bind(this)
+      ),
+      subscribe: (callback) => core.subscribe(callback)
+    });
+    super({
+      getState: () => core.getState(),
+      subscribe: (callback) => stateBinding.subscribe(callback)
+    });
+    this._getState = stateBinding.getState.bind(stateBinding);
+  }
+  get attachments() {
+    return this.getState()?.attachments ?? EMPTY_ARRAY;
+  }
+  getState() {
+    return this._getState();
+  }
+  // TODO replace with events
+  _focusListeners = /* @__PURE__ */ new Set();
+  focus() {
+    this._focusListeners.forEach((callback) => callback());
+  }
+  onFocus(callback) {
+    this._focusListeners.add(callback);
+    return () => this._focusListeners.delete(callback);
+  }
+  unstable_getAttachmentByIndex(idx) {
+    return new ThreadComposerAttachmentRuntime(
+      new ShallowMemoizeSubject({
+        getState: () => {
+          const attachments = this.getState().attachments;
+          const attachment = attachments[idx];
+          if (!attachment) return SKIP_UPDATE;
+          return {
+            ...attachment,
+            attachment,
+            source: "thread-composer"
+          };
+        },
+        subscribe: (callback) => this._core.subscribe(callback)
+      }),
+      this._core
+    );
+  }
+};
+var EditComposerRuntime = class extends ComposerRuntime {
+  constructor(core, _beginEdit) {
+    const stateBinding = new LazyMemoizeSubject({
+      getState: () => getEditComposerState(core.getState(), this._beginEdit),
+      subscribe: (callback) => core.subscribe(callback)
+    });
+    super({
+      getState: () => core.getState(),
+      subscribe: (callback) => stateBinding.subscribe(callback)
+    });
+    this._beginEdit = _beginEdit;
+    this._getState = stateBinding.getState.bind(stateBinding);
+  }
+  get type() {
+    return "edit";
+  }
+  _getState;
+  getState() {
+    return this._getState();
+  }
+  beginEdit() {
+    this._beginEdit();
+  }
+  /**
+   * @deprecated Use `beginEdit()` instead. This will be removed in 0.6.0.
+   */
+  edit() {
+    this.beginEdit();
+  }
+  unstable_getAttachmentByIndex(idx) {
+    return new EditComposerAttachmentRuntime(
+      new ShallowMemoizeSubject({
+        getState: () => {
+          const attachments = this.getState().attachments;
+          const attachment = attachments[idx];
+          if (!attachment) return SKIP_UPDATE;
+          return {
+            ...attachment,
+            attachment,
+            source: "edit-composer"
+          };
+        },
+        subscribe: (callback) => this._core.subscribe(callback)
+      }),
+      this._core
+    );
+  }
+};
+
+// src/api/subscribable/NestedSubscriptionSubject.ts
+var NestedSubscriptionSubject = class extends BaseSubject {
+  constructor(binding) {
+    super();
+    this.binding = binding;
+  }
+  getState() {
+    return this.binding.getState();
+  }
+  _connect() {
+    const callback = () => {
+      this.notifySubscribers();
+    };
+    let lastState = this.binding.getState();
+    let innerUnsubscribe = lastState?.subscribe(callback);
+    const onRuntimeUpdate = () => {
+      const newState = this.binding.getState();
+      if (newState === lastState) return;
+      lastState = newState;
+      innerUnsubscribe?.();
+      innerUnsubscribe = this.binding.getState()?.subscribe(callback);
+      callback();
+    };
+    const outerUnsubscribe = this.binding.subscribe(onRuntimeUpdate);
+    return () => {
+      outerUnsubscribe?.();
+      innerUnsubscribe?.();
+    };
+  }
+};
+
+// src/api/MessageRuntime.ts
+var COMPLETE_STATUS2 = {
+  type: "complete"
+};
+var toContentPartStatus = (message, partIndex, part) => {
+  if (message.role !== "assistant") return COMPLETE_STATUS2;
+  const isLastPart = partIndex === Math.max(0, message.content.length - 1);
+  if (part.type !== "tool-call") {
+    if ("reason" in message.status && message.status.reason === "tool-calls" && isLastPart)
+      throw new Error(
+        "Encountered unexpected requires-action status. This is likely an internal bug in assistant-ui."
+      );
+    return isLastPart ? message.status : COMPLETE_STATUS2;
+  }
+  if (!!part.result) {
+    return COMPLETE_STATUS2;
+  }
+  return message.status;
+};
+var EMPTY_CONTENT = Object.freeze({ type: "text", text: "" });
+var getContentPartState = (message, partIndex) => {
+  let part = message.content[partIndex];
+  if (!part) {
+    if (message.content.length === 0 && partIndex === 0) {
+      part = EMPTY_CONTENT;
+    } else {
+      return SKIP_UPDATE;
+    }
+  } else if (message.content.length === 1 && part.type === "text" && part.text.length === 0) {
+    part = EMPTY_CONTENT;
+  }
+  const status = toContentPartStatus(message, partIndex, part);
+  return Object.freeze({ ...part, part, status });
+};
+var MessageRuntime = class {
+  constructor(_core, _threadBinding) {
+    this._core = _core;
+    this._threadBinding = _threadBinding;
+  }
+  composer = new EditComposerRuntime(
+    new NestedSubscriptionSubject({
+      getState: () => this._threadBinding.getState().getEditComposer(this._core.getState().id),
+      subscribe: (callback) => this._threadBinding.subscribe(callback)
+    }),
+    () => this._threadBinding.getState().beginEdit(this._core.getState().id)
+  );
+  getState() {
+    return this._core.getState();
+  }
+  // TODO improve type
+  unstable_edit(message) {
+    const state = this._core.getState();
+    if (!state) throw new Error("Message is not available");
+    this._threadBinding.getState().append({
+      ...message,
+      parentId: state.parentId
+    });
+  }
+  reload() {
+    const state = this._core.getState();
+    if (!state) throw new Error("Message is not available");
+    if (state.role !== "assistant")
+      throw new Error("Can only reload assistant messages");
+    this._threadBinding.getState().startRun(state.parentId);
+  }
+  speak() {
+    const state = this._core.getState();
+    if (!state) throw new Error("Message is not available");
+    return this._threadBinding.getState().speak(state.id);
+  }
+  submitFeedback({ type }) {
+    const state = this._core.getState();
+    if (!state) throw new Error("Message is not available");
+    this._threadBinding.getState().submitFeedback({
+      messageId: state.id,
+      type
+    });
+  }
+  switchToBranch({
+    position,
+    branchId
+  }) {
+    const state = this._core.getState();
+    if (!state) throw new Error("Message is not available");
+    if (branchId && position) {
+      throw new Error("May not specify both branchId and position");
+    } else if (!branchId && !position) {
+      throw new Error("Must specify either branchId or position");
+    }
+    const thread = this._threadBinding.getState();
+    const branches = thread.getBranches(state.id);
+    let targetBranch = branchId;
+    if (position === "previous") {
+      targetBranch = branches[state.branchNumber - 2];
+    } else if (position === "next") {
+      targetBranch = branches[state.branchNumber];
+    }
+    if (!targetBranch) throw new Error("Branch not found");
+    this._threadBinding.getState().switchToBranch(targetBranch);
+  }
+  subscribe(callback) {
+    return this._core.subscribe(callback);
+  }
+  unstable_getContentPartByIndex(idx) {
+    if (idx < 0) throw new Error("Message index must be >= 0");
+    return new ContentPartRuntime(
+      new ShallowMemoizeSubject({
+        getState: () => {
+          return getContentPartState(this.getState(), idx);
+        },
+        subscribe: (callback) => this._core.subscribe(callback)
+      }),
+      this._core,
+      this._threadBinding
+    );
+  }
+  unstable_getAttachmentByIndex(idx) {
+    return new MessageAttachmentRuntime(
+      new ShallowMemoizeSubject({
+        getState: () => {
+          const attachments = this.getState().attachments;
+          const attachment = attachments?.[idx];
+          if (!attachment) return SKIP_UPDATE;
+          return {
+            ...attachment,
+            attachment,
+            source: "message"
+          };
+        },
+        subscribe: (callback) => this._core.subscribe(callback)
+      })
+    );
+  }
+};
+
+// src/primitives/message/MessageContent.tsx
+import { jsx as jsx21, jsxs } from "react/jsx-runtime";
+var ToolUIDisplay = ({
+  UI,
+  ...props
+}) => {
+  const Render = useToolUIs((s) => s.getToolUI(props.toolName)) ?? UI;
+  if (!Render) return null;
+  return /* @__PURE__ */ jsx21(Render, { ...props });
+};
+var defaultComponents = {
+  Text: () => /* @__PURE__ */ jsxs("p", { style: { whiteSpace: "pre-line" }, children: [
+    /* @__PURE__ */ jsx21(ContentPartPrimitiveText, {}),
+    /* @__PURE__ */ jsx21(ContentPartPrimitiveInProgress, { children: /* @__PURE__ */ jsx21("span", { style: { fontFamily: "revert" }, children: " \u25CF" }) })
+  ] }),
+  Image: () => /* @__PURE__ */ jsx21(ContentPartPrimitiveImage, {}),
+  UI: () => /* @__PURE__ */ jsx21(ContentPartPrimitiveDisplay, {})
+};
+var MessageContentPartComponent = ({
+  components: {
+    Text: Text2 = defaultComponents.Text,
+    Empty,
+    Image: Image2 = defaultComponents.Image,
+    UI = defaultComponents.UI,
+    tools: { by_name = {}, Fallback: Fallback2 = void 0 } = {}
+  } = {}
+}) => {
+  const messageStore = useMessageStore();
+  const threadRuntime = useThreadRuntime();
+  const part = useContentPart();
+  const type = part.type;
+  switch (type) {
+    case "text":
+      if (part.status.type === "requires-action")
+        throw new Error("Encountered unexpected requires-action status");
+      if (part.part === EMPTY_CONTENT && !!Empty) {
+        return /* @__PURE__ */ jsx21(Empty, { status: part.status });
+      }
+      return /* @__PURE__ */ jsx21(Text2, { ...part, part });
+    case "image":
+      if (part.status.type === "requires-action")
+        throw new Error("Encountered unexpected requires-action status");
+      return /* @__PURE__ */ jsx21(Image2, { ...part, part });
+    case "ui":
+      if (part.status.type === "requires-action")
+        throw new Error("Encountered unexpected requires-action status");
+      return /* @__PURE__ */ jsx21(UI, { ...part, part });
+    case "tool-call": {
+      const Tool = by_name[part.toolName] || Fallback2;
+      const addResult = (result) => threadRuntime.addToolResult({
+        messageId: messageStore.getState().id,
+        toolName: part.toolName,
+        toolCallId: part.toolCallId,
+        result
+      });
+      return /* @__PURE__ */ jsx21(ToolUIDisplay, { ...part, part, UI: Tool, addResult });
+    }
+    default:
+      const unhandledType = type;
+      throw new Error(`Unknown content part type: ${unhandledType}`);
+  }
+};
+var MessageContentPartImpl = ({
+  partIndex,
+  components
+}) => {
+  const messageRuntime = useMessageRuntime();
+  const runtime = useMemo6(
+    () => messageRuntime.unstable_getContentPartByIndex(partIndex),
+    [messageRuntime, partIndex]
+  );
+  return /* @__PURE__ */ jsx21(ContentPartRuntimeProvider, { runtime, children: /* @__PURE__ */ jsx21(MessageContentPartComponent, { components }) });
+};
+var MessageContentPart = memo2(
+  MessageContentPartImpl,
+  (prev, next) => prev.partIndex === next.partIndex && prev.components?.Text === next.components?.Text && prev.components?.Image === next.components?.Image && prev.components?.UI === next.components?.UI && prev.components?.tools === next.components?.tools
+);
+var MessagePrimitiveContent = ({
+  components
+}) => {
+  const contentLength = useMessage((s) => s.content.length) || 1;
+  return Array.from({ length: contentLength }, (_, index) => /* @__PURE__ */ jsx21(MessageContentPart, { partIndex: index, components }, index));
+};
+MessagePrimitiveContent.displayName = "MessagePrimitive.Content";
+
+// src/primitives/message/MessageInProgress.tsx
+var MessagePrimitiveInProgress = () => {
+  return null;
+};
+MessagePrimitiveInProgress.displayName = "MessagePrimitive.InProgress";
+
+// src/primitives/message/MessageAttachments.tsx
+import { memo as memo3, useMemo as useMemo8 } from "react";
+
+// src/context/react/AttachmentContext.ts
+import { createContext as createContext6, useContext as useContext3 } from "react";
 var AttachmentContext = createContext6(
   null
 );
@@ -1757,84 +2240,91 @@ function useAttachmentContext(options) {
   const context = useContext3(AttachmentContext);
   if (!options?.optional && !context)
     throw new Error(
-      "This component must be used within a ComposerPrimitive.Attachments or MessagePrimitive.Attachments component."
+      "This component must be used within a ComposerPrimitive.Attachments or MessagePrimitive.Attachments component."
+    );
+  return context;
+}
+function useThreadComposerAttachmentContext(options) {
+  const context = useAttachmentContext(options);
+  if (!context) return null;
+  if (context.source !== "thread-composer")
+    throw new Error(
+      "This component must be used within a thread's ComposerPrimitive.Attachments component."
     );
   return context;
 }
-function useComposerAttachmentContext(options) {
+function useEditComposerAttachmentContext(options) {
   const context = useAttachmentContext(options);
   if (!context) return null;
-  if (context.type !== "composer")
+  if (context.source !== "edit-composer")
     throw new Error(
-      "This component must be used within a ComposerPrimitive.Attachments component."
+      "This component must be used within a messages's ComposerPrimitive.Attachments component."
     );
   return context;
 }
 function useMessageAttachmentContext(options) {
   const context = useAttachmentContext(options);
   if (!context) return null;
-  if (context.type !== "message")
+  if (context.source !== "message")
     throw new Error(
       "This component must be used within a MessagePrimitive.Attachments component."
     );
   return context;
 }
-var { useAttachment, useAttachmentStore } = createContextStoreHook(
+function useAttachmentRuntime(options) {
+  const attachmentRuntime = useAttachmentContext(options);
+  if (!attachmentRuntime) return null;
+  return attachmentRuntime.useAttachmentRuntime();
+}
+var { useAttachment } = createContextStoreHook(
   useAttachmentContext,
   "useAttachment"
 );
-var {
-  useAttachment: useComposerAttachment,
-  useAttachmentStore: useComposerAttachmentStore
-} = createContextStoreHook(useComposerAttachmentContext, "useAttachment");
-var {
-  useAttachment: useMessageAttachment,
-  useAttachmentStore: useMessageAttachmentStore
-} = createContextStoreHook(useMessageAttachmentContext, "useAttachment");
+var { useAttachment: useThreadComposerAttachment } = createContextStoreHook(useThreadComposerAttachmentContext, "useAttachment");
+var { useAttachment: useEditComposerAttachment } = createContextStoreHook(useEditComposerAttachmentContext, "useAttachment");
+var { useAttachment: useMessageAttachment } = createContextStoreHook(
+  useMessageAttachmentContext,
+  "useAttachment"
+);
 
-// src/context/providers/MessageAttachmentProvider.tsx
-import { useEffect as useEffect10, useState as useState8 } from "react";
-import { create as create9 } from "zustand";
+// src/context/providers/AttachmentRuntimeProvider.tsx
+import {
+  useEffect as useEffect10,
+  useMemo as useMemo7,
+  useState as useState8
+} from "react";
+import { create as create8 } from "zustand";
 import { jsx as jsx22 } from "react/jsx-runtime";
-var getAttachment = (message, useAttachment2, partIndex) => {
-  if (message.role !== "user") return null;
-  const attachments = message.attachments;
-  const attachment = attachments[partIndex];
-  if (!attachment) return null;
-  const currentState = useAttachment2?.getState();
-  if (currentState && currentState.attachment === attachment) return null;
-  return Object.freeze({ attachment });
-};
-var useMessageAttachmentContext2 = (partIndex) => {
-  const messageStore = useMessageStore();
-  const [context] = useState8(
-    () => {
-      const useAttachment2 = create9(
-        () => getAttachment(messageStore.getState(), void 0, partIndex)
-      );
-      return { type: "message", useAttachment: useAttachment2 };
-    }
-  );
+var useAttachmentRuntimeStore = (runtime) => {
+  const [store] = useState8(() => create8(() => runtime));
   useEffect10(() => {
-    const syncAttachment = (messageState) => {
-      const newState = getAttachment(
-        messageState,
-        context.useAttachment,
-        partIndex
-      );
-      if (!newState) return;
-      writableStore(context.useAttachment).setState(newState, true);
-    };
-    syncAttachment(messageStore.getState());
-    return messageStore.subscribe(syncAttachment);
-  }, [context, messageStore, partIndex]);
-  return context;
+    writableStore(store).setState(runtime, true);
+  }, [runtime, store]);
+  return store;
+};
+var useAttachmentStore = (runtime) => {
+  const [store] = useState8(() => create8(() => runtime.getState()));
+  useEffect10(() => {
+    const updateState = () => writableStore(store).setState(runtime.getState(), true);
+    updateState();
+    return runtime.subscribe(updateState);
+  }, [runtime, store]);
+  return store;
 };
-var MessageAttachmentProvider = ({
-  attachmentIndex: partIndex,
+var AttachmentRuntimeProvider = ({
+  runtime,
   children
 }) => {
-  const context = useMessageAttachmentContext2(partIndex);
+  const useAttachmentRuntime2 = useAttachmentRuntimeStore(runtime);
+  const useAttachment2 = useAttachmentStore(runtime);
+  const source = useAttachment2((s) => s.source);
+  const context = useMemo7(() => {
+    return {
+      source,
+      useAttachmentRuntime: useAttachmentRuntime2,
+      useAttachment: useAttachment2
+    };
+  }, [useAttachmentRuntime2, useAttachment2]);
   return /* @__PURE__ */ jsx22(AttachmentContext.Provider, { value: context, children });
 };
 
@@ -1862,7 +2352,12 @@ var AttachmentComponent = ({ components }) => {
   return /* @__PURE__ */ jsx23(Component, {});
 };
 var MessageAttachmentImpl = ({ components, attachmentIndex }) => {
-  return /* @__PURE__ */ jsx23(MessageAttachmentProvider, { attachmentIndex, children: /* @__PURE__ */ jsx23(AttachmentComponent, { components }) });
+  const messageRuntime = useMessageRuntime();
+  const runtime = useMemo8(
+    () => messageRuntime.unstable_getAttachmentByIndex(attachmentIndex),
+    [messageRuntime, attachmentIndex]
+  );
+  return /* @__PURE__ */ jsx23(AttachmentRuntimeProvider, { runtime, children: /* @__PURE__ */ jsx23(AttachmentComponent, { components }) });
 };
 var MessageAttachment = memo3(
   MessageAttachmentImpl,
@@ -2036,51 +2531,8 @@ var ComposerPrimitiveAddAttachment = createActionButton(
 );
 
 // src/primitives/composer/ComposerAttachments.tsx
-import { memo as memo4 } from "react";
-
-// src/context/providers/ComposerAttachmentProvider.tsx
-import { useEffect as useEffect12, useState as useState9 } from "react";
-import { create as create10 } from "zustand";
+import { memo as memo4, useMemo as useMemo9 } from "react";
 import { jsx as jsx27 } from "react/jsx-runtime";
-var getAttachment2 = ({ attachments }, useAttachment2, partIndex) => {
-  const attachment = attachments[partIndex];
-  if (!attachment) return null;
-  const currentState = useAttachment2?.getState();
-  if (currentState && currentState.attachment === attachment) return null;
-  return Object.freeze({ attachment });
-};
-var useComposerAttachmentContext2 = (partIndex) => {
-  const threadComposerStore = useThreadComposerStore();
-  const [context] = useState9(
-    () => {
-      const useAttachment2 = create10(
-        () => getAttachment2(threadComposerStore.getState(), void 0, partIndex)
-      );
-      return { type: "composer", useAttachment: useAttachment2 };
-    }
-  );
-  useEffect12(() => {
-    const syncAttachment = (composer) => {
-      const newState = getAttachment2(
-        composer,
-        context.useAttachment,
-        partIndex
-      );
-      if (!newState) return;
-      writableStore(context.useAttachment).setState(newState, true);
-    };
-    syncAttachment(threadComposerStore.getState());
-    return threadComposerStore.subscribe(syncAttachment);
-  }, [context, threadComposerStore, partIndex]);
-  return context;
-};
-var ComposerAttachmentProvider = ({ attachmentIndex: partIndex, children }) => {
-  const context = useComposerAttachmentContext2(partIndex);
-  return /* @__PURE__ */ jsx27(AttachmentContext.Provider, { value: context, children });
-};
-
-// src/primitives/composer/ComposerAttachments.tsx
-import { jsx as jsx28 } from "react/jsx-runtime";
 var getComponent2 = (components, attachment) => {
   const type = attachment.type;
   switch (type) {
@@ -2096,22 +2548,27 @@ var getComponent2 = (components, attachment) => {
   }
 };
 var AttachmentComponent2 = ({ components }) => {
-  const Component = useComposerAttachment(
-    (a) => getComponent2(components, a.attachment)
+  const Component = useThreadComposerAttachment(
+    (a) => getComponent2(components, a)
   );
   if (!Component) return null;
-  return /* @__PURE__ */ jsx28(Component, {});
+  return /* @__PURE__ */ jsx27(Component, {});
 };
 var ComposerAttachmentImpl = ({ components, attachmentIndex }) => {
-  return /* @__PURE__ */ jsx28(ComposerAttachmentProvider, { attachmentIndex, children: /* @__PURE__ */ jsx28(AttachmentComponent2, { components }) });
+  const composerRuntime = useComposerRuntime();
+  const runtime = useMemo9(
+    () => composerRuntime.unstable_getAttachmentByIndex(attachmentIndex),
+    [composerRuntime, attachmentIndex]
+  );
+  return /* @__PURE__ */ jsx27(AttachmentRuntimeProvider, { runtime, children: /* @__PURE__ */ jsx27(AttachmentComponent2, { components }) });
 };
 var ComposerAttachment = memo4(
   ComposerAttachmentImpl,
   (prev, next) => prev.attachmentIndex === next.attachmentIndex && prev.components?.Image === next.components?.Image && prev.components?.Document === next.components?.Document && prev.components?.File === next.components?.File && prev.components?.Attachment === next.components?.Attachment
 );
 var ComposerPrimitiveAttachments = ({ components }) => {
-  const attachmentsCount = useThreadComposer((s) => s.attachments.length);
-  return Array.from({ length: attachmentsCount }, (_, index) => /* @__PURE__ */ jsx28(
+  const attachmentsCount = useComposer((s) => s.attachments.length);
+  return Array.from({ length: attachmentsCount }, (_, index) => /* @__PURE__ */ jsx27(
     ComposerAttachment,
     {
       attachmentIndex: index,
@@ -2156,9 +2613,9 @@ __export(thread_exports, {
 // src/primitives/thread/ThreadRoot.tsx
 import { Primitive as Primitive11 } from "@radix-ui/react-primitive";
 import { forwardRef as forwardRef17 } from "react";
-import { jsx as jsx29 } from "react/jsx-runtime";
+import { jsx as jsx28 } from "react/jsx-runtime";
 var ThreadPrimitiveRoot = forwardRef17((props, ref) => {
-  return /* @__PURE__ */ jsx29(Primitive11.div, { ...props, ref });
+  return /* @__PURE__ */ jsx28(Primitive11.div, { ...props, ref });
 });
 ThreadPrimitiveRoot.displayName = "ThreadPrimitive.Root";
 
@@ -2232,11 +2689,11 @@ var useOnResizeContent = (callback) => {
 
 // src/utils/hooks/useOnScrollToBottom.tsx
 import { useCallbackRef as useCallbackRef4 } from "@radix-ui/react-use-callback-ref";
-import { useEffect as useEffect13 } from "react";
+import { useEffect as useEffect12 } from "react";
 var useOnScrollToBottom = (callback) => {
   const callbackRef = useCallbackRef4(callback);
   const threadViewportStore = useThreadViewportStore();
-  useEffect13(() => {
+  useEffect12(() => {
     return threadViewportStore.getState().onScrollToBottom(() => {
       callbackRef();
     });
@@ -2295,65 +2752,26 @@ var useThreadViewportAutoScroll = ({
 };
 
 // src/primitives/thread/ThreadViewport.tsx
-import { jsx as jsx30 } from "react/jsx-runtime";
+import { jsx as jsx29 } from "react/jsx-runtime";
 var ThreadPrimitiveViewport = forwardRef18(({ autoScroll, children, ...rest }, forwardedRef) => {
   const autoScrollRef = useThreadViewportAutoScroll({
     autoScroll
   });
   const ref = useComposedRefs4(forwardedRef, autoScrollRef);
-  return /* @__PURE__ */ jsx30(Primitive12.div, { ...rest, ref, children });
+  return /* @__PURE__ */ jsx29(Primitive12.div, { ...rest, ref, children });
 });
 ThreadPrimitiveViewport.displayName = "ThreadPrimitive.Viewport";
 
 // src/primitives/thread/ThreadMessages.tsx
-import { memo as memo5, useMemo as useMemo6 } from "react";
+import { memo as memo5, useMemo as useMemo10 } from "react";
 
 // src/context/providers/MessageRuntimeProvider.tsx
-import { useEffect as useEffect14, useState as useState10 } from "react";
-import { create as create13 } from "zustand";
-
-// src/context/stores/EditComposer.ts
-import { create as create11 } from "zustand";
-var makeEditComposerStore = ({
-  onEdit,
-  onSend
-}) => create11()((set, get) => ({
-  type: "edit",
-  get value() {
-    return get().text;
-  },
-  setValue(value) {
-    get().setText(value);
-  },
-  text: "",
-  setText: (text) => {
-    set({ text, isEmpty: text.trim().length === 0 });
-  },
-  canCancel: false,
-  isEditing: false,
-  isEmpty: true,
-  edit: () => {
-    const text = onEdit();
-    set({
-      isEditing: true,
-      canCancel: true,
-      isEmpty: text.trim().length === 0,
-      text
-    });
-  },
-  send: () => {
-    const text = get().text;
-    set({ isEditing: false, canCancel: false });
-    onSend(text);
-  },
-  cancel: () => {
-    set({ isEditing: false, canCancel: false });
-  }
-}));
+import { useEffect as useEffect13, useState as useState9 } from "react";
+import { create as create10 } from "zustand";
 
 // src/context/stores/MessageUtils.ts
-import { create as create12 } from "zustand";
-var makeMessageUtilsStore = () => create12((set) => {
+import { create as create9 } from "zustand";
+var makeMessageUtilsStore = () => create9((set) => {
   let utterance = null;
   return {
     isCopied: false,
@@ -2383,17 +2801,17 @@ var makeMessageUtilsStore = () => create12((set) => {
 });
 
 // src/context/providers/MessageRuntimeProvider.tsx
-import { jsx as jsx31 } from "react/jsx-runtime";
+import { jsx as jsx30 } from "react/jsx-runtime";
 var useMessageRuntimeStore = (runtime) => {
-  const [store] = useState10(() => create13(() => runtime));
-  useEffect14(() => {
+  const [store] = useState9(() => create10(() => runtime));
+  useEffect13(() => {
     writableStore(store).setState(runtime, true);
   }, [runtime, store]);
   return store;
 };
 var useMessageStore2 = (runtime) => {
-  const [store] = useState10(() => create13(() => runtime.getState()));
-  useEffect14(() => {
+  const [store] = useState9(() => create10(() => runtime.getState()));
+  useEffect13(() => {
     const updateState = () => writableStore(store).setState(runtime.getState(), true);
     updateState();
     return runtime.subscribe(updateState);
@@ -2401,33 +2819,17 @@ var useMessageStore2 = (runtime) => {
   return store;
 };
 var useMessageUtilsStore2 = () => {
-  const [store] = useState10(() => makeMessageUtilsStore());
+  const [store] = useState9(() => makeMessageUtilsStore());
   return store;
 };
 var useEditComposerStore2 = (useMessageRuntime2) => {
-  const [store] = useState10(
-    () => makeEditComposerStore({
-      onEdit: () => {
-        const text = getThreadMessageText(
-          useMessageRuntime2.getState().getState()
-        );
-        return text;
-      },
-      onSend: (text) => {
-        const message = useMessageRuntime2.getState().getState();
-        const previousText = getThreadMessageText(message);
-        if (previousText === text) return;
-        const nonTextParts = message.content.filter(
-          (part) => part.type !== "text" && part.type !== "ui"
-        );
-        useMessageRuntime2.getState().edit({
-          role: message.role,
-          content: [{ type: "text", text }, ...nonTextParts],
-          attachments: message.attachments ?? []
-        });
-      }
-    })
-  );
+  const runtime = useMessageRuntime2.getState().composer;
+  const [store] = useState9(() => create10(() => runtime.getState()));
+  useEffect13(() => {
+    const updateState = () => writableStore(store).setState(runtime.getState());
+    updateState();
+    return runtime.subscribe(updateState);
+  }, [runtime, store]);
   return store;
 };
 var MessageRuntimeProvider = ({
@@ -2438,14 +2840,14 @@ var MessageRuntimeProvider = ({
   const useMessage2 = useMessageStore2(runtime);
   const useMessageUtils2 = useMessageUtilsStore2();
   const useEditComposer2 = useEditComposerStore2(useMessageRuntime2);
-  const [context] = useState10(() => {
+  const [context] = useState9(() => {
     return { useMessageRuntime: useMessageRuntime2, useMessage: useMessage2, useMessageUtils: useMessageUtils2, useEditComposer: useEditComposer2 };
   });
-  return /* @__PURE__ */ jsx31(MessageContext.Provider, { value: context, children });
+  return /* @__PURE__ */ jsx30(MessageContext.Provider, { value: context, children });
 };
 
 // src/primitives/thread/ThreadMessages.tsx
-import { jsx as jsx32 } from "react/jsx-runtime";
+import { jsx as jsx31 } from "react/jsx-runtime";
 var isComponentsSame = (prev, next) => {
   return prev.Message === next.Message && prev.EditComposer === next.EditComposer && prev.UserEditComposer === next.UserEditComposer && prev.AssistantEditComposer === next.AssistantEditComposer && prev.SystemEditComposer === next.SystemEditComposer && prev.UserMessage === next.UserMessage && prev.AssistantMessage === next.AssistantMessage && prev.SystemMessage === next.SystemMessage;
 };
@@ -2481,18 +2883,18 @@ var ThreadMessageComponent = ({
   const role = useMessage((m) => m.role);
   const isEditing = useEditComposer((c) => c.isEditing);
   const Component = getComponent3(components, role, isEditing);
-  return /* @__PURE__ */ jsx32(Component, {});
+  return /* @__PURE__ */ jsx31(Component, {});
 };
 var ThreadMessageImpl = ({
   messageIndex,
   components
 }) => {
   const threadRuntime = useThreadRuntime();
-  const runtime = useMemo6(
+  const runtime = useMemo10(
     () => threadRuntime.unstable_getMesssageByIndex(messageIndex),
     [threadRuntime, messageIndex]
   );
-  return /* @__PURE__ */ jsx32(MessageRuntimeProvider, { runtime, children: /* @__PURE__ */ jsx32(ThreadMessageComponent, { components }) });
+  return /* @__PURE__ */ jsx31(MessageRuntimeProvider, { runtime, children: /* @__PURE__ */ jsx31(ThreadMessageComponent, { components }) });
 };
 var ThreadMessage = memo5(
   ThreadMessageImpl,
@@ -2501,9 +2903,9 @@ var ThreadMessage = memo5(
 var ThreadPrimitiveMessagesImpl = ({
   components
 }) => {
-  const messagesLength = useThreadMessages((t) => t.length);
+  const messagesLength = useThread((t) => t.messages.length);
   if (messagesLength === 0) return null;
-  return Array.from({ length: messagesLength }, (_, index) => /* @__PURE__ */ jsx32(ThreadMessage, { messageIndex: index, components }, index));
+  return Array.from({ length: messagesLength }, (_, index) => /* @__PURE__ */ jsx31(ThreadMessage, { messageIndex: index, components }, index));
 };
 ThreadPrimitiveMessagesImpl.displayName = "ThreadPrimitive.Messages";
 var ThreadPrimitiveMessages = memo5(
@@ -2545,7 +2947,7 @@ var subscribeToMainThread = (runtime, callback) => {
 };
 
 // src/runtimes/local/useLocalRuntime.tsx
-import { useInsertionEffect, useMemo as useMemo7, useState as useState12 } from "react";
+import { useInsertionEffect, useMemo as useMemo11, useState as useState11 } from "react";
 
 // src/runtimes/core/BaseAssistantRuntimeCore.tsx
 var BaseAssistantRuntimeCore = class {
@@ -2575,7 +2977,7 @@ var internal_exports = {};
 __export(internal_exports, {
   AssistantRuntime: () => AssistantRuntime,
   BaseAssistantRuntimeCore: () => BaseAssistantRuntimeCore,
-  BaseThreadComposerRuntimeCore: () => BaseThreadComposerRuntimeCore,
+  DefaultThreadComposerRuntimeCore: () => DefaultThreadComposerRuntimeCore,
   MessageRepository: () => MessageRepository,
   ProxyConfigProvider: () => ProxyConfigProvider,
   ThreadRuntime: () => ThreadRuntime,
@@ -2586,46 +2988,70 @@ __export(internal_exports, {
   withSmoothContextProvider: () => withSmoothContextProvider
 });
 
-// src/runtimes/utils/BaseThreadComposerRuntimeCore.tsx
-var BaseThreadComposerRuntimeCore = class {
-  constructor(runtime) {
-    this.runtime = runtime;
-    this.connect();
-  }
-  _attachmentAdapter;
+// src/runtimes/composer/BaseComposerRuntimeCore.tsx
+var isAttachmentComplete = (a) => a.status.type === "complete";
+var BaseComposerRuntimeCore = class {
+  isEditing = true;
   attachmentAccept = "*";
+  _attachments = [];
+  set attachments(value) {
+    this._attachments = value;
+    this.notifySubscribers();
+  }
+  get attachments() {
+    return this._attachments;
+  }
   get isEmpty() {
     return !this.text.trim() && !this.attachments.length;
   }
-  _canCancel = false;
-  get canCancel() {
-    return this._canCancel;
+  _text = "";
+  get text() {
+    return this._text;
   }
-  connect() {
-    return this.runtime.subscribe(() => {
-      if (this.canCancel !== this.runtime.capabilities.cancel) {
-        this._canCancel = this.runtime.capabilities.cancel;
-        this.notifySubscribers();
-      }
-    });
+  setText(value) {
+    this._text = value;
+    this.notifySubscribers();
+  }
+  reset() {
+    this._text = "";
+    this._attachments = [];
+    this.notifySubscribers();
+  }
+  async send() {
+    const attachments = this._attachmentAdapter ? await Promise.all(
+      this.attachments.map(async (a) => {
+        if (isAttachmentComplete(a)) return a;
+        const result = await this._attachmentAdapter.send(a);
+        if (result.status?.type !== "complete") {
+          result.status = { type: "complete" };
+        }
+        return result;
+      })
+    ) : [];
+    const message = {
+      role: "user",
+      content: this.text ? [{ type: "text", text: this.text }] : [],
+      attachments
+    };
+    this.reset();
+    this.handleSend(message);
   }
+  _attachmentAdapter;
   setAttachmentAdapter(adapter) {
     this._attachmentAdapter = adapter;
     const accept = adapter?.accept ?? "*";
     if (this.attachmentAccept !== accept) {
       this.attachmentAccept = accept;
-      return true;
+      this.notifySubscribers();
     }
-    return false;
-  }
-  _attachments = [];
-  get attachments() {
-    return this._attachments;
   }
   async addAttachment(file) {
     if (!this._attachmentAdapter)
       throw new Error("Attachments are not supported");
     const attachment = await this._attachmentAdapter.add({ file });
+    if (attachment.status === void 0) {
+      attachment.status = { type: "requires-action", reason: "composer-send" };
+    }
     this._attachments = [...this._attachments, attachment];
     this.notifySubscribers();
   }
@@ -2639,44 +3065,47 @@ var BaseThreadComposerRuntimeCore = class {
     this._attachments = this._attachments.toSpliced(index, 1);
     this.notifySubscribers();
   }
-  _text = "";
-  get text() {
-    return this._text;
+  _subscriptions = /* @__PURE__ */ new Set();
+  notifySubscribers() {
+    for (const callback of this._subscriptions) callback();
   }
-  setText(value) {
-    this._text = value;
-    this.notifySubscribers();
+  subscribe(callback) {
+    this._subscriptions.add(callback);
+    return () => this._subscriptions.delete(callback);
   }
-  reset() {
-    this._text = "";
-    this._attachments = [];
-    this.notifySubscribers();
+};
+
+// src/runtimes/composer/DefaultThreadComposerRuntimeCore.tsx
+var DefaultThreadComposerRuntimeCore = class extends BaseComposerRuntimeCore {
+  constructor(runtime) {
+    super();
+    this.runtime = runtime;
+    this.connect();
   }
-  async send() {
-    const attachments = this._attachmentAdapter ? await Promise.all(
-      this.attachments.map(
-        async (a) => await this._attachmentAdapter.send(a)
-      )
-    ) : [];
+  _canCancel = false;
+  get canCancel() {
+    return this._canCancel;
+  }
+  get attachments() {
+    return super.attachments;
+  }
+  connect() {
+    return this.runtime.subscribe(() => {
+      if (this.canCancel !== this.runtime.capabilities.cancel) {
+        this._canCancel = this.runtime.capabilities.cancel;
+        this.notifySubscribers();
+      }
+    });
+  }
+  async handleSend(message) {
     this.runtime.append({
-      parentId: this.runtime.messages.at(-1)?.id ?? null,
-      role: "user",
-      content: this.text ? [{ type: "text", text: this.text }] : [],
-      attachments
+      ...message,
+      parentId: this.runtime.messages.at(-1)?.id ?? null
     });
-    this.reset();
   }
   async cancel() {
     this.runtime.cancelRun();
   }
-  _subscriptions = /* @__PURE__ */ new Set();
-  notifySubscribers() {
-    for (const callback of this._subscriptions) callback();
-  }
-  subscribe(callback) {
-    this._subscriptions.add(callback);
-    return () => this._subscriptions.delete(callback);
-  }
 };
 
 // src/utils/ProxyConfigProvider.ts
@@ -2804,11 +3233,11 @@ var MessageRepository = class {
     }
   }
   getMessages() {
-    const messages = new Array(this.head?.level ?? 0);
+    const messages2 = new Array(this.head?.level ?? 0);
     for (let current = this.head; current; current = current.prev) {
-      messages[current.level] = current.current;
+      messages2[current.level] = current.current;
     }
-    return messages;
+    return messages2;
   }
   addOrUpdateMessage(parentId, message) {
     const existingItem = this.messages.get(message.id);
@@ -2934,11 +3363,11 @@ var MessageRepository = class {
       messages: exportItems
     };
   }
-  import({ headId, messages }) {
-    for (const { message, parentId } of messages) {
+  import({ headId, messages: messages2 }) {
+    for (const { message, parentId } of messages2) {
       this.addOrUpdateMessage(parentId, message);
     }
-    this.resetHead(headId ?? messages.at(-1)?.message.id ?? null);
+    this.resetHead(headId ?? messages2.at(-1)?.message.id ?? null);
   }
 };
 
@@ -2953,7 +3382,7 @@ import {
   forwardRef as forwardRef19
 } from "react";
 import classNames from "classnames";
-import { jsx as jsx33 } from "react/jsx-runtime";
+import { jsx as jsx32 } from "react/jsx-runtime";
 var withDefaultProps = ({
   className,
   ...defaultProps
@@ -2969,7 +3398,7 @@ var withDefaults = (Component, defaultProps) => {
   const WithDefaults = forwardRef19(
     (props, ref) => {
       const ComponentAsAny = Component;
-      return /* @__PURE__ */ jsx33(ComponentAsAny, { ...getProps(props), ref });
+      return /* @__PURE__ */ jsx32(ComponentAsAny, { ...getProps(props), ref });
     }
   );
   WithDefaults.displayName = "withDefaults(" + (typeof Component === "string" ? Component : Component.displayName) + ")";
@@ -2977,9 +3406,9 @@ var withDefaults = (Component, defaultProps) => {
 };
 
 // src/ui/base/tooltip.tsx
-import { jsx as jsx34 } from "react/jsx-runtime";
+import { jsx as jsx33 } from "react/jsx-runtime";
 var Tooltip = (props) => {
-  return /* @__PURE__ */ jsx34(TooltipPrimitive.Provider, { children: /* @__PURE__ */ jsx34(TooltipPrimitive.Root, { ...props }) });
+  return /* @__PURE__ */ jsx33(TooltipPrimitive.Provider, { children: /* @__PURE__ */ jsx33(TooltipPrimitive.Root, { ...props }) });
 };
 Tooltip.displayName = "Tooltip";
 var TooltipTrigger = TooltipPrimitive.Trigger;
@@ -2993,7 +3422,7 @@ TooltipContent.displayName = "TooltipContent";
 import { cva } from "class-variance-authority";
 import { Primitive as Primitive13 } from "@radix-ui/react-primitive";
 import { forwardRef as forwardRef20 } from "react";
-import { jsx as jsx35 } from "react/jsx-runtime";
+import { jsx as jsx34 } from "react/jsx-runtime";
 var buttonVariants = cva("aui-button", {
   variants: {
     variant: {
@@ -3013,7 +3442,7 @@ var buttonVariants = cva("aui-button", {
 });
 var Button = forwardRef20(
   ({ className, variant, size, ...props }, ref) => {
-    return /* @__PURE__ */ jsx35(
+    return /* @__PURE__ */ jsx34(
       Primitive13.button,
       {
         className: buttonVariants({ variant, size, className }),
@@ -3026,77 +3455,17 @@ var Button = forwardRef20(
 Button.displayName = "Button";
 
 // src/ui/base/tooltip-icon-button.tsx
-import { jsx as jsx36, jsxs as jsxs3 } from "react/jsx-runtime";
+import { jsx as jsx35, jsxs as jsxs2 } from "react/jsx-runtime";
 var TooltipIconButton = forwardRef21(({ children, tooltip, side = "bottom", ...rest }, ref) => {
-  return /* @__PURE__ */ jsxs3(Tooltip, { children: [
-    /* @__PURE__ */ jsx36(TooltipTrigger, { asChild: true, children: /* @__PURE__ */ jsxs3(Button, { variant: "ghost", size: "icon", ...rest, ref, children: [
+  return /* @__PURE__ */ jsxs2(Tooltip, { children: [
+    /* @__PURE__ */ jsx35(TooltipTrigger, { asChild: true, children: /* @__PURE__ */ jsxs2(Button, { variant: "ghost", size: "icon", ...rest, ref, children: [
       children,
-      /* @__PURE__ */ jsx36("span", { className: "aui-sr-only", children: tooltip })
+      /* @__PURE__ */ jsx35("span", { className: "aui-sr-only", children: tooltip })
     ] }) }),
-    /* @__PURE__ */ jsx36(TooltipContent, { side, children: tooltip })
+    /* @__PURE__ */ jsx35(TooltipContent, { side, children: tooltip })
   ] });
-});
-TooltipIconButton.displayName = "TooltipIconButton";
-
-// src/api/subscribable/BaseSubject.ts
-var BaseSubject = class {
-  _subscriptions = /* @__PURE__ */ new Set();
-  _connection;
-  get isConnected() {
-    return !!this._connection;
-  }
-  notifySubscribers() {
-    for (const callback of this._subscriptions) callback();
-  }
-  _updateConnection() {
-    if (this._subscriptions.size > 0) {
-      if (this._connection) return;
-      this._connection = this._connect();
-    } else {
-      this._connection?.();
-      this._connection = void 0;
-    }
-  }
-  subscribe(callback) {
-    this._subscriptions.add(callback);
-    this._updateConnection();
-    return () => {
-      this._subscriptions.delete(callback);
-      this._updateConnection();
-    };
-  }
-};
-
-// src/api/subscribable/NestedSubscriptionSubject.ts
-var NestedSubscriptionSubject = class extends BaseSubject {
-  constructor(binding) {
-    super();
-    this.binding = binding;
-  }
-  getState() {
-    return this.binding.getState();
-  }
-  _connect() {
-    const callback = () => {
-      this.notifySubscribers();
-    };
-    let lastState = this.binding.getState();
-    let innerUnsubscribe = lastState.subscribe(callback);
-    const onRuntimeUpdate = () => {
-      const newState = this.binding.getState();
-      if (newState === lastState) return;
-      lastState = newState;
-      innerUnsubscribe?.();
-      innerUnsubscribe = this.binding.getState().subscribe(callback);
-      callback();
-    };
-    const outerUnsubscribe = this.binding.subscribe(onRuntimeUpdate);
-    return () => {
-      outerUnsubscribe?.();
-      innerUnsubscribe();
-    };
-  }
-};
+});
+TooltipIconButton.displayName = "TooltipIconButton";
 
 // src/api/AssistantRuntime.ts
 var AssistantRuntime = class {
@@ -3128,245 +3497,11 @@ var AssistantRuntime = class {
   }
 };
 
-// src/api/ContentPartRuntime.ts
-var ContentPartRuntime = class {
-  constructor(contentBinding, messageApi, threadApi) {
-    this.contentBinding = contentBinding;
-    this.messageApi = messageApi;
-    this.threadApi = threadApi;
-  }
-  getState() {
-    return this.contentBinding.getState();
-  }
-  addToolResult(result) {
-    const message = this.messageApi.getState();
-    if (!message) throw new Error("Message is not available");
-    const state = this.contentBinding.getState();
-    if (!state) throw new Error("Content part is not available");
-    if (state.part.type !== "tool-call")
-      throw new Error("Tried to add tool result to non-tool content part");
-    const toolName = state.part.toolName;
-    const toolCallId = state.part.toolCallId;
-    this.threadApi.getState().addToolResult({
-      messageId: message.id,
-      toolName,
-      toolCallId,
-      result
-    });
-  }
-};
-
-// src/api/subscribable/shallowEqual.ts
-function shallowEqual(objA, objB) {
-  if (objA === void 0 && objB === void 0) return true;
-  if (objA === void 0) return false;
-  if (objB === void 0) return false;
-  for (const key of Object.keys(objA)) {
-    const valueA = objA[key];
-    const valueB = objB[key];
-    if (!Object.is(valueA, valueB)) return false;
-  }
-  return true;
-}
-
-// src/api/subscribable/ShallowMemoizeSubject.ts
-var ShallowMemoizeSubject = class extends BaseSubject {
-  constructor(binding) {
-    super();
-    this.binding = binding;
-    const state = binding.getState();
-    if (state === void 0)
-      throw new Error("Entry not available in the store");
-    this._previousState = state;
-  }
-  _previousState;
-  getState = () => {
-    if (!this.isConnected) this._syncState();
-    return this._previousState;
-  };
-  _syncState() {
-    const state = this.binding.getState();
-    if (state === void 0) return false;
-    if (shallowEqual(state, this._previousState)) return false;
-    this._previousState = state;
-    return true;
-  }
-  _connect() {
-    const callback = () => {
-      if (this._syncState()) {
-        this.notifySubscribers();
-      }
-    };
-    return this.binding.subscribe(callback);
-  }
-};
-
-// src/api/MessageRuntime.ts
-var MessageRuntime = class {
-  constructor(_core, _threadBinding) {
-    this._core = _core;
-    this._threadBinding = _threadBinding;
-  }
-  getState() {
-    return this._core.getState();
-  }
-  // TODO improve type
-  edit(message) {
-    const state = this._core.getState();
-    if (!state) throw new Error("Message is not available");
-    this._threadBinding.getState().append({
-      ...message,
-      parentId: state.parentId
-    });
-  }
-  reload() {
-    const state = this._core.getState();
-    if (!state) throw new Error("Message is not available");
-    this._threadBinding.getState().startRun(state.parentId);
-  }
-  speak() {
-    const state = this._core.getState();
-    if (!state) throw new Error("Message is not available");
-    this._threadBinding.getState().speak(state.id);
-  }
-  submitFeedback({ type }) {
-    const state = this._core.getState();
-    if (!state) throw new Error("Message is not available");
-    this._threadBinding.getState().submitFeedback({
-      messageId: state.id,
-      type
-    });
-  }
-  switchToBranch({
-    position,
-    branchId
-  }) {
-    const state = this._core.getState();
-    if (!state) throw new Error("Message is not available");
-    if (branchId && position) {
-      throw new Error("May not specify both branchId and position");
-    } else if (!branchId && !position) {
-      throw new Error("Must specify either branchId or position");
-    }
-    const thread = this._threadBinding.getState();
-    const branches = thread.getBranches(state.id);
-    let targetBranch = branchId;
-    if (position === "previous") {
-      targetBranch = branches[state.branchNumber - 2];
-    } else if (position === "next") {
-      targetBranch = branches[state.branchNumber];
-    }
-    if (!targetBranch) throw new Error("Branch not found");
-    this._threadBinding.getState().switchToBranch(targetBranch);
-  }
-  subscribe(callback) {
-    return this._core.subscribe(callback);
-  }
-  getContentPartByIdx(idx) {
-    if (idx < 0) throw new Error("Message index must be >= 0");
-    return new ContentPartRuntime(
-      new ShallowMemoizeSubject({
-        getState: () => {
-          const state = this.getState();
-          if (!state) return void 0;
-          const message = state.message;
-          const part = message.content[idx];
-          if (!part) return void 0;
-          return {
-            part,
-            status: toContentPartStatus(message, idx, part)
-          };
-        },
-        subscribe: (callback) => this._core.subscribe(callback)
-      }),
-      this._core,
-      this._threadBinding
-    );
-  }
-};
-
-// src/api/ThreadComposerRuntime.ts
-var getThreadComposerState = (runtime) => {
-  return Object.freeze({
-    canCancel: runtime.canCancel,
-    isEmpty: runtime.isEmpty,
-    text: runtime.text,
-    attachments: runtime.attachments,
-    attachmentAccept: runtime.attachmentAccept
-  });
-};
-var ThreadComposerRuntime = class {
-  constructor(_core) {
-    this._core = _core;
-  }
-  /**
-   * @deprecated Use `getState().isEmpty` instead. This will be removed in 0.6.0.
-   */
-  get isEmpty() {
-    return this._core.getState().isEmpty;
-  }
-  /**
-   * @deprecated Use `getState().canCancel` instead. This will be removed in 0.6.0.
-   */
-  get canCancel() {
-    return this._core.getState().canCancel;
-  }
-  /**
-   * @deprecated Use `getState().text` instead. This will be removed in 0.6.0.
-   */
-  get text() {
-    return this._core.getState().text;
-  }
-  /**
-   * @deprecated Use `getState().attachmentAccept` instead. This will be removed in 0.6.0.
-   */
-  get attachmentAccept() {
-    return this._core.getState().attachmentAccept;
-  }
-  // TODO should this instead return getAttachmentByIndex([idx]) instead?
-  /**
-   * @deprecated Use `getState().attachments` instead. This will be removed in 0.6.0.
-   */
-  get attachments() {
-    return this._core.getState().attachments;
-  }
-  getState() {
-    return getThreadComposerState(this._core.getState());
-  }
-  setText(text) {
-    this._core.getState().setText(text);
-  }
-  addAttachment(file) {
-    return this._core.getState().addAttachment(file);
-  }
-  // /**
-  //  * @deprecated Use `getAttachmentById(id).removeAttachment` instead. This will be removed in 0.6.0.
-  //  */
-  removeAttachment(attachmentId) {
-    return this._core.getState().removeAttachment(attachmentId);
-  }
-  /**
-   * @deprecated This method will be removed in 0.6.0. Submit feedback if you need this functionality.
-   */
-  reset() {
-    this._core.getState().reset();
-  }
-  send() {
-    this._core.getState().send();
-  }
-  cancel() {
-    this._core.getState().cancel();
-  }
-  subscribe(callback) {
-    return this._core.subscribe(callback);
-  }
-};
-
 // src/api/ThreadRuntime.ts
-var toAppendMessage = (messages, message) => {
+var toAppendMessage = (messages2, message) => {
   if (typeof message === "string") {
     return {
-      parentId: messages.at(-1)?.id ?? null,
+      parentId: messages2.at(-1)?.id ?? null,
       role: "user",
       content: [{ type: "text", text: message }],
       attachments: []
@@ -3376,7 +3511,7 @@ var toAppendMessage = (messages, message) => {
     return message;
   }
   return {
-    parentId: message.parentId ?? messages.at(-1)?.id ?? null,
+    parentId: message.parentId ?? messages2.at(-1)?.id ?? null,
     role: message.role ?? "user",
     content: message.content,
     attachments: message.attachments ?? []
@@ -3389,13 +3524,10 @@ var getThreadState = (runtime) => {
     capabilities: runtime.capabilities,
     isDisabled: runtime.isDisabled,
     isRunning: lastMessage?.role !== "assistant" ? false : lastMessage.status.type === "running",
-    unstable_synchronizer: runtime.unstable_synchronizer
+    messages: runtime.messages
   });
 };
 var ThreadRuntime = class {
-  constructor(_threadBinding) {
-    this._threadBinding = _threadBinding;
-  }
   // public path = "assistant.threads[main]"; // TODO
   /**
    * @deprecated Use `getState().threadId` instead. This will be removed in 0.6.0.
@@ -3421,13 +3553,27 @@ var ThreadRuntime = class {
   get capabilities() {
     return this.getState().capabilities;
   }
-  // TODO this should instead return getMessageByIndex([idx])
+  /**
+   * @deprecated Use `getState().messages` instead. This will be removed in 0.6.0.
+   */
   get messages() {
     return this._threadBinding.getState().messages;
   }
   unstable_getCore() {
     return this._threadBinding.getState();
   }
+  _threadBinding;
+  constructor(threadBinding) {
+    const stateBinding = new LazyMemoizeSubject({
+      getState: () => getThreadState(threadBinding.getState()),
+      subscribe: (callback) => threadBinding.subscribe(callback)
+    });
+    this._threadBinding = {
+      getState: () => threadBinding.getState(),
+      getStateState: () => stateBinding.getState(),
+      subscribe: (callback) => threadBinding.subscribe(callback)
+    };
+  }
   composer = new ThreadComposerRuntime(
     new NestedSubscriptionSubject({
       getState: () => this._threadBinding.getState().composer,
@@ -3435,7 +3581,7 @@ var ThreadRuntime = class {
     })
   );
   getState() {
-    return getThreadState(this._threadBinding.getState());
+    return this._threadBinding.getStateState();
   }
   append(message) {
     this._threadBinding.getState().append(
@@ -3484,6 +3630,18 @@ var ThreadRuntime = class {
   submitFeedback(options) {
     return this._threadBinding.getState().submitFeedback(options);
   }
+  /**
+   * @deprecated Use `getMesssageById(id).unstable_getMessageByIndex(idx).composer` instead. This will be removed in 0.6.0.
+   */
+  getEditComposer(messageId) {
+    return this._threadBinding.getState().getEditComposer(messageId);
+  }
+  /**
+   * @deprecated Use `getMesssageById(id).unstable_getMessageByIndex(idx).composer.beginEdit()` instead. This will be removed in 0.6.0.
+   */
+  beginEdit(messageId) {
+    return this._threadBinding.getState().beginEdit(messageId);
+  }
   export() {
     return this._threadBinding.getState().export();
   }
@@ -3495,15 +3653,15 @@ var ThreadRuntime = class {
     return new MessageRuntime(
       new ShallowMemoizeSubject({
         getState: () => {
-          const messages = this.messages;
-          const message = messages[idx];
-          if (!message) return void 0;
+          const messages2 = this.getState().messages;
+          const message = messages2[idx];
+          if (!message) return SKIP_UPDATE;
           const branches = this._threadBinding.getState().getBranches(message.id);
           return {
             ...message,
             message,
-            isLast: idx === messages.length - 1,
-            parentId: messages[idx - 1]?.id ?? null,
+            isLast: idx === messages2.length - 1,
+            parentId: messages2[idx - 1]?.id ?? null,
             branches,
             branchNumber: branches.indexOf(message.id) + 1,
             branchCount: branches.length
@@ -3518,12 +3676,12 @@ var ThreadRuntime = class {
 
 // src/runtimes/edge/converters/fromLanguageModelMessages.ts
 var fromLanguageModelMessages = (lm, { mergeRoundtrips }) => {
-  const messages = [];
+  const messages2 = [];
   for (const lmMessage of lm) {
     const role = lmMessage.role;
     switch (role) {
       case "system": {
-        messages.push({
+        messages2.push({
           role: "system",
           content: [
             {
@@ -3535,7 +3693,7 @@ var fromLanguageModelMessages = (lm, { mergeRoundtrips }) => {
         break;
       }
       case "user": {
-        messages.push({
+        messages2.push({
           role: "user",
           content: lmMessage.content.map((part) => {
             const type = part.type;
@@ -3555,6 +3713,9 @@ var fromLanguageModelMessages = (lm, { mergeRoundtrips }) => {
                 }
                 throw new Error("Only images with URL data are supported");
               }
+              case "file": {
+                throw new Error("File content parts are not supported");
+              }
               default: {
                 const unhandledType = type;
                 throw new Error(`Unknown content part type: ${unhandledType}`);
@@ -3578,20 +3739,20 @@ var fromLanguageModelMessages = (lm, { mergeRoundtrips }) => {
           return part;
         });
         if (mergeRoundtrips) {
-          const previousMessage = messages[messages.length - 1];
+          const previousMessage = messages2[messages2.length - 1];
           if (previousMessage?.role === "assistant") {
             previousMessage.content.push(...newContent);
             break;
           }
         }
-        messages.push({
+        messages2.push({
           role: "assistant",
           content: newContent
         });
         break;
       }
       case "tool": {
-        const previousMessage = messages[messages.length - 1];
+        const previousMessage = messages2[messages2.length - 1];
         if (previousMessage?.role !== "assistant")
           throw new Error(
             "A tool message must be preceded by an assistant message."
@@ -3617,7 +3778,7 @@ var fromLanguageModelMessages = (lm, { mergeRoundtrips }) => {
       }
     }
   }
-  return messages;
+  return messages2;
 };
 
 // src/runtimes/edge/converters/fromLanguageModelTools.ts
@@ -3680,7 +3841,7 @@ var streamUtils = {
 };
 
 // src/runtimes/edge/useEdgeRuntime.ts
-import { useState as useState11 } from "react";
+import { useState as useState10 } from "react";
 
 // src/runtimes/edge/streams/assistantDecoderStream.ts
 function assistantDecoderStream() {
@@ -3775,7 +3936,7 @@ var EdgeChatAdapter = class {
   constructor(options) {
     this.options = options;
   }
-  async *run({ messages, abortSignal, config }) {
+  async *run({ messages: messages2, abortSignal, config }) {
     const headers = new Headers(this.options.headers);
     headers.set("Content-Type", "application/json");
     const result = await fetch(this.options.api, {
@@ -3784,7 +3945,7 @@ var EdgeChatAdapter = class {
       credentials: this.options.credentials ?? "same-origin",
       body: JSON.stringify({
         system: config.system,
-        messages: toCoreMessages(messages),
+        messages: toCoreMessages(messages2),
         tools: config.tools ? toLanguageModelTools(config.tools) : [],
         ...config.callSettings,
         ...config.config,
@@ -3812,7 +3973,7 @@ var useEdgeRuntime = ({
   adapters,
   ...options
 }) => {
-  const [adapter] = useState11(() => new EdgeChatAdapter(options));
+  const [adapter] = useState10(() => new EdgeChatAdapter(options));
   return useLocalRuntime(adapter, {
     initialMessages,
     maxToolRoundtrips,
@@ -3823,6 +3984,43 @@ var useEdgeRuntime = ({
 // src/runtimes/local/shouldContinue.tsx
 var shouldContinue = (result) => result.status?.type === "requires-action" && result.status.reason === "tool-calls" && result.content.every((c) => c.type !== "tool-call" || !!c.result);
 
+// src/runtimes/composer/DefaultEditComposerRuntimeCore.tsx
+var DefaultEditComposerRuntimeCore = class extends BaseComposerRuntimeCore {
+  constructor(runtime, endEditCallback, { parentId, message }) {
+    super();
+    this.runtime = runtime;
+    this.endEditCallback = endEditCallback;
+    this._parentId = parentId;
+    this._previousText = getThreadMessageText(message);
+    this.setText(this._previousText);
+    this._nonTextParts = message.content.filter(
+      (part) => part.type !== "text" && part.type !== "ui"
+    );
+  }
+  get canCancel() {
+    return true;
+  }
+  _nonTextParts;
+  _previousText;
+  _parentId;
+  async handleSend(message) {
+    const text = getThreadMessageText(message);
+    if (text !== this._previousText) {
+      this.runtime.append({
+        ...message,
+        content: [...message.content, ...this._nonTextParts],
+        parentId: this._parentId
+      });
+    }
+    this.endEditCallback();
+    this.notifySubscribers();
+  }
+  async cancel() {
+    this.endEditCallback();
+    this.notifySubscribers();
+  }
+};
+
 // src/runtimes/local/LocalThreadRuntimeCore.tsx
 var LocalThreadRuntimeCore = class {
   constructor(configProvider, adapter, { initialMessages, ...options }) {
@@ -3832,8 +4030,8 @@ var LocalThreadRuntimeCore = class {
     this.options = options;
     if (initialMessages) {
       let parentId = null;
-      const messages = fromCoreMessages(initialMessages);
-      for (const message of messages) {
+      const messages2 = fromCoreMessages(initialMessages);
+      for (const message of messages2) {
         this.repository.addOrUpdateMessage(parentId, message);
         parentId = message.id;
       }
@@ -3857,7 +4055,7 @@ var LocalThreadRuntimeCore = class {
   get messages() {
     return this.repository.getMessages();
   }
-  composer = new BaseThreadComposerRuntimeCore(this);
+  composer = new DefaultThreadComposerRuntimeCore(this);
   getModelConfig() {
     return this.configProvider.getModelConfig();
   }
@@ -3886,6 +4084,23 @@ var LocalThreadRuntimeCore = class {
     }
     if (hasUpdates) this.notifySubscribers();
   }
+  _editComposers = /* @__PURE__ */ new Map();
+  getEditComposer(messageId) {
+    return this._editComposers.get(messageId);
+  }
+  beginEdit(messageId) {
+    if (this._editComposers.has(messageId))
+      throw new Error("Edit already in progress");
+    this._editComposers.set(
+      messageId,
+      new DefaultEditComposerRuntimeCore(
+        this,
+        () => this._editComposers.delete(messageId),
+        this.repository.getMessage(messageId)
+      )
+    );
+    this.notifySubscribers();
+  }
   getBranches(messageId) {
     return this.repository.getBranches(messageId);
   }
@@ -3920,7 +4135,7 @@ var LocalThreadRuntimeCore = class {
     } while (shouldContinue(message));
   }
   async performRoundtrip(parentId, message) {
-    const messages = this.repository.getMessages();
+    const messages2 = this.repository.getMessages();
     this.abortController?.abort();
     this.abortController = new AbortController();
     const initialContent = message.content;
@@ -3975,7 +4190,7 @@ var LocalThreadRuntimeCore = class {
     }
     try {
       const promiseOrGenerator = this.adapter.run({
-        messages,
+        messages: messages2,
         abortSignal: this.abortController.signal,
         config: this.configProvider.getModelConfig(),
         onUpdate: updateMessage
@@ -4117,10 +4332,10 @@ var LocalRuntimeCore = class extends BaseAssistantRuntimeCore {
   } = {}) {
     this.switchToThread(null);
     if (!initialMessages) return;
-    const messages = fromCoreMessages(initialMessages);
+    const messages2 = fromCoreMessages(initialMessages);
     this.thread.import({
-      messages: messages.map((m, idx) => ({
-        parentId: messages[idx - 1]?.id ?? null,
+      messages: messages2.map((m, idx) => ({
+        parentId: messages2[idx - 1]?.id ?? null,
         message: m
       }))
     });
@@ -4138,16 +4353,16 @@ var LocalRuntime = class extends AssistantRuntime {
   }
 };
 var useLocalRuntime = (adapter, options = {}) => {
-  const [runtime] = useState12(() => new LocalRuntimeCore(adapter, options));
+  const [runtime] = useState11(() => new LocalRuntimeCore(adapter, options));
   useInsertionEffect(() => {
     runtime.thread.adapter = adapter;
     runtime.thread.options = options;
   });
-  return useMemo7(() => new LocalRuntime(runtime), [runtime]);
+  return useMemo11(() => new LocalRuntime(runtime), [runtime]);
 };
 
 // src/runtimes/external-store/useExternalStoreRuntime.tsx
-import { useEffect as useEffect15, useMemo as useMemo8, useState as useState13 } from "react";
+import { useEffect as useEffect14, useMemo as useMemo12, useState as useState12 } from "react";
 
 // src/runtimes/external-store/getExternalStoreMessage.tsx
 var symbolInnerMessage = Symbol("innerMessage");
@@ -4158,8 +4373,8 @@ var getExternalStoreMessage = (message) => {
 // src/runtimes/external-store/ThreadMessageConverter.ts
 var ThreadMessageConverter = class {
   cache = /* @__PURE__ */ new WeakMap();
-  convertMessages(messages, converter) {
-    return messages.map((m, idx) => {
+  convertMessages(messages2, converter) {
+    return messages2.map((m, idx) => {
       const cached = this.cache.get(m);
       const newMessage = converter(cached, m, idx);
       this.cache.set(m, newMessage);
@@ -4252,8 +4467,8 @@ var fromThreadMessageLike = (like, fallbackId, fallbackStatus) => {
 };
 
 // src/runtimes/external-store/ExternalStoreThreadRuntimeCore.tsx
-var hasUpcomingMessage = (isRunning, messages) => {
-  return isRunning && messages[messages.length - 1]?.role !== "assistant";
+var hasUpcomingMessage = (isRunning, messages2) => {
+  return isRunning && messages2[messages2.length - 1]?.role !== "assistant";
 };
 var ExternalStoreThreadRuntimeCore = class {
   constructor(configProvider, store) {
@@ -4281,7 +4496,24 @@ var ExternalStoreThreadRuntimeCore = class {
   isDisabled;
   converter = new ThreadMessageConverter();
   _store;
-  composer = new BaseThreadComposerRuntimeCore(this);
+  composer = new DefaultThreadComposerRuntimeCore(this);
+  _editComposers = /* @__PURE__ */ new Map();
+  getEditComposer(messageId) {
+    return this._editComposers.get(messageId);
+  }
+  beginEdit(messageId) {
+    if (this._editComposers.has(messageId))
+      throw new Error("Edit already in progress");
+    this._editComposers.set(
+      messageId,
+      new DefaultEditComposerRuntimeCore(
+        this,
+        () => this._editComposers.delete(messageId),
+        this.repository.getMessage(messageId)
+      )
+    );
+    this.notifySubscribers();
+  }
   get store() {
     return this._store;
   }
@@ -4312,7 +4544,7 @@ var ExternalStoreThreadRuntimeCore = class {
         return;
       }
     }
-    const messages = !store.convertMessage ? store.messages : this.converter.convertMessages(store.messages, (cache, m, idx) => {
+    const messages2 = !store.convertMessage ? store.messages : this.converter.convertMessages(store.messages, (cache, m, idx) => {
       if (!store.convertMessage) return m;
       const isLast = idx === store.messages.length - 1;
       const autoStatus = getAutoStatus(isLast, isRunning);
@@ -4326,18 +4558,18 @@ var ExternalStoreThreadRuntimeCore = class {
       newMessage[symbolInnerMessage] = m;
       return newMessage;
     });
-    for (let i = 0; i < messages.length; i++) {
-      const message = messages[i];
-      const parent = messages[i - 1];
+    for (let i = 0; i < messages2.length; i++) {
+      const message = messages2[i];
+      const parent = messages2[i - 1];
       this.repository.addOrUpdateMessage(parent?.id ?? null, message);
     }
     if (this.assistantOptimisticId) {
       this.repository.deleteMessage(this.assistantOptimisticId);
       this.assistantOptimisticId = null;
     }
-    if (hasUpcomingMessage(isRunning, messages)) {
+    if (hasUpcomingMessage(isRunning, messages2)) {
       this.assistantOptimisticId = this.repository.appendOptimisticMessage(
-        messages.at(-1)?.id ?? null,
+        messages2.at(-1)?.id ?? null,
         {
           role: "assistant",
           content: []
@@ -4345,7 +4577,7 @@ var ExternalStoreThreadRuntimeCore = class {
       );
     }
     this.repository.resetHead(
-      this.assistantOptimisticId ?? messages.at(-1)?.id ?? null
+      this.assistantOptimisticId ?? messages2.at(-1)?.id ?? null
     );
     this.messages = this.repository.getMessages();
     this.notifySubscribers();
@@ -4387,19 +4619,19 @@ var ExternalStoreThreadRuntimeCore = class {
       this.repository.deleteMessage(this.assistantOptimisticId);
       this.assistantOptimisticId = null;
     }
-    let messages = this.repository.getMessages();
-    const previousMessage = messages[messages.length - 1];
-    if (previousMessage?.role === "user" && previousMessage.id === messages.at(-1)?.id) {
+    let messages2 = this.repository.getMessages();
+    const previousMessage = messages2[messages2.length - 1];
+    if (previousMessage?.role === "user" && previousMessage.id === messages2.at(-1)?.id) {
       this.repository.deleteMessage(previousMessage.id);
       if (!this.composer.text.trim()) {
         this.composer.setText(getThreadMessageText(previousMessage));
       }
-      messages = this.repository.getMessages();
+      messages2 = this.repository.getMessages();
     } else {
       this.notifySubscribers();
     }
     setTimeout(() => {
-      this.updateMessages(messages);
+      this.updateMessages(messages2);
     }, 0);
   }
   addToolResult(options) {
@@ -4423,9 +4655,9 @@ var ExternalStoreThreadRuntimeCore = class {
     this._subscriptions.add(callback);
     return () => this._subscriptions.delete(callback);
   }
-  updateMessages = (messages) => {
+  updateMessages = (messages2) => {
     this._store.setMessages?.(
-      messages.flatMap(getExternalStoreMessage).filter((m) => m != null)
+      messages2.flatMap(getExternalStoreMessage).filter((m) => m != null)
     );
   };
   import(repository) {
@@ -4483,21 +4715,21 @@ var ExternalStoreRuntimeCore = class extends BaseAssistantRuntimeCore {
 
 // src/runtimes/external-store/useExternalStoreRuntime.tsx
 var useExternalStoreRuntime = (store) => {
-  const [runtime] = useState13(() => new ExternalStoreRuntimeCore(store));
-  useEffect15(() => {
+  const [runtime] = useState12(() => new ExternalStoreRuntimeCore(store));
+  useEffect14(() => {
     runtime.thread.store = store;
   });
-  return useMemo8(() => new AssistantRuntime(runtime, ThreadRuntime), [runtime]);
+  return useMemo12(() => new AssistantRuntime(runtime, ThreadRuntime), [runtime]);
 };
 
 // src/runtimes/external-store/external-message-converter.tsx
-import { useMemo as useMemo9 } from "react";
-var joinExternalMessages = (messages) => {
+import { useMemo as useMemo13 } from "react";
+var joinExternalMessages = (messages2) => {
   const assistantMessage = {
     role: "assistant",
     content: []
   };
-  for (const output of messages) {
+  for (const output of messages2) {
     if (output.role === "tool") {
       const toolCallIdx = assistantMessage.content.findIndex(
         (c) => c.type === "tool-call" && c.toolCallId === output.toolCallId
@@ -4574,10 +4806,10 @@ var chunkExternalMessages = (callbackResults) => {
 };
 var useExternalMessageConverter = ({
   callback,
-  messages,
+  messages: messages2,
   isRunning
 }) => {
-  const state = useMemo9(
+  const state = useMemo13(
     () => ({
       callback,
       callbackCache: /* @__PURE__ */ new WeakMap(),
@@ -4586,9 +4818,9 @@ var useExternalMessageConverter = ({
     }),
     [callback]
   );
-  return useMemo9(() => {
+  return useMemo13(() => {
     const callbackResults = [];
-    for (const message of messages) {
+    for (const message of messages2) {
       let result = state.callbackCache.get(message);
       if (!result) {
         const output = state.callback(message);
@@ -4626,7 +4858,7 @@ var useExternalMessageConverter = ({
         return newMessage;
       }
     );
-  }, [state, messages, isRunning]);
+  }, [state, messages2, isRunning]);
 };
 var shallowArrayEqual = (a, b) => {
   if (a.length !== b.length) return false;
@@ -4637,20 +4869,20 @@ var shallowArrayEqual = (a, b) => {
 };
 
 // src/runtimes/dangerous-in-browser/useDangerousInBrowserRuntime.ts
-import { useState as useState14 } from "react";
+import { useState as useState13 } from "react";
 
 // src/runtimes/dangerous-in-browser/DangerousInBrowserAdapter.ts
 var DangerousInBrowserAdapter = class {
   constructor(options) {
     this.options = options;
   }
-  async *run({ messages, abortSignal, config }) {
+  async *run({ messages: messages2, abortSignal, config }) {
     const res = await getEdgeRuntimeStream({
       options: this.options,
       abortSignal,
       requestData: {
         system: config.system,
-        messages: toCoreMessages(messages),
+        messages: toCoreMessages(messages2),
         tools: config.tools ? toLanguageModelTools(config.tools) : [],
         ...config.callSettings,
         ...config.config
@@ -4668,7 +4900,7 @@ var useDangerousInBrowserRuntime = ({
   initialMessages,
   ...options
 }) => {
-  const [adapter] = useState14(() => new DangerousInBrowserAdapter(options));
+  const [adapter] = useState13(() => new DangerousInBrowserAdapter(options));
   return useLocalRuntime(adapter, { initialMessages });
 };
 
@@ -4721,12 +4953,15 @@ var SimpleImageAttachmentAdapter = class {
       id: state.file.name,
       type: "image",
       name: state.file.name,
-      file: state.file
+      contentType: state.file.type,
+      file: state.file,
+      status: { type: "requires-action", reason: "composer-send" }
     };
   }
   async send(attachment) {
     return {
       ...attachment,
+      status: { type: "complete" },
       content: [
         {
           type: "image",
@@ -4753,12 +4988,15 @@ var SimpleTextAttachmentAdapter = class {
       id: state.file.name,
       type: "document",
       name: state.file.name,
-      file: state.file
+      contentType: state.file.type,
+      file: state.file,
+      status: { type: "requires-action", reason: "composer-send" }
     };
   }
   async send(attachment) {
     return {
       ...attachment,
+      status: { type: "complete" },
       content: [
         {
           type: "text",
@@ -4841,7 +5079,14 @@ var CompositeAttachmentAdapter = class {
   async remove(attachment) {
     const adapters = this._adapters.slice();
     for (const adapter of adapters) {
-      if (fileMatchesAccept(attachment.file, adapter.accept)) {
+      if (fileMatchesAccept(
+        {
+          name: attachment.name,
+          type: attachment.contentType ?? "unknown/unknown"
+          // TODO remove after 0.6.0
+        },
+        adapter.accept
+      )) {
         return adapter.remove(attachment);
       }
     }
@@ -4854,7 +5099,7 @@ import {
   createContext as createContext7,
   useContext as useContext4
 } from "react";
-import { Fragment as Fragment3, jsx as jsx37 } from "react/jsx-runtime";
+import { Fragment as Fragment3, jsx as jsx36 } from "react/jsx-runtime";
 var ThreadConfigContext = createContext7({});
 var useThreadConfig = () => {
   return useContext4(ThreadConfigContext);
@@ -4864,14 +5109,14 @@ var ThreadConfigProvider = ({
   config
 }) => {
   const hasAssistant = !!useAssistantRuntime({ optional: true });
-  const configProvider = config && Object.keys(config ?? {}).length > 0 ? /* @__PURE__ */ jsx37(ThreadConfigContext.Provider, { value: config, children }) : /* @__PURE__ */ jsx37(Fragment3, { children });
+  const configProvider = config && Object.keys(config ?? {}).length > 0 ? /* @__PURE__ */ jsx36(ThreadConfigContext.Provider, { value: config, children }) : /* @__PURE__ */ jsx36(Fragment3, { children });
   if (!config?.runtime) return configProvider;
   if (hasAssistant) {
     throw new Error(
       "You provided a runtime to <Thread> while simulataneously using <AssistantRuntimeProvider>. This is not allowed."
     );
   }
-  return /* @__PURE__ */ jsx37(AssistantRuntimeProvider, { runtime: config.runtime, children: configProvider });
+  return /* @__PURE__ */ jsx36(AssistantRuntimeProvider, { runtime: config.runtime, children: configProvider });
 };
 ThreadConfigProvider.displayName = "ThreadConfigProvider";
 
@@ -4886,7 +5131,7 @@ import {
   ThumbsDownIcon,
   ThumbsUpIcon
 } from "lucide-react";
-import { Fragment as Fragment4, jsx as jsx38, jsxs as jsxs4 } from "react/jsx-runtime";
+import { Fragment as Fragment4, jsx as jsx37, jsxs as jsxs3 } from "react/jsx-runtime";
 var useAllowCopy = (ensureCapability = false) => {
   const { assistantMessage: { allowCopy = true } = {} } = useThreadConfig();
   const copySupported = useThread((t) => t.capabilities.unstable_copy);
@@ -4920,18 +5165,18 @@ var AssistantActionBar = () => {
   const allowFeedbackNegative = useAllowFeedbackNegative(true);
   if (!allowCopy && !allowReload && !allowSpeak && !allowFeedbackPositive && !allowFeedbackNegative)
     return null;
-  return /* @__PURE__ */ jsxs4(
+  return /* @__PURE__ */ jsxs3(
     AssistantActionBarRoot,
     {
       hideWhenRunning: true,
       autohide: "not-last",
       autohideFloat: "single-branch",
       children: [
-        allowSpeak && /* @__PURE__ */ jsx38(AssistantActionBarSpeechControl, {}),
-        allowCopy && /* @__PURE__ */ jsx38(AssistantActionBarCopy, {}),
-        allowReload && /* @__PURE__ */ jsx38(AssistantActionBarReload, {}),
-        allowFeedbackPositive && /* @__PURE__ */ jsx38(AssistantActionBarFeedbackPositive, {}),
-        allowFeedbackNegative && /* @__PURE__ */ jsx38(AssistantActionBarFeedbackNegative, {})
+        allowSpeak && /* @__PURE__ */ jsx37(AssistantActionBarSpeechControl, {}),
+        allowCopy && /* @__PURE__ */ jsx37(AssistantActionBarCopy, {}),
+        allowReload && /* @__PURE__ */ jsx37(AssistantActionBarReload, {}),
+        allowFeedbackPositive && /* @__PURE__ */ jsx37(AssistantActionBarFeedbackPositive, {}),
+        allowFeedbackNegative && /* @__PURE__ */ jsx37(AssistantActionBarFeedbackNegative, {})
       ]
     }
   );
@@ -4947,16 +5192,16 @@ var AssistantActionBarCopy = forwardRef22((props, ref) => {
       assistantMessage: { copy: { tooltip = "Copy" } = {} } = {}
     } = {}
   } = useThreadConfig();
-  return /* @__PURE__ */ jsx38(actionBar_exports.Copy, { asChild: true, children: /* @__PURE__ */ jsx38(TooltipIconButton, { tooltip, ...props, ref, children: props.children ?? /* @__PURE__ */ jsxs4(Fragment4, { children: [
-    /* @__PURE__ */ jsx38(message_exports.If, { copied: true, children: /* @__PURE__ */ jsx38(CheckIcon, {}) }),
-    /* @__PURE__ */ jsx38(message_exports.If, { copied: false, children: /* @__PURE__ */ jsx38(CopyIcon, {}) })
+  return /* @__PURE__ */ jsx37(actionBar_exports.Copy, { asChild: true, children: /* @__PURE__ */ jsx37(TooltipIconButton, { tooltip, ...props, ref, children: props.children ?? /* @__PURE__ */ jsxs3(Fragment4, { children: [
+    /* @__PURE__ */ jsx37(message_exports.If, { copied: true, children: /* @__PURE__ */ jsx37(CheckIcon, {}) }),
+    /* @__PURE__ */ jsx37(message_exports.If, { copied: false, children: /* @__PURE__ */ jsx37(CopyIcon, {}) })
   ] }) }) });
 });
 AssistantActionBarCopy.displayName = "AssistantActionBarCopy";
 var AssistantActionBarSpeechControl = () => {
-  return /* @__PURE__ */ jsxs4(Fragment4, { children: [
-    /* @__PURE__ */ jsx38(message_exports.If, { speaking: false, children: /* @__PURE__ */ jsx38(AssistantActionBarSpeak, {}) }),
-    /* @__PURE__ */ jsx38(message_exports.If, { speaking: true, children: /* @__PURE__ */ jsx38(AssistantActionBarStopSpeaking, {}) })
+  return /* @__PURE__ */ jsxs3(Fragment4, { children: [
+    /* @__PURE__ */ jsx37(message_exports.If, { speaking: false, children: /* @__PURE__ */ jsx37(AssistantActionBarSpeak, {}) }),
+    /* @__PURE__ */ jsx37(message_exports.If, { speaking: true, children: /* @__PURE__ */ jsx37(AssistantActionBarStopSpeaking, {}) })
   ] });
 };
 var AssistantActionBarSpeak = forwardRef22((props, ref) => {
@@ -4966,7 +5211,7 @@ var AssistantActionBarSpeak = forwardRef22((props, ref) => {
     } = {}
   } = useThreadConfig();
   const allowSpeak = useAllowSpeak();
-  return /* @__PURE__ */ jsx38(actionBar_exports.Speak, { disabled: !allowSpeak, asChild: true, children: /* @__PURE__ */ jsx38(TooltipIconButton, { tooltip, ...props, ref, children: props.children ?? /* @__PURE__ */ jsx38(AudioLinesIcon, {}) }) });
+  return /* @__PURE__ */ jsx37(actionBar_exports.Speak, { disabled: !allowSpeak, asChild: true, children: /* @__PURE__ */ jsx37(TooltipIconButton, { tooltip, ...props, ref, children: props.children ?? /* @__PURE__ */ jsx37(AudioLinesIcon, {}) }) });
 });
 AssistantActionBarSpeak.displayName = "AssistantActionBarSpeak";
 var AssistantActionBarStopSpeaking = forwardRef22((props, ref) => {
@@ -4978,7 +5223,7 @@ var AssistantActionBarStopSpeaking = forwardRef22((props, ref) => {
     } = {}
   } = useThreadConfig();
   const allowSpeak = useAllowSpeak();
-  return /* @__PURE__ */ jsx38(actionBar_exports.StopSpeaking, { disabled: !allowSpeak, asChild: true, children: /* @__PURE__ */ jsx38(TooltipIconButton, { tooltip: stopTooltip, ...props, ref, children: props.children ?? /* @__PURE__ */ jsx38(StopCircleIcon, {}) }) });
+  return /* @__PURE__ */ jsx37(actionBar_exports.StopSpeaking, { disabled: !allowSpeak, asChild: true, children: /* @__PURE__ */ jsx37(TooltipIconButton, { tooltip: stopTooltip, ...props, ref, children: props.children ?? /* @__PURE__ */ jsx37(StopCircleIcon, {}) }) });
 });
 AssistantActionBarStopSpeaking.displayName = "AssistantActionBarStopSpeaking";
 var AssistantActionBarReload = forwardRef22((props, ref) => {
@@ -4988,7 +5233,7 @@ var AssistantActionBarReload = forwardRef22((props, ref) => {
     } = {}
   } = useThreadConfig();
   const allowReload = useAllowReload();
-  return /* @__PURE__ */ jsx38(actionBar_exports.Reload, { disabled: !allowReload, asChild: true, children: /* @__PURE__ */ jsx38(TooltipIconButton, { tooltip, ...props, ref, children: props.children ?? /* @__PURE__ */ jsx38(RefreshCwIcon, {}) }) });
+  return /* @__PURE__ */ jsx37(actionBar_exports.Reload, { disabled: !allowReload, asChild: true, children: /* @__PURE__ */ jsx37(TooltipIconButton, { tooltip, ...props, ref, children: props.children ?? /* @__PURE__ */ jsx37(RefreshCwIcon, {}) }) });
 });
 AssistantActionBarReload.displayName = "AssistantActionBarReload";
 var AssistantActionBarFeedbackPositive = forwardRef22((props, ref) => {
@@ -5000,13 +5245,13 @@ var AssistantActionBarFeedbackPositive = forwardRef22((props, ref) => {
     } = {}
   } = useThreadConfig();
   const allowFeedbackPositive = useAllowFeedbackPositive();
-  return /* @__PURE__ */ jsx38(
+  return /* @__PURE__ */ jsx37(
     actionBar_exports.FeedbackPositive,
     {
       disabled: !allowFeedbackPositive,
       className: "aui-assistant-action-bar-feedback-positive",
       asChild: true,
-      children: /* @__PURE__ */ jsx38(TooltipIconButton, { tooltip, ...props, ref, children: props.children ?? /* @__PURE__ */ jsx38(ThumbsUpIcon, {}) })
+      children: /* @__PURE__ */ jsx37(TooltipIconButton, { tooltip, ...props, ref, children: props.children ?? /* @__PURE__ */ jsx37(ThumbsUpIcon, {}) })
     }
   );
 });
@@ -5020,13 +5265,13 @@ var AssistantActionBarFeedbackNegative = forwardRef22((props, ref) => {
     } = {}
   } = useThreadConfig();
   const allowFeedbackNegative = useAllowFeedbackNegative();
-  return /* @__PURE__ */ jsx38(
+  return /* @__PURE__ */ jsx37(
     actionBar_exports.FeedbackNegative,
     {
       disabled: !allowFeedbackNegative,
       className: "aui-assistant-action-bar-feedback-negative",
       asChild: true,
-      children: /* @__PURE__ */ jsx38(TooltipIconButton, { tooltip, ...props, ref, children: props.children ?? /* @__PURE__ */ jsx38(ThumbsDownIcon, {}) })
+      children: /* @__PURE__ */ jsx37(TooltipIconButton, { tooltip, ...props, ref, children: props.children ?? /* @__PURE__ */ jsx37(ThumbsDownIcon, {}) })
     }
   );
 });
@@ -5047,24 +5292,24 @@ var assistant_action_bar_default = Object.assign(
 );
 
 // src/ui/assistant-message.tsx
-import { forwardRef as forwardRef24, useMemo as useMemo10 } from "react";
+import { forwardRef as forwardRef24, useMemo as useMemo14 } from "react";
 
 // src/ui/branch-picker.tsx
 import { forwardRef as forwardRef23 } from "react";
 import { ChevronLeftIcon, ChevronRightIcon } from "lucide-react";
-import { jsx as jsx39, jsxs as jsxs5 } from "react/jsx-runtime";
+import { jsx as jsx38, jsxs as jsxs4 } from "react/jsx-runtime";
 var useAllowBranchPicker = (ensureCapability = false) => {
   const { branchPicker: { allowBranchPicker = true } = {} } = useThreadConfig();
   const branchPickerSupported = useThread((t) => t.capabilities.edit);
   return allowBranchPicker && (!ensureCapability || branchPickerSupported);
 };
 var BranchPicker = () => {
-  const allowBranchPicker = useAllowBranchPicker();
+  const allowBranchPicker = useAllowBranchPicker(true);
   if (!allowBranchPicker) return null;
-  return /* @__PURE__ */ jsxs5(BranchPickerRoot, { hideWhenSingleBranch: true, children: [
-    /* @__PURE__ */ jsx39(BranchPickerPrevious2, {}),
-    /* @__PURE__ */ jsx39(BranchPickerState, {}),
-    /* @__PURE__ */ jsx39(BranchPickerNext, {})
+  return /* @__PURE__ */ jsxs4(BranchPickerRoot, { hideWhenSingleBranch: true, children: [
+    /* @__PURE__ */ jsx38(BranchPickerPrevious2, {}),
+    /* @__PURE__ */ jsx38(BranchPickerState, {}),
+    /* @__PURE__ */ jsx38(BranchPickerNext, {})
   ] });
 };
 BranchPicker.displayName = "BranchPicker";
@@ -5079,17 +5324,17 @@ var BranchPickerPrevious2 = forwardRef23((props, ref) => {
     } = {}
   } = useThreadConfig();
   const allowBranchPicker = useAllowBranchPicker();
-  return /* @__PURE__ */ jsx39(branchPicker_exports.Previous, { disabled: !allowBranchPicker, asChild: true, children: /* @__PURE__ */ jsx39(TooltipIconButton, { tooltip, ...props, ref, children: props.children ?? /* @__PURE__ */ jsx39(ChevronLeftIcon, {}) }) });
+  return /* @__PURE__ */ jsx38(branchPicker_exports.Previous, { disabled: !allowBranchPicker, asChild: true, children: /* @__PURE__ */ jsx38(TooltipIconButton, { tooltip, ...props, ref, children: props.children ?? /* @__PURE__ */ jsx38(ChevronLeftIcon, {}) }) });
 });
 BranchPickerPrevious2.displayName = "BranchPickerPrevious";
 var BranchPickerStateWrapper = withDefaults("span", {
   className: "aui-branch-picker-state"
 });
 var BranchPickerState = forwardRef23((props, ref) => {
-  return /* @__PURE__ */ jsxs5(BranchPickerStateWrapper, { ...props, ref, children: [
-    /* @__PURE__ */ jsx39(branchPicker_exports.Number, {}),
+  return /* @__PURE__ */ jsxs4(BranchPickerStateWrapper, { ...props, ref, children: [
+    /* @__PURE__ */ jsx38(branchPicker_exports.Number, {}),
     " / ",
-    /* @__PURE__ */ jsx39(branchPicker_exports.Count, {})
+    /* @__PURE__ */ jsx38(branchPicker_exports.Count, {})
   ] });
 });
 BranchPickerState.displayName = "BranchPickerState";
@@ -5098,7 +5343,7 @@ var BranchPickerNext = forwardRef23((props, ref) => {
     strings: { branchPicker: { next: { tooltip = "Next" } = {} } = {} } = {}
   } = useThreadConfig();
   const allowBranchPicker = useAllowBranchPicker();
-  return /* @__PURE__ */ jsx39(branchPicker_exports.Next, { disabled: !allowBranchPicker, asChild: true, children: /* @__PURE__ */ jsx39(TooltipIconButton, { tooltip, ...props, ref, children: props.children ?? /* @__PURE__ */ jsx39(ChevronRightIcon, {}) }) });
+  return /* @__PURE__ */ jsx38(branchPicker_exports.Next, { disabled: !allowBranchPicker, asChild: true, children: /* @__PURE__ */ jsx38(TooltipIconButton, { tooltip, ...props, ref, children: props.children ?? /* @__PURE__ */ jsx38(ChevronRightIcon, {}) }) });
 });
 BranchPickerNext.displayName = "BranchPickerNext";
 var exports2 = {
@@ -5110,12 +5355,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 jsx40, jsxs as jsxs6 } from "react/jsx-runtime";
+import { jsx as jsx39, jsxs as jsxs5 } from "react/jsx-runtime";
 var Avatar = ({ src, alt, fallback }) => {
   if (src == null && fallback == null) return null;
-  return /* @__PURE__ */ jsxs6(AvatarRoot, { children: [
-    src != null && /* @__PURE__ */ jsx40(AvatarImage, { src, alt }),
-    fallback != null && /* @__PURE__ */ jsx40(AvatarFallback, { children: fallback })
+  return /* @__PURE__ */ jsxs5(AvatarRoot, { children: [
+    src != null && /* @__PURE__ */ jsx39(AvatarImage, { src, alt }),
+    fallback != null && /* @__PURE__ */ jsx39(AvatarFallback, { children: fallback })
   ] });
 };
 Avatar.displayName = "Avatar";
@@ -5134,10 +5379,10 @@ AvatarFallback.displayName = "AvatarFallback";
 
 // src/ui/content-part.tsx
 import classNames2 from "classnames";
-import { jsx as jsx41 } from "react/jsx-runtime";
+import { jsx as jsx40 } from "react/jsx-runtime";
 var Text = () => {
   const status = useSmoothStatus();
-  return /* @__PURE__ */ jsx41(
+  return /* @__PURE__ */ jsx40(
     contentPart_exports.Text,
     {
       className: classNames2(
@@ -5152,19 +5397,19 @@ var exports3 = { Text: withSmoothContextProvider(Text) };
 var content_part_default = exports3;
 
 // src/ui/assistant-message.tsx
-import { jsx as jsx42, jsxs as jsxs7 } from "react/jsx-runtime";
+import { jsx as jsx41, jsxs as jsxs6 } from "react/jsx-runtime";
 var AssistantMessage = () => {
-  return /* @__PURE__ */ jsxs7(AssistantMessageRoot, { children: [
-    /* @__PURE__ */ jsx42(AssistantMessageAvatar, {}),
-    /* @__PURE__ */ jsx42(AssistantMessageContent, {}),
-    /* @__PURE__ */ jsx42(branch_picker_default, {}),
-    /* @__PURE__ */ jsx42(assistant_action_bar_default, {})
+  return /* @__PURE__ */ jsxs6(AssistantMessageRoot, { children: [
+    /* @__PURE__ */ jsx41(AssistantMessageAvatar, {}),
+    /* @__PURE__ */ jsx41(AssistantMessageContent, {}),
+    /* @__PURE__ */ jsx41(branch_picker_default, {}),
+    /* @__PURE__ */ jsx41(assistant_action_bar_default, {})
   ] });
 };
 AssistantMessage.displayName = "AssistantMessage";
 var AssistantMessageAvatar = () => {
   const { assistantAvatar: avatar = { fallback: "A" } } = useThreadConfig();
-  return /* @__PURE__ */ jsx42(Avatar, { ...avatar });
+  return /* @__PURE__ */ jsx41(Avatar, { ...avatar });
 };
 var AssistantMessageRoot = withDefaults(message_exports.Root, {
   className: "aui-assistant-message-root"
@@ -5175,7 +5420,7 @@ var AssistantMessageContentWrapper = withDefaults("div", {
 });
 var AssistantMessageContent = forwardRef24(({ components: componentsProp, ...rest }, ref) => {
   const { tools, assistantMessage: { components = {} } = {} } = useThreadConfig();
-  const toolsComponents = useMemo10(
+  const toolsComponents = useMemo14(
     () => ({
       by_name: !tools ? void 0 : Object.fromEntries(
         tools.map((t) => [
@@ -5188,7 +5433,7 @@ var AssistantMessageContent = forwardRef24(({ components: componentsProp, ...res
     // eslint-disable-next-line react-hooks/exhaustive-deps
     [...tools ?? [], components.ToolFallback]
   );
-  return /* @__PURE__ */ jsx42(AssistantMessageContentWrapper, { ...rest, ref, children: /* @__PURE__ */ jsx42(
+  return /* @__PURE__ */ jsx41(AssistantMessageContentWrapper, { ...rest, ref, children: /* @__PURE__ */ jsx41(
     message_exports.Content,
     {
       components: {
@@ -5223,9 +5468,9 @@ import { forwardRef as forwardRef26 } from "react";
 import { PaperclipIcon, SendHorizontalIcon } from "lucide-react";
 
 // src/ui/base/CircleStopIcon.tsx
-import { jsx as jsx43 } from "react/jsx-runtime";
+import { jsx as jsx42 } from "react/jsx-runtime";
 var CircleStopIcon = () => {
-  return /* @__PURE__ */ jsx43(
+  return /* @__PURE__ */ jsx42(
     "svg",
     {
       xmlns: "http://www.w3.org/2000/svg",
@@ -5233,7 +5478,7 @@ var CircleStopIcon = () => {
       fill: "currentColor",
       width: "16",
       height: "16",
-      children: /* @__PURE__ */ jsx43("rect", { width: "10", height: "10", x: "3", y: "3", rx: "2" })
+      children: /* @__PURE__ */ jsx42("rect", { width: "10", height: "10", x: "3", y: "3", rx: "2" })
     }
   );
 };
@@ -5242,17 +5487,17 @@ CircleStopIcon.displayName = "CircleStopIcon";
 // src/ui/composer-attachment.tsx
 import { forwardRef as forwardRef25 } from "react";
 import { CircleXIcon } from "lucide-react";
-import { jsx as jsx44, jsxs as jsxs8 } from "react/jsx-runtime";
+import { jsx as jsx43, jsxs as jsxs7 } from "react/jsx-runtime";
 var ComposerAttachmentRoot = withDefaults("div", {
   className: "aui-composer-attachment-root"
 });
 ComposerAttachmentRoot.displayName = "ComposerAttachmentRoot";
 var ComposerAttachment2 = () => {
-  const attachment = useComposerAttachment((a) => a.attachment);
-  return /* @__PURE__ */ jsxs8(ComposerAttachmentRoot, { children: [
+  const attachment = useThreadComposerAttachment((a) => a.attachment);
+  return /* @__PURE__ */ jsxs7(ComposerAttachmentRoot, { children: [
     ".",
     attachment.name.split(".").pop(),
-    /* @__PURE__ */ jsx44(ComposerAttachmentRemove, {})
+    /* @__PURE__ */ jsx43(ComposerAttachmentRemove, {})
   ] });
 };
 ComposerAttachment2.displayName = "ComposerAttachment";
@@ -5262,12 +5507,11 @@ var ComposerAttachmentRemove = forwardRef25((props, ref) => {
       composer: { removeAttachment: { tooltip = "Remove file" } = {} } = {}
     } = {}
   } = useThreadConfig();
-  const composerStore = useThreadComposerStore();
-  const attachmentStore = useAttachmentStore();
+  const attachmentRuntime = useAttachmentRuntime();
   const handleRemoveAttachment = () => {
-    composerStore.getState().removeAttachment(attachmentStore.getState().attachment.id);
+    attachmentRuntime.remove();
   };
-  return /* @__PURE__ */ jsx44(
+  return /* @__PURE__ */ jsx43(
     TooltipIconButton,
     {
       tooltip,
@@ -5276,7 +5520,7 @@ var ComposerAttachmentRemove = forwardRef25((props, ref) => {
       ...props,
       onClick: handleRemoveAttachment,
       ref,
-      children: props.children ?? /* @__PURE__ */ jsx44(CircleXIcon, {})
+      children: props.children ?? /* @__PURE__ */ jsx43(CircleXIcon, {})
     }
   );
 });
@@ -5291,7 +5535,7 @@ var composer_attachment_default = Object.assign(
 );
 
 // src/ui/composer.tsx
-import { Fragment as Fragment5, jsx as jsx45, jsxs as jsxs9 } from "react/jsx-runtime";
+import { Fragment as Fragment5, jsx as jsx44, jsxs as jsxs8 } from "react/jsx-runtime";
 var useAllowAttachments = (ensureCapability = false) => {
   const { composer: { allowAttachments = true } = {} } = useThreadConfig();
   const attachmentsSupported = useThread((t) => t.capabilities.attachments);
@@ -5299,11 +5543,11 @@ var useAllowAttachments = (ensureCapability = false) => {
 };
 var Composer = () => {
   const allowAttachments = useAllowAttachments(true);
-  return /* @__PURE__ */ jsxs9(ComposerRoot, { children: [
-    allowAttachments && /* @__PURE__ */ jsx45(ComposerAttachments, {}),
-    allowAttachments && /* @__PURE__ */ jsx45(ComposerAddAttachment, {}),
-    /* @__PURE__ */ jsx45(ComposerInput, { autoFocus: true }),
-    /* @__PURE__ */ jsx45(ComposerAction, {})
+  return /* @__PURE__ */ jsxs8(ComposerRoot, { children: [
+    allowAttachments && /* @__PURE__ */ jsx44(ComposerAttachments, {}),
+    allowAttachments && /* @__PURE__ */ jsx44(ComposerAddAttachment, {}),
+    /* @__PURE__ */ jsx44(ComposerInput, { autoFocus: true }),
+    /* @__PURE__ */ jsx44(ComposerAction, {})
   ] });
 };
 Composer.displayName = "Composer";
@@ -5323,7 +5567,7 @@ var ComposerInput = forwardRef26(
         composer: { input: { placeholder = "Write a message..." } = {} } = {}
       } = {}
     } = useThreadConfig();
-    return /* @__PURE__ */ jsx45(ComposerInputStyled, { placeholder, ...props, ref });
+    return /* @__PURE__ */ jsx44(ComposerInputStyled, { placeholder, ...props, ref });
   }
 );
 ComposerInput.displayName = "ComposerInput";
@@ -5331,7 +5575,7 @@ var ComposerAttachmentsContainer = withDefaults("div", {
   className: "aui-composer-attachments"
 });
 var ComposerAttachments = ({ components }) => {
-  return /* @__PURE__ */ jsx45(ComposerAttachmentsContainer, { children: /* @__PURE__ */ jsx45(
+  return /* @__PURE__ */ jsx44(ComposerAttachmentsContainer, { children: /* @__PURE__ */ jsx44(
     composer_exports.Attachments,
     {
       components: {
@@ -5352,14 +5596,14 @@ var ComposerAddAttachment = forwardRef26((props, ref) => {
     } = {}
   } = useThreadConfig();
   const allowAttachments = useAllowAttachments();
-  return /* @__PURE__ */ jsx45(composer_exports.AddAttachment, { disabled: !allowAttachments, asChild: true, children: /* @__PURE__ */ jsx45(
+  return /* @__PURE__ */ jsx44(composer_exports.AddAttachment, { disabled: !allowAttachments, asChild: true, children: /* @__PURE__ */ jsx44(
     ComposerAttachButton,
     {
       tooltip,
       variant: "ghost",
       ...props,
       ref,
-      children: props.children ?? /* @__PURE__ */ jsx45(PaperclipIcon, {})
+      children: props.children ?? /* @__PURE__ */ jsx44(PaperclipIcon, {})
     }
   ) });
 });
@@ -5370,10 +5614,10 @@ var useAllowCancel = () => {
 };
 var ComposerAction = () => {
   const allowCancel = useAllowCancel();
-  if (!allowCancel) return /* @__PURE__ */ jsx45(ComposerSend, {});
-  return /* @__PURE__ */ jsxs9(Fragment5, { children: [
-    /* @__PURE__ */ jsx45(thread_exports.If, { running: false, children: /* @__PURE__ */ jsx45(ComposerSend, {}) }),
-    /* @__PURE__ */ jsx45(thread_exports.If, { running: true, children: /* @__PURE__ */ jsx45(ComposerCancel, {}) })
+  if (!allowCancel) return /* @__PURE__ */ jsx44(ComposerSend, {});
+  return /* @__PURE__ */ jsxs8(Fragment5, { children: [
+    /* @__PURE__ */ jsx44(thread_exports.If, { running: false, children: /* @__PURE__ */ jsx44(ComposerSend, {}) }),
+    /* @__PURE__ */ jsx44(thread_exports.If, { running: true, children: /* @__PURE__ */ jsx44(ComposerCancel, {}) })
   ] });
 };
 ComposerAction.displayName = "ComposerAction";
@@ -5385,7 +5629,7 @@ var ComposerSend = forwardRef26((props, ref) => {
   const {
     strings: { composer: { send: { tooltip = "Send" } = {} } = {} } = {}
   } = useThreadConfig();
-  return /* @__PURE__ */ jsx45(composer_exports.Send, { asChild: true, children: /* @__PURE__ */ jsx45(ComposerSendButton, { tooltip, ...props, ref, children: props.children ?? /* @__PURE__ */ jsx45(SendHorizontalIcon, {}) }) });
+  return /* @__PURE__ */ jsx44(composer_exports.Send, { asChild: true, children: /* @__PURE__ */ jsx44(ComposerSendButton, { tooltip, ...props, ref, children: props.children ?? /* @__PURE__ */ jsx44(SendHorizontalIcon, {}) }) });
 });
 ComposerSend.displayName = "ComposerSend";
 var ComposerCancelButton = withDefaults(TooltipIconButton, {
@@ -5396,7 +5640,7 @@ var ComposerCancel = forwardRef26((props, ref) => {
   const {
     strings: { composer: { cancel: { tooltip = "Cancel" } = {} } = {} } = {}
   } = useThreadConfig();
-  return /* @__PURE__ */ jsx45(composer_exports.Cancel, { asChild: true, children: /* @__PURE__ */ jsx45(ComposerCancelButton, { tooltip, ...props, ref, children: props.children ?? /* @__PURE__ */ jsx45(CircleStopIcon, {}) }) });
+  return /* @__PURE__ */ jsx44(composer_exports.Cancel, { asChild: true, children: /* @__PURE__ */ jsx44(ComposerCancelButton, { tooltip, ...props, ref, children: props.children ?? /* @__PURE__ */ jsx44(CircleStopIcon, {}) }) });
 });
 ComposerCancel.displayName = "ComposerCancel";
 var exports6 = {
@@ -5412,14 +5656,14 @@ var composer_default = Object.assign(Composer, exports6);
 
 // src/ui/thread-welcome.tsx
 import { forwardRef as forwardRef27 } from "react";
-import { jsx as jsx46, jsxs as jsxs10 } from "react/jsx-runtime";
+import { jsx as jsx45, jsxs as jsxs9 } from "react/jsx-runtime";
 var ThreadWelcome = () => {
-  return /* @__PURE__ */ jsxs10(ThreadWelcomeRoot, { children: [
-    /* @__PURE__ */ jsxs10(ThreadWelcomeCenter, { children: [
-      /* @__PURE__ */ jsx46(ThreadWelcomeAvatar, {}),
-      /* @__PURE__ */ jsx46(ThreadWelcomeMessage, {})
+  return /* @__PURE__ */ jsxs9(ThreadWelcomeRoot, { children: [
+    /* @__PURE__ */ jsxs9(ThreadWelcomeCenter, { children: [
+      /* @__PURE__ */ jsx45(ThreadWelcomeAvatar, {}),
+      /* @__PURE__ */ jsx45(ThreadWelcomeMessage, {})
     ] }),
-    /* @__PURE__ */ jsx46(ThreadWelcomeSuggestions, {})
+    /* @__PURE__ */ jsx45(ThreadWelcomeSuggestions, {})
   ] });
 };
 ThreadWelcome.displayName = "ThreadWelcome";
@@ -5431,20 +5675,20 @@ var ThreadWelcomeCenter = withDefaults("div", {
 });
 var ThreadWelcomeRoot = forwardRef27(
   (props, ref) => {
-    return /* @__PURE__ */ jsx46(thread_exports.Empty, { children: /* @__PURE__ */ jsx46(ThreadWelcomeRootStyled, { ...props, ref }) });
+    return /* @__PURE__ */ jsx45(thread_exports.Empty, { children: /* @__PURE__ */ jsx45(ThreadWelcomeRootStyled, { ...props, ref }) });
   }
 );
 ThreadWelcomeRoot.displayName = "ThreadWelcomeRoot";
 var ThreadWelcomeAvatar = () => {
   const { assistantAvatar: avatar = { fallback: "A" } } = useThreadConfig();
-  return /* @__PURE__ */ jsx46(Avatar, { ...avatar });
+  return /* @__PURE__ */ jsx45(Avatar, { ...avatar });
 };
 var ThreadWelcomeMessageStyled = withDefaults("p", {
   className: "aui-thread-welcome-message"
 });
 var ThreadWelcomeMessage = forwardRef27(({ message: messageProp, ...rest }, ref) => {
   const { welcome: { message = "How can I help you today?" } = {} } = useThreadConfig();
-  return /* @__PURE__ */ jsx46(ThreadWelcomeMessageStyled, { ...rest, ref, children: messageProp ?? message });
+  return /* @__PURE__ */ jsx45(ThreadWelcomeMessageStyled, { ...rest, ref, children: messageProp ?? message });
 });
 ThreadWelcomeMessage.displayName = "ThreadWelcomeMessage";
 var ThreadWelcomeSuggestionContainer = withDefaults("div", {
@@ -5456,21 +5700,21 @@ var ThreadWelcomeSuggestionStyled = withDefaults(thread_exports.Suggestion, {
 var ThreadWelcomeSuggestion = ({
   suggestion: { text, prompt }
 }) => {
-  return /* @__PURE__ */ jsx46(
+  return /* @__PURE__ */ jsx45(
     ThreadWelcomeSuggestionStyled,
     {
       prompt,
       method: "replace",
       autoSend: true,
-      children: /* @__PURE__ */ jsx46("span", { className: "aui-thread-welcome-suggestion-text", children: text ?? prompt })
+      children: /* @__PURE__ */ jsx45("span", { className: "aui-thread-welcome-suggestion-text", children: text ?? prompt })
     }
   );
 };
 var ThreadWelcomeSuggestions = () => {
   const { welcome: { suggestions } = {} } = useThreadConfig();
-  return /* @__PURE__ */ jsx46(ThreadWelcomeSuggestionContainer, { children: suggestions?.map((suggestion, idx) => {
+  return /* @__PURE__ */ jsx45(ThreadWelcomeSuggestionContainer, { children: suggestions?.map((suggestion, idx) => {
     const key = `${suggestion.prompt}-${idx}`;
-    return /* @__PURE__ */ jsx46(ThreadWelcomeSuggestion, { suggestion }, key);
+    return /* @__PURE__ */ jsx45(ThreadWelcomeSuggestion, { suggestion }, key);
   }) });
 };
 ThreadWelcomeSuggestions.displayName = "ThreadWelcomeSuggestions";
@@ -5490,7 +5734,7 @@ import { forwardRef as forwardRef29 } from "react";
 // src/ui/user-action-bar.tsx
 import { forwardRef as forwardRef28 } from "react";
 import { PencilIcon } from "lucide-react";
-import { jsx as jsx47 } from "react/jsx-runtime";
+import { jsx as jsx46 } from "react/jsx-runtime";
 var useAllowEdit = (ensureCapability = false) => {
   const { userMessage: { allowEdit = true } = {} } = useThreadConfig();
   const editSupported = useThread((t) => t.capabilities.edit);
@@ -5499,7 +5743,7 @@ var useAllowEdit = (ensureCapability = false) => {
 var UserActionBar = () => {
   const allowEdit = useAllowEdit(true);
   if (!allowEdit) return null;
-  return /* @__PURE__ */ jsx47(UserActionBarRoot, { hideWhenRunning: true, autohide: "not-last", children: /* @__PURE__ */ jsx47(UserActionBarEdit, {}) });
+  return /* @__PURE__ */ jsx46(UserActionBarRoot, { hideWhenRunning: true, autohide: "not-last", children: /* @__PURE__ */ jsx46(UserActionBarEdit, {}) });
 };
 UserActionBar.displayName = "UserActionBar";
 var UserActionBarRoot = withDefaults(actionBar_exports.Root, {
@@ -5511,7 +5755,7 @@ var UserActionBarEdit = forwardRef28((props, ref) => {
     strings: { userMessage: { edit: { tooltip = "Edit" } = {} } = {} } = {}
   } = useThreadConfig();
   const allowEdit = useAllowEdit();
-  return /* @__PURE__ */ jsx47(actionBar_exports.Edit, { disabled: !allowEdit, asChild: true, children: /* @__PURE__ */ jsx47(TooltipIconButton, { tooltip, ...props, ref, children: props.children ?? /* @__PURE__ */ jsx47(PencilIcon, {}) }) });
+  return /* @__PURE__ */ jsx46(actionBar_exports.Edit, { disabled: !allowEdit, asChild: true, children: /* @__PURE__ */ jsx46(TooltipIconButton, { tooltip, ...props, ref, children: props.children ?? /* @__PURE__ */ jsx46(PencilIcon, {}) }) });
 });
 UserActionBarEdit.displayName = "UserActionBarEdit";
 var exports8 = {
@@ -5521,14 +5765,14 @@ var exports8 = {
 var user_action_bar_default = Object.assign(UserActionBar, exports8);
 
 // src/ui/user-message-attachment.tsx
-import { jsxs as jsxs11 } from "react/jsx-runtime";
+import { jsxs as jsxs10 } from "react/jsx-runtime";
 var UserMessageAttachmentRoot = withDefaults("div", {
   className: "aui-user-message-attachment-root"
 });
 UserMessageAttachmentRoot.displayName = "UserMessageAttachmentRoot";
 var UserMessageAttachment = () => {
   const attachment = useAttachment((a) => a.attachment);
-  return /* @__PURE__ */ jsxs11(UserMessageAttachmentRoot, { children: [
+  return /* @__PURE__ */ jsxs10(UserMessageAttachmentRoot, { children: [
     ".",
     attachment.name.split(".").pop()
   ] });
@@ -5543,13 +5787,13 @@ var user_message_attachment_default = Object.assign(
 );
 
 // src/ui/user-message.tsx
-import { jsx as jsx48, jsxs as jsxs12 } from "react/jsx-runtime";
+import { jsx as jsx47, jsxs as jsxs11 } from "react/jsx-runtime";
 var UserMessage = () => {
-  return /* @__PURE__ */ jsxs12(UserMessageRoot, { children: [
-    /* @__PURE__ */ jsx48(UserMessageAttachments, {}),
-    /* @__PURE__ */ jsx48(user_action_bar_default, {}),
-    /* @__PURE__ */ jsx48(UserMessageContent, {}),
-    /* @__PURE__ */ jsx48(branch_picker_default, {})
+  return /* @__PURE__ */ jsxs11(UserMessageRoot, { children: [
+    /* @__PURE__ */ jsx47(UserMessageAttachments, {}),
+    /* @__PURE__ */ jsx47(user_action_bar_default, {}),
+    /* @__PURE__ */ jsx47(UserMessageContent, {}),
+    /* @__PURE__ */ jsx47(branch_picker_default, {})
   ] });
 };
 UserMessage.displayName = "UserMessage";
@@ -5562,7 +5806,7 @@ var UserMessageContentWrapper = withDefaults("div", {
 });
 var UserMessageContent = forwardRef29(
   ({ components, ...props }, ref) => {
-    return /* @__PURE__ */ jsx48(UserMessageContentWrapper, { ...props, ref, children: /* @__PURE__ */ jsx48(
+    return /* @__PURE__ */ jsx47(UserMessageContentWrapper, { ...props, ref, children: /* @__PURE__ */ jsx47(
       message_exports.Content,
       {
         components: {
@@ -5580,7 +5824,7 @@ var UserMessageAttachmentsContainer = withDefaults("div", {
 var UserMessageAttachments = ({
   components
 }) => {
-  return /* @__PURE__ */ jsx48(message_exports.If, { hasAttachments: true, children: /* @__PURE__ */ jsx48(UserMessageAttachmentsContainer, { children: /* @__PURE__ */ jsx48(
+  return /* @__PURE__ */ jsx47(message_exports.If, { hasAttachments: true, children: /* @__PURE__ */ jsx47(UserMessageAttachmentsContainer, { children: /* @__PURE__ */ jsx47(
     message_exports.Attachments,
     {
       components: {
@@ -5599,13 +5843,13 @@ var user_message_default = Object.assign(UserMessage, exports10);
 
 // src/ui/edit-composer.tsx
 import { forwardRef as forwardRef30 } from "react";
-import { jsx as jsx49, jsxs as jsxs13 } from "react/jsx-runtime";
+import { jsx as jsx48, jsxs as jsxs12 } from "react/jsx-runtime";
 var EditComposer = () => {
-  return /* @__PURE__ */ jsxs13(EditComposerRoot, { children: [
-    /* @__PURE__ */ jsx49(EditComposerInput, {}),
-    /* @__PURE__ */ jsxs13(EditComposerFooter, { children: [
-      /* @__PURE__ */ jsx49(EditComposerCancel, {}),
-      /* @__PURE__ */ jsx49(EditComposerSend, {})
+  return /* @__PURE__ */ jsxs12(EditComposerRoot, { children: [
+    /* @__PURE__ */ jsx48(EditComposerInput, {}),
+    /* @__PURE__ */ jsxs12(EditComposerFooter, { children: [
+      /* @__PURE__ */ jsx48(EditComposerCancel, {}),
+      /* @__PURE__ */ jsx48(EditComposerSend, {})
     ] })
   ] });
 };
@@ -5629,7 +5873,7 @@ var EditComposerCancel = forwardRef30(
         editComposer: { cancel: { label = "Cancel" } = {} } = {}
       } = {}
     } = useThreadConfig();
-    return /* @__PURE__ */ jsx49(composer_exports.Cancel, { asChild: true, children: /* @__PURE__ */ jsx49(Button, { variant: "ghost", ...props, ref, children: props.children ?? label }) });
+    return /* @__PURE__ */ jsx48(composer_exports.Cancel, { asChild: true, children: /* @__PURE__ */ jsx48(Button, { variant: "ghost", ...props, ref, children: props.children ?? label }) });
   }
 );
 EditComposerCancel.displayName = "EditComposerCancel";
@@ -5638,7 +5882,7 @@ var EditComposerSend = forwardRef30(
     const {
       strings: { editComposer: { send: { label = "Send" } = {} } = {} } = {}
     } = useThreadConfig();
-    return /* @__PURE__ */ jsx49(composer_exports.Send, { asChild: true, children: /* @__PURE__ */ jsx49(Button, { ...props, ref, children: props.children ?? label }) });
+    return /* @__PURE__ */ jsx48(composer_exports.Send, { asChild: true, children: /* @__PURE__ */ jsx48(Button, { ...props, ref, children: props.children ?? label }) });
   }
 );
 EditComposerSend.displayName = "EditComposerSend";
@@ -5652,14 +5896,14 @@ var exports11 = {
 var edit_composer_default = Object.assign(EditComposer, exports11);
 
 // src/ui/thread.tsx
-import { jsx as jsx50, jsxs as jsxs14 } from "react/jsx-runtime";
+import { jsx as jsx49, jsxs as jsxs13 } from "react/jsx-runtime";
 var Thread = (config) => {
-  return /* @__PURE__ */ jsx50(ThreadRoot, { config, children: /* @__PURE__ */ jsxs14(ThreadViewport, { children: [
-    /* @__PURE__ */ jsx50(thread_welcome_default, {}),
-    /* @__PURE__ */ jsx50(ThreadMessages, {}),
-    /* @__PURE__ */ jsxs14(ThreadViewportFooter, { children: [
-      /* @__PURE__ */ jsx50(ThreadScrollToBottom, {}),
-      /* @__PURE__ */ jsx50(composer_default, {})
+  return /* @__PURE__ */ jsx49(ThreadRoot, { config, children: /* @__PURE__ */ jsxs13(ThreadViewport, { children: [
+    /* @__PURE__ */ jsx49(thread_welcome_default, {}),
+    /* @__PURE__ */ jsx49(ThreadMessages, {}),
+    /* @__PURE__ */ jsxs13(ThreadViewportFooter, { children: [
+      /* @__PURE__ */ jsx49(ThreadScrollToBottom, {}),
+      /* @__PURE__ */ jsx49(composer_default, {})
     ] })
   ] }) });
 };
@@ -5668,7 +5912,7 @@ var ThreadRootStyled = withDefaults(thread_exports.Root, {
 });
 var ThreadRoot = forwardRef31(
   ({ config, ...props }, ref) => {
-    return /* @__PURE__ */ jsx50(ThreadConfigProvider, { config, children: /* @__PURE__ */ jsx50(ThreadRootStyled, { ...props, ref }) });
+    return /* @__PURE__ */ jsx49(ThreadConfigProvider, { config, children: /* @__PURE__ */ jsx49(ThreadRootStyled, { ...props, ref }) });
   }
 );
 ThreadRoot.displayName = "ThreadRoot";
@@ -5682,7 +5926,7 @@ var ThreadViewportFooter = withDefaults("div", {
 ThreadViewportFooter.displayName = "ThreadViewportFooter";
 var SystemMessage = () => null;
 var ThreadMessages = ({ components, ...rest }) => {
-  return /* @__PURE__ */ jsx50(
+  return /* @__PURE__ */ jsx49(
     thread_exports.Messages,
     {
       components: {
@@ -5706,7 +5950,7 @@ var ThreadScrollToBottom = forwardRef31((props, ref) => {
       thread: { scrollToBottom: { tooltip = "Scroll to bottom" } = {} } = {}
     } = {}
   } = useThreadConfig();
-  return /* @__PURE__ */ jsx50(thread_exports.ScrollToBottom, { asChild: true, children: /* @__PURE__ */ jsx50(ThreadScrollToBottomIconButton, { tooltip, ...props, ref, children: props.children ?? /* @__PURE__ */ jsx50(ArrowDownIcon, {}) }) });
+  return /* @__PURE__ */ jsx49(thread_exports.ScrollToBottom, { asChild: true, children: /* @__PURE__ */ jsx49(ThreadScrollToBottomIconButton, { tooltip, ...props, ref, children: props.children ?? /* @__PURE__ */ jsx49(ArrowDownIcon, {}) }) });
 });
 ThreadScrollToBottom.displayName = "ThreadScrollToBottom";
 var exports12 = {
@@ -5719,20 +5963,20 @@ var exports12 = {
 var thread_default = Object.assign(Thread, exports12);
 
 // src/ui/assistant-modal.tsx
-import { Fragment as Fragment6, jsx as jsx51, jsxs as jsxs15 } from "react/jsx-runtime";
+import { Fragment as Fragment6, jsx as jsx50, jsxs as jsxs14 } from "react/jsx-runtime";
 var AssistantModal = (config) => {
-  return /* @__PURE__ */ jsxs15(AssistantModalRoot, { config, children: [
-    /* @__PURE__ */ jsx51(AssistantModalTrigger, {}),
-    /* @__PURE__ */ jsx51(AssistantModalContent, { children: /* @__PURE__ */ jsx51(thread_default, {}) })
+  return /* @__PURE__ */ jsxs14(AssistantModalRoot, { config, children: [
+    /* @__PURE__ */ jsx50(AssistantModalTrigger, {}),
+    /* @__PURE__ */ jsx50(AssistantModalContent, { children: /* @__PURE__ */ jsx50(thread_default, {}) })
   ] });
 };
 AssistantModal.displayName = "AssistantModal";
 var AssistantModalRoot = ({ config, ...props }) => {
-  return /* @__PURE__ */ jsx51(ThreadConfigProvider, { config, children: /* @__PURE__ */ jsx51(assistantModal_exports.Root, { ...props }) });
+  return /* @__PURE__ */ jsx50(ThreadConfigProvider, { config, children: /* @__PURE__ */ jsx50(assistantModal_exports.Root, { ...props }) });
 };
 AssistantModalRoot.displayName = "AssistantModalRoot";
 var AssistantModalTrigger = forwardRef32((props, ref) => {
-  return /* @__PURE__ */ jsx51(AssistantModalAnchor, { children: /* @__PURE__ */ jsx51(assistantModal_exports.Trigger, { asChild: true, children: /* @__PURE__ */ jsx51(AssistantModalButton, { ...props, ref }) }) });
+  return /* @__PURE__ */ jsx50(AssistantModalAnchor, { children: /* @__PURE__ */ jsx50(assistantModal_exports.Trigger, { asChild: true, children: /* @__PURE__ */ jsx50(AssistantModalButton, { ...props, ref }) }) });
 });
 AssistantModalTrigger.displayName = "AssistantModalTrigger";
 var AssistantModalAnchor = withDefaults(assistantModal_exports.Anchor, {
@@ -5757,7 +6001,7 @@ var AssistantModalButton = forwardRef32(({ "data-state": state, ...rest }, ref)
     } = {}
   } = useThreadConfig();
   const tooltip = state === "open" ? openTooltip : closedTooltip;
-  return /* @__PURE__ */ jsx51(
+  return /* @__PURE__ */ jsx50(
     ModalButtonStyled,
     {
       side: "left",
@@ -5765,15 +6009,15 @@ var AssistantModalButton = forwardRef32(({ "data-state": state, ...rest }, ref)
       "data-state": state,
       ...rest,
       ref,
-      children: rest.children ?? /* @__PURE__ */ jsxs15(Fragment6, { children: [
-        /* @__PURE__ */ jsx51(
+      children: rest.children ?? /* @__PURE__ */ jsxs14(Fragment6, { children: [
+        /* @__PURE__ */ jsx50(
           BotIcon,
           {
             "data-state": state,
             className: "aui-modal-button-closed-icon"
           }
         ),
-        /* @__PURE__ */ jsx51(
+        /* @__PURE__ */ jsx50(
           ChevronDownIcon,
           {
             "data-state": state,
@@ -5866,12 +6110,14 @@ export {
   useComposerCancel,
   useComposerContext,
   useComposerIf,
+  useComposerRuntime,
   useComposerSend,
   useComposerStore,
   useContentPart,
   useContentPartContext,
   useContentPartDisplay,
   useContentPartImage,
+  useContentPartRuntime,
   useContentPartStore,
   useContentPartText,
   useDangerousInBrowserRuntime,