mirror of
https://github.com/Kilo-Org/kilocode.git
synced 2026-09-24 16:02:55 +08:00
fix(vscode): infer autocomplete provider from direct models
This commit is contained in:
@@ -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")
|
||||
|
||||
|
||||
Reference in New Issue
Block a user