Add PDF Parser Block (#1600)

This commit is contained in:
Shuchang Zheng 2025-01-20 12:33:54 -08:00 committed by GitHub
parent 8f3941b6b5
commit 1796af6df9
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
6 changed files with 143 additions and 2 deletions

View file

@ -46,6 +46,7 @@ from skyvern.forge.sdk.workflow.models.block import (
ForLoopBlock,
LoginBlock,
NavigationBlock,
PDFParserBlock,
SendEmailBlock,
TaskBlock,
TextPromptBlock,
@ -1469,6 +1470,14 @@ class WorkflowService:
file_type=block_yaml.file_type,
continue_on_failure=block_yaml.continue_on_failure,
)
elif block_yaml.block_type == BlockType.PDF_PARSER:
return PDFParserBlock(
label=block_yaml.label,
output_parameter=output_parameter,
file_url=block_yaml.file_url,
json_schema=block_yaml.json_schema,
continue_on_failure=block_yaml.continue_on_failure,
)
elif block_yaml.block_type == BlockType.VALIDATION:
validation_block_parameters = (
[parameters[parameter_key] for parameter_key in block_yaml.parameter_keys]