95 lines
3.0 KiB
JavaScript
95 lines
3.0 KiB
JavaScript
import test from 'brittle'
|
|
import { readFileSync } from 'node:fs'
|
|
import vm from 'node:vm'
|
|
|
|
const QVAC_SRC = readFileSync(new URL('../lib/agent/agent-qvac.js', import.meta.url), 'utf8')
|
|
const MODELS_SRC = readFileSync(
|
|
new URL('../lib/agent/agent-models.js', import.meta.url),
|
|
'utf8'
|
|
)
|
|
const TUI_SRC = readFileSync(new URL('../lib/agent/agent-tui.js', import.meta.url), 'utf8')
|
|
|
|
function load() {
|
|
const sandbox = { console }
|
|
vm.createContext(sandbox)
|
|
vm.runInContext(
|
|
QVAC_SRC +
|
|
'\n' +
|
|
MODELS_SRC +
|
|
'\n' +
|
|
TUI_SRC +
|
|
'\n;this.__exports = { bareAgentHonorExplicitQvacModel, bareAgentApplyLiveModel, bareAgentApplyProviderProfile, bareAgentQvacGetProfile, bareAgentQvacFindChatModel, bareAgentQvacResolveCtxSize, bareAgentQvacSyncSettingsForModel }',
|
|
sandbox
|
|
)
|
|
return sandbox.__exports
|
|
}
|
|
|
|
test('explicit QVAC model survives profile apply (live Discord/CLI switch)', async (t) => {
|
|
const { bareAgentApplyProviderProfile } = load()
|
|
const next = bareAgentApplyProviderProfile({
|
|
backend: 'qvac',
|
|
qvac_profile: 'recommended',
|
|
qvac_model: 'QWEN3_8B_INST_Q4_K_M',
|
|
model: 'QWEN3_8B_INST_Q4_K_M'
|
|
})
|
|
t.is(next.qvac_model, 'QWEN3_8B_INST_Q4_K_M')
|
|
t.is(next.model, 'QWEN3_8B_INST_Q4_K_M')
|
|
})
|
|
|
|
test('empty QVAC model fills from profile', async (t) => {
|
|
const { bareAgentApplyProviderProfile, bareAgentQvacGetProfile } = load()
|
|
const next = bareAgentApplyProviderProfile({
|
|
backend: 'qvac',
|
|
qvac_profile: 'strong'
|
|
})
|
|
t.is(next.qvac_model, bareAgentQvacGetProfile('strong').chatModel)
|
|
})
|
|
|
|
test('apply live model on QVAC sets both keys and profile when catalog-known', async (t) => {
|
|
const { bareAgentApplyLiveModel } = load()
|
|
const next = bareAgentApplyLiveModel(
|
|
{ backend: 'qvac', qvac_profile: 'recommended', qvac_model: 'QWEN3_1_7B_INST_Q4' },
|
|
'QWEN3_600M_INST_Q4'
|
|
)
|
|
t.is(next.qvac_model, 'QWEN3_600M_INST_Q4')
|
|
t.is(next.model, 'QWEN3_600M_INST_Q4')
|
|
t.is(next.qvac_profile, 'lite')
|
|
})
|
|
|
|
test('live QVAC model switch resets ctx to the new model card', async (t) => {
|
|
const { bareAgentApplyLiveModel, bareAgentQvacResolveCtxSize, bareAgentQvacGetProfile } =
|
|
load()
|
|
const next = bareAgentApplyLiveModel(
|
|
{
|
|
backend: 'qvac',
|
|
qvac_profile: 'recommended',
|
|
qvac_model: 'QWEN3_1_7B_INST_Q4',
|
|
qvac_ctx_size: 32768
|
|
},
|
|
'QWEN3_5_4B_MULTIMODAL_Q4_K_M'
|
|
)
|
|
t.is(next.qvac_model, 'QWEN3_5_4B_MULTIMODAL_Q4_K_M')
|
|
t.is(next.qvac_ctx_size, 0)
|
|
t.is(
|
|
bareAgentQvacResolveCtxSize(next, bareAgentQvacGetProfile(next.qvac_profile)),
|
|
262144
|
|
)
|
|
})
|
|
|
|
test('apply live model on REST only changes model', async (t) => {
|
|
const { bareAgentApplyLiveModel } = load()
|
|
const next = bareAgentApplyLiveModel(
|
|
{
|
|
backend: 'rest',
|
|
provider: 'groq',
|
|
rest_api_key: 'gsk',
|
|
rest_base_url: 'https://api.groq.com/openai/v1',
|
|
model: 'old'
|
|
},
|
|
'llama-3.1-8b-instant'
|
|
)
|
|
t.is(next.backend, 'rest')
|
|
t.is(next.model, 'llama-3.1-8b-instant')
|
|
t.absent(next.qvac_model)
|
|
})
|