text2text.js 845 B

123456789101112131415161718192021222324252627282930
  1. const config = require('../config');
  2. const PATH = 'chat/completions';
  3. const TIMEOUT_MS = 120000;
  4. // 普通 AI 请求参数(modelOverride 有值则优先,否则 config.MODEL_NAME,再否则默认)
  5. function getBody (prompt, modelOverride) {
  6. const model =
  7. (modelOverride && String(modelOverride).trim()) ||
  8. config.MODEL_NAME ||
  9. 'gpt-4.1';
  10. return {
  11. model,
  12. messages: [{ role: 'user', content: prompt }],
  13. stream: false
  14. };
  15. }
  16. // 豆包请求参数(可选 modelOverride 覆盖接入点 ID)
  17. function getDoubaoBody (prompt, modelOverride) {
  18. const model =
  19. (modelOverride && String(modelOverride).trim()) || config.DOUBAO_MODEL;
  20. return {
  21. model,
  22. messages: [{ role: 'user', content: prompt }],
  23. stream: false
  24. };
  25. }
  26. module.exports = { path: PATH, getBody, getDoubaoBody, timeoutMs: TIMEOUT_MS };