@assistant-ui/react 0.5.24 → 0.5.25

package/dist/index.mjs

@@ -159,11 +159,23 @@ var makeComposerStore = (useThreadMessages, useThreadActions) => {
 
 // src/context/stores/Thread.ts
 import { create as create4 } from "zustand";
+var getThreadStateFromRuntime = (runtime) => {
+  const lastMessage = runtime.messages.at(-1);
+  if (lastMessage?.role !== "assistant")
+    return Object.freeze({
+      isDisabled: runtime.isDisabled,
+      isRunning: false,
+      unstable_canAppendNew: runtime.isDisabled
+    });
+  return Object.freeze({
+    isDisabled: runtime.isDisabled,
+    isRunning: lastMessage.status.type === "running",
+    unstable_canAppendNew: !runtime.isDisabled && lastMessage.status.type !== "running" && lastMessage.status.type !== "requires-action"
+  });
+};
 var makeThreadStore = (runtimeRef) => {
-  return create4(() => ({
-    isDisabled: runtimeRef.getState().isDisabled,
-    isRunning: runtimeRef.getState().isRunning
-  }));
+  const runtime = runtimeRef.getState();
+  return create4(() => getThreadStateFromRuntime(runtime));
 };
 
 // src/context/stores/ThreadViewport.tsx
@@ -260,13 +272,11 @@ var ThreadProvider = ({
     useCallback2(
       (thread) => {
         const onThreadUpdate = () => {
-          const threadState = context.useThread.getState();
-          if (thread.isRunning !== threadState.isRunning || thread.isDisabled !== threadState.isDisabled) {
+          const oldState = context.useThread.getState();
+          const state = getThreadStateFromRuntime(thread);
+          if (oldState.isDisabled !== state.isDisabled || oldState.isRunning !== state.isRunning || oldState.unstable_canAppendNew !== state.unstable_canAppendNew) {
             context.useThread.setState(
-              Object.freeze({
-                isRunning: thread.isRunning,
-                isDisabled: thread.isDisabled
-              }),
+              getThreadStateFromRuntime(thread),
               true
             );
           }
@@ -520,8 +530,8 @@ var useCombinedStore = (stores, selector) => {
   return useCombined(selector);
 };
 
-// src/utils/getMessageText.tsx
-var getMessageText = (message) => {
+// src/utils/getThreadMessageText.tsx
+var getThreadMessageText = (message) => {
   const textParts = message.content.filter(
     (part) => part.type === "text"
   );
@@ -543,7 +553,7 @@ var useActionBarCopy = ({
     const { message } = useMessage.getState();
     const { setIsCopied } = useMessageUtils.getState();
     const { isEditing, value: composerValue } = useEditComposer.getState();
-    const valueToCopy = isEditing ? composerValue : getMessageText(message);
+    const valueToCopy = isEditing ? composerValue : getThreadMessageText(message);
     navigator.clipboard.writeText(valueToCopy).then(() => {
       setIsCopied(true);
       setTimeout(() => setIsCopied(false), copiedDuration);
@@ -670,7 +680,7 @@ var useComposerSend = () => {
   const { useComposer } = useComposerContext();
   const disabled = useCombinedStore(
     [useThread, useComposer],
-    (t, c) => t.isDisabled || t.isRunning || !c.isEditing || c.value.length === 0
+    (t, c) => !t.unstable_canAppendNew || !c.isEditing || c.value.length === 0
   );
   const callback = useCallback11(() => {
     const composerState = useComposer.getState();
@@ -1568,8 +1578,8 @@ var ComposerPrimitiveInput = forwardRef12(
     const handleKeyPress = (e) => {
       if (isDisabled) return;
       if (e.key === "Enter" && e.shiftKey === false) {
-        const isRunning = useThread.getState().isRunning;
-        if (!isRunning) {
+        const { unstable_canAppendNew } = useThread.getState();
+        if (unstable_canAppendNew) {
           e.preventDefault();
           textareaRef.current?.closest("form")?.requestSubmit();
         }
@@ -1902,7 +1912,7 @@ var useMessageContext2 = (messageIndex) => {
           throw new Error(
             "Tried to edit a non-user message. Editing is only supported for user messages. This is likely an internal bug in assistant-ui."
           );
-        const text = getMessageText(message);
+        const text = getThreadMessageText(message);
         return text;
       },
       onSend: (text) => {
@@ -2709,9 +2719,6 @@ var LocalThreadRuntime = class {
   get messages() {
     return this.repository.getMessages();
   }
-  get isRunning() {
-    return this.abortController != null;
-  }
   getBranches(messageId) {
     return this.repository.getBranches(messageId);
   }
@@ -2800,8 +2807,6 @@ var LocalThreadRuntime = class {
         updateMessage({
           status: { type: "complete", reason: "unknown" }
         });
-      } else {
-        this.notifySubscribers();
       }
     } catch (e) {
       this.abortController = null;
@@ -3016,7 +3021,6 @@ var ExternalStoreThreadRuntime = class {
   }
   messages = [];
   isDisabled = false;
-  isRunning = false;
   converter = new ThreadMessageConverter();
   _store;
   constructor(store) {
@@ -3075,7 +3079,6 @@ var ExternalStoreThreadRuntime = class {
     );
     this.messages = this.repository.getMessages();
     this.isDisabled = isDisabled;
-    this.isRunning = isRunning;
     for (const callback of this._subscriptions) callback();
   }
   getBranches(messageId) {