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

Implement DownloadToS3Block (#133)

Browse source
This commit is contained in:
Kerem Yilmaz 2024-03-28 16:46:54 -07:00 • committed by GitHub
parent 57062952b8
commit 3d1b146470
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
6 changed files with 128 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -25,6 +25,7 @@ from skyvern.forge.sdk.workflow.models.block import (
BlockType,
BlockTypeVar,
CodeBlock,
DownloadToS3Block,
ForLoopBlock,
TaskBlock,
TextPromptBlock,
@ -732,4 +733,10 @@ class WorkflowService:
json_schema=block_yaml.json_schema,
output_parameter=output_parameter,
)
elif block_yaml.block_type == BlockType.DOWNLOAD_TO_S3:
return DownloadToS3Block(
label=block_yaml.label,
output_parameter=output_parameter,
url=block_yaml.url,
)
raise ValueError(f"Invalid block type {block_yaml.block_type}")