@purpleschool/gptbot-tools 0.1.134 → 0.2.1-stage

package/writer/models/writer-source-bibliographic-metadata.schema.ts

@@ -0,0 +1,31 @@
+import { z } from 'zod';
+import { WRITER_SOURCE_TYPE } from '../enums';
+
+export const BookBibliographicMetadataSchema = z.object({
+    type: z.literal(WRITER_SOURCE_TYPE.BOOK),
+    title: z.string(),
+    authors: z.string(),
+    year: z.number(),
+    publicationCity: z.string(),
+    publisher: z.string(),
+    pageCount: z.number().nullable(),
+});
+
+export const JournalArticleBibliographicMetadataSchema = z.object({
+    type: z.literal(WRITER_SOURCE_TYPE.JOURNAL_ARTICLE),
+    title: z.string(),
+    authors: z.string(),
+    year: z.number(),
+    journalName: z.string(),
+    issueNumber: z.string(),
+    pageRange: z.string(),
+});
+
+export const WriterSourceBibliographicMetadataSchema = z.discriminatedUnion('type', [
+    BookBibliographicMetadataSchema,
+    JournalArticleBibliographicMetadataSchema,
+]);
+
+export type WriterSourceBibliographicMetadata = z.infer<
+    typeof WriterSourceBibliographicMetadataSchema
+>;

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

@@ -0,0 +1,22 @@
+import { z } from 'zod';
+import { WRITER_SOURCE_ORIGIN, WRITER_SOURCE_STATUS } from '../enums';
+import { WriterSourceBibliographicMetadataSchema } from './writer-source-bibliographic-metadata.schema';
+
+export const WriterSourceSchema = z.object({
+    uuid: z.string().uuid(),
+    query: z.string().nullable(),
+    url: z.string(),
+    snippet: z.string().nullable(),
+    inclusionReason: z.string().nullable(),
+    sourceOrigin: z.nativeEnum(WRITER_SOURCE_ORIGIN),
+    content: z.string().nullable(),
+    documentId: z.string().uuid(),
+    fileId: z.string().nullable(),
+    citation: z.string().nullable(),
+    status: z.nativeEnum(WRITER_SOURCE_STATUS),
+    bibliographicMetadata: WriterSourceBibliographicMetadataSchema.nullable(),
+    createdAt: z.date(),
+    updatedAt: z.date(),
+});
+
+export type WriterSource = z.infer<typeof WriterSourceSchema>;

package/writer/queries/get-writer-document-sources.query.ts

@@ -0,0 +1,15 @@
+import { z } from 'zod';
+import { ICommandResponseSchema } from '../../common/models/command-response.schema';
+import { WriterSourceSchema } from '../models';
+
+export namespace GetWriterDocumentSourcesQueryContract {
+    export const RequestSchema = z.object({
+        uuid: z.string().uuid(),
+        userId: z.string().uuid().nullable().optional(),
+        unregisteredUserId: z.string().uuid().nullable().optional(),
+    });
+    export type Request = z.infer<typeof RequestSchema>;
+
+    export const ResponseSchema = ICommandResponseSchema(z.array(WriterSourceSchema));
+    export type Response = z.infer<typeof ResponseSchema>;
+}

package/writer/queries/index.ts

@@ -1,3 +1,4 @@
 export * from './find-writer-documents.query';
 export * from './get-writer-config.query';
 export * from './find-writer-document-by-uuid.query';
+export * from './get-writer-document-sources.query';

package/writer/routes/writer.routes.ts

@@ -10,6 +10,12 @@ export const WRITER_AMQP_ROUTES = {
     TOGGLE_DOCUMENT_TITLE_PAGE: 'tools.writer.toggle-document-title-page.rpc',
     UPDATE_DOCUMENT_CONTENTS: 'tools.writer.update-document-contents.rpc',
     GENERATE_DOCUMENT_CONTENTS: 'tools.writer.generate-document-contents.rpc',
+    GENERATE_DOCUMENT_OUTLINE: 'tools.writer.generate-document-outline.rpc',
+    COLLECT_SOURCES: 'tools.writer.collect-sources.rpc',
+    GET_DOCUMENT_SOURCES: 'tools.writer.get-document-sources.rpc',
+    UPLOAD_SOURCE: 'tools.writer.upload-source.rpc',
+    CONFIRM_SOURCE_CITATION_METADATA: 'tools.writer.confirm-source-citation-metadata.rpc',
+    DISCARD_SOURCE: 'tools.writer.discard-source.rpc',
     GET_CONTENT_GENERATION_PRICE: 'tools.writer.content-generation.get-price.rpc',
     PARAPHRASE: 'tools.writer.actions.paraphrase.rpc',
     EXTEND_TEXT: 'tools.writer.actions.extend-text.rpc',