Merge pull request #351 from continuedev/history

Session History

1 files changed, 6 insertions, 0 deletions

diff --git a/extension/react-app/src/hooks/messenger.ts b/extension/react-app/src/hooks/messenger.ts
index ecf646c7..0bfbe00c 100644
--- a/extension/react-app/src/hooks/messenger.ts
+++ b/extension/react-app/src/hooks/messenger.ts
@@ -15,6 +15,8 @@ export abstract class Messenger {
   abstract sendAndReceive(messageType: string, data: any): Promise<any>;
 
   abstract onError(callback: (error: any) => void): void;
+
+  abstract close(): void;
 }
 
 export class WebsocketMessenger extends Messenger {
@@ -105,4 +107,8 @@ export class WebsocketMessenger extends Messenger {
   onError(callback: (error: any) => void): void {
     this.websocket.addEventListener("error", callback);
   }
+
+  close(): void {
+    this.websocket.close();
+  }
 }