fix(vscode): infer autocomplete provider from direct models

This commit is contained in:
Mark IJbema
2026-05-26 10:25:36 +02:00
parent e5b2bb8628
commit 6fb50a8942
2 changed files with 13 additions and 0 deletions
@@ -83,6 +83,11 @@ export function getAutocompleteModel(provider?: string, model?: string): Autocom
return DEFAULT_AUTOCOMPLETE_MODEL
}
if (!provider) {
const direct = models.find((m) => m.directProvider && m.modelID === model)
if (direct) return direct
}
const pid = provider || "kilo"
const mid = aliases[model ?? ""] ?? model
for (const m of models) {
@@ -33,6 +33,14 @@ describe("autocomplete settings", () => {
expect(buildAutocompleteSettingsMessage().settings.model).toBe("mercury-edit-2")
})
it("infers direct provider from native direct model choices", async () => {
state.set("model", "mercury-edit-2")
const { buildAutocompleteSettingsMessage } = await import("../settings")
expect(buildAutocompleteSettingsMessage().settings.provider).toBe("inception")
expect(buildAutocompleteSettingsMessage().settings.model).toBe("mercury-edit-2")
})
it("defaults to codestral when no model is set", async () => {
const { buildAutocompleteSettingsMessage } = await import("../settings")