genaicode 0.6.1 → 0.8.0

package/dist/prompt/steps/step-codegen-planning.js

@@ -0,0 +1,69 @@
+import { getFunctionDefs } from '../function-calling.js';
+import { putSystemMessage } from '../../main/common/content-bus.js';
+import { validateAndRecoverSingleResult } from './step-validate-recover.js';
+import { StepResult } from './steps-types.js';
+import { CODEGEN_SUMMARY_PROMPT } from '../static-prompts.js';
+export const PLANNING_PROMPT = `Please analyze the conversation so far and help plan the implementation:
+
+1. Review the conversation history and requirements
+2. Thoughtfully analyze the problem
+3. Create a step-by-step implementation plan
+4. List all files that will be affected during implementation
+
+Remember to:
+- Consider both direct and indirect dependencies
+- Think about how changes in one file might affect others
+- Include all necessary file paths
+- Explain your reasoning for each step
+- Consider edge cases and potential issues
+
+Output the plan using \`codegenPlanning\` function provding all the required fields.`;
+/**
+ * Execute the codegen planning step
+ * This step analyzes the conversation and produces a detailed implementation plan
+ * before proceeding with actual code generation
+ */
+export async function executeStepCodegenPlanning(generateContentFn, prompt, options) {
+    putSystemMessage('Starting codegen planning phase...');
+    // Add the planning prompt to the conversation
+    if (prompt.slice(-1)[0].type === 'user') {
+        prompt.slice(-1)[0].text += '\n\\n' + PLANNING_PROMPT;
+    }
+    else {
+        prompt.push({ type: 'user', text: PLANNING_PROMPT });
+    }
+    const planningRequest = [
+        prompt,
+        getFunctionDefs(),
+        'codegenPlanning',
+        options.temperature ?? 0.7,
+        options.cheap ?? true,
+        options,
+    ];
+    try {
+        let planningResult = await generateContentFn(...planningRequest);
+        planningResult = await validateAndRecoverSingleResult(planningRequest, planningResult, generateContentFn);
+        const codegenPlanningRequest = planningResult.find((call) => call.name === 'codegenPlanning');
+        if (codegenPlanningRequest) {
+            putSystemMessage('Planning phase completed successfully', codegenPlanningRequest);
+            // Store the planning response in conversation history
+            prompt.push({ type: 'assistant', functionCalls: planningResult });
+            prompt.push({
+                type: 'user',
+                functionResponses: planningResult.map((call) => ({ name: call.name, call_id: call.id })),
+                text: CODEGEN_SUMMARY_PROMPT,
+                cache: true,
+            });
+            return StepResult.CONTINUE;
+        }
+        else {
+            putSystemMessage('Did not receive codegen planning result');
+            return StepResult.BREAK;
+        }
+    }
+    catch (error) {
+        putSystemMessage('Error during planning phase: ' + error.message);
+        return StepResult.BREAK;
+    }
+}
+//# sourceMappingURL=step-codegen-planning.js.map

package/dist/prompt/steps/step-codegen-planning.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"step-codegen-planning.js","sourceRoot":"","sources":["../../../src/prompt/steps/step-codegen-planning.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,eAAe,EAAE,MAAM,wBAAwB,CAAC;AACzD,OAAO,EAAE,gBAAgB,EAAE,MAAM,kCAAkC,CAAC;AACpE,OAAO,EAAE,8BAA8B,EAAE,MAAM,4BAA4B,CAAC;AAC5E,OAAO,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAC9C,OAAO,EAAE,sBAAsB,EAAE,MAAM,sBAAsB,CAAC;AAE9D,MAAM,CAAC,MAAM,eAAe,GAAG;;;;;;;;;;;;;;qFAcsD,CAAC;AAEtF;;;;GAIG;AACH,MAAM,CAAC,KAAK,UAAU,0BAA0B,CAC9C,iBAA0C,EAC1C,MAAoB,EACpB,OAAuB;IAEvB,gBAAgB,CAAC,oCAAoC,CAAC,CAAC;IAEvD,8CAA8C;IAC9C,IAAI,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;QACxC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,IAAI,OAAO,GAAG,eAAe,CAAC;IACxD,CAAC;SAAM,CAAC;QACN,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,eAAe,EAAE,CAAC,CAAC;IACvD,CAAC;IAED,MAAM,eAAe,GAA2E;QAC9F,MAAM;QACN,eAAe,EAAE;QACjB,iBAAiB;QACjB,OAAO,CAAC,WAAW,IAAI,GAAG;QAC1B,OAAO,CAAC,KAAK,IAAI,IAAI;QACrB,OAAO;KACR,CAAC;IAEF,IAAI,CAAC;QACH,IAAI,cAAc,GAAG,MAAM,iBAAiB,CAAC,GAAG,eAAe,CAAC,CAAC;QACjE,cAAc,GAAG,MAAM,8BAA8B,CAAC,eAAe,EAAE,cAAc,EAAE,iBAAiB,CAAC,CAAC;QAE1G,MAAM,sBAAsB,GAAG,cAAc,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,iBAAiB,CAAC,CAAC;QAE9F,IAAI,sBAAsB,EAAE,CAAC;YAC3B,gBAAgB,CAAC,uCAAuC,EAAE,sBAAsB,CAAC,CAAC;YAElF,sDAAsD;YACtD,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,aAAa,EAAE,cAAc,EAAE,CAAC,CAAC;YAClE,MAAM,CAAC,IAAI,CAAC;gBACV,IAAI,EAAE,MAAM;gBACZ,iBAAiB,EAAE,cAAc,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,OAAO,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,CAAC;gBACxF,IAAI,EAAE,sBAAsB;gBAC5B,KAAK,EAAE,IAAI;aACZ,CAAC,CAAC;YAEH,OAAO,UAAU,CAAC,QAAQ,CAAC;QAC7B,CAAC;aAAM,CAAC;YACN,gBAAgB,CAAC,yCAAyC,CAAC,CAAC;YAC5D,OAAO,UAAU,CAAC,KAAK,CAAC;QAC1B,CAAC;IACH,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,gBAAgB,CAAC,+BAA+B,GAAI,KAAe,CAAC,OAAO,CAAC,CAAC;QAC7E,OAAO,UAAU,CAAC,KAAK,CAAC;IAC1B,CAAC;AACH,CAAC"}

package/dist/prompt/steps/step-codegen-summary.d.ts

@@ -0,0 +1,7 @@
+import { FunctionCall, FunctionDef, GenerateContentFunction, GenerateImageFunction, PromptItem } from '../../ai-service/common.js';
+import { CodegenOptions } from '../../main/codegen-types.js';
+/**
+ * Executes the codegen summary step.
+ * This step generates a summary of the planned code changes and processes each file update.
+ */
+export declare function executeStepCodegenSummary(generateContentFn: GenerateContentFunction, prompt: PromptItem[], functionDefs: FunctionDef[], options: CodegenOptions, waitIfPaused?: () => Promise<void>, generateImageFn?: GenerateImageFunction): Promise<FunctionCall[]>;

package/dist/prompt/steps/step-codegen-summary.js

@@ -0,0 +1,106 @@
+import assert from 'node:assert';
+import fs from 'fs';
+import mime from 'mime-types';
+import { validateAndRecoverSingleResult } from './step-validate-recover.js';
+import { putSystemMessage } from '../../main/common/content-bus.js';
+import { executeStepGenerateImage } from './step-generate-image.js';
+import { executeStepVerifyPatch } from './step-verify-patch.js';
+import { getPartialPromptTemplate } from '../static-prompts.js';
+import { executeStepEnsureContext } from './step-ensure-context.js';
+import { StepResult } from './steps-types.js';
+/**
+ * Executes the codegen summary step.
+ * This step generates a summary of the planned code changes and processes each file update.
+ */
+export async function executeStepCodegenSummary(generateContentFn, prompt, functionDefs, options, waitIfPaused = () => Promise.resolve(), generateImageFn) {
+    const baseRequest = [
+        prompt,
+        functionDefs,
+        'codegenSummary',
+        options.temperature ?? 0.7,
+        options.cheap ?? false,
+        options,
+    ];
+    let baseResult = await generateContentFn(...baseRequest);
+    let codegenSummaryRequest = baseResult.find((call) => call.name === 'codegenSummary');
+    if (codegenSummaryRequest) {
+        // Second stage: for each file request the actual code updates
+        baseResult = await validateAndRecoverSingleResult(baseRequest, baseResult, generateContentFn);
+        codegenSummaryRequest = baseResult.find((call) => call.name === 'codegenSummary');
+        putSystemMessage('Received codegen summary, will collect partial updates', codegenSummaryRequest);
+        // Sometimes the result happens to be a string
+        assert(Array.isArray(codegenSummaryRequest?.args?.fileUpdates), 'fileUpdates is not an array');
+        assert(Array.isArray(codegenSummaryRequest?.args.contextPaths), 'contextPaths is not an array');
+        // Ensure all necessary files are in context
+        const contextResult = await executeStepEnsureContext(prompt, codegenSummaryRequest, options);
+        if (contextResult === StepResult.BREAK) {
+            return baseResult;
+        }
+        // Store the first stage response entirely in conversation history
+        prompt.push({ type: 'assistant', functionCalls: baseResult });
+        prompt.push({
+            type: 'user',
+            functionResponses: baseResult.map((call) => ({ name: call.name, call_id: call.id })),
+            cache: true,
+        });
+        const result = [];
+        for (const file of codegenSummaryRequest.args.fileUpdates) {
+            putSystemMessage('Collecting partial update for: ' + file.filePath + ' using tool: ' + file.updateToolName, file);
+            // Check if execution is paused before proceeding
+            await waitIfPaused();
+            // this is needed, otherwise we will get an error
+            if (prompt.slice(-1)[0].type === 'user') {
+                prompt.slice(-1)[0].text = file.prompt ?? getPartialPromptTemplate(file.filePath);
+            }
+            else {
+                prompt.push({ type: 'user', text: file.prompt ?? getPartialPromptTemplate(file.filePath) });
+            }
+            if (options.vision && file.contextImageAssets) {
+                prompt.slice(-1)[0].images = file.contextImageAssets.map((path) => ({
+                    path,
+                    base64url: fs.readFileSync(path, 'base64'),
+                    mediaType: mime.lookup(path) || '',
+                }));
+            }
+            const partialRequest = [
+                prompt,
+                functionDefs,
+                file.updateToolName,
+                file.temperature ?? options.temperature,
+                file.cheap === true,
+                options,
+            ];
+            let partialResult = await generateContentFn(...partialRequest);
+            // Validate if function call is compliant with the schema
+            partialResult = await validateAndRecoverSingleResult(partialRequest, partialResult, generateContentFn);
+            putSystemMessage('Received partial update', partialResult);
+            // Handle image generation requests
+            const generateImageCall = partialResult.find((call) => call.name === 'generateImage');
+            if (generateImageCall && generateImageFn) {
+                partialResult.push(await executeStepGenerateImage(generateImageFn, generateImageCall));
+            }
+            else if (generateImageCall) {
+                console.warn('Image generation requested but generateImageFn not provided');
+            }
+            // Verify if patchFile is one of the functions called, and test if patch is valid and can be applied successfully
+            const patchFileCall = partialResult.find((call) => call.name === 'patchFile');
+            if (patchFileCall) {
+                partialResult = await executeStepVerifyPatch(patchFileCall.args, generateContentFn, prompt, functionDefs, file.temperature ?? options.temperature, file.cheap === true, options);
+            }
+            // add the code gen result to the context, as the subsequent code gen may depend on the result
+            prompt.push({ type: 'assistant', functionCalls: partialResult }, {
+                type: 'user',
+                text: 'Update applied.',
+                functionResponses: partialResult.map((call) => ({ name: call.name, call_id: call.id })),
+            });
+            result.push(...partialResult);
+        }
+        return result;
+    }
+    else {
+        // This is unexpected, if happens probably means no code updates.
+        putSystemMessage('Did not receive codegen summary, returning result.');
+        return baseResult;
+    }
+}
+//# sourceMappingURL=step-codegen-summary.js.map

package/dist/prompt/steps/step-codegen-summary.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"step-codegen-summary.js","sourceRoot":"","sources":["../../../src/prompt/steps/step-codegen-summary.ts"],"names":[],"mappings":"AAAA,OAAO,MAAM,MAAM,aAAa,CAAC;AACjC,OAAO,EAAE,MAAM,IAAI,CAAC;AACpB,OAAO,IAAI,MAAM,YAAY,CAAC;AAU9B,OAAO,EAAE,8BAA8B,EAAE,MAAM,4BAA4B,CAAC;AAC5E,OAAO,EAAE,gBAAgB,EAAE,MAAM,kCAAkC,CAAC;AACpE,OAAO,EAAE,wBAAwB,EAAE,MAAM,0BAA0B,CAAC;AACpE,OAAO,EAAE,sBAAsB,EAAE,MAAM,wBAAwB,CAAC;AAChE,OAAO,EAAE,wBAAwB,EAAE,MAAM,sBAAsB,CAAC;AAChE,OAAO,EAAE,wBAAwB,EAAE,MAAM,0BAA0B,CAAC;AACpE,OAAO,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAE9C;;;GAGG;AACH,MAAM,CAAC,KAAK,UAAU,yBAAyB,CAC7C,iBAA0C,EAC1C,MAAoB,EACpB,YAA2B,EAC3B,OAAuB,EACvB,eAAoC,GAAG,EAAE,CAAC,OAAO,CAAC,OAAO,EAAE,EAC3D,eAAuC;IAEvC,MAAM,WAAW,GAA2E;QAC1F,MAAM;QACN,YAAY;QACZ,gBAAgB;QAChB,OAAO,CAAC,WAAW,IAAI,GAAG;QAC1B,OAAO,CAAC,KAAK,IAAI,KAAK;QACtB,OAAO;KACR,CAAC;IACF,IAAI,UAAU,GAAG,MAAM,iBAAiB,CAAC,GAAG,WAAW,CAAC,CAAC;IAEzD,IAAI,qBAAqB,GAAG,UAAU,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,gBAAgB,CAAC,CAAC;IAEtF,IAAI,qBAAqB,EAAE,CAAC;QAC1B,8DAA8D;QAC9D,UAAU,GAAG,MAAM,8BAA8B,CAAC,WAAW,EAAE,UAAU,EAAE,iBAAiB,CAAC,CAAC;QAC9F,qBAAqB,GAAG,UAAU,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,gBAAgB,CAAC,CAAC;QAElF,gBAAgB,CAAC,wDAAwD,EAAE,qBAAqB,CAAC,CAAC;QAElG,8CAA8C;QAC9C,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,qBAAqB,EAAE,IAAI,EAAE,WAAW,CAAC,EAAE,6BAA6B,CAAC,CAAC;QAC/F,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,qBAAqB,EAAE,IAAI,CAAC,YAAY,CAAC,EAAE,8BAA8B,CAAC,CAAC;QAEhG,4CAA4C;QAC5C,MAAM,aAAa,GAAG,MAAM,wBAAwB,CAAC,MAAM,EAAE,qBAAqB,EAAE,OAAO,CAAC,CAAC;QAC7F,IAAI,aAAa,KAAK,UAAU,CAAC,KAAK,EAAE,CAAC;YACvC,OAAO,UAAU,CAAC;QACpB,CAAC;QAED,kEAAkE;QAClE,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,aAAa,EAAE,UAAU,EAAE,CAAC,CAAC;QAC9D,MAAM,CAAC,IAAI,CAAC;YACV,IAAI,EAAE,MAAM;YACZ,iBAAiB,EAAE,UAAU,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,OAAO,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,CAAC;YACpF,KAAK,EAAE,IAAI;SACZ,CAAC,CAAC;QAEH,MAAM,MAAM,GAAmB,EAAE,CAAC;QAElC,KAAK,MAAM,IAAI,IAAI,qBAAsB,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;YAC3D,gBAAgB,CAAC,iCAAiC,GAAG,IAAI,CAAC,QAAQ,GAAG,eAAe,GAAG,IAAI,CAAC,cAAc,EAAE,IAAI,CAAC,CAAC;YAElH,iDAAiD;YACjD,MAAM,YAAY,EAAE,CAAC;YAErB,iDAAiD;YACjD,IAAI,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;gBACxC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,GAAG,IAAI,CAAC,MAAM,IAAI,wBAAwB,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YACpF,CAAC;iBAAM,CAAC;gBACN,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,MAAM,IAAI,wBAAwB,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;YAC9F,CAAC;YAED,IAAI,OAAO,CAAC,MAAM,IAAI,IAAI,CAAC,kBAAkB,EAAE,CAAC;gBAC9C,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,MAAM,GAAG,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,CAAC,IAAY,EAAE,EAAE,CAAC,CAAC;oBAC1E,IAAI;oBACJ,SAAS,EAAE,EAAE,CAAC,YAAY,CAAC,IAAI,EAAE,QAAQ,CAAC;oBAC1C,SAAS,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,EAAE;iBACnC,CAAC,CAAC,CAAC;YACN,CAAC;YAED,MAAM,cAAc,GAA2E;gBAC7F,MAAM;gBACN,YAAY;gBACZ,IAAI,CAAC,cAAc;gBACnB,IAAI,CAAC,WAAW,IAAI,OAAO,CAAC,WAAW;gBACvC,IAAI,CAAC,KAAK,KAAK,IAAI;gBACnB,OAAO;aACR,CAAC;YACF,IAAI,aAAa,GAAG,MAAM,iBAAiB,CAAC,GAAG,cAAc,CAAC,CAAC;YAE/D,yDAAyD;YACzD,aAAa,GAAG,MAAM,8BAA8B,CAAC,cAAc,EAAE,aAAa,EAAE,iBAAiB,CAAC,CAAC;YAEvG,gBAAgB,CAAC,yBAAyB,EAAE,aAAa,CAAC,CAAC;YAE3D,mCAAmC;YACnC,MAAM,iBAAiB,GAAG,aAAa,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,eAAe,CAAC,CAAC;YACtF,IAAI,iBAAiB,IAAI,eAAe,EAAE,CAAC;gBACzC,aAAa,CAAC,IAAI,CAAC,MAAM,wBAAwB,CAAC,eAAe,EAAE,iBAAiB,CAAC,CAAC,CAAC;YACzF,CAAC;iBAAM,IAAI,iBAAiB,EAAE,CAAC;gBAC7B,OAAO,CAAC,IAAI,CAAC,6DAA6D,CAAC,CAAC;YAC9E,CAAC;YAED,iHAAiH;YACjH,MAAM,aAAa,GAAG,aAAa,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,WAAW,CAAC,CAAC;YAC9E,IAAI,aAAa,EAAE,CAAC;gBAClB,aAAa,GAAG,MAAM,sBAAsB,CAC1C,aAAa,CAAC,IAA2C,EACzD,iBAAiB,EACjB,MAAM,EACN,YAAY,EACZ,IAAI,CAAC,WAAW,IAAI,OAAO,CAAC,WAAW,EACvC,IAAI,CAAC,KAAK,KAAK,IAAI,EACnB,OAAO,CACR,CAAC;YACJ,CAAC;YAED,8FAA8F;YAC9F,MAAM,CAAC,IAAI,CACT,EAAE,IAAI,EAAE,WAAW,EAAE,aAAa,EAAE,aAAa,EAAE,EACnD;gBACE,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE,iBAAiB;gBACvB,iBAAiB,EAAE,aAAa,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,OAAO,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,CAAC;aACxF,CACF,CAAC;YAEF,MAAM,CAAC,IAAI,CAAC,GAAG,aAAa,CAAC,CAAC;QAChC,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;SAAM,CAAC;QACN,iEAAiE;QACjE,gBAAgB,CAAC,oDAAoD,CAAC,CAAC;QACvE,OAAO,UAAU,CAAC;IACpB,CAAC;AACH,CAAC"}

package/dist/prompt/steps/step-codegen-summary.test.d.ts

@@ -0,0 +1 @@
+export {};

package/dist/prompt/steps/step-codegen-summary.test.js

@@ -0,0 +1,136 @@
+import { describe, it, expect, vi, beforeEach } from 'vitest';
+import { executeStepCodegenSummary } from './step-codegen-summary.js';
+import { getSourceCode } from '../../files/read-files.js';
+vi.mock('../../files/find-files.js', () => ({
+    refreshFiles: vi.fn(),
+}));
+vi.mock('../../files/read-files.js', () => ({
+    getSourceCode: vi.fn(),
+}));
+vi.mock('../../main/config.js', () => ({
+    rootDir: '/test',
+    rcConfig: {
+        rootDir: '/test',
+        extensions: ['.js', '.ts', '.tsx', '.jsx'],
+    },
+    importantContext: {},
+}));
+describe('executeStepCodegenSummary', () => {
+    const mockGenerateContentFn = vi.fn();
+    const mockWaitIfPaused = vi.fn();
+    // Mock data
+    const mockPrompt = [{ type: 'user', text: 'test prompt' }];
+    const mockFunctionDefs = [
+        {
+            name: 'codegenSummary',
+            description: 'Function for generating codegen summary',
+            parameters: {
+                type: 'object',
+                properties: {
+                    explanation: { type: 'string', description: 'Explanation of changes' },
+                    fileUpdates: { type: 'array', description: 'List of file updates' },
+                    contextPaths: { type: 'array', description: 'List of context paths' },
+                },
+                required: ['explanation', 'fileUpdates', 'contextPaths'],
+            },
+        },
+    ];
+    const mockOptions = {
+        aiService: 'vertex-ai',
+        temperature: 0.7,
+        askQuestion: false,
+    };
+    beforeEach(() => {
+        vi.clearAllMocks();
+        mockWaitIfPaused.mockResolvedValue(undefined);
+        vi.mocked(getSourceCode).mockReturnValue({});
+    });
+    it('should handle successful codegen summary generation', async () => {
+        const mockCodegenSummary = {
+            name: 'codegenSummary',
+            args: {
+                explanation: 'Test explanation',
+                fileUpdates: [
+                    {
+                        filePath: '/test/path.ts',
+                        updateToolName: 'updateFile',
+                        prompt: 'Update test file',
+                    },
+                ],
+                contextPaths: ['/test/context.ts'],
+            },
+        };
+        const mockUpdateResult = {
+            name: 'updateFile',
+            args: {
+                filePath: '/test/path.ts',
+                newContent: 'updated content',
+                explanation: 'Updated file',
+            },
+        };
+        mockGenerateContentFn.mockResolvedValueOnce([mockCodegenSummary]).mockResolvedValueOnce([mockUpdateResult]);
+        const result = await executeStepCodegenSummary(mockGenerateContentFn, mockPrompt, mockFunctionDefs, mockOptions, mockWaitIfPaused);
+        expect(result).toHaveLength(1);
+        expect(result[0]).toEqual(mockUpdateResult);
+        expect(mockGenerateContentFn).toHaveBeenCalledTimes(2);
+    });
+    it('should handle validation and recovery of invalid codegen summary', async () => {
+        const invalidCodegenSummary = {
+            name: 'codegenSummary',
+            args: {
+            // Missing required fields
+            },
+        };
+        const validCodegenSummary = {
+            name: 'codegenSummary',
+            args: {
+                explanation: 'Recovered explanation',
+                fileUpdates: [],
+                contextPaths: [],
+            },
+        };
+        mockGenerateContentFn.mockResolvedValueOnce([invalidCodegenSummary]).mockResolvedValueOnce([validCodegenSummary]);
+        const result = await executeStepCodegenSummary(mockGenerateContentFn, mockPrompt, mockFunctionDefs, mockOptions, mockWaitIfPaused);
+        expect(result).toHaveLength(0);
+        expect(mockGenerateContentFn).toHaveBeenCalledTimes(2);
+    });
+    it('should handle errors in generate content function', async () => {
+        mockGenerateContentFn.mockRejectedValueOnce(new Error('Test error'));
+        await expect(executeStepCodegenSummary(mockGenerateContentFn, mockPrompt, mockFunctionDefs, mockOptions, mockWaitIfPaused)).rejects.toThrow('Test error');
+    });
+    it('should handle no codegen summary in response', async () => {
+        mockGenerateContentFn.mockResolvedValueOnce([
+            {
+                name: 'explanation',
+                args: { text: 'No changes needed' },
+            },
+        ]);
+        const result = await executeStepCodegenSummary(mockGenerateContentFn, mockPrompt, mockFunctionDefs, mockOptions, mockWaitIfPaused);
+        expect(result).toHaveLength(1);
+        expect(result[0].name).toBe('explanation');
+    });
+    it('should respect waitIfPaused during file updates', async () => {
+        const mockCodegenSummary = {
+            name: 'codegenSummary',
+            args: {
+                explanation: 'Test pause handling',
+                fileUpdates: [
+                    { filePath: '/test/file1.ts', updateToolName: 'updateFile', prompt: 'Update 1' },
+                    { filePath: '/test/file2.ts', updateToolName: 'updateFile', prompt: 'Update 2' },
+                ],
+                contextPaths: [],
+            },
+        };
+        mockGenerateContentFn
+            .mockResolvedValueOnce([mockCodegenSummary])
+            .mockResolvedValueOnce([
+            { name: 'updateFile', args: { filePath: '/test/file1.ts', newContent: 'test', explanation: 'test' } },
+        ])
+            .mockResolvedValueOnce([
+            { name: 'updateFile', args: { filePath: '/test/file2.ts', newContent: 'test', explanation: 'test' } },
+        ]);
+        await executeStepCodegenSummary(mockGenerateContentFn, mockPrompt, mockFunctionDefs, mockOptions, mockWaitIfPaused);
+        expect(mockWaitIfPaused).toHaveBeenCalledTimes(2);
+    });
+});
+//# sourceMappingURL=step-codegen-summary.test.js.map

package/dist/prompt/steps/step-codegen-summary.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"step-codegen-summary.test.js","sourceRoot":"","sources":["../../../src/prompt/steps/step-codegen-summary.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAC9D,OAAO,EAAE,yBAAyB,EAAE,MAAM,2BAA2B,CAAC;AAGtE,OAAO,EAAE,aAAa,EAAE,MAAM,2BAA2B,CAAC;AAE1D,EAAE,CAAC,IAAI,CAAC,2BAA2B,EAAE,GAAG,EAAE,CAAC,CAAC;IAC1C,YAAY,EAAE,EAAE,CAAC,EAAE,EAAE;CACtB,CAAC,CAAC,CAAC;AACJ,EAAE,CAAC,IAAI,CAAC,2BAA2B,EAAE,GAAG,EAAE,CAAC,CAAC;IAC1C,aAAa,EAAE,EAAE,CAAC,EAAE,EAAE;CACvB,CAAC,CAAC,CAAC;AACJ,EAAE,CAAC,IAAI,CAAC,sBAAsB,EAAE,GAAG,EAAE,CAAC,CAAC;IACrC,OAAO,EAAE,OAAO;IAChB,QAAQ,EAAE;QACR,OAAO,EAAE,OAAO;QAChB,UAAU,EAAE,CAAC,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,CAAC;KAC3C;IACD,gBAAgB,EAAE,EAAE;CACrB,CAAC,CAAC,CAAC;AAEJ,QAAQ,CAAC,2BAA2B,EAAE,GAAG,EAAE;IACzC,MAAM,qBAAqB,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;IACtC,MAAM,gBAAgB,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;IAEjC,YAAY;IACZ,MAAM,UAAU,GAAiB,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,aAAa,EAAE,CAAC,CAAC;IACzE,MAAM,gBAAgB,GAAkB;QACtC;YACE,IAAI,EAAE,gBAAgB;YACtB,WAAW,EAAE,yCAAyC;YACtD,UAAU,EAAE;gBACV,IAAI,EAAE,QAAQ;gBACd,UAAU,EAAE;oBACV,WAAW,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,wBAAwB,EAAE;oBACtE,WAAW,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,WAAW,EAAE,sBAAsB,EAAE;oBACnE,YAAY,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,WAAW,EAAE,uBAAuB,EAAE;iBACtE;gBACD,QAAQ,EAAE,CAAC,aAAa,EAAE,aAAa,EAAE,cAAc,CAAC;aACzD;SACF;KACF,CAAC;IAEF,MAAM,WAAW,GAAmB;QAClC,SAAS,EAAE,WAAW;QACtB,WAAW,EAAE,GAAG;QAChB,WAAW,EAAE,KAAK;KACnB,CAAC;IAEF,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;QACnB,gBAAgB,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;QAE9C,EAAE,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,eAAe,CAAC,EAAE,CAAC,CAAC;IAC/C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;QACnE,MAAM,kBAAkB,GAAiB;YACvC,IAAI,EAAE,gBAAgB;YACtB,IAAI,EAAE;gBACJ,WAAW,EAAE,kBAAkB;gBAC/B,WAAW,EAAE;oBACX;wBACE,QAAQ,EAAE,eAAe;wBACzB,cAAc,EAAE,YAAY;wBAC5B,MAAM,EAAE,kBAAkB;qBAC3B;iBACF;gBACD,YAAY,EAAE,CAAC,kBAAkB,CAAC;aACnC;SACF,CAAC;QAEF,MAAM,gBAAgB,GAAiB;YACrC,IAAI,EAAE,YAAY;YAClB,IAAI,EAAE;gBACJ,QAAQ,EAAE,eAAe;gBACzB,UAAU,EAAE,iBAAiB;gBAC7B,WAAW,EAAE,cAAc;aAC5B;SACF,CAAC;QAEF,qBAAqB,CAAC,qBAAqB,CAAC,CAAC,kBAAkB,CAAC,CAAC,CAAC,qBAAqB,CAAC,CAAC,gBAAgB,CAAC,CAAC,CAAC;QAE5G,MAAM,MAAM,GAAG,MAAM,yBAAyB,CAC5C,qBAAqB,EACrB,UAAU,EACV,gBAAgB,EAChB,WAAW,EACX,gBAAgB,CACjB,CAAC;QAEF,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC/B,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAC;QAC5C,MAAM,CAAC,qBAAqB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IACzD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kEAAkE,EAAE,KAAK,IAAI,EAAE;QAChF,MAAM,qBAAqB,GAAiB;YAC1C,IAAI,EAAE,gBAAgB;YACtB,IAAI,EAAE;YACJ,0BAA0B;aAC3B;SACF,CAAC;QAEF,MAAM,mBAAmB,GAAiB;YACxC,IAAI,EAAE,gBAAgB;YACtB,IAAI,EAAE;gBACJ,WAAW,EAAE,uBAAuB;gBACpC,WAAW,EAAE,EAAE;gBACf,YAAY,EAAE,EAAE;aACjB;SACF,CAAC;QAEF,qBAAqB,CAAC,qBAAqB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,qBAAqB,CAAC,CAAC,mBAAmB,CAAC,CAAC,CAAC;QAElH,MAAM,MAAM,GAAG,MAAM,yBAAyB,CAC5C,qBAAqB,EACrB,UAAU,EACV,gBAAgB,EAChB,WAAW,EACX,gBAAgB,CACjB,CAAC;QAEF,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC/B,MAAM,CAAC,qBAAqB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IACzD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mDAAmD,EAAE,KAAK,IAAI,EAAE;QACjE,qBAAqB,CAAC,qBAAqB,CAAC,IAAI,KAAK,CAAC,YAAY,CAAC,CAAC,CAAC;QAErE,MAAM,MAAM,CACV,yBAAyB,CAAC,qBAAqB,EAAE,UAAU,EAAE,gBAAgB,EAAE,WAAW,EAAE,gBAAgB,CAAC,CAC9G,CAAC,OAAO,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;IAClC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;QAC5D,qBAAqB,CAAC,qBAAqB,CAAC;YAC1C;gBACE,IAAI,EAAE,aAAa;gBACnB,IAAI,EAAE,EAAE,IAAI,EAAE,mBAAmB,EAAE;aACpC;SACF,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,MAAM,yBAAyB,CAC5C,qBAAqB,EACrB,UAAU,EACV,gBAAgB,EAChB,WAAW,EACX,gBAAgB,CACjB,CAAC;QAEF,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC/B,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;IAC7C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;QAC/D,MAAM,kBAAkB,GAAiB;YACvC,IAAI,EAAE,gBAAgB;YACtB,IAAI,EAAE;gBACJ,WAAW,EAAE,qBAAqB;gBAClC,WAAW,EAAE;oBACX,EAAE,QAAQ,EAAE,gBAAgB,EAAE,cAAc,EAAE,YAAY,EAAE,MAAM,EAAE,UAAU,EAAE;oBAChF,EAAE,QAAQ,EAAE,gBAAgB,EAAE,cAAc,EAAE,YAAY,EAAE,MAAM,EAAE,UAAU,EAAE;iBACjF;gBACD,YAAY,EAAE,EAAE;aACjB;SACF,CAAC;QAEF,qBAAqB;aAClB,qBAAqB,CAAC,CAAC,kBAAkB,CAAC,CAAC;aAC3C,qBAAqB,CAAC;YACrB,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,EAAE,QAAQ,EAAE,gBAAgB,EAAE,UAAU,EAAE,MAAM,EAAE,WAAW,EAAE,MAAM,EAAE,EAAE;SACtG,CAAC;aACD,qBAAqB,CAAC;YACrB,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,EAAE,QAAQ,EAAE,gBAAgB,EAAE,UAAU,EAAE,MAAM,EAAE,WAAW,EAAE,MAAM,EAAE,EAAE;SACtG,CAAC,CAAC;QAEL,MAAM,yBAAyB,CAAC,qBAAqB,EAAE,UAAU,EAAE,gBAAgB,EAAE,WAAW,EAAE,gBAAgB,CAAC,CAAC;QAEpH,MAAM,CAAC,gBAAgB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IACpD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}