@vellumai/assistant 0.10.0 → 0.10.1-staging.1

package/src/__tests__/db-conversation-inference-profile-migration.test.ts

@@ -3,7 +3,6 @@ import { afterAll, beforeEach, describe, expect, mock, test } from "bun:test";
 
 import { drizzle } from "drizzle-orm/bun-sqlite";
 
-import { removeTestDbFiles } from "./assert-not-live-db.js";
 import { makeMockLogger } from "./helpers/mock-logger.js";
 
 const originalBunTest = process.env.BUN_TEST;
@@ -15,11 +14,9 @@ mock.module("../util/logger.js", () => ({
 import { _resetDisplayOrderMigrationForTests } from "../memory/conversation-display-order-migration.js";
 import { _resetGroupMigrationForTests } from "../memory/conversation-group-migration.js";
 import { getSqliteFrom } from "../memory/db-connection.js";
-import { initializeDb } from "../memory/db-init.js";
 import { migrateAddConversationInferenceProfile } from "../memory/migrations/227-add-conversation-inference-profile.js";
 import { migrateRenameInferenceProfileSnakeCase } from "../memory/migrations/228-rename-inference-profile-snake-case.js";
 import * as schema from "../memory/schema.js";
-import { getDbPath } from "../util/platform.js";
 import { resetDbForTesting } from "./db-test-helpers.js";
 
 function createTestDb() {
@@ -73,7 +70,6 @@ function resetMigrationTestDb(): void {
   // again. Otherwise downstream tests hit `no such column: group_id`.
   _resetGroupMigrationForTests();
   _resetDisplayOrderMigrationForTests();
-  removeTestDbFiles(getDbPath());
 }
 
 describe("conversation inference profile migration", () => {
@@ -88,9 +84,13 @@ describe("conversation inference profile migration", () => {
   });
 
   test("fresh DB initialization includes nullable inference_profile column", () => {
-    initializeDb();
+    const db = createTestDb();
+    const raw = getSqliteFrom(db);
+
+    bootstrapPreInferenceProfileConversations(raw);
+    migrateAddConversationInferenceProfile(db);
+    migrateRenameInferenceProfileSnakeCase(db);
 
-    const raw = new Database(getDbPath());
     const columns = getColumnNames(raw);
     // Migration 228 renames the camelCase column added by 227 to snake_case to
     // match the rest of the table.
@@ -106,7 +106,6 @@ describe("conversation inference profile migration", () => {
 
     expect(inferenceProfileColumn).toBeDefined();
     expect(inferenceProfileColumn?.notnull).toBe(0);
-    raw.close();
   });
 
   test("migration upgrades the previous schema without disturbing existing rows", () => {

package/src/__tests__/db-llm-request-log-provider-migration.test.ts

@@ -11,7 +11,6 @@ import {
 
 import { drizzle } from "drizzle-orm/bun-sqlite";
 
-import { removeTestDbFiles } from "./assert-not-live-db.js";
 const originalBunTest = process.env.BUN_TEST;
 
 mock.module("../util/logger.js", () => ({
@@ -22,10 +21,8 @@ mock.module("../util/logger.js", () => ({
 }));
 
 import { getSqliteFrom } from "../memory/db-connection.js";
-import { initializeDb } from "../memory/db-init.js";
 import { migrateLlmRequestLogProvider } from "../memory/migrations/184-llm-request-log-provider.js";
 import * as schema from "../memory/schema.js";
-import { getDbPath } from "../util/platform.js";
 import { resetDbForTesting } from "./db-test-helpers.js";
 
 function createTestDb() {
@@ -61,12 +58,10 @@ describe("llm_request_logs provider migration", () => {
   beforeEach(() => {
     process.env.BUN_TEST = "0";
     resetDbForTesting();
-    removeTestDbFiles(getDbPath());
   });
 
   afterEach(() => {
     resetDbForTesting();
-    removeTestDbFiles(getDbPath());
   });
 
   afterAll(() => {
@@ -76,21 +71,21 @@ describe("llm_request_logs provider migration", () => {
       process.env.BUN_TEST = originalBunTest;
     }
     resetDbForTesting();
-    removeTestDbFiles(getDbPath());
   });
 
   test("fresh DB initialization includes llm_request_logs.provider", () => {
-    initializeDb();
+    const db = createTestDb();
+    const raw = getSqliteFrom(db);
+
+    bootstrapPreProviderLlmRequestLogs(raw);
+    migrateLlmRequestLogProvider(db);
 
-    const raw = new Database(getDbPath());
     const columns = getColumnInfo(raw);
 
     expect(columns.some((column) => column.name === "provider")).toBe(true);
     expect(columns.find((column) => column.name === "provider")?.notnull).toBe(
       0,
     );
-
-    raw.close();
   });
 
   test("migration upgrades the pre-provider schema without disturbing rows", () => {

package/src/__tests__/db-migration-rollback.test.ts

@@ -16,11 +16,23 @@ import { afterEach, describe, expect, test } from "bun:test";
 
 import { drizzle } from "drizzle-orm/bun-sqlite";
 
-import { getSqliteFrom } from "../memory/db-connection.js";
-import { downJobDeferrals } from "../memory/migrations/001-job-deferrals.js";
-import { downMemoryEntityRelationDedup } from "../memory/migrations/004-entity-relation-dedup.js";
-import { downMemoryItemsFingerprintScopeUnique } from "../memory/migrations/005-fingerprint-scope-unique.js";
-import { downMemoryItemsScopeSaltedFingerprints } from "../memory/migrations/006-scope-salted-fingerprints.js";
+import { type DrizzleDb, getSqliteFrom } from "../memory/db-connection.js";
+import {
+  downJobDeferrals,
+  migrateJobDeferrals,
+} from "../memory/migrations/001-job-deferrals.js";
+import {
+  downMemoryEntityRelationDedup,
+  migrateMemoryEntityRelationDedup,
+} from "../memory/migrations/004-entity-relation-dedup.js";
+import {
+  downMemoryItemsFingerprintScopeUnique,
+  migrateMemoryItemsFingerprintScopeUnique,
+} from "../memory/migrations/005-fingerprint-scope-unique.js";
+import {
+  downMemoryItemsScopeSaltedFingerprints,
+  migrateMemoryItemsScopeSaltedFingerprints,
+} from "../memory/migrations/006-scope-salted-fingerprints.js";
 import { downAssistantIdToSelf } from "../memory/migrations/007-assistant-id-to-self.js";
 import { downRemoveAssistantIdColumns } from "../memory/migrations/008-remove-assistant-id-columns.js";
 import { downLlmUsageEventsDropAssistantId } from "../memory/migrations/009-llm-usage-events-drop-assistant-id.js";
@@ -55,18 +67,17 @@ import { migrateDropCapabilityCardStateDown } from "../memory/migrations/176-dro
 import { migrateBackfillInlineAttachmentsToDiskDown } from "../memory/migrations/180-backfill-inline-attachments-to-disk.js";
 import { migrateRenameThreadStartersCheckpointsDown } from "../memory/migrations/181-rename-thread-starters-checkpoints.js";
 import { migrateBackfillAudioAttachmentMimeTypesDown } from "../memory/migrations/191-backfill-audio-attachment-mime-types.js";
+import { migrateLlmUsageAttribution } from "../memory/migrations/235-llm-usage-attribution.js";
 import {
-  migrateJobDeferrals,
-  migrateLlmUsageAttribution,
-  migrateMemoryEntityRelationDedup,
-  migrateMemoryItemsFingerprintScopeUnique,
-  migrateMemoryItemsScopeSaltedFingerprints,
   MIGRATION_REGISTRY,
   type MigrationRegistryEntry,
   type MigrationValidationResult,
+} from "../memory/migrations/registry.js";
+import { runMigrationSteps } from "../memory/migrations/run-migrations.js";
+import {
   rollbackMemoryMigration,
   validateMigrationState,
-} from "../memory/migrations/index.js";
+} from "../memory/migrations/validate-migration-state.js";
 import * as schema from "../memory/schema.js";
 
 // ---------------------------------------------------------------------------
@@ -525,11 +536,11 @@ describe("schema-drift recovery: migration handles unexpected schema state", ()
 
     // Insert a "started" checkpoint — simulates mid-migration crash.
     raw.exec(
-      `INSERT INTO memory_checkpoints (key, value, updated_at) VALUES ('migration_job_deferrals', 'started', ${now})`,
+      `INSERT INTO memory_checkpoints (key, value, updated_at) VALUES ('step:migrateJobDeferrals', 'started', ${now})`,
     );
     // A completed checkpoint should not be flagged.
     raw.exec(
-      `INSERT INTO memory_checkpoints (key, value, updated_at) VALUES ('migration_memory_entity_relations_dedup_v1', '1', ${now})`,
+      `INSERT INTO memory_checkpoints (key, value, updated_at) VALUES ('step:migrateMemoryEntityRelationDedup', '1', ${now})`,
     );
 
     // validateMigrationState logs warnings for crashed migrations and returns
@@ -537,9 +548,9 @@ describe("schema-drift recovery: migration handles unexpected schema state", ()
     // than re-deriving the crashed list from the raw DB — this verifies the
     // function itself detects the crash, not just that the data is present.
     const result: MigrationValidationResult = validateMigrationState(db);
-    expect(result.crashed).toContain("migration_job_deferrals");
+    expect(result.crashed).toContain("step:migrateJobDeferrals");
     expect(result.crashed).not.toContain(
-      "migration_memory_entity_relations_dedup_v1",
+      "step:migrateMemoryEntityRelationDedup",
     );
   });
 
@@ -554,11 +565,12 @@ describe("schema-drift recovery: migration handles unexpected schema state", ()
 
     const now = Date.now();
 
-    // Write the child migration (salted fingerprints) but NOT its parent
-    // (fingerprint_scope_unique). This violates the declared dependsOn.
+    // Write the child migration (salted fingerprints) step checkpoint but NOT
+    // its parent (fingerprint_scope_unique). This violates the declared
+    // dependsOn — the runner records `step:<stepName>` for completed steps.
     raw.exec(`
       INSERT INTO memory_checkpoints (key, value, updated_at)
-      VALUES ('migration_memory_items_scope_salted_fingerprints_v1', '1', ${now})
+      VALUES ('step:migrateMemoryItemsScopeSaltedFingerprints', '1', ${now})
     `);
 
     // validateMigrationState throws an IntegrityError on dependency violations
@@ -959,6 +971,7 @@ describe("rollbackMemoryMigration", () => {
     const testEntries: MigrationRegistryEntry[] = [
       {
         key: "test_rollback_v1000",
+        stepName: "test_rollback_v1000",
         version: 1000,
         description: "test migration v1000",
         down: () => {
@@ -967,6 +980,7 @@ describe("rollbackMemoryMigration", () => {
       },
       {
         key: "test_rollback_v1001",
+        stepName: "test_rollback_v1001",
         version: 1001,
         description: "test migration v1001",
         down: () => {
@@ -975,6 +989,7 @@ describe("rollbackMemoryMigration", () => {
       },
       {
         key: "test_rollback_v1002",
+        stepName: "test_rollback_v1002",
         version: 1002,
         description: "test migration v1002",
         down: () => {
@@ -985,10 +1000,10 @@ describe("rollbackMemoryMigration", () => {
 
     MIGRATION_REGISTRY.push(...testEntries);
 
-    // Simulate all three migrations as completed.
+    // Simulate all three migrations as completed via their step checkpoints.
     for (const entry of testEntries) {
       raw.exec(
-        `INSERT INTO memory_checkpoints (key, value, updated_at) VALUES ('${entry.key}', '1', ${now})`,
+        `INSERT INTO memory_checkpoints (key, value, updated_at) VALUES ('step:${entry.stepName}', '1', ${now})`,
       );
     }
 
@@ -1004,14 +1019,14 @@ describe("rollbackMemoryMigration", () => {
     // Checkpoints for rolled-back migrations should be deleted.
     const cp1001 = raw
       .query(
-        `SELECT 1 FROM memory_checkpoints WHERE key = 'test_rollback_v1001'`,
+        `SELECT 1 FROM memory_checkpoints WHERE key = 'step:test_rollback_v1001'`,
       )
       .get();
     expect(cp1001).toBeNull();
 
     const cp1002 = raw
       .query(
-        `SELECT 1 FROM memory_checkpoints WHERE key = 'test_rollback_v1002'`,
+        `SELECT 1 FROM memory_checkpoints WHERE key = 'step:test_rollback_v1002'`,
       )
       .get();
     expect(cp1002).toBeNull();
@@ -1019,13 +1034,78 @@ describe("rollbackMemoryMigration", () => {
     // Checkpoint for the migration at target version should still exist.
     const cp1000 = raw
       .query(
-        `SELECT value FROM memory_checkpoints WHERE key = 'test_rollback_v1000'`,
+        `SELECT value FROM memory_checkpoints WHERE key = 'step:test_rollback_v1000'`,
       )
       .get() as { value: string } | null;
     expect(cp1000).toBeTruthy();
     expect(cp1000!.value).toBe("1");
   });
 
+  test("clears forward-step checkpoints so a re-upgrade re-applies a rolled-back registry-backed step", async () => {
+    /**
+     * A registry-backed migration that is also a named forward step is tracked
+     * by the runner's `step:` checkpoint in memory_checkpoints. Rolling it back
+     * must clear the step checkpoint too, otherwise the next upgrade skips the
+     * step and never restores the rolled-back schema.
+     */
+    saveRegistry();
+
+    const db = createTestDb();
+    const raw = getRaw(db);
+    bootstrapCheckpointsTable(raw);
+
+    const hasTable = () =>
+      raw
+        .query(
+          `SELECT name FROM sqlite_master WHERE type = 'table' AND name = 'test_step_rollback_data'`,
+        )
+        .get() !== null;
+
+    // GIVEN a registry-backed migration whose forward body is a named step that
+    // creates a table. The step runner records `step:<functionName>` after a
+    // successful run; the registry entry's stepName maps to that key so rollback
+    // can find and clear it.
+    function migrateTestStepRollback(database: DrizzleDb): void {
+      getSqliteFrom(database).exec(
+        `CREATE TABLE IF NOT EXISTS test_step_rollback_data (id TEXT PRIMARY KEY)`,
+      );
+    }
+
+    // AND the migration is registered with a down() that drops that table.
+    MIGRATION_REGISTRY.push({
+      key: "test_step_rollback_v4000",
+      stepName: "migrateTestStepRollback",
+      version: 4000,
+      description: "test step rollback",
+      down: (database) => {
+        getSqliteFrom(database).exec(
+          `DROP TABLE IF EXISTS test_step_rollback_data`,
+        );
+      },
+    });
+
+    // AND the step has run once through the runner, recording the step checkpoint.
+    await runMigrationSteps(db, [migrateTestStepRollback]);
+    expect(hasTable()).toBe(true);
+    expect(
+      raw
+        .query(
+          `SELECT 1 FROM memory_checkpoints WHERE key = 'step:migrateTestStepRollback'`,
+        )
+        .get(),
+    ).toBeTruthy();
+
+    // AND the migration has since been rolled back below its version.
+    rollbackMemoryMigration(db, 3999);
+    expect(hasTable()).toBe(false);
+
+    // WHEN the runner executes the forward steps again on a later upgrade.
+    await runMigrationSteps(db, [migrateTestStepRollback]);
+
+    // THEN the rolled-back step re-applies instead of being skipped.
+    expect(hasTable()).toBe(true);
+  });
+
   test("handles transaction failure in down() — rolls back and preserves checkpoint", () => {
     saveRegistry();
 
@@ -1050,6 +1130,7 @@ describe("rollbackMemoryMigration", () => {
     // but a trigger will force the modification to fail.
     MIGRATION_REGISTRY.push({
       key: "test_fail_down_v3000",
+      stepName: "test_fail_down_v3000",
       version: 3000,
       description: "test migration with failing down()",
       down: (database) => {
@@ -1061,9 +1142,9 @@ describe("rollbackMemoryMigration", () => {
       },
     });
 
-    // Mark as completed.
+    // Mark as completed (via step checkpoint).
     raw.exec(
-      `INSERT INTO memory_checkpoints (key, value, updated_at) VALUES ('test_fail_down_v3000', '1', ${now})`,
+      `INSERT INTO memory_checkpoints (key, value, updated_at) VALUES ('step:test_fail_down_v3000', '1', ${now})`,
     );
 
     // Install a trigger to force the down() function to fail.
@@ -1091,7 +1172,7 @@ describe("rollbackMemoryMigration", () => {
     // written before down() was called and is preserved.
     const cp = raw
       .query(
-        `SELECT value FROM memory_checkpoints WHERE key = 'test_fail_down_v3000'`,
+        `SELECT value FROM memory_checkpoints WHERE key = 'step:test_fail_down_v3000'`,
       )
       .get() as { value: string } | null;
     expect(cp).toBeTruthy();
@@ -1130,6 +1211,7 @@ describe("rollbackMemoryMigration", () => {
     // wrapped every down() call in BEGIN/COMMIT.
     MIGRATION_REGISTRY.push({
       key: "test_self_txn_down_v3500",
+      stepName: "test_self_txn_down_v3500",
       version: 3500,
       description: "test migration with self-transactional down()",
       down: (database) => {
@@ -1142,9 +1224,9 @@ describe("rollbackMemoryMigration", () => {
       },
     });
 
-    // Mark as completed.
+    // Mark as completed (via step checkpoint).
     raw.exec(
-      `INSERT INTO memory_checkpoints (key, value, updated_at) VALUES ('test_self_txn_down_v3500', '1', ${now})`,
+      `INSERT INTO memory_checkpoints (key, value, updated_at) VALUES ('step:test_self_txn_down_v3500', '1', ${now})`,
     );
 
     // This should succeed — no nested transaction error.
@@ -1162,7 +1244,7 @@ describe("rollbackMemoryMigration", () => {
     // Checkpoint should be deleted.
     const cp = raw
       .query(
-        `SELECT 1 FROM memory_checkpoints WHERE key = 'test_self_txn_down_v3500'`,
+        `SELECT 1 FROM memory_checkpoints WHERE key = 'step:test_self_txn_down_v3500'`,
       )
       .get();
     expect(cp).toBeNull();
@@ -1183,6 +1265,7 @@ describe("rollbackMemoryMigration", () => {
     MIGRATION_REGISTRY.push(
       {
         key: "test_noop_v4000",
+        stepName: "test_noop_v4000",
         version: 4000,
         description: "test noop v4000",
         down: () => {
@@ -1191,6 +1274,7 @@ describe("rollbackMemoryMigration", () => {
       },
       {
         key: "test_noop_v4001",
+        stepName: "test_noop_v4001",
         version: 4001,
         description: "test noop v4001",
         down: () => {
@@ -1199,12 +1283,12 @@ describe("rollbackMemoryMigration", () => {
       },
     );
 
-    // Mark both as completed.
+    // Mark both as completed (via step checkpoints).
     raw.exec(
-      `INSERT INTO memory_checkpoints (key, value, updated_at) VALUES ('test_noop_v4000', '1', ${now})`,
+      `INSERT INTO memory_checkpoints (key, value, updated_at) VALUES ('step:test_noop_v4000', '1', ${now})`,
     );
     raw.exec(
-      `INSERT INTO memory_checkpoints (key, value, updated_at) VALUES ('test_noop_v4001', '1', ${now})`,
+      `INSERT INTO memory_checkpoints (key, value, updated_at) VALUES ('step:test_noop_v4001', '1', ${now})`,
     );
 
     // Roll back to version >= latest applied migration — should be a no-op.
@@ -1216,12 +1300,12 @@ describe("rollbackMemoryMigration", () => {
     // Both checkpoints should remain.
     const cp4000 = raw
       .query(
-        `SELECT value FROM memory_checkpoints WHERE key = 'test_noop_v4000'`,
+        `SELECT value FROM memory_checkpoints WHERE key = 'step:test_noop_v4000'`,
       )
       .get() as { value: string } | null;
     const cp4001 = raw
       .query(
-        `SELECT value FROM memory_checkpoints WHERE key = 'test_noop_v4001'`,
+        `SELECT value FROM memory_checkpoints WHERE key = 'step:test_noop_v4001'`,
       )
       .get() as { value: string } | null;
     expect(cp4000!.value).toBe("1");
@@ -1251,6 +1335,7 @@ describe("rollbackMemoryMigration", () => {
     MIGRATION_REGISTRY.push(
       {
         key: "test_parent_v5000",
+        stepName: "test_parent_v5000",
         version: 5000,
         description: "test parent migration",
         down: () => {
@@ -1259,6 +1344,7 @@ describe("rollbackMemoryMigration", () => {
       },
       {
         key: "test_child_v5001",
+        stepName: "test_child_v5001",
         version: 5001,
         dependsOn: ["test_parent_v5000"],
         description: "test child migration depending on parent",
@@ -1268,12 +1354,12 @@ describe("rollbackMemoryMigration", () => {
       },
     );
 
-    // Both are completed.
+    // Both are completed (via step checkpoints).
     raw.exec(
-      `INSERT INTO memory_checkpoints (key, value, updated_at) VALUES ('test_parent_v5000', '1', ${now})`,
+      `INSERT INTO memory_checkpoints (key, value, updated_at) VALUES ('step:test_parent_v5000', '1', ${now})`,
     );
     raw.exec(
-      `INSERT INTO memory_checkpoints (key, value, updated_at) VALUES ('test_child_v5001', '1', ${now})`,
+      `INSERT INTO memory_checkpoints (key, value, updated_at) VALUES ('step:test_child_v5001', '1', ${now})`,
     );
 
     // Roll back to version 4999 — both should be rolled back, child first.
@@ -1286,10 +1372,14 @@ describe("rollbackMemoryMigration", () => {
 
     // Both checkpoints should be deleted.
     const cpParent = raw
-      .query(`SELECT 1 FROM memory_checkpoints WHERE key = 'test_parent_v5000'`)
+      .query(
+        `SELECT 1 FROM memory_checkpoints WHERE key = 'step:test_parent_v5000'`,
+      )
       .get();
     const cpChild = raw
-      .query(`SELECT 1 FROM memory_checkpoints WHERE key = 'test_child_v5001'`)
+      .query(
+        `SELECT 1 FROM memory_checkpoints WHERE key = 'step:test_child_v5001'`,
+      )
       .get();
     expect(cpParent).toBeNull();
     expect(cpChild).toBeNull();

package/src/__tests__/db-proxy-transaction.test.ts

@@ -15,7 +15,7 @@ import { getSqlite } from "../memory/db-connection.js";
 import { initializeDb } from "../memory/db-init.js";
 import { RouteError } from "../runtime/routes/errors.js";
 
-initializeDb();
+await initializeDb();
 
 function resetTestTable(): void {
   const sqlite = getSqlite();

package/src/__tests__/db-schedule-syntax-migration.test.ts

@@ -306,16 +306,5 @@ describe("schedule_syntax column migration", () => {
       { id: "legacy-defer", description: "" },
       { id: "legacy-schedule", description: "Legacy report" },
     ]);
-
-    raw
-      .query("UPDATE cron_jobs SET description = ? WHERE id = ?")
-      .run("", "legacy-schedule");
-
-    migrateScheduleDescription(db);
-
-    const description = raw
-      .query("SELECT description FROM cron_jobs WHERE id = ?")
-      .get("legacy-schedule") as { description: string };
-    expect(description.description).toBe("");
   });
 });

package/src/__tests__/db-test-helpers.ts

@@ -1,5 +1,5 @@
 /**
- * Test-only utilities for resetting the assistant DB singleton.
+ * Test-only utilities for resetting the assistant DB singletons.
  *
  * Replaces the removed `resetDb` export from `db-connection.ts`. Lives
  * here (not in the source module) because production modules should not
@@ -7,52 +7,69 @@
  *
  * No source-module imports
  * ------------------------
- * This file has ZERO imports from `src/`. It accesses the DB singleton's
- * state via the shared `globalThis.vellumAssistant.dbSingleton` slot
+ * This file has ZERO imports from `src/`. It accesses the DB singletons'
+ * state via the shared `globalThis.vellumAssistant.dbSingletons` slots
  * that `src/memory/db-singleton.ts` also reads/writes. The slot shape
  * is duplicated here on purpose: keeping this file off the production
  * import graph is what protects the test preload from a broken
  * `node_modules` symlink (DB ghost #3). The two declarations MUST stay
  * in sync — if you change one, change the other.
  *
- * Production code that needs to close + reopen the DB (post-migration,
+ * Production code that needs to close + reopen the DBs (post-migration,
  * post-restore, post-vbundle-import, on shutdown) should use `resetDb()`
  * from `src/memory/db-connection.ts` instead.
  */
 
 // Mirrors `src/memory/db-singleton.ts`. Duplicated by design — see the
 // "No source-module imports" section above.
+type DbSlotKey = "main" | "logs" | "memory";
+
 type DbSlot = {
   db: unknown;
   closer: (() => void) | null;
 };
 
+type DbSlots = Record<DbSlotKey, DbSlot>;
+
 type VellumAssistantNamespace = {
-  dbSingleton?: DbSlot;
+  dbSingletons?: DbSlots;
 };
 
-function dbSlot(): DbSlot {
+function emptySlot(): DbSlot {
+  return { db: null, closer: null };
+}
+
+function dbSlots(): DbSlots {
   const g = globalThis as { vellumAssistant?: VellumAssistantNamespace };
   const ns = (g.vellumAssistant ??= {});
-  return (ns.dbSingleton ??= { db: null, closer: null });
+  return (ns.dbSingletons ??= {
+    main: emptySlot(),
+    logs: emptySlot(),
+    memory: emptySlot(),
+  });
 }
 
 /**
- * Close the active DB connection (if any) and drop the singleton.
+ * Close every active DB connection (main, logs, memory) and drop the
+ * singletons.
  *
- * Used by tests that nuke or replace the DB file mid-run — without this
- * reset, subsequent `getDb()` calls return a handle to the now-gone file.
- * Idempotent: safe to call when no connection has been opened.
+ * Used by tests that nuke or replace a DB file mid-run — without this
+ * reset, subsequent `getDb()`/`getLogsDb()`/`getMemoryDb()` calls return a
+ * handle to the now-gone file. Idempotent: safe to call when no connection
+ * has been opened.
  */
 export function resetDbForTesting(): void {
-  const s = dbSlot();
-  if (s.closer) {
-    try {
-      s.closer();
-    } catch {
-      /* best-effort close */
+  const slots = dbSlots();
+  for (const key of ["main", "logs", "memory"] as const) {
+    const s = slots[key];
+    if (s.closer) {
+      try {
+        s.closer();
+      } catch {
+        /* best-effort close */
+      }
     }
+    s.db = null;
+    s.closer = null;
   }
-  s.db = null;
-  s.closer = null;
 }

package/src/__tests__/delete-propagation.test.ts

@@ -35,7 +35,7 @@ import {
 import { _setDeleteLookupConfigForTests } from "../runtime/routes/inbound-message-handler.js";
 import { handleChannelInbound } from "./helpers/channel-test-adapter.js";
 
-initializeDb();
+await initializeDb();
 
 const TEST_BEARER_TOKEN = "test-token";
 // Slack `message_deleted` events stamp the deleted message's original author