@vellumai/assistant 0.10.0 → 0.10.1-staging.1

package/src/__tests__/conversation-process-callsite.test.ts

@@ -20,15 +20,11 @@ import type { Message, ProviderResponse } from "../providers/types.js";
 const captured: {
   callSite?: string;
   constructorMaxTokens?: unknown;
-  resolvedMaxTokens?: unknown;
-  resolvedHasMaxTokens?: boolean;
 } = {};
 
 function clearCaptured(): void {
   captured.callSite = undefined;
   captured.constructorMaxTokens = undefined;
-  captured.resolvedMaxTokens = undefined;
-  captured.resolvedHasMaxTokens = undefined;
 }
 
 mock.module("../util/logger.js", () => ({
@@ -207,14 +203,8 @@ mock.module("../agent/loop.js", () => ({
       provider?: unknown;
       systemPrompt?: string;
       config?: Record<string, unknown>;
-      resolveSystemPrompt?: (history: Message[]) => Record<string, unknown>;
     }) {
       captured.constructorMaxTokens = options?.config?.maxTokens;
-      const resolved = options?.resolveSystemPrompt?.([]);
-      captured.resolvedMaxTokens = resolved?.maxTokens;
-      captured.resolvedHasMaxTokens =
-        resolved !== undefined &&
-        Object.prototype.hasOwnProperty.call(resolved, "maxTokens");
     }
     getToolTokenBudget() {
       return 0;
@@ -246,6 +236,12 @@ mock.module("../agent/loop.js", () => ({
 
 mock.module("../plugins/defaults/compaction/window-manager.js", () => ({
   ContextWindowManager: class {
+    estimateInputTokens() {
+      return 0;
+    }
+    get tokenCountInputs() {
+      return { systemPrompt: "", tools: undefined };
+    }
     constructor() {}
     updateConfig() {}
     shouldCompact() {
@@ -375,8 +371,6 @@ describe("processMessage callSite threading", () => {
     await getOrCreateConversation("conv-store-default");
 
     expect(captured.constructorMaxTokens).toBeUndefined();
-    expect(captured.resolvedMaxTokens).toBeUndefined();
-    expect(captured.resolvedHasMaxTokens).toBe(false);
   });
 
   test("preserves explicit maxResponseTokens at conversation creation", async () => {
@@ -397,8 +391,6 @@ describe("processMessage callSite threading", () => {
     });
 
     expect(captured.constructorMaxTokens).toBe(1234);
-    expect(captured.resolvedMaxTokens).toBe(1234);
-    expect(captured.resolvedHasMaxTokens).toBe(true);
   });
 
   test("applies clientTimezone in the create and reuse transport metadata path", async () => {

package/src/__tests__/conversation-provider-retry-repair.test.ts

@@ -147,6 +147,12 @@ mock.module("../memory/retriever.js", () => ({
 
 mock.module("../plugins/defaults/compaction/window-manager.js", () => ({
   ContextWindowManager: class {
+    estimateInputTokens() {
+      return 0;
+    }
+    get tokenCountInputs() {
+      return { systemPrompt: "", tools: undefined };
+    }
     conversationId: string | undefined;
     constructor(options?: { conversationId?: string }) {
       this.conversationId = options?.conversationId;

package/src/__tests__/conversation-queue.test.ts

@@ -259,6 +259,12 @@ mock.module("../memory/retriever.js", () => ({
 
 mock.module("../plugins/defaults/compaction/window-manager.js", () => ({
   ContextWindowManager: class {
+    estimateInputTokens() {
+      return 0;
+    }
+    get tokenCountInputs() {
+      return { systemPrompt: "", tools: undefined };
+    }
     constructor() {}
     updateConfig() {}
     shouldCompact() {

package/src/__tests__/conversation-routes-disk-view.test.ts

@@ -90,7 +90,7 @@ mock.module("../config/loader.js", () => ({
   }),
 }));
 
-initializeDb();
+await initializeDb();
 
 const conversationInstances = new Map<string, Conversation>();
 

package/src/__tests__/conversation-runtime-assembly.test.ts

@@ -33,6 +33,40 @@ mock.module("../config/loader.js", () => ({
   },
 }));
 
+// `resolveTurnInboundActorContext` reads INFO (contact notes / interaction
+// count) via a local contactId join. Stub the join so the actor-context tests
+// can stage it without standing up the contacts schema.
+const realContactStoreForAssemblyTest =
+  await import("../contacts/contact-store.js");
+let contactInfoById: Record<
+  string,
+  { notes: string | null; interactionCount: number }
+> = {};
+mock.module("../contacts/contact-store.js", () => ({
+  ...realContactStoreForAssemblyTest,
+  findContactInfoById: (contactId: string) =>
+    contactInfoById[contactId] ?? null,
+}));
+
+// `resolveTurnInboundActorContext` must NOT re-resolve ACL via the actor-trust
+// resolver on the fresh-inbound path. Track calls so the tests can assert it
+// is never invoked.
+const realActorTrustResolverForAssemblyTest = await import(
+  "../runtime/actor-trust-resolver.js"
+);
+let resolveActorTrustCalls = 0;
+mock.module("../runtime/actor-trust-resolver.js", () => ({
+  ...realActorTrustResolverForAssemblyTest,
+  resolveActorTrust: (
+    ...args: Parameters<
+      typeof realActorTrustResolverForAssemblyTest.resolveActorTrust
+    >
+  ) => {
+    resolveActorTrustCalls += 1;
+    return realActorTrustResolverForAssemblyTest.resolveActorTrust(...args);
+  },
+}));
+
 // PKB search is mocked so the reminder-hints tests can assert behavior
 // without standing up Qdrant. The mock returns whatever is staged in
 // `pkbSearchResults` / `pkbSearchThrows` for the enclosing test.
@@ -88,6 +122,7 @@ import {
   loadSlackActiveThreadFocusBlock,
   loadSlackChronologicalContext,
   loadSlackChronologicalMessages,
+  NON_INTERACTIVE_CONTEXT_BLOCK,
   resolveChannelCapabilities,
   resolveTurnInboundActorContext,
   resolveTurnModelProfileLabel,
@@ -123,7 +158,7 @@ import { getWorkspacePromptPath } from "../util/platform.js";
 // `applyRuntimeInjections` self-resolves the Slack active-thread focus block by
 // reading the live conversation's persisted message rows, so the schema must
 // exist before any Slack-channel assembly test runs.
-initializeDb();
+await initializeDb();
 
 // The pkb-reminder injector derives PKB-active state from the workspace itself
 // — `readPkbContext()` returning content behind the personal-memory trust gate
@@ -941,6 +976,32 @@ describe("applyRuntimeInjections — injection mode", () => {
     expect(allText).toContain("<knowledge_base>");
   });
 
+  test("captures background-turn / channel-capabilities / non-interactive-context for metadata persistence", async () => {
+    // These three blocks must be captured into `blocks` so persistInjectionBlocks
+    // writes them to message metadata and loadFromDb rehydrates them on a
+    // reload/fork (background-source memory retrospective cache parity — see the
+    // rehydration-order tests in conversation-lifecycle.test.ts). background-turn
+    // fires only for a background/scheduled live conversation, so register one
+    // under the turn's conversation id.
+    setConversation("injection-mode-conv", {
+      conversationId: "injection-mode-conv",
+      workingDir: "/sandbox",
+      workspaceTopLevelContext: "",
+      workspaceTopLevelDirty: false,
+      surfaceState: new Map(),
+      channelCapabilities,
+      conversationType: "background",
+    } as never);
+
+    const { blocks } = await applyRuntimeInjections(baseMessages, fullOptions);
+
+    expect(blocks.backgroundTurnBlock).toContain("<background_turn>");
+    expect(blocks.channelCapabilitiesBlock).toContain("<channel_capabilities>");
+    expect(blocks.nonInteractiveContextBlock).toBe(
+      NON_INTERACTIVE_CONTEXT_BLOCK,
+    );
+  });
+
   test("explicit mode: 'full' behaves the same as default", async () => {
     const { messages: result } = await applyRuntimeInjections(baseMessages, {
       ...fullOptions,
@@ -1449,6 +1510,11 @@ describe("applyRuntimeInjections with nowScratchpad", () => {
 // ---------------------------------------------------------------------------
 
 describe("resolveTurnInboundActorContext", () => {
+  beforeEach(() => {
+    contactInfoById = {};
+    resolveActorTrustCalls = 0;
+  });
+
   /**
    * No trust context means there is no inbound actor to ground the turn on, so
    * the actor section is suppressed.
@@ -1456,7 +1522,7 @@ describe("resolveTurnInboundActorContext", () => {
   test("returns null when there is no trust context", () => {
     // GIVEN a turn with no trust context
     // WHEN the inbound actor context is resolved
-    const actorContext = resolveTurnInboundActorContext(undefined, "asst-1");
+    const actorContext = resolveTurnInboundActorContext(undefined);
 
     // THEN there is no actor context to inject
     expect(actorContext).toBeNull();
@@ -1467,8 +1533,7 @@ describe("resolveTurnInboundActorContext", () => {
    * only renders actor fields for non-guardian actors.
    */
   test("returns null on guardian turns", () => {
-    // GIVEN a guardian trust context (no requester chat id, so the direct
-    // projection branch is taken)
+    // GIVEN a guardian trust context
     const trustContext: TrustContext = {
       sourceChannel: "vellum",
       trustClass: "guardian",
@@ -1476,40 +1541,78 @@ describe("resolveTurnInboundActorContext", () => {
     };
 
     // WHEN the inbound actor context is resolved
-    const actorContext = resolveTurnInboundActorContext(trustContext, "asst-1");
+    const actorContext = resolveTurnInboundActorContext(trustContext);
 
     // THEN the actor section is suppressed for the owner
     expect(actorContext).toBeNull();
   });
 
   /**
-   * A non-guardian trust context without the identity fields needed for the
-   * unified actor-trust lookup is projected directly into the actor context.
+   * ACL fields (trust class, member status/policy, guardian identity) project
+   * from the verdict-derived trust context — without re-resolving via the
+   * actor-trust resolver.
    */
-  test("projects a non-guardian trust context directly", () => {
-    // GIVEN a trusted-contact trust context lacking requester chat/user ids
+  test("projects ACL from the verdict-derived trust context", () => {
+    // GIVEN a trusted-contact trust context carrying verdict-derived ACL fields
     const trustContext: TrustContext = {
       sourceChannel: "telegram",
       trustClass: "trusted_contact",
       requesterIdentifier: "@bob_handle",
       requesterDisplayName: "Bob",
       requesterSenderDisplayName: "Bobby",
+      requesterExternalUserId: "tg-123",
+      requesterChatId: "chat-1",
+      memberStatus: "active",
+      memberPolicy: "allow",
     };
 
     // WHEN the inbound actor context is resolved
-    const actorContext = resolveTurnInboundActorContext(trustContext, "asst-1");
+    const actorContext = resolveTurnInboundActorContext(trustContext);
 
-    // THEN the trust context is projected into the model-facing actor context
+    // THEN ACL is projected from the context and no ACL re-read happened
     expect(actorContext).toEqual({
       sourceChannel: "telegram",
-      canonicalActorIdentity: null,
+      canonicalActorIdentity: "tg-123",
       actorIdentifier: "@bob_handle",
       actorDisplayName: "Bob",
       actorSenderDisplayName: "Bobby",
       actorMemberDisplayName: undefined,
       trustClass: "trusted_contact",
       guardianIdentity: undefined,
+      memberStatus: "active",
+      memberPolicy: "allow",
     });
+    expect(resolveActorTrustCalls).toBe(0);
+  });
+
+  /**
+   * INFO fields (contact notes, interaction count) are joined locally by the
+   * verdict-stamped contact ID, not re-resolved through the ACL store.
+   */
+  test("populates INFO via the local contactId join", () => {
+    // GIVEN a contact whose INFO is available by id
+    contactInfoById["contact-42"] = {
+      notes: "prefers async updates",
+      interactionCount: 7,
+    };
+    const trustContext: TrustContext = {
+      sourceChannel: "telegram",
+      trustClass: "trusted_contact",
+      requesterExternalUserId: "tg-123",
+      requesterChatId: "chat-1",
+      requesterContactId: "contact-42",
+      memberStatus: "active",
+      memberPolicy: "allow",
+    };
+
+    // WHEN the inbound actor context is resolved
+    const actorContext = resolveTurnInboundActorContext(trustContext);
+
+    // THEN INFO comes from the local join and ACL stays verdict-derived
+    expect(actorContext?.contactNotes).toBe("prefers async updates");
+    expect(actorContext?.contactInteractionCount).toBe(7);
+    expect(actorContext?.memberStatus).toBe("active");
+    expect(resolveActorTrustCalls).toBe(0);
   });
 });
 

package/src/__tests__/conversation-slash-queue.test.ts

@@ -150,6 +150,12 @@ mock.module("../memory/retriever.js", () => ({
 
 mock.module("../plugins/defaults/compaction/window-manager.js", () => ({
   ContextWindowManager: class {
+    estimateInputTokens() {
+      return 0;
+    }
+    get tokenCountInputs() {
+      return { systemPrompt: "", tools: undefined };
+    }
     constructor() {}
     updateConfig() {}
     shouldCompact() {

package/src/__tests__/conversation-slash-unknown.test.ts

@@ -151,6 +151,12 @@ mock.module("../memory/retriever.js", () => ({
 
 mock.module("../plugins/defaults/compaction/window-manager.js", () => ({
   ContextWindowManager: class {
+    estimateInputTokens() {
+      return 0;
+    }
+    get tokenCountInputs() {
+      return { systemPrompt: "", tools: undefined };
+    }
     constructor() {}
     updateConfig() {}
     shouldCompact() {

package/src/__tests__/conversation-speed-override.test.ts

@@ -167,6 +167,12 @@ mock.module("../memory/retriever.js", () => ({
 
 mock.module("../plugins/defaults/compaction/window-manager.js", () => ({
   ContextWindowManager: class {
+    estimateInputTokens() {
+      return 0;
+    }
+    get tokenCountInputs() {
+      return { systemPrompt: "", tools: undefined };
+    }
     constructor() {}
     updateConfig() {}
     shouldCompact() {

package/src/__tests__/conversation-starter-routes.test.ts

@@ -14,7 +14,7 @@ mock.module("../prompts/user-reference.js", () => ({
   resolveUserReference: () => "Alice",
 }));
 
-import { getSqlite } from "../memory/db-connection.js";
+import { getMemorySqlite, getSqlite } from "../memory/db-connection.js";
 import { initializeDb } from "../memory/db-init.js";
 import {
   CONVERSATION_STARTERS_STALE_TTL_MS,
@@ -23,7 +23,7 @@ import {
 } from "../runtime/routes/conversation-starter-routes.js";
 import { RouteError } from "../runtime/routes/errors.js";
 
-initializeDb();
+await initializeDb();
 
 /**
  * Dispatch a request to the ROUTES array by matching endpoint + method.
@@ -66,7 +66,7 @@ function dispatch(path: string, method = "GET"): unknown | Promise<unknown> {
 function clearTables() {
   getSqlite().run("DELETE FROM conversation_starters");
   getSqlite().run("DELETE FROM memory_graph_nodes");
-  getSqlite().run("DELETE FROM memory_jobs");
+  getMemorySqlite()!.run("DELETE FROM memory_jobs");
   getSqlite().run("DELETE FROM memory_checkpoints");
 }
 
@@ -129,7 +129,7 @@ function setCheckpoint(key: string, value: string, updatedAt = Date.now()) {
 
 function insertStarterJob(scopeId = "default", status = "pending") {
   const now = Date.now();
-  getSqlite().run(
+  getMemorySqlite()!.run(
     `INSERT INTO memory_jobs (
       id, type, payload, status, attempts, deferrals, run_after, last_error,
       started_at, created_at, updated_at
@@ -140,7 +140,7 @@ function insertStarterJob(scopeId = "default", status = "pending") {
 
 function countStarterJobs() {
   return (
-    getSqlite()
+    getMemorySqlite()!
       .prepare(
         `SELECT COUNT(*) AS c FROM memory_jobs WHERE type = 'generate_conversation_starters'`,
       )

package/src/__tests__/conversation-store.test.ts

@@ -27,7 +27,7 @@ import { getDb } from "../memory/db-connection.js";
 import { initializeDb } from "../memory/db-init.js";
 import { skillLoadedEvents } from "../memory/schema.js";
 // Initialize db once before all tests
-initializeDb();
+await initializeDb();
 
 describe("deleteLastExchange", () => {
   beforeEach(() => {

package/src/__tests__/conversation-surfaces-activation-emit.test.ts

@@ -44,7 +44,7 @@ import { initializeDb } from "../memory/db-init.js";
 import { queryUnreportedOnboardingEvents } from "../memory/onboarding-events-store.js";
 import { activationSessions, onboardingEvents } from "../memory/schema.js";
 
-initializeDb();
+await initializeDb();
 
 interface ProcessMessageCall {
   content: string;

package/src/__tests__/conversation-sync-tags.test.ts

@@ -20,7 +20,7 @@ import { publishConversationTitleChanged } from "../runtime/sync/resource-sync-e
 import { resetDbForTesting } from "./db-test-helpers.js";
 import { waitFor } from "./helpers/wait-for.js";
 
-initializeDb();
+await initializeDb();
 
 function clearTables(): void {
   const db = getDb();

package/src/__tests__/conversation-usage.test.ts

@@ -48,7 +48,7 @@ import { UsageTrackingProvider } from "../providers/usage-tracking.js";
 import type { PricingUsage } from "../usage/types.js";
 import { resolvePricingForUsageWithOverrides } from "../util/pricing.js";
 
-initializeDb();
+await initializeDb();
 
 function createMockLlmConfig() {
   return {

package/src/__tests__/conversation-wipe.test.ts

@@ -14,12 +14,12 @@ import {
   getMessages,
   wipeConversation,
 } from "../memory/conversation-crud.js";
-import { getDb } from "../memory/db-connection.js";
+import { getDb, getLogsDb, getMemoryDb } from "../memory/db-connection.js";
 import { initializeDb } from "../memory/db-init.js";
 import { enqueueMemoryJob } from "../memory/jobs-store.js";
 
 // Initialize db once before all tests
-initializeDb();
+await initializeDb();
 
 describe("wipeConversation", () => {
   beforeEach(() => {
@@ -28,9 +28,10 @@ describe("wipeConversation", () => {
     db.run(`DELETE FROM memory_segments`);
     db.run(`DELETE FROM memory_summaries`);
     db.run(`DELETE FROM memory_embeddings`);
-    db.run(`DELETE FROM memory_jobs`);
+    // memory_jobs / llm_request_logs live in their own dedicated connections.
+    getMemoryDb()!.run(`DELETE FROM memory_jobs`);
     db.run(`DELETE FROM tool_invocations`);
-    db.run(`DELETE FROM llm_request_logs`);
+    getLogsDb()!.run(`DELETE FROM llm_request_logs`);
     db.run(`DELETE FROM messages`);
     db.run(`DELETE FROM conversations`);
   });
@@ -98,9 +99,9 @@ describe("wipeConversation", () => {
       skipIndexing: true,
     });
 
-    // Clear any jobs that might have been created by prior operations
-    const db = getDb();
-    db.run(`DELETE FROM memory_jobs`);
+    // Clear any jobs that might have been created by prior operations.
+    // memory_jobs lives in the dedicated memory connection.
+    getMemoryDb()!.run(`DELETE FROM memory_jobs`);
 
     enqueueMemoryJob("graph_extract", { conversationId: conv.id });
     enqueueMemoryJob("build_conversation_summary", {
@@ -110,7 +111,7 @@ describe("wipeConversation", () => {
     const result = wipeConversation(conv.id);
 
     const raw = (
-      getDb() as unknown as {
+      getMemoryDb() as unknown as {
         $client: import("bun:sqlite").Database;
       }
     ).$client;

package/src/__tests__/conversation-workspace-cache-state.test.ts

@@ -139,6 +139,12 @@ mock.module("../memory/retrieval-budget.js", () => ({
 
 mock.module("../plugins/defaults/compaction/window-manager.js", () => ({
   ContextWindowManager: class {
+    estimateInputTokens() {
+      return 0;
+    }
+    get tokenCountInputs() {
+      return { systemPrompt: "", tools: undefined };
+    }
     constructor() {}
     updateConfig() {}
     shouldCompact() {

package/src/__tests__/conversation-workspace-injection.test.ts

@@ -178,6 +178,12 @@ mock.module("../memory/retrieval-budget.js", () => ({
 }));
 mock.module("../plugins/defaults/compaction/window-manager.js", () => ({
   ContextWindowManager: class {
+    estimateInputTokens() {
+      return 0;
+    }
+    get tokenCountInputs() {
+      return { systemPrompt: "", tools: undefined };
+    }
     updateConfig() {}
     shouldCompact() {
       return { needed: false, estimatedTokens: 0 };

package/src/__tests__/conversation-workspace-tool-tracking.test.ts

@@ -175,6 +175,12 @@ mock.module("../memory/retrieval-budget.js", () => ({
 }));
 mock.module("../plugins/defaults/compaction/window-manager.js", () => ({
   ContextWindowManager: class {
+    estimateInputTokens() {
+      return 0;
+    }
+    get tokenCountInputs() {
+      return { systemPrompt: "", tools: undefined };
+    }
     updateConfig() {}
     shouldCompact() {
       return { needed: false, estimatedTokens: 0 };

package/src/__tests__/conversations-import-system-filter.test.ts

@@ -35,7 +35,7 @@ import { conversations, messages } from "../memory/schema.js";
 import { ROUTES } from "../runtime/routes/conversations-import-routes.js";
 import type { RouteHandlerArgs } from "../runtime/routes/types.js";
 
-initializeDb();
+await initializeDb();
 
 function resetTables() {
   const db = getDb();

package/src/__tests__/copy-composer-tc-templates.test.ts

@@ -234,6 +234,23 @@ describe("guardian_decision fallback copy", () => {
     expect(copy.body.length).toBeLessThan(longName.length);
     expect(copy.body).toContain("…");
   });
+
+  test("degrades to generic copy when the payload fails to parse", () => {
+    // A payload that doesn't match the schema (here, an out-of-range
+    // `decision`) is not trusted for identity rendering — the template must
+    // still produce safe, generic copy rather than throw or leak fields.
+    const signal = buildGuardianDecisionSignal({
+      decision: "totally-bogus",
+      requesterDisplayName: "Alice",
+    });
+    const result = composeFallbackCopy(signal, ["vellum"]);
+    const copy = result.vellum!;
+
+    expect(copy.title).toBe("Trusted Contact Decision");
+    expect(copy.body).toBe(
+      "Someone's access request has been denied by a guardian.",
+    );
+  });
 });
 
 // ── denied template ──────────────────────────────────────────────────────────

package/src/__tests__/credential-security-invariants.test.ts

@@ -222,7 +222,6 @@ describe("Invariant 2: no generic plaintext secret read API", () => {
       "runtime/routes/platform-routes.ts", // CLI platform connect/disconnect/status routes (CLI-migrated to IPC)
       "ipc/skill-routes/providers.ts", // host.providers.secureKeys.getProviderKey IPC route (out-of-process SkillHost companion)
       "daemon/external-plugins-bootstrap.ts", // reads credentials at plugin init (manifest.requiresCredential) via the CES-mediated getSecureKeyAsync path
-      "plugin-api/index.ts", // public @vellumai/plugin-api surface re-exports getSecureKeyAsync to dynamically-imported workspace plugins via the boot-time shim (compiled-binary plugin loading)
       "inbound/platform-callback-registration.ts", // managed credential lookup for platform base URL, assistant ID, and API key
       "tts/providers/elevenlabs-provider.ts", // ElevenLabs TTS API key lookup
       "tts/providers/deepgram-provider.ts", // Deepgram TTS API key lookup

package/src/__tests__/db-acp-history.test.ts

@@ -319,11 +319,11 @@ describe("acp_session_history cwd migration (272)", () => {
     // Second run short-circuits on the completed checkpoint.
     expect(() => migrateAcpSessionHistoryCwd(db)).not.toThrow();
 
-    // Even with the checkpoint cleared (simulating crash recovery), the
+    // Even with the step checkpoint cleared (simulating crash recovery), the
     // column-existence guard makes the ALTER a no-op.
     raw
       .query(`DELETE FROM memory_checkpoints WHERE key = ?`)
-      .run("migration_acp_session_history_cwd_v1");
+      .run("step:migrateAcpSessionHistoryCwd");
     expect(() => migrateAcpSessionHistoryCwd(db)).not.toThrow();
 
     const columns = raw

package/src/__tests__/db-conversation-fork-lineage-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";
 const originalBunTest = process.env.BUN_TEST;
 
 mock.module("../util/logger.js", () => ({
@@ -14,10 +13,8 @@ mock.module("../util/logger.js", () => ({
 }));
 
 import { getSqliteFrom } from "../memory/db-connection.js";
-import { initializeDb } from "../memory/db-init.js";
 import { migrateConversationForkLineage } from "../memory/migrations/183-add-conversation-fork-lineage.js";
 import * as schema from "../memory/schema.js";
-import { getDbPath } from "../util/platform.js";
 import { resetDbForTesting } from "./db-test-helpers.js";
 
 function createTestDb() {
@@ -68,7 +65,6 @@ function bootstrapPreLineageConversations(raw: Database): void {
 
 function resetMigrationTestDb(): void {
   resetDbForTesting();
-  removeTestDbFiles(getDbPath());
 }
 
 describe("conversation fork lineage migration", () => {
@@ -83,9 +79,12 @@ describe("conversation fork lineage migration", () => {
   });
 
   test("fresh DB initialization includes nullable lineage columns and parent lookup index", () => {
-    initializeDb();
+    const db = createTestDb();
+    const raw = getSqliteFrom(db);
+    bootstrapPreLineageConversations(raw);
+
+    migrateConversationForkLineage(db);
 
-    const raw = new Database(getDbPath());
     const columns = getColumnNames(raw);
 
     expect(columns).toContain("fork_parent_conversation_id");
@@ -107,7 +106,6 @@ describe("conversation fork lineage migration", () => {
 
     expect(forkColumns).toHaveLength(2);
     expect(forkColumns.every((column) => column.notnull === 0)).toBe(true);
-    raw.close();
   });
 
   test("migration upgrades the previous schema without disturbing existing conversation rows", () => {