@vellumai/assistant 0.10.0 → 0.10.1-staging.1

package/src/memory/migrations/176-drop-capability-card-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_drop_capability_card_state_v1";
 
 /**
  * Remove persisted capability-card state after the card feed was deleted.
@@ -11,28 +8,26 @@ const CHECKPOINT_KEY = "migration_drop_capability_card_state_v1";
  * state, and queued generation jobs are dead data and should not survive.
  */
 export function migrateDropCapabilityCardState(database: DrizzleDb): void {
-  withCrashRecovery(database, CHECKPOINT_KEY, () => {
-    const raw = getSqliteFrom(database);
-
-    const conversationStartersExists = raw
-      .query(
-        `SELECT 1 FROM sqlite_master WHERE type = 'table' AND name = 'conversation_starters'`,
-      )
-      .get();
-    if (conversationStartersExists) {
-      raw.exec(
-        /*sql*/ `DELETE FROM conversation_starters WHERE card_type = 'card'`,
-      );
-    }
+  const raw = getSqliteFrom(database);
 
+  const conversationStartersExists = raw
+    .query(
+      `SELECT 1 FROM sqlite_master WHERE type = 'table' AND name = 'conversation_starters'`,
+    )
+    .get();
+  if (conversationStartersExists) {
     raw.exec(
-      /*sql*/ `DELETE FROM memory_jobs WHERE type = 'generate_capability_cards'`,
+      /*sql*/ `DELETE FROM conversation_starters WHERE card_type = 'card'`,
     );
-    raw.exec(
-      /*sql*/ `DELETE FROM memory_checkpoints WHERE key LIKE 'capability_cards:%'`,
-    );
-    raw.exec(/*sql*/ `DROP TABLE IF EXISTS capability_card_categories`);
-  });
+  }
+
+  raw.exec(
+    /*sql*/ `DELETE FROM memory_jobs WHERE type = 'generate_capability_cards'`,
+  );
+  raw.exec(
+    /*sql*/ `DELETE FROM memory_checkpoints WHERE key LIKE 'capability_cards:%'`,
+  );
+  raw.exec(/*sql*/ `DROP TABLE IF EXISTS capability_card_categories`);
 }
 
 /**

package/src/memory/migrations/177-create-trace-events-table.ts

@@ -1,40 +1,35 @@
 import type { DrizzleDb } from "../db-connection.js";
 import { getSqliteFrom } from "../db-connection.js";
-import { withCrashRecovery } from "./validate-migration-state.js";
-
-const CHECKPOINT_KEY = "migration_create_trace_events_table_v1";
 
 /**
  * Create the trace_events table for persisting trace/activity events
  * so the Logs panel has data across sessions.
  */
 export function migrateCreateTraceEventsTable(database: DrizzleDb): void {
-  withCrashRecovery(database, CHECKPOINT_KEY, () => {
-    const raw = getSqliteFrom(database);
+  const raw = getSqliteFrom(database);
 
-    raw.exec(/*sql*/ `
-      CREATE TABLE IF NOT EXISTS trace_events (
-        event_id TEXT PRIMARY KEY,
-        conversation_id TEXT NOT NULL,
-        request_id TEXT,
-        timestamp_ms INTEGER NOT NULL,
-        sequence INTEGER NOT NULL,
-        kind TEXT NOT NULL,
-        status TEXT,
-        summary TEXT NOT NULL,
-        attributes_json TEXT,
-        created_at INTEGER NOT NULL
-      )
-    `);
+  raw.exec(/*sql*/ `
+    CREATE TABLE IF NOT EXISTS trace_events (
+      event_id TEXT PRIMARY KEY,
+      conversation_id TEXT NOT NULL,
+      request_id TEXT,
+      timestamp_ms INTEGER NOT NULL,
+      sequence INTEGER NOT NULL,
+      kind TEXT NOT NULL,
+      status TEXT,
+      summary TEXT NOT NULL,
+      attributes_json TEXT,
+      created_at INTEGER NOT NULL
+    )
+  `);
 
-    raw.exec(/*sql*/ `
-      CREATE INDEX IF NOT EXISTS idx_trace_events_conversation_id
-      ON trace_events (conversation_id)
-    `);
+  raw.exec(/*sql*/ `
+    CREATE INDEX IF NOT EXISTS idx_trace_events_conversation_id
+    ON trace_events (conversation_id)
+  `);
 
-    raw.exec(/*sql*/ `
-      CREATE INDEX IF NOT EXISTS idx_trace_events_conversation_timestamp
-      ON trace_events (conversation_id, timestamp_ms)
-    `);
-  });
+  raw.exec(/*sql*/ `
+    CREATE INDEX IF NOT EXISTS idx_trace_events_conversation_timestamp
+    ON trace_events (conversation_id, timestamp_ms)
+  `);
 }

package/src/memory/migrations/180-backfill-inline-attachments-to-disk.ts

@@ -1,7 +1,6 @@
 import { writeAttachmentToDisk } from "../attachments-store.js";
 import type { DrizzleDb } from "../db-connection.js";
 import { getSqliteFrom } from "../db-connection.js";
-import { withCrashRecovery } from "./validate-migration-state.js";
 
 /**
  * Backfill existing inline (base64-in-DB) attachments to disk.
@@ -16,53 +15,47 @@ import { withCrashRecovery } from "./validate-migration-state.js";
 export function migrateBackfillInlineAttachmentsToDisk(
   database: DrizzleDb,
 ): void {
-  withCrashRecovery(
-    database,
-    "migration_backfill_inline_attachments_v1",
-    () => {
-      const raw = getSqliteFrom(database);
-      const BATCH_SIZE = 50;
-      let totalMigrated = 0;
+  const raw = getSqliteFrom(database);
+  const BATCH_SIZE = 50;
+  let totalMigrated = 0;
 
-      for (;;) {
-        const rows = raw
-          .query(
-            `SELECT id, original_filename, data_base64 FROM attachments
-             WHERE (file_path IS NULL OR file_path = '')
-               AND data_base64 != ''
-               AND length(data_base64) > 0
-             LIMIT ?`,
-          )
-          .all(BATCH_SIZE) as Array<{
-          id: string;
-          original_filename: string;
-          data_base64: string;
-        }>;
+  for (;;) {
+    const rows = raw
+      .query(
+        `SELECT id, original_filename, data_base64 FROM attachments
+         WHERE (file_path IS NULL OR file_path = '')
+           AND data_base64 != ''
+           AND length(data_base64) > 0
+         LIMIT ?`,
+      )
+      .all(BATCH_SIZE) as Array<{
+      id: string;
+      original_filename: string;
+      data_base64: string;
+    }>;
 
-        if (rows.length === 0) break;
+    if (rows.length === 0) break;
 
-        for (const row of rows) {
-          const filePath = writeAttachmentToDisk(
-            row.data_base64,
-            row.original_filename,
-          );
-          raw
-            .query(
-              `UPDATE attachments SET file_path = ?, data_base64 = '' WHERE id = ?`,
-            )
-            .run(filePath, row.id);
-        }
+    for (const row of rows) {
+      const filePath = writeAttachmentToDisk(
+        row.data_base64,
+        row.original_filename,
+      );
+      raw
+        .query(
+          `UPDATE attachments SET file_path = ?, data_base64 = '' WHERE id = ?`,
+        )
+        .run(filePath, row.id);
+    }
 
-        totalMigrated += rows.length;
-      }
+    totalMigrated += rows.length;
+  }
 
-      if (totalMigrated > 0) {
-        // Log is intentionally not imported to keep migration self-contained;
-        // the checkpoint value records completion.
-        console.log(`Migrated ${totalMigrated} inline attachments to disk`);
-      }
-    },
-  );
+  if (totalMigrated > 0) {
+    // Log is intentionally not imported to keep migration self-contained;
+    // the checkpoint value records completion.
+    console.log(`Migrated ${totalMigrated} inline attachments to disk`);
+  }
 }
 
 /**

package/src/memory/migrations/181-rename-thread-starters-checkpoints.ts

@@ -1,6 +1,5 @@
 import type { DrizzleDb } from "../db-connection.js";
 import { getSqliteFrom } from "../db-connection.js";
-import { withCrashRecovery } from "./validate-migration-state.js";
 
 /**
  * Rename checkpoint keys from the old `thread_starters:` prefix to
@@ -10,10 +9,10 @@ import { withCrashRecovery } from "./validate-migration-state.js";
  *
  * This was originally appended to migration 174, but that migration
  * had already shipped with its own checkpoint key
- * (`migration_rename_thread_starters_table_v1`), so `withCrashRecovery`
+ * (`migration_rename_thread_starters_table_v1`), so the step runner
  * would skip the entire body for users who had already run it. Moving
  * the checkpoint-key rewrite into its own migration with an independent
- * checkpoint key ensures it runs for all users.
+ * step name ensures it runs for all users.
  *
  * The rename is collision-safe: if a database already has both old
  * `thread_starters:*` keys and new `conversation_starters:*` keys
@@ -23,26 +22,20 @@ import { withCrashRecovery } from "./validate-migration-state.js";
 export function migrateRenameThreadStartersCheckpoints(
   database: DrizzleDb,
 ): void {
-  withCrashRecovery(
-    database,
-    "migration_rename_thread_starters_checkpoints_v1",
-    () => {
-      const raw = getSqliteFrom(database);
+  const raw = getSqliteFrom(database);
 
-      // 1. Delete old thread_starters: keys where a corresponding
-      //    conversation_starters: key already exists (the newer key
-      //    written by updated code takes precedence).
-      raw.exec(/*sql*/ `DELETE FROM memory_checkpoints
-         WHERE key LIKE 'thread_starters:%'
-           AND replace(key, 'thread_starters:', 'conversation_starters:') IN (
-             SELECT key FROM memory_checkpoints WHERE key LIKE 'conversation_starters:%'
-           )`);
+  // 1. Delete old thread_starters: keys where a corresponding
+  //    conversation_starters: key already exists (the newer key
+  //    written by updated code takes precedence).
+  raw.exec(/*sql*/ `DELETE FROM memory_checkpoints
+     WHERE key LIKE 'thread_starters:%'
+       AND replace(key, 'thread_starters:', 'conversation_starters:') IN (
+         SELECT key FROM memory_checkpoints WHERE key LIKE 'conversation_starters:%'
+       )`);
 
-      // 2. Rename remaining old keys that have no collision.
-      raw.exec(
-        /*sql*/ `UPDATE memory_checkpoints SET key = replace(key, 'thread_starters:', 'conversation_starters:') WHERE key LIKE 'thread_starters:%'`,
-      );
-    },
+  // 2. Rename remaining old keys that have no collision.
+  raw.exec(
+    /*sql*/ `UPDATE memory_checkpoints SET key = replace(key, 'thread_starters:', 'conversation_starters:') WHERE key LIKE 'thread_starters:%'`,
   );
 }
 

package/src/memory/migrations/191-backfill-audio-attachment-mime-types.ts

@@ -1,6 +1,5 @@
 import type { DrizzleDb } from "../db-connection.js";
 import { getSqliteFrom } from "../db-connection.js";
-import { withCrashRecovery } from "./validate-migration-state.js";
 
 /**
  * Backfill MIME types for audio attachments that were stored with
@@ -23,31 +22,25 @@ const AUDIO_EXT_MIME: Record<string, string> = {
 export function migrateBackfillAudioAttachmentMimeTypes(
   database: DrizzleDb,
 ): void {
-  withCrashRecovery(
-    database,
-    "migration_backfill_audio_attachment_mime_types_v1",
-    () => {
-      const raw = getSqliteFrom(database);
+  const raw = getSqliteFrom(database);
 
-      for (const [ext, mime] of Object.entries(AUDIO_EXT_MIME)) {
-        const pattern = `%.${ext}`;
-        const result = raw
-          .query(
-            `UPDATE attachments
-             SET mime_type = ?, kind = 'document'
-             WHERE lower(original_filename) LIKE ?
-               AND mime_type = 'application/octet-stream'`,
-          )
-          .run(mime, pattern);
+  for (const [ext, mime] of Object.entries(AUDIO_EXT_MIME)) {
+    const pattern = `%.${ext}`;
+    const result = raw
+      .query(
+        `UPDATE attachments
+         SET mime_type = ?, kind = 'document'
+         WHERE lower(original_filename) LIKE ?
+           AND mime_type = 'application/octet-stream'`,
+      )
+      .run(mime, pattern);
 
-        if ((result as { changes?: number }).changes) {
-          console.log(
-            `Backfilled ${(result as { changes: number }).changes} .${ext} attachments → ${mime}`,
-          );
-        }
-      }
-    },
-  );
+    if ((result as { changes?: number }).changes) {
+      console.log(
+        `Backfilled ${(result as { changes: number }).changes} .${ext} attachments → ${mime}`,
+      );
+    }
+  }
 }
 
 /**

package/src/memory/migrations/192-contacts-user-file-column.ts

@@ -1,15 +1,12 @@
 import type { DrizzleDb } from "../db-connection.js";
 import { getSqliteFrom } from "../db-connection.js";
-import { withCrashRecovery } from "./validate-migration-state.js";
 
 export function migrateContactsUserFileColumn(database: DrizzleDb): void {
-  withCrashRecovery(database, "migration_contacts_user_file_column_v1", () => {
-    const raw = getSqliteFrom(database);
+  const raw = getSqliteFrom(database);
 
-    try {
-      raw.exec(/*sql*/ `ALTER TABLE contacts ADD COLUMN user_file TEXT`);
-    } catch {
-      /* already exists */
-    }
-  });
+  try {
+    raw.exec(/*sql*/ `ALTER TABLE contacts ADD COLUMN user_file TEXT`);
+  } catch {
+    /* already exists */
+  }
 }

package/src/memory/migrations/193-add-source-type-columns.ts

@@ -1,7 +1,6 @@
 import type { DrizzleDb } from "../db-connection.js";
 import { getSqliteFrom } from "../db-connection.js";
 import { tableHasColumn } from "./schema-introspection.js";
-import { withCrashRecovery } from "./validate-migration-state.js";
 
 /**
  * Add source_type and source_message_role columns to memory_items.
@@ -16,49 +15,47 @@ import { withCrashRecovery } from "./validate-migration-state.js";
  * 2. source_message_role from the earliest source message's role via subquery
  */
 export function migrateAddSourceTypeColumns(database: DrizzleDb): void {
-  withCrashRecovery(database, "migration_add_source_type_columns_v1", () => {
-    const raw = getSqliteFrom(database);
-
-    // Add source_type column if it doesn't exist
-    if (!tableHasColumn(database, "memory_items", "source_type")) {
-      raw.exec(
-        /*sql*/ `ALTER TABLE memory_items ADD COLUMN source_type TEXT NOT NULL DEFAULT 'extraction'`,
-      );
-    }
+  const raw = getSqliteFrom(database);
 
-    // Add source_message_role column if it doesn't exist
-    if (!tableHasColumn(database, "memory_items", "source_message_role")) {
-      raw.exec(
-        /*sql*/ `ALTER TABLE memory_items ADD COLUMN source_message_role TEXT`,
-      );
-    }
+  // Add source_type column if it doesn't exist
+  if (!tableHasColumn(database, "memory_items", "source_type")) {
+    raw.exec(
+      /*sql*/ `ALTER TABLE memory_items ADD COLUMN source_type TEXT NOT NULL DEFAULT 'extraction'`,
+    );
+  }
 
-    // Backfill source_type = 'tool' for items that were explicitly saved
+  // Add source_message_role column if it doesn't exist
+  if (!tableHasColumn(database, "memory_items", "source_message_role")) {
     raw.exec(
-      /*sql*/ `UPDATE memory_items SET source_type = 'tool' WHERE verification_state = 'user_confirmed'`,
+      /*sql*/ `ALTER TABLE memory_items ADD COLUMN source_message_role TEXT`,
     );
+  }
+
+  // Backfill source_type = 'tool' for items that were explicitly saved
+  raw.exec(
+    /*sql*/ `UPDATE memory_items SET source_type = 'tool' WHERE verification_state = 'user_confirmed'`,
+  );
 
-    // Backfill source_message_role from the earliest source message's role.
-    // Only backfill where source_message_role is currently NULL and a source
-    // message exists.
-    raw.exec(/*sql*/ `
-        UPDATE memory_items
-        SET source_message_role = (
-          SELECT m.role
+  // Backfill source_message_role from the earliest source message's role.
+  // Only backfill where source_message_role is currently NULL and a source
+  // message exists.
+  raw.exec(/*sql*/ `
+      UPDATE memory_items
+      SET source_message_role = (
+        SELECT m.role
+        FROM memory_item_sources mis
+        JOIN messages m ON m.id = mis.message_id
+        WHERE mis.memory_item_id = memory_items.id
+        ORDER BY mis.created_at ASC
+        LIMIT 1
+      )
+      WHERE source_message_role IS NULL
+        AND EXISTS (
+          SELECT 1
           FROM memory_item_sources mis
-          JOIN messages m ON m.id = mis.message_id
           WHERE mis.memory_item_id = memory_items.id
-          ORDER BY mis.created_at ASC
-          LIMIT 1
         )
-        WHERE source_message_role IS NULL
-          AND EXISTS (
-            SELECT 1
-            FROM memory_item_sources mis
-            WHERE mis.memory_item_id = memory_items.id
-          )
-      `);
-  });
+    `);
 }
 
 /**

package/src/memory/migrations/194-memory-recall-logs.ts

@@ -1,50 +1,45 @@
 import type { DrizzleDb } from "../db-connection.js";
 import { getSqliteFrom } from "../db-connection.js";
-import { withCrashRecovery } from "./validate-migration-state.js";
-
-const CHECKPOINT_KEY = "migration_create_memory_recall_logs_v1";
 
 /**
  * Create the memory_recall_logs table for the inspector memory tab.
  */
 export function migrateCreateMemoryRecallLogs(database: DrizzleDb): void {
-  withCrashRecovery(database, CHECKPOINT_KEY, () => {
-    const raw = getSqliteFrom(database);
+  const raw = getSqliteFrom(database);
 
-    raw.exec(/*sql*/ `
-      CREATE TABLE IF NOT EXISTS memory_recall_logs (
-        id TEXT PRIMARY KEY,
-        conversation_id TEXT NOT NULL,
-        message_id TEXT,
-        enabled INTEGER NOT NULL,
-        degraded INTEGER NOT NULL,
-        provider TEXT,
-        model TEXT,
-        degradation_json TEXT,
-        semantic_hits INTEGER NOT NULL,
-        merged_count INTEGER NOT NULL,
-        selected_count INTEGER NOT NULL,
-        tier1_count INTEGER NOT NULL,
-        tier2_count INTEGER NOT NULL,
-        hybrid_search_latency_ms INTEGER NOT NULL,
-        sparse_vector_used INTEGER NOT NULL,
-        injected_tokens INTEGER NOT NULL,
-        latency_ms INTEGER NOT NULL,
-        top_candidates_json TEXT NOT NULL,
-        injected_text TEXT,
-        reason TEXT,
-        created_at INTEGER NOT NULL
-      )
-    `);
+  raw.exec(/*sql*/ `
+    CREATE TABLE IF NOT EXISTS memory_recall_logs (
+      id TEXT PRIMARY KEY,
+      conversation_id TEXT NOT NULL,
+      message_id TEXT,
+      enabled INTEGER NOT NULL,
+      degraded INTEGER NOT NULL,
+      provider TEXT,
+      model TEXT,
+      degradation_json TEXT,
+      semantic_hits INTEGER NOT NULL,
+      merged_count INTEGER NOT NULL,
+      selected_count INTEGER NOT NULL,
+      tier1_count INTEGER NOT NULL,
+      tier2_count INTEGER NOT NULL,
+      hybrid_search_latency_ms INTEGER NOT NULL,
+      sparse_vector_used INTEGER NOT NULL,
+      injected_tokens INTEGER NOT NULL,
+      latency_ms INTEGER NOT NULL,
+      top_candidates_json TEXT NOT NULL,
+      injected_text TEXT,
+      reason TEXT,
+      created_at INTEGER NOT NULL
+    )
+  `);
 
-    raw.exec(/*sql*/ `
-      CREATE INDEX IF NOT EXISTS idx_memory_recall_logs_message_id
-      ON memory_recall_logs (message_id)
-    `);
+  raw.exec(/*sql*/ `
+    CREATE INDEX IF NOT EXISTS idx_memory_recall_logs_message_id
+    ON memory_recall_logs (message_id)
+  `);
 
-    raw.exec(/*sql*/ `
-      CREATE INDEX IF NOT EXISTS idx_memory_recall_logs_conversation_id
-      ON memory_recall_logs (conversation_id)
-    `);
-  });
+  raw.exec(/*sql*/ `
+    CREATE INDEX IF NOT EXISTS idx_memory_recall_logs_conversation_id
+    ON memory_recall_logs (conversation_id)
+  `);
 }