From e4d081e2d8eba1e33369391dfb92025a460eeaec Mon Sep 17 00:00:00 2001 From: MeDott29 Date: Wed, 6 Nov 2024 03:14:17 -0500 Subject: [PATCH 1/2] update to grok-beta --- core/src/core/models.ts | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/core/src/core/models.ts b/core/src/core/models.ts index 41a91f1eab2..3c8b275d4ed 100644 --- 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]: { From a233529c61e6a838d5e4143c0308630fe2315e6e Mon Sep 17 00:00:00 2001 From: MeDott29 Date: Wed, 6 Nov 2024 03:17:08 -0500 Subject: [PATCH 2/2] update to grok-beta --- core/src/core/generation.ts | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/core/src/core/generation.ts b/core/src/core/generation.ts index bfd1fdd9400..c9eda52e7a7 100644 --- 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, {