@vellumai/assistant 0.10.4 → 0.10.5-dev.202607022329.d89bf2f

package/src/__tests__/schedule-routes.test.ts

@@ -156,6 +156,7 @@ function recordUsageCostAt(
     { estimatedCostUsd, pricingStatus: "priced" },
   );
   rawRun(
+    "test:setUsageCreatedAt",
     "UPDATE llm_usage_events SET created_at = ? WHERE id = ?",
     createdAt,
     event.id,
@@ -174,6 +175,7 @@ function setScheduleRunWindow({
   status?: "ok" | "error" | "running";
 }) {
   rawRun(
+    "test:setCronRunWindow",
     "UPDATE cron_runs SET status = ?, started_at = ?, finished_at = ?, duration_ms = ?, created_at = ? WHERE id = ?",
     status,
     startedAt,
@@ -804,6 +806,7 @@ describe("schedule and heartbeat run metadata", () => {
     const runId = await createScheduleRun(job.id, conversation.id);
     await completeScheduleRun(runId, { status: "ok" });
     rawRun(
+      "test:setCronRunTimes",
       "UPDATE cron_runs SET started_at = ?, finished_at = ?, duration_ms = ?, created_at = ? WHERE id = ?",
       1000,
       2000,
@@ -849,6 +852,7 @@ describe("schedule and heartbeat run metadata", () => {
     });
     const archivedAt = 3000;
     rawRun(
+      "test:archiveConversation",
       "UPDATE conversations SET archived_at = ? WHERE id = ?",
       archivedAt,
       conversation.id,
@@ -926,6 +930,7 @@ describe("schedule and heartbeat run metadata", () => {
     const runId = insertPendingHeartbeatRun(900);
     startHeartbeatRun(runId);
     rawRun(
+      "test:setHeartbeatRunOk",
       "UPDATE heartbeat_runs SET status = 'ok', scheduled_for = ?, started_at = ?, finished_at = ?, duration_ms = ?, conversation_id = ?, created_at = ? WHERE id = ?",
       900,
       1000,
@@ -966,6 +971,7 @@ describe("schedule and heartbeat run metadata", () => {
     });
     const runId = insertPendingHeartbeatRun(1000);
     rawRun(
+      "test:setHeartbeatRunRunning",
       "UPDATE heartbeat_runs SET status = 'running', finished_at = NULL, conversation_id = ?, created_at = ? WHERE id = ?",
       conversation.id,
       1000,
@@ -1160,7 +1166,7 @@ describe("POST /schedules — create", () => {
   async function postCreate(body: Record<string, unknown>) {
     const route = findRoute("schedules", "POST");
     return (await route.handler({ body })) as {
-      schedules: Array<{ id: string; inferenceProfile: string | null }>;
+      schedule: { id: string; inferenceProfile: string | null };
     };
   }
 
@@ -1171,7 +1177,7 @@ describe("POST /schedules — create", () => {
       expression: "0 9 * * *",
       message: "good morning",
     });
-    expect(result.schedules).toHaveLength(1);
+    expect(result.schedule.id).toBeDefined();
     const job = listSchedules()[0];
     expect(job.name).toBe("Morning ping");
     expect(job.mode).toBe("execute");
@@ -1241,7 +1247,7 @@ describe("POST /schedules — create", () => {
     expect(job.description).toBe("Description fallback");
   });
 
-  test("rejects modes other than execute/workflow", async () => {
+  test("rejects modes other than execute/workflow/script", async () => {
     await expect(
       postCreate({
         name: "x",
@@ -1250,7 +1256,9 @@ describe("POST /schedules — create", () => {
         message: "hi",
         mode: "notify",
       }),
-    ).rejects.toThrow("Only 'execute' and 'workflow' modes are supported");
+    ).rejects.toThrow(
+      "Only 'execute', 'script', and 'workflow' modes are supported",
+    );
   });
 
   test("rejects an unparseable expression", async () => {
@@ -1327,7 +1335,7 @@ describe("POST /schedules — create", () => {
       message: "write the digest",
       inferenceProfile: "cost-optimized",
     });
-    expect(result.schedules[0].inferenceProfile).toBe("cost-optimized");
+    expect(result.schedule.inferenceProfile).toBe("cost-optimized");
     expect(listSchedules()[0].inferenceProfile).toBe("cost-optimized");
   });
 
@@ -1338,7 +1346,7 @@ describe("POST /schedules — create", () => {
       expression: "0 9 * * *",
       message: "hi",
     });
-    expect(result.schedules[0].inferenceProfile).toBeNull();
+    expect(result.schedule.inferenceProfile).toBeNull();
   });
 
   test("rejects an unknown inferenceProfile", async () => {

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

@@ -55,7 +55,9 @@ async function waitFor(predicate: () => boolean): Promise<void> {
   // predicate holds, so a larger budget costs nothing when events flow.
   const deadline = Date.now() + 2000;
   while (Date.now() < deadline) {
-    if (predicate()) return;
+    if (predicate()) {
+      return;
+    }
     await new Promise((resolve) => setTimeout(resolve, 5));
   }
   throw new Error("Timed out waiting for schedule-store event");

package/src/__tests__/scheduler-disk-pressure.test.ts

@@ -89,6 +89,18 @@ mock.module("../daemon/disk-pressure-background-gate.js", () => ({
   shouldLogDiskPressureBackgroundSkip: () => true,
 }));
 
+const mockProcessMessage = mock((..._args: unknown[]) => Promise.resolve());
+mock.module("../daemon/process-message.js", () => ({
+  processMessage: mockProcessMessage,
+}));
+
+const mockEmitNotificationSignal = mock((..._args: unknown[]) =>
+  Promise.resolve(),
+);
+mock.module("../notifications/emit-signal.js", () => ({
+  emitNotificationSignal: mockEmitNotificationSignal,
+}));
+
 import { getDb } from "../persistence/db-connection.js";
 import { initializeDb } from "../persistence/db-init.js";
 import { createSchedule } from "../schedule/schedule-store.js";
@@ -104,6 +116,8 @@ function rawDb(): import("bun:sqlite").Database {
 describe("scheduler disk pressure gate", () => {
   beforeEach(() => {
     locked = true;
+    mockProcessMessage.mockClear();
+    mockEmitNotificationSignal.mockClear();
     const db = getDb();
     db.run("DELETE FROM cron_runs");
     db.run("DELETE FROM cron_jobs");
@@ -122,14 +136,11 @@ describe("scheduler disk pressure gate", () => {
       nextRunAt: dueAt,
     });
 
-    const processMessage = mock(() => Promise.resolve());
-    const notify = mock(() => Promise.resolve());
-
-    const processed = await runScheduleOnce(processMessage, notify);
+    const processed = await runScheduleOnce();
 
     expect(processed).toBe(0);
-    expect(processMessage).not.toHaveBeenCalled();
-    expect(notify).not.toHaveBeenCalled();
+    expect(mockProcessMessage).not.toHaveBeenCalled();
+    expect(mockEmitNotificationSignal).not.toHaveBeenCalled();
 
     const row = rawDb()
       .query("SELECT status, next_run_at FROM cron_jobs WHERE id = ?")

package/src/__tests__/scheduler-recurrence.test.ts

@@ -50,6 +50,25 @@ mock.module("../runtime/background-job-runner.js", () => ({
   },
 }));
 
+// The scheduler dispatches reuse/task-path messages through `processMessage`;
+// route them to a per-test delegate so tests can capture those calls.
+let processMessageImpl: (
+  conversationId: string,
+  message: string,
+) => Promise<void> = async () => {};
+mock.module("../daemon/process-message.js", () => ({
+  processMessage: (conversationId: string, message: string) =>
+    processMessageImpl(conversationId, message),
+}));
+
+// Notify-mode firings go through `emitNotificationSignal`; capture them.
+const notifySignalCalls: Array<Record<string, unknown>> = [];
+mock.module("../notifications/emit-signal.js", () => ({
+  emitNotificationSignal: async (payload: Record<string, unknown>) => {
+    notifySignalCalls.push(payload);
+  },
+}));
+
 import { getDb } from "../persistence/db-connection.js";
 import { initializeDb } from "../persistence/db-init.js";
 import {
@@ -138,6 +157,8 @@ describe("scheduler RRULE execution", () => {
     db.run("DELETE FROM conversations");
     onRunBackgroundJobPrompt = null;
     runBackgroundJobShouldFail = false;
+    processMessageImpl = async () => {};
+    notifySignalCalls.length = 0;
   });
 
   test("RRULE schedule fires and creates cron_runs entry", async () => {
@@ -163,10 +184,7 @@ describe("scheduler RRULE execution", () => {
       processedMessages.push({ conversationId, message: prompt });
     };
 
-    const scheduler = startScheduler(
-      async () => {},
-      () => {},
-    );
+    const scheduler = startScheduler();
     await new Promise((resolve) => setTimeout(resolve, 500));
     scheduler.stop();
 
@@ -201,12 +219,12 @@ describe("scheduler RRULE execution", () => {
     const directCalls: { conversationId: string; message: string }[] = [];
     let onMessage: (() => void) | undefined;
     const messageReceived = new Promise<void>((r) => (onMessage = r));
-    const processMessage = async (conversationId: string, message: string) => {
+    processMessageImpl = async (conversationId, message) => {
       directCalls.push({ conversationId, message });
       onMessage?.();
     };
 
-    const scheduler = startScheduler(processMessage, () => {});
+    const scheduler = startScheduler();
     // The run_task path involves a dynamic import which can take >50ms in CI,
     // exceeding the patched setTimeout delay. Await the actual callback instead
     // of relying on a fixed timeout.
@@ -271,10 +289,7 @@ describe("scheduler RRULE execution", () => {
     };
 
     // First tick: the expired schedule should fire its final due run
-    const scheduler1 = startScheduler(
-      async () => {},
-      () => {},
-    );
+    const scheduler1 = startScheduler();
     await new Promise((resolve) => setTimeout(resolve, 500));
     scheduler1.stop();
 
@@ -294,10 +309,7 @@ describe("scheduler RRULE execution", () => {
 
     // Second tick: the disabled schedule must NOT fire again
     processedMessages.length = 0;
-    const scheduler2 = startScheduler(
-      async () => {},
-      () => {},
-    );
+    const scheduler2 = startScheduler();
     await new Promise((resolve) => setTimeout(resolve, 500));
     scheduler2.stop();
 
@@ -327,10 +339,7 @@ describe("scheduler RRULE execution", () => {
       processedMessages.push({ conversationId, message: prompt });
     };
 
-    const scheduler = startScheduler(
-      async () => {},
-      () => {},
-    );
+    const scheduler = startScheduler();
     await new Promise((resolve) => setTimeout(resolve, 500));
     scheduler.stop();
 
@@ -385,10 +394,7 @@ describe("scheduler RRULE execution", () => {
     // Force the schedule to be due
     forceScheduleDue(schedule.id);
 
-    const scheduler = startScheduler(
-      async () => {},
-      () => {},
-    );
+    const scheduler = startScheduler();
     await new Promise((resolve) => setTimeout(resolve, 500));
     scheduler.stop();
 
@@ -437,10 +443,7 @@ describe("scheduler RRULE execution", () => {
       processedMessages.push(prompt);
     };
 
-    const scheduler = startScheduler(
-      async () => {},
-      () => {},
-    );
+    const scheduler = startScheduler();
     await new Promise((resolve) => setTimeout(resolve, 500));
     scheduler.stop();
 
@@ -516,10 +519,7 @@ describe("scheduler RRULE execution", () => {
         processedMessages.push(prompt);
       };
 
-      const scheduler = startScheduler(
-        async () => {},
-        () => {},
-      );
+      const scheduler = startScheduler();
       await new Promise((resolve) => setTimeout(resolve, 500));
       scheduler.stop();
 
@@ -557,8 +557,7 @@ describe("scheduler RRULE execution", () => {
     forceScheduleDue(schedule.id);
     const forcedDueAt = getSchedule(schedule.id)!.nextRunAt;
 
-    const processMessage = async () => {};
-    const scheduler = startScheduler(processMessage, () => {});
+    const scheduler = startScheduler();
     await new Promise((resolve) => setTimeout(resolve, 500));
     scheduler.stop();
 
@@ -589,10 +588,7 @@ describe("scheduler RRULE execution", () => {
       processedMessages.push({ conversationId, message: prompt });
     };
 
-    const scheduler = startScheduler(
-      async () => {},
-      () => {},
-    );
+    const scheduler = startScheduler();
     await new Promise((resolve) => setTimeout(resolve, 500));
     scheduler.stop();
 
@@ -623,32 +619,19 @@ describe("scheduler RRULE execution", () => {
 
     expect(getSchedule(schedule.id)!.status).toBe("active");
 
-    const notifyCalls: Array<{
-      id: string;
-      label: string;
-      message: string;
-      routingIntent: string;
-      routingHints: Record<string, unknown>;
-    }> = [];
-    const notifyScheduleOneShot = (payload: {
-      id: string;
-      label: string;
-      message: string;
-      routingIntent: string;
-      routingHints: Record<string, unknown>;
-    }) => {
-      notifyCalls.push(payload);
-    };
-
-    const scheduler = startScheduler(async () => {}, notifyScheduleOneShot);
+    const scheduler = startScheduler();
     await new Promise((resolve) => setTimeout(resolve, 500));
     scheduler.stop();
 
-    expect(notifyCalls).toHaveLength(1);
-    expect(notifyCalls[0]).toEqual({
-      id: schedule.id,
-      label: "One-shot notify",
-      message: "Notify about this",
+    expect(notifySignalCalls).toHaveLength(1);
+    expect(notifySignalCalls[0]).toMatchObject({
+      sourceEventName: "schedule.notify",
+      sourceContextId: schedule.id,
+      contextPayload: {
+        scheduleId: schedule.id,
+        label: "One-shot notify",
+        message: "Notify about this",
+      },
       routingIntent: "multi_channel",
       routingHints: { channel: "slack" },
     });
@@ -671,10 +654,7 @@ describe("scheduler RRULE execution", () => {
 
     runBackgroundJobShouldFail = true;
 
-    const scheduler = startScheduler(
-      async () => {},
-      () => {},
-    );
+    const scheduler = startScheduler();
     await new Promise((resolve) => setTimeout(resolve, 500));
     scheduler.stop();
 
@@ -699,32 +679,19 @@ describe("scheduler RRULE execution", () => {
 
     forceScheduleDue(schedule.id);
 
-    const notifyCalls: Array<{
-      id: string;
-      label: string;
-      message: string;
-      routingIntent: string;
-      routingHints: Record<string, unknown>;
-    }> = [];
-    const notifyScheduleOneShot = (payload: {
-      id: string;
-      label: string;
-      message: string;
-      routingIntent: string;
-      routingHints: Record<string, unknown>;
-    }) => {
-      notifyCalls.push(payload);
-    };
-
-    const scheduler = startScheduler(async () => {}, notifyScheduleOneShot);
+    const scheduler = startScheduler();
     await new Promise((resolve) => setTimeout(resolve, 500));
     scheduler.stop();
 
-    expect(notifyCalls).toHaveLength(1);
-    expect(notifyCalls[0]).toEqual({
-      id: schedule.id,
-      label: "Recurring notify",
-      message: "Recurring notification",
+    expect(notifySignalCalls).toHaveLength(1);
+    expect(notifySignalCalls[0]).toMatchObject({
+      sourceEventName: "schedule.notify",
+      sourceContextId: schedule.id,
+      contextPayload: {
+        scheduleId: schedule.id,
+        label: "Recurring notify",
+        message: "Recurring notification",
+      },
       routingIntent: "single_channel",
       routingHints: { preferred: "email" },
     });
@@ -752,30 +719,13 @@ describe("scheduler RRULE execution", () => {
       },
     });
 
-    const notifyCalls: Array<{
-      id: string;
-      label: string;
-      message: string;
-      routingIntent: string;
-      routingHints: Record<string, unknown>;
-    }> = [];
-    const notifyScheduleOneShot = (payload: {
-      id: string;
-      label: string;
-      message: string;
-      routingIntent: string;
-      routingHints: Record<string, unknown>;
-    }) => {
-      notifyCalls.push(payload);
-    };
-
-    const scheduler = startScheduler(async () => {}, notifyScheduleOneShot);
+    const scheduler = startScheduler();
     await new Promise((resolve) => setTimeout(resolve, 500));
     scheduler.stop();
 
-    expect(notifyCalls).toHaveLength(1);
-    expect(notifyCalls[0].routingIntent).toBe("all_channels");
-    expect(notifyCalls[0].routingHints).toEqual({
+    expect(notifySignalCalls).toHaveLength(1);
+    expect(notifySignalCalls[0].routingIntent).toBe("all_channels");
+    expect(notifySignalCalls[0].routingHints).toEqual({
       requestedByUser: true,
       channelMentions: ["telegram", "slack"],
       priority: "high",

package/src/__tests__/scheduler-reuse-conversation.test.ts

@@ -86,9 +86,28 @@ mock.module("../runtime/background-job-runner.js", () => ({
   },
 }));
 
+// The scheduler's reuse path dispatches into the existing conversation through
+// `processMessage`; capture those calls into the same shared log as the
+// fresh-bootstrap runner so assertions don't need to know which path ran.
+const captureProcessedMessage = async (
+  conversationId: string,
+  message: string,
+): Promise<void> => {
+  processedMessages.push({ conversationId, message });
+};
+let processMessageImpl: (
+  conversationId: string,
+  message: string,
+) => Promise<void> = captureProcessedMessage;
+mock.module("../daemon/process-message.js", () => ({
+  processMessage: (conversationId: string, message: string) =>
+    processMessageImpl(conversationId, message),
+}));
+
 import { deleteConversation } from "../persistence/conversation-crud.js";
 import { getDb } from "../persistence/db-connection.js";
 import { initializeDb } from "../persistence/db-init.js";
+import { assistantEventHub } from "../runtime/assistant-event-hub.js";
 import { createSchedule, getScheduleRuns } from "../schedule/schedule-store.js";
 import { startScheduler } from "../schedule/scheduler.js";
 
@@ -148,6 +167,7 @@ describe("scheduler conversation reuse", () => {
     db.run("DELETE FROM messages");
     db.run("DELETE FROM conversations");
     processedMessages.length = 0;
+    processMessageImpl = captureProcessedMessage;
     runBackgroundJobOptions.length = 0;
     runBackgroundJobShouldFail = false;
     runBackgroundJobBootstrapFails = false;
@@ -173,11 +193,7 @@ describe("scheduler conversation reuse", () => {
     // WHEN the schedule fires for the first time
     forceScheduleDue(schedule.id);
 
-    const processMessage = async (conversationId: string, message: string) => {
-      processedMessages.push({ conversationId, message });
-    };
-
-    const scheduler1 = startScheduler(processMessage, () => {});
+    const scheduler1 = startScheduler();
     await new Promise((resolve) => setTimeout(resolve, 500));
     scheduler1.stop();
 
@@ -196,7 +212,7 @@ describe("scheduler conversation reuse", () => {
     forceScheduleDue(schedule.id);
     processedMessages.length = 0;
 
-    const scheduler2 = startScheduler(processMessage, () => {});
+    const scheduler2 = startScheduler();
     await new Promise((resolve) => setTimeout(resolve, 500));
     scheduler2.stop();
 
@@ -233,11 +249,7 @@ describe("scheduler conversation reuse", () => {
     // WHEN the schedule fires for the first time
     forceScheduleDue(schedule.id);
 
-    const processMessage = async (conversationId: string, message: string) => {
-      processedMessages.push({ conversationId, message });
-    };
-
-    const scheduler1 = startScheduler(processMessage, () => {});
+    const scheduler1 = startScheduler();
     await new Promise((resolve) => setTimeout(resolve, 500));
     scheduler1.stop();
 
@@ -249,7 +261,7 @@ describe("scheduler conversation reuse", () => {
     forceScheduleDue(schedule.id);
     processedMessages.length = 0;
 
-    const scheduler2 = startScheduler(processMessage, () => {});
+    const scheduler2 = startScheduler();
     await new Promise((resolve) => setTimeout(resolve, 500));
     scheduler2.stop();
 
@@ -277,11 +289,7 @@ describe("scheduler conversation reuse", () => {
     // WHEN the schedule fires for the first time
     forceScheduleDue(schedule.id);
 
-    const processMessage = async (conversationId: string, message: string) => {
-      processedMessages.push({ conversationId, message });
-    };
-
-    const scheduler1 = startScheduler(processMessage, () => {});
+    const scheduler1 = startScheduler();
     await new Promise((resolve) => setTimeout(resolve, 500));
     scheduler1.stop();
 
@@ -292,7 +300,7 @@ describe("scheduler conversation reuse", () => {
     forceScheduleDue(schedule.id);
     processedMessages.length = 0;
 
-    const scheduler2 = startScheduler(processMessage, () => {});
+    const scheduler2 = startScheduler();
     await new Promise((resolve) => setTimeout(resolve, 500));
     scheduler2.stop();
 
@@ -320,11 +328,7 @@ describe("scheduler conversation reuse", () => {
 
     forceScheduleDue(schedule.id);
 
-    const processMessage = async (conversationId: string, message: string) => {
-      processedMessages.push({ conversationId, message });
-    };
-
-    const scheduler1 = startScheduler(processMessage, () => {});
+    const scheduler1 = startScheduler();
     await new Promise((resolve) => setTimeout(resolve, 500));
     scheduler1.stop();
 
@@ -338,7 +342,7 @@ describe("scheduler conversation reuse", () => {
     forceScheduleDue(schedule.id);
     processedMessages.length = 0;
 
-    const scheduler2 = startScheduler(processMessage, () => {});
+    const scheduler2 = startScheduler();
     await new Promise((resolve) => setTimeout(resolve, 500));
     scheduler2.stop();
 
@@ -364,11 +368,7 @@ describe("scheduler conversation reuse", () => {
     });
 
     // WHEN the schedule fires
-    const processMessage = async (conversationId: string, message: string) => {
-      processedMessages.push({ conversationId, message });
-    };
-
-    const scheduler = startScheduler(processMessage, () => {});
+    const scheduler = startScheduler();
     await new Promise((resolve) => setTimeout(resolve, 500));
     scheduler.stop();
 
@@ -403,12 +403,12 @@ describe("scheduler conversation reuse", () => {
     forceScheduleDue(schedule.id);
 
     let shouldFail = false;
-    const processMessage = async (conversationId: string, message: string) => {
+    processMessageImpl = async (conversationId, message) => {
       processedMessages.push({ conversationId, message });
       if (shouldFail) throw new Error("Simulated failure");
     };
 
-    const scheduler1 = startScheduler(processMessage, () => {});
+    const scheduler1 = startScheduler();
     await new Promise((resolve) => setTimeout(resolve, 500));
     scheduler1.stop();
 
@@ -420,7 +420,7 @@ describe("scheduler conversation reuse", () => {
     processedMessages.length = 0;
     shouldFail = true;
 
-    const scheduler2 = startScheduler(processMessage, () => {});
+    const scheduler2 = startScheduler();
     await new Promise((resolve) => setTimeout(resolve, 500));
     scheduler2.stop();
 
@@ -434,7 +434,7 @@ describe("scheduler conversation reuse", () => {
     processedMessages.length = 0;
     shouldFail = false;
 
-    const scheduler3 = startScheduler(processMessage, () => {});
+    const scheduler3 = startScheduler();
     await new Promise((resolve) => setTimeout(resolve, 500));
     scheduler3.stop();
 
@@ -453,6 +453,7 @@ describe("scheduler talk-mode runner option propagation", () => {
     db.run("DELETE FROM messages");
     db.run("DELETE FROM conversations");
     processedMessages.length = 0;
+    processMessageImpl = captureProcessedMessage;
     runBackgroundJobOptions.length = 0;
     runBackgroundJobShouldFail = false;
     runBackgroundJobBootstrapFails = false;
@@ -470,8 +471,7 @@ describe("scheduler talk-mode runner option propagation", () => {
     });
     forceScheduleDue(schedule.id);
 
-    const processMessage = async () => {};
-    const scheduler = startScheduler(processMessage, () => {});
+    const scheduler = startScheduler();
     await new Promise((resolve) => setTimeout(resolve, 500));
     scheduler.stop();
 
@@ -495,8 +495,7 @@ describe("scheduler talk-mode runner option propagation", () => {
     });
     forceScheduleDue(schedule.id);
 
-    const processMessage = async () => {};
-    const scheduler = startScheduler(processMessage, () => {});
+    const scheduler = startScheduler();
     await new Promise((resolve) => setTimeout(resolve, 500));
     scheduler.stop();
 
@@ -524,8 +523,7 @@ describe("scheduler talk-mode runner option propagation", () => {
     forceScheduleDue(schedule.id);
     runBackgroundJobBootstrapFails = true;
 
-    const processMessage = async () => {};
-    const scheduler = startScheduler(processMessage, () => {});
+    const scheduler = startScheduler();
     await new Promise((resolve) => setTimeout(resolve, 500));
     scheduler.stop();
 
@@ -554,15 +552,22 @@ describe("scheduler talk-mode runner option propagation", () => {
       scheduleJobId: string;
       title: string;
     }> = [];
-    const processMessage = async () => {};
-    const scheduler = startScheduler(
-      processMessage,
-      () => {},
-      undefined,
-      (info) => sseCalls.push(info),
-    );
+    const subscription = assistantEventHub.subscribe({
+      type: "process",
+      callback: (event) => {
+        if (event.message.type === "schedule_conversation_created") {
+          sseCalls.push({
+            conversationId: event.message.conversationId,
+            scheduleJobId: event.message.scheduleJobId,
+            title: event.message.title,
+          });
+        }
+      },
+    });
+    const scheduler = startScheduler();
     await new Promise((resolve) => setTimeout(resolve, 500));
     scheduler.stop();
+    subscription.dispose();
 
     expect(sseCalls).toHaveLength(1);
     expect(sseCalls[0]).toMatchObject({