fix(runtimes/external-store): invalidate cache when isRunning changes…

… during autoStatus (#584)

.changeset/chatty-balloons-film.md

@@ -0,0 +1,5 @@
+---
+"@assistant-ui/react": patch
+---
+
+fix(runtimes/external-store): invalidate cache when isRunning changes during autoStatus

packages/react/src/runtimes/external-store/auto-status.tsx

@@ -1,13 +1,8 @@
-import { MessageStatus } from "../../types";
-
 const AUTO_STATUS_RUNNING = Object.freeze({ type: "running" });
 const AUTO_STATUS_COMPLETE = Object.freeze({
   type: "complete",
   reason: "unknown",
 });
 
-export const isAutoStatus = (status: MessageStatus) =>
-  status === AUTO_STATUS_RUNNING || status === AUTO_STATUS_COMPLETE;
-
 export const getAutoStatus = (isLast: boolean, isRunning: boolean) =>
   isLast && isRunning ? AUTO_STATUS_RUNNING : AUTO_STATUS_COMPLETE;

packages/react/src/runtimes/external-store/useExternalStoreSync.tsx

@@ -6,7 +6,7 @@ import {
 } from "./ThreadMessageConverter";
 import { ThreadMessage } from "../../types";
 import { symbolInnerMessage } from "./getExternalStoreMessage";
-import { getAutoStatus, isAutoStatus } from "./auto-status";
+import { getAutoStatus } from "./auto-status";
 import { fromThreadMessageLike } from "./ThreadMessageLike";
 
 type UpdateDataCallback = (
@@ -32,16 +32,9 @@ export const useExternalStoreSync = <T extends WeakKey>(
         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(
         converter(m, idx),
         idx.toString(),