diff --git a/packages/stream_chat_flutter/lib/src/attachment_actions_modal.dart b/packages/stream_chat_flutter/lib/src/attachment_actions_modal.dart
index 7b9f0d51..0c10ae46 100644
--- a/packages/stream_chat_flutter/lib/src/attachment_actions_modal.dart
+++ b/packages/stream_chat_flutter/lib/src/attachment_actions_modal.dart
@@ -394,14 +394,21 @@ class _DownloadProgress {
   int get toPercentage => (received * 100) ~/ total;
 }
 
+/// Class for custom attachment action
 class AttachmentAction {
-  String actionTitle;
-  Widget icon;
-  VoidCallback onTap;
-
+  /// Constructor for custom attachment action
   AttachmentAction({
     required this.actionTitle,
     required this.icon,
     required this.onTap,
   });
+
+  /// Title for the attachment action
+  String actionTitle;
+
+  /// Icon for the attachment action
+  Widget icon;
+
+  /// Callback for when the action is tapped
+  VoidCallback onTap;
 }