diff --git a/packages/core/src/models.ts b/packages/core/src/models.ts index 99e8507821a..585b5bb0c1e 100644 --- a/packages/core/src/models.ts +++ b/packages/core/src/models.ts @@ -189,15 +189,15 @@ export const models: Models = { [ModelClass.SMALL]: settings.SMALL_GOOGLE_MODEL || settings.GOOGLE_MODEL || - "gemini-1.5-flash-latest", + "gemini-2.0-flash-exp", [ModelClass.MEDIUM]: settings.MEDIUM_GOOGLE_MODEL || settings.GOOGLE_MODEL || - "gemini-1.5-flash-latest", + "gemini-2.0-flash-exp", [ModelClass.LARGE]: settings.LARGE_GOOGLE_MODEL || settings.GOOGLE_MODEL || - "gemini-1.5-pro-latest", + "gemini-2.0-flash-exp", [ModelClass.EMBEDDING]: settings.EMBEDDING_GOOGLE_MODEL || settings.GOOGLE_MODEL || diff --git a/packages/core/src/tests/models.test.ts b/packages/core/src/tests/models.test.ts index f336093cfdd..f469b6976c4 100644 --- a/packages/core/src/tests/models.test.ts +++ b/packages/core/src/tests/models.test.ts @@ -96,9 +96,9 @@ describe("Model Provider Configuration", () => { describe("Google Provider", () => { test("should have correct model mappings", () => { const googleModels = models[ModelProviderName.GOOGLE].model; - expect(googleModels[ModelClass.SMALL]).toBe("gemini-1.5-flash-latest"); - expect(googleModels[ModelClass.MEDIUM]).toBe("gemini-1.5-flash-latest"); - expect(googleModels[ModelClass.LARGE]).toBe("gemini-1.5-pro-latest"); + expect(googleModels[ModelClass.SMALL]).toBe("gemini-2.0-flash-exp"); + expect(googleModels[ModelClass.MEDIUM]).toBe("gemini-2.0-flash-exp"); + expect(googleModels[ModelClass.LARGE]).toBe("gemini-2.0-flash-exp"); }); }); });