memorix 1.1.7 → 1.1.8

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (174) hide show
  1. package/CHANGELOG.md +13 -0
  2. package/CLAUDE.md +6 -1
  3. package/README.md +21 -0
  4. package/README.zh-CN.md +21 -0
  5. package/TEAM.md +86 -86
  6. package/dist/cli/index.js +730 -150
  7. package/dist/cli/index.js.map +1 -1
  8. package/dist/dashboard/static/index.html +201 -201
  9. package/dist/dashboard/static/style.css +3584 -3584
  10. package/dist/index.js +15 -8
  11. package/dist/index.js.map +1 -1
  12. package/dist/memcode-runtime/CHANGELOG.md +13 -0
  13. package/dist/memcode-runtime/package.json +4 -4
  14. package/dist/sdk.js +15 -8
  15. package/dist/sdk.js.map +1 -1
  16. package/docs/AGENT_OPERATOR_PLAYBOOK.md +18 -0
  17. package/docs/DESIGN_DECISIONS.md +357 -357
  18. package/docs/SETUP.md +10 -0
  19. package/docs/dev-log/progress.txt +35 -26
  20. package/package.json +1 -1
  21. package/src/audit/index.ts +156 -156
  22. package/src/cli/commands/agent-integrations.ts +623 -0
  23. package/src/cli/commands/audit-list.ts +89 -89
  24. package/src/cli/commands/background.ts +659 -659
  25. package/src/cli/commands/cleanup.ts +255 -255
  26. package/src/cli/commands/doctor.ts +23 -0
  27. package/src/cli/commands/formation.ts +48 -48
  28. package/src/cli/commands/git-hook-install.ts +111 -111
  29. package/src/cli/commands/handoff.ts +66 -66
  30. package/src/cli/commands/hooks-status.ts +63 -63
  31. package/src/cli/commands/ingest-commit.ts +153 -153
  32. package/src/cli/commands/ingest-image.ts +73 -73
  33. package/src/cli/commands/ingest-log.ts +180 -180
  34. package/src/cli/commands/ingest.ts +44 -44
  35. package/src/cli/commands/integrate-shared.ts +15 -15
  36. package/src/cli/commands/lock.ts +96 -96
  37. package/src/cli/commands/message.ts +121 -121
  38. package/src/cli/commands/poll.ts +70 -70
  39. package/src/cli/commands/purge-all-memory.ts +85 -85
  40. package/src/cli/commands/purge-project-memory.ts +83 -83
  41. package/src/cli/commands/reasoning.ts +132 -132
  42. package/src/cli/commands/repair.ts +60 -0
  43. package/src/cli/commands/retention.ts +108 -108
  44. package/src/cli/commands/serve-shared.ts +118 -118
  45. package/src/cli/commands/setup.ts +3 -3
  46. package/src/cli/commands/skills.ts +123 -123
  47. package/src/cli/commands/task.ts +192 -192
  48. package/src/cli/commands/transfer.ts +73 -73
  49. package/src/cli/commands/uninstall-project-artifacts.ts +85 -85
  50. package/src/cli/index.ts +3 -1
  51. package/src/cli/tui/ChatView.tsx +234 -234
  52. package/src/cli/tui/CommandBar.tsx +312 -312
  53. package/src/cli/tui/ContextRail.tsx +118 -118
  54. package/src/cli/tui/HeaderBar.tsx +72 -72
  55. package/src/cli/tui/LogoBanner.tsx +51 -51
  56. package/src/cli/tui/Panels.tsx +632 -632
  57. package/src/cli/tui/Sidebar.tsx +179 -179
  58. package/src/cli/tui/chat-service.ts +742 -742
  59. package/src/cli/tui/data.ts +547 -547
  60. package/src/cli/tui/index.ts +41 -41
  61. package/src/cli/tui/markdown-render.tsx +371 -371
  62. package/src/cli/tui/theme.ts +178 -178
  63. package/src/cli/tui/use-mouse.ts +157 -157
  64. package/src/cli/tui/useNavigation.ts +56 -56
  65. package/src/cli/update-checker.ts +211 -211
  66. package/src/cli/version.ts +7 -7
  67. package/src/cli/workbench.ts +1 -1
  68. package/src/codegraph/context-pack.ts +5 -1
  69. package/src/codegraph/lite-provider.ts +5 -1
  70. package/src/codegraph/project-context.ts +5 -1
  71. package/src/compact/token-budget.ts +74 -74
  72. package/src/config/behavior.ts +59 -59
  73. package/src/dashboard/project-classification.ts +64 -64
  74. package/src/dashboard/static/index.html +201 -201
  75. package/src/dashboard/static/style.css +3584 -3584
  76. package/src/embedding/fastembed-provider.ts +142 -142
  77. package/src/embedding/transformers-provider.ts +111 -111
  78. package/src/git/extractor.ts +209 -209
  79. package/src/git/hooks-path.ts +85 -85
  80. package/src/git/noise-filter.ts +210 -210
  81. package/src/hooks/installers/index.ts +4 -4
  82. package/src/hooks/official-skills.ts +1 -1
  83. package/src/hooks/pattern-detector.ts +173 -173
  84. package/src/hooks/rules/memorix-agent-rules.md +2 -2
  85. package/src/hooks/significance-filter.ts +250 -250
  86. package/src/llm/memory-manager.ts +328 -328
  87. package/src/llm/provider.ts +885 -885
  88. package/src/llm/quality.ts +248 -248
  89. package/src/memory/attribution-guard.ts +249 -249
  90. package/src/memory/auto-relations.ts +107 -107
  91. package/src/memory/consolidation.ts +302 -302
  92. package/src/memory/disclosure-policy.ts +141 -141
  93. package/src/memory/entity-extractor.ts +197 -197
  94. package/src/memory/formation/evaluate.ts +217 -217
  95. package/src/memory/formation/extract.ts +361 -361
  96. package/src/memory/formation/index.ts +417 -417
  97. package/src/memory/formation/resolve.ts +344 -344
  98. package/src/memory/formation/types.ts +315 -315
  99. package/src/memory/freshness.ts +122 -122
  100. package/src/memory/graph.ts +197 -197
  101. package/src/memory/refs.ts +94 -94
  102. package/src/memory/retention.ts +433 -433
  103. package/src/memory/secret-filter.ts +79 -79
  104. package/src/memory/session.ts +523 -523
  105. package/src/multimodal/image-loader.ts +143 -143
  106. package/src/orchestrate/adapters/claude-stream.ts +192 -192
  107. package/src/orchestrate/adapters/claude.ts +111 -111
  108. package/src/orchestrate/adapters/codex-stream.ts +134 -134
  109. package/src/orchestrate/adapters/codex.ts +41 -41
  110. package/src/orchestrate/adapters/gemini-stream.ts +166 -166
  111. package/src/orchestrate/adapters/gemini.ts +42 -42
  112. package/src/orchestrate/adapters/index.ts +73 -73
  113. package/src/orchestrate/adapters/opencode-stream.ts +143 -143
  114. package/src/orchestrate/adapters/opencode.ts +47 -47
  115. package/src/orchestrate/adapters/spawn-helper.ts +286 -286
  116. package/src/orchestrate/adapters/types.ts +77 -77
  117. package/src/orchestrate/capability-router.ts +284 -284
  118. package/src/orchestrate/context-compact.ts +188 -188
  119. package/src/orchestrate/cost-tracker.ts +219 -219
  120. package/src/orchestrate/error-recovery.ts +191 -191
  121. package/src/orchestrate/evidence.ts +140 -140
  122. package/src/orchestrate/ledger.ts +110 -110
  123. package/src/orchestrate/memorix-bridge.ts +380 -380
  124. package/src/orchestrate/output-budget.ts +80 -80
  125. package/src/orchestrate/permission.ts +152 -152
  126. package/src/orchestrate/pipeline-trace.ts +131 -131
  127. package/src/orchestrate/prompt-builder.ts +155 -155
  128. package/src/orchestrate/ring-buffer.ts +37 -37
  129. package/src/orchestrate/task-graph.ts +389 -389
  130. package/src/orchestrate/verify-gate.ts +219 -219
  131. package/src/orchestrate/worktree.ts +232 -232
  132. package/src/project/aliases.ts +374 -374
  133. package/src/project/detector.ts +268 -268
  134. package/src/rules/adapters/claude-code.ts +99 -99
  135. package/src/rules/adapters/codex.ts +97 -97
  136. package/src/rules/adapters/copilot.ts +124 -124
  137. package/src/rules/adapters/cursor.ts +114 -114
  138. package/src/rules/adapters/kiro.ts +126 -126
  139. package/src/rules/adapters/trae.ts +56 -56
  140. package/src/rules/adapters/windsurf.ts +83 -83
  141. package/src/rules/syncer.ts +235 -235
  142. package/src/sdk.ts +327 -327
  143. package/src/search/intent-detector.ts +289 -289
  144. package/src/search/query-expansion.ts +52 -52
  145. package/src/server/formation-timeout.ts +27 -27
  146. package/src/skills/mini-skills.ts +386 -386
  147. package/src/store/chat-store.ts +119 -119
  148. package/src/store/file-lock.ts +100 -100
  149. package/src/store/graph-store.ts +249 -249
  150. package/src/store/mini-skill-store.ts +349 -349
  151. package/src/store/obs-store.ts +255 -255
  152. package/src/store/persistence-json.ts +212 -212
  153. package/src/store/persistence.ts +291 -291
  154. package/src/store/project-affinity.ts +195 -195
  155. package/src/store/session-store.ts +259 -259
  156. package/src/store/sqlite-store.ts +339 -339
  157. package/src/team/event-bus.ts +76 -76
  158. package/src/team/file-locks.ts +173 -173
  159. package/src/team/handoff.ts +167 -167
  160. package/src/team/messages.ts +203 -203
  161. package/src/team/poll.ts +132 -132
  162. package/src/team/tasks.ts +211 -211
  163. package/src/wiki/generator.ts +237 -237
  164. package/src/wiki/knowledge-graph.ts +334 -334
  165. package/src/wiki/types.ts +85 -85
  166. package/src/workspace/mcp-adapters/codex.ts +191 -191
  167. package/src/workspace/mcp-adapters/copilot.ts +105 -105
  168. package/src/workspace/mcp-adapters/cursor.ts +53 -53
  169. package/src/workspace/mcp-adapters/kiro.ts +64 -64
  170. package/src/workspace/mcp-adapters/opencode.ts +123 -123
  171. package/src/workspace/mcp-adapters/trae.ts +134 -134
  172. package/src/workspace/mcp-adapters/windsurf.ts +91 -91
  173. package/src/workspace/sanitizer.ts +60 -60
  174. package/src/workspace/workflow-sync.ts +131 -131
@@ -1,328 +1,328 @@
1
- /**
2
- * Memory Compact Engine
3
- *
4
- * Dual-mode memory management inspired by:
5
- * - Mem0's ADD/UPDATE/DELETE/NONE decision model with content merging
6
- * - Cipher's dual-mode architecture (LLM + heuristic fallback)
7
- *
8
- * Two paths:
9
- * 1. LLM mode (compactOnWrite): Single LLM call → extract facts + compare + decide + merge
10
- * 2. Free mode (heuristicCompact): Vector similarity → rule-based ADD/UPDATE/NONE
11
- *
12
- * Both paths produce a CompactDecision that the caller executes.
13
- */
14
-
15
- import { callLLM, isLLMEnabled } from './provider.js';
16
-
17
- /** The decision the compact engine makes for each memory operation */
18
- export type MemoryAction = 'ADD' | 'UPDATE' | 'DELETE' | 'NONE';
19
-
20
- /** Existing memory entry for comparison */
21
- export interface ExistingMemory {
22
- id: number;
23
- title: string;
24
- narrative: string;
25
- facts: string;
26
- score: number;
27
- }
28
-
29
- /** The compact decision — what to do with the new memory */
30
- export interface CompactDecision {
31
- action: MemoryAction;
32
- /** ID of existing memory to UPDATE/DELETE */
33
- targetId?: number;
34
- /** Brief explanation of why this action was chosen */
35
- reason: string;
36
- /** Merged narrative for UPDATE (Mem0-style rewrite) */
37
- mergedNarrative?: string;
38
- /** Merged facts for UPDATE */
39
- mergedFacts?: string[];
40
- /** LLM-extracted enriched facts (only in LLM mode) */
41
- enrichedFacts?: string[];
42
- /** Whether LLM was used for this decision */
43
- usedLLM: boolean;
44
- }
45
-
46
- /**
47
- * Unified Compact on Write prompt (Mem0-inspired, single LLM call).
48
- *
49
- * This prompt does 3 things in 1 call:
50
- * 1. Extract structured facts from the new content
51
- * 2. Compare with existing similar memories
52
- * 3. Decide ADD/UPDATE/DELETE/NONE and merge if needed
53
- */
54
- const COMPACT_ON_WRITE_PROMPT = `You are a smart coding memory manager. You control the memory of a cross-IDE coding assistant.
55
-
56
- You receive a NEW MEMORY to store and a list of EXISTING similar memories. Your job:
57
- 1. Extract the key facts from the new memory
58
- 2. Compare with existing memories
59
- 3. Decide the best action
60
-
61
- Actions:
62
- - ADD: New memory contains unique information. Store it as-is.
63
- - UPDATE: New memory supersedes or improves an existing one. Merge them into a single, comprehensive memory.
64
- - DELETE: An existing memory is outdated/contradicted by the new one. Remove it.
65
- - NONE: New memory is redundant. Existing memories already cover this. Skip storing.
66
-
67
- Decision rules:
68
- - Same topic updated (e.g., "MySQL → PostgreSQL"): UPDATE the old memory with merged content
69
- - Bug fixed that was reported as open: UPDATE the bug report to include the fix
70
- - Task completed that was tracked as in-progress: UPDATE to mark completed
71
- - Minor variation of existing memory: NONE (skip)
72
- - Completely new topic: ADD
73
- - Old info directly contradicted: DELETE the old one
74
- - Prefer UPDATE over ADD — keep memory count low, merge information
75
-
76
- For UPDATE: write a merged narrative that combines the best of both old and new, preserving all important details. Also merge the facts lists, removing duplicates.
77
-
78
- Respond in JSON only:
79
- {
80
- "action": "ADD" | "UPDATE" | "DELETE" | "NONE",
81
- "targetId": null or existing_memory_id_number,
82
- "reason": "brief explanation of decision",
83
- "mergedNarrative": "merged narrative text (required for UPDATE, null otherwise)",
84
- "mergedFacts": ["merged fact 1", "merged fact 2"] or null,
85
- "extractedFacts": ["fact extracted from new content 1", "fact 2"]
86
- }`;
87
-
88
- // ─── Heuristic Constants (Cipher-inspired) ───────────────────────────
89
-
90
- /** Similarity threshold above which we consider memories as "same topic" */
91
- const SIMILARITY_HIGH = 0.75;
92
- /** Similarity threshold for "related but different" */
93
- const SIMILARITY_MEDIUM = 0.5;
94
-
95
- // ─── LLM Mode ────────────────────────────────────────────────────────
96
-
97
- /**
98
- * Compact on Write — LLM mode.
99
- *
100
- * Single LLM call that extracts facts, compares with existing memories,
101
- * and decides ADD/UPDATE/DELETE/NONE with merged content.
102
- *
103
- * Inspired by Mem0's `get_update_memory_messages` but unified into 1 call.
104
- */
105
- export async function compactOnWrite(
106
- newMemory: { title: string; narrative: string; facts: string[] },
107
- existingMemories: ExistingMemory[],
108
- ): Promise<CompactDecision> {
109
- if (!isLLMEnabled()) {
110
- return heuristicCompact(newMemory, existingMemories);
111
- }
112
-
113
- if (existingMemories.length === 0) {
114
- return { action: 'ADD', reason: 'No existing memories to compare', usedLLM: false };
115
- }
116
-
117
- // Build the prompt with new + existing memories (Mem0's temp_uuid_mapping approach)
118
- const existingList = existingMemories
119
- .map(m => `[ID: ${m.id}] (similarity: ${m.score.toFixed(2)}) ${m.title}\n Narrative: ${m.narrative}\n Facts: ${m.facts}`)
120
- .join('\n\n');
121
-
122
- const userMessage = `NEW MEMORY:
123
- Title: ${newMemory.title}
124
- Narrative: ${newMemory.narrative}
125
- Facts: ${newMemory.facts.join('; ')}
126
-
127
- EXISTING SIMILAR MEMORIES:
128
- ${existingList}`;
129
-
130
- try {
131
- const response = await callLLM(COMPACT_ON_WRITE_PROMPT, userMessage);
132
-
133
- // Parse response — handle markdown code blocks
134
- let content = response.content.trim();
135
- if (content.startsWith('```')) {
136
- content = content.replace(/^```(?:json)?\s*/, '').replace(/\s*```$/, '');
137
- }
138
-
139
- const parsed = JSON.parse(content) as {
140
- action: string;
141
- targetId?: number | null;
142
- reason?: string;
143
- mergedNarrative?: string | null;
144
- mergedFacts?: string[] | null;
145
- extractedFacts?: string[] | null;
146
- };
147
-
148
- // Validate action
149
- const action = parsed.action?.toUpperCase() as MemoryAction;
150
- if (!action || !['ADD', 'UPDATE', 'DELETE', 'NONE'].includes(action)) {
151
- return { action: 'ADD', reason: 'LLM response invalid, defaulting to ADD', usedLLM: true };
152
- }
153
-
154
- // Validate targetId exists in existing memories for UPDATE/DELETE
155
- if ((action === 'UPDATE' || action === 'DELETE') && parsed.targetId != null) {
156
- const targetExists = existingMemories.some(m => m.id === parsed.targetId);
157
- if (!targetExists) {
158
- // LLM hallucinated an ID (common issue noted in Mem0's code)
159
- return { action: 'ADD', reason: `LLM referenced non-existent memory #${parsed.targetId}, defaulting to ADD`, usedLLM: true };
160
- }
161
- }
162
-
163
- return {
164
- action,
165
- targetId: parsed.targetId ?? undefined,
166
- reason: parsed.reason ?? 'LLM decision',
167
- mergedNarrative: parsed.mergedNarrative ?? undefined,
168
- mergedFacts: parsed.mergedFacts ?? undefined,
169
- enrichedFacts: parsed.extractedFacts ?? undefined,
170
- usedLLM: true,
171
- };
172
- } catch (err) {
173
- // LLM failed — fall back to heuristic
174
- console.error(`[memorix] LLM compact failed, falling back to heuristic:`, (err as Error)?.message ?? err);
175
- return heuristicCompact(newMemory, existingMemories);
176
- }
177
- }
178
-
179
- // ─── Free Mode (Heuristic) ───────────────────────────────────────────
180
-
181
- /**
182
- * Heuristic Compact — Free mode, no LLM needed.
183
- *
184
- * Uses vector similarity scores from search results to make decisions.
185
- * Inspired by Cipher's fallback logic in extract_and_operate_memory.ts.
186
- *
187
- * Decision logic:
188
- * - score >= 0.75: Very similar → check if new is more complete
189
- * - New is longer/richer → UPDATE (merge)
190
- * - Otherwise → NONE (skip, already covered)
191
- * - score >= 0.50: Related topic
192
- * - Same entity + same type → UPDATE if new has more facts
193
- * - Otherwise → ADD (different enough)
194
- * - score < 0.50: Different topic → ADD
195
- */
196
- export function heuristicCompact(
197
- newMemory: { title: string; narrative: string; facts: string[] },
198
- existingMemories: ExistingMemory[],
199
- ): CompactDecision {
200
- if (existingMemories.length === 0) {
201
- return { action: 'ADD', reason: 'No existing memories to compare', usedLLM: false };
202
- }
203
-
204
- const best = existingMemories[0]; // Already sorted by score
205
-
206
- // High similarity — very likely same topic
207
- if (best.score >= SIMILARITY_HIGH) {
208
- const newLength = newMemory.narrative.length + newMemory.facts.join(' ').length;
209
- const oldLength = best.narrative.length + best.facts.length;
210
-
211
- if (newLength > oldLength * 1.2) {
212
- // New memory is substantially richer — UPDATE with merged content
213
- const mergedNarrative = mergeTexts(best.narrative, newMemory.narrative);
214
- const mergedFacts = mergeFacts(best.facts, newMemory.facts.join('\n'));
215
- return {
216
- action: 'UPDATE',
217
- targetId: best.id,
218
- reason: `New memory is more comprehensive (${newLength} > ${oldLength} chars)`,
219
- mergedNarrative,
220
- mergedFacts,
221
- usedLLM: false,
222
- };
223
- }
224
-
225
- // New memory is not richer — skip it
226
- return {
227
- action: 'NONE',
228
- targetId: best.id,
229
- reason: `Existing memory #${best.id} already covers this topic (similarity: ${best.score.toFixed(2)})`,
230
- usedLLM: false,
231
- };
232
- }
233
-
234
- // Medium similarity — related but might be different enough
235
- if (best.score >= SIMILARITY_MEDIUM) {
236
- const newFactCount = newMemory.facts.length;
237
- const oldFactCount = best.facts.split('\n').filter(Boolean).length;
238
-
239
- if (newFactCount > oldFactCount) {
240
- // New has more facts — UPDATE
241
- const mergedNarrative = mergeTexts(best.narrative, newMemory.narrative);
242
- const mergedFacts = mergeFacts(best.facts, newMemory.facts.join('\n'));
243
- return {
244
- action: 'UPDATE',
245
- targetId: best.id,
246
- reason: `New memory has more facts (${newFactCount} > ${oldFactCount}), merging`,
247
- mergedNarrative,
248
- mergedFacts,
249
- usedLLM: false,
250
- };
251
- }
252
-
253
- // Related but not richer — ADD as separate memory
254
- return { action: 'ADD', reason: `Related to #${best.id} but different enough to keep separate`, usedLLM: false };
255
- }
256
-
257
- // Low similarity — new topic
258
- return { action: 'ADD', reason: 'No similar memories found', usedLLM: false };
259
- }
260
-
261
- // ─── Content Merging Utilities ───────────────────────────────────────
262
-
263
- /**
264
- * Merge two narrative texts, keeping the more comprehensive version
265
- * while preserving unique information from both.
266
- */
267
- function mergeTexts(oldText: string, newText: string): string {
268
- // If new text is significantly longer, prefer it
269
- if (newText.length > oldText.length * 1.5) return newText;
270
- // If old text is significantly longer, append new info
271
- if (oldText.length > newText.length * 1.5) return oldText;
272
- // Similar length — combine with separator
273
- return `${newText}\n\n[Previous context]: ${oldText}`;
274
- }
275
-
276
- /**
277
- * Merge two fact lists, removing duplicates.
278
- * oldFacts is newline-separated string, newFacts is array.
279
- */
280
- function mergeFacts(oldFactsStr: string, newFactsStr: string): string[] {
281
- const oldFacts = oldFactsStr.split('\n').filter(Boolean).map(f => f.trim());
282
- const newFacts = newFactsStr.split('\n').filter(Boolean).map(f => f.trim());
283
-
284
- const seen = new Set<string>();
285
- const merged: string[] = [];
286
-
287
- // Add new facts first (they're more recent)
288
- for (const f of newFacts) {
289
- const normalized = f.toLowerCase();
290
- if (!seen.has(normalized)) {
291
- seen.add(normalized);
292
- merged.push(f);
293
- }
294
- }
295
-
296
- // Add old facts that aren't duplicates
297
- for (const f of oldFacts) {
298
- const normalized = f.toLowerCase();
299
- if (!seen.has(normalized)) {
300
- seen.add(normalized);
301
- merged.push(f);
302
- }
303
- }
304
-
305
- return merged;
306
- }
307
-
308
- // ─── Batch Dedup (for memorix_deduplicate tool) ──────────────────────
309
-
310
- /**
311
- * LLM-powered dedup for batch cleanup.
312
- * Compares a new memory against existing ones and returns a decision.
313
- * Used by memorix_deduplicate tool.
314
- */
315
- export async function deduplicateMemory(
316
- newMemory: { title: string; narrative: string; facts: string[] },
317
- existingMemories: Array<{ id: number; title: string; narrative: string; facts: string }>,
318
- ): Promise<CompactDecision | null> {
319
- if (!isLLMEnabled()) return null;
320
- if (existingMemories.length === 0) return { action: 'ADD', reason: 'No existing memories', usedLLM: false };
321
-
322
- const asExisting: ExistingMemory[] = existingMemories.map(m => ({
323
- ...m,
324
- score: 0.8, // Batch dedup assumes high similarity (pre-filtered)
325
- }));
326
-
327
- return compactOnWrite(newMemory, asExisting);
328
- }
1
+ /**
2
+ * Memory Compact Engine
3
+ *
4
+ * Dual-mode memory management inspired by:
5
+ * - Mem0's ADD/UPDATE/DELETE/NONE decision model with content merging
6
+ * - Cipher's dual-mode architecture (LLM + heuristic fallback)
7
+ *
8
+ * Two paths:
9
+ * 1. LLM mode (compactOnWrite): Single LLM call → extract facts + compare + decide + merge
10
+ * 2. Free mode (heuristicCompact): Vector similarity → rule-based ADD/UPDATE/NONE
11
+ *
12
+ * Both paths produce a CompactDecision that the caller executes.
13
+ */
14
+
15
+ import { callLLM, isLLMEnabled } from './provider.js';
16
+
17
+ /** The decision the compact engine makes for each memory operation */
18
+ export type MemoryAction = 'ADD' | 'UPDATE' | 'DELETE' | 'NONE';
19
+
20
+ /** Existing memory entry for comparison */
21
+ export interface ExistingMemory {
22
+ id: number;
23
+ title: string;
24
+ narrative: string;
25
+ facts: string;
26
+ score: number;
27
+ }
28
+
29
+ /** The compact decision — what to do with the new memory */
30
+ export interface CompactDecision {
31
+ action: MemoryAction;
32
+ /** ID of existing memory to UPDATE/DELETE */
33
+ targetId?: number;
34
+ /** Brief explanation of why this action was chosen */
35
+ reason: string;
36
+ /** Merged narrative for UPDATE (Mem0-style rewrite) */
37
+ mergedNarrative?: string;
38
+ /** Merged facts for UPDATE */
39
+ mergedFacts?: string[];
40
+ /** LLM-extracted enriched facts (only in LLM mode) */
41
+ enrichedFacts?: string[];
42
+ /** Whether LLM was used for this decision */
43
+ usedLLM: boolean;
44
+ }
45
+
46
+ /**
47
+ * Unified Compact on Write prompt (Mem0-inspired, single LLM call).
48
+ *
49
+ * This prompt does 3 things in 1 call:
50
+ * 1. Extract structured facts from the new content
51
+ * 2. Compare with existing similar memories
52
+ * 3. Decide ADD/UPDATE/DELETE/NONE and merge if needed
53
+ */
54
+ const COMPACT_ON_WRITE_PROMPT = `You are a smart coding memory manager. You control the memory of a cross-IDE coding assistant.
55
+
56
+ You receive a NEW MEMORY to store and a list of EXISTING similar memories. Your job:
57
+ 1. Extract the key facts from the new memory
58
+ 2. Compare with existing memories
59
+ 3. Decide the best action
60
+
61
+ Actions:
62
+ - ADD: New memory contains unique information. Store it as-is.
63
+ - UPDATE: New memory supersedes or improves an existing one. Merge them into a single, comprehensive memory.
64
+ - DELETE: An existing memory is outdated/contradicted by the new one. Remove it.
65
+ - NONE: New memory is redundant. Existing memories already cover this. Skip storing.
66
+
67
+ Decision rules:
68
+ - Same topic updated (e.g., "MySQL → PostgreSQL"): UPDATE the old memory with merged content
69
+ - Bug fixed that was reported as open: UPDATE the bug report to include the fix
70
+ - Task completed that was tracked as in-progress: UPDATE to mark completed
71
+ - Minor variation of existing memory: NONE (skip)
72
+ - Completely new topic: ADD
73
+ - Old info directly contradicted: DELETE the old one
74
+ - Prefer UPDATE over ADD — keep memory count low, merge information
75
+
76
+ For UPDATE: write a merged narrative that combines the best of both old and new, preserving all important details. Also merge the facts lists, removing duplicates.
77
+
78
+ Respond in JSON only:
79
+ {
80
+ "action": "ADD" | "UPDATE" | "DELETE" | "NONE",
81
+ "targetId": null or existing_memory_id_number,
82
+ "reason": "brief explanation of decision",
83
+ "mergedNarrative": "merged narrative text (required for UPDATE, null otherwise)",
84
+ "mergedFacts": ["merged fact 1", "merged fact 2"] or null,
85
+ "extractedFacts": ["fact extracted from new content 1", "fact 2"]
86
+ }`;
87
+
88
+ // ─── Heuristic Constants (Cipher-inspired) ───────────────────────────
89
+
90
+ /** Similarity threshold above which we consider memories as "same topic" */
91
+ const SIMILARITY_HIGH = 0.75;
92
+ /** Similarity threshold for "related but different" */
93
+ const SIMILARITY_MEDIUM = 0.5;
94
+
95
+ // ─── LLM Mode ────────────────────────────────────────────────────────
96
+
97
+ /**
98
+ * Compact on Write — LLM mode.
99
+ *
100
+ * Single LLM call that extracts facts, compares with existing memories,
101
+ * and decides ADD/UPDATE/DELETE/NONE with merged content.
102
+ *
103
+ * Inspired by Mem0's `get_update_memory_messages` but unified into 1 call.
104
+ */
105
+ export async function compactOnWrite(
106
+ newMemory: { title: string; narrative: string; facts: string[] },
107
+ existingMemories: ExistingMemory[],
108
+ ): Promise<CompactDecision> {
109
+ if (!isLLMEnabled()) {
110
+ return heuristicCompact(newMemory, existingMemories);
111
+ }
112
+
113
+ if (existingMemories.length === 0) {
114
+ return { action: 'ADD', reason: 'No existing memories to compare', usedLLM: false };
115
+ }
116
+
117
+ // Build the prompt with new + existing memories (Mem0's temp_uuid_mapping approach)
118
+ const existingList = existingMemories
119
+ .map(m => `[ID: ${m.id}] (similarity: ${m.score.toFixed(2)}) ${m.title}\n Narrative: ${m.narrative}\n Facts: ${m.facts}`)
120
+ .join('\n\n');
121
+
122
+ const userMessage = `NEW MEMORY:
123
+ Title: ${newMemory.title}
124
+ Narrative: ${newMemory.narrative}
125
+ Facts: ${newMemory.facts.join('; ')}
126
+
127
+ EXISTING SIMILAR MEMORIES:
128
+ ${existingList}`;
129
+
130
+ try {
131
+ const response = await callLLM(COMPACT_ON_WRITE_PROMPT, userMessage);
132
+
133
+ // Parse response — handle markdown code blocks
134
+ let content = response.content.trim();
135
+ if (content.startsWith('```')) {
136
+ content = content.replace(/^```(?:json)?\s*/, '').replace(/\s*```$/, '');
137
+ }
138
+
139
+ const parsed = JSON.parse(content) as {
140
+ action: string;
141
+ targetId?: number | null;
142
+ reason?: string;
143
+ mergedNarrative?: string | null;
144
+ mergedFacts?: string[] | null;
145
+ extractedFacts?: string[] | null;
146
+ };
147
+
148
+ // Validate action
149
+ const action = parsed.action?.toUpperCase() as MemoryAction;
150
+ if (!action || !['ADD', 'UPDATE', 'DELETE', 'NONE'].includes(action)) {
151
+ return { action: 'ADD', reason: 'LLM response invalid, defaulting to ADD', usedLLM: true };
152
+ }
153
+
154
+ // Validate targetId exists in existing memories for UPDATE/DELETE
155
+ if ((action === 'UPDATE' || action === 'DELETE') && parsed.targetId != null) {
156
+ const targetExists = existingMemories.some(m => m.id === parsed.targetId);
157
+ if (!targetExists) {
158
+ // LLM hallucinated an ID (common issue noted in Mem0's code)
159
+ return { action: 'ADD', reason: `LLM referenced non-existent memory #${parsed.targetId}, defaulting to ADD`, usedLLM: true };
160
+ }
161
+ }
162
+
163
+ return {
164
+ action,
165
+ targetId: parsed.targetId ?? undefined,
166
+ reason: parsed.reason ?? 'LLM decision',
167
+ mergedNarrative: parsed.mergedNarrative ?? undefined,
168
+ mergedFacts: parsed.mergedFacts ?? undefined,
169
+ enrichedFacts: parsed.extractedFacts ?? undefined,
170
+ usedLLM: true,
171
+ };
172
+ } catch (err) {
173
+ // LLM failed — fall back to heuristic
174
+ console.error(`[memorix] LLM compact failed, falling back to heuristic:`, (err as Error)?.message ?? err);
175
+ return heuristicCompact(newMemory, existingMemories);
176
+ }
177
+ }
178
+
179
+ // ─── Free Mode (Heuristic) ───────────────────────────────────────────
180
+
181
+ /**
182
+ * Heuristic Compact — Free mode, no LLM needed.
183
+ *
184
+ * Uses vector similarity scores from search results to make decisions.
185
+ * Inspired by Cipher's fallback logic in extract_and_operate_memory.ts.
186
+ *
187
+ * Decision logic:
188
+ * - score >= 0.75: Very similar → check if new is more complete
189
+ * - New is longer/richer → UPDATE (merge)
190
+ * - Otherwise → NONE (skip, already covered)
191
+ * - score >= 0.50: Related topic
192
+ * - Same entity + same type → UPDATE if new has more facts
193
+ * - Otherwise → ADD (different enough)
194
+ * - score < 0.50: Different topic → ADD
195
+ */
196
+ export function heuristicCompact(
197
+ newMemory: { title: string; narrative: string; facts: string[] },
198
+ existingMemories: ExistingMemory[],
199
+ ): CompactDecision {
200
+ if (existingMemories.length === 0) {
201
+ return { action: 'ADD', reason: 'No existing memories to compare', usedLLM: false };
202
+ }
203
+
204
+ const best = existingMemories[0]; // Already sorted by score
205
+
206
+ // High similarity — very likely same topic
207
+ if (best.score >= SIMILARITY_HIGH) {
208
+ const newLength = newMemory.narrative.length + newMemory.facts.join(' ').length;
209
+ const oldLength = best.narrative.length + best.facts.length;
210
+
211
+ if (newLength > oldLength * 1.2) {
212
+ // New memory is substantially richer — UPDATE with merged content
213
+ const mergedNarrative = mergeTexts(best.narrative, newMemory.narrative);
214
+ const mergedFacts = mergeFacts(best.facts, newMemory.facts.join('\n'));
215
+ return {
216
+ action: 'UPDATE',
217
+ targetId: best.id,
218
+ reason: `New memory is more comprehensive (${newLength} > ${oldLength} chars)`,
219
+ mergedNarrative,
220
+ mergedFacts,
221
+ usedLLM: false,
222
+ };
223
+ }
224
+
225
+ // New memory is not richer — skip it
226
+ return {
227
+ action: 'NONE',
228
+ targetId: best.id,
229
+ reason: `Existing memory #${best.id} already covers this topic (similarity: ${best.score.toFixed(2)})`,
230
+ usedLLM: false,
231
+ };
232
+ }
233
+
234
+ // Medium similarity — related but might be different enough
235
+ if (best.score >= SIMILARITY_MEDIUM) {
236
+ const newFactCount = newMemory.facts.length;
237
+ const oldFactCount = best.facts.split('\n').filter(Boolean).length;
238
+
239
+ if (newFactCount > oldFactCount) {
240
+ // New has more facts — UPDATE
241
+ const mergedNarrative = mergeTexts(best.narrative, newMemory.narrative);
242
+ const mergedFacts = mergeFacts(best.facts, newMemory.facts.join('\n'));
243
+ return {
244
+ action: 'UPDATE',
245
+ targetId: best.id,
246
+ reason: `New memory has more facts (${newFactCount} > ${oldFactCount}), merging`,
247
+ mergedNarrative,
248
+ mergedFacts,
249
+ usedLLM: false,
250
+ };
251
+ }
252
+
253
+ // Related but not richer — ADD as separate memory
254
+ return { action: 'ADD', reason: `Related to #${best.id} but different enough to keep separate`, usedLLM: false };
255
+ }
256
+
257
+ // Low similarity — new topic
258
+ return { action: 'ADD', reason: 'No similar memories found', usedLLM: false };
259
+ }
260
+
261
+ // ─── Content Merging Utilities ───────────────────────────────────────
262
+
263
+ /**
264
+ * Merge two narrative texts, keeping the more comprehensive version
265
+ * while preserving unique information from both.
266
+ */
267
+ function mergeTexts(oldText: string, newText: string): string {
268
+ // If new text is significantly longer, prefer it
269
+ if (newText.length > oldText.length * 1.5) return newText;
270
+ // If old text is significantly longer, append new info
271
+ if (oldText.length > newText.length * 1.5) return oldText;
272
+ // Similar length — combine with separator
273
+ return `${newText}\n\n[Previous context]: ${oldText}`;
274
+ }
275
+
276
+ /**
277
+ * Merge two fact lists, removing duplicates.
278
+ * oldFacts is newline-separated string, newFacts is array.
279
+ */
280
+ function mergeFacts(oldFactsStr: string, newFactsStr: string): string[] {
281
+ const oldFacts = oldFactsStr.split('\n').filter(Boolean).map(f => f.trim());
282
+ const newFacts = newFactsStr.split('\n').filter(Boolean).map(f => f.trim());
283
+
284
+ const seen = new Set<string>();
285
+ const merged: string[] = [];
286
+
287
+ // Add new facts first (they're more recent)
288
+ for (const f of newFacts) {
289
+ const normalized = f.toLowerCase();
290
+ if (!seen.has(normalized)) {
291
+ seen.add(normalized);
292
+ merged.push(f);
293
+ }
294
+ }
295
+
296
+ // Add old facts that aren't duplicates
297
+ for (const f of oldFacts) {
298
+ const normalized = f.toLowerCase();
299
+ if (!seen.has(normalized)) {
300
+ seen.add(normalized);
301
+ merged.push(f);
302
+ }
303
+ }
304
+
305
+ return merged;
306
+ }
307
+
308
+ // ─── Batch Dedup (for memorix_deduplicate tool) ──────────────────────
309
+
310
+ /**
311
+ * LLM-powered dedup for batch cleanup.
312
+ * Compares a new memory against existing ones and returns a decision.
313
+ * Used by memorix_deduplicate tool.
314
+ */
315
+ export async function deduplicateMemory(
316
+ newMemory: { title: string; narrative: string; facts: string[] },
317
+ existingMemories: Array<{ id: number; title: string; narrative: string; facts: string }>,
318
+ ): Promise<CompactDecision | null> {
319
+ if (!isLLMEnabled()) return null;
320
+ if (existingMemories.length === 0) return { action: 'ADD', reason: 'No existing memories', usedLLM: false };
321
+
322
+ const asExisting: ExistingMemory[] = existingMemories.map(m => ({
323
+ ...m,
324
+ score: 0.8, // Batch dedup assumes high similarity (pre-filtered)
325
+ }));
326
+
327
+ return compactOnWrite(newMemory, asExisting);
328
+ }