@assistant-ui/react 0.12.11 → 0.12.12

package/src/primitives/message/MessageAttachments.tsx

@@ -1,115 +1,6 @@
 "use client";
 
-import type { CompleteAttachment } from "@assistant-ui/core";
-import { ComponentType, type FC, memo, useMemo } from "react";
-import { useAuiState } from "@assistant-ui/store";
-import { MessageAttachmentByIndexProvider } from "../../context/providers";
-
-export namespace MessagePrimitiveAttachments {
-  export type Props = {
-    components:
-      | {
-          Image?: ComponentType | undefined;
-          Document?: ComponentType | undefined;
-          File?: ComponentType | undefined;
-          Attachment?: ComponentType | undefined;
-        }
-      | undefined;
-  };
-}
-
-const getComponent = (
-  components: MessagePrimitiveAttachments.Props["components"],
-  attachment: CompleteAttachment,
-) => {
-  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: never = type;
-      throw new Error(`Unknown attachment type: ${_exhaustiveCheck}`);
-  }
-};
-
-const AttachmentComponent: FC<{
-  components: MessagePrimitiveAttachments.Props["components"];
-}> = ({ components }) => {
-  const attachment = useAuiState((s) => s.attachment);
-  if (!attachment) return null;
-
-  const Component = getComponent(components, attachment as CompleteAttachment);
-  if (!Component) return null;
-  return <Component />;
-};
-
-export namespace MessagePrimitiveAttachmentByIndex {
-  export type Props = {
-    index: number;
-    components?: MessagePrimitiveAttachments.Props["components"];
-  };
-}
-
-/**
- * Renders a single attachment at the specified index within the current message.
- *
- * This component provides direct access to render a specific attachment
- * from the message's attachment list using the provided component configuration.
- *
- * @example
- * ```tsx
- * <MessagePrimitive.AttachmentByIndex
- *   index={0}
- *   components={{
- *     Image: MyImageAttachment,
- *     Document: MyDocumentAttachment
- *   }}
- * />
- * ```
- */
-export const MessagePrimitiveAttachmentByIndex: FC<MessagePrimitiveAttachmentByIndex.Props> =
-  memo(
-    ({ index, components }) => {
-      return (
-        <MessageAttachmentByIndexProvider index={index}>
-          <AttachmentComponent components={components} />
-        </MessageAttachmentByIndexProvider>
-      );
-    },
-    (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";
-
-export const MessagePrimitiveAttachments: FC<
-  MessagePrimitiveAttachments.Props
-> = ({ components }) => {
-  const attachmentsCount = useAuiState((s) => {
-    if (s.message.role !== "user") return 0;
-    return s.message.attachments.length;
-  });
-
-  const attachmentElements = useMemo(() => {
-    return Array.from({ length: attachmentsCount }, (_, index) => (
-      <MessagePrimitiveAttachmentByIndex
-        key={index}
-        index={index}
-        components={components}
-      />
-    ));
-  }, [attachmentsCount, components]);
-
-  return attachmentElements;
-};
-
-MessagePrimitiveAttachments.displayName = "MessagePrimitive.Attachments";
+export {
+  MessagePrimitiveAttachments,
+  MessagePrimitiveAttachmentByIndex,
+} from "@assistant-ui/core/react";

package/src/primitives/message/MessageIf.ts

@@ -78,7 +78,7 @@ export namespace MessagePrimitiveIf {
 }
 
 /**
- * @deprecated Use `<AuiIf condition={({ message }) => ...} />` instead.
+ * @deprecated Use `<AuiIf condition={(s) => s.message...} />` instead.
  */
 export const MessagePrimitiveIf: FC<MessagePrimitiveIf.Props> = ({
   children,