@assistant-ui/react 0.2.4 → 0.3.1

package/dist/index.js

@@ -210,11 +210,14 @@ var makeBaseComposer = (set) => ({
 });
 
 // src/context/stores/Composer.ts
-var makeComposerStore = (useThread, useThreadMessages, useThreadActions) => {
+var makeComposerStore = (useThreadMessages, useThreadActions) => {
   const focusListeners = /* @__PURE__ */ new Set();
   return (0, import_zustand3.create)()((set, get, store) => {
     return {
       ...makeBaseComposer(set, get, store),
+      get canCancel() {
+        return useThreadActions.getState().capabilities.cancel;
+      },
       isEditing: true,
       send: () => {
         const { setValue, value } = get();
@@ -226,10 +229,7 @@ var makeComposerStore = (useThread, useThreadMessages, useThreadActions) => {
         });
       },
       cancel: () => {
-        const thread = useThread.getState();
-        if (!thread.isRunning) return false;
         useThreadActions.getState().cancelRun();
-        return true;
       },
       focus: () => {
         for (const listener of focusListeners) {
@@ -279,6 +279,9 @@ var import_zustand6 = require("zustand");
 var makeThreadActionStore = (runtimeRef) => {
   return (0, import_zustand6.create)(
     () => Object.freeze({
+      get capabilities() {
+        return runtimeRef.current.capabilities;
+      },
       getBranches: (messageId) => runtimeRef.current.getBranches(messageId),
       switchToBranch: (branchId) => runtimeRef.current.switchToBranch(branchId),
       startRun: (parentId) => runtimeRef.current.startRun(parentId),
@@ -310,11 +313,7 @@ var ThreadProvider = ({
     const useThreadMessages = makeThreadMessagesStore(runtimeRef);
     const useThreadActions = makeThreadActionStore(runtimeRef);
     const useViewport = makeThreadViewportStore();
-    const useComposer = makeComposerStore(
-      useThread,
-      useThreadMessages,
-      useThreadActions
-    );
+    const useComposer = makeComposerStore(useThreadMessages, useThreadActions);
     return {
       useThread,
       useThreadMessages,
@@ -584,7 +583,7 @@ var useActionBarCopy = ({
   const hasCopyableContent = useCombinedStore(
     [useMessage, useEditComposer],
     (m, c) => {
-      return !c.isEditing && m.message.content.some((c2) => c2.type === "text");
+      return !c.isEditing && m.message.content.some((c2) => c2.type === "text" && c2.text.length > 0);
     }
   );
   const callback = (0, import_react16.useCallback)(() => {
@@ -687,7 +686,7 @@ var useBranchPickerPrevious = () => {
 var import_react21 = require("react");
 var useComposerCancel = () => {
   const { useComposer } = useComposerContext();
-  const disabled = useComposer((c) => !c.isEditing);
+  const disabled = useComposer((c) => !c.canCancel);
   const callback = (0, import_react21.useCallback)(() => {
     const { cancel } = useComposer.getState();
     cancel();
@@ -757,7 +756,7 @@ var useContentPartText = () => {
       throw new Error(
         "ContentPartText can only be used inside text content parts."
       );
-    return c.part.text;
+    return c;
   });
   return text;
 };
@@ -819,15 +818,18 @@ var useThreadSuggestion = ({
   autoSend
 }) => {
   const { useThread, useComposer } = useThreadContext();
+  const append = useAppendMessage();
   const disabled = useThread((t) => t.isRunning);
   const callback = (0, import_react24.useCallback)(() => {
     const thread = useThread.getState();
     const composer = useComposer.getState();
-    composer.setValue(prompt);
     if (autoSend && !thread.isRunning) {
-      composer.send();
+      append(prompt);
+      composer.setValue("");
+    } else {
+      composer.setValue(prompt);
     }
-  }, [useThread, useComposer, prompt, autoSend]);
+  }, [useThread, useComposer, autoSend, append, prompt]);
   if (disabled) return null;
   return callback;
 };
@@ -1107,28 +1109,57 @@ __export(message_exports, {
 });
 
 // src/primitives/message/MessageRoot.tsx
-var import_primitive4 = require("@radix-ui/primitive");
 var import_react_primitive3 = require("@radix-ui/react-primitive");
+var import_react33 = require("react");
+
+// src/utils/hooks/useManagedRef.ts
 var import_react32 = require("react");
+var useManagedRef = (callback) => {
+  const cleanupRef = (0, import_react32.useRef)();
+  const ref = (0, import_react32.useCallback)(
+    (el) => {
+      if (cleanupRef.current) {
+        cleanupRef.current();
+      }
+      if (el) {
+        cleanupRef.current = callback(el);
+      }
+    },
+    [callback]
+  );
+  return ref;
+};
+
+// src/primitives/message/MessageRoot.tsx
+var import_react_compose_refs = require("@radix-ui/react-compose-refs");
 var import_jsx_runtime12 = require("react/jsx-runtime");
-var MessagePrimitiveRoot = (0, import_react32.forwardRef)(({ onMouseEnter, onMouseLeave, ...rest }, ref) => {
+var useIsHoveringRef = () => {
   const { useMessageUtils } = useMessageContext();
-  const setIsHovering = useMessageUtils((s) => s.setIsHovering);
-  const handleMouseEnter = () => {
-    setIsHovering(true);
-  };
-  const handleMouseLeave = () => {
-    setIsHovering(false);
-  };
-  return /* @__PURE__ */ (0, import_jsx_runtime12.jsx)(
-    import_react_primitive3.Primitive.div,
-    {
-      ...rest,
-      ref,
-      onMouseEnter: (0, import_primitive4.composeEventHandlers)(onMouseEnter, handleMouseEnter),
-      onMouseLeave: (0, import_primitive4.composeEventHandlers)(onMouseLeave, handleMouseLeave)
-    }
+  const callbackRef = (0, import_react33.useCallback)(
+    (el) => {
+      const setIsHovering = useMessageUtils.getState().setIsHovering;
+      const handleMouseEnter = () => {
+        setIsHovering(true);
+      };
+      const handleMouseLeave = () => {
+        setIsHovering(false);
+      };
+      el.addEventListener("mouseenter", handleMouseEnter);
+      el.addEventListener("mouseleave", handleMouseLeave);
+      return () => {
+        el.removeEventListener("mouseenter", handleMouseEnter);
+        el.removeEventListener("mouseleave", handleMouseLeave);
+        setIsHovering(false);
+      };
+    },
+    [useMessageUtils]
   );
+  return useManagedRef(callbackRef);
+};
+var MessagePrimitiveRoot = (0, import_react33.forwardRef)(({ onMouseEnter, onMouseLeave, ...rest }, forwardRef15) => {
+  const isHoveringRef = useIsHoveringRef();
+  const ref = (0, import_react_compose_refs.useComposedRefs)(forwardRef15, isHoveringRef);
+  return /* @__PURE__ */ (0, import_jsx_runtime12.jsx)(import_react_primitive3.Primitive.div, { ...rest, ref });
 });
 MessagePrimitiveRoot.displayName = "MessagePrimitive.Root";
 
@@ -1143,10 +1174,10 @@ var MessagePrimitiveIf = ({
 MessagePrimitiveIf.displayName = "MessagePrimitive.If";
 
 // src/primitives/message/MessageContent.tsx
-var import_react36 = require("react");
+var import_react37 = require("react");
 
 // src/context/providers/ContentPartProvider.tsx
-var import_react33 = require("react");
+var import_react34 = require("react");
 var import_zustand9 = require("zustand");
 var import_jsx_runtime13 = require("react/jsx-runtime");
 var syncContentPart = ({ message }, useContentPart, partIndex) => {
@@ -1165,14 +1196,14 @@ var syncContentPart = ({ message }, useContentPart, partIndex) => {
 };
 var useContentPartContext2 = (partIndex) => {
   const { useMessage } = useMessageContext();
-  const [context] = (0, import_react33.useState)(() => {
+  const [context] = (0, import_react34.useState)(() => {
     const useContentPart = (0, import_zustand9.create)(
       () => ({})
     );
     syncContentPart(useMessage.getState(), useContentPart, partIndex);
     return { useContentPart };
   });
-  (0, import_react33.useEffect)(() => {
+  (0, import_react34.useEffect)(() => {
     syncContentPart(useMessage.getState(), context.useContentPart, partIndex);
     return useMessage.subscribe((message) => {
       syncContentPart(message, context.useContentPart, partIndex);
@@ -1188,6 +1219,29 @@ var ContentPartProvider = ({
   return /* @__PURE__ */ (0, import_jsx_runtime13.jsx)(ContentPartContext.Provider, { value: context, children });
 };
 
+// src/primitives/contentPart/ContentPartText.tsx
+var import_react_primitive4 = require("@radix-ui/react-primitive");
+var import_react35 = require("react");
+var import_jsx_runtime14 = require("react/jsx-runtime");
+var ContentPartPrimitiveText = (0, import_react35.forwardRef)((props, forwardedRef) => {
+  const {
+    part: { text },
+    status
+  } = useContentPartText();
+  return /* @__PURE__ */ (0, import_jsx_runtime14.jsx)(import_react_primitive4.Primitive.span, { "data-status": status, ...props, ref: forwardedRef, children: text });
+});
+ContentPartPrimitiveText.displayName = "ContentPartPrimitive.Text";
+
+// src/primitives/contentPart/ContentPartImage.tsx
+var import_react_primitive5 = require("@radix-ui/react-primitive");
+var import_react36 = require("react");
+var import_jsx_runtime15 = require("react/jsx-runtime");
+var ContentPartPrimitiveImage = (0, import_react36.forwardRef)((props, forwardedRef) => {
+  const image = useContentPartImage();
+  return /* @__PURE__ */ (0, import_jsx_runtime15.jsx)(import_react_primitive5.Primitive.img, { src: image, ...props, ref: forwardedRef });
+});
+ContentPartPrimitiveImage.displayName = "ContentPartPrimitive.Image";
+
 // src/primitives/contentPart/ContentPartDisplay.tsx
 var ContentPartPrimitiveDisplay = () => {
   const display = useContentPartDisplay();
@@ -1195,61 +1249,31 @@ var ContentPartPrimitiveDisplay = () => {
 };
 ContentPartPrimitiveDisplay.displayName = "ContentPartPrimitive.Display";
 
-// src/utils/OutPortal.tsx
-var import_react34 = require("react");
-var import_jsx_runtime14 = require("react/jsx-runtime");
-var OutPortal = ({ node }) => {
-  const parentRef = (0, import_react34.useRef)(null);
-  (0, import_react34.useLayoutEffect)(() => {
-    const parent = parentRef.current;
-    if (!parent || !node) return;
-    parent.appendChild(node);
-    return () => {
-      parent.removeChild(node);
-    };
-  }, [node]);
-  if (!node) return null;
-  return /* @__PURE__ */ (0, import_jsx_runtime14.jsx)("span", { ref: parentRef });
-};
-
-// src/primitives/contentPart/ContentPartInProgressIndicator.tsx
-var import_jsx_runtime15 = require("react/jsx-runtime");
-var ContentPartPrimitiveInProgressIndicator = () => {
-  const { useMessageUtils } = useMessageContext();
+// src/primitives/contentPart/ContentPartInProgress.tsx
+var ContentPartPrimitiveInProgress = ({
+  children
+}) => {
   const { useContentPart } = useContentPartContext();
-  const indicator = useCombinedStore(
-    [useMessageUtils, useContentPart],
-    (m, c) => c.status === "in_progress" ? m.inProgressIndicator : null
-  );
-  return /* @__PURE__ */ (0, import_jsx_runtime15.jsx)(OutPortal, { node: indicator });
+  const isInProgress = useContentPart((c) => c.status === "in_progress");
+  return isInProgress ? children : null;
 };
-ContentPartPrimitiveInProgressIndicator.displayName = "ContentPartPrimitive.InProgressIndicator";
-
-// src/primitives/contentPart/ContentPartText.tsx
-var import_react_primitive4 = require("@radix-ui/react-primitive");
-var import_react35 = require("react");
-var import_jsx_runtime16 = require("react/jsx-runtime");
-var ContentPartPrimitiveText = (0, import_react35.forwardRef)((props, forwardedRef) => {
-  const text = useContentPartText();
-  return /* @__PURE__ */ (0, import_jsx_runtime16.jsx)(import_react_primitive4.Primitive.span, { ...props, ref: forwardedRef, children: text });
-});
-ContentPartPrimitiveText.displayName = "ContentPartPrimitive.Text";
+ContentPartPrimitiveInProgress.displayName = "ContentPartPrimitive.InProgress";
 
 // src/primitives/message/MessageContent.tsx
-var import_jsx_runtime17 = require("react/jsx-runtime");
+var import_jsx_runtime16 = require("react/jsx-runtime");
 var defaultComponents = {
-  Text: () => /* @__PURE__ */ (0, import_jsx_runtime17.jsxs)("p", { style: { whiteSpace: "pre-line" }, children: [
-    /* @__PURE__ */ (0, import_jsx_runtime17.jsx)(ContentPartPrimitiveText, {}),
-    /* @__PURE__ */ (0, import_jsx_runtime17.jsx)(ContentPartPrimitiveInProgressIndicator, {})
+  Text: () => /* @__PURE__ */ (0, import_jsx_runtime16.jsxs)("p", { style: { whiteSpace: "pre-line" }, children: [
+    /* @__PURE__ */ (0, import_jsx_runtime16.jsx)(ContentPartPrimitiveText, {}),
+    /* @__PURE__ */ (0, import_jsx_runtime16.jsx)(ContentPartPrimitiveInProgress, { children: /* @__PURE__ */ (0, import_jsx_runtime16.jsx)("span", { style: { fontFamily: "revert" }, children: " \u25CF" }) })
   ] }),
-  Image: () => null,
-  UI: () => /* @__PURE__ */ (0, import_jsx_runtime17.jsx)(ContentPartPrimitiveDisplay, {}),
+  Image: () => /* @__PURE__ */ (0, import_jsx_runtime16.jsx)(ContentPartPrimitiveImage, {}),
+  UI: () => /* @__PURE__ */ (0, import_jsx_runtime16.jsx)(ContentPartPrimitiveDisplay, {}),
   tools: {
     Fallback: (props) => {
       const { useToolUIs } = useAssistantContext();
       const Render = useToolUIs((s) => s.getToolUI(props.part.toolName));
       if (!Render) return null;
-      return /* @__PURE__ */ (0, import_jsx_runtime17.jsx)(Render, { ...props });
+      return /* @__PURE__ */ (0, import_jsx_runtime16.jsx)(Render, { ...props });
     }
   }
 };
@@ -1268,15 +1292,15 @@ var MessageContentPartComponent = ({
   const type = part.type;
   switch (type) {
     case "text":
-      return /* @__PURE__ */ (0, import_jsx_runtime17.jsx)(Text, { part, status });
+      return /* @__PURE__ */ (0, import_jsx_runtime16.jsx)(Text, { part, status });
     case "image":
-      return /* @__PURE__ */ (0, import_jsx_runtime17.jsx)(Image, { part, status });
+      return /* @__PURE__ */ (0, import_jsx_runtime16.jsx)(Image, { part, status });
     case "ui":
-      return /* @__PURE__ */ (0, import_jsx_runtime17.jsx)(UI, { part, status });
+      return /* @__PURE__ */ (0, import_jsx_runtime16.jsx)(UI, { part, status });
     case "tool-call": {
       const Tool = by_name[part.toolName] || Fallback;
       const addResult = (result) => addToolResult(part.toolCallId, result);
-      return /* @__PURE__ */ (0, import_jsx_runtime17.jsx)(Tool, { part, status, addResult });
+      return /* @__PURE__ */ (0, import_jsx_runtime16.jsx)(Tool, { part, status, addResult });
     }
     default:
       throw new Error(`Unknown content part type: ${type}`);
@@ -1286,9 +1310,9 @@ var MessageContentPartImpl = ({
   partIndex,
   components
 }) => {
-  return /* @__PURE__ */ (0, import_jsx_runtime17.jsx)(ContentPartProvider, { partIndex, children: /* @__PURE__ */ (0, import_jsx_runtime17.jsx)(MessageContentPartComponent, { components }) });
+  return /* @__PURE__ */ (0, import_jsx_runtime16.jsx)(ContentPartProvider, { partIndex, children: /* @__PURE__ */ (0, import_jsx_runtime16.jsx)(MessageContentPartComponent, { components }) });
 };
-var MessageContentPart = (0, import_react36.memo)(
+var MessageContentPart = (0, import_react37.memo)(
   MessageContentPartImpl,
   (prev, next) => prev.partIndex === next.partIndex && prev.components?.Text === next.components?.Text && prev.components?.Image === next.components?.Image && prev.components?.UI === next.components?.UI && prev.components?.tools === next.components?.tools
 );
@@ -1299,7 +1323,7 @@ var MessagePrimitiveContent = ({
   const contentLength = useMessage((s) => s.message.content.length);
   return new Array(contentLength).fill(null).map((_, idx) => {
     const partIndex = idx;
-    return /* @__PURE__ */ (0, import_jsx_runtime17.jsx)(
+    return /* @__PURE__ */ (0, import_jsx_runtime16.jsx)(
       MessageContentPart,
       {
         partIndex,
@@ -1312,21 +1336,15 @@ var MessagePrimitiveContent = ({
 MessagePrimitiveContent.displayName = "MessagePrimitive.Content";
 
 // src/primitives/message/MessageInProgress.tsx
-var import_react_dom = require("react-dom");
-var import_react_primitive5 = require("@radix-ui/react-primitive");
-var import_react37 = require("react");
-var import_jsx_runtime18 = require("react/jsx-runtime");
-var MessagePrimitiveInProgress = (0, import_react37.forwardRef)((props, ref) => {
-  const { useMessageUtils } = useMessageContext();
-  const portalNode = useMessageUtils((s) => s.inProgressIndicator);
-  return (0, import_react_dom.createPortal)(/* @__PURE__ */ (0, import_jsx_runtime18.jsx)(import_react_primitive5.Primitive.span, { ...props, ref }), portalNode);
-});
+var MessagePrimitiveInProgress = () => {
+  return null;
+};
 MessagePrimitiveInProgress.displayName = "MessagePrimitive.InProgress";
 
 // src/primitives/branchPicker/BranchPickerRoot.tsx
-var import_jsx_runtime19 = require("react/jsx-runtime");
+var import_jsx_runtime17 = require("react/jsx-runtime");
 var BranchPickerPrimitiveRoot = (0, import_react38.forwardRef)(({ hideWhenSingleBranch, ...rest }, ref) => {
-  return /* @__PURE__ */ (0, import_jsx_runtime19.jsx)(MessagePrimitiveIf, { hasBranches: hideWhenSingleBranch ? true : void 0, children: /* @__PURE__ */ (0, import_jsx_runtime19.jsx)(import_react_primitive6.Primitive.div, { ...rest, ref }) });
+  return /* @__PURE__ */ (0, import_jsx_runtime17.jsx)(MessagePrimitiveIf, { hasBranches: hideWhenSingleBranch ? true : void 0, children: /* @__PURE__ */ (0, import_jsx_runtime17.jsx)(import_react_primitive6.Primitive.div, { ...rest, ref }) });
 });
 BranchPickerPrimitiveRoot.displayName = "BranchPickerPrimitive.Root";
 
@@ -1341,10 +1359,10 @@ __export(composer_exports, {
 });
 
 // src/primitives/composer/ComposerRoot.tsx
-var import_primitive5 = require("@radix-ui/primitive");
+var import_primitive4 = require("@radix-ui/primitive");
 var import_react_primitive7 = require("@radix-ui/react-primitive");
 var import_react39 = require("react");
-var import_jsx_runtime20 = require("react/jsx-runtime");
+var import_jsx_runtime18 = require("react/jsx-runtime");
 var ComposerPrimitiveRoot = (0, import_react39.forwardRef)(({ onSubmit, ...rest }, forwardedRef) => {
   const send = useComposerSend();
   const handleSubmit = (e) => {
@@ -1352,25 +1370,25 @@ var ComposerPrimitiveRoot = (0, import_react39.forwardRef)(({ onSubmit, ...rest
     if (!send) return;
     send();
   };
-  return /* @__PURE__ */ (0, import_jsx_runtime20.jsx)(
+  return /* @__PURE__ */ (0, import_jsx_runtime18.jsx)(
     import_react_primitive7.Primitive.form,
     {
       ...rest,
       ref: forwardedRef,
-      onSubmit: (0, import_primitive5.composeEventHandlers)(onSubmit, handleSubmit)
+      onSubmit: (0, import_primitive4.composeEventHandlers)(onSubmit, handleSubmit)
     }
   );
 });
 ComposerPrimitiveRoot.displayName = "ComposerPrimitive.Root";
 
 // src/primitives/composer/ComposerInput.tsx
-var import_primitive6 = require("@radix-ui/primitive");
-var import_react_compose_refs = require("@radix-ui/react-compose-refs");
+var import_primitive5 = require("@radix-ui/primitive");
+var import_react_compose_refs2 = require("@radix-ui/react-compose-refs");
 var import_react_slot = require("@radix-ui/react-slot");
 var import_react40 = require("react");
 var import_react_textarea_autosize = __toESM(require("react-textarea-autosize"));
 var import_react_use_escape_keydown = require("@radix-ui/react-use-escape-keydown");
-var import_jsx_runtime21 = require("react/jsx-runtime");
+var import_jsx_runtime19 = require("react/jsx-runtime");
 var ComposerPrimitiveInput = (0, import_react40.forwardRef)(
   ({ autoFocus = false, asChild, disabled, onChange, onKeyDown, ...rest }, forwardedRef) => {
     const { useThread } = useThreadContext();
@@ -1381,10 +1399,11 @@ var ComposerPrimitiveInput = (0, import_react40.forwardRef)(
     });
     const Component = asChild ? import_react_slot.Slot : import_react_textarea_autosize.default;
     const textareaRef = (0, import_react40.useRef)(null);
-    const ref = (0, import_react_compose_refs.useComposedRefs)(forwardedRef, textareaRef);
+    const ref = (0, import_react_compose_refs2.useComposedRefs)(forwardedRef, textareaRef);
     (0, import_react_use_escape_keydown.useEscapeKeydown)((e) => {
       const composer = useComposer.getState();
-      if (composer.cancel()) {
+      if (composer.canCancel) {
+        composer.cancel();
         e.preventDefault();
       }
     });
@@ -1414,19 +1433,20 @@ var ComposerPrimitiveInput = (0, import_react40.forwardRef)(
         focus();
       }
     });
-    return /* @__PURE__ */ (0, import_jsx_runtime21.jsx)(
+    return /* @__PURE__ */ (0, import_jsx_runtime19.jsx)(
       Component,
       {
+        name: "input",
         value,
         ...rest,
         ref,
         disabled,
-        onChange: (0, import_primitive6.composeEventHandlers)(onChange, (e) => {
+        onChange: (0, import_primitive5.composeEventHandlers)(onChange, (e) => {
           const composerState = useComposer.getState();
           if (!composerState.isEditing) return;
           return composerState.setValue(e.target.value);
         }),
-        onKeyDown: (0, import_primitive6.composeEventHandlers)(onKeyDown, handleKeyPress)
+        onKeyDown: (0, import_primitive5.composeEventHandlers)(onKeyDown, handleKeyPress)
       }
     );
   }
@@ -1436,11 +1456,11 @@ ComposerPrimitiveInput.displayName = "ComposerPrimitive.Input";
 // src/primitives/composer/ComposerSend.tsx
 var import_react41 = require("react");
 var import_react_primitive8 = require("@radix-ui/react-primitive");
-var import_jsx_runtime22 = require("react/jsx-runtime");
+var import_jsx_runtime20 = require("react/jsx-runtime");
 var ComposerPrimitiveSend = (0, import_react41.forwardRef)(({ disabled, ...rest }, ref) => {
   const { useComposer } = useComposerContext();
   const hasValue = useComposer((c) => c.isEditing && c.value.length > 0);
-  return /* @__PURE__ */ (0, import_jsx_runtime22.jsx)(
+  return /* @__PURE__ */ (0, import_jsx_runtime20.jsx)(
     import_react_primitive8.Primitive.button,
     {
       type: "submit",
@@ -1473,20 +1493,10 @@ var contentPart_exports = {};
 __export(contentPart_exports, {
   Display: () => ContentPartPrimitiveDisplay,
   Image: () => ContentPartPrimitiveImage,
-  InProgressIndicator: () => ContentPartPrimitiveInProgressIndicator,
+  InProgress: () => ContentPartPrimitiveInProgress,
   Text: () => ContentPartPrimitiveText
 });
 
-// src/primitives/contentPart/ContentPartImage.tsx
-var import_react_primitive9 = require("@radix-ui/react-primitive");
-var import_react42 = require("react");
-var import_jsx_runtime23 = require("react/jsx-runtime");
-var ContentPartPrimitiveImage = (0, import_react42.forwardRef)((props, forwardedRef) => {
-  const image = useContentPartImage();
-  return /* @__PURE__ */ (0, import_jsx_runtime23.jsx)(import_react_primitive9.Primitive.img, { src: image, ...props, ref: forwardedRef });
-});
-ContentPartPrimitiveImage.displayName = "ContentPartPrimitive.Image";
-
 // src/primitives/thread/index.ts
 var thread_exports = {};
 __export(thread_exports, {
@@ -1500,11 +1510,11 @@ __export(thread_exports, {
 });
 
 // src/primitives/thread/ThreadRoot.tsx
-var import_react_primitive10 = require("@radix-ui/react-primitive");
-var import_react43 = require("react");
-var import_jsx_runtime24 = require("react/jsx-runtime");
-var ThreadPrimitiveRoot = (0, import_react43.forwardRef)((props, ref) => {
-  return /* @__PURE__ */ (0, import_jsx_runtime24.jsx)(import_react_primitive10.Primitive.div, { ...props, ref });
+var import_react_primitive9 = require("@radix-ui/react-primitive");
+var import_react42 = require("react");
+var import_jsx_runtime21 = require("react/jsx-runtime");
+var ThreadPrimitiveRoot = (0, import_react42.forwardRef)((props, ref) => {
+  return /* @__PURE__ */ (0, import_jsx_runtime21.jsx)(import_react_primitive9.Primitive.div, { ...props, ref });
 });
 ThreadPrimitiveRoot.displayName = "ThreadPrimitive.Root";
 
@@ -1528,40 +1538,20 @@ var ThreadPrimitiveIf = ({
 ThreadPrimitiveIf.displayName = "ThreadPrimitive.If";
 
 // src/primitives/thread/ThreadViewport.tsx
-var import_react_compose_refs3 = require("@radix-ui/react-compose-refs");
-var import_react_primitive11 = require("@radix-ui/react-primitive");
-var import_react48 = require("react");
+var import_react_compose_refs4 = require("@radix-ui/react-compose-refs");
+var import_react_primitive10 = require("@radix-ui/react-primitive");
+var import_react46 = require("react");
 
 // src/primitive-hooks/thread/useThreadViewportAutoScroll.tsx
-var import_react_compose_refs2 = require("@radix-ui/react-compose-refs");
-var import_react47 = require("react");
-
-// src/utils/hooks/useOnResizeContent.tsx
-var import_react_use_callback_ref2 = require("@radix-ui/react-use-callback-ref");
+var import_react_compose_refs3 = require("@radix-ui/react-compose-refs");
 var import_react45 = require("react");
 
-// src/utils/hooks/useManagedRef.ts
-var import_react44 = require("react");
-var useManagedRef = (callback) => {
-  const cleanupRef = (0, import_react44.useRef)();
-  const ref = (0, import_react44.useCallback)(
-    (el) => {
-      if (cleanupRef.current) {
-        cleanupRef.current();
-      }
-      if (el) {
-        cleanupRef.current = callback(el);
-      }
-    },
-    [callback]
-  );
-  return ref;
-};
-
 // src/utils/hooks/useOnResizeContent.tsx
+var import_react_use_callback_ref2 = require("@radix-ui/react-use-callback-ref");
+var import_react43 = require("react");
 var useOnResizeContent = (callback) => {
   const callbackRef = (0, import_react_use_callback_ref2.useCallbackRef)(callback);
-  const refCallback = (0, import_react45.useCallback)(
+  const refCallback = (0, import_react43.useCallback)(
     (el) => {
       const resizeObserver = new ResizeObserver(() => {
         callbackRef();
@@ -1598,11 +1588,11 @@ var useOnResizeContent = (callback) => {
 
 // src/utils/hooks/useOnScrollToBottom.tsx
 var import_react_use_callback_ref3 = require("@radix-ui/react-use-callback-ref");
-var import_react46 = require("react");
+var import_react44 = require("react");
 var useOnScrollToBottom = (callback) => {
   const callbackRef = (0, import_react_use_callback_ref3.useCallbackRef)(callback);
   const { useViewport } = useThreadContext();
-  (0, import_react46.useEffect)(() => {
+  (0, import_react44.useEffect)(() => {
     return useViewport.getState().onScrollToBottom(() => {
       callbackRef();
     });
@@ -1613,11 +1603,11 @@ var useOnScrollToBottom = (callback) => {
 var useThreadViewportAutoScroll = ({
   autoScroll = true
 }) => {
-  const divRef = (0, import_react47.useRef)(null);
+  const divRef = (0, import_react45.useRef)(null);
   const { useViewport } = useThreadContext();
-  const firstRenderRef = (0, import_react47.useRef)(true);
-  const lastScrollTop = (0, import_react47.useRef)(0);
-  const isScrollingToBottomRef = (0, import_react47.useRef)(false);
+  const firstRenderRef = (0, import_react45.useRef)(true);
+  const lastScrollTop = (0, import_react45.useRef)(0);
+  const isScrollingToBottomRef = (0, import_react45.useRef)(false);
   const scrollToBottom = () => {
     const div = divRef.current;
     if (!div || !autoScroll) return;
@@ -1655,7 +1645,7 @@ var useThreadViewportAutoScroll = ({
       el.removeEventListener("scroll", handleScroll);
     };
   });
-  const autoScrollRef = (0, import_react_compose_refs2.useComposedRefs)(resizeRef, scrollRef, divRef);
+  const autoScrollRef = (0, import_react_compose_refs3.useComposedRefs)(resizeRef, scrollRef, divRef);
   useOnScrollToBottom(() => {
     scrollToBottom();
   });
@@ -1663,21 +1653,21 @@ var useThreadViewportAutoScroll = ({
 };
 
 // src/primitives/thread/ThreadViewport.tsx
-var import_jsx_runtime25 = require("react/jsx-runtime");
-var ThreadPrimitiveViewport = (0, import_react48.forwardRef)(({ autoScroll, onScroll, children, ...rest }, forwardedRef) => {
+var import_jsx_runtime22 = require("react/jsx-runtime");
+var ThreadPrimitiveViewport = (0, import_react46.forwardRef)(({ autoScroll, onScroll, children, ...rest }, forwardedRef) => {
   const autoScrollRef = useThreadViewportAutoScroll({
     autoScroll
   });
-  const ref = (0, import_react_compose_refs3.useComposedRefs)(forwardedRef, autoScrollRef);
-  return /* @__PURE__ */ (0, import_jsx_runtime25.jsx)(import_react_primitive11.Primitive.div, { ...rest, ref, children });
+  const ref = (0, import_react_compose_refs4.useComposedRefs)(forwardedRef, autoScrollRef);
+  return /* @__PURE__ */ (0, import_jsx_runtime22.jsx)(import_react_primitive10.Primitive.div, { ...rest, ref, children });
 });
 ThreadPrimitiveViewport.displayName = "ThreadPrimitive.Viewport";
 
 // src/primitives/thread/ThreadMessages.tsx
-var import_react50 = require("react");
+var import_react48 = require("react");
 
 // src/context/providers/MessageProvider.tsx
-var import_react49 = require("react");
+var import_react47 = require("react");
 var import_zustand12 = require("zustand");
 
 // src/context/stores/EditComposer.ts
@@ -1687,27 +1677,25 @@ var makeEditComposerStore = ({
   onSend
 }) => (0, import_zustand10.create)()((set, get, store) => ({
   ...makeBaseComposer(set, get, store),
+  canCancel: false,
   isEditing: false,
   edit: () => {
     const value = onEdit();
-    set({ isEditing: true, value });
+    set({ isEditing: true, canCancel: true, value });
   },
   send: () => {
     const value = get().value;
-    set({ isEditing: false });
+    set({ isEditing: false, canCancel: false });
     onSend(value);
   },
   cancel: () => {
-    if (!get().isEditing) return false;
-    set({ isEditing: false });
-    return true;
+    set({ isEditing: false, canCancel: false });
   }
 }));
 
 // src/context/stores/MessageUtils.ts
 var import_zustand11 = require("zustand");
 var makeMessageUtilsStore = () => (0, import_zustand11.create)((set) => ({
-  inProgressIndicator: document.createElement("span"),
   isCopied: false,
   setIsCopied: (value) => {
     set({ isCopied: value });
@@ -1719,7 +1707,7 @@ var makeMessageUtilsStore = () => (0, import_zustand11.create)((set) => ({
 }));
 
 // src/context/providers/MessageProvider.tsx
-var import_jsx_runtime26 = require("react/jsx-runtime");
+var import_jsx_runtime23 = require("react/jsx-runtime");
 var getIsLast = (messages, message) => {
   return messages[messages.length - 1]?.id === message.id;
 };
@@ -1741,7 +1729,7 @@ var syncMessage = (messages, getBranches, useMessage, messageIndex) => {
 };
 var useMessageContext2 = (messageIndex) => {
   const { useThreadMessages, useThreadActions } = useThreadContext();
-  const [context] = (0, import_react49.useState)(() => {
+  const [context] = (0, import_react47.useState)(() => {
     const useMessage = (0, import_zustand12.create)(() => ({}));
     const useMessageUtils = makeMessageUtilsStore();
     const useEditComposer = makeEditComposerStore({
@@ -1778,7 +1766,7 @@ var useMessageContext2 = (messageIndex) => {
     );
     return { useMessage, useMessageUtils, useEditComposer };
   });
-  (0, import_react49.useEffect)(() => {
+  (0, import_react47.useEffect)(() => {
     return useThreadMessages.subscribe((thread) => {
       syncMessage(
         thread,
@@ -1795,11 +1783,11 @@ var MessageProvider = ({
   children
 }) => {
   const context = useMessageContext2(messageIndex);
-  return /* @__PURE__ */ (0, import_jsx_runtime26.jsx)(MessageContext.Provider, { value: context, children });
+  return /* @__PURE__ */ (0, import_jsx_runtime23.jsx)(MessageContext.Provider, { value: context, children });
 };
 
 // src/primitives/thread/ThreadMessages.tsx
-var import_jsx_runtime27 = require("react/jsx-runtime");
+var import_jsx_runtime24 = require("react/jsx-runtime");
 var getComponents = (components) => {
   return {
     EditComposer: components.EditComposer ?? components.UserMessage ?? components.Message,
@@ -1812,19 +1800,19 @@ var ThreadMessageImpl = ({
   components
 }) => {
   const { UserMessage, EditComposer, AssistantMessage } = getComponents(components);
-  return /* @__PURE__ */ (0, import_jsx_runtime27.jsxs)(MessageProvider, { messageIndex, children: [
-    /* @__PURE__ */ (0, import_jsx_runtime27.jsxs)(MessagePrimitiveIf, { user: true, children: [
-      /* @__PURE__ */ (0, import_jsx_runtime27.jsx)(ComposerPrimitiveIf, { editing: false, children: /* @__PURE__ */ (0, import_jsx_runtime27.jsx)(UserMessage, {}) }),
-      /* @__PURE__ */ (0, import_jsx_runtime27.jsx)(ComposerPrimitiveIf, { editing: true, children: /* @__PURE__ */ (0, import_jsx_runtime27.jsx)(EditComposer, {}) })
+  return /* @__PURE__ */ (0, import_jsx_runtime24.jsxs)(MessageProvider, { messageIndex, children: [
+    /* @__PURE__ */ (0, import_jsx_runtime24.jsxs)(MessagePrimitiveIf, { user: true, children: [
+      /* @__PURE__ */ (0, import_jsx_runtime24.jsx)(ComposerPrimitiveIf, { editing: false, children: /* @__PURE__ */ (0, import_jsx_runtime24.jsx)(UserMessage, {}) }),
+      /* @__PURE__ */ (0, import_jsx_runtime24.jsx)(ComposerPrimitiveIf, { editing: true, children: /* @__PURE__ */ (0, import_jsx_runtime24.jsx)(EditComposer, {}) })
     ] }),
-    /* @__PURE__ */ (0, import_jsx_runtime27.jsx)(MessagePrimitiveIf, { assistant: true, children: /* @__PURE__ */ (0, import_jsx_runtime27.jsx)(AssistantMessage, {}) })
+    /* @__PURE__ */ (0, import_jsx_runtime24.jsx)(MessagePrimitiveIf, { assistant: true, children: /* @__PURE__ */ (0, import_jsx_runtime24.jsx)(AssistantMessage, {}) })
   ] });
 };
-var ThreadMessage = (0, import_react50.memo)(
+var ThreadMessage = (0, import_react48.memo)(
   ThreadMessageImpl,
-  (prev, next) => prev.messageIndex === next.messageIndex && prev.components.UserMessage === next.components.UserMessage && prev.components.EditComposer === next.components.EditComposer && prev.components.AssistantMessage === next.components.AssistantMessage
+  (prev, next) => prev.messageIndex === next.messageIndex && prev.components.Message === next.components.Message && prev.components.UserMessage === next.components.UserMessage && prev.components.EditComposer === next.components.EditComposer && prev.components.AssistantMessage === next.components.AssistantMessage
 );
-var ThreadPrimitiveMessages = ({
+var ThreadPrimitiveMessagesImpl = ({
   components
 }) => {
   const { useThreadMessages } = useThreadContext();
@@ -1832,7 +1820,7 @@ var ThreadPrimitiveMessages = ({
   if (messagesLength === 0) return null;
   return new Array(messagesLength).fill(null).map((_, idx) => {
     const messageIndex = idx;
-    return /* @__PURE__ */ (0, import_jsx_runtime27.jsx)(
+    return /* @__PURE__ */ (0, import_jsx_runtime24.jsx)(
       ThreadMessage,
       {
         messageIndex,
@@ -1842,7 +1830,11 @@ var ThreadPrimitiveMessages = ({
     );
   });
 };
-ThreadPrimitiveMessages.displayName = "ThreadPrimitive.Messages";
+ThreadPrimitiveMessagesImpl.displayName = "ThreadPrimitive.Messages";
+var ThreadPrimitiveMessages = (0, import_react48.memo)(
+  ThreadPrimitiveMessagesImpl,
+  (prev, next) => prev.components?.Message === next.components?.Message && prev.components?.UserMessage === next.components?.UserMessage && prev.components?.EditComposer === next.components?.EditComposer && prev.components?.AssistantMessage === next.components?.AssistantMessage
+);
 
 // src/primitives/thread/ThreadScrollToBottom.tsx
 var ThreadPrimitiveScrollToBottom = createActionButton(
@@ -1857,7 +1849,7 @@ var ThreadPrimitiveSuggestion = createActionButton(
 );
 
 // src/runtime/local/useLocalRuntime.tsx
-var import_react51 = require("react");
+var import_react49 = require("react");
 
 // src/runtime/utils/idUtils.tsx
 var import_non_secure = require("nanoid/non-secure");
@@ -2026,6 +2018,9 @@ var BaseAssistantRuntime = class {
     this._thread = _thread;
     this._unsubscribe = this._thread.subscribe(this.subscriptionHandler);
   }
+  get capabilities() {
+    return this._thread.capabilities;
+  }
   _unsubscribe;
   get thread() {
     return this._thread;
@@ -2098,6 +2093,12 @@ var LocalRuntime = class extends BaseAssistantRuntime {
     );
   }
 };
+var CAPABILITIES = Object.freeze({
+  edit: true,
+  reload: true,
+  cancel: true,
+  copy: true
+});
 var LocalThreadRuntime = class {
   constructor(_configProviders, adapter) {
     this._configProviders = _configProviders;
@@ -2106,6 +2107,7 @@ var LocalThreadRuntime = class {
   _subscriptions = /* @__PURE__ */ new Set();
   abortController = null;
   repository = new MessageRepository();
+  capabilities = CAPABILITIES;
   get messages() {
     return this.repository.getMessages();
   }
@@ -2189,8 +2191,8 @@ var LocalThreadRuntime = class {
 
 // src/runtime/local/useLocalRuntime.tsx
 var useLocalRuntime = (adapter) => {
-  const [runtime] = (0, import_react51.useState)(() => new LocalRuntime(adapter));
-  (0, import_react51.useInsertionEffect)(() => {
+  const [runtime] = (0, import_react49.useState)(() => new LocalRuntime(adapter));
+  (0, import_react49.useInsertionEffect)(() => {
     runtime.adapter = adapter;
   });
   return runtime;