diff --git a/src/commands/model-picker.test.ts b/src/commands/model-picker.test.ts index 0d657bd7cc2..81f4f2a9ede 100644 --- a/src/commands/model-picker.test.ts +++ b/src/commands/model-picker.test.ts @@ -144,6 +144,13 @@ describe("promptDefaultModel", () => { }); expect(runProviderPluginAuthMethod).toHaveBeenCalledOnce(); + expect(resolvePluginProviders).toHaveBeenCalledWith({ + config, + workspaceDir: undefined, + env: undefined, + bundledProviderAllowlistCompat: true, + bundledProviderVitestCompat: true, + }); expect(result.model).toBe("vllm/meta-llama/Meta-Llama-3-8B-Instruct"); expect(result.config?.models?.providers?.vllm).toMatchObject({ baseUrl: "http://127.0.0.1:8000/v1", diff --git a/src/flows/model-picker.ts b/src/flows/model-picker.ts index 7ebb99206bc..226c36bd35a 100644 --- a/src/flows/model-picker.ts +++ b/src/flows/model-picker.ts @@ -287,6 +287,8 @@ async function maybeHandleProviderPluginSelection(params: { config: params.cfg, workspaceDir: params.workspaceDir, env: params.env, + bundledProviderAllowlistCompat: true, + bundledProviderVitestCompat: true, }); pluginResolution = pluginProviders.some( (provider) => normalizeProviderId(provider.id) === normalizeProviderId(params.selection), @@ -315,6 +317,8 @@ async function maybeHandleProviderPluginSelection(params: { config: params.cfg, workspaceDir: params.workspaceDir, env: params.env, + bundledProviderAllowlistCompat: true, + bundledProviderVitestCompat: true, }); } const resolved = resolveProviderPluginChoice({