84 lines
2.7 KiB
TypeScript
84 lines
2.7 KiB
TypeScript
import { expect, test } from "@playwright/test";
|
||
import { buildReferenceImageOrderPrompt } from "@/lib/ai/providers/reference-image-order";
|
||
import { collectMessageImageAttachments } from "@/lib/chat/message-image-attachments";
|
||
|
||
test.describe("image message contracts", () => {
|
||
test("normalizes every supported image part shape for message rendering", () => {
|
||
const attachments = collectMessageImageAttachments({
|
||
parts: [
|
||
{
|
||
type: "file",
|
||
name: "图一",
|
||
mediaType: "image/png",
|
||
url: "https://cdn.example.com/one.png",
|
||
},
|
||
{
|
||
type: "image",
|
||
name: "图二",
|
||
image: "https://cdn.example.com/two.jpg",
|
||
},
|
||
{
|
||
type: "image_url",
|
||
name: "图三",
|
||
image_url: { url: "https://cdn.example.com/three.webp" },
|
||
},
|
||
{
|
||
type: "input_image",
|
||
data: "https://cdn.example.com/four.png",
|
||
},
|
||
{
|
||
type: "file",
|
||
mediaType: "video/mp4",
|
||
url: "https://cdn.example.com/video.mp4",
|
||
},
|
||
],
|
||
attachments: [
|
||
{
|
||
name: "重复图一",
|
||
contentType: "image/png",
|
||
url: "https://cdn.example.com/one.png",
|
||
},
|
||
{
|
||
name: "历史附件",
|
||
contentType: "image/jpeg",
|
||
url: "https://cdn.example.com/five.jpg",
|
||
},
|
||
],
|
||
});
|
||
|
||
expect(attachments.map((attachment) => attachment.url)).toEqual([
|
||
"https://cdn.example.com/one.png",
|
||
"https://cdn.example.com/two.jpg",
|
||
"https://cdn.example.com/three.webp",
|
||
"https://cdn.example.com/four.png",
|
||
"https://cdn.example.com/five.jpg",
|
||
]);
|
||
expect(attachments.map((attachment) => attachment.name)).toEqual([
|
||
"图一",
|
||
"图二",
|
||
"图三",
|
||
"图片4",
|
||
"历史附件",
|
||
]);
|
||
});
|
||
|
||
test("adds explicit ordinal guidance only for multi-image edits", () => {
|
||
const prompt = "把图二的横杆加到图一的桌子上，其他不变";
|
||
|
||
expect(
|
||
buildReferenceImageOrderPrompt(prompt, ["https://cdn.example.com/one.png"])
|
||
).toBe(prompt);
|
||
|
||
const multiImagePrompt = buildReferenceImageOrderPrompt(prompt, [
|
||
"https://cdn.example.com/one.png",
|
||
"https://cdn.example.com/two.png",
|
||
]);
|
||
expect(multiImagePrompt).toContain("本次请求包含 2 张参考图");
|
||
expect(multiImagePrompt).toContain("不得忽略后续图片");
|
||
expect(multiImagePrompt).toContain("第二张图/图2/图二");
|
||
expect(multiImagePrompt).toContain("图A是对象素材来源，图B是目标底图");
|
||
expect(multiImagePrompt).toContain("不要输出拼图或并排对比图");
|
||
expect(multiImagePrompt.endsWith(prompt)).toBe(true);
|
||
});
|
||
});
|