@webiny/ai-powerups 6.6.0-alpha.3 → 6.6.0-alpha.5

package/api/features/Connections/ConnectionsHandler.js

@@ -0,0 +1,84 @@
+import { z } from "zod";
+import { Encryption } from "@webiny/api-core/features/encryption/index.js";
+import { Masker } from "@webiny/api-core/features/masker/index.js";
+import { AiPowerUpsSettingsGroupHandler } from "../shared/index.js";
+import { readLegacyProviderPresets } from "./legacyProviders.js";
+import { sdkNameFromModel } from "../shared/modelId.js";
+const inputSchema = z.object({
+    presets: z.array(z.object({
+        id: z.string().min(1),
+        name: z.string().min(1),
+        sdkName: z.string().min(1),
+        apiKey: z.string().nullish()
+    }))
+});
+class ConnectionsHandlerImpl {
+    constructor(encryption, masker){
+        this.encryption = encryption;
+        this.masker = masker;
+        this.name = "connections";
+        this.inputSchema = inputSchema;
+    }
+    mapFromStorage(persisted, all) {
+        if (persisted && "object" == typeof persisted) {
+            const data = persisted;
+            return {
+                presets: (data.presets ?? []).map((p)=>({
+                        id: p.id,
+                        name: p.name,
+                        sdkName: p.sdkName,
+                        apiKeyMasked: p.apiKeyMasked ?? "",
+                        apiKeyEncrypted: p.apiKeyEncrypted ?? ""
+                    }))
+            };
+        }
+        return {
+            presets: readLegacyProviderPresets(all).map((p)=>({
+                    id: p.id,
+                    name: p.name,
+                    sdkName: sdkNameFromModel(p.model),
+                    apiKeyMasked: p.apiKeyMasked ?? "",
+                    apiKeyEncrypted: p.apiKeyEncrypted ?? ""
+                }))
+        };
+    }
+    async mapToStorage(internal, existing) {
+        const input = internal;
+        const existingPresets = existing?.presets ?? [];
+        const presets = await Promise.all(input.presets.map(async (preset)=>{
+            const existingMatch = existingPresets.find((ep)=>ep.id === preset.id);
+            let apiKeyEncrypted;
+            let apiKeyMasked;
+            if (preset.apiKey && preset.apiKey !== existingMatch?.apiKeyMasked) {
+                apiKeyEncrypted = await this.encryption.encrypt(preset.apiKey);
+                apiKeyMasked = this.masker.mask(preset.apiKey, [
+                    8,
+                    4
+                ]);
+            } else {
+                apiKeyEncrypted = existingMatch?.apiKeyEncrypted ?? "";
+                apiKeyMasked = existingMatch?.apiKeyMasked ?? "";
+            }
+            return {
+                id: preset.id,
+                name: preset.name,
+                sdkName: preset.sdkName,
+                apiKeyEncrypted,
+                apiKeyMasked
+            };
+        }));
+        return {
+            presets
+        };
+    }
+}
+const ConnectionsHandler = AiPowerUpsSettingsGroupHandler.createImplementation({
+    implementation: ConnectionsHandlerImpl,
+    dependencies: [
+        Encryption,
+        Masker
+    ]
+});
+export default ConnectionsHandler;
+
+//# sourceMappingURL=ConnectionsHandler.js.map

package/api/features/Connections/ConnectionsHandler.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"api/features/Connections/ConnectionsHandler.js","sources":["../../../../src/api/features/Connections/ConnectionsHandler.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 { AiPowerUpsSettingsGroupHandler } from \"~/api/features/shared/index.js\";\nimport { readLegacyProviderPresets } from \"./legacyProviders.js\";\nimport { sdkNameFromModel } from \"~/api/features/shared/modelId.js\";\nimport type {\n    ConnectionsSettings,\n    PersistedAiConnectionPreset,\n    PersistedConnections\n} 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            sdkName: z.string().min(1),\n            apiKey: z.string().nullish()\n        })\n    )\n});\n\nclass ConnectionsHandlerImpl implements AiPowerUpsSettingsGroupHandler.Interface {\n    readonly name = \"connections\";\n    readonly inputSchema = inputSchema;\n\n    constructor(\n        private encryption: Encryption.Interface,\n        private masker: Masker.Interface\n    ) {}\n\n    mapFromStorage(persisted: unknown, all?: Record<string, unknown>): ConnectionsSettings {\n        if (persisted && typeof persisted === \"object\") {\n            const data = persisted as PersistedConnections;\n            return {\n                presets: (data.presets ?? []).map(p => ({\n                    id: p.id,\n                    name: p.name,\n                    sdkName: p.sdkName,\n                    apiKeyMasked: p.apiKeyMasked ?? \"\",\n                    apiKeyEncrypted: p.apiKeyEncrypted ?? \"\"\n                }))\n            };\n        }\n\n        /*\n         * No `connections` section yet, so derive one from the legacy `providers` presets. The keys\n         * are already encrypted in storage, which is why this can be a pure read: nothing has to be\n         * re-encrypted, and nothing is written back until the next save.\n         *\n         * Dropping the model here is safe. `ModelRoles` reads the same legacy section and lifts\n         * the first preset's model into the `standard` role, which is the only model the old code\n         * ever used.\n         */\n        return {\n            presets: readLegacyProviderPresets(all).map(p => ({\n                id: p.id,\n                name: p.name,\n                sdkName: sdkNameFromModel(p.model),\n                apiKeyMasked: p.apiKeyMasked ?? \"\",\n                apiKeyEncrypted: p.apiKeyEncrypted ?? \"\"\n            }))\n        };\n    }\n\n    async mapToStorage(internal: unknown, existing: unknown | null): Promise<PersistedConnections> {\n        const input = internal as ConnectionsSettings;\n        const existingPresets = (existing as ConnectionsSettings | null)?.presets ?? [];\n\n        const presets: PersistedAiConnectionPreset[] = 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                    // The form sends back the mask when the key was left alone. Carry it forward.\n                    apiKeyEncrypted = existingMatch?.apiKeyEncrypted ?? \"\";\n                    apiKeyMasked = existingMatch?.apiKeyMasked ?? \"\";\n                } else {\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                    sdkName: preset.sdkName,\n                    apiKeyEncrypted,\n                    apiKeyMasked\n                };\n            })\n        );\n\n        return { presets };\n    }\n}\n\nexport default AiPowerUpsSettingsGroupHandler.createImplementation({\n    implementation: ConnectionsHandlerImpl,\n    dependencies: [Encryption, Masker]\n});\n"],"names":["inputSchema","z","ConnectionsHandlerImpl","encryption","masker","persisted","all","data","p","readLegacyProviderPresets","sdkNameFromModel","internal","existing","input","existingPresets","presets","Promise","preset","existingMatch","ep","apiKeyEncrypted","apiKeyMasked","AiPowerUpsSettingsGroupHandler","Encryption","Masker"],"mappings":";;;;;;AAYA,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,SAASA,EAAE,MAAM,GAAG,GAAG,CAAC;QACxB,QAAQA,EAAE,MAAM,GAAG,OAAO;IAC9B;AAER;AAEA,MAAMC;IAIF,YACYC,UAAgC,EAChCC,MAAwB,CAClC;aAFUD,UAAU,GAAVA;aACAC,MAAM,GAANA;aALH,IAAI,GAAG;aACP,WAAW,GAAGJ;IAKpB;IAEH,eAAeK,SAAkB,EAAEC,GAA6B,EAAuB;QACnF,IAAID,aAAa,AAAqB,YAArB,OAAOA,WAAwB;YAC5C,MAAME,OAAOF;YACb,OAAO;gBACH,SAAUE,AAAAA,CAAAA,KAAK,OAAO,IAAI,EAAC,EAAG,GAAG,CAACC,CAAAA,IAAM;wBACpC,IAAIA,EAAE,EAAE;wBACR,MAAMA,EAAE,IAAI;wBACZ,SAASA,EAAE,OAAO;wBAClB,cAAcA,EAAE,YAAY,IAAI;wBAChC,iBAAiBA,EAAE,eAAe,IAAI;oBAC1C;YACJ;QACJ;QAWA,OAAO;YACH,SAASC,0BAA0BH,KAAK,GAAG,CAACE,CAAAA,IAAM;oBAC9C,IAAIA,EAAE,EAAE;oBACR,MAAMA,EAAE,IAAI;oBACZ,SAASE,iBAAiBF,EAAE,KAAK;oBACjC,cAAcA,EAAE,YAAY,IAAI;oBAChC,iBAAiBA,EAAE,eAAe,IAAI;gBAC1C;QACJ;IACJ;IAEA,MAAM,aAAaG,QAAiB,EAAEC,QAAwB,EAAiC;QAC3F,MAAMC,QAAQF;QACd,MAAMG,kBAAmBF,UAAyC,WAAW,EAAE;QAE/E,MAAMG,UAAyC,MAAMC,QAAQ,GAAG,CAC5DH,MAAM,OAAO,CAAC,GAAG,CAAC,OAAMI;YACpB,MAAMC,gBAAgBJ,gBAAgB,IAAI,CAACK,CAAAA,KAAMA,GAAG,EAAE,KAAKF,OAAO,EAAE;YAEpE,IAAIG;YACJ,IAAIC;YAEJ,IAAI,AAACJ,OAAO,MAAM,IAAIA,OAAO,MAAM,KAAKC,eAAe,cAIhD;gBACHE,kBAAkB,MAAM,IAAI,CAAC,UAAU,CAAC,OAAO,CAACH,OAAO,MAAM;gBAC7DI,eAAe,IAAI,CAAC,MAAM,CAAC,IAAI,CAACJ,OAAO,MAAM,EAAE;oBAAC;oBAAG;iBAAE;YACzD,OAPqE;gBAEjEG,kBAAkBF,eAAe,mBAAmB;gBACpDG,eAAeH,eAAe,gBAAgB;YAClD;YAKA,OAAO;gBACH,IAAID,OAAO,EAAE;gBACb,MAAMA,OAAO,IAAI;gBACjB,SAASA,OAAO,OAAO;gBACvBG;gBACAC;YACJ;QACJ;QAGJ,OAAO;YAAEN;QAAQ;IACrB;AACJ;AAEA,2BAAeO,+BAA+B,oBAAoB,CAAC;IAC/D,gBAAgBpB;IAChB,cAAc;QAACqB;QAAYC;KAAO;AACtC"}

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

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

package/api/features/Connections/feature.js

@@ -0,0 +1,13 @@
+import { createFeature } from "@webiny/feature/api";
+import ConnectionsHandler from "./ConnectionsHandler.js";
+import ConnectionsGraphQLMapper from "./ConnectionsGraphQLMapper.js";
+const ConnectionsFeature = createFeature({
+    name: "AiPowerUpsConnections",
+    register (container) {
+        container.register(ConnectionsHandler);
+        container.register(ConnectionsGraphQLMapper);
+    }
+});
+export { ConnectionsFeature };
+
+//# sourceMappingURL=feature.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"api/features/Connections/feature.js","sources":["../../../../src/api/features/Connections/feature.ts"],"sourcesContent":["import { createFeature } from \"@webiny/feature/api\";\nimport ConnectionsHandler from \"./ConnectionsHandler.js\";\nimport ConnectionsGraphQLMapper from \"./ConnectionsGraphQLMapper.js\";\n\nexport const ConnectionsFeature = createFeature({\n    name: \"AiPowerUpsConnections\",\n    register(container) {\n        container.register(ConnectionsHandler);\n        container.register(ConnectionsGraphQLMapper);\n    }\n});\n"],"names":["ConnectionsFeature","createFeature","container","ConnectionsHandler","ConnectionsGraphQLMapper"],"mappings":";;;AAIO,MAAMA,qBAAqBC,cAAc;IAC5C,MAAM;IACN,UAASC,SAAS;QACdA,UAAU,QAAQ,CAACC;QACnBD,UAAU,QAAQ,CAACE;IACvB;AACJ"}

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

@@ -0,0 +1,2 @@
+export type { AiConnectionPreset, ConnectionsSettings } from "./types.js";
+export { ConnectionsFeature } from "./feature.js";

package/api/features/Connections/index.js

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

package/api/features/Connections/legacyProviders.d.ts

@@ -0,0 +1,26 @@
+/**
+ * Reading the settings section `connections` replaces.
+ *
+ * `providers` bundled a credential with a single model. Nothing writes it any more, but an upgraded
+ * project still has one and has never saved since, so both `Connections` and `ModelRoles` derive
+ * their first value from it. That makes this the one piece of code allowed to reach into a section
+ * nobody owns.
+ *
+ * It lives in its own module, and not with the connection types, because it is the only part of
+ * this feature with an expiry date: when the legacy section is finally dropped, this file and its
+ * two callers go with it.
+ */
+/** Shape of a preset in the legacy `providers` section. */
+export interface LegacyProviderPreset {
+    id: string;
+    name: string;
+    model?: string;
+    apiKeyEncrypted?: string;
+    apiKeyMasked?: string;
+}
+/**
+ * Deliberately defensive. This parses a blob written by an older version against no schema, so
+ * every level is checked rather than asserted: a project that never had `providers` and one that
+ * has a malformed one both read as "nothing to migrate" instead of throwing on read.
+ */
+export declare const readLegacyProviderPresets: (all: Record<string, unknown> | undefined) => LegacyProviderPreset[];

package/api/features/Connections/legacyProviders.js

@@ -0,0 +1,9 @@
+const readLegacyProviderPresets = (all)=>{
+    const providers = all?.["providers"];
+    if (!providers || "object" != typeof providers) return [];
+    const presets = providers.presets;
+    return Array.isArray(presets) ? presets : [];
+};
+export { readLegacyProviderPresets };
+
+//# sourceMappingURL=legacyProviders.js.map

package/api/features/Connections/legacyProviders.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"api/features/Connections/legacyProviders.js","sources":["../../../../src/api/features/Connections/legacyProviders.ts"],"sourcesContent":["/**\n * Reading the settings section `connections` replaces.\n *\n * `providers` bundled a credential with a single model. Nothing writes it any more, but an upgraded\n * project still has one and has never saved since, so both `Connections` and `ModelRoles` derive\n * their first value from it. That makes this the one piece of code allowed to reach into a section\n * nobody owns.\n *\n * It lives in its own module, and not with the connection types, because it is the only part of\n * this feature with an expiry date: when the legacy section is finally dropped, this file and its\n * two callers go with it.\n */\n\n/** Shape of a preset in the legacy `providers` section. */\nexport interface LegacyProviderPreset {\n    id: string;\n    name: string;\n    model?: string;\n    apiKeyEncrypted?: string;\n    apiKeyMasked?: string;\n}\n\n/**\n * Deliberately defensive. This parses a blob written by an older version against no schema, so\n * every level is checked rather than asserted: a project that never had `providers` and one that\n * has a malformed one both read as \"nothing to migrate\" instead of throwing on read.\n */\nexport const readLegacyProviderPresets = (\n    all: Record<string, unknown> | undefined\n): LegacyProviderPreset[] => {\n    const providers = all?.[\"providers\"];\n    if (!providers || typeof providers !== \"object\") {\n        return [];\n    }\n\n    const presets = (providers as { presets?: unknown }).presets;\n    return Array.isArray(presets) ? (presets as LegacyProviderPreset[]) : [];\n};\n"],"names":["readLegacyProviderPresets","all","providers","presets","Array"],"mappings":"AA2BO,MAAMA,4BAA4B,CACrCC;IAEA,MAAMC,YAAYD,KAAK,CAAC,YAAY;IACpC,IAAI,CAACC,aAAa,AAAqB,YAArB,OAAOA,WACrB,OAAO,EAAE;IAGb,MAAMC,UAAWD,UAAoC,OAAO;IAC5D,OAAOE,MAAM,OAAO,CAACD,WAAYA,UAAqC,EAAE;AAC5E"}

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

@@ -0,0 +1,36 @@
+import type { IAiPowerUpsSettings } from "../../../api/types.js";
+/**
+ * A connection is one credential for one vendor. It carries no model.
+ *
+ * The section it replaces (`providers`) bundled a key with a single model, so a project that wanted
+ * two Anthropic models had to paste the same key twice and keep both copies in sync. Splitting the
+ * two means the key is entered once and every model that vendor offers is then selectable.
+ */
+export interface AiConnectionPreset {
+    id: string;
+    name: string;
+    /** AI SDK namespace, e.g. "anthropic" or "openai". Matches `AiSdkFactory.id`. */
+    sdkName: string;
+    /** Plaintext. Only ever set on the way in from the admin form, never persisted. */
+    apiKey?: string;
+    apiKeyMasked: string;
+    apiKeyEncrypted: string;
+}
+declare module "~/api/types.js" {
+    interface IAiPowerUpsSettings {
+        connections: {
+            presets: AiConnectionPreset[];
+        };
+    }
+}
+export type ConnectionsSettings = IAiPowerUpsSettings["connections"];
+export interface PersistedAiConnectionPreset {
+    id: string;
+    name: string;
+    sdkName: string;
+    apiKeyEncrypted: string;
+    apiKeyMasked: string;
+}
+export interface PersistedConnections {
+    presets: PersistedAiConnectionPreset[];
+}

package/api/features/GetSettings/GetSettingsRepository.js

@@ -15,7 +15,7 @@ class GetSettingsRepositoryImpl {
         const storeResult = await this.keyValueStore.get(AI_POWER_UPS_SETTINGS);
         const raw = storeResult.isOk() && storeResult.value ? storeResult.value : {};
         const result = {};
-        for (const handler of this.handlers)result[handler.name] = handler.mapFromStorage(raw[handler.name]);
+        for (const handler of this.handlers)result[handler.name] = handler.mapFromStorage(raw[handler.name], raw);
         for (const key of Object.keys(raw))if (!(key in result)) result[key] = raw[key];
         const mapped = result;
         this.cache.set(raw, mapped);

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/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/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/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,9 +1,9 @@
 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 { 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";
@@ -11,21 +11,17 @@ import { buildDomainPrompt } from "./buildPrompt.js";
 import { LlmJsonResponse } from "./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,
@@ -46,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,
@@ -60,11 +56,8 @@ 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 + formatAdditionalFilesContext(context.additionalFiles),
@@ -109,10 +102,9 @@ const WbGeneratePageContentUseCaseImplementation = WbGeneratePageContentUseCase.
     implementation: WbGeneratePageContentUseCaseImpl,
     dependencies: [
         AiPromptContextBuilder,
-        GetSettingsUseCase,
+        ResolveAiCapabilityUseCase,
         Ai,
         AiSdkTools,
-        Encryption,
         ListTagsUseCase
     ]
 });