Fixed types and created tests
This commit is contained in:
@@ -10,15 +10,15 @@ export class GrokProvider extends LLMProvider {
|
|||||||
this.model = config.model || 'grok-3';
|
this.model = config.model || 'grok-3';
|
||||||
}
|
}
|
||||||
|
|
||||||
getConfigured() {
|
get isConfigured() {
|
||||||
return !!this.apiKey;
|
return !!this.apiKey;
|
||||||
}
|
}
|
||||||
|
|
||||||
async complete(systemPrompt, userMessage, opts = {}) {
|
async complete(systemPrompt, userMessage, opts = {}) {
|
||||||
const res = await fetch('https://api.x.ai/v1/chat/completions', {
|
const res = await fetch('https://api.x.ai/v1/chat/completions', {
|
||||||
method: POST,
|
method: 'POST',
|
||||||
headers: {
|
headers: {
|
||||||
'Content-type': 'application/json',
|
'Content-Type': 'application/json',
|
||||||
Authorization: `Bearer ${this.apiKey}`,
|
Authorization: `Bearer ${this.apiKey}`,
|
||||||
},
|
},
|
||||||
body: JSON.stringify({
|
body: JSON.stringify({
|
||||||
|
|||||||
79
test/llm-grok.test.mjs
Normal file
79
test/llm-grok.test.mjs
Normal file
@@ -0,0 +1,79 @@
|
|||||||
|
// Grok provider — unit tests
|
||||||
|
// Uses Node.js built-in test runner (node:test) — no extra dependencies
|
||||||
|
|
||||||
|
import { describe, it, mock, beforeEach } from 'node:test';
|
||||||
|
import assert from 'node:assert/strict';
|
||||||
|
import { GrokProvider } from '../lib/llm/grok.mjs';
|
||||||
|
import { createLLMProvider } from '../lib/llm/index.mjs';
|
||||||
|
|
||||||
|
// ─── Unit Tests ───
|
||||||
|
|
||||||
|
describe('GrokProvider', () => {
|
||||||
|
it('should set defaults correctly', () => {
|
||||||
|
const provider = new GrokProvider({ apiKey: 'sk-test' });
|
||||||
|
assert.equal(provider.name, 'grok');
|
||||||
|
assert.equal(provider.model, 'grok-3');
|
||||||
|
assert.equal(provider.isConfigured, true);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should accept custom model', () => {
|
||||||
|
const provider = new GrokProvider({ apiKey: 'sk-test', model: 'grok-2' });
|
||||||
|
assert.equal(provider.model, 'grok-2');
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should report not configured without API key', () => {
|
||||||
|
const provider = new GrokProvider({});
|
||||||
|
assert.equal(provider.isConfigured, false);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should throw on API error', async () => {
|
||||||
|
const provider = new GrokProvider({ apiKey: 'sk-test' });
|
||||||
|
const originalFetch = globalThis.fetch;
|
||||||
|
globalThis.fetch = mock.fn(() =>
|
||||||
|
Promise.resolve({ ok: false, status: 401, text: () => Promise.resolve('Unauthorized') })
|
||||||
|
);
|
||||||
|
try {
|
||||||
|
await assert.rejects(
|
||||||
|
() => provider.complete('system', 'user'),
|
||||||
|
(err) => {
|
||||||
|
assert.match(err.message, /Grok API 401/);
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
);
|
||||||
|
} finally {
|
||||||
|
globalThis.fetch = originalFetch;
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
it('should parse successful response', async () => {
|
||||||
|
const provider = new GrokProvider({ apiKey: 'sk-test' });
|
||||||
|
const mockResponse = {
|
||||||
|
choices: [{ message: { content: 'Hello world' } }],
|
||||||
|
usage: { prompt_tokens: 10, completion_tokens: 5 },
|
||||||
|
model: 'grok-3'
|
||||||
|
};
|
||||||
|
const originalFetch = globalThis.fetch;
|
||||||
|
globalThis.fetch = mock.fn(() =>
|
||||||
|
Promise.resolve({ ok: true, json: () => Promise.resolve(mockResponse) })
|
||||||
|
);
|
||||||
|
try {
|
||||||
|
const result = await provider.complete('system', 'user');
|
||||||
|
assert.equal(result.text, 'Hello world');
|
||||||
|
assert.equal(result.usage.inputTokens, 10);
|
||||||
|
assert.equal(result.usage.outputTokens, 5);
|
||||||
|
assert.equal(result.model, 'grok-3');
|
||||||
|
} finally {
|
||||||
|
globalThis.fetch = originalFetch;
|
||||||
|
}
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
// ─── Factory Tests ───
|
||||||
|
|
||||||
|
describe('createLLMProvider', () => {
|
||||||
|
it('should create Grok provider', () => {
|
||||||
|
const provider = createLLMProvider({ provider: 'grok', apiKey: 'sk-test' });
|
||||||
|
assert.ok(provider instanceof GrokProvider);
|
||||||
|
assert.equal(provider.isConfigured, true);
|
||||||
|
});
|
||||||
|
});
|
||||||
Reference in New Issue
Block a user