utils.js

const { spawn } = require('child_process');
const fs = require('fs');
const path = require('path');

exports.translate = async (text, { key, file }) => {
  return new Promise((resolve) => {
    const payload = JSON.stringify({
      model: 'Qwen/Qwen2.5-72B-Instruct',
      messages: [
        {
          role: 'system',
          content: [
            {
              type: 'text',
              text: 'You are a professional, authentic machine translation engine. Especially skilled in translating tasks from English to Chinese related to the Rust programming language.',
            },
          ],
        },
        {
          role: 'user',
          content: [
            {
              type: 'text',
              text: `Translate the following source text to chinese: ${text}，Output translation directly without any additional text. Absolutely refrain from adding symbols or tags that were not in the original translation content.Remember, Keep ALL HTML TAG AND ATTRIBUTE, ONLY TRANSLATE CONTENT!`,
            },
          ],
        },
      ],
      temperature: 0,
      top_p: 0.95,
      max_tokens: 4096,
    });
    const apiKey = process.env.OPENAI_API_KEY;
    const url = process.env.OPENAI_URL;
    const res = spawn('curl', [
      '--request',
      'POST',
      '--url',
      url,
      '-H',
      'Content-Type: application/json',
      '-H',
      `Authorization: Bearer ${apiKey}`,
      '-d',
      payload,
    ]);
    let result = '';
    res.stdout.on('data', (data) => {
      result += data.toString();
    });
    res.stdout.on('close', () => {
      try {
        const json = JSON.parse(result);
        const translate = json.choices?.[0]?.message?.content;
        if (!translate) {
          console.error(`ai 翻译失败: [${text}] -> [${result}]`);
          // Note: 创建一个 error.txt ，如果不存在则创建
          const errorPath = path.resolve(__dirname, 'error.txt');
          if (!fs.existsSync(errorPath)) {
            fs.writeFileSync(errorPath, '');
          }
          // Note：追加到文件
          fs.appendFileSync(
            errorPath,
            `[${file}-${key}]: ${text} -> ${result}\n`
          );
          resolve('【翻译引擎出错，请联系作者】');
          return;
        }
        resolve(translate);
      } catch (e) {
        console.error(`ai 翻译失败: [${text}] -> [${result}]`);
        // Note: 创建一个 error.txt ，如果不存在则创建
        if (!fs.existsSync('error.txt')) {
          fs.writeFileSync('error.txt', '');
        }
        // Note：追加到文件
        fs.appendFileSync(
          'error.txt',
          `[${file}-${key}]: ${text} -> ${result}\n`
        );
        resolve('【翻译引擎出错，请联系作者】');
      }
    });
  });
};