teqo/tests/unit/agentPoolModels.unit.spec.ts

// @vitest-environment node

import { describe, expect, it } from 'vitest'

import { POOL_DEFAULT_MODEL_SLUG, resolvePoolModel } from '../../scripts/lib/agent-pool-models.mjs'

const API_MODELS = [
  {
    id: 'composer-2.5',
    aliases: ['composer-latest'],
    parameters: [{ id: 'fast', values: [{ value: 'false' }, { value: 'true' }] }],
  },
  {
    id: 'grok-4.5',
    parameters: [
      {
        id: 'effort',
        values: [{ value: 'low' }, { value: 'medium' }, { value: 'high' }],
      },
      { id: 'fast', values: [{ value: 'false' }, { value: 'true' }] },
    ],
  },
  {
    id: 'kimi-k3',
    parameters: [
      {
        id: 'reasoning',
        values: [{ value: 'low' }, { value: 'high' }, { value: 'max' }],
      },
    ],
  },
]

describe('resolvePoolModel', () => {
  it('resolves composer by exact id and alias', () => {
    expect(resolvePoolModel('composer-2.5', API_MODELS)).toEqual({
      model: { id: 'composer-2.5' },
      requested: 'composer-2.5',
      usedFallback: false,
    })
    expect(resolvePoolModel('composer-latest', API_MODELS).model).toEqual({ id: 'composer-2.5' })
  })

  it('maps the repo -fast suffix to the fast param when the table supports it', () => {
    const result = resolvePoolModel('composer-2.5-fast', API_MODELS)
    expect(result.model).toEqual({
      id: 'composer-2.5',
      params: [{ id: 'fast', value: 'true' }],
    })
    expect(result.usedFallback).toBe(false)
  })

  it.each([
    ['cursor-grok-4.5-low', 'low'],
    ['cursor-grok-4.5-medium', 'medium'],
    ['cursor-grok-4.5-high', 'high'],
  ] as const)('maps %s → grok-4.5 effort=%s', (slug, effort) => {
    expect(resolvePoolModel(slug, API_MODELS)).toEqual({
      model: { id: 'grok-4.5', params: [{ id: 'effort', value: effort }] },
      requested: slug,
      usedFallback: false,
    })
  })

  it('maps cursor-grok-4.5-high-fast → effort high + fast', () => {
    expect(resolvePoolModel('cursor-grok-4.5-high-fast', API_MODELS)).toEqual({
      model: {
        id: 'grok-4.5',
        params: [
          { id: 'effort', value: 'high' },
          { id: 'fast', value: 'true' },
        ],
      },
      requested: 'cursor-grok-4.5-high-fast',
      usedFallback: false,
    })
  })

  it('maps kimi-k3-low → kimi-k3 reasoning=low', () => {
    expect(resolvePoolModel('kimi-k3-low', API_MODELS)).toEqual({
      model: { id: 'kimi-k3', params: [{ id: 'reasoning', value: 'low' }] },
      requested: 'kimi-k3-low',
      usedFallback: false,
    })
  })

  it('falls back when a -fast slug base lacks the fast param', () => {
    const result = resolvePoolModel('kimi-k3-fast', API_MODELS)
    expect(result.model).toEqual({ id: POOL_DEFAULT_MODEL_SLUG })
    expect(result.usedFallback).toBe(true)
  })

  it('falls back to composer-2.5 with a warn for unknown slugs', () => {
    const result = resolvePoolModel('modelo-inventado-9', API_MODELS)
    expect(result.model).toEqual({ id: POOL_DEFAULT_MODEL_SLUG })
    expect(result.requested).toBe('modelo-inventado-9')
    expect(result.usedFallback).toBe(true)
    expect(result.warn).toContain('fallback')
  })

  it('uses the default silently when the issue declares no model', () => {
    for (const absent of [undefined, null, '', '   ']) {
      const result = resolvePoolModel(absent, API_MODELS)
      expect(result.model).toEqual({ id: POOL_DEFAULT_MODEL_SLUG })
      expect(result.usedFallback).toBe(false)
      expect(result.warn).toBeUndefined()
    }
  })

  it('still emits canonical Cloud shapes for known slugs when the table is empty', () => {
    expect(resolvePoolModel('cursor-grok-4.5-medium', [])).toEqual({
      model: { id: 'grok-4.5', params: [{ id: 'effort', value: 'medium' }] },
      requested: 'cursor-grok-4.5-medium',
      usedFallback: false,
    })
    expect(resolvePoolModel('kimi-k3-low', []).model).toEqual({
      id: 'kimi-k3',
      params: [{ id: 'reasoning', value: 'low' }],
    })
  })

  it('falls back for unknown slugs when the live table is unavailable', () => {
    const result = resolvePoolModel('qualquer-coisa', [])
    expect(result.model).toEqual({ id: POOL_DEFAULT_MODEL_SLUG })
    expect(result.warn).toContain('/v1/models')
  })

  it('ignores non-string frontmatter values', () => {
    const result = resolvePoolModel(42, API_MODELS)
    expect(result.model).toEqual({ id: POOL_DEFAULT_MODEL_SLUG })
    expect(result.usedFallback).toBe(false)
  })
})