get agent and start agent added

Author: rawatSingh2022

index.d.ts

@@ -19,11 +19,7 @@ declare class Codebolt {
     waitForConnection(): Promise<void>;
     websocket: WebSocket | null;
     fs: {
-        createFile: (fileName: string, source: string, filePath: string) => Promise<import("@codebolt/types").CreateFileResponse>; /**
-         * @method waitForConnection
-         * @description Waits for the WebSocket connection to open.
-         * @returns {Promise<void>} A promise that resolves when the WebSocket connection is open.
-         */
+        createFile: (fileName: string, source: string, filePath: string) => Promise<import("@codebolt/types").CreateFileResponse>;
         createFolder: (folderName: string, folderPath: string) => Promise<import("@codebolt/types").CreateFolderResponse>;
         readFile: (filePath: string) => Promise<import("@codebolt/types").ReadFileResponse>;
         updateFile: (filename: string, filePath: string, newContent: string) => Promise<import("@codebolt/types").UpdateFileResponse>;
@@ -193,10 +189,6 @@ declare class Codebolt {
     project: {
         getProjectSettings: (output: any) => void;
         getProjectPath: () => Promise<import("@codebolt/types").GetProjectPathResponse>;
-        /**
-         * @constructor
-         * @description Initializes the websocket connection.
-         */
         getRepoMap: (message: any) => Promise<import("@codebolt/types").GetProjectPathResponse>;
         runProject: () => void;
         getEditorFileStatus: () => Promise<unknown>;
@@ -249,6 +241,10 @@ declare class Codebolt {
         getMCPTool: (name: string) => Promise<any>;
         getEnabledMCPS: () => Promise<any>;
     };
+    AGENT: {
+        getAgent: (task: string) => Promise<any>;
+        startAgent: (task: string) => Promise<any>;
+    };
 }
 declare const _default: Codebolt;
 export default _default;

index.js

@@ -28,6 +28,7 @@ const tokenizer_1 = __importDefault(require("./modules/tokenizer"));
 const ws_1 = __importDefault(require("ws"));
 const history_1 = require("./modules/history");
 const mcp_1 = __importDefault(require("./modules/mcp"));
+const agent_1 = __importDefault(require("./modules/agent"));
 /**
  * @class Codebolt
  * @description This class provides a unified interface to interact with various modules.
@@ -62,6 +63,7 @@ class Codebolt {
         this.tokenizer = tokenizer_1.default;
         this.chatSummary = history_1.chatSummary;
         this.MCP = mcp_1.default;
+        this.AGENT = agent_1.default;
         this.websocket = websocket_1.default.getWebsocket;
     }
     /**

modules/agent.d.ts

@@ -0,0 +1,15 @@
+declare const codeboltAgent: {
+    /**
+     * Retrieves an agent based on the specified task.
+     * @param {string} task - The task for which an agent is needed.
+     * @returns {Promise<AgentResponse>} A promise that resolves with the agent details.
+     */
+    getAgent: (task: string) => Promise<any>;
+    /**
+     * Starts an agent for the specified task.
+     * @param {string} task - The task for which the agent should be started.
+     * @returns {Promise<void>} A promise that resolves when the agent has been successfully started.
+     */
+    startAgent: (task: string) => Promise<any>;
+};
+export default codeboltAgent;

modules/agent.js

@@ -0,0 +1,49 @@
+"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 codeboltAgent = {
+    /**
+     * Retrieves an agent based on the specified task.
+     * @param {string} task - The task for which an agent is needed.
+     * @returns {Promise<AgentResponse>} A promise that resolves with the agent details.
+     */
+    getAgent: (task) => {
+        return new Promise((resolve, reject) => {
+            websocket_1.default.getWebsocket.send(JSON.stringify({
+                "type": "agentEvent",
+                "action": "getAgentByTask",
+                "task": task
+            }));
+            websocket_1.default.getWebsocket.on('message', (data) => {
+                const response = JSON.parse(data);
+                if (response.type === "getAgentByTaskResponse") {
+                    resolve(response); // Resolve the Promise with the agent details
+                }
+            });
+        });
+    },
+    /**
+     * Starts an agent for the specified task.
+     * @param {string} task - The task for which the agent should be started.
+     * @returns {Promise<void>} A promise that resolves when the agent has been successfully started.
+     */
+    startAgent: (task) => {
+        return new Promise((resolve, reject) => {
+            websocket_1.default.getWebsocket.send(JSON.stringify({
+                "type": "agentEvent",
+                "action": "startAgent",
+                "task": task
+            }));
+            websocket_1.default.getWebsocket.on('message', (data) => {
+                const response = JSON.parse(data);
+                if (response.type === "startAgentResponse" && response.task === task) {
+                    resolve(response); // Resolve the Promise when the agent has been successfully started
+                }
+            });
+        });
+    }
+};
+exports.default = codeboltAgent;

package.json

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

src/.DS_Store

@@ -24,6 +24,8 @@ import WebSocket from 'ws';
 import { EventEmitter } from 'events';
 import {chatSummary} from './modules/history'
 import codeboltMCP from './modules/mcp';
+import cbagent from './modules/agent';
+
 
 
 /**
@@ -93,6 +95,7 @@ class Codebolt  { // Extend EventEmitter
     tokenizer = tokenizer;
     chatSummary=chatSummary;
     MCP = codeboltMCP;
+    AGENT = cbagent;
 }
 
 export default new Codebolt();

src/index.ts

@@ -0,0 +1,53 @@
+import { GetAgentStateResponse } from '@codebolt/types';
+import cbws from './websocket';
+
+const codeboltAgent = {
+    /**
+     * Retrieves an agent based on the specified task.
+     * @param {string} task - The task for which an agent is needed.
+     * @returns {Promise<AgentResponse>} A promise that resolves with the agent details.
+     */
+    getAgent: (task: string): Promise<any> => {
+        return new Promise((resolve, reject) => {
+            cbws.getWebsocket.send(JSON.stringify({
+                "type": "agentEvent",
+                "action": "getAgentByTask",
+                "task": task
+            }));
+            cbws.getWebsocket.on('message', (data: string) => {
+                const response = JSON.parse(data);
+                if (response.type === "getAgentByTaskResponse") {
+                    resolve(response); // Resolve the Promise with the agent details
+                }
+            });
+        });
+    },
+
+    /**
+     * Starts an agent for the specified task.
+     * @param {string} task - The task for which the agent should be started.
+     * @returns {Promise<void>} A promise that resolves when the agent has been successfully started.
+     */
+    startAgent: (task: string): Promise<any> => {
+        return new Promise((resolve, reject) => {
+            cbws.getWebsocket.send(JSON.stringify({
+                "type": "agentEvent",
+                "action": "startAgent",
+                "task": task
+            }));
+            cbws.getWebsocket.on('message', (data: string) => {
+                const response = JSON.parse(data);
+                if (response.type === "startAgentResponse" && response.task === task) {
+                    resolve(response); // Resolve the Promise when the agent has been successfully started
+                }
+            });
+        });
+    }
+}
+
+
+export default codeboltAgent;
+
+
+
+