@codebolt/codeboltjs 1.1.82 → 1.1.84

package/index.d.ts

@@ -243,7 +243,7 @@ declare class Codebolt {
     };
     AGENT: {
         getAgent: (task: string) => Promise<any>;
-        startAgent: (task: string) => Promise<any>;
+        startAgent: (agentId: string, task: string) => Promise<any>;
     };
 }
 declare const _default: Codebolt;

package/modules/agent.d.ts

@@ -10,6 +10,6 @@ declare const codeboltAgent: {
      * @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>;
+    startAgent: (agentId: string, task: string) => Promise<any>;
 };
 export default codeboltAgent;

package/modules/agent.js

@@ -30,16 +30,17 @@ const codeboltAgent = {
      * @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) => {
+    startAgent: (agentId, task) => {
         return new Promise((resolve, reject) => {
             websocket_1.default.getWebsocket.send(JSON.stringify({
                 "type": "agentEvent",
                 "action": "startAgent",
+                "agentId": agentId,
                 "task": task
             }));
             websocket_1.default.getWebsocket.on('message', (data) => {
                 const response = JSON.parse(data);
-                if (response.type === "startAgentResponse" && response.task === task) {
+                if (response.type === "taskCompletionResponse") {
                     resolve(response); // Resolve the Promise when the agent has been successfully started
                 }
             });

package/modules/websocket.js

@@ -16,8 +16,9 @@ class cbws {
     constructor() {
         const uniqueConnectionId = this.getUniqueConnectionId();
         const initialMessage = this.getInitialMessage();
-        console.log(uniqueConnectionId);
-        this.websocket = new ws_1.default(`ws://localhost:${process.env.SOCKET_PORT}/codebolt?id=${uniqueConnectionId}${process.env.Is_Dev ? '&dev=true' : ''}`);
+        const agentIdParam = process.env.AGENT_ID ? `&agentId=${process.env.AGENT_ID}` : '';
+        const parentIdParam = process.env.PARENT_ID ? `&parentId=${process.env.PARENT_ID}` : '';
+        this.websocket = new ws_1.default(`ws://localhost:${process.env.SOCKET_PORT}/codebolt?id=${uniqueConnectionId}${agentIdParam}${parentIdParam}${process.env.Is_Dev ? '&dev=true' : ''}`);
         this.initializeWebSocket(initialMessage).catch(error => {
             console.error("WebSocket connection failed:", error);
         });

package/package.json

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

package/src/modules/agent.ts

@@ -28,16 +28,17 @@ const codeboltAgent = {
      * @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> => {
+    startAgent: (agentId: string, task: string): Promise<any> => {
         return new Promise((resolve, reject) => {
             cbws.getWebsocket.send(JSON.stringify({
                 "type": "agentEvent",
                 "action": "startAgent",
+                "agentId": agentId,
                 "task": task
             }));
             cbws.getWebsocket.on('message', (data: string) => {
                 const response = JSON.parse(data);
-                if (response.type === "startAgentResponse" && response.task === task) {
+                if (response.type === "taskCompletionResponse") {
                     resolve(response); // Resolve the Promise when the agent has been successfully started
                 }
             });

package/src/modules/websocket.ts

@@ -14,8 +14,10 @@ class cbws {
     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' : ''}`);
+   
+        const agentIdParam = process.env.AGENT_ID ? `&agentId=${process.env.AGENT_ID}` : '';
+        const parentIdParam = process.env.PARENT_ID ? `&parentId=${process.env.PARENT_ID}` : '';
+        this.websocket = new WebSocket(`ws://localhost:${process.env.SOCKET_PORT}/codebolt?id=${uniqueConnectionId}${agentIdParam}${parentIdParam}${process.env.Is_Dev ? '&dev=true' : ''}`);
         this.initializeWebSocket(initialMessage).catch(error => {
             console.error("WebSocket connection failed:", error);
         });