From dd8be0712094402a9e7d8dbdf25ab47a8fd03485 Mon Sep 17 00:00:00 2001
From: GenkaOk <2690324+GenkaOk@users.noreply.github.com>
Date: Mon, 1 Apr 2024 07:32:06 +0300
Subject: [PATCH] Set name assistant (#2529)

---
 web/screens/Chat/SimpleTextMessage/index.tsx | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/web/screens/Chat/SimpleTextMessage/index.tsx b/web/screens/Chat/SimpleTextMessage/index.tsx
index 489def1c1..376d011eb 100644
--- a/web/screens/Chat/SimpleTextMessage/index.tsx
+++ b/web/screens/Chat/SimpleTextMessage/index.tsx
@@ -44,12 +44,14 @@ import {
   editMessageAtom,
   getCurrentChatMessagesAtom,
 } from '@/helpers/atoms/ChatMessage.atom'
+import { activeThreadAtom } from '@/helpers/atoms/Thread.atom'
 
 const SimpleTextMessage: React.FC<ThreadMessage> = (props) => {
   let text = ''
   const isUser = props.role === ChatCompletionRole.User
   const isSystem = props.role === ChatCompletionRole.System
   const editMessage = useAtomValue(editMessageAtom)
+  const activeThread = useAtomValue(activeThreadAtom)
 
   if (props.content && props.content.length > 0) {
     text = props.content[0]?.text?.value ?? ''
@@ -178,7 +180,9 @@ const SimpleTextMessage: React.FC<ThreadMessage> = (props) => {
             isUser && 'text-gray-500'
           )}
         >
-          {props.role}
+          {isUser
+            ? props.role
+            : activeThread?.assistants[0].assistant_name ?? props.role}
         </div>
         <p className="text-xs font-medium text-gray-400">
           {displayDate(props.created)}