john
95 lines
3.2 KiB
JavaScript
95 lines
3.2 KiB
JavaScript
import test from 'node:test';
|
|
import assert from 'node:assert/strict';
|
|
import ExcelJS from 'exceljs';
|
|
import {
|
|
buildAttachmentPayload,
|
|
extractFileAttachmentsFromMessage,
|
|
} from './tkmind-proxy.mjs';
|
|
|
|
test('extractFileAttachmentsFromMessage reads metadata.fileAttachments', () => {
|
|
const attachments = extractFileAttachmentsFromMessage({
|
|
metadata: {
|
|
fileAttachments: [
|
|
{
|
|
assetId: 'asset-1',
|
|
filename: 'report.xlsx',
|
|
downloadUrl: '/api/mindspace/v1/assets/asset-1/download',
|
|
mimeType: 'application/vnd.openxmlformats-officedocument.spreadsheetml.sheet',
|
|
},
|
|
],
|
|
},
|
|
content: [],
|
|
});
|
|
assert.equal(attachments.length, 1);
|
|
assert.equal(attachments[0].filename, 'report.xlsx');
|
|
});
|
|
|
|
test('buildAttachmentPayload injects extracted attachment text for agent execution', async () => {
|
|
const xlsxBuffer = Buffer.from('PK\x03\x04');
|
|
const payload = await buildAttachmentPayload({
|
|
userId: 'user-1',
|
|
userMessage: {
|
|
metadata: {
|
|
displayText: '请分析这份表格',
|
|
fileAttachments: [
|
|
{
|
|
assetId: 'asset-9',
|
|
filename: 'note.txt',
|
|
downloadUrl: '/api/mindspace/v1/assets/asset-9/download',
|
|
mimeType: 'text/plain',
|
|
},
|
|
],
|
|
},
|
|
content: [{ type: 'text', text: '请分析这份表格\n\n[文件1: note.txt]: /api/mindspace/v1/assets/asset-9/download' }],
|
|
},
|
|
localFetchAsset: async () => ({
|
|
buffer: Buffer.from('hello attachment'),
|
|
mimeType: 'text/plain',
|
|
}),
|
|
});
|
|
|
|
assert.ok(payload?.userMessage);
|
|
const text = payload.userMessage.content[0].text;
|
|
assert.match(text, /【TKMind 附件分析结果/);
|
|
assert.match(text, /hello attachment/);
|
|
assert.match(text, /oa\/note\.txt/);
|
|
assert.equal(payload.userMessage.metadata.displayText, '请分析这份表格');
|
|
});
|
|
|
|
test('buildAttachmentPayload tells the agent to prefer dedicated tools for xlsx', async () => {
|
|
const workbook = new ExcelJS.Workbook();
|
|
workbook.addWorksheet('销售').addRow(['地区', '销售额']);
|
|
const xlsxBuffer = Buffer.from(await workbook.xlsx.writeBuffer());
|
|
const payload = await buildAttachmentPayload({
|
|
userId: 'user-1',
|
|
userMessage: {
|
|
metadata: {
|
|
displayText: '分析一下',
|
|
fileAttachments: [
|
|
{
|
|
assetId: 'asset-xlsx',
|
|
filename: 'sales.xlsx',
|
|
downloadUrl: '/api/mindspace/v1/assets/asset-xlsx/download',
|
|
mimeType: 'application/vnd.openxmlformats-officedocument.spreadsheetml.sheet',
|
|
},
|
|
],
|
|
},
|
|
content: [{ type: 'text', text: '分析一下' }],
|
|
},
|
|
fetchImpl: async () => {
|
|
throw new Error('use local asset');
|
|
},
|
|
localFetchAsset: async () => ({
|
|
buffer: xlsxBuffer,
|
|
mimeType: 'application/vnd.openxmlformats-officedocument.spreadsheetml.sheet',
|
|
}),
|
|
});
|
|
|
|
const text = payload.userMessage.content[0].text;
|
|
assert.match(text, /Excel 分析提示/);
|
|
assert.match(text, /excel_inspect \/ excel_analyze \/ excel_chart/);
|
|
assert.match(text, /excel_report/);
|
|
assert.match(text, /禁止用 write_file \/ edit_file 手工抄写数字/);
|
|
assert.match(text, /工作区路径: oa\/sales\.xlsx/);
|
|
});
|