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,122 +1,122 @@
1
- /**
2
- * Unified Freshness Gate (Phase 3a)
3
- *
4
- * Replaces withFreshObservations() as the public API for all retrieval
5
- * surfaces. Checks both observation and mini-skill generation counters
6
- * to ensure the Orama index is fully up-to-date before any read.
7
- *
8
- * withFreshObservations() remains in observations.ts as @internal —
9
- * only called by this module and legacy test code.
10
- */
11
-
12
- import { ensureFreshObservations } from './observations.js';
13
- import { getMiniSkillStore, isMiniSkillStoreInitialized } from '../store/mini-skill-store.js';
14
- import { miniSkillToDocument } from '../skills/mini-skills.js';
15
- import { insert, remove } from '@orama/orama';
16
-
17
- // ── Mini-skill index state ──────────────────────────────────────
18
-
19
- let lastMiniSkillGeneration = -1;
20
-
21
- /**
22
- * Check if mini-skills have changed since our last index sync.
23
- * If stale, reindex all mini-skills in Orama.
24
- *
25
- * Returns true if the index was refreshed.
26
- */
27
- export async function ensureFreshMiniSkills(): Promise<boolean> {
28
- // Skip silently when the store has not been initialized yet —
29
- // observation-only paths never call initMiniSkillStore().
30
- if (!isMiniSkillStoreInitialized()) return false;
31
-
32
- try {
33
- const store = getMiniSkillStore();
34
- const wasStale = await store.ensureFresh();
35
- const currentGen = store.getGeneration();
36
-
37
- if (wasStale || currentGen !== lastMiniSkillGeneration) {
38
- await reindexMiniSkills();
39
- lastMiniSkillGeneration = currentGen;
40
- return true;
41
- }
42
- } catch (err) {
43
- // Best-effort — don't crash the read path, but log for diagnostics
44
- console.warn(`[memorix] ensureFreshMiniSkills failed: ${err instanceof Error ? err.message : err}`);
45
- }
46
- return false;
47
- }
48
-
49
- // Deterministic tracking of indexed mini-skill doc IDs — avoids empty-term search
50
- const indexedSkillDocIds = new Set<string>();
51
-
52
- /**
53
- * Reindex all mini-skills into the Orama database.
54
- * Uses deterministic doc ID tracking instead of empty-term search to ensure
55
- * all stale documents are removed reliably.
56
- */
57
- export async function reindexMiniSkills(): Promise<number> {
58
- // Lazy import to avoid circular dependency
59
- const { getDb } = await import('../store/orama-store.js');
60
- const database = await getDb();
61
-
62
- // Remove previously tracked mini-skill documents by their known IDs
63
- for (const docId of indexedSkillDocIds) {
64
- try { await remove(database, docId); } catch { /* already removed or never existed */ }
65
- }
66
- indexedSkillDocIds.clear();
67
-
68
- // Load and index all current mini-skills
69
- const store = getMiniSkillStore();
70
- const skills = await store.loadAll();
71
- let indexed = 0;
72
-
73
- for (const skill of skills) {
74
- try {
75
- const doc = miniSkillToDocument(skill);
76
- await insert(database, doc);
77
- indexedSkillDocIds.add(doc.id);
78
- indexed++;
79
- } catch (err) {
80
- console.error(`[memorix] Failed to index mini-skill ${skill.id}: ${err instanceof Error ? err.message : err}`);
81
- }
82
- }
83
-
84
- return indexed;
85
- }
86
-
87
- // ── Unified gate ────────────────────────────────────────────────
88
-
89
- /**
90
- * Ensure both observations and mini-skills are fresh in the Orama index.
91
- * Returns true if any data source was refreshed.
92
- */
93
- export async function ensureFreshIndex(): Promise<boolean> {
94
- let anyStale = false;
95
- const obsStale = await ensureFreshObservations();
96
- if (obsStale) anyStale = true;
97
- const skillsStale = await ensureFreshMiniSkills();
98
- if (skillsStale) anyStale = true;
99
- return anyStale;
100
- }
101
-
102
- /**
103
- * Centralized freshness gate — wraps a read-facing function with
104
- * ensureFreshIndex() so callers cannot forget the freshness check.
105
- *
106
- * Usage:
107
- * return withFreshIndex(async () => { ... read from Orama ... });
108
- *
109
- * Phase 3a: replaces withFreshObservations() at all retrieval call sites.
110
- */
111
- export async function withFreshIndex<T>(fn: () => T | Promise<T>): Promise<T> {
112
- await ensureFreshIndex();
113
- return fn();
114
- }
115
-
116
- /**
117
- * Reset mini-skill freshness tracking. Used in tests.
118
- */
119
- export function resetMiniSkillFreshness(): void {
120
- lastMiniSkillGeneration = -1;
121
- indexedSkillDocIds.clear();
122
- }
1
+ /**
2
+ * Unified Freshness Gate (Phase 3a)
3
+ *
4
+ * Replaces withFreshObservations() as the public API for all retrieval
5
+ * surfaces. Checks both observation and mini-skill generation counters
6
+ * to ensure the Orama index is fully up-to-date before any read.
7
+ *
8
+ * withFreshObservations() remains in observations.ts as @internal —
9
+ * only called by this module and legacy test code.
10
+ */
11
+
12
+ import { ensureFreshObservations } from './observations.js';
13
+ import { getMiniSkillStore, isMiniSkillStoreInitialized } from '../store/mini-skill-store.js';
14
+ import { miniSkillToDocument } from '../skills/mini-skills.js';
15
+ import { insert, remove } from '@orama/orama';
16
+
17
+ // ── Mini-skill index state ──────────────────────────────────────
18
+
19
+ let lastMiniSkillGeneration = -1;
20
+
21
+ /**
22
+ * Check if mini-skills have changed since our last index sync.
23
+ * If stale, reindex all mini-skills in Orama.
24
+ *
25
+ * Returns true if the index was refreshed.
26
+ */
27
+ export async function ensureFreshMiniSkills(): Promise<boolean> {
28
+ // Skip silently when the store has not been initialized yet —
29
+ // observation-only paths never call initMiniSkillStore().
30
+ if (!isMiniSkillStoreInitialized()) return false;
31
+
32
+ try {
33
+ const store = getMiniSkillStore();
34
+ const wasStale = await store.ensureFresh();
35
+ const currentGen = store.getGeneration();
36
+
37
+ if (wasStale || currentGen !== lastMiniSkillGeneration) {
38
+ await reindexMiniSkills();
39
+ lastMiniSkillGeneration = currentGen;
40
+ return true;
41
+ }
42
+ } catch (err) {
43
+ // Best-effort — don't crash the read path, but log for diagnostics
44
+ console.warn(`[memorix] ensureFreshMiniSkills failed: ${err instanceof Error ? err.message : err}`);
45
+ }
46
+ return false;
47
+ }
48
+
49
+ // Deterministic tracking of indexed mini-skill doc IDs — avoids empty-term search
50
+ const indexedSkillDocIds = new Set<string>();
51
+
52
+ /**
53
+ * Reindex all mini-skills into the Orama database.
54
+ * Uses deterministic doc ID tracking instead of empty-term search to ensure
55
+ * all stale documents are removed reliably.
56
+ */
57
+ export async function reindexMiniSkills(): Promise<number> {
58
+ // Lazy import to avoid circular dependency
59
+ const { getDb } = await import('../store/orama-store.js');
60
+ const database = await getDb();
61
+
62
+ // Remove previously tracked mini-skill documents by their known IDs
63
+ for (const docId of indexedSkillDocIds) {
64
+ try { await remove(database, docId); } catch { /* already removed or never existed */ }
65
+ }
66
+ indexedSkillDocIds.clear();
67
+
68
+ // Load and index all current mini-skills
69
+ const store = getMiniSkillStore();
70
+ const skills = await store.loadAll();
71
+ let indexed = 0;
72
+
73
+ for (const skill of skills) {
74
+ try {
75
+ const doc = miniSkillToDocument(skill);
76
+ await insert(database, doc);
77
+ indexedSkillDocIds.add(doc.id);
78
+ indexed++;
79
+ } catch (err) {
80
+ console.error(`[memorix] Failed to index mini-skill ${skill.id}: ${err instanceof Error ? err.message : err}`);
81
+ }
82
+ }
83
+
84
+ return indexed;
85
+ }
86
+
87
+ // ── Unified gate ────────────────────────────────────────────────
88
+
89
+ /**
90
+ * Ensure both observations and mini-skills are fresh in the Orama index.
91
+ * Returns true if any data source was refreshed.
92
+ */
93
+ export async function ensureFreshIndex(): Promise<boolean> {
94
+ let anyStale = false;
95
+ const obsStale = await ensureFreshObservations();
96
+ if (obsStale) anyStale = true;
97
+ const skillsStale = await ensureFreshMiniSkills();
98
+ if (skillsStale) anyStale = true;
99
+ return anyStale;
100
+ }
101
+
102
+ /**
103
+ * Centralized freshness gate — wraps a read-facing function with
104
+ * ensureFreshIndex() so callers cannot forget the freshness check.
105
+ *
106
+ * Usage:
107
+ * return withFreshIndex(async () => { ... read from Orama ... });
108
+ *
109
+ * Phase 3a: replaces withFreshObservations() at all retrieval call sites.
110
+ */
111
+ export async function withFreshIndex<T>(fn: () => T | Promise<T>): Promise<T> {
112
+ await ensureFreshIndex();
113
+ return fn();
114
+ }
115
+
116
+ /**
117
+ * Reset mini-skill freshness tracking. Used in tests.
118
+ */
119
+ export function resetMiniSkillFreshness(): void {
120
+ lastMiniSkillGeneration = -1;
121
+ indexedSkillDocIds.clear();
122
+ }
@@ -1,6 +1,7 @@
1
1
  import { auditMemoryQuality, type MemoryQualityAuditReport } from './quality-audit.js';
2
2
  import type { Observation } from '../types.js';
3
3
  import { getRetentionZone } from './retention.js';
4
+ import { isEligibleForAutomaticDelivery } from './admission.js';
4
5
 
5
6
  export interface GraphContextPacketOptions {
6
7
  projectId: string;
@@ -225,6 +226,8 @@ function scoreForPacket(obs: Observation, queryTokens: string[], referenceTime:
225
226
  source: obs.source ?? 'agent',
226
227
  sourceDetail: obs.sourceDetail ?? '',
227
228
  valueCategory: obs.valueCategory ?? '',
229
+ admissionState: obs.admissionState ?? '',
230
+ admissionReason: obs.admissionReason ?? '',
228
231
  }, referenceTime) !== 'active') {
229
232
  score -= 2;
230
233
  }
@@ -348,8 +351,11 @@ export function buildGraphContextPacket(
348
351
  ...audit.issues.orphans.map((entry) => entry.id),
349
352
  ...audit.issues.retentionCandidates.map((entry) => entry.id),
350
353
  ]);
351
- const filteredObservations = observations.filter((obs) => obs.projectId === options.projectId && !riskIds.has(obs.id));
352
- const baseObservations = filteredObservations.length > 0 ? filteredObservations : observations;
354
+ const deliveryEligible = observations.filter(
355
+ (obs) => obs.projectId === options.projectId && isEligibleForAutomaticDelivery(obs),
356
+ );
357
+ const filteredObservations = deliveryEligible.filter((obs) => !riskIds.has(obs.id));
358
+ const baseObservations = filteredObservations.length > 0 ? filteredObservations : deliveryEligible;
353
359
  const memories = pickMemories(baseObservations, options.projectId, options.query, options.limit ?? 5, referenceTime);
354
360
  const entities = buildEntities(memories);
355
361
  const entityNames = new Set(entities.map((entity) => entity.name));
@@ -1,197 +1,197 @@
1
- /**
2
- * Knowledge Graph Manager
3
- *
4
- * Manages the Entity-Relation knowledge graph.
5
- * Source: MCP Official Memory Server v0.6.3 (complete rewrite with same API).
6
- *
7
- * Key differences from official implementation:
8
- * - Uses per-project JSONL files (official uses single file)
9
- * - Async initialization with persistence layer
10
- * - Project-scoped operations
11
- */
12
-
13
- import type { Entity, Relation, KnowledgeGraph } from '../types.js';
14
- import { initGraphStore, getGraphStore } from '../store/graph-store.js';
15
-
16
- export class KnowledgeGraphManager {
17
- private entities: Entity[] = [];
18
- private relations: Relation[] = [];
19
- private projectDir: string;
20
- private initialized = false;
21
- /** Index: lowercase entity name → Entity for O(1) lookups */
22
- private entityIndex = new Map<string, Entity>();
23
-
24
- constructor(projectDir: string) {
25
- this.projectDir = projectDir;
26
- }
27
-
28
- /** Rebuild the entity name index */
29
- private rebuildIndex(): void {
30
- this.entityIndex.clear();
31
- for (const e of this.entities) {
32
- this.entityIndex.set(e.name.toLowerCase(), e);
33
- }
34
- }
35
-
36
- /** Load graph from SQLite on first access */
37
- async init(): Promise<void> {
38
- if (this.initialized) return;
39
- await initGraphStore(this.projectDir);
40
- const store = getGraphStore();
41
- this.entities = store.loadEntities();
42
- this.relations = store.loadRelations();
43
- this.rebuildIndex();
44
- this.initialized = true;
45
- }
46
-
47
- /** Find entity by name (case-insensitive, O(1)) */
48
- findEntityByName(name: string): Entity | undefined {
49
- return this.entityIndex.get(name.toLowerCase());
50
- }
51
-
52
- /** Get all entity names as a Set (lowercase, for fast membership checks) */
53
- getEntityNameSet(): Set<string> {
54
- return new Set(this.entityIndex.keys());
55
- }
56
-
57
- /** Persist current state to SQLite */
58
- private async save(): Promise<void> {
59
- const store = getGraphStore();
60
- store.replaceAll(this.entities, this.relations);
61
- }
62
-
63
- /** Create new entities (skip duplicates by name) */
64
- async createEntities(entities: Entity[]): Promise<Entity[]> {
65
- await this.init();
66
- const newEntities = entities.filter(
67
- (e) => !this.entityIndex.has(e.name.toLowerCase()),
68
- );
69
- this.entities.push(...newEntities);
70
- if (newEntities.length > 0) this.rebuildIndex();
71
- await this.save();
72
- return newEntities;
73
- }
74
-
75
- /** Create new relations (skip duplicates) */
76
- async createRelations(relations: Relation[]): Promise<Relation[]> {
77
- await this.init();
78
- const newRelations = relations.filter(
79
- (r) =>
80
- !this.relations.some(
81
- (existing) =>
82
- existing.from === r.from &&
83
- existing.to === r.to &&
84
- existing.relationType === r.relationType,
85
- ),
86
- );
87
- this.relations.push(...newRelations);
88
- await this.save();
89
- return newRelations;
90
- }
91
-
92
- /** Add observations to existing entities */
93
- async addObservations(
94
- observations: { entityName: string; contents: string[] }[],
95
- ): Promise<{ entityName: string; addedObservations: string[] }[]> {
96
- await this.init();
97
- const results = observations.map((o) => {
98
- const entity = this.entities.find((e) => e.name === o.entityName);
99
- if (!entity) {
100
- throw new Error(`Entity with name ${o.entityName} not found`);
101
- }
102
- const newObs = o.contents.filter((c) => !entity.observations.includes(c));
103
- entity.observations.push(...newObs);
104
- return { entityName: o.entityName, addedObservations: newObs };
105
- });
106
- await this.save();
107
- return results;
108
- }
109
-
110
- /** Delete entities and their associated relations */
111
- async deleteEntities(entityNames: string[]): Promise<void> {
112
- await this.init();
113
- this.entities = this.entities.filter((e) => !entityNames.includes(e.name));
114
- this.relations = this.relations.filter(
115
- (r) => !entityNames.includes(r.from) && !entityNames.includes(r.to),
116
- );
117
- this.rebuildIndex();
118
- await this.save();
119
- }
120
-
121
- /** Delete specific observations from entities */
122
- async deleteObservations(
123
- deletions: { entityName: string; observations: string[] }[],
124
- ): Promise<void> {
125
- await this.init();
126
- for (const d of deletions) {
127
- const entity = this.entities.find((e) => e.name === d.entityName);
128
- if (entity) {
129
- entity.observations = entity.observations.filter(
130
- (o) => !d.observations.includes(o),
131
- );
132
- }
133
- }
134
- await this.save();
135
- }
136
-
137
- /** Delete specific relations */
138
- async deleteRelations(relations: Relation[]): Promise<void> {
139
- await this.init();
140
- this.relations = this.relations.filter(
141
- (r) =>
142
- !relations.some(
143
- (del) =>
144
- r.from === del.from &&
145
- r.to === del.to &&
146
- r.relationType === del.relationType,
147
- ),
148
- );
149
- await this.save();
150
- }
151
-
152
- /** Get all entity names (for Formation Pipeline entity resolution) */
153
- getEntityNames(): string[] {
154
- return this.entities.map(e => e.name);
155
- }
156
-
157
- /** Read the entire graph */
158
- async readGraph(): Promise<KnowledgeGraph> {
159
- await this.init();
160
- return { entities: this.entities, relations: this.relations };
161
- }
162
-
163
- /** Search nodes by query string (upgraded from official's basic includes) */
164
- async searchNodes(query: string): Promise<KnowledgeGraph> {
165
- await this.init();
166
- const lowerQuery = query.toLowerCase();
167
-
168
- const filteredEntities = this.entities.filter(
169
- (e) =>
170
- e.name.toLowerCase().includes(lowerQuery) ||
171
- e.entityType.toLowerCase().includes(lowerQuery) ||
172
- e.observations.some((o) => o.toLowerCase().includes(lowerQuery)),
173
- );
174
-
175
- const filteredNames = new Set(filteredEntities.map((e) => e.name));
176
-
177
- const filteredRelations = this.relations.filter(
178
- (r) => filteredNames.has(r.from) && filteredNames.has(r.to),
179
- );
180
-
181
- return { entities: filteredEntities, relations: filteredRelations };
182
- }
183
-
184
- /** Open specific nodes by name */
185
- async openNodes(names: string[]): Promise<KnowledgeGraph> {
186
- await this.init();
187
-
188
- const filteredEntities = this.entities.filter((e) => names.includes(e.name));
189
- const filteredNames = new Set(filteredEntities.map((e) => e.name));
190
-
191
- const filteredRelations = this.relations.filter(
192
- (r) => filteredNames.has(r.from) && filteredNames.has(r.to),
193
- );
194
-
195
- return { entities: filteredEntities, relations: filteredRelations };
196
- }
197
- }
1
+ /**
2
+ * Knowledge Graph Manager
3
+ *
4
+ * Manages the Entity-Relation knowledge graph.
5
+ * Source: MCP Official Memory Server v0.6.3 (complete rewrite with same API).
6
+ *
7
+ * Key differences from official implementation:
8
+ * - Uses per-project JSONL files (official uses single file)
9
+ * - Async initialization with persistence layer
10
+ * - Project-scoped operations
11
+ */
12
+
13
+ import type { Entity, Relation, KnowledgeGraph } from '../types.js';
14
+ import { initGraphStore, getGraphStore } from '../store/graph-store.js';
15
+
16
+ export class KnowledgeGraphManager {
17
+ private entities: Entity[] = [];
18
+ private relations: Relation[] = [];
19
+ private projectDir: string;
20
+ private initialized = false;
21
+ /** Index: lowercase entity name → Entity for O(1) lookups */
22
+ private entityIndex = new Map<string, Entity>();
23
+
24
+ constructor(projectDir: string) {
25
+ this.projectDir = projectDir;
26
+ }
27
+
28
+ /** Rebuild the entity name index */
29
+ private rebuildIndex(): void {
30
+ this.entityIndex.clear();
31
+ for (const e of this.entities) {
32
+ this.entityIndex.set(e.name.toLowerCase(), e);
33
+ }
34
+ }
35
+
36
+ /** Load graph from SQLite on first access */
37
+ async init(): Promise<void> {
38
+ if (this.initialized) return;
39
+ await initGraphStore(this.projectDir);
40
+ const store = getGraphStore();
41
+ this.entities = store.loadEntities();
42
+ this.relations = store.loadRelations();
43
+ this.rebuildIndex();
44
+ this.initialized = true;
45
+ }
46
+
47
+ /** Find entity by name (case-insensitive, O(1)) */
48
+ findEntityByName(name: string): Entity | undefined {
49
+ return this.entityIndex.get(name.toLowerCase());
50
+ }
51
+
52
+ /** Get all entity names as a Set (lowercase, for fast membership checks) */
53
+ getEntityNameSet(): Set<string> {
54
+ return new Set(this.entityIndex.keys());
55
+ }
56
+
57
+ /** Persist current state to SQLite */
58
+ private async save(): Promise<void> {
59
+ const store = getGraphStore();
60
+ store.replaceAll(this.entities, this.relations);
61
+ }
62
+
63
+ /** Create new entities (skip duplicates by name) */
64
+ async createEntities(entities: Entity[]): Promise<Entity[]> {
65
+ await this.init();
66
+ const newEntities = entities.filter(
67
+ (e) => !this.entityIndex.has(e.name.toLowerCase()),
68
+ );
69
+ this.entities.push(...newEntities);
70
+ if (newEntities.length > 0) this.rebuildIndex();
71
+ await this.save();
72
+ return newEntities;
73
+ }
74
+
75
+ /** Create new relations (skip duplicates) */
76
+ async createRelations(relations: Relation[]): Promise<Relation[]> {
77
+ await this.init();
78
+ const newRelations = relations.filter(
79
+ (r) =>
80
+ !this.relations.some(
81
+ (existing) =>
82
+ existing.from === r.from &&
83
+ existing.to === r.to &&
84
+ existing.relationType === r.relationType,
85
+ ),
86
+ );
87
+ this.relations.push(...newRelations);
88
+ await this.save();
89
+ return newRelations;
90
+ }
91
+
92
+ /** Add observations to existing entities */
93
+ async addObservations(
94
+ observations: { entityName: string; contents: string[] }[],
95
+ ): Promise<{ entityName: string; addedObservations: string[] }[]> {
96
+ await this.init();
97
+ const results = observations.map((o) => {
98
+ const entity = this.entities.find((e) => e.name === o.entityName);
99
+ if (!entity) {
100
+ throw new Error(`Entity with name ${o.entityName} not found`);
101
+ }
102
+ const newObs = o.contents.filter((c) => !entity.observations.includes(c));
103
+ entity.observations.push(...newObs);
104
+ return { entityName: o.entityName, addedObservations: newObs };
105
+ });
106
+ await this.save();
107
+ return results;
108
+ }
109
+
110
+ /** Delete entities and their associated relations */
111
+ async deleteEntities(entityNames: string[]): Promise<void> {
112
+ await this.init();
113
+ this.entities = this.entities.filter((e) => !entityNames.includes(e.name));
114
+ this.relations = this.relations.filter(
115
+ (r) => !entityNames.includes(r.from) && !entityNames.includes(r.to),
116
+ );
117
+ this.rebuildIndex();
118
+ await this.save();
119
+ }
120
+
121
+ /** Delete specific observations from entities */
122
+ async deleteObservations(
123
+ deletions: { entityName: string; observations: string[] }[],
124
+ ): Promise<void> {
125
+ await this.init();
126
+ for (const d of deletions) {
127
+ const entity = this.entities.find((e) => e.name === d.entityName);
128
+ if (entity) {
129
+ entity.observations = entity.observations.filter(
130
+ (o) => !d.observations.includes(o),
131
+ );
132
+ }
133
+ }
134
+ await this.save();
135
+ }
136
+
137
+ /** Delete specific relations */
138
+ async deleteRelations(relations: Relation[]): Promise<void> {
139
+ await this.init();
140
+ this.relations = this.relations.filter(
141
+ (r) =>
142
+ !relations.some(
143
+ (del) =>
144
+ r.from === del.from &&
145
+ r.to === del.to &&
146
+ r.relationType === del.relationType,
147
+ ),
148
+ );
149
+ await this.save();
150
+ }
151
+
152
+ /** Get all entity names (for Formation Pipeline entity resolution) */
153
+ getEntityNames(): string[] {
154
+ return this.entities.map(e => e.name);
155
+ }
156
+
157
+ /** Read the entire graph */
158
+ async readGraph(): Promise<KnowledgeGraph> {
159
+ await this.init();
160
+ return { entities: this.entities, relations: this.relations };
161
+ }
162
+
163
+ /** Search nodes by query string (upgraded from official's basic includes) */
164
+ async searchNodes(query: string): Promise<KnowledgeGraph> {
165
+ await this.init();
166
+ const lowerQuery = query.toLowerCase();
167
+
168
+ const filteredEntities = this.entities.filter(
169
+ (e) =>
170
+ e.name.toLowerCase().includes(lowerQuery) ||
171
+ e.entityType.toLowerCase().includes(lowerQuery) ||
172
+ e.observations.some((o) => o.toLowerCase().includes(lowerQuery)),
173
+ );
174
+
175
+ const filteredNames = new Set(filteredEntities.map((e) => e.name));
176
+
177
+ const filteredRelations = this.relations.filter(
178
+ (r) => filteredNames.has(r.from) && filteredNames.has(r.to),
179
+ );
180
+
181
+ return { entities: filteredEntities, relations: filteredRelations };
182
+ }
183
+
184
+ /** Open specific nodes by name */
185
+ async openNodes(names: string[]): Promise<KnowledgeGraph> {
186
+ await this.init();
187
+
188
+ const filteredEntities = this.entities.filter((e) => names.includes(e.name));
189
+ const filteredNames = new Set(filteredEntities.map((e) => e.name));
190
+
191
+ const filteredRelations = this.relations.filter(
192
+ (r) => filteredNames.has(r.from) && filteredNames.has(r.to),
193
+ );
194
+
195
+ return { entities: filteredEntities, relations: filteredRelations };
196
+ }
197
+ }