@purpleschool/gptbot 0.7.34 → 0.7.35-texteditor

package/constants/writer/enums/index.ts

@@ -0,0 +1,4 @@
+export * from './writer-document-section-type.enum';
+export * from './writer-document-stage.enum';
+export * from './writer-ai-action-pricing-type.enum';
+export * from './writer-ai-action-type.enum';

package/constants/writer/enums/writer-ai-action-pricing-type.enum.ts

@@ -0,0 +1,4 @@
+export enum WRITER_AI_ACTION_PRICING_TYPE {
+    PER_CHARACTER = 'PER_CHARACTER',
+    FLAT = 'FLAT',
+}

package/constants/writer/enums/writer-ai-action-type.enum.ts

@@ -0,0 +1,7 @@
+export enum WRITER_AI_ACTION_TYPE {
+    PARAPHRASE = 'PARAPHRASE',
+    EXTEND_TEXT = 'EXTEND_TEXT',
+    SHORTEN_TEXT = 'SHORTEN_TEXT',
+    GENERATE_TEXT = 'GENERATE_TEXT',
+    FIX_ERRORS = 'FIX_ERRORS',
+}

package/constants/writer/enums/writer-document-section-type.enum.ts

@@ -0,0 +1,5 @@
+export enum WRITER_DOCUMENT_SECTION_TYPE {
+    INTRODUCTION = 'introduction',
+    CONCLUSION = 'conclusion',
+    SECTION = 'section',
+}

package/constants/writer/enums/writer-document-stage.enum.ts

@@ -0,0 +1,13 @@
+export enum WRITER_DOCUMENT_STAGE {
+    CREATED = 'CREATED',
+
+    GENERATING_OUTLINE = 'GENERATING_OUTLINE',
+    OUTLINE_GENERATED = 'OUTLINE_GENERATED',
+    OUTLINE_GENERATION_FAILED = 'OUTLINE_GENERATION_FAILED',
+
+    GENERATING_CONTENT = 'GENERATING_CONTENT',
+    CONTENT_GENERATED = 'CONTENT_GENERATED',
+    CONTENT_GENERATION_FAILED = 'CONTENT_GENERATION_FAILED',
+
+    DELETED = 'DELETED',
+}

package/constants/writer/index.ts

@@ -0,0 +1 @@
+export * from './enums';

package/helpers/index.ts

@@ -3,3 +3,4 @@ export * from './stt';
 export * from './subscription';
 export * from './tts';
 export * from './video';
+export * from './writer';

package/helpers/writer/calculate-writer-ai-action-price.util.ts

@@ -0,0 +1,20 @@
+import { WRITER_AI_ACTION_PRICING_TYPE } from '../../constants';
+import { WriterAiActionPricingRules } from '../../models';
+
+export function calculateWriterAiActionPrice(
+    pricingRules: WriterAiActionPricingRules,
+    selectionText: string,
+): number {
+    switch (pricingRules.type) {
+        case WRITER_AI_ACTION_PRICING_TYPE.PER_CHARACTER:
+            const characterCount = selectionText.length;
+            const price = characterCount * pricingRules.price;
+            return Math.max(1, Math.ceil(price));
+
+        case WRITER_AI_ACTION_PRICING_TYPE.FLAT:
+            return Math.max(1, Math.ceil(pricingRules.price));
+
+        default:
+            throw new Error(`Unknown pricing type`);
+    }
+}

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

@@ -0,0 +1,3 @@
+export function calculateWriterContentGenerationPrice(pages: number): number {
+    return Math.max(1, pages);
+}

package/helpers/writer/index.ts

@@ -0,0 +1,2 @@
+export * from './calculate-writer-ai-action-price.util';
+export * from './calculate-writer-content-generation-price.util';

package/models/tools/image-editor/image-editor-config.schema.ts

@@ -2,15 +2,6 @@ import { z } from 'zod';
 import { ImageEditorModelSchema } from './image-editor-model.schema';
 import { IconVariantsSchema } from '../../icon-variants.schema';
 
-export const SystemPromptParamsSchema = z.object({
-    imageSettings: z.object({
-        minImages: z.number(),
-        maxImages: z.number(),
-    }),
-});
-
-export type SystemPromptParams = z.infer<typeof SystemPromptParamsSchema>;
-
 export const SystemPromptSchema = z.object({
     uuid: z.string(),
     title: z.string(),
@@ -18,7 +9,6 @@ export const SystemPromptSchema = z.object({
     prompt: z.string(),
     allowUserPrompt: z.boolean(),
     groupId: z.string().uuid(),
-    params: SystemPromptParamsSchema,
     preview: z
         .object({
             before: z.string(),
@@ -26,7 +16,6 @@ export const SystemPromptSchema = z.object({
         })
         .optional()
         .nullable(),
-    order: z.number(),
 });
 
 export const PromptGroupSchema = z.object({
@@ -39,5 +28,5 @@ export const PromptGroupSchema = z.object({
 
 export const ImageEditorConfigSchema = z.object({
     models: z.array(ImageEditorModelSchema),
-    promptGroups: z.array(PromptGroupSchema),
+    promptGroups: z.array(SystemPromptSchema),
 });

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

@@ -1,6 +1,5 @@
 import { z } from 'zod';
 import { USER_REACTION } from '../../../constants';
-import { ToolJobSchema } from '../../tool-job.schema';
 
 export const ImageEditorJobParamsSchema = z.object({
     imageUrls: z.array(z.string()).optional(),
@@ -9,7 +8,7 @@ export const ImageEditorJobParamsSchema = z.object({
 
 export type ImageEditorJobParams = z.infer<typeof ImageEditorJobParamsSchema>;
 
-export const ImageEditorJobSchema = ToolJobSchema.extend({
+export const ImageEditorJobSchema = z.object({
     title: z.string(),
     prompt: z.string(),
     reaction: z.nativeEnum(USER_REACTION).nullable(),

package/models/tools/index.ts

@@ -3,4 +3,5 @@ export * from './paraphrase';
 export * from './stt';
 export * from './tts';
 export * from './video';
+export * from './writer';
 export * from './image-editor';

package/models/tools/writer/index.ts

@@ -0,0 +1,5 @@
+export * from './writer-config.schema';
+export * from './writer-ai-action.schema';
+export * from './writer-document-outline.schema';
+export * from './writer-document.schema';
+export * from './writer-document-type.schema';

package/models/tools/writer/writer-ai-action.schema.ts

@@ -0,0 +1,27 @@
+import { z } from 'zod';
+import { WRITER_AI_ACTION_PRICING_TYPE, WRITER_AI_ACTION_TYPE } from '../../../constants';
+
+export const WriterAiActionPricingRulesSchema = z.discriminatedUnion('type', [
+    z.object({
+        type: z.literal(WRITER_AI_ACTION_PRICING_TYPE.PER_CHARACTER),
+        price: z.number(),
+    }),
+    z.object({
+        type: z.literal(WRITER_AI_ACTION_PRICING_TYPE.FLAT),
+        price: z.number(),
+    }),
+]);
+export type WriterAiActionPricingRules = z.infer<typeof WriterAiActionPricingRulesSchema>;
+
+export const WriterAiActionSchema = z.object({
+    uuid: z.string(),
+    type: z.nativeEnum(WRITER_AI_ACTION_TYPE),
+    pricingRules: WriterAiActionPricingRulesSchema,
+    maxPromptLength: z.number(),
+    title: z.string(),
+    icon: z.string(),
+    order: z.number(),
+    createdAt: z.date(),
+    updatedAt: z.date(),
+});
+export type WriterAiAction = z.infer<typeof WriterAiActionSchema>;

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

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

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

@@ -0,0 +1,28 @@
+import { z } from 'zod';
+import { WRITER_DOCUMENT_SECTION_TYPE } from '../../../constants';
+
+export const WriterDocumentOutlineSubsectionSchema = z.object({
+    title: z
+        .string()
+        .min(3)
+        .max(120)
+        .describe(
+            `Concise title of the subsection. Meant to elaborate on its parent section's title.`,
+        ),
+});
+export type WriterDocumentOutlineSubsection = z.infer<typeof WriterDocumentOutlineSubsectionSchema>;
+
+export const WriterDocumentOutlineSectionSchema = z.object({
+    title: z.string().min(3).max(120).describe(`Concise title of the paper section.`),
+    type: z.nativeEnum(WRITER_DOCUMENT_SECTION_TYPE),
+    subsections: z
+        .array(WriterDocumentOutlineSubsectionSchema)
+        .optional()
+        .describe(
+            `Subsections of the paper section. Each subsection is a single idea that is a part of the parent section.`,
+        ),
+});
+
+export const WriterDocumentOutlineSchema = z.array(WriterDocumentOutlineSectionSchema);
+
+export type WriterDocumentOutline = z.infer<typeof WriterDocumentOutlineSchema>;

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

@@ -0,0 +1,12 @@
+import { z } from 'zod';
+
+export const WriterDocumentTypeSchema = z.object({
+    uuid: z.string().uuid(),
+    title: z.string(),
+    icon: z.string(),
+    order: z.number(),
+    promptInstructions: z.string().nullable(),
+    createdAt: z.date(),
+    updatedAt: z.date(),
+});
+export type WriterDocumentType = z.infer<typeof WriterDocumentTypeSchema>;

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

@@ -0,0 +1,21 @@
+import { z } from 'zod';
+import { WriterDocumentOutlineSchema } from './writer-document-outline.schema';
+import { WRITER_DOCUMENT_STAGE } from '../../../constants';
+
+export const WriterDocumentSchema = z.object({
+    uuid: z.string().uuid(),
+    prompt: z.string(),
+    topic: z.string(),
+    typeId: z.string(),
+    stage: z.nativeEnum(WRITER_DOCUMENT_STAGE),
+    outline: WriterDocumentOutlineSchema,
+    sourceMd: z.string().nullable(),
+    finalMd: z.string().nullable(),
+    userId: z.string().nullable(),
+    unregisteredUserId: z.string().nullable(),
+    pages: z.number(),
+    createdAt: z.date(),
+    updatedAt: z.date(),
+    completedAt: z.date().nullable(),
+});
+export type WriterDocument = z.infer<typeof WriterDocumentSchema>;

package/package.json

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