@codebolt/codeboltjs 2.0.12 → 2.0.15

package/dist/index.d.ts

@@ -1,302 +1,6 @@
-/// <reference types="node" />
-import WebSocket from 'ws';
-import { EventEmitter } from 'events';
-import type { ChatMessageFromUser, LLMResponse, UserMessage } from './types/socketMessageTypes';
-export type { Message, ToolCall, Tool, LLMInferenceParams, APIResponse, CodeboltConfig, ProgressCallback, ErrorCallback, SuccessCallback, CompletionCallback } from './types/libFunctionTypes';
-/**
- * @class Codebolt
- * @description This class provides a unified interface to interact with various modules.
- */
-declare class Codebolt {
-    websocket: WebSocket | null;
-    private isReady;
-    private readyPromise;
-    /**
-     * @constructor
-     * @description Initializes the websocket connection.
-     */
-    constructor();
-    /**
-     * @method initializeConnection
-     * @description Initializes the WebSocket connection asynchronously.
-     * @private
-     */
-    private initializeConnection;
-    /**
-     * @method waitForReady
-     * @description Waits for the Codebolt instance to be fully initialized.
-     * @returns {Promise<void>} A promise that resolves when the instance is ready.
-     */
-    waitForReady(): Promise<void>;
-    /**
-     * @method isReady
-     * @description Checks if the Codebolt instance is ready for use.
-     * @returns {boolean} True if the instance is ready, false otherwise.
-     */
-    get ready(): boolean;
-    fs: {
-        createFile: (fileName: string, source: string, filePath: string) => Promise<import("./types/socketMessageTypes").CreateFileResponse>;
-        createFolder: (folderName: string, folderPath: string) => Promise<import("./types/socketMessageTypes").CreateFolderResponse>;
-        readFile: (filePath: string) => Promise<import("./types/socketMessageTypes").ReadFileResponse>;
-        updateFile: (filename: string, filePath: string, newContent: string) => Promise<import("./types/socketMessageTypes").UpdateFileResponse>;
-        deleteFile: (filename: string, filePath: string) => Promise<import("./types/socketMessageTypes").DeleteFileResponse>;
-        deleteFolder: (foldername: string, folderpath: string) => Promise<import("./types/socketMessageTypes").DeleteFolderResponse>;
-        listFile: (folderPath: string, isRecursive?: boolean) => Promise<any>;
-        listCodeDefinitionNames: (path: string) => Promise<{
-            success: boolean;
-            result: any;
-        }>;
-        searchFiles: (path: string, regex: string, filePattern: string) => Promise<{
-            success: boolean;
-            result: any;
-        }>;
-        writeToFile: (relPath: string, newContent: string) => Promise<any>;
-        grepSearch: (path: string, query: string, includePattern?: string | undefined, excludePattern?: string | undefined, caseSensitive?: boolean) => Promise<{
-            success: boolean;
-            result: any;
-        }>;
-        fileSearch: (query: string) => Promise<{
-            success: boolean;
-            result: any;
-        }>;
-        editFileWithDiff: (targetFile: string, codeEdit: string, diffIdentifier: string, prompt: string, applyModel?: string | undefined) => Promise<{
-            success: boolean;
-            result: any;
-        }>;
-    };
-    git: {
-        init: (path: string) => Promise<import("./types/socketMessageTypes").GitInitResponse>;
-        pull: () => Promise<import("./types/socketMessageTypes").GitPullResponse>;
-        push: () => Promise<import("./types/socketMessageTypes").GitPushResponse>;
-        status: () => Promise<import("./types/socketMessageTypes").GitStatusResponse>;
-        addAll: () => Promise<import("./types/socketMessageTypes").AddResponse>;
-        commit: (message: string) => Promise<import("./types/socketMessageTypes").GitCommitResponse>;
-        checkout: (branch: string) => Promise<import("./types/socketMessageTypes").GitCheckoutResponse>;
-        branch: (branch: string) => Promise<import("./types/socketMessageTypes").GitBranchResponse>;
-        logs: (path: string) => Promise<import("./types/socketMessageTypes").GitLogsResponse>;
-        diff: (commitHash: string) => Promise<import("./types/socketMessageTypes").GitDiffResponse>;
-    };
-    llm: {
-        inference: (message: string, llmrole: string) => Promise<LLMResponse>;
-    };
-    browser: {
-        newPage: () => Promise<import("./types/socketMessageTypes").BrowserActionResponseData>;
-        getUrl: () => Promise<import("./types/socketMessageTypes").UrlResponse>;
-        goToPage: (url: string) => Promise<import("./types/socketMessageTypes").GoToPageResponse>;
-        screenshot: () => Promise<import("./types/socketMessageTypes").BrowserScreenshotResponse>;
-        getHTML: () => Promise<import("./types/socketMessageTypes").HtmlReceived>;
-        getMarkdown: () => Promise<import("./types/socketMessageTypes").GetMarkdownResponse>;
-        getPDF: () => void;
-        pdfToText: () => void;
-        getContent: () => Promise<import("./types/socketMessageTypes").GetContentResponse>;
-        getSnapShot: () => Promise<import("./types/socketMessageTypes").BrowserSnapshotResponse>;
-        getBrowserInfo: () => Promise<import("./types/socketMessageTypes").BrowserInfoResponse>;
-        extractText: () => Promise<import("./types/socketMessageTypes").ExtractTextResponse>;
-        close: () => void;
-        scroll: (direction: string, pixels: string) => Promise<import("./types/socketMessageTypes").BrowserActionResponseData>;
-        type: (elementid: string, text: string) => Promise<import("./types/socketMessageTypes").BrowserActionResponseData>;
-        click: (elementid: string) => Promise<import("./types/socketMessageTypes").BrowserActionResponseData>;
-        enter: () => Promise<import("./types/socketMessageTypes").BrowserActionResponseData>;
-        search: (elementid: string, query: string) => Promise<import("./types/socketMessageTypes").BrowserActionResponseData>;
-    };
-    chat: {
-        getChatHistory: () => Promise<import("./types/socketMessageTypes").ChatMessage[]>;
-        setRequestHandler: (handler: (request: any, response: (data: any) => void) => void | Promise<void>) => void;
-        sendMessage: (message: string, payload: any) => void;
-        waitforReply: (message: string) => Promise<UserMessage>;
-        processStarted: (onStopClicked?: ((message: any) => void) | undefined) => {
-            stopProcess: () => void;
-            cleanup: () => void;
-        } | {
-            stopProcess: () => void;
-            cleanup?: undefined;
-        };
-        stopProcess: () => void;
-        processFinished: () => void;
-        sendConfirmationRequest: (confirmationMessage: string, buttons?: string[], withFeedback?: boolean) => Promise<string>;
-        askQuestion: (question: string, buttons?: string[], withFeedback?: boolean) => Promise<string>;
-        sendNotificationEvent: (notificationMessage: string, type: "browser" | "terminal" | "git" | "debug" | "planner" | "editor" | "preview") => void;
-    };
-    terminal: {
-        eventEmitter: {
-            cleanup?: (() => void) | undefined;
-            [EventEmitter.captureRejectionSymbol]?<K>(error: Error, event: string | symbol, ...args: any[]): void;
-            addListener<K_1>(eventName: string | symbol, listener: (...args: any[]) => void): any;
-            on<K_2>(eventName: string | symbol, listener: (...args: any[]) => void): any;
-            once<K_3>(eventName: string | symbol, listener: (...args: any[]) => void): any;
-            removeListener<K_4>(eventName: string | symbol, listener: (...args: any[]) => void): any;
-            off<K_5>(eventName: string | symbol, listener: (...args: any[]) => void): any;
-            removeAllListeners(eventName?: string | symbol | undefined): any;
-            setMaxListeners(n: number): any;
-            getMaxListeners(): number;
-            listeners<K_6>(eventName: string | symbol): Function[];
-            rawListeners<K_7>(eventName: string | symbol): Function[];
-            emit<K_8>(eventName: string | symbol, ...args: any[]): boolean;
-            listenerCount<K_9>(eventName: string | symbol, listener?: Function | undefined): number;
-            prependListener<K_10>(eventName: string | symbol, listener: (...args: any[]) => void): any;
-            prependOnceListener<K_11>(eventName: string | symbol, listener: (...args: any[]) => void): any;
-            eventNames(): (string | symbol)[];
-        };
-        executeCommand: (command: string, returnEmptyStringOnSuccess?: boolean) => Promise<any>;
-        executeCommandRunUntilError: (command: string, executeInMain?: boolean) => Promise<import("./types/socketMessageTypes").CommandError>;
-        sendManualInterrupt(): Promise<import("./types/socketMessageTypes").TerminalInterruptResponse>;
-        executeCommandWithStream(command: string, executeInMain?: boolean): {
-            cleanup?: (() => void) | undefined;
-            [EventEmitter.captureRejectionSymbol]?<K>(error: Error, event: string | symbol, ...args: any[]): void;
-            addListener<K_1>(eventName: string | symbol, listener: (...args: any[]) => void): any;
-            on<K_2>(eventName: string | symbol, listener: (...args: any[]) => void): any;
-            once<K_3>(eventName: string | symbol, listener: (...args: any[]) => void): any;
-            removeListener<K_4>(eventName: string | symbol, listener: (...args: any[]) => void): any;
-            off<K_5>(eventName: string | symbol, listener: (...args: any[]) => void): any;
-            removeAllListeners(eventName?: string | symbol | undefined): any;
-            setMaxListeners(n: number): any;
-            getMaxListeners(): number;
-            listeners<K_6>(eventName: string | symbol): Function[];
-            rawListeners<K_7>(eventName: string | symbol): Function[];
-            emit<K_8>(eventName: string | symbol, ...args: any[]): boolean;
-            listenerCount<K_9>(eventName: string | symbol, listener?: Function | undefined): number;
-            prependListener<K_10>(eventName: string | symbol, listener: (...args: any[]) => void): any;
-            prependOnceListener<K_11>(eventName: string | symbol, listener: (...args: any[]) => void): any;
-            eventNames(): (string | symbol)[];
-        };
-    };
-    codeutils: {
-        getJsTree: (filePath?: string | undefined) => Promise<import("./types/InternalTypes").JSTreeResponse>;
-        getAllFilesAsMarkDown: () => Promise<string>;
-        performMatch: (matcherDefinition: object, problemPatterns: any[], problems?: any[]) => Promise<import("./types/socketMessageTypes").MatchProblemResponse>;
-        getMatcherList: () => Promise<import("./types/socketMessageTypes").GetMatcherListTreeResponse>;
-        matchDetail: (matcher: string) => Promise<import("./types/socketMessageTypes").getMatchDetail>;
-    };
-    crawler: {
-        start: () => void;
-        screenshot: () => void;
-        goToPage: (url: string) => void;
-        scroll: (direction: string) => void;
-        click: (id: string) => Promise<any>;
-    };
-    search: {
-        init: (engine?: string) => void;
-        search: (query: string) => Promise<string>;
-        get_first_link: (query: string) => Promise<string>;
-    };
-    knowledge: {};
-    rag: {
-        init: () => void;
-        add_file: (filename: string, file_path: string) => void;
-        retrieve_related_knowledge: (query: string, filename: string) => void;
-    };
-    codeparsers: {
-        getClassesInFile: (file: string) => Promise<{
-            error: string;
-        } | {
-            name: any;
-            location: string;
-        }[]>;
-        getFunctionsinClass: (file: string, className: string) => Promise<{
-            error: string;
-        } | {
-            name: string;
-            class: string;
-            location: string;
-        }[]>;
-        getAstTreeInFile: (file: string, className?: string | undefined) => Promise<import("./types/commonTypes").ASTNode | {
-            error: string;
-        }>;
-    };
-    outputparsers: {
-        parseJSON: (jsonString: string) => {
-            success: boolean;
-            parsed?: any;
-            error?: Error | undefined;
-        };
-        parseXML: (xmlString: string) => {
-            success: boolean;
-            parsed?: any;
-        };
-        parseCSV: (csvString: string) => {
-            success: boolean;
-            parsed?: any[] | undefined;
-            error?: Error | undefined;
-        };
-        parseText: (text: string) => {
-            success: boolean;
-            parsed: string[];
-        };
-        parseErrors: (output: any) => string[];
-        parseWarnings: (output: any) => string[];
-    };
-    project: {
-        getProjectSettings: () => Promise<any>;
-        getProjectPath: () => Promise<import("./types/socketMessageTypes").GetProjectPathResponse>;
-        getRepoMap: (message: any) => Promise<import("./types/socketMessageTypes").GetProjectPathResponse>;
-        runProject: () => void;
-        getEditorFileStatus: () => Promise<any>;
-    };
-    dbmemory: {
-        addKnowledge: (key: string, value: any) => Promise<import("./types/socketMessageTypes").MemorySetResponse>;
-        getKnowledge: (key: string) => Promise<import("./types/socketMessageTypes").MemoryGetResponse>;
-    };
-    cbstate: {
-        getApplicationState: () => Promise<import("./types/commonTypes").ApplicationState>;
-        addToAgentState: (key: string, value: string) => Promise<import("./types/socketMessageTypes").AddToAgentStateResponse>;
-        getAgentState: () => Promise<import("./types/socketMessageTypes").GetAgentStateResponse>;
-        getProjectState: () => Promise<any>;
-        updateProjectState: (key: string, value: any) => Promise<any>;
-    };
-    taskplaner: {
-        addTask: (task: string) => Promise<any>;
-        getTasks: () => Promise<any>;
-        updateTask: (task: string) => Promise<any>;
-    };
-    vectordb: {
-        getVector: (key: string) => Promise<import("./types/socketMessageTypes").GetVectorResponse>;
-        addVectorItem: (item: any) => Promise<import("./types/socketMessageTypes").AddVectorItemResponse>;
-        queryVectorItem: (key: string) => Promise<import("./types/socketMessageTypes").QueryVectorItemResponse>;
-        queryVectorItems: (items: [], dbPath: string) => Promise<import("./types/socketMessageTypes").QueryVectorItemResponse>;
-    };
-    debug: {
-        debug: (log: string, type: import("./modules/debug").logType) => Promise<import("./types/socketMessageTypes").DebugAddLogResponse>;
-        openDebugBrowser: (url: string, port: number) => Promise<import("./types/socketMessageTypes").OpenDebugBrowserResponse>;
-    };
-    tokenizer: {
-        addToken: (key: string) => Promise<import("./types/socketMessageTypes").AddTokenResponse>;
-        getToken: (key: string) => Promise<import("./types/socketMessageTypes").GetTokenResponse>;
-    };
-    chatSummary: {
-        summarizeAll: () => Promise<import("./types/socketMessageTypes").GetSummarizeAllResponse>;
-        summarize: (messages: {
-            role: string;
-            content: string;
-        }[], depth: number) => Promise<import("./types/socketMessageTypes").GetSummarizeResponse>;
-    };
-    mcp: {
-        getEnabledMCPServers: () => Promise<import("./types/socketMessageTypes").GetEnabledToolBoxesResponse>;
-        getLocalMCPServers: () => Promise<import("./types/socketMessageTypes").GetLocalToolBoxesResponse>;
-        getMentionedMCPServers: (userMessage: import("./utils").UserMessage) => Promise<import("./types/socketMessageTypes").GetAvailableToolBoxesResponse>;
-        searchAvailableMCPServers: (query: string) => Promise<import("./types/socketMessageTypes").SearchAvailableToolBoxesResponse>;
-        listMcpFromServers: (toolBoxes: string[]) => Promise<import("./types/socketMessageTypes").ListToolsFromToolBoxesResponse>;
-        configureMCPServer: (name: string, config: any) => Promise<import("./types/socketMessageTypes").ConfigureToolBoxResponse>;
-        getTools: (tools: {
-            toolbox: string;
-            toolName: string;
-        }[]) => Promise<import("./types/socketMessageTypes").GetToolsResponse>;
-        executeTool: (toolbox: string, toolName: string, params: any) => Promise<import("./types/socketMessageTypes").ExecuteToolResponse>;
-    };
-    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<import("./types/socketMessageTypes").FindAgentByTaskResponse>;
-        startAgent: (agentId: string, task: string) => Promise<import("./types/socketMessageTypes").TaskCompletionResponse>;
-        getAgentsList: (type?: import("./modules/agent").Agents) => Promise<import("./types/socketMessageTypes").ListAgentsResponse>;
-        getAgentsDetail: (agentList?: never[]) => Promise<import("./types/socketMessageTypes").AgentsDetailResponse>;
-    };
-    utils: {
-        editFileAndApplyDiff: (filePath: string, diff: string, diffIdentifier: string, prompt: string, applyModel?: string | undefined) => Promise<import("./types/socketMessageTypes").FsEditFileAndApplyDiffResponse>;
-    };
-    /**
-     * Sets up a listener for incoming messages with a direct handler function.
-     * @param {Function} handler - The handler function to call when a message is received.
-     * @returns {void}
-     */
-    onMessage(handler: (userMessage: ChatMessageFromUser) => void | Promise<void> | any | Promise<any>): void;
-}
+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 { 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;
+export { Codebolt };

package/dist/index.js

@@ -3,145 +3,19 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-const websocket_1 = __importDefault(require("./core/websocket"));
-const fs_1 = __importDefault(require("./modules/fs"));
-const llm_1 = __importDefault(require("./modules/llm"));
-const terminal_1 = __importDefault(require("./modules/terminal"));
-const browser_1 = __importDefault(require("./modules/browser"));
-const chat_1 = __importDefault(require("./modules/chat"));
-const codeutils_1 = __importDefault(require("./modules/codeutils"));
-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"));
-const dbmemory_1 = __importDefault(require("./modules/dbmemory"));
-const state_1 = __importDefault(require("./modules/state"));
-const task_1 = __importDefault(require("./modules/task"));
-const vectordb_1 = __importDefault(require("./modules/vectordb"));
-const debug_1 = __importDefault(require("./modules/debug"));
-const tokenizer_1 = __importDefault(require("./modules/tokenizer"));
-const history_1 = require("./modules/history");
-const mcp_1 = __importDefault(require("./modules/mcp"));
-const agent_1 = __importDefault(require("./modules/agent"));
-const utils_1 = __importDefault(require("./modules/utils"));
-/**
- * @class Codebolt
- * @description This class provides a unified interface to interact with various modules.
- */
-class Codebolt {
-    /**
-     * @constructor
-     * @description Initializes the websocket connection.
-     */
-    constructor() {
-        this.websocket = null;
-        this.isReady = false;
-        this.fs = fs_1.default;
-        this.git = git_1.default;
-        this.llm = llm_1.default;
-        this.browser = browser_1.default;
-        this.chat = chat_1.default;
-        this.terminal = terminal_1.default;
-        this.codeutils = codeutils_1.default;
-        this.crawler = crawler_1.default;
-        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.vectordb = vectordb_1.default;
-        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.utils = utils_1.default;
-        console.log("Codebolt Agent initialized");
-        this.readyPromise = this.initializeConnection();
-    }
-    /**
-     * @method initializeConnection
-     * @description Initializes the WebSocket connection asynchronously.
-     * @private
-     */
-    async initializeConnection() {
-        try {
-            await websocket_1.default.initializeWebSocket();
-            this.websocket = websocket_1.default.getWebsocket;
-            this.isReady = true;
-            console.log("Codebolt WebSocket connection established");
-        }
-        catch (error) {
-            console.error('Failed to initialize WebSocket connection:', error);
-            throw error;
-        }
-    }
-    /**
-     * @method waitForReady
-     * @description Waits for the Codebolt instance to be fully initialized.
-     * @returns {Promise<void>} A promise that resolves when the instance is ready.
-     */
-    async waitForReady() {
-        return this.readyPromise;
-    }
-    /**
-     * @method isReady
-     * @description Checks if the Codebolt instance is ready for use.
-     * @returns {boolean} True if the instance is ready, false otherwise.
-     */
-    get ready() {
-        return this.isReady;
-    }
-    /**
-     * Sets up a listener for incoming messages with a direct handler function.
-     * @param {Function} handler - The handler function to call when a message is received.
-     * @returns {void}
-     */
-    onMessage(handler) {
-        // Wait for the WebSocket to be ready before setting up the handler
-        this.waitForReady().then(() => {
-            const handleUserMessage = async (response) => {
-                console.log("Message received By Agent Library Starting Custom Agent Handler Logic");
-                if (response.type === "messageResponse") {
-                    try {
-                        const result = await handler(response.message);
-                        // Send processStoped with optional message
-                        const message = {
-                            "type": "processStoped"
-                        };
-                        // If handler returned data, include it as message
-                        if (result !== undefined && result !== null) {
-                            message.message = result;
-                        }
-                        websocket_1.default.messageManager.send(message);
-                    }
-                    catch (error) {
-                        console.error('Error in user message handler:', error);
-                        // Send processStoped even if there's an error
-                        websocket_1.default.messageManager.send({
-                            "type": "processStoped",
-                            "error": error instanceof Error ? error.message : "Unknown error occurred"
-                        });
-                    }
-                }
-            };
-            websocket_1.default.messageManager.on('message', handleUserMessage);
-        }).catch(error => {
-            console.error('Failed to set up message handler:', error);
-        });
-    }
-}
-const codebolt = new Codebolt();
+exports.Codebolt = void 0;
+const Codebolt_1 = __importDefault(require("./core/Codebolt"));
+exports.Codebolt = Codebolt_1.default;
+// ================================
+// Main Library Instance
+// ================================
+const codebolt = new Codebolt_1.default();
+// ================================
+// Export the Main Instance and Class
+// ================================
 // For ES6 modules (import)
 exports.default = codebolt;
 // For CommonJS compatibility (require)
 module.exports = codebolt;
 module.exports.default = codebolt;
+module.exports.Codebolt = Codebolt_1.default;

package/dist/modules/llm.d.ts

@@ -11,16 +11,36 @@ declare const cbllm: {
      * for the role is not found, it falls back to the default model for the current agent,
      * and ultimately to the default application-wide LLM if necessary.
      *
-     * @param message - The inference parameters including:
+     * @param params - The inference parameters including:
      *   - messages: Array of conversation messages
      *   - tools: Available tools for the model to use
      *   - tool_choice: How the model should use tools
-     *   - llmrole: The LLM role to determine which model to use
+     *   - full: Whether to return full response
      *   - max_tokens: Maximum number of tokens to generate
      *   - temperature: Temperature for response generation
      *   - stream: Whether to stream the response
      * @returns A promise that resolves with the LLM's response
      */
-    inference: (message: string, llmrole: string) => Promise<LLMResponse>;
+    inference: (params: {
+        messages: Message[];
+        tools?: any[];
+        tool_choice?: string;
+        full?: boolean;
+        llmrole?: string;
+        max_tokens?: number;
+        temperature?: number;
+        stream?: boolean;
+    }, llmrole?: string) => Promise<{
+        completion: any;
+    }>;
+    /**
+     * Legacy method for backward compatibility - converts simple string prompt to message format.
+     * @deprecated Use the new inference method with proper message format instead.
+     *
+     * @param {string} message - The input message or prompt to be sent to the LLM.
+     * @param {string} llmrole - The role of the LLM to determine which model to use.
+     * @returns {Promise<LLMResponse>} A promise that resolves with the LLM's response.
+     */
+    legacyInference: (message: string, llmrole: string) => Promise<LLMResponse>;
 };
 export default cbllm;

package/dist/modules/llm.js

@@ -14,17 +14,34 @@ const cbllm = {
      * for the role is not found, it falls back to the default model for the current agent,
      * and ultimately to the default application-wide LLM if necessary.
      *
-     * @param message - The inference parameters including:
+     * @param params - The inference parameters including:
      *   - messages: Array of conversation messages
      *   - tools: Available tools for the model to use
      *   - tool_choice: How the model should use tools
-     *   - llmrole: The LLM role to determine which model to use
+     *   - full: Whether to return full response
      *   - max_tokens: Maximum number of tokens to generate
      *   - temperature: Temperature for response generation
      *   - stream: Whether to stream the response
      * @returns A promise that resolves with the LLM's response
      */
-    inference: async (message, llmrole) => {
+    inference: async (params, llmrole) => {
+        return websocket_1.default.messageManager.sendAndWaitForResponse({
+            "type": "inference",
+            "message": {
+                prompt: params,
+                llmrole
+            },
+        }, "llmResponse");
+    },
+    /**
+     * Legacy method for backward compatibility - converts simple string prompt to message format.
+     * @deprecated Use the new inference method with proper message format instead.
+     *
+     * @param {string} message - The input message or prompt to be sent to the LLM.
+     * @param {string} llmrole - The role of the LLM to determine which model to use.
+     * @returns {Promise<LLMResponse>} A promise that resolves with the LLM's response.
+     */
+    legacyInference: async (message, llmrole) => {
         return websocket_1.default.messageManager.sendAndWaitForResponse({
             "type": "inference",
             "message": {

package/dist/modules/project.d.ts

@@ -1,13 +1,13 @@
-import { GetProjectPathResponse } from '../types/socketMessageTypes';
+import { GetProjectPathResponse, GetProjectSettingsResponse } from '../types/socketMessageTypes';
 /**
  * A module for interacting with project settings and paths.
  */
 declare const cbproject: {
     /**
      * Retrieves the project settings from the server.
-     * @returns {Promise<any>} A promise that resolves with the project settings response.
+     * @returns {Promise<GetProjectSettingsResponse>} A promise that resolves with the project settings response.
      */
-    getProjectSettings: () => Promise<any>;
+    getProjectSettings: () => Promise<GetProjectSettingsResponse>;
     /**
      * Retrieves the path of the current project.
      * @returns {Promise<GetProjectPathResponse>} A promise that resolves with the project path response.

package/dist/modules/project.js

@@ -10,7 +10,7 @@ const websocket_1 = __importDefault(require("../core/websocket"));
 const cbproject = {
     /**
      * Retrieves the project settings from the server.
-     * @returns {Promise<any>} A promise that resolves with the project settings response.
+     * @returns {Promise<GetProjectSettingsResponse>} A promise that resolves with the project settings response.
      */
     getProjectSettings: () => {
         return websocket_1.default.messageManager.sendAndWaitForResponse({

package/dist/modules/state.d.ts

@@ -1,4 +1,4 @@
-import { AddToAgentStateResponse, GetAgentStateResponse } from '../types/socketMessageTypes';
+import { AddToAgentStateResponse, GetAgentStateResponse, GetProjectStateResponse, UpdateProjectStateResponse } from '../types/socketMessageTypes';
 import { ApplicationState } from '../types/commonTypes';
 declare const cbstate: {
     /**
@@ -22,11 +22,13 @@ declare const cbstate: {
      * Retrieves the current project state from the server via WebSocket.
      * @returns {Promise<GetProjectStateResponse>} A promise that resolves with the project's state.
      */
-    getProjectState: () => Promise<any>;
+    getProjectState: () => Promise<GetProjectStateResponse>;
     /**
      * Updates the project state on the server via WebSocket.
+     * @param {string} key - The key to update in the project state.
+     * @param {any} value - The value to set for the key.
      * @returns {Promise<UpdateProjectStateResponse>} A promise that resolves with the response to the update request.
      */
-    updateProjectState: (key: string, value: any) => Promise<any>;
+    updateProjectState: (key: string, value: any) => Promise<UpdateProjectStateResponse>;
 };
 export default cbstate;

package/dist/modules/state.js

@@ -53,6 +53,8 @@ const cbstate = {
     },
     /**
      * Updates the project state on the server via WebSocket.
+     * @param {string} key - The key to update in the project state.
+     * @param {any} value - The value to set for the key.
      * @returns {Promise<UpdateProjectStateResponse>} A promise that resolves with the response to the update request.
      */
     updateProjectState: async (key, value) => {