@wix/ditto-codegen-public 1.0.204 → 1.0.205

package/dist/out.js

@@ -38994,6 +38994,58 @@ var require_sdkConfig = __commonJS({
   }
 });
 
+// ../codegen-common-logic/dist/generate-text-helpers.js
+var require_generate_text_helpers = __commonJS({
+  "../codegen-common-logic/dist/generate-text-helpers.js"(exports2) {
+    "use strict";
+    Object.defineProperty(exports2, "__esModule", { value: true });
+    exports2.extractToolResult = extractToolResult;
+    exports2.stopOnToolCall = stopOnToolCall;
+    exports2.stopOnNoToolCalls = stopOnNoToolCalls;
+    exports2.generateAgentText = generateAgentText;
+    var ai_1 = require_dist8();
+    var ditto_codegen_types_12 = require_dist();
+    function extractToolResult(toolCalls, toolName) {
+      const call = toolCalls.find((tc) => tc.toolName === toolName);
+      return call?.input ?? null;
+    }
+    function stopOnToolCall(toolName) {
+      return ({ steps }) => {
+        const lastStep = steps[steps.length - 1];
+        return lastStep?.toolCalls.some((tc) => tc.toolName === toolName && !("invalid" in tc) && !("error" in tc)) ?? false;
+      };
+    }
+    function stopOnNoToolCalls(maxSteps) {
+      return ({ steps }) => {
+        if (steps.length >= maxSteps) {
+          return true;
+        }
+        const lastStep = steps[steps.length - 1];
+        return lastStep?.toolCalls.length === 0;
+      };
+    }
+    async function generateAgentText(options) {
+      return (0, ai_1.generateText)({
+        model: options.model,
+        tools: options.tools,
+        stopWhen: options.stopWhen,
+        messages: [
+          {
+            role: "system",
+            content: options.systemPrompt,
+            // Anthropic-specific: enables prompt caching. Ignored by other providers.
+            providerOptions: (0, ditto_codegen_types_12.withCaching)("1h")
+          },
+          { role: "user", content: options.userMessage }
+        ],
+        maxRetries: 3,
+        temperature: 0,
+        experimental_telemetry: { isEnabled: true, functionId: options.agentName }
+      });
+    }
+  }
+});
+
 // ../codegen-common-logic/dist/index.js
 var require_dist9 = __commonJS({
   "../codegen-common-logic/dist/index.js"(exports2) {
@@ -39033,6 +39085,7 @@ var require_dist9 = __commonJS({
     __exportStar2(require_commonUseCasesGuide(), exports2);
     __exportStar2(require_system_prompts(), exports2);
     __exportStar2(require_sdkConfig(), exports2);
+    __exportStar2(require_generate_text_helpers(), exports2);
   }
 });
 
@@ -72599,6 +72652,104 @@ var require_prompt_selectors = __commonJS({
   }
 });
 
+// dist/agents/utils/api.js
+var require_api2 = __commonJS({
+  "dist/agents/utils/api.js"(exports2) {
+    "use strict";
+    Object.defineProperty(exports2, "__esModule", { value: true });
+    exports2.getApiDocumentation = exports2.extractApis = exports2.extractApiNames = exports2.extractApiDocumentation = void 0;
+    var isExtensionEnriched = (extension) => {
+      return extension.relatedApis?.some((api) => "documentation" in api || "methodSchema" in api) ?? false;
+    };
+    var extractApiDocumentation = (extension) => {
+      if (isExtensionEnriched(extension)) {
+        return { apiDocumentation: (0, exports2.getApiDocumentation)(extension), apiNames: [] };
+      }
+      return { apiDocumentation: "", apiNames: (0, exports2.extractApiNames)(extension) };
+    };
+    exports2.extractApiDocumentation = extractApiDocumentation;
+    var extractApiNames = (extension) => {
+      return extension.relatedApis?.map((api) => api.name).filter((name) => !!name) || [];
+    };
+    exports2.extractApiNames = extractApiNames;
+    var extractApis = (extension) => {
+      return extension.relatedApis?.map((api) => ({
+        name: api.name,
+        purpose: api.purpose,
+        documentation: api.documentation,
+        methodSchema: api.methodSchema
+      })) || [];
+    };
+    exports2.extractApis = extractApis;
+    var getApiDocumentation = (extension) => {
+      const apis = (0, exports2.extractApis)(extension);
+      const docsWithContent = apis.filter((api) => api.documentation || api.methodSchema);
+      if (docsWithContent.length === 0) {
+        return "";
+      }
+      return docsWithContent.map((api) => {
+        const parts = [];
+        parts.push(`<api name="${api.name}">`);
+        if (api.purpose) {
+          parts.push(`<purpose>${api.purpose}</purpose>`);
+        }
+        if (api.documentation) {
+          parts.push(`<documentation>${api.documentation}</documentation>`);
+        }
+        if (api.methodSchema) {
+          parts.push(`<methodSchema>${api.methodSchema}</methodSchema>`);
+        }
+        parts.push(`</api>`);
+        return parts.join("\n");
+      }).join("\n\n");
+    };
+    exports2.getApiDocumentation = getApiDocumentation;
+  }
+});
+
+// dist/agents/utils/file-operations.js
+var require_file_operations = __commonJS({
+  "dist/agents/utils/file-operations.js"(exports2) {
+    "use strict";
+    Object.defineProperty(exports2, "__esModule", { value: true });
+    exports2.FilesSchema = exports2.FileItemSchema = exports2.FileCreateOrUpdateSchema = void 0;
+    var zod_1 = require_zod();
+    var ditto_codegen_types_12 = require_dist();
+    exports2.FileCreateOrUpdateSchema = zod_1.z.object({
+      operation: zod_1.z.union([
+        zod_1.z.literal(ditto_codegen_types_12.ExtensionGenerationOperation.INSERT),
+        zod_1.z.literal(ditto_codegen_types_12.ExtensionGenerationOperation.UPDATE)
+      ]).describe("File operation: insert (new file), update (modify existing)").default(ditto_codegen_types_12.ExtensionGenerationOperation.INSERT),
+      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). Required for insert and update operations.")
+    });
+    var FileEditSchema = zod_1.z.object({
+      operation: zod_1.z.literal(ditto_codegen_types_12.ExtensionGenerationOperation.EDIT),
+      path: zod_1.z.string().describe("Relative file path from project root"),
+      replacements: zod_1.z.array(zod_1.z.object({
+        oldString: zod_1.z.string().describe("The text to replace (must match the file contents exactly, including all whitespace and indentation)"),
+        newString: zod_1.z.string().describe("The edited text to replace the oldString"),
+        replaceAll: zod_1.z.boolean().optional().describe("Replace all occurrences of oldString (default false)")
+      })).min(1).describe("List of string replacements to apply")
+    });
+    var FileDeleteSchema = zod_1.z.object({
+      operation: zod_1.z.literal(ditto_codegen_types_12.ExtensionGenerationOperation.DELETE),
+      path: zod_1.z.string().describe("Relative file path from project root")
+    });
+    exports2.FileItemSchema = zod_1.z.discriminatedUnion("operation", [
+      exports2.FileCreateOrUpdateSchema,
+      FileDeleteSchema,
+      FileEditSchema
+    ]).describe(`Representation of an operation that modifies a file on the local file system.
+Use the 'insert' operation to create a new file.
+Use the 'delete' operation to remove a file.
+To modify a file, you can either use the 'edit' operation to replace a specific strings with new strings (THIS IS THE PREFERRED WAY TO MODIFY A FILE), or use the 'update' operation to replace the entire file with new content.`);
+    exports2.FilesSchema = zod_1.z.object({
+      files: zod_1.z.array(exports2.FileItemSchema).default([]).describe("An array of files")
+    });
+  }
+});
+
 // dist/agents/utils/file-loading.js
 var require_file_loading = __commonJS({
   "dist/agents/utils/file-loading.js"(exports2) {
@@ -72765,101 +72916,43 @@ ${JSON.stringify(formattedOps, null, 2)}`;
   }
 });
 
-// dist/agents/utils/file-operations.js
-var require_file_operations = __commonJS({
-  "dist/agents/utils/file-operations.js"(exports2) {
-    "use strict";
-    Object.defineProperty(exports2, "__esModule", { value: true });
-    exports2.FilesSchema = exports2.FileItemSchema = exports2.FileCreateOrUpdateSchema = void 0;
-    var zod_1 = require_zod();
-    var ditto_codegen_types_12 = require_dist();
-    exports2.FileCreateOrUpdateSchema = zod_1.z.object({
-      operation: zod_1.z.union([
-        zod_1.z.literal(ditto_codegen_types_12.ExtensionGenerationOperation.INSERT),
-        zod_1.z.literal(ditto_codegen_types_12.ExtensionGenerationOperation.UPDATE)
-      ]).describe("File operation: insert (new file), update (modify existing)").default(ditto_codegen_types_12.ExtensionGenerationOperation.INSERT),
-      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). Required for insert and update operations.")
-    });
-    var FileEditSchema = zod_1.z.object({
-      operation: zod_1.z.literal(ditto_codegen_types_12.ExtensionGenerationOperation.EDIT),
-      path: zod_1.z.string().describe("Relative file path from project root"),
-      replacements: zod_1.z.array(zod_1.z.object({
-        oldString: zod_1.z.string().describe("The text to replace (must match the file contents exactly, including all whitespace and indentation)"),
-        newString: zod_1.z.string().describe("The edited text to replace the oldString"),
-        replaceAll: zod_1.z.boolean().optional().describe("Replace all occurrences of oldString (default false)")
-      })).min(1).describe("List of string replacements to apply")
-    });
-    var FileDeleteSchema = zod_1.z.object({
-      operation: zod_1.z.literal(ditto_codegen_types_12.ExtensionGenerationOperation.DELETE),
-      path: zod_1.z.string().describe("Relative file path from project root")
-    });
-    exports2.FileItemSchema = zod_1.z.discriminatedUnion("operation", [
-      exports2.FileCreateOrUpdateSchema,
-      FileDeleteSchema,
-      FileEditSchema
-    ]).describe(`Representation of an operation that modifies a file on the local file system.
-Use the 'insert' operation to create a new file.
-Use the 'delete' operation to remove a file.
-To modify a file, you can either use the 'edit' operation to replace a specific strings with new strings (THIS IS THE PREFERRED WAY TO MODIFY A FILE), or use the 'update' operation to replace the entire file with new content.`);
-    exports2.FilesSchema = zod_1.z.object({
-      files: zod_1.z.array(exports2.FileItemSchema).default([]).describe("An array of files")
-    });
-  }
-});
-
-// dist/agents/utils/api.js
-var require_api2 = __commonJS({
-  "dist/agents/utils/api.js"(exports2) {
+// dist/agents/BaseCodeGenerationAgent.js
+var require_BaseCodeGenerationAgent = __commonJS({
+  "dist/agents/BaseCodeGenerationAgent.js"(exports2) {
     "use strict";
     Object.defineProperty(exports2, "__esModule", { value: true });
-    exports2.getApiDocumentation = exports2.extractApis = exports2.extractApiNames = exports2.extractApiDocumentation = void 0;
-    var isExtensionEnriched = (extension) => {
-      return extension.relatedApis?.some((api) => "documentation" in api || "methodSchema" in api) ?? false;
-    };
-    var extractApiDocumentation = (extension) => {
-      if (isExtensionEnriched(extension)) {
-        return { apiDocumentation: (0, exports2.getApiDocumentation)(extension), apiNames: [] };
+    exports2.BaseCodeGenerationAgent = void 0;
+    var file_operations_1 = require_file_operations();
+    var userPrompt_1 = require_userPrompt();
+    var codeGenerationService_12 = require_codeGenerationService();
+    var constants_1 = require_constants();
+    var BaseCodeGenerationAgent = class {
+      constructor(useIteration = false) {
+        this.useIteration = useIteration;
       }
-      return { apiDocumentation: "", apiNames: (0, exports2.extractApiNames)(extension) };
-    };
-    exports2.extractApiDocumentation = extractApiDocumentation;
-    var extractApiNames = (extension) => {
-      return extension.relatedApis?.map((api) => api.name).filter((name) => !!name) || [];
-    };
-    exports2.extractApiNames = extractApiNames;
-    var extractApis = (extension) => {
-      return extension.relatedApis?.map((api) => ({
-        name: api.name,
-        purpose: api.purpose,
-        documentation: api.documentation,
-        methodSchema: api.methodSchema
-      })) || [];
-    };
-    exports2.extractApis = extractApis;
-    var getApiDocumentation = (extension) => {
-      const apis = (0, exports2.extractApis)(extension);
-      const docsWithContent = apis.filter((api) => api.documentation || api.methodSchema);
-      if (docsWithContent.length === 0) {
-        return "";
+      getGenerateOptions() {
+        return {
+          schema: file_operations_1.FilesSchema,
+          extractResult: (obj) => obj.files
+        };
+      }
+      async generate(params) {
+        const systemPrompt = await this.buildSystemPrompt(params);
+        const primaryAction = this.getPrimaryAction(params);
+        const userMessage = (0, userPrompt_1.buildUserPromptForCodeGenerationAgent)(params, primaryAction);
+        const { schema, model, extractResult } = this.getGenerateOptions();
+        const result = await codeGenerationService_12.codegenAIProxyService.generateCodegenObject({
+          userMessage,
+          systemPrompt,
+          provider: constants_1.LLM_PROVIDERS.ANTHROPIC,
+          model: model ?? constants_1.LLM_MODELS.CLAUDE_SONNET_4_5_20250929,
+          schema,
+          agentName: this.name
+        });
+        return extractResult(result.object);
       }
-      return docsWithContent.map((api) => {
-        const parts = [];
-        parts.push(`<api name="${api.name}">`);
-        if (api.purpose) {
-          parts.push(`<purpose>${api.purpose}</purpose>`);
-        }
-        if (api.documentation) {
-          parts.push(`<documentation>${api.documentation}</documentation>`);
-        }
-        if (api.methodSchema) {
-          parts.push(`<methodSchema>${api.methodSchema}</methodSchema>`);
-        }
-        parts.push(`</api>`);
-        return parts.join("\n");
-      }).join("\n\n");
     };
-    exports2.getApiDocumentation = getApiDocumentation;
+    exports2.BaseCodeGenerationAgent = BaseCodeGenerationAgent;
   }
 });
 
@@ -72877,33 +72970,19 @@ var require_DashboardPageAgent2 = __commonJS({
     var dashboard_page_prompt_1 = require_dashboard_page_prompt();
     var prompt_selectors_1 = require_prompt_selectors();
     var codegen_common_logic_1 = require_dist9();
-    var userPrompt_1 = require_userPrompt();
-    var file_operations_1 = require_file_operations();
     var codeGenerationService_12 = require_codeGenerationService();
     var constants_1 = require_constants();
     var codegen_dashboard_agents_1 = require_dist11();
     var api_1 = require_api2();
-    var DashboardPageAgent = class {
-      constructor(useIteration = false) {
-        this.useIteration = useIteration;
+    var BaseCodeGenerationAgent_1 = require_BaseCodeGenerationAgent();
+    var DashboardPageAgent = class extends BaseCodeGenerationAgent_1.BaseCodeGenerationAgent {
+      constructor() {
+        super(...arguments);
         this.name = "DashboardPageAgent";
       }
       getAgentDocumentation() {
         return (0, codegen_common_logic_1.getDashboardPageAgentDocumentation)();
       }
-      async buildSystemPrompt({ useData, useAutoPatterns, useApiSpec, useDynamicParameters, apiNames, apiDocumentation, autoPatternsSectionIds }) {
-        const systemPrompt = await (0, dashboard_page_prompt_1.dashboardPagePrompt)({
-          useData,
-          useAutoPatterns,
-          useApiSpec,
-          useDynamicParameters,
-          apiNames,
-          apiDocumentation,
-          useIteration: this.useIteration,
-          autoPatternsSectionIds
-        });
-        return systemPrompt;
-      }
       async getAutoPatternsSectionIds(useAutoPatterns, userRequestSummary) {
         if (!useAutoPatterns || !this.useIteration || !userRequestSummary) {
           return void 0;
@@ -72917,7 +72996,7 @@ var require_DashboardPageAgent2 = __commonJS({
         const result = await autoPatternsDocSelector.generate(userRequestSummary);
         return result.selectedSectionIds;
       }
-      // temporary fix to add all files in the auto patterns directory as relevant files
+      // Temporary fix to add all files in the auto patterns directory as relevant files
       addAutoPatternsRelevantFiles(params) {
         const relevantFilePaths = params.relevantFilePaths ?? [];
         const patternsJsonPath = relevantFilePaths.find((p) => p.endsWith("patterns.json"));
@@ -72935,8 +73014,8 @@ var require_DashboardPageAgent2 = __commonJS({
           }
         }
       }
-      async generate(params) {
-        const { blueprint, plan, userRequestSummary } = params;
+      async buildSystemPrompt(params) {
+        const { plan, userRequestSummary } = params;
         const useData = (0, prompt_selectors_1.shouldUseDataPrompt)(params);
         const useAutoPatterns = (0, prompt_selectors_1.shouldUseAutoPatterns)(params);
         const useApiSpec = Boolean(plan?.apiSpec);
@@ -72946,27 +73025,20 @@ var require_DashboardPageAgent2 = __commonJS({
           this.addAutoPatternsRelevantFiles(params);
         }
         const { apiDocumentation, apiNames } = (0, api_1.extractApiDocumentation)(params.extension);
-        const systemPrompt = await this.buildSystemPrompt({
+        return (0, dashboard_page_prompt_1.dashboardPagePrompt)({
           useData,
           useAutoPatterns,
           useApiSpec,
           useDynamicParameters,
           apiNames,
           apiDocumentation,
+          useIteration: this.useIteration,
           autoPatternsSectionIds
         });
-        const appName = blueprint?.appName ? `'${blueprint.appName}'` : "";
-        const primaryAction = `Customize the Wix CLI dashboard page scaffolding for the app ${appName}`;
-        const userMessage = (0, userPrompt_1.buildUserPromptForCodeGenerationAgent)(params, primaryAction);
-        const result = await codeGenerationService_12.codegenAIProxyService.generateCodegenObject({
-          userMessage,
-          systemPrompt,
-          provider: constants_1.LLM_PROVIDERS.ANTHROPIC,
-          model: constants_1.LLM_MODELS.CLAUDE_SONNET_4_5_20250929,
-          schema: file_operations_1.FilesSchema,
-          agentName: this.name
-        });
-        return result.object.files;
+      }
+      getPrimaryAction(params) {
+        const appName = params.blueprint?.appName ? `'${params.blueprint.appName}'` : "";
+        return `Customize the Wix CLI dashboard page scaffolding for the app ${appName}`;
       }
     };
     exports2.DashboardPageAgent = DashboardPageAgent;
@@ -73043,19 +73115,18 @@ var require_SiteComponentAgent2 = __commonJS({
     var api_1 = require_api2();
     var site_component_prompt_1 = require_site_component_prompt();
     var codegen_common_logic_1 = require_dist9();
-    var userPrompt_1 = require_userPrompt();
-    var file_operations_1 = require_file_operations();
-    var constants_1 = require_constants();
-    var codeGenerationService_12 = require_codeGenerationService();
-    var SiteComponentAgent = class {
-      constructor(useIteration = false) {
-        this.useIteration = useIteration;
+    var BaseCodeGenerationAgent_1 = require_BaseCodeGenerationAgent();
+    var SiteComponentAgent = class extends BaseCodeGenerationAgent_1.BaseCodeGenerationAgent {
+      constructor() {
+        super(...arguments);
         this.name = "SiteComponentAgent";
       }
       getAgentDocumentation() {
         return (0, codegen_common_logic_1.getSiteComponentAgentDocumentation)();
       }
-      buildSystemPrompt({ apiNames, apiDocumentation, useData }) {
+      buildSystemPrompt(params) {
+        const { apiDocumentation, apiNames } = (0, api_1.extractApiDocumentation)(params.extension);
+        const useData = Boolean(params.plan?.collections?.length);
         return (0, site_component_prompt_1.siteComponentPrompt)({
           apiNames,
           useData,
@@ -73063,27 +73134,9 @@ var require_SiteComponentAgent2 = __commonJS({
           apiDocumentation
         });
       }
-      async generate(params) {
-        const { blueprint, plan } = params;
-        const { apiDocumentation, apiNames } = (0, api_1.extractApiDocumentation)(params.extension);
-        const useData = Boolean(plan?.collections?.length);
-        const systemPrompt = this.buildSystemPrompt({
-          apiNames,
-          apiDocumentation,
-          useData
-        });
-        const appName = blueprint?.appName ? `'${blueprint.appName}'` : "";
-        const primaryAction = `Customize the Wix CLI site component scaffolding for the app ${appName}`;
-        const userMessage = (0, userPrompt_1.buildUserPromptForCodeGenerationAgent)(params, primaryAction);
-        const result = await codeGenerationService_12.codegenAIProxyService.generateCodegenObject({
-          userMessage,
-          systemPrompt,
-          provider: constants_1.LLM_PROVIDERS.ANTHROPIC,
-          model: constants_1.LLM_MODELS.CLAUDE_SONNET_4_5_20250929,
-          schema: file_operations_1.FilesSchema,
-          agentName: this.name
-        });
-        return result.object.files;
+      getPrimaryAction(params) {
+        const appName = params.blueprint?.appName ? `'${params.blueprint.appName}'` : "";
+        return `Customize the Wix CLI site component scaffolding for the app ${appName}`;
       }
     };
     exports2.SiteComponentAgent = SiteComponentAgent;
@@ -73206,19 +73259,18 @@ var require_CustomElementAgent2 = __commonJS({
     var api_1 = require_api2();
     var custom_element_prompt_1 = require_custom_element_prompt();
     var codegen_common_logic_1 = require_dist9();
-    var userPrompt_1 = require_userPrompt();
-    var codeGenerationService_12 = require_codeGenerationService();
-    var file_operations_1 = require_file_operations();
-    var constants_1 = require_constants();
-    var CustomElementAgent = class {
-      constructor(useIteration = false) {
-        this.useIteration = useIteration;
+    var BaseCodeGenerationAgent_1 = require_BaseCodeGenerationAgent();
+    var CustomElementAgent = class extends BaseCodeGenerationAgent_1.BaseCodeGenerationAgent {
+      constructor() {
+        super(...arguments);
         this.name = "CustomElementAgent";
       }
       getAgentDocumentation() {
         return (0, codegen_common_logic_1.getCustomElementAgentDocumentation)();
       }
-      buildSystemPrompt({ apiNames, apiDocumentation, useData }) {
+      buildSystemPrompt(params) {
+        const { apiDocumentation, apiNames } = (0, api_1.extractApiDocumentation)(params.extension);
+        const useData = Boolean(params.plan?.collections?.length);
         return (0, custom_element_prompt_1.customElementPrompt)({
           apiNames,
           useData,
@@ -73226,27 +73278,9 @@ var require_CustomElementAgent2 = __commonJS({
           apiDocumentation
         });
       }
-      async generate(params) {
-        const { blueprint, plan } = params;
-        const { apiDocumentation, apiNames } = (0, api_1.extractApiDocumentation)(params.extension);
-        const useData = Boolean(plan?.collections?.length);
-        const systemPrompt = this.buildSystemPrompt({
-          apiNames,
-          apiDocumentation,
-          useData
-        });
-        const appName = blueprint?.appName ? `'${blueprint.appName}'` : "";
-        const primaryAction = `Customize the Wix CLI custom element / site widget scaffolding for the app ${appName}`;
-        const userMessage = (0, userPrompt_1.buildUserPromptForCodeGenerationAgent)(params, primaryAction);
-        const result = await codeGenerationService_12.codegenAIProxyService.generateCodegenObject({
-          userMessage,
-          systemPrompt,
-          provider: constants_1.LLM_PROVIDERS.ANTHROPIC,
-          model: constants_1.LLM_MODELS.CLAUDE_SONNET_4_5_20250929,
-          schema: file_operations_1.FilesSchema,
-          agentName: this.name
-        });
-        return result?.object.files;
+      getPrimaryAction(params) {
+        const appName = params.blueprint?.appName ? `'${params.blueprint.appName}'` : "";
+        return `Customize the Wix CLI custom element / site widget scaffolding for the app ${appName}`;
       }
     };
     exports2.CustomElementAgent = CustomElementAgent;
@@ -73450,51 +73484,31 @@ var require_SPIAgent2 = __commonJS({
     var api_1 = require_api2();
     var prompt_selectors_1 = require_prompt_selectors();
     var codegen_common_logic_1 = require_dist9();
-    var userPrompt_1 = require_userPrompt();
-    var file_operations_1 = require_file_operations();
-    var constants_1 = require_constants();
-    var codeGenerationService_12 = require_codeGenerationService();
-    var SPIAgent = class {
-      constructor(useIteration = false) {
-        this.useIteration = useIteration;
+    var BaseCodeGenerationAgent_1 = require_BaseCodeGenerationAgent();
+    var SPIAgent = class extends BaseCodeGenerationAgent_1.BaseCodeGenerationAgent {
+      constructor() {
+        super(...arguments);
         this.name = "SPIAgent";
       }
       getAgentDocumentation() {
         return (0, codegen_common_logic_1.getSPIAgentDocumentation)();
       }
-      buildSystemPrompt({ apiNames, apiDocumentation, useData, spiNames }) {
-        const systemPrompt = (0, service_plugin_prompt_1.servicePluginPrompt)({
+      buildSystemPrompt(params) {
+        const { extension } = params;
+        const spiNames = extension.relatedSpis?.map((spi) => spi.name).filter((name) => !!name) || [];
+        const { apiDocumentation, apiNames } = (0, api_1.extractApiDocumentation)(params.extension);
+        const useData = (0, prompt_selectors_1.shouldUseDataPrompt)(params);
+        return (0, service_plugin_prompt_1.servicePluginPrompt)({
           apiNames,
           apiDocumentation,
           useData,
           useIteration: this.useIteration,
           spiNames
         });
-        return systemPrompt;
       }
-      async generate(params) {
-        const { extension, blueprint } = params;
-        const allSpiNames = extension.relatedSpis?.map((spi) => spi.name).filter((name) => !!name) || [];
-        const { apiDocumentation, apiNames } = (0, api_1.extractApiDocumentation)(params.extension);
-        const useData = (0, prompt_selectors_1.shouldUseDataPrompt)(params);
-        const systemPrompt = this.buildSystemPrompt({
-          apiNames,
-          apiDocumentation,
-          useData,
-          spiNames: allSpiNames
-        });
-        const appName = blueprint?.appName ? `'${blueprint.appName}'` : "";
-        const primaryAction = `Customize the Wix CLI service plugin scaffolding for the app ${appName}`;
-        const userMessage = (0, userPrompt_1.buildUserPromptForCodeGenerationAgent)(params, primaryAction);
-        const result = await codeGenerationService_12.codegenAIProxyService.generateCodegenObject({
-          userMessage,
-          systemPrompt,
-          provider: constants_1.LLM_PROVIDERS.ANTHROPIC,
-          model: constants_1.LLM_MODELS.CLAUDE_SONNET_4_5_20250929,
-          schema: file_operations_1.FilesSchema,
-          agentName: this.name
-        });
-        return result.object.files;
+      getPrimaryAction(params) {
+        const appName = params.blueprint?.appName ? `'${params.blueprint.appName}'` : "";
+        return `Customize the Wix CLI service plugin scaffolding for the app ${appName}`;
       }
     };
     exports2.SPIAgent = SPIAgent;
@@ -73573,19 +73587,18 @@ var require_BackendEventAgent2 = __commonJS({
     var api_1 = require_api2();
     var backend_event_prompt_1 = require_backend_event_prompt();
     var codegen_common_logic_1 = require_dist9();
-    var userPrompt_1 = require_userPrompt();
-    var file_operations_1 = require_file_operations();
-    var constants_1 = require_constants();
-    var codeGenerationService_12 = require_codeGenerationService();
-    var BackendEventAgent = class {
-      constructor(useIteration = false) {
-        this.useIteration = useIteration;
+    var BaseCodeGenerationAgent_1 = require_BaseCodeGenerationAgent();
+    var BackendEventAgent = class extends BaseCodeGenerationAgent_1.BaseCodeGenerationAgent {
+      constructor() {
+        super(...arguments);
         this.name = "BackendEventAgent";
       }
       getAgentDocumentation() {
         return (0, codegen_common_logic_1.getBackendEventAgentDocumentation)();
       }
-      buildSystemPrompt({ apiNames, apiDocumentation, useData }) {
+      buildSystemPrompt(params) {
+        const { apiDocumentation, apiNames } = (0, api_1.extractApiDocumentation)(params.extension);
+        const useData = (0, prompt_selectors_1.shouldUseDataPrompt)(params);
         return (0, backend_event_prompt_1.backendEventPrompt)({
           apiNames,
           useData,
@@ -73593,27 +73606,9 @@ var require_BackendEventAgent2 = __commonJS({
           apiDocumentation
         });
       }
-      async generate(params) {
-        const { blueprint } = params;
-        const { apiDocumentation, apiNames } = (0, api_1.extractApiDocumentation)(params.extension);
-        const useData = (0, prompt_selectors_1.shouldUseDataPrompt)(params);
-        const systemPrompt = this.buildSystemPrompt({
-          apiNames,
-          apiDocumentation,
-          useData
-        });
-        const appName = blueprint?.appName ? `'${blueprint.appName}'` : "";
-        const primaryAction = `Customize backend event scaffolding for the app ${appName}`;
-        const userMessage = (0, userPrompt_1.buildUserPromptForCodeGenerationAgent)(params, primaryAction);
-        const result = await codeGenerationService_12.codegenAIProxyService.generateCodegenObject({
-          userMessage,
-          systemPrompt,
-          provider: constants_1.LLM_PROVIDERS.ANTHROPIC,
-          model: constants_1.LLM_MODELS.CLAUDE_SONNET_4_5_20250929,
-          schema: file_operations_1.FilesSchema,
-          agentName: this.name
-        });
-        return result.object.files;
+      getPrimaryAction(params) {
+        const appName = params.blueprint?.appName ? `'${params.blueprint.appName}'` : "";
+        return `Customize backend event scaffolding for the app ${appName}`;
       }
     };
     exports2.BackendEventAgent = BackendEventAgent;
@@ -73738,21 +73733,20 @@ var require_BackendApiAgent2 = __commonJS({
     exports2.BackendApiAgent = void 0;
     var backend_api_prompt_1 = require_backend_api_prompt();
     var codegen_common_logic_1 = require_dist9();
-    var userPrompt_1 = require_userPrompt();
-    var file_operations_1 = require_file_operations();
-    var constants_1 = require_constants();
     var api_1 = require_api2();
     var prompt_selectors_1 = require_prompt_selectors();
-    var codeGenerationService_12 = require_codeGenerationService();
-    var BackendApiAgent = class {
-      constructor(useIteration = false) {
-        this.useIteration = useIteration;
+    var BaseCodeGenerationAgent_1 = require_BaseCodeGenerationAgent();
+    var BackendApiAgent = class extends BaseCodeGenerationAgent_1.BaseCodeGenerationAgent {
+      constructor() {
+        super(...arguments);
         this.name = "BackendApiAgent";
       }
       getAgentDocumentation() {
         return (0, codegen_common_logic_1.getBackendApiAgentDocumentation)();
       }
-      buildSystemPrompt({ apiNames, apiDocumentation, useData }) {
+      buildSystemPrompt(params) {
+        const { apiDocumentation, apiNames } = (0, api_1.extractApiDocumentation)(params.extension);
+        const useData = (0, prompt_selectors_1.shouldUseDataPrompt)(params);
         return (0, backend_api_prompt_1.backendApiPrompt)({
           apiNames,
           useData,
@@ -73760,26 +73754,8 @@ var require_BackendApiAgent2 = __commonJS({
           apiDocumentation
         });
       }
-      async generate(params) {
-        const { blueprint } = params;
-        const { apiDocumentation, apiNames } = (0, api_1.extractApiDocumentation)(params.extension);
-        const useData = (0, prompt_selectors_1.shouldUseDataPrompt)(params);
-        const systemPrompt = this.buildSystemPrompt({
-          apiNames,
-          apiDocumentation,
-          useData
-        });
-        const primaryAction = `Create Astro Server Endpoints for the app ${blueprint?.appName}`;
-        const userMessage = (0, userPrompt_1.buildUserPromptForCodeGenerationAgent)(params, primaryAction);
-        const result = await codeGenerationService_12.codegenAIProxyService.generateCodegenObject({
-          userMessage,
-          systemPrompt,
-          provider: constants_1.LLM_PROVIDERS.ANTHROPIC,
-          model: constants_1.LLM_MODELS.CLAUDE_SONNET_4_5_20250929,
-          schema: file_operations_1.FilesSchema,
-          agentName: this.name
-        });
-        return result.object.files;
+      getPrimaryAction(params) {
+        return `Create Astro Server Endpoints for the app ${params.blueprint?.appName}`;
       }
     };
     exports2.BackendApiAgent = BackendApiAgent;
@@ -74028,10 +74004,8 @@ var require_EmbeddedScriptAgent2 = __commonJS({
     var zod_1 = __importDefault2(require_zod());
     var embedded_script_prompt_1 = require_embedded_script_prompt();
     var codegen_common_logic_1 = require_dist9();
-    var userPrompt_1 = require_userPrompt();
     var file_operations_1 = require_file_operations();
-    var constants_1 = require_constants();
-    var codeGenerationService_12 = require_codeGenerationService();
+    var BaseCodeGenerationAgent_1 = require_BaseCodeGenerationAgent();
     var EmbeddedScriptPlacement;
     (function(EmbeddedScriptPlacement2) {
       EmbeddedScriptPlacement2["HEAD"] = "HEAD";
@@ -74050,15 +74024,18 @@ var require_EmbeddedScriptAgent2 = __commonJS({
       scriptType: zod_1.default.enum(Object.values(EmbeddedScriptType)),
       placement: zod_1.default.enum(Object.values(EmbeddedScriptPlacement))
     });
-    var EmbeddedScriptAgent = class {
-      constructor(useIteration = false) {
-        this.useIteration = useIteration;
+    var EmbeddedScriptAgent = class extends BaseCodeGenerationAgent_1.BaseCodeGenerationAgent {
+      constructor() {
+        super(...arguments);
         this.name = "EmbeddedScriptAgent";
       }
       getAgentDocumentation() {
         return (0, codegen_common_logic_1.getEmbeddedScriptAgentDocumentation)();
       }
-      buildSystemPrompt({ hasDynamicParameters, apiNames, apiDocumentation, useData }) {
+      buildSystemPrompt(params) {
+        const hasDynamicParameters = (0, prompt_selectors_1.shouldUseDynamicParametersPrompt)(params);
+        const { apiDocumentation, apiNames } = (0, api_1.extractApiDocumentation)(params.extension);
+        const useData = (0, prompt_selectors_1.shouldUseDataPrompt)(params);
         return (0, embedded_script_prompt_1.embeddedScriptPrompt)({
           hasDynamicParameters,
           apiNames,
@@ -74067,32 +74044,21 @@ var require_EmbeddedScriptAgent2 = __commonJS({
           apiDocumentation
         });
       }
-      async generate(params) {
-        const { blueprint } = params;
-        const hasDynamicParameters = (0, prompt_selectors_1.shouldUseDynamicParametersPrompt)(params);
-        const { apiDocumentation, apiNames } = (0, api_1.extractApiDocumentation)(params.extension);
-        const useData = (0, prompt_selectors_1.shouldUseDataPrompt)(params);
-        const systemPrompt = this.buildSystemPrompt({
-          hasDynamicParameters,
-          apiNames,
-          apiDocumentation,
-          useData
-        });
-        const appName = blueprint?.appName ? `'${blueprint.appName}'` : "";
-        const primaryAction = `Customize the Wix CLI embedded script scaffolding for the app ${appName}`;
-        const userMessage = (0, userPrompt_1.buildUserPromptForCodeGenerationAgent)(params, primaryAction);
-        const result = await codeGenerationService_12.codegenAIProxyService.generateCodegenObject({
-          userMessage,
-          systemPrompt,
-          provider: constants_1.LLM_PROVIDERS.ANTHROPIC,
-          model: constants_1.LLM_MODELS.CLAUDE_SONNET_4_5_20250929,
-          schema: EmbeddedScriptSchema,
-          agentName: this.name
-        });
+      getPrimaryAction(params) {
+        const appName = params.blueprint?.appName ? `'${params.blueprint.appName}'` : "";
+        return `Customize the Wix CLI embedded script scaffolding for the app ${appName}`;
+      }
+      getGenerateOptions() {
         return {
-          files: result.object.files,
-          scriptType: result.object.scriptType,
-          placement: result.object.placement
+          schema: EmbeddedScriptSchema,
+          extractResult: (obj) => {
+            const result = obj;
+            return {
+              files: result.files,
+              scriptType: result.scriptType,
+              placement: result.placement
+            };
+          }
         };
       }
     };
@@ -82208,9 +82174,9 @@ var require_IterationAgent = __commonJS({
     "use strict";
     Object.defineProperty(exports2, "__esModule", { value: true });
     exports2.IterationAgent = void 0;
-    var ai_1 = require_dist8();
     var iteration_agent_prompt_1 = require_iteration_agent_prompt();
     var AgentsRegistry_1 = require_AgentsRegistry();
+    var codegen_common_logic_1 = require_dist9();
     var ditto_codegen_types_12 = require_dist();
     var constants_1 = require_constants();
     var contextBuilders_1 = require_contextBuilders();
@@ -82224,13 +82190,6 @@ var require_IterationAgent = __commonJS({
       buildSystemPrompt() {
         return (0, iteration_agent_prompt_1.iterationAgentPrompt)();
       }
-      extractPlanResult(toolCalls) {
-        const submitCall = toolCalls.find((call) => call.toolName === SUBMIT_PLAN_TOOL_NAME);
-        if (!submitCall) {
-          throw new Error("Failed to generate iteration plan");
-        }
-        return submitCall.input;
-      }
       mapExtensionType(extensionType) {
         const supportedExtensionTypes = (0, AgentsRegistry_1.getSupportedExtensionTypes)();
         if (supportedExtensionTypes.includes(extensionType)) {
@@ -82268,7 +82227,8 @@ var require_IterationAgent = __commonJS({
           const model = (0, customAnthropicProvider_1.createCustomTextModel)()(constants_1.LLM_MODELS.CLAUDE_SONNET_4_5_20250929, {
             agentName: this.name
           });
-          const result = await (0, ai_1.generateText)({
+          const result = await (0, codegen_common_logic_1.generateAgentText)({
+            agentName: this.name,
             model,
             tools: {
               [tools_1.TOOL_NAMES.READ_FILE]: (0, tools_1.createReadFileTool)(outputPath),
@@ -82276,23 +82236,14 @@ var require_IterationAgent = __commonJS({
               [tools_1.TOOL_NAMES.GLOB]: (0, tools_1.createGlobTool)(outputPath),
               [SUBMIT_PLAN_TOOL_NAME]: (0, tools_1.createSubmitPlanTool)()
             },
-            stopWhen: ({ steps }) => {
-              const lastStep = steps[steps.length - 1];
-              return lastStep?.toolCalls.some((toolCall) => toolCall.toolName === SUBMIT_PLAN_TOOL_NAME && !("invalid" in toolCall) && !("error" in toolCall)) ?? false;
-            },
-            messages: [
-              {
-                role: "system",
-                content: this.buildSystemPrompt(),
-                providerOptions: (0, ditto_codegen_types_12.withCaching)("1h")
-              },
-              { role: "user", content: userMessage }
-            ],
-            maxRetries: 3,
-            temperature: 0,
-            experimental_telemetry: { isEnabled: true, functionId: this.name }
+            systemPrompt: this.buildSystemPrompt(),
+            userMessage,
+            stopWhen: (0, codegen_common_logic_1.stopOnToolCall)(SUBMIT_PLAN_TOOL_NAME)
           });
-          const planResult = this.extractPlanResult(result.toolCalls);
+          const planResult = (0, codegen_common_logic_1.extractToolResult)(result.toolCalls, SUBMIT_PLAN_TOOL_NAME);
+          if (!planResult) {
+            throw new Error("Failed to generate iteration plan");
+          }
           console.log("IterationAgent planResult:", JSON.stringify(planResult, null, 2));
           const resultObject = {
             currentExtensions: planResult.currentExtensions.map((ext) => ({
@@ -97697,7 +97648,7 @@ var require_SDKPickerAgent = __commonJS({
     Object.defineProperty(exports2, "__esModule", { value: true });
     exports2.SDKPickerAgent = void 0;
     var ditto_codegen_types_12 = require_dist();
-    var ai_1 = require_dist8();
+    var codegen_common_logic_1 = require_dist9();
     var WixMCPClient_1 = require_WixMCPClient();
     var tools_1 = require_tools3();
     var sdkEnricher_1 = require_sdkEnricher();
@@ -97708,21 +97659,6 @@ var require_SDKPickerAgent = __commonJS({
         this.mcpClient = new WixMCPClient_1.WixMCPClient();
         this.model = model;
       }
-      createStopCondition() {
-        return ({ steps }) => {
-          const lastStep = steps[steps.length - 1];
-          const hasSuccessfulSubmit = lastStep?.toolCalls.some((toolCall) => toolCall.toolName === tools_1.SDK_PICKER_TOOL_NAMES.SUBMIT_PLAN && !("invalid" in toolCall) && !("error" in toolCall)) ?? false;
-          return hasSuccessfulSubmit;
-        };
-      }
-      extractSDKPickerResult({ toolCalls }) {
-        const submitCall = toolCalls.find((call) => call.toolName === tools_1.SDK_PICKER_TOOL_NAMES.SUBMIT_PLAN);
-        if (!submitCall) {
-          console.error("No submit_plan tool call found");
-          return null;
-        }
-        return submitCall.input;
-      }
       async generate({ validIds, extensionDocumentation, userPrompt }) {
         if (validIds.length === 0) {
           return { extensions: [] };
@@ -97733,26 +97669,15 @@ var require_SDKPickerAgent = __commonJS({
             mcpClient: this.mcpClient,
             validIds
           });
-          const systemPrompt = (0, sdk_picker_agent_prompt_1.sdkPickerAgentPrompt)(extensionDocumentation);
-          const result = await (0, ai_1.generateText)({
+          const result = await (0, codegen_common_logic_1.generateAgentText)({
+            agentName: this.name,
             model: this.model,
             tools,
-            stopWhen: this.createStopCondition(),
-            messages: [
-              {
-                role: "system",
-                content: systemPrompt,
-                providerOptions: (0, ditto_codegen_types_12.withCaching)("1h")
-              },
-              { role: "user", content: userPrompt }
-            ],
-            maxRetries: 3,
-            temperature: 0,
-            experimental_telemetry: { isEnabled: true, functionId: this.name }
-          });
-          const apiResult = this.extractSDKPickerResult({
-            toolCalls: result.toolCalls
+            systemPrompt: (0, sdk_picker_agent_prompt_1.sdkPickerAgentPrompt)(extensionDocumentation),
+            userMessage: userPrompt,
+            stopWhen: (0, codegen_common_logic_1.stopOnToolCall)(tools_1.SDK_PICKER_TOOL_NAMES.SUBMIT_PLAN)
           });
+          const apiResult = (0, codegen_common_logic_1.extractToolResult)(result.toolCalls, tools_1.SDK_PICKER_TOOL_NAMES.SUBMIT_PLAN);
           if (!apiResult) {
             throw new Error("Failed to extract SDK picker result from tool calls");
           }
@@ -98133,8 +98058,8 @@ var require_ExtensionIndexerAgent = __commonJS({
     };
     Object.defineProperty(exports2, "__esModule", { value: true });
     exports2.ExtensionIndexerAgent = void 0;
-    var ai_1 = require_dist8();
     var ditto_codegen_types_12 = require_dist();
+    var codegen_common_logic_1 = require_dist9();
     var constants_1 = require_constants();
     var customAnthropicProvider_1 = require_customAnthropicProvider();
     var tools_1 = require_tools();
@@ -98175,7 +98100,8 @@ ${srcFolderStructure}
           const model = (0, customAnthropicProvider_1.createCustomTextModel)()(constants_1.LLM_MODELS.CLAUDE_HAIKU_4_5, {
             agentName: this.name
           });
-          await (0, ai_1.generateText)({
+          await (0, codegen_common_logic_1.generateAgentText)({
+            agentName: this.name,
             model,
             tools: {
               [tools_1.TOOL_NAMES.READ_FILE]: (0, tools_1.createReadFileTool)(outputPath),
@@ -98184,24 +98110,9 @@ ${srcFolderStructure}
               [tools_1.TOOL_NAMES.GREP]: (0, tools_1.createGrepTool)(outputPath),
               [tools_1.TOOL_NAMES.GLOB]: (0, tools_1.createGlobTool)(outputPath)
             },
-            stopWhen: ({ steps }) => {
-              if (steps.length >= MAX_STEPS) {
-                return true;
-              }
-              const lastStep = steps[steps.length - 1];
-              return lastStep?.toolCalls.length === 0;
-            },
-            messages: [
-              {
-                role: "system",
-                content: systemPrompt,
-                providerOptions: (0, ditto_codegen_types_12.withCaching)("1h")
-              },
-              { role: "user", content: userMessage }
-            ],
-            maxRetries: 3,
-            temperature: 0,
-            experimental_telemetry: { isEnabled: true, functionId: this.name }
+            systemPrompt,
+            userMessage,
+            stopWhen: (0, codegen_common_logic_1.stopOnNoToolCalls)(MAX_STEPS)
           });
         } catch (error) {
           throw (0, ditto_codegen_types_12.toCodegenError)(error);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@wix/ditto-codegen-public",
-  "version": "1.0.204",
+  "version": "1.0.205",
   "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": "b38838efbcff60c6aa56eabbec8126a86f885bfd2bbc736a6e526ec2"
+  "falconPackageHash": "30c73c66aeeddb26c76b80c1f980e0ffe8c2698f2958772df94da841"
 }