@purpleschool/gptbot 0.5.82 → 0.5.84

package/build/commands/message/create-image-message.command.js

@@ -6,12 +6,14 @@ const zod_1 = require("zod");
 const constants_1 = require("../../constants");
 var CreateImageMessageCommand;
 (function (CreateImageMessageCommand) {
-    CreateImageMessageCommand.RequestSchema = zod_1.z.intersection(models_1.MessageSchema.pick({ text: true }), zod_1.z.object({
+    CreateImageMessageCommand.RequestSchema = zod_1.z.object({
+        text: zod_1.z.string(),
         params: zod_1.z.array(zod_1.z.object({
             type: zod_1.z.nativeEnum(constants_1.AI_MODEL_CONFIG_PARAM),
             option: zod_1.z.string().uuid(),
         })),
-    }));
+        refinePrompt: zod_1.z.boolean().default(false),
+    });
     CreateImageMessageCommand.RequestParamSchema = models_1.ChatSchema.pick({
         uuid: true,
     });

package/build/constants/ai-model/enums/ai-model-config-selector-type.enum.js

@@ -4,5 +4,6 @@ exports.AI_MODEL_CONFIG_PARAM = void 0;
 var AI_MODEL_CONFIG_PARAM;
 (function (AI_MODEL_CONFIG_PARAM) {
     AI_MODEL_CONFIG_PARAM["IMAGE_SIZE"] = "image-size";
+    AI_MODEL_CONFIG_PARAM["IMAGE_STYLE"] = "image-style";
     AI_MODEL_CONFIG_PARAM["WEB_SEARCH"] = "web-search";
 })(AI_MODEL_CONFIG_PARAM || (exports.AI_MODEL_CONFIG_PARAM = AI_MODEL_CONFIG_PARAM = {}));

package/build/models/ai-model-config.schema.js

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.AiModelConfigSchema = exports.AiModelImageSizeParameter = exports.AiModelImageSizeOptionSchema = exports.AiModelConfigParameterSchema = void 0;
+exports.AiModelConfigSchema = exports.AiModelWebSearchParameter = exports.AiModelImageStyleParameter = exports.AiModelImageStyleOptionSchema = exports.AiModelImageSizeParameter = exports.AiModelImageSizeOptionSchema = exports.AiModelConfigParameterSchema = void 0;
 const zod_1 = require("zod");
 const constants_1 = require("../constants");
 exports.AiModelConfigParameterSchema = zod_1.z.object({
@@ -20,11 +20,25 @@ exports.AiModelImageSizeParameter = zod_1.z.object({
     title: zod_1.z.string(),
     options: zod_1.z.array(exports.AiModelImageSizeOptionSchema),
 });
+// IMAGE STYLE
+exports.AiModelImageStyleOptionSchema = exports.AiModelConfigParameterSchema.extend({
+    sample: zod_1.z.string(),
+});
+exports.AiModelImageStyleParameter = zod_1.z.object({
+    title: zod_1.z.string(),
+    options: zod_1.z.array(exports.AiModelImageStyleOptionSchema),
+});
+// WEB SEARCH
+exports.AiModelWebSearchParameter = zod_1.z.object({
+    webSearchPrice: zod_1.z.number(),
+});
 // CONFIG
 exports.AiModelConfigSchema = zod_1.z.object({
     parameters: zod_1.z
         .object({
         [constants_1.AI_MODEL_CONFIG_PARAM.IMAGE_SIZE]: exports.AiModelImageSizeParameter.optional(),
+        [constants_1.AI_MODEL_CONFIG_PARAM.IMAGE_STYLE]: exports.AiModelImageStyleParameter.optional(),
+        [constants_1.AI_MODEL_CONFIG_PARAM.WEB_SEARCH]: exports.AiModelWebSearchParameter.optional(),
     })
         .strict(),
 });

package/commands/message/create-image-message.command.ts

@@ -3,17 +3,16 @@ import { z } from 'zod';
 import { AI_MODEL_CONFIG_PARAM } from '../../constants';
 
 export namespace CreateImageMessageCommand {
-    export const RequestSchema = z.intersection(
-        MessageSchema.pick({ text: true }),
-        z.object({
-            params: z.array(
-                z.object({
-                    type: z.nativeEnum(AI_MODEL_CONFIG_PARAM),
-                    option: z.string().uuid(),
-                }),
-            ),
-        }),
-    );
+    export const RequestSchema = z.object({
+        text: z.string(),
+        params: z.array(
+            z.object({
+                type: z.nativeEnum(AI_MODEL_CONFIG_PARAM),
+                option: z.string().uuid(),
+            }),
+        ),
+        refinePrompt: z.boolean().default(false),
+    });
 
     export const RequestParamSchema = ChatSchema.pick({
         uuid: true,

package/constants/ai-model/enums/ai-model-config-selector-type.enum.ts

@@ -1,4 +1,5 @@
 export enum AI_MODEL_CONFIG_PARAM {
     IMAGE_SIZE = 'image-size',
+    IMAGE_STYLE = 'image-style',
     WEB_SEARCH = 'web-search',
 }

package/models/ai-model-config.schema.ts

@@ -22,12 +22,31 @@ export const AiModelImageSizeParameter = z.object({
     options: z.array(AiModelImageSizeOptionSchema),
 });
 
+// IMAGE STYLE
+
+export const AiModelImageStyleOptionSchema = AiModelConfigParameterSchema.extend({
+    sample: z.string(),
+});
+
+export const AiModelImageStyleParameter = z.object({
+    title: z.string(),
+    options: z.array(AiModelImageStyleOptionSchema),
+});
+
+// WEB SEARCH
+
+export const AiModelWebSearchParameter = z.object({
+    webSearchPrice: z.number(),
+});
+
 // CONFIG
 
 export const AiModelConfigSchema = z.object({
     parameters: z
         .object({
             [AI_MODEL_CONFIG_PARAM.IMAGE_SIZE]: AiModelImageSizeParameter.optional(),
+            [AI_MODEL_CONFIG_PARAM.IMAGE_STYLE]: AiModelImageStyleParameter.optional(),
+            [AI_MODEL_CONFIG_PARAM.WEB_SEARCH]: AiModelWebSearchParameter.optional(),
         })
         .strict(),
 });
@@ -35,4 +54,7 @@ export const AiModelConfigSchema = z.object({
 export type AiModelConfigParameter = z.infer<typeof AiModelConfigParameterSchema>;
 export type AiModelImageSizeOption = z.infer<typeof AiModelImageSizeOptionSchema>;
 export type AiModelImageSizeParameter = z.infer<typeof AiModelImageSizeParameter>;
+export type AiModelImageStyleOption = z.infer<typeof AiModelImageStyleOptionSchema>;
+export type AiModelImageStyleParameter = z.infer<typeof AiModelImageStyleParameter>;
+export type AiModelWebSeachParameter = z.infer<typeof AiModelWebSearchParameter>;
 export type AiModelConfig = z.infer<typeof AiModelConfigSchema>;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@purpleschool/gptbot",
-  "version": "0.5.82",
+  "version": "0.5.84",
   "description": "",
   "main": "build/index.js",
   "types": "build/index.d.ts",