@vellumai/assistant 0.10.0 → 0.10.1-staging.1

package/src/memory/migrations/217-conversation-host-access.ts

@@ -1,8 +1,5 @@
 import type { DrizzleDb } from "../db-connection.js";
 import { getSqliteFrom } from "../db-connection.js";
-import { withCrashRecovery } from "./validate-migration-state.js";
-
-const CHECKPOINT_KEY = "migration_conversation_host_access_v1";
 
 /**
  * Add conversation-scoped host access state with a safe default of disabled.
@@ -10,23 +7,21 @@ const CHECKPOINT_KEY = "migration_conversation_host_access_v1";
  * Idempotent: ALTER TABLE is guarded and the backfill only touches NULL rows.
  */
 export function migrateConversationHostAccess(database: DrizzleDb): void {
-  withCrashRecovery(database, CHECKPOINT_KEY, () => {
-    const raw = getSqliteFrom(database);
+  const raw = getSqliteFrom(database);
 
-    try {
-      raw.exec(
-        `ALTER TABLE conversations ADD COLUMN host_access INTEGER NOT NULL DEFAULT 0`,
-      );
-    } catch {
-      // Column already exists.
-    }
+  try {
+    raw.exec(
+      `ALTER TABLE conversations ADD COLUMN host_access INTEGER NOT NULL DEFAULT 0`,
+    );
+  } catch {
+    // Column already exists.
+  }
 
-    raw.exec(`
-      UPDATE conversations
-      SET host_access = 0
-      WHERE host_access IS NULL
-    `);
-  });
+  raw.exec(`
+    UPDATE conversations
+    SET host_access = 0
+    WHERE host_access IS NULL
+  `);
 }
 
 /**

package/src/memory/migrations/220-normalize-user-file-by-principal.ts

@@ -1,6 +1,5 @@
 import { computeUserFileBaseSlug } from "../../contacts/contact-store.js";
 import { type DrizzleDb, getSqliteFrom } from "../db-connection.js";
-import { withCrashRecovery } from "./validate-migration-state.js";
 
 /**
  * Reverse is a no-op. This migration only consolidates `user_file` across
@@ -85,106 +84,100 @@ export function isAutoIncrementedUserFile(
 export function migrateNormalizeUserFileByPrincipal(
   database: DrizzleDb,
 ): void {
-  withCrashRecovery(
-    database,
-    "migration_normalize_user_file_by_principal_v1",
-    () => {
-      const raw = getSqliteFrom(database);
+  const raw = getSqliteFrom(database);
 
-      const tableExists = raw
-        .query(
-          `SELECT 1 FROM sqlite_master WHERE type = 'table' AND name = 'contacts'`,
-        )
-        .get();
-      if (!tableExists) return;
+  const tableExists = raw
+    .query(
+      `SELECT 1 FROM sqlite_master WHERE type = 'table' AND name = 'contacts'`,
+    )
+    .get();
+  if (!tableExists) return;
 
-      const userFileColExists = raw
-        .query(
-          `SELECT 1 FROM pragma_table_info('contacts') WHERE name = 'user_file'`,
-        )
-        .get();
-      const principalColExists = raw
-        .query(
-          `SELECT 1 FROM pragma_table_info('contacts') WHERE name = 'principal_id'`,
-        )
-        .get();
-      if (!userFileColExists || !principalColExists) return;
+  const userFileColExists = raw
+    .query(
+      `SELECT 1 FROM pragma_table_info('contacts') WHERE name = 'user_file'`,
+    )
+    .get();
+  const principalColExists = raw
+    .query(
+      `SELECT 1 FROM pragma_table_info('contacts') WHERE name = 'principal_id'`,
+    )
+    .get();
+  if (!userFileColExists || !principalColExists) return;
 
-      try {
-        raw.exec("BEGIN");
+  try {
+    raw.exec("BEGIN");
 
-        const principals = raw
-          .query(
-            /*sql*/ `
-            SELECT principal_id
-            FROM contacts
-            WHERE principal_id IS NOT NULL
-            GROUP BY principal_id
-            HAVING COUNT(DISTINCT COALESCE(user_file, '')) > 1
-          `,
-          )
-          .all() as Array<{ principal_id: string }>;
+    const principals = raw
+      .query(
+        /*sql*/ `
+        SELECT principal_id
+        FROM contacts
+        WHERE principal_id IS NOT NULL
+        GROUP BY principal_id
+        HAVING COUNT(DISTINCT COALESCE(user_file, '')) > 1
+      `,
+      )
+      .all() as Array<{ principal_id: string }>;
 
-        // Fetch all non-null candidates and rank in JS. The auto-increment
-        // classification is a regex that SQLite's GLOB can't express cleanly
-        // (unbounded digit count, date-pattern exclusion), and keeping the
-        // logic in one place avoids SQL/JS drift.
-        const selectCandidates = raw.prepare(
-          /*sql*/ `
-          SELECT user_file, display_name, created_at, id FROM contacts
-          WHERE principal_id = ? AND user_file IS NOT NULL
-          `,
-        );
+    // Fetch all non-null candidates and rank in JS. The auto-increment
+    // classification is a regex that SQLite's GLOB can't express cleanly
+    // (unbounded digit count, date-pattern exclusion), and keeping the
+    // logic in one place avoids SQL/JS drift.
+    const selectCandidates = raw.prepare(
+      /*sql*/ `
+      SELECT user_file, display_name, created_at, id FROM contacts
+      WHERE principal_id = ? AND user_file IS NOT NULL
+      `,
+    );
 
-        const updateSiblings = raw.prepare(
-          /*sql*/ `
-          UPDATE contacts
-          SET user_file = ?, updated_at = ?
-          WHERE principal_id = ?
-            AND (user_file IS NULL OR user_file != ?)
-          `,
-        );
+    const updateSiblings = raw.prepare(
+      /*sql*/ `
+      UPDATE contacts
+      SET user_file = ?, updated_at = ?
+      WHERE principal_id = ?
+        AND (user_file IS NULL OR user_file != ?)
+      `,
+    );
 
-        for (const { principal_id } of principals) {
-          const candidates = selectCandidates.all(principal_id) as Array<{
-            user_file: string;
-            display_name: string;
-            created_at: number;
-            id: string;
-          }>;
-          if (candidates.length === 0) continue;
+    for (const { principal_id } of principals) {
+      const candidates = selectCandidates.all(principal_id) as Array<{
+        user_file: string;
+        display_name: string;
+        created_at: number;
+        id: string;
+      }>;
+      if (candidates.length === 0) continue;
 
-          candidates.sort((a, b) => {
-            const aAuto = isAutoIncrementedUserFile(a.user_file, a.display_name)
-              ? 1
-              : 0;
-            const bAuto = isAutoIncrementedUserFile(b.user_file, b.display_name)
-              ? 1
-              : 0;
-            if (aAuto !== bAuto) return aAuto - bAuto;
-            if (a.created_at !== b.created_at)
-              return a.created_at - b.created_at;
-            return a.id < b.id ? -1 : a.id > b.id ? 1 : 0;
-          });
+      candidates.sort((a, b) => {
+        const aAuto = isAutoIncrementedUserFile(a.user_file, a.display_name)
+          ? 1
+          : 0;
+        const bAuto = isAutoIncrementedUserFile(b.user_file, b.display_name)
+          ? 1
+          : 0;
+        if (aAuto !== bAuto) return aAuto - bAuto;
+        if (a.created_at !== b.created_at)
+          return a.created_at - b.created_at;
+        return a.id < b.id ? -1 : a.id > b.id ? 1 : 0;
+      });
 
-          const canonical = candidates[0]!.user_file;
-          updateSiblings.run(
-            canonical,
-            Date.now(),
-            principal_id,
-            canonical,
-          );
-        }
+      const canonical = candidates[0]!.user_file;
+      updateSiblings.run(
+        canonical,
+        Date.now(),
+        principal_id,
+        canonical,
+      );
+    }
 
-        raw.exec("COMMIT");
-      } catch (e) {
-        try {
-          raw.exec("ROLLBACK");
-        } catch {
-          /* no active transaction */
-        }
-        throw e;
-      }
-    },
-  );
+    raw.exec("COMMIT");
+  } catch (e) {
+    try {
+      raw.exec("ROLLBACK");
+    } catch {
+      /* no active transaction */
+    }
+    throw e;
+  }
 }

package/src/memory/migrations/222-strip-placeholder-sentinels-from-messages.ts

@@ -1,7 +1,6 @@
 import { isPlaceholderSentinelText } from "../../providers/placeholder-sentinels.js";
 import type { DrizzleDb } from "../db-connection.js";
 import { getSqliteFrom } from "../db-connection.js";
-import { withCrashRecovery } from "./validate-migration-state.js";
 
 /**
  * Strip provider placeholder sentinel text blocks from persisted assistant
@@ -25,59 +24,53 @@ import { withCrashRecovery } from "./validate-migration-state.js";
 export function migrateStripPlaceholderSentinelsFromMessages(
   database: DrizzleDb,
 ): void {
-  withCrashRecovery(
-    database,
-    "migration_strip_placeholder_sentinels_from_messages_v1",
-    () => {
-      const raw = getSqliteFrom(database);
+  const raw = getSqliteFrom(database);
 
-      const BATCH_SIZE = 100;
-      let lastRowid = 0;
+  const BATCH_SIZE = 100;
+  let lastRowid = 0;
 
-      for (;;) {
-        const rows = raw
-          .query(
-            `SELECT rowid, id, content FROM messages
-             WHERE role = 'assistant'
-               AND content LIKE '%__PLACEHOLDER__%'
-               AND rowid > ?
-             ORDER BY rowid
-             LIMIT ?`,
-          )
-          .all(lastRowid, BATCH_SIZE) as Array<{
-          rowid: number;
-          id: string;
-          content: string;
-        }>;
+  for (;;) {
+    const rows = raw
+      .query(
+        `SELECT rowid, id, content FROM messages
+         WHERE role = 'assistant'
+           AND content LIKE '%__PLACEHOLDER__%'
+           AND rowid > ?
+         ORDER BY rowid
+         LIMIT ?`,
+      )
+      .all(lastRowid, BATCH_SIZE) as Array<{
+      rowid: number;
+      id: string;
+      content: string;
+    }>;
 
-        if (rows.length === 0) break;
+    if (rows.length === 0) break;
 
-        for (const row of rows) {
-          lastRowid = row.rowid;
+    for (const row of rows) {
+      lastRowid = row.rowid;
 
-          let blocks: Array<Record<string, unknown>>;
-          try {
-            const parsed = JSON.parse(row.content);
-            if (!Array.isArray(parsed)) continue;
-            blocks = parsed;
-          } catch {
-            continue;
-          }
+      let blocks: Array<Record<string, unknown>>;
+      try {
+        const parsed = JSON.parse(row.content);
+        if (!Array.isArray(parsed)) continue;
+        blocks = parsed;
+      } catch {
+        continue;
+      }
 
-          const stripped = blocks.filter((b) => {
-            if (typeof b !== "object" || b === null) return false;
-            if (b.type !== "text") return true;
-            const text = typeof b.text === "string" ? b.text : "";
-            return !isPlaceholderSentinelText(text);
-          });
+      const stripped = blocks.filter((b) => {
+        if (typeof b !== "object" || b === null) return false;
+        if (b.type !== "text") return true;
+        const text = typeof b.text === "string" ? b.text : "";
+        return !isPlaceholderSentinelText(text);
+      });
 
-          if (stripped.length === blocks.length) continue;
+      if (stripped.length === blocks.length) continue;
 
-          raw
-            .query(`UPDATE messages SET content = ? WHERE id = ?`)
-            .run(JSON.stringify(stripped), row.id);
-        }
-      }
-    },
-  );
+      raw
+        .query(`UPDATE messages SET content = ? WHERE id = ?`)
+        .run(JSON.stringify(stripped), row.id);
+    }
+  }
 }

package/src/memory/migrations/230-acp-session-history.ts

@@ -1,8 +1,5 @@
 import type { DrizzleDb } from "../db-connection.js";
 import { getSqliteFrom } from "../db-connection.js";
-import { withCrashRecovery } from "./validate-migration-state.js";
-
-const CHECKPOINT_KEY = "migration_acp_session_history_v1";
 
 /**
  * Create the acp_session_history table for persisting completed ACP
@@ -10,32 +7,30 @@ const CHECKPOINT_KEY = "migration_acp_session_history_v1";
  * daemon restarts.
  */
 export function migrate230AcpSessionHistory(database: DrizzleDb): void {
-  withCrashRecovery(database, CHECKPOINT_KEY, () => {
-    const raw = getSqliteFrom(database);
+  const raw = getSqliteFrom(database);
 
-    raw.exec(/*sql*/ `
-      CREATE TABLE IF NOT EXISTS acp_session_history (
-        id TEXT PRIMARY KEY,
-        agent_id TEXT NOT NULL,
-        acp_session_id TEXT NOT NULL,
-        parent_conversation_id TEXT NOT NULL,
-        started_at INTEGER NOT NULL,
-        completed_at INTEGER,
-        status TEXT NOT NULL,
-        stop_reason TEXT,
-        error TEXT,
-        event_log_json TEXT NOT NULL DEFAULT '[]'
-      )
-    `);
+  raw.exec(/*sql*/ `
+    CREATE TABLE IF NOT EXISTS acp_session_history (
+      id TEXT PRIMARY KEY,
+      agent_id TEXT NOT NULL,
+      acp_session_id TEXT NOT NULL,
+      parent_conversation_id TEXT NOT NULL,
+      started_at INTEGER NOT NULL,
+      completed_at INTEGER,
+      status TEXT NOT NULL,
+      stop_reason TEXT,
+      error TEXT,
+      event_log_json TEXT NOT NULL DEFAULT '[]'
+    )
+  `);
 
-    raw.exec(/*sql*/ `
-      CREATE INDEX IF NOT EXISTS idx_acp_session_history_started_at
-      ON acp_session_history (started_at DESC)
-    `);
+  raw.exec(/*sql*/ `
+    CREATE INDEX IF NOT EXISTS idx_acp_session_history_started_at
+    ON acp_session_history (started_at DESC)
+  `);
 
-    raw.exec(/*sql*/ `
-      CREATE INDEX IF NOT EXISTS idx_acp_session_history_parent_conversation_id
-      ON acp_session_history (parent_conversation_id)
-    `);
-  });
+  raw.exec(/*sql*/ `
+    CREATE INDEX IF NOT EXISTS idx_acp_session_history_parent_conversation_id
+    ON acp_session_history (parent_conversation_id)
+  `);
 }

package/src/memory/migrations/231-repair-memory-graph-event-dates.ts

@@ -1,8 +1,6 @@
 import type { DrizzleDb } from "../db-connection.js";
 import { getSqliteFrom } from "../db-connection.js";
-import { withCrashRecovery } from "./validate-migration-state.js";
 
-const CHECKPOINT_KEY = "migration_repair_memory_graph_event_dates_v1";
 const REPAIR_CREATED_YEAR = 2026;
 const CORRUPT_EVENT_YEARS = [2024, 2025] as const;
 const DAY_MS = 24 * 60 * 60 * 1000;
@@ -76,67 +74,65 @@ export function repairMemoryGraphEventDate(
 export function migrate231RepairMemoryGraphEventDates(
   database: DrizzleDb,
 ): void {
-  withCrashRecovery(database, CHECKPOINT_KEY, () => {
-    const raw = getSqliteFrom(database);
-    const rows = raw
-      .query(
-        /*sql*/ `
-          SELECT id, created, event_date, content
-          FROM memory_graph_nodes
-          WHERE event_date IS NOT NULL
-            AND CAST(strftime('%Y', created / 1000, 'unixepoch') AS INTEGER) = ?
-            AND CAST(strftime('%Y', event_date / 1000, 'unixepoch') AS INTEGER) IN (?, ?)
-        `,
-      )
-      .all(
-        REPAIR_CREATED_YEAR,
-        ...CORRUPT_EVENT_YEARS,
-      ) as GraphNodeEventDateRow[];
-
-    const repairs: EventDateRepair[] = [];
-    for (const row of rows) {
-      const corrected = repairMemoryGraphEventDate(
-        row.created,
-        row.event_date,
-        row.content ?? "",
-      );
-      if (corrected == null) continue;
-      repairs.push({
-        id: row.id,
-        oldEventDate: row.event_date,
-        newEventDate: corrected,
-      });
-    }
+  const raw = getSqliteFrom(database);
+  const rows = raw
+    .query(
+      /*sql*/ `
+        SELECT id, created, event_date, content
+        FROM memory_graph_nodes
+        WHERE event_date IS NOT NULL
+          AND CAST(strftime('%Y', created / 1000, 'unixepoch') AS INTEGER) = ?
+          AND CAST(strftime('%Y', event_date / 1000, 'unixepoch') AS INTEGER) IN (?, ?)
+      `,
+    )
+    .all(
+      REPAIR_CREATED_YEAR,
+      ...CORRUPT_EVENT_YEARS,
+    ) as GraphNodeEventDateRow[];
+
+  const repairs: EventDateRepair[] = [];
+  for (const row of rows) {
+    const corrected = repairMemoryGraphEventDate(
+      row.created,
+      row.event_date,
+      row.content ?? "",
+    );
+    if (corrected == null) continue;
+    repairs.push({
+      id: row.id,
+      oldEventDate: row.event_date,
+      newEventDate: corrected,
+    });
+  }
 
-    if (repairs.length === 0) return;
-
-    const updateNode = raw.prepare(/*sql*/ `
-        UPDATE memory_graph_nodes
-        SET event_date = ?
-        WHERE id = ?
-          AND event_date = ?
-      `);
-    const updateTrigger = raw.prepare(/*sql*/ `
-        UPDATE memory_graph_triggers
-        SET event_date = ?
-        WHERE node_id = ?
-          AND event_date = ?
-      `);
-
-    raw.exec("BEGIN");
+  if (repairs.length === 0) return;
+
+  const updateNode = raw.prepare(/*sql*/ `
+      UPDATE memory_graph_nodes
+      SET event_date = ?
+      WHERE id = ?
+        AND event_date = ?
+    `);
+  const updateTrigger = raw.prepare(/*sql*/ `
+      UPDATE memory_graph_triggers
+      SET event_date = ?
+      WHERE node_id = ?
+        AND event_date = ?
+    `);
+
+  raw.exec("BEGIN");
+  try {
+    for (const repair of repairs) {
+      updateNode.run(repair.newEventDate, repair.id, repair.oldEventDate);
+      updateTrigger.run(repair.newEventDate, repair.id, repair.oldEventDate);
+    }
+    raw.exec("COMMIT");
+  } catch (error) {
     try {
-      for (const repair of repairs) {
-        updateNode.run(repair.newEventDate, repair.id, repair.oldEventDate);
-        updateTrigger.run(repair.newEventDate, repair.id, repair.oldEventDate);
-      }
-      raw.exec("COMMIT");
-    } catch (error) {
-      try {
-        raw.exec("ROLLBACK");
-      } catch {
-        // No active transaction.
-      }
-      throw error;
+      raw.exec("ROLLBACK");
+    } catch {
+      // No active transaction.
     }
-  });
+    throw error;
+  }
 }

package/src/memory/migrations/232-activation-state.ts

@@ -1,8 +1,5 @@
 import type { DrizzleDb } from "../db-connection.js";
 import { getSqliteFrom } from "../db-connection.js";
-import { withCrashRecovery } from "./validate-migration-state.js";
-
-const CHECKPOINT_KEY = "migration_activation_state_v1";
 
 /**
  * Create the activation_state table for memory v2 retrieval persistence.
@@ -17,19 +14,17 @@ const CHECKPOINT_KEY = "migration_activation_state_v1";
  * pattern: single-row-per-conversation snapshot rehydrated on resume.
  */
 export function migrateActivationState(database: DrizzleDb): void {
-  withCrashRecovery(database, CHECKPOINT_KEY, () => {
-    const raw = getSqliteFrom(database);
-    raw.exec(/*sql*/ `
-      CREATE TABLE IF NOT EXISTS activation_state (
-        conversation_id TEXT PRIMARY KEY,
-        message_id TEXT NOT NULL,
-        state_json TEXT NOT NULL,
-        ever_injected_json TEXT NOT NULL DEFAULT '[]',
-        current_turn INTEGER NOT NULL DEFAULT 0,
-        updated_at INTEGER NOT NULL
-      )
-    `);
-  });
+  const raw = getSqliteFrom(database);
+  raw.exec(/*sql*/ `
+    CREATE TABLE IF NOT EXISTS activation_state (
+      conversation_id TEXT PRIMARY KEY,
+      message_id TEXT NOT NULL,
+      state_json TEXT NOT NULL,
+      ever_injected_json TEXT NOT NULL DEFAULT '[]',
+      current_turn INTEGER NOT NULL DEFAULT 0,
+      updated_at INTEGER NOT NULL
+    )
+  `);
 }
 
 export function downActivationState(database: DrizzleDb): void {

package/src/memory/migrations/233-document-conversations.ts

@@ -1,8 +1,5 @@
 import type { DrizzleDb } from "../db-connection.js";
 import { getSqliteFrom } from "../db-connection.js";
-import { withCrashRecovery } from "./validate-migration-state.js";
-
-const CHECKPOINT_KEY = "migration_document_conversations_v1";
 
 /**
  * Create the document_conversations junction table.
@@ -26,29 +23,27 @@ const CHECKPOINT_KEY = "migration_document_conversations_v1";
  * so that pre-existing document–conversation relationships are preserved.
  */
 export function migrateCreateDocumentConversations(database: DrizzleDb): void {
-  withCrashRecovery(database, CHECKPOINT_KEY, () => {
-    const raw = getSqliteFrom(database);
+  const raw = getSqliteFrom(database);
 
-    raw.exec(/*sql*/ `
-      CREATE TABLE IF NOT EXISTS document_conversations (
-        surface_id TEXT NOT NULL,
-        conversation_id TEXT NOT NULL,
-        created_at INTEGER NOT NULL DEFAULT (strftime('%s','now') * 1000),
-        PRIMARY KEY (surface_id, conversation_id),
-        FOREIGN KEY (surface_id) REFERENCES documents(surface_id) ON DELETE CASCADE
-      )
-    `);
+  raw.exec(/*sql*/ `
+    CREATE TABLE IF NOT EXISTS document_conversations (
+      surface_id TEXT NOT NULL,
+      conversation_id TEXT NOT NULL,
+      created_at INTEGER NOT NULL DEFAULT (strftime('%s','now') * 1000),
+      PRIMARY KEY (surface_id, conversation_id),
+      FOREIGN KEY (surface_id) REFERENCES documents(surface_id) ON DELETE CASCADE
+    )
+  `);
 
-    raw.exec(/*sql*/ `
-      CREATE INDEX IF NOT EXISTS idx_doc_conv_conversation_id
-      ON document_conversations(conversation_id)
-    `);
+  raw.exec(/*sql*/ `
+    CREATE INDEX IF NOT EXISTS idx_doc_conv_conversation_id
+    ON document_conversations(conversation_id)
+  `);
 
-    // Backfill: seed junction table from existing documents.conversation_id
-    raw.exec(/*sql*/ `
-      INSERT OR IGNORE INTO document_conversations (surface_id, conversation_id, created_at)
-      SELECT surface_id, conversation_id, created_at
-      FROM documents
-    `);
-  });
+  // Backfill: seed junction table from existing documents.conversation_id
+  raw.exec(/*sql*/ `
+    INSERT OR IGNORE INTO document_conversations (surface_id, conversation_id, created_at)
+    SELECT surface_id, conversation_id, created_at
+    FROM documents
+  `);
 }