@assistant-ui/react 0.5.57 → 0.5.59

package/dist/index.d.mts

@@ -615,6 +615,7 @@ declare const AssistantRuntimeProvider: react.NamedExoticComponent<PropsWithChil
 
 type TextContentPartProviderProps = {
     text: string;
+    isRunning?: boolean | undefined;
 };
 declare const TextContentPartProvider: FC<PropsWithChildren<TextContentPartProviderProps>>;
 

package/dist/index.d.ts

@@ -615,6 +615,7 @@ declare const AssistantRuntimeProvider: react.NamedExoticComponent<PropsWithChil
 
 type TextContentPartProviderProps = {
     text: string;
+    isRunning?: boolean | undefined;
 };
 declare const TextContentPartProvider: FC<PropsWithChildren<TextContentPartProviderProps>>;
 

package/dist/index.js

@@ -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 = _react.useRef.call(void 0, id);
   const [displayedText, setDisplayedText] = _react.useState.call(void 0, text);
   const smoothStatusStore = useSmoothStatusStore({ optional: true });
@@ -2348,16 +2351,37 @@ var AssistantRuntimeProvider = _react.memo.call(void 0, AssistantRuntimeProvider
 
 
 
-var TextContentPartProvider = ({ children, text }) => {
+var COMPLETE_STATUS = {
+  type: "complete"
+};
+var RUNNING_STATUS = {
+  type: "running"
+};
+var TextContentPartProvider = ({ children, text, isRunning }) => {
   const [context] = _react.useState.call(void 0, () => {
     const useContentPart2 = _zustand.create.call(void 0, () => ({
-      status: { type: "complete" },
+      status: isRunning ? RUNNING_STATUS : COMPLETE_STATUS,
       part: { type: "text", text }
     }));
     return {
       useContentPart: useContentPart2
     };
   });
+  _react.useEffect.call(void 0, () => {
+    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__ */ _jsxruntime.jsx.call(void 0, ContentPartContext.Provider, { value: context, children });
 };
 
@@ -3284,21 +3308,21 @@ MessagePrimitiveIf.displayName = "MessagePrimitive.If";
 
 
 
-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;
 };