feat(ChatGPTBrowserClient): add optional onEventMessage parameter to …

…postConversation method to handle event messages from the server. (#297)
waylaidwanderer · Apr 12, 2023 · c34ae10 · c34ae10
1 parent c55bc25
commit c34ae10
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/src/ChatGPTBrowserClient.js b/src/ChatGPTBrowserClient.js
@@ -29,7 +29,7 @@ export default class ChatGPTBrowserClient {
         this.model = this.options.model || 'text-davinci-002-render-sha';
     }
 
-    async postConversation(conversation, onProgress, abortController = null) {
+    async postConversation(conversation, onProgress, abortController, onEventMessage = null) {
         const {
             action = 'next',
             conversationId,
@@ -133,6 +133,11 @@ export default class ChatGPTBrowserClient {
                         if (debug) {
                             console.debug(eventMessage);
                         }
+
+                        if (onEventMessage) {
+                            onEventMessage(eventMessage);
+                        }
+
                         if (!eventMessage.data || eventMessage.event === 'ping') {
                             return;
                         }
@@ -212,6 +217,7 @@ export default class ChatGPTBrowserClient {
             },
             opts.onProgress || (() => {}),
             opts.abortController || new AbortController(),
+            opts?.onEventMessage,
         );
 
         if (this.options.debug) {