mem0ai 2.4.6 → 3.0.0-beta.0

package/dist/oss/index.mjs

@@ -37,8 +37,7 @@ var MemoryConfigSchema = z.object({
     })
   }),
   historyDbPath: z.string().optional(),
-  customPrompt: z.string().optional(),
-  enableGraph: z.boolean().optional(),
+  customInstructions: z.string().optional(),
   graphStore: z.object({
     provider: z.string(),
     config: z.object({
@@ -50,7 +49,7 @@ var MemoryConfigSchema = z.object({
       provider: z.string(),
       config: z.record(z.string(), z.any())
     }).optional(),
-    customPrompt: z.string().optional()
+    customInstructions: z.string().optional()
   }).optional(),
   historyStore: z.object({
     provider: z.string(),
@@ -562,7 +561,7 @@ var MemoryVectorStore = class {
     );
     insertMany(vectors, ids, payloads);
   }
-  async search(query, limit = 10, filters) {
+  async search(query, topK = 10, filters) {
     if (query.length !== this.dimension) {
       throw new Error(
         `Query dimension mismatch. Expected ${this.dimension}, got ${query.length}`
@@ -592,7 +591,7 @@ var MemoryVectorStore = class {
       }
     }
     results.sort((a, b) => (b.score || 0) - (a.score || 0));
-    return results.slice(0, limit);
+    return results.slice(0, topK);
   }
   async get(vectorId) {
     const row = this.db.prepare(`SELECT * FROM vectors WHERE id = ?`).get(vectorId);
@@ -619,7 +618,7 @@ var MemoryVectorStore = class {
     this.db.exec(`DROP TABLE IF EXISTS vectors`);
     this.init();
   }
-  async list(filters, limit = 100) {
+  async list(filters, topK = 100) {
     const rows = this.db.prepare(`SELECT * FROM vectors`).all();
     const results = [];
     for (const row of rows) {
@@ -642,7 +641,7 @@ var MemoryVectorStore = class {
         });
       }
     }
-    return [results.slice(0, limit), results.length];
+    return [results.slice(0, topK), results.length];
   }
   async getUserId() {
     const row = this.db.prepare(`SELECT user_id FROM memory_migrations LIMIT 1`).get();
@@ -734,12 +733,12 @@ var Qdrant = class {
       points
     });
   }
-  async search(query, limit = 5, filters) {
+  async search(query, topK = 5, filters) {
     const queryFilter = this.createFilter(filters);
     const results = await this.client.search(this.collectionName, {
       vector: query,
       filter: queryFilter,
-      limit
+      limit: topK
     });
     return results.map((hit) => ({
       id: String(hit.id),
@@ -776,9 +775,9 @@ var Qdrant = class {
   async deleteCol() {
     await this.client.deleteCollection(this.collectionName);
   }
-  async list(filters, limit = 100) {
+  async list(filters, topK = 100) {
     const scrollRequest = {
-      limit,
+      limit: topK,
       filter: this.createFilter(filters),
       with_payload: true,
       with_vectors: false
@@ -948,7 +947,7 @@ var VectorizeDB = class {
       );
     }
   }
-  async search(query, limit = 5, filters) {
+  async search(query, topK = 5, filters) {
     var _a2, _b;
     try {
       const result = await ((_a2 = this.client) == null ? void 0 : _a2.vectorize.indexes.query(
@@ -958,7 +957,7 @@ var VectorizeDB = class {
           vector: query,
           filter: filters,
           returnMetadata: "all",
-          topK: limit
+          topK
         }
       ));
       return ((_b = result == null ? void 0 : result.matches) == null ? void 0 : _b.map((match) => ({
@@ -1055,7 +1054,7 @@ var VectorizeDB = class {
       );
     }
   }
-  async list(filters, limit = 20) {
+  async list(filters, topK = 20) {
     var _a2, _b;
     try {
       const result = await ((_a2 = this.client) == null ? void 0 : _a2.vectorize.indexes.query(
@@ -1065,7 +1064,7 @@ var VectorizeDB = class {
           vector: Array(this.dimensions).fill(0),
           // Dummy vector for listing
           filter: filters,
-          topK: limit,
+          topK,
           returnMetadata: "all"
         }
       ));
@@ -1506,7 +1505,7 @@ var RedisDB = class {
       throw error;
     }
   }
-  async search(query, limit = 5, filters) {
+  async search(query, topK = 5, filters) {
     const snakeFilters = filters ? toSnakeCase(filters) : void 0;
     const filterExpr = snakeFilters ? Object.entries(snakeFilters).filter(([_, value]) => value !== null).map(([key, value]) => `@${key}:{${value}}`).join(" ") : "*";
     const queryVector = new Float32Array(query).buffer;
@@ -1529,13 +1528,13 @@ var RedisDB = class {
       DIALECT: 2,
       LIMIT: {
         from: 0,
-        size: limit
+        size: topK
       }
     };
     try {
       const results = await this.client.ft.search(
         this.indexName,
-        `${filterExpr} =>[KNN ${limit} @embedding $vec AS __vector_score]`,
+        `${filterExpr} =>[KNN ${topK} @embedding $vec AS __vector_score]`,
         searchOptions
       );
       return results.documents.map((doc) => {
@@ -1700,7 +1699,7 @@ var RedisDB = class {
   async deleteCol() {
     await this.client.ft.dropIndex(this.indexName);
   }
-  async list(filters, limit = 100) {
+  async list(filters, topK = 100) {
     const snakeFilters = filters ? toSnakeCase(filters) : void 0;
     const filterExpr = snakeFilters ? Object.entries(snakeFilters).filter(([_, value]) => value !== null).map(([key, value]) => `@${key}:{${value}}`).join(" ") : "*";
     const searchOptions = {
@@ -1708,7 +1707,7 @@ var RedisDB = class {
       SORTDIR: "DESC",
       LIMIT: {
         from: 0,
-        size: limit
+        size: topK
       }
     };
     const results = await this.client.ft.search(
@@ -1991,11 +1990,11 @@ See the SQL migration instructions in the code comments.`
       throw error;
     }
   }
-  async search(query, limit = 5, filters) {
+  async search(query, topK = 5, filters) {
     try {
       const rpcQuery = {
         query_embedding: query,
-        match_count: limit
+        match_count: topK
       };
       if (filters) {
         rpcQuery.filter = filters;
@@ -2061,9 +2060,9 @@ See the SQL migration instructions in the code comments.`
       throw error;
     }
   }
-  async list(filters, limit = 100) {
+  async list(filters, topK = 100) {
     try {
-      let query = this.client.from(this.tableName).select("*", { count: "exact" }).limit(limit);
+      let query = this.client.from(this.tableName).select("*", { count: "exact" }).limit(topK);
       if (filters) {
         Object.entries(filters).forEach(([key, value]) => {
           query = query.eq(`${this.metadataColumnName}->>${key}`, value);
@@ -3130,7 +3129,7 @@ var LangchainVectorStore = class {
       await this.lcStore.addVectors(vectors, documents);
     }
   }
-  async search(query, limit = 5, filters) {
+  async search(query, topK = 5, filters) {
     if (this.dimension && query.length !== this.dimension) {
       throw new Error(
         `Query vector dimension mismatch. Expected ${this.dimension}, got ${query.length}`
@@ -3138,7 +3137,7 @@ var LangchainVectorStore = class {
     }
     const results = await this.lcStore.similaritySearchVectorWithScore(
       query,
-      limit
+      topK
       // Do not pass lcFilter here
     );
     return results.map(([doc, score]) => ({
@@ -3186,7 +3185,7 @@ var LangchainVectorStore = class {
       );
     }
   }
-  async list(filters, limit = 100) {
+  async list(filters, topK = 100) {
     console.error(
       `LangchainVectorStore: The 'list' method is not supported by the generic LangchainVectorStore wrapper.`
     );
@@ -3411,12 +3410,12 @@ var AzureAISearch = class {
   /**
    * Search for similar vectors
    */
-  async search(query, limit = 5, filters) {
+  async search(query, topK = 5, filters) {
     const filterExpression = filters ? this.buildFilterExpression(filters) : void 0;
     const vectorQuery = {
       kind: "vector",
       vector: query,
-      kNearestNeighborsCount: limit,
+      kNearestNeighborsCount: topK,
       fields: ["vector"]
     };
     let searchResults;
@@ -3427,7 +3426,7 @@ var AzureAISearch = class {
           filterMode: this.vectorFilterMode
         },
         filter: filterExpression,
-        top: limit,
+        top: topK,
         searchFields: ["payload"]
       });
     } else {
@@ -3437,7 +3436,7 @@ var AzureAISearch = class {
           filterMode: this.vectorFilterMode
         },
         filter: filterExpression,
-        top: limit
+        top: topK
       });
     }
     const results = [];
@@ -3543,11 +3542,11 @@ var AzureAISearch = class {
   /**
    * List all vectors in the index
    */
-  async list(filters, limit = 100) {
+  async list(filters, topK = 100) {
     const filterExpression = filters ? this.buildFilterExpression(filters) : void 0;
     const searchResults = await this.searchClient.search("*", {
       filter: filterExpression,
-      top: limit
+      top: topK
     });
     const results = [];
     for await (const result of searchResults.results) {
@@ -3777,10 +3776,10 @@ var PGVector = class {
       )
     );
   }
-  async search(query, limit = 5, filters) {
+  async search(query, topK = 5, filters) {
     const filterConditions = [];
     const queryVector = `[${query.join(",")}]`;
-    const filterValues = [queryVector, limit];
+    const filterValues = [queryVector, topK];
     let filterIndex = 3;
     if (filters) {
       for (const [key, value] of Object.entries(filters)) {
@@ -3843,7 +3842,7 @@ var PGVector = class {
     `);
     return result.rows.map((row) => row.table_name);
   }
-  async list(filters, limit = 100) {
+  async list(filters, topK = 100) {
     const filterConditions = [];
     const filterValues = [];
     let paramIndex = 1;
@@ -3866,7 +3865,7 @@ var PGVector = class {
       FROM ${this.collectionName}
       ${filterClause}
     `;
-    filterValues.push(limit);
+    filterValues.push(topK);
     const [listResult, countResult] = await Promise.all([
       this.client.query(listQuery, filterValues),
       this.client.query(countQuery, filterValues.slice(0, -1))
@@ -3904,6 +3903,38 @@ var PGVector = class {
   }
 };
 
+// src/oss/src/llms/deepseek.ts
+var DeepSeekLLM = class extends OpenAILLM {
+  constructor(config) {
+    const apiKey = config.apiKey || process.env.DEEPSEEK_API_KEY;
+    if (!apiKey) {
+      throw new Error("DeepSeek API key is required");
+    }
+    super({
+      ...config,
+      apiKey,
+      baseURL: config.baseURL || process.env.DEEPSEEK_API_BASE || "https://api.deepseek.com",
+      model: config.model || "deepseek-chat"
+    });
+  }
+  async generateResponse(messages, responseFormat, tools) {
+    try {
+      return await super.generateResponse(messages, responseFormat, tools);
+    } catch (err) {
+      const message = err instanceof Error ? err.message : String(err);
+      throw new Error(`DeepSeek LLM failed: ${message}`);
+    }
+  }
+  async generateChat(messages) {
+    try {
+      return await super.generateChat(messages);
+    } catch (err) {
+      const message = err instanceof Error ? err.message : String(err);
+      throw new Error(`DeepSeek LLM failed: ${message}`);
+    }
+  }
+};
+
 // src/oss/src/utils/factory.ts
 var EmbedderFactory = class {
   static create(provider, config) {
@@ -3950,6 +3981,8 @@ var LLMFactory = class {
         return new MistralLLM(config);
       case "langchain":
         return new LangchainLLM(config);
+      case "deepseek":
+        return new DeepSeekLLM(config);
       default:
         throw new Error(`Unsupported LLM provider: ${provider}`);
     }
@@ -4043,15 +4076,6 @@ var DEFAULT_MEMORY_CONFIG = {
       modelProperties: void 0
     }
   },
-  enableGraph: false,
-  graphStore: {
-    provider: "neo4j",
-    config: {
-      url: process.env.NEO4J_URL || "neo4j://localhost:7687",
-      username: process.env.NEO4J_USERNAME || "neo4j",
-      password: process.env.NEO4J_PASSWORD || "password"
-    }
-  },
   historyStore: {
     provider: "sqlite",
     config: {
@@ -4140,11 +4164,8 @@ var ConfigManager = class {
         })()
       },
       historyDbPath: userConfig.historyDbPath || ((_e = (_d = userConfig.historyStore) == null ? void 0 : _d.config) == null ? void 0 : _e.historyDbPath) || ((_g = (_f = DEFAULT_MEMORY_CONFIG.historyStore) == null ? void 0 : _f.config) == null ? void 0 : _g.historyDbPath),
-      customPrompt: userConfig.customPrompt,
-      graphStore: {
-        ...DEFAULT_MEMORY_CONFIG.graphStore,
-        ...userConfig.graphStore
-      },
+      customInstructions: userConfig.customInstructions,
+      graphStore: userConfig.graphStore ? { ...userConfig.graphStore } : void 0,
       historyStore: (() => {
         var _a3, _b2;
         const defaultHistoryStore = DEFAULT_MEMORY_CONFIG.historyStore;
@@ -4161,8 +4182,7 @@ var ConfigManager = class {
           }
         };
       })(),
-      disableHistory: userConfig.disableHistory || DEFAULT_MEMORY_CONFIG.disableHistory,
-      enableGraph: userConfig.enableGraph || DEFAULT_MEMORY_CONFIG.enableGraph
+      disableHistory: userConfig.disableHistory || DEFAULT_MEMORY_CONFIG.disableHistory
     };
     return MemoryConfigSchema.parse(mergedConfig);
   }
@@ -4345,7 +4365,7 @@ var MemoryGraph = class {
       relations: toBeAdded
     };
   }
-  async search(query, filters, limit = 100) {
+  async search(query, filters, topK = 100) {
     const entityTypeMap = await this._retrieveNodesFromData(query, filters);
     const searchOutput = await this._searchGraphDb(
       Object.keys(entityTypeMap),
@@ -4380,7 +4400,7 @@ var MemoryGraph = class {
       await session.close();
     }
   }
-  async getAll(filters, limit = 100) {
+  async getAll(filters, topK = 100) {
     const session = this.graph.session();
     try {
       const result = await session.run(
@@ -4389,7 +4409,7 @@ var MemoryGraph = class {
         RETURN n.name AS source, type(r) AS relationship, m.name AS target
         LIMIT toInteger($limit)
         `,
-        { user_id: filters["userId"], limit: Math.floor(Number(limit)) }
+        { user_id: filters["userId"], limit: Math.floor(Number(topK)) }
       );
       const finalResults = result.records.map((record) => ({
         source: record.get("source"),
@@ -4442,7 +4462,7 @@ var MemoryGraph = class {
   async _establishNodesRelationsFromData(data, filters, entityTypeMap) {
     var _a2;
     let messages;
-    if ((_a2 = this.config.graphStore) == null ? void 0 : _a2.customPrompt) {
+    if ((_a2 = this.config.graphStore) == null ? void 0 : _a2.customInstructions) {
       messages = [
         {
           role: "system",
@@ -4451,7 +4471,7 @@ var MemoryGraph = class {
             filters["userId"]
           ).replace(
             "CUSTOM_PROMPT",
-            `4. ${this.config.graphStore.customPrompt}`
+            `4. ${this.config.graphStore.customInstructions}`
           ) + "\nPlease provide your response in JSON format."
         },
         { role: "user", content: data }
@@ -4488,7 +4508,7 @@ Text: ${data}`
     logger.debug(`Extracted entities: ${JSON.stringify(entities)}`);
     return entities;
   }
-  async _searchGraphDb(nodeList, filters, limit = 100) {
+  async _searchGraphDb(nodeList, filters, topK = 100) {
     const resultRelations = [];
     const session = this.graph.session();
     try {
@@ -4521,7 +4541,7 @@ Text: ${data}`
           n_embedding: nEmbedding,
           threshold: this.threshold,
           user_id: filters["userId"],
-          limit: Math.floor(Number(limit))
+          limit: Math.floor(Number(topK))
         });
         resultRelations.push(
           ...result.records.map((record) => ({
@@ -4837,6 +4857,22 @@ try {
 }
 var POSTHOG_API_KEY = "phc_hgJkUVJFYtmaJqrvf6CYN67TIQ8yhXAkWzUn9AMU4yX";
 var POSTHOG_HOST = "https://us.i.posthog.com/i/v0/e/";
+var DEFAULT_SAMPLE_RATE = 0.1;
+var MEM0_TELEMETRY_SAMPLE_RATE = (() => {
+  var _a2;
+  try {
+    const raw = (_a2 = process == null ? void 0 : process.env) == null ? void 0 : _a2.MEM0_TELEMETRY_SAMPLE_RATE;
+    if (raw !== void 0) {
+      const parsed = Number(raw);
+      if (Number.isFinite(parsed) && parsed >= 0 && parsed <= 1) {
+        return parsed;
+      }
+    }
+  } catch (e) {
+  }
+  return DEFAULT_SAMPLE_RATE;
+})();
+var LIFECYCLE_EVENTS = /* @__PURE__ */ new Set(["init", "reset"]);
 var UnifiedTelemetry = class {
   constructor(projectApiKey, host) {
     this.apiKey = projectApiKey;
@@ -4881,6 +4917,10 @@ async function captureClientEvent(eventName, instance, additionalData = {}) {
     console.warn("No telemetry ID found for instance");
     return;
   }
+  const isLifecycle = LIFECYCLE_EVENTS.has(eventName);
+  if (!isLifecycle && Math.random() >= MEM0_TELEMETRY_SAMPLE_RATE) {
+    return;
+  }
   const eventData = {
     function: `${instance.constructor.name}`,
     method: eventName,
@@ -4888,7 +4928,9 @@ async function captureClientEvent(eventName, instance, additionalData = {}) {
     timestamp: (/* @__PURE__ */ new Date()).toISOString(),
     client_version: version,
     client_source: "nodejs",
-    ...additionalData
+    ...additionalData,
+    // sample_rate set AFTER the spread so callers can never override it
+    sample_rate: isLifecycle ? 1 : MEM0_TELEMETRY_SAMPLE_RATE
   };
   await telemetry.captureEvent(
     instance.telemetryId,
@@ -4901,7 +4943,7 @@ async function captureClientEvent(eventName, instance, additionalData = {}) {
 var Memory = class _Memory {
   constructor(config = {}) {
     this.config = ConfigManager.mergeConfig(config);
-    this.customPrompt = this.config.customPrompt;
+    this.customInstructions = this.config.customInstructions;
     this.embedder = EmbedderFactory.create(
       this.config.embedder.provider,
       this.config.embedder.config
@@ -4920,9 +4962,8 @@ var Memory = class _Memory {
     }
     this.collectionName = this.config.vectorStore.config.collectionName;
     this.apiVersion = this.config.version || "v1.0";
-    this.enableGraph = this.config.enableGraph || false;
     this.telemetryId = "anonymous";
-    if (this.enableGraph && this.config.graphStore) {
+    if (this.config.graphStore) {
       this.graphMemory = new MemoryGraph(this.config);
     }
     this._initPromise = this._autoInitialize().catch((error) => {
@@ -4977,8 +5018,7 @@ var Memory = class _Memory {
       await captureClientEvent("init", this, {
         api_version: this.apiVersion,
         client_type: "Memory",
-        collection_name: this.collectionName,
-        enable_graph: this.enableGraph
+        collection_name: this.collectionName
       });
     } catch (error) {
     }
@@ -5084,8 +5124,8 @@ var Memory = class _Memory {
       return returnedMemories;
     }
     const parsedMessages = messages.map((m) => m.content).join("\n");
-    const [systemPrompt, userPrompt] = this.customPrompt ? [
-      this.customPrompt.toLowerCase().includes("json") ? this.customPrompt : `${this.customPrompt}
+    const [systemPrompt, userPrompt] = this.customInstructions ? [
+      this.customInstructions.toLowerCase().includes("json") ? this.customInstructions : `${this.customInstructions}
 
 You MUST return a valid JSON object with a 'facts' key containing an array of strings.`,
       `Input:
@@ -5239,10 +5279,10 @@ ${parsedMessages}`
     await this._ensureInitialized();
     await this._captureEvent("search", {
       query_length: query.length,
-      limit: config.limit,
+      topK: config.topK,
       has_filters: !!config.filters
     });
-    const { userId, agentId, runId, limit = 100, filters = {} } = config;
+    const { userId, agentId, runId, topK = 100, filters = {} } = config;
     if (userId) filters.userId = userId;
     if (agentId) filters.agentId = agentId;
     if (runId) filters.runId = runId;
@@ -5254,7 +5294,7 @@ ${parsedMessages}`
     const queryEmbedding = await this.embedder.embed(query);
     const memories = await this.vectorStore.search(
       queryEmbedding,
-      limit,
+      topK,
       filters
     );
     let graphResults;
@@ -5370,17 +5410,17 @@ ${parsedMessages}`
   async getAll(config) {
     await this._ensureInitialized();
     await this._captureEvent("get_all", {
-      limit: config.limit,
+      topK: config.topK,
       has_user_id: !!config.userId,
       has_agent_id: !!config.agentId,
       has_run_id: !!config.runId
     });
-    const { userId, agentId, runId, limit = 100 } = config;
+    const { userId, agentId, runId, topK = 100 } = config;
     const filters = {};
     if (userId) filters.userId = userId;
     if (agentId) filters.agentId = agentId;
     if (runId) filters.runId = runId;
-    const [memories] = await this.vectorStore.list(filters, limit);
+    const [memories] = await this.vectorStore.list(filters, topK);
     const excludedKeys = /* @__PURE__ */ new Set([
       "userId",
       "agentId",