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

package/admin/presentation/WbContentGeneration/GenerateContentPresenter.js.map

@@ -1 +1 @@
-{"version":3,"file":"admin/presentation/WbContentGeneration/GenerateContentPresenter.js","sources":["../../../../src/admin/presentation/WbContentGeneration/GenerateContentPresenter.ts"],"sourcesContent":["import { makeAutoObservable, computed, toJS, runInAction } from \"mobx\";\nimport { z } from \"zod\";\nimport pick from \"lodash/pick.js\";\nimport { FormModelFactory, FormModel, ToolRegistry, ToolPipelineRunner } from \"@webiny/app-admin\";\nimport { GenerateContentPresenter, type CreateElementsFn } from \"./abstractions.js\";\nimport { GeneratePageContentUseCase } from \"~/admin/features/generatePageContent/index.js\";\nimport { GetSettingsUseCase } from \"~/admin/features/settings/getSettings/abstractions.js\";\nimport type { IAiPowerUpsSettings } from \"~/admin/features/settings/shared/abstractions.js\";\n\nconst SUBMIT_TIMEOUT_MS = 300_000;\n\nclass GenerateContentPresenterImpl implements GenerateContentPresenter.Interface {\n    private _loading = false;\n    private _submitting = false;\n    private _processing = false;\n    private _timedOut = false;\n    private _components: Record<string, any>[] = [];\n    private _createElements: CreateElementsFn = () => {};\n    private _form: FormModel.Interface | null = null;\n    private _settings: IAiPowerUpsSettings | null = null;\n    private _timeoutId: ReturnType<typeof setTimeout> | null = null;\n    private _intervalId: ReturnType<typeof setInterval> | null = null;\n    private _elapsedSeconds = 0;\n\n    constructor(\n        private toolRegistry: ToolRegistry.Interface,\n        private pipelineRunner: ToolPipelineRunner.Interface,\n        private generatePageContent: GeneratePageContentUseCase.Interface,\n        private formModelFactory: FormModelFactory.Interface,\n        private getSettings: GetSettingsUseCase.Interface\n    ) {\n        makeAutoObservable(this, { vm: computed }, { autoBind: true });\n    }\n\n    get vm(): GenerateContentPresenter.ViewModel {\n        return {\n            form: this._form ? this._form.vm : null,\n            loading: this._loading,\n            submitting: this._submitting,\n            processing: this._processing,\n            timedOut: this._timedOut,\n            elapsedSeconds: this._elapsedSeconds\n        };\n    }\n\n    async init(components: Record<string, any>[], createElements: CreateElementsFn): Promise<void> {\n        this._components = components;\n        this._createElements = createElements;\n\n        this._loading = true;\n\n        try {\n            const settings = await this.getSettings.execute();\n            runInAction(() => {\n                this._settings = settings;\n                this._form = this.buildForm();\n            });\n        } finally {\n            runInAction(() => {\n                this._loading = false;\n            });\n        }\n    }\n\n    async submit(): Promise<void> {\n        if (!this._form) {\n            return;\n        }\n\n        const data = await this._form.submit<{\n            prompt: string;\n            project?: string;\n            includedFiles?: string[];\n            readerPersona?: string;\n            writerPersona?: string;\n        }>();\n\n        if (!data) {\n            return;\n        }\n\n        runInAction(() => {\n            this._submitting = true;\n            this._timedOut = false;\n        });\n\n        this.startTimeout();\n\n        try {\n            const tools = this.toolRegistry.getTools().map(tool => ({\n                name: tool.name,\n                description: tool.description,\n                inputSchema: pick(z.toJSONSchema(tool.inputSchema), [\"properties\", \"type\"]),\n                outputSchema: pick(z.toJSONSchema(tool.outputSchema), [\"properties\", \"type\"])\n            }));\n\n            const excludedFileIds = this.computeExcludedFileIds(data.project, data.includedFiles);\n\n            await this.generatePageContent.execute({\n                prompt: data.prompt,\n                components: toJS(this._components),\n                tools,\n                projectId: data.project || null,\n                excludedFileIds,\n                readerPersonaId: data.readerPersona || null,\n                writerPersonaId: data.writerPersona || null\n            });\n        } catch {\n            this.clearTimeout();\n            runInAction(() => {\n                this._submitting = false;\n            });\n        }\n    }\n\n    async processAiResponse(responseText: string): Promise<void> {\n        this.clearTimeout();\n        runInAction(() => {\n            this._processing = true;\n            this._submitting = false;\n        });\n\n        try {\n            const aiResponseJson = JSON.parse(responseText);\n            const resolved = await this.pipelineRunner.resolve(aiResponseJson);\n            const items = Array.isArray(resolved) ? resolved : [resolved];\n\n            await this._createElements(\n                items.map((element: { component: string; inputs: Record<string, unknown> }) => ({\n                    componentName: element.component,\n                    parentId: \"root\",\n                    slot: \"children\",\n                    bindings: { inputs: element.inputs }\n                }))\n            );\n        } finally {\n            runInAction(() => {\n                this._processing = false;\n            });\n        }\n    }\n\n    cancelPrompt(): void {\n        this.clearTimeout();\n        this._processing = false;\n        this._submitting = false;\n    }\n\n    private startTimeout(): void {\n        this.clearTimeout();\n        this._elapsedSeconds = 0;\n        this._intervalId = setInterval(() => {\n            runInAction(() => {\n                this._elapsedSeconds++;\n            });\n        }, 1000);\n        this._timeoutId = setTimeout(() => {\n            runInAction(() => {\n                this._submitting = false;\n                this._timedOut = true;\n            });\n            this.clearInterval();\n        }, SUBMIT_TIMEOUT_MS);\n    }\n\n    private clearTimeout(): void {\n        if (this._timeoutId !== null) {\n            clearTimeout(this._timeoutId);\n            this._timeoutId = null;\n        }\n        this.clearInterval();\n    }\n\n    private clearInterval(): void {\n        if (this._intervalId !== null) {\n            clearInterval(this._intervalId);\n            this._intervalId = null;\n        }\n    }\n\n    private buildForm() {\n        return this.formModelFactory.create({\n            fields: fields => ({\n                project: fields\n                    .text()\n                    .label(\"Project\")\n                    .description(\"Select a predefined context to attach.\")\n                    .options(() => this.getProjectOptions())\n                    .afterChange((value, form) => {\n                        const projectId = value as string | undefined;\n                        const project = projectId\n                            ? this._settings?.projects?.presets?.find(p => p.id === projectId)\n                            : undefined;\n\n                        if (project?.defaultReaderPersonaId) {\n                            form.field(\"readerPersona\").setValue(project.defaultReaderPersonaId);\n                        }\n                        if (project?.defaultWriterPersonaId) {\n                            form.field(\"writerPersona\").setValue(project.defaultWriterPersonaId);\n                        }\n\n                        const fileIds = (project?.files ?? []).map(f => f.id);\n                        form.field(\"includedFiles\").setValue(fileIds);\n                    }),\n                includedFiles: fields\n                    .text()\n                    .label(\"Files included\")\n                    .description(\"Uncheck files to exclude them from this generation only.\")\n                    .options(() => this.getProjectFileOptions())\n                    .renderer(\"checkboxes\")\n                    .list()\n                    .hiddenWhen(form => {\n                        const projectId = form.field(\"project\").as(\"text\").getValue();\n                        if (!projectId) {\n                            return true;\n                        }\n                        const project = this._settings?.projects?.presets?.find(\n                            p => p.id === projectId\n                        );\n                        return !project?.files || project.files.length === 0;\n                    }),\n                readerPersona: fields\n                    .text()\n                    .label(\"Reader Persona\")\n                    .description(\"Select the target audience for the generated content.\")\n                    .options(() => this.getPersonaOptions(\"reader\")),\n                writerPersona: fields\n                    .text()\n                    .label(\"Writer Persona\")\n                    .description(\"Select the writing style for the generated content.\")\n                    .options(() => this.getPersonaOptions(\"writer\")),\n                prompt: fields\n                    .text()\n                    .label(\"Prompt\")\n                    .description(\"Describe the page content you want to generate.\")\n                    .required(\"Prompt is required\")\n                    .renderer(\"textarea\", { rows: 6 })\n            }),\n            layout: layout => [\n                layout.row(\"project\"),\n                layout.row(\"includedFiles\"),\n                layout.row(\"readerPersona\"),\n                layout.row(\"writerPersona\"),\n                layout.row(\"prompt\")\n            ]\n        });\n    }\n\n    private getSelectedProject() {\n        if (!this._form || !this._settings) {\n            return undefined;\n        }\n        const projectId = this._form.field(\"project\").getValue<string>();\n        if (!projectId) {\n            return undefined;\n        }\n        return this._settings.projects?.presets?.find(p => p.id === projectId);\n    }\n\n    private getProjectFileOptions() {\n        const project = this.getSelectedProject();\n        const files = project?.files;\n        if (!files || files.length === 0) {\n            return [];\n        }\n        return files.map(f => ({\n            label: f.name,\n            value: f.id\n        }));\n    }\n\n    private computeExcludedFileIds(projectId?: string, includedFiles?: string[]): string[] | null {\n        if (!projectId) {\n            return null;\n        }\n        const project = this._settings?.projects?.presets?.find(p => p.id === projectId);\n        const allFileIds = (project?.files ?? []).map(f => f.id);\n        if (allFileIds.length === 0) {\n            return null;\n        }\n        const included = new Set(includedFiles ?? []);\n        const excluded = allFileIds.filter(id => !included.has(id));\n        return excluded.length > 0 ? excluded : null;\n    }\n\n    private getProjectOptions() {\n        const presets = this._settings?.projects?.presets;\n        if (!presets || presets.length === 0) {\n            return [];\n        }\n        return presets.map(preset => ({\n            label: preset.name,\n            value: preset.id\n        }));\n    }\n\n    private getPersonaOptions(type: \"reader\" | \"writer\") {\n        const presets =\n            type === \"reader\"\n                ? this._settings?.readerPersonas?.presets\n                : this._settings?.writerPersonas?.presets;\n\n        if (!presets || presets.length === 0) {\n            return [];\n        }\n\n        return presets.map(preset => ({\n            label: preset.name,\n            value: preset.id\n        }));\n    }\n}\n\nexport const GenerateContentPresenterRegistration = GenerateContentPresenter.createImplementation({\n    implementation: GenerateContentPresenterImpl,\n    dependencies: [\n        ToolRegistry,\n        ToolPipelineRunner,\n        GeneratePageContentUseCase,\n        FormModelFactory,\n        GetSettingsUseCase\n    ]\n});\n"],"names":["SUBMIT_TIMEOUT_MS","GenerateContentPresenterImpl","toolRegistry","pipelineRunner","generatePageContent","formModelFactory","getSettings","makeAutoObservable","computed","components","createElements","settings","runInAction","data","tools","tool","pick","z","excludedFileIds","toJS","responseText","aiResponseJson","JSON","resolved","items","Array","element","setInterval","setTimeout","clearTimeout","clearInterval","fields","value","form","projectId","project","p","undefined","fileIds","f","layout","files","includedFiles","allFileIds","included","Set","excluded","id","presets","preset","type","GenerateContentPresenterRegistration","GenerateContentPresenter","ToolRegistry","ToolPipelineRunner","GeneratePageContentUseCase","FormModelFactory","GetSettingsUseCase"],"mappings":";;;;;;;AASA,MAAMA,oBAAoB;AAE1B,MAAMC;IAaF,YACYC,YAAoC,EACpCC,cAA4C,EAC5CC,mBAAyD,EACzDC,gBAA4C,EAC5CC,WAAyC,CACnD;aALUJ,YAAY,GAAZA;aACAC,cAAc,GAAdA;aACAC,mBAAmB,GAAnBA;aACAC,gBAAgB,GAAhBA;aACAC,WAAW,GAAXA;aAjBJ,QAAQ,GAAG;aACX,WAAW,GAAG;aACd,WAAW,GAAG;aACd,SAAS,GAAG;aACZ,WAAW,GAA0B,EAAE;aACvC,eAAe,GAAqB,KAAO;aAC3C,KAAK,GAA+B;aACpC,SAAS,GAA+B;aACxC,UAAU,GAAyC;aACnD,WAAW,GAA0C;aACrD,eAAe,GAAG;QAStBC,mBAAmB,IAAI,EAAE;YAAE,IAAIC;QAAS,GAAG;YAAE,UAAU;QAAK;IAChE;IAEA,IAAI,KAAyC;QACzC,OAAO;YACH,MAAM,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,EAAE,GAAG;YACnC,SAAS,IAAI,CAAC,QAAQ;YACtB,YAAY,IAAI,CAAC,WAAW;YAC5B,YAAY,IAAI,CAAC,WAAW;YAC5B,UAAU,IAAI,CAAC,SAAS;YACxB,gBAAgB,IAAI,CAAC,eAAe;QACxC;IACJ;IAEA,MAAM,KAAKC,UAAiC,EAAEC,cAAgC,EAAiB;QAC3F,IAAI,CAAC,WAAW,GAAGD;QACnB,IAAI,CAAC,eAAe,GAAGC;QAEvB,IAAI,CAAC,QAAQ,GAAG;QAEhB,IAAI;YACA,MAAMC,WAAW,MAAM,IAAI,CAAC,WAAW,CAAC,OAAO;YAC/CC,YAAY;gBACR,IAAI,CAAC,SAAS,GAAGD;gBACjB,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,SAAS;YAC/B;QACJ,SAAU;YACNC,YAAY;gBACR,IAAI,CAAC,QAAQ,GAAG;YACpB;QACJ;IACJ;IAEA,MAAM,SAAwB;QAC1B,IAAI,CAAC,IAAI,CAAC,KAAK,EACX;QAGJ,MAAMC,OAAO,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM;QAQpC,IAAI,CAACA,MACD;QAGJD,YAAY;YACR,IAAI,CAAC,WAAW,GAAG;YACnB,IAAI,CAAC,SAAS,GAAG;QACrB;QAEA,IAAI,CAAC,YAAY;QAEjB,IAAI;YACA,MAAME,QAAQ,IAAI,CAAC,YAAY,CAAC,QAAQ,GAAG,GAAG,CAACC,CAAAA,OAAS;oBACpD,MAAMA,KAAK,IAAI;oBACf,aAAaA,KAAK,WAAW;oBAC7B,aAAaC,KAAKC,EAAE,YAAY,CAACF,KAAK,WAAW,GAAG;wBAAC;wBAAc;qBAAO;oBAC1E,cAAcC,KAAKC,EAAE,YAAY,CAACF,KAAK,YAAY,GAAG;wBAAC;wBAAc;qBAAO;gBAChF;YAEA,MAAMG,kBAAkB,IAAI,CAAC,sBAAsB,CAACL,KAAK,OAAO,EAAEA,KAAK,aAAa;YAEpF,MAAM,IAAI,CAAC,mBAAmB,CAAC,OAAO,CAAC;gBACnC,QAAQA,KAAK,MAAM;gBACnB,YAAYM,KAAK,IAAI,CAAC,WAAW;gBACjCL;gBACA,WAAWD,KAAK,OAAO,IAAI;gBAC3BK;gBACA,iBAAiBL,KAAK,aAAa,IAAI;gBACvC,iBAAiBA,KAAK,aAAa,IAAI;YAC3C;QACJ,EAAE,OAAM;YACJ,IAAI,CAAC,YAAY;YACjBD,YAAY;gBACR,IAAI,CAAC,WAAW,GAAG;YACvB;QACJ;IACJ;IAEA,MAAM,kBAAkBQ,YAAoB,EAAiB;QACzD,IAAI,CAAC,YAAY;QACjBR,YAAY;YACR,IAAI,CAAC,WAAW,GAAG;YACnB,IAAI,CAAC,WAAW,GAAG;QACvB;QAEA,IAAI;YACA,MAAMS,iBAAiBC,KAAK,KAAK,CAACF;YAClC,MAAMG,WAAW,MAAM,IAAI,CAAC,cAAc,CAAC,OAAO,CAACF;YACnD,MAAMG,QAAQC,MAAM,OAAO,CAACF,YAAYA,WAAW;gBAACA;aAAS;YAE7D,MAAM,IAAI,CAAC,eAAe,CACtBC,MAAM,GAAG,CAAC,CAACE,UAAqE;oBAC5E,eAAeA,QAAQ,SAAS;oBAChC,UAAU;oBACV,MAAM;oBACN,UAAU;wBAAE,QAAQA,QAAQ,MAAM;oBAAC;gBACvC;QAER,SAAU;YACNd,YAAY;gBACR,IAAI,CAAC,WAAW,GAAG;YACvB;QACJ;IACJ;IAEA,eAAqB;QACjB,IAAI,CAAC,YAAY;QACjB,IAAI,CAAC,WAAW,GAAG;QACnB,IAAI,CAAC,WAAW,GAAG;IACvB;IAEQ,eAAqB;QACzB,IAAI,CAAC,YAAY;QACjB,IAAI,CAAC,eAAe,GAAG;QACvB,IAAI,CAAC,WAAW,GAAGe,YAAY;YAC3Bf,YAAY;gBACR,IAAI,CAAC,eAAe;YACxB;QACJ,GAAG;QACH,IAAI,CAAC,UAAU,GAAGgB,WAAW;YACzBhB,YAAY;gBACR,IAAI,CAAC,WAAW,GAAG;gBACnB,IAAI,CAAC,SAAS,GAAG;YACrB;YACA,IAAI,CAAC,aAAa;QACtB,GAAGZ;IACP;IAEQ,eAAqB;QACzB,IAAI,AAAoB,SAApB,IAAI,CAAC,UAAU,EAAW;YAC1B6B,aAAa,IAAI,CAAC,UAAU;YAC5B,IAAI,CAAC,UAAU,GAAG;QACtB;QACA,IAAI,CAAC,aAAa;IACtB;IAEQ,gBAAsB;QAC1B,IAAI,AAAqB,SAArB,IAAI,CAAC,WAAW,EAAW;YAC3BC,cAAc,IAAI,CAAC,WAAW;YAC9B,IAAI,CAAC,WAAW,GAAG;QACvB;IACJ;IAEQ,YAAY;QAChB,OAAO,IAAI,CAAC,gBAAgB,CAAC,MAAM,CAAC;YAChC,QAAQC,CAAAA,SAAW;oBACf,SAASA,OACJ,IAAI,GACJ,KAAK,CAAC,WACN,WAAW,CAAC,0CACZ,OAAO,CAAC,IAAM,IAAI,CAAC,iBAAiB,IACpC,WAAW,CAAC,CAACC,OAAOC;wBACjB,MAAMC,YAAYF;wBAClB,MAAMG,UAAUD,YACV,IAAI,CAAC,SAAS,EAAE,UAAU,SAAS,KAAKE,CAAAA,IAAKA,EAAE,EAAE,KAAKF,aACtDG;wBAEN,IAAIF,SAAS,wBACTF,KAAK,KAAK,CAAC,iBAAiB,QAAQ,CAACE,QAAQ,sBAAsB;wBAEvE,IAAIA,SAAS,wBACTF,KAAK,KAAK,CAAC,iBAAiB,QAAQ,CAACE,QAAQ,sBAAsB;wBAGvE,MAAMG,UAAWH,AAAAA,CAAAA,SAAS,SAAS,EAAC,EAAG,GAAG,CAACI,CAAAA,IAAKA,EAAE,EAAE;wBACpDN,KAAK,KAAK,CAAC,iBAAiB,QAAQ,CAACK;oBACzC;oBACJ,eAAeP,OACV,IAAI,GACJ,KAAK,CAAC,kBACN,WAAW,CAAC,4DACZ,OAAO,CAAC,IAAM,IAAI,CAAC,qBAAqB,IACxC,QAAQ,CAAC,cACT,IAAI,GACJ,UAAU,CAACE,CAAAA;wBACR,MAAMC,YAAYD,KAAK,KAAK,CAAC,WAAW,EAAE,CAAC,QAAQ,QAAQ;wBAC3D,IAAI,CAACC,WACD,OAAO;wBAEX,MAAMC,UAAU,IAAI,CAAC,SAAS,EAAE,UAAU,SAAS,KAC/CC,CAAAA,IAAKA,EAAE,EAAE,KAAKF;wBAElB,OAAO,CAACC,SAAS,SAASA,AAAyB,MAAzBA,QAAQ,KAAK,CAAC,MAAM;oBAClD;oBACJ,eAAeJ,OACV,IAAI,GACJ,KAAK,CAAC,kBACN,WAAW,CAAC,yDACZ,OAAO,CAAC,IAAM,IAAI,CAAC,iBAAiB,CAAC;oBAC1C,eAAeA,OACV,IAAI,GACJ,KAAK,CAAC,kBACN,WAAW,CAAC,uDACZ,OAAO,CAAC,IAAM,IAAI,CAAC,iBAAiB,CAAC;oBAC1C,QAAQA,OACH,IAAI,GACJ,KAAK,CAAC,UACN,WAAW,CAAC,mDACZ,QAAQ,CAAC,sBACT,QAAQ,CAAC,YAAY;wBAAE,MAAM;oBAAE;gBACxC;YACA,QAAQS,CAAAA,SAAU;oBACdA,OAAO,GAAG,CAAC;oBACXA,OAAO,GAAG,CAAC;oBACXA,OAAO,GAAG,CAAC;oBACXA,OAAO,GAAG,CAAC;oBACXA,OAAO,GAAG,CAAC;iBACd;QACL;IACJ;IAEQ,qBAAqB;QACzB,IAAI,CAAC,IAAI,CAAC,KAAK,IAAI,CAAC,IAAI,CAAC,SAAS,EAC9B;QAEJ,MAAMN,YAAY,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,WAAW,QAAQ;QACtD,IAAI,CAACA,WACD;QAEJ,OAAO,IAAI,CAAC,SAAS,CAAC,QAAQ,EAAE,SAAS,KAAKE,CAAAA,IAAKA,EAAE,EAAE,KAAKF;IAChE;IAEQ,wBAAwB;QAC5B,MAAMC,UAAU,IAAI,CAAC,kBAAkB;QACvC,MAAMM,QAAQN,SAAS;QACvB,IAAI,CAACM,SAASA,AAAiB,MAAjBA,MAAM,MAAM,EACtB,OAAO,EAAE;QAEb,OAAOA,MAAM,GAAG,CAACF,CAAAA,IAAM;gBACnB,OAAOA,EAAE,IAAI;gBACb,OAAOA,EAAE,EAAE;YACf;IACJ;IAEQ,uBAAuBL,SAAkB,EAAEQ,aAAwB,EAAmB;QAC1F,IAAI,CAACR,WACD,OAAO;QAEX,MAAMC,UAAU,IAAI,CAAC,SAAS,EAAE,UAAU,SAAS,KAAKC,CAAAA,IAAKA,EAAE,EAAE,KAAKF;QACtE,MAAMS,aAAcR,AAAAA,CAAAA,SAAS,SAAS,EAAC,EAAG,GAAG,CAACI,CAAAA,IAAKA,EAAE,EAAE;QACvD,IAAII,AAAsB,MAAtBA,WAAW,MAAM,EACjB,OAAO;QAEX,MAAMC,WAAW,IAAIC,IAAIH,iBAAiB,EAAE;QAC5C,MAAMI,WAAWH,WAAW,MAAM,CAACI,CAAAA,KAAM,CAACH,SAAS,GAAG,CAACG;QACvD,OAAOD,SAAS,MAAM,GAAG,IAAIA,WAAW;IAC5C;IAEQ,oBAAoB;QACxB,MAAME,UAAU,IAAI,CAAC,SAAS,EAAE,UAAU;QAC1C,IAAI,CAACA,WAAWA,AAAmB,MAAnBA,QAAQ,MAAM,EAC1B,OAAO,EAAE;QAEb,OAAOA,QAAQ,GAAG,CAACC,CAAAA,SAAW;gBAC1B,OAAOA,OAAO,IAAI;gBAClB,OAAOA,OAAO,EAAE;YACpB;IACJ;IAEQ,kBAAkBC,IAAyB,EAAE;QACjD,MAAMF,UACFE,AAAS,aAATA,OACM,IAAI,CAAC,SAAS,EAAE,gBAAgB,UAChC,IAAI,CAAC,SAAS,EAAE,gBAAgB;QAE1C,IAAI,CAACF,WAAWA,AAAmB,MAAnBA,QAAQ,MAAM,EAC1B,OAAO,EAAE;QAGb,OAAOA,QAAQ,GAAG,CAACC,CAAAA,SAAW;gBAC1B,OAAOA,OAAO,IAAI;gBAClB,OAAOA,OAAO,EAAE;YACpB;IACJ;AACJ;AAEO,MAAME,uCAAuCC,yBAAyB,oBAAoB,CAAC;IAC9F,gBAAgBnD;IAChB,cAAc;QACVoD;QACAC;QACAC;QACAC;QACAC;KACH;AACL"}
+{"version":3,"file":"admin/presentation/WbContentGeneration/GenerateContentPresenter.js","sources":["../../../../src/admin/presentation/WbContentGeneration/GenerateContentPresenter.ts"],"sourcesContent":["import { makeAutoObservable, computed, toJS, runInAction } from \"mobx\";\nimport { z } from \"zod\";\nimport pick from \"lodash/pick.js\";\nimport { ToolRegistry, ToolPipelineRunner } from \"@webiny/app-admin\";\nimport type { FormModel } from \"@webiny/app-admin\";\nimport type { FileValue } from \"@webiny/app-admin/features/formModel/abstractions.js\";\nimport { GenerateContentPresenter, type CreateElementsFn } from \"./abstractions.js\";\nimport { GeneratePageContentUseCase } from \"~/admin/features/generatePageContent/index.js\";\nimport { AiPromptFormFactory } from \"~/admin/presentation/AiPromptFormFactory/abstractions.js\";\n\nconst SUBMIT_TIMEOUT_MS = 300_000;\n\nclass GenerateContentPresenterImpl implements GenerateContentPresenter.Interface {\n    private _loading = false;\n    private _submitting = false;\n    private _processing = false;\n    private _timedOut = false;\n    private _components: Record<string, any>[] = [];\n    private _createElements: CreateElementsFn = () => {};\n    private _form: FormModel.Interface | null = null;\n    private _timeoutId: ReturnType<typeof setTimeout> | null = null;\n    private _intervalId: ReturnType<typeof setInterval> | null = null;\n    private _elapsedSeconds = 0;\n\n    constructor(\n        private toolRegistry: ToolRegistry.Interface,\n        private pipelineRunner: ToolPipelineRunner.Interface,\n        private generatePageContent: GeneratePageContentUseCase.Interface,\n        private formFactory: AiPromptFormFactory.Interface\n    ) {\n        makeAutoObservable(this, { vm: computed }, { autoBind: true });\n    }\n\n    get vm(): GenerateContentPresenter.ViewModel {\n        return {\n            form: this._form ? this._form.vm : null,\n            loading: this._loading,\n            submitting: this._submitting,\n            processing: this._processing,\n            timedOut: this._timedOut,\n            elapsedSeconds: this._elapsedSeconds\n        };\n    }\n\n    async init(components: Record<string, any>[], createElements: CreateElementsFn): Promise<void> {\n        this._components = components;\n        this._createElements = createElements;\n\n        this._form = this.formFactory.createForm({\n            promptDescription: \"Describe the page content you want to generate.\"\n        });\n    }\n\n    async submit(): Promise<void> {\n        if (!this._form) {\n            return;\n        }\n\n        const data = await this._form.submit<{\n            prompt: string;\n            project?: string;\n            includedFiles?: string[];\n            readerPersona?: string;\n            writerPersona?: string;\n            additionalFiles?: FileValue[];\n        }>();\n\n        if (!data) {\n            return;\n        }\n\n        runInAction(() => {\n            this._submitting = true;\n            this._timedOut = false;\n        });\n\n        this.startTimeout();\n\n        try {\n            const tools = this.toolRegistry.getTools().map(tool => ({\n                name: tool.name,\n                description: tool.description,\n                inputSchema: pick(z.toJSONSchema(tool.inputSchema), [\"properties\", \"type\"]),\n                outputSchema: pick(z.toJSONSchema(tool.outputSchema), [\"properties\", \"type\"])\n            }));\n\n            const excludedFileIds = this.formFactory.computeExcludedFileIds(this._form);\n            const additionalFileIds = data.additionalFiles?.map(f => f.id) ?? null;\n\n            await this.generatePageContent.execute({\n                prompt: data.prompt,\n                components: toJS(this._components),\n                tools,\n                projectId: data.project || null,\n                excludedFileIds,\n                readerPersonaId: data.readerPersona || null,\n                writerPersonaId: data.writerPersona || null,\n                additionalFileIds\n            });\n        } catch {\n            this.clearTimeout();\n            runInAction(() => {\n                this._submitting = false;\n            });\n        }\n    }\n\n    async processAiResponse(responseText: string): Promise<void> {\n        this.clearTimeout();\n        runInAction(() => {\n            this._processing = true;\n            this._submitting = false;\n        });\n\n        try {\n            const aiResponseJson = JSON.parse(responseText);\n            const resolved = await this.pipelineRunner.resolve(aiResponseJson);\n            const items = Array.isArray(resolved) ? resolved : [resolved];\n\n            await this._createElements(\n                items.map((element: { component: string; inputs: Record<string, unknown> }) => ({\n                    componentName: element.component,\n                    parentId: \"root\",\n                    slot: \"children\",\n                    bindings: { inputs: element.inputs }\n                }))\n            );\n        } finally {\n            runInAction(() => {\n                this._processing = false;\n            });\n        }\n    }\n\n    cancelPrompt(): void {\n        this.clearTimeout();\n        this._processing = false;\n        this._submitting = false;\n    }\n\n    private startTimeout(): void {\n        this.clearTimeout();\n        this._elapsedSeconds = 0;\n        this._intervalId = setInterval(() => {\n            runInAction(() => {\n                this._elapsedSeconds++;\n            });\n        }, 1000);\n        this._timeoutId = setTimeout(() => {\n            runInAction(() => {\n                this._submitting = false;\n                this._timedOut = true;\n            });\n            this.clearInterval();\n        }, SUBMIT_TIMEOUT_MS);\n    }\n\n    private clearTimeout(): void {\n        if (this._timeoutId !== null) {\n            clearTimeout(this._timeoutId);\n            this._timeoutId = null;\n        }\n        this.clearInterval();\n    }\n\n    private clearInterval(): void {\n        if (this._intervalId !== null) {\n            clearInterval(this._intervalId);\n            this._intervalId = null;\n        }\n    }\n}\n\nexport const GenerateContentPresenterRegistration = GenerateContentPresenter.createImplementation({\n    implementation: GenerateContentPresenterImpl,\n    dependencies: [\n        ToolRegistry,\n        ToolPipelineRunner,\n        GeneratePageContentUseCase,\n        AiPromptFormFactory\n    ]\n});\n"],"names":["SUBMIT_TIMEOUT_MS","GenerateContentPresenterImpl","toolRegistry","pipelineRunner","generatePageContent","formFactory","makeAutoObservable","computed","components","createElements","data","runInAction","tools","tool","pick","z","excludedFileIds","additionalFileIds","f","toJS","responseText","aiResponseJson","JSON","resolved","items","Array","element","setInterval","setTimeout","clearTimeout","clearInterval","GenerateContentPresenterRegistration","GenerateContentPresenter","ToolRegistry","ToolPipelineRunner","GeneratePageContentUseCase","AiPromptFormFactory"],"mappings":";;;;;;;AAUA,MAAMA,oBAAoB;AAE1B,MAAMC;IAYF,YACYC,YAAoC,EACpCC,cAA4C,EAC5CC,mBAAyD,EACzDC,WAA0C,CACpD;aAJUH,YAAY,GAAZA;aACAC,cAAc,GAAdA;aACAC,mBAAmB,GAAnBA;aACAC,WAAW,GAAXA;aAfJ,QAAQ,GAAG;aACX,WAAW,GAAG;aACd,WAAW,GAAG;aACd,SAAS,GAAG;aACZ,WAAW,GAA0B,EAAE;aACvC,eAAe,GAAqB,KAAO;aAC3C,KAAK,GAA+B;aACpC,UAAU,GAAyC;aACnD,WAAW,GAA0C;aACrD,eAAe,GAAG;QAQtBC,mBAAmB,IAAI,EAAE;YAAE,IAAIC;QAAS,GAAG;YAAE,UAAU;QAAK;IAChE;IAEA,IAAI,KAAyC;QACzC,OAAO;YACH,MAAM,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,EAAE,GAAG;YACnC,SAAS,IAAI,CAAC,QAAQ;YACtB,YAAY,IAAI,CAAC,WAAW;YAC5B,YAAY,IAAI,CAAC,WAAW;YAC5B,UAAU,IAAI,CAAC,SAAS;YACxB,gBAAgB,IAAI,CAAC,eAAe;QACxC;IACJ;IAEA,MAAM,KAAKC,UAAiC,EAAEC,cAAgC,EAAiB;QAC3F,IAAI,CAAC,WAAW,GAAGD;QACnB,IAAI,CAAC,eAAe,GAAGC;QAEvB,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,WAAW,CAAC,UAAU,CAAC;YACrC,mBAAmB;QACvB;IACJ;IAEA,MAAM,SAAwB;QAC1B,IAAI,CAAC,IAAI,CAAC,KAAK,EACX;QAGJ,MAAMC,OAAO,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM;QASpC,IAAI,CAACA,MACD;QAGJC,YAAY;YACR,IAAI,CAAC,WAAW,GAAG;YACnB,IAAI,CAAC,SAAS,GAAG;QACrB;QAEA,IAAI,CAAC,YAAY;QAEjB,IAAI;YACA,MAAMC,QAAQ,IAAI,CAAC,YAAY,CAAC,QAAQ,GAAG,GAAG,CAACC,CAAAA,OAAS;oBACpD,MAAMA,KAAK,IAAI;oBACf,aAAaA,KAAK,WAAW;oBAC7B,aAAaC,KAAKC,EAAE,YAAY,CAACF,KAAK,WAAW,GAAG;wBAAC;wBAAc;qBAAO;oBAC1E,cAAcC,KAAKC,EAAE,YAAY,CAACF,KAAK,YAAY,GAAG;wBAAC;wBAAc;qBAAO;gBAChF;YAEA,MAAMG,kBAAkB,IAAI,CAAC,WAAW,CAAC,sBAAsB,CAAC,IAAI,CAAC,KAAK;YAC1E,MAAMC,oBAAoBP,KAAK,eAAe,EAAE,IAAIQ,CAAAA,IAAKA,EAAE,EAAE,KAAK;YAElE,MAAM,IAAI,CAAC,mBAAmB,CAAC,OAAO,CAAC;gBACnC,QAAQR,KAAK,MAAM;gBACnB,YAAYS,KAAK,IAAI,CAAC,WAAW;gBACjCP;gBACA,WAAWF,KAAK,OAAO,IAAI;gBAC3BM;gBACA,iBAAiBN,KAAK,aAAa,IAAI;gBACvC,iBAAiBA,KAAK,aAAa,IAAI;gBACvCO;YACJ;QACJ,EAAE,OAAM;YACJ,IAAI,CAAC,YAAY;YACjBN,YAAY;gBACR,IAAI,CAAC,WAAW,GAAG;YACvB;QACJ;IACJ;IAEA,MAAM,kBAAkBS,YAAoB,EAAiB;QACzD,IAAI,CAAC,YAAY;QACjBT,YAAY;YACR,IAAI,CAAC,WAAW,GAAG;YACnB,IAAI,CAAC,WAAW,GAAG;QACvB;QAEA,IAAI;YACA,MAAMU,iBAAiBC,KAAK,KAAK,CAACF;YAClC,MAAMG,WAAW,MAAM,IAAI,CAAC,cAAc,CAAC,OAAO,CAACF;YACnD,MAAMG,QAAQC,MAAM,OAAO,CAACF,YAAYA,WAAW;gBAACA;aAAS;YAE7D,MAAM,IAAI,CAAC,eAAe,CACtBC,MAAM,GAAG,CAAC,CAACE,UAAqE;oBAC5E,eAAeA,QAAQ,SAAS;oBAChC,UAAU;oBACV,MAAM;oBACN,UAAU;wBAAE,QAAQA,QAAQ,MAAM;oBAAC;gBACvC;QAER,SAAU;YACNf,YAAY;gBACR,IAAI,CAAC,WAAW,GAAG;YACvB;QACJ;IACJ;IAEA,eAAqB;QACjB,IAAI,CAAC,YAAY;QACjB,IAAI,CAAC,WAAW,GAAG;QACnB,IAAI,CAAC,WAAW,GAAG;IACvB;IAEQ,eAAqB;QACzB,IAAI,CAAC,YAAY;QACjB,IAAI,CAAC,eAAe,GAAG;QACvB,IAAI,CAAC,WAAW,GAAGgB,YAAY;YAC3BhB,YAAY;gBACR,IAAI,CAAC,eAAe;YACxB;QACJ,GAAG;QACH,IAAI,CAAC,UAAU,GAAGiB,WAAW;YACzBjB,YAAY;gBACR,IAAI,CAAC,WAAW,GAAG;gBACnB,IAAI,CAAC,SAAS,GAAG;YACrB;YACA,IAAI,CAAC,aAAa;QACtB,GAAGX;IACP;IAEQ,eAAqB;QACzB,IAAI,AAAoB,SAApB,IAAI,CAAC,UAAU,EAAW;YAC1B6B,aAAa,IAAI,CAAC,UAAU;YAC5B,IAAI,CAAC,UAAU,GAAG;QACtB;QACA,IAAI,CAAC,aAAa;IACtB;IAEQ,gBAAsB;QAC1B,IAAI,AAAqB,SAArB,IAAI,CAAC,WAAW,EAAW;YAC3BC,cAAc,IAAI,CAAC,WAAW;YAC9B,IAAI,CAAC,WAAW,GAAG;QACvB;IACJ;AACJ;AAEO,MAAMC,uCAAuCC,yBAAyB,oBAAoB,CAAC;IAC9F,gBAAgB/B;IAChB,cAAc;QACVgC;QACAC;QACAC;QACAC;KACH;AACL"}

package/api/Extension.js

@@ -1,5 +1,4 @@
 import { createFeature } from "@webiny/feature/api";
-import { WcpContext } from "@webiny/api-core/features/wcp/WcpContext/index.js";
 import { BaseGraphQLSchema } from "./graphql/BaseGraphQLSchema.js";
 import AiPowerUpsSettingsGraphQLMapper from "./graphql/AiPowerUpsSettingsGraphQLMapper.js";
 import { AiPowerUpsSettingsCache } from "./features/shared/SettingsCache.js";
@@ -13,6 +12,12 @@ import { ProjectsFeature } from "./features/Projects/feature.js";
 import { AiPromptContextFeature } from "./features/AiPromptContext/feature.js";
 import { AiImageEnrichmentFeature } from "./features/AiImageEnrichment/feature.js";
 import { ExtractFrontmatterFeature } from "./features/ExtractFrontmatter/feature.js";
+import { CmsGenerateEntryContentFeature } from "./features/CmsGenerateEntryContent/feature.js";
+import { CmsResolveImageToolFeature } from "./features/CmsResolveImageTool/feature.js";
+import { AiChatProviderFeature } from "./features/AiChatProvider/index.js";
+import { CmsCompareEntryRevisionsFeature } from "./features/CmsCompareEntryRevisions/feature.js";
+import { WbTranslatePageFeature } from "./features/WbTranslatePage/feature.js";
+import { CmsCompareEntryRevisionsSchema } from "./graphql/CmsCompareEntryRevisionsSchema.js";
 const Extension = createFeature({
     name: "AiPowerUps",
     register (container) {
@@ -25,11 +30,16 @@ const Extension = createFeature({
         ProjectsFeature.register(container);
         AiPromptContextFeature.register(container);
         WbGeneratePageContentFeature.register(container);
+        CmsGenerateEntryContentFeature.register(container);
+        CmsResolveImageToolFeature.register(container);
+        AiChatProviderFeature.register(container);
         ExtractFrontmatterFeature.register(container);
-        const wcp = container.resolve(WcpContext);
-        if (wcp.canUseAiImageEnrichment()) AiImageEnrichmentFeature.register(container);
+        AiImageEnrichmentFeature.register(container);
+        CmsCompareEntryRevisionsFeature.register(container);
+        WbTranslatePageFeature.register(container);
         container.register(AiPowerUpsSettingsGraphQLMapper).inSingletonScope();
         container.register(BaseGraphQLSchema);
+        container.register(CmsCompareEntryRevisionsSchema);
     }
 });
 export { Extension };

package/api/Extension.js.map

@@ -1 +1 @@
-{"version":3,"file":"api/Extension.js","sources":["../../src/api/Extension.ts"],"sourcesContent":["import { createFeature } from \"@webiny/feature/api\";\nimport { WcpContext } from \"@webiny/api-core/features/wcp/WcpContext/index.js\";\nimport { BaseGraphQLSchema } from \"./graphql/BaseGraphQLSchema.js\";\nimport AiPowerUpsSettingsGraphQLMapperImpl from \"./graphql/AiPowerUpsSettingsGraphQLMapper.js\";\nimport { AiPowerUpsSettingsCache } from \"./features/shared/SettingsCache.js\";\nimport { GetSettingsFeature } from \"./features/GetSettings/feature.js\";\nimport { UpdateSettingsFeature } from \"./features/UpdateSettings/feature.js\";\nimport { WbGeneratePageContentFeature } from \"./features/WbGeneratePageContent/feature.js\";\nimport { ProvidersFeature } from \"./features/Providers/feature.js\";\nimport { ReaderPersonasFeature } from \"./features/ReaderPersonas/feature.js\";\nimport { WriterPersonasFeature } from \"./features/WriterPersonas/feature.js\";\nimport { ProjectsFeature } from \"./features/Projects/feature.js\";\nimport { AiPromptContextFeature } from \"./features/AiPromptContext/feature.js\";\nimport { AiImageEnrichmentFeature } from \"./features/AiImageEnrichment/feature.js\";\nimport { ExtractFrontmatterFeature } from \"./features/ExtractFrontmatter/feature.js\";\n\nexport const Extension = createFeature({\n    name: \"AiPowerUps\",\n    register(container) {\n        container.register(AiPowerUpsSettingsCache).inSingletonScope();\n\n        GetSettingsFeature.register(container);\n        UpdateSettingsFeature.register(container);\n        ProvidersFeature.register(container);\n        ReaderPersonasFeature.register(container);\n        WriterPersonasFeature.register(container);\n        ProjectsFeature.register(container);\n        AiPromptContextFeature.register(container);\n        WbGeneratePageContentFeature.register(container);\n        ExtractFrontmatterFeature.register(container);\n\n        const wcp = container.resolve(WcpContext);\n        if (wcp.canUseAiImageEnrichment()) {\n            AiImageEnrichmentFeature.register(container);\n        }\n\n        container.register(AiPowerUpsSettingsGraphQLMapperImpl).inSingletonScope();\n        container.register(BaseGraphQLSchema);\n    }\n});\n"],"names":["Extension","createFeature","container","AiPowerUpsSettingsCache","GetSettingsFeature","UpdateSettingsFeature","ProvidersFeature","ReaderPersonasFeature","WriterPersonasFeature","ProjectsFeature","AiPromptContextFeature","WbGeneratePageContentFeature","ExtractFrontmatterFeature","wcp","WcpContext","AiImageEnrichmentFeature","AiPowerUpsSettingsGraphQLMapperImpl","BaseGraphQLSchema"],"mappings":";;;;;;;;;;;;;;;AAgBO,MAAMA,YAAYC,cAAc;IACnC,MAAM;IACN,UAASC,SAAS;QACdA,UAAU,QAAQ,CAACC,yBAAyB,gBAAgB;QAE5DC,mBAAmB,QAAQ,CAACF;QAC5BG,sBAAsB,QAAQ,CAACH;QAC/BI,iBAAiB,QAAQ,CAACJ;QAC1BK,sBAAsB,QAAQ,CAACL;QAC/BM,sBAAsB,QAAQ,CAACN;QAC/BO,gBAAgB,QAAQ,CAACP;QACzBQ,uBAAuB,QAAQ,CAACR;QAChCS,6BAA6B,QAAQ,CAACT;QACtCU,0BAA0B,QAAQ,CAACV;QAEnC,MAAMW,MAAMX,UAAU,OAAO,CAACY;QAC9B,IAAID,IAAI,uBAAuB,IAC3BE,yBAAyB,QAAQ,CAACb;QAGtCA,UAAU,QAAQ,CAACc,iCAAqC,gBAAgB;QACxEd,UAAU,QAAQ,CAACe;IACvB;AACJ"}
+{"version":3,"file":"api/Extension.js","sources":["../../src/api/Extension.ts"],"sourcesContent":["import { createFeature } from \"@webiny/feature/api\";\nimport { BaseGraphQLSchema } from \"./graphql/BaseGraphQLSchema.js\";\nimport AiPowerUpsSettingsGraphQLMapperImpl from \"./graphql/AiPowerUpsSettingsGraphQLMapper.js\";\nimport { AiPowerUpsSettingsCache } from \"./features/shared/SettingsCache.js\";\nimport { GetSettingsFeature } from \"./features/GetSettings/feature.js\";\nimport { UpdateSettingsFeature } from \"./features/UpdateSettings/feature.js\";\nimport { WbGeneratePageContentFeature } from \"./features/WbGeneratePageContent/feature.js\";\nimport { ProvidersFeature } from \"./features/Providers/feature.js\";\nimport { ReaderPersonasFeature } from \"./features/ReaderPersonas/feature.js\";\nimport { WriterPersonasFeature } from \"./features/WriterPersonas/feature.js\";\nimport { ProjectsFeature } from \"./features/Projects/feature.js\";\nimport { AiPromptContextFeature } from \"./features/AiPromptContext/feature.js\";\nimport { AiImageEnrichmentFeature } from \"./features/AiImageEnrichment/feature.js\";\nimport { ExtractFrontmatterFeature } from \"./features/ExtractFrontmatter/feature.js\";\nimport { CmsGenerateEntryContentFeature } from \"./features/CmsGenerateEntryContent/feature.js\";\nimport { CmsResolveImageToolFeature } from \"./features/CmsResolveImageTool/feature.js\";\nimport { AiChatProviderFeature } from \"./features/AiChatProvider/index.js\";\nimport { CmsCompareEntryRevisionsFeature } from \"./features/CmsCompareEntryRevisions/feature.js\";\nimport { WbTranslatePageFeature } from \"./features/WbTranslatePage/feature.js\";\nimport { CmsCompareEntryRevisionsSchema } from \"./graphql/CmsCompareEntryRevisionsSchema.js\";\n\nexport const Extension = createFeature({\n    name: \"AiPowerUps\",\n    register(container) {\n        container.register(AiPowerUpsSettingsCache).inSingletonScope();\n\n        GetSettingsFeature.register(container);\n        UpdateSettingsFeature.register(container);\n        ProvidersFeature.register(container);\n        ReaderPersonasFeature.register(container);\n        WriterPersonasFeature.register(container);\n        ProjectsFeature.register(container);\n        AiPromptContextFeature.register(container);\n        WbGeneratePageContentFeature.register(container);\n        CmsGenerateEntryContentFeature.register(container);\n        CmsResolveImageToolFeature.register(container);\n        AiChatProviderFeature.register(container);\n        ExtractFrontmatterFeature.register(container);\n\n        // Registered unconditionally. The WCP license gate lives inside the feature's\n        // FileAfterCreate handler (trigger-time), because the license isn't loaded yet during this\n        // register() phase — a register-time canUse* check reads NullLicense and is always false.\n        AiImageEnrichmentFeature.register(container);\n\n        CmsCompareEntryRevisionsFeature.register(container);\n        WbTranslatePageFeature.register(container);\n\n        container.register(AiPowerUpsSettingsGraphQLMapperImpl).inSingletonScope();\n        container.register(BaseGraphQLSchema);\n        container.register(CmsCompareEntryRevisionsSchema);\n    }\n});\n"],"names":["Extension","createFeature","container","AiPowerUpsSettingsCache","GetSettingsFeature","UpdateSettingsFeature","ProvidersFeature","ReaderPersonasFeature","WriterPersonasFeature","ProjectsFeature","AiPromptContextFeature","WbGeneratePageContentFeature","CmsGenerateEntryContentFeature","CmsResolveImageToolFeature","AiChatProviderFeature","ExtractFrontmatterFeature","AiImageEnrichmentFeature","CmsCompareEntryRevisionsFeature","WbTranslatePageFeature","AiPowerUpsSettingsGraphQLMapperImpl","BaseGraphQLSchema","CmsCompareEntryRevisionsSchema"],"mappings":";;;;;;;;;;;;;;;;;;;;AAqBO,MAAMA,YAAYC,cAAc;IACnC,MAAM;IACN,UAASC,SAAS;QACdA,UAAU,QAAQ,CAACC,yBAAyB,gBAAgB;QAE5DC,mBAAmB,QAAQ,CAACF;QAC5BG,sBAAsB,QAAQ,CAACH;QAC/BI,iBAAiB,QAAQ,CAACJ;QAC1BK,sBAAsB,QAAQ,CAACL;QAC/BM,sBAAsB,QAAQ,CAACN;QAC/BO,gBAAgB,QAAQ,CAACP;QACzBQ,uBAAuB,QAAQ,CAACR;QAChCS,6BAA6B,QAAQ,CAACT;QACtCU,+BAA+B,QAAQ,CAACV;QACxCW,2BAA2B,QAAQ,CAACX;QACpCY,sBAAsB,QAAQ,CAACZ;QAC/Ba,0BAA0B,QAAQ,CAACb;QAKnCc,yBAAyB,QAAQ,CAACd;QAElCe,gCAAgC,QAAQ,CAACf;QACzCgB,uBAAuB,QAAQ,CAAChB;QAEhCA,UAAU,QAAQ,CAACiB,iCAAqC,gBAAgB;QACxEjB,UAAU,QAAQ,CAACkB;QACnBlB,UAAU,QAAQ,CAACmB;IACvB;AACJ"}

package/api/features/AiChatProvider/PowerUpsAiChatProvider.d.ts

@@ -0,0 +1,24 @@
+import { AiChatProvider as Abstraction } from "@webiny/ai-chat/api/index.js";
+import type { IAiChatProviderResolution } from "@webiny/ai-chat/api/index.js";
+import { Encryption } from "@webiny/api-core/features/encryption/index.js";
+import { GetSettingsUseCase } from "../../../api/features/GetSettings/index.js";
+/**
+ * Runs the assistant on the provider configured in AI Power-Ups settings.
+ *
+ * Overrides the environment-variable default so the model and its key are managed where every other
+ * AI feature in the project already manages them — in the admin UI, per tenant, with the key
+ * encrypted at rest. Same source `CmsGenerateEntryContent` and `CmsCompareEntryRevisions` read.
+ *
+ * Uses the FIRST configured provider, matching the other AI Power-Ups use cases. Picking per-feature
+ * providers is a settings design question, not something to invent here.
+ */
+declare class PowerUpsAiChatProviderImpl implements Abstraction.Interface {
+    private readonly getSettings;
+    private readonly encryption;
+    constructor(getSettings: GetSettingsUseCase.Interface, encryption: Encryption.Interface);
+    resolve(): Promise<IAiChatProviderResolution>;
+}
+export declare const PowerUpsAiChatProvider: typeof PowerUpsAiChatProviderImpl & {
+    __abstraction: import("@webiny/di").Abstraction<import("@webiny/ai-chat/api/abstractions").IAiChatProvider>;
+};
+export {};

package/api/features/AiChatProvider/PowerUpsAiChatProvider.js

@@ -0,0 +1,31 @@
+import { AiChatProvider } from "@webiny/ai-chat/api/index.js";
+import { Encryption } from "@webiny/api-core/features/encryption/index.js";
+import { GetSettingsUseCase } from "../GetSettings/index.js";
+class PowerUpsAiChatProviderImpl {
+    constructor(getSettings, encryption){
+        this.getSettings = getSettings;
+        this.encryption = encryption;
+    }
+    async resolve() {
+        const settingsResult = await this.getSettings.execute();
+        if (settingsResult.isFail()) throw new Error("Failed to load AI Power Ups settings.");
+        const provider = settingsResult.value.providers.presets[0];
+        if (!provider) throw new Error("No AI provider configured. Add one under Settings → AI Power-Ups → Providers.");
+        if (!provider.model) throw new Error(`The AI provider "${provider.name}" has no model selected. Pick one under Settings → AI Power-Ups → Providers.`);
+        if (!provider.apiKeyEncrypted) throw new Error(`The AI provider "${provider.name}" has no API key. Add one under Settings → AI Power-Ups → Providers.`);
+        return {
+            model: provider.model,
+            apiKey: await this.encryption.decrypt(provider.apiKeyEncrypted)
+        };
+    }
+}
+const PowerUpsAiChatProvider = AiChatProvider.createImplementation({
+    implementation: PowerUpsAiChatProviderImpl,
+    dependencies: [
+        GetSettingsUseCase,
+        Encryption
+    ]
+});
+export { PowerUpsAiChatProvider };
+
+//# sourceMappingURL=PowerUpsAiChatProvider.js.map

package/api/features/AiChatProvider/PowerUpsAiChatProvider.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"api/features/AiChatProvider/PowerUpsAiChatProvider.js","sources":["../../../../src/api/features/AiChatProvider/PowerUpsAiChatProvider.ts"],"sourcesContent":["import { AiChatProvider as Abstraction } from \"@webiny/ai-chat/api/index.js\";\nimport type { IAiChatProviderResolution } from \"@webiny/ai-chat/api/index.js\";\nimport { Encryption } from \"@webiny/api-core/features/encryption/index.js\";\nimport { GetSettingsUseCase } from \"~/api/features/GetSettings/index.js\";\n\n/**\n * Runs the assistant on the provider configured in AI Power-Ups settings.\n *\n * Overrides the environment-variable default so the model and its key are managed where every other\n * AI feature in the project already manages them — in the admin UI, per tenant, with the key\n * encrypted at rest. Same source `CmsGenerateEntryContent` and `CmsCompareEntryRevisions` read.\n *\n * Uses the FIRST configured provider, matching the other AI Power-Ups use cases. Picking per-feature\n * providers is a settings design question, not something to invent here.\n */\nclass PowerUpsAiChatProviderImpl implements Abstraction.Interface {\n    constructor(\n        private readonly getSettings: GetSettingsUseCase.Interface,\n        private readonly encryption: Encryption.Interface\n    ) {}\n\n    async resolve(): Promise<IAiChatProviderResolution> {\n        const settingsResult = await this.getSettings.execute();\n\n        if (settingsResult.isFail()) {\n            throw new Error(\"Failed to load AI Power Ups settings.\");\n        }\n\n        const provider = settingsResult.value.providers.presets[0];\n\n        if (!provider) {\n            throw new Error(\n                \"No AI provider configured. Add one under Settings → AI Power-Ups → Providers.\"\n            );\n        }\n\n        if (!provider.model) {\n            throw new Error(\n                `The AI provider \"${provider.name}\" has no model selected. Pick one under Settings → AI Power-Ups → Providers.`\n            );\n        }\n\n        /*\n         * A row can exist before a key is entered. Erroring here is deliberate: falling back to an\n         * environment variable when a provider IS configured is what made the previous behaviour so\n         * hard to reason about — the settings screen said one thing and the request used another.\n         */\n        if (!provider.apiKeyEncrypted) {\n            throw new Error(\n                `The AI provider \"${provider.name}\" has no API key. Add one under Settings → AI Power-Ups → Providers.`\n            );\n        }\n\n        return {\n            model: provider.model,\n            apiKey: await this.encryption.decrypt(provider.apiKeyEncrypted)\n        };\n    }\n}\n\nexport const PowerUpsAiChatProvider = Abstraction.createImplementation({\n    implementation: PowerUpsAiChatProviderImpl,\n    dependencies: [GetSettingsUseCase, Encryption]\n});\n"],"names":["PowerUpsAiChatProviderImpl","getSettings","encryption","settingsResult","Error","provider","PowerUpsAiChatProvider","Abstraction","GetSettingsUseCase","Encryption"],"mappings":";;;AAeA,MAAMA;IACF,YACqBC,WAAyC,EACzCC,UAAgC,CACnD;aAFmBD,WAAW,GAAXA;aACAC,UAAU,GAAVA;IAClB;IAEH,MAAM,UAA8C;QAChD,MAAMC,iBAAiB,MAAM,IAAI,CAAC,WAAW,CAAC,OAAO;QAErD,IAAIA,eAAe,MAAM,IACrB,MAAM,IAAIC,MAAM;QAGpB,MAAMC,WAAWF,eAAe,KAAK,CAAC,SAAS,CAAC,OAAO,CAAC,EAAE;QAE1D,IAAI,CAACE,UACD,MAAM,IAAID,MACN;QAIR,IAAI,CAACC,SAAS,KAAK,EACf,MAAM,IAAID,MACN,CAAC,iBAAiB,EAAEC,SAAS,IAAI,CAAC,4EAA4E,CAAC;QASvH,IAAI,CAACA,SAAS,eAAe,EACzB,MAAM,IAAID,MACN,CAAC,iBAAiB,EAAEC,SAAS,IAAI,CAAC,oEAAoE,CAAC;QAI/G,OAAO;YACH,OAAOA,SAAS,KAAK;YACrB,QAAQ,MAAM,IAAI,CAAC,UAAU,CAAC,OAAO,CAACA,SAAS,eAAe;QAClE;IACJ;AACJ;AAEO,MAAMC,yBAAyBC,eAAAA,oBAAgC,CAAC;IACnE,gBAAgBP;IAChB,cAAc;QAACQ;QAAoBC;KAAW;AAClD"}

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

@@ -0,0 +1,9 @@
+/**
+ * Points the admin AI assistant at the providers configured in AI Power-Ups.
+ *
+ * Registered after `AiChatFeature`'s environment default so this one wins.
+ */
+export declare const AiChatProviderFeature: {
+    name: string;
+    register(container: import("@webiny/di").Container): void;
+};

package/api/features/AiChatProvider/feature.js

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

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

@@ -0,0 +1 @@
+{"version":3,"file":"api/features/AiChatProvider/feature.js","sources":["../../../../src/api/features/AiChatProvider/feature.ts"],"sourcesContent":["import { createFeature } from \"@webiny/feature/api\";\nimport { PowerUpsAiChatProvider } from \"./PowerUpsAiChatProvider.js\";\n\n/**\n * Points the admin AI assistant at the providers configured in AI Power-Ups.\n *\n * Registered after `AiChatFeature`'s environment default so this one wins.\n */\nexport const AiChatProviderFeature = createFeature({\n    name: \"AiPowerUps/AiChatProvider\",\n    register(container) {\n        container.register(PowerUpsAiChatProvider);\n    }\n});\n"],"names":["AiChatProviderFeature","createFeature","container","PowerUpsAiChatProvider"],"mappings":";;AAQO,MAAMA,wBAAwBC,cAAc;IAC/C,MAAM;IACN,UAASC,SAAS;QACdA,UAAU,QAAQ,CAACC;IACvB;AACJ"}

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

@@ -0,0 +1,2 @@
+export { AiChatProviderFeature } from "./feature.js";
+export { PowerUpsAiChatProvider } from "./PowerUpsAiChatProvider.js";

package/api/features/AiChatProvider/index.js

@@ -0,0 +1,2 @@
+export { AiChatProviderFeature } from "./feature.js";
+export { PowerUpsAiChatProvider } from "./PowerUpsAiChatProvider.js";

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

@@ -0,0 +1,36 @@
+import { HttpRouteDefinition, HttpRouteHandler } from "@webiny/event-handler-core";
+import type { IHttpRequest, IHttpResponseBuilder } from "@webiny/event-handler-core";
+import { Ai } from "@webiny/api-core/features/ai/index.js";
+import { PrepareImageEnrichmentUseCase } from "./abstractions.js";
+/**
+ * Re-runs AI enrichment for a single file, streaming the model's output to the caller as
+ * server-sent events. Read-only: the result is a proposal the caller can accept or discard.
+ *
+ * Preparation (file lookup, image read, provider resolution) happens BEFORE the response opens, so
+ * anything knowable up front — missing file, wrong type, no provider — comes back as a real status
+ * code. Only the AI call itself streams; once the first byte is out the status is committed to 200
+ * and failures can only be reported as an `error` event.
+ */
+declare class AiImageEnrichmentStreamRouteImpl implements HttpRouteHandler.Interface {
+    private prepare;
+    private ai;
+    constructor(prepare: PrepareImageEnrichmentUseCase.Interface, ai: Ai.Interface);
+    handle(request: IHttpRequest, response: IHttpResponseBuilder): Promise<IHttpResponseBuilder>;
+    private enrich;
+}
+export declare const AiImageEnrichmentStreamRoute: typeof AiImageEnrichmentStreamRouteImpl & {
+    __abstraction: import("@webiny/di").Abstraction<import("@webiny/event-handler-core").IHttpRoute>;
+};
+declare class AiImageEnrichmentStreamRouteDefinitionImpl implements HttpRouteDefinition.Interface {
+    readonly name = "ai-image-enrichment-stream";
+    readonly method = "POST";
+    readonly path = "/stream/fm/files/:fileId/enrich";
+    readonly handler: typeof AiImageEnrichmentStreamRouteImpl & {
+        __abstraction: import("@webiny/di").Abstraction<import("@webiny/event-handler-core").IHttpRoute>;
+    };
+}
+/** What the router matches on. Zero dependencies, so building it costs nothing. */
+export declare const AiImageEnrichmentStreamRouteDefinition: typeof AiImageEnrichmentStreamRouteDefinitionImpl & {
+    __abstraction: import("@webiny/di").Abstraction<import("@webiny/event-handler-core").IHttpRouteDefinition>;
+};
+export {};

package/api/features/AiImageEnrichment/AiImageEnrichmentStreamRoute.js

@@ -0,0 +1,85 @@
+import { HttpRouteDefinition, HttpRouteHandler, toSseFrame } from "@webiny/event-handler-core";
+import { Ai } from "@webiny/api-core/features/ai/index.js";
+import { PrepareImageEnrichmentUseCase } from "./abstractions.js";
+import { buildEnrichmentAiRequest } from "./buildEnrichmentAiRequest.js";
+import { readEnrichmentPartial } from "./readEnrichmentPartial.js";
+import { imageEnrichmentErrorStatusCode } from "./imageEnrichmentErrorStatusCode.js";
+class AiImageEnrichmentStreamRouteImpl {
+    constructor(prepare, ai){
+        this.prepare = prepare;
+        this.ai = ai;
+    }
+    async handle(request, response) {
+        const fileId = request.pathParameters.fileId;
+        if (!fileId) return response.status(400).json({
+            message: "Missing file ID."
+        });
+        const preparedResult = await this.prepare.execute(fileId);
+        if (preparedResult.isFail()) {
+            const error = preparedResult.error;
+            const statusCode = imageEnrichmentErrorStatusCode(error);
+            return response.status(statusCode).json({
+                message: error.message,
+                code: error.code
+            });
+        }
+        const events = this.enrich(preparedResult.value);
+        return response.sse(events);
+    }
+    async *enrich(prepared) {
+        yield toSseFrame({
+            type: "start",
+            fileId: prepared.fileId,
+            model: prepared.model
+        });
+        let output = {
+            tags: [],
+            description: ""
+        };
+        try {
+            const request = buildEnrichmentAiRequest(prepared);
+            const stream = await this.ai.streamText(request);
+            for await (const partial of stream.partialOutputStream){
+                output = readEnrichmentPartial(partial);
+                yield toSseFrame({
+                    type: "partial",
+                    ...output
+                });
+            }
+        } catch (error) {
+            yield toSseFrame({
+                type: "error",
+                message: `AI enrichment failed: ${error instanceof Error ? error.message : String(error)}`
+            });
+            return;
+        }
+        yield toSseFrame({
+            type: "done",
+            fileId: prepared.fileId,
+            tags: output.tags,
+            description: output.description
+        });
+    }
+}
+const AiImageEnrichmentStreamRoute = HttpRouteHandler.createImplementation({
+    implementation: AiImageEnrichmentStreamRouteImpl,
+    dependencies: [
+        PrepareImageEnrichmentUseCase,
+        Ai
+    ]
+});
+class AiImageEnrichmentStreamRouteDefinitionImpl {
+    constructor(){
+        this.name = "ai-image-enrichment-stream";
+        this.method = "POST";
+        this.path = "/stream/fm/files/:fileId/enrich";
+        this.handler = AiImageEnrichmentStreamRoute;
+    }
+}
+const AiImageEnrichmentStreamRouteDefinition = HttpRouteDefinition.createImplementation({
+    implementation: AiImageEnrichmentStreamRouteDefinitionImpl,
+    dependencies: []
+});
+export { AiImageEnrichmentStreamRoute, AiImageEnrichmentStreamRouteDefinition };
+
+//# sourceMappingURL=AiImageEnrichmentStreamRoute.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"api/features/AiImageEnrichment/AiImageEnrichmentStreamRoute.js","sources":["../../../../src/api/features/AiImageEnrichment/AiImageEnrichmentStreamRoute.ts"],"sourcesContent":["import { HttpRouteDefinition, HttpRouteHandler, toSseFrame } from \"@webiny/event-handler-core\";\nimport type { IHttpRequest, IHttpResponseBuilder } from \"@webiny/event-handler-core\";\nimport { Ai } from \"@webiny/api-core/features/ai/index.js\";\nimport { PrepareImageEnrichmentUseCase } from \"./abstractions.js\";\nimport type { IPreparedImageEnrichment } from \"./abstractions.js\";\nimport { buildEnrichmentAiRequest } from \"./buildEnrichmentAiRequest.js\";\nimport { readEnrichmentPartial } from \"./readEnrichmentPartial.js\";\nimport { imageEnrichmentErrorStatusCode } from \"./imageEnrichmentErrorStatusCode.js\";\n\n/**\n * Re-runs AI enrichment for a single file, streaming the model's output to the caller as\n * server-sent events. Read-only: the result is a proposal the caller can accept or discard.\n *\n * Preparation (file lookup, image read, provider resolution) happens BEFORE the response opens, so\n * anything knowable up front — missing file, wrong type, no provider — comes back as a real status\n * code. Only the AI call itself streams; once the first byte is out the status is committed to 200\n * and failures can only be reported as an `error` event.\n */\nclass AiImageEnrichmentStreamRouteImpl implements HttpRouteHandler.Interface {\n    constructor(\n        private prepare: PrepareImageEnrichmentUseCase.Interface,\n        private ai: Ai.Interface\n    ) {}\n\n    async handle(\n        request: IHttpRequest,\n        response: IHttpResponseBuilder\n    ): Promise<IHttpResponseBuilder> {\n        const fileId = request.pathParameters.fileId;\n\n        if (!fileId) {\n            return response.status(400).json({ message: \"Missing file ID.\" });\n        }\n\n        const preparedResult = await this.prepare.execute(fileId);\n\n        if (preparedResult.isFail()) {\n            const error = preparedResult.error;\n\n            const statusCode = imageEnrichmentErrorStatusCode(error);\n\n            return response.status(statusCode).json({ message: error.message, code: error.code });\n        }\n\n        const events = this.enrich(preparedResult.value);\n\n        return response.sse(events);\n    }\n\n    private async *enrich(prepared: IPreparedImageEnrichment): AsyncGenerator<string> {\n        yield toSseFrame({ type: \"start\", fileId: prepared.fileId, model: prepared.model });\n\n        let output = { tags: [] as string[], description: \"\" };\n\n        try {\n            const request = buildEnrichmentAiRequest(prepared);\n            const stream = await this.ai.streamText(request);\n\n            for await (const partial of stream.partialOutputStream) {\n                output = readEnrichmentPartial(partial);\n                yield toSseFrame({ type: \"partial\", ...output });\n            }\n        } catch (error) {\n            yield toSseFrame({\n                type: \"error\",\n                message: `AI enrichment failed: ${\n                    error instanceof Error ? error.message : String(error)\n                }`\n            });\n            return;\n        }\n\n        // Nothing is persisted here. The route PROPOSES; the user accepts in the UI, and the save\n        // goes through the file's normal update path. The upload-time task still applies\n        // automatically — there is nobody to ask at that point.\n        yield toSseFrame({\n            type: \"done\",\n            fileId: prepared.fileId,\n            tags: output.tags,\n            description: output.description\n        });\n    }\n}\n\nexport const AiImageEnrichmentStreamRoute = HttpRouteHandler.createImplementation({\n    implementation: AiImageEnrichmentStreamRouteImpl,\n    dependencies: [PrepareImageEnrichmentUseCase, Ai]\n});\n\nclass AiImageEnrichmentStreamRouteDefinitionImpl implements HttpRouteDefinition.Interface {\n    readonly name = \"ai-image-enrichment-stream\";\n    readonly method = \"POST\";\n    readonly path = \"/stream/fm/files/:fileId/enrich\";\n    readonly handler = AiImageEnrichmentStreamRoute;\n}\n\n/** What the router matches on. Zero dependencies, so building it costs nothing. */\nexport const AiImageEnrichmentStreamRouteDefinition = HttpRouteDefinition.createImplementation({\n    implementation: AiImageEnrichmentStreamRouteDefinitionImpl,\n    dependencies: []\n});\n"],"names":["AiImageEnrichmentStreamRouteImpl","prepare","ai","request","response","fileId","preparedResult","error","statusCode","imageEnrichmentErrorStatusCode","events","prepared","toSseFrame","output","buildEnrichmentAiRequest","stream","partial","readEnrichmentPartial","Error","String","AiImageEnrichmentStreamRoute","HttpRouteHandler","PrepareImageEnrichmentUseCase","Ai","AiImageEnrichmentStreamRouteDefinitionImpl","AiImageEnrichmentStreamRouteDefinition","HttpRouteDefinition"],"mappings":";;;;;;AAkBA,MAAMA;IACF,YACYC,OAAgD,EAChDC,EAAgB,CAC1B;aAFUD,OAAO,GAAPA;aACAC,EAAE,GAAFA;IACT;IAEH,MAAM,OACFC,OAAqB,EACrBC,QAA8B,EACD;QAC7B,MAAMC,SAASF,QAAQ,cAAc,CAAC,MAAM;QAE5C,IAAI,CAACE,QACD,OAAOD,SAAS,MAAM,CAAC,KAAK,IAAI,CAAC;YAAE,SAAS;QAAmB;QAGnE,MAAME,iBAAiB,MAAM,IAAI,CAAC,OAAO,CAAC,OAAO,CAACD;QAElD,IAAIC,eAAe,MAAM,IAAI;YACzB,MAAMC,QAAQD,eAAe,KAAK;YAElC,MAAME,aAAaC,+BAA+BF;YAElD,OAAOH,SAAS,MAAM,CAACI,YAAY,IAAI,CAAC;gBAAE,SAASD,MAAM,OAAO;gBAAE,MAAMA,MAAM,IAAI;YAAC;QACvF;QAEA,MAAMG,SAAS,IAAI,CAAC,MAAM,CAACJ,eAAe,KAAK;QAE/C,OAAOF,SAAS,GAAG,CAACM;IACxB;IAEA,OAAe,OAAOC,QAAkC,EAA0B;QAC9E,MAAMC,WAAW;YAAE,MAAM;YAAS,QAAQD,SAAS,MAAM;YAAE,OAAOA,SAAS,KAAK;QAAC;QAEjF,IAAIE,SAAS;YAAE,MAAM,EAAE;YAAc,aAAa;QAAG;QAErD,IAAI;YACA,MAAMV,UAAUW,yBAAyBH;YACzC,MAAMI,SAAS,MAAM,IAAI,CAAC,EAAE,CAAC,UAAU,CAACZ;YAExC,WAAW,MAAMa,WAAWD,OAAO,mBAAmB,CAAE;gBACpDF,SAASI,sBAAsBD;gBAC/B,MAAMJ,WAAW;oBAAE,MAAM;oBAAW,GAAGC,MAAM;gBAAC;YAClD;QACJ,EAAE,OAAON,OAAO;YACZ,MAAMK,WAAW;gBACb,MAAM;gBACN,SAAS,CAAC,sBAAsB,EAC5BL,iBAAiBW,QAAQX,MAAM,OAAO,GAAGY,OAAOZ,QAClD;YACN;YACA;QACJ;QAKA,MAAMK,WAAW;YACb,MAAM;YACN,QAAQD,SAAS,MAAM;YACvB,MAAME,OAAO,IAAI;YACjB,aAAaA,OAAO,WAAW;QACnC;IACJ;AACJ;AAEO,MAAMO,+BAA+BC,iBAAiB,oBAAoB,CAAC;IAC9E,gBAAgBrB;IAChB,cAAc;QAACsB;QAA+BC;KAAG;AACrD;AAEA,MAAMC;;aACO,IAAI,GAAG;aACP,MAAM,GAAG;aACT,IAAI,GAAG;aACP,OAAO,GAAGJ;;AACvB;AAGO,MAAMK,yCAAyCC,oBAAoB,oBAAoB,CAAC;IAC3F,gBAAgBF;IAChB,cAAc,EAAE;AACpB"}

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

@@ -1,23 +1,19 @@
 import { TaskDefinition } from "@webiny/api-core/features/task/TaskDefinition/index.js";
 import { Ai } from "@webiny/api-core/features/ai/index.js";
-import { Encryption } from "@webiny/api-core/features/encryption/index.js";
-import { GetFileUseCase } from "@webiny/api-file-manager/features/file/GetFile/index.js";
-import { UpdateFileUseCase } from "@webiny/api-file-manager/features/file/UpdateFile/index.js";
-import { GetSettingsUseCase as FmGetSettingsUseCase } from "@webiny/api-file-manager/features/settings/GetSettings/abstractions.js";
-import { WebsocketService } from "@webiny/api-websockets/features/WebsocketService/index.js";
-import { GetSettingsUseCase } from "../../../api/features/GetSettings/index.js";
+import { ApplyImageEnrichmentUseCase, PrepareImageEnrichmentUseCase } from "./abstractions.js";
 export declare const AI_IMAGE_ENRICHMENT_TASK_ID = "fmAiImageEnrichment";
 export interface IAiImageEnrichmentTaskInput {
     fileId: string;
 }
+/**
+ * Background enrichment, triggered after a file is created. Shares its preparation and persistence
+ * with the streaming HTTP route (`AiImageEnrichmentStreamRoute`); the only difference is that this
+ * one waits for the whole AI response, because a background task has no one to stream to.
+ */
 declare class AiImageEnrichmentTaskImpl implements TaskDefinition.Interface<IAiImageEnrichmentTaskInput> {
-    private getFile;
-    private fmSettings;
-    private updateFile;
+    private prepare;
+    private apply;
     private ai;
-    private getSettings;
-    private encryption;
-    private websocketService?;
     id: string;
     title: string;
     description: string;
@@ -25,10 +21,10 @@ declare class AiImageEnrichmentTaskImpl implements TaskDefinition.Interface<IAiI
     isPrivate: boolean;
     databaseLogs: boolean;
     readonly selfCleanup: ("onAbort" | "onSuccess")[];
-    constructor(getFile: GetFileUseCase.Interface, fmSettings: FmGetSettingsUseCase.Interface, updateFile: UpdateFileUseCase.Interface, ai: Ai.Interface, getSettings: GetSettingsUseCase.Interface, encryption: Encryption.Interface, websocketService?: WebsocketService.Interface | undefined);
+    constructor(prepare: PrepareImageEnrichmentUseCase.Interface, apply: ApplyImageEnrichmentUseCase.Interface, ai: Ai.Interface);
     run({ input, controller }: TaskDefinition.RunParams<IAiImageEnrichmentTaskInput>): Promise<TaskDefinition.Result<IAiImageEnrichmentTaskInput>>;
 }
 export declare const AiImageEnrichmentTask: typeof AiImageEnrichmentTaskImpl & {
-    __abstraction: import("@webiny/di").Abstraction<import("@webiny/api-core/features/task/TaskDefinition/abstractions").ITaskDefinition<import("@webiny/api-core/features/task/TaskDefinition/abstractions").ITaskInput, import("@webiny/api-core/features/task/TaskDefinition/abstractions").ITaskOutput>>;
+    __abstraction: import("@webiny/di").Abstraction<import("@webiny/api-core/features/task/TaskDefinition/abstractions.js").ITaskDefinition<import("@webiny/api-core/features/task/TaskDefinition/abstractions.js").ITaskInput, import("@webiny/api-core/features/task/TaskDefinition/abstractions.js").ITaskOutput>>;
 };
 export {};

package/api/features/AiImageEnrichment/AiImageEnrichmentTask.js

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