@webiny/ai-powerups 0.0.0-unstable.0156a4c312

package/api/features/AiPromptContext/AiPromptContextBuilder.js

@@ -0,0 +1,147 @@
+import { GetSettingsUseCase } from "../GetSettings/index.js";
+import { GetFileContentsByIdUseCase } from "@webiny/api-file-manager/features/file/GetFileContentsById/index.js";
+import { GetFileUseCase } from "@webiny/api-file-manager/features/file/GetFile/index.js";
+import { TextExtractor } from "@webiny/api-core/features/ai/index.js";
+import { AiPromptContextBuilder, ProjectFileAssembler } from "./abstractions.js";
+import { WriterPersonaSection } from "./WriterPersonaSection.js";
+import { ReaderPersonaSection } from "./ReaderPersonaSection.js";
+import { ProjectSection } from "./ProjectSection.js";
+function resolvePersonas(settings, project, params) {
+    const readerPersonaId = params.readerPersonaId ?? project?.defaultReaderPersonaId ?? void 0;
+    const writerPersonaId = params.writerPersonaId ?? project?.defaultWriterPersonaId ?? void 0;
+    const readerPreset = readerPersonaId ? settings.readerPersonas?.presets?.find((p)=>p.id === readerPersonaId) : void 0;
+    const writerPreset = writerPersonaId ? settings.writerPersonas?.presets?.find((p)=>p.id === writerPersonaId) : void 0;
+    return {
+        reader: readerPreset ? {
+            name: readerPreset.name,
+            description: readerPreset.description,
+            style: readerPreset.style
+        } : void 0,
+        writer: writerPreset ? {
+            name: writerPreset.name,
+            description: writerPreset.description,
+            style: writerPreset.style
+        } : void 0
+    };
+}
+function formatContext(ctx) {
+    const sections = [];
+    if (ctx.writerPersona) sections.push(WriterPersonaSection.format(ctx.writerPersona));
+    if (ctx.readerPersona) sections.push(ReaderPersonaSection.format(ctx.readerPersona));
+    if (ctx.project) {
+        const projectText = ProjectSection.format(ctx.project);
+        if (projectText) sections.push(projectText);
+    }
+    return sections.length > 0 ? "\n\n" + sections.join("\n\n") : "";
+}
+function emptyContext(warnings) {
+    return {
+        project: void 0,
+        readerPersona: void 0,
+        writerPersona: void 0,
+        allProjectFiles: [],
+        additionalFiles: [],
+        excludedFileIds: new Set(),
+        cacheHit: false,
+        warnings,
+        toString () {
+            return "";
+        }
+    };
+}
+class AiPromptContextBuilderImpl {
+    constructor(getSettings, fileAssembler, getFileContents, getFile, textExtractor){
+        this.getSettings = getSettings;
+        this.fileAssembler = fileAssembler;
+        this.getFileContents = getFileContents;
+        this.getFile = getFile;
+        this.textExtractor = textExtractor;
+    }
+    async execute(params) {
+        const warnings = [];
+        const settingsResult = await this.getSettings.execute();
+        if (settingsResult.isFail()) {
+            warnings.push("Failed to load AI PowerUps settings.");
+            return emptyContext(warnings);
+        }
+        const settings = settingsResult.value;
+        const projectPreset = params.projectId ? settings.projects?.presets?.find((p)=>p.id === params.projectId) : void 0;
+        const { reader, writer } = resolvePersonas(settings, projectPreset, params);
+        const excludedFileIds = new Set(params.excludedFileIds ?? []);
+        let project;
+        let allProjectFiles = [];
+        let cacheHit = false;
+        if (projectPreset) {
+            const assembled = await this.fileAssembler.execute(projectPreset.id, projectPreset.version ?? 0, projectPreset.files ?? [], warnings);
+            allProjectFiles = assembled.files;
+            cacheHit = assembled.cacheHit;
+            const visibleFiles = excludedFileIds.size > 0 ? allProjectFiles.filter((f)=>!excludedFileIds.has(f.id)) : allProjectFiles;
+            project = {
+                name: projectPreset.name,
+                instructions: projectPreset.instructions,
+                files: visibleFiles,
+                totalTokens: visibleFiles.reduce((sum, f)=>sum + f.tokenCount, 0)
+            };
+        }
+        const additionalFiles = await this.fetchAdditionalFiles(params.additionalFileIds, warnings);
+        if (additionalFiles.length > 0) allProjectFiles = [
+            ...allProjectFiles,
+            ...additionalFiles
+        ];
+        const ctx = {
+            project,
+            readerPersona: reader,
+            writerPersona: writer,
+            allProjectFiles,
+            additionalFiles,
+            excludedFileIds,
+            cacheHit,
+            warnings,
+            toString () {
+                return formatContext(this);
+            }
+        };
+        return ctx;
+    }
+    async fetchAdditionalFiles(fileIds, warnings) {
+        if (!fileIds || 0 === fileIds.length) return [];
+        const results = await Promise.all(fileIds.map(async (fileId)=>{
+            const [contentsResult, damResult] = await Promise.all([
+                this.getFileContents.execute(fileId),
+                this.getFile.execute(fileId)
+            ]);
+            if (contentsResult.isFail()) {
+                warnings.push(`Failed to load additional file (${fileId}): ${contentsResult.error.message}`);
+                return null;
+            }
+            const name = damResult.isOk() ? damResult.value.name : fileId;
+            const mimeType = damResult.isOk() ? damResult.value.type : "application/octet-stream";
+            if (!this.textExtractor.canExtract(mimeType)) {
+                warnings.push(`Skipped additional file "${name}" (${fileId}): unsupported type "${mimeType}".`);
+                return null;
+            }
+            const { text: content } = await this.textExtractor.extract(contentsResult.value.buffer, mimeType);
+            return {
+                id: fileId,
+                name,
+                content,
+                description: damResult.isOk() ? damResult.value.description : void 0,
+                tokenCount: Math.ceil(content.length / 4)
+            };
+        }));
+        return results.filter((r)=>null !== r);
+    }
+}
+const AiPromptContextBuilder_AiPromptContextBuilder = AiPromptContextBuilder.createImplementation({
+    implementation: AiPromptContextBuilderImpl,
+    dependencies: [
+        GetSettingsUseCase,
+        ProjectFileAssembler,
+        GetFileContentsByIdUseCase,
+        GetFileUseCase,
+        TextExtractor
+    ]
+});
+export { AiPromptContextBuilder_AiPromptContextBuilder as AiPromptContextBuilder };
+
+//# sourceMappingURL=AiPromptContextBuilder.js.map

package/api/features/AiPromptContext/AiPromptContextBuilder.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"api/features/AiPromptContext/AiPromptContextBuilder.js","sources":["../../../../src/api/features/AiPromptContext/AiPromptContextBuilder.ts"],"sourcesContent":["import { GetSettingsUseCase } from \"~/api/features/GetSettings/index.js\";\nimport { GetFileContentsByIdUseCase } from \"@webiny/api-file-manager/features/file/GetFileContentsById/index.js\";\nimport { GetFileUseCase } from \"@webiny/api-file-manager/features/file/GetFile/index.js\";\nimport { TextExtractor } from \"@webiny/api-core/features/ai/index.js\";\nimport {\n    AiPromptContextBuilder as Abstraction,\n    ProjectFileAssembler,\n    type AiPromptContext,\n    type AiPromptContextParams,\n    type ProjectFileContent,\n    type ResolvedPersona,\n    type ResolvedProject\n} from \"./abstractions.js\";\nimport type { IAiPowerUpsSettings } from \"~/api/types.js\";\nimport type { ProjectPreset } from \"~/api/features/Projects/types.js\";\nimport { WriterPersonaSection } from \"./WriterPersonaSection.js\";\nimport { ReaderPersonaSection } from \"./ReaderPersonaSection.js\";\nimport { ProjectSection } from \"./ProjectSection.js\";\n\nfunction resolvePersonas(\n    settings: IAiPowerUpsSettings,\n    project: ProjectPreset | undefined,\n    params: AiPromptContextParams\n): { reader?: ResolvedPersona; writer?: ResolvedPersona } {\n    const readerPersonaId = params.readerPersonaId ?? project?.defaultReaderPersonaId ?? undefined;\n    const writerPersonaId = params.writerPersonaId ?? project?.defaultWriterPersonaId ?? undefined;\n\n    const readerPreset = readerPersonaId\n        ? settings.readerPersonas?.presets?.find(p => p.id === readerPersonaId)\n        : undefined;\n\n    const writerPreset = writerPersonaId\n        ? settings.writerPersonas?.presets?.find(p => p.id === writerPersonaId)\n        : undefined;\n\n    return {\n        reader: readerPreset\n            ? {\n                  name: readerPreset.name,\n                  description: readerPreset.description,\n                  style: readerPreset.style\n              }\n            : undefined,\n        writer: writerPreset\n            ? {\n                  name: writerPreset.name,\n                  description: writerPreset.description,\n                  style: writerPreset.style\n              }\n            : undefined\n    };\n}\n\nfunction formatContext(ctx: AiPromptContext): string {\n    const sections: string[] = [];\n\n    if (ctx.writerPersona) {\n        sections.push(WriterPersonaSection.format(ctx.writerPersona));\n    }\n    if (ctx.readerPersona) {\n        sections.push(ReaderPersonaSection.format(ctx.readerPersona));\n    }\n    if (ctx.project) {\n        const projectText = ProjectSection.format(ctx.project);\n        if (projectText) {\n            sections.push(projectText);\n        }\n    }\n    return sections.length > 0 ? \"\\n\\n\" + sections.join(\"\\n\\n\") : \"\";\n}\n\nfunction emptyContext(warnings: string[]): AiPromptContext {\n    return {\n        project: undefined,\n        readerPersona: undefined,\n        writerPersona: undefined,\n        allProjectFiles: [],\n        additionalFiles: [],\n        excludedFileIds: new Set(),\n        cacheHit: false,\n        warnings,\n        toString() {\n            return \"\";\n        }\n    };\n}\n\nclass AiPromptContextBuilderImpl implements Abstraction.Interface {\n    constructor(\n        private getSettings: GetSettingsUseCase.Interface,\n        private fileAssembler: ProjectFileAssembler.Interface,\n        private getFileContents: GetFileContentsByIdUseCase.Interface,\n        private getFile: GetFileUseCase.Interface,\n        private textExtractor: TextExtractor.Interface\n    ) {}\n\n    async execute(params: AiPromptContextParams): Promise<AiPromptContext> {\n        const warnings: string[] = [];\n\n        const settingsResult = await this.getSettings.execute();\n        if (settingsResult.isFail()) {\n            warnings.push(\"Failed to load AI PowerUps settings.\");\n            return emptyContext(warnings);\n        }\n\n        const settings = settingsResult.value;\n        const projectPreset = params.projectId\n            ? settings.projects?.presets?.find(p => p.id === params.projectId)\n            : undefined;\n\n        const { reader, writer } = resolvePersonas(settings, projectPreset, params);\n        const excludedFileIds = new Set(params.excludedFileIds ?? []);\n\n        let project: ResolvedProject | undefined;\n        let allProjectFiles: ProjectFileContent[] = [];\n        let cacheHit = false;\n\n        if (projectPreset) {\n            const assembled = await this.fileAssembler.execute(\n                projectPreset.id,\n                projectPreset.version ?? 0,\n                projectPreset.files ?? [],\n                warnings\n            );\n            allProjectFiles = assembled.files;\n            cacheHit = assembled.cacheHit;\n\n            const visibleFiles =\n                excludedFileIds.size > 0\n                    ? allProjectFiles.filter(f => !excludedFileIds.has(f.id))\n                    : allProjectFiles;\n\n            project = {\n                name: projectPreset.name,\n                instructions: projectPreset.instructions,\n                files: visibleFiles,\n                totalTokens: visibleFiles.reduce((sum, f) => sum + f.tokenCount, 0)\n            };\n        }\n\n        const additionalFiles = await this.fetchAdditionalFiles(params.additionalFileIds, warnings);\n        if (additionalFiles.length > 0) {\n            allProjectFiles = [...allProjectFiles, ...additionalFiles];\n        }\n\n        const ctx: AiPromptContext = {\n            project,\n            readerPersona: reader,\n            writerPersona: writer,\n            allProjectFiles,\n            additionalFiles,\n            excludedFileIds,\n            cacheHit,\n            warnings,\n            toString() {\n                return formatContext(this);\n            }\n        };\n\n        return ctx;\n    }\n\n    private async fetchAdditionalFiles(\n        fileIds: string[] | null | undefined,\n        warnings: string[]\n    ): Promise<ProjectFileContent[]> {\n        if (!fileIds || fileIds.length === 0) {\n            return [];\n        }\n\n        const results = await Promise.all(\n            fileIds.map(async (fileId): Promise<ProjectFileContent | null> => {\n                const [contentsResult, damResult] = await Promise.all([\n                    this.getFileContents.execute(fileId),\n                    this.getFile.execute(fileId)\n                ]);\n\n                if (contentsResult.isFail()) {\n                    warnings.push(\n                        `Failed to load additional file (${fileId}): ${contentsResult.error.message}`\n                    );\n                    return null;\n                }\n\n                const name = damResult.isOk() ? damResult.value.name : fileId;\n                const mimeType = damResult.isOk()\n                    ? damResult.value.type\n                    : \"application/octet-stream\";\n\n                if (!this.textExtractor.canExtract(mimeType)) {\n                    warnings.push(\n                        `Skipped additional file \"${name}\" (${fileId}): unsupported type \"${mimeType}\".`\n                    );\n                    return null;\n                }\n\n                const { text: content } = await this.textExtractor.extract(\n                    contentsResult.value.buffer,\n                    mimeType\n                );\n\n                return {\n                    id: fileId,\n                    name,\n                    content,\n                    description: damResult.isOk() ? damResult.value.description : undefined,\n                    tokenCount: Math.ceil(content.length / 4)\n                };\n            })\n        );\n\n        return results.filter((r): r is ProjectFileContent => r !== null);\n    }\n}\n\nexport const AiPromptContextBuilder = Abstraction.createImplementation({\n    implementation: AiPromptContextBuilderImpl,\n    dependencies: [\n        GetSettingsUseCase,\n        ProjectFileAssembler,\n        GetFileContentsByIdUseCase,\n        GetFileUseCase,\n        TextExtractor\n    ]\n});\n"],"names":["resolvePersonas","settings","project","params","readerPersonaId","undefined","writerPersonaId","readerPreset","p","writerPreset","formatContext","ctx","sections","WriterPersonaSection","ReaderPersonaSection","projectText","ProjectSection","emptyContext","warnings","Set","AiPromptContextBuilderImpl","getSettings","fileAssembler","getFileContents","getFile","textExtractor","settingsResult","projectPreset","reader","writer","excludedFileIds","allProjectFiles","cacheHit","assembled","visibleFiles","f","sum","additionalFiles","fileIds","results","Promise","fileId","contentsResult","damResult","name","mimeType","content","Math","r","AiPromptContextBuilder","Abstraction","GetSettingsUseCase","ProjectFileAssembler","GetFileContentsByIdUseCase","GetFileUseCase","TextExtractor"],"mappings":";;;;;;;;AAmBA,SAASA,gBACLC,QAA6B,EAC7BC,OAAkC,EAClCC,MAA6B;IAE7B,MAAMC,kBAAkBD,OAAO,eAAe,IAAID,SAAS,0BAA0BG;IACrF,MAAMC,kBAAkBH,OAAO,eAAe,IAAID,SAAS,0BAA0BG;IAErF,MAAME,eAAeH,kBACfH,SAAS,cAAc,EAAE,SAAS,KAAKO,CAAAA,IAAKA,EAAE,EAAE,KAAKJ,mBACrDC;IAEN,MAAMI,eAAeH,kBACfL,SAAS,cAAc,EAAE,SAAS,KAAKO,CAAAA,IAAKA,EAAE,EAAE,KAAKF,mBACrDD;IAEN,OAAO;QACH,QAAQE,eACF;YACI,MAAMA,aAAa,IAAI;YACvB,aAAaA,aAAa,WAAW;YACrC,OAAOA,aAAa,KAAK;QAC7B,IACAF;QACN,QAAQI,eACF;YACI,MAAMA,aAAa,IAAI;YACvB,aAAaA,aAAa,WAAW;YACrC,OAAOA,aAAa,KAAK;QAC7B,IACAJ;IACV;AACJ;AAEA,SAASK,cAAcC,GAAoB;IACvC,MAAMC,WAAqB,EAAE;IAE7B,IAAID,IAAI,aAAa,EACjBC,SAAS,IAAI,CAACC,qBAAqB,MAAM,CAACF,IAAI,aAAa;IAE/D,IAAIA,IAAI,aAAa,EACjBC,SAAS,IAAI,CAACE,qBAAqB,MAAM,CAACH,IAAI,aAAa;IAE/D,IAAIA,IAAI,OAAO,EAAE;QACb,MAAMI,cAAcC,eAAe,MAAM,CAACL,IAAI,OAAO;QACrD,IAAII,aACAH,SAAS,IAAI,CAACG;IAEtB;IACA,OAAOH,SAAS,MAAM,GAAG,IAAI,SAASA,SAAS,IAAI,CAAC,UAAU;AAClE;AAEA,SAASK,aAAaC,QAAkB;IACpC,OAAO;QACH,SAASb;QACT,eAAeA;QACf,eAAeA;QACf,iBAAiB,EAAE;QACnB,iBAAiB,EAAE;QACnB,iBAAiB,IAAIc;QACrB,UAAU;QACVD;QACA;YACI,OAAO;QACX;IACJ;AACJ;AAEA,MAAME;IACF,YACYC,WAAyC,EACzCC,aAA6C,EAC7CC,eAAqD,EACrDC,OAAiC,EACjCC,aAAsC,CAChD;aALUJ,WAAW,GAAXA;aACAC,aAAa,GAAbA;aACAC,eAAe,GAAfA;aACAC,OAAO,GAAPA;aACAC,aAAa,GAAbA;IACT;IAEH,MAAM,QAAQtB,MAA6B,EAA4B;QACnE,MAAMe,WAAqB,EAAE;QAE7B,MAAMQ,iBAAiB,MAAM,IAAI,CAAC,WAAW,CAAC,OAAO;QACrD,IAAIA,eAAe,MAAM,IAAI;YACzBR,SAAS,IAAI,CAAC;YACd,OAAOD,aAAaC;QACxB;QAEA,MAAMjB,WAAWyB,eAAe,KAAK;QACrC,MAAMC,gBAAgBxB,OAAO,SAAS,GAChCF,SAAS,QAAQ,EAAE,SAAS,KAAKO,CAAAA,IAAKA,EAAE,EAAE,KAAKL,OAAO,SAAS,IAC/DE;QAEN,MAAM,EAAEuB,MAAM,EAAEC,MAAM,EAAE,GAAG7B,gBAAgBC,UAAU0B,eAAexB;QACpE,MAAM2B,kBAAkB,IAAIX,IAAIhB,OAAO,eAAe,IAAI,EAAE;QAE5D,IAAID;QACJ,IAAI6B,kBAAwC,EAAE;QAC9C,IAAIC,WAAW;QAEf,IAAIL,eAAe;YACf,MAAMM,YAAY,MAAM,IAAI,CAAC,aAAa,CAAC,OAAO,CAC9CN,cAAc,EAAE,EAChBA,cAAc,OAAO,IAAI,GACzBA,cAAc,KAAK,IAAI,EAAE,EACzBT;YAEJa,kBAAkBE,UAAU,KAAK;YACjCD,WAAWC,UAAU,QAAQ;YAE7B,MAAMC,eACFJ,gBAAgB,IAAI,GAAG,IACjBC,gBAAgB,MAAM,CAACI,CAAAA,IAAK,CAACL,gBAAgB,GAAG,CAACK,EAAE,EAAE,KACrDJ;YAEV7B,UAAU;gBACN,MAAMyB,cAAc,IAAI;gBACxB,cAAcA,cAAc,YAAY;gBACxC,OAAOO;gBACP,aAAaA,aAAa,MAAM,CAAC,CAACE,KAAKD,IAAMC,MAAMD,EAAE,UAAU,EAAE;YACrE;QACJ;QAEA,MAAME,kBAAkB,MAAM,IAAI,CAAC,oBAAoB,CAAClC,OAAO,iBAAiB,EAAEe;QAClF,IAAImB,gBAAgB,MAAM,GAAG,GACzBN,kBAAkB;eAAIA;eAAoBM;SAAgB;QAG9D,MAAM1B,MAAuB;YACzBT;YACA,eAAe0B;YACf,eAAeC;YACfE;YACAM;YACAP;YACAE;YACAd;YACA;gBACI,OAAOR,cAAc,IAAI;YAC7B;QACJ;QAEA,OAAOC;IACX;IAEA,MAAc,qBACV2B,OAAoC,EACpCpB,QAAkB,EACW;QAC7B,IAAI,CAACoB,WAAWA,AAAmB,MAAnBA,QAAQ,MAAM,EAC1B,OAAO,EAAE;QAGb,MAAMC,UAAU,MAAMC,QAAQ,GAAG,CAC7BF,QAAQ,GAAG,CAAC,OAAOG;YACf,MAAM,CAACC,gBAAgBC,UAAU,GAAG,MAAMH,QAAQ,GAAG,CAAC;gBAClD,IAAI,CAAC,eAAe,CAAC,OAAO,CAACC;gBAC7B,IAAI,CAAC,OAAO,CAAC,OAAO,CAACA;aACxB;YAED,IAAIC,eAAe,MAAM,IAAI;gBACzBxB,SAAS,IAAI,CACT,CAAC,gCAAgC,EAAEuB,OAAO,GAAG,EAAEC,eAAe,KAAK,CAAC,OAAO,EAAE;gBAEjF,OAAO;YACX;YAEA,MAAME,OAAOD,UAAU,IAAI,KAAKA,UAAU,KAAK,CAAC,IAAI,GAAGF;YACvD,MAAMI,WAAWF,UAAU,IAAI,KACzBA,UAAU,KAAK,CAAC,IAAI,GACpB;YAEN,IAAI,CAAC,IAAI,CAAC,aAAa,CAAC,UAAU,CAACE,WAAW;gBAC1C3B,SAAS,IAAI,CACT,CAAC,yBAAyB,EAAE0B,KAAK,GAAG,EAAEH,OAAO,qBAAqB,EAAEI,SAAS,EAAE,CAAC;gBAEpF,OAAO;YACX;YAEA,MAAM,EAAE,MAAMC,OAAO,EAAE,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,OAAO,CACtDJ,eAAe,KAAK,CAAC,MAAM,EAC3BG;YAGJ,OAAO;gBACH,IAAIJ;gBACJG;gBACAE;gBACA,aAAaH,UAAU,IAAI,KAAKA,UAAU,KAAK,CAAC,WAAW,GAAGtC;gBAC9D,YAAY0C,KAAK,IAAI,CAACD,QAAQ,MAAM,GAAG;YAC3C;QACJ;QAGJ,OAAOP,QAAQ,MAAM,CAAC,CAACS,IAA+BA,AAAM,SAANA;IAC1D;AACJ;AAEO,MAAMC,gDAAyBC,uBAAAA,oBAAgC,CAAC;IACnE,gBAAgB9B;IAChB,cAAc;QACV+B;QACAC;QACAC;QACAC;QACAC;KACH;AACL"}

package/api/features/AiPromptContext/ProjectFileAssembler.d.ts

@@ -0,0 +1,22 @@
+import { GetFileContentsByIdUseCase } from "@webiny/api-file-manager/features/file/GetFileContentsById/index.js";
+import { GetFileUseCase } from "@webiny/api-file-manager/features/file/GetFile/index.js";
+import { ProjectFileAssembler as Abstraction, ProjectFileCache, type ProjectFileContent } from "./abstractions.js";
+declare class ProjectFileAssemblerImpl implements Abstraction.Interface {
+    private getFileContents;
+    private getFile;
+    private cache;
+    constructor(getFileContents: GetFileContentsByIdUseCase.Interface, getFile: GetFileUseCase.Interface, cache: ProjectFileCache.Interface);
+    execute(projectId: string, version: number, files: Array<{
+        id: string;
+        name: string;
+        mimeType: string;
+    }>, warnings: string[]): Promise<{
+        files: ProjectFileContent[];
+        cacheHit: boolean;
+    }>;
+    private assembleFiles;
+}
+export declare const ProjectFileAssembler: typeof ProjectFileAssemblerImpl & {
+    __abstraction: import("@webiny/di").Abstraction<import("./abstractions.js").IProjectFileAssembler>;
+};
+export {};

package/api/features/AiPromptContext/ProjectFileAssembler.js

@@ -0,0 +1,76 @@
+import { GetFileContentsByIdUseCase } from "@webiny/api-file-manager/features/file/GetFileContentsById/index.js";
+import { GetFileUseCase } from "@webiny/api-file-manager/features/file/GetFile/index.js";
+import { ProjectFileAssembler, ProjectFileCache } from "./abstractions.js";
+const SUPPORTED_MIME_PREFIXES = [
+    "text/"
+];
+const SUPPORTED_MIME_TYPES = new Set([
+    "application/json",
+    "application/csv"
+]);
+function isSupportedType(mimeType) {
+    if (SUPPORTED_MIME_PREFIXES.some((prefix)=>mimeType.startsWith(prefix))) return true;
+    return SUPPORTED_MIME_TYPES.has(mimeType);
+}
+function estimateTokenCount(text) {
+    return Math.ceil(text.length / 4);
+}
+class ProjectFileAssemblerImpl {
+    constructor(getFileContents, getFile, cache){
+        this.getFileContents = getFileContents;
+        this.getFile = getFile;
+        this.cache = cache;
+    }
+    async execute(projectId, version, files, warnings) {
+        const cached = await this.cache.get(projectId, version);
+        if (cached.hit) return {
+            files: cached.files,
+            cacheHit: true
+        };
+        const assembled = await this.assembleFiles(files, warnings);
+        await this.cache.set(projectId, version, assembled);
+        return {
+            files: assembled,
+            cacheHit: false
+        };
+    }
+    async assembleFiles(files, warnings) {
+        const supportedFiles = files.filter((f)=>isSupportedType(f.mimeType));
+        if (0 === supportedFiles.length) return [];
+        const results = await Promise.all(supportedFiles.map(async (file)=>{
+            const [contentsResult, damResult] = await Promise.all([
+                this.getFileContents.execute(file.id),
+                this.getFile.execute(file.id)
+            ]);
+            if (contentsResult.isFail()) {
+                const msg = `Failed to load file "${file.name}" (${file.id}): ${contentsResult.error.message}`;
+                console.warn("ProjectFileAssembler:", msg);
+                warnings.push(msg);
+                return null;
+            }
+            const content = contentsResult.value.buffer.toString("utf-8");
+            let description;
+            if (damResult.isFail()) console.warn(`ProjectFileAssembler: failed to load DAM record for "${file.name}" (${file.id})`);
+            else if (damResult.value.description) description = damResult.value.description;
+            return {
+                id: file.id,
+                name: file.name,
+                content,
+                description,
+                tokenCount: estimateTokenCount(content)
+            };
+        }));
+        return results.filter((r)=>null !== r);
+    }
+}
+const ProjectFileAssembler_ProjectFileAssembler = ProjectFileAssembler.createImplementation({
+    implementation: ProjectFileAssemblerImpl,
+    dependencies: [
+        GetFileContentsByIdUseCase,
+        GetFileUseCase,
+        ProjectFileCache
+    ]
+});
+export { ProjectFileAssembler_ProjectFileAssembler as ProjectFileAssembler };
+
+//# sourceMappingURL=ProjectFileAssembler.js.map

package/api/features/AiPromptContext/ProjectFileAssembler.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"api/features/AiPromptContext/ProjectFileAssembler.js","sources":["../../../../src/api/features/AiPromptContext/ProjectFileAssembler.ts"],"sourcesContent":["import { GetFileContentsByIdUseCase } from \"@webiny/api-file-manager/features/file/GetFileContentsById/index.js\";\nimport { GetFileUseCase } from \"@webiny/api-file-manager/features/file/GetFile/index.js\";\nimport {\n    ProjectFileAssembler as Abstraction,\n    ProjectFileCache,\n    type ProjectFileContent\n} from \"./abstractions.js\";\n\nconst SUPPORTED_MIME_PREFIXES = [\"text/\"];\nconst SUPPORTED_MIME_TYPES = new Set([\"application/json\", \"application/csv\"]);\n\nfunction isSupportedType(mimeType: string): boolean {\n    if (SUPPORTED_MIME_PREFIXES.some(prefix => mimeType.startsWith(prefix))) {\n        return true;\n    }\n    return SUPPORTED_MIME_TYPES.has(mimeType);\n}\n\nfunction estimateTokenCount(text: string): number {\n    return Math.ceil(text.length / 4);\n}\n\nclass ProjectFileAssemblerImpl implements Abstraction.Interface {\n    constructor(\n        private getFileContents: GetFileContentsByIdUseCase.Interface,\n        private getFile: GetFileUseCase.Interface,\n        private cache: ProjectFileCache.Interface\n    ) {}\n\n    async execute(\n        projectId: string,\n        version: number,\n        files: Array<{ id: string; name: string; mimeType: string }>,\n        warnings: string[]\n    ): Promise<{ files: ProjectFileContent[]; cacheHit: boolean }> {\n        const cached = await this.cache.get(projectId, version);\n        if (cached.hit) {\n            return { files: cached.files, cacheHit: true };\n        }\n\n        const assembled = await this.assembleFiles(files, warnings);\n        await this.cache.set(projectId, version, assembled);\n        return { files: assembled, cacheHit: false };\n    }\n\n    private async assembleFiles(\n        files: Array<{ id: string; name: string; mimeType: string }>,\n        warnings: string[]\n    ): Promise<ProjectFileContent[]> {\n        const supportedFiles = files.filter(f => isSupportedType(f.mimeType));\n\n        if (supportedFiles.length === 0) {\n            return [];\n        }\n\n        const results = await Promise.all(\n            supportedFiles.map(async (file): Promise<ProjectFileContent | null> => {\n                const [contentsResult, damResult] = await Promise.all([\n                    this.getFileContents.execute(file.id),\n                    this.getFile.execute(file.id)\n                ]);\n\n                if (contentsResult.isFail()) {\n                    const msg = `Failed to load file \"${file.name}\" (${file.id}): ${contentsResult.error.message}`;\n                    console.warn(\"ProjectFileAssembler:\", msg);\n                    warnings.push(msg);\n                    return null;\n                }\n\n                const content = contentsResult.value.buffer.toString(\"utf-8\");\n\n                let description: string | undefined;\n                if (damResult.isFail()) {\n                    console.warn(\n                        `ProjectFileAssembler: failed to load DAM record for \"${file.name}\" (${file.id})`\n                    );\n                } else if (damResult.value.description) {\n                    description = damResult.value.description;\n                }\n\n                return {\n                    id: file.id,\n                    name: file.name,\n                    content,\n                    description,\n                    tokenCount: estimateTokenCount(content)\n                };\n            })\n        );\n\n        return results.filter((r): r is ProjectFileContent => r !== null);\n    }\n}\n\nexport const ProjectFileAssembler = Abstraction.createImplementation({\n    implementation: ProjectFileAssemblerImpl,\n    dependencies: [GetFileContentsByIdUseCase, GetFileUseCase, ProjectFileCache]\n});\n"],"names":["SUPPORTED_MIME_PREFIXES","SUPPORTED_MIME_TYPES","Set","isSupportedType","mimeType","prefix","estimateTokenCount","text","Math","ProjectFileAssemblerImpl","getFileContents","getFile","cache","projectId","version","files","warnings","cached","assembled","supportedFiles","f","results","Promise","file","contentsResult","damResult","msg","console","content","description","r","ProjectFileAssembler","Abstraction","GetFileContentsByIdUseCase","GetFileUseCase","ProjectFileCache"],"mappings":";;;AAQA,MAAMA,0BAA0B;IAAC;CAAQ;AACzC,MAAMC,uBAAuB,IAAIC,IAAI;IAAC;IAAoB;CAAkB;AAE5E,SAASC,gBAAgBC,QAAgB;IACrC,IAAIJ,wBAAwB,IAAI,CAACK,CAAAA,SAAUD,SAAS,UAAU,CAACC,UAC3D,OAAO;IAEX,OAAOJ,qBAAqB,GAAG,CAACG;AACpC;AAEA,SAASE,mBAAmBC,IAAY;IACpC,OAAOC,KAAK,IAAI,CAACD,KAAK,MAAM,GAAG;AACnC;AAEA,MAAME;IACF,YACYC,eAAqD,EACrDC,OAAiC,EACjCC,KAAiC,CAC3C;aAHUF,eAAe,GAAfA;aACAC,OAAO,GAAPA;aACAC,KAAK,GAALA;IACT;IAEH,MAAM,QACFC,SAAiB,EACjBC,OAAe,EACfC,KAA4D,EAC5DC,QAAkB,EACyC;QAC3D,MAAMC,SAAS,MAAM,IAAI,CAAC,KAAK,CAAC,GAAG,CAACJ,WAAWC;QAC/C,IAAIG,OAAO,GAAG,EACV,OAAO;YAAE,OAAOA,OAAO,KAAK;YAAE,UAAU;QAAK;QAGjD,MAAMC,YAAY,MAAM,IAAI,CAAC,aAAa,CAACH,OAAOC;QAClD,MAAM,IAAI,CAAC,KAAK,CAAC,GAAG,CAACH,WAAWC,SAASI;QACzC,OAAO;YAAE,OAAOA;YAAW,UAAU;QAAM;IAC/C;IAEA,MAAc,cACVH,KAA4D,EAC5DC,QAAkB,EACW;QAC7B,MAAMG,iBAAiBJ,MAAM,MAAM,CAACK,CAAAA,IAAKjB,gBAAgBiB,EAAE,QAAQ;QAEnE,IAAID,AAA0B,MAA1BA,eAAe,MAAM,EACrB,OAAO,EAAE;QAGb,MAAME,UAAU,MAAMC,QAAQ,GAAG,CAC7BH,eAAe,GAAG,CAAC,OAAOI;YACtB,MAAM,CAACC,gBAAgBC,UAAU,GAAG,MAAMH,QAAQ,GAAG,CAAC;gBAClD,IAAI,CAAC,eAAe,CAAC,OAAO,CAACC,KAAK,EAAE;gBACpC,IAAI,CAAC,OAAO,CAAC,OAAO,CAACA,KAAK,EAAE;aAC/B;YAED,IAAIC,eAAe,MAAM,IAAI;gBACzB,MAAME,MAAM,CAAC,qBAAqB,EAAEH,KAAK,IAAI,CAAC,GAAG,EAAEA,KAAK,EAAE,CAAC,GAAG,EAAEC,eAAe,KAAK,CAAC,OAAO,EAAE;gBAC9FG,QAAQ,IAAI,CAAC,yBAAyBD;gBACtCV,SAAS,IAAI,CAACU;gBACd,OAAO;YACX;YAEA,MAAME,UAAUJ,eAAe,KAAK,CAAC,MAAM,CAAC,QAAQ,CAAC;YAErD,IAAIK;YACJ,IAAIJ,UAAU,MAAM,IAChBE,QAAQ,IAAI,CACR,CAAC,qDAAqD,EAAEJ,KAAK,IAAI,CAAC,GAAG,EAAEA,KAAK,EAAE,CAAC,CAAC,CAAC;iBAElF,IAAIE,UAAU,KAAK,CAAC,WAAW,EAClCI,cAAcJ,UAAU,KAAK,CAAC,WAAW;YAG7C,OAAO;gBACH,IAAIF,KAAK,EAAE;gBACX,MAAMA,KAAK,IAAI;gBACfK;gBACAC;gBACA,YAAYvB,mBAAmBsB;YACnC;QACJ;QAGJ,OAAOP,QAAQ,MAAM,CAAC,CAACS,IAA+BA,AAAM,SAANA;IAC1D;AACJ;AAEO,MAAMC,4CAAuBC,qBAAAA,oBAAgC,CAAC;IACjE,gBAAgBvB;IAChB,cAAc;QAACwB;QAA4BC;QAAgBC;KAAiB;AAChF"}

package/api/features/AiPromptContext/ProjectFileCache.d.ts

@@ -0,0 +1,16 @@
+import { GlobalKeyValueStore } from "@webiny/api-core/features/keyValueStore/index.js";
+import { ProjectFileCache as Abstraction, ProjectFileContent } from "./abstractions.js";
+declare class ProjectFileCacheImpl implements Abstraction.Interface {
+    private keyValueStore;
+    constructor(keyValueStore: GlobalKeyValueStore.Interface);
+    get(projectId: string, version: number): Promise<{
+        files: ProjectFileContent[];
+        hit: boolean;
+    }>;
+    set(projectId: string, version: number, files: ProjectFileContent[]): Promise<void>;
+    private cacheKey;
+}
+export declare const ProjectFileCache: typeof ProjectFileCacheImpl & {
+    __abstraction: import("@webiny/di").Abstraction<import("./abstractions.js").IProjectFileCache>;
+};
+export {};

package/api/features/AiPromptContext/ProjectFileCache.js

@@ -0,0 +1,61 @@
+import { compress, decompress } from "@webiny/utils/features/compression/legacy/gzip.js";
+import { GlobalKeyValueStore } from "@webiny/api-core/features/keyValueStore/index.js";
+import { ProjectFileCache } from "./abstractions.js";
+const CACHE_TTL_DAYS = 30;
+class ProjectFileCacheImpl {
+    constructor(keyValueStore){
+        this.keyValueStore = keyValueStore;
+    }
+    async get(projectId, version) {
+        const key = this.cacheKey(projectId, version);
+        const cached = await this.keyValueStore.get(key);
+        if (!cached.isFail() && cached.value) try {
+            const decompressed = await decompress(Buffer.from(cached.value, "base64"));
+            const parsed = JSON.parse(decompressed.toString("utf-8"));
+            return {
+                files: parsed.files,
+                hit: true
+            };
+        } catch (error) {
+            console.warn("ProjectFileCache: cache entry corrupted, rebuilding", {
+                key,
+                error
+            });
+        }
+        return {
+            files: [],
+            hit: false
+        };
+    }
+    async set(projectId, version, files) {
+        const key = this.cacheKey(projectId, version);
+        try {
+            const payload = {
+                files
+            };
+            const compressed = await compress(JSON.stringify(payload));
+            const expiresAt = new Date();
+            expiresAt.setDate(expiresAt.getDate() + CACHE_TTL_DAYS);
+            await this.keyValueStore.set(key, compressed.toString("base64"), {
+                expiresAt
+            });
+        } catch (error) {
+            console.warn("ProjectFileCache: cache write failed", {
+                key,
+                error
+            });
+        }
+    }
+    cacheKey(projectId, version) {
+        return `AiProjectContext/${projectId}/v${version}`;
+    }
+}
+const ProjectFileCache_ProjectFileCache = ProjectFileCache.createImplementation({
+    implementation: ProjectFileCacheImpl,
+    dependencies: [
+        GlobalKeyValueStore
+    ]
+});
+export { ProjectFileCache_ProjectFileCache as ProjectFileCache };
+
+//# sourceMappingURL=ProjectFileCache.js.map

package/api/features/AiPromptContext/ProjectFileCache.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"api/features/AiPromptContext/ProjectFileCache.js","sources":["../../../../src/api/features/AiPromptContext/ProjectFileCache.ts"],"sourcesContent":["import { compress, decompress } from \"@webiny/utils/features/compression/legacy/gzip.js\";\nimport { GlobalKeyValueStore } from \"@webiny/api-core/features/keyValueStore/index.js\";\nimport { ProjectFileCache as Abstraction, ProjectFileContent } from \"./abstractions.js\";\n\nconst CACHE_TTL_DAYS = 30;\n\ninterface CachedProjectContext {\n    files: ProjectFileContent[];\n}\n\nclass ProjectFileCacheImpl implements Abstraction.Interface {\n    constructor(private keyValueStore: GlobalKeyValueStore.Interface) {}\n\n    async get(\n        projectId: string,\n        version: number\n    ): Promise<{ files: ProjectFileContent[]; hit: boolean }> {\n        const key = this.cacheKey(projectId, version);\n        const cached = await this.keyValueStore.get<string>(key);\n\n        if (!cached.isFail() && cached.value) {\n            try {\n                const decompressed = await decompress(Buffer.from(cached.value, \"base64\"));\n                const parsed = JSON.parse(decompressed.toString(\"utf-8\")) as CachedProjectContext;\n                return { files: parsed.files, hit: true };\n            } catch (error) {\n                console.warn(\"ProjectFileCache: cache entry corrupted, rebuilding\", {\n                    key,\n                    error\n                });\n            }\n        }\n\n        return { files: [], hit: false };\n    }\n\n    async set(projectId: string, version: number, files: ProjectFileContent[]): Promise<void> {\n        const key = this.cacheKey(projectId, version);\n\n        try {\n            const payload: CachedProjectContext = { files };\n            const compressed = await compress(JSON.stringify(payload));\n            const expiresAt = new Date();\n            expiresAt.setDate(expiresAt.getDate() + CACHE_TTL_DAYS);\n            await this.keyValueStore.set(key, compressed.toString(\"base64\"), { expiresAt });\n        } catch (error) {\n            console.warn(\"ProjectFileCache: cache write failed\", { key, error });\n        }\n    }\n\n    private cacheKey(projectId: string, version: number): string {\n        return `AiProjectContext/${projectId}/v${version}`;\n    }\n}\n\nexport const ProjectFileCache = Abstraction.createImplementation({\n    implementation: ProjectFileCacheImpl,\n    dependencies: [GlobalKeyValueStore]\n});\n"],"names":["CACHE_TTL_DAYS","ProjectFileCacheImpl","keyValueStore","projectId","version","key","cached","decompressed","decompress","Buffer","parsed","JSON","error","console","files","payload","compressed","compress","expiresAt","Date","ProjectFileCache","Abstraction","GlobalKeyValueStore"],"mappings":";;;AAIA,MAAMA,iBAAiB;AAMvB,MAAMC;IACF,YAAoBC,aAA4C,CAAE;aAA9CA,aAAa,GAAbA;IAA+C;IAEnE,MAAM,IACFC,SAAiB,EACjBC,OAAe,EACuC;QACtD,MAAMC,MAAM,IAAI,CAAC,QAAQ,CAACF,WAAWC;QACrC,MAAME,SAAS,MAAM,IAAI,CAAC,aAAa,CAAC,GAAG,CAASD;QAEpD,IAAI,CAACC,OAAO,MAAM,MAAMA,OAAO,KAAK,EAChC,IAAI;YACA,MAAMC,eAAe,MAAMC,WAAWC,OAAO,IAAI,CAACH,OAAO,KAAK,EAAE;YAChE,MAAMI,SAASC,KAAK,KAAK,CAACJ,aAAa,QAAQ,CAAC;YAChD,OAAO;gBAAE,OAAOG,OAAO,KAAK;gBAAE,KAAK;YAAK;QAC5C,EAAE,OAAOE,OAAO;YACZC,QAAQ,IAAI,CAAC,uDAAuD;gBAChER;gBACAO;YACJ;QACJ;QAGJ,OAAO;YAAE,OAAO,EAAE;YAAE,KAAK;QAAM;IACnC;IAEA,MAAM,IAAIT,SAAiB,EAAEC,OAAe,EAAEU,KAA2B,EAAiB;QACtF,MAAMT,MAAM,IAAI,CAAC,QAAQ,CAACF,WAAWC;QAErC,IAAI;YACA,MAAMW,UAAgC;gBAAED;YAAM;YAC9C,MAAME,aAAa,MAAMC,SAASN,KAAK,SAAS,CAACI;YACjD,MAAMG,YAAY,IAAIC;YACtBD,UAAU,OAAO,CAACA,UAAU,OAAO,KAAKlB;YACxC,MAAM,IAAI,CAAC,aAAa,CAAC,GAAG,CAACK,KAAKW,WAAW,QAAQ,CAAC,WAAW;gBAAEE;YAAU;QACjF,EAAE,OAAON,OAAO;YACZC,QAAQ,IAAI,CAAC,wCAAwC;gBAAER;gBAAKO;YAAM;QACtE;IACJ;IAEQ,SAAST,SAAiB,EAAEC,OAAe,EAAU;QACzD,OAAO,CAAC,iBAAiB,EAAED,UAAU,EAAE,EAAEC,SAAS;IACtD;AACJ;AAEO,MAAMgB,oCAAmBC,iBAAAA,oBAAgC,CAAC;IAC7D,gBAAgBpB;IAChB,cAAc;QAACqB;KAAoB;AACvC"}

package/api/features/AiPromptContext/ProjectSection.d.ts

@@ -0,0 +1,4 @@
+import type { ResolvedProject } from "./abstractions.js";
+export declare class ProjectSection {
+    static format(project: ResolvedProject): string;
+}

package/api/features/AiPromptContext/ProjectSection.js

@@ -0,0 +1,35 @@
+class ProjectSection {
+    static format(project) {
+        const hasInstructions = !!project.instructions;
+        const hasFiles = project.files.length > 0;
+        if (!hasInstructions && !hasFiles) return "";
+        const parts = [
+            `## Project: ${project.name}`
+        ];
+        if (hasInstructions) parts.push(project.instructions);
+        if (hasFiles) parts.push([
+            "### Available reference files",
+            "",
+            `You have access to ${project.files.length} reference files for this project. Use the`,
+            "`read_project_file` tool to read any file when its contents are relevant",
+            "to the user's request. Read files only when needed — do not read all files",
+            "preemptively.",
+            "",
+            "Files:",
+            "",
+            ...project.files.map((file)=>{
+                const lines = [
+                    `- id: "${file.id}"`,
+                    `  name: "${file.name}"`,
+                    `  description: "${file.description || "(no description)"}"`,
+                    `  tokens: ~${file.tokenCount}`
+                ];
+                return lines.join("\n");
+            })
+        ].join("\n"));
+        return parts.join("\n\n");
+    }
+}
+export { ProjectSection };
+
+//# sourceMappingURL=ProjectSection.js.map

package/api/features/AiPromptContext/ProjectSection.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"api/features/AiPromptContext/ProjectSection.js","sources":["../../../../src/api/features/AiPromptContext/ProjectSection.ts"],"sourcesContent":["import type { ResolvedProject } from \"./abstractions.js\";\n\nexport class ProjectSection {\n    static format(project: ResolvedProject): string {\n        const hasInstructions = !!project.instructions;\n        const hasFiles = project.files.length > 0;\n\n        if (!hasInstructions && !hasFiles) {\n            return \"\";\n        }\n\n        const parts: string[] = [`## Project: ${project.name}`];\n\n        if (hasInstructions) {\n            parts.push(project.instructions!);\n        }\n\n        if (hasFiles) {\n            parts.push(\n                [\n                    \"### Available reference files\",\n                    \"\",\n                    `You have access to ${project.files.length} reference files for this project. Use the`,\n                    \"`read_project_file` tool to read any file when its contents are relevant\",\n                    \"to the user's request. Read files only when needed — do not read all files\",\n                    \"preemptively.\",\n                    \"\",\n                    \"Files:\",\n                    \"\",\n                    ...project.files.map(file => {\n                        const lines = [\n                            `- id: \"${file.id}\"`,\n                            `  name: \"${file.name}\"`,\n                            `  description: \"${file.description || \"(no description)\"}\"`,\n                            `  tokens: ~${file.tokenCount}`\n                        ];\n                        return lines.join(\"\\n\");\n                    })\n                ].join(\"\\n\")\n            );\n        }\n\n        return parts.join(\"\\n\\n\");\n    }\n}\n"],"names":["ProjectSection","project","hasInstructions","hasFiles","parts","file","lines"],"mappings":"AAEO,MAAMA;IACT,OAAO,OAAOC,OAAwB,EAAU;QAC5C,MAAMC,kBAAkB,CAAC,CAACD,QAAQ,YAAY;QAC9C,MAAME,WAAWF,QAAQ,KAAK,CAAC,MAAM,GAAG;QAExC,IAAI,CAACC,mBAAmB,CAACC,UACrB,OAAO;QAGX,MAAMC,QAAkB;YAAC,CAAC,YAAY,EAAEH,QAAQ,IAAI,EAAE;SAAC;QAEvD,IAAIC,iBACAE,MAAM,IAAI,CAACH,QAAQ,YAAY;QAGnC,IAAIE,UACAC,MAAM,IAAI,CACN;YACI;YACA;YACA,CAAC,mBAAmB,EAAEH,QAAQ,KAAK,CAAC,MAAM,CAAC,0CAA0C,CAAC;YACtF;YACA;YACA;YACA;YACA;YACA;eACGA,QAAQ,KAAK,CAAC,GAAG,CAACI,CAAAA;gBACjB,MAAMC,QAAQ;oBACV,CAAC,OAAO,EAAED,KAAK,EAAE,CAAC,CAAC,CAAC;oBACpB,CAAC,SAAS,EAAEA,KAAK,IAAI,CAAC,CAAC,CAAC;oBACxB,CAAC,gBAAgB,EAAEA,KAAK,WAAW,IAAI,mBAAmB,CAAC,CAAC;oBAC5D,CAAC,WAAW,EAAEA,KAAK,UAAU,EAAE;iBAClC;gBACD,OAAOC,MAAM,IAAI,CAAC;YACtB;SACH,CAAC,IAAI,CAAC;QAIf,OAAOF,MAAM,IAAI,CAAC;IACtB;AACJ"}

package/api/features/AiPromptContext/ReadProjectFileTool.d.ts

@@ -0,0 +1,3 @@
+import type { ToolSet } from "ai";
+import type { ProjectFileContent } from "./abstractions.js";
+export declare function createReadProjectFileTool(files: ProjectFileContent[], excludedFileIds: Set<string>): ToolSet;

package/api/features/AiPromptContext/ReadProjectFileTool.js

@@ -0,0 +1,36 @@
+import { z } from "zod";
+const inputSchema = z.object({
+    fileId: z.string().describe("The id of the file from the project manifest.")
+});
+function createReadProjectFileTool(files, excludedFileIds) {
+    const fileMap = new Map(files.map((f)=>[
+            f.id,
+            f
+        ]));
+    return {
+        read_project_file: {
+            description: "Read the full contents of a project reference file by its id. Use this when a file from the manifest contains information relevant to the user's request.",
+            inputSchema: inputSchema,
+            execute: async (input)=>{
+                const { fileId } = input;
+                if (excludedFileIds.has(fileId)) return {
+                    error: "File excluded for this generation."
+                };
+                const file = fileMap.get(fileId);
+                if (!file) {
+                    const availableIds = files.map((f)=>f.id).join(", ");
+                    return {
+                        error: `File not found. Available ids: ${availableIds}`
+                    };
+                }
+                return {
+                    content: file.content,
+                    name: file.name
+                };
+            }
+        }
+    };
+}
+export { createReadProjectFileTool };
+
+//# sourceMappingURL=ReadProjectFileTool.js.map

package/api/features/AiPromptContext/ReadProjectFileTool.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"api/features/AiPromptContext/ReadProjectFileTool.js","sources":["../../../../src/api/features/AiPromptContext/ReadProjectFileTool.ts"],"sourcesContent":["import { z } from \"zod\";\nimport type { ToolSet } from \"ai\";\nimport type { ProjectFileContent } from \"./abstractions.js\";\n\nconst inputSchema = z.object({\n    fileId: z.string().describe(\"The id of the file from the project manifest.\")\n});\n\nexport function createReadProjectFileTool(\n    files: ProjectFileContent[],\n    excludedFileIds: Set<string>\n): ToolSet {\n    const fileMap = new Map(files.map(f => [f.id, f]));\n\n    return {\n        read_project_file: {\n            description:\n                \"Read the full contents of a project reference file by its id. \" +\n                \"Use this when a file from the manifest contains information relevant to the user's request.\",\n            inputSchema,\n            execute: async (input: unknown) => {\n                const { fileId } = input as z.infer<typeof inputSchema>;\n\n                if (excludedFileIds.has(fileId)) {\n                    return { error: \"File excluded for this generation.\" };\n                }\n\n                const file = fileMap.get(fileId);\n                if (!file) {\n                    const availableIds = files.map(f => f.id).join(\", \");\n                    return {\n                        error: `File not found. Available ids: ${availableIds}`\n                    };\n                }\n\n                return {\n                    content: file.content,\n                    name: file.name\n                };\n            }\n        }\n    };\n}\n"],"names":["inputSchema","z","createReadProjectFileTool","files","excludedFileIds","fileMap","Map","f","input","fileId","file","availableIds"],"mappings":";AAIA,MAAMA,cAAcC,EAAE,MAAM,CAAC;IACzB,QAAQA,EAAE,MAAM,GAAG,QAAQ,CAAC;AAChC;AAEO,SAASC,0BACZC,KAA2B,EAC3BC,eAA4B;IAE5B,MAAMC,UAAU,IAAIC,IAAIH,MAAM,GAAG,CAACI,CAAAA,IAAK;YAACA,EAAE,EAAE;YAAEA;SAAE;IAEhD,OAAO;QACH,mBAAmB;YACf,aACI;YAEJP,aAAAA;YACA,SAAS,OAAOQ;gBACZ,MAAM,EAAEC,MAAM,EAAE,GAAGD;gBAEnB,IAAIJ,gBAAgB,GAAG,CAACK,SACpB,OAAO;oBAAE,OAAO;gBAAqC;gBAGzD,MAAMC,OAAOL,QAAQ,GAAG,CAACI;gBACzB,IAAI,CAACC,MAAM;oBACP,MAAMC,eAAeR,MAAM,GAAG,CAACI,CAAAA,IAAKA,EAAE,EAAE,EAAE,IAAI,CAAC;oBAC/C,OAAO;wBACH,OAAO,CAAC,+BAA+B,EAAEI,cAAc;oBAC3D;gBACJ;gBAEA,OAAO;oBACH,SAASD,KAAK,OAAO;oBACrB,MAAMA,KAAK,IAAI;gBACnB;YACJ;QACJ;IACJ;AACJ"}

package/api/features/AiPromptContext/ReaderPersonaSection.d.ts

@@ -0,0 +1,4 @@
+import type { ResolvedPersona } from "./abstractions.js";
+export declare class ReaderPersonaSection {
+    static format(persona: ResolvedPersona): string;
+}

package/api/features/AiPromptContext/ReaderPersonaSection.js

@@ -0,0 +1,10 @@
+class ReaderPersonaSection {
+    static format(persona) {
+        let section = `### Reader Persona (Target Audience)\n\nYou are writing for "${persona.name}": ${persona.description}`;
+        if (persona.style) section += `\nExpected tone/style: ${persona.style}`;
+        return section;
+    }
+}
+export { ReaderPersonaSection };
+
+//# sourceMappingURL=ReaderPersonaSection.js.map

package/api/features/AiPromptContext/ReaderPersonaSection.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"api/features/AiPromptContext/ReaderPersonaSection.js","sources":["../../../../src/api/features/AiPromptContext/ReaderPersonaSection.ts"],"sourcesContent":["import type { ResolvedPersona } from \"./abstractions.js\";\n\nexport class ReaderPersonaSection {\n    static format(persona: ResolvedPersona): string {\n        let section = `### Reader Persona (Target Audience)\\n\\nYou are writing for \"${persona.name}\": ${persona.description}`;\n\n        if (persona.style) {\n            section += `\\nExpected tone/style: ${persona.style}`;\n        }\n\n        return section;\n    }\n}\n"],"names":["ReaderPersonaSection","persona","section"],"mappings":"AAEO,MAAMA;IACT,OAAO,OAAOC,OAAwB,EAAU;QAC5C,IAAIC,UAAU,CAAC,6DAA6D,EAAED,QAAQ,IAAI,CAAC,GAAG,EAAEA,QAAQ,WAAW,EAAE;QAErH,IAAIA,QAAQ,KAAK,EACbC,WAAW,CAAC,uBAAuB,EAAED,QAAQ,KAAK,EAAE;QAGxD,OAAOC;IACX;AACJ"}

package/api/features/AiPromptContext/WriterPersonaSection.d.ts

@@ -0,0 +1,4 @@
+import type { ResolvedPersona } from "./abstractions.js";
+export declare class WriterPersonaSection {
+    static format(persona: ResolvedPersona): string;
+}

package/api/features/AiPromptContext/WriterPersonaSection.js

@@ -0,0 +1,10 @@
+class WriterPersonaSection {
+    static format(persona) {
+        let section = `### Writer Persona\n\nYou are writing as "${persona.name}": ${persona.description}`;
+        if (persona.style) section += `\nStyle: ${persona.style}`;
+        return section;
+    }
+}
+export { WriterPersonaSection };
+
+//# sourceMappingURL=WriterPersonaSection.js.map

package/api/features/AiPromptContext/WriterPersonaSection.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"api/features/AiPromptContext/WriterPersonaSection.js","sources":["../../../../src/api/features/AiPromptContext/WriterPersonaSection.ts"],"sourcesContent":["import type { ResolvedPersona } from \"./abstractions.js\";\n\nexport class WriterPersonaSection {\n    static format(persona: ResolvedPersona): string {\n        let section = `### Writer Persona\\n\\nYou are writing as \"${persona.name}\": ${persona.description}`;\n\n        if (persona.style) {\n            section += `\\nStyle: ${persona.style}`;\n        }\n\n        return section;\n    }\n}\n"],"names":["WriterPersonaSection","persona","section"],"mappings":"AAEO,MAAMA;IACT,OAAO,OAAOC,OAAwB,EAAU;QAC5C,IAAIC,UAAU,CAAC,0CAA0C,EAAED,QAAQ,IAAI,CAAC,GAAG,EAAEA,QAAQ,WAAW,EAAE;QAElG,IAAIA,QAAQ,KAAK,EACbC,WAAW,CAAC,SAAS,EAAED,QAAQ,KAAK,EAAE;QAG1C,OAAOC;IACX;AACJ"}

package/api/features/AiPromptContext/abstractions.d.ts

@@ -0,0 +1,71 @@
+export interface IProjectFileCache {
+    get(projectId: string, version: number): Promise<{
+        files: ProjectFileContent[];
+        hit: boolean;
+    }>;
+    set(projectId: string, version: number, files: ProjectFileContent[]): Promise<void>;
+}
+export declare const ProjectFileCache: import("@webiny/di").Abstraction<IProjectFileCache>;
+export declare namespace ProjectFileCache {
+    type Interface = IProjectFileCache;
+}
+export interface IProjectFileAssembler {
+    execute(projectId: string, version: number, files: Array<{
+        id: string;
+        name: string;
+        mimeType: string;
+    }>, warnings: string[]): Promise<{
+        files: ProjectFileContent[];
+        cacheHit: boolean;
+    }>;
+}
+export declare const ProjectFileAssembler: import("@webiny/di").Abstraction<IProjectFileAssembler>;
+export declare namespace ProjectFileAssembler {
+    type Interface = IProjectFileAssembler;
+}
+export interface AiPromptContextParams {
+    projectId?: string | null;
+    readerPersonaId?: string | null;
+    writerPersonaId?: string | null;
+    excludedFileIds?: string[] | null;
+    additionalFileIds?: string[] | null;
+}
+export interface ProjectFileContent {
+    id: string;
+    name: string;
+    content: string;
+    description?: string;
+    tokenCount: number;
+}
+export interface ResolvedProject {
+    name: string;
+    instructions?: string;
+    files: ProjectFileContent[];
+    totalTokens: number;
+}
+export interface ResolvedPersona {
+    name: string;
+    description: string;
+    style?: string;
+}
+export interface AiPromptContext {
+    project?: ResolvedProject;
+    readerPersona?: ResolvedPersona;
+    writerPersona?: ResolvedPersona;
+    allProjectFiles: ProjectFileContent[];
+    additionalFiles: ProjectFileContent[];
+    excludedFileIds: Set<string>;
+    cacheHit: boolean;
+    warnings: string[];
+    toString(): string;
+}
+export interface IAiPromptContextBuilder {
+    execute(params: AiPromptContextParams): Promise<AiPromptContext>;
+}
+export declare const AiPromptContextBuilder: import("@webiny/di").Abstraction<IAiPromptContextBuilder>;
+export declare namespace AiPromptContextBuilder {
+    type Interface = IAiPromptContextBuilder;
+    type Params = AiPromptContextParams;
+    type Context = AiPromptContext;
+}
+export declare function formatAdditionalFilesContext(files: ProjectFileContent[]): string;

package/api/features/AiPromptContext/abstractions.js

@@ -0,0 +1,15 @@
+import { createAbstraction } from "@webiny/feature/api";
+const ProjectFileCache = createAbstraction("ProjectFileCache");
+const ProjectFileAssembler = createAbstraction("ProjectFileAssembler");
+const AiPromptContextBuilder = createAbstraction("AiPowerUpsPromptContextBuilder");
+function formatAdditionalFilesContext(files) {
+    if (0 === files.length) return "";
+    const sections = files.map((f)=>{
+        const desc = f.description ? ` description="${f.description}"` : "";
+        return `<attached-file name="${f.name}"${desc}>\n${f.content}\n</attached-file>`;
+    });
+    return `\n\nThe user attached the following files as additional context for this generation:\n\n${sections.join("\n\n")}`;
+}
+export { AiPromptContextBuilder, ProjectFileAssembler, ProjectFileCache, formatAdditionalFilesContext };
+
+//# sourceMappingURL=abstractions.js.map

package/api/features/AiPromptContext/abstractions.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"api/features/AiPromptContext/abstractions.js","sources":["../../../../src/api/features/AiPromptContext/abstractions.ts"],"sourcesContent":["import { createAbstraction } from \"@webiny/feature/api\";\n\n// ============================================================================\n// ProjectFileCache\n// ============================================================================\n\nexport interface IProjectFileCache {\n    get(projectId: string, version: number): Promise<{ files: ProjectFileContent[]; hit: boolean }>;\n    set(projectId: string, version: number, files: ProjectFileContent[]): Promise<void>;\n}\n\nexport const ProjectFileCache = createAbstraction<IProjectFileCache>(\"ProjectFileCache\");\n\nexport namespace ProjectFileCache {\n    export type Interface = IProjectFileCache;\n}\n\n// ============================================================================\n// ProjectFileAssembler\n// ============================================================================\n\nexport interface IProjectFileAssembler {\n    execute(\n        projectId: string,\n        version: number,\n        files: Array<{ id: string; name: string; mimeType: string }>,\n        warnings: string[]\n    ): Promise<{ files: ProjectFileContent[]; cacheHit: boolean }>;\n}\n\nexport const ProjectFileAssembler =\n    createAbstraction<IProjectFileAssembler>(\"ProjectFileAssembler\");\n\nexport namespace ProjectFileAssembler {\n    export type Interface = IProjectFileAssembler;\n}\n\n// ============================================================================\n// AiPromptContext\n// ============================================================================\n\nexport interface AiPromptContextParams {\n    projectId?: string | null;\n    readerPersonaId?: string | null;\n    writerPersonaId?: string | null;\n    excludedFileIds?: string[] | null;\n    additionalFileIds?: string[] | null;\n}\n\nexport interface ProjectFileContent {\n    id: string;\n    name: string;\n    content: string;\n    description?: string;\n    tokenCount: number;\n}\n\nexport interface ResolvedProject {\n    name: string;\n    instructions?: string;\n    files: ProjectFileContent[];\n    totalTokens: number;\n}\n\nexport interface ResolvedPersona {\n    name: string;\n    description: string;\n    style?: string;\n}\n\nexport interface AiPromptContext {\n    project?: ResolvedProject;\n    readerPersona?: ResolvedPersona;\n    writerPersona?: ResolvedPersona;\n    allProjectFiles: ProjectFileContent[];\n    additionalFiles: ProjectFileContent[];\n    excludedFileIds: Set<string>;\n    cacheHit: boolean;\n    warnings: string[];\n    toString(): string;\n}\n\nexport interface IAiPromptContextBuilder {\n    execute(params: AiPromptContextParams): Promise<AiPromptContext>;\n}\n\nexport const AiPromptContextBuilder = createAbstraction<IAiPromptContextBuilder>(\n    \"AiPowerUpsPromptContextBuilder\"\n);\n\nexport namespace AiPromptContextBuilder {\n    export type Interface = IAiPromptContextBuilder;\n    export type Params = AiPromptContextParams;\n    export type Context = AiPromptContext;\n}\n\nexport function formatAdditionalFilesContext(files: ProjectFileContent[]): string {\n    if (files.length === 0) {\n        return \"\";\n    }\n    const sections = files.map(f => {\n        const desc = f.description ? ` description=\"${f.description}\"` : \"\";\n        return `<attached-file name=\"${f.name}\"${desc}>\\n${f.content}\\n</attached-file>`;\n    });\n    return `\\n\\nThe user attached the following files as additional context for this generation:\\n\\n${sections.join(\"\\n\\n\")}`;\n}\n"],"names":["ProjectFileCache","createAbstraction","ProjectFileAssembler","AiPromptContextBuilder","formatAdditionalFilesContext","files","sections","f","desc"],"mappings":";AAWO,MAAMA,mBAAmBC,kBAAqC;AAmB9D,MAAMC,uBACTD,kBAAyC;AAuDtC,MAAME,yBAAyBF,kBAClC;AASG,SAASG,6BAA6BC,KAA2B;IACpE,IAAIA,AAAiB,MAAjBA,MAAM,MAAM,EACZ,OAAO;IAEX,MAAMC,WAAWD,MAAM,GAAG,CAACE,CAAAA;QACvB,MAAMC,OAAOD,EAAE,WAAW,GAAG,CAAC,cAAc,EAAEA,EAAE,WAAW,CAAC,CAAC,CAAC,GAAG;QACjE,OAAO,CAAC,qBAAqB,EAAEA,EAAE,IAAI,CAAC,CAAC,EAAEC,KAAK,GAAG,EAAED,EAAE,OAAO,CAAC,kBAAkB,CAAC;IACpF;IACA,OAAO,CAAC,wFAAwF,EAAED,SAAS,IAAI,CAAC,SAAS;AAC7H"}

package/api/features/AiPromptContext/feature.d.ts

@@ -0,0 +1,4 @@
+export declare const AiPromptContextFeature: {
+    name: string;
+    register(container: import("@webiny/di").Container): void;
+};