claude-memory-layer 1.0.20 → 1.0.21

package/dist/core/index.js

@@ -91,8 +91,8 @@ var ConfigSchema = z.object({
   }).default({}),
   embedding: z.object({
     provider: z.enum(["local", "openai"]).default("local"),
-    model: z.string().default("jinaai/jina-embeddings-v5-text-nano"),
-    openaiModel: z.string().default("jinaai/jina-embeddings-v5-text-nano"),
+    model: z.string().default("jinaai/jina-embeddings-v5-text-nano-text-matching"),
+    openaiModel: z.string().default("jinaai/jina-embeddings-v5-text-nano-text-matching"),
     batchSize: z.number().default(32)
   }).default({}),
   retrieval: z.object({
@@ -3996,13 +3996,15 @@ var VectorStore = class {
 };
 
 // src/core/embedder.ts
-import { pipeline } from "@xenova/transformers";
+import { pipeline } from "@huggingface/transformers";
 var Embedder = class {
   pipeline = null;
   modelName;
+  activeModelName;
   initialized = false;
-  constructor(modelName = "jinaai/jina-embeddings-v5-text-nano") {
+  constructor(modelName = "jinaai/jina-embeddings-v5-text-nano-text-matching") {
     this.modelName = modelName;
+    this.activeModelName = modelName;
   }
   /**
    * Initialize the embedding pipeline
@@ -4010,8 +4012,21 @@ var Embedder = class {
   async initialize() {
     if (this.initialized)
       return;
-    this.pipeline = await pipeline("feature-extraction", this.modelName);
-    this.initialized = true;
+    try {
+      this.pipeline = await pipeline("feature-extraction", this.modelName);
+      this.activeModelName = this.modelName;
+      this.initialized = true;
+      return;
+    } catch (primaryError) {
+      const fallbackModel = process.env.CLAUDE_MEMORY_EMBEDDING_FALLBACK_MODEL || "onnx-community/embeddinggemma-300m-ONNX";
+      if (fallbackModel === this.modelName) {
+        throw primaryError;
+      }
+      console.warn(`[Embedder] Primary model failed (${this.modelName}). Falling back to ${fallbackModel}`);
+      this.pipeline = await pipeline("feature-extraction", fallbackModel);
+      this.activeModelName = fallbackModel;
+      this.initialized = true;
+    }
   }
   /**
    * Generate embedding for a single text
@@ -4028,7 +4043,7 @@ var Embedder = class {
     const vector = Array.from(output.data);
     return {
       vector,
-      model: this.modelName,
+      model: this.activeModelName,
       dimensions: vector.length
     };
   }
@@ -4052,7 +4067,7 @@ var Embedder = class {
         const vector = Array.from(output.data);
         results.push({
           vector,
-          model: this.modelName,
+          model: this.activeModelName,
           dimensions: vector.length
         });
       }
@@ -4076,7 +4091,7 @@ var Embedder = class {
    * Get model name
    */
   getModelName() {
-    return this.modelName;
+    return this.activeModelName;
   }
 };
 var defaultEmbedder = null;