fix(duplicates): workaround for duplicates

a proper fix will come later. Stopgap solution.
aws-samples · Feb 2, 2024 · c1ef47d · c1ef47d
1 parent 58876be
commit c1ef47d
Showing 1 changed file with 12 additions and 3 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
@@ -136,13 +136,22 @@ export function updateMessageHistoryRef(
       messageHistory.at(-1)?.type !== ChatBotMessageType.Human
     ) {
       const lastMessage = messageHistory.at(-1)!;
-      lastMessage.tokens = lastMessage.tokens || [];
+      lastMessage.tokens = lastMessage.tokens ?? [];
       if (hasToken) {
-        lastMessage.tokens.push(token);
+        // Workaround for text duplicates issue
+        if (
+          !lastMessage.tokens
+            .map((t) => t.sequenceNumber)
+            .includes(token.sequenceNumber)
+        ) {
+          lastMessage.tokens.push(token);
+        } else {
+          return;
+        }
+      } else {
       }
 
       lastMessage.tokens.sort((a, b) => a.sequenceNumber - b.sequenceNumber);
-      console.log(lastMessage);
       if (lastMessage.tokens.length > 0) {
         const lastRunId =
           lastMessage.tokens[lastMessage.tokens.length - 1].runId;