vikarti.anatra/Skyvern
1
0
Fork 0
mirror of https://github.com/Skyvern-AI/skyvern.git synced 2025-09-15 17:59:42 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

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
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
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")]