text2text.js 538 B

123456789101112131415161718192021222324
  1. const config = require('../config');
  2. const PATH = 'chat/completions';
  3. const TIMEOUT_MS = 120000;
  4. // 普通 AI 请求参数
  5. function getBody(prompt) {
  6. return {
  7. model: config.MODEL_NAME || 'gpt-4.1',
  8. messages: [{ role: 'user', content: prompt }],
  9. stream: false
  10. };
  11. }
  12. // 豆包请求参数
  13. function getDoubaoBody(prompt) {
  14. return {
  15. model: config.DOUBAO_MODEL,
  16. messages: [{ role: 'user', content: prompt }],
  17. stream: false
  18. };
  19. }
  20. module.exports = { path: PATH, getBody, getDoubaoBody, timeoutMs: TIMEOUT_MS };