From 49c21b6c69061d5c7f55b4ba637875f2790de64e Mon Sep 17 00:00:00 2001 From: Matt Rubens Date: Wed, 5 Feb 2025 11:57:51 -0500 Subject: [PATCH] Add new gemini models --- .changeset/violet-paws-own.md | 5 +++++ src/api/providers/__tests__/gemini.test.ts | 2 +- src/shared/api.ts | 26 +++++++++++++++++++++- 3 files changed, 31 insertions(+), 2 deletions(-) create mode 100644 .changeset/violet-paws-own.md diff --git a/.changeset/violet-paws-own.md b/.changeset/violet-paws-own.md new file mode 100644 index 000000000..1e313e6ea --- /dev/null +++ b/.changeset/violet-paws-own.md @@ -0,0 +1,5 @@ +--- +"roo-cline": patch +--- + +Add new Gemini models diff --git a/src/api/providers/__tests__/gemini.test.ts b/src/api/providers/__tests__/gemini.test.ts index e8a5594fa..e57ba2ea7 100644 --- a/src/api/providers/__tests__/gemini.test.ts +++ b/src/api/providers/__tests__/gemini.test.ts @@ -204,7 +204,7 @@ describe("GeminiHandler", () => { geminiApiKey: "test-key", }) const modelInfo = invalidHandler.getModel() - expect(modelInfo.id).toBe("gemini-2.0-flash-thinking-exp-01-21") // Default model + expect(modelInfo.id).toBe("gemini-2.0-flash-001") // Default model }) }) }) diff --git a/src/shared/api.ts b/src/shared/api.ts index 451c365a0..39bc2a69c 100644 --- a/src/shared/api.ts +++ b/src/shared/api.ts @@ -429,8 +429,32 @@ export const openAiModelInfoSaneDefaults: ModelInfo = { // Gemini // https://ai.google.dev/gemini-api/docs/models/gemini export type GeminiModelId = keyof typeof geminiModels -export const geminiDefaultModelId: GeminiModelId = "gemini-2.0-flash-thinking-exp-01-21" +export const geminiDefaultModelId: GeminiModelId = "gemini-2.0-flash-001" export const geminiModels = { + "gemini-2.0-flash-001": { + maxTokens: 8192, + contextWindow: 1_048_576, + supportsImages: true, + supportsPromptCache: false, + inputPrice: 0, + outputPrice: 0, + }, + "gemini-2.0-flash-lite-preview-02-05": { + maxTokens: 8192, + contextWindow: 1_048_576, + supportsImages: true, + supportsPromptCache: false, + inputPrice: 0, + outputPrice: 0, + }, + "gemini-2.0-pro-exp-02-05": { + maxTokens: 8192, + contextWindow: 2_097_152, + supportsImages: true, + supportsPromptCache: false, + inputPrice: 0, + outputPrice: 0, + }, "gemini-2.0-flash-thinking-exp-01-21": { maxTokens: 65_536, contextWindow: 1_048_576,