fix(browser-client): ignore any messages not from assistant role

waylaidwanderer · Feb 27, 2023 · aafdb30 · aafdb30
1 parent bb32168
commit aafdb30
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/ChatGPTBrowserClient.js b/src/ChatGPTBrowserClient.js
@@ -120,8 +120,12 @@ export default class ChatGPTBrowserClient {
                             return;
                         }
                         try {
-                            const lastMessage = lastEvent ? lastEvent.message.content.parts[0] : '';
                             const data = JSON.parse(message.data);
+                            // ignore any messages that are not from the assistant
+                            if (data.message?.author?.role !== 'assistant') {
+                                return;
+                            }
+                            const lastMessage = lastEvent ? lastEvent.message.content.parts[0] : '';
                             const newMessage = data.message.content.parts[0];
                             // get the difference between the current text and the previous text
                             const difference = newMessage.substring(lastMessage.length);