@assistant-ui/react 0.4.0 → 0.4.2

package/dist/index.mjs

@@ -831,21 +831,30 @@ var ActionBarPrimitiveRoot = forwardRef(({ hideWhenRunning, autohide, autohideFl
 ActionBarPrimitiveRoot.displayName = "ActionBarPrimitive.Root";
 
 // src/utils/createActionButton.tsx
-import { composeEventHandlers } from "@radix-ui/primitive";
-import { Primitive as Primitive2 } from "@radix-ui/react-primitive";
 import { forwardRef as forwardRef2 } from "react";
+import { Primitive as Primitive2 } from "@radix-ui/react-primitive";
+import { composeEventHandlers } from "@radix-ui/primitive";
 import { jsx as jsx5 } from "react/jsx-runtime";
-var createActionButton = (displayName, useActionButton) => {
+var createActionButton = (displayName, useActionButton, forwardProps = []) => {
   const ActionButton = forwardRef2((props, forwardedRef) => {
-    const callback = useActionButton(props);
+    const forwardedProps = {};
+    const primitiveProps = {};
+    Object.keys(props).forEach((key) => {
+      if (forwardProps.includes(key)) {
+        forwardedProps[key] = props[key];
+      } else {
+        primitiveProps[key] = props[key];
+      }
+    });
+    const callback = useActionButton(forwardedProps);
     return /* @__PURE__ */ jsx5(
       Primitive2.button,
       {
         type: "button",
         disabled: !callback,
-        ...props,
+        ...primitiveProps,
         ref: forwardedRef,
-        onClick: composeEventHandlers(props.onClick, () => {
+        onClick: composeEventHandlers(primitiveProps.onClick, () => {
           callback?.();
         })
       }
@@ -858,7 +867,8 @@ var createActionButton = (displayName, useActionButton) => {
 // src/primitives/actionBar/ActionBarCopy.tsx
 var ActionBarPrimitiveCopy = createActionButton(
   "ActionBarPrimitive.Copy",
-  useActionBarCopy
+  useActionBarCopy,
+  ["copiedDuration"]
 );
 
 // src/primitives/actionBar/ActionBarReload.tsx
@@ -1171,7 +1181,8 @@ import { forwardRef as forwardRef7 } from "react";
 // src/utils/hooks/useSmooth.tsx
 import { useEffect as useEffect8, useState as useState5 } from "react";
 var TextStreamAnimator = class {
-  constructor(setText) {
+  constructor(currentText, setText) {
+    this.currentText = currentText;
     this.setText = setText;
   }
   animationFrameId = null;
@@ -1192,33 +1203,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] = useState5(text);
   const [animatorRef] = useState5(
-    new TextStreamAnimator(setDisplayedText)
+    new TextStreamAnimator(text, setDisplayedText)
   );
   useEffect8(() => {
     if (!smooth) {
@@ -1227,6 +1236,7 @@ var useSmooth = (text, smooth = false) => {
     }
     if (!text.startsWith(animatorRef.targetText)) {
       setDisplayedText(text);
+      animatorRef.currentText = text;
       animatorRef.targetText = text;
       animatorRef.stop();
       return;
@@ -1885,7 +1895,8 @@ var ThreadPrimitiveScrollToBottom = createActionButton(
 // src/primitives/thread/ThreadSuggestion.tsx
 var ThreadPrimitiveSuggestion = createActionButton(
   "ThreadPrimitive.Suggestion",
-  useThreadSuggestion
+  useThreadSuggestion,
+  ["prompt", "autoSend", "method"]
 );
 
 // src/runtimes/local/useLocalRuntime.tsx