Log warnings for 'close' event reasons

Author: dlarocque

packages/ai/src/platform/browser/websocket.ts

@@ -134,7 +134,12 @@ export class BrowserWebSocketHandler implements WebSocketHandler {
       }
     };
 
-    const closeListener = (): void => {
+    const closeListener = (event: CloseEvent): void => {
+      if (event.reason) {
+        logger.warn(
+          `WebSocket connection closed by the server with reason: ${event.reason}`
+        );
+      }
       isClosed = true;
       if (resolvePromise) {
         resolvePromise();

packages/ai/src/platform/node/websocket.ts

@@ -16,6 +16,7 @@
  */
 
 import { AIError } from '../../errors';
+import { logger } from '../../logger';
 import { AIErrorCode } from '../../types';
 import { WebSocketHandler } from '../websocket';
 
@@ -145,7 +146,12 @@ export class NodeWebSocketHandler implements WebSocketHandler {
       }
     };
 
-    const closeListener = (): void => {
+    const closeListener = (event: CloseEvent): void => {
+      if (event.reason) {
+        logger.warn(
+          `WebSocket connection closed by the server with reason: ${event.reason}`
+        );
+      }
       isClosed = true;
       if (resolvePromise) {
         resolvePromise();