@alex900530/claude-persistent-memory 1.0.7 → 1.0.8

package/bin/setup.js

@@ -161,9 +161,9 @@ function setupHooks() {
   ok('  Hooks configured (5 hooks).');
 }
 
-// ── [4/5] .gitignore ────────────────────────────────────────────────────────
+// ── [4/6] .gitignore ────────────────────────────────────────────────────────
 function setupGitignore() {
-  log('[4/5] Updating .gitignore ...');
+  log('[4/6] Updating .gitignore ...');
 
   const entries = ['.claude-memory/', '.claude-memory.config.js', '.claude/', '.mcp.json'];
   let content = '';
@@ -187,9 +187,43 @@ function setupGitignore() {
   }
 }
 
-// ── [5/5] Services ──────────────────────────────────────────────────────────
+// ── [5/6] Pre-download embedding model ───────────────────────────────────────
+async function setupModel() {
+  log('[5/6] Preparing embedding model (bge-m3) ...');
+
+  const os = require('os');
+  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).`);
+      return;
+    }
+    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 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.');
+  }
+}
+
+// ── [6/6] Services ──────────────────────────────────────────────────────────
 function setupServices() {
-  log('[5/5] Setting up background services ...');
+  log('[6/6] Setting up background services ...');
 
   // Check if Azure credentials are configured
   let hasCredentials = false;
@@ -411,6 +445,7 @@ async function main() {
   setupMCP();
   setupHooks();
   setupGitignore();
+  await setupModel();
   setupServices();
 }
 

package/package.json

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