@codebolt/agent 2.2.2 → 2.2.5

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

@@ -1,4 +1,5 @@
-import { BaseMessageModifier, MessageModifierInput, ProcessedMessage } from '../../processor';
+import { FlatUserMessage } from '@codebolt/types/sdk';
+import { BaseMessageModifier, ProcessedMessage } from '../../processor';
 export interface ImageAttachmentMessageModifierOptions {
     passImageAs?: 'URL' | 'Base64' | 'Path';
     supportedFormats?: string[];
@@ -9,7 +10,7 @@ export declare class ImageAttachmentMessageModifier extends BaseMessageModifier
     private readonly supportedFormats;
     private readonly maxImageSize;
     constructor(options?: ImageAttachmentMessageModifierOptions);
-    modify(input: MessageModifierInput): Promise<ProcessedMessage>;
+    modify(originalRequest: FlatUserMessage, createdMessage: ProcessedMessage): Promise<ProcessedMessage>;
     private extractImageReferences;
     private isImagePath;
     private processImage;

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

@@ -44,9 +44,8 @@ class ImageAttachmentMessageModifier extends processor_1.BaseMessageModifier {
         this.supportedFormats = options.supportedFormats || ['.png', '.jpg', '.jpeg', '.gif', '.webp', '.bmp'];
         this.maxImageSize = options.maxImageSize || 10 * 1024 * 1024; // 10MB default
     }
-    async modify(input) {
+    async modify(originalRequest, createdMessage) {
         try {
-            const { originalRequest, createdMessage, context } = input;
             // Extract image references from the request
             const imageReferences = this.extractImageReferences(originalRequest);
             if (imageReferences.length === 0) {

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/workingDirectoryMessageModifier.d.ts

@@ -1,4 +1,6 @@
-import { BaseMessageModifier, MessageModifierInput, ProcessedMessage } from '../../processor';
+import { ProcessedMessage } from "@codebolt/types/agent";
+import { BaseMessageModifier } from "../base";
+import { FlatUserMessage } from "@codebolt/types/sdk";
 export interface WorkingDirectoryMessageModifierOptions {
     showFolderStructureSummary?: boolean;
     listFiles?: boolean;
@@ -6,6 +8,9 @@ export interface WorkingDirectoryMessageModifierOptions {
     listFilesLimit?: number;
     listFilesIgnoreFromGitignore?: boolean;
     listFilesIgnore?: string[];
+    maxItems?: number;
+    showFullDirectoryContext?: boolean;
+    [key: string]: unknown;
 }
 export declare class WorkingDirectoryMessageModifier extends BaseMessageModifier {
     private readonly showFolderStructureSummary;
@@ -14,9 +19,20 @@ export declare class WorkingDirectoryMessageModifier extends BaseMessageModifier
     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(input: MessageModifierInput): Promise<ProcessedMessage>;
+    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;
 }