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

add download block (#1266)

Browse source
This commit is contained in:
LawyZheng 2024-11-26 23:36:22 +08:00 • committed by GitHub
parent fcc83f6fe4
commit 7bc6ec8e46
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 48 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

6
skyvern/forge/sdk/workflow/models/block.py
View file

@ -77,6 +77,7 @@ class BlockType(StrEnum):
EXTRACTION = "extraction"
LOGIN = "login"
WAIT = "wait"
FILE_DOWNLOAD = "file_download"
class BlockStatus(StrEnum):
@ -1348,6 +1349,10 @@ class LoginBlock(BaseTaskBlock):
block_type: Literal[BlockType.LOGIN] = BlockType.LOGIN
class FileDownloadBlock(BaseTaskBlock):
block_type: Literal[BlockType.FILE_DOWNLOAD] = BlockType.FILE_DOWNLOAD
BlockSubclasses = Union[
ForLoopBlock,
TaskBlock,
@ -1363,5 +1368,6 @@ BlockSubclasses = Union[
ExtractionBlock,
LoginBlock,
WaitBlock,
FileDownloadBlock,
]
BlockTypeVar = Annotated[BlockSubclasses, Field(discriminator="block_type")]