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

trigger TerminateAction when totp code is not found (#1768)

Browse Source
This commit is contained in:
Shuchang Zheng
2025-02-14 01:50:38 +08:00
committed by GitHub
parent 4407c19417
commit 1759f06f65
3 changed files with 52 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -47,6 +47,7 @@ from skyvern.exceptions import (
NoIncrementalElementFoundForAutoCompletion,
NoIncrementalElementFoundForCustomSelection,
NoSuitableAutoCompleteOption,
NoTOTPVerificationCodeFound,
OptionIndexOutOfBound,
WrongElementToUploadFile,
)
@@ -2847,7 +2848,12 @@ async def poll_verification_code(
# check timeout
if datetime.utcnow() > timeout_datetime:
LOG.warning("Polling verification code timed out", workflow_id=workflow_id)
return None
raise NoTOTPVerificationCodeFound(
task_id=task_id,
workflow_run_id=workflow_run_id,
totp_verification_url=totp_verification_url,
totp_identifier=totp_identifier,
)
verification_code = None
if totp_verification_url:
verification_code = await _get_verification_code_from_url(task_id, totp_verification_url, org_token.token)