Skip to content

Commit 6dab2cd

Browse files
committed
fix: gemini-1.5 and gemini-exp to be migrated to geminiDefaultModelId
1 parent 5f326c4 commit 6dab2cd

File tree

3 files changed

+8
-8
lines changed

3 files changed

+8
-8
lines changed

src/api/providers/gemini.ts

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -48,19 +48,19 @@ export class GeminiHandler extends BaseProvider implements SingleCompletionHandl
4848
}
4949

5050
if (modelId.startsWith("gemini-1.5-pro-")) {
51-
return "gemini-2.0-pro-exp-02-05"
51+
return geminiDefaultModelId
5252
}
5353

5454
if (modelId.startsWith("gemini-1.5-flash-")) {
55-
return "gemini-2.0-flash-001"
55+
return geminiDefaultModelId
5656
}
5757

5858
if (modelId === "gemini-2.5-pro-exp-") {
5959
return "gemini-2.5-pro"
6060
}
6161

6262
if (modelId === "gemini-exp-1206") {
63-
return "gemini-2.0-pro-exp-02-05"
63+
return geminiDefaultModelId
6464
}
6565
}
6666

src/api/providers/vertex.ts

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,7 @@ export class VertexHandler extends GeminiHandler implements SingleCompletionHand
3232
}
3333

3434
if (modelId.startsWith("gemini-1.5-pro-")) {
35-
return "gemini-2.0-pro-exp-02-05"
35+
return "gemini-2.0-flash-001"
3636
}
3737

3838
if (modelId.startsWith("gemini-1.5-flash-")) {

webview-ui/src/components/ui/hooks/useSelectedModel.ts

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -55,19 +55,19 @@ function mapLegacyGeminiModel(modelId: string): string {
5555
}
5656

5757
if (modelId.startsWith("gemini-1.5-pro-")) {
58-
return "gemini-2.0-pro-exp-02-05"
58+
return geminiDefaultModelId
5959
}
6060

6161
if (modelId.startsWith("gemini-1.5-flash-")) {
62-
return "gemini-2.0-flash-001"
62+
return geminiDefaultModelId
6363
}
6464

6565
if (modelId === "gemini-2.5-pro-exp-03-25") {
6666
return "gemini-2.5-pro"
6767
}
6868

6969
if (modelId === "gemini-exp-1206") {
70-
return "gemini-2.0-pro-exp-02-05"
70+
return geminiDefaultModelId
7171
}
7272
}
7373

@@ -88,7 +88,7 @@ function mapLegacyVertexModel(modelId: string): string {
8888
}
8989

9090
if (modelId.startsWith("gemini-1.5-pro-")) {
91-
return "gemini-2.0-pro-exp-02-05"
91+
return "gemini-2.0-flash-001"
9292
}
9393

9494
if (modelId.startsWith("gemini-1.5-flash-")) {

0 commit comments

Comments
 (0)