mirror of
https://github.com/Skyvern-AI/skyvern.git
synced 2025-09-14 17:29:40 +00:00
support CUA navigation block (#2306)
This commit is contained in:
parent
945e5b1922
commit
869a94dbf4
3 changed files with 6 additions and 1 deletions
|
@ -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
|
||||||
|
|
|
@ -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
|
||||||
|
|
|
@ -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,
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue