Merge pull request #177 from l3vels/fix/chat-avatar

Fix: Chat Avatar
l3vels · Sep 27, 2023 · 03e6a1b · 03e6a1b
2 parents f92a575 + 1903ea0
commit 03e6a1b
Show file tree

Hide file tree

Showing 10 changed files with 22 additions and 15 deletions.
diff --git a/apps/server/controllers/chat.py b/apps/server/controllers/chat.py
@@ -250,7 +250,7 @@ def get_chat_messages(is_private_chat: bool, agent_id: Optional[UUID] = None, te
                  .filter(ChatMessageModel.session_id == session_id)
                  .order_by(ChatMessageModel.created_on.desc())
                  .limit(50)
-                 .options(joinedload(ChatMessageModel.agent), joinedload(ChatMessageModel.team), joinedload(ChatMessageModel.parent))
+                 .options(joinedload(ChatMessageModel.agent), joinedload(ChatMessageModel.team), joinedload(ChatMessageModel.parent), joinedload(ChatMessageModel.creator))
                  .all())
 
     chat_messages = [chat_message.to_dict() for chat_message in chat_messages]
@@ -286,7 +286,7 @@ def get_chat_messages(agent_id: Optional[UUID] = None, team_id: Optional[UUID] =
                  .filter(ChatMessageModel.session_id == session_id)
                  .order_by(ChatMessageModel.created_on.desc())
                  .limit(50)
-                 .options(joinedload(ChatMessageModel.agent), joinedload(ChatMessageModel.team), joinedload(ChatMessageModel.parent))
+                 .options(joinedload(ChatMessageModel.agent), joinedload(ChatMessageModel.team), joinedload(ChatMessageModel.parent), joinedload(ChatMessageModel.creator))
                  .all())
 
     chat_messages = [chat_message.to_dict() for chat_message in chat_messages]

diff --git a/apps/server/models/chat_message.py b/apps/server/models/chat_message.py
@@ -32,7 +32,7 @@ class ChatMessage(BaseModel):
 
     created_by = Column(UUID, ForeignKey('user.id', name='fk_created_by', ondelete='CASCADE'), nullable=True, index=True)
     modified_by = Column(UUID, ForeignKey('user.id', name='fk_modified_by', ondelete='CASCADE'), nullable=True, index=True)
-    creator = relationship("UserModel", foreign_keys=[created_by], lazy='select')
+    creator = relationship("UserModel", foreign_keys=[user_id], lazy='select')
 
     @classmethod
     def get_chat_message_by_id(cls, db, chat_message_id: UUID, account: AccountOutput):
@@ -72,4 +72,7 @@ def to_dict(self):
         if self.parent:
             data['parent'] = self.parent.to_dict()
 
+        if self.creator:
+            data['creator'] = self.creator.to_dict()
+
         return data
diff --git a/apps/server/typings/chat.py b/apps/server/typings/chat.py
@@ -28,6 +28,7 @@ class ChatMessageOutput(BaseModel):
     team_id: Optional[UUID] = None
     team: Optional[Dict] = None
     user_id: UUID
+    creator: Optional[Dict] = None
     account_id: UUID
     message: Dict
     thoughts: Optional[List[Dict]] = None

diff --git a/apps/ui/src/gql/chat/chatMessages.gql b/apps/ui/src/gql/chat/chatMessages.gql
@@ -15,5 +15,6 @@ query chatMessages($is_private_chat: Boolean!, $agent_id: String!, $team_id: Str
     agent
     team_id
     team
+    creator
   }
 }
diff --git a/apps/ui/src/gql/chat/chatMessagesHistory.gql b/apps/ui/src/gql/chat/chatMessagesHistory.gql
@@ -15,5 +15,6 @@ query chatMessages($is_private_chat: Boolean!, $agent_id: String!, $team_id: Str
     agent
     team_id
     team
+    creator
   }
 }
diff --git a/apps/ui/src/modals/AIChatModal/components/ChatHistory.tsx b/apps/ui/src/modals/AIChatModal/components/ChatHistory.tsx
@@ -79,7 +79,7 @@ const ChatHistory = () => {
     <StyledRoot>
       {(agentById || teamById) && (
         <StyledMembersWrapper>
-          <ChatMembers agentById={agentById} teamOfAgents={teamById} />
+          <ChatMembers agentById={agentById} teamOfAgents={teamById} isHistory />
         </StyledMembersWrapper>
       )}
 

diff --git a/apps/ui/src/modals/AIChatModal/components/ChatMembers/ChatMembers.tsx b/apps/ui/src/modals/AIChatModal/components/ChatMembers/ChatMembers.tsx
@@ -25,11 +25,11 @@ import MemberText from './components/MemberText'
 const ChatMembers = ({
   agentById,
   teamOfAgents,
-  userName,
+  isHistory,
 }: {
   agentById?: any
   teamOfAgents?: any
-  userName?: string
+  isHistory?: boolean
 }) => {
   const { user } = React.useContext(AuthContext)
 
@@ -57,10 +57,10 @@ const ChatMembers = ({
           <TabsContext activeTabId={activeTab}>
             <TabPanels noAnimation>
               <TabPanel>
-                {userName && (
+                {!isHistory && user?.name && (
                   <StyledAgentWrapper>
-                    <AvatarGenerator name={userName} size={30} />
-                    <MemberText name={userName} />
+                    <AvatarGenerator name={user.name} size={30} avatar={user.avatar} />
+                    <MemberText name={user.name} />
                   </StyledAgentWrapper>
                 )}
 
@@ -129,10 +129,10 @@ const ChatMembers = ({
           <TabsContext activeTabId={activeTab}>
             <TabPanels noAnimation>
               <TabPanel>
-                {userName && (
+                {!isHistory && user?.name && (
                   <StyledAgentWrapper>
-                    <AvatarGenerator name={userName} size={30} />
-                    <MemberText name={userName} />
+                    <AvatarGenerator name={user.name} size={30} avatar={user.avatar} />
+                    <MemberText name={user.name} />
                   </StyledAgentWrapper>
                 )}
 

diff --git a/apps/ui/src/modals/AIChatModal/components/ChatMessageList/ChatMessageListV2.tsx b/apps/ui/src/modals/AIChatModal/components/ChatMessageList/ChatMessageListV2.tsx
@@ -58,6 +58,7 @@ const ChatMessageListV2 = ({
       agentName: chat.agent?.name,
       teamName: chat.team?.name,
       avatar: chat?.agent?.avatar,
+      creator: chat?.creator,
     }
   })
 
@@ -169,7 +170,7 @@ const ChatMessageListV2 = ({
                     ))}
                 </StyledReplyMessageContainer>
                 <HumanMessage
-                  avatarImg={Avatar_3}
+                  avatarImg={chat.creator?.avatar}
                   userId={chat.user_id}
                   messageDate={chat.date}
                   messageText={chat.message}

diff --git a/apps/ui/src/modals/AIChatModal/components/ChatMessageList/components/HumanMessage.tsx b/apps/ui/src/modals/AIChatModal/components/ChatMessageList/components/HumanMessage.tsx
@@ -37,7 +37,7 @@ const HumanMessage = ({
     <>
       <StyledMessageWrapper>
         <StyledAvatarWrapper>
-          <AvatarGenerator name={authorName} size={50} />
+          <AvatarGenerator name={authorName} size={50} avatar={avatarImg} />
         </StyledAvatarWrapper>
 
         <StyledMainContent>

diff --git a/apps/ui/src/modals/AIChatModal/components/ChatV2.tsx b/apps/ui/src/modals/AIChatModal/components/ChatV2.tsx
@@ -292,7 +292,7 @@ const ChatV2 = ({ isPrivate = false }: ChatV2Props) => {
   return (
     <StyledWrapper>
       <StyledMembersWrapper>
-        <ChatMembers agentById={agentById} teamOfAgents={teamOfAgents} userName={user.name} />
+        <ChatMembers agentById={agentById} teamOfAgents={teamOfAgents} />
       </StyledMembersWrapper>
 
       <StyledMessages>
-Original file line number
+Diff line change
@@ Expand Up @@
         agent
         team_id
         team
+        creator
       }
     }