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

UrlBlock - a block to navigate to a url (#1461)

Browse Source
This commit is contained in:
Shuchang Zheng
2025-01-02 09:26:21 -08:00
committed by GitHub
parent 401d304dd0
commit b5d6e4184e
2 changed files with 13 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -83,6 +83,7 @@ class BlockType(StrEnum):
LOGIN = "login" LOGIN = "login"
WAIT = "wait" WAIT = "wait"
FILE_DOWNLOAD = "file_download" FILE_DOWNLOAD = "file_download"
GOTO_URL = "goto_url"
class BlockStatus(StrEnum): class BlockStatus(StrEnum):
@@ -1801,6 +1802,11 @@ class FileDownloadBlock(BaseTaskBlock):
block_type: Literal[BlockType.FILE_DOWNLOAD] = BlockType.FILE_DOWNLOAD block_type: Literal[BlockType.FILE_DOWNLOAD] = BlockType.FILE_DOWNLOAD
class UrlBlock(BaseTaskBlock):
block_type: Literal[BlockType.GOTO_URL] = BlockType.GOTO_URL
url: str
BlockSubclasses = Union[ BlockSubclasses = Union[
ForLoopBlock, ForLoopBlock,
TaskBlock, TaskBlock,
@@ -1817,5 +1823,6 @@ BlockSubclasses = Union[
LoginBlock, LoginBlock,
WaitBlock, WaitBlock,
FileDownloadBlock, FileDownloadBlock,
UrlBlock,
] ]
BlockTypeVar = Annotated[BlockSubclasses, Field(discriminator="block_type")] BlockTypeVar = Annotated[BlockSubclasses, Field(discriminator="block_type")]

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

@@ -310,6 +310,11 @@ class FileDownloadBlockYAML(BlockYAML):
cache_actions: bool = False cache_actions: bool = False
class UrlBlockYAML(BlockYAML):
block_type: Literal[BlockType.GOTO_URL] = BlockType.GOTO_URL # type: ignore
url: str
PARAMETER_YAML_SUBCLASSES = ( PARAMETER_YAML_SUBCLASSES = (
AWSSecretParameterYAML AWSSecretParameterYAML
| BitwardenLoginCredentialParameterYAML | BitwardenLoginCredentialParameterYAML
@@ -337,6 +342,7 @@ BLOCK_YAML_SUBCLASSES = (
| LoginBlockYAML | LoginBlockYAML
| WaitBlockYAML | WaitBlockYAML
| FileDownloadBlockYAML | FileDownloadBlockYAML
| UrlBlockYAML
) )
BLOCK_YAML_TYPES = Annotated[BLOCK_YAML_SUBCLASSES, Field(discriminator="block_type")] BLOCK_YAML_TYPES = Annotated[BLOCK_YAML_SUBCLASSES, Field(discriminator="block_type")]