fmode-image-analysis-smoke.js 4.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108
  1. #!/usr/bin/env node
  2. const fs = require('fs');
  3. const http = require('http');
  4. const os = require('os');
  5. const path = require('path');
  6. const { analyzeFmodeImage } = require('../mcp/src/tools/fmode-image-analysis');
  7. async function main() {
  8. const noToken = await analyzeFmodeImage({
  9. images: [tinyPngDataUrl()],
  10. prompt: '识别图片内容',
  11. allowEnvToken: false
  12. });
  13. assert(noToken.status === 'needs_token', 'missing token should return needs_token');
  14. assert(Array.isArray(noToken.errors) && noToken.errors.length === 0, 'missing token should keep errors empty');
  15. assert(!/Bearer\s+/i.test(JSON.stringify(noToken)), 'missing token output should not leak bearer headers');
  16. assert(/apig-pay/.test(noToken.assistantMessage), 'missing token output should include recharge link');
  17. const mock = await startMockFmode();
  18. try {
  19. const out = path.join(os.tmpdir(), `fmode-image-analysis-smoke-${Date.now()}`);
  20. const result = await analyzeFmodeImage({
  21. baseUrl: mock.url,
  22. path: 'v1/chat/completions',
  23. fmodeToken: 'mock-platform-token',
  24. images: [tinyPngDataUrl()],
  25. prompt: '请用中文识别图片中的文字和主体。',
  26. output: out,
  27. allowEnvToken: false
  28. });
  29. assert(result.status === 'ok', 'mock image analysis should return ok');
  30. assert(result.summary.model === 'doubao-seed-2-0-pro-260215', 'default model should be doubao-seed-2-0-pro-260215');
  31. assert(result.assistantMessage.includes('Fmode 图片分析结果'), 'assistant message should include result heading');
  32. assert(result.data.analysis.includes('测试图片'), 'analysis text should be parsed from mock response');
  33. assert(result.files.length === 2 && result.files.every(file => fs.existsSync(file)), 'output files should be written');
  34. const request = mock.requests[0];
  35. assert(request.headers.authorization === 'Bearer mock-platform-token', 'mock request should include platform token');
  36. assert(request.body.model === 'doubao-seed-2-0-pro-260215', 'request should use the configured default model');
  37. assert(Array.isArray(request.body.messages), 'request should use OpenAI-compatible messages');
  38. const userMessage = request.body.messages.find(item => item.role === 'user');
  39. assert(Array.isArray(userMessage.content), 'user content should be multimodal array');
  40. assert(userMessage.content.some(item => item.type === 'image_url'), 'user content should include image_url');
  41. assert(!JSON.stringify(result).includes('mock-platform-token'), 'public result should not leak token');
  42. } finally {
  43. await mock.close();
  44. }
  45. console.log('fmode image analysis smoke ok');
  46. }
  47. function startMockFmode() {
  48. const requests = [];
  49. const server = http.createServer((req, res) => {
  50. if (!req.url.startsWith('/v1/chat/completions')) {
  51. res.writeHead(404, { 'content-type': 'application/json' });
  52. res.end(JSON.stringify({ error: 'not_found' }));
  53. return;
  54. }
  55. let body = '';
  56. req.on('data', chunk => { body += chunk; });
  57. req.on('end', () => {
  58. requests.push({
  59. headers: req.headers,
  60. body: body ? JSON.parse(body) : {}
  61. });
  62. res.writeHead(200, { 'content-type': 'application/json' });
  63. res.end(JSON.stringify({
  64. choices: [
  65. {
  66. finish_reason: 'stop',
  67. message: {
  68. content: '这是一张测试图片,可见一个小色块;未发现明显风险。'
  69. }
  70. }
  71. ],
  72. usage: {
  73. prompt_tokens: 12,
  74. completion_tokens: 18,
  75. total_tokens: 30
  76. }
  77. }));
  78. });
  79. });
  80. return new Promise(resolve => {
  81. server.listen(0, '127.0.0.1', () => {
  82. resolve({
  83. url: `http://127.0.0.1:${server.address().port}`,
  84. requests,
  85. close: () => new Promise(done => server.close(done))
  86. });
  87. });
  88. });
  89. }
  90. function tinyPngDataUrl() {
  91. return 'data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAQAAAC1HAwCAAAAC0lEQVR42mP8/x8AAwMCAO+/p9sAAAAASUVORK5CYII=';
  92. }
  93. function assert(condition, message) {
  94. if (!condition) throw new Error(message);
  95. }
  96. main().catch(error => {
  97. console.error(error && error.stack ? error.stack : String(error));
  98. process.exit(1);
  99. });