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