@vellumai/assistant 0.8.5 → 0.8.7

package/src/__tests__/mcp-abort-signal.test.ts

@@ -8,7 +8,21 @@ mock.module("../security/secure-keys.js", () => ({
 }));
 
 mock.module("../config/env-registry.js", () => ({
+  getDebugStdoutLogs: () => false,
+  getIsContainerized: () => false,
   getIsPlatform: () => false,
+  isPlatformRemote: () => false,
+  getWorkspaceDirOverride: () => undefined,
+  getBackupDirOverride: () => undefined,
+  getBackupKeyPathOverride: () => undefined,
+  getCpuLimit: () => undefined,
+  getMinikubeStorageSize: () => undefined,
+  getProfilerRunId: () => undefined,
+  getProfilerMode: () => undefined,
+  getProfilerMaxBytes: () => undefined,
+  getProfilerMaxRuns: () => undefined,
+  getProfilerMinFreeMb: () => undefined,
+  checkUnrecognizedEnvVars: () => [],
 }));
 
 const { McpClient } = await import("../mcp/client.js");

package/src/__tests__/mcp-auth-routes.test.ts

@@ -24,19 +24,24 @@ mock.module("../mcp/mcp-auth-state.js", () => ({
   getMcpAuthState: mockGetMcpAuthState,
 }));
 
-mock.module("../config/loader.js", () => ({
-  loadRawConfig: () => ({
-    mcp: {
-      servers: {
-        "my-server": {
-          transport: { type: "sse", url: "https://mcp.example.com" },
-          enabled: true,
-          defaultRiskLevel: "high",
-        },
+const mockConfig = {
+  mcp: {
+    servers: {
+      "my-server": {
+        transport: { type: "sse", url: "https://mcp.example.com" },
+        enabled: true,
+        defaultRiskLevel: "high",
       },
     },
-  }),
+  },
+};
+mock.module("../config/loader.js", () => ({
+  loadRawConfig: () => mockConfig,
   saveRawConfig: () => {},
+  // route-policy → config/env transitively imports `getConfig`; stub it
+  // here so the mock surface matches what consumers expect.
+  getConfig: () => mockConfig,
+  getConfigReadOnly: () => mockConfig,
 }));
 
 mock.module("../util/logger.js", () => ({

package/src/__tests__/mcp-client-auth.test.ts

@@ -8,7 +8,21 @@ mock.module("../security/secure-keys.js", () => ({
 }));
 
 mock.module("../config/env-registry.js", () => ({
+  getDebugStdoutLogs: () => false,
+  getIsContainerized: () => false,
   getIsPlatform: () => false,
+  isPlatformRemote: () => false,
+  getWorkspaceDirOverride: () => undefined,
+  getBackupDirOverride: () => undefined,
+  getBackupKeyPathOverride: () => undefined,
+  getCpuLimit: () => undefined,
+  getMinikubeStorageSize: () => undefined,
+  getProfilerRunId: () => undefined,
+  getProfilerMode: () => undefined,
+  getProfilerMaxBytes: () => undefined,
+  getProfilerMaxRuns: () => undefined,
+  getProfilerMinFreeMb: () => undefined,
+  checkUnrecognizedEnvVars: () => [],
 }));
 
 const { McpClient } = await import("../mcp/client.js");

package/src/__tests__/mcp-health-check.test.ts

@@ -18,23 +18,28 @@ mock.module("../mcp/client.js", () => ({
   },
 }));
 
-mock.module("../config/loader.js", () => ({
-  loadRawConfig: () => ({
-    mcp: {
-      servers: {
-        test: {
-          transport: {
-            type: "streamable-http",
-            url: "https://example.com/mcp",
-          },
-          enabled: true,
-          defaultRiskLevel: "high",
-          maxTools: 20,
+const mockConfig = {
+  mcp: {
+    servers: {
+      test: {
+        transport: {
+          type: "streamable-http",
+          url: "https://example.com/mcp",
         },
+        enabled: true,
+        defaultRiskLevel: "high",
+        maxTools: 20,
       },
     },
-  }),
+  },
+};
+mock.module("../config/loader.js", () => ({
+  loadRawConfig: () => mockConfig,
   saveRawConfig: () => {},
+  // route-policy → config/env transitively imports `getConfig`; stub it
+  // here so the mock surface matches what consumers expect.
+  getConfig: () => mockConfig,
+  getConfigReadOnly: () => mockConfig,
 }));
 
 mock.module("../daemon/mcp-reload-service.js", () => ({

package/src/__tests__/memory-retrieval-pipeline.test.ts

@@ -29,7 +29,7 @@ import {
   type DefaultMemoryRetrievalDeps,
   defaultMemoryRetrievalPlugin,
   runDefaultMemoryRetrieval,
-} from "../plugins/defaults/memory-retrieval.js";
+} from "../plugins/defaults/memory-retrieval/register.js";
 import { DEFAULT_TIMEOUTS, runPipeline } from "../plugins/pipeline.js";
 import {
   getMiddlewaresFor,

package/src/__tests__/memory-v2-static-injector.test.ts

@@ -15,7 +15,7 @@
 
 import { describe, expect, test } from "bun:test";
 
-import { defaultInjectorsPlugin } from "../plugins/defaults/injectors.js";
+import { defaultInjectorsPlugin } from "../plugins/defaults/injectors/register.js";
 import type { Injector, TurnContext } from "../plugins/types.js";
 
 function findInjector(name: string): Injector {

package/src/__tests__/messaging-send-tool.test.ts

@@ -48,8 +48,10 @@ const addMessageMock = mock(
     conversationId: string,
     role: string,
     content: string,
-    _metadata?: Record<string, unknown>,
-    _opts?: { skipIndexing?: boolean },
+    _options?: {
+      metadata?: Record<string, unknown>;
+      skipIndexing?: boolean;
+    },
   ) => ({
     id: "xpost-msg-1",
     conversationId,
@@ -79,6 +81,7 @@ const getBindingByChannelChatMock = mock(
 mock.module("../memory/conversation-crud.js", () => ({
   addMessage: addMessageMock,
   getConversation: getConversationMock,
+  reserveMessage: mock(async () => ({ id: "msg-reserve" })),
 }));
 
 mock.module("../memory/conversation-disk-view.js", () => ({
@@ -196,8 +199,10 @@ describe("messaging-send tool", () => {
       "bound-conv-99",
       "assistant",
       JSON.stringify([{ type: "text", text: "hello from A" }]),
-      { automated: true, crossPostedFrom: "conv-A" },
-      { skipIndexing: true },
+      {
+        metadata: { automated: true, crossPostedFrom: "conv-A" },
+        skipIndexing: true,
+      },
     );
     expect(syncMessageToDiskMock).toHaveBeenCalledWith(
       "bound-conv-99",

package/src/__tests__/migration-export-http.test.ts

@@ -500,10 +500,10 @@ describe("export config sanitization", () => {
 
 describe("route policy registration", () => {
   test("migrations/export policy requires settings.write scope", async () => {
-    const { getPolicy } = await import("../runtime/auth/route-policy.js");
-    const policy = getPolicy("migrations/export");
+    const { ROUTES } = await import("../runtime/routes/index.js");
+    const policy = (ROUTES.find((r) => r.endpoint === "migrations/export")?.policy ?? null);
 
-    expect(policy).toBeDefined();
+    expect(policy).not.toBeNull();
     expect(policy?.requiredScopes).toContain("settings.write");
     expect(policy?.allowedPrincipalTypes).toContain("actor");
     expect(policy?.allowedPrincipalTypes).toContain("svc_gateway");
@@ -511,10 +511,10 @@ describe("route policy registration", () => {
   });
 
   test("migrations/validate policy is still registered", async () => {
-    const { getPolicy } = await import("../runtime/auth/route-policy.js");
-    const policy = getPolicy("migrations/validate");
+    const { ROUTES } = await import("../runtime/routes/index.js");
+    const policy = (ROUTES.find((r) => r.endpoint === "migrations/validate")?.policy ?? null);
 
-    expect(policy).toBeDefined();
+    expect(policy).not.toBeNull();
     expect(policy?.requiredScopes).toContain("settings.read");
   });
 });
@@ -525,10 +525,10 @@ describe("route policy registration", () => {
 
 describe("auth policy shape", () => {
   test("export policy requires settings.write and would deny without it", async () => {
-    const { getPolicy } = await import("../runtime/auth/route-policy.js");
-    const policy = getPolicy("migrations/export");
+    const { ROUTES } = await import("../runtime/routes/index.js");
+    const policy = (ROUTES.find((r) => r.endpoint === "migrations/export")?.policy ?? null);
 
-    expect(policy).toBeDefined();
+    expect(policy).not.toBeNull();
     // Verify the policy shape means a caller without settings.write would be denied
     expect(policy!.requiredScopes).toEqual(["settings.write"]);
     // Verify only the expected principal types are allowed
@@ -539,9 +539,9 @@ describe("auth policy shape", () => {
   });
 
   test("export policy differs from validate policy on scopes (validate is read-only)", async () => {
-    const { getPolicy } = await import("../runtime/auth/route-policy.js");
-    const exportPolicy = getPolicy("migrations/export");
-    const validatePolicy = getPolicy("migrations/validate");
+    const { ROUTES } = await import("../runtime/routes/index.js");
+    const exportPolicy = (ROUTES.find((r) => r.endpoint === "migrations/export")?.policy ?? null);
+    const validatePolicy = (ROUTES.find((r) => r.endpoint === "migrations/validate")?.policy ?? null);
 
     // validate is read-only so requires settings.read; export requires settings.write
     expect(exportPolicy!.requiredScopes).toEqual(["settings.write"]);

package/src/__tests__/migration-import-commit-http.test.ts

@@ -1129,10 +1129,10 @@ describe("commitImport — config sanitization", () => {
 
 describe("route policy registration", () => {
   test("migrations/import policy requires settings.write scope", async () => {
-    const { getPolicy } = await import("../runtime/auth/route-policy.js");
-    const policy = getPolicy("migrations/import");
+    const { ROUTES } = await import("../runtime/routes/index.js");
+    const policy = (ROUTES.find((r) => r.endpoint === "migrations/import")?.policy ?? null);
 
-    expect(policy).toBeDefined();
+    expect(policy).not.toBeNull();
     expect(policy?.requiredScopes).toContain("settings.write");
     expect(policy?.allowedPrincipalTypes).toContain("actor");
     expect(policy?.allowedPrincipalTypes).toContain("svc_gateway");
@@ -1140,11 +1140,11 @@ describe("route policy registration", () => {
   });
 
   test("import policy matches export/preflight but differs from validate on scopes", async () => {
-    const { getPolicy } = await import("../runtime/auth/route-policy.js");
-    const importPolicy = getPolicy("migrations/import");
-    const validatePolicy = getPolicy("migrations/validate");
-    const exportPolicy = getPolicy("migrations/export");
-    const preflightPolicy = getPolicy("migrations/import-preflight");
+    const { ROUTES } = await import("../runtime/routes/index.js");
+    const importPolicy = (ROUTES.find((r) => r.endpoint === "migrations/import")?.policy ?? null);
+    const validatePolicy = (ROUTES.find((r) => r.endpoint === "migrations/validate")?.policy ?? null);
+    const exportPolicy = (ROUTES.find((r) => r.endpoint === "migrations/export")?.policy ?? null);
+    const preflightPolicy = (ROUTES.find((r) => r.endpoint === "migrations/import-preflight")?.policy ?? null);
 
     // import, export, and preflight all require settings.write
     expect(importPolicy!.requiredScopes).toEqual(exportPolicy!.requiredScopes);

package/src/__tests__/migration-import-from-url.test.ts

@@ -118,13 +118,13 @@ mock.module("../config/env.js", () => ({
 // Imports (after mocks so module-level code picks up the stubs)
 // ---------------------------------------------------------------------------
 
-import { resetDb } from "../memory/db-connection.js";
 import { defaultV1Options } from "../runtime/migrations/__tests__/v1-test-helpers.js";
 import { buildVBundle } from "../runtime/migrations/vbundle-builder.js";
 import {
   _setUrlImportValidatorOptionsForTests,
   handleMigrationImport,
 } from "../runtime/routes/migration-routes.js";
+import { resetDbForTesting } from "./db-test-helpers.js";
 import { callHandler } from "./helpers/call-route-handler.js";
 // ---------------------------------------------------------------------------
 // Local http fixture server
@@ -448,7 +448,7 @@ describe("handleMigrationImport — no-swap path omits newer-migration warning",
 
     // Drop any cached Drizzle singleton so getDb() re-opens from the
     // seeded path above when the handler calls it post-import.
-    resetDb();
+    resetDbForTesting();
 
     // All-`vellum:*` credentials bundle: the streaming importer returns
     // ok=true with zero files_created/overwritten (no-swap success),
@@ -505,7 +505,7 @@ describe("handleMigrationImport — no-swap path omits newer-migration warning",
       await fixture.close();
       // Close the cached DB handle before the workspace dir gets rm'd
       // in afterEach so we don't leak WAL/SHM files across tests.
-      resetDb();
+      resetDbForTesting();
     }
   });
 });

package/src/__tests__/migration-import-preflight-http.test.ts

@@ -710,10 +710,10 @@ describe("DefaultPathResolver", () => {
 
 describe("route policy registration", () => {
   test("migrations/import-preflight policy requires settings.write scope", async () => {
-    const { getPolicy } = await import("../runtime/auth/route-policy.js");
-    const policy = getPolicy("migrations/import-preflight");
+    const { ROUTES } = await import("../runtime/routes/index.js");
+    const policy = (ROUTES.find((r) => r.endpoint === "migrations/import-preflight")?.policy ?? null);
 
-    expect(policy).toBeDefined();
+    expect(policy).not.toBeNull();
     expect(policy?.requiredScopes).toContain("settings.write");
     expect(policy?.allowedPrincipalTypes).toContain("actor");
     expect(policy?.allowedPrincipalTypes).toContain("svc_gateway");
@@ -721,10 +721,10 @@ describe("route policy registration", () => {
   });
 
   test("import-preflight policy matches export but differs from validate on scopes", async () => {
-    const { getPolicy } = await import("../runtime/auth/route-policy.js");
-    const preflightPolicy = getPolicy("migrations/import-preflight");
-    const validatePolicy = getPolicy("migrations/validate");
-    const exportPolicy = getPolicy("migrations/export");
+    const { ROUTES } = await import("../runtime/routes/index.js");
+    const preflightPolicy = (ROUTES.find((r) => r.endpoint === "migrations/import-preflight")?.policy ?? null);
+    const validatePolicy = (ROUTES.find((r) => r.endpoint === "migrations/validate")?.policy ?? null);
+    const exportPolicy = (ROUTES.find((r) => r.endpoint === "migrations/export")?.policy ?? null);
 
     // preflight and export both require settings.write
     expect(preflightPolicy!.requiredScopes).toEqual(

package/src/__tests__/migration-validate-http.test.ts

@@ -630,10 +630,10 @@ describe("handleMigrationValidate", () => {
 describe("route policy registration", () => {
   test("migrations/validate policy requires settings.read scope", async () => {
     // Import route-policy to verify the registration exists
-    const { getPolicy } = await import("../runtime/auth/route-policy.js");
-    const policy = getPolicy("migrations/validate");
+    const { ROUTES } = await import("../runtime/routes/index.js");
+    const policy = (ROUTES.find((r) => r.endpoint === "migrations/validate")?.policy ?? null);
 
-    expect(policy).toBeDefined();
+    expect(policy).not.toBeNull();
     expect(policy?.requiredScopes).toContain("settings.read");
     expect(policy?.allowedPrincipalTypes).toContain("actor");
     expect(policy?.allowedPrincipalTypes).toContain("svc_gateway");

package/src/__tests__/mock-gateway-ipc.ts

@@ -56,7 +56,9 @@ class FakePersistentIpcClient extends EventEmitter {
     if (simulateError) {
       throw new Error("Mock IPC socket error");
     }
-    return method in ipcResults ? ipcResults[method] : undefined;
+    if (method in ipcResults) return ipcResults[method];
+    if (method === "get_feature_flags") return GET_FEATURE_FLAGS_DEFAULT;
+    return undefined;
   }
 
   destroy(): void {
@@ -68,6 +70,18 @@ class FakePersistentIpcClient extends EventEmitter {
 // Register the mock (called once from test-preload.ts)
 // ---------------------------------------------------------------------------
 
+/**
+ * Sentinel returned by the mock for `get_feature_flags` when a test has not
+ * explicitly configured a result. Keeps the response non-empty so
+ * `initFeatureFlagOverrides()` does not enter its 7.75 s empty-result retry
+ * loop during tests that build the CLI program or otherwise trigger flag
+ * initialization. The sentinel key starts with a double underscore so it
+ * cannot collide with any real registry-declared flag.
+ */
+const GET_FEATURE_FLAGS_DEFAULT: Record<string, boolean> = {
+  __test_default__: false,
+};
+
 export function installGatewayIpcMock(): void {
   mock.module("@vellumai/gateway-client/ipc-client", () => ({
     ipcCall: async (
@@ -79,7 +93,9 @@ export function installGatewayIpcMock(): void {
         // Real ipcCall returns undefined on failure — mirror that behavior.
         return undefined;
       }
-      return method in ipcResults ? ipcResults[method] : undefined;
+      if (method in ipcResults) return ipcResults[method];
+      if (method === "get_feature_flags") return GET_FEATURE_FLAGS_DEFAULT;
+      return undefined;
     },
     PersistentIpcClient: FakePersistentIpcClient,
   }));

package/src/__tests__/model-intents.test.ts

@@ -20,7 +20,7 @@ describe("model intents", () => {
       "claude-haiku-4-5-20251001",
     );
     expect(resolveModelIntent("anthropic", "quality-optimized")).toBe(
-      "claude-opus-4-7",
+      "claude-opus-4-8",
     );
     expect(resolveModelIntent("anthropic", "vision-optimized")).toBe(
       "claude-opus-4-6",
@@ -44,9 +44,9 @@ describe("model intents", () => {
   });
 
   test("falls back to provider default for unknown providers", () => {
-    expect(getProviderDefaultModel("unknown-provider")).toBe("claude-opus-4-7");
+    expect(getProviderDefaultModel("unknown-provider")).toBe("claude-opus-4-8");
     expect(resolveModelIntent("unknown-provider", "quality-optimized")).toBe(
-      "claude-opus-4-7",
+      "claude-opus-4-8",
     );
   });
 });

package/src/__tests__/native-web-search.test.ts

@@ -51,8 +51,7 @@ mock.module("@anthropic-ai/sdk", () => ({
           _options?: Record<string, unknown>,
         ) => {
           lastStreamParams = JSON.parse(JSON.stringify(params));
-          const handlers: Record<string, ((...args: unknown[]) => void)[]> =
-            {};
+          const handlers: Record<string, ((...args: unknown[]) => void)[]> = {};
           return {
             on(event: string, cb: (...args: unknown[]) => void) {
               (handlers[event] ??= []).push(cb);
@@ -74,6 +73,7 @@ mock.module("@anthropic-ai/sdk", () => ({
 
 // Import after mocking
 import { AnthropicProvider } from "../providers/anthropic/client.js";
+import { isNativeWebSearchCapableProvider } from "../providers/registry.js";
 
 // ---------------------------------------------------------------------------
 // Helpers
@@ -99,6 +99,34 @@ const sampleTools: ToolDefinition[] = [
   },
 ];
 
+describe("Native Web Search — Selection Semantics", () => {
+  test("only native-capable inference providers/models request provider-native web search", () => {
+    expect(
+      isNativeWebSearchCapableProvider("anthropic", "claude-opus-4-7"),
+    ).toBe(true);
+    expect(isNativeWebSearchCapableProvider("openai", "gpt-5")).toBe(true);
+    expect(
+      isNativeWebSearchCapableProvider(
+        "openrouter",
+        "anthropic/claude-opus-4-7",
+      ),
+    ).toBe(true);
+
+    expect(
+      isNativeWebSearchCapableProvider(
+        "fireworks",
+        "accounts/fireworks/models/kimi-k2p6",
+      ),
+    ).toBe(false);
+    expect(isNativeWebSearchCapableProvider("gemini", "gemini-3.1-pro")).toBe(
+      false,
+    );
+    expect(isNativeWebSearchCapableProvider("openrouter", "openai/gpt-5")).toBe(
+      false,
+    );
+  });
+});
+
 // ---------------------------------------------------------------------------
 // Tests — Round-trip: fromAnthropicBlock
 // ---------------------------------------------------------------------------
@@ -296,7 +324,7 @@ describe("Native Web Search — Tool Filtering", () => {
       useNativeWebSearch: true,
     });
 
-    await provider.sendMessage([userMsg("Hi")], sampleTools);
+    await provider.sendMessage([userMsg("Hi")], { tools: sampleTools });
 
     const tools = lastStreamParams!.tools as Array<Record<string, unknown>>;
 
@@ -323,7 +351,7 @@ describe("Native Web Search — Tool Filtering", () => {
       useNativeWebSearch: false,
     });
 
-    await provider.sendMessage([userMsg("Hi")], sampleTools);
+    await provider.sendMessage([userMsg("Hi")], { tools: sampleTools });
 
     const tools = lastStreamParams!.tools as Array<Record<string, unknown>>;
 
@@ -357,7 +385,9 @@ describe("Native Web Search — Tool Filtering", () => {
       },
     ];
 
-    await provider.sendMessage([userMsg("Hi")], toolsWithoutWebSearch);
+    await provider.sendMessage([userMsg("Hi")], {
+      tools: toolsWithoutWebSearch,
+    });
 
     const tools = lastStreamParams!.tools as Array<Record<string, unknown>>;
     expect(tools).toHaveLength(1);
@@ -370,7 +400,7 @@ describe("Native Web Search — Tool Filtering", () => {
       useNativeWebSearch: true,
     });
 
-    await provider.sendMessage([userMsg("Hi")], sampleTools);
+    await provider.sendMessage([userMsg("Hi")], { tools: sampleTools });
 
     const tools = lastStreamParams!.tools as Array<{
       name: string;
@@ -417,14 +447,10 @@ describe("Native Web Search — Streaming Events", () => {
     ];
 
     const events: ProviderEvent[] = [];
-    await provider.sendMessage(
-      [userMsg("Search something")],
-      sampleTools,
-      undefined,
-      {
-        onEvent: (event) => events.push(event),
-      },
-    );
+    await provider.sendMessage([userMsg("Search something")], {
+      tools: sampleTools,
+      onEvent: (event) => events.push(event),
+    });
 
     const serverToolEvents = events.filter(
       (e) => e.type === "server_tool_start",
@@ -456,14 +482,10 @@ describe("Native Web Search — Streaming Events", () => {
     ];
 
     const events: ProviderEvent[] = [];
-    await provider.sendMessage(
-      [userMsg("Read a file")],
-      sampleTools,
-      undefined,
-      {
-        onEvent: (event) => events.push(event),
-      },
-    );
+    await provider.sendMessage([userMsg("Read a file")], {
+      tools: sampleTools,
+      onEvent: (event) => events.push(event),
+    });
 
     const serverToolEvents = events.filter(
       (e) => e.type === "server_tool_start",

package/src/__tests__/notification-decision-identity.test.ts

@@ -107,12 +107,9 @@ describe("identity context in notification decision engine", () => {
     mockIdentityContext = "I am Jarvis, a helpful assistant";
 
     configuredProvider = {
-      sendMessage: async (
-        _messages: unknown,
-        _tools: unknown,
-        systemPrompt: unknown,
-      ) => {
-        capturedSystemPrompt = systemPrompt as string;
+      sendMessage: async (...args: unknown[]) => {
+        const options = args[1] as { systemPrompt?: string } | undefined;
+        capturedSystemPrompt = options?.systemPrompt;
         return { content: [] };
       },
     };
@@ -146,12 +143,9 @@ describe("identity context in notification decision engine", () => {
     mockIdentityContext = null;
 
     configuredProvider = {
-      sendMessage: async (
-        _messages: unknown,
-        _tools: unknown,
-        systemPrompt: unknown,
-      ) => {
-        capturedSystemPrompt = systemPrompt as string;
+      sendMessage: async (...args: unknown[]) => {
+        const options = args[1] as { systemPrompt?: string } | undefined;
+        capturedSystemPrompt = options?.systemPrompt;
         return { content: [] };
       },
     };
@@ -185,12 +179,9 @@ describe("identity context in notification decision engine", () => {
     mockIdentityContext = "A".repeat(3000);
 
     configuredProvider = {
-      sendMessage: async (
-        _messages: unknown,
-        _tools: unknown,
-        systemPrompt: unknown,
-      ) => {
-        capturedSystemPrompt = systemPrompt as string;
+      sendMessage: async (...args: unknown[]) => {
+        const options = args[1] as { systemPrompt?: string } | undefined;
+        capturedSystemPrompt = options?.systemPrompt;
         return { content: [] };
       },
     };

package/src/__tests__/notification-decision-recipient-context.test.ts

@@ -104,12 +104,9 @@ function makeSignal(
 
 function setupLLMProvider() {
   configuredProvider = {
-    sendMessage: async (
-      _messages: unknown,
-      _tools: unknown,
-      systemPrompt: unknown,
-    ) => {
-      capturedSystemPrompt = systemPrompt as string;
+    sendMessage: async (...args: unknown[]) => {
+      const options = args[1] as { systemPrompt?: string } | undefined;
+      capturedSystemPrompt = options?.systemPrompt;
       return { content: [] };
     },
   };