@codebolt/agent 1.2.1 → 2.2.2

package/dist/builderpattern/agent.js

@@ -0,0 +1,324 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.Agent = void 0;
+const codeboltjs_1 = __importDefault(require("@codebolt/codeboltjs"));
+const { chat, mcp, llm, agent: codeboltAgent } = codeboltjs_1.default;
+// All interfaces moved to libFunctionTypes.ts
+/**
+ * Agent class that manages conversations with LLMs and tool executions.
+ * Handles the conversation flow, tool calls, and task completions.
+ */
+class Agent {
+    /**
+     * Creates a new Agent instance.
+     *
+     * @param tools - The tools available to the agent
+     * @param systemPrompt - The system prompt providing instructions to the LLM
+     * @param maxRun - Maximum number of conversation turns (0 means unlimited)
+     */
+    constructor(tools = [], systemPrompt, maxRun = 0) {
+        this.tools = tools;
+        this.userMessage = [];
+        this.apiConversationHistory = [];
+        this.maxRun = maxRun;
+        this.systemPrompt = systemPrompt;
+    }
+    /**
+     * Runs the agent on a specific task until completion or max runs reached.
+     *
+     * @param task - The task instruction to be executed
+     * @param successCondition - Optional function to determine if the task is successful
+     * @returns Promise with success status, error (if any), and the last assistant message
+     */
+    async run(task, successCondition = () => true) {
+        var _a, _b;
+        let mentaionedMCPSTool = await task.userMessage.getMentionedMcpsTools();
+        this.tools = [
+            ...this.tools,
+            ...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;
+        let userMessages = await task.toPrompt();
+        this.apiConversationHistory.push({ role: "user", content: userMessages });
+        let runcomplete = 0;
+        while (!completed && (runcomplete <= this.maxRun || this.maxRun === 0)) {
+            try {
+                runcomplete++;
+                const response = await this.attemptLlmRequest(this.apiConversationHistory, this.tools);
+                let isMessagePresentinReply = false;
+                for (const contentBlock of response.choices) {
+                    if (contentBlock.message) {
+                        isMessagePresentinReply = true;
+                        this.apiConversationHistory.push(contentBlock.message);
+                        if (contentBlock.message.content != null) {
+                            await chat.sendMessage(contentBlock.message.content, {});
+                        }
+                    }
+                }
+                if (!isMessagePresentinReply) {
+                    this.apiConversationHistory.push({
+                        role: "assistant",
+                        content: [{ type: "text", text: "Failure: I did not provide a response." }],
+                    });
+                }
+                try {
+                    let toolResults = [];
+                    let taskCompletedBlock;
+                    let userRejectedToolUse = false;
+                    const contentBlock = response.choices[0];
+                    if ((_a = contentBlock.message) === null || _a === void 0 ? void 0 : _a.tool_calls) {
+                        for (const tool of contentBlock.message.tool_calls) {
+                            try {
+                                const { toolInput, toolName, toolUseId } = this.getToolDetail(tool);
+                                if (!userRejectedToolUse) {
+                                    if (toolName.includes("attempt_completion")) {
+                                        taskCompletedBlock = tool;
+                                    }
+                                    else {
+                                        let [serverName] = toolName.replace('--', ':').split(':');
+                                        if (serverName == 'subagent') {
+                                            const agentResponse = await codeboltAgent.startAgent(toolName.replace("subagent--", ''), toolInput.task);
+                                            const [didUserReject, result] = [false, "tool result is successful"];
+                                            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 {
+                                            const [didUserReject, result] = await this.executeTool(toolName, toolInput);
+                                            // 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;
+                                            }
+                                        }
+                                    }
+                                }
+                                else {
+                                    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
+                                        };
+                                    }
+                                }
+                            }
+                            catch (error) {
+                                toolResults.push({
+                                    role: "tool",
+                                    tool_call_id: tool.id,
+                                    content: `please provide valid json string for tool.function.arguments String(error)`,
+                                });
+                            }
+                        }
+                    }
+                    if (taskCompletedBlock) {
+                        let [_, result] = await this.executeTool(taskCompletedBlock.function.name, JSON.parse(taskCompletedBlock.function.arguments || "{}"));
+                        if (result === "") {
+                            completed = true;
+                            result = "The user is satisfied with the 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 = toolResults;
+                    if (toolResults.length === 0) {
+                        nextUserMessage = [{
+                                role: "user",
+                                content: [{
+                                        type: "text",
+                                        text: "If you have completed the user's task, use the attempt_completion tool. If you require additional information from the user, use the ask_followup_question tool. Otherwise, if you have not completed the task and do not need additional information, then proceed with the next step of the task. (This is an automated message, so do not respond to it conversationally.)"
+                                    }]
+                            }];
+                        if (nextUserMessage) {
+                            this.apiConversationHistory.push(nextUserMessage[0]);
+                        }
+                    }
+                }
+                catch (error) {
+                    console.error("Error in agent tool call:", error);
+                    return { success: false, error: error instanceof Error ? error.message : String(error), message: null };
+                }
+            }
+            catch (error) {
+                console.error("Error in agent tool call:", error);
+                return { success: false, error: error instanceof Error ? error.message : String(error), message: null };
+            }
+        }
+        return {
+            success: completed,
+            error: null,
+            message: ((_b = this.apiConversationHistory
+                .filter(msg => msg.role === 'assistant')
+                .pop()) === null || _b === void 0 ? void 0 : _b.content) || ''
+        };
+    }
+    /**
+     * Attempts to make a request to the LLM with conversation history and tools.
+     *
+     * @param apiConversationHistory - The current conversation history
+     * @param tools - The tools available to the LLM
+     * @returns Promise with the LLM response
+     */
+    async attemptLlmRequest(apiConversationHistory, tools) {
+        try {
+            let systemPrompt = await this.systemPrompt.toPromptText();
+            const aiMessages = [
+                { role: "system", content: systemPrompt },
+                ...apiConversationHistory,
+            ];
+            const createParams = {
+                full: true,
+                messages: aiMessages,
+                tools: tools,
+                tool_choice: "auto",
+            };
+            //@ts-ignore
+            const { completion } = await llm.inference(createParams);
+            return completion;
+        }
+        catch (error) {
+            return this.attemptApiRequest();
+        }
+    }
+    /**
+     * Executes a tool with given name and input.
+     *
+     * @param toolName - The name of the tool to execute
+     * @param toolInput - The input parameters for the tool
+     * @returns Promise with tuple [userRejected, result]
+     */
+    async executeTool(toolName, toolInput) {
+        //codebolttools--readfile
+        const [toolboxName, actualToolName] = toolName.split('--');
+        console.log("Toolbox name: ", toolboxName, "Actual tool name: ", actualToolName);
+        const { data } = await mcp.executeTool(toolboxName, actualToolName, toolInput);
+        console.log("Tool result: ", data);
+        return data;
+    }
+    /**
+     * Starts a sub-agent to handle a specific task.
+     *
+     * @param agentName - The name of the sub-agent to start
+     * @param params - Parameters for the sub-agent
+     * @returns Promise with tuple [userRejected, result]
+     */
+    async startSubAgent(agentName, params) {
+        return [false, await codeboltAgent.startAgent(agentName, params.task)];
+    }
+    /**
+     * Extracts tool details from a tool call object.
+     *
+     * @param tool - The tool call object from the LLM response
+     * @returns ToolDetails object with name, input, and ID
+     */
+    getToolDetail(tool) {
+        return {
+            toolName: tool.function.name,
+            toolInput: JSON.parse(tool.function.arguments || "{}"),
+            toolUseId: tool.id
+        };
+    }
+    /**
+     * Creates a tool result object from the tool execution response.
+     *
+     * @param tool_call_id - The ID of the tool call
+     * @param content - The content returned by the tool
+     * @returns ToolResult object
+     */
+    getToolResult(tool_call_id, content) {
+        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
+        };
+    }
+    /**
+     * Fallback method for API requests in case of failures.
+     *
+     * @throws Error API request fallback not implemented
+     */
+    attemptApiRequest() {
+        throw new Error("API request fallback not implemented");
+    }
+}
+exports.Agent = Agent;

package/dist/builderpattern/followupquestionbuilder.d.ts

@@ -0,0 +1,75 @@
+import type { OpenAIMessage, OpenAITool, ToolResult, CodeboltAPI } from "../types/libFunctionTypes";
+/**
+ * Builds follow-up prompts for continuing conversations with tool results.
+ * Manages conversation history and summarization when conversations get too long.
+ */
+declare class FollowUpPromptBuilder {
+    /** Previous conversation messages */
+    private previousConversation;
+    /** Tool results to add to the conversation */
+    private toolResults;
+    /** Available tools for the conversation */
+    private tools;
+    /** The last LLM response, if available */
+    private llmResponse?;
+    /** Maximum conversation length before summarization */
+    private maxConversationLength;
+    /** Whether to force summarization */
+    private forceSummarization;
+    /** Codebolt API instance */
+    private codebolt?;
+    /**
+     * Creates a new FollowUpQuestionBuilder instance.
+     *
+     * @param codebolt - Optional codebolt API instance
+     */
+    constructor(codebolt?: CodeboltAPI);
+    /**
+     * Adds the previous conversation to the builder.
+     *
+     * @param previousPrompt - The previous prompt object containing messages and tools
+     * @returns The FollowUpQuestionBuilder instance for chaining
+     */
+    addPreviousConversation(previousPrompt: {
+        messages: OpenAIMessage[];
+        tools: OpenAITool[];
+        tool_choice?: string;
+    }, llmResponse: {
+        completion: any;
+    }): this;
+    addLLMResponseToConverstaion(llmResponse: {
+        completion: any;
+    }): this;
+    /**
+     * Adds tool execution results to the conversation.
+     *
+     * @param toolResults - Array of tool execution results
+     * @returns The FollowUpQuestionBuilder instance for chaining
+     */
+    addToolResult(toolResults: ToolResult[]): this;
+    /**
+     * Checks if the conversation is too long and sets up summarization with custom max length.
+     *
+     * @param maxLength - Maximum number of messages before summarization
+     * @returns The FollowUpQuestionBuilder instance for chaining
+     */
+    checkAndSummarizeConversationIfLong(maxLength: number): this;
+    /**
+     * Performs conversation summarization if needed.
+     *
+     * @returns Promise that resolves to the summarized messages
+     */
+    private performSummarization;
+    /**
+     * Builds the follow-up conversation prompt with tool results.
+     *
+     * @returns Promise that resolves to the conversation prompt object
+     */
+    build(): Promise<{
+        messages: OpenAIMessage[];
+        tools: OpenAITool[];
+        tool_choice: "auto";
+        full: boolean;
+    }>;
+}
+export { FollowUpPromptBuilder };

package/dist/builderpattern/followupquestionbuilder.js

@@ -0,0 +1,197 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.FollowUpPromptBuilder = void 0;
+const codeboltjs_1 = __importDefault(require("@codebolt/codeboltjs"));
+const { chatSummary } = codeboltjs_1.default;
+/**
+ * Builds follow-up prompts for continuing conversations with tool results.
+ * Manages conversation history and summarization when conversations get too long.
+ */
+class FollowUpPromptBuilder {
+    /**
+     * Creates a new FollowUpQuestionBuilder instance.
+     *
+     * @param codebolt - Optional codebolt API instance
+     */
+    constructor(codebolt) {
+        /** Previous conversation messages */
+        this.previousConversation = [];
+        /** Tool results to add to the conversation */
+        this.toolResults = [];
+        /** Available tools for the conversation */
+        this.tools = [];
+        /** Maximum conversation length before summarization */
+        this.maxConversationLength = 50;
+        /** Whether to force summarization */
+        this.forceSummarization = false;
+        this.codebolt = codebolt;
+    }
+    /**
+     * Adds the previous conversation to the builder.
+     *
+     * @param previousPrompt - The previous prompt object containing messages and tools
+     * @returns The FollowUpQuestionBuilder instance for chaining
+     */
+    addPreviousConversation(previousPrompt, llmResponse) {
+        this.previousConversation = [...previousPrompt.messages];
+        this.tools = [...previousPrompt.tools];
+        try {
+            // Resolve the response if it's a promise
+            const resolvedResponse = llmResponse;
+            if (!resolvedResponse || !resolvedResponse.completion) {
+                console.warn("Invalid LLM response provided");
+                return this;
+            }
+            const completion = resolvedResponse.completion;
+            let assistantMessage = null;
+            // Handle different response formats
+            if (completion.choices && completion.choices.length > 0) {
+                // OpenAI-style response with choices
+                const choice = completion.choices[0];
+                if (choice.message) {
+                    assistantMessage = {
+                        role: "assistant",
+                        content: choice.message.content || "",
+                        tool_calls: choice.message.tool_calls || undefined
+                    };
+                }
+            }
+            else if (completion.content) {
+                // Direct content response
+                assistantMessage = {
+                    role: "assistant",
+                    content: completion.content
+                };
+            }
+            else if (completion.message) {
+                // Message format response
+                assistantMessage = {
+                    role: "assistant",
+                    content: completion.message.content || "",
+                    tool_calls: completion.message.tool_calls || undefined
+                };
+            }
+            // Add the assistant message to conversation history
+            if (assistantMessage) {
+                this.previousConversation.push(assistantMessage);
+            }
+            else {
+                // Fallback for cases where no valid message is found
+                this.previousConversation.push({
+                    role: "assistant",
+                    content: "I apologize, but I was unable to provide a proper response."
+                });
+            }
+        }
+        catch (error) {
+            console.error("Error adding LLM response to conversation:", error);
+            // Add error message to conversation history
+            this.previousConversation.push({
+                role: "assistant",
+                content: "An error occurred while processing my response."
+            });
+        }
+        return this;
+    }
+    addLLMResponseToConverstaion(llmResponse) {
+        return this;
+    }
+    /**
+     * Adds tool execution results to the conversation.
+     *
+     * @param toolResults - Array of tool execution results
+     * @returns The FollowUpQuestionBuilder instance for chaining
+     */
+    addToolResult(toolResults) {
+        toolResults.forEach(toolResult => {
+            this.previousConversation.push(toolResult);
+        });
+        if (!toolResults.length) {
+            this.previousConversation.push({
+                role: "user",
+                content: [{
+                        type: "text",
+                        text: "If you have completed the user's task, use the attempt_completion tool. If you require additional information from the user, use the ask_followup_question tool. Otherwise, if you have not completed the task and do not need additional information, then proceed with the next step of the task. (This is an automated message, so do not respond to it conversationally.)"
+                    }]
+            });
+        }
+        return this;
+    }
+    /**
+     * Checks if the conversation is too long and sets up summarization with custom max length.
+     *
+     * @param maxLength - Maximum number of messages before summarization
+     * @returns The FollowUpQuestionBuilder instance for chaining
+     */
+    checkAndSummarizeConversationIfLong(maxLength) {
+        this.maxConversationLength = maxLength;
+        this.forceSummarization = this.previousConversation.length > maxLength;
+        return this;
+    }
+    /**
+     * Performs conversation summarization if needed.
+     *
+     * @returns Promise that resolves to the summarized messages
+     */
+    async performSummarization() {
+        const shouldSummarize = this.forceSummarization ||
+            this.previousConversation.length > this.maxConversationLength;
+        if (!shouldSummarize) {
+            return this.previousConversation;
+        }
+        try {
+            console.log("Summarizing conversation due to length:", this.previousConversation.length);
+            // Convert OpenAI messages to the format expected by chatSummary
+            const messagesToSummarize = this.previousConversation.map(msg => ({
+                role: msg.role,
+                content: typeof msg.content === 'string' ? msg.content :
+                    Array.isArray(msg.content) ? msg.content.map(c => c.text).join(' ') :
+                        String(msg.content)
+            }));
+            // Use the chat summary service to summarize the conversation
+            const summaryResponse = await chatSummary.summarize(messagesToSummarize, Math.floor(this.maxConversationLength / 2));
+            if (summaryResponse.payload || summaryResponse.summary) {
+                const summaryText = summaryResponse.payload || summaryResponse.summary || '';
+                // Keep the system message if it exists, and replace the rest with summary
+                const systemMessage = this.previousConversation.find(msg => msg.role === 'system');
+                const summarizedMessages = [];
+                if (systemMessage) {
+                    summarizedMessages.push(systemMessage);
+                }
+                // Add the summary as a system message
+                summarizedMessages.push({
+                    role: 'system',
+                    content: `Previous conversation summary: ${summaryText}`
+                });
+                // Keep the last few messages for context
+                const recentMessages = this.previousConversation.slice(-5);
+                summarizedMessages.push(...recentMessages);
+                return summarizedMessages;
+            }
+        }
+        catch (error) {
+            console.error("Error summarizing conversation:", error);
+        }
+        // If summarization fails, just return the original conversation
+        return this.previousConversation;
+    }
+    /**
+     * Builds the follow-up conversation prompt with tool results.
+     *
+     * @returns Promise that resolves to the conversation prompt object
+     */
+    async build() {
+        // Perform summarization if needed
+        let messages = this.previousConversation;
+        return {
+            messages,
+            tools: this.tools,
+            full: true,
+            tool_choice: "auto"
+        };
+    }
+}
+exports.FollowUpPromptBuilder = FollowUpPromptBuilder;

package/dist/builderpattern/index.d.ts

@@ -0,0 +1,14 @@
+/**
+ * @fileoverview Main entry point for CodeBolt Agent Utils
+ * @description Exports all agent utilities including Agent class, message builders, and prompt utilities
+ */
+export { Agent } from './agent';
+export { UserMessage } from './usermessage';
+export { SystemPrompt } from './systemprompt';
+export { TaskInstruction } from './taskInstruction';
+export { InitialPromptBuilder } from './promptbuilder';
+export { FollowUpPromptBuilder } from './followupquestionbuilder';
+export { LLMOutputHandler } from './llmoutputhandler';
+export type { Message, ToolResult, ToolDetails, OpenAIMessage, OpenAITool, ConversationEntry, UserMessageContent, CodeboltAPI, ToolCall } from '../types/libFunctionTypes';
+export type { MCPTool, Agent as AgentType, InitialUserMessage } from '../types/commonTypes';
+export type { UserMessage as CLIUserMessage } from '../types/socketMessageTypes';

package/dist/builderpattern/index.js

@@ -0,0 +1,23 @@
+"use strict";
+/**
+ * @fileoverview Main entry point for CodeBolt Agent Utils
+ * @description Exports all agent utilities including Agent class, message builders, and prompt utilities
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.LLMOutputHandler = exports.FollowUpPromptBuilder = exports.InitialPromptBuilder = exports.TaskInstruction = exports.SystemPrompt = exports.UserMessage = exports.Agent = void 0;
+// Core Agent functionality
+var agent_1 = require("./agent");
+Object.defineProperty(exports, "Agent", { enumerable: true, get: function () { return agent_1.Agent; } });
+// Message and prompt builders
+var usermessage_1 = require("./usermessage");
+Object.defineProperty(exports, "UserMessage", { enumerable: true, get: function () { return usermessage_1.UserMessage; } });
+var systemprompt_1 = require("./systemprompt");
+Object.defineProperty(exports, "SystemPrompt", { enumerable: true, get: function () { return systemprompt_1.SystemPrompt; } });
+var taskInstruction_1 = require("./taskInstruction");
+Object.defineProperty(exports, "TaskInstruction", { enumerable: true, get: function () { return taskInstruction_1.TaskInstruction; } });
+var promptbuilder_1 = require("./promptbuilder");
+Object.defineProperty(exports, "InitialPromptBuilder", { enumerable: true, get: function () { return promptbuilder_1.InitialPromptBuilder; } });
+var followupquestionbuilder_1 = require("./followupquestionbuilder");
+Object.defineProperty(exports, "FollowUpPromptBuilder", { enumerable: true, get: function () { return followupquestionbuilder_1.FollowUpPromptBuilder; } });
+var llmoutputhandler_1 = require("./llmoutputhandler");
+Object.defineProperty(exports, "LLMOutputHandler", { enumerable: true, get: function () { return llmoutputhandler_1.LLMOutputHandler; } });