feat(hermes): add model catalog controls

This commit is contained in:
晴天
2026-05-27 05:06:32 +08:00
parent 18c0164bdb
commit c9e4a380a9
8 changed files with 723 additions and 3 deletions

View File

@@ -127,6 +127,18 @@ test('Hermes 配置页会暴露基础模型结构化配置字段', () => {
}
})
test('Hermes 配置页会暴露模型目录结构化配置字段', () => {
for (const id of [
'hm-model-catalog-save',
'hm-model-catalog-enabled',
'hm-model-catalog-url',
'hm-model-catalog-ttl-hours',
'hm-model-catalog-providers-json',
]) {
assert.match(source, new RegExp(`id="${id}"`), `缺少 ${id}`)
}
})
test('Hermes 配置页会暴露 provider 超时覆盖结构化配置字段', () => {
for (const id of [
'hm-provider-overrides-save',

View File

@@ -0,0 +1,122 @@
import test from 'node:test'
import assert from 'node:assert/strict'
import {
buildHermesModelCatalogConfigValues,
mergeHermesModelCatalogConfig,
} from '../scripts/dev-api.js'
const DEFAULT_CATALOG_URL = 'https://hermes-agent.nousresearch.com/docs/api/model-catalog.json'
test('Hermes 模型目录配置读取会提供上游默认值', () => {
const values = buildHermesModelCatalogConfigValues({})
assert.deepEqual(values, {
modelCatalogEnabled: true,
modelCatalogUrl: DEFAULT_CATALOG_URL,
modelCatalogTtlHours: 24,
modelCatalogProvidersJson: '{}',
})
})
test('Hermes 模型目录配置读取会回显 YAML 字段', () => {
const values = buildHermesModelCatalogConfigValues({
model_catalog: {
enabled: false,
url: 'https://example.com/catalog.json',
ttl_hours: 6,
providers: {
openrouter: {
url: 'https://mirror.example.com/openrouter.json',
},
nous: {
url: 'https://mirror.example.com/nous.json',
},
},
},
})
assert.equal(values.modelCatalogEnabled, false)
assert.equal(values.modelCatalogUrl, 'https://example.com/catalog.json')
assert.equal(values.modelCatalogTtlHours, 6)
assert.deepEqual(JSON.parse(values.modelCatalogProvidersJson), {
openrouter: { url: 'https://mirror.example.com/openrouter.json' },
nous: { url: 'https://mirror.example.com/nous.json' },
})
})
test('Hermes 模型目录配置保存会保留未知字段并写入上游结构', () => {
const next = mergeHermesModelCatalogConfig({
model: { provider: 'openrouter' },
model_catalog: {
enabled: false,
url: 'https://old.example.com/catalog.json',
ttl_hours: 12,
providers: {
openrouter: {
url: 'https://old.example.com/openrouter.json',
},
},
custom_flag: 'keep-catalog',
},
streaming: { enabled: true },
}, {
modelCatalogEnabled: true,
modelCatalogUrl: 'https://catalog.example.com/model-catalog.json',
modelCatalogTtlHours: 48,
modelCatalogProvidersJson: JSON.stringify({
openrouter: { url: 'https://catalog.example.com/openrouter.json' },
nous: { url: 'https://catalog.example.com/nous.json' },
}),
})
assert.deepEqual(next.model, { provider: 'openrouter' })
assert.deepEqual(next.streaming, { enabled: true })
assert.equal(next.model_catalog.enabled, true)
assert.equal(next.model_catalog.url, 'https://catalog.example.com/model-catalog.json')
assert.equal(next.model_catalog.ttl_hours, 48)
assert.equal(next.model_catalog.providers.openrouter.url, 'https://catalog.example.com/openrouter.json')
assert.equal(next.model_catalog.providers.nous.url, 'https://catalog.example.com/nous.json')
assert.equal(next.model_catalog.custom_flag, 'keep-catalog')
})
test('Hermes 模型目录配置保存空 provider 覆盖会移除 providers', () => {
const next = mergeHermesModelCatalogConfig({
model_catalog: {
providers: {
openrouter: { url: 'https://old.example.com/openrouter.json' },
},
custom_flag: 'keep-catalog',
},
streaming: { enabled: true },
}, {
modelCatalogProvidersJson: '{}',
})
assert.equal(Object.hasOwn(next.model_catalog, 'providers'), false)
assert.equal(next.model_catalog.custom_flag, 'keep-catalog')
assert.deepEqual(next.streaming, { enabled: true })
})
test('Hermes 模型目录配置保存会拒绝非法 URL、TTL 和 provider JSON', () => {
assert.throws(
() => mergeHermesModelCatalogConfig({}, { modelCatalogUrl: 'ftp://example.com/catalog.json' }),
/model_catalog\.url/,
)
assert.throws(
() => mergeHermesModelCatalogConfig({}, { modelCatalogTtlHours: 0 }),
/model_catalog\.ttl_hours/,
)
assert.throws(
() => mergeHermesModelCatalogConfig({}, { modelCatalogProvidersJson: '[' }),
/model_catalog\.providers/,
)
assert.throws(
() => mergeHermesModelCatalogConfig({}, { modelCatalogProvidersJson: JSON.stringify({ 'bad provider': { url: 'https://example.com/catalog.json' } }) }),
/model_catalog\.providers\.bad provider/,
)
assert.throws(
() => mergeHermesModelCatalogConfig({}, { modelCatalogProvidersJson: JSON.stringify({ openrouter: { url: 'file:///tmp/catalog.json' } }) }),
/model_catalog\.providers\.openrouter\.url/,
)
})