115 lines
3.3 KiB
TypeScript
115 lines
3.3 KiB
TypeScript
/*
|
|
Copyright (C) 2023-2026 QuantumNous
|
|
|
|
This program is free software: you can redistribute it and/or modify
|
|
it under the terms of the GNU Affero General Public License as
|
|
published by the Free Software Foundation, either version 3 of the
|
|
License, or (at your option) any later version.
|
|
|
|
This program is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU Affero General Public License for more details.
|
|
|
|
You should have received a copy of the GNU Affero General Public License
|
|
along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
|
|
For commercial licensing, please contact support@quantumnous.com
|
|
*/
|
|
import { describe, expect, test } from 'vitest'
|
|
|
|
import { channelSchema } from '../../types'
|
|
import {
|
|
CHANNEL_FORM_DEFAULT_VALUES,
|
|
channelFormSchema,
|
|
transformChannelToFormDefaults,
|
|
transformFormDataToCreatePayload,
|
|
type ChannelFormValues,
|
|
} from '../channel-form'
|
|
|
|
const CHANNEL_TYPE_VERTEX_AI = 41
|
|
|
|
function vertexAPIKeyForm(
|
|
overrides: Partial<ChannelFormValues> = {}
|
|
): ChannelFormValues {
|
|
return {
|
|
...CHANNEL_FORM_DEFAULT_VALUES,
|
|
name: 'Vertex AI',
|
|
type: CHANNEL_TYPE_VERTEX_AI,
|
|
key: 'test-key',
|
|
models: 'gemini-4-argon',
|
|
other: '{"default":"global"}',
|
|
vertex_key_type: 'api_key',
|
|
...overrides,
|
|
}
|
|
}
|
|
|
|
function savedSettings(form: ChannelFormValues): Record<string, unknown> {
|
|
const payload = transformFormDataToCreatePayload(
|
|
channelFormSchema.parse(form)
|
|
)
|
|
return JSON.parse(payload.channel.settings ?? '{}')
|
|
}
|
|
|
|
describe('Vertex AI API Key project ID', () => {
|
|
test('saves the trimmed project ID in API Key mode', () => {
|
|
const settings = savedSettings(
|
|
vertexAPIKeyForm({ vertex_project_id: ' my-project ' })
|
|
)
|
|
|
|
expect(settings).toMatchObject({
|
|
vertex_key_type: 'api_key',
|
|
vertex_project_id: 'my-project',
|
|
})
|
|
})
|
|
|
|
test('accepts API Key mode without a project ID and saves none', () => {
|
|
const form = vertexAPIKeyForm({ vertex_project_id: ' ' })
|
|
|
|
expect(channelFormSchema.safeParse(form).success).toBe(true)
|
|
expect(savedSettings(form)).not.toHaveProperty('vertex_project_id')
|
|
})
|
|
|
|
test('drops a saved project ID after switching to JSON keys', () => {
|
|
const settings = savedSettings(
|
|
vertexAPIKeyForm({
|
|
vertex_key_type: 'json',
|
|
key: '{"project_id":"json-project"}',
|
|
vertex_project_id: 'my-project',
|
|
settings: JSON.stringify({
|
|
vertex_key_type: 'api_key',
|
|
vertex_project_id: 'my-project',
|
|
}),
|
|
})
|
|
)
|
|
|
|
expect(settings).toMatchObject({ vertex_key_type: 'json' })
|
|
expect(settings).not.toHaveProperty('vertex_project_id')
|
|
})
|
|
|
|
test('loads the saved project ID when editing a channel', () => {
|
|
const channel = channelSchema.parse({
|
|
id: 7,
|
|
name: 'Vertex AI',
|
|
type: CHANNEL_TYPE_VERTEX_AI,
|
|
key: '',
|
|
status: 1,
|
|
created_time: 1,
|
|
test_time: 0,
|
|
response_time: 0,
|
|
balance_updated_time: 0,
|
|
models: 'gemini-4-argon',
|
|
group: 'default',
|
|
other: '{"default":"global"}',
|
|
settings: JSON.stringify({
|
|
vertex_key_type: 'api_key',
|
|
vertex_project_id: 'my-project',
|
|
}),
|
|
})
|
|
|
|
expect(transformChannelToFormDefaults(channel).vertex_project_id).toBe(
|
|
'my-project'
|
|
)
|
|
})
|
|
})
|