atlas-pipeline-mcp 1.0.21 → 1.0.22

package/dist/tools/rag-search.js

@@ -0,0 +1,338 @@
+/**
+ * Atlas Server - RAG-Powered Codebase Search & Understanding
+ *
+ * Semantic code search using vector embeddings and RAG:
+ * - Natural language queries to find code
+ * - Semantic similarity matching (not just keyword search)
+ * - Code understanding through embeddings
+ * - Cross-file dependency discovery
+ * - Architecture pattern detection
+ * - Similar code detection
+ * - Knowledge graph construction
+ * - Multi-modal code analysis (code + docs + comments)
+ *
+ * @module rag-search
+ * @author Nishant Unavane
+ * @version 1.0.0
+ */
+import { promises as fs } from 'fs';
+import { join, relative, extname } from 'path';
+import { getActiveProvider, isNoLLMMode } from '../providers/index.js';
+import { logger, createTimer } from '../utils.js';
+import { z } from 'zod';
+// ============================================================================
+// Validation Schema
+// ============================================================================
+const RAGSearchOptionsSchema = z.object({
+    projectPath: z.string().min(1),
+    query: z.string().min(1),
+    searchType: z.enum(['semantic', 'keyword', 'hybrid']).optional(),
+    maxResults: z.number().min(1).max(100).optional(),
+    minSimilarity: z.number().min(0).max(1).optional(),
+    fileTypes: z.array(z.string()).optional(),
+    excludePaths: z.array(z.string()).optional(),
+    includeTests: z.boolean().optional(),
+    includeDocs: z.boolean().optional(),
+    chunkSize: z.number().optional(),
+    overlapSize: z.number().optional(),
+    useCache: z.boolean().optional(),
+});
+// ============================================================================
+// Code Indexing
+// ============================================================================
+/**
+ * Find all source files in project
+ */
+async function findSourceFiles(projectPath, fileTypes, excludePaths) {
+    const files = [];
+    async function scan(dir) {
+        try {
+            const entries = await fs.readdir(dir, { withFileTypes: true });
+            for (const entry of entries) {
+                const fullPath = join(dir, entry.name);
+                const relativePath = relative(projectPath, fullPath);
+                // Check if excluded
+                if (excludePaths.some(exclude => relativePath.startsWith(exclude))) {
+                    continue;
+                }
+                if (entry.isDirectory()) {
+                    if (!['node_modules', '.git', 'dist', 'build', 'coverage'].includes(entry.name)) {
+                        await scan(fullPath);
+                    }
+                }
+                else if (entry.isFile()) {
+                    const ext = extname(entry.name);
+                    if (fileTypes.includes(ext)) {
+                        files.push(fullPath);
+                    }
+                }
+            }
+        }
+        catch (error) {
+            logger.debug({ error, dir }, 'Error scanning directory');
+        }
+    }
+    await scan(projectPath);
+    return files;
+}
+/**
+ * Extract code chunks from file
+ */
+async function extractCodeChunks(filePath, chunkSize, overlapSize) {
+    const content = await fs.readFile(filePath, 'utf-8');
+    const lines = content.split('\n');
+    const chunks = [];
+    for (let i = 0; i < lines.length; i += chunkSize - overlapSize) {
+        const endLine = Math.min(i + chunkSize, lines.length);
+        const chunkLines = lines.slice(i, endLine);
+        chunks.push({
+            content: chunkLines.join('\n'),
+            lineStart: i + 1,
+            lineEnd: endLine,
+        });
+        if (endLine >= lines.length)
+            break;
+    }
+    return chunks;
+}
+/**
+ * Simple keyword-based search (fallback when no LLM)
+ */
+function keywordSearch(query, files) {
+    const matches = [];
+    const queryWords = query.toLowerCase().split(/\s+/).filter(Boolean);
+    for (const file of files) {
+        const lines = file.content.split('\n');
+        const contentLower = file.content.toLowerCase();
+        // Calculate relevance score
+        let score = 0;
+        for (const word of queryWords) {
+            const count = (contentLower.match(new RegExp(word, 'g')) || []).length;
+            score += count;
+        }
+        if (score > 0) {
+            // Find best matching snippet
+            for (let i = 0; i < lines.length; i++) {
+                const currentLine = lines[i];
+                if (!currentLine)
+                    continue;
+                const lineLower = currentLine.toLowerCase();
+                const lineScore = queryWords.filter(word => lineLower.includes(word)).length;
+                if (lineScore > 0) {
+                    const contextStart = Math.max(0, i - 2);
+                    const contextEnd = Math.min(lines.length, i + 3);
+                    matches.push({
+                        filePath: file.path,
+                        snippet: currentLine,
+                        context: lines.slice(contextStart, contextEnd).join('\n'),
+                        similarity: lineScore / queryWords.length,
+                        explanation: `Contains keywords: ${queryWords.join(', ')}`,
+                        lineStart: i + 1,
+                        lineEnd: i + 1,
+                        matchType: 'partial',
+                        metadata: {
+                            language: extname(file.path).substring(1),
+                            fileSize: file.content.length,
+                            lastModified: new Date(),
+                            category: detectCategory(currentLine),
+                        },
+                    });
+                }
+            }
+        }
+    }
+    return matches.sort((a, b) => b.similarity - a.similarity);
+}
+/**
+ * Detect code category from line
+ */
+function detectCategory(line) {
+    const trimmed = line.trim();
+    if (trimmed.startsWith('function ') || trimmed.includes('=> '))
+        return 'function';
+    if (trimmed.startsWith('class '))
+        return 'class';
+    if (trimmed.startsWith('interface '))
+        return 'interface';
+    if (trimmed.startsWith('type '))
+        return 'type';
+    if (trimmed.startsWith('import ') || trimmed.startsWith('from '))
+        return 'import';
+    if (trimmed.startsWith('//') || trimmed.startsWith('/*') || trimmed.startsWith('*'))
+        return 'comment';
+    return 'variable';
+}
+/**
+ * Build simple knowledge graph
+ */
+function buildKnowledgeGraph(files) {
+    const nodes = [];
+    const edges = [];
+    // Extract imports and exports
+    for (const file of files) {
+        const lines = file.content.split('\n');
+        for (let i = 0; i < lines.length; i++) {
+            const currentLine = lines[i];
+            if (!currentLine)
+                continue;
+            const line = currentLine.trim();
+            // Detect imports
+            const importMatch = line.match(/import\s+.*\s+from\s+['"](.+)['"]/);
+            if (importMatch && importMatch[1]) {
+                edges.push({
+                    from: file.path,
+                    to: importMatch[1],
+                    type: 'imports',
+                    weight: 1,
+                });
+            }
+            // Detect function definitions
+            const funcMatch = line.match(/(?:function|const|let|var)\s+(\w+)/);
+            if (funcMatch && funcMatch[1]) {
+                nodes.push({
+                    id: `${file.path}:${funcMatch[1]}`,
+                    label: funcMatch[1],
+                    type: 'function',
+                    filePath: file.path,
+                    importance: 0.5,
+                });
+            }
+            // Detect class definitions
+            const classMatch = line.match(/class\s+(\w+)/);
+            if (classMatch && classMatch[1]) {
+                nodes.push({
+                    id: `${file.path}:${classMatch[1]}`,
+                    label: classMatch[1],
+                    type: 'class',
+                    filePath: file.path,
+                    importance: 0.8,
+                });
+            }
+        }
+    }
+    return {
+        nodes,
+        edges,
+        clusters: [],
+    };
+}
+// ============================================================================
+// Main RAG Search Function
+// ============================================================================
+/**
+ * RAG-powered semantic code search
+ */
+export async function searchCodebase(options) {
+    const timer = createTimer();
+    const { projectPath, query, searchType = 'hybrid', maxResults = 10, minSimilarity = 0.3, fileTypes = ['.ts', '.tsx', '.js', '.jsx', '.py', '.java', '.go', '.rs'], excludePaths = ['node_modules', 'dist', 'build', '.git'], includeTests = false, includeDocs = true, chunkSize = 50, overlapSize = 5, useCache = true, } = RAGSearchOptionsSchema.parse(options);
+    logger.info({ query, projectPath, searchType }, 'Starting RAG code search');
+    // Find all source files
+    const files = await findSourceFiles(projectPath, fileTypes, excludePaths);
+    logger.info({ fileCount: files.length }, 'Source files indexed');
+    // Load file contents
+    const fileContents = await Promise.all(files.slice(0, 100).map(async (path) => ({
+        path,
+        content: await fs.readFile(path, 'utf-8').catch(() => ''),
+    })));
+    let results = [];
+    let semanticSearch = false;
+    // Try semantic search with LLM
+    if (!isNoLLMMode() && (searchType === 'semantic' || searchType === 'hybrid')) {
+        try {
+            const provider = await getActiveProvider();
+            semanticSearch = true;
+            // Use LLM to understand the query and find relevant code
+            const prompt = `You are a code search expert. Find code snippets that match this query: "${query}"
+
+Available files (showing first 5):
+${fileContents.slice(0, 5).map(f => `\n--- ${f.path} ---\n${f.content.substring(0, 500)}...`).join('\n')}
+
+Identify the TOP 5 most relevant code snippets. For each match, provide:
+1. File path
+2. Line numbers (estimate)
+3. Similarity score (0-1)
+4. Brief explanation of why it matches
+
+Format as JSON array:
+[
+  {
+    "filePath": "path/to/file.ts",
+    "lineStart": 10,
+    "lineEnd": 15,
+    "similarity": 0.95,
+    "explanation": "This function handles user authentication"
+  }
+]`;
+            const response = await provider.complete(prompt, {
+                temperature: 0.2,
+                maxTokens: 1000,
+            });
+            // Parse LLM response
+            const jsonMatch = response.text.match(/\[[\s\S]*\]/);
+            if (jsonMatch) {
+                const llmResults = JSON.parse(jsonMatch[0]);
+                for (const result of llmResults) {
+                    const fileContent = fileContents.find(f => f.path.includes(result.filePath));
+                    if (fileContent) {
+                        const lines = fileContent.content.split('\n');
+                        const snippet = lines.slice(result.lineStart - 1, result.lineEnd).join('\n');
+                        const contextStart = Math.max(0, result.lineStart - 3);
+                        const contextEnd = Math.min(lines.length, result.lineEnd + 3);
+                        results.push({
+                            filePath: fileContent.path,
+                            snippet,
+                            context: lines.slice(contextStart, contextEnd).join('\n'),
+                            similarity: result.similarity || 0.7,
+                            explanation: result.explanation || '',
+                            lineStart: result.lineStart || 1,
+                            lineEnd: result.lineEnd || 1,
+                            matchType: 'semantic',
+                            metadata: {
+                                language: extname(fileContent.path).substring(1),
+                                fileSize: fileContent.content.length,
+                                lastModified: new Date(),
+                                category: 'function',
+                            },
+                        });
+                    }
+                }
+            }
+        }
+        catch (error) {
+            logger.warn({ error }, 'Semantic search failed, falling back to keyword search');
+        }
+    }
+    // Fallback to keyword search
+    if (results.length === 0 || searchType === 'keyword' || searchType === 'hybrid') {
+        const keywordResults = keywordSearch(query, fileContents);
+        results = results.concat(keywordResults);
+    }
+    // Remove duplicates and apply filters
+    const uniqueResults = Array.from(new Map(results.map(r => [`${r.filePath}:${r.lineStart}`, r])).values());
+    const filteredResults = uniqueResults
+        .filter(r => r.similarity >= minSimilarity)
+        .sort((a, b) => b.similarity - a.similarity)
+        .slice(0, maxResults);
+    // Build knowledge graph
+    const knowledgeGraph = buildKnowledgeGraph(fileContents);
+    const searchTimeMs = timer.elapsed();
+    logger.info({
+        matches: filteredResults.length,
+        semanticSearch,
+        searchTimeMs
+    }, 'RAG search completed');
+    return {
+        query,
+        results: filteredResults,
+        totalMatches: uniqueResults.length,
+        searchTimeMs,
+        indexedFiles: files.length,
+        semanticSearch,
+        knowledgeGraph,
+    };
+}
+// ============================================================================
+// Export
+// ============================================================================
+export default searchCodebase;
+//# sourceMappingURL=rag-search.js.map

package/dist/tools/rag-search.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"rag-search.js","sourceRoot":"","sources":["../../src/tools/rag-search.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;GAgBG;AAEH,OAAO,EAAE,QAAQ,IAAI,EAAE,EAAE,MAAM,IAAI,CAAC;AACpC,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,MAAM,CAAC;AAC/C,OAAO,EAAE,iBAAiB,EAAE,WAAW,EAAE,MAAM,uBAAuB,CAAC;AACvE,OAAO,EAAE,MAAM,EAAE,WAAW,EAAc,MAAM,aAAa,CAAC;AAC9D,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AA6FxB,+EAA+E;AAC/E,oBAAoB;AACpB,+EAA+E;AAE/E,MAAM,sBAAsB,GAAG,CAAC,CAAC,MAAM,CAAC;IACtC,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC;IAC9B,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC;IACxB,UAAU,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,UAAU,EAAE,SAAS,EAAE,QAAQ,CAAC,CAAC,CAAC,QAAQ,EAAE;IAChE,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE;IACjD,aAAa,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE;IAClD,SAAS,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,QAAQ,EAAE;IACzC,YAAY,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,QAAQ,EAAE;IAC5C,YAAY,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,EAAE;IACpC,WAAW,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,EAAE;IACnC,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IAChC,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IAClC,QAAQ,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,EAAE;CACjC,CAAC,CAAC;AAEH,+EAA+E;AAC/E,gBAAgB;AAChB,+EAA+E;AAE/E;;GAEG;AACH,KAAK,UAAU,eAAe,CAC5B,WAAmB,EACnB,SAAmB,EACnB,YAAsB;IAEtB,MAAM,KAAK,GAAa,EAAE,CAAC;IAE3B,KAAK,UAAU,IAAI,CAAC,GAAW;QAC7B,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,MAAM,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC,CAAC;YAE/D,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;gBAC5B,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;gBACvC,MAAM,YAAY,GAAG,QAAQ,CAAC,WAAW,EAAE,QAAQ,CAAC,CAAC;gBAErD,oBAAoB;gBACpB,IAAI,YAAY,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,YAAY,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,EAAE,CAAC;oBACnE,SAAS;gBACX,CAAC;gBAED,IAAI,KAAK,CAAC,WAAW,EAAE,EAAE,CAAC;oBACxB,IAAI,CAAC,CAAC,cAAc,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,UAAU,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC;wBAChF,MAAM,IAAI,CAAC,QAAQ,CAAC,CAAC;oBACvB,CAAC;gBACH,CAAC;qBAAM,IAAI,KAAK,CAAC,MAAM,EAAE,EAAE,CAAC;oBAC1B,MAAM,GAAG,GAAG,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;oBAChC,IAAI,SAAS,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;wBAC5B,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;oBACvB,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,GAAG,EAAE,EAAE,0BAA0B,CAAC,CAAC;QAC3D,CAAC;IACH,CAAC;IAED,MAAM,IAAI,CAAC,WAAW,CAAC,CAAC;IACxB,OAAO,KAAK,CAAC;AACf,CAAC;AAED;;GAEG;AACH,KAAK,UAAU,iBAAiB,CAC9B,QAAgB,EAChB,SAAiB,EACjB,WAAmB;IAEnB,MAAM,OAAO,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;IACrD,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IAClC,MAAM,MAAM,GAAmE,EAAE,CAAC;IAElF,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,IAAI,SAAS,GAAG,WAAW,EAAE,CAAC;QAC/D,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,SAAS,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC;QACtD,MAAM,UAAU,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC;QAE3C,MAAM,CAAC,IAAI,CAAC;YACV,OAAO,EAAE,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC;YAC9B,SAAS,EAAE,CAAC,GAAG,CAAC;YAChB,OAAO,EAAE,OAAO;SACjB,CAAC,CAAC;QAEH,IAAI,OAAO,IAAI,KAAK,CAAC,MAAM;YAAE,MAAM;IACrC,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC;AAED;;GAEG;AACH,SAAS,aAAa,CACpB,KAAa,EACb,KAA+C;IAE/C,MAAM,OAAO,GAAgB,EAAE,CAAC;IAChC,MAAM,UAAU,GAAG,KAAK,CAAC,WAAW,EAAE,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IAEpE,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QACvC,MAAM,YAAY,GAAG,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC;QAEhD,4BAA4B;QAC5B,IAAI,KAAK,GAAG,CAAC,CAAC;QACd,KAAK,MAAM,IAAI,IAAI,UAAU,EAAE,CAAC;YAC9B,MAAM,KAAK,GAAG,CAAC,YAAY,CAAC,KAAK,CAAC,IAAI,MAAM,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC;YACvE,KAAK,IAAI,KAAK,CAAC;QACjB,CAAC;QAED,IAAI,KAAK,GAAG,CAAC,EAAE,CAAC;YACd,6BAA6B;YAC7B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;gBACtC,MAAM,WAAW,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;gBAC7B,IAAI,CAAC,WAAW;oBAAE,SAAS;gBAE3B,MAAM,SAAS,GAAG,WAAW,CAAC,WAAW,EAAE,CAAC;gBAC5C,MAAM,SAAS,GAAG,UAAU,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,SAAS,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,MAAM,CAAC;gBAE7E,IAAI,SAAS,GAAG,CAAC,EAAE,CAAC;oBAClB,MAAM,YAAY,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC;oBACxC,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC;oBAEjD,OAAO,CAAC,IAAI,CAAC;wBACX,QAAQ,EAAE,IAAI,CAAC,IAAI;wBACnB,OAAO,EAAE,WAAW;wBACpB,OAAO,EAAE,KAAK,CAAC,KAAK,CAAC,YAAY,EAAE,UAAU,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC;wBACzD,UAAU,EAAE,SAAS,GAAG,UAAU,CAAC,MAAM;wBACzC,WAAW,EAAE,sBAAsB,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE;wBAC1D,SAAS,EAAE,CAAC,GAAG,CAAC;wBAChB,OAAO,EAAE,CAAC,GAAG,CAAC;wBACd,SAAS,EAAE,SAAkB;wBAC7B,QAAQ,EAAE;4BACR,QAAQ,EAAE,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC;4BACzC,QAAQ,EAAE,IAAI,CAAC,OAAO,CAAC,MAAM;4BAC7B,YAAY,EAAE,IAAI,IAAI,EAAE;4BACxB,QAAQ,EAAE,cAAc,CAAC,WAAW,CAAC;yBACtC;qBACF,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,GAAG,CAAC,CAAC,UAAU,CAAC,CAAC;AAC7D,CAAC;AAED;;GAEG;AACH,SAAS,cAAc,CAAC,IAAY;IAClC,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC;IAE5B,IAAI,OAAO,CAAC,UAAU,CAAC,WAAW,CAAC,IAAI,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC;QAAE,OAAO,UAAU,CAAC;IAClF,IAAI,OAAO,CAAC,UAAU,CAAC,QAAQ,CAAC;QAAE,OAAO,OAAO,CAAC;IACjD,IAAI,OAAO,CAAC,UAAU,CAAC,YAAY,CAAC;QAAE,OAAO,WAAW,CAAC;IACzD,IAAI,OAAO,CAAC,UAAU,CAAC,OAAO,CAAC;QAAE,OAAO,MAAM,CAAC;IAC/C,IAAI,OAAO,CAAC,UAAU,CAAC,SAAS,CAAC,IAAI,OAAO,CAAC,UAAU,CAAC,OAAO,CAAC;QAAE,OAAO,QAAQ,CAAC;IAClF,IAAI,OAAO,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,OAAO,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,OAAO,CAAC,UAAU,CAAC,GAAG,CAAC;QAAE,OAAO,SAAS,CAAC;IAEtG,OAAO,UAAU,CAAC;AACpB,CAAC;AAED;;GAEG;AACH,SAAS,mBAAmB,CAAC,KAA+C;IAC1E,MAAM,KAAK,GAAgB,EAAE,CAAC;IAC9B,MAAM,KAAK,GAAgB,EAAE,CAAC;IAE9B,8BAA8B;IAC9B,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAEvC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YACtC,MAAM,WAAW,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;YAC7B,IAAI,CAAC,WAAW;gBAAE,SAAS;YAE3B,MAAM,IAAI,GAAG,WAAW,CAAC,IAAI,EAAE,CAAC;YAEhC,iBAAiB;YACjB,MAAM,WAAW,GAAG,IAAI,CAAC,KAAK,CAAC,mCAAmC,CAAC,CAAC;YACpE,IAAI,WAAW,IAAI,WAAW,CAAC,CAAC,CAAC,EAAE,CAAC;gBAClC,KAAK,CAAC,IAAI,CAAC;oBACT,IAAI,EAAE,IAAI,CAAC,IAAI;oBACf,EAAE,EAAE,WAAW,CAAC,CAAC,CAAC;oBAClB,IAAI,EAAE,SAAS;oBACf,MAAM,EAAE,CAAC;iBACV,CAAC,CAAC;YACL,CAAC;YAED,8BAA8B;YAC9B,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,oCAAoC,CAAC,CAAC;YACnE,IAAI,SAAS,IAAI,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC;gBAC9B,KAAK,CAAC,IAAI,CAAC;oBACT,EAAE,EAAE,GAAG,IAAI,CAAC,IAAI,IAAI,SAAS,CAAC,CAAC,CAAC,EAAE;oBAClC,KAAK,EAAE,SAAS,CAAC,CAAC,CAAC;oBACnB,IAAI,EAAE,UAAU;oBAChB,QAAQ,EAAE,IAAI,CAAC,IAAI;oBACnB,UAAU,EAAE,GAAG;iBAChB,CAAC,CAAC;YACL,CAAC;YAED,2BAA2B;YAC3B,MAAM,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,eAAe,CAAC,CAAC;YAC/C,IAAI,UAAU,IAAI,UAAU,CAAC,CAAC,CAAC,EAAE,CAAC;gBAChC,KAAK,CAAC,IAAI,CAAC;oBACT,EAAE,EAAE,GAAG,IAAI,CAAC,IAAI,IAAI,UAAU,CAAC,CAAC,CAAC,EAAE;oBACnC,KAAK,EAAE,UAAU,CAAC,CAAC,CAAC;oBACpB,IAAI,EAAE,OAAO;oBACb,QAAQ,EAAE,IAAI,CAAC,IAAI;oBACnB,UAAU,EAAE,GAAG;iBAChB,CAAC,CAAC;YACL,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO;QACL,KAAK;QACL,KAAK;QACL,QAAQ,EAAE,EAAE;KACb,CAAC;AACJ,CAAC;AAED,+EAA+E;AAC/E,2BAA2B;AAC3B,+EAA+E;AAE/E;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,cAAc,CAAC,OAAyB;IAC5D,MAAM,KAAK,GAAG,WAAW,EAAE,CAAC;IAE5B,MAAM,EACJ,WAAW,EACX,KAAK,EACL,UAAU,GAAG,QAAQ,EACrB,UAAU,GAAG,EAAE,EACf,aAAa,GAAG,GAAG,EACnB,SAAS,GAAG,CAAC,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,CAAC,EACxE,YAAY,GAAG,CAAC,cAAc,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,CAAC,EACxD,YAAY,GAAG,KAAK,EACpB,WAAW,GAAG,IAAI,EAClB,SAAS,GAAG,EAAE,EACd,WAAW,GAAG,CAAC,EACf,QAAQ,GAAG,IAAI,GAChB,GAAG,sBAAsB,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;IAE1C,MAAM,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,WAAW,EAAE,UAAU,EAAE,EAAE,0BAA0B,CAAC,CAAC;IAE5E,wBAAwB;IACxB,MAAM,KAAK,GAAG,MAAM,eAAe,CAAC,WAAW,EAAE,SAAS,EAAE,YAAY,CAAC,CAAC;IAC1E,MAAM,CAAC,IAAI,CAAC,EAAE,SAAS,EAAE,KAAK,CAAC,MAAM,EAAE,EAAE,sBAAsB,CAAC,CAAC;IAEjE,qBAAqB;IACrB,MAAM,YAAY,GAAG,MAAM,OAAO,CAAC,GAAG,CACpC,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC,GAAG,CAAC,KAAK,EAAE,IAAI,EAAE,EAAE,CAAC,CAAC;QACvC,IAAI;QACJ,OAAO,EAAE,MAAM,EAAE,CAAC,QAAQ,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC;KAC1D,CAAC,CAAC,CACJ,CAAC;IAEF,IAAI,OAAO,GAAgB,EAAE,CAAC;IAC9B,IAAI,cAAc,GAAG,KAAK,CAAC;IAE3B,+BAA+B;IAC/B,IAAI,CAAC,WAAW,EAAE,IAAI,CAAC,UAAU,KAAK,UAAU,IAAI,UAAU,KAAK,QAAQ,CAAC,EAAE,CAAC;QAC7E,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,iBAAiB,EAAE,CAAC;YAC3C,cAAc,GAAG,IAAI,CAAC;YAEtB,yDAAyD;YACzD,MAAM,MAAM,GAAG,4EAA4E,KAAK;;;EAGpG,YAAY,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,SAAS,CAAC,CAAC,IAAI,SAAS,CAAC,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,EAAE,GAAG,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC;;;;;;;;;;;;;;;;;EAiBtG,CAAC;YAEG,MAAM,QAAQ,GAAG,MAAM,QAAQ,CAAC,QAAQ,CAAC,MAAM,EAAE;gBAC/C,WAAW,EAAE,GAAG;gBAChB,SAAS,EAAE,IAAI;aAChB,CAAC,CAAC;YAEH,qBAAqB;YACrB,MAAM,SAAS,GAAG,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC;YACrD,IAAI,SAAS,EAAE,CAAC;gBACd,MAAM,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC;gBAE5C,KAAK,MAAM,MAAM,IAAI,UAAU,EAAE,CAAC;oBAChC,MAAM,WAAW,GAAG,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC;oBAC7E,IAAI,WAAW,EAAE,CAAC;wBAChB,MAAM,KAAK,GAAG,WAAW,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;wBAC9C,MAAM,OAAO,GAAG,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC,SAAS,GAAG,CAAC,EAAE,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;wBAC7E,MAAM,YAAY,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,MAAM,CAAC,SAAS,GAAG,CAAC,CAAC,CAAC;wBACvD,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,MAAM,EAAE,MAAM,CAAC,OAAO,GAAG,CAAC,CAAC,CAAC;wBAE9D,OAAO,CAAC,IAAI,CAAC;4BACX,QAAQ,EAAE,WAAW,CAAC,IAAI;4BAC1B,OAAO;4BACP,OAAO,EAAE,KAAK,CAAC,KAAK,CAAC,YAAY,EAAE,UAAU,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC;4BACzD,UAAU,EAAE,MAAM,CAAC,UAAU,IAAI,GAAG;4BACpC,WAAW,EAAE,MAAM,CAAC,WAAW,IAAI,EAAE;4BACrC,SAAS,EAAE,MAAM,CAAC,SAAS,IAAI,CAAC;4BAChC,OAAO,EAAE,MAAM,CAAC,OAAO,IAAI,CAAC;4BAC5B,SAAS,EAAE,UAAU;4BACrB,QAAQ,EAAE;gCACR,QAAQ,EAAE,OAAO,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC;gCAChD,QAAQ,EAAE,WAAW,CAAC,OAAO,CAAC,MAAM;gCACpC,YAAY,EAAE,IAAI,IAAI,EAAE;gCACxB,QAAQ,EAAE,UAAU;6BACrB;yBACF,CAAC,CAAC;oBACL,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,EAAE,wDAAwD,CAAC,CAAC;QACnF,CAAC;IACH,CAAC;IAED,6BAA6B;IAC7B,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,IAAI,UAAU,KAAK,SAAS,IAAI,UAAU,KAAK,QAAQ,EAAE,CAAC;QAChF,MAAM,cAAc,GAAG,aAAa,CAAC,KAAK,EAAE,YAAY,CAAC,CAAC;QAC1D,OAAO,GAAG,OAAO,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC;IAC3C,CAAC;IAED,sCAAsC;IACtC,MAAM,aAAa,GAAG,KAAK,CAAC,IAAI,CAC9B,IAAI,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,QAAQ,IAAI,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,MAAM,EAAE,CACxE,CAAC;IAEF,MAAM,eAAe,GAAG,aAAa;SAClC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,UAAU,IAAI,aAAa,CAAC;SAC1C,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,GAAG,CAAC,CAAC,UAAU,CAAC;SAC3C,KAAK,CAAC,CAAC,EAAE,UAAU,CAAC,CAAC;IAExB,wBAAwB;IACxB,MAAM,cAAc,GAAG,mBAAmB,CAAC,YAAY,CAAC,CAAC;IAEzD,MAAM,YAAY,GAAG,KAAK,CAAC,OAAO,EAAE,CAAC;IACrC,MAAM,CAAC,IAAI,CAAC;QACV,OAAO,EAAE,eAAe,CAAC,MAAM;QAC/B,cAAc;QACd,YAAY;KACb,EAAE,sBAAsB,CAAC,CAAC;IAE3B,OAAO;QACL,KAAK;QACL,OAAO,EAAE,eAAe;QACxB,YAAY,EAAE,aAAa,CAAC,MAAM;QAClC,YAAY;QACZ,YAAY,EAAE,KAAK,CAAC,MAAM;QAC1B,cAAc;QACd,cAAc;KACf,CAAC;AACJ,CAAC;AAED,+EAA+E;AAC/E,SAAS;AACT,+EAA+E;AAE/E,eAAe,cAAc,CAAC"}

package/package.json

@@ -1,8 +1,8 @@
 {
   "name": "atlas-pipeline-mcp",
-  "version": "1.0.21",
+  "version": "1.0.22",
   "mcpName": "io.github.IamNishant51/atlas-pipeline",
-  "description": "Model Context Protocol server with multi-stage AI pipeline - works with any LLM provider (Ollama, OpenAI, Anthropic). Compatible with Cursor, GitHub Copilot, Windsurf, and more. Features 15 tools including security scanning, test generation, code explanation, debugging assistance, and advanced sequential thinking.",
+  "description": "Advanced AI/ML-powered MCP server with RAG search, ML bug prediction, intelligent code migration, and advanced test generation. Works with any LLM (Ollama, OpenAI, Anthropic). 13 professional tools + 4 cutting-edge AI/ML tools. Compatible with Cursor, Windsurf, Claude Desktop.",
   "main": "dist/server.js",
   "bin": {
     "atlas-pipeline-mcp": "./dist/mcp.js",