not working

Author: utkarshx

package-lock.json

@@ -30,16 +30,29 @@ import {chatSummary} from './modules/history'
  * @description This class provides a unified interface to interact with various modules.
  */
 class Codebolt  { // Extend EventEmitter
+    private static instance: Codebolt | null = null;
+    private wsManager: cbws;
+    private vectorDB: vectorDB;
+
 
     /**
      * @constructor
      * @description Initializes the websocket connection.
      */
     constructor() {
-
-        this.websocket = cbws.getWebsocket;
+        this.wsManager = new cbws();
+        this.vectordb = new vectorDB(this.wsManager);
+        // this.websocket = cbws.getWebsocket;
+        // ... initialize other modules ...
     }
 
+    public static getInstance(): Codebolt {
+        if (!Codebolt.instance) {
+            Codebolt.instance = new Codebolt();
+        }
+        return Codebolt.instance;
+    }
+
     /**
      * @method waitForConnection
      * @description Waits for the WebSocket connection to open.
@@ -68,6 +81,14 @@ class Codebolt  { // Extend EventEmitter
         });
     }
 
+    async connect() {
+        await this.wsManager.connect();
+    }
+
+    async disconnect() {
+        await this.wsManager.disconnect();
+    }
+
     websocket: WebSocket | null = null;
     fs = cbfs;
     git = git;
@@ -93,6 +114,6 @@ class Codebolt  { // Extend EventEmitter
     chatSummary=chatSummary
 }
 
-export default new Codebolt();
+export default Codebolt.getInstance();
 
 // module.exports = new Codebolt();

src/index.ts

@@ -1,29 +1,36 @@
 import cbws from './websocket';
 import { AddVectorItemResponse,GetVectorResponse,QueryVectorItemResponse } from '@codebolt/types';
-const VectorDB = {
+class VectorDB {
+    private wsManager: cbws;
+    private ws: WebSocket;
+
+    constructor(wsManager: cbws) {
+        this.wsManager = wsManager;
+        this.ws = this.wsManager.getWebsocket();
+    }
     /**
      * Retrieves a vector from the vector database based on the provided key.
      *
      * @param {string} key - The key of the vector to retrieve.
      * @returns {Promise<GetVectorResponse>} A promise that resolves with the retrieved vector.
      */
-    getVector: async (key: string): Promise<GetVectorResponse> => {
+    getVector = async (key: string): Promise<GetVectorResponse> => {
         return new Promise((resolve, reject) => {
-            cbws.getWebsocket.send(JSON.stringify({
+            this.ws.send(JSON.stringify({
                 "type":"vectordbEvent",
                 "action": "getVector",
                 "message": {
                     item: key
                 },
             }));
-            cbws.getWebsocket.on('message', (data: string) => {
+            this.ws.on('message', (data: string) => {
                 const response = JSON.parse(data);
                 if (response.type === "getVectorResponse") {
                     resolve(response);
                 }
             });
         });
-    },
+    }
 
     /**
      * Adds a new vector item to the vector database.
@@ -32,71 +39,71 @@ const VectorDB = {
      * @param {any} item - The item to add to the vector.
      * @returns {Promise<AddVectorItemResponse>} A promise that resolves when the item is successfully added.
      */
-    addVectorItem: async ( item: any): Promise<AddVectorItemResponse> => {
+    addVectorItem =  async ( item: any): Promise<AddVectorItemResponse> => {
         return new Promise((resolve, reject) => {
-            cbws.getWebsocket.send(JSON.stringify({
+            this.ws.send(JSON.stringify({
                 "type":"vectordbEvent",
                 "action": "addVectorItem",
                 "message": {
                     item: item
                 },
             }));
-            cbws.getWebsocket.on('message', (data: string) => {
+            this.ws.on('message', (data: string) => {
                 const response = JSON.parse(data);
                 if (response.type === "addVectorItemResponse") {
                     resolve(response);
                 }
             });
         });
-    },
+    }
 
     /**
      * Queries a vector item from the vector database based on the provided key.
      *
      * @param {string} key - The key of the vector to query the item from.
      * @returns {Promise<QueryVectorItemResponse>} A promise that resolves with the queried vector item.
      */
-    queryVectorItem: async (key: string): Promise<QueryVectorItemResponse> => {
+    queryVectorItem =  async (key: string): Promise<QueryVectorItemResponse> => {
         return new Promise((resolve, reject) => {
-            cbws.getWebsocket.send(JSON.stringify({
+            this.ws.send(JSON.stringify({
                 "type":"vectordbEvent",
                 "action": "queryVectorItem",
                 "message": {
                     item: key
                 },
             }));
-            cbws.getWebsocket.on('message', (data: string) => {
+            this.ws.on('message', (data: string) => {
                 const response = JSON.parse(data);
                 if (response.type === "qeryVectorItemResponse") {
                     resolve(response);
                 }
             });
         });
-    },
+    }
     /**
      * Queries a vector item from the vector database based on the provided key.
      *
      * @param {string} key - The key of the vector to query the item from.
      * @returns {Promise<QueryVectorItemResponse>} A promise that resolves with the queried vector item.
      */
-    queryVectorItems: async (items: [],dbPath:string): Promise<QueryVectorItemResponse> => {
+    queryVectorItems = async (items: [],dbPath:string): Promise<QueryVectorItemResponse> => {
         return new Promise((resolve, reject) => {
-            cbws.getWebsocket.send(JSON.stringify({
+            this.ws.send(JSON.stringify({
                 "type":"vectordbEvent",
                 "action": "queryVectorItems",
                 "message": {
                     items,
                     dbPath
                 },
             }));
-            cbws.getWebsocket.on('message', (data: string) => {
+            this.ws.on('message', (data: string) => {
                 const response = JSON.parse(data);
                 if (response.type === "qeryVectorItemsResponse") {
                     resolve(response);
                 }
             });
         });
-    },
+    }
 };
 
 export default VectorDB;