@webiny/ai-powerups 6.4.11 → 6.6.0-alpha.1

package/api/features/WbGeneratePageContent/WbGeneratePageContentTask.js

@@ -1,14 +1,14 @@
 import { TaskDefinition } from "@webiny/api-core/features/task/TaskDefinition/index.js";
-import { WebsocketService } from "@webiny/api-websockets/features/WebsocketService/index.js";
+import { WebsocketsSendToIdentityUseCase } from "@webiny/api-websockets/features/SendToIdentity/abstractions.js";
 import { compress } from "@webiny/utils/features/compression/legacy/gzip.js";
 import { WbGeneratePageContentUseCase } from "./index.js";
 import { IdentityContext } from "@webiny/api-core/exports/api/security.js";
 const WB_GENERATE_PAGE_CONTENT_TASK_ID = "aiPowerUpsGeneratePageContent";
 class WbGeneratePageContentTaskImpl {
-    constructor(identityContext, generatePageContent, websocketService){
+    constructor(identityContext, generatePageContent, sendToIdentity){
         this.identityContext = identityContext;
         this.generatePageContent = generatePageContent;
-        this.websocketService = websocketService;
+        this.sendToIdentity = sendToIdentity;
         this.id = WB_GENERATE_PAGE_CONTENT_TASK_ID;
         this.title = "AI Power-Ups - Generate Page Content";
         this.description = "Generates page content using AI based on a user prompt.";
@@ -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({
@@ -40,7 +50,7 @@ class WbGeneratePageContentTaskImpl {
         return controller.response.done("Page content generated successfully.");
     }
     async sendContentToUser(identityId, payload, telemetry) {
-        await this.websocketService.send({
+        await this.sendToIdentity.execute({
             id: identityId
         }, {
             action: "aiPowerUps.generatePageContent.content",
@@ -52,7 +62,7 @@ class WbGeneratePageContentTaskImpl {
         });
     }
     async sendErrorToUser(identityId, message) {
-        await this.websocketService.send({
+        await this.sendToIdentity.execute({
             id: identityId
         }, {
             action: "aiPowerUps.generatePageContent.error",
@@ -67,7 +77,7 @@ const WbGeneratePageContentTask = TaskDefinition.createImplementation({
     dependencies: [
         IdentityContext,
         WbGeneratePageContentUseCase,
-        WebsocketService
+        WebsocketsSendToIdentityUseCase
     ]
 });
 export { WB_GENERATE_PAGE_CONTENT_TASK_ID, WbGeneratePageContentTask };

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 { WebsocketService } from \"@webiny/api-websockets/features/WebsocketService/index.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 websocketService: WebsocketService.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.websocketService.send(\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.websocketService.send(\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, WebsocketService]\n});\n"],"names":["WB_GENERATE_PAGE_CONTENT_TASK_ID","WbGeneratePageContentTaskImpl","identityContext","generatePageContent","websocketService","input","controller","result","identity","compressed","compress","payload","identityId","telemetry","message","WbGeneratePageContentTask","TaskDefinition","IdentityContext","WbGeneratePageContentUseCase","WebsocketService"],"mappings":";;;;;AAOO,MAAMA,mCAAmC;AAYhD,MAAMC;IAQF,YACYC,eAA0C,EAC1CC,mBAA2D,EAC3DC,gBAA4C,CACtD;aAHUF,eAAe,GAAfA;aACAC,mBAAmB,GAAnBA;aACAC,gBAAgB,GAAhBA;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,gBAAgB,CAAC,IAAI,CAC5B;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,gBAAgB,CAAC,IAAI,CAC5B;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;KAAiB;AACnF"}
+{"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

package/api/features/WbTranslatePage/LexicalParser.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"api/features/WbTranslatePage/LexicalParser.js","sources":["../../../../src/api/features/WbTranslatePage/LexicalParser.ts"],"sourcesContent":["import { createHtmlToLexicalParser } from \"@webiny/lexical-converter\";\nimport { LexicalParser as Abstraction } from \"./abstractions/LexicalParser.js\";\n\nconst parser = createHtmlToLexicalParser();\n\nclass LexicalParserImpl implements Abstraction.Interface {\n    private WindowClass: typeof import(\"happy-dom\").Window | null = null;\n\n    async parse(html: string): Promise<Record<string, unknown> | null> {\n        try {\n            const Window = await this.loadWindow();\n            const window = new Window();\n            const document = window.document;\n            document.body.innerHTML = html;\n            const result = parser(document as unknown as Document);\n            await window.happyDOM.close();\n            return result;\n        } catch (error) {\n            console.error(\"[LexicalParser] Failed to parse HTML to Lexical state:\", error);\n            console.error(\"[LexicalParser] Input HTML:\", html);\n            return null;\n        }\n    }\n\n    private async loadWindow() {\n        if (!this.WindowClass) {\n            const { Window } = await import(\"happy-dom\");\n            this.WindowClass = Window;\n        }\n        return this.WindowClass;\n    }\n}\n\nexport const LexicalParser = Abstraction.createImplementation({\n    implementation: LexicalParserImpl,\n    dependencies: []\n});\n"],"names":["parser","createHtmlToLexicalParser","LexicalParserImpl","html","Window","window","document","result","error","console","LexicalParser","Abstraction"],"mappings":";;AAGA,MAAMA,SAASC;AAEf,MAAMC;IAGF,MAAM,MAAMC,IAAY,EAA2C;QAC/D,IAAI;YACA,MAAMC,SAAS,MAAM,IAAI,CAAC,UAAU;YACpC,MAAMC,SAAS,IAAID;YACnB,MAAME,WAAWD,OAAO,QAAQ;YAChCC,SAAS,IAAI,CAAC,SAAS,GAAGH;YAC1B,MAAMI,SAASP,OAAOM;YACtB,MAAMD,OAAO,QAAQ,CAAC,KAAK;YAC3B,OAAOE;QACX,EAAE,OAAOC,OAAO;YACZC,QAAQ,KAAK,CAAC,0DAA0DD;YACxEC,QAAQ,KAAK,CAAC,+BAA+BN;YAC7C,OAAO;QACX;IACJ;IAEA,MAAc,aAAa;QACvB,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE;YACnB,MAAM,EAAEC,MAAM,EAAE,GAAG,MAAM,MAAM,CAAC;YAChC,IAAI,CAAC,WAAW,GAAGA;QACvB;QACA,OAAO,IAAI,CAAC,WAAW;IAC3B;;aAxBQ,WAAW,GAA6C;;AAyBpE;AAEO,MAAMM,8BAAgBC,cAAAA,oBAAgC,CAAC;IAC1D,gBAAgBT;IAChB,cAAc,EAAE;AACpB"}

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

@@ -0,0 +1,27 @@
+import { Ai } from "@webiny/api-core/features/ai/index.js";
+import { Encryption } from "@webiny/api-core/features/encryption/index.js";
+import { GetDefaultLanguageUseCase } from "@webiny/languages/exports/api/languages.js";
+import { TranslatePageUseCase } from "@webiny/api-website-builder/features/pages/TranslatePage/index.js";
+import { UpdatePageRepository } from "@webiny/api-website-builder/features/pages/UpdatePage/abstractions.js";
+import { GetSettingsUseCase } from "../../../api/features/GetSettings/index.js";
+import { LexicalParser } from "./abstractions/LexicalParser.js";
+declare class WbTranslatePageDecoratorImpl implements TranslatePageUseCase.Interface {
+    private getDefaultLanguage;
+    private getSettings;
+    private ai;
+    private encryption;
+    private lexicalParser;
+    private updatePageRepository;
+    private decoratee;
+    constructor(getDefaultLanguage: GetDefaultLanguageUseCase.Interface, getSettings: GetSettingsUseCase.Interface, ai: Ai.Interface, encryption: Encryption.Interface, lexicalParser: LexicalParser.Interface, updatePageRepository: UpdatePageRepository.Interface, decoratee: TranslatePageUseCase.Interface);
+    execute(params: TranslatePageUseCase.Params): TranslatePageUseCase.Return;
+    private resolveSourceLanguage;
+    private translate;
+    private extractBindings;
+    private slugify;
+    private buildUpdateData;
+}
+export declare const WbTranslatePageDecorator: typeof WbTranslatePageDecoratorImpl & {
+    __abstraction: import("@webiny/di").Abstraction<import("@webiny/api-website-builder/features/pages/TranslatePage/abstractions.js").ITranslatePageUseCase>;
+};
+export {};

package/api/features/WbTranslatePage/WbTranslatePageDecorator.js

@@ -0,0 +1,134 @@
+import set from "lodash/set";
+import { Result } from "@webiny/feature/api";
+import { Ai } from "@webiny/api-core/features/ai/index.js";
+import { Encryption } from "@webiny/api-core/features/encryption/index.js";
+import { GetDefaultLanguageUseCase } from "@webiny/languages/exports/api/languages.js";
+import { TranslatePageUseCase } from "@webiny/api-website-builder/features/pages/TranslatePage/index.js";
+import { UpdatePageRepository } from "@webiny/api-website-builder/features/pages/UpdatePage/abstractions.js";
+import { GetSettingsUseCase } from "../GetSettings/index.js";
+import { LexicalParser } from "./abstractions/LexicalParser.js";
+class WbTranslatePageDecoratorImpl {
+    constructor(getDefaultLanguage, getSettings, ai, encryption, lexicalParser, updatePageRepository, decoratee){
+        this.getDefaultLanguage = getDefaultLanguage;
+        this.getSettings = getSettings;
+        this.ai = ai;
+        this.encryption = encryption;
+        this.lexicalParser = lexicalParser;
+        this.updatePageRepository = updatePageRepository;
+        this.decoratee = decoratee;
+    }
+    async execute(params) {
+        const result = await this.decoratee.execute(params);
+        if (result.isFail()) return result;
+        const page = result.value;
+        const sourceLanguage = await this.resolveSourceLanguage(page);
+        const translations = await this.translate(page, sourceLanguage, params.languageCode);
+        if (!translations) return Result.ok(page);
+        const updateData = await this.buildUpdateData(page, translations);
+        const updateResult = await this.updatePageRepository.execute(page.id, updateData);
+        if (updateResult.isFail()) return Result.ok(page);
+        return Result.ok(updateResult.value);
+    }
+    async resolveSourceLanguage(page) {
+        if (page.properties["sourcePage"]) return page.properties["language"] ?? "en";
+        const result = await this.getDefaultLanguage.execute();
+        if (result.isOk()) return result.value.code;
+        return "en";
+    }
+    async translate(page, sourceLanguage, targetLanguage) {
+        const settingsResult = await this.getSettings.execute();
+        if (settingsResult.isFail()) return null;
+        const settings = settingsResult.value;
+        const firstProvider = settings.providers.presets[0];
+        if (!firstProvider) return null;
+        const apiKey = await this.encryption.decrypt(firstProvider.apiKeyEncrypted);
+        const properties = {
+            title: page.properties["title"],
+            snippet: page.properties["snippet"],
+            pathSlug: page.properties["title"]
+        };
+        const bindings = this.extractBindings(page.bindings);
+        const input = JSON.stringify({
+            properties,
+            bindings
+        });
+        const result = await this.ai.generateText({
+            model: firstProvider.model,
+            connection: {
+                sdkName: firstProvider.model.split("/")[0],
+                apiKey
+            },
+            system: `You are a professional translator. Translate all user-provided text to language code "${targetLanguage}", preserving placeholders and formatting. Return a JSON object with the same keys, only changing the values.`,
+            prompt: `Translate given key-value pairs from "${sourceLanguage}" to language code "${targetLanguage}". Do not modify the keys. "properties" is a simple key-value pair. "bindings" values are located in the "value" key. ${input}`,
+            temperature: 0.3
+        });
+        try {
+            return JSON.parse(result.text);
+        } catch  {
+            return null;
+        }
+    }
+    extractBindings(bindings) {
+        const result = {};
+        for(const elementId in bindings){
+            const element = bindings[elementId];
+            const inputs = element.inputs;
+            if (inputs) for(const inputName in inputs){
+                const input = inputs[inputName];
+                if ("text" === input.type || "longText" === input.type) result[`${elementId}.inputs.${inputName}`] = {
+                    type: input.type,
+                    value: input.static
+                };
+                else if ("lexical" === input.type && input.static?.html) result[`${elementId}.inputs.${inputName}`] = {
+                    type: input.type,
+                    value: input.static.html
+                };
+            }
+        }
+        return result;
+    }
+    slugify(text) {
+        return text.toLowerCase().normalize("NFKD").replace(/[̀-ͯ]/g, "").replace(/[^a-z0-9]+/g, "-").replace(/^-+|-+$/g, "");
+    }
+    async buildUpdateData(page, translations) {
+        const properties = {
+            ...page.properties
+        };
+        const bindings = structuredClone(page.bindings);
+        const translatedPathSlug = translations.properties["pathSlug"];
+        delete translations.properties["pathSlug"];
+        for (const [key, value] of Object.entries(translations.properties))properties[key] = value;
+        if (translatedPathSlug) {
+            const slug = this.slugify(translatedPathSlug);
+            if (slug) {
+                const languageCode = page.properties["language"];
+                properties["path"] = `/${languageCode}/${slug}`;
+            }
+        }
+        for (const [key, binding] of Object.entries(translations.bindings))if ("lexical" === binding.type) {
+            const state = await this.lexicalParser.parse(binding.value);
+            if (state) {
+                set(bindings, `${key}.static.state`, JSON.stringify(state));
+                set(bindings, `${key}.static.html`, binding.value);
+            }
+        } else set(bindings, `${key}.static`, binding.value);
+        return {
+            properties,
+            bindings
+        };
+    }
+}
+const WbTranslatePageDecorator = TranslatePageUseCase.createDecorator({
+    decorator: WbTranslatePageDecoratorImpl,
+    dependencies: [
+        GetDefaultLanguageUseCase,
+        GetSettingsUseCase,
+        Ai,
+        Encryption,
+        LexicalParser,
+        UpdatePageRepository
+    ]
+});
+export { WbTranslatePageDecorator };
+
+//# sourceMappingURL=WbTranslatePageDecorator.js.map

package/api/features/WbTranslatePage/WbTranslatePageDecorator.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"api/features/WbTranslatePage/WbTranslatePageDecorator.js","sources":["../../../../src/api/features/WbTranslatePage/WbTranslatePageDecorator.ts"],"sourcesContent":["import set from \"lodash/set\";\nimport { Result } from \"@webiny/feature/api\";\nimport { Ai } from \"@webiny/api-core/features/ai/index.js\";\nimport { Encryption } from \"@webiny/api-core/features/encryption/index.js\";\nimport { GetDefaultLanguageUseCase } from \"@webiny/languages/exports/api/languages.js\";\nimport { TranslatePageUseCase } from \"@webiny/api-website-builder/features/pages/TranslatePage/index.js\";\nimport { UpdatePageRepository } from \"@webiny/api-website-builder/features/pages/UpdatePage/abstractions.js\";\nimport type { WbPage } from \"@webiny/api-website-builder/domain/page/abstractions.js\";\nimport { GetSettingsUseCase } from \"~/api/features/GetSettings/index.js\";\nimport { LexicalParser } from \"./abstractions/LexicalParser.js\";\n\ntype InputType = \"text\" | \"longText\" | \"lexical\";\n\ninterface TranslatableBinding {\n    type: InputType;\n    value: string;\n}\n\ntype TranslatableBindings = Record<string, TranslatableBinding>;\n\ninterface TranslatedBindings {\n    [key: string]: { type: InputType; value: string };\n}\n\ninterface TranslatedData {\n    properties: Record<string, string>;\n    bindings: TranslatedBindings;\n}\n\nclass WbTranslatePageDecoratorImpl implements TranslatePageUseCase.Interface {\n    constructor(\n        private getDefaultLanguage: GetDefaultLanguageUseCase.Interface,\n        private getSettings: GetSettingsUseCase.Interface,\n        private ai: Ai.Interface,\n        private encryption: Encryption.Interface,\n        private lexicalParser: LexicalParser.Interface,\n        private updatePageRepository: UpdatePageRepository.Interface,\n        private decoratee: TranslatePageUseCase.Interface\n    ) {}\n\n    async execute(params: TranslatePageUseCase.Params): TranslatePageUseCase.Return {\n        const result = await this.decoratee.execute(params);\n        if (result.isFail()) {\n            return result;\n        }\n\n        const page = result.value;\n        const sourceLanguage = await this.resolveSourceLanguage(page);\n\n        const translations = await this.translate(page, sourceLanguage, params.languageCode);\n        if (!translations) {\n            return Result.ok(page);\n        }\n\n        const updateData = await this.buildUpdateData(page, translations);\n        const updateResult = await this.updatePageRepository.execute(page.id, updateData);\n        if (updateResult.isFail()) {\n            return Result.ok(page);\n        }\n\n        return Result.ok(updateResult.value);\n    }\n\n    private async resolveSourceLanguage(page: WbPage): Promise<string> {\n        if (page.properties[\"sourcePage\"]) {\n            return page.properties[\"language\"] ?? \"en\";\n        }\n\n        const result = await this.getDefaultLanguage.execute();\n        if (result.isOk()) {\n            return result.value.code;\n        }\n\n        return \"en\";\n    }\n\n    private async translate(\n        page: WbPage,\n        sourceLanguage: string,\n        targetLanguage: string\n    ): Promise<TranslatedData | null> {\n        const settingsResult = await this.getSettings.execute();\n        if (settingsResult.isFail()) {\n            return null;\n        }\n\n        const settings = settingsResult.value;\n        const firstProvider = settings.providers.presets[0];\n\n        if (!firstProvider) {\n            return null;\n        }\n\n        const apiKey = await this.encryption.decrypt(firstProvider.apiKeyEncrypted);\n\n        const properties = {\n            title: page.properties[\"title\"],\n            snippet: page.properties[\"snippet\"],\n            pathSlug: page.properties[\"title\"]\n        };\n        const bindings = this.extractBindings(page.bindings);\n        const input = JSON.stringify({ properties, bindings });\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: `You are a professional translator. Translate all user-provided text to language code \"${targetLanguage}\", preserving placeholders and formatting. Return a JSON object with the same keys, only changing the values.`,\n            prompt: `Translate given key-value pairs from \"${sourceLanguage}\" to language code \"${targetLanguage}\". Do not modify the keys. \"properties\" is a simple key-value pair. \"bindings\" values are located in the \"value\" key. ${input}`,\n            temperature: 0.3\n        });\n\n        try {\n            return JSON.parse(result.text);\n        } catch {\n            return null;\n        }\n    }\n\n    private extractBindings(bindings: Record<string, any>): TranslatableBindings {\n        const result: TranslatableBindings = {};\n\n        for (const elementId in bindings) {\n            const element = bindings[elementId];\n            const inputs = element.inputs;\n            if (!inputs) {\n                continue;\n            }\n\n            for (const inputName in inputs) {\n                const input = inputs[inputName];\n                if (input.type === \"text\" || input.type === \"longText\") {\n                    result[`${elementId}.inputs.${inputName}`] = {\n                        type: input.type,\n                        value: input.static\n                    };\n                } else if (input.type === \"lexical\" && input.static?.html) {\n                    result[`${elementId}.inputs.${inputName}`] = {\n                        type: input.type,\n                        value: input.static.html\n                    };\n                }\n            }\n        }\n\n        return result;\n    }\n\n    private slugify(text: string): string {\n        return text\n            .toLowerCase()\n            .normalize(\"NFKD\")\n            .replace(/[̀-ͯ]/g, \"\")\n            .replace(/[^a-z0-9]+/g, \"-\")\n            .replace(/^-+|-+$/g, \"\");\n    }\n\n    private async buildUpdateData(\n        page: WbPage,\n        translations: TranslatedData\n    ): Promise<{ properties: Record<string, any>; bindings: Record<string, any> }> {\n        const properties: Record<string, any> = { ...page.properties };\n        const bindings: Record<string, any> = structuredClone(page.bindings);\n\n        const translatedPathSlug = translations.properties[\"pathSlug\"];\n        delete translations.properties[\"pathSlug\"];\n\n        for (const [key, value] of Object.entries(translations.properties)) {\n            properties[key] = value;\n        }\n\n        if (translatedPathSlug) {\n            const slug = this.slugify(translatedPathSlug);\n            if (slug) {\n                const languageCode = page.properties[\"language\"];\n                properties[\"path\"] = `/${languageCode}/${slug}`;\n            }\n        }\n\n        for (const [key, binding] of Object.entries(translations.bindings)) {\n            if (binding.type === \"lexical\") {\n                const state = await this.lexicalParser.parse(binding.value);\n                if (state) {\n                    set(bindings, `${key}.static.state`, JSON.stringify(state));\n                    set(bindings, `${key}.static.html`, binding.value);\n                }\n            } else {\n                set(bindings, `${key}.static`, binding.value);\n            }\n        }\n\n        return { properties, bindings };\n    }\n}\n\nexport const WbTranslatePageDecorator = TranslatePageUseCase.createDecorator({\n    decorator: WbTranslatePageDecoratorImpl,\n    dependencies: [\n        GetDefaultLanguageUseCase,\n        GetSettingsUseCase,\n        Ai,\n        Encryption,\n        LexicalParser,\n        UpdatePageRepository\n    ]\n});\n"],"names":["WbTranslatePageDecoratorImpl","getDefaultLanguage","getSettings","ai","encryption","lexicalParser","updatePageRepository","decoratee","params","result","page","sourceLanguage","translations","Result","updateData","updateResult","targetLanguage","settingsResult","settings","firstProvider","apiKey","properties","bindings","input","JSON","elementId","element","inputs","inputName","text","structuredClone","translatedPathSlug","key","value","Object","slug","languageCode","binding","state","set","WbTranslatePageDecorator","TranslatePageUseCase","GetDefaultLanguageUseCase","GetSettingsUseCase","Ai","Encryption","LexicalParser","UpdatePageRepository"],"mappings":";;;;;;;;;AA6BA,MAAMA;IACF,YACYC,kBAAuD,EACvDC,WAAyC,EACzCC,EAAgB,EAChBC,UAAgC,EAChCC,aAAsC,EACtCC,oBAAoD,EACpDC,SAAyC,CACnD;aAPUN,kBAAkB,GAAlBA;aACAC,WAAW,GAAXA;aACAC,EAAE,GAAFA;aACAC,UAAU,GAAVA;aACAC,aAAa,GAAbA;aACAC,oBAAoB,GAApBA;aACAC,SAAS,GAATA;IACT;IAEH,MAAM,QAAQC,MAAmC,EAA+B;QAC5E,MAAMC,SAAS,MAAM,IAAI,CAAC,SAAS,CAAC,OAAO,CAACD;QAC5C,IAAIC,OAAO,MAAM,IACb,OAAOA;QAGX,MAAMC,OAAOD,OAAO,KAAK;QACzB,MAAME,iBAAiB,MAAM,IAAI,CAAC,qBAAqB,CAACD;QAExD,MAAME,eAAe,MAAM,IAAI,CAAC,SAAS,CAACF,MAAMC,gBAAgBH,OAAO,YAAY;QACnF,IAAI,CAACI,cACD,OAAOC,OAAO,EAAE,CAACH;QAGrB,MAAMI,aAAa,MAAM,IAAI,CAAC,eAAe,CAACJ,MAAME;QACpD,MAAMG,eAAe,MAAM,IAAI,CAAC,oBAAoB,CAAC,OAAO,CAACL,KAAK,EAAE,EAAEI;QACtE,IAAIC,aAAa,MAAM,IACnB,OAAOF,OAAO,EAAE,CAACH;QAGrB,OAAOG,OAAO,EAAE,CAACE,aAAa,KAAK;IACvC;IAEA,MAAc,sBAAsBL,IAAY,EAAmB;QAC/D,IAAIA,KAAK,UAAU,CAAC,aAAa,EAC7B,OAAOA,KAAK,UAAU,CAAC,WAAW,IAAI;QAG1C,MAAMD,SAAS,MAAM,IAAI,CAAC,kBAAkB,CAAC,OAAO;QACpD,IAAIA,OAAO,IAAI,IACX,OAAOA,OAAO,KAAK,CAAC,IAAI;QAG5B,OAAO;IACX;IAEA,MAAc,UACVC,IAAY,EACZC,cAAsB,EACtBK,cAAsB,EACQ;QAC9B,MAAMC,iBAAiB,MAAM,IAAI,CAAC,WAAW,CAAC,OAAO;QACrD,IAAIA,eAAe,MAAM,IACrB,OAAO;QAGX,MAAMC,WAAWD,eAAe,KAAK;QACrC,MAAME,gBAAgBD,SAAS,SAAS,CAAC,OAAO,CAAC,EAAE;QAEnD,IAAI,CAACC,eACD,OAAO;QAGX,MAAMC,SAAS,MAAM,IAAI,CAAC,UAAU,CAAC,OAAO,CAACD,cAAc,eAAe;QAE1E,MAAME,aAAa;YACf,OAAOX,KAAK,UAAU,CAAC,QAAQ;YAC/B,SAASA,KAAK,UAAU,CAAC,UAAU;YACnC,UAAUA,KAAK,UAAU,CAAC,QAAQ;QACtC;QACA,MAAMY,WAAW,IAAI,CAAC,eAAe,CAACZ,KAAK,QAAQ;QACnD,MAAMa,QAAQC,KAAK,SAAS,CAAC;YAAEH;YAAYC;QAAS;QAEpD,MAAMb,SAAS,MAAM,IAAI,CAAC,EAAE,CAAC,YAAY,CAAC;YACtC,OAAOU,cAAc,KAAK;YAC1B,YAAY;gBACR,SAASA,cAAc,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE;gBAC1CC;YACJ;YACA,QAAQ,CAAC,sFAAsF,EAAEJ,eAAe,6GAA6G,CAAC;YAC9N,QAAQ,CAAC,sCAAsC,EAAEL,eAAe,oBAAoB,EAAEK,eAAe,sHAAsH,EAAEO,OAAO;YACpO,aAAa;QACjB;QAEA,IAAI;YACA,OAAOC,KAAK,KAAK,CAACf,OAAO,IAAI;QACjC,EAAE,OAAM;YACJ,OAAO;QACX;IACJ;IAEQ,gBAAgBa,QAA6B,EAAwB;QACzE,MAAMb,SAA+B,CAAC;QAEtC,IAAK,MAAMgB,aAAaH,SAAU;YAC9B,MAAMI,UAAUJ,QAAQ,CAACG,UAAU;YACnC,MAAME,SAASD,QAAQ,MAAM;YAC7B,IAAKC,QAIL,IAAK,MAAMC,aAAaD,OAAQ;gBAC5B,MAAMJ,QAAQI,MAAM,CAACC,UAAU;gBAC/B,IAAIL,AAAe,WAAfA,MAAM,IAAI,IAAeA,AAAe,eAAfA,MAAM,IAAI,EACnCd,MAAM,CAAC,GAAGgB,UAAU,QAAQ,EAAEG,WAAW,CAAC,GAAG;oBACzC,MAAML,MAAM,IAAI;oBAChB,OAAOA,MAAM,MAAM;gBACvB;qBACG,IAAIA,AAAe,cAAfA,MAAM,IAAI,IAAkBA,MAAM,MAAM,EAAE,MACjDd,MAAM,CAAC,GAAGgB,UAAU,QAAQ,EAAEG,WAAW,CAAC,GAAG;oBACzC,MAAML,MAAM,IAAI;oBAChB,OAAOA,MAAM,MAAM,CAAC,IAAI;gBAC5B;YAER;QACJ;QAEA,OAAOd;IACX;IAEQ,QAAQoB,IAAY,EAAU;QAClC,OAAOA,KACF,WAAW,GACX,SAAS,CAAC,QACV,OAAO,CAAC,UAAU,IAClB,OAAO,CAAC,eAAe,KACvB,OAAO,CAAC,YAAY;IAC7B;IAEA,MAAc,gBACVnB,IAAY,EACZE,YAA4B,EAC+C;QAC3E,MAAMS,aAAkC;YAAE,GAAGX,KAAK,UAAU;QAAC;QAC7D,MAAMY,WAAgCQ,gBAAgBpB,KAAK,QAAQ;QAEnE,MAAMqB,qBAAqBnB,aAAa,UAAU,CAAC,WAAW;QAC9D,OAAOA,aAAa,UAAU,CAAC,WAAW;QAE1C,KAAK,MAAM,CAACoB,KAAKC,MAAM,IAAIC,OAAO,OAAO,CAACtB,aAAa,UAAU,EAC7DS,UAAU,CAACW,IAAI,GAAGC;QAGtB,IAAIF,oBAAoB;YACpB,MAAMI,OAAO,IAAI,CAAC,OAAO,CAACJ;YAC1B,IAAII,MAAM;gBACN,MAAMC,eAAe1B,KAAK,UAAU,CAAC,WAAW;gBAChDW,UAAU,CAAC,OAAO,GAAG,CAAC,CAAC,EAAEe,aAAa,CAAC,EAAED,MAAM;YACnD;QACJ;QAEA,KAAK,MAAM,CAACH,KAAKK,QAAQ,IAAIH,OAAO,OAAO,CAACtB,aAAa,QAAQ,EAC7D,IAAIyB,AAAiB,cAAjBA,QAAQ,IAAI,EAAgB;YAC5B,MAAMC,QAAQ,MAAM,IAAI,CAAC,aAAa,CAAC,KAAK,CAACD,QAAQ,KAAK;YAC1D,IAAIC,OAAO;gBACPC,IAAIjB,UAAU,GAAGU,IAAI,aAAa,CAAC,EAAER,KAAK,SAAS,CAACc;gBACpDC,IAAIjB,UAAU,GAAGU,IAAI,YAAY,CAAC,EAAEK,QAAQ,KAAK;YACrD;QACJ,OACIE,IAAIjB,UAAU,GAAGU,IAAI,OAAO,CAAC,EAAEK,QAAQ,KAAK;QAIpD,OAAO;YAAEhB;YAAYC;QAAS;IAClC;AACJ;AAEO,MAAMkB,2BAA2BC,qBAAqB,eAAe,CAAC;IACzE,WAAWzC;IACX,cAAc;QACV0C;QACAC;QACAC;QACAC;QACAC;QACAC;KACH;AACL"}

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

@@ -0,0 +1,7 @@
+export interface ILexicalParser {
+    parse(html: string): Promise<Record<string, unknown> | null>;
+}
+export declare const LexicalParser: import("@webiny/di").Abstraction<ILexicalParser>;
+export declare namespace LexicalParser {
+    type Interface = ILexicalParser;
+}

package/api/features/WbTranslatePage/abstractions/LexicalParser.js

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

package/api/features/WbTranslatePage/abstractions/LexicalParser.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"api/features/WbTranslatePage/abstractions/LexicalParser.js","sources":["../../../../../src/api/features/WbTranslatePage/abstractions/LexicalParser.ts"],"sourcesContent":["import { createAbstraction } from \"@webiny/feature/api\";\n\nexport interface ILexicalParser {\n    parse(html: string): Promise<Record<string, unknown> | null>;\n}\n\nexport const LexicalParser = createAbstraction<ILexicalParser>(\n    \"AiPowerUps/WbTranslatePage/LexicalParser\"\n);\n\nexport namespace LexicalParser {\n    export type Interface = ILexicalParser;\n}\n"],"names":["LexicalParser","createAbstraction"],"mappings":";AAMO,MAAMA,gBAAgBC,kBACzB"}

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

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

package/api/features/WbTranslatePage/feature.js

@@ -0,0 +1,13 @@
+import { createFeature } from "@webiny/feature/api";
+import { WbTranslatePageDecorator } from "./WbTranslatePageDecorator.js";
+import { LexicalParser } from "./LexicalParser.js";
+const WbTranslatePageFeature = createFeature({
+    name: "AiPowerUps/WbTranslatePage",
+    register (container) {
+        container.register(LexicalParser).inSingletonScope();
+        container.registerDecorator(WbTranslatePageDecorator);
+    }
+});
+export { WbTranslatePageFeature };
+
+//# sourceMappingURL=feature.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"api/features/WbTranslatePage/feature.js","sources":["../../../../src/api/features/WbTranslatePage/feature.ts"],"sourcesContent":["import { createFeature } from \"@webiny/feature/api\";\nimport { WbTranslatePageDecorator } from \"./WbTranslatePageDecorator.js\";\nimport { LexicalParser } from \"./LexicalParser.js\";\n\nexport const WbTranslatePageFeature = createFeature({\n    name: \"AiPowerUps/WbTranslatePage\",\n    register(container) {\n        container.register(LexicalParser).inSingletonScope();\n        container.registerDecorator(WbTranslatePageDecorator);\n    }\n});\n"],"names":["WbTranslatePageFeature","createFeature","container","LexicalParser","WbTranslatePageDecorator"],"mappings":";;;AAIO,MAAMA,yBAAyBC,cAAc;IAChD,MAAM;IACN,UAASC,SAAS;QACdA,UAAU,QAAQ,CAACC,eAAe,gBAAgB;QAClDD,UAAU,iBAAiB,CAACE;IAChC;AACJ"}

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

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

package/api/features/WbTranslatePage/index.js

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

package/api/graphql/BaseGraphQLSchema.d.ts

@@ -1,8 +1,8 @@
-import { CoreGraphQLSchemaFactory } from "@webiny/handler-graphql/graphql/abstractions.core.js";
+import { CoreGraphQLSchemaFactory } from "@webiny/api-graphql/graphql/abstractions.core.js";
 declare class BaseGraphQLSchemaImpl implements CoreGraphQLSchemaFactory.Interface {
     execute(builder: CoreGraphQLSchemaFactory.SchemaBuilder): Promise<CoreGraphQLSchemaFactory.SchemaBuilder>;
 }
 export declare const BaseGraphQLSchema: typeof BaseGraphQLSchemaImpl & {
-    __abstraction: import("@webiny/di").Abstraction<import("@webiny/handler-graphql/graphql/abstractions.public.js").IGraphQLSchemaFactory>;
+    __abstraction: import("@webiny/di").Abstraction<import("@webiny/api-graphql/graphql/abstractions.core.js").ICoreGraphQLSchemaFactory>;
 };
 export {};

package/api/graphql/BaseGraphQLSchema.js

@@ -1,11 +1,13 @@
-import { CoreGraphQLSchemaFactory } from "@webiny/handler-graphql/graphql/abstractions.core.js";
-import { ErrorResponse, Response } from "@webiny/handler-graphql/responses.js";
-import { AiModelRegistry } from "@webiny/api-core/features/ai/index.js";
+import { CoreGraphQLSchemaFactory } from "@webiny/api-graphql/graphql/abstractions.core.js";
+import { ErrorResponse, Response } from "@webiny/api-graphql/responses.js";
+import { Ai } from "@webiny/api-core/features/ai/index.js";
 import { TaskService } from "@webiny/api-core/features/task/TaskService/index.js";
+import { FeatureFlags } from "@webiny/api-core/features/featureFlags/abstractions.js";
 import { GetSettingsUseCase } from "../features/GetSettings/index.js";
 import { UpdateSettingsUseCase } from "../features/UpdateSettings/index.js";
 import { AiPowerUpsSettingsGraphQLMapper } from "./abstractions.js";
 import { WB_GENERATE_PAGE_CONTENT_TASK_ID } from "../features/WbGeneratePageContent/WbGeneratePageContentTask.js";
+import { CMS_GENERATE_ENTRY_CONTENT_TASK_ID } from "../features/CmsGenerateEntryContent/CmsGenerateEntryContentTask.js";
 class BaseGraphQLSchemaImpl {
     async execute(builder) {
         builder.addTypeDefs(`
@@ -43,6 +45,16 @@ class BaseGraphQLSchemaImpl {
                     excludedFileIds: [String!]
                     readerPersonaId: String
                     writerPersonaId: String
+                    additionalFileIds: [String!]
+                ): JSON!
+                generateEntryContent(
+                    prompt: String!
+                    modelId: String!
+                    projectId: String
+                    excludedFileIds: [String!]
+                    readerPersonaId: String
+                    writerPersonaId: String
+                    additionalFileIds: [String!]
                 ): JSON!
             }
 
@@ -65,9 +77,9 @@ class BaseGraphQLSchemaImpl {
         builder.addResolver({
             path: "AiPowerUpsQuery.listModels",
             dependencies: [
-                AiModelRegistry
+                Ai
             ],
-            resolver: (registry)=>async ()=>registry.listModels()
+            resolver: (ai)=>async ()=>ai.listModels()
         });
         builder.addResolver({
             path: "AiPowerUpsQuery.getSettings",
@@ -96,7 +108,34 @@ class BaseGraphQLSchemaImpl {
                             projectId: args.projectId ?? null,
                             excludedFileIds: args.excludedFileIds ?? null,
                             readerPersonaId: args.readerPersonaId ?? null,
-                            writerPersonaId: args.writerPersonaId ?? null
+                            writerPersonaId: args.writerPersonaId ?? null,
+                            additionalFileIds: args.additionalFileIds ?? null
+                        }
+                    });
+                    if (result.isFail()) throw result.error;
+                    return {
+                        id: result.value.id
+                    };
+                }
+        });
+        builder.addResolver({
+            path: "AiPowerUpsMutation.generateEntryContent",
+            dependencies: [
+                TaskService,
+                FeatureFlags
+            ],
+            resolver: (taskService, featureFlags)=>async ({ args })=>{
+                    if (!featureFlags.get().isEnabled("aiPowerups.cms.entryGeneration")) throw new Error("AI entry generation cannot be used because your project license does not permit it.");
+                    const result = await taskService.trigger({
+                        definition: CMS_GENERATE_ENTRY_CONTENT_TASK_ID,
+                        input: {
+                            prompt: args.prompt,
+                            modelId: args.modelId,
+                            projectId: args.projectId ?? null,
+                            excludedFileIds: args.excludedFileIds ?? null,
+                            readerPersonaId: args.readerPersonaId ?? null,
+                            writerPersonaId: args.writerPersonaId ?? null,
+                            additionalFileIds: args.additionalFileIds ?? null
                         }
                     });
                     if (result.isFail()) throw result.error;