Skip to content

Commit 86d66d5

Browse files
committed
fix: messages memoization
1 parent 90e1c83 commit 86d66d5

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

components/messages.tsx

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -69,8 +69,11 @@ function PureMessages({
6969
export const Messages = memo(PureMessages, (prevProps, nextProps) => {
7070
if (prevProps.isArtifactVisible && nextProps.isArtifactVisible) return true;
7171

72+
const prevIsLoading = (prevProps.status === 'submitted' || prevProps.status === 'streaming');
73+
const nextIsLoading = (nextProps.status === 'submitted' || nextProps.status === 'streaming');
74+
7275
if (prevProps.status !== nextProps.status) return false;
73-
if (prevProps.status && nextProps.status) return false;
76+
if (prevIsLoading && nextIsLoading) return false;
7477
if (prevProps.messages.length !== nextProps.messages.length) return false;
7578
if (!equal(prevProps.messages, nextProps.messages)) return false;
7679
if (!equal(prevProps.votes, nextProps.votes)) return false;

0 commit comments

Comments
 (0)