Merge pull request #1091 from oxSaturn/feat/configure-models-grok-2

feat: allow users to configure models for grok
elizaOS · Dec 14, 2024 · dadaae3 · dadaae3
2 parents 82b475f + 592287d
commit dadaae3
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 4 deletions.
diff --git a/.env.example b/.env.example
@@ -87,6 +87,12 @@ SMALL_REDPILL_MODEL=  # Default: gpt-4o-mini
 MEDIUM_REDPILL_MODEL= # Default: gpt-4o
 LARGE_REDPILL_MODEL=  # Default: gpt-4o
 
+# Grok Configuration
+SMALL_GROK_MODEL=       # Default: grok-2-1212
+MEDIUM_GROK_MODEL=      # Default: grok-2-1212
+LARGE_GROK_MODEL=       # Default: grok-2-1212
+EMBEDDING_GROK_MODEL=   # Default: grok-2-1212
+
 # Ollama Configuration
 OLLAMA_SERVER_URL= # Default: localhost:11434
 OLLAMA_MODEL=

diff --git a/packages/core/src/models.ts b/packages/core/src/models.ts
@@ -87,10 +87,10 @@ export const models: Models = {
         },
         endpoint: "https://api.x.ai/v1",
         model: {
-            [ModelClass.SMALL]: "grok-beta",
-            [ModelClass.MEDIUM]: "grok-beta",
-            [ModelClass.LARGE]: "grok-beta",
-            [ModelClass.EMBEDDING]: "grok-beta", // not sure about this one
+            [ModelClass.SMALL]: settings.SMALL_GROK_MODEL || "grok-2-1212",
+            [ModelClass.MEDIUM]: settings.MEDIUM_GROK_MODEL || "grok-2-1212",
+            [ModelClass.LARGE]: settings.LARGE_GROK_MODEL || "grok-2-1212",
+            [ModelClass.EMBEDDING]: settings.EMBEDDING_GROK_MODEL || "grok-2-1212", // not sure about this one
         },
     },
     [ModelProviderName.GROQ]: {