@codebolt/codeboltjs 1.1.75 → 1.1.77

package/src/modules/vectordb.ts

@@ -0,0 +1,102 @@
+import cbws from './websocket';
+import { AddVectorItemResponse,GetVectorResponse,QueryVectorItemResponse } from '@codebolt/types';
+const VectorDB = {
+    /**
+     * 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> => {
+        return new Promise((resolve, reject) => {
+            cbws.getWebsocket.send(JSON.stringify({
+                "type":"vectordbEvent",
+                "action": "getVector",
+                "message": {
+                    item: key
+                },
+            }));
+            cbws.getWebsocket.on('message', (data: string) => {
+                const response = JSON.parse(data);
+                if (response.type === "getVectorResponse") {
+                    resolve(response);
+                }
+            });
+        });
+    },
+
+    /**
+     * Adds a new vector item to the vector database.
+     *
+
+     * @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> => {
+        return new Promise((resolve, reject) => {
+            cbws.getWebsocket.send(JSON.stringify({
+                "type":"vectordbEvent",
+                "action": "addVectorItem",
+                "message": {
+                    item: item
+                },
+            }));
+            cbws.getWebsocket.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> => {
+        return new Promise((resolve, reject) => {
+            cbws.getWebsocket.send(JSON.stringify({
+                "type":"vectordbEvent",
+                "action": "queryVectorItem",
+                "message": {
+                    item: key
+                },
+            }));
+            cbws.getWebsocket.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> => {
+        return new Promise((resolve, reject) => {
+            cbws.getWebsocket.send(JSON.stringify({
+                "type":"vectordbEvent",
+                "action": "queryVectorItems",
+                "message": {
+                    items,
+                    dbPath
+                },
+            }));
+            cbws.getWebsocket.on('message', (data: string) => {
+                const response = JSON.parse(data);
+                if (response.type === "qeryVectorItemsResponse") {
+                    resolve(response);
+                }
+            });
+        });
+    },
+};
+
+export default VectorDB;

package/src/modules/websocket.ts

@@ -0,0 +1,89 @@
+import WebSocket from 'ws';
+import fs from 'fs';
+import yaml from 'js-yaml';
+
+/**
+ * Class representing a WebSocket connection.
+ */
+class cbws {
+    websocket: WebSocket;
+
+    /**
+     * Constructs a new cbws instance and initializes the WebSocket connection.
+     */
+    constructor() {
+        const uniqueConnectionId = this.getUniqueConnectionId();
+        const initialMessage = this.getInitialMessage();
+        console.log(uniqueConnectionId)
+        this.websocket = new WebSocket(`ws://localhost:${process.env.SOCKET_PORT}/codebolt?id=${uniqueConnectionId}${process.env.Is_Dev ? '&dev=true' : ''}`);
+        this.initializeWebSocket(initialMessage).catch(error => {
+            console.error("WebSocket connection failed:", error);
+        });
+    }
+    private getUniqueConnectionId(): string {
+        try {
+            let fileContents = fs.readFileSync('./codeboltagent.yaml', 'utf8');
+            let data: any = yaml.load(fileContents);
+            return data.unique_connectionid;
+        } catch (e) {
+            console.error('Unable to locate codeboltagent.yaml file.');
+            return '';
+        }
+    }
+
+    private getInitialMessage(): string {
+        try {
+            let fileContents = fs.readFileSync('./codeboltagent.yaml', 'utf8');
+            let data: any = yaml.load(fileContents);
+            return data.initial_message;
+        } catch (e) {
+            console.error('Unable to locate codeboltagent.yaml file.');
+            return '';
+        }
+    }
+
+    /**
+     * Initializes the WebSocket by setting up event listeners and returning a promise that resolves
+     * when the WebSocket connection is successfully opened.
+     * @returns {Promise<WebSocket>} A promise that resolves with the WebSocket instance.
+     */
+    private async initializeWebSocket(initialMessage: string): Promise<WebSocket> {
+        return new Promise((resolve, reject) => {
+            this.websocket.on('error', (error: Error) => {
+                console.log('WebSocket error:', error);
+                reject(error);
+            });
+
+            this.websocket.on('open', () => {
+                console.log('WebSocket connected');
+                // if (this.websocket) {
+                //     this.websocket.send(JSON.stringify({
+                //         "type": "sendMessage",
+                //         "message": initialMessage
+                //     }));
+                //     resolve(this.websocket);
+                // }
+            });
+
+            this.websocket.on('message', (data: WebSocket.Data) => {
+                // Handle incoming WebSocket messages here.
+                // console.log('WebSocket message received:', data);
+            });
+        });
+    }
+
+    /**
+     * Getter for the WebSocket instance. Throws an error if the WebSocket is not open.
+     * @returns {WebSocket} The WebSocket instance.
+     * @throws {Error} If the WebSocket is not open.
+     */
+    get getWebsocket(): WebSocket {
+        if (!this.websocket.OPEN) {
+            throw new Error('WebSocket is not open');
+        } else {
+            return this.websocket;
+        }
+    }
+}
+
+export default new cbws();