mirror of
https://github.com/wu736139669/hapi.git
synced 2026-08-09 07:29:51 +00:00
SSU-WEI HUANGandGitHub
b9eed7c071
* test: define Grok Build integration behavior * feat: add Grok Build agent integration * test: cover Grok permissions and resume paths * docs: add Grok Build setup guide * fix: scope Grok ACP discovery to session cwd * fix: align Grok permission UI semantics * docs: clarify Grok runner setup * test: require Grok create model and effort options * feat: add Grok create model and effort pickers * test: define Grok runtime parity behavior * feat: add Grok runtime ACP controls and discovery * fix: tighten Grok runtime controls * fix: suppress nonfatal Grok title quota errors * feat: support Grok Auto permission mode * feat: forward ACP native session titles for Grok * fix: guard Grok Windows shell arguments
306 lines
13 KiB
TypeScript
306 lines
13 KiB
TypeScript
import { describe, expect, it } from 'vitest'
|
|
import { getModelOptionsForFlavor, getNextModelForFlavor } from './modelOptions'
|
|
|
|
describe('getModelOptionsForFlavor', () => {
|
|
it('returns Gemini model options for gemini flavor', () => {
|
|
const options = getModelOptionsForFlavor('gemini')
|
|
expect(options[0]).toEqual({ value: null, label: 'Default' })
|
|
expect(options.some((o) => o.value === 'gemini-3-flash-preview')).toBe(true)
|
|
expect(options.some((o) => o.value === 'gemini-2.5-flash')).toBe(true)
|
|
})
|
|
|
|
it('returns Claude model options for claude flavor', () => {
|
|
const options = getModelOptionsForFlavor('claude')
|
|
expect(options[0]).toEqual({ value: null, label: 'Default' })
|
|
expect(options.some((o) => o.value === 'sonnet')).toBe(true)
|
|
expect(options.some((o) => o.value === 'opus')).toBe(true)
|
|
})
|
|
|
|
it('keeps Claude presets when explicit options only include Sonnet models', () => {
|
|
const options = getModelOptionsForFlavor('claude', null, [
|
|
{ value: null, label: 'Default' },
|
|
{ value: 'sonnet', label: 'Sonnet' },
|
|
{ value: 'sonnet[1m]', label: 'Sonnet 1M' }
|
|
])
|
|
expect(options).toEqual([
|
|
{ value: null, label: 'Default' },
|
|
{ value: 'sonnet', label: 'Sonnet' },
|
|
{ value: 'sonnet[1m]', label: 'Sonnet 1M' },
|
|
{ value: 'opus', label: 'Opus' },
|
|
{ value: 'opus[1m]', label: 'Opus 1M' },
|
|
{ value: 'fable', label: 'Fable' },
|
|
{ value: 'fable[1m]', label: 'Fable 1M' }
|
|
])
|
|
})
|
|
|
|
it('adds non-preset Claude options without hiding Opus presets', () => {
|
|
const options = getModelOptionsForFlavor('claude', null, [
|
|
{ value: 'claude-opus-4-1-20250805', label: 'Claude Opus 4.1' }
|
|
])
|
|
expect(options).toEqual([
|
|
{ value: null, label: 'Default' },
|
|
{ value: 'claude-opus-4-1-20250805', label: 'Claude Opus 4.1' },
|
|
{ value: 'sonnet', label: 'Sonnet' },
|
|
{ value: 'sonnet[1m]', label: 'Sonnet 1M' },
|
|
{ value: 'opus', label: 'Opus' },
|
|
{ value: 'opus[1m]', label: 'Opus 1M' },
|
|
{ value: 'fable', label: 'Fable' },
|
|
{ value: 'fable[1m]', label: 'Fable 1M' }
|
|
])
|
|
})
|
|
|
|
it('includes custom Gemini model from env/config in options', () => {
|
|
const options = getModelOptionsForFlavor('gemini', 'gemini-custom-experiment')
|
|
expect(options.some((o) => o.value === 'gemini-custom-experiment')).toBe(true)
|
|
})
|
|
|
|
it('does not duplicate a preset Gemini model', () => {
|
|
const options = getModelOptionsForFlavor('gemini', 'gemini-2.5-flash')
|
|
const flashCount = options.filter((o) => o.value === 'gemini-2.5-flash').length
|
|
expect(flashCount).toBe(1)
|
|
})
|
|
|
|
it('includes the current custom model when it is missing from explicit options', () => {
|
|
const options = getModelOptionsForFlavor('codex', 'gpt-legacy', [
|
|
{ value: 'gpt-5.5', label: 'GPT-5.5' }
|
|
])
|
|
expect(options).toEqual([
|
|
{ value: 'gpt-legacy', label: 'gpt-legacy' },
|
|
{ value: 'gpt-5.5', label: 'GPT-5.5' }
|
|
])
|
|
})
|
|
|
|
it('returns only the supplied custom options for opencode flavor (no claude fallback)', () => {
|
|
const options = getModelOptionsForFlavor('opencode', null, [
|
|
{ value: 'ollama/exaone:4.5-33b-q8', label: 'Ollama (SER8)/EXAONE 4.5 33B Q8' },
|
|
{ value: 'mlx/qwen3:0.6b', label: 'MLX/Qwen3 0.6B' }
|
|
])
|
|
expect(options).toEqual([
|
|
{ value: 'ollama/exaone:4.5-33b-q8', label: 'Ollama (SER8)/EXAONE 4.5 33B Q8' },
|
|
{ value: 'mlx/qwen3:0.6b', label: 'MLX/Qwen3 0.6B' }
|
|
])
|
|
})
|
|
|
|
it('returns an empty list for opencode flavor before models are discovered (no claude fallback)', () => {
|
|
const options = getModelOptionsForFlavor('opencode', null)
|
|
expect(options).toEqual([])
|
|
})
|
|
|
|
it('returns only default/current for cursor before models are discovered (no claude fallback)', () => {
|
|
const options = getModelOptionsForFlavor('cursor', 'composer-2.5')
|
|
expect(options).toEqual([
|
|
{ value: null, label: 'Default' },
|
|
{ value: 'composer-2.5', label: 'composer-2.5' }
|
|
])
|
|
})
|
|
|
|
it('returns dynamic cursor options when supplied', () => {
|
|
const options = getModelOptionsForFlavor('cursor', null, [
|
|
{ value: 'composer-2.5', label: 'Composer 2.5' },
|
|
{ value: 'gpt-5.5-high-fast', label: 'GPT-5.5 High Fast' }
|
|
])
|
|
expect(options).toEqual([
|
|
{ value: 'composer-2.5', label: 'Composer 2.5' },
|
|
{ value: 'gpt-5.5-high-fast', label: 'GPT-5.5 High Fast' }
|
|
])
|
|
})
|
|
|
|
it('does not inject raw wire id when dual picker base is already listed', () => {
|
|
const wire = 'claude-opus-4-8[thinking=true,context=300k,effort=high,fast=false]'
|
|
const options = getModelOptionsForFlavor('cursor', wire, [
|
|
{ value: null, label: 'Default' },
|
|
{ value: 'claude-opus-4-8', label: 'Claude Opus 4.8' },
|
|
{ value: 'composer-2.5', label: 'Composer 2.5' },
|
|
])
|
|
expect(options).toEqual([
|
|
{ value: null, label: 'Default' },
|
|
{ value: 'claude-opus-4-8', label: 'Claude Opus 4.8' },
|
|
{ value: 'composer-2.5', label: 'Composer 2.5' },
|
|
])
|
|
})
|
|
|
|
it('injects unknown wire id only when catalog lacks base and wire', () => {
|
|
const wire = 'claude-opus-4-9[effort=high,fast=false]'
|
|
const options = getModelOptionsForFlavor('cursor', wire, [
|
|
{ value: null, label: 'Default' },
|
|
{ value: 'composer-2.5', label: 'Composer 2.5' },
|
|
])
|
|
expect(options).toEqual([
|
|
{ value: null, label: 'Default' },
|
|
{ value: wire, label: wire },
|
|
{ value: 'composer-2.5', label: 'Composer 2.5' },
|
|
])
|
|
})
|
|
|
|
it('includes the current opencode model when it is missing from explicit options', () => {
|
|
const options = getModelOptionsForFlavor('opencode', 'ollama/legacy', [
|
|
{ value: 'ollama/exaone:4.5-33b-q8', label: 'Ollama EXAONE' }
|
|
])
|
|
expect(options).toEqual([
|
|
{ value: 'ollama/legacy', label: 'ollama/legacy' },
|
|
{ value: 'ollama/exaone:4.5-33b-q8', label: 'Ollama EXAONE' }
|
|
])
|
|
})
|
|
|
|
it('returns just the auto/default option for pi flavor (no Claude fallback)', () => {
|
|
const options = getModelOptionsForFlavor('pi')
|
|
expect(options).toEqual([{ value: null, label: 'Default' }])
|
|
})
|
|
|
|
it('keeps the current pi model in the options list when it is not auto', () => {
|
|
const options = getModelOptionsForFlavor('pi', 'claude-sonnet-4-5')
|
|
expect(options).toEqual([
|
|
{ value: null, label: 'Default' },
|
|
{ value: 'claude-sonnet-4-5', label: 'claude-sonnet-4-5' }
|
|
])
|
|
})
|
|
|
|
it('returns only default/current for grok without falling back to Claude models', () => {
|
|
expect(getModelOptionsForFlavor('grok')).toEqual([
|
|
{ value: null, label: 'Default' }
|
|
])
|
|
expect(getModelOptionsForFlavor('grok', 'grok-4.5')).toEqual([
|
|
{ value: null, label: 'Default' },
|
|
{ value: 'grok-4.5', label: 'grok-4.5' }
|
|
])
|
|
})
|
|
})
|
|
|
|
describe('getNextModelForFlavor', () => {
|
|
it('cycles Gemini models', () => {
|
|
const next = getNextModelForFlavor('gemini', null)
|
|
expect(next).not.toBeNull()
|
|
})
|
|
|
|
it('cycles Claude models', () => {
|
|
const next = getNextModelForFlavor('claude', null)
|
|
expect(next).not.toBeNull()
|
|
})
|
|
|
|
it('cycles through Claude presets when explicit options only include Sonnet models', () => {
|
|
const next = getNextModelForFlavor('claude', 'sonnet[1m]', [
|
|
{ value: 'sonnet', label: 'Sonnet' },
|
|
{ value: 'sonnet[1m]', label: 'Sonnet 1M' }
|
|
])
|
|
expect(next).toBe('opus')
|
|
})
|
|
|
|
it('cycles explicit model options', () => {
|
|
const next = getNextModelForFlavor('codex', 'gpt-5.5', [
|
|
{ value: 'gpt-5.5', label: 'GPT-5.5' },
|
|
{ value: 'gpt-5.4', label: 'GPT-5.4' }
|
|
])
|
|
expect(next).toBe('gpt-5.4')
|
|
})
|
|
|
|
it('does not choose auto when cycling explicit Codex model options from an unknown current model', () => {
|
|
const next = getNextModelForFlavor('codex', 'gpt-legacy', [
|
|
{ value: 'gpt-5.5', label: 'GPT-5.5' },
|
|
{ value: 'gpt-5.4', label: 'GPT-5.4' }
|
|
])
|
|
expect(next).toBe('gpt-5.5')
|
|
})
|
|
|
|
it('keeps the current opencode model when the dynamic list has not loaded (undefined customOptions)', () => {
|
|
const next = getNextModelForFlavor('opencode', 'ollama/exaone:4.5-33b-q8')
|
|
expect(next).toBe('ollama/exaone:4.5-33b-q8')
|
|
})
|
|
|
|
it('keeps the current opencode model when the dynamic list is empty', () => {
|
|
const next = getNextModelForFlavor('opencode', 'ollama/exaone:4.5-33b-q8', [])
|
|
expect(next).toBe('ollama/exaone:4.5-33b-q8')
|
|
})
|
|
|
|
it('returns null for opencode without a current model and without dynamic options (no Claude fallback)', () => {
|
|
const next = getNextModelForFlavor('opencode', null, [])
|
|
expect(next).toBeNull()
|
|
})
|
|
|
|
it('keeps the current cursor model when the dynamic list has not loaded', () => {
|
|
const next = getNextModelForFlavor('cursor', 'composer-2.5')
|
|
expect(next).toBe('composer-2.5')
|
|
})
|
|
|
|
it('keeps the current pi model on cycle (no Claude fallback)', () => {
|
|
// Pi has no predefined model list — Ctrl/Cmd+M must not cycle
|
|
// through Claude presets, which would push sonnet/opus ids into
|
|
// a Pi session via set-session-config.
|
|
const next = getNextModelForFlavor('pi', 'claude-sonnet-4-5')
|
|
expect(next).toBe('claude-sonnet-4-5')
|
|
})
|
|
|
|
it('keeps the current grok model on cycle (no Claude fallback)', () => {
|
|
expect(getNextModelForFlavor('grok', 'grok-4.5')).toBe('grok-4.5')
|
|
})
|
|
|
|
it('returns null for pi without a current model (no Claude fallback)', () => {
|
|
const next = getNextModelForFlavor('pi', null)
|
|
expect(next).toBeNull()
|
|
})
|
|
|
|
it('treats "auto" as null and returns null for pi (no Claude preset injection)', () => {
|
|
// normalizeCurrentModel maps 'auto' to null; a Pi session whose UI
|
|
// displays 'Auto' must not be switched to sonnet/opus by the
|
|
// cycler shortcut.
|
|
const next = getNextModelForFlavor('pi', 'auto')
|
|
expect(next).toBeNull()
|
|
})
|
|
|
|
it('treats "default" as null and returns null for pi', () => {
|
|
const next = getNextModelForFlavor('pi', 'default')
|
|
expect(next).toBeNull()
|
|
})
|
|
|
|
it('treats empty/whitespace strings as null for pi (no Claude preset injection)', () => {
|
|
expect(getNextModelForFlavor('pi', '')).toBeNull()
|
|
expect(getNextModelForFlavor('pi', ' ')).toBeNull()
|
|
})
|
|
|
|
it('trims surrounding whitespace from the current pi model', () => {
|
|
const next = getNextModelForFlavor('pi', ' claude-sonnet-4-5 ')
|
|
expect(next).toBe('claude-sonnet-4-5')
|
|
})
|
|
|
|
it('keeps a kimi current model on cycle (no Claude fallback)', () => {
|
|
expect(getNextModelForFlavor('kimi', 'kimi-k2-0711')).toBe('kimi-k2-0711')
|
|
expect(getNextModelForFlavor('kimi', null)).toBeNull()
|
|
})
|
|
|
|
it('keeps a cursor current model on cycle (no Claude fallback)', () => {
|
|
expect(getNextModelForFlavor('cursor', 'composer-2.5')).toBe('composer-2.5')
|
|
expect(getNextModelForFlavor('cursor', null)).toBeNull()
|
|
})
|
|
|
|
it('keeps an opencode current model on cycle (no Claude fallback)', () => {
|
|
expect(getNextModelForFlavor('opencode', 'ollama/legacy')).toBe('ollama/legacy')
|
|
expect(getNextModelForFlavor('opencode', null)).toBeNull()
|
|
})
|
|
})
|
|
|
|
describe('getModelOptionsForFlavor — pi normalize filter', () => {
|
|
it('drops "auto" and renders just the default option for pi', () => {
|
|
// 'auto' should be normalized to null, which equals the auto entry;
|
|
// we must not produce a duplicate { value: null, label: 'auto' } row.
|
|
const options = getModelOptionsForFlavor('pi', 'auto')
|
|
expect(options).toEqual([{ value: null, label: 'Default' }])
|
|
})
|
|
|
|
it('drops "default" and renders just the default option for pi', () => {
|
|
const options = getModelOptionsForFlavor('pi', 'default')
|
|
expect(options).toEqual([{ value: null, label: 'Default' }])
|
|
})
|
|
|
|
it('drops empty/whitespace currentModel for pi', () => {
|
|
expect(getModelOptionsForFlavor('pi', '')).toEqual([{ value: null, label: 'Default' }])
|
|
expect(getModelOptionsForFlavor('pi', ' ')).toEqual([{ value: null, label: 'Default' }])
|
|
})
|
|
|
|
it('trims whitespace from a real current pi model', () => {
|
|
const options = getModelOptionsForFlavor('pi', ' custom-model ')
|
|
expect(options).toEqual([
|
|
{ value: null, label: 'Default' },
|
|
{ value: 'custom-model', label: 'custom-model' }
|
|
])
|
|
})
|
|
})
|