lib/llm/__tests__/request-builder.test.ts

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);
  });
});