@bd7pil/opencode-deep-memory 0.6.0 → 0.8.0

package/dist/index.js

@@ -261,6 +261,7 @@ var PluginState = class {
   _ccrCache = /* @__PURE__ */ new Map();
   _lastInputTokens = 0;
   _lastNudgeMessageCount = /* @__PURE__ */ new Map();
+  _lastMemoryNudgeMessageCount = /* @__PURE__ */ new Map();
   _lastCCRCleanup = 0;
   _modelContextWindow = 0;
   agentOf(sessionID) {
@@ -274,6 +275,7 @@ var PluginState = class {
     this._models.delete(sessionID);
     this._lastUserText.delete(sessionID);
     this._lastNudgeMessageCount.delete(sessionID);
+    this._lastMemoryNudgeMessageCount.delete(sessionID);
   }
   recordModel(sessionID, model) {
     this._models.set(sessionID, model);
@@ -423,6 +425,13 @@ var PluginState = class {
     const last = this._lastNudgeMessageCount.get(sessionID);
     return last != null ? currentMessageCount - last : Number.POSITIVE_INFINITY;
   }
+  recordMemoryNudge(sessionID, messageCount) {
+    this._lastMemoryNudgeMessageCount.set(sessionID, messageCount);
+  }
+  messagesSinceLastMemoryNudge(sessionID, currentMessageCount) {
+    const last = this._lastMemoryNudgeMessageCount.get(sessionID);
+    return last != null ? currentMessageCount - last : Number.POSITIVE_INFINITY;
+  }
   setModelContextWindow(tokens) {
     if (tokens > 0) this._modelContextWindow = tokens;
   }
@@ -1127,7 +1136,7 @@ async function runDream(opts) {
         tools: {
           memory_search: true,
           memory_store: true,
-          memory_forget: false,
+          memory_forget: true,
           read: true,
           list: true
         }
@@ -1228,23 +1237,38 @@ async function handleSessionCreatedForDream(args) {
   }
   const notesPath = memoryFilePath("project", "notes", projectPath);
   let notesLines = 0;
+  let notesContent = "";
   try {
-    const content = fs5.readFileSync(notesPath, "utf8");
-    if (content.trim().length === 0) {
+    notesContent = fs5.readFileSync(notesPath, "utf8");
+    if (notesContent.trim().length === 0) {
       logger?.debug("auto-dream: notes.md is empty, skipping spawn");
       return;
     }
-    notesLines = content.split("\n").filter((l) => l.trim()).length;
+    notesLines = notesContent.split("\n").filter((l) => l.trim()).length;
   } catch {
     logger?.debug("auto-dream: notes.md not found, skipping spawn");
     return;
   }
   const memoryPath = memoryFilePath("project", "memory", projectPath);
   if (!fs5.existsSync(memoryPath) || fs5.statSync(memoryPath).size < 50) {
-    logger?.debug("auto-dream: MEMORY.md missing or too small, skipping", {
-      sessionID: info.id
-    });
-    return;
+    if (notesLines >= 5) {
+      try {
+        fs5.writeFileSync(memoryPath, notesContent, "utf8");
+        logger?.info("auto-dream: bootstrapped MEMORY.md from notes.md", {
+          notesLines
+        });
+      } catch (err) {
+        logger?.warn("auto-dream: failed to bootstrap MEMORY.md", {
+          error: err instanceof Error ? err.message : String(err)
+        });
+        return;
+      }
+    } else {
+      logger?.debug("auto-dream: MEMORY.md missing and notes too small, skipping", {
+        sessionID: info.id
+      });
+      return;
+    }
   }
   const isSevenDayDue = schedule.lastDream === null || Date.now() - Date.parse(schedule.lastDream) > DREAM_INTERVAL_MS;
   let isAccumulationDue = false;
@@ -1386,7 +1410,7 @@ async function runDistill(opts) {
         tools: {
           memory_search: true,
           memory_store: true,
-          memory_forget: false,
+          memory_forget: true,
           read: true,
           list: true
         }
@@ -15284,12 +15308,10 @@ function createCompactingHandler(args) {
 }
 
 // src/compress/pressure.ts
-var FALLBACK_MAX_CONTEXT = 128e3;
+var FALLBACK_MAX_CONTEXT = 1e6;
 var OPENCODE_COMPACTION_RATIO = 0.75;
-var THRESHOLDS = {
-  medium: 0.2,
-  high: 0.4
-};
+var PRESSURE_MEDIUM_TOKENS = 5e4;
+var PRESSURE_HIGH_TOKENS = 15e4;
 var calibratedMaxContext = 0;
 function calibrateFromCompaction(lastInputTokens) {
   if (lastInputTokens <= 0) return;
@@ -15341,20 +15363,23 @@ function extractTokensFromMessages(messages) {
   return total;
 }
 function extractInputTokensFromMessages(messages) {
+  let best = 0;
   for (let i = messages.length - 1; i >= 0; i--) {
     const msg = messages[i];
     for (const part of msg.parts) {
       if (typeof part !== "object" || part === null) continue;
       const p = part;
       if (p["type"] === "step-finish") {
-        const tokens = p.tokens;
-        if (tokens?.input && tokens.input > 0) {
-          return tokens.input;
-        }
+        const tokens = p;
+        const input = tokens.tokens?.input ?? 0;
+        const cached2 = tokens.tokens?.cached ?? 0;
+        const total = input + cached2;
+        if (total > best) best = total;
+        if (best > 0) return best;
       }
     }
   }
-  return 0;
+  return best;
 }
 function detectPressure(messages, modelContextWindow) {
   const ctx = maxContextFrom(modelContextWindow || 0);
@@ -15362,8 +15387,8 @@ function detectPressure(messages, modelContextWindow) {
   const estimated = inputTokens > 0 ? inputTokens : extractTokensFromMessages(messages);
   const ratio = Math.min(estimated / ctx, 1);
   let level;
-  if (ratio >= THRESHOLDS.high) level = "high";
-  else if (ratio >= THRESHOLDS.medium) level = "medium";
+  if (estimated >= PRESSURE_HIGH_TOKENS) level = "high";
+  else if (estimated >= PRESSURE_MEDIUM_TOKENS) level = "medium";
   else level = "low";
   return { level, ratio, estimatedTokens: estimated, maxContext: ctx };
 }
@@ -15386,17 +15411,17 @@ function buildNudgeText(level) {
 var MEMORY_NUDGE_COOLDOWN = 3;
 var DECISION_PATTERNS = [
   /\b(?:decided|decision|chose|chosen|picked|selected)\b/i,
-  /\b(?:采用|选择|决定|确定|选用)\b/,
+  /(?:采用|选择|决定|确定|选用)/,
   /\b(?:use|using|go with|went with)\b.*\b(?:because|since|due to)\b/i
 ];
 var CONSTRAINT_PATTERNS = [
   /\b(?:must not|cannot|should not|do not|never|always)\b/i,
   /\b(?:constraint|restriction|limitation|requirement)\b/i,
-  /\b(?:不能|必须|禁止|约束|限制|要求|务必)\b/
+  /(?:不能|必须|禁止|约束|限制|要求|务必)/
 ];
 var ERROR_FIX_PATTERNS = [
   /\b(?:fix|fixed|resolve|resolved|patch|corrected)\b/i,
-  /\b(?:修复|修复了|解决|解决了)\b/,
+  /(?:修复|修复了|解决|解决了)/,
   /\b(?:the (?:bug|error|issue) (?:was|is)|root cause)\b/i
 ];
 function detectMemoryNudge(messages, messagesSinceLastNudge) {
@@ -15410,13 +15435,14 @@ function detectMemoryNudge(messages, messagesSinceLastNudge) {
   const hasRecentToolError = recentMessages.some(
     (m) => m.parts.some((p) => p.type === "tool" && p.state?.status === "error")
   );
+  const recentAll = recentUserText + "\n" + recentAssistantText;
   if (hasRecentToolError && ERROR_FIX_PATTERNS.some((p) => p.test(recentAssistantText))) {
     return { injected: true, type: "gotcha" };
   }
-  if (CONSTRAINT_PATTERNS.some((p) => p.test(recentUserText))) {
+  if (CONSTRAINT_PATTERNS.some((p) => p.test(recentAll))) {
     return { injected: true, type: "constraint" };
   }
-  if (DECISION_PATTERNS.some((p) => p.test(recentAssistantText))) {
+  if (DECISION_PATTERNS.some((p) => p.test(recentAll))) {
     return { injected: true, type: "decision" };
   }
   return { injected: false, type: null };
@@ -15627,6 +15653,8 @@ var PROTECTED_TOOLS = /* @__PURE__ */ new Set([
 var NEVER_DEDUP = /* @__PURE__ */ new Set(["read", "bash", "grep", "glob", "find", "search"]);
 var ERROR_PURGE_TURN_THRESHOLD = 4;
 var PROTECTED_HEAD_SINGLE = 2;
+var ASSISTANT_COMPRESS_MIN_LENGTH = 500;
+var ASSISTANT_COMPRESS_SAVINGS_RATIO = 0.6;
 function simpleHash(s) {
   const len = s.length;
   const sampleSize = 500;
@@ -15640,12 +15668,32 @@ function simpleHash(s) {
   }
   return `${len}:${h.toString(36)}`;
 }
+function compressAssistantText(text) {
+  if (text.length < ASSISTANT_COMPRESS_MIN_LENGTH) return text;
+  const lines = text.split("\n");
+  const head = 3;
+  const tail = 3;
+  const kept = [];
+  for (let i = 0; i < lines.length; i++) {
+    const line = lines[i];
+    if (i < head || i >= lines.length - tail) {
+      kept.push(line);
+      continue;
+    }
+    if (/^#{1,3}\s/.test(line) || /error|fail|warning|critical|important/i.test(line) || /^\s*[-*]\s/.test(line) || /^\s*\d+\.\s/.test(line) || line.trim().startsWith("```") || /^\/[^\s:]+/.test(line)) {
+      kept.push(line);
+    }
+  }
+  const result = kept.join("\n");
+  return result.length < text.length * ASSISTANT_COMPRESS_SAVINGS_RATIO ? result : text;
+}
 function singlePassCompress(messages, state, protectedTail) {
   const stats = {
     toolDedup: 0,
     errorPurge: 0,
     toolOutputCompressed: 0,
     jsonCrushed: 0,
+    assistantCompressed: 0,
     ccrStored: 0
   };
   const totalMessages = messages.length;
@@ -15725,6 +15773,21 @@ function singlePassCompress(messages, state, protectedTail) {
         }
       }
     }
+    if (i < protectedTail && msg.info.role === "assistant") {
+      for (let j = msg.parts.length - 1; j >= 0; j--) {
+        const part = msg.parts[j];
+        if (typeof part !== "object" || part === null) continue;
+        const p = part;
+        if (p["type"] !== "text") continue;
+        const text = p["text"];
+        if (typeof text !== "string") continue;
+        const compressed = compressAssistantText(text);
+        if (compressed !== text) {
+          p["text"] = compressed;
+          stats.assistantCompressed++;
+        }
+      }
+    }
   }
   return stats;
 }
@@ -15766,27 +15829,30 @@ function runCompressionPipeline(ctx) {
     errorPurge: spStats.errorPurge,
     toolOutputCompressed: spStats.toolOutputCompressed,
     jsonCrushed: spStats.jsonCrushed,
+    assistantCompressed: spStats.assistantCompressed,
     ccrStored: spStats.ccrStored,
     nudgeInjected: false,
     pressureLevel: pressure.level,
     estimatedTokens: pressure.estimatedTokens
   };
   const sid = sessionID || "default";
-  const messagesSinceNudge = state.messagesSinceLastNudge(sid, messages.length);
-  if (shouldInjectNudge(pressure.level, messagesSinceNudge)) {
+  const currentMsgCount = messages.length;
+  const pressureSince = state.messagesSinceLastNudge(sid, currentMsgCount);
+  if (shouldInjectNudge(pressure.level, pressureSince)) {
     if (injectIntoLastAssistant(messages, buildNudgeText(pressure.level))) {
       stats.nudgeInjected = true;
-      state.recordNudge(sid, messages.length);
+      state.recordNudge(sid, currentMsgCount);
     }
   }
-  const memoryNudge = detectMemoryNudge(messages, state.messagesSinceLastNudge(sid, messages.length));
+  const memorySince = state.messagesSinceLastMemoryNudge(sid, currentMsgCount);
+  const memoryNudge = detectMemoryNudge(messages, memorySince);
   if (memoryNudge.injected) {
     if (injectIntoLastAssistant(messages, buildMemoryNudge(memoryNudge.type))) {
-      state.recordNudge(sid, messages.length);
+      state.recordMemoryNudge(sid, currentMsgCount);
       logger?.debug("compress: memory nudge", { type: memoryNudge.type });
     }
   }
-  const active = stats.toolDedup > 0 || stats.errorPurge > 0 || stats.toolOutputCompressed > 0 || stats.jsonCrushed > 0 || stats.nudgeInjected;
+  const active = stats.toolDedup > 0 || stats.errorPurge > 0 || stats.toolOutputCompressed > 0 || stats.jsonCrushed > 0 || stats.assistantCompressed > 0 || stats.nudgeInjected;
   if (active) {
     logger?.debug("compress: pipeline result", { ...stats });
   } else {
@@ -15976,7 +16042,7 @@ function createMessagesTransformHandler(state, logger) {
       logger
     });
     const ds = pipelineResult.stats;
-    if (ds.toolDedup > 0 || ds.errorPurge > 0 || ds.toolOutputCompressed > 0 || ds.jsonCrushed > 0 || ds.nudgeInjected) {
+    if (ds.toolDedup > 0 || ds.errorPurge > 0 || ds.toolOutputCompressed > 0 || ds.jsonCrushed > 0 || ds.assistantCompressed > 0 || ds.nudgeInjected) {
       logger?.debug("messages.transform: deep compression", { ...ds });
       state.mergeNotify({
         compression: stats,
@@ -16116,6 +16182,29 @@ function createNotifyHandler(client, logger) {
   };
 }
 
+// src/shared/model-limits.ts
+var KNOWN_MODEL_LIMITS = {
+  "deepseek-v4-pro": 1e6,
+  "deepseek-v4": 1e6,
+  "deepseek-v3": 64e3,
+  "deepseek-r1": 64e3,
+  "claude-opus-4": 2e5,
+  "claude-sonnet-4": 2e5,
+  "gpt-4o": 128e3,
+  "o1": 2e5,
+  "o3-mini": 2e5,
+  "gemini-2.5-pro": 1e6,
+  "gemini-2.5-flash": 1e6,
+  "qwen-max": 131072
+};
+function lookupModelLimit(modelID) {
+  if (KNOWN_MODEL_LIMITS[modelID]) return KNOWN_MODEL_LIMITS[modelID];
+  for (const [key, limit] of Object.entries(KNOWN_MODEL_LIMITS)) {
+    if (modelID.includes(key)) return limit;
+  }
+  return void 0;
+}
+
 // src/extract/enrich.ts
 import { stat } from "fs/promises";
 
@@ -16447,11 +16536,14 @@ var deepMemoryPlugin = async (input) => {
       const defaultModel = configResult.data?.model;
       if (typeof defaultModel === "string" && defaultModel.includes("/")) {
         const slashIdx = defaultModel.indexOf("/");
-        state.recordFallbackModel({
-          providerID: defaultModel.slice(0, slashIdx),
-          modelID: defaultModel.slice(slashIdx + 1)
-        });
-        logger.debug("resolved fallback model from config", { defaultModel });
+        const providerID = defaultModel.slice(0, slashIdx);
+        const modelID = defaultModel.slice(slashIdx + 1);
+        state.recordFallbackModel({ providerID, modelID });
+        const limit = lookupModelLimit(modelID);
+        if (limit) {
+          state.setModelContextWindow(limit);
+          logger.debug("resolved model context window", { modelID, limit });
+        }
       }
     }).catch((err) => {
       logger.debug("config.get failed, dream/distill will omit model", {