@wix/ditto-codegen-public 1.0.16 → 1.0.17

package/dist/out.js

@@ -114034,16 +114034,23 @@ var require_index_node = __commonJS({
 var require_utils14 = __commonJS({
   "dist/agents/utils.js"(exports2) {
     "use strict";
+    var __importDefault2 = exports2 && exports2.__importDefault || function(mod2) {
+      return mod2 && mod2.__esModule ? mod2 : { "default": mod2 };
+    };
     Object.defineProperty(exports2, "__esModule", { value: true });
-    exports2.getHttpClient = exports2.withCaching = exports2.GeneratedFileSchema = exports2.FileSchema = void 0;
+    exports2.buildUserPromptForCodeGenerationAgent = exports2.getHttpClient = exports2.withCaching = exports2.FileSchema = void 0;
+    exports2.loadRelevantFilesAsString = loadRelevantFilesAsString;
     var zod_1 = require_zod();
     var http_client_1 = require_index_node();
-    exports2.FileSchema = zod_1.z.object({
-      content: zod_1.z.string().describe("Complete file content as a string (for JSON files, stringify the object)")
+    var fs_1 = __importDefault2(require("fs"));
+    var path_1 = __importDefault2(require("path"));
+    var FileItemSchema = zod_1.z.object({
+      operation: zod_1.z.enum(["insert", "update", "delete"]).describe("File operation: insert (new file), update (modify existing), delete (remove file)").default("insert"),
+      path: zod_1.z.string().describe("Relative file path from project root").optional(),
+      content: zod_1.z.string().describe("Complete file content as a string (for JSON files, stringify the object). Required for insert and update operations.").optional()
     });
-    exports2.GeneratedFileSchema = zod_1.z.object({
-      path: zod_1.z.string().describe("Relative file path from project root"),
-      content: exports2.FileSchema.shape.content
+    exports2.FileSchema = zod_1.z.object({
+      files: zod_1.z.array(FileItemSchema).default([])
     });
     var withCaching = (ttl = "5m") => {
       return {
@@ -114068,6 +114075,78 @@ var require_utils14 = __commonJS({
       timeout: 6e6
     });
     exports2.getHttpClient = getHttpClient;
+    function loadRelevantFiles(relevantFilePaths = [], basePath) {
+      return relevantFilePaths.map((filePath) => {
+        const fullPath = path_1.default.isAbsolute(basePath) ? path_1.default.join(basePath, filePath) : path_1.default.join(process.cwd(), basePath, filePath);
+        try {
+          if (!fs_1.default.existsSync(fullPath)) {
+            return `Path: ${filePath}
+
+	[File does not exist]`;
+          }
+          const stats = fs_1.default.statSync(fullPath);
+          if (stats.isDirectory()) {
+            return `Path: ${filePath}
+
+	[Path is a directory, not a file]`;
+          }
+          const content = fs_1.default.readFileSync(fullPath, "utf8");
+          return `Path: ${filePath}
+
+	${content}`;
+        } catch (error) {
+          return `Path: ${filePath}
+
+	[Could not read file: ${error instanceof Error ? error.message : "Unknown error"}]`;
+        }
+      });
+    }
+    function loadRelevantFilesAsString(relevantFilePaths = [], basePath) {
+      const srcOnlyPaths = relevantFilePaths.filter((p) => {
+        const normalized = path_1.default.normalize(p);
+        return normalized.startsWith("src" + path_1.default.sep);
+      });
+      return loadRelevantFiles(srcOnlyPaths, basePath).join("\n\n---\n\n");
+    }
+    var buildUserPromptForCodeGenerationAgent = (params, primaryAction) => {
+      const { extension, blueprint, scaffold, userRequestSummary, relevantFilePaths, createdCollections, basePath } = params;
+      const contextSections = [];
+      if (extension.name)
+        contextSections.push(`Extension Name: ${extension.name}`);
+      if (extension.description)
+        contextSections.push(`Extension Description: ${extension.description}`);
+      if (blueprint?.summary) {
+        contextSections.push(`App Summary: ${blueprint.summary}`);
+      }
+      if (userRequestSummary) {
+        contextSections.push(`
+## USER REQUEST
+${userRequestSummary}`);
+      }
+      if (relevantFilePaths) {
+        const relevantFiles = loadRelevantFilesAsString(relevantFilePaths, basePath);
+        contextSections.push(`
+## RELEVANT FILES
+${relevantFiles}`);
+      }
+      if (scaffold) {
+        contextSections.push(`
+## CURRENT SCAFFOLDED FILE
+Path: ${scaffold.path}
+\`\`\`
+${scaffold.content}
+\`\`\``);
+      }
+      if (createdCollections) {
+        contextSections.push(`
+## CREATED COLLECTIONS
+${JSON.stringify(createdCollections, null, 2)}`);
+      }
+      const userMessage = `${primaryAction}:
+${contextSections.join("\n")}`;
+      return userMessage;
+    };
+    exports2.buildUserPromptForCodeGenerationAgent = buildUserPromptForCodeGenerationAgent;
   }
 });
 
@@ -114351,22 +114430,17 @@ var require_SPIAgent = __commonJS({
       buildSystemPrompt(spiNames = []) {
         return (0, servicePluginPrompt_1.getServicePluginPrompt)(spiNames);
       }
-      async generate(extension, blueprint, scaffold) {
+      async generate(params) {
+        const { extension, blueprint } = params;
         const examples = (0, load_examples_1.loadExamples)([load_examples_1.types.ServicePluginExtension]);
         const allSpiNames = extension.relatedSpis?.map((spi) => spi.name).filter((name) => !!name) || [];
         const systemPrompt = `${this.buildSystemPrompt(allSpiNames)}
 ${examples}
 `;
         console.log(`SPI Agent System Prompt length: ${systemPrompt.length} (is that what you expect?)`);
-        const userMessage = `Customize the Wix CLI service plugin scaffolding for the app "${blueprint.appName}":
-Extension Name: ${extension.name}
-Extension Description: ${extension.description}
-App Summary: ${blueprint.summary}
-## CURRENT SCAFFOLDED FILE
-Path: ${scaffold.path}
-\`\`\`
-${scaffold.content}
-\`\`\``;
+        const appName = blueprint?.appName ? `'${blueprint.appName}'` : "";
+        const primaryAction = `Customize the Wix CLI service plugin scaffolding for the app ${appName}`;
+        const userMessage = (0, utils_1.buildUserPromptForCodeGenerationAgent)(params, primaryAction);
         const model = (0, anthropic_1.createAnthropic)({
           apiKey: this.apiKey
         })("claude-sonnet-4-20250514");
@@ -114391,7 +114465,7 @@ ${scaffold.content}
           maxRetries: 3,
           temperature: 0
         });
-        return result.object;
+        return result.object.files;
       }
     };
     exports2.SPIAgent = SPIAgent;
@@ -115220,6 +115294,39 @@ ${blocks}
   }
 });
 
+// dist/system-prompts/dashboardPage/data.js
+var require_data = __commonJS({
+  "dist/system-prompts/dashboardPage/data.js"(exports2) {
+    "use strict";
+    Object.defineProperty(exports2, "__esModule", { value: true });
+    exports2.dataPrompt = void 0;
+    exports2.dataPrompt = `<wix_data_docs>
+  Summary:
+  - Read: items.query('Collection').filter/sort.limit.find() \u2192 { items, totalCount, hasNext }
+  - Write: items.insert | update | remove. Ensure collection permissions allow the action
+
+  Access data using the collection schema:
+  - Always use the exact field keys you defined in the collection schema.
+  - YOU MUST use the collection id exactly as you defined it in the collection schema.
+
+  Example - Insert / Update / Delete (if permissions allow):
+  import { items } from '@wix/data';
+
+  async function addItem(collectionId: string, itemData: WixDataItem) {
+    return items.insert(collectionId, itemData);
+  }
+
+  async function renameItem(collectionId: string, itemData: WixDataItem) {
+    await items.update(collectionId, itemData);
+  }
+
+  async function deleteItem(collectionId: string, itemId: WixDataItem) {
+    await items.remove(collectionId, itemId);
+  }
+</wix_data_docs>`;
+  }
+});
+
 // dist/system-prompts/dashboardPage/dashboardPagePrompt.js
 var require_dashboardPagePrompt = __commonJS({
   "dist/system-prompts/dashboardPage/dashboardPagePrompt.js"(exports2) {
@@ -115229,6 +115336,7 @@ var require_dashboardPagePrompt = __commonJS({
     var ecomPackage_1 = require_ecomPackage();
     var dashboardPackage_1 = require_dashboardPackage();
     var wdsPackage_1 = require_wdsPackage();
+    var data_1 = require_data();
     var wdsPackage_2 = require_wdsPackage();
     Object.defineProperty(exports2, "buildWdsSystemPrompt", { enumerable: true, get: function() {
       return wdsPackage_2.buildWdsSystemPrompt;
@@ -115257,7 +115365,7 @@ var require_dashboardPagePrompt = __commonJS({
       "ToggleSwitch",
       "InfoIcon"
     ];
-    var dashboardPagePrompt = async () => {
+    var dashboardPagePrompt = async (useData) => {
       const wdsPrompt = await (0, wdsPackage_1.buildWdsSystemPrompt)(listOfWdsComponents);
       return `
 <WIXCLI_DASHBOARD_PAGE_SYSTEM_PROMPT>
@@ -115321,8 +115429,6 @@ ${dashboardPackage_1.dashboardPackage}
 ${ecomPackage_1.ecomPackage}
 </api_references>
 
-
-
 <wds_reference>
 ${wdsPrompt}
 </wds_reference>
@@ -115336,30 +115442,7 @@ ${wdsPrompt}
 - Do NOT use // @ts-ignore or // @ts-expect-error; fix the types or add guards instead
 </typescript_quality_guidelines>
 
-<wix_data_docs>
-  Summary:
-  - Read: items.query('Collection').filter/sort.limit.find() \u2192 { items, totalCount, hasNext }
-  - Write: items.insert | update | remove. Ensure collection permissions allow the action
-
-  Access data using the collection schema:
-  - Always use the exact field keys you defined in the collection schema.
-  - YOU MUST use the collection id exactly as you defined it in the collection schema.
-
-  Example - Insert / Update / Delete (if permissions allow):
-  import { items } from '@wix/data';
-
-  async function addItem(collectionId: string, itemData: WixDataItem) {
-    return items.insert(collectionId, itemData);
-  }
-
-  async function renameItem(collectionId: string, itemData: WixDataItem) {
-    await items.update(collectionId, itemData);
-  }
-
-  async function deleteItem(collectionId: string, itemId: WixDataItem) {
-    await items.remove(collectionId, itemId);
-  }
-</wix_data_docs>
+${useData ? data_1.dataPrompt : ""}
 </WIXCLI_DASHBOARD_PAGE_SYSTEM_PROMPT>
 `;
     };
@@ -115383,28 +115466,20 @@ var require_DashboardPageAgent = __commonJS({
         this.apiKey = apiKey;
         this.name = "DashboardPageAgent";
       }
-      async buildSystemPrompt() {
-        return (0, dashboardPagePrompt_1.dashboardPagePrompt)();
+      async buildSystemPrompt(useData) {
+        return (0, dashboardPagePrompt_1.dashboardPagePrompt)(useData);
       }
-      async generate(extension, blueprint, scaffold, createdCollections) {
+      async generate(params) {
+        const { blueprint, createdCollections } = params;
         const examples = (0, load_examples_1.loadExamples)([load_examples_1.types.DashboardPage]);
-        const systemPrompt = `${await this.buildSystemPrompt()}
+        const useData = Boolean(createdCollections && createdCollections.length > 0);
+        const systemPrompt = `${await this.buildSystemPrompt(useData)}
 ${examples}
 `;
         console.log(`Dashboard Agent System Prompt length: ${systemPrompt.length} (is that what you expect?)`);
-        const userMessage = `Customize the Wix CLI dashboard page scaffolding for the app "${blueprint.appName}":
-Extension Name: ${extension.name}
-Extension Description: ${extension.description}
-App Summary: ${blueprint.summary}
-
-## CREATED COLLECTIONS
-${JSON.stringify(createdCollections, null, 2)}
-
-## CURRENT SCAFFOLDED FILE
-Path: ${scaffold.path}
-\`\`\`
-${scaffold.content}
-\`\`\``;
+        const appName = blueprint?.appName ? `'${blueprint.appName}'` : "";
+        const primaryAction = `Customize the Wix CLI dashboard page scaffolding for the app ${appName}`;
+        const userMessage = (0, utils_1.buildUserPromptForCodeGenerationAgent)(params, primaryAction);
         const model = (0, anthropic_1.createAnthropic)({ apiKey: this.apiKey })("claude-sonnet-4-20250514");
         const result = await (0, ai_1.generateObject)({
           model,
@@ -115428,7 +115503,7 @@ ${scaffold.content}
           maxRetries: 3,
           temperature: 0
         });
-        return result.object;
+        return result.object.files;
       }
     };
     exports2.DashboardPageAgent = DashboardPageAgent;
@@ -115437,7 +115512,7 @@ ${scaffold.content}
 });
 
 // dist/system-prompts/planner/data.js
-var require_data = __commonJS({
+var require_data2 = __commonJS({
   "dist/system-prompts/planner/data.js"(exports2) {
     "use strict";
     Object.defineProperty(exports2, "__esModule", { value: true });
@@ -115536,7 +115611,7 @@ var require_planner = __commonJS({
     "use strict";
     Object.defineProperty(exports2, "__esModule", { value: true });
     exports2.plannerPrompt = void 0;
-    var data_1 = require_data();
+    var data_1 = require_data2();
     var plannerPrompt = () => `
 <WIXCLI_PLANNER_SYSTEM_PROMPT>
 ${(0, data_1.cmsPlannerPrompt)()}
@@ -119122,18 +119197,12 @@ var require_SiteComponentAgent = __commonJS({
       async buildSystemPrompt() {
         return (0, siteComponentPrompt_1.siteComponentPrompt)();
       }
-      async generate(extension, blueprint, scaffold) {
+      async generate(params) {
+        const { blueprint } = params;
         const systemPrompt = `${await this.buildSystemPrompt()}`;
-        const userMessage = `Customize the Wix CLI dashboard page scaffolding for the app "${blueprint.appName}":
-Extension Name: ${extension.name}
-Extension Description: ${extension.description}
-App Summary: ${blueprint.summary}
-
-## CURRENT SCAFFOLDED FILE
-Path: ${scaffold.path}
-\`\`\`
-${scaffold.content}
-\`\`\``;
+        const appName = blueprint?.appName ? `'${blueprint.appName}'` : "";
+        const primaryAction = `Customize the Wix CLI site component scaffolding for the app ${appName}`;
+        const userMessage = (0, utils_1.buildUserPromptForCodeGenerationAgent)(params, primaryAction);
         const model = (0, anthropic_1.createAnthropic)({ apiKey: this.apiKey })("claude-sonnet-4-20250514");
         const result = await (0, ai_1.generateObject)({
           model,
@@ -119157,7 +119226,7 @@ ${scaffold.content}
           maxRetries: 3,
           temperature: 0
         });
-        return result.object;
+        return result.object.files;
       }
     };
     exports2.SiteComponentAgent = SiteComponentAgent;
@@ -119165,6 +119234,261 @@ ${scaffold.content}
   }
 });
 
+// dist/system-prompts/iterationAgent/iterationAgentPrompt.js
+var require_iterationAgentPrompt = __commonJS({
+  "dist/system-prompts/iterationAgent/iterationAgentPrompt.js"(exports2) {
+    "use strict";
+    Object.defineProperty(exports2, "__esModule", { value: true });
+    exports2.iterationAgentPrompt = void 0;
+    var iterationAgentPrompt = () => `
+<WIXCLI_ITERATION_AGENT_SYSTEM_PROMPT>
+
+<role>
+You are an iteration agent for a Wix app code generation system.
+Given the full chat history, current user request, blueprint, and all project files:
+1. Generate a clear summary of what the user wants to achieve
+2. Identify which existing extensions should be triggered to handle this request
+3. Map each triggered extension to the relevant file paths it needs to modify
+4. Propose new extensions if the request requires functionality not covered by existing ones
+</role>
+
+<supported_extension_types>
+
+<dashboard_page>
+**Use when:**
+- User wants to create a management interface/admin panel
+- Need to display data tables, forms, or configuration screens
+- Building settings pages, analytics dashboards, or content management interfaces
+- Example: 'Create a dashboard to manage FAQ items', 'Add settings page for the app'
+</dashboard_page>
+
+<site_component>
+**Use when:**
+- User wants to create a custom UI component for the site frontend
+- Need to build interactive widgets, forms, or display components for visitors
+- Creating custom elements that appear on the site (not admin interfaces)
+- Building components that integrate with site data or user interactions
+- Example: 'Create a product showcase widget', 'Add a contact form component', 'Build a testimonial slider'
+</site_component>
+
+<service_plugin>
+**Use when:**
+- User needs custom business logic for eCommerce (fees, shipping, discounts, validations)
+- Integrating with external payment providers
+- Custom pricing calculations for bookings
+- Form validation logic
+- Example: 'Calculate custom shipping rates', 'Add payment validation', 'Custom discount rules'
+
+**Available SPI Names (choose appropriate ones for relatedSpis):**
+- "ecom.shippingRates.getShippingRates" - For custom shipping rate calculations
+- "ecom.additionalFees.calculateAdditionalFees" - For calculating additional fees (handling, processing, etc.)
+- "ecom.paymentSettings.getPaymentSettings" - For payment configuration and 3D Secure settings
+- "ecom.validations.getValidationViolations" - For cart and checkout validations
+- "ecom.customTriggers.getEligibleTriggers" - For custom discount trigger logic
+- "ecom.customTriggers.listTriggers" - For listing available custom discount triggers
+- "ecom.giftCardsProvider.redeem" - For gift card redemption logic
+- "ecom.giftCardsProvider._void" - For voiding gift card transactions
+- "ecom.giftCardsProvider.getBalance" - For checking gift card balances
+- "ecom.taxCalculationProvider.calculateTax" - For custom tax calculations
+- "ecom.recommendationsProvider.getRecommendations" - For product recommendations
+- "ecom.catalog.getCatalogItems" - For external catalog integration
+
+**SPI Selection Guidelines:**
+- For shipping-related requests \u2192 use "ecom.shippingRates.getShippingRates"
+- For fee calculations \u2192 use "ecom.additionalFees.calculateAdditionalFees"
+- For payment processing \u2192 use "ecom.paymentSettings.getPaymentSettings"
+- For cart/checkout validation \u2192 use "ecom.validations.getValidationViolations"
+- For discount logic \u2192 use "ecom.customTriggers.getEligibleTriggers" or "ecom.customTriggers.listTriggers"
+- For gift card functionality \u2192 use appropriate "ecom.giftCardsProvider.*" methods
+- For tax calculations \u2192 use "ecom.taxCalculationProvider.calculateTax"
+- For product recommendations \u2192 use "ecom.recommendationsProvider.getRecommendations"
+- For external catalog \u2192 use "ecom.catalog.getCatalogItems"
+</service_plugin>
+
+</supported_extension_types>
+
+<extension_selection_logic>
+
+<selection_rules>
+1. **For UI/Admin interfaces**: Use DASHBOARD_PAGE
+2. **For site frontend components**: Use SITE_COMPONENT
+3. **For eCommerce business logic**: Use SERVICE_PLUGIN
+4. **For management interfaces**: Use DASHBOARD_PAGE
+5. **For eCommerce features**: Combine SERVICE_PLUGIN (logic) + DASHBOARD_PAGE (configuration UI)
+6. **For site features**: Combine SITE_COMPONENT (frontend) + SERVICE_PLUGIN (backend logic) + DASHBOARD_PAGE (admin interface)
+</selection_rules>
+
+<common_patterns>
+- **eCommerce Extensions**: SERVICE_PLUGIN + DASHBOARD_PAGE
+- **Configuration/Settings Apps**: DASHBOARD_PAGE only
+- **Site Frontend Features**: SITE_COMPONENT only
+- **Full-Featured Apps**: SITE_COMPONENT + SERVICE_PLUGIN + DASHBOARD_PAGE
+</common_patterns>
+
+</extension_selection_logic>
+
+<constraints>
+- Return the JSON schema with currentExtensions, additionalExtensions, and summary
+- Only propose additional extensions if existing ones cannot fulfill the user's request
+- Be specific about which file paths are relevant for each current extension (all extension types)
+- Consider the full chat history context when making decisions
+- Prioritize reusing existing extensions over creating new ones
+- **SUPPORTED EXTENSION TYPES ONLY**: DASHBOARD_PAGE, SERVICE_PLUGIN, and SITE_COMPONENT
+- **IMPORTANT**: The "type" field is ONLY for SERVICE_PLUGIN extensions - specify the SPI type
+- The "paths" field is for ALL extension types - specify relevant file paths to modify
+- Choose SPI types based on the specific eCommerce functionality requested (shipping, fees, validation, etc.)
+- Each extension should have a clear relevantUserRequest explaining what part of the request it addresses
+ - If files already exist for an extension (e.g., dashboard page component files or a service plugin \`plugin.ts\` under the relevant SPI directory), you MUST treat it as a currentExtensions item and update it, NOT as an additionalExtensions item
+ - For currentExtensions, ensure relevantUserRequest precisely reflects the change to be applied in those files (e.g., "Update static fee from 5 ILS to 10 ILS"), not a generic description
+ - Propose additionalExtensions ONLY when there is no existing file that matches the requested capability; avoid duplicates like proposing a new \`ecom.additionalFees.calculateAdditionalFees\` when one already exists in the project
+</constraints>
+
+<output_format>
+The response must be a valid JSON object matching the IterationPlanSchema:
+
+**Required Structure:**
+{
+  "currentExtensions": [
+    {
+      "extensionName": "SERVICE_PLUGIN",
+      "relatedSpis": ["ecom.shippingRates.getShippingRates"], // Only for SERVICE_PLUGIN extensions
+      "paths": ["src/backend/service-plugins/shipping/custom-rates/plugin.ts"],
+      "relevantUserRequest": "Calculate shipping rates based on product weight"
+    },
+    {
+      "extensionName": "DASHBOARD_PAGE",
+      "paths": ["src/dashboard/pages/AdminPage.tsx", "src/dashboard/components/SettingsForm.tsx"],
+      "relevantUserRequest": "Provide admin interface for managing shipping settings"
+    },
+    {
+      "extensionName": "SITE_COMPONENT",
+      "paths": ["src/site/components/ProductShowcase/component.tsx"],
+      "relevantUserRequest": "Create a product showcase widget for the site frontend"
+    }
+  ],
+  "additionalExtensions": [
+    {
+      "extensionName": "SERVICE_PLUGIN",
+      "relatedSpis": ["ecom.additionalFees.calculateAdditionalFees"], // Only for SERVICE_PLUGIN extensions
+      "relevantUserRequest": "Add handling fees for fragile items"
+    }
+  ],
+  "summary": "User wants to implement custom shipping rates and add handling fees for their eCommerce store"
+}
+
+**Key Points:**
+- currentExtensions: Existing extensions that should be triggered/modified (use extensionName of "SERVICE_PLUGIN", "DASHBOARD_PAGE", or "SITE_COMPONENT")
+- additionalExtensions: New extensions that need to be created (use extensionName of "SERVICE_PLUGIN", "DASHBOARD_PAGE", or "SITE_COMPONENT")
+- summary: Brief summary combining the chat history and current user request
+- "relatedSpis" field: ONLY for SERVICE_PLUGIN extensions - specify the SPI name
+- "paths" field: For ALL extension types - specify relevant file paths to modify
+- For currentExtensions, always include paths; for additionalExtensions, paths are not needed
+</output_format>
+
+</WIXCLI_ITERATION_AGENT_SYSTEM_PROMPT>
+`;
+    exports2.iterationAgentPrompt = iterationAgentPrompt;
+  }
+});
+
+// dist/agents/IterationAgent.js
+var require_IterationAgent = __commonJS({
+  "dist/agents/IterationAgent.js"(exports2) {
+    "use strict";
+    Object.defineProperty(exports2, "__esModule", { value: true });
+    exports2.IterationAgent = exports2.IterationPlanSchema = exports2.AdditionalExtensionSchema = exports2.CurrentExtensionSchema = void 0;
+    var anthropic_1 = require_dist5();
+    var ai_1 = require_dist7();
+    var zod_1 = require_zod();
+    var utils_1 = require_utils14();
+    var types_1 = require_types_impl();
+    var iterationAgentPrompt_1 = require_iterationAgentPrompt();
+    exports2.CurrentExtensionSchema = zod_1.z.object({
+      extensionName: zod_1.z.nativeEnum(types_1.ExtensionType).describe("The extension kind to trigger"),
+      relatedSpis: zod_1.z.array(zod_1.z.string()).optional().describe("Optional value, only for SPI extensions - specify the SPI types"),
+      paths: zod_1.z.array(zod_1.z.string()).describe("Paths relevant for this extension"),
+      relevantUserRequest: zod_1.z.string().describe("What part of the user request this extension addresses")
+    });
+    exports2.AdditionalExtensionSchema = zod_1.z.object({
+      extensionName: zod_1.z.nativeEnum(types_1.ExtensionType).describe("The extension kind to add"),
+      relatedSpis: zod_1.z.array(zod_1.z.string()).optional().describe("Optional value, only for SPI extensions - specify the SPI types"),
+      relevantUserRequest: zod_1.z.string().describe("What part of the user request this extension addresses")
+    });
+    exports2.IterationPlanSchema = zod_1.z.object({
+      currentExtensions: zod_1.z.array(exports2.CurrentExtensionSchema).describe("Existing extensions to be triggered"),
+      additionalExtensions: zod_1.z.array(exports2.AdditionalExtensionSchema).describe("New extensions to be created"),
+      summary: zod_1.z.string().describe("Summary of the chat with the user request")
+    });
+    var IterationAgent = class {
+      constructor(apiKey) {
+        this.apiKey = apiKey;
+        this.name = "IterationAgent";
+      }
+      buildSystemPrompt() {
+        return (0, iterationAgentPrompt_1.iterationAgentPrompt)();
+      }
+      mapExtensionType(extensionName) {
+        switch (extensionName) {
+          case "SERVICE_PLUGIN":
+            return types_1.ExtensionType.SERVICE_PLUGIN;
+          case "DASHBOARD_PAGE":
+            return types_1.ExtensionType.DASHBOARD_PAGE;
+          case "SITE_COMPONENT":
+            return types_1.ExtensionType.SITE_COMPONENT;
+          default:
+            throw new Error(`Unsupported extension type: ${extensionName}`);
+        }
+      }
+      async generate(outputPath, candidateFilePaths, currentUserRequest, chatHistory) {
+        const model = (0, anthropic_1.createAnthropic)({ apiKey: this.apiKey })("claude-3-5-haiku-latest");
+        const historyStr = (chatHistory || []).map((m) => `${m.role.toUpperCase()}: ${m.text}`).join("\n\n");
+        const allFilesContent = (0, utils_1.loadRelevantFilesAsString)(candidateFilePaths, outputPath);
+        const userContent = [
+          `
+Full Chat History:
+${historyStr}`,
+          `
+Current User Request: ${currentUserRequest}`,
+          `
+All Project Files (relative to output path):
+${candidateFilePaths.slice(0, 1e3).join("\n")}`,
+          `
+All Project Files Content:
+${allFilesContent}`
+        ].join("\n\n");
+        const result = await (0, ai_1.generateObject)({
+          model,
+          schema: exports2.IterationPlanSchema,
+          messages: [
+            {
+              role: "system",
+              content: this.buildSystemPrompt(),
+              providerOptions: (0, utils_1.withCaching)("1h")
+            },
+            { role: "user", content: userContent }
+          ],
+          maxRetries: 3,
+          temperature: 0,
+          experimental_telemetry: { isEnabled: true, functionId: this.name }
+        });
+        const resultObject = result.object;
+        resultObject.currentExtensions = result.object.currentExtensions.map((ext) => ({
+          ...ext,
+          extensionName: this.mapExtensionType(ext.extensionName)
+        }));
+        resultObject.additionalExtensions = result.object.additionalExtensions.map((ext) => ({
+          ...ext,
+          extensionName: this.mapExtensionType(ext.extensionName)
+        }));
+        console.log("IterationAgent result:", JSON.stringify(resultObject, null, 2));
+        return resultObject;
+      }
+    };
+    exports2.IterationAgent = IterationAgent;
+    exports2.default = IterationAgent;
+  }
+});
+
 // dist/agents/AgentsFactory.js
 var require_AgentsFactory = __commonJS({
   "dist/agents/AgentsFactory.js"(exports2) {
@@ -119181,6 +119505,7 @@ var require_AgentsFactory = __commonJS({
     var CMSAgent_1 = require_CMSAgent();
     var CMSDataAgent_1 = require_CMSDataAgent();
     var SiteComponentAgent_1 = __importDefault2(require_SiteComponentAgent());
+    var IterationAgent_1 = __importDefault2(require_IterationAgent());
     var AgentsFactory = class {
       constructor(apiKey) {
         this.apiKey = apiKey;
@@ -119197,6 +119522,8 @@ var require_AgentsFactory = __commonJS({
             return new CMSDataAgent_1.CMSDataAgent(this.apiKey);
           case "PLANNER":
             return new PlannerAgent_1.default(this.apiKey);
+          case "ITERATION_AGENT":
+            return new IterationAgent_1.default(this.apiKey);
           case types_1.ExtensionType.SITE_COMPONENT:
             return new SiteComponentAgent_1.default(this.apiKey);
           default:
@@ -119225,6 +119552,9 @@ var require_cli_listeners = __commonJS({
         console.log(`\u{1F527} Extensions to generate: ${info.extensionsCount}
 `);
       });
+      orchestrator.onEvent("start:iteration", () => {
+        console.log("\n\u{1F680} Starting code iteration for app");
+      });
       orchestrator.onEvent("scaffold:start", ({ extension }) => {
         console.log(`\u{1F4E6} Scaffolding ${extId(extension)}...`);
       });
@@ -119268,8 +119598,22 @@ var require_cli_listeners = __commonJS({
         timer.delete("PlannerAgent");
         console.log(`\u{1F916}\u2714\uFE0F  Generated plan in ${duration / 1e3}s`);
       });
+      orchestrator.onEvent("iterationAgent:start", () => {
+        timer.set("IterationAgent", Date.now());
+        console.log("\u{1F916} Generating iteration plan...");
+      });
+      orchestrator.onEvent("iterationAgent:done", ({ newExtensions, currentExtensions }) => {
+        const duration = Date.now() - timer.get("IterationAgent");
+        timer.delete("IterationAgent");
+        console.log(`\u{1F916}\u2714\uFE0F  Generated ${newExtensions} new extensions and modified ${currentExtensions} extensions in ${duration / 1e3}s`);
+      });
       orchestrator.onEvent("finish", ({ outputPath, durationMs }) => {
         console.log(`
+\u2705 Done in ${durationMs}ms`);
+        console.log(`\u{1F4C1} Files generated in: ${outputPath}`);
+      });
+      orchestrator.onEvent("finish:iteration", ({ outputPath, durationMs }) => {
+        console.log(`
 \u2705 Done in ${durationMs}ms`);
         console.log(`\u{1F4C1} Files generated in: ${outputPath}`);
       });
@@ -119575,6 +119919,10 @@ ${b}
   )`).join("\n")};
 `;
       const filePath = path_1.default.join(outputPath, "src", "extensions.ts");
+      const dir = path_1.default.dirname(filePath);
+      if (!fs_1.default.existsSync(dir)) {
+        fs_1.default.mkdirSync(dir, { recursive: true });
+      }
       fs_1.default.writeFileSync(filePath, content);
       console.log(`\u{1F4DD} Generated: ${path_1.default.relative(outputPath, filePath)}`);
     }
@@ -119868,24 +120216,15 @@ var require_orchestrator = __commonJS({
       return mod2 && mod2.__esModule ? mod2 : { "default": mod2 };
     };
     Object.defineProperty(exports2, "__esModule", { value: true });
-    exports2.DittoOrchestrator = exports2.FilesSchema = exports2.FileSchema = void 0;
+    exports2.DittoOrchestrator = void 0;
     var path_1 = __importDefault2(require("path"));
     var events_1 = require("events");
     var ditto_scaffolding_1 = require_dist11();
     var fs_1 = __importDefault2(require("fs"));
     var extensions_file_1 = require_extensions_file();
-    var zod_1 = require_zod();
     var ValidatorFactory_1 = __importDefault2(require_ValidatorFactory());
     var FixerFactory_1 = __importDefault2(require_FixerFactory());
     var MAX_FIX_ATTEMPTS = 3;
-    exports2.FileSchema = zod_1.z.object({
-      path: zod_1.z.string().describe("Relative file path from project root"),
-      content: zod_1.z.string().describe("Complete file content as a string (for JSON files, stringify the object)"),
-      type: zod_1.z.enum(["typescript", "json"]).describe("File type for syntax highlighting")
-    });
-    exports2.FilesSchema = zod_1.z.object({
-      files: zod_1.z.array(exports2.FileSchema).describe("Array of files to generate")
-    });
     var DittoOrchestrator = class extends events_1.EventEmitter {
       constructor(agentsFactory, apiKey) {
         super();
@@ -119900,21 +120239,54 @@ var require_orchestrator = __commonJS({
       emitEvent(event, payload) {
         return super.emit(event, payload);
       }
-      /**
-       * Generic method to generate code using Claude with tool execution
-       * @param systemPrompt - The system prompt to use
-       * @param userMessage - The initial user message
-       * @param outputPath - Where to output files
-       */
-      // Generation fully delegated to agents
-      writeFile(file, outputPath) {
-        const fullPath = path_1.default.join(outputPath, file.path);
-        const dir = path_1.default.dirname(fullPath);
-        if (!fs_1.default.existsSync(dir)) {
-          fs_1.default.mkdirSync(dir, { recursive: true });
+      writeFile(files, outputPath) {
+        if (!files) {
+          console.warn("\u26A0\uFE0F Skipping file operation: no files provided");
+          return;
+        }
+        for (const file of files) {
+          if (!file.path) {
+            console.warn("\u26A0\uFE0F Skipping file operation: no path specified");
+            continue;
+          }
+          const fullPath = path_1.default.join(outputPath, file.path);
+          switch (file.operation) {
+            case "insert":
+              if (!file.content) {
+                console.warn(`\u26A0\uFE0F Skipping insert operation for ${file.path}: no content provided`);
+                continue;
+              }
+              const dir = path_1.default.dirname(fullPath);
+              if (!fs_1.default.existsSync(dir)) {
+                fs_1.default.mkdirSync(dir, { recursive: true });
+              }
+              fs_1.default.writeFileSync(fullPath, file.content.trim());
+              console.log(`\u{1F4DD} Inserted: ${file.path}`);
+              break;
+            case "update":
+              if (!file.content) {
+                console.warn(`\u26A0\uFE0F Skipping update operation for ${file.path}: no content provided`);
+                continue;
+              }
+              if (!fs_1.default.existsSync(fullPath)) {
+                console.warn(`\u26A0\uFE0F Skipping update operation for ${file.path}: file does not exist`);
+                continue;
+              }
+              fs_1.default.writeFileSync(fullPath, file.content.trim());
+              console.log(`\u{1F4DD} Updated: ${file.path}`);
+              break;
+            case "delete":
+              if (fs_1.default.existsSync(fullPath)) {
+                fs_1.default.unlinkSync(fullPath);
+                console.log(`\u{1F5D1}\uFE0F Deleted: ${file.path}`);
+              } else {
+                console.warn(`\u26A0\uFE0F Skipping delete operation for ${file.path}: file does not exist`);
+              }
+              break;
+            default:
+              throw new Error(`Unknown operation "${file.operation}" for file ${file.path}`);
+          }
         }
-        fs_1.default.writeFileSync(fullPath, file.content.trim());
-        console.log(`\u{1F4DD} Generated: ${file.path}`);
       }
       async generatePlanAndResources(request) {
         const { siteId, accessToken, blueprint } = request;
@@ -119941,6 +120313,65 @@ var require_orchestrator = __commonJS({
         }
         return { createdCollections };
       }
+      async runIterationPlanningAndAugmentExtensions(outputPath, chatHistory) {
+        const iterationAgent = this.agentsFactory.getAgent({
+          type: "ITERATION_AGENT"
+        });
+        const candidateFiles = this.listGeneratedAppFiles(outputPath);
+        const currentUserRequest = chatHistory[chatHistory.length - 1]?.text || "";
+        const iterationPlan = await iterationAgent.generate(outputPath, candidateFiles, currentUserRequest, chatHistory);
+        return iterationPlan;
+      }
+      async processExtension({ extension, blueprint, outputPath, createdCollections }) {
+        this.emitEvent("scaffold:start", { extension });
+        const [scaffold] = await (0, ditto_scaffolding_1.copyScaffoldingTemplate)(extension, outputPath);
+        this.emitEvent("scaffold:done", {
+          extension,
+          scaffoldPath: scaffold.path
+        });
+        const agent = this.agentsFactory.getAgent(extension);
+        this.emitEvent("agent:start", { extension, name: agent.name });
+        const files = await agent.generate({
+          extension,
+          blueprint,
+          scaffold,
+          createdCollections,
+          basePath: outputPath
+        });
+        this.writeFile(files, outputPath);
+        this.emitEvent("agent:done", {
+          extension,
+          name: agent.name,
+          files
+        });
+      }
+      async processIterationExtension({ extension, paths, relevantUserRequest, outputPath, newExtension }) {
+        let scaffold;
+        if (newExtension) {
+          this.emitEvent("scaffold:start", { extension });
+          const [result] = await (0, ditto_scaffolding_1.copyScaffoldingTemplate)(extension, outputPath);
+          scaffold = result;
+          this.emitEvent("scaffold:done", {
+            extension,
+            scaffoldPath: scaffold.path
+          });
+        }
+        const agent = this.agentsFactory.getAgent(extension);
+        this.emitEvent("agent:start", { extension, name: agent.name });
+        const files = await agent.generate({
+          extension,
+          scaffold,
+          userRequestSummary: relevantUserRequest,
+          relevantFilePaths: paths,
+          basePath: outputPath
+        });
+        this.writeFile(files, outputPath);
+        this.emitEvent("agent:done", {
+          extension,
+          name: agent.name,
+          files
+        });
+      }
       async generateCMSData(request, createdCollections) {
         const { siteId, accessToken, blueprint } = request;
         this.emitEvent("cms-data:start", {
@@ -119959,10 +120390,6 @@ var require_orchestrator = __commonJS({
           totalItemsInserted
         });
       }
-      /**
-       * AI Generation Methods for customizing scaffolding
-       * Currently supports: SERVICE_PLUGIN, DASHBOARD_PAGE
-       */
       async generateCode(request) {
         const { blueprint, outputPath, siteId, accessToken } = request;
         this.emitEvent("start", {
@@ -119978,33 +120405,16 @@ var require_orchestrator = __commonJS({
           const planAndResourcesResult = await this.generatePlanAndResources(request);
           createdCollections.push(...planAndResourcesResult.createdCollections);
         }
-        const parallelTasks = extensions.map(async (extension) => {
-          this.emitEvent("scaffold:start", { extension });
-          const [scaffold] = await (0, ditto_scaffolding_1.copyScaffoldingTemplate)(extension, outputPath);
-          if (!scaffold) {
-            console.warn("Failed to scaffold for extension", extension);
-          }
-          this.emitEvent("scaffold:done", {
-            extension,
-            scaffoldPath: scaffold.path
-          });
-          const agent = this.agentsFactory.getAgent(extension);
-          this.emitEvent("agent:start", { extension, name: agent.name });
-          const file = await agent.generate(extension, blueprint, scaffold, createdCollections);
-          const files = [
-            { path: scaffold.path, content: file.content }
-          ];
-          this.writeFile(files[0], outputPath);
-          this.emitEvent("agent:done", {
-            extension,
-            name: agent.name,
-            files
-          });
-        });
+        const parallelTasks = extensions.map((extension) => this.processExtension({
+          extension,
+          blueprint,
+          outputPath,
+          createdCollections
+        }));
         if (createdCollections.length > 0 && siteId && accessToken) {
           parallelTasks.push(this.generateCMSData(request, createdCollections));
         }
-        await Promise.all(parallelTasks);
+        await Promise.all(parallelTasks.filter(Boolean));
         (0, extensions_file_1.generateExtensionsFile)(outputPath, extensions);
         const durationMsCodeGeneration = Date.now() - start;
         this.emitEvent("finish:code-generation", {
@@ -120018,6 +120428,80 @@ var require_orchestrator = __commonJS({
         const durationMs = Date.now() - start;
         this.emitEvent("finish", { outputPath, durationMs });
       }
+      async generateIterationCode(request) {
+        this.emitEvent("start:iteration", {});
+        const { outputPath, chatHistory } = request;
+        const start = Date.now();
+        this.emitEvent("iterationAgent:start", {});
+        const iterationPlan = await this.runIterationPlanningAndAugmentExtensions(outputPath, chatHistory);
+        this.emitEvent("iterationAgent:done", {
+          newExtensions: iterationPlan?.additionalExtensions?.length || 0,
+          currentExtensions: iterationPlan?.currentExtensions?.length || 0
+        });
+        const parallelTasks = iterationPlan?.additionalExtensions?.map((extension) => {
+          const extensionObj = {
+            type: extension.extensionName,
+            name: "",
+            description: `Generated from iteration: ${extension.relevantUserRequest}`,
+            relatedSpis: extension.relatedSpis?.map((spi) => ({
+              name: spi,
+              purpose: ""
+            })) || []
+          };
+          return this.processIterationExtension({
+            extension: extensionObj,
+            paths: [],
+            relevantUserRequest: iterationPlan.summary,
+            outputPath,
+            newExtension: true
+          });
+        }) || [];
+        if (iterationPlan?.currentExtensions && iterationPlan.currentExtensions.length > 0) {
+          parallelTasks.push(...iterationPlan.currentExtensions.map((extension) => {
+            const extensionObj = {
+              type: extension.extensionName,
+              name: "",
+              description: `Generated from iteration: ${extension.relevantUserRequest}`,
+              relatedSpis: extension.relatedSpis?.map((spi) => ({
+                name: spi,
+                purpose: ""
+              })) || []
+            };
+            return this.processIterationExtension({
+              extension: extensionObj,
+              paths: extension.paths,
+              relevantUserRequest: extension.relevantUserRequest,
+              outputPath,
+              newExtension: false
+            });
+          }));
+        }
+        await Promise.all(parallelTasks.filter(Boolean));
+        const finalExtensionsToWrite = iterationPlan?.additionalExtensions?.map((extension) => {
+          const extensionObj = {
+            type: extension.extensionName,
+            name: "",
+            description: `Generated from iteration: ${extension.relevantUserRequest}`,
+            relatedSpis: extension.relatedSpis?.map((spi) => ({
+              name: spi,
+              purpose: ""
+            })) || []
+          };
+          return extensionObj;
+        });
+        (0, extensions_file_1.generateExtensionsFile)(outputPath, finalExtensionsToWrite);
+        const durationMsCodeGeneration = Date.now() - start;
+        this.emitEvent("finish:code-generation", {
+          outputPath,
+          durationMs: durationMsCodeGeneration
+        });
+        await this.startFixFlow({
+          projectDir: outputPath,
+          apiKey: this.apiKey
+        });
+        const durationMs = Date.now() - start;
+        this.emitEvent("finish:iteration", { outputPath, durationMs });
+      }
       async startFixFlow(request) {
         const { projectDir, apiKey } = request;
         this.emitEvent("validation:start", { outputPath: projectDir });
@@ -120053,6 +120537,26 @@ ${unfixedValidationErrors}`)
         throw new Error(`Validation failed after ${MAX_FIX_ATTEMPTS} attempts. Errors: 
 ${unfixedValidationErrors}`);
       }
+      listGeneratedAppFiles(outputPath) {
+        const root = outputPath;
+        const srcDir = path_1.default.join(root, "src");
+        const results = [];
+        if (!fs_1.default.existsSync(srcDir)) {
+          return results;
+        }
+        const walk = (dir) => {
+          const entries = fs_1.default.existsSync(dir) ? fs_1.default.readdirSync(dir, { withFileTypes: true }) : [];
+          for (const entry of entries) {
+            const full = path_1.default.join(dir, entry.name);
+            if (entry.isDirectory())
+              walk(full);
+            else
+              results.push(path_1.default.relative(root, full));
+          }
+        };
+        walk(srcDir);
+        return results;
+      }
     };
     exports2.DittoOrchestrator = DittoOrchestrator;
     DittoOrchestrator.DEFAULT_OUTPUT_PATH = "generated-app";
@@ -120172,6 +120676,72 @@ var require_init_codegen = __commonJS({
   }
 });
 
+// dist/flows/iterate-codegen.js
+var require_iterate_codegen = __commonJS({
+  "dist/flows/iterate-codegen.js"(exports2) {
+    "use strict";
+    var __importDefault2 = exports2 && exports2.__importDefault || function(mod2) {
+      return mod2 && mod2.__esModule ? mod2 : { "default": mod2 };
+    };
+    Object.defineProperty(exports2, "__esModule", { value: true });
+    exports2.runIterateCodegenFlow = void 0;
+    var path_1 = __importDefault2(require("path"));
+    var __1 = require_index();
+    var AgentsFactory_1 = require_AgentsFactory();
+    var cli_listeners_1 = require_cli_listeners();
+    var context_1 = require_context();
+    var job_context_storage_1 = require_job_context_storage();
+    var orchestrator_1 = require_orchestrator();
+    var CodeGenService_1 = require_CodeGenService();
+    var utils_1 = require_utils18();
+    var slugify = (str = "") => {
+      return str.toLowerCase().replace(/ /g, "-");
+    };
+    var runIterateCodegenFlow = async (chatHistory) => {
+      const localJobContext = job_context_storage_1.jobContextStorage.getStore();
+      console.log(`[Iterate] Starting iterate codegen task: jobId=${localJobContext?.jobId}, taskId=${localJobContext?.taskId}`);
+      await __1.codeGenerationService.updateTask(localJobContext.jobId, localJobContext.taskId, CodeGenService_1.TaskStatus.RUNNING, {});
+      console.log(`[Init] Marked task RUNNING: jobId=${localJobContext.jobId}, taskId=${localJobContext.taskId}`);
+      try {
+        const outputDir = process.env.OUTPUT_PATH || orchestrator_1.DittoOrchestrator.DEFAULT_OUTPUT_PATH;
+        const outputPath = outputDir.startsWith("/") ? outputDir : path_1.default.join(process.cwd(), outputDir);
+        const agentsFactory = new AgentsFactory_1.AgentsFactory(context_1.ctx?.apiKey);
+        const orchestrator = new orchestrator_1.DittoOrchestrator(agentsFactory);
+        orchestrator.onEvent("agent:start", ({ extension }) => {
+          console.log(`[Agent] start: ${extension.name}`);
+          __1.codeGenerationService.addTask(localJobContext.jobId, {
+            id: `${localJobContext.taskId}-${slugify(extension.name)}`,
+            kind: "run_agent",
+            status: CodeGenService_1.TaskStatus.RUNNING,
+            name: "run_agent",
+            description: `Run agent for ${extension.name}`,
+            payload: { extension }
+          });
+        });
+        orchestrator.onEvent("agent:done", ({ extension, files }) => {
+          console.log(`[Agent] done: ${extension.name}`);
+          __1.codeGenerationService.updateTask(localJobContext.jobId, `${localJobContext.taskId}-${slugify(extension.name)}`, CodeGenService_1.TaskStatus.COMPLETED, { taskOutput: { files } });
+        });
+        (0, cli_listeners_1.attachOrchestratorListeners)(orchestrator);
+        await orchestrator.generateIterationCode({
+          chatHistory,
+          outputPath,
+          siteId: context_1.ctx.siteId,
+          accessToken: context_1.ctx.accessToken
+        });
+        await __1.codeGenerationService.updateTask(localJobContext.jobId, localJobContext.taskId, CodeGenService_1.TaskStatus.COMPLETED, {});
+        console.log(`[Init] Completed iterate codegen task: jobId=${localJobContext.jobId}, taskId=${localJobContext.taskId}`);
+      } catch (error) {
+        await __1.codeGenerationService.updateTask(localJobContext.jobId, localJobContext.taskId, CodeGenService_1.TaskStatus.FAILED, {
+          error: (0, utils_1.serializeError)(error)
+        });
+        console.error(`\u274C [Init] Failed iterate codegen task: jobId=${localJobContext.jobId}, taskId=${localJobContext.taskId}`, error);
+      }
+    };
+    exports2.runIterateCodegenFlow = runIterateCodegenFlow;
+  }
+});
+
 // dist/index.js
 var require_index = __commonJS({
   "dist/index.js"(exports2) {
@@ -120219,6 +120789,7 @@ var require_index = __commonJS({
     var CodeGenService_1 = __importStar2(require_CodeGenService());
     var job_context_storage_1 = require_job_context_storage();
     var init_codegen_1 = require_init_codegen();
+    var iterate_codegen_1 = require_iterate_codegen();
     var initializeTracing = async () => {
       try {
         await instrumentation_1.tracingReady;
@@ -120279,8 +120850,14 @@ var require_index = __commonJS({
       context_1.ctx.setJobStatus(job.jobId, CodeGenService_1.TaskStatus.RUNNING);
       console.log(`[Job] Marked job RUNNING: jobId=${job.jobId}, initTaskId=${task.id}`);
       console.log("Task Payload", JSON.stringify(task.payload, null, 2));
-      await job_context_storage_1.jobContextStorage.run({ jobId: job.jobId, taskId: task.id }, async () => {
-        await (0, init_codegen_1.runInitCodegenFlow)(task.payload.blueprint);
+      await job_context_storage_1.jobContextStorage.run({ jobId: job.jobId, taskId: task.id, kind: task.kind }, async () => {
+        const payload = task?.payload || {};
+        if (task.kind === "ITERATE_CODEGEN") {
+          const chatHistory = payload.history ?? [];
+          await (0, iterate_codegen_1.runIterateCodegenFlow)(chatHistory);
+        } else if (task.kind === "INIT_CODEGEN") {
+          await (0, init_codegen_1.runInitCodegenFlow)(payload.blueprint);
+        }
       });
     }
     (0, context_1.initCtx)("dafdsf").then((ctx) => {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@wix/ditto-codegen-public",
-  "version": "1.0.16",
+  "version": "1.0.17",
   "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.25.9"
   },
-  "falconPackageHash": "8ddfeb0447ecad792ac842099f8b1b9f7ff9a88a00c7dffe496475c3"
+  "falconPackageHash": "70fbc8a04729200a2076d5870d3ced40e10ef55b8ce324003c82c4bf"
 }