83 lines
3.4 KiB
TypeScript
83 lines
3.4 KiB
TypeScript
import { describe, expect, it } from 'vitest'
|
|
import {
|
|
AI_PROVIDER_FAMILY_META,
|
|
AI_PROVIDER_FAMILY_ORDER,
|
|
AI_PROVIDER_PRESETS,
|
|
createProviderFromPreset,
|
|
presetsByFamily,
|
|
resolveProviderFamily,
|
|
} from '../../src/ai/providerPresets'
|
|
|
|
describe('AI provider presets grouping', () => {
|
|
it('matches the expected UI group order', () => {
|
|
expect(AI_PROVIDER_FAMILY_ORDER).toEqual(['openai_compatible', 'standalone', 'plan', 'local', 'custom'])
|
|
expect(AI_PROVIDER_FAMILY_META.openai_compatible.label).toBe('OpenAI compatible')
|
|
expect(AI_PROVIDER_FAMILY_META.plan.label).toBe('Subscription plans')
|
|
expect(AI_PROVIDER_FAMILY_META.local.label).toBe('Self-hosted')
|
|
expect(AI_PROVIDER_FAMILY_META.custom.label).toBe('Custom')
|
|
})
|
|
|
|
it('lists OpenAI-compatible payg presets', () => {
|
|
expect(presetsByFamily('openai_compatible').map((item) => item.name)).toEqual([
|
|
'DeepSeek',
|
|
'OpenAI',
|
|
'Qwen',
|
|
'Kimi',
|
|
'Xiaomi MiMo',
|
|
'MiniMax',
|
|
])
|
|
})
|
|
|
|
it('lists Anthropic and Google Gemini after the separator group', () => {
|
|
expect(presetsByFamily('standalone').map((item) => item.name)).toEqual(['Anthropic', 'Google Gemini'])
|
|
})
|
|
|
|
it('lists Plan presets as GLM / MiniMax / Kimi / MiMo', () => {
|
|
expect(presetsByFamily('plan').map((item) => item.name)).toEqual(['GLM', 'MiniMax', 'Kimi', 'Xiaomi MiMo'])
|
|
expect(createProviderFromPreset('glm').defaultModelId).toBe('glm-5.3-plan')
|
|
expect(createProviderFromPreset('minimax-plan').defaultModelId).toBe('MiniMax-M3-plan')
|
|
expect(createProviderFromPreset('kimi-plan').defaultModelId).toBe('kimi-k3-plan')
|
|
expect(createProviderFromPreset('mimo-plan').defaultModelId).toBe('mimo-v2.5-pro-plan')
|
|
expect(createProviderFromPreset('mimo-plan').baseUrl).toBe('https://token-plan-cn.xiaomimimo.com/v1')
|
|
expect(createProviderFromPreset('zai').presetId).toBe('glm')
|
|
})
|
|
|
|
it('defaults DeepSeek to V4 Pro / Flash', () => {
|
|
const deepseek = createProviderFromPreset('deepseek')
|
|
expect(deepseek.defaultModelId).toBe('deepseek-v4-pro')
|
|
expect(deepseek.models.map((item) => item.model)).toEqual(['deepseek-v4-pro', 'deepseek-flash'])
|
|
})
|
|
|
|
it('uses current frontier model IDs from official catalogs', () => {
|
|
expect(createProviderFromPreset('openai').models.map((item) => item.model)).toEqual([
|
|
'gpt-6-astra',
|
|
'gpt-5.6-sol',
|
|
'gpt-5.6-terra',
|
|
'gpt-5.6-luna',
|
|
'gpt-5.5',
|
|
])
|
|
expect(createProviderFromPreset('anthropic').defaultModelId).toBe('claude-fable-5-1')
|
|
expect(createProviderFromPreset('google').defaultModelId).toBe('gemini-3.1-pro-preview')
|
|
expect(createProviderFromPreset('qwen').defaultModelId).toBe('qwen3.8-max')
|
|
expect(createProviderFromPreset('kimi').defaultModelId).toBe('kimi-k3')
|
|
})
|
|
|
|
it('lists local and custom at the bottom', () => {
|
|
expect(presetsByFamily('local').map((item) => item.name)).toEqual(['Ollama', 'VLLM'])
|
|
expect(presetsByFamily('custom').map((item) => item.name)).toEqual(['Custom'])
|
|
expect(createProviderFromPreset('custom').family).toBe('custom')
|
|
})
|
|
|
|
it('resolves legacy zhipu/zai to plan family', () => {
|
|
expect(
|
|
resolveProviderFamily({
|
|
presetId: 'zai',
|
|
protocol: 'openai-chat',
|
|
authStyle: 'bearer',
|
|
baseUrl: 'https://api.z.ai/api/paas/v4',
|
|
})
|
|
).toBe('plan')
|
|
expect(AI_PROVIDER_PRESETS.some((item) => item.id === 'mimo')).toBe(true)
|
|
})
|
|
})
|