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

fix the problem of verification_code not injected into string navigation goal (#603)

Browse Source
This commit is contained in:
Kerem Yilmaz
2024-07-12 15:20:59 -07:00
committed by GitHub
parent cdd731b862
commit 33ab70dd8b
1 changed files with 14 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
skyvern/forge/agent.py
View File

@@ -1009,9 +1009,21 @@ class ForgeAgent:
task: Task, task: Task,
) -> dict[str, Any] | list | str | None: ) -> dict[str, Any] | list | str | None:
final_navigation_payload = task.navigation_payload final_navigation_payload = task.navigation_payload
if isinstance(final_navigation_payload, dict) and task.totp_verification_url: if task.totp_verification_url:
if SPECIAL_FIELD_VERIFICATION_CODE not in final_navigation_payload: if (
isinstance(final_navigation_payload, dict)
and SPECIAL_FIELD_VERIFICATION_CODE not in final_navigation_payload
):
final_navigation_payload[SPECIAL_FIELD_VERIFICATION_CODE] = VERIFICATION_CODE_PLACEHOLDER final_navigation_payload[SPECIAL_FIELD_VERIFICATION_CODE] = VERIFICATION_CODE_PLACEHOLDER
elif (
isinstance(final_navigation_payload, str)
and SPECIAL_FIELD_VERIFICATION_CODE not in final_navigation_payload
):
final_navigation_payload = (
final_navigation_payload
+ "\n"
+ str({SPECIAL_FIELD_VERIFICATION_CODE: VERIFICATION_CODE_PLACEHOLDER})
)
return final_navigation_payload return final_navigation_payload
async def _get_action_results(self, task: Task) -> str: async def _get_action_results(self, task: Task) -> str: