38 lines
996 B
TypeScript
38 lines
996 B
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"]),
|
||
|
|
name: z.string().min(1).max(100),
|
||
|
|
url: z.string(),
|
||
|
|
});
|
||
|
|
|
||
|
|
const partSchema = z.union([textPartSchema, filePartSchema]);
|
||
|
|
|
||
|
|
const userMessageSchema = z.object({
|
||
|
|
id: z.string().uuid(),
|
||
|
|
role: z.enum(["user"]),
|
||
|
|
parts: z.array(partSchema),
|
||
|
|
});
|
||
|
|
|
||
|
|
const toolApprovalMessageSchema = z.object({
|
||
|
|
id: z.string(),
|
||
|
|
role: z.enum(["user", "assistant"]),
|
||
|
|
parts: z.array(z.record(z.unknown())),
|
||
|
|
});
|
||
|
|
|
||
|
|
export const postRequestBodySchema = z.object({
|
||
|
|
id: z.string().uuid(),
|
||
|
|
message: userMessageSchema.optional(),
|
||
|
|
messages: z.array(toolApprovalMessageSchema).optional(),
|
||
|
|
selectedChatModel: z.string(),
|
||
|
|
selectedVisibilityType: z.enum(["public", "private"]),
|
||
|
|
});
|
||
|
|
|
||
|
|
export type PostRequestBody = z.infer<typeof postRequestBodySchema>;
|