fix(streaming): fixing issues with streamed responses

aws-samples · Mar 5, 2024 · 4bf20ba · 4bf20ba
1 parent e3dd01a
commit 4bf20ba
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/lib/user-interface/react-app/src/components/chatbot/utils.ts b/lib/user-interface/react-app/src/components/chatbot/utils.ts
@@ -156,11 +156,11 @@ export function updateMessageHistoryRef(
         metadata = lastMessage.metadata;
       }
 
-      if (hasContent) {
+      if (hasContent || lastMessage.content.length > 0) {
         messageHistory[messageHistory.length - 1] = {
           ...lastMessage,
           type: ChatBotMessageType.AI,
-          content,
+          content: content ?? lastMessage.content,
           metadata,
           tokens: lastMessage.tokens,
         };