memorix 1.2.1 → 1.2.2

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 (199) hide show
  1. package/CHANGELOG.md +16 -0
  2. package/README.md +14 -2
  3. package/README.zh-CN.md +14 -2
  4. package/TEAM.md +86 -86
  5. package/dist/cli/index.js +15407 -13779
  6. package/dist/cli/index.js.map +1 -1
  7. package/dist/index.js +1321 -529
  8. package/dist/index.js.map +1 -1
  9. package/dist/maintenance-runner.d.ts +1 -1
  10. package/dist/maintenance-runner.js +8458 -8087
  11. package/dist/maintenance-runner.js.map +1 -1
  12. package/dist/memcode-runtime/CHANGELOG.md +16 -0
  13. package/dist/sdk.d.ts +7 -2
  14. package/dist/sdk.js +1349 -535
  15. package/dist/sdk.js.map +1 -1
  16. package/dist/types.d.ts +49 -1
  17. package/dist/types.js.map +1 -1
  18. package/docs/1.2.2-MEMORY-CONTROL-PLANE.md +434 -0
  19. package/docs/AGENT_OPERATOR_PLAYBOOK.md +4 -0
  20. package/docs/API_REFERENCE.md +24 -4
  21. package/docs/DESIGN_DECISIONS.md +357 -357
  22. package/docs/README.md +1 -1
  23. package/docs/dev-log/progress.txt +91 -11
  24. package/package.json +1 -1
  25. package/plugins/codex/memorix/.codex-plugin/plugin.json +1 -1
  26. package/src/audit/index.ts +156 -156
  27. package/src/cli/command-guide.ts +192 -0
  28. package/src/cli/commands/audit-list.ts +89 -89
  29. package/src/cli/commands/audit.ts +9 -4
  30. package/src/cli/commands/background.ts +659 -659
  31. package/src/cli/commands/cleanup.ts +5 -1
  32. package/src/cli/commands/codegraph.ts +15 -5
  33. package/src/cli/commands/context.ts +3 -2
  34. package/src/cli/commands/doctor.ts +4 -2
  35. package/src/cli/commands/explain.ts +9 -3
  36. package/src/cli/commands/formation.ts +48 -48
  37. package/src/cli/commands/git-hook-install.ts +111 -111
  38. package/src/cli/commands/handoff.ts +75 -61
  39. package/src/cli/commands/hooks-status.ts +63 -63
  40. package/src/cli/commands/identity.ts +116 -0
  41. package/src/cli/commands/ingest-commit.ts +153 -153
  42. package/src/cli/commands/ingest-image.ts +71 -69
  43. package/src/cli/commands/ingest-log.ts +180 -180
  44. package/src/cli/commands/ingest.ts +44 -44
  45. package/src/cli/commands/integrate-shared.ts +15 -15
  46. package/src/cli/commands/lock.ts +93 -92
  47. package/src/cli/commands/memory.ts +58 -21
  48. package/src/cli/commands/message.ts +123 -118
  49. package/src/cli/commands/operator-shared.ts +98 -3
  50. package/src/cli/commands/poll.ts +74 -64
  51. package/src/cli/commands/purge-all-memory.ts +85 -85
  52. package/src/cli/commands/purge-project-memory.ts +83 -83
  53. package/src/cli/commands/reasoning.ts +135 -121
  54. package/src/cli/commands/retention.ts +9 -4
  55. package/src/cli/commands/serve-http.ts +8 -2
  56. package/src/cli/commands/serve-shared.ts +118 -118
  57. package/src/cli/commands/session.ts +29 -3
  58. package/src/cli/commands/skills.ts +124 -119
  59. package/src/cli/commands/status.ts +4 -3
  60. package/src/cli/commands/task.ts +193 -184
  61. package/src/cli/commands/team.ts +14 -10
  62. package/src/cli/commands/transfer.ts +108 -55
  63. package/src/cli/commands/uninstall-project-artifacts.ts +85 -85
  64. package/src/cli/identity.ts +89 -0
  65. package/src/cli/index.ts +96 -19
  66. package/src/cli/invocation.ts +115 -0
  67. package/src/cli/tui/ChatView.tsx +234 -234
  68. package/src/cli/tui/CommandBar.tsx +312 -312
  69. package/src/cli/tui/ContextRail.tsx +118 -118
  70. package/src/cli/tui/HeaderBar.tsx +72 -72
  71. package/src/cli/tui/LogoBanner.tsx +51 -51
  72. package/src/cli/tui/Sidebar.tsx +179 -179
  73. package/src/cli/tui/chat-service.ts +41 -18
  74. package/src/cli/tui/data.ts +23 -44
  75. package/src/cli/tui/index.ts +41 -41
  76. package/src/cli/tui/markdown-render.tsx +371 -371
  77. package/src/cli/tui/operator-context.ts +60 -0
  78. package/src/cli/tui/use-mouse.ts +157 -157
  79. package/src/cli/tui/useNavigation.ts +56 -56
  80. package/src/cli/tui/views/MemoryView.tsx +10 -8
  81. package/src/cli/update-checker.ts +211 -211
  82. package/src/cli/version.ts +7 -7
  83. package/src/cli/workbench.ts +1 -1
  84. package/src/codegraph/auto-context.ts +31 -2
  85. package/src/codegraph/context-pack.ts +1 -0
  86. package/src/codegraph/project-context.ts +2 -0
  87. package/src/compact/engine.ts +26 -10
  88. package/src/compact/index-format.ts +25 -2
  89. package/src/compact/token-budget.ts +74 -74
  90. package/src/dashboard/project-classification.ts +64 -64
  91. package/src/dashboard/server.ts +46 -9
  92. package/src/embedding/fastembed-provider.ts +142 -142
  93. package/src/embedding/transformers-provider.ts +111 -111
  94. package/src/git/extractor.ts +209 -209
  95. package/src/git/hooks-path.ts +85 -85
  96. package/src/hooks/admission.ts +117 -0
  97. package/src/hooks/handler.ts +98 -91
  98. package/src/hooks/pattern-detector.ts +173 -173
  99. package/src/hooks/significance-filter.ts +250 -250
  100. package/src/knowledge/context-assembly.ts +97 -0
  101. package/src/knowledge/workset.ts +179 -10
  102. package/src/llm/memory-manager.ts +328 -328
  103. package/src/llm/provider.ts +885 -885
  104. package/src/llm/quality.ts +248 -248
  105. package/src/memory/admission.ts +57 -0
  106. package/src/memory/attribution-guard.ts +249 -249
  107. package/src/memory/consolidation.ts +13 -2
  108. package/src/memory/disclosure-policy.ts +140 -135
  109. package/src/memory/entity-extractor.ts +197 -197
  110. package/src/memory/export-import.ts +11 -3
  111. package/src/memory/formation/evaluate.ts +217 -217
  112. package/src/memory/formation/extract.ts +361 -361
  113. package/src/memory/formation/index.ts +417 -417
  114. package/src/memory/formation/resolve.ts +344 -344
  115. package/src/memory/formation/types.ts +315 -315
  116. package/src/memory/freshness.ts +122 -122
  117. package/src/memory/graph-context.ts +8 -2
  118. package/src/memory/graph.ts +197 -197
  119. package/src/memory/observations.ts +162 -4
  120. package/src/memory/quality-audit.ts +2 -0
  121. package/src/memory/refs.ts +94 -94
  122. package/src/memory/retention.ts +22 -2
  123. package/src/memory/secret-filter.ts +79 -79
  124. package/src/memory/session.ts +5 -2
  125. package/src/memory/visibility.ts +80 -0
  126. package/src/multimodal/image-loader.ts +143 -143
  127. package/src/orchestrate/adapters/claude-stream.ts +192 -192
  128. package/src/orchestrate/adapters/claude.ts +111 -111
  129. package/src/orchestrate/adapters/codex-stream.ts +134 -134
  130. package/src/orchestrate/adapters/codex.ts +41 -41
  131. package/src/orchestrate/adapters/gemini-stream.ts +166 -166
  132. package/src/orchestrate/adapters/gemini.ts +42 -42
  133. package/src/orchestrate/adapters/index.ts +73 -73
  134. package/src/orchestrate/adapters/opencode-stream.ts +143 -143
  135. package/src/orchestrate/adapters/opencode.ts +47 -47
  136. package/src/orchestrate/adapters/spawn-helper.ts +286 -286
  137. package/src/orchestrate/adapters/types.ts +77 -77
  138. package/src/orchestrate/capability-router.ts +284 -284
  139. package/src/orchestrate/context-compact.ts +188 -188
  140. package/src/orchestrate/cost-tracker.ts +219 -219
  141. package/src/orchestrate/error-recovery.ts +191 -191
  142. package/src/orchestrate/evidence.ts +140 -140
  143. package/src/orchestrate/ledger.ts +110 -110
  144. package/src/orchestrate/memorix-bridge.ts +378 -340
  145. package/src/orchestrate/output-budget.ts +80 -80
  146. package/src/orchestrate/permission.ts +152 -152
  147. package/src/orchestrate/pipeline-trace.ts +131 -131
  148. package/src/orchestrate/prompt-builder.ts +155 -155
  149. package/src/orchestrate/ring-buffer.ts +37 -37
  150. package/src/orchestrate/task-graph.ts +389 -389
  151. package/src/orchestrate/worktree.ts +232 -232
  152. package/src/project/aliases.ts +374 -374
  153. package/src/project/detector.ts +268 -268
  154. package/src/rules/adapters/claude-code.ts +99 -99
  155. package/src/rules/adapters/codex.ts +97 -97
  156. package/src/rules/adapters/copilot.ts +124 -124
  157. package/src/rules/adapters/cursor.ts +114 -114
  158. package/src/rules/adapters/kiro.ts +126 -126
  159. package/src/rules/adapters/trae.ts +56 -56
  160. package/src/rules/adapters/windsurf.ts +83 -83
  161. package/src/rules/syncer.ts +235 -235
  162. package/src/runtime/control-plane-maintenance.ts +1 -0
  163. package/src/runtime/isolated-maintenance.ts +1 -0
  164. package/src/runtime/lifecycle.ts +18 -0
  165. package/src/runtime/maintenance-jobs.ts +1 -0
  166. package/src/runtime/maintenance-runner.ts +2 -0
  167. package/src/runtime/project-maintenance.ts +89 -0
  168. package/src/sdk.ts +334 -304
  169. package/src/search/intent-detector.ts +289 -289
  170. package/src/search/query-expansion.ts +52 -52
  171. package/src/server/formation-timeout.ts +27 -27
  172. package/src/server.ts +260 -81
  173. package/src/skills/mini-skills.ts +386 -386
  174. package/src/store/chat-store.ts +119 -119
  175. package/src/store/graph-store.ts +249 -249
  176. package/src/store/mini-skill-store.ts +349 -349
  177. package/src/store/orama-store.ts +61 -6
  178. package/src/store/persistence-json.ts +212 -212
  179. package/src/store/persistence.ts +291 -291
  180. package/src/store/project-affinity.ts +195 -195
  181. package/src/store/sqlite-db.ts +23 -1
  182. package/src/store/sqlite-store.ts +12 -2
  183. package/src/team/event-bus.ts +76 -76
  184. package/src/team/file-locks.ts +173 -173
  185. package/src/team/handoff.ts +168 -161
  186. package/src/team/messages.ts +203 -203
  187. package/src/team/poll.ts +132 -132
  188. package/src/team/tasks.ts +211 -211
  189. package/src/types.ts +51 -0
  190. package/src/wiki/generator.ts +2 -0
  191. package/src/workspace/mcp-adapters/codex.ts +191 -191
  192. package/src/workspace/mcp-adapters/copilot.ts +105 -105
  193. package/src/workspace/mcp-adapters/cursor.ts +53 -53
  194. package/src/workspace/mcp-adapters/kiro.ts +64 -64
  195. package/src/workspace/mcp-adapters/opencode.ts +123 -123
  196. package/src/workspace/mcp-adapters/trae.ts +134 -134
  197. package/src/workspace/mcp-adapters/windsurf.ts +91 -91
  198. package/src/workspace/sanitizer.ts +60 -60
  199. 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
+ }