57 lines
1.8 KiB
JavaScript
57 lines
1.8 KiB
JavaScript
|
import test from 'node:test';
|
||
|
|
import assert from 'node:assert/strict';
|
||
|
|
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, '请分析这份表格');
|
||
|
|
});
|