dmitry.galkin/chatbot-template
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

refactor: replace isLoading with status (#861)

Browse source
This commit is contained in:
Jeremy 2025-03-11 15:33:18 -07:00 • committed by GitHub
parent 8e561dced4
commit 553a3d825a
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
10 changed files with 72 additions and 109 deletions
Download patch file Download diff file Expand all files Collapse all files

24
components/messages.tsx
View file

@ -1,29 +1,26 @@
import { ChatRequestOptions, Message } from 'ai';
import { Message } from 'ai';
import { PreviewMessage, ThinkingMessage } from './message';
import { useScrollToBottom } from './use-scroll-to-bottom';
import { Overview } from './overview';
import { memo } from 'react';
import { Vote } from '@/lib/db/schema';
import equal from 'fast-deep-equal';
import { UseChatHelpers } from '@ai-sdk/react';
interface MessagesProps {
chatId: string;
isLoading: boolean;
status: UseChatHelpers['status'];
votes: Array<Vote> | undefined;
messages: Array<Message>;
setMessages: (
messages: Message[] | ((messages: Message[]) => Message[]),
) => void;
reload: (
chatRequestOptions?: ChatRequestOptions,
) => Promise<string | null | undefined>;
setMessages: UseChatHelpers['setMessages'];
reload: UseChatHelpers['reload'];
isReadonly: boolean;
isArtifactVisible: boolean;
}
function PureMessages({
chatId,
isLoading,
status,
votes,
messages,
setMessages,
@ -43,10 +40,9 @@ function PureMessages({
{messages.map((message, index) => (
<PreviewMessage
key={message.id}
index={index}
chatId={chatId}
message={message}
isLoading={isLoading && messages.length - 1 === index}
isLoading={status === 'streaming' && messages.length - 1 === index}
vote={
votes
? votes.find((vote) => vote.messageId === message.id)
@ -58,7 +54,7 @@ function PureMessages({
/>
))}
{isLoading &&
{status === 'submitted' &&
messages.length > 0 &&
messages[messages.length - 1].role === 'user' && <ThinkingMessage />}
@ -73,8 +69,8 @@ function PureMessages({
export const Messages = memo(PureMessages, (prevProps, nextProps) => {
if (prevProps.isArtifactVisible && nextProps.isArtifactVisible) return true;
if (prevProps.isLoading !== nextProps.isLoading) return false;
if (prevProps.isLoading && nextProps.isLoading) return false;
if (prevProps.status !== nextProps.status) return false;
if (prevProps.status && nextProps.status) return false;
if (prevProps.messages.length !== nextProps.messages.length) return false;
if (!equal(prevProps.messages, nextProps.messages)) return false;
if (!equal(prevProps.votes, nextProps.votes)) return false;