@codebolt/agent 5.0.1 → 5.0.6

package/dist/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/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/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; } });

package/dist/llmoutputhandler.d.ts

@@ -0,0 +1,102 @@
+import type { OpenAIMessage, ToolResult, CodeboltAPI } from "./types/libFunctionTypes";
+/**
+ * Handles LLM output processing, tool execution, and completion detection.
+ * This class processes LLM responses and manages the conversation flow.
+ */
+declare class LLMOutputHandler {
+    /** The LLM response to process */
+    private llmResponse;
+    /** Whether the task has been completed */
+    private completed;
+    /** Tool results from execution */
+    private toolResults;
+    /** Next user message to be added to conversation */
+    private nextUserMessage;
+    /** Codebolt API instance */
+    private codebolt?;
+    /** Whether mcp have been executed */
+    private mcpExecuted;
+    /**
+     * Creates a new LLMOutputHandler instance.
+     *
+     * @param llmResponse - The LLM response object
+     * @param codebolt - Optional codebolt API instance
+     */
+    constructor(llmResponse: any, codebolt?: CodeboltAPI);
+    /**
+     * Checks if the task has been completed.
+     *
+     * @returns True if the task is completed, false otherwise
+     */
+    isCompleted(): boolean;
+    /**
+     * Gets the tool execution results.
+     *
+     * @returns Array of tool results
+     */
+    getToolResults(): ToolResult[];
+    /**
+     * Gets the next user message to be added to conversation.
+     *
+     * @returns The next user message or null
+     */
+    getNextUserMessage(): OpenAIMessage | null;
+    /**
+     * Sends the assistant's message to the user interface.
+     *
+     * @returns Promise that resolves when the message is sent
+     */
+    sendMessageToUser(): Promise<void>;
+    /**
+     * Executes all tool calls found in the LLM response.
+     *
+     * @returns Promise that resolves when all mcp are executed
+     */
+    runTools(): Promise<ToolResult[]>;
+    /**
+     * Extracts tool calls from the LLM response.
+     *
+     * @returns Array of tool calls or null if none found
+     */
+    private getToolCallsFromResponse;
+    /**
+     * 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
+     */
+    private getToolDetail;
+    /**
+     * 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]
+     */
+    private executeTool;
+    /**
+     * 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
+     */
+    private getToolResult;
+    /**
+     * Checks if the response has tool calls.
+     *
+     * @returns True if the response contains tool calls
+     */
+    hasToolCalls(): boolean;
+    /**
+     * Gets the assistant's message content from the response.
+     *
+     * @returns The assistant's message content or null
+     */
+    getAssistantMessage(): string | null;
+    /**
+     * Resets the handler state for reuse.
+     */
+    reset(): void;
+}
+export { LLMOutputHandler };