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

CUA feature - verification code / 2FA (#2174)

Browse Source
This commit is contained in:
Shuchang Zheng
2025-04-16 17:11:01 -07:00
committed by GitHub
parent 16fe1954c1
commit a9c8a1d889
5 changed files with 86 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

35
skyvern/webeye/actions/parse_actions.py
View File

@@ -27,8 +27,10 @@ from skyvern.webeye.actions.actions import (
SolveCaptchaAction,
TerminateAction,
UploadFileAction,
VerificationCodeAction,
WaitAction,
)
from skyvern.webeye.actions.handler import poll_verification_code
from skyvern.webeye.scraper.scraper import ScrapedPage
LOG = structlog.get_logger()
@@ -400,12 +402,33 @@ async def parse_cua_actions(
intention=reasoning,
)
elif skyvern_action_type == "get_verification_code":
# Currently we don't support verification code
# TODO: handle verification code by fetching the code and send it to CUA
action = TerminateAction(
reasoning=reasoning,
intention=reasoning,
)
if (task.totp_verification_url or task.totp_identifier) and task.organization_id:
LOG.info(
"Getting verification code for CUA",
task_id=task.task_id,
organization_id=task.organization_id,
workflow_run_id=task.workflow_run_id,
totp_verification_url=task.totp_verification_url,
totp_identifier=task.totp_identifier,
)
verification_code = await poll_verification_code(
task.task_id,
task.organization_id,
workflow_run_id=task.workflow_run_id,
totp_verification_url=task.totp_verification_url,
totp_identifier=task.totp_identifier,
)
reasoning = reasoning or f"Received verification code: {verification_code}"
action = VerificationCodeAction(
verification_code=verification_code,
reasoning=reasoning,
intention=reasoning,
)
else:
action = TerminateAction(
reasoning=reasoning,
intention=reasoning,
)
action.organization_id = task.organization_id
action.workflow_run_id = task.workflow_run_id