ag2 socket close hook

ag2ai · Feb 5, 2025 · f3daa6d · f3daa6d
1 parent 86bae56
commit f3daa6d
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/src/webRTC.ts b/src/webRTC.ts
@@ -1,3 +1,5 @@
+import type { WebSocketEvents } from 'vitest';
+
 interface AG2InitMessage {
   type: 'ag2.init';
   config: {
@@ -17,11 +19,15 @@ export class WebRTC {
   private microphone?: MediaStreamTrack;
   private ws: WebSocket | null;
   private pc: RTCPeerConnection | undefined;
+  public onAG2SocketClose: (ev: CloseEvent) => void;
 
   constructor(ag2SocketUrl: string, microphone?: MediaStreamTrack) {
     this.ag2SocketUrl = ag2SocketUrl;
     this.microphone = microphone;
     this.ws = null;
+    this.onAG2SocketClose = (ev: CloseEvent) => {
+      console.log('AG2 Websocket closed');
+    };
   }
 
   async disconnect(): Promise<void> {
@@ -138,6 +144,10 @@ export class WebRTC {
       console.log('web socket opened');
     };
 
+    this.ws.onclose = (event) => {
+      this.onAG2SocketClose(event);
+    };
+
     this.ws.onmessage = async (event) => {
       try {
         const message: AG2Message = JSON.parse(event.data);