rag-lite-ts 2.1.1 → 2.2.0

package/dist/esm/cli/indexer.js

@@ -0,0 +1,471 @@
+import { existsSync, statSync } from 'fs';
+import { resolve } from 'path';
+import { IngestionFactory } from '../factories/ingestion-factory.js';
+import { withCLIDatabaseAccess, setupCLICleanup, isDatabaseBusy } from '../core/cli-database-utils.js';
+import { EXIT_CODES, ConfigurationError } from '../core/config.js';
+/**
+ * Validate mode-specific model and strategy combinations
+ * Ensures that the selected model is compatible with the chosen mode
+ * and that reranking strategies are valid for the mode
+ */
+async function validateModeConfiguration(options) {
+    const mode = options.mode || 'text';
+    const model = options.embeddingModel;
+    // Define supported models for each mode
+    const textModels = [
+        'sentence-transformers/all-MiniLM-L6-v2',
+        'Xenova/all-mpnet-base-v2'
+    ];
+    const multimodalModels = [
+        'Xenova/clip-vit-base-patch32',
+        'Xenova/clip-vit-base-patch16'
+    ];
+    // Validate model compatibility with mode
+    if (model) {
+        if (mode === 'text' && !textModels.includes(model)) {
+            if (multimodalModels.includes(model)) {
+                throw new ConfigurationError(`Model '${model}' is a multimodal model but text mode was selected.\n` +
+                    `\n` +
+                    `To use this model, specify multimodal mode:\n` +
+                    `  raglite ingest <path> --mode multimodal --model ${model}\n` +
+                    `\n` +
+                    `Or choose a text model for text mode:\n` +
+                    `  ${textModels.map(m => `raglite ingest <path> --model ${m}`).join('\n  ')}\n`, EXIT_CODES.INVALID_ARGUMENTS);
+            }
+            else {
+                throw new ConfigurationError(`Model '${model}' is not supported for text mode.\n` +
+                    `\n` +
+                    `Supported models for text mode:\n` +
+                    `  ${textModels.join('\n  ')}\n` +
+                    `\n` +
+                    `Examples:\n` +
+                    `  raglite ingest <path> --model sentence-transformers/all-MiniLM-L6-v2\n` +
+                    `  raglite ingest <path> --model Xenova/all-mpnet-base-v2\n`, EXIT_CODES.INVALID_ARGUMENTS);
+            }
+        }
+        if (mode === 'multimodal' && !multimodalModels.includes(model)) {
+            if (textModels.includes(model)) {
+                throw new ConfigurationError(`Model '${model}' is a text-only model but multimodal mode was selected.\n` +
+                    `\n` +
+                    `To use this model, specify text mode:\n` +
+                    `  raglite ingest <path> --mode text --model ${model}\n` +
+                    `\n` +
+                    `Or choose a multimodal model for multimodal mode:\n` +
+                    `  ${multimodalModels.map(m => `raglite ingest <path> --mode multimodal --model ${m}`).join('\n  ')}\n`, EXIT_CODES.INVALID_ARGUMENTS);
+            }
+            else {
+                throw new ConfigurationError(`Model '${model}' is not supported for multimodal mode.\n` +
+                    `\n` +
+                    `Supported models for multimodal mode:\n` +
+                    `  ${multimodalModels.join('\n  ')}\n` +
+                    `\n` +
+                    `Example:\n` +
+                    `  raglite ingest <path> --mode multimodal --model Xenova/clip-vit-base-patch32\n`, EXIT_CODES.INVALID_ARGUMENTS);
+            }
+        }
+    }
+    // Log the final configuration
+    console.log('✅ Mode configuration validated successfully');
+    if (mode !== 'text') {
+        console.log(`   Mode: ${mode}`);
+    }
+    if (model) {
+        console.log(`   Model: ${model}`);
+    }
+}
+/**
+ * Run document ingestion from CLI
+ * @param path - File or directory path to ingest
+ * @param options - CLI options including model selection
+ */
+export async function runIngest(path, options = {}) {
+    try {
+        // Handle --rebuild-if-needed flag immediately to prevent dimension mismatch error
+        // Validate path exists
+        const resolvedPath = resolve(path);
+        if (!existsSync(resolvedPath)) {
+            console.error(`Error: Path does not exist: ${path}`);
+            console.error('');
+            console.error('Please check:');
+            console.error('- The path is spelled correctly');
+            console.error('- You have read permissions for the path');
+            console.error('- The file or directory exists');
+            console.error('');
+            console.error('Examples:');
+            console.error('  raglite ingest ./docs/           # Ingest directory');
+            console.error('  raglite ingest ./readme.md       # Ingest single file');
+            process.exit(EXIT_CODES.FILE_NOT_FOUND);
+        }
+        // Check if it's a file or directory
+        let stats;
+        try {
+            stats = statSync(resolvedPath);
+        }
+        catch (error) {
+            console.error(`Error: Cannot access path: ${path}`);
+            console.error('');
+            if (error instanceof Error && error.message.includes('EACCES')) {
+                console.error('Permission denied. Please check that you have read permissions for this path.');
+                process.exit(EXIT_CODES.PERMISSION_ERROR);
+            }
+            else {
+                console.error('The path exists but cannot be accessed. Please check permissions.');
+                process.exit(EXIT_CODES.FILE_NOT_FOUND);
+            }
+        }
+        const pathType = stats.isDirectory() ? 'directory' : 'file';
+        // Validate file type for single files
+        if (stats.isFile()) {
+            const mode = options.mode || 'text';
+            // Only formats with actual processing implementations
+            const textExtensions = ['.md', '.txt', '.mdx', '.pdf', '.docx'];
+            const imageExtensions = ['.jpg', '.jpeg', '.png', '.gif', '.webp', '.bmp'];
+            const validExtensions = mode === 'multimodal'
+                ? [...textExtensions, ...imageExtensions]
+                : textExtensions;
+            const hasValidExtension = validExtensions.some(ext => path.toLowerCase().endsWith(ext));
+            if (!hasValidExtension) {
+                console.error(`Error: Unsupported file type: ${path}`);
+                console.error('');
+                if (mode === 'multimodal') {
+                    console.error('Supported file types in multimodal mode:');
+                    console.error('  Text: .md, .txt, .mdx');
+                    console.error('  Documents: .pdf, .docx');
+                    console.error('  Images: .jpg, .jpeg, .png, .gif, .webp, .bmp');
+                }
+                else {
+                    console.error('Supported file types in text mode:');
+                    console.error('  Text: .md, .txt, .mdx');
+                    console.error('  Documents: .pdf, .docx');
+                    console.error('');
+                    console.error('For image files, use --mode multimodal:');
+                    console.error('  raglite ingest <path> --mode multimodal');
+                }
+                console.error('');
+                console.error('If you want to ingest multiple files, provide a directory path instead.');
+                process.exit(EXIT_CODES.INVALID_ARGUMENTS);
+            }
+        }
+        console.log(`Starting ingestion of ${pathType}: ${resolvedPath}`);
+        console.log('This may take a while for large document collections...');
+        console.log('');
+        // Prepare factory options
+        const factoryOptions = {};
+        if (options.model) {
+            factoryOptions.embeddingModel = options.model;
+            console.log(`Using embedding model: ${options.model}`);
+        }
+        if (options.mode) {
+            factoryOptions.mode = options.mode;
+            console.log(`Using processing mode: ${options.mode}`);
+        }
+        if (options.rebuildIfNeeded) {
+            factoryOptions.forceRebuild = true;
+            console.log('Force rebuild enabled due to rebuildIfNeeded option');
+            // Delete old index file immediately to prevent dimension mismatch errors
+            const indexPath = process.env.RAG_INDEX_FILE || './vector-index.bin';
+            const { existsSync, unlinkSync } = await import('fs');
+            if (existsSync(indexPath)) {
+                try {
+                    unlinkSync(indexPath);
+                    console.log('🗑️ Removed old index file to prevent dimension mismatch');
+                }
+                catch (error) {
+                    console.warn(`⚠️ Could not remove old index file: ${error}`);
+                }
+            }
+        }
+        // Validate mode-specific model and strategy combinations
+        await validateModeConfiguration(factoryOptions);
+        const dbPath = process.env.RAG_DB_FILE || './db.sqlite';
+        const indexPath = process.env.RAG_INDEX_FILE || './vector-index.bin';
+        // Setup graceful cleanup
+        setupCLICleanup(dbPath);
+        // Check if database is busy before starting
+        const busyStatus = await isDatabaseBusy(dbPath);
+        if (busyStatus.isBusy) {
+            console.log('⚠️  Database appears to be in use by another process');
+            console.log(`   Reason: ${busyStatus.reason}`);
+            console.log('   Attempting to proceed anyway...');
+            console.log('');
+        }
+        // Create ingestion pipeline using factory
+        let pipeline;
+        try {
+            // Create ingestion pipeline using IngestionFactory with database protection
+            pipeline = await withCLIDatabaseAccess(dbPath, () => IngestionFactory.create(dbPath, indexPath, factoryOptions), {
+                commandName: 'Ingestion command',
+                showProgress: true,
+                maxWaitMs: 15000 // Longer timeout for ingestion
+            });
+            const result = await pipeline.ingestPath(resolvedPath, { mode: factoryOptions.mode });
+            // Display final results
+            console.log('\n' + '='.repeat(50));
+            console.log('INGESTION SUMMARY');
+            console.log('='.repeat(50));
+            console.log(`Documents processed: ${result.documentsProcessed}`);
+            console.log(`Chunks created: ${result.chunksCreated}`);
+            console.log(`Embeddings generated: ${result.embeddingsGenerated}`);
+            if (result.documentErrors > 0) {
+                console.log(`Document errors: ${result.documentErrors}`);
+            }
+            if (result.embeddingErrors > 0) {
+                console.log(`Embedding errors: ${result.embeddingErrors}`);
+            }
+            console.log(`Total processing time: ${(result.processingTimeMs / 1000).toFixed(2)} seconds`);
+            if (result.chunksCreated > 0) {
+                const chunksPerSecond = (result.chunksCreated / (result.processingTimeMs / 1000)).toFixed(1);
+                console.log(`Processing rate: ${chunksPerSecond} chunks/second`);
+            }
+            console.log('\nIngestion completed successfully!');
+            // Display mode-specific information
+            const mode = options.mode || 'text';
+            if (mode === 'multimodal') {
+                console.log('✨ Multimodal mode enabled - you can now search across text and image content');
+            }
+            console.log('You can now search your documents using: raglite search "your query"');
+            console.log('');
+            console.log('💡 The search command will automatically detect and use the ingestion mode.');
+        }
+        finally {
+            if (pipeline) {
+                await pipeline.cleanup();
+            }
+            // Ensure clean exit for CLI commands
+            const { DatabaseConnectionManager } = await import('../core/database-connection-manager.js');
+            await DatabaseConnectionManager.closeAllConnections();
+            // Force exit for CLI commands to prevent hanging
+            setTimeout(() => {
+                process.exit(0);
+            }, 100);
+        }
+    }
+    catch (error) {
+        console.error('\n' + '='.repeat(50));
+        console.error('INGESTION FAILED');
+        console.error('='.repeat(50));
+        if (error instanceof ConfigurationError) {
+            console.error('Configuration Error:');
+            console.error(error.message);
+            process.exit(error.exitCode);
+        }
+        else if (error instanceof Error) {
+            console.error('Error:', error.message);
+            console.error('');
+            // Provide specific help for common error types
+            if (error.message.includes('ENOENT')) {
+                console.error('File/Directory Not Found:');
+                console.error('- Verify the path exists and is accessible');
+                console.error('- Check file and directory permissions');
+                console.error('- Ensure you have read access to the specified location');
+                process.exit(EXIT_CODES.FILE_NOT_FOUND);
+            }
+            else if (error.message.includes('EACCES') || error.message.includes('permission')) {
+                console.error('Permission Denied:');
+                console.error('- Check that you have read permissions for the files/directories');
+                console.error('- Ensure the database file is not locked by another process');
+                console.error('- Try running with appropriate permissions');
+                process.exit(EXIT_CODES.PERMISSION_ERROR);
+            }
+            else if (error.message.includes('SQLITE') || error.message.includes('database')) {
+                console.error('Database Error:');
+                console.error('- The database file may be corrupted or locked');
+                console.error('- Try running: raglite rebuild');
+                console.error('- Ensure no other RAG-lite processes are running');
+                console.error('- Check available disk space');
+                process.exit(EXIT_CODES.DATABASE_ERROR);
+            }
+            else if (error.message.includes('ONNX') || error.message.includes('model')) {
+                console.error('Model Loading Error:');
+                console.error('- The embedding model failed to load');
+                console.error('- This may happen on first run while downloading the model');
+                console.error('- Ensure you have internet connection for initial model download');
+                console.error('- Check available disk space in the models directory');
+                console.error('- Try running the command again');
+                process.exit(EXIT_CODES.MODEL_ERROR);
+            }
+            else if (error.message.includes('index') || error.message.includes('vector')) {
+                console.error('Vector Index Error:');
+                console.error('- The vector index may be corrupted');
+                console.error('- Try running: raglite rebuild');
+                console.error('- Check available disk space');
+                process.exit(EXIT_CODES.INDEX_ERROR);
+            }
+            else {
+                console.error('General Error:');
+                console.error('- An unexpected error occurred during ingestion');
+                console.error('- Check the error message above for more details');
+                console.error('- Try running the command again');
+                console.error('- If the problem persists, try: raglite rebuild');
+                process.exit(EXIT_CODES.GENERAL_ERROR);
+            }
+        }
+        else {
+            console.error('Unknown error:', String(error));
+            process.exit(EXIT_CODES.GENERAL_ERROR);
+        }
+    }
+}
+/**
+ * Run index rebuild from CLI
+ */
+export async function runRebuild() {
+    try {
+        console.log('Starting index rebuild...');
+        console.log('This will regenerate all embeddings and rebuild the vector index.');
+        console.log('This may take a while depending on your document collection size.');
+        console.log('');
+        console.log('Progress will be shown below...');
+        console.log('');
+        // Detect mode from existing database for rebuild
+        const dbPath = process.env.RAG_DB_FILE || './db.sqlite';
+        const indexPath = process.env.RAG_INDEX_FILE || './vector-index.bin';
+        let rebuildOptions = { forceRebuild: true };
+        if (existsSync(dbPath)) {
+            try {
+                // Import mode detection service
+                const { ModeDetectionService } = await import('../core/mode-detection-service.js');
+                const modeService = new ModeDetectionService(dbPath);
+                const systemInfo = await modeService.detectMode();
+                console.log(`🎯 Detected existing configuration:`);
+                console.log(`   Mode: ${systemInfo.mode}`);
+                console.log(`   Model: ${systemInfo.modelName}`);
+                console.log(`   Reranking: ${systemInfo.rerankingStrategy}`);
+                console.log('');
+                // Use the detected configuration for rebuild
+                rebuildOptions.mode = systemInfo.mode;
+                rebuildOptions.embeddingModel = systemInfo.modelName;
+                rebuildOptions.rerankingStrategy = systemInfo.rerankingStrategy;
+            }
+            catch (error) {
+                console.warn('⚠️  Could not detect existing mode configuration, using defaults');
+                console.warn(`   Error: ${error instanceof Error ? error.message : 'Unknown error'}`);
+            }
+        }
+        // Create ingestion pipeline with force rebuild using factory
+        const pipeline = await IngestionFactory.create(dbPath, indexPath, rebuildOptions);
+        try {
+            // Get all documents from database and re-ingest them
+            const { openDatabase } = await import('../core/db.js');
+            const db = await openDatabase(dbPath);
+            try {
+                const documents = await db.all('SELECT DISTINCT source FROM documents ORDER BY source');
+                if (documents.length === 0) {
+                    throw new Error('No documents found in database. Nothing to rebuild.');
+                }
+                console.log(`Found ${documents.length} document${documents.length === 1 ? '' : 's'} to rebuild`);
+                let totalResult = {
+                    documentsProcessed: 0,
+                    chunksCreated: 0,
+                    embeddingsGenerated: 0,
+                    documentErrors: 0,
+                    embeddingErrors: 0,
+                    processingTimeMs: 0
+                };
+                // Re-ingest each document
+                for (const doc of documents) {
+                    if (existsSync(doc.source)) {
+                        console.log(`Rebuilding: ${doc.source}`);
+                        const result = await pipeline.ingestFile(doc.source);
+                        totalResult.documentsProcessed += result.documentsProcessed;
+                        totalResult.chunksCreated += result.chunksCreated;
+                        totalResult.embeddingsGenerated += result.embeddingsGenerated;
+                        totalResult.documentErrors += result.documentErrors;
+                        totalResult.embeddingErrors += result.embeddingErrors;
+                        totalResult.processingTimeMs += result.processingTimeMs;
+                    }
+                    else {
+                        console.warn(`⚠️  Document not found, skipping: ${doc.source}`);
+                        totalResult.documentErrors++;
+                    }
+                }
+                console.log('\n' + '='.repeat(50));
+                console.log('REBUILD COMPLETE');
+                console.log('='.repeat(50));
+                console.log(`Documents processed: ${totalResult.documentsProcessed}`);
+                console.log(`Chunks created: ${totalResult.chunksCreated}`);
+                console.log(`Embeddings generated: ${totalResult.embeddingsGenerated}`);
+                if (totalResult.documentErrors > 0) {
+                    console.log(`Document errors: ${totalResult.documentErrors}`);
+                }
+                if (totalResult.embeddingErrors > 0) {
+                    console.log(`Embedding errors: ${totalResult.embeddingErrors}`);
+                }
+                console.log(`Total processing time: ${(totalResult.processingTimeMs / 1000).toFixed(2)} seconds`);
+                console.log('');
+                console.log('The vector index has been successfully rebuilt.');
+                console.log('All embeddings have been regenerated with the current model.');
+                console.log('');
+                console.log('You can now search your documents using: raglite search "your query"');
+            }
+            finally {
+                await db.close();
+            }
+        }
+        finally {
+            await pipeline.cleanup();
+            // Ensure clean exit for CLI commands
+            const { DatabaseConnectionManager } = await import('../core/database-connection-manager.js');
+            await DatabaseConnectionManager.closeAllConnections();
+            // Force exit for CLI commands to prevent hanging
+            setTimeout(() => {
+                process.exit(0);
+            }, 100);
+        }
+    }
+    catch (error) {
+        console.error('\n' + '='.repeat(50));
+        console.error('REBUILD FAILED');
+        console.error('='.repeat(50));
+        if (error instanceof ConfigurationError) {
+            console.error('Configuration Error:');
+            console.error(error.message);
+            process.exit(error.exitCode);
+        }
+        else if (error instanceof Error) {
+            console.error('Error:', error.message);
+            console.error('');
+            if (error.message.includes('No documents found') || error.message.includes('empty')) {
+                console.error('No Documents Found:');
+                console.error('- Make sure you have ingested documents first');
+                console.error('- Run: raglite ingest <path>');
+                console.error('- Check that your documents are in supported formats (.md, .txt)');
+                process.exit(EXIT_CODES.FILE_NOT_FOUND);
+            }
+            else if (error.message.includes('SQLITE') || error.message.includes('database')) {
+                console.error('Database Error:');
+                console.error('- The database file may be missing or corrupted');
+                console.error('- Try deleting db.sqlite and re-ingesting your documents');
+                console.error('- Ensure you have write permissions in the current directory');
+                process.exit(EXIT_CODES.DATABASE_ERROR);
+            }
+            else if (error.message.includes('model') || error.message.includes('ONNX')) {
+                console.error('Model Error:');
+                console.error('- The embedding model failed to load');
+                console.error('- Ensure you have internet connection for model download');
+                console.error('- Check available disk space');
+                console.error('- Try deleting the models directory and running again');
+                process.exit(EXIT_CODES.MODEL_ERROR);
+            }
+            else if (error.message.includes('index') || error.message.includes('vector')) {
+                console.error('Index Error:');
+                console.error('- Failed to rebuild the vector index');
+                console.error('- Try deleting vector-index.bin and running again');
+                console.error('- Check available disk space');
+                process.exit(EXIT_CODES.INDEX_ERROR);
+            }
+            else {
+                console.error('General Error:');
+                console.error('- An unexpected error occurred during rebuild');
+                console.error('- Try deleting db.sqlite and vector-index.bin, then re-ingest');
+                console.error('- Check available disk space and permissions');
+                process.exit(EXIT_CODES.GENERAL_ERROR);
+            }
+        }
+        else {
+            console.error('Unknown error:', String(error));
+            process.exit(EXIT_CODES.GENERAL_ERROR);
+        }
+    }
+}
+//# sourceMappingURL=indexer.js.map

package/dist/esm/cli/search.d.ts

@@ -0,0 +1,7 @@
+/**
+ * Run search from CLI
+ * @param query - Search query string or image file path
+ * @param options - CLI options including top-k and rerank settings
+ */
+export declare function runSearch(query: string, options?: Record<string, any>): Promise<void>;
+//# sourceMappingURL=search.d.ts.map