@assistant-ui/react
Version:
TypeScript/React library for AI Chat
60 lines (59 loc) • 2.31 kB
JavaScript
"use client";
// src/primitives/message/MessageAttachments.tsx
import { memo, useMemo } from "react";
import {
useAssistantState,
MessageAttachmentByIndexProvider
} from "../../context/index.js";
import { jsx } from "react/jsx-runtime";
var getComponent = (components, attachment) => {
const type = attachment.type;
switch (type) {
case "image":
return components?.Image ?? components?.Attachment;
case "document":
return components?.Document ?? components?.Attachment;
case "file":
return components?.File ?? components?.Attachment;
default:
const _exhaustiveCheck = type;
throw new Error(`Unknown attachment type: ${_exhaustiveCheck}`);
}
};
var AttachmentComponent = ({ components }) => {
const attachment = useAssistantState(({ attachment: attachment2 }) => attachment2);
if (!attachment) return null;
const Component = getComponent(components, attachment);
if (!Component) return null;
return /* @__PURE__ */ jsx(Component, {});
};
var MessagePrimitiveAttachmentByIndex = memo(
({ index, components }) => {
return /* @__PURE__ */ jsx(MessageAttachmentByIndexProvider, { index, children: /* @__PURE__ */ jsx(AttachmentComponent, { components }) });
},
(prev, next) => prev.index === next.index && prev.components?.Image === next.components?.Image && prev.components?.Document === next.components?.Document && prev.components?.File === next.components?.File && prev.components?.Attachment === next.components?.Attachment
);
MessagePrimitiveAttachmentByIndex.displayName = "MessagePrimitive.AttachmentByIndex";
var MessagePrimitiveAttachments = ({ components }) => {
const attachmentsCount = useAssistantState(({ message }) => {
if (message.role !== "user") return 0;
return message.attachments.length;
});
const attachmentElements = useMemo(() => {
return Array.from({ length: attachmentsCount }, (_, index) => /* @__PURE__ */ jsx(
MessagePrimitiveAttachmentByIndex,
{
index,
components
},
index
));
}, [attachmentsCount, components]);
return attachmentElements;
};
MessagePrimitiveAttachments.displayName = "MessagePrimitive.Attachments";
export {
MessagePrimitiveAttachmentByIndex,
MessagePrimitiveAttachments
};
//# sourceMappingURL=MessageAttachments.js.map