@webiny/ai-powerups 0.0.0-unstable.0d717d18dd

package/api/features/AiImageEnrichment/AiImageEnrichmentTask.js

@@ -0,0 +1,142 @@
+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 { GetFileContentsByIdUseCase } from "@webiny/api-file-manager/features/file/GetFileContentsById/abstractions.js";
+import { UpdateFileUseCase } from "@webiny/api-file-manager/features/file/UpdateFile/index.js";
+import { WebsocketsSendToIdentityUseCase } from "@webiny/api-websockets/features/SendToIdentity/abstractions.js";
+import { IdentityContext } from "@webiny/api-core/exports/api/security.js";
+import { 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, getFileContents, updateFile, ai, getSettings, encryption, identityContext, sendToIdentity){
+        this.getFile = getFile;
+        this.getFileContents = getFileContents;
+        this.updateFile = updateFile;
+        this.ai = ai;
+        this.getSettings = getSettings;
+        this.encryption = encryption;
+        this.identityContext = identityContext;
+        this.sendToIdentity = sendToIdentity;
+        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 contentsResult = await this.getFileContents.execute(input.fileId);
+        if (contentsResult.isFail()) return controller.response.error({
+            message: `Unable to read file contents: ${contentsResult.error.message}`
+        });
+        const imageBase64 = contentsResult.value.buffer.toString("base64");
+        const imageMediaType = contentsResult.value.contentType;
+        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: "file",
+                                data: imageBase64,
+                                mediaType: imageMediaType
+                            },
+                            {
+                                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}`
+        });
+        const identity = this.identityContext.getIdentity();
+        await this.sendToIdentity.execute({
+            id: identity.id
+        }, {
+            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,
+        GetFileContentsByIdUseCase,
+        UpdateFileUseCase,
+        Ai,
+        GetSettingsUseCase,
+        Encryption,
+        IdentityContext,
+        WebsocketsSendToIdentityUseCase
+    ]
+});
+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 { GetFileContentsByIdUseCase } from \"@webiny/api-file-manager/features/file/GetFileContentsById/abstractions.js\";\nimport { UpdateFileUseCase } from \"@webiny/api-file-manager/features/file/UpdateFile/index.js\";\nimport { WebsocketsSendToIdentityUseCase } from \"@webiny/api-websockets/features/SendToIdentity/abstractions.js\";\nimport { IdentityContext } from \"@webiny/api-core/exports/api/security.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 getFileContents: GetFileContentsByIdUseCase.Interface,\n        private updateFile: UpdateFileUseCase.Interface,\n        private ai: Ai.Interface,\n        private getSettings: GetSettingsUseCase.Interface,\n        private encryption: Encryption.Interface,\n        private identityContext: IdentityContext.Interface,\n        private sendToIdentity: WebsocketsSendToIdentityUseCase.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        // Read the image bytes and send them to the AI as base64, NOT a URL. A URL forces the\n        // provider to fetch the file — which fails for private/access-controlled files, leaks the\n        // domain, and can't reach a non-public origin (e.g. local dev). base64 is a portable standard\n        // that works on every setup. (Also resolves the AI-SDK \"image part\" deprecation.)\n        const contentsResult = await this.getFileContents.execute(input.fileId);\n        if (contentsResult.isFail()) {\n            return controller.response.error({\n                message: `Unable to read file contents: ${contentsResult.error.message}`\n            });\n        }\n        const imageBase64 = contentsResult.value.buffer.toString(\"base64\");\n        const imageMediaType = contentsResult.value.contentType;\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: \"file\",\n                                data: imageBase64,\n                                mediaType: imageMediaType\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        const identity = this.identityContext.getIdentity();\n        await this.sendToIdentity.execute(\n            { id: identity.id },\n            {\n                action: \"fm.file.enrichment\",\n                data: {\n                    id: file.id,\n                    tags: mergedTags,\n                    description\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        GetFileContentsByIdUseCase,\n        UpdateFileUseCase,\n        Ai,\n        GetSettingsUseCase,\n        Encryption,\n        IdentityContext,\n        WebsocketsSendToIdentityUseCase\n    ]\n});\n"],"names":["AI_IMAGE_ENRICHMENT_TASK_ID","AI_PROMPT","aiOutputSchema","Output","z","AiImageEnrichmentTaskImpl","getFile","getFileContents","updateFile","ai","getSettings","encryption","identityContext","sendToIdentity","input","controller","fileResult","file","contentsResult","imageBase64","imageMediaType","aiSettingsResult","aiSettings","firstProvider","tags","description","aiResult","error","console","Error","String","mergedTags","Set","updateResult","identity","AiImageEnrichmentTask","TaskDefinition","GetFileUseCase","GetFileContentsByIdUseCase","UpdateFileUseCase","Ai","GetSettingsUseCase","Encryption","IdentityContext","WebsocketsSendToIdentityUseCase"],"mappings":";;;;;;;;;;;AAYO,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,eAAqD,EACrDC,UAAuC,EACvCC,EAAgB,EAChBC,WAAyC,EACzCC,UAAgC,EAChCC,eAA0C,EAC1CC,cAAyD,CACnE;aARUP,OAAO,GAAPA;aACAC,eAAe,GAAfA;aACAC,UAAU,GAAVA;aACAC,EAAE,GAAFA;aACAC,WAAW,GAAXA;aACAC,UAAU,GAAVA;aACAC,eAAe,GAAfA;aACAC,cAAc,GAAdA;aAjBZ,EAAE,GAAGb;aACL,KAAK,GAAG;aACR,WAAW,GAAG;aACd,aAAa,GAAG;aAChB,SAAS,GAAG;aACZ,YAAY,GAAG;aAEC,WAAW,GAAG;YAAC;YAAsB;SAAmB;IAWrE;IAEH,MAAM,IAAI,EACNc,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;QAOpC,MAAMG,iBAAiB,MAAM,IAAI,CAAC,eAAe,CAAC,OAAO,CAACJ,MAAM,MAAM;QACtE,IAAII,eAAe,MAAM,IACrB,OAAOH,WAAW,QAAQ,CAAC,KAAK,CAAC;YAC7B,SAAS,CAAC,8BAA8B,EAAEG,eAAe,KAAK,CAAC,OAAO,EAAE;QAC5E;QAEJ,MAAMC,cAAcD,eAAe,KAAK,CAAC,MAAM,CAAC,QAAQ,CAAC;QACzD,MAAME,iBAAiBF,eAAe,KAAK,CAAC,WAAW;QAEvD,MAAMG,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,QAAQrB;gBACR,YAAY;oBACR,SAASqB,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,MAAMJ;gCACN,WAAWC;4BACf;4BACA;gCACI,MAAM;gCACN,MAAMnB;4BACV;yBACH;oBACL;iBACH;YACL;YAEAuB,OAAOE,SAAS,MAAM,CAAC,IAAI;YAC3BD,cAAcC,SAAS,MAAM,CAAC,WAAW;QAC7C,EAAE,OAAOC,OAAO;YACZC,QAAQ,GAAG,CAAC,SAASD,MAAM,OAAO;YAClC,OAAOZ,WAAW,QAAQ,CAAC,KAAK,CAAC;gBAC7B,SAAS,CAAC,sBAAsB,EAAEY,iBAAiBE,QAAQF,MAAM,OAAO,GAAGG,OAAOH,QAAQ;YAC9F;QACJ;QAEA,MAAMI,aAAa;eAAI,IAAIC,IAAI;mBAAIf,KAAK,IAAI;mBAAKO;aAAK;SAAE;QAExD,MAAMS,eAAe,MAAM,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC;YAC/C,IAAIhB,KAAK,EAAE;YACX,MAAMc;YACNN;QACJ;QAEA,IAAIQ,aAAa,MAAM,IACnB,OAAOlB,WAAW,QAAQ,CAAC,KAAK,CAAC;YAC7B,SAAS,CAAC,uBAAuB,EAAEkB,aAAa,KAAK,CAAC,OAAO,EAAE;QACnE;QAGJ,MAAMC,WAAW,IAAI,CAAC,eAAe,CAAC,WAAW;QACjD,MAAM,IAAI,CAAC,cAAc,CAAC,OAAO,CAC7B;YAAE,IAAIA,SAAS,EAAE;QAAC,GAClB;YACI,QAAQ;YACR,MAAM;gBACF,IAAIjB,KAAK,EAAE;gBACX,MAAMc;gBACNN;YACJ;QACJ;QAGJ,OAAOV,WAAW,QAAQ,CAAC,IAAI,CAAC;IACpC;AACJ;AAEO,MAAMoB,wBAAwBC,eAAe,oBAAoB,CAAC;IACrE,gBAAgB/B;IAChB,cAAc;QACVgC;QACAC;QACAC;QACAC;QACAC;QACAC;QACAC;QACAC;KACH;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,19 @@
+import { GetSettingsUseCase } from "../../../api/features/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 as Abstraction, ProjectFileAssembler, type AiPromptContext, type AiPromptContextParams } from "./abstractions.js";
+declare class AiPromptContextBuilderImpl implements Abstraction.Interface {
+    private getSettings;
+    private fileAssembler;
+    private getFileContents;
+    private getFile;
+    private textExtractor;
+    constructor(getSettings: GetSettingsUseCase.Interface, fileAssembler: ProjectFileAssembler.Interface, getFileContents: GetFileContentsByIdUseCase.Interface, getFile: GetFileUseCase.Interface, textExtractor: TextExtractor.Interface);
+    execute(params: AiPromptContextParams): Promise<AiPromptContext>;
+    private fetchAdditionalFiles;
+}
+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,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