From 197392a95ffe13528c39c91e924c6053fc4d5c54 Mon Sep 17 00:00:00 2001
From: Weves <chrisweaver101@gmail.com>
Date: Sat, 2 Mar 2024 23:19:31 -0800
Subject: [PATCH] Change AI Message name to Persona name

---
 web/src/app/chat/Chat.tsx             | 3 +++
 web/src/app/chat/ChatIntro.tsx        | 4 ++--
 web/src/app/chat/message/Messages.tsx | 6 +++++-
 3 files changed, 10 insertions(+), 3 deletions(-)

diff --git a/web/src/app/chat/Chat.tsx b/web/src/app/chat/Chat.tsx
index 686a25b2c..b3fe5ef95 100644
--- a/web/src/app/chat/Chat.tsx
+++ b/web/src/app/chat/Chat.tsx
@@ -564,6 +564,7 @@ export const Chat = ({
                           messageId={message.messageId}
                           content={message.message}
                           query={messageHistory[i]?.query || undefined}
+                          personaName={livePersona.name}
                           citedDocuments={getCitedDocumentsFromMessage(message)}
                           isComplete={
                             i !== messageHistory.length - 1 || !isStreaming
@@ -644,6 +645,7 @@ export const Chat = ({
                       <div key={i}>
                         <AIMessage
                           messageId={message.messageId}
+                          personaName={livePersona.name}
                           content={
                             <p className="text-red-700 text-sm my-auto">
                               {message.message}
@@ -661,6 +663,7 @@ export const Chat = ({
                     <div key={messageHistory.length}>
                       <AIMessage
                         messageId={null}
+                        personaName={livePersona.name}
                         content={
                           <div className="text-sm my-auto">
                             <ThreeDots
diff --git a/web/src/app/chat/ChatIntro.tsx b/web/src/app/chat/ChatIntro.tsx
index b19323440..84ae38f23 100644
--- a/web/src/app/chat/ChatIntro.tsx
+++ b/web/src/app/chat/ChatIntro.tsx
@@ -133,7 +133,7 @@ export function ChatIntro({
                       <p className="font-bold mb-1 mt-4 text-emphasis">
                         Knowledge Sets:{" "}
                       </p>
-                      <div className="flex flex-wrap gap-x-2">
+                      <div className="flex flex-wrap gap-2">
                         {selectedPersona.document_sets.map((documentSet) => (
                           <div key={documentSet.id} className="w-fit">
                             <HoverPopup
@@ -165,7 +165,7 @@ export function ChatIntro({
                       <p className="font-bold mb-1 mt-4 text-emphasis">
                         Connected Sources:{" "}
                       </p>
-                      <div className="flex flex-wrap gap-x-2">
+                      <div className="flex flex-wrap gap-2">
                         {availableSourceMetadata.map((sourceMetadata) => (
                           <span
                             key={sourceMetadata.internalName}
diff --git a/web/src/app/chat/message/Messages.tsx b/web/src/app/chat/message/Messages.tsx
index e17a9fa36..514451f38 100644
--- a/web/src/app/chat/message/Messages.tsx
+++ b/web/src/app/chat/message/Messages.tsx
@@ -33,6 +33,7 @@ export const AIMessage = ({
   messageId,
   content,
   query,
+  personaName,
   citedDocuments,
   isComplete,
   hasDocs,
@@ -46,6 +47,7 @@ export const AIMessage = ({
   messageId: number | null;
   content: string | JSX.Element;
   query?: string;
+  personaName?: string;
   citedDocuments?: [string, DanswerDocument][] | null;
   isComplete?: boolean;
   hasDocs?: boolean;
@@ -68,7 +70,9 @@ export const AIMessage = ({
               </div>
             </div>
 
-            <div className="font-bold text-emphasis ml-2 my-auto">Danswer</div>
+            <div className="font-bold text-emphasis ml-2 my-auto">
+              {personaName || "Danswer"}
+            </div>
 
             {query === undefined &&
               hasDocs &&