@empiricalrun/test-gen 0.42.28 → 0.43.0

package/CHANGELOG.md

@@ -1,5 +1,22 @@
 # @empiricalrun/test-gen
 
+## 0.43.0
+
+### Minor Changes
+
+- 46c0dab: feat: autofix workflow v1
+
+### Patch Changes
+
+- Updated dependencies [46c0dab]
+  - @empiricalrun/llm@0.9.35
+
+## 0.42.29
+
+### Patch Changes
+
+- f39500b: fix: removed explicit delays from master agent
+
 ## 0.42.28
 
 ### Patch Changes

package/dist/actions/click.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"click.d.ts","sourceRoot":"","sources":["../../src/actions/click.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,yBAAyB,EAAE,MAAM,UAAU,CAAC;AAIrD,eAAO,MAAM,4BAA4B,kBAAkB,CAAC;AAE5D,eAAO,MAAM,oBAAoB,EAAE,yBAuDlC,CAAC"}
+{"version":3,"file":"click.d.ts","sourceRoot":"","sources":["../../src/actions/click.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,yBAAyB,EAAE,MAAM,UAAU,CAAC;AAIrD,eAAO,MAAM,4BAA4B,kBAAkB,CAAC;AAE5D,eAAO,MAAM,oBAAoB,EAAE,yBAsDlC,CAAC"}

package/dist/actions/click.js

@@ -11,7 +11,6 @@ const clickActionGenerator = (page) => {
             const locator = await (0, utils_1.getPlaywrightLocatorUsingCssSelector)(selector, args.xpath, page.pwPageInstance, args?.elementAnnotation);
             const exec = new Function("page", `return page.${locator}.click({ timeout: 3000 })`);
             await exec(page.pwPageInstance);
-            await page.pwPageInstance.waitForTimeout(3000);
             return {
                 locator,
             };

package/dist/agent/diagnosis-agent/index.d.ts

@@ -0,0 +1,18 @@
+import { TraceClient } from "@empiricalrun/llm";
+import { TestErrorDiagnosisDetails, TestGenConfigOptions } from "@empiricalrun/shared-types";
+import { CustomLogger } from "../../bin/logger";
+/**
+ *
+ * inputs
+ * - task
+ * - diagnosis
+ */
+export declare function createTaskUsingFailureDiagnosis({ options, trace, diagnosis, logger, }: {
+    options?: TestGenConfigOptions;
+    trace?: TraceClient;
+    diagnosis: TestErrorDiagnosisDetails;
+    logger?: CustomLogger;
+}): Promise<{
+    task: string;
+}>;
+//# sourceMappingURL=index.d.ts.map

package/dist/agent/diagnosis-agent/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/agent/diagnosis-agent/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAyB,WAAW,EAAE,MAAM,mBAAmB,CAAC;AACvE,OAAO,EACL,yBAAyB,EACzB,oBAAoB,EACrB,MAAM,4BAA4B,CAAC;AAEpC,OAAO,EAAE,YAAY,EAAE,MAAM,kBAAkB,CAAC;AAuChD;;;;;GAKG;AACH,wBAAsB,+BAA+B,CAAC,EACpD,OAAO,EACP,KAAK,EACL,SAAS,EACT,MAAM,GACP,EAAE;IACD,OAAO,CAAC,EAAE,oBAAoB,CAAC;IAC/B,KAAK,CAAC,EAAE,WAAW,CAAC;IACpB,SAAS,EAAE,yBAAyB,CAAC;IACrC,MAAM,CAAC,EAAE,YAAY,CAAC;CACvB,GAAG,OAAO,CAAC;IAAE,IAAI,EAAE,MAAM,CAAA;CAAE,CAAC,CA8E5B"}

package/dist/agent/diagnosis-agent/index.js

@@ -0,0 +1,105 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.createTaskUsingFailureDiagnosis = void 0;
+const llm_1 = require("@empiricalrun/llm");
+const session_1 = require("../../session");
+const strict_mode_violation_1 = require("./strict-mode-violation");
+const session = (0, session_1.getSessionDetails)();
+const responseFormat = {
+    type: "json_schema",
+    json_schema: {
+        name: "test-case-auto-fix-summary",
+        strict: true,
+        schema: {
+            type: "object",
+            properties: {
+                observation: {
+                    type: "array",
+                    items: {
+                        type: "string",
+                    },
+                    description: "Detailed observation of what changed between successful and failed test screenshots",
+                },
+                action: {
+                    type: "string",
+                    description: "Direct action to fix the test in natural language without code snippets or options",
+                },
+            },
+            required: ["observation", "action"],
+            additionalProperties: false,
+        },
+    },
+};
+/**
+ *
+ * inputs
+ * - task
+ * - diagnosis
+ */
+async function createTaskUsingFailureDiagnosis({ options, trace, diagnosis, logger, }) {
+    trace =
+        trace ||
+            llm_1.langfuseInstance?.trace({
+                name: "infer-agent-task",
+                id: crypto.randomUUID(),
+                release: session.version,
+            });
+    const failureDiagnosisSpan = trace?.span({
+        name: "auto-fix",
+        input: {
+            diagnosisId: diagnosis.diagnosisId,
+            prjRepoName: options?.metadata.projectRepoName,
+        },
+    });
+    logger?.log("Trying to fix the test using failure diagnosis. Fetching key moments of the diagnosis");
+    const resp = await fetch(diagnosis.keyMomentsUrl);
+    // TODO: check for response to be not ok
+    if (resp.ok) {
+        logger?.success("Successfully fetched key moments of the diagnosis");
+    }
+    else {
+        logger?.warn("Failed to fetch key moments of the diagnosis");
+    }
+    const screenshotsData = await resp.json();
+    const llm = new llm_1.LLM({
+        provider: "openai",
+        defaultModel: "o1",
+        trace,
+    });
+    // TODO: make this dynamic in nature. the prompts should be made receipe
+    // which will help to get rid of if else logic
+    // receipe to have:
+    // 1. selection criteria
+    // 2. job to be done - in this case generate a prompt
+    let prompt;
+    if (diagnosis.failed_run_metadata.stack.includes("strict mode violation")) {
+        prompt = (0, strict_mode_violation_1.fixStrictModeViolationPrompt)({
+            screenshotsData,
+            diagnosis,
+        });
+    }
+    if (prompt) {
+        const llmResponse = await llm.createChatCompletion({
+            messages: prompt,
+            modelParameters: {
+                max_completion_tokens: 40000,
+            },
+            responseFormat,
+        });
+        const { observation, action } = JSON.parse(llmResponse?.content);
+        failureDiagnosisSpan?.update({
+            output: {
+                observation,
+                action,
+            },
+        });
+        return {
+            task: action,
+        };
+    }
+    // TODO: handle default prompt
+    return {
+        task: "",
+    };
+}
+exports.createTaskUsingFailureDiagnosis = createTaskUsingFailureDiagnosis;

package/dist/agent/diagnosis-agent/strict-mode-violation.d.ts

@@ -0,0 +1,9 @@
+import { TestErrorDiagnosisDetails } from "@empiricalrun/shared-types";
+export declare function fixStrictModeViolationPrompt({ screenshotsData, diagnosis, }: {
+    screenshotsData: {
+        success: string[];
+        failure: string[];
+    };
+    diagnosis: TestErrorDiagnosisDetails;
+}): import("openai/resources/index.mjs").ChatCompletionMessageParam[];
+//# sourceMappingURL=strict-mode-violation.d.ts.map

package/dist/agent/diagnosis-agent/strict-mode-violation.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"strict-mode-violation.d.ts","sourceRoot":"","sources":["../../../src/agent/diagnosis-agent/strict-mode-violation.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,yBAAyB,EAAE,MAAM,4BAA4B,CAAC;AAoBvE,wBAAgB,4BAA4B,CAAC,EAC3C,eAAe,EACf,SAAS,GACV,EAAE;IACD,eAAe,EAAE;QAAE,OAAO,EAAE,MAAM,EAAE,CAAC;QAAC,OAAO,EAAE,MAAM,EAAE,CAAA;KAAE,CAAC;IAC1D,SAAS,EAAE,yBAAyB,CAAC;CACtC,qEAiBA"}

package/dist/agent/diagnosis-agent/strict-mode-violation.js

@@ -0,0 +1,31 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.fixStrictModeViolationPrompt = void 0;
+const llm_1 = require("@empiricalrun/llm");
+const promptTemplate_0 = "{{#section \"system\"}}\nAs a software engineer, your task is to identify a fix for a failing Playwright test by analyzing screenshots of both the failed and successful test steps.\n\nKey issues for test failures include:\n- Duplicate elements for the same Playwright selector.\n\nInstructions:\n1. Examine the provided successful and failed test screenshots.\n2. Identify the correct element for action based on these observations on the successful test run screenshots.\n3. Sometimes the exact same locator is not available on failed test run screenshot, you need to identify the intent from successful test screenshots and apply that intent in failed test run screenshot to identify the right locator to interact with\n4. Evaluate the playwright selector options provided to you to execute the action. Pick the selector which best matches the intent of the test.\n5. Propose a precise action that addresses the issue.\n\nExample:\n- observation: \n  - Current step failure: await page.getByText(\"Audience\").click()\n  - Two similar buttons named \"Audience\" exist in the failed run screenshots\n  - The successful test run clicked on \"Untracked Audience\"\n  - The failed test run should click on \"Untracked Audience\"\n  - Available locators: await page.getByText(\"Untracked Audience\").click() contain the selector for Untracked Audience\n- action: Replace failing line with await page.getByText(\"Untracked Audience\").click()\n\nYour action should:\n- Be directly actionable and free of ambiguity, as it will guide another LLM to generate code.\n- Be in natural language and not just code snippet.\n- Be verified as feasible on the failure screen before responding.\n- Choose from the provided possible actions that can be executed on the failure screen.\n- Action should adhere to the format mentioned in the example, i.e. it should start with \"Replace the failing line\" and the updated code with replaced selector following it.\n\nEnsure the action is executable based on the failure screen context before providing it.\n{{/section}}\n\n{{#section \"user\"}}\nSuccessful test screenshots\n\n{{images successScreenshots}}\n\nFailed test screenshots\n\n{{images failedScreenshots}}\n\nStep where test failed:\n{{failingLine}}\n\nOptions for Playwright selectors to perform actions on a failed test screen:\n{{selectorOptions}}\n\n{{/section}}\n\n";
+function extractLocatorOptions(errorStack) {
+    // This regex matches a chain of locator API calls following the pattern:
+    // functionName(arguments) optionally chained with .functionName(arguments)
+    const regex = /aka\s+((?:[A-Za-z0-9_]+\([^)]*\)(?:\.[A-Za-z0-9_]+\([^)]*\))*))/g;
+    const options = [];
+    let match;
+    while ((match = regex.exec(errorStack)) !== null) {
+        if (match[1]) {
+            options.push(match[1]);
+        }
+    }
+    return options;
+}
+function fixStrictModeViolationPrompt({ screenshotsData, diagnosis, }) {
+    const compiledPrompt = (0, llm_1.compilePrompt)(promptTemplate_0, {
+        failingLine: diagnosis.failingLine,
+        successScreenshots: screenshotsData.success,
+        failedScreenshots: screenshotsData.failure,
+        selectorOptions: extractLocatorOptions(diagnosis.failed_run_metadata.stack).join("\n"),
+    }, {
+        imageDetail: "high",
+        modelProvider: "openai",
+    });
+    return compiledPrompt;
+}
+exports.fixStrictModeViolationPrompt = fixStrictModeViolationPrompt;

package/dist/agent/master/element-annotation.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"element-annotation.d.ts","sourceRoot":"","sources":["../../../src/agent/master/element-annotation.ts"],"names":[],"mappings":"AAAA,OAAO,EAAiB,GAAG,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AACpE,OAAO,EAAE,IAAI,EAAE,MAAM,YAAY,CAAC;AAQlC,OAAO,EAAE,oBAAoB,EAAE,MAAM,aAAa,CAAC;AAEnD,OAAO,EAAE,UAAU,EAAE,MAAM,qBAAqB,CAAC;AA0DjD,wBAAsB,oBAAoB,CAAC,EACzC,kBAAkB,EAClB,WAAW,EACX,mBAAmB,EACnB,KAAK,EACL,GAAG,EACH,OAAO,EACP,UAAU,GACX,EAAE;IACD,kBAAkB,EAAE,MAAM,CAAC;IAC3B,WAAW,EAAE,MAAM,CAAC;IACpB,mBAAmB,EAAE,MAAM,CAAC;IAC5B,KAAK,CAAC,EAAE,WAAW,CAAC;IACpB,GAAG,CAAC,EAAE,GAAG,CAAC;IACV,OAAO,CAAC,EAAE,oBAAoB,CAAC;IAC/B,UAAU,EAAE,oBAAoB,CAAC;CAClC,GAAG,OAAO,CAAC,MAAM,GAAG,SAAS,CAAC,CA8C9B;AAED,MAAM,MAAM,oBAAoB,GAAG;IACjC,UAAU,EACN,KAAK,GACL,UAAU,CAAC,IAAI,GACf,UAAU,CAAC,WAAW,GACtB,UAAU,CAAC,MAAM,CAAC;IACtB,aAAa,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;CACpC,CAAC;AAEF,wBAAsB,iBAAiB,CAAC,EACtC,IAAI,EACJ,UAAU,EACV,OAAO,GACR,EAAE;IACD,IAAI,EAAE,IAAI,CAAC;IACX,UAAU,EAAE,oBAAoB,CAAC;IACjC,OAAO,EAAE,oBAAoB,CAAC;CAC/B,GAAG,OAAO,CAAC;IACV,cAAc,EAAE;QAAE,SAAS,EAAE,MAAM,CAAC;QAAC,IAAI,EAAE,MAAM,CAAA;KAAE,EAAE,CAAC;IACtD,gBAAgB,EAAE,MAAM,CAAC;IACzB,uBAAuB,EAAE,MAAM,CAAC;CACjC,CAAC,CAqDD"}
+{"version":3,"file":"element-annotation.d.ts","sourceRoot":"","sources":["../../../src/agent/master/element-annotation.ts"],"names":[],"mappings":"AAAA,OAAO,EAAiB,GAAG,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AACpE,OAAO,EAAE,IAAI,EAAE,MAAM,YAAY,CAAC;AAQlC,OAAO,EAAE,oBAAoB,EAAE,MAAM,aAAa,CAAC;AAEnD,OAAO,EAAE,UAAU,EAAE,MAAM,qBAAqB,CAAC;AA0DjD,wBAAsB,oBAAoB,CAAC,EACzC,kBAAkB,EAClB,WAAW,EACX,mBAAmB,EACnB,KAAK,EACL,GAAG,EACH,OAAO,EACP,UAAU,GACX,EAAE;IACD,kBAAkB,EAAE,MAAM,CAAC;IAC3B,WAAW,EAAE,MAAM,CAAC;IACpB,mBAAmB,EAAE,MAAM,CAAC;IAC5B,KAAK,CAAC,EAAE,WAAW,CAAC;IACpB,GAAG,CAAC,EAAE,GAAG,CAAC;IACV,OAAO,CAAC,EAAE,oBAAoB,CAAC;IAC/B,UAAU,EAAE,oBAAoB,CAAC;CAClC,GAAG,OAAO,CAAC,MAAM,GAAG,SAAS,CAAC,CA8C9B;AAED,MAAM,MAAM,oBAAoB,GAAG;IACjC,UAAU,EACN,KAAK,GACL,UAAU,CAAC,IAAI,GACf,UAAU,CAAC,WAAW,GACtB,UAAU,CAAC,MAAM,CAAC;IACtB,aAAa,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;CACpC,CAAC;AAEF,wBAAsB,iBAAiB,CAAC,EACtC,IAAI,EACJ,UAAU,EACV,OAAO,GACR,EAAE;IACD,IAAI,EAAE,IAAI,CAAC;IACX,UAAU,EAAE,oBAAoB,CAAC;IACjC,OAAO,EAAE,oBAAoB,CAAC;CAC/B,GAAG,OAAO,CAAC;IACV,cAAc,EAAE;QAAE,SAAS,EAAE,MAAM,CAAC;QAAC,IAAI,EAAE,MAAM,CAAA;KAAE,EAAE,CAAC;IACtD,gBAAgB,EAAE,MAAM,CAAC;IACzB,uBAAuB,EAAE,MAAM,CAAC;CACjC,CAAC,CAoDD"}

package/dist/agent/master/element-annotation.js

@@ -120,7 +120,6 @@ async function getAnnotationKeys({ page, preference, options, }) {
         }));
         return annotations;
     }, { preference, options });
-    await page.waitForTimeout(2000);
     const annotationBuffer = await page.screenshot({
     // path: `screenshots/screenshot-${screenshotIndex++}.png`,
     });

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

@@ -1 +1 @@
-{"version":3,"file":"run.d.ts","sourceRoot":"","sources":["../../../src/agent/master/run.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,4BAA4B,CAAC;AAC3D,OAAO,EAAE,IAAI,EAAE,MAAM,YAAY,CAAC;AAclC,OAAO,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AACxC,OAAO,EACL,oBAAoB,EAErB,MAAM,aAAa,CAAC;AA6BrB,wBAAsB,0BAA0B,CAAC,EAC/C,IAAI,EACJ,IAAI,EACJ,QAAQ,EACR,QAAQ,EACR,OAAO,EACP,SAAS,GACV,EAAE;IACD,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,IAAI,CAAC;IACX,QAAQ,CAAC,EAAE,QAAQ,CAAC;IACpB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,OAAO,EAAE,oBAAoB,CAAC;IAC9B,SAAS,CAAC,EAAE,SAAS,CAAC;CACvB;;;GA0WA"}
+{"version":3,"file":"run.d.ts","sourceRoot":"","sources":["../../../src/agent/master/run.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,4BAA4B,CAAC;AAC3D,OAAO,EAAE,IAAI,EAAE,MAAM,YAAY,CAAC;AAclC,OAAO,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AACxC,OAAO,EACL,oBAAoB,EAErB,MAAM,aAAa,CAAC;AA6BrB,wBAAsB,0BAA0B,CAAC,EAC/C,IAAI,EACJ,IAAI,EACJ,QAAQ,EACR,QAAQ,EACR,OAAO,EACP,SAAS,GACV,EAAE;IACD,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,IAAI,CAAC;IACX,QAAQ,CAAC,EAAE,QAAQ,CAAC;IACpB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,OAAO,EAAE,oBAAoB,CAAC;IAC9B,SAAS,CAAC,EAAE,SAAS,CAAC;CACvB;;;GAsWA"}

package/dist/agent/master/run.js

@@ -37,8 +37,6 @@ async function createTestUsingMasterAgent({ task, page, testCase, specPath, opti
     const testgenUpdatesReporter = new reporter_1.TestGenUpdatesReporter();
     const session = (0, session_1.getSessionDetails)();
     const testGenPage = new page_1.TestGenPage(page, (0, utils_1.getPageVarName)());
-    // add timeout for the page to settle in
-    await page.waitForTimeout(3000);
     const trace = llm_1.langfuseInstance?.trace({
         name: "test-generator",
         id: crypto.randomUUID(),
@@ -182,7 +180,6 @@ async function createTestUsingMasterAgent({ task, page, testCase, specPath, opti
                     let shouldTriggerHintsFlow;
                     let hintsExecutionCompletion;
                     let elementAnnotation;
-                    await page.waitForTimeout(2000);
                     const actionType = toolCall.actionType;
                     let preference = {
                         actionType: "all",

package/dist/bin/index.js

@@ -35,6 +35,7 @@ const run_1 = require("../agent/browsing/run");
 const utils_1 = require("../agent/browsing/utils");
 const repo_edit_1 = require("../agent/codegen/repo-edit");
 const run_2 = require("../agent/codegen/run");
+const diagnosis_agent_1 = require("../agent/diagnosis-agent");
 const enrich_prompt_1 = require("../agent/enrich-prompt");
 const infer_agent_1 = require("../agent/infer-agent");
 const run_3 = require("../agent/planner/run");
@@ -112,6 +113,20 @@ async function runAgent(testGenConfig, span) {
         });
         return;
     }
+    // TODO: this needs to be moved to an orchestrator which decides what needs to be done first before executing the sub tasks
+    if (testGenConfig.testErrorDiagnosis &&
+        testGenConfig.testErrorDiagnosis.failingLine &&
+        // TODO: fix this hardcoding of user prompt - ideally its an auto fix intent
+        testCase.steps[0] == "Can you please fix the test") {
+        const { task: updatedTask } = await (0, diagnosis_agent_1.createTaskUsingFailureDiagnosis)({
+            options: testGenConfig.options,
+            trace,
+            diagnosis: testGenConfig.testErrorDiagnosis,
+        });
+        if (updatedTask) {
+            testCase.steps = [updatedTask];
+        }
+    }
     if (!agent || agent === "auto") {
         agent = await resolveAgentUsingTask({
             testCase,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@empiricalrun/test-gen",
-  "version": "0.42.28",
+  "version": "0.43.0",
   "publishConfig": {
     "registry": "https://registry.npmjs.org/",
     "access": "public"
@@ -72,7 +72,7 @@
     "ts-morph": "^23.0.0",
     "tsx": "^4.16.2",
     "typescript": "^5.3.3",
-    "@empiricalrun/llm": "^0.9.34",
+    "@empiricalrun/llm": "^0.9.35",
     "@empiricalrun/r2-uploader": "^0.3.8",
     "@empiricalrun/reporter": "^0.23.1"
   },
@@ -89,7 +89,7 @@
     "js-levenshtein": "^1.1.6",
     "playwright": "1.47.1",
     "ts-patch": "^3.3.0",
-    "@empiricalrun/shared-types": "0.0.3"
+    "@empiricalrun/shared-types": "0.0.4"
   },
   "scripts": {
     "dev": "tspc --build --watch",