Merge pull request elizaOS#216 from MeDott29/implement-grok-beta

Implement grok beta
MarcoMandar · Nov 6, 2024 · 39ce28e · 39ce28e
2 parents 84d5b10 + 6853b43
commit 39ce28e
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/core/src/core/generation.ts b/core/src/core/generation.ts
@@ -113,7 +113,8 @@ export async function generateText({
 
             case ModelProvider.GROK: {
                 prettyConsole.log("Initializing Grok model.");
-                const grok = createGroq({ apiKey });
+                const serverUrl = models[provider].endpoint;
+                const grok = createOpenAI({ apiKey, baseURL: serverUrl });
 
                 const { text: grokResponse } = await aiGenerateText({
                     model: grok.languageModel(model, {

diff --git a/core/src/core/models.ts b/core/src/core/models.ts
@@ -74,10 +74,10 @@ const models: Models = {
         },
         endpoint: "https://api.x.ai/v1",
         model: {
-            [ModelClass.SMALL]: "grok-2-beta",
-            [ModelClass.MEDIUM]: "grok-2-beta",
-            [ModelClass.LARGE]: "grok-2-beta",
-            [ModelClass.EMBEDDING]: "grok-2-beta", // not sure about this one
+            [ModelClass.SMALL]: "grok-beta",
+            [ModelClass.MEDIUM]: "grok-beta",
+            [ModelClass.LARGE]: "grok-beta",
+            [ModelClass.EMBEDDING]: "grok-beta", // not sure about this one
         },
     },
     [ModelProvider.GROQ]: {