vodorod/Dorod-Sky
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

fix custom selection issue (#1766)

Browse Source
This commit is contained in:
Shuchang Zheng
2025-02-13 21:11:14 +08:00
committed by GitHub
parent 6a5fd7aaaa
commit b411af56a6
1 changed files with 14 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
skyvern/webeye/actions/handler.py
View File

@@ -102,6 +102,7 @@ class CustomSingleSelectResult:
def __init__(self, skyvern_frame: SkyvernFrame) -> None: def __init__(self, skyvern_frame: SkyvernFrame) -> None:
self.reasoning: str | None = None self.reasoning: str | None = None
self.action_result: ActionResult | None = None self.action_result: ActionResult | None = None
self.action_type: ActionType | None = None
self.value: str | None = None self.value: str | None = None
self.dropdown_menu: SkyvernElement | None = None self.dropdown_menu: SkyvernElement | None = None
self.skyvern_frame = skyvern_frame self.skyvern_frame = skyvern_frame
@@ -2028,6 +2029,15 @@ async def sequentially_select_from_dropdown(
# it's for typing. it's been verified in `single_select_result.is_done()` # it's for typing. it's been verified in `single_select_result.is_done()`
assert single_select_result.dropdown_menu is not None assert single_select_result.dropdown_menu is not None
if single_select_result.action_type is not None and single_select_result.action_type == ActionType.INPUT_TEXT:
LOG.info(
"It's an input mini action, going to continue the select action",
step_id=step.step_id,
task_id=task.task_id,
)
continue
screenshot = await page.screenshot(timeout=settings.BROWSER_SCREENSHOT_TIMEOUT_MS) screenshot = await page.screenshot(timeout=settings.BROWSER_SCREENSHOT_TIMEOUT_MS)
mini_goal = ( mini_goal = (
input_or_select_context.field input_or_select_context.field
@@ -2161,9 +2171,10 @@ async def select_from_dropdown(
) )
action_type: str = json_response.get("action_type", "") action_type: str = json_response.get("action_type", "")
action_type = action_type.upper() action_type = action_type.lower()
single_select_result.action_type = ActionType(action_type)
element_id: str | None = json_response.get("id", None) element_id: str | None = json_response.get("id", None)
if not element_id or action_type not in ["CLICK", "INPUT_TEXT"]: if not element_id or action_type not in [ActionType.CLICK, ActionType.INPUT_TEXT]:
raise NoAvailableOptionFoundForCustomSelection(reason=json_response.get("reasoning")) raise NoAvailableOptionFoundForCustomSelection(reason=json_response.get("reasoning"))
if not force_select and target_value: if not force_select and target_value:
@@ -2176,7 +2187,7 @@ async def select_from_dropdown(
) )
return single_select_result return single_select_result
if value is not None and action_type == "INPUT_TEXT": if value is not None and action_type == ActionType.INPUT_TEXT:
LOG.info( LOG.info(
"No clickable option found, but found input element to search", "No clickable option found, but found input element to search",
element_id=element_id, element_id=element_id,