Implement DownloadToS3Block (#133)

This commit is contained in:
Kerem Yilmaz
2024-03-28 16:46:54 -07:00
committed by GitHub
parent 57062952b8
commit 3d1b146470
6 changed files with 128 additions and 3 deletions

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}")