test: preserve image model classification coverage

This commit is contained in:
134 2026-07-21 11:59:11 +08:00
parent 42a42e496d
commit ebc0680903
2 changed files with 28 additions and 1 deletions

View File

@ -57,7 +57,6 @@ const VIDEO_MODE_KEYS = new Set([
"audio-to-video",
"first-frame",
"first-last-frame",
"reference-images",
]);
const VIDEO_KIND_KEYS = new Set([

View File

@ -18,6 +18,34 @@ const atlasHunyuan3D = {
};
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", () => {
expect(inferAtlasCloudMediaKind(atlasHunyuan3D)).toBe("3d");