Merge branch 'main' of github.com:numan-ai/explainable_frontend

numan-ai · Jul 9, 2024 · daa2b09 · daa2b09
2 parents 29d6264 + f274fb2
commit daa2b09
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/src/api.ts b/src/api.ts
@@ -61,9 +61,10 @@ class WebSocketClient {
     this.ws.onmessage = (event) => {
       const message = event.data;
       const data = JSON.parse(message);
-      console.log('Got message', data);
-      if (this.messageCallbacks[data.type]) {
-        this.messageCallbacks[data.type].forEach(callback => callback(data.data));
+      if (Array.isArray(data)) {
+        data.forEach(this.processMessage);
+      } else {
+        this.processMessage(data);
       }
     };
 
@@ -75,6 +76,13 @@ class WebSocketClient {
     };
   }
 
+  private processMessage = (data: any) => {
+    console.log('Got message', data);
+    if (this.messageCallbacks[data.type]) {
+      this.messageCallbacks[data.type].forEach(callback => callback(data.data));
+    }
+  }
+
   public send(message: string) {
     if (!this.ws) {
       return;