From ee5f98f4518874b20b9b85c2f8f94738f2663411 Mon Sep 17 00:00:00 2001
From: Shuchang Zheng <wintonzheng0325@gmail.com>
Date: Thu, 15 May 2025 11:23:33 -0700
Subject: [PATCH] fix multi level selection issue (#2350)

---
 skyvern/webeye/actions/handler.py | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/skyvern/webeye/actions/handler.py b/skyvern/webeye/actions/handler.py
index 522f3d11..a96ac5b1 100644
--- a/skyvern/webeye/actions/handler.py
+++ b/skyvern/webeye/actions/handler.py
@@ -2286,6 +2286,7 @@ async def sequentially_select_from_dropdown(
     dropdown_menu_element: SkyvernElement | None = None,
     force_select: bool = False,
     target_value: str = "",
+    continue_until_click: bool = False,
 ) -> CustomSingleSelectResult | None:
     """
     TODO: support to return all values retrieved from the sequentially select
@@ -2411,6 +2412,14 @@ async def sequentially_select_from_dropdown(
             )
             continue
 
+        if continue_until_click:
+            LOG.info(
+                "Continue the selecting until the dropdown menu is closed",
+                step_id=step.step_id,
+                task_id=task.task_id,
+            )
+            continue
+
         screenshot = await page.screenshot(timeout=settings.BROWSER_SCREENSHOT_TIMEOUT_MS)
         mini_goal = (
             input_or_select_context.field