const config = require('../config');

const PATH = 'images/generations';
const TIMEOUT_MS = 180000;

// 普通 AI 请求参数（modelOverride 有值则优先，否则 dall-e-2）
function getBody (prompt, outputPath, modelOverride) {
  const model =
    (modelOverride && String(modelOverride).trim()) || 'dall-e-2';
  return {
    model,
    prompt,
    n: 1,
    size: '1024x1024',
    response_format: outputPath ? 'b64_json' : 'url'
  };
}

// 豆包文生图请求参数（优先用 DOUBAO_IMAGE_MODEL，未配置则用 DOUBAO_MODEL）
function getDoubaoBody (prompt, outputPath, modelOverride) {
  const model =
    (modelOverride && String(modelOverride).trim()) ||
    (config.DOUBAO_IMAGE_MODEL && config.DOUBAO_IMAGE_MODEL.trim()
      ? config.DOUBAO_IMAGE_MODEL.trim()
      : config.DOUBAO_MODEL);
  return {
    model,
    prompt,
    n: 1,
    size: '1024x1024',
    response_format: outputPath ? 'b64_json' : 'url'
  };
}

module.exports = { path: PATH, getBody, getDoubaoBody, timeoutMs: TIMEOUT_MS };