agentic-qe 2.5.9 → 2.6.0

package/dist/code-intelligence/rag/types.d.ts

@@ -0,0 +1,121 @@
+/**
+ * Types for RAG (Retrieval-Augmented Generation) Module
+ *
+ * Defines interfaces for context building, query processing,
+ * and response generation.
+ */
+export interface RAGConfig {
+    /**
+     * Maximum tokens in generated context.
+     */
+    maxContextTokens: number;
+    /**
+     * Number of chunks to retrieve.
+     */
+    topK: number;
+    /**
+     * Minimum relevance score for inclusion.
+     */
+    minRelevanceScore: number;
+    /**
+     * Enable graph expansion for related code.
+     */
+    enableGraphExpansion: boolean;
+    /**
+     * Depth of graph expansion.
+     */
+    graphExpansionDepth: number;
+    /**
+     * Enable response caching.
+     */
+    enableCaching: boolean;
+    /**
+     * Cache TTL in milliseconds.
+     */
+    cacheTtlMs: number;
+    /**
+     * Summarization strategy.
+     */
+    summarizationStrategy: 'tree' | 'map-reduce' | 'refine';
+    /**
+     * LLM model for summarization.
+     */
+    summarizationModel?: string;
+}
+export interface RetrievedContext {
+    /** Unique ID */
+    id: string;
+    /** Source file path */
+    filePath: string;
+    /** Code content */
+    content: string;
+    /** Start line in file */
+    startLine: number;
+    /** End line in file */
+    endLine: number;
+    /** Relevance score */
+    score: number;
+    /** Entity type (function, class, etc.) */
+    entityType?: string;
+    /** Entity name */
+    entityName?: string;
+    /** Programming language */
+    language?: string;
+    /** Related contexts from graph expansion */
+    relatedContexts?: RetrievedContext[];
+    /** Relationship type for related contexts */
+    relationship?: string;
+}
+export interface QueryContext {
+    /** Original user query */
+    query: string;
+    /** Parsed intent (if applicable) */
+    intent?: 'explain' | 'find' | 'modify' | 'debug' | 'generate';
+    /** Extracted entities from query */
+    entities?: string[];
+    /** Filter by language */
+    language?: string;
+    /** Filter by file pattern */
+    filePattern?: string;
+    /** Additional metadata */
+    metadata?: Record<string, unknown>;
+}
+export interface GeneratedContext {
+    /** Formatted context string for LLM */
+    contextText: string;
+    /** Token count estimate */
+    tokenCount: number;
+    /** Retrieved chunks used */
+    sources: RetrievedContext[];
+    /** Whether context was truncated */
+    truncated: boolean;
+    /** Summary if generated */
+    summary?: string;
+    /** Cache hit */
+    cached: boolean;
+}
+export interface RAGResponse {
+    /** Generated context */
+    context: GeneratedContext;
+    /** Processing metadata */
+    metadata: {
+        query: string;
+        retrievalTimeMs: number;
+        graphExpansionTimeMs: number;
+        formattingTimeMs: number;
+        totalTimeMs: number;
+        chunksRetrieved: number;
+        chunksIncluded: number;
+    };
+}
+export interface ContextTemplate {
+    /** Template name */
+    name: string;
+    /** Template for formatting context */
+    format: string;
+    /** Variables available in template */
+    variables: string[];
+}
+export declare const DEFAULT_RAG_CONFIG: RAGConfig;
+export declare const DEFAULT_CONTEXT_TEMPLATE: ContextTemplate;
+//# sourceMappingURL=types.d.ts.map

package/dist/code-intelligence/rag/types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../../src/code-intelligence/rag/types.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,MAAM,WAAW,SAAS;IACxB;;OAEG;IACH,gBAAgB,EAAE,MAAM,CAAC;IAEzB;;OAEG;IACH,IAAI,EAAE,MAAM,CAAC;IAEb;;OAEG;IACH,iBAAiB,EAAE,MAAM,CAAC;IAE1B;;OAEG;IACH,oBAAoB,EAAE,OAAO,CAAC;IAE9B;;OAEG;IACH,mBAAmB,EAAE,MAAM,CAAC;IAE5B;;OAEG;IACH,aAAa,EAAE,OAAO,CAAC;IAEvB;;OAEG;IACH,UAAU,EAAE,MAAM,CAAC;IAEnB;;OAEG;IACH,qBAAqB,EAAE,MAAM,GAAG,YAAY,GAAG,QAAQ,CAAC;IAExD;;OAEG;IACH,kBAAkB,CAAC,EAAE,MAAM,CAAC;CAC7B;AAED,MAAM,WAAW,gBAAgB;IAC/B,gBAAgB;IAChB,EAAE,EAAE,MAAM,CAAC;IAEX,uBAAuB;IACvB,QAAQ,EAAE,MAAM,CAAC;IAEjB,mBAAmB;IACnB,OAAO,EAAE,MAAM,CAAC;IAEhB,yBAAyB;IACzB,SAAS,EAAE,MAAM,CAAC;IAElB,uBAAuB;IACvB,OAAO,EAAE,MAAM,CAAC;IAEhB,sBAAsB;IACtB,KAAK,EAAE,MAAM,CAAC;IAEd,0CAA0C;IAC1C,UAAU,CAAC,EAAE,MAAM,CAAC;IAEpB,kBAAkB;IAClB,UAAU,CAAC,EAAE,MAAM,CAAC;IAEpB,2BAA2B;IAC3B,QAAQ,CAAC,EAAE,MAAM,CAAC;IAElB,4CAA4C;IAC5C,eAAe,CAAC,EAAE,gBAAgB,EAAE,CAAC;IAErC,6CAA6C;IAC7C,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB;AAED,MAAM,WAAW,YAAY;IAC3B,0BAA0B;IAC1B,KAAK,EAAE,MAAM,CAAC;IAEd,oCAAoC;IACpC,MAAM,CAAC,EAAE,SAAS,GAAG,MAAM,GAAG,QAAQ,GAAG,OAAO,GAAG,UAAU,CAAC;IAE9D,oCAAoC;IACpC,QAAQ,CAAC,EAAE,MAAM,EAAE,CAAC;IAEpB,yBAAyB;IACzB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAElB,6BAA6B;IAC7B,WAAW,CAAC,EAAE,MAAM,CAAC;IAErB,0BAA0B;IAC1B,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACpC;AAED,MAAM,WAAW,gBAAgB;IAC/B,uCAAuC;IACvC,WAAW,EAAE,MAAM,CAAC;IAEpB,2BAA2B;IAC3B,UAAU,EAAE,MAAM,CAAC;IAEnB,4BAA4B;IAC5B,OAAO,EAAE,gBAAgB,EAAE,CAAC;IAE5B,oCAAoC;IACpC,SAAS,EAAE,OAAO,CAAC;IAEnB,2BAA2B;IAC3B,OAAO,CAAC,EAAE,MAAM,CAAC;IAEjB,gBAAgB;IAChB,MAAM,EAAE,OAAO,CAAC;CACjB;AAED,MAAM,WAAW,WAAW;IAC1B,wBAAwB;IACxB,OAAO,EAAE,gBAAgB,CAAC;IAE1B,0BAA0B;IAC1B,QAAQ,EAAE;QACR,KAAK,EAAE,MAAM,CAAC;QACd,eAAe,EAAE,MAAM,CAAC;QACxB,oBAAoB,EAAE,MAAM,CAAC;QAC7B,gBAAgB,EAAE,MAAM,CAAC;QACzB,WAAW,EAAE,MAAM,CAAC;QACpB,eAAe,EAAE,MAAM,CAAC;QACxB,cAAc,EAAE,MAAM,CAAC;KACxB,CAAC;CACH;AAED,MAAM,WAAW,eAAe;IAC9B,oBAAoB;IACpB,IAAI,EAAE,MAAM,CAAC;IAEb,sCAAsC;IACtC,MAAM,EAAE,MAAM,CAAC;IAEf,sCAAsC;IACtC,SAAS,EAAE,MAAM,EAAE,CAAC;CACrB;AAED,eAAO,MAAM,kBAAkB,EAAE,SAShC,CAAC;AAEF,eAAO,MAAM,wBAAwB,EAAE,eA6BtC,CAAC"}

package/dist/code-intelligence/rag/types.js

@@ -0,0 +1,50 @@
+"use strict";
+/**
+ * Types for RAG (Retrieval-Augmented Generation) Module
+ *
+ * Defines interfaces for context building, query processing,
+ * and response generation.
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.DEFAULT_CONTEXT_TEMPLATE = exports.DEFAULT_RAG_CONFIG = void 0;
+exports.DEFAULT_RAG_CONFIG = {
+    maxContextTokens: 4096,
+    topK: 10,
+    minRelevanceScore: 0.5,
+    enableGraphExpansion: true,
+    graphExpansionDepth: 2,
+    enableCaching: true,
+    cacheTtlMs: 5 * 60 * 1000, // 5 minutes
+    summarizationStrategy: 'tree',
+};
+exports.DEFAULT_CONTEXT_TEMPLATE = {
+    name: 'default',
+    format: `# Relevant Code Context
+
+{{#each sources}}
+## {{filePath}} (lines {{startLine}}-{{endLine}})
+{{#if entityType}}Type: {{entityType}}{{/if}}
+{{#if entityName}}Name: {{entityName}}{{/if}}
+
+\`\`\`{{language}}
+{{content}}
+\`\`\`
+
+{{#if relatedContexts}}
+### Related Code:
+{{#each relatedContexts}}
+- **{{relationship}}**: {{filePath}} ({{entityName}})
+{{/each}}
+{{/if}}
+
+---
+{{/each}}
+
+{{#if summary}}
+## Summary
+{{summary}}
+{{/if}}
+`,
+    variables: ['sources', 'summary', 'query'],
+};
+//# sourceMappingURL=types.js.map

package/dist/code-intelligence/rag/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../../src/code-intelligence/rag/types.ts"],"names":[],"mappings":";AAAA;;;;;GAKG;;;AAuJU,QAAA,kBAAkB,GAAc;IAC3C,gBAAgB,EAAE,IAAI;IACtB,IAAI,EAAE,EAAE;IACR,iBAAiB,EAAE,GAAG;IACtB,oBAAoB,EAAE,IAAI;IAC1B,mBAAmB,EAAE,CAAC;IACtB,aAAa,EAAE,IAAI;IACnB,UAAU,EAAE,CAAC,GAAG,EAAE,GAAG,IAAI,EAAE,YAAY;IACvC,qBAAqB,EAAE,MAAM;CAC9B,CAAC;AAEW,QAAA,wBAAwB,GAAoB;IACvD,IAAI,EAAE,SAAS;IACf,MAAM,EAAE;;;;;;;;;;;;;;;;;;;;;;;;;CAyBT;IACC,SAAS,EAAE,CAAC,SAAS,EAAE,SAAS,EAAE,OAAO,CAAC;CAC3C,CAAC"}

package/dist/code-intelligence/retrieval/SemanticRetriever.d.ts

@@ -0,0 +1,62 @@
+/**
+ * Semantic Code Retriever
+ *
+ * Performs similarity-based code search using embeddings.
+ * Optimized for AST-chunked code with configurable top-k retrieval.
+ *
+ * Key Features:
+ * - Top-k retrieval (default: 5 chunks)
+ * - Similarity threshold filtering
+ * - Per-file deduplication
+ * - Context expansion
+ */
+import { NomicEmbedder } from '../embeddings/NomicEmbedder.js';
+import { RetrievalConfig, RetrievalResponse, StoredChunk } from './types.js';
+export declare class SemanticRetriever {
+    private embedder;
+    private config;
+    private chunks;
+    constructor(embedder: NomicEmbedder, config?: Partial<RetrievalConfig>);
+    /**
+     * Add chunks to the search index.
+     */
+    addChunks(chunks: StoredChunk[]): void;
+    /**
+     * Clear all indexed chunks.
+     */
+    clearIndex(): void;
+    /**
+     * Get current index size.
+     */
+    getIndexSize(): number;
+    /**
+     * Retrieve relevant code chunks for a query.
+     *
+     * @param query - Natural language or code query
+     * @param configOverride - Optional config override for this query
+     * @returns Retrieval response with ranked results
+     */
+    retrieve(query: string, configOverride?: Partial<RetrievalConfig>): Promise<RetrievalResponse>;
+    /**
+     * Retrieve and merge results into a single context string.
+     * Useful for providing context to LLMs.
+     */
+    retrieveAsContext(query: string, configOverride?: Partial<RetrievalConfig>): Promise<string>;
+    /**
+     * Get configuration.
+     */
+    getConfig(): RetrievalConfig;
+    /**
+     * Update configuration.
+     */
+    updateConfig(config: Partial<RetrievalConfig>): void;
+    /**
+     * Calculate cosine similarity between two vectors.
+     */
+    private cosineSimilarity;
+    /**
+     * Apply filtering, deduplication, and limits.
+     */
+    private applyFilters;
+}
+//# sourceMappingURL=SemanticRetriever.d.ts.map

package/dist/code-intelligence/retrieval/SemanticRetriever.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"SemanticRetriever.d.ts","sourceRoot":"","sources":["../../../src/code-intelligence/retrieval/SemanticRetriever.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;GAWG;AAEH,OAAO,EAAE,aAAa,EAAE,MAAM,gCAAgC,CAAC;AAC/D,OAAO,EACL,eAAe,EAEf,iBAAiB,EAEjB,WAAW,EAEZ,MAAM,YAAY,CAAC;AAEpB,qBAAa,iBAAiB;IAC5B,OAAO,CAAC,QAAQ,CAAgB;IAChC,OAAO,CAAC,MAAM,CAAkB;IAChC,OAAO,CAAC,MAAM,CAAuC;gBAGnD,QAAQ,EAAE,aAAa,EACvB,MAAM,GAAE,OAAO,CAAC,eAAe,CAAM;IAMvC;;OAEG;IACH,SAAS,CAAC,MAAM,EAAE,WAAW,EAAE,GAAG,IAAI;IAMtC;;OAEG;IACH,UAAU,IAAI,IAAI;IAIlB;;OAEG;IACH,YAAY,IAAI,MAAM;IAItB;;;;;;OAMG;IACG,QAAQ,CACZ,KAAK,EAAE,MAAM,EACb,cAAc,CAAC,EAAE,OAAO,CAAC,eAAe,CAAC,GACxC,OAAO,CAAC,iBAAiB,CAAC;IA2D7B;;;OAGG;IACG,iBAAiB,CACrB,KAAK,EAAE,MAAM,EACb,cAAc,CAAC,EAAE,OAAO,CAAC,eAAe,CAAC,GACxC,OAAO,CAAC,MAAM,CAAC;IAgBlB;;OAEG;IACH,SAAS,IAAI,eAAe;IAI5B;;OAEG;IACH,YAAY,CAAC,MAAM,EAAE,OAAO,CAAC,eAAe,CAAC,GAAG,IAAI;IAIpD;;OAEG;IACH,OAAO,CAAC,gBAAgB;IAuBxB;;OAEG;IACH,OAAO,CAAC,YAAY;CA6BrB"}

package/dist/code-intelligence/retrieval/SemanticRetriever.js

@@ -0,0 +1,174 @@
+"use strict";
+/**
+ * Semantic Code Retriever
+ *
+ * Performs similarity-based code search using embeddings.
+ * Optimized for AST-chunked code with configurable top-k retrieval.
+ *
+ * Key Features:
+ * - Top-k retrieval (default: 5 chunks)
+ * - Similarity threshold filtering
+ * - Per-file deduplication
+ * - Context expansion
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.SemanticRetriever = void 0;
+const types_js_1 = require("./types.js");
+class SemanticRetriever {
+    constructor(embedder, config = {}) {
+        this.chunks = new Map();
+        this.embedder = embedder;
+        this.config = { ...types_js_1.DEFAULT_RETRIEVAL_CONFIG, ...config };
+    }
+    /**
+     * Add chunks to the search index.
+     */
+    addChunks(chunks) {
+        for (const chunk of chunks) {
+            this.chunks.set(chunk.id, chunk);
+        }
+    }
+    /**
+     * Clear all indexed chunks.
+     */
+    clearIndex() {
+        this.chunks.clear();
+    }
+    /**
+     * Get current index size.
+     */
+    getIndexSize() {
+        return this.chunks.size;
+    }
+    /**
+     * Retrieve relevant code chunks for a query.
+     *
+     * @param query - Natural language or code query
+     * @param configOverride - Optional config override for this query
+     * @returns Retrieval response with ranked results
+     */
+    async retrieve(query, configOverride) {
+        const startTime = Date.now();
+        const config = { ...this.config, ...configOverride };
+        // Generate query embedding
+        const queryEmbedding = await this.embedder.embed(query);
+        // Calculate similarities for all chunks
+        const similarities = [];
+        for (const chunk of this.chunks.values()) {
+            const similarity = this.cosineSimilarity(queryEmbedding, chunk.embedding);
+            if (similarity >= config.minSimilarity) {
+                similarities.push({ chunk, similarity });
+            }
+        }
+        // Sort by similarity (descending)
+        similarities.sort((a, b) => b.similarity - a.similarity);
+        // Apply deduplication and limits
+        const results = this.applyFilters(similarities, config);
+        // Convert to results format
+        const retrievalResults = results.map(({ chunk, similarity }) => ({
+            chunkId: chunk.id,
+            filePath: chunk.filePath,
+            content: chunk.content,
+            startLine: chunk.startLine,
+            endLine: chunk.endLine,
+            similarity,
+            entityType: chunk.entityType,
+            entityName: chunk.entityName,
+            language: chunk.language,
+            metadata: chunk.metadata,
+        }));
+        // Calculate statistics
+        const uniqueFiles = new Set(retrievalResults.map(r => r.filePath)).size;
+        const avgSimilarity = retrievalResults.length > 0
+            ? retrievalResults.reduce((sum, r) => sum + r.similarity, 0) / retrievalResults.length
+            : 0;
+        const stats = {
+            totalChunksSearched: this.chunks.size,
+            chunksAboveThreshold: similarities.length,
+            chunksReturned: retrievalResults.length,
+            uniqueFiles,
+            avgSimilarity,
+        };
+        return {
+            results: retrievalResults,
+            query,
+            retrievalTimeMs: Date.now() - startTime,
+            stats,
+        };
+    }
+    /**
+     * Retrieve and merge results into a single context string.
+     * Useful for providing context to LLMs.
+     */
+    async retrieveAsContext(query, configOverride) {
+        const response = await this.retrieve(query, configOverride);
+        if (response.results.length === 0) {
+            return '';
+        }
+        // Format results as context
+        const contextParts = response.results.map((result, index) => {
+            const header = `// [${index + 1}] ${result.filePath}:${result.startLine}-${result.endLine} (${result.entityType}${result.entityName ? `: ${result.entityName}` : ''})`;
+            return `${header}\n${result.content}`;
+        });
+        return contextParts.join('\n\n');
+    }
+    /**
+     * Get configuration.
+     */
+    getConfig() {
+        return { ...this.config };
+    }
+    /**
+     * Update configuration.
+     */
+    updateConfig(config) {
+        this.config = { ...this.config, ...config };
+    }
+    /**
+     * Calculate cosine similarity between two vectors.
+     */
+    cosineSimilarity(a, b) {
+        if (a.length !== b.length) {
+            throw new Error(`Vector length mismatch: ${a.length} vs ${b.length}`);
+        }
+        let dotProduct = 0;
+        let normA = 0;
+        let normB = 0;
+        for (let i = 0; i < a.length; i++) {
+            dotProduct += a[i] * b[i];
+            normA += a[i] * a[i];
+            normB += b[i] * b[i];
+        }
+        const denominator = Math.sqrt(normA) * Math.sqrt(normB);
+        if (denominator === 0) {
+            return 0;
+        }
+        return dotProduct / denominator;
+    }
+    /**
+     * Apply filtering, deduplication, and limits.
+     */
+    applyFilters(similarities, config) {
+        if (!config.deduplicateByFile) {
+            // Simple case: just take top-k
+            return similarities.slice(0, config.topK);
+        }
+        // Track chunks per file
+        const fileChunkCounts = new Map();
+        const results = [];
+        for (const item of similarities) {
+            if (results.length >= config.topK) {
+                break;
+            }
+            const filePath = item.chunk.filePath;
+            const currentCount = fileChunkCounts.get(filePath) || 0;
+            if (currentCount < config.maxChunksPerFile) {
+                results.push(item);
+                fileChunkCounts.set(filePath, currentCount + 1);
+            }
+        }
+        return results;
+    }
+}
+exports.SemanticRetriever = SemanticRetriever;
+//# sourceMappingURL=SemanticRetriever.js.map

package/dist/code-intelligence/retrieval/SemanticRetriever.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"SemanticRetriever.js","sourceRoot":"","sources":["../../../src/code-intelligence/retrieval/SemanticRetriever.ts"],"names":[],"mappings":";AAAA;;;;;;;;;;;GAWG;;;AAGH,yCAOoB;AAEpB,MAAa,iBAAiB;IAK5B,YACE,QAAuB,EACvB,SAAmC,EAAE;QAJ/B,WAAM,GAA6B,IAAI,GAAG,EAAE,CAAC;QAMnD,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;QACzB,IAAI,CAAC,MAAM,GAAG,EAAE,GAAG,mCAAwB,EAAE,GAAG,MAAM,EAAE,CAAC;IAC3D,CAAC;IAED;;OAEG;IACH,SAAS,CAAC,MAAqB;QAC7B,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;YAC3B,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC;QACnC,CAAC;IACH,CAAC;IAED;;OAEG;IACH,UAAU;QACR,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;IACtB,CAAC;IAED;;OAEG;IACH,YAAY;QACV,OAAO,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC;IAC1B,CAAC;IAED;;;;;;OAMG;IACH,KAAK,CAAC,QAAQ,CACZ,KAAa,EACb,cAAyC;QAEzC,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC7B,MAAM,MAAM,GAAG,EAAE,GAAG,IAAI,CAAC,MAAM,EAAE,GAAG,cAAc,EAAE,CAAC;QAErD,2BAA2B;QAC3B,MAAM,cAAc,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;QAExD,wCAAwC;QACxC,MAAM,YAAY,GAAsD,EAAE,CAAC;QAE3E,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,EAAE,CAAC;YACzC,MAAM,UAAU,GAAG,IAAI,CAAC,gBAAgB,CAAC,cAAc,EAAE,KAAK,CAAC,SAAS,CAAC,CAAC;YAC1E,IAAI,UAAU,IAAI,MAAM,CAAC,aAAa,EAAE,CAAC;gBACvC,YAAY,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC,CAAC;YAC3C,CAAC;QACH,CAAC;QAED,kCAAkC;QAClC,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,GAAG,CAAC,CAAC,UAAU,CAAC,CAAC;QAEzD,iCAAiC;QACjC,MAAM,OAAO,GAAG,IAAI,CAAC,YAAY,CAAC,YAAY,EAAE,MAAM,CAAC,CAAC;QAExD,4BAA4B;QAC5B,MAAM,gBAAgB,GAAsB,OAAO,CAAC,GAAG,CAAC,CAAC,EAAE,KAAK,EAAE,UAAU,EAAE,EAAE,EAAE,CAAC,CAAC;YAClF,OAAO,EAAE,KAAK,CAAC,EAAE;YACjB,QAAQ,EAAE,KAAK,CAAC,QAAQ;YACxB,OAAO,EAAE,KAAK,CAAC,OAAO;YACtB,SAAS,EAAE,KAAK,CAAC,SAAS;YAC1B,OAAO,EAAE,KAAK,CAAC,OAAO;YACtB,UAAU;YACV,UAAU,EAAE,KAAK,CAAC,UAAU;YAC5B,UAAU,EAAE,KAAK,CAAC,UAAU;YAC5B,QAAQ,EAAE,KAAK,CAAC,QAAQ;YACxB,QAAQ,EAAE,KAAK,CAAC,QAAQ;SACzB,CAAC,CAAC,CAAC;QAEJ,uBAAuB;QACvB,MAAM,WAAW,GAAG,IAAI,GAAG,CAAC,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC;QACxE,MAAM,aAAa,GAAG,gBAAgB,CAAC,MAAM,GAAG,CAAC;YAC/C,CAAC,CAAC,gBAAgB,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,CAAC,UAAU,EAAE,CAAC,CAAC,GAAG,gBAAgB,CAAC,MAAM;YACtF,CAAC,CAAC,CAAC,CAAC;QAEN,MAAM,KAAK,GAAmB;YAC5B,mBAAmB,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI;YACrC,oBAAoB,EAAE,YAAY,CAAC,MAAM;YACzC,cAAc,EAAE,gBAAgB,CAAC,MAAM;YACvC,WAAW;YACX,aAAa;SACd,CAAC;QAEF,OAAO;YACL,OAAO,EAAE,gBAAgB;YACzB,KAAK;YACL,eAAe,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;YACvC,KAAK;SACN,CAAC;IACJ,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,iBAAiB,CACrB,KAAa,EACb,cAAyC;QAEzC,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,cAAc,CAAC,CAAC;QAE5D,IAAI,QAAQ,CAAC,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAClC,OAAO,EAAE,CAAC;QACZ,CAAC;QAED,4BAA4B;QAC5B,MAAM,YAAY,GAAG,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,KAAK,EAAE,EAAE;YAC1D,MAAM,MAAM,GAAG,OAAO,KAAK,GAAG,CAAC,KAAK,MAAM,CAAC,QAAQ,IAAI,MAAM,CAAC,SAAS,IAAI,MAAM,CAAC,OAAO,KAAK,MAAM,CAAC,UAAU,GAAG,MAAM,CAAC,UAAU,CAAC,CAAC,CAAC,KAAK,MAAM,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,GAAG,CAAC;YACvK,OAAO,GAAG,MAAM,KAAK,MAAM,CAAC,OAAO,EAAE,CAAC;QACxC,CAAC,CAAC,CAAC;QAEH,OAAO,YAAY,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IACnC,CAAC;IAED;;OAEG;IACH,SAAS;QACP,OAAO,EAAE,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC;IAC5B,CAAC;IAED;;OAEG;IACH,YAAY,CAAC,MAAgC;QAC3C,IAAI,CAAC,MAAM,GAAG,EAAE,GAAG,IAAI,CAAC,MAAM,EAAE,GAAG,MAAM,EAAE,CAAC;IAC9C,CAAC;IAED;;OAEG;IACK,gBAAgB,CAAC,CAAW,EAAE,CAAW;QAC/C,IAAI,CAAC,CAAC,MAAM,KAAK,CAAC,CAAC,MAAM,EAAE,CAAC;YAC1B,MAAM,IAAI,KAAK,CAAC,2BAA2B,CAAC,CAAC,MAAM,OAAO,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC;QACxE,CAAC;QAED,IAAI,UAAU,GAAG,CAAC,CAAC;QACnB,IAAI,KAAK,GAAG,CAAC,CAAC;QACd,IAAI,KAAK,GAAG,CAAC,CAAC;QAEd,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YAClC,UAAU,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;YAC1B,KAAK,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;YACrB,KAAK,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;QACvB,CAAC;QAED,MAAM,WAAW,GAAG,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACxD,IAAI,WAAW,KAAK,CAAC,EAAE,CAAC;YACtB,OAAO,CAAC,CAAC;QACX,CAAC;QAED,OAAO,UAAU,GAAG,WAAW,CAAC;IAClC,CAAC;IAED;;OAEG;IACK,YAAY,CAClB,YAA+D,EAC/D,MAAuB;QAEvB,IAAI,CAAC,MAAM,CAAC,iBAAiB,EAAE,CAAC;YAC9B,+BAA+B;YAC/B,OAAO,YAAY,CAAC,KAAK,CAAC,CAAC,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC;QAC5C,CAAC;QAED,wBAAwB;QACxB,MAAM,eAAe,GAAG,IAAI,GAAG,EAAkB,CAAC;QAClD,MAAM,OAAO,GAAsD,EAAE,CAAC;QAEtE,KAAK,MAAM,IAAI,IAAI,YAAY,EAAE,CAAC;YAChC,IAAI,OAAO,CAAC,MAAM,IAAI,MAAM,CAAC,IAAI,EAAE,CAAC;gBAClC,MAAM;YACR,CAAC;YAED,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC;YACrC,MAAM,YAAY,GAAG,eAAe,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;YAExD,IAAI,YAAY,GAAG,MAAM,CAAC,gBAAgB,EAAE,CAAC;gBAC3C,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBACnB,eAAe,CAAC,GAAG,CAAC,QAAQ,EAAE,YAAY,GAAG,CAAC,CAAC,CAAC;YAClD,CAAC;QACH,CAAC;QAED,OAAO,OAAO,CAAC;IACjB,CAAC;CACF;AAxMD,8CAwMC"}

package/dist/code-intelligence/retrieval/index.d.ts

@@ -0,0 +1,9 @@
+/**
+ * Code Intelligence Retrieval Module
+ *
+ * Semantic search for code using embeddings.
+ * Optimized for AST-chunked code with multi-chunk retrieval.
+ */
+export { SemanticRetriever } from './SemanticRetriever.js';
+export { RetrievalConfig, RetrievalResult, RetrievalResponse, RetrievalStats, StoredChunk, DEFAULT_RETRIEVAL_CONFIG, PRECISION_RETRIEVAL_CONFIG, RECALL_RETRIEVAL_CONFIG, } from './types.js';
+//# sourceMappingURL=index.d.ts.map

package/dist/code-intelligence/retrieval/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/code-intelligence/retrieval/index.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAAE,iBAAiB,EAAE,MAAM,wBAAwB,CAAC;AAC3D,OAAO,EACL,eAAe,EACf,eAAe,EACf,iBAAiB,EACjB,cAAc,EACd,WAAW,EACX,wBAAwB,EACxB,0BAA0B,EAC1B,uBAAuB,GACxB,MAAM,YAAY,CAAC"}

package/dist/code-intelligence/retrieval/index.js

@@ -0,0 +1,16 @@
+"use strict";
+/**
+ * Code Intelligence Retrieval Module
+ *
+ * Semantic search for code using embeddings.
+ * Optimized for AST-chunked code with multi-chunk retrieval.
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.RECALL_RETRIEVAL_CONFIG = exports.PRECISION_RETRIEVAL_CONFIG = exports.DEFAULT_RETRIEVAL_CONFIG = exports.SemanticRetriever = void 0;
+var SemanticRetriever_js_1 = require("./SemanticRetriever.js");
+Object.defineProperty(exports, "SemanticRetriever", { enumerable: true, get: function () { return SemanticRetriever_js_1.SemanticRetriever; } });
+var types_js_1 = require("./types.js");
+Object.defineProperty(exports, "DEFAULT_RETRIEVAL_CONFIG", { enumerable: true, get: function () { return types_js_1.DEFAULT_RETRIEVAL_CONFIG; } });
+Object.defineProperty(exports, "PRECISION_RETRIEVAL_CONFIG", { enumerable: true, get: function () { return types_js_1.PRECISION_RETRIEVAL_CONFIG; } });
+Object.defineProperty(exports, "RECALL_RETRIEVAL_CONFIG", { enumerable: true, get: function () { return types_js_1.RECALL_RETRIEVAL_CONFIG; } });
+//# sourceMappingURL=index.js.map

package/dist/code-intelligence/retrieval/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/code-intelligence/retrieval/index.ts"],"names":[],"mappings":";AAAA;;;;;GAKG;;;AAEH,+DAA2D;AAAlD,yHAAA,iBAAiB,OAAA;AAC1B,uCASoB;AAHlB,oHAAA,wBAAwB,OAAA;AACxB,sHAAA,0BAA0B,OAAA;AAC1B,mHAAA,uBAAuB,OAAA"}

package/dist/code-intelligence/retrieval/types.d.ts

@@ -0,0 +1,118 @@
+/**
+ * Types for Code Intelligence Retrieval System
+ *
+ * Supports multi-chunk retrieval with configurable top-k
+ * to balance semantic precision with context completeness.
+ */
+export interface RetrievalConfig {
+    /**
+     * Number of top chunks to retrieve per query.
+     * Higher values = more context but potentially less precision.
+     * Recommended: 3-5 for AST-chunked code (smaller semantic units)
+     * Default: 5
+     */
+    topK: number;
+    /**
+     * Minimum similarity score threshold (0-1).
+     * Chunks below this score are filtered out.
+     * Default: 0.5
+     */
+    minSimilarity: number;
+    /**
+     * Whether to deduplicate results from the same file.
+     * When true, consecutive chunks from same file are merged.
+     * Default: true
+     */
+    deduplicateByFile: boolean;
+    /**
+     * Maximum number of chunks to return from same file.
+     * Prevents single large file from dominating results.
+     * Default: 3
+     */
+    maxChunksPerFile: number;
+    /**
+     * Whether to include surrounding context for each chunk.
+     * Expands results to include adjacent chunks.
+     * Default: false
+     */
+    includeContext: boolean;
+    /**
+     * Number of chunks before/after to include as context.
+     * Only used when includeContext is true.
+     * Default: 1
+     */
+    contextWindow: number;
+}
+export interface RetrievalResult {
+    /** Unique chunk identifier */
+    chunkId: string;
+    /** File path containing this chunk */
+    filePath: string;
+    /** Chunk content */
+    content: string;
+    /** Start line in original file */
+    startLine: number;
+    /** End line in original file */
+    endLine: number;
+    /** Cosine similarity score (0-1) */
+    similarity: number;
+    /** Entity type (function, class, method, etc.) */
+    entityType: string;
+    /** Entity name if available */
+    entityName?: string;
+    /** Language of the code */
+    language: string;
+    /** Additional metadata */
+    metadata?: Record<string, unknown>;
+}
+export interface RetrievalResponse {
+    /** Retrieved chunks, ordered by relevance */
+    results: RetrievalResult[];
+    /** Query that was searched */
+    query: string;
+    /** Time taken for retrieval in milliseconds */
+    retrievalTimeMs: number;
+    /** Statistics about the retrieval */
+    stats: RetrievalStats;
+}
+export interface RetrievalStats {
+    /** Total chunks searched */
+    totalChunksSearched: number;
+    /** Chunks that passed similarity threshold */
+    chunksAboveThreshold: number;
+    /** Final chunks returned (after dedup/limits) */
+    chunksReturned: number;
+    /** Unique files in results */
+    uniqueFiles: number;
+    /** Average similarity of returned chunks */
+    avgSimilarity: number;
+}
+export interface StoredChunk {
+    id: string;
+    fileId: string;
+    filePath: string;
+    content: string;
+    startLine: number;
+    endLine: number;
+    entityType: string;
+    entityName?: string;
+    language: string;
+    embedding: number[];
+    metadata?: Record<string, unknown>;
+}
+/**
+ * Default retrieval configuration optimized for AST-chunked code.
+ * Uses top-k=5 to compensate for smaller semantic chunks.
+ */
+export declare const DEFAULT_RETRIEVAL_CONFIG: RetrievalConfig;
+/**
+ * High-precision configuration for specific queries.
+ * Use when you need exact matches over broad context.
+ */
+export declare const PRECISION_RETRIEVAL_CONFIG: RetrievalConfig;
+/**
+ * High-recall configuration for exploratory queries.
+ * Use when you want to find all potentially relevant code.
+ */
+export declare const RECALL_RETRIEVAL_CONFIG: RetrievalConfig;
+//# sourceMappingURL=types.d.ts.map

package/dist/code-intelligence/retrieval/types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../../src/code-intelligence/retrieval/types.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,MAAM,WAAW,eAAe;IAC9B;;;;;OAKG;IACH,IAAI,EAAE,MAAM,CAAC;IAEb;;;;OAIG;IACH,aAAa,EAAE,MAAM,CAAC;IAEtB;;;;OAIG;IACH,iBAAiB,EAAE,OAAO,CAAC;IAE3B;;;;OAIG;IACH,gBAAgB,EAAE,MAAM,CAAC;IAEzB;;;;OAIG;IACH,cAAc,EAAE,OAAO,CAAC;IAExB;;;;OAIG;IACH,aAAa,EAAE,MAAM,CAAC;CACvB;AAED,MAAM,WAAW,eAAe;IAC9B,8BAA8B;IAC9B,OAAO,EAAE,MAAM,CAAC;IAEhB,sCAAsC;IACtC,QAAQ,EAAE,MAAM,CAAC;IAEjB,oBAAoB;IACpB,OAAO,EAAE,MAAM,CAAC;IAEhB,kCAAkC;IAClC,SAAS,EAAE,MAAM,CAAC;IAElB,gCAAgC;IAChC,OAAO,EAAE,MAAM,CAAC;IAEhB,oCAAoC;IACpC,UAAU,EAAE,MAAM,CAAC;IAEnB,kDAAkD;IAClD,UAAU,EAAE,MAAM,CAAC;IAEnB,+BAA+B;IAC/B,UAAU,CAAC,EAAE,MAAM,CAAC;IAEpB,2BAA2B;IAC3B,QAAQ,EAAE,MAAM,CAAC;IAEjB,0BAA0B;IAC1B,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACpC;AAED,MAAM,WAAW,iBAAiB;IAChC,6CAA6C;IAC7C,OAAO,EAAE,eAAe,EAAE,CAAC;IAE3B,8BAA8B;IAC9B,KAAK,EAAE,MAAM,CAAC;IAEd,+CAA+C;IAC/C,eAAe,EAAE,MAAM,CAAC;IAExB,qCAAqC;IACrC,KAAK,EAAE,cAAc,CAAC;CACvB;AAED,MAAM,WAAW,cAAc;IAC7B,4BAA4B;IAC5B,mBAAmB,EAAE,MAAM,CAAC;IAE5B,8CAA8C;IAC9C,oBAAoB,EAAE,MAAM,CAAC;IAE7B,iDAAiD;IACjD,cAAc,EAAE,MAAM,CAAC;IAEvB,8BAA8B;IAC9B,WAAW,EAAE,MAAM,CAAC;IAEpB,4CAA4C;IAC5C,aAAa,EAAE,MAAM,CAAC;CACvB;AAED,MAAM,WAAW,WAAW;IAC1B,EAAE,EAAE,MAAM,CAAC;IACX,MAAM,EAAE,MAAM,CAAC;IACf,QAAQ,EAAE,MAAM,CAAC;IACjB,OAAO,EAAE,MAAM,CAAC;IAChB,SAAS,EAAE,MAAM,CAAC;IAClB,OAAO,EAAE,MAAM,CAAC;IAChB,UAAU,EAAE,MAAM,CAAC;IACnB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,QAAQ,EAAE,MAAM,CAAC;IACjB,SAAS,EAAE,MAAM,EAAE,CAAC;IACpB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACpC;AAED;;;GAGG;AACH,eAAO,MAAM,wBAAwB,EAAE,eAOtC,CAAC;AAEF;;;GAGG;AACH,eAAO,MAAM,0BAA0B,EAAE,eAOxC,CAAC;AAEF;;;GAGG;AACH,eAAO,MAAM,uBAAuB,EAAE,eAOrC,CAAC"}

package/dist/code-intelligence/retrieval/types.js

@@ -0,0 +1,46 @@
+"use strict";
+/**
+ * Types for Code Intelligence Retrieval System
+ *
+ * Supports multi-chunk retrieval with configurable top-k
+ * to balance semantic precision with context completeness.
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.RECALL_RETRIEVAL_CONFIG = exports.PRECISION_RETRIEVAL_CONFIG = exports.DEFAULT_RETRIEVAL_CONFIG = void 0;
+/**
+ * Default retrieval configuration optimized for AST-chunked code.
+ * Uses top-k=5 to compensate for smaller semantic chunks.
+ */
+exports.DEFAULT_RETRIEVAL_CONFIG = {
+    topK: 5, // Retrieve 5 chunks (recommended for AST chunking)
+    minSimilarity: 0.5, // 50% similarity threshold
+    deduplicateByFile: true,
+    maxChunksPerFile: 3,
+    includeContext: false,
+    contextWindow: 1,
+};
+/**
+ * High-precision configuration for specific queries.
+ * Use when you need exact matches over broad context.
+ */
+exports.PRECISION_RETRIEVAL_CONFIG = {
+    topK: 3,
+    minSimilarity: 0.7, // Higher threshold
+    deduplicateByFile: true,
+    maxChunksPerFile: 2,
+    includeContext: false,
+    contextWindow: 0,
+};
+/**
+ * High-recall configuration for exploratory queries.
+ * Use when you want to find all potentially relevant code.
+ */
+exports.RECALL_RETRIEVAL_CONFIG = {
+    topK: 10, // More results
+    minSimilarity: 0.4, // Lower threshold
+    deduplicateByFile: false,
+    maxChunksPerFile: 5,
+    includeContext: true,
+    contextWindow: 2,
+};
+//# sourceMappingURL=types.js.map

package/dist/code-intelligence/retrieval/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../../src/code-intelligence/retrieval/types.ts"],"names":[],"mappings":";AAAA;;;;;GAKG;;;AA4HH;;;GAGG;AACU,QAAA,wBAAwB,GAAoB;IACvD,IAAI,EAAE,CAAC,EAAe,mDAAmD;IACzE,aAAa,EAAE,GAAG,EAAI,2BAA2B;IACjD,iBAAiB,EAAE,IAAI;IACvB,gBAAgB,EAAE,CAAC;IACnB,cAAc,EAAE,KAAK;IACrB,aAAa,EAAE,CAAC;CACjB,CAAC;AAEF;;;GAGG;AACU,QAAA,0BAA0B,GAAoB;IACzD,IAAI,EAAE,CAAC;IACP,aAAa,EAAE,GAAG,EAAI,mBAAmB;IACzC,iBAAiB,EAAE,IAAI;IACvB,gBAAgB,EAAE,CAAC;IACnB,cAAc,EAAE,KAAK;IACrB,aAAa,EAAE,CAAC;CACjB,CAAC;AAEF;;;GAGG;AACU,QAAA,uBAAuB,GAAoB;IACtD,IAAI,EAAE,EAAE,EAAc,eAAe;IACrC,aAAa,EAAE,GAAG,EAAI,kBAAkB;IACxC,iBAAiB,EAAE,KAAK;IACxB,gBAAgB,EAAE,CAAC;IACnB,cAAc,EAAE,IAAI;IACpB,aAAa,EAAE,CAAC;CACjB,CAAC"}