@vellumai/assistant 0.10.0 → 0.10.1-staging.1

package/src/memory/migrations/234-memory-v2-activation-logs.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_memory_v2_activation_logs_v1";
 
 /**
  * Create the memory_v2_activation_logs table for per-turn v2 activation
@@ -19,34 +16,32 @@ const CHECKPOINT_KEY = "migration_memory_v2_activation_logs_v1";
  * message_id, by conversation_id, and ordering by created_at.
  */
 export function migrateMemoryV2ActivationLogs(database: DrizzleDb): void {
-  withCrashRecovery(database, CHECKPOINT_KEY, () => {
-    const raw = getSqliteFrom(database);
-    raw.exec(/*sql*/ `
-      CREATE TABLE IF NOT EXISTS memory_v2_activation_logs (
-        id TEXT PRIMARY KEY,
-        conversation_id TEXT NOT NULL,
-        message_id TEXT,
-        turn INTEGER NOT NULL,
-        mode TEXT NOT NULL,
-        concepts_json TEXT NOT NULL,
-        skills_json TEXT NOT NULL,
-        config_json TEXT NOT NULL,
-        created_at INTEGER NOT NULL
-      )
-    `);
-    raw.exec(/*sql*/ `
-      CREATE INDEX IF NOT EXISTS idx_memory_v2_activation_logs_message_id
-        ON memory_v2_activation_logs (message_id)
-    `);
-    raw.exec(/*sql*/ `
-      CREATE INDEX IF NOT EXISTS idx_memory_v2_activation_logs_conversation_id
-        ON memory_v2_activation_logs (conversation_id)
-    `);
-    raw.exec(/*sql*/ `
-      CREATE INDEX IF NOT EXISTS idx_memory_v2_activation_logs_created_at
-        ON memory_v2_activation_logs (created_at)
-    `);
-  });
+  const raw = getSqliteFrom(database);
+  raw.exec(/*sql*/ `
+    CREATE TABLE IF NOT EXISTS memory_v2_activation_logs (
+      id TEXT PRIMARY KEY,
+      conversation_id TEXT NOT NULL,
+      message_id TEXT,
+      turn INTEGER NOT NULL,
+      mode TEXT NOT NULL,
+      concepts_json TEXT NOT NULL,
+      skills_json TEXT NOT NULL,
+      config_json TEXT NOT NULL,
+      created_at INTEGER NOT NULL
+    )
+  `);
+  raw.exec(/*sql*/ `
+    CREATE INDEX IF NOT EXISTS idx_memory_v2_activation_logs_message_id
+      ON memory_v2_activation_logs (message_id)
+  `);
+  raw.exec(/*sql*/ `
+    CREATE INDEX IF NOT EXISTS idx_memory_v2_activation_logs_conversation_id
+      ON memory_v2_activation_logs (conversation_id)
+  `);
+  raw.exec(/*sql*/ `
+    CREATE INDEX IF NOT EXISTS idx_memory_v2_activation_logs_created_at
+      ON memory_v2_activation_logs (created_at)
+  `);
 }
 
 export function downMemoryV2ActivationLogs(database: DrizzleDb): void {

package/src/memory/migrations/235-slack-compaction-watermark.ts

@@ -1,8 +1,5 @@
 import type { DrizzleDb } from "../db-connection.js";
 import { tableHasColumn } from "./schema-introspection.js";
-import { withCrashRecovery } from "./validate-migration-state.js";
-
-const CHECKPOINT_KEY = "migration_slack_compaction_watermark_v1";
 
 const COLUMNS = [
   {
@@ -24,14 +21,12 @@ const COLUMNS = [
  * of local insertion order.
  */
 export function migrateSlackCompactionWatermark(database: DrizzleDb): void {
-  withCrashRecovery(database, CHECKPOINT_KEY, () => {
-    for (const column of COLUMNS) {
-      if (tableHasColumn(database, "conversations", column.name)) {
-        continue;
-      }
-      database.run(`ALTER TABLE conversations ADD COLUMN ${column.definition}`);
+  for (const column of COLUMNS) {
+    if (tableHasColumn(database, "conversations", column.name)) {
+      continue;
     }
-  });
+    database.run(`ALTER TABLE conversations ADD COLUMN ${column.definition}`);
+  }
 }
 
 export function downSlackCompactionWatermark(database: DrizzleDb): void {

package/src/memory/migrations/236-tool-invocations-matched-rule-id.ts

@@ -1,21 +1,16 @@
 import type { DrizzleDb } from "../db-connection.js";
 import { tableHasColumn } from "./schema-introspection.js";
-import { withCrashRecovery } from "./validate-migration-state.js";
-
-const CHECKPOINT_KEY = "migration_tool_invocations_matched_trust_rule_id_v1";
 
 /**
  * Add matched_trust_rule_id column to tool_invocations for audit and rule editor UI.
  */
 export function migrateToolInvocationsMatchedRuleId(database: DrizzleDb): void {
-  withCrashRecovery(database, CHECKPOINT_KEY, () => {
-    if (tableHasColumn(database, "tool_invocations", "matched_trust_rule_id")) {
-      return;
-    }
-    database.run(
-      `ALTER TABLE tool_invocations ADD COLUMN matched_trust_rule_id TEXT`,
-    );
-  });
+  if (tableHasColumn(database, "tool_invocations", "matched_trust_rule_id")) {
+    return;
+  }
+  database.run(
+    `ALTER TABLE tool_invocations ADD COLUMN matched_trust_rule_id TEXT`,
+  );
 }
 
 export function downToolInvocationsMatchedRuleId(database: DrizzleDb): void {

package/src/memory/migrations/237-heartbeat-runs.ts

@@ -1,9 +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";
-
-const CHECKPOINT_KEY = "migration_heartbeat_runs_v1";
 
 /**
  * Create the heartbeat_runs table for tracking heartbeat execution lifecycle.
@@ -13,30 +10,28 @@ const CHECKPOINT_KEY = "migration_heartbeat_runs_v1";
  * pending -> skipped/missed/superseded.
  */
 export function migrateHeartbeatRuns(database: DrizzleDb): void {
-  withCrashRecovery(database, CHECKPOINT_KEY, () => {
-    if (tableHasColumn(database, "heartbeat_runs", "id")) {
-      return;
-    }
-    const raw = getSqliteFrom(database);
-    raw.exec(/*sql*/ `
-      CREATE TABLE IF NOT EXISTS heartbeat_runs (
-        id TEXT PRIMARY KEY,
-        scheduled_for INTEGER NOT NULL,
-        started_at INTEGER,
-        finished_at INTEGER,
-        duration_ms INTEGER,
-        status TEXT NOT NULL,
-        skip_reason TEXT,
-        error TEXT,
-        conversation_id TEXT,
-        created_at INTEGER NOT NULL
-      )
-    `);
-    raw.exec(/*sql*/ `
-      CREATE INDEX IF NOT EXISTS idx_heartbeat_runs_scheduled_for
-        ON heartbeat_runs (scheduled_for)
-    `);
-  });
+  if (tableHasColumn(database, "heartbeat_runs", "id")) {
+    return;
+  }
+  const raw = getSqliteFrom(database);
+  raw.exec(/*sql*/ `
+    CREATE TABLE IF NOT EXISTS heartbeat_runs (
+      id TEXT PRIMARY KEY,
+      scheduled_for INTEGER NOT NULL,
+      started_at INTEGER,
+      finished_at INTEGER,
+      duration_ms INTEGER,
+      status TEXT NOT NULL,
+      skip_reason TEXT,
+      error TEXT,
+      conversation_id TEXT,
+      created_at INTEGER NOT NULL
+    )
+  `);
+  raw.exec(/*sql*/ `
+    CREATE INDEX IF NOT EXISTS idx_heartbeat_runs_scheduled_for
+      ON heartbeat_runs (scheduled_for)
+  `);
 }
 
 export function downHeartbeatRuns(database: DrizzleDb): void {

package/src/memory/migrations/239-trace-events-created-at-index.ts

@@ -1,18 +1,13 @@
 import type { DrizzleDb } from "../db-connection.js";
 import { getSqliteFrom } from "../db-connection.js";
-import { withCrashRecovery } from "./validate-migration-state.js";
-
-const CHECKPOINT_KEY = "migration_trace_events_created_at_index_v1";
 
 /**
  * Add an index on `trace_events.created_at` so the periodic prune job
  * can locate expired rows without a full table scan.
  */
 export function migrateTraceEventsCreatedAtIndex(database: DrizzleDb): void {
-  withCrashRecovery(database, CHECKPOINT_KEY, () => {
-    const raw = getSqliteFrom(database);
-    raw.exec(
-      `CREATE INDEX IF NOT EXISTS idx_trace_events_created_at ON trace_events(created_at)`,
-    );
-  });
+  const raw = getSqliteFrom(database);
+  raw.exec(
+    `CREATE INDEX IF NOT EXISTS idx_trace_events_created_at ON trace_events(created_at)`,
+  );
 }

package/src/memory/migrations/242-message-bookmarks.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_message_bookmarks_v1";
 
 /**
  * Create the message_bookmarks table for user-saved message bookmarks.
@@ -13,26 +10,24 @@ const CHECKPOINT_KEY = "migration_message_bookmarks_v1";
  * be bookmarked at most once at a time).
  */
 export function migrateMessageBookmarks(database: DrizzleDb): void {
-  withCrashRecovery(database, CHECKPOINT_KEY, () => {
-    const raw = getSqliteFrom(database);
+  const raw = getSqliteFrom(database);
 
-    raw.exec(/*sql*/ `
-      CREATE TABLE IF NOT EXISTS message_bookmarks (
-        id TEXT PRIMARY KEY,
-        message_id TEXT NOT NULL REFERENCES messages(id) ON DELETE CASCADE,
-        conversation_id TEXT NOT NULL REFERENCES conversations(id) ON DELETE CASCADE,
-        created_at INTEGER NOT NULL
-      )
-    `);
+  raw.exec(/*sql*/ `
+    CREATE TABLE IF NOT EXISTS message_bookmarks (
+      id TEXT PRIMARY KEY,
+      message_id TEXT NOT NULL REFERENCES messages(id) ON DELETE CASCADE,
+      conversation_id TEXT NOT NULL REFERENCES conversations(id) ON DELETE CASCADE,
+      created_at INTEGER NOT NULL
+    )
+  `);
 
-    raw.exec(/*sql*/ `
-      CREATE UNIQUE INDEX IF NOT EXISTS message_bookmarks_message_id_uniq
-      ON message_bookmarks (message_id)
-    `);
+  raw.exec(/*sql*/ `
+    CREATE UNIQUE INDEX IF NOT EXISTS message_bookmarks_message_id_uniq
+    ON message_bookmarks (message_id)
+  `);
 
-    raw.exec(/*sql*/ `
-      CREATE INDEX IF NOT EXISTS message_bookmarks_created_at_idx
-      ON message_bookmarks (created_at)
-    `);
-  });
+  raw.exec(/*sql*/ `
+    CREATE INDEX IF NOT EXISTS message_bookmarks_created_at_idx
+    ON message_bookmarks (created_at)
+  `);
 }

package/src/memory/migrations/245-memory-retrospective-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_memory_retrospective_state_v1";
 
 /**
  * Create the memory_retrospective_state table.
@@ -22,15 +19,13 @@ const CHECKPOINT_KEY = "migration_memory_retrospective_state_v1";
  * conversation. Without the cascade, deleted conversations would leak rows.
  */
 export function migrateMemoryRetrospectiveState(database: DrizzleDb): void {
-  withCrashRecovery(database, CHECKPOINT_KEY, () => {
-    const raw = getSqliteFrom(database);
+  const raw = getSqliteFrom(database);
 
-    raw.exec(/*sql*/ `
-      CREATE TABLE IF NOT EXISTS memory_retrospective_state (
-        conversation_id TEXT PRIMARY KEY REFERENCES conversations(id) ON DELETE CASCADE,
-        last_processed_message_id TEXT NOT NULL,
-        last_run_at INTEGER NOT NULL
-      )
-    `);
-  });
+  raw.exec(/*sql*/ `
+    CREATE TABLE IF NOT EXISTS memory_retrospective_state (
+      conversation_id TEXT PRIMARY KEY REFERENCES conversations(id) ON DELETE CASCADE,
+      last_processed_message_id TEXT NOT NULL,
+      last_run_at INTEGER NOT NULL
+    )
+  `);
 }

package/src/memory/migrations/249-normalize-slack-external-content.ts

@@ -7,9 +7,7 @@ import {
   unwrapExternalContentForDisplay,
 } from "../../security/untrusted-content.js";
 import { type DrizzleDb, getSqliteFrom } from "../db-connection.js";
-import { withCrashRecovery } from "./validate-migration-state.js";
 
-const CHECKPOINT_KEY = "migration_normalize_slack_external_content_v1";
 const BATCH_SIZE = 100;
 
 interface CandidateMessageRow {
@@ -28,49 +26,47 @@ interface NormalizedMessageRow {
 export function migrateNormalizeSlackExternalContent(
   database: DrizzleDb,
 ): void {
-  withCrashRecovery(database, CHECKPOINT_KEY, () => {
-    const raw = getSqliteFrom(database);
+  const raw = getSqliteFrom(database);
 
-    const tableExists = raw
+  const tableExists = raw
+    .query(
+      `SELECT 1 FROM sqlite_master WHERE type = 'table' AND name = 'messages'`,
+    )
+    .get();
+  if (!tableExists) return;
+
+  let lastRowid = 0;
+
+  for (;;) {
+    const rows = raw
       .query(
-        `SELECT 1 FROM sqlite_master WHERE type = 'table' AND name = 'messages'`,
+        /*sql*/ `
+          SELECT rowid, id, role, content, metadata
+          FROM messages
+          WHERE rowid > ?
+            AND metadata LIKE '%"slackMeta"%'
+            AND (
+              content LIKE '%<external_content%'
+              OR metadata NOT LIKE '%"provenanceTrustClass"%'
+            )
+          ORDER BY rowid
+          LIMIT ?
+        `,
       )
-      .get();
-    if (!tableExists) return;
-
-    let lastRowid = 0;
-
-    for (;;) {
-      const rows = raw
-        .query(
-          /*sql*/ `
-            SELECT rowid, id, role, content, metadata
-            FROM messages
-            WHERE rowid > ?
-              AND metadata LIKE '%"slackMeta"%'
-              AND (
-                content LIKE '%<external_content%'
-                OR metadata NOT LIKE '%"provenanceTrustClass"%'
-              )
-            ORDER BY rowid
-            LIMIT ?
-          `,
-        )
-        .all(lastRowid, BATCH_SIZE) as CandidateMessageRow[];
-
-      if (rows.length === 0) break;
-
-      for (const row of rows) {
-        lastRowid = row.rowid;
-        const normalized = normalizeSlackMessageRow(row);
-        if (!normalized) continue;
-
-        raw
-          .query(`UPDATE messages SET content = ?, metadata = ? WHERE id = ?`)
-          .run(normalized.content, normalized.metadata, row.id);
-      }
+      .all(lastRowid, BATCH_SIZE) as CandidateMessageRow[];
+
+    if (rows.length === 0) break;
+
+    for (const row of rows) {
+      lastRowid = row.rowid;
+      const normalized = normalizeSlackMessageRow(row);
+      if (!normalized) continue;
+
+      raw
+        .query(`UPDATE messages SET content = ?, metadata = ? WHERE id = ?`)
+        .run(normalized.content, normalized.metadata, row.id);
     }
-  });
+  }
 }
 
 export function downNormalizeSlackExternalContent(_database: DrizzleDb): void {

package/src/memory/migrations/251-a2a-tasks.ts

@@ -1,9 +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";
-
-const CHECKPOINT_KEY = "migration_a2a_tasks_v1";
 
 /**
  * Create the a2a_tasks table for tracking A2A request/response lifecycle.
@@ -12,35 +9,33 @@ const CHECKPOINT_KEY = "migration_a2a_tasks_v1";
  * progression through submitted -> working -> completed/failed/canceled/rejected.
  */
 export function migrateA2ATasks(database: DrizzleDb): void {
-  withCrashRecovery(database, CHECKPOINT_KEY, () => {
-    if (tableHasColumn(database, "a2a_tasks", "id")) {
-      return;
-    }
-    const raw = getSqliteFrom(database);
-    raw.exec(/*sql*/ `
-      CREATE TABLE IF NOT EXISTS a2a_tasks (
-        id TEXT PRIMARY KEY,
-        context_id TEXT,
-        conversation_id TEXT,
-        state TEXT NOT NULL DEFAULT 'submitted',
-        status_message TEXT,
-        request_message_json TEXT NOT NULL,
-        artifacts_json TEXT,
-        push_url TEXT,
-        sender_assistant_id TEXT NOT NULL,
-        created_at INTEGER NOT NULL,
-        updated_at INTEGER NOT NULL
-      )
-    `);
-    raw.exec(/*sql*/ `
-      CREATE INDEX IF NOT EXISTS idx_a2a_tasks_context
-        ON a2a_tasks (context_id)
-    `);
-    raw.exec(/*sql*/ `
-      CREATE INDEX IF NOT EXISTS idx_a2a_tasks_conversation
-        ON a2a_tasks (conversation_id)
-    `);
-  });
+  if (tableHasColumn(database, "a2a_tasks", "id")) {
+    return;
+  }
+  const raw = getSqliteFrom(database);
+  raw.exec(/*sql*/ `
+    CREATE TABLE IF NOT EXISTS a2a_tasks (
+      id TEXT PRIMARY KEY,
+      context_id TEXT,
+      conversation_id TEXT,
+      state TEXT NOT NULL DEFAULT 'submitted',
+      status_message TEXT,
+      request_message_json TEXT NOT NULL,
+      artifacts_json TEXT,
+      push_url TEXT,
+      sender_assistant_id TEXT NOT NULL,
+      created_at INTEGER NOT NULL,
+      updated_at INTEGER NOT NULL
+    )
+  `);
+  raw.exec(/*sql*/ `
+    CREATE INDEX IF NOT EXISTS idx_a2a_tasks_context
+      ON a2a_tasks (context_id)
+  `);
+  raw.exec(/*sql*/ `
+    CREATE INDEX IF NOT EXISTS idx_a2a_tasks_conversation
+      ON a2a_tasks (conversation_id)
+  `);
 }
 
 export function downA2ATasks(database: DrizzleDb): void {

package/src/memory/migrations/254-external-conversation-binding-chat-name.ts

@@ -1,27 +1,22 @@
 import type { DrizzleDb } from "../db-connection.js";
 import { tableHasColumn } from "./schema-introspection.js";
-import { withCrashRecovery } from "./validate-migration-state.js";
-
-const CHECKPOINT_KEY = "migration_external_conversation_binding_chat_name_v1";
 
 export function migrateExternalConversationBindingChatName(
   database: DrizzleDb,
 ): void {
-  withCrashRecovery(database, CHECKPOINT_KEY, () => {
-    if (
-      tableHasColumn(
-        database,
-        "external_conversation_bindings",
-        "external_chat_name",
-      )
-    ) {
-      return;
-    }
+  if (
+    tableHasColumn(
+      database,
+      "external_conversation_bindings",
+      "external_chat_name",
+    )
+  ) {
+    return;
+  }
 
-    database.run(
-      `ALTER TABLE external_conversation_bindings ADD COLUMN external_chat_name TEXT`,
-    );
-  });
+  database.run(
+    `ALTER TABLE external_conversation_bindings ADD COLUMN external_chat_name TEXT`,
+  );
 }
 
 export function downExternalConversationBindingChatName(

package/src/memory/migrations/255-channel-inbound-delivery-attempts.ts

@@ -1,24 +1,19 @@
 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";
-
-const CHECKPOINT_KEY = "migration_channel_inbound_delivery_attempts_v1";
 
 /** Add a delivery-specific retry counter for channel inbound events. */
 export function migrateChannelInboundDeliveryAttempts(
   database: DrizzleDb,
 ): void {
-  withCrashRecovery(database, CHECKPOINT_KEY, () => {
-    if (
-      tableHasColumn(database, "channel_inbound_events", "delivery_attempts")
-    ) {
-      return;
-    }
-    const raw = getSqliteFrom(database);
-    raw.exec(/*sql*/ `
-      ALTER TABLE channel_inbound_events
-        ADD COLUMN delivery_attempts INTEGER NOT NULL DEFAULT 0
-    `);
-  });
+  if (
+    tableHasColumn(database, "channel_inbound_events", "delivery_attempts")
+  ) {
+    return;
+  }
+  const raw = getSqliteFrom(database);
+  raw.exec(/*sql*/ `
+    ALTER TABLE channel_inbound_events
+      ADD COLUMN delivery_attempts INTEGER NOT NULL DEFAULT 0
+  `);
 }