@athenaintel/react 0.23.2 → 0.24.0

package/dist/chat/AthenaChat.cjs

@@ -1,23 +1,6 @@
 "use strict";
 Object.defineProperty(exports, Symbol.toStringTag, { value: "Module" });
 const jsxRuntime = require("react/jsx-runtime");
-require("../packages/agent-runtime/src/generated/runconfig-contract.cjs");
-const react$1 = require("react");
-require("@assistant-ui/tap");
-require("statewire");
-require("../packages/agent-runtime/src/statewire-suspended.cjs");
-require("../packages/agent-runtime/src/conversation-list.cjs");
-require("@tanstack/react-query");
-require("../packages/agent-runtime/src/statewire-state.cjs");
-require("../packages/agent-runtime/src/messages.cjs");
-require("@assistant-ui/store");
-require("../packages/agent-runtime/src/quota-safe-local-storage-session.cjs");
-require("../packages/agent-runtime/src/statewire-attach-observer.cjs");
-require("../packages/agent-runtime/src/thread-status.cjs");
-require("@assistant-ui/react-statewire");
-const githubWakeEvent = require("../packages/agent-runtime/src/github-wake-event.cjs");
-require("@assistant-ui/core");
-require("../packages/agent-runtime/src/thread-handoff.cjs");
 const tiptapText = require("./tiptap-text.cjs");
 const AthenaChatErrorBoundary = require("./AthenaChatErrorBoundary.cjs");
 const StatewireApprovalCard = require("./StatewireApprovalCard.cjs");
@@ -39,6 +22,13 @@ const chatThreadShell = require("../chat-ui/thread/chat-thread-shell.cjs");
 const statewireHistoryEdge = require("../chat-ui/thread/statewire-history-edge.cjs");
 const threadLoadingOverlay = require("../chat-ui/thread/thread-loading-overlay.cjs");
 const composerSendOrStop = require("../chat-ui/thread/composer-send-or-stop.cjs");
+const composerAction = require("../chat-ui/composer/composer-action.cjs");
+const athenaDiagnostics = require("../diagnostics/athena-diagnostics.cjs");
+const useDictation = require("../dictation/use-dictation.cjs");
+const DictationStrip = require("../dictation/DictationStrip.cjs");
+const dictationControls = require("../dictation/dictation-controls.cjs");
+const dictationState = require("../dictation/dictation-state.cjs");
+const useAthenaDictationEngine = require("../dictation/use-athena-dictation-engine.cjs");
 const tiptapComposer = require("../chat-ui/composer/tiptap-composer.cjs");
 const apiClient = require("../chat-ui/mentions/suggestions/data/api-client.cjs");
 const messageComponents = require("../chat-ui/messages/message-components.cjs");
@@ -50,6 +40,7 @@ const composerKeyboardHint = require("./composer-keyboard-hint.cjs");
 const usePrefersReducedMotion = require("../lib/use-prefers-reduced-motion.cjs");
 const react = require("@assistant-ui/react");
 const lucideReact = require("lucide-react");
+const react$1 = require("react");
 const threadIdContext = require("../threads/thread-id-context.cjs");
 const FileUploadButton = require("../composer/FileUploadButton.cjs");
 const ComposerQuotePreview = require("../composer/ComposerQuotePreview.cjs");
@@ -66,6 +57,7 @@ const index = require("../toolkits/index.cjs");
 const runActivity = require("./run-activity.cjs");
 const messageTime = require("./message-time.cjs");
 const threadScrollToBottom = require("./thread-scroll-to-bottom.cjs");
+const githubWakeEvent = require("../packages/agent-runtime/src/github-wake-event.cjs");
 const cnfast = require("cnfast");
 function QuotePostMessageBridge() {
   QuoteContext.useQuoteFromPostMessage();
@@ -144,7 +136,9 @@ const AthenaChat = ({
   mentionTools,
   welcomeSuggestions = athenaChatWelcome.DEFAULT_WELCOME_SUGGESTIONS,
   components,
-  groupToolCalls = true
+  groupToolCalls = true,
+  composer: composerSlots,
+  dictation = true
 }) => {
   var _a, _b, _c;
   const athenaConfig = AthenaContext.useAthenaConfig();
@@ -252,6 +246,8 @@ const AthenaChat = ({
             AthenaChatComposer,
             {
               tools,
+              slots: composerSlots,
+              dictation,
               welcomeSuggestionCount: welcomeSuggestions.length
             }
           ) })
@@ -310,7 +306,7 @@ const ATHENA_CHAT_MENTION_CATEGORIES = [
   "projects",
   "folders"
 ];
-const AthenaChatComposer = ({ tools, welcomeSuggestionCount }) => {
+const AthenaChatComposer = ({ tools, slots, dictation, welcomeSuggestionCount }) => {
   const aui = react.useAui();
   const threadId = threadIdContext.useAthenaStatewireThreadId();
   const { appUrl } = AthenaContext.useAthenaConfig();
@@ -319,6 +315,10 @@ const AthenaChatComposer = ({ tools, welcomeSuggestionCount }) => {
   const { attachFiles } = useAttachFiles.useAttachFiles();
   const composerRef = react$1.useRef(null);
   const [hasText, setHasText] = react$1.useState(false);
+  const isRunning = react.useAuiState((state) => {
+    var _a;
+    return ((_a = state.optional.thread) == null ? void 0 : _a.isRunning) ?? false;
+  });
   const [dragActive, setDragActive] = react$1.useState(false);
   const keyboardHint = composerKeyboardHint.useComposerKeyboardHint();
   const reducedMotion = usePrefersReducedMotion.usePrefersReducedMotion();
@@ -328,6 +328,29 @@ const AthenaChatComposer = ({ tools, welcomeSuggestionCount }) => {
     athenaChatWelcome.welcomeComposerDelayMs(welcomeSuggestionCount),
     reducedMotion || !mountedEmpty
   );
+  const dictationEnabled = dictation !== false;
+  const dictationOptions = typeof dictation === "object" ? dictation : void 0;
+  const appendSpokenText = react$1.useCallback(
+    (spoken) => {
+      if (!spoken) return;
+      aui.composer.setText(dictationState.composeDictationDraft(aui.composer.getState().text ?? "", spoken));
+    },
+    [aui]
+  );
+  const dictationEngine = useAthenaDictationEngine.useAthenaDictationEngine(dictationOptions == null ? void 0 : dictationOptions.engine);
+  const dictationCtl = useDictation.useDictation({
+    engine: dictationEngine,
+    lang: dictationOptions == null ? void 0 : dictationOptions.lang,
+    // A recognizer that gives up on its own (silence window) still hands
+    // over what it heard instead of dropping the words with the session.
+    onAutoEnd: appendSpokenText
+  });
+  const recording = dictationEnabled && dictationCtl.isActive;
+  const showMic = dictationEnabled && dictationCtl.isSupported;
+  const { isActive: dictationActive, cancel: cancelDictation } = dictationCtl;
+  react$1.useEffect(() => {
+    if (!dictationEnabled && dictationActive) cancelDictation();
+  }, [cancelDictation, dictationActive, dictationEnabled]);
   const rootCategories = react$1.useMemo(() => {
     const categories = [...ATHENA_CHAT_MENTION_CATEGORIES];
     if (tools.some((tool) => tool.type === "toolkit")) categories.push("toolkits");
@@ -383,6 +406,40 @@ const AthenaChatComposer = ({ tools, welcomeSuggestionCount }) => {
     }),
     [attachFiles]
   );
+  const stopDictation = react$1.useCallback(async () => {
+    appendSpokenText((await dictationCtl.stop()).text);
+  }, [appendSpokenText, dictationCtl]);
+  const sendDictation = react$1.useCallback(async () => {
+    var _a;
+    const { text: spoken, error } = await dictationCtl.stop();
+    const text = dictationState.composeDictationDraft(aui.composer.getState().text ?? "", spoken);
+    if (error) {
+      aui.composer.setText(text);
+      return;
+    }
+    const extras = attachmentsRef.current.length > 0 || quoteRef.current !== null;
+    if (!text && !extras) return;
+    aui.composer.setText(text);
+    const action = composerAction.resolveComposerAction({ hasDraft: true, isRunning });
+    if (action === "stop") return;
+    try {
+      await ((_a = slots == null ? void 0 : slots.onBeforeSend) == null ? void 0 : _a.call(slots, { action }));
+    } catch (error2) {
+      athenaDiagnostics.athenaDiagnostics.emit("composer.dictation", {
+        level: "warn",
+        data: {
+          phase: "before_send_rejected",
+          error: error2 instanceof Error ? error2.message : String(error2)
+        }
+      });
+      return;
+    }
+    if (extras) {
+      sendComposed(text);
+      return;
+    }
+    aui.composer.send({ steer: false });
+  }, [aui, dictationCtl, isRunning, sendComposed, slots]);
   return /* @__PURE__ */ jsxRuntime.jsx(ComposerDropZone.ComposerDropZone, { onDragActiveChange: setDragActive, children: /* @__PURE__ */ jsxRuntime.jsx(
     "div",
     {
@@ -399,23 +456,35 @@ const AthenaChatComposer = ({ tools, welcomeSuggestionCount }) => {
             /* @__PURE__ */ jsxRuntime.jsx(ComposerQuotePreview.ComposerQuotePreview, {}),
             /* @__PURE__ */ jsxRuntime.jsx(ComposerAttachmentPreview.ComposerAttachmentPreview, {})
           ] }),
-          editor: /* @__PURE__ */ jsxRuntime.jsx(
-            tiptapComposer.TiptapComposer,
-            {
-              ref: composerRef,
-              threadId,
-              appearance: "paper",
-              placeholder: "Type @ to tag an asset",
-              tools,
-              rootCategories,
-              onDraftStateChange: setHasText,
-              waitForHostUploads,
-              interceptSubmit,
-              fileIntake,
-              onSubmitted: keyboardHint.onSubmitted
-            }
-          ),
-          toolbarStart: /* @__PURE__ */ jsxRuntime.jsxs("div", { className: "flex items-center gap-1", children: [
+          editor: /* @__PURE__ */ jsxRuntime.jsxs(jsxRuntime.Fragment, { children: [
+            /* @__PURE__ */ jsxRuntime.jsx("div", { className: cnfast.cn(recording && "hidden"), children: /* @__PURE__ */ jsxRuntime.jsx(
+              tiptapComposer.TiptapComposer,
+              {
+                ref: composerRef,
+                threadId,
+                appearance: "paper",
+                placeholder: "Type @ to tag an asset",
+                tools,
+                rootCategories,
+                onDraftStateChange: setHasText,
+                waitForHostUploads,
+                interceptSubmit,
+                fileIntake,
+                onSubmitted: keyboardHint.onSubmitted
+              }
+            ) }),
+            recording && /* @__PURE__ */ jsxRuntime.jsx(
+              DictationStrip.DictationStrip,
+              {
+                status: dictationCtl.status,
+                transcript: dictationCtl.transcript,
+                stream: dictationCtl.stream,
+                stoppingLabel: dictationCtl.engine.stoppingLabel
+              }
+            ),
+            dictationEnabled && dictationCtl.status === dictationState.DICTATION_STATUSES.error && dictationCtl.error && /* @__PURE__ */ jsxRuntime.jsx("p", { className: "aui-dictation-error px-1.5 pb-1 text-destructive text-xs", role: "alert", children: dictationCtl.error.message })
+          ] }),
+          toolbarStart: /* @__PURE__ */ jsxRuntime.jsx("div", { className: "flex items-center gap-1", children: recording ? /* @__PURE__ */ jsxRuntime.jsx(dictationControls.DictationCancelButton, { onClick: dictationCtl.cancel }) : /* @__PURE__ */ jsxRuntime.jsxs(jsxRuntime.Fragment, { children: [
             /* @__PURE__ */ jsxRuntime.jsx(FileUploadButton.FileUploadButton, { className: "aui-composer-attach size-8 rounded-full p-0 disabled:opacity-30" }),
             /* @__PURE__ */ jsxRuntime.jsx(
               athenaComposerControls.AthenaComposerMentionButton,
@@ -425,17 +494,40 @@ const AthenaChatComposer = ({ tools, welcomeSuggestionCount }) => {
                   return (_b = (_a = composerRef.current) == null ? void 0 : _a.openMentionMenu) == null ? void 0 : _b.call(_a);
                 }
               }
+            ),
+            slots == null ? void 0 : slots.toolbarStart
+          ] }) }),
+          toolbarEnd: /* @__PURE__ */ jsxRuntime.jsx("div", { className: "flex items-center gap-1", children: recording ? /* @__PURE__ */ jsxRuntime.jsxs(jsxRuntime.Fragment, { children: [
+            /* @__PURE__ */ jsxRuntime.jsx(
+              dictationControls.DictationStopButton,
+              {
+                onClick: () => void stopDictation(),
+                disabled: dictationCtl.status === dictationState.DICTATION_STATUSES.stopping
+              }
+            ),
+            /* @__PURE__ */ jsxRuntime.jsx(
+              athenaComposerControls.AthenaComposerSendButton,
+              {
+                action: isRunning ? "queue" : "send",
+                onSend: () => void sendDictation(),
+                disabled: dictationCtl.status !== dictationState.DICTATION_STATUSES.listening || isUploading,
+                title: isUploading ? "Upload in progress..." : void 0
+              }
             )
-          ] }),
-          toolbarEnd: /* @__PURE__ */ jsxRuntime.jsx("div", { className: "flex items-center gap-1", children: /* @__PURE__ */ jsxRuntime.jsx(
-            composerSendOrStop.ComposerSendOrStop,
-            {
-              composerRef,
-              hasDraft: hasText || hasExtras,
-              sendBlockedReason: isUploading ? "Upload in progress..." : null,
-              components: athenaComposerControls.ATHENA_COMPOSER_COMPONENTS
-            }
-          ) }),
+          ] }) : /* @__PURE__ */ jsxRuntime.jsxs(jsxRuntime.Fragment, { children: [
+            slots == null ? void 0 : slots.toolbarEnd,
+            showMic && /* @__PURE__ */ jsxRuntime.jsx(dictationControls.DictationMicButton, { onClick: dictationCtl.start, disabled: isUploading }),
+            /* @__PURE__ */ jsxRuntime.jsx(
+              composerSendOrStop.ComposerSendOrStop,
+              {
+                composerRef,
+                hasDraft: hasText || hasExtras,
+                sendBlockedReason: isUploading ? "Upload in progress..." : null,
+                onBeforeSend: slots == null ? void 0 : slots.onBeforeSend,
+                components: athenaComposerControls.ATHENA_COMPOSER_COMPONENTS
+              }
+            )
+          ] }) }),
           belowSurface: /* @__PURE__ */ jsxRuntime.jsx(composerKeyboardHint.ComposerKeyboardHint, { visible: keyboardHint.visible })
         }
       )