@purpleschool/gptbot-tools 0.0.97 → 0.0.99

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

@@ -0,0 +1,32 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ForwardImageGenerationJobCommand = void 0;
+const zod_1 = require("zod");
+const command_response_schema_1 = require("../../common/models/command-response.schema");
+const models_1 = require("../models");
+const job_status_enum_1 = require("../../tools/enums/job-status.enum");
+const user_reaction_enum_1 = require("../../common/enums/user-reaction.enum");
+var ForwardImageGenerationJobCommand;
+(function (ForwardImageGenerationJobCommand) {
+    ForwardImageGenerationJobCommand.RequestSchema = zod_1.z.object({
+        uuid: zod_1.z.string().uuid(),
+        prompt: zod_1.z.string(),
+        title: zod_1.z.string(),
+        status: zod_1.z.nativeEnum(job_status_enum_1.JOB_STATUS),
+        error: zod_1.z.string().nullable(),
+        price: zod_1.z.number(),
+        reaction: zod_1.z.nativeEnum(user_reaction_enum_1.USER_REACTION).nullable(),
+        externalId: zod_1.z.string().nullable(),
+        output: zod_1.z.array(zod_1.z.string()),
+        params: models_1.ImageGenerationJobParamsSchema,
+        modelId: zod_1.z.string().uuid(),
+        presetId: zod_1.z.string().uuid().nullable().optional(),
+        userId: zod_1.z.string().uuid().nullable().optional(),
+        unregisteredUserId: zod_1.z.string().uuid().nullable().optional(),
+        isDeleted: zod_1.z.boolean(),
+        internalError: zod_1.z.string().nullable(),
+        createdAt: zod_1.z.coerce.date(),
+        updatedAt: zod_1.z.coerce.date(),
+    });
+    ForwardImageGenerationJobCommand.ResponseSchema = (0, command_response_schema_1.ICommandResponseSchema)(models_1.ImageGenerationJobSchema);
+})(ForwardImageGenerationJobCommand || (exports.ForwardImageGenerationJobCommand = ForwardImageGenerationJobCommand = {}));

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

@@ -15,6 +15,7 @@ var __exportStar = (this && this.__exportStar) || function(m, exports) {
 };
 Object.defineProperty(exports, "__esModule", { value: true });
 __exportStar(require("./execute-image-generation.command"), exports);
+__exportStar(require("./forward-image-generation-job.command"), exports);
 __exportStar(require("./save-image-generation-job.command"), exports);
 __exportStar(require("./soft-delete-image-generation-job-by-uuid.command"), exports);
 __exportStar(require("./soft-delete-image-generation-jobs-by-criteria.command"), exports);

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

@@ -4,6 +4,7 @@ exports.ImageGenerationJobSchema = exports.ImageGenerationJobParamsSchema = expo
 const zod_1 = require("zod");
 const tools_1 = require("../../tools");
 const common_1 = require("../../common");
+const enums_1 = require("../enums");
 exports.ImageGenerationFileAttachmentSchema = zod_1.z.object({
     uuid: zod_1.z.string(),
     url: zod_1.z.string(),
@@ -16,6 +17,7 @@ exports.ImageGenerationJobParamsSchema = zod_1.z.object({
     aspectRatio: zod_1.z.string(),
     attachedFiles: zod_1.z.array(exports.ImageGenerationFileAttachmentSchema),
     enhancePrompt: zod_1.z.boolean(),
+    resolution: zod_1.z.nativeEnum(enums_1.IMAGE_GENERATION_RESOLUTION).optional(),
 });
 exports.ImageGenerationJobSchema = zod_1.z.object({
     uuid: zod_1.z.string(),

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

@@ -9,9 +9,7 @@ exports.ImageGenerationRequestParamsSchema = zod_1.z.object({
     aspectRatio: zod_1.z.string(),
     attachedFiles: zod_1.z.array(image_generation_job_schema_1.ImageGenerationFileAttachmentSchema),
     enhancePrompt: zod_1.z.boolean(),
-    resolution: zod_1.z
-        .nativeEnum(enums_1.IMAGE_GENERATION_RESOLUTION)
-        .default(enums_1.IMAGE_GENERATION_RESOLUTION.MEDIUM),
+    resolution: zod_1.z.nativeEnum(enums_1.IMAGE_GENERATION_RESOLUTION).optional(),
 });
 exports.ImageGenerationModelParamsSchema = zod_1.z.object({
     aspectRatio: zod_1.z.record(zod_1.z.string(), zod_1.z.object({
@@ -19,7 +17,11 @@ exports.ImageGenerationModelParamsSchema = zod_1.z.object({
         height: zod_1.z.number(),
     })),
     quality: zod_1.z.string().optional(),
-    resolution: zod_1.z.nativeEnum(enums_1.IMAGE_GENERATION_RESOLUTION).optional(),
+    resolution: zod_1.z
+        .object({
+        options: zod_1.z.array(zod_1.z.nativeEnum(enums_1.IMAGE_GENERATION_RESOLUTION)),
+    })
+        .optional(),
 });
 exports.ImageModelPricingRuleConditionSchema = zod_1.z.object({
     resolution: zod_1.z.string().optional(),

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

@@ -4,6 +4,7 @@ exports.IMAGE_GENERATION_AMQP_ROUTES = void 0;
 exports.IMAGE_GENERATION_AMQP_ROUTES = {
     EXECUTE: 'tools.image-generation.execute.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',

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

@@ -4,6 +4,7 @@ exports.TOOL_CONTENT_TYPE = void 0;
 var TOOL_CONTENT_TYPE;
 (function (TOOL_CONTENT_TYPE) {
     TOOL_CONTENT_TYPE["IMAGE_EDITOR"] = "IMAGE_EDITOR";
+    TOOL_CONTENT_TYPE["IMAGE_GENERATION"] = "IMAGE_GENERATION";
     TOOL_CONTENT_TYPE["TEXT_TO_SPEECH"] = "TEXT_TO_SPEECH";
     TOOL_CONTENT_TYPE["SPEECH_TO_TEXT"] = "SPEECH_TO_TEXT";
     TOOL_CONTENT_TYPE["REPHRASE"] = "REPHRASE";

package/build/tools/models/global-tools-config.schema.js

@@ -7,6 +7,7 @@ const video_1 = require("../../video");
 const music_1 = require("../../music");
 const writer_1 = require("../../writer");
 const image_editor_1 = require("../../image-editor");
+const image_generation_1 = require("../../image-generation");
 const paraphrase_1 = require("../../paraphrase");
 const presentation_1 = require("../../presentation");
 const video_editor_1 = require("../../video-editor");
@@ -17,6 +18,7 @@ exports.GlobalToolsConfigSchema = zod_1.z.object({
     [enums_1.TOOL_CONTENT_TYPE.MUSIC]: music_1.MusicConfigSchema,
     [enums_1.TOOL_CONTENT_TYPE.WRITER]: writer_1.WriterConfigSchema,
     [enums_1.TOOL_CONTENT_TYPE.IMAGE_EDITOR]: image_editor_1.ImageEditorConfigSchema,
+    [enums_1.TOOL_CONTENT_TYPE.IMAGE_GENERATION]: image_generation_1.ImageGenerationConfigSchema,
     [enums_1.TOOL_CONTENT_TYPE.REPHRASE]: paraphrase_1.ParaphraseToolConfigSchema,
     [enums_1.TOOL_CONTENT_TYPE.PRESENTATION]: presentation_1.PresentationConfigSchema,
     [enums_1.TOOL_CONTENT_TYPE.VIDEO_EDITOR]: video_editor_1.VideoEditorConfigSchema,

package/build/tools/models/tools-with-configs.schema.js

@@ -7,6 +7,7 @@ const video_1 = require("../../video");
 const music_1 = require("../../music");
 const writer_1 = require("../../writer");
 const image_editor_1 = require("../../image-editor");
+const image_generation_1 = require("../../image-generation");
 const paraphrase_1 = require("../../paraphrase");
 const presentation_1 = require("../../presentation");
 const video_editor_1 = require("../../video-editor");
@@ -31,6 +32,10 @@ exports.ToolWithConfigSchema = zod_1.z.discriminatedUnion('contentType', [
         contentType: zod_1.z.literal(enums_1.TOOL_CONTENT_TYPE.IMAGE_EDITOR),
         config: image_editor_1.ImageEditorConfigSchema,
     })),
+    tool_schema_1.ToolSchema.merge(zod_1.z.object({
+        contentType: zod_1.z.literal(enums_1.TOOL_CONTENT_TYPE.IMAGE_GENERATION),
+        config: image_generation_1.ImageGenerationConfigSchema,
+    })),
     tool_schema_1.ToolSchema.merge(zod_1.z.object({
         contentType: zod_1.z.literal(enums_1.TOOL_CONTENT_TYPE.REPHRASE),
         config: paraphrase_1.ParaphraseToolConfigSchema,

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

@@ -0,0 +1,32 @@
+import { z } from 'zod';
+import { ICommandResponseSchema } from '../../common/models/command-response.schema';
+import { ImageGenerationJobSchema, ImageGenerationJobParamsSchema } from '../models';
+import { JOB_STATUS } from '../../tools/enums/job-status.enum';
+import { USER_REACTION } from '../../common/enums/user-reaction.enum';
+
+export namespace ForwardImageGenerationJobCommand {
+    export const RequestSchema = z.object({
+        uuid: z.string().uuid(),
+        prompt: z.string(),
+        title: z.string(),
+        status: z.nativeEnum(JOB_STATUS),
+        error: z.string().nullable(),
+        price: z.number(),
+        reaction: z.nativeEnum(USER_REACTION).nullable(),
+        externalId: z.string().nullable(),
+        output: z.array(z.string()),
+        params: ImageGenerationJobParamsSchema,
+        modelId: z.string().uuid(),
+        presetId: z.string().uuid().nullable().optional(),
+        userId: z.string().uuid().nullable().optional(),
+        unregisteredUserId: z.string().uuid().nullable().optional(),
+        isDeleted: z.boolean(),
+        internalError: z.string().nullable(),
+        createdAt: z.coerce.date(),
+        updatedAt: z.coerce.date(),
+    });
+    export type Request = z.infer<typeof RequestSchema>;
+
+    export const ResponseSchema = ICommandResponseSchema(ImageGenerationJobSchema);
+    export type Response = z.infer<typeof ResponseSchema>;
+}

package/image-generation/commands/index.ts

@@ -1,4 +1,5 @@
 export * from './execute-image-generation.command';
+export * from './forward-image-generation-job.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/models/image-generation-job.schema.ts

@@ -1,6 +1,7 @@
 import { z } from 'zod';
 import { JOB_STATUS } from '../../tools';
 import { USER_REACTION } from '../../common';
+import { IMAGE_GENERATION_RESOLUTION } from '../enums';
 
 export const ImageGenerationFileAttachmentSchema = z.object({
     uuid: z.string(),
@@ -16,6 +17,7 @@ export const ImageGenerationJobParamsSchema = z.object({
     aspectRatio: z.string(),
     attachedFiles: z.array(ImageGenerationFileAttachmentSchema),
     enhancePrompt: z.boolean(),
+    resolution: z.nativeEnum(IMAGE_GENERATION_RESOLUTION).optional(),
 });
 export type ImageGenerationJobParams = z.infer<typeof ImageGenerationJobParamsSchema>;
 

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

@@ -7,9 +7,7 @@ export const ImageGenerationRequestParamsSchema = z.object({
     aspectRatio: z.string(),
     attachedFiles: z.array(ImageGenerationFileAttachmentSchema),
     enhancePrompt: z.boolean(),
-    resolution: z
-        .nativeEnum(IMAGE_GENERATION_RESOLUTION)
-        .default(IMAGE_GENERATION_RESOLUTION.MEDIUM),
+    resolution: z.nativeEnum(IMAGE_GENERATION_RESOLUTION).optional(),
 });
 export type ImageGenerationRequestParams = z.infer<typeof ImageGenerationRequestParamsSchema>;
 
@@ -22,7 +20,11 @@ export const ImageGenerationModelParamsSchema = z.object({
         }),
     ),
     quality: z.string().optional(),
-    resolution: z.nativeEnum(IMAGE_GENERATION_RESOLUTION).optional(),
+    resolution: z
+        .object({
+            options: z.array(z.nativeEnum(IMAGE_GENERATION_RESOLUTION)),
+        })
+        .optional(),
 });
 export type ImageGenerationModelParams = z.infer<typeof ImageGenerationModelParamsSchema>;
 

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

@@ -1,6 +1,7 @@
 export const IMAGE_GENERATION_AMQP_ROUTES = {
     EXECUTE: 'tools.image-generation.execute.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',

package/package.json

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

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

@@ -1,5 +1,6 @@
 export enum TOOL_CONTENT_TYPE {
     IMAGE_EDITOR = 'IMAGE_EDITOR',
+    IMAGE_GENERATION = 'IMAGE_GENERATION',
     TEXT_TO_SPEECH = 'TEXT_TO_SPEECH',
     SPEECH_TO_TEXT = 'SPEECH_TO_TEXT',
     REPHRASE = 'REPHRASE',

package/tools/models/global-tools-config.schema.ts

@@ -4,6 +4,7 @@ import { VideoConfigSchema } from '../../video';
 import { MusicConfigSchema } from '../../music';
 import { WriterConfigSchema } from '../../writer';
 import { ImageEditorConfigSchema } from '../../image-editor';
+import { ImageGenerationConfigSchema } from '../../image-generation';
 import { ParaphraseToolConfigSchema } from '../../paraphrase';
 import { PresentationConfigSchema } from '../../presentation';
 import { VideoEditorConfigSchema } from '../../video-editor';
@@ -15,6 +16,7 @@ export const GlobalToolsConfigSchema = z.object({
     [TOOL_CONTENT_TYPE.MUSIC]: MusicConfigSchema,
     [TOOL_CONTENT_TYPE.WRITER]: WriterConfigSchema,
     [TOOL_CONTENT_TYPE.IMAGE_EDITOR]: ImageEditorConfigSchema,
+    [TOOL_CONTENT_TYPE.IMAGE_GENERATION]: ImageGenerationConfigSchema,
     [TOOL_CONTENT_TYPE.REPHRASE]: ParaphraseToolConfigSchema,
     [TOOL_CONTENT_TYPE.PRESENTATION]: PresentationConfigSchema,
     [TOOL_CONTENT_TYPE.VIDEO_EDITOR]: VideoEditorConfigSchema,

package/tools/models/tools-with-configs.schema.ts

@@ -4,6 +4,7 @@ import { VideoConfigSchema } from '../../video';
 import { MusicConfigSchema } from '../../music';
 import { WriterConfigSchema } from '../../writer';
 import { ImageEditorConfigSchema } from '../../image-editor';
+import { ImageGenerationConfigSchema } from '../../image-generation';
 import { ParaphraseToolConfigSchema } from '../../paraphrase';
 import { PresentationConfigSchema } from '../../presentation';
 import { VideoEditorConfigSchema } from '../../video-editor';
@@ -37,6 +38,12 @@ export const ToolWithConfigSchema = z.discriminatedUnion('contentType', [
             config: ImageEditorConfigSchema,
         }),
     ),
+    ToolSchema.merge(
+        z.object({
+            contentType: z.literal(TOOL_CONTENT_TYPE.IMAGE_GENERATION),
+            config: ImageGenerationConfigSchema,
+        }),
+    ),
     ToolSchema.merge(
         z.object({
             contentType: z.literal(TOOL_CONTENT_TYPE.REPHRASE),