@codebolt/agent 1.2.1 → 2.2.3

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

@@ -0,0 +1,59 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.HandleUrlMessageModifier = void 0;
+const processor_1 = require("../../processor");
+class HandleUrlMessageModifier extends processor_1.BaseMessageModifier {
+    constructor(options = {}) {
+        super({ context: options });
+        this.googleSearchEnabled = options.googleSearchEnabled || false;
+        this.fetchUrlContent = options.fetchUrlContent || false;
+    }
+    async modify(originalRequest, createdMessage) {
+        try {
+            // Check if the request contains URLs
+            const urls = this.extractUrls(originalRequest);
+            if (urls.length > 0) {
+                let contextContent = `URLs detected: ${urls.join(', ')}. ${this.googleSearchEnabled ? 'Google search is enabled.' : 'Google search is disabled.'}`;
+                // Fetch URL content if enabled
+                if (this.fetchUrlContent) {
+                    contextContent += '\n\nURL Content Fetching: Enabled (content would be fetched in a real implementation)';
+                    // In a real implementation, you would fetch the content here
+                    // const urlContents = await Promise.all(urls.map(url => this.fetchContent(url)));
+                }
+                // Add URL context to the message
+                const urlContextMessage = {
+                    role: 'system',
+                    content: contextContent,
+                    name: 'url-detector'
+                };
+                return {
+                    messages: [...createdMessage.messages, urlContextMessage],
+                    metadata: {
+                        ...createdMessage.metadata,
+                        urlsDetected: urls,
+                        googleSearchEnabled: this.googleSearchEnabled,
+                        fetchUrlContent: this.fetchUrlContent
+                    }
+                };
+            }
+            return createdMessage;
+        }
+        catch (error) {
+            console.error('Error in HandleUrlMessageModifier:', error);
+            throw error;
+        }
+    }
+    extractUrls(request) {
+        if (typeof request === 'string') {
+            const urlRegex = /https?:\/\/[^\s]+/g;
+            return request.match(urlRegex) || [];
+        }
+        if (typeof request === 'object' && request !== null) {
+            const requestString = JSON.stringify(request);
+            const urlRegex = /https?:\/\/[^\s]+/g;
+            return requestString.match(urlRegex) || [];
+        }
+        return [];
+    }
+}
+exports.HandleUrlMessageModifier = HandleUrlMessageModifier;

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

@@ -0,0 +1,34 @@
+import { ProcessedMessage } from "@codebolt/types/agent";
+import { BaseMessageModifier } from "../base";
+import { FlatUserMessage } from "@codebolt/types/sdk";
+export interface IdeContextOptions {
+    includeOpenFiles?: boolean;
+    includeActiveFile?: boolean;
+    includeCursorPosition?: boolean;
+    includeSelectedText?: boolean;
+    maxOpenFiles?: number;
+}
+export interface FileInfo {
+    path: string;
+    isActive?: boolean;
+    cursor?: {
+        line: number;
+        character: number;
+    };
+    selectedText?: string;
+}
+export interface IdeContext {
+    workspaceState?: {
+        openFiles: FileInfo[];
+    };
+}
+export declare class IdeContextModifier extends BaseMessageModifier {
+    private readonly options;
+    private lastSentIdeContext?;
+    private forceFullContext;
+    constructor(options?: IdeContextOptions);
+    modify(originalRequest: FlatUserMessage, createdMessage: ProcessedMessage): Promise<ProcessedMessage>;
+    private getIdeContext;
+    private getIdeContextParts;
+    setForceFullContext(force: boolean): void;
+}

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

@@ -0,0 +1,157 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.IdeContextModifier = void 0;
+const base_1 = require("../base");
+class IdeContextModifier extends base_1.BaseMessageModifier {
+    constructor(options = {}) {
+        super();
+        this.forceFullContext = true;
+        this.options = {
+            includeOpenFiles: options.includeOpenFiles !== false,
+            includeActiveFile: options.includeActiveFile !== false,
+            includeCursorPosition: options.includeCursorPosition !== false,
+            includeSelectedText: options.includeSelectedText !== false,
+            maxOpenFiles: options.maxOpenFiles || 10
+        };
+    }
+    modify(originalRequest, createdMessage) {
+        try {
+            // Get IDE context from the original request (activeFile, openedFiles)
+            const ideContext = this.getIdeContext(originalRequest, createdMessage.metadata);
+            if (!ideContext) {
+                return Promise.resolve(createdMessage);
+            }
+            const { contextParts, newIdeContext } = this.getIdeContextParts(this.forceFullContext, ideContext);
+            if (contextParts.length === 0) {
+                return Promise.resolve(createdMessage);
+            }
+            const ideContextMessage = {
+                role: 'system',
+                content: contextParts.join('\n')
+            };
+            // Find existing system message or add new one
+            const messages = [...createdMessage.message.messages];
+            const systemMessageIndex = messages.findIndex(msg => msg.role === 'system');
+            if (systemMessageIndex !== -1) {
+                // Append to existing system message
+                messages[systemMessageIndex] = {
+                    ...messages[systemMessageIndex],
+                    content: `${messages[systemMessageIndex].content}\n\n${ideContextMessage.content}`
+                };
+            }
+            else {
+                // Add new system message
+                messages.unshift(ideContextMessage);
+            }
+            this.lastSentIdeContext = newIdeContext;
+            this.forceFullContext = false;
+            return Promise.resolve({
+                message: {
+                    ...createdMessage.message,
+                    messages
+                },
+                metadata: {
+                    ...createdMessage.metadata,
+                    ideContextAdded: true,
+                    ideContextType: this.forceFullContext ? 'full' : 'incremental'
+                }
+            });
+        }
+        catch (error) {
+            console.error('Error in IdeContextModifier:', error);
+            return Promise.resolve(createdMessage);
+        }
+    }
+    getIdeContext(originalRequest, metadata) {
+        // Try to get IDE context from the metadata parameter first
+        if (metadata === null || metadata === void 0 ? void 0 : metadata.ideContext) {
+            return metadata.ideContext;
+        }
+        // Extract IDE context from the original message
+        const activeFile = originalRequest.activeFile;
+        const openedFiles = originalRequest.openedFiles;
+        if (!activeFile && (!openedFiles || openedFiles.length === 0)) {
+            return undefined;
+        }
+        const openFiles = [];
+        // Add active file if present (activeFile is just a path string)
+        if (activeFile) {
+            const activeFileInfo = {
+                path: activeFile,
+                isActive: true
+            };
+            openFiles.push(activeFileInfo);
+        }
+        // Add other opened files (openedFiles is array of path strings)
+        if (openedFiles && Array.isArray(openedFiles)) {
+            const otherFiles = openedFiles
+                .filter(filePath => filePath !== activeFile) // Exclude active file from others
+                .map(filePath => ({
+                path: filePath,
+                isActive: false
+            }));
+            openFiles.push(...otherFiles);
+        }
+        if (openFiles.length === 0) {
+            return undefined;
+        }
+        return {
+            workspaceState: {
+                openFiles
+            }
+        };
+    }
+    getIdeContextParts(forceFullContext, currentIdeContext) {
+        var _a;
+        if (!((_a = currentIdeContext === null || currentIdeContext === void 0 ? void 0 : currentIdeContext.workspaceState) === null || _a === void 0 ? void 0 : _a.openFiles)) {
+            return { contextParts: [], newIdeContext: undefined };
+        }
+        if (forceFullContext || !this.lastSentIdeContext) {
+            // Send full context as JSON
+            const openFiles = currentIdeContext.workspaceState.openFiles;
+            const activeFile = openFiles.find((f) => f.isActive);
+            const otherOpenFiles = openFiles
+                .filter((f) => !f.isActive)
+                .map((f) => f.path)
+                .slice(0, this.options.maxOpenFiles - 1); // Reserve space for active file
+            const contextData = {};
+            if (activeFile && this.options.includeActiveFile) {
+                const activeFileData = {
+                    path: activeFile.path
+                };
+                if (this.options.includeCursorPosition && activeFile.cursor) {
+                    activeFileData.cursor = {
+                        line: activeFile.cursor.line,
+                        character: activeFile.cursor.character,
+                    };
+                }
+                if (this.options.includeSelectedText && activeFile.selectedText) {
+                    activeFileData.selectedText = activeFile.selectedText;
+                }
+                contextData['activeFile'] = activeFileData;
+            }
+            if (this.options.includeOpenFiles && otherOpenFiles.length > 0) {
+                contextData['otherOpenFiles'] = otherOpenFiles;
+            }
+            if (Object.keys(contextData).length === 0) {
+                return { contextParts: [], newIdeContext: currentIdeContext };
+            }
+            const jsonString = JSON.stringify(contextData, null, 2);
+            const contextParts = [
+                "Here is the user's editor context as a JSON object. This is for your information only.",
+                '```json',
+                jsonString,
+                '```',
+                "Don't acknowledge this context in your response unless specifically asked about it."
+            ];
+            return { contextParts, newIdeContext: currentIdeContext };
+        }
+        // For incremental updates, we would compare with lastSentIdeContext
+        // and only send changes. For now, we'll skip incremental updates.
+        return { contextParts: [], newIdeContext: currentIdeContext };
+    }
+    setForceFullContext(force) {
+        this.forceFullContext = force;
+    }
+}
+exports.IdeContextModifier = IdeContextModifier;

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

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

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

@@ -0,0 +1,221 @@
+"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.ImageAttachmentMessageModifier = void 0;
+const processor_1 = require("../../processor");
+const fs = __importStar(require("fs/promises"));
+const path = __importStar(require("path"));
+class ImageAttachmentMessageModifier extends processor_1.BaseMessageModifier {
+    constructor(options = {}) {
+        super({ context: options });
+        this.passImageAs = options.passImageAs || 'URL';
+        this.supportedFormats = options.supportedFormats || ['.png', '.jpg', '.jpeg', '.gif', '.webp', '.bmp'];
+        this.maxImageSize = options.maxImageSize || 10 * 1024 * 1024; // 10MB default
+    }
+    async modify(originalRequest, createdMessage) {
+        try {
+            // Extract image references from the request
+            const imageReferences = this.extractImageReferences(originalRequest);
+            if (imageReferences.length === 0) {
+                return createdMessage;
+            }
+            const processedImages = [];
+            for (const imageRef of imageReferences) {
+                try {
+                    const processedImage = await this.processImage(imageRef);
+                    if (processedImage) {
+                        processedImages.push(processedImage);
+                    }
+                }
+                catch (error) {
+                    console.error(`Error processing image ${imageRef}:`, error);
+                }
+            }
+            if (processedImages.length === 0) {
+                return createdMessage;
+            }
+            // Add image context message
+            const imageContextMessage = {
+                role: 'system',
+                content: `Images attached to this conversation (${this.passImageAs} format):\n${processedImages.map((img, idx) => `${idx + 1}. ${img.description || img.reference}`).join('\n')}`,
+                name: 'image-attachments'
+            };
+            // Modify user messages to include image references
+            const modifiedMessages = createdMessage.messages.map(message => {
+                if (message.role === 'user') {
+                    return {
+                        ...message,
+                        attachments: processedImages
+                    };
+                }
+                return message;
+            });
+            return {
+                messages: [imageContextMessage, ...modifiedMessages],
+                metadata: {
+                    ...createdMessage.metadata,
+                    imagesAttached: processedImages.length,
+                    imageFormat: this.passImageAs,
+                    processedImages
+                }
+            };
+        }
+        catch (error) {
+            console.error('Error in ImageAttachmentMessageModifier:', error);
+            throw error;
+        }
+    }
+    extractImageReferences(request) {
+        const imageRefs = [];
+        if (typeof request === 'string') {
+            // Look for image file paths or URLs
+            const imageRegex = /(?:\.png|\.jpg|\.jpeg|\.gif|\.webp|\.bmp|https?:\/\/[^\s]+\.(?:png|jpg|jpeg|gif|webp|bmp))/gi;
+            const matches = request.match(imageRegex);
+            if (matches) {
+                imageRefs.push(...matches);
+            }
+        }
+        if (typeof request === 'object' && request !== null) {
+            // Look for image properties in the request object
+            const searchForImages = (obj) => {
+                if (Array.isArray(obj)) {
+                    obj.forEach(searchForImages);
+                }
+                else if (typeof obj === 'object' && obj !== null) {
+                    Object.values(obj).forEach(value => {
+                        if (typeof value === 'string' && this.isImagePath(value)) {
+                            imageRefs.push(value);
+                        }
+                        else if (typeof value === 'object') {
+                            searchForImages(value);
+                        }
+                    });
+                }
+            };
+            searchForImages(request);
+        }
+        return [...new Set(imageRefs)]; // Remove duplicates
+    }
+    isImagePath(str) {
+        const ext = path.extname(str).toLowerCase();
+        return this.supportedFormats.includes(ext) || /^https?:\/\/[^\s]+\.(?:png|jpg|jpeg|gif|webp|bmp)/i.test(str);
+    }
+    async processImage(imageRef) {
+        try {
+            if (imageRef.startsWith('http')) {
+                // URL image
+                return this.processUrlImage(imageRef);
+            }
+            else {
+                // Local file path
+                return this.processLocalImage(imageRef);
+            }
+        }
+        catch (error) {
+            console.error(`Error processing image ${imageRef}:`, error);
+            return null;
+        }
+    }
+    async processUrlImage(url) {
+        switch (this.passImageAs) {
+            case 'URL':
+                return {
+                    type: 'url',
+                    reference: url,
+                    description: `Image from URL: ${url}`
+                };
+            case 'Base64':
+                // In a real implementation, you would fetch and convert to base64
+                return {
+                    type: 'base64',
+                    reference: url,
+                    description: `Image from URL (would be converted to base64): ${url}`,
+                    note: 'Base64 conversion not implemented for URLs in this demo'
+                };
+            case 'Path':
+                return {
+                    type: 'path',
+                    reference: url,
+                    description: `Image URL: ${url}`
+                };
+            default:
+                return null;
+        }
+    }
+    async processLocalImage(filePath) {
+        try {
+            // Check if file exists and get stats
+            const stats = await fs.stat(filePath);
+            if (stats.size > this.maxImageSize) {
+                console.warn(`Image ${filePath} is too large (${stats.size} bytes)`);
+                return null;
+            }
+            switch (this.passImageAs) {
+                case 'Path':
+                    return {
+                        type: 'path',
+                        reference: filePath,
+                        description: `Local image: ${path.basename(filePath)}`,
+                        size: stats.size
+                    };
+                case 'Base64':
+                    const imageBuffer = await fs.readFile(filePath);
+                    const base64 = imageBuffer.toString('base64');
+                    const ext = path.extname(filePath).substring(1);
+                    return {
+                        type: 'base64',
+                        reference: `data:image/${ext};base64,${base64}`,
+                        description: `Local image (base64): ${path.basename(filePath)}`,
+                        size: stats.size
+                    };
+                case 'URL':
+                    // Convert local path to file:// URL
+                    return {
+                        type: 'url',
+                        reference: `file://${path.resolve(filePath)}`,
+                        description: `Local image (file URL): ${path.basename(filePath)}`,
+                        size: stats.size
+                    };
+                default:
+                    return null;
+            }
+        }
+        catch (error) {
+            console.error(`Error processing local image ${filePath}:`, error);
+            return null;
+        }
+    }
+}
+exports.ImageAttachmentMessageModifier = ImageAttachmentMessageModifier;

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

@@ -0,0 +1,11 @@
+export { EnvironmentContextModifier, type EnvironmentContextOptions } from './environmentContextModifier';
+export { CoreSystemPromptModifier, type CoreSystemPromptOptions } from './coreSystemPromptModifier';
+export { DirectoryContextModifier, type DirectoryContextOptions } from './directoryContextModifier';
+export { IdeContextModifier, type IdeContextOptions, type FileInfo, type IdeContext } from './ideContextModifier';
+export { AtFileProcessorModifier, type AtFileProcessorOptions } from './atFileProcessorModifier';
+export { ArgumentProcessorModifier, type ArgumentProcessorOptions } from './argumentProcessorModifier';
+export { MemoryImportModifier, type MemoryImportOptions } from './memoryImportModifier';
+export { LoopDetectionModifier, type LoopDetectionOptions } from '../postInferenceProcessors/loopDetectionModifier';
+export { ToolInjectionModifier, type ToolInjectionOptions } from './toolInjectionModifier';
+export { ChatRecordingModifier, type ChatRecordingOptions } from './chatRecordingModifier';
+export { ChatHistoryMessageModifier, type ChatHistoryMessageModifierOptions } from './chatHistoryMessageModifier';

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

@@ -0,0 +1,26 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ChatHistoryMessageModifier = exports.ChatRecordingModifier = exports.ToolInjectionModifier = exports.LoopDetectionModifier = exports.MemoryImportModifier = exports.ArgumentProcessorModifier = exports.AtFileProcessorModifier = exports.IdeContextModifier = exports.DirectoryContextModifier = exports.CoreSystemPromptModifier = exports.EnvironmentContextModifier = void 0;
+// Gemini-CLI equivalent modifiers
+var environmentContextModifier_1 = require("./environmentContextModifier");
+Object.defineProperty(exports, "EnvironmentContextModifier", { enumerable: true, get: function () { return environmentContextModifier_1.EnvironmentContextModifier; } });
+var coreSystemPromptModifier_1 = require("./coreSystemPromptModifier");
+Object.defineProperty(exports, "CoreSystemPromptModifier", { enumerable: true, get: function () { return coreSystemPromptModifier_1.CoreSystemPromptModifier; } });
+var directoryContextModifier_1 = require("./directoryContextModifier");
+Object.defineProperty(exports, "DirectoryContextModifier", { enumerable: true, get: function () { return directoryContextModifier_1.DirectoryContextModifier; } });
+var ideContextModifier_1 = require("./ideContextModifier");
+Object.defineProperty(exports, "IdeContextModifier", { enumerable: true, get: function () { return ideContextModifier_1.IdeContextModifier; } });
+var atFileProcessorModifier_1 = require("./atFileProcessorModifier");
+Object.defineProperty(exports, "AtFileProcessorModifier", { enumerable: true, get: function () { return atFileProcessorModifier_1.AtFileProcessorModifier; } });
+var argumentProcessorModifier_1 = require("./argumentProcessorModifier");
+Object.defineProperty(exports, "ArgumentProcessorModifier", { enumerable: true, get: function () { return argumentProcessorModifier_1.ArgumentProcessorModifier; } });
+var memoryImportModifier_1 = require("./memoryImportModifier");
+Object.defineProperty(exports, "MemoryImportModifier", { enumerable: true, get: function () { return memoryImportModifier_1.MemoryImportModifier; } });
+var loopDetectionModifier_1 = require("../postInferenceProcessors/loopDetectionModifier");
+Object.defineProperty(exports, "LoopDetectionModifier", { enumerable: true, get: function () { return loopDetectionModifier_1.LoopDetectionModifier; } });
+var toolInjectionModifier_1 = require("./toolInjectionModifier");
+Object.defineProperty(exports, "ToolInjectionModifier", { enumerable: true, get: function () { return toolInjectionModifier_1.ToolInjectionModifier; } });
+var chatRecordingModifier_1 = require("./chatRecordingModifier");
+Object.defineProperty(exports, "ChatRecordingModifier", { enumerable: true, get: function () { return chatRecordingModifier_1.ChatRecordingModifier; } });
+var chatHistoryMessageModifier_1 = require("./chatHistoryMessageModifier");
+Object.defineProperty(exports, "ChatHistoryMessageModifier", { enumerable: true, get: function () { return chatHistoryMessageModifier_1.ChatHistoryMessageModifier; } });

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

@@ -0,0 +1,15 @@
+import { ProcessedMessage } from "@codebolt/types/agent";
+import { BaseMessageModifier } from "../base";
+import { FlatUserMessage } from "@codebolt/types/sdk";
+export interface MemoryImportOptions {
+    enableMemoryImport?: boolean;
+    maxFileSize?: number;
+    allowedExtensions?: string[];
+    basePath?: string;
+    maxImports?: number;
+}
+export declare class MemoryImportModifier extends BaseMessageModifier {
+    private readonly options;
+    constructor(options?: MemoryImportOptions);
+    modify(originalRequest: FlatUserMessage, createdMessage: ProcessedMessage): Promise<ProcessedMessage>;
+}