memorix 1.2.1 → 1.2.2

package/src/embedding/fastembed-provider.ts

@@ -1,142 +1,142 @@
-/**
- * FastEmbed Provider
- *
- * Local ONNX-based embedding using fastembed (Qdrant).
- * Model: BAAI/bge-small-en-v1.5 (384 dimensions, ~30MB)
- *
- * This is an optional dependency — if fastembed is not installed,
- * the provider module gracefully falls back to fulltext-only search.
- *
- * Persistent disk cache: embeddings are saved to ~/.memorix/data/.embedding-cache.json
- * so server restarts don't need to regenerate them (saves minutes of CPU on 500+ obs).
- */
-
-import { createHash } from 'node:crypto';
-import { readFile, writeFile, mkdir } from 'node:fs/promises';
-import { join } from 'node:path';
-import { homedir } from 'node:os';
-import type { EmbeddingProvider } from './provider.js';
-
-const CACHE_DIR = process.env.MEMORIX_DATA_DIR || join(homedir(), '.memorix', 'data');
-const CACHE_FILE = join(CACHE_DIR, '.embedding-cache.json');
-
-// In-memory cache keyed by text hash → embedding
-const cache = new Map<string, number[]>();
-const MAX_CACHE_SIZE = 5000;
-let diskCacheDirty = false;
-
-function textHash(text: string): string {
-  return createHash('sha256').update(text).digest('hex').slice(0, 16);
-}
-
-async function loadDiskCache(): Promise<void> {
-  try {
-    const raw = await readFile(CACHE_FILE, 'utf-8');
-    const entries: [string, number[]][] = JSON.parse(raw);
-    for (const [k, v] of entries) cache.set(k, v);
-    console.error(`[memorix] Loaded ${entries.length} cached embeddings from disk`);
-  } catch {
-    // No cache file or corrupt — start fresh
-  }
-}
-
-async function saveDiskCache(): Promise<void> {
-  if (!diskCacheDirty) return;
-  try {
-    await mkdir(CACHE_DIR, { recursive: true });
-    const entries = Array.from(cache.entries());
-    await writeFile(CACHE_FILE, JSON.stringify(entries));
-    diskCacheDirty = false;
-  } catch {
-    // Ignore write errors — cache is an optimization, not critical
-  }
-}
-
-export class FastEmbedProvider implements EmbeddingProvider {
-  readonly name = 'fastembed-bge-small';
-  readonly dimensions = 384;
-
-  private model: { embed: (docs: string[], batchSize?: number) => AsyncGenerator<number[][]>; queryEmbed: (query: string) => Promise<number[]> };
-
-  private constructor(model: FastEmbedProvider['model']) {
-    this.model = model;
-  }
-
-  /**
-   * Initialize the FastEmbed provider.
-   * Downloads model on first use (~30MB), cached locally after.
-   * Loads persistent embedding cache from disk.
-   */
-  static async create(): Promise<FastEmbedProvider> {
-    // Dynamic import — throws if fastembed is not installed
-    const { EmbeddingModel, FlagEmbedding } = await import('fastembed');
-    const model = await FlagEmbedding.init({
-      model: EmbeddingModel.BGESmallENV15,
-    });
-    // Load disk cache before returning — subsequent embedBatch calls will hit cache
-    await loadDiskCache();
-    return new FastEmbedProvider(model);
-  }
-
-  async embed(text: string): Promise<number[]> {
-    const hash = textHash(text);
-    const cached = cache.get(hash);
-    if (cached) return cached;
-
-    const raw = await this.model.queryEmbed(text);
-    // Ensure plain number[] (fastembed may return Float32Array)
-    const result = Array.from(raw) as number[];
-    if (result.length !== this.dimensions) {
-      throw new Error(`Expected ${this.dimensions}d embedding, got ${result.length}d`);
-    }
-    this.cacheSet(hash, result);
-    return result;
-  }
-
-  async embedBatch(texts: string[]): Promise<number[][]> {
-    const results: number[][] = new Array(texts.length);
-    const uncachedIndices: number[] = [];
-    const uncachedTexts: string[] = [];
-
-    // Check cache for each text (by hash)
-    for (let i = 0; i < texts.length; i++) {
-      const hash = textHash(texts[i]);
-      const cached = cache.get(hash);
-      if (cached) {
-        results[i] = cached;
-      } else {
-        uncachedIndices.push(i);
-        uncachedTexts.push(texts[i]);
-      }
-    }
-
-    // Batch embed uncached texts
-    if (uncachedTexts.length > 0) {
-      console.error(`[memorix] Embedding ${uncachedTexts.length}/${texts.length} uncached texts (${texts.length - uncachedTexts.length} from cache)`);
-      let batchIdx = 0;
-      for await (const batch of this.model.embed(uncachedTexts, 64)) {
-        for (const vec of batch) {
-          const originalIdx = uncachedIndices[batchIdx];
-          const plain = Array.from(vec) as number[];
-          results[originalIdx] = plain;
-          this.cacheSet(textHash(uncachedTexts[batchIdx]), plain);
-          batchIdx++;
-        }
-      }
-      // Persist cache to disk after batch operations
-      await saveDiskCache();
-    }
-
-    return results;
-  }
-
-  private cacheSet(hash: string, value: number[]): void {
-    // Evict oldest entries if cache is full
-    if (cache.size >= MAX_CACHE_SIZE) {
-      const firstKey = cache.keys().next().value;
-      if (firstKey !== undefined) cache.delete(firstKey);
-    }
-    cache.set(hash, value);
-    diskCacheDirty = true;
-  }
-}
+/**
+ * FastEmbed Provider
+ *
+ * Local ONNX-based embedding using fastembed (Qdrant).
+ * Model: BAAI/bge-small-en-v1.5 (384 dimensions, ~30MB)
+ *
+ * This is an optional dependency — if fastembed is not installed,
+ * the provider module gracefully falls back to fulltext-only search.
+ *
+ * Persistent disk cache: embeddings are saved to ~/.memorix/data/.embedding-cache.json
+ * so server restarts don't need to regenerate them (saves minutes of CPU on 500+ obs).
+ */
+
+import { createHash } from 'node:crypto';
+import { readFile, writeFile, mkdir } from 'node:fs/promises';
+import { join } from 'node:path';
+import { homedir } from 'node:os';
+import type { EmbeddingProvider } from './provider.js';
+
+const CACHE_DIR = process.env.MEMORIX_DATA_DIR || join(homedir(), '.memorix', 'data');
+const CACHE_FILE = join(CACHE_DIR, '.embedding-cache.json');
+
+// In-memory cache keyed by text hash → embedding
+const cache = new Map<string, number[]>();
+const MAX_CACHE_SIZE = 5000;
+let diskCacheDirty = false;
+
+function textHash(text: string): string {
+  return createHash('sha256').update(text).digest('hex').slice(0, 16);
+}
+
+async function loadDiskCache(): Promise<void> {
+  try {
+    const raw = await readFile(CACHE_FILE, 'utf-8');
+    const entries: [string, number[]][] = JSON.parse(raw);
+    for (const [k, v] of entries) cache.set(k, v);
+    console.error(`[memorix] Loaded ${entries.length} cached embeddings from disk`);
+  } catch {
+    // No cache file or corrupt — start fresh
+  }
+}
+
+async function saveDiskCache(): Promise<void> {
+  if (!diskCacheDirty) return;
+  try {
+    await mkdir(CACHE_DIR, { recursive: true });
+    const entries = Array.from(cache.entries());
+    await writeFile(CACHE_FILE, JSON.stringify(entries));
+    diskCacheDirty = false;
+  } catch {
+    // Ignore write errors — cache is an optimization, not critical
+  }
+}
+
+export class FastEmbedProvider implements EmbeddingProvider {
+  readonly name = 'fastembed-bge-small';
+  readonly dimensions = 384;
+
+  private model: { embed: (docs: string[], batchSize?: number) => AsyncGenerator<number[][]>; queryEmbed: (query: string) => Promise<number[]> };
+
+  private constructor(model: FastEmbedProvider['model']) {
+    this.model = model;
+  }
+
+  /**
+   * Initialize the FastEmbed provider.
+   * Downloads model on first use (~30MB), cached locally after.
+   * Loads persistent embedding cache from disk.
+   */
+  static async create(): Promise<FastEmbedProvider> {
+    // Dynamic import — throws if fastembed is not installed
+    const { EmbeddingModel, FlagEmbedding } = await import('fastembed');
+    const model = await FlagEmbedding.init({
+      model: EmbeddingModel.BGESmallENV15,
+    });
+    // Load disk cache before returning — subsequent embedBatch calls will hit cache
+    await loadDiskCache();
+    return new FastEmbedProvider(model);
+  }
+
+  async embed(text: string): Promise<number[]> {
+    const hash = textHash(text);
+    const cached = cache.get(hash);
+    if (cached) return cached;
+
+    const raw = await this.model.queryEmbed(text);
+    // Ensure plain number[] (fastembed may return Float32Array)
+    const result = Array.from(raw) as number[];
+    if (result.length !== this.dimensions) {
+      throw new Error(`Expected ${this.dimensions}d embedding, got ${result.length}d`);
+    }
+    this.cacheSet(hash, result);
+    return result;
+  }
+
+  async embedBatch(texts: string[]): Promise<number[][]> {
+    const results: number[][] = new Array(texts.length);
+    const uncachedIndices: number[] = [];
+    const uncachedTexts: string[] = [];
+
+    // Check cache for each text (by hash)
+    for (let i = 0; i < texts.length; i++) {
+      const hash = textHash(texts[i]);
+      const cached = cache.get(hash);
+      if (cached) {
+        results[i] = cached;
+      } else {
+        uncachedIndices.push(i);
+        uncachedTexts.push(texts[i]);
+      }
+    }
+
+    // Batch embed uncached texts
+    if (uncachedTexts.length > 0) {
+      console.error(`[memorix] Embedding ${uncachedTexts.length}/${texts.length} uncached texts (${texts.length - uncachedTexts.length} from cache)`);
+      let batchIdx = 0;
+      for await (const batch of this.model.embed(uncachedTexts, 64)) {
+        for (const vec of batch) {
+          const originalIdx = uncachedIndices[batchIdx];
+          const plain = Array.from(vec) as number[];
+          results[originalIdx] = plain;
+          this.cacheSet(textHash(uncachedTexts[batchIdx]), plain);
+          batchIdx++;
+        }
+      }
+      // Persist cache to disk after batch operations
+      await saveDiskCache();
+    }
+
+    return results;
+  }
+
+  private cacheSet(hash: string, value: number[]): void {
+    // Evict oldest entries if cache is full
+    if (cache.size >= MAX_CACHE_SIZE) {
+      const firstKey = cache.keys().next().value;
+      if (firstKey !== undefined) cache.delete(firstKey);
+    }
+    cache.set(hash, value);
+    diskCacheDirty = true;
+  }
+}

package/src/embedding/transformers-provider.ts

@@ -1,111 +1,111 @@
-/**
- * Transformers.js Provider
- *
- * Pure JavaScript embedding using @huggingface/transformers (HuggingFace).
- * Model: Xenova/all-MiniLM-L6-v2 (384 dimensions, ~22MB quantized)
- *
- * Key advantages over fastembed:
- *   - No native ONNX binding required (pure JS / WASM)
- *   - Works out-of-the-box on Windows, macOS, Linux
- *   - Supports quantized models (q8, q4) for smaller footprint
- *
- * This is an optional dependency — if @huggingface/transformers is not
- * installed, the provider module gracefully falls back to the next option.
- *
- * Inspired by Mem0's multi-provider embedding architecture.
- */
-
-import type { EmbeddingProvider } from './provider.js';
-
-// In-memory LRU cache
-const cache = new Map<string, number[]>();
-const MAX_CACHE_SIZE = 5000;
-
-export class TransformersProvider implements EmbeddingProvider {
-    readonly name = 'transformers-minilm';
-    readonly dimensions = 384;
-
-    private extractor: any; // Pipeline instance
-
-    private constructor(extractor: any) {
-        this.extractor = extractor;
-    }
-
-    /**
-     * Initialize the Transformers.js provider.
-     * Downloads model on first use (~22MB quantized), cached locally after.
-     */
-    static async create(): Promise<TransformersProvider> {
-        // Dynamic import — throws if @huggingface/transformers is not installed
-        const { pipeline } = await import('@huggingface/transformers');
-        const extractor = await pipeline(
-            'feature-extraction',
-            'Xenova/all-MiniLM-L6-v2',
-            { dtype: 'q8' }, // Quantized for small footprint
-        );
-        return new TransformersProvider(extractor);
-    }
-
-    async embed(text: string): Promise<number[]> {
-        // Check cache first
-        const cached = cache.get(text);
-        if (cached) return cached;
-
-        const output = await this.extractor(text, {
-            pooling: 'mean',
-            normalize: true,
-        });
-
-        // output.tolist() returns [[...384 floats]]
-        const result: number[] = Array.from(output.tolist()[0]);
-        if (result.length !== this.dimensions) {
-            throw new Error(`Expected ${this.dimensions}d embedding, got ${result.length}d`);
-        }
-
-        this.cacheSet(text, result);
-        return result;
-    }
-
-    async embedBatch(texts: string[]): Promise<number[][]> {
-        const results: number[][] = new Array(texts.length);
-        const uncachedIndices: number[] = [];
-        const uncachedTexts: string[] = [];
-
-        // Check cache for each text
-        for (let i = 0; i < texts.length; i++) {
-            const cached = cache.get(texts[i]);
-            if (cached) {
-                results[i] = cached;
-            } else {
-                uncachedIndices.push(i);
-                uncachedTexts.push(texts[i]);
-            }
-        }
-
-        // Batch embed uncached texts
-        if (uncachedTexts.length > 0) {
-            const output = await this.extractor(uncachedTexts, {
-                pooling: 'mean',
-                normalize: true,
-            });
-            const allVecs: number[][] = output.tolist();
-
-            for (let i = 0; i < allVecs.length; i++) {
-                const vec = Array.from(allVecs[i]) as number[];
-                const originalIdx = uncachedIndices[i];
-                results[originalIdx] = vec;
-                this.cacheSet(uncachedTexts[i], vec);
-            }
-        }
-
-        return results;
-    }
-
-    private cacheSet(key: string, value: number[]): void {
-        if (cache.size >= MAX_CACHE_SIZE) {
-            const firstKey = cache.keys().next().value;
-            if (firstKey !== undefined) cache.delete(firstKey);
-        }
-        cache.set(key, value);
-    }
-}
+/**
+ * Transformers.js Provider
+ *
+ * Pure JavaScript embedding using @huggingface/transformers (HuggingFace).
+ * Model: Xenova/all-MiniLM-L6-v2 (384 dimensions, ~22MB quantized)
+ *
+ * Key advantages over fastembed:
+ *   - No native ONNX binding required (pure JS / WASM)
+ *   - Works out-of-the-box on Windows, macOS, Linux
+ *   - Supports quantized models (q8, q4) for smaller footprint
+ *
+ * This is an optional dependency — if @huggingface/transformers is not
+ * installed, the provider module gracefully falls back to the next option.
+ *
+ * Inspired by Mem0's multi-provider embedding architecture.
+ */
+
+import type { EmbeddingProvider } from './provider.js';
+
+// In-memory LRU cache
+const cache = new Map<string, number[]>();
+const MAX_CACHE_SIZE = 5000;
+
+export class TransformersProvider implements EmbeddingProvider {
+    readonly name = 'transformers-minilm';
+    readonly dimensions = 384;
+
+    private extractor: any; // Pipeline instance
+
+    private constructor(extractor: any) {
+        this.extractor = extractor;
+    }
+
+    /**
+     * Initialize the Transformers.js provider.
+     * Downloads model on first use (~22MB quantized), cached locally after.
+     */
+    static async create(): Promise<TransformersProvider> {
+        // Dynamic import — throws if @huggingface/transformers is not installed
+        const { pipeline } = await import('@huggingface/transformers');
+        const extractor = await pipeline(
+            'feature-extraction',
+            'Xenova/all-MiniLM-L6-v2',
+            { dtype: 'q8' }, // Quantized for small footprint
+        );
+        return new TransformersProvider(extractor);
+    }
+
+    async embed(text: string): Promise<number[]> {
+        // Check cache first
+        const cached = cache.get(text);
+        if (cached) return cached;
+
+        const output = await this.extractor(text, {
+            pooling: 'mean',
+            normalize: true,
+        });
+
+        // output.tolist() returns [[...384 floats]]
+        const result: number[] = Array.from(output.tolist()[0]);
+        if (result.length !== this.dimensions) {
+            throw new Error(`Expected ${this.dimensions}d embedding, got ${result.length}d`);
+        }
+
+        this.cacheSet(text, result);
+        return result;
+    }
+
+    async embedBatch(texts: string[]): Promise<number[][]> {
+        const results: number[][] = new Array(texts.length);
+        const uncachedIndices: number[] = [];
+        const uncachedTexts: string[] = [];
+
+        // Check cache for each text
+        for (let i = 0; i < texts.length; i++) {
+            const cached = cache.get(texts[i]);
+            if (cached) {
+                results[i] = cached;
+            } else {
+                uncachedIndices.push(i);
+                uncachedTexts.push(texts[i]);
+            }
+        }
+
+        // Batch embed uncached texts
+        if (uncachedTexts.length > 0) {
+            const output = await this.extractor(uncachedTexts, {
+                pooling: 'mean',
+                normalize: true,
+            });
+            const allVecs: number[][] = output.tolist();
+
+            for (let i = 0; i < allVecs.length; i++) {
+                const vec = Array.from(allVecs[i]) as number[];
+                const originalIdx = uncachedIndices[i];
+                results[originalIdx] = vec;
+                this.cacheSet(uncachedTexts[i], vec);
+            }
+        }
+
+        return results;
+    }
+
+    private cacheSet(key: string, value: number[]): void {
+        if (cache.size >= MAX_CACHE_SIZE) {
+            const firstKey = cache.keys().next().value;
+            if (firstKey !== undefined) cache.delete(firstKey);
+        }
+        cache.set(key, value);
+    }
+}