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

fix totp code bug (#4210)

Browse Source
This commit is contained in:
LawyZheng
2025-12-05 21:57:14 +08:00
committed by GitHub
parent e11414f988
commit 0cf52486de
1 changed files with 17 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
skyvern/forge/agent.py
View File

@@ -2993,6 +2993,23 @@ class ForgeAgent:
final_navigation_payload = (
final_navigation_payload + "\n" + str({SPECIAL_FIELD_VERIFICATION_CODE: verification_code})
)
elif isinstance(final_navigation_payload, list):
verification_code_dict = str({SPECIAL_FIELD_VERIFICATION_CODE: verification_code})
if verification_code_dict not in final_navigation_payload:
final_navigation_payload.append(verification_code_dict)
else:
LOG.warning(
"Verification code already exists in navigation payload",
final_navigation_payload=final_navigation_payload,
)
elif final_navigation_payload is None:
final_navigation_payload = {SPECIAL_FIELD_VERIFICATION_CODE: verification_code}
else:
LOG.warning(
"Didn't add verification code to navigation payload",
final_navigation_payload=final_navigation_payload,
)
if expire_verification_code:
current_context.totp_codes.pop(task.task_id)