Merge pull request #1815 from gmh5225/develop

elizaOS · Jan 4, 2025 · e115c0c · e115c0c
2 parents b043684 + f0f596c
commit e115c0c
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git 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
@@ -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");
         });
     });
 });