@assistant-ui/react 0.5.89 → 0.5.91

package/dist/index.mjs

@@ -10,7 +10,7 @@ import {
   toLanguageModelMessages,
   toLanguageModelTools,
   toolResultStream
-} from "./chunk-CKNRD26B.mjs";
+} from "./chunk-C57XMNDQ.mjs";
 import {
   __export
 } from "./chunk-BJPOCE4O.mjs";
@@ -2326,6 +2326,7 @@ var defaultComponents = {
     /* @__PURE__ */ jsx23(ContentPartPrimitiveInProgress, { children: /* @__PURE__ */ jsx23("span", { style: { fontFamily: "revert" }, children: " \u25CF" }) })
   ] }),
   Image: () => /* @__PURE__ */ jsx23(ContentPartPrimitiveImage, {}),
+  Unstable_Audio: () => null,
   UI: () => /* @__PURE__ */ jsx23(ContentPartPrimitiveDisplay, {})
 };
 var MessageContentPartComponent = ({
@@ -2334,33 +2335,32 @@ var MessageContentPartComponent = ({
     Empty,
     Image: Image2 = defaultComponents.Image,
     UI = defaultComponents.UI,
+    Unstable_Audio: Audio = defaultComponents.Unstable_Audio,
     tools: { by_name = {}, Fallback: Fallback2 = void 0 } = {}
   } = {}
 }) => {
   const contentPartRuntime = useContentPartRuntime();
   const part = useContentPart();
   const type = part.type;
+  if (type === "tool-call") {
+    const Tool = by_name[part.toolName] || Fallback2;
+    const addResult = (result) => contentPartRuntime.addToolResult(result);
+    return /* @__PURE__ */ jsx23(ToolUIDisplay, { ...part, part, UI: Tool, addResult });
+  }
+  if (part.status.type === "requires-action")
+    throw new Error("Encountered unexpected requires-action status");
   switch (type) {
     case "text":
-      if (part.status.type === "requires-action")
-        throw new Error("Encountered unexpected requires-action status");
       if (part.part === EMPTY_CONTENT && !!Empty) {
         return /* @__PURE__ */ jsx23(Empty, { status: part.status });
       }
       return /* @__PURE__ */ jsx23(Text2, { ...part, part });
     case "image":
-      if (part.status.type === "requires-action")
-        throw new Error("Encountered unexpected requires-action status");
       return /* @__PURE__ */ jsx23(Image2, { ...part, part });
+    case "audio":
+      return /* @__PURE__ */ jsx23(Audio, { ...part, part });
     case "ui":
-      if (part.status.type === "requires-action")
-        throw new Error("Encountered unexpected requires-action status");
       return /* @__PURE__ */ jsx23(UI, { ...part, part });
-    case "tool-call": {
-      const Tool = by_name[part.toolName] || Fallback2;
-      const addResult = (result) => contentPartRuntime.addToolResult(result);
-      return /* @__PURE__ */ jsx23(ToolUIDisplay, { ...part, part, UI: Tool, addResult });
-    }
     default:
       const unhandledType = type;
       throw new Error(`Unknown content part type: ${unhandledType}`);
@@ -2379,7 +2379,7 @@ var MessageContentPartImpl = ({
 };
 var MessageContentPart = memo2(
   MessageContentPartImpl,
-  (prev, next) => prev.partIndex === next.partIndex && prev.components?.Text === next.components?.Text && prev.components?.Image === next.components?.Image && prev.components?.UI === next.components?.UI && prev.components?.tools === next.components?.tools
+  (prev, next) => prev.partIndex === next.partIndex && prev.components?.Text === next.components?.Text && prev.components?.Image === next.components?.Image && prev.components?.Unstable_Audio === next.components?.Unstable_Audio && prev.components?.UI === next.components?.UI && prev.components?.tools === next.components?.tools
 );
 var MessagePrimitiveContent = ({
   components
@@ -5004,6 +5004,7 @@ var fromThreadMessageLike = (like, fallbackId, fallbackStatus) => {
             case "text":
             case "ui":
             case "image":
+            case "audio":
               return part;
             default: {
               const unhandledType = type;
@@ -6036,6 +6037,8 @@ DialogContent.displayName = DialogPrimitive.Content.displayName;
 
 // src/ui/attachment.tsx
 import { AvatarFallback as AvatarFallback2 } from "@radix-ui/react-avatar";
+import { useShallow } from "zustand/shallow";
+import { DialogTitle } from "@radix-ui/react-dialog";
 import { jsx as jsx46, jsxs as jsxs9 } from "react/jsx-runtime";
 var AttachmentRoot = withDefaults(attachment_exports.Root, {
   className: "aui-attachment-root"
@@ -6057,13 +6060,15 @@ var useFileSrc = (file) => {
   return src;
 };
 var useAttachmentSrc = () => {
-  const { file, src } = useAttachment((a) => {
-    if (a.type !== "image") return {};
-    if (a.file) return { file: a.file };
-    const src2 = a.content?.filter((c) => c.type === "image")[0]?.image;
-    if (!src2) return {};
-    return { src: src2 };
-  });
+  const { file, src } = useAttachment(
+    useShallow((a) => {
+      if (a.type !== "image") return {};
+      if (a.file) return { file: a.file };
+      const src2 = a.content?.filter((c) => c.type === "image")[0]?.image;
+      if (!src2) return {};
+      return { src: src2 };
+    })
+  );
   return useFileSrc(file) ?? src;
 };
 var AttachmentPreview = ({ src }) => {
@@ -6093,7 +6098,10 @@ var AttachmentPreviewDialog = ({ children }) => {
   if (!src) return children;
   return /* @__PURE__ */ jsxs9(Dialog, { children: [
     /* @__PURE__ */ jsx46(DialogTrigger, { className: "aui-attachment-preview-trigger", asChild: true, children }),
-    /* @__PURE__ */ jsx46(DialogContent, { children: /* @__PURE__ */ jsx46(AttachmentPreview, { src }) })
+    /* @__PURE__ */ jsxs9(DialogContent, { children: [
+      /* @__PURE__ */ jsx46(DialogTitle, { className: "aui-sr-only", children: "Image Attachment Preview" }),
+      /* @__PURE__ */ jsx46(AttachmentPreview, { src })
+    ] })
   ] });
 };
 var AttachmentThumb = () => {