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

package/build/tts/commands/index.js

@@ -19,4 +19,5 @@ __exportStar(require("./delete-all-tts-jobs.command"), exports);
 __exportStar(require("./retry-tts-job.command"), exports);
 __exportStar(require("./set-reaction-to-tts-job.command"), exports);
 __exportStar(require("./tts.command"), exports);
+__exportStar(require("./get-tts-price.command"), exports);
 __exportStar(require("./update-tts-job-title.command"), exports);

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

@@ -8,9 +8,10 @@ var RetryTTSJobCommand;
 (function (RetryTTSJobCommand) {
     RetryTTSJobCommand.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(),
+        tokenReservationId: zod_1.z.string().uuid(),
+        precalculatedPrice: zod_1.z.number(),
     });
     RetryTTSJobCommand.ResponseSchema = (0, command_response_schema_1.ICommandResponseSchema)(tts_job_schema_1.TTSJobSchema);
 })(RetryTTSJobCommand || (exports.RetryTTSJobCommand = RetryTTSJobCommand = {}));

package/build/tts/commands/tts.command.js

@@ -9,7 +9,6 @@ var TTSCommand;
     TTSCommand.RequestSchema = zod_1.z.object({
         userId: zod_1.z.string().uuid().nullable().optional(),
         unregisteredUserId: zod_1.z.string().uuid().nullable().optional(),
-        userBalance: zod_1.z.number(),
         modelId: zod_1.z.string(),
         voiceId: zod_1.z.string(),
         userInput: zod_1.z.string(),
@@ -21,6 +20,8 @@ var TTSCommand;
             style: zod_1.z.number().optional(),
         })
             .optional(),
+        tokenReservationId: zod_1.z.string().uuid(),
+        precalculatedPrice: zod_1.z.number(),
     });
     TTSCommand.ResponseSchema = (0, command_response_schema_1.ICommandResponseSchema)(tts_job_schema_1.TTSJobSchema);
 })(TTSCommand || (exports.TTSCommand = TTSCommand = {}));

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,5 +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

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

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

@@ -29,6 +29,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',
@@ -11,4 +12,5 @@ exports.VIDEO_AMQP_ROUTES = {
     SOFT_DELETE: 'tools.video.jobs.soft-delete.rpc',
     SOFT_DELETE_ALL: 'tools.video.jobs.soft-delete-all.rpc',
     RETRY: 'tools.video.jobs.retry.rpc',
+    UPDATE_JOB: 'tools.video.jobs.update.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

@@ -19,5 +19,7 @@ __exportStar(require("./delete-video-editor-job-by-uuid.command"), exports);
 __exportStar(require("./retry-video-editor-job.command"), exports);
 __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

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

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

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

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

@@ -3,11 +3,13 @@ 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',
     SET_REACTION: 'tools.video-editor.jobs.set-reaction.rpc',
     UPDATE_TITLE: 'tools.video-editor.jobs.update-title.rpc',
+    UPDATE_JOB: 'tools.video-editor.jobs.update.rpc',
     SOFT_DELETE: 'tools.video-editor.jobs.soft-delete.rpc',
     SOFT_DELETE_ALL: 'tools.video-editor.jobs.soft-delete-all.rpc',
     RETRY: 'tools.video-editor.jobs.retry.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,7 +1,9 @@
 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';
 export * from './update-image-editor-job-title.command';
+export * from './update-published-status-image-editor-jobs.command';
 export * from './image-editor-model';

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

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

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

@@ -26,7 +26,9 @@ export const ImageEditorJobSchema = z.object({
     attempts: z.array(z.any()),
     userId: z.string().nullable().optional(),
     unregisteredUserId: z.string().nullable().optional(),
+    isPublished: z.boolean(),
     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,10 +1,12 @@
 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',
     SET_REACTION: 'tools.image.jobs.set-reaction.rpc',
     UPDATE_TITLE: 'tools.image.jobs.update-title.rpc',
+    UPDATE_JOB: 'tools.image.jobs.update.rpc',
     SOFT_DELETE: 'tools.image.jobs.soft-delete.rpc',
     SOFT_DELETE_ALL: 'tools.image.jobs.soft-delete-all.rpc',
     RETRY: 'tools.image.jobs.retry.rpc',

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

@@ -1,17 +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';
 
 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,9 +1,11 @@
 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';
 export * from './set-reaction-to-image-generation-job.command';
 export * from './update-image-generation-job-title.command';
+export * from './update-published-status-image-generation-jobs.command';
 export * from './image-generation-model';
 export * from './retry-image-generation-job.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

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

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

@@ -28,8 +28,10 @@ export const ImageGenerationJobSchema = z.object({
     attempts: z.array(z.any()),
     userId: z.string().nullable().optional(),
     unregisteredUserId: z.string().nullable().optional(),
+    isPublished: z.boolean(),
     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,11 +1,13 @@
 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',
     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',
+    UPDATE_JOB: 'tools.image-generation.jobs.update.rpc',
     SOFT_DELETE: 'tools.image-generation.jobs.soft-delete.rpc',
     SOFT_DELETE_ALL: 'tools.image-generation.jobs.soft-delete-all.rpc',
     GET_MODEL_BY_UUID: 'tools.image-generation.model.get-by-uuid.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,7 +3,9 @@ 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';
 export * from './music-model';
+export * from './update-published-status-music-jobs.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(),
     });