Skip to content

Commit e115c0c

Browse files
authored
Merge pull request #1815 from gmh5225/develop
2 parents b043684 + f0f596c commit e115c0c

File tree

2 files changed

+6
-6
lines changed

2 files changed

+6
-6
lines changed

packages/core/src/models.ts

+3-3
Original file line numberDiff line numberDiff line change
@@ -189,15 +189,15 @@ export const models: Models = {
189189
[ModelClass.SMALL]:
190190
settings.SMALL_GOOGLE_MODEL ||
191191
settings.GOOGLE_MODEL ||
192-
"gemini-1.5-flash-latest",
192+
"gemini-2.0-flash-exp",
193193
[ModelClass.MEDIUM]:
194194
settings.MEDIUM_GOOGLE_MODEL ||
195195
settings.GOOGLE_MODEL ||
196-
"gemini-1.5-flash-latest",
196+
"gemini-2.0-flash-exp",
197197
[ModelClass.LARGE]:
198198
settings.LARGE_GOOGLE_MODEL ||
199199
settings.GOOGLE_MODEL ||
200-
"gemini-1.5-pro-latest",
200+
"gemini-2.0-flash-exp",
201201
[ModelClass.EMBEDDING]:
202202
settings.EMBEDDING_GOOGLE_MODEL ||
203203
settings.GOOGLE_MODEL ||

packages/core/src/tests/models.test.ts

+3-3
Original file line numberDiff line numberDiff line change
@@ -96,9 +96,9 @@ describe("Model Provider Configuration", () => {
9696
describe("Google Provider", () => {
9797
test("should have correct model mappings", () => {
9898
const googleModels = models[ModelProviderName.GOOGLE].model;
99-
expect(googleModels[ModelClass.SMALL]).toBe("gemini-1.5-flash-latest");
100-
expect(googleModels[ModelClass.MEDIUM]).toBe("gemini-1.5-flash-latest");
101-
expect(googleModels[ModelClass.LARGE]).toBe("gemini-1.5-pro-latest");
99+
expect(googleModels[ModelClass.SMALL]).toBe("gemini-2.0-flash-exp");
100+
expect(googleModels[ModelClass.MEDIUM]).toBe("gemini-2.0-flash-exp");
101+
expect(googleModels[ModelClass.LARGE]).toBe("gemini-2.0-flash-exp");
102102
});
103103
});
104104
});

0 commit comments

Comments
 (0)