fix: classify reference image models correctly

This commit is contained in:
134 2026-07-14 15:26:46 +08:00
parent f52ab780f1
commit 452effecbc
3 changed files with 36 additions and 1 deletions

View File

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

View File

@ -11,6 +11,7 @@ export default defineConfig({
/routes\/canvas-entry-performance\.test\.ts/,
/routes\/settings-visual-consistency\.test\.ts/,
/routes\/canvas-scale-adjust-intent\.test\.ts/,
/routes\/model-classification\.test\.ts/,
],
fullyParallel: true,
forbidOnly: !!process.env.CI,

View File

@ -0,0 +1,35 @@
import { expect, test } from "@playwright/test";
import {
isImageGenerationModelDescriptor,
isVideoModelDescriptor,
} from "@/lib/ai/model-classification";
test.describe("model 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);
});
});