diff --git a/components/Chat/ChatMessage.tsx b/components/Chat/ChatMessage.tsx index 5848b32..19481d8 100644 --- a/components/Chat/ChatMessage.tsx +++ b/components/Chat/ChatMessage.tsx @@ -1,5 +1,5 @@ import { Message } from '@/types/chat'; -import { IconCheck, IconCopy, IconEdit } from '@tabler/icons-react'; +import { IconCheck, IconCopy, IconEdit, IconUser, IconRobot } from '@tabler/icons-react'; import { useTranslation } from 'next-i18next'; import { FC, memo, useEffect, useRef, useState } from 'react'; import rehypeMathjax from 'rehype-mathjax'; @@ -81,7 +81,7 @@ export const ChatMessage: FC = memo( >
- {message.role === 'assistant' ? t('AI') : t('You')}: + {message.role === 'assistant' ? : }