1- import React , { useState } from 'react' ;
1+ import React , { useMemo , useState } from 'react' ;
22import clsx from 'clsx' ;
33
44import { MessageErrorIcon } from './icons' ;
@@ -89,6 +89,10 @@ const MessageSimpleWithContext = <
8989
9090 const hasAttachment = messageHasAttachments ( message ) ;
9191 const hasReactions = messageHasReactions ( message ) ;
92+ const isAIGenerated = useMemo ( ( ) => isMessageAIGenerated ?.( message ) , [
93+ isMessageAIGenerated ,
94+ message ,
95+ ] ) ;
9296
9397 if ( message . customType === CUSTOM_MESSAGE_TYPE . date ) {
9498 return null ;
@@ -102,7 +106,7 @@ const MessageSimpleWithContext = <
102106 const showReplyCountButton = ! threadList && ! ! message . reply_count ;
103107 const allowRetry = message . status === 'failed' && message . errorStatusCode !== 403 ;
104108 const isBounced = isMessageBounced ( message ) ;
105- const isEdited = isMessageEdited ( message ) && ! isMessageAIGenerated ?. ( message ) ;
109+ const isEdited = isMessageEdited ( message ) && ! isAIGenerated ;
106110
107111 let handleClick : ( ( ) => void ) | undefined = undefined ;
108112
@@ -188,7 +192,7 @@ const MessageSimpleWithContext = <
188192 { message . attachments ?. length && ! message . quoted_message ? (
189193 < Attachment actionHandler = { handleAction } attachments = { message . attachments } />
190194 ) : null }
191- { isMessageAIGenerated ?. ( message ) ? (
195+ { isAIGenerated ? (
192196 < StreamedMessageText message = { message } renderText = { renderText } />
193197 ) : (
194198 < MessageText message = { message } renderText = { renderText } />
0 commit comments