feat: added conditional rendering to messagetokens component

- Added conditional rendering to MessageTokens component.
- Added parameter 'isLastMessage' to MessageTokens component to determine conditional rendering based on message position.
This commit is contained in:
kangfenmao 2024-10-15 20:22:01 +08:00
parent 3cc8cfb43b
commit 14808649f8
2 changed files with 3 additions and 3 deletions

View File

@ -74,7 +74,7 @@ const MessageItem: FC<Props> = ({ message, index, lastMessage, showMenu = true,
<MessageContent message={message} model={model} /> <MessageContent message={message} model={model} />
{!lastMessage && showMenu && ( {!lastMessage && showMenu && (
<MessageFooter style={{ border: messageBorder, flexDirection: isLastMessage ? 'row-reverse' : undefined }}> <MessageFooter style={{ border: messageBorder, flexDirection: isLastMessage ? 'row-reverse' : undefined }}>
<MessgeTokens message={message} /> <MessgeTokens message={message} isLastMessage={isLastMessage} />
<MessageMenubar <MessageMenubar
message={message} message={message}
model={model} model={model}

View File

@ -3,7 +3,7 @@ import { EVENT_NAMES, EventEmitter } from '@renderer/services/event'
import { Message } from '@renderer/types' import { Message } from '@renderer/types'
import styled from 'styled-components' import styled from 'styled-components'
const MessgeTokens: React.FC<{ message: Message }> = ({ message }) => { const MessgeTokens: React.FC<{ message: Message; isLastMessage: boolean }> = ({ message, isLastMessage }) => {
const { generating } = useRuntime() const { generating } = useRuntime()
const locateMessage = () => { const locateMessage = () => {
@ -18,7 +18,7 @@ const MessgeTokens: React.FC<{ message: Message }> = ({ message }) => {
return <MessageMetadata onClick={locateMessage}>Tokens: {message?.usage?.total_tokens}</MessageMetadata> return <MessageMetadata onClick={locateMessage}>Tokens: {message?.usage?.total_tokens}</MessageMetadata>
} }
if (generating) { if (isLastMessage && generating) {
return null return null
} }