vodorod/Dorod-Sky
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

fix upload file storage class input style (#2618)

Browse Source
This commit is contained in:
Shuchang Zheng
2025-06-06 01:36:46 -07:00
committed by GitHub
parent 25cea85be0
commit 3a927aa4e2
1 changed files with 8 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
skyvern/forge/sdk/api/aws.py
View File

@@ -105,7 +105,10 @@ class AsyncAWSClient:
async with self.session.client(AWSClientType.S3, region_name=self.region_name) as client: async with self.session.client(AWSClientType.S3, region_name=self.region_name) as client:
parsed_uri = S3Uri(uri) parsed_uri = S3Uri(uri)
await client.upload_fileobj( await client.upload_fileobj(
file_obj, parsed_uri.bucket, parsed_uri.key, StorageClass=str(storage_class) file_obj,
parsed_uri.bucket,
parsed_uri.key,
ExtraArgs={"StorageClass": str(storage_class)},
) )
LOG.debug("Upload file stream success", uri=uri) LOG.debug("Upload file stream success", uri=uri)
return uri return uri
@@ -124,13 +127,14 @@ class AsyncAWSClient:
try: try:
async with self.session.client(AWSClientType.S3, region_name=self.region_name) as client: async with self.session.client(AWSClientType.S3, region_name=self.region_name) as client:
parsed_uri = S3Uri(uri) parsed_uri = S3Uri(uri)
extra_args: dict[str, Any] = {"ExtraArgs": {"Metadata": metadata}} if metadata else {} extra_args: dict[str, Any] = {"StorageClass": str(storage_class)}
if metadata:
extra_args["Metadata"] = metadata
await client.upload_file( await client.upload_file(
Filename=file_path, Filename=file_path,
Bucket=parsed_uri.bucket, Bucket=parsed_uri.bucket,
Key=parsed_uri.key, Key=parsed_uri.key,
StorageClass=str(storage_class), ExtraArgs=extra_args,
**extra_args,
) )
except Exception as e: except Exception as e:
LOG.exception("S3 upload failed.", uri=uri) LOG.exception("S3 upload failed.", uri=uri)