chore: move edited check upstream

Author: isekovanic

src/components/Message/MessageSimple.tsx

@@ -217,12 +217,10 @@ const MessageSimpleWithContext = <
                 </span>
               )}
               <MessageTimestamp customClass='str-chat__message-simple-timestamp' />
-              {isEdited && !message.ai_generated && (
+              {isEdited && (
                 <span className='str-chat__mesage-simple-edited'>{t<string>('Edited')}</span>
               )}
-              {isEdited && !message.ai_generated && (
-                <MessageEditedTimestamp calendar open={isEditedTimestampOpen} />
-              )}
+              {isEdited && <MessageEditedTimestamp calendar open={isEditedTimestampOpen} />}
             </div>
           )}
         </div>

src/components/Message/utils.tsx

@@ -494,5 +494,6 @@ export const isMessageBounced = <
 export const isMessageEdited = <
   StreamChatGenerics extends DefaultStreamChatGenerics = DefaultStreamChatGenerics
 >(
-  message: Pick<StreamMessage<StreamChatGenerics>, 'message_text_updated_at'>,
-) => !!message.message_text_updated_at;
+  message: Pick<StreamMessage<StreamChatGenerics>, 'message_text_updated_at'> &
+    Partial<Pick<StreamMessage<StreamChatGenerics>, 'ai_generated'>>,
+) => !!message.message_text_updated_at && !message.ai_generated;