fix: classify reference image models correctly
This commit is contained in:
parent
f52ab780f1
commit
452effecbc
@ -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([
|
||||
|
||||
@ -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,
|
||||
|
||||
35
tests/routes/model-classification.test.ts
Normal file
35
tests/routes/model-classification.test.ts
Normal 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);
|
||||
});
|
||||
});
|
||||
Loading…
x
Reference in New Issue
Block a user