Update prompt-input.tsx

Author: haydenbleasel

packages/elements/src/prompt-input.tsx

@@ -357,21 +357,14 @@ export type PromptInputAttachmentsProps = Omit<
 
 export function PromptInputAttachments({
   children,
-  className,
 }: PromptInputAttachmentsProps) {
   const attachments = usePromptInputAttachments();
 
-  if (!attachments.files.length) {
-    return null;
-  }
-
-  return (
-    <div className={cn("flex flex-wrap gap-1 p-2.5", className)}>
-      {attachments.files.map((file) => (
+  return attachments?.files.length > 0
+    ? attachments?.files.map((file) => (
         <Fragment key={file.id}>{children(file)}</Fragment>
-      ))}
-    </div>
-  );
+      ))
+    : null;
 }
 
 export type PromptInputActionAddAttachmentsProps = ComponentProps<