70 lines
2.1 KiB
JavaScript
70 lines
2.1 KiB
JavaScript
|
import assert from 'node:assert/strict'
|
|||
|
|
import test from 'node:test'
|
|||
|
|
|
|||
|
|
import {
|
|||
|
|
buildApplicationFieldsFromOntology,
|
|||
|
|
expandApplicationTimeWithDays,
|
|||
|
|
resolveApplicationReason,
|
|||
|
|
resolveApplicationTimeRange,
|
|||
|
|
resolvePromptField
|
|||
|
|
} from '../src/utils/expenseApplicationOntology.js'
|
|||
|
|
|
|||
|
|
const structuredApplicationPrompt = [
|
|||
|
|
'发生时间：2026-05-25',
|
|||
|
|
'地点：上海',
|
|||
|
|
'事由：支撑国网服务器部署',
|
|||
|
|
'天数：3天'
|
|||
|
|
].join('\n')
|
|||
|
|
|
|||
|
|
test('expense application fields use labeled reason and filter resolved missing slots', () => {
|
|||
|
|
const fields = buildApplicationFieldsFromOntology(
|
|||
|
|
{
|
|||
|
|
scenario: 'expense',
|
|||
|
|
intent: 'draft',
|
|||
|
|
entities: [],
|
|||
|
|
time_range: {},
|
|||
|
|
missing_slots: ['time_range', 'location', 'reason', 'amount']
|
|||
|
|
},
|
|||
|
|
structuredApplicationPrompt,
|
|||
|
|
{ name: '申请员工', departmentName: '交付部' }
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
assert.equal(fields.timeRange, '2026-05-25 至 2026-05-28')
|
|||
|
|
assert.equal(fields.location, '上海')
|
|||
|
|
assert.equal(fields.reason, '支撑国网服务器部署')
|
|||
|
|
assert.deepEqual(
|
|||
|
|
fields.missingSlots.map((item) => item.key),
|
|||
|
|
['amount']
|
|||
|
|
)
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
test('expense application prompt field parser supports multiline labels', () => {
|
|||
|
|
assert.equal(resolvePromptField(structuredApplicationPrompt, ['事由']), '支撑国网服务器部署')
|
|||
|
|
assert.equal(resolveApplicationReason(structuredApplicationPrompt), '支撑国网服务器部署')
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
test('expense application expands a single selected date with natural days', () => {
|
|||
|
|
const prompt = [
|
|||
|
|
'发生时间：2026-05-25',
|
|||
|
|
'去上海出差3天，支撑国网服务器部署'
|
|||
|
|
].join('\n')
|
|||
|
|
|
|||
|
|
assert.equal(expandApplicationTimeWithDays('2026-05-25', 3), '2026-05-25 至 2026-05-28')
|
|||
|
|
assert.equal(resolveApplicationTimeRange({ time_range: {} }, prompt), '2026-05-25 至 2026-05-28')
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
test('expense application keeps explicit time range before applying days', () => {
|
|||
|
|
assert.equal(
|
|||
|
|
resolveApplicationTimeRange(
|
|||
|
|
{
|
|||
|
|
time_range: {
|
|||
|
|
start_date: '2026-05-25',
|
|||
|
|
end_date: '2026-05-27'
|
|||
|
|
}
|
|||
|
|
},
|
|||
|
|
'去上海出差3天，支撑国网服务器部署'
|
|||
|
|
),
|
|||
|
|
'2026-05-25 至 2026-05-27'
|
|||
|
|
)
|
|||
|
|
})
|