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

package/api/Extension.js.map

@@ -1 +1 @@
-{"version":3,"file":"api/Extension.js","sources":["../../src/api/Extension.ts"],"sourcesContent":["import { createFeature } from \"@webiny/feature/api\";\nimport { WcpContext } from \"@webiny/api-core/features/wcp/WcpContext/index.js\";\nimport { BaseGraphQLSchema } from \"./graphql/BaseGraphQLSchema.js\";\nimport AiPowerUpsSettingsGraphQLMapperImpl from \"./graphql/AiPowerUpsSettingsGraphQLMapper.js\";\nimport { AiPowerUpsSettingsCache } from \"./features/shared/SettingsCache.js\";\nimport { GetSettingsFeature } from \"./features/GetSettings/feature.js\";\nimport { UpdateSettingsFeature } from \"./features/UpdateSettings/feature.js\";\nimport { WbGeneratePageContentFeature } from \"./features/WbGeneratePageContent/feature.js\";\nimport { ProvidersFeature } from \"./features/Providers/feature.js\";\nimport { ReaderPersonasFeature } from \"./features/ReaderPersonas/feature.js\";\nimport { WriterPersonasFeature } from \"./features/WriterPersonas/feature.js\";\nimport { ProjectsFeature } from \"./features/Projects/feature.js\";\nimport { AiPromptContextFeature } from \"./features/AiPromptContext/feature.js\";\nimport { AiImageEnrichmentFeature } from \"./features/AiImageEnrichment/feature.js\";\nimport { ExtractFrontmatterFeature } from \"./features/ExtractFrontmatter/feature.js\";\n\nexport const Extension = createFeature({\n    name: \"AiPowerUps\",\n    register(container) {\n        container.register(AiPowerUpsSettingsCache).inSingletonScope();\n\n        GetSettingsFeature.register(container);\n        UpdateSettingsFeature.register(container);\n        ProvidersFeature.register(container);\n        ReaderPersonasFeature.register(container);\n        WriterPersonasFeature.register(container);\n        ProjectsFeature.register(container);\n        AiPromptContextFeature.register(container);\n        WbGeneratePageContentFeature.register(container);\n        ExtractFrontmatterFeature.register(container);\n\n        const wcp = container.resolve(WcpContext);\n        if (wcp.canUseAiImageEnrichment()) {\n            AiImageEnrichmentFeature.register(container);\n        }\n\n        container.register(AiPowerUpsSettingsGraphQLMapperImpl).inSingletonScope();\n        container.register(BaseGraphQLSchema);\n    }\n});\n"],"names":["Extension","createFeature","container","AiPowerUpsSettingsCache","GetSettingsFeature","UpdateSettingsFeature","ProvidersFeature","ReaderPersonasFeature","WriterPersonasFeature","ProjectsFeature","AiPromptContextFeature","WbGeneratePageContentFeature","ExtractFrontmatterFeature","wcp","WcpContext","AiImageEnrichmentFeature","AiPowerUpsSettingsGraphQLMapperImpl","BaseGraphQLSchema"],"mappings":";;;;;;;;;;;;;;;AAgBO,MAAMA,YAAYC,cAAc;IACnC,MAAM;IACN,UAASC,SAAS;QACdA,UAAU,QAAQ,CAACC,yBAAyB,gBAAgB;QAE5DC,mBAAmB,QAAQ,CAACF;QAC5BG,sBAAsB,QAAQ,CAACH;QAC/BI,iBAAiB,QAAQ,CAACJ;QAC1BK,sBAAsB,QAAQ,CAACL;QAC/BM,sBAAsB,QAAQ,CAACN;QAC/BO,gBAAgB,QAAQ,CAACP;QACzBQ,uBAAuB,QAAQ,CAACR;QAChCS,6BAA6B,QAAQ,CAACT;QACtCU,0BAA0B,QAAQ,CAACV;QAEnC,MAAMW,MAAMX,UAAU,OAAO,CAACY;QAC9B,IAAID,IAAI,uBAAuB,IAC3BE,yBAAyB,QAAQ,CAACb;QAGtCA,UAAU,QAAQ,CAACc,iCAAqC,gBAAgB;QACxEd,UAAU,QAAQ,CAACe;IACvB;AACJ"}
+{"version":3,"file":"api/Extension.js","sources":["../../src/api/Extension.ts"],"sourcesContent":["import { createFeature } from \"@webiny/feature/api\";\nimport { WcpContext } from \"@webiny/api-core/features/wcp/WcpContext/index.js\";\nimport { BaseGraphQLSchema } from \"./graphql/BaseGraphQLSchema.js\";\nimport AiPowerUpsSettingsGraphQLMapperImpl from \"./graphql/AiPowerUpsSettingsGraphQLMapper.js\";\nimport { AiPowerUpsSettingsCache } from \"./features/shared/SettingsCache.js\";\nimport { GetSettingsFeature } from \"./features/GetSettings/feature.js\";\nimport { UpdateSettingsFeature } from \"./features/UpdateSettings/feature.js\";\nimport { WbGeneratePageContentFeature } from \"./features/WbGeneratePageContent/feature.js\";\nimport { ProvidersFeature } from \"./features/Providers/feature.js\";\nimport { ConnectionsFeature } from \"./features/Connections/feature.js\";\nimport { ModelRolesFeature } from \"./features/ModelRoles/feature.js\";\nimport { CapabilitiesFeature } from \"./features/Capabilities/feature.js\";\nimport { ReaderPersonasFeature } from \"./features/ReaderPersonas/feature.js\";\nimport { WriterPersonasFeature } from \"./features/WriterPersonas/feature.js\";\nimport { ProjectsFeature } from \"./features/Projects/feature.js\";\nimport { AiPromptContextFeature } from \"./features/AiPromptContext/feature.js\";\nimport { AiImageEnrichmentFeature } from \"./features/AiImageEnrichment/feature.js\";\nimport { ExtractFrontmatterFeature } from \"./features/ExtractFrontmatter/feature.js\";\nimport { CmsGenerateEntryContentFeature } from \"./features/CmsGenerateEntryContent/feature.js\";\nimport { CmsResolveImageToolFeature } from \"./features/CmsResolveImageTool/feature.js\";\nimport { CmsCompareEntryRevisionsFeature } from \"./features/CmsCompareEntryRevisions/feature.js\";\nimport { WbTranslatePageFeature } from \"./features/WbTranslatePage/feature.js\";\nimport { CmsCompareEntryRevisionsSchema } from \"./graphql/CmsCompareEntryRevisionsSchema.js\";\n\nexport const Extension = createFeature({\n    name: \"AiPowerUps\",\n    register(container) {\n        container.register(AiPowerUpsSettingsCache).inSingletonScope();\n\n        GetSettingsFeature.register(container);\n        UpdateSettingsFeature.register(container);\n        /*\n         * `ProvidersFeature` is still registered so the legacy `providers` section keeps round-\n         * tripping through storage. `Connections` and `ModelRoles` read it once to seed\n         * themselves; nothing else does. It goes away with the next breaking release.\n         */\n        ProvidersFeature.register(container);\n        ConnectionsFeature.register(container);\n        ModelRolesFeature.register(container);\n        CapabilitiesFeature.register(container);\n        ReaderPersonasFeature.register(container);\n        WriterPersonasFeature.register(container);\n        ProjectsFeature.register(container);\n        AiPromptContextFeature.register(container);\n        WbGeneratePageContentFeature.register(container);\n        CmsGenerateEntryContentFeature.register(container);\n        CmsResolveImageToolFeature.register(container);\n        ExtractFrontmatterFeature.register(container);\n\n        const wcp = container.resolve(WcpContext);\n        if (wcp.canUseAiImageEnrichment()) {\n            AiImageEnrichmentFeature.register(container);\n        }\n\n        CmsCompareEntryRevisionsFeature.register(container);\n        WbTranslatePageFeature.register(container);\n\n        container.register(AiPowerUpsSettingsGraphQLMapperImpl).inSingletonScope();\n        container.register(BaseGraphQLSchema);\n        container.register(CmsCompareEntryRevisionsSchema);\n    }\n});\n"],"names":["Extension","createFeature","container","AiPowerUpsSettingsCache","GetSettingsFeature","UpdateSettingsFeature","ProvidersFeature","ConnectionsFeature","ModelRolesFeature","CapabilitiesFeature","ReaderPersonasFeature","WriterPersonasFeature","ProjectsFeature","AiPromptContextFeature","WbGeneratePageContentFeature","CmsGenerateEntryContentFeature","CmsResolveImageToolFeature","ExtractFrontmatterFeature","wcp","WcpContext","AiImageEnrichmentFeature","CmsCompareEntryRevisionsFeature","WbTranslatePageFeature","AiPowerUpsSettingsGraphQLMapperImpl","BaseGraphQLSchema","CmsCompareEntryRevisionsSchema"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;AAwBO,MAAMA,YAAYC,cAAc;IACnC,MAAM;IACN,UAASC,SAAS;QACdA,UAAU,QAAQ,CAACC,yBAAyB,gBAAgB;QAE5DC,mBAAmB,QAAQ,CAACF;QAC5BG,sBAAsB,QAAQ,CAACH;QAM/BI,iBAAiB,QAAQ,CAACJ;QAC1BK,mBAAmB,QAAQ,CAACL;QAC5BM,kBAAkB,QAAQ,CAACN;QAC3BO,oBAAoB,QAAQ,CAACP;QAC7BQ,sBAAsB,QAAQ,CAACR;QAC/BS,sBAAsB,QAAQ,CAACT;QAC/BU,gBAAgB,QAAQ,CAACV;QACzBW,uBAAuB,QAAQ,CAACX;QAChCY,6BAA6B,QAAQ,CAACZ;QACtCa,+BAA+B,QAAQ,CAACb;QACxCc,2BAA2B,QAAQ,CAACd;QACpCe,0BAA0B,QAAQ,CAACf;QAEnC,MAAMgB,MAAMhB,UAAU,OAAO,CAACiB;QAC9B,IAAID,IAAI,uBAAuB,IAC3BE,yBAAyB,QAAQ,CAAClB;QAGtCmB,gCAAgC,QAAQ,CAACnB;QACzCoB,uBAAuB,QAAQ,CAACpB;QAEhCA,UAAU,QAAQ,CAACqB,iCAAqC,gBAAgB;QACxErB,UAAU,QAAQ,CAACsB;QACnBtB,UAAU,QAAQ,CAACuB;IACvB;AACJ"}

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

@@ -1,11 +1,12 @@
 import { TaskDefinition } from "@webiny/api-core/features/task/TaskDefinition/index.js";
 import { Ai } from "@webiny/api-core/features/ai/index.js";
-import { Encryption } from "@webiny/api-core/features/encryption/index.js";
+import { Logger } from "@webiny/api-core/features/logger/index.js";
 import { GetFileUseCase } from "@webiny/api-file-manager/features/file/GetFile/index.js";
 import { UpdateFileUseCase } from "@webiny/api-file-manager/features/file/UpdateFile/index.js";
 import { GetSettingsUseCase as FmGetSettingsUseCase } from "@webiny/api-file-manager/features/settings/GetSettings/abstractions.js";
-import { WebsocketService } from "@webiny/api-websockets/features/WebsocketService/index.js";
-import { GetSettingsUseCase } from "../../../api/features/GetSettings/index.js";
+import { WebsocketsSendToIdentityUseCase } from "@webiny/api-websockets/features/SendToIdentity/abstractions.js";
+import { IdentityContext } from "@webiny/api-core/exports/api/security.js";
+import { ResolveAiCapabilityUseCase } from "../../../api/features/Capabilities/index.js";
 export declare const AI_IMAGE_ENRICHMENT_TASK_ID = "fmAiImageEnrichment";
 export interface IAiImageEnrichmentTaskInput {
     fileId: string;
@@ -15,9 +16,10 @@ declare class AiImageEnrichmentTaskImpl implements TaskDefinition.Interface<IAiI
     private fmSettings;
     private updateFile;
     private ai;
-    private getSettings;
-    private encryption;
-    private websocketService?;
+    private resolveCapability;
+    private logger;
+    private identityContext;
+    private sendToIdentity;
     id: string;
     title: string;
     description: string;
@@ -25,10 +27,10 @@ declare class AiImageEnrichmentTaskImpl implements TaskDefinition.Interface<IAiI
     isPrivate: boolean;
     databaseLogs: boolean;
     readonly selfCleanup: ("onAbort" | "onSuccess")[];
-    constructor(getFile: GetFileUseCase.Interface, fmSettings: FmGetSettingsUseCase.Interface, updateFile: UpdateFileUseCase.Interface, ai: Ai.Interface, getSettings: GetSettingsUseCase.Interface, encryption: Encryption.Interface, websocketService?: WebsocketService.Interface | undefined);
+    constructor(getFile: GetFileUseCase.Interface, fmSettings: FmGetSettingsUseCase.Interface, updateFile: UpdateFileUseCase.Interface, ai: Ai.Interface, resolveCapability: ResolveAiCapabilityUseCase.Interface, logger: Logger.Interface, identityContext: IdentityContext.Interface, sendToIdentity: WebsocketsSendToIdentityUseCase.Interface);
     run({ input, controller }: TaskDefinition.RunParams<IAiImageEnrichmentTaskInput>): Promise<TaskDefinition.Result<IAiImageEnrichmentTaskInput>>;
 }
 export declare const AiImageEnrichmentTask: typeof AiImageEnrichmentTaskImpl & {
-    __abstraction: import("@webiny/di").Abstraction<import("@webiny/api-core/features/task/TaskDefinition/abstractions").ITaskDefinition<import("@webiny/api-core/features/task/TaskDefinition/abstractions").ITaskInput, import("@webiny/api-core/features/task/TaskDefinition/abstractions").ITaskOutput>>;
+    __abstraction: import("@webiny/di").Abstraction<import("@webiny/api-core/features/task/TaskDefinition/abstractions.js").ITaskDefinition<import("@webiny/api-core/features/task/TaskDefinition/abstractions.js").ITaskInput, import("@webiny/api-core/features/task/TaskDefinition/abstractions.js").ITaskOutput>>;
 };
 export {};

package/api/features/AiImageEnrichment/AiImageEnrichmentTask.js

@@ -2,14 +2,15 @@ import { Output } from "ai";
 import { z } from "zod";
 import { TaskDefinition } from "@webiny/api-core/features/task/TaskDefinition/index.js";
 import { Ai } from "@webiny/api-core/features/ai/index.js";
-import { Encryption } from "@webiny/api-core/features/encryption/index.js";
+import { Logger } from "@webiny/api-core/features/logger/index.js";
 import { GetFileUseCase } from "@webiny/api-file-manager/features/file/GetFile/index.js";
 import { UpdateFileUseCase } from "@webiny/api-file-manager/features/file/UpdateFile/index.js";
 import { GetSettingsUseCase } from "@webiny/api-file-manager/features/settings/GetSettings/abstractions.js";
-import { WebsocketService } from "@webiny/api-websockets/features/WebsocketService/index.js";
-import { GetSettingsUseCase as index_js_GetSettingsUseCase } from "../GetSettings/index.js";
+import { WebsocketsSendToIdentityUseCase } from "@webiny/api-websockets/features/SendToIdentity/abstractions.js";
+import { IdentityContext } from "@webiny/api-core/exports/api/security.js";
+import { ResolveAiCapabilityUseCase, withAdditionalInstructions } from "../Capabilities/index.js";
+import { FM_IMAGE_ENRICHMENT_CAPABILITY } from "./capability.js";
 const AI_IMAGE_ENRICHMENT_TASK_ID = "fmAiImageEnrichment";
-const AI_PROMPT = "Analyze this image and return up to 5 lowercase descriptive tags and one short sentence describing the image.";
 const aiOutputSchema = Output.object({
     schema: z.object({
         tags: z.array(z.string()),
@@ -17,14 +18,15 @@ const aiOutputSchema = Output.object({
     })
 });
 class AiImageEnrichmentTaskImpl {
-    constructor(getFile, fmSettings, updateFile, ai, getSettings, encryption, websocketService){
+    constructor(getFile, fmSettings, updateFile, ai, resolveCapability, logger, identityContext, sendToIdentity){
         this.getFile = getFile;
         this.fmSettings = fmSettings;
         this.updateFile = updateFile;
         this.ai = ai;
-        this.getSettings = getSettings;
-        this.encryption = encryption;
-        this.websocketService = websocketService;
+        this.resolveCapability = resolveCapability;
+        this.logger = logger;
+        this.identityContext = identityContext;
+        this.sendToIdentity = sendToIdentity;
         this.id = AI_IMAGE_ENRICHMENT_TASK_ID;
         this.title = "File Manager - AI Image Enrichment";
         this.description = "Automatically enriches uploaded images with AI-generated tags and description.";
@@ -47,25 +49,22 @@ class AiImageEnrichmentTaskImpl {
         const settingsResult = await this.fmSettings.execute();
         const srcPrefix = settingsResult.isOk() ? settingsResult.value.srcPrefix ?? "" : "";
         const imageUrl = `${srcPrefix}${file.key}`;
-        const aiSettingsResult = await this.getSettings.execute();
-        if (aiSettingsResult.isFail()) return controller.response.error({
-            message: "No AI provider configured. Add a provider in AI Power Ups settings."
-        });
-        const aiSettings = aiSettingsResult.value;
-        const firstProvider = aiSettings.providers.presets[0];
-        if (!firstProvider) return controller.response.done({
-            message: "No AI provider configured. Add a provider in AI Power Ups settings."
-        });
+        const resolved = await this.resolveCapability.execute(FM_IMAGE_ENRICHMENT_CAPABILITY);
+        if (resolved.isFail()) {
+            this.logger.warn({
+                fileId: input.fileId,
+                reason: resolved.error.message
+            }, "Skipping AI image enrichment.");
+            return controller.response.done(resolved.error.message);
+        }
+        const capability = resolved.value;
         let tags = [];
         let description = "";
         try {
             const aiResult = await this.ai.generateText({
-                model: firstProvider.model,
+                model: capability.model,
                 output: aiOutputSchema,
-                connection: {
-                    sdkName: firstProvider.model.split("/")[0],
-                    apiKey: await this.encryption.decrypt(firstProvider.apiKeyEncrypted)
-                },
+                connection: capability.connection,
                 messages: [
                     {
                         role: "user",
@@ -76,7 +75,7 @@ class AiImageEnrichmentTaskImpl {
                             },
                             {
                                 type: "text",
-                                text: AI_PROMPT
+                                text: withAdditionalInstructions(capability)
                             }
                         ]
                     }
@@ -104,17 +103,17 @@ class AiImageEnrichmentTaskImpl {
         if (updateResult.isFail()) return controller.response.error({
             message: `Failed to update file: ${updateResult.error.message}`
         });
-        if (this.websocketService) {
-            const connectionsResult = await this.websocketService.listConnections();
-            if (connectionsResult.isOk() && connectionsResult.value.length > 0) await this.websocketService.sendToConnections(connectionsResult.value, {
-                action: "fm.file.enrichment",
-                data: {
-                    id: file.id,
-                    tags: mergedTags,
-                    description
-                }
-            });
-        }
+        const identity = this.identityContext.getIdentity();
+        await this.sendToIdentity.execute({
+            id: identity.id
+        }, {
+            action: "fm.file.enrichment",
+            data: {
+                id: file.id,
+                tags: mergedTags,
+                description
+            }
+        });
         return controller.response.done("AI image enrichment completed successfully.");
     }
 }
@@ -125,14 +124,10 @@ const AiImageEnrichmentTask = TaskDefinition.createImplementation({
         GetSettingsUseCase,
         UpdateFileUseCase,
         Ai,
-        index_js_GetSettingsUseCase,
-        Encryption,
-        [
-            WebsocketService,
-            {
-                optional: true
-            }
-        ]
+        ResolveAiCapabilityUseCase,
+        Logger,
+        IdentityContext,
+        WebsocketsSendToIdentityUseCase
     ]
 });
 export { AI_IMAGE_ENRICHMENT_TASK_ID, AiImageEnrichmentTask };

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

@@ -1 +1 @@
-{"version":3,"file":"api/features/AiImageEnrichment/AiImageEnrichmentTask.js","sources":["../../../../src/api/features/AiImageEnrichment/AiImageEnrichmentTask.ts"],"sourcesContent":["import { Output } from \"ai\";\nimport { z } from \"zod\";\nimport { TaskDefinition } from \"@webiny/api-core/features/task/TaskDefinition/index.js\";\nimport { Ai } from \"@webiny/api-core/features/ai/index.js\";\nimport { Encryption } from \"@webiny/api-core/features/encryption/index.js\";\nimport { GetFileUseCase } from \"@webiny/api-file-manager/features/file/GetFile/index.js\";\nimport { UpdateFileUseCase } from \"@webiny/api-file-manager/features/file/UpdateFile/index.js\";\nimport { GetSettingsUseCase as FmGetSettingsUseCase } from \"@webiny/api-file-manager/features/settings/GetSettings/abstractions.js\";\nimport { WebsocketService } from \"@webiny/api-websockets/features/WebsocketService/index.js\";\nimport { GetSettingsUseCase } from \"~/api/features/GetSettings/index.js\";\n\nexport const AI_IMAGE_ENRICHMENT_TASK_ID = \"fmAiImageEnrichment\";\n\nconst AI_PROMPT =\n    \"Analyze this image and return up to 5 lowercase descriptive tags and one short sentence describing the image.\";\n\nconst aiOutputSchema = Output.object({\n    schema: z.object({\n        tags: z.array(z.string()),\n        description: z.string()\n    })\n});\n\nexport interface IAiImageEnrichmentTaskInput {\n    fileId: string;\n}\n\nclass AiImageEnrichmentTaskImpl implements TaskDefinition.Interface<IAiImageEnrichmentTaskInput> {\n    id = AI_IMAGE_ENRICHMENT_TASK_ID;\n    title = \"File Manager - AI Image Enrichment\";\n    description = \"Automatically enriches uploaded images with AI-generated tags and description.\";\n    maxIterations = 1;\n    isPrivate = true;\n    databaseLogs = false;\n\n    public readonly selfCleanup = [\"onSuccess\" as const, \"onAbort\" as const];\n\n    constructor(\n        private getFile: GetFileUseCase.Interface,\n        private fmSettings: FmGetSettingsUseCase.Interface,\n        private updateFile: UpdateFileUseCase.Interface,\n        private ai: Ai.Interface,\n        private getSettings: GetSettingsUseCase.Interface,\n        private encryption: Encryption.Interface,\n        private websocketService?: WebsocketService.Interface\n    ) {}\n\n    async run({\n        input,\n        controller\n    }: TaskDefinition.RunParams<IAiImageEnrichmentTaskInput>): Promise<\n        TaskDefinition.Result<IAiImageEnrichmentTaskInput>\n    > {\n        if (controller.runtime.isAborted()) {\n            return controller.response.aborted();\n        }\n\n        const fileResult = await this.getFile.execute(input.fileId);\n        if (fileResult.isFail()) {\n            return controller.response.error({\n                message: `File not found: ${input.fileId}`\n            });\n        }\n\n        const file = fileResult.value;\n\n        if (!file.type.startsWith(\"image/\")) {\n            return controller.response.done(\"File is not an image; skipping AI enrichment.\");\n        }\n\n        const settingsResult = await this.fmSettings.execute();\n        const srcPrefix = settingsResult.isOk() ? (settingsResult.value.srcPrefix ?? \"\") : \"\";\n        const imageUrl = `${srcPrefix}${file.key}`;\n\n        const aiSettingsResult = await this.getSettings.execute();\n\n        if (aiSettingsResult.isFail()) {\n            return controller.response.error({\n                message: \"No AI provider configured. Add a provider in AI Power Ups settings.\"\n            });\n        }\n\n        const aiSettings = aiSettingsResult.value;\n\n        const firstProvider = aiSettings.providers.presets[0];\n\n        if (!firstProvider) {\n            return controller.response.done({\n                message: \"No AI provider configured. Add a provider in AI Power Ups settings.\"\n            });\n        }\n\n        let tags: string[] = [];\n        let description = \"\";\n        try {\n            const aiResult = await this.ai.generateText({\n                model: firstProvider.model,\n                output: aiOutputSchema,\n                connection: {\n                    sdkName: firstProvider.model.split(\"/\")[0],\n                    apiKey: await this.encryption.decrypt(firstProvider.apiKeyEncrypted)\n                },\n                messages: [\n                    {\n                        role: \"user\",\n                        content: [\n                            {\n                                type: \"image\",\n                                image: new URL(imageUrl)\n                            },\n                            {\n                                type: \"text\",\n                                text: AI_PROMPT\n                            }\n                        ]\n                    }\n                ]\n            });\n\n            tags = aiResult.output.tags;\n            description = aiResult.output.description;\n        } catch (error) {\n            console.log(\"error\", error.message);\n            return controller.response.error({\n                message: `AI enrichment failed: ${error instanceof Error ? error.message : String(error)}`\n            });\n        }\n\n        const mergedTags = [...new Set([...file.tags, ...tags])];\n\n        const updateResult = await this.updateFile.execute({\n            id: file.id,\n            tags: mergedTags,\n            description\n        });\n\n        if (updateResult.isFail()) {\n            return controller.response.error({\n                message: `Failed to update file: ${updateResult.error.message}`\n            });\n        }\n\n        if (this.websocketService) {\n            const connectionsResult = await this.websocketService.listConnections();\n            if (connectionsResult.isOk() && connectionsResult.value.length > 0) {\n                await this.websocketService.sendToConnections(connectionsResult.value, {\n                    action: \"fm.file.enrichment\",\n                    data: {\n                        id: file.id,\n                        tags: mergedTags,\n                        description\n                    }\n                });\n            }\n        }\n\n        return controller.response.done(\"AI image enrichment completed successfully.\");\n    }\n}\n\nexport const AiImageEnrichmentTask = TaskDefinition.createImplementation({\n    implementation: AiImageEnrichmentTaskImpl,\n    dependencies: [\n        GetFileUseCase,\n        FmGetSettingsUseCase,\n        UpdateFileUseCase,\n        Ai,\n        GetSettingsUseCase,\n        Encryption,\n        [WebsocketService, { optional: true }]\n    ]\n});\n"],"names":["AI_IMAGE_ENRICHMENT_TASK_ID","AI_PROMPT","aiOutputSchema","Output","z","AiImageEnrichmentTaskImpl","getFile","fmSettings","updateFile","ai","getSettings","encryption","websocketService","input","controller","fileResult","file","settingsResult","srcPrefix","imageUrl","aiSettingsResult","aiSettings","firstProvider","tags","description","aiResult","URL","error","console","Error","String","mergedTags","Set","updateResult","connectionsResult","AiImageEnrichmentTask","TaskDefinition","GetFileUseCase","FmGetSettingsUseCase","UpdateFileUseCase","Ai","GetSettingsUseCase","Encryption","WebsocketService"],"mappings":";;;;;;;;;;AAWO,MAAMA,8BAA8B;AAE3C,MAAMC,YACF;AAEJ,MAAMC,iBAAiBC,OAAO,MAAM,CAAC;IACjC,QAAQC,EAAE,MAAM,CAAC;QACb,MAAMA,EAAE,KAAK,CAACA,EAAE,MAAM;QACtB,aAAaA,EAAE,MAAM;IACzB;AACJ;AAMA,MAAMC;IAUF,YACYC,OAAiC,EACjCC,UAA0C,EAC1CC,UAAuC,EACvCC,EAAgB,EAChBC,WAAyC,EACzCC,UAAgC,EAChCC,gBAA6C,CACvD;aAPUN,OAAO,GAAPA;aACAC,UAAU,GAAVA;aACAC,UAAU,GAAVA;aACAC,EAAE,GAAFA;aACAC,WAAW,GAAXA;aACAC,UAAU,GAAVA;aACAC,gBAAgB,GAAhBA;aAhBZ,EAAE,GAAGZ;aACL,KAAK,GAAG;aACR,WAAW,GAAG;aACd,aAAa,GAAG;aAChB,SAAS,GAAG;aACZ,YAAY,GAAG;aAEC,WAAW,GAAG;YAAC;YAAsB;SAAmB;IAUrE;IAEH,MAAM,IAAI,EACNa,KAAK,EACLC,UAAU,EAC0C,EAEtD;QACE,IAAIA,WAAW,OAAO,CAAC,SAAS,IAC5B,OAAOA,WAAW,QAAQ,CAAC,OAAO;QAGtC,MAAMC,aAAa,MAAM,IAAI,CAAC,OAAO,CAAC,OAAO,CAACF,MAAM,MAAM;QAC1D,IAAIE,WAAW,MAAM,IACjB,OAAOD,WAAW,QAAQ,CAAC,KAAK,CAAC;YAC7B,SAAS,CAAC,gBAAgB,EAAED,MAAM,MAAM,EAAE;QAC9C;QAGJ,MAAMG,OAAOD,WAAW,KAAK;QAE7B,IAAI,CAACC,KAAK,IAAI,CAAC,UAAU,CAAC,WACtB,OAAOF,WAAW,QAAQ,CAAC,IAAI,CAAC;QAGpC,MAAMG,iBAAiB,MAAM,IAAI,CAAC,UAAU,CAAC,OAAO;QACpD,MAAMC,YAAYD,eAAe,IAAI,KAAMA,eAAe,KAAK,CAAC,SAAS,IAAI,KAAM;QACnF,MAAME,WAAW,GAAGD,YAAYF,KAAK,GAAG,EAAE;QAE1C,MAAMI,mBAAmB,MAAM,IAAI,CAAC,WAAW,CAAC,OAAO;QAEvD,IAAIA,iBAAiB,MAAM,IACvB,OAAON,WAAW,QAAQ,CAAC,KAAK,CAAC;YAC7B,SAAS;QACb;QAGJ,MAAMO,aAAaD,iBAAiB,KAAK;QAEzC,MAAME,gBAAgBD,WAAW,SAAS,CAAC,OAAO,CAAC,EAAE;QAErD,IAAI,CAACC,eACD,OAAOR,WAAW,QAAQ,CAAC,IAAI,CAAC;YAC5B,SAAS;QACb;QAGJ,IAAIS,OAAiB,EAAE;QACvB,IAAIC,cAAc;QAClB,IAAI;YACA,MAAMC,WAAW,MAAM,IAAI,CAAC,EAAE,CAAC,YAAY,CAAC;gBACxC,OAAOH,cAAc,KAAK;gBAC1B,QAAQpB;gBACR,YAAY;oBACR,SAASoB,cAAc,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE;oBAC1C,QAAQ,MAAM,IAAI,CAAC,UAAU,CAAC,OAAO,CAACA,cAAc,eAAe;gBACvE;gBACA,UAAU;oBACN;wBACI,MAAM;wBACN,SAAS;4BACL;gCACI,MAAM;gCACN,OAAO,IAAII,IAAIP;4BACnB;4BACA;gCACI,MAAM;gCACN,MAAMlB;4BACV;yBACH;oBACL;iBACH;YACL;YAEAsB,OAAOE,SAAS,MAAM,CAAC,IAAI;YAC3BD,cAAcC,SAAS,MAAM,CAAC,WAAW;QAC7C,EAAE,OAAOE,OAAO;YACZC,QAAQ,GAAG,CAAC,SAASD,MAAM,OAAO;YAClC,OAAOb,WAAW,QAAQ,CAAC,KAAK,CAAC;gBAC7B,SAAS,CAAC,sBAAsB,EAAEa,iBAAiBE,QAAQF,MAAM,OAAO,GAAGG,OAAOH,QAAQ;YAC9F;QACJ;QAEA,MAAMI,aAAa;eAAI,IAAIC,IAAI;mBAAIhB,KAAK,IAAI;mBAAKO;aAAK;SAAE;QAExD,MAAMU,eAAe,MAAM,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC;YAC/C,IAAIjB,KAAK,EAAE;YACX,MAAMe;YACNP;QACJ;QAEA,IAAIS,aAAa,MAAM,IACnB,OAAOnB,WAAW,QAAQ,CAAC,KAAK,CAAC;YAC7B,SAAS,CAAC,uBAAuB,EAAEmB,aAAa,KAAK,CAAC,OAAO,EAAE;QACnE;QAGJ,IAAI,IAAI,CAAC,gBAAgB,EAAE;YACvB,MAAMC,oBAAoB,MAAM,IAAI,CAAC,gBAAgB,CAAC,eAAe;YACrE,IAAIA,kBAAkB,IAAI,MAAMA,kBAAkB,KAAK,CAAC,MAAM,GAAG,GAC7D,MAAM,IAAI,CAAC,gBAAgB,CAAC,iBAAiB,CAACA,kBAAkB,KAAK,EAAE;gBACnE,QAAQ;gBACR,MAAM;oBACF,IAAIlB,KAAK,EAAE;oBACX,MAAMe;oBACNP;gBACJ;YACJ;QAER;QAEA,OAAOV,WAAW,QAAQ,CAAC,IAAI,CAAC;IACpC;AACJ;AAEO,MAAMqB,wBAAwBC,eAAe,oBAAoB,CAAC;IACrE,gBAAgB/B;IAChB,cAAc;QACVgC;QACAC;QACAC;QACAC;QACAC;QACAC;QACA;YAACC;YAAkB;gBAAE,UAAU;YAAK;SAAE;KACzC;AACL"}
+{"version":3,"file":"api/features/AiImageEnrichment/AiImageEnrichmentTask.js","sources":["../../../../src/api/features/AiImageEnrichment/AiImageEnrichmentTask.ts"],"sourcesContent":["import { Output } from \"ai\";\nimport { z } from \"zod\";\nimport { TaskDefinition } from \"@webiny/api-core/features/task/TaskDefinition/index.js\";\nimport { Ai } from \"@webiny/api-core/features/ai/index.js\";\nimport { Logger } from \"@webiny/api-core/features/logger/index.js\";\nimport { GetFileUseCase } from \"@webiny/api-file-manager/features/file/GetFile/index.js\";\nimport { UpdateFileUseCase } from \"@webiny/api-file-manager/features/file/UpdateFile/index.js\";\nimport { GetSettingsUseCase as FmGetSettingsUseCase } from \"@webiny/api-file-manager/features/settings/GetSettings/abstractions.js\";\nimport { WebsocketsSendToIdentityUseCase } from \"@webiny/api-websockets/features/SendToIdentity/abstractions.js\";\nimport { IdentityContext } from \"@webiny/api-core/exports/api/security.js\";\nimport {\n    ResolveAiCapabilityUseCase,\n    withAdditionalInstructions\n} from \"~/api/features/Capabilities/index.js\";\nimport { FM_IMAGE_ENRICHMENT_CAPABILITY } from \"./capability.js\";\n\nexport const AI_IMAGE_ENRICHMENT_TASK_ID = \"fmAiImageEnrichment\";\n\nconst aiOutputSchema = Output.object({\n    schema: z.object({\n        tags: z.array(z.string()),\n        description: z.string()\n    })\n});\n\nexport interface IAiImageEnrichmentTaskInput {\n    fileId: string;\n}\n\nclass AiImageEnrichmentTaskImpl implements TaskDefinition.Interface<IAiImageEnrichmentTaskInput> {\n    id = AI_IMAGE_ENRICHMENT_TASK_ID;\n    title = \"File Manager - AI Image Enrichment\";\n    description = \"Automatically enriches uploaded images with AI-generated tags and description.\";\n    maxIterations = 1;\n    isPrivate = true;\n    databaseLogs = false;\n\n    public readonly selfCleanup = [\"onSuccess\" as const, \"onAbort\" as const];\n\n    constructor(\n        private getFile: GetFileUseCase.Interface,\n        private fmSettings: FmGetSettingsUseCase.Interface,\n        private updateFile: UpdateFileUseCase.Interface,\n        private ai: Ai.Interface,\n        private resolveCapability: ResolveAiCapabilityUseCase.Interface,\n        private logger: Logger.Interface,\n        private identityContext: IdentityContext.Interface,\n        private sendToIdentity: WebsocketsSendToIdentityUseCase.Interface\n    ) {}\n\n    async run({\n        input,\n        controller\n    }: TaskDefinition.RunParams<IAiImageEnrichmentTaskInput>): Promise<\n        TaskDefinition.Result<IAiImageEnrichmentTaskInput>\n    > {\n        if (controller.runtime.isAborted()) {\n            return controller.response.aborted();\n        }\n\n        const fileResult = await this.getFile.execute(input.fileId);\n        if (fileResult.isFail()) {\n            return controller.response.error({\n                message: `File not found: ${input.fileId}`\n            });\n        }\n\n        const file = fileResult.value;\n\n        if (!file.type.startsWith(\"image/\")) {\n            return controller.response.done(\"File is not an image; skipping AI enrichment.\");\n        }\n\n        const settingsResult = await this.fmSettings.execute();\n        const srcPrefix = settingsResult.isOk() ? (settingsResult.value.srcPrefix ?? \"\") : \"\";\n        const imageUrl = `${srcPrefix}${file.key}`;\n\n        const resolved = await this.resolveCapability.execute(FM_IMAGE_ENRICHMENT_CAPABILITY);\n\n        if (resolved.isFail()) {\n            /*\n             * A skip, not a failure. This feature is not licence-gated on this branch, so every\n             * project that uploads an image without AI configured would otherwise get a failed task\n             * on every upload. Today's behaviour for an unconfigured provider is the same soft\n             * done; the log is new, because a skip and a misconfiguration used to look identical.\n             */\n            this.logger.warn(\n                { fileId: input.fileId, reason: resolved.error.message },\n                \"Skipping AI image enrichment.\"\n            );\n            return controller.response.done(resolved.error.message);\n        }\n\n        const capability = resolved.value;\n\n        let tags: string[] = [];\n        let description = \"\";\n        try {\n            const aiResult = await this.ai.generateText({\n                model: capability.model,\n                output: aiOutputSchema,\n                connection: capability.connection,\n                messages: [\n                    {\n                        role: \"user\",\n                        content: [\n                            {\n                                type: \"image\",\n                                image: new URL(imageUrl)\n                            },\n                            {\n                                type: \"text\",\n                                text: withAdditionalInstructions(capability)\n                            }\n                        ]\n                    }\n                ]\n            });\n\n            tags = aiResult.output.tags;\n            description = aiResult.output.description;\n        } catch (error) {\n            console.log(\"error\", error.message);\n            return controller.response.error({\n                message: `AI enrichment failed: ${error instanceof Error ? error.message : String(error)}`\n            });\n        }\n\n        const mergedTags = [...new Set([...file.tags, ...tags])];\n\n        const updateResult = await this.updateFile.execute({\n            id: file.id,\n            tags: mergedTags,\n            description\n        });\n\n        if (updateResult.isFail()) {\n            return controller.response.error({\n                message: `Failed to update file: ${updateResult.error.message}`\n            });\n        }\n\n        const identity = this.identityContext.getIdentity();\n        await this.sendToIdentity.execute(\n            { id: identity.id },\n            {\n                action: \"fm.file.enrichment\",\n                data: {\n                    id: file.id,\n                    tags: mergedTags,\n                    description\n                }\n            }\n        );\n\n        return controller.response.done(\"AI image enrichment completed successfully.\");\n    }\n}\n\nexport const AiImageEnrichmentTask = TaskDefinition.createImplementation({\n    implementation: AiImageEnrichmentTaskImpl,\n    dependencies: [\n        GetFileUseCase,\n        FmGetSettingsUseCase,\n        UpdateFileUseCase,\n        Ai,\n        ResolveAiCapabilityUseCase,\n        Logger,\n        IdentityContext,\n        WebsocketsSendToIdentityUseCase\n    ]\n});\n"],"names":["AI_IMAGE_ENRICHMENT_TASK_ID","aiOutputSchema","Output","z","AiImageEnrichmentTaskImpl","getFile","fmSettings","updateFile","ai","resolveCapability","logger","identityContext","sendToIdentity","input","controller","fileResult","file","settingsResult","srcPrefix","imageUrl","resolved","FM_IMAGE_ENRICHMENT_CAPABILITY","capability","tags","description","aiResult","URL","withAdditionalInstructions","error","console","Error","String","mergedTags","Set","updateResult","identity","AiImageEnrichmentTask","TaskDefinition","GetFileUseCase","FmGetSettingsUseCase","UpdateFileUseCase","Ai","ResolveAiCapabilityUseCase","Logger","IdentityContext","WebsocketsSendToIdentityUseCase"],"mappings":";;;;;;;;;;;;AAgBO,MAAMA,8BAA8B;AAE3C,MAAMC,iBAAiBC,OAAO,MAAM,CAAC;IACjC,QAAQC,EAAE,MAAM,CAAC;QACb,MAAMA,EAAE,KAAK,CAACA,EAAE,MAAM;QACtB,aAAaA,EAAE,MAAM;IACzB;AACJ;AAMA,MAAMC;IAUF,YACYC,OAAiC,EACjCC,UAA0C,EAC1CC,UAAuC,EACvCC,EAAgB,EAChBC,iBAAuD,EACvDC,MAAwB,EACxBC,eAA0C,EAC1CC,cAAyD,CACnE;aARUP,OAAO,GAAPA;aACAC,UAAU,GAAVA;aACAC,UAAU,GAAVA;aACAC,EAAE,GAAFA;aACAC,iBAAiB,GAAjBA;aACAC,MAAM,GAANA;aACAC,eAAe,GAAfA;aACAC,cAAc,GAAdA;aAjBZ,EAAE,GAAGZ;aACL,KAAK,GAAG;aACR,WAAW,GAAG;aACd,aAAa,GAAG;aAChB,SAAS,GAAG;aACZ,YAAY,GAAG;aAEC,WAAW,GAAG;YAAC;YAAsB;SAAmB;IAWrE;IAEH,MAAM,IAAI,EACNa,KAAK,EACLC,UAAU,EAC0C,EAEtD;QACE,IAAIA,WAAW,OAAO,CAAC,SAAS,IAC5B,OAAOA,WAAW,QAAQ,CAAC,OAAO;QAGtC,MAAMC,aAAa,MAAM,IAAI,CAAC,OAAO,CAAC,OAAO,CAACF,MAAM,MAAM;QAC1D,IAAIE,WAAW,MAAM,IACjB,OAAOD,WAAW,QAAQ,CAAC,KAAK,CAAC;YAC7B,SAAS,CAAC,gBAAgB,EAAED,MAAM,MAAM,EAAE;QAC9C;QAGJ,MAAMG,OAAOD,WAAW,KAAK;QAE7B,IAAI,CAACC,KAAK,IAAI,CAAC,UAAU,CAAC,WACtB,OAAOF,WAAW,QAAQ,CAAC,IAAI,CAAC;QAGpC,MAAMG,iBAAiB,MAAM,IAAI,CAAC,UAAU,CAAC,OAAO;QACpD,MAAMC,YAAYD,eAAe,IAAI,KAAMA,eAAe,KAAK,CAAC,SAAS,IAAI,KAAM;QACnF,MAAME,WAAW,GAAGD,YAAYF,KAAK,GAAG,EAAE;QAE1C,MAAMI,WAAW,MAAM,IAAI,CAAC,iBAAiB,CAAC,OAAO,CAACC;QAEtD,IAAID,SAAS,MAAM,IAAI;YAOnB,IAAI,CAAC,MAAM,CAAC,IAAI,CACZ;gBAAE,QAAQP,MAAM,MAAM;gBAAE,QAAQO,SAAS,KAAK,CAAC,OAAO;YAAC,GACvD;YAEJ,OAAON,WAAW,QAAQ,CAAC,IAAI,CAACM,SAAS,KAAK,CAAC,OAAO;QAC1D;QAEA,MAAME,aAAaF,SAAS,KAAK;QAEjC,IAAIG,OAAiB,EAAE;QACvB,IAAIC,cAAc;QAClB,IAAI;YACA,MAAMC,WAAW,MAAM,IAAI,CAAC,EAAE,CAAC,YAAY,CAAC;gBACxC,OAAOH,WAAW,KAAK;gBACvB,QAAQrB;gBACR,YAAYqB,WAAW,UAAU;gBACjC,UAAU;oBACN;wBACI,MAAM;wBACN,SAAS;4BACL;gCACI,MAAM;gCACN,OAAO,IAAII,IAAIP;4BACnB;4BACA;gCACI,MAAM;gCACN,MAAMQ,2BAA2BL;4BACrC;yBACH;oBACL;iBACH;YACL;YAEAC,OAAOE,SAAS,MAAM,CAAC,IAAI;YAC3BD,cAAcC,SAAS,MAAM,CAAC,WAAW;QAC7C,EAAE,OAAOG,OAAO;YACZC,QAAQ,GAAG,CAAC,SAASD,MAAM,OAAO;YAClC,OAAOd,WAAW,QAAQ,CAAC,KAAK,CAAC;gBAC7B,SAAS,CAAC,sBAAsB,EAAEc,iBAAiBE,QAAQF,MAAM,OAAO,GAAGG,OAAOH,QAAQ;YAC9F;QACJ;QAEA,MAAMI,aAAa;eAAI,IAAIC,IAAI;mBAAIjB,KAAK,IAAI;mBAAKO;aAAK;SAAE;QAExD,MAAMW,eAAe,MAAM,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC;YAC/C,IAAIlB,KAAK,EAAE;YACX,MAAMgB;YACNR;QACJ;QAEA,IAAIU,aAAa,MAAM,IACnB,OAAOpB,WAAW,QAAQ,CAAC,KAAK,CAAC;YAC7B,SAAS,CAAC,uBAAuB,EAAEoB,aAAa,KAAK,CAAC,OAAO,EAAE;QACnE;QAGJ,MAAMC,WAAW,IAAI,CAAC,eAAe,CAAC,WAAW;QACjD,MAAM,IAAI,CAAC,cAAc,CAAC,OAAO,CAC7B;YAAE,IAAIA,SAAS,EAAE;QAAC,GAClB;YACI,QAAQ;YACR,MAAM;gBACF,IAAInB,KAAK,EAAE;gBACX,MAAMgB;gBACNR;YACJ;QACJ;QAGJ,OAAOV,WAAW,QAAQ,CAAC,IAAI,CAAC;IACpC;AACJ;AAEO,MAAMsB,wBAAwBC,eAAe,oBAAoB,CAAC;IACrE,gBAAgBjC;IAChB,cAAc;QACVkC;QACAC;QACAC;QACAC;QACAC;QACAC;QACAC;QACAC;KACH;AACL"}

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

@@ -0,0 +1,13 @@
+import { AiCapability } from "../../../api/features/Capabilities/index.js";
+export declare const FM_IMAGE_ENRICHMENT_CAPABILITY = "fm.imageEnrichment";
+declare class FmImageEnrichmentCapabilityImpl implements AiCapability.Interface {
+    readonly id = "fm.imageEnrichment";
+    readonly label = "Image enrichment";
+    readonly description = "Reads an uploaded image and writes its alt text, title and tags. Needs a model that accepts image input.";
+    readonly defaultRole: "vision";
+    readonly guidance = "Analyze this image and return up to 5 lowercase descriptive tags and one short sentence describing the image.";
+}
+export declare const FmImageEnrichmentCapability: typeof FmImageEnrichmentCapabilityImpl & {
+    __abstraction: import("@webiny/di").Abstraction<import("~/api/features/Capabilities/index.js").IAiCapability>;
+};
+export {};

package/api/features/AiImageEnrichment/capability.js

@@ -0,0 +1,19 @@
+import { AiCapability } from "../Capabilities/index.js";
+const FM_IMAGE_ENRICHMENT_CAPABILITY = "fm.imageEnrichment";
+const guidance = "Analyze this image and return up to 5 lowercase descriptive tags and one short sentence describing the image.";
+class FmImageEnrichmentCapabilityImpl {
+    constructor(){
+        this.id = FM_IMAGE_ENRICHMENT_CAPABILITY;
+        this.label = "Image enrichment";
+        this.description = "Reads an uploaded image and writes its alt text, title and tags. Needs a model that accepts image input.";
+        this.defaultRole = "vision";
+        this.guidance = guidance;
+    }
+}
+const FmImageEnrichmentCapability = AiCapability.createImplementation({
+    implementation: FmImageEnrichmentCapabilityImpl,
+    dependencies: []
+});
+export { FM_IMAGE_ENRICHMENT_CAPABILITY, FmImageEnrichmentCapability };
+
+//# sourceMappingURL=capability.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"api/features/AiImageEnrichment/capability.js","sources":["../../../../src/api/features/AiImageEnrichment/capability.ts"],"sourcesContent":["import { AiCapability } from \"~/api/features/Capabilities/index.js\";\n\nexport const FM_IMAGE_ENRICHMENT_CAPABILITY = \"fm.imageEnrichment\";\n\n/**\n * The prompt lives here rather than in the task now, because it is the capability's, and the\n * resolver hands it back with the project's additional instructions already appended.\n *\n * This is the one AI prompt whose structure does not come from the prose: `Output.object(...)` in\n * the task enforces the tags/description shape. House tag vocabularies go in additional\n * instructions (\"use our taxonomy\", \"no more than three tags\", \"British spelling\").\n */\nconst guidance =\n    \"Analyze this image and return up to 5 lowercase descriptive tags and one short sentence describing the image.\";\n\nclass FmImageEnrichmentCapabilityImpl implements AiCapability.Interface {\n    readonly id = FM_IMAGE_ENRICHMENT_CAPABILITY;\n    readonly label = \"Image enrichment\";\n    readonly description =\n        \"Reads an uploaded image and writes its alt text, title and tags. Needs a model that accepts image input.\";\n    readonly defaultRole = \"vision\" as const;\n    readonly guidance = guidance;\n}\n\nexport const FmImageEnrichmentCapability = AiCapability.createImplementation({\n    implementation: FmImageEnrichmentCapabilityImpl,\n    dependencies: []\n});\n"],"names":["FM_IMAGE_ENRICHMENT_CAPABILITY","guidance","FmImageEnrichmentCapabilityImpl","FmImageEnrichmentCapability","AiCapability"],"mappings":";AAEO,MAAMA,iCAAiC;AAU9C,MAAMC,WACF;AAEJ,MAAMC;;aACO,EAAE,GAAGF;aACL,KAAK,GAAG;aACR,WAAW,GAChB;aACK,WAAW,GAAG;aACd,QAAQ,GAAGC;;AACxB;AAEO,MAAME,8BAA8BC,aAAa,oBAAoB,CAAC;IACzE,gBAAgBF;IAChB,cAAc,EAAE;AACpB"}

package/api/features/AiImageEnrichment/feature.js

@@ -1,9 +1,11 @@
 import { createFeature } from "@webiny/feature/api";
 import { AiImageEnrichmentAfterCreateHandler } from "./AiImageEnrichmentAfterCreateHandler.js";
 import { AiImageEnrichmentTask } from "./AiImageEnrichmentTask.js";
+import { FmImageEnrichmentCapability } from "./capability.js";
 const AiImageEnrichmentFeature = createFeature({
     name: "AiPowerUps/AiImageEnrichment",
     register (container) {
+        container.register(FmImageEnrichmentCapability);
         container.register(AiImageEnrichmentAfterCreateHandler);
         container.register(AiImageEnrichmentTask);
     }

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

@@ -1 +1 @@
-{"version":3,"file":"api/features/AiImageEnrichment/feature.js","sources":["../../../../src/api/features/AiImageEnrichment/feature.ts"],"sourcesContent":["import { createFeature } from \"@webiny/feature/api\";\nimport { AiImageEnrichmentAfterCreateHandler } from \"./AiImageEnrichmentAfterCreateHandler.js\";\nimport { AiImageEnrichmentTask } from \"./AiImageEnrichmentTask.js\";\n\nexport const AiImageEnrichmentFeature = createFeature({\n    name: \"AiPowerUps/AiImageEnrichment\",\n    register(container) {\n        container.register(AiImageEnrichmentAfterCreateHandler);\n        container.register(AiImageEnrichmentTask);\n    }\n});\n"],"names":["AiImageEnrichmentFeature","createFeature","container","AiImageEnrichmentAfterCreateHandler","AiImageEnrichmentTask"],"mappings":";;;AAIO,MAAMA,2BAA2BC,cAAc;IAClD,MAAM;IACN,UAASC,SAAS;QACdA,UAAU,QAAQ,CAACC;QACnBD,UAAU,QAAQ,CAACE;IACvB;AACJ"}
+{"version":3,"file":"api/features/AiImageEnrichment/feature.js","sources":["../../../../src/api/features/AiImageEnrichment/feature.ts"],"sourcesContent":["import { createFeature } from \"@webiny/feature/api\";\nimport { AiImageEnrichmentAfterCreateHandler } from \"./AiImageEnrichmentAfterCreateHandler.js\";\nimport { AiImageEnrichmentTask } from \"./AiImageEnrichmentTask.js\";\nimport { FmImageEnrichmentCapability } from \"./capability.js\";\n\nexport const AiImageEnrichmentFeature = createFeature({\n    name: \"AiPowerUps/AiImageEnrichment\",\n    register(container) {\n        container.register(FmImageEnrichmentCapability);\n        container.register(AiImageEnrichmentAfterCreateHandler);\n        container.register(AiImageEnrichmentTask);\n    }\n});\n"],"names":["AiImageEnrichmentFeature","createFeature","container","FmImageEnrichmentCapability","AiImageEnrichmentAfterCreateHandler","AiImageEnrichmentTask"],"mappings":";;;;AAKO,MAAMA,2BAA2BC,cAAc;IAClD,MAAM;IACN,UAASC,SAAS;QACdA,UAAU,QAAQ,CAACC;QACnBD,UAAU,QAAQ,CAACE;QACnBF,UAAU,QAAQ,CAACG;IACvB;AACJ"}

package/api/features/AiPromptContext/AiPromptContextBuilder.d.ts

@@ -1,10 +1,17 @@
 import { GetSettingsUseCase } from "../../../api/features/GetSettings/index.js";
+import { GetFileContentsByIdUseCase } from "@webiny/api-file-manager/features/file/GetFileContentsById/index.js";
+import { GetFileUseCase } from "@webiny/api-file-manager/features/file/GetFile/index.js";
+import { TextExtractor } from "@webiny/api-core/features/ai/index.js";
 import { AiPromptContextBuilder as Abstraction, ProjectFileAssembler, type AiPromptContext, type AiPromptContextParams } from "./abstractions.js";
 declare class AiPromptContextBuilderImpl implements Abstraction.Interface {
     private getSettings;
     private fileAssembler;
-    constructor(getSettings: GetSettingsUseCase.Interface, fileAssembler: ProjectFileAssembler.Interface);
+    private getFileContents;
+    private getFile;
+    private textExtractor;
+    constructor(getSettings: GetSettingsUseCase.Interface, fileAssembler: ProjectFileAssembler.Interface, getFileContents: GetFileContentsByIdUseCase.Interface, getFile: GetFileUseCase.Interface, textExtractor: TextExtractor.Interface);
     execute(params: AiPromptContextParams): Promise<AiPromptContext>;
+    private fetchAdditionalFiles;
 }
 export declare const AiPromptContextBuilder: typeof AiPromptContextBuilderImpl & {
     __abstraction: import("@webiny/di").Abstraction<import("./abstractions.js").IAiPromptContextBuilder>;

package/api/features/AiPromptContext/AiPromptContextBuilder.js

@@ -1,4 +1,7 @@
 import { GetSettingsUseCase } from "../GetSettings/index.js";
+import { GetFileContentsByIdUseCase } from "@webiny/api-file-manager/features/file/GetFileContentsById/index.js";
+import { GetFileUseCase } from "@webiny/api-file-manager/features/file/GetFile/index.js";
+import { TextExtractor } from "@webiny/api-core/features/ai/index.js";
 import { AiPromptContextBuilder, ProjectFileAssembler } from "./abstractions.js";
 import { WriterPersonaSection } from "./WriterPersonaSection.js";
 import { ReaderPersonaSection } from "./ReaderPersonaSection.js";
@@ -37,6 +40,7 @@ function emptyContext(warnings) {
         readerPersona: void 0,
         writerPersona: void 0,
         allProjectFiles: [],
+        additionalFiles: [],
         excludedFileIds: new Set(),
         cacheHit: false,
         warnings,
@@ -46,9 +50,12 @@ function emptyContext(warnings) {
     };
 }
 class AiPromptContextBuilderImpl {
-    constructor(getSettings, fileAssembler){
+    constructor(getSettings, fileAssembler, getFileContents, getFile, textExtractor){
         this.getSettings = getSettings;
         this.fileAssembler = fileAssembler;
+        this.getFileContents = getFileContents;
+        this.getFile = getFile;
+        this.textExtractor = textExtractor;
     }
     async execute(params) {
         const warnings = [];
@@ -76,11 +83,17 @@ class AiPromptContextBuilderImpl {
                 totalTokens: visibleFiles.reduce((sum, f)=>sum + f.tokenCount, 0)
             };
         }
+        const additionalFiles = await this.fetchAdditionalFiles(params.additionalFileIds, warnings);
+        if (additionalFiles.length > 0) allProjectFiles = [
+            ...allProjectFiles,
+            ...additionalFiles
+        ];
         const ctx = {
             project,
             readerPersona: reader,
             writerPersona: writer,
             allProjectFiles,
+            additionalFiles,
             excludedFileIds,
             cacheHit,
             warnings,
@@ -90,12 +103,43 @@ class AiPromptContextBuilderImpl {
         };
         return ctx;
     }
+    async fetchAdditionalFiles(fileIds, warnings) {
+        if (!fileIds || 0 === fileIds.length) return [];
+        const results = await Promise.all(fileIds.map(async (fileId)=>{
+            const [contentsResult, damResult] = await Promise.all([
+                this.getFileContents.execute(fileId),
+                this.getFile.execute(fileId)
+            ]);
+            if (contentsResult.isFail()) {
+                warnings.push(`Failed to load additional file (${fileId}): ${contentsResult.error.message}`);
+                return null;
+            }
+            const name = damResult.isOk() ? damResult.value.name : fileId;
+            const mimeType = damResult.isOk() ? damResult.value.type : "application/octet-stream";
+            if (!this.textExtractor.canExtract(mimeType)) {
+                warnings.push(`Skipped additional file "${name}" (${fileId}): unsupported type "${mimeType}".`);
+                return null;
+            }
+            const { text: content } = await this.textExtractor.extract(contentsResult.value.buffer, mimeType);
+            return {
+                id: fileId,
+                name,
+                content,
+                description: damResult.isOk() ? damResult.value.description : void 0,
+                tokenCount: Math.ceil(content.length / 4)
+            };
+        }));
+        return results.filter((r)=>null !== r);
+    }
 }
 const AiPromptContextBuilder_AiPromptContextBuilder = AiPromptContextBuilder.createImplementation({
     implementation: AiPromptContextBuilderImpl,
     dependencies: [
         GetSettingsUseCase,
-        ProjectFileAssembler
+        ProjectFileAssembler,
+        GetFileContentsByIdUseCase,
+        GetFileUseCase,
+        TextExtractor
     ]
 });
 export { AiPromptContextBuilder_AiPromptContextBuilder as AiPromptContextBuilder };

package/api/features/AiPromptContext/AiPromptContextBuilder.js.map

@@ -1 +1 @@
-{"version":3,"file":"api/features/AiPromptContext/AiPromptContextBuilder.js","sources":["../../../../src/api/features/AiPromptContext/AiPromptContextBuilder.ts"],"sourcesContent":["import { GetSettingsUseCase } from \"~/api/features/GetSettings/index.js\";\nimport {\n    AiPromptContextBuilder as Abstraction,\n    ProjectFileAssembler,\n    type AiPromptContext,\n    type AiPromptContextParams,\n    type ProjectFileContent,\n    type ResolvedPersona,\n    type ResolvedProject\n} from \"./abstractions.js\";\nimport type { IAiPowerUpsSettings } from \"~/api/types.js\";\nimport type { ProjectPreset } from \"~/api/features/Projects/types.js\";\nimport { WriterPersonaSection } from \"./WriterPersonaSection.js\";\nimport { ReaderPersonaSection } from \"./ReaderPersonaSection.js\";\nimport { ProjectSection } from \"./ProjectSection.js\";\n\nfunction resolvePersonas(\n    settings: IAiPowerUpsSettings,\n    project: ProjectPreset | undefined,\n    params: AiPromptContextParams\n): { reader?: ResolvedPersona; writer?: ResolvedPersona } {\n    const readerPersonaId = params.readerPersonaId ?? project?.defaultReaderPersonaId ?? undefined;\n    const writerPersonaId = params.writerPersonaId ?? project?.defaultWriterPersonaId ?? undefined;\n\n    const readerPreset = readerPersonaId\n        ? settings.readerPersonas?.presets?.find(p => p.id === readerPersonaId)\n        : undefined;\n\n    const writerPreset = writerPersonaId\n        ? settings.writerPersonas?.presets?.find(p => p.id === writerPersonaId)\n        : undefined;\n\n    return {\n        reader: readerPreset\n            ? {\n                  name: readerPreset.name,\n                  description: readerPreset.description,\n                  style: readerPreset.style\n              }\n            : undefined,\n        writer: writerPreset\n            ? {\n                  name: writerPreset.name,\n                  description: writerPreset.description,\n                  style: writerPreset.style\n              }\n            : undefined\n    };\n}\n\nfunction formatContext(ctx: AiPromptContext): string {\n    const sections: string[] = [];\n\n    if (ctx.writerPersona) {\n        sections.push(WriterPersonaSection.format(ctx.writerPersona));\n    }\n    if (ctx.readerPersona) {\n        sections.push(ReaderPersonaSection.format(ctx.readerPersona));\n    }\n    if (ctx.project) {\n        const projectText = ProjectSection.format(ctx.project);\n        if (projectText) {\n            sections.push(projectText);\n        }\n    }\n\n    return sections.length > 0 ? \"\\n\\n\" + sections.join(\"\\n\\n\") : \"\";\n}\n\nfunction emptyContext(warnings: string[]): AiPromptContext {\n    return {\n        project: undefined,\n        readerPersona: undefined,\n        writerPersona: undefined,\n        allProjectFiles: [],\n        excludedFileIds: new Set(),\n        cacheHit: false,\n        warnings,\n        toString() {\n            return \"\";\n        }\n    };\n}\n\nclass AiPromptContextBuilderImpl implements Abstraction.Interface {\n    constructor(\n        private getSettings: GetSettingsUseCase.Interface,\n        private fileAssembler: ProjectFileAssembler.Interface\n    ) {}\n\n    async execute(params: AiPromptContextParams): Promise<AiPromptContext> {\n        const warnings: string[] = [];\n\n        const settingsResult = await this.getSettings.execute();\n        if (settingsResult.isFail()) {\n            warnings.push(\"Failed to load AI PowerUps settings.\");\n            return emptyContext(warnings);\n        }\n\n        const settings = settingsResult.value;\n        const projectPreset = params.projectId\n            ? settings.projects?.presets?.find(p => p.id === params.projectId)\n            : undefined;\n\n        const { reader, writer } = resolvePersonas(settings, projectPreset, params);\n        const excludedFileIds = new Set(params.excludedFileIds ?? []);\n\n        let project: ResolvedProject | undefined;\n        let allProjectFiles: ProjectFileContent[] = [];\n        let cacheHit = false;\n\n        if (projectPreset) {\n            const assembled = await this.fileAssembler.execute(\n                projectPreset.id,\n                projectPreset.version ?? 0,\n                projectPreset.files ?? [],\n                warnings\n            );\n            allProjectFiles = assembled.files;\n            cacheHit = assembled.cacheHit;\n\n            const visibleFiles =\n                excludedFileIds.size > 0\n                    ? allProjectFiles.filter(f => !excludedFileIds.has(f.id))\n                    : allProjectFiles;\n\n            project = {\n                name: projectPreset.name,\n                instructions: projectPreset.instructions,\n                files: visibleFiles,\n                totalTokens: visibleFiles.reduce((sum, f) => sum + f.tokenCount, 0)\n            };\n        }\n\n        const ctx: AiPromptContext = {\n            project,\n            readerPersona: reader,\n            writerPersona: writer,\n            allProjectFiles,\n            excludedFileIds,\n            cacheHit,\n            warnings,\n            toString() {\n                return formatContext(this);\n            }\n        };\n\n        return ctx;\n    }\n}\n\nexport const AiPromptContextBuilder = Abstraction.createImplementation({\n    implementation: AiPromptContextBuilderImpl,\n    dependencies: [GetSettingsUseCase, ProjectFileAssembler]\n});\n"],"names":["resolvePersonas","settings","project","params","readerPersonaId","undefined","writerPersonaId","readerPreset","p","writerPreset","formatContext","ctx","sections","WriterPersonaSection","ReaderPersonaSection","projectText","ProjectSection","emptyContext","warnings","Set","AiPromptContextBuilderImpl","getSettings","fileAssembler","settingsResult","projectPreset","reader","writer","excludedFileIds","allProjectFiles","cacheHit","assembled","visibleFiles","f","sum","AiPromptContextBuilder","Abstraction","GetSettingsUseCase","ProjectFileAssembler"],"mappings":";;;;;AAgBA,SAASA,gBACLC,QAA6B,EAC7BC,OAAkC,EAClCC,MAA6B;IAE7B,MAAMC,kBAAkBD,OAAO,eAAe,IAAID,SAAS,0BAA0BG;IACrF,MAAMC,kBAAkBH,OAAO,eAAe,IAAID,SAAS,0BAA0BG;IAErF,MAAME,eAAeH,kBACfH,SAAS,cAAc,EAAE,SAAS,KAAKO,CAAAA,IAAKA,EAAE,EAAE,KAAKJ,mBACrDC;IAEN,MAAMI,eAAeH,kBACfL,SAAS,cAAc,EAAE,SAAS,KAAKO,CAAAA,IAAKA,EAAE,EAAE,KAAKF,mBACrDD;IAEN,OAAO;QACH,QAAQE,eACF;YACI,MAAMA,aAAa,IAAI;YACvB,aAAaA,aAAa,WAAW;YACrC,OAAOA,aAAa,KAAK;QAC7B,IACAF;QACN,QAAQI,eACF;YACI,MAAMA,aAAa,IAAI;YACvB,aAAaA,aAAa,WAAW;YACrC,OAAOA,aAAa,KAAK;QAC7B,IACAJ;IACV;AACJ;AAEA,SAASK,cAAcC,GAAoB;IACvC,MAAMC,WAAqB,EAAE;IAE7B,IAAID,IAAI,aAAa,EACjBC,SAAS,IAAI,CAACC,qBAAqB,MAAM,CAACF,IAAI,aAAa;IAE/D,IAAIA,IAAI,aAAa,EACjBC,SAAS,IAAI,CAACE,qBAAqB,MAAM,CAACH,IAAI,aAAa;IAE/D,IAAIA,IAAI,OAAO,EAAE;QACb,MAAMI,cAAcC,eAAe,MAAM,CAACL,IAAI,OAAO;QACrD,IAAII,aACAH,SAAS,IAAI,CAACG;IAEtB;IAEA,OAAOH,SAAS,MAAM,GAAG,IAAI,SAASA,SAAS,IAAI,CAAC,UAAU;AAClE;AAEA,SAASK,aAAaC,QAAkB;IACpC,OAAO;QACH,SAASb;QACT,eAAeA;QACf,eAAeA;QACf,iBAAiB,EAAE;QACnB,iBAAiB,IAAIc;QACrB,UAAU;QACVD;QACA;YACI,OAAO;QACX;IACJ;AACJ;AAEA,MAAME;IACF,YACYC,WAAyC,EACzCC,aAA6C,CACvD;aAFUD,WAAW,GAAXA;aACAC,aAAa,GAAbA;IACT;IAEH,MAAM,QAAQnB,MAA6B,EAA4B;QACnE,MAAMe,WAAqB,EAAE;QAE7B,MAAMK,iBAAiB,MAAM,IAAI,CAAC,WAAW,CAAC,OAAO;QACrD,IAAIA,eAAe,MAAM,IAAI;YACzBL,SAAS,IAAI,CAAC;YACd,OAAOD,aAAaC;QACxB;QAEA,MAAMjB,WAAWsB,eAAe,KAAK;QACrC,MAAMC,gBAAgBrB,OAAO,SAAS,GAChCF,SAAS,QAAQ,EAAE,SAAS,KAAKO,CAAAA,IAAKA,EAAE,EAAE,KAAKL,OAAO,SAAS,IAC/DE;QAEN,MAAM,EAAEoB,MAAM,EAAEC,MAAM,EAAE,GAAG1B,gBAAgBC,UAAUuB,eAAerB;QACpE,MAAMwB,kBAAkB,IAAIR,IAAIhB,OAAO,eAAe,IAAI,EAAE;QAE5D,IAAID;QACJ,IAAI0B,kBAAwC,EAAE;QAC9C,IAAIC,WAAW;QAEf,IAAIL,eAAe;YACf,MAAMM,YAAY,MAAM,IAAI,CAAC,aAAa,CAAC,OAAO,CAC9CN,cAAc,EAAE,EAChBA,cAAc,OAAO,IAAI,GACzBA,cAAc,KAAK,IAAI,EAAE,EACzBN;YAEJU,kBAAkBE,UAAU,KAAK;YACjCD,WAAWC,UAAU,QAAQ;YAE7B,MAAMC,eACFJ,gBAAgB,IAAI,GAAG,IACjBC,gBAAgB,MAAM,CAACI,CAAAA,IAAK,CAACL,gBAAgB,GAAG,CAACK,EAAE,EAAE,KACrDJ;YAEV1B,UAAU;gBACN,MAAMsB,cAAc,IAAI;gBACxB,cAAcA,cAAc,YAAY;gBACxC,OAAOO;gBACP,aAAaA,aAAa,MAAM,CAAC,CAACE,KAAKD,IAAMC,MAAMD,EAAE,UAAU,EAAE;YACrE;QACJ;QAEA,MAAMrB,MAAuB;YACzBT;YACA,eAAeuB;YACf,eAAeC;YACfE;YACAD;YACAE;YACAX;YACA;gBACI,OAAOR,cAAc,IAAI;YAC7B;QACJ;QAEA,OAAOC;IACX;AACJ;AAEO,MAAMuB,gDAAyBC,uBAAAA,oBAAgC,CAAC;IACnE,gBAAgBf;IAChB,cAAc;QAACgB;QAAoBC;KAAqB;AAC5D"}
+{"version":3,"file":"api/features/AiPromptContext/AiPromptContextBuilder.js","sources":["../../../../src/api/features/AiPromptContext/AiPromptContextBuilder.ts"],"sourcesContent":["import { GetSettingsUseCase } from \"~/api/features/GetSettings/index.js\";\nimport { GetFileContentsByIdUseCase } from \"@webiny/api-file-manager/features/file/GetFileContentsById/index.js\";\nimport { GetFileUseCase } from \"@webiny/api-file-manager/features/file/GetFile/index.js\";\nimport { TextExtractor } from \"@webiny/api-core/features/ai/index.js\";\nimport {\n    AiPromptContextBuilder as Abstraction,\n    ProjectFileAssembler,\n    type AiPromptContext,\n    type AiPromptContextParams,\n    type ProjectFileContent,\n    type ResolvedPersona,\n    type ResolvedProject\n} from \"./abstractions.js\";\nimport type { IAiPowerUpsSettings } from \"~/api/types.js\";\nimport type { ProjectPreset } from \"~/api/features/Projects/types.js\";\nimport { WriterPersonaSection } from \"./WriterPersonaSection.js\";\nimport { ReaderPersonaSection } from \"./ReaderPersonaSection.js\";\nimport { ProjectSection } from \"./ProjectSection.js\";\n\nfunction resolvePersonas(\n    settings: IAiPowerUpsSettings,\n    project: ProjectPreset | undefined,\n    params: AiPromptContextParams\n): { reader?: ResolvedPersona; writer?: ResolvedPersona } {\n    const readerPersonaId = params.readerPersonaId ?? project?.defaultReaderPersonaId ?? undefined;\n    const writerPersonaId = params.writerPersonaId ?? project?.defaultWriterPersonaId ?? undefined;\n\n    const readerPreset = readerPersonaId\n        ? settings.readerPersonas?.presets?.find(p => p.id === readerPersonaId)\n        : undefined;\n\n    const writerPreset = writerPersonaId\n        ? settings.writerPersonas?.presets?.find(p => p.id === writerPersonaId)\n        : undefined;\n\n    return {\n        reader: readerPreset\n            ? {\n                  name: readerPreset.name,\n                  description: readerPreset.description,\n                  style: readerPreset.style\n              }\n            : undefined,\n        writer: writerPreset\n            ? {\n                  name: writerPreset.name,\n                  description: writerPreset.description,\n                  style: writerPreset.style\n              }\n            : undefined\n    };\n}\n\nfunction formatContext(ctx: AiPromptContext): string {\n    const sections: string[] = [];\n\n    if (ctx.writerPersona) {\n        sections.push(WriterPersonaSection.format(ctx.writerPersona));\n    }\n    if (ctx.readerPersona) {\n        sections.push(ReaderPersonaSection.format(ctx.readerPersona));\n    }\n    if (ctx.project) {\n        const projectText = ProjectSection.format(ctx.project);\n        if (projectText) {\n            sections.push(projectText);\n        }\n    }\n    return sections.length > 0 ? \"\\n\\n\" + sections.join(\"\\n\\n\") : \"\";\n}\n\nfunction emptyContext(warnings: string[]): AiPromptContext {\n    return {\n        project: undefined,\n        readerPersona: undefined,\n        writerPersona: undefined,\n        allProjectFiles: [],\n        additionalFiles: [],\n        excludedFileIds: new Set(),\n        cacheHit: false,\n        warnings,\n        toString() {\n            return \"\";\n        }\n    };\n}\n\nclass AiPromptContextBuilderImpl implements Abstraction.Interface {\n    constructor(\n        private getSettings: GetSettingsUseCase.Interface,\n        private fileAssembler: ProjectFileAssembler.Interface,\n        private getFileContents: GetFileContentsByIdUseCase.Interface,\n        private getFile: GetFileUseCase.Interface,\n        private textExtractor: TextExtractor.Interface\n    ) {}\n\n    async execute(params: AiPromptContextParams): Promise<AiPromptContext> {\n        const warnings: string[] = [];\n\n        const settingsResult = await this.getSettings.execute();\n        if (settingsResult.isFail()) {\n            warnings.push(\"Failed to load AI PowerUps settings.\");\n            return emptyContext(warnings);\n        }\n\n        const settings = settingsResult.value;\n        const projectPreset = params.projectId\n            ? settings.projects?.presets?.find(p => p.id === params.projectId)\n            : undefined;\n\n        const { reader, writer } = resolvePersonas(settings, projectPreset, params);\n        const excludedFileIds = new Set(params.excludedFileIds ?? []);\n\n        let project: ResolvedProject | undefined;\n        let allProjectFiles: ProjectFileContent[] = [];\n        let cacheHit = false;\n\n        if (projectPreset) {\n            const assembled = await this.fileAssembler.execute(\n                projectPreset.id,\n                projectPreset.version ?? 0,\n                projectPreset.files ?? [],\n                warnings\n            );\n            allProjectFiles = assembled.files;\n            cacheHit = assembled.cacheHit;\n\n            const visibleFiles =\n                excludedFileIds.size > 0\n                    ? allProjectFiles.filter(f => !excludedFileIds.has(f.id))\n                    : allProjectFiles;\n\n            project = {\n                name: projectPreset.name,\n                instructions: projectPreset.instructions,\n                files: visibleFiles,\n                totalTokens: visibleFiles.reduce((sum, f) => sum + f.tokenCount, 0)\n            };\n        }\n\n        const additionalFiles = await this.fetchAdditionalFiles(params.additionalFileIds, warnings);\n        if (additionalFiles.length > 0) {\n            allProjectFiles = [...allProjectFiles, ...additionalFiles];\n        }\n\n        const ctx: AiPromptContext = {\n            project,\n            readerPersona: reader,\n            writerPersona: writer,\n            allProjectFiles,\n            additionalFiles,\n            excludedFileIds,\n            cacheHit,\n            warnings,\n            toString() {\n                return formatContext(this);\n            }\n        };\n\n        return ctx;\n    }\n\n    private async fetchAdditionalFiles(\n        fileIds: string[] | null | undefined,\n        warnings: string[]\n    ): Promise<ProjectFileContent[]> {\n        if (!fileIds || fileIds.length === 0) {\n            return [];\n        }\n\n        const results = await Promise.all(\n            fileIds.map(async (fileId): Promise<ProjectFileContent | null> => {\n                const [contentsResult, damResult] = await Promise.all([\n                    this.getFileContents.execute(fileId),\n                    this.getFile.execute(fileId)\n                ]);\n\n                if (contentsResult.isFail()) {\n                    warnings.push(\n                        `Failed to load additional file (${fileId}): ${contentsResult.error.message}`\n                    );\n                    return null;\n                }\n\n                const name = damResult.isOk() ? damResult.value.name : fileId;\n                const mimeType = damResult.isOk()\n                    ? damResult.value.type\n                    : \"application/octet-stream\";\n\n                if (!this.textExtractor.canExtract(mimeType)) {\n                    warnings.push(\n                        `Skipped additional file \"${name}\" (${fileId}): unsupported type \"${mimeType}\".`\n                    );\n                    return null;\n                }\n\n                const { text: content } = await this.textExtractor.extract(\n                    contentsResult.value.buffer,\n                    mimeType\n                );\n\n                return {\n                    id: fileId,\n                    name,\n                    content,\n                    description: damResult.isOk() ? damResult.value.description : undefined,\n                    tokenCount: Math.ceil(content.length / 4)\n                };\n            })\n        );\n\n        return results.filter((r): r is ProjectFileContent => r !== null);\n    }\n}\n\nexport const AiPromptContextBuilder = Abstraction.createImplementation({\n    implementation: AiPromptContextBuilderImpl,\n    dependencies: [\n        GetSettingsUseCase,\n        ProjectFileAssembler,\n        GetFileContentsByIdUseCase,\n        GetFileUseCase,\n        TextExtractor\n    ]\n});\n"],"names":["resolvePersonas","settings","project","params","readerPersonaId","undefined","writerPersonaId","readerPreset","p","writerPreset","formatContext","ctx","sections","WriterPersonaSection","ReaderPersonaSection","projectText","ProjectSection","emptyContext","warnings","Set","AiPromptContextBuilderImpl","getSettings","fileAssembler","getFileContents","getFile","textExtractor","settingsResult","projectPreset","reader","writer","excludedFileIds","allProjectFiles","cacheHit","assembled","visibleFiles","f","sum","additionalFiles","fileIds","results","Promise","fileId","contentsResult","damResult","name","mimeType","content","Math","r","AiPromptContextBuilder","Abstraction","GetSettingsUseCase","ProjectFileAssembler","GetFileContentsByIdUseCase","GetFileUseCase","TextExtractor"],"mappings":";;;;;;;;AAmBA,SAASA,gBACLC,QAA6B,EAC7BC,OAAkC,EAClCC,MAA6B;IAE7B,MAAMC,kBAAkBD,OAAO,eAAe,IAAID,SAAS,0BAA0BG;IACrF,MAAMC,kBAAkBH,OAAO,eAAe,IAAID,SAAS,0BAA0BG;IAErF,MAAME,eAAeH,kBACfH,SAAS,cAAc,EAAE,SAAS,KAAKO,CAAAA,IAAKA,EAAE,EAAE,KAAKJ,mBACrDC;IAEN,MAAMI,eAAeH,kBACfL,SAAS,cAAc,EAAE,SAAS,KAAKO,CAAAA,IAAKA,EAAE,EAAE,KAAKF,mBACrDD;IAEN,OAAO;QACH,QAAQE,eACF;YACI,MAAMA,aAAa,IAAI;YACvB,aAAaA,aAAa,WAAW;YACrC,OAAOA,aAAa,KAAK;QAC7B,IACAF;QACN,QAAQI,eACF;YACI,MAAMA,aAAa,IAAI;YACvB,aAAaA,aAAa,WAAW;YACrC,OAAOA,aAAa,KAAK;QAC7B,IACAJ;IACV;AACJ;AAEA,SAASK,cAAcC,GAAoB;IACvC,MAAMC,WAAqB,EAAE;IAE7B,IAAID,IAAI,aAAa,EACjBC,SAAS,IAAI,CAACC,qBAAqB,MAAM,CAACF,IAAI,aAAa;IAE/D,IAAIA,IAAI,aAAa,EACjBC,SAAS,IAAI,CAACE,qBAAqB,MAAM,CAACH,IAAI,aAAa;IAE/D,IAAIA,IAAI,OAAO,EAAE;QACb,MAAMI,cAAcC,eAAe,MAAM,CAACL,IAAI,OAAO;QACrD,IAAII,aACAH,SAAS,IAAI,CAACG;IAEtB;IACA,OAAOH,SAAS,MAAM,GAAG,IAAI,SAASA,SAAS,IAAI,CAAC,UAAU;AAClE;AAEA,SAASK,aAAaC,QAAkB;IACpC,OAAO;QACH,SAASb;QACT,eAAeA;QACf,eAAeA;QACf,iBAAiB,EAAE;QACnB,iBAAiB,EAAE;QACnB,iBAAiB,IAAIc;QACrB,UAAU;QACVD;QACA;YACI,OAAO;QACX;IACJ;AACJ;AAEA,MAAME;IACF,YACYC,WAAyC,EACzCC,aAA6C,EAC7CC,eAAqD,EACrDC,OAAiC,EACjCC,aAAsC,CAChD;aALUJ,WAAW,GAAXA;aACAC,aAAa,GAAbA;aACAC,eAAe,GAAfA;aACAC,OAAO,GAAPA;aACAC,aAAa,GAAbA;IACT;IAEH,MAAM,QAAQtB,MAA6B,EAA4B;QACnE,MAAMe,WAAqB,EAAE;QAE7B,MAAMQ,iBAAiB,MAAM,IAAI,CAAC,WAAW,CAAC,OAAO;QACrD,IAAIA,eAAe,MAAM,IAAI;YACzBR,SAAS,IAAI,CAAC;YACd,OAAOD,aAAaC;QACxB;QAEA,MAAMjB,WAAWyB,eAAe,KAAK;QACrC,MAAMC,gBAAgBxB,OAAO,SAAS,GAChCF,SAAS,QAAQ,EAAE,SAAS,KAAKO,CAAAA,IAAKA,EAAE,EAAE,KAAKL,OAAO,SAAS,IAC/DE;QAEN,MAAM,EAAEuB,MAAM,EAAEC,MAAM,EAAE,GAAG7B,gBAAgBC,UAAU0B,eAAexB;QACpE,MAAM2B,kBAAkB,IAAIX,IAAIhB,OAAO,eAAe,IAAI,EAAE;QAE5D,IAAID;QACJ,IAAI6B,kBAAwC,EAAE;QAC9C,IAAIC,WAAW;QAEf,IAAIL,eAAe;YACf,MAAMM,YAAY,MAAM,IAAI,CAAC,aAAa,CAAC,OAAO,CAC9CN,cAAc,EAAE,EAChBA,cAAc,OAAO,IAAI,GACzBA,cAAc,KAAK,IAAI,EAAE,EACzBT;YAEJa,kBAAkBE,UAAU,KAAK;YACjCD,WAAWC,UAAU,QAAQ;YAE7B,MAAMC,eACFJ,gBAAgB,IAAI,GAAG,IACjBC,gBAAgB,MAAM,CAACI,CAAAA,IAAK,CAACL,gBAAgB,GAAG,CAACK,EAAE,EAAE,KACrDJ;YAEV7B,UAAU;gBACN,MAAMyB,cAAc,IAAI;gBACxB,cAAcA,cAAc,YAAY;gBACxC,OAAOO;gBACP,aAAaA,aAAa,MAAM,CAAC,CAACE,KAAKD,IAAMC,MAAMD,EAAE,UAAU,EAAE;YACrE;QACJ;QAEA,MAAME,kBAAkB,MAAM,IAAI,CAAC,oBAAoB,CAAClC,OAAO,iBAAiB,EAAEe;QAClF,IAAImB,gBAAgB,MAAM,GAAG,GACzBN,kBAAkB;eAAIA;eAAoBM;SAAgB;QAG9D,MAAM1B,MAAuB;YACzBT;YACA,eAAe0B;YACf,eAAeC;YACfE;YACAM;YACAP;YACAE;YACAd;YACA;gBACI,OAAOR,cAAc,IAAI;YAC7B;QACJ;QAEA,OAAOC;IACX;IAEA,MAAc,qBACV2B,OAAoC,EACpCpB,QAAkB,EACW;QAC7B,IAAI,CAACoB,WAAWA,AAAmB,MAAnBA,QAAQ,MAAM,EAC1B,OAAO,EAAE;QAGb,MAAMC,UAAU,MAAMC,QAAQ,GAAG,CAC7BF,QAAQ,GAAG,CAAC,OAAOG;YACf,MAAM,CAACC,gBAAgBC,UAAU,GAAG,MAAMH,QAAQ,GAAG,CAAC;gBAClD,IAAI,CAAC,eAAe,CAAC,OAAO,CAACC;gBAC7B,IAAI,CAAC,OAAO,CAAC,OAAO,CAACA;aACxB;YAED,IAAIC,eAAe,MAAM,IAAI;gBACzBxB,SAAS,IAAI,CACT,CAAC,gCAAgC,EAAEuB,OAAO,GAAG,EAAEC,eAAe,KAAK,CAAC,OAAO,EAAE;gBAEjF,OAAO;YACX;YAEA,MAAME,OAAOD,UAAU,IAAI,KAAKA,UAAU,KAAK,CAAC,IAAI,GAAGF;YACvD,MAAMI,WAAWF,UAAU,IAAI,KACzBA,UAAU,KAAK,CAAC,IAAI,GACpB;YAEN,IAAI,CAAC,IAAI,CAAC,aAAa,CAAC,UAAU,CAACE,WAAW;gBAC1C3B,SAAS,IAAI,CACT,CAAC,yBAAyB,EAAE0B,KAAK,GAAG,EAAEH,OAAO,qBAAqB,EAAEI,SAAS,EAAE,CAAC;gBAEpF,OAAO;YACX;YAEA,MAAM,EAAE,MAAMC,OAAO,EAAE,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,OAAO,CACtDJ,eAAe,KAAK,CAAC,MAAM,EAC3BG;YAGJ,OAAO;gBACH,IAAIJ;gBACJG;gBACAE;gBACA,aAAaH,UAAU,IAAI,KAAKA,UAAU,KAAK,CAAC,WAAW,GAAGtC;gBAC9D,YAAY0C,KAAK,IAAI,CAACD,QAAQ,MAAM,GAAG;YAC3C;QACJ;QAGJ,OAAOP,QAAQ,MAAM,CAAC,CAACS,IAA+BA,AAAM,SAANA;IAC1D;AACJ;AAEO,MAAMC,gDAAyBC,uBAAAA,oBAAgC,CAAC;IACnE,gBAAgB9B;IAChB,cAAc;QACV+B;QACAC;QACAC;QACAC;QACAC;KACH;AACL"}

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

@@ -28,6 +28,7 @@ export interface AiPromptContextParams {
     readerPersonaId?: string | null;
     writerPersonaId?: string | null;
     excludedFileIds?: string[] | null;
+    additionalFileIds?: string[] | null;
 }
 export interface ProjectFileContent {
     id: string;
@@ -52,6 +53,7 @@ export interface AiPromptContext {
     readerPersona?: ResolvedPersona;
     writerPersona?: ResolvedPersona;
     allProjectFiles: ProjectFileContent[];
+    additionalFiles: ProjectFileContent[];
     excludedFileIds: Set<string>;
     cacheHit: boolean;
     warnings: string[];
@@ -66,3 +68,4 @@ export declare namespace AiPromptContextBuilder {
     type Params = AiPromptContextParams;
     type Context = AiPromptContext;
 }
+export declare function formatAdditionalFilesContext(files: ProjectFileContent[]): string;

package/api/features/AiPromptContext/abstractions.js

@@ -2,6 +2,14 @@ import { createAbstraction } from "@webiny/feature/api";
 const ProjectFileCache = createAbstraction("ProjectFileCache");
 const ProjectFileAssembler = createAbstraction("ProjectFileAssembler");
 const AiPromptContextBuilder = createAbstraction("AiPowerUpsPromptContextBuilder");
-export { AiPromptContextBuilder, ProjectFileAssembler, ProjectFileCache };
+function formatAdditionalFilesContext(files) {
+    if (0 === files.length) return "";
+    const sections = files.map((f)=>{
+        const desc = f.description ? ` description="${f.description}"` : "";
+        return `<attached-file name="${f.name}"${desc}>\n${f.content}\n</attached-file>`;
+    });
+    return `\n\nThe user attached the following files as additional context for this generation:\n\n${sections.join("\n\n")}`;
+}
+export { AiPromptContextBuilder, ProjectFileAssembler, ProjectFileCache, formatAdditionalFilesContext };
 
 //# sourceMappingURL=abstractions.js.map

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

@@ -1 +1 @@
-{"version":3,"file":"api/features/AiPromptContext/abstractions.js","sources":["../../../../src/api/features/AiPromptContext/abstractions.ts"],"sourcesContent":["import { createAbstraction } from \"@webiny/feature/api\";\n\n// ============================================================================\n// ProjectFileCache\n// ============================================================================\n\nexport interface IProjectFileCache {\n    get(projectId: string, version: number): Promise<{ files: ProjectFileContent[]; hit: boolean }>;\n    set(projectId: string, version: number, files: ProjectFileContent[]): Promise<void>;\n}\n\nexport const ProjectFileCache = createAbstraction<IProjectFileCache>(\"ProjectFileCache\");\n\nexport namespace ProjectFileCache {\n    export type Interface = IProjectFileCache;\n}\n\n// ============================================================================\n// ProjectFileAssembler\n// ============================================================================\n\nexport interface IProjectFileAssembler {\n    execute(\n        projectId: string,\n        version: number,\n        files: Array<{ id: string; name: string; mimeType: string }>,\n        warnings: string[]\n    ): Promise<{ files: ProjectFileContent[]; cacheHit: boolean }>;\n}\n\nexport const ProjectFileAssembler =\n    createAbstraction<IProjectFileAssembler>(\"ProjectFileAssembler\");\n\nexport namespace ProjectFileAssembler {\n    export type Interface = IProjectFileAssembler;\n}\n\n// ============================================================================\n// AiPromptContext\n// ============================================================================\n\nexport interface AiPromptContextParams {\n    projectId?: string | null;\n    readerPersonaId?: string | null;\n    writerPersonaId?: string | null;\n    excludedFileIds?: string[] | null;\n}\n\nexport interface ProjectFileContent {\n    id: string;\n    name: string;\n    content: string;\n    description?: string;\n    tokenCount: number;\n}\n\nexport interface ResolvedProject {\n    name: string;\n    instructions?: string;\n    files: ProjectFileContent[];\n    totalTokens: number;\n}\n\nexport interface ResolvedPersona {\n    name: string;\n    description: string;\n    style?: string;\n}\n\nexport interface AiPromptContext {\n    project?: ResolvedProject;\n    readerPersona?: ResolvedPersona;\n    writerPersona?: ResolvedPersona;\n    allProjectFiles: ProjectFileContent[];\n    excludedFileIds: Set<string>;\n    cacheHit: boolean;\n    warnings: string[];\n    toString(): string;\n}\n\nexport interface IAiPromptContextBuilder {\n    execute(params: AiPromptContextParams): Promise<AiPromptContext>;\n}\n\nexport const AiPromptContextBuilder = createAbstraction<IAiPromptContextBuilder>(\n    \"AiPowerUpsPromptContextBuilder\"\n);\n\nexport namespace AiPromptContextBuilder {\n    export type Interface = IAiPromptContextBuilder;\n    export type Params = AiPromptContextParams;\n    export type Context = AiPromptContext;\n}\n"],"names":["ProjectFileCache","createAbstraction","ProjectFileAssembler","AiPromptContextBuilder"],"mappings":";AAWO,MAAMA,mBAAmBC,kBAAqC;AAmB9D,MAAMC,uBACTD,kBAAyC;AAqDtC,MAAME,yBAAyBF,kBAClC"}
+{"version":3,"file":"api/features/AiPromptContext/abstractions.js","sources":["../../../../src/api/features/AiPromptContext/abstractions.ts"],"sourcesContent":["import { createAbstraction } from \"@webiny/feature/api\";\n\n// ============================================================================\n// ProjectFileCache\n// ============================================================================\n\nexport interface IProjectFileCache {\n    get(projectId: string, version: number): Promise<{ files: ProjectFileContent[]; hit: boolean }>;\n    set(projectId: string, version: number, files: ProjectFileContent[]): Promise<void>;\n}\n\nexport const ProjectFileCache = createAbstraction<IProjectFileCache>(\"ProjectFileCache\");\n\nexport namespace ProjectFileCache {\n    export type Interface = IProjectFileCache;\n}\n\n// ============================================================================\n// ProjectFileAssembler\n// ============================================================================\n\nexport interface IProjectFileAssembler {\n    execute(\n        projectId: string,\n        version: number,\n        files: Array<{ id: string; name: string; mimeType: string }>,\n        warnings: string[]\n    ): Promise<{ files: ProjectFileContent[]; cacheHit: boolean }>;\n}\n\nexport const ProjectFileAssembler =\n    createAbstraction<IProjectFileAssembler>(\"ProjectFileAssembler\");\n\nexport namespace ProjectFileAssembler {\n    export type Interface = IProjectFileAssembler;\n}\n\n// ============================================================================\n// AiPromptContext\n// ============================================================================\n\nexport interface AiPromptContextParams {\n    projectId?: string | null;\n    readerPersonaId?: string | null;\n    writerPersonaId?: string | null;\n    excludedFileIds?: string[] | null;\n    additionalFileIds?: string[] | null;\n}\n\nexport interface ProjectFileContent {\n    id: string;\n    name: string;\n    content: string;\n    description?: string;\n    tokenCount: number;\n}\n\nexport interface ResolvedProject {\n    name: string;\n    instructions?: string;\n    files: ProjectFileContent[];\n    totalTokens: number;\n}\n\nexport interface ResolvedPersona {\n    name: string;\n    description: string;\n    style?: string;\n}\n\nexport interface AiPromptContext {\n    project?: ResolvedProject;\n    readerPersona?: ResolvedPersona;\n    writerPersona?: ResolvedPersona;\n    allProjectFiles: ProjectFileContent[];\n    additionalFiles: ProjectFileContent[];\n    excludedFileIds: Set<string>;\n    cacheHit: boolean;\n    warnings: string[];\n    toString(): string;\n}\n\nexport interface IAiPromptContextBuilder {\n    execute(params: AiPromptContextParams): Promise<AiPromptContext>;\n}\n\nexport const AiPromptContextBuilder = createAbstraction<IAiPromptContextBuilder>(\n    \"AiPowerUpsPromptContextBuilder\"\n);\n\nexport namespace AiPromptContextBuilder {\n    export type Interface = IAiPromptContextBuilder;\n    export type Params = AiPromptContextParams;\n    export type Context = AiPromptContext;\n}\n\nexport function formatAdditionalFilesContext(files: ProjectFileContent[]): string {\n    if (files.length === 0) {\n        return \"\";\n    }\n    const sections = files.map(f => {\n        const desc = f.description ? ` description=\"${f.description}\"` : \"\";\n        return `<attached-file name=\"${f.name}\"${desc}>\\n${f.content}\\n</attached-file>`;\n    });\n    return `\\n\\nThe user attached the following files as additional context for this generation:\\n\\n${sections.join(\"\\n\\n\")}`;\n}\n"],"names":["ProjectFileCache","createAbstraction","ProjectFileAssembler","AiPromptContextBuilder","formatAdditionalFilesContext","files","sections","f","desc"],"mappings":";AAWO,MAAMA,mBAAmBC,kBAAqC;AAmB9D,MAAMC,uBACTD,kBAAyC;AAuDtC,MAAME,yBAAyBF,kBAClC;AASG,SAASG,6BAA6BC,KAA2B;IACpE,IAAIA,AAAiB,MAAjBA,MAAM,MAAM,EACZ,OAAO;IAEX,MAAMC,WAAWD,MAAM,GAAG,CAACE,CAAAA;QACvB,MAAMC,OAAOD,EAAE,WAAW,GAAG,CAAC,cAAc,EAAEA,EAAE,WAAW,CAAC,CAAC,CAAC,GAAG;QACjE,OAAO,CAAC,qBAAqB,EAAEA,EAAE,IAAI,CAAC,CAAC,EAAEC,KAAK,GAAG,EAAED,EAAE,OAAO,CAAC,kBAAkB,CAAC;IACpF;IACA,OAAO,CAAC,wFAAwF,EAAED,SAAS,IAAI,CAAC,SAAS;AAC7H"}

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

@@ -1 +1 @@
-export { AiPromptContextBuilder, type ResolvedProject, type ResolvedPersona, type ProjectFileContent } from "./abstractions.js";
+export { AiPromptContextBuilder, formatAdditionalFilesContext, type ResolvedProject, type ResolvedPersona, type ProjectFileContent } from "./abstractions.js";

package/api/features/AiPromptContext/index.js

@@ -1 +1 @@
-export { AiPromptContextBuilder } from "./abstractions.js";
+export { AiPromptContextBuilder, formatAdditionalFilesContext } from "./abstractions.js";

package/api/features/Capabilities/CapabilitiesHandler.d.ts

@@ -0,0 +1,28 @@
+import { z } from "zod";
+import { AiPowerUpsSettingsGroupHandler } from "../../../api/features/shared/index.js";
+import type { CapabilitiesSettings } from "./types.js";
+import type { PersistedCapabilities } from "./types.js";
+declare class CapabilitiesHandlerImpl implements AiPowerUpsSettingsGroupHandler.Interface {
+    readonly name = "capabilities";
+    readonly inputSchema: z.ZodObject<{
+        items: z.ZodRecord<z.ZodString, z.ZodObject<{
+            enabled: z.ZodOptional<z.ZodNullable<z.ZodBoolean>>;
+            overrides: z.ZodOptional<z.ZodNullable<z.ZodObject<{
+                roleId: z.ZodOptional<z.ZodNullable<z.ZodUnion<readonly [z.ZodEnum<{
+                    fast: "fast";
+                    standard: "standard";
+                    vision: "vision";
+                }>, z.ZodLiteral<"">]>>>;
+                connectionId: z.ZodOptional<z.ZodNullable<z.ZodString>>;
+                model: z.ZodOptional<z.ZodNullable<z.ZodString>>;
+                additionalInstructions: z.ZodOptional<z.ZodNullable<z.ZodString>>;
+            }, z.core.$strip>>>;
+        }, z.core.$strip>>;
+    }, z.core.$strip>;
+    mapFromStorage(persisted: unknown): CapabilitiesSettings;
+    mapToStorage(internal: unknown): Promise<PersistedCapabilities>;
+}
+declare const _default: typeof CapabilitiesHandlerImpl & {
+    __abstraction: import("@webiny/di").Abstraction<import("../shared/abstractions.js").IAiPowerUpsSettingsGroupHandler>;
+};
+export default _default;