@purpleschool/gptbot 0.9.18 → 0.9.19-stage-2

package/api/controllers/http/subscription.ts

@@ -5,6 +5,7 @@ export const SUBSCRIPTION_PUBLIC_CONTROLLER = 'subscription/public' as const;
 export const SUBSCRIPTION_ROUTES = {
     FIND_BY_UUID: 'by/uuid',
     GET_ALL: 'all',
+    CUSTOM_ALL: 'custom/all',
     BUY: (uuid: string) => `${uuid}/buy`,
     FAST_BUY: 'fast/buy',
     GET_MY: 'my',
@@ -14,7 +15,6 @@ export const SUBSCRIPTION_ROUTES = {
     GET_AVAILABLE_DOWNGRADES: 'available-downgrades',
     UPGRADE: 'upgrade',
     DOWNGRADE: 'downgrade',
-    CREATE_CUSTOM: 'custom',
     FREE: 'free',
     SUMMARY: 'summary',
     RECOVER_PAST_DUE: (uuid: string) => `${uuid}/recover-past-due`,

package/api/routes.ts

@@ -151,12 +151,12 @@ export const REST_API = {
     },
     SUBSCRIPTION_PUBLIC: {
         GET: `${ROOT}/${CONTROLLERS.SUBSCRIPTION_PUBLIC_CONTROLLER}/${CONTROLLERS.SUBSCRIPTION_ROUTES.GET_ALL}`,
+        CUSTOM_ALL: `${ROOT}/${CONTROLLERS.SUBSCRIPTION_PUBLIC_CONTROLLER}/${CONTROLLERS.SUBSCRIPTION_ROUTES.CUSTOM_ALL}`,
         GET_BY_UUID: (uuid: string): string =>
             `${ROOT}/${CONTROLLERS.SUBSCRIPTION_PUBLIC_CONTROLLER}/${CONTROLLERS.SUBSCRIPTION_ROUTES.FIND_BY_UUID}/${uuid}`,
         FAST_BUY: (uuid: string) =>
             `${ROOT}/${CONTROLLERS.SUBSCRIPTION_PUBLIC_CONTROLLER}/${CONTROLLERS.SUBSCRIPTION_ROUTES.FAST_BUY}/${uuid}`,
         FREE: `${ROOT}/${CONTROLLERS.SUBSCRIPTION_PUBLIC_CONTROLLER}/${CONTROLLERS.SUBSCRIPTION_ROUTES.FREE}`,
-        CREATE_CUSTOM: `${ROOT}/${CONTROLLERS.SUBSCRIPTION_PUBLIC_CONTROLLER}/${CONTROLLERS.SUBSCRIPTION_ROUTES.CREATE_CUSTOM}`,
     },
     SUBSCRIPTION_PRIVATE: {
         GET: `${ROOT}/${CONTROLLERS.SUBSCRIPTION_PRIVATE_CONTROLLER}/${CONTROLLERS.SUBSCRIPTION_ROUTES.GET_ALL}`,
@@ -176,7 +176,6 @@ export const REST_API = {
             `${ROOT}/${CONTROLLERS.SUBSCRIPTION_PRIVATE_CONTROLLER}/${uuid}/${CONTROLLERS.SUBSCRIPTION_ROUTES.DOWNGRADE}`,
         GET_AVAILABLE_UPGRADES: `${ROOT}/${CONTROLLERS.SUBSCRIPTION_PRIVATE_CONTROLLER}/${CONTROLLERS.SUBSCRIPTION_ROUTES.GET_AVAILABLE_UPGRADES}`,
         GET_AVAILABLE_DOWNGRADES: `${ROOT}/${CONTROLLERS.SUBSCRIPTION_PRIVATE_CONTROLLER}/${CONTROLLERS.SUBSCRIPTION_ROUTES.GET_AVAILABLE_DOWNGRADES}`,
-        CREATE_CUSTOM: `${ROOT}/${CONTROLLERS.SUBSCRIPTION_PRIVATE_CONTROLLER}/${CONTROLLERS.SUBSCRIPTION_ROUTES.CREATE_CUSTOM}`,
         SUMMARY: `${ROOT}/${CONTROLLERS.SUBSCRIPTION_PRIVATE_CONTROLLER}/${CONTROLLERS.SUBSCRIPTION_ROUTES.SUMMARY}`,
         RECOVER_PAST_DUE: (uuid: string) =>
             `${ROOT}/${CONTROLLERS.SUBSCRIPTION_PRIVATE_CONTROLLER}/${CONTROLLERS.SUBSCRIPTION_ROUTES.RECOVER_PAST_DUE(uuid)}`,

package/build/api/controllers/http/subscription.js

@@ -6,6 +6,7 @@ exports.SUBSCRIPTION_PUBLIC_CONTROLLER = 'subscription/public';
 exports.SUBSCRIPTION_ROUTES = {
     FIND_BY_UUID: 'by/uuid',
     GET_ALL: 'all',
+    CUSTOM_ALL: 'custom/all',
     BUY: (uuid) => `${uuid}/buy`,
     FAST_BUY: 'fast/buy',
     GET_MY: 'my',
@@ -15,7 +16,6 @@ exports.SUBSCRIPTION_ROUTES = {
     GET_AVAILABLE_DOWNGRADES: 'available-downgrades',
     UPGRADE: 'upgrade',
     DOWNGRADE: 'downgrade',
-    CREATE_CUSTOM: 'custom',
     FREE: 'free',
     SUMMARY: 'summary',
     RECOVER_PAST_DUE: (uuid) => `${uuid}/recover-past-due`,

package/build/api/routes.js

@@ -159,10 +159,10 @@ exports.REST_API = {
     },
     SUBSCRIPTION_PUBLIC: {
         GET: `${exports.ROOT}/${CONTROLLERS.SUBSCRIPTION_PUBLIC_CONTROLLER}/${CONTROLLERS.SUBSCRIPTION_ROUTES.GET_ALL}`,
+        CUSTOM_ALL: `${exports.ROOT}/${CONTROLLERS.SUBSCRIPTION_PUBLIC_CONTROLLER}/${CONTROLLERS.SUBSCRIPTION_ROUTES.CUSTOM_ALL}`,
         GET_BY_UUID: (uuid) => `${exports.ROOT}/${CONTROLLERS.SUBSCRIPTION_PUBLIC_CONTROLLER}/${CONTROLLERS.SUBSCRIPTION_ROUTES.FIND_BY_UUID}/${uuid}`,
         FAST_BUY: (uuid) => `${exports.ROOT}/${CONTROLLERS.SUBSCRIPTION_PUBLIC_CONTROLLER}/${CONTROLLERS.SUBSCRIPTION_ROUTES.FAST_BUY}/${uuid}`,
         FREE: `${exports.ROOT}/${CONTROLLERS.SUBSCRIPTION_PUBLIC_CONTROLLER}/${CONTROLLERS.SUBSCRIPTION_ROUTES.FREE}`,
-        CREATE_CUSTOM: `${exports.ROOT}/${CONTROLLERS.SUBSCRIPTION_PUBLIC_CONTROLLER}/${CONTROLLERS.SUBSCRIPTION_ROUTES.CREATE_CUSTOM}`,
     },
     SUBSCRIPTION_PRIVATE: {
         GET: `${exports.ROOT}/${CONTROLLERS.SUBSCRIPTION_PRIVATE_CONTROLLER}/${CONTROLLERS.SUBSCRIPTION_ROUTES.GET_ALL}`,
@@ -177,7 +177,6 @@ exports.REST_API = {
         DOWNGRADE: (uuid) => `${exports.ROOT}/${CONTROLLERS.SUBSCRIPTION_PRIVATE_CONTROLLER}/${uuid}/${CONTROLLERS.SUBSCRIPTION_ROUTES.DOWNGRADE}`,
         GET_AVAILABLE_UPGRADES: `${exports.ROOT}/${CONTROLLERS.SUBSCRIPTION_PRIVATE_CONTROLLER}/${CONTROLLERS.SUBSCRIPTION_ROUTES.GET_AVAILABLE_UPGRADES}`,
         GET_AVAILABLE_DOWNGRADES: `${exports.ROOT}/${CONTROLLERS.SUBSCRIPTION_PRIVATE_CONTROLLER}/${CONTROLLERS.SUBSCRIPTION_ROUTES.GET_AVAILABLE_DOWNGRADES}`,
-        CREATE_CUSTOM: `${exports.ROOT}/${CONTROLLERS.SUBSCRIPTION_PRIVATE_CONTROLLER}/${CONTROLLERS.SUBSCRIPTION_ROUTES.CREATE_CUSTOM}`,
         SUMMARY: `${exports.ROOT}/${CONTROLLERS.SUBSCRIPTION_PRIVATE_CONTROLLER}/${CONTROLLERS.SUBSCRIPTION_ROUTES.SUMMARY}`,
         RECOVER_PAST_DUE: (uuid) => `${exports.ROOT}/${CONTROLLERS.SUBSCRIPTION_PRIVATE_CONTROLLER}/${CONTROLLERS.SUBSCRIPTION_ROUTES.RECOVER_PAST_DUE(uuid)}`,
         REFUND: (uuid) => `${exports.ROOT}/${CONTROLLERS.SUBSCRIPTION_PRIVATE_CONTROLLER}/${CONTROLLERS.SUBSCRIPTION_ROUTES.REFUND(uuid)}`,

package/build/commands/cabinet/get-user-statistics-by-month.command.js

@@ -7,7 +7,10 @@ var GetUserStatisticsByMonthCommand;
 (function (GetUserStatisticsByMonthCommand) {
     GetUserStatisticsByMonthCommand.RequestSchema = zod_1.z.object({
         origin: zod_1.z.nativeEnum(constants_1.JOB_REQUEST_ORIGIN).default(constants_1.JOB_REQUEST_ORIGIN.API).optional(),
-        metric: zod_1.z.nativeEnum(constants_1.STATISTICS_METRIC_TYPE).default(constants_1.STATISTICS_METRIC_TYPE.REQUESTS).optional(),
+        metric: zod_1.z
+            .nativeEnum(constants_1.STATISTICS_METRIC_TYPE)
+            .default(constants_1.STATISTICS_METRIC_TYPE.REQUESTS)
+            .optional(),
     });
     GetUserStatisticsByMonthCommand.UserStatisticsByMonthResponseSchema = zod_1.z.object({
         month: zod_1.z.string(),

package/build/commands/chat/create-chat.command.js

@@ -6,7 +6,6 @@ const zod_1 = require("zod");
 var CreateChatCommand;
 (function (CreateChatCommand) {
     CreateChatCommand.RequestSchema = models_1.ChatSchema.pick({
-        aIModelId: true,
         categoryId: true,
     }).extend({
         folderId: zod_1.z.string().uuid().optional(),

package/build/commands/chat/get-last-active-chat-command.js

@@ -7,7 +7,6 @@ var GetLastActiveChatCommand;
 (function (GetLastActiveChatCommand) {
     GetLastActiveChatCommand.RequestQuerySchema = models_1.ChatSchema.pick({
         categoryId: true,
-        aIModelId: true,
     });
     GetLastActiveChatCommand.Response = zod_1.z.object({
         data: models_1.ChatWithMessagesSchema,

package/build/commands/message/create-text-message.command.js

@@ -10,6 +10,7 @@ var CreateTextMessageCommand;
         text: zod_1.z.string(),
         files: zod_1.z.array(zod_1.z.string().uuid()).optional().default([]),
         features: zod_1.z.array(zod_1.z.nativeEnum(constants_1.AI_MODEL_FEATURE)).optional().default([]),
+        aiModelId: zod_1.z.string().uuid(),
     });
     CreateTextMessageCommand.RequestParamSchema = models_1.ChatSchema.pick({
         uuid: true,

package/build/commands/subscription/index.js

@@ -16,7 +16,6 @@ var __exportStar = (this && this.__exportStar) || function(m, exports) {
 Object.defineProperty(exports, "__esModule", { value: true });
 __exportStar(require("./buy-subscription.command"), exports);
 __exportStar(require("./cancel-subscription.command"), exports);
-__exportStar(require("./create-custom-subscription-plan.command"), exports);
 __exportStar(require("./create-subscription.command"), exports);
 __exportStar(require("./delete-subscription.command"), exports);
 __exportStar(require("./find-subscription.command"), exports);

package/build/constants/ai-model/enums/ai-model-content-type.enum.js

@@ -1,7 +1,8 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.AI_MODEL_CONTENT_TYPE = void 0;
-exports.AI_MODEL_CONTENT_TYPE = {
-    TEXT: 'TEXT',
-    IMAGE: 'IMAGE',
-};
+var AI_MODEL_CONTENT_TYPE;
+(function (AI_MODEL_CONTENT_TYPE) {
+    AI_MODEL_CONTENT_TYPE["TEXT"] = "TEXT";
+    AI_MODEL_CONTENT_TYPE["IMAGE"] = "IMAGE";
+})(AI_MODEL_CONTENT_TYPE || (exports.AI_MODEL_CONTENT_TYPE = AI_MODEL_CONTENT_TYPE = {}));

package/build/constants/model/enums/unified-model-content-type.enum.js

@@ -5,7 +5,6 @@ var UNIFIED_MODEL_CONTENT_TYPE;
 (function (UNIFIED_MODEL_CONTENT_TYPE) {
     // AI Model types
     UNIFIED_MODEL_CONTENT_TYPE["TEXT"] = "TEXT";
-    UNIFIED_MODEL_CONTENT_TYPE["IMAGE"] = "IMAGE";
     // Tool types
     UNIFIED_MODEL_CONTENT_TYPE["IMAGE_EDITOR"] = "IMAGE_EDITOR";
     UNIFIED_MODEL_CONTENT_TYPE["TEXT_TO_SPEECH"] = "TEXT_TO_SPEECH";

package/build/constants/subscription/enums/subscription-feature-type.enum.js

@@ -11,6 +11,7 @@ var SUBSCRIPTION_FEATURE_TYPE;
     SUBSCRIPTION_FEATURE_TYPE["ADVANCED_TOOLS_ACCESS"] = "advanced_tools_access";
     SUBSCRIPTION_FEATURE_TYPE["STT_MODEL_ACCESS"] = "stt_model_access";
     SUBSCRIPTION_FEATURE_TYPE["TTS_MODEL_ACCESS"] = "tts_model_access";
+    SUBSCRIPTION_FEATURE_TYPE["IMAGE_GENERATION_ACCESS"] = "image_generation_access";
     SUBSCRIPTION_FEATURE_TYPE["IMAGE_GENERATION_QUOTA"] = "image_generation_quota";
     SUBSCRIPTION_FEATURE_TYPE["TTS_QUOTA"] = "tts_quota";
     SUBSCRIPTION_FEATURE_TYPE["STT_QUOTA"] = "stt_quota";
@@ -33,7 +34,7 @@ var SUBSCRIPTION_FEATURE_KIND;
 (function (SUBSCRIPTION_FEATURE_KIND) {
     SUBSCRIPTION_FEATURE_KIND["TEXT"] = "text";
     SUBSCRIPTION_FEATURE_KIND["AI_MODEL"] = "ai_model";
-    SUBSCRIPTION_FEATURE_KIND["IMAGE"] = "image";
+    SUBSCRIPTION_FEATURE_KIND["IMAGE_GENERATION"] = "image_generation";
     SUBSCRIPTION_FEATURE_KIND["TTS"] = "tts";
     SUBSCRIPTION_FEATURE_KIND["STT"] = "stt";
     SUBSCRIPTION_FEATURE_KIND["PRESENTATIONS"] = "presentations";

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

@@ -15,7 +15,7 @@ exports.AiModelSchema = zod_1.z.object({
     order: zod_1.z.number(),
     canUse: zod_1.z.optional(zod_1.z.boolean()),
     status: zod_1.z.nativeEnum(constants_1.AI_MODEL_STATUS),
-    contentType: zod_1.z.enum(Object.values(constants_1.AI_MODEL_CONTENT_TYPE)),
+    contentType: zod_1.z.nativeEnum(constants_1.AI_MODEL_CONTENT_TYPE),
     features: zod_1.z.array(zod_1.z.nativeEnum(constants_1.AI_MODEL_FEATURE)),
     iconVariants: zod_1.z.object({
         light: zod_1.z.object({

package/build/models/chat.schema.js

@@ -8,7 +8,7 @@ exports.ChatSchema = zod_1.z.object({
     userId: zod_1.z.nullable(zod_1.z.string().uuid()),
     unregisteredUserId: zod_1.z.nullable(zod_1.z.string().uuid()),
     categoryId: zod_1.z.string().uuid(),
-    aIModelId: zod_1.z.string().uuid(),
+    aIModelId: zod_1.z.string().uuid().nullable(),
     title: zod_1.z.string(),
     chatStatus: zod_1.z.enum(Object.values(constants_1.CHAT_STATUS_ENUM)),
     folderId: zod_1.z.string().nullable(),

package/build/models/message.schema.js

@@ -9,6 +9,7 @@ exports.MessageSchema = zod_1.z.object({
     text: zod_1.z.string().max(10000),
     chatId: zod_1.z.nullable(zod_1.z.string().uuid()),
     role: zod_1.z.string(),
+    aiModelId: zod_1.z.string().nullable(),
     tokenUsage: zod_1.z.number(),
     userReaction: zod_1.z.nativeEnum(constants_1.USER_REACTION).nullable(),
     files: zod_1.z.array(file_schema_1.FileSchema),

package/build/models/prompt.schema.js

@@ -2,11 +2,9 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.PromptSchema = void 0;
 const zod_1 = require("zod");
-const constants_1 = require("../constants");
 exports.PromptSchema = zod_1.z.object({
     uuid: zod_1.z.string().uuid(),
     content: zod_1.z.string(),
-    type: zod_1.z.nativeEnum(constants_1.AI_MODEL_CONTENT_TYPE),
     topicId: zod_1.z.string().uuid().nullable(),
     title: zod_1.z.string().nullable(),
     createdAt: zod_1.z.date(),

package/build/models/subscription-feature.schema.js

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.SubscriptionFeatureSchema = exports.InteriorDesignModelAccessFeatureSchema = exports.InteriorDesignQuotaFeatureSchema = exports.ImageEditorModelAccessFeatureSchema = exports.ImageEditorQuotaFeatureSchema = exports.WriterModelAccessFeatureSchema = exports.WriterQuotaFeatureSchema = exports.MusicModelAccessFeatureSchema = exports.MusicQuotaFeatureSchema = exports.PresentationQuotaFeatureSchema = exports.VideoModelAccessFeatureSchema = exports.VideoQuotaFeatureSchema = exports.STTQuotaFeatureSchema = exports.STTModelAccessFeatureSchema = exports.TTSQuotaFeatureSchema = exports.ImageGenerationQuotaFeatureSchema = exports.TtsModelAccessFeatureSchema = exports.AdvancedToolsAccessFeatureSchema = exports.WebSearchFeatureSchema = exports.ExtendedContextFeatureSchema = exports.MaxInputLengthFeatureSchema = exports.RequestsQuotaFeatureSchema = exports.AiModelAccessFeatureSchema = exports.CarryoverBalanceFeatureSchema = exports.TokensFeatureSchema = exports.SubscriptionFeatureBaseSchema = void 0;
+exports.SubscriptionFeatureSchema = exports.InteriorDesignModelAccessFeatureSchema = exports.InteriorDesignQuotaFeatureSchema = exports.ImageEditorModelAccessFeatureSchema = exports.ImageEditorQuotaFeatureSchema = exports.WriterModelAccessFeatureSchema = exports.WriterQuotaFeatureSchema = exports.MusicModelAccessFeatureSchema = exports.MusicQuotaFeatureSchema = exports.PresentationQuotaFeatureSchema = exports.VideoModelAccessFeatureSchema = exports.VideoQuotaFeatureSchema = exports.STTQuotaFeatureSchema = exports.STTModelAccessFeatureSchema = exports.TTSQuotaFeatureSchema = exports.ImageGenerationQuotaFeatureSchema = exports.TtsModelAccessFeatureSchema = exports.AdvancedToolsAccessFeatureSchema = exports.WebSearchFeatureSchema = exports.ExtendedContextFeatureSchema = exports.MaxInputLengthFeatureSchema = exports.RequestsQuotaFeatureSchema = exports.ImageGenerationAccessFeatureSchema = exports.AiModelAccessFeatureSchema = exports.CarryoverBalanceFeatureSchema = exports.TokensFeatureSchema = exports.SubscriptionFeatureBaseSchema = void 0;
 const constants_1 = require("../constants");
 const zod_1 = require("zod");
 exports.SubscriptionFeatureBaseSchema = zod_1.z.object({
@@ -40,6 +40,14 @@ exports.AiModelAccessFeatureSchema = exports.SubscriptionFeatureBaseSchema.exten
     isAvailable: zod_1.z.boolean(),
     maxInputLength: zod_1.z.number(),
 });
+exports.ImageGenerationAccessFeatureSchema = exports.SubscriptionFeatureBaseSchema.extend({
+    type: zod_1.z.literal(constants_1.SUBSCRIPTION_FEATURE_TYPE.IMAGE_GENERATION_ACCESS),
+    kind: zod_1.z.literal(constants_1.SUBSCRIPTION_FEATURE_KIND.IMAGE_GENERATION),
+    order: zod_1.z.number(),
+    modelId: zod_1.z.string(),
+    isAvailable: zod_1.z.boolean(),
+    maxInputLength: zod_1.z.number(),
+});
 exports.RequestsQuotaFeatureSchema = exports.SubscriptionFeatureBaseSchema.extend({
     type: zod_1.z.literal(constants_1.SUBSCRIPTION_FEATURE_TYPE.TEXT_GENERATION_QUOTA),
     kind: zod_1.z.literal(constants_1.SUBSCRIPTION_FEATURE_KIND.TEXT),
@@ -74,7 +82,7 @@ exports.TtsModelAccessFeatureSchema = exports.SubscriptionFeatureBaseSchema.exte
     modelId: zod_1.z.string(),
 });
 exports.ImageGenerationQuotaFeatureSchema = exports.SubscriptionFeatureBaseSchema.extend({
-    kind: zod_1.z.literal(constants_1.SUBSCRIPTION_FEATURE_KIND.IMAGE),
+    kind: zod_1.z.literal(constants_1.SUBSCRIPTION_FEATURE_KIND.IMAGE_GENERATION),
     type: zod_1.z.literal(constants_1.SUBSCRIPTION_FEATURE_TYPE.IMAGE_GENERATION_QUOTA),
     value: zod_1.z.number(),
 });
@@ -179,4 +187,5 @@ exports.SubscriptionFeatureSchema = zod_1.z.union([
     exports.ImageEditorModelAccessFeatureSchema,
     exports.InteriorDesignQuotaFeatureSchema,
     exports.InteriorDesignModelAccessFeatureSchema,
+    exports.ImageGenerationAccessFeatureSchema,
 ]);

package/commands/cabinet/get-user-statistics-by-month.command.ts

@@ -2,10 +2,12 @@ import { z } from 'zod';
 import { JOB_REQUEST_ORIGIN, STATISTICS_METRIC_TYPE } from '../../constants';
 
 export namespace GetUserStatisticsByMonthCommand {
-
     export const RequestSchema = z.object({
         origin: z.nativeEnum(JOB_REQUEST_ORIGIN).default(JOB_REQUEST_ORIGIN.API).optional(),
-        metric: z.nativeEnum(STATISTICS_METRIC_TYPE).default(STATISTICS_METRIC_TYPE.REQUESTS).optional(),
+        metric: z
+            .nativeEnum(STATISTICS_METRIC_TYPE)
+            .default(STATISTICS_METRIC_TYPE.REQUESTS)
+            .optional(),
     });
 
     export type Request = z.infer<typeof RequestSchema>;

package/commands/chat/create-chat.command.ts

@@ -3,7 +3,6 @@ import { z } from 'zod';
 
 export namespace CreateChatCommand {
     export const RequestSchema = ChatSchema.pick({
-        aIModelId: true,
         categoryId: true,
     }).extend({
         folderId: z.string().uuid().optional(),

package/commands/chat/get-last-active-chat-command.ts

@@ -4,7 +4,6 @@ import { z } from 'zod';
 export namespace GetLastActiveChatCommand {
     export const RequestQuerySchema = ChatSchema.pick({
         categoryId: true,
-        aIModelId: true,
     });
 
     export const Response = z.object({

package/commands/message/create-text-message.command.ts

@@ -7,6 +7,7 @@ export namespace CreateTextMessageCommand {
         text: z.string(),
         files: z.array(z.string().uuid()).optional().default([]),
         features: z.array(z.nativeEnum(AI_MODEL_FEATURE)).optional().default([]),
+        aiModelId: z.string().uuid(),
     });
 
     export const RequestParamSchema = ChatSchema.pick({

package/commands/subscription/index.ts

@@ -1,6 +1,5 @@
 export * from './buy-subscription.command';
 export * from './cancel-subscription.command';
-export * from './create-custom-subscription-plan.command';
 export * from './create-subscription.command';
 export * from './delete-subscription.command';
 export * from './find-subscription.command';

package/constants/ai-model/enums/ai-model-content-type.enum.ts

@@ -1,7 +1,4 @@
-export const AI_MODEL_CONTENT_TYPE = {
-    TEXT: 'TEXT',
-    IMAGE: 'IMAGE',
-} as const;
-
-export type TAIModelContentTypeEnum =
-    (typeof AI_MODEL_CONTENT_TYPE)[keyof typeof AI_MODEL_CONTENT_TYPE];
+export enum AI_MODEL_CONTENT_TYPE {
+    TEXT = 'TEXT',
+    IMAGE = 'IMAGE',
+}

package/constants/cabinet/enums/index.ts

@@ -1,2 +1,2 @@
 export * from './group-by.enum';
-export * from './statistics-metric-type.enum';
+export * from './statistics-metric-type.enum';

package/constants/cabinet/enums/statistics-metric-type.enum.ts

@@ -1,4 +1,4 @@
 export enum STATISTICS_METRIC_TYPE {
-  REQUESTS = 'requests',
-  TOKENS = 'tokens',
-}
+    REQUESTS = 'requests',
+    TOKENS = 'tokens',
+}

package/constants/model/enums/unified-model-content-type.enum.ts

@@ -1,7 +1,6 @@
 export enum UNIFIED_MODEL_CONTENT_TYPE {
     // AI Model types
     TEXT = 'TEXT',
-    IMAGE = 'IMAGE',
     // Tool types
     IMAGE_EDITOR = 'IMAGE_EDITOR',
     TEXT_TO_SPEECH = 'TEXT_TO_SPEECH',

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

@@ -7,6 +7,7 @@ export enum SUBSCRIPTION_FEATURE_TYPE {
     ADVANCED_TOOLS_ACCESS = 'advanced_tools_access',
     STT_MODEL_ACCESS = 'stt_model_access',
     TTS_MODEL_ACCESS = 'tts_model_access',
+    IMAGE_GENERATION_ACCESS = 'image_generation_access',
     IMAGE_GENERATION_QUOTA = 'image_generation_quota',
     TTS_QUOTA = 'tts_quota',
     STT_QUOTA = 'stt_quota',
@@ -29,7 +30,7 @@ export enum SUBSCRIPTION_FEATURE_TYPE {
 export enum SUBSCRIPTION_FEATURE_KIND {
     TEXT = 'text',
     AI_MODEL = 'ai_model',
-    IMAGE = 'image',
+    IMAGE_GENERATION = 'image_generation',
     TTS = 'tts',
     STT = 'stt',
     PRESENTATIONS = 'presentations',

package/models/ai-model.schema.ts

@@ -1,10 +1,5 @@
 import { z } from 'zod';
-import {
-    AI_MODEL_CONTENT_TYPE,
-    AI_MODEL_FEATURE,
-    AI_MODEL_STATUS,
-    TAIModelContentTypeEnum,
-} from '../constants';
+import { AI_MODEL_CONTENT_TYPE, AI_MODEL_FEATURE, AI_MODEL_STATUS } from '../constants';
 import { AiModelConfigSchema } from './ai-model-config.schema';
 
 export const AiModelSchema = z.object({
@@ -18,7 +13,7 @@ export const AiModelSchema = z.object({
     order: z.number(),
     canUse: z.optional(z.boolean()),
     status: z.nativeEnum(AI_MODEL_STATUS),
-    contentType: z.enum(Object.values(AI_MODEL_CONTENT_TYPE) as [TAIModelContentTypeEnum]),
+    contentType: z.nativeEnum(AI_MODEL_CONTENT_TYPE),
     features: z.array(z.nativeEnum(AI_MODEL_FEATURE)),
     iconVariants: z.object({
         light: z.object({

package/models/chat.schema.ts

@@ -6,7 +6,7 @@ export const ChatSchema = z.object({
     userId: z.nullable(z.string().uuid()),
     unregisteredUserId: z.nullable(z.string().uuid()),
     categoryId: z.string().uuid(),
-    aIModelId: z.string().uuid(),
+    aIModelId: z.string().uuid().nullable(),
     title: z.string(),
     chatStatus: z.enum(Object.values(CHAT_STATUS_ENUM) as [TChatStatusEnum]),
     folderId: z.string().nullable(),

package/models/message.schema.ts

@@ -7,6 +7,7 @@ export const MessageSchema = z.object({
     text: z.string().max(10000),
     chatId: z.nullable(z.string().uuid()),
     role: z.string(),
+    aiModelId: z.string().nullable(),
     tokenUsage: z.number(),
     userReaction: z.nativeEnum(USER_REACTION).nullable(),
     files: z.array(FileSchema),

package/models/prompt.schema.ts

@@ -1,10 +1,8 @@
 import { z } from 'zod';
-import { AI_MODEL_CONTENT_TYPE } from '../constants';
 
 export const PromptSchema = z.object({
     uuid: z.string().uuid(),
     content: z.string(),
-    type: z.nativeEnum(AI_MODEL_CONTENT_TYPE),
     topicId: z.string().uuid().nullable(),
     title: z.string().nullable(),
 

package/models/subscription-feature.schema.ts

@@ -51,6 +51,16 @@ export const AiModelAccessFeatureSchema = SubscriptionFeatureBaseSchema.extend({
 });
 export type AiModelAccessFeature = z.infer<typeof AiModelAccessFeatureSchema>;
 
+export const ImageGenerationAccessFeatureSchema = SubscriptionFeatureBaseSchema.extend({
+    type: z.literal(SUBSCRIPTION_FEATURE_TYPE.IMAGE_GENERATION_ACCESS),
+    kind: z.literal(SUBSCRIPTION_FEATURE_KIND.IMAGE_GENERATION),
+    order: z.number(),
+    modelId: z.string(),
+    isAvailable: z.boolean(),
+    maxInputLength: z.number(),
+});
+export type ImageGenerationAccessFeature = z.infer<typeof ImageGenerationAccessFeatureSchema>;
+
 export const RequestsQuotaFeatureSchema = SubscriptionFeatureBaseSchema.extend({
     type: z.literal(SUBSCRIPTION_FEATURE_TYPE.TEXT_GENERATION_QUOTA),
     kind: z.literal(SUBSCRIPTION_FEATURE_KIND.TEXT),
@@ -97,7 +107,7 @@ export const TtsModelAccessFeatureSchema = SubscriptionFeatureBaseSchema.extend(
 export type TtsModelAccessFeature = z.infer<typeof TtsModelAccessFeatureSchema>;
 
 export const ImageGenerationQuotaFeatureSchema = SubscriptionFeatureBaseSchema.extend({
-    kind: z.literal(SUBSCRIPTION_FEATURE_KIND.IMAGE),
+    kind: z.literal(SUBSCRIPTION_FEATURE_KIND.IMAGE_GENERATION),
     type: z.literal(SUBSCRIPTION_FEATURE_TYPE.IMAGE_GENERATION_QUOTA),
     value: z.number(),
 });
@@ -234,6 +244,7 @@ export const SubscriptionFeatureSchema = z.union([
     ImageEditorModelAccessFeatureSchema,
     InteriorDesignQuotaFeatureSchema,
     InteriorDesignModelAccessFeatureSchema,
+    ImageGenerationAccessFeatureSchema,
 ]);
 
 export type SubscriptionFeature = z.infer<typeof SubscriptionFeatureSchema>;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@purpleschool/gptbot",
-  "version": "0.9.18",
+  "version": "0.9.19-stage-2",
   "description": "",
   "main": "build/index.js",
   "types": "build/index.d.ts",

package/build/commands/subscription/create-custom-subscription-plan.command.js

@@ -1,24 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.CreateCustomSubscriptionPlanCommand = void 0;
-const zod_1 = require("zod");
-const models_1 = require("../../models");
-var CreateCustomSubscriptionPlanCommand;
-(function (CreateCustomSubscriptionPlanCommand) {
-    CreateCustomSubscriptionPlanCommand.RequestSchema = zod_1.z.object({
-        tokens: zod_1.z
-            .number()
-            .refine((val) => val % 500 === 0, {
-            message: 'tokens must be a multiple of 500',
-        })
-            .refine((val) => val > 1000, {
-            message: 'tokens must be greater than 1000',
-        })
-            .refine((val) => val <= 10000, {
-            message: 'tokens must be equal or less than 10000',
-        }),
-    });
-    CreateCustomSubscriptionPlanCommand.ResponseSchema = zod_1.z.object({
-        data: zod_1.z.array(models_1.SubscriptionSchema),
-    });
-})(CreateCustomSubscriptionPlanCommand || (exports.CreateCustomSubscriptionPlanCommand = CreateCustomSubscriptionPlanCommand = {}));

package/commands/subscription/create-custom-subscription-plan.command.ts

@@ -1,26 +0,0 @@
-import { z } from 'zod';
-import { SubscriptionSchema } from '../../models';
-
-export namespace CreateCustomSubscriptionPlanCommand {
-    export const RequestSchema = z.object({
-        tokens: z
-            .number()
-            .refine((val) => val % 500 === 0, {
-                message: 'tokens must be a multiple of 500',
-            })
-            .refine((val) => val > 1000, {
-                message: 'tokens must be greater than 1000',
-            })
-            .refine((val) => val <= 10000, {
-                message: 'tokens must be equal or less than 10000',
-            }),
-    });
-
-    export type Request = z.infer<typeof RequestSchema>;
-
-    export const ResponseSchema = z.object({
-        data: z.array(SubscriptionSchema),
-    });
-
-    export type Response = z.infer<typeof ResponseSchema>;
-}