@webiny/ai-powerups 6.6.0-alpha.4 → 6.6.0-alpha.6

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 {\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"}
+{"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 { ListTagsUseCase } from \"@webiny/api-file-manager/features/file/ListTags/index.js\";\nimport {\n    ResolveAiCapabilityUseCase,\n    withAdditionalInstructions\n} from \"~/api/features/Capabilities/index.js\";\nimport { WB_GENERATE_PAGE_CAPABILITY } from \"./capability.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 resolveCapability: ResolveAiCapabilityUseCase.Interface,\n        private ai: Ai.Interface,\n        private aiSdkTools: AiSdkTools.Interface,\n        private listTags: ListTagsUseCase.Interface\n    ) {}\n\n    async execute(\n        params: WbGeneratePageContentParams\n    ): Promise<Result<GeneratePageContentResult, Error>> {\n        const resolved = await this.resolveCapability.execute(WB_GENERATE_PAGE_CAPABILITY);\n        if (resolved.isFail()) {\n            return Result.fail(resolved.error);\n        }\n\n        const capability = resolved.value;\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 = withAdditionalInstructions(\n            capability,\n            buildDomainPrompt(components, params.tools, imageTags) + context.toString()\n        );\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: capability.model,\n                connection: capability.connection,\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            ResolveAiCapabilityUseCase,\n            Ai,\n            AiSdkTools,\n            ListTagsUseCase\n        ]\n    });\n"],"names":["WbGeneratePageContentUseCaseImpl","promptContextBuilder","resolveCapability","ai","aiSdkTools","listTags","params","resolved","WB_GENERATE_PAGE_CAPABILITY","Result","capability","sdkTools","context","projectFileTool","createReadProjectFileTool","Object","tagsResult","imageTags","t","components","systemText","withAdditionalInstructions","buildDomainPrompt","system","aiResult","formatAdditionalFilesContext","stepCountIs","text","step","elements","LlmJsonResponse","componentFilter","ComponentFilter","output","JSON","filesRead","Set","toolCallsMade","call","input","telemetry","error","Error","String","WbGeneratePageContentUseCaseImplementation","WbGeneratePageContentUseCase","AiPromptContextBuilder","ResolveAiCapabilityUseCase","Ai","AiSdkTools","ListTagsUseCase"],"mappings":";;;;;;;;;;;;AAyBA,MAAMA;IACF,YACYC,oBAAsD,EACtDC,iBAAuD,EACvDC,EAAgB,EAChBC,UAAgC,EAChCC,QAAmC,CAC7C;aALUJ,oBAAoB,GAApBA;aACAC,iBAAiB,GAAjBA;aACAC,EAAE,GAAFA;aACAC,UAAU,GAAVA;aACAC,QAAQ,GAARA;IACT;IAEH,MAAM,QACFC,MAAmC,EACc;QACjD,MAAMC,WAAW,MAAM,IAAI,CAAC,iBAAiB,CAAC,OAAO,CAACC;QACtD,IAAID,SAAS,MAAM,IACf,OAAOE,OAAO,IAAI,CAACF,SAAS,KAAK;QAGrC,MAAMG,aAAaH,SAAS,KAAK;QAEjC,MAAMI,WAAW,IAAI,CAAC,UAAU,CAAC,UAAU;QAE3C,MAAMC,UAAU,MAAM,IAAI,CAAC,oBAAoB,CAAC,OAAO,CAAC;YACpD,WAAWN,OAAO,SAAS;YAC3B,iBAAiBA,OAAO,eAAe;YACvC,iBAAiBA,OAAO,eAAe;YACvC,iBAAiBA,OAAO,eAAe;YACvC,mBAAmBA,OAAO,iBAAiB;QAC/C;QAEA,IAAIM,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,aAAab,OAAO,UAAU;QACpC,MAAMc,aAAaC,2BACfX,YACAY,kBAAkBH,YAAYb,OAAO,KAAK,EAAEW,aAAaL,QAAQ,QAAQ;QAG7E,MAAMW,SAAS;YACX,MAAM;YACN,SAASH;YACT,iBAAiB;gBACb,WAAW;oBACP,cAAc;wBAAE,MAAM;oBAAY;gBACtC;YACJ;QACJ;QAEA,IAAI;YACA,MAAMI,WAAW,MAAM,IAAI,CAAC,EAAE,CAAC,YAAY,CAAC;gBACxC,OAAOd,WAAW,KAAK;gBACvB,YAAYA,WAAW,UAAU;gBACjCa;gBACA,YAAY;gBACZ,QAAQjB,OAAO,MAAM,GAAGmB,6BAA6Bb,QAAQ,eAAe;gBAC5E,GAAIG,OAAO,IAAI,CAACJ,UAAU,MAAM,GAAG,IAC7B;oBAAE,OAAOA;oBAAU,UAAUe,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,gBAAgBb;YAC5C,MAAMc,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,UAAUvB,QAAQ,QAAQ;gBAC1ByB;gBACA,YAAYb,SAAS,KAAK,CAAC,MAAM;gBACjC,gBAAgBT,OAAO,IAAI,CAACJ;gBAC5B,mBAAmBM;YACvB;YAEA,OAAOR,OAAO,EAAE,CAAC;gBAAEwB;gBAAQO;YAAU;QACzC,EAAE,OAAOC,OAAO;YACZ,OAAOhC,OAAO,IAAI,CACd,IAAIiC,MACA,CAAC,sBAAsB,EAAED,iBAAiBC,QAAQD,MAAM,OAAO,GAAGE,OAAOF,QAAQ;QAG7F;IACJ;AACJ;AAEO,MAAMG,6CACTC,6BAA6B,oBAAoB,CAAC;IAC9C,gBAAgB7C;IAChB,cAAc;QACV8C;QACAC;QACAC;QACAC;QACAC;KACH;AACL"}

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

@@ -0,0 +1,16 @@
+import { AiCapability } from "../../../api/features/Capabilities/index.js";
+export declare const WB_GENERATE_PAGE_CAPABILITY = "wb.generatePage";
+/**
+ * No `guidance`, for the same reason as CMS entry generation: the prompt carries the component
+ * catalog and the tool definitions, both computed per request.
+ */
+declare class WbGeneratePageCapabilityImpl implements AiCapability.Interface {
+    readonly id = "wb.generatePage";
+    readonly label = "Page generation";
+    readonly description = "Writes page content from a prompt using the Website Builder components available in the project.";
+    readonly defaultRole: "standard";
+}
+export declare const WbGeneratePageCapability: typeof WbGeneratePageCapabilityImpl & {
+    __abstraction: import("@webiny/di").Abstraction<import("~/api/features/Capabilities/index.js").IAiCapability>;
+};
+export {};

package/api/features/WbGeneratePageContent/capability.js

@@ -0,0 +1,17 @@
+import { AiCapability } from "../Capabilities/index.js";
+const WB_GENERATE_PAGE_CAPABILITY = "wb.generatePage";
+class WbGeneratePageCapabilityImpl {
+    constructor(){
+        this.id = WB_GENERATE_PAGE_CAPABILITY;
+        this.label = "Page generation";
+        this.description = "Writes page content from a prompt using the Website Builder components available in the project.";
+        this.defaultRole = "standard";
+    }
+}
+const WbGeneratePageCapability = AiCapability.createImplementation({
+    implementation: WbGeneratePageCapabilityImpl,
+    dependencies: []
+});
+export { WB_GENERATE_PAGE_CAPABILITY, WbGeneratePageCapability };
+
+//# sourceMappingURL=capability.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"api/features/WbGeneratePageContent/capability.js","sources":["../../../../src/api/features/WbGeneratePageContent/capability.ts"],"sourcesContent":["import { AiCapability } from \"~/api/features/Capabilities/index.js\";\n\nexport const WB_GENERATE_PAGE_CAPABILITY = \"wb.generatePage\";\n\n/**\n * No `guidance`, for the same reason as CMS entry generation: the prompt carries the component\n * catalog and the tool definitions, both computed per request.\n */\nclass WbGeneratePageCapabilityImpl implements AiCapability.Interface {\n    readonly id = WB_GENERATE_PAGE_CAPABILITY;\n    readonly label = \"Page generation\";\n    readonly description =\n        \"Writes page content from a prompt using the Website Builder components available in the project.\";\n    readonly defaultRole = \"standard\" as const;\n}\n\nexport const WbGeneratePageCapability = AiCapability.createImplementation({\n    implementation: WbGeneratePageCapabilityImpl,\n    dependencies: []\n});\n"],"names":["WB_GENERATE_PAGE_CAPABILITY","WbGeneratePageCapabilityImpl","WbGeneratePageCapability","AiCapability"],"mappings":";AAEO,MAAMA,8BAA8B;AAM3C,MAAMC;;aACO,EAAE,GAAGD;aACL,KAAK,GAAG;aACR,WAAW,GAChB;aACK,WAAW,GAAG;;AAC3B;AAEO,MAAME,2BAA2BC,aAAa,oBAAoB,CAAC;IACtE,gBAAgBF;IAChB,cAAc,EAAE;AACpB"}

package/api/features/WbGeneratePageContent/feature.js

@@ -1,9 +1,11 @@
 import { createFeature } from "@webiny/feature/api";
 import { WbGeneratePageContentUseCaseImplementation } from "./WbGeneratePageContentUseCase.js";
 import { WbGeneratePageContentTask } from "./WbGeneratePageContentTask.js";
+import { WbGeneratePageCapability } from "./capability.js";
 const WbGeneratePageContentFeature = createFeature({
     name: "AiPowerUps/WbGeneratePageContent",
     register (container) {
+        container.register(WbGeneratePageCapability);
         container.register(WbGeneratePageContentUseCaseImplementation);
         container.register(WbGeneratePageContentTask);
     }

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

@@ -1 +1 @@
-{"version":3,"file":"api/features/WbGeneratePageContent/feature.js","sources":["../../../../src/api/features/WbGeneratePageContent/feature.ts"],"sourcesContent":["import { createFeature } from \"@webiny/feature/api\";\nimport { WbGeneratePageContentUseCaseImplementation } from \"./WbGeneratePageContentUseCase.js\";\nimport { WbGeneratePageContentTask } from \"./WbGeneratePageContentTask.js\";\n\nexport const WbGeneratePageContentFeature = createFeature({\n    name: \"AiPowerUps/WbGeneratePageContent\",\n    register(container) {\n        container.register(WbGeneratePageContentUseCaseImplementation);\n        container.register(WbGeneratePageContentTask);\n    }\n});\n"],"names":["WbGeneratePageContentFeature","createFeature","container","WbGeneratePageContentUseCaseImplementation","WbGeneratePageContentTask"],"mappings":";;;AAIO,MAAMA,+BAA+BC,cAAc;IACtD,MAAM;IACN,UAASC,SAAS;QACdA,UAAU,QAAQ,CAACC;QACnBD,UAAU,QAAQ,CAACE;IACvB;AACJ"}
+{"version":3,"file":"api/features/WbGeneratePageContent/feature.js","sources":["../../../../src/api/features/WbGeneratePageContent/feature.ts"],"sourcesContent":["import { createFeature } from \"@webiny/feature/api\";\nimport { WbGeneratePageContentUseCaseImplementation } from \"./WbGeneratePageContentUseCase.js\";\nimport { WbGeneratePageContentTask } from \"./WbGeneratePageContentTask.js\";\nimport { WbGeneratePageCapability } from \"./capability.js\";\n\nexport const WbGeneratePageContentFeature = createFeature({\n    name: \"AiPowerUps/WbGeneratePageContent\",\n    register(container) {\n        container.register(WbGeneratePageCapability);\n        container.register(WbGeneratePageContentUseCaseImplementation);\n        container.register(WbGeneratePageContentTask);\n    }\n});\n"],"names":["WbGeneratePageContentFeature","createFeature","container","WbGeneratePageCapability","WbGeneratePageContentUseCaseImplementation","WbGeneratePageContentTask"],"mappings":";;;;AAKO,MAAMA,+BAA+BC,cAAc;IACtD,MAAM;IACN,UAASC,SAAS;QACdA,UAAU,QAAQ,CAACC;QACnBD,UAAU,QAAQ,CAACE;QACnBF,UAAU,QAAQ,CAACG;IACvB;AACJ"}

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

@@ -1,19 +1,19 @@
 import { Ai } from "@webiny/api-core/features/ai/index.js";
-import { Encryption } from "@webiny/api-core/features/encryption/index.js";
+import { Logger } from "@webiny/api-core/features/logger/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 { ResolveAiCapabilityUseCase } from "../../../api/features/Capabilities/index.js";
 import { LexicalParser } from "./abstractions/LexicalParser.js";
 declare class WbTranslatePageDecoratorImpl implements TranslatePageUseCase.Interface {
     private getDefaultLanguage;
-    private getSettings;
+    private resolveCapability;
+    private logger;
     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);
+    constructor(getDefaultLanguage: GetDefaultLanguageUseCase.Interface, resolveCapability: ResolveAiCapabilityUseCase.Interface, logger: Logger.Interface, ai: Ai.Interface, lexicalParser: LexicalParser.Interface, updatePageRepository: UpdatePageRepository.Interface, decoratee: TranslatePageUseCase.Interface);
     execute(params: TranslatePageUseCase.Params): TranslatePageUseCase.Return;
     private resolveSourceLanguage;
     private translate;

package/api/features/WbTranslatePage/WbTranslatePageDecorator.js

@@ -1,18 +1,19 @@
 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 { Logger } from "@webiny/api-core/features/logger/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 { ResolveAiCapabilityUseCase, withAdditionalInstructions } from "../Capabilities/index.js";
+import { WB_TRANSLATE_PAGE_CAPABILITY } from "./capability.js";
 import { LexicalParser } from "./abstractions/LexicalParser.js";
 class WbTranslatePageDecoratorImpl {
-    constructor(getDefaultLanguage, getSettings, ai, encryption, lexicalParser, updatePageRepository, decoratee){
+    constructor(getDefaultLanguage, resolveCapability, logger, ai, lexicalParser, updatePageRepository, decoratee){
         this.getDefaultLanguage = getDefaultLanguage;
-        this.getSettings = getSettings;
+        this.resolveCapability = resolveCapability;
+        this.logger = logger;
         this.ai = ai;
-        this.encryption = encryption;
         this.lexicalParser = lexicalParser;
         this.updatePageRepository = updatePageRepository;
         this.decoratee = decoratee;
@@ -36,12 +37,14 @@ class WbTranslatePageDecoratorImpl {
         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 resolved = await this.resolveCapability.execute(WB_TRANSLATE_PAGE_CAPABILITY);
+        if (resolved.isFail()) {
+            this.logger.warn({
+                reason: resolved.error.message
+            }, "Skipping AI page translation.");
+            return null;
+        }
+        const capability = resolved.value;
         const properties = {
             title: page.properties["title"],
             snippet: page.properties["snippet"],
@@ -53,12 +56,9 @@ class WbTranslatePageDecoratorImpl {
             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.`,
+            model: capability.model,
+            connection: capability.connection,
+            system: withAdditionalInstructions(capability),
             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
         });
@@ -122,9 +122,9 @@ const WbTranslatePageDecorator = TranslatePageUseCase.createDecorator({
     decorator: WbTranslatePageDecoratorImpl,
     dependencies: [
         GetDefaultLanguageUseCase,
-        GetSettingsUseCase,
+        ResolveAiCapabilityUseCase,
+        Logger,
         Ai,
-        Encryption,
         LexicalParser,
         UpdatePageRepository
     ]

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

@@ -1 +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"}
+{"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 { Logger } from \"@webiny/api-core/features/logger/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 {\n    ResolveAiCapabilityUseCase,\n    withAdditionalInstructions\n} from \"~/api/features/Capabilities/index.js\";\nimport { WB_TRANSLATE_PAGE_CAPABILITY } from \"./capability.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 resolveCapability: ResolveAiCapabilityUseCase.Interface,\n        private logger: Logger.Interface,\n        private ai: Ai.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 resolved = await this.resolveCapability.execute(WB_TRANSLATE_PAGE_CAPABILITY);\n        if (resolved.isFail()) {\n            /*\n             * Translation is a decorator on top of a successful page save, so it stays\n             * non-fatal: the page is already created, just untranslated. It does get logged\n             * now, because a silently skipped translation was indistinguishable from a\n             * misconfigured one.\n             */\n            this.logger.warn({ reason: resolved.error.message }, \"Skipping AI page translation.\");\n            return null;\n        }\n\n        const capability = resolved.value;\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: capability.model,\n            connection: capability.connection,\n            system: withAdditionalInstructions(capability),\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        ResolveAiCapabilityUseCase,\n        Logger,\n        Ai,\n        LexicalParser,\n        UpdatePageRepository\n    ]\n});\n"],"names":["WbTranslatePageDecoratorImpl","getDefaultLanguage","resolveCapability","logger","ai","lexicalParser","updatePageRepository","decoratee","params","result","page","sourceLanguage","translations","Result","updateData","updateResult","targetLanguage","resolved","WB_TRANSLATE_PAGE_CAPABILITY","capability","properties","bindings","input","JSON","withAdditionalInstructions","elementId","element","inputs","inputName","text","structuredClone","translatedPathSlug","key","value","Object","slug","languageCode","binding","state","set","WbTranslatePageDecorator","TranslatePageUseCase","GetDefaultLanguageUseCase","ResolveAiCapabilityUseCase","Logger","Ai","LexicalParser","UpdatePageRepository"],"mappings":";;;;;;;;;;AAiCA,MAAMA;IACF,YACYC,kBAAuD,EACvDC,iBAAuD,EACvDC,MAAwB,EACxBC,EAAgB,EAChBC,aAAsC,EACtCC,oBAAoD,EACpDC,SAAyC,CACnD;aAPUN,kBAAkB,GAAlBA;aACAC,iBAAiB,GAAjBA;aACAC,MAAM,GAANA;aACAC,EAAE,GAAFA;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,WAAW,MAAM,IAAI,CAAC,iBAAiB,CAAC,OAAO,CAACC;QACtD,IAAID,SAAS,MAAM,IAAI;YAOnB,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC;gBAAE,QAAQA,SAAS,KAAK,CAAC,OAAO;YAAC,GAAG;YACrD,OAAO;QACX;QAEA,MAAME,aAAaF,SAAS,KAAK;QAEjC,MAAMG,aAAa;YACf,OAAOV,KAAK,UAAU,CAAC,QAAQ;YAC/B,SAASA,KAAK,UAAU,CAAC,UAAU;YACnC,UAAUA,KAAK,UAAU,CAAC,QAAQ;QACtC;QACA,MAAMW,WAAW,IAAI,CAAC,eAAe,CAACX,KAAK,QAAQ;QACnD,MAAMY,QAAQC,KAAK,SAAS,CAAC;YAAEH;YAAYC;QAAS;QAEpD,MAAMZ,SAAS,MAAM,IAAI,CAAC,EAAE,CAAC,YAAY,CAAC;YACtC,OAAOU,WAAW,KAAK;YACvB,YAAYA,WAAW,UAAU;YACjC,QAAQK,2BAA2BL;YACnC,QAAQ,CAAC,sCAAsC,EAAER,eAAe,oBAAoB,EAAEK,eAAe,sHAAsH,EAAEM,OAAO;YACpO,aAAa;QACjB;QAEA,IAAI;YACA,OAAOC,KAAK,KAAK,CAACd,OAAO,IAAI;QACjC,EAAE,OAAM;YACJ,OAAO;QACX;IACJ;IAEQ,gBAAgBY,QAA6B,EAAwB;QACzE,MAAMZ,SAA+B,CAAC;QAEtC,IAAK,MAAMgB,aAAaJ,SAAU;YAC9B,MAAMK,UAAUL,QAAQ,CAACI,UAAU;YACnC,MAAME,SAASD,QAAQ,MAAM;YAC7B,IAAKC,QAIL,IAAK,MAAMC,aAAaD,OAAQ;gBAC5B,MAAML,QAAQK,MAAM,CAACC,UAAU;gBAC/B,IAAIN,AAAe,WAAfA,MAAM,IAAI,IAAeA,AAAe,eAAfA,MAAM,IAAI,EACnCb,MAAM,CAAC,GAAGgB,UAAU,QAAQ,EAAEG,WAAW,CAAC,GAAG;oBACzC,MAAMN,MAAM,IAAI;oBAChB,OAAOA,MAAM,MAAM;gBACvB;qBACG,IAAIA,AAAe,cAAfA,MAAM,IAAI,IAAkBA,MAAM,MAAM,EAAE,MACjDb,MAAM,CAAC,GAAGgB,UAAU,QAAQ,EAAEG,WAAW,CAAC,GAAG;oBACzC,MAAMN,MAAM,IAAI;oBAChB,OAAOA,MAAM,MAAM,CAAC,IAAI;gBAC5B;YAER;QACJ;QAEA,OAAOb;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,MAAMQ,aAAkC;YAAE,GAAGV,KAAK,UAAU;QAAC;QAC7D,MAAMW,WAAgCS,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,EAC7DQ,UAAU,CAACY,IAAI,GAAGC;QAGtB,IAAIF,oBAAoB;YACpB,MAAMI,OAAO,IAAI,CAAC,OAAO,CAACJ;YAC1B,IAAII,MAAM;gBACN,MAAMC,eAAe1B,KAAK,UAAU,CAAC,WAAW;gBAChDU,UAAU,CAAC,OAAO,GAAG,CAAC,CAAC,EAAEgB,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,IAAIlB,UAAU,GAAGW,IAAI,aAAa,CAAC,EAAET,KAAK,SAAS,CAACe;gBACpDC,IAAIlB,UAAU,GAAGW,IAAI,YAAY,CAAC,EAAEK,QAAQ,KAAK;YACrD;QACJ,OACIE,IAAIlB,UAAU,GAAGW,IAAI,OAAO,CAAC,EAAEK,QAAQ,KAAK;QAIpD,OAAO;YAAEjB;YAAYC;QAAS;IAClC;AACJ;AAEO,MAAMmB,2BAA2BC,qBAAqB,eAAe,CAAC;IACzE,WAAWzC;IACX,cAAc;QACV0C;QACAC;QACAC;QACAC;QACAC;QACAC;KACH;AACL"}

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

@@ -0,0 +1,13 @@
+import { AiCapability } from "../../../api/features/Capabilities/index.js";
+export declare const WB_TRANSLATE_PAGE_CAPABILITY = "wb.translatePage";
+declare class WbTranslatePageCapabilityImpl implements AiCapability.Interface {
+    readonly id = "wb.translatePage";
+    readonly label = "Page translation";
+    readonly description = "Translates the text on a page into another language, preserving placeholders.";
+    readonly defaultRole: "standard";
+    readonly guidance = "You are a professional translator. Translate all user-provided text to the requested target language, preserving placeholders and formatting. Return a JSON object with the same keys, only changing the values.";
+}
+export declare const WbTranslatePageCapability: typeof WbTranslatePageCapabilityImpl & {
+    __abstraction: import("@webiny/di").Abstraction<import("~/api/features/Capabilities/index.js").IAiCapability>;
+};
+export {};

package/api/features/WbTranslatePage/capability.js

@@ -0,0 +1,19 @@
+import { AiCapability } from "../Capabilities/index.js";
+const WB_TRANSLATE_PAGE_CAPABILITY = "wb.translatePage";
+const guidance = "You are a professional translator. Translate all user-provided text to the requested target language, preserving placeholders and formatting. Return a JSON object with the same keys, only changing the values.";
+class WbTranslatePageCapabilityImpl {
+    constructor(){
+        this.id = WB_TRANSLATE_PAGE_CAPABILITY;
+        this.label = "Page translation";
+        this.description = "Translates the text on a page into another language, preserving placeholders.";
+        this.defaultRole = "standard";
+        this.guidance = guidance;
+    }
+}
+const WbTranslatePageCapability = AiCapability.createImplementation({
+    implementation: WbTranslatePageCapabilityImpl,
+    dependencies: []
+});
+export { WB_TRANSLATE_PAGE_CAPABILITY, WbTranslatePageCapability };
+
+//# sourceMappingURL=capability.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"api/features/WbTranslatePage/capability.js","sources":["../../../../src/api/features/WbTranslatePage/capability.ts"],"sourcesContent":["import { AiCapability } from \"~/api/features/Capabilities/index.js\";\n\nexport const WB_TRANSLATE_PAGE_CAPABILITY = \"wb.translatePage\";\n\n/**\n * The target language used to be interpolated into this sentence. It moved to the user prompt so\n * the instructions could become fixed text a project can replace, which is worth doing here: house\n * translation rules (keep product names in English, use formal address, never translate these\n * twelve terms) are the most commonly requested prompt change in the whole feature set.\n */\nconst guidance =\n    \"You are a professional translator. Translate all user-provided text to the requested target language, preserving placeholders and formatting. Return a JSON object with the same keys, only changing the values.\";\n\nclass WbTranslatePageCapabilityImpl implements AiCapability.Interface {\n    readonly id = WB_TRANSLATE_PAGE_CAPABILITY;\n    readonly label = \"Page translation\";\n    readonly description =\n        \"Translates the text on a page into another language, preserving placeholders.\";\n    readonly defaultRole = \"standard\" as const;\n    readonly guidance = guidance;\n}\n\nexport const WbTranslatePageCapability = AiCapability.createImplementation({\n    implementation: WbTranslatePageCapabilityImpl,\n    dependencies: []\n});\n"],"names":["WB_TRANSLATE_PAGE_CAPABILITY","guidance","WbTranslatePageCapabilityImpl","WbTranslatePageCapability","AiCapability"],"mappings":";AAEO,MAAMA,+BAA+B;AAQ5C,MAAMC,WACF;AAEJ,MAAMC;;aACO,EAAE,GAAGF;aACL,KAAK,GAAG;aACR,WAAW,GAChB;aACK,WAAW,GAAG;aACd,QAAQ,GAAGC;;AACxB;AAEO,MAAME,4BAA4BC,aAAa,oBAAoB,CAAC;IACvE,gBAAgBF;IAChB,cAAc,EAAE;AACpB"}

package/api/features/WbTranslatePage/feature.js

@@ -1,9 +1,11 @@
 import { createFeature } from "@webiny/feature/api";
 import { WbTranslatePageDecorator } from "./WbTranslatePageDecorator.js";
 import { LexicalParser } from "./LexicalParser.js";
+import { WbTranslatePageCapability } from "./capability.js";
 const WbTranslatePageFeature = createFeature({
     name: "AiPowerUps/WbTranslatePage",
     register (container) {
+        container.register(WbTranslatePageCapability);
         container.register(LexicalParser).inSingletonScope();
         container.registerDecorator(WbTranslatePageDecorator);
     }

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

@@ -1 +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"}
+{"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\";\nimport { WbTranslatePageCapability } from \"./capability.js\";\n\nexport const WbTranslatePageFeature = createFeature({\n    name: \"AiPowerUps/WbTranslatePage\",\n    register(container) {\n        container.register(WbTranslatePageCapability);\n        container.register(LexicalParser).inSingletonScope();\n        container.registerDecorator(WbTranslatePageDecorator);\n    }\n});\n"],"names":["WbTranslatePageFeature","createFeature","container","WbTranslatePageCapability","LexicalParser","WbTranslatePageDecorator"],"mappings":";;;;AAKO,MAAMA,yBAAyBC,cAAc;IAChD,MAAM;IACN,UAASC,SAAS;QACdA,UAAU,QAAQ,CAACC;QACnBD,UAAU,QAAQ,CAACE,eAAe,gBAAgB;QAClDF,UAAU,iBAAiB,CAACG;IAChC;AACJ"}

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

@@ -2,7 +2,12 @@ import type { ZodType } from "zod";
 export interface IAiPowerUpsSettingsGroupHandler {
     readonly name: string;
     readonly inputSchema: ZodType<unknown>;
-    mapFromStorage(persisted: unknown): unknown;
+    /**
+     * `all` is the entire persisted settings blob. A group only needs it to read a section it does
+     * not own, which in practice means migrating away from one: `connections` and `modelRoles` both
+     * derive their defaults from the legacy `providers` section the first time they load.
+     */
+    mapFromStorage(persisted: unknown, all?: Record<string, unknown>): unknown;
     mapToStorage(internal: unknown, existing: unknown | null): Promise<unknown>;
 }
 export declare const AiPowerUpsSettingsGroupHandler: import("@webiny/di").Abstraction<IAiPowerUpsSettingsGroupHandler>;

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

@@ -1 +1 @@
-{"version":3,"file":"api/features/shared/abstractions.js","sources":["../../../../src/api/features/shared/abstractions.ts"],"sourcesContent":["import { createAbstraction } from \"@webiny/feature/api\";\nimport type { ZodType } from \"zod\";\n\nexport interface IAiPowerUpsSettingsGroupHandler {\n    readonly name: string;\n    readonly inputSchema: ZodType<unknown>;\n    mapFromStorage(persisted: unknown): unknown;\n    mapToStorage(internal: unknown, existing: unknown | null): Promise<unknown>;\n}\n\nexport const AiPowerUpsSettingsGroupHandler = createAbstraction<IAiPowerUpsSettingsGroupHandler>(\n    \"AiPowerUpsSettingsGroupHandler\"\n);\n\nexport namespace AiPowerUpsSettingsGroupHandler {\n    export type Interface = IAiPowerUpsSettingsGroupHandler;\n}\n\nexport interface IAiPowerUpsSettingsGroupGraphQLMapper {\n    readonly name: string;\n    toApi(internal: unknown): unknown | Promise<unknown>;\n    fromApi(api: unknown, existing: unknown | null): unknown | Promise<unknown>;\n}\n\nexport const AiPowerUpsSettingsGroupGraphQLMapper =\n    createAbstraction<IAiPowerUpsSettingsGroupGraphQLMapper>(\n        \"AiPowerUpsSettingsGroupGraphQLMapper\"\n    );\n\nexport namespace AiPowerUpsSettingsGroupGraphQLMapper {\n    export type Interface = IAiPowerUpsSettingsGroupGraphQLMapper;\n}\n\nimport type { IAiPowerUpsSettings } from \"~/api/types.js\";\n\nexport interface SettingsCacheEntry {\n    raw: Record<string, unknown>;\n    mapped: IAiPowerUpsSettings;\n}\n\nexport interface IAiPowerUpsSettingsCache {\n    get(): SettingsCacheEntry | null;\n    set(raw: Record<string, unknown>, mapped: IAiPowerUpsSettings): void;\n}\n\nexport const AiPowerUpsSettingsCache =\n    createAbstraction<IAiPowerUpsSettingsCache>(\"AiPowerUpsSettingsCache\");\n\nexport namespace AiPowerUpsSettingsCache {\n    export type Interface = IAiPowerUpsSettingsCache;\n}\n"],"names":["AiPowerUpsSettingsGroupHandler","createAbstraction","AiPowerUpsSettingsGroupGraphQLMapper","AiPowerUpsSettingsCache"],"mappings":";AAUO,MAAMA,iCAAiCC,kBAC1C;AAaG,MAAMC,uCACTD,kBACI;AAmBD,MAAME,0BACTF,kBAA4C"}
+{"version":3,"file":"api/features/shared/abstractions.js","sources":["../../../../src/api/features/shared/abstractions.ts"],"sourcesContent":["import { createAbstraction } from \"@webiny/feature/api\";\nimport type { ZodType } from \"zod\";\n\nexport interface IAiPowerUpsSettingsGroupHandler {\n    readonly name: string;\n    readonly inputSchema: ZodType<unknown>;\n    /**\n     * `all` is the entire persisted settings blob. A group only needs it to read a section it does\n     * not own, which in practice means migrating away from one: `connections` and `modelRoles` both\n     * derive their defaults from the legacy `providers` section the first time they load.\n     */\n    mapFromStorage(persisted: unknown, all?: Record<string, unknown>): unknown;\n    mapToStorage(internal: unknown, existing: unknown | null): Promise<unknown>;\n}\n\nexport const AiPowerUpsSettingsGroupHandler = createAbstraction<IAiPowerUpsSettingsGroupHandler>(\n    \"AiPowerUpsSettingsGroupHandler\"\n);\n\nexport namespace AiPowerUpsSettingsGroupHandler {\n    export type Interface = IAiPowerUpsSettingsGroupHandler;\n}\n\nexport interface IAiPowerUpsSettingsGroupGraphQLMapper {\n    readonly name: string;\n    toApi(internal: unknown): unknown | Promise<unknown>;\n    fromApi(api: unknown, existing: unknown | null): unknown | Promise<unknown>;\n}\n\nexport const AiPowerUpsSettingsGroupGraphQLMapper =\n    createAbstraction<IAiPowerUpsSettingsGroupGraphQLMapper>(\n        \"AiPowerUpsSettingsGroupGraphQLMapper\"\n    );\n\nexport namespace AiPowerUpsSettingsGroupGraphQLMapper {\n    export type Interface = IAiPowerUpsSettingsGroupGraphQLMapper;\n}\n\nimport type { IAiPowerUpsSettings } from \"~/api/types.js\";\n\nexport interface SettingsCacheEntry {\n    raw: Record<string, unknown>;\n    mapped: IAiPowerUpsSettings;\n}\n\nexport interface IAiPowerUpsSettingsCache {\n    get(): SettingsCacheEntry | null;\n    set(raw: Record<string, unknown>, mapped: IAiPowerUpsSettings): void;\n}\n\nexport const AiPowerUpsSettingsCache =\n    createAbstraction<IAiPowerUpsSettingsCache>(\"AiPowerUpsSettingsCache\");\n\nexport namespace AiPowerUpsSettingsCache {\n    export type Interface = IAiPowerUpsSettingsCache;\n}\n"],"names":["AiPowerUpsSettingsGroupHandler","createAbstraction","AiPowerUpsSettingsGroupGraphQLMapper","AiPowerUpsSettingsCache"],"mappings":";AAeO,MAAMA,iCAAiCC,kBAC1C;AAaG,MAAMC,uCACTD,kBACI;AAmBD,MAAME,0BACTF,kBAA4C"}

package/api/features/shared/modelId.d.ts

@@ -0,0 +1,11 @@
+/**
+ * Models are addressed as `"<sdkName>/<modelId>"`, e.g. `"anthropic/claude-sonnet-4-5"`.
+ *
+ * The vendor half has to be recoverable from the string on its own, because a stored role holds the
+ * composite and nothing else: resolving checks that the model's vendor matches the connection's
+ * before spending a request to find out.
+ *
+ * Shared rather than owned by `Connections`, because it is a fact about the model id format and
+ * both `Connections` and `Capabilities` read it.
+ */
+export declare const sdkNameFromModel: (model: string | undefined) => string;

package/api/features/shared/modelId.js

@@ -0,0 +1,4 @@
+const sdkNameFromModel = (model)=>(model ?? "").split("/")[0] ?? "";
+export { sdkNameFromModel };
+
+//# sourceMappingURL=modelId.js.map

package/api/features/shared/modelId.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"api/features/shared/modelId.js","sources":["../../../../src/api/features/shared/modelId.ts"],"sourcesContent":["/**\n * Models are addressed as `\"<sdkName>/<modelId>\"`, e.g. `\"anthropic/claude-sonnet-4-5\"`.\n *\n * The vendor half has to be recoverable from the string on its own, because a stored role holds the\n * composite and nothing else: resolving checks that the model's vendor matches the connection's\n * before spending a request to find out.\n *\n * Shared rather than owned by `Connections`, because it is a fact about the model id format and\n * both `Connections` and `Capabilities` read it.\n */\nexport const sdkNameFromModel = (model: string | undefined): string =>\n    (model ?? \"\").split(\"/\")[0] ?? \"\";\n"],"names":["sdkNameFromModel","model"],"mappings":"AAUO,MAAMA,mBAAmB,CAACC,QAC5BA,AAAAA,CAAAA,SAAS,EAAC,EAAG,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI"}

package/api/graphql/BaseGraphQLSchema.js

@@ -5,6 +5,7 @@ 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 { ListAiCapabilitiesUseCase } from "../features/Capabilities/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";
@@ -18,8 +19,16 @@ class BaseGraphQLSchemaImpl {
                 modelName: String!
             }
 
+            type AiCapability {
+                id: String!
+                label: String!
+                description: String!
+                defaultRole: String!
+            }
+
             type AiPowerUpsQuery {
                 listModels: [AiModel!]!
+                listCapabilities: [AiCapability!]!
                 getSettings: JSON
             }
 
@@ -81,6 +90,13 @@ class BaseGraphQLSchemaImpl {
             ],
             resolver: (ai)=>async ()=>ai.listModels()
         });
+        builder.addResolver({
+            path: "AiPowerUpsQuery.listCapabilities",
+            dependencies: [
+                ListAiCapabilitiesUseCase
+            ],
+            resolver: (useCase)=>async ()=>useCase.execute()
+        });
         builder.addResolver({
             path: "AiPowerUpsQuery.getSettings",
             dependencies: [

package/api/graphql/BaseGraphQLSchema.js.map

@@ -1 +1 @@
-{"version":3,"file":"api/graphql/BaseGraphQLSchema.js","sources":["../../../src/api/graphql/BaseGraphQLSchema.ts"],"sourcesContent":["import { CoreGraphQLSchemaFactory } from \"@webiny/api-graphql/graphql/abstractions.core.js\";\nimport { Response, ErrorResponse } from \"@webiny/api-graphql/responses.js\";\nimport { Ai } from \"@webiny/api-core/features/ai/index.js\";\nimport { TaskService } from \"@webiny/api-core/features/task/TaskService/index.js\";\nimport { FeatureFlags } from \"@webiny/api-core/features/featureFlags/abstractions.js\";\nimport { GetSettingsUseCase } from \"~/api/features/GetSettings/index.js\";\nimport { UpdateSettingsUseCase } from \"~/api/features/UpdateSettings/index.js\";\nimport { AiPowerUpsSettingsGraphQLMapper } from \"./abstractions.js\";\nimport {\n    WB_GENERATE_PAGE_CONTENT_TASK_ID,\n    type IWbGeneratePageContentTaskInput\n} from \"~/api/features/WbGeneratePageContent/WbGeneratePageContentTask.js\";\nimport {\n    CMS_GENERATE_ENTRY_CONTENT_TASK_ID,\n    type ICmsGenerateEntryContentTaskInput\n} from \"~/api/features/CmsGenerateEntryContent/CmsGenerateEntryContentTask.js\";\n\nclass BaseGraphQLSchemaImpl implements CoreGraphQLSchemaFactory.Interface {\n    async execute(\n        builder: CoreGraphQLSchemaFactory.SchemaBuilder\n    ): Promise<CoreGraphQLSchemaFactory.SchemaBuilder> {\n        builder.addTypeDefs(/* GraphQL */ `\n            type AiModel {\n                providerId: String!\n                providerName: String!\n                modelId: String!\n                modelName: String!\n            }\n\n            type AiPowerUpsQuery {\n                listModels: [AiModel!]!\n                getSettings: JSON\n            }\n\n            type AiPowerUpsError {\n                code: String\n                message: String\n                data: JSON\n                stack: String\n            }\n\n            type AiPowerUpsSettingsResponse {\n                data: JSON\n                error: AiPowerUpsError\n            }\n\n            type AiPowerUpsMutation {\n                updateSettings(input: JSON!): AiPowerUpsSettingsResponse!\n                generatePageContent(\n                    prompt: String!\n                    components: JSON!\n                    tools: JSON!\n                    projectId: String\n                    excludedFileIds: [String!]\n                    readerPersonaId: String\n                    writerPersonaId: String\n                    additionalFileIds: [String!]\n                ): JSON!\n                generateEntryContent(\n                    prompt: String!\n                    modelId: String!\n                    projectId: String\n                    excludedFileIds: [String!]\n                    readerPersonaId: String\n                    writerPersonaId: String\n                    additionalFileIds: [String!]\n                ): JSON!\n            }\n\n            extend type Query {\n                aiPowerUps: AiPowerUpsQuery\n            }\n\n            extend type Mutation {\n                aiPowerUps: AiPowerUpsMutation\n            }\n        `);\n\n        builder.addResolver({\n            path: \"Query.aiPowerUps\",\n            resolver: () => () => ({})\n        });\n\n        builder.addResolver({\n            path: \"Mutation.aiPowerUps\",\n            resolver: () => () => ({})\n        });\n\n        builder.addResolver({\n            path: \"AiPowerUpsQuery.listModels\",\n            dependencies: [Ai],\n            resolver: (ai: Ai.Interface) => {\n                return async () => ai.listModels();\n            }\n        });\n\n        builder.addResolver({\n            path: \"AiPowerUpsQuery.getSettings\",\n            dependencies: [GetSettingsUseCase, AiPowerUpsSettingsGraphQLMapper],\n            resolver: (\n                useCase: GetSettingsUseCase.Interface,\n                mapper: AiPowerUpsSettingsGraphQLMapper.Interface\n            ) => {\n                return async () => {\n                    const result = await useCase.execute();\n                    if (result.isFail()) {\n                        throw result.error;\n                    }\n\n                    return mapper.toApi(result.value);\n                };\n            }\n        });\n\n        builder.addResolver<IWbGeneratePageContentTaskInput>({\n            path: \"AiPowerUpsMutation.generatePageContent\",\n            dependencies: [TaskService],\n            resolver: (taskService: TaskService.Interface) => {\n                return async ({ args }) => {\n                    const result = await taskService.trigger<IWbGeneratePageContentTaskInput>({\n                        definition: WB_GENERATE_PAGE_CONTENT_TASK_ID,\n                        input: {\n                            prompt: args.prompt,\n                            components: args.components,\n                            tools: args.tools,\n                            projectId: args.projectId ?? null,\n                            excludedFileIds: args.excludedFileIds ?? null,\n                            readerPersonaId: args.readerPersonaId ?? null,\n                            writerPersonaId: args.writerPersonaId ?? null,\n                            additionalFileIds: args.additionalFileIds ?? null\n                        }\n                    });\n\n                    if (result.isFail()) {\n                        throw result.error;\n                    }\n\n                    return { id: result.value.id };\n                };\n            }\n        });\n\n        builder.addResolver<ICmsGenerateEntryContentTaskInput>({\n            path: \"AiPowerUpsMutation.generateEntryContent\",\n            dependencies: [TaskService, FeatureFlags],\n            resolver: (\n                taskService: TaskService.Interface,\n                featureFlags: FeatureFlags.Interface\n            ) => {\n                return async ({ args }) => {\n                    if (!featureFlags.get().isEnabled(\"aiPowerups.cms.entryGeneration\")) {\n                        throw new Error(\n                            \"AI entry generation cannot be used because your project license does not permit it.\"\n                        );\n                    }\n\n                    const result = await taskService.trigger<ICmsGenerateEntryContentTaskInput>({\n                        definition: CMS_GENERATE_ENTRY_CONTENT_TASK_ID,\n                        input: {\n                            prompt: args.prompt,\n                            modelId: args.modelId,\n                            projectId: args.projectId ?? null,\n                            excludedFileIds: args.excludedFileIds ?? null,\n                            readerPersonaId: args.readerPersonaId ?? null,\n                            writerPersonaId: args.writerPersonaId ?? null,\n                            additionalFileIds: args.additionalFileIds ?? null\n                        }\n                    });\n\n                    if (result.isFail()) {\n                        throw result.error;\n                    }\n\n                    return { id: result.value.id };\n                };\n            }\n        });\n\n        builder.addResolver<{ input: Record<string, unknown> }>({\n            path: \"AiPowerUpsMutation.updateSettings\",\n            dependencies: [\n                GetSettingsUseCase,\n                UpdateSettingsUseCase,\n                AiPowerUpsSettingsGraphQLMapper\n            ],\n            resolver: (\n                getSettings: GetSettingsUseCase.Interface,\n                updateSettings: UpdateSettingsUseCase.Interface,\n                mapper: AiPowerUpsSettingsGraphQLMapper.Interface\n            ) => {\n                return async ({ args }) => {\n                    const currentResult = await getSettings.execute();\n                    if (currentResult.isFail()) {\n                        return new ErrorResponse(currentResult.error);\n                    }\n\n                    const assembled = await mapper.fromApi(args.input, currentResult.value);\n                    const result = await updateSettings.execute(assembled);\n\n                    if (result.isFail()) {\n                        return new ErrorResponse(result.error);\n                    }\n\n                    return new Response(await mapper.toApi(result.value));\n                };\n            }\n        });\n\n        return builder;\n    }\n}\n\nexport const BaseGraphQLSchema = CoreGraphQLSchemaFactory.createImplementation({\n    implementation: BaseGraphQLSchemaImpl,\n    dependencies: []\n});\n"],"names":["BaseGraphQLSchemaImpl","builder","Ai","ai","GetSettingsUseCase","AiPowerUpsSettingsGraphQLMapper","useCase","mapper","result","TaskService","taskService","args","WB_GENERATE_PAGE_CONTENT_TASK_ID","FeatureFlags","featureFlags","Error","CMS_GENERATE_ENTRY_CONTENT_TASK_ID","UpdateSettingsUseCase","getSettings","updateSettings","currentResult","ErrorResponse","assembled","Response","BaseGraphQLSchema","CoreGraphQLSchemaFactory"],"mappings":";;;;;;;;;;AAiBA,MAAMA;IACF,MAAM,QACFC,OAA+C,EACA;QAC/CA,QAAQ,WAAW,CAAe,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;QAuDnC,CAAC;QAEDA,QAAQ,WAAW,CAAC;YAChB,MAAM;YACN,UAAU,IAAM,IAAO,EAAC;QAC5B;QAEAA,QAAQ,WAAW,CAAC;YAChB,MAAM;YACN,UAAU,IAAM,IAAO,EAAC;QAC5B;QAEAA,QAAQ,WAAW,CAAC;YAChB,MAAM;YACN,cAAc;gBAACC;aAAG;YAClB,UAAU,CAACC,KACA,UAAYA,GAAG,UAAU;QAExC;QAEAF,QAAQ,WAAW,CAAC;YAChB,MAAM;YACN,cAAc;gBAACG;gBAAoBC;aAAgC;YACnE,UAAU,CACNC,SACAC,SAEO;oBACH,MAAMC,SAAS,MAAMF,QAAQ,OAAO;oBACpC,IAAIE,OAAO,MAAM,IACb,MAAMA,OAAO,KAAK;oBAGtB,OAAOD,OAAO,KAAK,CAACC,OAAO,KAAK;gBACpC;QAER;QAEAP,QAAQ,WAAW,CAAkC;YACjD,MAAM;YACN,cAAc;gBAACQ;aAAY;YAC3B,UAAU,CAACC,cACA,OAAO,EAAEC,IAAI,EAAE;oBAClB,MAAMH,SAAS,MAAME,YAAY,OAAO,CAAkC;wBACtE,YAAYE;wBACZ,OAAO;4BACH,QAAQD,KAAK,MAAM;4BACnB,YAAYA,KAAK,UAAU;4BAC3B,OAAOA,KAAK,KAAK;4BACjB,WAAWA,KAAK,SAAS,IAAI;4BAC7B,iBAAiBA,KAAK,eAAe,IAAI;4BACzC,iBAAiBA,KAAK,eAAe,IAAI;4BACzC,iBAAiBA,KAAK,eAAe,IAAI;4BACzC,mBAAmBA,KAAK,iBAAiB,IAAI;wBACjD;oBACJ;oBAEA,IAAIH,OAAO,MAAM,IACb,MAAMA,OAAO,KAAK;oBAGtB,OAAO;wBAAE,IAAIA,OAAO,KAAK,CAAC,EAAE;oBAAC;gBACjC;QAER;QAEAP,QAAQ,WAAW,CAAoC;YACnD,MAAM;YACN,cAAc;gBAACQ;gBAAaI;aAAa;YACzC,UAAU,CACNH,aACAI,eAEO,OAAO,EAAEH,IAAI,EAAE;oBAClB,IAAI,CAACG,aAAa,GAAG,GAAG,SAAS,CAAC,mCAC9B,MAAM,IAAIC,MACN;oBAIR,MAAMP,SAAS,MAAME,YAAY,OAAO,CAAoC;wBACxE,YAAYM;wBACZ,OAAO;4BACH,QAAQL,KAAK,MAAM;4BACnB,SAASA,KAAK,OAAO;4BACrB,WAAWA,KAAK,SAAS,IAAI;4BAC7B,iBAAiBA,KAAK,eAAe,IAAI;4BACzC,iBAAiBA,KAAK,eAAe,IAAI;4BACzC,iBAAiBA,KAAK,eAAe,IAAI;4BACzC,mBAAmBA,KAAK,iBAAiB,IAAI;wBACjD;oBACJ;oBAEA,IAAIH,OAAO,MAAM,IACb,MAAMA,OAAO,KAAK;oBAGtB,OAAO;wBAAE,IAAIA,OAAO,KAAK,CAAC,EAAE;oBAAC;gBACjC;QAER;QAEAP,QAAQ,WAAW,CAAqC;YACpD,MAAM;YACN,cAAc;gBACVG;gBACAa;gBACAZ;aACH;YACD,UAAU,CACNa,aACAC,gBACAZ,SAEO,OAAO,EAAEI,IAAI,EAAE;oBAClB,MAAMS,gBAAgB,MAAMF,YAAY,OAAO;oBAC/C,IAAIE,cAAc,MAAM,IACpB,OAAO,IAAIC,cAAcD,cAAc,KAAK;oBAGhD,MAAME,YAAY,MAAMf,OAAO,OAAO,CAACI,KAAK,KAAK,EAAES,cAAc,KAAK;oBACtE,MAAMZ,SAAS,MAAMW,eAAe,OAAO,CAACG;oBAE5C,IAAId,OAAO,MAAM,IACb,OAAO,IAAIa,cAAcb,OAAO,KAAK;oBAGzC,OAAO,IAAIe,SAAS,MAAMhB,OAAO,KAAK,CAACC,OAAO,KAAK;gBACvD;QAER;QAEA,OAAOP;IACX;AACJ;AAEO,MAAMuB,oBAAoBC,yBAAyB,oBAAoB,CAAC;IAC3E,gBAAgBzB;IAChB,cAAc,EAAE;AACpB"}
+{"version":3,"file":"api/graphql/BaseGraphQLSchema.js","sources":["../../../src/api/graphql/BaseGraphQLSchema.ts"],"sourcesContent":["import { CoreGraphQLSchemaFactory } from \"@webiny/api-graphql/graphql/abstractions.core.js\";\nimport { Response, ErrorResponse } from \"@webiny/api-graphql/responses.js\";\nimport { Ai } from \"@webiny/api-core/features/ai/index.js\";\nimport { TaskService } from \"@webiny/api-core/features/task/TaskService/index.js\";\nimport { FeatureFlags } from \"@webiny/api-core/features/featureFlags/abstractions.js\";\nimport { GetSettingsUseCase } from \"~/api/features/GetSettings/index.js\";\nimport { UpdateSettingsUseCase } from \"~/api/features/UpdateSettings/index.js\";\nimport { ListAiCapabilitiesUseCase } from \"~/api/features/Capabilities/index.js\";\nimport { AiPowerUpsSettingsGraphQLMapper } from \"./abstractions.js\";\nimport {\n    WB_GENERATE_PAGE_CONTENT_TASK_ID,\n    type IWbGeneratePageContentTaskInput\n} from \"~/api/features/WbGeneratePageContent/WbGeneratePageContentTask.js\";\nimport {\n    CMS_GENERATE_ENTRY_CONTENT_TASK_ID,\n    type ICmsGenerateEntryContentTaskInput\n} from \"~/api/features/CmsGenerateEntryContent/CmsGenerateEntryContentTask.js\";\n\nclass BaseGraphQLSchemaImpl implements CoreGraphQLSchemaFactory.Interface {\n    async execute(\n        builder: CoreGraphQLSchemaFactory.SchemaBuilder\n    ): Promise<CoreGraphQLSchemaFactory.SchemaBuilder> {\n        builder.addTypeDefs(/* GraphQL */ `\n            type AiModel {\n                providerId: String!\n                providerName: String!\n                modelId: String!\n                modelName: String!\n            }\n\n            type AiCapability {\n                id: String!\n                label: String!\n                description: String!\n                defaultRole: String!\n            }\n\n            type AiPowerUpsQuery {\n                listModels: [AiModel!]!\n                listCapabilities: [AiCapability!]!\n                getSettings: JSON\n            }\n\n            type AiPowerUpsError {\n                code: String\n                message: String\n                data: JSON\n                stack: String\n            }\n\n            type AiPowerUpsSettingsResponse {\n                data: JSON\n                error: AiPowerUpsError\n            }\n\n            type AiPowerUpsMutation {\n                updateSettings(input: JSON!): AiPowerUpsSettingsResponse!\n                generatePageContent(\n                    prompt: String!\n                    components: JSON!\n                    tools: JSON!\n                    projectId: String\n                    excludedFileIds: [String!]\n                    readerPersonaId: String\n                    writerPersonaId: String\n                    additionalFileIds: [String!]\n                ): JSON!\n                generateEntryContent(\n                    prompt: String!\n                    modelId: String!\n                    projectId: String\n                    excludedFileIds: [String!]\n                    readerPersonaId: String\n                    writerPersonaId: String\n                    additionalFileIds: [String!]\n                ): JSON!\n            }\n\n            extend type Query {\n                aiPowerUps: AiPowerUpsQuery\n            }\n\n            extend type Mutation {\n                aiPowerUps: AiPowerUpsMutation\n            }\n        `);\n\n        builder.addResolver({\n            path: \"Query.aiPowerUps\",\n            resolver: () => () => ({})\n        });\n\n        builder.addResolver({\n            path: \"Mutation.aiPowerUps\",\n            resolver: () => () => ({})\n        });\n\n        builder.addResolver({\n            path: \"AiPowerUpsQuery.listModels\",\n            dependencies: [Ai],\n            resolver: (ai: Ai.Interface) => {\n                return async () => ai.listModels();\n            }\n        });\n\n        /*\n         * Whatever features registered. The admin screen has no list of its own, so a feature that\n         * registers a capability gets a settings row without touching the admin app.\n         */\n        builder.addResolver({\n            path: \"AiPowerUpsQuery.listCapabilities\",\n            dependencies: [ListAiCapabilitiesUseCase],\n            resolver: (useCase: ListAiCapabilitiesUseCase.Interface) => {\n                return async () => useCase.execute();\n            }\n        });\n\n        builder.addResolver({\n            path: \"AiPowerUpsQuery.getSettings\",\n            dependencies: [GetSettingsUseCase, AiPowerUpsSettingsGraphQLMapper],\n            resolver: (\n                useCase: GetSettingsUseCase.Interface,\n                mapper: AiPowerUpsSettingsGraphQLMapper.Interface\n            ) => {\n                return async () => {\n                    const result = await useCase.execute();\n                    if (result.isFail()) {\n                        throw result.error;\n                    }\n\n                    return mapper.toApi(result.value);\n                };\n            }\n        });\n\n        builder.addResolver<IWbGeneratePageContentTaskInput>({\n            path: \"AiPowerUpsMutation.generatePageContent\",\n            dependencies: [TaskService],\n            resolver: (taskService: TaskService.Interface) => {\n                return async ({ args }) => {\n                    const result = await taskService.trigger<IWbGeneratePageContentTaskInput>({\n                        definition: WB_GENERATE_PAGE_CONTENT_TASK_ID,\n                        input: {\n                            prompt: args.prompt,\n                            components: args.components,\n                            tools: args.tools,\n                            projectId: args.projectId ?? null,\n                            excludedFileIds: args.excludedFileIds ?? null,\n                            readerPersonaId: args.readerPersonaId ?? null,\n                            writerPersonaId: args.writerPersonaId ?? null,\n                            additionalFileIds: args.additionalFileIds ?? null\n                        }\n                    });\n\n                    if (result.isFail()) {\n                        throw result.error;\n                    }\n\n                    return { id: result.value.id };\n                };\n            }\n        });\n\n        builder.addResolver<ICmsGenerateEntryContentTaskInput>({\n            path: \"AiPowerUpsMutation.generateEntryContent\",\n            dependencies: [TaskService, FeatureFlags],\n            resolver: (\n                taskService: TaskService.Interface,\n                featureFlags: FeatureFlags.Interface\n            ) => {\n                return async ({ args }) => {\n                    if (!featureFlags.get().isEnabled(\"aiPowerups.cms.entryGeneration\")) {\n                        throw new Error(\n                            \"AI entry generation cannot be used because your project license does not permit it.\"\n                        );\n                    }\n\n                    const result = await taskService.trigger<ICmsGenerateEntryContentTaskInput>({\n                        definition: CMS_GENERATE_ENTRY_CONTENT_TASK_ID,\n                        input: {\n                            prompt: args.prompt,\n                            modelId: args.modelId,\n                            projectId: args.projectId ?? null,\n                            excludedFileIds: args.excludedFileIds ?? null,\n                            readerPersonaId: args.readerPersonaId ?? null,\n                            writerPersonaId: args.writerPersonaId ?? null,\n                            additionalFileIds: args.additionalFileIds ?? null\n                        }\n                    });\n\n                    if (result.isFail()) {\n                        throw result.error;\n                    }\n\n                    return { id: result.value.id };\n                };\n            }\n        });\n\n        builder.addResolver<{ input: Record<string, unknown> }>({\n            path: \"AiPowerUpsMutation.updateSettings\",\n            dependencies: [\n                GetSettingsUseCase,\n                UpdateSettingsUseCase,\n                AiPowerUpsSettingsGraphQLMapper\n            ],\n            resolver: (\n                getSettings: GetSettingsUseCase.Interface,\n                updateSettings: UpdateSettingsUseCase.Interface,\n                mapper: AiPowerUpsSettingsGraphQLMapper.Interface\n            ) => {\n                return async ({ args }) => {\n                    const currentResult = await getSettings.execute();\n                    if (currentResult.isFail()) {\n                        return new ErrorResponse(currentResult.error);\n                    }\n\n                    const assembled = await mapper.fromApi(args.input, currentResult.value);\n                    const result = await updateSettings.execute(assembled);\n\n                    if (result.isFail()) {\n                        return new ErrorResponse(result.error);\n                    }\n\n                    return new Response(await mapper.toApi(result.value));\n                };\n            }\n        });\n\n        return builder;\n    }\n}\n\nexport const BaseGraphQLSchema = CoreGraphQLSchemaFactory.createImplementation({\n    implementation: BaseGraphQLSchemaImpl,\n    dependencies: []\n});\n"],"names":["BaseGraphQLSchemaImpl","builder","Ai","ai","ListAiCapabilitiesUseCase","useCase","GetSettingsUseCase","AiPowerUpsSettingsGraphQLMapper","mapper","result","TaskService","taskService","args","WB_GENERATE_PAGE_CONTENT_TASK_ID","FeatureFlags","featureFlags","Error","CMS_GENERATE_ENTRY_CONTENT_TASK_ID","UpdateSettingsUseCase","getSettings","updateSettings","currentResult","ErrorResponse","assembled","Response","BaseGraphQLSchema","CoreGraphQLSchemaFactory"],"mappings":";;;;;;;;;;;AAkBA,MAAMA;IACF,MAAM,QACFC,OAA+C,EACA;QAC/CA,QAAQ,WAAW,CAAe,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;QA+DnC,CAAC;QAEDA,QAAQ,WAAW,CAAC;YAChB,MAAM;YACN,UAAU,IAAM,IAAO,EAAC;QAC5B;QAEAA,QAAQ,WAAW,CAAC;YAChB,MAAM;YACN,UAAU,IAAM,IAAO,EAAC;QAC5B;QAEAA,QAAQ,WAAW,CAAC;YAChB,MAAM;YACN,cAAc;gBAACC;aAAG;YAClB,UAAU,CAACC,KACA,UAAYA,GAAG,UAAU;QAExC;QAMAF,QAAQ,WAAW,CAAC;YAChB,MAAM;YACN,cAAc;gBAACG;aAA0B;YACzC,UAAU,CAACC,UACA,UAAYA,QAAQ,OAAO;QAE1C;QAEAJ,QAAQ,WAAW,CAAC;YAChB,MAAM;YACN,cAAc;gBAACK;gBAAoBC;aAAgC;YACnE,UAAU,CACNF,SACAG,SAEO;oBACH,MAAMC,SAAS,MAAMJ,QAAQ,OAAO;oBACpC,IAAII,OAAO,MAAM,IACb,MAAMA,OAAO,KAAK;oBAGtB,OAAOD,OAAO,KAAK,CAACC,OAAO,KAAK;gBACpC;QAER;QAEAR,QAAQ,WAAW,CAAkC;YACjD,MAAM;YACN,cAAc;gBAACS;aAAY;YAC3B,UAAU,CAACC,cACA,OAAO,EAAEC,IAAI,EAAE;oBAClB,MAAMH,SAAS,MAAME,YAAY,OAAO,CAAkC;wBACtE,YAAYE;wBACZ,OAAO;4BACH,QAAQD,KAAK,MAAM;4BACnB,YAAYA,KAAK,UAAU;4BAC3B,OAAOA,KAAK,KAAK;4BACjB,WAAWA,KAAK,SAAS,IAAI;4BAC7B,iBAAiBA,KAAK,eAAe,IAAI;4BACzC,iBAAiBA,KAAK,eAAe,IAAI;4BACzC,iBAAiBA,KAAK,eAAe,IAAI;4BACzC,mBAAmBA,KAAK,iBAAiB,IAAI;wBACjD;oBACJ;oBAEA,IAAIH,OAAO,MAAM,IACb,MAAMA,OAAO,KAAK;oBAGtB,OAAO;wBAAE,IAAIA,OAAO,KAAK,CAAC,EAAE;oBAAC;gBACjC;QAER;QAEAR,QAAQ,WAAW,CAAoC;YACnD,MAAM;YACN,cAAc;gBAACS;gBAAaI;aAAa;YACzC,UAAU,CACNH,aACAI,eAEO,OAAO,EAAEH,IAAI,EAAE;oBAClB,IAAI,CAACG,aAAa,GAAG,GAAG,SAAS,CAAC,mCAC9B,MAAM,IAAIC,MACN;oBAIR,MAAMP,SAAS,MAAME,YAAY,OAAO,CAAoC;wBACxE,YAAYM;wBACZ,OAAO;4BACH,QAAQL,KAAK,MAAM;4BACnB,SAASA,KAAK,OAAO;4BACrB,WAAWA,KAAK,SAAS,IAAI;4BAC7B,iBAAiBA,KAAK,eAAe,IAAI;4BACzC,iBAAiBA,KAAK,eAAe,IAAI;4BACzC,iBAAiBA,KAAK,eAAe,IAAI;4BACzC,mBAAmBA,KAAK,iBAAiB,IAAI;wBACjD;oBACJ;oBAEA,IAAIH,OAAO,MAAM,IACb,MAAMA,OAAO,KAAK;oBAGtB,OAAO;wBAAE,IAAIA,OAAO,KAAK,CAAC,EAAE;oBAAC;gBACjC;QAER;QAEAR,QAAQ,WAAW,CAAqC;YACpD,MAAM;YACN,cAAc;gBACVK;gBACAY;gBACAX;aACH;YACD,UAAU,CACNY,aACAC,gBACAZ,SAEO,OAAO,EAAEI,IAAI,EAAE;oBAClB,MAAMS,gBAAgB,MAAMF,YAAY,OAAO;oBAC/C,IAAIE,cAAc,MAAM,IACpB,OAAO,IAAIC,cAAcD,cAAc,KAAK;oBAGhD,MAAME,YAAY,MAAMf,OAAO,OAAO,CAACI,KAAK,KAAK,EAAES,cAAc,KAAK;oBACtE,MAAMZ,SAAS,MAAMW,eAAe,OAAO,CAACG;oBAE5C,IAAId,OAAO,MAAM,IACb,OAAO,IAAIa,cAAcb,OAAO,KAAK;oBAGzC,OAAO,IAAIe,SAAS,MAAMhB,OAAO,KAAK,CAACC,OAAO,KAAK;gBACvD;QAER;QAEA,OAAOR;IACX;AACJ;AAEO,MAAMwB,oBAAoBC,yBAAyB,oBAAoB,CAAC;IAC3E,gBAAgB1B;IAChB,cAAc,EAAE;AACpB"}

package/api/types.d.ts

@@ -1,4 +1,7 @@
 import "@webiny/background-tasks/api/features/TaskController/augmentation.js";
+import "~/api/features/Connections/types.js";
+import "~/api/features/ModelRoles/types.js";
+import "~/api/features/Capabilities/types.js";
 import "~/api/features/Providers/types.js";
 import "~/api/features/Projects/types.js";
 export interface IAiPowerUpsSettings {

package/api/types.js

@@ -1,3 +1,6 @@
 import "@webiny/background-tasks/api/features/TaskController/augmentation.js";
+import "./features/Connections/types.js";
+import "./features/ModelRoles/types.js";
+import "./features/Capabilities/types.js";
 import "./features/Providers/types.js";
 import "./features/Projects/types.js";

package/exports/api/ai-powerups.d.ts

@@ -1,2 +1,7 @@
 export { CmsGenerateEntryContentUseCase } from "../../api/features/CmsGenerateEntryContent/abstractions.js";
 export { GetSettingsUseCase } from "../../api/features/GetSettings/abstractions.js";
+export { AiCapability } from "../../api/features/Capabilities/abstractions.js";
+export { ResolveAiCapabilityUseCase } from "../../api/features/Capabilities/abstractions.js";
+export { withAdditionalInstructions } from "../../api/features/Capabilities/index.js";
+export { AI_MODEL_ROLE_IDS } from "../../api/features/ModelRoles/index.js";
+export type { AiModelRoleId } from "../../api/features/ModelRoles/index.js";

package/exports/api/ai-powerups.js

@@ -1,2 +1,5 @@
 export { CmsGenerateEntryContentUseCase } from "../../api/features/CmsGenerateEntryContent/abstractions.js";
 export { GetSettingsUseCase } from "../../api/features/GetSettings/abstractions.js";
+export { AiCapability, ResolveAiCapabilityUseCase } from "../../api/features/Capabilities/abstractions.js";
+export { withAdditionalInstructions } from "../../api/features/Capabilities/index.js";
+export { AI_MODEL_ROLE_IDS } from "../../api/features/ModelRoles/index.js";