@assistant-ui/react 0.5.9 → 0.5.11

package/dist/index.js

@@ -1284,7 +1284,7 @@ var toContentPartStatus = (message, partIndex, part) => {
   return message.status;
 };
 var EMPTY_CONTENT = Object.freeze({ type: "text", text: "" });
-var syncContentPart = ({ message }, useContentPart, partIndex) => {
+var getContentPartState = ({ message }, useContentPart, partIndex) => {
   let part = message.content[partIndex];
   if (!part) {
     if (message.content.length === 0 && partIndex === 0) {
@@ -1294,29 +1294,32 @@ var syncContentPart = ({ message }, useContentPart, partIndex) => {
     }
   }
   const status = toContentPartStatus(message, partIndex, part);
-  const currentState = useContentPart.getState();
-  if (currentState.part === part && currentState.status === status) return;
-  useContentPart.setState(
-    Object.freeze({
-      part,
-      status
-    })
-  );
+  const currentState = useContentPart?.getState();
+  if (currentState && currentState.part === part && currentState.status === status)
+    return null;
+  return Object.freeze({ part, status });
 };
 var useContentPartContext2 = (partIndex) => {
   const { useMessage } = useMessageContext();
   const [context] = (0, import_react34.useState)(() => {
     const useContentPart = (0, import_zustand10.create)(
-      () => ({})
+      () => getContentPartState(useMessage.getState(), void 0, partIndex)
     );
-    syncContentPart(useMessage.getState(), useContentPart, partIndex);
+    getContentPartState(useMessage.getState(), useContentPart, partIndex);
     return { useContentPart };
   });
   (0, import_react34.useEffect)(() => {
-    syncContentPart(useMessage.getState(), context.useContentPart, partIndex);
-    return useMessage.subscribe((message) => {
-      syncContentPart(message, context.useContentPart, partIndex);
-    });
+    const syncContentPart = (message) => {
+      const newState = getContentPartState(
+        message,
+        context.useContentPart,
+        partIndex
+      );
+      if (!newState) return;
+      context.useContentPart.setState(newState, true);
+    };
+    syncContentPart(useMessage.getState());
+    return useMessage.subscribe(syncContentPart);
   }, [context, useMessage, partIndex]);
   return context;
 };
@@ -1913,16 +1916,15 @@ var import_jsx_runtime23 = require("react/jsx-runtime");
 var getIsLast = (messages, message) => {
   return messages[messages.length - 1]?.id === message.id;
 };
-var syncMessage = (messages, getBranches, useMessage, messageIndex) => {
+var getMessageState = (messages, getBranches, useMessage, messageIndex) => {
   const parentId = messages[messageIndex - 1]?.id ?? null;
   const message = messages[messageIndex];
-  if (!message) return;
   const isLast = getIsLast(messages, message);
   const branches = getBranches(message.id);
-  const currentState = useMessage.getState();
-  if (currentState.message === message && currentState.parentId === parentId && currentState.branches === branches && currentState.isLast === isLast)
-    return;
-  useMessage.setState({
+  const currentState = useMessage?.getState();
+  if (currentState && currentState.message === message && currentState.parentId === parentId && currentState.branches === branches && currentState.isLast === isLast)
+    return null;
+  return Object.freeze({
     message,
     parentId,
     branches,
@@ -1932,7 +1934,14 @@ var syncMessage = (messages, getBranches, useMessage, messageIndex) => {
 var useMessageContext2 = (messageIndex) => {
   const { useThreadMessages, useThreadActions } = useThreadContext();
   const [context] = (0, import_react48.useState)(() => {
-    const useMessage = (0, import_zustand13.create)(() => ({}));
+    const useMessage = (0, import_zustand13.create)(
+      () => getMessageState(
+        useThreadMessages.getState(),
+        useThreadActions.getState().getBranches,
+        void 0,
+        messageIndex
+      )
+    );
     const useMessageUtils = makeMessageUtilsStore();
     const useEditComposer = makeEditComposerStore({
       onEdit: () => {
@@ -1960,23 +1969,24 @@ var useMessageContext2 = (messageIndex) => {
         });
       }
     });
-    syncMessage(
-      useThreadMessages.getState(),
-      useThreadActions.getState().getBranches,
-      useMessage,
-      messageIndex
-    );
     return { useMessage, useMessageUtils, useEditComposer };
   });
   (0, import_react48.useEffect)(() => {
-    return useThreadMessages.subscribe((thread) => {
-      syncMessage(
+    const syncMessage = (thread) => {
+      const newState = getMessageState(
         thread,
         useThreadActions.getState().getBranches,
         context.useMessage,
         messageIndex
       );
-    });
+      if (!newState) return;
+      context.useMessage.setState(
+        newState,
+        true
+      );
+    };
+    syncMessage(useThreadMessages.getState());
+    return useThreadMessages.subscribe(syncMessage);
   }, [useThreadMessages, useThreadActions, context, messageIndex]);
   return context;
 };