parse yaml criterion to block (#1383)

Skyvern-AI · Dec 13, 2024 · 31c1993 · 31c1993
1 parent 01c28e9
commit 31c1993
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/skyvern/forge/sdk/workflow/models/yaml.py b/skyvern/forge/sdk/workflow/models/yaml.py
@@ -131,6 +131,8 @@ class TaskBlockYAML(BlockYAML):
     totp_verification_url: str | None = None
     totp_identifier: str | None = None
     cache_actions: bool = False
+    complete_criterion: str | None = None
+    terminate_criterion: str | None = None
 
 
 class ForLoopBlockYAML(BlockYAML):
@@ -253,6 +255,8 @@ class NavigationBlockYAML(BlockYAML):
     totp_verification_url: str | None = None
     totp_identifier: str | None = None
     cache_actions: bool = False
+    complete_criterion: str | None = None
+    terminate_criterion: str | None = None
 
 
 class ExtractionBlockYAML(BlockYAML):
@@ -281,6 +285,8 @@ class LoginBlockYAML(BlockYAML):
     totp_verification_url: str | None = None
     totp_identifier: str | None = None
     cache_actions: bool = False
+    complete_criterion: str | None = None
+    terminate_criterion: str | None = None
 
 
 class WaitBlockYAML(BlockYAML):

diff --git a/skyvern/forge/sdk/workflow/service.py b/skyvern/forge/sdk/workflow/service.py
@@ -1313,6 +1313,8 @@ async def block_yaml_to_block(
                 totp_verification_url=block_yaml.totp_verification_url,
                 totp_identifier=block_yaml.totp_identifier,
                 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:
             loop_blocks = [
@@ -1481,6 +1483,8 @@ async def block_yaml_to_block(
                 totp_verification_url=block_yaml.totp_verification_url,
                 totp_identifier=block_yaml.totp_identifier,
                 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:
@@ -1523,6 +1527,8 @@ async def block_yaml_to_block(
                 totp_verification_url=block_yaml.totp_verification_url,
                 totp_identifier=block_yaml.totp_identifier,
                 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: