Skip to content

Commit d3397d6

Browse files
committed
fix: memoize response from isMessageAIGenerated
1 parent 8fd9e97 commit d3397d6

File tree

1 file changed

+7
-3
lines changed

1 file changed

+7
-3
lines changed

src/components/Message/MessageSimple.tsx

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
import React, { useState } from 'react';
1+
import React, { useMemo, useState } from 'react';
22
import clsx from 'clsx';
33

44
import { 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

Comments
 (0)