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

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

@@ -1,4 +1,4 @@
-import { TaskDefinition } from "@webiny/api-core/features/task/TaskDefinition/index.js";
+import { TaskDefinition, TaskHandler } from "@webiny/api-core/features/task/TaskDefinition/index.js";
 import { Ai } from "@webiny/api-core/features/ai/index.js";
 import { ApplyImageEnrichmentUseCase, PrepareImageEnrichmentUseCase } from "./abstractions.js";
 export declare const AI_IMAGE_ENRICHMENT_TASK_ID = "fmAiImageEnrichment";
@@ -10,10 +10,14 @@ export interface IAiImageEnrichmentTaskInput {
  * with the streaming HTTP route (`AiImageEnrichmentStreamRoute`); the only difference is that this
  * one waits for the whole AI response, because a background task has no one to stream to.
  */
-declare class AiImageEnrichmentTaskImpl implements TaskDefinition.Interface<IAiImageEnrichmentTaskInput> {
+declare class AiImageEnrichmentTaskHandlerImpl implements TaskHandler.Interface<IAiImageEnrichmentTaskInput> {
     private prepare;
     private apply;
     private ai;
+    constructor(prepare: PrepareImageEnrichmentUseCase.Interface, apply: ApplyImageEnrichmentUseCase.Interface, ai: Ai.Interface);
+    run({ input, controller }: TaskHandler.RunParams<IAiImageEnrichmentTaskInput>): Promise<TaskDefinition.Result<IAiImageEnrichmentTaskInput>>;
+}
+declare class AiImageEnrichmentTaskImpl implements TaskDefinition.Interface {
     id: string;
     title: string;
     description: string;
@@ -21,8 +25,9 @@ declare class AiImageEnrichmentTaskImpl implements TaskDefinition.Interface<IAiI
     isPrivate: boolean;
     databaseLogs: boolean;
     readonly selfCleanup: ("onAbort" | "onSuccess")[];
-    constructor(prepare: PrepareImageEnrichmentUseCase.Interface, apply: ApplyImageEnrichmentUseCase.Interface, ai: Ai.Interface);
-    run({ input, controller }: TaskDefinition.RunParams<IAiImageEnrichmentTaskInput>): Promise<TaskDefinition.Result<IAiImageEnrichmentTaskInput>>;
+    handler: typeof AiImageEnrichmentTaskHandlerImpl & {
+        __abstraction: import("@webiny/di").Abstraction<import("@webiny/api-core/features/task/TaskDefinition/abstractions.js").ITaskHandler<import("@webiny/api-core/features/task/TaskDefinition/abstractions.js").ITaskInput, import("@webiny/api-core/features/task/TaskDefinition/abstractions.js").ITaskOutput>>;
+    };
 }
 export declare const AiImageEnrichmentTask: typeof AiImageEnrichmentTaskImpl & {
     __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>>;

package/api/features/AiImageEnrichment/AiImageEnrichmentTask.js

@@ -1,24 +1,14 @@
-import { TaskDefinition } from "@webiny/api-core/features/task/TaskDefinition/index.js";
+import { TaskDefinition, TaskHandler } from "@webiny/api-core/features/task/TaskDefinition/index.js";
 import { Ai } from "@webiny/api-core/features/ai/index.js";
 import { ApplyImageEnrichmentUseCase, PrepareImageEnrichmentUseCase } from "./abstractions.js";
 import { buildEnrichmentAiRequest } from "./buildEnrichmentAiRequest.js";
 import { EnrichmentNotAnImageError } from "./errors.js";
 const AI_IMAGE_ENRICHMENT_TASK_ID = "fmAiImageEnrichment";
-class AiImageEnrichmentTaskImpl {
+class AiImageEnrichmentTaskHandlerImpl {
     constructor(prepare, apply, ai){
         this.prepare = prepare;
         this.apply = apply;
         this.ai = ai;
-        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.";
-        this.maxIterations = 1;
-        this.isPrivate = true;
-        this.databaseLogs = false;
-        this.selfCleanup = [
-            "onSuccess",
-            "onAbort"
-        ];
     }
     async run({ input, controller }) {
         if (controller.runtime.isAborted()) return controller.response.aborted();
@@ -54,14 +44,33 @@ class AiImageEnrichmentTaskImpl {
         return controller.response.done("AI image enrichment completed successfully.");
     }
 }
-const AiImageEnrichmentTask = TaskDefinition.createImplementation({
-    implementation: AiImageEnrichmentTaskImpl,
+const AiImageEnrichmentTaskHandler = TaskHandler.createImplementation({
+    implementation: AiImageEnrichmentTaskHandlerImpl,
     dependencies: [
         PrepareImageEnrichmentUseCase,
         ApplyImageEnrichmentUseCase,
         Ai
     ]
 });
+class AiImageEnrichmentTaskImpl {
+    constructor(){
+        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.";
+        this.maxIterations = 1;
+        this.isPrivate = true;
+        this.databaseLogs = false;
+        this.selfCleanup = [
+            "onSuccess",
+            "onAbort"
+        ];
+        this.handler = AiImageEnrichmentTaskHandler;
+    }
+}
+const AiImageEnrichmentTask = TaskDefinition.createImplementation({
+    implementation: AiImageEnrichmentTaskImpl,
+    dependencies: []
+});
 export { AI_IMAGE_ENRICHMENT_TASK_ID, AiImageEnrichmentTask };
 
 //# sourceMappingURL=AiImageEnrichmentTask.js.map

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 { TaskDefinition } from \"@webiny/api-core/features/task/TaskDefinition/index.js\";\nimport { Ai } from \"@webiny/api-core/features/ai/index.js\";\nimport { ApplyImageEnrichmentUseCase, PrepareImageEnrichmentUseCase } from \"./abstractions.js\";\nimport { buildEnrichmentAiRequest } from \"./buildEnrichmentAiRequest.js\";\nimport { EnrichmentNotAnImageError } from \"./errors.js\";\n\nexport const AI_IMAGE_ENRICHMENT_TASK_ID = \"fmAiImageEnrichment\";\n\nexport interface IAiImageEnrichmentTaskInput {\n    fileId: string;\n}\n\n/**\n * Background enrichment, triggered after a file is created. Shares its preparation and persistence\n * with the streaming HTTP route (`AiImageEnrichmentStreamRoute`); the only difference is that this\n * one waits for the whole AI response, because a background task has no one to stream to.\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 prepare: PrepareImageEnrichmentUseCase.Interface,\n        private apply: ApplyImageEnrichmentUseCase.Interface,\n        private ai: Ai.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 preparedResult = await this.prepare.execute(input.fileId);\n        if (preparedResult.isFail()) {\n            const error = preparedResult.error;\n            // A non-image isn't a failure — nothing to enrich, so the task is simply done.\n            if (error instanceof EnrichmentNotAnImageError) {\n                return controller.response.done(error.message);\n            }\n            return controller.response.error({ message: error.message });\n        }\n\n        const prepared = preparedResult.value;\n\n        let tags: string[];\n        let description: string;\n        try {\n            const request = buildEnrichmentAiRequest(prepared);\n            const aiResult = await this.ai.generateText(request);\n\n            tags = aiResult.output.tags;\n            description = aiResult.output.description;\n        } catch (error) {\n            return controller.response.error({\n                message: `AI enrichment failed: ${error instanceof Error ? error.message : String(error)}`\n            });\n        }\n\n        const appliedResult = await this.apply.execute({\n            fileId: prepared.fileId,\n            tags,\n            description\n        });\n\n        if (appliedResult.isFail()) {\n            return controller.response.error({ message: appliedResult.error.message });\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: [PrepareImageEnrichmentUseCase, ApplyImageEnrichmentUseCase, Ai]\n});\n"],"names":["AI_IMAGE_ENRICHMENT_TASK_ID","AiImageEnrichmentTaskImpl","prepare","apply","ai","input","controller","preparedResult","error","EnrichmentNotAnImageError","prepared","tags","description","request","buildEnrichmentAiRequest","aiResult","Error","String","appliedResult","AiImageEnrichmentTask","TaskDefinition","PrepareImageEnrichmentUseCase","ApplyImageEnrichmentUseCase","Ai"],"mappings":";;;;;AAMO,MAAMA,8BAA8B;AAW3C,MAAMC;IAUF,YACYC,OAAgD,EAChDC,KAA4C,EAC5CC,EAAgB,CAC1B;aAHUF,OAAO,GAAPA;aACAC,KAAK,GAALA;aACAC,EAAE,GAAFA;aAZZ,EAAE,GAAGJ;aACL,KAAK,GAAG;aACR,WAAW,GAAG;aACd,aAAa,GAAG;aAChB,SAAS,GAAG;aACZ,YAAY,GAAG;aAEC,WAAW,GAAG;YAAC;YAAsB;SAAmB;IAMrE;IAEH,MAAM,IAAI,EACNK,KAAK,EACLC,UAAU,EAC0C,EAEtD;QACE,IAAIA,WAAW,OAAO,CAAC,SAAS,IAC5B,OAAOA,WAAW,QAAQ,CAAC,OAAO;QAGtC,MAAMC,iBAAiB,MAAM,IAAI,CAAC,OAAO,CAAC,OAAO,CAACF,MAAM,MAAM;QAC9D,IAAIE,eAAe,MAAM,IAAI;YACzB,MAAMC,QAAQD,eAAe,KAAK;YAElC,IAAIC,iBAAiBC,2BACjB,OAAOH,WAAW,QAAQ,CAAC,IAAI,CAACE,MAAM,OAAO;YAEjD,OAAOF,WAAW,QAAQ,CAAC,KAAK,CAAC;gBAAE,SAASE,MAAM,OAAO;YAAC;QAC9D;QAEA,MAAME,WAAWH,eAAe,KAAK;QAErC,IAAII;QACJ,IAAIC;QACJ,IAAI;YACA,MAAMC,UAAUC,yBAAyBJ;YACzC,MAAMK,WAAW,MAAM,IAAI,CAAC,EAAE,CAAC,YAAY,CAACF;YAE5CF,OAAOI,SAAS,MAAM,CAAC,IAAI;YAC3BH,cAAcG,SAAS,MAAM,CAAC,WAAW;QAC7C,EAAE,OAAOP,OAAO;YACZ,OAAOF,WAAW,QAAQ,CAAC,KAAK,CAAC;gBAC7B,SAAS,CAAC,sBAAsB,EAAEE,iBAAiBQ,QAAQR,MAAM,OAAO,GAAGS,OAAOT,QAAQ;YAC9F;QACJ;QAEA,MAAMU,gBAAgB,MAAM,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC;YAC3C,QAAQR,SAAS,MAAM;YACvBC;YACAC;QACJ;QAEA,IAAIM,cAAc,MAAM,IACpB,OAAOZ,WAAW,QAAQ,CAAC,KAAK,CAAC;YAAE,SAASY,cAAc,KAAK,CAAC,OAAO;QAAC;QAG5E,OAAOZ,WAAW,QAAQ,CAAC,IAAI,CAAC;IACpC;AACJ;AAEO,MAAMa,wBAAwBC,eAAe,oBAAoB,CAAC;IACrE,gBAAgBnB;IAChB,cAAc;QAACoB;QAA+BC;QAA6BC;KAAG;AAClF"}
+{"version":3,"file":"api/features/AiImageEnrichment/AiImageEnrichmentTask.js","sources":["../../../../src/api/features/AiImageEnrichment/AiImageEnrichmentTask.ts"],"sourcesContent":["import {\n    TaskDefinition,\n    TaskHandler\n} from \"@webiny/api-core/features/task/TaskDefinition/index.js\";\nimport { Ai } from \"@webiny/api-core/features/ai/index.js\";\nimport { ApplyImageEnrichmentUseCase, PrepareImageEnrichmentUseCase } from \"./abstractions.js\";\nimport { buildEnrichmentAiRequest } from \"./buildEnrichmentAiRequest.js\";\nimport { EnrichmentNotAnImageError } from \"./errors.js\";\n\nexport const AI_IMAGE_ENRICHMENT_TASK_ID = \"fmAiImageEnrichment\";\n\nexport interface IAiImageEnrichmentTaskInput {\n    fileId: string;\n}\n\n/**\n * Background enrichment, triggered after a file is created. Shares its preparation and persistence\n * with the streaming HTTP route (`AiImageEnrichmentStreamRoute`); the only difference is that this\n * one waits for the whole AI response, because a background task has no one to stream to.\n */\nclass AiImageEnrichmentTaskHandlerImpl implements TaskHandler.Interface<IAiImageEnrichmentTaskInput> {\n    constructor(\n        private prepare: PrepareImageEnrichmentUseCase.Interface,\n        private apply: ApplyImageEnrichmentUseCase.Interface,\n        private ai: Ai.Interface\n    ) {}\n\n    async run({\n        input,\n        controller\n    }: TaskHandler.RunParams<IAiImageEnrichmentTaskInput>): Promise<\n        TaskDefinition.Result<IAiImageEnrichmentTaskInput>\n    > {\n        if (controller.runtime.isAborted()) {\n            return controller.response.aborted();\n        }\n\n        const preparedResult = await this.prepare.execute(input.fileId);\n        if (preparedResult.isFail()) {\n            const error = preparedResult.error;\n            // A non-image isn't a failure — nothing to enrich, so the task is simply done.\n            if (error instanceof EnrichmentNotAnImageError) {\n                return controller.response.done(error.message);\n            }\n            return controller.response.error({ message: error.message });\n        }\n\n        const prepared = preparedResult.value;\n\n        let tags: string[];\n        let description: string;\n        try {\n            const request = buildEnrichmentAiRequest(prepared);\n            const aiResult = await this.ai.generateText(request);\n\n            tags = aiResult.output.tags;\n            description = aiResult.output.description;\n        } catch (error) {\n            return controller.response.error({\n                message: `AI enrichment failed: ${error instanceof Error ? error.message : String(error)}`\n            });\n        }\n\n        const appliedResult = await this.apply.execute({\n            fileId: prepared.fileId,\n            tags,\n            description\n        });\n\n        if (appliedResult.isFail()) {\n            return controller.response.error({ message: appliedResult.error.message });\n        }\n\n        return controller.response.done(\"AI image enrichment completed successfully.\");\n    }\n}\n\nconst AiImageEnrichmentTaskHandler = TaskHandler.createImplementation({\n    implementation: AiImageEnrichmentTaskHandlerImpl,\n    dependencies: [PrepareImageEnrichmentUseCase, ApplyImageEnrichmentUseCase, Ai]\n});\n\nclass AiImageEnrichmentTaskImpl implements TaskDefinition.Interface {\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    public readonly selfCleanup = [\"onSuccess\" as const, \"onAbort\" as const];\n\n    handler = AiImageEnrichmentTaskHandler;\n}\n\nexport const AiImageEnrichmentTask = TaskDefinition.createImplementation({\n    implementation: AiImageEnrichmentTaskImpl,\n    dependencies: []\n});\n"],"names":["AI_IMAGE_ENRICHMENT_TASK_ID","AiImageEnrichmentTaskHandlerImpl","prepare","apply","ai","input","controller","preparedResult","error","EnrichmentNotAnImageError","prepared","tags","description","request","buildEnrichmentAiRequest","aiResult","Error","String","appliedResult","AiImageEnrichmentTaskHandler","TaskHandler","PrepareImageEnrichmentUseCase","ApplyImageEnrichmentUseCase","Ai","AiImageEnrichmentTaskImpl","AiImageEnrichmentTask","TaskDefinition"],"mappings":";;;;;AASO,MAAMA,8BAA8B;AAW3C,MAAMC;IACF,YACYC,OAAgD,EAChDC,KAA4C,EAC5CC,EAAgB,CAC1B;aAHUF,OAAO,GAAPA;aACAC,KAAK,GAALA;aACAC,EAAE,GAAFA;IACT;IAEH,MAAM,IAAI,EACNC,KAAK,EACLC,UAAU,EACuC,EAEnD;QACE,IAAIA,WAAW,OAAO,CAAC,SAAS,IAC5B,OAAOA,WAAW,QAAQ,CAAC,OAAO;QAGtC,MAAMC,iBAAiB,MAAM,IAAI,CAAC,OAAO,CAAC,OAAO,CAACF,MAAM,MAAM;QAC9D,IAAIE,eAAe,MAAM,IAAI;YACzB,MAAMC,QAAQD,eAAe,KAAK;YAElC,IAAIC,iBAAiBC,2BACjB,OAAOH,WAAW,QAAQ,CAAC,IAAI,CAACE,MAAM,OAAO;YAEjD,OAAOF,WAAW,QAAQ,CAAC,KAAK,CAAC;gBAAE,SAASE,MAAM,OAAO;YAAC;QAC9D;QAEA,MAAME,WAAWH,eAAe,KAAK;QAErC,IAAII;QACJ,IAAIC;QACJ,IAAI;YACA,MAAMC,UAAUC,yBAAyBJ;YACzC,MAAMK,WAAW,MAAM,IAAI,CAAC,EAAE,CAAC,YAAY,CAACF;YAE5CF,OAAOI,SAAS,MAAM,CAAC,IAAI;YAC3BH,cAAcG,SAAS,MAAM,CAAC,WAAW;QAC7C,EAAE,OAAOP,OAAO;YACZ,OAAOF,WAAW,QAAQ,CAAC,KAAK,CAAC;gBAC7B,SAAS,CAAC,sBAAsB,EAAEE,iBAAiBQ,QAAQR,MAAM,OAAO,GAAGS,OAAOT,QAAQ;YAC9F;QACJ;QAEA,MAAMU,gBAAgB,MAAM,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC;YAC3C,QAAQR,SAAS,MAAM;YACvBC;YACAC;QACJ;QAEA,IAAIM,cAAc,MAAM,IACpB,OAAOZ,WAAW,QAAQ,CAAC,KAAK,CAAC;YAAE,SAASY,cAAc,KAAK,CAAC,OAAO;QAAC;QAG5E,OAAOZ,WAAW,QAAQ,CAAC,IAAI,CAAC;IACpC;AACJ;AAEA,MAAMa,+BAA+BC,YAAY,oBAAoB,CAAC;IAClE,gBAAgBnB;IAChB,cAAc;QAACoB;QAA+BC;QAA6BC;KAAG;AAClF;AAEA,MAAMC;;aACF,EAAE,GAAGxB;aACL,KAAK,GAAG;aACR,WAAW,GAAG;aACd,aAAa,GAAG;aAChB,SAAS,GAAG;aACZ,YAAY,GAAG;aACC,WAAW,GAAG;YAAC;YAAsB;SAAmB;aAExE,OAAO,GAAGmB;;AACd;AAEO,MAAMM,wBAAwBC,eAAe,oBAAoB,CAAC;IACrE,gBAAgBF;IAChB,cAAc,EAAE;AACpB"}

package/api/features/CmsGenerateEntryContent/CmsGenerateEntryContentTask.d.ts

@@ -1,4 +1,4 @@
-import { TaskDefinition } from "@webiny/api-core/features/task/TaskDefinition/index.js";
+import { TaskDefinition, TaskHandler } from "@webiny/api-core/features/task/TaskDefinition/index.js";
 import { WebsocketsSendToIdentityUseCase } from "@webiny/api-websockets/features/SendToIdentity/abstractions.js";
 import { CmsGenerateEntryContentUseCase } from "../../../api/features/CmsGenerateEntryContent/index.js";
 import { IdentityContext } from "@webiny/api-core/exports/api/security.js";
@@ -12,20 +12,25 @@ export interface ICmsGenerateEntryContentTaskInput {
     writerPersonaId?: string | null;
     additionalFileIds?: string[] | null;
 }
-declare class CmsGenerateEntryContentTaskImpl implements TaskDefinition.Interface<ICmsGenerateEntryContentTaskInput> {
+declare class CmsGenerateEntryContentTaskHandlerImpl implements TaskHandler.Interface<ICmsGenerateEntryContentTaskInput> {
     private identityContext;
     private generateEntryContent;
     private sendToIdentity;
+    constructor(identityContext: IdentityContext.Interface, generateEntryContent: CmsGenerateEntryContentUseCase.Interface, sendToIdentity: WebsocketsSendToIdentityUseCase.Interface);
+    run({ input, controller }: TaskHandler.RunParams<ICmsGenerateEntryContentTaskInput>): Promise<TaskDefinition.Result<ICmsGenerateEntryContentTaskInput>>;
+    private sendContentToUser;
+    private sendErrorToUser;
+}
+declare class CmsGenerateEntryContentTaskImpl implements TaskDefinition.Interface {
     id: string;
     title: string;
     description: string;
     maxIterations: number;
     isPrivate: boolean;
     databaseLogs: boolean;
-    constructor(identityContext: IdentityContext.Interface, generateEntryContent: CmsGenerateEntryContentUseCase.Interface, sendToIdentity: WebsocketsSendToIdentityUseCase.Interface);
-    run({ input, controller }: TaskDefinition.RunParams<ICmsGenerateEntryContentTaskInput>): Promise<TaskDefinition.Result<ICmsGenerateEntryContentTaskInput>>;
-    private sendContentToUser;
-    private sendErrorToUser;
+    handler: typeof CmsGenerateEntryContentTaskHandlerImpl & {
+        __abstraction: import("@webiny/di").Abstraction<import("@webiny/api-core/features/task/TaskDefinition/abstractions").ITaskHandler<import("@webiny/api-core/features/task/TaskDefinition/abstractions").ITaskInput, import("@webiny/api-core/features/task/TaskDefinition/abstractions").ITaskOutput>>;
+    };
 }
 export declare const CmsGenerateEntryContentTask: typeof CmsGenerateEntryContentTaskImpl & {
     __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>>;

package/api/features/CmsGenerateEntryContent/CmsGenerateEntryContentTask.js

@@ -1,20 +1,14 @@
-import { TaskDefinition } from "@webiny/api-core/features/task/TaskDefinition/index.js";
+import { TaskDefinition, TaskHandler } from "@webiny/api-core/features/task/TaskDefinition/index.js";
 import { WebsocketsSendToIdentityUseCase } from "@webiny/api-websockets/features/SendToIdentity/abstractions.js";
 import { compressJson } from "@webiny/utils/features/compression/legacy/gzip.js";
 import { CmsGenerateEntryContentUseCase } from "./index.js";
 import { IdentityContext } from "@webiny/api-core/exports/api/security.js";
 const CMS_GENERATE_ENTRY_CONTENT_TASK_ID = "aiPowerUpsGenerateEntryContent";
-class CmsGenerateEntryContentTaskImpl {
+class CmsGenerateEntryContentTaskHandlerImpl {
     constructor(identityContext, generateEntryContent, sendToIdentity){
         this.identityContext = identityContext;
         this.generateEntryContent = generateEntryContent;
         this.sendToIdentity = sendToIdentity;
-        this.id = CMS_GENERATE_ENTRY_CONTENT_TASK_ID;
-        this.title = "AI Power-Ups - Generate Entry Content";
-        this.description = "Generates CMS entry content using AI based on a user prompt.";
-        this.maxIterations = 1;
-        this.isPrivate = true;
-        this.databaseLogs = false;
     }
     async run({ input, controller }) {
         if (controller.runtime.isAborted()) return controller.response.aborted();
@@ -71,14 +65,29 @@ class CmsGenerateEntryContentTaskImpl {
         });
     }
 }
-const CmsGenerateEntryContentTask = TaskDefinition.createImplementation({
-    implementation: CmsGenerateEntryContentTaskImpl,
+const CmsGenerateEntryContentTaskHandler = TaskHandler.createImplementation({
+    implementation: CmsGenerateEntryContentTaskHandlerImpl,
     dependencies: [
         IdentityContext,
         CmsGenerateEntryContentUseCase,
         WebsocketsSendToIdentityUseCase
     ]
 });
+class CmsGenerateEntryContentTaskImpl {
+    constructor(){
+        this.id = CMS_GENERATE_ENTRY_CONTENT_TASK_ID;
+        this.title = "AI Power-Ups - Generate Entry Content";
+        this.description = "Generates CMS entry content using AI based on a user prompt.";
+        this.maxIterations = 1;
+        this.isPrivate = true;
+        this.databaseLogs = false;
+        this.handler = CmsGenerateEntryContentTaskHandler;
+    }
+}
+const CmsGenerateEntryContentTask = TaskDefinition.createImplementation({
+    implementation: CmsGenerateEntryContentTaskImpl,
+    dependencies: []
+});
 export { CMS_GENERATE_ENTRY_CONTENT_TASK_ID, CmsGenerateEntryContentTask };
 
 //# sourceMappingURL=CmsGenerateEntryContentTask.js.map

package/api/features/CmsGenerateEntryContent/CmsGenerateEntryContentTask.js.map

@@ -1 +1 @@
-{"version":3,"file":"api/features/CmsGenerateEntryContent/CmsGenerateEntryContentTask.js","sources":["../../../../src/api/features/CmsGenerateEntryContent/CmsGenerateEntryContentTask.ts"],"sourcesContent":["import { TaskDefinition } from \"@webiny/api-core/features/task/TaskDefinition/index.js\";\nimport { WebsocketsSendToIdentityUseCase } from \"@webiny/api-websockets/features/SendToIdentity/abstractions.js\";\nimport { compressJson } from \"@webiny/utils/features/compression/legacy/gzip.js\";\nimport { CmsGenerateEntryContentUseCase } from \"~/api/features/CmsGenerateEntryContent/index.js\";\nimport type { GenerateEntryContentTelemetry } from \"~/api/features/CmsGenerateEntryContent/abstractions.js\";\nimport { IdentityContext } from \"@webiny/api-core/exports/api/security.js\";\n\nexport const CMS_GENERATE_ENTRY_CONTENT_TASK_ID = \"aiPowerUpsGenerateEntryContent\";\n\nexport interface ICmsGenerateEntryContentTaskInput {\n    prompt: string;\n    modelId: string;\n    projectId?: string | null;\n    excludedFileIds?: string[] | null;\n    readerPersonaId?: string | null;\n    writerPersonaId?: string | null;\n    additionalFileIds?: string[] | null;\n}\n\nclass CmsGenerateEntryContentTaskImpl implements TaskDefinition.Interface<ICmsGenerateEntryContentTaskInput> {\n    id = CMS_GENERATE_ENTRY_CONTENT_TASK_ID;\n    title = \"AI Power-Ups - Generate Entry Content\";\n    description = \"Generates CMS entry content using AI based on a user prompt.\";\n    maxIterations = 1;\n    isPrivate = true;\n    databaseLogs = false;\n\n    constructor(\n        private identityContext: IdentityContext.Interface,\n        private generateEntryContent: CmsGenerateEntryContentUseCase.Interface,\n        private sendToIdentity: WebsocketsSendToIdentityUseCase.Interface\n    ) {}\n\n    async run({\n        input,\n        controller\n    }: TaskDefinition.RunParams<ICmsGenerateEntryContentTaskInput>): Promise<\n        TaskDefinition.Result<ICmsGenerateEntryContentTaskInput>\n    > {\n        if (controller.runtime.isAborted()) {\n            return controller.response.aborted();\n        }\n\n        const identity = this.identityContext.getIdentity();\n\n        let result;\n        try {\n            result = await this.generateEntryContent.execute({\n                prompt: input.prompt,\n                modelId: input.modelId,\n                projectId: input.projectId,\n                excludedFileIds: input.excludedFileIds,\n                readerPersonaId: input.readerPersonaId,\n                writerPersonaId: input.writerPersonaId,\n                additionalFileIds: input.additionalFileIds\n            });\n        } catch (err) {\n            const message = err instanceof Error ? err.message : String(err);\n            await this.sendErrorToUser(identity.id, message);\n            return controller.response.error({ message });\n        }\n\n        if (result.isFail()) {\n            await this.sendErrorToUser(identity.id, result.error.message);\n\n            return controller.response.error({\n                message: result.error.message\n            });\n        }\n\n        // Serialize at the transport edge: the use case returns structured values; the\n        // websocket stream carries a gzip+base64 JSON string that the Admin app decodes.\n        const compressed = await compressJson(result.value.values);\n        const payload = compressed.toString(\"base64\");\n\n        await this.sendContentToUser(identity.id, payload, result.value.telemetry);\n\n        return controller.response.done(\"Entry content generated successfully.\");\n    }\n\n    private async sendContentToUser(\n        identityId: string,\n        payload: string,\n        telemetry: GenerateEntryContentTelemetry\n    ) {\n        await this.sendToIdentity.execute(\n            { id: identityId },\n            {\n                action: \"aiPowerUps.generateEntryContent.content\",\n                data: {\n                    compression: \"gzip\",\n                    value: payload,\n                    telemetry\n                }\n            }\n        );\n    }\n\n    private async sendErrorToUser(identityId: string, message: string) {\n        await this.sendToIdentity.execute(\n            { id: identityId },\n            {\n                action: \"aiPowerUps.generateEntryContent.error\",\n                data: {\n                    message\n                }\n            }\n        );\n    }\n}\n\nexport const CmsGenerateEntryContentTask = TaskDefinition.createImplementation({\n    implementation: CmsGenerateEntryContentTaskImpl,\n    dependencies: [IdentityContext, CmsGenerateEntryContentUseCase, WebsocketsSendToIdentityUseCase]\n});\n"],"names":["CMS_GENERATE_ENTRY_CONTENT_TASK_ID","CmsGenerateEntryContentTaskImpl","identityContext","generateEntryContent","sendToIdentity","input","controller","identity","result","err","message","Error","String","compressed","compressJson","payload","identityId","telemetry","CmsGenerateEntryContentTask","TaskDefinition","IdentityContext","CmsGenerateEntryContentUseCase","WebsocketsSendToIdentityUseCase"],"mappings":";;;;;AAOO,MAAMA,qCAAqC;AAYlD,MAAMC;IAQF,YACYC,eAA0C,EAC1CC,oBAA8D,EAC9DC,cAAyD,CACnE;aAHUF,eAAe,GAAfA;aACAC,oBAAoB,GAApBA;aACAC,cAAc,GAAdA;aAVZ,EAAE,GAAGJ;aACL,KAAK,GAAG;aACR,WAAW,GAAG;aACd,aAAa,GAAG;aAChB,SAAS,GAAG;aACZ,YAAY,GAAG;IAMZ;IAEH,MAAM,IAAI,EACNK,KAAK,EACLC,UAAU,EACgD,EAE5D;QACE,IAAIA,WAAW,OAAO,CAAC,SAAS,IAC5B,OAAOA,WAAW,QAAQ,CAAC,OAAO;QAGtC,MAAMC,WAAW,IAAI,CAAC,eAAe,CAAC,WAAW;QAEjD,IAAIC;QACJ,IAAI;YACAA,SAAS,MAAM,IAAI,CAAC,oBAAoB,CAAC,OAAO,CAAC;gBAC7C,QAAQH,MAAM,MAAM;gBACpB,SAASA,MAAM,OAAO;gBACtB,WAAWA,MAAM,SAAS;gBAC1B,iBAAiBA,MAAM,eAAe;gBACtC,iBAAiBA,MAAM,eAAe;gBACtC,iBAAiBA,MAAM,eAAe;gBACtC,mBAAmBA,MAAM,iBAAiB;YAC9C;QACJ,EAAE,OAAOI,KAAK;YACV,MAAMC,UAAUD,eAAeE,QAAQF,IAAI,OAAO,GAAGG,OAAOH;YAC5D,MAAM,IAAI,CAAC,eAAe,CAACF,SAAS,EAAE,EAAEG;YACxC,OAAOJ,WAAW,QAAQ,CAAC,KAAK,CAAC;gBAAEI;YAAQ;QAC/C;QAEA,IAAIF,OAAO,MAAM,IAAI;YACjB,MAAM,IAAI,CAAC,eAAe,CAACD,SAAS,EAAE,EAAEC,OAAO,KAAK,CAAC,OAAO;YAE5D,OAAOF,WAAW,QAAQ,CAAC,KAAK,CAAC;gBAC7B,SAASE,OAAO,KAAK,CAAC,OAAO;YACjC;QACJ;QAIA,MAAMK,aAAa,MAAMC,aAAaN,OAAO,KAAK,CAAC,MAAM;QACzD,MAAMO,UAAUF,WAAW,QAAQ,CAAC;QAEpC,MAAM,IAAI,CAAC,iBAAiB,CAACN,SAAS,EAAE,EAAEQ,SAASP,OAAO,KAAK,CAAC,SAAS;QAEzE,OAAOF,WAAW,QAAQ,CAAC,IAAI,CAAC;IACpC;IAEA,MAAc,kBACVU,UAAkB,EAClBD,OAAe,EACfE,SAAwC,EAC1C;QACE,MAAM,IAAI,CAAC,cAAc,CAAC,OAAO,CAC7B;YAAE,IAAID;QAAW,GACjB;YACI,QAAQ;YACR,MAAM;gBACF,aAAa;gBACb,OAAOD;gBACPE;YACJ;QACJ;IAER;IAEA,MAAc,gBAAgBD,UAAkB,EAAEN,OAAe,EAAE;QAC/D,MAAM,IAAI,CAAC,cAAc,CAAC,OAAO,CAC7B;YAAE,IAAIM;QAAW,GACjB;YACI,QAAQ;YACR,MAAM;gBACFN;YACJ;QACJ;IAER;AACJ;AAEO,MAAMQ,8BAA8BC,eAAe,oBAAoB,CAAC;IAC3E,gBAAgBlB;IAChB,cAAc;QAACmB;QAAiBC;QAAgCC;KAAgC;AACpG"}
+{"version":3,"file":"api/features/CmsGenerateEntryContent/CmsGenerateEntryContentTask.js","sources":["../../../../src/api/features/CmsGenerateEntryContent/CmsGenerateEntryContentTask.ts"],"sourcesContent":["import {\n    TaskDefinition,\n    TaskHandler\n} from \"@webiny/api-core/features/task/TaskDefinition/index.js\";\nimport { WebsocketsSendToIdentityUseCase } from \"@webiny/api-websockets/features/SendToIdentity/abstractions.js\";\nimport { compressJson } from \"@webiny/utils/features/compression/legacy/gzip.js\";\nimport { CmsGenerateEntryContentUseCase } from \"~/api/features/CmsGenerateEntryContent/index.js\";\nimport type { GenerateEntryContentTelemetry } from \"~/api/features/CmsGenerateEntryContent/abstractions.js\";\nimport { IdentityContext } from \"@webiny/api-core/exports/api/security.js\";\n\nexport const CMS_GENERATE_ENTRY_CONTENT_TASK_ID = \"aiPowerUpsGenerateEntryContent\";\n\nexport interface ICmsGenerateEntryContentTaskInput {\n    prompt: string;\n    modelId: string;\n    projectId?: string | null;\n    excludedFileIds?: string[] | null;\n    readerPersonaId?: string | null;\n    writerPersonaId?: string | null;\n    additionalFileIds?: string[] | null;\n}\n\nclass CmsGenerateEntryContentTaskHandlerImpl implements TaskHandler.Interface<ICmsGenerateEntryContentTaskInput> {\n    constructor(\n        private identityContext: IdentityContext.Interface,\n        private generateEntryContent: CmsGenerateEntryContentUseCase.Interface,\n        private sendToIdentity: WebsocketsSendToIdentityUseCase.Interface\n    ) {}\n\n    async run({\n        input,\n        controller\n    }: TaskHandler.RunParams<ICmsGenerateEntryContentTaskInput>): Promise<\n        TaskDefinition.Result<ICmsGenerateEntryContentTaskInput>\n    > {\n        if (controller.runtime.isAborted()) {\n            return controller.response.aborted();\n        }\n\n        const identity = this.identityContext.getIdentity();\n\n        let result;\n        try {\n            result = await this.generateEntryContent.execute({\n                prompt: input.prompt,\n                modelId: input.modelId,\n                projectId: input.projectId,\n                excludedFileIds: input.excludedFileIds,\n                readerPersonaId: input.readerPersonaId,\n                writerPersonaId: input.writerPersonaId,\n                additionalFileIds: input.additionalFileIds\n            });\n        } catch (err) {\n            const message = err instanceof Error ? err.message : String(err);\n            await this.sendErrorToUser(identity.id, message);\n            return controller.response.error({ message });\n        }\n\n        if (result.isFail()) {\n            await this.sendErrorToUser(identity.id, result.error.message);\n\n            return controller.response.error({\n                message: result.error.message\n            });\n        }\n\n        // Serialize at the transport edge: the use case returns structured values; the\n        // websocket stream carries a gzip+base64 JSON string that the Admin app decodes.\n        const compressed = await compressJson(result.value.values);\n        const payload = compressed.toString(\"base64\");\n\n        await this.sendContentToUser(identity.id, payload, result.value.telemetry);\n\n        return controller.response.done(\"Entry content generated successfully.\");\n    }\n\n    private async sendContentToUser(\n        identityId: string,\n        payload: string,\n        telemetry: GenerateEntryContentTelemetry\n    ) {\n        await this.sendToIdentity.execute(\n            { id: identityId },\n            {\n                action: \"aiPowerUps.generateEntryContent.content\",\n                data: {\n                    compression: \"gzip\",\n                    value: payload,\n                    telemetry\n                }\n            }\n        );\n    }\n\n    private async sendErrorToUser(identityId: string, message: string) {\n        await this.sendToIdentity.execute(\n            { id: identityId },\n            {\n                action: \"aiPowerUps.generateEntryContent.error\",\n                data: {\n                    message\n                }\n            }\n        );\n    }\n}\n\nconst CmsGenerateEntryContentTaskHandler = TaskHandler.createImplementation({\n    implementation: CmsGenerateEntryContentTaskHandlerImpl,\n    dependencies: [IdentityContext, CmsGenerateEntryContentUseCase, WebsocketsSendToIdentityUseCase]\n});\n\nclass CmsGenerateEntryContentTaskImpl implements TaskDefinition.Interface {\n    id = CMS_GENERATE_ENTRY_CONTENT_TASK_ID;\n    title = \"AI Power-Ups - Generate Entry Content\";\n    description = \"Generates CMS entry content using AI based on a user prompt.\";\n    maxIterations = 1;\n    isPrivate = true;\n    databaseLogs = false;\n\n    handler = CmsGenerateEntryContentTaskHandler;\n}\n\nexport const CmsGenerateEntryContentTask = TaskDefinition.createImplementation({\n    implementation: CmsGenerateEntryContentTaskImpl,\n    dependencies: []\n});\n"],"names":["CMS_GENERATE_ENTRY_CONTENT_TASK_ID","CmsGenerateEntryContentTaskHandlerImpl","identityContext","generateEntryContent","sendToIdentity","input","controller","identity","result","err","message","Error","String","compressed","compressJson","payload","identityId","telemetry","CmsGenerateEntryContentTaskHandler","TaskHandler","IdentityContext","CmsGenerateEntryContentUseCase","WebsocketsSendToIdentityUseCase","CmsGenerateEntryContentTaskImpl","CmsGenerateEntryContentTask","TaskDefinition"],"mappings":";;;;;AAUO,MAAMA,qCAAqC;AAYlD,MAAMC;IACF,YACYC,eAA0C,EAC1CC,oBAA8D,EAC9DC,cAAyD,CACnE;aAHUF,eAAe,GAAfA;aACAC,oBAAoB,GAApBA;aACAC,cAAc,GAAdA;IACT;IAEH,MAAM,IAAI,EACNC,KAAK,EACLC,UAAU,EAC6C,EAEzD;QACE,IAAIA,WAAW,OAAO,CAAC,SAAS,IAC5B,OAAOA,WAAW,QAAQ,CAAC,OAAO;QAGtC,MAAMC,WAAW,IAAI,CAAC,eAAe,CAAC,WAAW;QAEjD,IAAIC;QACJ,IAAI;YACAA,SAAS,MAAM,IAAI,CAAC,oBAAoB,CAAC,OAAO,CAAC;gBAC7C,QAAQH,MAAM,MAAM;gBACpB,SAASA,MAAM,OAAO;gBACtB,WAAWA,MAAM,SAAS;gBAC1B,iBAAiBA,MAAM,eAAe;gBACtC,iBAAiBA,MAAM,eAAe;gBACtC,iBAAiBA,MAAM,eAAe;gBACtC,mBAAmBA,MAAM,iBAAiB;YAC9C;QACJ,EAAE,OAAOI,KAAK;YACV,MAAMC,UAAUD,eAAeE,QAAQF,IAAI,OAAO,GAAGG,OAAOH;YAC5D,MAAM,IAAI,CAAC,eAAe,CAACF,SAAS,EAAE,EAAEG;YACxC,OAAOJ,WAAW,QAAQ,CAAC,KAAK,CAAC;gBAAEI;YAAQ;QAC/C;QAEA,IAAIF,OAAO,MAAM,IAAI;YACjB,MAAM,IAAI,CAAC,eAAe,CAACD,SAAS,EAAE,EAAEC,OAAO,KAAK,CAAC,OAAO;YAE5D,OAAOF,WAAW,QAAQ,CAAC,KAAK,CAAC;gBAC7B,SAASE,OAAO,KAAK,CAAC,OAAO;YACjC;QACJ;QAIA,MAAMK,aAAa,MAAMC,aAAaN,OAAO,KAAK,CAAC,MAAM;QACzD,MAAMO,UAAUF,WAAW,QAAQ,CAAC;QAEpC,MAAM,IAAI,CAAC,iBAAiB,CAACN,SAAS,EAAE,EAAEQ,SAASP,OAAO,KAAK,CAAC,SAAS;QAEzE,OAAOF,WAAW,QAAQ,CAAC,IAAI,CAAC;IACpC;IAEA,MAAc,kBACVU,UAAkB,EAClBD,OAAe,EACfE,SAAwC,EAC1C;QACE,MAAM,IAAI,CAAC,cAAc,CAAC,OAAO,CAC7B;YAAE,IAAID;QAAW,GACjB;YACI,QAAQ;YACR,MAAM;gBACF,aAAa;gBACb,OAAOD;gBACPE;YACJ;QACJ;IAER;IAEA,MAAc,gBAAgBD,UAAkB,EAAEN,OAAe,EAAE;QAC/D,MAAM,IAAI,CAAC,cAAc,CAAC,OAAO,CAC7B;YAAE,IAAIM;QAAW,GACjB;YACI,QAAQ;YACR,MAAM;gBACFN;YACJ;QACJ;IAER;AACJ;AAEA,MAAMQ,qCAAqCC,YAAY,oBAAoB,CAAC;IACxE,gBAAgBlB;IAChB,cAAc;QAACmB;QAAiBC;QAAgCC;KAAgC;AACpG;AAEA,MAAMC;;aACF,EAAE,GAAGvB;aACL,KAAK,GAAG;aACR,WAAW,GAAG;aACd,aAAa,GAAG;aAChB,SAAS,GAAG;aACZ,YAAY,GAAG;aAEf,OAAO,GAAGkB;;AACd;AAEO,MAAMM,8BAA8BC,eAAe,oBAAoB,CAAC;IAC3E,gBAAgBF;IAChB,cAAc,EAAE;AACpB"}

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

@@ -1,4 +1,4 @@
-import { TaskDefinition } from "@webiny/api-core/features/task/TaskDefinition/index.js";
+import { TaskDefinition, TaskHandler } from "@webiny/api-core/features/task/TaskDefinition/index.js";
 import { WebsocketsSendToIdentityUseCase } from "@webiny/api-websockets/features/SendToIdentity/abstractions.js";
 import { WbGeneratePageContentUseCase } from "../../../api/features/WbGeneratePageContent/index.js";
 import { IdentityContext } from "@webiny/api-core/exports/api/security.js";
@@ -13,20 +13,25 @@ export interface IWbGeneratePageContentTaskInput {
     writerPersonaId?: string | null;
     additionalFileIds?: string[] | null;
 }
-declare class WbGeneratePageContentTaskImpl implements TaskDefinition.Interface<IWbGeneratePageContentTaskInput> {
+declare class WbGeneratePageContentTaskHandlerImpl implements TaskHandler.Interface<IWbGeneratePageContentTaskInput> {
     private identityContext;
     private generatePageContent;
     private sendToIdentity;
+    constructor(identityContext: IdentityContext.Interface, generatePageContent: WbGeneratePageContentUseCase.Interface, sendToIdentity: WebsocketsSendToIdentityUseCase.Interface);
+    run({ input, controller }: TaskHandler.RunParams<IWbGeneratePageContentTaskInput>): Promise<TaskDefinition.Result<IWbGeneratePageContentTaskInput>>;
+    private sendContentToUser;
+    private sendErrorToUser;
+}
+declare class WbGeneratePageContentTaskImpl implements TaskDefinition.Interface {
     id: string;
     title: string;
     description: string;
     maxIterations: number;
     isPrivate: boolean;
     databaseLogs: boolean;
-    constructor(identityContext: IdentityContext.Interface, generatePageContent: WbGeneratePageContentUseCase.Interface, sendToIdentity: WebsocketsSendToIdentityUseCase.Interface);
-    run({ input, controller }: TaskDefinition.RunParams<IWbGeneratePageContentTaskInput>): Promise<TaskDefinition.Result<IWbGeneratePageContentTaskInput>>;
-    private sendContentToUser;
-    private sendErrorToUser;
+    handler: typeof WbGeneratePageContentTaskHandlerImpl & {
+        __abstraction: import("@webiny/di").Abstraction<import("@webiny/api-core/features/task/TaskDefinition/abstractions").ITaskHandler<import("@webiny/api-core/features/task/TaskDefinition/abstractions").ITaskInput, import("@webiny/api-core/features/task/TaskDefinition/abstractions").ITaskOutput>>;
+    };
 }
 export declare const WbGeneratePageContentTask: typeof WbGeneratePageContentTaskImpl & {
     __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>>;

package/api/features/WbGeneratePageContent/WbGeneratePageContentTask.js

@@ -1,20 +1,14 @@
-import { TaskDefinition } from "@webiny/api-core/features/task/TaskDefinition/index.js";
+import { TaskDefinition, TaskHandler } from "@webiny/api-core/features/task/TaskDefinition/index.js";
 import { WebsocketsSendToIdentityUseCase } from "@webiny/api-websockets/features/SendToIdentity/abstractions.js";
 import { compress } from "@webiny/utils/features/compression/legacy/gzip.js";
 import { WbGeneratePageContentUseCase } from "./index.js";
 import { IdentityContext } from "@webiny/api-core/exports/api/security.js";
 const WB_GENERATE_PAGE_CONTENT_TASK_ID = "aiPowerUpsGeneratePageContent";
-class WbGeneratePageContentTaskImpl {
+class WbGeneratePageContentTaskHandlerImpl {
     constructor(identityContext, generatePageContent, sendToIdentity){
         this.identityContext = identityContext;
         this.generatePageContent = generatePageContent;
         this.sendToIdentity = sendToIdentity;
-        this.id = WB_GENERATE_PAGE_CONTENT_TASK_ID;
-        this.title = "AI Power-Ups - Generate Page Content";
-        this.description = "Generates page content using AI based on a user prompt.";
-        this.maxIterations = 1;
-        this.isPrivate = true;
-        this.databaseLogs = false;
     }
     async run({ input, controller }) {
         if (controller.runtime.isAborted()) return controller.response.aborted();
@@ -72,14 +66,29 @@ class WbGeneratePageContentTaskImpl {
         });
     }
 }
-const WbGeneratePageContentTask = TaskDefinition.createImplementation({
-    implementation: WbGeneratePageContentTaskImpl,
+const WbGeneratePageContentTaskHandler = TaskHandler.createImplementation({
+    implementation: WbGeneratePageContentTaskHandlerImpl,
     dependencies: [
         IdentityContext,
         WbGeneratePageContentUseCase,
         WebsocketsSendToIdentityUseCase
     ]
 });
+class WbGeneratePageContentTaskImpl {
+    constructor(){
+        this.id = WB_GENERATE_PAGE_CONTENT_TASK_ID;
+        this.title = "AI Power-Ups - Generate Page Content";
+        this.description = "Generates page content using AI based on a user prompt.";
+        this.maxIterations = 1;
+        this.isPrivate = true;
+        this.databaseLogs = false;
+        this.handler = WbGeneratePageContentTaskHandler;
+    }
+}
+const WbGeneratePageContentTask = TaskDefinition.createImplementation({
+    implementation: WbGeneratePageContentTaskImpl,
+    dependencies: []
+});
 export { WB_GENERATE_PAGE_CONTENT_TASK_ID, WbGeneratePageContentTask };
 
 //# sourceMappingURL=WbGeneratePageContentTask.js.map

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

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

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@webiny/ai-powerups",
-	"version": "6.6.0-alpha.2",
+	"version": "6.6.0-alpha.4",
 	"type": "module",
 	"sideEffects": false,
 	"exports": {
@@ -18,28 +18,28 @@
 	"author": "Webiny Ltd.",
 	"license": "Webiny Enterprise",
 	"dependencies": {
-		"@webiny/admin-ui": "6.6.0-alpha.2",
-		"@webiny/ai-chat": "6.6.0-alpha.2",
-		"@webiny/api-core": "6.6.0-alpha.2",
-		"@webiny/api-file-manager": "6.6.0-alpha.2",
-		"@webiny/api-headless-cms": "6.6.0-alpha.2",
-		"@webiny/api-website-builder": "6.6.0-alpha.2",
-		"@webiny/api-websockets": "6.6.0-alpha.2",
-		"@webiny/app": "6.6.0-alpha.2",
-		"@webiny/app-admin": "6.6.0-alpha.2",
-		"@webiny/app-headless-cms": "6.6.0-alpha.2",
-		"@webiny/app-website-builder": "6.6.0-alpha.2",
-		"@webiny/app-websockets": "6.6.0-alpha.2",
-		"@webiny/background-tasks": "6.6.0-alpha.2",
+		"@webiny/admin-ui": "6.6.0-alpha.4",
+		"@webiny/ai-chat": "6.6.0-alpha.4",
+		"@webiny/api-core": "6.6.0-alpha.4",
+		"@webiny/api-file-manager": "6.6.0-alpha.4",
+		"@webiny/api-headless-cms": "6.6.0-alpha.4",
+		"@webiny/api-website-builder": "6.6.0-alpha.4",
+		"@webiny/api-websockets": "6.6.0-alpha.4",
+		"@webiny/app": "6.6.0-alpha.4",
+		"@webiny/app-admin": "6.6.0-alpha.4",
+		"@webiny/app-headless-cms": "6.6.0-alpha.4",
+		"@webiny/app-website-builder": "6.6.0-alpha.4",
+		"@webiny/app-websockets": "6.6.0-alpha.4",
+		"@webiny/background-tasks": "6.6.0-alpha.4",
 		"@webiny/di": "1.1.0",
-		"@webiny/event-handler-core": "6.6.0-alpha.2",
-		"@webiny/feature": "6.6.0-alpha.2",
-		"@webiny/icons": "6.6.0-alpha.2",
-		"@webiny/languages": "6.6.0-alpha.2",
-		"@webiny/lexical-converter": "6.6.0-alpha.2",
-		"@webiny/project": "6.6.0-alpha.2",
-		"@webiny/project-aws": "6.6.0-alpha.2",
-		"@webiny/utils": "6.6.0-alpha.2",
+		"@webiny/event-handler-core": "6.6.0-alpha.4",
+		"@webiny/feature": "6.6.0-alpha.4",
+		"@webiny/icons": "6.6.0-alpha.4",
+		"@webiny/languages": "6.6.0-alpha.4",
+		"@webiny/lexical-converter": "6.6.0-alpha.4",
+		"@webiny/project": "6.6.0-alpha.4",
+		"@webiny/project-aws": "6.6.0-alpha.4",
+		"@webiny/utils": "6.6.0-alpha.4",
 		"ai": "7.0.73",
 		"front-matter": "4.0.2",
 		"happy-dom": "20.11.6",
@@ -51,10 +51,10 @@
 		"zod": "4.4.3"
 	},
 	"devDependencies": {
-		"@webiny/api-graphql": "6.6.0-alpha.2",
-		"@webiny/build-tools": "6.6.0-alpha.2",
-		"@webiny/feature-flags": "6.6.0-alpha.2",
-		"@webiny/wcp": "6.6.0-alpha.2",
+		"@webiny/api-graphql": "6.6.0-alpha.4",
+		"@webiny/build-tools": "6.6.0-alpha.4",
+		"@webiny/feature-flags": "6.6.0-alpha.4",
+		"@webiny/wcp": "6.6.0-alpha.4",
 		"rimraf": "6.1.3",
 		"typescript": "7.0.2",
 		"vitest": "4.1.11"