changes

Author: rawatSingh2022

index.d.ts

@@ -237,14 +237,21 @@ declare class Codebolt {
             content: string;
         }[]>;
     };
-    MCP: {
-        executeTool: (toolName: string, params: any, mcpServer?: string | undefined) => Promise<any>;
-        getMcpTools: (tools: string[]) => Promise<any>;
-        getAllMCPTools: (mpcName: string) => Promise<any>;
-        getMCPTool: (name: string) => Promise<any>;
-        getEnabledMCPS: () => Promise<any>;
-    };
-    AGENT: {
+    tools: {
+        getEnabledToolBoxes: () => Promise<any>;
+        getLocalToolBoxes: () => Promise<any>;
+        getMentionedToolBoxes: (userMessage: import("./utils").UserMessage) => Promise<any>;
+        getAvailableToolBoxes: () => Promise<any>;
+        searchAvailableToolBoxes: (query: string) => Promise<any>;
+        listToolsFromToolBoxes: (toolBoxes: string[]) => Promise<any>;
+        configureToolBox: (name: string, config: any) => Promise<any>;
+        getTools: (tools: {
+            toolbox: string;
+            toolName: string;
+        }[]) => Promise<any[]>;
+        executeTool: (toolbox: string, toolName: string, params: any) => Promise<any>;
+    };
+    agent: {
         findAgent: (task: string, maxResult: number | undefined, agents: never[] | undefined, agentLocaltion: import("./modules/agent").AgentLocation | undefined, getFrom: import("./modules/agent").FilterUsing.USE_VECTOR_DB) => Promise<any>;
         startAgent: (agentId: string, task: string) => Promise<any>;
         getAgentsList: (type?: import("./modules/agent").Agents) => Promise<any>;

index.js

@@ -27,7 +27,7 @@ const debug_1 = __importDefault(require("./modules/debug"));
 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 tools_1 = __importDefault(require("./modules/tools"));
 const agent_1 = __importDefault(require("./modules/agent"));
 /**
  * @class Codebolt
@@ -62,8 +62,8 @@ class Codebolt {
         this.debug = debug_1.default;
         this.tokenizer = tokenizer_1.default;
         this.chatSummary = history_1.chatSummary;
-        this.MCP = mcp_1.default;
-        this.AGENT = agent_1.default;
+        this.tools = tools_1.default;
+        this.agent = agent_1.default;
         websocket_1.default.initializeWebSocket();
         this.websocket = websocket_1.default.getWebsocket;
     }

modules/agentlib/agent.js

@@ -5,7 +5,7 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.Agent = void 0;
 const chat_1 = __importDefault(require("./../chat"));
-const mcp_1 = __importDefault(require("./../mcp"));
+const tools_1 = __importDefault(require("./../tools"));
 const llm_1 = __importDefault(require("./../llm"));
 const agent_1 = __importDefault(require("./../agent"));
 class Agent {
@@ -160,7 +160,9 @@ class Agent {
         }
     }
     async executeTool(toolName, toolInput) {
-        return mcp_1.default.executeTool(toolName, toolInput);
+        //codebolttools--readfile
+        const [toolboxName, actualToolName] = toolName.split('--');
+        return tools_1.default.executeTool(toolboxName, actualToolName, toolInput);
     }
     async startSubAgent(agentName, params) {
         return agent_1.default.startAgent(agentName, params.task);

modules/agentlib/usermessage.d.ts

@@ -11,7 +11,7 @@ declare class UserMessage {
     message: Message;
     promptOverride: boolean;
     userMessages: UserMessageContent[];
-    mentaionedMCPS: string[];
+    mentionedMCPs: string[];
     constructor(message: Message, promptOverride?: boolean);
     getFiles(): void;
     toPrompt(bAttachFiles?: boolean, bAttachImages?: boolean, bAttachEnvironment?: boolean): Promise<UserMessageContent[]>;

modules/agentlib/usermessage.js

@@ -6,7 +6,7 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.UserMessage = void 0;
 const fs_1 = __importDefault(require("./../fs"));
 const project_1 = __importDefault(require("./../project"));
-const mcp_1 = __importDefault(require("./../mcp"));
+const tools_1 = __importDefault(require("./../tools"));
 class UserMessage {
     constructor(message, promptOverride = false) {
         this.getEnvironmentDetail = async (cwd) => {
@@ -21,7 +21,7 @@ class UserMessage {
         this.message = message;
         this.promptOverride = promptOverride;
         this.userMessages = [];
-        this.mentaionedMCPS = message.mentionedMCPs || [];
+        this.mentionedMCPs = message.mentionedMCPs || [];
     }
     getFiles() {
         // Implementation to be added
@@ -58,8 +58,8 @@ class UserMessage {
         return this.message.mentionedMCPs || [];
     }
     async getMentionedMcpsTools() {
-        if (this.mentaionedMCPS.length > 0) {
-            let tools = await mcp_1.default.getMcpTools(this.mentaionedMCPS);
+        if (this.mentionedMCPs.length > 0) {
+            let tools = await tools_1.default.listToolsFromToolBoxes(this.mentionedMCPs);
             return tools;
         }
         else {

modules/mcp.d.ts

@@ -4,5 +4,6 @@ declare const codeboltMCP: {
     getAllMCPTools: (mpcName: string) => Promise<any>;
     getMCPTool: (name: string) => Promise<any>;
     getEnabledMCPS: () => Promise<any>;
+    configureMCPTool: (name: string, config: any) => Promise<any>;
 };
 export default codeboltMCP;

modules/mcp.js

@@ -120,5 +120,29 @@ const codeboltMCP = {
             });
         });
     },
+    configureMCPTool: (name, config) => {
+        return new Promise((resolve, reject) => {
+            websocket_1.default.getWebsocket.send(JSON.stringify({
+                "type": "mcpEvent",
+                "action": "configureMCPTool",
+                "mcpName": name,
+                "config": config
+            }));
+            websocket_1.default.getWebsocket.on('message', (data) => {
+                try {
+                    const response = JSON.parse(data);
+                    if (response.type === "configureMCPToolResponse") {
+                        resolve(response.data);
+                    }
+                }
+                catch (error) {
+                    reject(new Error("Failed to parse response"));
+                }
+            });
+            websocket_1.default.getWebsocket.on('error', (error) => {
+                reject(error);
+            });
+        });
+    },
 };
 exports.default = codeboltMCP;