audrey 0.14.0 → 0.16.0

package/src/rollback.js

@@ -1,42 +1,42 @@
-import { safeJsonParse } from './utils.js';
-
-/**
- * @param {import('better-sqlite3').Database} db
- * @returns {Array<{ id: string, checkpoint_cursor: string|null, input_episode_ids: string, output_memory_ids: string, started_at: string, completed_at: string|null, status: string }>}
- */
-export function getConsolidationHistory(db) {
-  return db.prepare(`
-    SELECT id, checkpoint_cursor, input_episode_ids, output_memory_ids,
-           started_at, completed_at, status
-    FROM consolidation_runs ORDER BY started_at DESC
-  `).all();
-}
-
-/**
- * @param {import('better-sqlite3').Database} db
- * @param {string} runId
- * @returns {{ rolledBackMemories: number, restoredEpisodes: number }}
- */
-export function rollbackConsolidation(db, runId) {
-  const run = db.prepare('SELECT * FROM consolidation_runs WHERE id = ?').get(runId);
-  if (!run) throw new Error(`Consolidation run not found: ${runId}`);
-  if (run.status === 'rolled_back') throw new Error(`Run already rolled back: ${runId}`);
-
-  const outputIds = safeJsonParse(run.output_memory_ids, []);
-  const inputIds = safeJsonParse(run.input_episode_ids, []);
-
-  const doRollback = db.transaction(() => {
-    const markSemantics = db.prepare('UPDATE semantics SET state = ? WHERE id = ?');
-    const markProcedures = db.prepare('UPDATE procedures SET state = ? WHERE id = ?');
-    for (const id of outputIds) {
-      markSemantics.run('rolled_back', id);
-      markProcedures.run('rolled_back', id);
-    }
-    const unmark = db.prepare('UPDATE episodes SET consolidated = 0 WHERE id = ?');
-    for (const id of inputIds) { unmark.run(id); }
-    db.prepare('UPDATE consolidation_runs SET status = ? WHERE id = ?').run('rolled_back', runId);
-  });
-
-  doRollback();
-  return { rolledBackMemories: outputIds.length, restoredEpisodes: inputIds.length };
-}
+import { safeJsonParse } from './utils.js';
+
+/**
+ * @param {import('better-sqlite3').Database} db
+ * @returns {Array<{ id: string, checkpoint_cursor: string|null, input_episode_ids: string, output_memory_ids: string, started_at: string, completed_at: string|null, status: string }>}
+ */
+export function getConsolidationHistory(db) {
+  return db.prepare(`
+    SELECT id, checkpoint_cursor, input_episode_ids, output_memory_ids,
+           started_at, completed_at, status
+    FROM consolidation_runs ORDER BY started_at DESC
+  `).all();
+}
+
+/**
+ * @param {import('better-sqlite3').Database} db
+ * @param {string} runId
+ * @returns {{ rolledBackMemories: number, restoredEpisodes: number }}
+ */
+export function rollbackConsolidation(db, runId) {
+  const run = db.prepare('SELECT * FROM consolidation_runs WHERE id = ?').get(runId);
+  if (!run) throw new Error(`Consolidation run not found: ${runId}`);
+  if (run.status === 'rolled_back') throw new Error(`Run already rolled back: ${runId}`);
+
+  const outputIds = safeJsonParse(run.output_memory_ids, []);
+  const inputIds = safeJsonParse(run.input_episode_ids, []);
+
+  const doRollback = db.transaction(() => {
+    const markSemantics = db.prepare('UPDATE semantics SET state = ? WHERE id = ?');
+    const markProcedures = db.prepare('UPDATE procedures SET state = ? WHERE id = ?');
+    for (const id of outputIds) {
+      markSemantics.run('rolled_back', id);
+      markProcedures.run('rolled_back', id);
+    }
+    const unmark = db.prepare('UPDATE episodes SET consolidated = 0 WHERE id = ?');
+    for (const id of inputIds) { unmark.run(id); }
+    db.prepare('UPDATE consolidation_runs SET status = ? WHERE id = ?').run('rolled_back', runId);
+  });
+
+  doRollback();
+  return { rolledBackMemories: outputIds.length, restoredEpisodes: inputIds.length };
+}

package/src/ulid.js

@@ -1,18 +1,18 @@
-import { monotonicFactory } from 'ulid';
-import { createHash } from 'node:crypto';
-
-const monotonic = monotonicFactory();
-
-/** @returns {string} */
-export function generateId() {
-  return monotonic();
-}
-
-/**
- * @param {...*} parts
- * @returns {string}
- */
-export function generateDeterministicId(...parts) {
-  const input = JSON.stringify(parts);
-  return createHash('sha256').update(input).digest('hex').slice(0, 26);
-}
+import { monotonicFactory } from 'ulid';
+import { createHash } from 'node:crypto';
+
+const monotonic = monotonicFactory();
+
+/** @returns {string} */
+export function generateId() {
+  return monotonic();
+}
+
+/**
+ * @param {...*} parts
+ * @returns {string}
+ */
+export function generateDeterministicId(...parts) {
+  const input = JSON.stringify(parts);
+  return createHash('sha256').update(input).digest('hex').slice(0, 26);
+}

package/src/utils.js

@@ -1,38 +1,38 @@
-/**
- * @param {Buffer} bufA
- * @param {Buffer} bufB
- * @param {import('./embedding.js').EmbeddingProvider} provider
- * @returns {number}
- */
-export function cosineSimilarity(bufA, bufB, provider) {
-  const a = provider.bufferToVector(bufA);
-  const b = provider.bufferToVector(bufB);
-  let dot = 0, magA = 0, magB = 0;
-  for (let i = 0; i < a.length; i++) {
-    dot += a[i] * b[i];
-    magA += a[i] * a[i];
-    magB += b[i] * b[i];
-  }
-  const mag = Math.sqrt(magA) * Math.sqrt(magB);
-  return mag === 0 ? 0 : dot / mag;
-}
-
-/**
- * @param {string} dateStr
- * @param {Date} now
- * @returns {number}
- */
-export function daysBetween(dateStr, now) {
-  return Math.max(0, (now.getTime() - new Date(dateStr).getTime()) / (1000 * 60 * 60 * 24));
-}
-
-/**
- * @param {string | null | undefined} str
- * @param {*} [fallback=null]
- * @returns {*}
- */
-export function safeJsonParse(str, fallback = null) {
-  if (!str) return fallback;
-  try { return JSON.parse(str); }
-  catch { return fallback; }
-}
+/**
+ * @param {Buffer} bufA
+ * @param {Buffer} bufB
+ * @param {import('./embedding.js').EmbeddingProvider} provider
+ * @returns {number}
+ */
+export function cosineSimilarity(bufA, bufB, provider) {
+  const a = provider.bufferToVector(bufA);
+  const b = provider.bufferToVector(bufB);
+  let dot = 0, magA = 0, magB = 0;
+  for (let i = 0; i < a.length; i++) {
+    dot += a[i] * b[i];
+    magA += a[i] * a[i];
+    magB += b[i] * b[i];
+  }
+  const mag = Math.sqrt(magA) * Math.sqrt(magB);
+  return mag === 0 ? 0 : dot / mag;
+}
+
+/**
+ * @param {string} dateStr
+ * @param {Date} now
+ * @returns {number}
+ */
+export function daysBetween(dateStr, now) {
+  return Math.max(0, (now.getTime() - new Date(dateStr).getTime()) / (1000 * 60 * 60 * 24));
+}
+
+/**
+ * @param {string | null | undefined} str
+ * @param {*} [fallback=null]
+ * @returns {*}
+ */
+export function safeJsonParse(str, fallback = null) {
+  if (!str) return fallback;
+  try { return JSON.parse(str); }
+  catch { return fallback; }
+}

package/src/validate.js

@@ -1,172 +1,172 @@
-import { generateId } from './ulid.js';
-import { safeJsonParse } from './utils.js';
-import { buildContradictionDetectionPrompt } from './prompts.js';
-
-const REINFORCEMENT_THRESHOLD = 0.85;
-const CONTRADICTION_THRESHOLD = 0.60;
-
-/**
- * @param {import('better-sqlite3').Database} db
- * @param {import('./embedding.js').EmbeddingProvider} embeddingProvider
- * @param {{ id: string, content: string, source: string }} episode
- * @param {{ threshold?: number, contradictionThreshold?: number, llmProvider?: { json: (messages: any) => Promise<any> } }} [options]
- * @returns {Promise<{ action: string, semanticId?: string, similarity?: number, contradictionId?: string, resolution?: string }>}
- */
-export async function validateMemory(db, embeddingProvider, episode, options = {}) {
-  const {
-    threshold = REINFORCEMENT_THRESHOLD,
-    contradictionThreshold = CONTRADICTION_THRESHOLD,
-    llmProvider,
-  } = options;
-
-  const episodeVector = await embeddingProvider.embed(episode.content);
-  const episodeBuffer = embeddingProvider.vectorToBuffer(episodeVector);
-
-  const nearestSemantic = db.prepare(`
-    SELECT s.*, (1.0 - v.distance) AS similarity
-    FROM vec_semantics v
-    JOIN semantics s ON s.id = v.id
-    WHERE v.embedding MATCH ?
-      AND k = 1
-      AND (v.state = 'active' OR v.state = 'context_dependent')
-  `).get(episodeBuffer);
-
-  let bestMatch = null;
-  let bestSimilarity = 0;
-
-  if (nearestSemantic) {
-    bestMatch = nearestSemantic;
-    bestSimilarity = nearestSemantic.similarity;
-  }
-
-  if (bestMatch && bestSimilarity >= threshold) {
-    const evidenceIds = safeJsonParse(bestMatch.evidence_episode_ids, []);
-    if (!evidenceIds.includes(episode.id)) {
-      evidenceIds.push(episode.id);
-    }
-
-    const diversity = computeSourceDiversity(db, evidenceIds, episode);
-
-    const now = new Date().toISOString();
-    db.prepare(`
-      UPDATE semantics SET
-        supporting_count = supporting_count + 1,
-        evidence_episode_ids = ?,
-        evidence_count = ?,
-        source_type_diversity = ?,
-        last_reinforced_at = ?
-      WHERE id = ?
-    `).run(
-      JSON.stringify(evidenceIds),
-      evidenceIds.length,
-      diversity,
-      now,
-      bestMatch.id,
-    );
-
-    return {
-      action: 'reinforced',
-      semanticId: bestMatch.id,
-      similarity: bestSimilarity,
-    };
-  }
-
-  if (bestMatch && bestSimilarity >= contradictionThreshold && llmProvider) {
-    const messages = buildContradictionDetectionPrompt(episode.content, bestMatch.content);
-    const verdict = await llmProvider.json(messages);
-
-    if (verdict.contradicts) {
-      const resolution = verdict.resolution === 'context_dependent'
-        ? { type: 'context_dependent', conditions: verdict.conditions, explanation: verdict.explanation }
-        : verdict.resolution
-          ? { type: verdict.resolution, explanation: verdict.explanation }
-          : null;
-
-      const contradictionId = createContradiction(
-        db,
-        bestMatch.id,
-        'semantic',
-        episode.id,
-        'episodic',
-        resolution,
-      );
-
-      if (verdict.resolution === 'new_wins') {
-        db.prepare("UPDATE semantics SET state = 'disputed' WHERE id = ?").run(bestMatch.id);
-      } else if (verdict.resolution === 'context_dependent' && verdict.conditions) {
-        db.prepare("UPDATE semantics SET state = 'context_dependent', conditions = ? WHERE id = ?")
-          .run(JSON.stringify(verdict.conditions), bestMatch.id);
-      }
-
-      return {
-        action: 'contradiction',
-        contradictionId,
-        semanticId: bestMatch.id,
-        similarity: bestSimilarity,
-        resolution: verdict.resolution || null,
-      };
-    }
-  }
-
-  return { action: 'none' };
-}
-
-function computeSourceDiversity(db, evidenceIds, currentEpisode) {
-  const sourceTypes = new Set();
-  sourceTypes.add(currentEpisode.source);
-
-  if (evidenceIds.length > 0) {
-    const placeholders = evidenceIds.map(() => '?').join(',');
-    const rows = db.prepare(
-      `SELECT DISTINCT source FROM episodes WHERE id IN (${placeholders})`
-    ).all(...evidenceIds);
-    for (const row of rows) {
-      sourceTypes.add(row.source);
-    }
-  }
-
-  return sourceTypes.size;
-}
-
-/**
- * @param {import('better-sqlite3').Database} db
- * @param {string} claimAId
- * @param {string} claimAType
- * @param {string} claimBId
- * @param {string} claimBType
- * @param {object|null} resolution
- * @returns {string}
- */
-export function createContradiction(db, claimAId, claimAType, claimBId, claimBType, resolution) {
-  const id = generateId();
-  const now = new Date().toISOString();
-
-  const state = resolution ? 'resolved' : 'open';
-  const resolvedAt = resolution ? now : null;
-  const resolutionJson = resolution ? JSON.stringify(resolution) : null;
-
-  db.prepare(`
-    INSERT INTO contradictions (id, claim_a_id, claim_a_type, claim_b_id, claim_b_type,
-      state, resolution, resolved_at, created_at)
-    VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)
-  `).run(id, claimAId, claimAType, claimBId, claimBType, state, resolutionJson, resolvedAt, now);
-
-  return id;
-}
-
-/**
- * @param {import('better-sqlite3').Database} db
- * @param {string} contradictionId
- * @param {string} newEvidenceId
- * @returns {void}
- */
-export function reopenContradiction(db, contradictionId, newEvidenceId) {
-  const now = new Date().toISOString();
-  db.prepare(`
-    UPDATE contradictions SET
-      state = 'reopened',
-      reopen_evidence_id = ?,
-      reopened_at = ?
-    WHERE id = ?
-  `).run(newEvidenceId, now, contradictionId);
-}
+import { generateId } from './ulid.js';
+import { safeJsonParse } from './utils.js';
+import { buildContradictionDetectionPrompt } from './prompts.js';
+
+const REINFORCEMENT_THRESHOLD = 0.85;
+const CONTRADICTION_THRESHOLD = 0.60;
+
+/**
+ * @param {import('better-sqlite3').Database} db
+ * @param {import('./embedding.js').EmbeddingProvider} embeddingProvider
+ * @param {{ id: string, content: string, source: string }} episode
+ * @param {{ threshold?: number, contradictionThreshold?: number, llmProvider?: { json: (messages: any) => Promise<any> } }} [options]
+ * @returns {Promise<{ action: string, semanticId?: string, similarity?: number, contradictionId?: string, resolution?: string }>}
+ */
+export async function validateMemory(db, embeddingProvider, episode, options = {}) {
+  const {
+    threshold = REINFORCEMENT_THRESHOLD,
+    contradictionThreshold = CONTRADICTION_THRESHOLD,
+    llmProvider,
+  } = options;
+
+  const episodeVector = await embeddingProvider.embed(episode.content);
+  const episodeBuffer = embeddingProvider.vectorToBuffer(episodeVector);
+
+  const nearestSemantic = db.prepare(`
+    SELECT s.*, (1.0 - v.distance) AS similarity
+    FROM vec_semantics v
+    JOIN semantics s ON s.id = v.id
+    WHERE v.embedding MATCH ?
+      AND k = 1
+      AND (v.state = 'active' OR v.state = 'context_dependent')
+  `).get(episodeBuffer);
+
+  let bestMatch = null;
+  let bestSimilarity = 0;
+
+  if (nearestSemantic) {
+    bestMatch = nearestSemantic;
+    bestSimilarity = nearestSemantic.similarity;
+  }
+
+  if (bestMatch && bestSimilarity >= threshold) {
+    const evidenceIds = safeJsonParse(bestMatch.evidence_episode_ids, []);
+    if (!evidenceIds.includes(episode.id)) {
+      evidenceIds.push(episode.id);
+    }
+
+    const diversity = computeSourceDiversity(db, evidenceIds, episode);
+
+    const now = new Date().toISOString();
+    db.prepare(`
+      UPDATE semantics SET
+        supporting_count = supporting_count + 1,
+        evidence_episode_ids = ?,
+        evidence_count = ?,
+        source_type_diversity = ?,
+        last_reinforced_at = ?
+      WHERE id = ?
+    `).run(
+      JSON.stringify(evidenceIds),
+      evidenceIds.length,
+      diversity,
+      now,
+      bestMatch.id,
+    );
+
+    return {
+      action: 'reinforced',
+      semanticId: bestMatch.id,
+      similarity: bestSimilarity,
+    };
+  }
+
+  if (bestMatch && bestSimilarity >= contradictionThreshold && llmProvider) {
+    const messages = buildContradictionDetectionPrompt(episode.content, bestMatch.content);
+    const verdict = await llmProvider.json(messages);
+
+    if (verdict.contradicts) {
+      const resolution = verdict.resolution === 'context_dependent'
+        ? { type: 'context_dependent', conditions: verdict.conditions, explanation: verdict.explanation }
+        : verdict.resolution
+          ? { type: verdict.resolution, explanation: verdict.explanation }
+          : null;
+
+      const contradictionId = createContradiction(
+        db,
+        bestMatch.id,
+        'semantic',
+        episode.id,
+        'episodic',
+        resolution,
+      );
+
+      if (verdict.resolution === 'new_wins') {
+        db.prepare("UPDATE semantics SET state = 'disputed' WHERE id = ?").run(bestMatch.id);
+      } else if (verdict.resolution === 'context_dependent' && verdict.conditions) {
+        db.prepare("UPDATE semantics SET state = 'context_dependent', conditions = ? WHERE id = ?")
+          .run(JSON.stringify(verdict.conditions), bestMatch.id);
+      }
+
+      return {
+        action: 'contradiction',
+        contradictionId,
+        semanticId: bestMatch.id,
+        similarity: bestSimilarity,
+        resolution: verdict.resolution || null,
+      };
+    }
+  }
+
+  return { action: 'none' };
+}
+
+function computeSourceDiversity(db, evidenceIds, currentEpisode) {
+  const sourceTypes = new Set();
+  sourceTypes.add(currentEpisode.source);
+
+  if (evidenceIds.length > 0) {
+    const placeholders = evidenceIds.map(() => '?').join(',');
+    const rows = db.prepare(
+      `SELECT DISTINCT source FROM episodes WHERE id IN (${placeholders})`
+    ).all(...evidenceIds);
+    for (const row of rows) {
+      sourceTypes.add(row.source);
+    }
+  }
+
+  return sourceTypes.size;
+}
+
+/**
+ * @param {import('better-sqlite3').Database} db
+ * @param {string} claimAId
+ * @param {string} claimAType
+ * @param {string} claimBId
+ * @param {string} claimBType
+ * @param {object|null} resolution
+ * @returns {string}
+ */
+export function createContradiction(db, claimAId, claimAType, claimBId, claimBType, resolution) {
+  const id = generateId();
+  const now = new Date().toISOString();
+
+  const state = resolution ? 'resolved' : 'open';
+  const resolvedAt = resolution ? now : null;
+  const resolutionJson = resolution ? JSON.stringify(resolution) : null;
+
+  db.prepare(`
+    INSERT INTO contradictions (id, claim_a_id, claim_a_type, claim_b_id, claim_b_type,
+      state, resolution, resolved_at, created_at)
+    VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)
+  `).run(id, claimAId, claimAType, claimBId, claimBType, state, resolutionJson, resolvedAt, now);
+
+  return id;
+}
+
+/**
+ * @param {import('better-sqlite3').Database} db
+ * @param {string} contradictionId
+ * @param {string} newEvidenceId
+ * @returns {void}
+ */
+export function reopenContradiction(db, contradictionId, newEvidenceId) {
+  const now = new Date().toISOString();
+  db.prepare(`
+    UPDATE contradictions SET
+      state = 'reopened',
+      reopen_evidence_id = ?,
+      reopened_at = ?
+    WHERE id = ?
+  `).run(newEvidenceId, now, contradictionId);
+}