@webiny/ai-powerups 6.4.12-beta.4 → 6.5.0-beta.0

package/api/features/GetSettings/GetSettingsRepository.js.map

@@ -1 +1 @@
-{"version":3,"file":"api/features/GetSettings/GetSettingsRepository.js","sources":["../../../../src/api/features/GetSettings/GetSettingsRepository.ts"],"sourcesContent":["import { Result } from \"@webiny/feature/api\";\nimport { KeyValueStore } from \"@webiny/api-core/features/keyValueStore/index.js\";\nimport {\n    AiPowerUpsSettingsGroupHandler,\n    AiPowerUpsSettingsCache\n} from \"~/api/features/shared/index.js\";\nimport { GetSettingsRepository } from \"./abstractions.js\";\nimport type { IAiPowerUpsSettings } from \"~/api/types.js\";\nimport { AI_POWER_UPS_SETTINGS } from \"~/api/constants.js\";\n\nclass GetSettingsRepositoryImpl implements GetSettingsRepository.Interface {\n    constructor(\n        private keyValueStore: KeyValueStore.Interface,\n        private handlers: AiPowerUpsSettingsGroupHandler.Interface[],\n        private cache: AiPowerUpsSettingsCache.Interface\n    ) {}\n\n    async get(): Promise<Result<IAiPowerUpsSettings>> {\n        const cached = this.cache.get();\n        if (cached) {\n            return Result.ok(cached.mapped);\n        }\n\n        const storeResult =\n            await this.keyValueStore.get<Record<string, unknown>>(AI_POWER_UPS_SETTINGS);\n\n        const raw: Record<string, unknown> =\n            storeResult.isOk() && storeResult.value ? storeResult.value : {};\n\n        const result: Record<string, unknown> = {};\n\n        for (const handler of this.handlers) {\n            result[handler.name] = handler.mapFromStorage(raw[handler.name]);\n        }\n\n        for (const key of Object.keys(raw)) {\n            if (!(key in result)) {\n                result[key] = raw[key];\n            }\n        }\n\n        const mapped = result as unknown as IAiPowerUpsSettings;\n        this.cache.set(raw, mapped);\n\n        return Result.ok(mapped);\n    }\n}\n\nexport const GetSettingsRepositoryImplementation = GetSettingsRepository.createImplementation({\n    implementation: GetSettingsRepositoryImpl,\n    dependencies: [\n        KeyValueStore,\n        [AiPowerUpsSettingsGroupHandler, { multiple: true }],\n        AiPowerUpsSettingsCache\n    ]\n});\n"],"names":["GetSettingsRepositoryImpl","keyValueStore","handlers","cache","cached","Result","storeResult","AI_POWER_UPS_SETTINGS","raw","result","handler","key","Object","mapped","GetSettingsRepositoryImplementation","GetSettingsRepository","KeyValueStore","AiPowerUpsSettingsGroupHandler","AiPowerUpsSettingsCache"],"mappings":";;;;;AAUA,MAAMA;IACF,YACYC,aAAsC,EACtCC,QAAoD,EACpDC,KAAwC,CAClD;aAHUF,aAAa,GAAbA;aACAC,QAAQ,GAARA;aACAC,KAAK,GAALA;IACT;IAEH,MAAM,MAA4C;QAC9C,MAAMC,SAAS,IAAI,CAAC,KAAK,CAAC,GAAG;QAC7B,IAAIA,QACA,OAAOC,OAAO,EAAE,CAACD,OAAO,MAAM;QAGlC,MAAME,cACF,MAAM,IAAI,CAAC,aAAa,CAAC,GAAG,CAA0BC;QAE1D,MAAMC,MACFF,YAAY,IAAI,MAAMA,YAAY,KAAK,GAAGA,YAAY,KAAK,GAAG,CAAC;QAEnE,MAAMG,SAAkC,CAAC;QAEzC,KAAK,MAAMC,WAAW,IAAI,CAAC,QAAQ,CAC/BD,MAAM,CAACC,QAAQ,IAAI,CAAC,GAAGA,QAAQ,cAAc,CAACF,GAAG,CAACE,QAAQ,IAAI,CAAC;QAGnE,KAAK,MAAMC,OAAOC,OAAO,IAAI,CAACJ,KAC1B,IAAI,CAAEG,CAAAA,OAAOF,MAAK,GACdA,MAAM,CAACE,IAAI,GAAGH,GAAG,CAACG,IAAI;QAI9B,MAAME,SAASJ;QACf,IAAI,CAAC,KAAK,CAAC,GAAG,CAACD,KAAKK;QAEpB,OAAOR,OAAO,EAAE,CAACQ;IACrB;AACJ;AAEO,MAAMC,sCAAsCC,sBAAsB,oBAAoB,CAAC;IAC1F,gBAAgBf;IAChB,cAAc;QACVgB;QACA;YAACC;YAAgC;gBAAE,UAAU;YAAK;SAAE;QACpDC;KACH;AACL"}
+{"version":3,"file":"api/features/GetSettings/GetSettingsRepository.js","sources":["../../../../src/api/features/GetSettings/GetSettingsRepository.ts"],"sourcesContent":["import { Result } from \"@webiny/feature/api\";\nimport { KeyValueStore } from \"@webiny/api-core/features/keyValueStore/index.js\";\nimport {\n    AiPowerUpsSettingsGroupHandler,\n    AiPowerUpsSettingsCache\n} from \"~/api/features/shared/index.js\";\nimport { GetSettingsRepository } from \"./abstractions.js\";\nimport type { IAiPowerUpsSettings } from \"~/api/types.js\";\nimport { AI_POWER_UPS_SETTINGS } from \"~/api/constants.js\";\n\nclass GetSettingsRepositoryImpl implements GetSettingsRepository.Interface {\n    constructor(\n        private keyValueStore: KeyValueStore.Interface,\n        private handlers: AiPowerUpsSettingsGroupHandler.Interface[],\n        private cache: AiPowerUpsSettingsCache.Interface\n    ) {}\n\n    async get(): Promise<Result<IAiPowerUpsSettings>> {\n        const cached = this.cache.get();\n        if (cached) {\n            return Result.ok(cached.mapped);\n        }\n\n        const storeResult =\n            await this.keyValueStore.get<Record<string, unknown>>(AI_POWER_UPS_SETTINGS);\n\n        const raw: Record<string, unknown> =\n            storeResult.isOk() && storeResult.value ? storeResult.value : {};\n\n        const result: Record<string, unknown> = {};\n\n        for (const handler of this.handlers) {\n            result[handler.name] = handler.mapFromStorage(raw[handler.name], raw);\n        }\n\n        for (const key of Object.keys(raw)) {\n            if (!(key in result)) {\n                result[key] = raw[key];\n            }\n        }\n\n        const mapped = result as unknown as IAiPowerUpsSettings;\n        this.cache.set(raw, mapped);\n\n        return Result.ok(mapped);\n    }\n}\n\nexport const GetSettingsRepositoryImplementation = GetSettingsRepository.createImplementation({\n    implementation: GetSettingsRepositoryImpl,\n    dependencies: [\n        KeyValueStore,\n        [AiPowerUpsSettingsGroupHandler, { multiple: true }],\n        AiPowerUpsSettingsCache\n    ]\n});\n"],"names":["GetSettingsRepositoryImpl","keyValueStore","handlers","cache","cached","Result","storeResult","AI_POWER_UPS_SETTINGS","raw","result","handler","key","Object","mapped","GetSettingsRepositoryImplementation","GetSettingsRepository","KeyValueStore","AiPowerUpsSettingsGroupHandler","AiPowerUpsSettingsCache"],"mappings":";;;;;AAUA,MAAMA;IACF,YACYC,aAAsC,EACtCC,QAAoD,EACpDC,KAAwC,CAClD;aAHUF,aAAa,GAAbA;aACAC,QAAQ,GAARA;aACAC,KAAK,GAALA;IACT;IAEH,MAAM,MAA4C;QAC9C,MAAMC,SAAS,IAAI,CAAC,KAAK,CAAC,GAAG;QAC7B,IAAIA,QACA,OAAOC,OAAO,EAAE,CAACD,OAAO,MAAM;QAGlC,MAAME,cACF,MAAM,IAAI,CAAC,aAAa,CAAC,GAAG,CAA0BC;QAE1D,MAAMC,MACFF,YAAY,IAAI,MAAMA,YAAY,KAAK,GAAGA,YAAY,KAAK,GAAG,CAAC;QAEnE,MAAMG,SAAkC,CAAC;QAEzC,KAAK,MAAMC,WAAW,IAAI,CAAC,QAAQ,CAC/BD,MAAM,CAACC,QAAQ,IAAI,CAAC,GAAGA,QAAQ,cAAc,CAACF,GAAG,CAACE,QAAQ,IAAI,CAAC,EAAEF;QAGrE,KAAK,MAAMG,OAAOC,OAAO,IAAI,CAACJ,KAC1B,IAAI,CAAEG,CAAAA,OAAOF,MAAK,GACdA,MAAM,CAACE,IAAI,GAAGH,GAAG,CAACG,IAAI;QAI9B,MAAME,SAASJ;QACf,IAAI,CAAC,KAAK,CAAC,GAAG,CAACD,KAAKK;QAEpB,OAAOR,OAAO,EAAE,CAACQ;IACrB;AACJ;AAEO,MAAMC,sCAAsCC,sBAAsB,oBAAoB,CAAC;IAC1F,gBAAgBf;IAChB,cAAc;QACVgB;QACA;YAACC;YAAgC;gBAAE,UAAU;YAAK;SAAE;QACpDC;KACH;AACL"}

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

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

package/api/features/GetSettings/index.js

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

package/api/features/ModelRoles/ModelRolesHandler.d.ts

@@ -0,0 +1,15 @@
+import { z } from "zod";
+import { AiPowerUpsSettingsGroupHandler } from "../../../api/features/shared/index.js";
+import type { AiModelRoleAssignments, ModelRolesSettings, PersistedModelRoles } from "./types.js";
+declare class ModelRolesHandlerImpl implements AiPowerUpsSettingsGroupHandler.Interface {
+    readonly name = "modelRoles";
+    readonly inputSchema: z.ZodObject<{
+        roles: z.ZodType<AiModelRoleAssignments>;
+    }, z.core.$strip>;
+    mapFromStorage(persisted: unknown, all?: Record<string, unknown>): ModelRolesSettings;
+    mapToStorage(internal: unknown): Promise<PersistedModelRoles>;
+}
+declare const _default: typeof ModelRolesHandlerImpl & {
+    __abstraction: import("@webiny/di").Abstraction<import("../shared/abstractions.js").IAiPowerUpsSettingsGroupHandler>;
+};
+export default _default;

package/api/features/ModelRoles/ModelRolesHandler.js

@@ -0,0 +1,68 @@
+import { z } from "zod";
+import { AiPowerUpsSettingsGroupHandler } from "../shared/index.js";
+import { readLegacyProviderPresets } from "../Connections/legacyProviders.js";
+import { AI_MODEL_ROLE_IDS } from "./roles.js";
+import { emptyAssignment } from "./types.js";
+const formString = z.string().nullish();
+const assignmentSchema = z.object({
+    connectionId: formString,
+    model: formString
+});
+const inputSchema = z.object({
+    roles: z.object(Object.fromEntries(AI_MODEL_ROLE_IDS.map((id)=>[
+            id,
+            assignmentSchema
+        ])))
+});
+class ModelRolesHandlerImpl {
+    mapFromStorage(persisted, all) {
+        const stored = persisted ?? {};
+        const roles = Object.fromEntries(AI_MODEL_ROLE_IDS.map((id)=>{
+            const assignment = stored.roles?.[id];
+            return [
+                id,
+                {
+                    connectionId: assignment?.connectionId ?? "",
+                    model: assignment?.model ?? ""
+                }
+            ];
+        }));
+        if (roles.standard.model) return {
+            roles
+        };
+        const legacy = readLegacyProviderPresets(all)[0];
+        if (legacy?.model) roles.standard = {
+            connectionId: legacy.id,
+            model: legacy.model
+        };
+        return {
+            roles
+        };
+    }
+    async mapToStorage(internal) {
+        const input = internal;
+        return {
+            roles: Object.fromEntries(AI_MODEL_ROLE_IDS.map((id)=>{
+                const assignment = input.roles?.[id] ?? emptyAssignment();
+                return [
+                    id,
+                    {
+                        connectionId: assignment.connectionId ?? "",
+                        model: assignment.model ?? ""
+                    }
+                ];
+            }))
+        };
+    }
+    constructor(){
+        this.name = "modelRoles";
+        this.inputSchema = inputSchema;
+    }
+}
+const ModelRolesHandler = AiPowerUpsSettingsGroupHandler.createImplementation({
+    implementation: ModelRolesHandlerImpl,
+    dependencies: []
+});
+export default ModelRolesHandler;
+
+//# sourceMappingURL=ModelRolesHandler.js.map

package/api/features/ModelRoles/ModelRolesHandler.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"api/features/ModelRoles/ModelRolesHandler.js","sources":["../../../../src/api/features/ModelRoles/ModelRolesHandler.ts"],"sourcesContent":["import { z } from \"zod\";\nimport { AiPowerUpsSettingsGroupHandler } from \"~/api/features/shared/index.js\";\nimport { readLegacyProviderPresets } from \"~/api/features/Connections/legacyProviders.js\";\nimport { AI_MODEL_ROLE_IDS } from \"./roles.js\";\nimport { emptyAssignment } from \"./types.js\";\nimport type { AiModelRoleAssignments, ModelRolesSettings, PersistedModelRoles } from \"./types.js\";\n\n/*\n * The admin form sends `null`, not `\"\"`, for a field nobody has touched, and an unfilled role is\n * the normal state of two of the three. `nullish` keeps an untouched role from failing validation;\n * `mapToStorage` is what normalises it to `\"\"`.\n */\nconst formString = z.string().nullish();\n\nconst assignmentSchema = z.object({\n    connectionId: formString,\n    model: formString\n});\n\nconst inputSchema = z.object({\n    roles: z.object(\n        Object.fromEntries(AI_MODEL_ROLE_IDS.map(id => [id, assignmentSchema]))\n    ) as unknown as z.ZodType<AiModelRoleAssignments>\n});\n\nclass ModelRolesHandlerImpl implements AiPowerUpsSettingsGroupHandler.Interface {\n    readonly name = \"modelRoles\";\n    readonly inputSchema = inputSchema;\n\n    mapFromStorage(persisted: unknown, all?: Record<string, unknown>): ModelRolesSettings {\n        const stored = (persisted ?? {}) as PersistedModelRoles;\n        const roles = Object.fromEntries(\n            AI_MODEL_ROLE_IDS.map(id => {\n                const assignment = stored.roles?.[id];\n                return [\n                    id,\n                    {\n                        connectionId: assignment?.connectionId ?? \"\",\n                        model: assignment?.model ?? \"\"\n                    }\n                ];\n            })\n        ) as AiModelRoleAssignments;\n\n        if (roles.standard.model) {\n            return { roles };\n        }\n\n        /*\n         * Nothing has filled `standard` yet, so seed it from the legacy `providers` section. Every\n         * AI feature used to read `providers.presets[0]` and nothing else, so lifting exactly that\n         * preset into `standard` reproduces the old behaviour for a project that upgrades without\n         * touching the settings screen.\n         *\n         * `fast` and `vision` stay empty on purpose. Both fall back to `standard` at resolve time,\n         * so an untouched project keeps working, and the settings screen shows the fallback rather\n         * than pretending someone chose it.\n         */\n        const legacy = readLegacyProviderPresets(all)[0];\n\n        if (legacy?.model) {\n            roles.standard = { connectionId: legacy.id, model: legacy.model };\n        }\n\n        return { roles };\n    }\n\n    async mapToStorage(internal: unknown): Promise<PersistedModelRoles> {\n        const input = internal as ModelRolesSettings;\n\n        return {\n            roles: Object.fromEntries(\n                AI_MODEL_ROLE_IDS.map(id => {\n                    const assignment = input.roles?.[id] ?? emptyAssignment();\n                    return [\n                        id,\n                        {\n                            connectionId: assignment.connectionId ?? \"\",\n                            model: assignment.model ?? \"\"\n                        }\n                    ];\n                })\n            )\n        };\n    }\n}\n\nexport default AiPowerUpsSettingsGroupHandler.createImplementation({\n    implementation: ModelRolesHandlerImpl,\n    dependencies: []\n});\n"],"names":["formString","z","assignmentSchema","inputSchema","Object","AI_MODEL_ROLE_IDS","id","ModelRolesHandlerImpl","persisted","all","stored","roles","assignment","legacy","readLegacyProviderPresets","internal","input","emptyAssignment","AiPowerUpsSettingsGroupHandler"],"mappings":";;;;;AAYA,MAAMA,aAAaC,EAAE,MAAM,GAAG,OAAO;AAErC,MAAMC,mBAAmBD,EAAE,MAAM,CAAC;IAC9B,cAAcD;IACd,OAAOA;AACX;AAEA,MAAMG,cAAcF,EAAE,MAAM,CAAC;IACzB,OAAOA,EAAE,MAAM,CACXG,OAAO,WAAW,CAACC,kBAAkB,GAAG,CAACC,CAAAA,KAAM;YAACA;YAAIJ;SAAiB;AAE7E;AAEA,MAAMK;IAIF,eAAeC,SAAkB,EAAEC,GAA6B,EAAsB;QAClF,MAAMC,SAAUF,aAAa,CAAC;QAC9B,MAAMG,QAAQP,OAAO,WAAW,CAC5BC,kBAAkB,GAAG,CAACC,CAAAA;YAClB,MAAMM,aAAaF,OAAO,KAAK,EAAE,CAACJ,GAAG;YACrC,OAAO;gBACHA;gBACA;oBACI,cAAcM,YAAY,gBAAgB;oBAC1C,OAAOA,YAAY,SAAS;gBAChC;aACH;QACL;QAGJ,IAAID,MAAM,QAAQ,CAAC,KAAK,EACpB,OAAO;YAAEA;QAAM;QAanB,MAAME,SAASC,0BAA0BL,IAAI,CAAC,EAAE;QAEhD,IAAII,QAAQ,OACRF,MAAM,QAAQ,GAAG;YAAE,cAAcE,OAAO,EAAE;YAAE,OAAOA,OAAO,KAAK;QAAC;QAGpE,OAAO;YAAEF;QAAM;IACnB;IAEA,MAAM,aAAaI,QAAiB,EAAgC;QAChE,MAAMC,QAAQD;QAEd,OAAO;YACH,OAAOX,OAAO,WAAW,CACrBC,kBAAkB,GAAG,CAACC,CAAAA;gBAClB,MAAMM,aAAaI,MAAM,KAAK,EAAE,CAACV,GAAG,IAAIW;gBACxC,OAAO;oBACHX;oBACA;wBACI,cAAcM,WAAW,YAAY,IAAI;wBACzC,OAAOA,WAAW,KAAK,IAAI;oBAC/B;iBACH;YACL;QAER;IACJ;;aA1DS,IAAI,GAAG;aACP,WAAW,GAAGT;;AA0D3B;AAEA,0BAAee,+BAA+B,oBAAoB,CAAC;IAC/D,gBAAgBX;IAChB,cAAc,EAAE;AACpB"}

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

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

package/api/features/ModelRoles/feature.js

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

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

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

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

@@ -0,0 +1,4 @@
+export { AI_MODEL_ROLE_IDS, isAiModelRoleId } from "./roles.js";
+export type { AiModelRoleId } from "./roles.js";
+export type { AiModelRoleAssignment, AiModelRoleAssignments, ModelRolesSettings } from "./types.js";
+export { ModelRolesFeature } from "./feature.js";

package/api/features/ModelRoles/index.js

@@ -0,0 +1,2 @@
+export { AI_MODEL_ROLE_IDS, isAiModelRoleId } from "./roles.js";
+export { ModelRolesFeature } from "./feature.js";

package/api/features/ModelRoles/roles.d.ts

@@ -0,0 +1,16 @@
+/**
+ * Model roles are the indirection between an AI feature and a concrete model.
+ *
+ * A feature never names a model. It names a role, and the project decides which model fills that
+ * role. That keeps the settings screen small (one select per role, not one per feature) while still
+ * letting a project run cheap work on a cheap model.
+ *
+ * The set is deliberately fixed and closed. A project that needs a different model for one specific
+ * feature overrides that feature under Capabilities instead of inventing a role.
+ *
+ * Labels and descriptions are not here on purpose: they are UI text, and they live next to the
+ * settings form in `~/admin/presentation/modelRoles.js`.
+ */
+export declare const AI_MODEL_ROLE_IDS: readonly ["fast", "standard", "vision"];
+export type AiModelRoleId = (typeof AI_MODEL_ROLE_IDS)[number];
+export declare const isAiModelRoleId: (value: unknown) => value is AiModelRoleId;

package/api/features/ModelRoles/roles.js

@@ -0,0 +1,9 @@
+const AI_MODEL_ROLE_IDS = [
+    "fast",
+    "standard",
+    "vision"
+];
+const isAiModelRoleId = (value)=>"string" == typeof value && AI_MODEL_ROLE_IDS.includes(value);
+export { AI_MODEL_ROLE_IDS, isAiModelRoleId };
+
+//# sourceMappingURL=roles.js.map

package/api/features/ModelRoles/roles.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"api/features/ModelRoles/roles.js","sources":["../../../../src/api/features/ModelRoles/roles.ts"],"sourcesContent":["/**\n * Model roles are the indirection between an AI feature and a concrete model.\n *\n * A feature never names a model. It names a role, and the project decides which model fills that\n * role. That keeps the settings screen small (one select per role, not one per feature) while still\n * letting a project run cheap work on a cheap model.\n *\n * The set is deliberately fixed and closed. A project that needs a different model for one specific\n * feature overrides that feature under Capabilities instead of inventing a role.\n *\n * Labels and descriptions are not here on purpose: they are UI text, and they live next to the\n * settings form in `~/admin/presentation/modelRoles.js`.\n */\nexport const AI_MODEL_ROLE_IDS = [\"fast\", \"standard\", \"vision\"] as const;\n\nexport type AiModelRoleId = (typeof AI_MODEL_ROLE_IDS)[number];\n\nexport const isAiModelRoleId = (value: unknown): value is AiModelRoleId =>\n    typeof value === \"string\" && (AI_MODEL_ROLE_IDS as readonly string[]).includes(value);\n"],"names":["AI_MODEL_ROLE_IDS","isAiModelRoleId","value"],"mappings":"AAaO,MAAMA,oBAAoB;IAAC;IAAQ;IAAY;CAAS;AAIxD,MAAMC,kBAAkB,CAACC,QAC5B,AAAiB,YAAjB,OAAOA,SAAuBF,kBAAwC,QAAQ,CAACE"}

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

@@ -0,0 +1,21 @@
+import type { IAiPowerUpsSettings } from "../../../api/types.js";
+import type { AiModelRoleId } from "./roles.js";
+export interface AiModelRoleAssignment {
+    /** Empty string means the role is unfilled. */
+    connectionId: string;
+    /** Fully qualified, e.g. `"anthropic/claude-sonnet-4-5"`. Empty string means unfilled. */
+    model: string;
+}
+export type AiModelRoleAssignments = Record<AiModelRoleId, AiModelRoleAssignment>;
+declare module "~/api/types.js" {
+    interface IAiPowerUpsSettings {
+        modelRoles: {
+            roles: AiModelRoleAssignments;
+        };
+    }
+}
+export type ModelRolesSettings = IAiPowerUpsSettings["modelRoles"];
+export interface PersistedModelRoles {
+    roles?: Partial<Record<string, Partial<AiModelRoleAssignment>>>;
+}
+export declare const emptyAssignment: () => AiModelRoleAssignment;

package/api/features/ModelRoles/types.js

@@ -0,0 +1,7 @@
+const emptyAssignment = ()=>({
+        connectionId: "",
+        model: ""
+    });
+export { emptyAssignment };
+
+//# sourceMappingURL=types.js.map

package/api/features/ModelRoles/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"api/features/ModelRoles/types.js","sources":["../../../../src/api/features/ModelRoles/types.ts"],"sourcesContent":["import type { IAiPowerUpsSettings } from \"~/api/types.js\";\nimport type { AiModelRoleId } from \"./roles.js\";\n\nexport interface AiModelRoleAssignment {\n    /** Empty string means the role is unfilled. */\n    connectionId: string;\n    /** Fully qualified, e.g. `\"anthropic/claude-sonnet-4-5\"`. Empty string means unfilled. */\n    model: string;\n}\n\nexport type AiModelRoleAssignments = Record<AiModelRoleId, AiModelRoleAssignment>;\n\ndeclare module \"~/api/types.js\" {\n    interface IAiPowerUpsSettings {\n        modelRoles: {\n            roles: AiModelRoleAssignments;\n        };\n    }\n}\n\nexport type ModelRolesSettings = IAiPowerUpsSettings[\"modelRoles\"];\n\nexport interface PersistedModelRoles {\n    roles?: Partial<Record<string, Partial<AiModelRoleAssignment>>>;\n}\n\nexport const emptyAssignment = (): AiModelRoleAssignment => ({ connectionId: \"\", model: \"\" });\n"],"names":["emptyAssignment"],"mappings":"AA0BO,MAAMA,kBAAkB,IAA8B;QAAE,cAAc;QAAI,OAAO;IAAG"}

package/api/features/Providers/ProvidersHandler.js.map

@@ -1 +1 @@
-{"version":3,"file":"api/features/Providers/ProvidersHandler.js","sources":["../../../../src/api/features/Providers/ProvidersHandler.ts"],"sourcesContent":["import { z } from \"zod\";\nimport { Encryption } from \"@webiny/api-core/features/encryption/index.js\";\nimport { Masker } from \"@webiny/api-core/features/masker/index.js\";\nimport { AiModelRegistry } from \"@webiny/api-core/features/ai/index.js\";\nimport { AiPowerUpsSettingsGroupHandler } from \"~/api/features/shared/index.js\";\nimport type { PersistedProviderPreset, PersistedProviders, ProvidersSettings } from \"./types.js\";\n\nconst inputSchema = z.object({\n    presets: z.array(\n        z.object({\n            id: z.string().min(1),\n            name: z.string().min(1),\n            description: z.string().optional(),\n            model: z.string().min(1),\n            apiKey: z.string().optional()\n        })\n    )\n});\n\nclass ProvidersHandlerImpl implements AiPowerUpsSettingsGroupHandler.Interface {\n    readonly name = \"providers\";\n    readonly inputSchema = inputSchema;\n\n    constructor(\n        private encryption: Encryption.Interface,\n        private masker: Masker.Interface,\n        private modelRegistry: AiModelRegistry.Interface\n    ) {}\n\n    mapFromStorage(persisted: unknown): ProvidersSettings {\n        if (!persisted || typeof persisted !== \"object\") {\n            return { presets: [] };\n        }\n\n        const data = persisted as PersistedProviders;\n        const presets = (data.presets ?? []).map(p => ({\n            id: p.id,\n            name: p.name,\n            description: p.description,\n            model: p.model,\n            apiKeyMasked: p.apiKeyMasked ?? \"\",\n            apiKeyEncrypted: p.apiKeyEncrypted ?? \"\"\n        }));\n\n        return { presets };\n    }\n\n    async mapToStorage(internal: unknown, existing: unknown | null): Promise<PersistedProviders> {\n        const input = internal as ProvidersSettings;\n        const existingData = existing as ProvidersSettings | null;\n        const existingPresets = existingData?.presets ?? [];\n\n        // Validate that all preset models are in the registry.\n        const availableModels = await this.modelRegistry.listModels();\n        for (const preset of input.presets) {\n            if (!preset.model) {\n                continue;\n            }\n            const slashIndex = preset.model.indexOf(\"/\");\n            if (slashIndex === -1) {\n                throw new Error(\n                    `Invalid model ID \"${preset.model}\" in preset \"${preset.name}\". Expected format: \"<providerId>/<modelId>\".`\n                );\n            }\n            const providerId = preset.model.slice(0, slashIndex);\n            const modelId = preset.model.slice(slashIndex + 1);\n            const found = availableModels.some(\n                m => m.providerId === providerId && m.modelId === modelId\n            );\n            if (!found) {\n                throw new Error(\n                    `Model \"${preset.model}\" in preset \"${preset.name}\" is not available. Use listModels() to see available models.`\n                );\n            }\n        }\n\n        const presets: PersistedProviderPreset[] = await Promise.all(\n            input.presets.map(async preset => {\n                const existingMatch = existingPresets.find(ep => ep.id === preset.id);\n\n                let apiKeyEncrypted: string;\n                let apiKeyMasked: string;\n\n                if (!preset.apiKey || preset.apiKey === existingMatch?.apiKeyMasked) {\n                    // Carry forward existing encrypted key\n                    apiKeyEncrypted = existingMatch?.apiKeyEncrypted ?? \"\";\n                    apiKeyMasked = existingMatch?.apiKeyMasked ?? \"\";\n                } else {\n                    // New plaintext key — encrypt and mask\n                    apiKeyEncrypted = await this.encryption.encrypt(preset.apiKey);\n                    apiKeyMasked = this.masker.mask(preset.apiKey, [8, 4]);\n                }\n\n                return {\n                    id: preset.id,\n                    name: preset.name,\n                    description: preset.description,\n                    model: preset.model,\n                    apiKeyEncrypted,\n                    apiKeyMasked\n                };\n            })\n        );\n\n        return { presets };\n    }\n}\n\nexport default AiPowerUpsSettingsGroupHandler.createImplementation({\n    implementation: ProvidersHandlerImpl,\n    dependencies: [Encryption, Masker, AiModelRegistry]\n});\n"],"names":["inputSchema","z","ProvidersHandlerImpl","encryption","masker","modelRegistry","persisted","data","presets","p","internal","existing","input","existingData","existingPresets","availableModels","preset","slashIndex","Error","providerId","modelId","found","m","Promise","existingMatch","ep","apiKeyEncrypted","apiKeyMasked","AiPowerUpsSettingsGroupHandler","Encryption","Masker","AiModelRegistry"],"mappings":";;;;;AAOA,MAAMA,cAAcC,EAAE,MAAM,CAAC;IACzB,SAASA,EAAE,KAAK,CACZA,EAAE,MAAM,CAAC;QACL,IAAIA,EAAE,MAAM,GAAG,GAAG,CAAC;QACnB,MAAMA,EAAE,MAAM,GAAG,GAAG,CAAC;QACrB,aAAaA,EAAE,MAAM,GAAG,QAAQ;QAChC,OAAOA,EAAE,MAAM,GAAG,GAAG,CAAC;QACtB,QAAQA,EAAE,MAAM,GAAG,QAAQ;IAC/B;AAER;AAEA,MAAMC;IAIF,YACYC,UAAgC,EAChCC,MAAwB,EACxBC,aAAwC,CAClD;aAHUF,UAAU,GAAVA;aACAC,MAAM,GAANA;aACAC,aAAa,GAAbA;aANH,IAAI,GAAG;aACP,WAAW,GAAGL;IAMpB;IAEH,eAAeM,SAAkB,EAAqB;QAClD,IAAI,CAACA,aAAa,AAAqB,YAArB,OAAOA,WACrB,OAAO;YAAE,SAAS,EAAE;QAAC;QAGzB,MAAMC,OAAOD;QACb,MAAME,UAAWD,AAAAA,CAAAA,KAAK,OAAO,IAAI,EAAC,EAAG,GAAG,CAACE,CAAAA,IAAM;gBAC3C,IAAIA,EAAE,EAAE;gBACR,MAAMA,EAAE,IAAI;gBACZ,aAAaA,EAAE,WAAW;gBAC1B,OAAOA,EAAE,KAAK;gBACd,cAAcA,EAAE,YAAY,IAAI;gBAChC,iBAAiBA,EAAE,eAAe,IAAI;YAC1C;QAEA,OAAO;YAAED;QAAQ;IACrB;IAEA,MAAM,aAAaE,QAAiB,EAAEC,QAAwB,EAA+B;QACzF,MAAMC,QAAQF;QACd,MAAMG,eAAeF;QACrB,MAAMG,kBAAkBD,cAAc,WAAW,EAAE;QAGnD,MAAME,kBAAkB,MAAM,IAAI,CAAC,aAAa,CAAC,UAAU;QAC3D,KAAK,MAAMC,UAAUJ,MAAM,OAAO,CAAE;YAChC,IAAI,CAACI,OAAO,KAAK,EACb;YAEJ,MAAMC,aAAaD,OAAO,KAAK,CAAC,OAAO,CAAC;YACxC,IAAIC,AAAe,OAAfA,YACA,MAAM,IAAIC,MACN,CAAC,kBAAkB,EAAEF,OAAO,KAAK,CAAC,aAAa,EAAEA,OAAO,IAAI,CAAC,6CAA6C,CAAC;YAGnH,MAAMG,aAAaH,OAAO,KAAK,CAAC,KAAK,CAAC,GAAGC;YACzC,MAAMG,UAAUJ,OAAO,KAAK,CAAC,KAAK,CAACC,aAAa;YAChD,MAAMI,QAAQN,gBAAgB,IAAI,CAC9BO,CAAAA,IAAKA,EAAE,UAAU,KAAKH,cAAcG,EAAE,OAAO,KAAKF;YAEtD,IAAI,CAACC,OACD,MAAM,IAAIH,MACN,CAAC,OAAO,EAAEF,OAAO,KAAK,CAAC,aAAa,EAAEA,OAAO,IAAI,CAAC,6DAA6D,CAAC;QAG5H;QAEA,MAAMR,UAAqC,MAAMe,QAAQ,GAAG,CACxDX,MAAM,OAAO,CAAC,GAAG,CAAC,OAAMI;YACpB,MAAMQ,gBAAgBV,gBAAgB,IAAI,CAACW,CAAAA,KAAMA,GAAG,EAAE,KAAKT,OAAO,EAAE;YAEpE,IAAIU;YACJ,IAAIC;YAEJ,IAAI,AAACX,OAAO,MAAM,IAAIA,OAAO,MAAM,KAAKQ,eAAe,cAIhD;gBAEHE,kBAAkB,MAAM,IAAI,CAAC,UAAU,CAAC,OAAO,CAACV,OAAO,MAAM;gBAC7DW,eAAe,IAAI,CAAC,MAAM,CAAC,IAAI,CAACX,OAAO,MAAM,EAAE;oBAAC;oBAAG;iBAAE;YACzD,OARqE;gBAEjEU,kBAAkBF,eAAe,mBAAmB;gBACpDG,eAAeH,eAAe,gBAAgB;YAClD;YAMA,OAAO;gBACH,IAAIR,OAAO,EAAE;gBACb,MAAMA,OAAO,IAAI;gBACjB,aAAaA,OAAO,WAAW;gBAC/B,OAAOA,OAAO,KAAK;gBACnBU;gBACAC;YACJ;QACJ;QAGJ,OAAO;YAAEnB;QAAQ;IACrB;AACJ;AAEA,yBAAeoB,+BAA+B,oBAAoB,CAAC;IAC/D,gBAAgB1B;IAChB,cAAc;QAAC2B;QAAYC;QAAQC;KAAgB;AACvD"}
+{"version":3,"file":"api/features/Providers/ProvidersHandler.js","sources":["../../../../src/api/features/Providers/ProvidersHandler.ts"],"sourcesContent":["import { z } from \"zod\";\nimport { Encryption } from \"@webiny/api-core/features/encryption/index.js\";\nimport { Masker } from \"@webiny/api-core/features/masker/index.js\";\nimport { AiModelRegistry } from \"@webiny/api-core/features/ai/index.js\";\nimport { AiPowerUpsSettingsGroupHandler } from \"~/api/features/shared/index.js\";\nimport type { PersistedProviderPreset, PersistedProviders, ProvidersSettings } from \"./types.js\";\n\nconst inputSchema = z.object({\n    presets: z.array(\n        z.object({\n            id: z.string().min(1),\n            name: z.string().min(1),\n            description: z.string().optional(),\n            model: z.string().min(1),\n            apiKey: z.string().optional()\n        })\n    )\n});\n\nclass ProvidersHandlerImpl implements AiPowerUpsSettingsGroupHandler.Interface {\n    readonly name = \"providers\";\n    readonly inputSchema = inputSchema;\n\n    constructor(\n        private encryption: Encryption.Interface,\n        private masker: Masker.Interface,\n        private modelRegistry: AiModelRegistry.Interface\n    ) {}\n\n    mapFromStorage(persisted: unknown): ProvidersSettings {\n        if (!persisted || typeof persisted !== \"object\") {\n            return { presets: [] };\n        }\n\n        const data = persisted as PersistedProviders;\n        const presets = (data.presets ?? []).map(p => ({\n            id: p.id,\n            name: p.name,\n            description: p.description,\n            model: p.model,\n            apiKeyMasked: p.apiKeyMasked ?? \"\",\n            apiKeyEncrypted: p.apiKeyEncrypted ?? \"\"\n        }));\n\n        return { presets };\n    }\n\n    async mapToStorage(internal: unknown, existing: unknown | null): Promise<PersistedProviders> {\n        const input = internal as ProvidersSettings;\n        const existingData = existing as ProvidersSettings | null;\n        const existingPresets = existingData?.presets ?? [];\n\n        const availableModels = await this.modelRegistry.listModels();\n        for (const preset of input.presets) {\n            if (!preset.model) {\n                continue;\n            }\n            const slashIndex = preset.model.indexOf(\"/\");\n            if (slashIndex === -1) {\n                throw new Error(\n                    `Invalid model ID \"${preset.model}\" in preset \"${preset.name}\". Expected format: \"<providerId>/<modelId>\".`\n                );\n            }\n            const providerId = preset.model.slice(0, slashIndex);\n            const modelId = preset.model.slice(slashIndex + 1);\n            const found = availableModels.some(\n                m => m.providerId === providerId && m.modelId === modelId\n            );\n            if (!found) {\n                throw new Error(\n                    `Model \"${preset.model}\" in preset \"${preset.name}\" is not available. Use listModels() to see available models.`\n                );\n            }\n        }\n\n        const presets: PersistedProviderPreset[] = await Promise.all(\n            input.presets.map(async preset => {\n                const existingMatch = existingPresets.find(ep => ep.id === preset.id);\n\n                let apiKeyEncrypted: string;\n                let apiKeyMasked: string;\n\n                if (!preset.apiKey || preset.apiKey === existingMatch?.apiKeyMasked) {\n                    // Carry forward existing encrypted key\n                    apiKeyEncrypted = existingMatch?.apiKeyEncrypted ?? \"\";\n                    apiKeyMasked = existingMatch?.apiKeyMasked ?? \"\";\n                } else {\n                    // New plaintext key — encrypt and mask\n                    apiKeyEncrypted = await this.encryption.encrypt(preset.apiKey);\n                    apiKeyMasked = this.masker.mask(preset.apiKey, [8, 4]);\n                }\n\n                return {\n                    id: preset.id,\n                    name: preset.name,\n                    description: preset.description,\n                    model: preset.model,\n                    apiKeyEncrypted,\n                    apiKeyMasked\n                };\n            })\n        );\n\n        return { presets };\n    }\n}\n\nexport default AiPowerUpsSettingsGroupHandler.createImplementation({\n    implementation: ProvidersHandlerImpl,\n    dependencies: [Encryption, Masker, AiModelRegistry]\n});\n"],"names":["inputSchema","z","ProvidersHandlerImpl","encryption","masker","modelRegistry","persisted","data","presets","p","internal","existing","input","existingData","existingPresets","availableModels","preset","slashIndex","Error","providerId","modelId","found","m","Promise","existingMatch","ep","apiKeyEncrypted","apiKeyMasked","AiPowerUpsSettingsGroupHandler","Encryption","Masker","AiModelRegistry"],"mappings":";;;;;AAOA,MAAMA,cAAcC,EAAE,MAAM,CAAC;IACzB,SAASA,EAAE,KAAK,CACZA,EAAE,MAAM,CAAC;QACL,IAAIA,EAAE,MAAM,GAAG,GAAG,CAAC;QACnB,MAAMA,EAAE,MAAM,GAAG,GAAG,CAAC;QACrB,aAAaA,EAAE,MAAM,GAAG,QAAQ;QAChC,OAAOA,EAAE,MAAM,GAAG,GAAG,CAAC;QACtB,QAAQA,EAAE,MAAM,GAAG,QAAQ;IAC/B;AAER;AAEA,MAAMC;IAIF,YACYC,UAAgC,EAChCC,MAAwB,EACxBC,aAAwC,CAClD;aAHUF,UAAU,GAAVA;aACAC,MAAM,GAANA;aACAC,aAAa,GAAbA;aANH,IAAI,GAAG;aACP,WAAW,GAAGL;IAMpB;IAEH,eAAeM,SAAkB,EAAqB;QAClD,IAAI,CAACA,aAAa,AAAqB,YAArB,OAAOA,WACrB,OAAO;YAAE,SAAS,EAAE;QAAC;QAGzB,MAAMC,OAAOD;QACb,MAAME,UAAWD,AAAAA,CAAAA,KAAK,OAAO,IAAI,EAAC,EAAG,GAAG,CAACE,CAAAA,IAAM;gBAC3C,IAAIA,EAAE,EAAE;gBACR,MAAMA,EAAE,IAAI;gBACZ,aAAaA,EAAE,WAAW;gBAC1B,OAAOA,EAAE,KAAK;gBACd,cAAcA,EAAE,YAAY,IAAI;gBAChC,iBAAiBA,EAAE,eAAe,IAAI;YAC1C;QAEA,OAAO;YAAED;QAAQ;IACrB;IAEA,MAAM,aAAaE,QAAiB,EAAEC,QAAwB,EAA+B;QACzF,MAAMC,QAAQF;QACd,MAAMG,eAAeF;QACrB,MAAMG,kBAAkBD,cAAc,WAAW,EAAE;QAEnD,MAAME,kBAAkB,MAAM,IAAI,CAAC,aAAa,CAAC,UAAU;QAC3D,KAAK,MAAMC,UAAUJ,MAAM,OAAO,CAAE;YAChC,IAAI,CAACI,OAAO,KAAK,EACb;YAEJ,MAAMC,aAAaD,OAAO,KAAK,CAAC,OAAO,CAAC;YACxC,IAAIC,AAAe,OAAfA,YACA,MAAM,IAAIC,MACN,CAAC,kBAAkB,EAAEF,OAAO,KAAK,CAAC,aAAa,EAAEA,OAAO,IAAI,CAAC,6CAA6C,CAAC;YAGnH,MAAMG,aAAaH,OAAO,KAAK,CAAC,KAAK,CAAC,GAAGC;YACzC,MAAMG,UAAUJ,OAAO,KAAK,CAAC,KAAK,CAACC,aAAa;YAChD,MAAMI,QAAQN,gBAAgB,IAAI,CAC9BO,CAAAA,IAAKA,EAAE,UAAU,KAAKH,cAAcG,EAAE,OAAO,KAAKF;YAEtD,IAAI,CAACC,OACD,MAAM,IAAIH,MACN,CAAC,OAAO,EAAEF,OAAO,KAAK,CAAC,aAAa,EAAEA,OAAO,IAAI,CAAC,6DAA6D,CAAC;QAG5H;QAEA,MAAMR,UAAqC,MAAMe,QAAQ,GAAG,CACxDX,MAAM,OAAO,CAAC,GAAG,CAAC,OAAMI;YACpB,MAAMQ,gBAAgBV,gBAAgB,IAAI,CAACW,CAAAA,KAAMA,GAAG,EAAE,KAAKT,OAAO,EAAE;YAEpE,IAAIU;YACJ,IAAIC;YAEJ,IAAI,AAACX,OAAO,MAAM,IAAIA,OAAO,MAAM,KAAKQ,eAAe,cAIhD;gBAEHE,kBAAkB,MAAM,IAAI,CAAC,UAAU,CAAC,OAAO,CAACV,OAAO,MAAM;gBAC7DW,eAAe,IAAI,CAAC,MAAM,CAAC,IAAI,CAACX,OAAO,MAAM,EAAE;oBAAC;oBAAG;iBAAE;YACzD,OARqE;gBAEjEU,kBAAkBF,eAAe,mBAAmB;gBACpDG,eAAeH,eAAe,gBAAgB;YAClD;YAMA,OAAO;gBACH,IAAIR,OAAO,EAAE;gBACb,MAAMA,OAAO,IAAI;gBACjB,aAAaA,OAAO,WAAW;gBAC/B,OAAOA,OAAO,KAAK;gBACnBU;gBACAC;YACJ;QACJ;QAGJ,OAAO;YAAEnB;QAAQ;IACrB;AACJ;AAEA,yBAAeoB,+BAA+B,oBAAoB,CAAC;IAC/D,gBAAgB1B;IAChB,cAAc;QAAC2B;QAAYC;QAAQC;KAAgB;AACvD"}

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

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

package/api/features/UpdateSettings/UpdateSettingsRepository.js

@@ -19,7 +19,7 @@ class UpdateSettingsRepositoryImpl {
             raw = storeResult.isOk() && storeResult.value ? storeResult.value : {};
         }
         const existingInternal = {};
-        for (const handler of this.handlers)existingInternal[handler.name] = handler.mapFromStorage(raw[handler.name]);
+        for (const handler of this.handlers)existingInternal[handler.name] = handler.mapFromStorage(raw[handler.name], raw);
         const newSettings = input;
         const invalidFields = {};
         for (const handler of this.handlers){
@@ -46,7 +46,7 @@ class UpdateSettingsRepositoryImpl {
         const writeResult = await this.keyValueStore.set(AI_POWER_UPS_SETTINGS, persisted);
         if (writeResult.isFail()) return Result.fail(new Error(String(writeResult.error)));
         const result = {};
-        for (const handler of this.handlers)result[handler.name] = handler.mapFromStorage(persisted[handler.name]);
+        for (const handler of this.handlers)result[handler.name] = handler.mapFromStorage(persisted[handler.name], persisted);
         for (const key of Object.keys(persisted))if (!(key in result)) result[key] = persisted[key];
         const mapped = result;
         this.cache.set(persisted, mapped);

package/api/features/UpdateSettings/UpdateSettingsRepository.js.map

@@ -1 +1 @@
-{"version":3,"file":"api/features/UpdateSettings/UpdateSettingsRepository.js","sources":["../../../../src/api/features/UpdateSettings/UpdateSettingsRepository.ts"],"sourcesContent":["import { Result } from \"@webiny/feature/api\";\nimport { KeyValueStore } from \"@webiny/api-core/features/keyValueStore/index.js\";\nimport type { OutputErrors } from \"@webiny/utils/createZodError.js\";\nimport {\n    AiPowerUpsSettingsGroupHandler,\n    AiPowerUpsSettingsCache\n} from \"~/api/features/shared/index.js\";\nimport { UpdateSettingsRepository } from \"./abstractions.js\";\nimport { SettingsValidationError } from \"./errors.js\";\nimport type { IAiPowerUpsSettings } from \"~/api/types.js\";\nimport { AI_POWER_UPS_SETTINGS } from \"~/api/constants.js\";\n\nclass UpdateSettingsRepositoryImpl implements UpdateSettingsRepository.Interface {\n    constructor(\n        private keyValueStore: KeyValueStore.Interface,\n        private handlers: AiPowerUpsSettingsGroupHandler.Interface[],\n        private cache: AiPowerUpsSettingsCache.Interface\n    ) {}\n\n    async execute(input: IAiPowerUpsSettings): Promise<Result<IAiPowerUpsSettings, Error>> {\n        const cached = this.cache.get();\n        let raw: Record<string, unknown>;\n\n        if (cached) {\n            raw = cached.raw;\n        } else {\n            const storeResult =\n                await this.keyValueStore.get<Record<string, unknown>>(AI_POWER_UPS_SETTINGS);\n            raw = storeResult.isOk() && storeResult.value ? storeResult.value : {};\n        }\n\n        const existingInternal: Record<string, unknown> = {};\n        for (const handler of this.handlers) {\n            existingInternal[handler.name] = handler.mapFromStorage(raw[handler.name]);\n        }\n\n        const newSettings = input as unknown as Record<string, unknown>;\n        const invalidFields: OutputErrors = {};\n\n        for (const handler of this.handlers) {\n            const parseResult = await handler.inputSchema.safeParseAsync(newSettings[handler.name]);\n            if (!parseResult.success) {\n                for (const issue of parseResult.error.issues) {\n                    const fieldPath = [handler.name, ...issue.path.map(String)].join(\".\");\n                    invalidFields[fieldPath] = {\n                        code: issue.code,\n                        message: issue.message,\n                        data: { path: issue.path }\n                    };\n                }\n            }\n        }\n\n        if (Object.keys(invalidFields).length > 0) {\n            return Result.fail(new SettingsValidationError(invalidFields));\n        }\n\n        const persisted: Record<string, unknown> = { ...raw };\n        for (const handler of this.handlers) {\n            persisted[handler.name] = await handler.mapToStorage(\n                newSettings[handler.name],\n                existingInternal[handler.name]\n            );\n        }\n\n        const writeResult = await this.keyValueStore.set(AI_POWER_UPS_SETTINGS, persisted);\n        if (writeResult.isFail()) {\n            return Result.fail(new Error(String(writeResult.error)));\n        }\n\n        const result: Record<string, unknown> = {};\n        for (const handler of this.handlers) {\n            result[handler.name] = handler.mapFromStorage(persisted[handler.name]);\n        }\n\n        for (const key of Object.keys(persisted)) {\n            if (!(key in result)) {\n                result[key] = persisted[key];\n            }\n        }\n\n        const mapped = result as unknown as IAiPowerUpsSettings;\n        this.cache.set(persisted, mapped);\n\n        return Result.ok(mapped);\n    }\n}\n\nexport const UpdateSettingsRepositoryImplementation = UpdateSettingsRepository.createImplementation(\n    {\n        implementation: UpdateSettingsRepositoryImpl,\n        dependencies: [\n            KeyValueStore,\n            [AiPowerUpsSettingsGroupHandler, { multiple: true }],\n            AiPowerUpsSettingsCache\n        ]\n    }\n);\n"],"names":["UpdateSettingsRepositoryImpl","keyValueStore","handlers","cache","input","cached","raw","storeResult","AI_POWER_UPS_SETTINGS","existingInternal","handler","newSettings","invalidFields","parseResult","issue","fieldPath","String","Object","Result","SettingsValidationError","persisted","writeResult","Error","result","key","mapped","UpdateSettingsRepositoryImplementation","UpdateSettingsRepository","KeyValueStore","AiPowerUpsSettingsGroupHandler","AiPowerUpsSettingsCache"],"mappings":";;;;;;AAYA,MAAMA;IACF,YACYC,aAAsC,EACtCC,QAAoD,EACpDC,KAAwC,CAClD;aAHUF,aAAa,GAAbA;aACAC,QAAQ,GAARA;aACAC,KAAK,GAALA;IACT;IAEH,MAAM,QAAQC,KAA0B,EAA+C;QACnF,MAAMC,SAAS,IAAI,CAAC,KAAK,CAAC,GAAG;QAC7B,IAAIC;QAEJ,IAAID,QACAC,MAAMD,OAAO,GAAG;aACb;YACH,MAAME,cACF,MAAM,IAAI,CAAC,aAAa,CAAC,GAAG,CAA0BC;YAC1DF,MAAMC,YAAY,IAAI,MAAMA,YAAY,KAAK,GAAGA,YAAY,KAAK,GAAG,CAAC;QACzE;QAEA,MAAME,mBAA4C,CAAC;QACnD,KAAK,MAAMC,WAAW,IAAI,CAAC,QAAQ,CAC/BD,gBAAgB,CAACC,QAAQ,IAAI,CAAC,GAAGA,QAAQ,cAAc,CAACJ,GAAG,CAACI,QAAQ,IAAI,CAAC;QAG7E,MAAMC,cAAcP;QACpB,MAAMQ,gBAA8B,CAAC;QAErC,KAAK,MAAMF,WAAW,IAAI,CAAC,QAAQ,CAAE;YACjC,MAAMG,cAAc,MAAMH,QAAQ,WAAW,CAAC,cAAc,CAACC,WAAW,CAACD,QAAQ,IAAI,CAAC;YACtF,IAAI,CAACG,YAAY,OAAO,EACpB,KAAK,MAAMC,SAASD,YAAY,KAAK,CAAC,MAAM,CAAE;gBAC1C,MAAME,YAAY;oBAACL,QAAQ,IAAI;uBAAKI,MAAM,IAAI,CAAC,GAAG,CAACE;iBAAQ,CAAC,IAAI,CAAC;gBACjEJ,aAAa,CAACG,UAAU,GAAG;oBACvB,MAAMD,MAAM,IAAI;oBAChB,SAASA,MAAM,OAAO;oBACtB,MAAM;wBAAE,MAAMA,MAAM,IAAI;oBAAC;gBAC7B;YACJ;QAER;QAEA,IAAIG,OAAO,IAAI,CAACL,eAAe,MAAM,GAAG,GACpC,OAAOM,OAAO,IAAI,CAAC,IAAIC,wBAAwBP;QAGnD,MAAMQ,YAAqC;YAAE,GAAGd,GAAG;QAAC;QACpD,KAAK,MAAMI,WAAW,IAAI,CAAC,QAAQ,CAC/BU,SAAS,CAACV,QAAQ,IAAI,CAAC,GAAG,MAAMA,QAAQ,YAAY,CAChDC,WAAW,CAACD,QAAQ,IAAI,CAAC,EACzBD,gBAAgB,CAACC,QAAQ,IAAI,CAAC;QAItC,MAAMW,cAAc,MAAM,IAAI,CAAC,aAAa,CAAC,GAAG,CAACb,uBAAuBY;QACxE,IAAIC,YAAY,MAAM,IAClB,OAAOH,OAAO,IAAI,CAAC,IAAII,MAAMN,OAAOK,YAAY,KAAK;QAGzD,MAAME,SAAkC,CAAC;QACzC,KAAK,MAAMb,WAAW,IAAI,CAAC,QAAQ,CAC/Ba,MAAM,CAACb,QAAQ,IAAI,CAAC,GAAGA,QAAQ,cAAc,CAACU,SAAS,CAACV,QAAQ,IAAI,CAAC;QAGzE,KAAK,MAAMc,OAAOP,OAAO,IAAI,CAACG,WAC1B,IAAI,CAAEI,CAAAA,OAAOD,MAAK,GACdA,MAAM,CAACC,IAAI,GAAGJ,SAAS,CAACI,IAAI;QAIpC,MAAMC,SAASF;QACf,IAAI,CAAC,KAAK,CAAC,GAAG,CAACH,WAAWK;QAE1B,OAAOP,OAAO,EAAE,CAACO;IACrB;AACJ;AAEO,MAAMC,yCAAyCC,yBAAyB,oBAAoB,CAC/F;IACI,gBAAgB3B;IAChB,cAAc;QACV4B;QACA;YAACC;YAAgC;gBAAE,UAAU;YAAK;SAAE;QACpDC;KACH;AACL"}
+{"version":3,"file":"api/features/UpdateSettings/UpdateSettingsRepository.js","sources":["../../../../src/api/features/UpdateSettings/UpdateSettingsRepository.ts"],"sourcesContent":["import { Result } from \"@webiny/feature/api\";\nimport { KeyValueStore } from \"@webiny/api-core/features/keyValueStore/index.js\";\nimport type { OutputErrors } from \"@webiny/utils/createZodError.js\";\nimport {\n    AiPowerUpsSettingsGroupHandler,\n    AiPowerUpsSettingsCache\n} from \"~/api/features/shared/index.js\";\nimport { UpdateSettingsRepository } from \"./abstractions.js\";\nimport { SettingsValidationError } from \"./errors.js\";\nimport type { IAiPowerUpsSettings } from \"~/api/types.js\";\nimport { AI_POWER_UPS_SETTINGS } from \"~/api/constants.js\";\n\nclass UpdateSettingsRepositoryImpl implements UpdateSettingsRepository.Interface {\n    constructor(\n        private keyValueStore: KeyValueStore.Interface,\n        private handlers: AiPowerUpsSettingsGroupHandler.Interface[],\n        private cache: AiPowerUpsSettingsCache.Interface\n    ) {}\n\n    async execute(input: IAiPowerUpsSettings): Promise<Result<IAiPowerUpsSettings, Error>> {\n        const cached = this.cache.get();\n        let raw: Record<string, unknown>;\n\n        if (cached) {\n            raw = cached.raw;\n        } else {\n            const storeResult =\n                await this.keyValueStore.get<Record<string, unknown>>(AI_POWER_UPS_SETTINGS);\n            raw = storeResult.isOk() && storeResult.value ? storeResult.value : {};\n        }\n\n        const existingInternal: Record<string, unknown> = {};\n        for (const handler of this.handlers) {\n            /*\n             * `raw` matters here as much as it does on read. `connections` derives itself from the\n             * legacy `providers` section until it has one of its own, and this is what\n             * `mapToStorage` diffs against to carry forward an encrypted key the form only ever saw\n             * as a mask. Without it, the first save after an upgrade compares against an empty list\n             * and silently drops every stored key.\n             */\n            existingInternal[handler.name] = handler.mapFromStorage(raw[handler.name], raw);\n        }\n\n        const newSettings = input as unknown as Record<string, unknown>;\n        const invalidFields: OutputErrors = {};\n\n        for (const handler of this.handlers) {\n            const parseResult = await handler.inputSchema.safeParseAsync(newSettings[handler.name]);\n            if (!parseResult.success) {\n                for (const issue of parseResult.error.issues) {\n                    const fieldPath = [handler.name, ...issue.path.map(String)].join(\".\");\n                    invalidFields[fieldPath] = {\n                        code: issue.code,\n                        message: issue.message,\n                        data: { path: issue.path }\n                    };\n                }\n            }\n        }\n\n        if (Object.keys(invalidFields).length > 0) {\n            return Result.fail(new SettingsValidationError(invalidFields));\n        }\n\n        const persisted: Record<string, unknown> = { ...raw };\n        for (const handler of this.handlers) {\n            persisted[handler.name] = await handler.mapToStorage(\n                newSettings[handler.name],\n                existingInternal[handler.name]\n            );\n        }\n\n        const writeResult = await this.keyValueStore.set(AI_POWER_UPS_SETTINGS, persisted);\n        if (writeResult.isFail()) {\n            return Result.fail(new Error(String(writeResult.error)));\n        }\n\n        const result: Record<string, unknown> = {};\n        for (const handler of this.handlers) {\n            result[handler.name] = handler.mapFromStorage(persisted[handler.name], persisted);\n        }\n\n        for (const key of Object.keys(persisted)) {\n            if (!(key in result)) {\n                result[key] = persisted[key];\n            }\n        }\n\n        const mapped = result as unknown as IAiPowerUpsSettings;\n        this.cache.set(persisted, mapped);\n\n        return Result.ok(mapped);\n    }\n}\n\nexport const UpdateSettingsRepositoryImplementation = UpdateSettingsRepository.createImplementation(\n    {\n        implementation: UpdateSettingsRepositoryImpl,\n        dependencies: [\n            KeyValueStore,\n            [AiPowerUpsSettingsGroupHandler, { multiple: true }],\n            AiPowerUpsSettingsCache\n        ]\n    }\n);\n"],"names":["UpdateSettingsRepositoryImpl","keyValueStore","handlers","cache","input","cached","raw","storeResult","AI_POWER_UPS_SETTINGS","existingInternal","handler","newSettings","invalidFields","parseResult","issue","fieldPath","String","Object","Result","SettingsValidationError","persisted","writeResult","Error","result","key","mapped","UpdateSettingsRepositoryImplementation","UpdateSettingsRepository","KeyValueStore","AiPowerUpsSettingsGroupHandler","AiPowerUpsSettingsCache"],"mappings":";;;;;;AAYA,MAAMA;IACF,YACYC,aAAsC,EACtCC,QAAoD,EACpDC,KAAwC,CAClD;aAHUF,aAAa,GAAbA;aACAC,QAAQ,GAARA;aACAC,KAAK,GAALA;IACT;IAEH,MAAM,QAAQC,KAA0B,EAA+C;QACnF,MAAMC,SAAS,IAAI,CAAC,KAAK,CAAC,GAAG;QAC7B,IAAIC;QAEJ,IAAID,QACAC,MAAMD,OAAO,GAAG;aACb;YACH,MAAME,cACF,MAAM,IAAI,CAAC,aAAa,CAAC,GAAG,CAA0BC;YAC1DF,MAAMC,YAAY,IAAI,MAAMA,YAAY,KAAK,GAAGA,YAAY,KAAK,GAAG,CAAC;QACzE;QAEA,MAAME,mBAA4C,CAAC;QACnD,KAAK,MAAMC,WAAW,IAAI,CAAC,QAAQ,CAQ/BD,gBAAgB,CAACC,QAAQ,IAAI,CAAC,GAAGA,QAAQ,cAAc,CAACJ,GAAG,CAACI,QAAQ,IAAI,CAAC,EAAEJ;QAG/E,MAAMK,cAAcP;QACpB,MAAMQ,gBAA8B,CAAC;QAErC,KAAK,MAAMF,WAAW,IAAI,CAAC,QAAQ,CAAE;YACjC,MAAMG,cAAc,MAAMH,QAAQ,WAAW,CAAC,cAAc,CAACC,WAAW,CAACD,QAAQ,IAAI,CAAC;YACtF,IAAI,CAACG,YAAY,OAAO,EACpB,KAAK,MAAMC,SAASD,YAAY,KAAK,CAAC,MAAM,CAAE;gBAC1C,MAAME,YAAY;oBAACL,QAAQ,IAAI;uBAAKI,MAAM,IAAI,CAAC,GAAG,CAACE;iBAAQ,CAAC,IAAI,CAAC;gBACjEJ,aAAa,CAACG,UAAU,GAAG;oBACvB,MAAMD,MAAM,IAAI;oBAChB,SAASA,MAAM,OAAO;oBACtB,MAAM;wBAAE,MAAMA,MAAM,IAAI;oBAAC;gBAC7B;YACJ;QAER;QAEA,IAAIG,OAAO,IAAI,CAACL,eAAe,MAAM,GAAG,GACpC,OAAOM,OAAO,IAAI,CAAC,IAAIC,wBAAwBP;QAGnD,MAAMQ,YAAqC;YAAE,GAAGd,GAAG;QAAC;QACpD,KAAK,MAAMI,WAAW,IAAI,CAAC,QAAQ,CAC/BU,SAAS,CAACV,QAAQ,IAAI,CAAC,GAAG,MAAMA,QAAQ,YAAY,CAChDC,WAAW,CAACD,QAAQ,IAAI,CAAC,EACzBD,gBAAgB,CAACC,QAAQ,IAAI,CAAC;QAItC,MAAMW,cAAc,MAAM,IAAI,CAAC,aAAa,CAAC,GAAG,CAACb,uBAAuBY;QACxE,IAAIC,YAAY,MAAM,IAClB,OAAOH,OAAO,IAAI,CAAC,IAAII,MAAMN,OAAOK,YAAY,KAAK;QAGzD,MAAME,SAAkC,CAAC;QACzC,KAAK,MAAMb,WAAW,IAAI,CAAC,QAAQ,CAC/Ba,MAAM,CAACb,QAAQ,IAAI,CAAC,GAAGA,QAAQ,cAAc,CAACU,SAAS,CAACV,QAAQ,IAAI,CAAC,EAAEU;QAG3E,KAAK,MAAMI,OAAOP,OAAO,IAAI,CAACG,WAC1B,IAAI,CAAEI,CAAAA,OAAOD,MAAK,GACdA,MAAM,CAACC,IAAI,GAAGJ,SAAS,CAACI,IAAI;QAIpC,MAAMC,SAASF;QACf,IAAI,CAAC,KAAK,CAAC,GAAG,CAACH,WAAWK;QAE1B,OAAOP,OAAO,EAAE,CAACO;IACrB;AACJ;AAEO,MAAMC,yCAAyCC,yBAAyB,oBAAoB,CAC/F;IACI,gBAAgB3B;IAChB,cAAc;QACV4B;QACA;YAACC;YAAgC;gBAAE,UAAU;YAAK;SAAE;QACpDC;KACH;AACL"}

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

@@ -1,5 +1,5 @@
 import { TaskDefinition } from "@webiny/api-core/features/task/TaskDefinition/index.js";
-import { WebsocketService } from "@webiny/api-websockets/features/WebsocketService/index.js";
+import { WebsocketsSendToIdentityUseCase } from "@webiny/api-websockets/features/SendToIdentity/abstractions.js";
 import { WbGeneratePageContentUseCase } from "../../../api/features/WbGeneratePageContent/index.js";
 import { IdentityContext } from "@webiny/api-core/exports/api/security.js";
 export declare const WB_GENERATE_PAGE_CONTENT_TASK_ID = "aiPowerUpsGeneratePageContent";
@@ -11,18 +11,19 @@ export interface IWbGeneratePageContentTaskInput {
     excludedFileIds?: string[] | null;
     readerPersonaId?: string | null;
     writerPersonaId?: string | null;
+    additionalFileIds?: string[] | null;
 }
 declare class WbGeneratePageContentTaskImpl implements TaskDefinition.Interface<IWbGeneratePageContentTaskInput> {
     private identityContext;
     private generatePageContent;
-    private websocketService;
+    private sendToIdentity;
     id: string;
     title: string;
     description: string;
     maxIterations: number;
     isPrivate: boolean;
     databaseLogs: boolean;
-    constructor(identityContext: IdentityContext.Interface, generatePageContent: WbGeneratePageContentUseCase.Interface, websocketService: WebsocketService.Interface);
+    constructor(identityContext: IdentityContext.Interface, generatePageContent: WbGeneratePageContentUseCase.Interface, sendToIdentity: WebsocketsSendToIdentityUseCase.Interface);
     run({ input, controller }: TaskDefinition.RunParams<IWbGeneratePageContentTaskInput>): Promise<TaskDefinition.Result<IWbGeneratePageContentTaskInput>>;
     private sendContentToUser;
     private sendErrorToUser;

package/api/features/WbGeneratePageContent/WbGeneratePageContentTask.js

@@ -1,14 +1,14 @@
 import { TaskDefinition } from "@webiny/api-core/features/task/TaskDefinition/index.js";
-import { WebsocketService } from "@webiny/api-websockets/features/WebsocketService/index.js";
+import { WebsocketsSendToIdentityUseCase } from "@webiny/api-websockets/features/SendToIdentity/abstractions.js";
 import { compress } from "@webiny/utils/features/compression/legacy/gzip.js";
 import { WbGeneratePageContentUseCase } from "./index.js";
 import { IdentityContext } from "@webiny/api-core/exports/api/security.js";
 const WB_GENERATE_PAGE_CONTENT_TASK_ID = "aiPowerUpsGeneratePageContent";
 class WbGeneratePageContentTaskImpl {
-    constructor(identityContext, generatePageContent, websocketService){
+    constructor(identityContext, generatePageContent, sendToIdentity){
         this.identityContext = identityContext;
         this.generatePageContent = generatePageContent;
-        this.websocketService = websocketService;
+        this.sendToIdentity = sendToIdentity;
         this.id = WB_GENERATE_PAGE_CONTENT_TASK_ID;
         this.title = "AI Power-Ups - Generate Page Content";
         this.description = "Generates page content using AI based on a user prompt.";
@@ -18,16 +18,26 @@ class WbGeneratePageContentTaskImpl {
     }
     async run({ input, controller }) {
         if (controller.runtime.isAborted()) return controller.response.aborted();
-        const result = await this.generatePageContent.execute({
-            prompt: input.prompt,
-            components: input.components,
-            tools: input.tools,
-            projectId: input.projectId,
-            excludedFileIds: input.excludedFileIds,
-            readerPersonaId: input.readerPersonaId,
-            writerPersonaId: input.writerPersonaId
-        });
         const identity = this.identityContext.getIdentity();
+        let result;
+        try {
+            result = await this.generatePageContent.execute({
+                prompt: input.prompt,
+                components: input.components,
+                tools: input.tools,
+                projectId: input.projectId,
+                excludedFileIds: input.excludedFileIds,
+                readerPersonaId: input.readerPersonaId,
+                writerPersonaId: input.writerPersonaId,
+                additionalFileIds: input.additionalFileIds
+            });
+        } catch (err) {
+            const message = err instanceof Error ? err.message : String(err);
+            await this.sendErrorToUser(identity.id, message);
+            return controller.response.error({
+                message
+            });
+        }
         if (result.isFail()) {
             await this.sendErrorToUser(identity.id, result.error.message);
             return controller.response.error({
@@ -40,7 +50,7 @@ class WbGeneratePageContentTaskImpl {
         return controller.response.done("Page content generated successfully.");
     }
     async sendContentToUser(identityId, payload, telemetry) {
-        await this.websocketService.send({
+        await this.sendToIdentity.execute({
             id: identityId
         }, {
             action: "aiPowerUps.generatePageContent.content",
@@ -52,7 +62,7 @@ class WbGeneratePageContentTaskImpl {
         });
     }
     async sendErrorToUser(identityId, message) {
-        await this.websocketService.send({
+        await this.sendToIdentity.execute({
             id: identityId
         }, {
             action: "aiPowerUps.generatePageContent.error",
@@ -67,7 +77,7 @@ const WbGeneratePageContentTask = TaskDefinition.createImplementation({
     dependencies: [
         IdentityContext,
         WbGeneratePageContentUseCase,
-        WebsocketService
+        WebsocketsSendToIdentityUseCase
     ]
 });
 export { WB_GENERATE_PAGE_CONTENT_TASK_ID, WbGeneratePageContentTask };

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

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

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

@@ -1,20 +1,18 @@
 import { Result } from "@webiny/feature/api";
 import { Ai } from "@webiny/api-core/features/ai/index.js";
 import { AiSdkTools } from "@webiny/api-core/features/ai/index.js";
-import { Encryption } from "@webiny/api-core/features/encryption/index.js";
 import { ListTagsUseCase } from "@webiny/api-file-manager/features/file/ListTags/index.js";
-import { GetSettingsUseCase } from "../../../api/features/GetSettings/index.js";
+import { ResolveAiCapabilityUseCase } from "../../../api/features/Capabilities/index.js";
 import { AiPromptContextBuilder } from "../../../api/features/AiPromptContext/index.js";
 import { WbGeneratePageContentUseCase } from "./abstractions.js";
 import type { WbGeneratePageContentParams, GeneratePageContentResult } from "./abstractions.js";
 declare class WbGeneratePageContentUseCaseImpl implements WbGeneratePageContentUseCase.Interface {
     private promptContextBuilder;
-    private getSettings;
+    private resolveCapability;
     private ai;
     private aiSdkTools;
-    private encryption;
     private listTags;
-    constructor(promptContextBuilder: AiPromptContextBuilder.Interface, getSettings: GetSettingsUseCase.Interface, ai: Ai.Interface, aiSdkTools: AiSdkTools.Interface, encryption: Encryption.Interface, listTags: ListTagsUseCase.Interface);
+    constructor(promptContextBuilder: AiPromptContextBuilder.Interface, resolveCapability: ResolveAiCapabilityUseCase.Interface, ai: Ai.Interface, aiSdkTools: AiSdkTools.Interface, listTags: ListTagsUseCase.Interface);
     execute(params: WbGeneratePageContentParams): Promise<Result<GeneratePageContentResult, Error>>;
 }
 export declare const WbGeneratePageContentUseCaseImplementation: typeof WbGeneratePageContentUseCaseImpl & {

package/api/features/WbGeneratePageContent/WbGeneratePageContentUseCase.js

@@ -1,37 +1,34 @@
 import { stepCountIs } from "ai";
 import { Result } from "@webiny/feature/api";
 import { Ai, AiSdkTools } from "@webiny/api-core/features/ai/index.js";
-import { Encryption } from "@webiny/api-core/features/encryption/index.js";
 import { ListTagsUseCase } from "@webiny/api-file-manager/features/file/ListTags/index.js";
-import { GetSettingsUseCase } from "../GetSettings/index.js";
-import { AiPromptContextBuilder } from "../AiPromptContext/index.js";
+import { ResolveAiCapabilityUseCase, withAdditionalInstructions } from "../Capabilities/index.js";
+import { WB_GENERATE_PAGE_CAPABILITY } from "./capability.js";
+import { AiPromptContextBuilder, formatAdditionalFilesContext } from "../AiPromptContext/index.js";
 import { createReadProjectFileTool } from "../AiPromptContext/ReadProjectFileTool.js";
 import { WbGeneratePageContentUseCase } from "./abstractions.js";
 import { buildDomainPrompt } from "./buildPrompt.js";
-import { LlmJsonResponse } from "./LlmJsonResponse.js";
+import { LlmJsonResponse } from "../../../domain/LlmJsonResponse.js";
 import { ComponentFilter } from "./ComponentFilter.js";
 class WbGeneratePageContentUseCaseImpl {
-    constructor(promptContextBuilder, getSettings, ai, aiSdkTools, encryption, listTags){
+    constructor(promptContextBuilder, resolveCapability, ai, aiSdkTools, listTags){
         this.promptContextBuilder = promptContextBuilder;
-        this.getSettings = getSettings;
+        this.resolveCapability = resolveCapability;
         this.ai = ai;
         this.aiSdkTools = aiSdkTools;
-        this.encryption = encryption;
         this.listTags = listTags;
     }
     async execute(params) {
-        const settingsResult = await this.getSettings.execute();
-        if (settingsResult.isFail()) return Result.fail(new Error("Failed to load AI PowerUps settings."));
-        const settings = settingsResult.value;
-        const firstProvider = settings.providers.presets[0];
-        if (!firstProvider) return Result.fail(new Error("No AI provider configured. Add a provider in AI Power Ups settings."));
-        const apiKey = await this.encryption.decrypt(firstProvider.apiKeyEncrypted);
+        const resolved = await this.resolveCapability.execute(WB_GENERATE_PAGE_CAPABILITY);
+        if (resolved.isFail()) return Result.fail(resolved.error);
+        const capability = resolved.value;
         const sdkTools = this.aiSdkTools.getToolSet();
         const context = await this.promptContextBuilder.execute({
             projectId: params.projectId,
             readerPersonaId: params.readerPersonaId,
             writerPersonaId: params.writerPersonaId,
-            excludedFileIds: params.excludedFileIds
+            excludedFileIds: params.excludedFileIds,
+            additionalFileIds: params.additionalFileIds
         });
         if (context.allProjectFiles.length > 0) {
             const projectFileTool = createReadProjectFileTool(context.allProjectFiles, context.excludedFileIds);
@@ -45,7 +42,7 @@ class WbGeneratePageContentUseCaseImpl {
         });
         const imageTags = tagsResult.isOk() ? tagsResult.value.map((t)=>t.tag) : [];
         const components = params.components;
-        const systemText = buildDomainPrompt(components, params.tools, imageTags) + context.toString();
+        const systemText = withAdditionalInstructions(capability, buildDomainPrompt(components, params.tools, imageTags) + context.toString());
         const system = {
             role: "system",
             content: systemText,
@@ -59,14 +56,11 @@ class WbGeneratePageContentUseCaseImpl {
         };
         try {
             const aiResult = await this.ai.generateText({
-                model: firstProvider.model,
-                connection: {
-                    sdkName: firstProvider.model.split("/")[0],
-                    apiKey
-                },
+                model: capability.model,
+                connection: capability.connection,
                 system,
                 toolChoice: "auto",
-                prompt: params.prompt,
+                prompt: params.prompt + formatAdditionalFilesContext(context.additionalFiles),
                 ...Object.keys(sdkTools).length > 0 ? {
                     tools: sdkTools,
                     stopWhen: stepCountIs(20)
@@ -108,10 +102,9 @@ const WbGeneratePageContentUseCaseImplementation = WbGeneratePageContentUseCase.
     implementation: WbGeneratePageContentUseCaseImpl,
     dependencies: [
         AiPromptContextBuilder,
-        GetSettingsUseCase,
+        ResolveAiCapabilityUseCase,
         Ai,
         AiSdkTools,
-        Encryption,
         ListTagsUseCase
     ]
 });