@purpleschool/gptbot-tools 0.0.87 → 0.0.89

package/ai-vendor/index.ts

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

package/ai-vendor/models/ai-vendor.schema.ts

@@ -0,0 +1,13 @@
+import { z } from 'zod';
+import { IconVariantsSchema } from '../../common';
+
+export const AiVendorSchema = z.object({
+    uuid: z.string().uuid(),
+    title: z.string(),
+    iconVariants: IconVariantsSchema,
+    order: z.number(),
+    createdAt: z.date(),
+    updatedAt: z.date(),
+});
+
+export type AiVendor = z.infer<typeof AiVendorSchema>;

package/ai-vendor/models/index.ts

@@ -0,0 +1 @@
+export * from './ai-vendor.schema';

package/build/ai-vendor/index.js

@@ -0,0 +1,17 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __exportStar = (this && this.__exportStar) || function(m, exports) {
+    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+__exportStar(require("./models"), exports);

package/build/ai-vendor/models/ai-vendor.schema.js

@@ -0,0 +1,13 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.AiVendorSchema = void 0;
+const zod_1 = require("zod");
+const common_1 = require("../../common");
+exports.AiVendorSchema = zod_1.z.object({
+    uuid: zod_1.z.string().uuid(),
+    title: zod_1.z.string(),
+    iconVariants: common_1.IconVariantsSchema,
+    order: zod_1.z.number(),
+    createdAt: zod_1.z.date(),
+    updatedAt: zod_1.z.date(),
+});

package/build/ai-vendor/models/index.js

@@ -0,0 +1,17 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __exportStar = (this && this.__exportStar) || function(m, exports) {
+    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+__exportStar(require("./ai-vendor.schema"), exports);

package/build/common/errors/errors.js

@@ -1207,6 +1207,111 @@ exports.ERRORS = {
             httpCode: 500,
         },
     },
+    IMAGE_GENERATION: {
+        AI_ERROR: {
+            code: 'IMAGE_GENERATION.AI_ERROR',
+            message: 'Произошла ошибка во время запроса к ИИ',
+            httpCode: 500,
+        },
+        INSUFFICIENT_BALANCE: {
+            code: 'IMAGE_GENERATION.INSUFFICIENT_BALANCE',
+            message: 'У пользователя недостаточно средств',
+            httpCode: 400,
+        },
+        INVALID_PARAMS: {
+            code: 'IMAGE_GENERATION.INVALID_PARAMS',
+            message: 'Некорректные параметры изображения',
+            httpCode: 400,
+        },
+        DELETE_ERROR: {
+            code: 'IMAGE_GENERATION.DELETE_ERROR',
+            message: 'Произошла ошибка при удалении генерации изображения',
+            httpCode: 500,
+        },
+        MAX_PROMPT_LENGTH_EXCEEDED: {
+            code: 'IMAGE_GENERATION.MAX_PROMPT_LENGTH_EXCEEDED',
+            message: 'Превышена максимальная длина запроса',
+            httpCode: 400,
+        },
+        POLL_TIMEOUT: {
+            code: 'IMAGE_GENERATION.POLL_TIMEOUT',
+            message: 'Превышено время ожидания генерации изображения',
+            httpCode: 500,
+        },
+    },
+    IMAGE_GENERATION_MODEL: {
+        SAVE_ERROR: {
+            code: 'IMAGE_GENERATION_MODEL.SAVE_ERROR',
+            message: 'Произошла ошибка при сохранении модели для генерации изображений',
+            httpCode: 500,
+        },
+        FIND_ERROR: {
+            code: 'IMAGE_GENERATION_MODEL.FIND_ERROR',
+            message: 'Произошла ошибка при поиске модели для генерации изображений',
+            httpCode: 500,
+        },
+        NOT_FOUND: {
+            code: 'IMAGE_GENERATION_MODEL.NOT_FOUND',
+            message: 'Модель для генерации изображений не найдена',
+            httpCode: 404,
+        },
+        MODEL_INACTIVE: {
+            code: 'IMAGE_GENERATION_MODEL.MODEL_INACTIVE',
+            message: 'Модель для генерации изображений неактивна',
+            httpCode: 400,
+        },
+    },
+    IMAGE_GENERATION_JOB: {
+        SAVE_ERROR: {
+            code: 'IMAGE_GENERATION_JOB.SAVE_ERROR',
+            message: 'Произошла ошибка при сохранении задания на генерацию изображения',
+            httpCode: 500,
+        },
+        FIND_ERROR: {
+            code: 'IMAGE_GENERATION_JOB.FIND_ERROR',
+            message: 'Произошла ошибка при поиске задания на генерацию изображения',
+            httpCode: 500,
+        },
+        NOT_FOUND: {
+            code: 'IMAGE_GENERATION_JOB.NOT_FOUND',
+            message: 'Задание на генерацию изображения не найдено',
+            httpCode: 404,
+        },
+        NOT_AN_OWNER: {
+            code: 'IMAGE_GENERATION_JOB.NOT_AN_OWNER',
+            message: 'Пользователь не является владельцем задания на генерацию изображения',
+            httpCode: 403,
+        },
+        DELETE_ERROR: {
+            code: 'IMAGE_GENERATION_JOB.DELETE_ERROR',
+            message: 'Произошла ошибка при удалении задания на генерацию изображения',
+            httpCode: 500,
+        },
+    },
+    IMAGE_GENERATION_PRESET: {
+        FIND_ERROR: {
+            code: 'IMAGE_GENERATION_PRESET.FIND_ERROR',
+            message: 'Произошла ошибка при поиске пресета генерации изображения',
+            httpCode: 500,
+        },
+        NOT_FOUND: {
+            code: 'IMAGE_GENERATION_PRESET.NOT_FOUND',
+            message: 'Пресет генерации изображения не найден',
+            httpCode: 404,
+        },
+    },
+    AI_VENDOR: {
+        FIND_ERROR: {
+            code: 'AI_VENDOR.FIND_ERROR',
+            message: 'Произошла ошибка при поиске поставщика ИИ',
+            httpCode: 500,
+        },
+        NOT_FOUND: {
+            code: 'AI_VENDOR.NOT_FOUND',
+            message: 'Поставщик ИИ не найден',
+            httpCode: 404,
+        },
+    },
     CONTENT_POLICY: {
         PHOTOREALISTIC_PEOPLE: {
             code: 'CONTENT_POLICY.PHOTOREALISTIC_PEOPLE',

package/build/image-editor/enums/image-editor-model-type.enum.js

@@ -5,4 +5,5 @@ var IMAGE_EDITOR_MODEL_TYPE;
 (function (IMAGE_EDITOR_MODEL_TYPE) {
     IMAGE_EDITOR_MODEL_TYPE["NANO_BANANA_EDIT"] = "NANO_BANANA_EDIT";
     IMAGE_EDITOR_MODEL_TYPE["RECRAFT_REMOVE_BACKGROUND"] = "RECRAFT_REMOVE_BACKGROUND";
+    IMAGE_EDITOR_MODEL_TYPE["NANO_BANANA_PRO_EDIT"] = "NANO_BANANA_PRO_EDIT";
 })(IMAGE_EDITOR_MODEL_TYPE || (exports.IMAGE_EDITOR_MODEL_TYPE = IMAGE_EDITOR_MODEL_TYPE = {}));

package/build/image-editor/enums/image-editor-strategy.enum.js

@@ -6,4 +6,5 @@ var IMAGE_EDITING_STRATEGY;
     IMAGE_EDITING_STRATEGY["NANO_BANANA_POLLING"] = "NANO_BANANA_POLLING";
     IMAGE_EDITING_STRATEGY["NANO_BANANA_CALLBACK"] = "NANO_BANANA_CALLBACK";
     IMAGE_EDITING_STRATEGY["RECRAFT_REMOVE_BACKGROUND_CALLBACK"] = "RECRAFT_REMOVE_BACKGROUND_CALLBACK";
+    IMAGE_EDITING_STRATEGY["NANO_BANANA_PRO_CALLBACK"] = "NANO_BANANA_PRO_CALLBACK";
 })(IMAGE_EDITING_STRATEGY || (exports.IMAGE_EDITING_STRATEGY = IMAGE_EDITING_STRATEGY = {}));

package/build/image-generation/commands/execute-image-generation.command.js

@@ -0,0 +1,19 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ExecuteImageGenerationCommand = void 0;
+const zod_1 = require("zod");
+const command_response_schema_1 = require("../../common/models/command-response.schema");
+const models_1 = require("../models");
+var ExecuteImageGenerationCommand;
+(function (ExecuteImageGenerationCommand) {
+    ExecuteImageGenerationCommand.RequestSchema = zod_1.z.object({
+        userId: zod_1.z.string().uuid().nullable().optional(),
+        unregisteredUserId: zod_1.z.string().uuid().nullable().optional(),
+        prompt: zod_1.z.string(),
+        userBalance: zod_1.z.number(),
+        modelId: zod_1.z.string().uuid(),
+        params: models_1.ImageGenerationRequestParamsSchema,
+        presetId: zod_1.z.string().uuid().nullable().optional(),
+    });
+    ExecuteImageGenerationCommand.ResponseSchema = (0, command_response_schema_1.ICommandResponseSchema)(models_1.ImageGenerationJobSchema);
+})(ExecuteImageGenerationCommand || (exports.ExecuteImageGenerationCommand = ExecuteImageGenerationCommand = {}));

package/build/image-generation/commands/index.js

@@ -0,0 +1,22 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __exportStar = (this && this.__exportStar) || function(m, exports) {
+    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+__exportStar(require("./execute-image-generation.command"), exports);
+__exportStar(require("./save-image-generation-job.command"), exports);
+__exportStar(require("./soft-delete-image-generation-job-by-uuid.command"), exports);
+__exportStar(require("./soft-delete-image-generation-jobs-by-criteria.command"), exports);
+__exportStar(require("./set-reaction-to-image-generation-job.command"), exports);
+__exportStar(require("./update-image-generation-job-title.command"), exports);

package/build/image-generation/commands/save-image-generation-job.command.js

@@ -0,0 +1,17 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.SaveImageGenerationJobCommand = void 0;
+const zod_1 = require("zod");
+const command_response_schema_1 = require("../../common/models/command-response.schema");
+const models_1 = require("../models");
+var SaveImageGenerationJobCommand;
+(function (SaveImageGenerationJobCommand) {
+    SaveImageGenerationJobCommand.RequestSchema = zod_1.z.object({
+        userId: zod_1.z.string().uuid().nullable().optional(),
+        unregisteredUserId: zod_1.z.string().uuid().nullable().optional(),
+        prompt: zod_1.z.string(),
+        modelId: zod_1.z.string().uuid(),
+        userBalance: zod_1.z.number(),
+    });
+    SaveImageGenerationJobCommand.ResponseSchema = (0, command_response_schema_1.ICommandResponseSchema)(models_1.ImageGenerationJobSchema);
+})(SaveImageGenerationJobCommand || (exports.SaveImageGenerationJobCommand = SaveImageGenerationJobCommand = {}));

package/build/image-generation/commands/set-reaction-to-image-generation-job.command.js

@@ -0,0 +1,17 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.SetReactionToImageGenerationJobCommand = void 0;
+const zod_1 = require("zod");
+const common_1 = require("../../common");
+const command_response_schema_1 = require("../../common/models/command-response.schema");
+const models_1 = require("../models");
+var SetReactionToImageGenerationJobCommand;
+(function (SetReactionToImageGenerationJobCommand) {
+    SetReactionToImageGenerationJobCommand.RequestSchema = zod_1.z.object({
+        userId: zod_1.z.string().uuid().nullable().optional(),
+        unregisteredUserId: zod_1.z.string().uuid().nullable().optional(),
+        uuid: zod_1.z.string(),
+        reaction: zod_1.z.nativeEnum(common_1.USER_REACTION).nullable(),
+    });
+    SetReactionToImageGenerationJobCommand.ResponseSchema = (0, command_response_schema_1.ICommandResponseSchema)(models_1.ImageGenerationJobSchema);
+})(SetReactionToImageGenerationJobCommand || (exports.SetReactionToImageGenerationJobCommand = SetReactionToImageGenerationJobCommand = {}));

package/build/image-generation/commands/soft-delete-image-generation-job-by-uuid.command.js

@@ -0,0 +1,14 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.SoftDeleteImageGenerationJobByUuidCommand = void 0;
+const zod_1 = require("zod");
+const command_response_schema_1 = require("../../common/models/command-response.schema");
+var SoftDeleteImageGenerationJobByUuidCommand;
+(function (SoftDeleteImageGenerationJobByUuidCommand) {
+    SoftDeleteImageGenerationJobByUuidCommand.RequestSchema = zod_1.z.object({
+        userId: zod_1.z.string().uuid().nullable().optional(),
+        unregisteredUserId: zod_1.z.string().uuid().nullable().optional(),
+        uuid: zod_1.z.string(),
+    });
+    SoftDeleteImageGenerationJobByUuidCommand.ResponseSchema = (0, command_response_schema_1.ICommandResponseSchema)(zod_1.z.boolean());
+})(SoftDeleteImageGenerationJobByUuidCommand || (exports.SoftDeleteImageGenerationJobByUuidCommand = SoftDeleteImageGenerationJobByUuidCommand = {}));

package/build/image-generation/commands/soft-delete-image-generation-jobs-by-criteria.command.js

@@ -0,0 +1,13 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.SoftDeleteImageGenerationJobsByCriteriaCommand = void 0;
+const zod_1 = require("zod");
+const command_response_schema_1 = require("../../common/models/command-response.schema");
+var SoftDeleteImageGenerationJobsByCriteriaCommand;
+(function (SoftDeleteImageGenerationJobsByCriteriaCommand) {
+    SoftDeleteImageGenerationJobsByCriteriaCommand.RequestSchema = zod_1.z.object({
+        userId: zod_1.z.string().uuid().nullable().optional(),
+        unregisteredUserId: zod_1.z.string().uuid().nullable().optional(),
+    });
+    SoftDeleteImageGenerationJobsByCriteriaCommand.ResponseSchema = (0, command_response_schema_1.ICommandResponseSchema)(zod_1.z.boolean());
+})(SoftDeleteImageGenerationJobsByCriteriaCommand || (exports.SoftDeleteImageGenerationJobsByCriteriaCommand = SoftDeleteImageGenerationJobsByCriteriaCommand = {}));

package/build/image-generation/commands/update-image-generation-job-title.command.js

@@ -0,0 +1,16 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.UpdateImageGenerationJobTitleCommand = void 0;
+const zod_1 = require("zod");
+const command_response_schema_1 = require("../../common/models/command-response.schema");
+const models_1 = require("../models");
+var UpdateImageGenerationJobTitleCommand;
+(function (UpdateImageGenerationJobTitleCommand) {
+    UpdateImageGenerationJobTitleCommand.RequestSchema = zod_1.z.object({
+        userId: zod_1.z.string().uuid().nullable().optional(),
+        unregisteredUserId: zod_1.z.string().uuid().nullable().optional(),
+        uuid: zod_1.z.string(),
+        title: zod_1.z.string(),
+    });
+    UpdateImageGenerationJobTitleCommand.ResponseSchema = (0, command_response_schema_1.ICommandResponseSchema)(models_1.ImageGenerationJobSchema);
+})(UpdateImageGenerationJobTitleCommand || (exports.UpdateImageGenerationJobTitleCommand = UpdateImageGenerationJobTitleCommand = {}));

package/build/image-generation/enums/image-generation-model-strategy.enum.js

@@ -0,0 +1,20 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.IMAGE_GENERATION_MODEL_STRATEGY = void 0;
+var IMAGE_GENERATION_MODEL_STRATEGY;
+(function (IMAGE_GENERATION_MODEL_STRATEGY) {
+    // Together AI Models
+    IMAGE_GENERATION_MODEL_STRATEGY["FLUX_IMAGE_GENERATION"] = "FLUX_IMAGE_GENERATION";
+    IMAGE_GENERATION_MODEL_STRATEGY["FLUX_DEV_IMAGE_GENERATION"] = "FLUX_DEV_IMAGE_GENERATION";
+    IMAGE_GENERATION_MODEL_STRATEGY["STABLE_DIFFUSION_IMAGE_GENERATION"] = "STABLE_DIFFUSION_IMAGE_GENERATION";
+    // Grok AI
+    IMAGE_GENERATION_MODEL_STRATEGY["GROK_IMAGE_GENERATION"] = "GROK_IMAGE_GENERATION";
+    // OpenAI
+    IMAGE_GENERATION_MODEL_STRATEGY["DALLE_IMAGE_GENERATION"] = "DALLE_IMAGE_GENERATION";
+    IMAGE_GENERATION_MODEL_STRATEGY["OPENAI_IMAGE_1_GENERATION"] = "OPENAI_IMAGE_1_GENERATION";
+    // Google AI
+    IMAGE_GENERATION_MODEL_STRATEGY["GEMINI_IMAGE_GENERATION"] = "GEMINI_IMAGE_GENERATION";
+    IMAGE_GENERATION_MODEL_STRATEGY["IMAGEN_IMAGE_GENERATION"] = "IMAGEN_IMAGE_GENERATION";
+    // Midjourney
+    IMAGE_GENERATION_MODEL_STRATEGY["MIDJOURNEY_IMAGE_GENERATION"] = "MIDJOURNEY_IMAGE_GENERATION";
+})(IMAGE_GENERATION_MODEL_STRATEGY || (exports.IMAGE_GENERATION_MODEL_STRATEGY = IMAGE_GENERATION_MODEL_STRATEGY = {}));

package/build/image-generation/enums/index.js

@@ -0,0 +1,17 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __exportStar = (this && this.__exportStar) || function(m, exports) {
+    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+__exportStar(require("./image-generation-model-strategy.enum"), exports);

package/build/image-generation/index.js

@@ -0,0 +1,21 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __exportStar = (this && this.__exportStar) || function(m, exports) {
+    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+__exportStar(require("./enums"), exports);
+__exportStar(require("./models"), exports);
+__exportStar(require("./commands"), exports);
+__exportStar(require("./queries"), exports);
+__exportStar(require("./routes"), exports);

package/build/image-generation/models/image-generation-job.schema.js

@@ -0,0 +1,26 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ImageGenerationJobSchema = void 0;
+const zod_1 = require("zod");
+const tools_1 = require("../../tools");
+const common_1 = require("../../common");
+exports.ImageGenerationJobSchema = zod_1.z.object({
+    uuid: zod_1.z.string(),
+    prompt: zod_1.z.string(),
+    title: zod_1.z.string(),
+    status: zod_1.z.nativeEnum(tools_1.JOB_STATUS),
+    error: zod_1.z.string().nullable(),
+    price: zod_1.z.number(),
+    reaction: zod_1.z.nativeEnum(common_1.USER_REACTION).nullable(),
+    externalId: zod_1.z.string().nullable(),
+    params: zod_1.z.any(),
+    output: zod_1.z.array(zod_1.z.string()),
+    modelId: zod_1.z.string(),
+    attempts: zod_1.z.array(zod_1.z.any()),
+    userId: zod_1.z.string().nullable().optional(),
+    unregisteredUserId: zod_1.z.string().nullable().optional(),
+    isDeleted: zod_1.z.boolean(),
+    internalError: zod_1.z.string().nullable(),
+    createdAt: zod_1.z.date(),
+    updatedAt: zod_1.z.date(),
+});

package/build/image-generation/models/image-generation-model.schema.js

@@ -0,0 +1,26 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ImageGenerationModelSchema = exports.ImageGenerationRequestParamsSchema = void 0;
+const zod_1 = require("zod");
+const common_1 = require("../../common");
+const enums_1 = require("../enums");
+exports.ImageGenerationRequestParamsSchema = zod_1.z.object({
+    aspectRatio: zod_1.z.string(),
+    attachedImages: zod_1.z.array(zod_1.z.string()),
+    enhancePrompt: zod_1.z.boolean(),
+});
+exports.ImageGenerationModelSchema = zod_1.z.object({
+    uuid: zod_1.z.string(),
+    title: zod_1.z.string(),
+    description: zod_1.z.string(),
+    aiModel: zod_1.z.string(),
+    price: zod_1.z.number(),
+    order: zod_1.z.number(),
+    params: zod_1.z.any(),
+    strategy: zod_1.z.nativeEnum(enums_1.IMAGE_GENERATION_MODEL_STRATEGY),
+    status: zod_1.z.nativeEnum(common_1.TOOL_MODEL_STATUS),
+    maxPromptLength: zod_1.z.number(),
+    iconVariants: common_1.IconVariantsSchema,
+    createdAt: zod_1.z.date(),
+    updatedAt: zod_1.z.date(),
+});

package/build/image-generation/models/image-generation-preset.schema.js

@@ -0,0 +1,13 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ImageGenerationPresetSchema = void 0;
+const zod_1 = require("zod");
+exports.ImageGenerationPresetSchema = zod_1.z.object({
+    uuid: zod_1.z.string().uuid(),
+    title: zod_1.z.string(),
+    prompt: zod_1.z.string(),
+    sample: zod_1.z.string(),
+    order: zod_1.z.number(),
+    createdAt: zod_1.z.date(),
+    updatedAt: zod_1.z.date(),
+});

package/build/image-generation/models/index.js

@@ -0,0 +1,19 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __exportStar = (this && this.__exportStar) || function(m, exports) {
+    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+__exportStar(require("./image-generation-job.schema"), exports);
+__exportStar(require("./image-generation-model.schema"), exports);
+__exportStar(require("./image-generation-preset.schema"), exports);

package/build/image-generation/queries/find-image-generation-job-by-id.query.js

@@ -0,0 +1,19 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.FindImageGenerationJobByIdQuery = void 0;
+const zod_1 = require("zod");
+const command_response_schema_1 = require("../../common/models/command-response.schema");
+const models_1 = require("../models");
+var FindImageGenerationJobByIdQuery;
+(function (FindImageGenerationJobByIdQuery) {
+    FindImageGenerationJobByIdQuery.RequestSchema = zod_1.z
+        .object({
+        uuid: zod_1.z.string(),
+        userId: zod_1.z.string().uuid().nullable().optional(),
+        unregisteredUserId: zod_1.z.string().uuid().nullable().optional(),
+    })
+        .refine((data) => data.userId || data.unregisteredUserId, {
+        message: 'Either userId or unregisteredUserId must be provided',
+    });
+    FindImageGenerationJobByIdQuery.ResponseSchema = (0, command_response_schema_1.ICommandResponseSchema)(models_1.ImageGenerationJobSchema);
+})(FindImageGenerationJobByIdQuery || (exports.FindImageGenerationJobByIdQuery = FindImageGenerationJobByIdQuery = {}));

package/build/image-generation/queries/find-image-generation-jobs.query.js

@@ -0,0 +1,28 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.FindImageGenerationJobsQuery = void 0;
+const zod_1 = require("zod");
+const command_response_schema_1 = require("../../common/models/command-response.schema");
+const models_1 = require("../models");
+var FindImageGenerationJobsQuery;
+(function (FindImageGenerationJobsQuery) {
+    FindImageGenerationJobsQuery.RequestSchema = zod_1.z
+        .object({
+        userId: zod_1.z.string().uuid().optional(),
+        unregisteredUserId: zod_1.z.string().uuid().optional(),
+        title: zod_1.z.string().optional(),
+        limit: zod_1.z.coerce.number().min(1).optional(),
+        offset: zod_1.z.coerce.number().min(0).default(0).optional(),
+    })
+        .refine((data) => {
+        return Boolean(data.userId) !== Boolean(data.unregisteredUserId);
+    }, {
+        message: 'At least userId or unregisteredUserId must be present',
+        path: ['userId', 'unregisteredUserId'],
+    });
+    FindImageGenerationJobsQuery.ResponseSchema = (0, command_response_schema_1.ICommandResponseSchema)(zod_1.z.object({
+        data: zod_1.z.array(models_1.ImageGenerationJobSchema),
+        page: zod_1.z.number(),
+        totalPages: zod_1.z.number(),
+    }));
+})(FindImageGenerationJobsQuery || (exports.FindImageGenerationJobsQuery = FindImageGenerationJobsQuery = {}));

package/build/image-generation/queries/get-image-generation-config.query.js

@@ -0,0 +1,19 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.GetImageGenerationConfigQuery = void 0;
+const zod_1 = require("zod");
+const command_response_schema_1 = require("../../common/models/command-response.schema");
+const models_1 = require("../models");
+const ai_vendor_1 = require("../../ai-vendor");
+const VendorWithModelsSchema = ai_vendor_1.AiVendorSchema.extend({
+    models: zod_1.z.array(models_1.ImageGenerationModelSchema),
+});
+const ImageGenerationConfigSchema = zod_1.z.object({
+    vendors: zod_1.z.array(VendorWithModelsSchema),
+    presets: zod_1.z.array(models_1.ImageGenerationPresetSchema),
+});
+var GetImageGenerationConfigQuery;
+(function (GetImageGenerationConfigQuery) {
+    GetImageGenerationConfigQuery.RequestSchema = zod_1.z.object({});
+    GetImageGenerationConfigQuery.ResponseSchema = (0, command_response_schema_1.ICommandResponseSchema)(ImageGenerationConfigSchema);
+})(GetImageGenerationConfigQuery || (exports.GetImageGenerationConfigQuery = GetImageGenerationConfigQuery = {}));

package/build/image-generation/queries/index.js

@@ -0,0 +1,19 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __exportStar = (this && this.__exportStar) || function(m, exports) {
+    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+__exportStar(require("./get-image-generation-config.query"), exports);
+__exportStar(require("./find-image-generation-job-by-id.query"), exports);
+__exportStar(require("./find-image-generation-jobs.query"), exports);

package/build/image-generation/routes/image-generation.amqp.routes.js

@@ -0,0 +1,13 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.IMAGE_GENERATION_AMQP_ROUTES = void 0;
+exports.IMAGE_GENERATION_AMQP_ROUTES = {
+    EXECUTE: 'tools.image-generation.execute.rpc',
+    CONFIG: 'tools.image-generation.config.rpc',
+    GET_JOB: 'tools.image-generation.jobs.get.rpc',
+    LIST_JOBS: 'tools.image-generation.jobs.list.rpc',
+    SET_REACTION: 'tools.image-generation.jobs.set-reaction.rpc',
+    UPDATE_TITLE: 'tools.image-generation.jobs.update-title.rpc',
+    SOFT_DELETE: 'tools.image-generation.jobs.soft-delete.rpc',
+    SOFT_DELETE_ALL: 'tools.image-generation.jobs.soft-delete-all.rpc',
+};

package/build/image-generation/routes/image-generation.http.routes.js

@@ -0,0 +1,7 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.IMAGE_GENERATION_ROUTES = exports.IMAGE_GENERATION_CONTROLLER = void 0;
+exports.IMAGE_GENERATION_CONTROLLER = 'image-generation';
+exports.IMAGE_GENERATION_ROUTES = {
+    KIE_MIDJOURNEY_CALLBACK: 'kie/midjourney/callback/:jobId',
+};

package/build/image-generation/routes/index.js

@@ -0,0 +1,18 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __exportStar = (this && this.__exportStar) || function(m, exports) {
+    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+__exportStar(require("./image-generation.amqp.routes"), exports);
+__exportStar(require("./image-generation.http.routes"), exports);

package/build/index.js

@@ -46,8 +46,10 @@ __exportStar(require("./language"), exports);
 __exportStar(require("./video"), exports);
 __exportStar(require("./writer"), exports);
 __exportStar(require("./image-editor"), exports);
+__exportStar(require("./image-generation"), exports);
 __exportStar(require("./video-editor"), exports);
 __exportStar(require("./music"), exports);
+__exportStar(require("./ai-vendor"), exports);
 const common = __importStar(require("./common"));
 const paraphrase = __importStar(require("./paraphrase"));
 const tools = __importStar(require("./tools"));
@@ -58,8 +60,10 @@ const language = __importStar(require("./language"));
 const video = __importStar(require("./video"));
 const writer = __importStar(require("./writer"));
 const imageEditor = __importStar(require("./image-editor"));
+const imageGeneration = __importStar(require("./image-generation"));
 const videoEditor = __importStar(require("./video-editor"));
 const music = __importStar(require("./music"));
+const aiVendor = __importStar(require("./ai-vendor"));
 var ToolService;
 (function (ToolService) {
     ToolService.Common = common;
@@ -72,6 +76,8 @@ var ToolService;
     ToolService.Video = video;
     ToolService.Writer = writer;
     ToolService.ImageEditor = imageEditor;
+    ToolService.ImageGeneration = imageGeneration;
     ToolService.VideoEditor = videoEditor;
     ToolService.Music = music;
+    ToolService.AiVendor = aiVendor;
 })(ToolService || (exports.ToolService = ToolService = {}));

package/build/tools/enums/tool-type.enum.js

@@ -15,4 +15,5 @@ var TOOL_TYPE;
     TOOL_TYPE["VIDEO_EDITOR"] = "VIDEO_EDITOR";
     TOOL_TYPE["MUSIC"] = "MUSIC";
     TOOL_TYPE["MUSIC_ACTION"] = "MUSIC_ACTION";
+    TOOL_TYPE["IMAGE_GENERATION"] = "IMAGE_GENERATION";
 })(TOOL_TYPE || (exports.TOOL_TYPE = TOOL_TYPE = {}));

package/common/errors/errors.ts

@@ -1217,6 +1217,111 @@ export const ERRORS = {
             httpCode: 500,
         },
     },
+    IMAGE_GENERATION: {
+        AI_ERROR: {
+            code: 'IMAGE_GENERATION.AI_ERROR',
+            message: 'Произошла ошибка во время запроса к ИИ',
+            httpCode: 500,
+        },
+        INSUFFICIENT_BALANCE: {
+            code: 'IMAGE_GENERATION.INSUFFICIENT_BALANCE',
+            message: 'У пользователя недостаточно средств',
+            httpCode: 400,
+        },
+        INVALID_PARAMS: {
+            code: 'IMAGE_GENERATION.INVALID_PARAMS',
+            message: 'Некорректные параметры изображения',
+            httpCode: 400,
+        },
+        DELETE_ERROR: {
+            code: 'IMAGE_GENERATION.DELETE_ERROR',
+            message: 'Произошла ошибка при удалении генерации изображения',
+            httpCode: 500,
+        },
+        MAX_PROMPT_LENGTH_EXCEEDED: {
+            code: 'IMAGE_GENERATION.MAX_PROMPT_LENGTH_EXCEEDED',
+            message: 'Превышена максимальная длина запроса',
+            httpCode: 400,
+        },
+        POLL_TIMEOUT: {
+            code: 'IMAGE_GENERATION.POLL_TIMEOUT',
+            message: 'Превышено время ожидания генерации изображения',
+            httpCode: 500,
+        },
+    },
+    IMAGE_GENERATION_MODEL: {
+        SAVE_ERROR: {
+            code: 'IMAGE_GENERATION_MODEL.SAVE_ERROR',
+            message: 'Произошла ошибка при сохранении модели для генерации изображений',
+            httpCode: 500,
+        },
+        FIND_ERROR: {
+            code: 'IMAGE_GENERATION_MODEL.FIND_ERROR',
+            message: 'Произошла ошибка при поиске модели для генерации изображений',
+            httpCode: 500,
+        },
+        NOT_FOUND: {
+            code: 'IMAGE_GENERATION_MODEL.NOT_FOUND',
+            message: 'Модель для генерации изображений не найдена',
+            httpCode: 404,
+        },
+        MODEL_INACTIVE: {
+            code: 'IMAGE_GENERATION_MODEL.MODEL_INACTIVE',
+            message: 'Модель для генерации изображений неактивна',
+            httpCode: 400,
+        },
+    },
+    IMAGE_GENERATION_JOB: {
+        SAVE_ERROR: {
+            code: 'IMAGE_GENERATION_JOB.SAVE_ERROR',
+            message: 'Произошла ошибка при сохранении задания на генерацию изображения',
+            httpCode: 500,
+        },
+        FIND_ERROR: {
+            code: 'IMAGE_GENERATION_JOB.FIND_ERROR',
+            message: 'Произошла ошибка при поиске задания на генерацию изображения',
+            httpCode: 500,
+        },
+        NOT_FOUND: {
+            code: 'IMAGE_GENERATION_JOB.NOT_FOUND',
+            message: 'Задание на генерацию изображения не найдено',
+            httpCode: 404,
+        },
+        NOT_AN_OWNER: {
+            code: 'IMAGE_GENERATION_JOB.NOT_AN_OWNER',
+            message: 'Пользователь не является владельцем задания на генерацию изображения',
+            httpCode: 403,
+        },
+        DELETE_ERROR: {
+            code: 'IMAGE_GENERATION_JOB.DELETE_ERROR',
+            message: 'Произошла ошибка при удалении задания на генерацию изображения',
+            httpCode: 500,
+        },
+    },
+    IMAGE_GENERATION_PRESET: {
+        FIND_ERROR: {
+            code: 'IMAGE_GENERATION_PRESET.FIND_ERROR',
+            message: 'Произошла ошибка при поиске пресета генерации изображения',
+            httpCode: 500,
+        },
+        NOT_FOUND: {
+            code: 'IMAGE_GENERATION_PRESET.NOT_FOUND',
+            message: 'Пресет генерации изображения не найден',
+            httpCode: 404,
+        },
+    },
+    AI_VENDOR: {
+        FIND_ERROR: {
+            code: 'AI_VENDOR.FIND_ERROR',
+            message: 'Произошла ошибка при поиске поставщика ИИ',
+            httpCode: 500,
+        },
+        NOT_FOUND: {
+            code: 'AI_VENDOR.NOT_FOUND',
+            message: 'Поставщик ИИ не найден',
+            httpCode: 404,
+        },
+    },
     CONTENT_POLICY: {
         PHOTOREALISTIC_PEOPLE: {
             code: 'CONTENT_POLICY.PHOTOREALISTIC_PEOPLE',

package/image-editor/enums/image-editor-model-type.enum.ts

@@ -1,4 +1,5 @@
 export enum IMAGE_EDITOR_MODEL_TYPE {
     NANO_BANANA_EDIT = 'NANO_BANANA_EDIT',
     RECRAFT_REMOVE_BACKGROUND = 'RECRAFT_REMOVE_BACKGROUND',
+    NANO_BANANA_PRO_EDIT = 'NANO_BANANA_PRO_EDIT',
 }

package/image-editor/enums/image-editor-strategy.enum.ts

@@ -2,4 +2,5 @@ export enum IMAGE_EDITING_STRATEGY {
     NANO_BANANA_POLLING = 'NANO_BANANA_POLLING',
     NANO_BANANA_CALLBACK = 'NANO_BANANA_CALLBACK',
     RECRAFT_REMOVE_BACKGROUND_CALLBACK = 'RECRAFT_REMOVE_BACKGROUND_CALLBACK',
+    NANO_BANANA_PRO_CALLBACK = 'NANO_BANANA_PRO_CALLBACK',
 }

package/image-generation/commands/execute-image-generation.command.ts

@@ -0,0 +1,19 @@
+import { z } from 'zod';
+import { ICommandResponseSchema } from '../../common/models/command-response.schema';
+import { ImageGenerationJobSchema, ImageGenerationRequestParamsSchema } from '../models';
+
+export namespace ExecuteImageGenerationCommand {
+    export const RequestSchema = z.object({
+        userId: z.string().uuid().nullable().optional(),
+        unregisteredUserId: z.string().uuid().nullable().optional(),
+        prompt: z.string(),
+        userBalance: z.number(),
+        modelId: z.string().uuid(),
+        params: ImageGenerationRequestParamsSchema,
+        presetId: z.string().uuid().nullable().optional(),
+    });
+    export type Request = z.infer<typeof RequestSchema>;
+
+    export const ResponseSchema = ICommandResponseSchema(ImageGenerationJobSchema);
+    export type Response = z.infer<typeof ResponseSchema>;
+}

package/image-generation/commands/index.ts

@@ -0,0 +1,6 @@
+export * from './execute-image-generation.command';
+export * from './save-image-generation-job.command';
+export * from './soft-delete-image-generation-job-by-uuid.command';
+export * from './soft-delete-image-generation-jobs-by-criteria.command';
+export * from './set-reaction-to-image-generation-job.command';
+export * from './update-image-generation-job-title.command';

package/image-generation/commands/save-image-generation-job.command.ts

@@ -0,0 +1,17 @@
+import { z } from 'zod';
+import { ICommandResponseSchema } from '../../common/models/command-response.schema';
+import { ImageGenerationJobSchema } from '../models';
+
+export namespace SaveImageGenerationJobCommand {
+    export const RequestSchema = z.object({
+        userId: z.string().uuid().nullable().optional(),
+        unregisteredUserId: z.string().uuid().nullable().optional(),
+        prompt: z.string(),
+        modelId: z.string().uuid(),
+        userBalance: z.number(),
+    });
+    export type Request = z.infer<typeof RequestSchema>;
+
+    export const ResponseSchema = ICommandResponseSchema(ImageGenerationJobSchema);
+    export type Response = z.infer<typeof ResponseSchema>;
+}

package/image-generation/commands/set-reaction-to-image-generation-job.command.ts

@@ -0,0 +1,17 @@
+import { z } from 'zod';
+import { USER_REACTION } from '../../common';
+import { ICommandResponseSchema } from '../../common/models/command-response.schema';
+import { ImageGenerationJobSchema } from '../models';
+
+export namespace SetReactionToImageGenerationJobCommand {
+    export const RequestSchema = z.object({
+        userId: z.string().uuid().nullable().optional(),
+        unregisteredUserId: z.string().uuid().nullable().optional(),
+        uuid: z.string(),
+        reaction: z.nativeEnum(USER_REACTION).nullable(),
+    });
+    export type Request = z.infer<typeof RequestSchema>;
+
+    export const ResponseSchema = ICommandResponseSchema(ImageGenerationJobSchema);
+    export type Response = z.infer<typeof ResponseSchema>;
+}

package/image-generation/commands/soft-delete-image-generation-job-by-uuid.command.ts

@@ -0,0 +1,14 @@
+import { z } from 'zod';
+import { ICommandResponseSchema } from '../../common/models/command-response.schema';
+
+export namespace SoftDeleteImageGenerationJobByUuidCommand {
+    export const RequestSchema = z.object({
+        userId: z.string().uuid().nullable().optional(),
+        unregisteredUserId: z.string().uuid().nullable().optional(),
+        uuid: z.string(),
+    });
+    export type Request = z.infer<typeof RequestSchema>;
+
+    export const ResponseSchema = ICommandResponseSchema(z.boolean());
+    export type Response = z.infer<typeof ResponseSchema>;
+}

package/image-generation/commands/soft-delete-image-generation-jobs-by-criteria.command.ts

@@ -0,0 +1,13 @@
+import { z } from 'zod';
+import { ICommandResponseSchema } from '../../common/models/command-response.schema';
+
+export namespace SoftDeleteImageGenerationJobsByCriteriaCommand {
+    export const RequestSchema = z.object({
+        userId: z.string().uuid().nullable().optional(),
+        unregisteredUserId: z.string().uuid().nullable().optional(),
+    });
+    export type Request = z.infer<typeof RequestSchema>;
+
+    export const ResponseSchema = ICommandResponseSchema(z.boolean());
+    export type Response = z.infer<typeof ResponseSchema>;
+}

package/image-generation/commands/update-image-generation-job-title.command.ts

@@ -0,0 +1,16 @@
+import { z } from 'zod';
+import { ICommandResponseSchema } from '../../common/models/command-response.schema';
+import { ImageGenerationJobSchema } from '../models';
+
+export namespace UpdateImageGenerationJobTitleCommand {
+    export const RequestSchema = z.object({
+        userId: z.string().uuid().nullable().optional(),
+        unregisteredUserId: z.string().uuid().nullable().optional(),
+        uuid: z.string(),
+        title: z.string(),
+    });
+    export type Request = z.infer<typeof RequestSchema>;
+
+    export const ResponseSchema = ICommandResponseSchema(ImageGenerationJobSchema);
+    export type Response = z.infer<typeof ResponseSchema>;
+}

package/image-generation/enums/image-generation-model-strategy.enum.ts

@@ -0,0 +1,20 @@
+export enum IMAGE_GENERATION_MODEL_STRATEGY {
+    // Together AI Models
+    FLUX_IMAGE_GENERATION = 'FLUX_IMAGE_GENERATION',
+    FLUX_DEV_IMAGE_GENERATION = 'FLUX_DEV_IMAGE_GENERATION',
+    STABLE_DIFFUSION_IMAGE_GENERATION = 'STABLE_DIFFUSION_IMAGE_GENERATION',
+
+    // Grok AI
+    GROK_IMAGE_GENERATION = 'GROK_IMAGE_GENERATION',
+
+    // OpenAI
+    DALLE_IMAGE_GENERATION = 'DALLE_IMAGE_GENERATION',
+    OPENAI_IMAGE_1_GENERATION = 'OPENAI_IMAGE_1_GENERATION',
+
+    // Google AI
+    GEMINI_IMAGE_GENERATION = 'GEMINI_IMAGE_GENERATION',
+    IMAGEN_IMAGE_GENERATION = 'IMAGEN_IMAGE_GENERATION',
+
+    // Midjourney
+    MIDJOURNEY_IMAGE_GENERATION = 'MIDJOURNEY_IMAGE_GENERATION',
+}

package/image-generation/enums/index.ts

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

package/image-generation/index.ts

@@ -0,0 +1,5 @@
+export * from './enums';
+export * from './models';
+export * from './commands';
+export * from './queries';
+export * from './routes';

package/image-generation/models/image-generation-job.schema.ts

@@ -0,0 +1,26 @@
+import { z } from 'zod';
+import { JOB_STATUS } from '../../tools';
+import { USER_REACTION } from '../../common';
+
+export const ImageGenerationJobSchema = z.object({
+    uuid: z.string(),
+    prompt: z.string(),
+    title: z.string(),
+    status: z.nativeEnum(JOB_STATUS),
+    error: z.string().nullable(),
+    price: z.number(),
+    reaction: z.nativeEnum(USER_REACTION).nullable(),
+    externalId: z.string().nullable(),
+    params: z.any(),
+    output: z.array(z.string()),
+    modelId: z.string(),
+    attempts: z.array(z.any()),
+    userId: z.string().nullable().optional(),
+    unregisteredUserId: z.string().nullable().optional(),
+    isDeleted: z.boolean(),
+    internalError: z.string().nullable(),
+    createdAt: z.date(),
+    updatedAt: z.date(),
+});
+
+export type ImageGenerationJob = z.infer<typeof ImageGenerationJobSchema>;

package/image-generation/models/image-generation-model.schema.ts

@@ -0,0 +1,28 @@
+import { z } from 'zod';
+import { IconVariantsSchema, TOOL_MODEL_STATUS } from '../../common';
+import { IMAGE_GENERATION_MODEL_STRATEGY } from '../enums';
+
+export const ImageGenerationRequestParamsSchema = z.object({
+    aspectRatio: z.string(),
+    attachedImages: z.array(z.string()),
+    enhancePrompt: z.boolean(),
+});
+export type ImageGenerationRequestParams = z.infer<typeof ImageGenerationRequestParamsSchema>;
+
+export const ImageGenerationModelSchema = z.object({
+    uuid: z.string(),
+    title: z.string(),
+    description: z.string(),
+    aiModel: z.string(),
+    price: z.number(),
+    order: z.number(),
+    params: z.any(),
+    strategy: z.nativeEnum(IMAGE_GENERATION_MODEL_STRATEGY),
+    status: z.nativeEnum(TOOL_MODEL_STATUS),
+    maxPromptLength: z.number(),
+    iconVariants: IconVariantsSchema,
+    createdAt: z.date(),
+    updatedAt: z.date(),
+});
+
+export type ImageGenerationModel = z.infer<typeof ImageGenerationModelSchema>;

package/image-generation/models/image-generation-preset.schema.ts

@@ -0,0 +1,13 @@
+import { z } from 'zod';
+
+export const ImageGenerationPresetSchema = z.object({
+    uuid: z.string().uuid(),
+    title: z.string(),
+    prompt: z.string(),
+    sample: z.string(),
+    order: z.number(),
+    createdAt: z.date(),
+    updatedAt: z.date(),
+});
+
+export type ImageGenerationPreset = z.infer<typeof ImageGenerationPresetSchema>;

package/image-generation/models/index.ts

@@ -0,0 +1,3 @@
+export * from './image-generation-job.schema';
+export * from './image-generation-model.schema';
+export * from './image-generation-preset.schema';

package/image-generation/queries/find-image-generation-job-by-id.query.ts

@@ -0,0 +1,19 @@
+import { z } from 'zod';
+import { ICommandResponseSchema } from '../../common/models/command-response.schema';
+import { ImageGenerationJobSchema } from '../models';
+
+export namespace FindImageGenerationJobByIdQuery {
+    export const RequestSchema = z
+        .object({
+            uuid: z.string(),
+            userId: z.string().uuid().nullable().optional(),
+            unregisteredUserId: z.string().uuid().nullable().optional(),
+        })
+        .refine((data) => data.userId || data.unregisteredUserId, {
+            message: 'Either userId or unregisteredUserId must be provided',
+        });
+    export type Request = z.infer<typeof RequestSchema>;
+
+    export const ResponseSchema = ICommandResponseSchema(ImageGenerationJobSchema);
+    export type Response = z.infer<typeof ResponseSchema>;
+}

package/image-generation/queries/find-image-generation-jobs.query.ts

@@ -0,0 +1,33 @@
+import { z } from 'zod';
+import { ICommandResponseSchema } from '../../common/models/command-response.schema';
+import { ImageGenerationJobSchema } from '../models';
+
+export namespace FindImageGenerationJobsQuery {
+    export const RequestSchema = z
+        .object({
+            userId: z.string().uuid().optional(),
+            unregisteredUserId: z.string().uuid().optional(),
+            title: z.string().optional(),
+            limit: z.coerce.number().min(1).optional(),
+            offset: z.coerce.number().min(0).default(0).optional(),
+        })
+        .refine(
+            (data) => {
+                return Boolean(data.userId) !== Boolean(data.unregisteredUserId);
+            },
+            {
+                message: 'At least userId or unregisteredUserId must be present',
+                path: ['userId', 'unregisteredUserId'],
+            },
+        );
+    export type Request = z.infer<typeof RequestSchema>;
+
+    export const ResponseSchema = ICommandResponseSchema(
+        z.object({
+            data: z.array(ImageGenerationJobSchema),
+            page: z.number(),
+            totalPages: z.number(),
+        }),
+    );
+    export type Response = z.infer<typeof ResponseSchema>;
+}

package/image-generation/queries/get-image-generation-config.query.ts

@@ -0,0 +1,21 @@
+import { z } from 'zod';
+import { ICommandResponseSchema } from '../../common/models/command-response.schema';
+import { ImageGenerationModelSchema, ImageGenerationPresetSchema } from '../models';
+import { AiVendorSchema } from '../../ai-vendor';
+
+const VendorWithModelsSchema = AiVendorSchema.extend({
+    models: z.array(ImageGenerationModelSchema),
+});
+
+const ImageGenerationConfigSchema = z.object({
+    vendors: z.array(VendorWithModelsSchema),
+    presets: z.array(ImageGenerationPresetSchema),
+});
+
+export namespace GetImageGenerationConfigQuery {
+    export const RequestSchema = z.object({});
+    export type Request = z.infer<typeof RequestSchema>;
+
+    export const ResponseSchema = ICommandResponseSchema(ImageGenerationConfigSchema);
+    export type Response = z.infer<typeof ResponseSchema>;
+}

package/image-generation/queries/index.ts

@@ -0,0 +1,3 @@
+export * from './get-image-generation-config.query';
+export * from './find-image-generation-job-by-id.query';
+export * from './find-image-generation-jobs.query';

package/image-generation/routes/image-generation.amqp.routes.ts

@@ -0,0 +1,10 @@
+export const IMAGE_GENERATION_AMQP_ROUTES = {
+    EXECUTE: 'tools.image-generation.execute.rpc',
+    CONFIG: 'tools.image-generation.config.rpc',
+    GET_JOB: 'tools.image-generation.jobs.get.rpc',
+    LIST_JOBS: 'tools.image-generation.jobs.list.rpc',
+    SET_REACTION: 'tools.image-generation.jobs.set-reaction.rpc',
+    UPDATE_TITLE: 'tools.image-generation.jobs.update-title.rpc',
+    SOFT_DELETE: 'tools.image-generation.jobs.soft-delete.rpc',
+    SOFT_DELETE_ALL: 'tools.image-generation.jobs.soft-delete-all.rpc',
+} as const;

package/image-generation/routes/image-generation.http.routes.ts

@@ -0,0 +1,5 @@
+export const IMAGE_GENERATION_CONTROLLER = 'image-generation';
+
+export const IMAGE_GENERATION_ROUTES = {
+    KIE_MIDJOURNEY_CALLBACK: 'kie/midjourney/callback/:jobId',
+} as const;

package/image-generation/routes/index.ts

@@ -0,0 +1,2 @@
+export * from './image-generation.amqp.routes';
+export * from './image-generation.http.routes';

package/index.ts

@@ -7,8 +7,10 @@ export * from './language';
 export * from './video';
 export * from './writer';
 export * from './image-editor';
+export * from './image-generation';
 export * from './video-editor';
 export * from './music';
+export * from './ai-vendor';
 
 import * as common from './common';
 import * as paraphrase from './paraphrase';
@@ -20,8 +22,10 @@ import * as language from './language';
 import * as video from './video';
 import * as writer from './writer';
 import * as imageEditor from './image-editor';
+import * as imageGeneration from './image-generation';
 import * as videoEditor from './video-editor';
 import * as music from './music';
+import * as aiVendor from './ai-vendor';
 
 export namespace ToolService {
     export import Common = common;
@@ -34,6 +38,8 @@ export namespace ToolService {
     export import Video = video;
     export import Writer = writer;
     export import ImageEditor = imageEditor;
+    export import ImageGeneration = imageGeneration;
     export import VideoEditor = videoEditor;
     export import Music = music;
+    export import AiVendor = aiVendor;
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@purpleschool/gptbot-tools",
-  "version": "0.0.87",
+  "version": "0.0.89",
   "main": "build/index.js",
   "types": "build/index.d.ts",
   "scripts": {

package/tools/enums/tool-type.enum.ts

@@ -11,4 +11,5 @@ export enum TOOL_TYPE {
     VIDEO_EDITOR = 'VIDEO_EDITOR',
     MUSIC = 'MUSIC',
     MUSIC_ACTION = 'MUSIC_ACTION',
+    IMAGE_GENERATION = 'IMAGE_GENERATION',
 }