Smart Coding MCP

import path from "path"; import { cosineSimilarity } from "../lib/utils.js"; export class HybridSearch { constructor(embedder, cache, config) { this.embedder = embedder; this.cache = cache; this.config = config; } async search(query, maxResults) { const vectorStore = this.cache.getVectorStore(); if (vectorStore.length === 0) { return { results: [], message: "No code has been indexed yet. Please wait for initial indexing to complete." }; } // Generate query embedding const queryEmbed = await this.embedder(query, { pooling: "mean", normalize: true }); const queryVector = Array.from(queryEmbed.data); // Score all chunks const scoredChunks = vectorStore.map(chunk => { // Semantic similarity let score = cosineSimilarity(queryVector, chunk.vector) * this.config.semanticWeight; // Exact match boost const lowerQuery = query.toLowerCase(); const lowerContent = chunk.content.toLowerCase(); if (lowerContent.includes(lowerQuery)) { score += this.config.exactMatchBoost; } else { // Partial word matching const queryWords = lowerQuery.split(/\s+/); const matchedWords = queryWords.filter(word => word.length > 2 && lowerContent.includes(word) ).length; score += (matchedWords / queryWords.length) * 0.3; } return { ...chunk, score }; }); // Get top results const results = scoredChunks .sort((a, b) => b.score - a.score) .slice(0, maxResults); return { results, message: null }; } formatResults(results) { if (results.length === 0) { return "No matching code found for your query."; } return results.map((r, idx) => { const relPath = path.relative(this.config.searchDirectory, r.file); return `## Result ${idx + 1} (Relevance: ${(r.score * 100).toFixed(1)}%)\n` + `**File:** \`${relPath}\`\n` + `**Lines:** ${r.startLine}-${r.endLine}\n\n` + "```" + path.extname(r.file).slice(1) + "\n" + r.content + "\n" + "```\n"; }).join("\n"); } } // MCP Tool definition for this feature export function getToolDefinition(config) { return { name: "a_semantic_search", description: "Performs intelligent hybrid code search combining semantic understanding with exact text matching. Ideal for finding code by meaning (e.g., 'authentication logic', 'database queries') even with typos or variations. Returns the most relevant code snippets with file locations and line numbers.", inputSchema: { type: "object", properties: { query: { type: "string", description: "Search query - can be natural language (e.g., 'where do we handle user login') or specific terms" }, maxResults: { type: "number", description: "Maximum number of results to return (default: from config)", default: config.maxResults } }, required: ["query"] }, annotations: { title: "Semantic Code Search", readOnlyHint: true, destructiveHint: false, idempotentHint: true, openWorldHint: false } }; } // Tool handler export async function handleToolCall(request, hybridSearch) { const query = request.params.arguments.query; const maxResults = request.params.arguments.maxResults || hybridSearch.config.maxResults; const { results, message } = await hybridSearch.search(query, maxResults); if (message) { return { content: [{ type: "text", text: message }] }; } const formattedText = hybridSearch.formatResults(results); return { content: [{ type: "text", text: formattedText }] }; }