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

totp + proxy location support in observer (#1549)

Browse Source
This commit is contained in:
Shuchang Zheng
2025-01-14 09:06:44 -08:00
committed by GitHub
parent 0392763998
commit a0df333518
1 changed files with 9 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
skyvern/forge/sdk/services/observer_service.py
View File

@@ -437,6 +437,8 @@ async def run_observer_cruise_helper(
workflow_run_id=workflow_run_id,
original_url=original_url,
navigation_goal=navigation_goal,
totp_verification_url=observer_cruise.totp_verification_url,
totp_identifier=observer_cruise.totp_identifier,
)
task_history_record = {"type": task_type, "task": plan}
elif task_type == "loop":
@@ -498,7 +500,7 @@ async def run_observer_cruise_helper(
workflow_create_request = WorkflowCreateYAMLRequest(
title=workflow.title,
description=workflow.description,
proxy_location=ProxyLocation.RESIDENTIAL,
proxy_location=observer_cruise.proxy_location or ProxyLocation.RESIDENTIAL,
workflow_definition=workflow_definition_yaml,
)
LOG.info("Creating workflow from request", workflow_create_request=workflow_create_request)
@@ -994,6 +996,8 @@ async def _generate_navigation_task(
workflow_run_id: str,
navigation_goal: str,
original_url: str | None = None,
totp_verification_url: str | None = None,
totp_identifier: str | None = None,
) -> tuple[NavigationBlock, list[BLOCK_YAML_TYPES], list[PARAMETER_YAML_TYPES]]:
LOG.info("Generating navigation task", navigation_goal=navigation_goal, original_url=original_url)
label = f"navigation_{_generate_random_string()}"
@@ -1001,6 +1005,8 @@ async def _generate_navigation_task(
label=label,
url=original_url,
navigation_goal=navigation_goal,
totp_verification_url=totp_verification_url,
totp_identifier=totp_identifier,
)
output_parameter = await app.WORKFLOW_SERVICE.create_output_parameter_for_block(
workflow_id=workflow_id,
@@ -1011,6 +1017,8 @@ async def _generate_navigation_task(
label=label,
url=original_url,
navigation_goal=navigation_goal,
totp_verification_url=totp_verification_url,
totp_identifier=totp_identifier,
output_parameter=output_parameter,
),
[navigation_block_yaml],