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

Implement grok beta

Author: lalalune

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, {

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]: {