@assistant-ui/react 0.5.41 → 0.5.42

package/dist/index.mjs

@@ -1072,54 +1072,6 @@ var useEdgeRuntime = ({
 // src/runtimes/local/shouldContinue.tsx
 var shouldContinue = (result) => result.status?.type === "requires-action" && result.status.reason === "tool-calls" && result.content.every((c) => c.type !== "tool-call" || !!c.result);
 
-// src/utils/getThreadMessageText.tsx
-var getThreadMessageText = (message) => {
-  const textParts = message.content.filter(
-    (part) => part.type === "text"
-  );
-  return textParts.map((part) => part.text).join("\n\n");
-};
-
-// src/runtimes/speech/WebSpeechSynthesisAdapter.ts
-var WebSpeechSynthesisAdapter = class {
-  speak(message) {
-    const text = getThreadMessageText(message);
-    const utterance = new SpeechSynthesisUtterance(text);
-    let ended = false;
-    const endHandlers = /* @__PURE__ */ new Set();
-    const handleEnd = () => {
-      if (ended) return;
-      ended = true;
-      endHandlers.forEach((handler) => handler());
-    };
-    utterance.addEventListener("end", handleEnd);
-    utterance.addEventListener("error", handleEnd);
-    window.speechSynthesis.speak(utterance);
-    return {
-      stop: () => {
-        window.speechSynthesis.cancel();
-        handleEnd();
-      },
-      onEnd: (callback) => {
-        if (ended) {
-          let cancelled = false;
-          queueMicrotask(() => {
-            if (!cancelled) callback();
-          });
-          return () => {
-            cancelled = true;
-          };
-        } else {
-          endHandlers.add(callback);
-          return () => {
-            endHandlers.delete(callback);
-          };
-        }
-      }
-    };
-  }
-};
-
 // src/runtimes/local/LocalThreadRuntime.tsx
 var LocalThreadRuntime = class {
   constructor(configProvider, adapter, { initialMessages, ...options }) {
@@ -1158,6 +1110,9 @@ var LocalThreadRuntime = class {
     }
   };
   _options;
+  get options() {
+    return this._options;
+  }
   set options({ initialMessages, ...options }) {
     this._options = options;
     const canSpeak = options.adapters?.speech !== void 0;
@@ -1239,7 +1194,7 @@ var LocalThreadRuntime = class {
       this.repository.addOrUpdateMessage(parentId, message);
       this.notifySubscribers();
     };
-    const maxToolRoundtrips = this._options.maxToolRoundtrips ?? 1;
+    const maxToolRoundtrips = this.options.maxToolRoundtrips ?? 1;
     const toolRoundtrips = message.metadata?.roundtrips?.length ?? 0;
     if (toolRoundtrips > maxToolRoundtrips) {
       updateMessage({
@@ -1334,10 +1289,24 @@ var LocalThreadRuntime = class {
       this.performRoundtrip(parentId, message);
     }
   }
+  // TODO lift utterance state to thread runtime
+  _utterance;
   speak(messageId) {
+    const adapter = this.options.adapters?.speech;
+    if (!adapter) throw new Error("Speech adapter not configured");
     const { message } = this.repository.getMessage(messageId);
-    const adapter = new WebSpeechSynthesisAdapter();
-    return adapter.speak(message);
+    if (this._utterance) {
+      this._utterance.cancel();
+      this._utterance = void 0;
+    }
+    const utterance = adapter.speak(message);
+    utterance.onEnd(() => {
+      if (this._utterance === utterance) {
+        this._utterance = void 0;
+      }
+    });
+    this._utterance = utterance;
+    return this._utterance;
   }
   export() {
     return this.repository.export();
@@ -1497,6 +1466,14 @@ var fromThreadMessageLike = (like, fallbackId, fallbackStatus) => {
   }
 };
 
+// src/utils/getThreadMessageText.tsx
+var getThreadMessageText = (message) => {
+  const textParts = message.content.filter(
+    (part) => part.type === "text"
+  );
+  return textParts.map((part) => part.text).join("\n\n");
+};
+
 // src/runtimes/external-store/ExternalStoreThreadRuntime.tsx
 var hasUpcomingMessage = (isRunning, messages) => {
   return isRunning && messages[messages.length - 1]?.role !== "assistant";
@@ -1743,6 +1720,47 @@ var useDangerousInBrowserRuntime = ({
   return useLocalRuntime(adapter, { initialMessages });
 };
 
+// src/runtimes/speech/WebSpeechSynthesisAdapter.ts
+var WebSpeechSynthesisAdapter = class {
+  speak(message) {
+    const text = getThreadMessageText(message);
+    const utterance = new SpeechSynthesisUtterance(text);
+    const endHandlers = /* @__PURE__ */ new Set();
+    const handleEnd = (reason, error) => {
+      if (res.status.type === "ended") return;
+      res.status = { type: "ended", reason, error };
+      endHandlers.forEach((handler) => handler());
+    };
+    utterance.addEventListener("end", () => handleEnd("finished"));
+    utterance.addEventListener("error", (e) => handleEnd("error", e.error));
+    window.speechSynthesis.speak(utterance);
+    const res = {
+      status: { type: "running" },
+      cancel: () => {
+        window.speechSynthesis.cancel();
+        handleEnd("cancelled");
+      },
+      onEnd: (callback) => {
+        if (res.status.type === "ended") {
+          let cancelled = false;
+          queueMicrotask(() => {
+            if (!cancelled) callback();
+          });
+          return () => {
+            cancelled = true;
+          };
+        } else {
+          endHandlers.add(callback);
+          return () => {
+            endHandlers.delete(callback);
+          };
+        }
+      }
+    };
+    return res;
+  }
+};
+
 // src/context/providers/ThreadProvider.tsx
 import { jsx as jsx6, jsxs as jsxs2 } from "react/jsx-runtime";
 var ThreadProvider = ({
@@ -3335,7 +3353,7 @@ var makeMessageUtilsStore = () => create14((set) => {
     },
     isSpeaking: false,
     stopSpeaking: () => {
-      utterance?.stop();
+      utterance?.cancel();
     },
     addUtterance: (utt) => {
       utterance = utt;
@@ -3537,28 +3555,29 @@ import {
   StopCircleIcon
 } from "lucide-react";
 import { Fragment as Fragment4, jsx as jsx32, jsxs as jsxs5 } from "react/jsx-runtime";
-var useAllowCopy = () => {
+var useAllowCopy = (ensureCapability = false) => {
   const { assistantMessage: { allowCopy = true } = {} } = useThreadConfig();
   const { useThread } = useThreadContext();
   const copySupported = useThread((t) => t.capabilities.unstable_copy);
-  return copySupported && allowCopy;
+  return allowCopy && (!ensureCapability || copySupported);
 };
-var useAllowSpeak = () => {
+var useAllowSpeak = (ensureCapability = false) => {
   const { assistantMessage: { allowSpeak = true } = {} } = useThreadConfig();
   const { useThread } = useThreadContext();
   const speakSupported = useThread((t) => t.capabilities.speak);
-  return speakSupported && allowSpeak;
+  return allowSpeak && (!ensureCapability || speakSupported);
 };
-var useAllowReload = () => {
+var useAllowReload = (ensureCapability = false) => {
   const { assistantMessage: { allowReload = true } = {} } = useThreadConfig();
   const { useThread } = useThreadContext();
   const reloadSupported = useThread((t) => t.capabilities.reload);
-  return reloadSupported && allowReload;
+  return allowReload && (!ensureCapability || reloadSupported);
 };
 var AssistantActionBar = () => {
-  const allowCopy = useAllowCopy();
-  const allowReload = useAllowReload();
-  if (!allowCopy && !allowReload) return null;
+  const allowCopy = useAllowCopy(true);
+  const allowReload = useAllowReload(true);
+  const allowSpeak = useAllowSpeak(true);
+  if (!allowCopy && !allowReload && !allowSpeak) return null;
   return /* @__PURE__ */ jsxs5(
     AssistantActionBarRoot,
     {
@@ -3566,9 +3585,9 @@ var AssistantActionBar = () => {
       autohide: "not-last",
       autohideFloat: "single-branch",
       children: [
-        /* @__PURE__ */ jsx32(AssistantActionBarSpeechControl, {}),
-        /* @__PURE__ */ jsx32(AssistantActionBarCopy, {}),
-        /* @__PURE__ */ jsx32(AssistantActionBarReload, {})
+        allowSpeak && /* @__PURE__ */ jsx32(AssistantActionBarSpeechControl, {}),
+        allowCopy && /* @__PURE__ */ jsx32(AssistantActionBarCopy, {}),
+        allowReload && /* @__PURE__ */ jsx32(AssistantActionBarReload, {})
       ]
     }
   );
@@ -3584,8 +3603,6 @@ var AssistantActionBarCopy = forwardRef20((props, ref) => {
       assistantMessage: { copy: { tooltip = "Copy" } = {} } = {}
     } = {}
   } = useThreadConfig();
-  const allowCopy = useAllowCopy();
-  if (!allowCopy) return null;
   return /* @__PURE__ */ jsx32(actionBar_exports.Copy, { asChild: true, children: /* @__PURE__ */ jsx32(TooltipIconButton, { tooltip, ...props, ref, children: props.children ?? /* @__PURE__ */ jsxs5(Fragment4, { children: [
     /* @__PURE__ */ jsx32(message_exports.If, { copied: true, children: /* @__PURE__ */ jsx32(CheckIcon, {}) }),
     /* @__PURE__ */ jsx32(message_exports.If, { copied: false, children: /* @__PURE__ */ jsx32(CopyIcon, {}) })
@@ -3979,14 +3996,14 @@ import { forwardRef as forwardRef26 } from "react";
 import { forwardRef as forwardRef25 } from "react";
 import { PencilIcon } from "lucide-react";
 import { jsx as jsx40 } from "react/jsx-runtime";
-var useAllowEdit = () => {
+var useAllowEdit = (ensureCapability = false) => {
   const { userMessage: { allowEdit = true } = {} } = useThreadConfig();
   const { useThread } = useThreadContext();
   const editSupported = useThread((t) => t.capabilities.edit);
-  return editSupported && allowEdit;
+  return allowEdit && (!ensureCapability || editSupported);
 };
 var UserActionBar = () => {
-  const allowEdit = useAllowEdit();
+  const allowEdit = useAllowEdit(true);
   if (!allowEdit) return null;
   return /* @__PURE__ */ jsx40(UserActionBarRoot, { hideWhenRunning: true, autohide: "not-last", children: /* @__PURE__ */ jsx40(UserActionBarEdit, {}) });
 };