memorix 1.1.7 → 1.1.9

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 (185) hide show
  1. package/CHANGELOG.md +22 -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 +852 -214
  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 +129 -62
  11. package/dist/index.js.map +1 -1
  12. package/dist/memcode-runtime/CHANGELOG.md +22 -0
  13. package/dist/memcode-runtime/package.json +4 -4
  14. package/dist/sdk.js +129 -62
  15. package/dist/sdk.js.map +1 -1
  16. package/docs/AGENT_OPERATOR_PLAYBOOK.md +18 -0
  17. package/docs/API_REFERENCE.md +2 -0
  18. package/docs/CONFIGURATION.md +18 -0
  19. package/docs/DESIGN_DECISIONS.md +357 -357
  20. package/docs/SETUP.md +10 -0
  21. package/docs/dev-log/progress.txt +23 -30
  22. package/package.json +1 -1
  23. package/src/audit/index.ts +156 -156
  24. package/src/cli/commands/agent-integrations.ts +623 -0
  25. package/src/cli/commands/audit-list.ts +89 -89
  26. package/src/cli/commands/background.ts +659 -659
  27. package/src/cli/commands/cleanup.ts +255 -255
  28. package/src/cli/commands/codegraph.ts +4 -0
  29. package/src/cli/commands/config-get.ts +9 -2
  30. package/src/cli/commands/doctor.ts +26 -0
  31. package/src/cli/commands/formation.ts +48 -48
  32. package/src/cli/commands/git-hook-install.ts +111 -111
  33. package/src/cli/commands/handoff.ts +66 -66
  34. package/src/cli/commands/hooks-status.ts +63 -63
  35. package/src/cli/commands/ingest-commit.ts +153 -153
  36. package/src/cli/commands/ingest-image.ts +73 -73
  37. package/src/cli/commands/ingest-log.ts +180 -180
  38. package/src/cli/commands/ingest.ts +44 -44
  39. package/src/cli/commands/integrate-shared.ts +15 -15
  40. package/src/cli/commands/lock.ts +96 -96
  41. package/src/cli/commands/message.ts +121 -121
  42. package/src/cli/commands/poll.ts +70 -70
  43. package/src/cli/commands/purge-all-memory.ts +85 -85
  44. package/src/cli/commands/purge-project-memory.ts +83 -83
  45. package/src/cli/commands/reasoning.ts +132 -132
  46. package/src/cli/commands/repair.ts +60 -0
  47. package/src/cli/commands/retention.ts +108 -108
  48. package/src/cli/commands/serve-shared.ts +118 -118
  49. package/src/cli/commands/setup.ts +3 -3
  50. package/src/cli/commands/skills.ts +123 -123
  51. package/src/cli/commands/task.ts +192 -192
  52. package/src/cli/commands/transfer.ts +73 -73
  53. package/src/cli/commands/uninstall-project-artifacts.ts +85 -85
  54. package/src/cli/index.ts +3 -1
  55. package/src/cli/tui/ChatView.tsx +234 -234
  56. package/src/cli/tui/CommandBar.tsx +312 -312
  57. package/src/cli/tui/ContextRail.tsx +118 -118
  58. package/src/cli/tui/HeaderBar.tsx +72 -72
  59. package/src/cli/tui/LogoBanner.tsx +51 -51
  60. package/src/cli/tui/Panels.tsx +632 -632
  61. package/src/cli/tui/Sidebar.tsx +179 -179
  62. package/src/cli/tui/chat-service.ts +742 -742
  63. package/src/cli/tui/data.ts +547 -547
  64. package/src/cli/tui/index.ts +41 -41
  65. package/src/cli/tui/markdown-render.tsx +371 -371
  66. package/src/cli/tui/theme.ts +178 -178
  67. package/src/cli/tui/use-mouse.ts +157 -157
  68. package/src/cli/tui/useNavigation.ts +56 -56
  69. package/src/cli/update-checker.ts +211 -211
  70. package/src/cli/version.ts +7 -7
  71. package/src/cli/workbench.ts +1 -1
  72. package/src/codegraph/auto-context.ts +6 -0
  73. package/src/codegraph/context-pack.ts +7 -6
  74. package/src/codegraph/exclude.ts +47 -0
  75. package/src/codegraph/lite-provider.ts +5 -24
  76. package/src/codegraph/project-context.ts +13 -15
  77. package/src/compact/token-budget.ts +74 -74
  78. package/src/config/behavior.ts +59 -59
  79. package/src/config/resolved-config.ts +6 -0
  80. package/src/config/toml-loader.ts +4 -0
  81. package/src/config/yaml-loader.ts +7 -0
  82. package/src/dashboard/project-classification.ts +64 -64
  83. package/src/dashboard/static/index.html +201 -201
  84. package/src/dashboard/static/style.css +3584 -3584
  85. package/src/embedding/fastembed-provider.ts +142 -142
  86. package/src/embedding/transformers-provider.ts +111 -111
  87. package/src/git/extractor.ts +209 -209
  88. package/src/git/hooks-path.ts +85 -85
  89. package/src/git/noise-filter.ts +210 -210
  90. package/src/hooks/installers/index.ts +4 -4
  91. package/src/hooks/official-skills.ts +1 -1
  92. package/src/hooks/pattern-detector.ts +173 -173
  93. package/src/hooks/rules/memorix-agent-rules.md +2 -2
  94. package/src/hooks/significance-filter.ts +250 -250
  95. package/src/llm/memory-manager.ts +328 -328
  96. package/src/llm/provider.ts +885 -885
  97. package/src/llm/quality.ts +248 -248
  98. package/src/memory/attribution-guard.ts +249 -249
  99. package/src/memory/auto-relations.ts +107 -107
  100. package/src/memory/consolidation.ts +302 -302
  101. package/src/memory/disclosure-policy.ts +141 -141
  102. package/src/memory/entity-extractor.ts +197 -197
  103. package/src/memory/formation/evaluate.ts +217 -217
  104. package/src/memory/formation/extract.ts +361 -361
  105. package/src/memory/formation/index.ts +417 -417
  106. package/src/memory/formation/resolve.ts +344 -344
  107. package/src/memory/formation/types.ts +315 -315
  108. package/src/memory/freshness.ts +122 -122
  109. package/src/memory/graph.ts +197 -197
  110. package/src/memory/refs.ts +94 -94
  111. package/src/memory/retention.ts +433 -433
  112. package/src/memory/secret-filter.ts +79 -79
  113. package/src/memory/session.ts +523 -523
  114. package/src/multimodal/image-loader.ts +143 -143
  115. package/src/orchestrate/adapters/claude-stream.ts +192 -192
  116. package/src/orchestrate/adapters/claude.ts +111 -111
  117. package/src/orchestrate/adapters/codex-stream.ts +134 -134
  118. package/src/orchestrate/adapters/codex.ts +41 -41
  119. package/src/orchestrate/adapters/gemini-stream.ts +166 -166
  120. package/src/orchestrate/adapters/gemini.ts +42 -42
  121. package/src/orchestrate/adapters/index.ts +73 -73
  122. package/src/orchestrate/adapters/opencode-stream.ts +143 -143
  123. package/src/orchestrate/adapters/opencode.ts +47 -47
  124. package/src/orchestrate/adapters/spawn-helper.ts +286 -286
  125. package/src/orchestrate/adapters/types.ts +77 -77
  126. package/src/orchestrate/capability-router.ts +284 -284
  127. package/src/orchestrate/context-compact.ts +188 -188
  128. package/src/orchestrate/cost-tracker.ts +219 -219
  129. package/src/orchestrate/error-recovery.ts +191 -191
  130. package/src/orchestrate/evidence.ts +140 -140
  131. package/src/orchestrate/ledger.ts +110 -110
  132. package/src/orchestrate/memorix-bridge.ts +380 -380
  133. package/src/orchestrate/output-budget.ts +80 -80
  134. package/src/orchestrate/permission.ts +152 -152
  135. package/src/orchestrate/pipeline-trace.ts +131 -131
  136. package/src/orchestrate/prompt-builder.ts +155 -155
  137. package/src/orchestrate/ring-buffer.ts +37 -37
  138. package/src/orchestrate/task-graph.ts +389 -389
  139. package/src/orchestrate/verify-gate.ts +219 -219
  140. package/src/orchestrate/worktree.ts +232 -232
  141. package/src/project/aliases.ts +374 -374
  142. package/src/project/detector.ts +268 -268
  143. package/src/rules/adapters/claude-code.ts +99 -99
  144. package/src/rules/adapters/codex.ts +97 -97
  145. package/src/rules/adapters/copilot.ts +124 -124
  146. package/src/rules/adapters/cursor.ts +114 -114
  147. package/src/rules/adapters/kiro.ts +126 -126
  148. package/src/rules/adapters/trae.ts +56 -56
  149. package/src/rules/adapters/windsurf.ts +83 -83
  150. package/src/rules/syncer.ts +235 -235
  151. package/src/sdk.ts +327 -327
  152. package/src/search/intent-detector.ts +289 -289
  153. package/src/search/query-expansion.ts +52 -52
  154. package/src/server/formation-timeout.ts +27 -27
  155. package/src/server.ts +3 -0
  156. package/src/skills/mini-skills.ts +386 -386
  157. package/src/store/chat-store.ts +119 -119
  158. package/src/store/file-lock.ts +100 -100
  159. package/src/store/graph-store.ts +249 -249
  160. package/src/store/mini-skill-store.ts +349 -349
  161. package/src/store/obs-store.ts +255 -255
  162. package/src/store/orama-store.ts +15 -8
  163. package/src/store/persistence-json.ts +212 -212
  164. package/src/store/persistence.ts +291 -291
  165. package/src/store/project-affinity.ts +195 -195
  166. package/src/store/session-store.ts +259 -259
  167. package/src/store/sqlite-store.ts +339 -339
  168. package/src/team/event-bus.ts +76 -76
  169. package/src/team/file-locks.ts +173 -173
  170. package/src/team/handoff.ts +167 -167
  171. package/src/team/messages.ts +203 -203
  172. package/src/team/poll.ts +132 -132
  173. package/src/team/tasks.ts +211 -211
  174. package/src/wiki/generator.ts +237 -237
  175. package/src/wiki/knowledge-graph.ts +334 -334
  176. package/src/wiki/types.ts +85 -85
  177. package/src/workspace/mcp-adapters/codex.ts +191 -191
  178. package/src/workspace/mcp-adapters/copilot.ts +105 -105
  179. package/src/workspace/mcp-adapters/cursor.ts +53 -53
  180. package/src/workspace/mcp-adapters/kiro.ts +64 -64
  181. package/src/workspace/mcp-adapters/opencode.ts +123 -123
  182. package/src/workspace/mcp-adapters/trae.ts +134 -134
  183. package/src/workspace/mcp-adapters/windsurf.ts +91 -91
  184. package/src/workspace/sanitizer.ts +60 -60
  185. 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
+ }