@codebolt/codeboltjs 1.1.18 → 1.1.19

package/index.js

@@ -22,6 +22,7 @@ const git_1 = __importDefault(require("./modules/git"));
 const dbmemory_1 = __importDefault(require("./modules/dbmemory"));
 const state_1 = __importDefault(require("./modules/state"));
 const task_1 = __importDefault(require("./modules/task"));
+const vectordb_1 = __importDefault(require("./modules/vectordb"));
 const ws_1 = __importDefault(require("ws"));
 /**
  * @class Codebolt
@@ -52,6 +53,7 @@ class Codebolt {
         this.dbmemory = dbmemory_1.default;
         this.cbstate = state_1.default;
         this.taskplaner = task_1.default;
+        this.vectordb = vectordb_1.default;
         this.websocket = websocket_1.default.getWebsocket;
     }
     /**

package/modules/vectordb.d.ts

@@ -0,0 +1,25 @@
+declare 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<any>} A promise that resolves with the retrieved vector.
+     */
+    getVector: (key: string) => Promise<any>;
+    /**
+     * Adds a new vector item to the vector database.
+     *
+
+     * @param {any} item - The item to add to the vector.
+     * @returns {Promise<any>} A promise that resolves when the item is successfully added.
+     */
+    addVectorItem: (item: any) => Promise<any>;
+    /**
+     * 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<any>} A promise that resolves with the queried vector item.
+     */
+    queryVectorItem: (key: string) => Promise<any>;
+};
+export default VectorDB;

package/modules/vectordb.js

@@ -0,0 +1,76 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const websocket_1 = __importDefault(require("./websocket"));
+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<any>} A promise that resolves with the retrieved vector.
+     */
+    getVector: async (key) => {
+        return new Promise((resolve, reject) => {
+            websocket_1.default.getWebsocket.send(JSON.stringify({
+                "type": "getVector",
+                "message": {
+                    item: key
+                },
+            }));
+            websocket_1.default.getWebsocket.on('message', (data) => {
+                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<any>} A promise that resolves when the item is successfully added.
+     */
+    addVectorItem: async (item) => {
+        return new Promise((resolve, reject) => {
+            websocket_1.default.getWebsocket.send(JSON.stringify({
+                "type": "addVectorItem",
+                "message": {
+                    item: item
+                },
+            }));
+            websocket_1.default.getWebsocket.on('message', (data) => {
+                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<any>} A promise that resolves with the queried vector item.
+     */
+    queryVectorItem: async (key) => {
+        return new Promise((resolve, reject) => {
+            websocket_1.default.getWebsocket.send(JSON.stringify({
+                "type": "queryVectorItem",
+                "message": {
+                    item: key
+                },
+            }));
+            websocket_1.default.getWebsocket.on('message', (data) => {
+                const response = JSON.parse(data);
+                if (response.type === "qeryVectorItemResponse") {
+                    resolve(response);
+                }
+            });
+        });
+    },
+};
+exports.default = VectorDB;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@codebolt/codeboltjs",
-  "version": "1.1.18",
+  "version": "1.1.19",
   "description": "",
   "keywords": [],
   "author": "",

package/src/index.ts

@@ -17,6 +17,8 @@ import git from './modules/git';
 import dbmemory from './modules/dbmemory';
 import cbstate from './modules/state';
 import task from './modules/task';
+
+import vectorDB from './modules/vectordb';
 import WebSocket from 'ws';
 
 
@@ -90,7 +92,8 @@ class Codebolt {
     project = cbproject;
     dbmemory = dbmemory;
     cbstate=cbstate;
-    taskplaner=task
+    taskplaner=task;
+    vectordb=vectorDB;
 }
 
 // export default new Codebolt();

package/src/modules/vectordb.ts

@@ -0,0 +1,75 @@
+import cbws from './websocket';
+
+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<any>} A promise that resolves with the retrieved vector.
+     */
+    getVector: async (key: string): Promise<any> => {
+        return new Promise((resolve, reject) => {
+            cbws.getWebsocket.send(JSON.stringify({
+                "type": "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<any>} A promise that resolves when the item is successfully added.
+     */
+    addVectorItem: async ( item: any): Promise<any> => {
+        return new Promise((resolve, reject) => {
+            cbws.getWebsocket.send(JSON.stringify({
+                "type": "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<any>} A promise that resolves with the queried vector item.
+     */
+    queryVectorItem: async (key: string): Promise<any> => {
+        return new Promise((resolve, reject) => {
+            cbws.getWebsocket.send(JSON.stringify({
+                "type": "queryVectorItem",
+                "message": {
+                    item: key
+                },
+            }));
+            cbws.getWebsocket.on('message', (data: string) => {
+                const response = JSON.parse(data);
+                if (response.type === "qeryVectorItemResponse") {
+                    resolve(response);
+                }
+            });
+        });
+    },
+};
+
+export default VectorDB;