Merge branch 'main' of https://github.com/codeboltai/codeboltjs

Author: utkarshx

.gitignore

@@ -2,4 +2,4 @@ build
 node_modules
 modules
 index.js
-index.d.ts
+index.d.ts

package.json

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

src/index.ts

@@ -92,7 +92,7 @@ class Codebolt  { // Extend EventEmitter
     debug = debug;
     tokenizer = tokenizer;
     chatSummary=chatSummary;
-    codeboltMCP = codeboltMCP;
+    MCP = codeboltMCP;
 }
 
 export default new Codebolt();

src/modules/chat.ts

@@ -14,7 +14,6 @@ let eventEmitter = new CustomEventEmitter()
  * Chat module to interact with the WebSocket server.
  */
 const cbchat = {
-
     /**
      * Retrieves the chat history from the server.
      * @returns {Promise<ChatMessage[]>} A promise that resolves with an array of ChatMessage objects representing the chat history.
@@ -32,7 +31,6 @@ const cbchat = {
             })
         })
     },
-
     /**
      * Sets up a listener for incoming WebSocket messages and emits a custom event when a message is received.
      * @returns {EventEmitter} The event emitter used for emitting custom events.
@@ -53,7 +51,6 @@ const cbchat = {
         });
         return eventEmitter;
     },
-
     /**
      * Sends a message through the WebSocket connection.
      * @param {string} message - The message to be sent.
@@ -66,7 +63,6 @@ const cbchat = {
             payload
         }));
     },
-
     /**
      * Waits for a reply to a sent message.
      * @param {string} message - The message for which a reply is expected.
@@ -86,7 +82,6 @@ const cbchat = {
             });
         });
     },
-
     /**
      * Notifies the server that a process has started and sets up an event listener for stopProcessClicked events.
      * @returns An object containing the event emitter and a stopProcess method.
@@ -143,7 +138,6 @@ const cbchat = {
             "type": "processFinished"
         }));
     },
-
     /**
      * Sends a confirmation request to the server with two options: Yes or No.
      * @returns {Promise<string>} A promise that resolves with the server's response.
@@ -165,6 +159,23 @@ const cbchat = {
             });
         });
     },
+    askQuestion: (question: string, buttons: string[] = [],withFeedback:boolean=false): Promise<string> => {
+        return new Promise((resolve, reject) => {
+            cbws.getWebsocket.send(JSON.stringify({
+                "type": "confirmationRequest",
+                "message": question,
+                buttons: buttons,
+                withFeedback
+
+            }));
+            cbws.getWebsocket.on('message', (data: string) => {
+                const response = JSON.parse(data);
+                if (response.type === "confirmationResponse" || response.type === "feedbackResponse" ) {
+                    resolve(response); // Resolve the Promise with the server's response
+                }
+            });
+        });
+    },
     /**
  * Sends a notification event to the server.
  * @param {string} notificationMessage - The message to be sent in the notification.

src/modules/mcp.ts

@@ -1,15 +1,18 @@
 import cbws from './websocket';
 const codeboltMCP = {
-    getMcpList: (): Promise<any> => {
+
+    executeTool: ( toolName: string, params: any,mcpServer?: string,): Promise<any> => {
         return new Promise((resolve, reject) => {
             cbws.getWebsocket.send(JSON.stringify({
                 "type": "mcpEvent",
-                "action": "getMcpList"
+                "action": "executeTool",
+                "toolName": toolName,
+                "params": params
             }));
             cbws.getWebsocket.on('message', (data: string) => {
                 try {
                     const response = JSON.parse(data);
-                    if (response.type === "getMcpListResponse") {
+                    if (response.type === "executeToolResponse") {
                         resolve(response.data);
                     } else {
                         reject(new Error("Unexpected response type"));
@@ -23,19 +26,17 @@ const codeboltMCP = {
             });
         });
     },
-    executeTool: (mcpServer: string, toolName: string, params: any): Promise<any> => {
+    getMcpTools: (tools: string[]): Promise<any> => {
         return new Promise((resolve, reject) => {
             cbws.getWebsocket.send(JSON.stringify({
                 "type": "mcpEvent",
-                "action": "executeTool",
-                "mcpServer": mcpServer,
-                "toolName": toolName,
-                "params": params
+                "action": "getMcpTools",
+                "tools": tools
             }));
             cbws.getWebsocket.on('message', (data: string) => {
                 try {
                     const response = JSON.parse(data);
-                    if (response.type === "executeToolResponse" && response.toolName === toolName) {
+                    if (response.type === "getMcpToolsResponse") {
                         resolve(response.data);
                     } else {
                         reject(new Error("Unexpected response type"));
@@ -48,7 +49,79 @@ const codeboltMCP = {
                 reject(error);
             });
         });
-    }
+
+    },
+    getAllMCPTools: (mpcName: string): Promise<any> => {
+        return new Promise((resolve, reject) => {
+            cbws.getWebsocket.send(JSON.stringify({
+                "type": "mcpEvent",
+                "action": "getAllMCPTools",
+                "mpcName": mpcName
+            }));
+            cbws.getWebsocket.on('message', (data: string) => {
+                try {
+                    const response = JSON.parse(data);
+                    if (response.type === "getAllMCPToolsResponse") {
+                        resolve(response.data);
+                    } else {
+                        reject(new Error("Unexpected response type"));
+                    }
+                } catch (error) {
+                    reject(new Error("Failed to parse response"));
+                }
+            });
+            cbws.getWebsocket.on('error', (error: Error) => {
+                reject(error);
+            });
+        });
+    },
+    getMCPTool: (name: string): Promise<any> => {
+        return new Promise((resolve, reject) => {
+            cbws.getWebsocket.send(JSON.stringify({
+                "type": "mcpEvent",
+                "action": "getMCPTool",
+                "mcpName": name
+            }));
+            cbws.getWebsocket.on('message', (data: string) => {
+                try {
+                    const response = JSON.parse(data);
+                    if (response.type === "getMCPToolResponse") {
+                        resolve(response.data);
+                    } else {
+                        reject(new Error("Unexpected response type"));
+                    }
+                } catch (error) {
+                    reject(new Error("Failed to parse response"));
+                }
+            });
+            cbws.getWebsocket.on('error', (error: Error) => {
+                reject(error);
+            });
+        });
+    },
+    getEnabledMCPS: (): Promise<any> => {
+        return new Promise((resolve, reject) => {
+            cbws.getWebsocket.send(JSON.stringify({
+                "type": "mcpEvent",
+                "action": "getEnabledMCPS"
+            }));
+            cbws.getWebsocket.on('message', (data: string) => {
+                try {
+                    const response = JSON.parse(data);
+                    if (response.type === "getEnabledMCPSResponse") {
+                        resolve(response.data);
+                    } else {
+                        reject(new Error("Unexpected response type"));
+                    }
+                } catch (error) {
+                    reject(new Error("Failed to parse response"));
+                }
+            });
+            cbws.getWebsocket.on('error', (error: Error) => {
+                reject(error);
+            });
+        });
+    },
 }
 
 export default codeboltMCP;