Skip to content

Commit 4207d57

Browse files
committed
reuse constant
1 parent 13e2f61 commit 4207d57

File tree

2 files changed

+3
-9
lines changed

2 files changed

+3
-9
lines changed

src/services/ghost/new-auto-complete/NewAutocompleteModel.ts

Lines changed: 1 addition & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
import {
2+
AUTOCOMPLETE_PROVIDER_MODELS,
23
AutocompleteProviderKey,
34
defaultProviderUsabilityChecker,
45
getKiloBaseUriFromToken,
@@ -16,13 +17,6 @@ import Mistral from "../../continuedev/core/llm/llms/Mistral"
1617
import OpenRouter from "../../continuedev/core/llm/llms/OpenRouter"
1718
import KiloCode from "../../continuedev/core/llm/llms/KiloCode"
1819

19-
export const AUTOCOMPLETE_PROVIDER_MODELS = {
20-
mistral: "codestral-2508",
21-
kilocode: "codestral-2508",
22-
openrouter: "mistralai/codestral-2508",
23-
bedrock: "mistral.codestral-2508-v1:0",
24-
} as const
25-
2620
export class NewAutocompleteModel {
2721
private apiHandler: ApiHandler | null = null
2822
private profile: ProviderSettings | null = null

src/services/ghost/new-auto-complete/__tests__/NewAutocompleteModel.spec.ts

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,7 @@ describe("NewAutocompleteModel", () => {
3131
expect(result).toBeDefined()
3232
expect(Mistral).toHaveBeenCalledWith(
3333
expect.objectContaining({
34-
model: "codestral-2508",
34+
model: "codestral-latest",
3535
apiKey: "test-mistral-key",
3636
apiBase: "https://custom.mistral.ai/v1/",
3737
contextLength: 32000,
@@ -78,7 +78,7 @@ describe("NewAutocompleteModel", () => {
7878
expect(result).toBeDefined()
7979
expect(OpenAI).toHaveBeenCalledWith(
8080
expect.objectContaining({
81-
model: "codestral-2508",
81+
model: "mistralai/codestral-2508",
8282
apiKey: "test-kilocode-token",
8383
}),
8484
)

0 commit comments

Comments
 (0)