mirror of
https://github.com/EKKOLearnAI/hermes-web-ui.git
synced 2026-05-28 23:10:14 +00:00
ekko
73b919e07c
Prevents infinite retry loop on /api/hermes/available-models when accessing the app without a token, which triggers server-side 429 rate limiting. Both loadModels() and fetchProviders() now bail out early if hasApiKey() returns false. Closes #606 Co-authored-by: Claude Opus 4.7 <noreply@anthropic.com>
77 lines
2.5 KiB
TypeScript
77 lines
2.5 KiB
TypeScript
// @vitest-environment jsdom
|
|
import { beforeEach, describe, expect, it, vi } from 'vitest'
|
|
import { createPinia, setActivePinia } from 'pinia'
|
|
|
|
const mockSystemApi = vi.hoisted(() => ({
|
|
fetchAvailableModels: vi.fn(),
|
|
updateDefaultModel: vi.fn(),
|
|
addCustomProvider: vi.fn(),
|
|
removeCustomProvider: vi.fn(),
|
|
}))
|
|
|
|
vi.mock('@/api/hermes/system', () => mockSystemApi)
|
|
vi.mock('@/api/client', () => ({ hasApiKey: () => true }))
|
|
|
|
import { useAppStore } from '@/stores/hermes/app'
|
|
import { useModelsStore } from '@/stores/hermes/models'
|
|
|
|
describe('Models Store', () => {
|
|
beforeEach(() => {
|
|
setActivePinia(createPinia())
|
|
vi.clearAllMocks()
|
|
window.localStorage.clear()
|
|
})
|
|
|
|
it('keeps the sidebar model picker in sync after provider model visibility changes', async () => {
|
|
const visibleGroups = [
|
|
{
|
|
provider: 'deepseek',
|
|
label: 'DeepSeek',
|
|
base_url: 'https://api.deepseek.com/v1',
|
|
api_key: 'sk-test',
|
|
models: ['deepseek-v4-flash', 'deepseek-v4-pro'],
|
|
available_models: ['deepseek-v4-flash', 'deepseek-v4-pro'],
|
|
model_meta: {
|
|
'deepseek-v4-pro': { preview: true },
|
|
},
|
|
},
|
|
]
|
|
mockSystemApi.fetchAvailableModels.mockResolvedValue({
|
|
default: 'deepseek-v4-flash',
|
|
default_provider: 'deepseek',
|
|
groups: visibleGroups,
|
|
allProviders: visibleGroups,
|
|
model_visibility: {
|
|
deepseek: { mode: 'include', models: ['deepseek-v4-flash', 'deepseek-v4-pro'] },
|
|
},
|
|
})
|
|
|
|
const appStore = useAppStore()
|
|
appStore.modelGroups = [
|
|
{
|
|
provider: 'deepseek',
|
|
label: 'DeepSeek',
|
|
base_url: 'https://api.deepseek.com/v1',
|
|
api_key: 'sk-test',
|
|
models: ['deepseek-v4-flash'],
|
|
available_models: ['deepseek-v4-flash', 'deepseek-v4-pro'],
|
|
},
|
|
]
|
|
|
|
const modelsStore = useModelsStore()
|
|
await modelsStore.fetchProviders()
|
|
|
|
expect(modelsStore.providers[0].models).toEqual(['deepseek-v4-flash', 'deepseek-v4-pro'])
|
|
expect(appStore.modelGroups[0].models).toEqual(['deepseek-v4-flash', 'deepseek-v4-pro'])
|
|
expect(appStore.modelGroups[0].available_models).toEqual(['deepseek-v4-flash', 'deepseek-v4-pro'])
|
|
expect(appStore.modelGroups[0].model_meta).toEqual({
|
|
'deepseek-v4-pro': { preview: true },
|
|
})
|
|
expect(appStore.modelVisibility).toEqual({
|
|
deepseek: { mode: 'include', models: ['deepseek-v4-flash', 'deepseek-v4-pro'] },
|
|
})
|
|
expect(appStore.selectedModel).toBe('deepseek-v4-flash')
|
|
expect(appStore.selectedProvider).toBe('deepseek')
|
|
})
|
|
})
|