Merge pull request #924 from boxhock/fix-heurist-model-selection

fix: use of Heurist model env vars
elizaOS · Dec 9, 2024 · 621f3f4 · 621f3f4
2 parents 31143ae + a041953
commit 621f3f4
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/packages/core/src/models.ts b/packages/core/src/models.ts
@@ -289,11 +289,11 @@ export const models: Models = {
         },
         endpoint: "https://llm-gateway.heurist.xyz",
         model: {
-            [ModelClass.SMALL]: "meta-llama/llama-3-70b-instruct",
-            [ModelClass.MEDIUM]: "meta-llama/llama-3-70b-instruct",
-            [ModelClass.LARGE]: "meta-llama/llama-3.1-405b-instruct",
+            [ModelClass.SMALL]: settings.SMALL_HEURIST_MODEL || "meta-llama/llama-3-70b-instruct",
+            [ModelClass.MEDIUM]: settings.MEDIUM_HEURIST_MODEL || "meta-llama/llama-3-70b-instruct",
+            [ModelClass.LARGE]: settings.LARGE_HEURIST_MODEL || "meta-llama/llama-3.1-405b-instruct",
             [ModelClass.EMBEDDING]: "", //Add later,
-            [ModelClass.IMAGE]: "PepeXL",
+            [ModelClass.IMAGE]: settings.HEURIST_IMAGE_MODEL || "PepeXL",
         },
     },
     [ModelProviderName.GALADRIEL]: {