@purpleschool/gptbot 0.7.58-presentations → 0.7.59-presentations

package/constants/errors/errors.ts

@@ -2014,13 +2014,291 @@ export const ERRORS = {
         message: 'не удалось получить статистику пользователя',
         httpCode: 500,
     },
-    SLIDE_IMAGE_SLOT_UPDATE_ERROR: {
+    TOOL_GET_FORMATTED_ERROR: {
         code: 'A412',
+        message: 'Ошибка при получении отформатированных тулов',
+        httpCode: 500,
+    },
+    TOOL_FIND_ALL_ERROR: {
+        code: 'A413',
+        message: 'Ошибка при получении всех тулов',
+        httpCode: 500,
+    },
+    VIDEO_EDITOR_CONFIG_ERROR: {
+        code: 'A414',
+        message: 'Произошла ошибка при получении конфигурации инструмента для редактирования видео',
+        httpCode: 500,
+    },
+    VIDEO_EDITOR_SEND_REQUEST_ERROR: {
+        code: 'A415',
+        message: 'Произошла ошибка при отправке запроса на редактирование видео',
+        httpCode: 500,
+    },
+    VIDEO_EDITOR_FIND_JOBS_ERROR: {
+        code: 'A416',
+        message: 'Произошла ошибка при получении списка заданий на редактирование видео',
+        httpCode: 500,
+    },
+    VIDEO_EDITOR_FIND_JOB_ERROR: {
+        code: 'A417',
+        message: 'Произошла ошибка при получении задания на редактирование видео',
+        httpCode: 500,
+    },
+    VIDEO_EDITOR_RETRY_JOB_ERROR: {
+        code: 'A418',
+        message: 'Произошла ошибка при повторном запросе на редактирование видео',
+        httpCode: 500,
+    },
+    VIDEO_EDITOR_SET_REACTION_ERROR: {
+        code: 'A419',
+        message: 'Произошла ошибка при установке оценки задания на редактирование видео',
+        httpCode: 500,
+    },
+    VIDEO_EDITOR_UPDATE_TITLE_ERROR: {
+        code: 'A420',
+        message: 'Произошла ошибка при обновлении названия задания на редактирование видео',
+        httpCode: 500,
+    },
+    VIDEO_EDITOR_DELETE_JOB_ERROR: {
+        code: 'A421',
+        message: 'Произошла ошибка при удалении задания на редактирование видео',
+        httpCode: 500,
+    },
+    VIDEO_EDITOR_DELETE_ALL_JOBS_ERROR: {
+        code: 'A422',
+        message: 'Произошла ошибка при удалении всех заданий на редактирование видео',
+        httpCode: 500,
+    },
+    VIDEO_EDITOR_MODEL_NOT_FOUND: {
+        code: 'A423',
+        message: 'Не удалось найти указанную модель для редактирования видео',
+        httpCode: 400,
+    },
+    VIDEO_EDITOR_MODEL_INACTIVE: {
+        code: 'A424',
+        message: 'Модель для редактирования видео неактивна',
+        httpCode: 400,
+    },
+    VIDEO_EDITOR_MODEL_NOT_AVAILABLE: {
+        code: 'A425',
+        message: 'Модель для редактирования видео недоступна для данного пользователя',
+        httpCode: 403,
+    },
+    VIDEO_EDITOR_FILE_SIZE_EXCEEDED: {
+        code: 'A426',
+        message: 'Размер файла превышает допустимый лимит для выбранной модели',
+        httpCode: 400,
+    },
+    VIDEO_EDITOR_UNSUPPORTED_FORMAT: {
+        code: 'A427',
+        message: 'Формат файла не поддерживается выбранной моделью',
+        httpCode: 400,
+    },
+    FILE_ACCESS_DENIED: {
+        code: 'A428',
+        message: 'Доступ к файлу запрещен',
+        httpCode: 403,
+    },
+    VIDEO_EDITOR_FILE_DURATION_EXCEEDED: {
+        code: 'A429',
+        message: 'Продолжительность файла превышает допустимый лимит для выбранной модели',
+        httpCode: 400,
+    },
+    WEBMASTER_CREATE_ERROR: {
+        code: 'A430',
+        message: 'Ошибка при создании вебмастера',
+        httpCode: 500,
+    },
+    WEBMASTER_FIND_ERROR: {
+        code: 'A431',
+        message: 'Ошибка при получении вебмастера',
+        httpCode: 500,
+    },
+    WEBMASTER_UPDATE_ERROR: {
+        code: 'A432',
+        message: 'Ошибка при обновлении вебмастера',
+        httpCode: 500,
+    },
+    WEBMASTER_ALIAS_CONFLICT_ERROR: {
+        code: 'A433',
+        message: 'Выбранный alias уже занят',
+        httpCode: 409,
+    },
+    WEBMASTER_NOT_FOUND_ERROR: {
+        code: 'A434',
+        message: 'Запрашиваемый вебмастер не найден',
+        httpCode: 404,
+    },
+    WEBMASTER_OWNERSHIP_ERROR: {
+        code: 'A435',
+        message: 'Кабинет вебмастер не принадлежит юзеру, сделавшему запрос',
+        httpCode: 401,
+    },
+    WEBMASTER_BALANCE_CREATE_ERROR: {
+        code: 'A436',
+        message: 'Ошибка при создании записи баланса вебмастера',
+        httpCode: 500,
+    },
+    WEBMASTER_BALANCE_GET_ERROR: {
+        code: 'A437',
+        message: 'Ошибка при получении баланса вебмастера',
+        httpCode: 500,
+    },
+    WEBMASTER_BALANCE_GET_OVERVIEW_ERROR: {
+        code: 'A438',
+        message: 'Не удалось получить историю баланса вебмастера',
+        httpCode: 500,
+    },
+    WEBMASTER_INSUFFICIENT_BALANCE_ERROR: {
+        code: 'A439',
+        message: 'На балансе недостатчно средств для вывода',
+        httpCode: 400,
+    },
+    WEBMASTER_SEND_WITHDRAWAL_REQUEST_ERROR: {
+        code: 'A440',
+        message: 'Произошла ошибка при отправке запроса на вывод средств в Telegram для вебмастера',
+        httpCode: 500,
+    },
+    WEBMASTER_BALANCE_UPDATE_ERROR: {
+        code: 'A441',
+        message: 'Ошибка при обновлении записи баланса вебмастера',
+        httpCode: 500,
+    },
+    WEBMASTER_PROMOCODE_CONFLICT_ERROR: {
+        code: 'A442',
+        message: 'Вебмастер не может создать промокод для самого себя',
+        httpCode: 409,
+    },
+    WEBMASTER_CREATE_WITHDRAWAL_REQUEST_ERROR: {
+        code: 'A443',
+        message: 'Произошла ошибка при создании запроса на вывод средств для вебмастера',
+        httpCode: 500,
+    },
+    TELEGRAM_SEND_TO_CHANNEL_ERROR: {
+        code: 'A444',
+        message: 'Произошла ошибка при отправке сообщения в телеграм канал',
+        httpCode: 500,
+    },
+    WRITER_SET_REACTION_TO_WRITER_DOCUMENT_ERROR: {
+        code: 'A445',
+        message: 'Произошла ошибка при установке реакции на документ',
+        httpCode: 500,
+    },
+    MUSIC_CONFIG_ERROR: {
+        code: 'A446',
+        message: 'Произошла ошибка при получении конфигурации инструмента для генерации музыки',
+        httpCode: 500,
+    },
+    MUSIC_SEND_REQUEST_ERROR: {
+        code: 'A447',
+        message: 'Произошла ошибка при отправке запроса на генерацию музыки',
+        httpCode: 500,
+    },
+    MUSIC_FIND_JOBS_ERROR: {
+        code: 'A448',
+        message: 'Произошла ошибка при получении списка заданий на генерацию музыки',
+        httpCode: 500,
+    },
+    MUSIC_FIND_JOB_ERROR: {
+        code: 'A449',
+        message: 'Произошла ошибка при получении задания на генерацию музыки',
+        httpCode: 500,
+    },
+    MUSIC_RETRY_JOB_ERROR: {
+        code: 'A450',
+        message: 'Произошла ошибка при повторном запросе на генерацию музыки',
+        httpCode: 500,
+    },
+    MUSIC_SET_REACTION_ERROR: {
+        code: 'A451',
+        message: 'Произошла ошибка при установке оценки задания на генерацию музыки',
+        httpCode: 500,
+    },
+    MUSIC_UPDATE_TITLE_ERROR: {
+        code: 'A452',
+        message: 'Произошла ошибка при обновлении названия задания на генерацию музыки',
+        httpCode: 500,
+    },
+    MUSIC_DELETE_JOB_ERROR: {
+        code: 'A453',
+        message: 'Произошла ошибка при удалении задания на генерацию музыки',
+        httpCode: 500,
+    },
+    MUSIC_DELETE_ALL_JOBS_ERROR: {
+        code: 'A454',
+        message: 'Произошла ошибка при удалении всех заданий на генерацию музыки',
+        httpCode: 500,
+    },
+    MUSIC_MODEL_NOT_FOUND: {
+        code: 'A455',
+        message: 'Не удалось найти указанную модель для генерации музыки',
+        httpCode: 400,
+    },
+    MUSIC_MODEL_INACTIVE: {
+        code: 'A456',
+        message: 'Модель для генерации музыки неактивна',
+        httpCode: 400,
+    },
+    MUSIC_MODEL_NOT_AVAILABLE: {
+        code: 'A457',
+        message: 'Модель для генерации музыки недоступна для данного пользователя',
+        httpCode: 403,
+    },
+    MUSIC_MAX_SYMBOLS_EXCEEDED: {
+        code: 'A458',
+        message: 'Превышена максимальная количество символов',
+    },
+    MUSIC_INVALID_VOCAL: {
+        code: 'A459',
+        message: 'Выбран неверный голос для вокала',
+    },
+    MUSIC_JOB_DELETE_ERROR: {
+        code: 'A460',
+        message: 'Произошла ошибка при удалении сообщения',
+        httpCode: 500,
+    },
+    MUSIC_RETRY_REQUEST_ERROR: {
+        code: 'A461',
+        message: 'Произошла ошибка при повторном запросе на генерацию музыки',
+        httpCode: 500,
+    },
+    MUSIC_LYRICS_GENERATION_ERROR: {
+        code: 'A462',
+        message: 'Произошла ошибка при запросе на генерацию текста песни',
+        httpCode: 500,
+    },
+    MUSIC_CONVERT_TO_WAV_ERROR: {
+        code: 'A463',
+        message: 'Произошла ошибка при запросе на конвертацию mp3 в wav',
+        httpCode: 500,
+    },
+    WRITER_MODEL_NOT_FOUND: {
+        code: 'A464',
+        message: 'Не удалось найти указанную модель для Writer',
+        httpCode: 400,
+    },
+    WRITER_MODEL_INACTIVE: {
+        code: 'A465',
+        message: 'Модель Writer архивирована и недоступна для пользования',
+        httpCode: 403,
+    },
+    WRITER_MODEL_ACCESS_DENIED: {
+        code: 'A466',
+        message: 'У вас нет доступа к этой модели Writer',
+        httpCode: 403,
+    },
+    WEBMASTER_BONUS_APPLY_ERROR: {
+        code: 'A449',
+        message: 'Не удалось применить бонусы вебмастеру',
+        httpCode: 500,
+    },
+    SLIDE_IMAGE_SLOT_UPDATE_ERROR: {
+        code: 'A450',
         message: 'Произошла ошибка при обновлении изображения на слайде',
         httpCode: 500,
     },
     FILE_TYPE_NOT_SUPPORTED: {
-        code: 'A413',
+        code: 'A451',
         message: 'Неподдерживаемый тип файла',
         httpCode: 400,
     },

package/constants/file/enums/file-type.enum.ts

@@ -3,5 +3,6 @@ export enum FILE_TYPE {
     AUDIO = 'audio',
     TEXT = 'text',
     DOCUMENT = 'document',
+    VIDEO = 'video',
     OTHER = 'other',
 }

package/constants/file/file.constants.ts

@@ -4,6 +4,7 @@ export const MAX_IMAGE_UPLOAD_SIZE = 5 * 1024 * 1024; //  5 MB
 export const MAX_AUDIO_UPLOAD_SIZE = 500 * 1024 * 1024; // 500 MB
 export const MAX_TEXT_UPLOAD_SIZE = 10 * 1024 * 1024; // 10 MB
 export const MAX_DOC_UPLOAD_SIZE = 20 * 1024 * 1024; // 20 MB
+export const MAX_VIDEO_UPLOAD_SIZE = 500 * 1024 * 1024; // 500 MB
 
 export const SUPPORTED_FILES = new Map<
     string,
@@ -60,6 +61,9 @@ export const SUPPORTED_FILES = new Map<
     ], //.xlsx
 
     ['application/pdf', { type: FILE_TYPE.DOCUMENT, size: MAX_DOC_UPLOAD_SIZE }],
+
+    // Video formats
+    ['video/mp4', { type: FILE_TYPE.VIDEO, size: MAX_VIDEO_UPLOAD_SIZE }],
 ]);
 
 export const DELETE_UNUSED_FILES_AFTER_HOURS = 12;

package/constants/index.ts

@@ -31,7 +31,12 @@ export * from './promocode';
 export * from './ui-notification';
 export * from './review';
 export * from './tool-video';
+export * from './tool-video-editor';
 export * from './writer';
 export * from './tool-image-editor';
 export * from './feedback';
 export * from './daily-streak';
+export * from './cabinet';
+export * from './webmaster';
+export * from './webmaster-balance';
+export * from './tool-music';

package/constants/subscription/enums/subscription-feature-type.enum.ts

@@ -14,6 +14,8 @@ export enum SUBSCRIPTION_FEATURE_TYPE {
     VIDEO_MODEL_ACCESS = 'video_model_access',
     EXTENDED_CONTEXT = 'extended_context',
     PRESENTATION_QUOTA = 'presentation_quota',
+    TOKENS = 'tokens',
+    CARRYOVER_BALANCE = 'carryover_balance',
 }
 
 export enum SUBSCRIPTION_FEATURE_KIND {
@@ -25,4 +27,5 @@ export enum SUBSCRIPTION_FEATURE_KIND {
     PRESENTATIONS = 'presentations',
     VIDEO = 'video',
     PRESENTATION = 'presentation',
+    BASE = 'base',
 }

package/constants/tool-music/enums/index.ts

@@ -0,0 +1,5 @@
+export * from './suno-model-type.enum';
+export * from './music-strategy.enum';
+export * from './music-vocal-gender.enum';
+export * from './music-pricing-rule-type.enum';
+export * from './suno-mode-type.enum';

package/constants/tool-music/enums/music-pricing-rule-type.enum.ts

@@ -0,0 +1,3 @@
+export enum MUSIC_PRICING_RULE_TYPE {
+    FLAT = 'flat',
+}

package/constants/tool-music/enums/music-strategy.enum.ts

@@ -0,0 +1,3 @@
+export enum MUSIC_GENERATION_STRATEGY {
+    SUNO_POLLING = 'SUNO_POLLING',
+}

package/constants/tool-music/enums/music-vocal-gender.enum.ts

@@ -0,0 +1,4 @@
+export enum MUSIC_VOCAL_GENDER {
+    MALE = 'm',
+    FEMALE = 'f',
+}

package/constants/tool-music/enums/suno-mode-type.enum.ts

@@ -0,0 +1,6 @@
+export enum SUNO_MODE_TYPE {
+    SIMPLE = 'simple',
+    SIMPLE_INSTRUMENTAL = 'simple_instrumental',
+    CUSTOM = 'custom',
+    CUSTOM_INSTRUMENTAL = 'custom_instrumental',
+}

package/constants/tool-music/enums/suno-model-type.enum.ts

@@ -0,0 +1,6 @@
+export enum SUNO_MODEL_TYPE {
+    V4 = 'V4',
+    V4_5 = 'V4_5',
+    V4_5PLUS = 'V4_5PLUS',
+    V5 = 'V5',
+}

package/constants/tool-music/index.ts

@@ -0,0 +1 @@
+export * from './enums';

package/constants/tool-video-editor/enums/index.ts

@@ -0,0 +1 @@
+export * from './video-editor-model-strategy.enum';

package/constants/tool-video-editor/enums/video-editor-model-strategy.enum.ts

@@ -0,0 +1,3 @@
+export enum VIDEO_EDITOR_MODEL_STRATEGY {
+    LUMA_MODIFY_POLLING = 'LUMA_MODIFY_POLLING',
+}

package/constants/tool-video-editor/index.ts

@@ -0,0 +1 @@
+export * from './enums';

package/constants/transaction/enums/user-balance-status.enum.ts

@@ -1,6 +1,7 @@
 export enum USER_BALANCE_STATUS {
     bonus_balance = 'bonus_balance',
     subscription_balance = 'subscription_balance',
+    carried_over_balance = 'carried_over_balance',
     product_balance = 'product_balance',
     referral_bonus_balance = 'referral_bonus_balance',
 }

package/constants/webmaster/enums/index.ts

@@ -0,0 +1 @@
+export * from './organization-type.enum';

package/constants/webmaster/enums/organization-type.enum.ts

@@ -0,0 +1,5 @@
+export enum ORGANIZATION_TYPE {
+    INDIVIDUAL_ENTREPRENEUR = 'ИП',
+    LEGAL_ENTITY = 'Юридическое лицо',
+    SELF_EMPLOYED = 'Самозанятый',
+}

package/constants/webmaster/index.ts

@@ -0,0 +1 @@
+export * from './enums';

package/constants/webmaster-balance/enums/index.ts

@@ -0,0 +1,2 @@
+export * from './webmaster-balance-type.enum';
+export * from './webmaster-balance-status.enum';

package/constants/webmaster-balance/enums/webmaster-balance-status.enum.ts

@@ -0,0 +1,6 @@
+export enum WEBMASTER_BALANCE_STATUS {
+    SUCCESS = 'success',
+    AWAITING_CONFIRMATION = 'awaiting_confirmation',
+    CANCELLED = 'cancelled',
+    TELEGRAM_ERROR = 'telegram_error',
+}

package/constants/webmaster-balance/enums/webmaster-balance-type.enum.ts

@@ -0,0 +1,4 @@
+export enum WEBMASTER_BALANCE_TYPE {
+    DEPOSIT = 'deposit',
+    WITHDRAWAL = 'withdrawal',
+}

package/constants/webmaster-balance/index.ts

@@ -0,0 +1 @@
+export * from './enums';

package/constants/writer/enums/index.ts

@@ -2,3 +2,4 @@ export * from './writer-document-section-type.enum';
 export * from './writer-document-stage.enum';
 export * from './writer-ai-action-pricing-type.enum';
 export * from './writer-ai-action-type.enum';
+export * from './tool-model-status.enum';

package/constants/writer/enums/tool-model-status.enum.ts

@@ -0,0 +1,4 @@
+export enum TOOL_MODEL_STATUS {
+    ACTIVE = 'active',
+    INACTIVE = 'inactive',
+}

package/helpers/index.ts

@@ -4,4 +4,5 @@ export * from './stt';
 export * from './subscription';
 export * from './tts';
 export * from './video';
+export * from './video-editor';
 export * from './writer';

package/helpers/video-editor/calculate-video-editing-price.helper.ts

@@ -0,0 +1,9 @@
+export function calculateVideoEditingPrice({
+    duration,
+    pricePerSecond,
+}: {
+    duration: number;
+    pricePerSecond: number;
+}): number {
+    return Math.ceil(duration * pricePerSecond);
+}

package/helpers/video-editor/index.ts

@@ -0,0 +1 @@
+export * from './calculate-video-editing-price.helper';

package/helpers/writer/calculate-writer-content-generation-price.util.ts

@@ -1,3 +1,5 @@
-export function calculateWriterContentGenerationPrice(pages: number): number {
-    return Math.max(1, pages);
+import { WriterModel } from '../../models/tools/writer/writer-model.schema';
+
+export function calculateWriterContentGenerationPrice(pages: number, aiModel: WriterModel): number {
+    return Math.max(1, pages * aiModel.pricePerPage);
 }

package/models/index.ts

@@ -50,3 +50,6 @@ export * from './folder.schema';
 export * from './filters.schema';
 export * from './users-filter-notification.schema';
 export * from './daily-streak';
+export * from './webmaster.schema';
+export * from './webmaster-balance.schema';
+export * from './user-referrals.schema';

package/models/page.schema.ts

@@ -12,6 +12,8 @@ export const PageSchema = z.object({
     categoryId: z.string().uuid().nullable().optional(),
     aIModelId: z.string().uuid().nullable().optional(),
     toolId: z.string().uuid().nullable().optional(),
+    toolOptionId: z.string().uuid().nullable().optional(),
+    toolTitle: z.string().nullable(),
     type: z.nativeEnum(PageType),
 
     createdAt: z.date(),

package/models/subscription-feature.schema.ts

@@ -12,6 +12,34 @@ export const SubscriptionFeatureBaseSchema = z.object({
 });
 export type SubscriptionFeatureBase = z.infer<typeof SubscriptionFeatureBaseSchema>;
 
+const SubscriptionFeatureIconsSchema = z.object({
+    light: z.object({
+        png: z.string(),
+        svg: z.string(),
+    }),
+    dark: z.object({
+        png: z.string(),
+        svg: z.string(),
+    }),
+});
+
+export const TokensFeatureSchema = SubscriptionFeatureBaseSchema.extend({
+    kind: z.literal(SUBSCRIPTION_FEATURE_KIND.BASE),
+    type: z.literal(SUBSCRIPTION_FEATURE_TYPE.TOKENS),
+    tokens: z.number(),
+    icons: SubscriptionFeatureIconsSchema,
+});
+export type TokensFeature = z.infer<typeof TokensFeatureSchema>;
+
+export const CarryoverBalanceFeatureSchema = SubscriptionFeatureBaseSchema.extend({
+    kind: z.literal(SUBSCRIPTION_FEATURE_KIND.BASE),
+    type: z.literal(SUBSCRIPTION_FEATURE_TYPE.CARRYOVER_BALANCE),
+    isAvailable: z.boolean(),
+    tooltip: z.array(z.string()),
+    icons: SubscriptionFeatureIconsSchema,
+});
+export type CarryoverBalanceFeature = z.infer<typeof CarryoverBalanceFeatureSchema>;
+
 export const AiModelAccessFeatureSchema = SubscriptionFeatureBaseSchema.extend({
     type: z.literal(SUBSCRIPTION_FEATURE_TYPE.AI_MODEL_ACCESS),
     kind: z.literal(SUBSCRIPTION_FEATURE_KIND.AI_MODEL),
@@ -134,6 +162,8 @@ export const SubscriptionFeatureSchema = z.union([
     VideoModelAccessFeatureSchema,
     ExtendedContextFeatureSchema,
     PresentationQuotaFeatureSchema,
+    TokensFeatureSchema,
+    CarryoverBalanceFeatureSchema,
 ]);
 
 export type SubscriptionFeature = z.infer<typeof SubscriptionFeatureSchema>;

package/models/subscription.schema.ts

@@ -28,6 +28,7 @@ export const SubscriptionSchema = z.object({
     features: z.array(SubscriptionFeatureSchema),
     maxInputLength: z.number(),
     maxInputTokens: z.number(),
+    carryoverPercent: z.number(),
     createdAt: z.date(),
     updatedAt: z.date(),
 });

package/models/tool.schema.ts

@@ -1,4 +1,5 @@
 import { z } from 'zod';
+import { PageSchema } from './page.schema';
 
 export const ToolSchema = z.object({
     uuid: z.string().uuid(),
@@ -6,4 +7,16 @@ export const ToolSchema = z.object({
     description: z.string(),
     icon: z.string(),
     order: z.number(),
+    contentType: z.string(),
 });
+
+export const ToolFormattedSchema = z.intersection(
+    ToolSchema,
+    z.object({
+        pages: z.array(
+            PageSchema.pick({ alias: true, toolTitle: true, toolOptionId: true })
+                .optional()
+                .nullable(),
+        ),
+    }),
+);

package/models/tools/index.ts

@@ -6,3 +6,5 @@ export * from './tts';
 export * from './video';
 export * from './writer';
 export * from './image-editor';
+export * from './video-editor';
+export * from './music';

package/models/tools/music/index.ts

@@ -0,0 +1,4 @@
+export * from './music-config.schema';
+export * from './music-job.schema';
+export * from './music-model.schema';
+export * from './music-track.schema';

package/models/tools/music/music-config.schema.ts

@@ -0,0 +1,6 @@
+import { z } from 'zod';
+import { MusicModelSchema } from './music-model.schema';
+
+export const MusicConfigSchema = z.object({
+    models: z.array(MusicModelSchema),
+});

package/models/tools/music/music-job.schema.ts

@@ -0,0 +1,43 @@
+import { z } from 'zod';
+import { ToolJobSchema } from '../../tool-job.schema';
+import { MUSIC_VOCAL_GENDER, SUNO_MODE_TYPE, USER_REACTION } from '../../../constants';
+import { MusicTrackSchema } from './music-track.schema';
+
+export const MusicJobParamsSchema = z.discriminatedUnion('mode', [
+    z.object({
+        mode: z.literal(SUNO_MODE_TYPE.SIMPLE),
+        prompt: z.string().min(1).max(450),
+        style: z.string(),
+        vocalGender: z.nativeEnum(MUSIC_VOCAL_GENDER),
+    }),
+    z.object({
+        mode: z.literal(SUNO_MODE_TYPE.SIMPLE_INSTRUMENTAL),
+        prompt: z.string().min(1).max(450),
+        style: z.string(),
+    }),
+    z.object({
+        mode: z.literal(SUNO_MODE_TYPE.CUSTOM),
+        prompt: z.string().min(1).max(5000),
+        title: z.string().min(1).max(80),
+        style: z.string().min(1).max(1000),
+        vocalGender: z.nativeEnum(MUSIC_VOCAL_GENDER),
+    }),
+    z.object({
+        mode: z.literal(SUNO_MODE_TYPE.CUSTOM_INSTRUMENTAL),
+        title: z.string().min(1).max(80),
+        style: z.string().min(1).max(1000),
+    }),
+]);
+
+export type MusicJobParams = z.infer<typeof MusicJobParamsSchema>;
+
+export const MusicJobSchema = ToolJobSchema.extend({
+    title: z.string(),
+    prompt: z.string(),
+    reaction: z.nativeEnum(USER_REACTION).nullable(),
+    tracks: z.array(MusicTrackSchema),
+    params: MusicJobParamsSchema,
+    modelId: z.string(),
+});
+
+export type MusicJob = z.infer<typeof MusicJobSchema>;