@empiricalrun/test-gen 0.38.2 → 0.38.5

package/CHANGELOG.md

@@ -1,5 +1,23 @@
 # @empiricalrun/test-gen
 
+## 0.38.5
+
+### Patch Changes
+
+- 50c7a9a: feat: add support for page variables and state variables
+
+## 0.38.4
+
+### Patch Changes
+
+- be5370d: chore: send skill usage message on chat
+
+## 0.38.3
+
+### Patch Changes
+
+- ff76b50: fix: return first visible element matching with the CSS selector
+
 ## 0.38.2
 
 ### Patch Changes

package/dist/actions/assert.js

@@ -7,9 +7,9 @@ exports.PLAYWRIGHT_ASSERT_TEXT_VISIBILITY_ACTION_NAME = "assert_text_visibility"
 const assertTextVisibilityActionGenerator = (page) => {
     return {
         execute: async ({ args }) => {
-            const locator = await (0, utils_1.getPlaywrightLocatorUsingCssSelector)(args.css_selector, args.xpath, page, args?.elementAnnotation);
+            const locator = await (0, utils_1.getPlaywrightLocatorUsingCssSelector)(args.css_selector, args.xpath, page.pwPageInstance, args?.elementAnnotation);
             const exec = new Function("page", `return page.${locator}.isVisible({ timeout: 3000 })`);
-            await exec(page);
+            await exec(page.pwPageInstance);
             return {
                 locator,
             };
@@ -17,7 +17,7 @@ const assertTextVisibilityActionGenerator = (page) => {
         // TODO: args transformer to be kept at a single place
         template: (_, options) => {
             return {
-                code: `await expect(${(0, utils_1.getPageVarName)()}.${options.locator}).toBeVisible();`,
+                code: `await expect(${page.name}.${options.locator}).toBeVisible();`,
             };
         },
         name: exports.PLAYWRIGHT_ASSERT_TEXT_VISIBILITY_ACTION_NAME,

package/dist/actions/click.js

@@ -8,10 +8,10 @@ const clickActionGenerator = (page) => {
     return {
         execute: async ({ args }) => {
             const selector = args.css_selector;
-            const locator = await (0, utils_1.getPlaywrightLocatorUsingCssSelector)(selector, args.xpath, page, args?.elementAnnotation);
+            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);
-            await page.waitForTimeout(3000);
+            await exec(page.pwPageInstance);
+            await page.pwPageInstance.waitForTimeout(3000);
             return {
                 locator,
             };
@@ -19,7 +19,7 @@ const clickActionGenerator = (page) => {
         // TODO: args transformer to be kept at a single place
         template: (_, options) => {
             return {
-                code: `await ${(0, utils_1.getPageVarName)()}.${options.locator}.click();`,
+                code: `await ${page.name}.${options.locator}.click();`,
             };
         },
         name: exports.PLAYWRIGHT_CLICK_ACTION_NAME,

package/dist/actions/fill.js

@@ -11,10 +11,10 @@ const fillActionGenerator = (page, options) => {
     return {
         execute: async ({ args }) => {
             const css = args.css_selector;
-            const locator = await (0, utils_1.getPlaywrightLocatorUsingCssSelector)(css, args.xpath, page, args?.elementAnnotation);
+            const locator = await (0, utils_1.getPlaywrightLocatorUsingCssSelector)(css, args.xpath, page.pwPageInstance, args?.elementAnnotation);
             const textToFill = options?.stateVariables[args.variable_name] || args.text;
             const exec = new Function("page", `return page.${locator}.fill("${textToFill}", { timeout: 3000 })`);
-            await exec(page);
+            await exec(page.pwPageInstance);
             return {
                 locator,
             };
@@ -22,7 +22,7 @@ const fillActionGenerator = (page, options) => {
         // TODO: args transformer to be kept at a single place
         template: (args, options) => {
             return {
-                code: `await ${(0, utils_1.getPageVarName)()}.${options.locator}.fill("${args.text}");`,
+                code: `await ${page.name}.${options.locator}.fill("${args.text}");`,
             };
         },
         name: exports.PLAYWRIGHT_FILL_ACTION_NAME,

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

@@ -1 +1 @@
-{"version":3,"file":"goto.d.ts","sourceRoot":"","sources":["../../src/actions/goto.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,yBAAyB,EAAE,MAAM,UAAU,CAAC;AAIrD,eAAO,MAAM,2BAA2B,cAAc,CAAC;AAEvD,eAAO,MAAM,mBAAmB,EAAE,yBAuCjC,CAAC"}
+{"version":3,"file":"goto.d.ts","sourceRoot":"","sources":["../../src/actions/goto.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,yBAAyB,EAAE,MAAM,UAAU,CAAC;AAGrD,eAAO,MAAM,2BAA2B,cAAc,CAAC;AAEvD,eAAO,MAAM,mBAAmB,EAAE,yBAuCjC,CAAC"}

package/dist/actions/goto.js

@@ -3,20 +3,19 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.gotoActionGenerator = exports.PLAYWRIGHT_GOTO_ACTION_NAME = void 0;
 const utils_1 = require("../agent/browsing/utils");
 const constants_1 = require("./constants");
-const utils_2 = require("./utils");
 exports.PLAYWRIGHT_GOTO_ACTION_NAME = "page_goto";
 const gotoActionGenerator = (page) => {
     return {
         execute: async ({ args }) => {
             const url = args.url;
-            await page.goto(url);
-            await page.waitForTimeout(3000);
-            await (0, utils_1.injectPwLocatorGenerator)(page);
+            await page.pwPageInstance.goto(url);
+            await page.pwPageInstance.waitForTimeout(3000);
+            await (0, utils_1.injectPwLocatorGenerator)(page.pwPageInstance);
         },
         // TODO: args transformer to be kept at a single place
         template: (args) => {
             const url = args.url;
-            const code = `await ${(0, utils_2.getPageVarName)()}.goto("${url}");`;
+            const code = `await ${page.name}.goto("${url}");`;
             return {
                 code,
             };

package/dist/actions/hover.js

@@ -8,17 +8,17 @@ const hoverActionGenerator = (page) => {
     return {
         execute: async ({ args }) => {
             const selector = args.css_selector;
-            const locator = await (0, utils_1.getPlaywrightLocatorUsingCssSelector)(selector, args.xpath, page, args?.elementAnnotation);
+            const locator = await (0, utils_1.getPlaywrightLocatorUsingCssSelector)(selector, args.xpath, page.pwPageInstance, args?.elementAnnotation);
             const exec = new Function("page", `return page.${locator}.hover({ timeout: 3000 })`);
-            await exec(page);
-            await page.waitForTimeout(3000);
+            await exec(page.pwPageInstance);
+            await page.pwPageInstance.waitForTimeout(3000);
             return {
                 locator,
             };
         },
         // TODO: args transformer to be kept at a single place
         template: (args, options) => {
-            return { code: `await ${(0, utils_1.getPageVarName)()}.${options.locator}.hover();` };
+            return { code: `await ${page.name}.${options.locator}.hover();` };
         },
         name: exports.PLAYWRIGHT_HOVER_ACTION_NAME,
         schema: {

package/dist/actions/index.d.ts

@@ -1,12 +1,12 @@
 import { TraceClient } from "@empiricalrun/llm";
-import { Page } from "playwright";
+import { TestGenPage } from "../page";
 import { ActionSchema } from "../types";
 export declare class PlaywrightActions {
     private page;
     private stateVariables;
     private actionGenerators;
     private recordedActions;
-    constructor(page: Page, stateVariables?: Record<string, any>);
+    constructor(page: TestGenPage, stateVariables?: Record<string, any>);
     executeAction(name: string | undefined, args: Record<string, any>, trace?: TraceClient): Promise<string | undefined>;
     getBrowsingActionSchemas(): ActionSchema[];
     getMasterActionSchemas(): ActionSchema[];
@@ -24,5 +24,7 @@ export declare class PlaywrightActions {
      */
     isStuckInLoop(): boolean;
     isComplete(): boolean;
+    getStateVariables(): Record<string, any>;
+    setStateVariables(stateVariables: Record<string, any>): void;
 }
 //# sourceMappingURL=index.d.ts.map

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

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/actions/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAChD,OAAO,EAAE,IAAI,EAAE,MAAM,YAAY,CAAC;AAIlC,OAAO,EAAE,YAAY,EAA6B,MAAM,UAAU,CAAC;AAWnE,qBAAa,iBAAiB;IAQ1B,OAAO,CAAC,IAAI;IACZ,OAAO,CAAC,cAAc;IARxB,OAAO,CAAC,gBAAgB,CAA8B;IACtD,OAAO,CAAC,eAAe,CAInB;gBAEM,IAAI,EAAE,IAAI,EACV,cAAc,GAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAM;IAgB5C,aAAa,CACjB,IAAI,oBAAa,EACjB,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,EACzB,KAAK,CAAC,EAAE,WAAW,GAClB,OAAO,CAAC,MAAM,GAAG,SAAS,CAAC;IAgD9B,wBAAwB,IAAI,YAAY,EAAE;IAkB1C,sBAAsB,IAAI,YAAY,EAAE;IAUxC,YAAY,IAAI;QACd,IAAI,EAAE,MAAM,CAAC;QACb,WAAW,EAAE,MAAM,EAAE,CAAC;KACvB;IAUD,gBAAgB,CAAC,KAAK,EAAE,MAAM;IAK9B;;;;;;OAMG;IACH,aAAa,IAAI,OAAO;IAQxB,UAAU;CAUX"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/actions/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAGhD,OAAO,EAAE,WAAW,EAAE,MAAM,SAAS,CAAC;AAEtC,OAAO,EAAE,YAAY,EAA6B,MAAM,UAAU,CAAC;AAWnE,qBAAa,iBAAiB;IAQ1B,OAAO,CAAC,IAAI;IACZ,OAAO,CAAC,cAAc;IARxB,OAAO,CAAC,gBAAgB,CAA8B;IACtD,OAAO,CAAC,eAAe,CAInB;gBAEM,IAAI,EAAE,WAAW,EACjB,cAAc,GAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAM;IAgB5C,aAAa,CACjB,IAAI,oBAAa,EACjB,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,EACzB,KAAK,CAAC,EAAE,WAAW,GAClB,OAAO,CAAC,MAAM,GAAG,SAAS,CAAC;IAqD9B,wBAAwB,IAAI,YAAY,EAAE;IAmB1C,sBAAsB,IAAI,YAAY,EAAE;IAWxC,YAAY,IAAI;QACd,IAAI,EAAE,MAAM,CAAC;QACb,WAAW,EAAE,MAAM,EAAE,CAAC;KACvB;IAUD,gBAAgB,CAAC,KAAK,EAAE,MAAM;IAK9B;;;;;;OAMG;IACH,aAAa,IAAI,OAAO;IAQxB,UAAU;IAWV,iBAAiB;IAIjB,iBAAiB,CAAC,cAAc,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC;CAItD"}

package/dist/actions/index.js

@@ -35,7 +35,10 @@ class PlaywrightActions {
     }
     async executeAction(name = "", args, trace) {
         const [action] = this.actionGenerators
-            .map((a) => a(this.page, { stateVariables: this.stateVariables }))
+            .map((a) => a(this.page, {
+            stateVariables: this.stateVariables,
+            setStateVariables: this.setStateVariables.bind(this),
+        }))
             .filter((a) => a.name === name);
         if (!action) {
             throw Error(`No action registered for action: ${name}`);
@@ -93,6 +96,7 @@ class PlaywrightActions {
         ]
             .map((a) => a(this.page, {
             stateVariables: this.stateVariables,
+            setStateVariables: this.setStateVariables.bind(this),
         }))
             .map((a) => a.schema);
     }
@@ -100,6 +104,7 @@ class PlaywrightActions {
         return [skill_1.skillActionGenerator]
             .map((a) => a(this.page, {
             stateVariables: this.stateVariables,
+            setStateVariables: this.setStateVariables.bind(this),
         }))
             .map((a) => a.schema);
     }
@@ -135,5 +140,12 @@ class PlaywrightActions {
         this.recordedActions = this.recordedActions.filter((a) => a.name !== done_1.PLAYWRIGHT_DONE_ACTION_NAME);
         return !!doneAction;
     }
+    getStateVariables() {
+        return this.stateVariables;
+    }
+    setStateVariables(stateVariables) {
+        this.stateVariables = stateVariables;
+        console.log("setStateVariables", Object.keys(this.stateVariables));
+    }
 }
 exports.PlaywrightActions = PlaywrightActions;

package/dist/actions/press.js

@@ -8,9 +8,9 @@ const pressActionGenerator = (page) => {
     return {
         execute: async ({ args }) => {
             const css = args.css_selector;
-            const locator = await (0, utils_1.getPlaywrightLocatorUsingCssSelector)(css, args.xpath, page, args?.elementAnnotation);
+            const locator = await (0, utils_1.getPlaywrightLocatorUsingCssSelector)(css, args.xpath, page.pwPageInstance, args?.elementAnnotation);
             const exec = new Function("page", `return page.${locator}.press("${args.key}", { timeout: 3000 })`);
-            await exec(page);
+            await exec(page.pwPageInstance);
             return {
                 locator,
             };
@@ -18,7 +18,7 @@ const pressActionGenerator = (page) => {
         // TODO: args transformer to be kept at a single place
         template: (args, options) => {
             return {
-                code: `await ${(0, utils_1.getPageVarName)()}.${options.locator}.press("${args.key}");`,
+                code: `await ${page.name}.${options.locator}.press("${args.key}");`,
             };
         },
         name: exports.PLAYWRIGHT_PRESS_ACTION_NAME,

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

@@ -1 +1 @@
-{"version":3,"file":"skill.d.ts","sourceRoot":"","sources":["../../src/actions/skill.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,yBAAyB,EAAE,MAAM,UAAU,CAAC;AAGrD,eAAO,MAAM,WAAW,gBAAgB,CAAC;AAEzC,KAAK,KAAK,GAAG;IACX,QAAQ,EAAE,MAAM,CAAC;IACjB,QAAQ,EAAE,MAAM,CAAC;IACjB,YAAY,EAAE,MAAM,CAAC;IACrB,MAAM,EAAE,MAAM,CAAC;CAChB,CAAC;AAEF,cAAM,cAAc;IACN,OAAO,CAAC,MAAM;gBAAN,MAAM,EAAE,KAAK,EAAE;IAEnC,kBAAkB;IAIlB,YAAY,CAAC,MAAM,EAAE,KAAK,EAAE;CAG7B;AAED,eAAO,MAAM,cAAc,gBAAyB,CAAC;AAErD,eAAO,MAAM,oBAAoB,EAAE,yBA2FlC,CAAC"}
+{"version":3,"file":"skill.d.ts","sourceRoot":"","sources":["../../src/actions/skill.ts"],"names":[],"mappings":"AAKA,OAAO,EAAE,yBAAyB,EAAE,MAAM,UAAU,CAAC;AAGrD,eAAO,MAAM,WAAW,gBAAgB,CAAC;AAEzC,KAAK,KAAK,GAAG;IACX,QAAQ,EAAE,MAAM,CAAC;IACjB,QAAQ,EAAE,MAAM,CAAC;IACjB,YAAY,EAAE,MAAM,CAAC;IACrB,MAAM,EAAE,MAAM,CAAC;CAChB,CAAC;AAEF,cAAM,cAAc;IACN,OAAO,CAAC,MAAM;gBAAN,MAAM,EAAE,KAAK,EAAE;IAEnC,kBAAkB;IAIlB,YAAY,CAAC,MAAM,EAAE,KAAK,EAAE;CAG7B;AAED,eAAO,MAAM,cAAc,gBAAyB,CAAC;AAErD,eAAO,MAAM,oBAAoB,EAAE,yBA8GlC,CAAC"}

package/dist/actions/skill.js

@@ -4,8 +4,10 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
 };
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.skillActionGenerator = exports.testCaseSkills = exports.SKILL_USAGE = void 0;
+const promises_1 = __importDefault(require("fs/promises"));
 const api_1 = __importDefault(require("tsx/cjs/api"));
 const use_skill_1 = require("../agent/codegen/use-skill");
+const web_1 = require("../bin/utils/platform/web");
 const utils_1 = require("./utils");
 exports.SKILL_USAGE = "skill_usage";
 class TestCaseSkills {
@@ -34,23 +36,42 @@ const skillActionGenerator = (page, options) => {
             const [lastDir] = process.cwd().split("/").reverse();
             const dir = `${process.cwd()}/${lastDir}`;
             const module = await api_1.default.require(`./${skillFilePath}`, dir);
+            const skillFile = await promises_1.default.readFile(skillFilePath, "utf-8");
             const imports = Object.keys(module);
             const code = await (0, use_skill_1.generateSkillUsageCode)({
                 task: skillDetails.testStep,
                 sampleUsageMethod: skillDetails.usageExample,
                 scopeVariablesMapStr: JSON.stringify(options.stateVariables || {}, null, 2),
                 pageVariableName: (0, utils_1.getPageVarName)(),
+                skillMethodDefinition: skillFile,
                 trace,
             });
             console.log("Usage code generated", code);
             const stateVarKeys = Object.keys(options.stateVariables || {});
+            const newVariablesSpan = trace?.span({
+                name: "new-state-variables",
+            });
+            const newVariables = (0, web_1.getVariableDeclarationsFromCode)(code);
+            newVariablesSpan?.end({
+                output: {
+                    newVariables,
+                },
+            });
             const exec = new Function((0, utils_1.getPageVarName)(), ...imports, ...stateVarKeys, 
-            // assuming all POMs are async methods
+            // assuming all POMs are async methods and the LLM should respond with await keyword
             `
-        return (async () => { 
-          return ${code}
+        return (async () => {
+          ${code}
+          return {
+            ${newVariables.join(",")}
+          }
         })()`);
-            await exec(page, ...imports.map((i) => module[i]), ...stateVarKeys.map((s) => options.stateVariables[s]));
+            // extract the state variables after execution
+            const newStateVariables = await exec(page, ...imports.map((i) => module[i]), ...stateVarKeys.map((s) => options.stateVariables[s]));
+            options.setStateVariables({
+                ...options.stateVariables,
+                ...newStateVariables,
+            });
             return {
                 locator: code,
             };

package/dist/actions/text-content.js

@@ -8,9 +8,9 @@ const textContentActionGenerator = (page, options) => {
     return {
         execute: async (args) => {
             const css = args.css_selector;
-            const locator = await (0, utils_1.getPlaywrightLocatorUsingCssSelector)(css, args.xpath, page, args?.elementAnnotation);
+            const locator = await (0, utils_1.getPlaywrightLocatorUsingCssSelector)(css, args.xpath, page.pwPageInstance, args?.elementAnnotation);
             const exec = new Function("page", `return page.${locator}.textContent()`);
-            const value = (await exec(page));
+            const value = (await exec(page.pwPageInstance));
             if (options) {
                 options.stateVariables[args.variable_name] = value;
             }
@@ -19,7 +19,7 @@ const textContentActionGenerator = (page, options) => {
             };
         },
         template: (args, options) => {
-            const code = `const ${args.variable_name} = await ${(0, utils_1.getPageVarName)()}.${options.locator}.textContent();`;
+            const code = `const ${args.variable_name} = await ${page.name}.${options.locator}.textContent();`;
             return {
                 code,
             };

package/dist/actions/utils/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/actions/utils/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,EAAE,MAAM,YAAY,CAAC;AAElC,OAAO,CAAC,MAAM,CAAC;IACb,UAAU,MAAM;QACd,UAAU,EAAE,GAAG,CAAC;QAChB,MAAM,EAAE,GAAG,CAAC;KACb;CACF;AAED,wBAAsB,oCAAoC,CACxD,WAAW,EAAE,MAAM,EACnB,KAAK,EAAE,MAAM,EACb,IAAI,EAAE,IAAI,EACV,iBAAiB,CAAC,EAAE,MAAM,gBAkF3B;AAED,wBAAgB,cAAc,WAG7B"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/actions/utils/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,EAAE,MAAM,YAAY,CAAC;AAElC,OAAO,CAAC,MAAM,CAAC;IACb,UAAU,MAAM;QACd,UAAU,EAAE,GAAG,CAAC;QAChB,MAAM,EAAE,GAAG,CAAC;KACb;CACF;AAED,wBAAsB,oCAAoC,CACxD,WAAW,EAAE,MAAM,EACnB,KAAK,EAAE,MAAM,EACb,IAAI,EAAE,IAAI,EACV,iBAAiB,CAAC,EAAE,MAAM,gBAoF3B;AAED,wBAAgB,cAAc,WAG7B"}

package/dist/actions/utils/index.js

@@ -52,10 +52,12 @@ async function getPlaywrightLocatorUsingCssSelector(cssSelector, xpath, page, el
         if (!selectedElem) {
             const elements = window.jQuery(locator.cssForJq);
             let elIdx = 0;
-            Array.from(elements).forEach((el, i) => {
+            Array.from(elements).some((el, i) => {
                 if (window.jQuery(el).is(":visible")) {
                     elIdx = i;
+                    return true;
                 }
+                return false;
             });
             //@ts-ignore
             selectedElem = elements[elIdx];

package/dist/agent/codegen/skills-retriever.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"skills-retriever.d.ts","sourceRoot":"","sources":["../../../src/agent/codegen/skills-retriever.ts"],"names":[],"mappings":"AAAA,OAAO,EAAkB,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAYhE,OAAO,EAAE,QAAQ,EAAE,oBAAoB,EAAE,MAAM,aAAa,CAAC;AAG7D,eAAO,MAAM,cAAc;cAMf,QAAQ;;;;;;;;;;IA8CnB,CAAC;AAEF,wBAAsB,oBAAoB,CAAC,EACzC,QAAQ,EACR,OAAO,EACP,KAAK,GACN,EAAE;IACD,QAAQ,EAAE,QAAQ,CAAC;IACnB,OAAO,CAAC,EAAE,oBAAoB,CAAC;IAC/B,KAAK,CAAC,EAAE,WAAW,CAAC;CACrB;;;;;;KA6BA"}
+{"version":3,"file":"skills-retriever.d.ts","sourceRoot":"","sources":["../../../src/agent/codegen/skills-retriever.ts"],"names":[],"mappings":"AAAA,OAAO,EAAkB,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAgBhE,OAAO,EAAE,QAAQ,EAAE,oBAAoB,EAAE,MAAM,aAAa,CAAC;AAG7D,eAAO,MAAM,cAAc;cAMf,QAAQ;;;;;;;;;;IAoFnB,CAAC;AAEF,wBAAsB,oBAAoB,CAAC,EACzC,QAAQ,EACR,OAAO,EACP,KAAK,GACN,EAAE;IACD,QAAQ,EAAE,QAAQ,CAAC;IACnB,OAAO,CAAC,EAAE,oBAAoB,CAAC;IAC/B,KAAK,CAAC,EAAE,WAAW,CAAC;CACrB;;;;;;KA6BA"}

package/dist/agent/codegen/skills-retriever.js

@@ -43,7 +43,42 @@ const fetchPomSkills = async ({ testCase, pomFiles, options, trace, }) => {
         },
     });
     let response = firstShotMessage?.content || "";
-    const skills = (0, utils_1.extractTestStepsSuggestions)(response);
+    const { codePrompt } = await (0, context_1.contextForGeneration)();
+    const usageExampleUpdatePrompt = [
+        {
+            role: "system",
+            content: `
+You are provided with usage examples of page object models of a Playwright test repository and you are given a task to update the usage example of the page object models with the usage of them in tests. 
+
+You need to ensure all the pre-requisites of calling the method is also mentioned in the usage example. Use the tests code for same
+      `,
+        },
+        {
+            role: "user",
+            content: `
+Usage examples of page object models:
+
+${response}
+
+Tests files:
+
+${codePrompt}
+
+Before responding: 
+- ensure responding in the same format as provided to you in the usage example of page object models.
+- respond only with the methods and not any test block
+    `,
+        },
+    ];
+    const updatedUsageExampleMessage = await llm.createChatCompletion({
+        messages: usageExampleUpdatePrompt,
+        traceName: "fetch-pom-skills-usage-eg-update-llm",
+        modelParameters: {
+            ...constants_1.DEFAULT_MODEL_PARAMETERS,
+            ...options?.modelParameters,
+        },
+    });
+    const skills = (0, utils_1.extractTestStepsSuggestions)(updatedUsageExampleMessage?.content || "");
     fetchSkillsUsingPOMFilesSpan?.end({ output: { skills } });
     return skills;
 };

package/dist/agent/codegen/use-skill.d.ts

@@ -1,9 +1,10 @@
 import { TraceClient } from "@empiricalrun/llm";
-export declare function generateSkillUsageCode({ task, sampleUsageMethod, scopeVariablesMapStr, pageVariableName, trace, }: {
+export declare function generateSkillUsageCode({ task, sampleUsageMethod, scopeVariablesMapStr, pageVariableName, skillMethodDefinition, trace, }: {
     task: string;
     sampleUsageMethod: string;
     scopeVariablesMapStr: string;
     pageVariableName: string;
+    skillMethodDefinition: string;
     trace?: TraceClient;
 }): Promise<string>;
 //# sourceMappingURL=use-skill.d.ts.map

package/dist/agent/codegen/use-skill.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"use-skill.d.ts","sourceRoot":"","sources":["../../../src/agent/codegen/use-skill.ts"],"names":[],"mappings":"AAAA,OAAO,EAAkB,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAShE,wBAAsB,sBAAsB,CAAC,EAC3C,IAAI,EACJ,iBAAiB,EACjB,oBAAoB,EACpB,gBAAgB,EAChB,KAAK,GACN,EAAE;IACD,IAAI,EAAE,MAAM,CAAC;IACb,iBAAiB,EAAE,MAAM,CAAC;IAC1B,oBAAoB,EAAE,MAAM,CAAC;IAC7B,gBAAgB,EAAE,MAAM,CAAC;IACzB,KAAK,CAAC,EAAE,WAAW,CAAC;CACrB,mBA0CA"}
+{"version":3,"file":"use-skill.d.ts","sourceRoot":"","sources":["../../../src/agent/codegen/use-skill.ts"],"names":[],"mappings":"AAAA,OAAO,EAAkB,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAUhE,wBAAsB,sBAAsB,CAAC,EAC3C,IAAI,EACJ,iBAAiB,EACjB,oBAAoB,EACpB,gBAAgB,EAChB,qBAAqB,EACrB,KAAK,GACN,EAAE;IACD,IAAI,EAAE,MAAM,CAAC;IACb,iBAAiB,EAAE,MAAM,CAAC;IAC1B,oBAAoB,EAAE,MAAM,CAAC;IAC7B,gBAAgB,EAAE,MAAM,CAAC;IACzB,qBAAqB,EAAE,MAAM,CAAC;IAC9B,KAAK,CAAC,EAAE,WAAW,CAAC;CACrB,mBAiDA"}

package/dist/agent/codegen/use-skill.js

@@ -2,8 +2,11 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.generateSkillUsageCode = void 0;
 const llm_1 = require("@empiricalrun/llm");
+const logger_1 = require("../../bin/logger");
 const constants_1 = require("../../constants");
-async function generateSkillUsageCode({ task, sampleUsageMethod, scopeVariablesMapStr, pageVariableName, trace, }) {
+async function generateSkillUsageCode({ task, sampleUsageMethod, scopeVariablesMapStr, pageVariableName, skillMethodDefinition, trace, }) {
+    const logger = new logger_1.CustomLogger();
+    logger.log(`Generating code using skill usage example: ${sampleUsageMethod}`);
     const skillUsageSpan = trace?.span({
         name: "skill-usage",
         input: {
@@ -21,7 +24,8 @@ async function generateSkillUsageCode({ task, sampleUsageMethod, scopeVariablesM
         sampleUsageMethod,
         scopeVariablesMapStr,
         pageVariableName,
-    });
+        skillMethodDefinition,
+    }, 4);
     promptSpan?.end({ output: prompt });
     const llm = new llm_1.LLM({
         trace: skillUsageSpan,

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":"AAAA,OAAO,EAGL,GAAG,EACH,WAAW,EACZ,MAAM,mBAAmB,CAAC;AAG3B,OAAO,EAAE,IAAI,EAAE,MAAM,YAAY,CAAC;AAElC,OAAO,EAAE,iBAAiB,EAAE,MAAM,eAAe,CAAC;AAYlD,OAAO,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,aAAa,CAAC;AAClD,OAAO,EACL,oBAAoB,EAErB,MAAM,aAAa,CAAC;AAQrB,wBAAsB,aAAa,CAAC,EAClC,IAAI,EACJ,eAAe,EACf,aAAa,EACb,OAAO,EACP,KAAK,EACL,GAAG,EACH,OAAO,EACP,cAAc,EACd,uBAAuB,EACvB,OAAO,EACP,aAAa,EACb,QAAgB,EAChB,WAAW,GACZ,EAAE;IACD,IAAI,EAAE,MAAM,CAAC;IACb,eAAe,EAAE,MAAM,EAAE,CAAC;IAC1B,aAAa,EAAE,GAAG,EAAE,CAAC;IACrB,OAAO,EAAE,MAAM,CAAC;IAChB,KAAK,CAAC,EAAE,WAAW,CAAC;IACpB,GAAG,CAAC,EAAE,GAAG,CAAC;IACV,OAAO,CAAC,EAAE,oBAAoB,CAAC;IAC/B,cAAc,EAAE,MAAM,CAAC;IACvB,uBAAuB,CAAC,EAAE,MAAM,CAAC;IACjC,OAAO,EAAE,iBAAiB,CAAC;IAC3B,aAAa,EAAE,OAAO,CAAC;IACvB,QAAQ,EAAE,OAAO,CAAC;IAClB,WAAW,CAAC,EAAE,MAAM,EAAE,CAAC;CACxB,2FA2FA;AAGD,wBAAsB,0BAA0B,CAAC,EAC/C,IAAI,EACJ,IAAI,EACJ,QAAQ,EACR,OAAO,EACP,SAAS,GACV,EAAE;IACD,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,IAAI,CAAC;IACX,QAAQ,EAAE,QAAQ,CAAC;IACnB,OAAO,EAAE,oBAAoB,CAAC;IAC9B,SAAS,CAAC,EAAE,SAAS,CAAC;CACvB;;;GAuTA"}
+{"version":3,"file":"run.d.ts","sourceRoot":"","sources":["../../../src/agent/master/run.ts"],"names":[],"mappings":"AAAA,OAAO,EAGL,GAAG,EACH,WAAW,EACZ,MAAM,mBAAmB,CAAC;AAG3B,OAAO,EAAE,IAAI,EAAE,MAAM,YAAY,CAAC;AAElC,OAAO,EAAE,iBAAiB,EAAE,MAAM,eAAe,CAAC;AAclD,OAAO,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,aAAa,CAAC;AAClD,OAAO,EACL,oBAAoB,EAErB,MAAM,aAAa,CAAC;AAoBrB,wBAAsB,aAAa,CAAC,EAClC,IAAI,EACJ,eAAe,EACf,aAAa,EACb,OAAO,EACP,KAAK,EACL,GAAG,EACH,OAAO,EACP,cAAc,EACd,uBAAuB,EACvB,OAAO,EACP,aAAa,EACb,QAAgB,EAChB,WAAW,GACZ,EAAE;IACD,IAAI,EAAE,MAAM,CAAC;IACb,eAAe,EAAE,MAAM,EAAE,CAAC;IAC1B,aAAa,EAAE,GAAG,EAAE,CAAC;IACrB,OAAO,EAAE,MAAM,CAAC;IAChB,KAAK,CAAC,EAAE,WAAW,CAAC;IACpB,GAAG,CAAC,EAAE,GAAG,CAAC;IACV,OAAO,CAAC,EAAE,oBAAoB,CAAC;IAC/B,cAAc,EAAE,MAAM,CAAC;IACvB,uBAAuB,CAAC,EAAE,MAAM,CAAC;IACjC,OAAO,EAAE,iBAAiB,CAAC;IAC3B,aAAa,EAAE,OAAO,CAAC;IACvB,QAAQ,EAAE,OAAO,CAAC;IAClB,WAAW,CAAC,EAAE,MAAM,EAAE,CAAC;CACxB,2FA2FA;AAGD,wBAAsB,0BAA0B,CAAC,EAC/C,IAAI,EACJ,IAAI,EACJ,QAAQ,EACR,OAAO,EACP,SAAS,GACV,EAAE;IACD,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,IAAI,CAAC;IACX,QAAQ,EAAE,QAAQ,CAAC;IACnB,OAAO,EAAE,oBAAoB,CAAC;IAC9B,SAAS,CAAC,EAAE,SAAS,CAAC;CACvB;;;GAiUA"}

package/dist/agent/master/run.js

@@ -6,16 +6,27 @@ const vision_1 = require("@empiricalrun/llm/vision");
 const actions_1 = require("../../actions");
 const next_task_1 = require("../../actions/next-task");
 const skill_1 = require("../../actions/skill");
+const utils_1 = require("../../actions/utils");
 const logger_1 = require("../../bin/logger");
 const constants_1 = require("../../constants");
+const page_1 = require("../../page");
 const reporter_1 = require("../../reporter");
 const session_1 = require("../../session");
 const browsing_1 = require("../browsing");
-const utils_1 = require("../browsing/utils");
+const utils_2 = require("../browsing/utils");
 const skills_retriever_1 = require("../codegen/skills-retriever");
-const verification_1 = require("../verification");
+const run_time_planner_1 = require("../planner/run-time-planner");
 const with_hints_1 = require("./with-hints");
 const MAX_ERROR_COUNT = 2;
+function getPageVariables(stateVariables) {
+    const keys = Object.keys(stateVariables);
+    const pageVariables = keys.filter((key) => key.endsWith("Page") || key.endsWith("page"));
+    const pages = pageVariables.reduce((acc, key) => {
+        acc[key] = stateVariables[key];
+        return acc;
+    }, {});
+    return pages;
+}
 async function getNextAction({ task, executedActions, failedActions, pageUrl, trace, llm, options, pageScreenshot, annotatedPageScreenshot, actions, disableSkills, useHints = false, annotations, }) {
     const nextActionSpan = trace?.span({
         name: "master-agent-next-action",
@@ -105,6 +116,7 @@ async function createTestUsingMasterAgent({ task, page, testCase, options, scope
     const logger = new logger_1.CustomLogger({ useReporter: false });
     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({
@@ -141,8 +153,8 @@ async function createTestUsingMasterAgent({ task, page, testCase, options, scope
         options,
     });
     skill_1.testCaseSkills.updateSkills(skills);
-    const actions = new actions_1.PlaywrightActions(page, scopeVars);
-    await (0, utils_1.injectPwLocatorGenerator)(page);
+    const actions = new actions_1.PlaywrightActions(testGenPage, scopeVars);
+    await (0, utils_2.injectPwLocatorGenerator)(page);
     trace?.update({ input: { task } });
     let isGivenTaskDone = false;
     const masterAgentActions = [];
@@ -160,17 +172,25 @@ async function createTestUsingMasterAgent({ task, page, testCase, options, scope
                 failedActions,
             },
         });
-        if (masterAgentActions.length > 0) {
-            const verificationAgentResp = await (0, verification_1.verificationAgent)({
-                trace: masterAgentSpan,
-                task,
-                conversation: ["Successfully executed actions", ...masterAgentActions],
-            });
-            isGivenTaskDone = verificationAgentResp.isDone;
-            if (isGivenTaskDone) {
-                await testgenUpdatesReporter.sendMessage(`${verificationAgentResp.reason} Marking the task as done.`);
-                break;
-            }
+        const plannerResp = await (0, run_time_planner_1.runtimePlanner)({
+            trace: masterAgentSpan,
+            task,
+            conversation: ["Successfully executed actions", ...masterAgentActions],
+            pages: getPageVariables(actions.getStateVariables()),
+            currentPage: (0, utils_1.getPageVarName)(),
+        });
+        isGivenTaskDone = plannerResp.isDone;
+        if (isGivenTaskDone) {
+            await testgenUpdatesReporter.sendMessage(`${plannerResp.reason} Marking the task as done.`);
+            break;
+        }
+        if (actions.getStateVariables()[plannerResp.pageName]) {
+            // update page for the master agent
+            page = actions.getStateVariables()[plannerResp.pageName];
+            // update page in actions
+            testGenPage.updatePage({ page, name: plannerResp.pageName });
+            // inject scripts in the updated
+            await (0, utils_2.injectPwLocatorGenerator)(testGenPage.pwPageInstance);
         }
         const buffer = await page.screenshot({
         //This is done to improve element annotation accuracy, anyways it doesn't annotate elements which are out of viewport
@@ -261,7 +281,7 @@ async function createTestUsingMasterAgent({ task, page, testCase, options, scope
                     const result = await (0, with_hints_1.triggerHintsFlow)({
                         outputFromGetNextAction: output,
                         generatedAnnotations: annotations,
-                        page,
+                        page: testGenPage,
                         llm,
                         trace: triggerHintsFlowSpan,
                     });

package/dist/agent/master/with-hints.d.ts

@@ -1,6 +1,6 @@
 import { LLM, TraceClient } from "@empiricalrun/llm";
 import OpenAI from "openai";
-import { Page } from "playwright";
+import { TestGenPage } from "../../page";
 import { BrowsingAgentOptions } from "../browsing";
 export declare const getUserMessageWithForHints: ({ userMessage, options, pageScreenshot, annotatedPageScreenshot, }: {
     userMessage: OpenAI.ChatCompletionUserMessageParam;
@@ -14,7 +14,7 @@ export declare const triggerHintsFlow: ({ outputFromGetNextAction, generatedAnno
         elementAnnotation?: string;
     };
     generatedAnnotations: Record<string, any>;
-    page: Page;
+    page: TestGenPage;
     llm: LLM;
     trace?: TraceClient | undefined;
 }) => Promise<{

package/dist/agent/master/with-hints.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"with-hints.d.ts","sourceRoot":"","sources":["../../../src/agent/master/with-hints.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,GAAG,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAErD,OAAO,MAAM,MAAM,QAAQ,CAAC;AAC5B,OAAO,EAAE,IAAI,EAAE,MAAM,YAAY,CAAC;AAIlC,OAAO,EAAE,oBAAoB,EAAE,MAAM,aAAa,CAAC;AAEnD,eAAO,MAAM,0BAA0B;iBAMxB,OAAO,8BAA8B;;oBAElC,MAAM;6BACG,MAAM;MAC7B,MAAM,GAAG,OAAO,yBAAyB,EAiC5C,CAAC;AAEF,eAAO,MAAM,gBAAgB;6BAOF;QACvB,MAAM,EAAE,MAAM,CAAC;QACf,iBAAiB,CAAC,EAAE,MAAM,CAAC;KAC5B;0BACqB,OAAO,MAAM,EAAE,GAAG,CAAC;UACnC,IAAI;SACL,GAAG;;MAEN,QAAQ;IACV,sBAAsB,EAAE,OAAO,CAAC;IAChC,wBAAwB,EAAE,OAAO,qBAAqB,GAAG,SAAS,CAAC;CACpE,CAqGA,CAAC"}
+{"version":3,"file":"with-hints.d.ts","sourceRoot":"","sources":["../../../src/agent/master/with-hints.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,GAAG,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAErD,OAAO,MAAM,MAAM,QAAQ,CAAC;AAI5B,OAAO,EAAE,WAAW,EAAE,MAAM,YAAY,CAAC;AACzC,OAAO,EAAE,oBAAoB,EAAE,MAAM,aAAa,CAAC;AAEnD,eAAO,MAAM,0BAA0B;iBAMxB,OAAO,8BAA8B;;oBAElC,MAAM;6BACG,MAAM;MAC7B,MAAM,GAAG,OAAO,yBAAyB,EAiC5C,CAAC;AAEF,eAAO,MAAM,gBAAgB;6BAOF;QACvB,MAAM,EAAE,MAAM,CAAC;QACf,iBAAiB,CAAC,EAAE,MAAM,CAAC;KAC5B;0BACqB,OAAO,MAAM,EAAE,GAAG,CAAC;UACnC,WAAW;SACZ,GAAG;;MAEN,QAAQ;IACV,sBAAsB,EAAE,OAAO,CAAC;IAChC,wBAAwB,EAAE,OAAO,qBAAqB,GAAG,SAAS,CAAC;CACpE,CAqGA,CAAC"}

package/dist/agent/planner/run-time-planner.d.ts

@@ -0,0 +1,20 @@
+import { TraceClient } from "@empiricalrun/llm";
+/**
+ * This agent is used to divide the tasl into individual actions and then
+ * compare each action against the actions listed in the conversation.
+ * If the task is not fully completed, identify which specific actions are missing and suggest next steps to complete the task.
+ *
+ * This is very initial stage planner and needs iteration and currently forked from verification agent
+ */
+export declare function runtimePlanner({ trace, task, conversation, pages, currentPage, }: {
+    trace?: TraceClient;
+    conversation: string[];
+    task: string;
+    pages?: Record<string, any>;
+    currentPage?: string;
+}): Promise<{
+    pageName: string;
+    isDone: boolean;
+    reason: string;
+}>;
+//# sourceMappingURL=run-time-planner.d.ts.map

package/dist/agent/planner/run-time-planner.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"run-time-planner.d.ts","sourceRoot":"","sources":["../../../src/agent/planner/run-time-planner.ts"],"names":[],"mappings":"AAAA,OAAO,EAAO,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAGrD;;;;;;GAMG;AACH,wBAAsB,cAAc,CAAC,EACnC,KAAK,EACL,IAAI,EACJ,YAAY,EACZ,KAAK,EACL,WAAW,GACZ,EAAE;IACD,KAAK,CAAC,EAAE,WAAW,CAAC;IACpB,YAAY,EAAE,MAAM,EAAE,CAAC;IACvB,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IAC5B,WAAW,CAAC,EAAE,MAAM,CAAC;CACtB;;;;GA6GA"}

package/dist/agent/planner/run-time-planner.js

@@ -0,0 +1,121 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.runtimePlanner = void 0;
+const llm_1 = require("@empiricalrun/llm");
+/**
+ * This agent is used to divide the tasl into individual actions and then
+ * compare each action against the actions listed in the conversation.
+ * If the task is not fully completed, identify which specific actions are missing and suggest next steps to complete the task.
+ *
+ * This is very initial stage planner and needs iteration and currently forked from verification agent
+ */
+async function runtimePlanner({ trace, task, conversation, pages, currentPage, }) {
+    const runTimePlannerSpan = trace?.span({
+        name: "runtime-planner",
+        input: {
+            task,
+            conversation,
+        },
+    });
+    const llm = new llm_1.LLM({ provider: "openai" });
+    const prompt = [
+        {
+            role: "system",
+            content: `
+Given a conversation that lists only the actions that were successfully executed and a task comprising multiple actions, your goal is to analyse the conversation and determine if the entire task is completed.
+These conversations are between AI agents using Playwright to execute actions on browser. These agents already have access to browser tabs to execute steps. The successfully executed steps on browser post browser has opened, is provided to you as conversation.
+
+If the task is not fully completed, identify which specific actions are missing and suggest next steps to complete the task. Assume that the conversation provided is entirely truthful and no additional actions were performed beyond those listed.
+
+To fulfil your goal, follow these steps:
+- Divide the task into individual actions.
+- Compare each task action against the actions listed in the conversation.
+- Identify which actions have been executed and which have not.
+- If all actions are executed, respond with the task as done.
+- If any actions are missing, respond with the task as not done, listing all actions and specifying which are complete and which are missing.
+- If provided with list of pages, based on the next pending action and previously executed action, identify the page on which next action needs to be taken    
+    `,
+        },
+        {
+            role: "user",
+            content: `
+Task: ${task}
+
+Conversation:
+${conversation.join("\n")}
+
+Current page:
+${currentPage}
+        `,
+        },
+    ];
+    const response = await llm.createChatCompletion({
+        trace: runTimePlannerSpan,
+        traceName: "runtime-planner-llm",
+        model: "gpt-4o",
+        messages: prompt,
+        tools: [
+            {
+                type: "function",
+                function: {
+                    name: "task_done",
+                    description: "end the task by calling this method",
+                    parameters: {
+                        type: "object",
+                        properties: {
+                            actions: {
+                                type: "string",
+                                description: "actions extracted from task",
+                            },
+                            successful_actions: {
+                                type: "string",
+                                description: "successful actions mentioned in the conversation",
+                            },
+                            reason: {
+                                type: "string",
+                                description: "reasoning for identification of task status",
+                            },
+                            isDone: {
+                                type: "boolean",
+                                description: "whether the task is done",
+                            },
+                            pageName: {
+                                type: "string",
+                                enum: pages ? Object.keys(pages) : [],
+                                description: "page name for the next action.",
+                            },
+                        },
+                        required: ["isDone", "reason", "pageName"],
+                    },
+                },
+            },
+        ],
+        modelParameters: {
+            tool_choice: "required",
+            temperature: 0.5,
+        },
+    });
+    const toolCallResp = (response?.tool_calls || [])[0];
+    if (toolCallResp) {
+        const toolCall = JSON.parse(toolCallResp.function.arguments);
+        const output = {
+            pageName: toolCall.pageName,
+            isDone: toolCall.isDone,
+            reason: toolCall.reason,
+        };
+        runTimePlannerSpan?.end({
+            output,
+        });
+        return output;
+    }
+    const output = {
+        pageName: "",
+        isDone: false,
+        reason: "LLM failed to generate a valid response",
+    };
+    runTimePlannerSpan?.end({
+        output,
+    });
+    return output;
+}
+exports.runtimePlanner = runtimePlanner;

package/dist/bin/utils/context.d.ts

@@ -1,5 +1,5 @@
 export declare function createGitIgnoreFileFilter(): Promise<(pathname: string) => boolean>;
-export declare function contextForGeneration(file: string): Promise<{
+export declare function contextForGeneration(file?: string): Promise<{
     codePrompt: string | undefined;
     pomPrompt: string | undefined;
     testFileContent: string;

package/dist/bin/utils/context.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"context.d.ts","sourceRoot":"","sources":["../../../src/bin/utils/context.ts"],"names":[],"mappings":"AAKA,wBAAsB,yBAAyB,2CAS9C;AAED,wBAAsB,oBAAoB,CAAC,IAAI,EAAE,MAAM;;;;GAOtD;AAED,wBAAsB,iBAAiB,oBAYtC"}
+{"version":3,"file":"context.d.ts","sourceRoot":"","sources":["../../../src/bin/utils/context.ts"],"names":[],"mappings":"AAKA,wBAAsB,yBAAyB,2CAS9C;AAED,wBAAsB,oBAAoB,CAAC,IAAI,CAAC,EAAE,MAAM;;;;GAOvD;AAED,wBAAsB,iBAAiB,oBAYtC"}

package/dist/bin/utils/context.js

@@ -23,7 +23,7 @@ async function contextForGeneration(file) {
     return {
         codePrompt: await (0, fs_1.generatePromptFromDirectory)("./tests", filter),
         pomPrompt: await (0, fs_1.generatePromptFromDirectory)("./pages", filter),
-        testFileContent: await fs_extra_1.default.readFile(file, "utf-8"),
+        testFileContent: file ? await fs_extra_1.default.readFile(file, "utf-8") : "",
     };
 }
 exports.contextForGeneration = contextForGeneration;

package/dist/bin/utils/platform/web/index.d.ts

@@ -72,4 +72,5 @@ export declare function buildTestNamePrompt({ testName, suites, }: {
     testName: string;
     suites: string[];
 }): string;
+export declare function getVariableDeclarationsFromCode(sourceCode: string): string[];
 //# sourceMappingURL=index.d.ts.map

package/dist/bin/utils/platform/web/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../../src/bin/utils/platform/web/index.ts"],"names":[],"mappings":"AAMA,OAAO,EAGL,IAAI,EAEJ,UAAU,EAEX,MAAM,UAAU,CAAC;AAGlB,OAAO,EAAE,QAAQ,EAAE,MAAM,mBAAmB,CAAC;AAE7C,eAAO,MAAM,gCAAgC,eAC/B,UAAU,KACrB,MAgBF,CAAC;AAEF;;;;;;GAMG;AACH,wBAAgB,sBAAsB,CAAC,EACrC,YAAY,EACZ,MAAM,EACN,OAAO,GACR,EAAE;IACD,YAAY,EAAE,MAAM,CAAC;IACrB,MAAM,CAAC,EAAE,MAAM,EAAE,CAAC;IAClB,OAAO,EAAE,MAAM,CAAC;CACjB,GAAG;IACF,SAAS,EAAE,MAAM,GAAG,SAAS,CAAC;IAC9B,QAAQ,EAAE,IAAI,GAAG,SAAS,CAAC;IAC3B,SAAS,EAAE,MAAM,CAAC;CACnB,CA2CA;AAwBD,wBAAsB,0CAA0C,CAC9D,QAAQ,EAAE,MAAM,oBA+BjB;AAED;;;;;;;;;;;;;;;;;;;;;;GAsBG;AACH,wBAAgB,4BAA4B,CAC1C,IAAI,EAAE,IAAI,GAAG,SAAS,GACrB,IAAI,GAAG,SAAS,CA4BlB;AAED,wBAAgB,iBAAiB,CAAC,SAAS,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,MAAM,CAG5E;AAED,wBAAgB,kBAAkB,CAAC,QAAQ,EAAE,MAAM,GAAG,MAAM,EAAE,CA8C7D;AAED,wBAAsB,sBAAsB,CAC1C,OAAO,EAAE,MAAM,EACf,QAAQ,EAAE,MAAM,mCAWjB;AAED,wBAAsB,UAAU,CAAC,QAAQ,EAAE,MAAM,iBAShD;AAED,wBAAsB,UAAU,CAAC,QAAQ,EAAE,MAAM,iBAQhD;AAED,wBAAgB,YAAY,CAAC,QAAQ,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,EAAE,GAAG,EAAE,MAAM,UAE5E;AAED,wBAAsB,cAAc,CAAC,QAAQ,EAAE,MAAM,iBAMpD;AAED,wBAAgB,oBAAoB,CAAC,QAAQ,EAAE,MAAM,UAcpD;AAED,wBAAsB,iCAAiC,CAAC,QAAQ,EAAE,MAAM,+BAoBvE;AAED,wBAAgB,4BAA4B,CAC1C,QAAQ,EAAE,MAAM,EAChB,QAAQ,EAAE,MAAM,EAChB,aAAa,EAAE,MAAM,UA0CtB;AAED,eAAO,MAAM,6BAA6B;qBAKvB,MAAM;iBACV,MAAM;YACX,MAAM,EAAE;YA2DjB,CAAC;AAEF,eAAO,MAAM,iCAAiC,cACjC,MAAM,EAAE,gBACL,MAAM,sBAyBrB,CAAC;AAEF,wBAAsB,qBAAqB,CAAC,EAC1C,YAAY,EACZ,QAAQ,EACR,MAAM,GACP,EAAE;IACD,YAAY,EAAE,MAAM,CAAC;IACrB,QAAQ,EAAE,MAAM,CAAC;IACjB,MAAM,EAAE,MAAM,EAAE,CAAC;CAClB,iBAsDA;AAED,wBAAsB,uBAAuB,CAC3C,QAAQ,EAAE,MAAM,EAChB,cAAc,EAAE,MAAM,EAAE,iBA2BzB;AAED,wBAAgB,aAAa,CAAC,EAC5B,QAAQ,EACR,QAAQ,GACT,EAAE;IACD,QAAQ,EAAE,MAAM,CAAC;IACjB,QAAQ,EAAE,QAAQ,CAAC;CACpB,WAYA;AAED,wBAAgB,mBAAmB,CAAC,EAClC,QAAQ,EACR,MAAM,GACP,EAAE;IACD,QAAQ,EAAE,MAAM,CAAC;IACjB,MAAM,EAAE,MAAM,EAAE,CAAC;CAClB,UAOA"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../../src/bin/utils/platform/web/index.ts"],"names":[],"mappings":"AAMA,OAAO,EAGL,IAAI,EAEJ,UAAU,EAEX,MAAM,UAAU,CAAC;AAGlB,OAAO,EAAE,QAAQ,EAAE,MAAM,mBAAmB,CAAC;AAE7C,eAAO,MAAM,gCAAgC,eAC/B,UAAU,KACrB,MAgBF,CAAC;AAEF;;;;;;GAMG;AACH,wBAAgB,sBAAsB,CAAC,EACrC,YAAY,EACZ,MAAM,EACN,OAAO,GACR,EAAE;IACD,YAAY,EAAE,MAAM,CAAC;IACrB,MAAM,CAAC,EAAE,MAAM,EAAE,CAAC;IAClB,OAAO,EAAE,MAAM,CAAC;CACjB,GAAG;IACF,SAAS,EAAE,MAAM,GAAG,SAAS,CAAC;IAC9B,QAAQ,EAAE,IAAI,GAAG,SAAS,CAAC;IAC3B,SAAS,EAAE,MAAM,CAAC;CACnB,CA2CA;AAwBD,wBAAsB,0CAA0C,CAC9D,QAAQ,EAAE,MAAM,oBA+BjB;AAED;;;;;;;;;;;;;;;;;;;;;;GAsBG;AACH,wBAAgB,4BAA4B,CAC1C,IAAI,EAAE,IAAI,GAAG,SAAS,GACrB,IAAI,GAAG,SAAS,CA4BlB;AAED,wBAAgB,iBAAiB,CAAC,SAAS,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,MAAM,CAG5E;AAED,wBAAgB,kBAAkB,CAAC,QAAQ,EAAE,MAAM,GAAG,MAAM,EAAE,CA8C7D;AAED,wBAAsB,sBAAsB,CAC1C,OAAO,EAAE,MAAM,EACf,QAAQ,EAAE,MAAM,mCAWjB;AAED,wBAAsB,UAAU,CAAC,QAAQ,EAAE,MAAM,iBAShD;AAED,wBAAsB,UAAU,CAAC,QAAQ,EAAE,MAAM,iBAQhD;AAED,wBAAgB,YAAY,CAAC,QAAQ,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,EAAE,GAAG,EAAE,MAAM,UAE5E;AAED,wBAAsB,cAAc,CAAC,QAAQ,EAAE,MAAM,iBAMpD;AAED,wBAAgB,oBAAoB,CAAC,QAAQ,EAAE,MAAM,UAcpD;AAED,wBAAsB,iCAAiC,CAAC,QAAQ,EAAE,MAAM,+BAoBvE;AAED,wBAAgB,4BAA4B,CAC1C,QAAQ,EAAE,MAAM,EAChB,QAAQ,EAAE,MAAM,EAChB,aAAa,EAAE,MAAM,UA0CtB;AAED,eAAO,MAAM,6BAA6B;qBAKvB,MAAM;iBACV,MAAM;YACX,MAAM,EAAE;YA2DjB,CAAC;AAEF,eAAO,MAAM,iCAAiC,cACjC,MAAM,EAAE,gBACL,MAAM,sBAyBrB,CAAC;AAEF,wBAAsB,qBAAqB,CAAC,EAC1C,YAAY,EACZ,QAAQ,EACR,MAAM,GACP,EAAE;IACD,YAAY,EAAE,MAAM,CAAC;IACrB,QAAQ,EAAE,MAAM,CAAC;IACjB,MAAM,EAAE,MAAM,EAAE,CAAC;CAClB,iBAsDA;AAED,wBAAsB,uBAAuB,CAC3C,QAAQ,EAAE,MAAM,EAChB,cAAc,EAAE,MAAM,EAAE,iBA2BzB;AAED,wBAAgB,aAAa,CAAC,EAC5B,QAAQ,EACR,QAAQ,GACT,EAAE;IACD,QAAQ,EAAE,MAAM,CAAC;IACjB,QAAQ,EAAE,QAAQ,CAAC;CACpB,WAYA;AAED,wBAAgB,mBAAmB,CAAC,EAClC,QAAQ,EACR,MAAM,GACP,EAAE;IACD,QAAQ,EAAE,MAAM,CAAC;IACjB,MAAM,EAAE,MAAM,EAAE,CAAC;CAClB,UAOA;AAED,wBAAgB,+BAA+B,CAAC,UAAU,EAAE,MAAM,GAAG,MAAM,EAAE,CA4B5E"}

package/dist/bin/utils/platform/web/index.js

@@ -3,7 +3,7 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.buildTestNamePrompt = exports.isTestPresent = exports.appendScopeToCreateTest = exports.addUserContextFixture = exports.importAllExportsStmtFromFilePaths = exports.injectCodeSnippetBySuiteChain = exports.replaceCreateTestWithNewCode = exports.getPageVariableNameFromCreateTest = exports.getFixtureImportPath = exports.removeTestOnly = exports.addNewImport = exports.formatCode = exports.lintErrors = exports.stripAndPrependImports = exports.validateTypescript = exports.appendToTestBlock = exports.findFirstSerialDescribeBlock = exports.hasTopLevelDescribeConfigureWithSerialMode = exports.getTypescriptTestBlock = exports.getTestModuleAliasFromSourceFile = void 0;
+exports.getVariableDeclarationsFromCode = exports.buildTestNamePrompt = exports.isTestPresent = exports.appendScopeToCreateTest = exports.addUserContextFixture = exports.importAllExportsStmtFromFilePaths = exports.injectCodeSnippetBySuiteChain = exports.replaceCreateTestWithNewCode = exports.getPageVariableNameFromCreateTest = exports.getFixtureImportPath = exports.removeTestOnly = exports.addNewImport = exports.formatCode = exports.lintErrors = exports.stripAndPrependImports = exports.validateTypescript = exports.appendToTestBlock = exports.findFirstSerialDescribeBlock = exports.hasTopLevelDescribeConfigureWithSerialMode = exports.getTypescriptTestBlock = exports.getTestModuleAliasFromSourceFile = void 0;
 const eslint_1 = require("eslint");
 const fs_1 = require("fs");
 const fs_extra_1 = __importDefault(require("fs-extra"));
@@ -474,3 +474,29 @@ function buildTestNamePrompt({ testName, suites, }) {
         : testName;
 }
 exports.buildTestNamePrompt = buildTestNamePrompt;
+function getVariableDeclarationsFromCode(sourceCode) {
+    const project = new ts_morph_1.Project();
+    const sourceFile = project.createSourceFile("temp.ts", sourceCode);
+    // Extract variable declarations
+    const variables = sourceFile.getDescendantsOfKind(ts_morph_1.SyntaxKind.VariableDeclaration);
+    // Function to extract individual variables from destructured declarations
+    function extractVariables(variable) {
+        const nameNode = variable.getNameNode();
+        if (nameNode.getKind() === ts_morph_1.SyntaxKind.ObjectBindingPattern) {
+            // For object destructuring
+            return nameNode.getElements().map((element) => element.getName());
+        }
+        else if (nameNode.getKind() === ts_morph_1.SyntaxKind.ArrayBindingPattern) {
+            // For array destructuring
+            return nameNode.getElements().map((element) => element.getText());
+        }
+        else {
+            // For regular variable declarations
+            return [variable.getName()];
+        }
+    }
+    // Collect all variable names
+    const allVariables = variables.flatMap((variable) => extractVariables(variable));
+    return allVariables;
+}
+exports.getVariableDeclarationsFromCode = getVariableDeclarationsFromCode;

package/dist/evals/master-agent.evals.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"master-agent.evals.d.ts","sourceRoot":"","sources":["../../src/evals/master-agent.evals.ts"],"names":[],"mappings":"AAKA,OAAO,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAEpC,eAAO,MAAM,4BAA4B,EAAE,UA4C1C,CAAC;AAEF,eAAe,4BAA4B,CAAC"}
+{"version":3,"file":"master-agent.evals.d.ts","sourceRoot":"","sources":["../../src/evals/master-agent.evals.ts"],"names":[],"mappings":"AAMA,OAAO,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAEpC,eAAO,MAAM,4BAA4B,EAAE,UA4C1C,CAAC;AAEF,eAAe,4BAA4B,CAAC"}

package/dist/evals/master-agent.evals.js

@@ -4,11 +4,12 @@ exports.masterGetNextActionEvaluator = void 0;
 const actions_1 = require("../actions");
 const skill_1 = require("../actions/skill");
 const run_1 = require("../agent/master/run");
+const page_1 = require("../page");
 const masterGetNextActionEvaluator = async ({ item, trace, }) => {
     const { task, executedActions, failedActions, pageUrl, options, pageScreenshot, annotatedPageScreenshot, disableSkills, useHints, skills = [], annotations, } = item.input;
     const page = {};
     skill_1.testCaseSkills.updateSkills(skills);
-    const actions = new actions_1.PlaywrightActions(page);
+    const actions = new actions_1.PlaywrightActions(new page_1.TestGenPage(page, "page"));
     const output = await (0, run_1.getNextAction)({
         task,
         executedActions,

package/dist/page/index.d.ts

@@ -0,0 +1,11 @@
+import { Page } from "playwright";
+export declare class TestGenPage {
+    pwPageInstance: Page;
+    name: string;
+    constructor(pwPageInstance: Page, name: string);
+    updatePage({ page, name }: {
+        page: Page;
+        name: string;
+    }): void;
+}
+//# sourceMappingURL=index.d.ts.map

package/dist/page/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/page/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,EAAE,MAAM,YAAY,CAAC;AAElC,qBAAa,WAAW;IAEb,cAAc,EAAE,IAAI;IACpB,IAAI,EAAE,MAAM;gBADZ,cAAc,EAAE,IAAI,EACpB,IAAI,EAAE,MAAM;IAGrB,UAAU,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,EAAE;QAAE,IAAI,EAAE,IAAI,CAAC;QAAC,IAAI,EAAE,MAAM,CAAA;KAAE;CAIxD"}

package/dist/page/index.js

@@ -0,0 +1,16 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.TestGenPage = void 0;
+class TestGenPage {
+    pwPageInstance;
+    name;
+    constructor(pwPageInstance, name) {
+        this.pwPageInstance = pwPageInstance;
+        this.name = name;
+    }
+    updatePage({ page, name }) {
+        this.pwPageInstance = page;
+        this.name = name;
+    }
+}
+exports.TestGenPage = TestGenPage;

package/dist/types/index.d.ts

@@ -1,6 +1,6 @@
 import { LLMModel, LLMProvider, ModelParameters, TraceClient } from "@empiricalrun/llm";
 import OpenAI from "openai";
-import { Page } from "playwright";
+import { TestGenPage } from "../page";
 export type FileContent = {
     filePath: string;
     content: string;
@@ -44,8 +44,9 @@ export type TestCase = {
     suites: string[];
     ai_gist?: string;
 };
-export type PlaywrightActionGenerator = (page: Page, options: {
+export type PlaywrightActionGenerator = (page: TestGenPage, options: {
     stateVariables: Record<string, any>;
+    setStateVariables: (stateVariables: Record<string, any>) => void;
 }) => Action;
 export type ActionSchema = OpenAI.Chat.Completions.ChatCompletionTool;
 export type Action = {

package/dist/types/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/types/index.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,QAAQ,EACR,WAAW,EACX,eAAe,EACf,WAAW,EACZ,MAAM,mBAAmB,CAAC;AAC3B,OAAO,MAAM,MAAM,QAAQ,CAAC;AAC5B,OAAO,EAAE,IAAI,EAAE,MAAM,YAAY,CAAC;AAElC,MAAM,MAAM,WAAW,GAAG;IACxB,QAAQ,EAAE,MAAM,CAAC;IACjB,OAAO,EAAE,MAAM,CAAC;CACjB,CAAC;AAEF,MAAM,MAAM,WAAW,GAAG;IACxB,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;IACb,kBAAkB,EAAE,MAAM,EAAE,CAAC;CAC9B,CAAC;AAEF,MAAM,MAAM,KAAK,GAAG,MAAM,GAAG,QAAQ,GAAG,MAAM,GAAG,MAAM,CAAC;AAExD,MAAM,MAAM,oBAAoB,GAAG;IACjC,KAAK,EAAE,KAAK,CAAC;IACb,KAAK,EAAE,QAAQ,CAAC;IAChB,aAAa,EAAE,WAAW,CAAC;IAC3B,eAAe,CAAC,EAAE,eAAe,CAAC;IAClC,QAAQ,CAAC,EAAE,OAAO,CAAC;IACnB,QAAQ,EAAE;QACR,aAAa,EAAE,MAAM,CAAC;QACtB,YAAY,EAAE,MAAM,CAAC;QACrB,YAAY,EAAE,MAAM,CAAC;QACrB,eAAe,EAAE,MAAM,CAAC;QACxB,WAAW,EAAE,MAAM,CAAC;QACpB,WAAW,EAAE,aAAa,GAAG,YAAY,CAAC;KAC3C,CAAC;CACH,CAAC;AAEF,MAAM,MAAM,KAAK,GAAG;IAClB,GAAG,CAAC,EAAE,MAAM,CAAC;CACd,CAAC;AAEF,MAAM,MAAM,aAAa,GAAG;IAC1B,QAAQ,EAAE,MAAM,CAAC;IACjB,QAAQ,EAAE,QAAQ,CAAC;IACnB,KAAK,CAAC,EAAE,KAAK,CAAC;IACd,WAAW,CAAC,EAAE,WAAW,CAAC;IAC1B,OAAO,CAAC,EAAE,oBAAoB,CAAC;CAChC,CAAC;AAEF,MAAM,MAAM,QAAQ,GAAG;IACrB,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,EAAE,MAAM,EAAE,CAAC;IAChB,QAAQ,EAAE,MAAM,CAAC;IACjB,MAAM,EAAE,MAAM,EAAE,CAAC;IACjB,OAAO,CAAC,EAAE,MAAM,CAAC;CAClB,CAAC;AAEF,MAAM,MAAM,yBAAyB,GAAG,CACtC,IAAI,EAAE,IAAI,EACV,OAAO,EAAE;IACP,cAAc,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;CACrC,KACE,MAAM,CAAC;AAEZ,MAAM,MAAM,YAAY,GAAG,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,kBAAkB,CAAC;AAEtE,MAAM,MAAM,MAAM,GAAG;IACnB,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,EAAE,YAAY,CAAC;IACrB,OAAO,EAAE,CAAC,OAAO,EAAE;QACjB,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;QAC1B,KAAK,CAAC,EAAE,WAAW,CAAC;KACrB,KAAK,OAAO,CAAC;QAAE,OAAO,EAAE,MAAM,CAAA;KAAE,GAAG,IAAI,CAAC,CAAC;IAC1C,QAAQ,EAAE,CACR,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,EACzB,OAAO,EAAE;QAAE,OAAO,EAAE,MAAM,CAAA;KAAE,KACzB;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,UAAU,CAAC,EAAE,MAAM,CAAA;KAAE,CAAC;CAC5C,CAAC;AAEF,MAAM,MAAM,SAAS,GAAG,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/types/index.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,QAAQ,EACR,WAAW,EACX,eAAe,EACf,WAAW,EACZ,MAAM,mBAAmB,CAAC;AAC3B,OAAO,MAAM,MAAM,QAAQ,CAAC;AAE5B,OAAO,EAAE,WAAW,EAAE,MAAM,SAAS,CAAC;AAEtC,MAAM,MAAM,WAAW,GAAG;IACxB,QAAQ,EAAE,MAAM,CAAC;IACjB,OAAO,EAAE,MAAM,CAAC;CACjB,CAAC;AAEF,MAAM,MAAM,WAAW,GAAG;IACxB,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;IACb,kBAAkB,EAAE,MAAM,EAAE,CAAC;CAC9B,CAAC;AAEF,MAAM,MAAM,KAAK,GAAG,MAAM,GAAG,QAAQ,GAAG,MAAM,GAAG,MAAM,CAAC;AAExD,MAAM,MAAM,oBAAoB,GAAG;IACjC,KAAK,EAAE,KAAK,CAAC;IACb,KAAK,EAAE,QAAQ,CAAC;IAChB,aAAa,EAAE,WAAW,CAAC;IAC3B,eAAe,CAAC,EAAE,eAAe,CAAC;IAClC,QAAQ,CAAC,EAAE,OAAO,CAAC;IACnB,QAAQ,EAAE;QACR,aAAa,EAAE,MAAM,CAAC;QACtB,YAAY,EAAE,MAAM,CAAC;QACrB,YAAY,EAAE,MAAM,CAAC;QACrB,eAAe,EAAE,MAAM,CAAC;QACxB,WAAW,EAAE,MAAM,CAAC;QACpB,WAAW,EAAE,aAAa,GAAG,YAAY,CAAC;KAC3C,CAAC;CACH,CAAC;AAEF,MAAM,MAAM,KAAK,GAAG;IAClB,GAAG,CAAC,EAAE,MAAM,CAAC;CACd,CAAC;AAEF,MAAM,MAAM,aAAa,GAAG;IAC1B,QAAQ,EAAE,MAAM,CAAC;IACjB,QAAQ,EAAE,QAAQ,CAAC;IACnB,KAAK,CAAC,EAAE,KAAK,CAAC;IACd,WAAW,CAAC,EAAE,WAAW,CAAC;IAC1B,OAAO,CAAC,EAAE,oBAAoB,CAAC;CAChC,CAAC;AAEF,MAAM,MAAM,QAAQ,GAAG;IACrB,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,EAAE,MAAM,EAAE,CAAC;IAChB,QAAQ,EAAE,MAAM,CAAC;IACjB,MAAM,EAAE,MAAM,EAAE,CAAC;IACjB,OAAO,CAAC,EAAE,MAAM,CAAC;CAClB,CAAC;AAEF,MAAM,MAAM,yBAAyB,GAAG,CACtC,IAAI,EAAE,WAAW,EACjB,OAAO,EAAE;IACP,cAAc,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IACpC,iBAAiB,EAAE,CAAC,cAAc,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,KAAK,IAAI,CAAC;CAClE,KACE,MAAM,CAAC;AAEZ,MAAM,MAAM,YAAY,GAAG,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,kBAAkB,CAAC;AAEtE,MAAM,MAAM,MAAM,GAAG;IACnB,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,EAAE,YAAY,CAAC;IACrB,OAAO,EAAE,CAAC,OAAO,EAAE;QACjB,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;QAC1B,KAAK,CAAC,EAAE,WAAW,CAAC;KACrB,KAAK,OAAO,CAAC;QAAE,OAAO,EAAE,MAAM,CAAA;KAAE,GAAG,IAAI,CAAC,CAAC;IAC1C,QAAQ,EAAE,CACR,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,EACzB,OAAO,EAAE;QAAE,OAAO,EAAE,MAAM,CAAA;KAAE,KACzB;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,UAAU,CAAC,EAAE,MAAM,CAAA;KAAE,CAAC;CAC5C,CAAC;AAEF,MAAM,MAAM,SAAS,GAAG,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@empiricalrun/test-gen",
-  "version": "0.38.2",
+  "version": "0.38.5",
   "publishConfig": {
     "registry": "https://registry.npmjs.org/",
     "access": "public"