@alex900530/claude-persistent-memory 1.1.0 → 1.1.1

package/bin/setup.js

@@ -195,29 +195,46 @@ async function setupModel() {
   const cacheDir = path.join(os.homedir(), '.cache', 'huggingface', 'transformers-js');
   const modelDir = path.join(cacheDir, 'Xenova', 'bge-m3', 'onnx');
 
-  // Check if model is already cached
-  if (fs.existsSync(path.join(modelDir, 'model.onnx_data'))) {
-    const stat = fs.statSync(path.join(modelDir, 'model.onnx_data'));
-    if (stat.size > 300 * 1024 * 1024) { // > 300MB = likely complete
-      ok(`  Model already cached (${(stat.size / 1024 / 1024 / 1024).toFixed(1)}GB).`);
+  const modelFile = path.join(modelDir, 'model.onnx_data');
+  const modelBaseDir = path.join(cacheDir, 'Xenova', 'bge-m3');
+
+  // Helper: import transformers.js (handles both ESM and CJS re-export)
+  async function loadTransformers() {
+    const transformersPath = require.resolve('@huggingface/transformers', { paths: [PKG_DIR] });
+    const mod = await import(transformersPath);
+    const pipeline = mod.pipeline || mod.default?.pipeline;
+    const env = mod.env || mod.default?.env;
+    if (env) env.cacheDir = cacheDir;
+    return { pipeline, env };
+  }
+
+  // Check if model is already cached — verify by running inference
+  if (fs.existsSync(modelFile)) {
+    try {
+      const { pipeline } = await loadTransformers();
+      const extractor = await pipeline('feature-extraction', 'Xenova/bge-m3', { device: 'cpu' });
+      await extractor('test', { pooling: 'cls', normalize: true });
+      const stat = fs.statSync(modelFile);
+      ok(`  Model already cached and verified (${(stat.size / 1024 / 1024 / 1024).toFixed(1)}GB).`);
       return;
+    } catch (e) {
+      warn(`  Cached model is corrupt or incomplete, cleaning up ...`);
+      try { fs.rmSync(modelBaseDir, { recursive: true, force: true }); } catch {}
     }
-    warn('  Incomplete model cache detected, re-downloading ...');
   }
 
   log('  Downloading model (~2GB, this may take a few minutes) ...');
   try {
-    // Resolve from package's own dependencies
-    const transformersPath = require.resolve('@huggingface/transformers', { paths: [PKG_DIR] });
-    const { pipeline, env } = await import(transformersPath);
-    env.cacheDir = cacheDir;
+    const { pipeline } = await loadTransformers();
     const extractor = await pipeline('feature-extraction', 'Xenova/bge-m3', { device: 'cpu' });
-    // Quick test to verify model works
     await extractor('test', { pooling: 'cls', normalize: true });
     ok('  Model downloaded and verified.');
   } catch (e) {
-    warn(`  Model download failed: ${e.message}`);
-    warn('  The embedding server will retry on first start.');
+    // Clean up partial download
+    try { fs.rmSync(modelBaseDir, { recursive: true, force: true }); } catch {}
+    console.error(`${PREFIX} \x1b[31mModel download failed: ${e.message}\x1b[0m`);
+    console.error(`${PREFIX} Please retry: npm install @alex900530/claude-persistent-memory`);
+    process.exit(1);
   }
 }
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@alex900530/claude-persistent-memory",
-  "version": "1.1.0",
+  "version": "1.1.1",
   "description": "Persistent memory system for Claude Code — hybrid BM25 + vector search, LLM-driven structuring, automatic clustering",
   "main": "lib/memory-db.js",
   "bin": {