@purpleschool/gptbot-tools 0.0.57 → 0.0.59

package/build/common/errors/errors.js

@@ -670,6 +670,23 @@ exports.ERRORS = {
             httpCode: 404,
         },
     },
+    WRITER_MODEL: {
+        MODEL_NOT_ACTIVE: {
+            code: 'WRITER_MODEL.MODEL_NOT_ACTIVE',
+            message: 'Модель для генерации текста не активна',
+            httpCode: 400,
+        },
+        FIND_ERROR: {
+            code: 'WRITER_MODEL.FIND_ERROR',
+            message: 'Произошла ошибка при поиске модели для генерации текста',
+            httpCode: 500,
+        },
+        SAVE_ERROR: {
+            code: 'WRITER_MODEL.SAVE_ERROR',
+            message: 'Произошла ошибка при сохранении модели для генерации текста',
+            httpCode: 500,
+        },
+    },
     PROMPT_MODERATION: {
         AI_ERROR: {
             code: 'PROMPT_MODERATION.AI_ERROR',

package/build/writer/commands/create-writer-document.command.js

@@ -12,6 +12,7 @@ var CreateWriterDocumentCommand;
         prompt: zod_1.z.string(),
         numPages: zod_1.z.number(),
         documentTypeId: zod_1.z.string().uuid(),
+        modelId: zod_1.z.string().uuid(),
     });
     CreateWriterDocumentCommand.ResponseSchema = (0, common_1.ICommandResponseSchema)(models_1.WriterDocumentSchema);
 })(CreateWriterDocumentCommand || (exports.CreateWriterDocumentCommand = CreateWriterDocumentCommand = {}));

package/build/writer/enums/index.js

@@ -19,3 +19,4 @@ __exportStar(require("./writer-ai-action-pricing-type.enum"), exports);
 __exportStar(require("./writer-ai-action-type.enum"), exports);
 __exportStar(require("./writer-document-section-type.enum"), exports);
 __exportStar(require("./writer-document-stage.enum"), exports);
+__exportStar(require("./writer-content-generation-strategy.enum"), exports);

package/build/writer/enums/writer-content-generation-strategy.enum.js

@@ -0,0 +1,8 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.WRITER_CONTENT_GENERATION_STRATEGY = void 0;
+var WRITER_CONTENT_GENERATION_STRATEGY;
+(function (WRITER_CONTENT_GENERATION_STRATEGY) {
+    WRITER_CONTENT_GENERATION_STRATEGY["OPENAI"] = "openai";
+    WRITER_CONTENT_GENERATION_STRATEGY["PERPLEXITY"] = "perplexity";
+})(WRITER_CONTENT_GENERATION_STRATEGY || (exports.WRITER_CONTENT_GENERATION_STRATEGY = WRITER_CONTENT_GENERATION_STRATEGY = {}));

package/build/writer/index.js

@@ -20,3 +20,4 @@ __exportStar(require("./queries"), exports);
 __exportStar(require("./enums"), exports);
 __exportStar(require("./routes"), exports);
 __exportStar(require("./utils"), exports);
+__exportStar(require("./models"), exports);

package/build/writer/models/index.js

@@ -14,6 +14,7 @@ var __exportStar = (this && this.__exportStar) || function(m, exports) {
     for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
 };
 Object.defineProperty(exports, "__esModule", { value: true });
+__exportStar(require("./writer-model.schema"), exports);
 __exportStar(require("./writer-config.schema"), exports);
 __exportStar(require("./writer-ai-action.schema"), exports);
 __exportStar(require("./writer-document-outline.schema"), exports);

package/build/writer/models/writer-config.schema.js

@@ -4,8 +4,10 @@ exports.WriterConfigSchema = void 0;
 const zod_1 = require("zod");
 const writer_document_type_schema_1 = require("./writer-document-type.schema");
 const writer_ai_action_schema_1 = require("./writer-ai-action.schema");
+const writer_model_schema_1 = require("./writer-model.schema");
 exports.WriterConfigSchema = zod_1.z.object({
     maxPromptLength: zod_1.z.number(),
     documentTypes: zod_1.z.array(writer_document_type_schema_1.WriterDocumentTypeSchema),
     aiActions: zod_1.z.array(writer_ai_action_schema_1.WriterAiActionSchema),
+    models: zod_1.z.array(writer_model_schema_1.WriterModelSchema),
 });

package/build/writer/models/writer-document-type.schema.js

@@ -12,6 +12,7 @@ exports.WriterDocumentTypeSchema = zod_1.z.object({
     maxPages: zod_1.z.number(),
     pageRange: zod_1.z.array(zod_1.z.number()),
     needsTableOfContents: zod_1.z.boolean(),
+    needsCitations: zod_1.z.boolean(),
     createdAt: zod_1.z.date(),
     updatedAt: zod_1.z.date(),
 });

package/build/writer/models/writer-document.schema.js

@@ -16,6 +16,7 @@ exports.WriterDocumentSchema = zod_1.z.object({
     finalMd: zod_1.z.string().nullable(),
     userId: zod_1.z.string().nullable(),
     unregisteredUserId: zod_1.z.string().nullable(),
+    modelId: zod_1.z.string(),
     reaction: zod_1.z.nativeEnum(common_1.USER_REACTION).nullable(),
     pages: zod_1.z.number(),
     isDeleted: zod_1.z.boolean(),

package/build/writer/models/writer-model.schema.js

@@ -0,0 +1,21 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.WriterModelSchema = void 0;
+const zod_1 = require("zod");
+const common_1 = require("../../common");
+const common_2 = require("../../common");
+const writer_content_generation_strategy_enum_1 = require("../enums/writer-content-generation-strategy.enum");
+exports.WriterModelSchema = zod_1.z.object({
+    uuid: zod_1.z.string(),
+    title: zod_1.z.string(),
+    description: zod_1.z.string(),
+    aiModel: zod_1.z.string(),
+    pricePerPage: zod_1.z.number(),
+    order: zod_1.z.number(),
+    status: zod_1.z.nativeEnum(common_1.TOOL_MODEL_STATUS),
+    icons: common_2.IconVariantsSchema,
+    strategy: zod_1.z.nativeEnum(writer_content_generation_strategy_enum_1.WRITER_CONTENT_GENERATION_STRATEGY),
+    maxInputLength: zod_1.z.number(),
+    createdAt: zod_1.z.date(),
+    updatedAt: zod_1.z.date(),
+});

package/build/writer/utils/calculate-writer-content-generation-price.util.js

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.calculateWriterContentGenerationPrice = calculateWriterContentGenerationPrice;
-function calculateWriterContentGenerationPrice(pages) {
-    return Math.max(1, pages);
+function calculateWriterContentGenerationPrice(pages, model) {
+    return Math.max(1, pages * model.pricePerPage);
 }

package/common/errors/errors.ts

@@ -671,6 +671,23 @@ export const ERRORS = {
             httpCode: 404,
         },
     },
+    WRITER_MODEL: {
+        MODEL_NOT_ACTIVE: {
+            code: 'WRITER_MODEL.MODEL_NOT_ACTIVE',
+            message: 'Модель для генерации текста не активна',
+            httpCode: 400,
+        },
+        FIND_ERROR: {
+            code: 'WRITER_MODEL.FIND_ERROR',
+            message: 'Произошла ошибка при поиске модели для генерации текста',
+            httpCode: 500,
+        },
+        SAVE_ERROR: {
+            code: 'WRITER_MODEL.SAVE_ERROR',
+            message: 'Произошла ошибка при сохранении модели для генерации текста',
+            httpCode: 500,
+        },
+    },
     PROMPT_MODERATION: {
         AI_ERROR: {
             code: 'PROMPT_MODERATION.AI_ERROR',

package/package.json

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

package/writer/commands/create-writer-document.command.ts

@@ -9,6 +9,7 @@ export namespace CreateWriterDocumentCommand {
         prompt: z.string(),
         numPages: z.number(),
         documentTypeId: z.string().uuid(),
+        modelId: z.string().uuid(),
     });
     export type Request = z.infer<typeof RequestSchema>;
 

package/writer/enums/index.ts

@@ -3,3 +3,4 @@ export * from './writer-ai-action-pricing-type.enum';
 export * from './writer-ai-action-type.enum';
 export * from './writer-document-section-type.enum';
 export * from './writer-document-stage.enum';
+export * from './writer-content-generation-strategy.enum';

package/writer/enums/writer-content-generation-strategy.enum.ts

@@ -0,0 +1,4 @@
+export enum WRITER_CONTENT_GENERATION_STRATEGY {
+    OPENAI = 'openai',
+    PERPLEXITY = 'perplexity',
+}

package/writer/index.ts

@@ -4,3 +4,4 @@ export * from './queries';
 export * from './enums';
 export * from './routes';
 export * from './utils';
+export * from './models';

package/writer/models/index.ts

@@ -1,3 +1,4 @@
+export * from './writer-model.schema';
 export * from './writer-config.schema';
 export * from './writer-ai-action.schema';
 export * from './writer-document-outline.schema';

package/writer/models/writer-config.schema.ts

@@ -1,10 +1,12 @@
 import { z } from 'zod';
 import { WriterDocumentTypeSchema } from './writer-document-type.schema';
 import { WriterAiActionSchema } from './writer-ai-action.schema';
+import { WriterModelSchema } from './writer-model.schema';
 
 export const WriterConfigSchema = z.object({
     maxPromptLength: z.number(),
     documentTypes: z.array(WriterDocumentTypeSchema),
     aiActions: z.array(WriterAiActionSchema),
+    models: z.array(WriterModelSchema),
 });
 export type WriterConfig = z.infer<typeof WriterConfigSchema>;

package/writer/models/writer-document-type.schema.ts

@@ -10,6 +10,7 @@ export const WriterDocumentTypeSchema = z.object({
     maxPages: z.number(),
     pageRange: z.array(z.number()),
     needsTableOfContents: z.boolean(),
+    needsCitations: z.boolean(),
     createdAt: z.date(),
     updatedAt: z.date(),
 });

package/writer/models/writer-document.schema.ts

@@ -14,6 +14,7 @@ export const WriterDocumentSchema = z.object({
     finalMd: z.string().nullable(),
     userId: z.string().nullable(),
     unregisteredUserId: z.string().nullable(),
+    modelId: z.string(),
     reaction: z.nativeEnum(USER_REACTION).nullable(),
     pages: z.number(),
     isDeleted: z.boolean(),

package/writer/models/writer-model.schema.ts

@@ -0,0 +1,21 @@
+import { z } from 'zod';
+import { TOOL_MODEL_STATUS } from '../../common';
+import { IconVariantsSchema } from '../../common';
+import { WRITER_CONTENT_GENERATION_STRATEGY } from '../enums/writer-content-generation-strategy.enum';
+
+export const WriterModelSchema = z.object({
+    uuid: z.string(),
+    title: z.string(),
+    description: z.string(),
+    aiModel: z.string(),
+    pricePerPage: z.number(),
+    order: z.number(),
+    status: z.nativeEnum(TOOL_MODEL_STATUS),
+    icons: IconVariantsSchema,
+    strategy: z.nativeEnum(WRITER_CONTENT_GENERATION_STRATEGY),
+    maxInputLength: z.number(),
+    createdAt: z.date(),
+    updatedAt: z.date(),
+});
+
+export type WriterModel = z.infer<typeof WriterModelSchema>;

package/writer/utils/calculate-writer-content-generation-price.util.ts

@@ -1,3 +1,5 @@
-export function calculateWriterContentGenerationPrice(pages: number): number {
-    return Math.max(1, pages);
+import { WriterModel } from '../models';
+
+export function calculateWriterContentGenerationPrice(pages: number, model: WriterModel): number {
+    return Math.max(1, pages * model.pricePerPage);
 }