@webiny/ai-powerups 0.0.0-unstable.7be00a75a9

package/api/features/AiImageEnrichment/AiImageEnrichmentTask.js

@@ -0,0 +1,140 @@
+import { Output } from "ai";
+import { z } from "zod";
+import { TaskDefinition } from "@webiny/api-core/features/task/TaskDefinition/index.js";
+import { Ai } from "@webiny/api-core/features/ai/index.js";
+import { Encryption } from "@webiny/api-core/features/encryption/index.js";
+import { GetFileUseCase } from "@webiny/api-file-manager/features/file/GetFile/index.js";
+import { UpdateFileUseCase } from "@webiny/api-file-manager/features/file/UpdateFile/index.js";
+import { GetSettingsUseCase } from "@webiny/api-file-manager/features/settings/GetSettings/abstractions.js";
+import { WebsocketService } from "@webiny/api-websockets/features/WebsocketService/index.js";
+import { GetSettingsUseCase as index_js_GetSettingsUseCase } from "../GetSettings/index.js";
+const AI_IMAGE_ENRICHMENT_TASK_ID = "fmAiImageEnrichment";
+const AI_PROMPT = "Analyze this image and return up to 5 lowercase descriptive tags and one short sentence describing the image.";
+const aiOutputSchema = Output.object({
+    schema: z.object({
+        tags: z.array(z.string()),
+        description: z.string()
+    })
+});
+class AiImageEnrichmentTaskImpl {
+    constructor(getFile, fmSettings, updateFile, ai, getSettings, encryption, websocketService){
+        this.getFile = getFile;
+        this.fmSettings = fmSettings;
+        this.updateFile = updateFile;
+        this.ai = ai;
+        this.getSettings = getSettings;
+        this.encryption = encryption;
+        this.websocketService = websocketService;
+        this.id = AI_IMAGE_ENRICHMENT_TASK_ID;
+        this.title = "File Manager - AI Image Enrichment";
+        this.description = "Automatically enriches uploaded images with AI-generated tags and description.";
+        this.maxIterations = 1;
+        this.isPrivate = true;
+        this.databaseLogs = false;
+        this.selfCleanup = [
+            "onSuccess",
+            "onAbort"
+        ];
+    }
+    async run({ input, controller }) {
+        if (controller.runtime.isAborted()) return controller.response.aborted();
+        const fileResult = await this.getFile.execute(input.fileId);
+        if (fileResult.isFail()) return controller.response.error({
+            message: `File not found: ${input.fileId}`
+        });
+        const file = fileResult.value;
+        if (!file.type.startsWith("image/")) return controller.response.done("File is not an image; skipping AI enrichment.");
+        const settingsResult = await this.fmSettings.execute();
+        const srcPrefix = settingsResult.isOk() ? settingsResult.value.srcPrefix ?? "" : "";
+        const imageUrl = `${srcPrefix}${file.key}`;
+        const aiSettingsResult = await this.getSettings.execute();
+        if (aiSettingsResult.isFail()) return controller.response.error({
+            message: "No AI provider configured. Add a provider in AI Power Ups settings."
+        });
+        const aiSettings = aiSettingsResult.value;
+        const firstProvider = aiSettings.providers.presets[0];
+        if (!firstProvider) return controller.response.done({
+            message: "No AI provider configured. Add a provider in AI Power Ups settings."
+        });
+        let tags = [];
+        let description = "";
+        try {
+            const aiResult = await this.ai.generateText({
+                model: firstProvider.model,
+                output: aiOutputSchema,
+                connection: {
+                    sdkName: firstProvider.model.split("/")[0],
+                    apiKey: await this.encryption.decrypt(firstProvider.apiKeyEncrypted)
+                },
+                messages: [
+                    {
+                        role: "user",
+                        content: [
+                            {
+                                type: "image",
+                                image: new URL(imageUrl)
+                            },
+                            {
+                                type: "text",
+                                text: AI_PROMPT
+                            }
+                        ]
+                    }
+                ]
+            });
+            tags = aiResult.output.tags;
+            description = aiResult.output.description;
+        } catch (error) {
+            console.log("error", error.message);
+            return controller.response.error({
+                message: `AI enrichment failed: ${error instanceof Error ? error.message : String(error)}`
+            });
+        }
+        const mergedTags = [
+            ...new Set([
+                ...file.tags,
+                ...tags
+            ])
+        ];
+        const updateResult = await this.updateFile.execute({
+            id: file.id,
+            tags: mergedTags,
+            description
+        });
+        if (updateResult.isFail()) return controller.response.error({
+            message: `Failed to update file: ${updateResult.error.message}`
+        });
+        if (this.websocketService) {
+            const connectionsResult = await this.websocketService.listConnections();
+            if (connectionsResult.isOk() && connectionsResult.value.length > 0) await this.websocketService.sendToConnections(connectionsResult.value, {
+                action: "fm.file.enrichment",
+                data: {
+                    id: file.id,
+                    tags: mergedTags,
+                    description
+                }
+            });
+        }
+        return controller.response.done("AI image enrichment completed successfully.");
+    }
+}
+const AiImageEnrichmentTask = TaskDefinition.createImplementation({
+    implementation: AiImageEnrichmentTaskImpl,
+    dependencies: [
+        GetFileUseCase,
+        GetSettingsUseCase,
+        UpdateFileUseCase,
+        Ai,
+        index_js_GetSettingsUseCase,
+        Encryption,
+        [
+            WebsocketService,
+            {
+                optional: true
+            }
+        ]
+    ]
+});
+export { AI_IMAGE_ENRICHMENT_TASK_ID, AiImageEnrichmentTask };
+
+//# sourceMappingURL=AiImageEnrichmentTask.js.map

package/api/features/AiImageEnrichment/AiImageEnrichmentTask.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"api/features/AiImageEnrichment/AiImageEnrichmentTask.js","sources":["../../../../src/api/features/AiImageEnrichment/AiImageEnrichmentTask.ts"],"sourcesContent":["import { Output } from \"ai\";\nimport { z } from \"zod\";\nimport { TaskDefinition } from \"@webiny/api-core/features/task/TaskDefinition/index.js\";\nimport { Ai } from \"@webiny/api-core/features/ai/index.js\";\nimport { Encryption } from \"@webiny/api-core/features/encryption/index.js\";\nimport { GetFileUseCase } from \"@webiny/api-file-manager/features/file/GetFile/index.js\";\nimport { UpdateFileUseCase } from \"@webiny/api-file-manager/features/file/UpdateFile/index.js\";\nimport { GetSettingsUseCase as FmGetSettingsUseCase } from \"@webiny/api-file-manager/features/settings/GetSettings/abstractions.js\";\nimport { WebsocketService } from \"@webiny/api-websockets/features/WebsocketService/index.js\";\nimport { GetSettingsUseCase } from \"~/api/features/GetSettings/index.js\";\n\nexport const AI_IMAGE_ENRICHMENT_TASK_ID = \"fmAiImageEnrichment\";\n\nconst AI_PROMPT =\n    \"Analyze this image and return up to 5 lowercase descriptive tags and one short sentence describing the image.\";\n\nconst aiOutputSchema = Output.object({\n    schema: z.object({\n        tags: z.array(z.string()),\n        description: z.string()\n    })\n});\n\nexport interface IAiImageEnrichmentTaskInput {\n    fileId: string;\n}\n\nclass AiImageEnrichmentTaskImpl implements TaskDefinition.Interface<IAiImageEnrichmentTaskInput> {\n    id = AI_IMAGE_ENRICHMENT_TASK_ID;\n    title = \"File Manager - AI Image Enrichment\";\n    description = \"Automatically enriches uploaded images with AI-generated tags and description.\";\n    maxIterations = 1;\n    isPrivate = true;\n    databaseLogs = false;\n\n    public readonly selfCleanup = [\"onSuccess\" as const, \"onAbort\" as const];\n\n    constructor(\n        private getFile: GetFileUseCase.Interface,\n        private fmSettings: FmGetSettingsUseCase.Interface,\n        private updateFile: UpdateFileUseCase.Interface,\n        private ai: Ai.Interface,\n        private getSettings: GetSettingsUseCase.Interface,\n        private encryption: Encryption.Interface,\n        private websocketService?: WebsocketService.Interface\n    ) {}\n\n    async run({\n        input,\n        controller\n    }: TaskDefinition.RunParams<IAiImageEnrichmentTaskInput>): Promise<\n        TaskDefinition.Result<IAiImageEnrichmentTaskInput>\n    > {\n        if (controller.runtime.isAborted()) {\n            return controller.response.aborted();\n        }\n\n        const fileResult = await this.getFile.execute(input.fileId);\n        if (fileResult.isFail()) {\n            return controller.response.error({\n                message: `File not found: ${input.fileId}`\n            });\n        }\n\n        const file = fileResult.value;\n\n        if (!file.type.startsWith(\"image/\")) {\n            return controller.response.done(\"File is not an image; skipping AI enrichment.\");\n        }\n\n        const settingsResult = await this.fmSettings.execute();\n        const srcPrefix = settingsResult.isOk() ? (settingsResult.value.srcPrefix ?? \"\") : \"\";\n        const imageUrl = `${srcPrefix}${file.key}`;\n\n        const aiSettingsResult = await this.getSettings.execute();\n\n        if (aiSettingsResult.isFail()) {\n            return controller.response.error({\n                message: \"No AI provider configured. Add a provider in AI Power Ups settings.\"\n            });\n        }\n\n        const aiSettings = aiSettingsResult.value;\n\n        const firstProvider = aiSettings.providers.presets[0];\n\n        if (!firstProvider) {\n            return controller.response.done({\n                message: \"No AI provider configured. Add a provider in AI Power Ups settings.\"\n            });\n        }\n\n        let tags: string[] = [];\n        let description = \"\";\n        try {\n            const aiResult = await this.ai.generateText({\n                model: firstProvider.model,\n                output: aiOutputSchema,\n                connection: {\n                    sdkName: firstProvider.model.split(\"/\")[0],\n                    apiKey: await this.encryption.decrypt(firstProvider.apiKeyEncrypted)\n                },\n                messages: [\n                    {\n                        role: \"user\",\n                        content: [\n                            {\n                                type: \"image\",\n                                image: new URL(imageUrl)\n                            },\n                            {\n                                type: \"text\",\n                                text: AI_PROMPT\n                            }\n                        ]\n                    }\n                ]\n            });\n\n            tags = aiResult.output.tags;\n            description = aiResult.output.description;\n        } catch (error) {\n            console.log(\"error\", error.message);\n            return controller.response.error({\n                message: `AI enrichment failed: ${error instanceof Error ? error.message : String(error)}`\n            });\n        }\n\n        const mergedTags = [...new Set([...file.tags, ...tags])];\n\n        const updateResult = await this.updateFile.execute({\n            id: file.id,\n            tags: mergedTags,\n            description\n        });\n\n        if (updateResult.isFail()) {\n            return controller.response.error({\n                message: `Failed to update file: ${updateResult.error.message}`\n            });\n        }\n\n        if (this.websocketService) {\n            const connectionsResult = await this.websocketService.listConnections();\n            if (connectionsResult.isOk() && connectionsResult.value.length > 0) {\n                await this.websocketService.sendToConnections(connectionsResult.value, {\n                    action: \"fm.file.enrichment\",\n                    data: {\n                        id: file.id,\n                        tags: mergedTags,\n                        description\n                    }\n                });\n            }\n        }\n\n        return controller.response.done(\"AI image enrichment completed successfully.\");\n    }\n}\n\nexport const AiImageEnrichmentTask = TaskDefinition.createImplementation({\n    implementation: AiImageEnrichmentTaskImpl,\n    dependencies: [\n        GetFileUseCase,\n        FmGetSettingsUseCase,\n        UpdateFileUseCase,\n        Ai,\n        GetSettingsUseCase,\n        Encryption,\n        [WebsocketService, { optional: true }]\n    ]\n});\n"],"names":["AI_IMAGE_ENRICHMENT_TASK_ID","AI_PROMPT","aiOutputSchema","Output","z","AiImageEnrichmentTaskImpl","getFile","fmSettings","updateFile","ai","getSettings","encryption","websocketService","input","controller","fileResult","file","settingsResult","srcPrefix","imageUrl","aiSettingsResult","aiSettings","firstProvider","tags","description","aiResult","URL","error","console","Error","String","mergedTags","Set","updateResult","connectionsResult","AiImageEnrichmentTask","TaskDefinition","GetFileUseCase","FmGetSettingsUseCase","UpdateFileUseCase","Ai","GetSettingsUseCase","Encryption","WebsocketService"],"mappings":";;;;;;;;;;AAWO,MAAMA,8BAA8B;AAE3C,MAAMC,YACF;AAEJ,MAAMC,iBAAiBC,OAAO,MAAM,CAAC;IACjC,QAAQC,EAAE,MAAM,CAAC;QACb,MAAMA,EAAE,KAAK,CAACA,EAAE,MAAM;QACtB,aAAaA,EAAE,MAAM;IACzB;AACJ;AAMA,MAAMC;IAUF,YACYC,OAAiC,EACjCC,UAA0C,EAC1CC,UAAuC,EACvCC,EAAgB,EAChBC,WAAyC,EACzCC,UAAgC,EAChCC,gBAA6C,CACvD;aAPUN,OAAO,GAAPA;aACAC,UAAU,GAAVA;aACAC,UAAU,GAAVA;aACAC,EAAE,GAAFA;aACAC,WAAW,GAAXA;aACAC,UAAU,GAAVA;aACAC,gBAAgB,GAAhBA;aAhBZ,EAAE,GAAGZ;aACL,KAAK,GAAG;aACR,WAAW,GAAG;aACd,aAAa,GAAG;aAChB,SAAS,GAAG;aACZ,YAAY,GAAG;aAEC,WAAW,GAAG;YAAC;YAAsB;SAAmB;IAUrE;IAEH,MAAM,IAAI,EACNa,KAAK,EACLC,UAAU,EAC0C,EAEtD;QACE,IAAIA,WAAW,OAAO,CAAC,SAAS,IAC5B,OAAOA,WAAW,QAAQ,CAAC,OAAO;QAGtC,MAAMC,aAAa,MAAM,IAAI,CAAC,OAAO,CAAC,OAAO,CAACF,MAAM,MAAM;QAC1D,IAAIE,WAAW,MAAM,IACjB,OAAOD,WAAW,QAAQ,CAAC,KAAK,CAAC;YAC7B,SAAS,CAAC,gBAAgB,EAAED,MAAM,MAAM,EAAE;QAC9C;QAGJ,MAAMG,OAAOD,WAAW,KAAK;QAE7B,IAAI,CAACC,KAAK,IAAI,CAAC,UAAU,CAAC,WACtB,OAAOF,WAAW,QAAQ,CAAC,IAAI,CAAC;QAGpC,MAAMG,iBAAiB,MAAM,IAAI,CAAC,UAAU,CAAC,OAAO;QACpD,MAAMC,YAAYD,eAAe,IAAI,KAAMA,eAAe,KAAK,CAAC,SAAS,IAAI,KAAM;QACnF,MAAME,WAAW,GAAGD,YAAYF,KAAK,GAAG,EAAE;QAE1C,MAAMI,mBAAmB,MAAM,IAAI,CAAC,WAAW,CAAC,OAAO;QAEvD,IAAIA,iBAAiB,MAAM,IACvB,OAAON,WAAW,QAAQ,CAAC,KAAK,CAAC;YAC7B,SAAS;QACb;QAGJ,MAAMO,aAAaD,iBAAiB,KAAK;QAEzC,MAAME,gBAAgBD,WAAW,SAAS,CAAC,OAAO,CAAC,EAAE;QAErD,IAAI,CAACC,eACD,OAAOR,WAAW,QAAQ,CAAC,IAAI,CAAC;YAC5B,SAAS;QACb;QAGJ,IAAIS,OAAiB,EAAE;QACvB,IAAIC,cAAc;QAClB,IAAI;YACA,MAAMC,WAAW,MAAM,IAAI,CAAC,EAAE,CAAC,YAAY,CAAC;gBACxC,OAAOH,cAAc,KAAK;gBAC1B,QAAQpB;gBACR,YAAY;oBACR,SAASoB,cAAc,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE;oBAC1C,QAAQ,MAAM,IAAI,CAAC,UAAU,CAAC,OAAO,CAACA,cAAc,eAAe;gBACvE;gBACA,UAAU;oBACN;wBACI,MAAM;wBACN,SAAS;4BACL;gCACI,MAAM;gCACN,OAAO,IAAII,IAAIP;4BACnB;4BACA;gCACI,MAAM;gCACN,MAAMlB;4BACV;yBACH;oBACL;iBACH;YACL;YAEAsB,OAAOE,SAAS,MAAM,CAAC,IAAI;YAC3BD,cAAcC,SAAS,MAAM,CAAC,WAAW;QAC7C,EAAE,OAAOE,OAAO;YACZC,QAAQ,GAAG,CAAC,SAASD,MAAM,OAAO;YAClC,OAAOb,WAAW,QAAQ,CAAC,KAAK,CAAC;gBAC7B,SAAS,CAAC,sBAAsB,EAAEa,iBAAiBE,QAAQF,MAAM,OAAO,GAAGG,OAAOH,QAAQ;YAC9F;QACJ;QAEA,MAAMI,aAAa;eAAI,IAAIC,IAAI;mBAAIhB,KAAK,IAAI;mBAAKO;aAAK;SAAE;QAExD,MAAMU,eAAe,MAAM,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC;YAC/C,IAAIjB,KAAK,EAAE;YACX,MAAMe;YACNP;QACJ;QAEA,IAAIS,aAAa,MAAM,IACnB,OAAOnB,WAAW,QAAQ,CAAC,KAAK,CAAC;YAC7B,SAAS,CAAC,uBAAuB,EAAEmB,aAAa,KAAK,CAAC,OAAO,EAAE;QACnE;QAGJ,IAAI,IAAI,CAAC,gBAAgB,EAAE;YACvB,MAAMC,oBAAoB,MAAM,IAAI,CAAC,gBAAgB,CAAC,eAAe;YACrE,IAAIA,kBAAkB,IAAI,MAAMA,kBAAkB,KAAK,CAAC,MAAM,GAAG,GAC7D,MAAM,IAAI,CAAC,gBAAgB,CAAC,iBAAiB,CAACA,kBAAkB,KAAK,EAAE;gBACnE,QAAQ;gBACR,MAAM;oBACF,IAAIlB,KAAK,EAAE;oBACX,MAAMe;oBACNP;gBACJ;YACJ;QAER;QAEA,OAAOV,WAAW,QAAQ,CAAC,IAAI,CAAC;IACpC;AACJ;AAEO,MAAMqB,wBAAwBC,eAAe,oBAAoB,CAAC;IACrE,gBAAgB/B;IAChB,cAAc;QACVgC;QACAC;QACAC;QACAC;QACAC;QACAC;QACA;YAACC;YAAkB;gBAAE,UAAU;YAAK;SAAE;KACzC;AACL"}

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

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

package/api/features/AiImageEnrichment/feature.js

@@ -0,0 +1,13 @@
+import { createFeature } from "@webiny/feature/api";
+import { AiImageEnrichmentAfterCreateHandler } from "./AiImageEnrichmentAfterCreateHandler.js";
+import { AiImageEnrichmentTask } from "./AiImageEnrichmentTask.js";
+const AiImageEnrichmentFeature = createFeature({
+    name: "AiPowerUps/AiImageEnrichment",
+    register (container) {
+        container.register(AiImageEnrichmentAfterCreateHandler);
+        container.register(AiImageEnrichmentTask);
+    }
+});
+export { AiImageEnrichmentFeature };
+
+//# sourceMappingURL=feature.js.map

package/api/features/AiImageEnrichment/feature.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"api/features/AiImageEnrichment/feature.js","sources":["../../../../src/api/features/AiImageEnrichment/feature.ts"],"sourcesContent":["import { createFeature } from \"@webiny/feature/api\";\nimport { AiImageEnrichmentAfterCreateHandler } from \"./AiImageEnrichmentAfterCreateHandler.js\";\nimport { AiImageEnrichmentTask } from \"./AiImageEnrichmentTask.js\";\n\nexport const AiImageEnrichmentFeature = createFeature({\n    name: \"AiPowerUps/AiImageEnrichment\",\n    register(container) {\n        container.register(AiImageEnrichmentAfterCreateHandler);\n        container.register(AiImageEnrichmentTask);\n    }\n});\n"],"names":["AiImageEnrichmentFeature","createFeature","container","AiImageEnrichmentAfterCreateHandler","AiImageEnrichmentTask"],"mappings":";;;AAIO,MAAMA,2BAA2BC,cAAc;IAClD,MAAM;IACN,UAASC,SAAS;QACdA,UAAU,QAAQ,CAACC;QACnBD,UAAU,QAAQ,CAACE;IACvB;AACJ"}

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

@@ -0,0 +1,12 @@
+import { GetSettingsUseCase } from "../../../api/features/GetSettings/index.js";
+import { AiPromptContextBuilder as Abstraction, ProjectFileAssembler, type AiPromptContext, type AiPromptContextParams } from "./abstractions.js";
+declare class AiPromptContextBuilderImpl implements Abstraction.Interface {
+    private getSettings;
+    private fileAssembler;
+    constructor(getSettings: GetSettingsUseCase.Interface, fileAssembler: ProjectFileAssembler.Interface);
+    execute(params: AiPromptContextParams): Promise<AiPromptContext>;
+}
+export declare const AiPromptContextBuilder: typeof AiPromptContextBuilderImpl & {
+    __abstraction: import("@webiny/di").Abstraction<import("./abstractions.js").IAiPromptContextBuilder>;
+};
+export {};

package/api/features/AiPromptContext/AiPromptContextBuilder.js

@@ -0,0 +1,103 @@
+import { GetSettingsUseCase } from "../GetSettings/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: [],
+        excludedFileIds: new Set(),
+        cacheHit: false,
+        warnings,
+        toString () {
+            return "";
+        }
+    };
+}
+class AiPromptContextBuilderImpl {
+    constructor(getSettings, fileAssembler){
+        this.getSettings = getSettings;
+        this.fileAssembler = fileAssembler;
+    }
+    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 ctx = {
+            project,
+            readerPersona: reader,
+            writerPersona: writer,
+            allProjectFiles,
+            excludedFileIds,
+            cacheHit,
+            warnings,
+            toString () {
+                return formatContext(this);
+            }
+        };
+        return ctx;
+    }
+}
+const AiPromptContextBuilder_AiPromptContextBuilder = AiPromptContextBuilder.createImplementation({
+    implementation: AiPromptContextBuilderImpl,
+    dependencies: [
+        GetSettingsUseCase,
+        ProjectFileAssembler
+    ]
+});
+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 {\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\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        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    ) {}\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 ctx: AiPromptContext = {\n            project,\n            readerPersona: reader,\n            writerPersona: writer,\n            allProjectFiles,\n            excludedFileIds,\n            cacheHit,\n            warnings,\n            toString() {\n                return formatContext(this);\n            }\n        };\n\n        return ctx;\n    }\n}\n\nexport const AiPromptContextBuilder = Abstraction.createImplementation({\n    implementation: AiPromptContextBuilderImpl,\n    dependencies: [GetSettingsUseCase, ProjectFileAssembler]\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","settingsResult","projectPreset","reader","writer","excludedFileIds","allProjectFiles","cacheHit","assembled","visibleFiles","f","sum","AiPromptContextBuilder","Abstraction","GetSettingsUseCase","ProjectFileAssembler"],"mappings":";;;;;AAgBA,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;IAEA,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,IAAIc;QACrB,UAAU;QACVD;QACA;YACI,OAAO;QACX;IACJ;AACJ;AAEA,MAAME;IACF,YACYC,WAAyC,EACzCC,aAA6C,CACvD;aAFUD,WAAW,GAAXA;aACAC,aAAa,GAAbA;IACT;IAEH,MAAM,QAAQnB,MAA6B,EAA4B;QACnE,MAAMe,WAAqB,EAAE;QAE7B,MAAMK,iBAAiB,MAAM,IAAI,CAAC,WAAW,CAAC,OAAO;QACrD,IAAIA,eAAe,MAAM,IAAI;YACzBL,SAAS,IAAI,CAAC;YACd,OAAOD,aAAaC;QACxB;QAEA,MAAMjB,WAAWsB,eAAe,KAAK;QACrC,MAAMC,gBAAgBrB,OAAO,SAAS,GAChCF,SAAS,QAAQ,EAAE,SAAS,KAAKO,CAAAA,IAAKA,EAAE,EAAE,KAAKL,OAAO,SAAS,IAC/DE;QAEN,MAAM,EAAEoB,MAAM,EAAEC,MAAM,EAAE,GAAG1B,gBAAgBC,UAAUuB,eAAerB;QACpE,MAAMwB,kBAAkB,IAAIR,IAAIhB,OAAO,eAAe,IAAI,EAAE;QAE5D,IAAID;QACJ,IAAI0B,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,EACzBN;YAEJU,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;YAEV1B,UAAU;gBACN,MAAMsB,cAAc,IAAI;gBACxB,cAAcA,cAAc,YAAY;gBACxC,OAAOO;gBACP,aAAaA,aAAa,MAAM,CAAC,CAACE,KAAKD,IAAMC,MAAMD,EAAE,UAAU,EAAE;YACrE;QACJ;QAEA,MAAMrB,MAAuB;YACzBT;YACA,eAAeuB;YACf,eAAeC;YACfE;YACAD;YACAE;YACAX;YACA;gBACI,OAAOR,cAAc,IAAI;YAC7B;QACJ;QAEA,OAAOC;IACX;AACJ;AAEO,MAAMuB,gDAAyBC,uBAAAA,oBAAgC,CAAC;IACnE,gBAAgBf;IAChB,cAAc;QAACgB;QAAoBC;KAAqB;AAC5D"}

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;
+}