@h-rig/memory-plugin 0.0.6-alpha.156

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.
@@ -0,0 +1,1500 @@
1
+ // @bun
2
+ // packages/memory-plugin/src/cli.ts
3
+ import { existsSync } from "fs";
4
+ import { randomUUID } from "crypto";
5
+
6
+ // packages/memory-plugin/src/db.ts
7
+ import { Database } from "bun:sqlite";
8
+ import { mkdirSync } from "fs";
9
+ import { dirname } from "path";
10
+ import {
11
+ NO_MATCH_RETRIEVAL_CANONICAL_KEY
12
+ } from "@rig/contracts";
13
+ var SCHEMA_STATEMENTS = [
14
+ `CREATE TABLE IF NOT EXISTS memory_events (
15
+ event_id TEXT PRIMARY KEY,
16
+ event_type TEXT NOT NULL,
17
+ canonical_key TEXT NOT NULL,
18
+ summary TEXT,
19
+ kind TEXT,
20
+ category TEXT,
21
+ confidence REAL,
22
+ source_run_id TEXT,
23
+ source_task_id TEXT,
24
+ branch TEXT,
25
+ source_canonical_key TEXT,
26
+ replacement_canonical_key TEXT,
27
+ retrieval_query TEXT,
28
+ retrieval_rank INTEGER,
29
+ feedback_outcome TEXT,
30
+ details_json TEXT,
31
+ created_at TEXT NOT NULL
32
+ )`,
33
+ `CREATE TABLE IF NOT EXISTS memory_items (
34
+ canonical_key TEXT PRIMARY KEY,
35
+ summary TEXT NOT NULL,
36
+ kind TEXT,
37
+ category TEXT,
38
+ status TEXT NOT NULL,
39
+ confidence REAL NOT NULL,
40
+ source_run_id TEXT,
41
+ source_task_id TEXT,
42
+ branch TEXT,
43
+ details_json TEXT,
44
+ created_at TEXT NOT NULL,
45
+ updated_at TEXT NOT NULL,
46
+ last_event_id TEXT NOT NULL,
47
+ superseded_by TEXT,
48
+ embedding TEXT
49
+ )`,
50
+ `CREATE TABLE IF NOT EXISTS memory_links (
51
+ id INTEGER PRIMARY KEY AUTOINCREMENT,
52
+ from_key TEXT NOT NULL,
53
+ to_key TEXT NOT NULL,
54
+ relation TEXT NOT NULL,
55
+ source_event_id TEXT NOT NULL,
56
+ created_at TEXT NOT NULL
57
+ )`,
58
+ `CREATE TABLE IF NOT EXISTS task_runs (
59
+ run_id TEXT PRIMARY KEY,
60
+ task_id TEXT,
61
+ branch TEXT,
62
+ first_event_id TEXT,
63
+ last_event_id TEXT,
64
+ created_at TEXT NOT NULL,
65
+ updated_at TEXT NOT NULL
66
+ )`,
67
+ `CREATE TABLE IF NOT EXISTS retrieval_feedback (
68
+ id INTEGER PRIMARY KEY AUTOINCREMENT,
69
+ canonical_key TEXT NOT NULL,
70
+ outcome TEXT NOT NULL,
71
+ retrieval_rank INTEGER,
72
+ source_run_id TEXT,
73
+ source_task_id TEXT,
74
+ query_text TEXT,
75
+ source_event_id TEXT NOT NULL,
76
+ created_at TEXT NOT NULL
77
+ )`
78
+ ];
79
+ var MEMORY_EVENT_COLUMNS = [
80
+ ["summary", "TEXT"],
81
+ ["kind", "TEXT"],
82
+ ["category", "TEXT"],
83
+ ["confidence", "REAL"],
84
+ ["source_run_id", "TEXT"],
85
+ ["source_task_id", "TEXT"],
86
+ ["branch", "TEXT"],
87
+ ["source_canonical_key", "TEXT"],
88
+ ["replacement_canonical_key", "TEXT"],
89
+ ["retrieval_query", "TEXT"],
90
+ ["retrieval_rank", "INTEGER"],
91
+ ["feedback_outcome", "TEXT"],
92
+ ["details_json", "TEXT"]
93
+ ];
94
+ var MEMORY_ITEM_COLUMNS = [
95
+ ["kind", "TEXT"],
96
+ ["category", "TEXT"],
97
+ ["branch", "TEXT"],
98
+ ["details_json", "TEXT"],
99
+ ["superseded_by", "TEXT"],
100
+ ["embedding", "TEXT"]
101
+ ];
102
+ function jsonText(value) {
103
+ if (value == null) {
104
+ return null;
105
+ }
106
+ return JSON.stringify(value);
107
+ }
108
+ function parseJsonRecord(value) {
109
+ if (typeof value !== "string" || value.length === 0) {
110
+ return null;
111
+ }
112
+ return JSON.parse(value);
113
+ }
114
+ function parseEmbedding(value) {
115
+ if (typeof value !== "string" || value.length === 0) {
116
+ return null;
117
+ }
118
+ return JSON.parse(value);
119
+ }
120
+ function rowString(row, key) {
121
+ const value = row[key];
122
+ return value == null ? null : String(value);
123
+ }
124
+ function eventSummary(event) {
125
+ return "summary" in event ? event.summary : null;
126
+ }
127
+ function eventKind(event) {
128
+ return "kind" in event ? event.kind : null;
129
+ }
130
+ function eventCategory(event) {
131
+ return "category" in event ? event.category ?? null : null;
132
+ }
133
+ function eventConfidence(event) {
134
+ return "confidence" in event ? event.confidence : null;
135
+ }
136
+ function eventDetails(event) {
137
+ return "details" in event ? event.details ?? null : null;
138
+ }
139
+ function eventSourceCanonicalKey(event) {
140
+ return "sourceCanonicalKey" in event ? event.sourceCanonicalKey : null;
141
+ }
142
+ function eventReplacementCanonicalKey(event) {
143
+ return "replacementCanonicalKey" in event ? event.replacementCanonicalKey : null;
144
+ }
145
+ function eventRetrievalQuery(event) {
146
+ return "retrievalQuery" in event ? event.retrievalQuery ?? null : null;
147
+ }
148
+ function eventRetrievalRank(event) {
149
+ return "retrievalRank" in event ? event.retrievalRank : null;
150
+ }
151
+ function eventFeedbackOutcome(event) {
152
+ return "feedbackOutcome" in event ? event.feedbackOutcome : null;
153
+ }
154
+ function normalizeStatement(statement, args) {
155
+ if (typeof statement === "string") {
156
+ return { sql: statement, args };
157
+ }
158
+ return {
159
+ sql: statement.sql,
160
+ args: statement.args ?? args
161
+ };
162
+ }
163
+ function normalizeBindings(bindings) {
164
+ return (bindings ?? []).map((binding) => binding === undefined ? null : binding);
165
+ }
166
+ function statementVerb(sql) {
167
+ const match = sql.trim().match(/^[A-Za-z]+/);
168
+ return match ? match[0].toUpperCase() : "";
169
+ }
170
+ function isMutationStatement(sql) {
171
+ return new Set([
172
+ "ALTER",
173
+ "ATTACH",
174
+ "BEGIN",
175
+ "COMMIT",
176
+ "CREATE",
177
+ "DELETE",
178
+ "DETACH",
179
+ "DROP",
180
+ "END",
181
+ "INSERT",
182
+ "REINDEX",
183
+ "RELEASE",
184
+ "REPLACE",
185
+ "ROLLBACK",
186
+ "SAVEPOINT",
187
+ "UPDATE",
188
+ "VACUUM"
189
+ ]).has(statementVerb(sql));
190
+ }
191
+ function executeSqlite(sqlite, statement, args) {
192
+ const normalized = normalizeStatement(statement, args);
193
+ const bindings = normalizeBindings(normalized.args);
194
+ if (isMutationStatement(normalized.sql)) {
195
+ const result = bindings.length > 0 ? sqlite.run(normalized.sql, ...bindings) : sqlite.run(normalized.sql);
196
+ return {
197
+ rows: [],
198
+ rowsAffected: Number(result.changes)
199
+ };
200
+ }
201
+ const query = sqlite.query(normalized.sql);
202
+ const rows = bindings.length > 0 ? query.all(...bindings) : query.all();
203
+ return {
204
+ rows,
205
+ rowsAffected: 0
206
+ };
207
+ }
208
+ function createTransaction(sqlite, mode) {
209
+ sqlite.run(mode === "write" ? "BEGIN IMMEDIATE" : "BEGIN");
210
+ let active = true;
211
+ return {
212
+ async execute(statement, args) {
213
+ if (!active) {
214
+ throw new Error("memory transaction is closed");
215
+ }
216
+ return executeSqlite(sqlite, statement, args);
217
+ },
218
+ async commit() {
219
+ if (!active) {
220
+ return;
221
+ }
222
+ sqlite.run("COMMIT");
223
+ active = false;
224
+ },
225
+ async rollback() {
226
+ if (!active) {
227
+ return;
228
+ }
229
+ sqlite.run("ROLLBACK");
230
+ active = false;
231
+ },
232
+ close() {
233
+ if (!active) {
234
+ return;
235
+ }
236
+ try {
237
+ sqlite.run("ROLLBACK");
238
+ } catch {} finally {
239
+ active = false;
240
+ }
241
+ }
242
+ };
243
+ }
244
+ function createMemoryDbClient(sqlite) {
245
+ return {
246
+ async execute(statement, args) {
247
+ return executeSqlite(sqlite, statement, args);
248
+ },
249
+ async transaction(mode) {
250
+ return createTransaction(sqlite, mode);
251
+ },
252
+ close() {
253
+ sqlite.close();
254
+ }
255
+ };
256
+ }
257
+ async function listColumns(executor, tableName) {
258
+ const result = await executor.execute(`PRAGMA table_info(${tableName})`);
259
+ return new Set(result.rows.map((row) => String(row.name)));
260
+ }
261
+ async function ensureColumns(executor, tableName, columns) {
262
+ const existing = await listColumns(executor, tableName);
263
+ for (const [columnName, definition] of columns) {
264
+ if (!existing.has(columnName)) {
265
+ await executor.execute(`ALTER TABLE ${tableName} ADD COLUMN ${columnName} ${definition}`);
266
+ }
267
+ }
268
+ }
269
+ async function ensureSchema(db) {
270
+ for (const statement of SCHEMA_STATEMENTS) {
271
+ await db.client.execute(statement);
272
+ }
273
+ await ensureColumns(db.client, "memory_events", MEMORY_EVENT_COLUMNS);
274
+ await ensureColumns(db.client, "memory_items", MEMORY_ITEM_COLUMNS);
275
+ }
276
+ function toMemoryItemRow(row) {
277
+ return {
278
+ canonicalKey: String(row.canonical_key),
279
+ summary: String(row.summary),
280
+ kind: rowString(row, "kind"),
281
+ category: rowString(row, "category"),
282
+ status: String(row.status),
283
+ confidence: Number(row.confidence),
284
+ sourceRunId: rowString(row, "source_run_id"),
285
+ sourceTaskId: rowString(row, "source_task_id"),
286
+ branch: rowString(row, "branch"),
287
+ details: parseJsonRecord(row.details_json),
288
+ createdAt: String(row.created_at),
289
+ updatedAt: String(row.updated_at),
290
+ lastEventId: String(row.last_event_id),
291
+ supersededBy: rowString(row, "superseded_by"),
292
+ embedding: parseEmbedding(row.embedding)
293
+ };
294
+ }
295
+ async function hasMemoryItem(executor, canonicalKey) {
296
+ const result = await executor.execute({
297
+ sql: "SELECT 1 AS present FROM memory_items WHERE canonical_key = ? LIMIT 1",
298
+ args: [canonicalKey]
299
+ });
300
+ return result.rows.length > 0;
301
+ }
302
+ async function assertMemoryItemExists(executor, canonicalKey, errorPrefix) {
303
+ if (!await hasMemoryItem(executor, canonicalKey)) {
304
+ throw new Error(`${errorPrefix}: unknown memory ${canonicalKey}`);
305
+ }
306
+ }
307
+ async function upsertTaskRun(executor, event) {
308
+ if (!event.sourceRunId) {
309
+ return;
310
+ }
311
+ await executor.execute({
312
+ sql: `
313
+ INSERT INTO task_runs (
314
+ run_id,
315
+ task_id,
316
+ branch,
317
+ first_event_id,
318
+ last_event_id,
319
+ created_at,
320
+ updated_at
321
+ ) VALUES (?, ?, ?, ?, ?, ?, ?)
322
+ ON CONFLICT(run_id) DO UPDATE SET
323
+ task_id = COALESCE(excluded.task_id, task_runs.task_id),
324
+ branch = COALESCE(excluded.branch, task_runs.branch),
325
+ last_event_id = excluded.last_event_id,
326
+ updated_at = excluded.updated_at
327
+ `,
328
+ args: [
329
+ event.sourceRunId,
330
+ event.sourceTaskId ?? null,
331
+ event.branch ?? null,
332
+ event.eventId,
333
+ event.eventId,
334
+ event.createdAt,
335
+ event.createdAt
336
+ ]
337
+ });
338
+ }
339
+ async function insertEventRow(executor, event) {
340
+ await executor.execute({
341
+ sql: `
342
+ INSERT INTO memory_events (
343
+ event_id,
344
+ event_type,
345
+ canonical_key,
346
+ summary,
347
+ kind,
348
+ category,
349
+ confidence,
350
+ source_run_id,
351
+ source_task_id,
352
+ branch,
353
+ source_canonical_key,
354
+ replacement_canonical_key,
355
+ retrieval_query,
356
+ retrieval_rank,
357
+ feedback_outcome,
358
+ details_json,
359
+ created_at
360
+ ) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
361
+ `,
362
+ args: [
363
+ event.eventId,
364
+ event.eventType,
365
+ event.canonicalKey,
366
+ eventSummary(event),
367
+ eventKind(event),
368
+ eventCategory(event),
369
+ eventConfidence(event),
370
+ event.sourceRunId ?? null,
371
+ event.sourceTaskId ?? null,
372
+ event.branch ?? null,
373
+ eventSourceCanonicalKey(event),
374
+ eventReplacementCanonicalKey(event),
375
+ eventRetrievalQuery(event),
376
+ eventRetrievalRank(event),
377
+ eventFeedbackOutcome(event),
378
+ jsonText(eventDetails(event)),
379
+ event.createdAt
380
+ ]
381
+ });
382
+ }
383
+ async function upsertProjectedMemory(executor, event, status = "active") {
384
+ await executor.execute({
385
+ sql: `
386
+ INSERT INTO memory_items (
387
+ canonical_key,
388
+ summary,
389
+ kind,
390
+ category,
391
+ status,
392
+ confidence,
393
+ source_run_id,
394
+ source_task_id,
395
+ branch,
396
+ details_json,
397
+ created_at,
398
+ updated_at,
399
+ last_event_id,
400
+ superseded_by,
401
+ embedding
402
+ ) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, NULL)
403
+ ON CONFLICT(canonical_key) DO UPDATE SET
404
+ summary = excluded.summary,
405
+ kind = excluded.kind,
406
+ category = excluded.category,
407
+ status = excluded.status,
408
+ confidence = COALESCE(excluded.confidence, memory_items.confidence),
409
+ source_run_id = COALESCE(excluded.source_run_id, memory_items.source_run_id),
410
+ source_task_id = COALESCE(excluded.source_task_id, memory_items.source_task_id),
411
+ branch = COALESCE(excluded.branch, memory_items.branch),
412
+ details_json = COALESCE(excluded.details_json, memory_items.details_json),
413
+ updated_at = excluded.updated_at,
414
+ last_event_id = excluded.last_event_id,
415
+ superseded_by = excluded.superseded_by,
416
+ embedding = excluded.embedding
417
+ `,
418
+ args: [
419
+ event.canonicalKey,
420
+ eventSummary(event),
421
+ eventKind(event),
422
+ eventCategory(event),
423
+ status,
424
+ eventConfidence(event) ?? 0,
425
+ event.sourceRunId ?? null,
426
+ event.sourceTaskId ?? null,
427
+ event.branch ?? null,
428
+ jsonText(eventDetails(event)),
429
+ event.createdAt,
430
+ event.createdAt,
431
+ event.eventId,
432
+ eventReplacementCanonicalKey(event)
433
+ ]
434
+ });
435
+ }
436
+ async function insertMemoryLink(executor, fromKey, toKey, relation, event) {
437
+ await executor.execute({
438
+ sql: `
439
+ INSERT INTO memory_links (from_key, to_key, relation, source_event_id, created_at)
440
+ VALUES (?, ?, ?, ?, ?)
441
+ `,
442
+ args: [fromKey, toKey, relation, event.eventId, event.createdAt]
443
+ });
444
+ }
445
+ async function markSuperseded(executor, event) {
446
+ const result = await executor.execute({
447
+ sql: `
448
+ UPDATE memory_items
449
+ SET
450
+ status = 'superseded',
451
+ superseded_by = ?,
452
+ updated_at = ?,
453
+ last_event_id = ?
454
+ WHERE canonical_key = ?
455
+ `,
456
+ args: [eventReplacementCanonicalKey(event), event.createdAt, event.eventId, event.canonicalKey]
457
+ });
458
+ if ((result.rowsAffected ?? 0) === 0) {
459
+ throw new Error(`cannot supersede unknown memory ${event.canonicalKey}`);
460
+ }
461
+ if (eventReplacementCanonicalKey(event)) {
462
+ await insertMemoryLink(executor, event.canonicalKey, eventReplacementCanonicalKey(event), "superseded_by", event);
463
+ }
464
+ }
465
+ async function activeMemoryItem(executor, canonicalKey) {
466
+ const result = await executor.execute({
467
+ sql: `
468
+ SELECT
469
+ canonical_key,
470
+ summary,
471
+ kind,
472
+ category,
473
+ status,
474
+ confidence,
475
+ source_run_id,
476
+ source_task_id,
477
+ branch,
478
+ details_json,
479
+ created_at,
480
+ updated_at,
481
+ last_event_id,
482
+ superseded_by,
483
+ embedding
484
+ FROM memory_items
485
+ WHERE canonical_key = ?
486
+ LIMIT 1
487
+ `,
488
+ args: [canonicalKey]
489
+ });
490
+ const row = result.rows[0];
491
+ return row ? toMemoryItemRow(row) : null;
492
+ }
493
+ async function markRetracted(executor, event) {
494
+ const result = await executor.execute({
495
+ sql: `
496
+ UPDATE memory_items
497
+ SET
498
+ status = 'retracted',
499
+ updated_at = ?,
500
+ last_event_id = ?
501
+ WHERE canonical_key = ?
502
+ `,
503
+ args: [event.createdAt, event.eventId, event.canonicalKey]
504
+ });
505
+ if ((result.rowsAffected ?? 0) === 0) {
506
+ throw new Error(`cannot retract unknown memory ${event.canonicalKey}`);
507
+ }
508
+ }
509
+ async function recordRetrievalFeedback(executor, event) {
510
+ await executor.execute({
511
+ sql: `
512
+ INSERT INTO retrieval_feedback (
513
+ canonical_key,
514
+ outcome,
515
+ retrieval_rank,
516
+ source_run_id,
517
+ source_task_id,
518
+ query_text,
519
+ source_event_id,
520
+ created_at
521
+ ) VALUES (?, ?, ?, ?, ?, ?, ?, ?)
522
+ `,
523
+ args: [
524
+ event.canonicalKey,
525
+ eventFeedbackOutcome(event) ?? "ignored",
526
+ eventRetrievalRank(event),
527
+ event.sourceRunId ?? null,
528
+ event.sourceTaskId ?? null,
529
+ eventRetrievalQuery(event),
530
+ event.eventId,
531
+ event.createdAt
532
+ ]
533
+ });
534
+ }
535
+ async function applyReinforcement(executor, event) {
536
+ const existing = await activeMemoryItem(executor, event.canonicalKey);
537
+ if (!existing) {
538
+ throw new Error(`cannot reinforce unknown memory ${event.canonicalKey}`);
539
+ }
540
+ if (existing.status !== "active") {
541
+ throw new Error(`cannot reinforce non-active memory ${event.canonicalKey}`);
542
+ }
543
+ const result = await executor.execute({
544
+ sql: `
545
+ UPDATE memory_items
546
+ SET
547
+ status = 'active',
548
+ confidence = ?,
549
+ source_run_id = COALESCE(?, source_run_id),
550
+ source_task_id = COALESCE(?, source_task_id),
551
+ branch = COALESCE(?, branch),
552
+ updated_at = ?,
553
+ last_event_id = ?,
554
+ superseded_by = NULL
555
+ WHERE canonical_key = ?
556
+ `,
557
+ args: [
558
+ eventConfidence(event),
559
+ event.sourceRunId ?? null,
560
+ event.sourceTaskId ?? null,
561
+ event.branch ?? null,
562
+ event.createdAt,
563
+ event.eventId,
564
+ event.canonicalKey
565
+ ]
566
+ });
567
+ if ((result.rowsAffected ?? 0) === 0) {
568
+ throw new Error(`cannot reinforce unknown memory ${event.canonicalKey}`);
569
+ }
570
+ }
571
+ async function validateEventTargets(executor, event) {
572
+ switch (event.eventType) {
573
+ case "observed":
574
+ return;
575
+ case "promoted":
576
+ case "corrected":
577
+ await assertMemoryItemExists(executor, event.sourceCanonicalKey, `cannot ${event.eventType}`);
578
+ return;
579
+ case "superseded":
580
+ await assertMemoryItemExists(executor, event.replacementCanonicalKey, "cannot supersede");
581
+ await assertMemoryItemExists(executor, event.canonicalKey, "cannot supersede");
582
+ return;
583
+ case "retracted":
584
+ case "feedback_recorded":
585
+ await assertMemoryItemExists(executor, event.canonicalKey, `cannot ${event.eventType}`);
586
+ return;
587
+ case "reinforced":
588
+ await assertMemoryItemExists(executor, event.canonicalKey, "cannot reinforce");
589
+ return;
590
+ case "retrieved":
591
+ if (event.retrievalRank === 0 && event.canonicalKey === NO_MATCH_RETRIEVAL_CANONICAL_KEY) {
592
+ return;
593
+ }
594
+ await assertMemoryItemExists(executor, event.canonicalKey, `cannot ${event.eventType}`);
595
+ return;
596
+ default: {
597
+ const _exhaustive = event;
598
+ return _exhaustive;
599
+ }
600
+ }
601
+ }
602
+ async function openMemoryDb(dbPath) {
603
+ mkdirSync(dirname(dbPath), { recursive: true });
604
+ const sqlite = new Database(dbPath, { create: true, strict: true });
605
+ const client = createMemoryDbClient(sqlite);
606
+ const db = {
607
+ path: dbPath,
608
+ client,
609
+ async close() {
610
+ client.close();
611
+ }
612
+ };
613
+ await ensureSchema(db);
614
+ return db;
615
+ }
616
+ async function applyMemoryEvent(db, event) {
617
+ const transaction = await db.client.transaction("write");
618
+ try {
619
+ await validateEventTargets(transaction, event);
620
+ await insertEventRow(transaction, event);
621
+ await upsertTaskRun(transaction, event);
622
+ switch (event.eventType) {
623
+ case "observed":
624
+ await upsertProjectedMemory(transaction, event, "active");
625
+ break;
626
+ case "promoted":
627
+ await upsertProjectedMemory(transaction, event, "active");
628
+ if (event.sourceCanonicalKey) {
629
+ await insertMemoryLink(transaction, event.canonicalKey, event.sourceCanonicalKey, "promoted_from", event);
630
+ }
631
+ break;
632
+ case "corrected":
633
+ await upsertProjectedMemory(transaction, event, "active");
634
+ if (event.sourceCanonicalKey) {
635
+ await insertMemoryLink(transaction, event.canonicalKey, event.sourceCanonicalKey, "corrected_from", event);
636
+ }
637
+ break;
638
+ case "superseded":
639
+ await markSuperseded(transaction, event);
640
+ break;
641
+ case "retracted":
642
+ await markRetracted(transaction, event);
643
+ break;
644
+ case "reinforced":
645
+ await applyReinforcement(transaction, event);
646
+ break;
647
+ case "retrieved":
648
+ break;
649
+ case "feedback_recorded":
650
+ await recordRetrievalFeedback(transaction, event);
651
+ break;
652
+ }
653
+ await transaction.commit();
654
+ } catch (error) {
655
+ await transaction.rollback();
656
+ throw error;
657
+ } finally {
658
+ transaction.close();
659
+ }
660
+ }
661
+ async function setMemoryItemEmbedding(db, canonicalKey, embedding) {
662
+ await db.client.execute({
663
+ sql: `
664
+ UPDATE memory_items
665
+ SET embedding = ?
666
+ WHERE canonical_key = ?
667
+ `,
668
+ args: [JSON.stringify(embedding), canonicalKey]
669
+ });
670
+ }
671
+ async function getMemoryItemByKey(db, canonicalKey) {
672
+ const result = await db.client.execute({
673
+ sql: `
674
+ SELECT
675
+ canonical_key,
676
+ summary,
677
+ kind,
678
+ category,
679
+ status,
680
+ confidence,
681
+ source_run_id,
682
+ source_task_id,
683
+ branch,
684
+ details_json,
685
+ created_at,
686
+ updated_at,
687
+ last_event_id,
688
+ superseded_by,
689
+ embedding
690
+ FROM memory_items
691
+ WHERE canonical_key = ?
692
+ LIMIT 1
693
+ `,
694
+ args: [canonicalKey]
695
+ });
696
+ const row = result.rows[0];
697
+ return row ? toMemoryItemRow(row) : null;
698
+ }
699
+ async function listActiveMemoryItems(db) {
700
+ const result = await db.client.execute(`
701
+ SELECT
702
+ canonical_key,
703
+ summary,
704
+ kind,
705
+ category,
706
+ status,
707
+ confidence,
708
+ source_run_id,
709
+ source_task_id,
710
+ branch,
711
+ details_json,
712
+ created_at,
713
+ updated_at,
714
+ last_event_id,
715
+ superseded_by,
716
+ embedding
717
+ FROM memory_items
718
+ WHERE status = 'active'
719
+ ORDER BY canonical_key
720
+ `);
721
+ return result.rows.map((row) => toMemoryItemRow(row));
722
+ }
723
+ async function hasMemoryEvent(db, eventId) {
724
+ const result = await db.client.execute({
725
+ sql: "SELECT 1 AS present FROM memory_events WHERE event_id = ? LIMIT 1",
726
+ args: [eventId]
727
+ });
728
+ return result.rows.length > 0;
729
+ }
730
+
731
+ // packages/memory-plugin/src/embed.ts
732
+ import { createHash } from "crypto";
733
+ var DEFAULT_EMBEDDING_API_BASE_URL = "https://api.openai.com/v1";
734
+ var DEFAULT_EMBEDDING_MODEL = "text-embedding-3-small";
735
+ var DETERMINISTIC_EMBEDDER_MODE = "deterministic";
736
+ var DETERMINISTIC_VECTOR_DIMS = 32;
737
+ var MISSING_MEMORY_EMBEDDER_WARNING = "Shared memory embeddings disabled: OPENAI_API_KEY is not set. Continuing with lexical-only memory.";
738
+ var missingMemoryEmbedderWarningShown = false;
739
+ function embeddingTextForItem(item) {
740
+ const parts = [item.summary, item.kind ?? "", item.category ?? "", item.details ? JSON.stringify(item.details) : ""];
741
+ return parts.filter(Boolean).join(`
742
+ `);
743
+ }
744
+ function tokenizeForDeterministicEmbedding(text) {
745
+ const tokens = text.toLowerCase().match(/[a-z0-9_./:-]+/g) ?? [];
746
+ return tokens.length > 0 ? tokens : [text.toLowerCase()];
747
+ }
748
+ function deterministicTokenVector(token) {
749
+ const hash = createHash("sha256").update(token).digest();
750
+ const vector = new Array(DETERMINISTIC_VECTOR_DIMS).fill(0);
751
+ for (let index = 0;index < 8; index += 1) {
752
+ const slot = (hash[index] ?? 0) % DETERMINISTIC_VECTOR_DIMS;
753
+ const sign = ((hash[index + 8] ?? 0) & 1) === 0 ? 1 : -1;
754
+ const magnitude = ((hash[index + 16] ?? 0) + 1) / 256;
755
+ vector[slot] = (vector[slot] ?? 0) + sign * magnitude;
756
+ }
757
+ return vector;
758
+ }
759
+ function normalizeVector(vector) {
760
+ const norm = Math.sqrt(vector.reduce((sum, value) => sum + value * value, 0));
761
+ if (norm === 0) {
762
+ return vector.map(() => 0);
763
+ }
764
+ return vector.map((value) => Number((value / norm).toFixed(6)));
765
+ }
766
+ function createDeterministicMemoryEmbedder() {
767
+ return {
768
+ async embed(texts) {
769
+ return texts.map((text) => {
770
+ const vector = new Array(DETERMINISTIC_VECTOR_DIMS).fill(0);
771
+ for (const token of tokenizeForDeterministicEmbedding(text)) {
772
+ const tokenVector = deterministicTokenVector(token);
773
+ for (let index = 0;index < tokenVector.length; index += 1) {
774
+ vector[index] = (vector[index] ?? 0) + (tokenVector[index] ?? 0);
775
+ }
776
+ }
777
+ return normalizeVector(vector);
778
+ });
779
+ }
780
+ };
781
+ }
782
+ function embeddingResponseMessage(payload, status) {
783
+ if (typeof payload === "object" && payload !== null) {
784
+ const maybeMessage = payload.error?.message;
785
+ if (maybeMessage) {
786
+ return maybeMessage;
787
+ }
788
+ }
789
+ return `memory embedding request failed with status ${status}`;
790
+ }
791
+ function createOpenAiMemoryEmbedder(options) {
792
+ const apiKey = options.apiKey.trim();
793
+ const model = options.model?.trim() || DEFAULT_EMBEDDING_MODEL;
794
+ const apiBaseUrl = (options.apiBaseUrl?.trim() || DEFAULT_EMBEDDING_API_BASE_URL).replace(/\/+$/, "");
795
+ const fetchImpl = options.fetchImpl ?? fetch;
796
+ if (!apiKey) {
797
+ throw new Error("memory embedding api key must be non-empty");
798
+ }
799
+ return {
800
+ async embed(texts) {
801
+ if (texts.length === 0) {
802
+ return [];
803
+ }
804
+ const response = await fetchImpl(`${apiBaseUrl}/embeddings`, {
805
+ method: "POST",
806
+ headers: {
807
+ authorization: `Bearer ${apiKey}`,
808
+ "content-type": "application/json"
809
+ },
810
+ body: JSON.stringify({
811
+ input: texts,
812
+ model
813
+ })
814
+ });
815
+ const payload = await response.json();
816
+ if (!response.ok) {
817
+ throw new Error(embeddingResponseMessage(payload, response.status));
818
+ }
819
+ if (!Array.isArray(payload.data) || payload.data.length !== texts.length) {
820
+ throw new Error(`memory embedding provider returned ${payload.data?.length ?? 0} embeddings for ${texts.length} inputs`);
821
+ }
822
+ const byIndex = new Map;
823
+ for (const row of payload.data) {
824
+ if (typeof row.index !== "number" || !Array.isArray(row.embedding)) {
825
+ throw new Error("memory embedding provider returned an invalid response payload");
826
+ }
827
+ byIndex.set(row.index, row.embedding);
828
+ }
829
+ return texts.map((_text, index) => {
830
+ const embedding = byIndex.get(index);
831
+ if (!embedding) {
832
+ throw new Error(`memory embedding provider omitted embedding ${index}`);
833
+ }
834
+ return embedding;
835
+ });
836
+ }
837
+ };
838
+ }
839
+ function createConfiguredMemoryEmbedder(options = {}) {
840
+ const env = options.env ?? process.env;
841
+ const mode = env.RIG_MEMORY_EMBEDDER?.trim();
842
+ if (mode === DETERMINISTIC_EMBEDDER_MODE) {
843
+ return createDeterministicMemoryEmbedder();
844
+ }
845
+ const apiKey = env.OPENAI_API_KEY?.trim();
846
+ if (!apiKey) {
847
+ throw new Error("memory embeddings require OPENAI_API_KEY or RIG_MEMORY_EMBEDDER=deterministic");
848
+ }
849
+ return createOpenAiMemoryEmbedder({
850
+ apiKey,
851
+ model: env.RIG_MEMORY_EMBEDDING_MODEL?.trim() || DEFAULT_EMBEDDING_MODEL,
852
+ apiBaseUrl: options.apiBaseUrl ?? env.RIG_MEMORY_EMBEDDING_API_BASE_URL?.trim() ?? DEFAULT_EMBEDDING_API_BASE_URL,
853
+ fetchImpl: options.fetchImpl
854
+ });
855
+ }
856
+ function warnMissingMemoryEmbedderOnce(warn) {
857
+ if (missingMemoryEmbedderWarningShown) {
858
+ return;
859
+ }
860
+ missingMemoryEmbedderWarningShown = true;
861
+ warn(MISSING_MEMORY_EMBEDDER_WARNING);
862
+ }
863
+ function embeddingTargetsForEvent(event) {
864
+ switch (event.eventType) {
865
+ case "observed":
866
+ case "promoted":
867
+ case "corrected":
868
+ case "reinforced":
869
+ return [event.canonicalKey];
870
+ case "superseded":
871
+ case "retracted":
872
+ case "retrieved":
873
+ case "feedback_recorded":
874
+ return [];
875
+ }
876
+ }
877
+ function memoryEventRequiresEmbedding(event) {
878
+ return embeddingTargetsForEvent(event).length > 0;
879
+ }
880
+ function maybeCreateConfiguredMemoryEmbedder(options = {}, warn = (message) => {
881
+ console.warn(message);
882
+ }) {
883
+ const env = options.env ?? process.env;
884
+ const mode = env.RIG_MEMORY_EMBEDDER?.trim();
885
+ if (mode === DETERMINISTIC_EMBEDDER_MODE) {
886
+ return createConfiguredMemoryEmbedder(options);
887
+ }
888
+ const apiKey = env.OPENAI_API_KEY?.trim();
889
+ if (!apiKey) {
890
+ warnMissingMemoryEmbedderOnce(warn);
891
+ return null;
892
+ }
893
+ return createConfiguredMemoryEmbedder(options);
894
+ }
895
+ async function embedChangedMemoryItems(db, event, embedder) {
896
+ const targets = [...new Set(embeddingTargetsForEvent(event))];
897
+ if (targets.length === 0) {
898
+ return [];
899
+ }
900
+ const items = [];
901
+ for (const canonicalKey of targets) {
902
+ const item = await getMemoryItemByKey(db, canonicalKey);
903
+ if (item && item.status === "active") {
904
+ items.push(item);
905
+ }
906
+ }
907
+ if (items.length === 0) {
908
+ return [];
909
+ }
910
+ const texts = items.map(embeddingTextForItem);
911
+ const embeddings = await embedder.embed(texts);
912
+ for (const [index, item] of items.entries()) {
913
+ const embedding = embeddings[index];
914
+ if (!embedding) {
915
+ throw new Error(`missing embedding for ${item.canonicalKey}`);
916
+ }
917
+ await setMemoryItemEmbedding(db, item.canonicalKey, embedding);
918
+ }
919
+ return texts;
920
+ }
921
+
922
+ // packages/memory-plugin/src/query.ts
923
+ import {
924
+ DEFAULT_RUNTIME_MEMORY_RETRIEVAL
925
+ } from "@rig/runtime/control-plane/runtime/context";
926
+ var DEFAULT_RESULT_LIMIT = DEFAULT_RUNTIME_MEMORY_RETRIEVAL.topK;
927
+ var DAY_MS = 24 * 60 * 60 * 1000;
928
+ var MIN_VECTOR_MATCH_SCORE = 0.2;
929
+ function tokenize(text) {
930
+ return (text.toLowerCase().match(/[a-z0-9_./:-]+/g) ?? []).flatMap((token) => {
931
+ const split = token.split(/[./:_-]+/).filter(Boolean);
932
+ return split.length > 0 ? [token, ...split] : [token];
933
+ });
934
+ }
935
+ function lexicalScore(query, item) {
936
+ const queryTokens = [...new Set(tokenize(query))];
937
+ if (queryTokens.length === 0) {
938
+ return 0;
939
+ }
940
+ const haystackTokens = new Set(tokenize(item.summary));
941
+ let matched = 0;
942
+ for (const token of queryTokens) {
943
+ if (haystackTokens.has(token)) {
944
+ matched += 1;
945
+ }
946
+ }
947
+ return matched / queryTokens.length;
948
+ }
949
+ function cosineSimilarity(left, right) {
950
+ if (left.length === 0 || left.length !== right.length) {
951
+ return 0;
952
+ }
953
+ let dot = 0;
954
+ let leftNorm = 0;
955
+ let rightNorm = 0;
956
+ for (let index = 0;index < left.length; index += 1) {
957
+ const l = left[index] ?? 0;
958
+ const r = right[index] ?? 0;
959
+ dot += l * r;
960
+ leftNorm += l * l;
961
+ rightNorm += r * r;
962
+ }
963
+ if (leftNorm === 0 || rightNorm === 0) {
964
+ return 0;
965
+ }
966
+ return dot / Math.sqrt(leftNorm * rightNorm);
967
+ }
968
+ function recencyScore(item, now) {
969
+ const updatedAt = Date.parse(item.updatedAt);
970
+ if (Number.isNaN(updatedAt)) {
971
+ return 0;
972
+ }
973
+ const ageMs = Math.max(0, now.getTime() - updatedAt);
974
+ return 1 / (1 + ageMs / DAY_MS);
975
+ }
976
+ async function maybeEmbedQuery(items, query, embedder) {
977
+ if (!items.some((item) => Array.isArray(item.embedding) && item.embedding.length > 0)) {
978
+ return null;
979
+ }
980
+ try {
981
+ const resolvedEmbedder = embedder ?? createConfiguredMemoryEmbedder();
982
+ const [queryEmbedding] = await resolvedEmbedder.embed([query]);
983
+ return queryEmbedding ?? null;
984
+ } catch {
985
+ return null;
986
+ }
987
+ }
988
+ async function queryRelevantMemory(db, input) {
989
+ const retrieval = input.retrieval ?? DEFAULT_RUNTIME_MEMORY_RETRIEVAL;
990
+ const now = input.now instanceof Date ? input.now : new Date(input.now ?? Date.now());
991
+ const items = await listActiveMemoryItems(db);
992
+ if (items.length === 0) {
993
+ return [];
994
+ }
995
+ const queryEmbedding = await maybeEmbedQuery(items, input.query, input.embedder);
996
+ const scored = items.map((item) => {
997
+ const lexical = lexicalScore(input.query, item);
998
+ const vector = queryEmbedding && item.embedding ? cosineSimilarity(queryEmbedding, item.embedding) : 0;
999
+ const recency = recencyScore(item, now);
1000
+ const score = retrieval.lexicalWeight * lexical + retrieval.vectorWeight * vector + retrieval.recencyWeight * recency + retrieval.confidenceWeight * item.confidence;
1001
+ return {
1002
+ canonicalKey: item.canonicalKey,
1003
+ summary: item.summary,
1004
+ kind: item.kind,
1005
+ category: item.category,
1006
+ confidence: item.confidence,
1007
+ updatedAt: item.updatedAt,
1008
+ score,
1009
+ lexicalScore: lexical,
1010
+ vectorScore: vector,
1011
+ recencyScore: recency
1012
+ };
1013
+ }).filter((item) => item.lexicalScore > 0 || item.vectorScore >= MIN_VECTOR_MATCH_SCORE).sort((left, right) => right.score - left.score || Date.parse(right.updatedAt) - Date.parse(left.updatedAt) || right.confidence - left.confidence || left.canonicalKey.localeCompare(right.canonicalKey));
1014
+ return scored.slice(0, input.limit ?? retrieval.topK ?? DEFAULT_RESULT_LIMIT);
1015
+ }
1016
+ function formatMemoryQueryResults(results) {
1017
+ if (results.length === 0) {
1018
+ return "No shared memories matched.";
1019
+ }
1020
+ return results.map((result, index) => `${index + 1}. [${result.canonicalKey}] ${result.summary} (confidence ${result.confidence.toFixed(2)})`).join(`
1021
+ `);
1022
+ }
1023
+
1024
+ // packages/memory-plugin/src/write.ts
1025
+ import { nativePushRefWithLease, nativeWriteTreeCommit } from "@rig/runtime/control-plane/native/git-native";
1026
+
1027
+ // packages/memory-plugin/src/read.ts
1028
+ import { mkdtempSync, rmSync, writeFileSync } from "fs";
1029
+ import { tmpdir } from "os";
1030
+ import { join } from "path";
1031
+ import { nativeFetchRef, nativeReadBlobBytesAtRef } from "@rig/runtime/control-plane/native/git-native";
1032
+ import { resolveMonorepoRoot } from "@rig/runtime/control-plane/native/utils";
1033
+ var CANONICAL_MEMORY_DB_PATH = "rig/memory/project-memory.db";
1034
+ var DEFAULT_READ_DEPS = {
1035
+ fetchRef: nativeFetchRef,
1036
+ readBlobBytesAtRef: nativeReadBlobBytesAtRef,
1037
+ openMemoryDb,
1038
+ makeTempDir: () => mkdtempSync(join(tmpdir(), "memory-sync-read-")),
1039
+ removeDir: (path) => rmSync(path, { recursive: true, force: true })
1040
+ };
1041
+ function isMissingCanonicalMemoryBlobError(error) {
1042
+ const message = error instanceof Error ? error.message : String(error);
1043
+ return message.includes(`path '${CANONICAL_MEMORY_DB_PATH}' does not exist in`) || message.includes(`path '${CANONICAL_MEMORY_DB_PATH}' exists on disk, but not in`) || message.includes(`pathspec '${CANONICAL_MEMORY_DB_PATH}' did not match any file(s) known to git`);
1044
+ }
1045
+ async function validateReadableDatabase(dbPath, open) {
1046
+ const db = await open(dbPath);
1047
+ await db.close();
1048
+ }
1049
+ async function readCanonicalMemoryDb(projectRoot, deps = {}) {
1050
+ const readDeps = { ...DEFAULT_READ_DEPS, ...deps };
1051
+ const repoPath = resolveMonorepoRoot(projectRoot);
1052
+ let baseOid;
1053
+ let remoteUnavailable = false;
1054
+ try {
1055
+ baseOid = readDeps.fetchRef(repoPath, "origin", "main");
1056
+ } catch (error) {
1057
+ const message = error instanceof Error ? error.message : String(error);
1058
+ const originAbsent = /no such remote|remote (?:'|")?origin(?:'|")? not found/i.test(message);
1059
+ if (!originAbsent) {
1060
+ console.warn(`[rig-memory] canonical project memory unavailable (origin/main): ${message.split(`
1061
+ `)[0]} \u2014 starting this run with empty project memory.`);
1062
+ }
1063
+ baseOid = "";
1064
+ remoteUnavailable = true;
1065
+ }
1066
+ const tempDir = readDeps.makeTempDir();
1067
+ const dbPath = join(tempDir, "project-memory.db");
1068
+ let createdFresh = false;
1069
+ try {
1070
+ if (remoteUnavailable) {
1071
+ const db = await readDeps.openMemoryDb(dbPath);
1072
+ await db.close();
1073
+ createdFresh = true;
1074
+ } else {
1075
+ try {
1076
+ const bytes = readDeps.readBlobBytesAtRef(repoPath, baseOid, CANONICAL_MEMORY_DB_PATH);
1077
+ writeFileSync(dbPath, bytes);
1078
+ } catch (error) {
1079
+ if (!isMissingCanonicalMemoryBlobError(error)) {
1080
+ throw error;
1081
+ }
1082
+ const db = await readDeps.openMemoryDb(dbPath);
1083
+ await db.close();
1084
+ createdFresh = true;
1085
+ }
1086
+ }
1087
+ await validateReadableDatabase(dbPath, readDeps.openMemoryDb);
1088
+ return {
1089
+ repoPath,
1090
+ baseOid,
1091
+ dbPath,
1092
+ createdFresh,
1093
+ async cleanup() {
1094
+ readDeps.removeDir(tempDir);
1095
+ }
1096
+ };
1097
+ } catch (error) {
1098
+ readDeps.removeDir(tempDir);
1099
+ throw error;
1100
+ }
1101
+ }
1102
+
1103
+ // packages/memory-plugin/src/write.ts
1104
+ var CANONICAL_MEMORY_DB_PATH2 = "rig/memory/project-memory.db";
1105
+ var MAX_PROMOTION_ATTEMPTS = 2;
1106
+ var DEFAULT_WRITE_DEPS = {
1107
+ readCanonicalMemoryDb,
1108
+ openMemoryDb,
1109
+ applyMemoryEvent,
1110
+ hasMemoryEvent,
1111
+ embedChangedMemoryItems,
1112
+ writeTreeCommit: nativeWriteTreeCommit,
1113
+ pushRefWithLease: nativePushRefWithLease
1114
+ };
1115
+ async function promoteCanonicalMemoryEvent(projectRoot, input, deps = {}) {
1116
+ const writeDeps = { ...DEFAULT_WRITE_DEPS, ...deps };
1117
+ let lastError;
1118
+ for (let attempt = 0;attempt < MAX_PROMOTION_ATTEMPTS; attempt += 1) {
1119
+ const snapshot = await writeDeps.readCanonicalMemoryDb(projectRoot);
1120
+ try {
1121
+ const db = await writeDeps.openMemoryDb(snapshot.dbPath);
1122
+ try {
1123
+ if (await writeDeps.hasMemoryEvent(db, input.event.eventId)) {
1124
+ return {
1125
+ outcome: "applied",
1126
+ baseOid: snapshot.baseOid,
1127
+ commitOid: snapshot.baseOid
1128
+ };
1129
+ }
1130
+ await writeDeps.applyMemoryEvent(db, input.event);
1131
+ const embedder = input.embedder ?? (memoryEventRequiresEmbedding(input.event) ? maybeCreateConfiguredMemoryEmbedder() : null);
1132
+ if (embedder) {
1133
+ await writeDeps.embedChangedMemoryItems(db, input.event, embedder);
1134
+ }
1135
+ } finally {
1136
+ await db.close();
1137
+ }
1138
+ const updates = [
1139
+ {
1140
+ path: CANONICAL_MEMORY_DB_PATH2,
1141
+ sourceFilePath: snapshot.dbPath
1142
+ }
1143
+ ];
1144
+ const commitOid = writeDeps.writeTreeCommit(snapshot.repoPath, snapshot.baseOid, updates, input.message ?? `chore(memory): ${input.event.eventType} ${input.event.canonicalKey}`);
1145
+ try {
1146
+ await Promise.resolve(writeDeps.pushRefWithLease(snapshot.repoPath, commitOid, "refs/heads/main", snapshot.baseOid));
1147
+ return {
1148
+ outcome: "applied",
1149
+ baseOid: snapshot.baseOid,
1150
+ commitOid
1151
+ };
1152
+ } catch (error) {
1153
+ lastError = error;
1154
+ }
1155
+ } finally {
1156
+ await snapshot.cleanup();
1157
+ }
1158
+ }
1159
+ throw lastError instanceof Error ? lastError : new Error(String(lastError ?? "memory promotion failed"));
1160
+ }
1161
+
1162
+ // packages/memory-plugin/src/cli.ts
1163
+ import { NO_MATCH_RETRIEVAL_CANONICAL_KEY as NO_MATCH_RETRIEVAL_CANONICAL_KEY2 } from "@rig/contracts";
1164
+ function takeOption(args, option) {
1165
+ const rest = [];
1166
+ let value;
1167
+ for (let index = 0;index < args.length; index += 1) {
1168
+ const current = args[index];
1169
+ if (current === option) {
1170
+ if (value !== undefined) {
1171
+ throw new Error(`Duplicate option ${option}`);
1172
+ }
1173
+ const next = args[index + 1];
1174
+ if (!next || next.startsWith("-")) {
1175
+ throw new Error(`Missing value for ${option}`);
1176
+ }
1177
+ value = next;
1178
+ index += 1;
1179
+ continue;
1180
+ }
1181
+ if (current !== undefined) {
1182
+ rest.push(current);
1183
+ }
1184
+ }
1185
+ return { value, rest };
1186
+ }
1187
+ function usage(verb) {
1188
+ switch (verb) {
1189
+ case "observe":
1190
+ return "Usage: rig-agent memory observe <canonical-key> <summary> [--kind <kind>] [--category <category>] [--confidence <number>]";
1191
+ case "supersede":
1192
+ return "Usage: rig-agent memory supersede <canonical-key> --replacement <replacement-key>";
1193
+ case "retract":
1194
+ return "Usage: rig-agent memory retract <canonical-key>";
1195
+ case "reinforce":
1196
+ return "Usage: rig-agent memory reinforce <canonical-key> --confidence <number>";
1197
+ case "recall":
1198
+ return "Usage: rig-agent memory recall [query]";
1199
+ default:
1200
+ return [
1201
+ "Usage:",
1202
+ " rig-agent memory observe <canonical-key> <summary> [--kind <kind>] [--category <category>] [--confidence <number>]",
1203
+ " rig-agent memory supersede <canonical-key> --replacement <replacement-key>",
1204
+ " rig-agent memory retract <canonical-key>",
1205
+ " rig-agent memory reinforce <canonical-key> --confidence <number>",
1206
+ " rig-agent memory recall [query]"
1207
+ ].join(`
1208
+ `);
1209
+ }
1210
+ }
1211
+ function parseConfidence(raw, fallback) {
1212
+ if (!raw) {
1213
+ return fallback;
1214
+ }
1215
+ const value = Number(raw);
1216
+ if (!Number.isFinite(value) || value < 0 || value > 1) {
1217
+ throw new Error("Confidence must be a number between 0 and 1.");
1218
+ }
1219
+ return value;
1220
+ }
1221
+ function currentBranch(projectRoot) {
1222
+ const proc = Bun.spawnSync(["git", "rev-parse", "--abbrev-ref", "HEAD"], {
1223
+ cwd: projectRoot,
1224
+ stdout: "pipe",
1225
+ stderr: "pipe"
1226
+ });
1227
+ return proc.exitCode === 0 ? proc.stdout.toString().trim() || "main" : "main";
1228
+ }
1229
+ function requireRuntimeMemoryContext(runtimeContext) {
1230
+ if (!runtimeContext?.memory) {
1231
+ throw new Error("Shared memory is unavailable without a hydrated runtime-context.json.");
1232
+ }
1233
+ return runtimeContext.memory;
1234
+ }
1235
+ async function ensureRuntimeMemoryUsable(runtimeContext) {
1236
+ const memory = requireRuntimeMemoryContext(runtimeContext);
1237
+ if (!existsSync(memory.hydratedPath)) {
1238
+ throw new Error(`Shared memory database is missing: ${memory.hydratedPath}`);
1239
+ }
1240
+ const db = await openMemoryDb(memory.hydratedPath);
1241
+ await db.close();
1242
+ }
1243
+ async function syncRuntimeMemory(runtimeContext, event) {
1244
+ const memory = requireRuntimeMemoryContext(runtimeContext);
1245
+ const db = await openMemoryDb(memory.hydratedPath);
1246
+ try {
1247
+ if (await hasMemoryEvent(db, event.eventId)) {
1248
+ return;
1249
+ }
1250
+ await applyMemoryEvent(db, event);
1251
+ if (memoryEventRequiresEmbedding(event)) {
1252
+ const embedder = maybeCreateConfiguredMemoryEmbedder();
1253
+ if (!embedder) {
1254
+ return;
1255
+ }
1256
+ await embedChangedMemoryItems(db, event, embedder);
1257
+ }
1258
+ } finally {
1259
+ await db.close();
1260
+ }
1261
+ }
1262
+ async function lookupRelevantMemory(runtimeContext, query) {
1263
+ const memory = requireRuntimeMemoryContext(runtimeContext);
1264
+ const db = await openMemoryDb(memory.hydratedPath);
1265
+ try {
1266
+ return await queryRelevantMemory(db, {
1267
+ query,
1268
+ limit: memory.retrieval.topK,
1269
+ retrieval: memory.retrieval
1270
+ });
1271
+ } finally {
1272
+ await db.close();
1273
+ }
1274
+ }
1275
+ async function memorySummaryForKey(runtimeContext, canonicalKey) {
1276
+ const memory = requireRuntimeMemoryContext(runtimeContext);
1277
+ const db = await openMemoryDb(memory.hydratedPath);
1278
+ try {
1279
+ const item = await getMemoryItemByKey(db, canonicalKey);
1280
+ return item?.summary ?? null;
1281
+ } finally {
1282
+ await db.close();
1283
+ }
1284
+ }
1285
+ function formatRelatedMemorySection(results) {
1286
+ return formatMemoryQueryResults(results);
1287
+ }
1288
+ function defaultRecallQuery(options) {
1289
+ const role = options.runtimeContext?.role?.trim();
1290
+ const scopeTerms = (options.runtimeContext?.scopes ?? []).flatMap((scope) => scope.split("/").slice(-3)).map((term) => term.trim()).filter(Boolean);
1291
+ const validationTerms = (options.runtimeContext?.validation ?? []).map((command) => command.split(/\s+/)[0]?.trim() ?? "").filter(Boolean);
1292
+ const parts = [...new Set([role, ...scopeTerms, ...validationTerms].filter(Boolean))];
1293
+ return parts.join(" ").trim() || options.taskId;
1294
+ }
1295
+ function assertNoFlagTokens(args, usageText) {
1296
+ if (args.some((arg) => arg.startsWith("-"))) {
1297
+ throw new Error(usageText);
1298
+ }
1299
+ }
1300
+ function baseEventFields(options, branch) {
1301
+ return {
1302
+ eventId: `memory-${randomUUID()}`,
1303
+ sourceRunId: options.runtimeContext?.runtimeId || options.taskId,
1304
+ sourceTaskId: options.taskId,
1305
+ branch,
1306
+ createdAt: new Date().toISOString()
1307
+ };
1308
+ }
1309
+ function requestedPayload(verb, mutationKind, canonicalKey, extra) {
1310
+ return {
1311
+ command: verb,
1312
+ mutationKind,
1313
+ canonicalKey,
1314
+ ...extra
1315
+ };
1316
+ }
1317
+ function mutationKindForVerb(verb) {
1318
+ switch (verb) {
1319
+ case "observe":
1320
+ return "observed";
1321
+ case "supersede":
1322
+ return "superseded";
1323
+ case "retract":
1324
+ return "retracted";
1325
+ case "reinforce":
1326
+ return "reinforced";
1327
+ case "recall":
1328
+ return "retrieved";
1329
+ }
1330
+ }
1331
+ var DEFAULT_EXECUTE_MEMORY_COMMAND_DEPS = {
1332
+ currentBranch,
1333
+ ensureRuntimeMemoryUsable,
1334
+ syncRuntimeMemory,
1335
+ lookupRelevantMemory,
1336
+ memorySummaryForKey,
1337
+ promoteCanonicalMemoryEvent
1338
+ };
1339
+ async function executeMemoryCommand(options, deps = {}) {
1340
+ const [verbRaw, ...rest] = options.args;
1341
+ const verb = verbRaw;
1342
+ if (!verb || !["observe", "supersede", "retract", "reinforce", "recall"].includes(verb)) {
1343
+ throw new Error(usage());
1344
+ }
1345
+ const commandDeps = { ...DEFAULT_EXECUTE_MEMORY_COMMAND_DEPS, ...deps };
1346
+ let requested;
1347
+ try {
1348
+ switch (verb) {
1349
+ case "observe": {
1350
+ const kind = takeOption(rest, "--kind");
1351
+ const category = takeOption(kind.rest, "--category");
1352
+ const confidence = takeOption(category.rest, "--confidence");
1353
+ assertNoFlagTokens(confidence.rest, usage("observe"));
1354
+ const [canonicalKey, summary] = confidence.rest;
1355
+ if (!canonicalKey || !summary || confidence.rest.length !== 2) {
1356
+ throw new Error(usage("observe"));
1357
+ }
1358
+ await commandDeps.ensureRuntimeMemoryUsable(options.runtimeContext);
1359
+ const event = {
1360
+ ...baseEventFields(options, commandDeps.currentBranch(options.projectRoot)),
1361
+ eventType: "observed",
1362
+ canonicalKey,
1363
+ summary,
1364
+ kind: kind.value || "fact",
1365
+ category: category.value ?? null,
1366
+ confidence: parseConfidence(confidence.value, 0.7)
1367
+ };
1368
+ requested = requestedPayload("observe", "observed", canonicalKey);
1369
+ await options.eventBus?.emit("memory.command.requested", requested);
1370
+ await commandDeps.promoteCanonicalMemoryEvent(options.projectRoot, { event });
1371
+ await commandDeps.syncRuntimeMemory(options.runtimeContext, event);
1372
+ const relatedResults = await commandDeps.lookupRelevantMemory(options.runtimeContext, summary);
1373
+ const output = [`Observed memory ${canonicalKey}.`, "", "Related shared memories:", formatRelatedMemorySection(relatedResults)].join(`
1374
+ `);
1375
+ await options.eventBus?.emit("memory.command.applied", {
1376
+ ...requested,
1377
+ relatedCanonicalKeys: relatedResults.map((result) => result.canonicalKey)
1378
+ });
1379
+ return output;
1380
+ }
1381
+ case "supersede": {
1382
+ const replacement = takeOption(rest, "--replacement");
1383
+ assertNoFlagTokens(replacement.rest, usage("supersede"));
1384
+ const canonicalKey = replacement.rest[0];
1385
+ if (!canonicalKey || !replacement.value || replacement.rest.length !== 1) {
1386
+ throw new Error(usage("supersede"));
1387
+ }
1388
+ await commandDeps.ensureRuntimeMemoryUsable(options.runtimeContext);
1389
+ const event = {
1390
+ ...baseEventFields(options, commandDeps.currentBranch(options.projectRoot)),
1391
+ eventType: "superseded",
1392
+ canonicalKey,
1393
+ replacementCanonicalKey: replacement.value
1394
+ };
1395
+ requested = requestedPayload("supersede", "superseded", canonicalKey, { replacementCanonicalKey: replacement.value });
1396
+ await options.eventBus?.emit("memory.command.requested", requested);
1397
+ const relatedQuery = await commandDeps.memorySummaryForKey(options.runtimeContext, replacement.value) ?? replacement.value;
1398
+ await commandDeps.promoteCanonicalMemoryEvent(options.projectRoot, { event });
1399
+ await commandDeps.syncRuntimeMemory(options.runtimeContext, event);
1400
+ const relatedResults = await commandDeps.lookupRelevantMemory(options.runtimeContext, relatedQuery);
1401
+ const output = [`Superseded memory ${canonicalKey}.`, "", "Related shared memories:", formatRelatedMemorySection(relatedResults)].join(`
1402
+ `);
1403
+ await options.eventBus?.emit("memory.command.applied", {
1404
+ ...requested,
1405
+ relatedCanonicalKeys: relatedResults.map((result) => result.canonicalKey)
1406
+ });
1407
+ return output;
1408
+ }
1409
+ case "retract": {
1410
+ const canonicalKey = rest[0];
1411
+ assertNoFlagTokens(rest, usage("retract"));
1412
+ if (!canonicalKey || rest.length !== 1) {
1413
+ throw new Error(usage("retract"));
1414
+ }
1415
+ await commandDeps.ensureRuntimeMemoryUsable(options.runtimeContext);
1416
+ const event = {
1417
+ ...baseEventFields(options, commandDeps.currentBranch(options.projectRoot)),
1418
+ eventType: "retracted",
1419
+ canonicalKey
1420
+ };
1421
+ requested = requestedPayload("retract", "retracted", canonicalKey);
1422
+ await options.eventBus?.emit("memory.command.requested", requested);
1423
+ const relatedQuery = await commandDeps.memorySummaryForKey(options.runtimeContext, canonicalKey) ?? canonicalKey;
1424
+ await commandDeps.promoteCanonicalMemoryEvent(options.projectRoot, { event });
1425
+ await commandDeps.syncRuntimeMemory(options.runtimeContext, event);
1426
+ const relatedResults = await commandDeps.lookupRelevantMemory(options.runtimeContext, relatedQuery);
1427
+ const output = [`Retracted memory ${canonicalKey}.`, "", "Related shared memories:", formatRelatedMemorySection(relatedResults)].join(`
1428
+ `);
1429
+ await options.eventBus?.emit("memory.command.applied", {
1430
+ ...requested,
1431
+ relatedCanonicalKeys: relatedResults.map((result) => result.canonicalKey)
1432
+ });
1433
+ return output;
1434
+ }
1435
+ case "reinforce": {
1436
+ const confidence = takeOption(rest, "--confidence");
1437
+ assertNoFlagTokens(confidence.rest, usage("reinforce"));
1438
+ const canonicalKey = confidence.rest[0];
1439
+ if (!canonicalKey || !confidence.value || confidence.rest.length !== 1) {
1440
+ throw new Error(usage("reinforce"));
1441
+ }
1442
+ await commandDeps.ensureRuntimeMemoryUsable(options.runtimeContext);
1443
+ const event = {
1444
+ ...baseEventFields(options, commandDeps.currentBranch(options.projectRoot)),
1445
+ eventType: "reinforced",
1446
+ canonicalKey,
1447
+ confidence: parseConfidence(confidence.value, 0.7)
1448
+ };
1449
+ requested = requestedPayload("reinforce", "reinforced", canonicalKey);
1450
+ await options.eventBus?.emit("memory.command.requested", requested);
1451
+ const relatedQuery = await commandDeps.memorySummaryForKey(options.runtimeContext, canonicalKey) ?? canonicalKey;
1452
+ await commandDeps.promoteCanonicalMemoryEvent(options.projectRoot, { event });
1453
+ await commandDeps.syncRuntimeMemory(options.runtimeContext, event);
1454
+ const relatedResults = await commandDeps.lookupRelevantMemory(options.runtimeContext, relatedQuery);
1455
+ const output = [`Reinforced memory ${canonicalKey}.`, "", "Related shared memories:", formatRelatedMemorySection(relatedResults)].join(`
1456
+ `);
1457
+ await options.eventBus?.emit("memory.command.applied", {
1458
+ ...requested,
1459
+ relatedCanonicalKeys: relatedResults.map((result) => result.canonicalKey)
1460
+ });
1461
+ return output;
1462
+ }
1463
+ case "recall": {
1464
+ if (rest.some((arg) => arg.startsWith("--"))) {
1465
+ throw new Error(usage("recall"));
1466
+ }
1467
+ const query = rest.join(" ").trim() || defaultRecallQuery(options);
1468
+ await commandDeps.ensureRuntimeMemoryUsable(options.runtimeContext);
1469
+ requested = requestedPayload("recall", "retrieved", undefined, { query });
1470
+ await options.eventBus?.emit("memory.command.requested", requested);
1471
+ const relatedResults = await commandDeps.lookupRelevantMemory(options.runtimeContext, query);
1472
+ const topResult = relatedResults[0];
1473
+ const retrievedEvent = {
1474
+ ...baseEventFields(options, commandDeps.currentBranch(options.projectRoot)),
1475
+ eventType: "retrieved",
1476
+ canonicalKey: topResult?.canonicalKey ?? NO_MATCH_RETRIEVAL_CANONICAL_KEY2,
1477
+ retrievalQuery: query,
1478
+ retrievalRank: topResult ? 1 : 0
1479
+ };
1480
+ await commandDeps.promoteCanonicalMemoryEvent(options.projectRoot, { event: retrievedEvent });
1481
+ await commandDeps.syncRuntimeMemory(options.runtimeContext, retrievedEvent);
1482
+ await options.eventBus?.emit("memory.command.applied", {
1483
+ ...requested,
1484
+ retrievedCanonicalKeys: relatedResults.map((result) => result.canonicalKey)
1485
+ });
1486
+ return formatRelatedMemorySection(relatedResults);
1487
+ }
1488
+ }
1489
+ } catch (error) {
1490
+ const fallbackRequested = requested ?? requestedPayload(verb, mutationKindForVerb(verb), ["observe", "supersede", "retract", "reinforce"].includes(verb) ? rest[0] : undefined);
1491
+ await options.eventBus?.emit("memory.command.failed", {
1492
+ ...fallbackRequested,
1493
+ error: error instanceof Error ? error.message : String(error)
1494
+ });
1495
+ throw error;
1496
+ }
1497
+ }
1498
+ export {
1499
+ executeMemoryCommand
1500
+ };