Merge pull request #444 from miurla/feat/add-xai-provider

feat: add xAI (Grok) provider and models

.env.local.example

@@ -60,6 +60,9 @@ TAVILY_API_KEY=[YOUR_TAVILY_API_KEY]  # Get your API key at: https://app.tavily.
 # Fireworks
 # FIREWORKS_API_KEY=[YOUR_FIREWORKS_API_KEY]
 
+# xAI (Grok)
+# XAI_API_KEY=[YOUR_XAI_API_KEY]
+
 # OpenAI Compatible Model
 # NEXT_PUBLIC_OPENAI_COMPATIBLE_MODEL=
 # OPENAI_COMPATIBLE_API_KEY=

README.md

@@ -189,6 +189,9 @@ This will allow you to use Morphic as your default search engine in the browser.
 - DeepSeek
   - DeepSeek V3
   - DeepSeek R1
+- xAI
+  - grok-2
+  - grok-2-vision
 
 ## ⚡ AI SDK Implementation
 

docs/CONFIGURATION.md

@@ -165,6 +165,12 @@ NEXT_PUBLIC_AZURE_DEPLOYMENT_NAME=[YOUR_DEPLOYMENT_NAME]
 # FIREWORKS_API_KEY=[YOUR_API_KEY]
 ```
 
+### xAI
+
+```bash
+XAI_API_KEY=[YOUR_XAI_API_KEY]
+```
+
 ## Other Features
 
 ### Share Feature

lib/types/models.ts

@@ -95,5 +95,17 @@ export const models: Model[] = [
     name: process.env.NEXT_PUBLIC_OPENAI_COMPATIBLE_MODEL || 'Undefined',
     provider: 'OpenAI Compatible',
     providerId: 'openai-compatible'
+  },
+  {
+    id: 'grok-2-1212',
+    name: 'Grok 2',
+    provider: 'xAI',
+    providerId: 'xai'
+  },
+  {
+    id: 'grok-2-vision-1212',
+    name: 'Grok 2 Vision',
+    provider: 'xAI',
+    providerId: 'xai'
   }
 ]

lib/utils/registry.ts

@@ -5,6 +5,7 @@ import { createFireworks, fireworks } from '@ai-sdk/fireworks'
 import { google } from '@ai-sdk/google'
 import { groq } from '@ai-sdk/groq'
 import { createOpenAI, openai } from '@ai-sdk/openai'
+import { xai } from '@ai-sdk/xai'
 import {
   experimental_createProviderRegistry as createProviderRegistry,
   extractReasoningMiddleware,
@@ -34,7 +35,8 @@ export const registry = createProviderRegistry({
   'openai-compatible': createOpenAI({
     apiKey: process.env.OPENAI_COMPATIBLE_API_KEY,
     baseURL: process.env.OPENAI_COMPATIBLE_API_BASE_URL
-  })
+  }),
+  xai
 })
 
 export function getModel(model: string) {
@@ -102,6 +104,8 @@ export function isProviderEnabled(providerId: string): boolean {
       return !!process.env.DEEPSEEK_API_KEY
     case 'fireworks':
       return !!process.env.FIREWORKS_API_KEY
+    case 'xai':
+      return !!process.env.XAI_API_KEY
     case 'openai-compatible':
       return (
         !!process.env.OPENAI_COMPATIBLE_API_KEY &&

package.json

@@ -17,6 +17,7 @@
     "@ai-sdk/google": "^1.1.5",
     "@ai-sdk/groq": "^1.1.6",
     "@ai-sdk/openai": "^1.1.5",
+    "@ai-sdk/xai": "^1.1.10",
     "@radix-ui/react-alert-dialog": "^1.0.5",
     "@radix-ui/react-avatar": "^1.0.4",
     "@radix-ui/react-checkbox": "^1.0.4",