File size: 2,159 Bytes
253e783 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 | import { describe, expect, it } from "vitest";
import { generateImage } from "./runtime.js";
import type { ImageGenerationProvider } from "./types.js";
describe("image-generation output capabilities", () => {
it.each([
{ model: "extended", accepted: true },
{ model: "legacy", accepted: false },
{ model: "restricted", accepted: false },
])("filters output hints against $model model capabilities", async ({ model, accepted }) => {
let seenRequest: { quality?: string; outputFormat?: string; background?: string } | undefined;
const provider: ImageGenerationProvider = {
id: "test",
capabilities: {
generate: {},
edit: { enabled: true },
output: {
qualities: ["low"],
formats: ["png"],
backgrounds: ["opaque"],
qualitiesByModel: { extended: ["xhigh", "max"], restricted: [] },
formatsByModel: { extended: ["webp"], restricted: [] },
backgroundsByModel: { extended: ["transparent"], restricted: [] },
},
},
async generateImage(req) {
seenRequest = {
quality: req.quality,
outputFormat: req.outputFormat,
background: req.background,
};
return { images: [{ buffer: Buffer.from("image"), mimeType: "image/webp" }] };
},
};
const result = await generateImage(
{
cfg: {},
modelOverride: `test/${model}`,
prompt: "A sticker",
quality: "max",
outputFormat: "webp",
background: "transparent",
},
{
getProvider: (id) => (id === provider.id ? provider : undefined),
listProviders: () => [provider],
},
);
expect(seenRequest).toEqual({
quality: accepted ? "max" : undefined,
outputFormat: accepted ? "webp" : undefined,
background: accepted ? "transparent" : undefined,
});
expect(result.ignoredOverrides).toEqual(
accepted
? []
: [
{ key: "quality", value: "max" },
{ key: "outputFormat", value: "webp" },
{ key: "background", value: "transparent" },
],
);
});
});
|