vikarti.anatra/Skyvern
1
0
Fork 0
mirror of https://github.com/Skyvern-AI/skyvern.git synced 2025-09-10 06:15:18 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

Implement FileURLParserBlock and FILE_URL WorkflowParameterType (#559)

Browse source
This commit is contained in:
Kerem Yilmaz 2024-07-05 17:08:20 -07:00 • committed by GitHub
parent 8be94d7928
commit 6929a1d24d
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
7 changed files with 135 additions and 19 deletions
Download patch file Download diff file Expand all files Collapse all files

8
skyvern/forge/sdk/api/files.py
View file

@ -9,10 +9,18 @@ import structlog
from skyvern.constants import REPO_ROOT_DIR
from skyvern.exceptions import DownloadFileMaxSizeExceeded
from skyvern.forge.sdk.api.aws import AsyncAWSClient
LOG = structlog.get_logger()
async def download_from_s3(client: AsyncAWSClient, s3_uri: str) -> str:
downloaded_bytes = await client.download_file(uri=s3_uri)
file_path = tempfile.NamedTemporaryFile(delete=False)
file_path.write(downloaded_bytes)
return file_path.name
async def download_file(url: str, max_size_mb: int | None = None) -> str:
try:
async with aiohttp.ClientSession(raise_for_status=True) as session: