text2img.js 1020 B

1234567891011121314151617181920212223242526272829303132333435
  1. const config = require('../config');
  2. const PATH = 'images/generations';
  3. const TIMEOUT_MS = 180000;
  4. // 普通 AI 请求参数(modelOverride 有值则优先,否则 dall-e-2)
  5. function getBody (prompt, outputPath, modelOverride) {
  6. const model =
  7. (modelOverride && String(modelOverride).trim()) || 'dall-e-2';
  8. return {
  9. model,
  10. prompt,
  11. n: 1,
  12. size: '1024x1024',
  13. response_format: outputPath ? 'b64_json' : 'url'
  14. };
  15. }
  16. // 豆包文生图请求参数(优先用 DOUBAO_IMAGE_MODEL,未配置则用 DOUBAO_MODEL)
  17. function getDoubaoBody (prompt, outputPath, modelOverride) {
  18. const model =
  19. (modelOverride && String(modelOverride).trim()) ||
  20. (config.DOUBAO_IMAGE_MODEL && config.DOUBAO_IMAGE_MODEL.trim()
  21. ? config.DOUBAO_IMAGE_MODEL.trim()
  22. : config.DOUBAO_MODEL);
  23. return {
  24. model,
  25. prompt,
  26. n: 1,
  27. size: '1024x1024',
  28. response_format: outputPath ? 'b64_json' : 'url'
  29. };
  30. }
  31. module.exports = { path: PATH, getBody, getDoubaoBody, timeoutMs: TIMEOUT_MS };