From 39710ca59e3d33131ad5d81f7bf2388d2ecd705b Mon Sep 17 00:00:00 2001 From: Asher Foa Date: Tue, 17 Jun 2025 10:59:55 -0400 Subject: [PATCH] Simplify code, move some code out of the loop (#2685) --- skyvern/forge/sdk/artifact/storage/s3.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/skyvern/forge/sdk/artifact/storage/s3.py b/skyvern/forge/sdk/artifact/storage/s3.py index 840f1ba1..6175ffa0 100644 --- a/skyvern/forge/sdk/artifact/storage/s3.py +++ b/skyvern/forge/sdk/artifact/storage/s3.py @@ -186,13 +186,12 @@ class S3Storage(BaseStorage): files = os.listdir(download_dir) sc = await self._get_storage_class_for_org(organization_id) tags = await self._get_tags_for_org(organization_id) + base_uri = f"s3://{settings.AWS_S3_BUCKET_UPLOADS}/{DOWNLOAD_FILE_PREFIX}/{settings.ENV}/{organization_id}/{workflow_run_id or task_id}" for file in files: fpath = os.path.join(download_dir, file) if not os.path.isfile(fpath): continue - uri = f"s3://{settings.AWS_S3_BUCKET_UPLOADS}/{DOWNLOAD_FILE_PREFIX}/{settings.ENV}/{organization_id}/{workflow_run_id or task_id}/{file}" - - # Calculate SHA-256 checksum + uri = f"{base_uri}/{file}" checksum = calculate_sha256_for_file(fpath) LOG.info( "Calculated checksum for file",