@riotprompt/riotprompt 0.0.1

package/dist/builder.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"builder.cjs","sources":["../src/builder.ts"],"sourcesContent":["import path from \"path\";\nimport { z } from \"zod\";\nimport { ParametersSchema } from \"./items/parameters\";\nimport { SectionOptions, SectionOptionsSchema } from \"./items/section\";\nimport { DEFAULT_LOGGER, wrapLogger } from \"./logger\";\nimport { Content, Context, createPrompt, createSection, Instruction, Loader, Override, Parser, Prompt, Section, Weighted } from \"./riotprompt\";\n\nconst OptionSchema = z.object({\n    logger: z.any().optional().default(DEFAULT_LOGGER),\n    basePath: z.string(),\n    overridePath: z.string().optional().default(\"./\"),\n    overrides: z.boolean().optional().default(false),\n    parameters: ParametersSchema.optional().default({}),\n});\n\nexport type Options = z.infer<typeof OptionSchema>;\n\nexport type OptionsParam = Required<Pick<Options, 'basePath'>> & Partial<Omit<Options, 'basePath'>>;\n\nexport interface Instance {\n    addPersonaPath(contentPath: string, sectionOptions?: Partial<SectionOptions>): Promise<Instance>;\n    addContextPath(contentPath: string, sectionOptions?: Partial<SectionOptions>): Promise<Instance>;\n    addInstructionPath(contentPath: string, sectionOptions?: Partial<SectionOptions>): Promise<Instance>;\n    addContentPath(contentPath: string, sectionOptions?: Partial<SectionOptions>): Promise<Instance>;\n    addContent(content: string, sectionOptions?: Partial<SectionOptions>): Promise<Instance>;\n    addContext(context: string, sectionOptions?: Partial<SectionOptions>): Promise<Instance>;\n    loadContext(contextDirectories: string[], sectionOptions?: Partial<SectionOptions>): Promise<Instance>;\n    loadContent(contentDirectories: string[], sectionOptions?: Partial<SectionOptions>): Promise<Instance>;\n    build(): Promise<Prompt>;\n}\n\nexport const create = (builderOptions: OptionsParam): Instance => {\n    const options: Required<Options> = OptionSchema.parse(builderOptions) as Required<Options>;\n\n    const logger = wrapLogger(options.logger, 'Builder');\n    const parser = Parser.create({ logger });\n    const override = Override.create({\n        logger, configDir: options.overridePath || \"./\",\n        overrides: options.overrides || false\n    });\n    const loader = Loader.create({ logger });\n\n    const personaSection: Section<Instruction> = createSection({ title: \"Persona\" });\n    const contextSection: Section<Context> = createSection({ title: \"Context\" });\n    const instructionSection: Section<Instruction> = createSection({ title: \"Instruction\" });\n    const contentSection: Section<Content> = createSection({ title: \"Content\" });\n    const parameters = options.parameters;\n\n\n    const instance: Partial<Instance> = {}\n\n    const loadOptions = (sectionOptions: Partial<SectionOptions> = {}): SectionOptions => {\n        const currentOptions = SectionOptionsSchema.parse(sectionOptions);\n        return {\n            ...currentOptions,\n            parameters: {\n                ...parameters,\n                ...currentOptions.parameters\n            }\n        }\n    }\n\n    const loadDirectories = async <T extends Weighted>(\n        directories: string[],\n        sectionOptions: Partial<SectionOptions> = {}\n    ): Promise<Section<T>[]> => {\n        const currentOptions = loadOptions(sectionOptions);\n        logger.debug(\"Loading directories\", directories);\n        const sections: Section<T>[] = await loader.load<T>(directories, currentOptions);\n        return sections;\n    }\n\n    const loadContext = async (\n        contextDirectories: string[],\n        sectionOptions: Partial<SectionOptions> = {}\n    ): Promise<Instance> => {\n        const currentOptions = loadOptions(sectionOptions);\n        logger.debug('Loading context', contextDirectories);\n        const context: Section<Context>[] = await loadDirectories<Context>(contextDirectories, currentOptions);\n        contextSection.add(context);\n        return instance as Instance;\n    }\n    instance.loadContext = loadContext;\n\n    const loadContent = async (\n        contentDirectories: string[],\n        sectionOptions: Partial<SectionOptions> = {}\n    ): Promise<Instance> => {\n        const currentOptions = loadOptions(sectionOptions);\n        const content: Section<Content>[] = await loadDirectories<Content>(contentDirectories, currentOptions);\n        contentSection.add(content);\n        return instance as Instance;\n    }\n    instance.loadContent = loadContent;\n\n    const loadPath = async <T extends Weighted>(\n        contentPath: string,\n        sectionOptions: Partial<SectionOptions> = {}\n    ): Promise<Section<T>> => {\n        const currentOptions = loadOptions(sectionOptions);\n        const defaultPath = path.join(options.basePath as string, contentPath);\n        const section: Section<T> = await parser.parseFile<T>(defaultPath, currentOptions);\n        const overrideSection = await override.customize<T>(contentPath, section, currentOptions);\n        return overrideSection;\n    }\n\n    const addPersonaPath = async (\n        contentPath: string,\n        sectionOptions: Partial<SectionOptions> = {}\n    ): Promise<Instance> => {\n        const currentOptions = loadOptions(sectionOptions);\n        const persona: Section<Instruction> = await loadPath<Instruction>(contentPath, currentOptions);\n        personaSection.add(persona);\n        return instance as Instance;\n    }\n    instance.addPersonaPath = addPersonaPath;\n\n    const addContextPath = async (\n        contentPath: string,\n        sectionOptions: Partial<SectionOptions> = {}\n    ): Promise<Instance> => {\n        logger.debug(\"Adding context path\", contentPath);\n        const currentOptions = loadOptions(sectionOptions);\n        const context: Section<Context> = await loadPath<Context>(contentPath, currentOptions);\n        contextSection.add(context);\n        return instance as Instance;\n    }\n    instance.addContextPath = addContextPath;\n\n    const addInstructionPath = async (\n        contentPath: string,\n        sectionOptions: Partial<SectionOptions> = {}\n    ): Promise<Instance> => {\n        logger.debug(\"Adding instruction path\", contentPath);\n        const currentOptions = loadOptions(sectionOptions);\n        const instruction: Section<Instruction> = await loadPath<Instruction>(contentPath, currentOptions);\n        instructionSection.add(instruction);\n        return instance as Instance;\n    }\n    instance.addInstructionPath = addInstructionPath;\n\n    const addContentPath = async (\n        contentPath: string,\n        sectionOptions: Partial<SectionOptions> = {}\n    ): Promise<Instance> => {\n        logger.debug(\"Adding content path\", contentPath);\n        const currentOptions = loadOptions(sectionOptions);\n        const content: Section<Content> = await loadPath<Content>(contentPath, currentOptions);\n        contentSection.add(content);\n        return instance as Instance;\n    }\n    instance.addContentPath = addContentPath;\n\n    const addContent = async (\n        content: string | Buffer,\n        sectionOptions: Partial<SectionOptions> = {}\n    ): Promise<Instance> => {\n        logger.debug(\"Adding content\", typeof content);\n        const currentOptions = loadOptions(sectionOptions);\n        const parsedContentSection: Section<Content> = parser.parse<Content>(content, currentOptions);\n        contentSection.add(parsedContentSection);\n        return instance as Instance;\n    }\n    instance.addContent = addContent;\n\n    const addContext = async (\n        context: string | Buffer,\n        sectionOptions: Partial<SectionOptions> = {}\n    ): Promise<Instance> => {\n        logger.debug(\"Adding context\", typeof context);\n        const currentOptions = loadOptions(sectionOptions);\n        const parsedContextSection: Section<Context> = parser.parse<Context>(context, currentOptions);\n        contextSection.add(parsedContextSection);\n        return instance as Instance;\n    }\n    instance.addContext = addContext;\n\n    const build = async () => {\n        logger.debug(\"Building prompt\", {});\n        const prompt = createPrompt({ persona: personaSection, contexts: contextSection, instructions: instructionSection, contents: contentSection });\n        return prompt;\n    }\n    instance.build = build;\n\n    return instance as Instance;\n}\n"],"names":["OptionSchema","z","object","logger","any","optional","default","DEFAULT_LOGGER","basePath","string","overridePath","overrides","boolean","parameters","ParametersSchema","create","builderOptions","options","parse","wrapLogger","parser","Parser","override","Override","configDir","loader","Loader","personaSection","createSection","title","contextSection","instructionSection","contentSection","instance","loadOptions","sectionOptions","currentOptions","SectionOptionsSchema","loadDirectories","directories","debug","sections","load","loadContext","contextDirectories","context","add","loadContent","contentDirectories","content","loadPath","contentPath","defaultPath","path","join","section","parseFile","overrideSection","customize","addPersonaPath","persona","addContextPath","addInstructionPath","instruction","addContentPath","addContent","parsedContentSection","addContext","parsedContextSection","build","prompt","createPrompt","contexts","instructions","contents"],"mappings":";;;;;;;;;;;;;;;;AAOA,MAAMA,YAAAA,GAAeC,KAAEC,CAAAA,MAAM,CAAC;AAC1BC,IAAAA,MAAAA,EAAQF,MAAEG,GAAG,EAAA,CAAGC,QAAQ,EAAA,CAAGC,OAAO,CAACC,qBAAAA,CAAAA;AACnCC,IAAAA,QAAAA,EAAUP,MAAEQ,MAAM,EAAA;AAClBC,IAAAA,YAAAA,EAAcT,MAAEQ,MAAM,EAAA,CAAGJ,QAAQ,EAAA,CAAGC,OAAO,CAAC,IAAA,CAAA;AAC5CK,IAAAA,SAAAA,EAAWV,MAAEW,OAAO,EAAA,CAAGP,QAAQ,EAAA,CAAGC,OAAO,CAAC,KAAA,CAAA;AAC1CO,IAAAA,UAAAA,EAAYC,2BAAiBT,CAAAA,QAAQ,EAAGC,CAAAA,OAAO,CAAC,EAAC;AACrD,CAAA,CAAA;AAkBO,MAAMS,SAAS,CAACC,cAAAA,GAAAA;IACnB,MAAMC,OAAAA,GAA6BjB,YAAakB,CAAAA,KAAK,CAACF,cAAAA,CAAAA;AAEtD,IAAA,MAAMb,QAASgB,GAAAA,iBAAAA,CAAWF,OAAQd,CAAAA,MAAM,EAAE,SAAA,CAAA;IAC1C,MAAMiB,QAAAA,GAASC,aAAa,CAAC;AAAElB,gBAAAA;AAAO,KAAA,CAAA;IACtC,MAAMmB,UAAAA,GAAWC,eAAe,CAAC;AAC7BpB,gBAAAA,QAAAA;QAAQqB,SAAWP,EAAAA,OAAAA,CAAQP,YAAY,IAAI,IAAA;QAC3CC,SAAWM,EAAAA,OAAAA,CAAQN,SAAS,IAAI;AACpC,KAAA,CAAA;IACA,MAAMc,QAAAA,GAASC,aAAa,CAAC;AAAEvB,gBAAAA;AAAO,KAAA,CAAA;AAEtC,IAAA,MAAMwB,iBAAuCC,cAAc,CAAA;QAAEC,KAAO,EAAA;AAAU,KAAA,CAAA;AAC9E,IAAA,MAAMC,iBAAmCF,cAAc,CAAA;QAAEC,KAAO,EAAA;AAAU,KAAA,CAAA;AAC1E,IAAA,MAAME,qBAA2CH,cAAc,CAAA;QAAEC,KAAO,EAAA;AAAc,KAAA,CAAA;AACtF,IAAA,MAAMG,iBAAmCJ,cAAc,CAAA;QAAEC,KAAO,EAAA;AAAU,KAAA,CAAA;IAC1E,MAAMhB,UAAAA,GAAaI,QAAQJ,UAAU;AAGrC,IAAA,MAAMoB,WAA8B,EAAC;AAErC,IAAA,MAAMC,WAAc,GAAA,CAACC,cAA0C,GAAA,EAAE,GAAA;QAC7D,MAAMC,cAAAA,GAAiBC,4BAAqBnB,CAAAA,KAAK,CAACiB,cAAAA,CAAAA;QAClD,OAAO;AACH,YAAA,GAAGC,cAAc;YACjBvB,UAAY,EAAA;AACR,gBAAA,GAAGA,UAAU;AACb,gBAAA,GAAGuB,eAAevB;AACtB;AACJ,SAAA;AACJ,KAAA;AAEA,IAAA,MAAMyB,eAAkB,GAAA,OACpBC,WACAJ,EAAAA,cAAAA,GAA0C,EAAE,GAAA;AAE5C,QAAA,MAAMC,iBAAiBF,WAAYC,CAAAA,cAAAA,CAAAA;QACnChC,QAAOqC,CAAAA,KAAK,CAAC,qBAAuBD,EAAAA,WAAAA,CAAAA;AACpC,QAAA,MAAME,QAAyB,GAAA,MAAMhB,QAAOiB,CAAAA,IAAI,CAAIH,WAAaH,EAAAA,cAAAA,CAAAA;QACjE,OAAOK,QAAAA;AACX,KAAA;AAEA,IAAA,MAAME,WAAc,GAAA,OAChBC,kBACAT,EAAAA,cAAAA,GAA0C,EAAE,GAAA;AAE5C,QAAA,MAAMC,iBAAiBF,WAAYC,CAAAA,cAAAA,CAAAA;QACnChC,QAAOqC,CAAAA,KAAK,CAAC,iBAAmBI,EAAAA,kBAAAA,CAAAA;QAChC,MAAMC,OAAAA,GAA8B,MAAMP,eAAAA,CAAyBM,kBAAoBR,EAAAA,cAAAA,CAAAA;AACvFN,QAAAA,cAAAA,CAAegB,GAAG,CAACD,OAAAA,CAAAA;QACnB,OAAOZ,QAAAA;AACX,KAAA;AACAA,IAAAA,QAAAA,CAASU,WAAW,GAAGA,WAAAA;AAEvB,IAAA,MAAMI,WAAc,GAAA,OAChBC,kBACAb,EAAAA,cAAAA,GAA0C,EAAE,GAAA;AAE5C,QAAA,MAAMC,iBAAiBF,WAAYC,CAAAA,cAAAA,CAAAA;QACnC,MAAMc,OAAAA,GAA8B,MAAMX,eAAAA,CAAyBU,kBAAoBZ,EAAAA,cAAAA,CAAAA;AACvFJ,QAAAA,cAAAA,CAAec,GAAG,CAACG,OAAAA,CAAAA;QACnB,OAAOhB,QAAAA;AACX,KAAA;AACAA,IAAAA,QAAAA,CAASc,WAAW,GAAGA,WAAAA;AAEvB,IAAA,MAAMG,QAAW,GAAA,OACbC,WACAhB,EAAAA,cAAAA,GAA0C,EAAE,GAAA;AAE5C,QAAA,MAAMC,iBAAiBF,WAAYC,CAAAA,cAAAA,CAAAA;AACnC,QAAA,MAAMiB,cAAcC,IAAKC,CAAAA,IAAI,CAACrC,OAAAA,CAAQT,QAAQ,EAAY2C,WAAAA,CAAAA;AAC1D,QAAA,MAAMI,OAAsB,GAAA,MAAMnC,QAAOoC,CAAAA,SAAS,CAAIJ,WAAahB,EAAAA,cAAAA,CAAAA;AACnE,QAAA,MAAMqB,kBAAkB,MAAMnC,UAAAA,CAASoC,SAAS,CAAIP,aAAaI,OAASnB,EAAAA,cAAAA,CAAAA;QAC1E,OAAOqB,eAAAA;AACX,KAAA;AAEA,IAAA,MAAME,cAAiB,GAAA,OACnBR,WACAhB,EAAAA,cAAAA,GAA0C,EAAE,GAAA;AAE5C,QAAA,MAAMC,iBAAiBF,WAAYC,CAAAA,cAAAA,CAAAA;QACnC,MAAMyB,OAAAA,GAAgC,MAAMV,QAAAA,CAAsBC,WAAaf,EAAAA,cAAAA,CAAAA;AAC/ET,QAAAA,cAAAA,CAAemB,GAAG,CAACc,OAAAA,CAAAA;QACnB,OAAO3B,QAAAA;AACX,KAAA;AACAA,IAAAA,QAAAA,CAAS0B,cAAc,GAAGA,cAAAA;AAE1B,IAAA,MAAME,cAAiB,GAAA,OACnBV,WACAhB,EAAAA,cAAAA,GAA0C,EAAE,GAAA;QAE5ChC,QAAOqC,CAAAA,KAAK,CAAC,qBAAuBW,EAAAA,WAAAA,CAAAA;AACpC,QAAA,MAAMf,iBAAiBF,WAAYC,CAAAA,cAAAA,CAAAA;QACnC,MAAMU,OAAAA,GAA4B,MAAMK,QAAAA,CAAkBC,WAAaf,EAAAA,cAAAA,CAAAA;AACvEN,QAAAA,cAAAA,CAAegB,GAAG,CAACD,OAAAA,CAAAA;QACnB,OAAOZ,QAAAA;AACX,KAAA;AACAA,IAAAA,QAAAA,CAAS4B,cAAc,GAAGA,cAAAA;AAE1B,IAAA,MAAMC,kBAAqB,GAAA,OACvBX,WACAhB,EAAAA,cAAAA,GAA0C,EAAE,GAAA;QAE5ChC,QAAOqC,CAAAA,KAAK,CAAC,yBAA2BW,EAAAA,WAAAA,CAAAA;AACxC,QAAA,MAAMf,iBAAiBF,WAAYC,CAAAA,cAAAA,CAAAA;QACnC,MAAM4B,WAAAA,GAAoC,MAAMb,QAAAA,CAAsBC,WAAaf,EAAAA,cAAAA,CAAAA;AACnFL,QAAAA,kBAAAA,CAAmBe,GAAG,CAACiB,WAAAA,CAAAA;QACvB,OAAO9B,QAAAA;AACX,KAAA;AACAA,IAAAA,QAAAA,CAAS6B,kBAAkB,GAAGA,kBAAAA;AAE9B,IAAA,MAAME,cAAiB,GAAA,OACnBb,WACAhB,EAAAA,cAAAA,GAA0C,EAAE,GAAA;QAE5ChC,QAAOqC,CAAAA,KAAK,CAAC,qBAAuBW,EAAAA,WAAAA,CAAAA;AACpC,QAAA,MAAMf,iBAAiBF,WAAYC,CAAAA,cAAAA,CAAAA;QACnC,MAAMc,OAAAA,GAA4B,MAAMC,QAAAA,CAAkBC,WAAaf,EAAAA,cAAAA,CAAAA;AACvEJ,QAAAA,cAAAA,CAAec,GAAG,CAACG,OAAAA,CAAAA;QACnB,OAAOhB,QAAAA;AACX,KAAA;AACAA,IAAAA,QAAAA,CAAS+B,cAAc,GAAGA,cAAAA;AAE1B,IAAA,MAAMC,UAAa,GAAA,OACfhB,OACAd,EAAAA,cAAAA,GAA0C,EAAE,GAAA;QAE5ChC,QAAOqC,CAAAA,KAAK,CAAC,gBAAA,EAAkB,OAAOS,OAAAA,CAAAA;AACtC,QAAA,MAAMb,iBAAiBF,WAAYC,CAAAA,cAAAA,CAAAA;AACnC,QAAA,MAAM+B,oBAAyC9C,GAAAA,QAAAA,CAAOF,KAAK,CAAU+B,OAASb,EAAAA,cAAAA,CAAAA;AAC9EJ,QAAAA,cAAAA,CAAec,GAAG,CAACoB,oBAAAA,CAAAA;QACnB,OAAOjC,QAAAA;AACX,KAAA;AACAA,IAAAA,QAAAA,CAASgC,UAAU,GAAGA,UAAAA;AAEtB,IAAA,MAAME,UAAa,GAAA,OACftB,OACAV,EAAAA,cAAAA,GAA0C,EAAE,GAAA;QAE5ChC,QAAOqC,CAAAA,KAAK,CAAC,gBAAA,EAAkB,OAAOK,OAAAA,CAAAA;AACtC,QAAA,MAAMT,iBAAiBF,WAAYC,CAAAA,cAAAA,CAAAA;AACnC,QAAA,MAAMiC,oBAAyChD,GAAAA,QAAAA,CAAOF,KAAK,CAAU2B,OAAST,EAAAA,cAAAA,CAAAA;AAC9EN,QAAAA,cAAAA,CAAegB,GAAG,CAACsB,oBAAAA,CAAAA;QACnB,OAAOnC,QAAAA;AACX,KAAA;AACAA,IAAAA,QAAAA,CAASkC,UAAU,GAAGA,UAAAA;AAEtB,IAAA,MAAME,KAAQ,GAAA,UAAA;QACVlE,QAAOqC,CAAAA,KAAK,CAAC,iBAAA,EAAmB,EAAC,CAAA;AACjC,QAAA,MAAM8B,WAASC,aAAa,CAAA;YAAEX,OAASjC,EAAAA,cAAAA;YAAgB6C,QAAU1C,EAAAA,cAAAA;YAAgB2C,YAAc1C,EAAAA,kBAAAA;YAAoB2C,QAAU1C,EAAAA;AAAe,SAAA,CAAA;QAC5I,OAAOsC,QAAAA;AACX,KAAA;AACArC,IAAAA,QAAAA,CAASoC,KAAK,GAAGA,KAAAA;IAEjB,OAAOpC,QAAAA;AACX;;;;"}

package/dist/builder.d.ts

@@ -0,0 +1,37 @@
+import { z } from 'zod';
+import { SectionOptions } from './items/section';
+import { Prompt } from './riotprompt';
+declare const OptionSchema: z.ZodObject<{
+    logger: z.ZodDefault<z.ZodOptional<z.ZodAny>>;
+    basePath: z.ZodString;
+    overridePath: z.ZodDefault<z.ZodOptional<z.ZodString>>;
+    overrides: z.ZodDefault<z.ZodOptional<z.ZodBoolean>>;
+    parameters: z.ZodDefault<z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodUnion<[z.ZodString, z.ZodNumber, z.ZodBoolean, z.ZodArray<z.ZodUnion<[z.ZodString, z.ZodNumber, z.ZodBoolean]>, "many">]>>>>;
+}, "strip", z.ZodTypeAny, {
+    parameters: Record<string, string | number | boolean | (string | number | boolean)[]>;
+    overrides: boolean;
+    basePath: string;
+    overridePath: string;
+    logger?: any;
+}, {
+    basePath: string;
+    parameters?: Record<string, string | number | boolean | (string | number | boolean)[]> | undefined;
+    logger?: any;
+    overrides?: boolean | undefined;
+    overridePath?: string | undefined;
+}>;
+export type Options = z.infer<typeof OptionSchema>;
+export type OptionsParam = Required<Pick<Options, 'basePath'>> & Partial<Omit<Options, 'basePath'>>;
+export interface Instance {
+    addPersonaPath(contentPath: string, sectionOptions?: Partial<SectionOptions>): Promise<Instance>;
+    addContextPath(contentPath: string, sectionOptions?: Partial<SectionOptions>): Promise<Instance>;
+    addInstructionPath(contentPath: string, sectionOptions?: Partial<SectionOptions>): Promise<Instance>;
+    addContentPath(contentPath: string, sectionOptions?: Partial<SectionOptions>): Promise<Instance>;
+    addContent(content: string, sectionOptions?: Partial<SectionOptions>): Promise<Instance>;
+    addContext(context: string, sectionOptions?: Partial<SectionOptions>): Promise<Instance>;
+    loadContext(contextDirectories: string[], sectionOptions?: Partial<SectionOptions>): Promise<Instance>;
+    loadContent(contentDirectories: string[], sectionOptions?: Partial<SectionOptions>): Promise<Instance>;
+    build(): Promise<Prompt>;
+}
+export declare const create: (builderOptions: OptionsParam) => Instance;
+export {};

package/dist/builder.js

@@ -0,0 +1,148 @@
+import path__default from 'path';
+import { z } from 'zod';
+import { ParametersSchema } from './items/parameters.js';
+import { create as create$4, SectionOptionsSchema } from './items/section.js';
+import { DEFAULT_LOGGER, wrapLogger } from './logger.js';
+import './items/weighted.js';
+import { create as create$5 } from './prompt.js';
+import './formatter.js';
+import { create as create$1 } from './parser.js';
+import { create as create$3 } from './loader.js';
+import { create as create$2 } from './override.js';
+
+const OptionSchema = z.object({
+    logger: z.any().optional().default(DEFAULT_LOGGER),
+    basePath: z.string(),
+    overridePath: z.string().optional().default("./"),
+    overrides: z.boolean().optional().default(false),
+    parameters: ParametersSchema.optional().default({})
+});
+const create = (builderOptions)=>{
+    const options = OptionSchema.parse(builderOptions);
+    const logger = wrapLogger(options.logger, 'Builder');
+    const parser$1 = create$1({
+        logger
+    });
+    const override$1 = create$2({
+        logger,
+        configDir: options.overridePath || "./",
+        overrides: options.overrides || false
+    });
+    const loader$1 = create$3({
+        logger
+    });
+    const personaSection = create$4({
+        title: "Persona"
+    });
+    const contextSection = create$4({
+        title: "Context"
+    });
+    const instructionSection = create$4({
+        title: "Instruction"
+    });
+    const contentSection = create$4({
+        title: "Content"
+    });
+    const parameters = options.parameters;
+    const instance = {};
+    const loadOptions = (sectionOptions = {})=>{
+        const currentOptions = SectionOptionsSchema.parse(sectionOptions);
+        return {
+            ...currentOptions,
+            parameters: {
+                ...parameters,
+                ...currentOptions.parameters
+            }
+        };
+    };
+    const loadDirectories = async (directories, sectionOptions = {})=>{
+        const currentOptions = loadOptions(sectionOptions);
+        logger.debug("Loading directories", directories);
+        const sections = await loader$1.load(directories, currentOptions);
+        return sections;
+    };
+    const loadContext = async (contextDirectories, sectionOptions = {})=>{
+        const currentOptions = loadOptions(sectionOptions);
+        logger.debug('Loading context', contextDirectories);
+        const context = await loadDirectories(contextDirectories, currentOptions);
+        contextSection.add(context);
+        return instance;
+    };
+    instance.loadContext = loadContext;
+    const loadContent = async (contentDirectories, sectionOptions = {})=>{
+        const currentOptions = loadOptions(sectionOptions);
+        const content = await loadDirectories(contentDirectories, currentOptions);
+        contentSection.add(content);
+        return instance;
+    };
+    instance.loadContent = loadContent;
+    const loadPath = async (contentPath, sectionOptions = {})=>{
+        const currentOptions = loadOptions(sectionOptions);
+        const defaultPath = path__default.join(options.basePath, contentPath);
+        const section = await parser$1.parseFile(defaultPath, currentOptions);
+        const overrideSection = await override$1.customize(contentPath, section, currentOptions);
+        return overrideSection;
+    };
+    const addPersonaPath = async (contentPath, sectionOptions = {})=>{
+        const currentOptions = loadOptions(sectionOptions);
+        const persona = await loadPath(contentPath, currentOptions);
+        personaSection.add(persona);
+        return instance;
+    };
+    instance.addPersonaPath = addPersonaPath;
+    const addContextPath = async (contentPath, sectionOptions = {})=>{
+        logger.debug("Adding context path", contentPath);
+        const currentOptions = loadOptions(sectionOptions);
+        const context = await loadPath(contentPath, currentOptions);
+        contextSection.add(context);
+        return instance;
+    };
+    instance.addContextPath = addContextPath;
+    const addInstructionPath = async (contentPath, sectionOptions = {})=>{
+        logger.debug("Adding instruction path", contentPath);
+        const currentOptions = loadOptions(sectionOptions);
+        const instruction = await loadPath(contentPath, currentOptions);
+        instructionSection.add(instruction);
+        return instance;
+    };
+    instance.addInstructionPath = addInstructionPath;
+    const addContentPath = async (contentPath, sectionOptions = {})=>{
+        logger.debug("Adding content path", contentPath);
+        const currentOptions = loadOptions(sectionOptions);
+        const content = await loadPath(contentPath, currentOptions);
+        contentSection.add(content);
+        return instance;
+    };
+    instance.addContentPath = addContentPath;
+    const addContent = async (content, sectionOptions = {})=>{
+        logger.debug("Adding content", typeof content);
+        const currentOptions = loadOptions(sectionOptions);
+        const parsedContentSection = parser$1.parse(content, currentOptions);
+        contentSection.add(parsedContentSection);
+        return instance;
+    };
+    instance.addContent = addContent;
+    const addContext = async (context, sectionOptions = {})=>{
+        logger.debug("Adding context", typeof context);
+        const currentOptions = loadOptions(sectionOptions);
+        const parsedContextSection = parser$1.parse(context, currentOptions);
+        contextSection.add(parsedContextSection);
+        return instance;
+    };
+    instance.addContext = addContext;
+    const build = async ()=>{
+        logger.debug("Building prompt", {});
+        const prompt = create$5({
+            persona: personaSection,
+            contexts: contextSection,
+            instructions: instructionSection,
+            contents: contentSection
+        });
+        return prompt;
+    };
+    instance.build = build;
+    return instance;
+};
+
+export { create };
+//# sourceMappingURL=builder.js.map

package/dist/builder.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"builder.js","sources":["../src/builder.ts"],"sourcesContent":["import path from \"path\";\nimport { z } from \"zod\";\nimport { ParametersSchema } from \"./items/parameters\";\nimport { SectionOptions, SectionOptionsSchema } from \"./items/section\";\nimport { DEFAULT_LOGGER, wrapLogger } from \"./logger\";\nimport { Content, Context, createPrompt, createSection, Instruction, Loader, Override, Parser, Prompt, Section, Weighted } from \"./riotprompt\";\n\nconst OptionSchema = z.object({\n    logger: z.any().optional().default(DEFAULT_LOGGER),\n    basePath: z.string(),\n    overridePath: z.string().optional().default(\"./\"),\n    overrides: z.boolean().optional().default(false),\n    parameters: ParametersSchema.optional().default({}),\n});\n\nexport type Options = z.infer<typeof OptionSchema>;\n\nexport type OptionsParam = Required<Pick<Options, 'basePath'>> & Partial<Omit<Options, 'basePath'>>;\n\nexport interface Instance {\n    addPersonaPath(contentPath: string, sectionOptions?: Partial<SectionOptions>): Promise<Instance>;\n    addContextPath(contentPath: string, sectionOptions?: Partial<SectionOptions>): Promise<Instance>;\n    addInstructionPath(contentPath: string, sectionOptions?: Partial<SectionOptions>): Promise<Instance>;\n    addContentPath(contentPath: string, sectionOptions?: Partial<SectionOptions>): Promise<Instance>;\n    addContent(content: string, sectionOptions?: Partial<SectionOptions>): Promise<Instance>;\n    addContext(context: string, sectionOptions?: Partial<SectionOptions>): Promise<Instance>;\n    loadContext(contextDirectories: string[], sectionOptions?: Partial<SectionOptions>): Promise<Instance>;\n    loadContent(contentDirectories: string[], sectionOptions?: Partial<SectionOptions>): Promise<Instance>;\n    build(): Promise<Prompt>;\n}\n\nexport const create = (builderOptions: OptionsParam): Instance => {\n    const options: Required<Options> = OptionSchema.parse(builderOptions) as Required<Options>;\n\n    const logger = wrapLogger(options.logger, 'Builder');\n    const parser = Parser.create({ logger });\n    const override = Override.create({\n        logger, configDir: options.overridePath || \"./\",\n        overrides: options.overrides || false\n    });\n    const loader = Loader.create({ logger });\n\n    const personaSection: Section<Instruction> = createSection({ title: \"Persona\" });\n    const contextSection: Section<Context> = createSection({ title: \"Context\" });\n    const instructionSection: Section<Instruction> = createSection({ title: \"Instruction\" });\n    const contentSection: Section<Content> = createSection({ title: \"Content\" });\n    const parameters = options.parameters;\n\n\n    const instance: Partial<Instance> = {}\n\n    const loadOptions = (sectionOptions: Partial<SectionOptions> = {}): SectionOptions => {\n        const currentOptions = SectionOptionsSchema.parse(sectionOptions);\n        return {\n            ...currentOptions,\n            parameters: {\n                ...parameters,\n                ...currentOptions.parameters\n            }\n        }\n    }\n\n    const loadDirectories = async <T extends Weighted>(\n        directories: string[],\n        sectionOptions: Partial<SectionOptions> = {}\n    ): Promise<Section<T>[]> => {\n        const currentOptions = loadOptions(sectionOptions);\n        logger.debug(\"Loading directories\", directories);\n        const sections: Section<T>[] = await loader.load<T>(directories, currentOptions);\n        return sections;\n    }\n\n    const loadContext = async (\n        contextDirectories: string[],\n        sectionOptions: Partial<SectionOptions> = {}\n    ): Promise<Instance> => {\n        const currentOptions = loadOptions(sectionOptions);\n        logger.debug('Loading context', contextDirectories);\n        const context: Section<Context>[] = await loadDirectories<Context>(contextDirectories, currentOptions);\n        contextSection.add(context);\n        return instance as Instance;\n    }\n    instance.loadContext = loadContext;\n\n    const loadContent = async (\n        contentDirectories: string[],\n        sectionOptions: Partial<SectionOptions> = {}\n    ): Promise<Instance> => {\n        const currentOptions = loadOptions(sectionOptions);\n        const content: Section<Content>[] = await loadDirectories<Content>(contentDirectories, currentOptions);\n        contentSection.add(content);\n        return instance as Instance;\n    }\n    instance.loadContent = loadContent;\n\n    const loadPath = async <T extends Weighted>(\n        contentPath: string,\n        sectionOptions: Partial<SectionOptions> = {}\n    ): Promise<Section<T>> => {\n        const currentOptions = loadOptions(sectionOptions);\n        const defaultPath = path.join(options.basePath as string, contentPath);\n        const section: Section<T> = await parser.parseFile<T>(defaultPath, currentOptions);\n        const overrideSection = await override.customize<T>(contentPath, section, currentOptions);\n        return overrideSection;\n    }\n\n    const addPersonaPath = async (\n        contentPath: string,\n        sectionOptions: Partial<SectionOptions> = {}\n    ): Promise<Instance> => {\n        const currentOptions = loadOptions(sectionOptions);\n        const persona: Section<Instruction> = await loadPath<Instruction>(contentPath, currentOptions);\n        personaSection.add(persona);\n        return instance as Instance;\n    }\n    instance.addPersonaPath = addPersonaPath;\n\n    const addContextPath = async (\n        contentPath: string,\n        sectionOptions: Partial<SectionOptions> = {}\n    ): Promise<Instance> => {\n        logger.debug(\"Adding context path\", contentPath);\n        const currentOptions = loadOptions(sectionOptions);\n        const context: Section<Context> = await loadPath<Context>(contentPath, currentOptions);\n        contextSection.add(context);\n        return instance as Instance;\n    }\n    instance.addContextPath = addContextPath;\n\n    const addInstructionPath = async (\n        contentPath: string,\n        sectionOptions: Partial<SectionOptions> = {}\n    ): Promise<Instance> => {\n        logger.debug(\"Adding instruction path\", contentPath);\n        const currentOptions = loadOptions(sectionOptions);\n        const instruction: Section<Instruction> = await loadPath<Instruction>(contentPath, currentOptions);\n        instructionSection.add(instruction);\n        return instance as Instance;\n    }\n    instance.addInstructionPath = addInstructionPath;\n\n    const addContentPath = async (\n        contentPath: string,\n        sectionOptions: Partial<SectionOptions> = {}\n    ): Promise<Instance> => {\n        logger.debug(\"Adding content path\", contentPath);\n        const currentOptions = loadOptions(sectionOptions);\n        const content: Section<Content> = await loadPath<Content>(contentPath, currentOptions);\n        contentSection.add(content);\n        return instance as Instance;\n    }\n    instance.addContentPath = addContentPath;\n\n    const addContent = async (\n        content: string | Buffer,\n        sectionOptions: Partial<SectionOptions> = {}\n    ): Promise<Instance> => {\n        logger.debug(\"Adding content\", typeof content);\n        const currentOptions = loadOptions(sectionOptions);\n        const parsedContentSection: Section<Content> = parser.parse<Content>(content, currentOptions);\n        contentSection.add(parsedContentSection);\n        return instance as Instance;\n    }\n    instance.addContent = addContent;\n\n    const addContext = async (\n        context: string | Buffer,\n        sectionOptions: Partial<SectionOptions> = {}\n    ): Promise<Instance> => {\n        logger.debug(\"Adding context\", typeof context);\n        const currentOptions = loadOptions(sectionOptions);\n        const parsedContextSection: Section<Context> = parser.parse<Context>(context, currentOptions);\n        contextSection.add(parsedContextSection);\n        return instance as Instance;\n    }\n    instance.addContext = addContext;\n\n    const build = async () => {\n        logger.debug(\"Building prompt\", {});\n        const prompt = createPrompt({ persona: personaSection, contexts: contextSection, instructions: instructionSection, contents: contentSection });\n        return prompt;\n    }\n    instance.build = build;\n\n    return instance as Instance;\n}\n"],"names":["OptionSchema","z","object","logger","any","optional","default","DEFAULT_LOGGER","basePath","string","overridePath","overrides","boolean","parameters","ParametersSchema","create","builderOptions","options","parse","wrapLogger","parser","Parser","override","Override","configDir","loader","Loader","personaSection","createSection","title","contextSection","instructionSection","contentSection","instance","loadOptions","sectionOptions","currentOptions","SectionOptionsSchema","loadDirectories","directories","debug","sections","load","loadContext","contextDirectories","context","add","loadContent","contentDirectories","content","loadPath","contentPath","defaultPath","path","join","section","parseFile","overrideSection","customize","addPersonaPath","persona","addContextPath","addInstructionPath","instruction","addContentPath","addContent","parsedContentSection","addContext","parsedContextSection","build","prompt","createPrompt","contexts","instructions","contents"],"mappings":";;;;;;;;;;;;AAOA,MAAMA,YAAAA,GAAeC,CAAEC,CAAAA,MAAM,CAAC;AAC1BC,IAAAA,MAAAA,EAAQF,EAAEG,GAAG,EAAA,CAAGC,QAAQ,EAAA,CAAGC,OAAO,CAACC,cAAAA,CAAAA;AACnCC,IAAAA,QAAAA,EAAUP,EAAEQ,MAAM,EAAA;AAClBC,IAAAA,YAAAA,EAAcT,EAAEQ,MAAM,EAAA,CAAGJ,QAAQ,EAAA,CAAGC,OAAO,CAAC,IAAA,CAAA;AAC5CK,IAAAA,SAAAA,EAAWV,EAAEW,OAAO,EAAA,CAAGP,QAAQ,EAAA,CAAGC,OAAO,CAAC,KAAA,CAAA;AAC1CO,IAAAA,UAAAA,EAAYC,gBAAiBT,CAAAA,QAAQ,EAAGC,CAAAA,OAAO,CAAC,EAAC;AACrD,CAAA,CAAA;AAkBO,MAAMS,SAAS,CAACC,cAAAA,GAAAA;IACnB,MAAMC,OAAAA,GAA6BjB,YAAakB,CAAAA,KAAK,CAACF,cAAAA,CAAAA;AAEtD,IAAA,MAAMb,MAASgB,GAAAA,UAAAA,CAAWF,OAAQd,CAAAA,MAAM,EAAE,SAAA,CAAA;IAC1C,MAAMiB,QAAAA,GAASC,QAAa,CAAC;AAAElB,QAAAA;AAAO,KAAA,CAAA;IACtC,MAAMmB,UAAAA,GAAWC,QAAe,CAAC;AAC7BpB,QAAAA,MAAAA;QAAQqB,SAAWP,EAAAA,OAAAA,CAAQP,YAAY,IAAI,IAAA;QAC3CC,SAAWM,EAAAA,OAAAA,CAAQN,SAAS,IAAI;AACpC,KAAA,CAAA;IACA,MAAMc,QAAAA,GAASC,QAAa,CAAC;AAAEvB,QAAAA;AAAO,KAAA,CAAA;AAEtC,IAAA,MAAMwB,iBAAuCC,QAAc,CAAA;QAAEC,KAAO,EAAA;AAAU,KAAA,CAAA;AAC9E,IAAA,MAAMC,iBAAmCF,QAAc,CAAA;QAAEC,KAAO,EAAA;AAAU,KAAA,CAAA;AAC1E,IAAA,MAAME,qBAA2CH,QAAc,CAAA;QAAEC,KAAO,EAAA;AAAc,KAAA,CAAA;AACtF,IAAA,MAAMG,iBAAmCJ,QAAc,CAAA;QAAEC,KAAO,EAAA;AAAU,KAAA,CAAA;IAC1E,MAAMhB,UAAAA,GAAaI,QAAQJ,UAAU;AAGrC,IAAA,MAAMoB,WAA8B,EAAC;AAErC,IAAA,MAAMC,WAAc,GAAA,CAACC,cAA0C,GAAA,EAAE,GAAA;QAC7D,MAAMC,cAAAA,GAAiBC,oBAAqBnB,CAAAA,KAAK,CAACiB,cAAAA,CAAAA;QAClD,OAAO;AACH,YAAA,GAAGC,cAAc;YACjBvB,UAAY,EAAA;AACR,gBAAA,GAAGA,UAAU;AACb,gBAAA,GAAGuB,eAAevB;AACtB;AACJ,SAAA;AACJ,KAAA;AAEA,IAAA,MAAMyB,eAAkB,GAAA,OACpBC,WACAJ,EAAAA,cAAAA,GAA0C,EAAE,GAAA;AAE5C,QAAA,MAAMC,iBAAiBF,WAAYC,CAAAA,cAAAA,CAAAA;QACnChC,MAAOqC,CAAAA,KAAK,CAAC,qBAAuBD,EAAAA,WAAAA,CAAAA;AACpC,QAAA,MAAME,QAAyB,GAAA,MAAMhB,QAAOiB,CAAAA,IAAI,CAAIH,WAAaH,EAAAA,cAAAA,CAAAA;QACjE,OAAOK,QAAAA;AACX,KAAA;AAEA,IAAA,MAAME,WAAc,GAAA,OAChBC,kBACAT,EAAAA,cAAAA,GAA0C,EAAE,GAAA;AAE5C,QAAA,MAAMC,iBAAiBF,WAAYC,CAAAA,cAAAA,CAAAA;QACnChC,MAAOqC,CAAAA,KAAK,CAAC,iBAAmBI,EAAAA,kBAAAA,CAAAA;QAChC,MAAMC,OAAAA,GAA8B,MAAMP,eAAAA,CAAyBM,kBAAoBR,EAAAA,cAAAA,CAAAA;AACvFN,QAAAA,cAAAA,CAAegB,GAAG,CAACD,OAAAA,CAAAA;QACnB,OAAOZ,QAAAA;AACX,KAAA;AACAA,IAAAA,QAAAA,CAASU,WAAW,GAAGA,WAAAA;AAEvB,IAAA,MAAMI,WAAc,GAAA,OAChBC,kBACAb,EAAAA,cAAAA,GAA0C,EAAE,GAAA;AAE5C,QAAA,MAAMC,iBAAiBF,WAAYC,CAAAA,cAAAA,CAAAA;QACnC,MAAMc,OAAAA,GAA8B,MAAMX,eAAAA,CAAyBU,kBAAoBZ,EAAAA,cAAAA,CAAAA;AACvFJ,QAAAA,cAAAA,CAAec,GAAG,CAACG,OAAAA,CAAAA;QACnB,OAAOhB,QAAAA;AACX,KAAA;AACAA,IAAAA,QAAAA,CAASc,WAAW,GAAGA,WAAAA;AAEvB,IAAA,MAAMG,QAAW,GAAA,OACbC,WACAhB,EAAAA,cAAAA,GAA0C,EAAE,GAAA;AAE5C,QAAA,MAAMC,iBAAiBF,WAAYC,CAAAA,cAAAA,CAAAA;AACnC,QAAA,MAAMiB,cAAcC,aAAKC,CAAAA,IAAI,CAACrC,OAAAA,CAAQT,QAAQ,EAAY2C,WAAAA,CAAAA;AAC1D,QAAA,MAAMI,OAAsB,GAAA,MAAMnC,QAAOoC,CAAAA,SAAS,CAAIJ,WAAahB,EAAAA,cAAAA,CAAAA;AACnE,QAAA,MAAMqB,kBAAkB,MAAMnC,UAAAA,CAASoC,SAAS,CAAIP,aAAaI,OAASnB,EAAAA,cAAAA,CAAAA;QAC1E,OAAOqB,eAAAA;AACX,KAAA;AAEA,IAAA,MAAME,cAAiB,GAAA,OACnBR,WACAhB,EAAAA,cAAAA,GAA0C,EAAE,GAAA;AAE5C,QAAA,MAAMC,iBAAiBF,WAAYC,CAAAA,cAAAA,CAAAA;QACnC,MAAMyB,OAAAA,GAAgC,MAAMV,QAAAA,CAAsBC,WAAaf,EAAAA,cAAAA,CAAAA;AAC/ET,QAAAA,cAAAA,CAAemB,GAAG,CAACc,OAAAA,CAAAA;QACnB,OAAO3B,QAAAA;AACX,KAAA;AACAA,IAAAA,QAAAA,CAAS0B,cAAc,GAAGA,cAAAA;AAE1B,IAAA,MAAME,cAAiB,GAAA,OACnBV,WACAhB,EAAAA,cAAAA,GAA0C,EAAE,GAAA;QAE5ChC,MAAOqC,CAAAA,KAAK,CAAC,qBAAuBW,EAAAA,WAAAA,CAAAA;AACpC,QAAA,MAAMf,iBAAiBF,WAAYC,CAAAA,cAAAA,CAAAA;QACnC,MAAMU,OAAAA,GAA4B,MAAMK,QAAAA,CAAkBC,WAAaf,EAAAA,cAAAA,CAAAA;AACvEN,QAAAA,cAAAA,CAAegB,GAAG,CAACD,OAAAA,CAAAA;QACnB,OAAOZ,QAAAA;AACX,KAAA;AACAA,IAAAA,QAAAA,CAAS4B,cAAc,GAAGA,cAAAA;AAE1B,IAAA,MAAMC,kBAAqB,GAAA,OACvBX,WACAhB,EAAAA,cAAAA,GAA0C,EAAE,GAAA;QAE5ChC,MAAOqC,CAAAA,KAAK,CAAC,yBAA2BW,EAAAA,WAAAA,CAAAA;AACxC,QAAA,MAAMf,iBAAiBF,WAAYC,CAAAA,cAAAA,CAAAA;QACnC,MAAM4B,WAAAA,GAAoC,MAAMb,QAAAA,CAAsBC,WAAaf,EAAAA,cAAAA,CAAAA;AACnFL,QAAAA,kBAAAA,CAAmBe,GAAG,CAACiB,WAAAA,CAAAA;QACvB,OAAO9B,QAAAA;AACX,KAAA;AACAA,IAAAA,QAAAA,CAAS6B,kBAAkB,GAAGA,kBAAAA;AAE9B,IAAA,MAAME,cAAiB,GAAA,OACnBb,WACAhB,EAAAA,cAAAA,GAA0C,EAAE,GAAA;QAE5ChC,MAAOqC,CAAAA,KAAK,CAAC,qBAAuBW,EAAAA,WAAAA,CAAAA;AACpC,QAAA,MAAMf,iBAAiBF,WAAYC,CAAAA,cAAAA,CAAAA;QACnC,MAAMc,OAAAA,GAA4B,MAAMC,QAAAA,CAAkBC,WAAaf,EAAAA,cAAAA,CAAAA;AACvEJ,QAAAA,cAAAA,CAAec,GAAG,CAACG,OAAAA,CAAAA;QACnB,OAAOhB,QAAAA;AACX,KAAA;AACAA,IAAAA,QAAAA,CAAS+B,cAAc,GAAGA,cAAAA;AAE1B,IAAA,MAAMC,UAAa,GAAA,OACfhB,OACAd,EAAAA,cAAAA,GAA0C,EAAE,GAAA;QAE5ChC,MAAOqC,CAAAA,KAAK,CAAC,gBAAA,EAAkB,OAAOS,OAAAA,CAAAA;AACtC,QAAA,MAAMb,iBAAiBF,WAAYC,CAAAA,cAAAA,CAAAA;AACnC,QAAA,MAAM+B,oBAAyC9C,GAAAA,QAAAA,CAAOF,KAAK,CAAU+B,OAASb,EAAAA,cAAAA,CAAAA;AAC9EJ,QAAAA,cAAAA,CAAec,GAAG,CAACoB,oBAAAA,CAAAA;QACnB,OAAOjC,QAAAA;AACX,KAAA;AACAA,IAAAA,QAAAA,CAASgC,UAAU,GAAGA,UAAAA;AAEtB,IAAA,MAAME,UAAa,GAAA,OACftB,OACAV,EAAAA,cAAAA,GAA0C,EAAE,GAAA;QAE5ChC,MAAOqC,CAAAA,KAAK,CAAC,gBAAA,EAAkB,OAAOK,OAAAA,CAAAA;AACtC,QAAA,MAAMT,iBAAiBF,WAAYC,CAAAA,cAAAA,CAAAA;AACnC,QAAA,MAAMiC,oBAAyChD,GAAAA,QAAAA,CAAOF,KAAK,CAAU2B,OAAST,EAAAA,cAAAA,CAAAA;AAC9EN,QAAAA,cAAAA,CAAegB,GAAG,CAACsB,oBAAAA,CAAAA;QACnB,OAAOnC,QAAAA;AACX,KAAA;AACAA,IAAAA,QAAAA,CAASkC,UAAU,GAAGA,UAAAA;AAEtB,IAAA,MAAME,KAAQ,GAAA,UAAA;QACVlE,MAAOqC,CAAAA,KAAK,CAAC,iBAAA,EAAmB,EAAC,CAAA;AACjC,QAAA,MAAM8B,SAASC,QAAa,CAAA;YAAEX,OAASjC,EAAAA,cAAAA;YAAgB6C,QAAU1C,EAAAA,cAAAA;YAAgB2C,YAAc1C,EAAAA,kBAAAA;YAAoB2C,QAAU1C,EAAAA;AAAe,SAAA,CAAA;QAC5I,OAAOsC,MAAAA;AACX,KAAA;AACArC,IAAAA,QAAAA,CAASoC,KAAK,GAAGA,KAAAA;IAEjB,OAAOpC,QAAAA;AACX;;;;"}

package/dist/chat.cjs

@@ -0,0 +1,26 @@
+'use strict';
+
+Object.defineProperty(exports, Symbol.toStringTag, { value: 'Module' });
+
+const constants = require('./constants.cjs');
+
+const getPersonaRole = (model)=>{
+    if (model === "gpt-4o" || model === "gpt-4o-mini") {
+        return "system";
+    }
+    return constants.DEFAULT_PERSONA_ROLE;
+};
+const createRequest = (model)=>{
+    const messages = [];
+    return {
+        model,
+        messages,
+        addMessage: (message)=>{
+            messages.push(message);
+        }
+    };
+};
+
+exports.createRequest = createRequest;
+exports.getPersonaRole = getPersonaRole;
+//# sourceMappingURL=chat.cjs.map

package/dist/chat.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"chat.cjs","sources":["../src/chat.ts"],"sourcesContent":["import { DEFAULT_PERSONA_ROLE } from \"./constants\";\n\nexport type Role = \"user\" | \"assistant\" | \"system\" | \"developer\";\n\nexport type Model = \"gpt-4o\" | \"gpt-4o-mini\" | \"o1-preview\" | \"o1-mini\" | \"o1\" | \"o3-mini\" | \"o1-pro\";\n\nexport interface Message {\n    role: Role;\n    content: string | string[];\n    name?: string;\n}\n\nexport interface Request {\n    messages: Message[];\n    model: Model;\n\n    addMessage(message: Message): void;\n}\n\nexport const getPersonaRole = (model: Model): Role => {\n    if (model === \"gpt-4o\" || model === \"gpt-4o-mini\") {\n        return \"system\";\n    }\n    return DEFAULT_PERSONA_ROLE;\n}\n\nexport const createRequest = (model: Model): Request => {\n    const messages: Message[] = [];\n\n    return {\n        model,\n        messages,\n        addMessage: (message: Message) => {\n            messages.push(message);\n        }\n    }\n}\n"],"names":["getPersonaRole","model","DEFAULT_PERSONA_ROLE","createRequest","messages","addMessage","message","push"],"mappings":";;;;;;AAmBO,MAAMA,iBAAiB,CAACC,KAAAA,GAAAA;IAC3B,IAAIA,KAAAA,KAAU,QAAYA,IAAAA,KAAAA,KAAU,aAAe,EAAA;QAC/C,OAAO,QAAA;AACX;IACA,OAAOC,8BAAAA;AACX;AAEO,MAAMC,gBAAgB,CAACF,KAAAA,GAAAA;AAC1B,IAAA,MAAMG,WAAsB,EAAE;IAE9B,OAAO;AACHH,QAAAA,KAAAA;AACAG,QAAAA,QAAAA;AACAC,QAAAA,UAAAA,EAAY,CAACC,OAAAA,GAAAA;AACTF,YAAAA,QAAAA,CAASG,IAAI,CAACD,OAAAA,CAAAA;AAClB;AACJ,KAAA;AACJ;;;;;"}

package/dist/chat.d.ts

@@ -0,0 +1,14 @@
+export type Role = "user" | "assistant" | "system" | "developer";
+export type Model = "gpt-4o" | "gpt-4o-mini" | "o1-preview" | "o1-mini" | "o1" | "o3-mini" | "o1-pro";
+export interface Message {
+    role: Role;
+    content: string | string[];
+    name?: string;
+}
+export interface Request {
+    messages: Message[];
+    model: Model;
+    addMessage(message: Message): void;
+}
+export declare const getPersonaRole: (model: Model) => Role;
+export declare const createRequest: (model: Model) => Request;

package/dist/chat.js

@@ -0,0 +1,21 @@
+import { DEFAULT_PERSONA_ROLE } from './constants.js';
+
+const getPersonaRole = (model)=>{
+    if (model === "gpt-4o" || model === "gpt-4o-mini") {
+        return "system";
+    }
+    return DEFAULT_PERSONA_ROLE;
+};
+const createRequest = (model)=>{
+    const messages = [];
+    return {
+        model,
+        messages,
+        addMessage: (message)=>{
+            messages.push(message);
+        }
+    };
+};
+
+export { createRequest, getPersonaRole };
+//# sourceMappingURL=chat.js.map

package/dist/chat.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"chat.js","sources":["../src/chat.ts"],"sourcesContent":["import { DEFAULT_PERSONA_ROLE } from \"./constants\";\n\nexport type Role = \"user\" | \"assistant\" | \"system\" | \"developer\";\n\nexport type Model = \"gpt-4o\" | \"gpt-4o-mini\" | \"o1-preview\" | \"o1-mini\" | \"o1\" | \"o3-mini\" | \"o1-pro\";\n\nexport interface Message {\n    role: Role;\n    content: string | string[];\n    name?: string;\n}\n\nexport interface Request {\n    messages: Message[];\n    model: Model;\n\n    addMessage(message: Message): void;\n}\n\nexport const getPersonaRole = (model: Model): Role => {\n    if (model === \"gpt-4o\" || model === \"gpt-4o-mini\") {\n        return \"system\";\n    }\n    return DEFAULT_PERSONA_ROLE;\n}\n\nexport const createRequest = (model: Model): Request => {\n    const messages: Message[] = [];\n\n    return {\n        model,\n        messages,\n        addMessage: (message: Message) => {\n            messages.push(message);\n        }\n    }\n}\n"],"names":["getPersonaRole","model","DEFAULT_PERSONA_ROLE","createRequest","messages","addMessage","message","push"],"mappings":";;AAmBO,MAAMA,iBAAiB,CAACC,KAAAA,GAAAA;IAC3B,IAAIA,KAAAA,KAAU,QAAYA,IAAAA,KAAAA,KAAU,aAAe,EAAA;QAC/C,OAAO,QAAA;AACX;IACA,OAAOC,oBAAAA;AACX;AAEO,MAAMC,gBAAgB,CAACF,KAAAA,GAAAA;AAC1B,IAAA,MAAMG,WAAsB,EAAE;IAE9B,OAAO;AACHH,QAAAA,KAAAA;AACAG,QAAAA,QAAAA;AACAC,QAAAA,UAAAA,EAAY,CAACC,OAAAA,GAAAA;AACTF,YAAAA,QAAAA,CAASG,IAAI,CAACD,OAAAA,CAAAA;AAClB;AACJ,KAAA;AACJ;;;;"}

package/dist/constants.cjs

@@ -0,0 +1,34 @@
+'use strict';
+
+Object.defineProperty(exports, Symbol.toStringTag, { value: 'Module' });
+
+const DEFAULT_CHARACTER_ENCODING = "utf8";
+const LIBRARY_NAME = "riotprompt";
+const DEFAULT_PERSONA_ROLE = "developer";
+const DEFAULT_IGNORE_PATTERNS = [
+    "^\\..*",
+    "\\.(jpg|jpeg|png|gif|bmp|svg|webp|ico)$",
+    "\\.(mp3|wav|ogg|aac|flac)$",
+    "\\.(mp4|mov|avi|mkv|webm)$",
+    "\\.(pdf|doc|docx|xls|xlsx|ppt|pptx)$",
+    "\\.(zip|tar|gz|rar|7z)$" // Compressed files
+];
+const DEFAULT_SECTION_SEPARATOR = "tag";
+const DEFAULT_SECTION_INDENTATION = true;
+const DEFAULT_SECTION_TITLE_PROPERTY = "title";
+const DEFAULT_FORMAT_OPTIONS = {
+    sectionSeparator: DEFAULT_SECTION_SEPARATOR,
+    sectionIndentation: DEFAULT_SECTION_INDENTATION,
+    sectionTitleProperty: DEFAULT_SECTION_TITLE_PROPERTY,
+    sectionDepth: 0
+};
+
+exports.DEFAULT_CHARACTER_ENCODING = DEFAULT_CHARACTER_ENCODING;
+exports.DEFAULT_FORMAT_OPTIONS = DEFAULT_FORMAT_OPTIONS;
+exports.DEFAULT_IGNORE_PATTERNS = DEFAULT_IGNORE_PATTERNS;
+exports.DEFAULT_PERSONA_ROLE = DEFAULT_PERSONA_ROLE;
+exports.DEFAULT_SECTION_INDENTATION = DEFAULT_SECTION_INDENTATION;
+exports.DEFAULT_SECTION_SEPARATOR = DEFAULT_SECTION_SEPARATOR;
+exports.DEFAULT_SECTION_TITLE_PROPERTY = DEFAULT_SECTION_TITLE_PROPERTY;
+exports.LIBRARY_NAME = LIBRARY_NAME;
+//# sourceMappingURL=constants.cjs.map

package/dist/constants.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"constants.cjs","sources":["../src/constants.ts"],"sourcesContent":["import { SectionSeparator } from \"formatter\";\n\nimport { FormatOptions } from \"formatter\";\n\nexport const DEFAULT_CHARACTER_ENCODING = \"utf8\";\nexport const LIBRARY_NAME = \"riotprompt\";\n\nexport const DEFAULT_PERSONA_ROLE = \"developer\";\n\nexport const DEFAULT_INSTRUCTIONS_AREA_TITLE = \"Instructions\";\nexport const DEFAULT_CONTENTS_AREA_TITLE = \"Contents\";\nexport const DEFAULT_CONTEXT_AREA_TITLE = \"Context\";\n\nexport const DEFAULT_IGNORE_PATTERNS: string[] = [\n    \"^\\\\..*\", // Hidden files (e.g., .git, .DS_Store)\n    \"\\\\.(jpg|jpeg|png|gif|bmp|svg|webp|ico)$\", // Image files\n    \"\\\\.(mp3|wav|ogg|aac|flac)$\", // Audio files\n    \"\\\\.(mp4|mov|avi|mkv|webm)$\", // Video files\n    \"\\\\.(pdf|doc|docx|xls|xlsx|ppt|pptx)$\", // Document files\n    \"\\\\.(zip|tar|gz|rar|7z)$\" // Compressed files\n];\n\nexport const DEFAULT_SECTION_SEPARATOR: SectionSeparator = \"tag\";\nexport const DEFAULT_SECTION_INDENTATION = true;\nexport const DEFAULT_SECTION_TAG = \"section\";\nexport const DEFAULT_SECTION_TITLE_PROPERTY = \"title\";\n\nexport const DEFAULT_FORMAT_OPTIONS: FormatOptions = {\n    sectionSeparator: DEFAULT_SECTION_SEPARATOR,\n    sectionIndentation: DEFAULT_SECTION_INDENTATION,\n    sectionTitleProperty: DEFAULT_SECTION_TITLE_PROPERTY,\n    sectionDepth: 0,\n}\n"],"names":["DEFAULT_CHARACTER_ENCODING","LIBRARY_NAME","DEFAULT_PERSONA_ROLE","DEFAULT_IGNORE_PATTERNS","DEFAULT_SECTION_SEPARATOR","DEFAULT_SECTION_INDENTATION","DEFAULT_SECTION_TITLE_PROPERTY","DEFAULT_FORMAT_OPTIONS","sectionSeparator","sectionIndentation","sectionTitleProperty","sectionDepth"],"mappings":";;;;AAIO,MAAMA,6BAA6B;AACnC,MAAMC,eAAe;AAErB,MAAMC,uBAAuB;MAMvBC,uBAAoC,GAAA;AAC7C,IAAA,QAAA;AACA,IAAA,yCAAA;AACA,IAAA,4BAAA;AACA,IAAA,4BAAA;AACA,IAAA,sCAAA;AACA,IAAA,yBAAA;;AAGG,MAAMC,4BAA8C;AACpD,MAAMC,8BAA8B;AAEpC,MAAMC,iCAAiC;MAEjCC,sBAAwC,GAAA;IACjDC,gBAAkBJ,EAAAA,yBAAAA;IAClBK,kBAAoBJ,EAAAA,2BAAAA;IACpBK,oBAAsBJ,EAAAA,8BAAAA;IACtBK,YAAc,EAAA;AAClB;;;;;;;;;;;"}

package/dist/constants.d.ts

@@ -0,0 +1,13 @@
+import { SectionSeparator, FormatOptions } from './formatter';
+export declare const DEFAULT_CHARACTER_ENCODING = "utf8";
+export declare const LIBRARY_NAME = "riotprompt";
+export declare const DEFAULT_PERSONA_ROLE = "developer";
+export declare const DEFAULT_INSTRUCTIONS_AREA_TITLE = "Instructions";
+export declare const DEFAULT_CONTENTS_AREA_TITLE = "Contents";
+export declare const DEFAULT_CONTEXT_AREA_TITLE = "Context";
+export declare const DEFAULT_IGNORE_PATTERNS: string[];
+export declare const DEFAULT_SECTION_SEPARATOR: SectionSeparator;
+export declare const DEFAULT_SECTION_INDENTATION = true;
+export declare const DEFAULT_SECTION_TAG = "section";
+export declare const DEFAULT_SECTION_TITLE_PROPERTY = "title";
+export declare const DEFAULT_FORMAT_OPTIONS: FormatOptions;

package/dist/constants.js

@@ -0,0 +1,23 @@
+const DEFAULT_CHARACTER_ENCODING = "utf8";
+const LIBRARY_NAME = "riotprompt";
+const DEFAULT_PERSONA_ROLE = "developer";
+const DEFAULT_IGNORE_PATTERNS = [
+    "^\\..*",
+    "\\.(jpg|jpeg|png|gif|bmp|svg|webp|ico)$",
+    "\\.(mp3|wav|ogg|aac|flac)$",
+    "\\.(mp4|mov|avi|mkv|webm)$",
+    "\\.(pdf|doc|docx|xls|xlsx|ppt|pptx)$",
+    "\\.(zip|tar|gz|rar|7z)$" // Compressed files
+];
+const DEFAULT_SECTION_SEPARATOR = "tag";
+const DEFAULT_SECTION_INDENTATION = true;
+const DEFAULT_SECTION_TITLE_PROPERTY = "title";
+const DEFAULT_FORMAT_OPTIONS = {
+    sectionSeparator: DEFAULT_SECTION_SEPARATOR,
+    sectionIndentation: DEFAULT_SECTION_INDENTATION,
+    sectionTitleProperty: DEFAULT_SECTION_TITLE_PROPERTY,
+    sectionDepth: 0
+};
+
+export { DEFAULT_CHARACTER_ENCODING, DEFAULT_FORMAT_OPTIONS, DEFAULT_IGNORE_PATTERNS, DEFAULT_PERSONA_ROLE, DEFAULT_SECTION_INDENTATION, DEFAULT_SECTION_SEPARATOR, DEFAULT_SECTION_TITLE_PROPERTY, LIBRARY_NAME };
+//# sourceMappingURL=constants.js.map

package/dist/constants.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"constants.js","sources":["../src/constants.ts"],"sourcesContent":["import { SectionSeparator } from \"formatter\";\n\nimport { FormatOptions } from \"formatter\";\n\nexport const DEFAULT_CHARACTER_ENCODING = \"utf8\";\nexport const LIBRARY_NAME = \"riotprompt\";\n\nexport const DEFAULT_PERSONA_ROLE = \"developer\";\n\nexport const DEFAULT_INSTRUCTIONS_AREA_TITLE = \"Instructions\";\nexport const DEFAULT_CONTENTS_AREA_TITLE = \"Contents\";\nexport const DEFAULT_CONTEXT_AREA_TITLE = \"Context\";\n\nexport const DEFAULT_IGNORE_PATTERNS: string[] = [\n    \"^\\\\..*\", // Hidden files (e.g., .git, .DS_Store)\n    \"\\\\.(jpg|jpeg|png|gif|bmp|svg|webp|ico)$\", // Image files\n    \"\\\\.(mp3|wav|ogg|aac|flac)$\", // Audio files\n    \"\\\\.(mp4|mov|avi|mkv|webm)$\", // Video files\n    \"\\\\.(pdf|doc|docx|xls|xlsx|ppt|pptx)$\", // Document files\n    \"\\\\.(zip|tar|gz|rar|7z)$\" // Compressed files\n];\n\nexport const DEFAULT_SECTION_SEPARATOR: SectionSeparator = \"tag\";\nexport const DEFAULT_SECTION_INDENTATION = true;\nexport const DEFAULT_SECTION_TAG = \"section\";\nexport const DEFAULT_SECTION_TITLE_PROPERTY = \"title\";\n\nexport const DEFAULT_FORMAT_OPTIONS: FormatOptions = {\n    sectionSeparator: DEFAULT_SECTION_SEPARATOR,\n    sectionIndentation: DEFAULT_SECTION_INDENTATION,\n    sectionTitleProperty: DEFAULT_SECTION_TITLE_PROPERTY,\n    sectionDepth: 0,\n}\n"],"names":["DEFAULT_CHARACTER_ENCODING","LIBRARY_NAME","DEFAULT_PERSONA_ROLE","DEFAULT_IGNORE_PATTERNS","DEFAULT_SECTION_SEPARATOR","DEFAULT_SECTION_INDENTATION","DEFAULT_SECTION_TITLE_PROPERTY","DEFAULT_FORMAT_OPTIONS","sectionSeparator","sectionIndentation","sectionTitleProperty","sectionDepth"],"mappings":"AAIO,MAAMA,6BAA6B;AACnC,MAAMC,eAAe;AAErB,MAAMC,uBAAuB;MAMvBC,uBAAoC,GAAA;AAC7C,IAAA,QAAA;AACA,IAAA,yCAAA;AACA,IAAA,4BAAA;AACA,IAAA,4BAAA;AACA,IAAA,sCAAA;AACA,IAAA,yBAAA;;AAGG,MAAMC,4BAA8C;AACpD,MAAMC,8BAA8B;AAEpC,MAAMC,iCAAiC;MAEjCC,sBAAwC,GAAA;IACjDC,gBAAkBJ,EAAAA,yBAAAA;IAClBK,kBAAoBJ,EAAAA,2BAAAA;IACpBK,oBAAsBJ,EAAAA,8BAAAA;IACtBK,YAAc,EAAA;AAClB;;;;"}

package/dist/formatter.cjs

@@ -0,0 +1,139 @@
+'use strict';
+
+Object.defineProperty(exports, Symbol.toStringTag, { value: 'Module' });
+
+const zod = require('zod');
+const chat = require('./chat.cjs');
+const constants = require('./constants.cjs');
+const logger = require('./logger.cjs');
+const general = require('./util/general.cjs');
+
+const SectionSeparatorSchema = zod.z.enum([
+    "tag",
+    "markdown"
+]);
+const SectionTitlePropertySchema = zod.z.enum([
+    "title",
+    "name"
+]);
+const FormatOptionsSchema = zod.z.object({
+    sectionSeparator: SectionSeparatorSchema,
+    sectionIndentation: zod.z.boolean(),
+    sectionTitleProperty: SectionTitlePropertySchema,
+    sectionTitlePrefix: zod.z.string().optional(),
+    sectionTitleSeparator: zod.z.string().optional(),
+    sectionDepth: zod.z.number().default(1)
+});
+const OptionSchema = zod.z.object({
+    logger: zod.z.any().optional().default(logger.DEFAULT_LOGGER),
+    formatOptions: FormatOptionsSchema.partial().optional().default(constants.DEFAULT_FORMAT_OPTIONS)
+});
+// Type guard to check if an object is a Section
+function isSection(obj) {
+    return obj && typeof obj === 'object' && 'items' in obj && Array.isArray(obj.items);
+}
+// Type guard to check if an object is a Section
+function isWeighted(obj) {
+    return obj && typeof obj === 'object' && 'text' in obj;
+}
+const create = (formatterOptions)=>{
+    const options = OptionSchema.parse(formatterOptions || {});
+    const logger$1 = logger.wrapLogger(options.logger, 'Formatter');
+    let formatOptions = constants.DEFAULT_FORMAT_OPTIONS;
+    if (options === null || options === void 0 ? void 0 : options.formatOptions) {
+        formatOptions = {
+            ...formatOptions,
+            ...general.clean(options.formatOptions)
+        };
+    }
+    const formatPersona = (model, persona)=>{
+        logger$1.silly(`Formatting persona`);
+        if (persona) {
+            const formattedPersona = formatSection(persona);
+            return {
+                role: chat.getPersonaRole(model),
+                content: `${formattedPersona}`
+            };
+        } else {
+            throw new Error("Persona is required");
+        }
+    };
+    const format = (item, sectionDepth)=>{
+        logger$1.silly(`Formatting ${isSection(item) ? "section" : "item"} Item: %s`, general.stringifyJSON(item));
+        const currentSectionDepth = sectionDepth !== null && sectionDepth !== void 0 ? sectionDepth : formatOptions.sectionDepth;
+        logger$1.silly(`\t\tCurrent section depth: ${currentSectionDepth}`);
+        let result = "";
+        if (isSection(item)) {
+            result = formatSection(item, currentSectionDepth + 1);
+        } else if (isWeighted(item)) {
+            result = item.text;
+        } else {
+            //If the item is neither a section nor a weighted item, it is empty.
+            result = '';
+        }
+        return result;
+    };
+    const formatSection = (section, sectionDepth)=>{
+        logger$1.silly(`Formatting section`);
+        const currentSectionDepth = sectionDepth !== null && sectionDepth !== void 0 ? sectionDepth : formatOptions.sectionDepth;
+        logger$1.silly(`\t\tCurrent section depth: ${currentSectionDepth}`);
+        if (section) {
+            const formattedItems = section.items.map((item)=>format(item, currentSectionDepth)).join("\n\n");
+            if (formatOptions.sectionSeparator === "tag") {
+                var _section_title, _section_title1;
+                return `<${(_section_title = section.title) !== null && _section_title !== void 0 ? _section_title : "section"}>\n${formattedItems}\n</${(_section_title1 = section.title) !== null && _section_title1 !== void 0 ? _section_title1 : "section"}>`;
+            } else {
+                // Default depth to 1 if not provided, resulting in H2 (##) matching the test case.
+                const headingLevel = currentSectionDepth;
+                const hashes = '#'.repeat(headingLevel);
+                logger$1.silly(`\t\tHeading level: ${headingLevel}`);
+                logger$1.silly(`\t\tSection title: ${section.title}`);
+                return `${hashes} ${formatOptions.sectionTitlePrefix ? `${formatOptions.sectionTitlePrefix} ${formatOptions.sectionTitleSeparator} ` : ""}${section.title}\n\n${formattedItems}`;
+            }
+        } else {
+            return '';
+        }
+    };
+    // Helper function to format arrays of items or sections
+    const formatArray = (items, sectionDepth)=>{
+        logger$1.silly(`Formatting array`);
+        const currentSectionDepth = sectionDepth !== null && sectionDepth !== void 0 ? sectionDepth : formatOptions.sectionDepth;
+        return items.map((item)=>format(item, currentSectionDepth)).join("\n\n");
+    };
+    const formatPrompt = (model, prompt)=>{
+        logger$1.silly('Formatting prompt');
+        const chatRequest = chat.createRequest(model);
+        if (prompt.persona) {
+            [
+                prompt.persona
+            ].forEach((persona)=>{
+                chatRequest.addMessage(formatPersona(model, persona));
+            });
+        }
+        let formattedAreas = formatSection(prompt.instructions) + '\n\n';
+        if (prompt.contents) {
+            formattedAreas += formatSection(prompt.contents) + '\n\n';
+        }
+        if (prompt.contexts) {
+            formattedAreas += formatSection(prompt.contexts) + '\n\n';
+        }
+        chatRequest.addMessage({
+            role: "user",
+            content: formattedAreas
+        });
+        return chatRequest;
+    };
+    return {
+        formatPersona,
+        format,
+        formatPrompt,
+        formatArray
+    };
+};
+
+exports.FormatOptionsSchema = FormatOptionsSchema;
+exports.OptionSchema = OptionSchema;
+exports.SectionSeparatorSchema = SectionSeparatorSchema;
+exports.SectionTitlePropertySchema = SectionTitlePropertySchema;
+exports.create = create;
+//# sourceMappingURL=formatter.cjs.map

package/dist/formatter.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"formatter.cjs","sources":["../src/formatter.ts"],"sourcesContent":["import { Instruction } from \"riotprompt\";\nimport { z } from \"zod\";\nimport * as Chat from \"./chat\";\nimport { getPersonaRole, Message, Model } from \"./chat\";\nimport { DEFAULT_FORMAT_OPTIONS } from \"./constants\";\nimport { Section } from \"./items/section\";\nimport { Weighted } from \"./items/weighted\";\nimport { DEFAULT_LOGGER, wrapLogger } from \"./logger\";\nimport { Prompt } from \"./prompt\";\nimport { clean, stringifyJSON } from \"./util/general\";\n\nexport const SectionSeparatorSchema = z.enum([\"tag\", \"markdown\"]);\nexport const SectionTitlePropertySchema = z.enum([\"title\", \"name\"]);\n\nexport type SectionSeparator = z.infer<typeof SectionSeparatorSchema>;\nexport type SectionTitleProperty = z.infer<typeof SectionTitlePropertySchema>;\n\n\nexport const FormatOptionsSchema = z.object({\n    sectionSeparator: SectionSeparatorSchema,\n    sectionIndentation: z.boolean(),\n    sectionTitleProperty: SectionTitlePropertySchema,\n    sectionTitlePrefix: z.string().optional(),\n    sectionTitleSeparator: z.string().optional(),\n    sectionDepth: z.number().default(1),\n});\n\nexport type FormatOptions = z.infer<typeof FormatOptionsSchema>;\n\n\nexport const OptionSchema = z.object({\n    logger: z.any().optional().default(DEFAULT_LOGGER),\n    formatOptions: FormatOptionsSchema.partial().optional().default(DEFAULT_FORMAT_OPTIONS),\n});\n\nexport type Options = z.infer<typeof OptionSchema>;\n\nexport type OptionsParam = Partial<Options>;\n\nexport interface Instance {\n    formatPersona: (model: Model, persona: Section<Instruction>) => Message;\n    format: <T extends Weighted>(weightedText: T | Section<T>, sectionDepth?: number) => string;\n    formatArray: <T extends Weighted>(items: (T | Section<T>)[], sectionDepth?: number) => string;\n    formatPrompt: (model: Model, prompt: Prompt) => Chat.Request;\n}\n\n// Type guard to check if an object is a Section\nfunction isSection<T extends Weighted>(obj: T | Section<T>): obj is Section<T> {\n    return obj && typeof obj === 'object' && 'items' in obj && Array.isArray((obj as Section<T>).items);\n}\n\n// Type guard to check if an object is a Section\nfunction isWeighted<T extends Weighted>(obj: T | Section<T>): obj is T {\n    return obj && typeof obj === 'object' && 'text' in obj;\n}\n\n\nexport const create = (formatterOptions?: OptionsParam): Instance => {\n    const options: Required<Options> = OptionSchema.parse(formatterOptions || {}) as Required<Options>;\n\n    const logger = wrapLogger(options.logger, 'Formatter');\n\n    let formatOptions: FormatOptions = DEFAULT_FORMAT_OPTIONS;\n    if (options?.formatOptions) {\n        formatOptions = {\n            ...formatOptions,\n            ...clean(options.formatOptions),\n        };\n    }\n\n    const formatPersona = (model: Model, persona: Section<Instruction>): Message => {\n        logger.silly(`Formatting persona`);\n        if (persona) {\n            const formattedPersona = formatSection(persona);\n\n            return {\n                role: getPersonaRole(model),\n                content: `${formattedPersona}`,\n            }\n        } else {\n            throw new Error(\"Persona is required\");\n        }\n    }\n\n    const format = <T extends Weighted>(\n        item: T | Section<T>,\n        sectionDepth?: number,\n    ): string => {\n        logger.silly(`Formatting ${isSection(item) ? \"section\" : \"item\"} Item: %s`, stringifyJSON(item));\n        const currentSectionDepth = sectionDepth ?? formatOptions.sectionDepth;\n        logger.silly(`\\t\\tCurrent section depth: ${currentSectionDepth}`);\n\n        let result: string = \"\";\n        if (isSection(item)) {\n            result = formatSection(item, currentSectionDepth + 1);\n        } else if (isWeighted(item)) {\n            result = item.text;\n        } else {\n            //If the item is neither a section nor a weighted item, it is empty.\n            result = '';\n        }\n        return result;\n    }\n\n    const formatSection = <T extends Weighted>(section: Section<T>, sectionDepth?: number): string => {\n        logger.silly(`Formatting section`);\n        const currentSectionDepth = sectionDepth ?? formatOptions.sectionDepth;\n        logger.silly(`\\t\\tCurrent section depth: ${currentSectionDepth}`);\n\n        if (section) {\n            const formattedItems = section.items.map(item => format(item, currentSectionDepth)).join(\"\\n\\n\");\n\n            if (formatOptions.sectionSeparator === \"tag\") {\n                return `<${section.title ?? \"section\"}>\\n${formattedItems}\\n</${section.title ?? \"section\"}>`;\n            } else {\n                // Default depth to 1 if not provided, resulting in H2 (##) matching the test case.\n                const headingLevel = currentSectionDepth;\n                const hashes = '#'.repeat(headingLevel);\n                logger.silly(`\\t\\tHeading level: ${headingLevel}`);\n                logger.silly(`\\t\\tSection title: ${section.title}`);\n                return `${hashes} ${formatOptions.sectionTitlePrefix ? `${formatOptions.sectionTitlePrefix} ${formatOptions.sectionTitleSeparator} ` : \"\"}${section.title}\\n\\n${formattedItems}`;\n            }\n        } else {\n            return '';\n        }\n    }\n\n    // Helper function to format arrays of items or sections\n    const formatArray = <T extends Weighted>(\n        items: (T | Section<T>)[],\n        sectionDepth?: number\n    ): string => {\n        logger.silly(`Formatting array`);\n        const currentSectionDepth = sectionDepth ?? formatOptions.sectionDepth;\n        return items.map(item => format(item, currentSectionDepth)).join(\"\\n\\n\");\n    }\n\n    const formatPrompt = (model: Model, prompt: Prompt): Chat.Request => {\n        logger.silly('Formatting prompt');\n        const chatRequest: Chat.Request = Chat.createRequest(model);\n\n        if (prompt.persona) {\n            [prompt.persona].forEach((persona: Section<Instruction>) => {\n                chatRequest.addMessage(formatPersona(model, persona));\n            });\n        }\n\n        let formattedAreas: string = formatSection(prompt.instructions) + '\\n\\n';\n\n        if (prompt.contents) {\n            formattedAreas += formatSection(prompt.contents) + '\\n\\n';\n        }\n\n        if (prompt.contexts) {\n            formattedAreas += formatSection(prompt.contexts) + '\\n\\n';\n        }\n\n        chatRequest.addMessage({\n            role: \"user\",\n            content: formattedAreas,\n        });\n\n        return chatRequest;\n    }\n\n    return {\n        formatPersona,\n        format,\n        formatPrompt,\n        formatArray,\n    }\n}\n"],"names":["SectionSeparatorSchema","z","enum","SectionTitlePropertySchema","FormatOptionsSchema","object","sectionSeparator","sectionIndentation","boolean","sectionTitleProperty","sectionTitlePrefix","string","optional","sectionTitleSeparator","sectionDepth","number","default","OptionSchema","logger","any","DEFAULT_LOGGER","formatOptions","partial","DEFAULT_FORMAT_OPTIONS","isSection","obj","Array","isArray","items","isWeighted","create","formatterOptions","options","parse","wrapLogger","clean","formatPersona","model","persona","silly","formattedPersona","formatSection","role","getPersonaRole","content","Error","format","item","stringifyJSON","currentSectionDepth","result","text","section","formattedItems","map","join","title","headingLevel","hashes","repeat","formatArray","formatPrompt","prompt","chatRequest","Chat","forEach","addMessage","formattedAreas","instructions","contents","contexts"],"mappings":";;;;;;;;;;AAWaA,MAAAA,sBAAAA,GAAyBC,KAAEC,CAAAA,IAAI,CAAC;AAAC,IAAA,KAAA;AAAO,IAAA;CAAW;AACnDC,MAAAA,0BAAAA,GAA6BF,KAAEC,CAAAA,IAAI,CAAC;AAAC,IAAA,OAAA;AAAS,IAAA;CAAO;AAMrDE,MAAAA,mBAAAA,GAAsBH,KAAEI,CAAAA,MAAM,CAAC;IACxCC,gBAAkBN,EAAAA,sBAAAA;AAClBO,IAAAA,kBAAAA,EAAoBN,MAAEO,OAAO,EAAA;IAC7BC,oBAAsBN,EAAAA,0BAAAA;IACtBO,kBAAoBT,EAAAA,KAAAA,CAAEU,MAAM,EAAA,CAAGC,QAAQ,EAAA;IACvCC,qBAAuBZ,EAAAA,KAAAA,CAAEU,MAAM,EAAA,CAAGC,QAAQ,EAAA;AAC1CE,IAAAA,YAAAA,EAAcb,KAAEc,CAAAA,MAAM,EAAGC,CAAAA,OAAO,CAAC,CAAA;AACrC,CAAG;AAKUC,MAAAA,YAAAA,GAAehB,KAAEI,CAAAA,MAAM,CAAC;AACjCa,IAAAA,MAAAA,EAAQjB,MAAEkB,GAAG,EAAA,CAAGP,QAAQ,EAAA,CAAGI,OAAO,CAACI,qBAAAA,CAAAA;AACnCC,IAAAA,aAAAA,EAAejB,oBAAoBkB,OAAO,EAAA,CAAGV,QAAQ,EAAA,CAAGI,OAAO,CAACO,gCAAAA;AACpE,CAAG;AAaH;AACA,SAASC,UAA8BC,GAAmB,EAAA;IACtD,OAAOA,GAAAA,IAAO,OAAOA,GAAAA,KAAQ,QAAY,IAAA,OAAA,IAAWA,GAAOC,IAAAA,KAAAA,CAAMC,OAAO,CAAC,GAACF,CAAmBG,KAAK,CAAA;AACtG;AAEA;AACA,SAASC,WAA+BJ,GAAmB,EAAA;AACvD,IAAA,OAAOA,GAAO,IAAA,OAAOA,GAAQ,KAAA,QAAA,IAAY,MAAUA,IAAAA,GAAAA;AACvD;AAGO,MAAMK,SAAS,CAACC,gBAAAA,GAAAA;AACnB,IAAA,MAAMC,OAA6Bf,GAAAA,YAAAA,CAAagB,KAAK,CAACF,oBAAoB,EAAC,CAAA;AAE3E,IAAA,MAAMb,QAASgB,GAAAA,iBAAAA,CAAWF,OAAQd,CAAAA,MAAM,EAAE,WAAA,CAAA;AAE1C,IAAA,IAAIG,aAA+BE,GAAAA,gCAAAA;AACnC,IAAA,IAAIS,OAAAA,KAAAA,IAAAA,IAAAA,OAAAA,KAAAA,MAAAA,GAAAA,MAAAA,GAAAA,OAASX,CAAAA,aAAa,EAAE;QACxBA,aAAgB,GAAA;AACZ,YAAA,GAAGA,aAAa;YAChB,GAAGc,aAAAA,CAAMH,OAAQX,CAAAA,aAAa;AAClC,SAAA;AACJ;IAEA,MAAMe,aAAAA,GAAgB,CAACC,KAAcC,EAAAA,OAAAA,GAAAA;AACjCpB,QAAAA,QAAAA,CAAOqB,KAAK,CAAC,CAAC,kBAAkB,CAAC,CAAA;AACjC,QAAA,IAAID,OAAS,EAAA;AACT,YAAA,MAAME,mBAAmBC,aAAcH,CAAAA,OAAAA,CAAAA;YAEvC,OAAO;AACHI,gBAAAA,IAAAA,EAAMC,mBAAeN,CAAAA,KAAAA,CAAAA;AACrBO,gBAAAA,OAAAA,EAAS,GAAGJ,gBAAkB,CAAA;AAClC,aAAA;SACG,MAAA;AACH,YAAA,MAAM,IAAIK,KAAM,CAAA,qBAAA,CAAA;AACpB;AACJ,KAAA;IAEA,MAAMC,MAAAA,GAAS,CACXC,IACAjC,EAAAA,YAAAA,GAAAA;AAEAI,QAAAA,QAAAA,CAAOqB,KAAK,CAAC,CAAC,WAAW,EAAEf,SAAAA,CAAUuB,IAAQ,CAAA,GAAA,SAAA,GAAY,MAAO,CAAA,SAAS,CAAC,EAAEC,qBAAcD,CAAAA,IAAAA,CAAAA,CAAAA;AAC1F,QAAA,MAAME,mBAAsBnC,GAAAA,YAAAA,KAAAA,IAAAA,IAAAA,YAAAA,KAAAA,MAAAA,GAAAA,YAAAA,GAAgBO,cAAcP,YAAY;AACtEI,QAAAA,QAAAA,CAAOqB,KAAK,CAAC,CAAC,2BAA2B,EAAEU,mBAAqB,CAAA,CAAA,CAAA;AAEhE,QAAA,IAAIC,MAAiB,GAAA,EAAA;AACrB,QAAA,IAAI1B,UAAUuB,IAAO,CAAA,EAAA;YACjBG,MAAST,GAAAA,aAAAA,CAAcM,MAAME,mBAAsB,GAAA,CAAA,CAAA;SAChD,MAAA,IAAIpB,WAAWkB,IAAO,CAAA,EAAA;AACzBG,YAAAA,MAAAA,GAASH,KAAKI,IAAI;SACf,MAAA;;YAEHD,MAAS,GAAA,EAAA;AACb;QACA,OAAOA,MAAAA;AACX,KAAA;IAEA,MAAMT,aAAAA,GAAgB,CAAqBW,OAAqBtC,EAAAA,YAAAA,GAAAA;AAC5DI,QAAAA,QAAAA,CAAOqB,KAAK,CAAC,CAAC,kBAAkB,CAAC,CAAA;AACjC,QAAA,MAAMU,mBAAsBnC,GAAAA,YAAAA,KAAAA,IAAAA,IAAAA,YAAAA,KAAAA,MAAAA,GAAAA,YAAAA,GAAgBO,cAAcP,YAAY;AACtEI,QAAAA,QAAAA,CAAOqB,KAAK,CAAC,CAAC,2BAA2B,EAAEU,mBAAqB,CAAA,CAAA,CAAA;AAEhE,QAAA,IAAIG,OAAS,EAAA;AACT,YAAA,MAAMC,cAAiBD,GAAAA,OAAAA,CAAQxB,KAAK,CAAC0B,GAAG,CAACP,CAAAA,IAAAA,GAAQD,MAAOC,CAAAA,IAAAA,EAAME,mBAAsBM,CAAAA,CAAAA,CAAAA,IAAI,CAAC,MAAA,CAAA;YAEzF,IAAIlC,aAAAA,CAAcf,gBAAgB,KAAK,KAAO,EAAA;oBAC/B8C,cAAqDA,EAAAA,eAAAA;gBAAhE,OAAO,CAAC,CAAC,EAAEA,CAAAA,cAAAA,GAAAA,QAAQI,KAAK,MAAA,IAAA,IAAbJ,cAAAA,KAAAA,MAAAA,GAAAA,cAAAA,GAAiB,SAAU,CAAA,GAAG,EAAEC,cAAe,CAAA,IAAI,EAAED,CAAAA,eAAAA,GAAAA,OAAAA,CAAQI,KAAK,MAAA,IAAA,IAAbJ,eAAAA,KAAAA,MAAAA,GAAAA,eAAAA,GAAiB,SAAU,CAAA,CAAC,CAAC;aAC1F,MAAA;;AAEH,gBAAA,MAAMK,YAAeR,GAAAA,mBAAAA;gBACrB,MAAMS,MAAAA,GAAS,GAAIC,CAAAA,MAAM,CAACF,YAAAA,CAAAA;AAC1BvC,gBAAAA,QAAAA,CAAOqB,KAAK,CAAC,CAAC,mBAAmB,EAAEkB,YAAc,CAAA,CAAA,CAAA;AACjDvC,gBAAAA,QAAAA,CAAOqB,KAAK,CAAC,CAAC,mBAAmB,EAAEa,OAAAA,CAAQI,KAAK,CAAE,CAAA,CAAA;gBAClD,OAAO,CAAA,EAAGE,MAAO,CAAA,CAAC,EAAErC,aAAAA,CAAcX,kBAAkB,GAAG,CAAA,EAAGW,aAAcX,CAAAA,kBAAkB,CAAC,CAAC,EAAEW,aAAcR,CAAAA,qBAAqB,CAAC,CAAC,CAAC,GAAG,EAAKuC,CAAAA,EAAAA,OAAAA,CAAQI,KAAK,CAAC,IAAI,EAAEH,cAAgB,CAAA,CAAA;AACpL;SACG,MAAA;YACH,OAAO,EAAA;AACX;AACJ,KAAA;;IAGA,MAAMO,WAAAA,GAAc,CAChBhC,KACAd,EAAAA,YAAAA,GAAAA;AAEAI,QAAAA,QAAAA,CAAOqB,KAAK,CAAC,CAAC,gBAAgB,CAAC,CAAA;AAC/B,QAAA,MAAMU,mBAAsBnC,GAAAA,YAAAA,KAAAA,IAAAA,IAAAA,YAAAA,KAAAA,MAAAA,GAAAA,YAAAA,GAAgBO,cAAcP,YAAY;QACtE,OAAOc,KAAAA,CAAM0B,GAAG,CAACP,CAAAA,OAAQD,MAAOC,CAAAA,IAAAA,EAAME,mBAAsBM,CAAAA,CAAAA,CAAAA,IAAI,CAAC,MAAA,CAAA;AACrE,KAAA;IAEA,MAAMM,YAAAA,GAAe,CAACxB,KAAcyB,EAAAA,MAAAA,GAAAA;AAChC5C,QAAAA,QAAAA,CAAOqB,KAAK,CAAC,mBAAA,CAAA;QACb,MAAMwB,WAAAA,GAA4BC,kBAAkB,CAAC3B,KAAAA,CAAAA;QAErD,IAAIyB,MAAAA,CAAOxB,OAAO,EAAE;AAChB,YAAA;AAACwB,gBAAAA,MAAAA,CAAOxB;aAAQ,CAAC2B,OAAO,CAAC,CAAC3B,OAAAA,GAAAA;gBACtByB,WAAYG,CAAAA,UAAU,CAAC9B,aAAAA,CAAcC,KAAOC,EAAAA,OAAAA,CAAAA,CAAAA;AAChD,aAAA,CAAA;AACJ;AAEA,QAAA,IAAI6B,cAAyB1B,GAAAA,aAAAA,CAAcqB,MAAOM,CAAAA,YAAY,CAAI,GAAA,MAAA;QAElE,IAAIN,MAAAA,CAAOO,QAAQ,EAAE;YACjBF,cAAkB1B,IAAAA,aAAAA,CAAcqB,MAAOO,CAAAA,QAAQ,CAAI,GAAA,MAAA;AACvD;QAEA,IAAIP,MAAAA,CAAOQ,QAAQ,EAAE;YACjBH,cAAkB1B,IAAAA,aAAAA,CAAcqB,MAAOQ,CAAAA,QAAQ,CAAI,GAAA,MAAA;AACvD;AAEAP,QAAAA,WAAAA,CAAYG,UAAU,CAAC;YACnBxB,IAAM,EAAA,MAAA;YACNE,OAASuB,EAAAA;AACb,SAAA,CAAA;QAEA,OAAOJ,WAAAA;AACX,KAAA;IAEA,OAAO;AACH3B,QAAAA,aAAAA;AACAU,QAAAA,MAAAA;AACAe,QAAAA,YAAAA;AACAD,QAAAA;AACJ,KAAA;AACJ;;;;;;;;"}