@assistant-ui/react 0.5.11 → 0.5.14

package/dist/index.mjs

@@ -1198,7 +1198,7 @@ var getContentPartState = ({ message }, useContentPart, partIndex) => {
     if (message.content.length === 0 && partIndex === 0) {
       part = EMPTY_CONTENT;
     } else {
-      return;
+      return null;
     }
   }
   const status = toContentPartStatus(message, partIndex, part);
@@ -1834,6 +1834,7 @@ var getIsLast = (messages, message) => {
 var getMessageState = (messages, getBranches, useMessage, messageIndex) => {
   const parentId = messages[messageIndex - 1]?.id ?? null;
   const message = messages[messageIndex];
+  if (!message) return null;
   const isLast = getIsLast(messages, message);
   const branches = getBranches(message.id);
   const currentState = useMessage?.getState();
@@ -3582,7 +3583,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 {
@@ -3598,13 +3599,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;
     };
@@ -3770,10 +3862,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(() => {