@codebolt/agent 1.2.1 → 2.2.3

package/dist/processor-pieces/messageModifiers/memoryImportModifier.js

@@ -0,0 +1,129 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.MemoryImportModifier = void 0;
+const base_1 = require("../base");
+const fs = __importStar(require("fs"));
+const path = __importStar(require("path"));
+class MemoryImportModifier extends base_1.BaseMessageModifier {
+    constructor(options = {}) {
+        super();
+        this.options = {
+            enableMemoryImport: options.enableMemoryImport !== false,
+            maxFileSize: options.maxFileSize || 1024 * 1024, // 1MB
+            allowedExtensions: options.allowedExtensions || ['.md', '.txt', '.json', '.yml', '.yaml', '.xml'],
+            basePath: options.basePath || process.cwd(),
+            maxImports: options.maxImports || 10
+        };
+    }
+    async modify(originalRequest, createdMessage) {
+        try {
+            if (!this.options.enableMemoryImport) {
+                return createdMessage;
+            }
+            // Get the user message content to process
+            const userMessage = createdMessage.message.messages.find(msg => msg.role === 'user');
+            if (!userMessage || typeof userMessage.content !== 'string') {
+                return createdMessage;
+            }
+            // Look for @path/to/file patterns (memory import syntax)
+            const memoryImportPattern = /@([^\s@]+(?:\.[a-zA-Z0-9]+)?)/g;
+            let content = userMessage.content;
+            const processedImports = [];
+            let match;
+            let importCount = 0;
+            while ((match = memoryImportPattern.exec(userMessage.content)) !== null && importCount < this.options.maxImports) {
+                const filePath = match[1];
+                const fullPath = path.isAbsolute(filePath) ? filePath : path.join(this.options.basePath, filePath);
+                try {
+                    // Check if file exists and is allowed
+                    const stats = await fs.promises.stat(fullPath);
+                    if (!stats.isFile()) {
+                        continue;
+                    }
+                    if (stats.size > this.options.maxFileSize) {
+                        content = content.replace(match[0], `[File too large: ${filePath}]`);
+                        continue;
+                    }
+                    const ext = path.extname(filePath);
+                    if (!this.options.allowedExtensions.includes(ext)) {
+                        content = content.replace(match[0], `[Unsupported file type: ${filePath}]`);
+                        continue;
+                    }
+                    // Read and import the file content
+                    const fileContent = await fs.promises.readFile(fullPath, 'utf-8');
+                    const importedContent = `--- Content from ${filePath} ---\n${fileContent}\n--- End of ${filePath} ---`;
+                    content = content.replace(match[0], importedContent);
+                    processedImports.push(filePath);
+                    importCount++;
+                }
+                catch (error) {
+                    // Replace with error message if file can't be read
+                    content = content.replace(match[0], `[Error reading ${filePath}: ${error instanceof Error ? error.message : 'unknown error'}]`);
+                }
+            }
+            if (processedImports.length === 0) {
+                return createdMessage;
+            }
+            // Update the user message with processed content
+            const messages = createdMessage.message.messages.map(msg => {
+                if (msg.role === 'user' && msg === userMessage) {
+                    return {
+                        ...msg,
+                        content
+                    };
+                }
+                return msg;
+            });
+            return Promise.resolve({
+                message: {
+                    ...createdMessage.message,
+                    messages
+                },
+                metadata: {
+                    ...createdMessage.metadata,
+                    memoryImportsProcessed: true,
+                    importedFiles: processedImports,
+                    totalImports: processedImports.length
+                }
+            });
+        }
+        catch (error) {
+            console.error('Error in MemoryImportModifier:', error);
+            return createdMessage;
+        }
+    }
+}
+exports.MemoryImportModifier = MemoryImportModifier;

package/dist/processor-pieces/messageModifiers/mentionedFilesModifier.d.ts

@@ -0,0 +1,23 @@
+import { ProcessedMessage } from "@codebolt/types/agent";
+import { BaseMessageModifier } from "../base";
+import { FlatUserMessage } from "@codebolt/types/sdk";
+/**
+ * Message modifier that reads files and folders mentioned in the user message
+ * Uses mentionedFiles and mentionedFolders from FlatUserMessage
+ */
+export declare class MentionedFilesModifier extends BaseMessageModifier {
+    constructor();
+    modify(originalRequest: FlatUserMessage, createdMessage: ProcessedMessage): Promise<ProcessedMessage>;
+    /**
+     * Resolves and reads a file, handling both files and directories
+     */
+    private resolveAndReadFile;
+    /**
+     * Formats file content for inclusion in the message
+     */
+    private formatFileContent;
+    /**
+     * Updates the user message with file content
+     */
+    private updateUserMessage;
+}

package/dist/processor-pieces/messageModifiers/mentionedFilesModifier.js

@@ -0,0 +1,207 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.MentionedFilesModifier = void 0;
+const base_1 = require("../base");
+const fs = __importStar(require("node:fs/promises"));
+const path = __importStar(require("node:path"));
+const codeboltjs_1 = __importDefault(require("@codebolt/codeboltjs"));
+/**
+ * Message modifier that reads files and folders mentioned in the user message
+ * Uses mentionedFiles and mentionedFolders from FlatUserMessage
+ */
+class MentionedFilesModifier extends base_1.BaseMessageModifier {
+    constructor() {
+        super();
+    }
+    async modify(originalRequest, createdMessage) {
+        try {
+            const mentionedFiles = originalRequest.mentionedFiles || [];
+            const mentionedFolders = originalRequest.mentionedFolders || [];
+            const allMentionedPaths = [...mentionedFiles, ...mentionedFolders];
+            if (allMentionedPaths.length === 0) {
+                return createdMessage;
+            }
+            // Read all mentioned files and folders
+            const fileResults = [];
+            for (const mentionedPath of allMentionedPaths) {
+                try {
+                    const result = await this.resolveAndReadFile(mentionedPath);
+                    if (result) {
+                        fileResults.push(result);
+                    }
+                }
+                catch (error) {
+                    fileResults.push({
+                        path: mentionedPath,
+                        content: '',
+                        error: error instanceof Error ? error.message : 'Unknown error'
+                    });
+                }
+            }
+            if (fileResults.length === 0) {
+                return createdMessage;
+            }
+            // Add file content to the user message
+            const fileContentText = this.formatFileContent(fileResults);
+            const updatedMessages = this.updateUserMessage(createdMessage.message.messages || [], fileContentText);
+            return {
+                message: {
+                    ...createdMessage.message,
+                    messages: updatedMessages
+                },
+                metadata: {
+                    ...createdMessage.metadata,
+                    mentionedFilesAdded: true,
+                    resolvedFiles: fileResults.map(f => f.path)
+                }
+            };
+        }
+        catch (error) {
+            console.error('Error in MentionedFilesModifier:', error);
+            return createdMessage;
+        }
+    }
+    /**
+     * Resolves and reads a file, handling both files and directories
+     */
+    async resolveAndReadFile(pathName) {
+        try {
+            // pathName is already a full absolute path
+            const absolutePath = pathName;
+            // Get project path for security check and relative path display
+            let projectPath;
+            try {
+                const { projectPath: path } = await codeboltjs_1.default.project.getProjectPath();
+                projectPath = path || process.cwd();
+            }
+            catch (error) {
+                projectPath = process.cwd();
+            }
+            // Check if path is within project (basic security check)
+            const relativePath = path.relative(projectPath, absolutePath);
+            if (relativePath.startsWith('..')) {
+                throw new Error(`Path ${pathName} is outside project directory`);
+            }
+            const stats = await fs.stat(absolutePath);
+            if (stats.isDirectory()) {
+                // For directories, list contents
+                const files = await fs.readdir(absolutePath);
+                const fileList = files.join('\n');
+                return {
+                    path: relativePath || path.basename(pathName),
+                    content: `Directory contents:\n${fileList}`
+                };
+            }
+            else if (stats.isFile()) {
+                // Check file size (limit to reasonable size)
+                const maxFileSize = 1024 * 1024; // 1MB
+                if (stats.size > maxFileSize) {
+                    return {
+                        path: relativePath || path.basename(pathName),
+                        content: `File too large (${stats.size} bytes). Maximum allowed: ${maxFileSize} bytes.`
+                    };
+                }
+                // Read file content
+                const content = await fs.readFile(absolutePath, 'utf-8');
+                return {
+                    path: relativePath || path.basename(pathName),
+                    content
+                };
+            }
+        }
+        catch (error) {
+            if (error instanceof Error && 'code' in error && error.code === 'ENOENT') {
+                // File not found
+                return {
+                    path: pathName,
+                    content: '',
+                    error: `File not found: ${pathName}`
+                };
+            }
+            throw error;
+        }
+        return null;
+    }
+    /**
+     * Formats file content for inclusion in the message
+     */
+    formatFileContent(fileResults) {
+        const contentParts = [];
+        contentParts.push('\n--- Content from mentioned files ---');
+        for (const result of fileResults) {
+            if (result.error) {
+                contentParts.push(`\nError reading ${result.path}: ${result.error}`);
+            }
+            else {
+                contentParts.push(`\nContent from ${result.path}:`);
+                contentParts.push(result.content);
+            }
+        }
+        return contentParts.join('\n');
+    }
+    /**
+     * Updates the user message with file content
+     */
+    updateUserMessage(messages, fileContent) {
+        const updatedMessages = [...messages];
+        // Find the last user message and append file content
+        for (let i = updatedMessages.length - 1; i >= 0; i--) {
+            if (updatedMessages[i].role === 'user') {
+                const currentContent = updatedMessages[i].content;
+                if (Array.isArray(currentContent)) {
+                    // If content is array, add file content as new text element
+                    updatedMessages[i] = {
+                        ...updatedMessages[i],
+                        content: [...currentContent, { type: 'text', text: fileContent }]
+                    };
+                }
+                else {
+                    // If content is string, append file content
+                    updatedMessages[i] = {
+                        ...updatedMessages[i],
+                        content: (currentContent || '') + fileContent
+                    };
+                }
+                break;
+            }
+        }
+        return updatedMessages;
+    }
+}
+exports.MentionedFilesModifier = MentionedFilesModifier;

package/dist/processor-pieces/messageModifiers/simpleMessageModifier.d.ts

@@ -0,0 +1,7 @@
+import { ProcessedMessage } from "@codebolt/types/agent";
+import { BaseMessageModifier } from "../base";
+import { FlatUserMessage } from "@codebolt/types/sdk";
+export declare class SimpleMessageModifier extends BaseMessageModifier {
+    constructor();
+    modify(originalRequest: FlatUserMessage, createdMessage: ProcessedMessage, context?: Record<string, unknown>): Promise<ProcessedMessage>;
+}

package/dist/processor-pieces/messageModifiers/simpleMessageModifier.js

@@ -0,0 +1,26 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.SimpleMessageModifier = void 0;
+const base_1 = require("../base");
+class SimpleMessageModifier extends base_1.BaseMessageModifier {
+    constructor() {
+        super();
+    }
+    modify(originalRequest, createdMessage, context) {
+        const contextMessage = {
+            role: 'user',
+            content: "hi"
+        };
+        return Promise.resolve({
+            message: {
+                ...createdMessage.message,
+                messages: [...createdMessage.message.messages, contextMessage]
+            },
+            metadata: {
+                ...createdMessage.metadata,
+                userMessageAdded: true
+            }
+        });
+    }
+}
+exports.SimpleMessageModifier = SimpleMessageModifier;

package/dist/processor-pieces/messageModifiers/toolInjectionModifier.d.ts

@@ -0,0 +1,20 @@
+import { ProcessedMessage } from "@codebolt/types/agent";
+import { BaseMessageModifier } from "../base";
+import { FlatUserMessage } from "@codebolt/types/sdk";
+export interface ToolInjectionOptions {
+    toolsLocation?: 'Tool' | 'SystemMessage' | 'InsidePrompt';
+    includeToolDescriptions?: boolean;
+    maxToolsInMessage?: number;
+    giveToolExamples?: boolean;
+    maxToolExamples?: number;
+}
+export declare class ToolInjectionModifier extends BaseMessageModifier {
+    private readonly options;
+    constructor(options?: ToolInjectionOptions);
+    modify(originalRequest: FlatUserMessage, createdMessage: ProcessedMessage): Promise<ProcessedMessage>;
+    private addToolsInsidePrompt;
+    private addToolsAsSystemMessage;
+    private addToolsAsToolCalls;
+    private generateToolsInfo;
+    private generateToolExample;
+}

package/dist/processor-pieces/messageModifiers/toolInjectionModifier.js

@@ -0,0 +1,152 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ToolInjectionModifier = void 0;
+const base_1 = require("../base");
+const codeboltjs_1 = __importDefault(require("@codebolt/codeboltjs"));
+class ToolInjectionModifier extends base_1.BaseMessageModifier {
+    constructor(options = {}) {
+        super();
+        this.options = {
+            toolsLocation: options.toolsLocation || 'Tool',
+            includeToolDescriptions: options.includeToolDescriptions !== false,
+            maxToolsInMessage: options.maxToolsInMessage || 30,
+            giveToolExamples: options.giveToolExamples || false,
+            maxToolExamples: options.maxToolExamples || 2
+        };
+    }
+    async modify(originalRequest, createdMessage) {
+        try {
+            const toolsResponse = await codeboltjs_1.default.mcp.listMcpFromServers(['codebolt']);
+            const tools = (toolsResponse === null || toolsResponse === void 0 ? void 0 : toolsResponse.data) || [];
+            if (tools.length === 0) {
+                return createdMessage;
+            }
+            switch (this.options.toolsLocation) {
+                case 'InsidePrompt':
+                    //@ts-ignore
+                    return this.addToolsInsidePrompt(createdMessage, tools);
+                //@ts-ignore
+                case 'SystemMessage':
+                    //@ts-ignore
+                    return this.addToolsAsSystemMessage(createdMessage, tools);
+                case 'Tool':
+                    //@ts-ignore
+                    return this.addToolsAsToolCalls(createdMessage, tools);
+                default:
+                    //@ts-ignore
+                    return this.addToolsAsToolCalls(createdMessage, tools);
+            }
+        }
+        catch (error) {
+            console.error('Error in ToolInjectionModifier:', error);
+            return createdMessage;
+        }
+    }
+    addToolsInsidePrompt(createdMessage, tools) {
+        const toolsInfo = this.generateToolsInfo(tools);
+        // Modify the last user message to include tools information
+        const modifiedMessages = createdMessage.message.messages.map((message, index) => {
+            if (message.role === 'user' && index === createdMessage.message.messages.length - 1) {
+                return {
+                    ...message,
+                    content: `${message.content}\n\n--- Available Tools ---\n${toolsInfo}`
+                };
+            }
+            return message;
+        });
+        return {
+            message: {
+                ...createdMessage.message,
+                messages: modifiedMessages
+            },
+            metadata: {
+                ...createdMessage.metadata,
+                toolsInjected: true,
+                toolsLocation: 'InsidePrompt',
+                toolsCount: tools.length
+            }
+        };
+    }
+    addToolsAsSystemMessage(createdMessage, tools) {
+        const toolsInfo = this.generateToolsInfo(tools);
+        const toolsMessage = {
+            role: 'system',
+            content: `Available Tools:\n${toolsInfo}`
+        };
+        return {
+            message: {
+                ...createdMessage.message,
+                messages: [toolsMessage, ...createdMessage.message.messages]
+            },
+            metadata: {
+                ...createdMessage.metadata,
+                toolsInjected: true,
+                toolsLocation: 'SystemMessage',
+                toolsCount: tools.length
+            }
+        };
+    }
+    addToolsAsToolCalls(createdMessage, tools) {
+        return {
+            message: {
+                ...createdMessage.message,
+                tools,
+                tool_choice: 'auto'
+            },
+            metadata: {
+                ...createdMessage.metadata,
+                toolsInjected: true,
+                toolsLocation: 'Tool',
+                toolsCount: tools.length
+            }
+        };
+    }
+    generateToolsInfo(tools) {
+        const toolsInfo = [];
+        tools.slice(0, this.options.maxToolsInMessage).forEach((tool, index) => {
+            let toolInfo = `${index + 1}. **${tool.function.name}**`;
+            if (this.options.includeToolDescriptions && tool.function.description) {
+                toolInfo += `\n   Description: ${tool.function.description}`;
+            }
+            if (tool.function.parameters) {
+                const params = typeof tool.function.parameters === 'object'
+                    ? Object.keys(tool.function.parameters).join(', ')
+                    : String(tool.function.parameters);
+                toolInfo += `\n   Parameters: ${params}`;
+            }
+            if (this.options.giveToolExamples && index < this.options.maxToolExamples) {
+                const example = this.generateToolExample(tool);
+                if (example) {
+                    toolInfo += `\n   Example: ${example}`;
+                }
+            }
+            toolsInfo.push(toolInfo);
+        });
+        return toolsInfo.join('\n\n');
+    }
+    generateToolExample(tool) {
+        // Generate simple examples based on tool name
+        const toolName = tool.function.name.toLowerCase();
+        if (toolName.includes('read') && toolName.includes('file')) {
+            return `${tool.function.name}({ filePath: "example.txt" })`;
+        }
+        if (toolName.includes('write') && toolName.includes('file')) {
+            return `${tool.function.name}({ filePath: "output.txt", content: "Hello World" })`;
+        }
+        if (toolName.includes('delete') && toolName.includes('file')) {
+            return `${tool.function.name}({ filePath: "temp.txt" })`;
+        }
+        if (toolName.includes('move') && toolName.includes('file')) {
+            return `${tool.function.name}({ sourcePath: "old.txt", destinationPath: "new.txt" })`;
+        }
+        if (toolName.includes('copy') && toolName.includes('file')) {
+            return `${tool.function.name}({ sourcePath: "source.txt", destinationPath: "copy.txt" })`;
+        }
+        // Generic example
+        return `${tool.function.name}({ /* parameters */ })`;
+    }
+}
+exports.ToolInjectionModifier = ToolInjectionModifier;

package/dist/processor-pieces/messageModifiers/workingDirectoryMessageModifier.d.ts

@@ -0,0 +1,38 @@
+import { ProcessedMessage } from "@codebolt/types/agent";
+import { BaseMessageModifier } from "../base";
+import { FlatUserMessage } from "@codebolt/types/sdk";
+export interface WorkingDirectoryMessageModifierOptions {
+    showFolderStructureSummary?: boolean;
+    listFiles?: boolean;
+    listFilesDepth?: number;
+    listFilesLimit?: number;
+    listFilesIgnoreFromGitignore?: boolean;
+    listFilesIgnore?: string[];
+    maxItems?: number;
+    showFullDirectoryContext?: boolean;
+    [key: string]: unknown;
+}
+export declare class WorkingDirectoryMessageModifier extends BaseMessageModifier {
+    private readonly showFolderStructureSummary;
+    private readonly listFiles;
+    private readonly listFilesDepth;
+    private readonly listFilesLimit;
+    private readonly listFilesIgnoreFromGitignore;
+    private readonly listFilesIgnore;
+    private readonly maxItems;
+    private readonly showFullDirectoryContext;
+    private readonly defaultIgnoredFolders;
+    private readonly truncationIndicator;
+    constructor(options?: WorkingDirectoryMessageModifierOptions);
+    modify(originalRequest: FlatUserMessage, createdMessage: ProcessedMessage): Promise<ProcessedMessage>;
+    private generateFolderStructureSummary;
+    private generateFilesList;
+    private collectDirectories;
+    private collectFiles;
+    private shouldIgnore;
+    private getDirectoryContextString;
+    private getFolderStructure;
+    private readFullStructure;
+    private formatStructure;
+    private isTruncated;
+}