58 lines
1.8 KiB
TypeScript
58 lines
1.8 KiB
TypeScript
|
import { expect, test } from 'bun:test'
|
||
|
|
|
||
|
|
import * as v from 'valibot'
|
||
|
|
|
||
|
|
import {
|
||
|
|
modelProfileSchema,
|
||
|
|
type ModelProfileValidationContext
|
||
|
|
} from '@/app/ai/models/settings/profile-editor/schema'
|
||
|
|
|
||
|
|
const messages = {
|
||
|
|
requiredField: 'Required',
|
||
|
|
invalidURL: 'Invalid URL',
|
||
|
|
designToolsRequired: 'Tools required'
|
||
|
|
}
|
||
|
|
const context: ModelProfileValidationContext = {
|
||
|
|
customModel: false,
|
||
|
|
providerKind: 'api',
|
||
|
|
supportsCustomBaseURL: false,
|
||
|
|
design: true,
|
||
|
|
intent: 'save'
|
||
|
|
}
|
||
|
|
const values = {
|
||
|
|
name: 'Design',
|
||
|
|
modelID: 'known-model',
|
||
|
|
customModelID: '',
|
||
|
|
baseURL: '',
|
||
|
|
credential: false,
|
||
|
|
tools: true
|
||
|
|
}
|
||
|
|
|
||
|
|
function errors(options: Partial<ModelProfileValidationContext>, input = values) {
|
||
|
|
const result = v.safeParse(modelProfileSchema({ ...context, ...options }, messages), input)
|
||
|
|
return result.success ? {} : v.flatten(result.issues).nested
|
||
|
|
}
|
||
|
|
|
||
|
|
test('model requirements attach to the active field without fallback to an inactive model ID', () => {
|
||
|
|
expect(errors({}, { ...values, modelID: '' })).toEqual({ modelID: ['Required'] })
|
||
|
|
expect(errors({ customModel: true })).toEqual({ customModelID: ['Required'] })
|
||
|
|
expect(
|
||
|
|
errors({ customModel: true }, { ...values, modelID: '', customModelID: 'custom' })
|
||
|
|
).toEqual({})
|
||
|
|
})
|
||
|
|
|
||
|
|
test('tool support errors are forwarded by the schema to the active model field', () => {
|
||
|
|
expect(errors({}, { ...values, tools: false })).toEqual({ modelID: ['Tools required'] })
|
||
|
|
expect(
|
||
|
|
errors({ customModel: true }, { ...values, customModelID: 'custom', tools: false })
|
||
|
|
).toEqual({ customModelID: ['Tools required'] })
|
||
|
|
expect(errors({ design: false }, { ...values, tools: false })).toEqual({})
|
||
|
|
})
|
||
|
|
|
||
|
|
test('independent field errors survive a failed tool-support rule', () => {
|
||
|
|
expect(errors({}, { ...values, name: '', tools: false })).toEqual({
|
||
|
|
name: ['Required'],
|
||
|
|
modelID: ['Tools required']
|
||
|
|
})
|
||
|
|
})
|