fix(grok): align token caps and docs

This commit is contained in:
calesthio
2026-03-28 22:36:33 -07:00
parent 18233bcdff
commit 1fb50939ce
3 changed files with 76 additions and 22 deletions

View File

@@ -1,7 +1,7 @@
// 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 { describe, it, mock } from 'node:test';
import assert from 'node:assert/strict';
import { GrokProvider } from '../lib/llm/grok.mjs';
import { createLLMProvider } from '../lib/llm/index.mjs';
@@ -9,12 +9,12 @@ 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 set defaults correctly', () => {
const provider = new GrokProvider({ apiKey: 'sk-test' });
assert.equal(provider.name, 'grok');
assert.equal(provider.model, 'grok-4-latest');
assert.equal(provider.isConfigured, true);
});
it('should accept custom model', () => {
const provider = new GrokProvider({ apiKey: 'sk-test', model: 'grok-2' });
@@ -45,11 +45,11 @@ describe('GrokProvider', () => {
}
});
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 },
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;
@@ -63,9 +63,63 @@ describe('GrokProvider', () => {
assert.equal(result.usage.outputTokens, 5);
assert.equal(result.model, 'grok-3');
} finally {
globalThis.fetch = originalFetch;
}
});
globalThis.fetch = originalFetch;
}
});
it('should send correct request format', async () => {
const provider = new GrokProvider({ apiKey: 'sk-test-key', model: 'grok-4-latest' });
let capturedUrl, capturedOpts;
const originalFetch = globalThis.fetch;
globalThis.fetch = mock.fn((url, opts) => {
capturedUrl = url;
capturedOpts = opts;
return Promise.resolve({
ok: true,
json: () => Promise.resolve({
choices: [{ message: { content: 'ok' } }],
usage: { prompt_tokens: 1, completion_tokens: 1 },
model: 'grok-4-latest',
}),
});
});
try {
await provider.complete('system prompt', 'user message', { maxTokens: 2048 });
assert.equal(capturedUrl, 'https://api.x.ai/v1/chat/completions');
assert.equal(capturedOpts.method, 'POST');
const headers = capturedOpts.headers;
assert.equal(headers['Content-Type'], 'application/json');
assert.equal(headers.Authorization, 'Bearer sk-test-key');
const body = JSON.parse(capturedOpts.body);
assert.equal(body.model, 'grok-4-latest');
assert.equal(body.max_tokens, 2048);
assert.equal(body.messages[0].role, 'system');
assert.equal(body.messages[0].content, 'system prompt');
assert.equal(body.messages[1].role, 'user');
assert.equal(body.messages[1].content, 'user message');
} finally {
globalThis.fetch = originalFetch;
}
});
it('should handle empty response gracefully', async () => {
const provider = new GrokProvider({ apiKey: 'sk-test' });
const originalFetch = globalThis.fetch;
globalThis.fetch = mock.fn(() =>
Promise.resolve({
ok: true,
json: () => Promise.resolve({ choices: [], usage: {} }),
})
);
try {
const result = await provider.complete('sys', 'user');
assert.equal(result.text, '');
assert.equal(result.usage.inputTokens, 0);
assert.equal(result.usage.outputTokens, 0);
} finally {
globalThis.fetch = originalFetch;
}
});
});
// ─── Factory Tests ───
@@ -76,4 +130,4 @@ describe('createLLMProvider', () => {
assert.ok(provider instanceof GrokProvider);
assert.equal(provider.isConfigured, true);
});
});
});