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

package/src/pipeline/steps/fix-from-feedback.ts

@@ -1,105 +0,0 @@
-import { readFile, writeFile } from 'node:fs/promises';
-import type { LanguageModel } from 'ai';
-import { generateText } from 'ai';
-import { extractCodeBlocks } from '../../extract-code-block';
-import { buildLintFixPrompt, buildTestFixPrompt, buildTypeFixPrompt } from '../../prompt';
-import type { PipelineContext, StepResult } from '../run-pipeline';
-
-type ErrorCategory = 'type' | 'test' | 'lint' | 'mixed';
-
-function categorizeErrors(errorFeedback: string): ErrorCategory {
-  const hasTypeError = /error TS\d+|type.*error|typescript/i.test(errorFeedback);
-  const hasTestFailure = /test.*fail|vitest|expect.*to|assertion/i.test(errorFeedback);
-  const hasLintError = /lint|biome|eslint/i.test(errorFeedback);
-
-  const categories = [hasTypeError, hasTestFailure, hasLintError].filter(Boolean).length;
-
-  if (categories > 1) return 'mixed';
-  if (hasTypeError) return 'type';
-  if (hasTestFailure) return 'test';
-  if (hasLintError) return 'lint';
-
-  // Default to test errors if we can't categorize
-  return 'test';
-}
-
-export async function fixFromFeedbackStep(model: LanguageModel, ctx: PipelineContext): Promise<StepResult> {
-  if (!ctx.errorFeedback) {
-    return { success: true };
-  }
-
-  // Read existing files
-  const componentCode = await readFile(ctx.componentPath, 'utf-8');
-  const testCode = await readFile(ctx.testPath, 'utf-8');
-
-  ctx.componentCode = componentCode;
-  ctx.testCode = testCode;
-
-  const errorCategory = categorizeErrors(ctx.errorFeedback);
-
-  let system: string;
-  let prompt: string;
-
-  switch (errorCategory) {
-    case 'type': {
-      const typePrompt = buildTypeFixPrompt({
-        componentCode,
-        testCode,
-        errors: [ctx.errorFeedback],
-      });
-      system = typePrompt.system;
-      prompt = typePrompt.prompt;
-      break;
-    }
-    case 'test': {
-      const testPrompt = buildTestFixPrompt({
-        componentCode,
-        testCode,
-        failures: [ctx.errorFeedback],
-        testOutput: ctx.errorFeedback,
-      });
-      system = testPrompt.system;
-      prompt = testPrompt.prompt;
-      break;
-    }
-    case 'lint': {
-      const lintPrompt = buildLintFixPrompt({
-        componentCode,
-        testCode,
-        errors: [ctx.errorFeedback],
-      });
-      system = lintPrompt.system;
-      prompt = lintPrompt.prompt;
-      break;
-    }
-    case 'mixed':
-    default: {
-      // For mixed errors, use test fix prompt (most comprehensive)
-      const mixedPrompt = buildTestFixPrompt({
-        componentCode,
-        testCode,
-        failures: [ctx.errorFeedback],
-        testOutput: ctx.errorFeedback,
-      });
-      system = mixedPrompt.system;
-      prompt = mixedPrompt.prompt;
-      break;
-    }
-  }
-
-  const { text } = await generateText({ model, system, prompt });
-  ctx.llmCalls++;
-
-  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');
-  }
-
-  return { success: true };
-}