@purpleschool/gptbot 0.7.39 → 0.7.40-texteditor

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,3 @@
+export * from './calculate-writer-ai-action-price.util';
+export * from './calculate-writer-content-generation-price.util';
+export * from './calculate-page-count.util';

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,10 @@
+import { z } from 'zod';
+import { WriterDocumentTypeSchema } from './writer-document-type.schema';
+import { WriterAiActionSchema } from './writer-ai-action.schema';
+
+export const WriterConfigSchema = z.object({
+    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,15 @@
+import { z } from 'zod';
+import { IconVariantsSchema } from '../../icon-variants.schema';
+
+export const WriterDocumentTypeSchema = z.object({
+    uuid: z.string().uuid(),
+    title: z.string(),
+    icons: IconVariantsSchema,
+    order: z.number(),
+    minPages: z.number(),
+    maxPages: z.number(),
+    pageRange: z.array(z.number()),
+    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/models/user-to-subscription.schema.ts

@@ -9,6 +9,7 @@ export const UserToSubscriptionSchema = z.object({
     active: z.boolean(),
     tokenBalance: z.number(),
     status: z.string(),
+    startDate: z.date().nullable(),
     endDate: z.date().nullable(),
     marks: z.array(z.string()),
     intervalEndDate: z.date().nullable(),

package/package.json

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