codesift-mcp 0.1.0

package/dist/retrieval/codebase-retrieval.js

@@ -0,0 +1,472 @@
+import { loadConfig } from "../config.js";
+import { isTestFile } from "../utils/test-file.js";
+/**
+ * Estimate token count from a string. ~4 chars per token.
+ */
+function estimateTokens(text) {
+    return Math.ceil(text.length / 4);
+}
+/**
+ * Execute a single sub-query and return the result with token estimate.
+ */
+async function executeSubQuery(repo, query) {
+    const qType = query.type;
+    switch (qType) {
+        case "symbols": {
+            const { searchSymbols } = await import("../tools/search-tools.js");
+            const results = await searchSymbols(repo, query["query"], {
+                kind: query["kind"],
+                file_pattern: query["file_pattern"],
+                include_source: true,
+                top_k: query["top_k"] ?? 5,
+            });
+            const sourceLimit = query["source_chars"] ?? 200;
+            const data = results.map((r) => {
+                const sym = r.symbol;
+                return sourceLimit > 0 && sym.source && sym.source.length > sourceLimit
+                    ? { ...sym, source: sym.source.slice(0, sourceLimit) }
+                    : sym;
+            });
+            const text = JSON.stringify(data);
+            return { type: qType, data, tokens: estimateTokens(text) };
+        }
+        case "text": {
+            const { searchText } = await import("../tools/search-tools.js");
+            const results = await searchText(repo, query["query"], {
+                regex: query["regex"],
+                context_lines: query["context_lines"],
+                file_pattern: query["file_pattern"],
+            });
+            const text = JSON.stringify(results);
+            return { type: qType, data: results, tokens: estimateTokens(text) };
+        }
+        case "file_tree": {
+            const { getFileTree } = await import("../tools/outline-tools.js");
+            const result = await getFileTree(repo, {
+                path_prefix: (query["path"] ?? query["path_prefix"]),
+                name_pattern: query["name_pattern"],
+                depth: query["depth"],
+                compact: query["compact"],
+                min_symbols: query["min_symbols"],
+            });
+            const text = JSON.stringify(result);
+            return { type: qType, data: result, tokens: estimateTokens(text) };
+        }
+        case "outline": {
+            const { getFileOutline } = await import("../tools/outline-tools.js");
+            const result = await getFileOutline(repo, query["file_path"]);
+            const text = JSON.stringify(result);
+            return { type: qType, data: result, tokens: estimateTokens(text) };
+        }
+        case "references": {
+            const { findReferences } = await import("../tools/symbol-tools.js");
+            const results = await findReferences(repo, query["symbol_name"]);
+            const text = JSON.stringify(results);
+            return { type: qType, data: results, tokens: estimateTokens(text) };
+        }
+        case "call_chain": {
+            const { traceCallChain } = await import("../tools/graph-tools.js");
+            const result = await traceCallChain(repo, query["symbol_name"], query["direction"] ?? "callers", {
+                depth: query["depth"],
+                include_source: query["include_source"] ?? false,
+            });
+            const text = JSON.stringify(result);
+            return { type: qType, data: result, tokens: estimateTokens(text) };
+        }
+        case "impact": {
+            const { impactAnalysis } = await import("../tools/graph-tools.js");
+            const result = await impactAnalysis(repo, query["since"], {
+                depth: query["depth"],
+                until: query["until"],
+                include_source: query["include_source"] ?? false,
+            });
+            const text = JSON.stringify(result);
+            return { type: qType, data: result, tokens: estimateTokens(text) };
+        }
+        case "context": {
+            const { assembleContext } = await import("../tools/context-tools.js");
+            const result = await assembleContext(repo, query["query"], query["max_tokens"]);
+            const text = JSON.stringify(result);
+            return { type: qType, data: result, tokens: estimateTokens(text) };
+        }
+        case "knowledge_map": {
+            const { getKnowledgeMap } = await import("../tools/context-tools.js");
+            const result = await getKnowledgeMap(repo, query["focus"], query["depth"]);
+            const text = JSON.stringify(result);
+            return { type: qType, data: result, tokens: estimateTokens(text) };
+        }
+        case "semantic": {
+            const { getCodeIndex, getEmbeddingCache } = await import("../tools/index-tools.js");
+            const { createEmbeddingProvider, searchSemantic, cosineSimilarity } = await import("../search/semantic.js");
+            const { loadConfig: getConfig } = await import("../config.js");
+            const { getRepo } = await import("../storage/registry.js");
+            const { loadChunks, loadChunkEmbeddings, getChunkPath, getChunkEmbeddingPath } = await import("../storage/chunk-store.js");
+            const semanticConfig = getConfig();
+            if (!semanticConfig.embeddingProvider) {
+                throw new Error("No embedding provider configured. Set CODESIFT_VOYAGE_API_KEY, CODESIFT_OPENAI_API_KEY, or CODESIFT_OLLAMA_URL.");
+            }
+            const topK = query["top_k"] ?? 10;
+            const fileFilter = query["file_filter"];
+            const excludeTests = query["exclude_tests"] ?? true;
+            // Auto-decompose long queries for RRF merging (shared by both paths)
+            const provider = createEmbeddingProvider(semanticConfig.embeddingProvider, semanticConfig);
+            const subQueryTexts = decomposeQuery(query["query"]);
+            const vecs = await provider.embed(subQueryTexts);
+            // Try chunk-level semantic search first
+            const repoMeta = await getRepo(semanticConfig.registryPath, repo);
+            if (repoMeta) {
+                const chunkPath = getChunkPath(repoMeta.index_path);
+                const chunkEmbeddingPath = getChunkEmbeddingPath(repoMeta.index_path);
+                const [chunks, chunkEmbeddings] = await Promise.all([
+                    loadChunks(chunkPath),
+                    loadChunkEmbeddings(chunkEmbeddingPath),
+                ]);
+                if (chunks && chunkEmbeddings) {
+                    // Filter embeddings by file path substring and/or test file exclusion
+                    let filteredEmbeddings = chunkEmbeddings;
+                    if (fileFilter || excludeTests) {
+                        filteredEmbeddings = new Map([...chunkEmbeddings.entries()].filter(([id]) => {
+                            const chunkFile = chunks.get(id)?.file;
+                            if (!chunkFile)
+                                return false;
+                            if (fileFilter && !chunkFile.includes(fileFilter))
+                                return false;
+                            if (excludeTests && isTestFile(chunkFile))
+                                return false;
+                            return true;
+                        }));
+                    }
+                    // RRF over sub-query rankings (k=60 per standard RRF)
+                    const rrfScores = new Map();
+                    const rrfK = 60;
+                    for (const vec of vecs) {
+                        if (!vec)
+                            continue;
+                        const qEmbed = new Float32Array(vec);
+                        const subScores = [];
+                        for (const [id, chunkVec] of filteredEmbeddings) {
+                            if (chunkVec.length === qEmbed.length) {
+                                subScores.push({ id, score: cosineSimilarity(qEmbed, chunkVec) });
+                            }
+                        }
+                        subScores.sort((a, b) => b.score - a.score);
+                        subScores.forEach((s, rank) => {
+                            rrfScores.set(s.id, (rrfScores.get(s.id) ?? 0) + 1 / (rrfK + rank + 1));
+                        });
+                    }
+                    const topIds = [...rrfScores.entries()]
+                        .sort((a, b) => b[1] - a[1])
+                        .slice(0, topK)
+                        .map(([id]) => id);
+                    // Group chunks by file and sort by line number within each file
+                    const byFile = new Map();
+                    for (const id of topIds) {
+                        const chunk = chunks.get(id);
+                        if (!chunk)
+                            continue;
+                        const existing = byFile.get(chunk.file) ?? [];
+                        existing.push({ startLine: chunk.startLine, endLine: chunk.endLine, text: chunk.text });
+                        byFile.set(chunk.file, existing);
+                    }
+                    // Format as plain text — same style as Augment's context engine
+                    const sections = ["The following code sections were retrieved:"];
+                    for (const [file, fileChunks] of byFile) {
+                        fileChunks.sort((a, b) => a.startLine - b.startLine);
+                        // Merge overlapping/adjacent chunks to avoid duplicate lines
+                        const merged = [];
+                        for (const chunk of fileChunks) {
+                            const last = merged[merged.length - 1];
+                            if (last && chunk.startLine <= last.endLine + 5) {
+                                if (chunk.endLine > last.endLine) {
+                                    const overlapLines = last.endLine - chunk.startLine + 1;
+                                    const newLines = chunk.text.split("\n").slice(overlapLines);
+                                    last.text = last.text + "\n" + newLines.join("\n");
+                                    last.endLine = chunk.endLine;
+                                }
+                            }
+                            else {
+                                merged.push({ startLine: chunk.startLine, endLine: chunk.endLine, text: chunk.text });
+                            }
+                        }
+                        sections.push(`Path: ${file}`);
+                        for (const chunk of merged) {
+                            // Add line numbers to each line (matching Auggie's format)
+                            const lines = chunk.text.split("\n");
+                            const numbered = lines.map((line, i) => {
+                                const lineNo = String(chunk.startLine + i).padStart(6, " ");
+                                return `${lineNo}\t${line}`;
+                            }).join("\n");
+                            sections.push(numbered);
+                        }
+                        sections.push("...");
+                    }
+                    const text = sections.join("\n");
+                    return { type: qType, data: text, tokens: estimateTokens(text) };
+                }
+            }
+            // Fall back to symbol-level semantic search when no chunks are indexed
+            const index = await getCodeIndex(repo);
+            if (!index)
+                throw new Error(`Repository "${repo}" not found`);
+            const embeddings = await getEmbeddingCache(repo);
+            if (!embeddings)
+                throw new Error(`No embeddings for "${repo}". Run index_folder with an embedding provider configured.`);
+            const sourceLimit = query["source_chars"] ?? 200;
+            const symbolMap = new Map(index.symbols.map((s) => [s.id, s]));
+            // Filter embeddings by file path substring and/or test file exclusion
+            let filteredEmbeddings = embeddings;
+            if (fileFilter || excludeTests) {
+                filteredEmbeddings = new Map([...embeddings.entries()].filter(([id]) => {
+                    const symFile = symbolMap.get(id)?.file;
+                    if (!symFile)
+                        return false;
+                    if (fileFilter && !symFile.includes(fileFilter))
+                        return false;
+                    if (excludeTests && isTestFile(symFile))
+                        return false;
+                    return true;
+                }));
+            }
+            const primaryVec = vecs[0];
+            if (!primaryVec)
+                throw new Error("Embedding provider returned no vector");
+            const results = searchSemantic(new Float32Array(primaryVec), filteredEmbeddings, symbolMap, topK);
+            const data = results.map((r) => {
+                const sym = r.symbol;
+                return sourceLimit > 0 && sym.source && sym.source.length > sourceLimit
+                    ? { ...sym, source: sym.source.slice(0, sourceLimit) }
+                    : sym;
+            });
+            const text = JSON.stringify(data);
+            return { type: qType, data, tokens: estimateTokens(text) };
+        }
+        case "hybrid": {
+            // Hybrid: semantic embedding search + text/BM25 search, RRF-merged.
+            // Benefits: no need to choose semantic vs text — gets best of both.
+            const { getRepo: getRepoH } = await import("../storage/registry.js");
+            const { loadChunks: loadChunksH, loadChunkEmbeddings: loadChunkEmbeddingsH, getChunkPath: getChunkPathH, getChunkEmbeddingPath: getChunkEmbeddingPathH } = await import("../storage/chunk-store.js");
+            const { createEmbeddingProvider: createProviderH, cosineSimilarity: cosSimH } = await import("../search/semantic.js");
+            const { loadConfig: getConfigH } = await import("../config.js");
+            const { searchText: searchTextH } = await import("../tools/search-tools.js");
+            const hybridConfig = getConfigH();
+            if (!hybridConfig.embeddingProvider) {
+                throw new Error("No embedding provider configured.");
+            }
+            const topK = query["top_k"] ?? 10;
+            const fileFilter = query["file_filter"];
+            const excludeTests = query["exclude_tests"] ?? true;
+            const queryText = query["query"];
+            const repoMeta = await getRepoH(hybridConfig.registryPath, repo);
+            if (!repoMeta)
+                throw new Error(`Repository "${repo}" not found`);
+            const chunkPath = getChunkPathH(repoMeta.index_path);
+            const chunkEmbeddingPath = getChunkEmbeddingPathH(repoMeta.index_path);
+            // Run semantic embedding + text search in parallel
+            const provider = createProviderH(hybridConfig.embeddingProvider, hybridConfig);
+            const subQueryTexts = decomposeQuery(queryText);
+            const [chunks, chunkEmbeddings, textMatches, embVecs] = await Promise.all([
+                loadChunksH(chunkPath),
+                loadChunkEmbeddingsH(chunkEmbeddingPath),
+                searchTextH(repo, queryText, { file_pattern: fileFilter }).catch(() => []),
+                provider.embed(subQueryTexts),
+            ]);
+            if (!chunks || !chunkEmbeddings)
+                throw new Error(`No chunk index for "${repo}"`);
+            let filteredEmbeddings = chunkEmbeddings;
+            if (fileFilter || excludeTests) {
+                filteredEmbeddings = new Map([...chunkEmbeddings.entries()].filter(([id]) => {
+                    const chunkFile = chunks.get(id)?.file;
+                    if (!chunkFile)
+                        return false;
+                    if (fileFilter && !chunkFile.includes(fileFilter))
+                        return false;
+                    if (excludeTests && isTestFile(chunkFile))
+                        return false;
+                    return true;
+                }));
+            }
+            const rrfK = 60;
+            const rrfScores = new Map();
+            // 1. Semantic RRF contributions (one pass per decomposed sub-query)
+            for (const vec of embVecs) {
+                if (!vec)
+                    continue;
+                const qEmbed = new Float32Array(vec);
+                const subScores = [];
+                for (const [id, chunkVec] of filteredEmbeddings) {
+                    if (chunkVec.length === qEmbed.length) {
+                        subScores.push({ id, score: cosSimH(qEmbed, chunkVec) });
+                    }
+                }
+                subScores.sort((a, b) => b.score - a.score);
+                subScores.forEach((s, rank) => {
+                    rrfScores.set(s.id, (rrfScores.get(s.id) ?? 0) + 1 / (rrfK + rank + 1));
+                });
+            }
+            // 2. Text match RRF contributions — map match line → covering chunk
+            // Build file → sorted-chunks index for efficient line lookup
+            const fileToChunks = new Map();
+            for (const [id, chunk] of chunks) {
+                const list = fileToChunks.get(chunk.file) ?? [];
+                list.push({ id, startLine: chunk.startLine, endLine: chunk.endLine });
+                fileToChunks.set(chunk.file, list);
+            }
+            for (const list of fileToChunks.values()) {
+                list.sort((a, b) => a.startLine - b.startLine);
+            }
+            for (let rank = 0; rank < textMatches.length; rank++) {
+                const match = textMatches[rank];
+                if (!match)
+                    continue;
+                // Skip text matches from test files when excluding tests
+                if (excludeTests && isTestFile(match.file))
+                    continue;
+                const list = fileToChunks.get(match.file) ?? [];
+                for (const chunk of list) {
+                    if (chunk.startLine <= match.line && match.line <= chunk.endLine) {
+                        rrfScores.set(chunk.id, (rrfScores.get(chunk.id) ?? 0) + 1 / (rrfK + rank + 1));
+                        break;
+                    }
+                }
+            }
+            // Top_k by combined RRF score
+            const topIds = [...rrfScores.entries()]
+                .sort((a, b) => b[1] - a[1])
+                .slice(0, topK)
+                .map(([id]) => id);
+            // Format as plain text (same as semantic case)
+            const byFile = new Map();
+            for (const id of topIds) {
+                const chunk = chunks.get(id);
+                if (!chunk)
+                    continue;
+                // Double-check test file exclusion (text matches could contribute chunk IDs)
+                if (excludeTests && isTestFile(chunk.file))
+                    continue;
+                const existing = byFile.get(chunk.file) ?? [];
+                existing.push({ startLine: chunk.startLine, endLine: chunk.endLine, text: chunk.text });
+                byFile.set(chunk.file, existing);
+            }
+            const sections = ["The following code sections were retrieved:"];
+            for (const [file, fileChunks] of byFile) {
+                fileChunks.sort((a, b) => a.startLine - b.startLine);
+                const merged = [];
+                for (const chunk of fileChunks) {
+                    const last = merged[merged.length - 1];
+                    if (last && chunk.startLine <= last.endLine + 5) {
+                        if (chunk.endLine > last.endLine) {
+                            const overlapLines = last.endLine - chunk.startLine + 1;
+                            const newLines = chunk.text.split("\n").slice(overlapLines);
+                            last.text = last.text + "\n" + newLines.join("\n");
+                            last.endLine = chunk.endLine;
+                        }
+                    }
+                    else {
+                        merged.push({ startLine: chunk.startLine, endLine: chunk.endLine, text: chunk.text });
+                    }
+                }
+                sections.push(`Path: ${file}`);
+                for (const chunk of merged) {
+                    const lines = chunk.text.split("\n");
+                    const numbered = lines.map((line, i) => {
+                        const lineNo = String(chunk.startLine + i).padStart(6, " ");
+                        return `${lineNo}\t${line}`;
+                    }).join("\n");
+                    sections.push(numbered);
+                }
+                sections.push("...");
+            }
+            const hybridText = sections.join("\n");
+            return { type: qType, data: hybridText, tokens: estimateTokens(hybridText) };
+        }
+        default:
+            return {
+                type: qType,
+                data: { error: `Unknown sub-query type: ${qType}` },
+                tokens: 50,
+            };
+    }
+}
+/**
+ * Split a long query into sub-queries at natural connectors for RRF merging.
+ * Queries ≤ 8 words are returned as-is.
+ */
+function decomposeQuery(query) {
+    const words = query.split(/\s+/).filter(Boolean);
+    if (words.length <= 8)
+        return [query];
+    const splitWords = new Set(["and", "or", "from", "to", "with", "using", "for", "via", "then"]);
+    const lo = Math.floor(words.length * 0.35);
+    const hi = Math.floor(words.length * 0.65);
+    for (let i = lo; i <= hi; i++) {
+        if (splitWords.has((words[i] ?? "").toLowerCase())) {
+            const a = words.slice(0, i).join(" ");
+            const b = words.slice(i + 1).join(" ");
+            if (a.trim() && b.trim())
+                return [a, b];
+        }
+    }
+    const mid = Math.floor(words.length / 2);
+    return [words.slice(0, mid).join(" "), words.slice(mid).join(" ")];
+}
+/**
+ * Execute multiple sub-queries in a single batched call with shared token budget.
+ *
+ * Features:
+ * - Cross-query deduplication (same symbol in multiple results → included once)
+ * - Token budget enforcement
+ * - Max 20 sub-queries per call
+ * - Parallel execution where queries are independent
+ */
+export async function codebaseRetrieval(repo, queries, tokenBudget) {
+    const config = loadConfig();
+    const budget = tokenBudget ?? config.defaultTokenBudget;
+    const maxQueries = 20;
+    const limited = queries.slice(0, maxQueries);
+    // Execute all sub-queries in parallel
+    const subResults = await Promise.all(limited.map((q) => executeSubQuery(repo, q).catch((err) => {
+        const message = err instanceof Error ? err.message : String(err);
+        return {
+            type: q.type,
+            data: { error: message },
+            tokens: estimateTokens(message),
+        };
+    })));
+    // Enforce token budget — include results until budget is exceeded
+    const results = [];
+    let totalTokens = 0;
+    let truncated = false;
+    for (const result of subResults) {
+        if (totalTokens + result.tokens > budget) {
+            truncated = true;
+            // Try to include a truncated version
+            const remaining = budget - totalTokens;
+            if (remaining > 100) {
+                const truncatedText = JSON.stringify(result.data).slice(0, remaining * 4);
+                let truncatedData;
+                try {
+                    truncatedData = JSON.parse(truncatedText);
+                }
+                catch {
+                    truncatedData = { partial: true, note: "Result truncated to fit token budget" };
+                }
+                results.push({
+                    type: result.type,
+                    data: truncatedData,
+                    tokens: remaining,
+                });
+                totalTokens += remaining;
+            }
+            break;
+        }
+        results.push(result);
+        totalTokens += result.tokens;
+    }
+    return {
+        results,
+        total_tokens: totalTokens,
+        truncated,
+        query_count: limited.length,
+    };
+}
+//# sourceMappingURL=codebase-retrieval.js.map

package/dist/retrieval/codebase-retrieval.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"codebase-retrieval.js","sourceRoot":"","sources":["../../src/retrieval/codebase-retrieval.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,cAAc,CAAC;AAE1C,OAAO,EAAE,UAAU,EAAE,MAAM,uBAAuB,CAAC;AAenD;;GAEG;AACH,SAAS,cAAc,CAAC,IAAY;IAClC,OAAO,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;AACpC,CAAC;AAGD;;GAEG;AACH,KAAK,UAAU,eAAe,CAC5B,IAAY,EACZ,KAAe;IAEf,MAAM,KAAK,GAAG,KAAK,CAAC,IAAI,CAAC;IAEzB,QAAQ,KAAK,EAAE,CAAC;QACd,KAAK,SAAS,CAAC,CAAC,CAAC;YACf,MAAM,EAAE,aAAa,EAAE,GAAG,MAAM,MAAM,CAAC,0BAA0B,CAAC,CAAC;YACnE,MAAM,OAAO,GAAG,MAAM,aAAa,CAAC,IAAI,EAAE,KAAK,CAAC,OAAO,CAAW,EAAE;gBAClE,IAAI,EAAE,KAAK,CAAC,MAAM,CAA2B;gBAC7C,YAAY,EAAE,KAAK,CAAC,cAAc,CAAuB;gBACzD,cAAc,EAAE,IAAI;gBACpB,KAAK,EAAG,KAAK,CAAC,OAAO,CAAwB,IAAI,CAAC;aACnD,CAAC,CAAC;YACH,MAAM,WAAW,GAAI,KAAK,CAAC,cAAc,CAAwB,IAAI,GAAG,CAAC;YACzE,MAAM,IAAI,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;gBAC7B,MAAM,GAAG,GAAG,CAAC,CAAC,MAAM,CAAC;gBACrB,OAAO,WAAW,GAAG,CAAC,IAAI,GAAG,CAAC,MAAM,IAAI,GAAG,CAAC,MAAM,CAAC,MAAM,GAAG,WAAW;oBACrE,CAAC,CAAC,EAAE,GAAG,GAAG,EAAE,MAAM,EAAE,GAAG,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,WAAW,CAAC,EAAE;oBACtD,CAAC,CAAC,GAAG,CAAC;YACV,CAAC,CAAC,CAAC;YACH,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;YAClC,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,cAAc,CAAC,IAAI,CAAC,EAAE,CAAC;QAC7D,CAAC;QAED,KAAK,MAAM,CAAC,CAAC,CAAC;YACZ,MAAM,EAAE,UAAU,EAAE,GAAG,MAAM,MAAM,CAAC,0BAA0B,CAAC,CAAC;YAChE,MAAM,OAAO,GAAG,MAAM,UAAU,CAAC,IAAI,EAAE,KAAK,CAAC,OAAO,CAAW,EAAE;gBAC/D,KAAK,EAAE,KAAK,CAAC,OAAO,CAAwB;gBAC5C,aAAa,EAAE,KAAK,CAAC,eAAe,CAAuB;gBAC3D,YAAY,EAAE,KAAK,CAAC,cAAc,CAAuB;aAC1D,CAAC,CAAC;YACH,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;YACrC,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,EAAE,cAAc,CAAC,IAAI,CAAC,EAAE,CAAC;QACtE,CAAC;QAED,KAAK,WAAW,CAAC,CAAC,CAAC;YACjB,MAAM,EAAE,WAAW,EAAE,GAAG,MAAM,MAAM,CAAC,2BAA2B,CAAC,CAAC;YAClE,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,IAAI,EAAE;gBACrC,WAAW,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,aAAa,CAAC,CAAuB;gBAC1E,YAAY,EAAE,KAAK,CAAC,cAAc,CAAuB;gBACzD,KAAK,EAAE,KAAK,CAAC,OAAO,CAAuB;gBAC3C,OAAO,EAAE,KAAK,CAAC,SAAS,CAAwB;gBAChD,WAAW,EAAE,KAAK,CAAC,aAAa,CAAuB;aACxD,CAAC,CAAC;YACH,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;YACpC,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,cAAc,CAAC,IAAI,CAAC,EAAE,CAAC;QACrE,CAAC;QAED,KAAK,SAAS,CAAC,CAAC,CAAC;YACf,MAAM,EAAE,cAAc,EAAE,GAAG,MAAM,MAAM,CAAC,2BAA2B,CAAC,CAAC;YACrE,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC,IAAI,EAAE,KAAK,CAAC,WAAW,CAAW,CAAC,CAAC;YACxE,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;YACpC,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,cAAc,CAAC,IAAI,CAAC,EAAE,CAAC;QACrE,CAAC;QAED,KAAK,YAAY,CAAC,CAAC,CAAC;YAClB,MAAM,EAAE,cAAc,EAAE,GAAG,MAAM,MAAM,CAAC,0BAA0B,CAAC,CAAC;YACpE,MAAM,OAAO,GAAG,MAAM,cAAc,CAClC,IAAI,EACJ,KAAK,CAAC,aAAa,CAAW,CAC/B,CAAC;YACF,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;YACrC,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,EAAE,cAAc,CAAC,IAAI,CAAC,EAAE,CAAC;QACtE,CAAC;QAED,KAAK,YAAY,CAAC,CAAC,CAAC;YAClB,MAAM,EAAE,cAAc,EAAE,GAAG,MAAM,MAAM,CAAC,yBAAyB,CAAC,CAAC;YACnE,MAAM,MAAM,GAAG,MAAM,cAAc,CACjC,IAAI,EACJ,KAAK,CAAC,aAAa,CAAW,EAC7B,KAAK,CAAC,WAAW,CAA2B,IAAI,SAAS,EAC1D;gBACE,KAAK,EAAE,KAAK,CAAC,OAAO,CAAuB;gBAC3C,cAAc,EAAG,KAAK,CAAC,gBAAgB,CAAyB,IAAI,KAAK;aAC1E,CACF,CAAC;YACF,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;YACpC,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,cAAc,CAAC,IAAI,CAAC,EAAE,CAAC;QACrE,CAAC;QAED,KAAK,QAAQ,CAAC,CAAC,CAAC;YACd,MAAM,EAAE,cAAc,EAAE,GAAG,MAAM,MAAM,CAAC,yBAAyB,CAAC,CAAC;YACnE,MAAM,MAAM,GAAG,MAAM,cAAc,CACjC,IAAI,EACJ,KAAK,CAAC,OAAO,CAAW,EACxB;gBACE,KAAK,EAAE,KAAK,CAAC,OAAO,CAAuB;gBAC3C,KAAK,EAAE,KAAK,CAAC,OAAO,CAAuB;gBAC3C,cAAc,EAAG,KAAK,CAAC,gBAAgB,CAAyB,IAAI,KAAK;aAC1E,CACF,CAAC;YACF,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;YACpC,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,cAAc,CAAC,IAAI,CAAC,EAAE,CAAC;QACrE,CAAC;QAED,KAAK,SAAS,CAAC,CAAC,CAAC;YACf,MAAM,EAAE,eAAe,EAAE,GAAG,MAAM,MAAM,CAAC,2BAA2B,CAAC,CAAC;YACtE,MAAM,MAAM,GAAG,MAAM,eAAe,CAClC,IAAI,EACJ,KAAK,CAAC,OAAO,CAAW,EACxB,KAAK,CAAC,YAAY,CAAuB,CAC1C,CAAC;YACF,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;YACpC,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,cAAc,CAAC,IAAI,CAAC,EAAE,CAAC;QACrE,CAAC;QAED,KAAK,eAAe,CAAC,CAAC,CAAC;YACrB,MAAM,EAAE,eAAe,EAAE,GAAG,MAAM,MAAM,CAAC,2BAA2B,CAAC,CAAC;YACtE,MAAM,MAAM,GAAG,MAAM,eAAe,CAClC,IAAI,EACJ,KAAK,CAAC,OAAO,CAAuB,EACpC,KAAK,CAAC,OAAO,CAAuB,CACrC,CAAC;YACF,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;YACpC,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,cAAc,CAAC,IAAI,CAAC,EAAE,CAAC;QACrE,CAAC;QAED,KAAK,UAAU,CAAC,CAAC,CAAC;YAChB,MAAM,EAAE,YAAY,EAAE,iBAAiB,EAAE,GAAG,MAAM,MAAM,CAAC,yBAAyB,CAAC,CAAC;YACpF,MAAM,EAAE,uBAAuB,EAAE,cAAc,EAAE,gBAAgB,EAAE,GAAG,MAAM,MAAM,CAAC,uBAAuB,CAAC,CAAC;YAC5G,MAAM,EAAE,UAAU,EAAE,SAAS,EAAE,GAAG,MAAM,MAAM,CAAC,cAAc,CAAC,CAAC;YAC/D,MAAM,EAAE,OAAO,EAAE,GAAG,MAAM,MAAM,CAAC,wBAAwB,CAAC,CAAC;YAC3D,MAAM,EAAE,UAAU,EAAE,mBAAmB,EAAE,YAAY,EAAE,qBAAqB,EAAE,GAAG,MAAM,MAAM,CAAC,2BAA2B,CAAC,CAAC;YAE3H,MAAM,cAAc,GAAG,SAAS,EAAE,CAAC;YACnC,IAAI,CAAC,cAAc,CAAC,iBAAiB,EAAE,CAAC;gBACtC,MAAM,IAAI,KAAK,CAAC,iHAAiH,CAAC,CAAC;YACrI,CAAC;YAED,MAAM,IAAI,GAAI,KAAK,CAAC,OAAO,CAAwB,IAAI,EAAE,CAAC;YAC1D,MAAM,UAAU,GAAG,KAAK,CAAC,aAAa,CAAuB,CAAC;YAC9D,MAAM,YAAY,GAAI,KAAK,CAAC,eAAe,CAAyB,IAAI,IAAI,CAAC;YAE7E,qEAAqE;YACrE,MAAM,QAAQ,GAAG,uBAAuB,CAAC,cAAc,CAAC,iBAAiB,EAAE,cAAc,CAAC,CAAC;YAC3F,MAAM,aAAa,GAAG,cAAc,CAAC,KAAK,CAAC,OAAO,CAAW,CAAC,CAAC;YAC/D,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC;YAEjD,wCAAwC;YACxC,MAAM,QAAQ,GAAG,MAAM,OAAO,CAAC,cAAc,CAAC,YAAY,EAAE,IAAI,CAAC,CAAC;YAClE,IAAI,QAAQ,EAAE,CAAC;gBACb,MAAM,SAAS,GAAG,YAAY,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC;gBACpD,MAAM,kBAAkB,GAAG,qBAAqB,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC;gBACtE,MAAM,CAAC,MAAM,EAAE,eAAe,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;oBAClD,UAAU,CAAC,SAAS,CAAC;oBACrB,mBAAmB,CAAC,kBAAkB,CAAC;iBACxC,CAAC,CAAC;gBAEH,IAAI,MAAM,IAAI,eAAe,EAAE,CAAC;oBAC9B,sEAAsE;oBACtE,IAAI,kBAAkB,GAAG,eAAe,CAAC;oBACzC,IAAI,UAAU,IAAI,YAAY,EAAE,CAAC;wBAC/B,kBAAkB,GAAG,IAAI,GAAG,CAAC,CAAC,GAAG,eAAe,CAAC,OAAO,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE;4BAC1E,MAAM,SAAS,GAAG,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,IAAI,CAAC;4BACvC,IAAI,CAAC,SAAS;gCAAE,OAAO,KAAK,CAAC;4BAC7B,IAAI,UAAU,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,UAAU,CAAC;gCAAE,OAAO,KAAK,CAAC;4BAChE,IAAI,YAAY,IAAI,UAAU,CAAC,SAAS,CAAC;gCAAE,OAAO,KAAK,CAAC;4BACxD,OAAO,IAAI,CAAC;wBACd,CAAC,CAAC,CAAC,CAAC;oBACN,CAAC;oBAED,sDAAsD;oBACtD,MAAM,SAAS,GAAG,IAAI,GAAG,EAAkB,CAAC;oBAC5C,MAAM,IAAI,GAAG,EAAE,CAAC;oBAChB,KAAK,MAAM,GAAG,IAAI,IAAI,EAAE,CAAC;wBACvB,IAAI,CAAC,GAAG;4BAAE,SAAS;wBACnB,MAAM,MAAM,GAAG,IAAI,YAAY,CAAC,GAAG,CAAC,CAAC;wBACrC,MAAM,SAAS,GAAyC,EAAE,CAAC;wBAC3D,KAAK,MAAM,CAAC,EAAE,EAAE,QAAQ,CAAC,IAAI,kBAAkB,EAAE,CAAC;4BAChD,IAAI,QAAQ,CAAC,MAAM,KAAK,MAAM,CAAC,MAAM,EAAE,CAAC;gCACtC,SAAS,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,gBAAgB,CAAC,MAAM,EAAE,QAAQ,CAAC,EAAE,CAAC,CAAC;4BACpE,CAAC;wBACH,CAAC;wBACD,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC;wBAC5C,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,EAAE;4BAC5B,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,GAAG,IAAI,GAAG,CAAC,CAAC,CAAC,CAAC;wBAC1E,CAAC,CAAC,CAAC;oBACL,CAAC;oBACD,MAAM,MAAM,GAAG,CAAC,GAAG,SAAS,CAAC,OAAO,EAAE,CAAC;yBACpC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;yBAC3B,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC;yBACd,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC;oBAErB,gEAAgE;oBAChE,MAAM,MAAM,GAAG,IAAI,GAAG,EAAuE,CAAC;oBAC9F,KAAK,MAAM,EAAE,IAAI,MAAM,EAAE,CAAC;wBACxB,MAAM,KAAK,GAAG,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;wBAC7B,IAAI,CAAC,KAAK;4BAAE,SAAS;wBACrB,MAAM,QAAQ,GAAG,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;wBAC9C,QAAQ,CAAC,IAAI,CAAC,EAAE,SAAS,EAAE,KAAK,CAAC,SAAS,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC;wBACxF,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;oBACnC,CAAC;oBAED,gEAAgE;oBAChE,MAAM,QAAQ,GAAa,CAAC,6CAA6C,CAAC,CAAC;oBAC3E,KAAK,MAAM,CAAC,IAAI,EAAE,UAAU,CAAC,IAAI,MAAM,EAAE,CAAC;wBACxC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,GAAG,CAAC,CAAC,SAAS,CAAC,CAAC;wBACrD,6DAA6D;wBAC7D,MAAM,MAAM,GAAgE,EAAE,CAAC;wBAC/E,KAAK,MAAM,KAAK,IAAI,UAAU,EAAE,CAAC;4BAC/B,MAAM,IAAI,GAAG,MAAM,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;4BACvC,IAAI,IAAI,IAAI,KAAK,CAAC,SAAS,IAAI,IAAI,CAAC,OAAO,GAAG,CAAC,EAAE,CAAC;gCAChD,IAAI,KAAK,CAAC,OAAO,GAAG,IAAI,CAAC,OAAO,EAAE,CAAC;oCACjC,MAAM,YAAY,GAAG,IAAI,CAAC,OAAO,GAAG,KAAK,CAAC,SAAS,GAAG,CAAC,CAAC;oCACxD,MAAM,QAAQ,GAAG,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;oCAC5D,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC,IAAI,GAAG,IAAI,GAAG,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;oCACnD,IAAI,CAAC,OAAO,GAAG,KAAK,CAAC,OAAO,CAAC;gCAC/B,CAAC;4BACH,CAAC;iCAAM,CAAC;gCACN,MAAM,CAAC,IAAI,CAAC,EAAE,SAAS,EAAE,KAAK,CAAC,SAAS,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC;4BACxF,CAAC;wBACH,CAAC;wBACD,QAAQ,CAAC,IAAI,CAAC,SAAS,IAAI,EAAE,CAAC,CAAC;wBAC/B,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;4BAC3B,2DAA2D;4BAC3D,MAAM,KAAK,GAAG,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;4BACrC,MAAM,QAAQ,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,CAAC,EAAE,EAAE;gCACrC,MAAM,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,SAAS,GAAG,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;gCAC5D,OAAO,GAAG,MAAM,KAAK,IAAI,EAAE,CAAC;4BAC9B,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;4BACd,QAAQ,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;wBAC1B,CAAC;wBACD,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;oBACvB,CAAC;oBAED,MAAM,IAAI,GAAG,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;oBACjC,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE,cAAc,CAAC,IAAI,CAAC,EAAE,CAAC;gBACnE,CAAC;YACH,CAAC;YAED,uEAAuE;YACvE,MAAM,KAAK,GAAG,MAAM,YAAY,CAAC,IAAI,CAAC,CAAC;YACvC,IAAI,CAAC,KAAK;gBAAE,MAAM,IAAI,KAAK,CAAC,eAAe,IAAI,aAAa,CAAC,CAAC;YAE9D,MAAM,UAAU,GAAG,MAAM,iBAAiB,CAAC,IAAI,CAAC,CAAC;YACjD,IAAI,CAAC,UAAU;gBAAE,MAAM,IAAI,KAAK,CAAC,sBAAsB,IAAI,4DAA4D,CAAC,CAAC;YAEzH,MAAM,WAAW,GAAI,KAAK,CAAC,cAAc,CAAwB,IAAI,GAAG,CAAC;YACzE,MAAM,SAAS,GAAG,IAAI,GAAG,CAAC,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;YAE/D,sEAAsE;YACtE,IAAI,kBAAkB,GAAG,UAAU,CAAC;YACpC,IAAI,UAAU,IAAI,YAAY,EAAE,CAAC;gBAC/B,kBAAkB,GAAG,IAAI,GAAG,CAAC,CAAC,GAAG,UAAU,CAAC,OAAO,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE;oBACrE,MAAM,OAAO,GAAG,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,IAAI,CAAC;oBACxC,IAAI,CAAC,OAAO;wBAAE,OAAO,KAAK,CAAC;oBAC3B,IAAI,UAAU,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAC;wBAAE,OAAO,KAAK,CAAC;oBAC9D,IAAI,YAAY,IAAI,UAAU,CAAC,OAAO,CAAC;wBAAE,OAAO,KAAK,CAAC;oBACtD,OAAO,IAAI,CAAC;gBACd,CAAC,CAAC,CAAC,CAAC;YACN,CAAC;YAED,MAAM,UAAU,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;YAC3B,IAAI,CAAC,UAAU;gBAAE,MAAM,IAAI,KAAK,CAAC,uCAAuC,CAAC,CAAC;YAC1E,MAAM,OAAO,GAAG,cAAc,CAAC,IAAI,YAAY,CAAC,UAAU,CAAC,EAAE,kBAAkB,EAAE,SAAS,EAAE,IAAI,CAAC,CAAC;YAClG,MAAM,IAAI,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;gBAC7B,MAAM,GAAG,GAAG,CAAC,CAAC,MAAM,CAAC;gBACrB,OAAO,WAAW,GAAG,CAAC,IAAI,GAAG,CAAC,MAAM,IAAI,GAAG,CAAC,MAAM,CAAC,MAAM,GAAG,WAAW;oBACrE,CAAC,CAAC,EAAE,GAAG,GAAG,EAAE,MAAM,EAAE,GAAG,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,WAAW,CAAC,EAAE;oBACtD,CAAC,CAAC,GAAG,CAAC;YACV,CAAC,CAAC,CAAC;YACH,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;YAClC,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,cAAc,CAAC,IAAI,CAAC,EAAE,CAAC;QAC7D,CAAC;QAED,KAAK,QAAQ,CAAC,CAAC,CAAC;YACd,oEAAoE;YACpE,oEAAoE;YACpE,MAAM,EAAE,OAAO,EAAE,QAAQ,EAAE,GAAG,MAAM,MAAM,CAAC,wBAAwB,CAAC,CAAC;YACrE,MAAM,EAAE,UAAU,EAAE,WAAW,EAAE,mBAAmB,EAAE,oBAAoB,EAAE,YAAY,EAAE,aAAa,EAAE,qBAAqB,EAAE,sBAAsB,EAAE,GAAG,MAAM,MAAM,CAAC,2BAA2B,CAAC,CAAC;YACrM,MAAM,EAAE,uBAAuB,EAAE,eAAe,EAAE,gBAAgB,EAAE,OAAO,EAAE,GAAG,MAAM,MAAM,CAAC,uBAAuB,CAAC,CAAC;YACtH,MAAM,EAAE,UAAU,EAAE,UAAU,EAAE,GAAG,MAAM,MAAM,CAAC,cAAc,CAAC,CAAC;YAChE,MAAM,EAAE,UAAU,EAAE,WAAW,EAAE,GAAG,MAAM,MAAM,CAAC,0BAA0B,CAAC,CAAC;YAE7E,MAAM,YAAY,GAAG,UAAU,EAAE,CAAC;YAClC,IAAI,CAAC,YAAY,CAAC,iBAAiB,EAAE,CAAC;gBACpC,MAAM,IAAI,KAAK,CAAC,mCAAmC,CAAC,CAAC;YACvD,CAAC;YAED,MAAM,IAAI,GAAI,KAAK,CAAC,OAAO,CAAwB,IAAI,EAAE,CAAC;YAC1D,MAAM,UAAU,GAAG,KAAK,CAAC,aAAa,CAAuB,CAAC;YAC9D,MAAM,YAAY,GAAI,KAAK,CAAC,eAAe,CAAyB,IAAI,IAAI,CAAC;YAC7E,MAAM,SAAS,GAAG,KAAK,CAAC,OAAO,CAAW,CAAC;YAE3C,MAAM,QAAQ,GAAG,MAAM,QAAQ,CAAC,YAAY,CAAC,YAAY,EAAE,IAAI,CAAC,CAAC;YACjE,IAAI,CAAC,QAAQ;gBAAE,MAAM,IAAI,KAAK,CAAC,eAAe,IAAI,aAAa,CAAC,CAAC;YAEjE,MAAM,SAAS,GAAG,aAAa,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC;YACrD,MAAM,kBAAkB,GAAG,sBAAsB,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC;YAEvE,mDAAmD;YACnD,MAAM,QAAQ,GAAG,eAAe,CAAC,YAAY,CAAC,iBAAiB,EAAE,YAAY,CAAC,CAAC;YAC/E,MAAM,aAAa,GAAG,cAAc,CAAC,SAAS,CAAC,CAAC;YAChD,MAAM,CAAC,MAAM,EAAE,eAAe,EAAE,WAAW,EAAE,OAAO,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;gBACxE,WAAW,CAAC,SAAS,CAAC;gBACtB,oBAAoB,CAAC,kBAAkB,CAAC;gBACxC,WAAW,CAAC,IAAI,EAAE,SAAS,EAAE,EAAE,YAAY,EAAE,UAAU,EAAE,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC;gBAC1E,QAAQ,CAAC,KAAK,CAAC,aAAa,CAAC;aAC9B,CAAC,CAAC;YAEH,IAAI,CAAC,MAAM,IAAI,CAAC,eAAe;gBAAE,MAAM,IAAI,KAAK,CAAC,uBAAuB,IAAI,GAAG,CAAC,CAAC;YAEjF,IAAI,kBAAkB,GAAG,eAAe,CAAC;YACzC,IAAI,UAAU,IAAI,YAAY,EAAE,CAAC;gBAC/B,kBAAkB,GAAG,IAAI,GAAG,CAAC,CAAC,GAAG,eAAe,CAAC,OAAO,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE;oBAC1E,MAAM,SAAS,GAAG,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,IAAI,CAAC;oBACvC,IAAI,CAAC,SAAS;wBAAE,OAAO,KAAK,CAAC;oBAC7B,IAAI,UAAU,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,UAAU,CAAC;wBAAE,OAAO,KAAK,CAAC;oBAChE,IAAI,YAAY,IAAI,UAAU,CAAC,SAAS,CAAC;wBAAE,OAAO,KAAK,CAAC;oBACxD,OAAO,IAAI,CAAC;gBACd,CAAC,CAAC,CAAC,CAAC;YACN,CAAC;YAED,MAAM,IAAI,GAAG,EAAE,CAAC;YAChB,MAAM,SAAS,GAAG,IAAI,GAAG,EAAkB,CAAC;YAE5C,oEAAoE;YACpE,KAAK,MAAM,GAAG,IAAI,OAAO,EAAE,CAAC;gBAC1B,IAAI,CAAC,GAAG;oBAAE,SAAS;gBACnB,MAAM,MAAM,GAAG,IAAI,YAAY,CAAC,GAAG,CAAC,CAAC;gBACrC,MAAM,SAAS,GAAyC,EAAE,CAAC;gBAC3D,KAAK,MAAM,CAAC,EAAE,EAAE,QAAQ,CAAC,IAAI,kBAAkB,EAAE,CAAC;oBAChD,IAAI,QAAQ,CAAC,MAAM,KAAK,MAAM,CAAC,MAAM,EAAE,CAAC;wBACtC,SAAS,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,OAAO,CAAC,MAAM,EAAE,QAAQ,CAAC,EAAE,CAAC,CAAC;oBAC3D,CAAC;gBACH,CAAC;gBACD,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC;gBAC5C,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,EAAE;oBAC5B,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,GAAG,IAAI,GAAG,CAAC,CAAC,CAAC,CAAC;gBAC1E,CAAC,CAAC,CAAC;YACL,CAAC;YAED,oEAAoE;YACpE,6DAA6D;YAC7D,MAAM,YAAY,GAAG,IAAI,GAAG,EAAqE,CAAC;YAClG,KAAK,MAAM,CAAC,EAAE,EAAE,KAAK,CAAC,IAAI,MAAM,EAAE,CAAC;gBACjC,MAAM,IAAI,GAAG,YAAY,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;gBAChD,IAAI,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,SAAS,EAAE,KAAK,CAAC,SAAS,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;gBACtE,YAAY,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;YACrC,CAAC;YACD,KAAK,MAAM,IAAI,IAAI,YAAY,CAAC,MAAM,EAAE,EAAE,CAAC;gBACzC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,GAAG,CAAC,CAAC,SAAS,CAAC,CAAC;YACjD,CAAC;YAED,KAAK,IAAI,IAAI,GAAG,CAAC,EAAE,IAAI,GAAG,WAAW,CAAC,MAAM,EAAE,IAAI,EAAE,EAAE,CAAC;gBACrD,MAAM,KAAK,GAAG,WAAW,CAAC,IAAI,CAAC,CAAC;gBAChC,IAAI,CAAC,KAAK;oBAAE,SAAS;gBACrB,yDAAyD;gBACzD,IAAI,YAAY,IAAI,UAAU,CAAC,KAAK,CAAC,IAAI,CAAC;oBAAE,SAAS;gBACrD,MAAM,IAAI,GAAG,YAAY,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;gBAChD,KAAK,MAAM,KAAK,IAAI,IAAI,EAAE,CAAC;oBACzB,IAAI,KAAK,CAAC,SAAS,IAAI,KAAK,CAAC,IAAI,IAAI,KAAK,CAAC,IAAI,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;wBACjE,SAAS,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,GAAG,IAAI,GAAG,CAAC,CAAC,CAAC,CAAC;wBAChF,MAAM;oBACR,CAAC;gBACH,CAAC;YACH,CAAC;YAED,8BAA8B;YAC9B,MAAM,MAAM,GAAG,CAAC,GAAG,SAAS,CAAC,OAAO,EAAE,CAAC;iBACpC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;iBAC3B,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC;iBACd,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC;YAErB,+CAA+C;YAC/C,MAAM,MAAM,GAAG,IAAI,GAAG,EAAuE,CAAC;YAC9F,KAAK,MAAM,EAAE,IAAI,MAAM,EAAE,CAAC;gBACxB,MAAM,KAAK,GAAG,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;gBAC7B,IAAI,CAAC,KAAK;oBAAE,SAAS;gBACrB,6EAA6E;gBAC7E,IAAI,YAAY,IAAI,UAAU,CAAC,KAAK,CAAC,IAAI,CAAC;oBAAE,SAAS;gBACrD,MAAM,QAAQ,GAAG,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;gBAC9C,QAAQ,CAAC,IAAI,CAAC,EAAE,SAAS,EAAE,KAAK,CAAC,SAAS,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC;gBACxF,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;YACnC,CAAC;YAED,MAAM,QAAQ,GAAa,CAAC,6CAA6C,CAAC,CAAC;YAC3E,KAAK,MAAM,CAAC,IAAI,EAAE,UAAU,CAAC,IAAI,MAAM,EAAE,CAAC;gBACxC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,GAAG,CAAC,CAAC,SAAS,CAAC,CAAC;gBACrD,MAAM,MAAM,GAAgE,EAAE,CAAC;gBAC/E,KAAK,MAAM,KAAK,IAAI,UAAU,EAAE,CAAC;oBAC/B,MAAM,IAAI,GAAG,MAAM,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;oBACvC,IAAI,IAAI,IAAI,KAAK,CAAC,SAAS,IAAI,IAAI,CAAC,OAAO,GAAG,CAAC,EAAE,CAAC;wBAChD,IAAI,KAAK,CAAC,OAAO,GAAG,IAAI,CAAC,OAAO,EAAE,CAAC;4BACjC,MAAM,YAAY,GAAG,IAAI,CAAC,OAAO,GAAG,KAAK,CAAC,SAAS,GAAG,CAAC,CAAC;4BACxD,MAAM,QAAQ,GAAG,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;4BAC5D,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC,IAAI,GAAG,IAAI,GAAG,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;4BACnD,IAAI,CAAC,OAAO,GAAG,KAAK,CAAC,OAAO,CAAC;wBAC/B,CAAC;oBACH,CAAC;yBAAM,CAAC;wBACN,MAAM,CAAC,IAAI,CAAC,EAAE,SAAS,EAAE,KAAK,CAAC,SAAS,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC;oBACxF,CAAC;gBACH,CAAC;gBACD,QAAQ,CAAC,IAAI,CAAC,SAAS,IAAI,EAAE,CAAC,CAAC;gBAC/B,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;oBAC3B,MAAM,KAAK,GAAG,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;oBACrC,MAAM,QAAQ,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,CAAC,EAAE,EAAE;wBACrC,MAAM,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,SAAS,GAAG,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;wBAC5D,OAAO,GAAG,MAAM,KAAK,IAAI,EAAE,CAAC;oBAC9B,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;oBACd,QAAQ,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;gBAC1B,CAAC;gBACD,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACvB,CAAC;YAED,MAAM,UAAU,GAAG,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACvC,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,UAAU,EAAE,MAAM,EAAE,cAAc,CAAC,UAAU,CAAC,EAAE,CAAC;QAC/E,CAAC;QAED;YACE,OAAO;gBACL,IAAI,EAAE,KAAK;gBACX,IAAI,EAAE,EAAE,KAAK,EAAE,2BAA2B,KAAK,EAAE,EAAE;gBACnD,MAAM,EAAE,EAAE;aACX,CAAC;IACN,CAAC;AACH,CAAC;AAED;;;GAGG;AACH,SAAS,cAAc,CAAC,KAAa;IACnC,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IACjD,IAAI,KAAK,CAAC,MAAM,IAAI,CAAC;QAAE,OAAO,CAAC,KAAK,CAAC,CAAC;IAEtC,MAAM,UAAU,GAAG,IAAI,GAAG,CAAC,CAAC,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,CAAC,CAAC,CAAC;IAC/F,MAAM,EAAE,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,MAAM,GAAG,IAAI,CAAC,CAAC;IAC3C,MAAM,EAAE,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,MAAM,GAAG,IAAI,CAAC,CAAC;IAE3C,KAAK,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC;QAC9B,IAAI,UAAU,CAAC,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,WAAW,EAAE,CAAC,EAAE,CAAC;YACnD,MAAM,CAAC,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACtC,MAAM,CAAC,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACvC,IAAI,CAAC,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC,IAAI,EAAE;gBAAE,OAAO,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;QAC1C,CAAC;IACH,CAAC;IAED,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;IACzC,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;AACrE,CAAC;AASD;;;;;;;;GAQG;AACH,MAAM,CAAC,KAAK,UAAU,iBAAiB,CACrC,IAAY,EACZ,OAAmB,EACnB,WAAoB;IAEpB,MAAM,MAAM,GAAG,UAAU,EAAE,CAAC;IAC5B,MAAM,MAAM,GAAG,WAAW,IAAI,MAAM,CAAC,kBAAkB,CAAC;IACxD,MAAM,UAAU,GAAG,EAAE,CAAC;IAEtB,MAAM,OAAO,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,UAAU,CAAC,CAAC;IAE7C,sCAAsC;IACtC,MAAM,UAAU,GAAG,MAAM,OAAO,CAAC,GAAG,CAClC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAChB,eAAe,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,GAAY,EAAE,EAAE;QAC9C,MAAM,OAAO,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QACjE,OAAO;YACL,IAAI,EAAE,CAAC,CAAC,IAAI;YACZ,IAAI,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE;YACxB,MAAM,EAAE,cAAc,CAAC,OAAO,CAAC;SACP,CAAC;IAC7B,CAAC,CAAC,CACH,CACF,CAAC;IAEF,kEAAkE;IAClE,MAAM,OAAO,GAAqB,EAAE,CAAC;IACrC,IAAI,WAAW,GAAG,CAAC,CAAC;IACpB,IAAI,SAAS,GAAG,KAAK,CAAC;IAEtB,KAAK,MAAM,MAAM,IAAI,UAAU,EAAE,CAAC;QAChC,IAAI,WAAW,GAAG,MAAM,CAAC,MAAM,GAAG,MAAM,EAAE,CAAC;YACzC,SAAS,GAAG,IAAI,CAAC;YACjB,qCAAqC;YACrC,MAAM,SAAS,GAAG,MAAM,GAAG,WAAW,CAAC;YACvC,IAAI,SAAS,GAAG,GAAG,EAAE,CAAC;gBACpB,MAAM,aAAa,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,SAAS,GAAG,CAAC,CAAC,CAAC;gBAC1E,IAAI,aAAsB,CAAC;gBAC3B,IAAI,CAAC;oBACH,aAAa,GAAG,IAAI,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC;gBAC5C,CAAC;gBAAC,MAAM,CAAC;oBACP,aAAa,GAAG,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,sCAAsC,EAAE,CAAC;gBAClF,CAAC;gBACD,OAAO,CAAC,IAAI,CAAC;oBACX,IAAI,EAAE,MAAM,CAAC,IAAI;oBACjB,IAAI,EAAE,aAAa;oBACnB,MAAM,EAAE,SAAS;iBAClB,CAAC,CAAC;gBACH,WAAW,IAAI,SAAS,CAAC;YAC3B,CAAC;YACD,MAAM;QACR,CAAC;QACD,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACrB,WAAW,IAAI,MAAM,CAAC,MAAM,CAAC;IAC/B,CAAC;IAED,OAAO;QACL,OAAO;QACP,YAAY,EAAE,WAAW;QACzB,SAAS;QACT,WAAW,EAAE,OAAO,CAAC,MAAM;KAC5B,CAAC;AACJ,CAAC"}

package/dist/search/bm25.d.ts

@@ -0,0 +1,22 @@
+import type { CodeSymbol, SearchResult } from "../types.js";
+type FieldName = "name" | "signature" | "docstring" | "body";
+export interface BM25Index {
+    /** Per-field inverted index: token -> Map<symbolId, termFrequency> */
+    fields: Record<FieldName, Map<string, Map<string, number>>>;
+    /** Per-field average document length (in tokens) */
+    avgFieldLengths: Record<FieldName, number>;
+    /** Total number of indexed documents */
+    docCount: number;
+    /** Symbol lookup by ID */
+    symbols: Map<string, CodeSymbol>;
+}
+/**
+ * General-purpose tokenizer for signature, docstring, and body text.
+ * Splits on non-alphanumeric chars, applies camelCase/snake_case splitting,
+ * lowercases, and filters tokens shorter than 2 chars.
+ */
+export declare function tokenizeText(text: string): string[];
+export declare function buildBM25Index(symbols: CodeSymbol[]): BM25Index;
+export declare function searchBM25(index: BM25Index, query: string, topK: number, fieldWeights: Record<FieldName, number>): SearchResult[];
+export {};
+//# sourceMappingURL=bm25.d.ts.map

package/dist/search/bm25.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"bm25.d.ts","sourceRoot":"","sources":["../../src/search/bm25.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,UAAU,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAe5D,KAAK,SAAS,GAAG,MAAM,GAAG,WAAW,GAAG,WAAW,GAAG,MAAM,CAAC;AAE7D,MAAM,WAAW,SAAS;IACxB,sEAAsE;IACtE,MAAM,EAAE,MAAM,CAAC,SAAS,EAAE,GAAG,CAAC,MAAM,EAAE,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC,CAAC;IAC5D,oDAAoD;IACpD,eAAe,EAAE,MAAM,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC;IAC3C,wCAAwC;IACxC,QAAQ,EAAE,MAAM,CAAC;IACjB,0BAA0B;IAC1B,OAAO,EAAE,GAAG,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;CAClC;AAED;;;;GAIG;AACH,wBAAgB,YAAY,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM,EAAE,CAqBnD;AAqBD,wBAAgB,cAAc,CAAC,OAAO,EAAE,UAAU,EAAE,GAAG,SAAS,CAgD/D;AAED,wBAAgB,UAAU,CACxB,KAAK,EAAE,SAAS,EAChB,KAAK,EAAE,MAAM,EACb,IAAI,EAAE,MAAM,EACZ,YAAY,EAAE,MAAM,CAAC,SAAS,EAAE,MAAM,CAAC,GACtC,YAAY,EAAE,CAiGhB"}