@purpleschool/gptbot-tools 0.0.128 → 0.0.130-stage-2

package/build/video/commands/get-video-price.command.js

@@ -0,0 +1,17 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.GetVideoPriceCommand = void 0;
+const zod_1 = require("zod");
+const command_response_schema_1 = require("../../common/models/command-response.schema");
+const models_1 = require("../models");
+var GetVideoPriceCommand;
+(function (GetVideoPriceCommand) {
+    GetVideoPriceCommand.RequestSchema = zod_1.z.object({
+        modelId: zod_1.z.string(),
+        params: models_1.VideoGenerationRequestParamsSchema,
+    });
+    GetVideoPriceCommand.ResponseDataSchema = zod_1.z.object({
+        price: zod_1.z.number(),
+    });
+    GetVideoPriceCommand.ResponseSchema = (0, command_response_schema_1.ICommandResponseSchema)(GetVideoPriceCommand.ResponseDataSchema);
+})(GetVideoPriceCommand || (exports.GetVideoPriceCommand = GetVideoPriceCommand = {}));

package/build/video/commands/index.js

@@ -19,6 +19,7 @@ __exportStar(require("./delete-all-video-jobs.command"), exports);
 __exportStar(require("./retry-video-job.command"), exports);
 __exportStar(require("./set-reaction-to-video-job.command"), exports);
 __exportStar(require("./video.command"), exports);
+__exportStar(require("./get-video-price.command"), exports);
 __exportStar(require("./update-video-job-title.command"), exports);
 __exportStar(require("./video-model"), exports);
 __exportStar(require("./update-published-status-video-jobs.command"), exports);

package/build/video/commands/retry-video-job.command.js

@@ -9,12 +9,13 @@ var RetryVideoJobCommand;
 (function (RetryVideoJobCommand) {
     RetryVideoJobCommand.RequestSchema = zod_1.z.object({
         uuid: zod_1.z.string().uuid(),
-        userBalance: zod_1.z.number(),
         userId: zod_1.z.string().uuid().nullable().optional(),
         unregisteredUserId: zod_1.z.string().uuid().nullable().optional(),
         prompt: zod_1.z.string().optional(),
         modelId: zod_1.z.string().uuid().optional(),
         params: models_1.VideoGenerationRequestParamsSchema.optional(),
+        tokenReservationId: zod_1.z.string().uuid(),
+        precalculatedPrice: zod_1.z.number(),
     });
     RetryVideoJobCommand.ResponseSchema = (0, command_response_schema_1.ICommandResponseSchema)(video_job_schema_1.VideoJobSchema);
 })(RetryVideoJobCommand || (exports.RetryVideoJobCommand = RetryVideoJobCommand = {}));

package/build/video/commands/video.command.js

@@ -11,9 +11,10 @@ var VideoCommand;
         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(),
         params: models_1.VideoGenerationRequestParamsSchema,
+        tokenReservationId: zod_1.z.string().uuid(),
+        precalculatedPrice: zod_1.z.number(),
     });
     VideoCommand.ResponseSchema = (0, command_response_schema_1.ICommandResponseSchema)(video_job_schema_1.VideoJobSchema);
 })(VideoCommand || (exports.VideoCommand = VideoCommand = {}));

package/build/video/models/video-job.schema.js

@@ -33,6 +33,7 @@ exports.VideoJobSchema = zod_1.z.object({
     attempts: zod_1.z.array(zod_1.z.any()),
     userId: zod_1.z.string().nullable().optional(),
     unregisteredUserId: zod_1.z.string().nullable().optional(),
+    tokenReservationId: zod_1.z.string().nullable().optional(),
     isDeleted: zod_1.z.boolean(),
     createdAt: zod_1.z.date(),
     completedAt: zod_1.z.date().nullable().optional(),

package/build/video/routes/video.amqp.routes.js

@@ -3,6 +3,7 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.VIDEO_AMQP_ROUTES = void 0;
 exports.VIDEO_AMQP_ROUTES = {
     EXECUTE: 'tools.video.execute.rpc',
+    GET_PRICE: 'tools.video.get-price.rpc',
     CONFIG: 'tools.video.config.rpc',
     GET_JOB: 'tools.video.jobs.get.rpc',
     LIST_JOBS: 'tools.video.jobs.list.rpc',

package/build/video-editor/commands/edit-video.command.js

@@ -13,8 +13,9 @@ var EditVideoCommand;
         inputVideoId: zod_1.z.string().uuid(),
         inputVideoUrl: zod_1.z.string(),
         duration: zod_1.z.number(),
-        userBalance: zod_1.z.number(),
         modelId: zod_1.z.string().uuid(),
+        tokenReservationId: zod_1.z.string().uuid(),
+        precalculatedPrice: zod_1.z.number(),
     });
     EditVideoCommand.ResponseSchema = (0, command_response_schema_1.ICommandResponseSchema)(models_1.VideoEditorJobSchema);
 })(EditVideoCommand || (exports.EditVideoCommand = EditVideoCommand = {}));

package/build/video-editor/commands/get-video-editor-price.command.js

@@ -0,0 +1,15 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.GetVideoEditorPriceCommand = void 0;
+const zod_1 = require("zod");
+const command_response_schema_1 = require("../../common/models/command-response.schema");
+var GetVideoEditorPriceCommand;
+(function (GetVideoEditorPriceCommand) {
+    GetVideoEditorPriceCommand.RequestSchema = zod_1.z.object({
+        modelId: zod_1.z.string(),
+    });
+    GetVideoEditorPriceCommand.ResponseDataSchema = zod_1.z.object({
+        price: zod_1.z.number(),
+    });
+    GetVideoEditorPriceCommand.ResponseSchema = (0, command_response_schema_1.ICommandResponseSchema)(GetVideoEditorPriceCommand.ResponseDataSchema);
+})(GetVideoEditorPriceCommand || (exports.GetVideoEditorPriceCommand = GetVideoEditorPriceCommand = {}));

package/build/video-editor/commands/index.js

@@ -21,4 +21,5 @@ __exportStar(require("./set-reaction-to-video-editor-job.command"), exports);
 __exportStar(require("./update-video-editor-job-title.command"), exports);
 __exportStar(require("./update-published-status-video-editor-jobs.command"), exports);
 __exportStar(require("./edit-video.command"), exports);
+__exportStar(require("./get-video-editor-price.command"), exports);
 __exportStar(require("./video-editor-model"), exports);

package/build/video-editor/commands/retry-video-editor-job.command.js

@@ -8,7 +8,6 @@ var RetryVideoEditorJobCommand;
 (function (RetryVideoEditorJobCommand) {
     RetryVideoEditorJobCommand.RequestSchema = zod_1.z.object({
         uuid: zod_1.z.string().uuid(),
-        userBalance: zod_1.z.number(),
         userId: zod_1.z.string().uuid().nullable().optional(),
         unregisteredUserId: zod_1.z.string().uuid().nullable().optional(),
         prompt: zod_1.z.string().optional(),
@@ -16,6 +15,8 @@ var RetryVideoEditorJobCommand;
         inputVideoUrl: zod_1.z.string().optional(),
         duration: zod_1.z.number().optional(),
         modelId: zod_1.z.string().uuid().optional(),
+        tokenReservationId: zod_1.z.string().uuid(),
+        precalculatedPrice: zod_1.z.number(),
     });
     RetryVideoEditorJobCommand.ResponseSchema = (0, command_response_schema_1.ICommandResponseSchema)(video_editor_job_schema_1.VideoEditorJobSchema);
 })(RetryVideoEditorJobCommand || (exports.RetryVideoEditorJobCommand = RetryVideoEditorJobCommand = {}));

package/build/video-editor/models/video-editor-job.schema.js

@@ -23,6 +23,7 @@ exports.VideoEditorJobSchema = zod_1.z.object({
     attempts: zod_1.z.array(zod_1.z.any()),
     userId: zod_1.z.string().nullable().optional(),
     unregisteredUserId: zod_1.z.string().nullable().optional(),
+    tokenReservationId: zod_1.z.string().nullable().optional(),
     isPublished: zod_1.z.boolean(),
     postId: zod_1.z.string().nullable(),
     isDeleted: zod_1.z.boolean(),

package/build/video-editor/routes/video-editor.amqp.routes.js

@@ -3,6 +3,7 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.VIDEO_EDITOR_AMQP_ROUTES = void 0;
 exports.VIDEO_EDITOR_AMQP_ROUTES = {
     EXECUTE: 'tools.video-editor.execute.rpc',
+    GET_PRICE: 'tools.video-editor.get-price.rpc',
     CONFIG: 'tools.video-editor.config.rpc',
     GET_JOB: 'tools.video-editor.jobs.get.rpc',
     LIST_JOBS: 'tools.video-editor.jobs.list.rpc',

package/build/writer/commands/generate-document-contents.command.js

@@ -10,7 +10,8 @@ var GenerateWriterDocumentContentsCommand;
         documentId: zod_1.z.string().uuid(),
         userId: zod_1.z.string().uuid().nullable().optional(),
         unregisteredUserId: zod_1.z.string().uuid().nullable().optional(),
-        userBalance: zod_1.z.number(),
+        tokenReservationId: zod_1.z.string().uuid(),
+        precalculatedPrice: zod_1.z.number(),
         modelId: zod_1.z.string(),
     });
     GenerateWriterDocumentContentsCommand.ResponseSchema = (0, common_1.ICommandResponseSchema)(models_1.WriterDocumentSchema);

package/build/writer/commands/get-writer-content-generation-price.command.js

@@ -0,0 +1,16 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.GetWriterContentGenerationPriceCommand = void 0;
+const zod_1 = require("zod");
+const common_1 = require("../../common");
+var GetWriterContentGenerationPriceCommand;
+(function (GetWriterContentGenerationPriceCommand) {
+    GetWriterContentGenerationPriceCommand.RequestSchema = zod_1.z.object({
+        documentId: zod_1.z.string().uuid(),
+        modelId: zod_1.z.string().uuid(),
+    });
+    GetWriterContentGenerationPriceCommand.ResponseDataSchema = zod_1.z.object({
+        price: zod_1.z.number(),
+    });
+    GetWriterContentGenerationPriceCommand.ResponseSchema = (0, common_1.ICommandResponseSchema)(GetWriterContentGenerationPriceCommand.ResponseDataSchema);
+})(GetWriterContentGenerationPriceCommand || (exports.GetWriterContentGenerationPriceCommand = GetWriterContentGenerationPriceCommand = {}));

package/build/writer/commands/index.js

@@ -20,6 +20,7 @@ __exportStar(require("./delete-all-user-writer-documents.command"), exports);
 __exportStar(require("./update-writer-document-outline.command"), exports);
 __exportStar(require("./export-writer-document-as-docx.command"), exports);
 __exportStar(require("./generate-document-contents.command"), exports);
+__exportStar(require("./get-writer-content-generation-price.command"), exports);
 __exportStar(require("./writer-paraphrase.command"), exports);
 __exportStar(require("./writer-extend-text.command"), exports);
 __exportStar(require("./writer-shorten-text.command"), exports);

package/build/writer/models/writer-document.schema.js

@@ -23,6 +23,7 @@ exports.WriterDocumentSchema = zod_1.z.object({
     dislikeReason: zod_1.z.string().nullable(),
     pages: zod_1.z.number(),
     isDeleted: zod_1.z.boolean(),
+    tokenReservationId: zod_1.z.string().nullable().optional(),
     createdAt: zod_1.z.date(),
     updatedAt: zod_1.z.date(),
     completedAt: zod_1.z.date().nullable(),

package/build/writer/routes/writer.routes.js

@@ -11,6 +11,7 @@ exports.WRITER_AMQP_ROUTES = {
     UPDATE_DOCUMENT_OUTLINE: 'tools.writer.update-document-outline.rpc',
     UPDATE_DOCUMENT_CONTENTS: 'tools.writer.update-document-contents.rpc',
     GENERATE_DOCUMENT_CONTENTS: 'tools.writer.generate-document-contents.rpc',
+    GET_CONTENT_GENERATION_PRICE: 'tools.writer.content-generation.get-price.rpc',
     PARAPHRASE: 'tools.writer.actions.paraphrase.rpc',
     EXTEND_TEXT: 'tools.writer.actions.extend-text.rpc',
     SHORTEN_TEXT: 'tools.writer.actions.shorten-text.rpc',

package/image-editor/commands/get-image-editor-price.command.ts

@@ -0,0 +1,26 @@
+import { z } from 'zod';
+import { ICommandResponseSchema } from '../../common/models/command-response.schema';
+import { AttachedFileSchema } from '../../common';
+
+export namespace GetImageEditorPriceCommand {
+    export const RequestSchema = z.object({
+        modelId: z.string().uuid(),
+        userId: z.string().uuid().nullable().optional(),
+        unregisteredUserId: z.string().uuid().nullable().optional(),
+        prompt: z.string(),
+        params: z.object({
+            attachedFiles: z.array(AttachedFileSchema),
+            systemPromptId: z.string().optional(),
+        }),
+        userHasActiveSubscriptionOrProduct: z.boolean(),
+    });
+    export type Request = z.infer<typeof RequestSchema>;
+
+    export const ResponseDataSchema = z.object({
+        price: z.number(),
+    });
+    export type ResponseData = z.infer<typeof ResponseDataSchema>;
+
+    export const ResponseSchema = ICommandResponseSchema(ResponseDataSchema);
+    export type Response = z.infer<typeof ResponseSchema>;
+}

package/image-editor/commands/image-editor.command.ts

@@ -8,13 +8,14 @@ export namespace ImageEditorCommand {
         userId: z.string().uuid().nullable().optional(),
         unregisteredUserId: z.string().uuid().nullable().optional(),
         prompt: z.string(),
-        userBalance: z.number(),
         modelId: z.string(),
         params: z.object({
             attachedFiles: z.array(AttachedFileSchema),
             systemPromptId: z.string().optional(),
         }),
         userHasActiveSubscriptionOrProduct: z.boolean(),
+        tokenReservationId: z.string().uuid(),
+        precalculatedPrice: z.number(),
     });
     export type Request = z.infer<typeof RequestSchema>;
 

package/image-editor/commands/index.ts

@@ -1,5 +1,6 @@
 export * from './delete-image-editor-job-by-uuid.command';
 export * from './delete-all-image-editor-jobs.command';
+export * from './get-image-editor-price.command';
 export * from './retry-image-editor-job.command';
 export * from './set-reaction-to-image-editor-job.command';
 export * from './image-editor.command';

package/image-editor/commands/retry-image-editor-job.command.ts

@@ -5,10 +5,11 @@ import { ImageEditorJobParamsSchema, ImageEditorJobSchema } from '../models';
 export namespace RetryImageEditorJobCommand {
     export const RequestSchema = z.object({
         uuid: z.string().uuid(),
-        userBalance: z.number(),
         userId: z.string().uuid().nullable().optional(),
         unregisteredUserId: z.string().uuid().nullable().optional(),
         userHasActiveSubscriptionOrProduct: z.boolean(),
+        tokenReservationId: z.string().uuid(),
+        precalculatedPrice: z.number(),
         modelId: z.string().optional(),
         prompt: z.string().optional(),
         params: ImageEditorJobParamsSchema.optional(),

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

@@ -29,6 +29,7 @@ export const ImageEditorJobSchema = z.object({
     isPublished: z.boolean(),
     postId: z.string().nullable(),
     isDeleted: z.boolean(),
+    tokenReservationId: z.string().nullable().optional(),
     createdAt: z.date(),
     completedAt: z.date().nullable().optional(),
     updatedAt: z.date(),

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

@@ -1,5 +1,6 @@
 export const IMAGE_EDITOR_AMQP_ROUTES = {
     EXECUTE: 'tools.image.execute.rpc',
+    GET_PRICE: 'tools.image.get-price.rpc',
     CONFIG: 'tools.image.config.rpc',
     GET_JOB: 'tools.image.jobs.get.rpc',
     LIST_JOBS: 'tools.image.jobs.list.rpc',

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

@@ -1,19 +1,20 @@
 import { z } from 'zod';
 import { ICommandResponseSchema } from '../../common/models/command-response.schema';
-import { ImageGenerationJobSchema, ImageGenerationRequestParamsSchema } from '../models';
 import { JOB_REQUEST_ORIGIN } from '../../tools';
+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(),
         origin: z.nativeEnum(JOB_REQUEST_ORIGIN).default(JOB_REQUEST_ORIGIN.API),
         userHasActiveSubscriptionOrProduct: z.boolean(),
+        tokenReservationId: z.string().uuid(),
+        precalculatedPrice: z.number(),
     });
     export type Request = z.infer<typeof RequestSchema>;
 

package/image-generation/commands/get-image-generation-price.command.ts

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

package/image-generation/commands/index.ts

@@ -1,5 +1,6 @@
 export * from './execute-image-generation.command';
 export * from './forward-image-generation-job.command';
+export * from './get-image-generation-price.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';

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

@@ -7,12 +7,13 @@ export namespace RetryImageGenerationJobCommand {
         uuid: z.string().uuid(),
         userId: z.string().uuid().nullable().optional(),
         unregisteredUserId: z.string().uuid().nullable().optional(),
-        prompt: z.string().optional(),
-        userBalance: z.number(),
+        userHasActiveSubscriptionOrProduct: z.boolean(),
+        tokenReservationId: z.string().uuid(),
+        precalculatedPrice: z.number(),
         modelId: z.string().uuid().optional(),
+        prompt: z.string().optional(),
         params: ImageGenerationRequestParamsSchema.optional(),
         presetId: z.string().uuid().nullable().optional(),
-        userHasActiveSubscriptionOrProduct: z.boolean(),
     });
     export type Request = z.infer<typeof RequestSchema>;
 

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

@@ -32,6 +32,7 @@ export const ImageGenerationJobSchema = z.object({
     postId: z.string().nullable(),
     isDeleted: z.boolean(),
     internalError: z.string().nullable(),
+    tokenReservationId: z.string().nullable().optional(),
     createdAt: z.date(),
     updatedAt: z.date(),
 });

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

@@ -1,5 +1,6 @@
 export const IMAGE_GENERATION_AMQP_ROUTES = {
     EXECUTE: 'tools.image-generation.execute.rpc',
+    GET_PRICE: 'tools.image-generation.get-price.rpc',
     CONFIG: 'tools.image-generation.config.rpc',
     FORWARD_JOB: 'tools.image-generation.jobs.forward.rpc',
     GET_JOB: 'tools.image-generation.jobs.get.rpc',

package/music/commands/generate-music.command.ts

@@ -7,9 +7,11 @@ export namespace GenerateMusicCommand {
     export const RequestSchema = z.object({
         userId: z.string().uuid().nullable().optional(),
         unregisteredUserId: z.string().uuid().nullable().optional(),
-        userBalance: z.number(),
         modelId: z.string(),
         params: MusicJobParamsSchema,
+        tokenReservationId: z.string().uuid(),
+        precalculatedPrice: z.number(),
+        userHasActiveSubscriptionOrProduct: z.boolean(),
     });
     export type Request = z.infer<typeof RequestSchema>;
 

package/music/commands/get-music-price.command.ts

@@ -0,0 +1,22 @@
+import { z } from 'zod';
+import { ICommandResponseSchema } from '../../common/models/command-response.schema';
+import { MusicJobParamsSchema } from '../models';
+
+export namespace GetMusicPriceCommand {
+    export const RequestSchema = z.object({
+        modelId: z.string(),
+        userId: z.string().uuid().nullable().optional(),
+        unregisteredUserId: z.string().uuid().nullable().optional(),
+        params: MusicJobParamsSchema,
+        userHasActiveSubscriptionOrProduct: z.boolean(),
+    });
+    export type Request = z.infer<typeof RequestSchema>;
+
+    export const ResponseDataSchema = z.object({
+        price: z.number(),
+    });
+    export type ResponseData = z.infer<typeof ResponseDataSchema>;
+
+    export const ResponseSchema = ICommandResponseSchema(ResponseDataSchema);
+    export type Response = z.infer<typeof ResponseSchema>;
+}

package/music/commands/index.ts

@@ -3,6 +3,7 @@ export * from './delete-all-music-jobs.command';
 export * from './retry-music-job.command';
 export * from './set-reaction-to-music-job.command';
 export * from './generate-music.command';
+export * from './get-music-price.command';
 export * from './update-music-job-title.command';
 export * from './convert-to-wav.command';
 export * from './generate-lyrics.command';

package/music/commands/retry-music-job.command.ts

@@ -5,11 +5,13 @@ import { MusicJobParamsSchema, MusicJobSchema } from '../models/music-job.schema
 export namespace RetryMusicJobCommand {
     export const RequestSchema = z.object({
         uuid: z.string().uuid(),
-        userBalance: z.number(),
         userId: z.string().uuid().nullable().optional(),
         unregisteredUserId: z.string().uuid().nullable().optional(),
+        tokenReservationId: z.string().uuid(),
+        precalculatedPrice: z.number(),
         modelId: z.string().uuid().optional(),
         params: MusicJobParamsSchema.optional(),
+        userHasActiveSubscriptionOrProduct: z.boolean(),
     });
     export type Request = z.infer<typeof RequestSchema>;
 

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

@@ -48,6 +48,7 @@ export const MusicJobSchema = z.object({
     userId: z.string().nullable().optional(),
     unregisteredUserId: z.string().nullable().optional(),
     isDeleted: z.boolean(),
+    tokenReservationId: z.string().nullable().optional(),
     createdAt: z.date(),
     completedAt: z.date().nullable().optional(),
     updatedAt: z.date(),

package/music/models/music-model.schema.ts

@@ -20,6 +20,19 @@ export const MusicModelParamsSchema = z.object({
 
 export type MusicModelParams = z.infer<typeof MusicModelParamsSchema>;
 
+export const MusicModelPricingRuleConditionSchema = z.object({
+    withoutSub: z.boolean().optional(),
+});
+export type MusicModelPricingRuleCondition = z.infer<typeof MusicModelPricingRuleConditionSchema>;
+
+export const MusicModelPricingRulesSchema = z.array(
+    z.object({
+        condition: MusicModelPricingRuleConditionSchema,
+        value: z.number(),
+    }),
+);
+export type MusicModelPricingRules = z.infer<typeof MusicModelPricingRulesSchema>;
+
 export const MusicModelSchema = z.object({
     uuid: z.string(),
     title: z.string(),
@@ -33,6 +46,7 @@ export const MusicModelSchema = z.object({
     maxPromptLength: z.number(),
     maxLyricsLength: z.number(),
     params: MusicModelParamsSchema,
+    pricingRules: MusicModelPricingRulesSchema,
     createdAt: z.date(),
     updatedAt: z.date(),
 });

package/music/models/music-track.schema.ts

@@ -9,6 +9,7 @@ export const MusicTrackSchema = z.object({
     prompt: z.string().nullable(),
     tags: z.string(),
     duration: z.number(),
+    waveformPeaks: z.array(z.number()),
     isPublished: z.boolean().optional(),
     postId: z.string().nullable().optional(),
 

package/music/routes/music.amqp.routes.ts

@@ -1,5 +1,6 @@
 export const MUSIC_AMQP_ROUTES = {
     EXECUTE: 'tools.music.execute.rpc',
+    GET_PRICE: 'tools.music.get-price.rpc',
     GENERATE_LYRICS: 'tools.music.generate-lyrics.rpc',
     CONVERT_TO_WAV: 'tools.music.convert-to-wav.rpc',
     CONFIG: 'tools.music.config.rpc',

package/package.json

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

package/paraphrase/commands/get-paraphrase-price.command.ts

@@ -0,0 +1,16 @@
+import { z } from 'zod';
+import { ICommandResponseSchema } from '../../common/models/command-response.schema';
+
+export namespace GetParaphrasePriceCommand {
+    export const RequestSchema = z.object({
+        typeId: z.string().uuid(),
+    });
+    export type Request = z.infer<typeof RequestSchema>;
+
+    export const ResponseDataSchema = z.object({
+        price: z.number(),
+    });
+
+    export const ResponseSchema = ICommandResponseSchema(ResponseDataSchema);
+    export type Response = z.infer<typeof ResponseSchema>;
+}

package/paraphrase/commands/index.ts

@@ -4,3 +4,4 @@ export * from './delete-all-paraphrase-jobs.command';
 export * from './set-reaction-to-paraphrase-job.command';
 export * from './update-paraphrase-job-title.command';
 export * from './retry-paraphrase-job.command';
+export * from './get-paraphrase-price.command';

package/paraphrase/commands/paraphrase.command.ts

@@ -7,11 +7,12 @@ export namespace ParaphraseCommand {
     export const RequestSchema = z.object({
         userId: z.string().uuid().nullable().optional(),
         unregisteredUserId: z.string().uuid().nullable().optional(),
-        userBalance: z.number(),
         prompt: z.string(),
         typeId: z.string().uuid(),
         styleId: z.string().uuid(),
         intensity: z.nativeEnum(PARAPHRASING_INTENSITY),
+        tokenReservationId: z.string().uuid(),
+        precalculatedPrice: z.number(),
     });
     export type Request = z.infer<typeof RequestSchema>;
 

package/paraphrase/commands/retry-paraphrase-job.command.ts

@@ -5,9 +5,10 @@ import { ParaphraseJobSchema } from '../models/paraphrase-job.schema';
 export namespace RetryParaphraseJobCommand {
     export const RequestSchema = z.object({
         uuid: z.string().uuid(),
-        userBalance: z.number(),
         userId: z.string().uuid().nullable().optional(),
         unregisteredUserId: z.string().uuid().nullable().optional(),
+        tokenReservationId: z.string().uuid(),
+        precalculatedPrice: z.number(),
     });
     export type Request = z.infer<typeof RequestSchema>;
 

package/paraphrase/models/paraphrase-job.schema.ts

@@ -19,6 +19,7 @@ export const ParaphraseJobSchema = z.object({
     typeId: z.string(),
     styleId: z.string(),
     intensity: z.nativeEnum(PARAPHRASING_INTENSITY),
+    tokenReservationId: z.string().nullable().optional(),
     createdAt: z.date(),
     updatedAt: z.date(),
     completedAt: z.date().nullable(),

package/paraphrase/routes/paraphrase.amqp.routes.ts

@@ -1,5 +1,6 @@
 export const PARAPHRASE_AMQP_ROUTES = {
     EXECUTE: 'tools.paraphrase.execute.rpc',
+    GET_PRICE: 'tools.paraphrase.get-price.rpc',
     CONFIG: 'tools.paraphrase.config.rpc',
     GET_JOB: 'tools.paraphrase.jobs.get.rpc',
     LIST_JOBS: 'tools.paraphrase.jobs.list.rpc',

package/presentation/commands/generate-slides.command.ts

@@ -7,7 +7,8 @@ export namespace GenerateSlidesCommand {
         uuid: z.string().uuid(),
         userId: z.string().uuid().nullable().optional(),
         unregisteredUserId: z.string().uuid().nullable().optional(),
-        userBalance: z.number(),
+        tokenReservationId: z.string().uuid(),
+        precalculatedPrice: z.number(),
     });
     export type Request = z.infer<typeof RequestSchema>;
 

package/presentation/commands/get-presentation-slides-generation-price.command.ts

@@ -0,0 +1,16 @@
+import { z } from 'zod';
+import { ICommandResponseSchema } from '../../common';
+
+export namespace GetPresentationSlidesGenerationPriceCommand {
+    export const RequestSchema = z.object({
+        slideCount: z.number().int().min(1),
+    });
+    export type Request = z.infer<typeof RequestSchema>;
+
+    export const ResponseDataSchema = z.object({
+        price: z.number(),
+    });
+
+    export const ResponseSchema = ICommandResponseSchema(ResponseDataSchema);
+    export type Response = z.infer<typeof ResponseSchema>;
+}

package/presentation/commands/index.ts

@@ -7,6 +7,7 @@ export * from './generate-presentation-outline.command';
 export * from './generate-and-insert-slide.command';
 export * from './export-as-pptx.command';
 export * from './generate-slides.command';
+export * from './get-presentation-slides-generation-price.command';
 export * from './reposition-slide-outline.command';
 export * from './update-slide-outline.command';
 export * from './update-presentation.command';

package/presentation/models/presentation.schema.ts

@@ -19,6 +19,7 @@ export const PresentationSchema = z.object({
     unregisteredUserId: z.string().uuid().nullable().optional(),
     slideCount: z.number(),
     isDeleted: z.boolean(),
+    tokenReservationId: z.string().nullable().optional(),
     lastContentUpdateAt: z.date().nullable(),
     lastPptxExportedAt: z.date().nullable(),
     createdAt: z.date(),

package/presentation/routes/presentation.routes.ts

@@ -4,6 +4,7 @@ export const PRESENTATION_AMQP_ROUTES = {
     FIND_BY_UUID: 'tools.presentation.find-by-uuid.rpc',
     GENERATE_PRESENTATION_OUTLINE: 'tools.presentation.generate-outline.rpc',
     GENERATE_PRESENTATION_SLIDES: 'tools.presentation.generate-slides.rpc',
+    GET_SLIDES_GENERATION_PRICE: 'tools.presentation.slides.get-price.rpc',
     GENERATE_AND_INSERT_SLIDE: 'tools.presentation.generate-and-insert-slide.rpc',
     UPDATE_PRESENTATION_OUTLINE: 'tools.presentation.update-presentation-outline.rpc',
     LIST_PRESENTATIONS: 'tools.presentation.list-presentations.rpc',