Finished the Grok LLM implementation

This commit is contained in:
3byss
2026-03-20 00:46:44 -06:00
parent c3c06ff586
commit 17b8d9ed87
5 changed files with 63 additions and 4 deletions

53
lib/llm/grok.mjs Normal file
View File

@@ -0,0 +1,53 @@
// Grok Provider - raw fetch, no SDK
import { LLMProvider } from './provider.mjs';
export class GrokProvider extends LLMProvider {
constructor(config) {
super(config);
this.name = 'grok';
this.apiKey = config.apiKey;
this.model = config.model || 'grok-3';
}
getConfigured() {
return !!this.apiKey;
}
async complete(systemPrompt, userMessage, opts = {}) {
const res = await fetch('https://api.x.ai/v1/chat/completions', {
method: POST,
headers: {
'Content-type': 'application/json',
Authorization: `Bearer ${this.apiKey}`,
},
body: JSON.stringify({
messages: [
{ role: 'system', content: systemPrompt },
{ role: 'user', content: userMessage },
],
model: this.model,
stream: false,
temperature: 0,
}),
signal: AbortSignal.timeout(opts.timeout || 60000),
});
if (!res.ok) {
const err = await res.text().catch(() => '');
throw new Error(`Grok API ${res.status}: ${err.substring(0, 200)}`);
}
const data = await res.json();
const text = data.choices?.[0].message?.content || '';
return {
text,
usage: {
inputTokens: data.usage?.prompt_tokens || 0,
outputTokens: data.usage?.completion_tokens || 0,
},
model: data.model || this.model,
};
}
}

View File

@@ -7,6 +7,7 @@ import { GeminiProvider } from './gemini.mjs';
import { CodexProvider } from './codex.mjs';
import { MiniMaxProvider } from './minimax.mjs';
import { MistralProvider } from './mistral.mjs';
import { GrokProvider } from './grok.mjs';
export { LLMProvider } from './provider.mjs';
export { AnthropicProvider } from './anthropic.mjs';
@@ -16,6 +17,8 @@ export { GeminiProvider } from './gemini.mjs';
export { CodexProvider } from './codex.mjs';
export { MiniMaxProvider } from './minimax.mjs';
export { MistralProvider } from './mistral.mjs';
export { GrokProvider } from './grok.mjs';
/**
* Create an LLM provider based on config.
@@ -42,6 +45,8 @@ export function createLLMProvider(llmConfig) {
return new MiniMaxProvider({ apiKey, model });
case 'mistral':
return new MistralProvider({ apiKey, model });
case 'grok':
return new GrokProvider({ apiKey, model });
default:
console.warn(`[LLM] Unknown provider "${provider}". LLM features disabled.`);
return null;