mirror of
https://github.com/CherryHQ/cherry-studio.git
synced 2026-01-08 14:29:15 +08:00
feat(MessageHeader): add agent name display for assistant messages in agent view
This commit is contained in:
parent
1ef6de1869
commit
90264f6ec9
@ -4,9 +4,11 @@ import UserPopup from '@renderer/components/Popups/UserPopup'
|
|||||||
import { APP_NAME, AppLogo, isLocalAi } from '@renderer/config/env'
|
import { APP_NAME, AppLogo, isLocalAi } from '@renderer/config/env'
|
||||||
import { getModelLogo } from '@renderer/config/models'
|
import { getModelLogo } from '@renderer/config/models'
|
||||||
import { useTheme } from '@renderer/context/ThemeProvider'
|
import { useTheme } from '@renderer/context/ThemeProvider'
|
||||||
|
import { useAgent } from '@renderer/hooks/agents/useAgent'
|
||||||
import useAvatar from '@renderer/hooks/useAvatar'
|
import useAvatar from '@renderer/hooks/useAvatar'
|
||||||
import { useChatContext } from '@renderer/hooks/useChatContext'
|
import { useChatContext } from '@renderer/hooks/useChatContext'
|
||||||
import { useMinappPopup } from '@renderer/hooks/useMinappPopup'
|
import { useMinappPopup } from '@renderer/hooks/useMinappPopup'
|
||||||
|
import { useRuntime } from '@renderer/hooks/useRuntime'
|
||||||
import { useMessageStyle, useSettings } from '@renderer/hooks/useSettings'
|
import { useMessageStyle, useSettings } from '@renderer/hooks/useSettings'
|
||||||
import { getMessageModelId } from '@renderer/services/MessagesService'
|
import { getMessageModelId } from '@renderer/services/MessagesService'
|
||||||
import { getModelName } from '@renderer/services/ModelService'
|
import { getModelName } from '@renderer/services/ModelService'
|
||||||
@ -37,6 +39,10 @@ const MessageHeader: FC<Props> = memo(({ assistant, model, message, topic, isGro
|
|||||||
const avatar = useAvatar()
|
const avatar = useAvatar()
|
||||||
const { theme } = useTheme()
|
const { theme } = useTheme()
|
||||||
const { userName, sidebarIcons } = useSettings()
|
const { userName, sidebarIcons } = useSettings()
|
||||||
|
const { chat } = useRuntime()
|
||||||
|
const { activeTopicOrSession, activeAgentId } = chat
|
||||||
|
const { agent } = useAgent(activeAgentId)
|
||||||
|
const isAgentView = activeTopicOrSession === 'session'
|
||||||
const { t } = useTranslation()
|
const { t } = useTranslation()
|
||||||
const { isBubbleStyle } = useMessageStyle()
|
const { isBubbleStyle } = useMessageStyle()
|
||||||
const { openMinappById } = useMinappPopup()
|
const { openMinappById } = useMinappPopup()
|
||||||
@ -52,12 +58,16 @@ const MessageHeader: FC<Props> = memo(({ assistant, model, message, topic, isGro
|
|||||||
return APP_NAME
|
return APP_NAME
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if (isAgentView && message.role === 'assistant') {
|
||||||
|
return agent?.name ?? t('common.unknown')
|
||||||
|
}
|
||||||
|
|
||||||
if (message.role === 'assistant') {
|
if (message.role === 'assistant') {
|
||||||
return getModelName(model) || getMessageModelId(message) || ''
|
return getModelName(model) || getMessageModelId(message) || ''
|
||||||
}
|
}
|
||||||
|
|
||||||
return userName || t('common.you')
|
return userName || t('common.you')
|
||||||
}, [message, model, t, userName])
|
}, [agent?.name, isAgentView, message, model, t, userName])
|
||||||
|
|
||||||
const isAssistantMessage = message.role === 'assistant'
|
const isAssistantMessage = message.role === 'assistant'
|
||||||
const isUserMessage = message.role === 'user'
|
const isUserMessage = message.role === 'user'
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user