driftdetect-cortex 0.9.41

package/dist/orchestrators/retrieval-orchestrator.js

@@ -0,0 +1,319 @@
+/**
+ * Retrieval Orchestrator
+ *
+ * Orchestrates memory retrieval with compression, session
+ * deduplication, and prediction caching.
+ *
+ * @module orchestrators/retrieval-orchestrator
+ */
+/**
+ * Retrieval Orchestrator
+ *
+ * Coordinates retrieval, compression, session tracking,
+ * and prediction caching.
+ */
+export class RetrievalOrchestrator {
+    storage;
+    compressor;
+    budgetManager;
+    sessionManager;
+    predictionCache;
+    constructor(storage, options) {
+        this.storage = storage;
+        this.compressor = options?.compressor ?? null;
+        this.budgetManager = options?.budgetManager ?? null;
+        this.sessionManager = options?.sessionManager ?? null;
+        this.predictionCache = options?.predictionCache ?? null;
+    }
+    /**
+     * Retrieve memories with full orchestration
+     */
+    async retrieve(context) {
+        const startTime = Date.now();
+        const activeSession = await this.sessionManager?.getActiveSession();
+        const sessionId = activeSession?.id ?? 'default';
+        // Step 1: Gather candidates
+        const candidates = await this.gatherCandidates(context);
+        // Step 2: Score and rank
+        const scored = await this.scoreAndRank(candidates, context);
+        // Step 3: Compress and fit to budget
+        const budget = context.maxTokens ?? 2000;
+        const compressed = await this.compressAndFit(scored, budget);
+        // Step 4: Deduplicate with session
+        const { deduplicated, deduplicatedCount } = await this.deduplicateWithSession(compressed, sessionId);
+        // Step 5: Track in session
+        const tokensUsed = this.estimateTokens(deduplicated);
+        if (this.sessionManager && activeSession) {
+            for (const mem of deduplicated) {
+                await this.sessionManager.recordMemoryLoaded(sessionId, mem.compressed.memoryId, mem.compressed.tokenCount);
+            }
+        }
+        // Calculate prediction stats
+        const predictionStats = this.calculatePredictionStats(candidates);
+        return {
+            memories: deduplicated,
+            totalCandidates: candidates.length,
+            tokensUsed,
+            retrievalTime: Date.now() - startTime,
+            session: {
+                deduplicatedCount,
+                sessionId,
+                totalTokensSent: activeSession?.tokensSent ?? tokensUsed,
+            },
+            prediction: predictionStats,
+        };
+    }
+    /**
+     * Get the budget manager (for external use)
+     */
+    getBudgetManager() {
+        return this.budgetManager;
+    }
+    /**
+     * Gather candidate memories
+     */
+    async gatherCandidates(context) {
+        const candidates = [];
+        const seen = new Set();
+        // Get from active file
+        if (context.activeFile) {
+            const fileMemories = await this.storage.findByFile(context.activeFile);
+            for (const mem of fileMemories) {
+                if (!seen.has(mem.id)) {
+                    seen.add(mem.id);
+                    candidates.push(mem);
+                }
+            }
+        }
+        // Get from relevant patterns
+        if (context.relevantPatterns) {
+            for (const patternId of context.relevantPatterns) {
+                const patternMemories = await this.storage.findByPattern(patternId);
+                for (const mem of patternMemories) {
+                    if (!seen.has(mem.id)) {
+                        seen.add(mem.id);
+                        candidates.push(mem);
+                    }
+                }
+            }
+        }
+        // Get from prediction cache
+        if (this.predictionCache && context.activeFile) {
+            const predicted = await this.predictionCache.getForFile(context.activeFile);
+            if (predicted) {
+                for (const pred of predicted) {
+                    const mem = await this.storage.read(pred.memoryId);
+                    if (mem && !seen.has(mem.id)) {
+                        seen.add(mem.id);
+                        candidates.push(mem);
+                    }
+                }
+            }
+        }
+        // Get by intent-based search
+        const intentTypes = this.getTypesForIntent(context.intent);
+        for (const type of intentTypes) {
+            const typeMemories = await this.storage.findByType(type, { limit: 20 });
+            for (const mem of typeMemories) {
+                if (!seen.has(mem.id)) {
+                    seen.add(mem.id);
+                    candidates.push(mem);
+                }
+            }
+        }
+        return candidates;
+    }
+    /**
+     * Score and rank candidates
+     */
+    async scoreAndRank(candidates, context) {
+        const scored = [];
+        for (const memory of candidates) {
+            const score = this.calculateScore(memory, context);
+            scored.push({
+                memory,
+                score,
+                relevanceReason: this.getRelevanceReason(memory, context),
+            });
+        }
+        // Sort by score descending
+        scored.sort((a, b) => b.score - a.score);
+        return scored;
+    }
+    /**
+     * Calculate relevance score for a memory
+     */
+    calculateScore(memory, context) {
+        let score = memory.confidence;
+        // Boost for matching intent
+        const intentTypes = this.getTypesForIntent(context.intent);
+        if (intentTypes.includes(memory.type)) {
+            score += 0.2;
+        }
+        // Boost for linked to active file
+        if (context.activeFile && memory.linkedFiles?.includes(context.activeFile)) {
+            score += 0.3;
+        }
+        // Boost for linked to relevant patterns
+        if (context.relevantPatterns && memory.linkedPatterns) {
+            const overlap = memory.linkedPatterns.filter(p => context.relevantPatterns.includes(p));
+            score += overlap.length * 0.1;
+        }
+        // Boost for high importance
+        if (memory.importance === 'critical') {
+            score += 0.2;
+        }
+        else if (memory.importance === 'high') {
+            score += 0.1;
+        }
+        // Boost for recent access
+        if (memory.accessCount > 10) {
+            score += 0.1;
+        }
+        return Math.min(score, 1.0);
+    }
+    /**
+     * Get relevance reason for a memory
+     */
+    getRelevanceReason(memory, context) {
+        if (context.activeFile && memory.linkedFiles?.includes(context.activeFile)) {
+            return 'Linked to active file';
+        }
+        if (context.relevantPatterns && memory.linkedPatterns?.some(p => context.relevantPatterns.includes(p))) {
+            return 'Linked to relevant pattern';
+        }
+        const intentTypes = this.getTypesForIntent(context.intent);
+        if (intentTypes.includes(memory.type)) {
+            return `Relevant for ${context.intent}`;
+        }
+        return 'General relevance';
+    }
+    /**
+     * Compress and fit memories to budget
+     */
+    async compressAndFit(scored, budget) {
+        const result = [];
+        let usedTokens = 0;
+        for (const { memory, score, relevanceReason } of scored) {
+            // Estimate tokens for this memory
+            const memoryTokens = this.estimateMemoryTokens(memory);
+            // Check if it fits
+            if (usedTokens + memoryTokens <= budget) {
+                if (this.compressor) {
+                    const compressed = this.compressor.compress(memory, 3);
+                    result.push({
+                        compressed,
+                        relevanceScore: score,
+                        relevanceReason,
+                    });
+                    usedTokens += compressed.tokenCount;
+                }
+                else {
+                    // No compressor - create a simple compressed memory
+                    result.push({
+                        compressed: {
+                            memoryId: memory.id,
+                            level: 3,
+                            output: { id: memory.id, type: memory.type, importance: memory.importance, tokens: memoryTokens },
+                            tokenCount: memoryTokens,
+                            originalTokenCount: memoryTokens,
+                            compressionRatio: 1,
+                            compressedAt: new Date().toISOString(),
+                        },
+                        relevanceScore: score,
+                        relevanceReason,
+                    });
+                    usedTokens += memoryTokens;
+                }
+            }
+            else if (this.compressor) {
+                // Try to compress to fit
+                const compressed = this.compressor.compressToFit(memory, budget - usedTokens);
+                if (compressed.tokenCount <= budget - usedTokens) {
+                    result.push({
+                        compressed,
+                        relevanceScore: score,
+                        relevanceReason,
+                    });
+                    usedTokens += compressed.tokenCount;
+                }
+            }
+            // Stop if budget exhausted
+            if (usedTokens >= budget * 0.95) {
+                break;
+            }
+        }
+        return result;
+    }
+    /**
+     * Deduplicate with session context
+     */
+    async deduplicateWithSession(memories, sessionId) {
+        if (!this.sessionManager) {
+            return { deduplicated: memories, deduplicatedCount: 0 };
+        }
+        const activeSession = await this.sessionManager.getActiveSession();
+        if (!activeSession || activeSession.id !== sessionId) {
+            return { deduplicated: memories, deduplicatedCount: 0 };
+        }
+        const deduplicated = [];
+        let deduplicatedCount = 0;
+        for (const mem of memories) {
+            if (!activeSession.loadedMemories.has(mem.compressed.memoryId)) {
+                deduplicated.push(mem);
+            }
+            else {
+                deduplicatedCount++;
+            }
+        }
+        return { deduplicated, deduplicatedCount };
+    }
+    /**
+     * Get memory types relevant for an intent
+     */
+    getTypesForIntent(intent) {
+        switch (intent) {
+            case 'add_feature':
+                return ['pattern_rationale', 'tribal', 'procedural', 'constraint_override'];
+            case 'fix_bug':
+                return ['tribal', 'code_smell', 'decision_context'];
+            case 'refactor':
+                return ['pattern_rationale', 'decision_context', 'code_smell'];
+            case 'security_audit':
+                return ['tribal', 'constraint_override', 'code_smell'];
+            case 'understand_code':
+                return ['pattern_rationale', 'decision_context', 'semantic'];
+            case 'add_test':
+                return ['procedural', 'pattern_rationale'];
+            default:
+                return ['tribal', 'pattern_rationale', 'procedural'];
+        }
+    }
+    /**
+     * Estimate tokens for memories
+     */
+    estimateTokens(memories) {
+        return memories.reduce((sum, m) => sum + m.compressed.tokenCount, 0);
+    }
+    /**
+     * Estimate tokens for a single memory
+     */
+    estimateMemoryTokens(memory) {
+        const json = JSON.stringify(memory);
+        return Math.ceil(json.length / 4);
+    }
+    /**
+     * Calculate prediction statistics
+     */
+    calculatePredictionStats(candidates) {
+        if (!this.predictionCache) {
+            return { usedPredictions: false, cacheHitRate: 0 };
+        }
+        const stats = this.predictionCache.getStats();
+        return {
+            usedPredictions: candidates.length > 0,
+            cacheHitRate: stats.hitRate,
+        };
+    }
+}
+//# sourceMappingURL=retrieval-orchestrator.js.map

package/dist/orchestrators/retrieval-orchestrator.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"retrieval-orchestrator.js","sourceRoot":"","sources":["../../src/orchestrators/retrieval-orchestrator.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AA8DH;;;;;GAKG;AACH,MAAM,OAAO,qBAAqB;IACxB,OAAO,CAAiB;IACxB,UAAU,CAAkC;IAC5C,aAAa,CAA8B;IAC3C,cAAc,CAA+B;IAC7C,eAAe,CAAyB;IAEhD,YACE,OAAuB,EACvB,OAKC;QAED,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACvB,IAAI,CAAC,UAAU,GAAG,OAAO,EAAE,UAAU,IAAI,IAAI,CAAC;QAC9C,IAAI,CAAC,aAAa,GAAG,OAAO,EAAE,aAAa,IAAI,IAAI,CAAC;QACpD,IAAI,CAAC,cAAc,GAAG,OAAO,EAAE,cAAc,IAAI,IAAI,CAAC;QACtD,IAAI,CAAC,eAAe,GAAG,OAAO,EAAE,eAAe,IAAI,IAAI,CAAC;IAC1D,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,QAAQ,CAAC,OAAyB;QACtC,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC7B,MAAM,aAAa,GAAG,MAAM,IAAI,CAAC,cAAc,EAAE,gBAAgB,EAAE,CAAC;QACpE,MAAM,SAAS,GAAG,aAAa,EAAE,EAAE,IAAI,SAAS,CAAC;QAEjD,4BAA4B;QAC5B,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC;QAExD,yBAAyB;QACzB,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;QAE5D,qCAAqC;QACrC,MAAM,MAAM,GAAG,OAAO,CAAC,SAAS,IAAI,IAAI,CAAC;QACzC,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QAE7D,mCAAmC;QACnC,MAAM,EAAE,YAAY,EAAE,iBAAiB,EAAE,GAAG,MAAM,IAAI,CAAC,sBAAsB,CAAC,UAAU,EAAE,SAAS,CAAC,CAAC;QAErG,2BAA2B;QAC3B,MAAM,UAAU,GAAG,IAAI,CAAC,cAAc,CAAC,YAAY,CAAC,CAAC;QACrD,IAAI,IAAI,CAAC,cAAc,IAAI,aAAa,EAAE,CAAC;YACzC,KAAK,MAAM,GAAG,IAAI,YAAY,EAAE,CAAC;gBAC/B,MAAM,IAAI,CAAC,cAAc,CAAC,kBAAkB,CAC1C,SAAS,EACT,GAAG,CAAC,UAAU,CAAC,QAAQ,EACvB,GAAG,CAAC,UAAU,CAAC,UAAU,CAC1B,CAAC;YACJ,CAAC;QACH,CAAC;QAED,6BAA6B;QAC7B,MAAM,eAAe,GAAG,IAAI,CAAC,wBAAwB,CAAC,UAAU,CAAC,CAAC;QAElE,OAAO;YACL,QAAQ,EAAE,YAAY;YACtB,eAAe,EAAE,UAAU,CAAC,MAAM;YAClC,UAAU;YACV,aAAa,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;YACrC,OAAO,EAAE;gBACP,iBAAiB;gBACjB,SAAS;gBACT,eAAe,EAAE,aAAa,EAAE,UAAU,IAAI,UAAU;aACzD;YACD,UAAU,EAAE,eAAe;SAC5B,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,gBAAgB;QACd,OAAO,IAAI,CAAC,aAAa,CAAC;IAC5B,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,gBAAgB,CAAC,OAAyB;QACtD,MAAM,UAAU,GAAa,EAAE,CAAC;QAChC,MAAM,IAAI,GAAG,IAAI,GAAG,EAAU,CAAC;QAE/B,uBAAuB;QACvB,IAAI,OAAO,CAAC,UAAU,EAAE,CAAC;YACvB,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;YACvE,KAAK,MAAM,GAAG,IAAI,YAAY,EAAE,CAAC;gBAC/B,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC;oBACtB,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;oBACjB,UAAU,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;gBACvB,CAAC;YACH,CAAC;QACH,CAAC;QAED,6BAA6B;QAC7B,IAAI,OAAO,CAAC,gBAAgB,EAAE,CAAC;YAC7B,KAAK,MAAM,SAAS,IAAI,OAAO,CAAC,gBAAgB,EAAE,CAAC;gBACjD,MAAM,eAAe,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,SAAS,CAAC,CAAC;gBACpE,KAAK,MAAM,GAAG,IAAI,eAAe,EAAE,CAAC;oBAClC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC;wBACtB,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;wBACjB,UAAU,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;oBACvB,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;QAED,4BAA4B;QAC5B,IAAI,IAAI,CAAC,eAAe,IAAI,OAAO,CAAC,UAAU,EAAE,CAAC;YAC/C,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,eAAe,CAAC,UAAU,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;YAC5E,IAAI,SAAS,EAAE,CAAC;gBACd,KAAK,MAAM,IAAI,IAAI,SAAS,EAAE,CAAC;oBAC7B,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;oBACnD,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC;wBAC7B,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;wBACjB,UAAU,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;oBACvB,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;QAED,6BAA6B;QAC7B,MAAM,WAAW,GAAG,IAAI,CAAC,iBAAiB,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QAC3D,KAAK,MAAM,IAAI,IAAI,WAAW,EAAE,CAAC;YAC/B,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,IAAI,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC;YACxE,KAAK,MAAM,GAAG,IAAI,YAAY,EAAE,CAAC;gBAC/B,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC;oBACtB,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;oBACjB,UAAU,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;gBACvB,CAAC;YACH,CAAC;QACH,CAAC;QAED,OAAO,UAAU,CAAC;IACpB,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,YAAY,CAAC,UAAoB,EAAE,OAAyB;QACxE,MAAM,MAAM,GAAmB,EAAE,CAAC;QAElC,KAAK,MAAM,MAAM,IAAI,UAAU,EAAE,CAAC;YAChC,MAAM,KAAK,GAAG,IAAI,CAAC,cAAc,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;YACnD,MAAM,CAAC,IAAI,CAAC;gBACV,MAAM;gBACN,KAAK;gBACL,eAAe,EAAE,IAAI,CAAC,kBAAkB,CAAC,MAAM,EAAE,OAAO,CAAC;aAC1D,CAAC,CAAC;QACL,CAAC;QAED,2BAA2B;QAC3B,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC;QAEzC,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;OAEG;IACK,cAAc,CAAC,MAAc,EAAE,OAAyB;QAC9D,IAAI,KAAK,GAAG,MAAM,CAAC,UAAU,CAAC;QAE9B,4BAA4B;QAC5B,MAAM,WAAW,GAAG,IAAI,CAAC,iBAAiB,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QAC3D,IAAI,WAAW,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC;YACtC,KAAK,IAAI,GAAG,CAAC;QACf,CAAC;QAED,kCAAkC;QAClC,IAAI,OAAO,CAAC,UAAU,IAAI,MAAM,CAAC,WAAW,EAAE,QAAQ,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE,CAAC;YAC3E,KAAK,IAAI,GAAG,CAAC;QACf,CAAC;QAED,wCAAwC;QACxC,IAAI,OAAO,CAAC,gBAAgB,IAAI,MAAM,CAAC,cAAc,EAAE,CAAC;YACtD,MAAM,OAAO,GAAG,MAAM,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,OAAO,CAAC,gBAAiB,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC;YACzF,KAAK,IAAI,OAAO,CAAC,MAAM,GAAG,GAAG,CAAC;QAChC,CAAC;QAED,4BAA4B;QAC5B,IAAI,MAAM,CAAC,UAAU,KAAK,UAAU,EAAE,CAAC;YACrC,KAAK,IAAI,GAAG,CAAC;QACf,CAAC;aAAM,IAAI,MAAM,CAAC,UAAU,KAAK,MAAM,EAAE,CAAC;YACxC,KAAK,IAAI,GAAG,CAAC;QACf,CAAC;QAED,0BAA0B;QAC1B,IAAI,MAAM,CAAC,WAAW,GAAG,EAAE,EAAE,CAAC;YAC5B,KAAK,IAAI,GAAG,CAAC;QACf,CAAC;QAED,OAAO,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC;IAC9B,CAAC;IAED;;OAEG;IACK,kBAAkB,CAAC,MAAc,EAAE,OAAyB;QAClE,IAAI,OAAO,CAAC,UAAU,IAAI,MAAM,CAAC,WAAW,EAAE,QAAQ,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE,CAAC;YAC3E,OAAO,uBAAuB,CAAC;QACjC,CAAC;QACD,IAAI,OAAO,CAAC,gBAAgB,IAAI,MAAM,CAAC,cAAc,EAAE,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,OAAO,CAAC,gBAAiB,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;YACxG,OAAO,4BAA4B,CAAC;QACtC,CAAC;QACD,MAAM,WAAW,GAAG,IAAI,CAAC,iBAAiB,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QAC3D,IAAI,WAAW,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC;YACtC,OAAO,gBAAgB,OAAO,CAAC,MAAM,EAAE,CAAC;QAC1C,CAAC;QACD,OAAO,mBAAmB,CAAC;IAC7B,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,cAAc,CAAC,MAAsB,EAAE,MAAc;QACjE,MAAM,MAAM,GAAsB,EAAE,CAAC;QACrC,IAAI,UAAU,GAAG,CAAC,CAAC;QAEnB,KAAK,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,eAAe,EAAE,IAAI,MAAM,EAAE,CAAC;YACxD,kCAAkC;YAClC,MAAM,YAAY,GAAG,IAAI,CAAC,oBAAoB,CAAC,MAAM,CAAC,CAAC;YAEvD,mBAAmB;YACnB,IAAI,UAAU,GAAG,YAAY,IAAI,MAAM,EAAE,CAAC;gBACxC,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;oBACpB,MAAM,UAAU,GAAG,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;oBACvD,MAAM,CAAC,IAAI,CAAC;wBACV,UAAU;wBACV,cAAc,EAAE,KAAK;wBACrB,eAAe;qBAChB,CAAC,CAAC;oBACH,UAAU,IAAI,UAAU,CAAC,UAAU,CAAC;gBACtC,CAAC;qBAAM,CAAC;oBACN,oDAAoD;oBACpD,MAAM,CAAC,IAAI,CAAC;wBACV,UAAU,EAAE;4BACV,QAAQ,EAAE,MAAM,CAAC,EAAE;4BACnB,KAAK,EAAE,CAAC;4BACR,MAAM,EAAE,EAAE,EAAE,EAAE,MAAM,CAAC,EAAE,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE,UAAU,EAAE,MAAM,CAAC,UAAU,EAAE,MAAM,EAAE,YAAY,EAAE;4BACjG,UAAU,EAAE,YAAY;4BACxB,kBAAkB,EAAE,YAAY;4BAChC,gBAAgB,EAAE,CAAC;4BACnB,YAAY,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;yBACvC;wBACD,cAAc,EAAE,KAAK;wBACrB,eAAe;qBAChB,CAAC,CAAC;oBACH,UAAU,IAAI,YAAY,CAAC;gBAC7B,CAAC;YACH,CAAC;iBAAM,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;gBAC3B,yBAAyB;gBACzB,MAAM,UAAU,GAAG,IAAI,CAAC,UAAU,CAAC,aAAa,CAAC,MAAM,EAAE,MAAM,GAAG,UAAU,CAAC,CAAC;gBAC9E,IAAI,UAAU,CAAC,UAAU,IAAI,MAAM,GAAG,UAAU,EAAE,CAAC;oBACjD,MAAM,CAAC,IAAI,CAAC;wBACV,UAAU;wBACV,cAAc,EAAE,KAAK;wBACrB,eAAe;qBAChB,CAAC,CAAC;oBACH,UAAU,IAAI,UAAU,CAAC,UAAU,CAAC;gBACtC,CAAC;YACH,CAAC;YAED,2BAA2B;YAC3B,IAAI,UAAU,IAAI,MAAM,GAAG,IAAI,EAAE,CAAC;gBAChC,MAAM;YACR,CAAC;QACH,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,sBAAsB,CAClC,QAA2B,EAC3B,SAAiB;QAEjB,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,CAAC;YACzB,OAAO,EAAE,YAAY,EAAE,QAAQ,EAAE,iBAAiB,EAAE,CAAC,EAAE,CAAC;QAC1D,CAAC;QAED,MAAM,aAAa,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,gBAAgB,EAAE,CAAC;QACnE,IAAI,CAAC,aAAa,IAAI,aAAa,CAAC,EAAE,KAAK,SAAS,EAAE,CAAC;YACrD,OAAO,EAAE,YAAY,EAAE,QAAQ,EAAE,iBAAiB,EAAE,CAAC,EAAE,CAAC;QAC1D,CAAC;QAED,MAAM,YAAY,GAAsB,EAAE,CAAC;QAC3C,IAAI,iBAAiB,GAAG,CAAC,CAAC;QAE1B,KAAK,MAAM,GAAG,IAAI,QAAQ,EAAE,CAAC;YAC3B,IAAI,CAAC,aAAa,CAAC,cAAc,CAAC,GAAG,CAAC,GAAG,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;gBAC/D,YAAY,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACzB,CAAC;iBAAM,CAAC;gBACN,iBAAiB,EAAE,CAAC;YACtB,CAAC;QACH,CAAC;QAED,OAAO,EAAE,YAAY,EAAE,iBAAiB,EAAE,CAAC;IAC7C,CAAC;IAED;;OAEG;IACK,iBAAiB,CAAC,MAAc;QACtC,QAAQ,MAAM,EAAE,CAAC;YACf,KAAK,aAAa;gBAChB,OAAO,CAAC,mBAAmB,EAAE,QAAQ,EAAE,YAAY,EAAE,qBAAqB,CAAC,CAAC;YAC9E,KAAK,SAAS;gBACZ,OAAO,CAAC,QAAQ,EAAE,YAAY,EAAE,kBAAkB,CAAC,CAAC;YACtD,KAAK,UAAU;gBACb,OAAO,CAAC,mBAAmB,EAAE,kBAAkB,EAAE,YAAY,CAAC,CAAC;YACjE,KAAK,gBAAgB;gBACnB,OAAO,CAAC,QAAQ,EAAE,qBAAqB,EAAE,YAAY,CAAC,CAAC;YACzD,KAAK,iBAAiB;gBACpB,OAAO,CAAC,mBAAmB,EAAE,kBAAkB,EAAE,UAAU,CAAC,CAAC;YAC/D,KAAK,UAAU;gBACb,OAAO,CAAC,YAAY,EAAE,mBAAmB,CAAC,CAAC;YAC7C;gBACE,OAAO,CAAC,QAAQ,EAAE,mBAAmB,EAAE,YAAY,CAAC,CAAC;QACzD,CAAC;IACH,CAAC;IAED;;OAEG;IACK,cAAc,CAAC,QAA2B;QAChD,OAAO,QAAQ,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,CAAC,UAAU,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC;IACvE,CAAC;IAED;;OAEG;IACK,oBAAoB,CAAC,MAAc;QACzC,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;QACpC,OAAO,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;IACpC,CAAC;IAED;;OAEG;IACK,wBAAwB,CAAC,UAAoB;QACnD,IAAI,CAAC,IAAI,CAAC,eAAe,EAAE,CAAC;YAC1B,OAAO,EAAE,eAAe,EAAE,KAAK,EAAE,YAAY,EAAE,CAAC,EAAE,CAAC;QACrD,CAAC;QAED,MAAM,KAAK,GAAG,IAAI,CAAC,eAAe,CAAC,QAAQ,EAAE,CAAC;QAC9C,OAAO;YACL,eAAe,EAAE,UAAU,CAAC,MAAM,GAAG,CAAC;YACtC,YAAY,EAAE,KAAK,CAAC,OAAO;SAC5B,CAAC;IACJ,CAAC;CACF"}

package/dist/prediction/cache/index.d.ts

@@ -0,0 +1,10 @@
+/**
+ * Cache Submodule
+ *
+ * Exports prediction caching and preloading components.
+ *
+ * @module prediction/cache
+ */
+export { PredictionCache, type PredictionCacheConfig } from './prediction-cache.js';
+export { EmbeddingPreloader, type EmbeddingPreloaderConfig, type PreloadResult } from './preloader.js';
+//# sourceMappingURL=index.d.ts.map

package/dist/prediction/cache/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/prediction/cache/index.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAEH,OAAO,EAAE,eAAe,EAAE,KAAK,qBAAqB,EAAE,MAAM,uBAAuB,CAAC;AACpF,OAAO,EAAE,kBAAkB,EAAE,KAAK,wBAAwB,EAAE,KAAK,aAAa,EAAE,MAAM,gBAAgB,CAAC"}

package/dist/prediction/cache/index.js

@@ -0,0 +1,10 @@
+/**
+ * Cache Submodule
+ *
+ * Exports prediction caching and preloading components.
+ *
+ * @module prediction/cache
+ */
+export { PredictionCache } from './prediction-cache.js';
+export { EmbeddingPreloader } from './preloader.js';
+//# sourceMappingURL=index.js.map

package/dist/prediction/cache/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/prediction/cache/index.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAEH,OAAO,EAAE,eAAe,EAA8B,MAAM,uBAAuB,CAAC;AACpF,OAAO,EAAE,kBAAkB,EAAqD,MAAM,gBAAgB,CAAC"}

package/dist/prediction/cache/prediction-cache.d.ts

@@ -0,0 +1,97 @@
+/**
+ * Prediction Cache
+ *
+ * Caches predictions for fast retrieval.
+ * Enables efficient prediction reuse when
+ * context hasn't changed significantly.
+ *
+ * @module prediction/cache/prediction-cache
+ */
+import type { PredictedMemory, CachedPrediction, PredictionCacheStats, PredictionSignals } from '../types.js';
+/**
+ * Configuration for prediction cache
+ */
+export interface PredictionCacheConfig {
+    /** Maximum cache entries */
+    maxEntries: number;
+    /** Default TTL in milliseconds */
+    defaultTtlMs: number;
+    /** Whether to use file-based cache keys */
+    useFileBasedKeys: boolean;
+}
+/**
+ * Prediction Cache
+ *
+ * Caches predictions for fast retrieval.
+ */
+export declare class PredictionCache {
+    private config;
+    private cache;
+    private stats;
+    private predictionTimes;
+    constructor(config?: Partial<PredictionCacheConfig>);
+    /**
+     * Get cached predictions for a file
+     */
+    getForFile(file: string): Promise<PredictedMemory[] | null>;
+    /**
+     * Get cached predictions by key
+     */
+    get(key: string): Promise<PredictedMemory[] | null>;
+    /**
+     * Cache predictions
+     */
+    set(key: string, predictions: PredictedMemory[], signals: PredictionSignals, ttlMs?: number): Promise<void>;
+    /**
+     * Cache predictions for a file
+     */
+    setForFile(file: string, predictions: PredictedMemory[], signals: PredictionSignals, ttlMs?: number): Promise<void>;
+    /**
+     * Handle file opened event
+     */
+    onFileOpened(file: string): Promise<PredictedMemory[] | null>;
+    /**
+     * Check if predictions cover a query
+     */
+    predictionsCoverQuery(predictions: PredictedMemory[], query: string): boolean;
+    /**
+     * Invalidate cache for a file
+     */
+    invalidateForFile(file: string): void;
+    /**
+     * Invalidate cache entries containing a memory
+     */
+    invalidateForMemory(memoryId: string): void;
+    /**
+     * Record prediction time for stats
+     */
+    recordPredictionTime(timeMs: number): void;
+    /**
+     * Record embedding preload
+     */
+    recordEmbeddingPreload(count: number): void;
+    /**
+     * Get cache statistics
+     */
+    getStats(): PredictionCacheStats;
+    /**
+     * Clear the cache
+     */
+    clear(): void;
+    /**
+     * Get all cached entries
+     */
+    getAll(): CachedPrediction[];
+    /**
+     * Check if cache has entry for key
+     */
+    has(key: string): boolean;
+    /**
+     * Generate cache key from signals
+     */
+    generateKey(signals: PredictionSignals): string;
+    private generateFileKey;
+    private updateHitRate;
+    private evictOldest;
+}
+//# sourceMappingURL=prediction-cache.d.ts.map

package/dist/prediction/cache/prediction-cache.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"prediction-cache.d.ts","sourceRoot":"","sources":["../../../src/prediction/cache/prediction-cache.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AAEH,OAAO,KAAK,EACV,eAAe,EACf,gBAAgB,EAChB,oBAAoB,EACpB,iBAAiB,EAClB,MAAM,aAAa,CAAC;AAGrB;;GAEG;AACH,MAAM,WAAW,qBAAqB;IACpC,4BAA4B;IAC5B,UAAU,EAAE,MAAM,CAAC;IACnB,kCAAkC;IAClC,YAAY,EAAE,MAAM,CAAC;IACrB,2CAA2C;IAC3C,gBAAgB,EAAE,OAAO,CAAC;CAC3B;AAWD;;;;GAIG;AACH,qBAAa,eAAe;IAC1B,OAAO,CAAC,MAAM,CAAwB;IACtC,OAAO,CAAC,KAAK,CAA4C;IACzD,OAAO,CAAC,KAAK,CAOX;IACF,OAAO,CAAC,eAAe,CAAgB;gBAE3B,MAAM,CAAC,EAAE,OAAO,CAAC,qBAAqB,CAAC;IAInD;;OAEG;IACG,UAAU,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,eAAe,EAAE,GAAG,IAAI,CAAC;IAKjE;;OAEG;IACG,GAAG,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC,eAAe,EAAE,GAAG,IAAI,CAAC;IAuBzD;;OAEG;IACG,GAAG,CACP,GAAG,EAAE,MAAM,EACX,WAAW,EAAE,eAAe,EAAE,EAC9B,OAAO,EAAE,iBAAiB,EAC1B,KAAK,CAAC,EAAE,MAAM,GACb,OAAO,CAAC,IAAI,CAAC;IAsBhB;;OAEG;IACG,UAAU,CACd,IAAI,EAAE,MAAM,EACZ,WAAW,EAAE,eAAe,EAAE,EAC9B,OAAO,EAAE,iBAAiB,EAC1B,KAAK,CAAC,EAAE,MAAM,GACb,OAAO,CAAC,IAAI,CAAC;IAKhB;;OAEG;IACG,YAAY,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,eAAe,EAAE,GAAG,IAAI,CAAC;IAInE;;OAEG;IACH,qBAAqB,CACnB,WAAW,EAAE,eAAe,EAAE,EAC9B,KAAK,EAAE,MAAM,GACZ,OAAO;IAqBV;;OAEG;IACH,iBAAiB,CAAC,IAAI,EAAE,MAAM,GAAG,IAAI;IAMrC;;OAEG;IACH,mBAAmB,CAAC,QAAQ,EAAE,MAAM,GAAG,IAAI;IAgB3C;;OAEG;IACH,oBAAoB,CAAC,MAAM,EAAE,MAAM,GAAG,IAAI;IAa1C;;OAEG;IACH,sBAAsB,CAAC,KAAK,EAAE,MAAM,GAAG,IAAI;IAI3C;;OAEG;IACH,QAAQ,IAAI,oBAAoB;IAIhC;;OAEG;IACH,KAAK,IAAI,IAAI;IASb;;OAEG;IACH,MAAM,IAAI,gBAAgB,EAAE;IAI5B;;OAEG;IACH,GAAG,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO;IAczB;;OAEG;IACH,WAAW,CAAC,OAAO,EAAE,iBAAiB,GAAG,MAAM;IAiB/C,OAAO,CAAC,eAAe;IAKvB,OAAO,CAAC,aAAa;IAKrB,OAAO,CAAC,WAAW;CAkBpB"}