@aman_asmuei/amem 0.18.1 → 0.20.0

package/dist/embeddings.js

@@ -1,251 +0,0 @@
-export function cosineSimilarity(a, b) {
-    let dot = 0;
-    let normA = 0;
-    let normB = 0;
-    for (let i = 0; i < a.length; i++) {
-        dot += a[i] * b[i];
-        normA += a[i] * a[i];
-        normB += b[i] * b[i];
-    }
-    const denom = Math.sqrt(normA) * Math.sqrt(normB);
-    if (denom === 0)
-        return 0;
-    return dot / denom;
-}
-export function findTopK(query, candidates, k) {
-    const scored = candidates.map((c) => ({
-        id: c.id,
-        similarity: cosineSimilarity(query, c.embedding),
-        data: c.data,
-    }));
-    scored.sort((a, b) => b.similarity - a.similarity);
-    return scored.slice(0, k);
-}
-let pipelineInstance = null;
-let pipelineLoading = null;
-// LRU-style embedding cache to avoid recomputing identical queries
-const EMBEDDING_CACHE_MAX = 128;
-const embeddingCache = new Map();
-function cacheGet(key) {
-    const val = embeddingCache.get(key);
-    if (val) {
-        // Move to end (most recently used)
-        embeddingCache.delete(key);
-        embeddingCache.set(key, val);
-    }
-    return val;
-}
-function cachePut(key, val) {
-    if (embeddingCache.size >= EMBEDDING_CACHE_MAX) {
-        // Evict oldest (first) entry
-        const oldest = embeddingCache.keys().next().value;
-        if (oldest !== undefined)
-            embeddingCache.delete(oldest);
-    }
-    embeddingCache.set(key, val);
-}
-/** Skip embedding loading entirely (useful for CLI commands that need fast exit). */
-let embeddingDisabled = false;
-export function disableEmbeddings() {
-    embeddingDisabled = true;
-}
-async function getEmbeddingPipeline() {
-    if (embeddingDisabled)
-        return null;
-    if (pipelineInstance)
-        return pipelineInstance;
-    if (pipelineLoading)
-        return pipelineLoading;
-    pipelineLoading = (async () => {
-        const LOAD_TIMEOUT_MS = 120000;
-        async function attemptLoad() {
-            console.error("[amem] Loading embedding model — this may take a moment on first run (downloading model)...");
-            const startTime = Date.now();
-            const loadPromise = (async () => {
-                const { loadConfig } = await import("./config.js");
-                const config = loadConfig();
-                const mod = await import("@huggingface/transformers");
-                return await mod.pipeline("feature-extraction", config.embeddingModel);
-            })();
-            // Log progress so the user knows we're still working
-            const progressInterval = setInterval(() => {
-                const elapsed = Math.round((Date.now() - startTime) / 1000);
-                console.error(`[amem] Still loading embedding model... (${elapsed}s elapsed)`);
-            }, 15000);
-            try {
-                return await Promise.race([
-                    loadPromise,
-                    new Promise((resolve) => setTimeout(() => resolve(null), LOAD_TIMEOUT_MS)),
-                ]);
-            }
-            finally {
-                clearInterval(progressInterval);
-            }
-        }
-        try {
-            const result = await attemptLoad();
-            if (result) {
-                pipelineInstance = result;
-                console.error("[amem] Embedding model loaded — semantic search enabled");
-                return pipelineInstance;
-            }
-            console.error("[amem] Embedding model load timed out after 120s — using keyword matching. Try running again once download completes.");
-            return null;
-        }
-        catch (error) {
-            const msg = error instanceof Error ? error.message : String(error);
-            // If the cached model is corrupted, clear it and retry once
-            if (msg.includes("Protobuf parsing failed") || msg.includes("invalid model")) {
-                console.error("[amem] Corrupted model cache detected — clearing and retrying...");
-                try {
-                    const fs = await import("node:fs");
-                    const path = await import("node:path");
-                    const os = await import("node:os");
-                    const { loadConfig } = await import("./config.js");
-                    const cfg = loadConfig();
-                    const modelParts = cfg.embeddingModel.split("/");
-                    const modelOrg = modelParts[0] ?? "Xenova";
-                    const modelName = modelParts[1] ?? cfg.embeddingModel;
-                    // Clear the HuggingFace cache for this model
-                    const cacheLocations = [
-                        path.join(os.homedir(), ".cache", "huggingface", "transformers", modelOrg, modelName),
-                    ];
-                    // Also try to find the cache inside node_modules
-                    try {
-                        const modPath = import.meta.resolve?.("@huggingface/transformers") ?? "";
-                        if (modPath) {
-                            const modDir = path.dirname(modPath.replace("file://", ""));
-                            cacheLocations.push(path.join(modDir, ".cache", modelOrg, modelName));
-                        }
-                    }
-                    catch { }
-                    for (const loc of cacheLocations) {
-                        if (fs.existsSync(loc)) {
-                            fs.rmSync(loc, { recursive: true, force: true });
-                            console.error(`[amem] Cleared cache: ${loc}`);
-                        }
-                    }
-                    // Retry once
-                    const retryResult = await attemptLoad();
-                    if (retryResult) {
-                        pipelineInstance = retryResult;
-                        console.error("[amem] Model re-downloaded successfully — semantic search enabled");
-                        return pipelineInstance;
-                    }
-                }
-                catch (retryError) {
-                    console.error("[amem] Cache recovery failed:", retryError instanceof Error ? retryError.message : String(retryError));
-                }
-            }
-            console.error("[amem] Embeddings unavailable — using keyword matching (still works great!):", msg);
-            console.error("[amem] To enable semantic search: npm install @huggingface/transformers");
-            return null;
-        }
-    })();
-    return pipelineLoading;
-}
-/**
- * Pre-warm the embedding pipeline in the background.
- * Call this at startup so the model is ready when the first query arrives.
- */
-export function preloadEmbeddings() {
-    // Fire-and-forget — don't block startup
-    getEmbeddingPipeline().catch(() => { });
-}
-export async function generateEmbedding(text) {
-    // Check cache first
-    const cached = cacheGet(text);
-    if (cached)
-        return cached;
-    const extractor = await getEmbeddingPipeline();
-    if (!extractor)
-        return null;
-    const result = await extractor(text, { pooling: "mean", normalize: true });
-    const embedding = new Float32Array(result.data);
-    cachePut(text, embedding);
-    return embedding;
-}
-export async function isEmbeddingAvailable() {
-    const extractor = await getEmbeddingPipeline();
-    return extractor !== null;
-}
-let rerankerInstance = null;
-let rerankerLoading = null;
-let rerankerFailed = false;
-async function getCrossEncoderPipeline() {
-    if (rerankerInstance)
-        return rerankerInstance;
-    if (rerankerFailed)
-        return null;
-    if (rerankerLoading)
-        return rerankerLoading;
-    rerankerLoading = (async () => {
-        try {
-            const mod = await import("@huggingface/transformers");
-            const classifier = await mod.pipeline("text-classification", "Xenova/ms-marco-MiniLM-L-6-v2");
-            rerankerInstance = async (texts) => {
-                const results = [];
-                // Process one at a time to avoid memory spikes
-                for (const pair of texts) {
-                    const result = await classifier(pair.text, { text_pair: pair.text_pair, topk: 1 });
-                    const arr = Array.isArray(result) ? result : [result];
-                    results.push(arr);
-                }
-                return results;
-            };
-            return rerankerInstance;
-        }
-        catch (error) {
-            console.error("[amem] Cross-encoder reranker unavailable — skipping rerank step:", error instanceof Error ? error.message : String(error));
-            rerankerFailed = true;
-            return null;
-        }
-    })();
-    return rerankerLoading;
-}
-/**
- * Cross-encoder reranking: takes a query and a list of candidates,
- * scores each (query, candidate) pair with a cross-encoder model,
- * and returns candidates re-sorted by cross-encoder score.
- *
- * This is the final pass in the retrieval pipeline — after semantic + FTS + graph
- * have produced candidates, the cross-encoder provides the most accurate scoring
- * by attending to the full (query, document) pair jointly.
- *
- * Falls back to original scores if the cross-encoder model isn't available.
- */
-export async function rerankWithCrossEncoder(query, candidates, topK) {
-    if (candidates.length === 0)
-        return [];
-    if (candidates.length <= 1)
-        return candidates;
-    const scorer = await getCrossEncoderPipeline();
-    if (!scorer) {
-        // Fallback: return candidates as-is (already sorted by multi-strategy score)
-        return candidates.slice(0, topK);
-    }
-    try {
-        const pairs = candidates.map(c => ({
-            text: query,
-            text_pair: c.content.slice(0, 512), // Truncate long docs for cross-encoder
-        }));
-        const scores = await scorer(pairs);
-        // Merge cross-encoder scores with candidates
-        const reranked = candidates.map((c, i) => {
-            // Cross-encoder outputs relevance score; higher = more relevant
-            const ceScore = scores[i]?.[0]?.score ?? 0;
-            return { ...c, score: ceScore };
-        });
-        reranked.sort((a, b) => b.score - a.score);
-        return reranked.slice(0, topK);
-    }
-    catch (error) {
-        console.error("[amem] Cross-encoder reranking failed, using original scores:", error instanceof Error ? error.message : String(error));
-        return candidates.slice(0, topK);
-    }
-}
-export async function isRerankerAvailable() {
-    const scorer = await getCrossEncoderPipeline();
-    return scorer !== null;
-}
-//# sourceMappingURL=embeddings.js.map

package/dist/embeddings.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"embeddings.js","sourceRoot":"","sources":["../src/embeddings.ts"],"names":[],"mappings":"AAAA,MAAM,UAAU,gBAAgB,CAAC,CAAe,EAAE,CAAe;IAC/D,IAAI,GAAG,GAAG,CAAC,CAAC;IACZ,IAAI,KAAK,GAAG,CAAC,CAAC;IACd,IAAI,KAAK,GAAG,CAAC,CAAC;IACd,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QAClC,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;QACnB,KAAK,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;QACrB,KAAK,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;IACvB,CAAC;IACD,MAAM,KAAK,GAAG,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAClD,IAAI,KAAK,KAAK,CAAC;QAAE,OAAO,CAAC,CAAC;IAC1B,OAAO,GAAG,GAAG,KAAK,CAAC;AACrB,CAAC;AAcD,MAAM,UAAU,QAAQ,CACtB,KAAmB,EACnB,UAAmC,EACnC,CAAS;IAET,MAAM,MAAM,GAAG,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;QACpC,EAAE,EAAE,CAAC,CAAC,EAAE;QACR,UAAU,EAAE,gBAAgB,CAAC,KAAK,EAAE,CAAC,CAAC,SAAS,CAAC;QAChD,IAAI,EAAE,CAAC,CAAC,IAAI;KACb,CAAC,CAAC,CAAC;IACJ,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,GAAG,CAAC,CAAC,UAAU,CAAC,CAAC;IACnD,OAAO,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;AAC5B,CAAC;AAOD,IAAI,gBAAgB,GAA4B,IAAI,CAAC;AACrD,IAAI,eAAe,GAA4C,IAAI,CAAC;AAEpE,mEAAmE;AACnE,MAAM,mBAAmB,GAAG,GAAG,CAAC;AAChC,MAAM,cAAc,GAAG,IAAI,GAAG,EAAwB,CAAC;AAEvD,SAAS,QAAQ,CAAC,GAAW;IAC3B,MAAM,GAAG,GAAG,cAAc,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;IACpC,IAAI,GAAG,EAAE,CAAC;QACR,mCAAmC;QACnC,cAAc,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QAC3B,cAAc,CAAC,GAAG,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;IAC/B,CAAC;IACD,OAAO,GAAG,CAAC;AACb,CAAC;AAED,SAAS,QAAQ,CAAC,GAAW,EAAE,GAAiB;IAC9C,IAAI,cAAc,CAAC,IAAI,IAAI,mBAAmB,EAAE,CAAC;QAC/C,6BAA6B;QAC7B,MAAM,MAAM,GAAG,cAAc,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC;QAClD,IAAI,MAAM,KAAK,SAAS;YAAE,cAAc,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;IAC1D,CAAC;IACD,cAAc,CAAC,GAAG,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;AAC/B,CAAC;AAED,qFAAqF;AACrF,IAAI,iBAAiB,GAAG,KAAK,CAAC;AAE9B,MAAM,UAAU,iBAAiB;IAC/B,iBAAiB,GAAG,IAAI,CAAC;AAC3B,CAAC;AAED,KAAK,UAAU,oBAAoB;IACjC,IAAI,iBAAiB;QAAE,OAAO,IAAI,CAAC;IACnC,IAAI,gBAAgB;QAAE,OAAO,gBAAgB,CAAC;IAC9C,IAAI,eAAe;QAAE,OAAO,eAAe,CAAC;IAE5C,eAAe,GAAG,CAAC,KAAK,IAAI,EAAE;QAC5B,MAAM,eAAe,GAAG,MAAM,CAAC;QAE/B,KAAK,UAAU,WAAW;YACxB,OAAO,CAAC,KAAK,CAAC,6FAA6F,CAAC,CAAC;YAC7G,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAC7B,MAAM,WAAW,GAAG,CAAC,KAAK,IAAI,EAAE;gBAC9B,MAAM,EAAE,UAAU,EAAE,GAAG,MAAM,MAAM,CAAC,aAAa,CAAC,CAAC;gBACnD,MAAM,MAAM,GAAG,UAAU,EAAE,CAAC;gBAC5B,MAAM,GAAG,GAAG,MAAM,MAAM,CAAC,2BAA2B,CAAC,CAAC;gBACtD,OAAO,MAAM,GAAG,CAAC,QAAQ,CACvB,oBAAoB,EACpB,MAAM,CAAC,cAAc,CACS,CAAC;YACnC,CAAC,CAAC,EAAE,CAAC;YAEL,qDAAqD;YACrD,MAAM,gBAAgB,GAAG,WAAW,CAAC,GAAG,EAAE;gBACxC,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC,GAAG,IAAI,CAAC,CAAC;gBAC5D,OAAO,CAAC,KAAK,CAAC,4CAA4C,OAAO,YAAY,CAAC,CAAC;YACjF,CAAC,EAAE,KAAK,CAAC,CAAC;YAEV,IAAI,CAAC;gBACH,OAAO,MAAM,OAAO,CAAC,IAAI,CAAC;oBACxB,WAAW;oBACX,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,eAAe,CAAC,CAAC;iBACjF,CAAC,CAAC;YACL,CAAC;oBAAS,CAAC;gBACT,aAAa,CAAC,gBAAgB,CAAC,CAAC;YAClC,CAAC;QACH,CAAC;QAED,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,WAAW,EAAE,CAAC;YACnC,IAAI,MAAM,EAAE,CAAC;gBACX,gBAAgB,GAAG,MAAM,CAAC;gBAC1B,OAAO,CAAC,KAAK,CAAC,yDAAyD,CAAC,CAAC;gBACzE,OAAO,gBAAgB,CAAC;YAC1B,CAAC;YAED,OAAO,CAAC,KAAK,CAAC,uHAAuH,CAAC,CAAC;YACvI,OAAO,IAAI,CAAC;QACd,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,GAAG,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YACnE,4DAA4D;YAC5D,IAAI,GAAG,CAAC,QAAQ,CAAC,yBAAyB,CAAC,IAAI,GAAG,CAAC,QAAQ,CAAC,eAAe,CAAC,EAAE,CAAC;gBAC7E,OAAO,CAAC,KAAK,CAAC,kEAAkE,CAAC,CAAC;gBAClF,IAAI,CAAC;oBACH,MAAM,EAAE,GAAG,MAAM,MAAM,CAAC,SAAS,CAAC,CAAC;oBACnC,MAAM,IAAI,GAAG,MAAM,MAAM,CAAC,WAAW,CAAC,CAAC;oBACvC,MAAM,EAAE,GAAG,MAAM,MAAM,CAAC,SAAS,CAAC,CAAC;oBACnC,MAAM,EAAE,UAAU,EAAE,GAAG,MAAM,MAAM,CAAC,aAAa,CAAC,CAAC;oBACnD,MAAM,GAAG,GAAG,UAAU,EAAE,CAAC;oBACzB,MAAM,UAAU,GAAG,GAAG,CAAC,cAAc,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;oBACjD,MAAM,QAAQ,GAAG,UAAU,CAAC,CAAC,CAAC,IAAI,QAAQ,CAAC;oBAC3C,MAAM,SAAS,GAAG,UAAU,CAAC,CAAC,CAAC,IAAI,GAAG,CAAC,cAAc,CAAC;oBACtD,6CAA6C;oBAC7C,MAAM,cAAc,GAAG;wBACrB,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE,EAAE,QAAQ,EAAE,aAAa,EAAE,cAAc,EAAE,QAAQ,EAAE,SAAS,CAAC;qBACtF,CAAC;oBACF,iDAAiD;oBACjD,IAAI,CAAC;wBACH,MAAM,OAAO,GAAG,MAAM,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,2BAA2B,CAAC,IAAI,EAAE,CAAC;wBACzE,IAAI,OAAO,EAAE,CAAC;4BACZ,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC,CAAC;4BAC5D,cAAc,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,QAAQ,EAAE,QAAQ,EAAE,SAAS,CAAC,CAAC,CAAC;wBACxE,CAAC;oBACH,CAAC;oBAAC,MAAM,CAAC,CAAA,CAAC;oBACV,KAAK,MAAM,GAAG,IAAI,cAAc,EAAE,CAAC;wBACjC,IAAI,EAAE,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;4BACvB,EAAE,CAAC,MAAM,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;4BACjD,OAAO,CAAC,KAAK,CAAC,yBAAyB,GAAG,EAAE,CAAC,CAAC;wBAChD,CAAC;oBACH,CAAC;oBACD,aAAa;oBACb,MAAM,WAAW,GAAG,MAAM,WAAW,EAAE,CAAC;oBACxC,IAAI,WAAW,EAAE,CAAC;wBAChB,gBAAgB,GAAG,WAAW,CAAC;wBAE/B,OAAO,CAAC,KAAK,CAAC,mEAAmE,CAAC,CAAC;wBACnF,OAAO,gBAAgB,CAAC;oBAC1B,CAAC;gBACH,CAAC;gBAAC,OAAO,UAAU,EAAE,CAAC;oBACpB,OAAO,CAAC,KAAK,CAAC,+BAA+B,EAAE,UAAU,YAAY,KAAK,CAAC,CAAC,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,CAAC;gBACxH,CAAC;YACH,CAAC;YAED,OAAO,CAAC,KAAK,CAAC,8EAA8E,EAAE,GAAG,CAAC,CAAC;YACnG,OAAO,CAAC,KAAK,CAAC,yEAAyE,CAAC,CAAC;YACzF,OAAO,IAAI,CAAC;QACd,CAAC;IACH,CAAC,CAAC,EAAE,CAAC;IAEL,OAAO,eAAe,CAAC;AACzB,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,iBAAiB;IAC/B,wCAAwC;IACxC,oBAAoB,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;AACzC,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,iBAAiB,CACrC,IAAY;IAEZ,oBAAoB;IACpB,MAAM,MAAM,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC;IAC9B,IAAI,MAAM;QAAE,OAAO,MAAM,CAAC;IAE1B,MAAM,SAAS,GAAG,MAAM,oBAAoB,EAAE,CAAC;IAC/C,IAAI,CAAC,SAAS;QAAE,OAAO,IAAI,CAAC;IAE5B,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,IAAI,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IAC3E,MAAM,SAAS,GAAG,IAAI,YAAY,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;IAChD,QAAQ,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC;IAC1B,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,oBAAoB;IACxC,MAAM,SAAS,GAAG,MAAM,oBAAoB,EAAE,CAAC;IAC/C,OAAO,SAAS,KAAK,IAAI,CAAC;AAC5B,CAAC;AAQD,IAAI,gBAAgB,GAA8B,IAAI,CAAC;AACvD,IAAI,eAAe,GAA8C,IAAI,CAAC;AACtE,IAAI,cAAc,GAAG,KAAK,CAAC;AAE3B,KAAK,UAAU,uBAAuB;IACpC,IAAI,gBAAgB;QAAE,OAAO,gBAAgB,CAAC;IAC9C,IAAI,cAAc;QAAE,OAAO,IAAI,CAAC;IAChC,IAAI,eAAe;QAAE,OAAO,eAAe,CAAC;IAE5C,eAAe,GAAG,CAAC,KAAK,IAAI,EAAE;QAC5B,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,MAAM,MAAM,CAAC,2BAA2B,CAAC,CAAC;YACtD,MAAM,UAAU,GAAG,MAAO,GAAG,CAAC,QAAqB,CACjD,qBAAqB,EACrB,+BAA+B,CAChC,CAAC;YAEF,gBAAgB,GAAG,KAAK,EAAE,KAAK,EAAE,EAAE;gBACjC,MAAM,OAAO,GAA8C,EAAE,CAAC;gBAC9D,+CAA+C;gBAC/C,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;oBACzB,MAAM,MAAM,GAAG,MAAO,UAAuB,CAAC,IAAI,CAAC,IAAI,EAAE,EAAE,SAAS,EAAE,IAAI,CAAC,SAAS,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC;oBACjG,MAAM,GAAG,GAAG,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC;oBACtD,OAAO,CAAC,IAAI,CAAC,GAAyC,CAAC,CAAC;gBAC1D,CAAC;gBACD,OAAO,OAAO,CAAC;YACjB,CAAC,CAAC;YACF,OAAO,gBAAgB,CAAC;QAC1B,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,mEAAmE,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;YAC3I,cAAc,GAAG,IAAI,CAAC;YACtB,OAAO,IAAI,CAAC;QACd,CAAC;IACH,CAAC,CAAC,EAAE,CAAC;IAEL,OAAO,eAAe,CAAC;AACzB,CAAC;AAQD;;;;;;;;;;GAUG;AACH,MAAM,CAAC,KAAK,UAAU,sBAAsB,CAC1C,KAAa,EACb,UAA6B,EAC7B,IAAY;IAEZ,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,EAAE,CAAC;IACvC,IAAI,UAAU,CAAC,MAAM,IAAI,CAAC;QAAE,OAAO,UAAU,CAAC;IAE9C,MAAM,MAAM,GAAG,MAAM,uBAAuB,EAAE,CAAC;IAC/C,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,6EAA6E;QAC7E,OAAO,UAAU,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC;IACnC,CAAC;IAED,IAAI,CAAC;QACH,MAAM,KAAK,GAAG,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;YACjC,IAAI,EAAE,KAAK;YACX,SAAS,EAAE,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,uCAAuC;SAC5E,CAAC,CAAC,CAAC;QAEJ,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,KAAK,CAAC,CAAC;QAEnC,6CAA6C;QAC7C,MAAM,QAAQ,GAAG,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE;YACvC,gEAAgE;YAChE,MAAM,OAAO,GAAG,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,IAAI,CAAC,CAAC;YAC3C,OAAO,EAAE,GAAG,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC;QAClC,CAAC,CAAC,CAAC;QAEH,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC;QAC3C,OAAO,QAAQ,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC;IACjC,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,KAAK,CAAC,+DAA+D,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;QACvI,OAAO,UAAU,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC;IACnC,CAAC;AACH,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,mBAAmB;IACvC,MAAM,MAAM,GAAG,MAAM,uBAAuB,EAAE,CAAC;IAC/C,OAAO,MAAM,KAAK,IAAI,CAAC;AACzB,CAAC"}

package/dist/embeddings.test.d.ts

@@ -1 +0,0 @@
-export {};

package/dist/embeddings.test.js

@@ -1,106 +0,0 @@
-import { describe, it, expect } from "vitest";
-import { cosineSimilarity, findTopK } from "./embeddings.js";
-describe("cosineSimilarity", () => {
-    it("returns 1.0 for identical normalized vectors", () => {
-        const v = new Float32Array([1 / Math.sqrt(2), 1 / Math.sqrt(2)]);
-        expect(cosineSimilarity(v, v)).toBeCloseTo(1.0, 5);
-    });
-    it("returns 0 for orthogonal vectors", () => {
-        const a = new Float32Array([1, 0]);
-        const b = new Float32Array([0, 1]);
-        expect(cosineSimilarity(a, b)).toBeCloseTo(0, 5);
-    });
-    it("returns -1 for opposite vectors", () => {
-        const a = new Float32Array([1, 0]);
-        const b = new Float32Array([-1, 0]);
-        expect(cosineSimilarity(a, b)).toBeCloseTo(-1.0, 5);
-    });
-    it("returns 0 when a is a zero vector", () => {
-        const a = new Float32Array([0, 0, 0]);
-        const b = new Float32Array([1, 2, 3]);
-        expect(cosineSimilarity(a, b)).toBe(0);
-    });
-    it("returns 0 when b is a zero vector", () => {
-        const a = new Float32Array([1, 2, 3]);
-        const b = new Float32Array([0, 0, 0]);
-        expect(cosineSimilarity(a, b)).toBe(0);
-    });
-    it("returns 0 when both are zero vectors", () => {
-        const a = new Float32Array([0, 0]);
-        const b = new Float32Array([0, 0]);
-        expect(cosineSimilarity(a, b)).toBe(0);
-    });
-    it("computes correct similarity for known vectors", () => {
-        // [1, 2, 3] and [4, 5, 6]
-        // dot = 4+10+18 = 32, normA = sqrt(14), normB = sqrt(77)
-        // similarity = 32 / (sqrt(14) * sqrt(77)) = 32 / sqrt(1078)
-        const a = new Float32Array([1, 2, 3]);
-        const b = new Float32Array([4, 5, 6]);
-        const expected = 32 / (Math.sqrt(14) * Math.sqrt(77));
-        expect(cosineSimilarity(a, b)).toBeCloseTo(expected, 5);
-    });
-    it("is symmetric: sim(a,b) === sim(b,a)", () => {
-        const a = new Float32Array([0.3, 0.7, 0.1]);
-        const b = new Float32Array([0.5, 0.2, 0.9]);
-        expect(cosineSimilarity(a, b)).toBeCloseTo(cosineSimilarity(b, a), 10);
-    });
-    it("handles single-element vectors", () => {
-        const a = new Float32Array([3]);
-        const b = new Float32Array([5]);
-        expect(cosineSimilarity(a, b)).toBeCloseTo(1.0, 5);
-    });
-    it("handles negative values correctly", () => {
-        const a = new Float32Array([-1, -2]);
-        const b = new Float32Array([-1, -2]);
-        expect(cosineSimilarity(a, b)).toBeCloseTo(1.0, 5);
-    });
-});
-describe("findTopK", () => {
-    it("returns top K most similar candidates", () => {
-        const query = new Float32Array([1, 0, 0]);
-        const candidates = [
-            { id: "a", embedding: new Float32Array([1, 0, 0]), data: "exact" },
-            { id: "b", embedding: new Float32Array([0, 1, 0]), data: "orthogonal" },
-            { id: "c", embedding: new Float32Array([0.9, 0.1, 0]), data: "close" },
-        ];
-        const results = findTopK(query, candidates, 2);
-        expect(results).toHaveLength(2);
-        expect(results[0].id).toBe("a"); // exact match
-        expect(results[1].id).toBe("c"); // close match
-        expect(results[0].similarity).toBeGreaterThan(results[1].similarity);
-    });
-    it("returns all candidates when k > candidates.length", () => {
-        const query = new Float32Array([1, 0]);
-        const candidates = [
-            { id: "a", embedding: new Float32Array([1, 0]), data: "x" },
-        ];
-        const results = findTopK(query, candidates, 5);
-        expect(results).toHaveLength(1);
-    });
-    it("returns empty array for empty candidates", () => {
-        const query = new Float32Array([1, 0]);
-        const results = findTopK(query, [], 5);
-        expect(results).toHaveLength(0);
-    });
-    it("preserves data field in results", () => {
-        const query = new Float32Array([1, 0]);
-        const candidates = [
-            { id: "a", embedding: new Float32Array([1, 0]), data: { foo: "bar" } },
-        ];
-        const results = findTopK(query, candidates, 1);
-        expect(results[0].data).toEqual({ foo: "bar" });
-    });
-    it("sorts results by descending similarity", () => {
-        const query = new Float32Array([1, 0, 0]);
-        const candidates = [
-            { id: "low", embedding: new Float32Array([0, 0, 1]), data: 1 },
-            { id: "high", embedding: new Float32Array([1, 0, 0]), data: 2 },
-            { id: "mid", embedding: new Float32Array([0.7, 0.7, 0]), data: 3 },
-        ];
-        const results = findTopK(query, candidates, 3);
-        for (let i = 0; i < results.length - 1; i++) {
-            expect(results[i].similarity).toBeGreaterThanOrEqual(results[i + 1].similarity);
-        }
-    });
-});
-//# sourceMappingURL=embeddings.test.js.map

package/dist/embeddings.test.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"embeddings.test.js","sourceRoot":"","sources":["../src/embeddings.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,gBAAgB,EAAE,QAAQ,EAAE,MAAM,iBAAiB,CAAC;AAE7D,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;IAChC,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;QACtD,MAAM,CAAC,GAAG,IAAI,YAAY,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QACjE,MAAM,CAAC,gBAAgB,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC;IACrD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kCAAkC,EAAE,GAAG,EAAE;QAC1C,MAAM,CAAC,GAAG,IAAI,YAAY,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;QACnC,MAAM,CAAC,GAAG,IAAI,YAAY,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;QACnC,MAAM,CAAC,gBAAgB,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IACnD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;QACzC,MAAM,CAAC,GAAG,IAAI,YAAY,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;QACnC,MAAM,CAAC,GAAG,IAAI,YAAY,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;QACpC,MAAM,CAAC,gBAAgB,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC;IACtD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mCAAmC,EAAE,GAAG,EAAE;QAC3C,MAAM,CAAC,GAAG,IAAI,YAAY,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;QACtC,MAAM,CAAC,GAAG,IAAI,YAAY,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;QACtC,MAAM,CAAC,gBAAgB,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACzC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mCAAmC,EAAE,GAAG,EAAE;QAC3C,MAAM,CAAC,GAAG,IAAI,YAAY,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;QACtC,MAAM,CAAC,GAAG,IAAI,YAAY,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;QACtC,MAAM,CAAC,gBAAgB,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACzC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sCAAsC,EAAE,GAAG,EAAE;QAC9C,MAAM,CAAC,GAAG,IAAI,YAAY,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;QACnC,MAAM,CAAC,GAAG,IAAI,YAAY,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;QACnC,MAAM,CAAC,gBAAgB,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACzC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;QACvD,0BAA0B;QAC1B,yDAAyD;QACzD,4DAA4D;QAC5D,MAAM,CAAC,GAAG,IAAI,YAAY,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;QACtC,MAAM,CAAC,GAAG,IAAI,YAAY,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;QACtC,MAAM,QAAQ,GAAG,EAAE,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC;QACtD,MAAM,CAAC,gBAAgB,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC;IAC1D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qCAAqC,EAAE,GAAG,EAAE;QAC7C,MAAM,CAAC,GAAG,IAAI,YAAY,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC;QAC5C,MAAM,CAAC,GAAG,IAAI,YAAY,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC;QAC5C,MAAM,CAAC,gBAAgB,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,gBAAgB,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;IACzE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gCAAgC,EAAE,GAAG,EAAE;QACxC,MAAM,CAAC,GAAG,IAAI,YAAY,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QAChC,MAAM,CAAC,GAAG,IAAI,YAAY,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QAChC,MAAM,CAAC,gBAAgB,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC;IACrD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mCAAmC,EAAE,GAAG,EAAE;QAC3C,MAAM,CAAC,GAAG,IAAI,YAAY,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;QACrC,MAAM,CAAC,GAAG,IAAI,YAAY,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;QACrC,MAAM,CAAC,gBAAgB,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC;IACrD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,UAAU,EAAE,GAAG,EAAE;IACxB,EAAE,CAAC,uCAAuC,EAAE,GAAG,EAAE;QAC/C,MAAM,KAAK,GAAG,IAAI,YAAY,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;QAC1C,MAAM,UAAU,GAAG;YACjB,EAAE,EAAE,EAAE,GAAG,EAAE,SAAS,EAAE,IAAI,YAAY,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE;YAClE,EAAE,EAAE,EAAE,GAAG,EAAE,SAAS,EAAE,IAAI,YAAY,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE;YACvE,EAAE,EAAE,EAAE,GAAG,EAAE,SAAS,EAAE,IAAI,YAAY,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE;SACvE,CAAC;QAEF,MAAM,OAAO,GAAG,QAAQ,CAAC,KAAK,EAAE,UAAU,EAAE,CAAC,CAAC,CAAC;QAC/C,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAChC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,cAAc;QAC/C,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,cAAc;QAC/C,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC;IACvE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mDAAmD,EAAE,GAAG,EAAE;QAC3D,MAAM,KAAK,GAAG,IAAI,YAAY,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;QACvC,MAAM,UAAU,GAAG;YACjB,EAAE,EAAE,EAAE,GAAG,EAAE,SAAS,EAAE,IAAI,YAAY,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,GAAG,EAAE;SAC5D,CAAC;QAEF,MAAM,OAAO,GAAG,QAAQ,CAAC,KAAK,EAAE,UAAU,EAAE,CAAC,CAAC,CAAC;QAC/C,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;IAClC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;QAClD,MAAM,KAAK,GAAG,IAAI,YAAY,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;QACvC,MAAM,OAAO,GAAG,QAAQ,CAAC,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;QACvC,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;IAClC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;QACzC,MAAM,KAAK,GAAG,IAAI,YAAY,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;QACvC,MAAM,UAAU,GAAG;YACjB,EAAE,EAAE,EAAE,GAAG,EAAE,SAAS,EAAE,IAAI,YAAY,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,EAAE,GAAG,EAAE,KAAK,EAAE,EAAE;SACvE,CAAC;QAEF,MAAM,OAAO,GAAG,QAAQ,CAAC,KAAK,EAAE,UAAU,EAAE,CAAC,CAAC,CAAC;QAC/C,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,EAAE,GAAG,EAAE,KAAK,EAAE,CAAC,CAAC;IAClD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wCAAwC,EAAE,GAAG,EAAE;QAChD,MAAM,KAAK,GAAG,IAAI,YAAY,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;QAC1C,MAAM,UAAU,GAAG;YACjB,EAAE,EAAE,EAAE,KAAK,EAAE,SAAS,EAAE,IAAI,YAAY,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE;YAC9D,EAAE,EAAE,EAAE,MAAM,EAAE,SAAS,EAAE,IAAI,YAAY,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE;YAC/D,EAAE,EAAE,EAAE,KAAK,EAAE,SAAS,EAAE,IAAI,YAAY,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE;SACnE,CAAC;QAEF,MAAM,OAAO,GAAG,QAAQ,CAAC,KAAK,EAAE,UAAU,EAAE,CAAC,CAAC,CAAC;QAC/C,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;YAC5C,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,sBAAsB,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC;QAClF,CAAC;IACH,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/extractor.d.ts

@@ -1,13 +0,0 @@
-import type { MemoryTypeValue } from "./memory.js";
-export interface ConversationTurn {
-    role: "user" | "assistant" | "system";
-    content: string;
-}
-export interface ExtractedMemory {
-    content: string;
-    type: MemoryTypeValue;
-    confidence: number;
-    tags: string[];
-    source: string;
-}
-export declare function extractMemories(turns: ConversationTurn[]): ExtractedMemory[];

package/dist/extractor.js

@@ -1,89 +0,0 @@
-const EXTRACTION_PATTERNS = [
-    // Corrections — user telling the AI "don't do X" or "always do Y"
-    {
-        type: "correction",
-        patterns: [
-            /\b(?:don'?t|never|stop|no,?\s+(?:don'?t|never|not))\b.*\b(?:use|do|add|include|write|put|make|create)\b/i,
-            /\b(?:always|must|should always|never ever)\b.*\b(?:use|do|add|include|write|make)\b/i,
-            /\bthat'?s (?:wrong|incorrect|not right)\b/i,
-            /\bno,?\s+(?:that|this|it) (?:should|needs to|must)\b/i,
-        ],
-        confidence: 0.95,
-        roles: ["user"],
-    },
-    // Decisions — "we decided", "let's go with", "the approach is"
-    {
-        type: "decision",
-        patterns: [
-            /\b(?:we (?:decided|chose|agreed)|let'?s (?:go with|use|stick with))\b/i,
-            /\b(?:the (?:decision|approach|plan|strategy) is)\b/i,
-            /\b(?:we'?re (?:going to|gonna)|we'll)\b.*\b(?:use|switch to|migrate to|adopt)\b/i,
-        ],
-        confidence: 0.85,
-        roles: ["user"],
-    },
-    // Preferences — "I prefer", "I like to", "I want"
-    {
-        type: "preference",
-        patterns: [
-            /\bi (?:prefer|like to|want(?:ed)? to|tend to)\b/i,
-            /\bmy (?:preference|style|approach|convention) is\b/i,
-            /\bplease (?:always|use|keep|make sure)\b/i,
-        ],
-        confidence: 0.8,
-        roles: ["user"],
-    },
-    // Patterns — "in this project we", "our convention is"
-    {
-        type: "pattern",
-        patterns: [
-            /\b(?:in this (?:project|repo|codebase)|our (?:convention|standard|pattern|practice))\b/i,
-            /\bwe (?:usually|typically|always|normally)\b/i,
-            /\bthe (?:convention|standard|pattern|practice) (?:here |in this )?is\b/i,
-        ],
-        confidence: 0.7,
-        roles: ["user"],
-    },
-    // Topology — "X is in", "you can find X at"
-    {
-        type: "topology",
-        patterns: [
-            /\b(?:you(?:'ll| can| will)? find|(?:it|that|the \w+) (?:is|lives|sits) (?:in|at|under))\b.*(?:src\/|lib\/|config\/|\.\w+)/i,
-            /\b(?:the (?:config|settings|env|database|api|routes?) (?:is|are|lives?) (?:in|at))\b/i,
-        ],
-        confidence: 0.7,
-        roles: ["user"],
-    },
-];
-export function extractMemories(turns) {
-    const extracted = [];
-    for (const turn of turns) {
-        const text = turn.content.trim();
-        // Skip questions — they express uncertainty, not assertions
-        if (text.endsWith("?"))
-            continue;
-        // Skip very short messages — not enough signal
-        if (text.length < 15)
-            continue;
-        for (const pattern of EXTRACTION_PATTERNS) {
-            if (!pattern.roles.includes(turn.role))
-                continue;
-            for (const regex of pattern.patterns) {
-                if (regex.test(text)) {
-                    if (extracted.some(e => e.content === text))
-                        break;
-                    extracted.push({
-                        content: text,
-                        type: pattern.type,
-                        confidence: pattern.confidence,
-                        tags: ["auto-extracted"],
-                        source: "conversation-extractor",
-                    });
-                    break;
-                }
-            }
-        }
-    }
-    return extracted;
-}
-//# sourceMappingURL=extractor.js.map

package/dist/extractor.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"extractor.js","sourceRoot":"","sources":["../src/extractor.ts"],"names":[],"mappings":"AAsBA,MAAM,mBAAmB,GAAc;IACrC,kEAAkE;IAClE;QACE,IAAI,EAAE,YAAY;QAClB,QAAQ,EAAE;YACR,0GAA0G;YAC1G,sFAAsF;YACtF,4CAA4C;YAC5C,uDAAuD;SACxD;QACD,UAAU,EAAE,IAAI;QAChB,KAAK,EAAE,CAAC,MAAM,CAAC;KAChB;IACD,+DAA+D;IAC/D;QACE,IAAI,EAAE,UAAU;QAChB,QAAQ,EAAE;YACR,wEAAwE;YACxE,qDAAqD;YACrD,kFAAkF;SACnF;QACD,UAAU,EAAE,IAAI;QAChB,KAAK,EAAE,CAAC,MAAM,CAAC;KAChB;IACD,kDAAkD;IAClD;QACE,IAAI,EAAE,YAAY;QAClB,QAAQ,EAAE;YACR,kDAAkD;YAClD,qDAAqD;YACrD,2CAA2C;SAC5C;QACD,UAAU,EAAE,GAAG;QACf,KAAK,EAAE,CAAC,MAAM,CAAC;KAChB;IACD,uDAAuD;IACvD;QACE,IAAI,EAAE,SAAS;QACf,QAAQ,EAAE;YACR,yFAAyF;YACzF,+CAA+C;YAC/C,yEAAyE;SAC1E;QACD,UAAU,EAAE,GAAG;QACf,KAAK,EAAE,CAAC,MAAM,CAAC;KAChB;IACD,4CAA4C;IAC5C;QACE,IAAI,EAAE,UAAU;QAChB,QAAQ,EAAE;YACR,4HAA4H;YAC5H,uFAAuF;SACxF;QACD,UAAU,EAAE,GAAG;QACf,KAAK,EAAE,CAAC,MAAM,CAAC;KAChB;CACF,CAAC;AAEF,MAAM,UAAU,eAAe,CAAC,KAAyB;IACvD,MAAM,SAAS,GAAsB,EAAE,CAAC;IAExC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,MAAM,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC;QACjC,4DAA4D;QAC5D,IAAI,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC;YAAE,SAAS;QACjC,+CAA+C;QAC/C,IAAI,IAAI,CAAC,MAAM,GAAG,EAAE;YAAE,SAAS;QAE/B,KAAK,MAAM,OAAO,IAAI,mBAAmB,EAAE,CAAC;YAC1C,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC;gBAAE,SAAS;YAEjD,KAAK,MAAM,KAAK,IAAI,OAAO,CAAC,QAAQ,EAAE,CAAC;gBACrC,IAAI,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;oBACrB,IAAI,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,KAAK,IAAI,CAAC;wBAAE,MAAM;oBAEnD,SAAS,CAAC,IAAI,CAAC;wBACb,OAAO,EAAE,IAAI;wBACb,IAAI,EAAE,OAAO,CAAC,IAAI;wBAClB,UAAU,EAAE,OAAO,CAAC,UAAU;wBAC9B,IAAI,EAAE,CAAC,gBAAgB,CAAC;wBACxB,MAAM,EAAE,wBAAwB;qBACjC,CAAC,CAAC;oBACH,MAAM;gBACR,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,SAAS,CAAC;AACnB,CAAC"}

package/dist/memory.d.ts

@@ -1,133 +0,0 @@
-import type { AmemDatabase } from "./database.js";
-import { VectorIndex } from "./ann.js";
-export declare const MemoryType: {
-    readonly CORRECTION: "correction";
-    readonly DECISION: "decision";
-    readonly PATTERN: "pattern";
-    readonly PREFERENCE: "preference";
-    readonly TOPOLOGY: "topology";
-    readonly FACT: "fact";
-};
-export type MemoryTypeValue = (typeof MemoryType)[keyof typeof MemoryType];
-export declare const IMPORTANCE_WEIGHTS: Record<MemoryTypeValue, number>;
-export interface Memory {
-    id: string;
-    content: string;
-    type: MemoryTypeValue;
-    tags: string[];
-    confidence: number;
-    accessCount: number;
-    createdAt: number;
-    lastAccessed: number;
-    source: string;
-    embedding: Float32Array | null;
-    scope: string;
-    validFrom: number;
-    validUntil: number | null;
-    tier: 'core' | 'working' | 'archival';
-}
-export interface ScoreInput {
-    relevance: number;
-    confidence: number;
-    lastAccessed: number;
-    importance: number;
-    now: number;
-}
-export declare function computeScore(input: ScoreInput): number;
-export interface ConflictResult {
-    isConflict: boolean;
-    similarity: number;
-}
-export declare function detectConflict(newContent: string, existingContent: string, similarity: number): ConflictResult;
-export interface RecallOptions {
-    query: string | null;
-    queryEmbedding?: Float32Array | null;
-    limit: number;
-    type?: MemoryTypeValue;
-    tag?: string;
-    minConfidence?: number;
-    scope?: string;
-    explain?: boolean;
-    /** Filter out expired memories (valid_until < now). Default: true */
-    filterExpired?: boolean;
-    /** Only return memories from this tier */
-    tier?: Memory["tier"];
-}
-export interface RecalledMemory extends Memory {
-    score: number;
-}
-export interface ScoreExplanation {
-    relevance: number;
-    relevanceSource: "semantic" | "keyword" | "default";
-    recency: number;
-    hoursSinceAccess: number;
-    confidence: number;
-    importance: number;
-    importanceLabel: string;
-    finalScore: number;
-}
-export interface ExplainedMemory extends RecalledMemory {
-    explanation: ScoreExplanation;
-}
-export declare function recallMemories(db: AmemDatabase, options: RecallOptions): (RecalledMemory | ExplainedMemory)[];
-export interface ConsolidationOptions {
-    maxStaleDays: number;
-    minConfidence: number;
-    minAccessCount: number;
-    dryRun: boolean;
-    enableDecay?: boolean;
-    decayFactor?: number;
-}
-export interface ConsolidationAction {
-    action: "merged" | "pruned" | "promoted" | "decayed";
-    memoryIds: string[];
-    description: string;
-}
-export interface ConsolidationReport {
-    merged: number;
-    pruned: number;
-    promoted: number;
-    decayed: number;
-    actions: ConsolidationAction[];
-    healthScore: number;
-    before: {
-        total: number;
-    };
-    after: {
-        total: number;
-    };
-}
-export declare function consolidateMemories(db: AmemDatabase, cosineSim: (a: Float32Array, b: Float32Array) => number, options: ConsolidationOptions): ConsolidationReport;
-export declare function buildVectorIndex(db: AmemDatabase): VectorIndex;
-export declare function getVectorIndex(): VectorIndex | null;
-export interface MultiStrategyOptions {
-    query: string;
-    queryEmbedding: Float32Array | null;
-    limit: number;
-    scope?: string;
-    weights?: {
-        semantic: number;
-        fts: number;
-        graph: number;
-        temporal: number;
-    };
-    /** Enable cross-encoder reranking as the final pass. Default: false (uses config). */
-    rerank?: boolean;
-    /** How many candidates to feed to the reranker. Default: 20. */
-    rerankerTopK?: number;
-}
-/**
- * Multi-strategy retrieval: combines semantic search, FTS5, knowledge graph traversal,
- * and temporal recency into a unified ranking. Each strategy votes independently,
- * then scores are merged with configurable weights.
- */
-export declare function multiStrategyRecall(db: AmemDatabase, options: MultiStrategyOptions): Promise<RecalledMemory[]>;
-/**
- * When storing a new memory that contradicts an existing one,
- * auto-expire the old memory instead of requiring manual consolidation.
- */
-export declare function autoExpireContradictions(db: AmemDatabase, newContent: string, newEmbedding: Float32Array | null, newType: MemoryTypeValue): {
-    expired: string[];
-    reason: string;
-};
-export { sanitizeContent } from "./config.js";