62 lines
3.6 KiB
JavaScript
62 lines
3.6 KiB
JavaScript
import { readFileSync } from 'node:fs'
|
||
import { test } from 'node:test'
|
||
import assert from 'node:assert/strict'
|
||
|
||
const root = new URL('..', import.meta.url)
|
||
const read = (path) => readFileSync(new URL(path, root), 'utf8')
|
||
|
||
const page = read('app/pages/drama/[id]/episode/[episodeNumber].vue')
|
||
const useAgent = read('app/composables/useAgent.ts')
|
||
const useApi = read('app/composables/useApi.ts')
|
||
|
||
test('workbench offers model selectors for rewrite, image and video generation', () => {
|
||
// 三个选择器（自定义 ModelSelect 弹窗组件，非原生 select）
|
||
assert.match(page, /<ModelSelect/)
|
||
assert.match(page, /v-model="chatModel"/)
|
||
assert.match(page, /v-model="imageModel"/)
|
||
assert.match(page, /v-model="videoModel"/)
|
||
assert.doesNotMatch(page, /<select v-model="chatModel"/)
|
||
// 选项汇总该类型全部启用配置的模型（去重、按优先级），API 返回的 model 可能已是数组
|
||
assert.match(page, /function collectModelOptions\(cfgs\)/)
|
||
assert.match(page, /if \(Array\.isArray\(raw\)\) return raw\.filter\(Boolean\)/)
|
||
assert.match(page, /collectModelOptions\(textConfigs\.value\)/)
|
||
assert.match(page, /collectModelOptions\(imageConfigs\.value\)/)
|
||
assert.match(page, /collectModelOptions\(videoConfigs\.value\)/)
|
||
assert.match(page, /aiConfigAPI\.list\('text'\)/)
|
||
})
|
||
|
||
test('model select dropdown is a custom designed popover', () => {
|
||
const component = read('app/components/ModelSelect.vue')
|
||
|
||
assert.match(component, /Teleport to="body"/)
|
||
assert.match(component, /model-select-menu/)
|
||
assert.match(component, /model-select-option/)
|
||
assert.match(component, /Check :size="12"/)
|
||
assert.match(component, /model-select-backdrop/)
|
||
assert.match(component, /emit\('update:modelValue', model\)/)
|
||
// 多配置时展示来源配置名
|
||
assert.match(component, /showConfig/)
|
||
assert.match(component, /opt-config/)
|
||
})
|
||
|
||
test('selected models are sent with rewrite, image and video generation requests', () => {
|
||
// 选中模型时连同其所属配置一起调用
|
||
assert.match(page, /function ownerConfigId\(options, model\)/)
|
||
// 改写（Agent）透传模型与配置
|
||
assert.match(useAgent, /model: model \|\| undefined/)
|
||
assert.match(useAgent, /config_id: configId \|\| undefined/)
|
||
assert.match(page, /runAgent\('script_rewriter'[\s\S]*?chatModelOverride\(\), chatConfigId\(\)\)/)
|
||
// 图片生成透传模型与配置
|
||
assert.match(useApi, /generateImage: \(id: number, episodeId: number, model\?: string, configId\?: number, textModel\?: string, textConfigId\?: number\)/)
|
||
// 文本模型覆盖随生图/提取请求透传（缺提示词时后端触发提示词 Agent）
|
||
assert.match(useApi, /text_model: textModel \|\| undefined/)
|
||
assert.match(useApi, /text_config_id: textConfigId \|\| undefined/)
|
||
assert.match(useApi, /config_id: configId \|\| undefined/)
|
||
assert.match(page, /characterAPI\.generateImage\(id, epId\.value, imageModel\.value \|\| undefined, ownerConfigId\(imageModelOptions\.value, imageModel\.value\), chatModelOverride\(\), chatConfigId\(\)\)/)
|
||
assert.match(page, /sceneAPI\.generateImage\(id, epId\.value, imageModel\.value \|\| undefined, ownerConfigId\(imageModelOptions\.value, imageModel\.value\), chatModelOverride\(\), chatConfigId\(\)\)/)
|
||
assert.match(page, /characterAPI\.batchImages\(ids, epId\.value, imageModel\.value \|\| undefined, ownerConfigId\(imageModelOptions\.value, imageModel\.value\), chatModelOverride\(\), chatConfigId\(\)\)/)
|
||
// 视频生成透传模型与配置
|
||
assert.match(page, /model: videoModel\.value \|\| undefined/)
|
||
assert.match(page, /config_id: ownerConfigId\(videoModelOptions\.value, videoModel\.value\)/)
|
||
})
|