File size: 2,766 Bytes
391c43e | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 | import { describe, it, expect } from 'vitest';
import { getApiEndpoint, buildHeaders, resolveTemperature } from '@/lib/llm/request-builder';
import { getProvider } from '@/lib/llm/providers/registry';
describe('getApiEndpoint', () => {
it('opencode-go + minimax model + anthropic wire → /messages on opencode-go base URL', () => {
const result = getApiEndpoint('opencode-go', getProvider('opencode-go'), 'minimax-m2.7', {}, undefined, 'anthropic');
expect(result).toBe('https://opencode.ai/zen/go/v1/messages');
});
it('anthropic provider + anthropic wire → api.anthropic.com/v1/messages', () => {
const result = getApiEndpoint('anthropic', getProvider('anthropic'), 'claude-x', {}, undefined, 'anthropic');
expect(result).toBe('https://api.anthropic.com/v1/messages');
});
it('opencode-go + glm model + openai wire → /chat/completions on opencode-go base URL', () => {
const result = getApiEndpoint('opencode-go', getProvider('opencode-go'), 'glm-5.2', {}, undefined, 'openai');
expect(result).toBe('https://opencode.ai/zen/go/v1/chat/completions');
});
});
describe('buildHeaders', () => {
it('opencode-go + anthropic wire → x-api-key and anthropic-version, no anthropic-beta', () => {
const headers = buildHeaders('opencode-go', 'sk-x', null, getProvider('opencode-go'), 'anthropic');
expect(headers['x-api-key']).toBe('sk-x');
expect(headers['anthropic-version']).toBe('2023-06-01');
expect(headers['anthropic-beta']).toBeUndefined();
});
it('anthropic provider + anthropic wire → x-api-key AND anthropic-beta (supportsFunctions=true)', () => {
const config = getProvider('anthropic');
expect(config.supportsFunctions).toBe(true); // guard assertion
const headers = buildHeaders('anthropic', 'sk-x', null, config, 'anthropic');
expect(headers['x-api-key']).toBe('sk-x');
expect(headers['anthropic-beta']).toBe('tools-2024-04-04');
});
it('opencode-go + openai wire → Authorization Bearer, no x-api-key', () => {
const headers = buildHeaders('opencode-go', 'sk-x', null, getProvider('opencode-go'), 'openai');
expect(headers['Authorization']).toBe('Bearer sk-x');
expect(headers['x-api-key']).toBeUndefined();
});
});
describe('resolveTemperature', () => {
it('opencode-go + kimi- model → 1', () => {
expect(resolveTemperature('opencode-go', 'kimi-k2.7-code')).toBe(1);
});
it('opencode-go + non-kimi model → 0.7', () => {
expect(resolveTemperature('opencode-go', 'glm-5.2')).toBe(0.7);
});
it('openai + gpt-5-nano → 1', () => {
expect(resolveTemperature('openai', 'gpt-5-nano')).toBe(1);
});
it('anthropic + claude-x → 0.7', () => {
expect(resolveTemperature('anthropic', 'claude-x')).toBe(0.7);
});
});
|