40 lines
949 B
TypeScript
40 lines
949 B
TypeScript
import { beforeEach, describe, expect, it } from 'vitest'
|
|
import {
|
|
assertOfficialModelRegistered,
|
|
isOfficialModelRegistered,
|
|
registerOfficialModel,
|
|
resetOfficialModelRegistryForTest,
|
|
} from '@/lib/providers/official/model-registry'
|
|
|
|
describe('official model registry', () => {
|
|
beforeEach(() => {
|
|
resetOfficialModelRegistryForTest()
|
|
})
|
|
|
|
it('throws MODEL_NOT_REGISTERED when model is absent', () => {
|
|
expect(() =>
|
|
assertOfficialModelRegistered({
|
|
provider: 'bailian',
|
|
modality: 'llm',
|
|
modelId: 'qwen-plus',
|
|
}),
|
|
).toThrow(/MODEL_NOT_REGISTERED/)
|
|
})
|
|
|
|
it('accepts registered official model', () => {
|
|
registerOfficialModel({
|
|
provider: 'siliconflow',
|
|
modality: 'image',
|
|
modelId: 'sf-image',
|
|
})
|
|
|
|
expect(
|
|
isOfficialModelRegistered({
|
|
provider: 'siliconflow',
|
|
modality: 'image',
|
|
modelId: 'sf-image',
|
|
}),
|
|
).toBe(true)
|
|
})
|
|
})
|