@assistant-ui/react 0.4.0 → 0.4.1

package/dist/index.js

@@ -1244,7 +1244,8 @@ var import_react36 = require("react");
 // src/utils/hooks/useSmooth.tsx
 var import_react35 = require("react");
 var TextStreamAnimator = class {
-  constructor(setText) {
+  constructor(currentText, setText) {
+    this.currentText = currentText;
     this.setText = setText;
   }
   animationFrameId = null;
@@ -1265,33 +1266,31 @@ var TextStreamAnimator = class {
     const currentTime = Date.now();
     const deltaTime = currentTime - this.lastUpdateTime;
     let timeToConsume = deltaTime;
-    this.setText((currentText) => {
-      const targetText = this.targetText;
-      if (currentText === targetText) {
-        this.animationFrameId = null;
-        return currentText;
-      }
-      const remainingChars = targetText.length - currentText.length;
-      const baseTimePerChar = Math.min(5, 250 / remainingChars);
-      let charsToAdd = 0;
-      while (timeToConsume >= baseTimePerChar && charsToAdd < remainingChars) {
-        charsToAdd++;
-        timeToConsume -= baseTimePerChar;
-      }
+    const remainingChars = this.targetText.length - this.currentText.length;
+    const baseTimePerChar = Math.min(5, 250 / remainingChars);
+    let charsToAdd = 0;
+    while (timeToConsume >= baseTimePerChar && charsToAdd < remainingChars) {
+      charsToAdd++;
+      timeToConsume -= baseTimePerChar;
+    }
+    if (charsToAdd !== remainingChars) {
       this.animationFrameId = requestAnimationFrame(this.animate);
-      if (charsToAdd === 0) {
-        return currentText;
-      }
-      const newText = targetText.slice(0, currentText.length + charsToAdd);
-      this.lastUpdateTime = currentTime - timeToConsume;
-      return newText;
-    });
+    } else {
+      this.animationFrameId = null;
+    }
+    if (charsToAdd === 0) return;
+    this.currentText = this.targetText.slice(
+      0,
+      this.currentText.length + charsToAdd
+    );
+    this.lastUpdateTime = currentTime - timeToConsume;
+    this.setText(this.currentText);
   };
 };
 var useSmooth = (text, smooth = false) => {
   const [displayedText, setDisplayedText] = (0, import_react35.useState)(text);
   const [animatorRef] = (0, import_react35.useState)(
-    new TextStreamAnimator(setDisplayedText)
+    new TextStreamAnimator(text, setDisplayedText)
   );
   (0, import_react35.useEffect)(() => {
     if (!smooth) {
@@ -1300,6 +1299,7 @@ var useSmooth = (text, smooth = false) => {
     }
     if (!text.startsWith(animatorRef.targetText)) {
       setDisplayedText(text);
+      animatorRef.currentText = text;
       animatorRef.targetText = text;
       animatorRef.stop();
       return;