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.
- package/CHANGELOG.md +13 -0
- package/CLAUDE.md +6 -1
- package/README.md +21 -0
- package/README.zh-CN.md +21 -0
- package/TEAM.md +86 -86
- package/dist/cli/index.js +730 -150
- package/dist/cli/index.js.map +1 -1
- package/dist/dashboard/static/index.html +201 -201
- package/dist/dashboard/static/style.css +3584 -3584
- package/dist/index.js +15 -8
- package/dist/index.js.map +1 -1
- package/dist/memcode-runtime/CHANGELOG.md +13 -0
- package/dist/memcode-runtime/package.json +4 -4
- package/dist/sdk.js +15 -8
- package/dist/sdk.js.map +1 -1
- package/docs/AGENT_OPERATOR_PLAYBOOK.md +18 -0
- package/docs/DESIGN_DECISIONS.md +357 -357
- package/docs/SETUP.md +10 -0
- package/docs/dev-log/progress.txt +35 -26
- package/package.json +1 -1
- package/src/audit/index.ts +156 -156
- package/src/cli/commands/agent-integrations.ts +623 -0
- package/src/cli/commands/audit-list.ts +89 -89
- package/src/cli/commands/background.ts +659 -659
- package/src/cli/commands/cleanup.ts +255 -255
- package/src/cli/commands/doctor.ts +23 -0
- package/src/cli/commands/formation.ts +48 -48
- package/src/cli/commands/git-hook-install.ts +111 -111
- package/src/cli/commands/handoff.ts +66 -66
- package/src/cli/commands/hooks-status.ts +63 -63
- package/src/cli/commands/ingest-commit.ts +153 -153
- package/src/cli/commands/ingest-image.ts +73 -73
- package/src/cli/commands/ingest-log.ts +180 -180
- package/src/cli/commands/ingest.ts +44 -44
- package/src/cli/commands/integrate-shared.ts +15 -15
- package/src/cli/commands/lock.ts +96 -96
- package/src/cli/commands/message.ts +121 -121
- package/src/cli/commands/poll.ts +70 -70
- package/src/cli/commands/purge-all-memory.ts +85 -85
- package/src/cli/commands/purge-project-memory.ts +83 -83
- package/src/cli/commands/reasoning.ts +132 -132
- package/src/cli/commands/repair.ts +60 -0
- package/src/cli/commands/retention.ts +108 -108
- package/src/cli/commands/serve-shared.ts +118 -118
- package/src/cli/commands/setup.ts +3 -3
- package/src/cli/commands/skills.ts +123 -123
- package/src/cli/commands/task.ts +192 -192
- package/src/cli/commands/transfer.ts +73 -73
- package/src/cli/commands/uninstall-project-artifacts.ts +85 -85
- package/src/cli/index.ts +3 -1
- package/src/cli/tui/ChatView.tsx +234 -234
- package/src/cli/tui/CommandBar.tsx +312 -312
- package/src/cli/tui/ContextRail.tsx +118 -118
- package/src/cli/tui/HeaderBar.tsx +72 -72
- package/src/cli/tui/LogoBanner.tsx +51 -51
- package/src/cli/tui/Panels.tsx +632 -632
- package/src/cli/tui/Sidebar.tsx +179 -179
- package/src/cli/tui/chat-service.ts +742 -742
- package/src/cli/tui/data.ts +547 -547
- package/src/cli/tui/index.ts +41 -41
- package/src/cli/tui/markdown-render.tsx +371 -371
- package/src/cli/tui/theme.ts +178 -178
- package/src/cli/tui/use-mouse.ts +157 -157
- package/src/cli/tui/useNavigation.ts +56 -56
- package/src/cli/update-checker.ts +211 -211
- package/src/cli/version.ts +7 -7
- package/src/cli/workbench.ts +1 -1
- package/src/codegraph/context-pack.ts +5 -1
- package/src/codegraph/lite-provider.ts +5 -1
- package/src/codegraph/project-context.ts +5 -1
- package/src/compact/token-budget.ts +74 -74
- package/src/config/behavior.ts +59 -59
- package/src/dashboard/project-classification.ts +64 -64
- package/src/dashboard/static/index.html +201 -201
- package/src/dashboard/static/style.css +3584 -3584
- package/src/embedding/fastembed-provider.ts +142 -142
- package/src/embedding/transformers-provider.ts +111 -111
- package/src/git/extractor.ts +209 -209
- package/src/git/hooks-path.ts +85 -85
- package/src/git/noise-filter.ts +210 -210
- package/src/hooks/installers/index.ts +4 -4
- package/src/hooks/official-skills.ts +1 -1
- package/src/hooks/pattern-detector.ts +173 -173
- package/src/hooks/rules/memorix-agent-rules.md +2 -2
- package/src/hooks/significance-filter.ts +250 -250
- package/src/llm/memory-manager.ts +328 -328
- package/src/llm/provider.ts +885 -885
- package/src/llm/quality.ts +248 -248
- package/src/memory/attribution-guard.ts +249 -249
- package/src/memory/auto-relations.ts +107 -107
- package/src/memory/consolidation.ts +302 -302
- package/src/memory/disclosure-policy.ts +141 -141
- package/src/memory/entity-extractor.ts +197 -197
- package/src/memory/formation/evaluate.ts +217 -217
- package/src/memory/formation/extract.ts +361 -361
- package/src/memory/formation/index.ts +417 -417
- package/src/memory/formation/resolve.ts +344 -344
- package/src/memory/formation/types.ts +315 -315
- package/src/memory/freshness.ts +122 -122
- package/src/memory/graph.ts +197 -197
- package/src/memory/refs.ts +94 -94
- package/src/memory/retention.ts +433 -433
- package/src/memory/secret-filter.ts +79 -79
- package/src/memory/session.ts +523 -523
- package/src/multimodal/image-loader.ts +143 -143
- package/src/orchestrate/adapters/claude-stream.ts +192 -192
- package/src/orchestrate/adapters/claude.ts +111 -111
- package/src/orchestrate/adapters/codex-stream.ts +134 -134
- package/src/orchestrate/adapters/codex.ts +41 -41
- package/src/orchestrate/adapters/gemini-stream.ts +166 -166
- package/src/orchestrate/adapters/gemini.ts +42 -42
- package/src/orchestrate/adapters/index.ts +73 -73
- package/src/orchestrate/adapters/opencode-stream.ts +143 -143
- package/src/orchestrate/adapters/opencode.ts +47 -47
- package/src/orchestrate/adapters/spawn-helper.ts +286 -286
- package/src/orchestrate/adapters/types.ts +77 -77
- package/src/orchestrate/capability-router.ts +284 -284
- package/src/orchestrate/context-compact.ts +188 -188
- package/src/orchestrate/cost-tracker.ts +219 -219
- package/src/orchestrate/error-recovery.ts +191 -191
- package/src/orchestrate/evidence.ts +140 -140
- package/src/orchestrate/ledger.ts +110 -110
- package/src/orchestrate/memorix-bridge.ts +380 -380
- package/src/orchestrate/output-budget.ts +80 -80
- package/src/orchestrate/permission.ts +152 -152
- package/src/orchestrate/pipeline-trace.ts +131 -131
- package/src/orchestrate/prompt-builder.ts +155 -155
- package/src/orchestrate/ring-buffer.ts +37 -37
- package/src/orchestrate/task-graph.ts +389 -389
- package/src/orchestrate/verify-gate.ts +219 -219
- package/src/orchestrate/worktree.ts +232 -232
- package/src/project/aliases.ts +374 -374
- package/src/project/detector.ts +268 -268
- package/src/rules/adapters/claude-code.ts +99 -99
- package/src/rules/adapters/codex.ts +97 -97
- package/src/rules/adapters/copilot.ts +124 -124
- package/src/rules/adapters/cursor.ts +114 -114
- package/src/rules/adapters/kiro.ts +126 -126
- package/src/rules/adapters/trae.ts +56 -56
- package/src/rules/adapters/windsurf.ts +83 -83
- package/src/rules/syncer.ts +235 -235
- package/src/sdk.ts +327 -327
- package/src/search/intent-detector.ts +289 -289
- package/src/search/query-expansion.ts +52 -52
- package/src/server/formation-timeout.ts +27 -27
- package/src/skills/mini-skills.ts +386 -386
- package/src/store/chat-store.ts +119 -119
- package/src/store/file-lock.ts +100 -100
- package/src/store/graph-store.ts +249 -249
- package/src/store/mini-skill-store.ts +349 -349
- package/src/store/obs-store.ts +255 -255
- package/src/store/persistence-json.ts +212 -212
- package/src/store/persistence.ts +291 -291
- package/src/store/project-affinity.ts +195 -195
- package/src/store/session-store.ts +259 -259
- package/src/store/sqlite-store.ts +339 -339
- package/src/team/event-bus.ts +76 -76
- package/src/team/file-locks.ts +173 -173
- package/src/team/handoff.ts +167 -167
- package/src/team/messages.ts +203 -203
- package/src/team/poll.ts +132 -132
- package/src/team/tasks.ts +211 -211
- package/src/wiki/generator.ts +237 -237
- package/src/wiki/knowledge-graph.ts +334 -334
- package/src/wiki/types.ts +85 -85
- package/src/workspace/mcp-adapters/codex.ts +191 -191
- package/src/workspace/mcp-adapters/copilot.ts +105 -105
- package/src/workspace/mcp-adapters/cursor.ts +53 -53
- package/src/workspace/mcp-adapters/kiro.ts +64 -64
- package/src/workspace/mcp-adapters/opencode.ts +123 -123
- package/src/workspace/mcp-adapters/trae.ts +134 -134
- package/src/workspace/mcp-adapters/windsurf.ts +91 -91
- package/src/workspace/sanitizer.ts +60 -60
- 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
|
+
}
|