From 9cafc09d6f9c5ef0a347af1c6b37617f33a86ae3 Mon Sep 17 00:00:00 2001
From: Lucas Oliveira <lucas.edu.oli@hotmail.com>
Date: Sat, 24 Feb 2024 23:01:36 +0100
Subject: [PATCH] Fixed message not being displayed

---
 src/frontend/src/components/newChatView/chatMessage/index.tsx | 2 +-
 src/frontend/src/modals/formModal/index.tsx                   | 1 +
 src/frontend/src/types/components/index.ts                    | 1 +
 3 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/frontend/src/components/newChatView/chatMessage/index.tsx b/src/frontend/src/components/newChatView/chatMessage/index.tsx
index 36fb3955e..31f4f0345 100644
--- a/src/frontend/src/components/newChatView/chatMessage/index.tsx
+++ b/src/frontend/src/components/newChatView/chatMessage/index.tsx
@@ -70,7 +70,7 @@ export default function ChatMessage({
 
   useEffect(() => {
     // This effect is specifically for calling updateChat after streaming ends
-    if (!isStreaming && streamUrl === null) {
+    if (!isStreaming && streamUrl) {
       if (updateChat) {
         updateChat(chat, chatMessage, streamUrl);
       }
diff --git a/src/frontend/src/modals/formModal/index.tsx b/src/frontend/src/modals/formModal/index.tsx
index dfaf871ed..af6e9eace 100644
--- a/src/frontend/src/modals/formModal/index.tsx
+++ b/src/frontend/src/modals/formModal/index.tsx
@@ -576,6 +576,7 @@ export default function FormModal({
                           chatHistory.length - 1 === index ? true : false
                         }
                         key={index}
+                        updateChat={() => {}}
                       />
                     ))
                   ) : (
diff --git a/src/frontend/src/types/components/index.ts b/src/frontend/src/types/components/index.ts
index 3a8563cf1..bc8582118 100644
--- a/src/frontend/src/types/components/index.ts
+++ b/src/frontend/src/types/components/index.ts
@@ -524,6 +524,7 @@ export type chatMessagePropsType = {
   chat: ChatMessageType;
   lockChat: boolean;
   lastMessage: boolean;
+  updateChat: (chat: ChatMessageType, message: string, stream_url: string) => void;
 };
 
 export type formModalPropsType = {