@codebolt/codeboltjs 2.0.4 → 2.0.5

package/dist/modules/vectordb.js

@@ -3,7 +3,7 @@ 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 websocket_1 = __importDefault(require("../core/websocket"));
 const VectorDB = {
     /**
      * Retrieves a vector from the vector database based on the provided key.
@@ -12,21 +12,13 @@ const VectorDB = {
      * @returns {Promise<GetVectorResponse>} A promise that resolves with the retrieved vector.
      */
     getVector: async (key) => {
-        return new Promise((resolve, reject) => {
-            websocket_1.default.getWebsocket.send(JSON.stringify({
-                "type": "vectordbEvent",
-                "action": "getVector",
-                "message": {
-                    item: key
-                },
-            }));
-            websocket_1.default.getWebsocket.on('message', (data) => {
-                const response = JSON.parse(data);
-                if (response.type === "getVectorResponse") {
-                    resolve(response);
-                }
-            });
-        });
+        return websocket_1.default.messageManager.sendAndWaitForResponse({
+            "type": "vectordbEvent",
+            "action": "getVector",
+            "message": {
+                item: key
+            },
+        }, "getVectorResponse");
     },
     /**
      * Adds a new vector item to the vector database.
@@ -36,21 +28,13 @@ const VectorDB = {
      * @returns {Promise<AddVectorItemResponse>} 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": "vectordbEvent",
-                "action": "addVectorItem",
-                "message": {
-                    item: item
-                },
-            }));
-            websocket_1.default.getWebsocket.on('message', (data) => {
-                const response = JSON.parse(data);
-                if (response.type === "addVectorItemResponse") {
-                    resolve(response);
-                }
-            });
-        });
+        return websocket_1.default.messageManager.sendAndWaitForResponse({
+            "type": "vectordbEvent",
+            "action": "addVectorItem",
+            "message": {
+                item: item
+            },
+        }, "addVectorItemResponse");
     },
     /**
      * Queries a vector item from the vector database based on the provided key.
@@ -59,21 +43,13 @@ const VectorDB = {
      * @returns {Promise<QueryVectorItemResponse>} 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": "vectordbEvent",
-                "action": "queryVectorItem",
-                "message": {
-                    item: key
-                },
-            }));
-            websocket_1.default.getWebsocket.on('message', (data) => {
-                const response = JSON.parse(data);
-                if (response.type === "qeryVectorItemResponse") {
-                    resolve(response);
-                }
-            });
-        });
+        return websocket_1.default.messageManager.sendAndWaitForResponse({
+            "type": "vectordbEvent",
+            "action": "queryVectorItem",
+            "message": {
+                item: key
+            },
+        }, "qeryVectorItemResponse");
     },
     /**
      * Queries a vector item from the vector database based on the provided key.
@@ -82,22 +58,14 @@ const VectorDB = {
      * @returns {Promise<QueryVectorItemResponse>} A promise that resolves with the queried vector item.
      */
     queryVectorItems: async (items, dbPath) => {
-        return new Promise((resolve, reject) => {
-            websocket_1.default.getWebsocket.send(JSON.stringify({
-                "type": "vectordbEvent",
-                "action": "queryVectorItems",
-                "message": {
-                    items,
-                    dbPath
-                },
-            }));
-            websocket_1.default.getWebsocket.on('message', (data) => {
-                const response = JSON.parse(data);
-                if (response.type === "qeryVectorItemsResponse") {
-                    resolve(response);
-                }
-            });
-        });
+        return websocket_1.default.messageManager.sendAndWaitForResponse({
+            "type": "vectordbEvent",
+            "action": "queryVectorItems",
+            "message": {
+                items,
+                dbPath
+            },
+        }, "qeryVectorItemsResponse");
     },
 };
 exports.default = VectorDB;

package/dist/utils.d.ts

@@ -1,4 +1,4 @@
-export { ToolBox } from './modules/toolBox';
+export { ToolBox } from './utils/toolBox';
 export { TaskInstruction } from './agentlib/taskInstruction';
 export { UserMessage } from './agentlib/usermessage';
 export { SystemPrompt } from './agentlib/systemprompt';

package/dist/utils.js

@@ -1,7 +1,7 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.Agent = exports.SystemPrompt = exports.UserMessage = exports.TaskInstruction = exports.ToolBox = void 0;
-var toolBox_1 = require("./modules/toolBox");
+var toolBox_1 = require("./utils/toolBox");
 Object.defineProperty(exports, "ToolBox", { enumerable: true, get: function () { return toolBox_1.ToolBox; } });
 var taskInstruction_1 = require("./agentlib/taskInstruction");
 Object.defineProperty(exports, "TaskInstruction", { enumerable: true, get: function () { return taskInstruction_1.TaskInstruction; } });

package/package.json

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

package/dist/utils/editFile.js

@@ -1,30 +0,0 @@
-"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("../modules/websocket"));
-const cbutils = {
-    editFileAndApplyDiff: (filePath, diff, diffIdentifier, prompt, applyModel) => {
-        return new Promise((resolve, reject) => {
-            websocket_1.default.getWebsocket.send(JSON.stringify({
-                "type": "fsEvent",
-                "action": "editFileAndApplyDiff",
-                message: {
-                    filePath,
-                    diff,
-                    diffIdentifier,
-                    prompt,
-                    applyModel
-                }
-            }));
-            websocket_1.default.getWebsocket.on('message', (data) => {
-                const response = JSON.parse(data);
-                if (response.type === "editFileAndApplyDiffResponse") {
-                    resolve(response);
-                }
-            });
-        });
-    }
-};
-exports.default = cbutils;