diff --git a/src/renderer/src/pages/home/Messages/MessageHeader.tsx b/src/renderer/src/pages/home/Messages/MessageHeader.tsx index 077506e95e..95a6d219cb 100644 --- a/src/renderer/src/pages/home/Messages/MessageHeader.tsx +++ b/src/renderer/src/pages/home/Messages/MessageHeader.tsx @@ -4,9 +4,11 @@ import UserPopup from '@renderer/components/Popups/UserPopup' import { APP_NAME, AppLogo, isLocalAi } from '@renderer/config/env' import { getModelLogo } from '@renderer/config/models' import { useTheme } from '@renderer/context/ThemeProvider' +import { useAgent } from '@renderer/hooks/agents/useAgent' import useAvatar from '@renderer/hooks/useAvatar' import { useChatContext } from '@renderer/hooks/useChatContext' import { useMinappPopup } from '@renderer/hooks/useMinappPopup' +import { useRuntime } from '@renderer/hooks/useRuntime' import { useMessageStyle, useSettings } from '@renderer/hooks/useSettings' import { getMessageModelId } from '@renderer/services/MessagesService' import { getModelName } from '@renderer/services/ModelService' @@ -37,6 +39,10 @@ const MessageHeader: FC = memo(({ assistant, model, message, topic, isGro const avatar = useAvatar() const { theme } = useTheme() const { userName, sidebarIcons } = useSettings() + const { chat } = useRuntime() + const { activeTopicOrSession, activeAgentId } = chat + const { agent } = useAgent(activeAgentId) + const isAgentView = activeTopicOrSession === 'session' const { t } = useTranslation() const { isBubbleStyle } = useMessageStyle() const { openMinappById } = useMinappPopup() @@ -52,12 +58,16 @@ const MessageHeader: FC = memo(({ assistant, model, message, topic, isGro return APP_NAME } + if (isAgentView && message.role === 'assistant') { + return agent?.name ?? t('common.unknown') + } + if (message.role === 'assistant') { return getModelName(model) || getMessageModelId(message) || '' } return userName || t('common.you') - }, [message, model, t, userName]) + }, [agent?.name, isAgentView, message, model, t, userName]) const isAssistantMessage = message.role === 'assistant' const isUserMessage = message.role === 'user'