Implement FileURLParserBlock and FILE_URL WorkflowParameterType (#559)

This commit is contained in:
Kerem Yilmaz
2024-07-05 17:08:20 -07:00
committed by GitHub
parent 8be94d7928
commit 6929a1d24d
7 changed files with 135 additions and 19 deletions

View File

@@ -25,6 +25,7 @@ from skyvern.forge.sdk.workflow.models.block import (
BlockTypeVar,
CodeBlock,
DownloadToS3Block,
FileParserBlock,
ForLoopBlock,
SendEmailBlock,
TaskBlock,
@@ -1052,4 +1053,12 @@ class WorkflowService:
file_attachments=block_yaml.file_attachments or [],
continue_on_failure=block_yaml.continue_on_failure,
)
elif block_yaml.block_type == BlockType.FILE_URL_PARSER:
return FileParserBlock(
label=block_yaml.label,
output_parameter=output_parameter,
file_url=block_yaml.file_url,
file_type=block_yaml.file_type,
continue_on_failure=block_yaml.continue_on_failure,
)
raise ValueError(f"Invalid block type {block_yaml.block_type}")