From 012aec0cd5ded393165630ddbbd1f5f4b1cdc048 Mon Sep 17 00:00:00 2001 From: Shuchang Zheng Date: Fri, 12 Sep 2025 13:01:17 -0700 Subject: [PATCH] Support skyvern.upload_file (#3424) --- skyvern/__init__.py | 2 ++ skyvern/services/script_service.py | 2 +- 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/skyvern/__init__.py b/skyvern/__init__.py index c94a11fa..75f22af0 100644 --- a/skyvern/__init__.py +++ b/skyvern/__init__.py @@ -44,6 +44,7 @@ from skyvern.services.script_service import ( # noqa: E402 run_script, # noqa: E402 run_task, # noqa: E402 send_email, # noqa: E402 + upload_file, # noqa: E402 wait, # noqa: E402 ) # noqa: E402 @@ -68,6 +69,7 @@ __all__ = [ "run_task", "send_email", "setup", + "upload_file", "wait", "workflow", ] diff --git a/skyvern/services/script_service.py b/skyvern/services/script_service.py index 2013e335..67833f33 100644 --- a/skyvern/services/script_service.py +++ b/skyvern/services/script_service.py @@ -1479,7 +1479,7 @@ async def upload_file( label=block_validation_output.label, output_parameter=block_validation_output.output_parameter, parameters=parameters or [], - storage_type=storage_type, + storage_type=FileStorageType(storage_type), s3_bucket=s3_bucket, aws_access_key_id=aws_access_key_id, aws_secret_access_key=aws_secret_access_key,