@webiny/ai-powerups 6.6.0-alpha.0 → 6.6.0-alpha.2

package/api/features/CmsCompareEntryRevisions/CmsCompareEntryRevisionsUseCase.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"api/features/CmsCompareEntryRevisions/CmsCompareEntryRevisionsUseCase.js","sources":["../../../../src/api/features/CmsCompareEntryRevisions/CmsCompareEntryRevisionsUseCase.ts"],"sourcesContent":["import { Ai } from \"@webiny/api-core/features/ai/index.js\";\nimport { Encryption } from \"@webiny/api-core/features/encryption/index.js\";\nimport { GetModelUseCase } from \"@webiny/api-headless-cms/features/contentModel/GetModel/index.js\";\nimport { ModelToAstConverter } from \"@webiny/api-headless-cms/features/contentModel/ModelToAstConverter/index.js\";\nimport { CmsModelToJsonSchemaConverter } from \"@webiny/api-headless-cms/utils/contentModelToJsonSchema/index.js\";\nimport { GetRevisionByIdUseCase } from \"@webiny/api-headless-cms/features/contentEntry/GetRevisionById/index.js\";\nimport { GetSettingsUseCase } from \"~/api/features/GetSettings/index.js\";\nimport { CmsCompareEntryRevisionsUseCase } from \"./abstractions.js\";\nimport type {\n    ICmsCompareEntryRevisionsParams,\n    ICmsCompareEntryRevisionsResult\n} from \"./abstractions.js\";\n\nconst SYSTEM_PROMPT = `You are a content intelligence assistant specialized in version comparison for headless CMS platforms.\n\nI will provide:\n\n1. The content model JSON Schema that defines the structure of the entry.\n2. Two versions of the content entry values in JSON format: Version A and Version B.\n\nYour task:\n\n- Parse both versions according to the JSON Schema.\n- Identify all differences between Version A and Version B.\n- For each difference, explain:\n    - The field name (use the human-readable label from the schema description if available)\n    - The value in Version A\n    - The value in Version B\n    - A brief summary of the change (e.g., \"Title changed from 'Old' to 'New'\")\n- If nested fields or objects exist, perform a deep comparison.\n- Return the comparison in clean HTML format.\n\nOutput format:\n\n<div class=\"comparison-report\">\n    <table class=\"comparison-table\">\n        <thead>\n            <tr>\n                <th>Field</th>\n                <th>Version A</th>\n                <th>Version B</th>\n                <th>Change Summary</th>\n            </tr>\n        </thead>\n        <tbody>\n            <tr>\n                <td><strong>title</strong></td>\n                <td>Launch Plan</td>\n                <td>Updated Launch Plan</td>\n                <td>Title changed from 'Launch Plan' to 'Updated Launch Plan'</td>\n            </tr>\n        </tbody>\n    </table>\n</div>\n\nIf no differences are found, return: <div class=\"no-changes\"><h3>No differences detected between Version A and Version B.</h3></div>\n\nUse semantic HTML with appropriate CSS classes for styling. Do not include <style> tags or CSS — only return the HTML structure.\nFor rich text or complex nested values, show a concise summary rather than raw JSON.`;\n\nclass CmsCompareEntryRevisionsUseCaseImpl implements CmsCompareEntryRevisionsUseCase.Interface {\n    constructor(\n        private getSettings: GetSettingsUseCase.Interface,\n        private ai: Ai.Interface,\n        private encryption: Encryption.Interface,\n        private getModel: GetModelUseCase.Interface,\n        private getRevisionById: GetRevisionByIdUseCase.Interface,\n        private modelToAst: ModelToAstConverter.Interface\n    ) {}\n\n    async execute(\n        params: ICmsCompareEntryRevisionsParams\n    ): Promise<ICmsCompareEntryRevisionsResult> {\n        const { modelId, revisionId1, revisionId2 } = params;\n\n        const settingsResult = await this.getSettings.execute();\n        if (settingsResult.isFail()) {\n            throw new Error(\"Failed to load AI Power Ups settings.\");\n        }\n\n        const settings = settingsResult.value;\n        const firstProvider = settings.providers.presets[0];\n\n        if (!firstProvider) {\n            throw new Error(\"No AI provider configured. Add a provider in AI Power Ups settings.\");\n        }\n\n        const modelResult = await this.getModel.execute(modelId);\n        if (modelResult.isFail()) {\n            throw new Error(`Content model \"${modelId}\" not found.`);\n        }\n        const model = modelResult.value;\n\n        const [rev1Result, rev2Result] = await Promise.all([\n            this.getRevisionById.execute(model, revisionId1),\n            this.getRevisionById.execute(model, revisionId2)\n        ]);\n\n        if (rev1Result.isFail()) {\n            throw new Error(`Revision \"${revisionId1}\" not found.`);\n        }\n        if (rev2Result.isFail()) {\n            throw new Error(`Revision \"${revisionId2}\" not found.`);\n        }\n\n        const revision1 = rev1Result.value;\n        const revision2 = rev2Result.value;\n\n        const modelAst = this.modelToAst.toAst(model);\n        const jsonSchemaConverter = new CmsModelToJsonSchemaConverter();\n        const entrySchema = jsonSchemaConverter.convert(modelAst, {\n            name: model.name,\n            description: model.description\n        });\n\n        const apiKey = await this.encryption.decrypt(firstProvider.apiKeyEncrypted);\n\n        const userPrompt = `CONTENT MODEL JSON SCHEMA:\n${JSON.stringify(entrySchema, null, 2)}\n\nVERSION A (Revision #${revision1.version}):\n${JSON.stringify(revision1.values, null, 2)}\n\nVERSION B (Revision #${revision2.version}):\n${JSON.stringify(revision2.values, null, 2)}`;\n\n        const result = await this.ai.generateText({\n            model: firstProvider.model,\n            connection: {\n                sdkName: firstProvider.model.split(\"/\")[0],\n                apiKey\n            },\n            system: SYSTEM_PROMPT,\n            prompt: userPrompt,\n            temperature: 0.3\n        });\n\n        const text = result.text;\n\n        const noChangesMatch = text.match(/No differences detected/);\n        let summary = \"Content comparison completed\";\n        if (noChangesMatch) {\n            summary = \"No differences detected between versions\";\n        } else {\n            const headingMatch = text.match(/<h[2-3][^>]*>([^<]+)<\\/h[2-3]>/);\n            const trCount = (text.match(/<tr>/g) || []).length - 1;\n            if (headingMatch) {\n                summary = headingMatch[1].trim();\n            } else if (trCount > 0) {\n                summary = `${trCount} field${trCount === 1 ? \"\" : \"s\"} changed`;\n            }\n        }\n\n        return { html: text, summary };\n    }\n}\n\nexport const CmsCompareEntryRevisionsUseCaseImplementation =\n    CmsCompareEntryRevisionsUseCase.createImplementation({\n        implementation: CmsCompareEntryRevisionsUseCaseImpl,\n        dependencies: [\n            GetSettingsUseCase,\n            Ai,\n            Encryption,\n            GetModelUseCase,\n            GetRevisionByIdUseCase,\n            ModelToAstConverter\n        ]\n    });\n"],"names":["SYSTEM_PROMPT","CmsCompareEntryRevisionsUseCaseImpl","getSettings","ai","encryption","getModel","getRevisionById","modelToAst","params","modelId","revisionId1","revisionId2","settingsResult","Error","settings","firstProvider","modelResult","model","rev1Result","rev2Result","Promise","revision1","revision2","modelAst","jsonSchemaConverter","CmsModelToJsonSchemaConverter","entrySchema","apiKey","userPrompt","JSON","result","text","noChangesMatch","summary","headingMatch","trCount","CmsCompareEntryRevisionsUseCaseImplementation","CmsCompareEntryRevisionsUseCase","GetSettingsUseCase","Ai","Encryption","GetModelUseCase","GetRevisionByIdUseCase","ModelToAstConverter"],"mappings":";;;;;;;;AAaA,MAAMA,gBAAgB,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;oFA6C6D,CAAC;AAErF,MAAMC;IACF,YACYC,WAAyC,EACzCC,EAAgB,EAChBC,UAAgC,EAChCC,QAAmC,EACnCC,eAAiD,EACjDC,UAAyC,CACnD;aANUL,WAAW,GAAXA;aACAC,EAAE,GAAFA;aACAC,UAAU,GAAVA;aACAC,QAAQ,GAARA;aACAC,eAAe,GAAfA;aACAC,UAAU,GAAVA;IACT;IAEH,MAAM,QACFC,MAAuC,EACC;QACxC,MAAM,EAAEC,OAAO,EAAEC,WAAW,EAAEC,WAAW,EAAE,GAAGH;QAE9C,MAAMI,iBAAiB,MAAM,IAAI,CAAC,WAAW,CAAC,OAAO;QACrD,IAAIA,eAAe,MAAM,IACrB,MAAM,IAAIC,MAAM;QAGpB,MAAMC,WAAWF,eAAe,KAAK;QACrC,MAAMG,gBAAgBD,SAAS,SAAS,CAAC,OAAO,CAAC,EAAE;QAEnD,IAAI,CAACC,eACD,MAAM,IAAIF,MAAM;QAGpB,MAAMG,cAAc,MAAM,IAAI,CAAC,QAAQ,CAAC,OAAO,CAACP;QAChD,IAAIO,YAAY,MAAM,IAClB,MAAM,IAAIH,MAAM,CAAC,eAAe,EAAEJ,QAAQ,YAAY,CAAC;QAE3D,MAAMQ,QAAQD,YAAY,KAAK;QAE/B,MAAM,CAACE,YAAYC,WAAW,GAAG,MAAMC,QAAQ,GAAG,CAAC;YAC/C,IAAI,CAAC,eAAe,CAAC,OAAO,CAACH,OAAOP;YACpC,IAAI,CAAC,eAAe,CAAC,OAAO,CAACO,OAAON;SACvC;QAED,IAAIO,WAAW,MAAM,IACjB,MAAM,IAAIL,MAAM,CAAC,UAAU,EAAEH,YAAY,YAAY,CAAC;QAE1D,IAAIS,WAAW,MAAM,IACjB,MAAM,IAAIN,MAAM,CAAC,UAAU,EAAEF,YAAY,YAAY,CAAC;QAG1D,MAAMU,YAAYH,WAAW,KAAK;QAClC,MAAMI,YAAYH,WAAW,KAAK;QAElC,MAAMI,WAAW,IAAI,CAAC,UAAU,CAAC,KAAK,CAACN;QACvC,MAAMO,sBAAsB,IAAIC;QAChC,MAAMC,cAAcF,oBAAoB,OAAO,CAACD,UAAU;YACtD,MAAMN,MAAM,IAAI;YAChB,aAAaA,MAAM,WAAW;QAClC;QAEA,MAAMU,SAAS,MAAM,IAAI,CAAC,UAAU,CAAC,OAAO,CAACZ,cAAc,eAAe;QAE1E,MAAMa,aAAa,CAAC;AAC5B,EAAEC,KAAK,SAAS,CAACH,aAAa,MAAM,GAAG;;qBAElB,EAAEL,UAAU,OAAO,CAAC;AACzC,EAAEQ,KAAK,SAAS,CAACR,UAAU,MAAM,EAAE,MAAM,GAAG;;qBAEvB,EAAEC,UAAU,OAAO,CAAC;AACzC,EAAEO,KAAK,SAAS,CAACP,UAAU,MAAM,EAAE,MAAM,IAAI;QAErC,MAAMQ,SAAS,MAAM,IAAI,CAAC,EAAE,CAAC,YAAY,CAAC;YACtC,OAAOf,cAAc,KAAK;YAC1B,YAAY;gBACR,SAASA,cAAc,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE;gBAC1CY;YACJ;YACA,QAAQ3B;YACR,QAAQ4B;YACR,aAAa;QACjB;QAEA,MAAMG,OAAOD,OAAO,IAAI;QAExB,MAAME,iBAAiBD,KAAK,KAAK,CAAC;QAClC,IAAIE,UAAU;QACd,IAAID,gBACAC,UAAU;aACP;YACH,MAAMC,eAAeH,KAAK,KAAK,CAAC;YAChC,MAAMI,UAAWJ,AAAAA,CAAAA,KAAK,KAAK,CAAC,YAAY,EAAC,EAAG,MAAM,GAAG;YACrD,IAAIG,cACAD,UAAUC,YAAY,CAAC,EAAE,CAAC,IAAI;iBAC3B,IAAIC,UAAU,GACjBF,UAAU,GAAGE,QAAQ,MAAM,EAAEA,AAAY,MAAZA,UAAgB,KAAK,IAAI,QAAQ,CAAC;QAEvE;QAEA,OAAO;YAAE,MAAMJ;YAAME;QAAQ;IACjC;AACJ;AAEO,MAAMG,gDACTC,gCAAgC,oBAAoB,CAAC;IACjD,gBAAgBpC;IAChB,cAAc;QACVqC;QACAC;QACAC;QACAC;QACAC;QACAC;KACH;AACL"}

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

@@ -0,0 +1,18 @@
+export interface ICmsCompareEntryRevisionsResult {
+    html: string;
+    summary: string;
+}
+export interface ICmsCompareEntryRevisionsParams {
+    modelId: string;
+    revisionId1: string;
+    revisionId2: string;
+}
+export interface ICmsCompareEntryRevisionsUseCase {
+    execute(params: ICmsCompareEntryRevisionsParams): Promise<ICmsCompareEntryRevisionsResult>;
+}
+export declare const CmsCompareEntryRevisionsUseCase: import("@webiny/di").Abstraction<ICmsCompareEntryRevisionsUseCase>;
+export declare namespace CmsCompareEntryRevisionsUseCase {
+    type Interface = ICmsCompareEntryRevisionsUseCase;
+    type Params = ICmsCompareEntryRevisionsParams;
+    type Result = ICmsCompareEntryRevisionsResult;
+}

package/api/features/CmsCompareEntryRevisions/abstractions.js

@@ -0,0 +1,5 @@
+import { createAbstraction } from "@webiny/feature/api";
+const CmsCompareEntryRevisionsUseCase = createAbstraction("AiPowerUpsCmsCompareEntryRevisionsUseCase");
+export { CmsCompareEntryRevisionsUseCase };
+
+//# sourceMappingURL=abstractions.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"api/features/CmsCompareEntryRevisions/abstractions.js","sources":["../../../../src/api/features/CmsCompareEntryRevisions/abstractions.ts"],"sourcesContent":["import { createAbstraction } from \"@webiny/feature/api\";\n\nexport interface ICmsCompareEntryRevisionsResult {\n    html: string;\n    summary: string;\n}\n\nexport interface ICmsCompareEntryRevisionsParams {\n    modelId: string;\n    revisionId1: string;\n    revisionId2: string;\n}\n\nexport interface ICmsCompareEntryRevisionsUseCase {\n    execute(params: ICmsCompareEntryRevisionsParams): Promise<ICmsCompareEntryRevisionsResult>;\n}\n\nexport const CmsCompareEntryRevisionsUseCase = createAbstraction<ICmsCompareEntryRevisionsUseCase>(\n    \"AiPowerUpsCmsCompareEntryRevisionsUseCase\"\n);\n\nexport namespace CmsCompareEntryRevisionsUseCase {\n    export type Interface = ICmsCompareEntryRevisionsUseCase;\n    export type Params = ICmsCompareEntryRevisionsParams;\n    export type Result = ICmsCompareEntryRevisionsResult;\n}\n"],"names":["CmsCompareEntryRevisionsUseCase","createAbstraction"],"mappings":";AAiBO,MAAMA,kCAAkCC,kBAC3C"}

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

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

package/api/features/CmsCompareEntryRevisions/feature.js

@@ -0,0 +1,11 @@
+import { createFeature } from "@webiny/feature/api";
+import { CmsCompareEntryRevisionsUseCaseImplementation } from "./CmsCompareEntryRevisionsUseCase.js";
+const CmsCompareEntryRevisionsFeature = createFeature({
+    name: "AiPowerUps/CmsCompareEntryRevisions",
+    register (container) {
+        container.register(CmsCompareEntryRevisionsUseCaseImplementation);
+    }
+});
+export { CmsCompareEntryRevisionsFeature };
+
+//# sourceMappingURL=feature.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"api/features/CmsCompareEntryRevisions/feature.js","sources":["../../../../src/api/features/CmsCompareEntryRevisions/feature.ts"],"sourcesContent":["import { createFeature } from \"@webiny/feature/api\";\nimport { CmsCompareEntryRevisionsUseCaseImplementation } from \"./CmsCompareEntryRevisionsUseCase.js\";\n\nexport const CmsCompareEntryRevisionsFeature = createFeature({\n    name: \"AiPowerUps/CmsCompareEntryRevisions\",\n    register(container) {\n        container.register(CmsCompareEntryRevisionsUseCaseImplementation);\n    }\n});\n"],"names":["CmsCompareEntryRevisionsFeature","createFeature","container","CmsCompareEntryRevisionsUseCaseImplementation"],"mappings":";;AAGO,MAAMA,kCAAkCC,cAAc;IACzD,MAAM;IACN,UAASC,SAAS;QACdA,UAAU,QAAQ,CAACC;IACvB;AACJ"}

package/api/features/CmsCompareEntryRevisions/index.d.ts

@@ -0,0 +1,2 @@
+export { CmsCompareEntryRevisionsFeature } from "./feature.js";
+export { CmsCompareEntryRevisionsUseCase } from "./abstractions.js";

package/api/features/CmsCompareEntryRevisions/index.js

@@ -0,0 +1,2 @@
+export { CmsCompareEntryRevisionsFeature } from "./feature.js";
+export { CmsCompareEntryRevisionsUseCase } from "./abstractions.js";

package/api/features/CmsGenerateEntryContent/CmsGenerateEntryContentTask.d.ts

@@ -10,6 +10,7 @@ export interface ICmsGenerateEntryContentTaskInput {
     excludedFileIds?: string[] | null;
     readerPersonaId?: string | null;
     writerPersonaId?: string | null;
+    additionalFileIds?: string[] | null;
 }
 declare class CmsGenerateEntryContentTaskImpl implements TaskDefinition.Interface<ICmsGenerateEntryContentTaskInput> {
     private identityContext;

package/api/features/CmsGenerateEntryContent/CmsGenerateEntryContentTask.js

@@ -1,6 +1,6 @@
 import { TaskDefinition } from "@webiny/api-core/features/task/TaskDefinition/index.js";
 import { WebsocketsSendToIdentityUseCase } from "@webiny/api-websockets/features/SendToIdentity/abstractions.js";
-import { compress } from "@webiny/utils/features/compression/legacy/gzip.js";
+import { compressJson } from "@webiny/utils/features/compression/legacy/gzip.js";
 import { CmsGenerateEntryContentUseCase } from "./index.js";
 import { IdentityContext } from "@webiny/api-core/exports/api/security.js";
 const CMS_GENERATE_ENTRY_CONTENT_TASK_ID = "aiPowerUpsGenerateEntryContent";
@@ -18,22 +18,32 @@ class CmsGenerateEntryContentTaskImpl {
     }
     async run({ input, controller }) {
         if (controller.runtime.isAborted()) return controller.response.aborted();
-        const result = await this.generateEntryContent.execute({
-            prompt: input.prompt,
-            modelId: input.modelId,
-            projectId: input.projectId,
-            excludedFileIds: input.excludedFileIds,
-            readerPersonaId: input.readerPersonaId,
-            writerPersonaId: input.writerPersonaId
-        });
         const identity = this.identityContext.getIdentity();
+        let result;
+        try {
+            result = await this.generateEntryContent.execute({
+                prompt: input.prompt,
+                modelId: input.modelId,
+                projectId: input.projectId,
+                excludedFileIds: input.excludedFileIds,
+                readerPersonaId: input.readerPersonaId,
+                writerPersonaId: input.writerPersonaId,
+                additionalFileIds: input.additionalFileIds
+            });
+        } catch (err) {
+            const message = err instanceof Error ? err.message : String(err);
+            await this.sendErrorToUser(identity.id, message);
+            return controller.response.error({
+                message
+            });
+        }
         if (result.isFail()) {
             await this.sendErrorToUser(identity.id, result.error.message);
             return controller.response.error({
                 message: result.error.message
             });
         }
-        const compressed = await compress(result.value.output);
+        const compressed = await compressJson(result.value.values);
         const payload = compressed.toString("base64");
         await this.sendContentToUser(identity.id, payload, result.value.telemetry);
         return controller.response.done("Entry content generated successfully.");

package/api/features/CmsGenerateEntryContent/CmsGenerateEntryContentTask.js.map

@@ -1 +1 @@
-{"version":3,"file":"api/features/CmsGenerateEntryContent/CmsGenerateEntryContentTask.js","sources":["../../../../src/api/features/CmsGenerateEntryContent/CmsGenerateEntryContentTask.ts"],"sourcesContent":["import { TaskDefinition } from \"@webiny/api-core/features/task/TaskDefinition/index.js\";\nimport { WebsocketsSendToIdentityUseCase } from \"@webiny/api-websockets/features/SendToIdentity/abstractions.js\";\nimport { compress } from \"@webiny/utils/features/compression/legacy/gzip.js\";\nimport { CmsGenerateEntryContentUseCase } from \"~/api/features/CmsGenerateEntryContent/index.js\";\nimport type { GenerateEntryContentTelemetry } from \"~/api/features/CmsGenerateEntryContent/abstractions.js\";\nimport { IdentityContext } from \"@webiny/api-core/exports/api/security.js\";\n\nexport const CMS_GENERATE_ENTRY_CONTENT_TASK_ID = \"aiPowerUpsGenerateEntryContent\";\n\nexport interface ICmsGenerateEntryContentTaskInput {\n    prompt: string;\n    modelId: string;\n    projectId?: string | null;\n    excludedFileIds?: string[] | null;\n    readerPersonaId?: string | null;\n    writerPersonaId?: string | null;\n}\n\nclass CmsGenerateEntryContentTaskImpl implements TaskDefinition.Interface<ICmsGenerateEntryContentTaskInput> {\n    id = CMS_GENERATE_ENTRY_CONTENT_TASK_ID;\n    title = \"AI Power-Ups - Generate Entry Content\";\n    description = \"Generates CMS entry content using AI based on a user prompt.\";\n    maxIterations = 1;\n    isPrivate = true;\n    databaseLogs = false;\n\n    constructor(\n        private identityContext: IdentityContext.Interface,\n        private generateEntryContent: CmsGenerateEntryContentUseCase.Interface,\n        private sendToIdentity: WebsocketsSendToIdentityUseCase.Interface\n    ) {}\n\n    async run({\n        input,\n        controller\n    }: TaskDefinition.RunParams<ICmsGenerateEntryContentTaskInput>): Promise<\n        TaskDefinition.Result<ICmsGenerateEntryContentTaskInput>\n    > {\n        if (controller.runtime.isAborted()) {\n            return controller.response.aborted();\n        }\n\n        const result = await this.generateEntryContent.execute({\n            prompt: input.prompt,\n            modelId: input.modelId,\n            projectId: input.projectId,\n            excludedFileIds: input.excludedFileIds,\n            readerPersonaId: input.readerPersonaId,\n            writerPersonaId: input.writerPersonaId\n        });\n\n        const identity = this.identityContext.getIdentity();\n\n        if (result.isFail()) {\n            await this.sendErrorToUser(identity.id, result.error.message);\n\n            return controller.response.error({\n                message: result.error.message\n            });\n        }\n\n        const compressed = await compress(result.value.output);\n        const payload = compressed.toString(\"base64\");\n\n        await this.sendContentToUser(identity.id, payload, result.value.telemetry);\n\n        return controller.response.done(\"Entry content generated successfully.\");\n    }\n\n    private async sendContentToUser(\n        identityId: string,\n        payload: string,\n        telemetry: GenerateEntryContentTelemetry\n    ) {\n        await this.sendToIdentity.execute(\n            { id: identityId },\n            {\n                action: \"aiPowerUps.generateEntryContent.content\",\n                data: {\n                    compression: \"gzip\",\n                    value: payload,\n                    telemetry\n                }\n            }\n        );\n    }\n\n    private async sendErrorToUser(identityId: string, message: string) {\n        await this.sendToIdentity.execute(\n            { id: identityId },\n            {\n                action: \"aiPowerUps.generateEntryContent.error\",\n                data: {\n                    message\n                }\n            }\n        );\n    }\n}\n\nexport const CmsGenerateEntryContentTask = TaskDefinition.createImplementation({\n    implementation: CmsGenerateEntryContentTaskImpl,\n    dependencies: [IdentityContext, CmsGenerateEntryContentUseCase, WebsocketsSendToIdentityUseCase]\n});\n"],"names":["CMS_GENERATE_ENTRY_CONTENT_TASK_ID","CmsGenerateEntryContentTaskImpl","identityContext","generateEntryContent","sendToIdentity","input","controller","result","identity","compressed","compress","payload","identityId","telemetry","message","CmsGenerateEntryContentTask","TaskDefinition","IdentityContext","CmsGenerateEntryContentUseCase","WebsocketsSendToIdentityUseCase"],"mappings":";;;;;AAOO,MAAMA,qCAAqC;AAWlD,MAAMC;IAQF,YACYC,eAA0C,EAC1CC,oBAA8D,EAC9DC,cAAyD,CACnE;aAHUF,eAAe,GAAfA;aACAC,oBAAoB,GAApBA;aACAC,cAAc,GAAdA;aAVZ,EAAE,GAAGJ;aACL,KAAK,GAAG;aACR,WAAW,GAAG;aACd,aAAa,GAAG;aAChB,SAAS,GAAG;aACZ,YAAY,GAAG;IAMZ;IAEH,MAAM,IAAI,EACNK,KAAK,EACLC,UAAU,EACgD,EAE5D;QACE,IAAIA,WAAW,OAAO,CAAC,SAAS,IAC5B,OAAOA,WAAW,QAAQ,CAAC,OAAO;QAGtC,MAAMC,SAAS,MAAM,IAAI,CAAC,oBAAoB,CAAC,OAAO,CAAC;YACnD,QAAQF,MAAM,MAAM;YACpB,SAASA,MAAM,OAAO;YACtB,WAAWA,MAAM,SAAS;YAC1B,iBAAiBA,MAAM,eAAe;YACtC,iBAAiBA,MAAM,eAAe;YACtC,iBAAiBA,MAAM,eAAe;QAC1C;QAEA,MAAMG,WAAW,IAAI,CAAC,eAAe,CAAC,WAAW;QAEjD,IAAID,OAAO,MAAM,IAAI;YACjB,MAAM,IAAI,CAAC,eAAe,CAACC,SAAS,EAAE,EAAED,OAAO,KAAK,CAAC,OAAO;YAE5D,OAAOD,WAAW,QAAQ,CAAC,KAAK,CAAC;gBAC7B,SAASC,OAAO,KAAK,CAAC,OAAO;YACjC;QACJ;QAEA,MAAME,aAAa,MAAMC,SAASH,OAAO,KAAK,CAAC,MAAM;QACrD,MAAMI,UAAUF,WAAW,QAAQ,CAAC;QAEpC,MAAM,IAAI,CAAC,iBAAiB,CAACD,SAAS,EAAE,EAAEG,SAASJ,OAAO,KAAK,CAAC,SAAS;QAEzE,OAAOD,WAAW,QAAQ,CAAC,IAAI,CAAC;IACpC;IAEA,MAAc,kBACVM,UAAkB,EAClBD,OAAe,EACfE,SAAwC,EAC1C;QACE,MAAM,IAAI,CAAC,cAAc,CAAC,OAAO,CAC7B;YAAE,IAAID;QAAW,GACjB;YACI,QAAQ;YACR,MAAM;gBACF,aAAa;gBACb,OAAOD;gBACPE;YACJ;QACJ;IAER;IAEA,MAAc,gBAAgBD,UAAkB,EAAEE,OAAe,EAAE;QAC/D,MAAM,IAAI,CAAC,cAAc,CAAC,OAAO,CAC7B;YAAE,IAAIF;QAAW,GACjB;YACI,QAAQ;YACR,MAAM;gBACFE;YACJ;QACJ;IAER;AACJ;AAEO,MAAMC,8BAA8BC,eAAe,oBAAoB,CAAC;IAC3E,gBAAgBf;IAChB,cAAc;QAACgB;QAAiBC;QAAgCC;KAAgC;AACpG"}
+{"version":3,"file":"api/features/CmsGenerateEntryContent/CmsGenerateEntryContentTask.js","sources":["../../../../src/api/features/CmsGenerateEntryContent/CmsGenerateEntryContentTask.ts"],"sourcesContent":["import { TaskDefinition } from \"@webiny/api-core/features/task/TaskDefinition/index.js\";\nimport { WebsocketsSendToIdentityUseCase } from \"@webiny/api-websockets/features/SendToIdentity/abstractions.js\";\nimport { compressJson } from \"@webiny/utils/features/compression/legacy/gzip.js\";\nimport { CmsGenerateEntryContentUseCase } from \"~/api/features/CmsGenerateEntryContent/index.js\";\nimport type { GenerateEntryContentTelemetry } from \"~/api/features/CmsGenerateEntryContent/abstractions.js\";\nimport { IdentityContext } from \"@webiny/api-core/exports/api/security.js\";\n\nexport const CMS_GENERATE_ENTRY_CONTENT_TASK_ID = \"aiPowerUpsGenerateEntryContent\";\n\nexport interface ICmsGenerateEntryContentTaskInput {\n    prompt: string;\n    modelId: string;\n    projectId?: string | null;\n    excludedFileIds?: string[] | null;\n    readerPersonaId?: string | null;\n    writerPersonaId?: string | null;\n    additionalFileIds?: string[] | null;\n}\n\nclass CmsGenerateEntryContentTaskImpl implements TaskDefinition.Interface<ICmsGenerateEntryContentTaskInput> {\n    id = CMS_GENERATE_ENTRY_CONTENT_TASK_ID;\n    title = \"AI Power-Ups - Generate Entry Content\";\n    description = \"Generates CMS entry content using AI based on a user prompt.\";\n    maxIterations = 1;\n    isPrivate = true;\n    databaseLogs = false;\n\n    constructor(\n        private identityContext: IdentityContext.Interface,\n        private generateEntryContent: CmsGenerateEntryContentUseCase.Interface,\n        private sendToIdentity: WebsocketsSendToIdentityUseCase.Interface\n    ) {}\n\n    async run({\n        input,\n        controller\n    }: TaskDefinition.RunParams<ICmsGenerateEntryContentTaskInput>): Promise<\n        TaskDefinition.Result<ICmsGenerateEntryContentTaskInput>\n    > {\n        if (controller.runtime.isAborted()) {\n            return controller.response.aborted();\n        }\n\n        const identity = this.identityContext.getIdentity();\n\n        let result;\n        try {\n            result = await this.generateEntryContent.execute({\n                prompt: input.prompt,\n                modelId: input.modelId,\n                projectId: input.projectId,\n                excludedFileIds: input.excludedFileIds,\n                readerPersonaId: input.readerPersonaId,\n                writerPersonaId: input.writerPersonaId,\n                additionalFileIds: input.additionalFileIds\n            });\n        } catch (err) {\n            const message = err instanceof Error ? err.message : String(err);\n            await this.sendErrorToUser(identity.id, message);\n            return controller.response.error({ message });\n        }\n\n        if (result.isFail()) {\n            await this.sendErrorToUser(identity.id, result.error.message);\n\n            return controller.response.error({\n                message: result.error.message\n            });\n        }\n\n        // Serialize at the transport edge: the use case returns structured values; the\n        // websocket stream carries a gzip+base64 JSON string that the Admin app decodes.\n        const compressed = await compressJson(result.value.values);\n        const payload = compressed.toString(\"base64\");\n\n        await this.sendContentToUser(identity.id, payload, result.value.telemetry);\n\n        return controller.response.done(\"Entry content generated successfully.\");\n    }\n\n    private async sendContentToUser(\n        identityId: string,\n        payload: string,\n        telemetry: GenerateEntryContentTelemetry\n    ) {\n        await this.sendToIdentity.execute(\n            { id: identityId },\n            {\n                action: \"aiPowerUps.generateEntryContent.content\",\n                data: {\n                    compression: \"gzip\",\n                    value: payload,\n                    telemetry\n                }\n            }\n        );\n    }\n\n    private async sendErrorToUser(identityId: string, message: string) {\n        await this.sendToIdentity.execute(\n            { id: identityId },\n            {\n                action: \"aiPowerUps.generateEntryContent.error\",\n                data: {\n                    message\n                }\n            }\n        );\n    }\n}\n\nexport const CmsGenerateEntryContentTask = TaskDefinition.createImplementation({\n    implementation: CmsGenerateEntryContentTaskImpl,\n    dependencies: [IdentityContext, CmsGenerateEntryContentUseCase, WebsocketsSendToIdentityUseCase]\n});\n"],"names":["CMS_GENERATE_ENTRY_CONTENT_TASK_ID","CmsGenerateEntryContentTaskImpl","identityContext","generateEntryContent","sendToIdentity","input","controller","identity","result","err","message","Error","String","compressed","compressJson","payload","identityId","telemetry","CmsGenerateEntryContentTask","TaskDefinition","IdentityContext","CmsGenerateEntryContentUseCase","WebsocketsSendToIdentityUseCase"],"mappings":";;;;;AAOO,MAAMA,qCAAqC;AAYlD,MAAMC;IAQF,YACYC,eAA0C,EAC1CC,oBAA8D,EAC9DC,cAAyD,CACnE;aAHUF,eAAe,GAAfA;aACAC,oBAAoB,GAApBA;aACAC,cAAc,GAAdA;aAVZ,EAAE,GAAGJ;aACL,KAAK,GAAG;aACR,WAAW,GAAG;aACd,aAAa,GAAG;aAChB,SAAS,GAAG;aACZ,YAAY,GAAG;IAMZ;IAEH,MAAM,IAAI,EACNK,KAAK,EACLC,UAAU,EACgD,EAE5D;QACE,IAAIA,WAAW,OAAO,CAAC,SAAS,IAC5B,OAAOA,WAAW,QAAQ,CAAC,OAAO;QAGtC,MAAMC,WAAW,IAAI,CAAC,eAAe,CAAC,WAAW;QAEjD,IAAIC;QACJ,IAAI;YACAA,SAAS,MAAM,IAAI,CAAC,oBAAoB,CAAC,OAAO,CAAC;gBAC7C,QAAQH,MAAM,MAAM;gBACpB,SAASA,MAAM,OAAO;gBACtB,WAAWA,MAAM,SAAS;gBAC1B,iBAAiBA,MAAM,eAAe;gBACtC,iBAAiBA,MAAM,eAAe;gBACtC,iBAAiBA,MAAM,eAAe;gBACtC,mBAAmBA,MAAM,iBAAiB;YAC9C;QACJ,EAAE,OAAOI,KAAK;YACV,MAAMC,UAAUD,eAAeE,QAAQF,IAAI,OAAO,GAAGG,OAAOH;YAC5D,MAAM,IAAI,CAAC,eAAe,CAACF,SAAS,EAAE,EAAEG;YACxC,OAAOJ,WAAW,QAAQ,CAAC,KAAK,CAAC;gBAAEI;YAAQ;QAC/C;QAEA,IAAIF,OAAO,MAAM,IAAI;YACjB,MAAM,IAAI,CAAC,eAAe,CAACD,SAAS,EAAE,EAAEC,OAAO,KAAK,CAAC,OAAO;YAE5D,OAAOF,WAAW,QAAQ,CAAC,KAAK,CAAC;gBAC7B,SAASE,OAAO,KAAK,CAAC,OAAO;YACjC;QACJ;QAIA,MAAMK,aAAa,MAAMC,aAAaN,OAAO,KAAK,CAAC,MAAM;QACzD,MAAMO,UAAUF,WAAW,QAAQ,CAAC;QAEpC,MAAM,IAAI,CAAC,iBAAiB,CAACN,SAAS,EAAE,EAAEQ,SAASP,OAAO,KAAK,CAAC,SAAS;QAEzE,OAAOF,WAAW,QAAQ,CAAC,IAAI,CAAC;IACpC;IAEA,MAAc,kBACVU,UAAkB,EAClBD,OAAe,EACfE,SAAwC,EAC1C;QACE,MAAM,IAAI,CAAC,cAAc,CAAC,OAAO,CAC7B;YAAE,IAAID;QAAW,GACjB;YACI,QAAQ;YACR,MAAM;gBACF,aAAa;gBACb,OAAOD;gBACPE;YACJ;QACJ;IAER;IAEA,MAAc,gBAAgBD,UAAkB,EAAEN,OAAe,EAAE;QAC/D,MAAM,IAAI,CAAC,cAAc,CAAC,OAAO,CAC7B;YAAE,IAAIM;QAAW,GACjB;YACI,QAAQ;YACR,MAAM;gBACFN;YACJ;QACJ;IAER;AACJ;AAEO,MAAMQ,8BAA8BC,eAAe,oBAAoB,CAAC;IAC3E,gBAAgBlB;IAChB,cAAc;QAACmB;QAAiBC;QAAgCC;KAAgC;AACpG"}

package/api/features/CmsGenerateEntryContent/CmsGenerateEntryContentUseCase.d.ts

@@ -21,7 +21,7 @@ declare class CmsGenerateEntryContentUseCaseImpl implements CmsGenerateEntryCont
     private modelToAst;
     private toolPipelineRunner;
     constructor(promptContextBuilder: AiPromptContextBuilder.Interface, getSettings: GetSettingsUseCase.Interface, ai: Ai.Interface, aiSdkTools: AiSdkTools.Interface, encryption: Encryption.Interface, listTags: ListTagsUseCase.Interface, getModel: GetModelUseCase.Interface, modelToAst: ModelToAstConverter.Interface, toolPipelineRunner: AiToolPipelineRunner.Interface);
-    execute(params: CmsGenerateEntryContentParams): Promise<Result<GenerateEntryContentResult, Error>>;
+    execute<TValues = Record<string, any>>(params: CmsGenerateEntryContentParams): Promise<Result<GenerateEntryContentResult<TValues>, Error>>;
 }
 export declare const CmsGenerateEntryContentUseCaseImplementation: typeof CmsGenerateEntryContentUseCaseImpl & {
     __abstraction: import("@webiny/di").Abstraction<import("./abstractions.js").ICmsGenerateEntryContentUseCase>;

package/api/features/CmsGenerateEntryContent/CmsGenerateEntryContentUseCase.js

@@ -7,7 +7,7 @@ import { GetModelUseCase } from "@webiny/api-headless-cms/features/contentModel/
 import { ModelToAstConverter } from "@webiny/api-headless-cms/features/contentModel/ModelToAstConverter/index.js";
 import { CmsModelToJsonSchemaConverter } from "@webiny/api-headless-cms/utils/contentModelToJsonSchema/index.js";
 import { GetSettingsUseCase } from "../GetSettings/index.js";
-import { AiPromptContextBuilder } from "../AiPromptContext/index.js";
+import { AiPromptContextBuilder, formatAdditionalFilesContext } from "../AiPromptContext/index.js";
 import { createReadProjectFileTool } from "../AiPromptContext/ReadProjectFileTool.js";
 import { CmsGenerateEntryContentUseCase } from "./abstractions.js";
 import { buildEntryPrompt } from "./buildPrompt.js";
@@ -46,7 +46,8 @@ class CmsGenerateEntryContentUseCaseImpl {
             projectId: params.projectId,
             readerPersonaId: params.readerPersonaId,
             writerPersonaId: params.writerPersonaId,
-            excludedFileIds: params.excludedFileIds
+            excludedFileIds: params.excludedFileIds,
+            additionalFileIds: params.additionalFileIds
         });
         if (context.allProjectFiles.length > 0) {
             const projectFileTool = createReadProjectFileTool(context.allProjectFiles, context.excludedFileIds);
@@ -80,7 +81,7 @@ class CmsGenerateEntryContentUseCaseImpl {
                 },
                 system,
                 toolChoice: "auto",
-                prompt: params.prompt,
+                prompt: params.prompt + formatAdditionalFilesContext(context.additionalFiles),
                 ...Object.keys(sdkTools).length > 0 ? {
                     tools: sdkTools,
                     stopWhen: stepCountIs(20)
@@ -93,7 +94,6 @@ class CmsGenerateEntryContentUseCaseImpl {
                 resolved = await this.toolPipelineRunner.resolve(entry);
                 await injectDynamicZoneTypenames(resolved, modelAst, model.singularApiName);
             }
-            const output = JSON.stringify(resolved);
             const filesRead = new Set();
             let toolCallsMade = 0;
             for (const step of aiResult.steps)for (const call of step.toolCalls){
@@ -114,7 +114,7 @@ class CmsGenerateEntryContentUseCaseImpl {
                 imageTagsInPrompt: imageTags
             };
             return Result.ok({
-                output,
+                values: resolved ?? {},
                 telemetry
             });
         } catch (error) {

package/api/features/CmsGenerateEntryContent/CmsGenerateEntryContentUseCase.js.map

@@ -1 +1 @@
-{"version":3,"file":"api/features/CmsGenerateEntryContent/CmsGenerateEntryContentUseCase.js","sources":["../../../../src/api/features/CmsGenerateEntryContent/CmsGenerateEntryContentUseCase.ts"],"sourcesContent":["import { stepCountIs } from \"ai\";\nimport { Result } from \"@webiny/feature/api\";\nimport { Ai } from \"@webiny/api-core/features/ai/index.js\";\nimport { AiSdkTools } from \"@webiny/api-core/features/ai/index.js\";\nimport { AiToolPipelineRunner } from \"@webiny/api-core/features/ai/index.js\";\nimport { Encryption } from \"@webiny/api-core/features/encryption/index.js\";\nimport { ListTagsUseCase } from \"@webiny/api-file-manager/features/file/ListTags/index.js\";\nimport { GetModelUseCase } from \"@webiny/api-headless-cms/features/contentModel/GetModel/index.js\";\nimport { ModelToAstConverter } from \"@webiny/api-headless-cms/features/contentModel/ModelToAstConverter/index.js\";\nimport { CmsModelToJsonSchemaConverter } from \"@webiny/api-headless-cms/utils/contentModelToJsonSchema/index.js\";\nimport { GetSettingsUseCase } from \"~/api/features/GetSettings/index.js\";\nimport { AiPromptContextBuilder } from \"~/api/features/AiPromptContext/index.js\";\nimport { createReadProjectFileTool } from \"~/api/features/AiPromptContext/ReadProjectFileTool.js\";\nimport { CmsGenerateEntryContentUseCase } from \"./abstractions.js\";\nimport type {\n    CmsGenerateEntryContentParams,\n    GenerateEntryContentResult,\n    GenerateEntryContentTelemetry\n} from \"./abstractions.js\";\nimport { buildEntryPrompt } from \"./buildPrompt.js\";\nimport { LlmJsonResponse } from \"../WbGeneratePageContent/LlmJsonResponse.js\";\nimport { injectDynamicZoneTypenames } from \"./injectDynamicZoneTypenames.js\";\n\nclass CmsGenerateEntryContentUseCaseImpl implements CmsGenerateEntryContentUseCase.Interface {\n    constructor(\n        private promptContextBuilder: AiPromptContextBuilder.Interface,\n        private getSettings: GetSettingsUseCase.Interface,\n        private ai: Ai.Interface,\n        private aiSdkTools: AiSdkTools.Interface,\n        private encryption: Encryption.Interface,\n        private listTags: ListTagsUseCase.Interface,\n        private getModel: GetModelUseCase.Interface,\n        private modelToAst: ModelToAstConverter.Interface,\n        private toolPipelineRunner: AiToolPipelineRunner.Interface\n    ) {}\n\n    async execute(\n        params: CmsGenerateEntryContentParams\n    ): Promise<Result<GenerateEntryContentResult, Error>> {\n        const settingsResult = await this.getSettings.execute();\n        if (settingsResult.isFail()) {\n            return Result.fail(new Error(\"Failed to load AI PowerUps settings.\"));\n        }\n\n        const settings = settingsResult.value;\n        const firstProvider = settings.providers.presets[0];\n\n        if (!firstProvider) {\n            return Result.fail(\n                new Error(\"No AI provider configured. Add a provider in AI Power Ups settings.\")\n            );\n        }\n\n        const modelResult = await this.getModel.execute(params.modelId);\n        if (modelResult.isFail()) {\n            return Result.fail(new Error(`Content model \"${params.modelId}\" not found.`));\n        }\n\n        const model = modelResult.value;\n        const modelAst = this.modelToAst.toAst(model);\n        const jsonSchemaConverter = new CmsModelToJsonSchemaConverter();\n        const entrySchema = jsonSchemaConverter.convert(modelAst, {\n            name: model.name,\n            description: model.description\n        });\n\n        const apiKey = await this.encryption.decrypt(firstProvider.apiKeyEncrypted);\n        const sdkTools = this.aiSdkTools.getToolSet();\n\n        const context = await this.promptContextBuilder.execute({\n            projectId: params.projectId,\n            readerPersonaId: params.readerPersonaId,\n            writerPersonaId: params.writerPersonaId,\n            excludedFileIds: params.excludedFileIds\n        });\n\n        if (context.allProjectFiles.length > 0) {\n            const projectFileTool = createReadProjectFileTool(\n                context.allProjectFiles,\n                context.excludedFileIds\n            );\n            Object.assign(sdkTools, projectFileTool);\n        }\n\n        const tagsResult = await this.listTags.execute({\n            where: { type_startsWith: \"image/\" },\n            limit: 100\n        });\n        const imageTags = tagsResult.isOk() ? tagsResult.value.map(t => t.tag) : [];\n\n        const systemText =\n            buildEntryPrompt(model.name, entrySchema, imageTags) + context.toString();\n\n        const system = {\n            role: \"system\" as const,\n            content: systemText,\n            providerOptions: {\n                anthropic: {\n                    cacheControl: { type: \"ephemeral\" }\n                }\n            }\n        };\n\n        try {\n            const aiResult = await this.ai.generateText({\n                model: firstProvider.model,\n                connection: {\n                    sdkName: firstProvider.model.split(\"/\")[0],\n                    apiKey\n                },\n                system,\n                toolChoice: \"auto\",\n                prompt: params.prompt,\n                ...(Object.keys(sdkTools).length > 0\n                    ? { tools: sdkTools, stopWhen: stepCountIs(20) }\n                    : {})\n            });\n\n            const text =\n                aiResult.text ||\n                (aiResult.steps.filter(step => step.text.length > 0).pop()?.text ?? \"\");\n\n            const entry = LlmJsonResponse.fromRawText(text).toArray().pop();\n\n            let resolved: Record<string, any> | undefined;\n\n            if (entry) {\n                resolved = (await this.toolPipelineRunner.resolve(entry)) as Record<string, any>;\n                await injectDynamicZoneTypenames(resolved, modelAst, model.singularApiName);\n            }\n\n            const output = JSON.stringify(resolved);\n\n            const filesRead = new Set<string>();\n            let toolCallsMade = 0;\n            for (const step of aiResult.steps) {\n                for (const call of step.toolCalls) {\n                    toolCallsMade++;\n                    if (call.toolName === \"read_project_file\") {\n                        const input = call.input as { fileId?: string };\n                        if (input.fileId) {\n                            filesRead.add(input.fileId);\n                        }\n                    }\n                }\n            }\n\n            const telemetry: GenerateEntryContentTelemetry = {\n                filesRead: [...filesRead],\n                cacheHit: context.cacheHit,\n                toolCallsMade,\n                totalSteps: aiResult.steps.length,\n                toolsAvailable: Object.keys(sdkTools),\n                imageTagsInPrompt: imageTags\n            };\n\n            return Result.ok({ output, telemetry });\n        } catch (error) {\n            return Result.fail(\n                new Error(\n                    `AI generation failed: ${error instanceof Error ? error.message : String(error)}`\n                )\n            );\n        }\n    }\n}\n\nexport const CmsGenerateEntryContentUseCaseImplementation =\n    CmsGenerateEntryContentUseCase.createImplementation({\n        implementation: CmsGenerateEntryContentUseCaseImpl,\n        dependencies: [\n            AiPromptContextBuilder,\n            GetSettingsUseCase,\n            Ai,\n            AiSdkTools,\n            Encryption,\n            ListTagsUseCase,\n            GetModelUseCase,\n            ModelToAstConverter,\n            AiToolPipelineRunner\n        ]\n    });\n"],"names":["CmsGenerateEntryContentUseCaseImpl","promptContextBuilder","getSettings","ai","aiSdkTools","encryption","listTags","getModel","modelToAst","toolPipelineRunner","params","settingsResult","Result","Error","settings","firstProvider","modelResult","model","modelAst","jsonSchemaConverter","CmsModelToJsonSchemaConverter","entrySchema","apiKey","sdkTools","context","projectFileTool","createReadProjectFileTool","Object","tagsResult","imageTags","t","systemText","buildEntryPrompt","system","aiResult","stepCountIs","text","step","entry","LlmJsonResponse","resolved","injectDynamicZoneTypenames","output","JSON","filesRead","Set","toolCallsMade","call","input","telemetry","error","String","CmsGenerateEntryContentUseCaseImplementation","CmsGenerateEntryContentUseCase","AiPromptContextBuilder","GetSettingsUseCase","Ai","AiSdkTools","Encryption","ListTagsUseCase","GetModelUseCase","ModelToAstConverter","AiToolPipelineRunner"],"mappings":";;;;;;;;;;;;;;;AAuBA,MAAMA;IACF,YACYC,oBAAsD,EACtDC,WAAyC,EACzCC,EAAgB,EAChBC,UAAgC,EAChCC,UAAgC,EAChCC,QAAmC,EACnCC,QAAmC,EACnCC,UAAyC,EACzCC,kBAAkD,CAC5D;aATUR,oBAAoB,GAApBA;aACAC,WAAW,GAAXA;aACAC,EAAE,GAAFA;aACAC,UAAU,GAAVA;aACAC,UAAU,GAAVA;aACAC,QAAQ,GAARA;aACAC,QAAQ,GAARA;aACAC,UAAU,GAAVA;aACAC,kBAAkB,GAAlBA;IACT;IAEH,MAAM,QACFC,MAAqC,EACa;QAClD,MAAMC,iBAAiB,MAAM,IAAI,CAAC,WAAW,CAAC,OAAO;QACrD,IAAIA,eAAe,MAAM,IACrB,OAAOC,OAAO,IAAI,CAAC,IAAIC,MAAM;QAGjC,MAAMC,WAAWH,eAAe,KAAK;QACrC,MAAMI,gBAAgBD,SAAS,SAAS,CAAC,OAAO,CAAC,EAAE;QAEnD,IAAI,CAACC,eACD,OAAOH,OAAO,IAAI,CACd,IAAIC,MAAM;QAIlB,MAAMG,cAAc,MAAM,IAAI,CAAC,QAAQ,CAAC,OAAO,CAACN,OAAO,OAAO;QAC9D,IAAIM,YAAY,MAAM,IAClB,OAAOJ,OAAO,IAAI,CAAC,IAAIC,MAAM,CAAC,eAAe,EAAEH,OAAO,OAAO,CAAC,YAAY,CAAC;QAG/E,MAAMO,QAAQD,YAAY,KAAK;QAC/B,MAAME,WAAW,IAAI,CAAC,UAAU,CAAC,KAAK,CAACD;QACvC,MAAME,sBAAsB,IAAIC;QAChC,MAAMC,cAAcF,oBAAoB,OAAO,CAACD,UAAU;YACtD,MAAMD,MAAM,IAAI;YAChB,aAAaA,MAAM,WAAW;QAClC;QAEA,MAAMK,SAAS,MAAM,IAAI,CAAC,UAAU,CAAC,OAAO,CAACP,cAAc,eAAe;QAC1E,MAAMQ,WAAW,IAAI,CAAC,UAAU,CAAC,UAAU;QAE3C,MAAMC,UAAU,MAAM,IAAI,CAAC,oBAAoB,CAAC,OAAO,CAAC;YACpD,WAAWd,OAAO,SAAS;YAC3B,iBAAiBA,OAAO,eAAe;YACvC,iBAAiBA,OAAO,eAAe;YACvC,iBAAiBA,OAAO,eAAe;QAC3C;QAEA,IAAIc,QAAQ,eAAe,CAAC,MAAM,GAAG,GAAG;YACpC,MAAMC,kBAAkBC,0BACpBF,QAAQ,eAAe,EACvBA,QAAQ,eAAe;YAE3BG,OAAO,MAAM,CAACJ,UAAUE;QAC5B;QAEA,MAAMG,aAAa,MAAM,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC;YAC3C,OAAO;gBAAE,iBAAiB;YAAS;YACnC,OAAO;QACX;QACA,MAAMC,YAAYD,WAAW,IAAI,KAAKA,WAAW,KAAK,CAAC,GAAG,CAACE,CAAAA,IAAKA,EAAE,GAAG,IAAI,EAAE;QAE3E,MAAMC,aACFC,iBAAiBf,MAAM,IAAI,EAAEI,aAAaQ,aAAaL,QAAQ,QAAQ;QAE3E,MAAMS,SAAS;YACX,MAAM;YACN,SAASF;YACT,iBAAiB;gBACb,WAAW;oBACP,cAAc;wBAAE,MAAM;oBAAY;gBACtC;YACJ;QACJ;QAEA,IAAI;YACA,MAAMG,WAAW,MAAM,IAAI,CAAC,EAAE,CAAC,YAAY,CAAC;gBACxC,OAAOnB,cAAc,KAAK;gBAC1B,YAAY;oBACR,SAASA,cAAc,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE;oBAC1CO;gBACJ;gBACAW;gBACA,YAAY;gBACZ,QAAQvB,OAAO,MAAM;gBACrB,GAAIiB,OAAO,IAAI,CAACJ,UAAU,MAAM,GAAG,IAC7B;oBAAE,OAAOA;oBAAU,UAAUY,YAAY;gBAAI,IAC7C,CAAC,CAAC;YACZ;YAEA,MAAMC,OACFF,SAAS,IAAI,IACZA,CAAAA,SAAS,KAAK,CAAC,MAAM,CAACG,CAAAA,OAAQA,KAAK,IAAI,CAAC,MAAM,GAAG,GAAG,GAAG,IAAI,QAAQ,EAAC;YAEzE,MAAMC,QAAQC,gBAAgB,WAAW,CAACH,MAAM,OAAO,GAAG,GAAG;YAE7D,IAAII;YAEJ,IAAIF,OAAO;gBACPE,WAAY,MAAM,IAAI,CAAC,kBAAkB,CAAC,OAAO,CAACF;gBAClD,MAAMG,2BAA2BD,UAAUtB,UAAUD,MAAM,eAAe;YAC9E;YAEA,MAAMyB,SAASC,KAAK,SAAS,CAACH;YAE9B,MAAMI,YAAY,IAAIC;YACtB,IAAIC,gBAAgB;YACpB,KAAK,MAAMT,QAAQH,SAAS,KAAK,CAC7B,KAAK,MAAMa,QAAQV,KAAK,SAAS,CAAE;gBAC/BS;gBACA,IAAIC,AAAkB,wBAAlBA,KAAK,QAAQ,EAA0B;oBACvC,MAAMC,QAAQD,KAAK,KAAK;oBACxB,IAAIC,MAAM,MAAM,EACZJ,UAAU,GAAG,CAACI,MAAM,MAAM;gBAElC;YACJ;YAGJ,MAAMC,YAA2C;gBAC7C,WAAW;uBAAIL;iBAAU;gBACzB,UAAUpB,QAAQ,QAAQ;gBAC1BsB;gBACA,YAAYZ,SAAS,KAAK,CAAC,MAAM;gBACjC,gBAAgBP,OAAO,IAAI,CAACJ;gBAC5B,mBAAmBM;YACvB;YAEA,OAAOjB,OAAO,EAAE,CAAC;gBAAE8B;gBAAQO;YAAU;QACzC,EAAE,OAAOC,OAAO;YACZ,OAAOtC,OAAO,IAAI,CACd,IAAIC,MACA,CAAC,sBAAsB,EAAEqC,iBAAiBrC,QAAQqC,MAAM,OAAO,GAAGC,OAAOD,QAAQ;QAG7F;IACJ;AACJ;AAEO,MAAME,+CACTC,+BAA+B,oBAAoB,CAAC;IAChD,gBAAgBrD;IAChB,cAAc;QACVsD;QACAC;QACAC;QACAC;QACAC;QACAC;QACAC;QACAC;QACAC;KACH;AACL"}
+{"version":3,"file":"api/features/CmsGenerateEntryContent/CmsGenerateEntryContentUseCase.js","sources":["../../../../src/api/features/CmsGenerateEntryContent/CmsGenerateEntryContentUseCase.ts"],"sourcesContent":["import { stepCountIs } from \"ai\";\nimport { Result } from \"@webiny/feature/api\";\nimport { Ai } from \"@webiny/api-core/features/ai/index.js\";\nimport { AiSdkTools } from \"@webiny/api-core/features/ai/index.js\";\nimport { AiToolPipelineRunner } from \"@webiny/api-core/features/ai/index.js\";\nimport { Encryption } from \"@webiny/api-core/features/encryption/index.js\";\nimport { ListTagsUseCase } from \"@webiny/api-file-manager/features/file/ListTags/index.js\";\nimport { GetModelUseCase } from \"@webiny/api-headless-cms/features/contentModel/GetModel/index.js\";\nimport { ModelToAstConverter } from \"@webiny/api-headless-cms/features/contentModel/ModelToAstConverter/index.js\";\nimport { CmsModelToJsonSchemaConverter } from \"@webiny/api-headless-cms/utils/contentModelToJsonSchema/index.js\";\nimport { GetSettingsUseCase } from \"~/api/features/GetSettings/index.js\";\nimport {\n    AiPromptContextBuilder,\n    formatAdditionalFilesContext\n} from \"~/api/features/AiPromptContext/index.js\";\nimport { createReadProjectFileTool } from \"~/api/features/AiPromptContext/ReadProjectFileTool.js\";\nimport { CmsGenerateEntryContentUseCase } from \"./abstractions.js\";\nimport type {\n    CmsGenerateEntryContentParams,\n    GenerateEntryContentResult,\n    GenerateEntryContentTelemetry\n} from \"./abstractions.js\";\nimport { buildEntryPrompt } from \"./buildPrompt.js\";\nimport { LlmJsonResponse } from \"../WbGeneratePageContent/LlmJsonResponse.js\";\nimport { injectDynamicZoneTypenames } from \"./injectDynamicZoneTypenames.js\";\n\nclass CmsGenerateEntryContentUseCaseImpl implements CmsGenerateEntryContentUseCase.Interface {\n    constructor(\n        private promptContextBuilder: AiPromptContextBuilder.Interface,\n        private getSettings: GetSettingsUseCase.Interface,\n        private ai: Ai.Interface,\n        private aiSdkTools: AiSdkTools.Interface,\n        private encryption: Encryption.Interface,\n        private listTags: ListTagsUseCase.Interface,\n        private getModel: GetModelUseCase.Interface,\n        private modelToAst: ModelToAstConverter.Interface,\n        private toolPipelineRunner: AiToolPipelineRunner.Interface\n    ) {}\n\n    async execute<TValues = Record<string, any>>(\n        params: CmsGenerateEntryContentParams\n    ): Promise<Result<GenerateEntryContentResult<TValues>, Error>> {\n        const settingsResult = await this.getSettings.execute();\n        if (settingsResult.isFail()) {\n            return Result.fail(new Error(\"Failed to load AI PowerUps settings.\"));\n        }\n\n        const settings = settingsResult.value;\n        const firstProvider = settings.providers.presets[0];\n\n        if (!firstProvider) {\n            return Result.fail(\n                new Error(\"No AI provider configured. Add a provider in AI Power Ups settings.\")\n            );\n        }\n\n        const modelResult = await this.getModel.execute(params.modelId);\n        if (modelResult.isFail()) {\n            return Result.fail(new Error(`Content model \"${params.modelId}\" not found.`));\n        }\n\n        const model = modelResult.value;\n        const modelAst = this.modelToAst.toAst(model);\n        const jsonSchemaConverter = new CmsModelToJsonSchemaConverter();\n        const entrySchema = jsonSchemaConverter.convert(modelAst, {\n            name: model.name,\n            description: model.description\n        });\n\n        const apiKey = await this.encryption.decrypt(firstProvider.apiKeyEncrypted);\n        const sdkTools = this.aiSdkTools.getToolSet();\n\n        const context = await this.promptContextBuilder.execute({\n            projectId: params.projectId,\n            readerPersonaId: params.readerPersonaId,\n            writerPersonaId: params.writerPersonaId,\n            excludedFileIds: params.excludedFileIds,\n            additionalFileIds: params.additionalFileIds\n        });\n\n        if (context.allProjectFiles.length > 0) {\n            const projectFileTool = createReadProjectFileTool(\n                context.allProjectFiles,\n                context.excludedFileIds\n            );\n            Object.assign(sdkTools, projectFileTool);\n        }\n\n        const tagsResult = await this.listTags.execute({\n            where: { type_startsWith: \"image/\" },\n            limit: 100\n        });\n        const imageTags = tagsResult.isOk() ? tagsResult.value.map(t => t.tag) : [];\n\n        const systemText =\n            buildEntryPrompt(model.name, entrySchema, imageTags) + context.toString();\n\n        const system = {\n            role: \"system\" as const,\n            content: systemText,\n            providerOptions: {\n                anthropic: {\n                    cacheControl: { type: \"ephemeral\" }\n                }\n            }\n        };\n\n        try {\n            const aiResult = await this.ai.generateText({\n                model: firstProvider.model,\n                connection: {\n                    sdkName: firstProvider.model.split(\"/\")[0],\n                    apiKey\n                },\n                system,\n                toolChoice: \"auto\",\n                prompt: params.prompt + formatAdditionalFilesContext(context.additionalFiles),\n                ...(Object.keys(sdkTools).length > 0\n                    ? { tools: sdkTools, stopWhen: stepCountIs(20) }\n                    : {})\n            });\n\n            const text =\n                aiResult.text ||\n                (aiResult.steps.filter(step => step.text.length > 0).pop()?.text ?? \"\");\n\n            const entry = LlmJsonResponse.fromRawText(text).toArray().pop();\n\n            let resolved: Record<string, any> | undefined;\n\n            if (entry) {\n                resolved = (await this.toolPipelineRunner.resolve(entry)) as Record<string, any>;\n                await injectDynamicZoneTypenames(resolved, modelAst, model.singularApiName);\n            }\n\n            const filesRead = new Set<string>();\n            let toolCallsMade = 0;\n            for (const step of aiResult.steps) {\n                for (const call of step.toolCalls) {\n                    toolCallsMade++;\n                    if (call.toolName === \"read_project_file\") {\n                        const input = call.input as { fileId?: string };\n                        if (input.fileId) {\n                            filesRead.add(input.fileId);\n                        }\n                    }\n                }\n            }\n\n            const telemetry: GenerateEntryContentTelemetry = {\n                filesRead: [...filesRead],\n                cacheHit: context.cacheHit,\n                toolCallsMade,\n                totalSteps: aiResult.steps.length,\n                toolsAvailable: Object.keys(sdkTools),\n                imageTagsInPrompt: imageTags\n            };\n\n            // Boundary cast: the AI output is dynamic JSON, so this is the one honest point\n            // where we hand the resolved values to the caller's chosen `TValues` shape.\n            return Result.ok({ values: (resolved ?? {}) as TValues, telemetry });\n        } catch (error) {\n            return Result.fail(\n                new Error(\n                    `AI generation failed: ${error instanceof Error ? error.message : String(error)}`\n                )\n            );\n        }\n    }\n}\n\nexport const CmsGenerateEntryContentUseCaseImplementation =\n    CmsGenerateEntryContentUseCase.createImplementation({\n        implementation: CmsGenerateEntryContentUseCaseImpl,\n        dependencies: [\n            AiPromptContextBuilder,\n            GetSettingsUseCase,\n            Ai,\n            AiSdkTools,\n            Encryption,\n            ListTagsUseCase,\n            GetModelUseCase,\n            ModelToAstConverter,\n            AiToolPipelineRunner\n        ]\n    });\n"],"names":["CmsGenerateEntryContentUseCaseImpl","promptContextBuilder","getSettings","ai","aiSdkTools","encryption","listTags","getModel","modelToAst","toolPipelineRunner","params","settingsResult","Result","Error","settings","firstProvider","modelResult","model","modelAst","jsonSchemaConverter","CmsModelToJsonSchemaConverter","entrySchema","apiKey","sdkTools","context","projectFileTool","createReadProjectFileTool","Object","tagsResult","imageTags","t","systemText","buildEntryPrompt","system","aiResult","formatAdditionalFilesContext","stepCountIs","text","step","entry","LlmJsonResponse","resolved","injectDynamicZoneTypenames","filesRead","Set","toolCallsMade","call","input","telemetry","error","String","CmsGenerateEntryContentUseCaseImplementation","CmsGenerateEntryContentUseCase","AiPromptContextBuilder","GetSettingsUseCase","Ai","AiSdkTools","Encryption","ListTagsUseCase","GetModelUseCase","ModelToAstConverter","AiToolPipelineRunner"],"mappings":";;;;;;;;;;;;;;;AA0BA,MAAMA;IACF,YACYC,oBAAsD,EACtDC,WAAyC,EACzCC,EAAgB,EAChBC,UAAgC,EAChCC,UAAgC,EAChCC,QAAmC,EACnCC,QAAmC,EACnCC,UAAyC,EACzCC,kBAAkD,CAC5D;aATUR,oBAAoB,GAApBA;aACAC,WAAW,GAAXA;aACAC,EAAE,GAAFA;aACAC,UAAU,GAAVA;aACAC,UAAU,GAAVA;aACAC,QAAQ,GAARA;aACAC,QAAQ,GAARA;aACAC,UAAU,GAAVA;aACAC,kBAAkB,GAAlBA;IACT;IAEH,MAAM,QACFC,MAAqC,EACsB;QAC3D,MAAMC,iBAAiB,MAAM,IAAI,CAAC,WAAW,CAAC,OAAO;QACrD,IAAIA,eAAe,MAAM,IACrB,OAAOC,OAAO,IAAI,CAAC,IAAIC,MAAM;QAGjC,MAAMC,WAAWH,eAAe,KAAK;QACrC,MAAMI,gBAAgBD,SAAS,SAAS,CAAC,OAAO,CAAC,EAAE;QAEnD,IAAI,CAACC,eACD,OAAOH,OAAO,IAAI,CACd,IAAIC,MAAM;QAIlB,MAAMG,cAAc,MAAM,IAAI,CAAC,QAAQ,CAAC,OAAO,CAACN,OAAO,OAAO;QAC9D,IAAIM,YAAY,MAAM,IAClB,OAAOJ,OAAO,IAAI,CAAC,IAAIC,MAAM,CAAC,eAAe,EAAEH,OAAO,OAAO,CAAC,YAAY,CAAC;QAG/E,MAAMO,QAAQD,YAAY,KAAK;QAC/B,MAAME,WAAW,IAAI,CAAC,UAAU,CAAC,KAAK,CAACD;QACvC,MAAME,sBAAsB,IAAIC;QAChC,MAAMC,cAAcF,oBAAoB,OAAO,CAACD,UAAU;YACtD,MAAMD,MAAM,IAAI;YAChB,aAAaA,MAAM,WAAW;QAClC;QAEA,MAAMK,SAAS,MAAM,IAAI,CAAC,UAAU,CAAC,OAAO,CAACP,cAAc,eAAe;QAC1E,MAAMQ,WAAW,IAAI,CAAC,UAAU,CAAC,UAAU;QAE3C,MAAMC,UAAU,MAAM,IAAI,CAAC,oBAAoB,CAAC,OAAO,CAAC;YACpD,WAAWd,OAAO,SAAS;YAC3B,iBAAiBA,OAAO,eAAe;YACvC,iBAAiBA,OAAO,eAAe;YACvC,iBAAiBA,OAAO,eAAe;YACvC,mBAAmBA,OAAO,iBAAiB;QAC/C;QAEA,IAAIc,QAAQ,eAAe,CAAC,MAAM,GAAG,GAAG;YACpC,MAAMC,kBAAkBC,0BACpBF,QAAQ,eAAe,EACvBA,QAAQ,eAAe;YAE3BG,OAAO,MAAM,CAACJ,UAAUE;QAC5B;QAEA,MAAMG,aAAa,MAAM,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC;YAC3C,OAAO;gBAAE,iBAAiB;YAAS;YACnC,OAAO;QACX;QACA,MAAMC,YAAYD,WAAW,IAAI,KAAKA,WAAW,KAAK,CAAC,GAAG,CAACE,CAAAA,IAAKA,EAAE,GAAG,IAAI,EAAE;QAE3E,MAAMC,aACFC,iBAAiBf,MAAM,IAAI,EAAEI,aAAaQ,aAAaL,QAAQ,QAAQ;QAE3E,MAAMS,SAAS;YACX,MAAM;YACN,SAASF;YACT,iBAAiB;gBACb,WAAW;oBACP,cAAc;wBAAE,MAAM;oBAAY;gBACtC;YACJ;QACJ;QAEA,IAAI;YACA,MAAMG,WAAW,MAAM,IAAI,CAAC,EAAE,CAAC,YAAY,CAAC;gBACxC,OAAOnB,cAAc,KAAK;gBAC1B,YAAY;oBACR,SAASA,cAAc,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE;oBAC1CO;gBACJ;gBACAW;gBACA,YAAY;gBACZ,QAAQvB,OAAO,MAAM,GAAGyB,6BAA6BX,QAAQ,eAAe;gBAC5E,GAAIG,OAAO,IAAI,CAACJ,UAAU,MAAM,GAAG,IAC7B;oBAAE,OAAOA;oBAAU,UAAUa,YAAY;gBAAI,IAC7C,CAAC,CAAC;YACZ;YAEA,MAAMC,OACFH,SAAS,IAAI,IACZA,CAAAA,SAAS,KAAK,CAAC,MAAM,CAACI,CAAAA,OAAQA,KAAK,IAAI,CAAC,MAAM,GAAG,GAAG,GAAG,IAAI,QAAQ,EAAC;YAEzE,MAAMC,QAAQC,gBAAgB,WAAW,CAACH,MAAM,OAAO,GAAG,GAAG;YAE7D,IAAII;YAEJ,IAAIF,OAAO;gBACPE,WAAY,MAAM,IAAI,CAAC,kBAAkB,CAAC,OAAO,CAACF;gBAClD,MAAMG,2BAA2BD,UAAUvB,UAAUD,MAAM,eAAe;YAC9E;YAEA,MAAM0B,YAAY,IAAIC;YACtB,IAAIC,gBAAgB;YACpB,KAAK,MAAMP,QAAQJ,SAAS,KAAK,CAC7B,KAAK,MAAMY,QAAQR,KAAK,SAAS,CAAE;gBAC/BO;gBACA,IAAIC,AAAkB,wBAAlBA,KAAK,QAAQ,EAA0B;oBACvC,MAAMC,QAAQD,KAAK,KAAK;oBACxB,IAAIC,MAAM,MAAM,EACZJ,UAAU,GAAG,CAACI,MAAM,MAAM;gBAElC;YACJ;YAGJ,MAAMC,YAA2C;gBAC7C,WAAW;uBAAIL;iBAAU;gBACzB,UAAUnB,QAAQ,QAAQ;gBAC1BqB;gBACA,YAAYX,SAAS,KAAK,CAAC,MAAM;gBACjC,gBAAgBP,OAAO,IAAI,CAACJ;gBAC5B,mBAAmBM;YACvB;YAIA,OAAOjB,OAAO,EAAE,CAAC;gBAAE,QAAS6B,YAAY,CAAC;gBAAeO;YAAU;QACtE,EAAE,OAAOC,OAAO;YACZ,OAAOrC,OAAO,IAAI,CACd,IAAIC,MACA,CAAC,sBAAsB,EAAEoC,iBAAiBpC,QAAQoC,MAAM,OAAO,GAAGC,OAAOD,QAAQ;QAG7F;IACJ;AACJ;AAEO,MAAME,+CACTC,+BAA+B,oBAAoB,CAAC;IAChD,gBAAgBpD;IAChB,cAAc;QACVqD;QACAC;QACAC;QACAC;QACAC;QACAC;QACAC;QACAC;QACAC;KACH;AACL"}

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

@@ -6,6 +6,7 @@ export interface CmsGenerateEntryContentParams {
     excludedFileIds?: string[] | null;
     readerPersonaId?: string | null;
     writerPersonaId?: string | null;
+    additionalFileIds?: string[] | null;
 }
 export interface GenerateEntryContentTelemetry {
     filesRead: string[];
@@ -15,12 +16,18 @@ export interface GenerateEntryContentTelemetry {
     toolsAvailable: string[];
     imageTagsInPrompt: string[];
 }
-export interface GenerateEntryContentResult {
-    output: string;
+export interface GenerateEntryContentResult<TValues = Record<string, any>> {
+    /**
+     * The generated entry values, keyed by fieldId. Callers that consume the result
+     * directly (e.g. a bulk action) can read fields straight off this object; transport
+     * edges (the background task / websocket stream) serialize it themselves. Pass a type
+     * argument to `execute` to shape this (e.g. `execute<{ aiSummary?: string }>`).
+     */
+    values: TValues;
     telemetry: GenerateEntryContentTelemetry;
 }
 export interface ICmsGenerateEntryContentUseCase {
-    execute(params: CmsGenerateEntryContentParams): Promise<Result<GenerateEntryContentResult, Error>>;
+    execute<TValues = Record<string, any>>(params: CmsGenerateEntryContentParams): Promise<Result<GenerateEntryContentResult<TValues>, Error>>;
 }
 export declare const CmsGenerateEntryContentUseCase: import("@webiny/di").Abstraction<ICmsGenerateEntryContentUseCase>;
 export declare namespace CmsGenerateEntryContentUseCase {

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

@@ -1 +1 @@
-{"version":3,"file":"api/features/CmsGenerateEntryContent/abstractions.js","sources":["../../../../src/api/features/CmsGenerateEntryContent/abstractions.ts"],"sourcesContent":["import { createAbstraction, Result } from \"@webiny/feature/api\";\n\nexport interface CmsGenerateEntryContentParams {\n    prompt: string;\n    modelId: string;\n    projectId?: string | null;\n    excludedFileIds?: string[] | null;\n    readerPersonaId?: string | null;\n    writerPersonaId?: string | null;\n}\n\nexport interface GenerateEntryContentTelemetry {\n    filesRead: string[];\n    cacheHit: boolean;\n    toolCallsMade: number;\n    totalSteps: number;\n    toolsAvailable: string[];\n    imageTagsInPrompt: string[];\n}\n\nexport interface GenerateEntryContentResult {\n    output: string;\n    telemetry: GenerateEntryContentTelemetry;\n}\n\nexport interface ICmsGenerateEntryContentUseCase {\n    execute(\n        params: CmsGenerateEntryContentParams\n    ): Promise<Result<GenerateEntryContentResult, Error>>;\n}\n\nexport const CmsGenerateEntryContentUseCase = createAbstraction<ICmsGenerateEntryContentUseCase>(\n    \"AiPowerUpsCmsGenerateEntryContentUseCase\"\n);\n\nexport namespace CmsGenerateEntryContentUseCase {\n    export type Interface = ICmsGenerateEntryContentUseCase;\n    export type Params = CmsGenerateEntryContentParams;\n}\n"],"names":["CmsGenerateEntryContentUseCase","createAbstraction"],"mappings":";AA+BO,MAAMA,iCAAiCC,kBAC1C"}
+{"version":3,"file":"api/features/CmsGenerateEntryContent/abstractions.js","sources":["../../../../src/api/features/CmsGenerateEntryContent/abstractions.ts"],"sourcesContent":["import { createAbstraction, Result } from \"@webiny/feature/api\";\n\nexport interface CmsGenerateEntryContentParams {\n    prompt: string;\n    modelId: string;\n    projectId?: string | null;\n    excludedFileIds?: string[] | null;\n    readerPersonaId?: string | null;\n    writerPersonaId?: string | null;\n    additionalFileIds?: string[] | null;\n}\n\nexport interface GenerateEntryContentTelemetry {\n    filesRead: string[];\n    cacheHit: boolean;\n    toolCallsMade: number;\n    totalSteps: number;\n    toolsAvailable: string[];\n    imageTagsInPrompt: string[];\n}\n\nexport interface GenerateEntryContentResult<TValues = Record<string, any>> {\n    /**\n     * The generated entry values, keyed by fieldId. Callers that consume the result\n     * directly (e.g. a bulk action) can read fields straight off this object; transport\n     * edges (the background task / websocket stream) serialize it themselves. Pass a type\n     * argument to `execute` to shape this (e.g. `execute<{ aiSummary?: string }>`).\n     */\n    values: TValues;\n    telemetry: GenerateEntryContentTelemetry;\n}\n\nexport interface ICmsGenerateEntryContentUseCase {\n    execute<TValues = Record<string, any>>(\n        params: CmsGenerateEntryContentParams\n    ): Promise<Result<GenerateEntryContentResult<TValues>, Error>>;\n}\n\nexport const CmsGenerateEntryContentUseCase = createAbstraction<ICmsGenerateEntryContentUseCase>(\n    \"AiPowerUpsCmsGenerateEntryContentUseCase\"\n);\n\nexport namespace CmsGenerateEntryContentUseCase {\n    export type Interface = ICmsGenerateEntryContentUseCase;\n    export type Params = CmsGenerateEntryContentParams;\n}\n"],"names":["CmsGenerateEntryContentUseCase","createAbstraction"],"mappings":";AAsCO,MAAMA,iCAAiCC,kBAC1C"}

package/api/features/GetSettings/index.d.ts

@@ -1 +1,2 @@
+import "../Providers/types.js";
 export { GetSettingsUseCase } from "./abstractions.js";

package/api/features/GetSettings/index.js

@@ -1 +1,2 @@
+import "../Providers/types.js";
 export { GetSettingsUseCase } from "./abstractions.js";

package/api/features/Providers/types.d.ts

@@ -1,4 +1,4 @@
-import type { IAiPowerUpsSettings } from "../../../api/types.js";
+import type { IAiPowerUpsSettings } from "../../types.js";
 export interface ProviderPreset {
     id: string;
     name: string;
@@ -8,7 +8,7 @@ export interface ProviderPreset {
     apiKeyMasked: string;
     apiKeyEncrypted: string;
 }
-declare module "~/api/types.js" {
+declare module "../../types.js" {
     interface IAiPowerUpsSettings {
         providers: {
             presets: ProviderPreset[];

package/api/features/WbGeneratePageContent/WbGeneratePageContentTask.d.ts

@@ -11,6 +11,7 @@ export interface IWbGeneratePageContentTaskInput {
     excludedFileIds?: string[] | null;
     readerPersonaId?: string | null;
     writerPersonaId?: string | null;
+    additionalFileIds?: string[] | null;
 }
 declare class WbGeneratePageContentTaskImpl implements TaskDefinition.Interface<IWbGeneratePageContentTaskInput> {
     private identityContext;

package/api/features/WbGeneratePageContent/WbGeneratePageContentTask.js

@@ -18,16 +18,26 @@ class WbGeneratePageContentTaskImpl {
     }
     async run({ input, controller }) {
         if (controller.runtime.isAborted()) return controller.response.aborted();
-        const result = await this.generatePageContent.execute({
-            prompt: input.prompt,
-            components: input.components,
-            tools: input.tools,
-            projectId: input.projectId,
-            excludedFileIds: input.excludedFileIds,
-            readerPersonaId: input.readerPersonaId,
-            writerPersonaId: input.writerPersonaId
-        });
         const identity = this.identityContext.getIdentity();
+        let result;
+        try {
+            result = await this.generatePageContent.execute({
+                prompt: input.prompt,
+                components: input.components,
+                tools: input.tools,
+                projectId: input.projectId,
+                excludedFileIds: input.excludedFileIds,
+                readerPersonaId: input.readerPersonaId,
+                writerPersonaId: input.writerPersonaId,
+                additionalFileIds: input.additionalFileIds
+            });
+        } catch (err) {
+            const message = err instanceof Error ? err.message : String(err);
+            await this.sendErrorToUser(identity.id, message);
+            return controller.response.error({
+                message
+            });
+        }
         if (result.isFail()) {
             await this.sendErrorToUser(identity.id, result.error.message);
             return controller.response.error({

package/api/features/WbGeneratePageContent/WbGeneratePageContentTask.js.map

@@ -1 +1 @@
-{"version":3,"file":"api/features/WbGeneratePageContent/WbGeneratePageContentTask.js","sources":["../../../../src/api/features/WbGeneratePageContent/WbGeneratePageContentTask.ts"],"sourcesContent":["import { TaskDefinition } from \"@webiny/api-core/features/task/TaskDefinition/index.js\";\nimport { WebsocketsSendToIdentityUseCase } from \"@webiny/api-websockets/features/SendToIdentity/abstractions.js\";\nimport { compress } from \"@webiny/utils/features/compression/legacy/gzip.js\";\nimport { WbGeneratePageContentUseCase } from \"~/api/features/WbGeneratePageContent/index.js\";\nimport type { GenerationTelemetry } from \"~/api/features/WbGeneratePageContent/abstractions.js\";\nimport { IdentityContext } from \"@webiny/api-core/exports/api/security.js\";\n\nexport const WB_GENERATE_PAGE_CONTENT_TASK_ID = \"aiPowerUpsGeneratePageContent\";\n\nexport interface IWbGeneratePageContentTaskInput {\n    prompt: string;\n    components: unknown;\n    tools: unknown;\n    projectId?: string | null;\n    excludedFileIds?: string[] | null;\n    readerPersonaId?: string | null;\n    writerPersonaId?: string | null;\n}\n\nclass WbGeneratePageContentTaskImpl implements TaskDefinition.Interface<IWbGeneratePageContentTaskInput> {\n    id = WB_GENERATE_PAGE_CONTENT_TASK_ID;\n    title = \"AI Power-Ups - Generate Page Content\";\n    description = \"Generates page content using AI based on a user prompt.\";\n    maxIterations = 1;\n    isPrivate = true;\n    databaseLogs = false;\n\n    constructor(\n        private identityContext: IdentityContext.Interface,\n        private generatePageContent: WbGeneratePageContentUseCase.Interface,\n        private sendToIdentity: WebsocketsSendToIdentityUseCase.Interface\n    ) {}\n\n    async run({\n        input,\n        controller\n    }: TaskDefinition.RunParams<IWbGeneratePageContentTaskInput>): Promise<\n        TaskDefinition.Result<IWbGeneratePageContentTaskInput>\n    > {\n        if (controller.runtime.isAborted()) {\n            return controller.response.aborted();\n        }\n\n        const result = await this.generatePageContent.execute({\n            prompt: input.prompt,\n            components: input.components,\n            tools: input.tools,\n            projectId: input.projectId,\n            excludedFileIds: input.excludedFileIds,\n            readerPersonaId: input.readerPersonaId,\n            writerPersonaId: input.writerPersonaId\n        });\n\n        const identity = this.identityContext.getIdentity();\n\n        if (result.isFail()) {\n            await this.sendErrorToUser(identity.id, result.error.message);\n\n            return controller.response.error({\n                message: result.error.message\n            });\n        }\n\n        const compressed = await compress(result.value.output);\n        const payload = compressed.toString(\"base64\");\n\n        await this.sendContentToUser(identity.id, payload, result.value.telemetry);\n\n        return controller.response.done(\"Page content generated successfully.\");\n    }\n\n    private async sendContentToUser(\n        identityId: string,\n        payload: string,\n        telemetry: GenerationTelemetry\n    ) {\n        await this.sendToIdentity.execute(\n            { id: identityId },\n            {\n                action: \"aiPowerUps.generatePageContent.content\",\n                data: {\n                    compression: \"gzip\",\n                    value: payload,\n                    telemetry\n                }\n            }\n        );\n    }\n\n    private async sendErrorToUser(identityId: string, message: string) {\n        await this.sendToIdentity.execute(\n            { id: identityId },\n            {\n                action: \"aiPowerUps.generatePageContent.error\",\n                data: {\n                    message\n                }\n            }\n        );\n    }\n}\n\nexport const WbGeneratePageContentTask = TaskDefinition.createImplementation({\n    implementation: WbGeneratePageContentTaskImpl,\n    dependencies: [IdentityContext, WbGeneratePageContentUseCase, WebsocketsSendToIdentityUseCase]\n});\n"],"names":["WB_GENERATE_PAGE_CONTENT_TASK_ID","WbGeneratePageContentTaskImpl","identityContext","generatePageContent","sendToIdentity","input","controller","result","identity","compressed","compress","payload","identityId","telemetry","message","WbGeneratePageContentTask","TaskDefinition","IdentityContext","WbGeneratePageContentUseCase","WebsocketsSendToIdentityUseCase"],"mappings":";;;;;AAOO,MAAMA,mCAAmC;AAYhD,MAAMC;IAQF,YACYC,eAA0C,EAC1CC,mBAA2D,EAC3DC,cAAyD,CACnE;aAHUF,eAAe,GAAfA;aACAC,mBAAmB,GAAnBA;aACAC,cAAc,GAAdA;aAVZ,EAAE,GAAGJ;aACL,KAAK,GAAG;aACR,WAAW,GAAG;aACd,aAAa,GAAG;aAChB,SAAS,GAAG;aACZ,YAAY,GAAG;IAMZ;IAEH,MAAM,IAAI,EACNK,KAAK,EACLC,UAAU,EAC8C,EAE1D;QACE,IAAIA,WAAW,OAAO,CAAC,SAAS,IAC5B,OAAOA,WAAW,QAAQ,CAAC,OAAO;QAGtC,MAAMC,SAAS,MAAM,IAAI,CAAC,mBAAmB,CAAC,OAAO,CAAC;YAClD,QAAQF,MAAM,MAAM;YACpB,YAAYA,MAAM,UAAU;YAC5B,OAAOA,MAAM,KAAK;YAClB,WAAWA,MAAM,SAAS;YAC1B,iBAAiBA,MAAM,eAAe;YACtC,iBAAiBA,MAAM,eAAe;YACtC,iBAAiBA,MAAM,eAAe;QAC1C;QAEA,MAAMG,WAAW,IAAI,CAAC,eAAe,CAAC,WAAW;QAEjD,IAAID,OAAO,MAAM,IAAI;YACjB,MAAM,IAAI,CAAC,eAAe,CAACC,SAAS,EAAE,EAAED,OAAO,KAAK,CAAC,OAAO;YAE5D,OAAOD,WAAW,QAAQ,CAAC,KAAK,CAAC;gBAC7B,SAASC,OAAO,KAAK,CAAC,OAAO;YACjC;QACJ;QAEA,MAAME,aAAa,MAAMC,SAASH,OAAO,KAAK,CAAC,MAAM;QACrD,MAAMI,UAAUF,WAAW,QAAQ,CAAC;QAEpC,MAAM,IAAI,CAAC,iBAAiB,CAACD,SAAS,EAAE,EAAEG,SAASJ,OAAO,KAAK,CAAC,SAAS;QAEzE,OAAOD,WAAW,QAAQ,CAAC,IAAI,CAAC;IACpC;IAEA,MAAc,kBACVM,UAAkB,EAClBD,OAAe,EACfE,SAA8B,EAChC;QACE,MAAM,IAAI,CAAC,cAAc,CAAC,OAAO,CAC7B;YAAE,IAAID;QAAW,GACjB;YACI,QAAQ;YACR,MAAM;gBACF,aAAa;gBACb,OAAOD;gBACPE;YACJ;QACJ;IAER;IAEA,MAAc,gBAAgBD,UAAkB,EAAEE,OAAe,EAAE;QAC/D,MAAM,IAAI,CAAC,cAAc,CAAC,OAAO,CAC7B;YAAE,IAAIF;QAAW,GACjB;YACI,QAAQ;YACR,MAAM;gBACFE;YACJ;QACJ;IAER;AACJ;AAEO,MAAMC,4BAA4BC,eAAe,oBAAoB,CAAC;IACzE,gBAAgBf;IAChB,cAAc;QAACgB;QAAiBC;QAA8BC;KAAgC;AAClG"}
+{"version":3,"file":"api/features/WbGeneratePageContent/WbGeneratePageContentTask.js","sources":["../../../../src/api/features/WbGeneratePageContent/WbGeneratePageContentTask.ts"],"sourcesContent":["import { TaskDefinition } from \"@webiny/api-core/features/task/TaskDefinition/index.js\";\nimport { WebsocketsSendToIdentityUseCase } from \"@webiny/api-websockets/features/SendToIdentity/abstractions.js\";\nimport { compress } from \"@webiny/utils/features/compression/legacy/gzip.js\";\nimport { WbGeneratePageContentUseCase } from \"~/api/features/WbGeneratePageContent/index.js\";\nimport type { GenerationTelemetry } from \"~/api/features/WbGeneratePageContent/abstractions.js\";\nimport { IdentityContext } from \"@webiny/api-core/exports/api/security.js\";\n\nexport const WB_GENERATE_PAGE_CONTENT_TASK_ID = \"aiPowerUpsGeneratePageContent\";\n\nexport interface IWbGeneratePageContentTaskInput {\n    prompt: string;\n    components: unknown;\n    tools: unknown;\n    projectId?: string | null;\n    excludedFileIds?: string[] | null;\n    readerPersonaId?: string | null;\n    writerPersonaId?: string | null;\n    additionalFileIds?: string[] | null;\n}\n\nclass WbGeneratePageContentTaskImpl implements TaskDefinition.Interface<IWbGeneratePageContentTaskInput> {\n    id = WB_GENERATE_PAGE_CONTENT_TASK_ID;\n    title = \"AI Power-Ups - Generate Page Content\";\n    description = \"Generates page content using AI based on a user prompt.\";\n    maxIterations = 1;\n    isPrivate = true;\n    databaseLogs = false;\n\n    constructor(\n        private identityContext: IdentityContext.Interface,\n        private generatePageContent: WbGeneratePageContentUseCase.Interface,\n        private sendToIdentity: WebsocketsSendToIdentityUseCase.Interface\n    ) {}\n\n    async run({\n        input,\n        controller\n    }: TaskDefinition.RunParams<IWbGeneratePageContentTaskInput>): Promise<\n        TaskDefinition.Result<IWbGeneratePageContentTaskInput>\n    > {\n        if (controller.runtime.isAborted()) {\n            return controller.response.aborted();\n        }\n\n        const identity = this.identityContext.getIdentity();\n\n        let result;\n        try {\n            result = await this.generatePageContent.execute({\n                prompt: input.prompt,\n                components: input.components,\n                tools: input.tools,\n                projectId: input.projectId,\n                excludedFileIds: input.excludedFileIds,\n                readerPersonaId: input.readerPersonaId,\n                writerPersonaId: input.writerPersonaId,\n                additionalFileIds: input.additionalFileIds\n            });\n        } catch (err) {\n            const message = err instanceof Error ? err.message : String(err);\n            await this.sendErrorToUser(identity.id, message);\n            return controller.response.error({ message });\n        }\n\n        if (result.isFail()) {\n            await this.sendErrorToUser(identity.id, result.error.message);\n\n            return controller.response.error({\n                message: result.error.message\n            });\n        }\n\n        const compressed = await compress(result.value.output);\n        const payload = compressed.toString(\"base64\");\n\n        await this.sendContentToUser(identity.id, payload, result.value.telemetry);\n\n        return controller.response.done(\"Page content generated successfully.\");\n    }\n\n    private async sendContentToUser(\n        identityId: string,\n        payload: string,\n        telemetry: GenerationTelemetry\n    ) {\n        await this.sendToIdentity.execute(\n            { id: identityId },\n            {\n                action: \"aiPowerUps.generatePageContent.content\",\n                data: {\n                    compression: \"gzip\",\n                    value: payload,\n                    telemetry\n                }\n            }\n        );\n    }\n\n    private async sendErrorToUser(identityId: string, message: string) {\n        await this.sendToIdentity.execute(\n            { id: identityId },\n            {\n                action: \"aiPowerUps.generatePageContent.error\",\n                data: {\n                    message\n                }\n            }\n        );\n    }\n}\n\nexport const WbGeneratePageContentTask = TaskDefinition.createImplementation({\n    implementation: WbGeneratePageContentTaskImpl,\n    dependencies: [IdentityContext, WbGeneratePageContentUseCase, WebsocketsSendToIdentityUseCase]\n});\n"],"names":["WB_GENERATE_PAGE_CONTENT_TASK_ID","WbGeneratePageContentTaskImpl","identityContext","generatePageContent","sendToIdentity","input","controller","identity","result","err","message","Error","String","compressed","compress","payload","identityId","telemetry","WbGeneratePageContentTask","TaskDefinition","IdentityContext","WbGeneratePageContentUseCase","WebsocketsSendToIdentityUseCase"],"mappings":";;;;;AAOO,MAAMA,mCAAmC;AAahD,MAAMC;IAQF,YACYC,eAA0C,EAC1CC,mBAA2D,EAC3DC,cAAyD,CACnE;aAHUF,eAAe,GAAfA;aACAC,mBAAmB,GAAnBA;aACAC,cAAc,GAAdA;aAVZ,EAAE,GAAGJ;aACL,KAAK,GAAG;aACR,WAAW,GAAG;aACd,aAAa,GAAG;aAChB,SAAS,GAAG;aACZ,YAAY,GAAG;IAMZ;IAEH,MAAM,IAAI,EACNK,KAAK,EACLC,UAAU,EAC8C,EAE1D;QACE,IAAIA,WAAW,OAAO,CAAC,SAAS,IAC5B,OAAOA,WAAW,QAAQ,CAAC,OAAO;QAGtC,MAAMC,WAAW,IAAI,CAAC,eAAe,CAAC,WAAW;QAEjD,IAAIC;QACJ,IAAI;YACAA,SAAS,MAAM,IAAI,CAAC,mBAAmB,CAAC,OAAO,CAAC;gBAC5C,QAAQH,MAAM,MAAM;gBACpB,YAAYA,MAAM,UAAU;gBAC5B,OAAOA,MAAM,KAAK;gBAClB,WAAWA,MAAM,SAAS;gBAC1B,iBAAiBA,MAAM,eAAe;gBACtC,iBAAiBA,MAAM,eAAe;gBACtC,iBAAiBA,MAAM,eAAe;gBACtC,mBAAmBA,MAAM,iBAAiB;YAC9C;QACJ,EAAE,OAAOI,KAAK;YACV,MAAMC,UAAUD,eAAeE,QAAQF,IAAI,OAAO,GAAGG,OAAOH;YAC5D,MAAM,IAAI,CAAC,eAAe,CAACF,SAAS,EAAE,EAAEG;YACxC,OAAOJ,WAAW,QAAQ,CAAC,KAAK,CAAC;gBAAEI;YAAQ;QAC/C;QAEA,IAAIF,OAAO,MAAM,IAAI;YACjB,MAAM,IAAI,CAAC,eAAe,CAACD,SAAS,EAAE,EAAEC,OAAO,KAAK,CAAC,OAAO;YAE5D,OAAOF,WAAW,QAAQ,CAAC,KAAK,CAAC;gBAC7B,SAASE,OAAO,KAAK,CAAC,OAAO;YACjC;QACJ;QAEA,MAAMK,aAAa,MAAMC,SAASN,OAAO,KAAK,CAAC,MAAM;QACrD,MAAMO,UAAUF,WAAW,QAAQ,CAAC;QAEpC,MAAM,IAAI,CAAC,iBAAiB,CAACN,SAAS,EAAE,EAAEQ,SAASP,OAAO,KAAK,CAAC,SAAS;QAEzE,OAAOF,WAAW,QAAQ,CAAC,IAAI,CAAC;IACpC;IAEA,MAAc,kBACVU,UAAkB,EAClBD,OAAe,EACfE,SAA8B,EAChC;QACE,MAAM,IAAI,CAAC,cAAc,CAAC,OAAO,CAC7B;YAAE,IAAID;QAAW,GACjB;YACI,QAAQ;YACR,MAAM;gBACF,aAAa;gBACb,OAAOD;gBACPE;YACJ;QACJ;IAER;IAEA,MAAc,gBAAgBD,UAAkB,EAAEN,OAAe,EAAE;QAC/D,MAAM,IAAI,CAAC,cAAc,CAAC,OAAO,CAC7B;YAAE,IAAIM;QAAW,GACjB;YACI,QAAQ;YACR,MAAM;gBACFN;YACJ;QACJ;IAER;AACJ;AAEO,MAAMQ,4BAA4BC,eAAe,oBAAoB,CAAC;IACzE,gBAAgBlB;IAChB,cAAc;QAACmB;QAAiBC;QAA8BC;KAAgC;AAClG"}

package/api/features/WbGeneratePageContent/WbGeneratePageContentUseCase.js

@@ -4,7 +4,7 @@ import { Ai, AiSdkTools } from "@webiny/api-core/features/ai/index.js";
 import { Encryption } from "@webiny/api-core/features/encryption/index.js";
 import { ListTagsUseCase } from "@webiny/api-file-manager/features/file/ListTags/index.js";
 import { GetSettingsUseCase } from "../GetSettings/index.js";
-import { AiPromptContextBuilder } from "../AiPromptContext/index.js";
+import { AiPromptContextBuilder, formatAdditionalFilesContext } from "../AiPromptContext/index.js";
 import { createReadProjectFileTool } from "../AiPromptContext/ReadProjectFileTool.js";
 import { WbGeneratePageContentUseCase } from "./abstractions.js";
 import { buildDomainPrompt } from "./buildPrompt.js";
@@ -31,7 +31,8 @@ class WbGeneratePageContentUseCaseImpl {
             projectId: params.projectId,
             readerPersonaId: params.readerPersonaId,
             writerPersonaId: params.writerPersonaId,
-            excludedFileIds: params.excludedFileIds
+            excludedFileIds: params.excludedFileIds,
+            additionalFileIds: params.additionalFileIds
         });
         if (context.allProjectFiles.length > 0) {
             const projectFileTool = createReadProjectFileTool(context.allProjectFiles, context.excludedFileIds);
@@ -66,7 +67,7 @@ class WbGeneratePageContentUseCaseImpl {
                 },
                 system,
                 toolChoice: "auto",
-                prompt: params.prompt,
+                prompt: params.prompt + formatAdditionalFilesContext(context.additionalFiles),
                 ...Object.keys(sdkTools).length > 0 ? {
                     tools: sdkTools,
                     stopWhen: stepCountIs(20)

package/api/features/WbGeneratePageContent/WbGeneratePageContentUseCase.js.map

@@ -1 +1 @@
-{"version":3,"file":"api/features/WbGeneratePageContent/WbGeneratePageContentUseCase.js","sources":["../../../../src/api/features/WbGeneratePageContent/WbGeneratePageContentUseCase.ts"],"sourcesContent":["import { stepCountIs } from \"ai\";\nimport { Result } from \"@webiny/feature/api\";\nimport { Ai } from \"@webiny/api-core/features/ai/index.js\";\nimport { AiSdkTools } from \"@webiny/api-core/features/ai/index.js\";\nimport { Encryption } from \"@webiny/api-core/features/encryption/index.js\";\nimport { ListTagsUseCase } from \"@webiny/api-file-manager/features/file/ListTags/index.js\";\nimport { GetSettingsUseCase } from \"~/api/features/GetSettings/index.js\";\nimport { AiPromptContextBuilder } from \"~/api/features/AiPromptContext/index.js\";\nimport { createReadProjectFileTool } from \"~/api/features/AiPromptContext/ReadProjectFileTool.js\";\nimport { WbGeneratePageContentUseCase } from \"./abstractions.js\";\nimport type {\n    WbGeneratePageContentParams,\n    GeneratePageContentResult,\n    GenerationTelemetry\n} from \"./abstractions.js\";\nimport { buildDomainPrompt } from \"./buildPrompt.js\";\nimport { LlmJsonResponse } from \"./LlmJsonResponse.js\";\nimport { ComponentFilter } from \"./ComponentFilter.js\";\n\nclass WbGeneratePageContentUseCaseImpl implements WbGeneratePageContentUseCase.Interface {\n    constructor(\n        private promptContextBuilder: AiPromptContextBuilder.Interface,\n        private getSettings: GetSettingsUseCase.Interface,\n        private ai: Ai.Interface,\n        private aiSdkTools: AiSdkTools.Interface,\n        private encryption: Encryption.Interface,\n        private listTags: ListTagsUseCase.Interface\n    ) {}\n\n    async execute(\n        params: WbGeneratePageContentParams\n    ): Promise<Result<GeneratePageContentResult, Error>> {\n        const settingsResult = await this.getSettings.execute();\n        if (settingsResult.isFail()) {\n            return Result.fail(new Error(\"Failed to load AI PowerUps settings.\"));\n        }\n\n        const settings = settingsResult.value;\n        const firstProvider = settings.providers.presets[0];\n\n        if (!firstProvider) {\n            return Result.fail(\n                new Error(\"No AI provider configured. Add a provider in AI Power Ups settings.\")\n            );\n        }\n\n        const apiKey = await this.encryption.decrypt(firstProvider.apiKeyEncrypted);\n\n        const sdkTools = this.aiSdkTools.getToolSet();\n\n        const context = await this.promptContextBuilder.execute({\n            projectId: params.projectId,\n            readerPersonaId: params.readerPersonaId,\n            writerPersonaId: params.writerPersonaId,\n            excludedFileIds: params.excludedFileIds\n        });\n\n        if (context.allProjectFiles.length > 0) {\n            const projectFileTool = createReadProjectFileTool(\n                context.allProjectFiles,\n                context.excludedFileIds\n            );\n            Object.assign(sdkTools, projectFileTool);\n        }\n\n        // Fetch image tags so the prompt can enumerate what's queryable; without\n        // this the model invents tag/IDs instead of calling listImagesByTag.\n        const tagsResult = await this.listTags.execute({\n            where: { type_startsWith: \"image/\" },\n            limit: 100\n        });\n        const imageTags = tagsResult.isOk() ? tagsResult.value.map(t => t.tag) : [];\n\n        const components = params.components as Array<{ name: string }>;\n        const systemText =\n            buildDomainPrompt(components, params.tools, imageTags) + context.toString();\n\n        const system = {\n            role: \"system\" as const,\n            content: systemText,\n            providerOptions: {\n                anthropic: {\n                    cacheControl: { type: \"ephemeral\" }\n                }\n            }\n        };\n\n        try {\n            const aiResult = await this.ai.generateText({\n                model: firstProvider.model,\n                connection: {\n                    sdkName: firstProvider.model.split(\"/\")[0],\n                    apiKey\n                },\n                system,\n                toolChoice: \"auto\",\n                prompt: params.prompt,\n                ...(Object.keys(sdkTools).length > 0\n                    ? { tools: sdkTools, stopWhen: stepCountIs(20) }\n                    : {})\n            });\n\n            const text =\n                aiResult.text ||\n                (aiResult.steps.filter(step => step.text.length > 0).pop()?.text ?? \"\");\n\n            const elements = LlmJsonResponse.fromRawText(text).toArray();\n            const componentFilter = new ComponentFilter(components);\n            const output = JSON.stringify(componentFilter.filter(elements));\n\n            const filesRead = new Set<string>();\n            let toolCallsMade = 0;\n            for (const step of aiResult.steps) {\n                for (const call of step.toolCalls) {\n                    toolCallsMade++;\n                    if (call.toolName === \"read_project_file\") {\n                        const input = call.input as { fileId?: string };\n                        if (input.fileId) {\n                            filesRead.add(input.fileId);\n                        }\n                    }\n                }\n            }\n\n            const telemetry: GenerationTelemetry = {\n                filesRead: [...filesRead],\n                cacheHit: context.cacheHit,\n                toolCallsMade,\n                totalSteps: aiResult.steps.length,\n                toolsAvailable: Object.keys(sdkTools),\n                imageTagsInPrompt: imageTags\n            };\n\n            return Result.ok({ output, telemetry });\n        } catch (error) {\n            return Result.fail(\n                new Error(\n                    `AI generation failed: ${error instanceof Error ? error.message : String(error)}`\n                )\n            );\n        }\n    }\n}\n\nexport const WbGeneratePageContentUseCaseImplementation =\n    WbGeneratePageContentUseCase.createImplementation({\n        implementation: WbGeneratePageContentUseCaseImpl,\n        dependencies: [\n            AiPromptContextBuilder,\n            GetSettingsUseCase,\n            Ai,\n            AiSdkTools,\n            Encryption,\n            ListTagsUseCase\n        ]\n    });\n"],"names":["WbGeneratePageContentUseCaseImpl","promptContextBuilder","getSettings","ai","aiSdkTools","encryption","listTags","params","settingsResult","Result","Error","settings","firstProvider","apiKey","sdkTools","context","projectFileTool","createReadProjectFileTool","Object","tagsResult","imageTags","t","components","systemText","buildDomainPrompt","system","aiResult","stepCountIs","text","step","elements","LlmJsonResponse","componentFilter","ComponentFilter","output","JSON","filesRead","Set","toolCallsMade","call","input","telemetry","error","String","WbGeneratePageContentUseCaseImplementation","WbGeneratePageContentUseCase","AiPromptContextBuilder","GetSettingsUseCase","Ai","AiSdkTools","Encryption","ListTagsUseCase"],"mappings":";;;;;;;;;;;;AAmBA,MAAMA;IACF,YACYC,oBAAsD,EACtDC,WAAyC,EACzCC,EAAgB,EAChBC,UAAgC,EAChCC,UAAgC,EAChCC,QAAmC,CAC7C;aANUL,oBAAoB,GAApBA;aACAC,WAAW,GAAXA;aACAC,EAAE,GAAFA;aACAC,UAAU,GAAVA;aACAC,UAAU,GAAVA;aACAC,QAAQ,GAARA;IACT;IAEH,MAAM,QACFC,MAAmC,EACc;QACjD,MAAMC,iBAAiB,MAAM,IAAI,CAAC,WAAW,CAAC,OAAO;QACrD,IAAIA,eAAe,MAAM,IACrB,OAAOC,OAAO,IAAI,CAAC,IAAIC,MAAM;QAGjC,MAAMC,WAAWH,eAAe,KAAK;QACrC,MAAMI,gBAAgBD,SAAS,SAAS,CAAC,OAAO,CAAC,EAAE;QAEnD,IAAI,CAACC,eACD,OAAOH,OAAO,IAAI,CACd,IAAIC,MAAM;QAIlB,MAAMG,SAAS,MAAM,IAAI,CAAC,UAAU,CAAC,OAAO,CAACD,cAAc,eAAe;QAE1E,MAAME,WAAW,IAAI,CAAC,UAAU,CAAC,UAAU;QAE3C,MAAMC,UAAU,MAAM,IAAI,CAAC,oBAAoB,CAAC,OAAO,CAAC;YACpD,WAAWR,OAAO,SAAS;YAC3B,iBAAiBA,OAAO,eAAe;YACvC,iBAAiBA,OAAO,eAAe;YACvC,iBAAiBA,OAAO,eAAe;QAC3C;QAEA,IAAIQ,QAAQ,eAAe,CAAC,MAAM,GAAG,GAAG;YACpC,MAAMC,kBAAkBC,0BACpBF,QAAQ,eAAe,EACvBA,QAAQ,eAAe;YAE3BG,OAAO,MAAM,CAACJ,UAAUE;QAC5B;QAIA,MAAMG,aAAa,MAAM,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC;YAC3C,OAAO;gBAAE,iBAAiB;YAAS;YACnC,OAAO;QACX;QACA,MAAMC,YAAYD,WAAW,IAAI,KAAKA,WAAW,KAAK,CAAC,GAAG,CAACE,CAAAA,IAAKA,EAAE,GAAG,IAAI,EAAE;QAE3E,MAAMC,aAAaf,OAAO,UAAU;QACpC,MAAMgB,aACFC,kBAAkBF,YAAYf,OAAO,KAAK,EAAEa,aAAaL,QAAQ,QAAQ;QAE7E,MAAMU,SAAS;YACX,MAAM;YACN,SAASF;YACT,iBAAiB;gBACb,WAAW;oBACP,cAAc;wBAAE,MAAM;oBAAY;gBACtC;YACJ;QACJ;QAEA,IAAI;YACA,MAAMG,WAAW,MAAM,IAAI,CAAC,EAAE,CAAC,YAAY,CAAC;gBACxC,OAAOd,cAAc,KAAK;gBAC1B,YAAY;oBACR,SAASA,cAAc,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE;oBAC1CC;gBACJ;gBACAY;gBACA,YAAY;gBACZ,QAAQlB,OAAO,MAAM;gBACrB,GAAIW,OAAO,IAAI,CAACJ,UAAU,MAAM,GAAG,IAC7B;oBAAE,OAAOA;oBAAU,UAAUa,YAAY;gBAAI,IAC7C,CAAC,CAAC;YACZ;YAEA,MAAMC,OACFF,SAAS,IAAI,IACZA,CAAAA,SAAS,KAAK,CAAC,MAAM,CAACG,CAAAA,OAAQA,KAAK,IAAI,CAAC,MAAM,GAAG,GAAG,GAAG,IAAI,QAAQ,EAAC;YAEzE,MAAMC,WAAWC,gBAAgB,WAAW,CAACH,MAAM,OAAO;YAC1D,MAAMI,kBAAkB,IAAIC,gBAAgBX;YAC5C,MAAMY,SAASC,KAAK,SAAS,CAACH,gBAAgB,MAAM,CAACF;YAErD,MAAMM,YAAY,IAAIC;YACtB,IAAIC,gBAAgB;YACpB,KAAK,MAAMT,QAAQH,SAAS,KAAK,CAC7B,KAAK,MAAMa,QAAQV,KAAK,SAAS,CAAE;gBAC/BS;gBACA,IAAIC,AAAkB,wBAAlBA,KAAK,QAAQ,EAA0B;oBACvC,MAAMC,QAAQD,KAAK,KAAK;oBACxB,IAAIC,MAAM,MAAM,EACZJ,UAAU,GAAG,CAACI,MAAM,MAAM;gBAElC;YACJ;YAGJ,MAAMC,YAAiC;gBACnC,WAAW;uBAAIL;iBAAU;gBACzB,UAAUrB,QAAQ,QAAQ;gBAC1BuB;gBACA,YAAYZ,SAAS,KAAK,CAAC,MAAM;gBACjC,gBAAgBR,OAAO,IAAI,CAACJ;gBAC5B,mBAAmBM;YACvB;YAEA,OAAOX,OAAO,EAAE,CAAC;gBAAEyB;gBAAQO;YAAU;QACzC,EAAE,OAAOC,OAAO;YACZ,OAAOjC,OAAO,IAAI,CACd,IAAIC,MACA,CAAC,sBAAsB,EAAEgC,iBAAiBhC,QAAQgC,MAAM,OAAO,GAAGC,OAAOD,QAAQ;QAG7F;IACJ;AACJ;AAEO,MAAME,6CACTC,6BAA6B,oBAAoB,CAAC;IAC9C,gBAAgB7C;IAChB,cAAc;QACV8C;QACAC;QACAC;QACAC;QACAC;QACAC;KACH;AACL"}
+{"version":3,"file":"api/features/WbGeneratePageContent/WbGeneratePageContentUseCase.js","sources":["../../../../src/api/features/WbGeneratePageContent/WbGeneratePageContentUseCase.ts"],"sourcesContent":["import { stepCountIs } from \"ai\";\nimport { Result } from \"@webiny/feature/api\";\nimport { Ai } from \"@webiny/api-core/features/ai/index.js\";\nimport { AiSdkTools } from \"@webiny/api-core/features/ai/index.js\";\nimport { Encryption } from \"@webiny/api-core/features/encryption/index.js\";\nimport { ListTagsUseCase } from \"@webiny/api-file-manager/features/file/ListTags/index.js\";\nimport { GetSettingsUseCase } from \"~/api/features/GetSettings/index.js\";\nimport {\n    AiPromptContextBuilder,\n    formatAdditionalFilesContext\n} from \"~/api/features/AiPromptContext/index.js\";\nimport { createReadProjectFileTool } from \"~/api/features/AiPromptContext/ReadProjectFileTool.js\";\nimport { WbGeneratePageContentUseCase } from \"./abstractions.js\";\nimport type {\n    WbGeneratePageContentParams,\n    GeneratePageContentResult,\n    GenerationTelemetry\n} from \"./abstractions.js\";\nimport { buildDomainPrompt } from \"./buildPrompt.js\";\nimport { LlmJsonResponse } from \"./LlmJsonResponse.js\";\nimport { ComponentFilter } from \"./ComponentFilter.js\";\n\nclass WbGeneratePageContentUseCaseImpl implements WbGeneratePageContentUseCase.Interface {\n    constructor(\n        private promptContextBuilder: AiPromptContextBuilder.Interface,\n        private getSettings: GetSettingsUseCase.Interface,\n        private ai: Ai.Interface,\n        private aiSdkTools: AiSdkTools.Interface,\n        private encryption: Encryption.Interface,\n        private listTags: ListTagsUseCase.Interface\n    ) {}\n\n    async execute(\n        params: WbGeneratePageContentParams\n    ): Promise<Result<GeneratePageContentResult, Error>> {\n        const settingsResult = await this.getSettings.execute();\n        if (settingsResult.isFail()) {\n            return Result.fail(new Error(\"Failed to load AI PowerUps settings.\"));\n        }\n\n        const settings = settingsResult.value;\n        const firstProvider = settings.providers.presets[0];\n\n        if (!firstProvider) {\n            return Result.fail(\n                new Error(\"No AI provider configured. Add a provider in AI Power Ups settings.\")\n            );\n        }\n\n        const apiKey = await this.encryption.decrypt(firstProvider.apiKeyEncrypted);\n\n        const sdkTools = this.aiSdkTools.getToolSet();\n\n        const context = await this.promptContextBuilder.execute({\n            projectId: params.projectId,\n            readerPersonaId: params.readerPersonaId,\n            writerPersonaId: params.writerPersonaId,\n            excludedFileIds: params.excludedFileIds,\n            additionalFileIds: params.additionalFileIds\n        });\n\n        if (context.allProjectFiles.length > 0) {\n            const projectFileTool = createReadProjectFileTool(\n                context.allProjectFiles,\n                context.excludedFileIds\n            );\n            Object.assign(sdkTools, projectFileTool);\n        }\n\n        // Fetch image tags so the prompt can enumerate what's queryable; without\n        // this the model invents tag/IDs instead of calling listImagesByTag.\n        const tagsResult = await this.listTags.execute({\n            where: { type_startsWith: \"image/\" },\n            limit: 100\n        });\n        const imageTags = tagsResult.isOk() ? tagsResult.value.map(t => t.tag) : [];\n\n        const components = params.components as Array<{ name: string }>;\n        const systemText =\n            buildDomainPrompt(components, params.tools, imageTags) + context.toString();\n\n        const system = {\n            role: \"system\" as const,\n            content: systemText,\n            providerOptions: {\n                anthropic: {\n                    cacheControl: { type: \"ephemeral\" }\n                }\n            }\n        };\n\n        try {\n            const aiResult = await this.ai.generateText({\n                model: firstProvider.model,\n                connection: {\n                    sdkName: firstProvider.model.split(\"/\")[0],\n                    apiKey\n                },\n                system,\n                toolChoice: \"auto\",\n                prompt: params.prompt + formatAdditionalFilesContext(context.additionalFiles),\n                ...(Object.keys(sdkTools).length > 0\n                    ? { tools: sdkTools, stopWhen: stepCountIs(20) }\n                    : {})\n            });\n\n            const text =\n                aiResult.text ||\n                (aiResult.steps.filter(step => step.text.length > 0).pop()?.text ?? \"\");\n\n            const elements = LlmJsonResponse.fromRawText(text).toArray();\n            const componentFilter = new ComponentFilter(components);\n            const output = JSON.stringify(componentFilter.filter(elements));\n\n            const filesRead = new Set<string>();\n            let toolCallsMade = 0;\n            for (const step of aiResult.steps) {\n                for (const call of step.toolCalls) {\n                    toolCallsMade++;\n                    if (call.toolName === \"read_project_file\") {\n                        const input = call.input as { fileId?: string };\n                        if (input.fileId) {\n                            filesRead.add(input.fileId);\n                        }\n                    }\n                }\n            }\n\n            const telemetry: GenerationTelemetry = {\n                filesRead: [...filesRead],\n                cacheHit: context.cacheHit,\n                toolCallsMade,\n                totalSteps: aiResult.steps.length,\n                toolsAvailable: Object.keys(sdkTools),\n                imageTagsInPrompt: imageTags\n            };\n\n            return Result.ok({ output, telemetry });\n        } catch (error) {\n            return Result.fail(\n                new Error(\n                    `AI generation failed: ${error instanceof Error ? error.message : String(error)}`\n                )\n            );\n        }\n    }\n}\n\nexport const WbGeneratePageContentUseCaseImplementation =\n    WbGeneratePageContentUseCase.createImplementation({\n        implementation: WbGeneratePageContentUseCaseImpl,\n        dependencies: [\n            AiPromptContextBuilder,\n            GetSettingsUseCase,\n            Ai,\n            AiSdkTools,\n            Encryption,\n            ListTagsUseCase\n        ]\n    });\n"],"names":["WbGeneratePageContentUseCaseImpl","promptContextBuilder","getSettings","ai","aiSdkTools","encryption","listTags","params","settingsResult","Result","Error","settings","firstProvider","apiKey","sdkTools","context","projectFileTool","createReadProjectFileTool","Object","tagsResult","imageTags","t","components","systemText","buildDomainPrompt","system","aiResult","formatAdditionalFilesContext","stepCountIs","text","step","elements","LlmJsonResponse","componentFilter","ComponentFilter","output","JSON","filesRead","Set","toolCallsMade","call","input","telemetry","error","String","WbGeneratePageContentUseCaseImplementation","WbGeneratePageContentUseCase","AiPromptContextBuilder","GetSettingsUseCase","Ai","AiSdkTools","Encryption","ListTagsUseCase"],"mappings":";;;;;;;;;;;;AAsBA,MAAMA;IACF,YACYC,oBAAsD,EACtDC,WAAyC,EACzCC,EAAgB,EAChBC,UAAgC,EAChCC,UAAgC,EAChCC,QAAmC,CAC7C;aANUL,oBAAoB,GAApBA;aACAC,WAAW,GAAXA;aACAC,EAAE,GAAFA;aACAC,UAAU,GAAVA;aACAC,UAAU,GAAVA;aACAC,QAAQ,GAARA;IACT;IAEH,MAAM,QACFC,MAAmC,EACc;QACjD,MAAMC,iBAAiB,MAAM,IAAI,CAAC,WAAW,CAAC,OAAO;QACrD,IAAIA,eAAe,MAAM,IACrB,OAAOC,OAAO,IAAI,CAAC,IAAIC,MAAM;QAGjC,MAAMC,WAAWH,eAAe,KAAK;QACrC,MAAMI,gBAAgBD,SAAS,SAAS,CAAC,OAAO,CAAC,EAAE;QAEnD,IAAI,CAACC,eACD,OAAOH,OAAO,IAAI,CACd,IAAIC,MAAM;QAIlB,MAAMG,SAAS,MAAM,IAAI,CAAC,UAAU,CAAC,OAAO,CAACD,cAAc,eAAe;QAE1E,MAAME,WAAW,IAAI,CAAC,UAAU,CAAC,UAAU;QAE3C,MAAMC,UAAU,MAAM,IAAI,CAAC,oBAAoB,CAAC,OAAO,CAAC;YACpD,WAAWR,OAAO,SAAS;YAC3B,iBAAiBA,OAAO,eAAe;YACvC,iBAAiBA,OAAO,eAAe;YACvC,iBAAiBA,OAAO,eAAe;YACvC,mBAAmBA,OAAO,iBAAiB;QAC/C;QAEA,IAAIQ,QAAQ,eAAe,CAAC,MAAM,GAAG,GAAG;YACpC,MAAMC,kBAAkBC,0BACpBF,QAAQ,eAAe,EACvBA,QAAQ,eAAe;YAE3BG,OAAO,MAAM,CAACJ,UAAUE;QAC5B;QAIA,MAAMG,aAAa,MAAM,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC;YAC3C,OAAO;gBAAE,iBAAiB;YAAS;YACnC,OAAO;QACX;QACA,MAAMC,YAAYD,WAAW,IAAI,KAAKA,WAAW,KAAK,CAAC,GAAG,CAACE,CAAAA,IAAKA,EAAE,GAAG,IAAI,EAAE;QAE3E,MAAMC,aAAaf,OAAO,UAAU;QACpC,MAAMgB,aACFC,kBAAkBF,YAAYf,OAAO,KAAK,EAAEa,aAAaL,QAAQ,QAAQ;QAE7E,MAAMU,SAAS;YACX,MAAM;YACN,SAASF;YACT,iBAAiB;gBACb,WAAW;oBACP,cAAc;wBAAE,MAAM;oBAAY;gBACtC;YACJ;QACJ;QAEA,IAAI;YACA,MAAMG,WAAW,MAAM,IAAI,CAAC,EAAE,CAAC,YAAY,CAAC;gBACxC,OAAOd,cAAc,KAAK;gBAC1B,YAAY;oBACR,SAASA,cAAc,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE;oBAC1CC;gBACJ;gBACAY;gBACA,YAAY;gBACZ,QAAQlB,OAAO,MAAM,GAAGoB,6BAA6BZ,QAAQ,eAAe;gBAC5E,GAAIG,OAAO,IAAI,CAACJ,UAAU,MAAM,GAAG,IAC7B;oBAAE,OAAOA;oBAAU,UAAUc,YAAY;gBAAI,IAC7C,CAAC,CAAC;YACZ;YAEA,MAAMC,OACFH,SAAS,IAAI,IACZA,CAAAA,SAAS,KAAK,CAAC,MAAM,CAACI,CAAAA,OAAQA,KAAK,IAAI,CAAC,MAAM,GAAG,GAAG,GAAG,IAAI,QAAQ,EAAC;YAEzE,MAAMC,WAAWC,gBAAgB,WAAW,CAACH,MAAM,OAAO;YAC1D,MAAMI,kBAAkB,IAAIC,gBAAgBZ;YAC5C,MAAMa,SAASC,KAAK,SAAS,CAACH,gBAAgB,MAAM,CAACF;YAErD,MAAMM,YAAY,IAAIC;YACtB,IAAIC,gBAAgB;YACpB,KAAK,MAAMT,QAAQJ,SAAS,KAAK,CAC7B,KAAK,MAAMc,QAAQV,KAAK,SAAS,CAAE;gBAC/BS;gBACA,IAAIC,AAAkB,wBAAlBA,KAAK,QAAQ,EAA0B;oBACvC,MAAMC,QAAQD,KAAK,KAAK;oBACxB,IAAIC,MAAM,MAAM,EACZJ,UAAU,GAAG,CAACI,MAAM,MAAM;gBAElC;YACJ;YAGJ,MAAMC,YAAiC;gBACnC,WAAW;uBAAIL;iBAAU;gBACzB,UAAUtB,QAAQ,QAAQ;gBAC1BwB;gBACA,YAAYb,SAAS,KAAK,CAAC,MAAM;gBACjC,gBAAgBR,OAAO,IAAI,CAACJ;gBAC5B,mBAAmBM;YACvB;YAEA,OAAOX,OAAO,EAAE,CAAC;gBAAE0B;gBAAQO;YAAU;QACzC,EAAE,OAAOC,OAAO;YACZ,OAAOlC,OAAO,IAAI,CACd,IAAIC,MACA,CAAC,sBAAsB,EAAEiC,iBAAiBjC,QAAQiC,MAAM,OAAO,GAAGC,OAAOD,QAAQ;QAG7F;IACJ;AACJ;AAEO,MAAME,6CACTC,6BAA6B,oBAAoB,CAAC;IAC9C,gBAAgB9C;IAChB,cAAc;QACV+C;QACAC;QACAC;QACAC;QACAC;QACAC;KACH;AACL"}

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

@@ -7,6 +7,7 @@ export interface WbGeneratePageContentParams {
     excludedFileIds?: string[] | null;
     readerPersonaId?: string | null;
     writerPersonaId?: string | null;
+    additionalFileIds?: string[] | null;
 }
 export interface GenerationTelemetry {
     filesRead: string[];

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

@@ -1 +1 @@
-{"version":3,"file":"api/features/WbGeneratePageContent/abstractions.js","sources":["../../../../src/api/features/WbGeneratePageContent/abstractions.ts"],"sourcesContent":["import { createAbstraction, Result } from \"@webiny/feature/api\";\n\nexport interface WbGeneratePageContentParams {\n    prompt: string;\n    components: unknown;\n    tools: unknown;\n    projectId?: string | null;\n    excludedFileIds?: string[] | null;\n    readerPersonaId?: string | null;\n    writerPersonaId?: string | null;\n}\n\nexport interface GenerationTelemetry {\n    filesRead: string[];\n    cacheHit: boolean;\n    toolCallsMade: number;\n    totalSteps: number;\n    toolsAvailable: string[];\n    imageTagsInPrompt: string[];\n}\n\nexport interface GeneratePageContentResult {\n    output: string;\n    telemetry: GenerationTelemetry;\n}\n\nexport interface IWbGeneratePageContentUseCase {\n    execute(params: WbGeneratePageContentParams): Promise<Result<GeneratePageContentResult, Error>>;\n}\n\nexport const WbGeneratePageContentUseCase = createAbstraction<IWbGeneratePageContentUseCase>(\n    \"AiPowerUpsWbGeneratePageContentUseCase\"\n);\n\nexport namespace WbGeneratePageContentUseCase {\n    export type Interface = IWbGeneratePageContentUseCase;\n    export type Params = WbGeneratePageContentParams;\n}\n"],"names":["WbGeneratePageContentUseCase","createAbstraction"],"mappings":";AA8BO,MAAMA,+BAA+BC,kBACxC"}
+{"version":3,"file":"api/features/WbGeneratePageContent/abstractions.js","sources":["../../../../src/api/features/WbGeneratePageContent/abstractions.ts"],"sourcesContent":["import { createAbstraction, Result } from \"@webiny/feature/api\";\n\nexport interface WbGeneratePageContentParams {\n    prompt: string;\n    components: unknown;\n    tools: unknown;\n    projectId?: string | null;\n    excludedFileIds?: string[] | null;\n    readerPersonaId?: string | null;\n    writerPersonaId?: string | null;\n    additionalFileIds?: string[] | null;\n}\n\nexport interface GenerationTelemetry {\n    filesRead: string[];\n    cacheHit: boolean;\n    toolCallsMade: number;\n    totalSteps: number;\n    toolsAvailable: string[];\n    imageTagsInPrompt: string[];\n}\n\nexport interface GeneratePageContentResult {\n    output: string;\n    telemetry: GenerationTelemetry;\n}\n\nexport interface IWbGeneratePageContentUseCase {\n    execute(params: WbGeneratePageContentParams): Promise<Result<GeneratePageContentResult, Error>>;\n}\n\nexport const WbGeneratePageContentUseCase = createAbstraction<IWbGeneratePageContentUseCase>(\n    \"AiPowerUpsWbGeneratePageContentUseCase\"\n);\n\nexport namespace WbGeneratePageContentUseCase {\n    export type Interface = IWbGeneratePageContentUseCase;\n    export type Params = WbGeneratePageContentParams;\n}\n"],"names":["WbGeneratePageContentUseCase","createAbstraction"],"mappings":";AA+BO,MAAMA,+BAA+BC,kBACxC"}

package/api/features/WbTranslatePage/LexicalParser.d.ts

@@ -0,0 +1,10 @@
+import { LexicalParser as Abstraction } from "./abstractions/LexicalParser.js";
+declare class LexicalParserImpl implements Abstraction.Interface {
+    private WindowClass;
+    parse(html: string): Promise<Record<string, unknown> | null>;
+    private loadWindow;
+}
+export declare const LexicalParser: typeof LexicalParserImpl & {
+    __abstraction: import("@webiny/di").Abstraction<import("./abstractions/LexicalParser.js").ILexicalParser>;
+};
+export {};

package/api/features/WbTranslatePage/LexicalParser.js

@@ -0,0 +1,37 @@
+import { createHtmlToLexicalParser } from "@webiny/lexical-converter";
+import { LexicalParser } from "./abstractions/LexicalParser.js";
+const parser = createHtmlToLexicalParser();
+class LexicalParserImpl {
+    async parse(html) {
+        try {
+            const Window = await this.loadWindow();
+            const window = new Window();
+            const document = window.document;
+            document.body.innerHTML = html;
+            const result = parser(document);
+            await window.happyDOM.close();
+            return result;
+        } catch (error) {
+            console.error("[LexicalParser] Failed to parse HTML to Lexical state:", error);
+            console.error("[LexicalParser] Input HTML:", html);
+            return null;
+        }
+    }
+    async loadWindow() {
+        if (!this.WindowClass) {
+            const { Window } = await import("happy-dom");
+            this.WindowClass = Window;
+        }
+        return this.WindowClass;
+    }
+    constructor(){
+        this.WindowClass = null;
+    }
+}
+const LexicalParser_LexicalParser = LexicalParser.createImplementation({
+    implementation: LexicalParserImpl,
+    dependencies: []
+});
+export { LexicalParser_LexicalParser as LexicalParser };
+
+//# sourceMappingURL=LexicalParser.js.map