@purpleschool/gptbot-tools 0.0.95 → 0.0.96

package/build/image-generation/enums/image-generation-model-strategy.enum.js

@@ -17,4 +17,5 @@ var IMAGE_GENERATION_MODEL_STRATEGY;
     IMAGE_GENERATION_MODEL_STRATEGY["IMAGEN_IMAGE_GENERATION"] = "IMAGEN_IMAGE_GENERATION";
     // Midjourney
     IMAGE_GENERATION_MODEL_STRATEGY["MIDJOURNEY_IMAGE_GENERATION"] = "MIDJOURNEY_IMAGE_GENERATION";
+    IMAGE_GENERATION_MODEL_STRATEGY["NANO_BANANA_IMAGE_GENERATION"] = "NANO_BANANA_IMAGE_GENERATION";
 })(IMAGE_GENERATION_MODEL_STRATEGY || (exports.IMAGE_GENERATION_MODEL_STRATEGY = IMAGE_GENERATION_MODEL_STRATEGY = {}));

package/build/image-generation/enums/image-generation-models.enum.js

@@ -0,0 +1,8 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.IMAGE_GENERATION_MODELS = void 0;
+var IMAGE_GENERATION_MODELS;
+(function (IMAGE_GENERATION_MODELS) {
+    IMAGE_GENERATION_MODELS["NANO_BANANA"] = "google/nano-banana";
+    IMAGE_GENERATION_MODELS["NANO_BANANA_PRO"] = "nano-banana-pro";
+})(IMAGE_GENERATION_MODELS || (exports.IMAGE_GENERATION_MODELS = IMAGE_GENERATION_MODELS = {}));

package/build/image-generation/enums/image-generation-resolution.enum.js

@@ -0,0 +1,9 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.IMAGE_GENERATION_RESOLUTION = void 0;
+var IMAGE_GENERATION_RESOLUTION;
+(function (IMAGE_GENERATION_RESOLUTION) {
+    IMAGE_GENERATION_RESOLUTION["LOW"] = "1K";
+    IMAGE_GENERATION_RESOLUTION["MEDIUM"] = "2K";
+    IMAGE_GENERATION_RESOLUTION["HIGH"] = "4K";
+})(IMAGE_GENERATION_RESOLUTION || (exports.IMAGE_GENERATION_RESOLUTION = IMAGE_GENERATION_RESOLUTION = {}));

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

@@ -15,3 +15,5 @@ var __exportStar = (this && this.__exportStar) || function(m, exports) {
 };
 Object.defineProperty(exports, "__esModule", { value: true });
 __exportStar(require("./image-generation-model-strategy.enum"), exports);
+__exportStar(require("./image-generation-resolution.enum"), exports);
+__exportStar(require("./image-generation-models.enum"), exports);

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

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.ImageGenerationModelSchema = exports.ImageGenerationModelParamsSchema = exports.ImageGenerationRequestParamsSchema = void 0;
+exports.ImageGenerationModelSchema = exports.ImageModelPricingRulesSchema = exports.ImageModelPricingRuleConditionSchema = exports.ImageGenerationModelParamsSchema = exports.ImageGenerationRequestParamsSchema = void 0;
 const zod_1 = require("zod");
 const common_1 = require("../../common");
 const enums_1 = require("../enums");
@@ -9,6 +9,9 @@ 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),
 });
 exports.ImageGenerationModelParamsSchema = zod_1.z.object({
     aspectRatio: zod_1.z.record(zod_1.z.string(), zod_1.z.object({
@@ -16,7 +19,15 @@ 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(),
 });
+exports.ImageModelPricingRuleConditionSchema = zod_1.z.object({
+    resolution: zod_1.z.string().optional(),
+});
+exports.ImageModelPricingRulesSchema = zod_1.z.array(zod_1.z.object({
+    condition: exports.ImageModelPricingRuleConditionSchema,
+    value: zod_1.z.number(),
+}));
 exports.ImageGenerationModelSchema = zod_1.z.object({
     uuid: zod_1.z.string(),
     title: zod_1.z.string(),
@@ -31,6 +42,7 @@ exports.ImageGenerationModelSchema = zod_1.z.object({
     iconVariants: common_1.IconVariantsSchema,
     supportsImageAttachment: zod_1.z.boolean(),
     maxAttachedImages: zod_1.z.number(),
+    pricingRules: exports.ImageModelPricingRulesSchema,
     createdAt: zod_1.z.date(),
     updatedAt: zod_1.z.date(),
 });

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

@@ -4,4 +4,5 @@ exports.IMAGE_GENERATION_ROUTES = exports.IMAGE_GENERATION_CONTROLLER = void 0;
 exports.IMAGE_GENERATION_CONTROLLER = 'image-generation';
 exports.IMAGE_GENERATION_ROUTES = {
     KIE_MIDJOURNEY_CALLBACK: 'kie/midjourney/callback',
+    KIE_NANO_BANANA_CALLBACK: 'kie/nano-banana/callback',
 };

package/image-generation/enums/image-generation-model-strategy.enum.ts

@@ -17,4 +17,5 @@ export enum IMAGE_GENERATION_MODEL_STRATEGY {
 
     // Midjourney
     MIDJOURNEY_IMAGE_GENERATION = 'MIDJOURNEY_IMAGE_GENERATION',
+    NANO_BANANA_IMAGE_GENERATION = 'NANO_BANANA_IMAGE_GENERATION',
 }

package/image-generation/enums/image-generation-models.enum.ts

@@ -0,0 +1,4 @@
+export enum IMAGE_GENERATION_MODELS {
+    NANO_BANANA = 'google/nano-banana',
+    NANO_BANANA_PRO = 'nano-banana-pro',
+}

package/image-generation/enums/image-generation-resolution.enum.ts

@@ -0,0 +1,5 @@
+export enum IMAGE_GENERATION_RESOLUTION {
+    LOW = '1K',
+    MEDIUM = '2K',
+    HIGH = '4K',
+}

package/image-generation/enums/index.ts

@@ -1 +1,3 @@
 export * from './image-generation-model-strategy.enum';
+export * from './image-generation-resolution.enum';
+export * from './image-generation-models.enum';

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

@@ -1,12 +1,15 @@
 import { z } from 'zod';
 import { IconVariantsSchema, TOOL_MODEL_STATUS } from '../../common';
-import { IMAGE_GENERATION_MODEL_STRATEGY } from '../enums';
+import { IMAGE_GENERATION_MODEL_STRATEGY, IMAGE_GENERATION_RESOLUTION } from '../enums';
 import { ImageGenerationFileAttachmentSchema } from './image-generation-job.schema';
 
 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),
 });
 export type ImageGenerationRequestParams = z.infer<typeof ImageGenerationRequestParamsSchema>;
 
@@ -19,9 +22,23 @@ export const ImageGenerationModelParamsSchema = z.object({
         }),
     ),
     quality: z.string().optional(),
+    resolution: z.nativeEnum(IMAGE_GENERATION_RESOLUTION).optional(),
 });
 export type ImageGenerationModelParams = z.infer<typeof ImageGenerationModelParamsSchema>;
 
+export const ImageModelPricingRuleConditionSchema = z.object({
+    resolution: z.string().optional(),
+});
+export type ImageModelPricingRuleCondition = z.infer<typeof ImageModelPricingRuleConditionSchema>;
+
+export const ImageModelPricingRulesSchema = z.array(
+    z.object({
+        condition: ImageModelPricingRuleConditionSchema,
+        value: z.number(),
+    }),
+);
+export type ImageModelPricingRules = z.infer<typeof ImageModelPricingRulesSchema>;
+
 export const ImageGenerationModelSchema = z.object({
     uuid: z.string(),
     title: z.string(),
@@ -36,6 +53,7 @@ export const ImageGenerationModelSchema = z.object({
     iconVariants: IconVariantsSchema,
     supportsImageAttachment: z.boolean(),
     maxAttachedImages: z.number(),
+    pricingRules: ImageModelPricingRulesSchema,
     createdAt: z.date(),
     updatedAt: z.date(),
 });

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

@@ -2,4 +2,5 @@ export const IMAGE_GENERATION_CONTROLLER = 'image-generation';
 
 export const IMAGE_GENERATION_ROUTES = {
     KIE_MIDJOURNEY_CALLBACK: 'kie/midjourney/callback',
+    KIE_NANO_BANANA_CALLBACK: 'kie/nano-banana/callback',
 } as const;

package/package.json

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