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

send workflow_run_id to totp endpoint (#2120)

Browse Source
This commit is contained in:
Shuchang Zheng
2025-04-09 13:51:16 -07:00
committed by GitHub
parent e3251edb05
commit 7d5ab76d73
2 changed files with 7 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
skyvern/forge/agent.py
View File

@@ -2297,6 +2297,7 @@ class ForgeAgent:
verification_code = await poll_verification_code( verification_code = await poll_verification_code(
task.task_id, task.task_id,
task.organization_id, task.organization_id,
workflow_run_id=task.workflow_run_id,
totp_verification_url=task.totp_verification_url, totp_verification_url=task.totp_verification_url,
totp_identifier=task.totp_identifier, totp_identifier=task.totp_identifier,
) )

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

@@ -2887,7 +2887,12 @@ async def poll_verification_code(
) )
verification_code = None verification_code = None
if totp_verification_url: if totp_verification_url:
verification_code = await _get_verification_code_from_url(task_id, totp_verification_url, org_token.token) verification_code = await _get_verification_code_from_url(
task_id,
totp_verification_url,
org_token.token,
workflow_run_id=workflow_run_id,
)
elif totp_identifier: elif totp_identifier:
verification_code = await _get_verification_code_from_db( verification_code = await _get_verification_code_from_db(
task_id, organization_id, totp_identifier, workflow_id=workflow_id task_id, organization_id, totp_identifier, workflow_id=workflow_id