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

add max_steps_per_run to task (#297)

Browse Source 
Co-authored-by: Shuchang Zheng <wintonzheng0325@gmail.com>
This commit is contained in:
Kerem Yilmaz
2024-05-11 14:13:21 -07:00
committed by GitHub
parent 6feddbde6a
commit 270642c60c
8 changed files with 39 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
skyvern/forge/sdk/workflow/models/block.py
View File

@@ -111,6 +111,7 @@ class TaskBlock(Block):
# error code to error description for the LLM
error_code_mapping: dict[str, str] | None = None
max_retries: int = 0
max_steps_per_run: int | None = None
parameters: list[PARAMETER_TYPE] = []
def get_all_parameters(

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

@@ -84,6 +84,7 @@ class TaskBlockYAML(BlockYAML):
data_schema: dict[str, Any] | None = None
error_code_mapping: dict[str, str] | None = None
max_retries: int = 0
max_steps_per_run: int | None = None
parameter_keys: list[str] | None = None