💄 style: 修正 assistant 消息没有 background 的问题

close lobehub#42
bentwnghk · Aug 5, 2023 · 812e976 · 812e976
1 parent 5c8eb9c
commit 812e976
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/src/store/session/slices/chat/selectors/chat.ts b/src/store/session/slices/chat/selectors/chat.ts
@@ -19,6 +19,7 @@ export const getChatsById =
       session,
       {
         assistant: agentSelectors.currentAgentAvatar(s),
+        assistantBackground: agentSelectors.currentAgentBackgroundColor(s),
         user: useSettings.getState().settings.avatar || DEFAULT_USER_AVATAR,
       },
       s.activeTopicId,

diff --git a/src/store/session/slices/chat/selectors/utils.ts b/src/store/session/slices/chat/selectors/utils.ts
@@ -4,7 +4,7 @@ import { LobeAgentSession } from '@/types/session';
 
 export const organizeChats = (
   session: LobeAgentSession,
-  avatar: { assistant: string; user: string },
+  avatar: { assistant: string; assistantBackground?: string; user: string },
   topicId?: string,
 ) => {
   const getMeta = (message: ChatMessage) => {
@@ -22,6 +22,7 @@ export const organizeChats = (
       case 'assistant': {
         return {
           avatar: avatar.assistant,
+          backgroundColor: avatar.assistantBackground,
           title: session.meta.title,
         };
       }