@vellumai/assistant 0.10.0 → 0.10.1-staging.1

package/src/__tests__/jobs-store-upsert-debounced.test.ts

@@ -15,21 +15,18 @@ mock.module("../config/loader.js", () => ({
 
 import { eq } from "drizzle-orm";
 
-import { getDb } from "../memory/db-connection.js";
+import { getMemoryDb } from "../memory/db-connection.js";
 import { initializeDb } from "../memory/db-init.js";
-import {
-  enqueueMemoryJob,
-  upsertDebouncedJob,
-} from "../memory/jobs-store.js";
+import { enqueueMemoryJob, upsertDebouncedJob } from "../memory/jobs-store.js";
 import { memoryJobs } from "../memory/schema.js";
 
 describe("upsertDebouncedJob payload refresh", () => {
-  beforeAll(() => {
-    initializeDb();
+  beforeAll(async () => {
+    await initializeDb();
   });
 
   beforeEach(() => {
-    const db = getDb();
+    const db = getMemoryDb()!;
     db.run("DELETE FROM memory_jobs");
   });
 
@@ -49,7 +46,7 @@ describe("upsertDebouncedJob payload refresh", () => {
       Date.now(),
     );
 
-    const db = getDb();
+    const db = getMemoryDb()!;
     const rows = db
       .select()
       .from(memoryJobs)
@@ -79,7 +76,7 @@ describe("upsertDebouncedJob payload refresh", () => {
       Date.now(),
     );
 
-    const db = getDb();
+    const db = getMemoryDb()!;
     const rows = db
       .select()
       .from(memoryJobs)
@@ -106,7 +103,7 @@ describe("upsertDebouncedJob payload refresh", () => {
       runAfterNew,
     );
 
-    const db = getDb();
+    const db = getMemoryDb()!;
     const rows = db
       .select()
       .from(memoryJobs)
@@ -124,7 +121,7 @@ describe("upsertDebouncedJob payload refresh", () => {
       Date.now(),
     );
 
-    const db = getDb();
+    const db = getMemoryDb()!;
     const rows = db
       .select()
       .from(memoryJobs)

package/src/__tests__/list-messages-attachments.test.ts

@@ -35,7 +35,7 @@ import { getDb } from "../memory/db-connection.js";
 import { initializeDb } from "../memory/db-init.js";
 import { handleListMessages } from "../runtime/routes/conversation-routes.js";
 
-initializeDb();
+await initializeDb();
 
 function resetTables() {
   const db = getDb();
@@ -164,6 +164,47 @@ describe("handleListMessages attachments", () => {
     expect(imgAtt!.data).toBe(IMAGE_BASE64);
     expect(docAtt!.data).toBeUndefined();
   });
+
+  test("attachment-only assistant message synthesizes contentBlocks", async () => {
+    // When the assistant's entire response was a <vellum-attachment/> tag,
+    // parseDirectives strips it → cleanText is empty → renderHistoryContent
+    // drops the empty text block → contentBlocks is []. The serializer must
+    // synthesize attachment blocks from msgAttachments so the client has a
+    // block to anchor the attachment chip.
+    const conv = createConversation();
+    // Persist the post-strip content: an empty text block (what
+    // cleanAssistantContent leaves after stripping the directive tag).
+    const msg = await addMessage(
+      conv.id,
+      "assistant",
+      JSON.stringify([{ type: "text", text: "" }]),
+    );
+    const stored = uploadAttachment("output.png", "image/png", IMAGE_BASE64);
+    linkAttachmentToMessage(msg.id, stored.id, 0);
+
+    const response = handleListMessages(createTestArgs(conv.id));
+    const body = response as {
+      messages: {
+        attachments?: AttachmentPayload[];
+        contentBlocks?: Array<{
+          type: string;
+          attachment?: { id: string; filename: string };
+        }>;
+      }[];
+    };
+
+    expect(body.messages).toHaveLength(1);
+    // Attachments are always on the wire
+    expect(body.messages[0].attachments).toBeDefined();
+    expect(body.messages[0].attachments).toHaveLength(1);
+    // contentBlocks must be synthesized — not omitted
+    expect(body.messages[0].contentBlocks).toBeDefined();
+    expect(body.messages[0].contentBlocks).toHaveLength(1);
+    expect(body.messages[0].contentBlocks![0].type).toBe("attachment");
+    expect(body.messages[0].contentBlocks![0].attachment!.filename).toBe(
+      "output.png",
+    );
+  });
 });
 
 describe("handleListMessages no_response filtering", () => {

package/src/__tests__/list-messages-client-message-id.test.ts

@@ -30,7 +30,7 @@ import { getDb } from "../memory/db-connection.js";
 import { initializeDb } from "../memory/db-init.js";
 import { handleListMessages } from "../runtime/routes/conversation-routes.js";
 
-initializeDb();
+await initializeDb();
 
 function resetTables() {
   const db = getDb();

package/src/__tests__/list-messages-hidden-metadata.test.ts

@@ -40,7 +40,7 @@ import {
 import { messages } from "../memory/schema.js";
 import { handleListMessages } from "../runtime/routes/conversation-routes.js";
 
-initializeDb();
+await initializeDb();
 
 function resetTables() {
   const db = getDb();

package/src/__tests__/list-messages-page-latest.test.ts

@@ -49,7 +49,7 @@ import {
 import { handleListMessages } from "../runtime/routes/conversation-routes.js";
 import { BadRequestError } from "../runtime/routes/errors.js";
 
-initializeDb();
+await initializeDb();
 
 function resetTables() {
   const db = getDb();

package/src/__tests__/list-messages-tool-merge.test.ts

@@ -30,7 +30,7 @@ import { getDb } from "../memory/db-connection.js";
 import { initializeDb } from "../memory/db-init.js";
 import { handleListMessages } from "../runtime/routes/conversation-routes.js";
 
-initializeDb();
+await initializeDb();
 
 function resetTables() {
   const db = getDb();

package/src/__tests__/llm-context-route-provider.test.ts

@@ -7,13 +7,13 @@ mock.module("../util/logger.js", () => ({
     }),
 }));
 
-import { getDb } from "../memory/db-connection.js";
+import { getLogsDb } from "../memory/db-connection.js";
 import { initializeDb } from "../memory/db-init.js";
 import { llmRequestLogs } from "../memory/schema.js";
 import { ROUTES } from "../runtime/routes/conversation-query-routes.js";
 import { NotFoundError } from "../runtime/routes/errors.js";
 
-initializeDb();
+await initializeDb();
 
 const llmContextRoute = ROUTES.find(
   (r) => r.method === "GET" && r.endpoint === "messages/:id/llm-context",
@@ -32,7 +32,7 @@ function dispatchLogPayload(logId: string) {
 }
 
 function clearRequestLogs(): void {
-  getDb().delete(llmRequestLogs).run();
+  getLogsDb()!.delete(llmRequestLogs).run();
 }
 
 function seedRequestLog(overrides: {
@@ -43,7 +43,7 @@ function seedRequestLog(overrides: {
   responsePayload: string;
   createdAt?: number;
 }): void {
-  getDb()
+  getLogsDb()!
     .insert(llmRequestLogs)
     .values({
       id: overrides.id,
@@ -250,6 +250,71 @@ describe("GET /v1/messages/:id/llm-context provider preference", () => {
   });
 });
 
+describe("GET /v1/messages/:id/llm-context estimated cost cache accounting", () => {
+  const anthropicRequestPayload = JSON.stringify({
+    model: "claude-opus-4-6",
+    messages: [{ role: "user", content: "Hello there." }],
+  });
+
+  // Anthropic Messages usage reports `input_tokens` net of cache; cache-read is
+  // a separate, additive bucket. 19,091 fresh + 17,096 cache-read on Opus 4.6
+  // ($5/M input, $0.50/M cache-read, $25/M output).
+  const anthropicResponsePayload = JSON.stringify({
+    model: "claude-opus-4-6",
+    stop_reason: "end_turn",
+    content: [{ type: "text", text: "Hello back." }],
+    usage: {
+      input_tokens: 19091,
+      cache_read_input_tokens: 17096,
+      output_tokens: 440,
+    },
+  });
+
+  // 19091*5/1e6 + 17096*0.5/1e6 + 440*25/1e6
+  const expectedCostUsd = 0.115003;
+
+  test("prices full input at full rate for OpenRouter→Anthropic (cache not double-subtracted)", async () => {
+    seedRequestLog({
+      id: "log-cost-openrouter-anthropic",
+      messageId: "msg-cost-openrouter-anthropic",
+      provider: "openrouter",
+      requestPayload: anthropicRequestPayload,
+      responsePayload: anthropicResponsePayload,
+    });
+
+    const body = (await dispatchLlmContext(
+      "msg-cost-openrouter-anthropic",
+    )) as {
+      logs: Array<{ summary?: { estimatedCostUsd?: number | null } }>;
+    };
+
+    const cost = body.logs[0]?.summary?.estimatedCostUsd;
+    expect(cost).toBeCloseTo(expectedCostUsd, 5);
+    // The pre-fix bug subtracted cache-read from the already-net input, pricing
+    // only 1,995 fresh tokens → ~$0.0295. Guard against regressing to that.
+    expect(cost).toBeGreaterThan(0.11);
+  });
+
+  test("prices full input at full rate for native Anthropic", async () => {
+    seedRequestLog({
+      id: "log-cost-anthropic",
+      messageId: "msg-cost-anthropic",
+      provider: "anthropic",
+      requestPayload: anthropicRequestPayload,
+      responsePayload: anthropicResponsePayload,
+    });
+
+    const body = (await dispatchLlmContext("msg-cost-anthropic")) as {
+      logs: Array<{ summary?: { estimatedCostUsd?: number | null } }>;
+    };
+
+    expect(body.logs[0]?.summary?.estimatedCostUsd).toBeCloseTo(
+      expectedCostUsd,
+      5,
+    );
+  });
+});
+
 describe("GET /v1/llm-request-logs/:id/payload", () => {
   test("returns parsed payloads for a valid log", async () => {
     const reqPayload = JSON.stringify({ model: "gpt-4.1", messages: [] });

package/src/__tests__/llm-request-log-agent-loop-exit-reason.test.ts

@@ -25,7 +25,7 @@ mock.module("../config/loader.js", () => ({
   }),
 }));
 
-import { getDb } from "../memory/db-connection.js";
+import { getLogsDb } from "../memory/db-connection.js";
 import { initializeDb } from "../memory/db-init.js";
 import {
   getRequestLogById,
@@ -34,11 +34,11 @@ import {
 } from "../memory/llm-request-log-store.js";
 import { llmRequestLogs } from "../memory/schema.js";
 
-initializeDb();
+await initializeDb();
 
+// llm_request_logs lives in the dedicated logs connection.
 function resetLogs(): void {
-  const db = getDb();
-  db.delete(llmRequestLogs).run();
+  getLogsDb()!.delete(llmRequestLogs).run();
 }
 
 describe("setAgentLoopExitReasonOnLatestLog", () => {
@@ -105,7 +105,11 @@ describe("setAgentLoopExitReasonOnLatestLog", () => {
 
     // Current run lands a new log, then exits.
     Bun.sleepSync(2);
-    const current = recordRequestLog("conv-1", '{"cur_req":1}', '{"cur_res":1}');
+    const current = recordRequestLog(
+      "conv-1",
+      '{"cur_req":1}',
+      '{"cur_res":1}',
+    );
     setAgentLoopExitReasonOnLatestLog("conv-1", "yield_to_user");
 
     expect(getRequestLogById(prev)?.agentLoopExitReason).toBe("no_tool_calls");

package/src/__tests__/llm-request-log-call-site.test.ts

@@ -24,7 +24,7 @@ mock.module("../config/loader.js", () => ({
   }),
 }));
 
-import { getDb, getSqliteFrom } from "../memory/db-connection.js";
+import { getLogsDb, getSqliteFrom } from "../memory/db-connection.js";
 import { initializeDb } from "../memory/db-init.js";
 import {
   getRequestLogById,
@@ -33,11 +33,11 @@ import {
 import { migrateLlmRequestLogCallSite } from "../memory/migrations/264-llm-request-log-call-site.js";
 import { llmRequestLogs } from "../memory/schema.js";
 
-initializeDb();
+await initializeDb();
 
+// llm_request_logs lives in the dedicated logs connection.
 function resetLogs(): void {
-  const db = getDb();
-  db.delete(llmRequestLogs).run();
+  getLogsDb()!.delete(llmRequestLogs).run();
 }
 
 describe("recordRequestLog call_site stamping", () => {
@@ -100,7 +100,7 @@ describe("recordRequestLog call_site stamping", () => {
 
 describe("migrateLlmRequestLogCallSite", () => {
   test("adds the call_site column when missing", () => {
-    const db = getDb();
+    const db = getLogsDb()!;
     const raw = getSqliteFrom(db);
 
     // Drop the column if present (simulate pre-264 state). SQLite supports
@@ -126,7 +126,7 @@ describe("migrateLlmRequestLogCallSite", () => {
   });
 
   test("is idempotent — second run is a no-op", () => {
-    const db = getDb();
+    const db = getLogsDb()!;
     // First run (column may or may not exist depending on test order; either
     // path is fine for the idempotency contract).
     migrateLlmRequestLogCallSite(db);

package/src/__tests__/llm-request-log-turn-query.test.ts

@@ -25,7 +25,7 @@ import {
   createConversation,
   forkConversation,
 } from "../memory/conversation-crud.js";
-import { getDb } from "../memory/db-connection.js";
+import { getDb, getLogsDb } from "../memory/db-connection.js";
 import { initializeDb } from "../memory/db-init.js";
 import {
   backfillMessageIdOnLogs,
@@ -35,11 +35,16 @@ import {
 } from "../memory/llm-request-log-store.js";
 import { llmRequestLogs, toolInvocations } from "../memory/schema.js";
 
-initializeDb();
+await initializeDb();
+
+// llm_request_logs lives in the dedicated logs connection.
+function logsDb() {
+  return getLogsDb()!;
+}
 
 function resetTables(): void {
   const db = getDb();
-  db.delete(llmRequestLogs).run();
+  logsDb().delete(llmRequestLogs).run();
   db.delete(toolInvocations).run();
   db.run("DELETE FROM message_attachments");
   db.run("DELETE FROM attachments");
@@ -220,7 +225,8 @@ describe("getRequestLogsByMessageId — turn-aware query", () => {
     );
 
     // Orphaned log 1: message_id points to a deleted message
-    db.insert(llmRequestLogs)
+    logsDb()
+      .insert(llmRequestLogs)
       .values({
         id: "log-orphan-1",
         conversationId: conv.id,
@@ -233,7 +239,8 @@ describe("getRequestLogsByMessageId — turn-aware query", () => {
       .run();
 
     // Orphaned log 2
-    db.insert(llmRequestLogs)
+    logsDb()
+      .insert(llmRequestLogs)
       .values({
         id: "log-orphan-2",
         conversationId: conv.id,
@@ -246,7 +253,8 @@ describe("getRequestLogsByMessageId — turn-aware query", () => {
       .run();
 
     // Surviving log: backfilled to the surviving assistant message
-    db.insert(llmRequestLogs)
+    logsDb()
+      .insert(llmRequestLogs)
       .values({
         id: "log-surviving",
         conversationId: conv.id,
@@ -280,7 +288,8 @@ describe("getRequestLogsByMessageId — turn-aware query", () => {
     );
 
     // Unlinked log: messageId is NULL (backfill hasn't run yet)
-    db.insert(llmRequestLogs)
+    logsDb()
+      .insert(llmRequestLogs)
       .values({
         id: "log-unlinked-1",
         conversationId: conv.id,
@@ -293,7 +302,8 @@ describe("getRequestLogsByMessageId — turn-aware query", () => {
       .run();
 
     // Linked log: already backfilled to the assistant message
-    db.insert(llmRequestLogs)
+    logsDb()
+      .insert(llmRequestLogs)
       .values({
         id: "log-linked-1",
         conversationId: conv.id,
@@ -311,7 +321,7 @@ describe("getRequestLogsByMessageId — turn-aware query", () => {
     expect(logs[1]?.id).toBe("log-linked-1");
 
     // Verify opportunistic backfill ran: the unlinked log should now have a messageId
-    const backfilledLog = db
+    const backfilledLog = logsDb()
       .select({ messageId: llmRequestLogs.messageId })
       .from(llmRequestLogs)
       .where(sql`${llmRequestLogs.id} = 'log-unlinked-1'`)
@@ -332,7 +342,8 @@ describe("getRequestLogsByMessageId — turn-aware query", () => {
     db.run(
       sql`INSERT INTO messages (id, conversation_id, role, content, created_at) VALUES ('a1-a', ${convA.id}, 'assistant', '"Hi A"', ${T + 10000})`,
     );
-    db.insert(llmRequestLogs)
+    logsDb()
+      .insert(llmRequestLogs)
       .values({
         id: "log-conv-a",
         conversationId: convA.id,
@@ -351,7 +362,8 @@ describe("getRequestLogsByMessageId — turn-aware query", () => {
     db.run(
       sql`INSERT INTO messages (id, conversation_id, role, content, created_at) VALUES ('a1-b', ${convB.id}, 'assistant', '"Hi B"', ${T + 10000})`,
     );
-    db.insert(llmRequestLogs)
+    logsDb()
+      .insert(llmRequestLogs)
       .values({
         id: "log-conv-b",
         conversationId: convB.id,
@@ -386,7 +398,8 @@ describe("getRequestLogsByMessageId — turn-aware query", () => {
     db.run(
       sql`INSERT INTO messages (id, conversation_id, role, content, created_at) VALUES ('a1-t', ${conv.id}, 'assistant', '"Answer 1"', ${T + 10000})`,
     );
-    db.insert(llmRequestLogs)
+    logsDb()
+      .insert(llmRequestLogs)
       .values({
         id: "log-turn1-unlinked",
         conversationId: conv.id,
@@ -405,7 +418,8 @@ describe("getRequestLogsByMessageId — turn-aware query", () => {
     db.run(
       sql`INSERT INTO messages (id, conversation_id, role, content, created_at) VALUES ('a2-t', ${conv.id}, 'assistant', '"Answer 2"', ${T + 70000})`,
     );
-    db.insert(llmRequestLogs)
+    logsDb()
+      .insert(llmRequestLogs)
       .values({
         id: "log-turn2-unlinked",
         conversationId: conv.id,

package/src/__tests__/llm-usage-store.test.ts

@@ -21,7 +21,7 @@ import {
 import type { PricingResult, UsageEventInput } from "../usage/types.js";
 
 // Initialize db once before all tests
-initializeDb();
+await initializeDb();
 
 function makeInput(overrides?: Partial<UsageEventInput>): UsageEventInput {
   return {
@@ -1306,6 +1306,41 @@ describe("getUsageGroupBreakdown", () => {
     expect(groups[0].totalOutputTokens).toBe(125);
     expect(groups[0].totalEstimatedCostUsd).toBeCloseTo(0.05);
     expect(groups[0].eventCount).toBe(2);
+    // No user messages were inserted, so the conversation has zero turns.
+    expect(groups[0].turnCount).toBe(0);
+  });
+
+  test("counts conversation turns from eligible user messages when grouping by conversation", () => {
+    const db = getDb();
+    const conversationId = "conv-turns-1";
+    const now = Date.now();
+    db.run(
+      `INSERT INTO conversations (id, title, created_at, updated_at) VALUES ('${conversationId}', 'Turn counting', ${now}, ${now})`,
+    );
+    // Two user turns; the second turn has two LLM calls, but the count comes
+    // from the user messages (2), not the number of LLM calls (3).
+    db.run(
+      `INSERT INTO messages (id, conversation_id, role, content, created_at) VALUES ('u1', '${conversationId}', 'user', 'first', 500)`,
+    );
+    db.run(
+      `INSERT INTO messages (id, conversation_id, role, content, created_at) VALUES ('u2', '${conversationId}', 'user', 'second', 2000)`,
+    );
+    // Tool-result user messages are not real turns and must be excluded.
+    db.run(
+      `INSERT INTO messages (id, conversation_id, role, content, created_at) VALUES ('tr1', '${conversationId}', 'user', '[{"type":"tool_result","tool_use_id":"x","content":""}]', 2100)`,
+    );
+    insertEventAt(1000, { conversationId, inputTokens: 100 });
+    insertEventAt(2500, { conversationId, inputTokens: 100 });
+    insertEventAt(2600, { conversationId, inputTokens: 100 });
+
+    const groups = getUsageGroupBreakdown(
+      { from: 0, to: 5000 },
+      "conversation",
+    );
+    expect(groups).toHaveLength(1);
+    expect(groups[0].groupId).toBe(conversationId);
+    expect(groups[0].eventCount).toBe(3);
+    expect(groups[0].turnCount).toBe(2);
   });
 
   test("returns groupId null for the Other bucket when grouping by conversation and events have no conversation id", () => {
@@ -1329,6 +1364,8 @@ describe("getUsageGroupBreakdown", () => {
     expect(groups[0].groupId).toBeNull();
     expect(groups[0].totalInputTokens).toBe(300);
     expect(groups[0].eventCount).toBe(2);
+    // The Other bucket has no parent conversation, so turns are undefined.
+    expect(groups[0].turnCount).toBeNull();
   });
 
   test("returns groupId null for every row when grouping by a non-conversation dimension", () => {
@@ -1348,6 +1385,8 @@ describe("getUsageGroupBreakdown", () => {
     expect(groups.length).toBeGreaterThan(0);
     for (const row of groups) {
       expect(row.groupId).toBeNull();
+      // Turns are only computed for the conversation grouping.
+      expect(row.turnCount).toBeNull();
     }
   });
 });

package/src/__tests__/log-export-routes.test.ts

@@ -29,7 +29,7 @@ import { assistantEventHub } from "../runtime/assistant-event-hub.js";
 import { ROUTES } from "../runtime/routes/log-export-routes.js";
 import { getLogsDir } from "../util/platform.js";
 
-initializeDb();
+await initializeDb();
 
 const exportRoute = ROUTES.find((r) => r.endpoint === "export")!;
 

package/src/__tests__/log-export-workspace.test.ts

@@ -38,7 +38,7 @@ mock.module("../util/secure-keys.js", () => ({
   getSecureKeyAsync: async () => undefined,
 }));
 
-import { getDb } from "../memory/db-connection.js";
+import { getDb, getLogsDb } from "../memory/db-connection.js";
 import { initializeDb } from "../memory/db-init.js";
 import {
   conversations,
@@ -60,7 +60,7 @@ import {
   SYNTHETIC_OPENAI_PROJECT_KEY,
 } from "./secret-fixtures.js";
 
-initializeDb();
+await initializeDb();
 
 // ---------------------------------------------------------------------------
 // Helpers
@@ -552,7 +552,7 @@ describe("POST /v1/export — manifest truncatedSections", () => {
         createdAt: now + i,
       }),
     );
-    db.insert(llmRequestLogs).values(rows).run();
+    getLogsDb()!.insert(llmRequestLogs).values(rows).run();
 
     const manifest = await readManifest({ full: true });
     expect(manifest.truncatedSections).toEqual(["llm-request-logs"]);

package/src/__tests__/memory-jobs-worker-lanes.test.ts

@@ -110,7 +110,7 @@ mock.module("../memory/db-maintenance.js", () => ({
   maybeRunDbMaintenance: () => {},
 }));
 
-import { getDb } from "../memory/db-connection.js";
+import { getMemoryDb } from "../memory/db-connection.js";
 import { initializeDb } from "../memory/db-init.js";
 import { enqueueMemoryJob } from "../memory/jobs-store.js";
 import { runMemoryJobsOnce } from "../memory/jobs-worker.js";
@@ -118,12 +118,12 @@ import { _resetQdrantBreaker } from "../memory/qdrant-circuit-breaker.js";
 import { memoryJobs } from "../memory/schema.js";
 
 describe("memory jobs worker lane scheduling", () => {
-  beforeAll(() => {
-    initializeDb();
+  beforeAll(async () => {
+    await initializeDb();
   });
 
   beforeEach(() => {
-    const db = getDb();
+    const db = getMemoryDb()!;
     db.run("DELETE FROM memory_jobs");
     completions.length = 0;
     _resetQdrantBreaker();
@@ -185,7 +185,7 @@ describe("memory jobs worker lane scheduling", () => {
 function countSlowByStatus(
   status: "pending" | "running" | "completed",
 ): number {
-  const db = getDb();
+  const db = getMemoryDb()!;
   return db
     .select()
     .from(memoryJobs)

package/src/__tests__/memory-recall-log-store.test.ts

@@ -28,7 +28,7 @@ import {
 } from "../memory/memory-recall-log-store.js";
 import { memoryRecallLogs } from "../memory/schema.js";
 
-initializeDb();
+await initializeDb();
 
 function resetTables(): void {
   const db = getDb();

package/src/__tests__/memory-upsert-concurrency.test.ts

@@ -60,20 +60,20 @@ mock.module("../config/loader.js", () => ({
   invalidateConfigCache: () => {},
 }));
 
-import { getDb } from "../memory/db-connection.js";
+import { getDb, getMemorySqlite } from "../memory/db-connection.js";
 import { initializeDb } from "../memory/db-init.js";
 import { indexMessageNow } from "../memory/indexer.js";
 import { conversations, memorySegments, messages } from "../memory/schema.js";
 
 // Initialize DB once for the entire file. Each test cleans its own tables.
-initializeDb();
+await initializeDb();
 
 function resetTables() {
   const db = getDb();
   db.run("DELETE FROM memory_embeddings");
   db.run("DELETE FROM memory_graph_nodes");
   db.run("DELETE FROM memory_segments");
-  db.run("DELETE FROM memory_jobs");
+  getMemorySqlite()!.run("DELETE FROM memory_jobs");
   db.run("DELETE FROM messages");
   db.run("DELETE FROM conversations");
 }
@@ -550,4 +550,3 @@ describe("memory segment job atomicity under repeated indexer invocations", () =
     expect(storedSegments.length).toBe(firstCount);
   });
 });
-

package/src/__tests__/messages-after-tiebreaker.test.ts

@@ -14,7 +14,7 @@ import { getDb } from "../memory/db-connection.js";
 import { initializeDb } from "../memory/db-init.js";
 import { conversations, messages } from "../memory/schema.js";
 
-initializeDb();
+await initializeDb();
 
 const CONV_ID = "conv-tiebreaker";