support CUA navigation block (#2306)
Some checks are pending
Run tests and pre-commit / test (push) Waiting to run
Run tests and pre-commit / fe-lint-build (push) Waiting to run
Publish Fern Docs / run (push) Waiting to run
Sync to skyvern-cloud / sync (push) Waiting to run

This commit is contained in:
Shuchang Zheng 2025-05-07 13:04:02 -07:00 committed by GitHub
parent 945e5b1922
commit 869a94dbf4
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 6 additions and 1 deletions

View file

@ -71,6 +71,7 @@ from skyvern.forge.sdk.workflow.models.parameter import (
OutputParameter, OutputParameter,
WorkflowParameter, WorkflowParameter,
) )
from skyvern.schemas.runs import RunEngine
from skyvern.utils.url_validators import prepend_scheme_and_validate_url from skyvern.utils.url_validators import prepend_scheme_and_validate_url
from skyvern.webeye.browser_factory import BrowserState from skyvern.webeye.browser_factory import BrowserState
from skyvern.webeye.utils.page import SkyvernFrame from skyvern.webeye.utils.page import SkyvernFrame
@ -349,6 +350,7 @@ class BaseTaskBlock(Block):
task_type: str = TaskType.general task_type: str = TaskType.general
url: str | None = None url: str | None = None
title: str = "" title: str = ""
engine: RunEngine = RunEngine.skyvern_v1
complete_criterion: str | None = None complete_criterion: str | None = None
terminate_criterion: str | None = None terminate_criterion: str | None = None
navigation_goal: str | None = None navigation_goal: str | None = None
@ -623,6 +625,7 @@ class BaseTaskBlock(Block):
browser_session_id=browser_session_id, browser_session_id=browser_session_id,
close_browser_on_completion=browser_session_id is None, close_browser_on_completion=browser_session_id is None,
complete_verification=self.complete_verification, complete_verification=self.complete_verification,
engine=self.engine,
) )
except Exception as e: except Exception as e:
# Make sure the task is marked as failed in the database before raising the exception # Make sure the task is marked as failed in the database before raising the exception

View file

@ -8,7 +8,7 @@ from skyvern.forge.sdk.workflow.models.block import BlockType, FileType
from skyvern.forge.sdk.workflow.models.constants import FileStorageType from skyvern.forge.sdk.workflow.models.constants import FileStorageType
from skyvern.forge.sdk.workflow.models.parameter import ParameterType, WorkflowParameterType from skyvern.forge.sdk.workflow.models.parameter import ParameterType, WorkflowParameterType
from skyvern.forge.sdk.workflow.models.workflow import WorkflowStatus from skyvern.forge.sdk.workflow.models.workflow import WorkflowStatus
from skyvern.schemas.runs import ProxyLocation from skyvern.schemas.runs import ProxyLocation, RunEngine
class ParameterYAML(BaseModel, abc.ABC): class ParameterYAML(BaseModel, abc.ABC):
@ -275,6 +275,7 @@ class NavigationBlockYAML(BlockYAML):
navigation_goal: str navigation_goal: str
url: str | None = None url: str | None = None
title: str = "" title: str = ""
engine: RunEngine = RunEngine.skyvern_v1
error_code_mapping: dict[str, str] | None = None error_code_mapping: dict[str, str] | None = None
max_retries: int = 0 max_retries: int = 0
max_steps_per_run: int | None = None max_steps_per_run: int | None = None

View file

@ -1790,6 +1790,7 @@ class WorkflowService:
label=block_yaml.label, label=block_yaml.label,
url=block_yaml.url, url=block_yaml.url,
title=block_yaml.title, title=block_yaml.title,
engine=block_yaml.engine,
parameters=navigation_block_parameters, parameters=navigation_block_parameters,
output_parameter=output_parameter, output_parameter=output_parameter,
navigation_goal=block_yaml.navigation_goal, navigation_goal=block_yaml.navigation_goal,