@elizaos/plugin-memory 1.0.3 → 1.0.5

package/dist/node/index.node.js

@@ -117,14 +117,17 @@ class MemoryService extends Service {
     this.sessionMessageCounts = new Map;
     this.lastExtractionCheckpoints = new Map;
     this.memoryConfig = {
-      shortTermSummarizationThreshold: 5,
+      shortTermSummarizationThreshold: 16,
       shortTermRetainRecent: 10,
+      shortTermSummarizationInterval: 10,
       longTermExtractionEnabled: true,
       longTermVectorSearchEnabled: false,
       longTermConfidenceThreshold: 0.7,
+      longTermExtractionThreshold: 20,
       longTermExtractionInterval: 5,
       summaryModelType: "TEXT_LARGE",
-      summaryMaxTokens: 2500
+      summaryMaxTokens: 2500,
+      summaryMaxNewMessages: 20
     };
   }
   static async start(runtime) {
@@ -145,6 +148,14 @@ class MemoryService extends Service {
     if (retainRecent) {
       this.memoryConfig.shortTermRetainRecent = parseInt(retainRecent, 10);
     }
+    const summarizationInterval = runtime.getSetting("MEMORY_SUMMARIZATION_INTERVAL");
+    if (summarizationInterval) {
+      this.memoryConfig.shortTermSummarizationInterval = parseInt(summarizationInterval, 10);
+    }
+    const maxNewMessages = runtime.getSetting("MEMORY_MAX_NEW_MESSAGES");
+    if (maxNewMessages) {
+      this.memoryConfig.summaryMaxNewMessages = parseInt(maxNewMessages, 10);
+    }
     const longTermEnabled = runtime.getSetting("MEMORY_LONG_TERM_ENABLED");
     if (longTermEnabled === "false") {
       this.memoryConfig.longTermExtractionEnabled = false;
@@ -155,10 +166,21 @@ class MemoryService extends Service {
     if (confidenceThreshold) {
       this.memoryConfig.longTermConfidenceThreshold = parseFloat(confidenceThreshold);
     }
+    const extractionThreshold = runtime.getSetting("MEMORY_EXTRACTION_THRESHOLD");
+    if (extractionThreshold) {
+      this.memoryConfig.longTermExtractionThreshold = parseInt(extractionThreshold, 10);
+    }
+    const extractionInterval = runtime.getSetting("MEMORY_EXTRACTION_INTERVAL");
+    if (extractionInterval) {
+      this.memoryConfig.longTermExtractionInterval = parseInt(extractionInterval, 10);
+    }
     logger.info({
       summarizationThreshold: this.memoryConfig.shortTermSummarizationThreshold,
+      summarizationInterval: this.memoryConfig.shortTermSummarizationInterval,
+      maxNewMessages: this.memoryConfig.summaryMaxNewMessages,
       retainRecent: this.memoryConfig.shortTermRetainRecent,
       longTermEnabled: this.memoryConfig.longTermExtractionEnabled,
+      extractionThreshold: this.memoryConfig.longTermExtractionThreshold,
       extractionInterval: this.memoryConfig.longTermExtractionInterval,
       confidenceThreshold: this.memoryConfig.longTermConfidenceThreshold
     }, "MemoryService initialized");
@@ -219,14 +241,26 @@ class MemoryService extends Service {
     }
   }
   async shouldRunExtraction(entityId, roomId, currentMessageCount) {
+    const threshold = this.memoryConfig.longTermExtractionThreshold;
     const interval = this.memoryConfig.longTermExtractionInterval;
+    if (currentMessageCount < threshold) {
+      logger.debug({
+        entityId,
+        roomId,
+        currentMessageCount,
+        threshold,
+        shouldRun: false
+      }, "Extraction check: below threshold");
+      return false;
+    }
     const lastCheckpoint = await this.getLastExtractionCheckpoint(entityId, roomId);
     const currentCheckpoint = Math.floor(currentMessageCount / interval) * interval;
-    const shouldRun = currentMessageCount >= interval && currentCheckpoint > lastCheckpoint;
+    const shouldRun = currentMessageCount >= threshold && currentCheckpoint > lastCheckpoint;
     logger.debug({
       entityId,
       roomId,
       currentMessageCount,
+      threshold,
       interval,
       lastCheckpoint,
       currentCheckpoint,
@@ -294,7 +328,7 @@ class MemoryService extends Service {
       accessCount: row.accessCount
     }));
   }
-  async updateLongTermMemory(id, updates) {
+  async updateLongTermMemory(id, entityId, updates) {
     const db = this.getDb();
     const updateData = {
       updatedAt: new Date
@@ -317,13 +351,13 @@ class MemoryService extends Service {
     if (updates.accessCount !== undefined) {
       updateData.accessCount = updates.accessCount;
     }
-    await db.update(longTermMemories).set(updateData).where(eq(longTermMemories.id, id));
-    logger.info(`Updated long-term memory: ${id}`);
+    await db.update(longTermMemories).set(updateData).where(and(eq(longTermMemories.id, id), eq(longTermMemories.agentId, this.runtime.agentId), eq(longTermMemories.entityId, entityId)));
+    logger.info(`Updated long-term memory: ${id} for entity ${entityId}`);
   }
-  async deleteLongTermMemory(id) {
+  async deleteLongTermMemory(id, entityId) {
     const db = this.getDb();
-    await db.delete(longTermMemories).where(eq(longTermMemories.id, id));
-    logger.info(`Deleted long-term memory: ${id}`);
+    await db.delete(longTermMemories).where(and(eq(longTermMemories.id, id), eq(longTermMemories.agentId, this.runtime.agentId), eq(longTermMemories.entityId, entityId)));
+    logger.info(`Deleted long-term memory: ${id} for entity ${entityId}`);
   }
   async getCurrentSessionSummary(roomId) {
     const db = this.getDb();
@@ -378,7 +412,7 @@ class MemoryService extends Service {
     logger.info(`Stored session summary for room ${newSummary.roomId}`);
     return newSummary;
   }
-  async updateSessionSummary(id, updates) {
+  async updateSessionSummary(id, roomId, updates) {
     const db = this.getDb();
     const updateData = {
       updatedAt: new Date
@@ -404,8 +438,8 @@ class MemoryService extends Service {
     if (updates.embedding !== undefined) {
       updateData.embedding = updates.embedding;
     }
-    await db.update(sessionSummaries).set(updateData).where(eq(sessionSummaries.id, id));
-    logger.info(`Updated session summary: ${id}`);
+    await db.update(sessionSummaries).set(updateData).where(and(eq(sessionSummaries.id, id), eq(sessionSummaries.agentId, this.runtime.agentId), eq(sessionSummaries.roomId, roomId)));
+    logger.info(`Updated session summary: ${id} for room ${roomId}`);
   }
   async getSessionSummaries(roomId, limit = 5) {
     const db = this.getDb();
@@ -589,14 +623,31 @@ var summarizationEvaluator = {
     }
     const config = memoryService.getConfig();
     const currentMessageCount = await runtime.countMemories(message.roomId, false, "messages");
-    const shouldSummarize = currentMessageCount >= config.shortTermSummarizationThreshold;
-    logger2.debug({
-      roomId: message.roomId,
-      currentMessageCount,
-      threshold: config.shortTermSummarizationThreshold,
-      shouldSummarize
-    }, "Summarization validation check");
-    return shouldSummarize;
+    const existingSummary = await memoryService.getCurrentSessionSummary(message.roomId);
+    if (!existingSummary) {
+      const shouldSummarize = currentMessageCount >= config.shortTermSummarizationThreshold;
+      logger2.debug({
+        roomId: message.roomId,
+        currentMessageCount,
+        threshold: config.shortTermSummarizationThreshold,
+        shouldSummarize,
+        reason: "initial_summary_check"
+      }, "Summarization validation check");
+      return shouldSummarize;
+    } else {
+      const newMessageCount = currentMessageCount - existingSummary.lastMessageOffset;
+      const shouldUpdate = newMessageCount >= config.shortTermSummarizationInterval;
+      logger2.debug({
+        roomId: message.roomId,
+        currentMessageCount,
+        lastOffset: existingSummary.lastMessageOffset,
+        newMessageCount,
+        interval: config.shortTermSummarizationInterval,
+        shouldUpdate,
+        reason: "summary_update_check"
+      }, "Summarization validation check");
+      return shouldUpdate;
+    }
   },
   handler: async (runtime, message) => {
     const memoryService = runtime.getService("memory");
@@ -611,15 +662,25 @@ var summarizationEvaluator = {
       const existingSummary = await memoryService.getCurrentSessionSummary(roomId);
       const lastOffset = existingSummary?.lastMessageOffset || 0;
       const totalMessageCount = await runtime.countMemories(roomId, false, "messages");
+      const newMessageCount = totalMessageCount - lastOffset;
+      const maxNewMessages = config.summaryMaxNewMessages || 50;
+      const messagesToFetch = Math.min(newMessageCount, maxNewMessages);
+      if (messagesToFetch === 0) {
+        logger2.debug("No new messages to summarize");
+        return;
+      }
+      if (newMessageCount > maxNewMessages) {
+        logger2.warn(`Capping new messages at ${maxNewMessages} (${newMessageCount} available). Oldest messages will be skipped.`);
+      }
       const newMessages = await runtime.getMemories({
         tableName: "messages",
         roomId,
-        count: config.shortTermSummarizationThreshold,
+        count: messagesToFetch,
         unique: false,
         start: lastOffset
       });
       if (newMessages.length === 0) {
-        logger2.debug("No new messages to summarize");
+        logger2.debug("No new messages retrieved");
         return;
       }
       const sortedMessages = newMessages.sort((a, b) => (a.createdAt || 0) - (b.createdAt || 0));
@@ -664,7 +725,7 @@ var summarizationEvaluator = {
       const startTime = existingSummary ? existingSummary.startTime : firstMessage?.createdAt && firstMessage.createdAt > 0 ? new Date(firstMessage.createdAt) : new Date;
       const endTime = lastMessage?.createdAt && lastMessage.createdAt > 0 ? new Date(lastMessage.createdAt) : new Date;
       if (existingSummary) {
-        await memoryService.updateSessionSummary(existingSummary.id, {
+        await memoryService.updateSessionSummary(existingSummary.id, roomId, {
           summary: summaryResult.summary,
           messageCount: existingSummary.messageCount + sortedMessages.length,
           lastMessageOffset: newOffset,
@@ -723,7 +784,7 @@ var LongTermMemoryCategory;
 // src/evaluators/long-term-extraction.ts
 var extractionTemplate = `# Task: Extract Long-Term Memory
 
-You are analyzing a conversation to extract facts that should be remembered long-term about the user.
+You are analyzing a conversation to extract ONLY the most important, persistent facts about the user that should be remembered long-term.
 
 # Recent Messages
 {{recentMessages}}
@@ -742,12 +803,41 @@ You are analyzing a conversation to extract facts that should be remembered long
 8. **definitions**: Custom terms, acronyms, glossaries
 9. **behavioral_patterns**: How the user tends to interact
 
+# STRICT EXTRACTION CRITERIA
+
+**DO EXTRACT** - Facts that are:
+- Explicitly stated personal information (name, role, profession)
+- Repeated patterns across multiple conversations (3+ occurrences)
+- Core preferences stated with clear emphasis ("I always prefer...", "I never want...")
+- Important ongoing projects or goals mentioned multiple times
+- Persistent technical constraints or requirements
+- Domain expertise demonstrated consistently
+- Custom terminology the user defines and uses repeatedly
+
+**DO NOT EXTRACT** - Facts that are:
+- One-time requests or isolated tasks (e.g., "generate an image or similar")
+- Single interactions without pattern repetition
+- Casual preferences from single occurrences
+- Testing or exploratory questions
+- Temporary context or transient information
+- General courtesy or social patterns (greetings, thank yous)
+- Random or playful requests that don't indicate lasting preference
+- Information that could change or is situational
+
+# Quality Standards
+- **Confidence threshold**: Only extract if confidence >= 0.8
+- **Persistence required**: Must see evidence across multiple messages or strong explicit statement
+- **Meaningful value**: Would this fact improve future interactions weeks from now?
+- **Not redundant**: Avoid if existing memories already cover this information
+
 # Instructions
-Extract any NEW information that should be remembered long-term. For each item:
+Extract ONLY truly important NEW information that meets the strict criteria above. For each item:
 - Determine which category it belongs to
 - Write a clear, factual statement
-- Assess confidence (0.0 to 1.0)
-- Only include information explicitly stated or strongly implied
+- Assess confidence (0.0 to 1.0) - BE CONSERVATIVE
+- Require strong evidence before extraction
+
+**When in doubt, DO NOT extract.** It's better to miss temporary information than to clutter long-term memory.
 
 If there are no new long-term facts to extract, respond with <memories></memories>
 
@@ -897,7 +987,7 @@ var shortTermMemoryProvider = {
       if (!memoryService) {
         return {
           data: { summaries: [], recentMessages: [], mode: "disabled" },
-          values: { sessionSummaries: "", recentMessages: "" },
+          values: {},
           text: ""
         };
       }
@@ -978,9 +1068,7 @@ ${runText}`;
               mode: "full_conversation"
             },
             values: {
-              sessionSummaries: "",
-              recentMessages: "",
-              recentActionResults: ""
+              recentMessage: "No recent message available."
             },
             text: "No recent messages available"
           };
@@ -1018,10 +1106,14 @@ ${runText}`;
             mode: "full_conversation"
           },
           values: {
-            sessionSummaries: "",
-            recentMessages: isPostFormat ? recentPosts : recentMessages,
-            recentActionResults: actionResultsText,
-            recentMessage
+            ...(isPostFormat ? recentPosts : recentMessages) && {
+              recentMessages: isPostFormat ? recentPosts : recentMessages
+            },
+            ...recentPosts && { recentPosts },
+            ...actionResultsText && { recentActionResults: actionResultsText },
+            ...recentMessage && { recentMessage },
+            ...receivedMessageHeader && { receivedMessageHeader },
+            ...focusHeader && { focusHeader }
           },
           text: text4
         };
@@ -1091,8 +1183,10 @@ ${runText}`;
             mode: "summarized"
           },
           values: {
-            sessionSummaries: summaryText,
-            recentMessages: recentMessagesText
+            ...summaryText && { sessionSummaries: summaryText },
+            ...recentMessagesText && { recentMessages: recentMessagesText },
+            ...receivedMessageHeader && { receivedMessageHeader },
+            ...focusHeader && { focusHeader }
           },
           text: text4
         };
@@ -1101,7 +1195,7 @@ ${runText}`;
       logger4.error({ error }, "Error in shortTermMemoryProvider:");
       return {
         data: { summaries: [], recentMessages: [], mode: "error" },
-        values: { sessionSummaries: "", recentMessages: "" },
+        values: {},
         text: "Error retrieving conversation context."
       };
     }
@@ -1193,4 +1287,4 @@ export {
   LongTermMemoryCategory
 };
 
-//# debugId=F2E3A5D3351D51D164756E2164756E21
+//# debugId=374CC1B0E70335CE64756E2164756E21