@empiricalrun/test-gen 0.38.4 → 0.38.6

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
@@ -182,17 +202,17 @@ async function createTestUsingMasterAgent({ task, page, testCase, options, scope
         const pageScreenshot = buffer.toString("base64");
         let output;
         let generatedCodeSteps = [];
-        let annotations;
         let annotatedPageScreenshot;
+        let annotationKeys = [];
         if (useHints) {
             await page.waitForTimeout(2000);
-            const annotationResult = await page.evaluate(() => {
+            annotationKeys = await page.evaluate(() => {
                 // @ts-ignore
-                window.annotationInstance = window.annotateClickableElements();
+                // eslint-disable-next-line no-undef
+                window.annotationInstance = annotateClickableElements();
                 // @ts-ignore
-                return window.annotationInstance;
+                return Object.keys(window.annotationInstance.annotations);
             });
-            annotations = annotationResult?.annotations || {};
             await page.waitForTimeout(2000);
             const annonationBuffer = await page.screenshot({
             //This is done to improve element annotation accuracy, anyways it doesn't annotate elements which are out of viewport
@@ -216,7 +236,6 @@ async function createTestUsingMasterAgent({ task, page, testCase, options, scope
         if (await (0, session_1.shouldStopSession)()) {
             break;
         }
-        const annotationKeys = annotations ? Object.keys(annotations) : [];
         const toolCall = await getNextAction({
             task,
             executedActions: masterAgentActions,
@@ -255,13 +274,13 @@ async function createTestUsingMasterAgent({ task, page, testCase, options, scope
                         name: "trigger-hints-flow",
                         input: {
                             outputFromGetNextAction: output,
-                            generatedAnnotations: annotations,
+                            generatedAnnotations: annotationKeys,
                         },
                     });
                     const result = await (0, with_hints_1.triggerHintsFlow)({
                         outputFromGetNextAction: output,
-                        generatedAnnotations: annotations,
-                        page,
+                        generatedAnnotations: annotationKeys,
+                        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,CAuGA,CAAC"}

package/dist/agent/master/with-hints.js

@@ -37,12 +37,12 @@ const triggerHintsFlow = async ({ outputFromGetNextAction, generatedAnnotations,
     try {
         const hasElementAnnotation = outputFromGetNextAction?.elementAnnotation?.length &&
             outputFromGetNextAction?.elementAnnotation?.trim()?.length &&
-            outputFromGetNextAction?.elementAnnotation in
-                (generatedAnnotations || {});
+            generatedAnnotations?.includes(outputFromGetNextAction?.elementAnnotation);
         trace?.event({
             name: "has-element-annotation",
             output: {
                 hasElementAnnotation,
+                generatedAnnotations,
             },
         });
         if (!hasElementAnnotation) {

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;