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

parse yaml criterion to block (#1383)

Browse Source
This commit is contained in:
LawyZheng
2024-12-13 16:54:48 +08:00
committed by GitHub
parent 01c28e9a4d
commit 31c19938ff
2 changed files with 12 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -131,6 +131,8 @@ class TaskBlockYAML(BlockYAML):
totp_verification_url: str | None = None totp_verification_url: str | None = None
totp_identifier: str | None = None totp_identifier: str | None = None
cache_actions: bool = False cache_actions: bool = False
complete_criterion: str | None = None
terminate_criterion: str | None = None
class ForLoopBlockYAML(BlockYAML): class ForLoopBlockYAML(BlockYAML):
@@ -253,6 +255,8 @@ class NavigationBlockYAML(BlockYAML):
totp_verification_url: str | None = None totp_verification_url: str | None = None
totp_identifier: str | None = None totp_identifier: str | None = None
cache_actions: bool = False cache_actions: bool = False
complete_criterion: str | None = None
terminate_criterion: str | None = None
class ExtractionBlockYAML(BlockYAML): class ExtractionBlockYAML(BlockYAML):
@@ -281,6 +285,8 @@ class LoginBlockYAML(BlockYAML):
totp_verification_url: str | None = None totp_verification_url: str | None = None
totp_identifier: str | None = None totp_identifier: str | None = None
cache_actions: bool = False cache_actions: bool = False
complete_criterion: str | None = None
terminate_criterion: str | None = None
class WaitBlockYAML(BlockYAML): class WaitBlockYAML(BlockYAML):

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

@@ -1313,6 +1313,8 @@ class WorkflowService:
totp_verification_url=block_yaml.totp_verification_url, totp_verification_url=block_yaml.totp_verification_url,
totp_identifier=block_yaml.totp_identifier, totp_identifier=block_yaml.totp_identifier,
cache_actions=block_yaml.cache_actions, cache_actions=block_yaml.cache_actions,
complete_criterion=block_yaml.complete_criterion,
terminate_criterion=block_yaml.terminate_criterion,
) )
elif block_yaml.block_type == BlockType.FOR_LOOP: elif block_yaml.block_type == BlockType.FOR_LOOP:
loop_blocks = [ loop_blocks = [
@@ -1481,6 +1483,8 @@ class WorkflowService:
totp_verification_url=block_yaml.totp_verification_url, totp_verification_url=block_yaml.totp_verification_url,
totp_identifier=block_yaml.totp_identifier, totp_identifier=block_yaml.totp_identifier,
cache_actions=block_yaml.cache_actions, cache_actions=block_yaml.cache_actions,
complete_criterion=block_yaml.complete_criterion,
terminate_criterion=block_yaml.terminate_criterion,
) )
elif block_yaml.block_type == BlockType.EXTRACTION: elif block_yaml.block_type == BlockType.EXTRACTION:
@@ -1523,6 +1527,8 @@ class WorkflowService:
totp_verification_url=block_yaml.totp_verification_url, totp_verification_url=block_yaml.totp_verification_url,
totp_identifier=block_yaml.totp_identifier, totp_identifier=block_yaml.totp_identifier,
cache_actions=block_yaml.cache_actions, cache_actions=block_yaml.cache_actions,
complete_criterion=block_yaml.complete_criterion,
terminate_criterion=block_yaml.terminate_criterion,
) )
elif block_yaml.block_type == BlockType.WAIT: elif block_yaml.block_type == BlockType.WAIT: