whai/cherry-studio
1
1
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

feat: added conditional rendering to messagetokens component

Browse Source 
- 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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/renderer/src/pages/home/Messages/Message.tsx
View File

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

4
src/renderer/src/pages/home/Messages/MessageTokens.tsx
View File

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