@empiricalrun/test-gen 0.73.1 → 0.74.0

package/dist/agent/codegen/generate-code-apply-changes.js

@@ -1,379 +0,0 @@
-"use strict";
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.systemPromptBuilderForRepoEdit = systemPromptBuilderForRepoEdit;
-exports.generateCodeAndApplyChanges = generateCodeAndApplyChanges;
-const llm_1 = require("@empiricalrun/llm");
-const fs_1 = __importDefault(require("fs"));
-const path_1 = require("path");
-const constants_1 = require("../../constants");
-const types_1 = require("./types");
-const utils_1 = require("./utils");
-function getCodeEditorToolCalls() {
-    const strReplace = {
-        name: "code-block-replace",
-        description: "replace the code block with the provided code",
-        schema: {
-            type: "function",
-            function: {
-                name: types_1.CodeEditorToolCall.STR_REPLACE,
-                description: "replace the code block with the provided code",
-                parameters: {
-                    type: "object",
-                    properties: {
-                        reason: {
-                            type: "string",
-                            description: "reason for why the current block is selected for replacement and why is it needed.",
-                        },
-                        filePath: {
-                            type: "string",
-                            description: "path of the file where the code block is present",
-                        },
-                        oldCode: {
-                            type: "string",
-                            description: "old code block to be replaced",
-                        },
-                        newCode: {
-                            type: "string",
-                            description: "new code block to replace the old code block",
-                        },
-                    },
-                    required: ["reason", "filePath", "oldCode", "newCode"],
-                    additionalProperties: false,
-                },
-            },
-        },
-    };
-    const createFile = {
-        name: "create_file",
-        description: "create a new file with the provided code if the file doesn't exist",
-        schema: {
-            type: "function",
-            function: {
-                name: types_1.CodeEditorToolCall.CREATE_FILE,
-                description: "create a new file with the provided code if the file doesn't exist",
-                parameters: {
-                    type: "object",
-                    properties: {
-                        reason: {
-                            type: "string",
-                            description: "reason for why the file is needed and what is the purpose of the file.",
-                        },
-                        filePath: {
-                            type: "string",
-                            description: "path of the file that needs to be created",
-                        },
-                        code: {
-                            type: "string",
-                            description: "code to be written to the file",
-                        },
-                    },
-                    required: ["reason", "filePath", "code"],
-                    additionalProperties: false,
-                },
-            },
-        },
-    };
-    return [strReplace, createFile].map((tool) => tool.schema);
-}
-function getPlannerToolCalls() {
-    const planner = {
-        name: "change_plan",
-        description: "Provides the plan for the changes to be made",
-        schema: {
-            type: "function",
-            function: {
-                name: "change_plan",
-                description: "Provides the plan for the changes to be made",
-                parameters: {
-                    type: "object",
-                    properties: {
-                        reason: {
-                            type: "string",
-                            description: "Reason for providing the plan. Why a certain file needed the change ? Are you sure that all the changes required are present in the plan ?",
-                        },
-                        plan: {
-                            type: "string",
-                            description: "Plan given to code agent to execute. The plan should be readable and easy to understand.",
-                        },
-                    },
-                    additionalProperties: false,
-                    required: ["reason", "plan"],
-                },
-            },
-        },
-    };
-    const exit = {
-        name: "exit",
-        description: "Called when there are no more changes to be made. All the changes are done.",
-        schema: {
-            type: "function",
-            function: {
-                name: "exit",
-                description: "Called when there are no more changes to be made. All the changes are done",
-                parameters: {
-                    type: "object",
-                    properties: {
-                        reason: {
-                            type: "string",
-                            description: "Reason for marking the task as done.",
-                        },
-                    },
-                    additionalProperties: false,
-                },
-            },
-        },
-    };
-    return [planner, exit].map((tool) => tool.schema);
-}
-async function getPlanForCodeEditorAgent({ prompt, trace, }) {
-    const promptSpan = trace?.span({
-        name: "repo-edit-planner-agent",
-        input: {
-            prompt,
-        },
-    });
-    const llm = new llm_1.LLM({
-        trace: promptSpan,
-        provider: "openai",
-        defaultModel: "o3-mini-2025-01-31",
-        providerApiKey: constants_1.MODEL_API_KEYS["openai"],
-    });
-    const completion = await llm.createChatCompletion({
-        messages: prompt,
-        modelParameters: {
-            max_completion_tokens: constants_1.DEFAULT_O1_MODEL_PARAMETERS.max_completion_tokens,
-            reasoning_effort: "high",
-            tool_choice: "required",
-        },
-        trace: promptSpan,
-        tools: getPlannerToolCalls(),
-    });
-    promptSpan?.end({ output: { completion } });
-    if (!completion?.tool_calls || completion?.tool_calls?.length === 0) {
-        return;
-    }
-    if (completion.tool_calls[0].function.name === "exit") {
-        return;
-    }
-    if (completion.tool_calls[0].function.name === "change_plan") {
-        const args = completion.tool_calls[0].function.arguments;
-        const plan = JSON.parse(args).plan;
-        return plan;
-    }
-}
-function systemPromptBuilderForRepoEdit(files) {
-    const prompt = `
-You are an expert QA automation engineer well versed in Playwright and Typescript and is given an objective to update tests in the repository on the basis of task provided. 
-
-Directory structure for the repository:
-└── /
-  ├── tests/
-  ├────test-data/
-  │    └── index.ts
-  ├── pages/
-  ├── playwright.config.ts
-  ├── app_knowledge.md
-  ├── .eslintrc.js
-  ├── package.json
-  └── tsconfig.json
-
-Explanation of repository structure:
-- tests: this is a directory where all tests and fixtures are kept. All spec files here end with ".spec.ts" as naming convention. There is an exception for Playwright fixtures file. Fixtures file is named as "fixtures.ts"
-- test-data: this is a directory where all the test data are kept.
-- pages: this is a directory where all reusable page object model methods are kept. Page object model methods are written in a pure functional convention. These methods are reusable methods created to be used across tests. This is also a directory where all the utility methods are kept which can be executed independent of the Playwright tests.
-
-Coding principles and guidelines:
-
-## Code Style and Structure
-- Write concise, maintainable, and technically accurate TypeScript code with relevant examples.
-- Use functional and declarative programming patterns; avoid classes.
-- Use types to describe the shape of data and the behavior of functions.
-- Favour iteration and modularisation to adhere to DRY principles and avoid code duplication.
-- Use descriptive variable names with auxiliary verbs (e.g., isLoading, hasError).
-- Organise files systematically: each file should contain only related content, such as exported components, subcomponents, helpers, static content, and types.
-- Page object models are pure functions and are stateless.
-
-## Naming Conventions
-- Use lowercase with dashes for directories (e.g., tests/auth-wizard).
-- Favour named exports for functions.
-
-## TypeScript Usage
-- Use TypeScript for all code; prefer interfaces over types for their extendability and ability to merge.
-- Avoid enums; use maps instead for better type safety and flexibility.
-- Use functional components with TypeScript interfaces.
-
-## Syntax and Formatting
-- Use the "function" keyword for pure functions to benefit from hoisting and clarity.
-
-You will also be provided with current test files, fixtures and page object models for you to use and update code as per the task provided to you.
-
-Here is the list of the files in the repository:
-${files}
-`;
-    return prompt;
-}
-function userPromptBuilderForStrReplace(plan) {
-    const prompt = `
-You need to work to achieve the plan provided to you.
-
-Plan:
-${plan}
-
-In order to execute the plan, FOLLOW BELOW STEPS:
-- First go through the files in the repository and understand the code and the dependencies.
-- Read the plan and figure out what are the changes that needs to be made.
-- List down the changes required to execute the plan.
-- Pick each change and identify the file path which needs the change, the reason for change and the code change they need.
-- Use 'create_file' tool to create a new file if the file doesn't exist.
-- Use separate 'str_replace' tool to make the changes for each update, even if the changes are in the same file.
-- You can provide multiple 'str_replace' tool calls in the response.
-- After making the changes check if the changes are made to all the files if not apply the changes to the remaining files.
-- Also ensure that no extra changes are made to the code.
-
-Notes for using the 'str_replace' command:
-* The 'oldCode' parameter should match EXACTLY one or more consecutive lines from the original file. Be mindful of whitespaces!
-* If the 'oldCode' parameter is not unique in the file, the replacement will not be performed. Make sure to include enough context in 'oldCode' to make it unique
-* The 'newCode' parameter should contain the edited lines that should replace the 'oldCode'
-
-Notes for using the 'create_file' command:
-* The 'create_file' command cannot be used if the specified 'filePath' already exists as a file.
-`;
-    return prompt;
-}
-function userPromptBuilderForPlanner(task) {
-    return `
-You need to work towards completing the task provided to you.
-
-Task: 
-${task}
-
-In order to execute the task, FOLLOW BELOW STEPS:
-- First go through the files in the repository and understand the code and the dependencies.
-- Read the task and figure out what are the changes that needs to be made.
-- List down the changes required for the given task.
-- If there are no changes to be made, respond with "exit" tool call to exit the conversation.
-- Craft a prompt for the agent to make the changes to the code.
-- The plan should be readable and easy to understand.
-- The plan should contain the details of the changes to be made such as file path, test name etc.
-- If there are multiple changes in the same file, mention those separately in the plan.
-`;
-}
-function deDupUpdatedFiles(updatedFiles) {
-    return updatedFiles.filter((change, index, self) => index ===
-        self.findIndex((existing) => existing.filePath === change.filePath));
-}
-async function generateCodeAndApplyChanges({ task, trace, logger, getRelevantFiles, }) {
-    // Reducing this from 5 to 3, if the required changes are getting missed will change it back.
-    let planRetries = 3;
-    let updatedFiles = [];
-    while (planRetries--) {
-        const generateCodeAndApplyChangesSpan = trace?.span({
-            name: "generate-code-apply-changes",
-            input: {
-                task,
-            },
-        });
-        const { prompt: files } = await getRelevantFiles();
-        if (!files) {
-            return deDupUpdatedFiles(updatedFiles);
-        }
-        let strReplacePlan = await getPlanForCodeEditorAgent({
-            prompt: [
-                {
-                    role: "system",
-                    content: systemPromptBuilderForRepoEdit(files),
-                },
-                {
-                    role: "user",
-                    content: userPromptBuilderForPlanner(task),
-                },
-            ],
-            trace: generateCodeAndApplyChangesSpan,
-            logger,
-        });
-        if (!strReplacePlan) {
-            await (0, llm_1.flushAllTraces)();
-            return deDupUpdatedFiles(updatedFiles);
-        }
-        const promptForStrReplace = [
-            {
-                role: "system",
-                content: systemPromptBuilderForRepoEdit(files),
-            },
-            {
-                role: "user",
-                content: userPromptBuilderForStrReplace(strReplacePlan),
-            },
-        ];
-        const codeEditorSpan = generateCodeAndApplyChangesSpan?.span({
-            name: "code-editor-agent",
-            input: {
-                prompt: promptForStrReplace,
-            },
-        });
-        const llm = new llm_1.LLM({
-            trace: codeEditorSpan,
-            provider: "anthropic",
-            defaultModel: "claude-3-5-sonnet-20240620",
-            providerApiKey: constants_1.MODEL_API_KEYS["anthropic"],
-        });
-        const completion = await llm.createChatCompletion({
-            messages: promptForStrReplace,
-            modelParameters: {
-                ...constants_1.DEFAULT_MODEL_PARAMETERS,
-                temperature: 0.1,
-                tool_choice: "required",
-            },
-            trace: codeEditorSpan,
-            tools: getCodeEditorToolCalls(),
-        });
-        codeEditorSpan?.end({ output: { completion } });
-        if (!completion?.tool_calls || completion?.tool_calls?.length === 0) {
-            continue;
-        }
-        // Filter out the tool calls which are for creating new files
-        const createFileToolCalls = completion.tool_calls.filter((tc) => tc.function.name === types_1.CodeEditorToolCall.CREATE_FILE);
-        if (createFileToolCalls.length > 0) {
-            console.log(`create_file tool calls: `, createFileToolCalls);
-        }
-        await Promise.all(createFileToolCalls.map((tc) => {
-            return (async () => {
-                const args = JSON.parse(tc.function.arguments);
-                updatedFiles.push({
-                    filePath: args.filePath,
-                    oldCode: "",
-                    newCode: args.code,
-                    reason: args.reason,
-                });
-                fs_1.default.mkdirSync((0, path_1.dirname)(args.filePath), { recursive: true });
-                fs_1.default.writeFileSync(args.filePath, args.code, "utf-8");
-                console.log(`Created file: ${args.filePath}`);
-            })();
-        }));
-        const strReplaceToolCalls = completion.tool_calls.filter((tc) => tc.function.name === types_1.CodeEditorToolCall.STR_REPLACE);
-        if (strReplaceToolCalls.length > 0) {
-            console.log(`str_replace tool calls: `, strReplaceToolCalls);
-        }
-        // Filter out the tool calls which are for replacing code in existing files
-        const fileChanges = strReplaceToolCalls
-            .map((toolCall) => JSON.parse(toolCall.function.arguments))
-            .filter((f) => f.filePath && fs_1.default.existsSync(f.filePath));
-        // We add all the suggested changes to the updatedFiles array
-        // This is used to validate and format files later
-        updatedFiles.push(...fileChanges);
-        // applyChangesResponse contains the errors occurred while applying the changes
-        const updates = await (0, utils_1.applyFileChangesUsingStrReplace)({
-            trace: codeEditorSpan,
-            fileChanges,
-            logger,
-        });
-        const failedCodeUpdates = updates.filter((f) => f?.error);
-        if (failedCodeUpdates.length > 0) {
-            logger?.log(`Failed to apply changes, retrying...`, failedCodeUpdates);
-        }
-    }
-    return deDupUpdatedFiles(updatedFiles);
-}

package/dist/agent/codegen/repo-edit.d.ts

@@ -1,23 +0,0 @@
-import { TraceClient } from "@empiricalrun/llm";
-import { ChatCompletionMessageParam } from "openai/resources/index.mjs";
-import { CustomLogger } from "../../bin/logger";
-import { CodeUpdate } from "./types";
-export declare function generateCodeUsingRepoAgent({ task, trace, repoFiles, logger, }: {
-    trace?: TraceClient;
-    task: string;
-    repoFiles: string;
-    logger?: CustomLogger;
-}): Promise<{
-    prompt: ChatCompletionMessageParam[];
-    agentResponse: string;
-    fileChanges: CodeUpdate[];
-}>;
-export declare function repoEditAgent({ trace, task, logger, }: {
-    trace?: TraceClient;
-    task: string;
-    logger?: CustomLogger;
-}): Promise<{
-    prompt: ChatCompletionMessageParam[];
-    fileChanges: CodeUpdate[];
-}>;
-//# sourceMappingURL=repo-edit.d.ts.map

package/dist/agent/codegen/repo-edit.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"repo-edit.d.ts","sourceRoot":"","sources":["../../../src/agent/codegen/repo-edit.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAGhD,OAAO,EAAE,0BAA0B,EAAE,MAAM,4BAA4B,CAAC;AAExE,OAAO,EAAE,YAAY,EAAE,MAAM,kBAAkB,CAAC;AAOhD,OAAO,EAAE,UAAU,EAAE,MAAM,SAAS,CAAC;AAGrC,wBAAsB,0BAA0B,CAAC,EAC/C,IAAI,EACJ,KAAK,EACL,SAAS,EAET,MAAM,GACP,EAAE;IACD,KAAK,CAAC,EAAE,WAAW,CAAC;IACpB,IAAI,EAAE,MAAM,CAAC;IACb,SAAS,EAAE,MAAM,CAAC;IAClB,MAAM,CAAC,EAAE,YAAY,CAAC;CACvB,GAAG,OAAO,CAAC;IACV,MAAM,EAAE,0BAA0B,EAAE,CAAC;IACrC,aAAa,EAAE,MAAM,CAAC;IACtB,WAAW,EAAE,UAAU,EAAE,CAAC;CAC3B,CAAC,CA6CD;AAED,wBAAsB,aAAa,CAAC,EAClC,KAAK,EACL,IAAI,EACJ,MAAM,GACP,EAAE;IACD,KAAK,CAAC,EAAE,WAAW,CAAC;IACpB,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,CAAC,EAAE,YAAY,CAAC;CACvB,GAAG,OAAO,CAAC;IACV,MAAM,EAAE,0BAA0B,EAAE,CAAC;IACrC,WAAW,EAAE,UAAU,EAAE,CAAC;CAC3B,CAAC,CA8BD"}

package/dist/agent/codegen/repo-edit.js

@@ -1,81 +0,0 @@
-"use strict";
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.generateCodeUsingRepoAgent = generateCodeUsingRepoAgent;
-exports.repoEditAgent = repoEditAgent;
-const fs_1 = __importDefault(require("fs"));
-const context_1 = require("../../bin/utils/context");
-const web_1 = require("../../bin/utils/platform/web");
-const generate_code_apply_changes_1 = require("./generate-code-apply-changes");
-const utils_1 = require("./utils");
-async function generateCodeUsingRepoAgent({ task, trace, repoFiles, logger, }) {
-    const repoEditSpan = trace?.span({
-        name: "repo-edit",
-        input: { task },
-    });
-    const prompt = [
-        {
-            role: "system",
-            content: (0, generate_code_apply_changes_1.systemPromptBuilderForRepoEdit)(repoFiles),
-        },
-        {
-            role: "user",
-            content: `
-You need to work towards completing the task provided to you.
-
-Task: 
-${task}
-
-In order to execute the task, FOLLOW BELOW STEPS:
-- First go through the files in the repository and understand the code and the dependencies.
-- Read the task and figure out what are the changes that needs to be made.
-- List down the changes required for the given task.
-- Create a clear, detailed plan which will be used as a prompt for the code change agent. Include specifics like file paths and test names. If there are multiple changes in one file, list them separately.
-- The plan should be readable and easy to understand.
-
-If all the changes are done, respond with "exit" tool call otherwise respond with "change-plan" tool call.
-  `,
-        },
-    ];
-    const fileChanges = await (0, generate_code_apply_changes_1.generateCodeAndApplyChanges)({
-        task,
-        trace: repoEditSpan,
-        getRelevantFiles: async () => await (0, context_1.generateTxtForRepository)(),
-        logger,
-    });
-    repoEditSpan?.end({
-        output: { fileChanges },
-    });
-    return {
-        prompt,
-        fileChanges,
-        agentResponse: "",
-    };
-}
-async function repoEditAgent({ trace, task, logger, }) {
-    logger?.log(`Starting repo agent: [trace](${trace?.getTraceUrl()})`);
-    const { prompt: repoFiles } = await (0, context_1.generateTxtForRepository)();
-    const repoAgentOutput = await generateCodeUsingRepoAgent({
-        task,
-        trace,
-        repoFiles: repoFiles,
-        logger,
-    });
-    const updates = repoAgentOutput.fileChanges;
-    await (0, utils_1.validateTypesAndFormatCode)({
-        validateTypes: true,
-        trace,
-        testCase: {},
-        filePaths: updates.map((f) => f.filePath),
-    });
-    await Promise.all(updates.map((f) => {
-        return (async () => {
-            if (fs_1.default.existsSync(f.filePath)) {
-                await (0, web_1.lintErrors)(f.filePath);
-            }
-        })();
-    }));
-    return repoAgentOutput;
-}

package/dist/agent/codegen/run.d.ts

@@ -1,20 +0,0 @@
-import { TraceClient } from "@empiricalrun/llm";
-import type { TestCase } from "@empiricalrun/shared-types";
-import OpenAI from "openai";
-import { CreateTestCodeUpdate } from "./types";
-export declare function createTestWithCodeAgent({ testCase, file, repoFiles, trace, }: {
-    testCase: TestCase;
-    file: string;
-    repoFiles: string | undefined;
-    trace?: TraceClient;
-}): Promise<{
-    prompt: OpenAI.Chat.Completions.ChatCompletionMessageParam[];
-    agentResponse: string;
-    fileChanges: CreateTestCodeUpdate[];
-}>;
-export declare function generateTestWithCodegen({ testCase, file, trace, }: {
-    testCase: TestCase;
-    file: string;
-    trace?: TraceClient;
-}): Promise<TestCase[] | void>;
-//# sourceMappingURL=run.d.ts.map

package/dist/agent/codegen/run.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"run.d.ts","sourceRoot":"","sources":["../../../src/agent/codegen/run.ts"],"names":[],"mappings":"AAAA,OAAO,EAAsB,WAAW,EAAE,MAAM,mBAAmB,CAAC;AACpE,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,4BAA4B,CAAC;AAE3D,OAAO,MAAM,MAAM,QAAQ,CAAC;AAa5B,OAAO,EAAE,oBAAoB,EAAE,MAAM,SAAS,CAAC;AAQ/C,wBAAsB,uBAAuB,CAAC,EAC5C,QAAQ,EACR,IAAI,EACJ,SAAS,EACT,KAAK,GACN,EAAE;IACD,QAAQ,EAAE,QAAQ,CAAC;IACnB,IAAI,EAAE,MAAM,CAAC;IACb,SAAS,EAAE,MAAM,GAAG,SAAS,CAAC;IAC9B,KAAK,CAAC,EAAE,WAAW,CAAC;CACrB,GAAG,OAAO,CAAC;IACV,MAAM,EAAE,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,0BAA0B,EAAE,CAAC;IAC7D,aAAa,EAAE,MAAM,CAAC;IACtB,WAAW,EAAE,oBAAoB,EAAE,CAAC;CACrC,CAAC,CAyDD;AAED,wBAAsB,uBAAuB,CAAC,EAC5C,QAAQ,EACR,IAAI,EACJ,KAAK,GACN,EAAE;IACD,QAAQ,EAAE,QAAQ,CAAC;IACnB,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,CAAC,EAAE,WAAW,CAAC;CACrB,GAAG,OAAO,CAAC,QAAQ,EAAE,GAAG,IAAI,CAAC,CA2D7B"}

package/dist/agent/codegen/run.js

@@ -1,116 +0,0 @@
-"use strict";
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.createTestWithCodeAgent = createTestWithCodeAgent;
-exports.generateTestWithCodegen = generateTestWithCodegen;
-const llm_1 = require("@empiricalrun/llm");
-const fs_1 = __importDefault(require("fs"));
-const logger_1 = require("../../bin/logger");
-const context_1 = require("../../bin/utils/context");
-const web_1 = require("../../bin/utils/platform/web");
-const constants_1 = require("../../constants");
-const promptTemplate_0 = "{{#section \"system\"}}\nYou are a software test engineer who is given a task to write test cases in a Playwright test repository.\nYou will be provided with current test files, fixtures and page object models for you to use and update code as per the\ntask provided to you.\nYou need to respond with file path, the new test code block to be updated in the file and the reason for the change.\n\nHere is the list of files:\n\n{{repoFiles}}\n{{/section}}\n\n{{#section \"user\"}}\nTask: {{task}}\n\n## Follow these instructions before responding with output:\n- Scan the file '{{testFilePath}}' and check whether test case with name '{{testCaseName}}' exists in the file\n- If it exists do not update the file\n- else respond with <file_path></file_path>, <code_block></code_block> and <change></change> as xml tags\n- Only respond with single code_block in the response containing only the test code block to be updated\n{{/section}}";
-const update_flow_1 = require("./update-flow");
-const utils_1 = require("./utils");
-async function createTestWithCodeAgent({ testCase, file, repoFiles, trace, }) {
-    const task = `
-Create a new test case with name '${testCase.name}' at file path '${file}'
-${testCase.suites.length
-        ? `
-Put this test case inside describe blocks named: ${testCase.suites.join(" > ")}`
-        : ""}
-
-In the above test, write code that performs these actions and ONLY these actions:
-${testCase.steps.join("\n")}`;
-    const prompt = (0, llm_1.compilePrompt)(promptTemplate_0, {
-        task,
-        repoFiles,
-        testCaseName: testCase.name,
-        testFilePath: file,
-    });
-    const createTestSpan = trace?.span({
-        name: "create-test",
-        input: {
-            testCase,
-            repoFiles,
-            testFilePath: file,
-            task,
-        },
-    });
-    const llm = new llm_1.LLM({
-        trace,
-        provider: "anthropic",
-        defaultModel: "claude-3-5-sonnet-20240620",
-        providerApiKey: constants_1.MODEL_API_KEYS["anthropic"],
-    });
-    const updatedUsageExampleMessage = await llm.createChatCompletion({
-        messages: prompt,
-        trace,
-        traceName: "create-test-llm",
-        modelParameters: {
-            ...constants_1.DEFAULT_MODEL_PARAMETERS,
-        },
-    });
-    const updates = (0, utils_1.extractCreateTestUpdates)(updatedUsageExampleMessage?.content);
-    createTestSpan?.end({
-        output: { updates },
-    });
-    console.log("Updates: ", updates);
-    return {
-        prompt,
-        agentResponse: updatedUsageExampleMessage?.content,
-        fileChanges: updates,
-    };
-}
-async function generateTestWithCodegen({ testCase, file, trace, }) {
-    const logger = new logger_1.CustomLogger();
-    if (!fs_1.default.existsSync(file)) {
-        logger.log(`Creating a new spec file: ${file}`);
-        fs_1.default.writeFileSync(file, "");
-    }
-    const context = await (0, context_1.contextForGeneration)(file);
-    const { testFileContent } = context;
-    const { testBlock } = (0, web_1.getTypescriptTestBlock)({
-        scenarioName: testCase?.name,
-        content: testFileContent,
-        suites: testCase?.suites,
-    });
-    // Update the test if it already exists
-    const isUpdate = !!testBlock;
-    if (isUpdate) {
-        logger.log(`Updating the test '${testCase.name}': [View trace](${trace?.getTraceUrl()})`);
-        return await (0, update_flow_1.updateTest)({
-            testCase,
-            file,
-            trace,
-        });
-    }
-    logger?.log(`Creating a new test case '${testCase.name}': [trace](${trace?.getTraceUrl()})`);
-    // Create the test if it doesn't exist
-    const { prompt: repoFiles } = await (0, context_1.generateTxtForRepository)();
-    const updates = await createTestWithCodeAgent({
-        testCase,
-        file,
-        repoFiles,
-        trace,
-    });
-    await (0, utils_1.applyFileChangesForCreateTest)({
-        trace,
-        fileChanges: updates.fileChanges,
-    });
-    await (0, utils_1.validateTypesAndFormatCode)({
-        validateTypes: true,
-        trace,
-        testCase: {},
-        filePaths: updates.fileChanges.map((f) => f.filePath),
-    });
-    await Promise.all(updates.fileChanges.map((f) => {
-        return (async () => {
-            if (fs_1.default.existsSync(f.filePath)) {
-                await (0, web_1.lintErrors)(f.filePath);
-            }
-        })();
-    }));
-}

package/dist/agent/enrich-prompt/index.d.ts

@@ -1,12 +0,0 @@
-import { TraceClient } from "@empiricalrun/llm";
-export declare const enrichPromptWithFailingLine: ({ trace, testBlock, testFilePath, suggestionForFix, }: {
-    testBlock: string;
-    testFilePath: string;
-    suggestionForFix: string;
-    trace?: TraceClient;
-}) => Promise<{
-    output: string;
-    is_user_message_enriched: boolean;
-    reason_for_output: string;
-}>;
-//# sourceMappingURL=index.d.ts.map

package/dist/agent/enrich-prompt/index.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/agent/enrich-prompt/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAsB,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAmCpE,eAAO,MAAM,2BAA2B,GAAU,uDAK/C;IACD,SAAS,EAAE,MAAM,CAAC;IAClB,YAAY,EAAE,MAAM,CAAC;IACrB,gBAAgB,EAAE,MAAM,CAAC;IACzB,KAAK,CAAC,EAAE,WAAW,CAAC;CACrB,KAAG,OAAO,CAAC;IACV,MAAM,EAAE,MAAM,CAAC;IACf,wBAAwB,EAAE,OAAO,CAAC;IAClC,iBAAiB,EAAE,MAAM,CAAC;CAC3B,CAwDA,CAAC"}