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

update need verification prompt (#2207)

Browse Source 
Co-authored-by: lawyzheng <lawyzheng1106@gmail.com>
This commit is contained in:
Shuchang Zheng
2025-04-22 00:19:35 +08:00
committed by GitHub
parent f2eef5aa6d
commit f9f3940fbd
3 changed files with 10 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
skyvern/forge/agent.py
View File

@@ -2520,8 +2520,12 @@ class ForgeAgent:
browser_state: BrowserState,
json_response: dict[str, Any],
) -> dict[str, Any]:
need_verification_code = json_response.get("need_verification_code")
if need_verification_code and (task.totp_verification_url or task.totp_identifier) and task.organization_id:
place_to_enter_verification_code = json_response.get("place_to_enter_verification_code")
if (
place_to_enter_verification_code
and (task.totp_verification_url or task.totp_identifier)
and task.organization_id
):
LOG.info("Need verification code", step_id=step.step_id)
verification_code = await poll_verification_code(
task.task_id,