94 lines
3.3 KiB
TypeScript
94 lines
3.3 KiB
TypeScript
import { z } from "zod";
|
|
|
|
const textPartSchema = z.object({
|
|
type: z.enum(["text"]),
|
|
text: z.string().min(1).max(2000),
|
|
});
|
|
|
|
const filePartSchema = z.object({
|
|
type: z.enum(["file"]),
|
|
mediaType: z.enum(["image/jpeg", "image/png", "image/webp", "image/gif", "image/jpg", "image/bmp", "image/svg+xml"]),
|
|
name: z.string().min(1).max(100),
|
|
url: z.string().url(),
|
|
assetId: z.string().uuid().optional(),
|
|
});
|
|
|
|
const partSchema = z.union([textPartSchema, filePartSchema]);
|
|
|
|
// 视频参数 schema
|
|
const videoParamsSchema = z.object({
|
|
mode: z.enum(["text-to-video", "first-frame", "first-last-frame", "reference-images"]),
|
|
resolution: z.enum(["480p", "720p", "1080p"]).optional(),
|
|
ratio: z.string().optional(),
|
|
duration: z.number().optional(),
|
|
generateAudio: z.boolean().optional(),
|
|
watermark: z.boolean().optional(),
|
|
cameraFixed: z.boolean().optional(),
|
|
cameraMove: z.string().optional(),
|
|
fps: z.number().optional(),
|
|
seed: z.number().optional(),
|
|
returnLastFrame: z.boolean().optional(),
|
|
}).optional();
|
|
|
|
// 🆕 图片生成参数 schema
|
|
const imageParamsSchema = z.object({
|
|
aspectRatio: z.string().optional(),
|
|
imageSize: z.string().optional(),
|
|
imageCount: z.number().min(1).max(10).optional(),
|
|
// 阿里生图专用参数
|
|
size: z.string().optional(), // 自定义分辨率，如 "1024*1024"
|
|
negativePrompt: z.string().optional(), // 反向提示词
|
|
promptExtend: z.boolean().optional(), // 提示词智能优化
|
|
watermark: z.boolean().optional(), // 水印
|
|
seed: z.number().optional(), // 随机种子
|
|
n: z.number().min(1).max(10).optional(), // 兼容 n 参数
|
|
}).optional();
|
|
|
|
const contextPayloadSchema = z
|
|
.object({
|
|
source: z.enum(["chat", "canvas", "realtime-canvas"]).optional(),
|
|
projectId: z.string().uuid().optional(),
|
|
selectedNodeIds: z.array(z.string().min(1).max(128)).max(20).optional(),
|
|
selectedAssetIds: z.array(z.string().uuid()).max(20).optional(),
|
|
focusPolicy: z.enum(["auto", "selected", "latest_result", "original"]).optional(),
|
|
canvasImageNodes: z
|
|
.array(
|
|
z.object({
|
|
nodeId: z.string().min(1).max(128),
|
|
assetId: z.string().uuid().optional(),
|
|
url: z.string().url(),
|
|
title: z.string().max(256).optional(),
|
|
prompt: z.string().max(2000).optional(),
|
|
x: z.number(),
|
|
y: z.number(),
|
|
width: z.number().optional(),
|
|
height: z.number().optional(),
|
|
isSelected: z.boolean().optional(),
|
|
}),
|
|
)
|
|
.max(20)
|
|
.optional(),
|
|
})
|
|
.optional();
|
|
|
|
export const postRequestBodySchema = z.object({
|
|
id: z.string().uuid(),
|
|
message: z.object({
|
|
id: z.string().uuid(),
|
|
role: z.enum(["user"]),
|
|
parts: z.array(partSchema),
|
|
}),
|
|
hiddenPrompt: z.string().max(4000).optional(),
|
|
autoContext: z.string().max(4000).optional(),
|
|
agentPresetId: z.string().uuid().optional(),
|
|
// 支持预定义模型名称或数据库中的模型 UUID
|
|
selectedChatModel: z.string().min(1).max(128),
|
|
selectedVisibilityType: z.enum(["public", "private"]),
|
|
deepThinkingEnabled: z.boolean().optional(),
|
|
videoParams: videoParamsSchema, // 视频参数
|
|
imageParams: imageParamsSchema, // 🆕 图片生成参数
|
|
contextPayload: contextPayloadSchema,
|
|
});
|
|
|
|
export type PostRequestBody = z.infer<typeof postRequestBodySchema>;
|