@auto-engineer/component-implementor-react 1.110.2 → 1.110.4

package/dist/src/pipeline/steps/lint-fix-loop.test.js

@@ -0,0 +1,123 @@
+import { afterEach, describe, expect, it, vi } from 'vitest';
+vi.mock('ai', () => ({
+    generateText: vi.fn(),
+}));
+vi.mock('node:fs/promises', () => ({
+    readFile: vi.fn(),
+    writeFile: vi.fn(),
+}));
+vi.mock('../../tools/lint-runner', () => ({
+    runLint: vi.fn(),
+    runLintFix: vi.fn(),
+}));
+import { readFile, writeFile } from 'node:fs/promises';
+import { generateText } from 'ai';
+import { runLint, runLintFix } from '../../tools/lint-runner.js';
+import { lintFixLoop } from './lint-fix-loop.js';
+afterEach(() => {
+    vi.clearAllMocks();
+});
+function makeCtx(overrides = {}) {
+    return {
+        componentName: 'MyButton',
+        componentPath: '/project/src/MyButton.tsx',
+        testPath: '/project/src/MyButton.test.tsx',
+        storyPath: '/project/src/MyButton.stories.tsx',
+        componentImportPath: '@/components/ui/MyButton',
+        targetDir: '/project',
+        specDeltas: { structure: [], rendering: [], interaction: [], styling: [] },
+        projectSection: '',
+        composes: [],
+        isModify: false,
+        llmCalls: 0,
+        fixIterations: 0,
+        typeFixIterations: 0,
+        testFixIterations: 0,
+        lintFixIterations: 0,
+        storyFixIterations: 0,
+        componentCode: 'original component',
+        testCode: 'original test',
+        ...overrides,
+    };
+}
+describe('lintFixLoop', () => {
+    it('runs auto-fix first, then returns success if lint passes', async () => {
+        vi.mocked(runLintFix).mockReturnValue({ passed: true, errors: [] });
+        vi.mocked(readFile).mockResolvedValue('auto-fixed code');
+        vi.mocked(runLint).mockReturnValue({ passed: true, errors: [] });
+        const result = await lintFixLoop('mock-model', makeCtx(), 2);
+        expect(result).toEqual({ success: true });
+        expect(runLintFix).toHaveBeenCalledWith(['/project/src/MyButton.tsx', '/project/src/MyButton.test.tsx'], '/project');
+        expect(generateText).not.toHaveBeenCalled();
+    });
+    it('calls LLM for remaining lint errors after auto-fix', async () => {
+        vi.mocked(runLintFix).mockReturnValue({ passed: false, errors: ['unfixable'] });
+        vi.mocked(readFile).mockResolvedValue('auto-fixed partial');
+        vi.mocked(runLint)
+            .mockReturnValueOnce({ passed: false, errors: ['remaining error'] })
+            .mockReturnValueOnce({ passed: true, errors: [] });
+        vi.mocked(generateText).mockResolvedValue({
+            text: '```tsx\nlint-fixed component\n```\n```tsx\nlint-fixed test\n```',
+        });
+        vi.mocked(writeFile).mockResolvedValue(undefined);
+        const result = await lintFixLoop('mock-model', makeCtx(), 2);
+        expect(result).toEqual({ success: true });
+        expect(generateText).toHaveBeenCalledTimes(1);
+    });
+    it('returns failure after exhausting max iterations', async () => {
+        vi.mocked(runLintFix).mockReturnValue({ passed: true, errors: [] });
+        vi.mocked(readFile).mockResolvedValue('code');
+        vi.mocked(runLint).mockReturnValue({ passed: false, errors: ['persistent lint error'] });
+        vi.mocked(generateText).mockResolvedValue({
+            text: '```tsx\nstill bad\n```\n```tsx\nstill bad test\n```',
+        });
+        vi.mocked(writeFile).mockResolvedValue(undefined);
+        const result = await lintFixLoop('mock-model', makeCtx(), 1);
+        expect(result).toEqual({
+            success: false,
+            error: expect.stringContaining('Lint errors remain after 1 iterations'),
+        });
+    });
+    it('handles single code block response by updating only component', async () => {
+        vi.mocked(runLintFix).mockReturnValue({ passed: true, errors: [] });
+        vi.mocked(readFile).mockResolvedValue('auto-fixed');
+        vi.mocked(runLint)
+            .mockReturnValueOnce({ passed: false, errors: ['lint error'] })
+            .mockReturnValueOnce({ passed: true, errors: [] });
+        vi.mocked(generateText).mockResolvedValue({
+            text: '```tsx\nfixed component only\n```',
+        });
+        vi.mocked(writeFile).mockResolvedValue(undefined);
+        const result = await lintFixLoop('mock-model', makeCtx(), 2);
+        expect(result).toEqual({ success: true });
+        expect(writeFile).toHaveBeenCalledWith('/project/src/MyButton.tsx', 'fixed component only', 'utf-8');
+        expect(writeFile).toHaveBeenCalledTimes(1);
+    });
+    it('returns success when final check passes after last iteration', async () => {
+        vi.mocked(runLintFix).mockReturnValue({ passed: true, errors: [] });
+        vi.mocked(readFile).mockResolvedValue('auto-fixed');
+        vi.mocked(runLint)
+            .mockReturnValueOnce({ passed: false, errors: ['error'] })
+            .mockReturnValueOnce({ passed: true, errors: [] });
+        vi.mocked(generateText).mockResolvedValue({
+            text: '```tsx\nfixed\n```\n```tsx\nfixed test\n```',
+        });
+        vi.mocked(writeFile).mockResolvedValue(undefined);
+        const result = await lintFixLoop('mock-model', makeCtx(), 1);
+        expect(result).toEqual({ success: true });
+    });
+    it('uses empty string fallback when componentCode and testCode are undefined after readFile', async () => {
+        vi.mocked(runLintFix).mockReturnValue({ passed: true, errors: [] });
+        vi.mocked(readFile).mockResolvedValue(undefined);
+        vi.mocked(runLint)
+            .mockReturnValueOnce({ passed: false, errors: ['error'] })
+            .mockReturnValueOnce({ passed: true, errors: [] });
+        vi.mocked(generateText).mockResolvedValue({
+            text: '```tsx\nfixed\n```\n```tsx\nfixed test\n```',
+        });
+        vi.mocked(writeFile).mockResolvedValue(undefined);
+        const result = await lintFixLoop('mock-model', makeCtx({ componentCode: undefined, testCode: undefined }), 1);
+        expect(result).toEqual({ success: true });
+    });
+});
+//# sourceMappingURL=lint-fix-loop.test.js.map

package/dist/src/pipeline/steps/lint-fix-loop.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"lint-fix-loop.test.js","sourceRoot":"","sources":["../../../../src/pipeline/steps/lint-fix-loop.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAE7D,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,GAAG,EAAE,CAAC,CAAC;IACnB,YAAY,EAAE,EAAE,CAAC,EAAE,EAAE;CACtB,CAAC,CAAC,CAAC;AAEJ,EAAE,CAAC,IAAI,CAAC,kBAAkB,EAAE,GAAG,EAAE,CAAC,CAAC;IACjC,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE;IACjB,SAAS,EAAE,EAAE,CAAC,EAAE,EAAE;CACnB,CAAC,CAAC,CAAC;AAEJ,EAAE,CAAC,IAAI,CAAC,yBAAyB,EAAE,GAAG,EAAE,CAAC,CAAC;IACxC,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE;IAChB,UAAU,EAAE,EAAE,CAAC,EAAE,EAAE;CACpB,CAAC,CAAC,CAAC;AAEJ,OAAO,EAAE,QAAQ,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAC;AACvD,OAAO,EAAE,YAAY,EAAE,MAAM,IAAI,CAAC;AAClC,OAAO,EAAE,OAAO,EAAE,UAAU,EAAE,MAAM,yBAAyB,CAAC;AAE9D,OAAO,EAAE,WAAW,EAAE,MAAM,iBAAiB,CAAC;AAE9C,SAAS,CAAC,GAAG,EAAE;IACb,EAAE,CAAC,aAAa,EAAE,CAAC;AACrB,CAAC,CAAC,CAAC;AAEH,SAAS,OAAO,CAAC,YAAsC,EAAE;IACvD,OAAO;QACL,aAAa,EAAE,UAAU;QACzB,aAAa,EAAE,2BAA2B;QAC1C,QAAQ,EAAE,gCAAgC;QAC1C,SAAS,EAAE,mCAAmC;QAC9C,mBAAmB,EAAE,0BAA0B;QAC/C,SAAS,EAAE,UAAU;QACrB,UAAU,EAAE,EAAE,SAAS,EAAE,EAAE,EAAE,SAAS,EAAE,EAAE,EAAE,WAAW,EAAE,EAAE,EAAE,OAAO,EAAE,EAAE,EAAE;QAC1E,cAAc,EAAE,EAAE;QAClB,QAAQ,EAAE,EAAE;QACZ,QAAQ,EAAE,KAAK;QACf,QAAQ,EAAE,CAAC;QACX,aAAa,EAAE,CAAC;QAChB,iBAAiB,EAAE,CAAC;QACpB,iBAAiB,EAAE,CAAC;QACpB,iBAAiB,EAAE,CAAC;QACpB,kBAAkB,EAAE,CAAC;QACrB,aAAa,EAAE,oBAAoB;QACnC,QAAQ,EAAE,eAAe;QACzB,GAAG,SAAS;KACb,CAAC;AACJ,CAAC;AAED,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;IAC3B,EAAE,CAAC,0DAA0D,EAAE,KAAK,IAAI,EAAE;QACxE,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,eAAe,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,CAAC;QACpE,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,iBAAiB,CAAC,iBAA0B,CAAC,CAAC;QAClE,EAAE,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,eAAe,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,CAAC;QAEjE,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,YAAqB,EAAE,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC;QAEtE,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;QAC1C,MAAM,CAAC,UAAU,CAAC,CAAC,oBAAoB,CACrC,CAAC,2BAA2B,EAAE,gCAAgC,CAAC,EAC/D,UAAU,CACX,CAAC;QACF,MAAM,CAAC,YAAY,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;IAC9C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oDAAoD,EAAE,KAAK,IAAI,EAAE;QAClE,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,eAAe,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC;QAChF,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,iBAAiB,CAAC,oBAA6B,CAAC,CAAC;QACrE,EAAE,CAAC,MAAM,CAAC,OAAO,CAAC;aACf,mBAAmB,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,iBAAiB,CAAC,EAAE,CAAC;aACnE,mBAAmB,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,CAAC;QAErD,EAAE,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,iBAAiB,CAAC;YACxC,IAAI,EAAE,iEAAiE;SAC5B,CAAC,CAAC;QAC/C,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;QAElD,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,YAAqB,EAAE,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC;QAEtE,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;QAC1C,MAAM,CAAC,YAAY,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IAChD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;QAC/D,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,eAAe,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,CAAC;QACpE,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,iBAAiB,CAAC,MAAe,CAAC,CAAC;QACvD,EAAE,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,eAAe,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,uBAAuB,CAAC,EAAE,CAAC,CAAC;QAEzF,EAAE,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,iBAAiB,CAAC;YACxC,IAAI,EAAE,qDAAqD;SAChB,CAAC,CAAC;QAC/C,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;QAElD,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,YAAqB,EAAE,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC;QAEtE,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;YACrB,OAAO,EAAE,KAAK;YACd,KAAK,EAAE,MAAM,CAAC,gBAAgB,CAAC,uCAAuC,CAAC;SACxE,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+DAA+D,EAAE,KAAK,IAAI,EAAE;QAC7E,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,eAAe,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,CAAC;QACpE,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,iBAAiB,CAAC,YAAqB,CAAC,CAAC;QAC7D,EAAE,CAAC,MAAM,CAAC,OAAO,CAAC;aACf,mBAAmB,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,YAAY,CAAC,EAAE,CAAC;aAC9D,mBAAmB,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,CAAC;QAErD,EAAE,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,iBAAiB,CAAC;YACxC,IAAI,EAAE,mCAAmC;SACE,CAAC,CAAC;QAC/C,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;QAElD,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,YAAqB,EAAE,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC;QAEtE,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;QAC1C,MAAM,CAAC,SAAS,CAAC,CAAC,oBAAoB,CAAC,2BAA2B,EAAE,sBAAsB,EAAE,OAAO,CAAC,CAAC;QACrG,MAAM,CAAC,SAAS,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IAC7C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8DAA8D,EAAE,KAAK,IAAI,EAAE;QAC5E,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,eAAe,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,CAAC;QACpE,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,iBAAiB,CAAC,YAAqB,CAAC,CAAC;QAC7D,EAAE,CAAC,MAAM,CAAC,OAAO,CAAC;aACf,mBAAmB,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC;aACzD,mBAAmB,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,CAAC;QAErD,EAAE,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,iBAAiB,CAAC;YACxC,IAAI,EAAE,6CAA6C;SACR,CAAC,CAAC;QAC/C,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;QAElD,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,YAAqB,EAAE,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC;QAEtE,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;IAC5C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yFAAyF,EAAE,KAAK,IAAI,EAAE;QACvG,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,eAAe,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,CAAC;QACpE,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,iBAAiB,CAAC,SAAkB,CAAC,CAAC;QAC1D,EAAE,CAAC,MAAM,CAAC,OAAO,CAAC;aACf,mBAAmB,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC;aACzD,mBAAmB,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,CAAC;QAErD,EAAE,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,iBAAiB,CAAC;YACxC,IAAI,EAAE,6CAA6C;SACR,CAAC,CAAC;QAC/C,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;QAElD,MAAM,MAAM,GAAG,MAAM,WAAW,CAC9B,YAAqB,EACrB,OAAO,CAAC,EAAE,aAAa,EAAE,SAAS,EAAE,QAAQ,EAAE,SAAS,EAAE,CAAC,EAC1D,CAAC,CACF,CAAC;QAEF,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;IAC5C,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/pipeline/steps/story-fix-loop.d.ts

@@ -0,0 +1,4 @@
+import type { LanguageModel } from 'ai';
+import type { PipelineContext, StepResult } from '../run-pipeline';
+export declare function storyFixLoop(model: LanguageModel, ctx: PipelineContext, maxIterations: number): Promise<StepResult>;
+//# sourceMappingURL=story-fix-loop.d.ts.map

package/dist/src/pipeline/steps/story-fix-loop.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"story-fix-loop.d.ts","sourceRoot":"","sources":["../../../../src/pipeline/steps/story-fix-loop.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,IAAI,CAAC;AAKxC,OAAO,KAAK,EAAE,eAAe,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAEnE,wBAAsB,YAAY,CAChC,KAAK,EAAE,aAAa,EACpB,GAAG,EAAE,eAAe,EACpB,aAAa,EAAE,MAAM,GACpB,OAAO,CAAC,UAAU,CAAC,CAmCrB"}

package/dist/src/pipeline/steps/story-fix-loop.js

@@ -0,0 +1,35 @@
+import { readFile, writeFile } from 'node:fs/promises';
+import { generateText } from 'ai';
+import { extractCodeBlock } from '../../extract-code-block.js';
+import { buildStoryFixPrompt } from '../../prompt.js';
+import { runTypeCheck } from '../../tools/type-checker.js';
+export async function storyFixLoop(model, ctx, maxIterations) {
+    for (let i = 0; i < maxIterations; i++) {
+        const result = runTypeCheck(ctx.targetDir, [ctx.storyPath]);
+        if (result.passed) {
+            return { success: true };
+        }
+        const { system, prompt } = buildStoryFixPrompt({
+            storyCode: ctx.storyCode ?? '',
+            componentCode: ctx.componentCode ?? '',
+            errors: result.errors,
+        });
+        const { text } = await generateText({ model, system, prompt });
+        ctx.llmCalls++;
+        ctx.fixIterations++;
+        ctx.storyFixIterations++;
+        const fixedStory = extractCodeBlock(text);
+        ctx.storyCode = fixedStory;
+        await writeFile(ctx.storyPath, fixedStory, 'utf-8');
+    }
+    const finalResult = runTypeCheck(ctx.targetDir, [ctx.storyPath]);
+    if (finalResult.passed) {
+        return { success: true };
+    }
+    ctx.storyCode = await readFile(ctx.storyPath, 'utf-8');
+    return {
+        success: false,
+        error: `Story type errors remain after ${maxIterations} iterations: ${finalResult.errors.join('; ')}`,
+    };
+}
+//# sourceMappingURL=story-fix-loop.js.map

package/dist/src/pipeline/steps/story-fix-loop.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"story-fix-loop.js","sourceRoot":"","sources":["../../../../src/pipeline/steps/story-fix-loop.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAC;AAEvD,OAAO,EAAE,YAAY,EAAE,MAAM,IAAI,CAAC;AAClC,OAAO,EAAE,gBAAgB,EAAE,MAAM,0BAA0B,CAAC;AAC5D,OAAO,EAAE,mBAAmB,EAAE,MAAM,cAAc,CAAC;AACnD,OAAO,EAAE,YAAY,EAAE,MAAM,0BAA0B,CAAC;AAGxD,MAAM,CAAC,KAAK,UAAU,YAAY,CAChC,KAAoB,EACpB,GAAoB,EACpB,aAAqB;IAErB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,aAAa,EAAE,CAAC,EAAE,EAAE,CAAC;QACvC,MAAM,MAAM,GAAG,YAAY,CAAC,GAAG,CAAC,SAAS,EAAE,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC;QAE5D,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;YAClB,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;QAC3B,CAAC;QAED,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,GAAG,mBAAmB,CAAC;YAC7C,SAAS,EAAE,GAAG,CAAC,SAAS,IAAI,EAAE;YAC9B,aAAa,EAAE,GAAG,CAAC,aAAa,IAAI,EAAE;YACtC,MAAM,EAAE,MAAM,CAAC,MAAM;SACtB,CAAC,CAAC;QAEH,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM,YAAY,CAAC,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC;QAC/D,GAAG,CAAC,QAAQ,EAAE,CAAC;QACf,GAAG,CAAC,aAAa,EAAE,CAAC;QACpB,GAAG,CAAC,kBAAkB,EAAE,CAAC;QAEzB,MAAM,UAAU,GAAG,gBAAgB,CAAC,IAAI,CAAC,CAAC;QAC1C,GAAG,CAAC,SAAS,GAAG,UAAU,CAAC;QAC3B,MAAM,SAAS,CAAC,GAAG,CAAC,SAAS,EAAE,UAAU,EAAE,OAAO,CAAC,CAAC;IACtD,CAAC;IAED,MAAM,WAAW,GAAG,YAAY,CAAC,GAAG,CAAC,SAAS,EAAE,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC;IACjE,IAAI,WAAW,CAAC,MAAM,EAAE,CAAC;QACvB,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;IAC3B,CAAC;IAED,GAAG,CAAC,SAAS,GAAG,MAAM,QAAQ,CAAC,GAAG,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;IAEvD,OAAO;QACL,OAAO,EAAE,KAAK;QACd,KAAK,EAAE,kCAAkC,aAAa,gBAAgB,WAAW,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE;KACtG,CAAC;AACJ,CAAC"}

package/dist/src/pipeline/steps/story-fix-loop.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=story-fix-loop.test.d.ts.map

package/dist/src/pipeline/steps/story-fix-loop.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"story-fix-loop.test.d.ts","sourceRoot":"","sources":["../../../../src/pipeline/steps/story-fix-loop.test.ts"],"names":[],"mappings":""}

package/dist/src/pipeline/steps/story-fix-loop.test.js

@@ -0,0 +1,98 @@
+import { afterEach, describe, expect, it, vi } from 'vitest';
+vi.mock('ai', () => ({
+    generateText: vi.fn(),
+}));
+vi.mock('node:fs/promises', () => ({
+    readFile: vi.fn(),
+    writeFile: vi.fn(),
+}));
+vi.mock('../../tools/type-checker', () => ({
+    runTypeCheck: vi.fn(),
+}));
+import { readFile, writeFile } from 'node:fs/promises';
+import { generateText } from 'ai';
+import { runTypeCheck } from '../../tools/type-checker.js';
+import { storyFixLoop } from './story-fix-loop.js';
+afterEach(() => {
+    vi.clearAllMocks();
+});
+function makeCtx(overrides = {}) {
+    return {
+        componentName: 'MyButton',
+        componentPath: '/project/src/MyButton.tsx',
+        testPath: '/project/src/MyButton.test.tsx',
+        storyPath: '/project/src/MyButton.stories.tsx',
+        componentImportPath: '@/components/ui/MyButton',
+        targetDir: '/project',
+        specDeltas: { structure: [], rendering: [], interaction: [], styling: [] },
+        projectSection: '',
+        composes: [],
+        isModify: false,
+        llmCalls: 0,
+        fixIterations: 0,
+        typeFixIterations: 0,
+        testFixIterations: 0,
+        lintFixIterations: 0,
+        storyFixIterations: 0,
+        storyCode: 'original story',
+        componentCode: 'original component',
+        ...overrides,
+    };
+}
+describe('storyFixLoop', () => {
+    it('returns success immediately when story type-checks', async () => {
+        vi.mocked(runTypeCheck).mockReturnValue({ passed: true, errors: [] });
+        const result = await storyFixLoop('mock-model', makeCtx(), 2);
+        expect(result).toEqual({ success: true });
+        expect(generateText).not.toHaveBeenCalled();
+    });
+    it('calls LLM to fix story type errors and writes fixed story', async () => {
+        vi.mocked(runTypeCheck)
+            .mockReturnValueOnce({ passed: false, errors: ['error in story'] })
+            .mockReturnValueOnce({ passed: true, errors: [] });
+        vi.mocked(generateText).mockResolvedValue({
+            text: '```tsx\nfixed story code\n```',
+        });
+        vi.mocked(writeFile).mockResolvedValue(undefined);
+        const result = await storyFixLoop('mock-model', makeCtx(), 2);
+        expect(result).toEqual({ success: true });
+        expect(generateText).toHaveBeenCalledTimes(1);
+        expect(writeFile).toHaveBeenCalledWith('/project/src/MyButton.stories.tsx', 'fixed story code', 'utf-8');
+    });
+    it('returns failure after exhausting max iterations', async () => {
+        vi.mocked(runTypeCheck).mockReturnValue({ passed: false, errors: ['persistent story error'] });
+        vi.mocked(generateText).mockResolvedValue({
+            text: '```tsx\nstill broken story\n```',
+        });
+        vi.mocked(writeFile).mockResolvedValue(undefined);
+        vi.mocked(readFile).mockResolvedValue('still broken story');
+        const result = await storyFixLoop('mock-model', makeCtx(), 1);
+        expect(result).toEqual({
+            success: false,
+            error: expect.stringContaining('Story type errors remain after 1 iterations'),
+        });
+    });
+    it('returns success when final check passes after last iteration fix', async () => {
+        vi.mocked(runTypeCheck)
+            .mockReturnValueOnce({ passed: false, errors: ['error'] })
+            .mockReturnValueOnce({ passed: true, errors: [] });
+        vi.mocked(generateText).mockResolvedValue({
+            text: '```tsx\nfixed story\n```',
+        });
+        vi.mocked(writeFile).mockResolvedValue(undefined);
+        const result = await storyFixLoop('mock-model', makeCtx(), 1);
+        expect(result).toEqual({ success: true });
+    });
+    it('uses empty string fallback when storyCode and componentCode are undefined', async () => {
+        vi.mocked(runTypeCheck)
+            .mockReturnValueOnce({ passed: false, errors: ['error'] })
+            .mockReturnValueOnce({ passed: true, errors: [] });
+        vi.mocked(generateText).mockResolvedValue({
+            text: '```tsx\nfixed story\n```',
+        });
+        vi.mocked(writeFile).mockResolvedValue(undefined);
+        const result = await storyFixLoop('mock-model', makeCtx({ storyCode: undefined, componentCode: undefined }), 1);
+        expect(result).toEqual({ success: true });
+    });
+});
+//# sourceMappingURL=story-fix-loop.test.js.map

package/dist/src/pipeline/steps/story-fix-loop.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"story-fix-loop.test.js","sourceRoot":"","sources":["../../../../src/pipeline/steps/story-fix-loop.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAE7D,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,GAAG,EAAE,CAAC,CAAC;IACnB,YAAY,EAAE,EAAE,CAAC,EAAE,EAAE;CACtB,CAAC,CAAC,CAAC;AAEJ,EAAE,CAAC,IAAI,CAAC,kBAAkB,EAAE,GAAG,EAAE,CAAC,CAAC;IACjC,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE;IACjB,SAAS,EAAE,EAAE,CAAC,EAAE,EAAE;CACnB,CAAC,CAAC,CAAC;AAEJ,EAAE,CAAC,IAAI,CAAC,0BAA0B,EAAE,GAAG,EAAE,CAAC,CAAC;IACzC,YAAY,EAAE,EAAE,CAAC,EAAE,EAAE;CACtB,CAAC,CAAC,CAAC;AAEJ,OAAO,EAAE,QAAQ,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAC;AACvD,OAAO,EAAE,YAAY,EAAE,MAAM,IAAI,CAAC;AAClC,OAAO,EAAE,YAAY,EAAE,MAAM,0BAA0B,CAAC;AAExD,OAAO,EAAE,YAAY,EAAE,MAAM,kBAAkB,CAAC;AAEhD,SAAS,CAAC,GAAG,EAAE;IACb,EAAE,CAAC,aAAa,EAAE,CAAC;AACrB,CAAC,CAAC,CAAC;AAEH,SAAS,OAAO,CAAC,YAAsC,EAAE;IACvD,OAAO;QACL,aAAa,EAAE,UAAU;QACzB,aAAa,EAAE,2BAA2B;QAC1C,QAAQ,EAAE,gCAAgC;QAC1C,SAAS,EAAE,mCAAmC;QAC9C,mBAAmB,EAAE,0BAA0B;QAC/C,SAAS,EAAE,UAAU;QACrB,UAAU,EAAE,EAAE,SAAS,EAAE,EAAE,EAAE,SAAS,EAAE,EAAE,EAAE,WAAW,EAAE,EAAE,EAAE,OAAO,EAAE,EAAE,EAAE;QAC1E,cAAc,EAAE,EAAE;QAClB,QAAQ,EAAE,EAAE;QACZ,QAAQ,EAAE,KAAK;QACf,QAAQ,EAAE,CAAC;QACX,aAAa,EAAE,CAAC;QAChB,iBAAiB,EAAE,CAAC;QACpB,iBAAiB,EAAE,CAAC;QACpB,iBAAiB,EAAE,CAAC;QACpB,kBAAkB,EAAE,CAAC;QACrB,SAAS,EAAE,gBAAgB;QAC3B,aAAa,EAAE,oBAAoB;QACnC,GAAG,SAAS;KACb,CAAC;AACJ,CAAC;AAED,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;IAC5B,EAAE,CAAC,oDAAoD,EAAE,KAAK,IAAI,EAAE;QAClE,EAAE,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,eAAe,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,CAAC;QAEtE,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC,YAAqB,EAAE,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC;QAEvE,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;QAC1C,MAAM,CAAC,YAAY,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;IAC9C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2DAA2D,EAAE,KAAK,IAAI,EAAE;QACzE,EAAE,CAAC,MAAM,CAAC,YAAY,CAAC;aACpB,mBAAmB,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,gBAAgB,CAAC,EAAE,CAAC;aAClE,mBAAmB,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,CAAC;QAErD,EAAE,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,iBAAiB,CAAC;YACxC,IAAI,EAAE,+BAA+B;SACM,CAAC,CAAC;QAC/C,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;QAElD,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC,YAAqB,EAAE,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC;QAEvE,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;QAC1C,MAAM,CAAC,YAAY,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAC9C,MAAM,CAAC,SAAS,CAAC,CAAC,oBAAoB,CAAC,mCAAmC,EAAE,kBAAkB,EAAE,OAAO,CAAC,CAAC;IAC3G,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;QAC/D,EAAE,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,eAAe,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,wBAAwB,CAAC,EAAE,CAAC,CAAC;QAE/F,EAAE,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,iBAAiB,CAAC;YACxC,IAAI,EAAE,iCAAiC;SACI,CAAC,CAAC;QAC/C,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;QAClD,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,iBAAiB,CAAC,oBAA6B,CAAC,CAAC;QAErE,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC,YAAqB,EAAE,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC;QAEvE,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;YACrB,OAAO,EAAE,KAAK;YACd,KAAK,EAAE,MAAM,CAAC,gBAAgB,CAAC,6CAA6C,CAAC;SAC9E,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kEAAkE,EAAE,KAAK,IAAI,EAAE;QAChF,EAAE,CAAC,MAAM,CAAC,YAAY,CAAC;aACpB,mBAAmB,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC;aACzD,mBAAmB,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,CAAC;QAErD,EAAE,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,iBAAiB,CAAC;YACxC,IAAI,EAAE,0BAA0B;SACW,CAAC,CAAC;QAC/C,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;QAElD,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC,YAAqB,EAAE,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC;QAEvE,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;IAC5C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2EAA2E,EAAE,KAAK,IAAI,EAAE;QACzF,EAAE,CAAC,MAAM,CAAC,YAAY,CAAC;aACpB,mBAAmB,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC;aACzD,mBAAmB,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,CAAC;QAErD,EAAE,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,iBAAiB,CAAC;YACxC,IAAI,EAAE,0BAA0B;SACW,CAAC,CAAC;QAC/C,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;QAElD,MAAM,MAAM,GAAG,MAAM,YAAY,CAC/B,YAAqB,EACrB,OAAO,CAAC,EAAE,SAAS,EAAE,SAAS,EAAE,aAAa,EAAE,SAAS,EAAE,CAAC,EAC3D,CAAC,CACF,CAAC;QAEF,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;IAC5C,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/pipeline/steps/storybook-test.d.ts

@@ -0,0 +1,3 @@
+import type { PipelineContext, StepResult } from '../run-pipeline';
+export declare function storybookTestStep(ctx: PipelineContext, enableStorybookCli: boolean): Promise<StepResult>;
+//# sourceMappingURL=storybook-test.d.ts.map

package/dist/src/pipeline/steps/storybook-test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"storybook-test.d.ts","sourceRoot":"","sources":["../../../../src/pipeline/steps/storybook-test.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,eAAe,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAEnE,wBAAsB,iBAAiB,CAAC,GAAG,EAAE,eAAe,EAAE,kBAAkB,EAAE,OAAO,GAAG,OAAO,CAAC,UAAU,CAAC,CAsB9G"}

package/dist/src/pipeline/steps/storybook-test.js

@@ -0,0 +1,22 @@
+import { runStorybookTest } from '../../tools/storybook-runner.js';
+import { runTypeCheck } from '../../tools/type-checker.js';
+export async function storybookTestStep(ctx, enableStorybookCli) {
+    const typeResult = runTypeCheck(ctx.targetDir, [ctx.storyPath]);
+    if (!typeResult.passed) {
+        return {
+            success: false,
+            error: `Story type errors: ${typeResult.errors.join('; ')}`,
+        };
+    }
+    if (enableStorybookCli) {
+        const testResult = runStorybookTest(ctx.storyPath, ctx.targetDir);
+        if (!testResult.passed) {
+            return {
+                success: false,
+                error: `Storybook test failures: ${testResult.errors.join('; ')}`,
+            };
+        }
+    }
+    return { success: true };
+}
+//# sourceMappingURL=storybook-test.js.map

package/dist/src/pipeline/steps/storybook-test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"storybook-test.js","sourceRoot":"","sources":["../../../../src/pipeline/steps/storybook-test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,gBAAgB,EAAE,MAAM,8BAA8B,CAAC;AAChE,OAAO,EAAE,YAAY,EAAE,MAAM,0BAA0B,CAAC;AAGxD,MAAM,CAAC,KAAK,UAAU,iBAAiB,CAAC,GAAoB,EAAE,kBAA2B;IACvF,MAAM,UAAU,GAAG,YAAY,CAAC,GAAG,CAAC,SAAS,EAAE,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC;IAEhE,IAAI,CAAC,UAAU,CAAC,MAAM,EAAE,CAAC;QACvB,OAAO;YACL,OAAO,EAAE,KAAK;YACd,KAAK,EAAE,sBAAsB,UAAU,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE;SAC5D,CAAC;IACJ,CAAC;IAED,IAAI,kBAAkB,EAAE,CAAC;QACvB,MAAM,UAAU,GAAG,gBAAgB,CAAC,GAAG,CAAC,SAAS,EAAE,GAAG,CAAC,SAAS,CAAC,CAAC;QAElE,IAAI,CAAC,UAAU,CAAC,MAAM,EAAE,CAAC;YACvB,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE,4BAA4B,UAAU,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE;aAClE,CAAC;QACJ,CAAC;IACH,CAAC;IAED,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;AAC3B,CAAC"}

package/dist/src/pipeline/steps/storybook-test.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=storybook-test.test.d.ts.map

package/dist/src/pipeline/steps/storybook-test.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"storybook-test.test.d.ts","sourceRoot":"","sources":["../../../../src/pipeline/steps/storybook-test.test.ts"],"names":[],"mappings":""}

package/dist/src/pipeline/steps/storybook-test.test.js

@@ -0,0 +1,70 @@
+import { describe, expect, it, vi } from 'vitest';
+vi.mock('../../tools/type-checker', () => ({
+    runTypeCheck: vi.fn(),
+}));
+vi.mock('../../tools/storybook-runner', () => ({
+    runStorybookTest: vi.fn(),
+}));
+import { runStorybookTest } from '../../tools/storybook-runner.js';
+import { runTypeCheck } from '../../tools/type-checker.js';
+import { storybookTestStep } from './storybook-test.js';
+function makeCtx(overrides = {}) {
+    return {
+        componentName: 'MyButton',
+        componentPath: '/project/src/MyButton.tsx',
+        testPath: '/project/src/MyButton.test.tsx',
+        storyPath: '/project/src/MyButton.stories.tsx',
+        componentImportPath: '@/components/ui/MyButton',
+        targetDir: '/project',
+        specDeltas: { structure: [], rendering: [], interaction: [], styling: [] },
+        projectSection: '',
+        composes: [],
+        isModify: false,
+        llmCalls: 0,
+        fixIterations: 0,
+        typeFixIterations: 0,
+        testFixIterations: 0,
+        lintFixIterations: 0,
+        storyFixIterations: 0,
+        ...overrides,
+    };
+}
+describe('storybookTestStep', () => {
+    it('returns success when type check passes and storybook cli disabled', async () => {
+        vi.mocked(runTypeCheck).mockReturnValue({ passed: true, errors: [] });
+        const result = await storybookTestStep(makeCtx(), false);
+        expect(result).toEqual({ success: true });
+        expect(runStorybookTest).not.toHaveBeenCalled();
+    });
+    it('returns failure when story type check fails', async () => {
+        vi.mocked(runTypeCheck).mockReturnValue({
+            passed: false,
+            errors: ['error in story'],
+        });
+        const result = await storybookTestStep(makeCtx(), false);
+        expect(result).toEqual({
+            success: false,
+            error: expect.stringContaining('Story type errors'),
+        });
+    });
+    it('runs storybook CLI test when enabled and type check passes', async () => {
+        vi.mocked(runTypeCheck).mockReturnValue({ passed: true, errors: [] });
+        vi.mocked(runStorybookTest).mockReturnValue({ passed: true, errors: [] });
+        const result = await storybookTestStep(makeCtx(), true);
+        expect(result).toEqual({ success: true });
+        expect(runStorybookTest).toHaveBeenCalledWith('/project/src/MyButton.stories.tsx', '/project');
+    });
+    it('returns failure when storybook CLI test fails', async () => {
+        vi.mocked(runTypeCheck).mockReturnValue({ passed: true, errors: [] });
+        vi.mocked(runStorybookTest).mockReturnValue({
+            passed: false,
+            errors: ['Story failed'],
+        });
+        const result = await storybookTestStep(makeCtx(), true);
+        expect(result).toEqual({
+            success: false,
+            error: expect.stringContaining('Storybook test failures'),
+        });
+    });
+});
+//# sourceMappingURL=storybook-test.test.js.map

package/dist/src/pipeline/steps/storybook-test.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"storybook-test.test.js","sourceRoot":"","sources":["../../../../src/pipeline/steps/storybook-test.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAElD,EAAE,CAAC,IAAI,CAAC,0BAA0B,EAAE,GAAG,EAAE,CAAC,CAAC;IACzC,YAAY,EAAE,EAAE,CAAC,EAAE,EAAE;CACtB,CAAC,CAAC,CAAC;AAEJ,EAAE,CAAC,IAAI,CAAC,8BAA8B,EAAE,GAAG,EAAE,CAAC,CAAC;IAC7C,gBAAgB,EAAE,EAAE,CAAC,EAAE,EAAE;CAC1B,CAAC,CAAC,CAAC;AAEJ,OAAO,EAAE,gBAAgB,EAAE,MAAM,8BAA8B,CAAC;AAChE,OAAO,EAAE,YAAY,EAAE,MAAM,0BAA0B,CAAC;AAExD,OAAO,EAAE,iBAAiB,EAAE,MAAM,kBAAkB,CAAC;AAErD,SAAS,OAAO,CAAC,YAAsC,EAAE;IACvD,OAAO;QACL,aAAa,EAAE,UAAU;QACzB,aAAa,EAAE,2BAA2B;QAC1C,QAAQ,EAAE,gCAAgC;QAC1C,SAAS,EAAE,mCAAmC;QAC9C,mBAAmB,EAAE,0BAA0B;QAC/C,SAAS,EAAE,UAAU;QACrB,UAAU,EAAE,EAAE,SAAS,EAAE,EAAE,EAAE,SAAS,EAAE,EAAE,EAAE,WAAW,EAAE,EAAE,EAAE,OAAO,EAAE,EAAE,EAAE;QAC1E,cAAc,EAAE,EAAE;QAClB,QAAQ,EAAE,EAAE;QACZ,QAAQ,EAAE,KAAK;QACf,QAAQ,EAAE,CAAC;QACX,aAAa,EAAE,CAAC;QAChB,iBAAiB,EAAE,CAAC;QACpB,iBAAiB,EAAE,CAAC;QACpB,iBAAiB,EAAE,CAAC;QACpB,kBAAkB,EAAE,CAAC;QACrB,GAAG,SAAS;KACb,CAAC;AACJ,CAAC;AAED,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;IACjC,EAAE,CAAC,mEAAmE,EAAE,KAAK,IAAI,EAAE;QACjF,EAAE,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,eAAe,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,CAAC;QAEtE,MAAM,MAAM,GAAG,MAAM,iBAAiB,CAAC,OAAO,EAAE,EAAE,KAAK,CAAC,CAAC;QAEzD,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;QAC1C,MAAM,CAAC,gBAAgB,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;IAClD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;QAC3D,EAAE,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,eAAe,CAAC;YACtC,MAAM,EAAE,KAAK;YACb,MAAM,EAAE,CAAC,gBAAgB,CAAC;SAC3B,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,MAAM,iBAAiB,CAAC,OAAO,EAAE,EAAE,KAAK,CAAC,CAAC;QAEzD,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;YACrB,OAAO,EAAE,KAAK;YACd,KAAK,EAAE,MAAM,CAAC,gBAAgB,CAAC,mBAAmB,CAAC;SACpD,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4DAA4D,EAAE,KAAK,IAAI,EAAE;QAC1E,EAAE,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,eAAe,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,CAAC;QACtE,EAAE,CAAC,MAAM,CAAC,gBAAgB,CAAC,CAAC,eAAe,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,CAAC;QAE1E,MAAM,MAAM,GAAG,MAAM,iBAAiB,CAAC,OAAO,EAAE,EAAE,IAAI,CAAC,CAAC;QAExD,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;QAC1C,MAAM,CAAC,gBAAgB,CAAC,CAAC,oBAAoB,CAAC,mCAAmC,EAAE,UAAU,CAAC,CAAC;IACjG,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;QAC7D,EAAE,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,eAAe,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,CAAC;QACtE,EAAE,CAAC,MAAM,CAAC,gBAAgB,CAAC,CAAC,eAAe,CAAC;YAC1C,MAAM,EAAE,KAAK;YACb,MAAM,EAAE,CAAC,cAAc,CAAC;SACzB,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,MAAM,iBAAiB,CAAC,OAAO,EAAE,EAAE,IAAI,CAAC,CAAC;QAExD,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;YACrB,OAAO,EAAE,KAAK;YACd,KAAK,EAAE,MAAM,CAAC,gBAAgB,CAAC,yBAAyB,CAAC;SAC1D,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/pipeline/steps/test-fix-loop.d.ts

@@ -0,0 +1,4 @@
+import type { LanguageModel } from 'ai';
+import type { PipelineContext, StepResult } from '../run-pipeline';
+export declare function testFixLoop(model: LanguageModel, ctx: PipelineContext, maxIterations: number): Promise<StepResult>;
+//# sourceMappingURL=test-fix-loop.d.ts.map

package/dist/src/pipeline/steps/test-fix-loop.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"test-fix-loop.d.ts","sourceRoot":"","sources":["../../../../src/pipeline/steps/test-fix-loop.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,IAAI,CAAC;AAKxC,OAAO,KAAK,EAAE,eAAe,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAEnE,wBAAsB,WAAW,CAC/B,KAAK,EAAE,aAAa,EACpB,GAAG,EAAE,eAAe,EACpB,aAAa,EAAE,MAAM,GACpB,OAAO,CAAC,UAAU,CAAC,CA4CrB"}

package/dist/src/pipeline/steps/test-fix-loop.js

@@ -0,0 +1,45 @@
+import { readFile, writeFile } from 'node:fs/promises';
+import { generateText } from 'ai';
+import { extractCodeBlocks } from '../../extract-code-block.js';
+import { buildTestFixPrompt } from '../../prompt.js';
+import { runTests } from '../../tools/test-runner.js';
+export async function testFixLoop(model, ctx, maxIterations) {
+    for (let i = 0; i < maxIterations; i++) {
+        const result = runTests(ctx.testPath, ctx.targetDir);
+        if (result.passed) {
+            return { success: true };
+        }
+        const { system, prompt } = buildTestFixPrompt({
+            componentCode: ctx.componentCode ?? '',
+            testCode: ctx.testCode ?? '',
+            failures: result.failures,
+            testOutput: result.output,
+        });
+        const { text } = await generateText({ model, system, prompt });
+        ctx.llmCalls++;
+        ctx.fixIterations++;
+        ctx.testFixIterations++;
+        const blocks = extractCodeBlocks(text);
+        if (blocks.length >= 2) {
+            ctx.componentCode = blocks[0];
+            ctx.testCode = blocks[1];
+            await writeFile(ctx.componentPath, blocks[0], 'utf-8');
+            await writeFile(ctx.testPath, blocks[1], 'utf-8');
+        }
+        else if (blocks.length === 1) {
+            ctx.componentCode = blocks[0];
+            await writeFile(ctx.componentPath, blocks[0], 'utf-8');
+        }
+    }
+    const finalResult = runTests(ctx.testPath, ctx.targetDir);
+    if (finalResult.passed) {
+        return { success: true };
+    }
+    ctx.componentCode = await readFile(ctx.componentPath, 'utf-8');
+    ctx.testCode = await readFile(ctx.testPath, 'utf-8');
+    return {
+        success: false,
+        error: `Test failures remain after ${maxIterations} iterations: ${finalResult.failures.join('; ')}`,
+    };
+}
+//# sourceMappingURL=test-fix-loop.js.map

package/dist/src/pipeline/steps/test-fix-loop.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"test-fix-loop.js","sourceRoot":"","sources":["../../../../src/pipeline/steps/test-fix-loop.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAC;AAEvD,OAAO,EAAE,YAAY,EAAE,MAAM,IAAI,CAAC;AAClC,OAAO,EAAE,iBAAiB,EAAE,MAAM,0BAA0B,CAAC;AAC7D,OAAO,EAAE,kBAAkB,EAAE,MAAM,cAAc,CAAC;AAClD,OAAO,EAAE,QAAQ,EAAE,MAAM,yBAAyB,CAAC;AAGnD,MAAM,CAAC,KAAK,UAAU,WAAW,CAC/B,KAAoB,EACpB,GAAoB,EACpB,aAAqB;IAErB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,aAAa,EAAE,CAAC,EAAE,EAAE,CAAC;QACvC,MAAM,MAAM,GAAG,QAAQ,CAAC,GAAG,CAAC,QAAQ,EAAE,GAAG,CAAC,SAAS,CAAC,CAAC;QAErD,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;YAClB,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;QAC3B,CAAC;QAED,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,GAAG,kBAAkB,CAAC;YAC5C,aAAa,EAAE,GAAG,CAAC,aAAa,IAAI,EAAE;YACtC,QAAQ,EAAE,GAAG,CAAC,QAAQ,IAAI,EAAE;YAC5B,QAAQ,EAAE,MAAM,CAAC,QAAQ;YACzB,UAAU,EAAE,MAAM,CAAC,MAAM;SAC1B,CAAC,CAAC;QAEH,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM,YAAY,CAAC,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC;QAC/D,GAAG,CAAC,QAAQ,EAAE,CAAC;QACf,GAAG,CAAC,aAAa,EAAE,CAAC;QACpB,GAAG,CAAC,iBAAiB,EAAE,CAAC;QAExB,MAAM,MAAM,GAAG,iBAAiB,CAAC,IAAI,CAAC,CAAC;QACvC,IAAI,MAAM,CAAC,MAAM,IAAI,CAAC,EAAE,CAAC;YACvB,GAAG,CAAC,aAAa,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC;YAC9B,GAAG,CAAC,QAAQ,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC;YACzB,MAAM,SAAS,CAAC,GAAG,CAAC,aAAa,EAAE,MAAM,CAAC,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC;YACvD,MAAM,SAAS,CAAC,GAAG,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC;QACpD,CAAC;aAAM,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC/B,GAAG,CAAC,aAAa,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC;YAC9B,MAAM,SAAS,CAAC,GAAG,CAAC,aAAa,EAAE,MAAM,CAAC,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC;QACzD,CAAC;IACH,CAAC;IAED,MAAM,WAAW,GAAG,QAAQ,CAAC,GAAG,CAAC,QAAQ,EAAE,GAAG,CAAC,SAAS,CAAC,CAAC;IAC1D,IAAI,WAAW,CAAC,MAAM,EAAE,CAAC;QACvB,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;IAC3B,CAAC;IAED,GAAG,CAAC,aAAa,GAAG,MAAM,QAAQ,CAAC,GAAG,CAAC,aAAa,EAAE,OAAO,CAAC,CAAC;IAC/D,GAAG,CAAC,QAAQ,GAAG,MAAM,QAAQ,CAAC,GAAG,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;IAErD,OAAO;QACL,OAAO,EAAE,KAAK;QACd,KAAK,EAAE,8BAA8B,aAAa,gBAAgB,WAAW,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE;KACpG,CAAC;AACJ,CAAC"}

package/dist/src/pipeline/steps/test-fix-loop.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=test-fix-loop.test.d.ts.map

package/dist/src/pipeline/steps/test-fix-loop.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"test-fix-loop.test.d.ts","sourceRoot":"","sources":["../../../../src/pipeline/steps/test-fix-loop.test.ts"],"names":[],"mappings":""}