| 1 | import { createZodDto } from '@yikart/common' |
| 2 | import { z } from 'zod' |
| 3 | |
| 4 | export enum DraftGenerationMemoryContentType { |
| 5 | Video = 'video', |
| 6 | ImageText = 'image-text', |
| 7 | } |
| 8 | |
| 9 | export const ListDraftGenerationMemoryDtoSchema = z.object({ |
| 10 | contentType: z.enum(DraftGenerationMemoryContentType).optional().describe('Memory 类型：video 视频草稿，image-text 图文草稿'), |
| 11 | }) |
| 12 | export class ListDraftGenerationMemoryDto extends createZodDto(ListDraftGenerationMemoryDtoSchema, 'ListDraftGenerationMemoryDto') {} |
| 13 | |
| 14 | export const RegenerateDraftGenerationMemoryDtoSchema = z.object({ |
| 15 | contentType: z.enum(DraftGenerationMemoryContentType).describe('Memory 类型：video 视频草稿，image-text 图文草稿'), |
| 16 | plannerModel: z.string().optional().describe('草稿规划模型名称，不传时使用默认规划模型'), |
| 17 | }) |
| 18 | export class RegenerateDraftGenerationMemoryDto extends createZodDto(RegenerateDraftGenerationMemoryDtoSchema, 'RegenerateDraftGenerationMemoryDto') {} |
| 19 |