@codebolt/codeboltjs 1.1.92 → 1.1.95

package/src/modules/agentlib/agent.ts

@@ -1,5 +1,5 @@
 import chatlib from "./../chat"
-import mcp from "./../mcp"
+import tools from "./../tools"
 import llm from "./../llm"
 import codeboltAgent from "./../agent"
 import { SystemPrompt } from "./systemprompt";
@@ -16,6 +16,7 @@ interface ToolResult {
     role: 'tool';
     tool_call_id: string;
     content: any;
+    userMessage?: any;
 }
 
 interface ToolDetails {
@@ -26,28 +27,20 @@ interface ToolDetails {
 
 class Agent {
     private tools: any[];
-    private subAgents: any[];
+    // private subAgents: any[];
     private apiConversationHistory: Message[];
     private maxRun: number;
     private systemPrompt: SystemPrompt;
     private userMessage: Message[];
+    private nextUserMessage:any;
 
 
-    constructor(tools: any = [], systemPrompt: SystemPrompt, maxRun: number = 0, subAgents: any[] = []) {
+    constructor(tools: any = [], systemPrompt: SystemPrompt, maxRun: number = 0) {
         this.tools = tools;
         this.userMessage = [];
         this.apiConversationHistory = [];
         this.maxRun = maxRun;
         this.systemPrompt = systemPrompt;
-        this.subAgents = subAgents;
-        this.subAgents = subAgents.map(subagent => {
-            subagent.function.name = `subagent--${subagent.function.name}`;
-            return subagent;
-        });
-        this.tools = this.tools.concat(subAgents.map(subagent => ({
-            ...subagent
-        })));
-
 
     }
 
@@ -55,10 +48,36 @@ class Agent {
 
 
         let mentaionedMCPSTool: any[] = await task.userMessage.getMentionedMcpsTools();
+
         this.tools = [
             ...this.tools,
-            ...mentaionedMCPSTool
+            ...mentaionedMCPSTool,
+    
         ]
+        let mentionedAgents = await task.userMessage.getMentionedAgents();
+
+        // Transform agents into tool format
+        const agentTools = mentionedAgents.map(agent => {
+            return {
+                type: "function",
+                function: {
+                    name: `subagent--${agent.unique_id}`,
+                    description: agent.longDescription || agent.description,
+                    parameters: {
+                        type: "object",
+                        properties: {
+                            task: {
+                                type: "string",
+                                description: "The task to be executed by the tool."
+                            }
+                        },
+                        required: ["task"]
+                    }
+                }
+            };
+        });
+        
+        this.tools = this.tools.concat(agentTools);
 
 
         let completed = false;
@@ -112,20 +131,44 @@ class Agent {
 
                                         const agentResponse = await codeboltAgent.startAgent(toolName.replace("subagent--", ''), toolInput.task);
                                         console.log("got sub agent resonse  result", agentResponse);
-                                        const [didUserReject, result] = [false,"tool result is successful"];
-                                       console.log("got sub agent resonse  result", didUserReject, result); 
-                                      
-                                        toolResults.push(this.getToolResult(toolUseId, result));
+                                        const [didUserReject, result] = [false, "tool result is successful"];
+                                        console.log("got sub agent resonse  result", didUserReject, result);
+                                        let toolResult = this.getToolResult(toolUseId, result)
+                                        toolResults.push({
+                                            role: "tool",
+                                            tool_call_id: toolResult.tool_call_id,
+                                            content: toolResult.content,
+
+                                        });
+                                        if (toolResult.userMessage) {
+                                            this.nextUserMessage = {
+                                                role: "user",
+                                                content: toolResult.userMessage
+                                            }
+                                        }
                                         if (didUserReject) {
                                             userRejectedToolUse = true;
                                         }
+
                                     }
                                     else {
                                         console.log("calling tool with params", toolName, toolInput);
                                         const [didUserReject, result] = await this.executeTool(toolName, toolInput);
                                         console.log("tool result", result);
-                                        toolResults.push(this.getToolResult(toolUseId, result));
-
+                                        // toolResults.push(this.getToolResult(toolUseId, result));
+                                        let toolResult = this.getToolResult(toolUseId, result)
+                                        toolResults.push({
+                                            role: "tool",
+                                            tool_call_id: toolResult.tool_call_id,
+                                            content: toolResult.content,
+
+                                        });
+                                        if (toolResult.userMessage) {
+                                            this.nextUserMessage = {
+                                                role: "user",
+                                                content: toolResult.userMessage
+                                            }
+                                        }
                                         if (didUserReject) {
                                             userRejectedToolUse = true;
                                         }
@@ -133,7 +176,21 @@ class Agent {
 
                                 }
                             } else {
-                                toolResults.push(this.getToolResult(toolUseId, "Skipping tool execution due to previous tool user rejection."));
+                                let toolResult = this.getToolResult(toolUseId, "Skipping tool execution due to previous tool user rejection.")
+
+                                toolResults.push({
+                                    role: "tool",
+                                    tool_call_id: toolResult.tool_call_id,
+                                    content: toolResult.content,
+
+                                });
+                                if (toolResult.userMessage) {
+                                    this.nextUserMessage = {
+                                        role: "user",
+                                        content: toolResult.userMessage
+                                    }
+                                }
+
                             }
                         }
                     }
@@ -148,10 +205,26 @@ class Agent {
                             completed = true;
                             result = "The user is satisfied with the result.";
                         }
-                        toolResults.push(this.getToolResult(taskCompletedBlock.id, result));
+                        let toolResult = this.getToolResult(taskCompletedBlock.id, result)
+                        toolResults.push({
+                            role: "tool",
+                            tool_call_id: toolResult.tool_call_id,
+                            content: toolResult.content,
+
+                        });
+                        if (toolResult.userMessage) {
+                            this.nextUserMessage = {
+                                role: "user",
+                                content: toolResult.userMessage
+                            }
+                        }
+
                     }
 
                     this.apiConversationHistory.push(...toolResults);
+                    if (this.nextUserMessage) {
+                        this.apiConversationHistory.push(this.nextUserMessage);
+                    }
                     let nextUserMessage: Message[] = toolResults;
 
                     if (toolResults.length === 0) {
@@ -212,7 +285,9 @@ class Agent {
     }
 
     private async executeTool(toolName: string, toolInput: any): Promise<[boolean, any]> {
-        return mcp.executeTool(toolName, toolInput);
+        //codebolttools--readfile
+        const [toolboxName, actualToolName] = toolName.split('--');
+        return tools.executeTool(toolboxName, actualToolName, toolInput);
     }
     private async startSubAgent(agentName: string, params: any): Promise<[boolean, any]> {
         return codeboltAgent.startAgent(agentName, params.task);
@@ -226,11 +301,24 @@ class Agent {
         };
     }
 
-    private getToolResult(tool_call_id: string, content: any): ToolResult {
+    private getToolResult(tool_call_id: string, content: string): ToolResult {
+        let userMessage=undefined
+        try {
+            let parsed = JSON.parse(content);
+
+            if (parsed.payload && parsed.payload.content) {
+                content = `The browser action has been executed. The  screenshot have been captured for your analysis. The tool response is provided in the next user message`
+                // this.apiConversationHistory.push()
+                userMessage = parsed.payload.content
+            }
+        } catch (error) {
+
+        }
         return {
             role: "tool",
             tool_call_id,
             content,
+            userMessage
         };
     }
 

package/src/modules/agentlib/usermessage.ts

@@ -1,13 +1,21 @@
 import cbfs from "./../fs";
 import project from "./../project";
-import mcp from "./../mcp";
-import { escape } from "querystring";
-
-
+import mcp from "./../tools";
+interface agent {
+  
+        description: string;
+        title: string;
+        id: number;
+        agent_id: string;
+        unique_id: string;
+        longDescription: string;
+    
+}
 interface Message {
     userMessage: string;
     mentionedFiles?: string[];
     mentionedMCPs: string[];
+    mentionedAgents: agent[];
 }
 
 export interface UserMessageContent {
@@ -24,13 +32,13 @@ class UserMessage {
     message: Message;
     promptOverride: boolean;
     userMessages: UserMessageContent[];
-    mentaionedMCPS: string[];
+    mentionedMCPs: string[];
 
     constructor(message: Message, promptOverride: boolean = false) {
         this.message = message;
         this.promptOverride = promptOverride;
         this.userMessages = [];
-        this.mentaionedMCPS = message.mentionedMCPs || [];
+        this.mentionedMCPs = message.mentionedMCPs || [];
     }
 
     getFiles(): void {
@@ -70,12 +78,17 @@ class UserMessage {
         return this.userMessages;
     }
 
-    getMentionedMcps(): string[] {
-        return this.message.mentionedMCPs || [];
+    getMentionedAgents() {
+        //TODO : get config in tool format if neede
+        return this.message.mentionedAgents || [];
+    }
+
+    getMentionedMcps() {
+      return this.message.mentionedMCPs || [];
     }
     async getMentionedMcpsTools() {
-        if (this.mentaionedMCPS.length > 0) {
-            let tools = await mcp.getMcpTools(this.mentaionedMCPS)
+        if (this.mentionedMCPs.length > 0) {
+            let tools = await mcp.listToolsFromToolBoxes(this.mentionedMCPs)
             return tools
         }
         else {