fix node test

pinterest · Aug 14, 2023 · 997c8ee · 997c8ee
1 parent f0cf15b
commit 997c8ee
Showing 1 changed file with 13 additions and 13 deletions.
diff --git a/querybook/webapp/lib/ai-assistant/ai-assistant-socketio.ts b/querybook/webapp/lib/ai-assistant/ai-assistant-socketio.ts
@@ -13,14 +13,26 @@ class AIAssistantSocket {
     private socket: Socket = null;
     private socketPromise: Promise<Socket> = null;
 
-    constructor() {
+    public constructor() {
         this.setupSocket();
     }
 
     public onSocketConnect(socket: Socket) {
         socket.emit('subscribe');
     }
 
+    public requestAIAssistant = (command: AICommandType, payload: object) => {
+        this.socket.emit(AI_ASSISTANT_REQUEST_EVENT, command, payload);
+    };
+
+    public addAIListener = (listener) => {
+        this.socket.on(AI_ASSISTANT_RESPONSE_EVENT, listener);
+    };
+
+    public removeAIListener = (listener) => {
+        this.socket.off(AI_ASSISTANT_RESPONSE_EVENT, listener);
+    };
+
     private setupSocket = async () => {
         if (this.socket) {
             return this.socket;
@@ -44,18 +56,6 @@ class AIAssistantSocket {
 
         return this.socket;
     };
-
-    public requestAIAssistant = (command: AICommandType, payload: object) => {
-        this.socket.emit(AI_ASSISTANT_REQUEST_EVENT, command, payload);
-    };
-
-    public addAIListener = (listener) => {
-        this.socket.on(AI_ASSISTANT_RESPONSE_EVENT, listener);
-    };
-
-    public removeAIListener = (listener) => {
-        this.socket.off(AI_ASSISTANT_RESPONSE_EVENT, listener);
-    };
 }
 
 const defaultSocket = new AIAssistantSocket();