@assistant-ui/react 0.5.57 → 0.5.59

package/dist/index.mjs

@@ -774,7 +774,10 @@ var useSmooth = (state, smooth = false) => {
   const {
     part: { text }
   } = state;
-  const id = useMessage((m) => m.message.id);
+  const id = useMessage({
+    optional: true,
+    selector: (m) => m.message.id
+  });
   const idRef = useRef(id);
   const [displayedText, setDisplayedText] = useState2(text);
   const smoothStatusStore = useSmoothStatusStore({ optional: true });
@@ -2345,19 +2348,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 });
 };
 
@@ -2452,11 +2476,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: {
@@ -2485,10 +2509,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);
@@ -2506,10 +2530,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
     };
@@ -3048,11 +3072,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();
     });
@@ -3281,24 +3305,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;
 };
@@ -3328,7 +3352,7 @@ var useContentPartContext2 = (partIndex) => {
     );
     return { useContentPart: useContentPart2 };
   });
-  useEffect9(() => {
+  useEffect10(() => {
     const syncContentPart = (message) => {
       const newState = getContentPartState(
         message,
@@ -3535,7 +3559,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) => {
@@ -3557,7 +3581,7 @@ var useMessageAttachmentContext2 = (partIndex) => {
       return { type: "message", useAttachment: useAttachment2 };
     }
   );
-  useEffect10(() => {
+  useEffect11(() => {
     const syncAttachment = (messageState) => {
       const newState = getAttachment(
         messageState,
@@ -3677,7 +3701,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";
@@ -3730,7 +3754,7 @@ var ComposerPrimitiveInput = forwardRef16(
         textareaRef.current.value.length
       );
     }, [autoFocusEnabled]);
-    useEffect11(() => focus(), [focus]);
+    useEffect12(() => focus(), [focus]);
     useOnComposerFocus(() => {
       if (composerStore.getState().type === "thread") {
         focus();
@@ -3778,7 +3802,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) => {
@@ -3798,7 +3822,7 @@ var useComposerAttachmentContext2 = (partIndex) => {
       return { type: "composer", useAttachment: useAttachment2 };
     }
   );
-  useEffect12(() => {
+  useEffect13(() => {
     const syncAttachment = (composer) => {
       const newState = getAttachment2(
         composer,
@@ -3971,11 +3995,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();
     });
@@ -4048,7 +4072,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
@@ -4174,7 +4198,7 @@ var useMessageContext2 = (messageIndex) => {
     });
     return { useMessage: useMessage2, useMessageUtils: useMessageUtils2, useEditComposer: useEditComposer2 };
   });
-  useEffect14(() => {
+  useEffect15(() => {
     const syncMessage = (thread) => {
       const newState = getMessageState(
         thread,