9router/open-sse/handlers/imageProviders/sdwebui.js

21 lines
784 B
JavaScript
Raw Normal View History

2026-05-04 00:29:02 -04:00
// SD WebUI (AUTOMATIC1111) — local, noAuth
import { nowSec } from "./_base.js";
import { PROVIDER_MEDIA } from "../../providers/index.js";
const BASE_URL = PROVIDER_MEDIA["sdwebui"]?.imageConfig?.baseUrl;
2026-05-04 00:29:02 -04:00
export default {
noAuth: true,
buildUrl: () => BASE_URL,
2026-05-04 00:29:02 -04:00
buildHeaders: () => ({ "Content-Type": "application/json" }),
buildBody: (_model, body) => {
const { prompt, n = 1, size = "1024x1024" } = body;
const [width, height] = size.split("x").map(Number);
return { prompt, width: width || 512, height: height || 512, steps: 20, batch_size: n };
},
normalize: (responseBody) => {
const images = Array.isArray(responseBody.images) ? responseBody.images.map((img) => ({ b64_json: img })) : [];
return { created: nowSec(), data: images };
},
};