@purpleschool/gptbot-tools 0.0.118 → 0.0.119-stage-2

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

@@ -25,6 +25,7 @@ exports.TTSJobSchema = zod_1.z.object({
     duration: zod_1.z.number(),
     userId: zod_1.z.string().nullable(),
     unregisteredUserId: zod_1.z.string().nullable(),
+    tokenReservationId: zod_1.z.string().nullable().optional(),
     isDeleted: zod_1.z.boolean(),
     params: exports.TTSJobParamsSchema,
     createdAt: zod_1.z.date(),

package/build/tts/routes/tts.amqp.routes.js

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

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/update-published-status-video-jobs.command.js

@@ -12,8 +12,14 @@ var UpdatePublishedStatusVideoJobsCommand;
             userId: zod_1.z.string().uuid().nullable().optional(),
             unregisteredUserId: zod_1.z.string().uuid().nullable().optional(),
         }),
-        data: zod_1.z.object({
+        data: zod_1.z
+            .object({
             isPublished: zod_1.z.boolean(),
+            postId: zod_1.z.string().nullable().optional(),
+        })
+            .refine((data) => (data.isPublished ? !!data.postId : true), {
+            message: 'Post ID is required when job is published',
+            path: ['postId'],
         }),
     });
     UpdatePublishedStatusVideoJobsCommand.ResponseSchema = (0, command_response_schema_1.ICommandResponseSchema)(models_1.VideoJobSchema);

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

@@ -17,6 +17,8 @@ exports.VideoJobSchema = zod_1.z.object({
     uuid: zod_1.z.string(),
     prompt: zod_1.z.string(),
     reaction: zod_1.z.nativeEnum(common_1.USER_REACTION).nullable(),
+    isPublished: zod_1.z.boolean(),
+    postId: zod_1.z.string().nullable(),
     dislikeReason: zod_1.z.string().nullable(),
     externalId: zod_1.z.string().nullable(),
     title: zod_1.z.string(),
@@ -29,6 +31,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/commands/update-published-status-video-editor-jobs.command.js

@@ -12,8 +12,14 @@ var UpdatePublishedStatusVideoEditorJobsCommand;
             userId: zod_1.z.string().uuid().nullable().optional(),
             unregisteredUserId: zod_1.z.string().uuid().nullable().optional(),
         }),
-        data: zod_1.z.object({
+        data: zod_1.z
+            .object({
             isPublished: zod_1.z.boolean(),
+            postId: zod_1.z.string().nullable().optional(),
+        })
+            .refine((data) => (data.isPublished ? !!data.postId : true), {
+            message: 'Post ID is required when job is published',
+            path: ['postId'],
         }),
     });
     UpdatePublishedStatusVideoEditorJobsCommand.ResponseSchema = (0, command_response_schema_1.ICommandResponseSchema)(video_editor_job_schema_1.VideoEditorJobSchema);

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

@@ -23,7 +23,9 @@ 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(),
     createdAt: zod_1.z.date(),
     completedAt: zod_1.z.date().nullable().optional(),

package/build/video-editor/queries/find-video-editor-job-by-id.query.js

@@ -6,14 +6,10 @@ const command_response_schema_1 = require("../../common/models/command-response.
 const video_editor_job_schema_1 = require("../models/video-editor-job.schema");
 var FindVideoEditorJobByIdQuery;
 (function (FindVideoEditorJobByIdQuery) {
-    FindVideoEditorJobByIdQuery.RequestSchema = zod_1.z
-        .object({
+    FindVideoEditorJobByIdQuery.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',
     });
     FindVideoEditorJobByIdQuery.ResponseSchema = (0, command_response_schema_1.ICommandResponseSchema)(video_editor_job_schema_1.VideoEditorJobSchema);
 })(FindVideoEditorJobByIdQuery || (exports.FindVideoEditorJobByIdQuery = FindVideoEditorJobByIdQuery = {}));

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

@@ -21,6 +21,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/commands/update-published-status-image-editor-jobs.command.ts

@@ -9,9 +9,15 @@ export namespace UpdatePublishedStatusImageEditorJobsCommand {
             userId: z.string().uuid().nullable().optional(),
             unregisteredUserId: z.string().uuid().nullable().optional(),
         }),
-        data: z.object({
-            isPublished: z.boolean(),
-        }),
+        data: z
+            .object({
+                isPublished: z.boolean(),
+                postId: z.string().nullable().optional(),
+            })
+            .refine((data) => (data.isPublished ? !!data.postId : true), {
+                message: 'Post ID is required when job is published',
+                path: ['postId'],
+            }),
     });
     export type Request = z.infer<typeof RequestSchema>;
 

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

@@ -27,7 +27,9 @@ export const ImageEditorJobSchema = z.object({
     userId: z.string().nullable().optional(),
     unregisteredUserId: z.string().nullable().optional(),
     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/commands/update-published-status-image-generation-jobs.command.ts

@@ -9,9 +9,15 @@ export namespace UpdatePublishedStatusImageGenerationJobsCommand {
             userId: z.string().uuid().nullable().optional(),
             unregisteredUserId: z.string().uuid().nullable().optional(),
         }),
-        data: z.object({
-            isPublished: z.boolean(),
-        }),
+        data: z
+            .object({
+                isPublished: z.boolean(),
+                postId: z.string().nullable().optional(),
+            })
+            .refine((data) => (data.isPublished ? !!data.postId : true), {
+                message: 'Post ID is required when job is published',
+                path: ['postId'],
+            }),
     });
     export type Request = z.infer<typeof RequestSchema>;
 

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

@@ -29,8 +29,10 @@ export const ImageGenerationJobSchema = z.object({
     userId: z.string().nullable().optional(),
     unregisteredUserId: z.string().nullable().optional(),
     isPublished: z.boolean(),
+    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/queries/find-image-generation-job-by-id.query.ts

@@ -3,15 +3,11 @@ import { ICommandResponseSchema } from '../../common/models/command-response.sch
 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 const RequestSchema = z.object({
+        uuid: z.string(),
+        userId: z.string().uuid().nullable().optional(),
+        unregisteredUserId: z.string().uuid().nullable().optional(),
+    });
     export type Request = z.infer<typeof RequestSchema>;
 
     export const ResponseSchema = ICommandResponseSchema(ImageGenerationJobSchema);

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,10 @@ 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(),
     });
     export type Request = z.infer<typeof RequestSchema>;
 

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

@@ -0,0 +1,21 @@
+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,
+    });
+    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,9 +5,10 @@ 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(),
     });

package/music/commands/update-published-status-music-jobs.command.ts

@@ -9,9 +9,15 @@ export namespace UpdatePublishedStatusMusicJobsCommand {
             userId: z.string().uuid().nullable().optional(),
             unregisteredUserId: z.string().uuid().nullable().optional(),
         }),
-        data: z.object({
-            isPublished: z.boolean(),
-        }),
+        data: z
+            .object({
+                isPublished: z.boolean(),
+                postId: z.string().nullable().optional(),
+            })
+            .refine((data) => (data.isPublished ? !!data.postId : true), {
+                message: 'Post ID is required when job is published',
+                path: ['postId'],
+            }),
     });
     export type Request = z.infer<typeof RequestSchema>;
 

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

@@ -48,7 +48,9 @@ export const MusicJobSchema = z.object({
     userId: z.string().nullable().optional(),
     unregisteredUserId: z.string().nullable().optional(),
     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/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.118",
+  "version": "0.0.119-stage-2",
   "main": "build/index.js",
   "types": "build/index.d.ts",
   "scripts": {