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

add engine to more blocks (#2457)

Browse Source
This commit is contained in:
Shuchang Zheng
2025-05-26 08:19:42 -07:00
committed by GitHub
parent 9d32249d4c
commit 2438bcacbf
19 changed files with 185 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
skyvern/forge/sdk/workflow/models/yaml.py
View File

@@ -128,6 +128,7 @@ class TaskBlockYAML(BlockYAML):
url: str | None = None
title: str = ""
engine: RunEngine = RunEngine.skyvern_v1
navigation_goal: str | None = None
data_extraction_goal: str | None = None
data_schema: dict[str, Any] | list | None = None
@@ -259,6 +260,7 @@ class ActionBlockYAML(BlockYAML):
url: str | None = None
title: str = ""
engine: RunEngine = RunEngine.skyvern_v1
navigation_goal: str | None = None
error_code_mapping: dict[str, str] | None = None
max_retries: int = 0
@@ -298,6 +300,7 @@ class ExtractionBlockYAML(BlockYAML):
data_extraction_goal: str
url: str | None = None
title: str = ""
engine: RunEngine = RunEngine.skyvern_v1
data_schema: dict[str, Any] | list | None = None
max_retries: int = 0
max_steps_per_run: int | None = None
@@ -310,6 +313,7 @@ class LoginBlockYAML(BlockYAML):
url: str | None = None
title: str = ""
engine: RunEngine = RunEngine.skyvern_v1
navigation_goal: str | None = None
error_code_mapping: dict[str, str] | None = None
max_retries: int = 0
@@ -334,6 +338,7 @@ class FileDownloadBlockYAML(BlockYAML):
navigation_goal: str
url: str | None = None
title: str = ""
engine: RunEngine = RunEngine.skyvern_v1
error_code_mapping: dict[str, str] | None = None
max_retries: int = 0
max_steps_per_run: int | None = None

5
skyvern/forge/sdk/workflow/service.py
View File

@@ -1642,6 +1642,7 @@ class WorkflowService:
label=block_yaml.label,
url=block_yaml.url,
title=block_yaml.title,
engine=block_yaml.engine,
parameters=task_block_parameters,
output_parameter=output_parameter,
navigation_goal=block_yaml.navigation_goal,
@@ -1812,6 +1813,7 @@ class WorkflowService:
label=block_yaml.label,
url=block_yaml.url,
title=block_yaml.title,
engine=block_yaml.engine,
task_type=TaskType.action,
parameters=action_block_parameters,
output_parameter=output_parameter,
@@ -1868,6 +1870,7 @@ class WorkflowService:
label=block_yaml.label,
url=block_yaml.url,
title=block_yaml.title,
engine=block_yaml.engine,
parameters=extraction_block_parameters,
output_parameter=output_parameter,
data_extraction_goal=block_yaml.data_extraction_goal,
@@ -1889,6 +1892,7 @@ class WorkflowService:
label=block_yaml.label,
url=block_yaml.url,
title=block_yaml.title,
engine=block_yaml.engine,
parameters=login_block_parameters,
output_parameter=output_parameter,
navigation_goal=block_yaml.navigation_goal,
@@ -1925,6 +1929,7 @@ class WorkflowService:
label=block_yaml.label,
url=block_yaml.url,
title=block_yaml.title,
engine=block_yaml.engine,
parameters=file_download_block_parameters,
output_parameter=output_parameter,
navigation_goal=block_yaml.navigation_goal,