@assistant-ui/react 0.5.14 → 0.5.16

package/dist/index.mjs

@@ -209,6 +209,7 @@ var makeComposerStore = (useThreadMessages, useThreadActions) => {
 import { create as create4 } from "zustand";
 var makeThreadStore = (runtimeRef) => {
   return create4(() => ({
+    isDisabled: runtimeRef.getState().isDisabled,
     isRunning: runtimeRef.getState().isRunning
   }));
 };
@@ -307,10 +308,12 @@ var ThreadProvider = ({
     useCallback2(
       (thread) => {
         const onThreadUpdate = () => {
-          if (thread.isRunning !== context.useThread.getState().isRunning) {
+          const threadState = context.useThread.getState();
+          if (thread.isRunning !== threadState.isRunning || thread.isDisabled !== threadState.isDisabled) {
             context.useThread.setState(
               Object.freeze({
-                isRunning: thread.isRunning
+                isRunning: thread.isRunning,
+                isDisabled: thread.isDisabled
               }),
               true
             );
@@ -621,7 +624,7 @@ var useActionBarReload = () => {
   const { useMessage } = useMessageContext();
   const disabled = useCombinedStore(
     [useThread, useMessage],
-    (t, m) => t.isRunning || m.message.role !== "assistant"
+    (t, m) => t.isRunning || t.isDisabled || m.message.role !== "assistant"
   );
   const callback = useCallback7(() => {
     const { parentId } = useMessage.getState();
@@ -707,9 +710,16 @@ var useComposerIf = (props) => {
 // src/primitive-hooks/composer/useComposerSend.tsx
 import { useCallback as useCallback11 } from "react";
 var useComposerSend = () => {
-  const { useViewport, useComposer: useNewComposer } = useThreadContext();
+  const {
+    useThread,
+    useViewport,
+    useComposer: useNewComposer
+  } = useThreadContext();
   const { useComposer } = useComposerContext();
-  const disabled = useComposer((c) => !c.isEditing || c.value.length === 0);
+  const disabled = useCombinedStore(
+    [useThread, useComposer],
+    (t, c) => t.isDisabled || t.isRunning || !c.isEditing || c.value.length === 0
+  );
   const callback = useCallback11(() => {
     const composerState = useComposer.getState();
     if (!composerState.isEditing) return;
@@ -788,6 +798,8 @@ var useThreadIf = (props) => {
       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;
     }
   );
@@ -819,7 +831,7 @@ var useThreadSuggestion = ({
 }) => {
   const { useThread, useComposer } = useThreadContext();
   const append = useAppendMessage();
-  const disabled = useThread((t) => t.isRunning);
+  const disabled = useThread((t) => t.isDisabled);
   const callback = useCallback13(() => {
     const thread = useThread.getState();
     const composer = useComposer.getState();
@@ -1512,7 +1524,14 @@ import TextareaAutosize from "react-textarea-autosize";
 import { useEscapeKeydown } from "@radix-ui/react-use-escape-keydown";
 import { jsx as jsx19 } from "react/jsx-runtime";
 var ComposerPrimitiveInput = forwardRef11(
-  ({ autoFocus = false, asChild, disabled, onChange, onKeyDown, ...rest }, forwardedRef) => {
+  ({
+    autoFocus = false,
+    asChild,
+    disabled: disabledProp,
+    onChange,
+    onKeyDown,
+    ...rest
+  }, forwardedRef) => {
     const { useThread } = useThreadContext();
     const { useComposer, type } = useComposerContext();
     const value = useComposer((c) => {
@@ -1520,6 +1539,7 @@ var ComposerPrimitiveInput = forwardRef11(
       return c.value;
     });
     const Component = asChild ? Slot : TextareaAutosize;
+    const isDisabled = useThread((t) => t.isDisabled) ?? disabledProp ?? false;
     const textareaRef = useRef4(null);
     const ref = useComposedRefs2(forwardedRef, textareaRef);
     useEscapeKeydown((e) => {
@@ -1530,7 +1550,7 @@ var ComposerPrimitiveInput = forwardRef11(
       }
     });
     const handleKeyPress = (e) => {
-      if (disabled) return;
+      if (isDisabled) return;
       if (e.key === "Enter" && e.shiftKey === false) {
         const isRunning = useThread.getState().isRunning;
         if (!isRunning) {
@@ -1539,7 +1559,7 @@ var ComposerPrimitiveInput = forwardRef11(
         }
       }
     };
-    const autoFocusEnabled = autoFocus && !disabled;
+    const autoFocusEnabled = autoFocus && !isDisabled;
     const focus = useCallback15(() => {
       const textarea = textareaRef.current;
       if (!textarea || !autoFocusEnabled) return;
@@ -1562,7 +1582,7 @@ var ComposerPrimitiveInput = forwardRef11(
         value,
         ...rest,
         ref,
-        disabled,
+        disabled: isDisabled,
         onChange: composeEventHandlers5(onChange, (e) => {
           const composerState = useComposer.getState();
           if (!composerState.isEditing) return;
@@ -3392,6 +3412,7 @@ var LocalThreadRuntime = class {
   abortController = null;
   repository = new MessageRepository();
   capabilities = CAPABILITIES;
+  isDisabled = false;
   get messages() {
     return this.repository.getMessages();
   }
@@ -3605,7 +3626,6 @@ var AUTO_STATUS_COMPLETE = Object.freeze({
   type: "complete",
   reason: "unknown"
 });
-var isAutoStatus = (status) => status === AUTO_STATUS_RUNNING || status === AUTO_STATUS_COMPLETE;
 var getAutoStatus = (isLast, isRunning) => isLast && isRunning ? AUTO_STATUS_RUNNING : AUTO_STATUS_COMPLETE;
 
 // src/runtimes/external-store/ThreadMessageLike.tsx
@@ -3689,9 +3709,8 @@ var useExternalStoreSync = (adapter, updateData) => {
         adapterRef.current.messages.at(-1) === m,
         adapterRef.current.isRunning ?? false
       );
-      if (cache && (cache.role !== "assistant" || !isAutoStatus(cache.status) || cache.status.type === autoStatus.type))
+      if (cache && (cache.role !== "assistant" || cache.status === autoStatus))
         return cache;
-      if (cache) return cache;
       const newMessage = fromThreadMessageLike(
         converter2(m, idx),
         idx.toString(),
@@ -3704,6 +3723,7 @@ var useExternalStoreSync = (adapter, updateData) => {
   }, [adapter.convertMessage]);
   useEffect11(() => {
     updateData(
+      adapter.isDisabled ?? false,
       adapter.isRunning ?? false,
       converter.convertMessages(adapter.messages, convertCallback)
     );
@@ -3711,8 +3731,9 @@ var useExternalStoreSync = (adapter, updateData) => {
     updateData,
     converter,
     convertCallback,
-    adapter.messages,
-    adapter.isRunning
+    adapter.isDisabled,
+    adapter.isRunning,
+    adapter.messages
   ]);
 };
 
@@ -3723,6 +3744,9 @@ var hasUpcomingMessage = (isRunning, messages) => {
 var ExternalStoreThreadRuntime = class {
   constructor(store) {
     this.store = store;
+    this.isDisabled = store.isDisabled ?? false;
+    this.isRunning = store.isRunning ?? false;
+    this.messages = store.messages;
     this.useStore = create14(() => ({
       store
     }));
@@ -3740,8 +3764,9 @@ var ExternalStoreThreadRuntime = class {
       copy: this.store.onCopy !== null
     };
   }
-  messages = [];
-  isRunning = false;
+  messages;
+  isDisabled;
+  isRunning;
   getBranches(messageId) {
     return this.repository.getBranches(messageId);
   }
@@ -3792,7 +3817,7 @@ var ExternalStoreThreadRuntime = class {
       this.useStore.setState({ store: this.store });
     }
   }
-  updateData = (isRunning, vm) => {
+  updateData = (isDisabled, isRunning, vm) => {
     for (let i = 0; i < vm.length; i++) {
       const message = vm[i];
       const parent = vm[i - 1];
@@ -3815,6 +3840,7 @@ var ExternalStoreThreadRuntime = class {
       this.assistantOptimisticId ?? vm.at(-1)?.id ?? null
     );
     this.messages = this.repository.getMessages();
+    this.isDisabled = isDisabled;
     this.isRunning = isRunning;
     for (const callback of this._subscriptions) callback();
   };