moflo 4.0.2 → 4.0.3

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 (89) hide show
  1. package/package.json +114 -110
  2. package/v3/@claude-flow/cli/dist/src/memory/memory-bridge.js +194 -81
  3. package/v3/@claude-flow/cli/dist/src/memory/memory-initializer.js +1892 -1841
  4. package/v3/@claude-flow/memory/README.md +587 -0
  5. package/v3/@claude-flow/memory/dist/agent-memory-scope.d.ts +131 -0
  6. package/v3/@claude-flow/memory/dist/agent-memory-scope.js +223 -0
  7. package/v3/@claude-flow/memory/dist/agent-memory-scope.test.d.ts +8 -0
  8. package/v3/@claude-flow/memory/dist/agent-memory-scope.test.js +463 -0
  9. package/v3/@claude-flow/memory/dist/agentdb-adapter.d.ts +165 -0
  10. package/v3/@claude-flow/memory/dist/agentdb-adapter.js +806 -0
  11. package/v3/@claude-flow/memory/dist/agentdb-backend.d.ts +214 -0
  12. package/v3/@claude-flow/memory/dist/agentdb-backend.js +844 -0
  13. package/v3/@claude-flow/memory/dist/agentdb-backend.test.d.ts +7 -0
  14. package/v3/@claude-flow/memory/dist/agentdb-backend.test.js +258 -0
  15. package/v3/@claude-flow/memory/dist/application/commands/delete-memory.command.d.ts +65 -0
  16. package/v3/@claude-flow/memory/dist/application/commands/delete-memory.command.js +129 -0
  17. package/v3/@claude-flow/memory/dist/application/commands/store-memory.command.d.ts +48 -0
  18. package/v3/@claude-flow/memory/dist/application/commands/store-memory.command.js +72 -0
  19. package/v3/@claude-flow/memory/dist/application/index.d.ts +12 -0
  20. package/v3/@claude-flow/memory/dist/application/index.js +15 -0
  21. package/v3/@claude-flow/memory/dist/application/queries/search-memory.query.d.ts +72 -0
  22. package/v3/@claude-flow/memory/dist/application/queries/search-memory.query.js +143 -0
  23. package/v3/@claude-flow/memory/dist/application/services/memory-application-service.d.ts +121 -0
  24. package/v3/@claude-flow/memory/dist/application/services/memory-application-service.js +190 -0
  25. package/v3/@claude-flow/memory/dist/auto-memory-bridge.d.ts +226 -0
  26. package/v3/@claude-flow/memory/dist/auto-memory-bridge.js +709 -0
  27. package/v3/@claude-flow/memory/dist/auto-memory-bridge.test.d.ts +8 -0
  28. package/v3/@claude-flow/memory/dist/auto-memory-bridge.test.js +754 -0
  29. package/v3/@claude-flow/memory/dist/benchmark.test.d.ts +2 -0
  30. package/v3/@claude-flow/memory/dist/benchmark.test.js +277 -0
  31. package/v3/@claude-flow/memory/dist/cache-manager.d.ts +134 -0
  32. package/v3/@claude-flow/memory/dist/cache-manager.js +407 -0
  33. package/v3/@claude-flow/memory/dist/controller-registry.d.ts +216 -0
  34. package/v3/@claude-flow/memory/dist/controller-registry.js +893 -0
  35. package/v3/@claude-flow/memory/dist/controller-registry.test.d.ts +14 -0
  36. package/v3/@claude-flow/memory/dist/controller-registry.test.js +636 -0
  37. package/v3/@claude-flow/memory/dist/database-provider.d.ts +87 -0
  38. package/v3/@claude-flow/memory/dist/database-provider.js +410 -0
  39. package/v3/@claude-flow/memory/dist/database-provider.test.d.ts +7 -0
  40. package/v3/@claude-flow/memory/dist/database-provider.test.js +285 -0
  41. package/v3/@claude-flow/memory/dist/domain/entities/memory-entry.d.ts +143 -0
  42. package/v3/@claude-flow/memory/dist/domain/entities/memory-entry.js +226 -0
  43. package/v3/@claude-flow/memory/dist/domain/index.d.ts +11 -0
  44. package/v3/@claude-flow/memory/dist/domain/index.js +12 -0
  45. package/v3/@claude-flow/memory/dist/domain/repositories/memory-repository.interface.d.ts +102 -0
  46. package/v3/@claude-flow/memory/dist/domain/repositories/memory-repository.interface.js +11 -0
  47. package/v3/@claude-flow/memory/dist/domain/services/memory-domain-service.d.ts +105 -0
  48. package/v3/@claude-flow/memory/dist/domain/services/memory-domain-service.js +297 -0
  49. package/v3/@claude-flow/memory/dist/hnsw-index.d.ts +111 -0
  50. package/v3/@claude-flow/memory/dist/hnsw-index.js +781 -0
  51. package/v3/@claude-flow/memory/dist/hnsw-lite.d.ts +23 -0
  52. package/v3/@claude-flow/memory/dist/hnsw-lite.js +168 -0
  53. package/v3/@claude-flow/memory/dist/hybrid-backend.d.ts +245 -0
  54. package/v3/@claude-flow/memory/dist/hybrid-backend.js +569 -0
  55. package/v3/@claude-flow/memory/dist/hybrid-backend.test.d.ts +8 -0
  56. package/v3/@claude-flow/memory/dist/hybrid-backend.test.js +320 -0
  57. package/v3/@claude-flow/memory/dist/index.d.ts +208 -0
  58. package/v3/@claude-flow/memory/dist/index.js +362 -0
  59. package/v3/@claude-flow/memory/dist/infrastructure/index.d.ts +17 -0
  60. package/v3/@claude-flow/memory/dist/infrastructure/index.js +16 -0
  61. package/v3/@claude-flow/memory/dist/infrastructure/repositories/hybrid-memory-repository.d.ts +66 -0
  62. package/v3/@claude-flow/memory/dist/infrastructure/repositories/hybrid-memory-repository.js +409 -0
  63. package/v3/@claude-flow/memory/dist/learning-bridge.d.ts +137 -0
  64. package/v3/@claude-flow/memory/dist/learning-bridge.js +335 -0
  65. package/v3/@claude-flow/memory/dist/learning-bridge.test.d.ts +8 -0
  66. package/v3/@claude-flow/memory/dist/learning-bridge.test.js +578 -0
  67. package/v3/@claude-flow/memory/dist/memory-graph.d.ts +100 -0
  68. package/v3/@claude-flow/memory/dist/memory-graph.js +333 -0
  69. package/v3/@claude-flow/memory/dist/memory-graph.test.d.ts +8 -0
  70. package/v3/@claude-flow/memory/dist/memory-graph.test.js +609 -0
  71. package/v3/@claude-flow/memory/dist/migration.d.ts +68 -0
  72. package/v3/@claude-flow/memory/dist/migration.js +513 -0
  73. package/v3/@claude-flow/memory/dist/persistent-sona.d.ts +144 -0
  74. package/v3/@claude-flow/memory/dist/persistent-sona.js +332 -0
  75. package/v3/@claude-flow/memory/dist/query-builder.d.ts +211 -0
  76. package/v3/@claude-flow/memory/dist/query-builder.js +438 -0
  77. package/v3/@claude-flow/memory/dist/rvf-backend.d.ts +51 -0
  78. package/v3/@claude-flow/memory/dist/rvf-backend.js +481 -0
  79. package/v3/@claude-flow/memory/dist/rvf-learning-store.d.ts +139 -0
  80. package/v3/@claude-flow/memory/dist/rvf-learning-store.js +295 -0
  81. package/v3/@claude-flow/memory/dist/rvf-migration.d.ts +45 -0
  82. package/v3/@claude-flow/memory/dist/rvf-migration.js +254 -0
  83. package/v3/@claude-flow/memory/dist/sqlite-backend.d.ts +121 -0
  84. package/v3/@claude-flow/memory/dist/sqlite-backend.js +564 -0
  85. package/v3/@claude-flow/memory/dist/sqljs-backend.d.ts +128 -0
  86. package/v3/@claude-flow/memory/dist/sqljs-backend.js +601 -0
  87. package/v3/@claude-flow/memory/dist/types.d.ts +484 -0
  88. package/v3/@claude-flow/memory/dist/types.js +58 -0
  89. package/v3/@claude-flow/memory/package.json +46 -0
@@ -0,0 +1,609 @@
1
+ /**
2
+ * Tests for MemoryGraph - Knowledge Graph Module
3
+ *
4
+ * TDD London School (mock-first) tests for graph construction,
5
+ * PageRank computation, community detection, and graph-aware ranking.
6
+ */
7
+ import { describe, it, expect, beforeEach, vi } from 'vitest';
8
+ import { MemoryGraph } from './memory-graph.js';
9
+ import { createDefaultEntry } from './types.js';
10
+ // ===== Test Helpers =====
11
+ function makeEntry(id, refs = [], meta) {
12
+ const entry = createDefaultEntry({
13
+ key: id,
14
+ content: `content-${id}`,
15
+ references: refs,
16
+ metadata: meta,
17
+ });
18
+ // Override the auto-generated id with a deterministic one for testing
19
+ return { ...entry, id };
20
+ }
21
+ function createMockBackend(entries) {
22
+ return {
23
+ initialize: vi.fn().mockResolvedValue(undefined),
24
+ shutdown: vi.fn().mockResolvedValue(undefined),
25
+ store: vi.fn().mockResolvedValue(undefined),
26
+ get: vi.fn().mockImplementation(async (id) => {
27
+ return entries.find((e) => e.id === id) || null;
28
+ }),
29
+ getByKey: vi.fn().mockResolvedValue(null),
30
+ update: vi.fn().mockImplementation(async (id, _update) => {
31
+ return entries.find((e) => e.id === id) || null;
32
+ }),
33
+ delete: vi.fn().mockResolvedValue(true),
34
+ query: vi.fn().mockResolvedValue(entries),
35
+ search: vi.fn().mockResolvedValue([]),
36
+ bulkInsert: vi.fn().mockResolvedValue(undefined),
37
+ bulkDelete: vi.fn().mockResolvedValue(0),
38
+ count: vi.fn().mockResolvedValue(entries.length),
39
+ listNamespaces: vi.fn().mockResolvedValue(['default']),
40
+ clearNamespace: vi.fn().mockResolvedValue(0),
41
+ getStats: vi.fn().mockResolvedValue({
42
+ totalEntries: entries.length,
43
+ entriesByNamespace: {},
44
+ entriesByType: {},
45
+ memoryUsage: 0,
46
+ avgQueryTime: 0,
47
+ avgSearchTime: 0,
48
+ }),
49
+ healthCheck: vi.fn().mockResolvedValue({
50
+ status: 'healthy',
51
+ components: {
52
+ storage: { status: 'healthy', latency: 0 },
53
+ index: { status: 'healthy', latency: 0 },
54
+ cache: { status: 'healthy', latency: 0 },
55
+ },
56
+ timestamp: Date.now(),
57
+ issues: [],
58
+ recommendations: [],
59
+ }),
60
+ };
61
+ }
62
+ // ===== Tests =====
63
+ describe('MemoryGraph', () => {
64
+ let graph;
65
+ beforeEach(() => {
66
+ graph = new MemoryGraph();
67
+ });
68
+ // ===== Constructor =====
69
+ describe('constructor', () => {
70
+ it('should create with default configuration', () => {
71
+ const g = new MemoryGraph();
72
+ const stats = g.getStats();
73
+ expect(stats.nodeCount).toBe(0);
74
+ expect(stats.edgeCount).toBe(0);
75
+ });
76
+ it('should accept custom configuration', () => {
77
+ const config = {
78
+ pageRankDamping: 0.9,
79
+ maxNodes: 100,
80
+ similarityThreshold: 0.5,
81
+ };
82
+ const g = new MemoryGraph(config);
83
+ const stats = g.getStats();
84
+ expect(stats.nodeCount).toBe(0);
85
+ });
86
+ });
87
+ // ===== addNode / removeNode =====
88
+ describe('addNode', () => {
89
+ it('should add a node from a MemoryEntry', () => {
90
+ const entry = makeEntry('node-1');
91
+ graph.addNode(entry);
92
+ expect(graph.getStats().nodeCount).toBe(1);
93
+ });
94
+ it('should extract category from metadata', () => {
95
+ const entry = makeEntry('node-1', [], { category: 'security' });
96
+ graph.addNode(entry);
97
+ // Verify through getTopNodes after computing pagerank
98
+ graph.computePageRank();
99
+ const top = graph.getTopNodes(1);
100
+ expect(top.length).toBe(1);
101
+ expect(top[0].id).toBe('node-1');
102
+ });
103
+ it('should ignore when at maxNodes capacity', () => {
104
+ const g = new MemoryGraph({ maxNodes: 2 });
105
+ g.addNode(makeEntry('a'));
106
+ g.addNode(makeEntry('b'));
107
+ g.addNode(makeEntry('c'));
108
+ expect(g.getStats().nodeCount).toBe(2);
109
+ });
110
+ it('should allow re-adding an existing node without counting toward capacity', () => {
111
+ const g = new MemoryGraph({ maxNodes: 2 });
112
+ g.addNode(makeEntry('a'));
113
+ g.addNode(makeEntry('b'));
114
+ // Re-add existing node should succeed
115
+ g.addNode(makeEntry('a'));
116
+ expect(g.getStats().nodeCount).toBe(2);
117
+ });
118
+ it('should mark graph as dirty', () => {
119
+ graph.addNode(makeEntry('node-1'));
120
+ expect(graph.getStats().pageRankComputed).toBe(false);
121
+ });
122
+ });
123
+ describe('removeNode', () => {
124
+ it('should remove a node and clean up edges', () => {
125
+ const a = makeEntry('a');
126
+ const b = makeEntry('b');
127
+ graph.addNode(a);
128
+ graph.addNode(b);
129
+ graph.addEdge('a', 'b', 'reference');
130
+ expect(graph.getStats().edgeCount).toBe(1);
131
+ graph.removeNode('a');
132
+ expect(graph.getStats().nodeCount).toBe(1);
133
+ expect(graph.getStats().edgeCount).toBe(0);
134
+ });
135
+ it('should remove incoming edges to the deleted node', () => {
136
+ graph.addNode(makeEntry('a'));
137
+ graph.addNode(makeEntry('b'));
138
+ graph.addNode(makeEntry('c'));
139
+ graph.addEdge('a', 'b', 'reference');
140
+ graph.addEdge('c', 'b', 'reference');
141
+ expect(graph.getStats().edgeCount).toBe(2);
142
+ graph.removeNode('b');
143
+ expect(graph.getStats().edgeCount).toBe(0);
144
+ });
145
+ it('should handle removing a non-existent node gracefully', () => {
146
+ expect(() => graph.removeNode('does-not-exist')).not.toThrow();
147
+ });
148
+ it('should clean up pageRank and community entries', () => {
149
+ graph.addNode(makeEntry('a'));
150
+ graph.computePageRank();
151
+ graph.detectCommunities();
152
+ graph.removeNode('a');
153
+ const top = graph.getTopNodes(10);
154
+ expect(top.length).toBe(0);
155
+ });
156
+ });
157
+ // ===== addEdge =====
158
+ describe('addEdge', () => {
159
+ beforeEach(() => {
160
+ graph.addNode(makeEntry('a'));
161
+ graph.addNode(makeEntry('b'));
162
+ graph.addNode(makeEntry('c'));
163
+ });
164
+ it('should add an edge between existing nodes', () => {
165
+ graph.addEdge('a', 'b', 'reference');
166
+ expect(graph.getStats().edgeCount).toBe(1);
167
+ });
168
+ it('should update weight to max when edge already exists', () => {
169
+ graph.addEdge('a', 'b', 'reference', 0.5);
170
+ graph.addEdge('a', 'b', 'reference', 0.8);
171
+ expect(graph.getStats().edgeCount).toBe(1);
172
+ // The weight should now be 0.8 (max of 0.5 and 0.8)
173
+ });
174
+ it('should not downgrade weight when adding with lower value', () => {
175
+ graph.addEdge('a', 'b', 'reference', 0.9);
176
+ graph.addEdge('a', 'b', 'reference', 0.3);
177
+ // Edge count should remain 1 (updated, not duplicated)
178
+ expect(graph.getStats().edgeCount).toBe(1);
179
+ });
180
+ it('should skip when source node is missing', () => {
181
+ graph.addEdge('missing', 'b', 'reference');
182
+ expect(graph.getStats().edgeCount).toBe(0);
183
+ });
184
+ it('should skip when target node is missing', () => {
185
+ graph.addEdge('a', 'missing', 'reference');
186
+ expect(graph.getStats().edgeCount).toBe(0);
187
+ });
188
+ it('should create reverse edge entry', () => {
189
+ graph.addEdge('a', 'b', 'reference');
190
+ // Verify via getNeighbors traversal (reverse edges are used internally)
191
+ // If we remove 'a', 'b' should lose its incoming edge
192
+ graph.removeNode('a');
193
+ expect(graph.getStats().edgeCount).toBe(0);
194
+ });
195
+ it('should support multiple edge types', () => {
196
+ graph.addEdge('a', 'b', 'reference');
197
+ graph.addEdge('a', 'c', 'similar', 0.9);
198
+ expect(graph.getStats().edgeCount).toBe(2);
199
+ });
200
+ });
201
+ // ===== buildFromBackend =====
202
+ describe('buildFromBackend', () => {
203
+ it('should build graph from backend entries with references', async () => {
204
+ const entries = [
205
+ makeEntry('a', ['b', 'c']),
206
+ makeEntry('b', ['c']),
207
+ makeEntry('c', []),
208
+ ];
209
+ const backend = createMockBackend(entries);
210
+ await graph.buildFromBackend(backend);
211
+ expect(graph.getStats().nodeCount).toBe(3);
212
+ // a->b, a->c, b->c = 3 edges
213
+ expect(graph.getStats().edgeCount).toBe(3);
214
+ });
215
+ it('should handle an empty backend', async () => {
216
+ const backend = createMockBackend([]);
217
+ await graph.buildFromBackend(backend);
218
+ expect(graph.getStats().nodeCount).toBe(0);
219
+ expect(graph.getStats().edgeCount).toBe(0);
220
+ });
221
+ it('should respect maxNodes limit', async () => {
222
+ const g = new MemoryGraph({ maxNodes: 2 });
223
+ const entries = [makeEntry('a'), makeEntry('b'), makeEntry('c')];
224
+ const backend = createMockBackend(entries);
225
+ await g.buildFromBackend(backend);
226
+ expect(g.getStats().nodeCount).toBe(2);
227
+ });
228
+ it('should skip reference edges to nodes not in graph', async () => {
229
+ // 'a' references 'z' which is not in the entries
230
+ const entries = [makeEntry('a', ['z']), makeEntry('b', [])];
231
+ const backend = createMockBackend(entries);
232
+ await graph.buildFromBackend(backend);
233
+ expect(graph.getStats().nodeCount).toBe(2);
234
+ // a->z should be skipped because 'z' is not a node
235
+ expect(graph.getStats().edgeCount).toBe(0);
236
+ });
237
+ it('should emit graph:built event', async () => {
238
+ const handler = vi.fn();
239
+ graph.on('graph:built', handler);
240
+ const backend = createMockBackend([makeEntry('a')]);
241
+ await graph.buildFromBackend(backend);
242
+ expect(handler).toHaveBeenCalledWith({ nodeCount: 1 });
243
+ });
244
+ });
245
+ // ===== computePageRank =====
246
+ describe('computePageRank', () => {
247
+ it('should return empty map for empty graph', () => {
248
+ const ranks = graph.computePageRank();
249
+ expect(ranks.size).toBe(0);
250
+ });
251
+ it('should compute rank of 1.0 for a single node', () => {
252
+ graph.addNode(makeEntry('a'));
253
+ const ranks = graph.computePageRank();
254
+ expect(ranks.size).toBe(1);
255
+ expect(ranks.get('a')).toBeCloseTo(1.0, 5);
256
+ });
257
+ it('should converge for two connected nodes', () => {
258
+ graph.addNode(makeEntry('a'));
259
+ graph.addNode(makeEntry('b'));
260
+ graph.addEdge('a', 'b', 'reference');
261
+ const ranks = graph.computePageRank();
262
+ expect(ranks.size).toBe(2);
263
+ // Both should have positive values that sum close to 1
264
+ const total = (ranks.get('a') || 0) + (ranks.get('b') || 0);
265
+ expect(total).toBeCloseTo(1.0, 2);
266
+ });
267
+ it('should give higher rank to node with more incoming edges', () => {
268
+ graph.addNode(makeEntry('a'));
269
+ graph.addNode(makeEntry('b'));
270
+ graph.addNode(makeEntry('c'));
271
+ graph.addEdge('a', 'c', 'reference');
272
+ graph.addEdge('b', 'c', 'reference');
273
+ const ranks = graph.computePageRank();
274
+ const rankC = ranks.get('c') || 0;
275
+ const rankA = ranks.get('a') || 0;
276
+ const rankB = ranks.get('b') || 0;
277
+ expect(rankC).toBeGreaterThan(rankA);
278
+ expect(rankC).toBeGreaterThan(rankB);
279
+ });
280
+ it('should respect damping factor', () => {
281
+ const g = new MemoryGraph({ pageRankDamping: 0.5 });
282
+ g.addNode(makeEntry('a'));
283
+ g.addNode(makeEntry('b'));
284
+ g.addEdge('a', 'b', 'reference');
285
+ const ranks = g.computePageRank();
286
+ // With damping 0.5, the influence of links is reduced
287
+ expect(ranks.get('a')).toBeDefined();
288
+ expect(ranks.get('b')).toBeDefined();
289
+ });
290
+ it('should converge within maxIterations', () => {
291
+ const handler = vi.fn();
292
+ graph.on('pagerank:computed', handler);
293
+ graph.addNode(makeEntry('a'));
294
+ graph.addNode(makeEntry('b'));
295
+ graph.addEdge('a', 'b', 'reference');
296
+ graph.computePageRank();
297
+ expect(handler).toHaveBeenCalledTimes(1);
298
+ const { iterations } = handler.mock.calls[0][0];
299
+ expect(iterations).toBeLessThanOrEqual(50);
300
+ expect(iterations).toBeGreaterThan(0);
301
+ });
302
+ it('should handle disconnected components', () => {
303
+ graph.addNode(makeEntry('a'));
304
+ graph.addNode(makeEntry('b'));
305
+ graph.addNode(makeEntry('c'));
306
+ graph.addNode(makeEntry('d'));
307
+ graph.addEdge('a', 'b', 'reference');
308
+ graph.addEdge('c', 'd', 'reference');
309
+ const ranks = graph.computePageRank();
310
+ expect(ranks.size).toBe(4);
311
+ // Each component should have similar rank distribution
312
+ const total = [...ranks.values()].reduce((s, v) => s + v, 0);
313
+ expect(total).toBeCloseTo(1.0, 2);
314
+ });
315
+ it('should mark graph as not dirty after computation', () => {
316
+ graph.addNode(makeEntry('a'));
317
+ expect(graph.getStats().pageRankComputed).toBe(false);
318
+ graph.computePageRank();
319
+ expect(graph.getStats().pageRankComputed).toBe(true);
320
+ });
321
+ it('should emit pagerank:computed event', () => {
322
+ const handler = vi.fn();
323
+ graph.on('pagerank:computed', handler);
324
+ graph.computePageRank();
325
+ expect(handler).toHaveBeenCalledTimes(1);
326
+ });
327
+ });
328
+ // ===== detectCommunities =====
329
+ describe('detectCommunities', () => {
330
+ it('should assign isolated nodes to individual communities', () => {
331
+ graph.addNode(makeEntry('a'));
332
+ graph.addNode(makeEntry('b'));
333
+ graph.addNode(makeEntry('c'));
334
+ const communities = graph.detectCommunities();
335
+ const unique = new Set(communities.values());
336
+ expect(unique.size).toBe(3);
337
+ });
338
+ it('should group connected nodes into the same community', () => {
339
+ graph.addNode(makeEntry('a'));
340
+ graph.addNode(makeEntry('b'));
341
+ graph.addEdge('a', 'b', 'reference');
342
+ graph.addEdge('b', 'a', 'reference');
343
+ const communities = graph.detectCommunities();
344
+ // With bidirectional edges, they should converge
345
+ expect(communities.get('a')).toBe(communities.get('b'));
346
+ });
347
+ it('should detect two disconnected clusters', () => {
348
+ // Cluster 1: a <-> b
349
+ graph.addNode(makeEntry('a'));
350
+ graph.addNode(makeEntry('b'));
351
+ graph.addEdge('a', 'b', 'reference');
352
+ graph.addEdge('b', 'a', 'reference');
353
+ // Cluster 2: c <-> d
354
+ graph.addNode(makeEntry('c'));
355
+ graph.addNode(makeEntry('d'));
356
+ graph.addEdge('c', 'd', 'reference');
357
+ graph.addEdge('d', 'c', 'reference');
358
+ const communities = graph.detectCommunities();
359
+ expect(communities.get('a')).toBe(communities.get('b'));
360
+ expect(communities.get('c')).toBe(communities.get('d'));
361
+ expect(communities.get('a')).not.toBe(communities.get('c'));
362
+ });
363
+ it('should emit communities:detected event', () => {
364
+ const handler = vi.fn();
365
+ graph.on('communities:detected', handler);
366
+ graph.addNode(makeEntry('a'));
367
+ graph.detectCommunities();
368
+ expect(handler).toHaveBeenCalledTimes(1);
369
+ expect(handler.mock.calls[0][0].communityCount).toBe(1);
370
+ });
371
+ it('should handle empty graph', () => {
372
+ const communities = graph.detectCommunities();
373
+ expect(communities.size).toBe(0);
374
+ });
375
+ });
376
+ // ===== rankWithGraph =====
377
+ describe('rankWithGraph', () => {
378
+ it('should blend vector score and pagerank', () => {
379
+ graph.addNode(makeEntry('a'));
380
+ graph.addNode(makeEntry('b'));
381
+ graph.addNode(makeEntry('c'));
382
+ graph.addEdge('a', 'c', 'reference');
383
+ graph.addEdge('b', 'c', 'reference');
384
+ const searchResults = [
385
+ { entry: makeEntry('a'), score: 0.9, distance: 0.1 },
386
+ { entry: makeEntry('c'), score: 0.7, distance: 0.3 },
387
+ ];
388
+ const ranked = graph.rankWithGraph(searchResults);
389
+ expect(ranked.length).toBe(2);
390
+ // Each result should have all fields
391
+ expect(ranked[0].score).toBeDefined();
392
+ expect(ranked[0].pageRank).toBeDefined();
393
+ expect(ranked[0].combinedScore).toBeDefined();
394
+ });
395
+ it('should respect alpha parameter', () => {
396
+ graph.addNode(makeEntry('a'));
397
+ graph.addNode(makeEntry('b'));
398
+ graph.addEdge('a', 'b', 'reference');
399
+ const results = [
400
+ { entry: makeEntry('a'), score: 0.5, distance: 0.5 },
401
+ { entry: makeEntry('b'), score: 0.5, distance: 0.5 },
402
+ ];
403
+ // With alpha=1.0, only vector score matters
404
+ const ranked1 = graph.rankWithGraph(results, 1.0);
405
+ expect(ranked1[0].combinedScore).toBeCloseTo(0.5, 3);
406
+ // With alpha=0.0, only pagerank matters
407
+ const ranked0 = graph.rankWithGraph(results, 0.0);
408
+ // Node b has an incoming edge so should rank higher
409
+ expect(ranked0[0].entry.id).toBe('b');
410
+ });
411
+ it('should handle entries not in graph', () => {
412
+ graph.addNode(makeEntry('a'));
413
+ graph.computePageRank();
414
+ const results = [
415
+ { entry: makeEntry('a'), score: 0.8, distance: 0.2 },
416
+ { entry: makeEntry('unknown'), score: 0.9, distance: 0.1 },
417
+ ];
418
+ const ranked = graph.rankWithGraph(results);
419
+ expect(ranked.length).toBe(2);
420
+ // Unknown entry should have pageRank of 0
421
+ const unknownResult = ranked.find((r) => r.entry.id === 'unknown');
422
+ expect(unknownResult?.pageRank).toBe(0);
423
+ });
424
+ it('should sort results by combinedScore descending', () => {
425
+ graph.addNode(makeEntry('a'));
426
+ graph.addNode(makeEntry('b'));
427
+ graph.addNode(makeEntry('c'));
428
+ graph.addEdge('a', 'c', 'reference');
429
+ graph.addEdge('b', 'c', 'reference');
430
+ const results = [
431
+ { entry: makeEntry('a'), score: 0.5, distance: 0.5 },
432
+ { entry: makeEntry('b'), score: 0.5, distance: 0.5 },
433
+ { entry: makeEntry('c'), score: 0.5, distance: 0.5 },
434
+ ];
435
+ const ranked = graph.rankWithGraph(results, 0.5);
436
+ for (let i = 0; i < ranked.length - 1; i++) {
437
+ expect(ranked[i].combinedScore).toBeGreaterThanOrEqual(ranked[i + 1].combinedScore);
438
+ }
439
+ });
440
+ it('should include community info when communities are detected', () => {
441
+ graph.addNode(makeEntry('a'));
442
+ graph.addNode(makeEntry('b'));
443
+ graph.addEdge('a', 'b', 'reference');
444
+ graph.detectCommunities();
445
+ const results = [
446
+ { entry: makeEntry('a'), score: 0.8, distance: 0.2 },
447
+ ];
448
+ const ranked = graph.rankWithGraph(results);
449
+ expect(ranked[0].community).toBeDefined();
450
+ });
451
+ });
452
+ // ===== getTopNodes =====
453
+ describe('getTopNodes', () => {
454
+ it('should return top N nodes by pageRank', () => {
455
+ graph.addNode(makeEntry('a'));
456
+ graph.addNode(makeEntry('b'));
457
+ graph.addNode(makeEntry('c'));
458
+ graph.addEdge('a', 'c', 'reference');
459
+ graph.addEdge('b', 'c', 'reference');
460
+ const top = graph.getTopNodes(1);
461
+ expect(top.length).toBe(1);
462
+ expect(top[0].id).toBe('c');
463
+ });
464
+ it('should handle requesting more nodes than available', () => {
465
+ graph.addNode(makeEntry('a'));
466
+ graph.addNode(makeEntry('b'));
467
+ const top = graph.getTopNodes(10);
468
+ expect(top.length).toBe(2);
469
+ });
470
+ it('should return empty array for empty graph', () => {
471
+ const top = graph.getTopNodes(5);
472
+ expect(top.length).toBe(0);
473
+ });
474
+ it('should include community label', () => {
475
+ graph.addNode(makeEntry('a'));
476
+ graph.detectCommunities();
477
+ const top = graph.getTopNodes(1);
478
+ expect(top[0].community).toBeDefined();
479
+ });
480
+ });
481
+ // ===== getNeighbors =====
482
+ describe('getNeighbors', () => {
483
+ beforeEach(() => {
484
+ graph.addNode(makeEntry('a'));
485
+ graph.addNode(makeEntry('b'));
486
+ graph.addNode(makeEntry('c'));
487
+ graph.addNode(makeEntry('d'));
488
+ graph.addEdge('a', 'b', 'reference');
489
+ graph.addEdge('b', 'c', 'reference');
490
+ graph.addEdge('c', 'd', 'reference');
491
+ });
492
+ it('should return direct neighbors at depth 1', () => {
493
+ const neighbors = graph.getNeighbors('a', 1);
494
+ expect(neighbors.size).toBe(1);
495
+ expect(neighbors.has('b')).toBe(true);
496
+ });
497
+ it('should return extended neighbors at depth 2', () => {
498
+ const neighbors = graph.getNeighbors('a', 2);
499
+ expect(neighbors.size).toBe(2);
500
+ expect(neighbors.has('b')).toBe(true);
501
+ expect(neighbors.has('c')).toBe(true);
502
+ });
503
+ it('should return all reachable nodes at depth 3', () => {
504
+ const neighbors = graph.getNeighbors('a', 3);
505
+ expect(neighbors.size).toBe(3);
506
+ expect(neighbors.has('b')).toBe(true);
507
+ expect(neighbors.has('c')).toBe(true);
508
+ expect(neighbors.has('d')).toBe(true);
509
+ });
510
+ it('should handle node with no neighbors', () => {
511
+ const neighbors = graph.getNeighbors('d', 1);
512
+ expect(neighbors.size).toBe(0);
513
+ });
514
+ it('should not include the start node in results', () => {
515
+ const neighbors = graph.getNeighbors('a', 10);
516
+ expect(neighbors.has('a')).toBe(false);
517
+ });
518
+ it('should default to depth 1', () => {
519
+ const neighbors = graph.getNeighbors('a');
520
+ expect(neighbors.size).toBe(1);
521
+ expect(neighbors.has('b')).toBe(true);
522
+ });
523
+ });
524
+ // ===== addSimilarityEdges =====
525
+ describe('addSimilarityEdges', () => {
526
+ it('should add edges for similar entries above threshold', async () => {
527
+ const embedding = new Float32Array([1, 0, 0]);
528
+ const entryA = { ...makeEntry('a'), embedding };
529
+ const entryB = makeEntry('b');
530
+ graph.addNode(entryA);
531
+ graph.addNode(entryB);
532
+ const searchResults = [
533
+ { entry: entryB, score: 0.9, distance: 0.1 },
534
+ ];
535
+ const backend = createMockBackend([entryA, entryB]);
536
+ backend.search.mockResolvedValue(searchResults);
537
+ const added = await graph.addSimilarityEdges(backend, 'a');
538
+ expect(added).toBe(1);
539
+ expect(graph.getStats().edgeCount).toBe(1);
540
+ });
541
+ it('should return 0 if entry has no embedding', async () => {
542
+ const entry = makeEntry('a');
543
+ graph.addNode(entry);
544
+ const backend = createMockBackend([entry]);
545
+ const added = await graph.addSimilarityEdges(backend, 'a');
546
+ expect(added).toBe(0);
547
+ });
548
+ it('should return 0 if entry does not exist', async () => {
549
+ const backend = createMockBackend([]);
550
+ const added = await graph.addSimilarityEdges(backend, 'missing');
551
+ expect(added).toBe(0);
552
+ });
553
+ it('should skip self-references in search results', async () => {
554
+ const embedding = new Float32Array([1, 0, 0]);
555
+ const entryA = { ...makeEntry('a'), embedding };
556
+ graph.addNode(entryA);
557
+ const searchResults = [
558
+ { entry: entryA, score: 1.0, distance: 0.0 },
559
+ ];
560
+ const backend = createMockBackend([entryA]);
561
+ backend.search.mockResolvedValue(searchResults);
562
+ const added = await graph.addSimilarityEdges(backend, 'a');
563
+ expect(added).toBe(0);
564
+ });
565
+ });
566
+ // ===== getStats =====
567
+ describe('getStats', () => {
568
+ it('should return correct initial stats', () => {
569
+ const stats = graph.getStats();
570
+ expect(stats.nodeCount).toBe(0);
571
+ expect(stats.edgeCount).toBe(0);
572
+ expect(stats.avgDegree).toBe(0);
573
+ expect(stats.communityCount).toBe(0);
574
+ expect(stats.pageRankComputed).toBe(false);
575
+ expect(stats.maxPageRank).toBe(0);
576
+ expect(stats.minPageRank).toBe(0);
577
+ });
578
+ it('should reflect graph state after adding nodes and edges', () => {
579
+ graph.addNode(makeEntry('a'));
580
+ graph.addNode(makeEntry('b'));
581
+ graph.addNode(makeEntry('c'));
582
+ graph.addEdge('a', 'b', 'reference');
583
+ graph.addEdge('a', 'c', 'reference');
584
+ const stats = graph.getStats();
585
+ expect(stats.nodeCount).toBe(3);
586
+ expect(stats.edgeCount).toBe(2);
587
+ expect(stats.avgDegree).toBeCloseTo(2 / 3, 5);
588
+ });
589
+ it('should report pageRank min/max after computation', () => {
590
+ graph.addNode(makeEntry('a'));
591
+ graph.addNode(makeEntry('b'));
592
+ graph.addEdge('a', 'b', 'reference');
593
+ graph.computePageRank();
594
+ const stats = graph.getStats();
595
+ expect(stats.pageRankComputed).toBe(true);
596
+ expect(stats.maxPageRank).toBeGreaterThan(0);
597
+ expect(stats.minPageRank).toBeGreaterThan(0);
598
+ expect(stats.maxPageRank).toBeGreaterThanOrEqual(stats.minPageRank);
599
+ });
600
+ it('should count communities after detection', () => {
601
+ graph.addNode(makeEntry('a'));
602
+ graph.addNode(makeEntry('b'));
603
+ graph.detectCommunities();
604
+ const stats = graph.getStats();
605
+ expect(stats.communityCount).toBe(2);
606
+ });
607
+ });
608
+ });
609
+ //# sourceMappingURL=memory-graph.test.js.map
@@ -0,0 +1,68 @@
1
+ /**
2
+ * V3 Memory Migration Utility
3
+ *
4
+ * Migrates data from legacy memory systems (SQLite, Markdown, JSON, etc.)
5
+ * to the unified AgentDB-backed memory system with HNSW indexing.
6
+ *
7
+ * @module v3/memory/migration
8
+ */
9
+ import { EventEmitter } from 'node:events';
10
+ import { MigrationConfig, MigrationProgress, MigrationResult, MigrationSource, EmbeddingGenerator } from './types.js';
11
+ import { AgentDBAdapter } from './agentdb-adapter.js';
12
+ /**
13
+ * Memory Migration Manager
14
+ *
15
+ * Handles migration from:
16
+ * - SQLite backends (.db files)
17
+ * - Markdown backends (.md files)
18
+ * - JSON memory stores (.json files)
19
+ * - MemoryManager instances
20
+ * - SwarmMemory instances
21
+ * - DistributedMemory instances
22
+ */
23
+ export declare class MemoryMigrator extends EventEmitter {
24
+ private config;
25
+ private target;
26
+ private embeddingGenerator?;
27
+ private progress;
28
+ constructor(target: AgentDBAdapter, config: Partial<MigrationConfig>, embeddingGenerator?: EmbeddingGenerator);
29
+ /**
30
+ * Run the migration
31
+ */
32
+ migrate(): Promise<MigrationResult>;
33
+ /**
34
+ * Get current migration progress
35
+ */
36
+ getProgress(): MigrationProgress;
37
+ private loadFromSource;
38
+ private loadFromSQLite;
39
+ private loadFromMarkdown;
40
+ private loadFromJSON;
41
+ private loadFromMemoryManager;
42
+ private loadFromSwarmMemory;
43
+ private loadFromDistributedMemory;
44
+ private processBatch;
45
+ private transformEntry;
46
+ private initializeProgress;
47
+ private validateEntry;
48
+ private addError;
49
+ private parseTimestamp;
50
+ private isValidMemoryType;
51
+ private estimateTimeRemaining;
52
+ private generateSummary;
53
+ private walkDirectory;
54
+ private parseMarkdownEntry;
55
+ }
56
+ /**
57
+ * Convenience function to create a migrator
58
+ */
59
+ export declare function createMigrator(target: AgentDBAdapter, source: MigrationSource, sourcePath: string, options?: Partial<MigrationConfig>, embeddingGenerator?: EmbeddingGenerator): MemoryMigrator;
60
+ /**
61
+ * Migrate from multiple sources
62
+ */
63
+ export declare function migrateMultipleSources(target: AgentDBAdapter, sources: Array<{
64
+ source: MigrationSource;
65
+ path: string;
66
+ }>, options?: Partial<MigrationConfig>, embeddingGenerator?: EmbeddingGenerator): Promise<MigrationResult[]>;
67
+ export default MemoryMigrator;
68
+ //# sourceMappingURL=migration.d.ts.map