feat: MessageRuntime.unstable_getCopyText API (#1011)

.changeset/thin-mayflies-love.md

@@ -0,0 +1,5 @@
+---
+"@assistant-ui/react": patch
+---
+
+refactor!: simplify SpeechSynthesisAdapter to accept a text string

packages/react/src/api/MessageRuntime.ts

@@ -12,6 +12,7 @@ import {
   ContentPartStatus,
   ToolCallContentPartStatus,
 } from "../types/AssistantTypes";
+import { getThreadMessageText } from "../utils/getThreadMessageText";
 import {
   AttachmentRuntime,
   AttachmentState,
@@ -128,6 +129,7 @@ export type MessageRuntime = {
     position?: "previous" | "next" | undefined;
     branchId?: string | undefined;
   }): void;
+  unstable_getCopyText(): string;
 
   subscribe(callback: () => void): Unsubscribe;
 
@@ -223,6 +225,10 @@ export class MessageRuntimeImpl implements MessageRuntime {
     this._threadBinding.getState().switchToBranch(targetBranch);
   }
 
+  public unstable_getCopyText() {
+    return getThreadMessageText(this.getState());
+  }
+
   public subscribe(callback: () => void) {
     return this._core.subscribe(callback);
   }

packages/react/src/primitive-hooks/actionBar/index.ts

@@ -1,10 +1,10 @@
 export {
   /**
-   * @deprecated Use `useMessageRuntime().copy()` instead. This will be removed in 0.6.
+   * @deprecated Use `useMessageRuntime().unstable_getCopyText()` instead. This will be removed in 0.6.
    */
   useActionBarCopy,
   /**
-   * @deprecated Use `useMessageRuntime().copy()` instead. This will be removed in 0.6.
+   * @deprecated Use `useMessageRuntime().unstable_getCopyText()` instead. This will be removed in 0.6.
    */
   type UseActionBarCopyProps,
 } from "./useActionBarCopy";

packages/react/src/primitive-hooks/actionBar/useActionBarCopy.tsx

@@ -1,10 +1,9 @@
 import { useCallback } from "react";
 import {
+  useMessage,
   useMessageRuntime,
-  useMessageUtilsStore,
+  useMessageUtils,
 } from "../../context/react/MessageContext";
-import { useCombinedStore } from "../../utils/combined/useCombinedStore";
-import { getThreadMessageText } from "../../utils/getThreadMessageText";
 import { useComposerRuntime } from "../../context";
 
 export type UseActionBarCopyProps = {
@@ -16,32 +15,26 @@ export const useActionBarCopy = ({
 }: UseActionBarCopyProps = {}) => {
   const messageRuntime = useMessageRuntime();
   const composerRuntime = useComposerRuntime();
-  const messageUtilsStore = useMessageUtilsStore();
-  const hasCopyableContent = useCombinedStore(
-    [messageRuntime, composerRuntime],
-    (message, c) => {
-      return (
-        !c.isEditing &&
-        (message.role !== "assistant" || message.status.type !== "running") &&
-        message.content.some((c) => c.type === "text" && c.text.length > 0)
-      );
-    },
-  );
+  const setIsCopied = useMessageUtils((s) => s.setIsCopied);
+  const hasCopyableContent = useMessage((message) => {
+    return (
+      (message.role !== "assistant" || message.status.type !== "running") &&
+      message.content.some((c) => c.type === "text" && c.text.length > 0)
+    );
+  });
 
   const callback = useCallback(() => {
-    const message = messageRuntime.getState();
-    const { setIsCopied } = messageUtilsStore.getState();
     const { isEditing, text: composerValue } = composerRuntime.getState();
 
     const valueToCopy = isEditing
       ? composerValue
-      : getThreadMessageText(message);
+      : messageRuntime.unstable_getCopyText();
 
     navigator.clipboard.writeText(valueToCopy).then(() => {
       setIsCopied(true);
       setTimeout(() => setIsCopied(false), copiedDuration);
     });
-  }, [messageRuntime, messageUtilsStore, composerRuntime, copiedDuration]);
+  }, [messageRuntime, setIsCopied, composerRuntime, copiedDuration]);
 
   if (!hasCopyableContent) return null;
   return callback;