@assistant-ui/react 0.5.10 → 0.5.12

package/dist/index.mjs

@@ -1192,7 +1192,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) {
@@ -1202,29 +1202,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] = useState4(() => {
     const useContentPart = create10(
-      () => ({})
+      () => getContentPartState(useMessage.getState(), void 0, partIndex)
     );
-    syncContentPart(useMessage.getState(), useContentPart, partIndex);
+    getContentPartState(useMessage.getState(), useContentPart, partIndex);
     return { useContentPart };
   });
   useEffect6(() => {
-    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;
 };
@@ -1828,16 +1831,15 @@ import { jsx as jsx23 } from "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,
@@ -1847,7 +1849,14 @@ var syncMessage = (messages, getBranches, useMessage, messageIndex) => {
 var useMessageContext2 = (messageIndex) => {
   const { useThreadMessages, useThreadActions } = useThreadContext();
   const [context] = useState6(() => {
-    const useMessage = create13(() => ({}));
+    const useMessage = create13(
+      () => getMessageState(
+        useThreadMessages.getState(),
+        useThreadActions.getState().getBranches,
+        void 0,
+        messageIndex
+      )
+    );
     const useMessageUtils = makeMessageUtilsStore();
     const useEditComposer = makeEditComposerStore({
       onEdit: () => {
@@ -1875,23 +1884,24 @@ var useMessageContext2 = (messageIndex) => {
         });
       }
     });
-    syncMessage(
-      useThreadMessages.getState(),
-      useThreadActions.getState().getBranches,
-      useMessage,
-      messageIndex
-    );
     return { useMessage, useMessageUtils, useEditComposer };
   });
   useEffect10(() => {
-    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;
 };
@@ -3572,7 +3582,7 @@ var getExternalStoreMessage = (message) => {
 };
 
 // src/runtimes/external-store/useExternalStoreSync.tsx
-import { useEffect as useEffect11, useMemo as useMemo3 } from "react";
+import { useEffect as useEffect11, useInsertionEffect as useInsertionEffect4, useMemo as useMemo3, useRef as useRef6 } from "react";
 
 // src/runtimes/external-store/ThreadMessageConverter.ts
 var ThreadMessageConverter = class {
@@ -3588,13 +3598,104 @@ var ThreadMessageConverter = class {
   }
 };
 
+// src/runtimes/external-store/auto-status.tsx
+var AUTO_STATUS_RUNNING = Object.freeze({ type: "running" });
+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
+var fromThreadMessageLike = (like, fallbackId, fallbackStatus) => {
+  const { role, content, id, createdAt, status } = like;
+  const common = {
+    id: id ?? fallbackId,
+    createdAt: createdAt ?? /* @__PURE__ */ new Date()
+  };
+  switch (role) {
+    case "assistant":
+      return {
+        ...common,
+        role,
+        content: content.map((part) => {
+          const type = part.type;
+          switch (type) {
+            case "text":
+            case "ui":
+              return part;
+            case "tool-call": {
+              if ("argsText" in part) return part;
+              return {
+                ...part,
+                argsText: JSON.stringify(part.args)
+              };
+            }
+            default: {
+              const unhandledType = type;
+              throw new Error(`Unknown content part type: ${unhandledType}`);
+            }
+          }
+        }),
+        status: status ?? fallbackStatus
+      };
+    case "user":
+      return {
+        ...common,
+        role,
+        content: content.map((part) => {
+          const type = part.type;
+          switch (type) {
+            case "text":
+            case "ui":
+            case "image":
+              return part;
+            default: {
+              const unhandledType = type;
+              throw new Error(`Unknown content part type: ${unhandledType}`);
+            }
+          }
+        })
+      };
+    case "system":
+      if (content.length !== 1 || content[0].type !== "text")
+        throw new Error(
+          "System messages must have exactly one text content part."
+        );
+      return {
+        ...common,
+        role,
+        content
+      };
+    default: {
+      const unsupportedRole = role;
+      throw new Error(`Unknown message role: ${unsupportedRole}`);
+    }
+  }
+};
+
 // src/runtimes/external-store/useExternalStoreSync.tsx
 var useExternalStoreSync = (adapter, updateData) => {
+  const adapterRef = useRef6(adapter);
+  useInsertionEffect4(() => {
+    adapterRef.current = adapter;
+  });
   const [converter, convertCallback] = useMemo3(() => {
     const converter2 = adapter.convertMessage ?? ((m) => m);
     const convertCallback2 = (cache, m, idx) => {
+      const autoStatus = getAutoStatus(
+        adapterRef.current.messages.at(-1) === m,
+        adapterRef.current.isRunning ?? false
+      );
+      if (cache && (cache.role !== "assistant" || !isAutoStatus(cache.status) || cache.status.type === autoStatus.type))
+        return cache;
       if (cache) return cache;
-      const newMessage = converter2(m, idx);
+      const newMessage = fromThreadMessageLike(
+        converter2(m, idx),
+        idx.toString(),
+        autoStatus
+      );
       newMessage[symbolInnerMessage] = m;
       return newMessage;
     };
@@ -3760,10 +3861,10 @@ var ExternalStoreRuntime = class extends BaseAssistantRuntime {
 };
 
 // src/runtimes/external-store/useExternalStoreRuntime.tsx
-import { useEffect as useEffect12, useInsertionEffect as useInsertionEffect4, useState as useState9 } from "react";
+import { useEffect as useEffect12, useInsertionEffect as useInsertionEffect5, useState as useState9 } from "react";
 var useExternalStoreRuntime = (store) => {
   const [runtime] = useState9(() => new ExternalStoreRuntime(store));
-  useInsertionEffect4(() => {
+  useInsertionEffect5(() => {
     runtime.store = store;
   });
   useEffect12(() => {