test: preserve image model classification coverage
This commit is contained in:
parent
42a42e496d
commit
ebc0680903
@ -57,7 +57,6 @@ const VIDEO_MODE_KEYS = new Set([
|
|||||||
"audio-to-video",
|
"audio-to-video",
|
||||||
"first-frame",
|
"first-frame",
|
||||||
"first-last-frame",
|
"first-last-frame",
|
||||||
"reference-images",
|
|
||||||
]);
|
]);
|
||||||
|
|
||||||
const VIDEO_KIND_KEYS = new Set([
|
const VIDEO_KIND_KEYS = new Set([
|
||||||
|
|||||||
@ -18,6 +18,34 @@ const atlasHunyuan3D = {
|
|||||||
};
|
};
|
||||||
|
|
||||||
test.describe("AI model media classification", () => {
|
test.describe("AI model media classification", () => {
|
||||||
|
test("keeps multi-reference BananaImage models classified as image models", () => {
|
||||||
|
const descriptor = {
|
||||||
|
providerType: "bananaimage",
|
||||||
|
modelId: "gemini-3-pro-image-preview",
|
||||||
|
capabilities: {
|
||||||
|
supportedModes: [
|
||||||
|
"text-to-image",
|
||||||
|
"image-to-image",
|
||||||
|
"reference-images",
|
||||||
|
],
|
||||||
|
},
|
||||||
|
supportsImageGen: true,
|
||||||
|
};
|
||||||
|
|
||||||
|
expect(isVideoModelDescriptor(descriptor)).toBe(false);
|
||||||
|
expect(isImageGenerationModelDescriptor(descriptor)).toBe(true);
|
||||||
|
});
|
||||||
|
|
||||||
|
test("still recognizes explicit video generation modes", () => {
|
||||||
|
expect(
|
||||||
|
isVideoModelDescriptor({
|
||||||
|
providerType: "custom",
|
||||||
|
modelId: "custom-video-model",
|
||||||
|
capabilities: { supportedModes: ["text-to-video"] },
|
||||||
|
})
|
||||||
|
).toBe(true);
|
||||||
|
});
|
||||||
|
|
||||||
test("keeps Atlas 3D entries out of image and video categories", () => {
|
test("keeps Atlas 3D entries out of image and video categories", () => {
|
||||||
expect(inferAtlasCloudMediaKind(atlasHunyuan3D)).toBe("3d");
|
expect(inferAtlasCloudMediaKind(atlasHunyuan3D)).toBe("3d");
|
||||||
|
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user