@assistant-ui/react 0.5.56 → 0.5.58

package/dist/index.mjs

@@ -1839,8 +1839,8 @@ var ExternalStoreRuntime = class extends BaseAssistantRuntime {
     if (!this.store.onSwitchToNewThread)
       throw new Error("Runtime does not support switching to new threads.");
     this.thread = new ExternalStoreThreadRuntime({
-      messages: [],
-      onNew: this.store.onNew
+      ...this.store,
+      messages: []
     });
     await this.store.onSwitchToNewThread();
   }
@@ -1849,8 +1849,9 @@ var ExternalStoreRuntime = class extends BaseAssistantRuntime {
       if (!this.store.onSwitchToThread)
         throw new Error("Runtime does not support switching threads.");
       this.thread = new ExternalStoreThreadRuntime({
-        messages: [],
-        onNew: this.store.onNew
+        ...this.store,
+        messages: []
+        // ignore messages until rerender
       });
       this.store.onSwitchToThread(threadId);
     } else {
@@ -2344,19 +2345,40 @@ var AssistantRuntimeProviderImpl = ({ children, runtime }) => {
 var AssistantRuntimeProvider = memo(AssistantRuntimeProviderImpl);
 
 // src/context/providers/TextContentPartProvider.tsx
-import { useState as useState9 } from "react";
+import { useEffect as useEffect5, useState as useState9 } from "react";
 import { create as create11 } from "zustand";
 import { jsx as jsx9 } from "react/jsx-runtime";
-var TextContentPartProvider = ({ children, text }) => {
+var COMPLETE_STATUS = {
+  type: "complete"
+};
+var RUNNING_STATUS = {
+  type: "running"
+};
+var TextContentPartProvider = ({ children, text, isRunning }) => {
   const [context] = useState9(() => {
     const useContentPart2 = create11(() => ({
-      status: { type: "complete" },
+      status: isRunning ? RUNNING_STATUS : COMPLETE_STATUS,
       part: { type: "text", text }
     }));
     return {
       useContentPart: useContentPart2
     };
   });
+  useEffect5(() => {
+    const state = context.useContentPart.getState();
+    const textUpdated = state.part.text !== text;
+    const targetTextPart = textUpdated ? { type: "text", text } : state.part;
+    const targetStatus = isRunning ? RUNNING_STATUS : COMPLETE_STATUS;
+    const statusUpdated = state.status !== targetStatus;
+    if (!textUpdated && !statusUpdated) return;
+    writableStore(context.useContentPart).setState(
+      {
+        part: targetTextPart,
+        status: targetStatus
+      },
+      true
+    );
+  }, [context, isRunning, text]);
   return /* @__PURE__ */ jsx9(ContentPartContext.Provider, { value: context, children });
 };
 
@@ -2451,11 +2473,11 @@ var useSwitchToNewThread = () => {
 };
 
 // src/model-config/useAssistantTool.tsx
-import { useEffect as useEffect5 } from "react";
+import { useEffect as useEffect6 } from "react";
 var useAssistantTool = (tool) => {
   const assistantActionsStore = useAssistantActionsStore();
   const toolUIsStore = useToolUIsStore();
-  useEffect5(() => {
+  useEffect6(() => {
     const { toolName, render, ...rest } = tool;
     const config = {
       tools: {
@@ -2484,10 +2506,10 @@ var makeAssistantTool = (tool) => {
 };
 
 // src/model-config/useAssistantToolUI.tsx
-import { useEffect as useEffect6 } from "react";
+import { useEffect as useEffect7 } from "react";
 var useAssistantToolUI = (tool) => {
   const toolUIsStore = useToolUIsStore();
-  useEffect6(() => {
+  useEffect7(() => {
     if (!tool) return;
     const { toolName, render } = tool;
     return toolUIsStore.getState().setToolUI(toolName, render);
@@ -2505,10 +2527,10 @@ var makeAssistantToolUI = (tool) => {
 };
 
 // src/model-config/useAssistantInstructions.tsx
-import { useEffect as useEffect7 } from "react";
+import { useEffect as useEffect8 } from "react";
 var useAssistantInstructions = (instruction) => {
   const actionsStore = useAssistantActionsStore();
-  useEffect7(() => {
+  useEffect8(() => {
     const config = {
       system: instruction
     };
@@ -3047,11 +3069,11 @@ import { composeEventHandlers as composeEventHandlers3 } from "@radix-ui/primiti
 
 // src/utils/hooks/useOnComposerFocus.tsx
 import { useCallbackRef as useCallbackRef2 } from "@radix-ui/react-use-callback-ref";
-import { useEffect as useEffect8 } from "react";
+import { useEffect as useEffect9 } from "react";
 var useOnComposerFocus = (callback) => {
   const callbackRef = useCallbackRef2(callback);
   const threadComposerStore = useThreadComposerStore();
-  useEffect8(() => {
+  useEffect9(() => {
     return threadComposerStore.getState().onFocus(() => {
       callbackRef();
     });
@@ -3280,24 +3302,24 @@ MessagePrimitiveIf.displayName = "MessagePrimitive.If";
 import { memo as memo2 } from "react";
 
 // src/context/providers/ContentPartProvider.tsx
-import { useEffect as useEffect9, useState as useState11 } from "react";
+import { useEffect as useEffect10, useState as useState11 } from "react";
 import { create as create12 } from "zustand";
 import { jsx as jsx20 } from "react/jsx-runtime";
-var COMPLETE_STATUS = {
+var COMPLETE_STATUS2 = {
   type: "complete"
 };
 var toContentPartStatus = (message, partIndex, part) => {
-  if (message.role !== "assistant") return COMPLETE_STATUS;
+  if (message.role !== "assistant") return COMPLETE_STATUS2;
   const isLastPart = partIndex === Math.max(0, message.content.length - 1);
   if (part.type !== "tool-call") {
     if ("reason" in message.status && message.status.reason === "tool-calls" && isLastPart)
       throw new Error(
         "Encountered unexpected requires-action status. This is likely an internal bug in assistant-ui."
       );
-    return isLastPart ? message.status : COMPLETE_STATUS;
+    return isLastPart ? message.status : COMPLETE_STATUS2;
   }
   if (!!part.result) {
-    return COMPLETE_STATUS;
+    return COMPLETE_STATUS2;
   }
   return message.status;
 };
@@ -3327,7 +3349,7 @@ var useContentPartContext2 = (partIndex) => {
     );
     return { useContentPart: useContentPart2 };
   });
-  useEffect9(() => {
+  useEffect10(() => {
     const syncContentPart = (message) => {
       const newState = getContentPartState(
         message,
@@ -3534,7 +3556,7 @@ var {
 } = createContextStoreHook(useMessageAttachmentContext, "useAttachment");
 
 // src/context/providers/MessageAttachmentProvider.tsx
-import { useEffect as useEffect10, useState as useState12 } from "react";
+import { useEffect as useEffect11, useState as useState12 } from "react";
 import { create as create13 } from "zustand";
 import { jsx as jsx24 } from "react/jsx-runtime";
 var getAttachment = ({ message }, useAttachment2, partIndex) => {
@@ -3556,7 +3578,7 @@ var useMessageAttachmentContext2 = (partIndex) => {
       return { type: "message", useAttachment: useAttachment2 };
     }
   );
-  useEffect10(() => {
+  useEffect11(() => {
     const syncAttachment = (messageState) => {
       const newState = getAttachment(
         messageState,
@@ -3676,7 +3698,7 @@ import { Slot } from "@radix-ui/react-slot";
 import {
   forwardRef as forwardRef16,
   useCallback as useCallback17,
-  useEffect as useEffect11,
+  useEffect as useEffect12,
   useRef as useRef4
 } from "react";
 import TextareaAutosize from "react-textarea-autosize";
@@ -3729,7 +3751,7 @@ var ComposerPrimitiveInput = forwardRef16(
         textareaRef.current.value.length
       );
     }, [autoFocusEnabled]);
-    useEffect11(() => focus(), [focus]);
+    useEffect12(() => focus(), [focus]);
     useOnComposerFocus(() => {
       if (composerStore.getState().type === "thread") {
         focus();
@@ -3777,7 +3799,7 @@ var ComposerPrimitiveAddAttachment = createActionButton(
 import { memo as memo4 } from "react";
 
 // src/context/providers/ComposerAttachmentProvider.tsx
-import { useEffect as useEffect12, useState as useState13 } from "react";
+import { useEffect as useEffect13, useState as useState13 } from "react";
 import { create as create14 } from "zustand";
 import { jsx as jsx29 } from "react/jsx-runtime";
 var getAttachment2 = ({ attachments }, useAttachment2, partIndex) => {
@@ -3797,7 +3819,7 @@ var useComposerAttachmentContext2 = (partIndex) => {
       return { type: "composer", useAttachment: useAttachment2 };
     }
   );
-  useEffect12(() => {
+  useEffect13(() => {
     const syncAttachment = (composer) => {
       const newState = getAttachment2(
         composer,
@@ -3970,11 +3992,11 @@ var useOnResizeContent = (callback) => {
 
 // src/utils/hooks/useOnScrollToBottom.tsx
 import { useCallbackRef as useCallbackRef4 } from "@radix-ui/react-use-callback-ref";
-import { useEffect as useEffect13 } from "react";
+import { useEffect as useEffect14 } from "react";
 var useOnScrollToBottom = (callback) => {
   const callbackRef = useCallbackRef4(callback);
   const threadViewportStore = useThreadViewportStore();
-  useEffect13(() => {
+  useEffect14(() => {
     return threadViewportStore.getState().onScrollToBottom(() => {
       callbackRef();
     });
@@ -4047,7 +4069,7 @@ ThreadPrimitiveViewport.displayName = "ThreadPrimitive.Viewport";
 import { memo as memo5 } from "react";
 
 // src/context/providers/MessageProvider.tsx
-import { useEffect as useEffect14, useState as useState14 } from "react";
+import { useEffect as useEffect15, useState as useState14 } from "react";
 import { create as create17 } from "zustand";
 
 // src/context/stores/EditComposer.ts
@@ -4173,7 +4195,7 @@ var useMessageContext2 = (messageIndex) => {
     });
     return { useMessage: useMessage2, useMessageUtils: useMessageUtils2, useEditComposer: useEditComposer2 };
   });
-  useEffect14(() => {
+  useEffect15(() => {
     const syncMessage = (thread) => {
       const newState = getMessageState(
         thread,