@assistant-ui/react 0.12.9 → 0.12.10

package/src/hooks/useMessageQuote.ts

@@ -0,0 +1,27 @@
+"use client";
+
+import { useAuiState } from "@assistant-ui/store";
+import type { QuoteInfo } from "../types/QuoteTypes";
+
+/**
+ * Hook that returns the quote info for the current message, if any.
+ *
+ * Reads from `message.metadata.custom.quote`.
+ *
+ * @example
+ * ```tsx
+ * function QuoteBlock() {
+ *   const quote = useMessageQuote();
+ *   if (!quote) return null;
+ *   return <blockquote>{quote.text}</blockquote>;
+ * }
+ * ```
+ */
+export const useMessageQuote = (): QuoteInfo | undefined => {
+  return useAuiState(
+    (s) =>
+      (s.message.metadata?.custom as Record<string, unknown>)?.quote as
+        | QuoteInfo
+        | undefined,
+  );
+};

package/src/legacy-runtime/AssistantRuntimeProvider.tsx

@@ -32,10 +32,7 @@ export const AssistantRuntimeProviderImpl: FC<
   const aui = useAui({ threads: RuntimeAdapter(runtime) }, { parent: parent });
 
   useEffect(() => {
-    if (
-      typeof process === "undefined" ||
-      process.env["NODE_ENV"] === "production"
-    )
+    if (typeof process === "undefined" || process.env.NODE_ENV === "production")
       return;
     return DevToolsProviderApi.register(aui);
   }, [aui]);

package/src/legacy-runtime/client/ComposerRuntimeClient.ts

@@ -100,6 +100,7 @@ export const ComposerClient = resource(
         isEmpty: runtimeState.isEmpty,
         type: runtimeState.type ?? "thread",
         dictation: runtimeState.dictation,
+        quote: runtimeState.quote,
       };
     }, [runtimeState, attachments.state]);
 
@@ -122,6 +123,7 @@ export const ComposerClient = resource(
           }),
         startDictation: runtime.startDictation,
         stopDictation: runtime.stopDictation,
+        setQuote: runtime.setQuote,
         attachment: (selector) => {
           if ("id" in selector) {
             return attachments.get({ key: selector.id });

package/src/legacy-runtime/hooks/AssistantContext.ts

@@ -62,6 +62,6 @@ const useThreadListRuntime = (opt: {
 }): ThreadListRuntime | null => useAssistantRuntime(opt)?.threads ?? null;
 
 /**
- * @deprecated Use `useAuiState(({ threads }) => threads)` instead. See migration guide: https://assistant-ui.com/docs/migrations/v0-12
+ * @deprecated Use `useAuiState((s) => s.threads)` instead. See migration guide: https://assistant-ui.com/docs/migrations/v0-12
  */
 export const useThreadList = createStateHookForRuntime(useThreadListRuntime);

package/src/legacy-runtime/hooks/AttachmentContext.ts

@@ -84,7 +84,7 @@ export function useMessageAttachmentRuntime(options?: {
 }
 
 /**
- * @deprecated Use `useAuiState(({ attachment }) => attachment)` instead. See migration guide: https://assistant-ui.com/docs/migrations/v0-12
+ * @deprecated Use `useAuiState((s) => s.attachment)` instead. See migration guide: https://assistant-ui.com/docs/migrations/v0-12
  */
 export const useAttachment = createStateHookForRuntime(useAttachmentRuntime);
 

package/src/legacy-runtime/hooks/ComposerContext.ts

@@ -44,8 +44,8 @@ import { createStateHookForRuntime } from "../../context/react/utils/createState
  * // After:
  * function ComposerActions() {
  *   const aui = useAui();
- *   const canSend = useAuiState(({ composer }) => composer.canSend);
- *   const canCancel = useAuiState(({ composer }) => composer.canCancel);
+ *   const canSend = useAuiState((s) => s.composer.canSend);
+ *   const canCancel = useAuiState((s) => s.composer.canCancel);
  *   const handleSend = () => {
  *     if (canSend) {
  *       aui.composer().send();
@@ -87,7 +87,7 @@ export function useComposerRuntime(options?: {
 }
 
 /**
- * @deprecated Use `useAuiState(({ composer }) => composer)` instead. See migration guide: https://assistant-ui.com/docs/migrations/v0-12
+ * @deprecated Use `useAuiState((s) => s.composer)` instead. See migration guide: https://assistant-ui.com/docs/migrations/v0-12
  *
  * Hook to access the current composer state.
  *
@@ -115,9 +115,9 @@ export function useComposerRuntime(options?: {
  *
  * // After:
  * function ComposerStatus() {
- *   const text = useAuiState(({ composer }) => composer.text);
- *   const canSend = useAuiState(({ composer }) => composer.canSend);
- *   const attachmentCount = useAuiState(({ composer }) => composer.attachments.length);
+ *   const text = useAuiState((s) => s.composer.text);
+ *   const canSend = useAuiState((s) => s.composer.canSend);
+ *   const attachmentCount = useAuiState((s) => s.composer.attachments.length);
  *   return (
  *     <div>
  *       Text: {text.length} chars,

package/src/legacy-runtime/hooks/MessageContext.ts

@@ -76,7 +76,7 @@ export function useMessageRuntime(options?: {
 }
 
 /**
- * @deprecated Use `useAuiState(({ message }) => message)` instead. See migration guide: https://assistant-ui.com/docs/migrations/v0-12
+ * @deprecated Use `useAuiState((s) => s.message)` instead. See migration guide: https://assistant-ui.com/docs/migrations/v0-12
  *
  * Hook to access the current message state.
  *
@@ -102,9 +102,9 @@ export function useMessageRuntime(options?: {
  *
  * // After:
  * function MessageContent() {
- *   const role = useAuiState(({ message }) => message.role);
- *   const content = useAuiState(({ message }) => message.content);
- *   const isLoading = useAuiState(({ message }) => message.status.type === "running");
+ *   const role = useAuiState((s) => s.message.role);
+ *   const content = useAuiState((s) => s.message.content);
+ *   const isLoading = useAuiState((s) => s.message.status.type === "running");
  *   return (
  *     <div className={`message-${role}`}>
  *       {isLoading ? "Loading..." : content.map(part => part.text).join("")}
@@ -120,7 +120,7 @@ const useEditComposerRuntime = (opt: {
 }): EditComposerRuntime | null => useMessageRuntime(opt)?.composer ?? null;
 
 /**
- * @deprecated Use `useAuiState(({ message }) => message.composer)` instead. See migration guide: https://assistant-ui.com/docs/migrations/v0-12
+ * @deprecated Use `useAuiState((s) => s.message.composer)` instead. See migration guide: https://assistant-ui.com/docs/migrations/v0-12
  */
 export const useEditComposer = createStateHookForRuntime(
   useEditComposerRuntime,

package/src/legacy-runtime/hooks/MessagePartContext.ts

@@ -27,6 +27,6 @@ export function useMessagePartRuntime(options?: {
 }
 
 /**
- * @deprecated Use `useAuiState(({ part }) => part)` instead. See migration guide: https://assistant-ui.com/docs/migrations/v0-12
+ * @deprecated Use `useAuiState((s) => s.part)` instead. See migration guide: https://assistant-ui.com/docs/migrations/v0-12
  */
 export const useMessagePart = createStateHookForRuntime(useMessagePartRuntime);

package/src/legacy-runtime/hooks/ThreadContext.ts

@@ -58,7 +58,7 @@ export function useThreadRuntime(options?: { optional?: boolean | undefined }) {
 }
 
 /**
- * @deprecated Use `useAuiState(({ thread }) => thread)` instead. See migration guide: https://assistant-ui.com/docs/migrations/v0-12
+ * @deprecated Use `useAuiState((s) => s.thread)` instead. See migration guide: https://assistant-ui.com/docs/migrations/v0-12
  *
  * Hook to access the current thread state.
  *
@@ -79,8 +79,8 @@ export function useThreadRuntime(options?: { optional?: boolean | undefined }) {
  *
  * // After:
  * function ThreadStatus() {
- *   const isRunning = useAuiState(({ thread }) => thread.isRunning);
- *   const messageCount = useAuiState(({ thread }) => thread.messages.length);
+ *   const isRunning = useAuiState((s) => s.thread.isRunning);
+ *   const messageCount = useAuiState((s) => s.thread.messages.length);
  *   return <div>Running: {isRunning}, Messages: {messageCount}</div>;
  * }
  * ```
@@ -92,14 +92,14 @@ const useThreadComposerRuntime = (opt: {
 }): ThreadComposerRuntime | null => useThreadRuntime(opt)?.composer ?? null;
 
 /**
- * @deprecated Use `useAuiState(({ thread }) => thread.composer)` instead. See migration guide: https://assistant-ui.com/docs/migrations/v0-12
+ * @deprecated Use `useAuiState((s) => s.thread.composer)` instead. See migration guide: https://assistant-ui.com/docs/migrations/v0-12
  */
 export const useThreadComposer = createStateHookForRuntime(
   useThreadComposerRuntime,
 );
 
 /**
- * @deprecated Use `useAuiState(({ thread }) => thread.modelContext)` instead. See migration guide: https://assistant-ui.com/docs/migrations/v0-12
+ * @deprecated Use `useAuiState((s) => s.thread.modelContext)` instead. See migration guide: https://assistant-ui.com/docs/migrations/v0-12
  */
 export function useThreadModelContext(options?: {
   optional?: false | undefined;

package/src/legacy-runtime/hooks/ThreadListItemContext.ts

@@ -29,7 +29,7 @@ export function useThreadListItemRuntime(options?: {
 }
 
 /**
- * @deprecated Use `useAuiState(({ threadListItem }) => threadListItem)` instead. See migration guide: https://assistant-ui.com/docs/migrations/v0-12
+ * @deprecated Use `useAuiState((s) => s.threadListItem)` instead. See migration guide: https://assistant-ui.com/docs/migrations/v0-12
  */
 export const useThreadListItem = createStateHookForRuntime(
   useThreadListItemRuntime,

package/src/legacy-runtime/runtime/ComposerRuntime.ts

@@ -10,6 +10,7 @@ import type {
 } from "../runtime-cores/core/ComposerRuntimeCore";
 import type { Unsubscribe } from "../../types";
 import type { MessageRole, RunConfig } from "../../types/AssistantTypes";
+import type { QuoteInfo } from "../../types/QuoteTypes";
 import type {
   ThreadComposerRuntimeCoreBinding,
   EditComposerRuntimeCoreBinding,
@@ -51,6 +52,12 @@ type BaseComposerState = {
    * Undefined when dictation is not active.
    */
   readonly dictation: DictationState | undefined;
+
+  /**
+   * The currently quoted text, if any.
+   * Undefined when no quote is set.
+   */
+  readonly quote: QuoteInfo | undefined;
 };
 
 export type ThreadComposerState = BaseComposerState & {
@@ -83,6 +90,7 @@ const getThreadComposerState = (
     runConfig: runtime?.runConfig ?? EMPTY_OBJECT,
     attachmentAccept: runtime?.attachmentAccept ?? "",
     dictation: runtime?.dictation,
+    quote: runtime?.quote,
 
     value: runtime?.text ?? "",
   });
@@ -104,6 +112,7 @@ const getEditComposerState = (
     runConfig: runtime?.runConfig ?? EMPTY_OBJECT,
     attachmentAccept: runtime?.attachmentAccept ?? "",
     dictation: runtime?.dictation,
+    quote: runtime?.quote,
 
     value: runtime?.text ?? "",
   });
@@ -192,6 +201,12 @@ export type ComposerRuntime = {
    */
   stopDictation(): void;
 
+  /**
+   * Set a quote for the next message. Pass undefined to clear.
+   * @param quote The quote info to set, or undefined to clear.
+   */
+  setQuote(quote: QuoteInfo | undefined): void;
+
   /**
    * @deprecated This API is still under active development and might change without notice.
    */
@@ -224,6 +239,7 @@ export abstract class ComposerRuntimeImpl implements ComposerRuntime {
     this.getAttachmentByIndex = this.getAttachmentByIndex.bind(this);
     this.startDictation = this.startDictation.bind(this);
     this.stopDictation = this.stopDictation.bind(this);
+    this.setQuote = this.setQuote.bind(this);
     this.unstable_on = this.unstable_on.bind(this);
   }
 
@@ -289,6 +305,12 @@ export abstract class ComposerRuntimeImpl implements ComposerRuntime {
     core.stopDictation();
   }
 
+  public setQuote(quote: QuoteInfo | undefined) {
+    const core = this._core.getState();
+    if (!core) throw new Error("Composer is not available");
+    core.setQuote(quote);
+  }
+
   public subscribe(callback: () => void) {
     return this._core.subscribe(callback);
   }

package/src/legacy-runtime/runtime-cores/assistant-transport/types.ts

@@ -31,6 +31,8 @@ export type AssistantMessage = {
 export type AddMessageCommand = {
   readonly type: "add-message";
   readonly message: UserMessage | AssistantMessage;
+  readonly parentId: string | null;
+  readonly sourceId: string | null;
 };
 
 export type AddToolResultCommand = {
@@ -109,6 +111,9 @@ export type AssistantTransportOptions<T> = {
     updateState: (updater: (state: T) => T) => void;
     error?: Error;
   }) => void;
+  capabilities?: {
+    edit?: boolean;
+  };
   adapters?: {
     attachments?: AttachmentAdapter | undefined;
     history?: ThreadHistoryAdapter | undefined;

package/src/legacy-runtime/runtime-cores/assistant-transport/useAssistantTransportRuntime.tsx

@@ -35,6 +35,36 @@ import { InMemoryThreadListAdapter } from "../remote-thread-list/adapter/in-memo
 import { useAui, useAuiState } from "@assistant-ui/store";
 import { UserExternalState } from "../../../augmentations";
 
+const convertAppendMessageToCommand = (
+  message: AppendMessage,
+): AddMessageCommand => {
+  if (message.role !== "user")
+    throw new Error("Only user messages are supported");
+
+  const parts: UserMessagePart[] = [];
+  const content = [
+    ...message.content,
+    ...(message.attachments?.flatMap((a) => a.content) ?? []),
+  ];
+  for (const contentPart of content) {
+    if (contentPart.type === "text") {
+      parts.push({ type: "text", text: contentPart.text });
+    } else if (contentPart.type === "image") {
+      parts.push({ type: "image", image: contentPart.image });
+    }
+  }
+
+  return {
+    type: "add-message",
+    message: {
+      role: "user",
+      parts,
+    },
+    parentId: message.parentId,
+    sourceId: message.sourceId,
+  };
+};
+
 const symbolAssistantTransportExtras = Symbol("assistant-transport-extras");
 type AssistantTransportExtras = {
   [symbolAssistantTransportExtras]: true;
@@ -74,8 +104,8 @@ export function useAssistantTransportState<T>(
 export function useAssistantTransportState<T>(
   selector: (state: UserExternalState) => T = (t) => t as T,
 ): T | UserExternalState {
-  return useAuiState(({ thread }) =>
-    selector(asAssistantTransportExtras(thread.extras).state),
+  return useAuiState((s) =>
+    selector(asAssistantTransportExtras(s.thread.extras).state),
   );
 }
 
@@ -90,7 +120,7 @@ const useAssistantTransportThreadRuntime = <T,>(
     onQueue: () => runManager.schedule(),
   });
 
-  const threadId = useAuiState(({ threadListItem }) => threadListItem.remoteId);
+  const threadId = useAuiState((s) => s.threadListItem.remoteId);
 
   const runManager = useRunManager({
     onRun: async (signal: AbortSignal) => {
@@ -254,37 +284,17 @@ const useAssistantTransportThreadRuntime = <T,>(
       state: agentStateRef.current as UserExternalState,
     } satisfies AssistantTransportExtras,
     onNew: async (message: AppendMessage): Promise<void> => {
-      if (message.role !== "user")
-        throw new Error("Only user messages are supported");
-
-      // Store parentId for the request
       parentIdRef.current = message.parentId;
-
-      // Convert AppendMessage to AddMessageCommand
-      const parts: UserMessagePart[] = [];
-
-      const content = [
-        ...message.content,
-        ...(message.attachments?.flatMap((a) => a.content) ?? []),
-      ];
-      for (const contentPart of content) {
-        if (contentPart.type === "text") {
-          parts.push({ type: "text", text: contentPart.text });
-        } else if (contentPart.type === "image") {
-          parts.push({ type: "image", image: contentPart.image });
-        }
-      }
-
-      const command: AddMessageCommand = {
-        type: "add-message",
-        message: {
-          role: "user",
-          parts,
-        },
-      };
-
+      const command = convertAppendMessageToCommand(message);
       commandQueue.enqueue(command);
     },
+    ...(options.capabilities?.edit && {
+      onEdit: async (message: AppendMessage): Promise<void> => {
+        parentIdRef.current = message.parentId;
+        const command = convertAppendMessageToCommand(message);
+        commandQueue.enqueue(command);
+      },
+    }),
     onCancel: async () => {
       runManager.cancel();
       await toolInvocations.abort();

package/src/legacy-runtime/runtime-cores/assistant-transport/useToolInvocations.ts

@@ -181,7 +181,7 @@ export function useToolInvocations({
 
               if (content.argsText !== lastState.argsText) {
                 if (lastState.argsComplete) {
-                  if (process.env["NODE_ENV"] !== "production") {
+                  if (process.env.NODE_ENV !== "production") {
                     console.warn(
                       "argsText updated after controller was closed:",
                       {
@@ -233,6 +233,13 @@ export function useToolInvocations({
               }
 
               if (content.result !== undefined && !lastState.hasResult) {
+                lastToolStates.current[content.toolCallId] = {
+                  hasResult: true,
+                  argsComplete: true,
+                  argsText: lastState.argsText,
+                  controller: lastState.controller,
+                };
+
                 lastState.controller.setResponse(
                   new ToolResponse({
                     result: content.result as ReadonlyJSONValue,
@@ -241,13 +248,6 @@ export function useToolInvocations({
                   }),
                 );
                 lastState.controller.close();
-
-                lastToolStates.current[content.toolCallId] = {
-                  hasResult: true,
-                  argsComplete: true,
-                  argsText: lastState.argsText,
-                  controller: lastState.controller,
-                };
               }
             }
 

package/src/legacy-runtime/runtime-cores/composer/BaseComposerRuntimeCore.tsx

@@ -11,6 +11,7 @@ import type {
   DictationState,
 } from "../core/ComposerRuntimeCore";
 import type { MessageRole, RunConfig } from "../../../types/AssistantTypes";
+import type { QuoteInfo } from "../../../types/QuoteTypes";
 import { BaseSubscribable } from "../remote-thread-list/BaseSubscribable";
 import type { DictationAdapter } from "../adapters/speech/SpeechAdapterTypes";
 
@@ -64,6 +65,19 @@ export abstract class BaseComposerRuntimeCore
     return this._runConfig;
   }
 
+  private _quote: QuoteInfo | undefined = undefined;
+
+  get quote() {
+    return this._quote;
+  }
+
+  public setQuote(quote: QuoteInfo | undefined) {
+    if (this._quote === quote) return;
+
+    this._quote = quote;
+    this._notifySubscribers();
+  }
+
   public setText(value: string) {
     if (this._text === value) return;
 
@@ -112,13 +126,15 @@ export abstract class BaseComposerRuntimeCore
       this._attachments.length === 0 &&
       this._text === "" &&
       this._role === "user" &&
-      Object.keys(this._runConfig).length === 0
+      Object.keys(this._runConfig).length === 0 &&
+      this._quote === undefined
     ) {
       return;
     }
 
     this._role = "user";
     this._runConfig = {};
+    this._quote = undefined;
 
     const task = this._onClearAttachments();
     this._emptyTextAndAttachments();
@@ -151,14 +167,17 @@ export abstract class BaseComposerRuntimeCore
         : [];
 
     const text = this.text;
+    const quote = this._quote;
+    this._quote = undefined;
     this._emptyTextAndAttachments();
+
     const message: Omit<AppendMessage, "parentId" | "sourceId"> = {
       createdAt: new Date(),
       role: this.role,
       content: text ? [{ type: "text", text }] : [],
       attachments: await attachments,
       runConfig: this.runConfig,
-      metadata: { custom: {} },
+      metadata: { custom: { ...(quote ? { quote } : {}) } },
     };
 
     this.handleSend(message);

package/src/legacy-runtime/runtime-cores/core/ComposerRuntimeCore.tsx

@@ -5,6 +5,7 @@ import type {
 } from "../../../types";
 import type { MessageRole, RunConfig } from "../../../types/AssistantTypes";
 import type { DictationAdapter } from "../adapters/speech/SpeechAdapterTypes";
+import type { QuoteInfo } from "../../../types/QuoteTypes";
 
 export type ComposerRuntimeEventType = "send" | "attachmentAdd";
 
@@ -49,6 +50,9 @@ export type ComposerRuntimeCore = Readonly<{
   runConfig: RunConfig;
   setRunConfig: (runConfig: RunConfig) => void;
 
+  quote: QuoteInfo | undefined;
+  setQuote: (quote: QuoteInfo | undefined) => void;
+
   reset: () => Promise<void>;
   clearAttachments: () => Promise<void>;
 

package/src/legacy-runtime/runtime-cores/external-store/createMessageConverter.tsx

@@ -63,8 +63,8 @@ export const createMessageConverter = <T extends object>(
         return undefined;
       });
 
-      const messageMessages = useAuiState<T[]>(({ message }) =>
-        getExternalStoreMessages(message),
+      const messageMessages = useAuiState<T[]>((s) =>
+        getExternalStoreMessages(s.message),
       );
 
       const messages = partMessages ?? messageMessages;

package/src/legacy-runtime/runtime-cores/local/useLocalRuntime.tsx

@@ -27,9 +27,7 @@ const useLocalThreadRuntime = (
   const [runtime] = useState(() => new LocalRuntimeCore(opt, initialMessages));
 
   const threadIdRef = useRef<string | undefined>(undefined);
-  threadIdRef.current = useAuiState(
-    ({ threadListItem }) => threadListItem.remoteId,
-  );
+  threadIdRef.current = useAuiState((s) => s.threadListItem.remoteId);
 
   useEffect(() => {
     runtime.threads

package/src/legacy-runtime/runtime-cores/remote-thread-list/EMPTY_THREAD_CORE.tsx

@@ -129,6 +129,12 @@ export const EMPTY_THREAD_CORE: ThreadRuntimeCore = {
       // noop
     },
 
+    quote: undefined,
+
+    setQuote() {
+      throw EMPTY_THREAD_ERROR;
+    },
+
     subscribe() {
       return () => {};
     },

package/src/primitives/actionBar/ActionBarCopy.tsx

@@ -37,15 +37,16 @@ const useActionBarPrimitiveCopy = ({
   copiedDuration?: number | undefined;
 } = {}) => {
   const aui = useAui();
-  const hasCopyableContent = useAuiState(({ message }) => {
+  const hasCopyableContent = useAuiState((s) => {
     return (
-      (message.role !== "assistant" || message.status?.type !== "running") &&
-      message.parts.some((c) => c.type === "text" && c.text.length > 0)
+      (s.message.role !== "assistant" ||
+        s.message.status?.type !== "running") &&
+      s.message.parts.some((c) => c.type === "text" && c.text.length > 0)
     );
   });
 
-  const isEditing = useAuiState(({ composer }) => composer.isEditing);
-  const composerValue = useAuiState(({ composer }) => composer.text);
+  const isEditing = useAuiState((s) => s.composer.isEditing);
+  const composerValue = useAuiState((s) => s.composer.text);
 
   const callback = useCallback(() => {
     const valueToCopy = isEditing ? composerValue : aui.message().getCopyText();
@@ -89,7 +90,7 @@ export const ActionBarPrimitiveCopy = forwardRef<
   ActionBarPrimitiveCopy.Element,
   ActionBarPrimitiveCopy.Props
 >(({ copiedDuration, onClick, disabled, ...props }, forwardedRef) => {
-  const isCopied = useAuiState(({ message }) => message.isCopied);
+  const isCopied = useAuiState((s) => s.message.isCopied);
   const callback = useActionBarPrimitiveCopy({ copiedDuration });
   return (
     <Primitive.button

package/src/primitives/actionBar/ActionBarEdit.tsx

@@ -31,7 +31,7 @@ import { useAuiState, useAui } from "@assistant-ui/store";
  */
 const useActionBarEdit = () => {
   const aui = useAui();
-  const disabled = useAuiState(({ composer }) => composer.isEditing);
+  const disabled = useAuiState((s) => s.composer.isEditing);
 
   const callback = useCallback(() => {
     aui.composer().beginEdit();

package/src/primitives/actionBar/ActionBarExportMarkdown.tsx

@@ -14,10 +14,11 @@ const useActionBarExportMarkdown = ({
   onExport?: ((content: string) => void | Promise<void>) | undefined;
 } = {}) => {
   const aui = useAui();
-  const hasExportableContent = useAuiState(({ message }) => {
+  const hasExportableContent = useAuiState((s) => {
     return (
-      (message.role !== "assistant" || message.status?.type !== "running") &&
-      message.parts.some((c) => c.type === "text" && c.text.length > 0)
+      (s.message.role !== "assistant" ||
+        s.message.status?.type !== "running") &&
+      s.message.parts.some((c) => c.type === "text" && c.text.length > 0)
     );
   });
 

package/src/primitives/actionBar/ActionBarSpeak.tsx

@@ -14,10 +14,11 @@ const useActionBarSpeak = () => {
     aui.message().speak();
   }, [aui]);
 
-  const hasSpeakableContent = useAuiState(({ message }) => {
+  const hasSpeakableContent = useAuiState((s) => {
     return (
-      (message.role !== "assistant" || message.status?.type !== "running") &&
-      message.parts.some((c) => c.type === "text" && c.text.length > 0)
+      (s.message.role !== "assistant" ||
+        s.message.status?.type !== "running") &&
+      s.message.parts.some((c) => c.type === "text" && c.text.length > 0)
     );
   });
 

package/src/primitives/actionBar/ActionBarStopSpeaking.tsx

@@ -10,7 +10,7 @@ import { useAuiState, useAui } from "@assistant-ui/store";
 
 const useActionBarStopSpeaking = () => {
   const aui = useAui();
-  const isSpeaking = useAuiState(({ message }) => message.speech != null);
+  const isSpeaking = useAuiState((s) => s.message.speech != null);
 
   const callback = useCallback(() => {
     aui.message().stopSpeaking();

package/src/primitives/actionBar/useActionBarFloatStatus.tsx

@@ -19,22 +19,22 @@ export const useActionBarFloatStatus = ({
   autohide,
   autohideFloat,
 }: UseActionBarFloatStatusProps) => {
-  return useAuiState(({ thread, message }) => {
-    if (hideWhenRunning && thread.isRunning) return HideAndFloatStatus.Hidden;
+  return useAuiState((s) => {
+    if (hideWhenRunning && s.thread.isRunning) return HideAndFloatStatus.Hidden;
 
     const autohideEnabled =
-      autohide === "always" || (autohide === "not-last" && !message.isLast);
+      autohide === "always" || (autohide === "not-last" && !s.message.isLast);
 
     // normal status
     if (!autohideEnabled) return HideAndFloatStatus.Normal;
 
     // hidden status
-    if (!message.isHovering) return HideAndFloatStatus.Hidden;
+    if (!s.message.isHovering) return HideAndFloatStatus.Hidden;
 
     // floating status
     if (
       autohideFloat === "always" ||
-      (autohideFloat === "single-branch" && message.branchCount <= 1)
+      (autohideFloat === "single-branch" && s.message.branchCount <= 1)
     )
       return HideAndFloatStatus.Floating;