@codebolt/codeboltjs 2.2.1 → 2.2.3

package/dist/core/Codebolt.js

@@ -14,7 +14,6 @@ const crawler_1 = __importDefault(require("../modules/crawler"));
 const search_1 = __importDefault(require("../modules/search"));
 const knowledge_1 = __importDefault(require("../modules/knowledge"));
 const rag_1 = __importDefault(require("../modules/rag"));
-const codeparsers_1 = __importDefault(require("../modules/codeparsers"));
 const outputparsers_1 = __importDefault(require("../modules/outputparsers"));
 const project_1 = __importDefault(require("../modules/project"));
 const git_1 = __importDefault(require("../modules/git"));
@@ -29,6 +28,8 @@ const mcp_1 = __importDefault(require("../modules/mcp"));
 const agent_1 = __importDefault(require("../modules/agent"));
 const utils_1 = __importDefault(require("../modules/utils"));
 const notificationfunctions_1 = require("../notificationfunctions");
+const user_message_manager_1 = require("../modules/user-message-manager");
+const user_message_utilities_1 = require("../modules/user-message-utilities");
 /**
  * @class Codebolt
  * @description This class provides a unified interface to interact with various modules.
@@ -53,12 +54,11 @@ class Codebolt {
         this.search = search_1.default;
         this.knowledge = knowledge_1.default;
         this.rag = rag_1.default;
-        this.codeparsers = codeparsers_1.default;
         this.outputparsers = outputparsers_1.default;
         this.project = project_1.default;
         this.dbmemory = dbmemory_1.default;
         this.cbstate = state_1.default;
-        this.taskplaner = task_1.default;
+        this.task = task_1.default;
         this.vectordb = vectordb_1.default;
         this.debug = debug_1.default;
         this.tokenizer = tokenizer_1.default;
@@ -67,6 +67,10 @@ class Codebolt {
         this.agent = agent_1.default;
         this.utils = utils_1.default;
         this.notify = notificationfunctions_1.notificationFunctions;
+        /**
+         * User message utilities for accessing current user message and context
+         */
+        this.userMessage = user_message_utilities_1.userMessageUtilities;
         console.log("Codebolt Agent initialized");
         this.readyPromise = this.initializeConnection();
     }
@@ -148,12 +152,11 @@ class Codebolt {
         this.waitForReady().then(() => {
             const handleUserMessage = async (response) => {
                 var _a, _b;
-                console.log("Message received By Agent Library Starting Custom Agent Handler Logic");
+                console.log("Message received By Agent Library Starting Custom Agent Handler Logic", response);
                 if (response.type === "messageResponse") {
                     try {
                         // Extract user-facing message from internal socket message
                         const userMessage = {
-                            type: response.type,
                             userMessage: response.message.userMessage,
                             currentFile: response.message.currentFile,
                             mentionedFiles: response.message.mentionedFiles || [],
@@ -169,8 +172,12 @@ class Codebolt {
                             threadId: response.message.threadId,
                             selection: response.message.selection,
                             remixPrompt: response.message.remixPrompt,
-                            mentionedAgents: response.message.mentionedAgents
+                            mentionedAgents: response.message.mentionedAgents || [],
+                            activeFile: response.message.activeFile,
+                            openedFiles: response.message.activeFile
                         };
+                        // Automatically save the user message globally
+                        user_message_manager_1.userMessageManager.saveMessage(userMessage);
                         const result = await handler(userMessage);
                         // Send processStoped with optional message
                         const message = {
@@ -197,5 +204,259 @@ class Codebolt {
             console.error('Failed to set up message handler:', error);
         });
     }
+    onRawMessage(handler) {
+        this.waitForReady().then(() => {
+            const handleRawUserMessage = async (response) => {
+                if (response.type != "messageResponse" && response.type != "providerStart" && response.type != "providerAgentStart") {
+                    handler(response);
+                }
+            };
+            websocket_1.default.messageManager.on('message', handleRawUserMessage);
+        }).catch(error => {
+            console.error('Failed to set up message handler:', error);
+        });
+    }
+    /**
+     * Sets up a listener for provider start events.
+     * @param {Function} handler - The handler function to call when provider starts.
+     * @returns {void}
+     */
+    onProviderStart(handler) {
+        this.waitForReady().then(() => {
+            const handleProviderStart = async (response) => {
+                if (response.type === "providerStart") {
+                    try {
+                        const result = await handler(response || {});
+                        const message = {
+                            "type": "remoteProviderEvent",
+                            "action": "providerStartResponse"
+                        };
+                        if (result !== undefined && result !== null) {
+                            message.message = result;
+                        }
+                        websocket_1.default.messageManager.send(message);
+                    }
+                    catch (error) {
+                        console.error('Error in provider start handler:', error);
+                        websocket_1.default.messageManager.send({
+                            "type": "providerStartResponse",
+                            "error": error instanceof Error ? error.message : "Unknown error occurred"
+                        });
+                    }
+                }
+            };
+            websocket_1.default.messageManager.on('message', handleProviderStart);
+        }).catch(error => {
+            console.error('Failed to set up provider start handler:', error);
+        });
+    }
+    /**
+     * Sets up a listener for provider agent start events.
+     * @param {Function} handler - The handler function to call when provider agent starts.
+     * @returns {void}
+     */
+    onProviderAgentStart(handler) {
+        this.waitForReady().then(() => {
+            const handleProviderAgentStart = async (response) => {
+                console.log("Provider agent start event received");
+                if (response.type === "providerAgentStart") {
+                    try {
+                        const result = await handler(response.userMessage);
+                        const message = {
+                            "type": "remoteProviderEvent",
+                            "action": "providerAgentStartResponse"
+                        };
+                        if (result !== undefined && result !== null) {
+                            message.message = result;
+                        }
+                        websocket_1.default.messageManager.send(message);
+                    }
+                    catch (error) {
+                        console.error('Error in provider agent start handler:', error);
+                        websocket_1.default.messageManager.send({
+                            "type": "remoteProviderEvent",
+                            "action": "providerAgentStartResponse",
+                            "error": error instanceof Error ? error.message : "Unknown error occurred"
+                        });
+                    }
+                }
+            };
+            websocket_1.default.messageManager.on('message', handleProviderAgentStart);
+        }).catch(error => {
+            console.error('Failed to set up provider agent start handler:', error);
+        });
+    }
+    /**
+     * Sets up a listener for provider stop events.
+     * @param {Function} handler - The handler function to call when provider stops.
+     * @returns {void}
+     */
+    onProviderStop(handler) {
+        this.waitForReady().then(() => {
+            const handleProviderStop = async (response) => {
+                if (response.type === "providerStop") {
+                    try {
+                        const result = await handler(response || {});
+                        const message = {
+                            "type": "remoteProviderEvent",
+                            "action": "providerStopResponse",
+                        };
+                        if (result !== undefined && result !== null) {
+                            message.message = result;
+                        }
+                        websocket_1.default.messageManager.send(message);
+                    }
+                    catch (error) {
+                        console.error('Error in provider stop handler:', error);
+                        websocket_1.default.messageManager.send({
+                            "type": "remoteProviderEvent",
+                            "action": "providerStopResponse",
+                            "error": error instanceof Error ? error.message : "Unknown error occurred"
+                        });
+                    }
+                }
+            };
+            websocket_1.default.messageManager.on('message', handleProviderStop);
+        }).catch(error => {
+            console.error('Failed to set up provider stop handler:', error);
+        });
+    }
+    /**
+     * Sets up a listener for get diff files events.
+     * @param {Function} handler - The handler function to call when diff files are requested.
+     * @returns {void}
+     */
+    onGetDiffFiles(handler) {
+        this.waitForReady().then(() => {
+            const handleGetDiffFiles = async (response) => {
+                console.log("Get diff files event received");
+                if (response.type === "providerGetDiffFiles") {
+                    try {
+                        const result = await handler();
+                        const message = {
+                            "type": "remoteProviderEvent",
+                            "action": "providerDiffFilesResponse",
+                        };
+                        if (result !== undefined && result !== null) {
+                            message.message = result;
+                        }
+                        websocket_1.default.messageManager.send(message);
+                    }
+                    catch (error) {
+                        console.error('Error in get diff files handler:', error);
+                        websocket_1.default.messageManager.send({
+                            "type": "getDiffFiles",
+                            "error": error instanceof Error ? error.message : "Unknown error occurred"
+                        });
+                    }
+                }
+            };
+            websocket_1.default.messageManager.on('message', handleGetDiffFiles);
+        }).catch(error => {
+            console.error('Failed to set up get diff files handler:', error);
+        });
+    }
+    /**
+     * Sets up a listener for close signal events.
+     * @param {Function} handler - The handler function to call when close signal is received.
+     * @returns {void}
+     */
+    onCloseSignal(handler) {
+        this.waitForReady().then(() => {
+            const handleCloseSignal = async (response) => {
+                console.log("Close signal event received");
+                if (response.type === "closeSignal") {
+                    try {
+                        const result = await handler();
+                        const message = {
+                            "type": "closeSignalResponse"
+                        };
+                        if (result !== undefined && result !== null) {
+                            message.message = result;
+                        }
+                        websocket_1.default.messageManager.send(message);
+                    }
+                    catch (error) {
+                        console.error('Error in close signal handler:', error);
+                        websocket_1.default.messageManager.send({
+                            "type": "closeSignalResponse",
+                            "error": error instanceof Error ? error.message : "Unknown error occurred"
+                        });
+                    }
+                }
+            };
+            websocket_1.default.messageManager.on('message', handleCloseSignal);
+        }).catch(error => {
+            console.error('Failed to set up close signal handler:', error);
+        });
+    }
+    /**
+     * Sets up a listener for create patch request events.
+     * @param {Function} handler - The handler function to call when patch request is created.
+     * @returns {void}
+     */
+    onCreatePatchRequest(handler) {
+        this.waitForReady().then(() => {
+            const handleCreatePatchRequest = async (response) => {
+                console.log("Create patch request event received");
+                if (response.type === "createPatchRequest") {
+                    try {
+                        const result = await handler();
+                        const message = {
+                            "type": "createPatchRequestResponse"
+                        };
+                        if (result !== undefined && result !== null) {
+                            message.message = result;
+                        }
+                        websocket_1.default.messageManager.send(message);
+                    }
+                    catch (error) {
+                        console.error('Error in create patch request handler:', error);
+                        websocket_1.default.messageManager.send({
+                            "type": "createPatchRequestResponse",
+                            "error": error instanceof Error ? error.message : "Unknown error occurred"
+                        });
+                    }
+                }
+            };
+            websocket_1.default.messageManager.on('message', handleCreatePatchRequest);
+        }).catch(error => {
+            console.error('Failed to set up create patch request handler:', error);
+        });
+    }
+    /**
+     * Sets up a listener for create pull request events.
+     * @param {Function} handler - The handler function to call when pull request is created.
+     * @returns {void}
+     */
+    onCreatePullRequestRequest(handler) {
+        this.waitForReady().then(() => {
+            const handleCreatePullRequestRequest = async (response) => {
+                console.log("Create pull request event received");
+                if (response.type === "createPullRequestRequest") {
+                    try {
+                        const result = await handler();
+                        const message = {
+                            "type": "createPullRequestRequestResponse"
+                        };
+                        if (result !== undefined && result !== null) {
+                            message.message = result;
+                        }
+                        websocket_1.default.messageManager.send(message);
+                    }
+                    catch (error) {
+                        console.error('Error in create pull request handler:', error);
+                        websocket_1.default.messageManager.send({
+                            "type": "createPullRequestRequestResponse",
+                            "error": error instanceof Error ? error.message : "Unknown error occurred"
+                        });
+                    }
+                }
+            };
+            websocket_1.default.messageManager.on('message', handleCreatePullRequestRequest);
+        }).catch(error => {
+            console.error('Failed to set up create pull request handler:', error);
+        });
+    }
 }
 exports.default = Codebolt;

package/dist/core/websocket.js

@@ -25,16 +25,14 @@ class cbws {
     }
     getUniqueConnectionId() {
         try {
-            console.log('[WebSocket] Reading unique connection ID from codeboltagent.yaml');
-            let fileContents = fs_1.default.readFileSync('./codeboltagent.yaml', 'utf8');
-            let data = js_yaml_1.default.load(fileContents);
-            const connectionId = data.unique_connectionid;
-            console.log('[WebSocket] Successfully retrieved connection ID:', connectionId);
-            return connectionId;
+            // Generate a unique ID using timestamp and random string
+            const timestamp = Date.now().toString(36);
+            const randomPart = Math.random().toString(36).substring(2, 15);
+            return `ws_${timestamp}_${randomPart}`;
         }
         catch (e) {
-            console.error('[WebSocket] Unable to locate codeboltagent.yaml file:', e);
-            return '';
+            console.error('[WebSocket] Error generating unique connection ID:', e);
+            return `ws_fallback_${Date.now()}`;
         }
     }
     getInitialMessage() {
@@ -59,7 +57,7 @@ class cbws {
     async initializeWebSocket() {
         console.log('[WebSocket] Starting WebSocket initialization');
         const uniqueConnectionId = this.getUniqueConnectionId();
-        const initialMessage = this.getInitialMessage();
+        // const initialMessage = "Hello" //this.getInitialMessage();
         const agentIdParam = process.env.agentId ? `&agentId=${process.env.agentId}` : '';
         const parentIdParam = process.env.parentId ? `&parentId=${process.env.parentId}` : '';
         const parentAgentInstanceIdParam = process.env.parentAgentInstanceId ? `&parentAgentInstanceId=${process.env.parentAgentInstanceId}` : '';
@@ -69,7 +67,7 @@ class cbws {
         const threadToken = process.env.threadToken || null;
         console.log('[WebSocket] Logging all relevant variables:');
         console.log('uniqueConnectionId:', uniqueConnectionId);
-        console.log('initialMessage:', initialMessage);
+        // console.log('initialMessage:', initialMessage);
         console.log('agentIdParam:', agentIdParam);
         console.log('parentIdParam:', parentIdParam);
         console.log('parentAgentInstanceIdParam:', parentAgentInstanceIdParam);
@@ -127,7 +125,7 @@ class cbws {
                 this.initialized = false;
             });
             this.websocket.on('message', (data) => {
-                console.log('[WebSocket] Message received:', data.toString().substring(0, 100) + (data.toString().length > 100 ? '...' : ''));
+                console.log('[WebSocket] Message received:', data.toString());
             });
             this.websocket.on('ping', () => {
                 console.log('[WebSocket] Ping received');

package/dist/index.d.ts

@@ -1,5 +1,5 @@
 import Codebolt from './core/Codebolt';
-export type { Message, ToolCall, Tool, UserMessage, LLMInferenceParams, APIResponse, CodeboltConfig, ProgressCallback, ErrorCallback, SuccessCallback, CompletionCallback, OpenAIMessage, OpenAITool, ConversationEntry, ToolResult, ToolDetails, UserMessageContent, CodeboltAPI, ReadFileOptions, WriteFileOptions, ListFilesOptions, SearchFilesOptions, GrepSearchOptions, BrowserNavigationOptions, BrowserScreenshotOptions, BrowserElementSelector, TerminalExecuteOptions, GitCommitOptions, GitLogOptions, LLMChatOptions, VectorAddOptions, VectorQueryOptions, AgentMessageHandler, AgentConfiguration, MemorySetOptions, MemoryGetOptions, TaskCreateOptions, TaskUpdateOptions, CodeAnalysisOptions, CodeParseOptions, DebugLogOptions, ProjectInfo, CrawlerOptions, MCPExecuteOptions, MCPConfigureOptions, StateUpdateOptions, ChatSendOptions, ChatHistoryOptions, NotificationOptions, PaginationOptions, FilterOptions, AsyncOperationOptions, APIEventMap } from './types/libFunctionTypes';
+export type { Message, ToolCall, Tool, UserMessage, LLMInferenceParams, APIResponse, CodeboltConfig, ProgressCallback, ErrorCallback, SuccessCallback, CompletionCallback, OpenAIMessage, OpenAITool, ConversationEntry, ToolResult, ToolDetails, UserMessageContent, CodeboltAPI, ReadFileOptions, WriteFileOptions, ListFilesOptions, SearchFilesOptions, GrepSearchOptions, BrowserNavigationOptions, BrowserScreenshotOptions, BrowserElementSelector, TerminalExecuteOptions, LLMChatOptions, VectorAddOptions, VectorQueryOptions, AgentMessageHandler, AgentConfiguration, MemorySetOptions, MemoryGetOptions, TaskCreateOptions, TaskUpdateOptions, CodeAnalysisOptions, CodeParseOptions, DebugLogOptions, ProjectInfo, CrawlerOptions, MCPExecuteOptions, MCPConfigureOptions, StateUpdateOptions, ChatSendOptions, ChatHistoryOptions, NotificationOptions, PaginationOptions, FilterOptions, AsyncOperationOptions, APIEventMap } from './types/libFunctionTypes';
 export type { GitFileStatus, StatusResult, CommitSummary, DiffResult, AgentFunction, AgentDetail, Agent, Task, VectorItem, VectorQueryResult, FileEntry, SearchMatch, SearchResult, BrowserElement, CodeIssue, CodeAnalysis, CodeMatcher, MCPTool, MCPServer, ASTNode, Notification, DeepPartial, DeepRequired, Optional, Required } from './types/commonTypes';
 declare const codebolt: Codebolt;
 export default codebolt;

package/dist/modules/agent.d.ts

@@ -1,19 +1,5 @@
-import { FindAgentByTaskResponse, ListAgentsResponse, AgentsDetailResponse, TaskCompletionResponse } from '../types/socketMessageTypes';
-export declare enum AgentLocation {
-    ALL = "all",
-    LOCAL_ONLY = "local_only",
-    REMOTE_ONLY = "remote_only"
-}
-export declare enum Agents {
-    LOCAL = "local",
-    ALL = "all",
-    DOWNLOADED = "downloaded"
-}
-export declare enum FilterUsing {
-    USE_AI = "use_ai",
-    USE_VECTOR_DB = "use_vector_db",
-    USE_BOTH = "use_both"
-}
+import { FindAgentByTaskResponse, ListAgentsResponse, AgentsDetailResponse, TaskCompletionResponse } from '@codebolt/types/sdk';
+import { AgentLocation, Agents, FilterUsing } from '@codebolt/types/enum';
 declare const codeboltAgent: {
     /**
      * Retrieves an agent based on the specified task.

package/dist/modules/agent.js

@@ -3,42 +3,24 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.FilterUsing = exports.Agents = exports.AgentLocation = void 0;
 const websocket_1 = __importDefault(require("../core/websocket"));
-var AgentLocation;
-(function (AgentLocation) {
-    AgentLocation["ALL"] = "all";
-    AgentLocation["LOCAL_ONLY"] = "local_only";
-    AgentLocation["REMOTE_ONLY"] = "remote_only";
-})(AgentLocation || (exports.AgentLocation = AgentLocation = {}));
-var Agents;
-(function (Agents) {
-    Agents["LOCAL"] = "local";
-    Agents["ALL"] = "all";
-    Agents["DOWNLOADED"] = "downloaded";
-})(Agents || (exports.Agents = Agents = {}));
-var FilterUsing;
-(function (FilterUsing) {
-    FilterUsing["USE_AI"] = "use_ai";
-    FilterUsing["USE_VECTOR_DB"] = "use_vector_db";
-    FilterUsing["USE_BOTH"] = "use_both";
-})(FilterUsing || (exports.FilterUsing = FilterUsing = {}));
+const enum_1 = require("@codebolt/types/enum");
 const codeboltAgent = {
     /**
      * Retrieves an agent based on the specified task.
      * @param {string} task - The task for which an agent is needed.
      * @returns {Promise<FindAgentByTaskResponse>} A promise that resolves with the agent details.
      */
-    findAgent: (task, maxResult = 1, agents = [], agentLocaltion = AgentLocation.ALL, getFrom) => {
+    findAgent: (task, maxResult = 1, agents = [], agentLocaltion = enum_1.AgentLocation.ALL, getFrom) => {
         return websocket_1.default.messageManager.sendAndWaitForResponse({
-            "type": "agentEvent",
-            "action": "findAgent",
+            "type": enum_1.EventType.AGENT_EVENT,
+            "action": enum_1.AgentAction.FIND_AGENT,
             "task": task,
             "agents": agents, // for filter in vector db
             "maxResult": maxResult,
             "location": agentLocaltion,
             "getFrom": getFrom
-        }, "findAgentByTaskResponse");
+        }, enum_1.AgentResponseType.FIND_AGENT_BY_TASK_RESPONSE);
     },
     /**
      * Starts an agent for the specified task.
@@ -47,22 +29,22 @@ const codeboltAgent = {
      */
     startAgent: (agentId, task) => {
         return websocket_1.default.messageManager.sendAndWaitForResponse({
-            "type": "agentEvent",
-            "action": "startAgent",
+            "type": enum_1.EventType.AGENT_EVENT,
+            "action": enum_1.AgentAction.START_AGENT,
             "agentId": agentId,
             "task": task
-        }, "taskCompletionResponse");
+        }, enum_1.AgentResponseType.TASK_COMPLETION_RESPONSE);
     },
     /**
      * Lists all available agents.
      * @returns {Promise<ListAgentsResponse>} A promise that resolves with the list of agents.
      */
-    getAgentsList: (type = Agents.DOWNLOADED) => {
+    getAgentsList: (type = enum_1.Agents.DOWNLOADED) => {
         return websocket_1.default.messageManager.sendAndWaitForResponse({
-            "type": "agentEvent",
-            "action": "listAgents",
+            "type": enum_1.EventType.AGENT_EVENT,
+            "action": enum_1.AgentAction.LIST_AGENTS,
             "agentType": type,
-        }, "listAgentsResponse");
+        }, enum_1.AgentResponseType.LIST_AGENTS_RESPONSE);
     },
     /**
      * Lists all available agents.
@@ -70,10 +52,10 @@ const codeboltAgent = {
      */
     getAgentsDetail: (agentList = []) => {
         return websocket_1.default.messageManager.sendAndWaitForResponse({
-            "type": "agentEvent",
-            "action": "agentsDetail",
+            "type": enum_1.EventType.AGENT_EVENT,
+            "action": enum_1.AgentAction.AGENTS_DETAIL,
             "agentList": agentList
-        }, "agentsDetailResponse");
+        }, enum_1.AgentResponseType.AGENTS_DETAIL_RESPONSE);
     }
 };
 exports.default = codeboltAgent;

package/dist/modules/browser.d.ts

@@ -1,4 +1,4 @@
-import { GoToPageResponse, UrlResponse, GetMarkdownResponse, HtmlReceived, ExtractTextResponse, GetContentResponse, BrowserActionResponseData, BrowserScreenshotResponse, BrowserInfoResponse, BrowserSnapshotResponse } from '../types/socketMessageTypes';
+import { GoToPageResponse, UrlResponse, GetMarkdownResponse, HtmlReceived, ExtractTextResponse, GetContentResponse, BrowserActionResponseData, BrowserScreenshotResponse, BrowserInfoResponse, BrowserSnapshotResponse } from '@codebolt/types/sdk';
 /**
  * A module for interacting with a browser through WebSockets.
  */