| 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" }, |
| ], |
| ); |
| }); |
| }); |
|
|
