Update models.ts

elizaOS · Dec 6, 2024 · a2568b5 · a2568b5
1 parent 7fb73d2
commit a2568b5
Showing 1 changed file with 12 additions and 4 deletions.
diff --git a/packages/core/src/models.ts b/packages/core/src/models.ts
@@ -179,10 +179,18 @@ export const models: Models = {
             temperature: 0.7,
         },
         model: {
-            [ModelClass.SMALL]: "gemini-1.5-flash-latest",
-            [ModelClass.MEDIUM]: "gemini-1.5-flash-latest",
-            [ModelClass.LARGE]: "gemini-1.5-pro-latest",
-            [ModelClass.EMBEDDING]: "text-embedding-004",
+            [ModelClass.SMALL]: settings.SMALL_GOOGLE_MODEL ||
+                settings.GOOGLE_MODEL ||
+                "gemini-1.5-flash-latest",
+            [ModelClass.MEDIUM]: settings.MEDIUM_GOOGLE_MODEL ||
+                settings.GOOGLE_MODEL ||
+                "gemini-1.5-flash-latest",
+            [ModelClass.LARGE]: settings.LARGE_GOOGLE_MODEL ||
+                settings.GOOGLE_MODEL ||
+                "gemini-1.5-pro-latest",
+            [ModelClass.EMBEDDING]: settings.EMBEDDING_GOOGLE_MODEL ||
+                settings.GOOGLE_MODEL ||
+                "text-embedding-004",
         },
     },
     [ModelProviderName.REDPILL]: {