diff --git a/web/screens/Thread/ThreadCenterPanel/ChatItem/index.tsx b/web/screens/Thread/ThreadCenterPanel/ChatItem/index.tsx index 10d408211..1fa3ef115 100644 --- a/web/screens/Thread/ThreadCenterPanel/ChatItem/index.tsx +++ b/web/screens/Thread/ThreadCenterPanel/ChatItem/index.tsx @@ -54,7 +54,12 @@ const ChatItem = forwardRef((message, ref) => { <> {status !== MessageStatus.Error && content?.length > 0 && (
- +
)} {errorMessage && !message.loadModelError && ( diff --git a/web/screens/Thread/ThreadCenterPanel/TextMessage/index.tsx b/web/screens/Thread/ThreadCenterPanel/TextMessage/index.tsx index fa39f8209..f65b9b967 100644 --- a/web/screens/Thread/ThreadCenterPanel/TextMessage/index.tsx +++ b/web/screens/Thread/ThreadCenterPanel/TextMessage/index.tsx @@ -19,19 +19,18 @@ import { MarkdownTextMessage } from './MarkdownTextMessage' import { editMessageAtom, - getCurrentChatMessagesAtom, tokenSpeedAtom, } from '@/helpers/atoms/ChatMessage.atom' import { activeThreadAtom } from '@/helpers/atoms/Thread.atom' -const MessageContainer: React.FC = (props) => { +const MessageContainer: React.FC< + ThreadMessage & { isCurrentMessage: boolean } +> = (props) => { const isUser = props.role === ChatCompletionRole.User const isSystem = props.role === ChatCompletionRole.System const editMessage = useAtomValue(editMessageAtom) const activeThread = useAtomValue(activeThreadAtom) - const tokenSpeed = useAtomValue(tokenSpeedAtom) - const messages = useAtomValue(getCurrentChatMessagesAtom) const text = useMemo( () => props.content[0]?.text?.value ?? '', @@ -81,16 +80,6 @@ const MessageContainer: React.FC = (props) => {

{displayDate(props.created)}

-
- -
{tokenSpeed && tokenSpeed.message === props.id && tokenSpeed.tokenSpeed > 0 && ( @@ -100,39 +89,52 @@ const MessageContainer: React.FC = (props) => { )} -
- <> - {messageType === ContentType.Image && ( - +
+
+ > + +
+
+ <> + {messageType === ContentType.Image && ( + + )} + {messageType === ContentType.Pdf && ( + + )} - {editMessage === props.id ? ( -
- -
- ) : ( -
- -
- )} - + {editMessage === props.id ? ( +
+ +
+ ) : ( +
+ +
+ )} + +
)