@wix/ditto-codegen-public 1.0.223 → 1.0.225

package/dist/out.js

@@ -105088,48 +105088,6 @@ export default appBuilder;
   }
 });
 
-// dist/orchestrator-error-helpers.js
-var require_orchestrator_error_helpers = __commonJS({
-  "dist/orchestrator-error-helpers.js"(exports2) {
-    "use strict";
-    Object.defineProperty(exports2, "__esModule", { value: true });
-    exports2.createExtensionErrorHandler = createExtensionErrorHandler;
-    exports2.throwIfFailures = throwIfFailures;
-    var ditto_codegen_types_12 = require_dist();
-    function createExtensionErrorHandler(eventEmitter, extension) {
-      return (error) => {
-        const codegenError = (0, ditto_codegen_types_12.toCodegenError)(error);
-        eventEmitter.emitEvent("agent:error", {
-          extension,
-          error: codegenError
-        });
-        console.error(`\u274C Failed to process extension ${extension.name}:`, error);
-        throw codegenError;
-      };
-    }
-    function throwIfFailures(results) {
-      const failures = results.map((result, index) => {
-        if (result.status === "rejected") {
-          return {
-            index,
-            error: result.reason
-          };
-        }
-        return null;
-      }).filter((f) => f !== null);
-      if (failures.length > 0) {
-        const errorMessages = failures.map((f) => {
-          const codegenError = (0, ditto_codegen_types_12.toCodegenError)(f.error);
-          return `Task ${f.index} failed: ${codegenError.message}`;
-        });
-        const message = `Failed to process ${failures.length} of ${results.length} parallel tasks:
-${errorMessages.join("\n")}`;
-        throw (0, ditto_codegen_types_12.toCodegenError)(new Error(message));
-      }
-    }
-  }
-});
-
 // dist/utils.js
 var require_utils15 = __commonJS({
   "dist/utils.js"(exports2) {
@@ -105202,6 +105160,337 @@ var require_utils15 = __commonJS({
   }
 });
 
+// ../../node_modules/@wix/ambassador-ctp-ditto-v1-blueprint/build/cjs/http.impl.js
+var require_http_impl2 = __commonJS({
+  "../../node_modules/@wix/ambassador-ctp-ditto-v1-blueprint/build/cjs/http.impl.js"(exports2) {
+    "use strict";
+    Object.defineProperty(exports2, "__esModule", { value: true });
+    exports2.getRequiredPermissions = exports2.editBlueprint = exports2.createAppName = exports2.createBlueprint = void 0;
+    var ambassador_1 = require_ambassador_index();
+    var metro_runtime_1 = require_cjs2();
+    var _createAppNameRequest = {};
+    var _createAppNameResponse = {};
+    var _createBlueprintRequest = {};
+    var _createBlueprintResponse = {};
+    var _editBlueprintRequest = {};
+    var _editBlueprintResponse = {};
+    var _getRequiredPermissionsRequest = {};
+    var _getRequiredPermissionsResponse = {};
+    function resolveComWixpressDittoBlueprintServiceApiDittoBlueprintServiceUrl(opts) {
+      var domainToMappings = {
+        "manage._base_domain_": [
+          {
+            srcPath: "/_api/ditto-blueprint-service-web",
+            destPath: ""
+          }
+        ]
+      };
+      return (0, metro_runtime_1.resolveUrl)(Object.assign(opts, { domainToMappings }));
+    }
+    function createBlueprint(payload) {
+      var _a = (0, ambassador_1.serializer)(_createBlueprintRequest, {}), toReq = _a.toJSON, fromReq = _a.fromJSON;
+      var fromRes = (0, ambassador_1.serializer)(_createBlueprintResponse, {}).fromJSON;
+      function __createBlueprint(_a2) {
+        var host = _a2.host;
+        var serializedData = toReq(payload);
+        var metadata = {
+          entityFqdn: "wix.ctp.ditto.v1.blueprint",
+          method: "POST",
+          methodFqn: "com.wixpress.ditto_blueprint_service.api.DittoBlueprintService.CreateBlueprint",
+          migrationOptions: {
+            optInTransformResponse: true
+          },
+          url: resolveComWixpressDittoBlueprintServiceApiDittoBlueprintServiceUrl({
+            protoPath: "/v1/blueprints",
+            data: serializedData,
+            host
+          }),
+          data: serializedData,
+          transformResponse: fromRes
+        };
+        return metadata;
+      }
+      __createBlueprint.fromReq = fromReq;
+      __createBlueprint.__isAmbassador = true;
+      return __createBlueprint;
+    }
+    exports2.createBlueprint = createBlueprint;
+    function createAppName(payload) {
+      var _a = (0, ambassador_1.serializer)(_createAppNameRequest, {}), toReq = _a.toJSON, fromReq = _a.fromJSON;
+      var fromRes = (0, ambassador_1.serializer)(_createAppNameResponse, {}).fromJSON;
+      function __createAppName(_a2) {
+        var host = _a2.host;
+        var serializedData = toReq(payload);
+        var metadata = {
+          entityFqdn: "wix.ctp.ditto.v1.blueprint",
+          method: "POST",
+          methodFqn: "com.wixpress.ditto_blueprint_service.api.DittoBlueprintService.CreateAppName",
+          migrationOptions: {
+            optInTransformResponse: true
+          },
+          url: resolveComWixpressDittoBlueprintServiceApiDittoBlueprintServiceUrl({
+            protoPath: "/v1/blueprints/app-name",
+            data: serializedData,
+            host
+          }),
+          data: serializedData,
+          transformResponse: fromRes
+        };
+        return metadata;
+      }
+      __createAppName.fromReq = fromReq;
+      __createAppName.__isAmbassador = true;
+      return __createAppName;
+    }
+    exports2.createAppName = createAppName;
+    function editBlueprint(payload) {
+      var _a = (0, ambassador_1.serializer)(_editBlueprintRequest, {}), toReq = _a.toJSON, fromReq = _a.fromJSON;
+      var fromRes = (0, ambassador_1.serializer)(_editBlueprintResponse, {}).fromJSON;
+      function __editBlueprint(_a2) {
+        var host = _a2.host;
+        var serializedData = toReq(payload);
+        var metadata = {
+          entityFqdn: "wix.ctp.ditto.v1.blueprint",
+          method: "POST",
+          methodFqn: "com.wixpress.ditto_blueprint_service.api.DittoBlueprintService.EditBlueprint",
+          migrationOptions: {
+            optInTransformResponse: true
+          },
+          url: resolveComWixpressDittoBlueprintServiceApiDittoBlueprintServiceUrl({
+            protoPath: "/v1/blueprints/edit",
+            data: serializedData,
+            host
+          }),
+          data: serializedData,
+          transformResponse: fromRes
+        };
+        return metadata;
+      }
+      __editBlueprint.fromReq = fromReq;
+      __editBlueprint.__isAmbassador = true;
+      return __editBlueprint;
+    }
+    exports2.editBlueprint = editBlueprint;
+    function getRequiredPermissions(payload) {
+      var _a = (0, ambassador_1.serializer)(_getRequiredPermissionsRequest, {}), toReq = _a.toJSON, fromReq = _a.fromJSON;
+      var fromRes = (0, ambassador_1.serializer)(_getRequiredPermissionsResponse, {}).fromJSON;
+      function __getRequiredPermissions(_a2) {
+        var host = _a2.host;
+        var serializedData = toReq(payload);
+        var metadata = {
+          entityFqdn: "wix.ctp.ditto.v1.blueprint",
+          method: "POST",
+          methodFqn: "com.wixpress.ditto_blueprint_service.api.DittoBlueprintService.GetRequiredPermissions",
+          migrationOptions: {
+            optInTransformResponse: true
+          },
+          url: resolveComWixpressDittoBlueprintServiceApiDittoBlueprintServiceUrl({
+            protoPath: "/v1/blueprints/required-permissions",
+            data: serializedData,
+            host
+          }),
+          data: serializedData,
+          transformResponse: fromRes
+        };
+        return metadata;
+      }
+      __getRequiredPermissions.fromReq = fromReq;
+      __getRequiredPermissions.__isAmbassador = true;
+      return __getRequiredPermissions;
+    }
+    exports2.getRequiredPermissions = getRequiredPermissions;
+  }
+});
+
+// dist/flows/codegen-flow-helpers.js
+var require_codegen_flow_helpers = __commonJS({
+  "dist/flows/codegen-flow-helpers.js"(exports2) {
+    "use strict";
+    var __importDefault2 = exports2 && exports2.__importDefault || function(mod) {
+      return mod && mod.__esModule ? mod : { "default": mod };
+    };
+    Object.defineProperty(exports2, "__esModule", { value: true });
+    exports2.addAgentTask = addAgentTask;
+    exports2.completeAgentTask = completeAgentTask;
+    exports2.failAgentTask = failAgentTask;
+    exports2.addUpdateExtensionAgentTask = addUpdateExtensionAgentTask;
+    exports2.completeUpdateExtensionAgentTask = completeUpdateExtensionAgentTask;
+    exports2.failUpdateExtensionAgentTask = failUpdateExtensionAgentTask;
+    exports2.updateJobPayload = updateJobPayload;
+    exports2.updateParentTaskStatus = updateParentTaskStatus;
+    exports2.getOutputPath = getOutputPath;
+    exports2.getRequiredPermissions = getRequiredPermissions;
+    var path_1 = __importDefault2(require("path"));
+    var codeGenerationService_12 = require_codeGenerationService();
+    var ditto_codegen_types_12 = require_dist();
+    var utils_1 = require_utils15();
+    var http_1 = require_http_impl2();
+    var http_client_1 = require_index_node();
+    var orchestrator_1 = require_orchestrator();
+    var job_context_storage_12 = require_job_context_storage();
+    var slugify = (str = "") => {
+      return str.toLowerCase().replace(/ /g, "-");
+    };
+    function getJobContext() {
+      const store = job_context_storage_12.jobContextStorage.getStore();
+      if (!store?.jobId || !store?.taskId) {
+        return null;
+      }
+      return { jobId: store.jobId, taskId: store.taskId };
+    }
+    function getAgentTaskId(parentTaskId, extension) {
+      return `${parentTaskId}-${slugify(extension.name)}`;
+    }
+    async function addAgentTask(extension, agentName) {
+      const jobContext = getJobContext();
+      if (!jobContext) {
+        console.log(`[Agent] start: ${extension.name} (no job context)`);
+        return;
+      }
+      console.log(`[Agent] start: ${extension.name}`);
+      const taskId = getAgentTaskId(jobContext.taskId, extension);
+      await codeGenerationService_12.codeGenerationService.addTask(jobContext.jobId, {
+        id: taskId,
+        kind: ditto_codegen_types_12.TaskKind.RUN_AGENT,
+        status: ditto_codegen_types_12.Status.RUNNING,
+        name: "run_agent",
+        description: `Run agent for ${extension.name}`,
+        payload: { extension },
+        agentName
+      });
+    }
+    async function completeAgentTask(extension, files) {
+      const jobContext = getJobContext();
+      if (!jobContext) {
+        console.log(`[Agent] done: ${extension.name} (no job context)`);
+        return;
+      }
+      console.log(`[Agent] done: ${extension.name}`);
+      const taskId = getAgentTaskId(jobContext.taskId, extension);
+      await codeGenerationService_12.codeGenerationService.updateTask(jobContext.jobId, taskId, ditto_codegen_types_12.Status.COMPLETED, { taskOutput: { files } });
+    }
+    async function failAgentTask(extension, error) {
+      const jobContext = getJobContext();
+      if (!jobContext) {
+        console.error(`[Agent] error: ${extension.name} (no job context)`, error);
+        return;
+      }
+      console.error(`[Agent] error: ${extension.name}`, error);
+      const taskId = getAgentTaskId(jobContext.taskId, extension);
+      await codeGenerationService_12.codeGenerationService.updateTask(jobContext.jobId, taskId, ditto_codegen_types_12.Status.FAILED, { error: (0, utils_1.serializeError)(error) });
+    }
+    async function addUpdateExtensionAgentTask() {
+      const jobContext = getJobContext();
+      if (!jobContext) {
+        console.log(`[Agent] start: iterating existing code (no job context)`);
+        return;
+      }
+      console.log(`[Agent] start: iterating existing code`);
+      const taskId = `${jobContext.taskId}-update-extension-agent`;
+      await codeGenerationService_12.codeGenerationService.addTask(jobContext.jobId, {
+        id: taskId,
+        kind: ditto_codegen_types_12.TaskKind.RUN_AGENT,
+        status: ditto_codegen_types_12.Status.RUNNING,
+        name: "run_agent",
+        description: "Iterating existing code",
+        agentName: "UpdateExtensionAgent"
+      });
+    }
+    async function completeUpdateExtensionAgentTask(files) {
+      const jobContext = getJobContext();
+      if (!jobContext) {
+        console.log(`[Agent] done: iterating existing code (no job context)`);
+        return;
+      }
+      console.log(`[Agent] done: iterating existing code`);
+      const taskId = `${jobContext.taskId}-update-extension-agent`;
+      await codeGenerationService_12.codeGenerationService.updateTask(jobContext.jobId, taskId, ditto_codegen_types_12.Status.COMPLETED, { taskOutput: { files } });
+    }
+    async function failUpdateExtensionAgentTask(error) {
+      const jobContext = getJobContext();
+      if (!jobContext) {
+        console.error(`[Agent] error: iterating existing code (no job context)`, error);
+        return;
+      }
+      console.error(`[Agent] error: iterating existing code`, error);
+      const taskId = `${jobContext.taskId}-update-extension-agent`;
+      await codeGenerationService_12.codeGenerationService.updateTask(jobContext.jobId, taskId, ditto_codegen_types_12.Status.FAILED, { error: (0, utils_1.serializeError)(error) });
+    }
+    async function updateJobPayload(jobContext, payload) {
+      await codeGenerationService_12.codeGenerationService.updateJob(jobContext.jobId, payload);
+    }
+    async function updateParentTaskStatus(jobContext, status, error) {
+      await codeGenerationService_12.codeGenerationService.updateTask(jobContext.jobId, jobContext.taskId, status, error ? { error: (0, utils_1.serializeError)(error) } : {});
+    }
+    function getOutputPath() {
+      const outputDir = process.env.OUTPUT_PATH || orchestrator_1.DittoOrchestrator.DEFAULT_OUTPUT_PATH;
+      return outputDir.startsWith("/") ? outputDir : path_1.default.join(process.cwd(), outputDir);
+    }
+    async function getRequiredPermissions(extensions) {
+      const httpClient = (0, http_client_1.createHttpClient)({
+        baseURL: "https://manage.wix.com"
+      });
+      return httpClient.request((0, http_1.getRequiredPermissions)({
+        relatedApis: getRelatedApis(extensions ?? []),
+        extensionTypes: getExtensionTypes(extensions ?? [])
+      }));
+    }
+    function getExtensionTypes(extensions) {
+      return Array.from(new Set((extensions ?? []).map(({ type }) => type).filter(Boolean)));
+    }
+    function getRelatedApis(extensions) {
+      return Array.from(new Set((extensions ?? []).flatMap((ext) => [
+        ...(ext.relatedApis ?? []).map(({ name }) => name).filter(Boolean),
+        ...(ext.relatedSpis ?? []).map(({ name }) => name).filter(Boolean)
+      ])));
+    }
+  }
+});
+
+// dist/orchestrator-error-helpers.js
+var require_orchestrator_error_helpers = __commonJS({
+  "dist/orchestrator-error-helpers.js"(exports2) {
+    "use strict";
+    Object.defineProperty(exports2, "__esModule", { value: true });
+    exports2.createExtensionErrorHandler = createExtensionErrorHandler;
+    exports2.throwIfFailures = throwIfFailures;
+    var ditto_codegen_types_12 = require_dist();
+    var codegen_flow_helpers_1 = require_codegen_flow_helpers();
+    function createExtensionErrorHandler(eventEmitter, extension) {
+      return async (error) => {
+        const codegenError = (0, ditto_codegen_types_12.toCodegenError)(error);
+        await (0, codegen_flow_helpers_1.failAgentTask)(extension, codegenError);
+        eventEmitter.emitEvent("agent:error", {
+          extension,
+          error: codegenError
+        });
+        console.error(`\u274C Failed to process extension ${extension.name}:`, error);
+        throw codegenError;
+      };
+    }
+    function throwIfFailures(results) {
+      const failures = results.map((result, index) => {
+        if (result.status === "rejected") {
+          return {
+            index,
+            error: result.reason
+          };
+        }
+        return null;
+      }).filter((f) => f !== null);
+      if (failures.length > 0) {
+        const errorMessages = failures.map((f) => {
+          const codegenError = (0, ditto_codegen_types_12.toCodegenError)(f.error);
+          return `Task ${f.index} failed: ${codegenError.message}`;
+        });
+        const message = `Failed to process ${failures.length} of ${results.length} parallel tasks:
+${errorMessages.join("\n")}`;
+        throw (0, ditto_codegen_types_12.toCodegenError)(new Error(message));
+      }
+    }
+  }
+});
+
 // dist/validators/TypescriptValidator.js
 var require_TypescriptValidator = __commonJS({
   "dist/validators/TypescriptValidator.js"(exports2) {
@@ -105428,7 +105717,7 @@ var require_BuildFixerAgent = __commonJS({
 WORKFLOW:
 1. Analyze the build output to identify the error type and files involved
 2. Read the relevant file(s) to understand the context
-3. Only AFTER reading, decide what to fix and submit
+3. ALWAYS call submit_fix with your fix - this is REQUIRED
 
 EXTENSION-VALIDATOR ERRORS (most common):
 Format: "[field] message (... → ./path/source.ts)"
@@ -105443,12 +105732,12 @@ OTHER BUILD ERRORS:
 - Read the file mentioned in the error
 - Use the error message and file content to determine the fix
 
-IMPORTANT:
-- ALWAYS read files FIRST before submitting a fix
-- Read only the files you NEED - typically just 1-2 files
+CRITICAL RULES:
+- You MUST call submit_fix at the end - never respond with just text
+- Read files FIRST, then submit the fix
+- Read only 1-2 files maximum
 - Make minimal changes - only fix what's broken
-- Preserve code style and formatting
-- Be efficient - submit the fix quickly after reading`;
+- Preserve code style and formatting`;
       }
       formatPreviousBuildOutput(previousBuildOutput) {
         if (!previousBuildOutput) {
@@ -105474,11 +105763,11 @@ IMPORTANT:
       2. Read the relevant file(s) first to understand the context
       3. After reading, submit the fix for the file that needs changing
     `;
+        console.log(`\u{1F527} BuildFixerAgent starting with tools...`);
+        const model = (0, customAnthropicProvider_1.createCustomTextModel)()(constants_1.LLM_MODELS.CLAUDE_HAIKU_4_5, {
+          agentName: this.name
+        });
         try {
-          console.log(`\u{1F527} BuildFixerAgent starting with tools...`);
-          const model = (0, customAnthropicProvider_1.createCustomTextModel)()(constants_1.LLM_MODELS.CLAUDE_HAIKU_4_5, {
-            agentName: this.name
-          });
           const result = await (0, codegen_common_logic_1.generateAgentText)({
             agentName: this.name,
             model,
@@ -105492,7 +105781,9 @@ IMPORTANT:
           });
           const fixResult = (0, codegen_common_logic_1.extractToolResult)(result.toolCalls, SUBMIT_FIX_TOOL_NAME);
           if (!fixResult) {
-            throw new Error("BuildFixerAgent did not submit a fix");
+            throw new ditto_codegen_types_12.AutoFixError("BuildFixerAgent did not submit a fix", {
+              validationType: ditto_codegen_types_12.ValidationType.BUILD
+            });
           }
           console.log(`\u{1F527} BuildFixerAgent fix submitted for: ${fixResult.filePath}`);
           return fixResult;
@@ -106377,7 +106668,8 @@ var require_finalize_extension_generation = __commonJS({
     var file_collector_1 = require_file_collector();
     var orchestrator_error_helpers_1 = require_orchestrator_error_helpers();
     var write_file_1 = require_write_file();
-    function finalizeExtensionGeneration(options, eventEmitter) {
+    var codegen_flow_helpers_1 = require_codegen_flow_helpers();
+    async function finalizeExtensionGeneration(options, eventEmitter) {
       const { extension, agentName, agentFiles, scaffolds = [], scaffoldPath, outputPath, generatedConfig } = options;
       try {
         const filesToCheck = agentFiles.map((f) => f.path || "");
@@ -106396,6 +106688,7 @@ var require_finalize_extension_generation = __commonJS({
           outputPath,
           existingFiles
         });
+        await (0, codegen_flow_helpers_1.completeAgentTask)(extension, allCreatedFiles);
         eventEmitter.emitEvent("agent:done", {
           extension,
           name: agentName,
@@ -106403,7 +106696,8 @@ var require_finalize_extension_generation = __commonJS({
         });
         return extensionFilePath;
       } catch (error) {
-        throw (0, orchestrator_error_helpers_1.createExtensionErrorHandler)(eventEmitter, extension)(error);
+        await (0, orchestrator_error_helpers_1.createExtensionErrorHandler)(eventEmitter, extension)(error);
+        throw error;
       }
     }
   }
@@ -106449,10 +106743,12 @@ var require_extensionProcessors = __commonJS({
     exports2.processStandardExtension = processStandardExtension;
     var finalize_extension_generation_1 = require_finalize_extension_generation();
     var create_scaffolding_1 = require_create_scaffolding();
+    var codegen_flow_helpers_1 = require_codegen_flow_helpers();
     async function processStandardExtension(params) {
       const { extension, outputPath, plan, agentsFactory, eventEmitter, blueprint, userRequestSummary, previousResources } = params;
       const scaffolds = await (0, create_scaffolding_1.createScaffolding)(extension, outputPath, eventEmitter);
       const agent = agentsFactory.getAgent(extension);
+      await (0, codegen_flow_helpers_1.addAgentTask)(extension, agent.name);
       eventEmitter.emitEvent("agent:start", {
         extension,
         name: agent.name
@@ -106466,7 +106762,7 @@ var require_extensionProcessors = __commonJS({
         ...userRequestSummary && { userRequestSummary },
         ...previousResources && { previousResources }
       });
-      (0, finalize_extension_generation_1.finalizeExtensionGeneration)({
+      await (0, finalize_extension_generation_1.finalizeExtensionGeneration)({
         extension,
         agentName: agent.name,
         agentFiles: files,
@@ -106487,6 +106783,7 @@ var require_dashboardPage = __commonJS({
     var finalize_extension_generation_1 = require_finalize_extension_generation();
     var extensionProcessors_1 = require_extensionProcessors();
     var ditto_codegen_types_12 = require_dist();
+    var codegen_flow_helpers_1 = require_codegen_flow_helpers();
     async function processDashboardPage(params) {
       const { extension, outputPath, plan, agentsFactory, eventEmitter, blueprint, userRequestSummary, previousResources } = params;
       const decisionAgent = agentsFactory.getAgent({
@@ -106528,6 +106825,7 @@ var require_dashboardPage = __commonJS({
       const autoPatternsGenerator = agentsFactory.getAgent({
         type: "AUTO_PATTERNS_GENERATOR"
       });
+      await (0, codegen_flow_helpers_1.addAgentTask)(extension, "DashboardAgent");
       eventEmitter.emitEvent("agent:start", {
         extension,
         name: "DashboardAgent"
@@ -106538,7 +106836,7 @@ var require_dashboardPage = __commonJS({
         extensionName: extension.name || ""
       });
       const pagePath = files[0].path || "";
-      (0, finalize_extension_generation_1.finalizeExtensionGeneration)({
+      await (0, finalize_extension_generation_1.finalizeExtensionGeneration)({
         extension,
         agentName: "DashboardAgent",
         agentFiles: files,
@@ -106560,6 +106858,7 @@ var require_servicePlugin = __commonJS({
     var finalize_extension_generation_1 = require_finalize_extension_generation();
     var create_scaffolding_1 = require_create_scaffolding();
     var ditto_codegen_types_12 = require_dist();
+    var codegen_flow_helpers_1 = require_codegen_flow_helpers();
     async function processMultipleServicePluginScaffolds(params) {
       const { extension, scaffolds, outputPath, plan, agentsFactory, eventEmitter, blueprint, userRequestSummary, previousResources } = params;
       if (!extension.relatedSpis || extension.relatedSpis.length === 0) {
@@ -106580,6 +106879,7 @@ var require_servicePlugin = __commonJS({
           relatedSpis: [spi]
         };
         const agent = agentsFactory.getAgent(spiExtension);
+        await (0, codegen_flow_helpers_1.addAgentTask)(spiExtension, agent.name);
         eventEmitter.emitEvent("agent:start", {
           extension: spiExtension,
           name: agent.name
@@ -106593,7 +106893,7 @@ var require_servicePlugin = __commonJS({
           ...userRequestSummary && { userRequestSummary },
           ...previousResources && { previousResources }
         });
-        (0, finalize_extension_generation_1.finalizeExtensionGeneration)({
+        await (0, finalize_extension_generation_1.finalizeExtensionGeneration)({
           extension: spiExtension,
           agentName: agent.name,
           agentFiles: files,
@@ -106612,6 +106912,7 @@ var require_servicePlugin = __commonJS({
         return processMultipleServicePluginScaffolds({ ...params, scaffolds });
       }
       const agent = agentsFactory.getAgent(extension);
+      await (0, codegen_flow_helpers_1.addAgentTask)(extension, agent.name);
       eventEmitter.emitEvent("agent:start", { extension, name: agent.name });
       const files = await agent.generate({
         extension,
@@ -106622,7 +106923,7 @@ var require_servicePlugin = __commonJS({
         ...userRequestSummary && { userRequestSummary },
         ...previousResources && { previousResources }
       });
-      (0, finalize_extension_generation_1.finalizeExtensionGeneration)({
+      await (0, finalize_extension_generation_1.finalizeExtensionGeneration)({
         extension,
         agentName: agent.name,
         agentFiles: files,
@@ -106643,10 +106944,12 @@ var require_embeddedScript = __commonJS({
     var types_1 = require_types_impl2();
     var finalize_extension_generation_1 = require_finalize_extension_generation();
     var create_scaffolding_1 = require_create_scaffolding();
+    var codegen_flow_helpers_1 = require_codegen_flow_helpers();
     async function processEmbeddedScriptExtension(params) {
       const { extension, outputPath, plan, agentsFactory, eventEmitter, blueprint, userRequestSummary, previousResources } = params;
       const scaffolds = await (0, create_scaffolding_1.createScaffolding)(extension, outputPath, eventEmitter);
       const embeddedScriptAgent = agentsFactory.getAgent(extension);
+      await (0, codegen_flow_helpers_1.addAgentTask)(extension, embeddedScriptAgent.name);
       eventEmitter.emitEvent("agent:start", {
         extension,
         name: embeddedScriptAgent.name
@@ -106660,7 +106963,7 @@ var require_embeddedScript = __commonJS({
         ...userRequestSummary && { userRequestSummary },
         ...previousResources && { previousResources }
       });
-      (0, finalize_extension_generation_1.finalizeExtensionGeneration)({
+      await (0, finalize_extension_generation_1.finalizeExtensionGeneration)({
         extension,
         agentName: embeddedScriptAgent.name,
         agentFiles: files,
@@ -106684,9 +106987,11 @@ var require_backendApi = __commonJS({
     Object.defineProperty(exports2, "__esModule", { value: true });
     exports2.processBackendApiExtension = processBackendApiExtension;
     var finalize_extension_generation_1 = require_finalize_extension_generation();
+    var codegen_flow_helpers_1 = require_codegen_flow_helpers();
     async function processBackendApiExtension(params) {
       const { extension, outputPath, plan, agentsFactory, eventEmitter, blueprint, userRequestSummary, previousResources } = params;
       const backendApiAgent = agentsFactory.getAgent(extension);
+      await (0, codegen_flow_helpers_1.addAgentTask)(extension, backendApiAgent.name);
       eventEmitter.emitEvent("agent:start", {
         extension,
         name: backendApiAgent.name
@@ -106699,7 +107004,7 @@ var require_backendApi = __commonJS({
         ...userRequestSummary && { userRequestSummary },
         ...previousResources && { previousResources }
       });
-      (0, finalize_extension_generation_1.finalizeExtensionGeneration)({
+      await (0, finalize_extension_generation_1.finalizeExtensionGeneration)({
         extension,
         agentName: backendApiAgent.name,
         agentFiles: files,
@@ -106965,255 +107270,6 @@ var require_orchestrator = __commonJS({
   }
 });
 
-// ../../node_modules/@wix/ambassador-ctp-ditto-v1-blueprint/build/cjs/http.impl.js
-var require_http_impl2 = __commonJS({
-  "../../node_modules/@wix/ambassador-ctp-ditto-v1-blueprint/build/cjs/http.impl.js"(exports2) {
-    "use strict";
-    Object.defineProperty(exports2, "__esModule", { value: true });
-    exports2.getRequiredPermissions = exports2.editBlueprint = exports2.createAppName = exports2.createBlueprint = void 0;
-    var ambassador_1 = require_ambassador_index();
-    var metro_runtime_1 = require_cjs2();
-    var _createAppNameRequest = {};
-    var _createAppNameResponse = {};
-    var _createBlueprintRequest = {};
-    var _createBlueprintResponse = {};
-    var _editBlueprintRequest = {};
-    var _editBlueprintResponse = {};
-    var _getRequiredPermissionsRequest = {};
-    var _getRequiredPermissionsResponse = {};
-    function resolveComWixpressDittoBlueprintServiceApiDittoBlueprintServiceUrl(opts) {
-      var domainToMappings = {
-        "manage._base_domain_": [
-          {
-            srcPath: "/_api/ditto-blueprint-service-web",
-            destPath: ""
-          }
-        ]
-      };
-      return (0, metro_runtime_1.resolveUrl)(Object.assign(opts, { domainToMappings }));
-    }
-    function createBlueprint(payload) {
-      var _a = (0, ambassador_1.serializer)(_createBlueprintRequest, {}), toReq = _a.toJSON, fromReq = _a.fromJSON;
-      var fromRes = (0, ambassador_1.serializer)(_createBlueprintResponse, {}).fromJSON;
-      function __createBlueprint(_a2) {
-        var host = _a2.host;
-        var serializedData = toReq(payload);
-        var metadata = {
-          entityFqdn: "wix.ctp.ditto.v1.blueprint",
-          method: "POST",
-          methodFqn: "com.wixpress.ditto_blueprint_service.api.DittoBlueprintService.CreateBlueprint",
-          migrationOptions: {
-            optInTransformResponse: true
-          },
-          url: resolveComWixpressDittoBlueprintServiceApiDittoBlueprintServiceUrl({
-            protoPath: "/v1/blueprints",
-            data: serializedData,
-            host
-          }),
-          data: serializedData,
-          transformResponse: fromRes
-        };
-        return metadata;
-      }
-      __createBlueprint.fromReq = fromReq;
-      __createBlueprint.__isAmbassador = true;
-      return __createBlueprint;
-    }
-    exports2.createBlueprint = createBlueprint;
-    function createAppName(payload) {
-      var _a = (0, ambassador_1.serializer)(_createAppNameRequest, {}), toReq = _a.toJSON, fromReq = _a.fromJSON;
-      var fromRes = (0, ambassador_1.serializer)(_createAppNameResponse, {}).fromJSON;
-      function __createAppName(_a2) {
-        var host = _a2.host;
-        var serializedData = toReq(payload);
-        var metadata = {
-          entityFqdn: "wix.ctp.ditto.v1.blueprint",
-          method: "POST",
-          methodFqn: "com.wixpress.ditto_blueprint_service.api.DittoBlueprintService.CreateAppName",
-          migrationOptions: {
-            optInTransformResponse: true
-          },
-          url: resolveComWixpressDittoBlueprintServiceApiDittoBlueprintServiceUrl({
-            protoPath: "/v1/blueprints/app-name",
-            data: serializedData,
-            host
-          }),
-          data: serializedData,
-          transformResponse: fromRes
-        };
-        return metadata;
-      }
-      __createAppName.fromReq = fromReq;
-      __createAppName.__isAmbassador = true;
-      return __createAppName;
-    }
-    exports2.createAppName = createAppName;
-    function editBlueprint(payload) {
-      var _a = (0, ambassador_1.serializer)(_editBlueprintRequest, {}), toReq = _a.toJSON, fromReq = _a.fromJSON;
-      var fromRes = (0, ambassador_1.serializer)(_editBlueprintResponse, {}).fromJSON;
-      function __editBlueprint(_a2) {
-        var host = _a2.host;
-        var serializedData = toReq(payload);
-        var metadata = {
-          entityFqdn: "wix.ctp.ditto.v1.blueprint",
-          method: "POST",
-          methodFqn: "com.wixpress.ditto_blueprint_service.api.DittoBlueprintService.EditBlueprint",
-          migrationOptions: {
-            optInTransformResponse: true
-          },
-          url: resolveComWixpressDittoBlueprintServiceApiDittoBlueprintServiceUrl({
-            protoPath: "/v1/blueprints/edit",
-            data: serializedData,
-            host
-          }),
-          data: serializedData,
-          transformResponse: fromRes
-        };
-        return metadata;
-      }
-      __editBlueprint.fromReq = fromReq;
-      __editBlueprint.__isAmbassador = true;
-      return __editBlueprint;
-    }
-    exports2.editBlueprint = editBlueprint;
-    function getRequiredPermissions(payload) {
-      var _a = (0, ambassador_1.serializer)(_getRequiredPermissionsRequest, {}), toReq = _a.toJSON, fromReq = _a.fromJSON;
-      var fromRes = (0, ambassador_1.serializer)(_getRequiredPermissionsResponse, {}).fromJSON;
-      function __getRequiredPermissions(_a2) {
-        var host = _a2.host;
-        var serializedData = toReq(payload);
-        var metadata = {
-          entityFqdn: "wix.ctp.ditto.v1.blueprint",
-          method: "POST",
-          methodFqn: "com.wixpress.ditto_blueprint_service.api.DittoBlueprintService.GetRequiredPermissions",
-          migrationOptions: {
-            optInTransformResponse: true
-          },
-          url: resolveComWixpressDittoBlueprintServiceApiDittoBlueprintServiceUrl({
-            protoPath: "/v1/blueprints/required-permissions",
-            data: serializedData,
-            host
-          }),
-          data: serializedData,
-          transformResponse: fromRes
-        };
-        return metadata;
-      }
-      __getRequiredPermissions.fromReq = fromReq;
-      __getRequiredPermissions.__isAmbassador = true;
-      return __getRequiredPermissions;
-    }
-    exports2.getRequiredPermissions = getRequiredPermissions;
-  }
-});
-
-// dist/flows/codegen-flow-helpers.js
-var require_codegen_flow_helpers = __commonJS({
-  "dist/flows/codegen-flow-helpers.js"(exports2) {
-    "use strict";
-    var __importDefault2 = exports2 && exports2.__importDefault || function(mod) {
-      return mod && mod.__esModule ? mod : { "default": mod };
-    };
-    Object.defineProperty(exports2, "__esModule", { value: true });
-    exports2.setupAgentTaskTracking = setupAgentTaskTracking;
-    exports2.updateJobPayload = updateJobPayload;
-    exports2.updateParentTaskStatus = updateParentTaskStatus;
-    exports2.getOutputPath = getOutputPath;
-    exports2.getRequiredPermissions = getRequiredPermissions;
-    var path_1 = __importDefault2(require("path"));
-    var codeGenerationService_12 = require_codeGenerationService();
-    var ditto_codegen_types_12 = require_dist();
-    var utils_1 = require_utils15();
-    var http_1 = require_http_impl2();
-    var http_client_1 = require_index_node();
-    var orchestrator_1 = require_orchestrator();
-    var AgentsFactory_1 = require_AgentsFactory();
-    var slugify = (str = "") => {
-      return str.toLowerCase().replace(/ /g, "-");
-    };
-    function setupAgentTaskTracking(eventEmitter, jobContext) {
-      const getTaskId = (extension) => {
-        return `${jobContext.taskId}-${slugify(extension.name)}`;
-      };
-      eventEmitter.onEvent("agent:start", ({ extension }) => {
-        console.log(`[Agent] start: ${extension.name}`);
-        const taskId = getTaskId(extension);
-        codeGenerationService_12.codeGenerationService.addTask(jobContext.jobId, {
-          id: taskId,
-          kind: ditto_codegen_types_12.TaskKind.RUN_AGENT,
-          status: ditto_codegen_types_12.Status.RUNNING,
-          name: "run_agent",
-          description: `Run agent for ${extension.name}`,
-          payload: { extension },
-          agentName: new AgentsFactory_1.AgentsFactory().getAgent(extension).name
-        });
-      });
-      eventEmitter.onEvent("agent:done", ({ extension, files }) => {
-        console.log(`[Agent] done: ${extension.name}`);
-        const taskId = getTaskId(extension);
-        codeGenerationService_12.codeGenerationService.updateTask(jobContext.jobId, taskId, ditto_codegen_types_12.Status.COMPLETED, { taskOutput: { files } });
-      });
-      eventEmitter.onEvent("agent:error", async ({ extension, error }) => {
-        console.error(`[Agent] error: ${extension.name}`, error);
-        const taskId = getTaskId(extension);
-        await codeGenerationService_12.codeGenerationService.updateTask(jobContext.jobId, taskId, ditto_codegen_types_12.Status.FAILED, {
-          error: (0, utils_1.serializeError)(error)
-        });
-      });
-      const updateExtensionAgentTaskId = `${jobContext.taskId}-update-extension-agent`;
-      eventEmitter.onEvent("updateExtensionAgent:start", () => {
-        console.log(`[Agent] start: iterating existing code`);
-        codeGenerationService_12.codeGenerationService.addTask(jobContext.jobId, {
-          id: updateExtensionAgentTaskId,
-          kind: ditto_codegen_types_12.TaskKind.RUN_AGENT,
-          status: ditto_codegen_types_12.Status.RUNNING,
-          name: "run_agent",
-          description: "Iterating existing code",
-          agentName: "UpdateExtensionAgent"
-        });
-      });
-      eventEmitter.onEvent("updateExtensionAgent:done", ({ files }) => {
-        console.log(`[Agent] done: iterating existing code`);
-        codeGenerationService_12.codeGenerationService.updateTask(jobContext.jobId, updateExtensionAgentTaskId, ditto_codegen_types_12.Status.COMPLETED, { taskOutput: { files } });
-      });
-      eventEmitter.onEvent("updateExtensionAgent:error", async ({ error }) => {
-        console.error(`[Agent] error: iterating existing code`, error);
-        await codeGenerationService_12.codeGenerationService.updateTask(jobContext.jobId, updateExtensionAgentTaskId, ditto_codegen_types_12.Status.FAILED, {
-          error: (0, utils_1.serializeError)(error)
-        });
-      });
-    }
-    async function updateJobPayload(jobContext, payload) {
-      await codeGenerationService_12.codeGenerationService.updateJob(jobContext.jobId, payload);
-    }
-    async function updateParentTaskStatus(jobContext, status, error) {
-      await codeGenerationService_12.codeGenerationService.updateTask(jobContext.jobId, jobContext.taskId, status, error ? { error: (0, utils_1.serializeError)(error) } : {});
-    }
-    function getOutputPath() {
-      const outputDir = process.env.OUTPUT_PATH || orchestrator_1.DittoOrchestrator.DEFAULT_OUTPUT_PATH;
-      return outputDir.startsWith("/") ? outputDir : path_1.default.join(process.cwd(), outputDir);
-    }
-    async function getRequiredPermissions(extensions) {
-      const httpClient = (0, http_client_1.createHttpClient)({
-        baseURL: "https://manage.wix.com"
-      });
-      return httpClient.request((0, http_1.getRequiredPermissions)({
-        relatedApis: getRelatedApis(extensions ?? []),
-        extensionTypes: getExtensionTypes(extensions ?? [])
-      }));
-    }
-    function getExtensionTypes(extensions) {
-      return Array.from(new Set((extensions ?? []).map(({ type }) => type).filter(Boolean)));
-    }
-    function getRelatedApis(extensions) {
-      return Array.from(new Set((extensions ?? []).flatMap((ext) => [
-        ...(ext.relatedApis ?? []).map(({ name }) => name).filter(Boolean),
-        ...(ext.relatedSpis ?? []).map(({ name }) => name).filter(Boolean)
-      ])));
-    }
-  }
-});
-
 // dist/flows/init-codegen.js
 var require_init_codegen = __commonJS({
   "dist/flows/init-codegen.js"(exports2) {
@@ -107239,7 +107295,6 @@ var require_init_codegen = __commonJS({
         const outputPath = (0, codegen_flow_helpers_1.getOutputPath)();
         const agentsFactory = new AgentsFactory_1.AgentsFactory(false);
         const orchestrator = new orchestrator_1.DittoOrchestrator(agentsFactory);
-        (0, codegen_flow_helpers_1.setupAgentTaskTracking)(orchestrator, localJobContext);
         (0, cli_listeners_1.attachOrchestratorListeners)(orchestrator);
         try {
           const requiredPermissionsResponse = await (0, codegen_flow_helpers_1.getRequiredPermissions)(blueprint?.extensions ?? []);
@@ -321045,6 +321100,7 @@ var require_IterationOrchestrator = __commonJS({
     var write_file_1 = require_write_file();
     var projectContext_1 = require_projectContext();
     var ditto_codegen_types_12 = require_dist();
+    var codegen_flow_helpers_1 = require_codegen_flow_helpers();
     var IterationOrchestrator = class extends DittoEventEmitter_1.DittoEventEmitter {
       constructor(agentsFactory) {
         super();
@@ -321143,6 +321199,7 @@ var require_IterationOrchestrator = __commonJS({
         return iterationPlan;
       }
       async processExistingExtensions({ currentExtensions, userRequestSummary, outputPath, plan, previousResources }) {
+        await (0, codegen_flow_helpers_1.addUpdateExtensionAgentTask)();
         this.emitEvent("updateExtensionAgent:start", {});
         try {
           const updateExtensionAgent = this.agentsFactory.getAgent({
@@ -321155,10 +321212,12 @@ var require_IterationOrchestrator = __commonJS({
             plan,
             previousResources
           });
+          await (0, codegen_flow_helpers_1.completeUpdateExtensionAgentTask)(result.filesModified);
           this.emitEvent("updateExtensionAgent:done", {
             files: result.filesModified
           });
         } catch (error) {
+          await (0, codegen_flow_helpers_1.failUpdateExtensionAgentTask)(error);
           this.emitEvent("updateExtensionAgent:error", {
             error: error instanceof Error ? error : new Error(String(error))
           });
@@ -321310,7 +321369,6 @@ var require_iterate_codegen = __commonJS({
         const outputPath = (0, codegen_flow_helpers_1.getOutputPath)();
         const agentsFactory = new AgentsFactory_1.AgentsFactory(true);
         const iterationOrchestrator = new IterationOrchestrator_1.IterationOrchestrator(agentsFactory);
-        (0, codegen_flow_helpers_1.setupAgentTaskTracking)(iterationOrchestrator, localJobContext);
         (0, cli_listeners_1.attachOrchestratorListeners)(iterationOrchestrator);
         const { iterationPlan, previousResources } = await iterationOrchestrator.getContextualIterationPlan({
           outputPath,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@wix/ditto-codegen-public",
-  "version": "1.0.223",
+  "version": "1.0.225",
   "description": "AI-powered Wix CLI app generator - standalone executable",
   "scripts": {
     "build": "node build.mjs",
@@ -24,5 +24,5 @@
     "@wix/ditto-codegen": "1.0.0",
     "esbuild": "^0.27.2"
   },
-  "falconPackageHash": "b4a09affaa1c6925b5e726fbc1140c778b980df29e8e7cb10648460f"
+  "falconPackageHash": "2e7d0b6dea61740e43a2e46910e0abf3709a19aed36a59e7b403b2cc"
 }