agent-working-memory 0.5.0 → 0.5.2

package/src/storage/sqlite.ts

@@ -197,9 +197,15 @@ export class EngramStore {
         checkpoint_at TEXT,
         last_consolidation_at TEXT,
         last_mini_consolidation_at TEXT,
+        consolidation_cycle_count INTEGER NOT NULL DEFAULT 0,
         updated_at TEXT NOT NULL DEFAULT (datetime('now'))
       )
     `);
+
+    // Migration: add consolidation_cycle_count if missing (existing DBs)
+    try {
+      this.db.exec(`ALTER TABLE conscious_state ADD COLUMN consolidation_cycle_count INTEGER NOT NULL DEFAULT 0`);
+    } catch { /* column already exists */ }
   }
 
   // --- Engram CRUD ---
@@ -926,6 +932,7 @@ export class EngramStore {
           last_mini_consolidation_at = ?,
           write_count_since_consolidation = 0,
           recall_count_since_consolidation = 0,
+          consolidation_cycle_count = consolidation_cycle_count + 1,
           updated_at = ?
         WHERE agent_id = ?
       `).run(now, now, now, agentId);
@@ -943,6 +950,13 @@ export class EngramStore {
     }));
   }
 
+  getConsolidationCycleCount(agentId: string): number {
+    const row = this.db.prepare(
+      'SELECT consolidation_cycle_count FROM conscious_state WHERE agent_id = ?',
+    ).get(agentId) as { consolidation_cycle_count: number } | undefined;
+    return row?.consolidation_cycle_count ?? 0;
+  }
+
   close(): void {
     this.db.close();
   }