vikarti.anatra/Skyvern
1
0
Fork 0
mirror of https://github.com/Skyvern-AI/skyvern.git synced 2025-09-09 13:56:08 +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

25
skyvern/forge/sdk/workflow/service.py
View file

@ -38,6 +38,7 @@ from skyvern.forge.sdk.workflow.models.block import (
CodeBlock,
DownloadToS3Block,
ExtractionBlock,
FileDownloadBlock,
FileParserBlock,
ForLoopBlock,
LoginBlock,
@ -1474,4 +1475,28 @@ class WorkflowService:
output_parameter=output_parameter,
)
elif block_yaml.block_type == BlockType.FILE_DOWNLOAD:
file_download_block_parameters = (
[parameters[parameter_key] for parameter_key in block_yaml.parameter_keys]
if block_yaml.parameter_keys
else []
)
return FileDownloadBlock(
label=block_yaml.label,
url=block_yaml.url,
title=block_yaml.title,
parameters=file_download_block_parameters,
output_parameter=output_parameter,
navigation_goal=block_yaml.navigation_goal,
error_code_mapping=block_yaml.error_code_mapping,
max_steps_per_run=block_yaml.max_steps_per_run,
max_retries=block_yaml.max_retries,
download_suffix=block_yaml.download_suffix,
continue_on_failure=block_yaml.continue_on_failure,
totp_verification_url=block_yaml.totp_verification_url,
totp_identifier=block_yaml.totp_identifier,
cache_actions=block_yaml.cache_actions,
complete_on_download=True,
)
raise ValueError(f"Invalid block type {block_yaml.block_type}")